Skybrake stuff, supposes a particular MAC.
[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","FEATURE","AGC_CONFIG","?","?",
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 regbits" % sys.argv[0];
49     print "%s pyregs" % sys.argv[0];
50
51     print "%s sniff\n\tSniffs packets by current config." % sys.argv[0];
52     print "%s sniffob\n\tSniffs OpenBeacon traffic." % sys.argv[0];
53     print "%s snifftp\n\tSniffs Turning Point Clicker traffic." % sys.argv[0];
54     print "%s snifftp\n\tSniffs SparkFun Dongle traffic." % sys.argv[0];
55     
56     print "%s sniffprom\n\tSniffs promiscuously." % sys.argv[0];
57     
58     print "%s hosttp\n\tHosts Turning Point Clicker traffic." % sys.argv[0];
59
60     print "%s carrier [freq]\n\tHolds a carrier on [freq] Hz." % sys.argv[0];
61     sys.exit();
62
63 #Initialize FET and set baud rate
64 client=GoodFETNRF();
65 client.serInit()
66
67 client.NRFsetup();
68
69 if(sys.argv[1]=="info"):
70     printconfig();
71
72 if(sys.argv[1]=="test"):
73     print "Old registers:"
74     printconfig();
75     
76     # Set PWR_UP=1 and PRIM_RX=0 in CONFIG.
77     client.poke(0x00,2);
78     #Delay of 1.5ms by round-trip.
79     
80     print "\n\n";
81     
82     #Try all data rates
83     for foo in [250*10**3,
84                 1*10**6,
85                 2*10**6]:
86         client.RF_setrate(foo);
87         if(client.RF_getrate()!=foo):
88             print "ERROR Rate %i not supported.  Got %i instead." % (foo,
89                                                                      client.RF_getrate());
90     
91     print "\n\n";
92     client.poke(0x0A,0xDEADBEEF,5);
93     #print "SMAC set to %010x" % client.RF_getsmac();
94     if client.RF_getsmac()!=0xdeadbeef:
95         print "ERROR: Failed to set MAC address.";
96     print "Final registers:"
97     printconfig();
98     
99 if(sys.argv[1]=="carrier"):
100     if len(sys.argv)>2:
101         client.RF_setfreq(eval(sys.argv[2]));
102     client.RF_carrier();
103     printconfig();
104     print "\nHolding a carrier wave.";
105     while(1):
106         time.sleep(1);
107 if(sys.argv[1]=="regs"):
108     for r in range(0,0x20):
109         print "r[0x%02x]=0x%010x //%16s " % (r,client.peek(r),regnames[r]);
110 if(sys.argv[1]=="pyregs"):
111     for r in range(0,0x20):
112         print "client.set(0x%02x,0x%010x); #%16s " % (r,client.peek(r),regnames[r]);
113
114 if(sys.argv[1]=="peek"):
115     start=0x0000;
116     if(len(sys.argv)>2):
117         start=int(sys.argv[2],16);
118     stop=start;
119     if(len(sys.argv)>3):
120         stop=int(sys.argv[3],16);
121     print "Peeking from %02x to %02x." % (start,stop);
122     while start<=stop:
123         print "%02x: %010x" % (start,client.peek(start));
124         start=start+1;
125 if(sys.argv[1]=="poke"):
126     start=0x0000;
127     val=0x00;
128     if(len(sys.argv)>2):
129         start=int(sys.argv[2],16);
130     if(len(sys.argv)>3):
131         val=int(sys.argv[3],16);
132     print "Poking %02x to become %010x." % (start,val);
133     
134     client.poke(start,val);
135     print "Poked to %04x" % client.peek(start);
136
137 if(sys.argv[1]=="sniffob"):
138     #Reversal of transmitter code from nRF_CMD.c of OpenBeacon
139     #TODO remove all poke() calls.
140     
141     client.poke(0x00,0x00); #Stop nRF
142     client.poke(0x01,0x00); #Disable Shockburst
143     client.poke(0x02,0x01); #Set RX Pipe 0
144     
145     client.RF_setfreq(2481 * 10**6);
146     client.poke(0x06,0x09); #2MBps, -18dBm in RF_SETUP
147     client.poke(0x07,0x78); #Reset status register
148     
149     #OpenBeacon defines these in little endian as follows.
150     client.RF_setmaclen(5); # SETUP_AW for 5-byte addresses.
151     #0x01, 0x02, 0x03, 0x02, 0x01
152     client.RF_setsmac(0x0102030201);
153     #'O', 'C', 'A', 'E', 'B'
154     client.RF_settmac(0x424541434F);
155     
156     #Set packet length of 16.
157     client.RF_setpacketlen(16);
158     
159     #Power radio, prime for RX, one-byte checksum.
160     client.poke(0x00,0x70|0x03|0x08); #0x08 for one byte, 0x04 for two.
161     
162     print "Listening as %010x on %i MHz" % (client.RF_getsmac(),
163                                            client.RF_getfreq()/10**6);
164     #Now we're ready to get packets.
165     while 1:
166         packet=None;
167         while packet==None:
168             #time.sleep(0.1);
169             packet=client.RF_rxpacket();
170         printpacket(packet);
171         sys.stdout.flush();
172
173 if(sys.argv[1]=="regbits"):
174     print "Scanning registers to determine which bits are valid."
175     regbits=range(0,0x30);
176     for r in range(0,0x30):
177         old=client.peek(r);
178         #Which bits can be set?
179         client.poke(r,0xFF);
180         ones=client.peek(r);
181         #Which bits can be clear?
182         client.poke(r,0x00);
183         zeroes=client.peek(r);
184         regbits[r]=(ones & (~zeroes));
185     for r in range(0,0x30):
186         if regbits[r]!=0:
187             print "r[0x%02x] masked %02x // %s" % (r,regbits[r], regnames[r]);
188 if(sys.argv[1]=="sniffprom"):
189     #Reversal of transmitter code from nRF_CMD.c of OpenBeacon
190     #TODO remove all poke() calls.
191     
192     client.poke(0x00,0x00); #Stop nRF
193     client.poke(0x01,0x00); #Disable Shockburst
194     client.poke(0x02,0x01); #Set RX Pipe 0
195     
196     client.RF_setfreq(2481 * 10**6);
197     client.poke(0x06,0x09); #2MBps, -18dBm in RF_SETUP
198     client.poke(0x07,0x78); #Reset status register
199     
200     #OpenBeacon defines these in little endian as follows.
201     client.RF_setmaclen(5); # SETUP_AW for 3-byte addresses.
202     #0x01, 0x02, 0x03, 0x02, 0x01
203     #client.RF_setsmac(0x0102030201);
204     #client.RF_setsmac(0x0000000201);
205     client.RF_setsmac(0x0102);
206     #'O', 'C', 'A', 'E', 'B'
207     #client.RF_settmac(0x424541434F);
208     
209     #Mac packet length, illegally 0-length address field.
210     client.RF_setpacketlen(32);
211     client.poke(0x03,0);
212     
213     #Power radio, prime for RX, one-byte checksum.
214     client.poke(0x00,0x70|0x03); #0x08 for one byte, 0x04 for two.
215     
216     print "Listening as %010x on %i MHz" % (client.RF_getsmac(),
217                                            client.RF_getfreq()/10**6);
218     #Now we're ready to get packets.
219     while 1:
220         packet=None;
221         while packet==None:
222             #time.sleep(0.1);
223             packet=client.RF_rxpacket();
224         printpacket(packet);
225         sys.stdout.flush();
226
227
228
229 if(sys.argv[1]=="sniffskybrake"):
230     #Reversal of transmitter code from nRF_CMD.c of OpenBeacon
231     #TODO remove all poke() calls.
232     
233     client.poke(0x00,0x00); #Stop nRF
234     client.poke(0x01,0x00); #Disable Shockburst
235     client.poke(0x02,0x01); #Set RX Pipe 0
236     
237     client.RF_setfreq(2439 * 10**6);
238     client.poke(0x06,0x00); #1MBps
239     client.poke(0x07,0x78); #Reset status register
240     
241     #OpenBeacon defines these in little endian as follows.
242     client.RF_setmaclen(2); # SETUP_AW for 3-byte addresses.
243     #0x01, 0x02, 0x03, 0x02, 0x01
244     
245     client.RF_setsmac(0x070700d2c4); #reversed
246     
247     #client.RF_setsmac(0xd2c4);
248     #client.RF_setsmac(0);
249     
250     #Mac packet length, illegally 0-length address field.
251     client.RF_setpacketlen(32);
252     
253     #Power radio, prime for RX, one-byte checksum.
254     client.poke(0x00,0x70|0x03); #0x08 for one byte, 0x04 for two.
255     
256     print "Listening as %010x on %i MHz" % (client.RF_getsmac(),
257                                            client.RF_getfreq()/10**6);
258     print "%i byte mac match." % client.RF_getmaclen();
259     #Now we're ready to get packets.
260     while 1:
261         packet=None;
262         while packet==None:
263             #time.sleep(0.1);
264             packet=client.RF_rxpacket();
265         printpacket(packet);
266         sys.stdout.flush();
267
268 if(sys.argv[1]=="sniffsf"):
269     #Reversal of transmitter code from nRF_CMD.c of OpenBeacon
270     #TODO remove all poke() calls.
271     
272     client.poke(0x00,0x00); #Stop nRF
273     client.poke(0x01,0x00); #Disable Shockburst
274     client.poke(0x02,0x01); #Set RX Pipe 0
275     
276     client.RF_setfreq(2402 * 10**6);
277     client.poke(0x06,0x07); #1Mbps
278     client.poke(0x07,0x78); #Reset status register
279     
280     #OpenBeacon defines these in little endian as follows.
281     client.RF_setmaclen(5); # SETUP_AW for 5-byte addresses.
282     #0x01, 0x02, 0x03, 0x02, 0x01
283     client.RF_setsmac(0xe7e7e7e7e7);
284     #'O', 'C', 'A', 'E', 'B'
285     client.RF_settmac(0xe7e7e7e7e7);
286     
287     #Set packet length of 16.
288     client.RF_setpacketlen(4);
289     
290     #Power radio, prime for RX, one-byte checksum.
291     client.poke(0x00,0x70|0x03|0x08); #0x08 for one byte, 0x04 for two.
292     
293     print "Listening as %010x on %i MHz" % (client.RF_getsmac(),
294                                            client.RF_getfreq()/10**6);
295     #Now we're ready to get packets.
296     while 1:
297         packet=None;
298         while packet==None:
299             #time.sleep(0.1);
300             packet=client.RF_rxpacket();
301         printpacket(packet);
302         sys.stdout.flush();
303
304 if(sys.argv[1]=="snifftp"):
305     client.poke(0x00,0x00); #Stop nRF
306     client.poke(0x01,0x00); #Disable Shockburst
307     client.poke(0x02,0x01); #Set RX Pipe 0
308     
309     client.RF_setfreq((2400+0x29) * 10**6);
310     client.poke(0x06,0x00); #1Mbps
311     client.poke(0x07,0x78); #Reset status register
312     
313     client.RF_setmaclen(3); # SETUP_AW for 3-byte addresses.
314     client.RF_setsmac(0x123456);
315     client.RF_setpacketlen(4);
316     
317     #Power radio, prime for RX, two-byte checksum.
318     client.poke(0x00,0x70|0x03|0x04|0x08);
319     
320     print "Listening as %010x on %i MHz" % (client.RF_getsmac(),
321                                            client.RF_getfreq()/10**6);
322     #Now we're ready to get packets.
323     while 1:
324         packet=None;
325         while packet==None:
326             #time.sleep(0.1);
327             packet=client.RF_rxpacket();
328         printpacket(packet);
329         sys.stdout.flush();
330
331 if(sys.argv[1]=="hosttp"):
332     client.poke(0x00,0x00); #Stop nRF
333     client.poke(0x01,0x00); #Disable Shockburst
334     client.poke(0x02,0x01); #Set RX Pipe 0
335     
336     chan=0x29;
337
338     client.RF_setfreq((2400+chan) * 10**6);
339     client.poke(0x06,0x00); #1Mbps
340     client.poke(0x07,0x78); #Reset status register
341     
342     client.RF_setmaclen(3); # SETUP_AW for 3-byte addresses.
343     client.RF_setsmac(0x123456);
344     client.RF_setpacketlen(4);
345     
346     #Power radio, prime for RX, two-byte checksum.
347     client.poke(0x00,0x70|0x03|0x04|0x08);
348     
349     print "Listening as %010x on %i MHz" % (client.RF_getsmac(),
350                                            client.RF_getfreq()/10**6);
351     #Now we're ready to get packets.
352     while 1:
353         packet=None;
354         while packet==None:
355             packet=client.RF_rxpacket();
356         mac=((ord(packet[0])<<16)+
357              (ord(packet[1])<<8)+
358              ord(packet[2]));
359         key=packet[3];
360         print "%c from %06x" % (key,mac);
361         sys.stdout.flush();
362
363 if(sys.argv[1]=="sniff"):
364     #client.poke(0x00,0x00); #Stop nRF
365     client.poke(0x07,0x78); #Reset status register
366     
367     #Power radio, prime for RX, checksum.
368     client.poke(0x00,0x70|0x03|0x08);
369     
370     print "Listening as %010x on %i MHz" % (client.RF_getsmac(),
371                                            client.RF_getfreq()/10**6);
372     #Now we're ready to get packets.
373     
374     while 1:
375         packet=None;
376         while packet==None:
377             #time.sleep(0.1);
378             packet=client.RF_rxpacket();
379         printpacket(packet);
380         sys.stdout.flush();
381 if(sys.argv[1]=="explore"):
382     #client.poke(0x00,0x00); #Stop nRF
383     client.poke(0x07,0x78); #Reset status register
384     
385     #Power radio, prime for RX, no checksum.
386     client.poke(0x00,0x70|0x03);
387     
388     #Set packet length of 32.
389     #Without checksums, extra data will mix in.
390     client.RF_setpacketlen(32);
391     client.RF_setmaclen(3); # shortest address length
392     
393     #Now we're ready to get packets.
394     for smac in [0x0102030201, 0]:
395         client.RF_setsmac(smac);
396         for chan in range(0,0x80):
397             client.RF_setfreq((2400+chan) * 10**6);
398             time.sleep(1);
399             packet=client.RF_rxpacket();
400             if packet!=None:
401                 print "Listening as %010x on %i MHz" % (client.RF_getsmac(),
402                                                         client.RF_getfreq()/10**6);
403                 printpacket(packet);
404                 sys.stdout.flush();