git.rot13.org Git - goodfet/blob - client/goodfet.ccspi

   1 #!/usr/bin/env python
   2
   3 #GoodFET Chipcon SPI Client
   4 # (C) 2011 Travis Goodspeed
   5
   6 #N.B.,
   7 #Might be CC2420 Specific
   8
   9 import sys;
  10 import binascii;
  11 import array, time;
  12
  13 from GoodFETCCSPI import GoodFETCCSPI;
  14
  15 def printpacket(packet):
  16     s="";
  17     i=0;
  18     for foo in packet:
  19         s="%s %02x" % (s,ord(foo));
  20     print "#%s" % s;
  21
  22
  23 if(len(sys.argv)==1):
  24     print "Usage: %s verb [objects]\n" % sys.argv[0];
  25     print "%s info" % sys.argv[0];
  26     print "%s regs" % sys.argv[0];
  27     print "%s test" % sys.argv[0];
  28     print "%s peek 0x$start [0x$stop]" % sys.argv[0];
  29     print "%s poke 0x$adr 0x$val" % sys.argv[0];
  30
  31     print "\n%s rssi" % sys.argv[0];
  32     print "%s sniff" % sys.argv[0];
  33
  34     sys.exit();
  35
  36 #Initialize FET and set baud rate
  37 client=GoodFETCCSPI();
  38 client.serInit()
  39
  40
  41 client.setup();
  42
  43 #Dummy read.
  44 #Might read as all ones if chip has a startup delay.
  45
  46 if(sys.argv[1]=="info"):
  47     print "Found   %s" % client.identstr();
  48     print "Freq:   %05f MHz" % (client.RF_getfreq()/(10**6));
  49     print "Status: %s" % client.status();
  50 if(sys.argv[1]=="regs"):
  51     for adr in range(0x10,0x40): #*1024):
  52         val=client.peek(adr);
  53         print "%04x:=0x%04x" % (adr,val);
  54 if(sys.argv[1]=="test"):
  55     data=client.trans([0x20, 0xde, 0xad]);
  56     print "%02x %02x" % (ord(data[1]), ord(data[2]));
  57     data=client.trans([0x40|0x20, 0xde, 0xad]);
  58     print "%02x %02x" % (ord(data[1]), ord(data[2]));
  59 if(sys.argv[1]=="rssi"):
  60     if len(sys.argv)>2:
  61         client.RF_setfreq(eval(sys.argv[2]));
  62     print "Listening on %f MHz." % (client.RF_getfreq()/10.0**6);
  63
  64     client.strobe(0x02); #Calibrate
  65     time.sleep(1);
  66
  67     while 1:
  68         client.CC_RFST_RX();
  69         #client.strobe(0x03); #SRXON
  70         rssi=client.RF_getrssi();
  71         #client.CC_RFST_IDLE(); #idle
  72         time.sleep(0.01);
  73         string="";
  74         for foo in range(0,rssi>>2):
  75             string=("%s."%string);
  76         print "%02x %04i %s" % (rssi,rssi, string);
  77 if(sys.argv[1]=="sniff"):
  78     #Promiscuous mode.
  79     client.RF_promiscuity(1);
  80
  81     if len(sys.argv)>2:
  82         freq=eval(sys.argv[2]);
  83         if freq>100:
  84             client.RF_setfreq(freq);
  85         else:
  86             print "Channels not yet supported."
  87     client.CC_RFST_RX();
  88     print "Listening as %010x on %i MHz" % (client.RF_getsmac(),
  89                                             client.RF_getfreq()/10**6);
  90     #Now we're ready to get packets.
  91
  92     while 1:
  93         packet=None;
  94         while packet==None:
  95             packet=client.RF_rxpacket();
  96         printpacket(packet);
  97         sys.stdout.flush();
  98 if(sys.argv[1]=="txtest"):
  99     if len(sys.argv)>2:
 100         freq=eval(sys.argv[2]);
 101         if freq>100:
 102             client.RF_setfreq(freq);
 103         else:
 104             print "Channels not yet supported."
 105     print "Transmitting as %010x on %i MHz" % (
 106         client.RF_getsmac(),
 107         client.RF_getfreq()/10**6);
 108
 109     while 1:
 110         client.RF_txpacket([0x0f, 0x01, 0x08, 0x82,
 111                             0xff, 0xff, 0xff, 0xff,
 112                             0x4d, 0x7d, 0x09, 0x00,
 113                             0x1f, 0x00, 0xc0]);
 114         print client.status();
 115         print;
 116
 117 if(sys.argv[1]=="peek"):
 118     start=0x0000;
 119     if(len(sys.argv)>2):
 120         start=int(sys.argv[2],16);
 121     stop=start;
 122     if(len(sys.argv)>3):
 123         stop=int(sys.argv[3],16);
 124     print "Peeking from %04x to %04x." % (start,stop);
 125     while start<=stop:
 126         print "%04x: 0x%04x" % (start,client.peek(start));
 127         start=start+1;
 128 if(sys.argv[1]=="poke"):
 129     start=0x0000;
 130     val=0x00;
 131     if(len(sys.argv)>2):
 132         start=int(sys.argv[2],16);
 133     if(len(sys.argv)>3):
 134         val=int(sys.argv[3],16);
 135     print "Poking r%02x to become 0x%04x." % (start,val);
 136
 137     client.poke(start,val);
 138