1
0
mirror of git://projects.qi-hardware.com/antorcha.git synced 2024-11-22 13:44:59 +02:00

fw/: addition of boot loader (WIP) and assorted cleanup and improvements

The boot loader currently uses the protocol switch intended for the
application. This makes it too big to fit in the very limiting
constraints the ATmega168 poses - and there's not even cryptographic
authentication yet. We'll have to dumb it down quite a bit.
This commit is contained in:
Werner Almesberger 2012-06-18 10:56:43 -03:00
parent df6ceb8394
commit 56476539cf
16 changed files with 651 additions and 52 deletions

4
NOTES
View File

@ -90,6 +90,6 @@ electrical, LEDs:
even less demanding even less demanding
authentication: authentication:
- shared secret (128 bit, MD5-hashed text with 128 bit salt) - shared secret (128 bit, SHA1-hashed text with 128 bit salt)
- salt (128 bit) - salt (128 bit)
- MD5 - SHA1 from avrcryptolib

View File

@ -1,8 +1,8 @@
# #
# Makefile - Makefile of the ATUSB firmware # Makefile - Makefile of the Antorcha firmware
# #
# Written 2010-2011 by Werner Almesberger # Written 2012 by Werner Almesberger
# Copyright 2010-2011 by Werner Almesberger # Copyright 2012 by Werner Almesberger
# #
# This program is free software; you can redistribute it and/or modify # 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 # it under the terms of the GNU General Public License as published by
@ -14,13 +14,16 @@ SHELL = /bin/bash
NAME = antorcha NAME = antorcha
CFLAGS = -g -mmcu=$(CHIP) -DBOOT_ADDR=$(BOOT_ADDR) \ CFLAGS = -g -mmcu=$(CHIP) \
-DBOOT_ADDR=$(BOOT_ADDR) -DAPP_ADDR=$(APP_ADDR) -DAPP_END=$(APP_END) \
-Wall -Wextra -Wshadow -Werror -Wno-unused-parameter \ -Wall -Wextra -Wshadow -Werror -Wno-unused-parameter \
-Wmissing-prototypes -Wmissing-declarations -Wstrict-prototypes -Wmissing-prototypes -Wmissing-declarations -Wstrict-prototypes
CHIP=atmega168 CHIP = atmega168
HOST=jlime HOST = jlime
BOOT_ADDR=0x7000 BOOT_ADDR = 0
APP_ADDR = 0x1000
APP_END = 0x4000
AVR_PREFIX = $(BIN_PATH) avr- AVR_PREFIX = $(BIN_PATH) avr-
CC = $(AVR_PREFIX)gcc CC = $(AVR_PREFIX)gcc
@ -28,11 +31,9 @@ OBJCOPY = $(AVR_PREFIX)objcopy
#OBJDUMP = $(AVR_PREFIX)objdump #OBJDUMP = $(AVR_PREFIX)objdump
SIZE = $(AVR_PREFIX)size SIZE = $(AVR_PREFIX)size
USB_ID = 20b7:1540 OBJS = $(NAME).o $(COMMON_OBJS)
BOOT_OBJS = boot.o flash.o fw.o $(COMMON_OBJS)
OBJS = $(NAME).o rf.o spi.o COMMON_OBJS = dispatch.o hash.o rf.o spi.o
CFLAGS += -Iinclude
# ----- Verbosity control ----------------------------------------------------- # ----- Verbosity control -----------------------------------------------------
@ -56,18 +57,19 @@ endif
# ----- Rules ----------------------------------------------------------------- # ----- Rules -----------------------------------------------------------------
.PHONY: all clean upload prog update version.c bindist .PHONY: all clean upload prog update version.c
.PHONY: prog-app prog-read on off reset .PHONY: prog-app prog-read on off reset
all: $(NAME).bin all: $(NAME).bin boot.bin
$(NAME).elf: $(OBJS) $(NAME).elf: $(OBJS)
$(MAKE) version.o $(MAKE) version.o
$(CC) $(CFLAGS) -o $@ $(OBJS) version.o $(CC) $(CFLAGS) -o $@ $(OBJS) version.o \
-Wl,--section-start=.text=$(APP_ADDR)
#boot.elf: $(BOOT_OBJS) boot.elf: $(BOOT_OBJS)
# $(CC) $(CFLAGS) -o $@ $(BOOT_OBJS) \ $(CC) $(CFLAGS) -o $@ $(BOOT_OBJS) \
# -Wl,--section-start=.text=$(BOOT_ADDR) -Wl,--section-start=.text=$(BOOT_ADDR)
%.bin: %.elf %.bin: %.elf
$(BUILD) $(OBJCOPY) -j .text -j .data -O binary $< $@ $(BUILD) $(OBJCOPY) -j .text -j .data -O binary $< $@
@ -120,25 +122,10 @@ MKDEP = \
-include $(OBJS:.o=.d) -include $(OBJS:.o=.d)
# ----- Distribution ----------------------------------------------------------
BINDIST_BASE=http://downloads.qi-hardware.com/people/werner/wpan/bindist
ATUSB_BIN_NAME=atusb-`git rev-parse HEAD | cut -c 1-7`.bin
bindist:
qippl atusb.bin wpan/bindist/$(ATUSB_BIN_NAME)
@echo $(BINDIST_BASE)/$(ATUSB_BIN_NAME)
@echo md5sum: `md5sum atusb.bin | sed 's/ .*//'`
@echo atrf-id: \
`sed '/.*number = \(.*\);/s//#\1/p;d' version.c` \
`sed '/.*date = "\(.*\)";/s//\1/p;d' version.c`
# ----- Programming and device control ---------------------------------------- # ----- Programming and device control ----------------------------------------
#upload: $(NAME).bin boot.hex upload: $(NAME).bin boot.hex
# scp $(NAME).bin boot.hex $(HOST): scp $(NAME).bin boot.hex $(HOST):
upload: $(NAME).bin
scp $(NAME).bin $(HOST):
# lfuse: external clock, slow start-up # lfuse: external clock, slow start-up
# hfuse: 4 kB boot loader, reset into boot loader # hfuse: 4 kB boot loader, reset into boot loader
@ -146,20 +133,20 @@ upload: $(NAME).bin
# Note: when trying to program 0xef, we get back 0x2f, failing # Note: when trying to program 0xef, we get back 0x2f, failing
# verification. So we just program 0x2f. # verification. So we just program 0x2f.
prog: prog-app:
ssh $(HOST) avrdude -F -p $(CHIP) -c nanonote_antorcha -e \ ssh $(HOST) avrdude -F -p $(CHIP) -c nanonote_antorcha -e \
-U flash:w:antorcha.bin:r -U flash:w:antorcha.bin:r
# -U lfuse:w:0x60:m # -U lfuse:w:0x60:m
#prog: prog:
# ssh $(HOST) avrdude -F -p $(CHIP) -c nanonote_atusb -e \ ssh $(HOST) avrdude -F -p $(CHIP) -c nanonote_antorcha -e \
# -U flash:w:boot.hex:i \ -U flash:w:boot.hex:i
# -U lfuse:w:0x60:m \ # -U lfuse:w:0x60:m \
# -U hfuse:w:0xd8:m \ # -U hfuse:w:0xd8:m \
# -U lock:w:0x2f:m # -U lock:w:0x2f:m
prog-read: prog-read:
ssh $(HOST) avrdude -F -p $(CHIP) -c nanonote_atusb \ ssh $(HOST) avrdude -F -p $(CHIP) -c nanonote_antorcha \
-U flash:r:mcu.bin:r -U flash:r:mcu.bin:r
on: on:

81
fw/boot.c Normal file
View File

@ -0,0 +1,81 @@
/*
* fw/boot.c - Antorcha boot loader
*
* Written 2012 by Werner Almesberger
* Copyright 2012 Werner Almesberger
*
* 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.
*/
#include <avr/io.h>
#include <avr/pgmspace.h>
#include "io.h"
#include "rf.h"
#include "proto.h"
#include "dispatch.h"
#include "fw.h"
#define MS_TO_LOOP(ms) ((uint32_t) (ms)*335)
static void wait_upload(void)
{
uint8_t buf[PAYLOAD+5];
uint32_t i;
uint8_t got;
restart:
for (i = 0; i != MS_TO_LOOP(2000); i++) {
got = rf_recv(buf, sizeof(buf));
if (got && dispatch(buf, got, fw_protos))
goto restart;
}
}
int main(void)
{
volatile int zero = 0;
/* Port B has two LEDs and the rest is SPI */
PORTB = MASK(B, RF_nSS) | MASK(B, RF_nRST);
DDRB = MASK(B, RF_SCLK) | MASK(B, RF_MOSI) | MASK(B, RF_nSS) |
MASK(B, RF_nRST) | 0xc0;
/* All port C pins drive LEDs */
PORTC = 0;
DDRC = 0x3f;
/* All port D pins drive LEDs */
PORTD = 0;
DDRD = 0xff;
/* disable pull-ups */
MCUCR |= 1 << PUD;
rf_init();
/*
* Switch the LED inside the loop so that we get a short pulse one can
* observe on a scope.
*/
do {
SET(LED_B8);
wait_upload();
CLR(LED_B8);
while (1);
} while (pgm_read_byte(zero) != 0xff);
((void (*)(void)) 0)();
/* not reached */
return 0;
}

104
fw/dispatch.c Normal file
View File

@ -0,0 +1,104 @@
/*
* fw/dispatch.c - Wireless protocol dispatcher
*
* Written 2012 by Werner Almesberger
* Copyright 2012 Werner Almesberger
*
* 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.
*/
#include <stddef.h>
#include <stdbool.h>
#include <stdint.h>
#define F_CPU 8000000UL
#include <util/delay.h>
#include "rf.h"
#include "proto.h"
#include "dispatch.h"
#include "io.h"
static uint8_t seq; /* last sequence number seen */
static uint8_t type; /* type currently being processed */
static uint8_t limit; /* last limit seen */
static const struct handler *curr_proto = NULL;
static void send_ack(const uint8_t *buf)
{
uint8_t ack[3] = { buf[0]+1, buf[1], 0 };
SET(LED_B6);
_delay_ms(1);
rf_send(ack, sizeof(ack));
CLR(LED_B6);
}
static bool answer_ping(const uint8_t *buf)
{
uint8_t pong[] = { PONG, 0, 0 };
if (buf[1])
return 0;
rf_send(pong, sizeof(pong));
return 1;
}
bool dispatch(const uint8_t *buf, uint8_t len, const struct handler **protos)
{
SET(LED_B7);
CLR(LED_B7);
if (len == 3 && buf[0] == PING)
return answer_ping(buf);
if (len != 64+3)
return 0;
if (!buf[1]) {
while (*protos) {
if ((*protos)->type == buf[0])
break;
protos++;
}
if (!*protos)
return 0;
if (!(*protos)->first(buf+3))
return 0;
curr_proto = *protos;
type = buf[0];
seq = 0;
limit = buf[2];
send_ack(buf);
return 1;
}
if (!curr_proto)
return 0;
if (buf[0] != type)
return 0;
if (buf[1] > limit)
return 0;
if (buf[2] != limit)
return 0;
if (buf[1] == seq) {
send_ack(buf);
return 0;
}
if (buf[1] != seq+1)
return 0;
if (!curr_proto->more(buf[1], limit, buf+3))
return 0;
seq++;
send_ack(buf);
return 1;
}

29
fw/dispatch.h Normal file
View File

@ -0,0 +1,29 @@
/*
* fw/dispatch.h - Wireless protocol dispatcher
*
* Written 2012 by Werner Almesberger
* Copyright 2012 Werner Almesberger
*
* 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.
*/
#ifndef DISPATCH_H
#define DISPATCH_H
#include <stdbool.h>
#include <stdint.h>
struct handler {
enum pck_type type;
bool (*first)(const uint8_t *payload);
bool (*more)(uint8_t seq, uint8_t limit, const uint8_t *payload);
};
bool dispatch(const uint8_t *buf, uint8_t len, const struct handler **protos);
#endif /* !PROTO_H */

73
fw/flash.c Normal file
View File

@ -0,0 +1,73 @@
/*
* fw/flash.c - Flash interface
*
* Written 2011, 2012 by Werner Almesberger
* Copyright 2011, 2012 Werner Almesberger
*
* 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.
*/
/*
* Adapted from ben-wpan/atusb/fw/flash.c
*/
#include <stdint.h>
#include <avr/boot.h>
//#include <avr/pgmspace.h>
#include "flash.h"
static uint32_t payload;
void flash_start(uint32_t addr)
{
payload = addr;
}
int flash_can_write(uint16_t size)
{
return payload <= APP_END-size;
}
void flash_write(const uint8_t *buf, uint16_t size)
{
static uint8_t last;
const uint8_t *p;
for (p = buf; p != buf+size; p++) {
if (!(payload & (SPM_PAGESIZE-1))) {
boot_page_erase(payload);
boot_spm_busy_wait();
}
if (payload & 1)
boot_page_fill(payload, last | (*p << 8));
else
last = *p;
payload++;
if (!(payload & (SPM_PAGESIZE-1))) {
boot_page_write(payload-SPM_PAGESIZE);
boot_spm_busy_wait();
}
}
}
void flash_end_write(void)
{
if (payload & (SPM_PAGESIZE-1)) {
boot_page_write(payload & ~(SPM_PAGESIZE-1));
boot_spm_busy_wait();
}
boot_rww_enable();
}

24
fw/flash.h Normal file
View File

@ -0,0 +1,24 @@
/*
* fw/flash.h - Flash interface
*
* Written 2012 by Werner Almesberger
* Copyright 2012 Werner Almesberger
*
* 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.
*/
#ifndef FLASH_H
#define FLASH_H
#include <stdint.h>
void flash_start(uint32_t addr);
int flash_can_write(uint16_t size);
void flash_write(const uint8_t *buf, uint16_t size);
void flash_end_write(void);
#endif /* !FLASH_H */

133
fw/fw.c Normal file
View File

@ -0,0 +1,133 @@
/*
* fw/fw.h - Firmware upload protocols
*
* Written 2012 by Werner Almesberger
* Copyright 2012 Werner Almesberger
*
* 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.
*/
#include <stddef.h>
#include <stdbool.h>
#include <stdint.h>
#include "hash.h"
#include "flash.h"
#include "proto.h"
#include "dispatch.h"
#include "fw.h"
static const uint8_t unlock_secret[] = {
#include "unlock-secret.inc"
};
static void panic(void)
{
/* ??? */
}
/* ----- Unlocking --------------------------------------------------------- */
static bool unlocked = 0;
static bool unlock_failed;
static bool unlock_first(const uint8_t *payload)
{
hash_init();
hash_merge(unlock_secret, sizeof(unlock_secret));
hash_merge(payload, PAYLOAD);
unlocked = 0;
unlock_failed = 0;
return 1;
}
static bool unlock_more(uint8_t seq, uint8_t limit, const uint8_t *payload)
{
switch (seq) {
case 1:
hash_merge(payload, PAYLOAD);
hash_end();
break;
case 2:
if (!hash_eq(payload, PAYLOAD, 0))
unlock_failed = 1;
break;
case 3:
if (unlock_failed)
return 1;
if (hash_eq(payload, PAYLOAD, PAYLOAD))
unlocked = 1;
else
unlock_failed = 1;
break;
default:
return 0;
}
return 1;
}
static const struct handler unlock_proto = {
.type = UNLOCK,
.first = unlock_first,
.more = unlock_more,
};
/* ----- Firmware upload --------------------------------------------------- */
static bool fw_first(const uint8_t *payload)
{
// if (!unlocked)
// return 0;
hash_init();
hash_merge(payload, PAYLOAD);
flash_start(APP_ADDR);
flash_write(payload, PAYLOAD);
return 1;
}
static bool fw_more(uint8_t seq, uint8_t limit, const uint8_t *payload)
{
if (!flash_can_write(PAYLOAD))
return 0;
if (seq != limit) {
hash_merge(payload, PAYLOAD);
flash_write(payload, PAYLOAD);
return 1;
}
flash_end_write();
hash_end();
if (!hash_eq(payload, PAYLOAD, 0))
panic();
return 1;
}
static const struct handler fw_proto = {
.type = FIRMWARE,
.first = fw_first,
.more = fw_more,
};
/* ----- Protocol table ---------------------------------------------------- */
const struct handler *fw_protos[] = {
&unlock_proto,
&fw_proto,
NULL
};

21
fw/fw.h Normal file
View File

@ -0,0 +1,21 @@
/*
* fw/fw.h - Firmware upload protocols
*
* Written 2012 by Werner Almesberger
* Copyright 2012 Werner Almesberger
*
* 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.
*/
#ifndef FW_H
#define FW_H
#include "proto.h"
extern const struct handler *fw_protos[];
#endif /* !FW_H */

61
fw/hash.c Normal file
View File

@ -0,0 +1,61 @@
/*
* fw/hash.h - Secure hash
*
* Written 2012 by Werner Almesberger
* Copyright 2012 Werner Almesberger
*
* 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.
*/
#include <stdbool.h>
#include <stdint.h>
#include <string.h>
#include "hash.h"
#define HASH_SIZE 128
static uint8_t hash[HASH_SIZE];
void hash_init(void)
{
memset(hash, 0, HASH_SIZE);
}
void hash_merge(const uint8_t *buf, uint8_t len)
{
uint8_t i;
for (i = 0; i != len; i++)
hash[i & (HASH_SIZE-1)] ^= buf[i];
}
void hash_end(void)
{
}
bool hash_eq(const uint8_t *buf, uint8_t len, uint8_t off)
{
uint8_t i;
for (i = 0; i != len; i++)
if (hash[(off+i) & (HASH_SIZE-1)] != buf[i])
return 0;
return 1;
}
void hash_cp(uint8_t *buf, uint8_t len, uint8_t off)
{
memcpy(buf, hash+off, len);
}

26
fw/hash.h Normal file
View File

@ -0,0 +1,26 @@
/*
* fw/hash.h - Secure hash
*
* Written 2012 by Werner Almesberger
* Copyright 2012 Werner Almesberger
*
* 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.
*/
#ifndef HASH_H
#define HASH_H
#include <stdbool.h>
#include <stdint.h>
void hash_init(void);
void hash_merge(const uint8_t *buf, uint8_t len);
void hash_end(void);
bool hash_eq(const uint8_t *buf, uint8_t len, uint8_t off);
void hash_cp(uint8_t *buf, uint8_t len, uint8_t off);
#endif /* !HASH_H */

View File

@ -13,6 +13,11 @@
#ifndef IO_H #ifndef IO_H
#define IO_H #define IO_H
#include <avr/io.h>
#define LED_B6 D, 5
#define LED_B7 D, 6
#define LED_B8 D, 7 #define LED_B8 D, 7
#define RF_SCLK B, 5 #define RF_SCLK B, 5

31
fw/proto.h Normal file
View File

@ -0,0 +1,31 @@
/*
* fw/proto.h - Wireless protocol constants
*
* Written 2012 by Werner Almesberger
* Copyright 2012 Werner Almesberger
*
* 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.
*/
#ifndef PROTO_H
#define PROTO_H
#define PAYLOAD 64 /* most messages use a fixed 64 bytes payload */
enum pck_type {
PING = 0, /* version query */
PONG = 1, /* version response */
UNLOCK = 2, /* unlock firmware upload */
UNLOCK_ACK = 3, /* unlock acknowledgement */
FIRMWARE = 4, /* firmware upload */
FIRMWARE_ACK = 5, /* firmware upload acknowledgement */
IMAGE = 6, /* image upload */
IMAGE_ACK = 7, /* image upload acknowledgement */
PARAM = 8, /* parameter upload */
PARAM_ACK = 9, /* parameter upload acknowledgement */
};
#endif /* !PROTO_H */

21
fw/rf.c
View File

@ -44,8 +44,8 @@ static uint8_t reg_read(uint8_t reg)
uint8_t value; uint8_t value;
spi_begin(); spi_begin();
spi_io(AT86RF230_REG_READ | reg); spi_send(AT86RF230_REG_READ | reg);
value = spi_io(0); value = spi_recv();
spi_end(); spi_end();
return value; return value;
@ -55,11 +55,12 @@ static uint8_t reg_read(uint8_t reg)
static void reg_write(uint8_t reg, uint8_t value) static void reg_write(uint8_t reg, uint8_t value)
{ {
spi_begin(); spi_begin();
spi_io(AT86RF230_REG_WRITE | reg); spi_send(AT86RF230_REG_WRITE | reg);
spi_io(value); spi_send(value);
spi_end(); spi_end();
} }
void rf_init(void) void rf_init(void)
{ {
spi_init(); spi_init();
@ -89,10 +90,10 @@ void rf_send(const void *buf, uint8_t size)
_delay_us(1); /* tTR9 = 1 us */ _delay_us(1); /* tTR9 = 1 us */
spi_begin(); spi_begin();
spi_io(AT86RF230_BUF_WRITE); spi_send(AT86RF230_BUF_WRITE);
spi_io(size+2); /* CRC */ spi_send(size+2); /* CRC */
for (i = 0; i != size; i++) for (i = 0; i != size; i++)
spi_io(((const uint8_t *) buf)[i]); spi_send(((const uint8_t *) buf)[i]);
spi_end(); spi_end();
reg_read(REG_IRQ_STATUS); reg_read(REG_IRQ_STATUS);
@ -122,8 +123,8 @@ uint8_t rf_recv(void *buf, uint8_t size)
return 0; return 0;
spi_begin(); spi_begin();
spi_io(AT86RF230_BUF_READ); spi_send(AT86RF230_BUF_READ);
len = spi_io(0); len = spi_recv();
if (!len || (len & 0x80)) { if (!len || (len & 0x80)) {
spi_end(); spi_end();
return 0; return 0;
@ -131,7 +132,7 @@ uint8_t rf_recv(void *buf, uint8_t size)
if (size > len) if (size > len)
size = len; size = len;
for (i = 0; i != size; i++) for (i = 0; i != size; i++)
((uint8_t *) buf)[i] = spi_io(0); ((uint8_t *) buf)[i] = spi_recv();
spi_end(); spi_end();
return len; return len;
} }

View File

@ -21,4 +21,26 @@ void spi_begin(void);
uint8_t spi_io(uint8_t v); uint8_t spi_io(uint8_t v);
void spi_end(void); void spi_end(void);
/*
* Interestingly, the #defines below produce slightly shorter code (6 bytes)
* than the inlines.
*/
#if 0
static inline void spi_send(uint8_t v)
{
spi_io(v);
}
static inline uint8_t spi_recv(void)
{
return spi_io(0);
}
#endif
#define spi_send(v) spi_io(v)
#define spi_recv() spi_io(0)
#endif /* !SPI_H */ #endif /* !SPI_H */

1
fw/unlock-secret.inc Normal file
View File

@ -0,0 +1 @@
1, 2, 3