Add a vclean_unlocked() call that allows HAMMER to try to get rid of a
[dragonfly.git] / contrib / wpa_supplicant-0.5.8 / crypto_libtomcrypt.c
blobe82097f10a00f1dd5702ee75ba97c4f6dc8d7697
1 /*
2 * WPA Supplicant / Crypto wrapper for LibTomCrypt (for internal TLSv1)
3 * Copyright (c) 2005-2006, Jouni Malinen <j@w1.fi>
5 * This program is free software; you can redistribute it and/or modify
6 * it under the terms of the GNU General Public License version 2 as
7 * published by the Free Software Foundation.
9 * Alternatively, this software may be distributed under the terms of BSD
10 * license.
12 * See README and COPYING for more details.
15 #include "includes.h"
16 #include <tomcrypt.h>
18 #include "common.h"
19 #include "rc4.h"
20 #include "crypto.h"
22 #ifndef mp_init_multi
23 #define mp_init_multi ltc_init_multi
24 #define mp_clear_multi ltc_deinit_multi
25 #define mp_unsigned_bin_size(a) ltc_mp.unsigned_size(a)
26 #define mp_to_unsigned_bin(a, b) ltc_mp.unsigned_write(a, b)
27 #define mp_read_unsigned_bin(a, b, c) ltc_mp.unsigned_read(a, b, c)
28 #define mp_exptmod(a,b,c,d) ltc_mp.exptmod(a,b,c,d)
29 #endif
32 void md4_vector(size_t num_elem, const u8 *addr[], const size_t *len, u8 *mac)
34 hash_state md;
35 size_t i;
37 md4_init(&md);
38 for (i = 0; i < num_elem; i++)
39 md4_process(&md, addr[i], len[i]);
40 md4_done(&md, mac);
44 void des_encrypt(const u8 *clear, const u8 *key, u8 *cypher)
46 u8 pkey[8], next, tmp;
47 int i;
48 symmetric_key skey;
50 /* Add parity bits to the key */
51 next = 0;
52 for (i = 0; i < 7; i++) {
53 tmp = key[i];
54 pkey[i] = (tmp >> i) | next | 1;
55 next = tmp << (7 - i);
57 pkey[i] = next | 1;
59 des_setup(pkey, 8, 0, &skey);
60 des_ecb_encrypt(clear, cypher, &skey);
61 des_done(&skey);
65 #ifdef EAP_TLS_FUNCS
66 void md5_vector(size_t num_elem, const u8 *addr[], const size_t *len, u8 *mac)
68 hash_state md;
69 size_t i;
71 md5_init(&md);
72 for (i = 0; i < num_elem; i++)
73 md5_process(&md, addr[i], len[i]);
74 md5_done(&md, mac);
78 void sha1_vector(size_t num_elem, const u8 *addr[], const size_t *len, u8 *mac)
80 hash_state md;
81 size_t i;
83 sha1_init(&md);
84 for (i = 0; i < num_elem; i++)
85 sha1_process(&md, addr[i], len[i]);
86 sha1_done(&md, mac);
90 void * aes_encrypt_init(const u8 *key, size_t len)
92 symmetric_key *skey;
93 skey = os_malloc(sizeof(*skey));
94 if (skey == NULL)
95 return NULL;
96 if (aes_setup(key, len, 0, skey) != CRYPT_OK) {
97 os_free(skey);
98 return NULL;
100 return skey;
104 void aes_encrypt(void *ctx, const u8 *plain, u8 *crypt)
106 symmetric_key *skey = ctx;
107 aes_ecb_encrypt(plain, crypt, skey);
111 void aes_encrypt_deinit(void *ctx)
113 symmetric_key *skey = ctx;
114 aes_done(skey);
115 os_free(skey);
119 void * aes_decrypt_init(const u8 *key, size_t len)
121 symmetric_key *skey;
122 skey = os_malloc(sizeof(*skey));
123 if (skey == NULL)
124 return NULL;
125 if (aes_setup(key, len, 0, skey) != CRYPT_OK) {
126 os_free(skey);
127 return NULL;
129 return skey;
133 void aes_decrypt(void *ctx, const u8 *crypt, u8 *plain)
135 symmetric_key *skey = ctx;
136 aes_ecb_encrypt(plain, (u8 *) crypt, skey);
140 void aes_decrypt_deinit(void *ctx)
142 symmetric_key *skey = ctx;
143 aes_done(skey);
144 os_free(skey);
148 #ifdef CONFIG_TLS_INTERNAL
150 struct crypto_hash {
151 enum crypto_hash_alg alg;
152 int error;
153 union {
154 hash_state md;
155 hmac_state hmac;
156 } u;
160 struct crypto_hash * crypto_hash_init(enum crypto_hash_alg alg, const u8 *key,
161 size_t key_len)
163 struct crypto_hash *ctx;
165 ctx = os_zalloc(sizeof(*ctx));
166 if (ctx == NULL)
167 return NULL;
169 ctx->alg = alg;
171 switch (alg) {
172 case CRYPTO_HASH_ALG_MD5:
173 if (md5_init(&ctx->u.md) != CRYPT_OK)
174 goto fail;
175 break;
176 case CRYPTO_HASH_ALG_SHA1:
177 if (sha1_init(&ctx->u.md) != CRYPT_OK)
178 goto fail;
179 break;
180 case CRYPTO_HASH_ALG_HMAC_MD5:
181 if (hmac_init(&ctx->u.hmac, find_hash("md5"), key, key_len) !=
182 CRYPT_OK)
183 goto fail;
184 break;
185 case CRYPTO_HASH_ALG_HMAC_SHA1:
186 if (hmac_init(&ctx->u.hmac, find_hash("sha1"), key, key_len) !=
187 CRYPT_OK)
188 goto fail;
189 break;
190 default:
191 goto fail;
194 return ctx;
196 fail:
197 os_free(ctx);
198 return NULL;
201 void crypto_hash_update(struct crypto_hash *ctx, const u8 *data, size_t len)
203 if (ctx == NULL || ctx->error)
204 return;
206 switch (ctx->alg) {
207 case CRYPTO_HASH_ALG_MD5:
208 ctx->error = md5_process(&ctx->u.md, data, len) != CRYPT_OK;
209 break;
210 case CRYPTO_HASH_ALG_SHA1:
211 ctx->error = sha1_process(&ctx->u.md, data, len) != CRYPT_OK;
212 break;
213 case CRYPTO_HASH_ALG_HMAC_MD5:
214 case CRYPTO_HASH_ALG_HMAC_SHA1:
215 ctx->error = hmac_process(&ctx->u.hmac, data, len) != CRYPT_OK;
216 break;
221 int crypto_hash_finish(struct crypto_hash *ctx, u8 *mac, size_t *len)
223 int ret = 0;
224 unsigned long clen;
226 if (ctx == NULL)
227 return -2;
229 if (mac == NULL || len == NULL) {
230 os_free(ctx);
231 return 0;
234 if (ctx->error) {
235 os_free(ctx);
236 return -2;
239 switch (ctx->alg) {
240 case CRYPTO_HASH_ALG_MD5:
241 if (*len < 16) {
242 *len = 16;
243 os_free(ctx);
244 return -1;
246 *len = 16;
247 if (md5_done(&ctx->u.md, mac) != CRYPT_OK)
248 ret = -2;
249 break;
250 case CRYPTO_HASH_ALG_SHA1:
251 if (*len < 20) {
252 *len = 20;
253 os_free(ctx);
254 return -1;
256 *len = 20;
257 if (sha1_done(&ctx->u.md, mac) != CRYPT_OK)
258 ret = -2;
259 break;
260 case CRYPTO_HASH_ALG_HMAC_SHA1:
261 if (*len < 20) {
262 *len = 20;
263 os_free(ctx);
264 return -1;
266 /* continue */
267 case CRYPTO_HASH_ALG_HMAC_MD5:
268 if (*len < 16) {
269 *len = 16;
270 os_free(ctx);
271 return -1;
273 clen = *len;
274 if (hmac_done(&ctx->u.hmac, mac, &clen) != CRYPT_OK) {
275 os_free(ctx);
276 return -1;
278 *len = clen;
279 break;
280 default:
281 ret = -2;
282 break;
285 os_free(ctx);
287 return ret;
291 struct crypto_cipher {
292 int rc4;
293 union {
294 symmetric_CBC cbc;
295 struct {
296 size_t used_bytes;
297 u8 key[16];
298 size_t keylen;
299 } rc4;
300 } u;
304 struct crypto_cipher * crypto_cipher_init(enum crypto_cipher_alg alg,
305 const u8 *iv, const u8 *key,
306 size_t key_len)
308 struct crypto_cipher *ctx;
309 int idx, res, rc4 = 0;
311 switch (alg) {
312 case CRYPTO_CIPHER_ALG_AES:
313 idx = find_cipher("aes");
314 break;
315 case CRYPTO_CIPHER_ALG_3DES:
316 idx = find_cipher("3des");
317 break;
318 case CRYPTO_CIPHER_ALG_DES:
319 idx = find_cipher("des");
320 break;
321 case CRYPTO_CIPHER_ALG_RC2:
322 idx = find_cipher("rc2");
323 break;
324 case CRYPTO_CIPHER_ALG_RC4:
325 idx = -1;
326 rc4 = 1;
327 break;
328 default:
329 return NULL;
332 ctx = os_zalloc(sizeof(*ctx));
333 if (ctx == NULL)
334 return NULL;
336 if (rc4) {
337 ctx->rc4 = 1;
338 if (key_len > sizeof(ctx->u.rc4.key)) {
339 os_free(ctx);
340 return NULL;
342 ctx->u.rc4.keylen = key_len;
343 os_memcpy(ctx->u.rc4.key, key, key_len);
344 } else {
345 res = cbc_start(idx, iv, key, key_len, 0, &ctx->u.cbc);
346 if (res != CRYPT_OK) {
347 wpa_printf(MSG_DEBUG, "LibTomCrypt: Cipher start "
348 "failed: %s", error_to_string(res));
349 os_free(ctx);
350 return NULL;
354 return ctx;
357 int crypto_cipher_encrypt(struct crypto_cipher *ctx, const u8 *plain,
358 u8 *crypt, size_t len)
360 int res;
362 if (ctx->rc4) {
363 if (plain != crypt)
364 os_memcpy(crypt, plain, len);
365 rc4_skip(ctx->u.rc4.key, ctx->u.rc4.keylen,
366 ctx->u.rc4.used_bytes, crypt, len);
367 ctx->u.rc4.used_bytes += len;
368 return 0;
371 res = cbc_encrypt(plain, crypt, len, &ctx->u.cbc);
372 if (res != CRYPT_OK) {
373 wpa_printf(MSG_DEBUG, "LibTomCrypt: CBC encryption "
374 "failed: %s", error_to_string(res));
375 return -1;
377 return 0;
381 int crypto_cipher_decrypt(struct crypto_cipher *ctx, const u8 *crypt,
382 u8 *plain, size_t len)
384 int res;
386 if (ctx->rc4) {
387 if (plain != crypt)
388 os_memcpy(plain, crypt, len);
389 rc4_skip(ctx->u.rc4.key, ctx->u.rc4.keylen,
390 ctx->u.rc4.used_bytes, plain, len);
391 ctx->u.rc4.used_bytes += len;
392 return 0;
395 res = cbc_decrypt(crypt, plain, len, &ctx->u.cbc);
396 if (res != CRYPT_OK) {
397 wpa_printf(MSG_DEBUG, "LibTomCrypt: CBC decryption "
398 "failed: %s", error_to_string(res));
399 return -1;
402 return 0;
406 void crypto_cipher_deinit(struct crypto_cipher *ctx)
408 if (!ctx->rc4)
409 cbc_done(&ctx->u.cbc);
410 os_free(ctx);
414 struct crypto_public_key {
415 rsa_key rsa;
418 struct crypto_private_key {
419 rsa_key rsa;
423 struct crypto_public_key * crypto_public_key_import(const u8 *key, size_t len)
425 int res;
426 struct crypto_public_key *pk;
428 pk = os_zalloc(sizeof(*pk));
429 if (pk == NULL)
430 return NULL;
432 res = rsa_import(key, len, &pk->rsa);
433 if (res != CRYPT_OK) {
434 wpa_printf(MSG_ERROR, "LibTomCrypt: Failed to import "
435 "public key (res=%d '%s')",
436 res, error_to_string(res));
437 os_free(pk);
438 return NULL;
441 if (pk->rsa.type != PK_PUBLIC) {
442 wpa_printf(MSG_ERROR, "LibTomCrypt: Public key was not of "
443 "correct type");
444 rsa_free(&pk->rsa);
445 os_free(pk);
446 return NULL;
449 return pk;
453 struct crypto_private_key * crypto_private_key_import(const u8 *key,
454 size_t len)
456 int res;
457 struct crypto_private_key *pk;
459 pk = os_zalloc(sizeof(*pk));
460 if (pk == NULL)
461 return NULL;
463 res = rsa_import(key, len, &pk->rsa);
464 if (res != CRYPT_OK) {
465 wpa_printf(MSG_ERROR, "LibTomCrypt: Failed to import "
466 "private key (res=%d '%s')",
467 res, error_to_string(res));
468 os_free(pk);
469 return NULL;
472 if (pk->rsa.type != PK_PRIVATE) {
473 wpa_printf(MSG_ERROR, "LibTomCrypt: Private key was not of "
474 "correct type");
475 rsa_free(&pk->rsa);
476 os_free(pk);
477 return NULL;
480 return pk;
484 struct crypto_public_key * crypto_public_key_from_cert(const u8 *buf,
485 size_t len)
487 /* No X.509 support in LibTomCrypt */
488 return NULL;
492 static int pkcs1_generate_encryption_block(u8 block_type, size_t modlen,
493 const u8 *in, size_t inlen,
494 u8 *out, size_t *outlen)
496 size_t ps_len;
497 u8 *pos;
500 * PKCS #1 v1.5, 8.1:
502 * EB = 00 || BT || PS || 00 || D
503 * BT = 00 or 01 for private-key operation; 02 for public-key operation
504 * PS = k-3-||D||; at least eight octets
505 * (BT=0: PS=0x00, BT=1: PS=0xff, BT=2: PS=pseudorandom non-zero)
506 * k = length of modulus in octets (modlen)
509 if (modlen < 12 || modlen > *outlen || inlen > modlen - 11) {
510 wpa_printf(MSG_DEBUG, "PKCS #1: %s - Invalid buffer "
511 "lengths (modlen=%lu outlen=%lu inlen=%lu)",
512 __func__, (unsigned long) modlen,
513 (unsigned long) *outlen,
514 (unsigned long) inlen);
515 return -1;
518 pos = out;
519 *pos++ = 0x00;
520 *pos++ = block_type; /* BT */
521 ps_len = modlen - inlen - 3;
522 switch (block_type) {
523 case 0:
524 os_memset(pos, 0x00, ps_len);
525 pos += ps_len;
526 break;
527 case 1:
528 os_memset(pos, 0xff, ps_len);
529 pos += ps_len;
530 break;
531 case 2:
532 if (os_get_random(pos, ps_len) < 0) {
533 wpa_printf(MSG_DEBUG, "PKCS #1: %s - Failed to get "
534 "random data for PS", __func__);
535 return -1;
537 while (ps_len--) {
538 if (*pos == 0x00)
539 *pos = 0x01;
540 pos++;
542 break;
543 default:
544 wpa_printf(MSG_DEBUG, "PKCS #1: %s - Unsupported block type "
545 "%d", __func__, block_type);
546 return -1;
548 *pos++ = 0x00;
549 os_memcpy(pos, in, inlen); /* D */
551 return 0;
555 static int crypto_rsa_encrypt_pkcs1(int block_type, rsa_key *key, int key_type,
556 const u8 *in, size_t inlen,
557 u8 *out, size_t *outlen)
559 unsigned long len, modlen;
560 int res;
562 modlen = mp_unsigned_bin_size(key->N);
564 if (pkcs1_generate_encryption_block(block_type, modlen, in, inlen,
565 out, outlen) < 0)
566 return -1;
568 len = *outlen;
569 res = rsa_exptmod(out, modlen, out, &len, key_type, key);
570 if (res != CRYPT_OK) {
571 wpa_printf(MSG_DEBUG, "LibTomCrypt: rsa_exptmod failed: %s",
572 error_to_string(res));
573 return -1;
575 *outlen = len;
577 return 0;
581 int crypto_public_key_encrypt_pkcs1_v15(struct crypto_public_key *key,
582 const u8 *in, size_t inlen,
583 u8 *out, size_t *outlen)
585 return crypto_rsa_encrypt_pkcs1(2, &key->rsa, PK_PUBLIC, in, inlen,
586 out, outlen);
590 int crypto_private_key_sign_pkcs1(struct crypto_private_key *key,
591 const u8 *in, size_t inlen,
592 u8 *out, size_t *outlen)
594 return crypto_rsa_encrypt_pkcs1(1, &key->rsa, PK_PRIVATE, in, inlen,
595 out, outlen);
599 void crypto_public_key_free(struct crypto_public_key *key)
601 if (key) {
602 rsa_free(&key->rsa);
603 os_free(key);
608 void crypto_private_key_free(struct crypto_private_key *key)
610 if (key) {
611 rsa_free(&key->rsa);
612 os_free(key);
617 int crypto_public_key_decrypt_pkcs1(struct crypto_public_key *key,
618 const u8 *crypt, size_t crypt_len,
619 u8 *plain, size_t *plain_len)
621 int res;
622 unsigned long len;
623 u8 *pos;
625 len = *plain_len;
626 res = rsa_exptmod(crypt, crypt_len, plain, &len, PK_PUBLIC,
627 &key->rsa);
628 if (res != CRYPT_OK) {
629 wpa_printf(MSG_DEBUG, "LibTomCrypt: rsa_exptmod failed: %s",
630 error_to_string(res));
631 return -1;
635 * PKCS #1 v1.5, 8.1:
637 * EB = 00 || BT || PS || 00 || D
638 * BT = 01
639 * PS = k-3-||D|| times FF
640 * k = length of modulus in octets
643 if (len < 3 + 8 + 16 /* min hash len */ ||
644 plain[0] != 0x00 || plain[1] != 0x01 || plain[2] != 0xff) {
645 wpa_printf(MSG_INFO, "LibTomCrypt: Invalid signature EB "
646 "structure");
647 return -1;
650 pos = plain + 3;
651 while (pos < plain + len && *pos == 0xff)
652 pos++;
653 if (pos - plain - 2 < 8) {
654 /* PKCS #1 v1.5, 8.1: At least eight octets long PS */
655 wpa_printf(MSG_INFO, "LibTomCrypt: Too short signature "
656 "padding");
657 return -1;
660 if (pos + 16 /* min hash len */ >= plain + len || *pos != 0x00) {
661 wpa_printf(MSG_INFO, "LibTomCrypt: Invalid signature EB "
662 "structure (2)");
663 return -1;
665 pos++;
666 len -= pos - plain;
668 /* Strip PKCS #1 header */
669 os_memmove(plain, pos, len);
670 *plain_len = len;
672 return 0;
676 int crypto_global_init(void)
678 ltc_mp = tfm_desc;
679 /* TODO: only register algorithms that are really needed */
680 if (register_hash(&md4_desc) < 0 ||
681 register_hash(&md5_desc) < 0 ||
682 register_hash(&sha1_desc) < 0 ||
683 register_cipher(&aes_desc) < 0 ||
684 register_cipher(&des_desc) < 0 ||
685 register_cipher(&des3_desc) < 0) {
686 wpa_printf(MSG_ERROR, "TLSv1: Failed to register "
687 "hash/cipher functions");
688 return -1;
691 return 0;
695 void crypto_global_deinit(void)
700 #ifdef EAP_FAST
702 int crypto_mod_exp(const u8 *base, size_t base_len,
703 const u8 *power, size_t power_len,
704 const u8 *modulus, size_t modulus_len,
705 u8 *result, size_t *result_len)
707 void *b, *p, *m, *r;
709 if (mp_init_multi(&b, &p, &m, &r, NULL) != CRYPT_OK)
710 return -1;
712 if (mp_read_unsigned_bin(b, (u8 *) base, base_len) != CRYPT_OK ||
713 mp_read_unsigned_bin(p, (u8 *) power, power_len) != CRYPT_OK ||
714 mp_read_unsigned_bin(m, (u8 *) modulus, modulus_len) != CRYPT_OK)
715 goto fail;
717 if (mp_exptmod(b, p, m, r) != CRYPT_OK)
718 goto fail;
720 *result_len = mp_unsigned_bin_size(r);
721 if (mp_to_unsigned_bin(r, result) != CRYPT_OK)
722 goto fail;
724 mp_clear_multi(b, p, m, r, NULL);
725 return 0;
727 fail:
728 mp_clear_multi(b, p, m, r, NULL);
729 return -1;
732 #endif /* EAP_FAST */
734 #endif /* CONFIG_TLS_INTERNAL */
736 #endif /* EAP_TLS_FUNCS */