keychain/securityd/Regressions/secd-64-circlereset.m

   1 //
   2 //  secd-64-circlereset.c
   3 //  sec
   4 //
   5 //  Created by Richard Murphy on 7/22/15.
   6 //
   7 //
   8
   9
  10
  11 #include <Security/SecBase.h>
  12 #include <Security/SecItem.h>
  13
  14 #include <CoreFoundation/CFDictionary.h>
  15
  16 #include "keychain/SecureObjectSync/SOSAccount.h"
  17 #include <Security/SecureObjectSync/SOSCloudCircle.h>
  18 #include "keychain/SecureObjectSync/SOSInternal.h"
  19 #include "keychain/SecureObjectSync/SOSUserKeygen.h"
  20 #include "keychain/SecureObjectSync/SOSTransport.h"
  21 #include "keychain/SecureObjectSync/SOSAccountTrustClassic+Circle.h"
  22
  23 #include <stdlib.h>
  24 #include <unistd.h>
  25
  26 #include "secd_regressions.h"
  27 #include "SOSTestDataSource.h"
  28
  29 #include "SOSRegressionUtilities.h"
  30 #include <utilities/SecCFWrappers.h>
  31 #include <Security/SecKeyPriv.h>
  32
  33 #include "keychain/securityd/SOSCloudCircleServer.h"
  34
  35 #include "SOSAccountTesting.h"
  36
  37 #include "SecdTestKeychainUtilities.h"
  38 #if SOS_ENABLED
  39
  40 static int64_t getCurrentGenCount(SOSAccount* account) {
  41     SOSAccountTrustClassic* trust = account.trust;
  42     return SOSCircleGetGenerationSint(trust.trustedCircle);
  43 }
  44
  45 static bool SOSAccountResetWithGenCountValue(SOSAccount* account, int64_t gcount, CFErrorRef* error) {
  46     if (!SOSAccountHasPublicKey(account, error))
  47         return false;
  48     __block bool result = true;
  49     SOSAccountTrustClassic* trust = account.trust;
  50
  51     result &= [account.trust resetAllRings:account err:error];
  52
  53     trust.fullPeerInfo = nil;
  54     [trust setDepartureCode:kSOSWithdrewMembership];
  55
  56     result &= [account.trust modifyCircle:account.circle_transport err:error action:^(SOSCircleRef circle) {
  57         SOSGenCountRef gencount = SOSGenerationCreateWithValue(gcount);
  58         result = SOSCircleResetToEmpty(circle, error);
  59         SOSCircleSetGeneration(circle, gencount);
  60         CFReleaseNull(gencount);
  61         return result;
  62     }];
  63
  64     if (!result) {
  65         secerror("error: %@", error ? *error : NULL);
  66     }
  67
  68     return result;
  69 }
  70
  71 static SOSCircleRef SOSCircleCreateWithGenCount(int64_t gcount) {
  72     SOSCircleRef c = SOSCircleCreate(kCFAllocatorDefault, CFSTR("a"), NULL);
  73     SOSGenCountRef gencount = SOSGenerationCreateWithValue(gcount);
  74     SOSCircleSetGeneration(c, gencount);
  75     CFReleaseNull(gencount);
  76     return c;
  77 }
  78
  79 static void tests(void)
  80 {
  81     CFErrorRef error = NULL;
  82     CFDataRef cfpassword = CFDataCreate(NULL, (uint8_t *) "FooFooFoo", 10);
  83     CFStringRef cfaccount = CFSTR("test@test.org");
  84
  85     SOSCircleRef c1 = SOSCircleCreateWithGenCount(1);
  86     SOSCircleRef c99 = SOSCircleCreateWithGenCount(99);
  87     ok(SOSCircleIsOlderGeneration(c1, c99), "Is Comparison working correctly?", NULL);
  88     CFReleaseNull(c1);
  89     CFReleaseNull(c99);
  90
  91
  92     CFMutableDictionaryRef changes = CFDictionaryCreateMutableForCFTypes(kCFAllocatorDefault);
  93     SOSAccount* alice_account = CreateAccountForLocalChanges(CFSTR("Alice"), CFSTR("TestSource"));
  94     SOSAccount* bob_account = CreateAccountForLocalChanges(CFSTR("Bob"), CFSTR("TestSource"));
  95
  96     // Setup Circle with Bob and Alice in it
  97     ok(SOSAccountAssertUserCredentialsAndUpdate(bob_account, cfaccount, cfpassword, &error), "Credential setting (%@)", error);
  98     is(ProcessChangesUntilNoChange(changes, alice_account, bob_account, NULL), 1, "updates");
  99     ok(SOSAccountAssertUserCredentialsAndUpdate(alice_account, cfaccount, cfpassword, &error), "Credential setting (%@)", error);
 100     CFReleaseNull(error);
 101     ok(SOSAccountResetToOffering_wTxn(alice_account, &error), "Reset to offering (%@)", error);
 102     CFReleaseNull(error);
 103     is(ProcessChangesUntilNoChange(changes, alice_account, bob_account, NULL), 2, "updates");
 104     ok(SOSAccountJoinCircles_wTxn(bob_account, &error), "Bob Applies (%@)", error);
 105     CFReleaseNull(error);
 106     is(ProcessChangesUntilNoChange(changes, alice_account, bob_account, NULL), 2, "updates");
 107     {
 108         CFArrayRef applicants = SOSAccountCopyApplicants(alice_account, &error);
 109
 110         ok(applicants && CFArrayGetCount(applicants) == 1, "See one applicant %@ (%@)", applicants, error);
 111         ok(SOSAccountAcceptApplicants(alice_account, applicants, &error), "Alice accepts (%@)", error);
 112         CFReleaseNull(error);
 113         CFReleaseNull(applicants);
 114     }
 115     is(ProcessChangesUntilNoChange(changes, alice_account, bob_account, NULL), 3, "updates");
 116     accounts_agree("bob&alice pair", bob_account, alice_account);
 117     CFArrayRef peers = SOSAccountCopyPeers(alice_account, &error);
 118     ok(peers && CFArrayGetCount(peers) == 2, "See two peers %@ (%@)", peers, error);
 119     CFReleaseNull(peers);
 120
 121     uint64_t cnt = getCurrentGenCount(alice_account);
 122
 123     ok(SOSAccountResetWithGenCountValue(alice_account, cnt-1, &error), "Alice resets the circle to empty with old value");
 124     CFReleaseNull(error);
 125
 126     is(ProcessChangesUntilNoChange(changes, alice_account, bob_account, NULL), 1, "updates");
 127     is([bob_account getCircleStatus:&error], 0, "Bob Survives bad circle post");
 128     is([alice_account getCircleStatus:&error], 1, "Alice does not survive bad circle post");
 129
 130     CFReleaseNull(cfpassword);
 131     alice_account = nil;
 132     bob_account = nil;
 133     SOSTestCleanup();
 134 }
 135 #endif
 136
 137 int secd_64_circlereset(int argc, char *const *argv)
 138 {
 139 #if SOS_ENABLED
 140     plan_tests(35);
 141     secd_test_setup_temp_keychain(__FUNCTION__, NULL);
 142     tests();
 143 #else
 144     plan_tests(0);
 145 #endif
 146     return 0;
 147 }