X-Git-Url: http://git.rot13.org/?p=goodfet;a=blobdiff_plain;f=client%2FGoodFETNRF.py;h=74be559b800994770b6c1cfc29a61d3b967f3f4c;hp=0c62cdfd1777e88e11c50c0ae017a55c39f1d63e;hb=cf47ccc725fd13d7216400149190328c6077a010;hpb=1536c622c446a0928fb2e9e44cdc67e7cceb6e7e

diff --git a/client/GoodFETNRF.py b/client/GoodFETNRF.py
index 0c62cdf..74be559 100644
--- a/client/GoodFETNRF.py
+++ b/client/GoodFETNRF.py
@@ -26,9 +26,15 @@ class GoodFETNRF(GoodFET):
         self.writecmd(self.NRFAPP,0x00,len(data),data);
         return self.data;
     
-    def peek(self,reg,bytes=1):
+    def peek(self,reg,bytes=-1):
         """Read an NRF Register.  For long regs, result is flipped."""
         data=[reg,0,0,0,0,0];
+        
+        #Automatically calibrate the len.
+        if bytes==-1:
+            bytes=1;
+            if reg==0x0a or reg==0x0b or reg==0x10: bytes=5;
+        
         self.writecmd(self.NRFAPP,0x02,len(data),data);
         toret=0;
         for i in range(0,bytes):
@@ -40,29 +46,90 @@ class GoodFETNRF(GoodFET):
         for i in range(0,bytes):
             data=data+[(val>>(8*i))&0xFF];
         self.writecmd(self.NRFAPP,0x03,len(data),data);
-        if self.peek(reg,bytes)!=val:
+        if self.peek(reg,bytes)!=val and reg!=0x07:
             print "Warning, failed to set register %02x." %reg;
         return;
     
     def status(self):
         """Read the status byte."""
-        self.poke(0x07,0x78); #Reset status
         status=self.peek(0x07);
         print "Status=%02x" % status;
     
     #Radio stuff begins here.
-    def RF_freq(self,frequency):
+    def RF_setfreq(self,frequency):
         """Set the frequency in Hz."""
         
         #On the NRF24L01+, register 0x05 is the offset in
         #MHz above 2400.
         
-        mhz=frequency/1000000-2400;
-        print "Setting channel %i." % mhz 
-        self.poke(0x05,mhz);
+        chan=frequency/1000000-2400;
+        self.poke(0x05,chan);
+    def RF_getfreq(self):
+        """Get the frequency in Hz."""
+        
+        #On the NRF24L01+, register 0x05 is the offset in
+        #MHz above 2400.
+        
+        return (2400+self.peek(0x05))*10**6
+        self.poke(0x05,chan);
     def RF_getsmac(self):
         """Return the source MAC address."""
         
         #Register 0A is RX_ADDR_P0, five bytes.
         mac=self.peek(0x0A, 5);
         return mac;
+    def RF_setsmac(self,mac):
+        """Set the source MAC address."""
+        
+        #Register 0A is RX_ADDR_P0, five bytes.
+        self.poke(0x0A, mac, 5);
+        return mac;
+    def RF_gettmac(self):
+        """Return the target MAC address."""
+        
+        #Register 0x10 is TX_ADDR, five bytes.
+        mac=self.peek(0x10, 5);
+        return mac;
+    def RF_settmac(self,mac):
+        """Set the target MAC address."""
+        
+        #Register 0x10 is TX_ADDR, five bytes.
+        self.poke(0x10, mac, 5);
+        return mac;
+
+    def RF_rxpacket(self):
+        """Get a packet from the radio.  Returns None if none is waiting."""
+        if self.peek(0x07) & 0x40:
+            #Packet has arrived.
+            self.writecmd(self.NRFAPP,0x80,0,None); #RX Packet
+            data=self.data;
+            self.poke(0x07,0x40);#clear bit.
+            return data;
+        elif self.peek(0x07)==0:
+            self.writecmd(self.NRFAPP,0x82,0,None); #Flush
+            self.poke(0x07,0x40);#clear bit.
+        return None;
+    packetlen=16;
+    def RF_setpacketlen(self,len=16):
+        """Set the number of bytes in the expected payload."""
+        self.poke(0x11,len);
+        self.packetlen=len;
+    def RF_getpacketlen(self):
+        """Set the number of bytes in the expected payload."""
+        len=self.peek(0x11);
+        self.packetlen=len;
+        return len;
+    maclen=5;
+    def RF_getmaclen(self):
+        """Get the number of bytes in the MAC address."""
+        choices=["illegal", 3, 4, 5];
+        choice=self.peek(0x03)&3;
+        self.maclen=choices[choice];
+        return self.maclen;
+    def RF_setmaclen(self,len):
+        """Set the number of bytes in the MAC address."""
+        choices=["illegal", "illegal", "illegal", 
+                 1, 2, 3];
+        choice=choices[len];
+        self.poke(0x03,choice);
+        self.maclen=len;