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