X-Git-Url: http://git.rot13.org/?p=goodfet;a=blobdiff_plain;f=client%2FGoodFET.py;h=f80a46ded7691923542e5ec8be4a6ae67788d253;hp=231106203b0fda516e2f50b5a2b63aabddc0ab1d;hb=da2dad53f6596dbb3049ad14044b9b754dbf4245;hpb=c5a56b2284ba0cd554cb44ed221eb91489607b9e diff --git a/client/GoodFET.py b/client/GoodFET.py index 2311062..f80a46d 100755 --- a/client/GoodFET.py +++ b/client/GoodFET.py @@ -6,10 +6,11 @@ # This code is ugly as sin, for bootstrapping the firmware only. # Rewrite cleanly as soon as is convenient. -import sys, time, string, cStringIO, struct, glob, serial +import sys, time, string, cStringIO, struct, glob, serial, os; class GoodFET: + """GoodFET Client Library""" def __init__(self, *args, **kargs): self.data=[0]; def timeout(self): @@ -17,6 +18,10 @@ class GoodFET: def serInit(self, port=None): """Open the serial port""" + if port is None and os.environ.get("GOODFET")!=None: + glob_list = glob.glob(os.environ.get("GOODFET")); + if len(glob_list) > 0: + port = glob_list[0]; if port is None: glob_list = glob.glob("/dev/tty.usbserial*"); if len(glob_list) > 0: @@ -28,13 +33,7 @@ class GoodFET: self.serialport = serial.Serial( port, - #300, - #2400, - #4800, #9600, - #19200, - #38400, - #57600, 115200, parity = serial.PARITY_NONE ) @@ -48,8 +47,8 @@ class GoodFET: self.readcmd(); #Read the first command. if(self.verb!=0x7F): print "Verb %02x is wrong. Incorrect firmware?" % self.verb; - print "Connected." - def writecmd(self, app, verb, count, data): + #print "Connected." + def writecmd(self, app, verb, count=0, data=[], blocks=1): """Write a command and some data to the GoodFET.""" self.serialport.write(chr(app)); self.serialport.write(chr(verb)); @@ -58,14 +57,15 @@ class GoodFET: if count!=0: for d in data: self.serialport.write(chr(d)); - self.readcmd(); #Uncomment this later, to ensure a response. - def readcmd(self): + + self.readcmd(blocks); #Uncomment this later, to ensure a response. + def readcmd(self,blocks=1): """Read a reply from the GoodFET.""" self.app=ord(self.serialport.read(1)); self.verb=ord(self.serialport.read(1)); self.count=ord(self.serialport.read(1)); - if self.count>0: - self.data=self.serialport.read(self.count); + self.data=self.serialport.read(self.count*blocks); + return self.data; #print "READ %02x %02x %02x " % (self.app, self.verb, self.count); #Monitor stuff @@ -141,7 +141,7 @@ class GoodFET: """Self-test several functions through the monitor.""" print "Performing monitor self-test."; - if self.peekword(0x0c00)!=0x0c04: + if self.peekword(0x0c00)!=0x0c04 and self.peekword(0x0c00)!=0x0c06: print "ERROR Fetched wrong value from 0x0c04."; self.pokebyte(0x0021,0); #Drop LED if self.peekbyte(0x0021)!=0: @@ -180,6 +180,11 @@ class GoodFET: 0xC22013: "MX25L4005", 0x204011: "M45PE10" }; + JEDECsizes={0x14: 0x100000, + 0x13: 0x080000, + 0x12: 0x040000, + 0x11: 0x020000} + JEDECsize=0; def SPIjedec(self): """Grab an SPI Flash ROM's JEDEC bytes.""" data=[0x9f, 0, 0, 0]; @@ -188,6 +193,9 @@ class GoodFET: self.JEDECmanufacturer=ord(data[1]); self.JEDECtype=ord(data[2]); self.JEDECcapacity=ord(data[3]); + self.JEDECsize=self.JEDECsizes.get(self.JEDECcapacity); + if self.JEDECsize==None: + self.JEDECsize=0; self.JEDECdevice=(ord(data[1])<<16)+(ord(data[2])<<8)+ord(data[3]); return data; def SPIpeek(self,adr): @@ -199,13 +207,14 @@ class GoodFET: 0]; self.SPItrans(data); return ord(self.data[4]); - def SPIpeekblock(self,adr): - """Grab a byte from an SPI Flash ROM.""" + def SPIpeekblock(self,adr,blocks=1): + """Grab a few block from an SPI Flash ROM. Block size is unknown""" data=[(adr&0xFF0000)>>16, (adr&0xFF00)>>8, - adr&0xFF]; + adr&0xFF, + blocks]; - self.writecmd(0x01,0x02,3,data); + self.writecmd(0x01,0x02,4,data,blocks); return self.data; def SPIpokebyte(self,adr,val): @@ -243,16 +252,32 @@ class GoodFET: if device==0: device="???" return "%s %s" % (man,device); + def MSP430setup(self): """Move the FET into the MSP430 JTAG application.""" print "Initializing MSP430."; self.writecmd(0x11,0x10,0,self.data); - - + def I2Csetup(self): + """Move the FET into the I2C application.""" + self.writecmd(0x02,0x10,0,self.data); #SPI/SETUP + def I2Cstart(self): + """Start an I2C transaction.""" + self.writecmd(0x02,0x20,0,self.data); #SPI/SETUP + def I2Cstop(self): + """Stop an I2C transaction.""" + self.writecmd(0x02,0x21,0,self.data); #SPI/SETUP + def I2Cread(self,len=1): + """Read len bytes by I2C.""" + self.writecmd(0x02,0x00,1,[len]); #SPI/SETUP + return self.data; + def I2Cwrite(self,bytes): + """Write bytes by I2C.""" + self.writecmd(0x02,0x01,len(bytes),bytes); #SPI/SETUP + return ord(self.data[0]); def CCsetup(self): """Move the FET into the CC2430/CC2530 application.""" - print "Initializing Chipcon."; + #print "Initializing Chipcon."; self.writecmd(0x30,0x10,0,self.data); def CCrd_config(self): """Read the config register of a Chipcon.""" @@ -351,7 +376,7 @@ class GoodFET: #print "Status: %s." % self.CCstatusstr(); self.CCreleasecpu(); self.CChaltcpu(); - print "Status: %s." % self.CCstatusstr(); + #print "Status: %s." % self.CCstatusstr(); def CCstop(self): """Stop debugging.""" @@ -436,7 +461,7 @@ class GoodFET: def CCtest(self): self.CCreleasecpu(); self.CChaltcpu(); - print "Status: %s" % self.CCstatusstr(); + #print "Status: %s" % self.CCstatusstr(); #Grab ident three times, should be equal. ident1=self.CCident(); @@ -447,26 +472,22 @@ class GoodFET: print "%04x, %04x, %04x" % (ident1, ident2, ident3); #Single step, printing PC. - #print "Tracing execution at startup." + print "Tracing execution at startup." for i in range(1,15): pc=self.CCgetPC(); byte=self.CCpeekcodebyte(i); - print "PC=%04x, %02x" % (pc, byte); + #print "PC=%04x, %02x" % (pc, byte); self.CCstep_instr(); - #print "Verifying that debugging a NOP doesn't affect the PC." + print "Verifying that debugging a NOP doesn't affect the PC." for i in range(1,15): pc=self.CCgetPC(); self.CCdebuginstr([0x00]); if(pc!=self.CCgetPC()): print "ERROR: PC changed during CCdebuginstr([NOP])!"; - for i in range(0xE500,0xE600): - byte=self.CCpeekdatabyte(i); - print "data %04x: %02x" % (i,byte); - self.CCpokedatabyte(i,i&0xFF); - byte=self.CCpeekdatabyte(i); - print "data %04x: %02x" % (i,byte); - print "Status: %s." % self.CCstatusstr(); + + + #print "Status: %s." % self.CCstatusstr(); #Exit debugger self.CCstop(); print "Done.";