root/trunk/tool/renetcolGUI.py @ 87

Revision 87, 54.4 KB (checked in by andreu, 14 years ago)

tickets #25 and #34

  • Property svn:eol-style set to native
Line 
1
2##  File: renetcolGUI.py
3 
4##  Authors: ANDREU Francois-Xavier
5 
6##  Copyright (C) 2005-2006-2007-2008 GIP RENATER
7 
8
9##   This file is part of renetcol.
10 
11##   renetcol is free software; you can redistribute it and/or modify
12##   it under the terms of the GNU General Public License as published by
13##   the Free Software Foundation; either version 2 of the License, or
14##   (at your option) any later version.
15 
16##   renetcol is distributed in the hope that it will be useful,
17##   but WITHOUT ANY WARRANTY; without even the implied warranty of
18##   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
19##   GNU General Public License for more details.
20 
21##   You should have received a copy of the GNU General Public License
22##   along with renetcol; if not, write to the Free Software
23##   Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
24 
25
26# Make Windows actually find the stuff installed
27# You must check up if it's the good path to GTK
28gtkdir = 'C:/Program Files/Fichiers communs/GTK/'
29
30import os
31import binascii
32import struct
33import time
34os.environ['PATH'] += ";%s/lib;%s/bin" % (gtkdir, gtkdir)
35
36import operator
37import threading
38import string
39import re
40import socket
41import sys
42import gtk
43import traceback
44import pango
45import commands
46from optparse import OptionParser
47
48########### DEFAULT VALUES, YOU MUST CHANGE IT ###############################
49IPversion = 4
50localAddr = ''
51localAddr6 = '0::'
52localPort = 2222
53collectorAddr = "127.0.0.1"
54collectorAddr6 = "0::"
55collectorPort = 52571  # if you change this port number, apply the
56                       # modification on the renetcolSender.h file
57                       # and you'll need a new compilation of the colllector
58##############################################################################
59
60myInputControlThread = None
61printValue =      [ 0,1,1,0,1,0,0,1,1,0, # 0 to 9
62                    0,1,1,0,0,0,0,0,0,0, # 10...
63                    1,0,0,0,0,0,0,1,1,0, #
64                    0,0,0,0,0,0,0,0,0,0, #
65                    0,0,0,0,0,0,1,1,0,0, #
66                    0,0,0,0,0,0,0,0,0,0, #
67                    0,0,0,0,0,0,0,0,0,0, #
68                    1,1,1,1,1,1,1,1,1,1, #
69                    0,0,0,0,0,0,0,0,0,0,  # 80 to 89
70                    0,0,0,0,0,0,0,0,0,0, #
71                    0,0,0,0,0,0,0,0,0,0, #
72                    0,0,0,0,0,0,0,0,0,0, #
73                    0,0,0,0,0,0,0,0,0,0, #
74                    0,0,0,0,0,0,0,0,0,0 # 130 to 139
75                    ]
76firstParseValue = [ 0,0,0,0,0,0,0,0,0,0,
77                    0,0,0,0,0,0,0,0,0,0,
78                    0,0,0,0,0,0,0,0,0,0,
79                    0,0,0,0,0,0,0,0,0,0,
80                    0,0,0,0,0,0,0,0,0,0,
81                    0,0,0,0,0,0,0,0,0,0,
82                    0,0,0,0,0,0,0,0,0,0,
83                    0,0,0,0,0,0,0,0,0,0,
84                    0,0,0,0,0,0,0,0,0,0,
85                    0,0,0,0,0,0,0,0,0,0, #
86                    0,0,0,0,0,0,0,0,0,0, #
87                    0,0,0,0,0,0,0,0,0,0, #
88                    0,0,0,0,0,0,0,0,0,0, #
89                    0,0,0,0,0,0,0,0,0,0 # 130 to 139
90                    ]
91checkUpValues =   [ 0,0,0,0,0,0,0,0,0,0,
92                    0,0,0,0,0,0,0,0,0,0,
93                    0,0,0,0,0,0,0,0,0,0,
94                    0,0,0,0,0,0,0,0,0,0,
95                    0,0,0,0,0,0,0,0,0,0,
96                    0,0,0,0,0,0,0,0,0,0,
97                    0,0,0,0,0,0,0,0,0,0,
98                    0,0,0,0,0,0,0,0,0,0,
99                    0,0,0,0,0,0,0,0,0,0,
100                    0,0,0,0,0,0,0,0,0,0, #
101                    0,0,0,0,0,0,0,0,0,0, #
102                    0,0,0,0,0,0,0,0,0,0, #
103                    0,0,0,0,0,0,0,0,0,0, #
104                    0,0,0,0,0,0,0,0,0,0 # 130 to 139
105                    ]
106fieldsName =  [ "","IN_BYT","IN_PKTS","","PROT","TOS","TCP_FL","S_PORT","IPv4@S","S_MASK",
107                "InSNMP","D_PORT","IPv4@D","D_MASK","OutSNMP","NextHop","S_AS","D_AS","","",
108                "","Time","","","","","","IPv6@S","IPv6@D","",
109                "","","","","","","","","","",
110                "","","","","","","M_T_L_T","M_T_L_@","","",
111                "","","","","","","","","","",
112                "","","","","","","","","","ROUTER",
113                "M_L_1","M_L_2","M_L_3","M_L_4","M_L_5","M_L_6","M_L_7","M_L_8","M_L_9","M_L_10",
114                "","","","","","","","","","", # 80 to 89
115                "","","","","","","","","","",
116                "","","","","","","","","","",
117                "","","","","","","","","","",
118                "","","","","","","","","","",
119                "","","","","","","","","","" # 130 to 139
120                ]
121flowCpt = 0
122mask = [ 0,0,0 ]
123spaceSep = " "
124tabSep = "\t"
125freeze = 0
126record = 0
127record_file_name = ""
128record_file = None
129is_already_see = 0
130collectorRule=""
131coll_window = gtk.Window()
132router=0
133tpl_def = {}
134os_type = 0
135oldTpl = 0
136fromRouter = ""
137myTimer = None
138myCPT = 1
139
140class InputControl(threading.Thread):
141    def kill(self, timeout):
142        self.imRunning = 0
143        time.sleep(1)
144        print "\n.\n..\n..."
145        time.sleep(1)
146        print "....\n.....\n......"       
147        time.sleep(1)
148        if self.myFlowInput:
149            self.myFlowInput.socket_close()
150        self.join(timeout)
151       
152    def __init__(self, address, port):
153        self.imRunning = 1
154        self.currentflow = None
155        threading.Thread.__init__(self)
156        self.myFlowInput = FlowInput(address, port)
157
158    def run(self):
159        global freeze, record
160        while self.imRunning:
161            if self.myFlowInput:
162                tmp = self.myFlowInput.get_flow()
163                if (tmp[0]!=None):
164                    self.currentflow = Flow(tmp)
165                    if (freeze==0) & self.currentflow.enable():
166                        self.currentflow.print_flow2()
167                    if (record==1) & self.currentflow.enable():
168                        self.currentflow.write_flow2()
169##                else:
170##                    self.myFlowInput = None
171   
172class Flow:
173    def __init__(self, data):
174        self.flow = [ data[0], data[1] ]
175
176# flow selection
177    def enable(self):
178        global firstParseValue, checkUpValues, tpl_def, fromRouter
179        res = 1
180        resS = 1
181        resD = 1
182        for i in range (0, len(tpl_def[self.flow[0]])):
183            f = tpl_def[self.flow[0]][i][0]
184            # or f==15 or f==18 or f==47
185            if ( f==8 ) and ( checkUpValues[f]==1 ):
186                resS = resS & ( ((struct.unpack('>L',(socket.inet_aton(self.flow[1][i])))[0] & mask[1-1]) == firstParseValue[f]) )
187            elif ( f==12 ) and ( checkUpValues[f]==1 ):
188                resD = resD & ( ((struct.unpack('>L',(socket.inet_aton(self.flow[1][i])))[0] & mask[1-1]) == firstParseValue[f]) )
189            elif ( i==27 or i==28 ):
190                pass
191            elif ( (f==1) and (checkUpValues[f]==1) ):
192                res = res & ( (self.flow[1][i] <= ((firstParseValue[f])+((firstParseValue[f])*5/100))) and ((self.flow[1][i] >= ((firstParseValue[f])-((firstParseValue[f])*5/100)))) )
193            elif (checkUpValues[f]==1):
194                res = res & (self.flow[1][i] == firstParseValue[f])
195        if ( checkUpValues[69]==1 ):
196            res = res & ( (fromRouter == firstParseValue[69]) )
197        res = res & (resS or resD)
198        return res
199
200# flow print
201    def print_flow2(self):
202        global printValue, tpl_def, flowCpt, fieldsName, oldTpl, os_type, fromRouter
203        myliste = []
204        myTpl = []
205        strFlow = ""
206        strField = ""
207        underscore_line = ""
208        flowCpt+=1
209        if (flowCpt%60 == 0 or oldTpl != self.flow[0]):
210##        if (flowCpt%100 == 0):
211            flowCpt = 0
212            for i in range (0, len(tpl_def[self.flow[0]])):
213                f = tpl_def[self.flow[0]][i][0]
214                if printValue[f]:
215                    if ( f==27 or f==28 or f==62 or f==63 ):
216                        strField += str(fieldsName[f])
217                        l = len(str(fieldsName[f]))
218                        if ((40-l)%8 == 0):
219                            tabNb = int((40-l)/8)
220                        else:
221                            tabNb = int((40-l)/8) + 1
222                        for j in range (0, tabNb):
223                            strField += "\t"
224                        for k in range (0, 40):
225                            underscore_line += "-"
226                    elif ( f==8 or f==15 or f==12 or f==18 or f==47 ):
227                        strField += str(fieldsName[f])
228                        l = len(str(fieldsName[f]))
229                        if ((16-l)%8 == 0):
230                            tabNb = int((16-l)/8)
231                        else:
232                            tabNb = int((16-l)/8) + 1
233                        for j in range (0, tabNb):
234                            strField += "\t"
235                        for k in range (0, 16):
236                            underscore_line += "-"
237                    else:
238                        strField += str(fieldsName[f])
239                        strField += "\t"
240                        for k in range (0, 8):
241                            underscore_line += "-"
242            if (oldTpl != self.flow[0]):
243                print " "
244            if (os_type == 0):
245                esc = '\x1b['
246                sep = ';'
247                end = 'm'
248                if (printValue[69]):
249                    rt = str(fieldsName[69])
250                    toprint = esc+"37"+sep+"44"+end+strField+rt+esc+"0"+end
251                else:
252                    toprint = esc+"37"+sep+"44"+end+strField+esc+"0"+end
253                print toprint
254            else:
255                if (printValue[69]):
256                    strField += str(fieldsName[69])
257                print strField
258                print underscore_line
259        for i in range (0, len(tpl_def[self.flow[0]])):
260            f = tpl_def[self.flow[0]][i][0]
261            if printValue[f]:
262                if ( f==27 or f==28 or f==62 or f==63 ):
263                    strFlow += str(self.flow[1][i])
264                    l = len(str(self.flow[1][i]))
265                    if ((40-l)%8 == 0):
266                        tabNb = int((40-l)/8)
267                    else:
268                        tabNb = int((40-l)/8) + 1
269                    for j in range (0, tabNb):
270                        strFlow += "\t"
271                elif ( f==8 or f==12 or f==15 or f==18 or f==47 ):
272                    strFlow += str(self.flow[1][i])
273                    l = len(str(self.flow[1][i]))
274                    if ((16-l)%8 == 0):
275                        tabNb = int((16-l)/8)
276                    else:
277                        tabNb = int((16-l)/8) + 1
278                    for j in range (0, tabNb):
279                        strFlow += "\t"
280                elif (f==21):
281                    myliste += [str(self.flow[1][i]-self.flow[1][i+1])]
282                    strFlow += str(self.flow[1][i]-self.flow[1][i+1])
283                    strFlow += "\t"
284                elif (f==22):
285                    pass
286                else:
287                    myliste += [str(self.flow[1][i])]
288                    strFlow += str(self.flow[1][i])
289                    for k in range (len(str(self.flow[1][i])), 7):
290                        strFlow += " "
291                    strFlow += "\t"
292        if (printValue[69]):
293            strFlow += fromRouter
294        print strFlow
295        oldTpl = self.flow[0]
296
297# flow record
298    def write_flow2(self):
299        global printValue, record_file, tpl_def
300        myliste = []
301        for i in range (0, len(tpl_def[self.flow[0]])):
302            f = tpl_def[self.flow[0]][i][0]
303            if printValue[f]:
304                if (f==21):
305                    myliste += [str(self.flow[1][i]-self.flow[1][i+1])]
306                elif (f==22):
307                    pass
308                else:
309                    myliste += [str(self.flow[1][i])]
310        theflow = string.join(myliste, "\t")+"\t"+fromRouter+"\n"
311        if (record_file!=None):
312            record_file.write(theflow)
313       
314class FlowInput:
315    def __init__(self, h, p):
316        self.HOST = h
317        self.PORT = p
318        if (IPversion == 4):
319            self.socketnumber = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
320        elif (IPversion == 6):
321            self.socketnumber = socket.socket(socket.AF_INET6, socket.SOCK_DGRAM)
322        else:
323            print "IP protocol ", IPversion ," not supported"
324            exit(0)
325        self.socketnumber.bind((self.HOST, self.PORT))
326   
327    def get_flow(self):
328        global collectorRule, tpl_def, os_type, fromRouter
329        if (self.socketnumber!=None):
330            data = self.socketnumber.recvfrom(1024)
331            flow = data[0]
332            code = struct.unpack('<H', flow[0:2])
333            if (code[0] == 1): ## a def packet
334                routerSrc = struct.unpack('<L', flow[2:6])
335                sourceId = struct.unpack('<L', flow[6:10])
336                tplId = struct.unpack('<H', flow[10:12])
337                mykeylst = []
338                mykeylst += [str(routerSrc[0])]
339                mykeylst += [str(sourceId[0])]
340                mykeylst += [str(tplId[0])]
341                mykeystr = string.join(mykeylst,"");
342                fields = []
343                fieldNb = struct.unpack('<H', flow[12:14])
344                for i in range(0, fieldNb[0]):
345                    fd = struct.unpack('<H',flow[14+(i*4):16+(i*4)])
346                    tp = struct.unpack('<H',flow[14+(2+i*4):16+(2+i*4)])
347                    fields.insert(i,(fd[0],tp[0]))
348                tpl_def[mykeystr] = fields
349                return ( None, None)
350            if (code[0]==11): ## a data packet
351                routerSrc = struct.unpack('<L', flow[4:8])
352                tmp =  struct.unpack('<L', flow[4:8])
353                fromRouter = socket.inet_ntoa(struct.pack('>L', tmp[0]))
354                sourceId = struct.unpack('<L', flow[8:12])
355                tplId = struct.unpack('<H', flow[12:14])
356                mykeylst = []
357                mykeylst += [str(routerSrc[0])]
358                mykeylst += [str(sourceId[0])]
359                mykeylst += [str(tplId[0])]
360                mykeystr = string.join(mykeylst,"")
361                if (tpl_def.has_key(mykeystr)== True):
362                    field_list = []
363                    index = 0
364                    for i in range (0, len(tpl_def[mykeystr])):
365                        if (tpl_def[mykeystr][i][1]==1):
366                            value = struct.unpack('<B',flow[14+index:15+index])
367                            field_list.insert(i,value[0])
368                        if (tpl_def[mykeystr][i][1]==2):
369                            value = struct.unpack('>H',flow[14+index:16+index])
370                            field_list.insert(i,value[0])
371                        if (tpl_def[mykeystr][i][1]==3):
372                            value = struct.unpack('<BBB',flow[14+index:17+index])
373                            valueTmp = struct.pack('<BBBB',value[2],value[1],value[0],0)
374                            valueFinal = struct.unpack('<L', valueTmp)
375                            valueFinal2 = (valueFinal[0])>>4
376                            field_list.insert(i,valueFinal2)
377                        if (tpl_def[mykeystr][i][1]==4):
378                            if (tpl_def[mykeystr][i][0]==8 or tpl_def[mykeystr][i][0]==12 or tpl_def[mykeystr][i][0]==15 or tpl_def[mykeystr][i][0]==18 or tpl_def[mykeystr][i][0]==47):
379                                value = socket.inet_ntoa(flow[14+index:18+index])
380                                field_list.insert(i,value)
381                            else:
382                                value = struct.unpack('<L', flow[14+index:18+index])
383                                value = struct.unpack('>L', flow[14+index:18+index])
384                                field_list.insert(i,value[0])
385                        if (tpl_def[mykeystr][i][1]==16):
386                            if (os_type==1):
387                                field_list.insert(i,nt_inet_ntop(flow[14+index:30+index]))
388                            else:
389                                tmp = socket.inet_ntop(socket.AF_INET6,flow[14+index:30+index])
390                                field_list.insert(i,tmp)
391                        index += tpl_def[mykeystr][i][1]
392                    return mykeystr, field_list
393##                else:
394##                  print "I'm waiting the tpl def for the key", mykeystr
395##                return (routerSrc[0], field_list)
396            return ( None, None)
397        return ( None, None)
398
399    def socket_close(self):
400        if self.socketnumber:
401           try :
402               self.socketnumber.shutdown(2)
403               self.socketnumber = None
404           except socket.error:
405              pass
406
407class FileSel:
408    def file_ok_sel(self, w):
409        global record, record_file, record_file_name
410        record = 1
411        record_file_name = self.fileSel.get_filename()
412        record_file = file(record_file_name, 'w')
413        print "%s" % self.fileSel.get_filename()
414        self.fileSel.destroy()
415
416    def destroy(self, widget):
417        self.fileSel.destroy()
418
419    def __init__(self):
420        self.fileSel = gtk.FileSelection("File Selection")
421        self.fileSel.connect("destroy", self.destroy)
422        self.fileSel.ok_button.connect("clicked", self.file_ok_sel)
423        self.fileSel.cancel_button.connect("clicked",
424                                           lambda w: self.fileSel.destroy())
425        self.fileSel.set_filename("flow_record.txt")
426        self.fileSel.show()
427
428def print_hello(widget, event):
429    print "Hello, World!"
430
431def nt_inet_ntop(packed_ip):
432    cpt = 0
433    double = 0
434    string_ip = ""
435    tmp = binascii.hexlify(packed_ip[0:2])
436    if ( int(tmp,16)!=0 ):
437        string_ip += tmp
438        tmp = binascii.hexlify(packed_ip[2:4])
439        if ( int(tmp,16)==0 ):
440            cpt+=1
441            double = 1
442            if (cpt==1):
443                string_ip += ":"
444        else:
445            cpt=0
446            string_ip += ":"+tmp
447        tmp = binascii.hexlify(packed_ip[4:6])
448        if ( int(tmp,16)==0 ):
449            cpt+=1
450            double = 1
451            if (cpt==1):
452                string_ip += ":"
453        else:
454            cpt=0
455            string_ip += ":"+tmp
456        tmp = binascii.hexlify(packed_ip[6:8])
457        if ( int(tmp,16)==0 ):
458            if (double==1 and cpt==0):
459                string_ip += ":0"
460            else:
461                cpt+=1
462                double = 1
463                if (cpt==1):
464                    string_ip += ":"
465        else:
466            cpt=0
467            string_ip += ":"+tmp
468        tmp = binascii.hexlify(packed_ip[8:10])
469        if ( int(tmp,16)==0 ):
470            if (double==1 and cpt==0):
471                string_ip += ":0"
472            else:
473                cpt+=1
474                double = 1
475                if (cpt==1):
476                    string_ip += ":"
477        else:
478            cpt=0
479            string_ip += ":"+tmp
480        tmp = binascii.hexlify(packed_ip[10:12])
481        if ( int(tmp,16)==0 ):
482            if (double==1 and cpt==0):
483                string_ip += ":0"
484            else:
485                cpt+=1
486                double = 1
487                if (cpt==1):
488                    string_ip += ":"
489        else:
490            cpt=0
491            string_ip += ":"+tmp
492        tmp = binascii.hexlify(packed_ip[12:14])
493        if ( int(tmp,16)==0 ):
494            if (double==1 and cpt==0):
495                string_ip += ":0"
496            else:
497                cpt+=1
498                double = 1
499                if (cpt==1):
500                    string_ip += ":"
501        else:
502            cpt=0
503            string_ip += ":"+tmp
504        tmp = binascii.hexlify(packed_ip[14:16])
505        if ( int(tmp,16) != 0 ):
506            string_ip += ":"+tmp
507        else:
508            string_ip += ":"
509    else:
510        string_ip = "::"
511    return string_ip
512
513def get_main_menu(self, window):
514    accel_group = gtk.AccelGroup()
515    item_factory = gtk.ItemFactory(gtk.MenuBar, "<main>", accel_group)
516    item_factory.create_items(menu_items)
517    window.add_accel_group(accel_group)
518    item_factory = item_factory
519    return item_factory.get_widget("<main>")
520
521def reception(widget, event):
522    global myInputControlThread
523    global localAddr, localPort
524    myInputControlThread = InputControl(localAddr, localPort)
525    myInputControlThread.start()
526
527def close_recept(widget, event):
528    global myInputControlThread
529    if myInputControlThread != None:
530        myInputControlThread.kill(1)
531        myInputControlThread = None
532        print "\n\n\n\n\n Input Stream closed. \n\n\n\n\n"
533
534def about_dialog(widget, event):
535    dialog = gtk.MessageDialog(
536        parent         = None,
537        flags          = gtk.DIALOG_MODAL,
538        type           = gtk.MESSAGE_INFO,
539        buttons        = gtk.BUTTONS_OK,
540        message_format = "Author : FX Andreu\nemail andreu@renater.fr\nCopyright (C) 2005-2008 GIP RENATER\n\nThis file is part of renetcol.\n\nrenetcol is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation; either version 2 of the License, or (at your option) any later version.\n\nrenetcol is distributed in the hope that it will be useful,but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details.\n\nYou should have received a copy of the GNU General Public License along with renetcol; if not, write to the Free Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA ")
541    dialog.set_title('About')
542    dialog.connect('response', lambda dialog, response: dialog.destroy())
543    dialog.show()
544
545def check_dialog(widget, event, text):
546    dialog = gtk.MessageDialog(
547        parent         = None,
548        flags          = gtk.DIALOG_MODAL,
549        type           = gtk.MESSAGE_INFO,
550        buttons        = gtk.BUTTONS_OK,
551        message_format = text)
552    dialog.set_title('Value check ')
553    dialog.connect('response', lambda dialog, response: dialog.destroy())
554    dialog.show()
555
556def info_dialog(widget, event, text):
557    dialog = gtk.MessageDialog(
558        parent         = None,
559        flags          = gtk.DIALOG_MODAL,
560        type           = gtk.MESSAGE_INFO,
561        buttons        = gtk.BUTTONS_OK,
562        message_format = text)
563    dialog.set_title('Information ')
564    dialog.connect('response', lambda dialog, response: dialog.destroy())
565    dialog.show()
566
567def callback_freeze(widget, data=None):
568    global freeze
569    if widget.get_active() == 1:
570        freeze = 1
571        print "FREEZE ACTIF"
572    if widget.get_active() == 0:
573        freeze = 0
574        print "FREEZE NOT ACTIF"
575
576def wind_destroy(widget, data=None):
577    widget.destroy()
578   
579def callback_record(widget, data=None):
580    global record, record_file, record_file_name
581    if widget.get_active() == 1:
582        FileSel()
583    if widget.get_active() == 0:
584        record = 0
585        time.sleep(2)
586        record_file.close()
587
588def strsend(target, message):
589   global collectorAddr, collectorPort
590   s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
591   s.connect((collectorAddr, collectorPort))
592   s.send(message)
593   s.close()
594
595def callback_refresh(widget, parent):
596    global collectorAddr
597    strsend(collectorAddr, 'REFRESH')
598    parent.destroy()
599   
600def callback_delete_rule(widget, parent):
601    global collectorAddr, collectorPort
602    if widget.get_active() == 1:
603        sent_rule(widget, "DELETE")
604        widget.set_active(0)
605        parent.destroy()
606
607def callback_check_rule(widget, data=None):
608    global router
609    if widget.get_active()==1:
610        router=1
611
612def timer_action():
613    global record_file, record_file_name, myTimer, myCPT
614    myCPT+=1
615    myTimer.cancel()
616    my_new_file_name = "%s_%s_%d" % (record_file_name,time.strftime("%Y%m%d", time.gmtime()),myCPT)
617    ##my_new_file_name = "%s_%d" % (record_file_name,myCPT)
618    print my_new_file_name
619    new_record_file = file(my_new_file_name, 'w')
620    old_record_file = record_file
621    record_file = new_record_file
622    old_record_file.close()
623    now2 = time.time()
624    reste = 360-(operator.mod(now2,360))
625    myTimer = threading.Timer(360.0,timer_action)
626    myTimer.start()
627
628def callback_RotRec(widget, data=None):
629    global record, myTimer
630    if ((record == 0) and (widget.get_active() == 1)):
631        info_dialog(widget, None, "You must before activated the record ! ")
632        widget.set_active(0)
633    else:
634        if widget.get_active() == 1:
635            print "STARTING RECORD ROTATION"
636            now2 = time.time()
637            reste = 360-(operator.mod(now2,360))
638            myTimer = threading.Timer(360,timer_action)
639            myTimer.start()
640        if widget.get_active() == 0:
641            myTimer.cancel()
642            print "STOPPING RECORD ROTATION"
643
644def callback_collector(widget, controlWindow):
645    global is_already_see, collectorRule, myInputControlThread, coll_window, collectorAddr
646    if ((myInputControlThread == None) and (widget.get_active() == 1)):
647        info_dialog(widget, None, "You must before activated the stream reception ! (File menu)")
648        widget.set_active(0)
649    else:
650        if widget.get_active() == 1:
651            widget.set_active(0)
652            coll_window.connect("destroy", wind_destroy)
653            coll_window.set_title("Collector control ")
654            coll_window.set_size_request(500, 250)
655            coll_vbox = gtk.VBox(False, 1)
656            coll_vbox.set_border_width(1)
657            coll_window.add(coll_vbox)
658            coll_frame = gtk.Frame("Enter your filter (format described in tip), validate by enter")
659            coll_vbox.pack_start(coll_frame, False, False, 5)
660            coll_hbox = gtk.VBox(False, 0)
661            coll_hbox.set_border_width(5)
662            coll_frame.add(coll_hbox)
663##            button = gtk.CheckButton(" router ")
664##            button.connect("toggled", callback_check_rule, None)
665##            coll_hbox.pack_start(button, False, False, 0)
666##            mainTT.set_tip(button, "Check it to capture all flows from one router and enter the IPv4 address of this router. IN TEST")
667##            button.show()
668            coll_pbbox = gtk.HButtonBox()
669            coll_hbox.pack_start(coll_pbbox, False, True, 5)
670            prefix_entry(249, "Required Format : \n<field> = <value>[ & <field> = <value>]\
671            \nlike \
672            \n8 = 10.0.0.4 \
673            \n\t(all flows from IP Source 10.0.0.4)\
674            \nor 8 = 10.0.1.0/24 \
675            \n\t(all flows with IP Source in subnet 10.0.1.0/24)\
676            \nor 27 = 2001::1 (IPv6 address only on Linux system)\
677            \n\t(all flows from IP Source 2001::1)\
678            \nor 27 = 2001::/64 \
679            \n\t(all flows with IP Source in subnet 2001::/64)\
680            \nor 0 = 10.0.0.1 & 4 = 1\
681            \n\t(all ICMP flows from router 10.0.0.1)\
682            \nonly equal operator in this release\
683            \nmaximum 4 logical '&' for one filter (limit 249 chars)\
684            \n----------------------------------\
685            \n---- NetFlow Fields list ------\
686            \nROUTER_SRC_ADDR:\t 0\
687            \nIN_BYTES:\t\t\t 1\
688            \nL4_PROT:\t\t\t 4\
689            \nL4_SRC_PORT:\t\t 7\
690            \nIPV4_SRC_ADDR:\t\t 8\
691            \nINPUT_SNMP:\t\t\t10\
692            \nL4_DST_PORT:\t\t11\
693            \nIPV4_DST_ADDR:\t\t12\
694            \nOUTPUT_SNMP:\t\t14\
695            \nIPV4_NEXT_HOP:\t\t15\
696            \nSRC_AS:\t\t\t\t16\
697            \nDST_AS:\t\t\t\t17\
698            \nBGP_IPV4_NEXT_HOP:\t18\
699            \nIPV6_SRC_ADDR:\t\t27\
700            \nIPV6_DST_ADDR:\t\t28\
701            \nIPV6_NEXT_HOP:\t\t62\
702            \nBPG_IPV6_NEXT_HOP:\t63\
703            \nIP_PROT_VERSION:\t60\
704            \n---------------------------", coll_pbbox, "", 0)
705            coll_frame2 = gtk.Frame("Disable rules")
706            coll_vbox.pack_start(coll_frame2, False, False, 5)
707            coll_hbox2 = gtk.VBox(False, 0)
708            coll_hbox2.set_border_width(5)
709            coll_frame2.add(coll_hbox2)
710            coll_pbbox2 = gtk.HButtonBox()
711            coll_hbox2.pack_start(coll_pbbox2, False, True, 5)
712            buttondelete = gtk.ToggleButton("Delete")
713            buttondelete.connect("toggled", callback_delete_rule, coll_window)
714            mainTT.set_tip(buttondelete, "Delete all your rules of the collector configuration.")
715            coll_hbox2.pack_start(buttondelete, False, False, 2)
716            buttondelete.show()
717            coll_vbox.show()
718            coll_window.show_all()
719            if (is_already_see==0):
720                info_dialog(widget, None, "You should register only one rule for each GUI parser ! Think of using the DELETE button !")
721                is_already_see=1
722
723def callback_fields_printed( widget, data=None):
724    global printValue
725    printValue[data] = widget.get_active()
726
727def callback_check_ip( widget, data=None):
728    global src_Or_Dst_IP
729    if widget.get_active()==1:
730        if checkUpValues[1]:
731            src_Or_Dst_IP = 1
732        else:
733            widget.set_active(0)
734            check_dialog(widget, None, "First, enter a source address.")
735    else:
736            src_Or_Dst_IP = 0
737
738def callback_comboR(widget, data=None):
739    global comboRouter, routerHash
740    nop = 0
741    print struct.unpack('>L',socket.inet_aton(routerHash[comboRouter.entry.get_text()]))
742   
743def callback_src_mask(widget, data=None):
744    nop = 0
745
746def callback_dst_mask(widget, data=None):
747    nop = 0
748
749def callback_tos(widget, data=None):
750    nop = 0
751
752def callback_prot(widget, data=None):
753    nop = 0
754
755def create_button(title, state, callback_function, ptr_var, parent, tooltip_Text):
756    global mainTT
757    button = gtk.ToggleButton(title)
758    if state==1:
759        button.set_active(True)
760    else:
761        button.set_active(False)
762    button.connect("toggled", callback_function, ptr_var)
763    if tooltip_Text!=None:
764        mainTT.set_tip(button, tooltip_Text)
765    parent.pack_start(button, False, True, 2)
766    button.show()
767
768def create_entry(maxLen, tooltip_Text, parent, lab, id):
769    global mainTT
770    vbox = gtk.VBox(False, 0)
771    parent.pack_start(vbox, False, True, 0)
772    label = gtk.Label(lab)
773    label.set_alignment(0, 0.5)
774    vbox.pack_start(label, False, True, 0)
775    entry = gtk.Entry(max=maxLen)
776    entry.set_max_length(maxLen)
777    entry.connect("activate", enter_callback, entry, id)
778    entry.set_text("")
779    entry.select_region(0, len(entry.get_text()))
780    if tooltip_Text!=None:
781        mainTT.set_tip(entry, tooltip_Text)
782    vbox.pack_start(entry, False, True, 0)
783    entry.show()
784
785def sent_rule(widget, msg):
786    global collectorAddr, collectorPort, collectorAddr6, IPversion
787    s = None
788    if (IPversion==4):
789        tmpaf =  socket.AF_INET
790        addr = collectorAddr
791        port = collectorPort
792    elif (IPversion == 6):
793        tmpaf =  socket.AF_INET6
794        addr = collectorAddr6
795        port = collectorPort
796    else:
797        print "Wrong IP version : ", IPversion
798        exit(0)
799    for res in socket.getaddrinfo(addr, port, tmpaf, socket.SOCK_STREAM):
800        af, socktype, proto, canonname, sa = res
801        try:
802            s = socket.socket(af, socktype, proto)
803        except socket.error, msg:
804            s = None
805            continue
806        try:
807            s.connect(sa)
808        except socket.error, msg:
809            s.close()
810            s = None
811            continue
812        break
813    if s is None:
814        check_dialog( widget, None, "could not open socket")
815    else:
816        s.send(msg)
817        check_dialog( widget, None, "Action was sent")
818        s.close()
819
820def sent_rule_from_cli(msg):
821    global collectorAddr, collectorPort, collectorAddr6, IPversion
822    s = None
823    if (IPversion==4):
824        tmpaf =  socket.AF_INET
825        addr = collectorAddr
826        port = collectorPort
827    elif (IPversion == 6):
828        tmpaf =  socket.AF_INET6
829        addr = collectorAddr6
830        port = collectorPort
831    else:
832        print "Wrong IP version : ", IPversion
833        exit(0)
834    for res in socket.getaddrinfo(addr, port, tmpaf, socket.SOCK_STREAM):
835        af, socktype, proto, canonname, sa = res
836        try:
837            s = socket.socket(af, socktype, proto)
838        except socket.error, msg:
839            s = None
840            continue
841        try:
842            s.connect(sa)
843        except socket.error, msg:
844            s.close()
845            s = None
846            continue
847        break
848    if s is None:
849        check_dialog( None, None, "could not open socket")
850    else:
851        s.send(msg)
852        check_dialog( None, None, "Action was sent")
853        s.close()
854
855# dialog box to collector control
856def prefix_callback( widget, entry, id):
857    global collPrefix, coll_window, router, localPort
858    entry_text = entry.get_text()
859    is_good = 0
860    mymsg = []
861    if (id == 0):
862        if ( entry_text==""):
863            collPrefix = 0
864        else:
865            ## here we parse the filter, possibility of many filter concatened with '&' operator
866            filters = entry_text.split('&')
867            if len(filters)>5:
868                check_dialog( widget, None, "No more 4 '&' !")
869                is_ok = 0
870            else:
871                is_ok = 1
872                for f in filters:
873                    rule = f.split(' ')
874                    if len(mymsg)==0:
875                        field = rule[0]
876                        operator = rule[1]
877                        value = rule[2]
878                        mymsg = "C" + "_" + field + "_" + operator + "_" + value
879                    else:
880                        field = rule[1]
881                        operator = rule[2]
882                        value = rule[3]                       
883                        mymsg += "_C" + "_" + field + "_" + operator + "_" + value
884                    print f
885                    print mymsg
886                    if (field=='0' or field=='8' or field=='12' or field=='15' or field=='18' or field=='47'):
887                        if (re.match('^\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}(/\d{1,2})?$',value)):
888                            splitter = value.split('/')
889                            adip = struct.unpack('>L',socket.inet_aton(splitter[0]))[0]
890                            if len(splitter)>1:
891                                bits = int(splitter[1])
892                                if bits <= 32 and bits > 0:
893                                    mask[1] = 0xffffffffL << (32 - bits)
894                                    is_good = 1
895                                else:
896                                    check_dialog( widget, None, "Wrong value in mask!")
897                                    is_good = 0
898                            else:
899                                value = value+"/32"
900                                mask[1] = 0xffffffffL
901                                is_good = 1
902                        else:
903                            check_dialog( widget, None, "Wrong value !")
904                    if (field=='27' or field=='28' or field=='62'):
905                        ipv6ad_mask = value.split('/')
906                        if len(ipv6ad_mask)==1:
907                            try:
908                                socket.gethostbyaddr(value)
909                                is_good = 1
910                            except socket.gaierror:
911                                check_dialog( widget, None, "Wrong value !")
912                            except socket.herror: ## can't resolv
913                                check_dialog( widget, None, "I can't resolv !")
914                            except socket.error:
915                                check_dialog( widget, None, "Error in gethostbyaddr()!")
916                        else:
917                            is_good = 1
918                    ##here, check the prefix value?
919                    if (field=='60' or field=='46'):
920                        if (int(value) < 256):
921                            is_good = 1
922                        else:
923                            check_dialog( widget, None, "Wrong value, must be < 256 !")
924                            is_good = 0
925                    if (field=='1'):
926                        if (int(value) < 4294967295):
927                            is_good = 1
928                        else:
929                            check_dialog( widget, None, "Wrong value, must be < 4294967295 !")
930                            is_good = 0
931                    if (field=='7' or field=='11' or field=='4'):
932                        if (int(value) < 65536):
933                            is_good = 1
934                        else:
935                            check_dialog( widget, None, "Wrong value, must be < 65536 !")
936                            is_good = 0
937                    if (field=='10' or field=='14' or field=='16' or field=='17'):
938                        if (int(value) < 65536):
939                            is_good = 1
940                        else:
941                            check_dialog( widget, None, "Wrong value, must be < 65536 !")
942                            is_good = 0
943            if (is_good==1 and is_ok==1):
944                ##collPrefix = adip & mask[1]
945                ## Here tcp exchange between collector & client
946                ##msg = str(localPort) +" "+ field +" "+ operator+" "+ value
947                msg = str(localPort) + " " + mymsg
948                print msg
949                sent_rule(widget, msg)
950                widget.set_editable(False)
951                coll_window.destroy()
952    else:
953        check_dialog( widget, None, "Internal program error ;) !")
954
955# collector control button
956def prefix_entry(maxLen, tooltip_Text, parent, lab, id):
957    global mainTT
958    vbox = gtk.VBox(False, 0)
959    parent.pack_start(vbox, False, True, 0)
960    label = gtk.Label(lab)
961    label.set_alignment(0, 0.5)
962    vbox.pack_start(label, False, True, 0)
963    entry = gtk.Entry(max=maxLen)
964    entry.set_max_length(maxLen)
965    entry.connect("activate", prefix_callback, entry, id)
966    entry.set_text("")
967    entry.select_region(0, len(entry.get_text()))
968    if tooltip_Text!=None:
969        mainTT.set_tip(entry, tooltip_Text)
970    vbox.pack_start(entry, False, True, 0)
971    entry.show()
972
973def user_quit(widget, data=None):
974    global myInputControlThread, localPort, record_file, myTimer
975    if myInputControlThread != None:
976        myInputControlThread.kill(1)
977        myInputControlThread = None
978        s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
979        s.connect(('localhost', localPort))
980        s.send("FIN")
981        s.close()
982        if (record_file!=None):
983            record_file.close()
984        sent_rule_from_cli("DELETE")
985        print "\n\n\n\n\n Input Stream closed. \n\n\n\n\n Bye :-) \n\n\n\n"
986    if (myTimer != None):
987        myTimer.cancel()
988    gtk.main_quit()
989
990# Flow Selection
991def enter_callback( widget, entry, id):
992    global firstParseValue
993    entry_text = entry.get_text()
994    if id == 0:
995        if (entry_text==""):
996            checkUpValues[0] = 0
997        else:
998            if (re.search('\D',entry_text)) or (int(entry_text) >= 99) or (int(entry_text) <= 0):
999                check_dialog( widget, None, "Wrong value !")
1000            else:
1001                firstParseValue[0] = int(entry_text)
1002                checkUpValues[0] = 1
1003    if id == 1:
1004        if ( entry_text==""):
1005            checkUpValues[8] = 0
1006            checkUpValues[12] = 0
1007        else:
1008            if (re.match('^\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}(/\d{1,2})?$',entry_text)):
1009                splitter = entry_text.split('/')
1010                adip = struct.unpack('>L',socket.inet_aton(splitter[0]))[0]
1011                if len(splitter)>1:
1012                    bits = int(splitter[1])
1013                    if bits <= 32 and bits > 0:
1014                        mask[0] = 0xffffffffL << (32 - bits)
1015                    else:
1016                        check_dialog( widget, None, "Wrong value in mask!")
1017                else:
1018                    mask[0] = 0xffffffffL
1019                firstParseValue[8] = adip & mask[0]
1020                checkUpValues[8] = 1
1021                firstParseValue[12] = adip & mask[0]
1022                checkUpValues[12] = 1
1023            else:
1024                check_dialog( widget, None, "Wrong value !")
1025    if id == 2:
1026        if ( entry_text==""):
1027            checkUpValues[2] = 0
1028        else:
1029            check_dialog( widget, None, "NOT YET IMPLEMEMTED")
1030            ## pb with IPv6 and getaddrinfo() ??
1031##            splitter = entry_text.split('/')
1032##            try:
1033##                socket.gethostbyaddr(splitter[0])
1034##            except socket.gaierror:
1035##                check_dialog( widget, None, "Wrong value !")
1036##            except socket.herror: ## can't resolv
1037##                pass
1038##            except socket.error:
1039##                pass
1040    if id == 3:
1041        if ( entry_text==""):
1042            checkUpValues[15] = 0
1043        else:
1044            if (re.match('^\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}(/\d{1,2})?$',entry_text)):
1045                splitter = entry_text.split('/')
1046                adip = struct.unpack('>L',socket.inet_aton(splitter[0]))[0]
1047                if len(splitter)>1:
1048                    bits = int(splitter[1])
1049                    if bits <= 32 and bits > 0:
1050                        mask[2] = 0xffffffffL << (32 - bits)
1051                    else:
1052                        check_dialog( widget, None, "Wrong value in mask!")
1053                else:
1054                    mask[2] = 0xffffffffL
1055                firstParseValue[15] = adip & mask[2]
1056                checkUpValues[15] = 1
1057            else:
1058                check_dialog( widget, None, "Wrong value !")
1059    if id == 4: ## In|Out
1060        if ( entry_text==""):
1061            checkUpValues[10] = 0
1062            checkUpValues[14] = 0
1063        else:
1064            if (re.search('\D',entry_text)) or (int(entry_text) > 65535) or (int(entry_text) < 0):
1065                check_dialog( widget, None, "Wrong value !")
1066            else:
1067                firstParseValue[10] = int(entry_text)
1068                checkUpValues[10] = 1
1069                firstParseValue[14] = int(entry_text)
1070                checkUpValues[14] = 1
1071    if id == 6: ## bytes
1072        if ( entry_text==""):
1073            checkUpValues[1] = 0
1074        else:
1075            if (re.search('\D',entry_text)) or (int(entry_text) > 999999999) or (int(entry_text) <= 0):
1076                check_dialog( widget, None, "Wrong value !")
1077            else:
1078                firstParseValue[1] = int(entry_text)
1079                if ( int(entry_text) == 0 ):
1080                    firstParseValue[1] = 1000000000
1081                else:
1082                    firstParseValue[1] = int(entry_text)
1083                checkUpValues[1] = 1
1084    if id == 7: ## pkts
1085        if ( entry_text==""):
1086            checkUpValues[2] = 0
1087        else:
1088            if (re.search('\D',entry_text)) or (int(entry_text) > 9999) or (int(entry_text) <= 0):
1089                check_dialog( widget, None, "Wrong value !")
1090            else:
1091                firstParseValue[2] = int(entry_text)
1092                checkUpValues[2] = 1
1093    if id == 8: ## time no concordance with nfv9 field number
1094        if ( entry_text==""):
1095            checkUpValues[0] = 0
1096        else:
1097            if (re.search('\D',entry_text)) or (int(entry_text) > 99) or (int(entry_text) < 0):
1098                check_dialog( widget, None, "Wrong value !")
1099            else:
1100                firstParseValue[0] = int(entry_text)
1101                checkUpValues[0] = 1
1102    if id == 9: ## port range
1103        if ( entry_text==""):
1104            checkUpValues[7] = 0
1105        else:
1106            port_list = string.split(entry_text)
1107            if (len(port_list)!=2)or((re.search('\D',port_list[0])) or (int(port_list[0]) > 65535) or (int(port_list[0]) <= 0) or (re.search('\D',port_list[1])) or (int(port_list[1]) > 65535) or (int(port_list[1]) <= 0)):
1108                check_dialog( widget, None, "Wrong value !")
1109            else:
1110                firstParseValue[7] = int(port_list[0])
1111                firstParseValue[11] = int(port_list[1])
1112                checkUpValues[7] = 1
1113    if id == 11: ## tcp flags
1114        if ( entry_text==""):
1115            checkUpValues[6] = 0
1116        else:
1117            if (re.search('\D',entry_text)) or (int(entry_text) > 255) or (int(entry_text) <= 0):
1118                check_dialog( widget, None, "Wrong value !")
1119            else:
1120                firstParseValue[6] = int(entry_text)
1121                checkUpValues[6] = 1
1122    if id == 12: ## prot
1123        if ( entry_text==""):
1124            checkUpValues[4] = 0
1125        else:
1126            if (re.search('\D',entry_text)) or (int(entry_text) > 255) or (int(entry_text) <= 0):
1127                check_dialog( widget, None, "Wrong value !")
1128            else:
1129                firstParseValue[4] = int(entry_text)
1130                checkUpValues[4] = 1
1131    if id == 13: ## tos
1132        if ( entry_text==""):
1133            checkUpValues[5] = 0
1134        else:
1135            if (re.search('\D',entry_text)) or (int(entry_text) > 255) or (int(entry_text) < 0):
1136                check_dialog( widget, None, "Wrong value !")
1137            else:
1138                firstParseValue[5] = int(entry_text)
1139                checkUpValues[5] = 1
1140    if id == 14: ## src AS
1141        if ( entry_text==""):
1142            checkUpValues[16] = 0
1143        else:
1144            if (re.search('\D',entry_text)) or (int(entry_text) > 65535) or (int(entry_text) < 0):
1145                check_dialog( widget, None, "Wrong value !")
1146            else:
1147                firstParseValue[16] = int(entry_text)
1148                checkUpValues[16] = 1
1149    if id == 15: ## dst AS
1150        if ( entry_text==""):
1151            checkUpValues[17] = 0
1152        else:
1153            if (re.search('\D',entry_text)) or (int(entry_text) > 65535) or (int(entry_text) < 0):
1154                check_dialog( widget, None, "Wrong value !")
1155            else:
1156                firstParseValue[17] = int(entry_text)
1157                checkUpValues[17] = 1
1158    if id == 18: ## from router
1159        if ( entry_text==""):
1160            checkUpValues[69] = 0
1161        else:
1162            if (re.match('^\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}(/\d{1,2})?$',entry_text)):
1163                splitter = entry_text.split('/')
1164                adip = struct.unpack('>L',socket.inet_aton(splitter[0]))[0]
1165                if len(splitter)>1:
1166                    bits = int(splitter[1])
1167                    if bits <= 32 and bits > 0:
1168                        mask[0] = 0xffffffffL << (32 - bits)
1169                    else:
1170                        check_dialog( widget, None, "Wrong value in mask!")
1171                else:
1172                    mask[0] = 0xffffffffL
1173                firstParseValue[69] = splitter[0]
1174                checkUpValues[69] = 1
1175            else:
1176                check_dialog( widget, None, "Wrong value !")
1177   
1178if __name__ == "__main__":
1179    global window, mainTT, comboRouter, os_type, printShit
1180    global localAddr, localPort
1181   
1182    gtk.gdk.threads_init()
1183   
1184    if (os.name =="nt"):
1185        os_type = 1
1186    elif (os.name =="posix"):
1187        os_type = 0
1188    else:
1189        os_type = 3
1190
1191    parser = OptionParser()
1192    parser.add_option("-r", "--router", dest="rtr", help ="router address used for NetFlow data export")
1193    parser.add_option("-i", "--index", dest="ind", help ="Interface index get with snmpwalk")
1194    (options, args) = parser.parse_args()
1195    rtr = options.rtr
1196    ind = options.ind
1197
1198    if len(sys.argv)==5:
1199        myInputControlThread = InputControl(localAddr, localPort)
1200        myInputControlThread.start()
1201        msg = str(localPort) + " C_0_=_" + rtr + "_C_10_=_" + ind
1202        print msg
1203        sent_rule_from_cli(msg)
1204       
1205    print "---------------------------------------------------------------"
1206    print "In this version, you can only apply rules on the collector based on the following fields:"
1207    print "IN_BYTES: 1,"
1208    print "L4_PROT: 4,"
1209    print "L4_SRC_PORT: 7,"
1210    print "IPV4_SRC_ADDR: 8,"
1211    print "INPUT_SNMP: 10,"
1212    print "L4_DST_PORT: 11,"
1213    print "IPV4_DST_ADDR: 12,"
1214    print "OUTPUT_SNMP: 14,"
1215    print "IPV4_NEXT_HOP: 15,"
1216    print "SRC_AS: 16,"
1217    print "DST_AS: 17,"
1218    print "BGP_IPV4_NEXT_HOP: 18,"
1219    print "IPV6_SRC_ADDR: 27,"
1220    print "IPV6_DST_ADDR: 28,"
1221    print "MPLS_TOP_LABEL_TYPE: 46,"
1222    print "IPV6_NEXT_HOP: 62,"
1223    print "BPG_IPV6_NEXT_HOP: 63,"
1224    print "IP_PROTOCOL_VERSION: 60, please note that the IPv4 templates don't have this field"
1225    print "ROUTER_SRC_ADDR: 0, !!! it's not a real number for a field, but it's a great function"
1226    print "----------------------------------------------------------------"
1227   
1228    mainTT = gtk.Tooltips()
1229    window = gtk.Window(gtk.WINDOW_TOPLEVEL)   
1230    window.connect("destroy", gtk.main_quit, "WM destroy")
1231    window.set_title("RenetcolGUI")
1232    window.set_size_request(850, 550)
1233   
1234    main_vbox = gtk.VBox(False, 1)
1235    main_vbox.set_border_width(1)
1236    window.add(main_vbox)
1237
1238##  menu
1239    file_menu = gtk.Menu()
1240    receive_item = gtk.MenuItem("Receive Stream")
1241    stop_item = gtk.MenuItem("Stop Stream")
1242    quit_item  = gtk.MenuItem("Quit")
1243    file_menu.append(receive_item)
1244    file_menu.append(stop_item)
1245    file_menu.append(quit_item)
1246    receive_item.connect("activate", reception, None)
1247    stop_item.connect("activate", close_recept, None)
1248    quit_item.connect("activate", user_quit, None)
1249    receive_item.show()
1250    stop_item.show()
1251    quit_item.show()
1252
1253    help_menu = gtk.Menu()
1254    about_item = gtk.MenuItem("About")
1255    help_menu.append(about_item)
1256    about_item.connect("activate", about_dialog, None)
1257    about_item.show()
1258
1259    menu_bar = gtk.MenuBar()
1260    main_vbox.pack_start(menu_bar, False, False, 0)
1261    menu_bar.show()
1262    file_item = gtk.MenuItem("File")
1263    file_item.show()
1264    help_item = gtk.MenuItem("Help")
1265    help_item.show()
1266
1267    file_item.set_submenu(file_menu)
1268    menu_bar.append(file_item)
1269    menu_bar.append(file_menu)
1270    file_item.set_right_justified(0)
1271
1272    help_item.set_submenu(help_menu)
1273    menu_bar.append(help_item)
1274    menu_bar.append(help_menu)
1275    help_item.set_right_justified(1)
1276
1277##  first frame
1278    frame = gtk.Frame("Input Stream Control")
1279    main_vbox.pack_start(frame, False, True, 0)
1280    hbox = gtk.HBox(False, 0)
1281    hbox.set_border_width(5)
1282    frame.add(hbox)
1283
1284    ## freeze button
1285    buttonfreeze = gtk.ToggleButton("Freeze")
1286    buttonfreeze.connect("toggled", callback_freeze, None)
1287    mainTT.set_tip(buttonfreeze, "Stop/Start the output in the window.")
1288    hbox.pack_start(buttonfreeze, False, False, 2)
1289    buttonfreeze.show()
1290    ## record button
1291    buttonrecord = gtk.ToggleButton("Record")
1292    buttonrecord.connect("toggled", callback_record, None)
1293    mainTT.set_tip(buttonrecord, "Stop/Start the record in a text file.")
1294    hbox.pack_start(buttonrecord, False, False, 2)
1295    buttonrecord.show()
1296    ## incremental record
1297    buttonRotRec = gtk.ToggleButton("Rot. Record")
1298    buttonRotRec.connect("toggled", callback_RotRec, None)
1299    mainTT.set_tip(buttonRotRec, "Stop/Start the record in a different file for each day.")
1300    hbox.pack_start(buttonRotRec, False, False, 2)
1301    buttonRotRec.show()
1302    ## control collector button
1303    buttoncollector = gtk.ToggleButton("Collector Control")
1304    buttoncollector.connect("toggled", callback_collector, None)
1305    mainTT.set_tip(buttoncollector, "Change the rule which is applied on the collector. You must have activated the input stream before.")
1306    hbox.pack_start(buttoncollector, False, False, 2)
1307    buttoncollector.show()
1308
1309    ipv4_vbox = gtk.VBox(False, 1)
1310    ipv4_vbox.set_border_width(1)
1311    main_vbox.add(ipv4_vbox)
1312##  second frame
1313    frame = gtk.Frame("Select printed fields")
1314    ipv4_vbox.pack_start(frame, False, False, 0)
1315   
1316    vbox = gtk.VBox(False, 0)
1317    vbox.set_border_width(5)
1318    frame.add(vbox)
1319    bbox = gtk.HButtonBox()
1320    vbox.pack_start(bbox, False, True, 5)
1321   
1322    create_button("Src \n IP", 1, callback_fields_printed, 8, bbox, "Source IP")
1323    create_button("Dst\n IP", 1, callback_fields_printed, 12, bbox, "Destination IP")
1324    create_button("Flw\n IP", 0, callback_fields_printed, 15, bbox, "Following IP")
1325    create_button("Index\n In", 0, callback_fields_printed, 10, bbox, "SNMP Index Input")
1326    create_button("Index\n Out", 0, callback_fields_printed, 14, bbox, "SNMP Index Output")
1327    create_button("Size (P)", 1, callback_fields_printed, 2, bbox, "Packets number")
1328    create_button("Size (B)", 1, callback_fields_printed, 1, bbox, "Size in Bytes")
1329    create_button("Time", 0, callback_fields_printed, 0, bbox, "Age of flow")
1330
1331    b2box = gtk.HButtonBox()
1332    vbox.pack_start(b2box, False, True, 5)
1333    create_button("Src\n Port", 1, callback_fields_printed, 7, b2box, "TCP/UDP source port")
1334    create_button("Dst\n Port", 1, callback_fields_printed, 11, b2box, "TCP/UDP destination port")
1335    create_button("TCP\n Flags", 0, callback_fields_printed, 6, b2box, "TCP Flags")
1336    create_button("Prot", 1, callback_fields_printed, 4, b2box, "Protocol")
1337    create_button("Tos", 0, callback_fields_printed, 5, b2box, "Type of service")
1338    create_button("Src\nAs", 0, callback_fields_printed, 16, b2box, "Source AS")
1339    create_button("Dst\nAs", 0, callback_fields_printed, 17, b2box, "Destination AS")
1340    create_button("Src\nMask", 0, callback_fields_printed, 9, b2box, "Source Mask")
1341    create_button("Dst\nMask", 0, callback_fields_printed, 13, b2box, "Destination Mask")
1342
1343    b3box = gtk.HButtonBox()
1344    vbox.pack_start(b3box, False, True, 5)
1345    create_button("Router", 0, callback_fields_printed, 69, b3box, "Router address")
1346
1347##  third frame
1348    frame = gtk.Frame("Parsing values")
1349    ipv4_vbox.pack_start(frame, False, False, 5)
1350    hbox = gtk.VBox(False, 0)
1351    hbox.set_border_width(5)
1352    frame.add(hbox)
1353    pbbox = gtk.HButtonBox()
1354    hbox.pack_start(pbbox, False, True, 5)
1355##  here entry button
1356    create_entry(18, "Required Format : \n an IPv4 address like x.x.x.x \n or a network address like x.x.x.x/x", pbbox, "IPv4 address :", 1)
1357    create_entry(39, "Required Format : \n an IPv6 address like x:x:x:x:x::x \n or a network address like x:x:x/x NOT YET SUPPORTED", pbbox, "(IPv6 address :)", 2)
1358    create_entry(18, "Required Format : \n an IP address like x.x.x.x \n or a network address like x.x.x.x/x", pbbox, "Flw IP ad. :", 3)
1359    dbbox = gtk.HButtonBox()
1360    hbox.pack_start(dbbox, False, True, 5)
1361    create_entry(5, "Required Format : a number less than 65535", dbbox, "SNMP Index In|Out :", 4)
1362    create_entry(4, "Required Format : a number less than 9999", dbbox, "Packet NB :", 7)
1363    create_entry(17, "Required Format : a number less than 999999999, each flow who the size equal this number +/- 5%", dbbox, "Bytes :", 6)
1364    tbbox = gtk.HButtonBox()
1365    hbox.pack_start(tbbox, False, True, 5)
1366    create_entry(2, "Required Format : a number less than 99", tbbox, "Time :", 8)
1367    create_entry(11, "Required Format : a range, each number less than 65535 and space as separator: \"6881 6889\"", tbbox, "Port :", 9)
1368    #create_entry(5, "Required Format : a number less than 65535", tbbox, "Dst port :", 10)
1369    create_entry(3, "Required Format : a number less than 255", tbbox, "TCP Flag :", 11)
1370    fbbox = gtk.HButtonBox()
1371    hbox.pack_start(fbbox, False, True, 5)
1372    create_entry(5, "Required Format : a number less than 65535", fbbox, "Protocol :", 12)
1373    create_entry(3, "Required Format : a number less than 255", fbbox, "ToS :", 13)
1374    create_entry(5, "Required Format : a number less than 65535", fbbox, "Src AS :", 14)
1375    create_entry(5, "Required Format : a number less than 65535", fbbox, "Dst AS :", 15)
1376    kbbox = gtk.HButtonBox()
1377    hbox.pack_start(kbbox, False, True, 5)
1378    create_entry(18, "Required Format : \n an IP address like x.x.x.x or x.x.x.x/x", kbbox, "The flows sending by this Router :", 18)
1379
1380##  Show
1381    vbox.show()
1382    main_vbox.show()
1383    window.show_all()
1384    mainTT.enable()
1385
1386##  gtk main loop
1387    gtk.gdk.threads_enter()
1388    gtk.main()
1389    gtk.gdk.threads_leave()
Note: See TracBrowser for help on using the browser.