Quelle  rt2x00crypto.c   Sprache: C

// SPDX-License-Identifier: GPL-2.0-or-later
/*
 Copyright (C) 2004 - 2009 Ivo van Doorn <IvDoorn@gmail.com>
 <http://rt2x00.serialmonkey.com>

 */

/*
 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->cipher) {
 case WLAN_CIPHER_SUITE_WEP40:
  return CIPHER_WEP64;
 case WLAN_CIPHER_SUITE_WEP104:
  return CIPHER_WEP128;
 case WLAN_CIPHER_SUITE_TKIP:
  return CIPHER_TKIP;
 case WLAN_CIPHER_SUITE_CCMP:
  return CIPHER_AES;
 default:
  return CIPHER_NONE;
 }
}

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

 if (!rt2x00_has_cap_hw_crypto(rt2x00dev) || !hw_key)
  return;

 __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 = txdesc->header_length;
 txdesc->iv_len = hw_key->iv_len;

 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 rt2x00_dev *rt2x00dev,
          struct sk_buff *skb)
{
 struct ieee80211_tx_info *tx_info = IEEE80211_SKB_CB(skb);
 struct ieee80211_key_conf *key = tx_info->control.hw_key;
 unsigned int overhead = 0;

 if (!rt2x00_has_cap_hw_crypto(rt2x00dev) || !key)
  return overhead;

 /*
  * 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->cipher == WLAN_CIPHER_SUITE_TKIP)
   overhead += 8;
 }

 return overhead;
}

void rt2x00crypto_tx_copy_iv(struct sk_buff *skb, struct txentry_desc *txdesc)
{
 struct skb_frame_desc *skbdesc = get_skb_frame_desc(skb);

 if (unlikely(!txdesc->iv_len))
  return;

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

void rt2x00crypto_tx_remove_iv(struct sk_buff *skb, struct txentry_desc *txdesc)
{
 struct skb_frame_desc *skbdesc = get_skb_frame_desc(skb);

 if (unlikely(!txdesc->iv_len))
  return;

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

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

 /* Pull buffer to correct size */
 skb_pull(skb, txdesc->iv_len);
 txdesc->length -= txdesc->iv_len;

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

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

 if (!(skbdesc->flags & SKBDESC_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 &= ~SKBDESC_IV_STRIPPED;
}

void rt2x00crypto_rx_insert_iv(struct sk_buff *skb,
          unsigned int header_length,
          struct rxdone_entry_desc *rxdesc)
{
 unsigned int payload_len = rxdesc->size - header_length;
 unsigned int align = ALIGN_SIZE(skb, 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. There are 2 possibilities
  * either the alignment is already present between
  * the 802.11 header and payload. In that case we
  * have to move the header less than the iv_len
  * since we can use the already available l2pad bytes
  * for the iv data.
  * When the alignment must be added manually we must
  * move the header more then iv_len since we must
  * make room for the payload move as well.
 */
 if (rxdesc->dev_flags & RXDONE_L2PAD) {
  skb_push(skb, iv_len - align);
  skb_put(skb, icv_len);

  /* Move ieee80211 header */
  memmove(skb->data + transfer,
   skb->data + transfer + (iv_len - align),
   header_length);
  transfer += header_length;
 } else {
  skb_push(skb, iv_len + align);
  skb_put(skb, icv_len - align);

  /* 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 for alignment purposes. Note that
  * this is only needed when no l2 padding is present.
 */
 if (!(rxdesc->dev_flags & RXDONE_L2PAD)) {
  memmove(skb->data + transfer,
   skb->data + transfer + align,
   payload_len);
 }

 /*
  * NOTE: Always count the payload as transferred,
  * 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;
}