+++ /dev/null
-//
-// ClientInfoByNotification.m
-// Security
-//
-// Created by murf on 4/12/18.
-//
-
-#import <XCTest/XCTest.h>
-#import <Foundation/Foundation.h>
-#import <Foundation/NSXPCConnection_Private.h>
-#import <Security/Security.h>
-#include <Security/SecureObjectSync/SOSCloudCircle.h>
-#include <Security/SecureObjectSync/SOSCloudCircleInternal.h>
-
-#undef DOVIEWMACRO
-#define DOVIEWMACRO(VIEWNAME, DEFSTRING, CMDSTRING, SYSTEM, DEFAULTSETTING, INITIALSYNCSETTING, ALWAYSONSETTING, BACKUPSETTING, V0SETTING) \
-const CFStringRef k##SYSTEM##View##VIEWNAME = CFSTR(DEFSTRING);
-#include "Security/SecureObjectSync/ViewList.list"
-
-#import "DeviceSimulatorProtocol.h"
-#import "MultiDeviceNetworking.h"
-#import <objc/runtime.h>
-
-
-#if 1
-@interface MDDevice2 : NSObject<DeviceSimulatorProtocol>
-@property NSXPCConnection *connection;
-@property NSString *name;
-- (instancetype)initWithConnection:(NSXPCConnection *)connection;
-@end
-
-#pragma clang diagnostic push
-#pragma clang diagnostic ignored "-Wprotocol"
-@implementation MDDevice2
-
-- (instancetype)initWithConnection:(NSXPCConnection *)connection
-{
- self = [super init];
- if (self) {
- self.connection = connection;
- }
- return self;
-}
-
-/* Oh, ObjC, you are my friend */
-- (void)forwardInvocation:(NSInvocation *)invocation
-{
- struct objc_method_description desc = protocol_getMethodDescription(@protocol(DeviceSimulatorProtocol), [invocation selector], true, true);
- if (desc.name == NULL) {
- [super forwardInvocation:invocation];
- } else {
- __block bool dooooooEeeeetExclamationPoint = true;
- id object = [self.connection synchronousRemoteObjectProxyWithErrorHandler:^(NSError * _Nonnull error) {
- NSLog(@"peer failed with: %@", error);
- dooooooEeeeetExclamationPoint = false;
- //abort();
- }];
- if(dooooooEeeeetExclamationPoint) {
- [invocation invokeWithTarget:object];
- }
- }
-}
-@end
-#pragma clang diagnostic pop
-#endif
-
-@interface ClientInfoByNotification : XCTestCase
-@property NSMutableDictionary<NSString *,MDDevice2 *> *connections;
-@property MultiDeviceNetworking *network;
-@property MDDevice2 *masterDevice;
-@end
-
-static NSString *testInstanceUUID;
-
-@implementation ClientInfoByNotification
-
-
-+ (void)setUp
-{
- testInstanceUUID = [[NSUUID UUID] UUIDString];
-}
-
-- (void)setUp
-{
- self.connections = [NSMutableDictionary dictionary];
- self.network = [[MultiDeviceNetworking alloc] init];
- [self runSigninWithAdditionalDevices:0];
-}
-
-- (void)tearDown
-{
- __block uint64_t totalUserUsec = 0, totalSysUsec = 0;
- NSMutableDictionary *result = [NSMutableDictionary dictionary];
-
- for (NSString *name in self.connections) {
- MDDevice2 *device = self.connections[name];
- NSLog(@"device: %@", name);
- [device diagnosticsCPUUsage:^(bool success, uint64_t user_usec, uint64_t sys_usec, NSError *error) {
- NSLog(@"%@: %d: u:%llu s:%llu", device.name, success, (unsigned long long)user_usec, (unsigned long long)sys_usec);
- totalUserUsec += user_usec;
- totalSysUsec += sys_usec;
- result[[NSString stringWithFormat:@"cpu-%@", name]] = @{ @"user_usec" : @(user_usec), @"system_usec" : @(sys_usec)};
- }];
- }
-
- result[@"cpu-total"] = @{ @"user_usec" : @(totalUserUsec), @"system_usec" : @(totalSysUsec)};
-
- NSLog(@"Total: u:%llu s:%llu", (unsigned long long)totalUserUsec, (unsigned long long)totalSysUsec);
-
- /* XXX check for leaks in all devices */
- for (NSString *name in self.connections) {
- MDDevice2 *device = self.connections[name];
- [device.connection invalidate];
- }
- self.connections = NULL;
- [self.network dumpKVSState];
- [self.network dumpCounters];
- [self.network disconnectAll];
- self.network = NULL;
-
- NSData * jsonData = [NSJSONSerialization dataWithJSONObject:result options:0 error:NULL];
-
- [jsonData writeToFile:[NSString stringWithFormat:@"/tmp/test-result-%@", [self name]] atomically:NO];
-
-}
-
-//MARK: - Device logic
-
-- (MDDevice2 *)device:(NSString *)name model:(NSString *)model version:(NSString *)version
-{
- MDDevice2 *device = self.connections[name];
- if (device != NULL) {
- return NULL;
- }
-
- NSXPCConnection *conn = [[NSXPCConnection alloc] initWithServiceName:@"com.apple.Security.DeviceSimulator"];
- conn.remoteObjectInterface = [NSXPCInterface interfaceWithProtocol:@protocol(DeviceSimulatorProtocol)];
- [conn _setUUID:[NSUUID UUID]]; // select a random instance
- [conn resume];
-
- device = [[MDDevice2 alloc] initWithConnection:conn];
- device.name = name;
-
- self.connections[name] = device;
-
- [device setDevice:name
- version:version
- model:model
- testInstance:testInstanceUUID
- network:[self.network endpoint]
- complete:^(BOOL success) {
- if (!success) {
- abort();
- }
- }];
- return device;
-}
-
-- (void)runSigninWithAdditionalDevices:(unsigned)additionalDeviceCount {
- signal(SIGPIPE, SIG_IGN);
- _masterDevice = [self device:@"ipad" model:@"iPad" version:@"15E143a"];
- NSMutableArray<MDDevice2 *> *otherDevices = [NSMutableArray array];
-
- [_masterDevice setupSOSCircle:@"user" password:@"foo" complete:^void(bool success, NSError *error) {
- XCTAssert(success, "Expect success: %@", error);
- }];
-
- [_masterDevice sosCircleStatus:^void(SOSCCStatus status, NSError *error) {
- XCTAssertEqual(status, kSOSCCInCircle, @"expected to be in circle: %@", error);
- }];
-
- sleep(4); // give "kvs" a chance to get the word out.
- for (unsigned n = 0; n < additionalDeviceCount; n++) {
- MDDevice2 *dev = [self device:[NSString stringWithFormat:@"mac-%u", n] model:@"Mac Pro" version:@"17E121"];
-
- [dev setupSOSCircle:@"user" password:@"foo" complete:^void(bool success, NSError *error) {
- XCTAssert(success, "%u: Expect success: %@", n, error);
- }];
-
- __block NSString *devPeerID = NULL;
- [dev sosRequestToJoin:^(bool success, NSString *peerID, NSError *error) {
- XCTAssert(success, "%u: Expect success: %@", n, error);
- XCTAssertNotEqual(peerID, NULL, @"%u: expected to find peerID for peer2", n);
- devPeerID = peerID;
- }];
-
- __block NSError *localErr = nil;
- __block bool done = false;
- for(int tries=0; tries < 5; tries++) {
-
- [_masterDevice sosApprovePeer:devPeerID complete:^(BOOL success, NSError *error) {
- localErr = [error copy];
- if(success) {
- localErr = nil;
- done = true;
- }
- }];
- if(done) break;
- sleep(10);
- }
- XCTAssert(done, "%u: Expect success (for approve of %@): %@", n, devPeerID, localErr);
-
- [otherDevices addObject:dev];
- }
-}
-
-- (void)testSOSIsThisDeviceInCircle
-{
- [self measureBlock:^{
- for(int i=0; i<100; i++) {
- [self.masterDevice sosCircleStatus:^void(SOSCCStatus status, NSError *error) {
- XCTAssertEqual(status, kSOSCCInCircle, @"expected to be in circle: %@", error);
- }];
- }
- }];
-}
-
-- (void)testSOSIsThisDeviceInCircleNonCached
-{
- [self measureBlock:^{
- for(int i=0; i<100; i++) {
- [self.masterDevice sosCircleStatusNonCached:^void(SOSCCStatus status, NSError *error) {
- XCTAssertEqual(status, kSOSCCInCircle, @"expected to be in circle: %@", error);
- }];
- }
- }];
-}
-
-
-#if TARGET_OS_IPHONE
-- (void)testSOSViews2
-{
- [self.masterDevice sosViewStatus: (__bridge NSString*)kSOSViewAutofillPasswords withCompletion: ^void(SOSCCStatus status, NSError *error) {
- XCTAssertEqual(status, kSOSCCViewNotMember, @"expected to be not in view: %@", error);
- }];
- [self.masterDevice sosViewStatus: (__bridge NSString*)kSOSViewHomeKit withCompletion: ^void(SOSCCStatus status, NSError *error) {
- XCTAssertEqual(status, kSOSCCViewMember, @"expected to be in view: %@", error);
- }];
- [self.masterDevice sosViewStatus: (__bridge NSString*)kSOSViewWiFi withCompletion: ^void(SOSCCStatus status, NSError *error) {
- XCTAssertEqual(status, kSOSCCViewNotMember, @"expected to be not in view: %@", error);
- }];
- [self.masterDevice sosViewStatus: (__bridge NSString*)kSOSViewContinuityUnlock withCompletion: ^void(SOSCCStatus status, NSError *error) {
- XCTAssertEqual(status, kSOSCCViewMember, @"expected to be in view: %@", error);
- }];
-
- [self.masterDevice sosEnableAllViews:^(BOOL success, NSError *error) {
- XCTAssert(success, "Expected to enable all views");
- }];
-
- //uint64_t bitmask = [self.masterDevice sosCachedCircleBitmask];
- //XCTAssertEqual(bitmask, 0, "Expected bitmask to be %llx", bitmask);
-
-
- [self.masterDevice sosViewStatus: (__bridge NSString*)kSOSViewWiFi withCompletion: ^void(SOSCCStatus status, NSError *error) {
- XCTAssertEqual(status, kSOSCCViewMember, @"expected to be in view: %@", error);
- }];
-
- [self.masterDevice sosCachedViewBitmask:^(uint64_t bitmask) {
- XCTAssertEqual(bitmask, 33554367, @"expected bitmask of %llx", bitmask);
- }];
-
- [self measureBlock:^{
- for(int i=0; i<100; i++) {
- [self.masterDevice sosViewStatus: (__bridge NSString*)kSOSViewHomeKit withCompletion: ^void(SOSCCStatus status, NSError *error) {
- }];
- }
- }];
-}
-#endif
-
-
-- (void)testSOSMultiView
-{
- [self.masterDevice sosEnableAllViews:^(BOOL success, NSError *error) {
- XCTAssert(success, "Expected to enable all views");
- }];
-
- [self.masterDevice sosICKStatus: ^void(bool status) {
- XCTAssert(status, "Expected to enable iCloud Keychain");
- }];
-
- [self measureBlock:^{
- for(int i=0; i<100; i++) {
- [self.masterDevice sosICKStatus: ^void(bool status) {
- }];
- }
- }];
-}
-
-- (void) testMaster2
-{
- [self testSOSMultiView];
-}
-
-@end
projects / apple / security.git / blobdiff