1/*
2 * AES-128-CMAC with TLen 16 for IEEE 802.11w BIP
3 * Copyright 2008, 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
10#include <linux/kernel.h>
11#include <linux/types.h>
12#include <linux/crypto.h>
13#include <linux/export.h>
14#include <linux/err.h>
15#include <crypto/aes.h>
16
17#include <net/mac80211.h>
18#include "key.h"
19#include "aes_cmac.h"
20
21#define CMAC_TLEN 8 /* CMAC TLen = 64 bits (8 octets) */
22#define CMAC_TLEN_256 16 /* CMAC TLen = 128 bits (16 octets) */
23#define AAD_LEN 20
24
25
26static void gf_mulx(u8 *pad)
27{
28	int i, carry;
29
30	carry = pad[0] & 0x80;
31	for (i = 0; i < AES_BLOCK_SIZE - 1; i++)
32		pad[i] = (pad[i] << 1) | (pad[i + 1] >> 7);
33	pad[AES_BLOCK_SIZE - 1] <<= 1;
34	if (carry)
35		pad[AES_BLOCK_SIZE - 1] ^= 0x87;
36}
37
38static void aes_cmac_vector(struct crypto_cipher *tfm, size_t num_elem,
39			    const u8 *addr[], const size_t *len, u8 *mac,
40			    size_t mac_len)
41{
42	u8 cbc[AES_BLOCK_SIZE], pad[AES_BLOCK_SIZE];
43	const u8 *pos, *end;
44	size_t i, e, left, total_len;
45
46	memset(cbc, 0, AES_BLOCK_SIZE);
47
48	total_len = 0;
49	for (e = 0; e < num_elem; e++)
50		total_len += len[e];
51	left = total_len;
52
53	e = 0;
54	pos = addr[0];
55	end = pos + len[0];
56
57	while (left >= AES_BLOCK_SIZE) {
58		for (i = 0; i < AES_BLOCK_SIZE; i++) {
59			cbc[i] ^= *pos++;
60			if (pos >= end) {
61				e++;
62				pos = addr[e];
63				end = pos + len[e];
64			}
65		}
66		if (left > AES_BLOCK_SIZE)
67			crypto_cipher_encrypt_one(tfm, cbc, cbc);
68		left -= AES_BLOCK_SIZE;
69	}
70
71	memset(pad, 0, AES_BLOCK_SIZE);
72	crypto_cipher_encrypt_one(tfm, pad, pad);
73	gf_mulx(pad);
74
75	if (left || total_len == 0) {
76		for (i = 0; i < left; i++) {
77			cbc[i] ^= *pos++;
78			if (pos >= end) {
79				e++;
80				pos = addr[e];
81				end = pos + len[e];
82			}
83		}
84		cbc[left] ^= 0x80;
85		gf_mulx(pad);
86	}
87
88	for (i = 0; i < AES_BLOCK_SIZE; i++)
89		pad[i] ^= cbc[i];
90	crypto_cipher_encrypt_one(tfm, pad, pad);
91	memcpy(mac, pad, mac_len);
92}
93
94
95void ieee80211_aes_cmac(struct crypto_cipher *tfm, const u8 *aad,
96			const u8 *data, size_t data_len, u8 *mic)
97{
98	const u8 *addr[3];
99	size_t len[3];
100	u8 zero[CMAC_TLEN];
101
102	memset(zero, 0, CMAC_TLEN);
103	addr[0] = aad;
104	len[0] = AAD_LEN;
105	addr[1] = data;
106	len[1] = data_len - CMAC_TLEN;
107	addr[2] = zero;
108	len[2] = CMAC_TLEN;
109
110	aes_cmac_vector(tfm, 3, addr, len, mic, CMAC_TLEN);
111}
112
113void ieee80211_aes_cmac_256(struct crypto_cipher *tfm, const u8 *aad,
114			    const u8 *data, size_t data_len, u8 *mic)
115{
116	const u8 *addr[3];
117	size_t len[3];
118	u8 zero[CMAC_TLEN_256];
119
120	memset(zero, 0, CMAC_TLEN_256);
121	addr[0] = aad;
122	len[0] = AAD_LEN;
123	addr[1] = data;
124	len[1] = data_len - CMAC_TLEN_256;
125	addr[2] = zero;
126	len[2] = CMAC_TLEN_256;
127
128	aes_cmac_vector(tfm, 3, addr, len, mic, CMAC_TLEN_256);
129}
130
131struct crypto_cipher *ieee80211_aes_cmac_key_setup(const u8 key[],
132						   size_t key_len)
133{
134	struct crypto_cipher *tfm;
135
136	tfm = crypto_alloc_cipher("aes", 0, CRYPTO_ALG_ASYNC);
137	if (!IS_ERR(tfm))
138		crypto_cipher_setkey(tfm, key, key_len);
139
140	return tfm;
141}
142
143
144void ieee80211_aes_cmac_key_free(struct crypto_cipher *tfm)
145{
146	crypto_free_cipher(tfm);
147}
148
149void ieee80211_aes_cmac_calculate_k1_k2(struct ieee80211_key_conf *keyconf,
150					u8 *k1, u8 *k2)
151{
152	u8 l[AES_BLOCK_SIZE] = {};
153	struct ieee80211_key *key =
154		container_of(keyconf, struct ieee80211_key, conf);
155
156	crypto_cipher_encrypt_one(key->u.aes_cmac.tfm, l, l);
157
158	memcpy(k1, l, AES_BLOCK_SIZE);
159	gf_mulx(k1);
160
161	memcpy(k2, k1, AES_BLOCK_SIZE);
162	gf_mulx(k2);
163}
164EXPORT_SYMBOL(ieee80211_aes_cmac_calculate_k1_k2);
165