more changes to TX test code

[goodfet] / client / GoodFETAVR.py

diff --git a/client/GoodFETAVR.py b/client/GoodFETAVR.py
index f8189bf..8a9adc8 100644 (file)
--- a/client/GoodFETAVR.py
+++ b/client/GoodFETAVR.py
@@ -5,7 +5,7 @@
 #
 # This code is being rewritten and refactored.  You've been warned!
 
-import sys, time, string, cStringIO, struct, glob, serial, os;
+import sys, time, string, cStringIO, struct, glob, os;
 
 from GoodFET import GoodFET;
 
@@ -57,6 +57,7 @@ class GoodFETAVR(GoodFET):
         0x9406: "ATmega168",
         0x9405: "ATmega169",
         0x9502: "ATmega32",
+        0x958a: "ATmega32U2", #TODO add the other U series.
         0x9501: "ATmega323",
         0x9503: "ATmega325",
         0x9504: "ATmega3250",
@@ -80,6 +81,9 @@ class GoodFETAVR(GoodFET):
         0x9302: "ATmega85",
         0x9305: "ATmega83",
         0x9601: "ATmega603",
+        
+        #These are missing from the Fenceline DB.
+        0x960a: "ATmega644P",
         };
     
     def setup(self):
@@ -116,7 +120,8 @@ class GoodFETAVR(GoodFET):
         """Read the target's lockbits."""
         self.writecmd(self.AVRAPP,0x92,1,[bits]);
         return self.lockbits();
-    
+    def lock(self):
+        self.setlockbits(0xFC);
     def eeprompeek(self, adr):
         """Read a byte of the target's EEPROM."""
         self.writecmd(self.AVRAPP,0x81 ,2,