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         i=i+1;
  20         if i>ord(packet[1])+2: break;
  21         if i<2: continue;
  22         s="%s %02x" % (s,ord(foo));
  23     print "%s" % s;
  24
  25
  26 if(len(sys.argv)==1):
  27     print "Usage: %s verb [objects]\n" % sys.argv[0];
  28     print "%s regs" % sys.argv[0];
  29     print "%s test" % sys.argv[0];
  30     print "%s peek 0x$start [0x$stop]" % sys.argv[0];
  31     print "%s poke 0x$adr 0x$val" % sys.argv[0];
  32     sys.exit();
  33
  34 #Initialize FET and set baud rate
  35 client=GoodFETCCSPI();
  36 client.serInit()
  37
  38
  39 client.setup();
  40
  41 #Dummy read.
  42 #Might read as all ones if chip has a startup delay.
  43
  44 if(sys.argv[1]=="info"):
  45     print "Found %s" % client.identstr();
  46     print "Freq: %05.05f MHz" % (client.RF_getfreq()/(10**6));
  47 if(sys.argv[1]=="regs"):
  48     for adr in range(0x10,0x40): #*1024):
  49         val=client.peek(adr);
  50         print "%04x:=0x%04x" % (adr,val);
  51 if(sys.argv[1]=="test"):
  52     data=client.trans([0x20, 0xde, 0xad]);
  53     print "%02x %02x" % (ord(data[1]), ord(data[2]));
  54     data=client.trans([0x40|0x20, 0xde, 0xad]);
  55     print "%02x %02x" % (ord(data[1]), ord(data[2]));
  56 if(sys.argv[1]=="rssi"):
  57     if len(sys.argv)>2:
  58         client.RF_setfreq(eval(sys.argv[2]));
  59     print "Listening on %f MHz." % (client.RF_getfreq()/10.0**6);
  60
  61     client.strobe(0x02); #Calibrate
  62     time.sleep(1);
  63
  64     while 1:
  65         client.CC_RFST_RX();
  66         #client.strobe(0x03); #SRXON
  67         rssi=client.RF_getrssi();
  68         #client.CC_RFST_IDLE(); #idle
  69         time.sleep(0.01);
  70         string="";
  71         for foo in range(0,rssi>>2):
  72             string=("%s."%string);
  73         print "%02x %04i %s" % (rssi,rssi, string);
  74 if(sys.argv[1]=="sniff"):
  75     #if len(sys.argv)>2:
  76         #print "Set MAC to %s" % sys.argv[2];
  77         #client.tune(sys.argv[2]);
  78         #client.RF_setmaclen(5);
  79
  80     client.CC_RFST_RX();
  81     print "Listening as %010x on %i MHz" % (client.RF_getsmac(),
  82                                             client.RF_getfreq()/10**6);
  83     #Now we're ready to get packets.
  84
  85     while 1:
  86         packet=None;
  87         while packet==None:
  88             #time.sleep(0.1);
  89             packet=client.RF_rxpacket();
  90         printpacket(packet);
  91         sys.stdout.flush();
  92 if(sys.argv[1]=="peek"):
  93     start=0x0000;
  94     if(len(sys.argv)>2):
  95         start=int(sys.argv[2],16);
  96     stop=start;
  97     if(len(sys.argv)>3):
  98         stop=int(sys.argv[3],16);
  99     print "Peeking from %04x to %04x." % (start,stop);
 100     while start<=stop:
 101         print "%04x: 0x%04x" % (start,client.peek(start));
 102         start=start+1;
 103 if(sys.argv[1]=="poke"):
 104     start=0x0000;
 105     val=0x00;
 106     if(len(sys.argv)>2):
 107         start=int(sys.argv[2],16);
 108     if(len(sys.argv)>3):
 109         val=int(sys.argv[3],16);
 110     print "Poking r%02x to become 0x%04x." % (start,val);
 111
 112     client.poke(start,val);
 113