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 peekcode 0x$start [0x$stop]" % sys.argv[0];
  27     sys.exit();
  28
  29 #Initailize FET and set baud rate
  30 client=GoodFETCC();
  31 client.serInit()
  32
  33 #Connect to target
  34 client.CCsetup();
  35 client.CCstart();
  36
  37 if(sys.argv[1]=="test"):
  38     client.CCtest();
  39 if(sys.argv[1]=="deadtest"):
  40     for i in range(1,10):
  41         print "IDENT as %s" % client.CCidentstr();
  42 if(sys.argv[1]=="dumpcode"):
  43     f = sys.argv[2];
  44     start=0x0000;
  45     stop=0xFFFF;
  46     if(len(sys.argv)>3):
  47         start=int(sys.argv[3],16);
  48     if(len(sys.argv)>4):
  49         stop=int(sys.argv[4],16);
  50
  51     print "Dumping code from %04x to %04x as %s." % (start,stop,f);
  52     h = IntelHex(None);
  53     i=start;
  54     while i<=stop:
  55         h[i]=client.CCpeekcodebyte(i);
  56         if(i%0x100==0):
  57             print "Dumped %04x."%i;
  58         i+=1;
  59     h.write_hex_file(f);
  60 if(sys.argv[1]=="dumpdata"):
  61     f = sys.argv[2];
  62     start=0xE000;
  63     stop=0xFFFF;
  64     if(len(sys.argv)>3):
  65         start=int(sys.argv[3],16);
  66     if(len(sys.argv)>4):
  67         stop=int(sys.argv[4],16);
  68
  69     print "Dumping data from %04x to %04x as %s." % (start,stop,f);
  70     h = IntelHex(None);
  71     i=start;
  72     while i<=stop:
  73         h[i]=client.CCpeekdatabyte(i);
  74         if(i%0x100==0):
  75             print "Dumped %04x."%i;
  76         i+=1;
  77     h.write_hex_file(f);
  78 if(sys.argv[1]=="status"):
  79     print "Status: %s" %client.CCstatusstr();
  80 if(sys.argv[1]=="erase"):
  81     print "Status: %s" % client.CCstatusstr();
  82     client.CCchiperase();
  83     print "Status: %s" %client.CCstatusstr();
  84
  85 if(sys.argv[1]=="peekinfo"):
  86     print "Select info flash."
  87     client.CCwr_config(1);
  88     print "Config is %02x" % client.CCrd_config();
  89
  90     start=0x0000;
  91     if(len(sys.argv)>2):
  92         start=int(sys.argv[2],16);
  93     stop=start;
  94     if(len(sys.argv)>3):
  95         stop=int(sys.argv[3],16);
  96     print "Peeking from %04x to %04x." % (start,stop);
  97     while start<=stop:
  98         print "%04x: %02x" % (start,client.CCpeekcodebyte(start));
  99         start=start+1;
 100
 101 if(sys.argv[1]=="config"):
 102     print "Config is %02x" % client.CCrd_config();
 103
 104 if(sys.argv[1]=="flash"):
 105      f=sys.argv[2];
 106      start=0;
 107      stop=0xFFFF;
 108      if(len(sys.argv)>3):
 109          start=int(sys.argv[3],16);
 110      if(len(sys.argv)>4):
 111          stop=int(sys.argv[4],16);
 112
 113      h = IntelHex(f);
 114      page = 0x0000;
 115      pagelen = 2048; #2kB pages in 32-bit words
 116      bcount = 0;
 117
 118      print "Wiping Flash."
 119      #Wipe all of flash.
 120      client.CCchiperase();
 121      #Wipe the RAM buffer for the next flash page.
 122      client.CCeraseflashbuffer();
 123      for i in h._buf.keys():
 124          while(i>page+pagelen):
 125              if bcount>0:
 126                  client.CCflashpage(page);
 127                  client.CCeraseflashbuffer();
 128                  bcount=0;
 129                  print "Flashed page at %06x" % page
 130              page+=pagelen;
 131
 132          #Place byte into buffer.
 133          client.CCpokedatabyte(0xF000+i-page,
 134                                h[i]);
 135          bcount+=1;
 136          if(i%0x100==0):
 137                 print "Buffering %04x to %06x" % (i,page);
 138      #last page
 139      client.CCflashpage(page);
 140      print "Flashed final page at %06x" % page;
 141
 142 if(sys.argv[1]=="lock"):
 143     print "Status: %s" %client.CCstatusstr();
 144     client.CClockchip();
 145     print "Status: %s" %client.CCstatusstr();
 146 if(sys.argv[1]=="flashpage"):
 147     target=0;
 148     if(len(sys.argv)>2):
 149         target=int(sys.argv[2],16);
 150     print "Writing a page of flash from 0xF000 in XDATA"
 151     client.CCflashpage(target);
 152 if(sys.argv[1]=="erasebuffer"):
 153     print "Erasing flash buffer.";
 154     client.CCeraseflashbuffer();
 155
 156 if(sys.argv[1]=="writedata"):
 157     f=sys.argv[2];
 158     start=0;
 159     stop=0xFFFF;
 160     if(len(sys.argv)>3):
 161         start=int(sys.argv[3],16);
 162     if(len(sys.argv)>4):
 163         stop=int(sys.argv[4],16);
 164
 165     h = IntelHex(f);
 166
 167     for i in h._buf.keys():
 168         if(i>=start and i<=stop):
 169             client.CCpokedatabyte(i,h[i]);
 170             if(i%0x100==0):
 171                 print "%04x" % i;
 172 #if(sys.argv[1]=="flashtest"):
 173 #    client.CCflashtest();
 174 if(sys.argv[1]=="peekdata"):
 175     start=0x0000;
 176     if(len(sys.argv)>2):
 177         start=int(sys.argv[2],16);
 178     stop=start;
 179     if(len(sys.argv)>3):
 180         stop=int(sys.argv[3],16);
 181     print "Peeking from %04x to %04x." % (start,stop);
 182     while start<=stop:
 183         print "%04x: %02x" % (start,client.CCpeekdatabyte(start));
 184         start=start+1;
 185 if(sys.argv[1]=="peekcode"):
 186     start=0x0000;
 187     if(len(sys.argv)>2):
 188         start=int(sys.argv[2],16);
 189     stop=start;
 190     if(len(sys.argv)>3):
 191         stop=int(sys.argv[3],16);
 192     print "Peeking from %04x to %04x." % (start,stop);
 193     while start<=stop:
 194         print "%04x: %02x" % (start,client.CCpeekcodebyte(start));
 195         start=start+1;
 196 if(sys.argv[1]=="pokedata"):
 197     start=0x0000;
 198     val=0x00;
 199     if(len(sys.argv)>2):
 200         start=int(sys.argv[2],16);
 201     if(len(sys.argv)>3):
 202         val=int(sys.argv[3],16);
 203     print "Poking %04x to become %02x." % (start,val);
 204     client.CCpokedatabyte(start,val);
 205
 206 client.CCstop();