1/*
2 * WPA Supplicant / Crypto wrapper for LibTomCrypt (for internal TLSv1)
3 * Copyright (c) 2005-2006, Jouni Malinen <j@w1.fi>
4 *
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.
8 *
9 * Alternatively, this software may be distributed under the terms of BSD
10 * license.
11 *
12 * See README and COPYING for more details.
13 */
14
15#include "includes.h"
16#include <tomcrypt.h>
17
18#include "common.h"
19#include "rc4.h"
20#include "crypto.h"
21
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
30
31
32void md4_vector(size_t num_elem, const u8 *addr[], const size_t *len, u8 *mac)
33{
34	hash_state md;
35	size_t i;
36
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);
41}
42
43
44void des_encrypt(const u8 *clear, const u8 *key, u8 *cypher)
45{
46	u8 pkey[8], next, tmp;
47	int i;
48	symmetric_key skey;
49
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);
56	}
57	pkey[i] = next | 1;
58
59	des_setup(pkey, 8, 0, &skey);
60	des_ecb_encrypt(clear, cypher, &skey);
61	des_done(&skey);
62}
63
64
65#ifdef EAP_TLS_FUNCS
66void md5_vector(size_t num_elem, const u8 *addr[], const size_t *len, u8 *mac)
67{
68	hash_state md;
69	size_t i;
70
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);
75}
76
77
78void sha1_vector(size_t num_elem, const u8 *addr[], const size_t *len, u8 *mac)
79{
80	hash_state md;
81	size_t i;
82
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);
87}
88
89
90void * aes_encrypt_init(const u8 *key, size_t len)
91{
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;
99	}
100	return skey;
101}
102
103
104void aes_encrypt(void *ctx, const u8 *plain, u8 *crypt)
105{
106	symmetric_key *skey = ctx;
107	aes_ecb_encrypt(plain, crypt, skey);
108}
109
110
111void aes_encrypt_deinit(void *ctx)
112{
113	symmetric_key *skey = ctx;
114	aes_done(skey);
115	os_free(skey);
116}
117
118
119void * aes_decrypt_init(const u8 *key, size_t len)
120{
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;
128	}
129	return skey;
130}
131
132
133void aes_decrypt(void *ctx, const u8 *crypt, u8 *plain)
134{
135	symmetric_key *skey = ctx;
136	aes_ecb_encrypt(plain, (u8 *) crypt, skey);
137}
138
139
140void aes_decrypt_deinit(void *ctx)
141{
142	symmetric_key *skey = ctx;
143	aes_done(skey);
144	os_free(skey);
145}
146
147
148#ifdef CONFIG_TLS_INTERNAL
149
150struct crypto_hash {
151	enum crypto_hash_alg alg;
152	int error;
153	union {
154		hash_state md;
155		hmac_state hmac;
156	} u;
157};
158
159
160struct crypto_hash * crypto_hash_init(enum crypto_hash_alg alg, const u8 *key,
161				      size_t key_len)
162{
163	struct crypto_hash *ctx;
164
165	ctx = os_zalloc(sizeof(*ctx));
166	if (ctx == NULL)
167		return NULL;
168
169	ctx->alg = alg;
170
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;
192	}
193
194	return ctx;
195
196fail:
197	os_free(ctx);
198	return NULL;
199}
200
201void crypto_hash_update(struct crypto_hash *ctx, const u8 *data, size_t len)
202{
203	if (ctx == NULL || ctx->error)
204		return;
205
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;
217	}
218}
219
220
221int crypto_hash_finish(struct crypto_hash *ctx, u8 *mac, size_t *len)
222{
223	int ret = 0;
224	unsigned long clen;
225
226	if (ctx == NULL)
227		return -2;
228
229	if (mac == NULL || len == NULL) {
230		os_free(ctx);
231		return 0;
232	}
233
234	if (ctx->error) {
235		os_free(ctx);
236		return -2;
237	}
238
239	switch (ctx->alg) {
240	case CRYPTO_HASH_ALG_MD5:
241		if (*len < 16) {
242			*len = 16;
243			os_free(ctx);
244			return -1;
245		}
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;
255		}
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;
265		}
266		/* continue */
267	case CRYPTO_HASH_ALG_HMAC_MD5:
268		if (*len < 16) {
269			*len = 16;
270			os_free(ctx);
271			return -1;
272		}
273		clen = *len;
274		if (hmac_done(&ctx->u.hmac, mac, &clen) != CRYPT_OK) {
275			os_free(ctx);
276			return -1;
277		}
278		*len = clen;
279		break;
280	default:
281		ret = -2;
282		break;
283	}
284
285	os_free(ctx);
286
287	return ret;
288}
289
290
291struct 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;
301};
302
303
304struct crypto_cipher * crypto_cipher_init(enum crypto_cipher_alg alg,
305					  const u8 *iv, const u8 *key,
306					  size_t key_len)
307{
308	struct crypto_cipher *ctx;
309	int idx, res, rc4 = 0;
310
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;
330	}
331
332	ctx = os_zalloc(sizeof(*ctx));
333	if (ctx == NULL)
334		return NULL;
335
336	if (rc4) {
337		ctx->rc4 = 1;
338		if (key_len > sizeof(ctx->u.rc4.key)) {
339			os_free(ctx);
340			return NULL;
341		}
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;
351		}
352	}
353
354	return ctx;
355}
356
357int crypto_cipher_encrypt(struct crypto_cipher *ctx, const u8 *plain,
358			  u8 *crypt, size_t len)
359{
360	int res;
361
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;
369	}
370
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;
376	}
377	return 0;
378}
379
380
381int crypto_cipher_decrypt(struct crypto_cipher *ctx, const u8 *crypt,
382			  u8 *plain, size_t len)
383{
384	int res;
385
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;
393	}
394
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;
400	}
401
402	return 0;
403}
404
405
406void crypto_cipher_deinit(struct crypto_cipher *ctx)
407{
408	if (!ctx->rc4)
409		cbc_done(&ctx->u.cbc);
410	os_free(ctx);
411}
412
413
414struct crypto_public_key {
415	rsa_key rsa;
416};
417
418struct crypto_private_key {
419	rsa_key rsa;
420};
421
422
423struct crypto_public_key * crypto_public_key_import(const u8 *key, size_t len)
424{
425	int res;
426	struct crypto_public_key *pk;
427
428	pk = os_zalloc(sizeof(*pk));
429	if (pk == NULL)
430		return NULL;
431
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;
439	}
440
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;
447	}
448
449	return pk;
450}
451
452
453struct crypto_private_key * crypto_private_key_import(const u8 *key,
454						      size_t len)
455{
456	int res;
457	struct crypto_private_key *pk;
458
459	pk = os_zalloc(sizeof(*pk));
460	if (pk == NULL)
461		return NULL;
462
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;
470	}
471
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;
478	}
479
480	return pk;
481}
482
483
484struct crypto_public_key * crypto_public_key_from_cert(const u8 *buf,
485						       size_t len)
486{
487	/* No X.509 support in LibTomCrypt */
488	return NULL;
489}
490
491
492static int pkcs1_generate_encryption_block(u8 block_type, size_t modlen,
493					   const u8 *in, size_t inlen,
494					   u8 *out, size_t *outlen)
495{
496	size_t ps_len;
497	u8 *pos;
498
499	/*
500	 * PKCS #1 v1.5, 8.1:
501	 *
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)
507	 */
508
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;
516	}
517
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;
536		}
537		while (ps_len--) {
538			if (*pos == 0x00)
539				*pos = 0x01;
540			pos++;
541		}
542		break;
543	default:
544		wpa_printf(MSG_DEBUG, "PKCS #1: %s - Unsupported block type "
545			   "%d", __func__, block_type);
546		return -1;
547	}
548	*pos++ = 0x00;
549	os_memcpy(pos, in, inlen); /* D */
550
551	return 0;
552}
553
554
555static 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)
558{
559	unsigned long len, modlen;
560	int res;
561
562	modlen = mp_unsigned_bin_size(key->N);
563
564	if (pkcs1_generate_encryption_block(block_type, modlen, in, inlen,
565					    out, outlen) < 0)
566		return -1;
567
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;
574	}
575	*outlen = len;
576
577	return 0;
578}
579
580
581int crypto_public_key_encrypt_pkcs1_v15(struct crypto_public_key *key,
582					const u8 *in, size_t inlen,
583					u8 *out, size_t *outlen)
584{
585	return crypto_rsa_encrypt_pkcs1(2, &key->rsa, PK_PUBLIC, in, inlen,
586					out, outlen);
587}
588
589
590int crypto_private_key_sign_pkcs1(struct crypto_private_key *key,
591				  const u8 *in, size_t inlen,
592				  u8 *out, size_t *outlen)
593{
594	return crypto_rsa_encrypt_pkcs1(1, &key->rsa, PK_PRIVATE, in, inlen,
595					out, outlen);
596}
597
598
599void crypto_public_key_free(struct crypto_public_key *key)
600{
601	if (key) {
602		rsa_free(&key->rsa);
603		os_free(key);
604	}
605}
606
607
608void crypto_private_key_free(struct crypto_private_key *key)
609{
610	if (key) {
611		rsa_free(&key->rsa);
612		os_free(key);
613	}
614}
615
616
617int 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)
620{
621	int res;
622	unsigned long len;
623	u8 *pos;
624
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;
632	}
633
634	/*
635	 * PKCS #1 v1.5, 8.1:
636	 *
637	 * EB = 00 || BT || PS || 00 || D
638	 * BT = 01
639	 * PS = k-3-||D|| times FF
640	 * k = length of modulus in octets
641	 */
642
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;
648	}
649
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;
658	}
659
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;
664	}
665	pos++;
666	len -= pos - plain;
667
668	/* Strip PKCS #1 header */
669	os_memmove(plain, pos, len);
670	*plain_len = len;
671
672	return 0;
673}
674
675
676int crypto_global_init(void)
677{
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;
689	}
690
691	return 0;
692}
693
694
695void crypto_global_deinit(void)
696{
697}
698
699
700#ifdef EAP_FAST
701
702int 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)
706{
707	void *b, *p, *m, *r;
708
709	if (mp_init_multi(&b, &p, &m, &r, NULL) != CRYPT_OK)
710		return -1;
711
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;
716
717	if (mp_exptmod(b, p, m, r) != CRYPT_OK)
718		goto fail;
719
720	*result_len = mp_unsigned_bin_size(r);
721	if (mp_to_unsigned_bin(r, result) != CRYPT_OK)
722		goto fail;
723
724	mp_clear_multi(b, p, m, r, NULL);
725	return 0;
726
727fail:
728	mp_clear_multi(b, p, m, r, NULL);
729	return -1;
730}
731
732#endif /* EAP_FAST */
733
734#endif /* CONFIG_TLS_INTERNAL */
735
736#endif /* EAP_TLS_FUNCS */
737