keychain/securityd/Regressions/secd-230-keybagtable.m

   1 /*
   2  * Copyright (c) 2013-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 #include "secd_regressions.h"
  26
  27 #include "keychain/securityd/SecDbItem.h"
  28 #include <utilities/array_size.h>
  29 #include <utilities/SecCFWrappers.h>
  30 #include <utilities/SecFileLocations.h>
  31 #include <utilities/fileIo.h>
  32
  33 #include "keychain/securityd/SecItemServer.h"
  34
  35 #include <Security/SecBasePriv.h>
  36 #include <Security/SecItemBackup.h>
  37
  38 #include <TargetConditionals.h>
  39 #include <AssertMacros.h>
  40
  41 #import "SecBackupKeybagEntry.h"
  42
  43 #if 0
  44
  45 <rdar://problem/30685971> Add test for keybag table add SPI
  46 <rdar://problem/30412884> Create an SPI to add a keybag to the keychain database (Keybag Table)
  47
  48 // original secd_35_keychain_migrate_inet
  49
  50 sudo defaults write /Library/Preferences/com.apple.security V10SchemaUpgradeTest -bool true
  51 sudo defaults read /Library/Preferences/com.apple.security V10SchemaUpgradeTest
  52
  53 #endif
  54
  55 #if USE_KEYSTORE
  56 #include "OSX/utilities/SecAKSWrappers.h"
  57
  58 #include "SecdTestKeychainUtilities.h"
  59
  60 static const bool kTestCustomKeybag = false;
  61 static const bool kTestLocalKeybag = false;
  62
  63 void SecAccessGroupsSetCurrent(CFArrayRef accessGroups);
  64 CFArrayRef SecAccessGroupsGetCurrent(void);
  65
  66 #define kSecdTestCreateCustomKeybagTestCount 6
  67 #define kSecdTestLocalKeybagTestCount 1
  68 #define kSecdTestKeybagtableTestCount 5
  69 #define kSecdTestAddItemTestCount 2
  70
  71 #define DATA_ARG(x) (x) ? CFDataGetBytePtr((x)) : NULL, (x) ? (int)CFDataGetLength((x)) : 0
  72
  73 // copied from si-33-keychain-backup.c
  74 static CFDataRef create_keybag(keybag_handle_t bag_type, CFDataRef password)
  75 {
  76     keybag_handle_t handle = bad_keybag_handle;
  77
  78     if (aks_create_bag(DATA_ARG(password), bag_type, &handle) == 0) {
  79         void * keybag = NULL;
  80         int keybag_size = 0;
  81         if (aks_save_bag(handle, &keybag, &keybag_size) == 0) {
  82             return CFDataCreate(kCFAllocatorDefault, keybag, keybag_size);
  83         }
  84     }
  85
  86     return CFDataCreate(kCFAllocatorDefault, NULL, 0);
  87 }
  88
  89 static bool createCustomKeybag() {
  90     /* custom keybag */
  91     keybag_handle_t keybag;
  92     keybag_state_t state;
  93     char *passcode="password";
  94     int passcode_len=(int)strlen(passcode);
  95     const bool kTestLockedKeybag = false;
  96
  97     ok(kAKSReturnSuccess==aks_create_bag(passcode, passcode_len, kAppleKeyStoreDeviceBag, &keybag), "create keybag");
  98     ok(kAKSReturnSuccess==aks_get_lock_state(keybag, &state), "get keybag state");
  99     ok(!(state&keybag_state_locked), "keybag unlocked");
 100     SecItemServerSetKeychainKeybag(keybag);
 101
 102     if (kTestLockedKeybag) {
 103         /* lock */
 104         ok(kAKSReturnSuccess==aks_lock_bag(keybag), "lock keybag");
 105         ok(kAKSReturnSuccess==aks_get_lock_state(keybag, &state), "get keybag state");
 106         ok(state&keybag_state_locked, "keybag locked");
 107     }
 108
 109     return true;
 110 }
 111
 112 static int keychainTestEnvironment(const char *environmentName, dispatch_block_t do_in_reset, dispatch_block_t do_in_environment) {
 113     //
 114     // Setup phase
 115     //
 116     CFArrayRef old_ag = SecAccessGroupsGetCurrent();
 117     CFMutableArrayRef test_ag = CFArrayCreateMutableCopy(NULL, 0, old_ag);
 118     CFArrayAppendValue(test_ag, CFSTR("test"));
 119     SecAccessGroupsSetCurrent(test_ag);
 120
 121     secd_test_setup_temp_keychain(environmentName, do_in_reset);
 122     bool haveCustomKeybag = kTestCustomKeybag && createCustomKeybag();
 123
 124     // Perform tasks in the test keychain environment
 125     if (do_in_environment)
 126         do_in_environment();
 127
 128     //
 129     // Cleanup phase
 130     //
 131
 132     // Reset keybag
 133     if (haveCustomKeybag)
 134         SecItemServerResetKeychainKeybag();
 135
 136     // Reset server accessgroups
 137     SecAccessGroupsSetCurrent(old_ag);
 138     CFReleaseSafe(test_ag);
 139     // Reset custom $HOME
 140     SetCustomHomePath(NULL);
 141     SecKeychainDbReset(NULL);
 142     return 0;
 143 }
 144
 145 static int addOneItemTest(NSString *account) {
 146     /* Creating a password */
 147     const char *v_data = "test";
 148     CFDataRef pwdata = CFDataCreate(NULL, (UInt8 *)v_data, strlen(v_data));
 149
 150     NSDictionary *item = @{
 151         (__bridge NSString *)kSecClass : (__bridge NSString *)kSecClassInternetPassword,
 152         (__bridge NSString *)kSecAttrServer : @"members.spamcop.net",
 153         (__bridge NSString *)kSecAttrAccount : account, // e.g. @"smith",
 154         (__bridge NSString *)kSecAttrPort : @80,
 155         (__bridge NSString *)kSecAttrProtocol : @"http",
 156         (__bridge NSString *)kSecAttrAuthenticationType : @"dflt",
 157         (__bridge NSString *)kSecValueData : (__bridge NSData *)pwdata
 158     };
 159
 160     ok_status(SecItemAdd((CFDictionaryRef)item, NULL), "add internet password, while unlocked");
 161     CFReleaseSafe(pwdata);
 162     return 0;
 163 }
 164
 165 static int localKeybagTest() {
 166     const char *pass = "sup3rsekretpassc0de";
 167     CFDataRef password = CFDataCreate(NULL, (UInt8 *)pass, strlen(pass));
 168     CFDataRef keybag = create_keybag(kAppleKeyStoreAsymmetricBackupBag, password);
 169     ok(keybag != NULL);
 170     CFReleaseNull(keybag);
 171     CFReleaseNull(password);
 172     return 0;
 173 }
 174
 175 static int test_keybagtable() {
 176     CFErrorRef error = NULL;
 177     const char *pass = "sup3rsekretpassc0de";
 178     CFDataRef password = CFDataCreate(NULL, (UInt8 *)pass, strlen(pass));
 179     CFDataRef identifier = NULL;
 180     CFURLRef pathinfo = NULL;
 181
 182     ok(SecBackupKeybagAdd(password, &identifier, &pathinfo, &error));
 183     CFReleaseNull(error);
 184
 185     NSDictionary *deleteQuery = @{(__bridge NSString *)kSecAttrPublicKeyHash:(__bridge NSData *)identifier};
 186     ok(SecBackupKeybagDelete((__bridge CFDictionaryRef)deleteQuery, &error));
 187
 188     ok(SecBackupKeybagAdd(password, &identifier, &pathinfo, &error));
 189     CFReleaseNull(error);
 190
 191     ok(SecBackupKeybagAdd(password, &identifier, &pathinfo, &error));
 192     CFReleaseNull(error);
 193
 194     NSDictionary *deleteAllQuery = @{(id)kSecMatchLimit: (id)kSecMatchLimitAll};
 195     ok(SecBackupKeybagDelete((__bridge CFDictionaryRef)deleteAllQuery, &error));
 196
 197     CFReleaseNull(identifier);
 198     CFReleaseNull(pathinfo);
 199     CFReleaseNull(password);
 200     CFReleaseNull(error);
 201     return 0;
 202 }
 203
 204 static void showHomeURL() {
 205 #if DEBUG
 206     CFURLRef homeURL = SecCopyHomeURL();
 207     NSLog(@"Home URL for test : %@", homeURL);
 208     CFReleaseSafe(homeURL);
 209 #endif
 210 }
 211
 212 int secd_230_keybagtable(int argc, char *const *argv)
 213 {
 214     int testcount = kSecdTestSetupTestCount + kSecdTestKeybagtableTestCount + kSecdTestAddItemTestCount;
 215     if (kTestLocalKeybag)
 216         testcount += kSecdTestLocalKeybagTestCount;
 217     if (kTestCustomKeybag)
 218         testcount += kSecdTestCreateCustomKeybagTestCount;
 219     plan_tests(testcount);
 220
 221     dispatch_block_t run_tests = ^{
 222         showHomeURL();
 223         if (kTestLocalKeybag)
 224             localKeybagTest();
 225         addOneItemTest(@"smith");
 226         test_keybagtable();
 227         addOneItemTest(@"jones");
 228     };
 229
 230     dispatch_block_t do_in_reset = NULL;
 231     dispatch_block_t do_in_environment = run_tests;
 232
 233     keychainTestEnvironment("secd_230_keybagtable", do_in_reset, do_in_environment);
 234
 235     return 0;
 236 }
 237
 238 #else
 239
 240 int secd_230_keybagtable(int argc, char *const *argv)
 241 {
 242     plan_tests(1);
 243     secLogDisable();
 244
 245     todo("Not yet working in simulator");
 246
 247     TODO: {
 248         ok(false);
 249     }
 250     /* not implemented in simulator (no keybag) */
 251     return 0;
 252 }
 253 #endif