2010-12-05 22:59:12 +00:00
|
|
|
/*
|
2010-12-30 12:19:25 +00:00
|
|
|
* This file is part of the libopencm3 project.
|
2010-12-05 22:59:12 +00:00
|
|
|
*
|
2010-12-29 15:43:26 +00:00
|
|
|
* Copyright (C) 2010 Gareth McMullin <gareth@blacksphere.co.nz>
|
2010-12-05 22:59:12 +00:00
|
|
|
*
|
2012-03-02 10:23:11 +00:00
|
|
|
* This library is free software: you can redistribute it and/or modify
|
|
|
|
* it under the terms of the GNU Lesser General Public License as published by
|
2010-12-05 22:59:12 +00:00
|
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
|
|
* (at your option) any later version.
|
|
|
|
*
|
2012-03-02 10:23:11 +00:00
|
|
|
* This library is distributed in the hope that it will be useful,
|
2010-12-05 22:59:12 +00:00
|
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
2012-03-02 10:23:11 +00:00
|
|
|
* GNU Lesser General Public License for more details.
|
2010-12-05 22:59:12 +00:00
|
|
|
*
|
2012-03-02 10:23:11 +00:00
|
|
|
* You should have received a copy of the GNU Lesser General Public License
|
|
|
|
* along with this library. If not, see <http://www.gnu.org/licenses/>.
|
2010-12-05 22:59:12 +00:00
|
|
|
*/
|
|
|
|
|
2010-12-29 16:02:36 +00:00
|
|
|
#include <string.h>
|
2011-10-13 03:58:27 +00:00
|
|
|
#include <libopencm3/stm32/f1/rcc.h>
|
|
|
|
#include <libopencm3/stm32/f1/gpio.h>
|
|
|
|
#include <libopencm3/stm32/f1/flash.h>
|
|
|
|
#include <libopencm3/stm32/f1/scb.h>
|
2010-12-31 17:18:39 +00:00
|
|
|
#include <libopencm3/usb/usbd.h>
|
|
|
|
#include <libopencm3/usb/dfu.h>
|
2010-12-05 22:59:12 +00:00
|
|
|
|
|
|
|
#define APP_ADDRESS 0x08002000
|
|
|
|
|
|
|
|
/* Commands sent with wBlockNum == 0 as per ST implementation. */
|
2011-11-02 19:48:17 +00:00
|
|
|
#define CMD_SETADDR 0x21
|
|
|
|
#define CMD_ERASE 0x41
|
2010-12-05 22:59:12 +00:00
|
|
|
|
|
|
|
/* We need a special large control buffer for this device: */
|
|
|
|
u8 usbd_control_buffer[1024];
|
|
|
|
|
|
|
|
static enum dfu_state usbdfu_state = STATE_DFU_IDLE;
|
|
|
|
|
|
|
|
static struct {
|
|
|
|
u8 buf[sizeof(usbd_control_buffer)];
|
|
|
|
u16 len;
|
|
|
|
u32 addr;
|
|
|
|
u16 blocknum;
|
|
|
|
} prog;
|
|
|
|
|
|
|
|
const struct usb_device_descriptor dev = {
|
2011-11-02 19:48:17 +00:00
|
|
|
.bLength = USB_DT_DEVICE_SIZE,
|
|
|
|
.bDescriptorType = USB_DT_DEVICE,
|
|
|
|
.bcdUSB = 0x0200,
|
|
|
|
.bDeviceClass = 0,
|
|
|
|
.bDeviceSubClass = 0,
|
|
|
|
.bDeviceProtocol = 0,
|
|
|
|
.bMaxPacketSize0 = 64,
|
|
|
|
.idVendor = 0x0483,
|
|
|
|
.idProduct = 0xDF11,
|
|
|
|
.bcdDevice = 0x0200,
|
|
|
|
.iManufacturer = 1,
|
|
|
|
.iProduct = 2,
|
|
|
|
.iSerialNumber = 3,
|
|
|
|
.bNumConfigurations = 1,
|
2010-12-05 22:59:12 +00:00
|
|
|
};
|
|
|
|
|
|
|
|
const struct usb_dfu_descriptor dfu_function = {
|
|
|
|
.bLength = sizeof(struct usb_dfu_descriptor),
|
|
|
|
.bDescriptorType = DFU_FUNCTIONAL,
|
|
|
|
.bmAttributes = USB_DFU_CAN_DOWNLOAD | USB_DFU_WILL_DETACH,
|
|
|
|
.wDetachTimeout = 255,
|
|
|
|
.wTransferSize = 1024,
|
|
|
|
.bcdDFUVersion = 0x011A,
|
|
|
|
};
|
|
|
|
|
|
|
|
const struct usb_interface_descriptor iface = {
|
|
|
|
.bLength = USB_DT_INTERFACE_SIZE,
|
|
|
|
.bDescriptorType = USB_DT_INTERFACE,
|
|
|
|
.bInterfaceNumber = 0,
|
|
|
|
.bAlternateSetting = 0,
|
|
|
|
.bNumEndpoints = 0,
|
|
|
|
.bInterfaceClass = 0xFE, /* Device Firmware Upgrade */
|
|
|
|
.bInterfaceSubClass = 1,
|
|
|
|
.bInterfaceProtocol = 2,
|
|
|
|
|
2011-11-02 19:48:17 +00:00
|
|
|
/* The ST Microelectronics DfuSe application needs this string.
|
2010-12-05 22:59:12 +00:00
|
|
|
* The format isn't documented... */
|
|
|
|
.iInterface = 4,
|
|
|
|
|
|
|
|
.extra = &dfu_function,
|
|
|
|
.extralen = sizeof(dfu_function),
|
|
|
|
};
|
|
|
|
|
|
|
|
const struct usb_interface ifaces[] = {{
|
|
|
|
.num_altsetting = 1,
|
|
|
|
.altsetting = &iface,
|
|
|
|
}};
|
|
|
|
|
|
|
|
const struct usb_config_descriptor config = {
|
|
|
|
.bLength = USB_DT_CONFIGURATION_SIZE,
|
|
|
|
.bDescriptorType = USB_DT_CONFIGURATION,
|
|
|
|
.wTotalLength = 0,
|
|
|
|
.bNumInterfaces = 1,
|
|
|
|
.bConfigurationValue = 1,
|
|
|
|
.iConfiguration = 0,
|
|
|
|
.bmAttributes = 0xC0,
|
|
|
|
.bMaxPower = 0x32,
|
|
|
|
|
|
|
|
.interface = ifaces,
|
|
|
|
};
|
|
|
|
|
|
|
|
static const char *usb_strings[] = {
|
|
|
|
"x",
|
|
|
|
"Black Sphere Technologies",
|
|
|
|
"DFU Demo",
|
|
|
|
"DEMO",
|
2011-11-02 19:48:17 +00:00
|
|
|
/* This string is used by ST Microelectronics' DfuSe utility. */
|
2011-11-12 09:27:34 +00:00
|
|
|
"@Internal Flash /0x08000000/8*001Ka,56*001Kg",
|
2010-12-05 22:59:12 +00:00
|
|
|
};
|
|
|
|
|
|
|
|
static u8 usbdfu_getstatus(u32 *bwPollTimeout)
|
|
|
|
{
|
2011-11-02 19:48:17 +00:00
|
|
|
switch (usbdfu_state) {
|
2010-12-05 22:59:12 +00:00
|
|
|
case STATE_DFU_DNLOAD_SYNC:
|
2011-11-02 19:48:17 +00:00
|
|
|
usbdfu_state = STATE_DFU_DNBUSY;
|
2010-12-05 22:59:12 +00:00
|
|
|
*bwPollTimeout = 100;
|
|
|
|
return DFU_STATUS_OK;
|
|
|
|
case STATE_DFU_MANIFEST_SYNC:
|
2011-11-02 19:48:17 +00:00
|
|
|
/* Device will reset when read is complete. */
|
2010-12-05 22:59:12 +00:00
|
|
|
usbdfu_state = STATE_DFU_MANIFEST;
|
|
|
|
return DFU_STATUS_OK;
|
|
|
|
default:
|
|
|
|
return DFU_STATUS_OK;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
static void usbdfu_getstatus_complete(struct usb_setup_data *req)
|
|
|
|
{
|
|
|
|
int i;
|
|
|
|
(void)req;
|
|
|
|
|
2011-11-02 19:48:17 +00:00
|
|
|
switch (usbdfu_state) {
|
2010-12-05 22:59:12 +00:00
|
|
|
case STATE_DFU_DNBUSY:
|
|
|
|
flash_unlock();
|
2011-11-02 19:48:17 +00:00
|
|
|
if (prog.blocknum == 0) {
|
|
|
|
switch (prog.buf[0]) {
|
2010-12-05 22:59:12 +00:00
|
|
|
case CMD_ERASE:
|
2011-11-02 19:48:17 +00:00
|
|
|
flash_erase_page(*(u32 *)(prog.buf + 1));
|
2010-12-05 22:59:12 +00:00
|
|
|
case CMD_SETADDR:
|
2011-11-02 19:48:17 +00:00
|
|
|
prog.addr = *(u32 *)(prog.buf + 1);
|
2010-12-05 22:59:12 +00:00
|
|
|
}
|
|
|
|
} else {
|
2011-11-02 19:48:17 +00:00
|
|
|
u32 baseaddr = prog.addr + ((prog.blocknum - 2) *
|
|
|
|
dfu_function.wTransferSize);
|
|
|
|
for (i = 0; i < prog.len; i += 2)
|
|
|
|
flash_program_half_word(baseaddr + i,
|
|
|
|
*(u16 *)(prog.buf + i));
|
2010-12-05 22:59:12 +00:00
|
|
|
}
|
|
|
|
flash_lock();
|
|
|
|
|
2011-11-02 19:48:17 +00:00
|
|
|
/* Jump straight to dfuDNLOAD-IDLE, skipping dfuDNLOAD-SYNC. */
|
2010-12-05 22:59:12 +00:00
|
|
|
usbdfu_state = STATE_DFU_DNLOAD_IDLE;
|
|
|
|
return;
|
|
|
|
case STATE_DFU_MANIFEST:
|
|
|
|
/* USB device must detach, we just reset... */
|
|
|
|
scb_reset_system();
|
2011-11-02 19:48:17 +00:00
|
|
|
return; /* Will never return. */
|
2010-12-05 22:59:12 +00:00
|
|
|
default:
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2011-11-02 19:48:17 +00:00
|
|
|
static int usbdfu_control_request(struct usb_setup_data *req, u8 **buf,
|
2011-03-13 01:50:53 +00:00
|
|
|
u16 *len, void (**complete)(struct usb_setup_data *req))
|
2010-12-05 22:59:12 +00:00
|
|
|
{
|
2011-11-02 19:48:17 +00:00
|
|
|
if ((req->bmRequestType & 0x7F) != 0x21)
|
|
|
|
return 0; /* Only accept class request. */
|
2010-12-05 22:59:12 +00:00
|
|
|
|
2011-11-02 19:48:17 +00:00
|
|
|
switch (req->bRequest) {
|
2010-12-05 22:59:12 +00:00
|
|
|
case DFU_DNLOAD:
|
2011-11-02 19:48:17 +00:00
|
|
|
if ((len == NULL) || (*len == 0)) {
|
2011-03-13 01:50:53 +00:00
|
|
|
usbdfu_state = STATE_DFU_MANIFEST_SYNC;
|
|
|
|
return 1;
|
|
|
|
} else {
|
2011-11-02 19:48:17 +00:00
|
|
|
/* Copy download data for use on GET_STATUS. */
|
2011-03-13 01:50:53 +00:00
|
|
|
prog.blocknum = req->wValue;
|
|
|
|
prog.len = *len;
|
|
|
|
memcpy(prog.buf, *buf, *len);
|
|
|
|
usbdfu_state = STATE_DFU_DNLOAD_SYNC;
|
|
|
|
return 1;
|
|
|
|
}
|
2010-12-05 22:59:12 +00:00
|
|
|
case DFU_CLRSTATUS:
|
2011-11-02 19:48:17 +00:00
|
|
|
/* Clear error and return to dfuIDLE. */
|
|
|
|
if (usbdfu_state == STATE_DFU_ERROR)
|
2010-12-05 22:59:12 +00:00
|
|
|
usbdfu_state = STATE_DFU_IDLE;
|
|
|
|
return 1;
|
|
|
|
case DFU_ABORT:
|
2011-11-02 19:48:17 +00:00
|
|
|
/* Abort returns to dfuIDLE state. */
|
2010-12-05 22:59:12 +00:00
|
|
|
usbdfu_state = STATE_DFU_IDLE;
|
|
|
|
return 1;
|
|
|
|
case DFU_UPLOAD:
|
2011-11-02 19:48:17 +00:00
|
|
|
/* Upload not supported for now. */
|
2010-12-05 22:59:12 +00:00
|
|
|
return 0;
|
|
|
|
case DFU_GETSTATUS: {
|
|
|
|
u32 bwPollTimeout = 0; /* 24-bit integer in DFU class spec */
|
|
|
|
(*buf)[0] = usbdfu_getstatus(&bwPollTimeout);
|
|
|
|
(*buf)[1] = bwPollTimeout & 0xFF;
|
|
|
|
(*buf)[2] = (bwPollTimeout >> 8) & 0xFF;
|
|
|
|
(*buf)[3] = (bwPollTimeout >> 16) & 0xFF;
|
|
|
|
(*buf)[4] = usbdfu_state;
|
2011-11-02 19:48:17 +00:00
|
|
|
(*buf)[5] = 0; /* iString not used here */
|
2010-12-05 22:59:12 +00:00
|
|
|
*len = 6;
|
|
|
|
*complete = usbdfu_getstatus_complete;
|
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
case DFU_GETSTATE:
|
2011-11-02 19:48:17 +00:00
|
|
|
/* Return state with no state transision. */
|
2010-12-05 22:59:12 +00:00
|
|
|
*buf[0] = usbdfu_state;
|
|
|
|
*len = 1;
|
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
int main(void)
|
|
|
|
{
|
|
|
|
rcc_peripheral_enable_clock(&RCC_APB2ENR, RCC_APB2ENR_IOPAEN);
|
2011-11-02 19:48:17 +00:00
|
|
|
|
|
|
|
if (!gpio_get(GPIOA, GPIO10)) {
|
|
|
|
/* Boot the application if it's valid. */
|
|
|
|
if ((*(volatile u32 *)APP_ADDRESS & 0x2FFE0000) == 0x20000000) {
|
|
|
|
/* Set vector table base address. */
|
2010-12-05 22:59:12 +00:00
|
|
|
SCB_VTOR = APP_ADDRESS & 0xFFFF;
|
2011-11-02 19:48:17 +00:00
|
|
|
/* Initialise master stack pointer. */
|
|
|
|
asm volatile("msr msp, %0"::"g"
|
|
|
|
(*(volatile u32 *)APP_ADDRESS));
|
|
|
|
/* Jump to application. */
|
|
|
|
(*(void (**)())(APP_ADDRESS + 4))();
|
2010-12-05 22:59:12 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2011-11-02 19:48:17 +00:00
|
|
|
rcc_clock_setup_in_hsi_out_48mhz();
|
2010-12-05 22:59:12 +00:00
|
|
|
|
2011-10-29 22:48:55 +00:00
|
|
|
rcc_peripheral_enable_clock(&RCC_APB2ENR, RCC_APB2ENR_IOPCEN);
|
2011-12-17 01:11:39 +00:00
|
|
|
rcc_peripheral_enable_clock(&RCC_AHBENR, RCC_AHBENR_OTGFSEN);
|
2010-12-05 22:59:12 +00:00
|
|
|
|
2011-10-29 22:48:55 +00:00
|
|
|
gpio_set(GPIOC, GPIO2);
|
2011-11-02 19:48:17 +00:00
|
|
|
gpio_set_mode(GPIOC, GPIO_MODE_OUTPUT_50_MHZ,
|
|
|
|
GPIO_CNF_OUTPUT_PUSHPULL, GPIO2);
|
2010-12-05 22:59:12 +00:00
|
|
|
|
2011-10-29 22:29:49 +00:00
|
|
|
usbd_init(&stm32f107_usb_driver, &dev, &config, usb_strings);
|
2010-12-05 22:59:12 +00:00
|
|
|
usbd_set_control_buffer_size(sizeof(usbd_control_buffer));
|
2011-03-13 01:50:53 +00:00
|
|
|
usbd_register_control_callback(
|
|
|
|
USB_REQ_TYPE_CLASS | USB_REQ_TYPE_INTERFACE,
|
|
|
|
USB_REQ_TYPE_TYPE | USB_REQ_TYPE_RECIPIENT,
|
|
|
|
usbdfu_control_request);
|
2010-12-05 22:59:12 +00:00
|
|
|
|
2011-10-29 22:48:55 +00:00
|
|
|
gpio_clear(GPIOC, GPIO2);
|
2010-12-05 22:59:12 +00:00
|
|
|
|
2011-11-02 19:48:17 +00:00
|
|
|
while (1)
|
2010-12-05 22:59:12 +00:00
|
|
|
usbd_poll();
|
|
|
|
}
|