[flashkit-mdc.git] / flashkit.c

/*
 * Tool for USB serial communication with Krikzz's FlashKit-MD
 * Copyright (c) 2017 notaz
 *
 * 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 <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <stdint.h>
#include <assert.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <arpa/inet.h> // hton
#include <termios.h>
#include <unistd.h>

#ifndef min
#define min(x, y) ((x) < (y) ? (x) : (y))
#define max(x, y) ((x) > (y) ? (x) : (y))
#endif

enum dev_cmd {
	CMD_ADDR = 0,
        CMD_LEN = 1,
        CMD_RD = 2,
        CMD_WR = 3,
        CMD_RY = 4,
        CMD_DELAY = 5,
};
#define PAR_MODE8  (1 << 4)
#define PAR_DEV_ID (1 << 5)
#define PAR_SINGE  (1 << 6)
#define PAR_INC    (1 << 7)

static int setup(int fd)
{
	struct termios tty;
	int ret;

	memset(&tty, 0, sizeof(tty));

	ret = tcgetattr(fd, &tty);
	if (ret != 0)
	{
		perror("tcgetattr");
		return 1;
	}

	tty.c_iflag &= ~(IGNBRK | BRKINT | PARMRK | ISTRIP
			| INLCR | IGNCR | ICRNL | IXON);
	tty.c_oflag &= ~OPOST;
	tty.c_lflag &= ~(ECHO | ECHONL | ICANON | ISIG | IEXTEN);
	tty.c_cflag &= ~(CSIZE | PARENB);
	tty.c_cflag |= CS8;

	//tty.c_cc[VMIN] = 1;
	//tty.c_cc[VTIME] = 5;            // 0.5 seconds read timeout

	ret = tcsetattr(fd, TCSANOW, &tty);
	if (ret != 0) {
		perror("tcsetattr");
		return ret;
	}

	return 0;
}

static int write_serial(int fd, const void *data, size_t size)
{
	int ret;

	ret = write(fd, data, size);
	if (ret != size) {
		fprintf(stderr, "write %d/%zd: ", ret, size);
		perror("");
		exit(1);
	}

	return 0;
}

static int read_serial(int fd, void *data, size_t size)
{
	size_t got = 0;
	int ret;

	while (got < size) {
		ret = read(fd, (char *)data + got, size - got);
		if (ret <= 0) {
			fprintf(stderr, "read %d %zd/%zd: ",
				ret, got, size);
			perror("");
			exit(1);
		}
		got += ret;
	}

	return 0;
}

/* addr arg is always byte address */
static void set_addr8(int fd, uint32_t addr)
{
	uint8_t cmd[6] = {
		CMD_ADDR, addr >> 16,
		CMD_ADDR, addr >> 8,
		CMD_ADDR, addr >> 0
	};
	write_serial(fd, cmd, sizeof(cmd));
}

static void set_addr16(int fd, uint32_t addr)
{
	set_addr8(fd, addr >> 1);
}

static uint16_t read_bus8(int fd, uint32_t addr)
{
	uint8_t cmd[7] = {
		CMD_ADDR, addr >> 16,
		CMD_ADDR, addr >> 8,
		CMD_ADDR, addr >> 0,
		CMD_RD | PAR_SINGE | PAR_MODE8
	};
	uint8_t r;

	write_serial(fd, cmd, sizeof(cmd));
	read_serial(fd, &r, sizeof(r));
	return r;
}

static uint16_t read_bus16(int fd, uint32_t addr)
{
	uint8_t cmd[7] = {
		CMD_ADDR, addr >> 17,
		CMD_ADDR, addr >> 9,
		CMD_ADDR, addr >> 1,
		CMD_RD | PAR_SINGE
	};
	uint16_t r;

	write_serial(fd, cmd, sizeof(cmd));
	read_serial(fd, &r, sizeof(r));
	return ntohs(r);
}

static void write_bus8(int fd, uint32_t addr, uint16_t d)
{
	uint8_t cmd[8] = {
		CMD_ADDR, addr >> 16,
		CMD_ADDR, addr >> 8,
		CMD_ADDR, addr >> 0,
		CMD_WR | PAR_SINGE | PAR_MODE8,
		d
	};

	write_serial(fd, cmd, sizeof(cmd));
}

static void write_bus16(int fd, uint32_t addr, uint16_t d)
{
	uint8_t cmd[9] = {
		CMD_ADDR, addr >> 17,
		CMD_ADDR, addr >> 9,
		CMD_ADDR, addr >> 1,
		CMD_WR | PAR_SINGE,
		d >> 8, d
	};

	write_serial(fd, cmd, sizeof(cmd));
}

static void read_block8(int fd, void *dst, uint32_t size)
{
	// PAR_MODE8 does not work, so read as 16bit and throw away MSB
	uint8_t tmp[0x10000], *d8 = dst;
	uint8_t cmd[5] = {
		CMD_LEN, size >> 8,
		CMD_LEN, size >> 0,
		CMD_RD | PAR_INC
	};
	int i;

	assert(size <= 0x10000);
	write_serial(fd, cmd, sizeof(cmd));
	read_serial(fd, dst, size);
	read_serial(fd, tmp, size);

	for (i = 0; i < size / 2; i++)
		d8[i] = d8[i * 2 + 1];
	d8 += size / 2;
	for (i = 0; i < size / 2; i++)
		d8[i] = tmp[i * 2 + 1];
}

static void read_block16(int fd, void *dst, uint32_t size)
{
	uint8_t cmd[5] = {
		CMD_LEN, size >> 9,
		CMD_LEN, size >> 1,
		CMD_RD | PAR_INC
	};

	assert(size <= 0x10000);
	write_serial(fd, cmd, sizeof(cmd));
	read_serial(fd, dst, size);
}

static void flash_seq_write8(int fd, uint32_t addr, const uint8_t *d)
{
	uint8_t cmd[] = {
		// unlock
		CMD_ADDR, 0,
		CMD_ADDR, 0x0a,
		CMD_ADDR, 0xaa,
		CMD_WR | PAR_SINGE | PAR_MODE8, 0xaa,
		CMD_ADDR, 0,
		CMD_ADDR, 0x05,
		CMD_ADDR, 0x55,
		CMD_WR | PAR_SINGE | PAR_MODE8, 0x55,
		// program setup
		CMD_ADDR, 0,
		CMD_ADDR, 0x0a,
		CMD_ADDR, 0xaa,
		CMD_WR | PAR_SINGE | PAR_MODE8, 0xa0,
		// program data
		CMD_ADDR, addr >> 16,
		CMD_ADDR, addr >> 8,
		CMD_ADDR, addr >> 0,
		CMD_WR | PAR_SINGE | PAR_MODE8, *d,
		CMD_RY
	};

	write_serial(fd, cmd, sizeof(cmd));
}

static void flash_seq_write16(int fd, uint32_t addr, const uint8_t *d)
{
	uint8_t cmd[] = {
		// unlock
		CMD_ADDR, 0,
		CMD_ADDR, 0x05,
		CMD_ADDR, 0x55,
		CMD_WR | PAR_SINGE | PAR_MODE8, 0xaa,
		CMD_ADDR, 0,
		CMD_ADDR, 0x02,
		CMD_ADDR, 0xaa,
		CMD_WR | PAR_SINGE | PAR_MODE8, 0x55,
		// program setup
		CMD_ADDR, 0,
		CMD_ADDR, 0x05,
		CMD_ADDR, 0x55,
		CMD_WR | PAR_SINGE | PAR_MODE8, 0xa0,
		// program data
		CMD_ADDR, addr >> 17,
		CMD_ADDR, addr >> 9,
		CMD_ADDR, addr >> 1,
		CMD_WR | PAR_SINGE, d[0], d[1],
		CMD_RY
	};

	write_serial(fd, cmd, sizeof(cmd));
}

static const struct iof
{
	void     (*set_addr)(int fd, uint32_t addr);
	uint16_t (*read_bus)(int fd, uint32_t addr);
	void     (*write_bus)(int fd, uint32_t addr, uint16_t d);
	void     (*read_block)(int fd, void *dst, uint32_t size);
	void     (*flash_seq_write)(int fd, uint32_t addr, const uint8_t *d);
}
io_ops[] =
{
	{ set_addr8,  read_bus8,  write_bus8,  read_block8,  flash_seq_write8  },
	{ set_addr16, read_bus16, write_bus16, read_block16, flash_seq_write16 },
};

static const struct iof *io = &io_ops[1];

static uint16_t flash_seq_r(int fd, uint8_t cmd, uint32_t addr)
{
	// unlock
	io->write_bus(fd, 0xaaa, 0xaa);
	io->write_bus(fd, 0x555, 0x55);

	io->write_bus(fd, 0xaaa, cmd);
	return io->read_bus(fd, addr);
}

static void flash_seq_erase(int fd, uint32_t addr)
{
	// printf("erase %06x\n", addr);
	io->write_bus(fd, 0xaaa, 0xaa);
	io->write_bus(fd, 0x555, 0x55);
	io->write_bus(fd, 0xaaa, 0x80);

	io->write_bus(fd, 0xaaa, 0xaa);
	io->write_bus(fd, 0x555, 0x55);
	io->write_bus(fd, addr, 0x30);
}

// status wait + dummy read to cause a wait?
static uint16_t ry_read(int fd)
{
	uint8_t cmd[2] = { CMD_RY, CMD_RD | PAR_SINGE };
	uint16_t rv = 0;

	write_serial(fd, cmd, sizeof(cmd));
	read_serial(fd, &rv, sizeof(rv));
	return ntohs(rv);
}

static void set_delay(int fd, uint8_t delay)
{
	uint8_t cmd[2] = { CMD_DELAY, delay };

	write_serial(fd, cmd, sizeof(cmd));
}

static struct flash_info {
	uint16_t mid;
	uint16_t did;
	uint32_t size;
	uint16_t region_cnt;
	struct {
		uint32_t block_size;
		uint32_t block_count;
		uint32_t start;
		uint32_t size;
	} region[4];
} info;

static void read_info(int fd)
{
	static const uint16_t qry[3] = { 'Q', 'R', 'Y' };
	uint32_t total = 0;
	uint16_t resp[3];
	uint32_t i, a;

	info.mid = flash_seq_r(fd, 0x90, 0); // autoselect
	info.did = io->read_bus(fd, 2);

	// could enter CFI directly, but there seems to be a "stack"
	// of modes, so 2 exits would be needed
	io->write_bus(fd, 0, 0xf0);

	io->write_bus(fd, 0xaa, 0x98); // CFI Query
	resp[0] = io->read_bus(fd, 0x20);
	resp[1] = io->read_bus(fd, 0x22);
	resp[2] = io->read_bus(fd, 0x24);
	if (memcmp(resp, qry, sizeof(resp))) {
		fprintf(stderr, "unexpected CFI response: %04x %04x %04x\n",
			resp[0], resp[1], resp[2]);
		exit(1);
	}
	info.size = 1u << io->read_bus(fd, 0x4e);
	info.region_cnt = io->read_bus(fd, 0x58);
	assert(0 < info.region_cnt && info.region_cnt <= 4);
	for (i = 0, a = 0x5a; i < info.region_cnt; i++, a += 8) {
		info.region[i].block_count = io->read_bus(fd, a + 0) + 1;
		info.region[i].block_count += io->read_bus(fd, a + 2) << 8;
		info.region[i].block_size = io->read_bus(fd, a + 4) << 8;
		info.region[i].block_size |= io->read_bus(fd, a + 6) << 16;
		info.region[i].start = total;
		info.region[i].size =
			info.region[i].block_size * info.region[i].block_count;
		assert(info.region[i].size);
		total += info.region[i].size;
	}

	io->write_bus(fd, 0, 0xf0); // flash reset

	printf("Flash info:\n");
	printf("Manufacturer ID: %04x\n", info.mid);
	printf("Device ID: %04x\n", info.did);
	printf("size: %u\n", info.size);
	printf("Erase Block Regions: %u\n", info.region_cnt);
	for (i = 0; i < info.region_cnt; i++)
		printf("  %5u x %u\n", info.region[i].block_size,
			info.region[i].block_count);
	if (info.size != total)
		fprintf(stderr, "warning: total is %u, bad CFI?\n", total);
}

static uint32_t get_block_addr(uint32_t addr, uint32_t blk_offset)
{
	uint32_t i, base, faddr;

	assert(info.region_cnt);
	assert(info.size);

	// get a flash address to allow mapper hardware
	faddr = addr & (info.size - 1);
	base = addr & ~(info.size - 1);

	for (i = 0; i < info.region_cnt; i++) {
		if (info.region[i].start <= faddr
		    && faddr < info.region[i].start + info.region[i].size)
		{
			uint32_t blk = (faddr - info.region[i].start)
					/ info.region[i].block_size
					+ blk_offset;
			return base + info.region[i].start
				+ blk * info.region[i].block_size;
		}
	}

	fprintf(stderr, "\naddress out of range: 0x%x\n", addr);
	exit(1);
}

static void print_progress(uint32_t done, uint32_t total)
{
	int i, step;

	printf("\r%06x/%06x |", done, total);

	step = (total + 19) / 20;
	for (i = step; i <= total; i += step)
		fputc(done >= i ? '=' : '-', stdout);
	printf("| %3d%%", done * 100 / total);
	fflush(stdout);
	if (done >= total)
		fputc('\n', stdout);
}

static FILE *open_prep_read(const char *fname, long *size)
{
	FILE *f = fopen(fname, "rb");
	if (!f) {
		fprintf(stderr, "fopen %s: ", fname);
		perror("");
		exit(1);
	}
	if (*size <= 0) {
		fseek(f, 0, SEEK_END);
		*size = ftell(f);
		fseek(f, 0, SEEK_SET);
	}
	if (*size <= 0) {
		fprintf(stderr, "size of %s is %ld\n", fname, *size);
		exit(1);
	}
	return f;
}

static void usage(const char *argv0)
{
	printf("usage:\n"
		"%s [options]\n"
		"  -d <ttydevice>      (default /dev/ttyUSB0)\n"
		"  -r <file> [size]    dump the cart (default 4MB)\n"
		"  -w <file> [size]    program the flash (def. file size)\n"
		"  -s <file> [size]    simple write (SRAM, etc, def. file size)\n"
		"  -e <size>           erase (rounds to block size)\n"
		"  -a <start_address>  read/write start address (default 0)\n"
		"  -8                  8bit flash\n"
		"  -v                  verify written data\n"
		"  -i                  get info about the flash chip\n"
		, argv0);
	exit(1);
}

static void invarg(int argc, char *argv[], int arg)
{
	if (arg < argc)
		fprintf(stderr, "invalid arg %d: \"%s\"\n", arg, argv[arg]);
	else
		fprintf(stderr, "missing required argument %d\n", arg);
	exit(1);
}

static void *getarg(int argc, char *argv[], int arg)
{
	if (arg >= argc)
		invarg(argc, argv, arg);
	return argv[arg];
}

static uint8_t g_block[0x10000];
static uint8_t g_block2[0x10000];

int main(int argc, char *argv[])
{
	const char *portname = "/dev/ttyUSB0";
	const char *fname_w = NULL;
	const char *fname_r = NULL;
	const char *fname_ws = NULL;
	long size_w = 0;
	long size_r = 0;
	long size_ws = 0;
	long size_e = 0;
	long size_v = 0;
	long len, address_in = 0;
	long a, a_blk, end;
	int do_info = 0;
	int do_verify = 0;
	int write_step = 2;
	FILE *f_w = NULL;
	FILE *f_r = NULL;
	FILE *f_ws = NULL;
	uint8_t id[2] = { 0, 0 };
	uint8_t cmd;
	uint16_t rv;
	int arg = 1;
	int fd;

	if (argc < 2 || !strcmp(argv[1], "-h") || !strcmp(argv[1], "--help"))
		usage(argv[0]);

	for (arg = 1; arg < argc; arg++) {
		if (!strcmp(argv[arg], "-d")) {
			portname = getarg(argc, argv, ++arg);
			continue;
		}
		if (!strcmp(argv[arg], "-r")) {
			fname_r = getarg(argc, argv, ++arg);
			if (arg + 1 < argc && argv[arg + 1][0] != '-') {
				size_r = strtol(argv[++arg], NULL, 0);
				if (size_r <= 0)
					invarg(argc, argv, arg);
			}
			continue;
		}
		if (!strcmp(argv[arg], "-w")) {
			fname_w = getarg(argc, argv, ++arg);
			if (arg + 1 < argc && argv[arg + 1][0] != '-') {
				size_w = strtol(argv[++arg], NULL, 0);
				if (size_w <= 0)
					invarg(argc, argv, arg);
			}
			continue;
		}
		if (!strcmp(argv[arg], "-s")) {
			fname_ws = getarg(argc, argv, ++arg);
			if (arg + 1 < argc && argv[arg + 1][0] != '-') {
				size_ws = strtol(argv[++arg], NULL, 0);
				if (size_ws <= 0)
					invarg(argc, argv, arg);
			}
			continue;
		}
		if (!strcmp(argv[arg], "-a")) {
			address_in = strtol(getarg(argc, argv, ++arg), NULL, 0);
			if (address_in < 0 || (address_in & 1))
				invarg(argc, argv, arg);
			continue;
		}
		if (!strcmp(argv[arg], "-e")) {
			size_e = strtol(getarg(argc, argv, ++arg), NULL, 0);
			if (size_e <= 0)
				invarg(argc, argv, arg);
			continue;
		}
		if (!strcmp(argv[arg], "-8")) {
			io = &io_ops[0];
			write_step = 1;
			continue;
		}
		if (!strcmp(argv[arg], "-v")) {
			do_verify = 1;
			continue;
		}
		if (!strcmp(argv[arg], "-i")) {
			do_info = 1;
			continue;
		}
		invarg(argc, argv, arg);
	}

	if (fname_r && size_r == 0)
		size_r = 0x400000;

	if (fname_w) {
		f_w = open_prep_read(fname_w, &size_w);
		if (size_e < size_w)
			size_e = size_w;
		if (do_verify)
			size_v = size_w;
	}

	fd = open(portname, O_RDWR | O_NOCTTY | O_SYNC);
	if (fd < 0) {
		fprintf(stderr, "open %s: ", portname);
		perror("");
		return 1;
	}

	if (fname_ws)
		f_ws = open_prep_read(fname_ws, &size_ws);

	setup(fd);

	cmd = CMD_RD | PAR_SINGE | PAR_DEV_ID;
	write_serial(fd, &cmd, sizeof(cmd));
	read_serial(fd, id, sizeof(id));
	if (id[0] != id[1] || id[0] == 0) {
		fprintf(stderr, "unexpected id: %02x %02x\n", id[0], id[1]);
		return 1;
	}
	printf("flashkit id: %02x\n", id[0]);

	set_delay(fd, 1);
	io->write_bus(fd, 0, 0xf0); // flash reset

	if (do_info || size_e)
		read_info(fd);

	if (size_e) {
		// set_delay(fd, 0); // ?
		a_blk = get_block_addr(address_in, 0);
		end = address_in + size_e;

		printf("erasing %ld bytes:\n", size_e);
		print_progress(0, size_e);
		for (a = address_in; a < end; ) {
			flash_seq_erase(fd, a_blk);
			rv = ry_read(fd);
			if (rv != 0xffff) {
				fprintf(stderr, "\nerase error: %lx %04x\n",
					a_blk, rv);
			}

			a_blk = get_block_addr(a_blk, 1);
			a += a_blk - a;
			print_progress(a - address_in, size_e);
		}
	}
	if (f_w != NULL) {
		uint8_t b[2];
		set_delay(fd, 0);
		printf("flashing %ld bytes:\n", size_w);
		for (a = 0; a < size_w; a += write_step) {
			ssize_t r;

			b[1] = 0xff;
			len = min(size_w - a, write_step);
			r = fread(b, 1, len, f_w);
			if (r != len) {
				perror("\nfread");
				return 1;
			}
			io->flash_seq_write(fd, address_in + a, b);

			if (!(a & 0x3ff))
				print_progress(a, size_w);
		}
		print_progress(a, size_w);
		rv = ry_read(fd);
		if (write_step == 2 && rv != ((b[0] << 8) | b[1]))
			fprintf(stderr, "warning: last bytes: %04x %02x%02x\n",
				rv, b[0], b[1]);
		rewind(f_w);
		set_delay(fd, 1);
	}
	if (f_ws != NULL) {
		printf("writing %ld bytes:\n", size_ws);
		for (a = 0; a < size_ws; a += write_step) {
			uint16_t b = 0xffff;
			ssize_t r;

			len = min(size_ws - a, write_step);
			r = fread(&b, 1, len, f_ws);
			if (r != len) {
				perror("\nfread");
				return 1;
			}
			if (write_step == 2)
				b = htons(b);
			io->write_bus(fd, address_in + a, b);

			if (!(a & 0x3ff))
				print_progress(a, size_ws);
		}
		print_progress(a, size_ws);
	}

	if (fname_r || size_v) {
		long blks, blks_v, done, verify_diff = 0;

		blks = (size_r + sizeof(g_block) - 1) / sizeof(g_block);
		blks_v = (size_v + sizeof(g_block) - 1) / sizeof(g_block);
		blks = max(blks, blks_v);
		if (fname_r) {
			f_r = fopen(fname_r, "wb");
			if (!f_r) {
				fprintf(stderr, "fopen %s: ", fname_r);
				perror("");
				return 1;
			}
		}

		printf("reading %ld bytes:\n", max(size_r, size_v));
		print_progress(0, blks * sizeof(g_block));
		io->set_addr(fd, address_in);
		for (done = 0; done < size_r || done < size_v; ) {
			io->read_block(fd, g_block, sizeof(g_block));
			if (f_r && done < size_r) {
				len = min(size_r - done, sizeof(g_block));
				if (fwrite(g_block, 1, len, f_r) != len) {
					perror("fwrite");
					return 1;
				}
			}
			if (done < size_v) {
				len = min(size_v - done, sizeof(g_block));
				if (fread(g_block2, 1, len, f_w) != len) {
					perror("fread");
					return 1;
				}
				verify_diff |= memcmp(g_block, g_block2, len);
			}
			done += sizeof(g_block);
			print_progress(done, blks * sizeof(g_block));
		}
		if (verify_diff) {
			fprintf(stderr, "verify FAILED\n");
			return 1;
		}
	}
	if (f_r)
		fclose(f_r);
	if (f_w)
		fclose(f_w);

	return 0;
}
Commit	Line	Data
d9502b8d	1	/*
	2	* Tool for USB serial communication with Krikzz's FlashKit-MD
	3	* Copyright (c) 2017 notaz
	4	*
	5	* Permission is hereby granted, free of charge, to any person obtaining
	6	* a copy of this software and associated documentation files (the
	7	* "Software"), to deal in the Software without restriction, including
	8	* without limitation the rights to use, copy, modify, merge, publish,
	9	* distribute, sublicense, and/or sell copies of the Software, and to
	10	* permit persons to whom the Software is furnished to do so, subject to
	11	* the following conditions:
	12	*
	13	* The above copyright notice and this permission notice shall be
	14	* included in all copies or substantial portions of the Software.
	15	*
	16	* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
	17	* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
	18	* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
	19	* NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
	20	* BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
	21	* ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
	22	* CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
	23	* SOFTWARE.
	24	*/
	25
	26	#include <stdio.h>
	27	#include <string.h>
	28	#include <stdlib.h>
	29	#include <stdint.h>
	30	#include <assert.h>
	31	#include <sys/types.h>
	32	#include <sys/stat.h>
	33	#include <fcntl.h>
	34	#include <arpa/inet.h> // hton
	35	#include <termios.h>
	36	#include <unistd.h>
	37
	38	#ifndef min
	39	#define min(x, y) ((x) < (y) ? (x) : (y))
	40	#define max(x, y) ((x) > (y) ? (x) : (y))
	41	#endif
	42
	43	enum dev_cmd {
	44	CMD_ADDR = 0,
	45	CMD_LEN = 1,
	46	CMD_RD = 2,
	47	CMD_WR = 3,
	48	CMD_RY = 4,
	49	CMD_DELAY = 5,
	50	};
	51	#define PAR_MODE8 (1 << 4)
	52	#define PAR_DEV_ID (1 << 5)
	53	#define PAR_SINGE (1 << 6)
	54	#define PAR_INC (1 << 7)
	55
	56	static int setup(int fd)
	57	{
	58	struct termios tty;
	59	int ret;
	60
	61	memset(&tty, 0, sizeof(tty));
	62
	63	ret = tcgetattr(fd, &tty);
	64	if (ret != 0)
65	{
66	perror("tcgetattr");
67	return 1;
68	}
69
70	tty.c_iflag &= ~(IGNBRK \| BRKINT \| PARMRK \| ISTRIP
71	\| INLCR \| IGNCR \| ICRNL \| IXON);
72	tty.c_oflag &= ~OPOST;
73	tty.c_lflag &= ~(ECHO \| ECHONL \| ICANON \| ISIG \| IEXTEN);
74	tty.c_cflag &= ~(CSIZE \| PARENB);
75	tty.c_cflag \|= CS8;
76
77	//tty.c_cc[VMIN] = 1;
78	//tty.c_cc[VTIME] = 5; // 0.5 seconds read timeout
79
80	ret = tcsetattr(fd, TCSANOW, &tty);
81	if (ret != 0) {
82	perror("tcsetattr");
83	return ret;
84	}
85
86	return 0;
87	}
88
89	static int write_serial(int fd, const void *data, size_t size)
90	{
91	int ret;
92
93	ret = write(fd, data, size);
94	if (ret != size) {
95	fprintf(stderr, "write %d/%zd: ", ret, size);
96	perror("");
97	exit(1);
98	}
99
100	return 0;
101	}
102
103	static int read_serial(int fd, void *data, size_t size)
104	{
105	size_t got = 0;
106	int ret;
107
108	while (got < size) {
109	ret = read(fd, (char *)data + got, size - got);
110	if (ret <= 0) {
111	fprintf(stderr, "read %d %zd/%zd: ",
112	ret, got, size);
113	perror("");
114	exit(1);
115	}
116	got += ret;
117	}
118
119	return 0;
120	}
121
c097fb89	122	/* addr arg is always byte address */
c097fb89	123	static void set_addr8(int fd, uint32_t addr)
d9502b8d	124	{
d9502b8d	125	uint8_t cmd[6] = {
c097fb89	126	CMD_ADDR, addr >> 16,
	127	CMD_ADDR, addr >> 8,
	128	CMD_ADDR, addr >> 0
	129	};
	130	write_serial(fd, cmd, sizeof(cmd));
	131	}
	132
	133	static void set_addr16(int fd, uint32_t addr)
	134	{
	135	set_addr8(fd, addr >> 1);
	136	}
	137
	138	static uint16_t read_bus8(int fd, uint32_t addr)
	139	{
	140	uint8_t cmd[7] = {
	141	CMD_ADDR, addr >> 16,
	142	CMD_ADDR, addr >> 8,
	143	CMD_ADDR, addr >> 0,
	144	CMD_RD \| PAR_SINGE \| PAR_MODE8
d9502b8d	145	};
c097fb89	146	uint8_t r;
c097fb89	147
d9502b8d	148	write_serial(fd, cmd, sizeof(cmd));
c097fb89	149	read_serial(fd, &r, sizeof(r));
c097fb89	150	return r;
d9502b8d	151	}
d9502b8d	152
c097fb89	153	static uint16_t read_bus16(int fd, uint32_t addr)
d9502b8d	154	{
	155	uint8_t cmd[7] = {
	156	CMD_ADDR, addr >> 17,
	157	CMD_ADDR, addr >> 9,
	158	CMD_ADDR, addr >> 1,
	159	CMD_RD \| PAR_SINGE
	160	};
	161	uint16_t r;
	162
	163	write_serial(fd, cmd, sizeof(cmd));
	164	read_serial(fd, &r, sizeof(r));
	165	return ntohs(r);
	166	}
	167
c097fb89	168	static void write_bus8(int fd, uint32_t addr, uint16_t d)
	169	{
	170	uint8_t cmd[8] = {
	171	CMD_ADDR, addr >> 16,
	172	CMD_ADDR, addr >> 8,
	173	CMD_ADDR, addr >> 0,
	174	CMD_WR \| PAR_SINGE \| PAR_MODE8,
	175	d
	176	};
	177
	178	write_serial(fd, cmd, sizeof(cmd));
	179	}
	180
	181	static void write_bus16(int fd, uint32_t addr, uint16_t d)
d9502b8d	182	{
	183	uint8_t cmd[9] = {
	184	CMD_ADDR, addr >> 17,
	185	CMD_ADDR, addr >> 9,
	186	CMD_ADDR, addr >> 1,
	187	CMD_WR \| PAR_SINGE,
	188	d >> 8, d
	189	};
	190
	191	write_serial(fd, cmd, sizeof(cmd));
	192	}
	193
c097fb89	194	static void read_block8(int fd, void *dst, uint32_t size)
d9502b8d	195	{
c097fb89	196	// PAR_MODE8 does not work, so read as 16bit and throw away MSB
c097fb89	197	uint8_t tmp[0x10000], *d8 = dst;
d9502b8d	198	uint8_t cmd[5] = {
c097fb89	199	CMD_LEN, size >> 8,
c097fb89	200	CMD_LEN, size >> 0,
d9502b8d	201	CMD_RD \| PAR_INC
d9502b8d	202	};
c097fb89	203	int i;
d9502b8d	204
	205	assert(size <= 0x10000);
	206	write_serial(fd, cmd, sizeof(cmd));
	207	read_serial(fd, dst, size);
c097fb89	208	read_serial(fd, tmp, size);
	209
	210	for (i = 0; i < size / 2; i++)
	211	d8[i] = d8[i * 2 + 1];
	212	d8 += size / 2;
	213	for (i = 0; i < size / 2; i++)
	214	d8[i] = tmp[i * 2 + 1];
d9502b8d	215	}
d9502b8d	216
c097fb89	217	static void read_block16(int fd, void *dst, uint32_t size)
d9502b8d	218	{
c097fb89	219	uint8_t cmd[5] = {
	220	CMD_LEN, size >> 9,
	221	CMD_LEN, size >> 1,
	222	CMD_RD \| PAR_INC
	223	};
d9502b8d	224
c097fb89	225	assert(size <= 0x10000);
	226	write_serial(fd, cmd, sizeof(cmd));
	227	read_serial(fd, dst, size);
d9502b8d	228	}
d9502b8d	229
c097fb89	230	static void flash_seq_write8(int fd, uint32_t addr, const uint8_t *d)
d9502b8d	231	{
c097fb89	232	uint8_t cmd[] = {
	233	// unlock
	234	CMD_ADDR, 0,
	235	CMD_ADDR, 0x0a,
	236	CMD_ADDR, 0xaa,
	237	CMD_WR \| PAR_SINGE \| PAR_MODE8, 0xaa,
	238	CMD_ADDR, 0,
	239	CMD_ADDR, 0x05,
	240	CMD_ADDR, 0x55,
	241	CMD_WR \| PAR_SINGE \| PAR_MODE8, 0x55,
	242	// program setup
	243	CMD_ADDR, 0,
	244	CMD_ADDR, 0x0a,
	245	CMD_ADDR, 0xaa,
	246	CMD_WR \| PAR_SINGE \| PAR_MODE8, 0xa0,
	247	// program data
	248	CMD_ADDR, addr >> 16,
	249	CMD_ADDR, addr >> 8,
	250	CMD_ADDR, addr >> 0,
	251	CMD_WR \| PAR_SINGE \| PAR_MODE8, *d,
	252	CMD_RY
	253	};
d9502b8d	254
c097fb89	255	write_serial(fd, cmd, sizeof(cmd));
d9502b8d	256	}
d9502b8d	257
c097fb89	258	static void flash_seq_write16(int fd, uint32_t addr, const uint8_t *d)
d9502b8d	259	{
	260	uint8_t cmd[] = {
	261	// unlock
	262	CMD_ADDR, 0,
	263	CMD_ADDR, 0x05,
	264	CMD_ADDR, 0x55,
	265	CMD_WR \| PAR_SINGE \| PAR_MODE8, 0xaa,
	266	CMD_ADDR, 0,
	267	CMD_ADDR, 0x02,
	268	CMD_ADDR, 0xaa,
	269	CMD_WR \| PAR_SINGE \| PAR_MODE8, 0x55,
	270	// program setup
	271	CMD_ADDR, 0,
	272	CMD_ADDR, 0x05,
	273	CMD_ADDR, 0x55,
	274	CMD_WR \| PAR_SINGE \| PAR_MODE8, 0xa0,
	275	// program data
	276	CMD_ADDR, addr >> 17,
	277	CMD_ADDR, addr >> 9,
	278	CMD_ADDR, addr >> 1,
	279	CMD_WR \| PAR_SINGE, d[0], d[1],
	280	CMD_RY
	281	};
	282
	283	write_serial(fd, cmd, sizeof(cmd));
	284	}
	285
c097fb89	286	static const struct iof
	287	{
	288	void (*set_addr)(int fd, uint32_t addr);
	289	uint16_t (*read_bus)(int fd, uint32_t addr);
	290	void (*write_bus)(int fd, uint32_t addr, uint16_t d);
	291	void (read_block)(int fd, void dst, uint32_t size);
	292	void (flash_seq_write)(int fd, uint32_t addr, const uint8_t d);
	293	}
	294	io_ops[] =
	295	{
	296	{ set_addr8, read_bus8, write_bus8, read_block8, flash_seq_write8 },
	297	{ set_addr16, read_bus16, write_bus16, read_block16, flash_seq_write16 },
	298	};
	299
	300	static const struct iof *io = &io_ops[1];
	301
	302	static uint16_t flash_seq_r(int fd, uint8_t cmd, uint32_t addr)
	303	{
	304	// unlock
	305	io->write_bus(fd, 0xaaa, 0xaa);
	306	io->write_bus(fd, 0x555, 0x55);
	307
	308	io->write_bus(fd, 0xaaa, cmd);
	309	return io->read_bus(fd, addr);
	310	}
	311
	312	static void flash_seq_erase(int fd, uint32_t addr)
	313	{
	314	// printf("erase %06x\n", addr);
	315	io->write_bus(fd, 0xaaa, 0xaa);
	316	io->write_bus(fd, 0x555, 0x55);
	317	io->write_bus(fd, 0xaaa, 0x80);
	318
	319	io->write_bus(fd, 0xaaa, 0xaa);
	320	io->write_bus(fd, 0x555, 0x55);
	321	io->write_bus(fd, addr, 0x30);
	322	}
	323
d9502b8d	324	// status wait + dummy read to cause a wait?
	325	static uint16_t ry_read(int fd)
	326	{
	327	uint8_t cmd[2] = { CMD_RY, CMD_RD \| PAR_SINGE };
	328	uint16_t rv = 0;
	329
	330	write_serial(fd, cmd, sizeof(cmd));
	331	read_serial(fd, &rv, sizeof(rv));
	332	return ntohs(rv);
	333	}
	334
	335	static void set_delay(int fd, uint8_t delay)
	336	{
	337	uint8_t cmd[2] = { CMD_DELAY, delay };
	338
	339	write_serial(fd, cmd, sizeof(cmd));
	340	}
	341
	342	static struct flash_info {
	343	uint16_t mid;
	344	uint16_t did;
	345	uint32_t size;
	346	uint16_t region_cnt;
	347	struct {
	348	uint32_t block_size;
	349	uint32_t block_count;
	350	uint32_t start;
	351	uint32_t size;
	352	} region[4];
	353	} info;
	354
	355	static void read_info(int fd)
	356	{
	357	static const uint16_t qry[3] = { 'Q', 'R', 'Y' };
	358	uint32_t total = 0;
	359	uint16_t resp[3];
	360	uint32_t i, a;
	361
	362	info.mid = flash_seq_r(fd, 0x90, 0); // autoselect
c097fb89	363	info.did = io->read_bus(fd, 2);
d9502b8d	364
	365	// could enter CFI directly, but there seems to be a "stack"
	366	// of modes, so 2 exits would be needed
c097fb89	367	io->write_bus(fd, 0, 0xf0);
d9502b8d	368
c097fb89	369	io->write_bus(fd, 0xaa, 0x98); // CFI Query
	370	resp[0] = io->read_bus(fd, 0x20);
	371	resp[1] = io->read_bus(fd, 0x22);
	372	resp[2] = io->read_bus(fd, 0x24);
d9502b8d	373	if (memcmp(resp, qry, sizeof(resp))) {
	374	fprintf(stderr, "unexpected CFI response: %04x %04x %04x\n",
	375	resp[0], resp[1], resp[2]);
	376	exit(1);
	377	}
c097fb89	378	info.size = 1u << io->read_bus(fd, 0x4e);
c097fb89	379	info.region_cnt = io->read_bus(fd, 0x58);
d9502b8d	380	assert(0 < info.region_cnt && info.region_cnt <= 4);
d9502b8d	381	for (i = 0, a = 0x5a; i < info.region_cnt; i++, a += 8) {
c097fb89	382	info.region[i].block_count = io->read_bus(fd, a + 0) + 1;
	383	info.region[i].block_count += io->read_bus(fd, a + 2) << 8;
	384	info.region[i].block_size = io->read_bus(fd, a + 4) << 8;
	385	info.region[i].block_size \|= io->read_bus(fd, a + 6) << 16;
d9502b8d	386	info.region[i].start = total;
	387	info.region[i].size =
	388	info.region[i].block_size * info.region[i].block_count;
	389	assert(info.region[i].size);
	390	total += info.region[i].size;
	391	}
	392
c097fb89	393	io->write_bus(fd, 0, 0xf0); // flash reset
d9502b8d	394
	395	printf("Flash info:\n");
	396	printf("Manufacturer ID: %04x\n", info.mid);
	397	printf("Device ID: %04x\n", info.did);
	398	printf("size: %u\n", info.size);
	399	printf("Erase Block Regions: %u\n", info.region_cnt);
	400	for (i = 0; i < info.region_cnt; i++)
	401	printf(" %5u x %u\n", info.region[i].block_size,
	402	info.region[i].block_count);
	403	if (info.size != total)
	404	fprintf(stderr, "warning: total is %u, bad CFI?\n", total);
	405	}
	406
	407	static uint32_t get_block_addr(uint32_t addr, uint32_t blk_offset)
	408	{
f34ac796	409	uint32_t i, base, faddr;
d9502b8d	410
d9502b8d	411	assert(info.region_cnt);
f34ac796	412	assert(info.size);
	413
	414	// get a flash address to allow mapper hardware
	415	faddr = addr & (info.size - 1);
	416	base = addr & ~(info.size - 1);
	417
d9502b8d	418	for (i = 0; i < info.region_cnt; i++) {
f34ac796	419	if (info.region[i].start <= faddr
f34ac796	420	&& faddr < info.region[i].start + info.region[i].size)
d9502b8d	421	{
f34ac796	422	uint32_t blk = (faddr - info.region[i].start)
d9502b8d	423	/ info.region[i].block_size
d9502b8d	424	+ blk_offset;
f34ac796	425	return base + info.region[i].start
d9502b8d	426	+ blk * info.region[i].block_size;
	427	}
	428	}
	429
	430	fprintf(stderr, "\naddress out of range: 0x%x\n", addr);
	431	exit(1);
	432	}
	433
	434	static void print_progress(uint32_t done, uint32_t total)
	435	{
	436	int i, step;
	437
c097fb89	438	printf("\r%06x/%06x \|", done, total);
d9502b8d	439
	440	step = (total + 19) / 20;
	441	for (i = step; i <= total; i += step)
	442	fputc(done >= i ? '=' : '-', stdout);
	443	printf("\| %3d%%", done * 100 / total);
	444	fflush(stdout);
	445	if (done >= total)
	446	fputc('\n', stdout);
	447	}
	448
78ad8139	449	static FILE open_prep_read(const char fname, long *size)
	450	{
	451	FILE *f = fopen(fname, "rb");
	452	if (!f) {
	453	fprintf(stderr, "fopen %s: ", fname);
	454	perror("");
	455	exit(1);
	456	}
	457	if (*size <= 0) {
	458	fseek(f, 0, SEEK_END);
	459	*size = ftell(f);
	460	fseek(f, 0, SEEK_SET);
	461	}
	462	if (*size <= 0) {
	463	fprintf(stderr, "size of %s is %ld\n", fname, *size);
	464	exit(1);
	465	}
	466	return f;
	467	}
	468
d9502b8d	469	static void usage(const char *argv0)
	470	{
	471	printf("usage:\n"
	472	"%s [options]\n"
	473	" -d <ttydevice> (default /dev/ttyUSB0)\n"
	474	" -r <file> [size] dump the cart (default 4MB)\n"
78ad8139	475	" -w <file> [size] program the flash (def. file size)\n"
78ad8139	476	" -s <file> [size] simple write (SRAM, etc, def. file size)\n"
d9502b8d	477	" -e <size> erase (rounds to block size)\n"
78ad8139	478	" -a <start_address> read/write start address (default 0)\n"
c097fb89	479	" -8 8bit flash\n"
d9502b8d	480	" -v verify written data\n"
	481	" -i get info about the flash chip\n"
	482	, argv0);
	483	exit(1);
	484	}
	485
	486	static void invarg(int argc, char *argv[], int arg)
	487	{
	488	if (arg < argc)
	489	fprintf(stderr, "invalid arg %d: \"%s\"\n", arg, argv[arg]);
	490	else
	491	fprintf(stderr, "missing required argument %d\n", arg);
	492	exit(1);
	493	}
	494
	495	static void getarg(int argc, char argv[], int arg)
	496	{
	497	if (arg >= argc)
	498	invarg(argc, argv, arg);
	499	return argv[arg];
	500	}
	501
	502	static uint8_t g_block[0x10000];
	503	static uint8_t g_block2[0x10000];
	504
	505	int main(int argc, char *argv[])
	506	{
	507	const char *portname = "/dev/ttyUSB0";
	508	const char *fname_w = NULL;
	509	const char *fname_r = NULL;
78ad8139	510	const char *fname_ws = NULL;
d9502b8d	511	long size_w = 0;
d9502b8d	512	long size_r = 0;
78ad8139	513	long size_ws = 0;
d9502b8d	514	long size_e = 0;
	515	long size_v = 0;
	516	long len, address_in = 0;
	517	long a, a_blk, end;
	518	int do_info = 0;
	519	int do_verify = 0;
c097fb89	520	int write_step = 2;
d9502b8d	521	FILE *f_w = NULL;
d9502b8d	522	FILE *f_r = NULL;
78ad8139	523	FILE *f_ws = NULL;
d9502b8d	524	uint8_t id[2] = { 0, 0 };
	525	uint8_t cmd;
	526	uint16_t rv;
	527	int arg = 1;
	528	int fd;
	529
	530	if (argc < 2 \|\| !strcmp(argv[1], "-h") \|\| !strcmp(argv[1], "--help"))
	531	usage(argv[0]);
	532
	533	for (arg = 1; arg < argc; arg++) {
	534	if (!strcmp(argv[arg], "-d")) {
	535	portname = getarg(argc, argv, ++arg);
	536	continue;
	537	}
	538	if (!strcmp(argv[arg], "-r")) {
	539	fname_r = getarg(argc, argv, ++arg);
	540	if (arg + 1 < argc && argv[arg + 1][0] != '-') {
	541	size_r = strtol(argv[++arg], NULL, 0);
	542	if (size_r <= 0)
	543	invarg(argc, argv, arg);
	544	}
	545	continue;
	546	}
	547	if (!strcmp(argv[arg], "-w")) {
	548	fname_w = getarg(argc, argv, ++arg);
	549	if (arg + 1 < argc && argv[arg + 1][0] != '-') {
	550	size_w = strtol(argv[++arg], NULL, 0);
	551	if (size_w <= 0)
	552	invarg(argc, argv, arg);
	553	}
	554	continue;
	555	}
78ad8139	556	if (!strcmp(argv[arg], "-s")) {
	557	fname_ws = getarg(argc, argv, ++arg);
	558	if (arg + 1 < argc && argv[arg + 1][0] != '-') {
	559	size_ws = strtol(argv[++arg], NULL, 0);
	560	if (size_ws <= 0)
	561	invarg(argc, argv, arg);
	562	}
	563	continue;
	564	}
d9502b8d	565	if (!strcmp(argv[arg], "-a")) {
	566	address_in = strtol(getarg(argc, argv, ++arg), NULL, 0);
	567	if (address_in < 0 \|\| (address_in & 1))
	568	invarg(argc, argv, arg);
	569	continue;
	570	}
	571	if (!strcmp(argv[arg], "-e")) {
	572	size_e = strtol(getarg(argc, argv, ++arg), NULL, 0);
	573	if (size_e <= 0)
	574	invarg(argc, argv, arg);
	575	continue;
	576	}
c097fb89	577	if (!strcmp(argv[arg], "-8")) {
	578	io = &io_ops[0];
	579	write_step = 1;
	580	continue;
	581	}
d9502b8d	582	if (!strcmp(argv[arg], "-v")) {
	583	do_verify = 1;
	584	continue;
	585	}
	586	if (!strcmp(argv[arg], "-i")) {
	587	do_info = 1;
	588	continue;
	589	}
	590	invarg(argc, argv, arg);
	591	}
	592
	593	if (fname_r && size_r == 0)
	594	size_r = 0x400000;
	595
	596	if (fname_w) {
78ad8139	597	f_w = open_prep_read(fname_w, &size_w);
d9502b8d	598	if (size_e < size_w)
	599	size_e = size_w;
	600	if (do_verify)
	601	size_v = size_w;
	602	}
	603
	604	fd = open(portname, O_RDWR \| O_NOCTTY \| O_SYNC);
	605	if (fd < 0) {
	606	fprintf(stderr, "open %s: ", portname);
	607	perror("");
	608	return 1;
	609	}
	610
78ad8139	611	if (fname_ws)
	612	f_ws = open_prep_read(fname_ws, &size_ws);
	613
d9502b8d	614	setup(fd);
	615
	616	cmd = CMD_RD \| PAR_SINGE \| PAR_DEV_ID;
	617	write_serial(fd, &cmd, sizeof(cmd));
	618	read_serial(fd, id, sizeof(id));
	619	if (id[0] != id[1] \|\| id[0] == 0) {
	620	fprintf(stderr, "unexpected id: %02x %02x\n", id[0], id[1]);
	621	return 1;
	622	}
	623	printf("flashkit id: %02x\n", id[0]);
	624
	625	set_delay(fd, 1);
c097fb89	626	io->write_bus(fd, 0, 0xf0); // flash reset
d9502b8d	627
	628	if (do_info \|\| size_e)
	629	read_info(fd);
	630
	631	if (size_e) {
	632	// set_delay(fd, 0); // ?
	633	a_blk = get_block_addr(address_in, 0);
	634	end = address_in + size_e;
	635
	636	printf("erasing %ld bytes:\n", size_e);
	637	print_progress(0, size_e);
	638	for (a = address_in; a < end; ) {
	639	flash_seq_erase(fd, a_blk);
	640	rv = ry_read(fd);
	641	if (rv != 0xffff) {
	642	fprintf(stderr, "\nerase error: %lx %04x\n",
	643	a_blk, rv);
	644	}
	645
	646	a_blk = get_block_addr(a_blk, 1);
	647	a += a_blk - a;
	648	print_progress(a - address_in, size_e);
	649	}
	650	}
	651	if (f_w != NULL) {
	652	uint8_t b[2];
	653	set_delay(fd, 0);
78ad8139	654	printf("flashing %ld bytes:\n", size_w);
c097fb89	655	for (a = 0; a < size_w; a += write_step) {
d9502b8d	656	ssize_t r;
	657
	658	b[1] = 0xff;
c097fb89	659	len = min(size_w - a, write_step);
d9502b8d	660	r = fread(b, 1, len, f_w);
	661	if (r != len) {
	662	perror("\nfread");
	663	return 1;
	664	}
c097fb89	665	io->flash_seq_write(fd, address_in + a, b);
d9502b8d	666
c097fb89	667	if (!(a & 0x3ff))
d9502b8d	668	print_progress(a, size_w);
	669	}
	670	print_progress(a, size_w);
	671	rv = ry_read(fd);
c097fb89	672	if (write_step == 2 && rv != ((b[0] << 8) \| b[1]))
d9502b8d	673	fprintf(stderr, "warning: last bytes: %04x %02x%02x\n",
	674	rv, b[0], b[1]);
	675	rewind(f_w);
	676	set_delay(fd, 1);
	677	}
78ad8139	678	if (f_ws != NULL) {
	679	printf("writing %ld bytes:\n", size_ws);
	680	for (a = 0; a < size_ws; a += write_step) {
	681	uint16_t b = 0xffff;
	682	ssize_t r;
	683
	684	len = min(size_ws - a, write_step);
	685	r = fread(&b, 1, len, f_ws);
	686	if (r != len) {
	687	perror("\nfread");
	688	return 1;
	689	}
	690	if (write_step == 2)
	691	b = htons(b);
	692	io->write_bus(fd, address_in + a, b);
	693
	694	if (!(a & 0x3ff))
	695	print_progress(a, size_ws);
	696	}
	697	print_progress(a, size_ws);
	698	}
d9502b8d	699
	700	if (fname_r \|\| size_v) {
	701	long blks, blks_v, done, verify_diff = 0;
	702
	703	blks = (size_r + sizeof(g_block) - 1) / sizeof(g_block);
	704	blks_v = (size_v + sizeof(g_block) - 1) / sizeof(g_block);
	705	blks = max(blks, blks_v);
	706	if (fname_r) {
	707	f_r = fopen(fname_r, "wb");
	708	if (!f_r) {
	709	fprintf(stderr, "fopen %s: ", fname_r);
	710	perror("");
	711	return 1;
	712	}
	713	}
	714
	715	printf("reading %ld bytes:\n", max(size_r, size_v));
	716	print_progress(0, blks * sizeof(g_block));
c097fb89	717	io->set_addr(fd, address_in);
d9502b8d	718	for (done = 0; done < size_r \|\| done < size_v; ) {
c097fb89	719	io->read_block(fd, g_block, sizeof(g_block));
d9502b8d	720	if (f_r && done < size_r) {
	721	len = min(size_r - done, sizeof(g_block));
	722	if (fwrite(g_block, 1, len, f_r) != len) {
	723	perror("fwrite");
	724	return 1;
	725	}
	726	}
	727	if (done < size_v) {
	728	len = min(size_v - done, sizeof(g_block));
	729	if (fread(g_block2, 1, len, f_w) != len) {
	730	perror("fread");
	731	return 1;
	732	}
	733	verify_diff \|= memcmp(g_block, g_block2, len);
	734	}
	735	done += sizeof(g_block);
	736	print_progress(done, blks * sizeof(g_block));
	737	}
	738	if (verify_diff) {
	739	fprintf(stderr, "verify FAILED\n");
	740	return 1;
	741	}
	742	}
	743	if (f_r)
	744	fclose(f_r);
	745	if (f_w)
	746	fclose(f_w);
	747
	748	return 0;
	749	}