goodfet.msp430 selftest is working!
[goodfet] / 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","?","?","?","?","?",
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 pyregs" % sys.argv[0];
49     print "%s sniff\n\tSniffs packets by current config." % sys.argv[0];
50     print "%s sniffob\n\tSniffs OpenBeacon traffic." % sys.argv[0];
51     print "%s carrier [freq]\n\tHolds a carrier on [freq] Hz." % sys.argv[0];
52     sys.exit();
53
54 #Initialize FET and set baud rate
55 client=GoodFETNRF();
56 client.serInit()
57
58 client.NRFsetup();
59
60 if(sys.argv[1]=="info"):
61     printconfig();
62
63 if(sys.argv[1]=="test"):
64     print "Old registers:"
65     printconfig();
66     
67     # Set PWR_UP=1 and PRIM_RX=0 in CONFIG.
68     client.poke(0x00,2);
69     #Delay of 1.5ms by round-trip.
70     
71     print "\n\n";
72     
73     #Try all data rates
74     for foo in [250*10**3,
75                 1*10**6,
76                 2*10**6]:
77         client.RF_setrate(foo);
78         if(client.RF_getrate()!=foo):
79             print "ERROR Rate %i not supported.  Got %i instead." % (foo,
80                                                                      client.RF_getrate());
81     
82     print "\n\n";
83     client.poke(0x0A,0xDEADBEEF,5);
84     #print "SMAC set to %010x" % client.RF_getsmac();
85     if client.RF_getsmac()!=0xdeadbeef:
86         print "ERROR: Failed to set MAC address.";
87     print "Final registers:"
88     printconfig();
89     
90 if(sys.argv[1]=="carrier"):
91     if len(sys.argv)>2:
92         client.RF_setfreq(eval(sys.argv[2]));
93     client.RF_carrier();
94     printconfig();
95     print "\nHolding a carrier wave.";
96     while(1):
97         time.sleep(1);
98 if(sys.argv[1]=="regs"):
99     for r in range(0,0x20):
100         print "r[0x%02x]=0x%010x //%16s " % (r,client.peek(r),regnames[r]);
101 if(sys.argv[1]=="pyregs"):
102     for r in range(0,0x20):
103         print "client.set(0x%02x,0x%010x); #%16s " % (r,client.peek(r),regnames[r]);
104
105 if(sys.argv[1]=="peek"):
106     start=0x0000;
107     if(len(sys.argv)>2):
108         start=int(sys.argv[2],16);
109     stop=start;
110     if(len(sys.argv)>3):
111         stop=int(sys.argv[3],16);
112     print "Peeking from %02x to %02x." % (start,stop);
113     while start<=stop:
114         print "%02x: %010x" % (start,client.peek(start));
115         start=start+1;
116 if(sys.argv[1]=="poke"):
117     start=0x0000;
118     val=0x00;
119     if(len(sys.argv)>2):
120         start=int(sys.argv[2],16);
121     if(len(sys.argv)>3):
122         val=int(sys.argv[3],16);
123     print "Poking %02x to become %010x." % (start,val);
124     
125     client.poke(start,val);
126     print "Poked to %04x" % client.peek(start);
127
128 if(sys.argv[1]=="sniffob"):
129     #Reversal of transmitter code from nRF_CMD.c of OpenBeacon
130     #TODO remove all poke() calls.
131     
132     client.poke(0x00,0x00); #Stop nRF
133     client.poke(0x01,0x00); #Disable Shockburst
134     client.poke(0x02,0x01); #Set RX Pipe 0
135     
136     client.RF_setfreq(2481 * 10**6);
137     client.poke(0x06,0x09); #2MBps, -18dBm in RF_SETUP
138     client.poke(0x07,0x78); #Reset status register
139     
140     #OpenBeacon defines these in little endian as follows.
141     client.RF_setmaclen(5); # SETUP_AW for 5-byte addresses.
142     #0x01, 0x02, 0x03, 0x02, 0x01
143     client.RF_setsmac(0x0102030201);
144     #'O', 'C', 'A', 'E', 'B'
145     client.RF_settmac(0x424541434F);
146     
147     #Set packet length of 16.
148     client.RF_setpacketlen(16);
149     
150     #Power radio, prime for RX, one-byte checksum.
151     client.poke(0x00,0x70|0x03|0x08); #0x08 for one byte, 0x04 for two.
152     
153     print "Listening as %010x on %i MHz" % (client.RF_getsmac(),
154                                            client.RF_getfreq()/10**6);
155     #Now we're ready to get packets.
156     while 1:
157         packet=None;
158         while packet==None:
159             #time.sleep(0.1);
160             packet=client.RF_rxpacket();
161         printpacket(packet);
162         sys.stdout.flush();
163
164 if(sys.argv[1]=="sniff"):
165     #client.poke(0x00,0x00); #Stop nRF
166     client.poke(0x07,0x78); #Reset status register
167     
168     #Power radio, prime for RX, checksum.
169     client.poke(0x00,0x70|0x03|0x08);
170     
171     print "Listening as %010x on %i MHz" % (client.RF_getsmac(),
172                                            client.RF_getfreq()/10**6);
173     #Now we're ready to get packets.
174     
175     while 1:
176         packet=None;
177         while packet==None:
178             #time.sleep(0.1);
179             packet=client.RF_rxpacket();
180         printpacket(packet);
181         sys.stdout.flush();
182 if(sys.argv[1]=="explore"):
183     #client.poke(0x00,0x00); #Stop nRF
184     client.poke(0x07,0x78); #Reset status register
185     
186     #Power radio, prime for RX, no checksum.
187     client.poke(0x00,0x70|0x03);
188     
189     #Set packet length of 32.
190     #Without checksums, extra data will mix in.
191     client.RF_setpacketlen(32);
192     client.RF_setmaclen(3); # shortest address length
193     
194     #Now we're ready to get packets.
195     for smac in [0x0102030201, 0]:
196         client.RF_setsmac(smac);
197         for chan in range(0,0x80):
198             client.RF_setfreq((2400+chan) * 10**6);
199             time.sleep(1);
200             packet=client.RF_rxpacket();
201             if packet!=None:
202                 print "Listening as %010x on %i MHz" % (client.RF_getsmac(),
203                                                         client.RF_getfreq()/10**6);
204                 printpacket(packet);
205                 sys.stdout.flush();