keychain/ot/OTResetOperation.m

   1 /*
   2  * Copyright (c) 2018 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 #if OCTAGON
  24
  25 #import "keychain/ot/OTResetOperation.h"
  26
  27 #import "keychain/ot/ObjCImprovements.h"
  28 #import "keychain/TrustedPeersHelper/TrustedPeersHelperProtocol.h"
  29
  30 @interface OTResetOperation ()
  31 @property NSString* containerName;
  32 @property NSString* contextID;
  33 @property id<NSXPCProxyCreating> cuttlefishXPC;
  34
  35 // Since we're making callback based async calls, use this operation trick to hold off the ending of this operation
  36 @property NSOperation* finishedOp;
  37 @end
  38
  39 @implementation OTResetOperation
  40 @synthesize intendedState = _intendedState;
  41 @synthesize nextState = _nextState;
  42
  43 - (instancetype)init:(NSString*)containerName
  44            contextID:(NSString*)contextID
  45        intendedState:(OctagonState*)intendedState
  46           errorState:(OctagonState*)errorState
  47        cuttlefishXPC:(id<NSXPCProxyCreating>)cuttlefishXPC
  48 {
  49     if((self = [super init])) {
  50         _intendedState = intendedState;
  51         _nextState = errorState;
  52
  53         _containerName = containerName;
  54         _contextID = contextID;
  55         _cuttlefishXPC = cuttlefishXPC;
  56     }
  57     return self;
  58 }
  59
  60 - (void)groupStart
  61 {
  62     secnotice("octagon-authkit", "Attempting to reset octagon");
  63
  64     self.finishedOp = [[NSOperation alloc] init];
  65     [self dependOnBeforeGroupFinished:self.finishedOp];
  66
  67     WEAKIFY(self);
  68     [[self.cuttlefishXPC remoteObjectProxyWithErrorHandler:^(NSError * _Nonnull error) {
  69         STRONGIFY(self);
  70         secerror("octagon: Can't talk with TrustedPeersHelper: %@", error);
  71         [[CKKSAnalytics logger] logRecoverableError:error forEvent:OctagonEventReset withAttributes:NULL];
  72         self.error = error;
  73         [self runBeforeGroupFinished:self.finishedOp];
  74
  75     }] resetWithContainer:self.containerName
  76                   context:self.contextID
  77                     reply:^(NSError * _Nullable error) {
  78          STRONGIFY(self);
  79          [[CKKSAnalytics logger] logResultForEvent:OctagonEventReset hardFailure:true result:error];
  80
  81          if(error) {
  82              secnotice("octagon", "Unable to reset for (%@,%@): %@", self.containerName, self.contextID, error);
  83              self.error = error;
  84          } else {
  85              secnotice("octagon", "Successfully reset Octagon");
  86              self.nextState = self.intendedState;
  87          }
  88
  89          [self runBeforeGroupFinished:self.finishedOp];
  90      }];
  91 }
  92
  93 @end
  94
  95 #endif // OCTAGON