2 * Copyright © 2011 Keith Packard <keithp@keithp.com>
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.
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.
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.
20 #define AO_CMAC_KEY_LEN AO_AES_LEN
21 #define AO_CMAC_MAX_LEN (128 - AO_CMAC_KEY_LEN)
23 static __xdata uint8_t ao_radio_cmac_mutex;
24 __pdata int16_t ao_radio_cmac_rssi;
25 static __xdata uint8_t cmac_data[AO_CMAC_MAX_LEN + AO_CMAC_KEY_LEN + 2 + AO_CMAC_KEY_LEN];
26 static __pdata uint8_t ao_radio_cmac_len;
33 b = ao_cmd_hexchar(getchar());
35 ao_cmd_status = ao_cmd_lex_error;
51 round_len(uint8_t len)
55 /* Make sure we transfer at least one packet, and
56 * then make sure every packet is full. Note that
57 * there is no length encoded, and that the receiver
58 * must deal with any extra bytes in the packet
60 if (len < AO_CMAC_KEY_LEN)
61 len = AO_CMAC_KEY_LEN;
62 rem = len % AO_CMAC_KEY_LEN;
64 len += (AO_CMAC_KEY_LEN - rem);
69 * Sign and deliver the data sitting in the cmac buffer
72 radio_cmac_send(uint8_t len) __reentrant
77 /* Make sure the AES key is loaded */
84 ao_mutex_get(&ao_aes_mutex);
85 ao_aes_set_mode(ao_aes_mode_cbc_mac);
86 ao_aes_set_key(ao_config.aes_key);
88 for (i = 0; i < len; i += AO_CMAC_KEY_LEN) {
89 if (i + AO_CMAC_KEY_LEN < len)
90 ao_aes_run(&cmac_data[i], NULL);
92 ao_aes_run(&cmac_data[i], &cmac_data[len]);
94 ao_mutex_put(&ao_aes_mutex);
96 ao_radio_send(cmac_data, len + AO_CMAC_KEY_LEN);
100 * Receive and validate an incoming packet
104 radio_cmac_recv(uint8_t len, uint16_t timeout) __reentrant
108 len = round_len(len);
115 i = ao_radio_recv(cmac_data, len + AO_CMAC_KEY_LEN + 2);
119 ao_radio_cmac_rssi = 0;
120 return AO_RADIO_CMAC_TIMEOUT;
123 ao_radio_cmac_rssi = (int16_t) (((int8_t) cmac_data[len + AO_CMAC_KEY_LEN]) >> 1) - 74;
124 if (!(cmac_data[len + AO_CMAC_KEY_LEN +1] & PKT_APPEND_STATUS_1_CRC_OK))
125 return AO_RADIO_CMAC_CRC_ERROR;
129 /* Compute the packet signature
131 ao_mutex_get(&ao_aes_mutex);
132 ao_aes_set_mode(ao_aes_mode_cbc_mac);
133 ao_aes_set_key(ao_config.aes_key);
135 for (i = 0; i < len; i += AO_CMAC_KEY_LEN) {
136 if (i + AO_CMAC_KEY_LEN < len)
137 ao_aes_run(&cmac_data[i], NULL);
139 ao_aes_run(&cmac_data[i], &cmac_data[len + AO_CMAC_KEY_LEN + 2]);
141 ao_mutex_put(&ao_aes_mutex);
143 /* Check the packet signature against the signature provided
147 if (memcmp(&cmac_data[len],
148 &cmac_data[len + AO_CMAC_KEY_LEN + 2],
149 AO_CMAC_KEY_LEN) != 0) {
150 return AO_RADIO_CMAC_MAC_ERROR;
153 return AO_RADIO_CMAC_OK;
157 ao_radio_cmac_send(__xdata void *packet, uint8_t len) __reentrant
159 if (len > AO_CMAC_MAX_LEN)
160 return AO_RADIO_CMAC_LEN_ERROR;
161 ao_mutex_get(&ao_radio_cmac_mutex);
162 memcpy(cmac_data, packet, len);
163 radio_cmac_send(len);
164 ao_mutex_put(&ao_radio_cmac_mutex);
165 return AO_RADIO_CMAC_OK;
169 ao_radio_cmac_recv(__xdata void *packet, uint8_t len, uint16_t timeout) __reentrant
172 if (len > AO_CMAC_MAX_LEN)
173 return AO_RADIO_CMAC_LEN_ERROR;
174 ao_mutex_get(&ao_radio_cmac_mutex);
175 i = radio_cmac_recv(len, timeout);
176 if (i == AO_RADIO_CMAC_OK)
177 memcpy(packet, cmac_data, len);
178 ao_mutex_put(&ao_radio_cmac_mutex);
183 radio_cmac_send_cmd(void) __reentrant
189 if (ao_cmd_status != ao_cmd_success)
192 if (len > AO_CMAC_MAX_LEN) {
193 ao_cmd_status = ao_cmd_syntax_error;
197 ao_mutex_get(&ao_radio_cmac_mutex);
199 for (i = 0; i < len; i++) {
200 cmac_data[i] = getbyte();
201 if (ao_cmd_status != ao_cmd_success)
204 radio_cmac_send(len);
205 ao_mutex_put(&ao_radio_cmac_mutex);
209 radio_cmac_recv_cmd(void) __reentrant
215 if (ao_cmd_status != ao_cmd_success)
219 if (ao_cmd_status != ao_cmd_success)
221 timeout = AO_MS_TO_TICKS(ao_cmd_lex_i);
222 ao_mutex_get(&ao_radio_cmac_mutex);
223 i = radio_cmac_recv(len, timeout);
224 if (i == AO_RADIO_CMAC_OK) {
226 for (i = 0; i < len; i++)
227 printf("%02x", cmac_data[i]);
228 printf (" %d\n", ao_radio_cmac_rssi);
230 printf ("ERROR %d %d\n", i, ao_radio_cmac_rssi);
231 ao_mutex_put(&ao_radio_cmac_mutex);
234 static __xdata struct ao_launch_command command;
235 static __xdata struct ao_launch_query query;
236 static pdata uint16_t launch_serial;
237 static pdata uint8_t launch_channel;
238 static pdata uint16_t tick_offset;
241 launch_args(void) __reentrant
244 launch_serial = ao_cmd_lex_i;
246 launch_channel = ao_cmd_lex_i;
253 int8_t r = AO_RADIO_CMAC_OK;
255 tick_offset = ao_time();
256 for (i = 0; i < 10; i++) {
257 printf ("."); flush();
258 command.tick = ao_time();
259 command.serial = launch_serial;
260 command.cmd = AO_LAUNCH_QUERY;
261 command.channel = launch_channel;
262 ao_radio_cmac_send(&command, sizeof (command));
263 r = ao_radio_cmac_recv(&query, sizeof (query), AO_MS_TO_TICKS(500));
264 if (r == AO_RADIO_CMAC_OK)
267 tick_offset -= query.tick;
268 printf("\n"); flush();
273 launch_report_cmd(void) __reentrant
278 if (ao_cmd_status != ao_cmd_success)
282 case AO_RADIO_CMAC_OK:
284 switch (query.arm_status) {
285 case ao_igniter_ready:
286 case ao_igniter_active:
290 printf("Disarmed: ");
292 switch (query.igniter_status) {
296 case ao_igniter_ready:
297 printf("igniter good\n");
299 case ao_igniter_open:
300 printf("igniter bad\n");
304 printf("Invalid channel %d\n", launch_channel);
306 printf("Rssi: %d\n", ao_radio_cmac_rssi);
309 printf("Error %d\n", r);
315 launch_arm(void) __reentrant
317 command.tick = ao_time() - tick_offset;
318 command.serial = launch_serial;
319 command.cmd = AO_LAUNCH_ARM;
320 command.channel = launch_channel;
321 ao_radio_cmac_send(&command, sizeof (command));
325 launch_ignite(void) __reentrant
327 command.tick = ao_time() - tick_offset;
328 command.serial = launch_serial;
329 command.cmd = AO_LAUNCH_FIRE;
331 ao_radio_cmac_send(&command, sizeof (command));
335 launch_fire_cmd(void) __reentrant
337 static __xdata struct ao_launch_command command;
345 if (ao_cmd_status != ao_cmd_success)
348 if (r != AO_RADIO_CMAC_OK) {
349 printf("query failed %d\n", r);
353 for (i = 0; i < 4; i++) {
354 printf("arm %d\n", i); flush();
358 secs = secs * 10 - 5;
361 for (i = 0; i < secs; i++) {
362 printf("fire %d\n", i); flush();
364 ao_delay(AO_MS_TO_TICKS(100));
369 launch_arm_cmd(void) __reentrant
375 if (r != AO_RADIO_CMAC_OK) {
376 printf("query failed %d\n", r);
379 for (i = 0; i < 4; i++)
384 launch_ignite_cmd(void) __reentrant
388 for (i = 0; i < 4; i++)
392 static __code struct ao_cmds ao_radio_cmac_cmds[] = {
393 { radio_cmac_send_cmd, "s <length>\0Send AES-CMAC packet. Bytes to send follow on next line" },
394 { radio_cmac_recv_cmd, "S <length> <timeout>\0Receive AES-CMAC packet. Timeout in ms" },
395 { launch_report_cmd, "l <serial> <channel>\0Get remote launch status" },
396 { launch_fire_cmd, "f <serial> <channel> <secs>\0Fire remote igniter" },
397 { launch_arm_cmd, "a <serial> <channel>\0Arm remote igniter" },
398 { launch_ignite_cmd, "i <serial> <channel>\0Pulse remote igniter" },
403 ao_radio_cmac_init(void)
405 ao_cmd_register(&ao_radio_cmac_cmds[0]);