#
# 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;
import sqlite3;
fmt = ("B", "<H", None, "<L")
"values(?,?,?,?,?);", (
adr,name,memory,size,comment));
#print "Set %s=%s." % (name,adr);
+class GoodFETbtser:
+ """py-bluez class for emulating py-serial."""
+ def __init__(self,btaddr):
+ import bluetooth;
+ if btaddr==None or btaddr=="none" or btaddr=="bluetooth":
+ print "performing inquiry..."
+ nearby_devices = bluetooth.discover_devices(lookup_names = True)
+ print "found %d devices" % len(nearby_devices)
+ for addr, name in nearby_devices:
+ print " %s - '%s'" % (addr, name)
+ #TODO switch to wildcards.
+ if name=='FireFly-A6BD':
+ btaddr=addr;
+ if name=='RN42-A94A':
+ btaddr=addr;
+
+ print "Please set $GOODFET to the address of your device.";
+ sys.exit();
+ print "Identified GoodFET at %s" % btaddr;
+ # Manually use the portnumber.
+ port=1;
+
+ print "Connecting to %s on port %i." % (btaddr, port);
+ sock=bluetooth.BluetoothSocket(bluetooth.RFCOMM);
+ self.sock=sock;
+ sock.connect((btaddr,port));
+ sock.settimeout(10); #IMPORTANT Must be patient.
+
+ ##This is what we'd do for a normal reset.
+ #str="";
+ #while not str.endswith("goodfet.sf.net/"):
+ # str=self.read(64);
+ # print str;
+
+ # Instead, just return and hope for the best.
+ return;
+
+ def write(self,msg):
+ """Send traffic."""
+ import time;
+ self.sock.send(msg);
+ #time.sleep(0.1);
+ return;
+ def read(self,length):
+ """Read traffic."""
+ data="";
+ while len(data)<length:
+ data=data+self.sock.recv(length-len(data));
+ return data;
class GoodFET:
"""GoodFET Client Library"""
def timeout(self):
print "timeout\n";
def serInit(self, port=None, timeout=2, attemptlimit=None):
+ """Open a serial port of some kind."""
+ import re;
+
+ if port==None:
+ port=os.environ.get("GOODFET");
+ if port=="bluetooth" or (port is not None and re.match("..:..:..:..:..:..",port)):
+ self.btInit(port,2,attemptlimit);
+ else:
+ self.pyserInit(port,timeout,attemptlimit);
+ def btInit(self, port, timeout, attemptlimit):
+ """Open a bluetooth port.""";
+ #self.verbose=True; #For debugging BT.
+ self.serialport=GoodFETbtser(port);
+
+ def pyserInit(self, port, timeout, attemptlimit):
"""Open the serial port"""
# Make timeout None to wait forever, 0 for non-blocking mode.
+ import serial;
+ fixserial=False;
if os.name=='nt' and sys.version.find('64 bit')!=-1:
print "WARNING: PySerial requires a 32-bit Python build in Windows.";
a=1;
baud=115200;
- if(os.environ.get("platform")=='arduino'):
+ if(os.environ.get("platform")=='arduino' or os.environ.get("board")=='arduino'):
baud=19200; #Slower, for now.
self.serialport = serial.Serial(
port,
#print "'%s'!=\n'%s'" % (self.data,"http://goodfet.sf.net/");
if attemptlimit is not None and attempts >= attemptlimit:
return
- elif attempts>2:
- print "Resyncing.";
- self.serialport.flushInput()
- self.serialport.flushOutput()
+ elif attempts==2 and os.environ.get("board")!='telosb':
+ print "See the GoodFET FAQ about missing info flash.";
+ self.serialport.setTimeout(0.2);
+ #self.serialport.flushInput()
+ #self.serialport.flushOutput()
#TelosB reset, prefer software to I2C SPST Switch.
- if(os.environ.get("platform")=='telosb'):
+ if (os.environ.get("board")=='telosb'):
#print "TelosB Reset";
self.telosBReset();
+ elif (os.environ.get("board")=='z1'):
+ self.bslResetZ1(invokeBSL=0);
+ elif (os.environ.get("board")=='apimote1') or (os.environ.get("board")=='apimote'):
+ #Explicitly set RTS and DTR to halt board.
+ self.serialport.setRTS(1);
+ self.serialport.setDTR(1);
+ #RTS pin, not DTR is used for reset.
+ self.serialport.setRTS(0);
+ #print "Resetting Apimote not yet tested.";
else:
#Explicitly set RTS and DTR to halt board.
self.serialport.setRTS(1);
attempts=attempts+1;
self.readcmd(); #Read the first command.
#print "Got %02x,%02x:'%s'" % (self.app,self.verb,self.data);
+ if self.verb!=0x7f:
+ #Retry again. This usually times out, but helps connect.
+ self.readcmd();
+ #print "Retry got %02x,%02x:'%s'" % (self.app,self.verb,self.data);
#Here we have a connection, but maybe not a good one.
#print "We have a connection."
connected=1;
olds=self.infostring();
clocking=self.monitorclocking();
- #if(os.environ.get("platform")!='arduino'):
for foo in range(1,30):
if not self.monitorecho():
if self.verbose:
print "Comm error on %i try, resyncing out of %s." % (foo,
clocking);
- connected=0;
- break;
+ connected=0;
+ break;
if self.verbose: print "Connected after %02i attempts." % attempts;
self.mon_connected();
self.serialport.setTimeout(12);
self.telosI2CWriteByte( 0x90 | (addr << 1) )
self.telosI2CWriteByte( cmdbyte )
self.telosI2CStop()
+ def bslResetZ1(self, invokeBSL=0):
+ '''
+ Applies BSL entry sequence on RST/NMI and TEST/VPP pins
+ Parameters:
+ invokeBSL = 1: complete sequence
+ invokeBSL = 0: only RST/NMI pin accessed
+
+ By now only BSL mode is accessed
+ '''
+
+ #if DEBUG > 1: sys.stderr.write("* bslReset(invokeBSL=%s)\n" % invokeBSL)
+ if invokeBSL:
+ #sys.stderr.write("in Z1 bsl reset...\n")
+ time.sleep(0.1)
+ self.writepicROM(0xFF, 0xFF)
+ time.sleep(0.1)
+ #sys.stderr.write("z1 bsl reset done...\n")
+ else:
+ #sys.stderr.write("in Z1 reset...\n")
+ time.sleep(0.1)
+ self.writepicROM(0xFF, 0xFE)
+ time.sleep(0.1)
+ #sys.stderr.write("z1 reset done...\n")
+ def writepicROM(self, address, data):
+ ''' Writes data to @address'''
+ for i in range(7,-1,-1):
+ self.picROMclock((address >> i) & 0x01)
+ self.picROMclock(0)
+ recbuf = 0
+ for i in range(7,-1,-1):
+ s = ((data >> i) & 0x01)
+ #print s
+ if i < 1:
+ r = not self.picROMclock(s, True)
+ else:
+ r = not self.picROMclock(s)
+ recbuf = (recbuf << 1) + r
+
+ self.picROMclock(0, True)
+ #k = 1
+ #while not self.serial.getCTS():
+ # pass
+ #time.sleep(0.1)
+ return recbuf
+ def readpicROM(self, address):
+ ''' reads a byte from @address'''
+ for i in range(7,-1,-1):
+ self.picROMclock((address >> i) & 0x01)
+ self.picROMclock(1)
+ recbuf = 0
+ r = 0
+ for i in range(7,-1,-1):
+ r = self.picROMclock(0)
+ recbuf = (recbuf << 1) + r
+ self.picROMclock(r)
+ #time.sleep(0.1)
+ return recbuf
+
+ #This seems more reliable when slowed.
+ def picROMclock(self, masterout, slow = True):
+ #print "setting masterout to "+str(masterout)
+ self.serialport.setRTS(masterout)
+ self.serialport.setDTR(1)
+ #time.sleep(0.02)
+ self.serialport.setDTR(0)
+ if slow:
+ time.sleep(0.02)
+ return self.serialport.getCTS()
+
+ def picROMfastclock(self, masterout):
+ #print "setting masterout to "+str(masterout)
+ self.serialport.setRTS(masterout)
+ self.serialport.setDTR(1)
+ self.serialport.setDTR(0)
+ time.sleep(0.02)
+ return self.serialport.getCTS()
def telosBReset(self,invokeBSL=0):
# "BSL entry sequence at dedicated JTAG pins"
+(ord(self.serialport.read(1))<<8)
);
- #if self.verbose:
- #print "Rx: ( 0x%02x, 0x%02x, 0x%04x )" % ( self.app, self.verb, self.count )
+ if self.verbose:
+ print "Rx: ( 0x%02x, 0x%02x, 0x%04x )" % ( self.app, self.verb, self.count )
#Debugging string; print, but wait.
if self.app==0xFF:
self.writecmd(0,0x03,3,self.data);
return ord(self.data[0]);
def poke16(self,address,value):
+ """Set a word of memory by the monitor."""
+ self.MONpoke16(address,value);
+ def MONpoke16(self,address,value):
"""Set a word of memory by the monitor."""
self.pokebyte(address,value&0xFF);
self.pokebyte(address,(value>>8)&0xFF);
data="The quick brown fox jumped over the lazy dog.";
self.writecmd(self.MONITORAPP,0x81,len(data),data);
if self.data!=data:
- print "Comm error recognized by monitorecho(), got:\n%s" % self.data;
+ if self.verbose:
+ print "Comm error recognized by monitorecho(), got:\n%s" % self.data;
return 0;
return 1;
print "Clocked at %s" % self.monitorclocking();
return 1;
+ def testleds(self):
+ print "Flashing LEDs"
+ self.writecmd(self.MONITORAPP,0xD0,0,"")
+ try:
+ print "Flashed %d LED." % ord(self.data)
+ except:
+ print "Unable to process response:", self.data
+ return 1
+
def monitor_list_apps(self, full=False):
self.monitor_info()
old_value = self.besilent
self.MONpoke16(0x56, clock);
def monitorgetclock(self):
"""Get the clocking value."""
- if(os.environ.get("platform")=='arduino'):
+ if(os.environ.get("platform")=='arduino' or os.environ.get("board")=='arduino'):
return 0xDEAD;
#Check for MSP430 before peeking this.
return self.MONpeek16(0x56);
# every client.
def infostring(self):
- if(os.environ.get("platform")=='arduino'):
+ if(os.environ.get("platform")=='arduino' or os.environ.get("board")=='arduino'):
return "Arduino";
else:
a=self.MONpeek8(0xff0);
projects / goodfet / blobdiff