129 lines
4.2 KiB
Python
129 lines
4.2 KiB
Python
#!/usr/bin/env python3
|
|
|
|
import sqlite3
|
|
from sqlite3 import Error
|
|
import tkinter as tk
|
|
from tkinter import ttk
|
|
from scapy.all import *
|
|
from datetime import datetime
|
|
|
|
def create_connection(db_file):
|
|
conn = None
|
|
try:
|
|
conn = sqlite3.connect(db_file)
|
|
except Error as e:
|
|
print(e)
|
|
return conn
|
|
|
|
def select_all_pcaprecords(conn):
|
|
cur = conn.cursor()
|
|
cur.execute("SELECT * FROM pcap_records")
|
|
|
|
rows = cur.fetchall()
|
|
return rows
|
|
|
|
def open_popup(text_display):
|
|
top = tk.Toplevel(mainwindow)
|
|
top.geometry("550x250")
|
|
top.title("Analysis Results!")
|
|
popup_text = tk.Text(top, height=10, width=65,font=('Consolas',10))
|
|
popup_text.grid(row=0,column=0,padx=5,pady=5,sticky=tk.W)
|
|
popup_text.insert(tk.END,text_display)
|
|
|
|
def callback(event):
|
|
filepath = "./outputfiles/"
|
|
# get the index of the mouse click
|
|
index = event.widget.index("@%s,%s" % (event.x, event.y))
|
|
|
|
# get the indices of all "adj" tags
|
|
tag_indices = list(event.widget.tag_ranges('tag'))
|
|
|
|
# iterate them pairwise (start and end index)
|
|
for start, end in zip(tag_indices[0::2], tag_indices[1::2]):
|
|
# check if the tag matches the mouse click index
|
|
if event.widget.compare(start, '<=', index) and event.widget.compare(index, '<', end):
|
|
# return string between tag start and end
|
|
filename = filepath + event.widget.get(start, end)
|
|
print(filename)
|
|
file1 = open(filename,'r')
|
|
lines = file1.readlines()
|
|
d = ""
|
|
for line in lines:
|
|
d += line.strip() + "\n"
|
|
|
|
open_popup(d)
|
|
|
|
def display_to_text(in_data):
|
|
displaytext.tag_config("tag",foreground="blue")
|
|
displaytext.tag_bind("tag","<Button-1>", callback)
|
|
displaytext.insert(tk.END, 'ID' + "\t" + "PCAP Filename" + " " + "Date" + "\t\t\t\t" + "Time" + "\t" + "Output File\n")
|
|
displaytext.insert(tk.END, '==' + "\t" + "==========================" + "\t" + "========" + "\t" + "====" + "\t" + "===========\n")
|
|
for row in in_data:
|
|
displaytext.insert(tk.END, str(row[0]) + "\t" + row[1] + " \t\t\t" + row[2] + "\t" + row[3] + " " + row[4] + "\t")
|
|
displaytext.insert(tk.END,row[4],"tag")
|
|
displaytext.insert(tk.END,"\n")
|
|
|
|
def get_current_date():
|
|
now = datetime.now()
|
|
return now.strftime('%Y%m%d')
|
|
|
|
def get_current_time():
|
|
now = datetime.now()
|
|
return now.strftime('%H%M%S')
|
|
|
|
|
|
def analysispcap():
|
|
pcapFile = getfiletextbox.get()
|
|
pcap = rdpcap(pcapFile)
|
|
numofpackets = len(pcap)
|
|
resultstextbox.insert(tk.END,numofpackets)
|
|
|
|
def saveresult():
|
|
pcapFile = getfiletextbox.get()
|
|
analysis_date = get_current_date()
|
|
analysis_time = get_current_time()
|
|
analysis_output = resultstextbox.get("1.0","end-1c")
|
|
output_filename = analysis_date + analysis_time + '.txt'
|
|
print(pcapFile)
|
|
print(analysis_date)
|
|
print(analysis_time)
|
|
print(analysis_output)
|
|
print(output_filename)
|
|
|
|
# ====================== Main Start Here =====================================================
|
|
|
|
database = r"saspdemo.db"
|
|
conn = create_connection(database)
|
|
rows = select_all_pcaprecords(conn)
|
|
data = []
|
|
|
|
for row in rows:
|
|
data.append([row[0], row[1], row[2], row[3], row[4]])
|
|
|
|
mainwindow = tk.Tk()
|
|
mainwindow.title("SASP Part B Assignment AY2223 - PCAP Analysis By Chan Tai Man")
|
|
tabControl = ttk.Notebook(mainwindow)
|
|
|
|
tab1 = ttk.Frame(tabControl)
|
|
tab2 = ttk.Frame(tabControl)
|
|
|
|
tabControl.add(tab1, text ='Network Traffic Analysis')
|
|
tabControl.add(tab2, text ='History')
|
|
tabControl.pack(expand = 1, fill ="both")
|
|
|
|
# =================== tab 1 GUI Layout ========================================================
|
|
|
|
getfilelabel = tk.Label(tab1,text='Start to work on your assignment',fg='red',font=('Consolas',12))
|
|
getfilelabel.grid(row=0,column=0,padx=5,sticky=tk.W)
|
|
|
|
|
|
# =================== tab 2 GUI Layout ========================================================
|
|
|
|
displaybtn = tk.Button(tab2, text="DISPLAY", fg='blue', width=20, command=lambda:display_to_text(data))
|
|
displaybtn.grid(row=0,column=0, padx=5,pady=10,sticky=tk.W)
|
|
|
|
displaytext = tk.Text(tab2,height=10, width=100,font=('Consolas',10))
|
|
displaytext.grid(row=2,column=0,padx=5,pady=5,sticky=tk.W)
|
|
|
|
mainwindow.mainloop()
|