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 bytes" % client.RF_getpacketlen();
  28     print "MacLen    %2i bytes" % 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 regs" % sys.argv[0];
  35     print "%s sniffob\n\tSniffs OpenBeacon traffic." % sys.argv[0];
  36     print "%s carrier [freq]\n\tHolds a carrier on [freq] Hz." % sys.argv[0];
  37     sys.exit();
  38
  39 #Initialize FET and set baud rate
  40 client=GoodFETNRF();
  41 client.serInit()
  42
  43 client.NRFsetup();
  44
  45 if(sys.argv[1]=="info"):
  46     printconfig();
  47
  48 if(sys.argv[1]=="test"):
  49     # Set PWR_UP=1 and PRIM_RX=0 in CONFIG.
  50     client.poke(0x00,2);
  51     #Delay of 1.5ms by round-trip.
  52     # Set CONT_WAVE, PLL_LOCK, and 0dBm in RF_SETUP
  53     client.poke(0x06,8+10+4+2);
  54     client.RF_setfreq(2480 * 10**6);
  55
  56     #Print registers, just for fun.
  57     printconfig();
  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]=="carrier"):
  65     if len(sys.argv)>2:
  66         client.RF_setfreq(eval(sys.argv[2]));
  67     client.RF_carrier();
  68     printconfig();
  69     print "\nHolding a carrier wave.";
  70     while(1):
  71         time.sleep(1);
  72 if(sys.argv[1]=="regs"):
  73     for r in range(0,0x20):
  74         print "r[0x%02x]=0x%010x" % (r,client.peek(r));
  75
  76 if(sys.argv[1]=="sniffob"):
  77     #Reversal of transmitter code from nRF_CMD.c of OpenBeacon
  78     #TODO remove all poke() calls.
  79
  80     client.poke(0x00,0x00); #Stop nRF
  81     client.poke(0x01,0x00); #Disable Shockburst
  82     client.poke(0x02,0x01); #Set RX Pipe 0
  83     client.RF_setmaclen(5); # SETUP_AW for 5-byte addresses.
  84     client.RF_setfreq(2481 * 10**6);
  85     client.poke(0x06,0x09); #2MBps, -18dBm in RF_SETUP
  86     client.poke(0x07,0x78); #Reset status register
  87
  88     #OpenBeacon defines these in little endian as follows.
  89     #0x01, 0x02, 0x03, 0x02, 0x01
  90     client.RF_setsmac(0x0102030201);
  91     #'O', 'C', 'A', 'E', 'B'
  92     client.RF_settmac(0x424541434F);
  93
  94     #Set packet length of 16.
  95     client.RF_setpacketlen(16);
  96
  97     #Power radio, prime for RX, checksum.
  98     client.poke(0x00,0x70|0x03|0x08);
  99
 100     print "Listening as %010x on %i MHz" % (client.RF_getsmac(),
 101                                            client.RF_getfreq()/10**6);
 102     #Now we're ready to get packets.
 103
 104     while 1:
 105         packet=None;
 106         while packet==None:
 107             #time.sleep(0.1);
 108             packet=client.RF_rxpacket();
 109         printpacket(packet);
 110         sys.stdout.flush();