X-Git-Url: https://git.saurik.com/apple/security.git/blobdiff_plain/80e2389990082500d76eb566d4946be3e786c3ef..d8f41ccd20de16f8ebe2ccc84d47bf1cb2b26bbb:/Security/libsecurity_utilities/lib/coderepository.h

diff --git a/Security/libsecurity_utilities/lib/coderepository.h b/Security/libsecurity_utilities/lib/coderepository.h
new file mode 100644
index 00000000..400dcf73
--- /dev/null
+++ b/Security/libsecurity_utilities/lib/coderepository.h
@@ -0,0 +1,110 @@
+/*
+ * Copyright (c) 2004,2011,2014 Apple Inc. All Rights Reserved.
+ * 
+ * @APPLE_LICENSE_HEADER_START@
+ * 
+ * This file contains Original Code and/or Modifications of Original Code
+ * as defined in and that are subject to the Apple Public Source License
+ * Version 2.0 (the 'License'). You may not use this file except in
+ * compliance with the License. Please obtain a copy of the License at
+ * http://www.opensource.apple.com/apsl/ and read it before using this
+ * file.
+ * 
+ * The Original Code and all software distributed under the License are
+ * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
+ * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
+ * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
+ * Please see the License for the specific language governing rights and
+ * limitations under the License.
+ * 
+ * @APPLE_LICENSE_HEADER_END@
+ */
+
+
+//
+// bundlerepository - directory search paths for bundles
+//
+#ifndef _H_BUNDLEREPOSITORY
+#define _H_BUNDLEREPOSITORY
+
+#include <security_utilities/cfutilities.h>
+#include <security_utilities/refcount.h>
+#include <CoreFoundation/CoreFoundation.h>
+#include <string>
+#include <vector>
+
+
+namespace Security {
+
+
+//
+// PathList abstracts a directory search path.
+// It's not really powerful enough to be useful on its own.
+//
+class PathList {
+public:
+	PathList();
+	PathList(const string &subPath, const char *suffix = NULL,
+		const char *envar = NULL, bool forUser = true);
+	virtual ~PathList();
+	
+	void addDirectory(const string &dirPath);
+
+protected:
+	vector<string> mPaths;
+	string mSuffix;
+	IFDEBUG(string mDebugOverride);
+};
+
+
+//
+// CodeRepository<Code> represents all code objects within the PathList search path,
+// represented forcibly as objects of type Code.
+//
+template <class Code>
+class CodeRepository : public vector<RefPointer<Code> >, public PathList {
+public:
+	CodeRepository() { }				// empty - populate with paths
+	CodeRepository(const string &subPath, const char *suffix = NULL,
+		const char *envar = NULL, bool forUser = true)
+		: PathList(subPath, suffix, envar, forUser) { }
+	
+	void update();
+};
+
+
+//
+// The generic implementation of update works with subclasses of GenericBundle,
+// represented through CFBundleRefs collected via CFBundle.
+// (Technically, this would work with anything that has a constructor from CFBundleRef.)
+// If we ever wanted a CodeRepository<ExecutableTool>, we'd specialize update() to deal with
+// ExecutableTool's slightly different constructor.
+//
+template <class Code>
+void CodeRepository<Code>::update()
+{
+	vector<RefPointer<Code> > result;
+	for (vector<string>::const_iterator it = mPaths.begin(); it != mPaths.end(); it++) {
+		if (CFRef<CFArrayRef> bundles = CFBundleCreateBundlesFromDirectory(NULL,
+				CFTempURL(*it, true), mSuffix.empty() ? NULL : CFStringRef(CFTempString(mSuffix)))) {
+			CFIndex count = CFArrayGetCount(bundles);
+			secdebug("coderep", "%p directory %s has %ld entries", this, it->c_str(), count);
+			for (CFIndex n = 0; n < count; n++)
+				try {
+					result.push_back(new Code((CFBundleRef)CFArrayGetValueAtIndex(bundles, n)));
+				} catch (...) {
+					secdebug("coderep", "%p exception creating %s (skipped)",
+						this, cfString(CFBundleRef(CFArrayGetValueAtIndex(bundles, n))).c_str());
+				}
+		} else
+			secdebug("coderep", "directory %s bundle read failed", it->c_str());
+	}
+	secdebug("coderep", "%p total of %ld items in list", this, result.size());
+	this->swap(result);
+}
+
+
+} // end namespace Security
+
+#endif //_H_BUNDLEREPOSITORY