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 shortenOutput:(BOOL)shortenOutput {
 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             NSString *syncingPolicy = pop(global, @"policy", NSString);
 310             NSString *viewsFromPolicy = pop(global, @"viewsFromPolicy", NSString);
 311
 312             if(!shortenOutput) {
 313                 printf("================================================================================\n\n");
 314                 printf("Global state:\n\n");
 315             }
 316
 317             printf("Syncing Policy:       %s\n", [[syncingPolicy description] UTF8String]);
 318             printf("Views from policy:    %s\n", [[viewsFromPolicy description] UTF8String]);
 319
 320             if(!shortenOutput) {
 321                 printf("Reachability:         %s\n", [[reachability description] UTF8String]);
 322                 printf("Retry:                %s\n", [[retry description] UTF8String]);
 323                 printf("CK DeviceID:          %s\n", [[ckdeviceID description] UTF8String]);
 324                 printf("CK DeviceID Error:    %s\n", [[ckdeviceIDError description] UTF8String]);
 325                 printf("Lock state:           %s\n", [[lockStateTracker description] UTF8String]);
 326                 printf("Last CKKS push:       %s\n", [[lastCKKSPush description] UTF8String]);
 327             }
 328
 329             printf("\n");
 330         }
 331
 332         NSArray* remainingViews = result.count > 1 ? [result subarrayWithRange:NSMakeRange(1, result.count-1)] : @[];
 333
 334         if(remainingViews.count == 0u) {
 335             printf("No CKKS views are active.\n");
 336         }
 337
 338         for(NSDictionary* viewStatus in remainingViews) {
 339             if(shortenOutput) {
 340                 NSMutableDictionary* status = [viewStatus mutableCopy];
 341
 342                 NSString* viewName = pop(status, @"view", NSString);
 343                 NSString* keystate = pop(status, @"keystate", NSString);
 344
 345                 printf("%-25s: %s\n", [viewName UTF8String], [keystate UTF8String]);
 346                 continue;
 347             }
 348
 349             NSMutableDictionary* status = [viewStatus mutableCopy];
 350
 351             NSString* viewName = pop(status,@"view", NSString);
 352             NSString* accountStatus = pop(status,@"ckaccountstatus", NSString);
 353             NSString* accountTracker = pop(status,@"accounttracker", NSString);
 354             NSString* fetcher = pop(status,@"fetcher", NSString);
 355             NSString* zoneCreated = pop(status,@"zoneCreated", NSString);
 356             NSString* zoneSubscribed = pop(status,@"zoneSubscribed", NSString);
 357             NSString* zoneInitializeScheduler = pop(status,@"zoneInitializeScheduler", NSString);
 358             NSString* keystate = pop(status,@"keystate", NSString);
 359             NSString* statusError = pop(status,@"statusError", NSString);
 360             NSString* itemSyncEnabled = pop(status,@"itemsyncing", NSString);
 361             NSString* currentTLK =    pop(status,@"currentTLK", NSString);
 362             NSString* currentClassA = pop(status,@"currentClassA", NSString);
 363             NSString* currentClassC = pop(status,@"currentClassC", NSString);
 364             NSString* currentTLKPtr =    pop(status,@"currentTLKPtr", NSString);
 365             NSString* currentClassAPtr = pop(status,@"currentClassAPtr", NSString);
 366             NSString* currentClassCPtr = pop(status,@"currentClassCPtr", NSString);
 367             NSArray* launchSequence = pop(status, @"launchSequence", NSArray);
 368
 369             NSDictionary* oqe = pop(status,@"oqe", NSDictionary);
 370             NSDictionary* iqe = pop(status,@"iqe", NSDictionary);
 371             NSDictionary* keys = pop(status,@"keys", NSDictionary);
 372             NSDictionary* ckmirror = pop(status,@"ckmirror", NSDictionary);
 373             NSArray* devicestates = pop(status, @"devicestates", NSArray);
 374             NSArray* tlkshares = pop(status, @"tlkshares", NSArray);
 375
 376             NSString* lastIncomingQueueOperation          = pop(status,@"lastIncomingQueueOperation", NSString);
 377             NSString* lastNewTLKOperation                 = pop(status,@"lastNewTLKOperation", NSString);
 378             NSString* lastOutgoingQueueOperation          = pop(status,@"lastOutgoingQueueOperation", NSString);
 379             NSString* lastProcessReceivedKeysOperation    = pop(status,@"lastProcessReceivedKeysOperation", NSString);
 380             NSString* lastReencryptOutgoingItemsOperation = pop(status,@"lastReencryptOutgoingItemsOperation", NSString);
 381
 382             printf("================================================================================\n\n");
 383
 384             printf("View: %s\n\n", [viewName UTF8String]);
 385
 386             if(statusError != nil) {
 387                 printf("ERROR FETCHING STATUS: %s\n\n", [statusError UTF8String]);
 388             }
 389
 390             printf("CloudKit account:     %s\n", [accountStatus UTF8String]);
 391             printf("Account tracker:      %s\n", [accountTracker UTF8String]);
 392
 393             if(!([zoneCreated isEqualToString:@"yes"] && [zoneSubscribed isEqualToString:@"yes"])) {
 394                 printf("CK Zone Created:            %s\n", [[zoneCreated description] UTF8String]);
 395                 printf("CK Zone Subscribed:         %s\n", [[zoneSubscribed description] UTF8String]);
 396                 printf("CK Zone initialize retry:   %s\n", [[zoneInitializeScheduler description] UTF8String]);
 397                 printf("\n");
 398             }
 399
 400             printf("Key state:            %s\n", [keystate UTF8String]);
 401             printf("Current TLK:          %s\n", currentTLK != nil
 402                    ? [currentTLK    UTF8String]
 403                    : [[NSString stringWithFormat:@"missing; pointer is %@", currentTLKPtr] UTF8String]);
 404             printf("Current ClassA:       %s\n", currentClassA != nil
 405                    ? [currentClassA UTF8String]
 406                    : [[NSString stringWithFormat:@"missing; pointer is %@", currentClassAPtr] UTF8String]);
 407             printf("Current ClassC:       %s\n", currentClassC != nil
 408                    ? [currentClassC UTF8String]
 409                    : [[NSString stringWithFormat:@"missing; pointer is %@", currentClassCPtr] UTF8String]);
 410
 411             printf("TLK shares:           %s\n", [[tlkshares description] UTF8String]);
 412
 413             printf("Item syncing:          %s\n", [[itemSyncEnabled description] UTF8String]);
 414             printf("Outgoing Queue counts: %s\n", [[oqe description] UTF8String]);
 415             printf("Incoming Queue counts: %s\n", [[iqe description] UTF8String]);
 416             printf("Key counts: %s\n", [[keys description] UTF8String]);
 417
 418             printf("Item counts (by key):  %s\n", [[ckmirror description] UTF8String]);
 419             printf("Peer states:           %s\n", [[devicestates description] UTF8String]);
 420
 421             printf("zone change fetcher:                 %s\n", [[fetcher description] UTF8String]);
 422             printf("lastIncomingQueueOperation:          %s\n", lastIncomingQueueOperation          == nil ? "never" : [lastIncomingQueueOperation          UTF8String]);
 423             printf("lastNewTLKOperation:                 %s\n", lastNewTLKOperation                 == nil ? "never" : [lastNewTLKOperation                 UTF8String]);
 424             printf("lastOutgoingQueueOperation:          %s\n", lastOutgoingQueueOperation          == nil ? "never" : [lastOutgoingQueueOperation          UTF8String]);
 425             printf("lastProcessReceivedKeysOperation:    %s\n", lastProcessReceivedKeysOperation    == nil ? "never" : [lastProcessReceivedKeysOperation    UTF8String]);
 426             printf("lastReencryptOutgoingItemsOperation: %s\n", lastReencryptOutgoingItemsOperation == nil ? "never" : [lastReencryptOutgoingItemsOperation UTF8String]);
 427
 428             printf("Launch sequence:\n");
 429             for (NSString *event in launchSequence) {
 430                 printf("\t%s\n", [[event description] UTF8String]);
 431             }
 432
 433             if(status.allKeys.count > 0u) {
 434                 printf("\nExtra information: %s\n", [[status description] UTF8String]);
 435             }
 436             printf("\n");
 437         }
 438         dispatch_semaphore_signal(sema);
 439     }];
 440
 441     if(dispatch_semaphore_wait(sema, dispatch_time(DISPATCH_TIME_NOW, NSEC_PER_SEC * 30)) != 0) {
 442         printf("\n\nError: timed out waiting for response\n");
 443     }
 444 #endif // OCTAGON
 445 }
 446
 447 - (long)fetch:(NSString*)view {
 448     __block long ret = 0;
 449 #if OCTAGON
 450     dispatch_semaphore_t sema = dispatch_semaphore_create(0);
 451
 452     [self.control rpcFetchAndProcessChanges:view reply:^(NSError* error) {
 453         if(error) {
 454             printf("Error fetching: %s\n", [[error description] UTF8String]);
 455             ret = (error.code == 0 ? -1 : error.code);
 456         } else {
 457             printf("Complete.\n");
 458             ret = 0;
 459         }
 460
 461         dispatch_semaphore_signal(sema);
 462     }];
 463
 464     // The maximum device-side delay to start a fetch is 120s, so we must wait longer than that for a response.
 465     if(dispatch_semaphore_wait(sema, dispatch_time(DISPATCH_TIME_NOW, NSEC_PER_SEC * 135)) != 0) {
 466         printf("\n\nError: timed out waiting for response\n");
 467         return -1;
 468     }
 469 #endif // OCTAGON
 470     return ret;
 471 }
 472
 473 - (long)push:(NSString*)view {
 474     __block long ret = 0;
 475 #if OCTAGON
 476     dispatch_semaphore_t sema = dispatch_semaphore_create(0);
 477
 478     [self.control rpcPushOutgoingChanges:view reply:^(NSError* error) {
 479         if(error) {
 480             printf("Error pushing: %s\n", [[error description] UTF8String]);
 481             ret = (error.code == 0 ? -1 : error.code);
 482         } else {
 483             printf("Complete.\n");
 484             ret = 0;
 485         }
 486         dispatch_semaphore_signal(sema);
 487     }];
 488
 489     if(dispatch_semaphore_wait(sema, dispatch_time(DISPATCH_TIME_NOW, NSEC_PER_SEC * 65)) != 0) {
 490         printf("\n\nError: timed out waiting for response\n");
 491         return -1;
 492     }
 493
 494 #endif // OCTAGON
 495     return ret;
 496 }
 497
 498 - (long)ckmetric {
 499     __block long ret = 0;
 500 #if OCTAGON
 501     dispatch_semaphore_t sema = dispatch_semaphore_create(0);
 502
 503     [self.control rpcCKMetric:@"testMetric" attributes:@{ @"testAttribute" : @"value" } reply:^(NSError* error) {
 504         if(error) {
 505             printf("Error sending metric: %s\n", [[error description] UTF8String]);
 506             ret = (error.code == 0 ? -1 : error.code);
 507         } else {
 508             printf("Complete.\n");
 509             ret = 0;
 510         }
 511         dispatch_semaphore_signal(sema);
 512     }];
 513
 514     if(dispatch_semaphore_wait(sema, dispatch_time(DISPATCH_TIME_NOW, NSEC_PER_SEC * 65)) != 0) {
 515         printf("\n\nError: timed out waiting for response\n");
 516         return -1;
 517     }
 518 #endif // OCTAGON
 519     return ret;
 520
 521 }
 522
 523 @end
 524
 525
 526 static int perfCounters = false;
 527 static int status = false;
 528 static int resync = false;
 529 static int reset = false;
 530 static int resetCloudKit = false;
 531 static int fetch = false;
 532 static int push = false;
 533 static int json = false;
 534 static int shortOutput = false;
 535 static int ckmetric = false;
 536
 537 static char* viewArg = NULL;
 538
 539 int main(int argc, char **argv)
 540 {
 541     static struct argument options[] = {
 542         { .shortname='p', .longname="perfcounters", .flag=&perfCounters, .flagval=true, .description="Print CKKS performance counters"},
 543         { .shortname='j', .longname="json", .flag=&json, .flagval=true, .description="Output in JSON format"},
 544         { .shortname='s', .longname="short", .flag=&shortOutput, .flagval=true, .description="Output a short format"},
 545         { .shortname='v', .longname="view", .argument=&viewArg, .description="Operate on a single view"},
 546
 547         { .command="status", .flag=&status, .flagval=true, .description="Report status on CKKS views"},
 548         { .command="fetch", .flag=&fetch, .flagval=true, .description="Fetch all new changes in CloudKit and attempt to process them"},
 549         { .command="push", .flag=&push, .flagval=true, .description="Push all pending local changes to CloudKit"},
 550         { .command="resync", .flag=&resync, .flagval=true, .description="Resync all data with what's in CloudKit"},
 551         { .command="reset", .flag=&reset, .flagval=true, .description="All local data will be wiped, and data refetched from CloudKit"},
 552         { .command="reset-cloudkit", .flag=&resetCloudKit, .flagval=true, .description="All data in CloudKit will be removed and replaced with what's local"},
 553         { .command="ckmetric", .flag=&ckmetric, .flagval=true, .description="Push CloudKit metric"},
 554         {}
 555     };
 556
 557     static struct arguments args = {
 558         .programname="ckksctl",
 559         .description="Control and report on CKKS",
 560         .arguments = options,
 561     };
 562
 563     if(!options_parse(argc, argv, &args)) {
 564         printf("\n");
 565         print_usage(&args);
 566         return -1;
 567     }
 568
 569     @autoreleasepool {
 570         NSError* error = nil;
 571
 572         CKKSControl* rpc = [CKKSControl CKKSControlObject:false error:&error];
 573         if(error || !rpc) {
 574             errx(1, "no CKKSControl failed: %s", [[error description] UTF8String]);
 575         }
 576
 577         CKKSControlCLI* ctl = [[CKKSControlCLI alloc] initWithCKKSControl:rpc];
 578
 579         NSString* view = viewArg ? [NSString stringWithCString: viewArg encoding: NSUTF8StringEncoding] : nil;
 580
 581         if(status) {
 582             // Complicated logic, but you can choose any combination of (json, perfcounters) that you like.
 583             NSMutableDictionary *statusDict = [[NSMutableDictionary alloc] init];
 584             if(perfCounters) {
 585                 statusDict[@"performance"] = [ctl fetchPerformanceCounters];
 586             }
 587             if (json) {
 588                 statusDict[@"status"] = [ctl fetchStatus:view];
 589             }
 590             if(json || perfCounters) {
 591                print_result(statusDict, true);
 592                 printf("\n");
 593             }
 594
 595             if(!json) {
 596                 [ctl printHumanReadableStatus:view shortenOutput:shortOutput];
 597             }
 598             return 0;
 599         } else if(perfCounters) {
 600             NSMutableDictionary *statusDict = [[NSMutableDictionary alloc] init];
 601             statusDict[@"performance"] = [ctl fetchPerformanceCounters];
 602             print_result(statusDict, false);
 603
 604         } else if(fetch) {
 605             return (int)[ctl fetch:view];
 606         } else if(push) {
 607             return (int)[ctl push:view];
 608         } else if(reset) {
 609             return (int)[ctl resetLocal:view];
 610         } else if(resetCloudKit) {
 611             return (int)[ctl resetCloudKit:view];
 612         } else if(resync) {
 613             return (int)[ctl resync:view];
 614         } else if(ckmetric) {
 615             return (int)[ctl ckmetric];
 616         } else {
 617             print_usage(&args);
 618             return -1;
 619         }
 620     }
 621     return 0;
 622 }