keychain/ckksctl/ckksctl.m

   1 //
   2 //  Security
   3 //
   4
   5 #import <Foundation/Foundation.h>
   6 #import <Foundation/NSXPCConnection_Private.h>
   7 #import <Security/Security.h>
   8 #import <Security/SecItemPriv.h>
   9 #import <xpc/xpc.h>
  10 #import <err.h>
  11
  12 #import "keychain/ckks/CKKS.h"
  13 #import "keychain/ckks/CKKSControl.h"
  14
  15 #include "lib/SecArgParse.h"
  16
  17 static void nsprintf(NSString *fmt, ...) NS_FORMAT_FUNCTION(1, 2);
  18 static void print_result(NSDictionary *dict, bool json_flag);
  19 static void print_dict(NSDictionary *dict, int ind);
  20 static void print_array(NSArray *array, int ind);
  21 static void print_entry(id k, id v, int ind);
  22
  23 static void nsprintf(NSString *fmt, ...)
  24 {
  25     va_list ap;
  26     va_start(ap, fmt);
  27     NSString *str = [[NSString alloc] initWithFormat:fmt arguments:ap];
  28     va_end(ap);
  29
  30     puts([str UTF8String]);
  31 #if !__has_feature(objc_arc)
  32     [str release];
  33 #endif
  34 }
  35
  36 // Mutual recursion to set up an object for jsonification
  37 static NSDictionary* cleanDictionaryForJSON(NSDictionary* dict);
  38
  39 static id cleanObjectForJSON(id obj) {
  40     if(!obj) {
  41         return nil;
  42     }
  43     if([obj isKindOfClass:[NSError class]]) {
  44         NSError* obje = (NSError*) obj;
  45         NSMutableDictionary* newErrorDict = [@{@"code": @(obje.code), @"domain": obje.domain} mutableCopy];
  46         newErrorDict[@"userInfo"] = cleanDictionaryForJSON(obje.userInfo);
  47         return newErrorDict;
  48     } else if([NSJSONSerialization isValidJSONObject:obj]) {
  49         return obj;
  50
  51     } else if ([obj isKindOfClass: [NSNumber class]]) {
  52         return obj;
  53
  54     } else if([obj isKindOfClass: [NSData class]]) {
  55         NSData* dataObj = (NSData*)obj;
  56         return [dataObj base64EncodedStringWithOptions:0];
  57
  58     } else if ([obj isKindOfClass: [NSDictionary class]]) {
  59         return cleanDictionaryForJSON((NSDictionary*) obj);
  60
  61     } else if ([obj isKindOfClass: [NSArray class]]) {
  62         NSArray* arrayObj = (NSArray*)obj;
  63         NSMutableArray* cleanArray = [NSMutableArray arrayWithCapacity:arrayObj.count];
  64
  65         for(id x in arrayObj) {
  66             [cleanArray addObject: cleanObjectForJSON(x)];
  67         }
  68         return cleanArray;
  69
  70     } else {
  71         return [obj description];
  72     }
  73 }
  74
  75 static NSDictionary* cleanDictionaryForJSON(NSDictionary* dict) {
  76     if(!dict) {
  77         return nil;
  78     }
  79     NSMutableDictionary* mutDict = [dict mutableCopy];
  80     for(id key in mutDict.allKeys) {
  81         id obj = mutDict[key];
  82         mutDict[key] = cleanObjectForJSON(obj);
  83     }
  84     return mutDict;
  85 }
  86
  87 static void print_result(NSDictionary *dict, bool json_flag)
  88 {
  89     if (json_flag) {
  90         NSError *err;
  91
  92         NSData *json = [NSJSONSerialization dataWithJSONObject:cleanDictionaryForJSON(dict)
  93                                                        options:(NSJSONWritingPrettyPrinted | NSJSONWritingSortedKeys)
  94                                                          error:&err];
  95         if (!json) {
  96             NSLog(@"error: %@", err.localizedDescription);
  97         } else {
  98             printf("%s", [[[NSString alloc] initWithData:json encoding:NSUTF8StringEncoding] UTF8String]);
  99         }
 100     } else {
 101         print_dict(dict, 0);
 102     }
 103 }
 104
 105 static void print_dict(NSDictionary *dict, int ind)
 106 {
 107     NSArray *sortedKeys = [[dict allKeys] sortedArrayUsingSelector:@selector(localizedCaseInsensitiveCompare:)];
 108     for (id k in sortedKeys) {
 109         id v = dict[k];
 110         print_entry(k, v, ind);
 111     }
 112 }
 113
 114 static void print_array(NSArray *array, int ind)
 115 {
 116     [array enumerateObjectsUsingBlock:^(id v, NSUInteger i, BOOL *stop __unused) {
 117         print_entry(@(i), v, ind);
 118     }];
 119 }
 120
 121 static void print_entry(id k, id v, int ind)
 122 {
 123     if ([v isKindOfClass:[NSDictionary class]]) {
 124         if (ind == 0) {
 125             nsprintf(@"\n%*s%@ -", ind * 4, "", k);
 126             nsprintf(@"%*s========================", ind * 4, "");
 127         } else if (ind == 1) {
 128             nsprintf(@"\n%*s%@ -", ind * 4, "", k);
 129             nsprintf(@"%*s------------------------", ind * 4, "");
 130         } else {
 131             nsprintf(@"%*s%@ -", ind * 4, "", k);
 132         }
 133
 134         print_dict(v, ind + 1);
 135     } else if ([v isKindOfClass:[NSArray class]]) {
 136         nsprintf(@"%*s%@ -", ind * 4, "", k);
 137         print_array(v, ind + 1);
 138     } else {
 139         nsprintf(@"%*s%@: %@", ind * 4, "", k, v);
 140     }
 141 }
 142
 143 @interface CKKSControlCLI : NSObject
 144 @property CKKSControl* control;
 145 @end
 146
 147 @implementation CKKSControlCLI
 148
 149 - (instancetype) initWithCKKSControl:(CKKSControl*)control {
 150     if ((self = [super init])) {
 151         _control = control;
 152     }
 153
 154     return self;
 155 }
 156
 157 - (NSDictionary<NSString *, id> *)fetchPerformanceCounters
 158 {
 159     NSMutableDictionary *perfDict = [[NSMutableDictionary alloc] init];
 160 #if OCTAGON
 161     dispatch_semaphore_t sema = dispatch_semaphore_create(0);
 162
 163     [self.control rpcPerformanceCounters:^(NSDictionary<NSString *,NSNumber *> * counters, NSError * error) {
 164         if(error) {
 165             perfDict[@"error"] = [error description];
 166         }
 167
 168         [counters enumerateKeysAndObjectsUsingBlock:^(NSString * key, NSNumber * obj, BOOL *stop) {
 169             perfDict[key] = obj;
 170         }];
 171
 172         dispatch_semaphore_signal(sema);
 173     }];
 174
 175     if(dispatch_semaphore_wait(sema, dispatch_time(DISPATCH_TIME_NOW, NSEC_PER_SEC * 60)) != 0) {
 176         perfDict[@"error"] = @"timed out waiting for response";
 177     }
 178 #endif
 179
 180     return perfDict;
 181 }
 182
 183 - (long)resetLocal:(NSString*)view {
 184     __block long ret = 0;
 185 #if OCTAGON
 186     printf("Beginning local reset for %s...\n", view ? [[view description] UTF8String] : "all zones");
 187     dispatch_semaphore_t sema = dispatch_semaphore_create(0);
 188
 189     [self.control rpcResetLocal:view
 190                           reply:^(NSError *error) {
 191                               if(error == NULL) {
 192                                   printf("reset complete.\n");
 193                                   ret = 0;
 194                               } else {
 195                                   nsprintf(@"reset error: %@\n", error);
 196                                   ret = error.code;
 197                               }
 198                               dispatch_semaphore_signal(sema);
 199                           }];
 200
 201     if(dispatch_semaphore_wait(sema, dispatch_time(DISPATCH_TIME_NOW, NSEC_PER_SEC * 60 * 2)) != 0) {
 202         printf("\n\nError: timed out waiting for response\n");
 203         return -1;
 204     }
 205 #endif // OCTAGON
 206     return ret;
 207 }
 208
 209 - (long)resetCloudKit:(NSString*)view {
 210     __block long ret = 0;
 211 #if OCTAGON
 212     printf("Beginning CloudKit reset for %s...\n", view ? [[view description] UTF8String] : "all zones");
 213     dispatch_semaphore_t sema = dispatch_semaphore_create(0);
 214
 215     [self.control rpcResetCloudKit:view reason:@"ckksctl" reply:^(NSError* error){
 216         if(error == NULL) {
 217             printf("CloudKit Reset complete.\n");
 218             ret = 0;
 219         } else {
 220             nsprintf(@"Reset error: %@\n", error);
 221             ret = error.code;
 222         }
 223         dispatch_semaphore_signal(sema);
 224     }];
 225
 226     if(dispatch_semaphore_wait(sema, dispatch_time(DISPATCH_TIME_NOW, NSEC_PER_SEC * 60 * 5)) != 0) {
 227         printf("\n\nError: timed out waiting for response\n");
 228         return -1;
 229     }
 230 #endif // OCTAON
 231     return ret;
 232 }
 233
 234 - (long)resync:(NSString*)view {
 235     __block long ret = 0;
 236 #if OCTAGON
 237     printf("Beginning resync for %s...\n", view ? [[view description] UTF8String] : "all zones");
 238     dispatch_semaphore_t sema = dispatch_semaphore_create(0);
 239
 240     [self.control rpcResync:view reply:^(NSError* error){
 241         if(error == NULL) {
 242             printf("resync success.\n");
 243             ret = 0;
 244         } else {
 245             nsprintf(@"resync errored: %@\n", error);
 246             ret = error.code;
 247         }
 248         dispatch_semaphore_signal(sema);
 249     }];
 250
 251     if(dispatch_semaphore_wait(sema, dispatch_time(DISPATCH_TIME_NOW, NSEC_PER_SEC * 60 * 2)) != 0) {
 252         printf("\n\nError: timed out waiting for response\n");
 253         return -1;
 254     }
 255 #endif // OCTAGON
 256     return ret;
 257 }
 258
 259 - (NSDictionary<NSString *, id> *)fetchStatus: (NSString*) view {
 260     NSMutableDictionary *status = [[NSMutableDictionary alloc] init];
 261 #if OCTAGON
 262     dispatch_semaphore_t sema = dispatch_semaphore_create(0);
 263
 264     [self.control rpcStatus: view reply: ^(NSArray<NSDictionary*>* result, NSError* error) {
 265         if(error) {
 266             status[@"error"] = [error description];
 267         }
 268
 269         if(result.count <= 1u) {
 270             printf("No CKKS views are active.\n");
 271         }
 272
 273
 274         for(NSDictionary* viewStatus in result) {
 275             status[viewStatus[@"view"]] = viewStatus;
 276         }
 277         dispatch_semaphore_signal(sema);
 278     }];
 279
 280     if(dispatch_semaphore_wait(sema, dispatch_time(DISPATCH_TIME_NOW, NSEC_PER_SEC * 30)) != 0) {
 281         status[@"error"] = @"timed out";
 282     }
 283 #endif // OCTAGON
 284     return status;
 285 }
 286
 287 - (void)printHumanReadableStatus: (NSString*) view {
 288 #if OCTAGON
 289     dispatch_semaphore_t sema = dispatch_semaphore_create(0);
 290
 291     [self.control rpcStatus: view reply: ^(NSArray<NSDictionary*>* result, NSError* error) {
 292         if(error) {
 293             printf("ERROR FETCHING STATUS: %s\n", [[error description] UTF8String]);
 294         }
 295
 296 #define pop(d, key, cls) ({ id x = d[key]; d[key] = nil; [x isKindOfClass: [cls class]] ? x : nil; })
 297
 298         // First result is always global state
 299         // Ideally, this would come in another parameter, but we can't change the protocol until
 300         // <rdar://problem/33583242> CKKS: remove PCS's use of CKKSControlProtocol
 301         NSMutableDictionary* global = [result[0] mutableCopy];
 302         if(global) {
 303             NSString* reachability = pop(global, @"reachability", NSString);
 304             NSString* ckdeviceID = pop(global, @"ckdeviceID", NSString);
 305             NSString* ckdeviceIDError = pop(global, @"ckdeviceIDError", NSString);
 306             NSString* lockStateTracker = pop(global,@"lockstatetracker", NSString);
 307             NSString* retry = pop(global,@"cloudkitRetryAfter", NSString);
 308             NSDate *lastCKKSPush = pop(global, @"lastCKKSPush", NSDate);
 309
 310             printf("================================================================================\n\n");
 311             printf("Global state:\n\n");
 312             printf("Reachability:         %s\n", [[reachability description] UTF8String]);
 313             printf("Retry:                %s\n", [[retry description] UTF8String]);
 314             printf("CK DeviceID:          %s\n", [[ckdeviceID description] UTF8String]);
 315             printf("CK DeviceID Error:    %s\n", [[ckdeviceIDError description] UTF8String]);
 316             printf("Lock state:           %s\n", [[lockStateTracker description] UTF8String]);
 317             printf("Last CKKS push:       %s\n", [[lastCKKSPush description] UTF8String]);
 318
 319             printf("\n");
 320         }
 321
 322         NSArray* remainingViews = result.count > 1 ? [result subarrayWithRange:NSMakeRange(1, result.count-1)] : @[];
 323
 324         if(remainingViews.count == 0u) {
 325             printf("No CKKS views are active.\n");
 326         }
 327
 328         for(NSDictionary* viewStatus in remainingViews) {
 329             NSMutableDictionary* status = [viewStatus mutableCopy];
 330
 331             NSString* viewName = pop(status,@"view", NSString);
 332             NSString* accountStatus = pop(status,@"ckaccountstatus", NSString);
 333             NSString* accountTracker = pop(status,@"accounttracker", NSString);
 334             NSString* fetcher = pop(status,@"fetcher", NSString);
 335             NSString* zoneCreated = pop(status,@"zoneCreated", NSString);
 336             NSString* zoneCreatedError = pop(status,@"zoneCreatedError", NSString);
 337             NSString* zoneSubscribed = pop(status,@"zoneSubscribed", NSString);
 338             NSString* zoneSubscribedError = pop(status,@"zoneSubscribedError", NSString);
 339             NSString* zoneInitializeScheduler = pop(status,@"zoneInitializeScheduler", NSString);
 340             NSString* keystate = pop(status,@"keystate", NSString);
 341             NSString* keyStateError = pop(status,@"keyStateError", NSString);
 342             NSString* statusError = pop(status,@"statusError", NSString);
 343             NSString* currentTLK =    pop(status,@"currentTLK", NSString);
 344             NSString* currentClassA = pop(status,@"currentClassA", NSString);
 345             NSString* currentClassC = pop(status,@"currentClassC", NSString);
 346             NSString* currentTLKPtr =    pop(status,@"currentTLKPtr", NSString);
 347             NSString* currentClassAPtr = pop(status,@"currentClassAPtr", NSString);
 348             NSString* currentClassCPtr = pop(status,@"currentClassCPtr", NSString);
 349             NSString* currentManifestGeneration = pop(status,@"currentManifestGen", NSString);
 350             NSArray* launchSequence = pop(status, @"launchSequence", NSArray);
 351
 352             NSDictionary* oqe = pop(status,@"oqe", NSDictionary);
 353             NSDictionary* iqe = pop(status,@"iqe", NSDictionary);
 354             NSDictionary* keys = pop(status,@"keys", NSDictionary);
 355             NSDictionary* ckmirror = pop(status,@"ckmirror", NSDictionary);
 356             NSArray* devicestates = pop(status, @"devicestates", NSArray);
 357             NSArray* tlkshares = pop(status, @"tlkshares", NSArray);
 358
 359             NSString* zoneSetupOperation                  = pop(status,@"zoneSetupOperation", NSString);
 360             NSString* keyStateOperation                   = pop(status,@"keyStateOperation", NSString);
 361             NSString* lastIncomingQueueOperation          = pop(status,@"lastIncomingQueueOperation", NSString);
 362             NSString* lastNewTLKOperation                 = pop(status,@"lastNewTLKOperation", NSString);
 363             NSString* lastOutgoingQueueOperation          = pop(status,@"lastOutgoingQueueOperation", NSString);
 364             NSString* lastProcessReceivedKeysOperation    = pop(status,@"lastProcessReceivedKeysOperation", NSString);
 365             NSString* lastReencryptOutgoingItemsOperation = pop(status,@"lastReencryptOutgoingItemsOperation", NSString);
 366             NSString* lastScanLocalItemsOperation         = pop(status,@"lastScanLocalItemsOperation", NSString);
 367
 368             printf("================================================================================\n\n");
 369
 370             printf("View: %s\n\n", [viewName UTF8String]);
 371
 372             if(statusError != nil) {
 373                 printf("ERROR FETCHING STATUS: %s\n\n", [statusError UTF8String]);
 374             }
 375
 376             printf("CloudKit account:     %s\n", [accountStatus UTF8String]);
 377             printf("Account tracker:      %s\n", [accountTracker UTF8String]);
 378
 379             if(!([zoneCreated isEqualToString:@"yes"] && [zoneSubscribed isEqualToString:@"yes"])) {
 380                 printf("CK Zone Created:            %s\n", [[zoneCreated description] UTF8String]);
 381                 printf("CK Zone Created error:      %s\n", [[zoneCreatedError description] UTF8String]);
 382
 383                 printf("CK Zone Subscribed:         %s\n", [[zoneSubscribed description] UTF8String]);
 384                 printf("CK Zone Subscription error: %s\n", [[zoneSubscribedError description] UTF8String]);
 385                 printf("CK Zone initialize retry:   %s\n", [[zoneInitializeScheduler description] UTF8String]);
 386                 printf("\n");
 387             }
 388
 389             printf("Key state:            %s\n", [keystate UTF8String]);
 390             if(keyStateError != nil) {
 391                 printf("Key State Error: %s\n", [keyStateError UTF8String]);
 392             }
 393             printf("Current TLK:          %s\n", currentTLK != nil
 394                    ? [currentTLK    UTF8String]
 395                    : [[NSString stringWithFormat:@"missing; pointer is %@", currentTLKPtr] UTF8String]);
 396             printf("Current ClassA:       %s\n", currentClassA != nil
 397                    ? [currentClassA UTF8String]
 398                    : [[NSString stringWithFormat:@"missing; pointer is %@", currentClassAPtr] UTF8String]);
 399             printf("Current ClassC:       %s\n", currentClassC != nil
 400                    ? [currentClassC UTF8String]
 401                    : [[NSString stringWithFormat:@"missing; pointer is %@", currentClassCPtr] UTF8String]);
 402
 403             printf("TLK shares:           %s\n", [[tlkshares description] UTF8String]);
 404
 405             printf("Outgoing Queue counts: %s\n", [[oqe description] UTF8String]);
 406             printf("Incoming Queue counts: %s\n", [[iqe description] UTF8String]);
 407             printf("Key counts: %s\n", [[keys description] UTF8String]);
 408             printf("latest manifest generation: %s\n", currentManifestGeneration == nil ? "null" : currentManifestGeneration.UTF8String);
 409
 410             printf("Item counts (by key):  %s\n", [[ckmirror description] UTF8String]);
 411             printf("Peer states:           %s\n", [[devicestates description] UTF8String]);
 412
 413             printf("zone change fetcher:                 %s\n", [[fetcher description] UTF8String]);
 414             printf("zoneSetupOperation:                  %s\n", zoneSetupOperation                  == nil ? "never" : [zoneSetupOperation                  UTF8String]);
 415             printf("keyStateOperation:                   %s\n", keyStateOperation                   == nil ? "never" : [keyStateOperation                   UTF8String]);
 416             printf("lastIncomingQueueOperation:          %s\n", lastIncomingQueueOperation          == nil ? "never" : [lastIncomingQueueOperation          UTF8String]);
 417             printf("lastNewTLKOperation:                 %s\n", lastNewTLKOperation                 == nil ? "never" : [lastNewTLKOperation                 UTF8String]);
 418             printf("lastOutgoingQueueOperation:          %s\n", lastOutgoingQueueOperation          == nil ? "never" : [lastOutgoingQueueOperation          UTF8String]);
 419             printf("lastProcessReceivedKeysOperation:    %s\n", lastProcessReceivedKeysOperation    == nil ? "never" : [lastProcessReceivedKeysOperation    UTF8String]);
 420             printf("lastReencryptOutgoingItemsOperation: %s\n", lastReencryptOutgoingItemsOperation == nil ? "never" : [lastReencryptOutgoingItemsOperation UTF8String]);
 421             printf("lastScanLocalItemsOperation:         %s\n", lastScanLocalItemsOperation         == nil ? "never" : [lastScanLocalItemsOperation         UTF8String]);
 422
 423             printf("Launch sequence:\n");
 424             for (NSString *event in launchSequence) {
 425                 printf("\t%s\n", [[event description] UTF8String]);
 426             }
 427
 428             if(status.allKeys.count > 0u) {
 429                 printf("\nExtra information: %s\n", [[status description] UTF8String]);
 430             }
 431             printf("\n");
 432         }
 433         dispatch_semaphore_signal(sema);
 434     }];
 435
 436     if(dispatch_semaphore_wait(sema, dispatch_time(DISPATCH_TIME_NOW, NSEC_PER_SEC * 30)) != 0) {
 437         printf("\n\nError: timed out waiting for response\n");
 438     }
 439 #endif // OCTAGON
 440 }
 441
 442 - (long)fetch:(NSString*)view {
 443     __block long ret = 0;
 444 #if OCTAGON
 445     dispatch_semaphore_t sema = dispatch_semaphore_create(0);
 446
 447     [self.control rpcFetchAndProcessChanges:view reply:^(NSError* error) {
 448         if(error) {
 449             printf("Error fetching: %s\n", [[error description] UTF8String]);
 450             ret = (error.code == 0 ? -1 : error.code);
 451         } else {
 452             printf("Complete.\n");
 453             ret = 0;
 454         }
 455
 456         dispatch_semaphore_signal(sema);
 457     }];
 458
 459     if(dispatch_semaphore_wait(sema, dispatch_time(DISPATCH_TIME_NOW, NSEC_PER_SEC * 65)) != 0) {
 460         printf("\n\nError: timed out waiting for response\n");
 461         return -1;
 462     }
 463 #endif // OCTAGON
 464     return ret;
 465 }
 466
 467 - (long)push:(NSString*)view {
 468     __block long ret = 0;
 469 #if OCTAGON
 470     dispatch_semaphore_t sema = dispatch_semaphore_create(0);
 471
 472     [self.control rpcPushOutgoingChanges:view reply:^(NSError* error) {
 473         if(error) {
 474             printf("Error pushing: %s\n", [[error description] UTF8String]);
 475             ret = (error.code == 0 ? -1 : error.code);
 476         } else {
 477             printf("Complete.\n");
 478             ret = 0;
 479         }
 480         dispatch_semaphore_signal(sema);
 481     }];
 482
 483     if(dispatch_semaphore_wait(sema, dispatch_time(DISPATCH_TIME_NOW, NSEC_PER_SEC * 65)) != 0) {
 484         printf("\n\nError: timed out waiting for response\n");
 485         return -1;
 486     }
 487
 488 #endif // OCTAGON
 489     return ret;
 490 }
 491
 492 - (long)ckmetric {
 493     __block long ret = 0;
 494 #if OCTAGON
 495     dispatch_semaphore_t sema = dispatch_semaphore_create(0);
 496
 497     [self.control rpcCKMetric:@"testMetric" attributes:@{ @"testAttribute" : @"value" } reply:^(NSError* error) {
 498         if(error) {
 499             printf("Error sending metric: %s\n", [[error description] UTF8String]);
 500             ret = (error.code == 0 ? -1 : error.code);
 501         } else {
 502             printf("Complete.\n");
 503             ret = 0;
 504         }
 505         dispatch_semaphore_signal(sema);
 506     }];
 507
 508     if(dispatch_semaphore_wait(sema, dispatch_time(DISPATCH_TIME_NOW, NSEC_PER_SEC * 65)) != 0) {
 509         printf("\n\nError: timed out waiting for response\n");
 510         return -1;
 511     }
 512 #endif // OCTAGON
 513     return ret;
 514
 515 }
 516
 517 @end
 518
 519
 520 static int perfCounters = false;
 521 static int status = false;
 522 static int resync = false;
 523 static int reset = false;
 524 static int resetCloudKit = false;
 525 static int fetch = false;
 526 static int push = false;
 527 static int json = false;
 528 static int ckmetric = false;
 529
 530 static char* viewArg = NULL;
 531
 532 int main(int argc, char **argv)
 533 {
 534     static struct argument options[] = {
 535         { .shortname='p', .longname="perfcounters", .flag=&perfCounters, .flagval=true, .description="Print CKKS performance counters"},
 536         { .shortname='j', .longname="json", .flag=&json, .flagval=true, .description="Output in JSON format"},
 537         { .shortname='v', .longname="view", .argument=&viewArg, .description="Operate on a single view"},
 538
 539         { .command="status", .flag=&status, .flagval=true, .description="Report status on CKKS views"},
 540         { .command="fetch", .flag=&fetch, .flagval=true, .description="Fetch all new changes in CloudKit and attempt to process them"},
 541         { .command="push", .flag=&push, .flagval=true, .description="Push all pending local changes to CloudKit"},
 542         { .command="resync", .flag=&resync, .flagval=true, .description="Resync all data with what's in CloudKit"},
 543         { .command="reset", .flag=&reset, .flagval=true, .description="All local data will be wiped, and data refetched from CloudKit"},
 544         { .command="reset-cloudkit", .flag=&resetCloudKit, .flagval=true, .description="All data in CloudKit will be removed and replaced with what's local"},
 545         { .command="ckmetric", .flag=&ckmetric, .flagval=true, .description="Push CloudKit metric"},
 546         {}
 547     };
 548
 549     static struct arguments args = {
 550         .programname="ckksctl",
 551         .description="Control and report on CKKS",
 552         .arguments = options,
 553     };
 554
 555     if(!options_parse(argc, argv, &args)) {
 556         printf("\n");
 557         print_usage(&args);
 558         return -1;
 559     }
 560
 561     @autoreleasepool {
 562         NSError* error = nil;
 563
 564         CKKSControl* rpc = [CKKSControl CKKSControlObject:false error:&error];
 565         if(error || !rpc) {
 566             errx(1, "no CKKSControl failed: %s", [[error description] UTF8String]);
 567         }
 568
 569         CKKSControlCLI* ctl = [[CKKSControlCLI alloc] initWithCKKSControl:rpc];
 570
 571         NSString* view = viewArg ? [NSString stringWithCString: viewArg encoding: NSUTF8StringEncoding] : nil;
 572
 573         if(status) {
 574             // Complicated logic, but you can choose any combination of (json, perfcounters) that you like.
 575             NSMutableDictionary *statusDict = [[NSMutableDictionary alloc] init];
 576             if(perfCounters) {
 577                 statusDict[@"performance"] = [ctl fetchPerformanceCounters];
 578             }
 579             if (json) {
 580                 statusDict[@"status"] = [ctl fetchStatus:view];
 581             }
 582             if(json || perfCounters) {
 583                print_result(statusDict, true);
 584                 printf("\n");
 585             }
 586
 587             if(!json) {
 588                 [ctl printHumanReadableStatus:view];
 589             }
 590             return 0;
 591         } else if(perfCounters) {
 592             NSMutableDictionary *statusDict = [[NSMutableDictionary alloc] init];
 593             statusDict[@"performance"] = [ctl fetchPerformanceCounters];
 594             print_result(statusDict, false);
 595
 596         } else if(fetch) {
 597             return (int)[ctl fetch:view];
 598         } else if(push) {
 599             return (int)[ctl push:view];
 600         } else if(reset) {
 601             return (int)[ctl resetLocal:view];
 602         } else if(resetCloudKit) {
 603             return (int)[ctl resetCloudKit:view];
 604         } else if(resync) {
 605             return (int)[ctl resync:view];
 606         } else if(ckmetric) {
 607             return (int)[ctl ckmetric];
 608         } else {
 609             print_usage(&args);
 610             return -1;
 611         }
 612     }
 613     return 0;
 614 }