Tidy up the flashing tool

usb-flasher/build.sh

@@ -2,4 +2,5 @@
 
 set -e
 
-gcc -o kbprog-usb kbprog-usb.c
+gcc -o ppkb-flasher flasher.c
+gcc -o ppkb-debugger debugger.c

usb-flasher/common.c

@@ -0,0 +1,205 @@
+#include <assert.h>
+#include <errno.h>
+#include <fcntl.h>
+#include <inttypes.h>
+#include <stdarg.h>
+#include <stdbool.h>
+#include <stdint.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <sys/stat.h>
+#include <time.h>
+#include <sys/poll.h>
+#include <sys/ioctl.h>
+#include <unistd.h>
+#include <sys/types.h>
+#include <dirent.h>
+#include <getopt.h>
+
+#include <linux/usbdevice_fs.h>
+
+static bool verbose;
+#define debug(args...) { if (verbose) printf(args); }
+
+static void syscall_error(int is_err, const char* fmt, ...)
+{
+	va_list ap;
+
+	if (!is_err)
+		return;
+
+	fprintf(stderr, "ERROR: ");
+	va_start(ap, fmt);
+	vfprintf(stderr, fmt, ap);
+	va_end(ap);
+	fprintf(stderr, ": %s\n", strerror(errno));
+
+	exit(1);
+}
+
+static void error(const char* fmt, ...)
+{
+	va_list ap;
+
+	fprintf(stderr, "ERROR: ");
+	va_start(ap, fmt);
+	vfprintf(stderr, fmt, ap);
+	va_end(ap);
+	fprintf(stderr, "\n");
+
+	exit(1);
+}
+
+static bool read_file(const char* path, char* buf, size_t size)
+{
+	int fd;
+	ssize_t ret;
+
+	fd = open(path, O_RDONLY);
+	if (fd < 0)
+		return false;
+
+	ret = read(fd, buf, size);
+	close(fd);
+	if (ret < 0)
+		return false;
+
+	if (ret < size) {
+		buf[ret] = 0;
+		return true;
+	} else {
+		buf[size - 1] = 0;
+		return false;
+	}
+}
+
+static int open_usb_dev(uint16_t vid, uint16_t pid)
+{
+	char path[256], buf[256];
+	struct dirent *e;
+	unsigned e_vid, e_pid, bus, dev;
+	int fd = -1, ret;
+	DIR* d;
+
+        d = opendir("/sys/bus/usb/devices");
+	syscall_error(d == NULL, "opendir(/sys/bus/usb/devices) failed");
+
+	while (true) {
+		errno = 0;
+		e = readdir(d);
+		syscall_error(e == NULL && errno, "readdir(/sys/bus/usb/devices) failed");
+		if (!e)
+			break;
+
+                if (!strcmp(e->d_name, ".") || !strcmp(e->d_name, ".."))
+			continue;
+
+		snprintf(path, sizeof path,
+			 "/sys/bus/usb/devices/%s/idVendor", e->d_name);
+		if (!read_file(path, buf, sizeof buf))
+			continue;
+
+		ret = sscanf(buf, "%x", &e_vid);
+		if (ret != 1)
+			error("Failed to parse %s", path);
+
+		snprintf(path, sizeof path,
+			 "/sys/bus/usb/devices/%s/idProduct", e->d_name);
+		if (!read_file(path, buf, sizeof buf))
+			continue;
+
+		ret = sscanf(buf, "%x", &e_pid);
+		if (ret != 1)
+			error("Failed to parse %s", path);
+
+		if (e_vid == vid && e_pid == pid) {
+			snprintf(path, sizeof path,
+				 "/sys/bus/usb/devices/%s/busnum", e->d_name);
+			if (!read_file(path, buf, sizeof buf))
+				error("Failed to read %s", path);
+
+			ret = sscanf(buf, "%u", &bus);
+			if (ret != 1)
+				error("Failed to parse %s", path);
+
+			snprintf(path, sizeof path,
+				 "/sys/bus/usb/devices/%s/devnum", e->d_name);
+			if (!read_file(path, buf, sizeof buf))
+				error("Failed to read %s", path);
+
+			ret = sscanf(buf, "%u", &dev);
+			if (ret != 1)
+				error("Failed to parse %s", path);
+
+			snprintf(path, sizeof path,
+				 "/dev/bus/usb/%03u/%03u", bus, dev);
+
+			debug("Found %04x:%04x at %s\n", e_vid, e_pid, path);
+
+			fd = open(path, O_RDWR);
+			syscall_error(fd < 0, "open(%s) failed", path);
+			break;
+		}
+	}
+
+	errno = ENOENT;
+	closedir(d);
+	return fd;
+}
+
+static int handle_urb(int usb_fd, struct usbdevfs_urb* urb, int timeout)
+{
+	int ret;
+	struct usbdevfs_urb* reaped_urb;
+	int retries = 0;
+
+retry:
+        ret = ioctl(usb_fd, USBDEVFS_SUBMITURB, urb);
+        if (ret < 0)
+        	return ret;
+
+	struct pollfd fd = {
+		.fd = usb_fd,
+		.events = POLLOUT,
+	};
+
+	ret = poll(&fd, 1, timeout);
+	if (ret <= 0) {
+		if (ret == 0)
+			errno = ETIMEDOUT;
+
+		int save_errno = errno;
+
+		// on timeout or other poll error, we need to discard and reap the submitted URB
+		ret = ioctl(usb_fd, USBDEVFS_DISCARDURB, urb);
+
+		// even if discard fails, URB may still be reapable, we need to try reaping anyway
+		ret = ioctl(usb_fd, USBDEVFS_REAPURBNDELAY, &reaped_urb);
+
+		// reap must immediately succeed, otherwise this is fatal
+		syscall_error(ret < 0, "USBDEVFS_REAPURBNDELAY failed");
+
+		errno = save_errno;
+		return -1;
+	}
+
+	// hopefully POLLERR means we get some error immediately on reap
+
+	ret = ioctl(usb_fd, USBDEVFS_REAPURB, &reaped_urb);
+        if (ret < 0)
+        	return ret;
+
+	// EPROTO errors are recoverable
+	if (urb->status == -71 && retries < 3) {
+		retries++;
+		goto retry;
+	}
+
+	if (urb->status != 0) {
+		errno = -urb->status;
+		return -1;
+	}
+
+	return 0;
+}

usb-flasher/debugger.c

@@ -0,0 +1,195 @@
+/**
+ * USB debuggign tool for Pinephone keyboard
+ *
+ * Copyright (C) 2021  Ondřej Jirman <megi@xff.cz>
+ *
+ * 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 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+#define DEBUG 1
+#include "common.c"
+
+int kb_open(void)
+{
+	int ret, fd;
+
+	// first check if keyboard USB device is available, if it is
+	// we need to first switch to bootloader mode
+
+	fd = open_usb_dev(0x04f3, 0xb001);
+	if (fd >= 0) {
+		struct usbdevfs_disconnect_claim dc = {
+			.interface = 0,
+		};
+
+		ret = ioctl(fd, USBDEVFS_DISCONNECT_CLAIM, &dc);
+		syscall_error(ret < 0, "USBDEVFS_DISCONNECT_CLAIM failed");
+	}
+
+	return fd;
+}
+
+static int usb_fd = -1;
+
+int command(uint8_t req[8])
+{
+	int ret;
+	struct usbdevfs_urb urb = {
+		.type = USBDEVFS_URB_TYPE_INTERRUPT,
+		.endpoint = 0x01,
+//		.flags = USBDEVFS_URB_ZERO_PACKET,
+		.buffer = req,
+		.buffer_length = 8,
+		.actual_length = 8,
+	};
+
+	ret = handle_urb(usb_fd, &urb, 100);
+	if (ret)
+		return ret;
+
+	debug("CMD:");
+	for (int i = 0; i < urb.actual_length; i++)
+		debug(" %02hhx", req[i]);
+	debug("\n");
+	
+	return 0;
+}
+
+int response(uint8_t res[8])
+{
+	int ret;
+	struct usbdevfs_urb urb = {
+		.type = USBDEVFS_URB_TYPE_INTERRUPT,
+		.endpoint = 0x82,
+		.flags = USBDEVFS_URB_SHORT_NOT_OK,
+		.buffer = res,
+		.buffer_length = 8,
+		.actual_length = 0,
+	};
+
+	ret = handle_urb(usb_fd, &urb, 100);
+	if (ret)
+		return ret;
+
+	if (urb.actual_length != 8)
+		error("Status response size invalid, must be 8 bytes, got %d", urb.actual_length);
+
+	debug("RES:");
+	for (int i = 0; i < urb.actual_length; i++)
+		debug(" %02hhx", res[i]);
+	debug("\n");
+	
+	return 0;
+}
+
+ssize_t xwrite(int fd, uint8_t* buf, size_t len)
+{
+	size_t off = 0;
+	
+	while (off < len) {
+		ssize_t ret = write(fd, buf + off, len - off);
+		if (ret < 0)
+			return ret;
+		
+		off += ret;
+	}
+	
+	return off;
+}
+
+int read_stdout(void)
+{
+	int ret;
+
+	uint8_t buf[64];
+	struct usbdevfs_urb urb = {
+		.type = USBDEVFS_URB_TYPE_INTERRUPT,
+		.endpoint = 0x83,
+		.buffer = buf,
+		.buffer_length = 64,
+		.actual_length = 0,
+	};
+
+	ret = handle_urb(usb_fd, &urb, 10);
+	if (ret)
+		return ret;
+
+	if (urb.actual_length > 0) {
+		ssize_t rv = xwrite(1, buf, urb.actual_length);
+		if (rv < 0)
+			return -1;
+	}
+
+	return 0;
+}
+
+int read_keys(uint8_t buf[12])
+{
+	int ret;
+	struct usbdevfs_urb urb = {
+		.type = USBDEVFS_URB_TYPE_INTERRUPT,
+		.endpoint = 0x84,
+		.flags = USBDEVFS_URB_SHORT_NOT_OK,
+		.buffer = buf,
+		.buffer_length = 12,
+		.actual_length = 0,
+	};
+
+	ret = handle_urb(usb_fd, &urb, 10);
+	if (ret)
+		return ret;
+
+	return 0;
+}
+
+void print_bitmap(uint8_t* map)
+{
+//	printf("\033[H");
+	for (int r = 0; r < 6; r++) {
+		if (r == 0) {
+			printf("   C");
+			for (int c = 0; c < 12; c++)
+				printf("%-3d", c + 1);
+			printf("\n");
+		}
+
+		printf("R%d", r + 1);
+		for (int c = 0; c < 12; c++)
+			printf("  %s", map[c] & (1u << r) ? "X" : ".");
+		printf("\n");
+	}
+}
+
+int main(int ac, char* av[])
+{
+	int ret;
+	uint8_t keys[12];
+	
+	usb_fd = kb_open();
+	if (usb_fd < 0)
+		error("Failed to open the keyboard");
+	
+	int i = 0;
+	while (1) {
+		ret = read_stdout();
+
+		ret = read_keys(keys);
+		if (ret == 0)
+			print_bitmap(keys);
+		
+		i++;
+	}
+
+	return 0;
+}

usb-flasher/flasher.c

@@ -17,164 +17,12 @@
  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  */
 
-// {{{ includes
-
-#include <assert.h>
-#include <errno.h>
-#include <fcntl.h>
-#include <inttypes.h>
-#include <stdarg.h>
-#include <stdbool.h>
-#include <stdint.h>
-#include <stdio.h>
-#include <stdlib.h>
-#include <string.h>
-#include <sys/stat.h>
-#include <time.h>
-#include <sys/ioctl.h>
-#include <unistd.h>
-#include <sys/types.h>
-#include <dirent.h>
-
-#include <linux/usbdevice_fs.h>
-
-#define DEBUG 1
-
-#if DEBUG
-#define debug(args...) printf(args)
-#else
-#define debug(args...)
-#endif
+#define VERSION "1.0"
+#include "common.c"
 
 // }}}
 // {{{ utils
 
-static void syscall_error(int is_err, const char* fmt, ...)
-{
-	va_list ap;
-
-	if (!is_err)
-		return;
-
-	fprintf(stderr, "ERROR: ");
-	va_start(ap, fmt);
-	vfprintf(stderr, fmt, ap);
-	va_end(ap);
-	fprintf(stderr, ": %s\n", strerror(errno));
-
-	exit(1);
-}
-
-static void error(const char* fmt, ...)
-{
-	va_list ap;
-
-	fprintf(stderr, "ERROR: ");
-	va_start(ap, fmt);
-	vfprintf(stderr, fmt, ap);
-	va_end(ap);
-	fprintf(stderr, "\n");
-
-	exit(1);
-}
-
-bool read_file(const char* path, char* buf, size_t size)
-{
-	int fd;
-	ssize_t ret;
-
-	fd = open(path, O_RDONLY);
-	if (fd < 0)
-		return false;
-
-	ret = read(fd, buf, size);
-	close(fd);
-	if (ret < 0)
-		return false;
-
-	if (ret < size) {
-		buf[ret] = 0;
-		return true;
-	} else {
-		buf[size - 1] = 0;
-		return false;
-	}
-}
-
-static int open_usb_dev(uint16_t vid, uint16_t pid)
-{
-	char path[256], buf[256];
-	struct dirent *e;
-	unsigned e_vid, e_pid, bus, dev;
-	int fd = -1, ret;
-	DIR* d;
-
-        d = opendir("/sys/bus/usb/devices");
-	syscall_error(d == NULL, "opendir(/sys/bus/usb/devices) failed");
-
-	while (true) {
-		errno = 0;
-		e = readdir(d);
-		syscall_error(e == NULL && errno, "readdir(/sys/bus/usb/devices) failed");
-		if (!e)
-			break;
-
-                if (!strcmp(e->d_name, ".") || !strcmp(e->d_name, ".."))
-			continue;
-
-		snprintf(path, sizeof path,
-			 "/sys/bus/usb/devices/%s/idVendor", e->d_name);
-		if (!read_file(path, buf, sizeof buf))
-			continue;
-
-		ret = sscanf(buf, "%x", &e_vid);
-		if (ret != 1)
-			error("Failed to parse %s", path);
-
-		snprintf(path, sizeof path,
-			 "/sys/bus/usb/devices/%s/idProduct", e->d_name);
-		if (!read_file(path, buf, sizeof buf))
-			continue;
-
-		ret = sscanf(buf, "%x", &e_pid);
-		if (ret != 1)
-			error("Failed to parse %s", path);
-
-		if (e_vid == vid && e_pid == pid) {
-			snprintf(path, sizeof path,
-				 "/sys/bus/usb/devices/%s/busnum", e->d_name);
-			if (!read_file(path, buf, sizeof buf))
-				error("Failed to read %s", path);
-
-			ret = sscanf(buf, "%u", &bus);
-			if (ret != 1)
-				error("Failed to parse %s", path);
-
-			snprintf(path, sizeof path,
-				 "/sys/bus/usb/devices/%s/devnum", e->d_name);
-			if (!read_file(path, buf, sizeof buf))
-				error("Failed to read %s", path);
-
-			ret = sscanf(buf, "%u", &dev);
-			if (ret != 1)
-				error("Failed to parse %s", path);
-
-			snprintf(path, sizeof path,
-				 "/dev/bus/usb/%03u/%03u", bus, dev);
-
-			debug("Found %04x:%04x at %s\n", e_vid, e_pid, path);
-
-			fd = open(path, O_RDWR);
-			syscall_error(fd < 0, "open(%s) failed", path);
-			break;
-		}
-	}
-
-	errno = ENOENT;
-	closedir(d);
-	return fd;
-}
-
 int bootloader_open(void)
 {
 	int ret, fd;
@@ -185,6 +33,8 @@ int bootloader_open(void)
 
 	fd = open_usb_dev(0x04f3, 0x1812);
 	if (fd >= 0) {
+		printf("Found USB device for the vendor firmware, swithing to USB bootloader\n");
+
 		for (unsigned i = 0; i <= 1; i++) {
 			struct usbdevfs_disconnect_claim dc = {
 				.interface = i,
@@ -223,14 +73,46 @@ int bootloader_open(void)
 
 		had_switch = true;
 		close(fd);
+		goto wait_bl;
 	}
 
-	// open the bootloader USB device (wait for it if we just switched to bootloader mode)
+	fd = open_usb_dev(0x04f3, 0xb001);
+	if (fd >= 0) {
+		printf("Found debugger USB device, swithing to USB bootloader\n");
 
+		struct usbdevfs_disconnect_claim dc = {
+			.interface = 0,
+		};
+
+		ret = ioctl(fd, USBDEVFS_DISCONNECT_CLAIM, &dc);
+		syscall_error(ret < 0, "USBDEVFS_DISCONNECT_CLAIM failed");
+
+		uint8_t buf[8] = { 0x01, };
+		
+		struct usbdevfs_urb urb = {
+			.type = USBDEVFS_URB_TYPE_INTERRUPT,
+			.endpoint = 0x01,
+			.buffer = buf,
+			.buffer_length = 8,
+			.actual_length = 0,
+		};
+
+		ret = handle_urb(fd, &urb, 500);
+		syscall_error(ret < 0, "handle_urb failed");
+
+		had_switch = true;
+		close(fd);
+		goto wait_bl;
+	}
+
+wait_bl:
+	// open the bootloader USB device (wait for it if we just switched to bootloader mode)
 	for (int i = 0;; i++) {
 		fd = open_usb_dev(0x04f3, 0x0905);
-		if (fd >= 0)
+		if (fd >= 0) {
+			printf("Found USB bootloader device\n");
 			break;
+		}
 
 		if (!had_switch)
 			error("Bootloader USB device not found");
@@ -261,12 +143,11 @@ enum {
 	EP_DATAIN = 0x84,
 };
 
-static int bootloader_fd = -1;
+static int usb_fd = -1;
 
 void bootloader_command(uint8_t req[8])
 {
 	int ret;
-	struct usbdevfs_urb* reaped_urb;
 	struct usbdevfs_urb urb = {
 		.type = USBDEVFS_URB_TYPE_INTERRUPT,
 		.endpoint = EP_CMD,
@@ -276,14 +157,8 @@ void bootloader_command(uint8_t req[8])
 		.actual_length = 8,
 	};
 
-        ret = ioctl(bootloader_fd, USBDEVFS_SUBMITURB, &urb);
-	syscall_error(ret < 0, "USBDEVFS_SUBMITURB failed");
-
-	ret = ioctl(bootloader_fd, USBDEVFS_REAPURB, &reaped_urb);
-	syscall_error(ret < 0, "USBDEVFS_REAPURB failed");
-
-	if (urb.status != 0)
-		error("URB failed with status=%s (%d)", strerror(urb.status), urb.status);
+	ret = handle_urb(usb_fd, &urb, 1000);
+	syscall_error(ret < 0, "handle_urb failed");
 
 	debug("CMD:");
 	for (int i = 0; i < urb.actual_length; i++)
@@ -294,7 +169,6 @@ void bootloader_command(uint8_t req[8])
 void bootloader_status(uint8_t res[4])
 {
 	int ret;
-	struct usbdevfs_urb* reaped_urb;
 	struct usbdevfs_urb urb = {
 		.type = USBDEVFS_URB_TYPE_INTERRUPT,
 		.endpoint = EP_STATUS,
@@ -304,16 +178,8 @@ void bootloader_status(uint8_t res[4])
 		.actual_length = 0,
 	};
 
-        ret = ioctl(bootloader_fd, USBDEVFS_SUBMITURB, &urb);
-	syscall_error(ret < 0, "USBDEVFS_SUBMITURB failed");
-
-	ret = ioctl(bootloader_fd, USBDEVFS_REAPURB, &reaped_urb);
-	syscall_error(ret < 0, "USBDEVFS_REAPURB failed");
-	
-	if (urb.status != 0)
-		error("URB failed with status=%s (%d)", strerror(urb.status), urb.status);
-	if (urb.actual_length != 4)
-		error("Status response too short, must be 4 bytes, got %d", urb.actual_length);
+	ret = handle_urb(usb_fd, &urb, 1000);
+	syscall_error(ret < 0, "handle_urb failed");
 	
 	debug("RES:");
 	for (int i = 0; i < urb.actual_length; i++)
@@ -373,7 +239,6 @@ void bootloader_standard_status_check(void)
 int bootloader_read_data(uint8_t res[64])
 {
 	int ret;
-	struct usbdevfs_urb* reaped_urb;
 	struct usbdevfs_urb urb = {
 		.type = USBDEVFS_URB_TYPE_INTERRUPT,
 		.endpoint = EP_DATAIN,
@@ -384,29 +249,20 @@ int bootloader_read_data(uint8_t res[64])
 //		.usercontext = (void*)(uintptr_t)0,
 	};
 
-        ret = ioctl(bootloader_fd, USBDEVFS_SUBMITURB, &urb);
-	syscall_error(ret < 0, "USBDEVFS_SUBMITURB failed");
-
-	ret = ioctl(bootloader_fd, USBDEVFS_REAPURB, &reaped_urb);
-	syscall_error(ret < 0, "USBDEVFS_REAPURB failed");
+	ret = handle_urb(usb_fd, &urb, 1000);
+	syscall_error(ret < 0, "handle_urb failed");
 
 	debug("DATA:");
 	for (int i = 0; i < urb.actual_length; i++)
 		debug(" %02hhx", res[i]);
 	debug("\n");
 
-	if (urb.status != 0)
-		error("URB failed with status=%s (%d)", strerror(urb.status), urb.status);
-	if (urb.actual_length != 64)
-		error("Data response too short, must be 64 bytes, got %d", urb.actual_length);
-
 	return urb.actual_length;
 }
 
 void bootloader_write_data(uint8_t res[64])
 {
 	int ret;
-	struct usbdevfs_urb* reaped_urb;
 	struct usbdevfs_urb urb = {
 		.type = USBDEVFS_URB_TYPE_INTERRUPT,
 		.endpoint = EP_DATAOUT,
@@ -417,14 +273,8 @@ void bootloader_write_data(uint8_t res[64])
 //		.usercontext = (void*)(uintptr_t)0,
 	};
 
-        ret = ioctl(bootloader_fd, USBDEVFS_SUBMITURB, &urb);
-	syscall_error(ret < 0, "USBDEVFS_SUBMITURB failed");
-
-	ret = ioctl(bootloader_fd, USBDEVFS_REAPURB, &reaped_urb);
-	syscall_error(ret < 0, "USBDEVFS_REAPURB failed");
-
-	if (urb.status != 0)
-		error("URB failed with status=%s (%d)", strerror(urb.status), urb.status);
+	ret = handle_urb(usb_fd, &urb, 1000);
+	syscall_error(ret < 0, "handle_urb failed");
 
 	debug("DATA:");
 	for (int i = 0; i < urb.actual_length; i++)
@@ -705,33 +555,198 @@ const char* boot_cond_text(uint8_t status)
 	}
 }
 
+static void usage(void)
+{
+	printf(
+	       "Usage: ppkb-flasher [--rom-in <path>] [--rom-out <path>] [--verbose]\n"
+	       "                    [--help] [<read|write|info|reset>...]\n"
+	       "\n"
+	       "Options:\n"
+	       "  -i, --rom-in <path>   Specify path to binary file you want to flash.\n"
+	       "  -o, --rom-out <path>  Specify path where you want to store the contents\n"
+	       "                        of code ROM read from the device.\n"
+	       "  -s, --rom-size <size> Specify how many bytes of code rom to flash\n"
+	       "                        starting from offset 0x2000 in the rom file.\n"
+	       "  -v, --verbose         Show details of what's going on.\n"
+	       "  -h, --help            This help.\n"
+	       "\n"
+	       "Commands:\n"
+	       "  info      Display information about the firmware.\n"
+	       "  read      Read ROM from the device to --rom-out file.\n"
+	       "  write     Flash ROM file to the device from --rom-in.\n"
+	       "  reset     Perform software reset of the MCU.\n"
+	       "\n"
+	       "Format of the ROM files is a flat binary. Only the part of it starting\n"
+	       "from 0x2000 will be flashed. Use -s to specify how many bytes to write.\n"
+	       "\n"
+	       "Pinephone keyboard flashing tool " VERSION "\n"
+	       "Written by Ondrej Jirman <megi@xff.cz>, 2021\n"
+	       "Licensed under GPLv3, see https://xff.cz/git/pinephone-keyboard/ for\n"
+	       "more information.\n"
+	);
+
+	exit(2);
+}
+
 int main(int ac, char* av[])
 {
-	bootloader_fd = bootloader_open();
+	char* rom_in = NULL;
+	char* rom_out = NULL;
+	int size = 0x1000;
+	int ret;
 
-	if (ac == 2 && !strcmp(av[1], "reset")) {
-		cmd_software_reset();
-		return 0;
+	while (1) {
+		int option_index = 0;
+		struct option long_options[] = {
+			{ "rom-in",  required_argument, 0, 'i' },
+			{ "rom-out", required_argument, 0, 'o' },
+			{ "size"   , required_argument, 0, 's' },
+			{ "verbose", no_argument,       0, 'v' },
+			{ "help",    no_argument,       0, 'h' },
+			{ 0,         0,                 0,  0  }
+		};
+
+		int c = getopt_long(ac, av, "i:o:s:vh", long_options, &option_index);
+		if (c == -1)
+			break;
+
+		switch (c) {
+		case 'o':
+			rom_out = strdup(optarg);
+			break;
+		case 'i':
+			rom_in = strdup(optarg);
+			break;
+		case 's':
+			if (strstr("0x", optarg) == optarg) {
+				errno = 0;
+				char* next = NULL;
+				size = strtol(optarg + 2, &next, 16);
+				if (errno || next == optarg + 2) {
+					printf("ERROR: Can't parse --size %s\n\n", optarg);
+					usage();
+				}
+			} else {
+				errno = 0;
+				char* next = NULL;
+				size = strtol(optarg, &next, 10);
+				if (errno || next == optarg) {
+					printf("ERROR: Can't parse --size %s\n\n", optarg);
+					usage();
+				}
+			}
+			break;
+		case 'v':
+			verbose = 1;
+			break;
+		case 'h':
+		case '?':
+		default:
+			usage();
+			break;
+		}
 	}
 
+	if (optind == ac)
+		usage();
+
+	if (size < 64) {
+		printf("ERROR: --size 0x%04x too small\n\n", size);
+		usage();
+	}
+
+	if (size > 0x6000) {
+		printf("ERROR: --size 0x%04x too large\n\n", size);
+		usage();
+	}
+
+	if (size % 64 != 0) {
+		printf("ERROR: --size 0x%04x is not multiple of 64\n\n", size);
+		usage();
+	}
+
+	for (int i = optind; i < ac; i++) {
+		if (!strcmp(av[i], "read")) {
+			if (!rom_out) {
+				printf("ERROR: You must specify target file to write rom contents to via --rom-out\n\n");
+				usage();
+			}
+		} else if (!strcmp(av[i], "write")) {
+			if (!rom_in) {
+				printf("ERROR: You must source file for flashing via --rom-in\n\n");
+				usage();
+			}
+		} else if (!strcmp(av[i], "info")) {
+			;
+		} else if (!strcmp(av[i], "reset")) {
+			;
+		} else {
+			printf("ERROR: Unknown command: %s\n\n", av[i]);
+			usage();
+		}
+	}
+
+	printf("Searching for bootloader USB device\n");
+	usb_fd = bootloader_open();
+
+	printf("Resetting bootloader state\n");
 	cmd_abort();
 
-	printf("FW=0x%04hx BOOT=0x%04hx\n", cmd_get_ver_fw(), cmd_get_ver_spec());
+	for (int i = optind; i < ac; i++) {
+		if (!strcmp(av[i], "read")) {
+			uint8_t rom[0x8000];
+			memset(rom, 0xff, sizeof rom);
+
+			printf("Reading code ROM\n");
+			cmd_read_rom(rom, 0, 0x8000);
+
+			int fd = open(rom_out, O_WRONLY | O_CREAT | O_TRUNC, 0666);
+			if (fd >= 0) {
+				ssize_t wr = write(fd, rom, 0x8000);
+				syscall_error(wr < 0, "write failed");
+				close(fd);
+			}
+		} else if (!strcmp(av[i], "write")) {
+			int fd;
+
+			uint8_t rom[0x8000];
+			memset(rom, 0xff, sizeof rom);
+
+			fd = open(rom_in, O_RDONLY);
+			syscall_error(fd < 0, "open(%s) failed", rom_in);
+			ssize_t len = read(fd, rom, 0x8000);
+			syscall_error(len < 0, "read failed");
+			close(fd);
+			if (len != 0x8000)
+				error("Invalid ROM file (%s) size (%d), must be 32768 bytes", rom_in, (int)len);
+
+			printf("Unlocking\n");
+			cmd_unlock();
+
+			printf("Entering flasing mode\n");
+			cmd_entry_iap();
+
+			printf("Flashing code ROM\n");
+			cmd_write_rom(rom + 0x2000, 0x2000, size);
+
+			printf("Finishing flashing\n");
+			cmd_finished_iap();
+		} else if (!strcmp(av[i], "info")) {
+			printf("Bootlaoder version: 0x%04hx\n", cmd_get_ver_spec());
+			printf("Firmware version: 0x%04hx\n", cmd_get_ver_fw());
 
 			uint8_t opts[128];
 			cmd_read_option(opts);
 
 			uint16_t icid = (opts[124] | opts[125] << 8) ^ (opts[121] | opts[122] << 8);
-	printf("ICID=%04hx\n", icid);
-
-	cmd_unlock();
+			printf("ICID: 0x%04hx\n", icid);
 
 			uint8_t bootcond = cmd_boot_condition();
-	printf("Booted via %s\n", boot_cond_text(bootcond));
+			printf("Booted via: %s\n", boot_cond_text(bootcond));
 
 			uint16_t csum_boot = cmd_get_checksum(CHECKSUM_TYPE_BOOT);
 			uint16_t csum_app = cmd_get_checksum(CHECKSUM_TYPE_MAIN);
-	printf("Checksums: boot=%04hx app=%04hx\n", csum_boot, csum_app);
+			printf("Checksums: boot=0x%04hx app=0x%04hx\n", csum_boot, csum_app);
 
 			/*
 			 * Checksums: boot=d355 app=449b
@@ -740,6 +755,8 @@ int main(int ac, char* av[])
 			 *
 			 *  CODE0 at 116: 0xff
 			 *   - 24MHz intosc mode, WDT disabled, 256kHz low freq mode
+			 *  CODE2:
+			 *   - BIT(0)
 			 *  CODE3 at 119: 0xff
 			 *   - eeprom and hw reset disabled
 			 *   - BITS(2..0) = reset button config
@@ -751,36 +768,26 @@ int main(int ac, char* av[])
 			 * ...: ff ff ff ff
 			 *
 			 * 120: df     // read by bootloader & 0x6 = 0x6: 24MHz,
-	 * 121: 0a 4f  // part of ICID (ICID is this value XORed with CSUM at 124, wtf?)
-	 * 123: 20     // ???
+			 * 121: 0a     // part of ICID (ICID is this value XORed with CSUM at 124, wtf?)
+			 * 122: 4f     // part of ICID (ICID is this value XORed with CSUM at 125, wtf?)
+			 * 123: 20     // ??? some count?
 			 * 124: 9b 44  // checksum (written by CMD_WRITECHECKSUM)
 			 * 126: 49     // ???
 			 * 127: aa     // app OK flag (0xaa = ok)
 			 */
 
-	int fd;
-	uint8_t rom[0x8000];
-	memset(rom, 0xff, sizeof rom);
-
-	fd = open("fw.bin", O_RDONLY);
-	syscall_error(fd < 0, "open(fw.bin) failed");
-	ssize_t len = read(fd, rom, 0x8000);
-	close(fd);
-	if (len != 0x8000)
-		error("Invalid rom size, must be 32768 bytes");
-
-	cmd_entry_iap();
-	cmd_write_rom(rom + 0x2000, 0x2000, 0x600);
-	cmd_finished_iap();
-
-        cmd_read_rom(rom, 0, 0x8000);
-	fd = open("rom.bin", O_WRONLY | O_CREAT | O_TRUNC, 0666);
-	if (fd >= 0) {
-		write(fd, rom, 0x8000);
-		close(fd);
-	}
-
+			printf("Option ROM dump:\n");
+			for (int i = 0; i < 128; i++)
+				if (opts[i] != 0xff)
+					printf("0x%02x (%d): 0x%02hhx\n", i + 128, i, opts[i]);
+		} else if (!strcmp(av[i], "reset")) {
+			printf("Restarting the MCU\n");
 			cmd_software_reset();
+		} else {
+			printf("ERROR: Unknown command: %s\n\n", av[i]);
+			usage();
+		}
+	}
 
 	return 0;
 }

usb-flasher/fw.bin

@@ -1 +0,0 @@
-../firmware/build/fw.bin