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
  16 def printpacket(packet):
  17     s="";
  18     i=0;
  19     for foo in packet:
  20         i=i+1;
  21         s="%s %02x" % (s,foo);
  22     print "# %s" %s;
  23
  24 simplepacketcount=0;
  25 def handlesimplicitipacket(packet):
  26     s="";
  27     i=0;
  28     global simplepacketcount;
  29     simplepacketcount=simplepacketcount+1;
  30
  31     len=packet[0];
  32     if len<12: return;
  33
  34     dst=[packet[1],
  35              packet[2],
  36              packet[3],
  37              packet[4]];
  38     src=[packet[5],
  39              packet[6],
  40              packet[7],
  41              packet[8]];
  42     port=packet[9];
  43     info=packet[10];
  44     seq=packet[11];
  45     #payload begins at byte 10.
  46
  47     if packet[len+2]&0x80==0:
  48         print "# Dropped broken packet.";
  49     elif port==0x20:
  50         #data packet
  51         counter=packet[11];
  52         button=packet[12];
  53         x=packet[13];
  54         y=packet[14];
  55         z=packet[15];
  56         print "%09i %03i %4i %4i %4i" % (simplepacketcount,button,x,y,z);
  57         sys.stdout.flush();
  58     elif port==0x02:
  59         #Link request.  Gotta send a proper reply to get data.
  60         tid=packet[13];
  61         #14 ff ff ff ff 3c b7 e3 98
  62         #02 03 c9
  63         #01 97
  64         #ef be ad de 3d 00 02
  65         reply=[0x10,
  66                src[0], src[1], src[2], src[3],
  67                0x78,0x56,0x34,0x10, #my address.
  68                port, 0x21, seq,
  69                0x81, tid,         #reply, tid
  70
  71                0x20,0x00,0xad,0xde, #link token
  72                0x00];               #no security
  73         #printpacket(reply);
  74         print "#FIXME FAST: repeatedly broadcasting ACK to catch LINK on the next attempt.";
  75         for foo in range(1,50):
  76             client.RF_txpacket(reply);
  77
  78         pass;
  79     elif port==0x03:
  80         #print "Join request.";
  81         #printpacket(packet);
  82         if packet[12]!=1:
  83             print "Not a join request.  WTF?";
  84             return;
  85         tid=packet[13];
  86         reply=[0x12, #reply is one byte shorter
  87                src[0], src[1], src[2], src[3],
  88                0x78,0x56,0x34,0x10, #my address.
  89                port, 0x21, seq,
  90                0x81, tid,         #reply, tid
  91
  92                0xef,0xbe,0xad,0xde, #Join token
  93                0x00];             #no security
  94         #printpacket(reply);
  95         print "#FIXME FAST: repeatedly broadcasting ACK to catch JOIN on the next attempt.";
  96         #printpacket(reply);
  97         for foo in range(1,50):
  98             client.RF_txpacket(reply);
  99
 100
 101     elif port==0x04:
 102         print "Security request.";
 103     elif port==0x05:
 104         print "Frequency request.";
 105     elif port==0x06:
 106         print "Management request.";
 107     else:
 108         print "Unknown Port %02x" %port;
 109
 110 if(len(sys.argv)==1):
 111     print "Usage: %s verb [objects]\n" % sys.argv[0];
 112     print "%s erase" % sys.argv[0];
 113     print "%s flash $foo.hex" % sys.argv[0];
 114     print "%s test" % sys.argv[0];
 115     print "%s term" % sys.argv[0];
 116     print "%s info" % sys.argv[0];
 117     print "%s infotest" % sys.argv[0];
 118     print "%s halt"  % sys.argv[0];
 119     print "%s regs" % sys.argv[0];
 120     print "%s dumpcode $foo.hex [0x$start 0x$stop]" % sys.argv[0];
 121     print "%s dumpdata $foo.hex [0x$start 0x$stop]" % sys.argv[0];
 122     print "%s writedata $foo.hex [0x$start 0x$stop]" % sys.argv[0];
 123     print "%s verify $foo.hex [0x$start 0x$stop]" % sys.argv[0];
 124     print "%s peekdata 0x$start [0x$stop]" % sys.argv[0];
 125     print "%s pokedata 0x$adr 0x$val" % sys.argv[0];
 126     print "%s peek 0x$iram" % sys.argv[0];
 127     print "%s poke 0x$iram 0x$val" % sys.argv[0];
 128     print "%s peekcode 0x$start [0x$stop]" % sys.argv[0];
 129     print "\n"
 130     print "%s rssi [freq]\n\tGraphs signal strength on [freq] Hz." % sys.argv[0];
 131     print "%s carrier [freq]\n\tHolds a carrier on [freq] Hz." % sys.argv[0];
 132     print "%s reflex [freq]\n\tJams on [freq] Hz." % sys.argv[0];
 133     print "%s sniffsimpliciti [us|eu|lf]\n\tSniffs SimpliciTI packets." % sys.argv[0];
 134     print "%s sniffdash7 [lf]\n\tSniffs Dash7. (untested)" % sys.argv[0];
 135
 136
 137     sys.exit();
 138
 139 #Initailize FET and set baud rate
 140 #client=GoodFET.GoodFETCC.GoodFETCC();
 141 client=GoodFETCC();
 142 client.serInit()
 143
 144 #Connect to target
 145 client.setup();
 146 client.start();
 147
 148
 149
 150 if(sys.argv[1]=="carrier"):
 151     if len(sys.argv)>2:
 152         client.RF_setfreq(eval(sys.argv[2]));
 153     client.RF_carrier();
 154     while(1):
 155         time.sleep(1);
 156
 157 if(sys.argv[1]=="reflex"):
 158     client.CC1110_crystal();
 159     client.RF_idle();
 160
 161     client.config_simpliciti();
 162
 163     threshold=100;
 164     if len(sys.argv)>2:
 165         client.RF_setfreq(eval(sys.argv[2]));
 166     print "Listening on %f MHz." % (client.RF_getfreq()/10**6);
 167     print "Jamming if RSSI>=%i" % threshold;
 168
 169     client.pokebyte(0xFE00,threshold,"xdata"); #Write threshold to shellcode.
 170     client.shellcodefile("reflex.ihx");
 171     rssi=0;
 172     while 1:
 173         while(0==client.ishalted()):
 174             rssi=0;
 175         rssi=client.peek8(0xFE00,"xdata");
 176         print "Activated jamming with RSSI of %i, going again for another packet." % rssi;
 177         #client.CCdebuginstr([0x02, 0xf0, 0x00]); #ljmp 0xF000
 178         client.resume();
 179
 180
 181 if(sys.argv[1]=="rssi"):
 182     client.CC1110_crystal();
 183     client.RF_idle();
 184
 185     client.config_simpliciti();
 186
 187     if len(sys.argv)>2:
 188         client.RF_setfreq(eval(sys.argv[2]));
 189     print "Listening on %f MHz." % (client.RF_getfreq()/10.0**6);
 190
 191     #FIXME, ugly
 192     RFST=0xDFE1
 193     client.CC_RFST_CAL();
 194     time.sleep(1);
 195
 196     while 1:
 197         client.CC_RFST_RX();
 198         rssi=client.RF_getrssi();
 199         client.CC_RFST_IDLE(); #idle
 200         time.sleep(0.01);
 201         string="";
 202         for foo in range(0,rssi>>2):
 203             string=("%s."%string);
 204         print "%02x %04i %s" % (rssi,rssi, string);
 205
 206 if(sys.argv[1]=="sniff"):
 207     client.CC1110_crystal();
 208     client.RF_idle();
 209
 210     #client.config_simpliciti(region);
 211
 212     print "Listening as %x on %f MHz" % (client.RF_getsmac(),
 213                                            client.RF_getfreq()/10.0**6);
 214     #Now we're ready to get packets.
 215     while 1:
 216         packet=None;
 217         while packet==None:
 218             packet=client.RF_rxpacket();
 219         printpacket(packet);
 220         sys.stdout.flush();
 221
 222 if(sys.argv[1]=="sniffsimpliciti"):
 223     region="us";
 224     if len(sys.argv)>2:
 225         region=sys.argv[2];
 226
 227     client.CC1110_crystal();
 228     client.RF_idle();
 229
 230     client.config_simpliciti(region);
 231
 232     print "Listening as %x on %f MHz" % (client.RF_getsmac(),
 233                                            client.RF_getfreq()/10.0**6);
 234     #Now we're ready to get packets.
 235     while 1:
 236         packet=None;
 237         while packet==None:
 238             packet=client.RF_rxpacket();
 239         printpacket(packet);
 240         sys.stdout.flush();
 241 if(sys.argv[1]=="sniffdash7"):
 242     region="lf";
 243     if len(sys.argv)>2:
 244         region=sys.argv[2];
 245
 246     client.CC1110_crystal();
 247     client.RF_idle();
 248
 249     client.config_dash7(region);
 250
 251     print "Listening as %x on %f MHz" % (client.RF_getsmac(),
 252                                            client.RF_getfreq()/10.0**6);
 253     #Now we're ready to get packets.
 254     while 1:
 255         packet=None;
 256         while packet==None:
 257             packet=client.RF_rxpacket();
 258         printpacket(packet);
 259         sys.stdout.flush();
 260
 261 if(sys.argv[1]=="simpliciti"):
 262     region="us";
 263     if len(sys.argv)>2:
 264         region=sys.argv[2];
 265
 266     client.CC1110_crystal();
 267     client.RF_idle();
 268
 269     client.config_simpliciti(region);
 270
 271     print "# Listening as %x on %f MHz" % (client.RF_getsmac(),
 272                                            client.RF_getfreq()/10.0**6);
 273     #Now we're ready to get packets.
 274     while 1:
 275         packet=None;
 276         while packet==None:
 277             packet=client.RF_rxpacket();
 278         handlesimplicitipacket(packet);
 279         sys.stdout.flush();
 280
 281
 282
 283 if(sys.argv[1]=="term"):
 284     GoodFETConsole(client).run();
 285 if(sys.argv[1]=="test"):
 286     client.test();
 287 if(sys.argv[1]=="deadtest"):
 288     for i in range(1,10):
 289         print "IDENT as %s" % client.CCidentstr();
 290 if(sys.argv[1]=="dumpcode"):
 291     f = sys.argv[2];
 292     start=0x0000;
 293     stop=0xFFFF;
 294     if(len(sys.argv)>3):
 295         start=int(sys.argv[3],16);
 296     if(len(sys.argv)>4):
 297         stop=int(sys.argv[4],16);
 298
 299     print "Dumping code from %04x to %04x as %s." % (start,stop,f);
 300     h = IntelHex(None);
 301     i=start;
 302     while i<=stop:
 303         h[i]=client.CCpeekcodebyte(i);
 304         if(i%0x100==0):
 305             print "Dumped %04x."%i;
 306         i+=1;
 307     h.write_hex_file(f);
 308 if(sys.argv[1]=="dumpdata"):
 309     f = sys.argv[2];
 310     start=0xE000;
 311     stop=0xFFFF;
 312     if(len(sys.argv)>3):
 313         start=int(sys.argv[3],16);
 314     if(len(sys.argv)>4):
 315         stop=int(sys.argv[4],16);
 316
 317     print "Dumping data from %04x to %04x as %s." % (start,stop,f);
 318     h = IntelHex(None);
 319     i=start;
 320     while i<=stop:
 321         h[i]=client.CCpeekdatabyte(i);
 322         if(i%0x100==0):
 323             print "Dumped %04x."%i;
 324         i+=1;
 325     h.write_hex_file(f);
 326 if(sys.argv[1]=="status"):
 327     print "Status: %s" %client.status();
 328 if(sys.argv[1]=="halt"):
 329     print "Halting CPU."
 330     client.halt();
 331
 332 if(sys.argv[1]=="infotest"):
 333     while 1:
 334         client.start();
 335         print "Ident   %s" % client.CCidentstr();
 336 if(sys.argv[1]=="info"):
 337     print "Ident   %s" % client.CCidentstr();
 338
 339     try:
 340         print "Freq    %10.3f MHz" % (client.RF_getfreq()/10**6);
 341         print "RSSI    %02x" % client.RF_getrssi();
 342     except:
 343         print "Freq, RSSI, etc unknown.  Install SmartRF7.";
 344     #print "Rate    %10i kbps" % (client.RF_getrate()/1000);
 345     #print "PacketLen %02i bytes" % client.RF_getpacketlen();
 346     #print "SMAC  0x%010x" % client.RF_getsmac();
 347     #print "TMAC  0x%010x" % client.RF_gettmac();
 348
 349 if(sys.argv[1]=="regs"):
 350     client.CMDrs();
 351
 352 if(sys.argv[1]=="erase"):
 353     print "Status: %s" % client.status();
 354     client.CCchiperase();
 355     print "Status: %s" %client.status();
 356
 357 if(sys.argv[1]=="peekinfo"):
 358     print "Select info flash."
 359     client.CCwr_config(1);
 360     print "Config is %02x" % client.CCrd_config();
 361
 362     start=0x0000;
 363     if(len(sys.argv)>2):
 364         start=int(sys.argv[2],16);
 365     stop=start;
 366     if(len(sys.argv)>3):
 367         stop=int(sys.argv[3],16);
 368     print "Peeking from %04x to %04x." % (start,stop);
 369     while start<=stop:
 370         print "%04x: %02x" % (start,client.CCpeekcodebyte(start));
 371         start=start+1;
 372 if(sys.argv[1]=="poke"):
 373     client.CCpokeirambyte(int(sys.argv[2],16),
 374                           int(sys.argv[3],16));
 375 if(sys.argv[1]=="randtest"):
 376     #Seed RNG
 377     client.CCpokeirambyte(0xBD,0x01); #RNDH=0x01
 378     client.CCpokeirambyte(0xB4,0x04); #ADCCON1=0x04
 379     client.CCpokeirambyte(0xBD,0x01); #RNDH=0x01
 380     client.CCpokeirambyte(0xB4,0x04); #ADCCON1=0x04
 381
 382     #Dump values
 383     for foo in range(1,10):
 384         print "%02x" % client.CCpeekirambyte(0xBD); #RNDH
 385         client.CCpokeirambyte(0xB4,0x04); #ADCCON1=0x04
 386         client.CCreleasecpu();
 387         client.CChaltcpu();
 388     print "%02x" % client.CCpeekdatabyte(0xDF61); #CHIP ID
 389 if(sys.argv[1]=="adctest"):
 390     # ADCTest 0xDF3A 0xDF3B
 391     print "ADC TEST %02x%02x" % (
 392         client.CCpeekdatabyte(0xDF3A),
 393         client.CCpeekdatabyte(0xDF3B));
 394 if(sys.argv[1]=="config"):
 395     print "Config is %02x" % client.CCrd_config();
 396
 397 if(sys.argv[1]=="flash"):
 398      f=sys.argv[2];
 399      start=0;
 400      stop=0xFFFF;
 401      if(len(sys.argv)>3):
 402          start=int(sys.argv[3],16);
 403      if(len(sys.argv)>4):
 404          stop=int(sys.argv[4],16);
 405
 406      client.flash(f);
 407 if(sys.argv[1]=="lock"):
 408     print "Status: %s" %client.status();
 409     client.CClockchip();
 410     print "Status: %s" %client.status();
 411 if(sys.argv[1]=="flashpage"):
 412     target=0;
 413     if(len(sys.argv)>2):
 414         target=int(sys.argv[2],16);
 415     print "Writing a page of flash from 0xF000 in XDATA"
 416     client.CCflashpage(target);
 417 if(sys.argv[1]=="erasebuffer"):
 418     print "Erasing flash buffer.";
 419     client.CCeraseflashbuffer();
 420
 421 if(sys.argv[1]=="writedata"):
 422     f=sys.argv[2];
 423     start=0;
 424     stop=0xFFFF;
 425     if(len(sys.argv)>3):
 426         start=int(sys.argv[3],16);
 427     if(len(sys.argv)>4):
 428         stop=int(sys.argv[4],16);
 429
 430     h = IntelHex(f);
 431
 432     for i in h._buf.keys():
 433         if(i>=start and i<=stop):
 434             client.CCpokedatabyte(i,h[i]);
 435             if(i%0x100==0):
 436                 print "%04x" % i;
 437 #if(sys.argv[1]=="flashtest"):
 438 #    client.CCflashtest();
 439 if(sys.argv[1]=="peekdata"):
 440     start=0x0000;
 441     if(len(sys.argv)>2):
 442         start=int(sys.argv[2],16);
 443     stop=start;
 444     if(len(sys.argv)>3):
 445         stop=int(sys.argv[3],16);
 446     print "Peeking from %04x to %04x." % (start,stop);
 447     while start<=stop:
 448         print "%04x: %02x" % (start,client.CCpeekdatabyte(start));
 449         start=start+1;
 450 if(sys.argv[1]=="peek"):
 451     start=0x0000;
 452     if(len(sys.argv)>2):
 453         start=int(sys.argv[2],16);
 454     stop=start;
 455     if(len(sys.argv)>3):
 456         stop=int(sys.argv[3],16);
 457     print "Peeking from %04x to %04x." % (start,stop);
 458     while start<=stop:
 459         print "%04x: %02x" % (start,client.CCpeekirambyte(start));
 460         start=start+1;
 461 if(sys.argv[1]=="verify"):
 462     f=sys.argv[2];
 463     start=0;
 464     stop=0xFFFF;
 465     if(len(sys.argv)>3):
 466         start=int(sys.argv[3],16);
 467     if(len(sys.argv)>4):
 468         stop=int(sys.argv[4],16);
 469
 470     h = IntelHex(f);
 471     for i in h._buf.keys():
 472         if(i>=start and i<stop):
 473             peek=client.CCpeekcodebyte(i)
 474             if(h[i]!=peek):
 475                 print "ERROR at %04x, found %02x not %02x"%(i,peek,h[i]);
 476             if(i%0x100==0):
 477                 print "%04x" % i;
 478 if(sys.argv[1]=="peekcode"):
 479     start=0x0000;
 480     if(len(sys.argv)>2):
 481         start=int(sys.argv[2],16);
 482     stop=start;
 483     if(len(sys.argv)>3):
 484         stop=int(sys.argv[3],16);
 485     print "Peeking from %04x to %04x." % (start,stop);
 486     while start<=stop:
 487         print "%04x: %02x" % (start,client.CCpeekcodebyte(start));
 488         start=start+1;
 489 if(sys.argv[1]=="pokedata"):
 490     start=0x0000;
 491     val=0x00;
 492     if(len(sys.argv)>2):
 493         start=int(sys.argv[2],16);
 494     if(len(sys.argv)>3):
 495         val=int(sys.argv[3],16);
 496     print "Poking %04x to become %02x." % (start,val);
 497     client.CCpokedatabyte(start,val);
 498
 499 client.stop();