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