minor bug fix for ROM programmer

2 files changed, 11 insertions, 5 deletions

diff --git a/sw/programmer/linux/src/flash.c b/sw/programmer/linux/src/flash.c
index 48189da..6fcd09f 100644
--- a/sw/programmer/linux/src/flash.c
+++ b/sw/programmer/linux/src/flash.c
@@ -28,19 +28,24 @@ int flash_write(const char *romfile, void (*log)(const char *))
 		goto exit_romfd;
 
 	while ((head.size = read(romfd, buffer, FLASH_BLOCK_SIZE))) {
+        if (head.size < 0) {
+            log("[!] Error while reading from file\n");
+            break;
+        }
+                
 		head.addr = (uint16_t) lseek(romfd, 0, SEEK_CUR);
 
 		written = write(flash_serial_fd, &head, sizeof(struct flash_blk));
-		if (written < 0) {
-			// error
+		if (written != sizeof(struct flash_blk)) {
+            log("[!] Some bytes of flash_blk haven't been written\n");
 		}
 
 		written = write(flash_serial_fd, buffer, head.size);		
-		if (written < 0) {
-			// error
+		if (written != head.size) {
+            log("[#] Some bytes might not have been written\n");
 		}
 
-		char logbuf[64];
+        char logbuf[64];
 		sprintf(logbuf, "[@] Written %d bytes at address %d\n", head.size, head.addr);
 		log(logbuf);
 	}
diff --git a/sw/programmer/linux/src/ui.c b/sw/programmer/linux/src/ui.c
index 87dc0c7..b53ff0b 100644
--- a/sw/programmer/linux/src/ui.c
+++ b/sw/programmer/linux/src/ui.c
@@ -18,6 +18,7 @@ void ui_init(int *argc, char **argv[])
 
     /* connect objects to callbacks */
     window = GTK_WINDOW(gtk_builder_get_object(ui_builder, "window"));
+    g_signal_connect(window, "delete_event", G_CALLBACK(flash_close), NULL);
     g_signal_connect(window, "destroy", G_CALLBACK(gtk_main_quit), NULL);
 
     connectbtn = GTK_BUTTON(gtk_builder_get_object(ui_builder, "connectbtn"));