git.rot13.org Git - goodfet/blob - client/goodfet.cc

   1 #!/usr/bin/env python
   2 # GoodFET Chipcon Example
   3 #
   4 # (C) 2009 Travis Goodspeed <travis at radiantmachines.com>
   5 #
   6 # This code is being rewritten and refactored.  You've been warned!
   7
   8 import sys;
   9 import binascii;
  10
  11 from GoodFETCC import GoodFETCC;
  12 from GoodFETConsole import GoodFETConsole;
  13 from intelhex import IntelHex;
  14
  15 if(len(sys.argv)==1):
  16     print "Usage: %s verb [objects]\n" % sys.argv[0];
  17     print "%s erase" % sys.argv[0];
  18     print "%s flash $foo.hex" % sys.argv[0];
  19     print "%s test" % sys.argv[0];
  20     print "%s term" % sys.argv[0];
  21     print "%s info" % sys.argv[0];
  22     print "%s regs" % sys.argv[0];
  23     print "%s dumpcode $foo.hex [0x$start 0x$stop]" % sys.argv[0];
  24     print "%s dumpdata $foo.hex [0x$start 0x$stop]" % sys.argv[0];
  25     print "%s writedata $foo.hex [0x$start 0x$stop]" % sys.argv[0];
  26     print "%s verify $foo.hex [0x$start 0x$stop]" % sys.argv[0];
  27     print "%s peekdata 0x$start [0x$stop]" % sys.argv[0];
  28     print "%s pokedata 0x$adr 0x$val" % sys.argv[0];
  29     print "%s peek 0x$iram" % sys.argv[0];
  30     print "%s poke 0x$iram 0x$val" % sys.argv[0];
  31     print "%s peekcode 0x$start [0x$stop]" % sys.argv[0];
  32     sys.exit();
  33
  34 #Initailize FET and set baud rate
  35 #client=GoodFET.GoodFETCC.GoodFETCC();
  36 client=GoodFETCC();
  37 client.serInit()
  38
  39 #Connect to target
  40 client.setup();
  41 client.start();
  42
  43
  44 if(sys.argv[1]=="explore"):
  45     print "Exploring undefined commands."
  46     print "Status: %s" %client.status();
  47
  48     cmd=0x04; #read status
  49     for foo in range(0,0x5):
  50         client.CCcmd([(0x0F<<3)|(0x00)|0x03,0x09<<3]);
  51         print "Status %02x: %s" % (foo,client.status());
  52     for foo in range(0,3):
  53         print "PC: %04x" % client.CCgetPC();
  54 if(sys.argv[1]=="term"):
  55     GoodFETConsole(client).run();
  56 if(sys.argv[1]=="test"):
  57     client.test();
  58 if(sys.argv[1]=="deadtest"):
  59     for i in range(1,10):
  60         print "IDENT as %s" % client.CCidentstr();
  61 if(sys.argv[1]=="dumpcode"):
  62     f = sys.argv[2];
  63     start=0x0000;
  64     stop=0xFFFF;
  65     if(len(sys.argv)>3):
  66         start=int(sys.argv[3],16);
  67     if(len(sys.argv)>4):
  68         stop=int(sys.argv[4],16);
  69
  70     print "Dumping code from %04x to %04x as %s." % (start,stop,f);
  71     h = IntelHex(None);
  72     i=start;
  73     while i<=stop:
  74         h[i]=client.CCpeekcodebyte(i);
  75         if(i%0x100==0):
  76             print "Dumped %04x."%i;
  77         i+=1;
  78     h.write_hex_file(f);
  79 if(sys.argv[1]=="dumpdata"):
  80     f = sys.argv[2];
  81     start=0xE000;
  82     stop=0xFFFF;
  83     if(len(sys.argv)>3):
  84         start=int(sys.argv[3],16);
  85     if(len(sys.argv)>4):
  86         stop=int(sys.argv[4],16);
  87
  88     print "Dumping data from %04x to %04x as %s." % (start,stop,f);
  89     h = IntelHex(None);
  90     i=start;
  91     while i<=stop:
  92         h[i]=client.CCpeekdatabyte(i);
  93         if(i%0x100==0):
  94             print "Dumped %04x."%i;
  95         i+=1;
  96     h.write_hex_file(f);
  97 if(sys.argv[1]=="status"):
  98     print "Status: %s" %client.status();
  99 if(sys.argv[1]=="info"):
 100     print "Ident   %s" % client.CCidentstr();
 101
 102     try:
 103         print "Freq    %10.3f MHz" % (client.RF_getfreq()/10**6);
 104         print "RSSI    %02x" % client.RF_getrssi();
 105     except:
 106         print "Freq, RSSI, etc unknown.  Install SmartRF7.";
 107     #print "Rate    %10i kbps" % (client.RF_getrate()/1000);
 108     #print "PacketLen %02i bytes" % client.RF_getpacketlen();
 109     #print "SMAC  0x%010x" % client.RF_getsmac();
 110     #print "TMAC  0x%010x" % client.RF_gettmac();
 111
 112 if(sys.argv[1]=="regs"):
 113     client.CMDrs();
 114
 115 if(sys.argv[1]=="erase"):
 116     print "Status: %s" % client.status();
 117     client.CCchiperase();
 118     print "Status: %s" %client.status();
 119
 120 if(sys.argv[1]=="peekinfo"):
 121     print "Select info flash."
 122     client.CCwr_config(1);
 123     print "Config is %02x" % client.CCrd_config();
 124
 125     start=0x0000;
 126     if(len(sys.argv)>2):
 127         start=int(sys.argv[2],16);
 128     stop=start;
 129     if(len(sys.argv)>3):
 130         stop=int(sys.argv[3],16);
 131     print "Peeking from %04x to %04x." % (start,stop);
 132     while start<=stop:
 133         print "%04x: %02x" % (start,client.CCpeekcodebyte(start));
 134         start=start+1;
 135 if(sys.argv[1]=="poke"):
 136     client.CCpokeirambyte(int(sys.argv[2],16),
 137                           int(sys.argv[3],16));
 138 if(sys.argv[1]=="randtest"):
 139     #Seed RNG
 140     client.CCpokeirambyte(0xBD,0x01); #RNDH=0x01
 141     client.CCpokeirambyte(0xB4,0x04); #ADCCON1=0x04
 142     client.CCpokeirambyte(0xBD,0x01); #RNDH=0x01
 143     client.CCpokeirambyte(0xB4,0x04); #ADCCON1=0x04
 144
 145     #Dump values
 146     for foo in range(1,10):
 147         print "%02x" % client.CCpeekirambyte(0xBD); #RNDH
 148         client.CCpokeirambyte(0xB4,0x04); #ADCCON1=0x04
 149         client.CCreleasecpu();
 150         client.CChaltcpu();
 151     print "%02x" % client.CCpeekdatabyte(0xDF61); #CHIP ID
 152 if(sys.argv[1]=="adctest"):
 153     # ADCTest 0xDF3A 0xDF3B
 154     print "ADC TEST %02x%02x" % (
 155         client.CCpeekdatabyte(0xDF3A),
 156         client.CCpeekdatabyte(0xDF3B));
 157 if(sys.argv[1]=="config"):
 158     print "Config is %02x" % client.CCrd_config();
 159
 160 if(sys.argv[1]=="flash"):
 161      f=sys.argv[2];
 162      start=0;
 163      stop=0xFFFF;
 164      if(len(sys.argv)>3):
 165          start=int(sys.argv[3],16);
 166      if(len(sys.argv)>4):
 167          stop=int(sys.argv[4],16);
 168
 169      client.flash(f);
 170 if(sys.argv[1]=="lock"):
 171     print "Status: %s" %client.status();
 172     client.CClockchip();
 173     print "Status: %s" %client.status();
 174 if(sys.argv[1]=="flashpage"):
 175     target=0;
 176     if(len(sys.argv)>2):
 177         target=int(sys.argv[2],16);
 178     print "Writing a page of flash from 0xF000 in XDATA"
 179     client.CCflashpage(target);
 180 if(sys.argv[1]=="erasebuffer"):
 181     print "Erasing flash buffer.";
 182     client.CCeraseflashbuffer();
 183
 184 if(sys.argv[1]=="writedata"):
 185     f=sys.argv[2];
 186     start=0;
 187     stop=0xFFFF;
 188     if(len(sys.argv)>3):
 189         start=int(sys.argv[3],16);
 190     if(len(sys.argv)>4):
 191         stop=int(sys.argv[4],16);
 192
 193     h = IntelHex(f);
 194
 195     for i in h._buf.keys():
 196         if(i>=start and i<=stop):
 197             client.CCpokedatabyte(i,h[i]);
 198             if(i%0x100==0):
 199                 print "%04x" % i;
 200 #if(sys.argv[1]=="flashtest"):
 201 #    client.CCflashtest();
 202 if(sys.argv[1]=="peekdata"):
 203     start=0x0000;
 204     if(len(sys.argv)>2):
 205         start=int(sys.argv[2],16);
 206     stop=start;
 207     if(len(sys.argv)>3):
 208         stop=int(sys.argv[3],16);
 209     print "Peeking from %04x to %04x." % (start,stop);
 210     while start<=stop:
 211         print "%04x: %02x" % (start,client.CCpeekdatabyte(start));
 212         start=start+1;
 213 if(sys.argv[1]=="peek"):
 214     start=0x0000;
 215     if(len(sys.argv)>2):
 216         start=int(sys.argv[2],16);
 217     stop=start;
 218     if(len(sys.argv)>3):
 219         stop=int(sys.argv[3],16);
 220     print "Peeking from %04x to %04x." % (start,stop);
 221     while start<=stop:
 222         print "%04x: %02x" % (start,client.CCpeekirambyte(start));
 223         start=start+1;
 224 if(sys.argv[1]=="verify"):
 225     f=sys.argv[2];
 226     start=0;
 227     stop=0xFFFF;
 228     if(len(sys.argv)>3):
 229         start=int(sys.argv[3],16);
 230     if(len(sys.argv)>4):
 231         stop=int(sys.argv[4],16);
 232
 233     h = IntelHex(f);
 234     for i in h._buf.keys():
 235         if(i>=start and i<stop):
 236             peek=client.CCpeekcodebyte(i)
 237             if(h[i]!=peek):
 238                 print "ERROR at %04x, found %02x not %02x"%(i,peek,h[i]);
 239             if(i%0x100==0):
 240                 print "%04x" % i;
 241 if(sys.argv[1]=="peekcode"):
 242     start=0x0000;
 243     if(len(sys.argv)>2):
 244         start=int(sys.argv[2],16);
 245     stop=start;
 246     if(len(sys.argv)>3):
 247         stop=int(sys.argv[3],16);
 248     print "Peeking from %04x to %04x." % (start,stop);
 249     while start<=stop:
 250         print "%04x: %02x" % (start,client.CCpeekcodebyte(start));
 251         start=start+1;
 252 if(sys.argv[1]=="pokedata"):
 253     start=0x0000;
 254     val=0x00;
 255     if(len(sys.argv)>2):
 256         start=int(sys.argv[2],16);
 257     if(len(sys.argv)>3):
 258         val=int(sys.argv[3],16);
 259     print "Poking %04x to become %02x." % (start,val);
 260     client.CCpokedatabyte(start,val);
 261
 262 client.stop();