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