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