350 lines
10 KiB
C
350 lines
10 KiB
C
/*
|
|
* Copyright (c) 2010, Loughborough University - Computer Science
|
|
* All rights reserved.
|
|
*
|
|
* Redistribution and use in source and binary forms, with or without
|
|
* modification, are permitted provided that the following conditions
|
|
* are met:
|
|
* 1. Redistributions of source code must retain the above copyright
|
|
* notice, this list of conditions and the following disclaimer.
|
|
* 2. Redistributions in binary form must reproduce the above copyright
|
|
* notice, this list of conditions and the following disclaimer in the
|
|
* documentation and/or other materials provided with the distribution.
|
|
* 3. Neither the name of the Institute nor the names of its contributors
|
|
* may be used to endorse or promote products derived from this software
|
|
* without specific prior written permission.
|
|
*
|
|
* THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND
|
|
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
|
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
|
* ARE DISCLAIMED. IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE
|
|
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
|
|
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
|
|
* OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
|
|
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
|
|
* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
|
|
* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
|
|
* SUCH DAMAGE.
|
|
*
|
|
* This file is part of the Contiki operating system.
|
|
*/
|
|
|
|
/**
|
|
* \file
|
|
* Disco server sources
|
|
* (embedded part of the DISCOBALL project)
|
|
*
|
|
* It objective is to receive a code file over UDP, store it in
|
|
* external flash and disseminate it to other nodes of the
|
|
* 6LoWPAN network.
|
|
*
|
|
* For this to work, the image must be co-hosted with the BooTTY!
|
|
* bootloader, which will move the image from external to internal
|
|
* flash.
|
|
*
|
|
* To link this application in your contiki image, all you need to
|
|
* do is to add this line:
|
|
* OFFSET_FIRMWARE=1
|
|
* to your project's makefile
|
|
*
|
|
* \author
|
|
* George Oikonomou - <oikonomou@users.sourceforge.net>
|
|
*/
|
|
|
|
#include "contiki.h"
|
|
#include "contiki-net.h"
|
|
#include "sys/clock.h"
|
|
#include "sys/ctimer.h"
|
|
#include "dev/watchdog.h"
|
|
|
|
#include "dev/n740.h"
|
|
#include "dev/m25p16.h"
|
|
|
|
#include "disco.h"
|
|
/*---------------------------------------------------------------------------*/
|
|
#define DEBUG DEBUG_NONE
|
|
#include "net/ip/uip-debug.h"
|
|
/*---------------------------------------------------------------------------*/
|
|
#if BATMON_CONF_ENABLED
|
|
void batmon_log(uint8_t trigger);
|
|
|
|
#define LOG_TRIGGER_OAP_DISCO_START 0x01
|
|
#define LOG_TRIGGER_OAP_DISCO_DONE 0x02
|
|
#define LOG_TRIGGER_OAP_DISCO_ABORT 0x03
|
|
#else
|
|
#define batmon_log(t) do { } while(0);
|
|
#endif
|
|
/*---------------------------------------------------------------------------*/
|
|
static struct uip_udp_conn *server_conn;
|
|
static struct disco_request_pdu *req;
|
|
static struct disco_response_pdu resp;
|
|
static struct disco_seed seed;
|
|
static uint8_t state;
|
|
static uint8_t sector;
|
|
static uint16_t interval;
|
|
static struct ctimer disco_timer;
|
|
|
|
#define UIP_IP_BUF ((struct uip_ip_hdr *)&uip_buf[UIP_LLH_LEN])
|
|
#define UIP_UDP_BUF ((struct uip_udp_hdr *)&uip_buf[uip_l2_l3_hdr_len])
|
|
|
|
extern uint16_t uip_len;
|
|
extern void *uip_appdata;
|
|
|
|
__xdata __at(BOOTTY_CMD_LOCATION) static uint8_t bd;
|
|
/*---------------------------------------------------------------------------*/
|
|
static void timer_handler(void *p);
|
|
/*---------------------------------------------------------------------------*/
|
|
static void
|
|
abort() CC_NON_BANKED
|
|
{
|
|
PRINTF("Disco: Abort @ %lu\n", clock_seconds());
|
|
n740_analog_deactivate();
|
|
m25p16_dp();
|
|
n740_analog_activate();
|
|
state = DISCO_STATE_LISTENING;
|
|
memset(&seed, 0, sizeof(seed));
|
|
ctimer_stop(&disco_timer);
|
|
batmon_log(LOG_TRIGGER_OAP_DISCO_ABORT);
|
|
}
|
|
/*---------------------------------------------------------------------------*/
|
|
static void
|
|
restart_timer(uint16_t t) CC_NON_BANKED
|
|
{
|
|
interval = t;
|
|
ctimer_stop(&disco_timer);
|
|
ctimer_set(&disco_timer, interval, timer_handler, &state);
|
|
}
|
|
/*---------------------------------------------------------------------------*/
|
|
static void
|
|
timer_handler(void *p)
|
|
{
|
|
uint8_t *s = p;
|
|
uint8_t wip;
|
|
|
|
PRINTF("Disco: @ %lu, s: %u\n", clock_seconds(), *s);
|
|
|
|
if(*s == DISCO_STATE_PREPARING) {
|
|
n740_analog_deactivate();
|
|
wip = M25P16_WIP();
|
|
n740_analog_activate();
|
|
|
|
if(wip) {
|
|
restart_timer(DISCO_TIMEOUT_PREPARE);
|
|
} else {
|
|
PRINTF("Disco: Erased %u\n", sector);
|
|
if((sector & 1) == 0) {
|
|
sector++;
|
|
PRINTF("Disco: Next %u\n", sector);
|
|
n740_analog_deactivate();
|
|
m25p16_se(sector);
|
|
n740_analog_activate();
|
|
restart_timer(DISCO_TIMEOUT_PREPARE);
|
|
} else {
|
|
PRINTF("Disco: Ready\n");
|
|
*s = DISCO_STATE_READY;
|
|
resp.status = DISCO_CMD_INIT;
|
|
restart_timer(DISCO_TIMEOUT_ABORT);
|
|
server_conn->rport = seed.port;
|
|
uip_ipaddr_copy(&server_conn->ripaddr, &seed.addr);
|
|
uip_udp_packet_send(server_conn, &resp, DISCO_RESP_LEN_INIT);
|
|
|
|
/* Restore server connection to allow data from any node */
|
|
uip_create_unspecified(&server_conn->ripaddr);
|
|
server_conn->rport = 0;
|
|
}
|
|
}
|
|
} else if(*s == DISCO_STATE_READY) {
|
|
abort();
|
|
} else if(*s == DISCO_STATE_REBOOTING) {
|
|
watchdog_reboot();
|
|
}
|
|
}
|
|
/*---------------------------------------------------------------------------*/
|
|
static uint8_t
|
|
is_protected(uint8_t a) CC_NON_BANKED
|
|
{
|
|
uint8_t bp = M25P16_BP() >> 2;
|
|
|
|
if(bp > 5) {
|
|
return SECTOR_PROTECTED;
|
|
}
|
|
|
|
bp -= 1;
|
|
|
|
if(a >= (32 - (1 << bp))) {
|
|
return SECTOR_PROTECTED;
|
|
}
|
|
return SECTOR_UNPROTECTED;
|
|
}
|
|
/*---------------------------------------------------------------------------*/
|
|
static uint8_t
|
|
cmd_init() CC_NON_BANKED
|
|
{
|
|
PRINTF("Disco: Init 0x%02x\n", req->addr[0]);
|
|
if(uip_datalen() != DISCO_LEN_INIT) {
|
|
PRINTF("Disco: Bad len (%u)\n", uip_datalen());
|
|
resp.status = DISCO_ERR_BAD_LEN;
|
|
return DISCO_RESP_LEN_ERR;
|
|
}
|
|
n740_analog_deactivate();
|
|
m25p16_res();
|
|
sector = 2 * req->addr[0];
|
|
if(is_protected(sector) == SECTOR_PROTECTED
|
|
|| is_protected(sector + 1) == SECTOR_PROTECTED) {
|
|
resp.status = DISCO_ERR_PROTECTED;
|
|
n740_analog_activate();
|
|
return DISCO_RESP_LEN_ERR;
|
|
}
|
|
m25p16_se(sector);
|
|
n740_analog_activate();
|
|
state = DISCO_STATE_PREPARING;
|
|
restart_timer(DISCO_TIMEOUT_PREPARE);
|
|
|
|
/* Store the sender's address/port so we can reply when ready */
|
|
seed.port = UIP_UDP_BUF->srcport;
|
|
uip_ipaddr_copy(&seed.addr, &UIP_IP_BUF->srcipaddr);
|
|
PRINTF("Disco: OK\n");
|
|
|
|
batmon_log(LOG_TRIGGER_OAP_DISCO_START);
|
|
|
|
return DISCO_RESPONSE_NONE;
|
|
}
|
|
/*---------------------------------------------------------------------------*/
|
|
static uint8_t
|
|
cmd_write() CC_NON_BANKED
|
|
{
|
|
PRINTF("Disco: Write 0x%02x%02x%02x\n", req->addr[0], req->addr[1],
|
|
req->addr[2]);
|
|
if(uip_datalen() != DISCO_LEN_WRITE) {
|
|
resp.status = DISCO_ERR_BAD_LEN;
|
|
return DISCO_RESP_LEN_ERR;
|
|
}
|
|
restart_timer(DISCO_TIMEOUT_ABORT);
|
|
n740_analog_deactivate();
|
|
m25p16_pp(req->addr, req->data, DISCO_FLEN_DATA);
|
|
watchdog_periodic();
|
|
while(M25P16_WIP());
|
|
n740_analog_activate();
|
|
resp.status = DISCO_CMD_WRITE;
|
|
memcpy(resp.addr, req->addr, DISCO_FLEN_ADDR);
|
|
return DISCO_RESP_LEN_WRITE;
|
|
}
|
|
/*---------------------------------------------------------------------------*/
|
|
static uint8_t
|
|
cmd_switch() CC_NON_BANKED
|
|
{
|
|
PRINTF("Disco: Switch 0x%02x\n", req->addr[0]);
|
|
if(uip_datalen() != DISCO_LEN_SWITCH) {
|
|
resp.status = DISCO_ERR_BAD_LEN;
|
|
return DISCO_RESP_LEN_ERR;
|
|
}
|
|
if(req->addr[0] > 15) {
|
|
resp.status = DISCO_ERR_BAD_OFFSET;
|
|
return DISCO_RESP_LEN_ERR;
|
|
}
|
|
|
|
bd = BOOTTY_CMD_COPY_IMAGE;
|
|
bd |= req->addr[0];
|
|
|
|
resp.status = DISCO_CMD_SWITCH;
|
|
resp.addr[0] = req->addr[0];
|
|
|
|
restart_timer(DISCO_TIMEOUT_REBOOT);
|
|
state = DISCO_STATE_REBOOTING;
|
|
|
|
return DISCO_RESP_LEN_SWITCH;
|
|
}
|
|
/*---------------------------------------------------------------------------*/
|
|
static uint8_t
|
|
cmd_done() CC_NON_BANKED
|
|
{
|
|
PRINTF("Disco: Done\n");
|
|
if(uip_datalen() != DISCO_LEN_DONE) {
|
|
resp.status = DISCO_ERR_BAD_LEN;
|
|
return DISCO_RESP_LEN_ERR;
|
|
}
|
|
resp.status = DISCO_CMD_DONE;
|
|
|
|
batmon_log(LOG_TRIGGER_OAP_DISCO_DONE);
|
|
|
|
return DISCO_RESP_LEN_DONE;
|
|
}
|
|
/*---------------------------------------------------------------------------*/
|
|
static uint8_t
|
|
event_handler(process_event_t ev) CC_NON_BANKED
|
|
{
|
|
uint8_t rv = DISCO_RESPONSE_NONE;
|
|
|
|
if(ev != tcpip_event) {
|
|
return rv;
|
|
}
|
|
|
|
/* Always accept CMD_DONE */
|
|
if(req->cmd == DISCO_CMD_DONE) {
|
|
return cmd_done();
|
|
}
|
|
|
|
/* Always accept switch too */
|
|
if(req->cmd == DISCO_CMD_SWITCH) {
|
|
return cmd_switch();
|
|
}
|
|
|
|
switch(state) {
|
|
case DISCO_STATE_LISTENING:
|
|
req = uip_appdata;
|
|
if(req->cmd == DISCO_CMD_INIT) {
|
|
rv = cmd_init();
|
|
}
|
|
break;
|
|
case DISCO_STATE_PREPARING:
|
|
PRINTF("Disco: Not Ready\n");
|
|
resp.status = DISCO_ERR_NOT_READY;
|
|
rv = DISCO_RESP_LEN_ERR;
|
|
break;
|
|
case DISCO_STATE_READY:
|
|
req = uip_appdata;
|
|
if(req->cmd == DISCO_CMD_WRITE) {
|
|
rv = cmd_write();
|
|
} else if(req->cmd == DISCO_CMD_INIT) {
|
|
resp.status = DISCO_ERR_INIT_DONE;
|
|
rv = DISCO_RESP_LEN_ERR;
|
|
} else if(req->cmd == DISCO_CMD_SWITCH) {
|
|
rv = cmd_switch();
|
|
}
|
|
break;
|
|
}
|
|
return rv;
|
|
}
|
|
/*---------------------------------------------------------------------------*/
|
|
PROCESS(disco_process, "Disco Server Process");
|
|
/*---------------------------------------------------------------------------*/
|
|
PROCESS_THREAD(disco_process, ev, data)
|
|
{
|
|
uint8_t len;
|
|
|
|
PROCESS_BEGIN();
|
|
|
|
PRINTF("Disco Server\n");
|
|
|
|
server_conn = udp_new(NULL, UIP_HTONS(0), NULL);
|
|
udp_bind(server_conn, UIP_HTONS(DISCO_UDP_PORT));
|
|
|
|
state = DISCO_STATE_LISTENING;
|
|
|
|
while(1) {
|
|
PROCESS_YIELD();
|
|
len = event_handler(ev);
|
|
|
|
if(len > 0) {
|
|
server_conn->rport = UIP_UDP_BUF->srcport;
|
|
uip_ipaddr_copy(&server_conn->ripaddr, &UIP_IP_BUF->srcipaddr);
|
|
uip_udp_packet_send(server_conn, &resp, len);
|
|
/* Restore server connection to allow data from any node */
|
|
uip_create_unspecified(&server_conn->ripaddr);
|
|
server_conn->rport = 0;
|
|
}
|
|
}
|
|
|
|
PROCESS_END();
|
|
}
|
|
/*---------------------------------------------------------------------------*/
|