Security-59306.11.20.tar.gz

[apple/security.git] / keychain / analytics / TestResourceUsage.m

diff --git a/keychain/analytics/TestResourceUsage.m b/keychain/analytics/TestResourceUsage.m
new file mode 100644 (file)
index 0000000..eb4cea5
--- /dev/null
+++ b/keychain/analytics/TestResourceUsage.m
@@ -0,0 +1,109 @@
+/*
+ * Copyright (c) 2018 Apple Inc. All Rights Reserved.
+ *
+ * @APPLE_LICENSE_HEADER_START@
+ *
+ * This file contains Original Code and/or Modifications of Original Code
+ * as defined in and that are subject to the Apple Public Source License
+ * Version 2.0 (the 'License'). You may not use this file except in
+ * compliance with the License. Please obtain a copy of the License at
+ * http://www.opensource.apple.com/apsl/ and read it before using this
+ * file.
+ *
+ * The Original Code and all software distributed under the License are
+ * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
+ * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
+ * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
+ * Please see the License for the specific language governing rights and
+ * limitations under the License.
+ *
+ * @APPLE_LICENSE_HEADER_END@
+ */
+
+#import "TestResourceUsage.h"
+#import <libproc.h>
+#import <mach/mach_time.h>
+#import <mach/message.h>
+#import <os/assumes.h>
+#import <stdio.h>
+
+@interface TestResourceUsage () <XCTestObservation>
+@property (assign) rusage_info_current startResource;
+@property (assign) uint64_t startMachTime;
+@property (assign) uint64_t stopMachTime;
+@end
+
+
+@implementation TestResourceUsage
+
++ (TestResourceUsage *)sharedManager
+{
+    static TestResourceUsage *manager = nil;
+    static dispatch_once_t onceToken;
+    dispatch_once(&onceToken, ^{
+        manager = [TestResourceUsage new];
+    });
+    return manager;
+}
+
++ (void)monitorTestResourceUsage
+{
+    static dispatch_once_t onceToken;
+    TestResourceUsage *manager = [TestResourceUsage sharedManager];
+    dispatch_once(&onceToken, ^{
+        [[XCTestObservationCenter sharedTestObservationCenter] addTestObserver:manager];
+    });
+}
+
+- (void)testCaseWillStart:(XCTestCase *)testCase
+{
+    proc_pid_rusage(getpid(), RUSAGE_INFO_CURRENT, (rusage_info_t *)&_startResource);
+    _startMachTime = mach_absolute_time();
+}
+
+- (void)testCaseDidFinish:(XCTestCase *)testCase
+{
+    _stopMachTime = mach_absolute_time();
+
+    rusage_info_current endResource;
+    proc_pid_rusage(getpid(), RUSAGE_INFO_CURRENT, (rusage_info_t *)&endResource);
+
+    uint64_t startUserTime = [self nsecondsFromMachTime:self.startResource.ri_user_time];
+    uint64_t endUserTime = [self nsecondsFromMachTime:endResource.ri_user_time];
+    uint64_t startSysrTime = [self nsecondsFromMachTime:self.startResource.ri_system_time];
+    uint64_t endSysTime = [self nsecondsFromMachTime:endResource.ri_system_time];
+
+    uint64_t diskUsage = endResource.ri_logical_writes - self.startResource.ri_logical_writes;
+
+    [self BATSUnitTestToken:@"DiskUsage" value:@(diskUsage) testcase:testCase];
+    [self BATSUnitTestToken:@"CPUUserTime" value:@(endUserTime - startUserTime) testcase:testCase];
+    [self BATSUnitTestToken:@"CPUSysTime" value:@(endSysTime - startSysrTime) testcase:testCase];
+    [self BATSUnitTestToken:@"WallTime" value:@(_stopMachTime - _startMachTime) testcase:testCase];
+}
+
+- (void)BATSUnitTestToken:(NSString *)name value:(id)value testcase:(XCTestCase *)testCase
+{
+    printf("%s", [[NSString stringWithFormat:@"[RESULT_KEY] TestResourceUsage:%s:%@\n[RESULT_VALUE] %@\n",
+                   object_getClassName([testCase class]), name, value] UTF8String]);
+}
+
+- (uint64_t)nsecondsFromMachTime:(uint64_t)machTime
+{
+    static dispatch_once_t once;
+    static uint64_t ratio;
+    dispatch_once(&once, ^{
+        mach_timebase_info_data_t tbi;
+        if (os_assumes_zero(mach_timebase_info(&tbi)) == KERN_SUCCESS) {
+            ratio = tbi.numer / tbi.denom;
+        } else {
+            ratio = 1;
+        }
+    });
+
+    return (machTime * ratio);
+}
+
+
+
+@end