src/bsp/lk/app/lpcboot/lpcboot.c

/*
 * Copyright (c) 2015 Brian Swetland
 *
 * Permission is hereby granted, free of charge, to any person obtaining
 * a copy of this software and associated documentation files
 * (the "Software"), to deal in the Software without restriction,
 * including without limitation the rights to use, copy, modify, merge,
 * publish, distribute, sublicense, and/or sell copies of the Software,
 * and to permit persons to whom the Software is furnished to do so,
 * subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be
 * included in all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
 * IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
 * CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
 * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
 * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 */

#include <app.h>
#include <err.h>
#include <debug.h>
#include <string.h>
#include <stdlib.h>
#include <printf.h>
#include <dev/udc.h>

#include <platform.h>
#include <arch/arm.h>
#include <kernel/thread.h>
#include <kernel/event.h>
#include <kernel/timer.h>

#include <platform/lpc43xx-gpio.h>

#define PIN_LED		PIN(1,1)
#define GPIO_LED	GPIO(0,8)

void spifi_init(void);
void spifi_page_program(u32 addr, u32 *ptr, u32 count);
void spifi_sector_erase(u32 addr);
int spifi_verify_erased(u32 addr, u32 count);
int spifi_verify_page(u32 addr, u32 *ptr);

static event_t txevt = EVENT_INITIAL_VALUE(txevt, 0, 0);
static event_t rxevt = EVENT_INITIAL_VALUE(rxevt, 0, 0);

static udc_request_t *txreq;
static udc_request_t *rxreq;
static udc_endpoint_t *txept;
static udc_endpoint_t *rxept;

static volatile int online;
static volatile int txstatus;
static volatile int rxstatus;
static volatile unsigned rxactual;

static void lpcboot_notify(udc_gadget_t *gadget, unsigned event) {
	if (event == UDC_EVENT_ONLINE) {
		online = 1;
	} else {
		online = 0;
	}
}

static void rx_complete(udc_request_t *req, unsigned actual, int status) {
	rxactual = actual;
	rxstatus = status;
	event_signal(&rxevt, 0);
}

static void tx_complete(udc_request_t *req, unsigned actual, int status) {
	txstatus = status;
	event_signal(&txevt, 0);
}

void usb_xmit(void *data, unsigned len) {
	event_unsignal(&txevt);
	txreq->buffer = data;
	txreq->length = len;
	txstatus = 1;
	udc_request_queue(txept, txreq);
	event_wait(&txevt);
}

int usb_recv(void *data, unsigned len, lk_time_t timeout) {
	event_unsignal(&rxevt);
	rxreq->buffer = data;
	rxreq->length = len;
	rxstatus = 1;
	udc_request_queue(rxept, rxreq);
	if (event_wait_timeout(&rxevt, timeout)) {
		return ERR_TIMED_OUT;
	}
	return rxactual;
}

static udc_device_t lpcboot_device = {
	.vendor_id = 0x1209,
	.product_id = 0x5039,
	.version_id = 0x0100,
};

static udc_endpoint_t *lpcboot_endpoints[2];

static udc_gadget_t lpcboot_gadget = {
	.notify = lpcboot_notify,
	.ifc_class = 0xFF,
	.ifc_subclass = 0xFF,
	.ifc_protocol = 0xFF,
	.ifc_endpoints = 2,
	.ept = lpcboot_endpoints,
};

static void lpcboot_init(const struct app_descriptor *app)
{
	udc_init(&lpcboot_device);
	lpcboot_endpoints[0] = txept = udc_endpoint_alloc(UDC_BULK_IN, 512);
	lpcboot_endpoints[1] = rxept = udc_endpoint_alloc(UDC_BULK_OUT, 512);
	txreq = udc_request_alloc();
	rxreq = udc_request_alloc();
	rxreq->complete = rx_complete;
	txreq->complete = tx_complete;
	udc_register_gadget(&lpcboot_gadget);
}

#define RAM_BASE	0x10000000
#define RAM_SIZE	(128 * 1024)

#define BOOT_BASE	0
#define BOOT_SIZE	(32 * 1024)

#define ROM_BASE	(32 * 1024)
#define ROM_SIZE	(128 * 1024)

struct device_info {
	u8 part[16];
	u8 board[16];
	u32 version;
	u32 ram_base;
	u32 ram_size;
	u32 rom_base;
	u32 rom_size;
	u32 unused0;
	u32 unused1;
	u32 unused2;
};

struct device_info DEVICE = {
	.part = "LPC43xx",
	.board = TARGET,
	.version = 0x0001000,
	.ram_base = RAM_BASE,
	.ram_size = RAM_SIZE,
	.rom_base = ROM_BASE,
	.rom_size = ROM_SIZE,
};


#define MAGIC1		0xAA113377
#define MAGIC2		0xAA773311
#define MAGIC1_ADDR	0x20003FF8
#define MAGIC2_ADDR	0x20003FFC

void boot_app(void) {
	writel(MAGIC1, MAGIC1_ADDR);
	writel(MAGIC2, MAGIC2_ADDR);
}

int erase_page(u32 addr) {
	spifi_sector_erase(addr);
	return spifi_verify_erased(addr, 0x1000/4);
}

int write_page(u32 addr, void *ptr) {
	unsigned n;
	u32 *x = ptr;
	for (n = 0; n < 16; n++) {
		spifi_page_program(addr, x, 256 / 4);
		if (spifi_verify_page(addr, x)) return -1;
		addr += 256;
		x += (256 / 4);
	}
	return 0;
}

static uint32_t ram[4096/4];

void handle(u32 magic, u32 cmd, u32 arg) {
	u32 reply[2];
	u32 addr, xfer;
	int err = 0;

	if (magic != 0xDB00A5A5)
		return;

	reply[0] = magic;
	reply[1] = -1;

	switch (cmd) {
	case 'E':
		reply[1] = erase_page(ROM_BASE);
		break;
	case 'W':
	case 'w':
		if (cmd == 'W') {
			if (arg > ROM_SIZE)
				break;
			addr = ROM_BASE;
		} else {
			if (arg > BOOT_SIZE)
				break;
			addr = BOOT_BASE;
		}
		reply[1] = 0;
		usb_xmit(reply, 8);
		while (arg > 0) {
			xfer = (arg > 4096) ? 4096 : arg;
			usb_recv(ram, xfer, INFINITE_TIME);
			if (!err) err = erase_page(addr);
			if (!err) err = write_page(addr, ram);
			addr += 4096;
			arg -= xfer;
		}
		printf("flash %s\n", err ? "ERROR" : "OK");
		reply[1] = err;
		break;
#if WITH_BOOT_TO_RAM
	case 'X':
		if (arg > RAM_SIZE)
			break;
		reply[1] = 0;
		usb_xmit(reply, 8);
		usb_recv(ram, arg);
		usb_xmit(reply, 8);

		/* let last txn clear */
		usb_recv_timeout(buf, 64, 10);

		boot_image(ram);
		break;
#endif
	case 'Q':
		reply[1] = 0;
		usb_xmit(reply, 8);
		usb_xmit(&DEVICE, sizeof(DEVICE));
		return;
	case 'A':
		boot_app();
	case 'R':
		/* reboot "normally" */
		reply[1] = 0;
		usb_xmit(reply, 8);
		udc_stop();
		platform_halt(HALT_ACTION_REBOOT, HALT_REASON_SW_RESET);
	default:
		break;
	}
	usb_xmit(reply, 8);
}

static short led_idx = 0;
static short led_delay[] = { 500, 100, 100, 100, };
static short led_state[] = {   1,   0,   1,   0, };
static timer_t led_timer = TIMER_INITIAL_VALUE(led_timer);

static enum handler_return led_timer_cb(timer_t *timer, lk_time_t now, void *arg) {
	gpio_set(GPIO_LED, led_state[led_idx]);
	timer_set_oneshot(timer, led_delay[led_idx], led_timer_cb, NULL);
	led_idx++;
	if (led_idx == (sizeof(led_state)/sizeof(led_state[0]))) {
		led_idx = 0;
	}
	return 0;
}

static void lpcboot_entry(const struct app_descriptor *app, void *args)
{
	lk_time_t timeout;
	int r;
	u32 buf[64/4];

#if 0
	timeout = INFINITE_TIME;
#else
	if (readl(32768) != 0) {
		timeout = 3000;
	} else {
		timeout = INFINITE_TIME;
	}
#endif

	pin_config(PIN_LED, PIN_MODE(0) | PIN_PLAIN);
	gpio_config(GPIO_LED, GPIO_OUTPUT);
	led_timer_cb(&led_timer, 0, NULL);

	udc_start();
	spifi_init();
	for (;;) {
		if (!online) {
			thread_yield();
			continue;
		}
		r = usb_recv(buf, 64, timeout);
		if (r == ERR_TIMED_OUT) {
			boot_app();
			platform_halt(HALT_ACTION_REBOOT, HALT_REASON_SW_RESET);
		}
		if (r == 12) {
			handle(buf[0], buf[1], buf[2]);
			timeout = INFINITE_TIME;
		}
	}
}

APP_START(usbtest)
	.init = lpcboot_init,
	.entry = lpcboot_entry,
APP_END