1/*-
2 * Copyright (c) 2002-2008 Sam Leffler, Errno Consulting
3 * All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 * 1. Redistributions of source code must retain the above copyright
9 *    notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 *    notice, this list of conditions and the following disclaimer in the
12 *    documentation and/or other materials provided with the distribution.
13 *
14 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
15 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
16 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
17 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
18 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
19 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
20 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
21 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
22 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
23 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
24 */
25
26#include <sys/cdefs.h>
27__FBSDID("$FreeBSD$");
28
29/*
30 * IEEE 802.11i TKIP crypto support.
31 *
32 * Part of this module is derived from similar code in the Host
33 * AP driver. The code is used with the consent of the author and
34 * it's license is included below.
35 */
36#include "opt_wlan.h"
37
38#include <sys/param.h>
39#include <sys/systm.h>
40#include <sys/mbuf.h>
41#include <sys/malloc.h>
42#include <sys/kernel.h>
43#include <sys/module.h>
44#include <sys/endian.h>
45
46#include <sys/socket.h>
47
48#include <net/if.h>
49#include <net/if_media.h>
50#include <net/ethernet.h>
51
52#include <net80211/ieee80211_var.h>
53
54static	void *tkip_attach(struct ieee80211vap *, struct ieee80211_key *);
55static	void tkip_detach(struct ieee80211_key *);
56static	int tkip_setkey(struct ieee80211_key *);
57static	int tkip_encap(struct ieee80211_key *, struct mbuf *m, uint8_t keyid);
58static	int tkip_enmic(struct ieee80211_key *, struct mbuf *, int);
59static	int tkip_decap(struct ieee80211_key *, struct mbuf *, int);
60static	int tkip_demic(struct ieee80211_key *, struct mbuf *, int);
61
62static const struct ieee80211_cipher tkip  = {
63	.ic_name	= "TKIP",
64	.ic_cipher	= IEEE80211_CIPHER_TKIP,
65	.ic_header	= IEEE80211_WEP_IVLEN + IEEE80211_WEP_KIDLEN +
66			  IEEE80211_WEP_EXTIVLEN,
67	.ic_trailer	= IEEE80211_WEP_CRCLEN,
68	.ic_miclen	= IEEE80211_WEP_MICLEN,
69	.ic_attach	= tkip_attach,
70	.ic_detach	= tkip_detach,
71	.ic_setkey	= tkip_setkey,
72	.ic_encap	= tkip_encap,
73	.ic_decap	= tkip_decap,
74	.ic_enmic	= tkip_enmic,
75	.ic_demic	= tkip_demic,
76};
77
78typedef	uint8_t u8;
79typedef	uint16_t u16;
80typedef	uint32_t __u32;
81typedef	uint32_t u32;
82
83struct tkip_ctx {
84	struct ieee80211vap *tc_vap;	/* for diagnostics+statistics */
85
86	u16	tx_ttak[5];
87	int	tx_phase1_done;
88	u8	tx_rc4key[16];		/* XXX for test module; make locals? */
89
90	u16	rx_ttak[5];
91	int	rx_phase1_done;
92	u8	rx_rc4key[16];		/* XXX for test module; make locals? */
93	uint64_t rx_rsc;		/* held until MIC verified */
94};
95
96static	void michael_mic(struct tkip_ctx *, const u8 *key,
97		struct mbuf *m, u_int off, size_t data_len,
98		u8 mic[IEEE80211_WEP_MICLEN]);
99static	int tkip_encrypt(struct tkip_ctx *, struct ieee80211_key *,
100		struct mbuf *, int hdr_len);
101static	int tkip_decrypt(struct tkip_ctx *, struct ieee80211_key *,
102		struct mbuf *, int hdr_len);
103
104/* number of references from net80211 layer */
105static	int nrefs = 0;
106
107static void *
108tkip_attach(struct ieee80211vap *vap, struct ieee80211_key *k)
109{
110	struct tkip_ctx *ctx;
111
112	ctx = (struct tkip_ctx *) malloc(sizeof(struct tkip_ctx),
113		M_80211_CRYPTO, M_NOWAIT | M_ZERO);
114	if (ctx == NULL) {
115		vap->iv_stats.is_crypto_nomem++;
116		return NULL;
117	}
118
119	ctx->tc_vap = vap;
120	nrefs++;			/* NB: we assume caller locking */
121	return ctx;
122}
123
124static void
125tkip_detach(struct ieee80211_key *k)
126{
127	struct tkip_ctx *ctx = k->wk_private;
128
129	free(ctx, M_80211_CRYPTO);
130	KASSERT(nrefs > 0, ("imbalanced attach/detach"));
131	nrefs--;			/* NB: we assume caller locking */
132}
133
134static int
135tkip_setkey(struct ieee80211_key *k)
136{
137	struct tkip_ctx *ctx = k->wk_private;
138
139	if (k->wk_keylen != (128/NBBY)) {
140		(void) ctx;		/* XXX */
141		IEEE80211_DPRINTF(ctx->tc_vap, IEEE80211_MSG_CRYPTO,
142			"%s: Invalid key length %u, expecting %u\n",
143			__func__, k->wk_keylen, 128/NBBY);
144		return 0;
145	}
146	k->wk_keytsc = 1;		/* TSC starts at 1 */
147	ctx->rx_phase1_done = 0;
148	return 1;
149}
150
151/*
152 * Add privacy headers and do any s/w encryption required.
153 */
154static int
155tkip_encap(struct ieee80211_key *k, struct mbuf *m, uint8_t keyid)
156{
157	struct tkip_ctx *ctx = k->wk_private;
158	struct ieee80211vap *vap = ctx->tc_vap;
159	struct ieee80211com *ic = vap->iv_ic;
160	uint8_t *ivp;
161	int hdrlen;
162
163	/*
164	 * Handle TKIP counter measures requirement.
165	 */
166	if (vap->iv_flags & IEEE80211_F_COUNTERM) {
167#ifdef IEEE80211_DEBUG
168		struct ieee80211_frame *wh = mtod(m, struct ieee80211_frame *);
169#endif
170
171		IEEE80211_NOTE_MAC(vap, IEEE80211_MSG_CRYPTO, wh->i_addr2,
172		    "discard frame due to countermeasures (%s)", __func__);
173		vap->iv_stats.is_crypto_tkipcm++;
174		return 0;
175	}
176	hdrlen = ieee80211_hdrspace(ic, mtod(m, void *));
177
178	/*
179	 * Copy down 802.11 header and add the IV, KeyID, and ExtIV.
180	 */
181	M_PREPEND(m, tkip.ic_header, M_NOWAIT);
182	if (m == NULL)
183		return 0;
184	ivp = mtod(m, uint8_t *);
185	memmove(ivp, ivp + tkip.ic_header, hdrlen);
186	ivp += hdrlen;
187
188	ivp[0] = k->wk_keytsc >> 8;		/* TSC1 */
189	ivp[1] = (ivp[0] | 0x20) & 0x7f;	/* WEP seed */
190	ivp[2] = k->wk_keytsc >> 0;		/* TSC0 */
191	ivp[3] = keyid | IEEE80211_WEP_EXTIV;	/* KeyID | ExtID */
192	ivp[4] = k->wk_keytsc >> 16;		/* TSC2 */
193	ivp[5] = k->wk_keytsc >> 24;		/* TSC3 */
194	ivp[6] = k->wk_keytsc >> 32;		/* TSC4 */
195	ivp[7] = k->wk_keytsc >> 40;		/* TSC5 */
196
197	/*
198	 * Finally, do software encrypt if neeed.
199	 */
200	if (k->wk_flags & IEEE80211_KEY_SWENCRYPT) {
201		if (!tkip_encrypt(ctx, k, m, hdrlen))
202			return 0;
203		/* NB: tkip_encrypt handles wk_keytsc */
204	} else
205		k->wk_keytsc++;
206
207	return 1;
208}
209
210/*
211 * Add MIC to the frame as needed.
212 */
213static int
214tkip_enmic(struct ieee80211_key *k, struct mbuf *m, int force)
215{
216	struct tkip_ctx *ctx = k->wk_private;
217
218	if (force || (k->wk_flags & IEEE80211_KEY_SWENMIC)) {
219		struct ieee80211_frame *wh = mtod(m, struct ieee80211_frame *);
220		struct ieee80211vap *vap = ctx->tc_vap;
221		struct ieee80211com *ic = vap->iv_ic;
222		int hdrlen;
223		uint8_t mic[IEEE80211_WEP_MICLEN];
224
225		vap->iv_stats.is_crypto_tkipenmic++;
226
227		hdrlen = ieee80211_hdrspace(ic, wh);
228
229		michael_mic(ctx, k->wk_txmic,
230			m, hdrlen, m->m_pkthdr.len - hdrlen, mic);
231		return m_append(m, tkip.ic_miclen, mic);
232	}
233	return 1;
234}
235
236static __inline uint64_t
237READ_6(uint8_t b0, uint8_t b1, uint8_t b2, uint8_t b3, uint8_t b4, uint8_t b5)
238{
239	uint32_t iv32 = (b0 << 0) | (b1 << 8) | (b2 << 16) | (b3 << 24);
240	uint16_t iv16 = (b4 << 0) | (b5 << 8);
241	return (((uint64_t)iv16) << 32) | iv32;
242}
243
244/*
245 * Validate and strip privacy headers (and trailer) for a
246 * received frame.  If necessary, decrypt the frame using
247 * the specified key.
248 */
249static int
250tkip_decap(struct ieee80211_key *k, struct mbuf *m, int hdrlen)
251{
252	struct tkip_ctx *ctx = k->wk_private;
253	struct ieee80211vap *vap = ctx->tc_vap;
254	struct ieee80211_frame *wh;
255	uint8_t *ivp, tid;
256
257	/*
258	 * Header should have extended IV and sequence number;
259	 * verify the former and validate the latter.
260	 */
261	wh = mtod(m, struct ieee80211_frame *);
262	ivp = mtod(m, uint8_t *) + hdrlen;
263	if ((ivp[IEEE80211_WEP_IVLEN] & IEEE80211_WEP_EXTIV) == 0) {
264		/*
265		 * No extended IV; discard frame.
266		 */
267		IEEE80211_NOTE_MAC(vap, IEEE80211_MSG_CRYPTO, wh->i_addr2,
268		    "%s", "missing ExtIV for TKIP cipher");
269		vap->iv_stats.is_rx_tkipformat++;
270		return 0;
271	}
272	/*
273	 * Handle TKIP counter measures requirement.
274	 */
275	if (vap->iv_flags & IEEE80211_F_COUNTERM) {
276		IEEE80211_NOTE_MAC(vap, IEEE80211_MSG_CRYPTO, wh->i_addr2,
277		    "discard frame due to countermeasures (%s)", __func__);
278		vap->iv_stats.is_crypto_tkipcm++;
279		return 0;
280	}
281
282	tid = ieee80211_gettid(wh);
283	ctx->rx_rsc = READ_6(ivp[2], ivp[0], ivp[4], ivp[5], ivp[6], ivp[7]);
284	if (ctx->rx_rsc <= k->wk_keyrsc[tid] &&
285	    (k->wk_flags & IEEE80211_KEY_NOREPLAY) == 0) {
286		/*
287		 * Replay violation; notify upper layer.
288		 */
289		ieee80211_notify_replay_failure(vap, wh, k, ctx->rx_rsc, tid);
290		vap->iv_stats.is_rx_tkipreplay++;
291		return 0;
292	}
293	/*
294	 * NB: We can't update the rsc in the key until MIC is verified.
295	 *
296	 * We assume we are not preempted between doing the check above
297	 * and updating wk_keyrsc when stripping the MIC in tkip_demic.
298	 * Otherwise we might process another packet and discard it as
299	 * a replay.
300	 */
301
302	/*
303	 * Check if the device handled the decrypt in hardware.
304	 * If so we just strip the header; otherwise we need to
305	 * handle the decrypt in software.
306	 */
307	if ((k->wk_flags & IEEE80211_KEY_SWDECRYPT) &&
308	    !tkip_decrypt(ctx, k, m, hdrlen))
309		return 0;
310
311	/*
312	 * Copy up 802.11 header and strip crypto bits.
313	 */
314	memmove(mtod(m, uint8_t *) + tkip.ic_header, mtod(m, void *), hdrlen);
315	m_adj(m, tkip.ic_header);
316	m_adj(m, -tkip.ic_trailer);
317
318	return 1;
319}
320
321/*
322 * Verify and strip MIC from the frame.
323 */
324static int
325tkip_demic(struct ieee80211_key *k, struct mbuf *m, int force)
326{
327	struct tkip_ctx *ctx = k->wk_private;
328	struct ieee80211_frame *wh;
329	uint8_t tid;
330
331	wh = mtod(m, struct ieee80211_frame *);
332	if ((k->wk_flags & IEEE80211_KEY_SWDEMIC) || force) {
333		struct ieee80211vap *vap = ctx->tc_vap;
334		int hdrlen = ieee80211_hdrspace(vap->iv_ic, wh);
335		u8 mic[IEEE80211_WEP_MICLEN];
336		u8 mic0[IEEE80211_WEP_MICLEN];
337
338		vap->iv_stats.is_crypto_tkipdemic++;
339
340		michael_mic(ctx, k->wk_rxmic,
341			m, hdrlen, m->m_pkthdr.len - (hdrlen + tkip.ic_miclen),
342			mic);
343		m_copydata(m, m->m_pkthdr.len - tkip.ic_miclen,
344			tkip.ic_miclen, mic0);
345		if (memcmp(mic, mic0, tkip.ic_miclen)) {
346			/* NB: 802.11 layer handles statistic and debug msg */
347			ieee80211_notify_michael_failure(vap, wh,
348				k->wk_rxkeyix != IEEE80211_KEYIX_NONE ?
349					k->wk_rxkeyix : k->wk_keyix);
350			return 0;
351		}
352	}
353	/*
354	 * Strip MIC from the tail.
355	 */
356	m_adj(m, -tkip.ic_miclen);
357
358	/*
359	 * Ok to update rsc now that MIC has been verified.
360	 */
361	tid = ieee80211_gettid(wh);
362	k->wk_keyrsc[tid] = ctx->rx_rsc;
363
364	return 1;
365}
366
367/*
368 * Host AP crypt: host-based TKIP encryption implementation for Host AP driver
369 *
370 * Copyright (c) 2003-2004, Jouni Malinen <jkmaline@cc.hut.fi>
371 *
372 * This program is free software; you can redistribute it and/or modify
373 * it under the terms of the GNU General Public License version 2 as
374 * published by the Free Software Foundation. See README and COPYING for
375 * more details.
376 *
377 * Alternatively, this software may be distributed under the terms of BSD
378 * license.
379 */
380
381static const __u32 crc32_table[256] = {
382	0x00000000L, 0x77073096L, 0xee0e612cL, 0x990951baL, 0x076dc419L,
383	0x706af48fL, 0xe963a535L, 0x9e6495a3L, 0x0edb8832L, 0x79dcb8a4L,
384	0xe0d5e91eL, 0x97d2d988L, 0x09b64c2bL, 0x7eb17cbdL, 0xe7b82d07L,
385	0x90bf1d91L, 0x1db71064L, 0x6ab020f2L, 0xf3b97148L, 0x84be41deL,
386	0x1adad47dL, 0x6ddde4ebL, 0xf4d4b551L, 0x83d385c7L, 0x136c9856L,
387	0x646ba8c0L, 0xfd62f97aL, 0x8a65c9ecL, 0x14015c4fL, 0x63066cd9L,
388	0xfa0f3d63L, 0x8d080df5L, 0x3b6e20c8L, 0x4c69105eL, 0xd56041e4L,
389	0xa2677172L, 0x3c03e4d1L, 0x4b04d447L, 0xd20d85fdL, 0xa50ab56bL,
390	0x35b5a8faL, 0x42b2986cL, 0xdbbbc9d6L, 0xacbcf940L, 0x32d86ce3L,
391	0x45df5c75L, 0xdcd60dcfL, 0xabd13d59L, 0x26d930acL, 0x51de003aL,
392	0xc8d75180L, 0xbfd06116L, 0x21b4f4b5L, 0x56b3c423L, 0xcfba9599L,
393	0xb8bda50fL, 0x2802b89eL, 0x5f058808L, 0xc60cd9b2L, 0xb10be924L,
394	0x2f6f7c87L, 0x58684c11L, 0xc1611dabL, 0xb6662d3dL, 0x76dc4190L,
395	0x01db7106L, 0x98d220bcL, 0xefd5102aL, 0x71b18589L, 0x06b6b51fL,
396	0x9fbfe4a5L, 0xe8b8d433L, 0x7807c9a2L, 0x0f00f934L, 0x9609a88eL,
397	0xe10e9818L, 0x7f6a0dbbL, 0x086d3d2dL, 0x91646c97L, 0xe6635c01L,
398	0x6b6b51f4L, 0x1c6c6162L, 0x856530d8L, 0xf262004eL, 0x6c0695edL,
399	0x1b01a57bL, 0x8208f4c1L, 0xf50fc457L, 0x65b0d9c6L, 0x12b7e950L,
400	0x8bbeb8eaL, 0xfcb9887cL, 0x62dd1ddfL, 0x15da2d49L, 0x8cd37cf3L,
401	0xfbd44c65L, 0x4db26158L, 0x3ab551ceL, 0xa3bc0074L, 0xd4bb30e2L,
402	0x4adfa541L, 0x3dd895d7L, 0xa4d1c46dL, 0xd3d6f4fbL, 0x4369e96aL,
403	0x346ed9fcL, 0xad678846L, 0xda60b8d0L, 0x44042d73L, 0x33031de5L,
404	0xaa0a4c5fL, 0xdd0d7cc9L, 0x5005713cL, 0x270241aaL, 0xbe0b1010L,
405	0xc90c2086L, 0x5768b525L, 0x206f85b3L, 0xb966d409L, 0xce61e49fL,
406	0x5edef90eL, 0x29d9c998L, 0xb0d09822L, 0xc7d7a8b4L, 0x59b33d17L,
407	0x2eb40d81L, 0xb7bd5c3bL, 0xc0ba6cadL, 0xedb88320L, 0x9abfb3b6L,
408	0x03b6e20cL, 0x74b1d29aL, 0xead54739L, 0x9dd277afL, 0x04db2615L,
409	0x73dc1683L, 0xe3630b12L, 0x94643b84L, 0x0d6d6a3eL, 0x7a6a5aa8L,
410	0xe40ecf0bL, 0x9309ff9dL, 0x0a00ae27L, 0x7d079eb1L, 0xf00f9344L,
411	0x8708a3d2L, 0x1e01f268L, 0x6906c2feL, 0xf762575dL, 0x806567cbL,
412	0x196c3671L, 0x6e6b06e7L, 0xfed41b76L, 0x89d32be0L, 0x10da7a5aL,
413	0x67dd4accL, 0xf9b9df6fL, 0x8ebeeff9L, 0x17b7be43L, 0x60b08ed5L,
414	0xd6d6a3e8L, 0xa1d1937eL, 0x38d8c2c4L, 0x4fdff252L, 0xd1bb67f1L,
415	0xa6bc5767L, 0x3fb506ddL, 0x48b2364bL, 0xd80d2bdaL, 0xaf0a1b4cL,
416	0x36034af6L, 0x41047a60L, 0xdf60efc3L, 0xa867df55L, 0x316e8eefL,
417	0x4669be79L, 0xcb61b38cL, 0xbc66831aL, 0x256fd2a0L, 0x5268e236L,
418	0xcc0c7795L, 0xbb0b4703L, 0x220216b9L, 0x5505262fL, 0xc5ba3bbeL,
419	0xb2bd0b28L, 0x2bb45a92L, 0x5cb36a04L, 0xc2d7ffa7L, 0xb5d0cf31L,
420	0x2cd99e8bL, 0x5bdeae1dL, 0x9b64c2b0L, 0xec63f226L, 0x756aa39cL,
421	0x026d930aL, 0x9c0906a9L, 0xeb0e363fL, 0x72076785L, 0x05005713L,
422	0x95bf4a82L, 0xe2b87a14L, 0x7bb12baeL, 0x0cb61b38L, 0x92d28e9bL,
423	0xe5d5be0dL, 0x7cdcefb7L, 0x0bdbdf21L, 0x86d3d2d4L, 0xf1d4e242L,
424	0x68ddb3f8L, 0x1fda836eL, 0x81be16cdL, 0xf6b9265bL, 0x6fb077e1L,
425	0x18b74777L, 0x88085ae6L, 0xff0f6a70L, 0x66063bcaL, 0x11010b5cL,
426	0x8f659effL, 0xf862ae69L, 0x616bffd3L, 0x166ccf45L, 0xa00ae278L,
427	0xd70dd2eeL, 0x4e048354L, 0x3903b3c2L, 0xa7672661L, 0xd06016f7L,
428	0x4969474dL, 0x3e6e77dbL, 0xaed16a4aL, 0xd9d65adcL, 0x40df0b66L,
429	0x37d83bf0L, 0xa9bcae53L, 0xdebb9ec5L, 0x47b2cf7fL, 0x30b5ffe9L,
430	0xbdbdf21cL, 0xcabac28aL, 0x53b39330L, 0x24b4a3a6L, 0xbad03605L,
431	0xcdd70693L, 0x54de5729L, 0x23d967bfL, 0xb3667a2eL, 0xc4614ab8L,
432	0x5d681b02L, 0x2a6f2b94L, 0xb40bbe37L, 0xc30c8ea1L, 0x5a05df1bL,
433	0x2d02ef8dL
434};
435
436static __inline u16 RotR1(u16 val)
437{
438	return (val >> 1) | (val << 15);
439}
440
441static __inline u8 Lo8(u16 val)
442{
443	return val & 0xff;
444}
445
446static __inline u8 Hi8(u16 val)
447{
448	return val >> 8;
449}
450
451static __inline u16 Lo16(u32 val)
452{
453	return val & 0xffff;
454}
455
456static __inline u16 Hi16(u32 val)
457{
458	return val >> 16;
459}
460
461static __inline u16 Mk16(u8 hi, u8 lo)
462{
463	return lo | (((u16) hi) << 8);
464}
465
466static __inline u16 Mk16_le(const u16 *v)
467{
468	return le16toh(*v);
469}
470
471static const u16 Sbox[256] = {
472	0xC6A5, 0xF884, 0xEE99, 0xF68D, 0xFF0D, 0xD6BD, 0xDEB1, 0x9154,
473	0x6050, 0x0203, 0xCEA9, 0x567D, 0xE719, 0xB562, 0x4DE6, 0xEC9A,
474	0x8F45, 0x1F9D, 0x8940, 0xFA87, 0xEF15, 0xB2EB, 0x8EC9, 0xFB0B,
475	0x41EC, 0xB367, 0x5FFD, 0x45EA, 0x23BF, 0x53F7, 0xE496, 0x9B5B,
476	0x75C2, 0xE11C, 0x3DAE, 0x4C6A, 0x6C5A, 0x7E41, 0xF502, 0x834F,
477	0x685C, 0x51F4, 0xD134, 0xF908, 0xE293, 0xAB73, 0x6253, 0x2A3F,
478	0x080C, 0x9552, 0x4665, 0x9D5E, 0x3028, 0x37A1, 0x0A0F, 0x2FB5,
479	0x0E09, 0x2436, 0x1B9B, 0xDF3D, 0xCD26, 0x4E69, 0x7FCD, 0xEA9F,
480	0x121B, 0x1D9E, 0x5874, 0x342E, 0x362D, 0xDCB2, 0xB4EE, 0x5BFB,
481	0xA4F6, 0x764D, 0xB761, 0x7DCE, 0x527B, 0xDD3E, 0x5E71, 0x1397,
482	0xA6F5, 0xB968, 0x0000, 0xC12C, 0x4060, 0xE31F, 0x79C8, 0xB6ED,
483	0xD4BE, 0x8D46, 0x67D9, 0x724B, 0x94DE, 0x98D4, 0xB0E8, 0x854A,
484	0xBB6B, 0xC52A, 0x4FE5, 0xED16, 0x86C5, 0x9AD7, 0x6655, 0x1194,
485	0x8ACF, 0xE910, 0x0406, 0xFE81, 0xA0F0, 0x7844, 0x25BA, 0x4BE3,
486	0xA2F3, 0x5DFE, 0x80C0, 0x058A, 0x3FAD, 0x21BC, 0x7048, 0xF104,
487	0x63DF, 0x77C1, 0xAF75, 0x4263, 0x2030, 0xE51A, 0xFD0E, 0xBF6D,
488	0x814C, 0x1814, 0x2635, 0xC32F, 0xBEE1, 0x35A2, 0x88CC, 0x2E39,
489	0x9357, 0x55F2, 0xFC82, 0x7A47, 0xC8AC, 0xBAE7, 0x322B, 0xE695,
490	0xC0A0, 0x1998, 0x9ED1, 0xA37F, 0x4466, 0x547E, 0x3BAB, 0x0B83,
491	0x8CCA, 0xC729, 0x6BD3, 0x283C, 0xA779, 0xBCE2, 0x161D, 0xAD76,
492	0xDB3B, 0x6456, 0x744E, 0x141E, 0x92DB, 0x0C0A, 0x486C, 0xB8E4,
493	0x9F5D, 0xBD6E, 0x43EF, 0xC4A6, 0x39A8, 0x31A4, 0xD337, 0xF28B,
494	0xD532, 0x8B43, 0x6E59, 0xDAB7, 0x018C, 0xB164, 0x9CD2, 0x49E0,
495	0xD8B4, 0xACFA, 0xF307, 0xCF25, 0xCAAF, 0xF48E, 0x47E9, 0x1018,
496	0x6FD5, 0xF088, 0x4A6F, 0x5C72, 0x3824, 0x57F1, 0x73C7, 0x9751,
497	0xCB23, 0xA17C, 0xE89C, 0x3E21, 0x96DD, 0x61DC, 0x0D86, 0x0F85,
498	0xE090, 0x7C42, 0x71C4, 0xCCAA, 0x90D8, 0x0605, 0xF701, 0x1C12,
499	0xC2A3, 0x6A5F, 0xAEF9, 0x69D0, 0x1791, 0x9958, 0x3A27, 0x27B9,
500	0xD938, 0xEB13, 0x2BB3, 0x2233, 0xD2BB, 0xA970, 0x0789, 0x33A7,
501	0x2DB6, 0x3C22, 0x1592, 0xC920, 0x8749, 0xAAFF, 0x5078, 0xA57A,
502	0x038F, 0x59F8, 0x0980, 0x1A17, 0x65DA, 0xD731, 0x84C6, 0xD0B8,
503	0x82C3, 0x29B0, 0x5A77, 0x1E11, 0x7BCB, 0xA8FC, 0x6DD6, 0x2C3A,
504};
505
506static __inline u16 _S_(u16 v)
507{
508	u16 t = Sbox[Hi8(v)];
509	return Sbox[Lo8(v)] ^ ((t << 8) | (t >> 8));
510}
511
512#define PHASE1_LOOP_COUNT 8
513
514static void tkip_mixing_phase1(u16 *TTAK, const u8 *TK, const u8 *TA, u32 IV32)
515{
516	int i, j;
517
518	/* Initialize the 80-bit TTAK from TSC (IV32) and TA[0..5] */
519	TTAK[0] = Lo16(IV32);
520	TTAK[1] = Hi16(IV32);
521	TTAK[2] = Mk16(TA[1], TA[0]);
522	TTAK[3] = Mk16(TA[3], TA[2]);
523	TTAK[4] = Mk16(TA[5], TA[4]);
524
525	for (i = 0; i < PHASE1_LOOP_COUNT; i++) {
526		j = 2 * (i & 1);
527		TTAK[0] += _S_(TTAK[4] ^ Mk16(TK[1 + j], TK[0 + j]));
528		TTAK[1] += _S_(TTAK[0] ^ Mk16(TK[5 + j], TK[4 + j]));
529		TTAK[2] += _S_(TTAK[1] ^ Mk16(TK[9 + j], TK[8 + j]));
530		TTAK[3] += _S_(TTAK[2] ^ Mk16(TK[13 + j], TK[12 + j]));
531		TTAK[4] += _S_(TTAK[3] ^ Mk16(TK[1 + j], TK[0 + j])) + i;
532	}
533}
534
535#ifndef _BYTE_ORDER
536#error "Don't know native byte order"
537#endif
538
539static void tkip_mixing_phase2(u8 *WEPSeed, const u8 *TK, const u16 *TTAK,
540			       u16 IV16)
541{
542	/* Make temporary area overlap WEP seed so that the final copy can be
543	 * avoided on little endian hosts. */
544	u16 *PPK = (u16 *) &WEPSeed[4];
545
546	/* Step 1 - make copy of TTAK and bring in TSC */
547	PPK[0] = TTAK[0];
548	PPK[1] = TTAK[1];
549	PPK[2] = TTAK[2];
550	PPK[3] = TTAK[3];
551	PPK[4] = TTAK[4];
552	PPK[5] = TTAK[4] + IV16;
553
554	/* Step 2 - 96-bit bijective mixing using S-box */
555	PPK[0] += _S_(PPK[5] ^ Mk16_le((const u16 *) &TK[0]));
556	PPK[1] += _S_(PPK[0] ^ Mk16_le((const u16 *) &TK[2]));
557	PPK[2] += _S_(PPK[1] ^ Mk16_le((const u16 *) &TK[4]));
558	PPK[3] += _S_(PPK[2] ^ Mk16_le((const u16 *) &TK[6]));
559	PPK[4] += _S_(PPK[3] ^ Mk16_le((const u16 *) &TK[8]));
560	PPK[5] += _S_(PPK[4] ^ Mk16_le((const u16 *) &TK[10]));
561
562	PPK[0] += RotR1(PPK[5] ^ Mk16_le((const u16 *) &TK[12]));
563	PPK[1] += RotR1(PPK[0] ^ Mk16_le((const u16 *) &TK[14]));
564	PPK[2] += RotR1(PPK[1]);
565	PPK[3] += RotR1(PPK[2]);
566	PPK[4] += RotR1(PPK[3]);
567	PPK[5] += RotR1(PPK[4]);
568
569	/* Step 3 - bring in last of TK bits, assign 24-bit WEP IV value
570	 * WEPSeed[0..2] is transmitted as WEP IV */
571	WEPSeed[0] = Hi8(IV16);
572	WEPSeed[1] = (Hi8(IV16) | 0x20) & 0x7F;
573	WEPSeed[2] = Lo8(IV16);
574	WEPSeed[3] = Lo8((PPK[5] ^ Mk16_le((const u16 *) &TK[0])) >> 1);
575
576#if _BYTE_ORDER == _BIG_ENDIAN
577	{
578		int i;
579		for (i = 0; i < 6; i++)
580			PPK[i] = (PPK[i] << 8) | (PPK[i] >> 8);
581	}
582#endif
583}
584
585static void
586wep_encrypt(u8 *key, struct mbuf *m0, u_int off, size_t data_len,
587	uint8_t icv[IEEE80211_WEP_CRCLEN])
588{
589	u32 i, j, k, crc;
590	size_t buflen;
591	u8 S[256];
592	u8 *pos;
593	struct mbuf *m;
594#define S_SWAP(a,b) do { u8 t = S[a]; S[a] = S[b]; S[b] = t; } while(0)
595
596	/* Setup RC4 state */
597	for (i = 0; i < 256; i++)
598		S[i] = i;
599	j = 0;
600	for (i = 0; i < 256; i++) {
601		j = (j + S[i] + key[i & 0x0f]) & 0xff;
602		S_SWAP(i, j);
603	}
604
605	/* Compute CRC32 over unencrypted data and apply RC4 to data */
606	crc = ~0;
607	i = j = 0;
608	m = m0;
609	pos = mtod(m, uint8_t *) + off;
610	buflen = m->m_len - off;
611	for (;;) {
612		if (buflen > data_len)
613			buflen = data_len;
614		data_len -= buflen;
615		for (k = 0; k < buflen; k++) {
616			crc = crc32_table[(crc ^ *pos) & 0xff] ^ (crc >> 8);
617			i = (i + 1) & 0xff;
618			j = (j + S[i]) & 0xff;
619			S_SWAP(i, j);
620			*pos++ ^= S[(S[i] + S[j]) & 0xff];
621		}
622		m = m->m_next;
623		if (m == NULL) {
624			KASSERT(data_len == 0,
625			    ("out of buffers with data_len %zu\n", data_len));
626			break;
627		}
628		pos = mtod(m, uint8_t *);
629		buflen = m->m_len;
630	}
631	crc = ~crc;
632
633	/* Append little-endian CRC32 and encrypt it to produce ICV */
634	icv[0] = crc;
635	icv[1] = crc >> 8;
636	icv[2] = crc >> 16;
637	icv[3] = crc >> 24;
638	for (k = 0; k < IEEE80211_WEP_CRCLEN; k++) {
639		i = (i + 1) & 0xff;
640		j = (j + S[i]) & 0xff;
641		S_SWAP(i, j);
642		icv[k] ^= S[(S[i] + S[j]) & 0xff];
643	}
644}
645
646static int
647wep_decrypt(u8 *key, struct mbuf *m, u_int off, size_t data_len)
648{
649	u32 i, j, k, crc;
650	u8 S[256];
651	u8 *pos, icv[4];
652	size_t buflen;
653
654	/* Setup RC4 state */
655	for (i = 0; i < 256; i++)
656		S[i] = i;
657	j = 0;
658	for (i = 0; i < 256; i++) {
659		j = (j + S[i] + key[i & 0x0f]) & 0xff;
660		S_SWAP(i, j);
661	}
662
663	/* Apply RC4 to data and compute CRC32 over decrypted data */
664	crc = ~0;
665	i = j = 0;
666	pos = mtod(m, uint8_t *) + off;
667	buflen = m->m_len - off;
668	for (;;) {
669		if (buflen > data_len)
670			buflen = data_len;
671		data_len -= buflen;
672		for (k = 0; k < buflen; k++) {
673			i = (i + 1) & 0xff;
674			j = (j + S[i]) & 0xff;
675			S_SWAP(i, j);
676			*pos ^= S[(S[i] + S[j]) & 0xff];
677			crc = crc32_table[(crc ^ *pos) & 0xff] ^ (crc >> 8);
678			pos++;
679		}
680		m = m->m_next;
681		if (m == NULL) {
682			KASSERT(data_len == 0,
683			    ("out of buffers with data_len %zu\n", data_len));
684			break;
685		}
686		pos = mtod(m, uint8_t *);
687		buflen = m->m_len;
688	}
689	crc = ~crc;
690
691	/* Encrypt little-endian CRC32 and verify that it matches with the
692	 * received ICV */
693	icv[0] = crc;
694	icv[1] = crc >> 8;
695	icv[2] = crc >> 16;
696	icv[3] = crc >> 24;
697	for (k = 0; k < 4; k++) {
698		i = (i + 1) & 0xff;
699		j = (j + S[i]) & 0xff;
700		S_SWAP(i, j);
701		if ((icv[k] ^ S[(S[i] + S[j]) & 0xff]) != *pos++) {
702			/* ICV mismatch - drop frame */
703			return -1;
704		}
705	}
706
707	return 0;
708}
709
710
711static __inline u32 rotl(u32 val, int bits)
712{
713	return (val << bits) | (val >> (32 - bits));
714}
715
716
717static __inline u32 rotr(u32 val, int bits)
718{
719	return (val >> bits) | (val << (32 - bits));
720}
721
722
723static __inline u32 xswap(u32 val)
724{
725	return ((val & 0x00ff00ff) << 8) | ((val & 0xff00ff00) >> 8);
726}
727
728
729#define michael_block(l, r)	\
730do {				\
731	r ^= rotl(l, 17);	\
732	l += r;			\
733	r ^= xswap(l);		\
734	l += r;			\
735	r ^= rotl(l, 3);	\
736	l += r;			\
737	r ^= rotr(l, 2);	\
738	l += r;			\
739} while (0)
740
741
742static __inline u32 get_le32_split(u8 b0, u8 b1, u8 b2, u8 b3)
743{
744	return b0 | (b1 << 8) | (b2 << 16) | (b3 << 24);
745}
746
747static __inline u32 get_le32(const u8 *p)
748{
749	return get_le32_split(p[0], p[1], p[2], p[3]);
750}
751
752
753static __inline void put_le32(u8 *p, u32 v)
754{
755	p[0] = v;
756	p[1] = v >> 8;
757	p[2] = v >> 16;
758	p[3] = v >> 24;
759}
760
761/*
762 * Craft pseudo header used to calculate the MIC.
763 */
764static void
765michael_mic_hdr(const struct ieee80211_frame *wh0, uint8_t hdr[16])
766{
767	const struct ieee80211_frame_addr4 *wh =
768		(const struct ieee80211_frame_addr4 *) wh0;
769
770	switch (wh->i_fc[1] & IEEE80211_FC1_DIR_MASK) {
771	case IEEE80211_FC1_DIR_NODS:
772		IEEE80211_ADDR_COPY(hdr, wh->i_addr1); /* DA */
773		IEEE80211_ADDR_COPY(hdr + IEEE80211_ADDR_LEN, wh->i_addr2);
774		break;
775	case IEEE80211_FC1_DIR_TODS:
776		IEEE80211_ADDR_COPY(hdr, wh->i_addr3); /* DA */
777		IEEE80211_ADDR_COPY(hdr + IEEE80211_ADDR_LEN, wh->i_addr2);
778		break;
779	case IEEE80211_FC1_DIR_FROMDS:
780		IEEE80211_ADDR_COPY(hdr, wh->i_addr1); /* DA */
781		IEEE80211_ADDR_COPY(hdr + IEEE80211_ADDR_LEN, wh->i_addr3);
782		break;
783	case IEEE80211_FC1_DIR_DSTODS:
784		IEEE80211_ADDR_COPY(hdr, wh->i_addr3); /* DA */
785		IEEE80211_ADDR_COPY(hdr + IEEE80211_ADDR_LEN, wh->i_addr4);
786		break;
787	}
788
789	if (wh->i_fc[0] & IEEE80211_FC0_SUBTYPE_QOS) {
790		const struct ieee80211_qosframe *qwh =
791			(const struct ieee80211_qosframe *) wh;
792		hdr[12] = qwh->i_qos[0] & IEEE80211_QOS_TID;
793	} else
794		hdr[12] = 0;
795	hdr[13] = hdr[14] = hdr[15] = 0; /* reserved */
796}
797
798static void
799michael_mic(struct tkip_ctx *ctx, const u8 *key,
800	struct mbuf *m, u_int off, size_t data_len,
801	u8 mic[IEEE80211_WEP_MICLEN])
802{
803	uint8_t hdr[16];
804	u32 l, r;
805	const uint8_t *data;
806	u_int space;
807
808	michael_mic_hdr(mtod(m, struct ieee80211_frame *), hdr);
809
810	l = get_le32(key);
811	r = get_le32(key + 4);
812
813	/* Michael MIC pseudo header: DA, SA, 3 x 0, Priority */
814	l ^= get_le32(hdr);
815	michael_block(l, r);
816	l ^= get_le32(&hdr[4]);
817	michael_block(l, r);
818	l ^= get_le32(&hdr[8]);
819	michael_block(l, r);
820	l ^= get_le32(&hdr[12]);
821	michael_block(l, r);
822
823	/* first buffer has special handling */
824	data = mtod(m, const uint8_t *) + off;
825	space = m->m_len - off;
826	for (;;) {
827		if (space > data_len)
828			space = data_len;
829		/* collect 32-bit blocks from current buffer */
830		while (space >= sizeof(uint32_t)) {
831			l ^= get_le32(data);
832			michael_block(l, r);
833			data += sizeof(uint32_t), space -= sizeof(uint32_t);
834			data_len -= sizeof(uint32_t);
835		}
836		/*
837		 * NB: when space is zero we make one more trip around
838		 * the loop to advance to the next mbuf where there is
839		 * data.  This handles the case where there are 4*n
840		 * bytes in an mbuf followed by <4 bytes in a later mbuf.
841		 * By making an extra trip we'll drop out of the loop
842		 * with m pointing at the mbuf with 3 bytes and space
843		 * set as required by the remainder handling below.
844		 */
845		if (data_len == 0 ||
846		    (data_len < sizeof(uint32_t) && space != 0))
847			break;
848		m = m->m_next;
849		if (m == NULL) {
850			KASSERT(0, ("out of data, data_len %zu\n", data_len));
851			break;
852		}
853		if (space != 0) {
854			const uint8_t *data_next;
855			/*
856			 * Block straddles buffers, split references.
857			 */
858			data_next = mtod(m, const uint8_t *);
859			KASSERT(m->m_len >= sizeof(uint32_t) - space,
860				("not enough data in following buffer, "
861				"m_len %u need %zu\n", m->m_len,
862				sizeof(uint32_t) - space));
863			switch (space) {
864			case 1:
865				l ^= get_le32_split(data[0], data_next[0],
866					data_next[1], data_next[2]);
867				data = data_next + 3;
868				space = m->m_len - 3;
869				break;
870			case 2:
871				l ^= get_le32_split(data[0], data[1],
872					data_next[0], data_next[1]);
873				data = data_next + 2;
874				space = m->m_len - 2;
875				break;
876			case 3:
877				l ^= get_le32_split(data[0], data[1],
878					data[2], data_next[0]);
879				data = data_next + 1;
880				space = m->m_len - 1;
881				break;
882			}
883			michael_block(l, r);
884			data_len -= sizeof(uint32_t);
885		} else {
886			/*
887			 * Setup for next buffer.
888			 */
889			data = mtod(m, const uint8_t *);
890			space = m->m_len;
891		}
892	}
893	/*
894	 * Catch degenerate cases like mbuf[4*n+1 bytes] followed by
895	 * mbuf[2 bytes].  I don't believe these should happen; if they
896	 * do then we'll need more involved logic.
897	 */
898	KASSERT(data_len <= space,
899	    ("not enough data, data_len %zu space %u\n", data_len, space));
900
901	/* Last block and padding (0x5a, 4..7 x 0) */
902	switch (data_len) {
903	case 0:
904		l ^= get_le32_split(0x5a, 0, 0, 0);
905		break;
906	case 1:
907		l ^= get_le32_split(data[0], 0x5a, 0, 0);
908		break;
909	case 2:
910		l ^= get_le32_split(data[0], data[1], 0x5a, 0);
911		break;
912	case 3:
913		l ^= get_le32_split(data[0], data[1], data[2], 0x5a);
914		break;
915	}
916	michael_block(l, r);
917	/* l ^= 0; */
918	michael_block(l, r);
919
920	put_le32(mic, l);
921	put_le32(mic + 4, r);
922}
923
924static int
925tkip_encrypt(struct tkip_ctx *ctx, struct ieee80211_key *key,
926	struct mbuf *m, int hdrlen)
927{
928	struct ieee80211_frame *wh;
929	uint8_t icv[IEEE80211_WEP_CRCLEN];
930
931	ctx->tc_vap->iv_stats.is_crypto_tkip++;
932
933	wh = mtod(m, struct ieee80211_frame *);
934	if (!ctx->tx_phase1_done) {
935		tkip_mixing_phase1(ctx->tx_ttak, key->wk_key, wh->i_addr2,
936				   (u32)(key->wk_keytsc >> 16));
937		ctx->tx_phase1_done = 1;
938	}
939	tkip_mixing_phase2(ctx->tx_rc4key, key->wk_key, ctx->tx_ttak,
940		(u16) key->wk_keytsc);
941
942	wep_encrypt(ctx->tx_rc4key,
943		m, hdrlen + tkip.ic_header,
944		m->m_pkthdr.len - (hdrlen + tkip.ic_header),
945		icv);
946	(void) m_append(m, IEEE80211_WEP_CRCLEN, icv);	/* XXX check return */
947
948	key->wk_keytsc++;
949	if ((u16)(key->wk_keytsc) == 0)
950		ctx->tx_phase1_done = 0;
951	return 1;
952}
953
954static int
955tkip_decrypt(struct tkip_ctx *ctx, struct ieee80211_key *key,
956	struct mbuf *m, int hdrlen)
957{
958	struct ieee80211_frame *wh;
959	struct ieee80211vap *vap = ctx->tc_vap;
960	u32 iv32;
961	u16 iv16;
962	u8 tid;
963
964	vap->iv_stats.is_crypto_tkip++;
965
966	wh = mtod(m, struct ieee80211_frame *);
967	/* NB: tkip_decap already verified header and left seq in rx_rsc */
968	iv16 = (u16) ctx->rx_rsc;
969	iv32 = (u32) (ctx->rx_rsc >> 16);
970
971	tid = ieee80211_gettid(wh);
972	if (iv32 != (u32)(key->wk_keyrsc[tid] >> 16) || !ctx->rx_phase1_done) {
973		tkip_mixing_phase1(ctx->rx_ttak, key->wk_key,
974			wh->i_addr2, iv32);
975		ctx->rx_phase1_done = 1;
976	}
977	tkip_mixing_phase2(ctx->rx_rc4key, key->wk_key, ctx->rx_ttak, iv16);
978
979	/* NB: m is unstripped; deduct headers + ICV to get payload */
980	if (wep_decrypt(ctx->rx_rc4key,
981		m, hdrlen + tkip.ic_header,
982	        m->m_pkthdr.len - (hdrlen + tkip.ic_header + tkip.ic_trailer))) {
983		if (iv32 != (u32)(key->wk_keyrsc[tid] >> 16)) {
984			/* Previously cached Phase1 result was already lost, so
985			 * it needs to be recalculated for the next packet. */
986			ctx->rx_phase1_done = 0;
987		}
988		IEEE80211_NOTE_MAC(vap, IEEE80211_MSG_CRYPTO, wh->i_addr2,
989		    "%s", "TKIP ICV mismatch on decrypt");
990		vap->iv_stats.is_rx_tkipicv++;
991		return 0;
992	}
993	return 1;
994}
995
996/*
997 * Module glue.
998 */
999IEEE80211_CRYPTO_MODULE(tkip, 1);
1000