mirror of
git://projects.qi-hardware.com/ben-wpan.git
synced 2024-11-05 14:25:01 +02:00
ead7ade51e
This reverts commit 39e3d4e61a.
The log isn't so useful in its present state because we may have to
unplug/replug to get out of a troublesome condition.
280 lines
6.1 KiB
C
280 lines
6.1 KiB
C
/*
|
|
* fw/ep0.c - EP0 extension protocol
|
|
*
|
|
* Written 2008-2011 by Werner Almesberger
|
|
* Copyright 2008-2011 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 <stdint.h>
|
|
#include <string.h>
|
|
|
|
#include <avr/io.h>
|
|
|
|
#ifndef NULL
|
|
#define NULL 0
|
|
#endif
|
|
|
|
#include "usb.h"
|
|
#include "dfu.h"
|
|
|
|
#include "at86rf230.h"
|
|
#include "atusb/ep0.h"
|
|
#include "version.h"
|
|
#include "board.h"
|
|
#include "sernum.h"
|
|
#include "spi.h"
|
|
|
|
|
|
#define HW_TYPE HW_TYPE_110131
|
|
|
|
#define debug(...)
|
|
#define error(...)
|
|
|
|
|
|
static const uint8_t id[] = { EP0ATUSB_MAJOR, EP0ATUSB_MINOR, HW_TYPE };
|
|
static uint8_t buf[MAX_PSDU+3]; /* command, PHDR, and LQI */
|
|
static uint8_t size;
|
|
|
|
|
|
static void do_buf_write(void *user)
|
|
{
|
|
uint8_t i;
|
|
|
|
spi_begin();
|
|
for (i = 0; i != size; i++)
|
|
spi_send(buf[i]);
|
|
spi_end();
|
|
}
|
|
|
|
|
|
#define BUILD_OFFSET 7 /* '#' plus "65535" plus ' ' */
|
|
|
|
|
|
static int my_setup(const struct setup_request *setup)
|
|
{
|
|
uint16_t req = setup->bmRequestType | setup->bRequest << 8;
|
|
unsigned tmp;
|
|
uint8_t i;
|
|
uint64_t tmp64;
|
|
|
|
switch (req) {
|
|
case ATUSB_FROM_DEV(ATUSB_ID):
|
|
debug("ATUSB_ID\n");
|
|
if (setup->wLength > 3)
|
|
return 0;
|
|
usb_send(&eps[0], id, setup->wLength, NULL, NULL);
|
|
return 1;
|
|
case ATUSB_FROM_DEV(ATUSB_BUILD):
|
|
debug("ATUSB_BUILD\n");
|
|
tmp = build_number;
|
|
for (i = BUILD_OFFSET-2; tmp; i--) {
|
|
buf[i] = (tmp % 10)+'0';
|
|
tmp /= 10;
|
|
}
|
|
buf[i] = '#';
|
|
buf[BUILD_OFFSET-1] = ' ';
|
|
for (size = 0; build_date[size]; size++)
|
|
buf[BUILD_OFFSET+size] = build_date[size];
|
|
size += BUILD_OFFSET-i;
|
|
if (size > setup->wLength)
|
|
return 0;
|
|
usb_send(&eps[0], buf+i, size, NULL, NULL);
|
|
return 1;
|
|
|
|
case ATUSB_TO_DEV(ATUSB_RESET):
|
|
debug("ATUSB_RESET\n");
|
|
reset_cpu();
|
|
while (1);
|
|
|
|
case ATUSB_TO_DEV(ATUSB_RF_RESET):
|
|
debug("ATUSB_RF_RESET\n");
|
|
reset_rf();
|
|
//ep_send_zlp(EP_CTRL);
|
|
return 1;
|
|
|
|
case ATUSB_FROM_DEV(ATUSB_POLL_INT):
|
|
debug("ATUSB_POLL_INT\n");
|
|
if (setup->wLength < 1)
|
|
return 0;
|
|
*buf = read_irq();
|
|
usb_send(&eps[0], buf, 1, NULL, NULL);
|
|
return 1;
|
|
|
|
case ATUSB_FROM_DEV(ATUSB_TIMER):
|
|
debug("ATUSB_TIMER\n");
|
|
size = setup->wLength;
|
|
if (size > sizeof(tmp64))
|
|
size = sizeof(tmp64);
|
|
tmp64 = timer_read();
|
|
memcpy(buf, &tmp64, sizeof(tmp64));
|
|
usb_send(&eps[0], buf, size, NULL, NULL);
|
|
return 1;
|
|
|
|
case ATUSB_FROM_DEV(ATUSB_GPIO):
|
|
debug("ATUSB_GPIO\n");
|
|
if (setup->wLength < 3)
|
|
return 0;
|
|
if (!gpio(setup->wIndex, setup->wValue, setup->wValue >> 8,
|
|
setup->wIndex >> 8, buf))
|
|
return 0;
|
|
usb_send(&eps[0], buf, 3, NULL, NULL);
|
|
return 1;
|
|
case ATUSB_TO_DEV(ATUSB_GPIO_CLEANUP):
|
|
gpio_cleanup();
|
|
return 1;
|
|
|
|
case ATUSB_TO_DEV(ATUSB_SLP_TR):
|
|
debug("ATUSB_SLP_TR\n");
|
|
slp_tr();
|
|
return 1;
|
|
|
|
case ATUSB_TO_DEV(ATUSB_REG_WRITE):
|
|
debug("ATUSB_REG_WRITE\n");
|
|
spi_begin();
|
|
spi_send(AT86RF230_REG_WRITE | setup->wIndex);
|
|
spi_send(setup->wValue);
|
|
spi_end();
|
|
//ep_send_zlp(EP_CTRL);
|
|
return 1;
|
|
case ATUSB_FROM_DEV(ATUSB_REG_READ):
|
|
debug("ATUSB_REG_READ\n");
|
|
spi_begin();
|
|
spi_send(AT86RF230_REG_READ | setup->wIndex);
|
|
*buf = spi_recv();
|
|
spi_end();
|
|
usb_send(&eps[0], buf, 1, NULL, NULL);
|
|
return 1;
|
|
|
|
case ATUSB_TO_DEV(ATUSB_BUF_WRITE):
|
|
debug("ATUSB_BUF_WRITE\n");
|
|
if (setup->wLength < 1)
|
|
return 0;
|
|
if (setup->wLength > MAX_PSDU)
|
|
return 0;
|
|
buf[0] = AT86RF230_BUF_WRITE;
|
|
buf[1] = setup->wLength;
|
|
size = setup->wLength+2;
|
|
usb_recv(&eps[0], buf+2, setup->wLength, do_buf_write, NULL);
|
|
return 1;
|
|
case ATUSB_FROM_DEV(ATUSB_BUF_READ):
|
|
debug("ATUSB_BUF_READ\n");
|
|
if (setup->wLength < 2) /* PHR+LQI */
|
|
return 0;
|
|
if (setup->wLength > MAX_PSDU+2) /* PHR+PSDU+LQI */
|
|
return 0;
|
|
spi_begin();
|
|
spi_send(AT86RF230_BUF_READ);
|
|
size = spi_recv();
|
|
if (size >= setup->wLength)
|
|
size = setup->wLength-1;
|
|
for (i = 0; i != size+1; i++)
|
|
buf[i] = spi_recv();
|
|
spi_end();
|
|
usb_send(&eps[0], buf, size+1, NULL, NULL);
|
|
return 1;
|
|
|
|
case ATUSB_TO_DEV(ATUSB_SRAM_WRITE):
|
|
debug("ATUSB_SRAM_WRITE\n");
|
|
if (setup->wIndex > SRAM_SIZE)
|
|
return 0;
|
|
if (setup->wIndex+setup->wLength > SRAM_SIZE)
|
|
return 0;
|
|
buf[0] = AT86RF230_SRAM_WRITE;
|
|
buf[1] = setup->wIndex;
|
|
size = setup->wLength+2;
|
|
usb_recv(&eps[0], buf+2, setup->wLength, do_buf_write, NULL);
|
|
return 1;
|
|
case ATUSB_FROM_DEV(ATUSB_SRAM_READ):
|
|
debug("ATUSB_SRAM_READ\n");
|
|
if (setup->wIndex > SRAM_SIZE)
|
|
return 0;
|
|
if (setup->wIndex+setup->wLength > SRAM_SIZE)
|
|
return 0;
|
|
spi_begin();
|
|
spi_send(AT86RF230_SRAM_READ);
|
|
spi_send(setup->wIndex);
|
|
for (i = 0; i != setup->wLength; i++)
|
|
buf[i] = spi_recv();
|
|
spi_end();
|
|
usb_send(&eps[0], buf, setup->wLength, NULL, NULL);
|
|
return 1;
|
|
|
|
case ATUSB_TO_DEV(ATUSB_SPI_WRITE):
|
|
size = setup->wLength+2;
|
|
if (size > sizeof(buf))
|
|
return 0;
|
|
buf[0] = setup->wValue;
|
|
buf[1] = setup->wIndex;
|
|
if (setup->wLength)
|
|
usb_recv(&eps[0], buf+2, setup->wLength,
|
|
do_buf_write, NULL);
|
|
else
|
|
do_buf_write(NULL);
|
|
return 1;
|
|
case ATUSB_FROM_DEV(ATUSB_SPI_READ1):
|
|
case ATUSB_FROM_DEV(ATUSB_SPI_READ2):
|
|
spi_begin();
|
|
spi_send(setup->wValue);
|
|
if (req == ATUSB_FROM_DEV(ATUSB_SPI_READ2))
|
|
spi_send(setup->wIndex);
|
|
for (i = 0; i != setup->wLength; i++)
|
|
buf[i] = spi_recv();
|
|
spi_end();
|
|
usb_send(&eps[0], buf, setup->wLength, NULL, NULL);
|
|
return 1;
|
|
|
|
default:
|
|
error("Unrecognized SETUP: 0x%02x 0x%02x ...\n",
|
|
setup->bmRequestType, setup->bRequest);
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
|
|
static int my_dfu_setup(const struct setup_request *setup)
|
|
{
|
|
switch (setup->bmRequestType | setup->bRequest << 8) {
|
|
case DFU_TO_DEV(DFU_DETACH):
|
|
/* @@@ should use wTimeout */
|
|
dfu.state = appDETACH;
|
|
return 1;
|
|
default:
|
|
return dfu_setup_common(setup);
|
|
}
|
|
}
|
|
|
|
|
|
static void my_set_interface(int nth)
|
|
{
|
|
if (nth) {
|
|
user_setup = my_dfu_setup;
|
|
user_get_descriptor = dfu_my_descr;
|
|
dfu.state = appIDLE;
|
|
} else {
|
|
user_setup = my_setup;
|
|
user_get_descriptor = sernum_get_descr;
|
|
}
|
|
}
|
|
|
|
|
|
static void my_reset(void)
|
|
{
|
|
if (dfu.state == appDETACH)
|
|
reset_cpu();
|
|
}
|
|
|
|
|
|
void ep0_init(void)
|
|
{
|
|
user_setup = my_setup;
|
|
user_set_interface = my_set_interface;
|
|
my_set_interface(0);
|
|
user_reset = my_reset;
|
|
}
|