Initial Commit - Copy from Altus Metrum AltOS

altoslib/AltosSelfFlash.java

@@ -0,0 +1,209 @@
+/*
+ * Copyright © 2013 Keith Packard <keithp@keithp.com>
+ *
+ * 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 2 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, write to the Free Software Foundation, Inc.,
+ * 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA.
+ */
+
+package org.altusmetrum.altoslib_14;
+
+import java.io.*;
+
+public class AltosSelfFlash extends AltosProgrammer {
+	File			file;
+	FileInputStream		input;
+	AltosHexfile		image;
+	AltosLink		link;
+	boolean			aborted;
+	AltosFlashListener	listener;
+	AltosRomconfig		rom_config;
+
+	void action(String s, int percent) {
+		if (listener != null && !aborted)
+			listener.position(s, percent);
+	}
+
+	void action(int part, int total) {
+		int percent = 100 * part / total;
+		action(String.format("%d/%d (%d%%)",
+				     part, total, percent),
+		       percent);
+	}
+
+	byte[] read_memory(long addr, int len) throws InterruptedException, IOException {
+		int b;
+		byte[]	data = new byte[len];
+
+		for (int offset = 0; offset < len; offset += 0x100) {
+			link.printf("R %x\n", addr + offset);
+			byte[]	reply = link.get_binary_reply(5000, 0x100);
+
+			if (reply == null)
+				throw new IOException("Read device memory timeout");
+			for (b = 0; b < 0x100 && b + offset < len; b++)
+				data[b+offset] = reply[b];
+		}
+		return data;
+	}
+
+	AltosHexfile read_hexfile(long addr, int len) throws InterruptedException {
+		try {
+			byte[] mem = read_memory(addr, len);
+
+			AltosHexfile	hexfile = new AltosHexfile(mem, addr);
+
+			if (image != null)
+				hexfile.add_symbols(image);
+			return hexfile;
+		} catch (IOException ie) {
+			return null;
+		}
+	}
+
+	void write_memory(long addr, byte[] data, int start, int len) {
+		int b;
+		link.printf("W %x\n", addr);
+		link.flush_output();
+		for (b = 0; b < len; b++)
+			link.putchar(data[start + b]);
+		for (; b < 0x100; b++)
+			link.putchar((byte) 0xff);
+	}
+
+	void reboot() {
+		link.printf("a\n");
+		link.flush_output();
+	}
+
+	public void flash() {
+		try {
+			if (!check_rom_config())
+				throw new IOException("Invalid rom config settings");
+
+			/*
+			 * Store desired config values into image
+			 */
+			rom_config.write(image);
+
+			int remain = image.data.length;
+			long flash_addr = image.address;
+			int image_start = 0;
+
+			action(AltosFlashListener.flash_start, 0);
+			action(0, image.data.length);
+			while (remain > 0 && !aborted) {
+				int this_time = remain;
+				if (this_time > 0x100)
+					this_time = 0x100;
+
+				if (link != null) {
+					/* write the data */
+					write_memory(flash_addr, image.data, image_start, this_time);
+
+					byte[] check = read_memory(flash_addr, this_time);
+					for (int i = 0; i < this_time; i++)
+						if (check[i] != image.data[image_start + i])
+							throw new IOException(String.format("Flash write failed at 0x%x (%02x != %02x)",
+											    image.address + image_start + i,
+											    check[i], image.data[image_start + i]));
+				} else {
+					Thread.sleep(100);
+				}
+
+				remain -= this_time;
+				flash_addr += this_time;
+				image_start += this_time;
+
+				action(image.data.length - remain, image.data.length);
+			}
+			if (!aborted) {
+				action(AltosFlashListener.flash_done, 100);
+			}
+			close();
+		} catch (IOException ie) {
+			action(ie.getMessage(), -1);
+			abort();
+		} catch (InterruptedException ie) {
+			abort();
+		}
+	}
+
+	public void close() {
+		if (link != null) {
+			reboot();
+			try {
+				link.close();
+			} catch (InterruptedException ie) {
+			}
+			link = null;
+		}
+	}
+
+	synchronized public void abort() {
+		aborted = true;
+		close();
+	}
+
+	private AltosHexfile get_rom() throws InterruptedException {
+		try {
+			long base = AltosRomconfig.fetch_base(image);
+			long bounds = AltosRomconfig.fetch_bounds(image);
+
+			if (link.debug)
+				System.out.printf("rom base %x bounds %x\n", base, bounds);
+			return read_hexfile(base, (int) (bounds - base));
+		} catch (AltosNoSymbol ns) {
+			return null;
+		}
+	}
+
+	public boolean check_rom_config() throws InterruptedException {
+		if (link == null) {
+			return true;
+		}
+		if (rom_config == null) {
+			AltosHexfile hexfile = get_rom();
+			if (hexfile != null)
+				rom_config = new AltosRomconfig(hexfile);
+		}
+		return rom_config != null && rom_config.valid();
+	}
+
+	public void set_romconfig (AltosRomconfig romconfig) {
+		rom_config = romconfig;
+	}
+
+	public AltosRomconfig target_romconfig(AltosUsbId usb_id, String usb_product) throws InterruptedException {
+		if (!check_rom_config())
+			return null;
+		if (rom_config.usb_id == null)
+			rom_config.usb_id = usb_id;
+		if (rom_config.usb_product == null)
+			rom_config.usb_product = usb_product;
+		return rom_config;
+	}
+
+	public AltosRomconfig image_romconfig() {
+		return new AltosRomconfig(image);
+	}
+
+	public AltosSelfFlash(File file, AltosLink link, AltosFlashListener listener)
+		throws IOException, FileNotFoundException, InterruptedException {
+		this.file = file;
+		this.link = link;
+		this.listener = listener;
+		input = new FileInputStream(file);
+		image = new AltosHexfile(input);
+	}
+}