scripts: add kwboot tool
This tool is used with Marvell EBU SoC to trigger the UART boot mode provided by the SoC BootROM, and push the bootloader image to the target using the Xmodem protocol. It has been taken from the U-Boot source code, with minor modifications to make it work with Armada 370/XP platforms. Signed-off-by: Thomas Petazzoni <thomas.petazzoni@free-electrons.com> Tested-by: Gregory CLEMENT <gregory.clement@free-electrons.com> Signed-off-by: Sascha Hauer <s.hauer@pengutronix.de>
This commit is contained in:
parent
a2a728c416
commit
0535713bbf
3 changed files with 719 additions and 1 deletions
1
scripts/.gitignore
vendored
1
scripts/.gitignore
vendored
|
@ -3,6 +3,7 @@ bin2c
|
|||
gen_netx_image
|
||||
kallsyms
|
||||
kwbimage
|
||||
kwboot
|
||||
mk-am35xx-spi-image
|
||||
mkimage
|
||||
mkublheader
|
||||
|
|
|
@ -8,7 +8,7 @@ hostprogs-$(CONFIG_KALLSYMS) += kallsyms
|
|||
hostprogs-y += bin2c
|
||||
hostprogs-y += mkimage
|
||||
hostprogs-y += bareboxenv
|
||||
hostprogs-$(CONFIG_ARCH_MVEBU) += kwbimage
|
||||
hostprogs-$(CONFIG_ARCH_MVEBU) += kwbimage kwboot
|
||||
hostprogs-$(CONFIG_ARCH_NETX) += gen_netx_image
|
||||
hostprogs-$(CONFIG_ARCH_OMAP) += omap_signGP mk-am35xx-spi-image
|
||||
hostprogs-$(CONFIG_ARCH_S5PCxx) += s5p_cksum
|
||||
|
|
717
scripts/kwboot.c
Normal file
717
scripts/kwboot.c
Normal file
|
@ -0,0 +1,717 @@
|
|||
/*
|
||||
* Boot a Marvell Kirkwood SoC, with Xmodem over UART0.
|
||||
*
|
||||
* (c) 2012 Daniel Stodden <daniel.stodden@gmail.com>
|
||||
*
|
||||
* References: marvell.com, "88F6180, 88F6190, 88F6192, and 88F6281
|
||||
* Integrated Controller: Functional Specifications" December 2,
|
||||
* 2008. Chapter 24.2 "BootROM Firmware".
|
||||
*/
|
||||
|
||||
#include <stdlib.h>
|
||||
#include <stdio.h>
|
||||
#include <string.h>
|
||||
#include <stdarg.h>
|
||||
#include <libgen.h>
|
||||
#include <fcntl.h>
|
||||
#include <errno.h>
|
||||
#include <unistd.h>
|
||||
#include <stdint.h>
|
||||
#include <termios.h>
|
||||
#include <sys/mman.h>
|
||||
#include <sys/stat.h>
|
||||
|
||||
/*
|
||||
* Marvell BootROM UART Sensing
|
||||
*/
|
||||
|
||||
static unsigned char kwboot_msg_boot[] = {
|
||||
0xBB, 0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77
|
||||
};
|
||||
|
||||
static unsigned char kwboot_msg_debug[] = {
|
||||
0xDD, 0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77
|
||||
};
|
||||
|
||||
#define KWBOOT_MSG_REQ_DELAY 1000 /* ms */
|
||||
#define KWBOOT_MSG_RSP_TIMEO 1000 /* ms */
|
||||
|
||||
/*
|
||||
* Xmodem Transfers
|
||||
*/
|
||||
|
||||
#define SOH 1 /* sender start of block header */
|
||||
#define EOT 4 /* sender end of block transfer */
|
||||
#define ACK 6 /* target block ack */
|
||||
#define NAK 21 /* target block negative ack */
|
||||
#define CAN 24 /* target/sender transfer cancellation */
|
||||
|
||||
struct kwboot_block {
|
||||
uint8_t soh;
|
||||
uint8_t pnum;
|
||||
uint8_t _pnum;
|
||||
uint8_t data[128];
|
||||
uint8_t csum;
|
||||
} __attribute((packed));
|
||||
|
||||
#define KWBOOT_BLK_RSP_TIMEO 1000 /* ms */
|
||||
|
||||
static int kwboot_verbose;
|
||||
|
||||
static void
|
||||
kwboot_printv(const char *fmt, ...)
|
||||
{
|
||||
va_list ap;
|
||||
|
||||
if (kwboot_verbose) {
|
||||
va_start(ap, fmt);
|
||||
vprintf(fmt, ap);
|
||||
va_end(ap);
|
||||
fflush(stdout);
|
||||
}
|
||||
}
|
||||
|
||||
static void
|
||||
__spinner(void)
|
||||
{
|
||||
const char seq[] = { '-', '\\', '|', '/' };
|
||||
const int div = 8;
|
||||
static int state, bs;
|
||||
|
||||
if (state % div == 0) {
|
||||
fputc(bs, stdout);
|
||||
fputc(seq[state / div % sizeof(seq)], stdout);
|
||||
fflush(stdout);
|
||||
}
|
||||
|
||||
bs = '\b';
|
||||
state++;
|
||||
}
|
||||
|
||||
static void
|
||||
kwboot_spinner(void)
|
||||
{
|
||||
if (kwboot_verbose)
|
||||
__spinner();
|
||||
}
|
||||
|
||||
static void
|
||||
__progress(int pct, char c)
|
||||
{
|
||||
const int width = 70;
|
||||
static const char *nl = "";
|
||||
static int pos;
|
||||
|
||||
if (pos % width == 0)
|
||||
printf("%s%3d %% [", nl, pct);
|
||||
|
||||
fputc(c, stdout);
|
||||
|
||||
nl = "]\n";
|
||||
pos++;
|
||||
|
||||
if (pct == 100) {
|
||||
while (pos++ < width)
|
||||
fputc(' ', stdout);
|
||||
fputs(nl, stdout);
|
||||
}
|
||||
|
||||
fflush(stdout);
|
||||
|
||||
}
|
||||
|
||||
static void
|
||||
kwboot_progress(int _pct, char c)
|
||||
{
|
||||
static int pct;
|
||||
|
||||
if (_pct != -1)
|
||||
pct = _pct;
|
||||
|
||||
if (kwboot_verbose)
|
||||
__progress(pct, c);
|
||||
}
|
||||
|
||||
static int
|
||||
kwboot_tty_recv(int fd, void *buf, size_t len, int timeo)
|
||||
{
|
||||
int rc, nfds;
|
||||
fd_set rfds;
|
||||
struct timeval tv;
|
||||
ssize_t n;
|
||||
|
||||
rc = -1;
|
||||
|
||||
FD_ZERO(&rfds);
|
||||
FD_SET(fd, &rfds);
|
||||
|
||||
tv.tv_sec = 0;
|
||||
tv.tv_usec = timeo * 1000;
|
||||
if (tv.tv_usec > 1000000) {
|
||||
tv.tv_sec += tv.tv_usec / 1000000;
|
||||
tv.tv_usec %= 1000000;
|
||||
}
|
||||
|
||||
do {
|
||||
nfds = select(fd + 1, &rfds, NULL, NULL, &tv);
|
||||
if (nfds < 0)
|
||||
goto out;
|
||||
if (!nfds) {
|
||||
errno = ETIMEDOUT;
|
||||
goto out;
|
||||
}
|
||||
|
||||
n = read(fd, buf, len);
|
||||
if (n < 0)
|
||||
goto out;
|
||||
|
||||
buf = (char *)buf + n;
|
||||
len -= n;
|
||||
} while (len > 0);
|
||||
|
||||
rc = 0;
|
||||
out:
|
||||
return rc;
|
||||
}
|
||||
|
||||
static int
|
||||
kwboot_tty_send(int fd, const void *buf, size_t len)
|
||||
{
|
||||
int rc;
|
||||
ssize_t n;
|
||||
|
||||
rc = -1;
|
||||
|
||||
do {
|
||||
n = write(fd, buf, len);
|
||||
if (n < 0)
|
||||
goto out;
|
||||
|
||||
buf = (char *)buf + n;
|
||||
len -= n;
|
||||
} while (len > 0);
|
||||
|
||||
rc = tcdrain(fd);
|
||||
out:
|
||||
return rc;
|
||||
}
|
||||
|
||||
static int
|
||||
kwboot_tty_send_char(int fd, unsigned char c)
|
||||
{
|
||||
return kwboot_tty_send(fd, &c, 1);
|
||||
}
|
||||
|
||||
static speed_t
|
||||
kwboot_tty_speed(int baudrate)
|
||||
{
|
||||
switch (baudrate) {
|
||||
case 115200:
|
||||
return B115200;
|
||||
case 57600:
|
||||
return B57600;
|
||||
case 38400:
|
||||
return B38400;
|
||||
case 19200:
|
||||
return B19200;
|
||||
case 9600:
|
||||
return B9600;
|
||||
}
|
||||
|
||||
return -1;
|
||||
}
|
||||
|
||||
static int
|
||||
kwboot_open_tty(const char *path, speed_t speed)
|
||||
{
|
||||
int rc, fd;
|
||||
struct termios tio;
|
||||
|
||||
rc = -1;
|
||||
|
||||
fd = open(path, O_RDWR|O_NOCTTY|O_NDELAY);
|
||||
if (fd < 0)
|
||||
goto out;
|
||||
|
||||
memset(&tio, 0, sizeof(tio));
|
||||
|
||||
tio.c_iflag = 0;
|
||||
tio.c_cflag = CREAD|CLOCAL|CS8;
|
||||
|
||||
tio.c_cc[VMIN] = 1;
|
||||
tio.c_cc[VTIME] = 10;
|
||||
|
||||
cfsetospeed(&tio, speed);
|
||||
cfsetispeed(&tio, speed);
|
||||
|
||||
rc = tcsetattr(fd, TCSANOW, &tio);
|
||||
if (rc)
|
||||
goto out;
|
||||
|
||||
rc = fd;
|
||||
out:
|
||||
if (rc < 0) {
|
||||
if (fd >= 0)
|
||||
close(fd);
|
||||
}
|
||||
|
||||
return rc;
|
||||
}
|
||||
|
||||
static int
|
||||
kwboot_bootmsg(int tty, void *msg)
|
||||
{
|
||||
int rc;
|
||||
char c;
|
||||
|
||||
kwboot_printv("Sending boot message. Please reboot the target...");
|
||||
|
||||
do {
|
||||
rc = tcflush(tty, TCIOFLUSH);
|
||||
if (rc)
|
||||
break;
|
||||
|
||||
rc = kwboot_tty_send(tty, msg, 8);
|
||||
if (rc) {
|
||||
usleep(KWBOOT_MSG_REQ_DELAY * 1000);
|
||||
continue;
|
||||
}
|
||||
|
||||
rc = kwboot_tty_recv(tty, &c, 1, KWBOOT_MSG_RSP_TIMEO);
|
||||
|
||||
kwboot_spinner();
|
||||
|
||||
} while (rc || c != NAK);
|
||||
|
||||
kwboot_printv("\n");
|
||||
|
||||
return rc;
|
||||
}
|
||||
|
||||
static int
|
||||
kwboot_debugmsg(int tty, void *msg)
|
||||
{
|
||||
int rc;
|
||||
|
||||
kwboot_printv("Sending debug message. Please reboot the target...");
|
||||
|
||||
do {
|
||||
char buf[16];
|
||||
|
||||
rc = tcflush(tty, TCIOFLUSH);
|
||||
if (rc)
|
||||
break;
|
||||
|
||||
rc = kwboot_tty_send(tty, msg, 8);
|
||||
if (rc) {
|
||||
usleep(KWBOOT_MSG_REQ_DELAY * 1000);
|
||||
continue;
|
||||
}
|
||||
|
||||
rc = kwboot_tty_recv(tty, buf, 16, KWBOOT_MSG_RSP_TIMEO);
|
||||
|
||||
kwboot_spinner();
|
||||
|
||||
} while (rc);
|
||||
|
||||
kwboot_printv("\n");
|
||||
|
||||
return rc;
|
||||
}
|
||||
|
||||
static int
|
||||
kwboot_xm_makeblock(struct kwboot_block *block, const void *data,
|
||||
size_t size, int pnum)
|
||||
{
|
||||
const size_t blksz = sizeof(block->data);
|
||||
size_t n;
|
||||
int i;
|
||||
|
||||
block->pnum = pnum;
|
||||
block->_pnum = ~block->pnum;
|
||||
|
||||
n = size < blksz ? size : blksz;
|
||||
memcpy(&block->data[0], data, n);
|
||||
memset(&block->data[n], 0, blksz - n);
|
||||
|
||||
block->csum = 0;
|
||||
for (i = 0; i < n; i++)
|
||||
block->csum += block->data[i];
|
||||
|
||||
return n;
|
||||
}
|
||||
|
||||
static int
|
||||
kwboot_xm_sendblock(int fd, struct kwboot_block *block)
|
||||
{
|
||||
int rc, retries;
|
||||
char c;
|
||||
|
||||
retries = 16;
|
||||
do {
|
||||
rc = kwboot_tty_send(fd, block, sizeof(*block));
|
||||
if (rc)
|
||||
break;
|
||||
|
||||
do {
|
||||
rc = kwboot_tty_recv(fd, &c, 1, KWBOOT_BLK_RSP_TIMEO);
|
||||
if (rc)
|
||||
break;
|
||||
|
||||
if (c != ACK && c!= NAK && c != CAN)
|
||||
printf("%c", c);
|
||||
|
||||
} while (c != ACK && c != NAK && c != CAN);
|
||||
|
||||
if (c != ACK)
|
||||
kwboot_progress(-1, '+');
|
||||
|
||||
} while (c == NAK && retries-- > 0);
|
||||
|
||||
rc = -1;
|
||||
|
||||
switch (c) {
|
||||
case ACK:
|
||||
rc = 0;
|
||||
break;
|
||||
case NAK:
|
||||
errno = EBADMSG;
|
||||
break;
|
||||
case CAN:
|
||||
errno = ECANCELED;
|
||||
break;
|
||||
default:
|
||||
errno = EPROTO;
|
||||
break;
|
||||
}
|
||||
|
||||
return rc;
|
||||
}
|
||||
|
||||
static int
|
||||
kwboot_xmodem(int tty, const void *_data, size_t size)
|
||||
{
|
||||
const uint8_t *data = _data;
|
||||
int rc, pnum, N, err;
|
||||
|
||||
pnum = 1;
|
||||
N = 0;
|
||||
|
||||
kwboot_printv("Sending boot image...\n");
|
||||
|
||||
do {
|
||||
struct kwboot_block block;
|
||||
int n;
|
||||
|
||||
n = kwboot_xm_makeblock(&block,
|
||||
data + N, size - N,
|
||||
pnum++);
|
||||
if (n < 0)
|
||||
goto can;
|
||||
|
||||
if (!n)
|
||||
break;
|
||||
|
||||
rc = kwboot_xm_sendblock(tty, &block);
|
||||
if (rc)
|
||||
goto out;
|
||||
|
||||
N += n;
|
||||
kwboot_progress(N * 100 / size, '.');
|
||||
} while (1);
|
||||
|
||||
rc = kwboot_tty_send_char(tty, EOT);
|
||||
|
||||
out:
|
||||
return rc;
|
||||
|
||||
can:
|
||||
err = errno;
|
||||
kwboot_tty_send_char(tty, CAN);
|
||||
errno = err;
|
||||
goto out;
|
||||
}
|
||||
|
||||
static int
|
||||
kwboot_term_pipe(int in, int out, char *quit, int *s)
|
||||
{
|
||||
ssize_t nin, nout;
|
||||
char _buf[128], *buf = _buf;
|
||||
|
||||
nin = read(in, buf, sizeof(buf));
|
||||
if (nin < 0)
|
||||
return -1;
|
||||
|
||||
if (quit) {
|
||||
int i;
|
||||
|
||||
for (i = 0; i < nin; i++) {
|
||||
if (*buf == quit[*s]) {
|
||||
(*s)++;
|
||||
if (!quit[*s])
|
||||
return 0;
|
||||
buf++;
|
||||
nin--;
|
||||
} else
|
||||
while (*s > 0) {
|
||||
nout = write(out, quit, *s);
|
||||
if (nout <= 0)
|
||||
return -1;
|
||||
(*s) -= nout;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
while (nin > 0) {
|
||||
nout = write(out, buf, nin);
|
||||
if (nout <= 0)
|
||||
return -1;
|
||||
nin -= nout;
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
static int
|
||||
kwboot_terminal(int tty)
|
||||
{
|
||||
int rc, in, s;
|
||||
char *quit = "\34c";
|
||||
struct termios otio, tio;
|
||||
|
||||
rc = -1;
|
||||
|
||||
in = STDIN_FILENO;
|
||||
if (isatty(in)) {
|
||||
rc = tcgetattr(in, &otio);
|
||||
if (!rc) {
|
||||
tio = otio;
|
||||
cfmakeraw(&tio);
|
||||
rc = tcsetattr(in, TCSANOW, &tio);
|
||||
}
|
||||
if (rc) {
|
||||
perror("tcsetattr");
|
||||
goto out;
|
||||
}
|
||||
|
||||
kwboot_printv("[Type Ctrl-%c + %c to quit]\r\n",
|
||||
quit[0]|0100, quit[1]);
|
||||
} else
|
||||
in = -1;
|
||||
|
||||
rc = 0;
|
||||
s = 0;
|
||||
|
||||
do {
|
||||
fd_set rfds;
|
||||
int nfds = 0;
|
||||
|
||||
FD_SET(tty, &rfds);
|
||||
nfds = nfds < tty ? tty : nfds;
|
||||
|
||||
if (in >= 0) {
|
||||
FD_SET(in, &rfds);
|
||||
nfds = nfds < in ? in : nfds;
|
||||
}
|
||||
|
||||
nfds = select(nfds + 1, &rfds, NULL, NULL, NULL);
|
||||
if (nfds < 0)
|
||||
break;
|
||||
|
||||
if (FD_ISSET(tty, &rfds)) {
|
||||
rc = kwboot_term_pipe(tty, STDOUT_FILENO, NULL, NULL);
|
||||
if (rc)
|
||||
break;
|
||||
}
|
||||
|
||||
if (FD_ISSET(in, &rfds)) {
|
||||
rc = kwboot_term_pipe(in, tty, quit, &s);
|
||||
if (rc)
|
||||
break;
|
||||
}
|
||||
} while (quit[s] != 0);
|
||||
|
||||
tcsetattr(in, TCSANOW, &otio);
|
||||
out:
|
||||
return rc;
|
||||
}
|
||||
|
||||
static void *
|
||||
kwboot_mmap_image(const char *path, size_t *size, int prot)
|
||||
{
|
||||
int rc, fd, flags;
|
||||
struct stat st;
|
||||
void *img;
|
||||
|
||||
rc = -1;
|
||||
fd = -1;
|
||||
img = NULL;
|
||||
|
||||
fd = open(path, O_RDONLY);
|
||||
if (fd < 0)
|
||||
goto out;
|
||||
|
||||
rc = fstat(fd, &st);
|
||||
if (rc)
|
||||
goto out;
|
||||
|
||||
flags = (prot & PROT_WRITE) ? MAP_PRIVATE : MAP_SHARED;
|
||||
|
||||
img = mmap(NULL, st.st_size, prot, flags, fd, 0);
|
||||
if (img == MAP_FAILED) {
|
||||
img = NULL;
|
||||
goto out;
|
||||
}
|
||||
|
||||
rc = 0;
|
||||
*size = st.st_size;
|
||||
out:
|
||||
if (rc && img) {
|
||||
munmap(img, st.st_size);
|
||||
img = NULL;
|
||||
}
|
||||
if (fd >= 0)
|
||||
close(fd);
|
||||
|
||||
return img;
|
||||
}
|
||||
|
||||
static void
|
||||
kwboot_usage(FILE *stream, char *progname)
|
||||
{
|
||||
fprintf(stream,
|
||||
"Usage: %s [-d | -b <image>] [ -p ] [ -t ] "
|
||||
"[-B <baud> ] <TTY>\n", progname);
|
||||
fprintf(stream, "\n");
|
||||
fprintf(stream, " -b <image>: boot <image>\n");
|
||||
fprintf(stream, " -d: enter debug mode\n");
|
||||
fprintf(stream, "\n");
|
||||
fprintf(stream, " -t: mini terminal\n");
|
||||
fprintf(stream, "\n");
|
||||
fprintf(stream, " -B <baud>: set baud rate\n");
|
||||
fprintf(stream, "\n");
|
||||
}
|
||||
|
||||
int
|
||||
main(int argc, char **argv)
|
||||
{
|
||||
const char *ttypath, *imgpath;
|
||||
int rv, rc, tty, term;
|
||||
void *bootmsg;
|
||||
void *debugmsg;
|
||||
void *img;
|
||||
size_t size;
|
||||
speed_t speed;
|
||||
|
||||
rv = 1;
|
||||
tty = -1;
|
||||
bootmsg = NULL;
|
||||
debugmsg = NULL;
|
||||
imgpath = NULL;
|
||||
img = NULL;
|
||||
term = 0;
|
||||
size = 0;
|
||||
speed = B115200;
|
||||
|
||||
kwboot_verbose = isatty(STDOUT_FILENO);
|
||||
|
||||
do {
|
||||
int c = getopt(argc, argv, "hb:dptB:");
|
||||
if (c < 0)
|
||||
break;
|
||||
|
||||
switch (c) {
|
||||
case 'b':
|
||||
bootmsg = kwboot_msg_boot;
|
||||
imgpath = optarg;
|
||||
break;
|
||||
|
||||
case 'd':
|
||||
debugmsg = kwboot_msg_debug;
|
||||
break;
|
||||
|
||||
case 't':
|
||||
term = 1;
|
||||
break;
|
||||
|
||||
case 'B':
|
||||
speed = kwboot_tty_speed(atoi(optarg));
|
||||
if (speed == -1)
|
||||
goto usage;
|
||||
break;
|
||||
|
||||
case 'h':
|
||||
rv = 0;
|
||||
default:
|
||||
goto usage;
|
||||
}
|
||||
} while (1);
|
||||
|
||||
if (!bootmsg && !term && !debugmsg)
|
||||
goto usage;
|
||||
|
||||
if (argc - optind < 1)
|
||||
goto usage;
|
||||
|
||||
ttypath = argv[optind++];
|
||||
|
||||
tty = kwboot_open_tty(ttypath, speed);
|
||||
if (tty < 0) {
|
||||
perror(ttypath);
|
||||
goto out;
|
||||
}
|
||||
|
||||
if (imgpath) {
|
||||
img = kwboot_mmap_image(imgpath, &size, PROT_READ);
|
||||
if (!img) {
|
||||
perror(imgpath);
|
||||
goto out;
|
||||
}
|
||||
}
|
||||
|
||||
if (debugmsg) {
|
||||
rc = kwboot_debugmsg(tty, debugmsg);
|
||||
if (rc) {
|
||||
perror("debugmsg");
|
||||
goto out;
|
||||
}
|
||||
}
|
||||
|
||||
if (bootmsg) {
|
||||
rc = kwboot_bootmsg(tty, bootmsg);
|
||||
if (rc) {
|
||||
perror("bootmsg");
|
||||
goto out;
|
||||
}
|
||||
}
|
||||
|
||||
if (img) {
|
||||
rc = kwboot_xmodem(tty, img, size);
|
||||
if (rc) {
|
||||
perror("xmodem");
|
||||
goto out;
|
||||
}
|
||||
}
|
||||
|
||||
if (term) {
|
||||
rc = kwboot_terminal(tty);
|
||||
if (rc && !(errno == EINTR)) {
|
||||
perror("terminal");
|
||||
goto out;
|
||||
}
|
||||
}
|
||||
|
||||
rv = 0;
|
||||
out:
|
||||
if (tty >= 0)
|
||||
close(tty);
|
||||
|
||||
if (img)
|
||||
munmap(img, size);
|
||||
|
||||
return rv;
|
||||
|
||||
usage:
|
||||
kwboot_usage(rv ? stderr : stdout, basename(argv[0]));
|
||||
goto out;
|
||||
}
|
Loading…
Reference in a new issue