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