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