git.gag.com Git - fw/altos/blob - src/kernel/ao_radio_cmac.c

   1 /*
   2  * Copyright © 2011 Keith Packard <keithp@keithp.com>
   3  *
   4  * This program is free software; you can redistribute it and/or modify
   5  * it under the terms of the GNU General Public License as published by
   6  * the Free Software Foundation; version 2 of the License.
   7  *
   8  * This program is distributed in the hope that it will be useful, but
   9  * WITHOUT ANY WARRANTY; without even the implied warranty of
  10  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  11  * General Public License for more details.
  12  *
  13  * You should have received a copy of the GNU General Public License along
  14  * with this program; if not, write to the Free Software Foundation, Inc.,
  15  * 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA.
  16  */
  17
  18 #include <ao.h>
  19 #include <ao_radio_cmac.h>
  20
  21 static __xdata uint8_t ao_radio_cmac_mutex;
  22 __pdata int8_t ao_radio_cmac_rssi;
  23 static __xdata uint8_t cmac_data[AO_CMAC_MAX_LEN + AO_CMAC_KEY_LEN + 2 + AO_CMAC_KEY_LEN];
  24
  25 static uint8_t
  26 round_len(uint8_t len)
  27 {
  28         uint8_t rem;
  29
  30         /* Make sure we transfer at least one packet, and
  31          * then make sure every packet is full. Note that
  32          * there is no length encoded, and that the receiver
  33          * must deal with any extra bytes in the packet
  34          */
  35         if (len < AO_CMAC_KEY_LEN)
  36                 len = AO_CMAC_KEY_LEN;
  37         rem = len % AO_CMAC_KEY_LEN;
  38         if (rem != 0)
  39                 len += (AO_CMAC_KEY_LEN - rem);
  40         return len;
  41 }
  42
  43 /*
  44  * Sign and deliver the data sitting in the cmac buffer
  45  */
  46 static void
  47 radio_cmac_send(uint8_t len) __reentrant
  48 {
  49         uint8_t i;
  50
  51         len = round_len(len);
  52         /* Make sure the AES key is loaded */
  53         ao_config_get();
  54
  55 #if HAS_MONITOR
  56         ao_monitor_set(0);
  57 #endif
  58
  59         ao_mutex_get(&ao_aes_mutex);
  60         ao_aes_set_mode(ao_aes_mode_cbc_mac);
  61         ao_aes_set_key(ao_config.aes_key);
  62         ao_aes_zero_iv();
  63         for (i = 0; i < len; i += AO_CMAC_KEY_LEN) {
  64                 if (i + AO_CMAC_KEY_LEN < len)
  65                         ao_aes_run(&cmac_data[i], NULL);
  66                 else
  67                         ao_aes_run(&cmac_data[i], &cmac_data[len]);
  68         }
  69         ao_mutex_put(&ao_aes_mutex);
  70
  71         ao_radio_send(cmac_data, len + AO_CMAC_KEY_LEN);
  72 }
  73
  74 /*
  75  * Receive and validate an incoming packet
  76  */
  77
  78 static int8_t
  79 radio_cmac_recv(uint8_t len, uint16_t timeout) __reentrant
  80 {
  81         uint8_t i;
  82
  83         len = round_len(len);
  84 #if HAS_MONITOR
  85         ao_monitor_set(0);
  86 #endif
  87         i = ao_radio_recv(cmac_data, len + AO_CMAC_KEY_LEN + 2, timeout);
  88
  89         if (!i) {
  90                 ao_radio_cmac_rssi = 0;
  91                 return AO_RADIO_CMAC_TIMEOUT;
  92         }
  93
  94         if (!(cmac_data[len + AO_CMAC_KEY_LEN +1] & AO_RADIO_STATUS_CRC_OK))
  95                 return AO_RADIO_CMAC_CRC_ERROR;
  96
  97         ao_config_get();
  98
  99         /* Compute the packet signature
 100          */
 101         ao_mutex_get(&ao_aes_mutex);
 102         ao_aes_set_mode(ao_aes_mode_cbc_mac);
 103         ao_aes_set_key(ao_config.aes_key);
 104         ao_aes_zero_iv();
 105         for (i = 0; i < len; i += AO_CMAC_KEY_LEN) {
 106                 if (i + AO_CMAC_KEY_LEN < len)
 107                         ao_aes_run(&cmac_data[i], NULL);
 108                 else
 109                         ao_aes_run(&cmac_data[i], &cmac_data[len + AO_CMAC_KEY_LEN + 2]);
 110         }
 111         ao_mutex_put(&ao_aes_mutex);
 112
 113         /* Check the packet signature against the signature provided
 114          * over the link
 115          */
 116
 117         if (memcmp(&cmac_data[len],
 118                    &cmac_data[len + AO_CMAC_KEY_LEN + 2],
 119                    AO_CMAC_KEY_LEN) != 0) {
 120                 return AO_RADIO_CMAC_MAC_ERROR;
 121         }
 122
 123         ao_radio_cmac_rssi = ao_radio_rssi;
 124
 125         return AO_RADIO_CMAC_OK;
 126 }
 127
 128 int8_t
 129 ao_radio_cmac_send(__xdata void *packet, uint8_t len) __reentrant
 130 {
 131         if (len > AO_CMAC_MAX_LEN)
 132                 return AO_RADIO_CMAC_LEN_ERROR;
 133         ao_mutex_get(&ao_radio_cmac_mutex);
 134         ao_xmemcpy(cmac_data, packet, len);
 135 #if AO_LED_TX
 136         ao_led_on(AO_LED_TX);
 137 #endif
 138         radio_cmac_send(len);
 139 #if AO_LED_TX
 140         ao_led_off(AO_LED_TX);
 141 #endif
 142         ao_mutex_put(&ao_radio_cmac_mutex);
 143         return AO_RADIO_CMAC_OK;
 144 }
 145
 146 int8_t
 147 ao_radio_cmac_recv(__xdata void *packet, uint8_t len, uint16_t timeout) __reentrant
 148 {
 149         int8_t  i;
 150         if (len > AO_CMAC_MAX_LEN)
 151                 return AO_RADIO_CMAC_LEN_ERROR;
 152         ao_mutex_get(&ao_radio_cmac_mutex);
 153 #if AO_LED_RX
 154         ao_led_on(AO_LED_RX);
 155 #endif
 156         i = radio_cmac_recv(len, timeout);
 157 #if AO_LED_RX
 158         ao_led_off(AO_LED_RX);
 159 #endif
 160         if (i == AO_RADIO_CMAC_OK)
 161                 ao_xmemcpy(packet, cmac_data, len);
 162         ao_mutex_put(&ao_radio_cmac_mutex);
 163         return i;
 164 }