1*00b67f09SDavid van Moolenbroek /* $NetBSD: random.c,v 1.1.1.4 2014/12/10 03:34:28 christos Exp $ */
2*00b67f09SDavid van Moolenbroek
3*00b67f09SDavid van Moolenbroek /*
4*00b67f09SDavid van Moolenbroek * Copyright (C) 2014 Internet Systems Consortium, Inc. ("ISC")
5*00b67f09SDavid van Moolenbroek *
6*00b67f09SDavid van Moolenbroek * Permission to use, copy, modify, and/or distribute this software for any
7*00b67f09SDavid van Moolenbroek * purpose with or without fee is hereby granted, provided that the above
8*00b67f09SDavid van Moolenbroek * copyright notice and this permission notice appear in all copies.
9*00b67f09SDavid van Moolenbroek *
10*00b67f09SDavid van Moolenbroek * THE SOFTWARE IS PROVIDED "AS IS" AND ISC DISCLAIMS ALL WARRANTIES WITH
11*00b67f09SDavid van Moolenbroek * REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
12*00b67f09SDavid van Moolenbroek * AND FITNESS. IN NO EVENT SHALL ISC BE LIABLE FOR ANY SPECIAL, DIRECT,
13*00b67f09SDavid van Moolenbroek * INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
14*00b67f09SDavid van Moolenbroek * LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE
15*00b67f09SDavid van Moolenbroek * OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
16*00b67f09SDavid van Moolenbroek * PERFORMANCE OF THIS SOFTWARE.
17*00b67f09SDavid van Moolenbroek */
18*00b67f09SDavid van Moolenbroek
19*00b67f09SDavid van Moolenbroek /*
20*00b67f09SDavid van Moolenbroek * Portions copyright (c) 2008 Nominet UK. All rights reserved.
21*00b67f09SDavid van Moolenbroek *
22*00b67f09SDavid van Moolenbroek * Redistribution and use in source and binary forms, with or without
23*00b67f09SDavid van Moolenbroek * modification, are permitted provided that the following conditions
24*00b67f09SDavid van Moolenbroek * are met:
25*00b67f09SDavid van Moolenbroek * 1. Redistributions of source code must retain the above copyright
26*00b67f09SDavid van Moolenbroek * notice, this list of conditions and the following disclaimer.
27*00b67f09SDavid van Moolenbroek * 2. Redistributions in binary form must reproduce the above copyright
28*00b67f09SDavid van Moolenbroek * notice, this list of conditions and the following disclaimer in the
29*00b67f09SDavid van Moolenbroek * documentation and/or other materials provided with the distribution.
30*00b67f09SDavid van Moolenbroek *
31*00b67f09SDavid van Moolenbroek * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
32*00b67f09SDavid van Moolenbroek * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
33*00b67f09SDavid van Moolenbroek * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
34*00b67f09SDavid van Moolenbroek * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
35*00b67f09SDavid van Moolenbroek * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
36*00b67f09SDavid van Moolenbroek * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
37*00b67f09SDavid van Moolenbroek * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
38*00b67f09SDavid van Moolenbroek * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
39*00b67f09SDavid van Moolenbroek * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
40*00b67f09SDavid van Moolenbroek * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
41*00b67f09SDavid van Moolenbroek */
42*00b67f09SDavid van Moolenbroek
43*00b67f09SDavid van Moolenbroek /* Id */
44*00b67f09SDavid van Moolenbroek
45*00b67f09SDavid van Moolenbroek /* random [-m module] [-s $slot] [-n count] */
46*00b67f09SDavid van Moolenbroek
47*00b67f09SDavid van Moolenbroek /*! \file */
48*00b67f09SDavid van Moolenbroek
49*00b67f09SDavid van Moolenbroek #include <config.h>
50*00b67f09SDavid van Moolenbroek
51*00b67f09SDavid van Moolenbroek #include <stdio.h>
52*00b67f09SDavid van Moolenbroek #include <stdlib.h>
53*00b67f09SDavid van Moolenbroek #include <string.h>
54*00b67f09SDavid van Moolenbroek #include <time.h>
55*00b67f09SDavid van Moolenbroek #include <unistd.h>
56*00b67f09SDavid van Moolenbroek
57*00b67f09SDavid van Moolenbroek #include <isc/commandline.h>
58*00b67f09SDavid van Moolenbroek #include <isc/result.h>
59*00b67f09SDavid van Moolenbroek #include <isc/types.h>
60*00b67f09SDavid van Moolenbroek
61*00b67f09SDavid van Moolenbroek #include <pk11/pk11.h>
62*00b67f09SDavid van Moolenbroek #include <pk11/result.h>
63*00b67f09SDavid van Moolenbroek
64*00b67f09SDavid van Moolenbroek #ifndef HAVE_CLOCK_GETTIME
65*00b67f09SDavid van Moolenbroek #ifndef CLOCK_REALTIME
66*00b67f09SDavid van Moolenbroek #define CLOCK_REALTIME 0
67*00b67f09SDavid van Moolenbroek #endif
68*00b67f09SDavid van Moolenbroek
69*00b67f09SDavid van Moolenbroek int
clock_gettime(int32_t id,struct timespec * tp)70*00b67f09SDavid van Moolenbroek clock_gettime(int32_t id, struct timespec *tp)
71*00b67f09SDavid van Moolenbroek {
72*00b67f09SDavid van Moolenbroek struct timeval tv;
73*00b67f09SDavid van Moolenbroek int result;
74*00b67f09SDavid van Moolenbroek
75*00b67f09SDavid van Moolenbroek result = gettimeofday(&tv, NULL);
76*00b67f09SDavid van Moolenbroek if (result)
77*00b67f09SDavid van Moolenbroek return (result);
78*00b67f09SDavid van Moolenbroek tp->tv_sec = tv.tv_sec;
79*00b67f09SDavid van Moolenbroek tp->tv_nsec = (long) tv.tv_usec * 1000;
80*00b67f09SDavid van Moolenbroek return (result);
81*00b67f09SDavid van Moolenbroek }
82*00b67f09SDavid van Moolenbroek #endif
83*00b67f09SDavid van Moolenbroek
84*00b67f09SDavid van Moolenbroek CK_BYTE buf[1024];
85*00b67f09SDavid van Moolenbroek
86*00b67f09SDavid van Moolenbroek int
main(int argc,char * argv[])87*00b67f09SDavid van Moolenbroek main(int argc, char *argv[]) {
88*00b67f09SDavid van Moolenbroek isc_result_t result;
89*00b67f09SDavid van Moolenbroek CK_RV rv;
90*00b67f09SDavid van Moolenbroek CK_SLOT_ID slot = 0;
91*00b67f09SDavid van Moolenbroek CK_SESSION_HANDLE hSession = CK_INVALID_HANDLE;
92*00b67f09SDavid van Moolenbroek CK_ULONG len = sizeof(buf);
93*00b67f09SDavid van Moolenbroek pk11_context_t pctx;
94*00b67f09SDavid van Moolenbroek pk11_optype_t op_type = OP_RAND;
95*00b67f09SDavid van Moolenbroek char *lib_name = NULL;
96*00b67f09SDavid van Moolenbroek int error = 0;
97*00b67f09SDavid van Moolenbroek int c, errflg = 0;
98*00b67f09SDavid van Moolenbroek unsigned int count = 1000;
99*00b67f09SDavid van Moolenbroek unsigned int i;
100*00b67f09SDavid van Moolenbroek struct timespec starttime;
101*00b67f09SDavid van Moolenbroek struct timespec endtime;
102*00b67f09SDavid van Moolenbroek
103*00b67f09SDavid van Moolenbroek while ((c = isc_commandline_parse(argc, argv, ":m:s:n:")) != -1) {
104*00b67f09SDavid van Moolenbroek switch (c) {
105*00b67f09SDavid van Moolenbroek case 'm':
106*00b67f09SDavid van Moolenbroek lib_name = isc_commandline_argument;
107*00b67f09SDavid van Moolenbroek break;
108*00b67f09SDavid van Moolenbroek case 's':
109*00b67f09SDavid van Moolenbroek slot = atoi(isc_commandline_argument);
110*00b67f09SDavid van Moolenbroek op_type = OP_ANY;
111*00b67f09SDavid van Moolenbroek break;
112*00b67f09SDavid van Moolenbroek case 'n':
113*00b67f09SDavid van Moolenbroek count = atoi(isc_commandline_argument);
114*00b67f09SDavid van Moolenbroek break;
115*00b67f09SDavid van Moolenbroek case ':':
116*00b67f09SDavid van Moolenbroek fprintf(stderr,
117*00b67f09SDavid van Moolenbroek "Option -%c requires an operand\n",
118*00b67f09SDavid van Moolenbroek isc_commandline_option);
119*00b67f09SDavid van Moolenbroek errflg++;
120*00b67f09SDavid van Moolenbroek break;
121*00b67f09SDavid van Moolenbroek case '?':
122*00b67f09SDavid van Moolenbroek default:
123*00b67f09SDavid van Moolenbroek fprintf(stderr, "Unrecognised option: -%c\n",
124*00b67f09SDavid van Moolenbroek isc_commandline_option);
125*00b67f09SDavid van Moolenbroek errflg++;
126*00b67f09SDavid van Moolenbroek }
127*00b67f09SDavid van Moolenbroek }
128*00b67f09SDavid van Moolenbroek
129*00b67f09SDavid van Moolenbroek if (errflg) {
130*00b67f09SDavid van Moolenbroek fprintf(stderr, "Usage:\n");
131*00b67f09SDavid van Moolenbroek fprintf(stderr,
132*00b67f09SDavid van Moolenbroek "\trandom [-m module] [-s slot] [-n count]\n");
133*00b67f09SDavid van Moolenbroek exit(1);
134*00b67f09SDavid van Moolenbroek }
135*00b67f09SDavid van Moolenbroek
136*00b67f09SDavid van Moolenbroek pk11_result_register();
137*00b67f09SDavid van Moolenbroek
138*00b67f09SDavid van Moolenbroek /* Initialize the CRYPTOKI library */
139*00b67f09SDavid van Moolenbroek if (lib_name != NULL)
140*00b67f09SDavid van Moolenbroek pk11_set_lib_name(lib_name);
141*00b67f09SDavid van Moolenbroek
142*00b67f09SDavid van Moolenbroek result = pk11_get_session(&pctx, op_type, ISC_FALSE, ISC_FALSE,
143*00b67f09SDavid van Moolenbroek ISC_FALSE, NULL, slot);
144*00b67f09SDavid van Moolenbroek if ((result != ISC_R_SUCCESS) &&
145*00b67f09SDavid van Moolenbroek (result != PK11_R_NODIGESTSERVICE) &&
146*00b67f09SDavid van Moolenbroek (result != PK11_R_NOAESSERVICE)) {
147*00b67f09SDavid van Moolenbroek fprintf(stderr, "Error initializing PKCS#11: %s\n",
148*00b67f09SDavid van Moolenbroek isc_result_totext(result));
149*00b67f09SDavid van Moolenbroek exit(1);
150*00b67f09SDavid van Moolenbroek }
151*00b67f09SDavid van Moolenbroek
152*00b67f09SDavid van Moolenbroek hSession = pctx.session;
153*00b67f09SDavid van Moolenbroek
154*00b67f09SDavid van Moolenbroek if (clock_gettime(CLOCK_REALTIME, &starttime) < 0) {
155*00b67f09SDavid van Moolenbroek perror("clock_gettime(start)");
156*00b67f09SDavid van Moolenbroek goto exit_session;
157*00b67f09SDavid van Moolenbroek }
158*00b67f09SDavid van Moolenbroek
159*00b67f09SDavid van Moolenbroek for (i = 0; i < count; i++) {
160*00b67f09SDavid van Moolenbroek /* Get random bytes */
161*00b67f09SDavid van Moolenbroek rv = pkcs_C_GenerateRandom(hSession, buf, len);
162*00b67f09SDavid van Moolenbroek if (rv != CKR_OK) {
163*00b67f09SDavid van Moolenbroek fprintf(stderr,
164*00b67f09SDavid van Moolenbroek "C_GenerateRandom[%u]: Error = 0x%.8lX\n",
165*00b67f09SDavid van Moolenbroek i, rv);
166*00b67f09SDavid van Moolenbroek error = 1;
167*00b67f09SDavid van Moolenbroek break;
168*00b67f09SDavid van Moolenbroek }
169*00b67f09SDavid van Moolenbroek }
170*00b67f09SDavid van Moolenbroek
171*00b67f09SDavid van Moolenbroek if (clock_gettime(CLOCK_REALTIME, &endtime) < 0) {
172*00b67f09SDavid van Moolenbroek perror("clock_gettime(end)");
173*00b67f09SDavid van Moolenbroek goto exit_session;
174*00b67f09SDavid van Moolenbroek }
175*00b67f09SDavid van Moolenbroek
176*00b67f09SDavid van Moolenbroek endtime.tv_sec -= starttime.tv_sec;
177*00b67f09SDavid van Moolenbroek endtime.tv_nsec -= starttime.tv_nsec;
178*00b67f09SDavid van Moolenbroek while (endtime.tv_nsec < 0) {
179*00b67f09SDavid van Moolenbroek endtime.tv_sec -= 1;
180*00b67f09SDavid van Moolenbroek endtime.tv_nsec += 1000000000;
181*00b67f09SDavid van Moolenbroek }
182*00b67f09SDavid van Moolenbroek printf("%uK random bytes in %ld.%09lds\n", i,
183*00b67f09SDavid van Moolenbroek endtime.tv_sec, endtime.tv_nsec);
184*00b67f09SDavid van Moolenbroek if (i > 0)
185*00b67f09SDavid van Moolenbroek printf("%g random bytes/s\n",
186*00b67f09SDavid van Moolenbroek 1024 * i / ((double) endtime.tv_sec +
187*00b67f09SDavid van Moolenbroek (double) endtime.tv_nsec / 1000000000.));
188*00b67f09SDavid van Moolenbroek
189*00b67f09SDavid van Moolenbroek exit_session:
190*00b67f09SDavid van Moolenbroek pk11_return_session(&pctx);
191*00b67f09SDavid van Moolenbroek (void) pk11_finalize();
192*00b67f09SDavid van Moolenbroek
193*00b67f09SDavid van Moolenbroek exit(error);
194*00b67f09SDavid van Moolenbroek }
195