-//
-// MultiDeviceNetworking.m
-// Security
-//
-
-#import "MultiDeviceNetworking.h"
-#import "MultiDeviceNetworkingProtocol.h"
-
-@interface MDNCounters ()
-@property (assign) unsigned long kvsSyncAndWait;
-@property (assign) unsigned long kvsFlush;
-@property (assign) unsigned long kvsSend;
-@property (assign) unsigned long kvsRecv;
-@property (assign) unsigned long kvsRecvAll;
-@property (strong) NSMutableDictionary<NSString *,NSNumber *> *kvsKeys;
-
-- (void)addCountToKey:(NSString *)key;
-@end
-
-
-@interface MDNConnection : NSObject <MultiDeviceNetworkingProtocol>
-@property (weak) MultiDeviceNetworking *network;
-@property NSXPCConnection *inConnection;
-@property NSXPCConnection *outConnection;
-@property MDNCounters *counters;
-@end
-
-@interface MultiDeviceNetworking () <NSXPCListenerDelegate>
-@property NSXPCListener *networkListener;
-@property NSMutableDictionary *kvs;
-@property NSMutableArray<MDNConnection *> *connections;
-@property dispatch_queue_t serialQueue;
-@property NSMutableDictionary<NSString *, XCTestExpectation *> *expectations;
-@end
-
-@implementation MDNCounters
-
-- (instancetype)init {
- if ((self = [super init]) == NULL) {
- return nil;
- }
- self.kvsKeys = [NSMutableDictionary dictionary];
- return self;
-}
-
-- (NSDictionary *)summary{
- NSDictionary *kvsKeys = @{};
- @synchronized(self.kvsKeys) {
- kvsKeys = [self.kvsKeys copy];
- }
- return @{
- @"kvsSyncAndWait" : @(self.kvsSyncAndWait),
- @"kvsFlush" : @(self.kvsFlush),
- @"kvsSend" : @(self.kvsSend),
- @"kvsRecv" : @(self.kvsRecv),
- @"kvsRecvAll" : @(self.kvsRecvAll),
- @"kvsKeys" : kvsKeys,
- };
-}
-- (NSString *)description
-{
- return [NSString stringWithFormat:@"<MDNCounters: %@>", [self summary]];
-}
-- (void)addCountToKey:(NSString *)key
-{
- @synchronized(self.kvsKeys) {
- NSNumber *number = self.kvsKeys[key];
- self.kvsKeys[key] = @([number longValue] + 1);
- }
-}
-
-@end
-
-@implementation MultiDeviceNetworking
-
-- (instancetype)init
-{
- self = [super init];
- if (self) {
- self.networkListener = [NSXPCListener anonymousListener];
- self.networkListener.delegate = self;
- [self.networkListener resume];
- self.kvs = [[NSMutableDictionary alloc] init];
- self.connections = [NSMutableArray array];
- self.serialQueue = dispatch_queue_create("MultiDeviceNetworking.flushQueue", NULL);
- self.expectations = [NSMutableDictionary dictionary];
- }
- return self;
-}
-
-- (NSXPCListenerEndpoint *)endpoint
-{
- return [self.networkListener endpoint];
-}
-
-- (void)dumpKVSState
-{
- @synchronized(self.kvs) {
- puts("KVS STATE");
- [self.kvs enumerateKeysAndObjectsUsingBlock:^(id _Nonnull key, id _Nonnull obj, BOOL * _Nonnull __unused stop) {
- puts([[NSString stringWithFormat:@"%@ - %@", key, obj] UTF8String]);
- }];
- }
-}
-
-- (void)dumpCounters
-{
- @synchronized(self.connections) {
- puts("Network counters:");
- for (MDNConnection *conn in self.connections) {
- puts([[NSString stringWithFormat:@"%@", conn.counters] UTF8String]);
- }
- }
-}
-
-
-- (void)disconnectAll
-{
- @synchronized(self.connections) {
- for (MDNConnection *conn in self.connections) {
- [conn.inConnection invalidate];
- [conn.outConnection invalidate];
- }
- self.connections = [NSMutableArray array];
- }
-}
-
-- (void)setTestExpectation:(XCTestExpectation *)expectation forKey:(NSString *)key
-{
- self.expectations[key] = expectation;
-}
-
-- (void)clearTestExpectations
-{
- self.expectations = [NSMutableDictionary dictionary];
-}
-
-- (void)fulfill:(NSString *)key
-{
- [self.expectations[key] fulfill];
-}
-
-
-
-//MARK: - setup listener
-
-- (BOOL)listener:(NSXPCListener *)listener shouldAcceptNewConnection:(NSXPCConnection *)newConnection
-{
- newConnection.exportedInterface = [NSXPCInterface interfaceWithProtocol:@protocol(MultiDeviceNetworkingProtocol)];
-
- MDNConnection *conn = [[MDNConnection alloc] init];
- conn.network = self;
- conn.inConnection = newConnection;
- newConnection.exportedObject = conn;
-
- [self.connections addObject:conn];
- [newConnection resume];
-
- return YES;
-}
-
-@end
-
-
-//MARK: - KVS fun
-
-@implementation MDNConnection
-
-- (instancetype)init
-{
- if ((self = [super init]) == nil)
- return nil;
- _counters = [[MDNCounters alloc] init];
- return self;
-}
-
-- (void)MDNRegisterCallback:(NSXPCListenerEndpoint *)callback complete:(MDNComplete)complete
-{
- self.outConnection = [[NSXPCConnection alloc] initWithListenerEndpoint:callback];
- self.outConnection.remoteObjectInterface = [NSXPCInterface interfaceWithProtocol:@protocol(MultiDeviceNetworkingCallbackProtocol)];
-
- __typeof(self) weakSelf = self;
- self.outConnection.invalidationHandler = ^{
- __typeof(self) strongSelf = weakSelf;
- strongSelf.outConnection = nil;
- };
-
-
- [self.outConnection resume];
- complete(NULL, NULL);
-}
-
-- (void)MDNCloudPut:(NSDictionary *)values complete:(MDNComplete)complete {
- MultiDeviceNetworking *network = self.network;
- @synchronized(network.kvs) {
- [network.kvs setValuesForKeysWithDictionary:values];
- }
- /* interact with test expections so that tests can check that something happned in KVS */
- [network fulfill:@"Network"];
- for (NSString *key in values.allKeys) {
- NSString *dataSummary = @"";
- id value = values[key];
- if ([value isKindOfClass:[NSString class]]) {
- dataSummary = [NSString stringWithFormat:@" = string[%ld]", [(NSString *)value length]];
- } else if ([value isKindOfClass:[NSData class]]) {
- NSUInteger length = [(NSData *)value length];
- NSData *subdata = [(NSData *)value subdataWithRange:NSMakeRange(0, MIN(length, 4))];
- dataSummary = [NSString stringWithFormat:@" = data[%lu][%@]", (unsigned long)length, subdata];
- } else {
- dataSummary = [NSString stringWithFormat:@" = other(%@)", [value description]];
- }
- NSLog(@"KVS key update: %@%@", key, dataSummary);
- [network fulfill:key];
- [self.counters addCountToKey:key];
- }
-
-
- self.counters.kvsSend++;
- for (MDNConnection *conn in network.connections) {
- if (conn == self || conn.outConnection == NULL) {
- continue;
- }
- conn.counters.kvsRecv++;
- [[conn.outConnection remoteObjectProxy] MDNCItemsChanged:values complete:^(NSDictionary *returnedValues, NSError *error) {
- ;
- }];
- }
- complete(@{}, NULL);
-}
-
-- (void)MDNCloudsynchronizeAndWait:(NSDictionary *)values complete:(MDNComplete)complete {
- MultiDeviceNetworking *network = self.network;
- NSDictionary *kvsCopy = NULL;
- @synchronized(network.kvs) {
- kvsCopy = [network.kvs copy];
- }
- self.counters.kvsSyncAndWait++;
- [[self.outConnection synchronousRemoteObjectProxyWithErrorHandler:^(NSError * _Nonnull error) {
- NSLog(@"foo: %@", error);
- //abort();
- }] MDNCItemsChanged:kvsCopy complete:^(NSDictionary *returnedValues, NSError *error) {
- }];
- dispatch_async(network.serialQueue, ^{
- complete(@{}, NULL);
- });
-}
-
-- (void)MDNCloudGet:(NSArray *)keys complete:(MDNComplete)complete{
- MultiDeviceNetworking *network = self.network;
- NSLog(@"asking for: %@", keys);
- self.counters.kvsRecv++;
- NSMutableDictionary *reply = [NSMutableDictionary dictionary];
- @synchronized(network.kvs) {
- for (id key in keys) {
- reply[key] = network.kvs[key];
- }
- }
- complete(reply, NULL);
-}
-
-- (void)MDNCloudGetAll:(MDNComplete)complete
-{
- MultiDeviceNetworking *network = self.network;
- NSDictionary *kvsCopy = NULL;
- self.counters.kvsRecvAll++;
- @synchronized(network.kvs) {
- kvsCopy = [network.kvs copy];
- }
- complete(kvsCopy, NULL);
-}
-
-- (void)MDNCloudRemoveKeys:(NSArray<NSString *> *)keys complete:(MDNComplete)complete
-{
- MultiDeviceNetworking *network = self.network;
- @synchronized(network.kvs) {
- if (keys) {
- for (NSString *key in keys) {
- network.kvs[key] = NULL;
- }
- } else {
- network.kvs = [NSMutableDictionary dictionary];
- }
- }
- complete(NULL, NULL);
-}
-
-- (void)MDNCloudFlush:(MDNComplete)complete
-{
- self.counters.kvsFlush++;
- dispatch_async(self.network.serialQueue, ^{
- complete(@{}, NULL);
- });
-}
-
-@end
projects / apple / security.git / blobdiff