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: %05.05f MHz" % (client.RF_getfreq()/(10**6));
  49 if(sys.argv[1]=="regs"):
  50     for adr in range(0x10,0x40): #*1024):
  51         val=client.peek(adr);
  52         print "%04x:=0x%04x" % (adr,val);
  53 if(sys.argv[1]=="test"):
  54     data=client.trans([0x20, 0xde, 0xad]);
  55     print "%02x %02x" % (ord(data[1]), ord(data[2]));
  56     data=client.trans([0x40|0x20, 0xde, 0xad]);
  57     print "%02x %02x" % (ord(data[1]), ord(data[2]));
  58 if(sys.argv[1]=="rssi"):
  59     if len(sys.argv)>2:
  60         client.RF_setfreq(eval(sys.argv[2]));
  61     print "Listening on %f MHz." % (client.RF_getfreq()/10.0**6);
  62
  63     client.strobe(0x02); #Calibrate
  64     time.sleep(1);
  65
  66     while 1:
  67         client.CC_RFST_RX();
  68         #client.strobe(0x03); #SRXON
  69         rssi=client.RF_getrssi();
  70         #client.CC_RFST_IDLE(); #idle
  71         time.sleep(0.01);
  72         string="";
  73         for foo in range(0,rssi>>2):
  74             string=("%s."%string);
  75         print "%02x %04i %s" % (rssi,rssi, string);
  76 if(sys.argv[1]=="sniff"):
  77     if len(sys.argv)>2:
  78         freq=eval(sys.argv[2]);
  79         if freq>100:
  80             client.RF_setfreq(freq);
  81         else:
  82             print "Channels not yet supported."
  83     client.CC_RFST_RX();
  84     print "Listening as %010x on %i MHz" % (client.RF_getsmac(),
  85                                             client.RF_getfreq()/10**6);
  86     #Now we're ready to get packets.
  87
  88     while 1:
  89         packet=None;
  90         while packet==None:
  91             #time.sleep(0.1);
  92             packet=client.RF_rxpacket();
  93         printpacket(packet);
  94         sys.stdout.flush();
  95 if(sys.argv[1]=="peek"):
  96     start=0x0000;
  97     if(len(sys.argv)>2):
  98         start=int(sys.argv[2],16);
  99     stop=start;
 100     if(len(sys.argv)>3):
 101         stop=int(sys.argv[3],16);
 102     print "Peeking from %04x to %04x." % (start,stop);
 103     while start<=stop:
 104         print "%04x: 0x%04x" % (start,client.peek(start));
 105         start=start+1;
 106 if(sys.argv[1]=="poke"):
 107     start=0x0000;
 108     val=0x00;
 109     if(len(sys.argv)>2):
 110         start=int(sys.argv[2],16);
 111     if(len(sys.argv)>3):
 112         val=int(sys.argv[3],16);
 113     print "Poking r%02x to become 0x%04x." % (start,val);
 114
 115     client.poke(start,val);
 116