--- /dev/null
+/*
+ * hashTimeSA.cpp - measure performance of digest ops, standalone version (no
+ * dependency on Security.framewortk or on CommonCrypto portion of libSystem).
+ */
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <CommonCrypto/CommonDigest.h> /* static lib used in Tiger */
+#include "MD5.h" /* CryptKit version used in Panther and prior */
+#include "SHA1.h" /* ditto */
+#include <Security/cssmtype.h> /* for ALGID values */
+#include <Security/cssmapple.h> /* more ALGID values */
+#include <CoreFoundation/CFDate.h>
+
+/* enumerate digest algorithms our way */
+typedef int HT_Alg;
+enum {
+ HA_MD5 = 0,
+ HA_SHA1,
+ HA_SHA224,
+ HA_SHA256,
+ HA_SHA384,
+ HA_SHA512
+};
+
+#define FIRST_ALG HA_MD5
+#define LAST_ALG HA_SHA512
+
+static void usage(char **argv)
+{
+ printf("Usage: %s c|k [option ...]\n", argv[0]);
+ printf(" c=CommonCrypto; k=CryptKit\n");
+ printf("Options:\n");
+ printf(" a=alg; default=all\n");
+ printf(" algs: m : MD5\n");
+ printf(" s : SHA1\n");
+ printf(" 4 : SHA224\n");
+ printf(" 2 : SHA256\n");
+ printf(" 3 : SHA384\n");
+ printf(" 5 : SHA512\n");
+ printf(" l=loops (only valid if testspec is given)\n");
+ printf(" v verify digest by printing it\n");
+ exit(1);
+}
+
+static void dumpDigest(
+ const unsigned char *digest,
+ unsigned len)
+{
+ for(unsigned dex=0; dex<len; dex++) {
+ printf("%02X", *digest++);
+ if((dex % 4) == 3) {
+ printf(" ");
+ }
+ }
+ printf("\n");
+}
+
+/* sort-of random, but repeatable */
+static void initPtext(
+ unsigned char *ptext,
+ unsigned len)
+{
+ srandom(1);
+ for(unsigned dex=0; dex<len; dex++) {
+ *ptext++ = random();
+ }
+}
+
+/* passed to each test */
+typedef struct {
+ unsigned loops;
+ CSSM_ALGORITHMS algId; // MD5, SHA1
+ bool dumpDigest;
+} TestParams;
+
+#define MAX_DIGEST_SIZE 64 // we provide, no malloc below CSSM
+
+#define PTEXT_SIZE 1000 // to digest in bytes
+#define INNER_LOOPS 500
+
+
+/* SHA1 digest is not orthoganal, fix up here */
+static void ckSha1Final(
+ void *ctx,
+ unsigned char *digest)
+{
+ sha1GetDigest((sha1Obj)ctx, digest);
+}
+
+typedef void (*ckInitFcn)(void *digestCtx);
+typedef void (*ckUpdateFcn)(void *digestCtx, const void *data, unsigned len);
+typedef void (*ckFinalFcn)(void *digestCtx, unsigned char *digest);
+
+static CSSM_RETURN hashDataRateCryptKit(
+ TestParams *params)
+{
+ ckUpdateFcn updatePtr = NULL;
+ ckFinalFcn finalPtr = NULL;
+ ckInitFcn initPtr = NULL;
+ struct MD5Context md5;
+ sha1Obj sha;
+ void *ctx;
+
+ unsigned loop;
+ unsigned iloop;
+ double startTime, endTime;
+ double timeSpent, timeSpentMs;
+ uint8 ptext[PTEXT_SIZE];
+ uint8 digest[MAX_DIGEST_SIZE];
+ unsigned digestLen = 16;
+
+ /* we reuse this one inside the loop */
+ switch(params->algId) {
+ case CSSM_ALGID_SHA1:
+ sha = sha1Alloc();
+ ctx = sha;
+ initPtr = (ckInitFcn)sha1Reinit;
+ updatePtr = (ckUpdateFcn)sha1AddData;
+ finalPtr = (ckFinalFcn)ckSha1Final;
+ digestLen = 20;
+ break;
+ case CSSM_ALGID_MD5:
+ ctx = &md5;
+ initPtr = (ckInitFcn)MD5Init;
+ updatePtr = (ckUpdateFcn)MD5Update;
+ finalPtr = (ckFinalFcn)MD5Final;
+ break;
+ default:
+ printf("***Sorry, CryptKit can only do SHA1 and MD5.\n");
+ return 1;
+ }
+
+ /* random data, const thru the loops */
+ initPtext(ptext, PTEXT_SIZE);
+
+ /* start critical timing loop */
+ startTime = CFAbsoluteTimeGetCurrent();
+ for(loop=0; loop<params->loops; loop++) {
+ initPtr(ctx);
+ for(iloop=0; iloop<INNER_LOOPS; iloop++) {
+ updatePtr(ctx, ptext, PTEXT_SIZE);
+ }
+ finalPtr(ctx, digest);
+ }
+ endTime = CFAbsoluteTimeGetCurrent();
+ timeSpent = endTime - startTime;
+ timeSpentMs = timeSpent * 1000.0;
+
+ float bytesPerLoop = INNER_LOOPS * PTEXT_SIZE;
+ float totalBytes = params->loops * bytesPerLoop;
+
+ /* careful, KByte = 1024, ms = 1/1000 */
+ printf(" Digest %.0f bytes : %u ops in %.2f ms; %f ms/op, %.0f KBytes/s\n",
+ bytesPerLoop, params->loops,
+ timeSpentMs, timeSpentMs / (double)params->loops,
+ ((float)totalBytes / 1024.0) / timeSpent);
+ if(params->dumpDigest) {
+ dumpDigest(digest, digestLen);
+ }
+ return CSSM_OK;
+}
+
+typedef union {
+ CC_MD5_CTX md5;
+ CC_SHA1_CTX sha;
+ CC_SHA256_CTX sha256;
+ CC_SHA512_CTX sha512;
+} CC_CTX;
+
+typedef void (*ccInitFcn)(void *digestCtx);
+typedef void (*ccUpdateFcn)(void *digestCtx, const void *data, CC_LONG len);
+typedef void (*ccFinalFcn)(unsigned char *digest, void *digestCtx);
+
+static CSSM_RETURN hashDataRateCommonCrypto(
+ TestParams *params)
+{
+ CC_CTX ctx;
+ ccUpdateFcn updatePtr = NULL;
+ ccFinalFcn finalPtr = NULL;
+ ccInitFcn initPtr = NULL;
+ unsigned loop;
+ unsigned iloop;
+ double startTime, endTime;
+ double timeSpent, timeSpentMs;
+ uint8 ptext[PTEXT_SIZE];
+ uint8 digest[MAX_DIGEST_SIZE];
+ unsigned digestLen = 16;
+
+ /* we reuse this one inside the loop */
+ switch(params->algId) {
+ case CSSM_ALGID_SHA1:
+ initPtr = (ccInitFcn)CC_SHA1_Init;
+ updatePtr = (ccUpdateFcn)CC_SHA1_Update;
+ finalPtr = (ccFinalFcn)CC_SHA1_Final;
+ digestLen = 20;
+ break;
+ case CSSM_ALGID_SHA224:
+ initPtr = (ccInitFcn)CC_SHA224_Init;
+ updatePtr = (ccUpdateFcn)CC_SHA224_Update;
+ finalPtr = (ccFinalFcn)CC_SHA224_Final;
+ digestLen = 28;
+ break;
+ case CSSM_ALGID_SHA256:
+ initPtr = (ccInitFcn)CC_SHA256_Init;
+ updatePtr = (ccUpdateFcn)CC_SHA256_Update;
+ finalPtr = (ccFinalFcn)CC_SHA256_Final;
+ digestLen = 32;
+ break;
+ case CSSM_ALGID_SHA384:
+ initPtr = (ccInitFcn)CC_SHA384_Init;
+ updatePtr = (ccUpdateFcn)CC_SHA384_Update;
+ finalPtr = (ccFinalFcn)CC_SHA384_Final;
+ digestLen = 48;
+ break;
+ case CSSM_ALGID_SHA512:
+ initPtr = (ccInitFcn)CC_SHA512_Init;
+ updatePtr = (ccUpdateFcn)CC_SHA512_Update;
+ finalPtr = (ccFinalFcn)CC_SHA512_Final;
+ digestLen = 64;
+ break;
+ case CSSM_ALGID_MD5:
+ initPtr = (ccInitFcn)CC_MD5_Init;
+ updatePtr = (ccUpdateFcn)CC_MD5_Update;
+ finalPtr = (ccFinalFcn)CC_MD5_Final;
+ digestLen = 16;
+ break;
+ default:
+ printf("***BRRRZAP!\n");
+ return 1;
+ }
+
+ /* random data, const thru the loops */
+ initPtext(ptext, PTEXT_SIZE);
+
+ /* start critical timing loop */
+ startTime = CFAbsoluteTimeGetCurrent();
+ for(loop=0; loop<params->loops; loop++) {
+ initPtr(&ctx);
+ for(iloop=0; iloop<INNER_LOOPS; iloop++) {
+ updatePtr(&ctx, ptext, PTEXT_SIZE);
+ }
+ finalPtr(digest, &ctx);
+ }
+ endTime = CFAbsoluteTimeGetCurrent();
+ timeSpent = endTime - startTime;
+ timeSpentMs = timeSpent * 1000.0;
+
+ float bytesPerLoop = INNER_LOOPS * PTEXT_SIZE;
+ float totalBytes = params->loops * bytesPerLoop;
+
+ /* careful, KByte = 1024, ms = 1/1000 */
+ printf(" Digest %.0f bytes : %u ops in %.2f ms; %f ms/op, %.0f KBytes/s\n",
+ bytesPerLoop, params->loops,
+ timeSpentMs, timeSpentMs / (double)params->loops,
+ ((float)totalBytes / 1024.0) / timeSpent);
+ if(params->dumpDigest) {
+ dumpDigest(digest, digestLen);
+ }
+ return CSSM_OK;
+}
+
+typedef CSSM_RETURN (*testRunFcn)(TestParams *testParams);
+
+/*
+ * Static declaration of a test
+ */
+typedef struct {
+ const char *testName;
+ unsigned loops;
+ testRunFcn run;
+ char testSpec; // for t=xxx cmd line opt
+} TestDefs;
+
+static TestDefs testDefsCryptKit =
+{ "Large data digest, CryptKit",
+ 1000,
+ hashDataRateCryptKit,
+ 'd',
+};
+
+static TestDefs testDefsCommonCrypto =
+{ "Large data digest, CommonCrypto",
+ 1000,
+ hashDataRateCommonCrypto,
+ 'd',
+};
+
+static void algToAlgId(
+ HT_Alg alg,
+ CSSM_ALGORITHMS *algId,
+ const char **algStr)
+{
+ switch(alg) {
+ case HA_MD5:
+ *algId = CSSM_ALGID_MD5;
+ *algStr = "MD5";
+ break;
+ case HA_SHA1:
+ *algId = CSSM_ALGID_SHA1;
+ *algStr = "SHA1";
+ break;
+ case HA_SHA224:
+ *algId = CSSM_ALGID_SHA224;
+ *algStr = "SHA224";
+ break;
+ case HA_SHA256:
+ *algId = CSSM_ALGID_SHA256;
+ *algStr = "SHA256";
+ break;
+ case HA_SHA384:
+ *algId = CSSM_ALGID_SHA384;
+ *algStr = "SHA384";
+ break;
+ case HA_SHA512:
+ *algId = CSSM_ALGID_SHA512;
+ *algStr = "SHA512";
+ break;
+ default:
+ printf("***algToAlgId screwup\n");
+ exit(1);
+ }
+}
+
+int main(int argc, char **argv)
+{
+ TestParams testParams;
+ TestDefs *testDefs = NULL;
+ CSSM_RETURN crtn;
+ int arg;
+ char *argp;
+ unsigned cmdLoops = 0; // can be specified in cmd line
+ // if not, use TestDefs.loops
+ HT_Alg alg;
+ const char *algStr;
+ int firstAlg = FIRST_ALG;
+ int lastAlg = LAST_ALG;
+
+ memset(&testParams, 0, sizeof(testParams));
+
+ if(argc < 2) {
+ usage(argv);
+ }
+ switch(argv[1][0]) {
+ case 'c':
+ testDefs = &testDefsCommonCrypto;
+ break;
+ case 'k':
+ testDefs = &testDefsCryptKit;
+ break;
+ default:
+ usage(argv);
+ }
+
+ for(arg=2; arg<argc; arg++) {
+ argp = argv[arg];
+ switch(argp[0]) {
+ case 'l':
+ cmdLoops = atoi(&argp[2]);
+ break;
+ case 'a':
+ if(argp[1] == '\0') {
+ usage(argv);
+ }
+ switch(argp[2]) {
+ case 'm':
+ firstAlg = lastAlg = HA_MD5;
+ break;
+ case 's':
+ firstAlg = lastAlg = HA_SHA1;
+ break;
+ case '4':
+ firstAlg = lastAlg = HA_SHA224;
+ break;
+ case '2':
+ firstAlg = lastAlg = HA_SHA256;
+ break;
+ case '3':
+ firstAlg = lastAlg = HA_SHA384;
+ break;
+ case '5':
+ firstAlg = lastAlg = HA_SHA512;
+ break;
+ default:
+ usage(argv);
+ }
+ break;
+ case 'v':
+ testParams.dumpDigest = true;
+ break;
+ default:
+ usage(argv);
+ }
+ }
+
+ printf("%s:\n", testDefs->testName);
+ if(cmdLoops) {
+ /* user specified */
+ testParams.loops = cmdLoops;
+ }
+ else {
+ /* default */
+ testParams.loops = testDefs->loops;
+ }
+ if((lastAlg > HA_SHA1) && (testDefs == &testDefsCryptKit)) {
+ /* CryptKit can only do MD5 and SHA1 */
+ lastAlg = HA_SHA1;
+ }
+ for(alg=firstAlg; alg<=lastAlg; alg++) {
+ algToAlgId(alg, &testParams.algId, &algStr);
+ printf(" === %s ===\n", algStr);
+ crtn = testDefs->run(&testParams);
+ if(crtn) {
+ printf("***Error detected in test, somehow....aborting.\n");
+ exit(1);
+ }
+ }
+ return 0;
+}
projects / apple / security.git / blobdiff