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]=="carrier"):
  47     if len(sys.argv)>2:
  48         client.RF_setfreq(eval(sys.argv[2]));
  49     while 1:
  50         client.RF_carrier();
  51     while(1):
  52         time.sleep(1);
  53
  54 if(sys.argv[1]=="modulated_spectrum"):
  55     if len(sys.argv)>2:
  56         client.RF_setfreq(eval(sys.argv[2]));
  57     while 1:
  58         client.RF_modulated_spectrum();
  59     while(1):
  60         time.sleep(1);
  61
  62 if(sys.argv[1]=="info"):
  63     print "Found   %s" % client.identstr();
  64     print "Freq:   %05f MHz" % (client.RF_getfreq()/(10**6));
  65     print "Status: %s" % client.status();
  66 if(sys.argv[1]=="regs"):
  67     for adr in range(0x10,0x40): #*1024):
  68         val=client.peek(adr);
  69         print "%04x:=0x%04x" % (adr,val);
  70 if(sys.argv[1]=="test"):
  71     data=client.trans([0x20, 0xde, 0xad]);
  72     print "%02x %02x" % (ord(data[1]), ord(data[2]));
  73     data=client.trans([0x40|0x20, 0xde, 0xad]);
  74     print "%02x %02x" % (ord(data[1]), ord(data[2]));
  75 if(sys.argv[1]=="rssi"):
  76     if len(sys.argv)>2:
  77         client.RF_setfreq(eval(sys.argv[2]));
  78     print "Listening on %f MHz." % (client.RF_getfreq()/10.0**6);
  79
  80     client.strobe(0x02); #Calibrate
  81     time.sleep(1);
  82
  83     while 1:
  84         client.CC_RFST_RX();
  85         #client.strobe(0x03); #SRXON
  86         rssi=client.RF_getrssi();
  87         #client.CC_RFST_IDLE(); #idle
  88         time.sleep(0.01);
  89         string="";
  90         for foo in range(0,rssi>>2):
  91             string=("%s."%string);
  92         print "%02x %04i %s" % (rssi,rssi, string);
  93 if(sys.argv[1]=="sniff"):
  94     #Promiscuous mode.
  95     client.RF_promiscuity(1);
  96
  97     if len(sys.argv)>2:
  98         freq=eval(sys.argv[2]);
  99         if freq>100:
 100             client.RF_setfreq(freq);
 101         else:
 102             print "Channels not yet supported."
 103     client.CC_RFST_RX();
 104     print "Listening as %010x on %i MHz" % (client.RF_getsmac(),
 105                                             client.RF_getfreq()/10**6);
 106     #Now we're ready to get packets.
 107
 108     while 1:
 109         packet=None;
 110         while packet==None:
 111             packet=client.RF_rxpacket();
 112         printpacket(packet);
 113         sys.stdout.flush();
 114 if(sys.argv[1]=="txtest"):
 115     if len(sys.argv)>2:
 116         freq=eval(sys.argv[2]);
 117         if freq>100:
 118             client.RF_setfreq(freq);
 119         else:
 120             print "Channels not yet supported."
 121     print "Transmitting as %010x on %i MHz" % (
 122         client.RF_getsmac(),
 123         client.RF_getfreq()/10**6);
 124
 125
 126     while 1:
 127         client.RF_txpacket([0x0f, 0x01, 0x08, 0x82,
 128                             0xff, 0xff, 0xff, 0xff,
 129                             0xde, 0xad, 0xbe, 0xef,
 130                             0xba, 0xbe, 0xc0]);
 131
 132 if(sys.argv[1]=="peek"):
 133     start=0x0000;
 134     if(len(sys.argv)>2):
 135         start=int(sys.argv[2],16);
 136     stop=start;
 137     if(len(sys.argv)>3):
 138         stop=int(sys.argv[3],16);
 139     print "Peeking from %04x to %04x." % (start,stop);
 140     while start<=stop:
 141         print "%04x: 0x%04x" % (start,client.peek(start));
 142         start=start+1;
 143 if(sys.argv[1]=="poke"):
 144     start=0x0000;
 145     val=0x00;
 146     if(len(sys.argv)>2):
 147         start=int(sys.argv[2],16);
 148     if(len(sys.argv)>3):
 149         val=int(sys.argv[3],16);
 150     print "Poking r%02x to become 0x%04x." % (start,val);
 151
 152     client.poke(start,val);
 153