--- /dev/null
+/*
+ * Measure performance of SecureTransport - setup and sustained data
+ * throughput. Single process version, no sockets - all data transfer
+ * between client and server is via local memory shared between two
+ * threads.
+ *
+ * Written by Doug Mitchell.
+ */
+#include <Security/SecureTransport.h>
+#include <clAppUtils/sslAppUtils.h>
+#include <utilLib/fileIo.h>
+#include <utilLib/common.h>
+#include <clAppUtils/ringBufferIo.h>
+#include <clAppUtils/sslRingBufferThreads.h>
+
+#include <CoreServices/../Frameworks/CarbonCore.framework/Headers/MacErrors.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <time.h>
+#include <ctype.h>
+#include <sys/param.h>
+#include <CoreFoundation/CoreFoundation.h>
+#include <security_utilities/devrandom.h>
+
+#define DEFAULT_KC "localcert" /* default keychain */
+#define DEFAULT_XFER (1024 * 1024 * 20) /* total xfer size in bytes */
+
+/* we might make these user-tweakable */
+#define DEFAULT_NUM_BUFS 16
+#define DEFAULT_BUF_SIZE 2048 /* in the ring buffers */
+#define DEFAULT_CHUNK 1024 /* bytes to write per SSLWrite() */
+
+static void usage(char **argv)
+{
+ printf("Usage: %s [option ...]\n", argv[0]);
+ printf("Options:\n");
+ printf(" -k keychain (default = %s)\n", DEFAULT_KC);
+ printf(" -x transferSize (default=%d; 0=forever)\n", DEFAULT_XFER);
+ printf(" -c cipher (default = RSA/AES128\n");
+ printf(" ciphers: a=RSA/AES128; r=RSA/RC4; d=RSA/DES; D=RSA/3DES;\n");
+ printf(" h=DHA/RC4; H=DH/DSS/DES; A=AES256\n");
+ printf(" -v version (t|2|3; default = t(TLS1)\n");
+ printf(" -w password (unlock server keychain with password)\n");
+ printf(" -a (enable client authentication)\n");
+ printf(" -p (pause on error)\n");
+ printf(" -m (pause for malloc debug)\n");
+ exit(1);
+}
+
+int main(int argc, char **argv)
+{
+ RingBuffer serverToClientRing;
+ RingBuffer clientToServerRing;
+ unsigned numBufs = DEFAULT_NUM_BUFS;
+ unsigned bufSize = DEFAULT_BUF_SIZE;
+ unsigned chunkSize = DEFAULT_CHUNK;
+ unsigned char clientBuf[DEFAULT_CHUNK];
+ unsigned char serverBuf[DEFAULT_CHUNK];
+ CFArrayRef idArray; /* for SSLSetCertificate */
+ CFArrayRef anchorArray; /* trusted roots */
+ SslRingBufferArgs clientArgs;
+ SslRingBufferArgs serverArgs;
+ SecKeychainRef kcRef = NULL;
+ SecCertificateRef anchorCert = NULL;
+ SecIdentityRef idRef = NULL;
+ bool abortFlag = false;
+ pthread_t client_thread = NULL;
+ int result;
+ bool diffieHellman = true; /* FIXME needs work */
+ OSStatus ortn;
+
+ /* user-spec'd variables */
+ char *kcName = DEFAULT_KC;
+ unsigned xferSize = DEFAULT_XFER;
+ SSLCipherSuite cipherSuite = TLS_RSA_WITH_AES_128_CBC_SHA;
+ SSLProtocol prot = kTLSProtocol1;
+ char password[200];
+ bool clientAuthEnable = false;
+ bool pauseOnError = false;
+ bool runForever = false;
+ bool mallocPause = false;
+
+ password[0] = 0;
+
+ extern int optind;
+ extern char *optarg;
+ int arg;
+ optind = 1;
+ while ((arg = getopt(argc, argv, "k:x:c:v:w:aBpm")) != -1) {
+ switch (arg) {
+ case 'k':
+ kcName = optarg;
+ break;
+ case 'x':
+ {
+ unsigned xsize = atoi(optarg);
+ if(xsize == 0) {
+ runForever = true;
+ /* and leave xferSize alone */
+ }
+ else {
+ xferSize = xsize;
+ }
+ break;
+ }
+ case 'c':
+ switch(optarg[0]) {
+ case 'a':
+ cipherSuite = TLS_RSA_WITH_AES_128_CBC_SHA;
+ break;
+ case 'r':
+ cipherSuite = SSL_RSA_WITH_RC4_128_SHA;
+ break;
+ case 'd':
+ cipherSuite = SSL_RSA_WITH_DES_CBC_SHA;
+ break;
+ case 'D':
+ cipherSuite = SSL_RSA_WITH_3DES_EDE_CBC_SHA;
+ break;
+ case 'h':
+ cipherSuite = SSL_DH_anon_WITH_RC4_128_MD5;
+ diffieHellman = true;
+ break;
+ case 'H':
+ cipherSuite = SSL_DHE_DSS_WITH_DES_CBC_SHA;
+ diffieHellman = true;
+ break;
+ case 'A':
+ cipherSuite = TLS_RSA_WITH_AES_256_CBC_SHA;
+ break;
+ default:
+ usage(argv);
+ }
+ break;
+ case 'v':
+ switch(optarg[0]) {
+ case 't':
+ prot = kTLSProtocol1;
+ break;
+ case '2':
+ prot = kSSLProtocol2;
+ break;
+ case '3':
+ prot = kSSLProtocol3;
+ break;
+ default:
+ usage(argv);
+ }
+ break;
+ case 'w':
+ strcpy(password, optarg);
+ break;
+ case 'a':
+ clientAuthEnable = true;
+ break;
+ case 'p':
+ pauseOnError = true;
+ break;
+ case 'm':
+ mallocPause = true;
+ break;
+ default:
+ usage(argv);
+ }
+ }
+ if(optind != argc) {
+ usage(argv);
+ }
+
+ /* set up ring buffers */
+ ringBufSetup(&serverToClientRing, "serveToClient", numBufs, bufSize);
+ ringBufSetup(&clientToServerRing, "clientToServe", numBufs, bufSize);
+
+ /* get server SecIdentity */
+ idArray = getSslCerts(kcName,
+ CSSM_FALSE, /* encryptOnly */
+ CSSM_FALSE, /* completeCertChain */
+ NULL, /* anchorFile */
+ &kcRef);
+ if(idArray == NULL) {
+ printf("***Can't get signing cert from %s\n", kcName);
+ exit(1);
+ }
+ idRef = (SecIdentityRef)CFArrayGetValueAtIndex(idArray, 0);
+ ortn = SecIdentityCopyCertificate(idRef, &anchorCert);
+ if(ortn) {
+ cssmPerror("SecIdentityCopyCertificate", ortn);
+ exit(1);
+ }
+ anchorArray = CFArrayCreate(NULL, (const void **)&anchorCert,
+ 1, &kCFTypeArrayCallBacks);
+
+ /* unlock keychain? */
+ if(password[0]) {
+ ortn = SecKeychainUnlock(kcRef, strlen(password), password, true);
+ if(ortn) {
+ cssmPerror("SecKeychainUnlock", ortn);
+ /* oh well */
+ }
+ }
+ CFRelease(kcRef);
+
+ if(mallocPause) {
+ fpurge(stdin);
+ printf("Pausing for MallocDebug setup. CR to proceed: ");
+ getchar();
+ }
+
+ /* set up server side */
+ memset(&serverArgs, 0, sizeof(serverArgs));
+ serverArgs.idArray = idArray;
+ serverArgs.trustedRoots = anchorArray;
+ serverArgs.xferSize = xferSize;
+ serverArgs.xferBuf = serverBuf;
+ serverArgs.chunkSize = chunkSize;
+ serverArgs.runForever = runForever;
+ serverArgs.cipherSuite = cipherSuite;
+ serverArgs.prot = prot;
+ serverArgs.ringWrite = &serverToClientRing;
+ serverArgs.ringRead = &clientToServerRing;
+ serverArgs.goFlag = &clientArgs.iAmReady;
+ serverArgs.abortFlag = &abortFlag;
+ serverArgs.pauseOnError = pauseOnError;
+
+ /* set up client side */
+ memset(&clientArgs, 0, sizeof(clientArgs));
+ clientArgs.idArray = NULL; /* until we do client auth */
+ clientArgs.trustedRoots = anchorArray;
+ clientArgs.xferSize = xferSize;
+ clientArgs.xferBuf = clientBuf;
+ clientArgs.chunkSize = chunkSize;
+ clientArgs.runForever = runForever;
+ clientArgs.cipherSuite = cipherSuite;
+ clientArgs.prot = prot;
+ clientArgs.ringWrite = &clientToServerRing;
+ clientArgs.ringRead = &serverToClientRing;
+ clientArgs.goFlag = &serverArgs.iAmReady;
+ clientArgs.abortFlag = &abortFlag;
+ clientArgs.pauseOnError = pauseOnError;
+
+ /* fire up client thread */
+ result = pthread_create(&client_thread, NULL,
+ sslRbClientThread, &clientArgs);
+ if(result) {
+ printf("***pthread_create returned %d, aborting\n", result);
+ exit(1);
+ }
+
+ /*
+ * And the server pseudo thread. This returns when all data has been transferred.
+ */
+ ortn = sslRbServerThread(&serverArgs);
+
+ if(abortFlag) {
+ printf("***Test aborted.\n");
+ exit(1);
+ }
+
+ printf("\n");
+
+ if(mallocPause) {
+ fpurge(stdin);
+ printf("End of test. Pausing for MallocDebug analysis. CR to proceed: ");
+ getchar();
+ }
+
+ printf("SSL Protocol Version : %s\n",
+ sslGetProtocolVersionString(serverArgs.negotiatedProt));
+ printf("SSL Cipher : %s\n",
+ sslGetCipherSuiteString(serverArgs.negotiatedCipher));
+
+ printf("SSL Handshake : %f s\n",
+ serverArgs.startData - serverArgs.startHandshake);
+ printf("Data Transfer : %u bytes in %f s\n", (unsigned)xferSize,
+ serverArgs.endData - serverArgs.startHandshake);
+ printf(" : %.1f Kbytes/s\n",
+ xferSize / (serverArgs.endData - serverArgs.startHandshake) / 1024.0);
+ return 0;
+}
+
+
+
projects / apple / security.git / blobdiff