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 def printpacket(packet):
  15     s="";
  16     i=0;
  17     for foo in packet:
  18         i=i+1;
  19         if i>client.packetlen: break;
  20         s="%s %02x" % (s,ord(foo));
  21     print "%s" %s;
  22
  23 def printconfig():
  24     print "SMAC  0x%010x" % client.RF_getsmac();
  25     print "TMAC  0x%010x" % client.RF_gettmac();
  26     print "Freq    %10i MHz" % (client.RF_getfreq()/10**6);
  27     print "PacketLen %02i" % client.RF_getpacketlen();
  28     print "MacLen    %02i" % client.RF_getmaclen();
  29
  30 if(len(sys.argv)==1):
  31     print "Usage: %s verb [objects]\n" % sys.argv[0];
  32     print "%s info" % sys.argv[0];
  33     print "%s test" % sys.argv[0];
  34     print "%s sniffob\n\tSniffs OpenBeacon traffic." % sys.argv[0];
  35     print "%s regs" % sys.argv[0];
  36     sys.exit();
  37
  38 #Initialize FET and set baud rate
  39 client=GoodFETNRF();
  40 client.serInit()
  41
  42 client.NRFsetup();
  43
  44 if(sys.argv[1]=="info"):
  45     printconfig();
  46
  47 if(sys.argv[1]=="test"):
  48     # Set PWR_UP=1 and PRIM_RX=0 in CONFIG.
  49     client.poke(0x00,2);
  50     #Delay of 1.5ms by round-trip.
  51     # Set CONT_WAVE, PLL_LOCK, and 0dBm in RF_SETUP
  52     client.poke(0x06,8+10+4+2);
  53     client.RF_setfreq(2480 * 10**6);
  54
  55     #Print registers, just for fun.
  56     print "SMAC  0x%010x" % client.RF_getsmac();
  57     print "Freq    %10i MHz" % (client.RF_getfreq()/10**6);
  58
  59     client.poke(0x0A,0xDEADBEEF,5);
  60     print "SMAC set to %010x" % client.RF_getsmac();
  61     if client.RF_getsmac()!=0xdeadbeef:
  62         print "ERROR: Failed to set MAC address.";
  63
  64 if(sys.argv[1]=="regs"):
  65     for r in range(0,0x20):
  66         print "r[0x%02x]=0x%010x" % (r,client.peek(r));
  67
  68 if(sys.argv[1]=="sniffob"):
  69     #Reversal of transmitter code from nRF_CMD.c of OpenBeacon
  70     #TODO remove all poke() calls.
  71
  72     client.poke(0x00,0x00); #Stop nRF
  73     client.poke(0x01,0x00); #Disable Shockburst
  74     client.poke(0x02,0x01); #Set RX Pipe 0
  75     client.RF_setmaclen(5); # SETUP_AW for 5-byte addresses.
  76     client.RF_setfreq(2481 * 10**6);
  77     client.poke(0x06,0x09); #2MBps, -18dBm in RF_SETUP
  78     client.poke(0x07,0x78); #Reset status register
  79
  80     #OpenBeacon defines these in little endian as follows.
  81     #0x01, 0x02, 0x03, 0x02, 0x01
  82     client.RF_setsmac(0x0102030201);
  83     #'O', 'C', 'A', 'E', 'B'
  84     client.RF_settmac(0x424541434F);
  85
  86     #Set packet length of 16.
  87     client.RF_setpacketlen(16);
  88
  89     #Power radio, prime for RX, checksum.
  90     client.poke(0x00,0x70|0x03|0x08);
  91
  92     print "Listening as %010x on %i MHz" % (client.RF_getsmac(),
  93                                            client.RF_getfreq()/10**6);
  94     #Now we're ready to get packets.
  95
  96     while 1:
  97         packet=None;
  98         while packet==None:
  99             #time.sleep(0.1);
 100             packet=client.RF_rxpacket();
 101         printpacket(packet);
 102         sys.stdout.flush();