[deliverable/linux.git] / drivers / net / wireless / rt2x00 / rt2x00crypto.c

/*
	Copyright (C) 2004 - 2008 rt2x00 SourceForge Project
	<http://rt2x00.serialmonkey.com>

	This program is free software; you can redistribute it and/or modify
	it under the terms of the GNU General Public License as published by
	the Free Software Foundation; either version 2 of the License, or
	(at your option) any later version.

	This program is distributed in the hope that it will be useful,
	but WITHOUT ANY WARRANTY; without even the implied warranty of
	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
	GNU General Public License for more details.

	You should have received a copy of the GNU General Public License
	along with this program; if not, write to the
	Free Software Foundation, Inc.,
	59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
 */

/*
	Module: rt2x00lib
	Abstract: rt2x00 crypto specific routines.
 */

#include <linux/kernel.h>
#include <linux/module.h>

#include "rt2x00.h"
#include "rt2x00lib.h"

enum cipher rt2x00crypto_key_to_cipher(struct ieee80211_key_conf *key)
{
	switch (key->alg) {
	case ALG_WEP:
		if (key->keylen == LEN_WEP40)
			return CIPHER_WEP64;
		else
			return CIPHER_WEP128;
	case ALG_TKIP:
		return CIPHER_TKIP;
	case ALG_CCMP:
		return CIPHER_AES;
	default:
		return CIPHER_NONE;
	}
}

void rt2x00crypto_create_tx_descriptor(struct queue_entry *entry,
				       struct txentry_desc *txdesc)
{
	struct ieee80211_tx_info *tx_info = IEEE80211_SKB_CB(entry->skb);
	struct ieee80211_key_conf *hw_key = tx_info->control.hw_key;

	__set_bit(ENTRY_TXD_ENCRYPT, &txdesc->flags);

	txdesc->cipher = rt2x00crypto_key_to_cipher(hw_key);

	if (hw_key->flags & IEEE80211_KEY_FLAG_PAIRWISE)
		__set_bit(ENTRY_TXD_ENCRYPT_PAIRWISE, &txdesc->flags);

	txdesc->key_idx = hw_key->hw_key_idx;
	txdesc->iv_offset = ieee80211_get_hdrlen_from_skb(entry->skb);

	if (!(hw_key->flags & IEEE80211_KEY_FLAG_GENERATE_IV))
		__set_bit(ENTRY_TXD_ENCRYPT_IV, &txdesc->flags);

	if (!(hw_key->flags & IEEE80211_KEY_FLAG_GENERATE_MMIC))
		__set_bit(ENTRY_TXD_ENCRYPT_MMIC, &txdesc->flags);
}

unsigned int rt2x00crypto_tx_overhead(struct ieee80211_tx_info *tx_info)
{
	struct ieee80211_key_conf *key = tx_info->control.hw_key;
	unsigned int overhead = 0;

	/*
	 * Extend frame length to include IV/EIV/ICV/MMIC,
	 * note that these lengths should only be added when
	 * mac80211 does not generate it.
	 */
	overhead += key->icv_len;

	if (!(key->flags & IEEE80211_KEY_FLAG_GENERATE_IV))
		overhead += key->iv_len;

	if (!(key->flags & IEEE80211_KEY_FLAG_GENERATE_MMIC)) {
		if (key->alg == ALG_TKIP)
			overhead += 8;
	}

	return overhead;
}

void rt2x00crypto_tx_copy_iv(struct sk_buff *skb, unsigned int iv_len)
{
	struct skb_frame_desc *skbdesc = get_skb_frame_desc(skb);
	unsigned int header_length = ieee80211_get_hdrlen_from_skb(skb);

	if (unlikely(!iv_len))
		return;

	/* Copy IV/EIV data */
	memcpy(skbdesc->iv, skb->data + header_length, iv_len);
}

void rt2x00crypto_tx_remove_iv(struct sk_buff *skb, unsigned int iv_len)
{
	struct skb_frame_desc *skbdesc = get_skb_frame_desc(skb);
	unsigned int header_length = ieee80211_get_hdrlen_from_skb(skb);

	if (unlikely(!iv_len))
		return;

	/* Copy IV/EIV data */
	memcpy(skbdesc->iv, skb->data + header_length, iv_len);

	/* Move ieee80211 header */
	memmove(skb->data + iv_len, skb->data, header_length);

	/* Pull buffer to correct size */
	skb_pull(skb, iv_len);

	/* IV/EIV data has officially be stripped */
	skbdesc->flags |= FRAME_DESC_IV_STRIPPED;
}

void rt2x00crypto_tx_insert_iv(struct sk_buff *skb)
{
	struct skb_frame_desc *skbdesc = get_skb_frame_desc(skb);
	unsigned int header_length = ieee80211_get_hdrlen_from_skb(skb);
	const unsigned int iv_len =
	    ((!!(skbdesc->iv[0])) * 4) + ((!!(skbdesc->iv[1])) * 4);

	if (!(skbdesc->flags & FRAME_DESC_IV_STRIPPED))
		return;

	skb_push(skb, iv_len);

	/* Move ieee80211 header */
	memmove(skb->data, skb->data + iv_len, header_length);

	/* Copy IV/EIV data */
	memcpy(skb->data + header_length, skbdesc->iv, iv_len);

	/* IV/EIV data has returned into the frame */
	skbdesc->flags &= ~FRAME_DESC_IV_STRIPPED;
}

void rt2x00crypto_rx_insert_iv(struct sk_buff *skb, unsigned int align,
			       unsigned int header_length,
			       struct rxdone_entry_desc *rxdesc)
{
	unsigned int payload_len = rxdesc->size - header_length;
	unsigned int iv_len;
	unsigned int icv_len;
	unsigned int transfer = 0;

	/*
	 * WEP64/WEP128: Provides IV & ICV
	 * TKIP: Provides IV/EIV & ICV
	 * AES: Provies IV/EIV & ICV
	 */
	switch (rxdesc->cipher) {
	case CIPHER_WEP64:
	case CIPHER_WEP128:
		iv_len = 4;
		icv_len = 4;
		break;
	case CIPHER_TKIP:
		iv_len = 8;
		icv_len = 4;
		break;
	case CIPHER_AES:
		iv_len = 8;
		icv_len = 8;
		break;
	default:
		/* Unsupport type */
		return;
	}

	/*
	 * Make room for new data, note that we increase both
	 * headsize and tailsize when required. The tailsize is
	 * only needed when ICV data needs to be inserted and
	 * the padding is smaller than the ICV data.
	 * When alignment requirements is greater than the
	 * ICV data we must trim the skb to the correct size
	 * because we need to remove the extra bytes.
	 */
	skb_push(skb, iv_len + align);
	if (align < icv_len)
		skb_put(skb, icv_len - align);
	else if (align > icv_len)
		skb_trim(skb, rxdesc->size + iv_len + icv_len);

	/* Move ieee80211 header */
	memmove(skb->data + transfer,
		skb->data + transfer + iv_len + align,
		header_length);
	transfer += header_length;

	/* Copy IV/EIV data */
	memcpy(skb->data + transfer, rxdesc->iv, iv_len);
	transfer += iv_len;

	/* Move payload */
	if (align) {
		memmove(skb->data + transfer,
			skb->data + transfer + align,
			payload_len);
	}

	/*
	 * NOTE: Always count the payload as transfered,
	 * even when alignment was set to zero. This is required
	 * for determining the correct offset for the ICV data.
	 */
	transfer += payload_len;

	/*
	 * Copy ICV data
	 * AES appends 8 bytes, we can't fill the upper
	 * 4 bytes, but mac80211 doesn't care about what
	 * we provide here anyway and strips it immediately.
	 */
	memcpy(skb->data + transfer, &rxdesc->icv, 4);
	transfer += icv_len;

	/* IV/EIV/ICV has been inserted into frame */
	rxdesc->size = transfer;
	rxdesc->flags &= ~RX_FLAG_IV_STRIPPED;
}
Commit	Line	Data
2bb057d0 ID	1	/*
	2	Copyright (C) 2004 - 2008 rt2x00 SourceForge Project
	3	<http://rt2x00.serialmonkey.com>
	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 as published by
	7	the Free Software Foundation; either version 2 of the License, or
	8	(at your option) any later version.
	9
	10	This program is distributed in the hope that it will be useful,
	11	but WITHOUT ANY WARRANTY; without even the implied warranty of
	12	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
	13	GNU General Public License for more details.
	14
	15	You should have received a copy of the GNU General Public License
	16	along with this program; if not, write to the
	17	Free Software Foundation, Inc.,
	18	59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
	19	*/
	20
	21	/*
	22	Module: rt2x00lib
	23	Abstract: rt2x00 crypto specific routines.
	24	*/
	25
	26	#include <linux/kernel.h>
	27	#include <linux/module.h>
	28
	29	#include "rt2x00.h"
	30	#include "rt2x00lib.h"
	31
	32	enum cipher rt2x00crypto_key_to_cipher(struct ieee80211_key_conf *key)
	33	{
	34	switch (key->alg) {
	35	case ALG_WEP:
	36	if (key->keylen == LEN_WEP40)
	37	return CIPHER_WEP64;
	38	else
	39	return CIPHER_WEP128;
	40	case ALG_TKIP:
	41	return CIPHER_TKIP;
	42	case ALG_CCMP:
	43	return CIPHER_AES;
	44	default:
	45	return CIPHER_NONE;
	46	}
	47	}
	48
9c3444d3 ID	49	void rt2x00crypto_create_tx_descriptor(struct queue_entry *entry,
	50	struct txentry_desc *txdesc)
	51	{
	52	struct ieee80211_tx_info *tx_info = IEEE80211_SKB_CB(entry->skb);
	53	struct ieee80211_key_conf *hw_key = tx_info->control.hw_key;
	54
	55	__set_bit(ENTRY_TXD_ENCRYPT, &txdesc->flags);
	56
	57	txdesc->cipher = rt2x00crypto_key_to_cipher(hw_key);
	58
	59	if (hw_key->flags & IEEE80211_KEY_FLAG_PAIRWISE)
	60	__set_bit(ENTRY_TXD_ENCRYPT_PAIRWISE, &txdesc->flags);
	61
	62	txdesc->key_idx = hw_key->hw_key_idx;
	63	txdesc->iv_offset = ieee80211_get_hdrlen_from_skb(entry->skb);
	64
	65	if (!(hw_key->flags & IEEE80211_KEY_FLAG_GENERATE_IV))
	66	__set_bit(ENTRY_TXD_ENCRYPT_IV, &txdesc->flags);
	67
	68	if (!(hw_key->flags & IEEE80211_KEY_FLAG_GENERATE_MMIC))
	69	__set_bit(ENTRY_TXD_ENCRYPT_MMIC, &txdesc->flags);
	70	}
	71
2bb057d0 ID	72	unsigned int rt2x00crypto_tx_overhead(struct ieee80211_tx_info *tx_info)
	73	{
	74	struct ieee80211_key_conf *key = tx_info->control.hw_key;
	75	unsigned int overhead = 0;
	76
	77	/*
	78	* Extend frame length to include IV/EIV/ICV/MMIC,
	79	* note that these lengths should only be added when
	80	* mac80211 does not generate it.
	81	*/
76708dee	82	overhead += key->icv_len;
2bb057d0 ID	83
2bb057d0 ID	84	if (!(key->flags & IEEE80211_KEY_FLAG_GENERATE_IV))
76708dee	85	overhead += key->iv_len;
2bb057d0 ID	86
	87	if (!(key->flags & IEEE80211_KEY_FLAG_GENERATE_MMIC)) {
	88	if (key->alg == ALG_TKIP)
	89	overhead += 8;
	90	}
	91
	92	return overhead;
	93	}
	94
dddfb478 ID	95	void rt2x00crypto_tx_copy_iv(struct sk_buff *skb, unsigned int iv_len)
	96	{
	97	struct skb_frame_desc *skbdesc = get_skb_frame_desc(skb);
	98	unsigned int header_length = ieee80211_get_hdrlen_from_skb(skb);
	99
	100	if (unlikely(!iv_len))
	101	return;
	102
	103	/* Copy IV/EIV data */
	104	memcpy(skbdesc->iv, skb->data + header_length, iv_len);
	105	}
	106
2bb057d0 ID	107	void rt2x00crypto_tx_remove_iv(struct sk_buff *skb, unsigned int iv_len)
	108	{
	109	struct skb_frame_desc *skbdesc = get_skb_frame_desc(skb);
	110	unsigned int header_length = ieee80211_get_hdrlen_from_skb(skb);
	111
	112	if (unlikely(!iv_len))
	113	return;
	114
	115	/* Copy IV/EIV data */
1ce9cdac	116	memcpy(skbdesc->iv, skb->data + header_length, iv_len);
2bb057d0 ID	117
	118	/* Move ieee80211 header */
	119	memmove(skb->data + iv_len, skb->data, header_length);
	120
	121	/* Pull buffer to correct size */
	122	skb_pull(skb, iv_len);
	123
	124	/* IV/EIV data has officially be stripped */
	125	skbdesc->flags \|= FRAME_DESC_IV_STRIPPED;
	126	}
	127
	128	void rt2x00crypto_tx_insert_iv(struct sk_buff *skb)
	129	{
	130	struct skb_frame_desc *skbdesc = get_skb_frame_desc(skb);
	131	unsigned int header_length = ieee80211_get_hdrlen_from_skb(skb);
	132	const unsigned int iv_len =
1ce9cdac	133	((!!(skbdesc->iv[0])) * 4) + ((!!(skbdesc->iv[1])) * 4);
2bb057d0 ID	134
	135	if (!(skbdesc->flags & FRAME_DESC_IV_STRIPPED))
	136	return;
	137
	138	skb_push(skb, iv_len);
	139
	140	/* Move ieee80211 header */
	141	memmove(skb->data, skb->data + iv_len, header_length);
	142
	143	/* Copy IV/EIV data */
1ce9cdac	144	memcpy(skb->data + header_length, skbdesc->iv, iv_len);
2bb057d0 ID	145
	146	/* IV/EIV data has returned into the frame */
	147	skbdesc->flags &= ~FRAME_DESC_IV_STRIPPED;
	148	}
	149
	150	void rt2x00crypto_rx_insert_iv(struct sk_buff *skb, unsigned int align,
	151	unsigned int header_length,
	152	struct rxdone_entry_desc *rxdesc)
	153	{
	154	unsigned int payload_len = rxdesc->size - header_length;
	155	unsigned int iv_len;
	156	unsigned int icv_len;
	157	unsigned int transfer = 0;
	158
	159	/*
	160	* WEP64/WEP128: Provides IV & ICV
	161	* TKIP: Provides IV/EIV & ICV
	162	* AES: Provies IV/EIV & ICV
	163	*/
	164	switch (rxdesc->cipher) {
	165	case CIPHER_WEP64:
	166	case CIPHER_WEP128:
	167	iv_len = 4;
	168	icv_len = 4;
	169	break;
	170	case CIPHER_TKIP:
	171	iv_len = 8;
	172	icv_len = 4;
	173	break;
	174	case CIPHER_AES:
	175	iv_len = 8;
	176	icv_len = 8;
	177	break;
	178	default:
	179	/* Unsupport type */
	180	return;
	181	}
	182
	183	/*
	184	* Make room for new data, note that we increase both
	185	* headsize and tailsize when required. The tailsize is
	186	* only needed when ICV data needs to be inserted and
025dfdaf FS	187	* the padding is smaller than the ICV data.
025dfdaf FS	188	* When alignment requirements is greater than the
2bb057d0 ID	189	* ICV data we must trim the skb to the correct size
	190	* because we need to remove the extra bytes.
	191	*/
	192	skb_push(skb, iv_len + align);
	193	if (align < icv_len)
	194	skb_put(skb, icv_len - align);
	195	else if (align > icv_len)
	196	skb_trim(skb, rxdesc->size + iv_len + icv_len);
	197
	198	/* Move ieee80211 header */
	199	memmove(skb->data + transfer,
	200	skb->data + transfer + iv_len + align,
	201	header_length);
	202	transfer += header_length;
	203
1ce9cdac ID	204	/* Copy IV/EIV data */
	205	memcpy(skb->data + transfer, rxdesc->iv, iv_len);
	206	transfer += iv_len;
2bb057d0 ID	207
	208	/* Move payload */
	209	if (align) {
	210	memmove(skb->data + transfer,
	211	skb->data + transfer + align,
	212	payload_len);
	213	}
	214
	215	/*
	216	* NOTE: Always count the payload as transfered,
	217	* even when alignment was set to zero. This is required
	218	* for determining the correct offset for the ICV data.
	219	*/
	220	transfer += payload_len;
	221
1ce9cdac ID	222	/*
	223	* Copy ICV data
	224	* AES appends 8 bytes, we can't fill the upper
	225	* 4 bytes, but mac80211 doesn't care about what
	226	* we provide here anyway and strips it immediately.
	227	*/
	228	memcpy(skb->data + transfer, &rxdesc->icv, 4);
	229	transfer += icv_len;
2bb057d0 ID	230
	231	/* IV/EIV/ICV has been inserted into frame */
	232	rxdesc->size = transfer;
	233	rxdesc->flags &= ~RX_FLAG_IV_STRIPPED;
	234	}