1/* 2 * Copyright (c) 2000-2001,2004,2011-2012,2014 Apple Inc. All Rights Reserved. 3 * 4 * @APPLE_LICENSE_HEADER_START@ 5 * 6 * This file contains Original Code and/or Modifications of Original Code 7 * as defined in and that are subject to the Apple Public Source License 8 * Version 2.0 (the 'License'). You may not use this file except in 9 * compliance with the License. Please obtain a copy of the License at 10 * http://www.opensource.apple.com/apsl/ and read it before using this 11 * file. 12 * 13 * The Original Code and all software distributed under the License are 14 * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER 15 * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES, 16 * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY, 17 * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT. 18 * Please see the License for the specific language governing rights and 19 * limitations under the License. 20 * 21 * @APPLE_LICENSE_HEADER_END@ 22 */ 23 24 25// 26// PrimaryKey.cpp 27// 28 29#include "PrimaryKey.h" 30 31using namespace KeychainCore; 32using namespace CssmClient; 33 34 35PrimaryKeyImpl::PrimaryKeyImpl(const CSSM_DATA &data) 36: CssmDataContainer(data.Data, data.Length), mMutex(Mutex::recursive) 37{ 38 39//@@@ do bounds checking here, throw if invalid 40 41} 42 43PrimaryKeyImpl::PrimaryKeyImpl(const DbAttributes &primaryKeyAttrs) : mMutex(Mutex::recursive) 44{ 45 Length = sizeof(uint32); 46 for (uint32 ix = 0; ix < primaryKeyAttrs.size(); ++ix) 47 { 48 if (primaryKeyAttrs.at(ix).size() == 0) 49 MacOSError::throwMe(errSecInvalidKeychain); 50 51 Length += sizeof(uint32) + primaryKeyAttrs.at(ix).Value[0].Length; 52 } 53 54 // Careful with exceptions 55 Data = mAllocator.alloc<uint8>((UInt32)Length); 56 uint8 *p = Data; 57 58 putUInt32(p, primaryKeyAttrs.recordType()); 59 for (uint32 ix = 0; ix < primaryKeyAttrs.size(); ++ix) 60 { 61 UInt32 len = (UInt32)primaryKeyAttrs.at(ix).Value[0].Length; 62 putUInt32(p, len); 63 memcpy(p, primaryKeyAttrs.at(ix).Value[0].Data, len); 64 p += len; 65 } 66} 67 68CssmClient::DbCursor 69PrimaryKeyImpl::createCursor(const Keychain &keychain) 70{ 71 StLock<Mutex>_(mMutex); 72 DbCursor cursor(keychain->database()); 73 74 // @@@ Set up cursor to find item with this. 75 uint8 *p = Data; 76 uint32 left = (uint32)Length; 77 if (left < sizeof(*p)) 78 MacOSError::throwMe(errSecNoSuchAttr); // XXX Not really but whatever. 79 80 CSSM_DB_RECORDTYPE rt = getUInt32(p, left); 81 const CssmAutoDbRecordAttributeInfo &infos = keychain->primaryKeyInfosFor(rt); 82 83 cursor->recordType(rt); 84 cursor->conjunctive(CSSM_DB_AND); 85 for (uint32 ix = 0; ix < infos.size(); ++ix) 86 { 87 uint32 len = getUInt32(p, left); 88 89 if (left < len) 90 MacOSError::throwMe(errSecNoSuchAttr); // XXX Not really but whatever. 91 92 CssmData value(p, len); 93 left -= len; 94 p += len; 95 96 cursor->add(CSSM_DB_EQUAL, infos.at(ix), value); 97 } 98 99 return cursor; 100} 101 102 103void 104PrimaryKeyImpl::putUInt32(uint8 *&p, uint32 value) 105{ 106 *p++ = (value >> 24); 107 *p++ = (value >> 16) & 0xff; 108 *p++ = (value >> 8) & 0xff; 109 *p++ = value & 0xff; 110} 111 112uint32 113PrimaryKeyImpl::getUInt32(uint8 *&p, uint32 &left) const 114{ 115 if (left < sizeof(uint32)) 116 MacOSError::throwMe(errSecNoSuchAttr); // XXX Not really but whatever. 117 118 119 // @@@ Assumes data written in big endian. 120 uint32 value = (p[0] << 24) + (p[1] << 16) + (p[2] << 8) + p[3]; 121 p += sizeof(uint32); 122 left -= sizeof(uint32); 123 return value; 124} 125 126 127 128CSSM_DB_RECORDTYPE 129PrimaryKeyImpl::recordType() const 130{ 131 uint8 *data = Data; 132 uint32 length = (uint32)Length; 133 return getUInt32(data, length); 134} 135