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