// // speed-test.m // libsecurity_transform // // Created by J Osborne on 8/17/10. // Copyright 2010 Apple. All rights reserved. // #import "speed-test.h" #include "SecTransform.h" #include "SecExternalSourceTransform.h" #include "SecNullTransform.h" #include <assert.h> @implementation speed_test @end UInt8 zeros[1024]; typedef void (^push_block_t)(CFDataRef d); void timed_test(NSString *name, float seconds, SecTransformRef tr, push_block_t push) { __block int num_out = 0; __block int num_in = 0; __block int timeout_out = -1; __block int timeout_in = -1; volatile __block bool done; static CFDataRef z = CFDataCreateWithBytesNoCopy(NULL, zeros, sizeof(zeros), NULL); dispatch_after(dispatch_time(DISPATCH_TIME_NOW, static_cast<int64_t>(seconds * NSEC_PER_SEC)), dispatch_get_global_queue(DISPATCH_QUEUE_PRIORITY_HIGH, 0), ^{ done = true; timeout_out = num_out; timeout_in = num_in; }); dispatch_group_t dg = dispatch_group_create(); dispatch_group_enter(dg); dispatch_queue_t q = dispatch_queue_create("counter", NULL); SecTransformExecuteAsync(tr, q, ^(CFTypeRef message, CFErrorRef error, Boolean isFinal) { if (message) { num_out++; } if (error) { NSLog(@"Error %@ while running %@", error, name); } if (isFinal) { dispatch_group_leave(dg); } }); while (!done) { push(z); num_in++; } push(NULL); dispatch_group_wait(dg, DISPATCH_TIME_FOREVER); NSString *m = [NSString stringWithFormat:@"%@ %d in, %d out times in %f seconds, %d stragglers\n", name, timeout_in, timeout_out, seconds, num_out - timeout_out]; [m writeToFile:@"/dev/stdout" atomically:NO encoding:NSUTF8StringEncoding error:NULL]; } int main(int argc, char *argv[]) { NSAutoreleasePool *ap = [[NSAutoreleasePool alloc] init]; float seconds = 5.0; { SecTransformRef x = SecExternalSourceTransformCreate(NULL); //SecTransformRef t = SecEncodeTransformCreate(kSecBase64Encoding, NULL); SecTransformRef t = SecNullTransformCreate(); SecTransformRef g = SecTransformCreateGroupTransform(); assert(x && t && g); SecTransformConnectTransforms(x, kSecTransformOutputAttributeName, t, kSecTransformInputAttributeName, g, NULL); timed_test(@"external source", seconds, t, ^(CFDataRef d){ SecExternalSourceSetValue(x, d, NULL); }); } // This second test has issues with the stock transform framwork -- it don't think the graph is valid (missing input) { //SecTransformRef t = SecEncodeTransformCreate(kSecBase64Encoding, NULL); SecTransformRef t = SecNullTransformCreate(); assert(t); timed_test(@"set INPUT", seconds, t, ^(CFDataRef d){ SecTransformSetAttribute(t, kSecTransformInputAttributeName, d, NULL); }); } }