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" % sys.argv[0];
  26     print "%s sniffdisect" % 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         client.RF_setfreq(eval(sys.argv[2]));
  72     print "Listening on %f MHz." % (client.RF_getfreq()/10.0**6);
  73
  74     client.strobe(0x02); #Calibrate
  75     time.sleep(1);
  76
  77     while 1:
  78         client.CC_RFST_RX();
  79         #client.strobe(0x03); #SRXON
  80         rssi=client.RF_getrssi();
  81         #client.CC_RFST_IDLE(); #idle
  82         time.sleep(0.01);
  83         string="";
  84         for foo in range(0,rssi>>2):
  85             string=("%s."%string);
  86         print "%02x %04i %s" % (rssi,rssi, string);
  87 if(sys.argv[1]=="sniff" or sys.argv[1]=="sniffdisect"):
  88     #Promiscuous mode.
  89     client.RF_promiscuity(1);
  90
  91     if len(sys.argv)>2:
  92         freq=eval(sys.argv[2]);
  93         if freq>100:
  94             client.RF_setfreq(freq);
  95         else:
  96             print "Channels not yet supported."
  97     client.CC_RFST_RX();
  98     print "Listening as %010x on %i MHz" % (client.RF_getsmac(),
  99                                             client.RF_getfreq()/10**6);
 100     #Now we're ready to get packets.
 101     while 1:
 102         packet=None;
 103         while packet==None:
 104             packet=client.RF_rxpacket();
 105         if sys.argv[1]=="sniffdisect":
 106             client.printdisect(packet);
 107         else:
 108             client.printpacket(packet);
 109         sys.stdout.flush();
 110 if(sys.argv[1]=="txtest"):
 111     if len(sys.argv)>2:
 112         freq=eval(sys.argv[2]);
 113         if freq>100:
 114             client.RF_setfreq(freq);
 115         else:
 116             print "Channels not yet supported."
 117     print "Transmitting DEADBEEF as %010x on %i MHz" % (
 118         client.RF_getsmac(),
 119         client.RF_getfreq()/10**6);
 120
 121     while 1:
 122         client.RF_txpacket([0x0f, 0x01, 0x08, 0x82,
 123                             0xff, 0xff, 0xff, 0xff,
 124                             0xde, 0xad, 0xbe, 0xef,
 125                             0xba, 0xbe, 0xc0]);
 126
 127 if(sys.argv[1]=="peek"):
 128     start=0x0000;
 129     if(len(sys.argv)>2):
 130         start=int(sys.argv[2],16);
 131     stop=start;
 132     if(len(sys.argv)>3):
 133         stop=int(sys.argv[3],16);
 134     print "Peeking from %04x to %04x." % (start,stop);
 135     while start<=stop:
 136         print "%04x: 0x%04x" % (start,client.peek(start));
 137         start=start+1;
 138 if(sys.argv[1]=="poke"):
 139     start=0x0000;
 140     val=0x00;
 141     if(len(sys.argv)>2):
 142         start=int(sys.argv[2],16);
 143     if(len(sys.argv)>3):
 144         val=int(sys.argv[3],16);
 145     print "Poking r%02x to become 0x%04x." % (start,val);
 146
 147     client.poke(start,val);
 148