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

diff --git a/Security/libsecurity_apple_x509_cl/lib/LockedMap.h b/Security/libsecurity_apple_x509_cl/lib/LockedMap.h
new file mode 100644
index 00000000..2386a755
--- /dev/null
+++ b/Security/libsecurity_apple_x509_cl/lib/LockedMap.h
@@ -0,0 +1,92 @@
+/*
+ * Copyright (c) 2000-2001,2011,2014 Apple Inc. All Rights Reserved.
+ * 
+ * The contents of this file constitute Original Code as defined in and are
+ * subject to the Apple Public Source License Version 1.2 (the 'License').
+ * You may not use this file except in compliance with the License. Please obtain
+ * a copy of the License at http://www.apple.com/publicsource and read it before
+ * using this file.
+ * 
+ * This 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.
+ */
+
+
+/*
+ * LockedMap.h - STL-style map with attached Mutex
+ *
+ * Copyright (c) 2000,2011,2014 Apple Inc. 
+ */
+ 
+#ifndef	_LOCKED_MAP_H_
+#define _LOCKED_MAP_H_
+
+#include <map>
+#include <security_utilities/threading.h>
+
+template <class KeyType, class ValueType>
+class LockedMap
+{
+private:
+	typedef std::map<KeyType, ValueType *> MapType;
+	MapType					mMap;
+	Mutex					mMapLock;
+	
+	/* low-level lookup, cacheMapLock held on entry and exit */
+	ValueType 				
+	*lookupEntryLocked(KeyType key) 
+		{
+			// don't create new entry if desired entry isn't there
+			typename MapType::iterator it = mMap.find(key);
+			if(it == mMap.end()) {
+				return NULL;
+			}
+			return it->second;
+		}
+
+public:
+	/* high level maintenance */
+	void 
+	addEntry(ValueType &value, KeyType key)
+		{
+			StLock<Mutex> _(mMapLock);
+			mMap[key] = &value;
+		}
+		
+	ValueType				
+	*lookupEntry(KeyType key)
+		{
+			StLock<Mutex> _(mMapLock);
+			return lookupEntryLocked(key);
+		}
+		
+	void	
+	removeEntry(KeyType key)
+		{
+			StLock<Mutex> _(mMapLock);
+
+			ValueType *value = lookupEntryLocked(key);
+			if(value != NULL) {
+				mMap.erase(key);
+			}
+		}
+		
+	ValueType	
+	*removeFirstEntry()
+		{
+			StLock<Mutex> _(mMapLock);
+			typename MapType::iterator it = mMap.begin();
+			if(it == mMap.end()) {
+				return NULL;
+			}
+			ValueType *rtn = it->second;
+			mMap.erase(it->first);
+			return rtn;
+		}
+};
+
+#endif	/* _LOCKED_MAP_H_ */