1 # Copyright 2006 clauz at ninux dot org
2 # released under the GNU Public License
3 #
4 # OLSR Topology Logger. Dumps dot files from the OLSR dot plugin (http://www.olsr.org).
5 # Usage: olsrtopologylogger.py [-t timeoutseconds] [-s sleepseconds] [{+,-}h]
6 # [-x n] hostname [port]
7 #
8 # -t timeoutseconds set the read timeout to timeoutseconds (default 120).
9 # -s sleepseconds pause sleepseconds between reads (default 300).
10 # +h use human-readable date and time for filenames.
11 # -h use floating point seconds since the epoch for filenames (default).
12 # -x n read and dump to file n times (-1=infinite) (default -1)
13 # hostname the hostname
14 # port the port (default 2004)
15 #
16 # example: olsrtopologylogger.py -x 3 -s 3 localhost
17
18 import telnetlib
19 import time
20 import sys
21
22 params={
23 'PORT':2004, #tcp
24 'HOST':'127.0.0.1',
25 'TIMEOUT':120,
26 'SLEEP':300,
27 'HUMANREADABLE':False,
28 'TIMES':-1,
29 'EXT':'.dot'
30 }
31
32 def readfromdotplugin(host='127.0.0.1',port='2004',timeout=120, exitonerror=True):
33 try:
34 dotcon=telnetlib.Telnet(host,port)
35 except:
36 sys.stderr.write("Error. Can't connect to %s:%s.\n" % (host,port))
37 if exitonerror:
38 sys.exit(2)
39 else:
40 return ""
41 dotoutput=""
42 dotoutput=dotcon.read_until('}',timeout)
43 dotoutput+='\n'
44 dotcon.close()
45 return dotoutput
46 #readfromdotplugin
47
48 def gettimestamp(humanreadable=False):
49 if humanreadable:
50 ts=time.asctime()
51 ts=ts.replace(' ','_')
52 ts=ts.replace(':','.')
53 else:
54 ts=time.time() #seconds since the epoch
55 return ts
56 #gettimestamp
57
58 def processoptions(argvlist,params):
59 if len(argvlist)<2:
60 instructions= "OLSR Topology Logger. Dumps dot files from the OLSR dot plugin (http://www.olsr.org).\n"
61 instructions+="Usage: %s [-t timeoutseconds] [-s sleepseconds] [{+,-}h] \n" % (argvlist[0])
62 instructions+="\t\t[-x n] hostname [port]\n\n"
63 instructions+="\t-t timeoutseconds\tset the read timeout to timeoutseconds (default %d).\n" % (params['TIMEOUT'],)
64 instructions+="\t-s sleepseconds \tpause sleepseconds between reads (default %d).\n" % (params['SLEEP'],)
65 instructions+="\t+h \tuse human-readable date and time for filenames"
66 if params['HUMANREADABLE']:
67 instructions+=" (default).\n"
68 else:
69 instructions+=".\n"
70 instructions+="\t-h \tuse floating point seconds since the epoch for filenames"
71 if not params['HUMANREADABLE']:
72 instructions+=" (default).\n"
73 else:
74 instructions+=".\n"
75 instructions+="\t-x n \tread and dump to file n times (-1=infinite) (default %s)\n" % (params['TIMES'],)
76 instructions+="\thostname \tthe hostname\n"
77 instructions+="\tport \tthe port (default %s)\n" % (params['PORT'],)
78 # print instructions
79 sys.stderr.write(instructions)
80 sys.exit(1)
81 else:
82 i=1
83 while i<len(argvlist):
84 arg=argvlist[i]
85 if arg=='+h':
86 params['HUMANREADABLE']=True
87 elif arg=='-h':
88 params['HUMANREADABLE']=False
89 elif arg=='-t':
90 try:
91 timeoutseconds=argvlist[i+1]
92 params['TIMEOUT']=int(timeoutseconds)
93 i+=1
94 except:
95 pass
96 elif arg=='-s':
97 try:
98 sleepseconds=argvlist[i+1]
99 params['SLEEP']=int(sleepseconds)
100 i+=1
101 except:
102 pass
103 elif arg=='-x':
104 try:
105 n=argvlist[i+1]
106 params['TIMES']=int(n)
107 i+=1
108 except:
109 pass
110 else:
111 params['HOST']=arg
112 try:
113 port=argvlist[i+1]
114 params['PORT']=int(port)
115 i+=1
116 except:
117 pass
118 i+=1
119 #processoptions
120
121
122 if __name__=="__main__":
123
124 processoptions(sys.argv,params)
125
126 try:
127 firstiteration=True
128 iterations=params['TIMES']
129 while iterations==-1 or iterations>0:
130 print "reading..."
131 dotoutput=readfromdotplugin(params['HOST'],params['PORT'],params['TIMEOUT'],firstiteration)
132 firstiteration=False
133 timestamp=gettimestamp(params['HUMANREADABLE'])
134
135 filename='%s' % (timestamp,)
136 filename+=params['EXT']
137
138 try:
139 try:
140 dotfile=file(filename,'r')
141 dotfile.close()
142 erstr="Warning! File %s already exists! Skipping...",(filename,)
143 # sys.stderr.write(erstr)
144 print erstr
145 except IOError:
146 dotfile=file(filename,'w')
147 dotfile.write(dotoutput)
148 dotfile.close()
149 print "%s: %s created." % (time.asctime(),filename)
150 except:
151 sys.stderr.write("I/O Error!!")
152 sys.exit(2)
153
154
155 print "sleeping %d seconds..." % (params['SLEEP'],)
156 time.sleep(params['SLEEP'])
157
158 if iterations>0:
159 iterations-=1
160 #while
161 except KeyboardInterrupt:
162 pass
