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 intelhex import IntelHex;
  13
  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 info" % sys.argv[0];
  19     print "%s dumpcode $foo.hex [0x$start 0x$stop]" % sys.argv[0];
  20     print "%s dumpdata $foo.hex [0x$start 0x$stop]" % sys.argv[0];
  21     print "%s erase" % sys.argv[0];
  22     print "%s writedata $foo.hex [0x$start 0x$stop]" % sys.argv[0];
  23     print "%s verify $foo.hex [0x$start 0x$stop]" % sys.argv[0];
  24     print "%s peekdata 0x$start [0x$stop]" % sys.argv[0];
  25     print "%s pokedata 0x$adr 0x$val" % sys.argv[0];
  26     print "%s peek 0x$iram" % sys.argv[0];
  27     print "%s poke 0x$iram 0x$val" % sys.argv[0];
  28     print "%s peekcode 0x$start [0x$stop]" % sys.argv[0];
  29     sys.exit();
  30
  31 #Initailize FET and set baud rate
  32 client=GoodFETCC();
  33 client.serInit()
  34
  35 #Connect to target
  36 client.setup();
  37 client.start();
  38
  39 if(sys.argv[1]=="test"):
  40     client.CCtest();
  41 if(sys.argv[1]=="deadtest"):
  42     for i in range(1,10):
  43         print "IDENT as %s" % client.CCidentstr();
  44 if(sys.argv[1]=="dumpcode"):
  45     f = sys.argv[2];
  46     start=0x0000;
  47     stop=0xFFFF;
  48     if(len(sys.argv)>3):
  49         start=int(sys.argv[3],16);
  50     if(len(sys.argv)>4):
  51         stop=int(sys.argv[4],16);
  52
  53     print "Dumping code from %04x to %04x as %s." % (start,stop,f);
  54     h = IntelHex(None);
  55     i=start;
  56     while i<=stop:
  57         h[i]=client.CCpeekcodebyte(i);
  58         if(i%0x100==0):
  59             print "Dumped %04x."%i;
  60         i+=1;
  61     h.write_hex_file(f);
  62 if(sys.argv[1]=="dumpdata"):
  63     f = sys.argv[2];
  64     start=0xE000;
  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 data from %04x to %04x as %s." % (start,stop,f);
  72     h = IntelHex(None);
  73     i=start;
  74     while i<=stop:
  75         h[i]=client.CCpeekdatabyte(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]=="status"):
  81     print "Status: %s" %client.CCstatusstr();
  82 if(sys.argv[1]=="erase"):
  83     print "Status: %s" % client.CCstatusstr();
  84     client.CCchiperase();
  85     print "Status: %s" %client.CCstatusstr();
  86
  87 if(sys.argv[1]=="peekinfo"):
  88     print "Select info flash."
  89     client.CCwr_config(1);
  90     print "Config is %02x" % client.CCrd_config();
  91
  92     start=0x0000;
  93     if(len(sys.argv)>2):
  94         start=int(sys.argv[2],16);
  95     stop=start;
  96     if(len(sys.argv)>3):
  97         stop=int(sys.argv[3],16);
  98     print "Peeking from %04x to %04x." % (start,stop);
  99     while start<=stop:
 100         print "%04x: %02x" % (start,client.CCpeekcodebyte(start));
 101         start=start+1;
 102 if(sys.argv[1]=="poke"):
 103     client.CCpokeirambyte(int(sys.argv[2],16),
 104                           int(sys.argv[3],16));
 105 if(sys.argv[1]=="randtest"):
 106     #Seed RNG
 107     client.CCpokeirambyte(0xBD,0x01); #RNDH=0x01
 108     client.CCpokeirambyte(0xB4,0x04); #ADCCON1=0x04
 109     client.CCpokeirambyte(0xBD,0x01); #RNDH=0x01
 110     client.CCpokeirambyte(0xB4,0x04); #ADCCON1=0x04
 111
 112     #Dump values
 113     for foo in range(1,10):
 114         print "%02x" % client.CCpeekirambyte(0xBD); #RNDH
 115         client.CCpokeirambyte(0xB4,0x04); #ADCCON1=0x04
 116         client.CCreleasecpu();
 117         client.CChaltcpu();
 118     print "%02x" % client.CCpeekdatabyte(0xDF61); #CHIP ID
 119 if(sys.argv[1]=="adctest"):
 120     # ADCTest 0xDF3A 0xDF3B
 121     print "ADC TEST %02x%02x" % (
 122         client.CCpeekdatabyte(0xDF3A),
 123         client.CCpeekdatabyte(0xDF3B));
 124 if(sys.argv[1]=="config"):
 125     print "Config is %02x" % client.CCrd_config();
 126
 127 if(sys.argv[1]=="flash"):
 128      f=sys.argv[2];
 129      start=0;
 130      stop=0xFFFF;
 131      if(len(sys.argv)>3):
 132          start=int(sys.argv[3],16);
 133      if(len(sys.argv)>4):
 134          stop=int(sys.argv[4],16);
 135
 136      h = IntelHex(f);
 137      page = 0x0000;
 138      pagelen = 2048; #2kB pages in 32-bit words
 139      bcount = 0;
 140
 141      print "Wiping Flash."
 142      #Wipe all of flash.
 143      #client.CCchiperase();
 144      #Wipe the RAM buffer for the next flash page.
 145      #client.CCeraseflashbuffer();
 146      for i in h._buf.keys():
 147          while(i>page+pagelen):
 148              if bcount>0:
 149                  client.CCflashpage(page);
 150                  #client.CCeraseflashbuffer();
 151                  bcount=0;
 152                  print "Flashed page at %06x" % page
 153              page+=pagelen;
 154
 155          #Place byte into buffer.
 156          client.CCpokedatabyte(0xF000+i-page,
 157                                h[i]);
 158          bcount+=1;
 159          if(i%0x100==0):
 160                 print "Buffering %04x toward %06x" % (i,page);
 161      #last page
 162      client.CCflashpage(page);
 163      print "Flashed final page at %06x" % page;
 164
 165 if(sys.argv[1]=="lock"):
 166     print "Status: %s" %client.CCstatusstr();
 167     client.CClockchip();
 168     print "Status: %s" %client.CCstatusstr();
 169 if(sys.argv[1]=="flashpage"):
 170     target=0;
 171     if(len(sys.argv)>2):
 172         target=int(sys.argv[2],16);
 173     print "Writing a page of flash from 0xF000 in XDATA"
 174     client.CCflashpage(target);
 175 if(sys.argv[1]=="erasebuffer"):
 176     print "Erasing flash buffer.";
 177     client.CCeraseflashbuffer();
 178
 179 if(sys.argv[1]=="writedata"):
 180     f=sys.argv[2];
 181     start=0;
 182     stop=0xFFFF;
 183     if(len(sys.argv)>3):
 184         start=int(sys.argv[3],16);
 185     if(len(sys.argv)>4):
 186         stop=int(sys.argv[4],16);
 187
 188     h = IntelHex(f);
 189
 190     for i in h._buf.keys():
 191         if(i>=start and i<=stop):
 192             client.CCpokedatabyte(i,h[i]);
 193             if(i%0x100==0):
 194                 print "%04x" % i;
 195 #if(sys.argv[1]=="flashtest"):
 196 #    client.CCflashtest();
 197 if(sys.argv[1]=="peekdata"):
 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.CCpeekdatabyte(start));
 207         start=start+1;
 208 if(sys.argv[1]=="peek"):
 209     start=0x0000;
 210     if(len(sys.argv)>2):
 211         start=int(sys.argv[2],16);
 212     stop=start;
 213     if(len(sys.argv)>3):
 214         stop=int(sys.argv[3],16);
 215     print "Peeking from %04x to %04x." % (start,stop);
 216     while start<=stop:
 217         print "%04x: %02x" % (start,client.CCpeekirambyte(start));
 218         start=start+1;
 219
 220 if(sys.argv[1]=="peekcode"):
 221     start=0x0000;
 222     if(len(sys.argv)>2):
 223         start=int(sys.argv[2],16);
 224     stop=start;
 225     if(len(sys.argv)>3):
 226         stop=int(sys.argv[3],16);
 227     print "Peeking from %04x to %04x." % (start,stop);
 228     while start<=stop:
 229         print "%04x: %02x" % (start,client.CCpeekcodebyte(start));
 230         start=start+1;
 231 if(sys.argv[1]=="pokedata"):
 232     start=0x0000;
 233     val=0x00;
 234     if(len(sys.argv)>2):
 235         start=int(sys.argv[2],16);
 236     if(len(sys.argv)>3):
 237         val=int(sys.argv[3],16);
 238     print "Poking %04x to become %02x." % (start,val);
 239     client.CCpokedatabyte(start,val);
 240
 241 client.stop();