xref: /freebsd/crypto/heimdal/lib/krb5/test_crypto.c (revision 271c3a9060f2ee55607ebe146523f888e1db2654)
1 /*
2  * Copyright (c) 2003-2005 Kungliga Tekniska H�gskolan
3  * (Royal Institute of Technology, Stockholm, Sweden).
4  * All rights reserved.
5  *
6  * Redistribution and use in source and binary forms, with or without
7  * modification, are permitted provided that the following conditions
8  * are met:
9  *
10  * 1. Redistributions of source code must retain the above copyright
11  *    notice, this list of conditions and the following disclaimer.
12  *
13  * 2. Redistributions in binary form must reproduce the above copyright
14  *    notice, this list of conditions and the following disclaimer in the
15  *    documentation and/or other materials provided with the distribution.
16  *
17  * 3. Neither the name of KTH nor the names of its contributors may be
18  *    used to endorse or promote products derived from this software without
19  *    specific prior written permission.
20  *
21  * THIS SOFTWARE IS PROVIDED BY KTH AND ITS CONTRIBUTORS ``AS IS'' AND ANY
22  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
24  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL KTH OR ITS CONTRIBUTORS BE
25  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
26  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
27  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR
28  * BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
29  * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
30  * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
31  * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. */
32 
33 #include "krb5_locl.h"
34 #include <err.h>
35 #include <getarg.h>
36 
37 RCSID("$Id: test_crypto.c 16290 2005-11-24 09:57:50Z lha $");
38 
39 static void
40 time_encryption(krb5_context context, size_t size,
41 		krb5_enctype etype, int iterations)
42 {
43     struct timeval tv1, tv2;
44     krb5_error_code ret;
45     krb5_keyblock key;
46     krb5_crypto crypto;
47     krb5_data data;
48     char *etype_name;
49     void *buf;
50     int i;
51 
52     ret = krb5_generate_random_keyblock(context, etype, &key);
53     if (ret)
54 	krb5_err(context, 1, ret, "krb5_generate_random_keyblock");
55 
56     ret = krb5_enctype_to_string(context, etype, &etype_name);
57     if (ret)
58 	krb5_err(context, 1, ret, "krb5_enctype_to_string");
59 
60     buf = malloc(size);
61     if (buf == NULL)
62 	krb5_errx(context, 1, "out of memory");
63     memset(buf, 0, size);
64 
65     ret = krb5_crypto_init(context, &key, 0, &crypto);
66     if (ret)
67 	krb5_err(context, 1, ret, "krb5_crypto_init");
68 
69     gettimeofday(&tv1, NULL);
70 
71     for (i = 0; i < iterations; i++) {
72 	ret = krb5_encrypt(context, crypto, 0, buf, size, &data);
73 	if (ret)
74 	    krb5_err(context, 1, ret, "encrypt: %d", i);
75 	krb5_data_free(&data);
76     }
77 
78     gettimeofday(&tv2, NULL);
79 
80     timevalsub(&tv2, &tv1);
81 
82     printf("%s size: %7lu iterations: %d time: %3ld.%06ld\n",
83 	   etype_name, (unsigned long)size, iterations,
84 	   (long)tv2.tv_sec, (long)tv2.tv_usec);
85 
86     free(buf);
87     free(etype_name);
88     krb5_crypto_destroy(context, crypto);
89     krb5_free_keyblock_contents(context, &key);
90 }
91 
92 static void
93 time_s2k(krb5_context context,
94 	 krb5_enctype etype,
95 	 const char *password,
96 	 krb5_salt salt,
97 	 int iterations)
98 {
99     struct timeval tv1, tv2;
100     krb5_error_code ret;
101     krb5_keyblock key;
102     krb5_data opaque;
103     char *etype_name;
104     int i;
105 
106     ret = krb5_enctype_to_string(context, etype, &etype_name);
107     if (ret)
108 	krb5_err(context, 1, ret, "krb5_enctype_to_string");
109 
110     opaque.data = NULL;
111     opaque.length = 0;
112 
113     gettimeofday(&tv1, NULL);
114 
115     for (i = 0; i < iterations; i++) {
116 	ret = krb5_string_to_key_salt_opaque(context, etype, password, salt,
117 					 opaque, &key);
118 	if (ret)
119 	    krb5_err(context, 1, ret, "krb5_string_to_key_data_salt_opaque");
120 	krb5_free_keyblock_contents(context, &key);
121     }
122 
123     gettimeofday(&tv2, NULL);
124 
125     timevalsub(&tv2, &tv1);
126 
127     printf("%s string2key %d iterations time: %3ld.%06ld\n",
128 	   etype_name, iterations, (long)tv2.tv_sec, (long)tv2.tv_usec);
129     free(etype_name);
130 
131 }
132 
133 static int version_flag = 0;
134 static int help_flag	= 0;
135 
136 static struct getargs args[] = {
137     {"version",	0,	arg_flag,	&version_flag,
138      "print version", NULL },
139     {"help",	0,	arg_flag,	&help_flag,
140      NULL, NULL }
141 };
142 
143 static void
144 usage (int ret)
145 {
146     arg_printusage (args,
147 		    sizeof(args)/sizeof(*args),
148 		    NULL,
149 		    "");
150     exit (ret);
151 }
152 
153 int
154 main(int argc, char **argv)
155 {
156     krb5_context context;
157     krb5_error_code ret;
158     int i, enciter, s2kiter;
159     int optidx = 0;
160     krb5_salt salt;
161 
162     krb5_enctype enctypes[] = {
163 	ETYPE_DES_CBC_CRC,
164 	ETYPE_DES3_CBC_SHA1,
165 	ETYPE_ARCFOUR_HMAC_MD5,
166 	ETYPE_AES128_CTS_HMAC_SHA1_96,
167 	ETYPE_AES256_CTS_HMAC_SHA1_96
168     };
169 
170     setprogname(argv[0]);
171 
172     if(getarg(args, sizeof(args) / sizeof(args[0]), argc, argv, &optidx))
173 	usage(1);
174 
175     if (help_flag)
176 	usage (0);
177 
178     if(version_flag){
179 	print_version(NULL);
180 	exit(0);
181     }
182 
183     argc -= optidx;
184     argv += optidx;
185 
186     salt.salttype = KRB5_PW_SALT;
187     salt.saltvalue.data = NULL;
188     salt.saltvalue.length = 0;
189 
190     ret = krb5_init_context(&context);
191     if (ret)
192 	errx (1, "krb5_init_context failed: %d", ret);
193 
194     enciter = 1000;
195     s2kiter = 100;
196 
197     for (i = 0; i < sizeof(enctypes)/sizeof(enctypes[0]); i++) {
198 
199 	time_encryption(context, 16, enctypes[i], enciter);
200 	time_encryption(context, 32, enctypes[i], enciter);
201 	time_encryption(context, 512, enctypes[i], enciter);
202 	time_encryption(context, 1024, enctypes[i], enciter);
203 	time_encryption(context, 2048, enctypes[i], enciter);
204 	time_encryption(context, 4096, enctypes[i], enciter);
205 	time_encryption(context, 8192, enctypes[i], enciter);
206 	time_encryption(context, 16384, enctypes[i], enciter);
207 	time_encryption(context, 32768, enctypes[i], enciter);
208 
209 	time_s2k(context, enctypes[i], "mYsecreitPassword", salt, s2kiter);
210     }
211 
212     krb5_free_context(context);
213 
214     return 0;
215 }
216