keychain/ot/OTFetchViewsOperation.m

   1 /*
   2  * Copyright (c) 2019 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 #if OCTAGON
  25
  26 #import "keychain/ot/OTFetchViewsOperation.h"
  27 #import "keychain/ot/ObjCImprovements.h"
  28 #import "keychain/TrustedPeersHelper/TrustedPeersHelperProtocol.h"
  29 #import "keychain/ot/categories/OTAccountMetadataClassC+KeychainSupport.h"
  30 #import "keychain/ckks/CKKSAnalytics.h"
  31
  32 @interface OTFetchViewsOperation ()
  33 @property OTOperationDependencies* deps;
  34 @end
  35
  36 @implementation OTFetchViewsOperation
  37 @synthesize intendedState = _intendedState;
  38 @synthesize nextState = _nextState;
  39
  40 - (instancetype)initWithDependencies:(OTOperationDependencies*)dependencies
  41                        intendedState:(OctagonState*)intendedState
  42                           errorState:(OctagonState*)errorState
  43 {
  44     if ((self = [super init])) {
  45         _deps = dependencies;
  46
  47         _intendedState = intendedState;
  48         _nextState = errorState;
  49     }
  50     return self;
  51 }
  52
  53 - (void)groupStart
  54 {
  55     secnotice("octagon", "fetching views");
  56
  57     WEAKIFY(self);
  58     [self.deps.cuttlefishXPCWrapper fetchCurrentPolicyWithContainer:self.deps.containerName
  59                                                             context:self.deps.contextID
  60                                                               reply:^(NSSet<NSString*>* _Nullable viewList,
  61                                                                       TPPolicy* _Nullable policy,
  62                                                                       NSError* _Nullable error) {
  63         STRONGIFY(self);
  64         [[CKKSAnalytics logger] logResultForEvent:OctagonEventFetchViews hardFailure:true result:error];
  65
  66         if (error) {
  67             secerror("octagon: failed to retrieve policy+views: %@", error);
  68             self.error = error;
  69             return;
  70         }
  71
  72         secnotice("octagon-ckks", "Received policy %@ with view list: %@", policy, viewList);
  73         // Write them down before continuing
  74
  75         NSError* stateError = nil;
  76         [self.deps.stateHolder persistAccountChanges:^OTAccountMetadataClassC * _Nullable(OTAccountMetadataClassC * _Nonnull metadata) {
  77             metadata.syncingViews = [viewList mutableCopy];
  78             [metadata setTPPolicy:policy];
  79             return metadata;
  80         } error:&stateError];
  81
  82         if(stateError) {
  83             secerror("octagon: failed to save policy+views: %@", stateError);
  84             self.error = stateError;
  85             return;
  86         }
  87
  88         [self.deps.viewManager setSyncingViews:viewList sortingPolicy:policy];
  89
  90         self.nextState = self.intendedState;
  91     }];
  92 }
  93
  94 @end
  95
  96 #endif // OCTAGON