secd-20-keychain_upgrade.m   [plain text]

/*
* Copyright (c) 2015 Apple Inc. All Rights Reserved.
*
* @APPLE_LICENSE_HEADER_START@
*
* This file contains Original Code and/or Modifications of Original Code
* as defined in and that are subject to the Apple Public Source License
* Version 2.0 (the 'License'). You may not use this file except in
* compliance with the License. Please obtain a copy of the License at
* http://www.opensource.apple.com/apsl/ and read it before using this
* file.
*
* The 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.
*
* @APPLE_LICENSE_HEADER_END@
*/

/*
* This is to fool os services to not provide the Keychain manager
* interface tht doens't work since we don't have unified headers
* between iOS and OS X. rdar://23405418/
*/
#define __KEYCHAINCORE__ 1

#import <Foundation/Foundation.h>
#import <CoreFoundation/CoreFoundation.h>
#import <Security/SecBase.h>
#import <Security/SecItem.h>
#import <Security/SecItemPriv.h>
#import <Security/SecInternal.h>
#import <utilities/SecCFRelease.h>
#import <utilities/SecFileLocations.h>
#import "keychain/securityd/SecItemServer.h"

#import <stdlib.h>

#include "secd_regressions.h"
#include "SecdTestKeychainUtilities.h"
#include "server_security_helpers.h"

static int ckmirror_row_exists = 0;
static int ckmirror_row_callback(void* unused, int count, char **data, char **columns)
{
ckmirror_row_exists = 1;
for (int i = 0; i < count; i++) {
if(strcmp(columns[i], "ckzone") == 0) {
is(strcmp(data[i], "ckzone"), 0, "Data is expected 'ckzone'");
}
}

return 0;
}

static void
keychain_upgrade(bool musr, const char *dbname)
{
OSStatus res;

secd_test_setup_temp_keychain(dbname, NULL);

#if TARGET_OS_IOS
if (musr)
SecSecuritySetMusrMode(true, 502, 502);
#endif

#if TARGET_OS_IPHONE
/*
* Check system keychain migration
*/

res = SecItemAdd((__bridge CFDictionaryRef)@{
(id)kSecClass : (id)kSecClassGenericPassword,
(id)kSecAttrAccount : @"system-label-me",
(id)kSecUseSystemKeychain : (id)kCFBooleanTrue,
(id)kSecValueData : [NSData dataWithBytes:"some data" length:9],
}, NULL);
is(res, 0, "SecItemAdd(system)");
#endif

/*
* Check user keychain
*/

res = SecItemAdd((__bridge CFDictionaryRef)@{
(id)kSecClass : (id)kSecClassGenericPassword,
(id)kSecAttrAccount : @"user-label-me",
(id)kSecValueData : [NSData dataWithBytes:"some data" length:9],
}, NULL);
is(res, 0, "SecItemAdd(user)");

NSString *keychain_path = CFBridgingRelease(__SecKeychainCopyPath());

// Add a row to a non-item table
/* Create a new keychain sqlite db */
sqlite3 *db = NULL;

is(sqlite3_open([keychain_path UTF8String], &db), SQLITE_OK, "open db");
is(sqlite3_exec(db, "INSERT into ckmirror VALUES(\"ckzone\", \"importantuuid\", \"keyuuid\", 0, \"asdf\", \"qwer\", \"ckrecord\", 0, 0, NULL, NULL, NULL);", NULL, NULL, NULL), SQLITE_OK, "row added to ckmirror table");
is(sqlite3_close(db), SQLITE_OK, "close db");

SecKeychainDbForceClose();
SecKeychainDbReset(^{

/* Create a new keychain sqlite db */
sqlite3 *db;

is(sqlite3_open([keychain_path UTF8String], &db), SQLITE_OK, "create keychain");
is(sqlite3_exec(db, "UPDATE tversion SET minor = minor - 1", NULL, NULL, NULL), SQLITE_OK,
"\"downgrade\" keychain");
is(sqlite3_close(db), SQLITE_OK, "close db");
});

#if TARGET_OS_IPHONE
res = SecItemCopyMatching((__bridge CFDictionaryRef)@{
(id)kSecClass : (id)kSecClassGenericPassword,
(id)kSecAttrAccount : @"system-label-me",
(id)kSecUseSystemKeychain : (id)kCFBooleanTrue,
}, NULL);
is(res, 0, "SecItemCopyMatching(system)");
#endif

res = SecItemCopyMatching((__bridge CFDictionaryRef)@{
(id)kSecClass : (id)kSecClassGenericPassword,
(id)kSecAttrAccount : @"user-label-me",
}, NULL);
is(res, 0, "SecItemCopyMatching(user)");

char* err = NULL;

is(sqlite3_open([keychain_path UTF8String], &db), SQLITE_OK, "open db");
is(sqlite3_exec(db, "select * from ckmirror;", ckmirror_row_callback, NULL, &err), SQLITE_OK, "row added to ckmirror table");
is(sqlite3_close(db), SQLITE_OK, "close db");
is(ckmirror_row_exists, 1, "SQLite found a row in the ckmirror table");

#if TARGET_OS_IOS
if (musr)
SecSecuritySetMusrMode(false, 501, -1);
#endif

secd_test_teardown_delete_temp_keychain(dbname);
}

int
secd_20_keychain_upgrade(int argc, char *const *argv)
{
#if TARGET_OS_IPHONE
#define have_system_keychain_tests 2
#else
#define have_system_keychain_tests 0
#endif

plan_tests((kSecdTestSetupTestCount + 5 + have_system_keychain_tests + 8) * 2);

CFArrayRef currentACL = CFRetainSafe(SecAccessGroupsGetCurrent());

NSMutableArray *newACL = [NSMutableArray arrayWithArray:(__bridge NSArray *)currentACL];
[newACL addObjectsFromArray:@[
@"com.apple.private.system-keychain",
@"com.apple.private.syncbubble-keychain",
@"com.apple.private.migrate-musr-system-keychain",
]];

SecAccessGroupsSetCurrent((__bridge CFArrayRef)newACL);

keychain_upgrade(false, "secd_20_keychain_upgrade");
keychain_upgrade(true, "secd_20_keychain_upgrade-musr");

SecAccessGroupsSetCurrent(currentACL);
CFReleaseNull(currentACL);

return 0;
}