133 lines
3.1 KiB
Python
133 lines
3.1 KiB
Python
|
#!/usr/bin/python
|
||
|
|
# (C) 2012, Mariano Alvira <mar@devl.org>
|
||
|
|
|
||
|
|
import sys,os,time
|
||
|
|
from termios import *
|
||
|
|
from struct import *
|
||
|
|
import re
|
||
|
|
|
||
|
|
if len(sys.argv) < 3:
|
||
|
|
sys.stderr.write( "Usage: %s tty channel\n" %(sys.argv[0]))
|
||
|
|
sys.stderr.write( " channel = 11-26\n")
|
||
|
|
sys.exit(2)
|
||
|
|
|
||
|
|
# change the channel
|
||
|
|
# rftest-rx increments it's channel everytime you send a character and returns "channel: num"
|
||
|
|
# send a character until we get to the right channel
|
||
|
|
|
||
|
|
try:
|
||
|
|
serport = os.open(sys.argv[1], os.O_RDWR|os.O_NOCTTY)
|
||
|
|
except IOError:
|
||
|
|
print "error opening port"
|
||
|
|
sys.exit(2)
|
||
|
|
|
||
|
|
try:
|
||
|
|
oldattrs = tcgetattr(serport)
|
||
|
|
attrs = [IGNPAR, oldattrs[1], 0, 0, B115200, B115200, oldattrs[6]]
|
||
|
|
attrs[2] = B115200 | CS8 | CLOCAL | CREAD
|
||
|
|
attrs[6][VTIME] = 0
|
||
|
|
attrs[6][VMIN] = 1
|
||
|
|
tcflush(serport, TCIFLUSH)
|
||
|
|
tcsetattr(serport, TCSANOW, attrs)
|
||
|
|
except IOError:
|
||
|
|
print "IOError in termios"
|
||
|
|
sys.exit(2)
|
||
|
|
|
||
|
|
chan = ''
|
||
|
|
while chan != int(sys.argv[2]) - 11:
|
||
|
|
os.write(serport, ' ')
|
||
|
|
chanstr = ''
|
||
|
|
while 1:
|
||
|
|
chanstr += os.read(serport, 1)
|
||
|
|
m = re.match(".*channel: (\w+)\s+", chanstr)
|
||
|
|
if m != None:
|
||
|
|
chan = int(m.group(1))
|
||
|
|
break
|
||
|
|
|
||
|
|
os.close(serport)
|
||
|
|
|
||
|
|
try:
|
||
|
|
serport = open(sys.argv[1], 'r+')
|
||
|
|
except IOError:
|
||
|
|
print "error opening port"
|
||
|
|
sys.exit(2)
|
||
|
|
|
||
|
|
|
||
|
|
### PCAP setup
|
||
|
|
MAGIC = 0xa1b2c3d4;
|
||
|
|
MAJOR = 2;
|
||
|
|
MINOR = 4;
|
||
|
|
ZONE = 0;
|
||
|
|
SIG = 0;
|
||
|
|
SNAPLEN = 0xffff;
|
||
|
|
NETWORK = 230; # 802.15.4 no FCS
|
||
|
|
|
||
|
|
# output overall PCAP header
|
||
|
|
sys.stdout.write(pack('<LHHLLLL', MAGIC, MAJOR, MINOR, ZONE, SIG, SNAPLEN, NETWORK))
|
||
|
|
|
||
|
|
count = 0
|
||
|
|
fileempty = 1
|
||
|
|
newpacket = 0
|
||
|
|
|
||
|
|
try:
|
||
|
|
while 1:
|
||
|
|
line = serport.readline().rstrip()
|
||
|
|
|
||
|
|
m_rftestline = re.match(".*rftest-rx --- len 0x(\w\w).*", line)
|
||
|
|
|
||
|
|
if m_rftestline != None:
|
||
|
|
newpacket = 1
|
||
|
|
t = time.time()
|
||
|
|
sec = int(t)
|
||
|
|
usec = (t - sec) * 100000
|
||
|
|
length = int(m_rftestline.group(1), 16)
|
||
|
|
# sys.stderr.write(line + "\n")
|
||
|
|
# sys.stderr.write("rftestline: %d %d %d\n" % (sec, usec, length))
|
||
|
|
continue
|
||
|
|
|
||
|
|
# if this is a new packet, add a packet header
|
||
|
|
if newpacket == 1:
|
||
|
|
newpacket = 0
|
||
|
|
sys.stdout.write(pack('<LLLL',sec,usec,length,length))
|
||
|
|
|
||
|
|
count += 1
|
||
|
|
sys.stderr.write("\x1b[G") # Move the cursor up one
|
||
|
|
sys.stderr.write("RX: %d" % count)
|
||
|
|
|
||
|
|
# clear file empty flag
|
||
|
|
if fileempty:
|
||
|
|
fileempty = 0
|
||
|
|
if fileempty == 0 :
|
||
|
|
# write payload
|
||
|
|
for d in line.split(' '):
|
||
|
|
# do a match because their might be a \r floating around
|
||
|
|
m = re.match('.*(\w\w).*', d)
|
||
|
|
sys.stdout.write(pack('<B', int(m.group(1),16)))
|
||
|
|
|
||
|
|
|
||
|
|
# cn.recv_block()
|
||
|
|
# if cn.data != None:
|
||
|
|
|
||
|
|
# # sys.stderr.write(cn.data)
|
||
|
|
# # sys.stderr.write(cn.data)
|
||
|
|
# # sys.stderr.write(str(time.time()))
|
||
|
|
|
||
|
|
# t = time.time()
|
||
|
|
# sec = int(t)
|
||
|
|
# usec = (t - sec) * 100000
|
||
|
|
|
||
|
|
# # sys.stderr.write("new packet: %d %d %d" % (sec, usec, len(cn.data)))
|
||
|
|
|
||
|
|
# count += 1
|
||
|
|
# sys.stderr.write("\x1b[G") # Move the cursor up one
|
||
|
|
# sys.stderr.write("RX: %d" % count)
|
||
|
|
|
||
|
|
# sys.stdout.write(pack('<LLLL',sec,usec,len(cn.data),len(cn.data)))
|
||
|
|
# sys.stdout.write(cn.data)
|
||
|
|
|
||
|
|
except KeyboardInterrupt:
|
||
|
|
# cn.close()
|
||
|
|
sys.exit(2)
|
||
|
|
|
||
|
|
|