OSX/sec/securityd/Regressions/secd-20-keychain_upgrade.m

   1 /*
   2  * Copyright (c) 2015 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  * This is to fool os services to not provide the Keychain manager
  26  * interface tht doens't work since we don't have unified headers
  27  * between iOS and OS X. rdar://23405418/
  28  */
  29 #define __KEYCHAINCORE__ 1
  30
  31
  32 #import <Foundation/Foundation.h>
  33 #import <CoreFoundation/CoreFoundation.h>
  34 #import <Security/SecBase.h>
  35 #import <Security/SecItem.h>
  36 #import <Security/SecItemPriv.h>
  37 #import <Security/SecInternal.h>
  38 #import <utilities/SecCFRelease.h>
  39 #import <utilities/SecFileLocations.h>
  40 #import <securityd/SecItemServer.h>
  41
  42 #import <stdlib.h>
  43
  44 #include "secd_regressions.h"
  45 #include "SecdTestKeychainUtilities.h"
  46
  47 static int ckmirror_row_exists = 0;
  48 static int ckmirror_row_callback(void* unused, int count, char **data, char **columns)
  49 {
  50     ckmirror_row_exists = 1;
  51     for (int i = 0; i < count; i++) {
  52         if(strcmp(columns[i], "ckzone") == 0) {
  53             is(strcmp(data[i], "ckzone"), 0, "Data is expected 'ckzone'");
  54         }
  55     }
  56
  57     return 0;
  58 }
  59
  60 static void
  61 keychain_upgrade(bool musr, const char *dbname)
  62 {
  63     OSStatus res;
  64
  65     secd_test_setup_temp_keychain(dbname, NULL);
  66
  67 #if TARGET_OS_IOS
  68     if (musr)
  69         SecSecuritySetMusrMode(true, 502, 502);
  70 #endif
  71
  72 #if TARGET_OS_IPHONE
  73     /*
  74      * Check system keychain migration
  75      */
  76
  77     res = SecItemAdd((CFDictionaryRef)@{
  78         (id)kSecClass :  (id)kSecClassGenericPassword,
  79         (id)kSecAttrAccount :  @"system-label-me",
  80         (id)kSecUseSystemKeychain : (id)kCFBooleanTrue,
  81         (id)kSecValueData : [NSData dataWithBytes:"some data" length:9],
  82     }, NULL);
  83     is(res, 0, "SecItemAdd(system)");
  84 #endif
  85
  86     /*
  87      * Check user keychain
  88      */
  89
  90     res = SecItemAdd((CFDictionaryRef)@{
  91         (id)kSecClass :  (id)kSecClassGenericPassword,
  92         (id)kSecAttrAccount :  @"user-label-me",
  93         (id)kSecValueData : [NSData dataWithBytes:"some data" length:9],
  94     }, NULL);
  95     is(res, 0, "SecItemAdd(user)");
  96
  97     NSString *keychain_path = CFBridgingRelease(__SecKeychainCopyPath());
  98
  99     // Add a row to a non-item table
 100     /* Create a new keychain sqlite db */
 101     sqlite3 *db = NULL;
 102
 103     is(sqlite3_open([keychain_path UTF8String], &db), SQLITE_OK, "open db");
 104     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");
 105     is(sqlite3_close(db), SQLITE_OK, "close db");
 106
 107     SecKeychainDbReset(^{
 108
 109         /* Create a new keychain sqlite db */
 110         sqlite3 *db;
 111
 112         is(sqlite3_open([keychain_path UTF8String], &db), SQLITE_OK, "create keychain");
 113         is(sqlite3_exec(db, "UPDATE tversion SET minor = minor - 1", NULL, NULL, NULL), SQLITE_OK,
 114            "\"downgrade\" keychain");
 115         is(sqlite3_close(db), SQLITE_OK, "close db");
 116     });
 117
 118 #if TARGET_OS_IPHONE
 119     res = SecItemCopyMatching((CFDictionaryRef)@{
 120         (id)kSecClass :  (id)kSecClassGenericPassword,
 121         (id)kSecAttrAccount :  @"system-label-me",
 122         (id)kSecUseSystemKeychain : (id)kCFBooleanTrue,
 123     }, NULL);
 124     is(res, 0, "SecItemCopyMatching(system)");
 125 #endif
 126
 127     res = SecItemCopyMatching((CFDictionaryRef)@{
 128         (id)kSecClass :  (id)kSecClassGenericPassword,
 129         (id)kSecAttrAccount :  @"user-label-me",
 130     }, NULL);
 131     is(res, 0, "SecItemCopyMatching(user)");
 132
 133     char* err = NULL;
 134
 135     is(sqlite3_open([keychain_path UTF8String], &db), SQLITE_OK, "open db");
 136     is(sqlite3_exec(db, "select * from ckmirror;", ckmirror_row_callback, NULL, &err), SQLITE_OK, "row added to ckmirror table");
 137     is(sqlite3_close(db), SQLITE_OK, "close db");
 138     is(ckmirror_row_exists, 1, "SQLite found a row in the ckmirror table");
 139
 140 #if TARGET_OS_IOS
 141     if (musr)
 142         SecSecuritySetMusrMode(false, 501, -1);
 143 #endif
 144 }
 145
 146 void SecAccessGroupsSetCurrent(CFArrayRef accessGroups);
 147 CFArrayRef SecAccessGroupsGetCurrent(void);
 148
 149 int
 150 secd_20_keychain_upgrade(int argc, char *const *argv)
 151 {
 152 #if TARGET_OS_IPHONE
 153 #define have_system_keychain_tests 2
 154 #else
 155 #define have_system_keychain_tests 0
 156 #endif
 157
 158     plan_tests((kSecdTestSetupTestCount + 5 + have_system_keychain_tests + 8) * 2);
 159
 160     CFArrayRef currentACL = SecAccessGroupsGetCurrent();
 161
 162     NSMutableArray *newACL = [NSMutableArray arrayWithArray:(__bridge NSArray *)currentACL];
 163     [newACL addObjectsFromArray:@[
 164          @"com.apple.private.system-keychain",
 165          @"com.apple.private.syncbubble-keychain",
 166          @"com.apple.private.migrate-musr-system-keychain",
 167     ]];
 168
 169     SecAccessGroupsSetCurrent((__bridge CFArrayRef)newACL);
 170
 171     keychain_upgrade(false, "secd_20_keychain_upgrade");
 172     keychain_upgrade(true,  "secd_20_keychain_upgrade-musr");
 173
 174     SecAccessGroupsSetCurrent(currentACL);
 175
 176     return 0;
 177 }