git.rot13.org Git - goodfet/blob - client/goodfet.nrf

   1 #!/usr/bin/env python
   2
   3 #GoodFET SPI Flash Client
   4 #by Travis Goodspeed
   5
   6 import sys;
   7 import binascii;
   8 import array;
   9 import time;
  10
  11 from GoodFETNRF import GoodFETNRF;
  12 from intelhex import IntelHex;
  13
  14
  15 regnames=["CONFIG","EN_AA","EN_RXADDR","SETUP_AW","SETUP_RET",
  16           "RF_CH","RF_SETUP","STATUS","OBSERVE_TX","RPD",
  17           "RX_ADDR_P0","RX_ADDR_P1","RX_ADDR_P2","RX_ADDR_P3","RX_ADDR_P4","RX_ADDR_P5",
  18           "TX_ADDR",
  19           "RX_PW_P0","RX_PW_P1","RX_PW_P2","RX_PW_P3","RX_PW_P4","RX_PW_P5",
  20           "FIFO_STATUS","?",
  21           "?","?","?","DYNPD","FEATURE","AGC_CONFIG","?","?",
  22           "?","?","?","?","?","?","?","?"];
  23
  24 def printpacket(packet):
  25     s="";
  26     i=0;
  27     for foo in packet:
  28         i=i+1;
  29         if i>client.packetlen: break;
  30         s="%s %02x" % (s,ord(foo));
  31     print "%s" %s;
  32
  33 def printconfig():
  34     print "Encoding %s" % client.RF_getenc();
  35     print "Freq    %10i MHz" % (client.RF_getfreq()/10**6);
  36     print "Rate    %10i kbps" % (client.RF_getrate()/1000);
  37     print "PacketLen %02i bytes" % client.RF_getpacketlen();
  38     #print "MacLen    %2i bytes" % client.RF_getmaclen();
  39     print "SMAC  0x%010x" % client.RF_getsmac();
  40     print "TMAC  0x%010x" % client.RF_gettmac();
  41
  42
  43 if(len(sys.argv)==1):
  44     print "Usage: %s verb [objects]\n" % sys.argv[0];
  45     print "%s info" % sys.argv[0];
  46     print "%s test" % sys.argv[0];
  47     print "%s regs" % sys.argv[0];
  48     print "%s regbits" % sys.argv[0];
  49     print "%s pyregs" % sys.argv[0];
  50
  51     print "%s sniff\n\tSniffs packets by current config." % sys.argv[0];
  52     print "%s sniffob\n\tSniffs OpenBeacon traffic." % sys.argv[0];
  53     print "%s snifftp\n\tSniffs Turning Point Clicker traffic." % sys.argv[0];
  54     print "%s snifftp\n\tSniffs SparkFun Dongle traffic." % sys.argv[0];
  55
  56     print "%s sniffprom\n\tSniffs promiscuously." % sys.argv[0];
  57
  58     print "%s hosttp\n\tHosts Turning Point Clicker traffic." % sys.argv[0];
  59
  60     print "%s carrier [freq]\n\tHolds a carrier on [freq] Hz." % sys.argv[0];
  61     sys.exit();
  62
  63 #Initialize FET and set baud rate
  64 client=GoodFETNRF();
  65 client.serInit()
  66
  67 client.NRFsetup();
  68
  69 if(sys.argv[1]=="info"):
  70     printconfig();
  71
  72 if(sys.argv[1]=="test"):
  73     print "Old registers:"
  74     printconfig();
  75
  76     # Set PWR_UP=1 and PRIM_RX=0 in CONFIG.
  77     client.poke(0x00,2);
  78     #Delay of 1.5ms by round-trip.
  79
  80     print "\n\n";
  81
  82     #Try all data rates
  83     for foo in [250*10**3,
  84                 1*10**6,
  85                 2*10**6]:
  86         client.RF_setrate(foo);
  87         if(client.RF_getrate()!=foo):
  88             print "ERROR Rate %i not supported.  Got %i instead." % (foo,
  89                                                                      client.RF_getrate());
  90
  91     print "\n\n";
  92     client.poke(0x0A,0xDEADBEEF,5);
  93     #print "SMAC set to %010x" % client.RF_getsmac();
  94     if client.RF_getsmac()!=0xdeadbeef:
  95         print "ERROR: Failed to set MAC address.";
  96     print "Final registers:"
  97     printconfig();
  98
  99 if(sys.argv[1]=="carrier"):
 100     if len(sys.argv)>2:
 101         client.RF_setfreq(eval(sys.argv[2]));
 102     client.RF_carrier();
 103     printconfig();
 104     print "\nHolding a carrier wave.";
 105     while(1):
 106         time.sleep(1);
 107 if(sys.argv[1]=="regs"):
 108     for r in range(0,0x20):
 109         print "r[0x%02x]=0x%010x //%16s " % (r,client.peek(r),regnames[r]);
 110 if(sys.argv[1]=="pyregs"):
 111     for r in range(0,0x20):
 112         print "client.set(0x%02x,0x%010x); #%16s " % (r,client.peek(r),regnames[r]);
 113
 114 if(sys.argv[1]=="peek"):
 115     start=0x0000;
 116     if(len(sys.argv)>2):
 117         start=int(sys.argv[2],16);
 118     stop=start;
 119     if(len(sys.argv)>3):
 120         stop=int(sys.argv[3],16);
 121     print "Peeking from %02x to %02x." % (start,stop);
 122     while start<=stop:
 123         print "%02x: %010x" % (start,client.peek(start));
 124         start=start+1;
 125 if(sys.argv[1]=="poke"):
 126     start=0x0000;
 127     val=0x00;
 128     if(len(sys.argv)>2):
 129         start=int(sys.argv[2],16);
 130     if(len(sys.argv)>3):
 131         val=int(sys.argv[3],16);
 132     print "Poking %02x to become %010x." % (start,val);
 133
 134     client.poke(start,val);
 135     print "Poked to %04x" % client.peek(start);
 136
 137 if(sys.argv[1]=="sniffob"):
 138     #Reversal of transmitter code from nRF_CMD.c of OpenBeacon
 139     #TODO remove all poke() calls.
 140
 141     client.poke(0x00,0x00); #Stop nRF
 142     client.poke(0x01,0x00); #Disable Shockburst
 143     client.poke(0x02,0x01); #Set RX Pipe 0
 144
 145     client.RF_setfreq(2481 * 10**6);
 146     client.poke(0x06,0x09); #2MBps, -18dBm in RF_SETUP
 147     client.poke(0x07,0x78); #Reset status register
 148
 149     #OpenBeacon defines these in little endian as follows.
 150     client.RF_setmaclen(5); # SETUP_AW for 5-byte addresses.
 151     #0x01, 0x02, 0x03, 0x02, 0x01
 152     client.RF_setsmac(0x0102030201);
 153     #'O', 'C', 'A', 'E', 'B'
 154     client.RF_settmac(0x424541434F);
 155
 156     #Set packet length of 16.
 157     client.RF_setpacketlen(16);
 158
 159     #Power radio, prime for RX, one-byte checksum.
 160     client.poke(0x00,0x70|0x03|0x08); #0x08 for one byte, 0x04 for two.
 161
 162     print "Listening as %010x on %i MHz" % (client.RF_getsmac(),
 163                                            client.RF_getfreq()/10**6);
 164     #Now we're ready to get packets.
 165     while 1:
 166         packet=None;
 167         while packet==None:
 168             #time.sleep(0.1);
 169             packet=client.RF_rxpacket();
 170         printpacket(packet);
 171         sys.stdout.flush();
 172
 173 if(sys.argv[1]=="regbits"):
 174     print "Scanning registers to determine which bits are valid."
 175     regbits=range(0,0x30);
 176     for r in range(0,0x30):
 177         old=client.peek(r);
 178         #Which bits can be set?
 179         client.poke(r,0xFF);
 180         ones=client.peek(r);
 181         #Which bits can be clear?
 182         client.poke(r,0x00);
 183         zeroes=client.peek(r);
 184         regbits[r]=(ones & (~zeroes));
 185     for r in range(0,0x30):
 186         if regbits[r]!=0:
 187             print "r[0x%02x] masked %02x // %s" % (r,regbits[r], regnames[r]);
 188 if(sys.argv[1]=="sniffprom"):
 189     #Reversal of transmitter code from nRF_CMD.c of OpenBeacon
 190     #TODO remove all poke() calls.
 191
 192     client.poke(0x00,0x00); #Stop nRF
 193     client.poke(0x01,0x00); #Disable Shockburst
 194     client.poke(0x02,0x01); #Set RX Pipe 0
 195
 196     client.RF_setfreq(2481 * 10**6);
 197     client.poke(0x06,0x09); #2MBps, -18dBm in RF_SETUP
 198     client.poke(0x07,0x78); #Reset status register
 199
 200     #OpenBeacon defines these in little endian as follows.
 201     client.RF_setmaclen(5); # SETUP_AW for 3-byte addresses.
 202     #0x01, 0x02, 0x03, 0x02, 0x01
 203     #client.RF_setsmac(0x0102030201);
 204     #client.RF_setsmac(0x0000000201);
 205     client.RF_setsmac(0x0102);
 206     #'O', 'C', 'A', 'E', 'B'
 207     #client.RF_settmac(0x424541434F);
 208
 209     #Mac packet length, illegally 0-length address field.
 210     client.RF_setpacketlen(32);
 211     client.poke(0x03,0);
 212
 213     #Power radio, prime for RX, one-byte checksum.
 214     client.poke(0x00,0x70|0x03); #0x08 for one byte, 0x04 for two.
 215
 216     print "Listening as %010x on %i MHz" % (client.RF_getsmac(),
 217                                            client.RF_getfreq()/10**6);
 218     #Now we're ready to get packets.
 219     while 1:
 220         packet=None;
 221         while packet==None:
 222             #time.sleep(0.1);
 223             packet=client.RF_rxpacket();
 224         printpacket(packet);
 225         sys.stdout.flush();
 226
 227
 228
 229 if(sys.argv[1]=="sniffskybrake"):
 230     #Reversal of transmitter code from nRF_CMD.c of OpenBeacon
 231     #TODO remove all poke() calls.
 232
 233     client.poke(0x00,0x00); #Stop nRF
 234     client.poke(0x01,0x00); #Disable Shockburst
 235     client.poke(0x02,0x01); #Set RX Pipe 0
 236
 237     client.RF_setfreq(2439 * 10**6);
 238     client.poke(0x06,0x00); #1MBps
 239     client.poke(0x07,0x78); #Reset status register
 240
 241     #OpenBeacon defines these in little endian as follows.
 242     client.RF_setmaclen(2); # SETUP_AW for 3-byte addresses.
 243     #0x01, 0x02, 0x03, 0x02, 0x01
 244
 245     client.RF_setsmac(0x070700d2c4); #reversed
 246
 247     #client.RF_setsmac(0xd2c4);
 248     #client.RF_setsmac(0);
 249
 250     #Mac packet length, illegally 0-length address field.
 251     client.RF_setpacketlen(32);
 252
 253     #Power radio, prime for RX, one-byte checksum.
 254     client.poke(0x00,0x70|0x03); #0x08 for one byte, 0x04 for two.
 255
 256     print "Listening as %010x on %i MHz" % (client.RF_getsmac(),
 257                                            client.RF_getfreq()/10**6);
 258     print "%i byte mac match." % client.RF_getmaclen();
 259     #Now we're ready to get packets.
 260     while 1:
 261         packet=None;
 262         while packet==None:
 263             #time.sleep(0.1);
 264             packet=client.RF_rxpacket();
 265         printpacket(packet);
 266         sys.stdout.flush();
 267
 268 if(sys.argv[1]=="sniffsf"):
 269     #Reversal of transmitter code from nRF_CMD.c of OpenBeacon
 270     #TODO remove all poke() calls.
 271
 272     client.poke(0x00,0x00); #Stop nRF
 273     client.poke(0x01,0x00); #Disable Shockburst
 274     client.poke(0x02,0x01); #Set RX Pipe 0
 275
 276     client.RF_setfreq(2402 * 10**6);
 277     client.poke(0x06,0x07); #1Mbps
 278     client.poke(0x07,0x78); #Reset status register
 279
 280     #OpenBeacon defines these in little endian as follows.
 281     client.RF_setmaclen(5); # SETUP_AW for 5-byte addresses.
 282     #0x01, 0x02, 0x03, 0x02, 0x01
 283     client.RF_setsmac(0xe7e7e7e7e7);
 284     #'O', 'C', 'A', 'E', 'B'
 285     client.RF_settmac(0xe7e7e7e7e7);
 286
 287     #Set packet length of 16.
 288     client.RF_setpacketlen(4);
 289
 290     #Power radio, prime for RX, one-byte checksum.
 291     client.poke(0x00,0x70|0x03|0x08); #0x08 for one byte, 0x04 for two.
 292
 293     print "Listening as %010x on %i MHz" % (client.RF_getsmac(),
 294                                            client.RF_getfreq()/10**6);
 295     #Now we're ready to get packets.
 296     while 1:
 297         packet=None;
 298         while packet==None:
 299             #time.sleep(0.1);
 300             packet=client.RF_rxpacket();
 301         printpacket(packet);
 302         sys.stdout.flush();
 303
 304 if(sys.argv[1]=="snifftp"):
 305     client.poke(0x00,0x00); #Stop nRF
 306     client.poke(0x01,0x00); #Disable Shockburst
 307     client.poke(0x02,0x01); #Set RX Pipe 0
 308
 309     client.RF_setfreq((2400+0x29) * 10**6);
 310     client.poke(0x06,0x00); #1Mbps
 311     client.poke(0x07,0x78); #Reset status register
 312
 313     client.RF_setmaclen(3); # SETUP_AW for 3-byte addresses.
 314     client.RF_setsmac(0x123456);
 315     client.RF_setpacketlen(4);
 316
 317     #Power radio, prime for RX, two-byte checksum.
 318     client.poke(0x00,0x70|0x03|0x04|0x08);
 319
 320     print "Listening as %010x on %i MHz" % (client.RF_getsmac(),
 321                                            client.RF_getfreq()/10**6);
 322     #Now we're ready to get packets.
 323     while 1:
 324         packet=None;
 325         while packet==None:
 326             #time.sleep(0.1);
 327             packet=client.RF_rxpacket();
 328         printpacket(packet);
 329         sys.stdout.flush();
 330
 331 if(sys.argv[1]=="hosttp"):
 332     client.poke(0x00,0x00); #Stop nRF
 333     client.poke(0x01,0x00); #Disable Shockburst
 334     client.poke(0x02,0x01); #Set RX Pipe 0
 335
 336     chan=0x29;
 337
 338     client.RF_setfreq((2400+chan) * 10**6);
 339     client.poke(0x06,0x00); #1Mbps
 340     client.poke(0x07,0x78); #Reset status register
 341
 342     client.RF_setmaclen(3); # SETUP_AW for 3-byte addresses.
 343     client.RF_setsmac(0x123456);
 344     client.RF_setpacketlen(4);
 345
 346     #Power radio, prime for RX, two-byte checksum.
 347     client.poke(0x00,0x70|0x03|0x04|0x08);
 348
 349     print "Listening as %010x on %i MHz" % (client.RF_getsmac(),
 350                                            client.RF_getfreq()/10**6);
 351     #Now we're ready to get packets.
 352     while 1:
 353         packet=None;
 354         while packet==None:
 355             packet=client.RF_rxpacket();
 356         mac=((ord(packet[0])<<16)+
 357              (ord(packet[1])<<8)+
 358              ord(packet[2]));
 359         key=packet[3];
 360         print "%c from %06x" % (key,mac);
 361         sys.stdout.flush();
 362
 363 if(sys.argv[1]=="sniff"):
 364     #client.poke(0x00,0x00); #Stop nRF
 365     client.poke(0x07,0x78); #Reset status register
 366
 367     #Power radio, prime for RX, checksum.
 368     client.poke(0x00,0x70|0x03|0x08);
 369
 370     print "Listening as %010x on %i MHz" % (client.RF_getsmac(),
 371                                            client.RF_getfreq()/10**6);
 372     #Now we're ready to get packets.
 373
 374     while 1:
 375         packet=None;
 376         while packet==None:
 377             #time.sleep(0.1);
 378             packet=client.RF_rxpacket();
 379         printpacket(packet);
 380         sys.stdout.flush();
 381 if(sys.argv[1]=="explore"):
 382     #client.poke(0x00,0x00); #Stop nRF
 383     client.poke(0x07,0x78); #Reset status register
 384
 385     #Power radio, prime for RX, no checksum.
 386     client.poke(0x00,0x70|0x03);
 387
 388     #Set packet length of 32.
 389     #Without checksums, extra data will mix in.
 390     client.RF_setpacketlen(32);
 391     client.RF_setmaclen(3); # shortest address length
 392
 393     #Now we're ready to get packets.
 394     for smac in [0x0102030201, 0]:
 395         client.RF_setsmac(smac);
 396         for chan in range(0,0x80):
 397             client.RF_setfreq((2400+chan) * 10**6);
 398             time.sleep(1);
 399             packet=client.RF_rxpacket();
 400             if packet!=None:
 401                 print "Listening as %010x on %i MHz" % (client.RF_getsmac(),
 402                                                         client.RF_getfreq()/10**6);
 403                 printpacket(packet);
 404                 sys.stdout.flush();