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