projects / apple / security.git / blame
| Commit | Line | Data |
|---|---|---|
| b1ab9ed8 | 1 | /* |
| d8f41ccd | 2 | * Copyright (c) 2000-2004,2011,2014 Apple Inc. All Rights Reserved. |
| b1ab9ed8 A |
3 | * |
| 4 | * The contents of this file constitute Original Code as defined in and are | |
| 5 | * subject to the Apple Public Source License Version 1.2 (the 'License'). | |
| 6 | * You may not use this file except in compliance with the License. Please obtain | |
| 7 | * a copy of the License at http://www.apple.com/publicsource and read it before | |
| 8 | * using this file. | |
| 9 | * | |
| 10 | * This Original Code and all software distributed under the License are | |
| 11 | * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER EXPRESS | |
| 12 | * OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES, INCLUDING WITHOUT | |
| 13 | * LIMITATION, ANY WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR | |
| 14 | * PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT. Please see the License for the | |
| 15 | * specific language governing rights and limitations under the License. | |
| 16 | */ | |
| 17 | ||
| 18 | ||
| 19 | // | |
| 20 | // mdsclient - friendly interface to CDSA MDS API | |
| 21 | // | |
| 22 | #include <security_cdsa_client/mdsclient.h> | |
| 23 | ||
| 24 | ||
| 25 | namespace Security { | |
| 26 | namespace CssmClient { | |
| 27 | ||
| 28 | ||
| 29 | // | |
| 30 | // DLAccess gets a virtual destructor just in case | |
| 31 | // | |
| 32 | DLAccess::~DLAccess() | |
| 33 | { } | |
| 34 | ||
| 35 | ||
| 36 | // | |
| 37 | // Basic Record objects (abstract) | |
| 38 | // | |
| 39 | Record::Record(const char * const * names) | |
| 40 | : CssmAutoData(Allocator::standard(Allocator::sensitive)) | |
| 41 | { | |
| 42 | addAttributes(names); | |
| 43 | } | |
| 44 | ||
| 45 | void Record::addAttributes(const char * const * name) | |
| 46 | { | |
| 47 | if (name) | |
| 48 | while (*name) | |
| 49 | mAttributes.add(CssmDbAttributeInfo(*name++)); | |
| 50 | } | |
| 51 | ||
| 52 | Record::~Record() | |
| 53 | { } | |
| 54 | ||
| 55 | ||
| 56 | // | |
| 57 | // Tables and their components (non-template common features) | |
| 58 | // | |
| 59 | TableBase::TableBase(DLAccess &source, CSSM_DB_RECORDTYPE type, bool getData /* = true */) | |
| 60 | : database(source), mRecordType(type), mGetData(getData) | |
| 61 | { | |
| 62 | } | |
| 63 | ||
| 64 | TableBase::Handle::~Handle() | |
| 65 | { | |
| 66 | if (query) | |
| 67 | mAccess->dlAbortQuery(query); | |
| 68 | } | |
| 69 | ||
| 70 | TableBase::Uid::~Uid() | |
| 71 | { | |
| 72 | if (uid) | |
| 73 | mAccess->dlFreeUniqueId(uid); | |
| 74 | } | |
| 75 | ||
| 76 | TableBase::Iterator::Iterator(DLAccess *ac, CSSM_HANDLE query, | |
| 77 | CSSM_DB_UNIQUE_RECORD *id, Record *record, bool getData) | |
| 78 | : mAccess(ac), mQuery(new Handle(ac, query)), | |
| 79 | mUid(new Uid(ac, id)), mRecord(record), mGetData(getData) | |
| 80 | { } | |
| 81 | ||
| 82 | ||
| 83 | void TableBase::Iterator::advance(Record *fill) | |
| 84 | { | |
| 85 | RefPointer<Record> newRecord = fill; // hold it safely | |
| 86 | CSSM_DB_UNIQUE_RECORD *id; | |
| 87 | CssmAutoData data(mAccess->allocator()); | |
| 88 | if (mAccess->dlGetNext(mQuery->query, newRecord->attributes(), | |
| 89 | mGetData ? &data.get() : NULL, id)) { | |
| 90 | if (mGetData) | |
| 91 | newRecord->recordData() = data; | |
| 92 | mUid = new Uid(mAccess, id); | |
| 93 | mRecord = newRecord; | |
| 94 | } else { | |
| 95 | mQuery->query = CSSM_INVALID_HANDLE; // was automatically aborted | |
| 96 | // release all iterator resources and make me == end() | |
| 97 | mQuery = NULL; | |
| 98 | mUid = NULL; | |
| 99 | mRecord = NULL; | |
| 100 | } | |
| 101 | } | |
| 102 | ||
| 103 | ||
| 104 | uint32 TableBase::erase(const CSSM_QUERY &query) | |
| 105 | { | |
| 106 | CSSM_DB_UNIQUE_RECORD *id; | |
| 107 | CssmDbRecordAttributeData noAttributes; | |
| 108 | CSSM_HANDLE handle = database.dlGetFirst(query, noAttributes, NULL, id); | |
| 109 | if (handle == CSSM_INVALID_HANDLE) | |
| 110 | return 0; // no match, nothing erased | |
| 111 | uint32 count = 0; | |
| 112 | do { | |
| 113 | database.dlDeleteRecord(id); | |
| 114 | count++; | |
| 115 | database.dlFreeUniqueId(id); | |
| 116 | } while (database.dlGetNext(handle, noAttributes, NULL, id)); | |
| 117 | return count; | |
| 118 | } | |
| 119 | ||
| 120 | uint32 TableBase::erase(const Query &query) | |
| 121 | { | |
| 122 | return erase(query.cssmQuery()); | |
| 123 | } | |
| 124 | ||
| 125 | ||
| 126 | } // end namespace CssmClient | |
| 127 | } // end namespace Security |