--- /dev/null
+/*
+ * Copyright (c) 2000-2001,2011-2012,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.
+ */
+
+
+//
+// MultiDLDb implementation.
+//
+
+#include <security_cdsa_client/multidldb.h>
+#include <security_cdsa_client/securestorage.h>
+
+
+
+namespace Security
+{
+
+using namespace CssmClient;
+
+namespace CssmClient
+{
+
+//
+// MultiDLDbDbCursorImpl declaration
+//
+class MultiDLDbDbCursorImpl : public DbCursorImpl
+{
+public:
+ MultiDLDbDbCursorImpl(const MultiDLDb &parent, const CSSM_QUERY &query, Allocator &allocator);
+ MultiDLDbDbCursorImpl(const MultiDLDb &parent, uint32 capacity, Allocator &allocator);
+ virtual ~MultiDLDbDbCursorImpl();
+
+ bool next(DbAttributes *attributes, ::CssmDataContainer *data, DbUniqueRecord &uniqueId);
+private:
+ MultiDLDb multiDLDb() { return parent<MultiDLDb>(); }
+ void activate();
+ void deactivate();
+
+ MultiDLDbImpl::ListRef mListRef;
+ MultiDLDbImpl::List::const_iterator mNext;
+ MultiDLDbImpl::List::const_iterator mEnd;
+ DbCursor mCursor;
+};
+
+} // end namespace CssmClient
+
+} // end namespace Security
+
+//
+// MultiDLDbImpl
+//
+MultiDLDbImpl::MultiDLDbImpl(const vector<DLDbIdentifier> &list, bool useSecureStorage, const Cssm &cssm)
+: ObjectImpl(cssm), mListRef(list), mUseSecureStorage(useSecureStorage)
+{
+}
+
+MultiDLDbImpl::MultiDLDbImpl(const vector<DLDbIdentifier> &list, bool useSecureStorage)
+: ObjectImpl(Cssm::standard()), mListRef(list), mUseSecureStorage(useSecureStorage)
+{
+}
+
+MultiDLDbImpl::~MultiDLDbImpl()
+{
+ deactivate();
+}
+
+Db
+MultiDLDbImpl::database(const DLDbIdentifier &dlDbIdentifier)
+{
+ StLock<Mutex> _(mLock);
+ DbMap::const_iterator it = mDbMap.find(dlDbIdentifier);
+ if (it != mDbMap.end())
+ return it->second;
+
+ Module module(dlDbIdentifier.ssuid().guid(), cssm());
+ DL dl;
+ if (dlDbIdentifier.ssuid().subserviceType() & CSSM_SERVICE_CSP)
+ {
+ if (mUseSecureStorage)
+ dl = SSCSPDL(module);
+ else
+ dl = CSPDL(module);
+ }
+ else
+ dl = DL(module);
+
+ dl->subserviceId(dlDbIdentifier.ssuid().subserviceId());
+ dl->version(dlDbIdentifier.ssuid().version());
+ Db db(dl, dlDbIdentifier.dbName());
+ if (find(mListRef->begin(), mListRef->end(), dlDbIdentifier) != mListRef->end())
+ mDbMap.insert(DbMap::value_type(dlDbIdentifier, db));
+
+ return db;
+}
+
+void
+MultiDLDbImpl::list(const vector<DLDbIdentifier> &list)
+{
+ StLock<Mutex> _(mLock);
+ set<DLDbIdentifier> oldList(mListRef->begin(), mListRef->end());
+ mListRef = ListRef(list);
+ set<DLDbIdentifier> newList(mListRef->begin(), mListRef->end());
+ vector<DLDbIdentifier> obsolete;
+ back_insert_iterator<vector<DLDbIdentifier> > ii(obsolete);
+ // Remove all db's from the map that were in oldList but are not in mListRef.
+ set_difference(oldList.begin(), oldList.end(), newList.begin(), newList.end(), ii);
+ for (vector<DLDbIdentifier>::const_iterator it = obsolete.begin(); it != obsolete.end(); ++it)
+ mDbMap.erase(*it);
+}
+
+DbCursorImpl *
+MultiDLDbImpl::newDbCursor(const CSSM_QUERY &query, Allocator &allocator)
+{
+ return new MultiDLDbDbCursorImpl(MultiDLDb(this), query, allocator);
+}
+
+DbCursorImpl *
+MultiDLDbImpl::newDbCursor(uint32 capacity, Allocator &allocator)
+{
+ return new MultiDLDbDbCursorImpl(MultiDLDb(this), capacity, allocator);
+}
+
+void
+MultiDLDbImpl::activate()
+{
+}
+
+void
+MultiDLDbImpl::deactivate()
+{
+ StLock<Mutex> _(mLock);
+ mDbMap.erase(mDbMap.begin(), mDbMap.end());
+}
+
+
+//
+// MultiDLDbDbCursorImpl
+//
+MultiDLDbDbCursorImpl::MultiDLDbDbCursorImpl(const MultiDLDb &parent,
+ const CSSM_QUERY &query, Allocator &allocator)
+: DbCursorImpl(parent, query, allocator)
+{
+}
+
+MultiDLDbDbCursorImpl::MultiDLDbDbCursorImpl(const MultiDLDb &parent,
+ uint32 capacity, Allocator &allocator)
+: DbCursorImpl(parent, capacity, allocator)
+{
+}
+
+MultiDLDbDbCursorImpl::~MultiDLDbDbCursorImpl()
+{
+ try
+ {
+ deactivate();
+ }
+ catch(...) {}
+}
+
+bool
+MultiDLDbDbCursorImpl::next(DbAttributes *attributes, ::CssmDataContainer *data, DbUniqueRecord &uniqueId)
+{
+ activate();
+ for (;;)
+ {
+ if (!mCursor)
+ {
+ if (mNext == mEnd)
+ {
+ // This is how it ends.
+ deactivate();
+ return false;
+ }
+
+ mCursor = DbCursor(multiDLDb()->database(*mNext++), *this);
+ }
+
+ try
+ {
+ if (mCursor->next(attributes, data, uniqueId))
+ return true;
+ }
+
+ catch(const CommonError &err)
+ {
+ OSStatus status = err.osStatus();
+ if(status != CSSMERR_DL_DATASTORE_DOESNOT_EXIST)
+ throw;
+ }
+
+
+
+ mCursor = DbCursor();
+ }
+}
+
+void
+MultiDLDbDbCursorImpl::activate()
+{
+ StLock<Mutex> _(mActivateMutex);
+ if (!mActive)
+ {
+ mListRef = multiDLDb()->listRef();
+ mNext = mListRef->begin();
+ mEnd = mListRef->end();
+ mActive = true;
+ }
+}
+
+void
+MultiDLDbDbCursorImpl::deactivate()
+{
+ StLock<Mutex> _(mActivateMutex);
+ if (mActive)
+ {
+ mActive = false;
+ mListRef = MultiDLDbImpl::ListRef();
+ mNext = mEnd;
+ mCursor = DbCursor();
+ }
+}
+
projects / apple / security.git / blobdiff