X-Git-Url: https://git.saurik.com/apple/security.git/blobdiff_plain/fa7225c82381bac4432a6edf16f53b5370238d85..7e6b461318c8a779d91381531435a68ee4e8b6ed:/OSX/libsecurity_keychain/lib/defaultcreds.cpp?ds=sidebyside

diff --git a/OSX/libsecurity_keychain/lib/defaultcreds.cpp b/OSX/libsecurity_keychain/lib/defaultcreds.cpp
index cee6e780..d1b862d7 100644
--- a/OSX/libsecurity_keychain/lib/defaultcreds.cpp
+++ b/OSX/libsecurity_keychain/lib/defaultcreds.cpp
@@ -77,6 +77,9 @@ bool DefaultCredentials::operator () (Db database)
 					case CSSM_APPLE_UNLOCK_TYPE_WRAPPED_PRIVATE:
 						keyReferral(**it);
 						break;
+					case CSSM_APPLE_UNLOCK_TYPE_KEYBAG:
+						keybagReferral(**it);
+						break;
 					default:
 						secinfo("kcreferral", "referral type %lu (to %s) not supported",
 							(unsigned long)(*it)->type(), (*it)->dbName().c_str());
@@ -97,7 +100,7 @@ bool DefaultCredentials::operator () (Db database)
 
 //
 // Process a single referral record. This will handle all known types
-// of referrals.
+// of referrals, other than keybag (see keybagReferral).
 //
 void DefaultCredentials::keyReferral(const UnlockReferralRecord &ref)
 {
@@ -136,7 +139,7 @@ bool DefaultCredentials::unlockKey(const UnlockReferralRecord &ref, const Keycha
 		CSSM_DB_RECORDTYPE recordType =
 			(ref.type() == CSSM_APPLE_UNLOCK_TYPE_KEY_DIRECT) ?
 				CSSM_DL_DB_RECORD_SYMMETRIC_KEY : CSSM_DL_DB_RECORD_PRIVATE_KEY;
-		KCCursor cursor(list, recordType, &search);
+		KCCursor cursor(list, (SecItemClass) recordType, &search);
 		
 		Item keyItem;
 		while (cursor->next(keyItem)) {
@@ -170,6 +173,21 @@ bool DefaultCredentials::unlockKey(const UnlockReferralRecord &ref, const Keycha
 	return foundSome;
 }
 
+void
+DefaultCredentials::keybagReferral(const UnlockReferralRecord &ref)
+{
+	secinfo("kcreferral", "processing type %ld referral", (long)ref.type());
+
+	try {
+		// assemble and add CSSM_SAMPLE_TYPE_KEYCHAIN_LOCK item
+		append(TypedList(allocator, CSSM_SAMPLE_TYPE_KEYCHAIN_LOCK,
+			new(allocator) ListElement(CSSM_WORDID_KEYBAG_KEY),
+			new(allocator) ListElement(allocator, CssmData::wrap(ref.dbGuid())),
+			new(allocator) ListElement(allocator, ref.get())
+			));
+	} catch (...) {
+	}
+}
 
 //
 // Take the official keychain search list, and return those keychains whose