firestarter 1.3.2__tar.gz → 1.3.3__tar.gz

{firestarter-1.3.2 → firestarter-1.3.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: firestarter
-Version: 1.3.2
+Version: 1.3.3
 Summary: EPROM burner software for the Relatively Universal ROM Programmer
 Author-email: Henrik Olsson <henols@gmail.com>
 Description-Content-Type: text/markdown

firestarter-1.3.3/firestarter/__init__.py

@@ -0,0 +1 @@
+__version__ = "1.3.3"

{firestarter-1.3.2 → firestarter-1.3.3}/firestarter/eprom_operations.py

@@ -11,13 +11,13 @@ import os
 import time
 
 try:
-    from .serial_comm import find_programmer, wait_for_response
-    from .database import get_eprom
-    from .utils import extract_hex_to_decimal, print_progress, print_progress_bar
+    from .serial_comm import find_programmer, wait_for_response, consume_response
+    from .database import get_eprom, search_chip_id
+    from .utils import extract_hex_to_decimal, print_progress_bar
 except ImportError:
-    from serial_comm import find_programmer, wait_for_response
-    from database import get_eprom
-    from utils import extract_hex_to_decimal, print_progress, print_progress_bar
+    from serial_comm import find_programmer, wait_for_response, consume_response
+    from database import get_eprom, search_chip_id
+    from utils import extract_hex_to_decimal, print_progress_bar
 
 # Constants
 BUFFER_SIZE = 512
@@ -48,7 +48,7 @@ def read(eprom_name, output_file=None, force=False):
         eprom["force"] = True
 
     if not output_file:
-        output_file = f"{eprom_name}.bin"
+        output_file = f"{eprom_name.upper()}.bin"
     print(f"Reading EPROM {eprom_name}, saving to {output_file}")
 
     ser = find_programmer(eprom)
@@ -78,24 +78,26 @@ def read(eprom_name, output_file=None, force=False):
                         prefix=f"Address: 0x{from_address:X} - 0x{to_address:X}",
                         suffix=f"- {time.time() - start_time:.2f}s",
                     )
+
                     ser.write("OK".encode("ascii"))
                     ser.flush()
                 elif resp == "OK":
                     break
                 elif resp == "ERROR":
-                    print(f"\nError: {info}")
-                    return 1
-                else:
-                    print(f"\nUnexpected response: {info}")
+                    # print(f"\nError: {info}")
                     return 1
+                # else:
+                #     print(f"\nUnexpected response: {info}")
+                #     return 1
 
             print(f"\nRead complete in: {time.time() - start_time:.2f} seconds")
             print(f"Data saved to {output_file}")
     except Exception as e:
         print(f"Error while reading: {e}")
     finally:
+        consume_response(ser)
         ser.close()
-
+    return 0
 
 def write(eprom_name, input_file, address=None, ignore_blank_check=False, force=False):
     """
@@ -146,46 +148,42 @@ def write(eprom_name, input_file, address=None, ignore_blank_check=False, force=
             while True:
                 data = file.read(BUFFER_SIZE)
                 if not data:
+                    print("\nEnd of file reached")
                     ser.write(int(0).to_bytes(2, byteorder="big"))
                     ser.flush()
                     resp, info = wait_for_response(ser)
-                    print("\nEnd of file reached")
-                    print(info)
+                    while resp != "OK":
+                        if resp == "ERROR":
+                            return 1
+                        resp, info = wait_for_response(ser)
                     break
 
                 ser.write(len(data).to_bytes(2, byteorder="big"))
                 ser.flush()
                 resp, info = wait_for_response(ser)
-                if resp == "ERROR":
-                    print(f"\nError writing: {info}")
-                    return 1
+                while resp != "OK":
+                    if resp == "ERROR":
+                        return 1
+                    resp, info = wait_for_response(ser)
+
                 nr_bytes = ser.write(data)
                 bytes_written += nr_bytes
                 ser.flush()
-
-                while True:
-                    resp, info = wait_for_response(ser)
-                    if resp == "OK":
-                        # percent = int(bytes_written / file_size * 100)
-                        # print_progress(
-                        #     percent,
-                        #     bytes_written - nr_bytes + write_address,
-                        #     bytes_written + write_address,
-                        # )
-                        from_address = bytes_written - nr_bytes + write_address
-                        to_address = bytes_written + write_address
-                        print_progress_bar(
-                            bytes_written / BUFFER_SIZE,
-                            total_iterations,
-                            prefix=f"Address: 0x{from_address:X} - 0x{to_address:X}",
-                            suffix=f"- {time.time() - start_time:.2f}s",
-                        )
-                        break
-                    elif resp == "ERROR":
-                        print(f"\nError: {info}")
+                # print(f"Bytes written: {bytes_written}")
+                resp, info = wait_for_response(ser)
+                while resp != "OK":
+                    if resp == "ERROR":
                         return 1
-                    else:
-                        print(f"Warning: {info}")
+                    resp, info = wait_for_response(ser)
+
+                from_address = bytes_written - nr_bytes + write_address
+                to_address = bytes_written + write_address
+                print_progress_bar(
+                    bytes_written / BUFFER_SIZE,
+                    total_iterations,
+                    prefix=f"Address: 0x{from_address:X} - 0x{to_address:X}",
+                    suffix=f"- {time.time() - start_time:.2f}s",
+                )
                 if bytes_written == mem_size:
                     break
 
@@ -194,6 +192,7 @@ def write(eprom_name, input_file, address=None, ignore_blank_check=False, force=
         print(f"Error while writing: {e}")
         return 1
     finally:
+        consume_response(ser)
         ser.close()
     return 0
 
@@ -211,15 +210,21 @@ def erase(eprom_name):
         return 1
 
     eprom["state"] = STATE_ERASE
-    ser = find_programmer(eprom)
-    if not ser:
-        return 1
-
-    resp, info = wait_for_response(ser)
-    if resp == "OK":
-        print(f"EPROM {eprom_name} erased successfully.")
-    elif resp == "ERROR":
-        print(f"Error erasing EPROM {eprom_name}: {info}")
+    try:
+        ser = find_programmer(eprom)
+        if not ser:
+            return 1
+
+        resp, info = wait_for_response(ser)
+        if resp == "OK":
+            print(f"EPROM {eprom_name} erased successfully.")
+            return 0
+        elif resp == "ERROR":
+            print(f"Error erasing EPROM {eprom_name}")
+            return 1
+    finally:
+        consume_response(ser)
+        ser.close()
 
 
 def check_chip_id(eprom_name):
@@ -238,17 +243,26 @@ def check_chip_id(eprom_name):
     ser = find_programmer(eprom)
     if not ser:
         return 1
-
-    resp, info = wait_for_response(ser)
-    if resp == "OK":
-        print(f"Chip ID check passed for {eprom_name}: {info}")
-    elif resp == "ERROR":
-        print(f"Chip ID check failed for {eprom_name}: {info}")
+    try:
+        resp, info = wait_for_response(ser)
+        if resp == "OK":
+            print(f"Chip ID check passed for {eprom_name}: {info}")
+            return 0
         chip_id = extract_hex_to_decimal(info)
-        print(
-            f"Extracted chip ID: {chip_id}" if chip_id else "Failed to extract chip ID."
-        )
-
+        if not chip_id:
+            print(f"Failed to extract chip ID.")
+        else:
+            eproms = search_chip_id(chip_id)
+            if not eproms:
+                print(f"Chip ID 0x{chip_id:x} not found in the database.")
+            else:
+                print(f"Chip ID 0x{chip_id:x} found in the database:")
+            for eprom in eproms:
+                print(eprom)
+        return 1
+    finally:
+        consume_response(ser)
+        ser.close()
 
 def blank_check(eprom_name):
     """
@@ -263,12 +277,16 @@ def blank_check(eprom_name):
         return 1
 
     eprom["state"] = STATE_CHECK_BLANK
-    ser = find_programmer(eprom)
-    if not ser:
-        return 1
-
-    resp, info = wait_for_response(ser)
-    if resp == "OK":
-        print(f"EPROM {eprom_name} is blank.")
-    elif resp == "ERROR":
-        print(f"Blank check failed for {eprom_name}: {info}")
+    try:
+        ser = find_programmer(eprom)
+        if not ser:
+            return 1
+
+        resp, info = wait_for_response(ser)
+        if resp == "OK":
+            print(f"EPROM {eprom_name} is blank.")
+        elif resp == "ERROR":
+            print(f"Blank check failed for {eprom_name}")
+    finally:
+        consume_response(ser)
+        ser.close()

{firestarter-1.3.2 → firestarter-1.3.3}/firestarter/firmware.py

@@ -11,12 +11,12 @@ import os
 import requests
 
 try:
-    from .serial_comm import find_programmer, wait_for_response, find_comports
+    from .serial_comm import find_programmer, wait_for_response, find_comports, consume_response
     from .config import get_config_value, set_config_value
     from .avr_tool import Avrdude
     from .utils import verbose
 except ImportError:
-    from serial_comm import find_programmer, wait_for_response, find_comports
+    from serial_comm import find_programmer, wait_for_response, find_comports, consume_response
     from config import get_config_value, set_config_value
     from avr_tool import Avrdude
     from utils import verbose
@@ -103,6 +103,7 @@ def firmware_check(port=None):
         )
         return False, ser.portstr, url
     finally:
+        consume_response(ser)
         ser.close()
 
 

{firestarter-1.3.2 → firestarter-1.3.3}/firestarter/hardware.py

@@ -1,8 +1,9 @@
 import time
+
 try:
-    from .serial_comm import find_programmer, wait_for_response
+    from .serial_comm import find_programmer, wait_for_response, consume_response
 except ImportError:
-    from serial_comm import find_programmer, wait_for_response
+    from serial_comm import find_programmer, wait_for_response, consume_response
 
 STATE_READ_VPP = 11
 STATE_READ_VPE = 12
@@ -32,6 +33,7 @@ def hardware():
             print(f"Failed to read hardware revision. {resp}: {version}")
             return 1
     finally:
+        consume_response(ser)
         ser.close()
     return 0
 
@@ -53,15 +55,20 @@ def config(rev=None, r1=None, r2=None):
     ser = find_programmer(data)
     if not ser:
         return 1
-    ser.write("OK".encode("ascii"))
-    r, version = wait_for_response(ser)
-    if r == "OK":
-        print(f"Config: {version}")
-    else:
-        print(r)
-        return 1
+    try:
+        # ser.write("OK".encode("ascii"))
+        r, version = wait_for_response(ser)
+        if r == "OK":
+            print(f"Config: {version}")
+        else:
+            # print(r)
+            return 1
+    finally:
+        consume_response(ser)
+        ser.close()
     return 0
 
+
 def read_vpp(timeout=None):
     """
     Reads the VPP voltage from the programmer.
@@ -94,6 +101,7 @@ def read_voltage(state, timeout=None):
             print(f"Error reading {type} voltage: {info}")
             return 1
         ser.write("OK".encode("ascii"))
+        ser.flush()
 
         if timeout:
             start = time.time()
@@ -104,5 +112,10 @@ def read_voltage(state, timeout=None):
                 print()
                 return 0
             ser.write("OK".encode("ascii"))
+            ser.flush()
+    except Exception as e:
+        print(f"Error while reading {type} voltage: {e}")
     finally:
+        consume_response(ser)
         ser.close()
+    return 1

{firestarter-1.3.2 → firestarter-1.3.3}/firestarter/serial_comm.py

@@ -54,10 +54,8 @@ def check_port(port, data, baud_rate=BAUD_RATE):
         res, msg = wait_for_response(ser)
         while res != "OK":
             if res == "ERROR":
-                raise Exception(msg)
-            if res == "TIMEOUT_ERROR":
                 return None
-            print(f"{res} - {msg}")
+            res, msg = wait_for_response(ser)
 
         if verbose():
             print(f"Programmer: {msg}")
@@ -143,6 +141,54 @@ def find_programmer(data, port=None):
     return None
 
 
+def read_response(ser):
+    while ser.in_waiting > 0:
+        byte_array = ser.readline()
+        res = read_filtered_bytes(byte_array)
+        type = None
+        msg = None
+        try:
+            if res:
+                if "OK:" in res:
+                    msg = res.split("OK:")[-1].strip()
+                    type = "OK"
+                    return
+                elif "INFO:" in res:
+                    msg = res.split("INFO:")[-1].strip()
+                    type = "INFO"
+                    return
+                elif "DEBUG:" in res:
+                    msg = res.split("DEBUG:")[-1].strip()
+                    type = "DEBUG"
+                    return
+                elif "ERROR:" in res:
+                    msg = res.split("ERROR:")[-1].strip()
+                    type = "ERROR"
+                    return
+                elif "WARN:" in res:
+                    msg = res.split("WARN:")[-1].strip()
+                    type = "WARN"
+                    return
+                elif "DATA:" in res:
+                    msg = res.split("DATA:")[-1].strip()
+                    type = "DATA"
+                    return
+                else:
+                    return
+        finally:
+            write_feedback(type, msg)
+            # if type == "ERROR":
+            #     raise Exception(msg)
+            return type, msg
+    return None, None
+
+
+def consume_response(ser):
+    time.sleep(0.1)
+    while read_response(ser)[0] != None:
+        time.sleep(0.3)
+
+
 def wait_for_response(ser):
     """
     Waits for a response from the serial connection.
@@ -155,41 +201,13 @@ def wait_for_response(ser):
     """
     timeout = time.time() + 2  # Set timeout period
     while time.time() < timeout:
-        if ser.in_waiting > 0:
-            byte_array = ser.readline()
-            res = read_filtered_bytes(byte_array)
-            msg = None
-            type = None
-            try:
-                if res:
-                    if "OK:" in res:
-                        msg = res.split("OK:")[-1].strip()
-                        type = "OK"
-                        return
-                    elif "INFO:" in res:
-                        msg = res.split("INFO:")[-1].strip()
-                        write_feedback("INFO", msg)
-                        # return "INFO", msg
-                    elif "ERROR:" in res:
-                        msg = res.split("ERROR:")[-1].strip()
-                        type = "ERROR"
-                        return
-                    elif "WARN:" in res:
-                        msg = res.split("WARN:")[-1].strip()
-                        type = "WARN"
-                        return
-                    elif "DATA:" in res:
-                        msg = res.split("DATA:")[-1].strip()
-                        type = "DATA"
-                        return
-                    timeout = time.time() + 2  # Reset timeout
-            finally:
-                write_feedback(type, msg)
-                if type:
-                    return type, msg
+        type, msg = read_response(ser)
+        if type and type != "INFO" and type != "DEBUG" and type != "WARN":
+            return type, msg
+        # time.sleep(0.1)
     msg = f"Timeout, no response on {ser.portstr}"
-    write_feedback("ERROR", msg)
-    return "TIMEOUT_ERROR", msg
+    write_feedback("Error", msg)
+    raise Exception(msg)
 
 
 def write_feedback(type, msg):
@@ -199,7 +217,7 @@ def write_feedback(type, msg):
     Args:
         msg (str): The feedback message.
     """
-    if msg and (verbose() or type == "ERROR" or type == "WARN"):
+    if msg and (verbose() or (type.upper() == "ERROR" or type == "WARN")):
         print(f"{type}: {msg}")
 
 

{firestarter-1.3.2 → firestarter-1.3.3}/firestarter.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: firestarter
-Version: 1.3.2
+Version: 1.3.3
 Summary: EPROM burner software for the Relatively Universal ROM Programmer
 Author-email: Henrik Olsson <henols@gmail.com>
 Description-Content-Type: text/markdown

{firestarter-1.3.2 → firestarter-1.3.3}/firestarter_test.sh

@@ -12,7 +12,7 @@ if [ ! -d "$TEMP_DIR" ]; then
 fi
 
 # Trap to clean up the temporary files on exit or interrupt
-trap "rm -rf $TEMP_DIR; echo 'Cleaned up temp files'; exit" EXIT
+# trap "rm -rf $TEMP_DIR; echo 'Cleaned up temp files'; exit" EXIT
 
 # Convert TARGET_NAME to uppercase
 EPROM_NAME=$(echo "$EPROM_NAME" | tr '[:lower:]' '[:upper:]')
@@ -104,7 +104,7 @@ then
     exit 1
 fi
 echo
-sleep 0.5
+sleep 1
 echo "---------------------------------"
 echo "VPP"
 echo "---------------------------------"
@@ -115,7 +115,7 @@ then
     exit 1
 fi
 echo
-sleep 0.5
+sleep 1
 echo "---------------------------------"
 echo "VPE"
 echo "---------------------------------"
@@ -126,7 +126,7 @@ then
     exit 1
 fi
 echo
-sleep 0.5
+sleep 1
 
 # ------------------------------ FIRMWARE TESTS ------------------------------
 

{firestarter-1.3.2 → firestarter-1.3.3}/write_test.sh

@@ -15,7 +15,7 @@ fi
 rm -f "$TEMP_DIR"/*
 
 # Trap to clean up the temporary files on exit or interrupt
-trap "rm -rf $TEMP_DIR; echo 'Cleaned up temp files'; exit" EXIT
+# trap "rm -rf $TEMP_DIR; echo 'Cleaned up temp files'; exit" EXIT
 
 # Convert TARGET_NAME to uppercase
 EPROM_NAME=$(echo "$EPROM_NAME" | tr '[:lower:]' '[:upper:]')

firestarter-1.3.2/firestarter/__init__.py

@@ -1 +0,0 @@
-__version__ = "1.3.2"