gecko/security/nss/cmd/libpkix/perf/nss_threads.c
2008-06-06 08:40:11 -04:00

198 lines
6.2 KiB
C

/* ***** BEGIN LICENSE BLOCK *****
* Version: MPL 1.1/GPL 2.0/LGPL 2.1
*
* The contents of this file are subject to the Mozilla Public License Version
* 1.1 (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
* http://www.mozilla.org/MPL/
*
* Software distributed under the License is distributed on an "AS IS" basis,
* WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
* for the specific language governing rights and limitations under the
* License.
*
* The Original Code is the PKIX-C library.
*
* The Initial Developer of the Original Code is
* Sun Microsystems, Inc.
* Portions created by the Initial Developer are
* Copyright 2004-2007 Sun Microsystems, Inc. All Rights Reserved.
*
* Contributor(s):
* Sun Microsystems, Inc.
*
* Alternatively, the contents of this file may be used under the terms of
* either the GNU General Public License Version 2 or later (the "GPL"), or
* the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
* in which case the provisions of the GPL or the LGPL are applicable instead
* of those above. If you wish to allow use of your version of this file only
* under the terms of either the GPL or the LGPL, and not to allow others to
* use your version of this file under the terms of the MPL, indicate your
* decision by deleting the provisions above and replace them with the notice
* and other provisions required by the GPL or the LGPL. If you do not delete
* the provisions above, a recipient may use your version of this file under
* the terms of any one of the MPL, the GPL or the LGPL.
*
* ***** END LICENSE BLOCK ***** */
/*
* nssThreads.c
*
* NSS Performance Evaluation application (multi-threaded)
*
*/
#include <stdio.h>
#include <string.h>
#include "secutil.h"
#include "nspr.h"
#include "prtypes.h"
#include "prtime.h"
#include "prlong.h"
#include "pk11func.h"
#include "secasn1.h"
#include "cert.h"
#include "cryptohi.h"
#include "secoid.h"
#include "certdb.h"
#include "nss.h"
typedef struct ThreadDataStr tData;
struct ThreadDataStr {
CERTCertificate* cert;
PRIntervalTime duration;
PRUint32 iterations;
};
static void ThreadEntry(void* data)
{
tData* tdata = (tData*) data;
PRIntervalTime duration = tdata->duration;
PRTime now = PR_Now();
PRIntervalTime start = PR_IntervalNow();
PR_ASSERT(duration);
if (!duration)
{
return;
}
do {
SECStatus rv = CERT_VerifyCertificate
(CERT_GetDefaultCertDB(),
tdata->cert,
PR_TRUE,
certificateUsageEmailSigner,
now,
NULL,
NULL,
NULL);
if (rv != SECSuccess)
{
(void) fprintf(stderr, "Validation failed.\n");
PORT_Assert(0);
return;
}
tdata->iterations ++;
} while ((PR_IntervalNow() - start) < duration);
}
static void Test(CERTCertificate* cert, PRIntervalTime duration, PRUint32 threads)
{
tData data;
tData** alldata;
PRIntervalTime starttime, endtime, elapsed;
PRUint32 msecs;
float total = 0;
PRThread** pthreads = NULL;
PRUint32 i = 0;
data.duration = duration;
data.cert = cert;
data.iterations = 0;
starttime = PR_IntervalNow();
pthreads = (PRThread**)PR_Malloc(threads*sizeof (PRThread*));
alldata = (tData**)PR_Malloc(threads*sizeof (tData*));
for (i = 0; i < threads; i++)
{
alldata[i] = (tData*)PR_Malloc(sizeof (tData));
*alldata[i] = data;
pthreads[i] =
PR_CreateThread(PR_USER_THREAD,
ThreadEntry,
(void*) alldata[i],
PR_PRIORITY_NORMAL,
PR_GLOBAL_THREAD,
PR_JOINABLE_THREAD,
0);
}
for (i = 0; i < threads; i++)
{
tData* args = alldata[i];
PR_JoinThread(pthreads[i]);
total += args->iterations;
PR_Free((void*)args);
}
PR_Free((void*) pthreads);
PR_Free((void*) alldata);
endtime = PR_IntervalNow();
endtime = PR_IntervalNow();
elapsed = endtime - starttime;
msecs = PR_IntervalToMilliseconds(elapsed);
total /= msecs;
total *= 1000;
(void) fprintf(stdout, "%f operations per second.\n", total);
}
static void finish(char* message, int code)
{
(void) printf(message);
exit(code);
}
static void usage(char* progname)
{
(void) printf("Usage : %s <duration> <threads> <certnickname>\n\n",
progname);
finish("", 0);
}
int nss_threads(int argc, char** argv)
{
SECStatus rv = SECSuccess;
CERTCertDBHandle *handle = NULL;
CERTCertificate* cert = NULL;
PRIntervalTime duration = PR_SecondsToInterval(1);
PRUint32 threads = 1;
if (argc != 4)
{
usage(argv[0]);
}
if (atoi(argv[1]) > 0)
{
duration = PR_SecondsToInterval(atoi(argv[1]));
}
if (atoi(argv[2]) > 0)
{
threads = atoi(argv[2]);
}
handle = CERT_GetDefaultCertDB();
PR_ASSERT(handle);
cert = CERT_FindCertByNicknameOrEmailAddr(handle, argv[3]);
if (!cert)
{
finish("Unable to find certificate.\n", 1);
}
Test(cert, duration, threads);
CERT_DestroyCertificate(cert);
return (0);
}