9c39e4d7f007d7c8679142e669ecc75c0a1753c4
[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 printmspacket(packet,offset=1):
34     keyword=client.RF_getsmac();
35     #print "keyword=%010x" % key;
36     key=[];
37     for foo in range(0,5):
38         key.append(keyword&0xFF);
39         keyword=(keyword>>8);
40         #print "Keybyte %02x" % key[foo];
41     i=0;
42     s="";
43     for foo in packet:
44         if i>=4:
45             s="%s %02x" % (s,ord(foo)^key[(i+offset)%5]);
46         else:
47             s="%s %02x" % (s,ord(foo));
48         i=i+1;
49     print "%s" % (s);
50 def printconfig():
51     print "Encoding %s" % client.RF_getenc();
52     print "Freq    %10i MHz" % (client.RF_getfreq()/10**6);
53     print "Rate    %10i kbps" % (client.RF_getrate()/1000);
54     print "PacketLen %02i bytes" % client.RF_getpacketlen();
55     #print "MacLen    %2i bytes" % client.RF_getmaclen();
56     print "SMAC  0x%010x" % client.RF_getsmac();
57     print "TMAC  0x%010x" % client.RF_gettmac();
58
59
60 if(len(sys.argv)==1):
61     print "Usage: %s verb [objects]\n" % sys.argv[0];
62     print "%s info" % sys.argv[0];
63     print "%s test" % sys.argv[0];
64     print "%s regs" % sys.argv[0];
65     print "%s regbits" % sys.argv[0];
66     print "%s pyregs" % sys.argv[0];
67     print "";
68     print "%s tune aa|55,mac,r5,r6\n\tTunes to a configuration." % sys.argv[0];
69     print "%s sniff\n\tSniffs packets by current config." % sys.argv[0];
70     print "%s sniffob\n\tSniffs OpenBeacon traffic." % sys.argv[0];
71     print "%s snifftp\n\tSniffs Turning Point Clicker traffic." % sys.argv[0];
72     print "%s sniffsf\n\tSniffs SparkFun Dongle traffic." % sys.argv[0];
73     print "";
74     print "%s sniffmacs \n\tSniffs for MAC addresses on the present channel." % sys.argv[0];
75     print "%s sniffprom [0xaa|0x55]\n\tSniffs promiscuously for a preamble of 0xAA or 0x55" % sys.argv[0];
76     print "%s autotune\n\tSearches for a valid destination address." % sys.argv[0];
77     print "";
78     print "%s sniffskybrake\n\tSniffs skybrake. [broken?]" % sys.argv[0];
79     print "%s sniffmskb\n\tSniffs MS KB. [broken?]" % sys.argv[0];
80     
81     print "%s hosttp\n\tHosts Turning Point Clicker traffic." % sys.argv[0];
82
83     print "%s carrier [freq]\n\tHolds a carrier on [freq] Hz." % sys.argv[0];
84     sys.exit();
85
86 #Initialize FET and set baud rate
87 client=GoodFETNRF();
88 client.serInit()
89
90 client.NRFsetup();
91
92 if(sys.argv[1]=="info"):
93     printconfig();
94
95 if(sys.argv[1]=="test"):
96     print "Old registers:"
97     printconfig();
98     
99     # Set PWR_UP=1 and PRIM_RX=0 in CONFIG.
100     client.poke(0x00,2);
101     #Delay of 1.5ms by round-trip.
102     
103     print "\n\n";
104     
105     #Try all data rates
106     for foo in [250*10**3,
107                 1*10**6,
108                 2*10**6]:
109         client.RF_setrate(foo);
110         if(client.RF_getrate()!=foo):
111             print "ERROR Rate %i not supported.  Got %i instead." % (foo,
112                                                                      client.RF_getrate());
113     
114     print "\n\n";
115     client.poke(0x0A,0xDEADBEEF,5);
116     #print "SMAC set to %010x" % client.RF_getsmac();
117     if client.RF_getsmac()!=0xdeadbeef:
118         print "ERROR: Failed to set MAC address.";
119     print "Final registers:"
120     printconfig();
121
122 if(sys.argv[1]=="carrier"):
123     if len(sys.argv)>2:
124         client.RF_setfreq(eval(sys.argv[2]));
125     client.RF_carrier();
126     printconfig();
127     print "\nHolding a carrier wave.";
128     while(1):
129         time.sleep(1);
130
131 if(sys.argv[1]=="tune"):
132     if len(sys.argv)>2:
133         client.tune(sys.argv[2]);
134     else:
135         print "Specify a tuning, such as 'aa,c78c65805e,14,09'";
136 if(sys.argv[1]=="regs"):
137     for r in range(0,0x20):
138         print "r[0x%02x]=0x%010x //%16s " % (r,client.peek(r),regnames[r]);
139 if(sys.argv[1]=="pyregs"):
140     for r in range(0,0x20):
141         print "client.set(0x%02x,0x%010x); #%16s " % (r,client.peek(r),regnames[r]);
142
143 if(sys.argv[1]=="peek"):
144     start=0x0000;
145     if(len(sys.argv)>2):
146         start=int(sys.argv[2],16);
147     stop=start;
148     if(len(sys.argv)>3):
149         stop=int(sys.argv[3],16);
150     print "Peeking from %02x to %02x." % (start,stop);
151     while start<=stop:
152         print "%02x: %010x" % (start,client.peek(start));
153         start=start+1;
154 if(sys.argv[1]=="poke"):
155     start=0x0000;
156     val=0x00;
157     if(len(sys.argv)>2):
158         start=int(sys.argv[2],16);
159     if(len(sys.argv)>3):
160         val=int(sys.argv[3],16);
161     print "Poking %02x to become %010x." % (start,val);
162     
163     client.poke(start,val);
164     print "Poked to %04x" % client.peek(start);
165
166 if(sys.argv[1]=="sniffob"):
167     #Reversal of transmitter code from nRF_CMD.c of OpenBeacon
168     #TODO remove all poke() calls.
169     
170     client.poke(0x00,0x00); #Stop nRF
171     client.poke(0x01,0x00); #Disable Shockburst
172     client.poke(0x02,0x01); #Set RX Pipe 0
173     
174     client.RF_setfreq(2481 * 10**6);
175     client.poke(0x06,0x09); #2MBps, -18dBm in RF_SETUP
176     client.poke(0x07,0x78); #Reset status register
177     
178     #OpenBeacon defines these in little endian as follows.
179     client.RF_setmaclen(5); # SETUP_AW for 5-byte addresses.
180     #0x01, 0x02, 0x03, 0x02, 0x01
181     client.RF_setsmac(0x0102030201);
182     #'O', 'C', 'A', 'E', 'B'
183     client.RF_settmac(0x424541434F);
184     
185     #Set packet length of 16.
186     client.RF_setpacketlen(16);
187     
188     #Power radio, prime for RX, one-byte checksum.
189     client.poke(0x00,0x70|0x03|0x08); #0x08 for one byte, 0x04 for two.
190     
191     print "Listening as %010x on %i MHz" % (client.RF_getsmac(),
192                                            client.RF_getfreq()/10**6);
193     #Now we're ready to get packets.
194     while 1:
195         packet=None;
196         while packet==None:
197             #time.sleep(0.1);
198             packet=client.RF_rxpacket();
199         printpacket(packet);
200         sys.stdout.flush();
201
202 if(sys.argv[1]=="regbits"):
203     print "Scanning registers to determine which bits are valid."
204     regbits=range(0,0x30);
205     for r in range(0,0x30):
206         old=client.peek(r);
207         #Which bits can be set?
208         client.poke(r,0xFF);
209         ones=client.peek(r);
210         #Which bits can be clear?
211         client.poke(r,0x00);
212         zeroes=client.peek(r);
213         regbits[r]=(ones & (~zeroes));
214     for r in range(0,0x30):
215         if regbits[r]!=0:
216             print "r[0x%02x] masked %02x // %s" % (r,regbits[r], regnames[r]);
217 if(sys.argv[1]=="sniffprom"):
218     #Reversal of transmitter code from nRF_CMD.c of OpenBeacon
219     #TODO remove all poke() calls.
220     
221     client.poke(0x00,0x00); #Stop nRF
222     client.poke(0x01,0x00); #Disable Shockburst
223     client.poke(0x02,0x01); #Set RX Pipe 0
224     
225     #client.RF_setfreq(2481 * 10**6);
226     #client.poke(0x06,0x09); #2MBps, -18dBm in RF_SETUP
227     client.poke(0x07,0x78); #Reset status register
228     
229     #OpenBeacon defines these in little endian as follows.
230     client.RF_setmaclen(2); # SETUP_AW for shortest
231     
232     #It's better to have a known fragment, when one is available.
233     #client.RF_setsmac(0x00AA);
234     #client.RF_setsmac(0x0055);
235     
236     #Should end in 55 or AA depending upon the packet.
237     tail=0x55
238     if(len(sys.argv)>2):
239         tail=int(sys.argv[2],16);
240     else:
241         print "Please specify a tail of 0xAA or 0x55.";
242         sys.exit(1);
243     client.RF_setsmac(tail);
244     
245     #Longest length.
246     client.RF_setpacketlen(32);
247     
248     #Power radio, prime for RX, no checksum
249     client.poke(0x00,0x70|0x03); #0x08 for checksum, 0x04 for two.
250     
251     print "Listening as %010x on %i MHz" % (client.RF_getsmac(),
252                                            client.RF_getfreq()/10**6);
253     #Now we're ready to get packets.
254     while 1:
255         packet=None;
256         while packet==None:
257             #time.sleep(0.1);
258             packet=client.RF_rxpacket();
259         printpacket(packet);
260         sys.stdout.flush();
261
262 class AutoTuner():
263     """This guesses addresses by searching through packets."""
264     addresses={};
265     client=None;
266     
267     #Limits on search space, because you usually know what you're looking for.
268     rate=False;
269     chan=False;
270     sync=False;
271     startch=0; #Useful for forcing an early match.
272     def init(self,goodfet,
273              rate=True,chan=True,sync=True):
274         """Initializes a link to the GoodFET for autotuning."""
275         self.client=goodfet;
276         self.rate=rate;
277         self.chan=chan;
278         self.sync=sync;
279         
280         client.poke(0x00,0x00); #Stop nRF
281         client.poke(0x01,0x00); #Disable Shockburst
282         client.poke(0x02,0x01); #Set RX Pipe 0
283         
284         #Disable shockburst.
285         client.poke(0x1C,0x00);
286         client.poke(0x1D,0x00);
287     
288         client.RF_setmaclen(2); # SETUP_AW for shortest
289         
290         #historic
291         #client.RF_setsmac(0x00AA);
292         #client.RF_setsmac(0x0055);
293         
294         client.poke(0x00,0x70|0x03); #prime radio.
295         
296         return;
297     def packetaddr(self,packet,justmac=False):
298         """Returns a loaded packet address, including channel and rate."""
299         
300         sync=self.client.RF_getsmac()&0xFF;
301         
302         mac="";
303         #MAC,RF_CH,RATE
304         for i in range(0,5):
305             mac="%s%02x" % (mac,ord(packet[i]));
306         if justmac:
307             return mac;
308         ch=self.client.peek(0x05);
309         rate=self.client.peek(0x06);
310         return "%02x,%s,%02x,%02x" % (
311             sync,mac,ch,rate);
312     def validmac(self,packet):
313         sync=self.client.RF_getsmac()&0xFF;
314         mac=self.packetaddr(packet,justmac=True);
315         if (ord(packet[0])&0x80)^(sync&0x80):
316             #print "%02x%02x invalid entry." % (sync,ord(packet[0]));
317             #This is a special kind of failure.  Freq is probably right, but MAC is wrong.
318             return False;
319         if mac=='5555555555' or mac=='aaaaaaaaaa' or mac=='0000000000':
320             return False;
321         return True;
322         
323     def handle(self,packet):
324         """Handles a packet."""
325         #printpacket(packet);
326         
327         if not self.validmac(packet):
328             #print "Dropped packet from %s" % self.packetaddr(packet,justmac=True);
329             #printpacket(packet);
330             return;
331         
332         addr=self.packetaddr(packet);
333         
334         #Increment the address count.
335         count=0;
336         try:
337             count=self.addresses[addr];
338         except:
339             pass;
340         self.addresses[addr]=count+1;
341         rate=count*1.0/len(self.addresses);
342         if self.addresses[addr]>1 or rate>0.01:
343             print "'%s' looks valid\t%i\t%0.5f" % (
344                 addr,count,rate);
345         return addr;
346     tunecount=0;
347     def selftune(self,threshold=2,forever=False,
348                  delay=5.0):
349         """Tunes to the first strong signal.
350         It's important that this not get triggered by false positives."""
351         
352         while 1:
353             self.retune();
354             start=time.mktime(time.localtime());
355             sys.stdout.flush();
356             while (time.mktime(time.localtime())-start) < delay:
357                 packet=None;
358                 while packet==None:
359                     packet=client.RF_rxpacket();
360                 addr=guesser.handle(packet);
361                 try:
362                     count=self.addresses[addr];
363                 except:
364                     count=0;
365                 if count>threshold and forever==False:
366                     #Tune it in here?
367                     client.tune(addr);
368                     return addr;
369             
370         
371     def retune(self):
372         """Tunes to another channel or preamble looking for the next packet."""
373         count=self.tunecount;
374         self.tunecount=count+1;
375         
376         #Swap the SYNC value most often.
377         if self.sync:
378             sync=0x00AA;
379             if count&1:
380                 sync=0x0055;
381             self.client.RF_setsmac(sync);
382             count=(count>>1);
383         
384         if self.rate:
385             #Then the data rate.
386             rate=0;
387             
388             #This swaps between 1Mbps and 2Mbps.
389             #TODO add support for 256kbps, if anyone uses it.
390             if count&1:
391                 rate=rate|0x08;
392             #print "Setting rate to 0x%02x" % rate;
393             if(rate==0x20):
394                 rate=0x08;
395             self.client.poke(0x06,rate);
396             count=(count>>1);
397         
398         if self.chan:
399             self.client.poke(0x05,
400                              (count+self.startch)&0x7f);
401             print "Tuned to %i MHz" % (
402                 self.client.RF_getfreq()
403                 /(10**6));
404         #Grab two packets to clear buffers.
405         #Should retune only after a few packets to reduce this delay.
406         packet=client.RF_rxpacket();
407         packet=client.RF_rxpacket();
408         return;
409         
410         
411 if(sys.argv[1]=="autotune"):
412     #Reversal of transmitter code from nRF_CMD.c of OpenBeacon
413     #TODO remove all poke() calls.
414     guesser=AutoTuner();
415     guesser.init(client,rate=True,sync=True,chan=True);
416     
417     
418     #client.RF_setfreq(2481 * 10**6);
419     client.poke(0x06,0x09); #2MBps, -18dBm in RF_SETUP
420     client.poke(0x07,0x78); #Reset status register
421     
422     #This is determined by the MAC, which we don't yet know.
423     
424     #Longest length.
425     client.RF_setpacketlen(32);
426     
427     #Power radio, prime for RX, no checksum
428     client.poke(0x00,0x70|0x03); #0x08 for checksum, 0x04 for two.
429     
430     print "Autotuning on %i MHz" % (
431         client.RF_getfreq()/10**6);
432     print "sync,mac,r5,r6";
433     #Now we're ready to get packets.
434     guesser.selftune(threshold=2,
435                      forever=True);
436
437 if(sys.argv[1]=="sniffmacs"):
438     #Reversal of transmitter code from nRF_CMD.c of OpenBeacon
439     #TODO remove all poke() calls.
440     guesser=AutoTuner();
441     guesser.init(client,rate=False,sync=True,chan=False);
442     
443     #Longest length.
444     client.RF_setpacketlen(32);
445     
446     #Power radio, prime for RX, no checksum
447     client.poke(0x00,0x70|0x03); #0x08 for checksum, 0x04 for two.
448     
449     print "Holding autotune on %i MHz" % (
450         client.RF_getfreq()/10**6);
451     print "sync,mac,r5,r6";
452     #Now we're ready to get packets.
453     guesser.selftune(threshold=2,
454                      forever=True);
455
456 if(sys.argv[1]=="sniffmskb"):
457     #MSWK 3000 v2.0
458     #TODO remove all poke() calls.
459     
460     client.poke(0x00,0x00); #Stop nRF
461     client.poke(0x01,0x00); #Disable Shockburst
462     client.poke(0x02,0x01); #Set RX Pipe 0
463     
464     client.poke(0x06,0x09); #2MBps, -18dBm in RF_SETUP
465     client.poke(0x07,0x78); #Reset status register
466     
467     #This is the address of a specific keyboard.
468     #Other keyboards will be different.
469     
470     client.RF_setmaclen(5);
471     
472     #Known pairs.  The channel and the low address bytes must match.
473     #client.RF_setfreq((2400+0x13) * 10**6);
474     #client.RF_setsmac(0xc00a3598cd);
475     
476     #client.RF_setfreq((2400+0x15) * 10**6);
477     #client.RF_setsmac(0xc10446facd);
478     
479     #Mac packet length, illegally 0-length address field.
480     client.RF_setpacketlen(16);
481     
482     #aa,c00a3598cd,13,09
483     if len(sys.argv)>2:
484         client.tune(sys.argv[2]);
485     else:
486         
487         print "Searching for a keyboard.";
488         
489         guesser=AutoTuner();
490         guesser.init(client, rate=False, sync=True, chan=True);
491         guesser.selftune(threshold=4,forever=False,
492                          delay=10.0);
493     
494     client.poke(0x00,0x00); #Stop nRF
495     client.poke(0x01,0x00); #Disable Shockburst
496     client.poke(0x02,0x01); #Set RX Pipe 0
497     client.RF_setmaclen(5);
498     
499     #Finally, dynamic payload lengths need to be enabled.
500     #client.poke(0x01,0x01);
501     client.poke(0x1C,0x01);
502     client.poke(0x1D,0x06);
503     
504     client.poke(0x00,0x70|0x03); #prime radio.
505     print "Listening as %010x on %i MHz" % (client.RF_getsmac(),
506                                            client.RF_getfreq()/10**6);
507     #Now we're ready to get packets.
508     while 1:
509         packet=None;
510         while packet==None:
511             #time.sleep(1);
512             packet=client.RF_rxpacket();
513             #print ".";
514         printmspacket(packet);
515         sys.stdout.flush();
516
517 if(sys.argv[1]=="sniffant"):
518     #Prototyped on Garmin device.
519     #Channel hopping is pretty damned fast, hard to follow.
520     #This doesn't really work yet, still experimenting.
521     
522     #Might be more effective to sniff knowing the MFG ID and Dev. ID,
523     #as these predict a lot of the MAC address.
524     
525     client.poke(0x00,0x00); #Stop nRF
526     client.poke(0x01,0x00); #Disable Shockburst
527     client.poke(0x02,0x01); #Set RX Pipe 0
528     
529     client.poke(0x05,57); #broadcast-only channel
530     client.poke(0x06,0x00); #1MBps
531     client.poke(0x07,0x78); #Reset status register
532     
533     #Is this appropriate?  Might be 3.
534     client.RF_setmaclen(5);
535     
536     
537     #Mac packet length, illegally 0-length address field.
538     client.RF_setpacketlen(16);
539     
540     if len(sys.argv)>2:
541         client.tune(sys.argv[2]);
542     else:
543         
544         print "Searching for ANT+.";
545         
546         guesser=AutoTuner();
547         guesser.init(client, rate=False, sync=True, chan=True);
548         guesser.selftune(threshold=2,forever=False,
549                          delay=9.0);
550     
551     client.poke(0x00,0x00); #Stop nRF
552     client.poke(0x01,0x00); #Disable Shockburst
553     client.poke(0x02,0x01); #Set RX Pipe 0
554     client.RF_setmaclen(5);
555     
556     
557     client.poke(0x00,0x70|0x03); #prime radio.
558     print "Dumping ANT as %010x on %i MHz" % (client.RF_getsmac(),
559                                               client.RF_getfreq()/10**6);
560     #Now we're ready to get packets.
561     while 1:
562         packet=None;
563         while packet==None:
564             #time.sleep(1);
565             packet=client.RF_rxpacket();
566             #print ".";
567         printpacket(packet);
568         sys.stdout.flush();
569
570
571
572 if(sys.argv[1]=="sniffskybrake"):
573     #Reversal of transmitter code from nRF_CMD.c of OpenBeacon
574     #TODO remove all poke() calls.
575     
576     client.poke(0x00,0x00); #Stop nRF
577     client.poke(0x01,0x00); #Disable Shockburst
578     client.poke(0x02,0x01); #Set RX Pipe 0
579     
580     client.RF_setfreq(2439 * 10**6);
581     client.poke(0x06,0x00); #1MBps
582     client.poke(0x07,0x78); #Reset status register
583     
584     #OpenBeacon defines these in little endian as follows.
585     client.RF_setmaclen(2); # SETUP_AW for 3-byte addresses.
586     #0x01, 0x02, 0x03, 0x02, 0x01
587     
588     client.RF_setsmac(0x070700d2c4); #reversed
589     
590     #client.RF_setsmac(0xd2c4);
591     #client.RF_setsmac(0);
592     
593     #Mac packet length, illegally 0-length address field.
594     client.RF_setpacketlen(32);
595     
596     #Power radio, prime for RX, one-byte checksum.
597     client.poke(0x00,0x70|0x03); #0x08 for one byte, 0x04 for two.
598     
599     print "Listening as %010x on %i MHz" % (client.RF_getsmac(),
600                                            client.RF_getfreq()/10**6);
601     print "%i byte mac match." % client.RF_getmaclen();
602     #Now we're ready to get packets.
603     while 1:
604         packet=None;
605         while packet==None:
606             #time.sleep(0.1);
607             packet=client.RF_rxpacket();
608         printpacket(packet);
609         sys.stdout.flush();
610
611 if(sys.argv[1]=="sniffsf"):
612     #Reversal of transmitter code from nRF_CMD.c of OpenBeacon
613     #TODO remove all poke() calls.
614     
615     client.poke(0x00,0x00); #Stop nRF
616     client.poke(0x01,0x00); #Disable Shockburst
617     client.poke(0x02,0x01); #Set RX Pipe 0
618     
619     client.RF_setfreq(2402 * 10**6);
620     client.poke(0x06,0x07); #1Mbps
621     client.poke(0x07,0x78); #Reset status register
622     
623     #OpenBeacon defines these in little endian as follows.
624     client.RF_setmaclen(5); # SETUP_AW for 5-byte addresses.
625     client.RF_setsmac(0xe7e7e7e7e7);
626     client.RF_settmac(0xe7e7e7e7e7);
627     
628     #Set packet length of 16.
629     client.RF_setpacketlen(4);
630     
631     #Power radio, prime for RX, one-byte checksum.
632     client.poke(0x00,0x70|0x03|0x08); #0x08 for one byte, 0x04 for two.
633     
634     print "Listening as %010x on %i MHz" % (client.RF_getsmac(),
635                                            client.RF_getfreq()/10**6);
636     #Now we're ready to get packets.
637     while 1:
638         packet=None;
639         while packet==None:
640             #time.sleep(0.1);
641             packet=client.RF_rxpacket();
642         printpacket(packet);
643         sys.stdout.flush();
644
645 if(sys.argv[1]=="snifftp"):
646     client.poke(0x00,0x00); #Stop nRF
647     client.poke(0x01,0x00); #Disable Shockburst
648     client.poke(0x02,0x01); #Set RX Pipe 0
649     
650     client.RF_setfreq((2400+0x29) * 10**6);
651     client.poke(0x06,0x00); #1Mbps
652     client.poke(0x07,0x78); #Reset status register
653     
654     client.RF_setmaclen(3); # SETUP_AW for 3-byte addresses.
655     client.RF_setsmac(0x123456);
656     client.RF_setpacketlen(4);
657     
658     #Power radio, prime for RX, two-byte checksum.
659     client.poke(0x00,0x70|0x03|0x04|0x08);
660     
661     print "Listening as %010x on %i MHz" % (client.RF_getsmac(),
662                                            client.RF_getfreq()/10**6);
663     #Now we're ready to get packets.
664     while 1:
665         packet=None;
666         while packet==None:
667             #time.sleep(0.1);
668             packet=client.RF_rxpacket();
669         printpacket(packet);
670         sys.stdout.flush();
671
672 if(sys.argv[1]=="hosttp"):
673     client.poke(0x00,0x00); #Stop nRF
674     client.poke(0x01,0x00); #Disable Shockburst
675     client.poke(0x02,0x01); #Set RX Pipe 0
676     
677     chan=0x29;
678
679     client.RF_setfreq((2400+chan) * 10**6);
680     client.poke(0x06,0x00); #1Mbps
681     client.poke(0x07,0x78); #Reset status register
682     
683     client.RF_setmaclen(3); # SETUP_AW for 3-byte addresses.
684     client.RF_setsmac(0x123456);
685     client.RF_setpacketlen(4);
686     
687     #Power radio, prime for RX, two-byte checksum.
688     client.poke(0x00,0x70|0x03|0x04|0x08);
689     
690     print "Listening as %010x on %i MHz" % (client.RF_getsmac(),
691                                            client.RF_getfreq()/10**6);
692     #Now we're ready to get packets.
693     while 1:
694         packet=None;
695         while packet==None:
696             packet=client.RF_rxpacket();
697         mac=((ord(packet[0])<<16)+
698              (ord(packet[1])<<8)+
699              ord(packet[2]));
700         key=packet[3];
701         print "%c from %06x" % (key,mac);
702         sys.stdout.flush();
703
704 if(sys.argv[1]=="sniff"):
705     if len(sys.argv)>2:
706         print "Set MAC to %s" % sys.argv[2];
707         client.tune(sys.argv[2]);
708         client.RF_setmaclen(5);
709         
710     #client.poke(0x00,0x00); #Stop nRF
711     client.poke(0x07,0x78); #Reset status register
712     
713     #Power radio, prime for RX, no checksum.
714     client.poke(0x00,0x70|0x03);
715     #Mac packet length.
716     client.RF_setpacketlen(32);
717     #Mac length, reduced
718     #client.RF_setmaclen(3); # SETUP_AW for shortest
719     
720     print "Listening as %010x on %i MHz" % (client.RF_getsmac(),
721                                            client.RF_getfreq()/10**6);
722     #Now we're ready to get packets.
723     
724     while 1:
725         packet=None;
726         while packet==None:
727             #time.sleep(0.1);
728             packet=client.RF_rxpacket();
729         printpacket(packet);
730         sys.stdout.flush();
731 if(sys.argv[1]=="explore"):
732     #client.poke(0x00,0x00); #Stop nRF
733     client.poke(0x07,0x78); #Reset status register
734     
735     #Power radio, prime for RX, no checksum.
736     client.poke(0x00,0x70|0x03);
737     
738     #Set packet length of 32.
739     #Without checksums, extra data will mix in.
740     client.RF_setpacketlen(32);
741     client.RF_setmaclen(3); # shortest address length
742     
743     #Now we're ready to get packets.
744     for smac in [0x0102030201, 0]:
745         client.RF_setsmac(smac);
746         for chan in range(0,0x80):
747             client.RF_setfreq((2400+chan) * 10**6);
748             time.sleep(1);
749             packet=client.RF_rxpacket();
750             if packet!=None:
751                 print "Listening as %010x on %i MHz" % (client.RF_getsmac(),
752                                                         client.RF_getfreq()/10**6);
753                 printpacket(packet);
754                 sys.stdout.flush();