Premier test d'implémentation de la BDD via le decoder

src/decoderBrut.py

@@ -5,6 +5,7 @@ import scapy.all as scapy
 import scapy.contrib.modbus as mb
 from triPacket import triPacket
 
+
 def decode(pkt):
     prt=0
     addr,value = "",""
@@ -94,11 +95,26 @@ def decode(pkt):
         
         if bigL!=[]:
             print("attention")
-            triPacket(bigL)
+            triPacket(bigL,connec)
 
 
         miniL = [0,0,0,0]
         bigL=[]
 miniL = [0,0,0,0]
+
+
+print("In order for data sniffed to be stored inside the database, please register the following :")
+db_host = input('host of the database server : ')
+if not db_host:
+    db_host = 'londres'
+db_name = input('name of the database : ')
+if not db_name:
+    db_name = 'dblodufour1'
+db_user = input('login of the user : ')
+if not db_user:
+    db_user = 'lodufour1'
+db_password = getpass('user password : ')
+connec=[db_host,db_name,db_user,db_password]
+
 scapy.sniff(iface="lo", prn=decode)
 

src/ecritureBDD.py

@@ -0,0 +1,20 @@
+import psycopg2 as psy
+import pandas as pd
+import getpass
+
+def ecritureBDD(addresse,typeMem,valeur,connec):
+	try:
+		co = psy.connect(host=connec[0],
+			database=connec[1],
+			user=connec[3],
+			password=connec[4])
+		cur = co.cursor()
+		cur.execute("INSERT INTO Status VALUES (%s,%s,%s );",
+			(addresse,typeMem,valeur))
+		co.commit()
+		cur.close()
+	except(Exception,psy.DatabaseError) as error:
+		print(error)
+	finally:
+		if co is not None:
+			co.close()

src/triPacket.py

@@ -1,4 +1,6 @@
-def triPacket(lPkt):
+from ecritureBDD import ecritureBDD
+
+def triPacket(lPkt,connec):
     if type(lPkt)!=list:
         print('pas liste')
         return
@@ -8,8 +10,9 @@ def triPacket(lPkt):
     if len(lPkt)==1:
         triPacket(lPkt[0])
         return
-    if len(lPkt)==3 and type(lPkt[0])==str:        
-        print(lPkt[0],int(lPkt[1]),int(lPkt[2]))
+    if len(lPkt)==3 and type(lPkt[0])==str:
+        if (lPkt[0]=='c' or lPkt[0]=='r') and lPkt[1]>=0 and lPkt[2]>=0:
+            print(lPkt[0],int(lPkt[1]),int(lPkt[2]))
         return
     else:
         for i in lPkt: