root/trunk/tool/renetcolGUI_0_0_2.py @ 30

Revision 2, 44.9 KB (checked in by andreu, 17 years ago)

First RENETCOL CVS Integration

  • Property svn:eol-style set to native
Line 
1
2##  File: renetcolGUI_0_0_2.py
3 
4##  Authors: ANDREU Francois-Xavier
5 
6##  Copyright (C) 2005 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 threading
37import string
38import re
39import socket
40import sys
41import gtk
42import traceback
43import pango
44import commands
45
46########### DEFAULT VALUES, YOU MUST CHANGE IT ###############################
47IPversion = 4
48localAddr = ''
49localAddr6 = '2001:660:3001:4012:30ec:9207:2a75:1583'
50localPort = 2222
51collectorAddr = "193.51.183.185"
52collectorAddr6 = "2001:660:3000:1011:10::"
53collectorPort = 52571  # if you change it, apply the modification on
54                       # the renetcolSender.h file and recompil the collector
55##############################################################################
56
57myInputControlThread = None
58printValue =      [ 0,1,1,0,1,0,0,1,1,0,
59                    0,1,1,0,0,0,0,0,0,0,
60                    1,0,0,0,0,0,0,1,1,0,
61                    0,0,0,0,0,0,0,0,0,0,
62                    0,0,0,0,0,0,0,0,0,0,
63                    0,0,0,0,0,0,0,0,0,0,
64                    0,0,0,0,0,0,0,0,0,0,
65                    0,0,0,0,0,0,0,0,0,0,
66                    0,0,0,0,0,0,0,0,0,0
67                    ]
68firstParseValue = [ 0,0,0,0,0,0,0,0,0,0,
69                    0,0,0,0,0,0,0,0,0,0,
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,
75                    0,0,0,0,0,0,0,0,0,0,
76                    0,0,0,0,0,0,0,0,0,0
77                    ]
78checkUpValues =   [ 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                    ]
88fieldsName =     [ "","IN_BYT","IN_PKTS","","PROT","TOS","TCP_FL","S_PORT","IPv4@S","S_MASK",
89                   "InSNMP","D_PORT","IPv4@D","D_MASK","OutSNMP","NextHop","S_AS","D_AS","","",
90                   "","Time","","","","","","IPv6@S","IPv6@D","",
91                   "","","","","","","","","","",
92                   "","","","","","","","","","",
93                   "","","","","","","","","","",
94                   "","","","","","","","","","ROUTER",
95                   "","","","","","","","","","",
96                   "","","","","","","","","",""
97                   ]
98flowCpt = 0
99mask = [ 0,0,0 ]
100spaceSep = " "
101tabSep = "\t"
102freeze = 0
103record = 0
104record_file_name = ""
105record_file = None
106is_already_see = 0
107collectorRule=""
108coll_window = gtk.Window()
109router=0
110tpl_def = {}
111os_type = 0
112oldTpl = 0
113fromRouter = ""
114
115class InputControl(threading.Thread):
116    def kill(self, timeout):
117        self.imRunning = 0
118        time.sleep(1)
119        print "\n.\n..\n..."
120        time.sleep(1)
121        print "....\n.....\n......"       
122        time.sleep(1)
123        if self.myFlowInput:
124            self.myFlowInput.socket_close()
125        self.join(timeout)
126       
127    def __init__(self, address, port):
128        self.imRunning = 1
129        self.currentflow = None
130        threading.Thread.__init__(self)
131        self.myFlowInput = FlowInput(address, port)
132
133    def run(self):
134        global freeze, record
135        while self.imRunning:
136            if self.myFlowInput:
137                tmp = self.myFlowInput.get_flow()
138                if (tmp[0]!=None):
139                    self.currentflow = Flow(tmp)
140                    if (freeze==0) & self.currentflow.enable():
141                        self.currentflow.print_flow2()
142                    if (record==1) & self.currentflow.enable():
143                        self.currentflow.write_flow2()
144##                else:
145##                    self.myFlowInput = None
146   
147class Flow:
148    def __init__(self, data):
149        self.flow = [ data[0], data[1] ]
150
151    def enable(self):
152        global firstParseValue, checkUpValues, tpl_def
153        res = 1
154        for i in range (0, len(tpl_def[self.flow[0]])):
155            f = tpl_def[self.flow[0]][i][0]
156            if ( f==8 or f==12 ) and ( checkUpValues[f]==1 ):
157                res = res & ( ((struct.unpack('>L',(socket.inet_aton(self.flow[1][i])))[0] & mask[1-1]) == firstParseValue[f])| ((struct.unpack('>L',(socket.inet_aton(self.flow[1][i])))[0] & mask[1-1]) == firstParseValue[f]) )
158            elif ( i==27 or i==28 ):
159                pass
160            elif (checkUpValues[f]==1):
161                res = res & (self.flow[1][i] == firstParseValue[f])
162        return res
163               
164    def print_flow2(self):
165        global printValue, tpl_def, flowCpt, fieldsName, oldTpl, os_type, fromRouter
166        myliste = []
167        myTpl = []
168        strFlow = ""
169        strField = ""
170        underscore_line = ""
171        flowCpt+=1
172        if (flowCpt%20 == 0 or oldTpl != self.flow[0]):
173            flowCpt = 0
174            for i in range (0, len(tpl_def[self.flow[0]])):
175                f = tpl_def[self.flow[0]][i][0]
176                if printValue[f]:
177                    if ( f==27 or f==28 or f==62 or f==63 ):
178                        strField += str(fieldsName[f])
179                        l = len(str(fieldsName[f]))
180                        if ((40-l)%8 == 0):
181                            tabNb = int((40-l)/8)
182                        else:
183                            tabNb = int((40-l)/8) + 1
184                        for j in range (0, tabNb):
185                            strField += "\t"
186                        for k in range (0, 40):
187                            underscore_line += "-"
188                    elif ( f==8 or f==15 or f==12 ):
189                        strField += str(fieldsName[f])
190                        l = len(str(fieldsName[f]))
191                        if ((16-l)%8 == 0):
192                            tabNb = int((16-l)/8)
193                        else:
194                            tabNb = int((16-l)/8) + 1
195                        for j in range (0, tabNb):
196                            strField += "\t"
197                        for k in range (0, 16):
198                            underscore_line += "-"
199                    else:
200                        strField += str(fieldsName[f])
201                        strField += "\t"
202                        for k in range (0, 8):
203                            underscore_line += "-"
204            if (oldTpl != self.flow[0]):
205                print " "
206            if (os_type == 0):
207                esc = '\x1b['
208                sep = ';'
209                end = 'm'
210                if (printValue[69]):
211                    rt = str(fieldsName[69])
212                    toprint = esc+"37"+sep+"44"+end+strField+rt+esc+"0"+end
213                else:
214                    toprint = esc+"37"+sep+"44"+end+strField+esc+"0"+end
215                print toprint
216            else:
217                if (printValue[69]):
218                    strField += str(fieldsName[69])
219                print strField
220                print underscore_line
221        for i in range (0, len(tpl_def[self.flow[0]])):
222            f = tpl_def[self.flow[0]][i][0]
223            if printValue[f]:
224                if ( f==27 or f==28 or f==62 or f==63 ):
225                    strFlow += str(self.flow[1][i])
226                    l = len(str(self.flow[1][i]))
227                    if ((40-l)%8 == 0):
228                        tabNb = int((40-l)/8)
229                    else:
230                        tabNb = int((40-l)/8) + 1
231                    for j in range (0, tabNb):
232                        strFlow += "\t"
233                elif ( f==8 or f==12 or f==15 ):
234                    strFlow += str(self.flow[1][i])
235                    l = len(str(self.flow[1][i]))
236                    if ((16-l)%8 == 0):
237                        tabNb = int((16-l)/8)
238                    else:
239                        tabNb = int((16-l)/8) + 1
240                    for j in range (0, tabNb):
241                        strFlow += "\t"
242                elif (f==21):
243                    myliste += [str(self.flow[1][i]-self.flow[1][i+1])]
244                    strFlow += str(self.flow[1][i]-self.flow[1][i+1])
245                    strFlow += "\t"
246                elif (f==22):
247                    pass
248                else:
249                    myliste += [str(self.flow[1][i])]
250                    strFlow += str(self.flow[1][i])
251                    for k in range (len(str(self.flow[1][i])), 7):
252                        strFlow += " "
253                    strFlow += "\t"
254        if (printValue[69]):
255            strFlow += fromRouter
256        print strFlow
257        oldTpl = self.flow[0]
258
259    def write_flow2(self):
260        global printValue, record_file, tpl_def
261        myliste = []
262        for i in range (0, len(tpl_def[self.flow[0]])):
263            f = tpl_def[self.flow[0]][i][0]
264            if printValue[f]:
265                if (f==21):
266                    myliste += [str(self.flow[1][i]-self.flow[1][i+1])]
267                elif (f==22):
268                    pass
269                else:
270                    myliste += [str(self.flow[1][i])]
271        theflow = string.join(myliste, "\t")+"\n" 
272        record_file.write(theflow)
273       
274class FlowInput:
275    def __init__(self, h, p):
276        self.HOST = h
277        self.PORT = p
278        if (IPversion == 4):
279            self.socketnumber = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
280        elif (IPversion == 6):
281            self.socketnumber = socket.socket(socket.AF_INET6, socket.SOCK_DGRAM)
282        else:
283            print "IP protocol ", IPversion ," not supported"
284            exit(0)
285        self.socketnumber.bind((self.HOST, self.PORT))
286   
287    def get_flow(self):
288        global collectorRule, tpl_def, os_type, fromRouter
289        if (self.socketnumber!=None):
290            data = self.socketnumber.recvfrom(1024)
291            flow = data[0]
292            code = struct.unpack('<H', flow[0:2])
293            if (code[0] == 1): ## a def packet
294                routerSrc = struct.unpack('<L', flow[2:6])
295                sourceId = struct.unpack('<L', flow[6:10])
296                tplId = struct.unpack('<H', flow[10:12])
297                mykeylst = []
298                mykeylst += [str(routerSrc[0])]
299                mykeylst += [str(sourceId[0])]
300                mykeylst += [str(tplId[0])]
301                mykeystr = string.join(mykeylst,"");
302                fields = []
303                fieldNb = struct.unpack('<H', flow[12:14])
304                for i in range(0, fieldNb[0]):
305                    fd = struct.unpack('<H',flow[14+(i*4):16+(i*4)])
306                    tp = struct.unpack('<H',flow[14+(2+i*4):16+(2+i*4)])
307                    fields.insert(i,(fd[0],tp[0]))
308                tpl_def[mykeystr] = fields
309                return ( None, None)
310            if (code[0]==11): ## a data packet
311                routerSrc = struct.unpack('<L', flow[4:8])
312                tmp =  struct.unpack('<L', flow[4:8])
313                fromRouter = socket.inet_ntoa(struct.pack('>L', tmp[0]))
314                sourceId = struct.unpack('<L', flow[8:12])
315                tplId = struct.unpack('<H', flow[12:14])
316                mykeylst = []
317                mykeylst += [str(routerSrc[0])]
318                mykeylst += [str(sourceId[0])]
319                mykeylst += [str(tplId[0])]
320                mykeystr = string.join(mykeylst,"")
321                if (tpl_def.has_key(mykeystr)== True):
322                    field_list = []
323                    index = 0
324                    for i in range (0, len(tpl_def[mykeystr])):
325                        if (tpl_def[mykeystr][i][1]==1):
326                            value = struct.unpack('<B',flow[14+index:15+index])
327                            field_list.insert(i,value[0])
328                        if (tpl_def[mykeystr][i][1]==2):
329                            value = struct.unpack('>H',flow[14+index:16+index])
330                            field_list.insert(i,value[0])
331                        if (tpl_def[mykeystr][i][1]==4):
332                            if (tpl_def[mykeystr][i][0]==8 or tpl_def[mykeystr][i][0]==12 or tpl_def[mykeystr][i][0]==15):
333                                value = socket.inet_ntoa(flow[14+index:18+index])
334                                field_list.insert(i,value)
335                            else:
336                                value = struct.unpack('<L', flow[14+index:18+index])
337                                value = struct.unpack('>L', flow[14+index:18+index])
338                                field_list.insert(i,value[0])
339                        if (tpl_def[mykeystr][i][1]==16):
340                            if (os_type==1):
341                                field_list.insert(i,nt_inet_ntop(flow[14+index:30+index]))
342                            else:
343                                tmp = socket.inet_ntop(socket.AF_INET6,flow[14+index:30+index])
344                                field_list.insert(i,tmp)
345                        index += tpl_def[mykeystr][i][1]
346                    return mykeystr, field_list
347##                else:
348##                  print "I'm waiting the tpl def for the key", mykeystr
349##                return (routerSrc[0], field_list)
350            return ( None, None)
351        return ( None, None)
352
353    def socket_close(self):
354        if self.socketnumber:
355           try :
356               self.socketnumber.shutdown(2)
357               self.socketnumber = None
358           except socket.error:
359              pass
360
361class FileSel:
362    def file_ok_sel(self, w):
363        global record, record_file, record_file_name
364        record = 1
365        record_file_name = self.fileSel.get_filename()
366        record_file = file(record_file_name, 'w')
367        print "%s" % self.fileSel.get_filename()
368        self.fileSel.destroy()
369
370    def destroy(self, widget):
371        self.fileSel.destroy()
372
373    def __init__(self):
374        self.fileSel = gtk.FileSelection("File Selection")
375        self.fileSel.connect("destroy", self.destroy)
376        self.fileSel.ok_button.connect("clicked", self.file_ok_sel)
377        self.fileSel.cancel_button.connect("clicked",
378                                           lambda w: self.fileSel.destroy())
379        self.fileSel.set_filename("flow_record.txt")
380        self.fileSel.show()
381
382def print_hello(widget, event):
383    print "Hello, World!"
384
385def nt_inet_ntop(packed_ip):
386    cpt = 0
387    double = 0
388    string_ip = ""
389    tmp = binascii.hexlify(packed_ip[0:2])
390    if ( int(tmp,16)!=0 ):
391        string_ip += tmp
392        tmp = binascii.hexlify(packed_ip[2:4])
393        if ( int(tmp,16)==0 ):
394            cpt+=1
395            double = 1
396            if (cpt==1):
397                string_ip += ":"
398        else:
399            cpt=0
400            string_ip += ":"+tmp
401        tmp = binascii.hexlify(packed_ip[4:6])
402        if ( int(tmp,16)==0 ):
403            cpt+=1
404            double = 1
405            if (cpt==1):
406                string_ip += ":"
407        else:
408            cpt=0
409            string_ip += ":"+tmp
410        tmp = binascii.hexlify(packed_ip[6:8])
411        if ( int(tmp,16)==0 ):
412            if (double==1 and cpt==0):
413                string_ip += ":0"
414            else:
415                cpt+=1
416                double = 1
417                if (cpt==1):
418                    string_ip += ":"
419        else:
420            cpt=0
421            string_ip += ":"+tmp
422        tmp = binascii.hexlify(packed_ip[8:10])
423        if ( int(tmp,16)==0 ):
424            if (double==1 and cpt==0):
425                string_ip += ":0"
426            else:
427                cpt+=1
428                double = 1
429                if (cpt==1):
430                    string_ip += ":"
431        else:
432            cpt=0
433            string_ip += ":"+tmp
434        tmp = binascii.hexlify(packed_ip[10:12])
435        if ( int(tmp,16)==0 ):
436            if (double==1 and cpt==0):
437                string_ip += ":0"
438            else:
439                cpt+=1
440                double = 1
441                if (cpt==1):
442                    string_ip += ":"
443        else:
444            cpt=0
445            string_ip += ":"+tmp
446        tmp = binascii.hexlify(packed_ip[12:14])
447        if ( int(tmp,16)==0 ):
448            if (double==1 and cpt==0):
449                string_ip += ":0"
450            else:
451                cpt+=1
452                double = 1
453                if (cpt==1):
454                    string_ip += ":"
455        else:
456            cpt=0
457            string_ip += ":"+tmp
458        tmp = binascii.hexlify(packed_ip[14:16])
459        if ( int(tmp,16) != 0 ):
460            string_ip += ":"+tmp
461        else:
462            string_ip += ":"
463    else:
464        string_ip = "::"
465    return string_ip
466
467def get_main_menu(self, window):
468    accel_group = gtk.AccelGroup()
469    item_factory = gtk.ItemFactory(gtk.MenuBar, "<main>", accel_group)
470    item_factory.create_items(menu_items)
471    window.add_accel_group(accel_group)
472    item_factory = item_factory
473    return item_factory.get_widget("<main>")
474
475def reception(widget, event):
476    global myInputControlThread
477    global localAddr, localPort
478    myInputControlThread = InputControl(localAddr, localPort)
479    myInputControlThread.start()
480
481def close_recept(widget, event):
482    global myInputControlThread
483    if myInputControlThread != None:
484        myInputControlThread.kill(1)
485        myInputControlThread = None
486        print "\n\n\n\n\n Input Stream closed. \n\n\n\n\n"
487
488def about_dialog(widget, event):
489    dialog = gtk.MessageDialog(
490        parent         = None,
491        flags          = gtk.DIALOG_MODAL,
492        type           = gtk.MESSAGE_INFO,
493        buttons        = gtk.BUTTONS_OK,
494        message_format = "Author : FX Andreu\nemail andreu@renater.fr\nCopyright (C) 2005 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 ")
495    dialog.set_title('About')
496    dialog.connect('response', lambda dialog, response: dialog.destroy())
497    dialog.show()
498
499def check_dialog(widget, event, text):
500    dialog = gtk.MessageDialog(
501        parent         = None,
502        flags          = gtk.DIALOG_MODAL,
503        type           = gtk.MESSAGE_INFO,
504        buttons        = gtk.BUTTONS_OK,
505        message_format = text)
506    dialog.set_title('Value check ')
507    dialog.connect('response', lambda dialog, response: dialog.destroy())
508    dialog.show()
509
510def info_dialog(widget, event, text):
511    dialog = gtk.MessageDialog(
512        parent         = None,
513        flags          = gtk.DIALOG_MODAL,
514        type           = gtk.MESSAGE_INFO,
515        buttons        = gtk.BUTTONS_OK,
516        message_format = text)
517    dialog.set_title('Information ')
518    dialog.connect('response', lambda dialog, response: dialog.destroy())
519    dialog.show()
520
521def callback_freeze(widget, data=None):
522    global freeze
523    if widget.get_active() == 1:
524        freeze = 1
525        print "FREEZE ACTIF"
526    if widget.get_active() == 0:
527        freeze = 0
528        print "FREEZE NOT ACTIF"
529
530def wind_destroy(widget, data=None):
531    widget.destroy()
532   
533def callback_record(widget, data=None):
534    global record, record_file, record_file_name
535    if widget.get_active() == 1:
536        FileSel()
537    if widget.get_active() == 0:
538        record = 0
539        time.sleep(2)
540        record_file.close()
541
542def strsend(target, message):
543   global collectorAddr, collectorPort
544   s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
545   s.connect((collectorAddr, collectorPort))
546   s.send(message)
547   s.close()
548
549def callback_refresh(widget, parent):
550    global collectorAddr
551    strsend(collectorAddr, 'REFRESH')
552    parent.destroy()
553   
554def callback_delete_rule(widget, parent):
555    global collectorAddr, collectorPort
556    if widget.get_active() == 1:
557        sent_rule(widget, "DELETE")
558        widget.set_active(0)
559        parent.destroy()
560
561def callback_check_rule(widget, data=None):
562    global router
563    if widget.get_active()==1:
564        router=1
565
566def callback_collector(widget, controlWindow):
567    global is_already_see, collectorRule, myInputControlThread, coll_window, collectorAddr
568    if ((myInputControlThread == None) and (widget.get_active() == 1)):
569        info_dialog(widget, None, "You must before activated the stream reception ! (File menu)")
570        widget.set_active(0)
571    else:
572        if widget.get_active() == 1:
573            widget.set_active(0)
574            coll_window.connect("destroy", wind_destroy)
575            coll_window.set_title("Collector control ")
576            coll_window.set_size_request(500, 250)
577            coll_vbox = gtk.VBox(False, 1)
578            coll_vbox.set_border_width(1)
579            coll_window.add(coll_vbox)
580            coll_frame = gtk.Frame("The prefix/IP address/router, validate by enter")
581            coll_vbox.pack_start(coll_frame, False, False, 5)
582            coll_hbox = gtk.VBox(False, 0)
583            coll_hbox.set_border_width(5)
584            coll_frame.add(coll_hbox)
585            button = gtk.CheckButton(" router ")
586            button.connect("toggled", callback_check_rule, None)
587            coll_hbox.pack_start(button, False, False, 0)
588            mainTT.set_tip(button, "Check it to capture all flows from one router and enter the IPv4 address of this router. NOT SUPPORTED IN THIS RELEASE")
589            button.show()
590            coll_pbbox = gtk.HButtonBox()
591            coll_hbox.pack_start(coll_pbbox, False, True, 5)
592            prefix_entry(45, "Required Format : \n <field> <operation> <address>\nlike 8 = 10.0.0.1 \nor 27 = 2001::1 (IPv6 address only on Linux system)\nonly '=' operator for the moment\n------------------------------\nIn this version 0.0.2, you can compare the following fields:\nIPV4_SRC_ADDR: 8\nIPV4_DST_ADDR: 12\nIPV4_NEXT_HOP: 15\nBGP_IPV4_NEXT_HOP: 18\nIPV6_SRC_ADDR: 27\nIPV6_DST_ADDR: 28\nIPV6_NEXT_HOP: 62\nBPG_IPV6_NEXT_HOP: 63\nIP_PROTOCOL_VERSION: 60\n---------------------------", coll_pbbox, "", 0)
593            coll_frame2 = gtk.Frame("Disable rules")
594            coll_vbox.pack_start(coll_frame2, False, False, 5)
595            coll_hbox2 = gtk.VBox(False, 0)
596            coll_hbox2.set_border_width(5)
597            coll_frame2.add(coll_hbox2)
598            coll_pbbox2 = gtk.HButtonBox()
599            coll_hbox2.pack_start(coll_pbbox2, False, True, 5)
600            buttondelete = gtk.ToggleButton("Delete")
601            buttondelete.connect("toggled", callback_delete_rule, coll_window)
602            mainTT.set_tip(buttondelete, "Delete all your rules of the collector configuration.")
603            coll_hbox2.pack_start(buttondelete, False, False, 2)
604            buttondelete.show()
605            coll_vbox.show()
606            coll_window.show_all()
607            if (is_already_see==0):
608                info_dialog(widget, None, "You should registered only one rule for each GUI parser ! Think of using the DELETE button !")
609                is_already_see=1
610
611def callback_fields_printed( widget, data=None):
612    global printValue
613    printValue[data] = widget.get_active()
614
615def callback_check_ip( widget, data=None):
616    global src_Or_Dst_IP
617    if widget.get_active()==1:
618        if checkUpValues[1]:
619            src_Or_Dst_IP = 1
620        else:
621            widget.set_active(0)
622            check_dialog(widget, None, "First, enter a source address.")
623    else:
624            src_Or_Dst_IP = 0
625
626def callback_comboR(widget, data=None):
627    global comboRouter, routerHash
628    nop = 0
629    print struct.unpack('>L',socket.inet_aton(routerHash[comboRouter.entry.get_text()]))
630   
631def callback_src_mask(widget, data=None):
632    nop = 0
633
634def callback_dst_mask(widget, data=None):
635    nop = 0
636
637def callback_tos(widget, data=None):
638    nop = 0
639
640def callback_prot(widget, data=None):
641    nop = 0
642
643def create_button(title, state, callback_function, ptr_var, parent, tooltip_Text):
644    global mainTT
645    button = gtk.ToggleButton(title)
646    if state==1:
647        button.set_active(True)
648    else:
649        button.set_active(False)
650    button.connect("toggled", callback_function, ptr_var)
651    if tooltip_Text!=None:
652        mainTT.set_tip(button, tooltip_Text)
653    parent.pack_start(button, False, True, 2)
654    button.show()
655
656def create_entry(maxLen, tooltip_Text, parent, lab, id):
657    global mainTT
658    vbox = gtk.VBox(False, 0)
659    parent.pack_start(vbox, False, True, 0)
660    label = gtk.Label(lab)
661    label.set_alignment(0, 0.5)
662    vbox.pack_start(label, False, True, 0)
663    entry = gtk.Entry(max=maxLen)
664    entry.set_max_length(maxLen)
665    entry.connect("activate", enter_callback, entry, id)
666    entry.set_text("")
667    entry.select_region(0, len(entry.get_text()))
668    if tooltip_Text!=None:
669        mainTT.set_tip(entry, tooltip_Text)
670    vbox.pack_start(entry, False, True, 0)
671    entry.show()
672
673def sent_rule(widget, msg):
674    global collectorAddr, collectorPort, collectorAddr6, IPversion
675    s = None
676    if (IPversion==4):
677        tmpaf =  socket.AF_INET
678        addr = collectorAddr
679        port = collectorPort
680    elif (IPversion == 6):
681        tmpaf =  socket.AF_INET6
682        addr = collectorAddr6
683        port = collectorPort
684    else:
685        print "Wrong IP version : ", IPversion
686        exit(0)
687    for res in socket.getaddrinfo(addr, port, tmpaf, socket.SOCK_STREAM):
688        af, socktype, proto, canonname, sa = res
689        try:
690            s = socket.socket(af, socktype, proto)
691        except socket.error, msg:
692            s = None
693            continue
694        try:
695            s.connect(sa)
696        except socket.error, msg:
697            s.close()
698            s = None
699            continue
700        break
701    if s is None:
702        check_dialog( widget, None, "could not open socket")
703    else:
704        s.send(msg)
705        check_dialog( widget, None, "Action was sent")
706        s.close()
707
708def prefix_callback( widget, entry, id):
709    global collPrefix, coll_window, router, localPort
710    entry_text = entry.get_text()
711    is_good = 0
712    if (id == 0):
713        if ( entry_text==""):
714            collPrefix = 0
715        else:
716            rule = entry_text.split(' ')
717            field = rule[0]
718            operator = rule[1]
719            value = rule[2]
720            if (field=='8' or field=='12' or field=='15' or field=='18'):
721                if (re.match('^\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}(/\d{1,2})?$',value)):
722                    splitter = value.split('/')
723                    adip = struct.unpack('>L',socket.inet_aton(splitter[0]))[0]
724                    if len(splitter)>1:
725                        bits = int(splitter[1])
726                        if bits <= 32 and bits > 0:
727                            mask[1] = 0xffffffffL << (32 - bits)
728                            is_good = 1
729                        else:
730                            check_dialog( widget, None, "Wrong value in mask!")
731                            is_good = 0
732                    else:
733                        mask[1] = 0xffffffffL
734                        is_good = 1
735                else:
736                    check_dialog( widget, None, "Wrong value !")
737            if (field=='27' or field=='28' or field=='62'):
738                try:
739                    socket.gethostbyaddr(value)
740                    is_good = 1
741                except socket.gaierror:
742                    check_dialog( widget, None, "Wrong value !")
743                except socket.herror: ## can't resolv
744                    check_dialog( widget, None, "I can't resolv !")
745                except socket.error:
746                    check_dialog( widget, None, "Error in gethostbyaddr()!")
747            if (field=='60'):
748                if (int(value) < 256):
749                    is_good = 1
750            if (is_good==1):
751                ##collPrefix = adip & mask[1]
752                ## Here tcp exchange between collector & client
753                msg = str(localPort) +" "+ field +" "+ operator+" "+ value
754                sent_rule(widget, msg)
755                widget.set_editable(False)
756                coll_window.destroy()
757    else:
758        check_dialog( widget, None, "Internal program error ;) !")
759
760def prefix_entry(maxLen, tooltip_Text, parent, lab, id):
761    global mainTT
762    vbox = gtk.VBox(False, 0)
763    parent.pack_start(vbox, False, True, 0)
764    label = gtk.Label(lab)
765    label.set_alignment(0, 0.5)
766    vbox.pack_start(label, False, True, 0)
767    entry = gtk.Entry(max=maxLen)
768    entry.set_max_length(maxLen)
769    entry.connect("activate", prefix_callback, entry, id)
770    entry.set_text("")
771    entry.select_region(0, len(entry.get_text()))
772    if tooltip_Text!=None:
773        mainTT.set_tip(entry, tooltip_Text)
774    vbox.pack_start(entry, False, True, 0)
775    entry.show()
776
777def user_quit(widget, data=None):
778    global myInputControlThread, localPort, record_file
779    if myInputControlThread != None:
780        myInputControlThread.kill(1)
781        myInputControlThread = None
782        s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
783        s.connect(('localhost', localPort))
784        s.send("FIN")
785        s.close()
786        if (record_file!=None):
787            record_file.close()
788        print "\n\n\n\n\n Input Stream closed. \n\n\n\n\n By :-) \n\n\n\n"
789    gtk.main_quit()
790
791def enter_callback( widget, entry, id):
792    global firstParseValue
793    entry_text = entry.get_text()
794    if id == 0:
795        if (entry_text==""):
796            checkUpValues[0] = 0
797        else:
798            if (re.search('\D',entry_text)) or (int(entry_text) >= 99) or (int(entry_text) <= 0):
799                check_dialog( widget, None, "Wrong value !")
800            else:
801                firstParseValue[0] = int(entry_text)
802                checkUpValues[0] = 1
803    if id == 1:
804        if ( entry_text==""):
805            checkUpValues[8] = 0
806            checkUpValues[12] = 0
807        else:
808            if (re.match('^\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}(/\d{1,2})?$',entry_text)):
809                splitter = entry_text.split('/')
810                adip = struct.unpack('>L',socket.inet_aton(splitter[0]))[0]
811                if len(splitter)>1:
812                    bits = int(splitter[1])
813                    if bits <= 32 and bits > 0:
814                        mask[0] = 0xffffffffL << (32 - bits)
815                    else:
816                        check_dialog( widget, None, "Wrong value in mask!")
817                else:
818                    mask[0] = 0xffffffffL
819                firstParseValue[8] = adip & mask[0]
820                checkUpValues[8] = 1
821                firstParseValue[12] = adip & mask[0]
822                checkUpValues[12] = 1
823            else:
824                check_dialog( widget, None, "Wrong value !")
825    if id == 2:
826        if ( entry_text==""):
827            checkUpValues[2] = 0
828        else:
829            check_dialog( widget, None, "NOT YET IMPLEMEMTED")
830            ## pb with IPv6 and getaddrinfo() ??
831##            splitter = entry_text.split('/')
832##            try:
833##                socket.gethostbyaddr(splitter[0])
834##            except socket.gaierror:
835##                check_dialog( widget, None, "Wrong value !")
836##            except socket.herror: ## can't resolv
837##                pass
838##            except socket.error:
839##                pass
840    if id == 3:
841        if ( entry_text==""):
842            checkUpValues[15] = 0
843        else:
844            if (re.match('^\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}(/\d{1,2})?$',entry_text)):
845                splitter = entry_text.split('/')
846                adip = struct.unpack('>L',socket.inet_aton(splitter[0]))[0]
847                if len(splitter)>1:
848                    bits = int(splitter[1])
849                    if bits <= 32 and bits > 0:
850                        mask[2] = 0xffffffffL << (32 - bits)
851                    else:
852                        check_dialog( widget, None, "Wrong value in mask!")
853                else:
854                    mask[2] = 0xffffffffL
855                firstParseValue[15] = adip & mask[2]
856                checkUpValues[15] = 1
857            else:
858                check_dialog( widget, None, "Wrong value !")
859    if id == 4: ## In|Out
860        if ( entry_text==""):
861            checkUpValues[10] = 0
862            checkUpValues[14] = 0
863        else:
864            if (re.search('\D',entry_text)) or (int(entry_text) > 65535) or (int(entry_text) < 0):
865                check_dialog( widget, None, "Wrong value !")
866            else:
867                firstParseValue[10] = int(entry_text)
868                checkUpValues[10] = 1
869                firstParseValue[14] = int(entry_text)
870                checkUpValues[14] = 1
871    if id == 6: ## bytes
872        if ( entry_text==""):
873            checkUpValues[1] = 0
874        else:
875            size_list = string.split(entry_text)
876            if (len(size_list)!=2)or((re.search('\D',size_list[0]))or(int(size_list[0])>999999999)or(int(size_list[0])<=0) or (re.search('\D',size_list[1]))or(int(size_list[1])>999999999)or(int(size_list[1])<=0)):
877                check_dialog( widget, None, "Wrong value !")
878            else:
879                firstParseValue[1] = int(size_list[0])
880                if ( int(size_list[1]) == 999999999 ):
881                    firstParseValue[1] = 100000000000
882                else:
883                    firstParseValue[1] = int(size_list[1])
884                checkUpValues[1] = 1
885    if id == 7: ## pkts
886        if ( entry_text==""):
887            checkUpValues[2] = 0
888        else:
889            if (re.search('\D',entry_text)) or (int(entry_text) > 9999) or (int(entry_text) <= 0):
890                check_dialog( widget, None, "Wrong value !")
891            else:
892                firstParseValue[2] = int(entry_text)
893                checkUpValues[2] = 1
894    if id == 8: ## time no concordance with nfv9 field number
895        if ( entry_text==""):
896            checkUpValues[0] = 0
897        else:
898            if (re.search('\D',entry_text)) or (int(entry_text) > 99) or (int(entry_text) < 0):
899                check_dialog( widget, None, "Wrong value !")
900            else:
901                firstParseValue[0] = int(entry_text)
902                checkUpValues[0] = 1
903    if id == 9: ## port range
904        if ( entry_text==""):
905            checkUpValues[7] = 0
906        else:
907            port_list = string.split(entry_text)
908            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)):
909                check_dialog( widget, None, "Wrong value !")
910            else:
911                firstParseValue[7] = int(port_list[0])
912                firstParseValue[11] = int(port_list[1])
913                checkUpValues[7] = 1
914    if id == 11: ## tcp flags
915        if ( entry_text==""):
916            checkUpValues[6] = 0
917        else:
918            if (re.search('\D',entry_text)) or (int(entry_text) > 255) or (int(entry_text) <= 0):
919                check_dialog( widget, None, "Wrong value !")
920            else:
921                firstParseValue[6] = int(entry_text)
922                checkUpValues[6] = 1
923    if id == 12: ## prot
924        if ( entry_text==""):
925            checkUpValues[4] = 0
926        else:
927            if (re.search('\D',entry_text)) or (int(entry_text) > 255) or (int(entry_text) <= 0):
928                check_dialog( widget, None, "Wrong value !")
929            else:
930                firstParseValue[4] = int(entry_text)
931                checkUpValues[4] = 1
932    if id == 13: ## tos
933        if ( entry_text==""):
934            checkUpValues[5] = 0
935        else:
936            if (re.search('\D',entry_text)) or (int(entry_text) > 255) or (int(entry_text) < 0):
937                check_dialog( widget, None, "Wrong value !")
938            else:
939                firstParseValue[5] = int(entry_text)
940                checkUpValues[5] = 1
941    if id == 14: ## src AS
942        if ( entry_text==""):
943            checkUpValues[16] = 0
944        else:
945            if (re.search('\D',entry_text)) or (int(entry_text) > 65535) or (int(entry_text) < 0):
946                check_dialog( widget, None, "Wrong value !")
947            else:
948                firstParseValue[16] = int(entry_text)
949                checkUpValues[16] = 1
950    if id == 15: ## dst AS
951        if ( entry_text==""):
952            checkUpValues[17] = 0
953        else:
954            if (re.search('\D',entry_text)) or (int(entry_text) > 65535) or (int(entry_text) < 0):
955                check_dialog( widget, None, "Wrong value !")
956            else:
957                firstParseValue[17] = int(entry_text)
958                checkUpValues[17] = 1
959    if id == 18: ## from router
960        if ( entry_text==""):
961            checkUpValues[69] = 0
962        else:
963            if (re.match('^\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}?$',entry_text)):
964                adip = struct.unpack('>L',socket.inet_aton(entry_text))[0]
965                firstParseValue[69] = adip
966                checkUpValues[69] = 1
967            else:
968                check_dialog( widget, None, "Wrong value !")
969   
970if __name__ == "__main__":
971    global window, mainTT, comboRouter, os_type, printShit
972   
973    gtk.threads_init()
974   
975    if (os.name =="nt"):
976        os_type = 1
977    elif (os.name =="posix"):
978        os_type = 0
979    else:
980        os_type = 3
981
982    print "---------------------------------------------------------------"
983    print "In this version 0.0.2, you can apply rules on the collector based on the following fields:"
984    print "IPV4_SRC_ADDR: 8,"
985    print "IPV4_DST_ADDR: 12,"
986    print "IPV4_NEXT_HOP: 15,"
987    print "BGP_IPV4_NEXT_HOP: 18,"
988    print "IPV6_SRC_ADDR: 27,"
989    print "IPV6_DST_ADDR: 28,"
990    print "IPV6_NEXT_HOP: 62,"
991    print "BPG_IPV6_NEXT_HOP: 63,"
992    print "IP_PROTOCOL_VERSION: 60"
993    print "----------------------------------------------------------------"
994   
995    mainTT = gtk.Tooltips()
996    window = gtk.Window(gtk.WINDOW_TOPLEVEL)   
997    window.connect("destroy", gtk.main_quit, "WM destroy")
998    window.set_title("RenetcolGUI")
999    window.set_size_request(850, 550)
1000   
1001    main_vbox = gtk.VBox(False, 1)
1002    main_vbox.set_border_width(1)
1003    window.add(main_vbox)
1004
1005##  menu
1006    file_menu = gtk.Menu()
1007    receive_item = gtk.MenuItem("Receive Stream")
1008    stop_item = gtk.MenuItem("Stop Stream")
1009    quit_item  = gtk.MenuItem("Quit")
1010    file_menu.append(receive_item)
1011    file_menu.append(stop_item)
1012    file_menu.append(quit_item)
1013    receive_item.connect("activate", reception, None)
1014    stop_item.connect("activate", close_recept, None)
1015    quit_item.connect("activate", user_quit, None)
1016    receive_item.show()
1017    stop_item.show()
1018    quit_item.show()
1019
1020    help_menu = gtk.Menu()
1021    about_item = gtk.MenuItem("About")
1022    help_menu.append(about_item)
1023    about_item.connect("activate", about_dialog, None)
1024    about_item.show()
1025
1026    menu_bar = gtk.MenuBar()
1027    main_vbox.pack_start(menu_bar, False, False, 0)
1028    menu_bar.show()
1029    file_item = gtk.MenuItem("File")
1030    file_item.show()
1031    help_item = gtk.MenuItem("Help")
1032    help_item.show()
1033
1034    file_item.set_submenu(file_menu)
1035    menu_bar.append(file_item)
1036    menu_bar.append(file_menu)
1037    file_item.set_right_justified(0)
1038
1039    help_item.set_submenu(help_menu)
1040    menu_bar.append(help_item)
1041    menu_bar.append(help_menu)
1042    help_item.set_right_justified(1)
1043
1044##  first frame
1045    frame = gtk.Frame("Input Stream Control")
1046    main_vbox.pack_start(frame, False, True, 0)
1047    hbox = gtk.HBox(False, 0)
1048    hbox.set_border_width(5)
1049    frame.add(hbox)
1050
1051    ## freeze button
1052    buttonfreeze = gtk.ToggleButton("Freeze")
1053    buttonfreeze.connect("toggled", callback_freeze, None)
1054    mainTT.set_tip(buttonfreeze, "Stop/Start the output in the window.")
1055    hbox.pack_start(buttonfreeze, False, False, 2)
1056    buttonfreeze.show()
1057    ## record button
1058    buttonrecord = gtk.ToggleButton("Record")
1059    buttonrecord.connect("toggled", callback_record, None)
1060    mainTT.set_tip(buttonrecord, "Stop/Start the record in a text file.")
1061    hbox.pack_start(buttonrecord, False, False, 2)
1062    buttonrecord.show()
1063    ## control collector button
1064    buttoncollector = gtk.ToggleButton("Collector Control")
1065    buttoncollector.connect("toggled", callback_collector, None)
1066    mainTT.set_tip(buttoncollector, "Change the rule which is applied on the collector. You must have activated the input stream before.")
1067    hbox.pack_start(buttoncollector, False, False, 2)
1068    buttoncollector.show()
1069
1070    ipv4_vbox = gtk.VBox(False, 1)
1071    ipv4_vbox.set_border_width(1)
1072    main_vbox.add(ipv4_vbox)
1073##  second frame
1074    frame = gtk.Frame("Select printed fields")
1075    ipv4_vbox.pack_start(frame, False, False, 0)
1076   
1077    vbox = gtk.VBox(False, 0)
1078    vbox.set_border_width(5)
1079    frame.add(vbox)
1080    bbox = gtk.HButtonBox()
1081    vbox.pack_start(bbox, False, True, 5)
1082   
1083    create_button("Src \n IP", 1, callback_fields_printed, 8, bbox, "Source IP")
1084    create_button("Dst\n IP", 1, callback_fields_printed, 12, bbox, "Destination IP")
1085    create_button("Flw\n IP", 0, callback_fields_printed, 15, bbox, "Following IP")
1086    create_button("Index\n In", 0, callback_fields_printed, 10, bbox, "SNMP Index Input")
1087    create_button("Index\n Out", 0, callback_fields_printed, 14, bbox, "SNMP Index Output")
1088    create_button("Size (P)", 1, callback_fields_printed, 2, bbox, "Packets number")
1089    create_button("Size (B)", 1, callback_fields_printed, 1, bbox, "Size in Bytes")
1090    create_button("Time", 0, callback_fields_printed, 0, bbox, "Age of flow")
1091
1092    b2box = gtk.HButtonBox()
1093    vbox.pack_start(b2box, False, True, 5)
1094    create_button("Src\n Port", 1, callback_fields_printed, 7, b2box, "TCP/UDP source port")
1095    create_button("Dst\n Port", 1, callback_fields_printed, 11, b2box, "TCP/UDP destination port")
1096    create_button("TCP\n Flags", 0, callback_fields_printed, 6, b2box, "TCP Flags")
1097    create_button("Prot", 1, callback_fields_printed, 4, b2box, "Protocol")
1098    create_button("Tos", 0, callback_fields_printed, 5, b2box, "Type of service")
1099    create_button("Src\nAs", 0, callback_fields_printed, 16, b2box, "Source AS")
1100    create_button("Dst\nAs", 0, callback_fields_printed, 17, b2box, "Destination AS")
1101    create_button("Src\nMask", 0, callback_fields_printed, 9, b2box, "Source Mask")
1102    create_button("Dst\nMask", 0, callback_fields_printed, 13, b2box, "Destination Mask")
1103
1104    b3box = gtk.HButtonBox()
1105    vbox.pack_start(b3box, False, True, 5)
1106    create_button("Router", 0, callback_fields_printed, 69, b3box, "Router address")
1107
1108##  third frame
1109    frame = gtk.Frame("Parsing values")
1110    ipv4_vbox.pack_start(frame, False, False, 5)
1111    hbox = gtk.VBox(False, 0)
1112    hbox.set_border_width(5)
1113    frame.add(hbox)
1114    pbbox = gtk.HButtonBox()
1115    hbox.pack_start(pbbox, False, True, 5)
1116##  here entry button
1117    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)
1118    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)
1119    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)
1120    dbbox = gtk.HButtonBox()
1121    hbox.pack_start(dbbox, False, True, 5)
1122    create_entry(5, "Required Format : a number less than 65535", dbbox, "SNMP Index In|Out :", 4)
1123    create_entry(4, "Required Format : a number less than 9999", dbbox, "Packet NB :", 7)
1124    create_entry(17, "Required Format : a range, each number less than 999999999 and space as separator: \"48 54\"", dbbox, "Bytes :", 6)
1125    tbbox = gtk.HButtonBox()
1126    hbox.pack_start(tbbox, False, True, 5)
1127    create_entry(2, "Required Format : a number less than 99", tbbox, "Time :", 8)
1128    create_entry(11, "Required Format : a range, each number less than 65535 and space as separator: \"6881 6889\"", tbbox, "Port :", 9)
1129    #create_entry(5, "Required Format : a number less than 65535", tbbox, "Dst port :", 10)
1130    create_entry(3, "Required Format : a number less than 255", tbbox, "TCP Flag :", 11)
1131    fbbox = gtk.HButtonBox()
1132    hbox.pack_start(fbbox, False, True, 5)
1133    create_entry(5, "Required Format : a number less than 65535", fbbox, "Protocol :", 12)
1134    create_entry(3, "Required Format : a number less than 255", fbbox, "ToS :", 13)
1135    create_entry(5, "Required Format : a number less than 65535", fbbox, "Src AS :", 14)
1136    create_entry(5, "Required Format : a number less than 65535", fbbox, "Dst AS :", 15)
1137    kbbox = gtk.HButtonBox()
1138    hbox.pack_start(kbbox, False, True, 5)
1139    create_entry(18, "Required Format : \n an IP address like x.x.x.x", kbbox, "The flows from this Router :", 18)
1140
1141##  Show
1142    vbox.show()
1143    main_vbox.show()
1144    window.show_all()
1145    mainTT.enable()
1146
1147##  gtk main loop
1148    gtk.threads_enter()
1149    gtk.main()
1150    gtk.threads_leave()
Note: See TracBrowser for help on using the browser.