keychain/ckks/tests/CKKSMockSOSPresentAdapter.m

   1 #import "keychain/ckks/CKKSListenerCollection.h"
   2 #import "keychain/ckks/tests/CKKSMockSOSPresentAdapter.h"
   3
   4 @interface CKKSMockSOSPresentAdapter()
   5 @property CKKSListenerCollection* peerChangeListeners;
   6 @end
   7
   8 @implementation CKKSMockSOSPresentAdapter
   9 @synthesize essential = _essential;
  10 @synthesize sosEnabled = _sosEnabled;
  11
  12 - (instancetype)initWithSelfPeer:(CKKSSOSSelfPeer*)selfPeer
  13                     trustedPeers:(NSSet<CKKSSOSPeer*>*)trustedPeers
  14                        essential:(BOOL)essential
  15 {
  16     if((self = [super init])) {
  17         _sosEnabled = true;
  18         _essential = essential;
  19
  20         _circleStatus = kSOSCCInCircle;
  21
  22         _excludeSelfPeerFromTrustSet = false;
  23
  24         _peerChangeListeners = [[CKKSListenerCollection alloc] initWithName:@"ckks-mock-sos"];
  25
  26         _selfPeer = selfPeer;
  27         _trustedPeers = [trustedPeers mutableCopy];
  28     }
  29     return self;
  30 }
  31
  32 - (NSString*)providerID
  33 {
  34     return [NSString stringWithFormat:@"[CKKSMockSOSPresentAdapter: %@]", self.selfPeer.peerID];
  35 }
  36
  37 - (SOSCCStatus)circleStatus:(NSError * _Nullable __autoreleasing * _Nullable)error
  38 {
  39     if(!self.sosEnabled || self.circleStatus == kSOSCCError) {
  40         if(error && self.circleStatus == kSOSCCError) {
  41             *error = [NSError errorWithDomain:(__bridge NSString*)kSOSErrorDomain code:self.circleStatus userInfo:nil];
  42         }
  43         return kSOSCCError;
  44     }
  45
  46     return self.circleStatus;
  47 }
  48
  49 // I currently don't understand when SOS returns a self or not. I've seen it return a self while not in kSOSCCInCircle,
  50 // which seems wrong. So, always return a self, unless we're in an obvious error state.
  51 - (id<CKKSSelfPeer> _Nullable)currentSOSSelf:(NSError * _Nullable __autoreleasing * _Nullable)error
  52 {
  53     if(self.selfPeerError) {
  54         if(error) {
  55             *error = self.selfPeerError;
  56         }
  57         return nil;
  58     }
  59
  60     if(self.sosEnabled && self.circleStatus == kSOSCCInCircle) {
  61         return self.selfPeer;
  62     } else {
  63         if(error) {
  64             *error = [NSError errorWithDomain:(__bridge NSString*)kSOSErrorDomain code:self.circleStatus userInfo:nil];
  65         }
  66         return nil;
  67     }
  68 }
  69
  70 - (CKKSSelves * _Nullable)fetchSelfPeers:(NSError *__autoreleasing  _Nullable * _Nullable)error
  71 {
  72     id<CKKSSelfPeer> peer = [self currentSOSSelf:error];
  73     if(!peer) {
  74         return nil;
  75     }
  76
  77     return [[CKKSSelves alloc] initWithCurrent:peer allSelves:nil];
  78 }
  79
  80 - (NSSet<id<CKKSRemotePeerProtocol>> * _Nullable)fetchTrustedPeers:(NSError * _Nullable __autoreleasing * _Nullable)error
  81 {
  82     if(self.trustedPeersError) {
  83         if(*error) {
  84             *error = self.trustedPeersError;
  85         }
  86         return nil;
  87     }
  88
  89     // TODO: I'm actually not entirely sure what SOS does if it's not in circle?
  90     if(self.sosEnabled && self.circleStatus == kSOSCCInCircle) {
  91         if(self.excludeSelfPeerFromTrustSet) {
  92             return self.trustedPeers;
  93         } else {
  94             return [self allPeers];
  95         }
  96     } else {
  97         if(error) {
  98             *error = [NSError errorWithDomain:(__bridge NSString*)kSOSErrorDomain code:kSOSCCNotInCircle userInfo:nil];
  99         }
 100         return nil;
 101     }
 102 }
 103
 104 - (void)updateOctagonKeySetWithAccount:(nonnull id<CKKSSelfPeer>)currentSelfPeer error:(NSError *__autoreleasing  _Nullable * _Nullable)error {
 105     if(self.updateOctagonKeySetListener) {
 106         self.updateOctagonKeySetListener(currentSelfPeer);
 107     }
 108     return;
 109 }
 110
 111 - (void)registerForPeerChangeUpdates:(nonnull id<CKKSPeerUpdateListener>)listener {
 112     [self.peerChangeListeners registerListener:listener];
 113 }
 114
 115 - (void)sendSelfPeerChangedUpdate {
 116     [self.peerChangeListeners iterateListeners: ^(id<CKKSPeerUpdateListener> listener) {
 117         [listener selfPeerChanged:self];
 118     }];
 119 }
 120
 121 - (void)sendTrustedPeerSetChangedUpdate {
 122     [self.peerChangeListeners iterateListeners: ^(id<CKKSPeerUpdateListener> listener) {
 123         [listener trustedPeerSetChanged:self];
 124     }];
 125 }
 126
 127 - (nonnull CKKSPeerProviderState *)currentState {
 128     return [CKKSPeerProviderState createFromProvider:self];
 129 }
 130
 131 - (NSSet<id<CKKSRemotePeerProtocol>>*)allPeers
 132 {
 133     // include the self peer, but as a CKKSSOSPeer object instead of a self peer
 134     CKKSSOSPeer* s = [[CKKSSOSPeer alloc] initWithSOSPeerID:self.selfPeer.peerID
 135                                         encryptionPublicKey:self.selfPeer.publicEncryptionKey
 136                                            signingPublicKey:self.selfPeer.publicSigningKey
 137                                                    viewList:self.selfPeer.viewList];
 138
 139     return [self.trustedPeers setByAddingObject: s];
 140 }
 141
 142 @end