Security-57031.1.35.tar.gz

[apple/security.git] / 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 (file)
index 0000000..400dcf7
--- /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