|
|
|
|
@ -0,0 +1,954 @@
|
|
|
|
|
import ipaddress
|
|
|
|
|
import sys
|
|
|
|
|
import threading
|
|
|
|
|
import ttkbootstrap as tk
|
|
|
|
|
from ttkbootstrap import *
|
|
|
|
|
from ttkbootstrap import ttk
|
|
|
|
|
from tkinter import messagebox
|
|
|
|
|
import re
|
|
|
|
|
from PIL import ImageTk, Image
|
|
|
|
|
import platform
|
|
|
|
|
|
|
|
|
|
from SimObjs import SimPacket, SimHost, AllSimConnect, SimRouter, SimSwitch, SimHub, SimBase
|
|
|
|
|
from dbUtil import search, execute_sql, delete_obj, truncate_db
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def validate_ip_address(ip_address):
|
|
|
|
|
"""
|
|
|
|
|
匹配ip地址格式是否规范
|
|
|
|
|
:param ip_address: IP地址
|
|
|
|
|
:return: Boolean
|
|
|
|
|
"""
|
|
|
|
|
# 定义IP地址的正则表达式模式
|
|
|
|
|
pattern_with_subnet = r'^(\d{1,3})\.(\d{1,3})\.(\d{1,3})\.(\d{1,3})/(\d{1,2})$'
|
|
|
|
|
pattern_without_subnet = r'^(\d{1,3})\.(\d{1,3})\.(\d{1,3})\.(\d{1,3})$'
|
|
|
|
|
# 使用re模块进行匹配
|
|
|
|
|
match_with_subnet = re.match(pattern_with_subnet, ip_address)
|
|
|
|
|
match_without_subnet = re.match(pattern_without_subnet, ip_address)
|
|
|
|
|
if match_with_subnet:
|
|
|
|
|
# 带有子网掩码的IP地址
|
|
|
|
|
# 检查每个组件的取值范围是否在 0-255 之间
|
|
|
|
|
for group in match_with_subnet.groups()[:4]:
|
|
|
|
|
if not (0 <= int(group) <= 255):
|
|
|
|
|
return False
|
|
|
|
|
# 检查子网掩码的取值范围是否在 0-32 之间
|
|
|
|
|
subnet_mask = int(match_with_subnet.groups()[4])
|
|
|
|
|
if not (0 <= subnet_mask <= 32):
|
|
|
|
|
return False
|
|
|
|
|
return True
|
|
|
|
|
elif match_without_subnet:
|
|
|
|
|
# 不带子网掩码的IP地址
|
|
|
|
|
# 检查每个组件的取值范围是否在 0-255 之间
|
|
|
|
|
for group in match_without_subnet.groups():
|
|
|
|
|
if not (0 <= int(group) <= 255):
|
|
|
|
|
return False
|
|
|
|
|
return True
|
|
|
|
|
else:
|
|
|
|
|
# IP地址格式不正确
|
|
|
|
|
return False
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class RouterConfigWindow(tk.Toplevel):
|
|
|
|
|
def __init__(self, parent, router_obj):
|
|
|
|
|
super().__init__(parent)
|
|
|
|
|
self.geometry("435x433+350+200")
|
|
|
|
|
self.title(f"{router_obj.ObjLabel}路由表配置")
|
|
|
|
|
self.router_obj = router_obj
|
|
|
|
|
self.interface_entries = []
|
|
|
|
|
self.router_table = {}
|
|
|
|
|
self.create_interface_inputs()
|
|
|
|
|
self.create_router_table()
|
|
|
|
|
|
|
|
|
|
def create_interface_inputs(self):
|
|
|
|
|
label_text = ["接口1", "接口2", "接口3", "接口4"]
|
|
|
|
|
for i in range(4):
|
|
|
|
|
label = tk.Label(self, text=label_text[i])
|
|
|
|
|
label.grid(row=i, column=0, padx=10, pady=5, sticky="w")
|
|
|
|
|
entry = tk.Entry(self, width=20)
|
|
|
|
|
entry.grid(row=i, column=1, padx=10, pady=5, sticky="w")
|
|
|
|
|
self.interface_entries.append(entry)
|
|
|
|
|
button = tk.Button(self, text="添加", command=lambda index=i: self.add_router_entry(index))
|
|
|
|
|
button.grid(row=i, column=2, padx=10, pady=5)
|
|
|
|
|
lab = LabelFrame(self, text="示例")
|
|
|
|
|
lab.grid(row=4, column=0, columnspan=3, sticky=W, padx=20)
|
|
|
|
|
Label(lab, text="10.1.2.0/24 或者 10.1.2.12" if self.router_obj.ObjType == 2 else "MAC11").pack()
|
|
|
|
|
|
|
|
|
|
def create_router_table(self):
|
|
|
|
|
def on_right_click(event):
|
|
|
|
|
row = self.router_treeview.identify_row(event.y) # 获取鼠标位置的行索引
|
|
|
|
|
if row:
|
|
|
|
|
self.router_treeview.selection_set(row) # 选中该行
|
|
|
|
|
delete_menu.post(event.x_root, event.y_root) # 在鼠标位置弹出删除菜单
|
|
|
|
|
|
|
|
|
|
def delete_row():
|
|
|
|
|
selected_items = self.router_treeview.selection() # 获取选中的行
|
|
|
|
|
for item in selected_items:
|
|
|
|
|
ifs, network = int(self.router_treeview.item(item)["values"][0][-1:]), self.router_treeview.item(item)["values"][1]
|
|
|
|
|
self.router_obj.delete_config(ifs, network)
|
|
|
|
|
self.router_treeview.delete(item)
|
|
|
|
|
self.router_table_frame = tk.Frame(self)
|
|
|
|
|
self.router_table_frame.grid(row=5, column=0, columnspan=3, padx=10, pady=5)
|
|
|
|
|
self.router_treeview = ttk.Treeview(self.router_table_frame, columns=("Interface", "Route"), show="headings")
|
|
|
|
|
self.router_treeview.heading("Interface", text="接口")
|
|
|
|
|
self.router_treeview.heading("Route", text="网段")
|
|
|
|
|
self.router_treeview.pack(side="left", fill="both")
|
|
|
|
|
scrollbar = ttk.Scrollbar(self.router_table_frame, orient="vertical", command=self.router_treeview.yview)
|
|
|
|
|
scrollbar.pack(side="right", fill="y")
|
|
|
|
|
self.router_treeview.configure(yscrollcommand=scrollbar.set)
|
|
|
|
|
self.router_table = self.router_obj.router_table
|
|
|
|
|
self.router_treeview.bind("<Button-3>", on_right_click)
|
|
|
|
|
# 创建删除菜单
|
|
|
|
|
delete_menu = tk.Menu(root, tearoff=False)
|
|
|
|
|
delete_menu.add_command(label="删除", command=delete_row)
|
|
|
|
|
self.update_router_table()
|
|
|
|
|
|
|
|
|
|
def add_router_entry(self, index):
|
|
|
|
|
entry_text = self.interface_entries[index].get()
|
|
|
|
|
try:
|
|
|
|
|
ipaddress.ip_network(entry_text)
|
|
|
|
|
if isinstance(self.router_obj, SimRouter):
|
|
|
|
|
if not validate_ip_address(entry_text):
|
|
|
|
|
messagebox.showerror("注意", message="添加的网段信息格式不合格")
|
|
|
|
|
self.interface_entries[index].delete(0, tk.END)
|
|
|
|
|
self.focus_set()
|
|
|
|
|
return
|
|
|
|
|
if entry_text:
|
|
|
|
|
if index + 1 in self.router_table:
|
|
|
|
|
self.router_table[index + 1].append(entry_text)
|
|
|
|
|
else:
|
|
|
|
|
self.router_table[index + 1] = [entry_text]
|
|
|
|
|
self.interface_entries[index].delete(0, tk.END)
|
|
|
|
|
self.router_obj.add_config(entry_text, index + 1)
|
|
|
|
|
self.update_router_table()
|
|
|
|
|
except:
|
|
|
|
|
messagebox.showerror("注意", message="网段格式错误！网段示例如下：\n10.1.2.0/24\n10.1.2.12")
|
|
|
|
|
return
|
|
|
|
|
|
|
|
|
|
def update_router_table(self):
|
|
|
|
|
self.router_treeview.delete(*self.router_treeview.get_children())
|
|
|
|
|
for i, entrys in self.router_table.items():
|
|
|
|
|
for entry in entrys:
|
|
|
|
|
self.router_treeview.insert("", "end", values=(f"接口{i}", entry))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class SwitchConfigWindow(RouterConfigWindow):
|
|
|
|
|
def __init__(self, parent, router_obj):
|
|
|
|
|
super().__init__(parent, router_obj)
|
|
|
|
|
self.geometry("435x433+350+200")
|
|
|
|
|
self.title(f"{router_obj.ObjLabel}交换表配置")
|
|
|
|
|
self.router_obj = router_obj
|
|
|
|
|
self.interface_entries = []
|
|
|
|
|
self.router_table = {}
|
|
|
|
|
self.create_interface_inputs()
|
|
|
|
|
self.create_router_table()
|
|
|
|
|
|
|
|
|
|
def create_router_table(self):
|
|
|
|
|
def on_right_click(event):
|
|
|
|
|
row = self.router_treeview.identify_row(event.y) # 获取鼠标位置的行索引
|
|
|
|
|
if row:
|
|
|
|
|
self.router_treeview.selection_set(row) # 选中该行
|
|
|
|
|
delete_menu.post(event.x_root, event.y_root) # 在鼠标位置弹出删除菜单
|
|
|
|
|
|
|
|
|
|
def delete_row():
|
|
|
|
|
selected_items = self.router_treeview.selection() # 获取选中的行
|
|
|
|
|
for item in selected_items:
|
|
|
|
|
ifs, network = int(self.router_treeview.item(item)["values"][0][-1:]), self.router_treeview.item(item)["values"][1]
|
|
|
|
|
self.router_obj.delete_config(ifs, network)
|
|
|
|
|
self.router_treeview.delete(item)
|
|
|
|
|
|
|
|
|
|
self.router_table_frame = tk.Frame(self)
|
|
|
|
|
self.router_table_frame.grid(row=5, column=0, columnspan=3, padx=10, pady=5)
|
|
|
|
|
self.router_treeview = ttk.Treeview(self.router_table_frame, columns=("Interface", "Route"), show="headings")
|
|
|
|
|
self.router_treeview.heading("Interface", text="接口")
|
|
|
|
|
self.router_treeview.heading("Route", text="mac")
|
|
|
|
|
self.router_treeview.pack(side="left", fill="both")
|
|
|
|
|
scrollbar = ttk.Scrollbar(self.router_table_frame, orient="vertical", command=self.router_treeview.yview)
|
|
|
|
|
scrollbar.pack(side="right", fill="y")
|
|
|
|
|
self.router_treeview.configure(yscrollcommand=scrollbar.set)
|
|
|
|
|
self.router_table = self.router_obj.mac_table
|
|
|
|
|
self.router_treeview.bind("<Button-3>", on_right_click)
|
|
|
|
|
# 创建删除菜单
|
|
|
|
|
delete_menu = tk.Menu(root, tearoff=False)
|
|
|
|
|
delete_menu.add_command(label="删除", command=delete_row)
|
|
|
|
|
self.update_router_table()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def add_router_entry(self, index):
|
|
|
|
|
entry_text = self.interface_entries[index].get()
|
|
|
|
|
if isinstance(self.router_obj, SimRouter):
|
|
|
|
|
if not validate_ip_address(entry_text):
|
|
|
|
|
messagebox.showerror("注意", message="添加的网段信息格式不合格")
|
|
|
|
|
self.interface_entries[index].delete(0, tk.END)
|
|
|
|
|
self.focus_set()
|
|
|
|
|
return
|
|
|
|
|
if entry_text:
|
|
|
|
|
if index + 1 in self.router_table:
|
|
|
|
|
self.router_table[index + 1].append(entry_text)
|
|
|
|
|
else:
|
|
|
|
|
self.router_table[index + 1] = [entry_text]
|
|
|
|
|
self.interface_entries[index].delete(0, tk.END)
|
|
|
|
|
self.router_obj.add_config(entry_text, index + 1)
|
|
|
|
|
self.update_router_table()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class NetWorkAnalog(Canvas):
|
|
|
|
|
def __init__(self, master, **kwargs):
|
|
|
|
|
super().__init__(master, **kwargs)
|
|
|
|
|
self.master = master
|
|
|
|
|
self.router_img = ImageTk.PhotoImage(
|
|
|
|
|
Image.open(sys.path[0] + "/../datas/images/路由器.png").resize((60, 60)))
|
|
|
|
|
self.switch_img = ImageTk.PhotoImage(
|
|
|
|
|
Image.open(sys.path[0] + "/../datas/images/交换机.png").resize((60, 60)))
|
|
|
|
|
self.hub_img = ImageTk.PhotoImage(
|
|
|
|
|
Image.open(sys.path[0] + "/../datas/images/集线器.png").resize((60, 60)))
|
|
|
|
|
self.host_img = ImageTk.PhotoImage(
|
|
|
|
|
Image.open(sys.path[0] + "/../datas/images/主机.png").resize((60, 60)))
|
|
|
|
|
self.chose = self.host_img
|
|
|
|
|
self.AllSimObjs = {}
|
|
|
|
|
self.conns = []
|
|
|
|
|
self.drawLine = True # 画线标志
|
|
|
|
|
self.line_start_obj = None
|
|
|
|
|
self.line_end_obj = None
|
|
|
|
|
self.line_start_ifs = None
|
|
|
|
|
self.line_end_ifs = None
|
|
|
|
|
self.chose_obj = None
|
|
|
|
|
self.show_label_flag = True
|
|
|
|
|
self.show_interface_flag = True
|
|
|
|
|
self.create_widget()
|
|
|
|
|
|
|
|
|
|
def is_chose(self):
|
|
|
|
|
"""
|
|
|
|
|
当被选中时，绘制选中框
|
|
|
|
|
:return:
|
|
|
|
|
"""
|
|
|
|
|
self.delete("rectangle")
|
|
|
|
|
self.create_rectangle(self.chose_obj.ObjX - 25, self.chose_obj.ObjY - 25, self.chose_obj.ObjX + 25,
|
|
|
|
|
self.chose_obj.ObjY + 25, outline="red", tags="rectangle")
|
|
|
|
|
|
|
|
|
|
def tag_bind_event(self):
|
|
|
|
|
"""
|
|
|
|
|
为每个子组件绑定事件
|
|
|
|
|
:return:
|
|
|
|
|
"""
|
|
|
|
|
|
|
|
|
|
def init():
|
|
|
|
|
"""
|
|
|
|
|
初始化方法，将连接对象赋值为初始化值
|
|
|
|
|
:return:
|
|
|
|
|
"""
|
|
|
|
|
self.line_start_obj = None
|
|
|
|
|
self.line_end_obj = None
|
|
|
|
|
self.line_start_ifs = None
|
|
|
|
|
self.line_end_ifs = None
|
|
|
|
|
|
|
|
|
|
def show_menu(event, tag: SimBase):
|
|
|
|
|
"""
|
|
|
|
|
右击组件弹出选择接口框
|
|
|
|
|
:param event:
|
|
|
|
|
:param tag:
|
|
|
|
|
:return:
|
|
|
|
|
"""
|
|
|
|
|
menu = Menu(self, tearoff=0)
|
|
|
|
|
flag = False
|
|
|
|
|
for conn in tag.connections:
|
|
|
|
|
i = tag.connections.index(conn)
|
|
|
|
|
if not isinstance(conn, AllSimConnect):
|
|
|
|
|
menu.add_command(label=f"接口{i + 1}", command=lambda num=i + 1: interface_selected(num, tag))
|
|
|
|
|
flag = True
|
|
|
|
|
if not flag:
|
|
|
|
|
menu.add_command(label="暂无可用接口", state="disabled")
|
|
|
|
|
menu.post(event.x_root, event.y_root)
|
|
|
|
|
|
|
|
|
|
def interface_selected(interface, tag):
|
|
|
|
|
"""
|
|
|
|
|
选择接口回调
|
|
|
|
|
:param interface:
|
|
|
|
|
:return:
|
|
|
|
|
"""
|
|
|
|
|
if self.line_start_ifs is None:
|
|
|
|
|
self.line_start_ifs = interface
|
|
|
|
|
self.line_start_obj = tag
|
|
|
|
|
self.drawLine = True
|
|
|
|
|
self.bind("<Motion>", lambda event: right_motion(event))
|
|
|
|
|
self.bind("<Escape>", quit)
|
|
|
|
|
self.focus_set() # 获取焦点
|
|
|
|
|
else:
|
|
|
|
|
self.line_end_ifs = interface
|
|
|
|
|
self.line_end_obj = tag
|
|
|
|
|
self.delete("Line")
|
|
|
|
|
flag = False
|
|
|
|
|
if self.line_start_obj.ObjID == self.line_end_obj.ObjID:
|
|
|
|
|
messagebox.showerror("注意！", message="不能连接自己")
|
|
|
|
|
return
|
|
|
|
|
conn = AllSimConnect(self, self.line_start_obj, self.line_start_ifs, self.line_end_obj,
|
|
|
|
|
self.line_end_ifs)
|
|
|
|
|
for conn_obj in self.line_start_obj.connections: # 判断两个连接对象是否已经连接过了
|
|
|
|
|
if conn_obj == conn:
|
|
|
|
|
flag = True
|
|
|
|
|
if not flag:
|
|
|
|
|
self.line_start_obj.connections[self.line_start_ifs - 1] = conn
|
|
|
|
|
self.line_end_obj.connections[self.line_end_ifs - 1] = conn
|
|
|
|
|
conn.save()
|
|
|
|
|
self.conns.append(conn)
|
|
|
|
|
init()
|
|
|
|
|
self.unbind("<Motion>")
|
|
|
|
|
else:
|
|
|
|
|
del conn
|
|
|
|
|
self.delete("Line")
|
|
|
|
|
self.unbind("<Motion>")
|
|
|
|
|
init()
|
|
|
|
|
messagebox.showerror("注意！", message="已经连接过了")
|
|
|
|
|
|
|
|
|
|
def right_motion(event):
|
|
|
|
|
"""
|
|
|
|
|
移动鼠标
|
|
|
|
|
:param event:
|
|
|
|
|
:param tag:
|
|
|
|
|
:return:
|
|
|
|
|
"""
|
|
|
|
|
if self.drawLine:
|
|
|
|
|
self.delete("Line")
|
|
|
|
|
x, y = event.x, event.y
|
|
|
|
|
if not (150 < x < 650 and 40 < y < 490):
|
|
|
|
|
if x < 150:
|
|
|
|
|
x = 150
|
|
|
|
|
elif x > 650:
|
|
|
|
|
x = 650
|
|
|
|
|
if y < 40:
|
|
|
|
|
y = 40
|
|
|
|
|
elif y > 490:
|
|
|
|
|
y = 490
|
|
|
|
|
self.create_line(self.line_start_obj.ObjX, self.line_start_obj.ObjY, x + 8, y + 8, fill="#5b9bd5",
|
|
|
|
|
width=1,
|
|
|
|
|
tags="Line")
|
|
|
|
|
return
|
|
|
|
|
self.create_line(self.line_start_obj.ObjX, self.line_start_obj.ObjY, x + 8, y + 8, fill="#5b9bd5",
|
|
|
|
|
width=1, tags="Line")
|
|
|
|
|
|
|
|
|
|
def quit(event):
|
|
|
|
|
self.delete("Line") # 删除刚刚产生的连接线
|
|
|
|
|
self.drawLine = False # 关闭画线标志
|
|
|
|
|
init()
|
|
|
|
|
|
|
|
|
|
for tag_id, tag in self.AllSimObjs.items():
|
|
|
|
|
self.tag_bind(tag.ObjID, "<Button-3>", lambda event, tag=tag: show_menu(event, tag)) # 绑定右击事件
|
|
|
|
|
|
|
|
|
|
def bind_event(self, component, name):
|
|
|
|
|
# todo: 绑定事件
|
|
|
|
|
"""
|
|
|
|
|
绑定事件
|
|
|
|
|
:param component: 组件对象
|
|
|
|
|
:param name: 组件名称
|
|
|
|
|
"""
|
|
|
|
|
def move(event, name):
|
|
|
|
|
"""
|
|
|
|
|
鼠标左键松开事件
|
|
|
|
|
:param event: 事件对象
|
|
|
|
|
"""
|
|
|
|
|
if 170 < event.x < 630 and 60 < event.y < 470: # 在方框内，无变化
|
|
|
|
|
if name == "路由器":
|
|
|
|
|
tag = SimRouter(self, event.x, event.y)
|
|
|
|
|
elif name == "集线器":
|
|
|
|
|
tag = SimHub(self, event.x, event.y)
|
|
|
|
|
elif name == "交换机":
|
|
|
|
|
tag = SimSwitch(self, event.x, event.y)
|
|
|
|
|
else:
|
|
|
|
|
tag = SimHost(self, event.x, event.y)
|
|
|
|
|
self.AllSimObjs[tag.ObjID] = tag
|
|
|
|
|
tag.save()
|
|
|
|
|
self.tag_bind_event()
|
|
|
|
|
else:
|
|
|
|
|
self.delete("L")
|
|
|
|
|
|
|
|
|
|
def motion(event, name):
|
|
|
|
|
"""
|
|
|
|
|
鼠标拖动事件
|
|
|
|
|
:param event: 事件对象
|
|
|
|
|
"""
|
|
|
|
|
if name == "路由器":
|
|
|
|
|
self.chose = self.router_img
|
|
|
|
|
elif name == "集线器":
|
|
|
|
|
self.chose = self.hub_img
|
|
|
|
|
elif name == "交换机":
|
|
|
|
|
self.chose = self.switch_img
|
|
|
|
|
else:
|
|
|
|
|
self.chose = self.host_img
|
|
|
|
|
self.delete("L")
|
|
|
|
|
if 170 < event.x < 630 and 60 < event.y < 470: # 在方框内，无变化
|
|
|
|
|
pass
|
|
|
|
|
else: # 在方框外，显示禁止放置标识
|
|
|
|
|
self.create_oval(event.x - 10, event.y - 45, event.x + 10, event.y - 25, outline="red", width=2,
|
|
|
|
|
tags="L")
|
|
|
|
|
self.create_line(event.x - 7, event.y - 42, event.x + 8, event.y - 27, fill="red", width=2, tags="L")
|
|
|
|
|
self.create_image(event.x - 30, event.y - 30, image=self.chose, anchor="nw", tags="L")
|
|
|
|
|
|
|
|
|
|
self.tag_bind(component, "<ButtonRelease-1>", lambda event: move(event, name))
|
|
|
|
|
self.tag_bind(component, "<B1-Motion>", lambda event: motion(event, name))
|
|
|
|
|
|
|
|
|
|
def reload_data(self):
|
|
|
|
|
# todo: 加载上一次程序运行的数据
|
|
|
|
|
"""
|
|
|
|
|
加载上一次程序运行时的数据
|
|
|
|
|
:return:
|
|
|
|
|
"""
|
|
|
|
|
self.AllSimObjs = {}
|
|
|
|
|
self.conns = []
|
|
|
|
|
sim_obj_sql = "select * from sim_objs"
|
|
|
|
|
sim_obj_data = search(sim_obj_sql)
|
|
|
|
|
for index, row in sim_obj_data.iterrows(): # 初始化组件对象
|
|
|
|
|
sim_type = row["ObjType"]
|
|
|
|
|
ObjX = row["ObjX"]
|
|
|
|
|
ObjY = row["ObjY"]
|
|
|
|
|
ConfigCorrect = row["ConfigCorrect"]
|
|
|
|
|
ObjLable = row["ObjLabel"]
|
|
|
|
|
ObjID = row["ObjID"]
|
|
|
|
|
if sim_type == 1:
|
|
|
|
|
tag = SimHost(self, ObjX, ObjY, ObjID, ConfigCorrect, ObjLable)
|
|
|
|
|
elif sim_type == 2:
|
|
|
|
|
tag = SimRouter(self, ObjX, ObjY, ObjID, ConfigCorrect, ObjLable)
|
|
|
|
|
elif sim_type == 3:
|
|
|
|
|
tag = SimSwitch(self, ObjX, ObjY, ObjID, ConfigCorrect, ObjLable)
|
|
|
|
|
else:
|
|
|
|
|
tag = SimHub(self, ObjX, ObjY, ObjID, ConfigCorrect, ObjLable)
|
|
|
|
|
self.AllSimObjs[tag.ObjID] = tag
|
|
|
|
|
|
|
|
|
|
sim_conn_sql = "select s.conn_id, ConfigCorrect, node_id, node_ifs from sim_conn s join conn_config c on s.conn_id=c.conn_id"
|
|
|
|
|
sim_conn_data = search(sim_conn_sql)
|
|
|
|
|
conn_datas = {}
|
|
|
|
|
for index, conn in sim_conn_data.iterrows():
|
|
|
|
|
if (conn["conn_id"], conn["ConfigCorrect"]) not in conn_datas:
|
|
|
|
|
conn_datas[(conn["conn_id"], conn["ConfigCorrect"])] = [(conn["node_id"], conn["node_ifs"])]
|
|
|
|
|
else:
|
|
|
|
|
conn_datas[(conn["conn_id"], conn["ConfigCorrect"])].append((conn["node_id"], conn["node_ifs"]))
|
|
|
|
|
for key, value in conn_datas.items():
|
|
|
|
|
conn_obj = AllSimConnect(self, self.AllSimObjs[value[0][0]], value[0][1],
|
|
|
|
|
self.AllSimObjs[value[1][0]], value[1][1], key[1])
|
|
|
|
|
self.AllSimObjs[value[0][0]].connections[value[0][1] - 1] = conn_obj # 将连接对象传入组件对象
|
|
|
|
|
self.AllSimObjs[value[1][0]].connections[value[1][1] - 1] = conn_obj
|
|
|
|
|
self.conns.append(conn_obj)
|
|
|
|
|
conn_obj.ConfigCorrect = key[1]
|
|
|
|
|
self.tag_bind_event()
|
|
|
|
|
|
|
|
|
|
def delete_obj(self):
|
|
|
|
|
# todo: 删除对象
|
|
|
|
|
"""
|
|
|
|
|
选中删除对象
|
|
|
|
|
:return:
|
|
|
|
|
"""
|
|
|
|
|
if self.chose_obj is None:
|
|
|
|
|
messagebox.showerror("注意", message="请先选择要删除的对象！")
|
|
|
|
|
return
|
|
|
|
|
ask = messagebox.askquestion(title='确认操作', message='确认删除该对象吗？')
|
|
|
|
|
if ask == "no":
|
|
|
|
|
return
|
|
|
|
|
self.delete(self.chose_obj.ObjID) # 删除图片
|
|
|
|
|
self.delete(self.chose_obj.ObjID + "text") # 删除标签
|
|
|
|
|
for conn in self.chose_obj.connections: # 删除该对象的所有连接线
|
|
|
|
|
if isinstance(conn, AllSimConnect):
|
|
|
|
|
conn.delete_line()
|
|
|
|
|
delete_sql = f"delete from sim_conn where conn_id in (select conn_id from conn_config where node_id='{self.chose_obj.ObjID}')"
|
|
|
|
|
execute_sql(delete_sql)
|
|
|
|
|
delete_obj(self.chose_obj.ObjID)
|
|
|
|
|
self.delete("rectangle")
|
|
|
|
|
self.reload_data()
|
|
|
|
|
|
|
|
|
|
def delete_line(self):
|
|
|
|
|
# todo: 删除连接线
|
|
|
|
|
"""
|
|
|
|
|
删除连接线
|
|
|
|
|
:return:
|
|
|
|
|
"""
|
|
|
|
|
if self.chose_obj is None:
|
|
|
|
|
messagebox.showerror("注意", message="请先选择要删除连接线的对象！")
|
|
|
|
|
return
|
|
|
|
|
conn_sql = f"""
|
|
|
|
|
select
|
|
|
|
|
s.conn_id, ConfigCorrect, node_id, node_ifs
|
|
|
|
|
from sim_conn s
|
|
|
|
|
join conn_config c on s.conn_id=c.conn_id
|
|
|
|
|
where node_id='{self.chose_obj.ObjID}'
|
|
|
|
|
"""
|
|
|
|
|
conn_data = search(conn_sql)
|
|
|
|
|
conn_names = {}
|
|
|
|
|
for index, conn in conn_data.iterrows():
|
|
|
|
|
if conn["conn_id"] not in conn_names:
|
|
|
|
|
conn_names[conn["conn_id"]] = [(conn["node_id"], conn["node_ifs"])]
|
|
|
|
|
else:
|
|
|
|
|
conn_names[conn["conn_id"]].append((conn["node_id"], conn["node_ifs"]))
|
|
|
|
|
child_d = tk.Toplevel()
|
|
|
|
|
child_d.title(f"{self.chose_obj.ObjLabel}的连接线配置")
|
|
|
|
|
child_d.geometry('300x200+450+200')
|
|
|
|
|
child_d.grab_set()
|
|
|
|
|
cv1 = Canvas(child_d)
|
|
|
|
|
cv1.pack()
|
|
|
|
|
value = StringVar()
|
|
|
|
|
combobox = ttk.Combobox(
|
|
|
|
|
master=child_d, # 父容器
|
|
|
|
|
height=12, # 高度,下拉显示的条目数量
|
|
|
|
|
width=15, # 宽度
|
|
|
|
|
state='readonly', # readonly(只可选)
|
|
|
|
|
font=('', 18), # 字体
|
|
|
|
|
textvariable=value, # 通过StringVar设置可改变的值
|
|
|
|
|
values=list(conn_names.keys()), # 设置下拉框的选项
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
def del_line():
|
|
|
|
|
if value.get() == "":
|
|
|
|
|
messagebox.showerror("注意", message="请选择需要删除的连接线！")
|
|
|
|
|
return
|
|
|
|
|
conn_sql = f"""
|
|
|
|
|
select
|
|
|
|
|
conn_id, node_id, node_ifs
|
|
|
|
|
from conn_config
|
|
|
|
|
where conn_id='{value.get()}'
|
|
|
|
|
"""
|
|
|
|
|
conn_data = search(conn_sql)
|
|
|
|
|
delete_sql = f"delete from sim_conn where conn_id='{value.get()}'"
|
|
|
|
|
execute_sql(delete_sql)
|
|
|
|
|
conn_names = {}
|
|
|
|
|
for index, conn in conn_data.iterrows():
|
|
|
|
|
if conn["conn_id"] not in conn_names:
|
|
|
|
|
conn_names[conn["conn_id"]] = [(conn["node_id"], conn["node_ifs"])]
|
|
|
|
|
else:
|
|
|
|
|
conn_names[conn["conn_id"]].append((conn["node_id"], conn["node_ifs"]))
|
|
|
|
|
for data in conn_names[value.get()]:
|
|
|
|
|
self.AllSimObjs[data[0]].connections[data[1] - 1].delete_line()
|
|
|
|
|
self.AllSimObjs[data[0]].connections[data[1] - 1] = None
|
|
|
|
|
child_d.destroy()
|
|
|
|
|
messagebox.showinfo("提示", f"连接线{value.get()}删除成功！")
|
|
|
|
|
|
|
|
|
|
btn_yes = tk.Button(child_d, text='确定', font=('黑体', 12), height=1, command=del_line)
|
|
|
|
|
btn_yes.place(x=240, y=20)
|
|
|
|
|
combobox.place(x=20, y=20)
|
|
|
|
|
|
|
|
|
|
def update_tag_name(self):
|
|
|
|
|
# todo: 更新组件名称
|
|
|
|
|
"""
|
|
|
|
|
更新组件名称
|
|
|
|
|
:return:
|
|
|
|
|
"""
|
|
|
|
|
if self.chose_obj is None:
|
|
|
|
|
messagebox.showerror("注意", message="请先选择要更新的对象！")
|
|
|
|
|
return
|
|
|
|
|
child1 = tk.Toplevel()
|
|
|
|
|
child1.title(self.chose_obj.ObjLabel + "的标签信息")
|
|
|
|
|
child1.geometry('240x100+450+250')
|
|
|
|
|
child1.grab_set() # 设置组件焦点抓取。使焦点在释放之前永远保持在这个组件上,只能在这个组件上操作
|
|
|
|
|
tk.Label(child1, text='原标签:' + self.chose_obj.ObjLabel,
|
|
|
|
|
font=('黑体', 12)).grid(row=0, column=0, columnspan=2, sticky='w')
|
|
|
|
|
tk.Label(child1, text='新标签:', font=('黑体', 12)).grid(row=1, column=0, sticky='w') # ,sticky='w'靠左显示
|
|
|
|
|
new_name = tk.Entry(child1, font=('黑体', 12), textvariable=tk.StringVar())
|
|
|
|
|
new_name.grid(row=1, column=1)
|
|
|
|
|
|
|
|
|
|
def update_tag():
|
|
|
|
|
name = new_name.get()
|
|
|
|
|
if name == "":
|
|
|
|
|
messagebox.showerror("注意", message="请输入新名称！")
|
|
|
|
|
return
|
|
|
|
|
self.chose_obj.ObjLabel = name
|
|
|
|
|
self.chose_obj.update()
|
|
|
|
|
for conn in self.chose_obj.connections:
|
|
|
|
|
if isinstance(conn, AllSimConnect):
|
|
|
|
|
conn.update_info(self.chose_obj.ObjID)
|
|
|
|
|
child1.destroy()
|
|
|
|
|
messagebox.showinfo("提示", message="修改成功！")
|
|
|
|
|
|
|
|
|
|
tk.Button(child1, text='确定', font=('黑体', 10), height=1, command=update_tag).grid(row=2, column=0,
|
|
|
|
|
sticky='e')
|
|
|
|
|
tk.Button(child1, text='取消', font=('黑体', 10), height=1, command=child1.destroy).grid(row=2, column=1,
|
|
|
|
|
sticky='e')
|
|
|
|
|
|
|
|
|
|
def network_config(self):
|
|
|
|
|
# todo: 网络配置
|
|
|
|
|
"""
|
|
|
|
|
网络配置
|
|
|
|
|
:return:
|
|
|
|
|
"""
|
|
|
|
|
if self.chose_obj is None:
|
|
|
|
|
messagebox.showerror("注意", message="请先选择要配置的对象！")
|
|
|
|
|
return
|
|
|
|
|
if len(self.chose_obj.get_config()) == 0:
|
|
|
|
|
messagebox.showerror("注意", message="请先给对象添加连接线!")
|
|
|
|
|
return
|
|
|
|
|
child_r = tk.Toplevel()
|
|
|
|
|
child_r.title(self.chose_obj.ObjLabel + "的网络配置信息")
|
|
|
|
|
child_r.geometry('530x395+350+200')
|
|
|
|
|
child_r.grab_set() # 设置组件焦点抓取。使焦点在释放之前永远保持在这个组件上,只能在这个组件上操作
|
|
|
|
|
ifs_frame = tk.Frame(child_r)
|
|
|
|
|
ifs_frame.grid(row=1, column=0, columnspan=4) # 装接口的框架
|
|
|
|
|
ifs_frame_YN = tk.Frame(child_r)
|
|
|
|
|
ifs_frame_YN.grid(row=2, column=0, columnspan=4)
|
|
|
|
|
num = 0
|
|
|
|
|
datas = {}
|
|
|
|
|
for conn in self.chose_obj.connections:
|
|
|
|
|
if isinstance(conn, AllSimConnect):
|
|
|
|
|
index = self.chose_obj.connections.index(conn)
|
|
|
|
|
num_label = tk.LabelFrame(ifs_frame, text=f'接口{index + 1}')
|
|
|
|
|
num_label.grid(row=num, column=0, padx=18, ipady=5)
|
|
|
|
|
tk.Label(num_label, text='MAC:', font=('黑体', 10)).grid(row=0, column=0)
|
|
|
|
|
mac_en = tk.Entry(num_label, font=('黑体', 12), textvariable=tk.StringVar(), state=DISABLED if self.chose_obj.ObjType not in [1, 2, 3] else NORMAL)
|
|
|
|
|
mac_en.insert('0', str(self.chose_obj.interface[index].get("mac", "")))
|
|
|
|
|
mac_en.grid(row=0, column=1, padx=10)
|
|
|
|
|
tk.Label(num_label, text='IP:', font=('黑体', 10)).grid(row=1, column=0)
|
|
|
|
|
ip_en = tk.Entry(num_label, font=('黑体', 12), textvariable=tk.StringVar(), state=DISABLED if self.chose_obj.ObjType not in [1, 2] else NORMAL)
|
|
|
|
|
ip_en.insert('0', str(self.chose_obj.interface[index].get("ip", "")))
|
|
|
|
|
ip_en.grid(row=1, column=1, padx=10)
|
|
|
|
|
tk.Label(num_label, text='端口:', font=('黑体', 10)).grid(row=0, column=2)
|
|
|
|
|
port_en = tk.Entry(num_label, font=('黑体', 12), textvariable=tk.StringVar(), state=DISABLED if self.chose_obj.ObjType != 1 else NORMAL)
|
|
|
|
|
port_en.insert('0', str(self.chose_obj.interface[index].get("conn_port", "")))
|
|
|
|
|
port_en.grid(row=0, column=3, padx=10)
|
|
|
|
|
tk.Label(num_label, text='应用层地址:', font=('黑体', 10)).grid(row=1, column=2)
|
|
|
|
|
addr_en = tk.Entry(num_label, font=('黑体', 12), textvariable=tk.StringVar(), state=DISABLED if self.chose_obj.ObjType != 1 else NORMAL)
|
|
|
|
|
addr_en.insert('0', str(self.chose_obj.interface[index].get("addr", "")))
|
|
|
|
|
addr_en.grid(row=1, column=3, padx=10)
|
|
|
|
|
num += 1
|
|
|
|
|
datas[index + 1] = {"mac": mac_en, "ip": ip_en, "conn_port": port_en, "addr": addr_en}
|
|
|
|
|
|
|
|
|
|
num_label = tk.LabelFrame(ifs_frame, text=f'示例')
|
|
|
|
|
num_label.grid(row=num, column=0, padx=18, ipady=5)
|
|
|
|
|
tk.Label(num_label, text='MAC:', font=('黑体', 10)).grid(row=0, column=0)
|
|
|
|
|
mac_en = tk.Entry(num_label, font=('黑体', 12), textvariable=tk.StringVar())
|
|
|
|
|
mac_en.insert('0', "MAC11")
|
|
|
|
|
mac_en.config(state=READONLY)
|
|
|
|
|
mac_en.grid(row=0, column=1, padx=10)
|
|
|
|
|
tk.Label(num_label, text='IP:', font=('黑体', 10)).grid(row=1, column=0)
|
|
|
|
|
ip_en = tk.Entry(num_label, font=('黑体', 12), textvariable=tk.StringVar())
|
|
|
|
|
ip_en.insert(0, "10.1.1.10")
|
|
|
|
|
ip_en.config(state=READONLY)
|
|
|
|
|
ip_en.grid(row=1, column=1, padx=10)
|
|
|
|
|
tk.Label(num_label, text='端口:', font=('黑体', 10)).grid(row=0, column=2)
|
|
|
|
|
port_en = tk.Entry(num_label, font=('黑体', 12), textvariable=tk.StringVar())
|
|
|
|
|
port_en.insert('0', "80")
|
|
|
|
|
port_en.config(state=READONLY)
|
|
|
|
|
port_en.grid(row=0, column=3, padx=10)
|
|
|
|
|
tk.Label(num_label, text='应用层地址:', font=('黑体', 10)).grid(row=1, column=2)
|
|
|
|
|
addr_en = tk.Entry(num_label, font=('黑体', 12), textvariable=tk.StringVar())
|
|
|
|
|
addr_en.insert('0', "10.1.2.10:10810:Name3")
|
|
|
|
|
addr_en.config(state=READONLY)
|
|
|
|
|
addr_en.grid(row=1, column=3, padx=10)
|
|
|
|
|
def commit():
|
|
|
|
|
self.chose_obj.config(datas)
|
|
|
|
|
child_r.destroy()
|
|
|
|
|
|
|
|
|
|
tk.Button(ifs_frame_YN, text='确定', font=('黑体', 10), height=1, command=commit).grid(row=0, column=0,
|
|
|
|
|
padx=20) # ,sticky="w"
|
|
|
|
|
tk.Button(ifs_frame_YN, text='取消', font=('黑体', 10), height=1,
|
|
|
|
|
command=child_r.destroy).grid(row=0, column=2, padx=20)
|
|
|
|
|
|
|
|
|
|
def router_table_config(self):
|
|
|
|
|
# todo: 路由表配置
|
|
|
|
|
"""
|
|
|
|
|
路由表配置
|
|
|
|
|
:return:
|
|
|
|
|
"""
|
|
|
|
|
if self.chose_obj is None:
|
|
|
|
|
messagebox.showerror("注意", message="请先选择要配置的对象！")
|
|
|
|
|
return
|
|
|
|
|
if self.chose_obj.ObjType != 2:
|
|
|
|
|
messagebox.showerror("注意", message="请选择路由器对象!")
|
|
|
|
|
return
|
|
|
|
|
RouterConfigWindow(self, self.chose_obj)
|
|
|
|
|
|
|
|
|
|
def mac_table_config(self):
|
|
|
|
|
# todo: 交换表配置
|
|
|
|
|
"""
|
|
|
|
|
交换表配置
|
|
|
|
|
:return:
|
|
|
|
|
"""
|
|
|
|
|
if self.chose_obj is None:
|
|
|
|
|
messagebox.showerror("注意", message="请先选择要配置的对象！")
|
|
|
|
|
return
|
|
|
|
|
if self.chose_obj.ObjType != 3:
|
|
|
|
|
messagebox.showerror("注意", message="请选择交换机对象!")
|
|
|
|
|
return
|
|
|
|
|
SwitchConfigWindow(self, self.chose_obj)
|
|
|
|
|
|
|
|
|
|
def show_label(self):
|
|
|
|
|
"""
|
|
|
|
|
显示/不显示标签
|
|
|
|
|
:return:
|
|
|
|
|
"""
|
|
|
|
|
self.show_label_flag = not self.show_label_flag
|
|
|
|
|
if self.show_label_flag:
|
|
|
|
|
for tag in self.AllSimObjs.values():
|
|
|
|
|
tag.create_img()
|
|
|
|
|
else:
|
|
|
|
|
for tag in self.AllSimObjs.values():
|
|
|
|
|
self.delete(tag.ObjID + "text")
|
|
|
|
|
|
|
|
|
|
def show_interface(self):
|
|
|
|
|
# todo: 显示/不显示接口
|
|
|
|
|
"""
|
|
|
|
|
显示/不显示接口
|
|
|
|
|
:return:
|
|
|
|
|
"""
|
|
|
|
|
self.show_interface_flag = not self.show_interface_flag
|
|
|
|
|
if self.show_interface_flag:
|
|
|
|
|
for conn in self.conns:
|
|
|
|
|
conn.draw_line()
|
|
|
|
|
else:
|
|
|
|
|
for conn in self.conns:
|
|
|
|
|
self.delete(conn.NobjS.ObjID + conn.NobjE.ObjID)
|
|
|
|
|
|
|
|
|
|
def show_network_config(self):
|
|
|
|
|
# todo: 显示网络配置信息
|
|
|
|
|
"""
|
|
|
|
|
显示网络配置信息
|
|
|
|
|
:return:
|
|
|
|
|
"""
|
|
|
|
|
if self.chose_obj is None:
|
|
|
|
|
messagebox.showerror("注意", message="请先选择要显示的对象！")
|
|
|
|
|
return
|
|
|
|
|
self.delete("netSet")
|
|
|
|
|
self.create_text(740, 120, text="(" + self.chose_obj.ObjLabel + ")", anchor="n", font=('微软雅黑', 10, 'bold'),
|
|
|
|
|
fill="#7030a0", tags="netSet")
|
|
|
|
|
self.create_text(675, 145, text=self.chose_obj,
|
|
|
|
|
anchor="nw", font=('宋体', 11), tags="netSet")
|
|
|
|
|
|
|
|
|
|
def show_router_config(self):
|
|
|
|
|
# todo: 显示路由表交换表信息
|
|
|
|
|
"""
|
|
|
|
|
显示路由交换表信息
|
|
|
|
|
:return:
|
|
|
|
|
"""
|
|
|
|
|
if self.chose_obj is None:
|
|
|
|
|
messagebox.showerror("注意", message="请先选择要显示的对象！")
|
|
|
|
|
return
|
|
|
|
|
if self.chose_obj.ObjType != 2 and self.chose_obj.ObjType != 3:
|
|
|
|
|
messagebox.showerror("注意", message="请选择路由器/交换机对象！")
|
|
|
|
|
return
|
|
|
|
|
self.delete("routerSet")
|
|
|
|
|
self.create_text(905, 120, text="(" + self.chose_obj.ObjLabel + ")", anchor="n", font=('微软雅黑', 10, 'bold'),
|
|
|
|
|
fill="#7030a0", tags="routerSet")
|
|
|
|
|
self.create_text(835, 145, text=self.chose_obj.get_table_config(),
|
|
|
|
|
anchor="nw", font=('宋体', 11), tags="routerSet")
|
|
|
|
|
|
|
|
|
|
def send_packet(self):
|
|
|
|
|
"""
|
|
|
|
|
发送数据包
|
|
|
|
|
:return:
|
|
|
|
|
"""
|
|
|
|
|
if self.chose_obj is None:
|
|
|
|
|
messagebox.showerror("注意", message="请先选择要显示的对象！")
|
|
|
|
|
return
|
|
|
|
|
if self.chose_obj.ConfigCorrect != 1:
|
|
|
|
|
messagebox.showerror("注意", message="请先对选择对象进行网络配置！")
|
|
|
|
|
return
|
|
|
|
|
if self.chose_obj.ObjType != 1:
|
|
|
|
|
messagebox.showerror("注意", message="请选择主机对象！")
|
|
|
|
|
return
|
|
|
|
|
child2 = tk.Toplevel()
|
|
|
|
|
child2.title("数据包配置")
|
|
|
|
|
child2.geometry('240x100+450+250')
|
|
|
|
|
child2.grab_set() # 设置组件焦点抓取。使焦点在释放之前永远保持在这个组件上,只能在这个组件上操作
|
|
|
|
|
tk.Label(child2, text='目的IP:', font=('黑体', 12)).grid(row=0, column=0, columnspan=2, sticky='w')
|
|
|
|
|
packet_ip = tk.Entry(child2, font=('黑体', 12), textvariable=tk.StringVar())
|
|
|
|
|
packet_ip.grid(row=0, column=1)
|
|
|
|
|
tk.Label(child2, text='目的MAC:', font=('黑体', 12)).grid(row=1, column=0, sticky='w') # ,sticky='w'靠左显示
|
|
|
|
|
packet_mac = tk.Entry(child2, font=('黑体', 12), textvariable=tk.StringVar())
|
|
|
|
|
packet_mac.grid(row=1, column=1)
|
|
|
|
|
tk.Label(child2, text='消息:', font=('黑体', 12)).grid(row=2, column=0, sticky='w') # ,sticky='w'靠左显示
|
|
|
|
|
packet_message = tk.Entry(child2, font=('黑体', 12), textvariable=tk.StringVar())
|
|
|
|
|
packet_message.grid(row=2, column=1)
|
|
|
|
|
|
|
|
|
|
def send():
|
|
|
|
|
"""
|
|
|
|
|
发送数据包
|
|
|
|
|
:return:
|
|
|
|
|
"""
|
|
|
|
|
if packet_ip.get() == "":
|
|
|
|
|
messagebox.showerror("注意", message="ip地址不能为空")
|
|
|
|
|
return
|
|
|
|
|
if not validate_ip_address(packet_ip.get()):
|
|
|
|
|
messagebox.showerror("注意", message="IP地址不规范！")
|
|
|
|
|
return
|
|
|
|
|
if packet_mac.get() == "":
|
|
|
|
|
messagebox.showerror("注意", message="mac地址不能为空！")
|
|
|
|
|
return
|
|
|
|
|
if packet_message.get() == "":
|
|
|
|
|
messagebox.showerror("注意", message="消息不能为空！")
|
|
|
|
|
return
|
|
|
|
|
self.chose_obj.create_packet(packet_ip.get(),
|
|
|
|
|
packet_mac.get(),
|
|
|
|
|
packet_message.get())
|
|
|
|
|
child2.destroy()
|
|
|
|
|
|
|
|
|
|
tk.Button(child2, text='确定', font=('黑体', 10), height=1, command=send).grid(row=3, column=0, sticky='e')
|
|
|
|
|
tk.Button(child2, text='取消', font=('黑体', 10), height=1, command=child2.destroy).grid(row=3, column=1,
|
|
|
|
|
sticky='e')
|
|
|
|
|
|
|
|
|
|
def send_packet_list(self):
|
|
|
|
|
"""
|
|
|
|
|
批量发送数据包
|
|
|
|
|
:return:
|
|
|
|
|
"""
|
|
|
|
|
hosts = {}
|
|
|
|
|
for tag in self.AllSimObjs.values():
|
|
|
|
|
if tag.ObjType == 1 and tag.ConfigCorrect == 1:
|
|
|
|
|
hosts[tag.ObjLabel] = tag.ObjID
|
|
|
|
|
child2 = tk.Toplevel()
|
|
|
|
|
child2.title("批量数据包配置")
|
|
|
|
|
child2.geometry('400x420+450+200')
|
|
|
|
|
tk.Label(child2, text='目的IP:', font=('黑体', 12)).grid(row=0, column=0, columnspan=2, sticky='w', ipady=10)
|
|
|
|
|
packet_ip = tk.Entry(child2, font=('黑体', 12), textvariable=tk.StringVar())
|
|
|
|
|
packet_ip.grid(row=0, column=1, pady=5)
|
|
|
|
|
tk.Label(child2, text='目的MAC:', font=('黑体', 12)).grid(row=1, column=0, sticky='w',
|
|
|
|
|
pady=5) # ,sticky='w'靠左显示
|
|
|
|
|
packet_mac = tk.Entry(child2, font=('黑体', 12), textvariable=tk.StringVar())
|
|
|
|
|
packet_mac.grid(row=1, column=1, pady=5)
|
|
|
|
|
tk.Label(child2, text='消息:', font=('黑体', 12)).grid(row=2, column=0, sticky='w', pady=5) # ,sticky='w'靠左显示
|
|
|
|
|
packet_message = tk.Entry(child2, font=('黑体', 12), textvariable=tk.StringVar())
|
|
|
|
|
packet_message.grid(row=2, column=1, pady=5)
|
|
|
|
|
host = StringVar()
|
|
|
|
|
tk.Label(child2, text='发送主机:', font=('黑体', 12)).grid(row=3, column=0, sticky='w',
|
|
|
|
|
pady=5) # ,sticky='w'靠左显示
|
|
|
|
|
combobox = ttk.Combobox(
|
|
|
|
|
master=child2, # 父容器
|
|
|
|
|
height=5, # 高度,下拉显示的条目数量
|
|
|
|
|
width=12, # 宽度
|
|
|
|
|
state='readonly', # readonly(只可选)
|
|
|
|
|
font=('', 18), # 字体
|
|
|
|
|
textvariable=host, # 通过StringVar设置可改变的值
|
|
|
|
|
values=list(hosts.keys()), # 设置下拉框的选项
|
|
|
|
|
)
|
|
|
|
|
combobox.grid(row=3, column=1, pady=5)
|
|
|
|
|
packet_frame = tk.Frame(child2)
|
|
|
|
|
packet_frame.grid(row=5, column=0, columnspan=3, padx=10, pady=5)
|
|
|
|
|
packet_treeview = ttk.Treeview(packet_frame, columns=("source_host", "ip", "mac", "message"), show="headings")
|
|
|
|
|
packet_treeview.heading("source_host", text="发送主机")
|
|
|
|
|
packet_treeview.column("source_host", width=60, anchor=CENTER)
|
|
|
|
|
packet_treeview.heading("ip", text="IP")
|
|
|
|
|
packet_treeview.column("ip", width=120, anchor=CENTER)
|
|
|
|
|
packet_treeview.heading("mac", text="MAC")
|
|
|
|
|
packet_treeview.column("mac", width=60, anchor=CENTER)
|
|
|
|
|
packet_treeview.heading("message", text="消息")
|
|
|
|
|
packet_treeview.column("message", width=120, anchor=CENTER)
|
|
|
|
|
packet_treeview.pack(side="left", fill="both")
|
|
|
|
|
scrollbar = ttk.Scrollbar(packet_frame, orient="vertical", command=packet_treeview.yview)
|
|
|
|
|
scrollbar.pack(side="right", fill="y")
|
|
|
|
|
packet_treeview.configure(yscrollcommand=scrollbar.set)
|
|
|
|
|
packets = []
|
|
|
|
|
|
|
|
|
|
def add():
|
|
|
|
|
ip = packet_ip.get()
|
|
|
|
|
mac = packet_mac.get()
|
|
|
|
|
message = packet_message.get()
|
|
|
|
|
chose_host = host.get()
|
|
|
|
|
if ip == "" or mac == "" or message == "" or chose_host == "":
|
|
|
|
|
messagebox.showerror("注意", message="输入框不能为空")
|
|
|
|
|
return
|
|
|
|
|
packet = SimPacket(self.AllSimObjs[hosts[chose_host]].interface[0]["ip"],
|
|
|
|
|
self.AllSimObjs[hosts[chose_host]].interface[0]["mac"],
|
|
|
|
|
ip, mac, message)
|
|
|
|
|
packets.append((self.AllSimObjs[hosts[chose_host]], packet))
|
|
|
|
|
packet_treeview.delete(*packet_treeview.get_children())
|
|
|
|
|
for datas in packets:
|
|
|
|
|
tag: SimBase = datas[0]
|
|
|
|
|
packet_data: SimPacket = datas[1]
|
|
|
|
|
packet_treeview.insert("", "end", values=(
|
|
|
|
|
tag.ObjLabel, packet_data.destination_ip, packet_data.destination_mac, packet_data.message))
|
|
|
|
|
|
|
|
|
|
def send():
|
|
|
|
|
if len(packets) == 0:
|
|
|
|
|
messagebox.showerror("注意", message="请至少添加一条数据包！")
|
|
|
|
|
return
|
|
|
|
|
for data in packets:
|
|
|
|
|
threading.Thread(target=data[0].send, args=(data[1],)).start()
|
|
|
|
|
child2.destroy()
|
|
|
|
|
|
|
|
|
|
tk.Button(child2, text="添加数据包", font=('黑体', 12), height=3, command=add).grid(row=0, column=2, rowspan=4)
|
|
|
|
|
button_frame = Frame(child2)
|
|
|
|
|
tk.Button(button_frame, text="发送", font=('黑体', 12), height=1, command=send).pack(side=RIGHT, padx=20)
|
|
|
|
|
tk.Button(button_frame, text="取消", font=('黑体', 12), height=1, command=child2.destroy).pack(side=RIGHT)
|
|
|
|
|
button_frame.grid(row=6, column=0, columnspan=3)
|
|
|
|
|
|
|
|
|
|
def clear_canvas(self):
|
|
|
|
|
"""
|
|
|
|
|
清除画布
|
|
|
|
|
:return:
|
|
|
|
|
"""
|
|
|
|
|
ask = messagebox.askquestion(title='确认操作', message='确认要清除画布吗？')
|
|
|
|
|
if ask == "no":
|
|
|
|
|
return
|
|
|
|
|
truncate_db() # 清除数据库
|
|
|
|
|
for tag_id, tag in self.AllSimObjs.items():
|
|
|
|
|
self.delete(tag_id)
|
|
|
|
|
self.delete(tag_id + "text")
|
|
|
|
|
for conn in self.conns:
|
|
|
|
|
conn.delete_line()
|
|
|
|
|
self.delete("rectangle")
|
|
|
|
|
self.AllSimObjs.clear()
|
|
|
|
|
self.conns.clear()
|
|
|
|
|
|
|
|
|
|
def create_widget(self):
|
|
|
|
|
# todo: 创建页面
|
|
|
|
|
"""
|
|
|
|
|
创建整体页面布局
|
|
|
|
|
:return:
|
|
|
|
|
"""
|
|
|
|
|
self.create_rectangle(150, 40, 650, 490, outline="#7f6000", width=3) # 矩形框，左上角坐标，右下角坐标
|
|
|
|
|
self.create_rectangle(660, 100, 815, 400, outline="#ffff00", width=3, fill="#f4b88e") # 矩形框，左上角坐标，右下角坐标
|
|
|
|
|
self.create_text(735, 105, text="网络配置信息", anchor="n", font=('微软雅黑', 11, 'bold'))
|
|
|
|
|
self.create_rectangle(825, 100, 1000, 400, outline="#ffff00", width=3, fill="#f4b88e") # 矩形框，左上角坐标，右下角坐标
|
|
|
|
|
self.create_text(915, 105, text="路由/交换表信息", anchor="n", font=('微软雅黑', 11, 'bold'))
|
|
|
|
|
# 显示左边的固定图片
|
|
|
|
|
self.create_text(80 - 55, 120 + 15, text="路由器", anchor="nw", font=('微软雅黑', 14, 'bold')) # 显示文字
|
|
|
|
|
router = self.create_image(80, 120, image=self.router_img, anchor="nw")
|
|
|
|
|
self.create_text(80 - 55, 190 + 15, text="交换机", anchor="nw", font=('微软雅黑', 14, 'bold')) # 显示文字
|
|
|
|
|
switch = self.create_image(80, 190, image=self.switch_img, anchor="nw")
|
|
|
|
|
self.create_text(80 - 55, 260 + 15, text="集线器", anchor="nw", font=('微软雅黑', 14, 'bold')) # 显示文字
|
|
|
|
|
hub = self.create_image(80, 260, image=self.hub_img, anchor="nw")
|
|
|
|
|
self.create_text(80 - 55, 330 + 15, text="主机", anchor="nw", font=('微软雅黑', 14, 'bold')) # 显示文字
|
|
|
|
|
host = self.create_image(80, 330, image=self.host_img, anchor="nw")
|
|
|
|
|
self.bind_event(router, "路由器")
|
|
|
|
|
self.bind_event(switch, "交换机")
|
|
|
|
|
self.bind_event(hub, "集线器")
|
|
|
|
|
self.bind_event(host, "主机")
|
|
|
|
|
|
|
|
|
|
# 创建一个菜单栏，这里我们可以把他理解成一个容器，在窗口的上方
|
|
|
|
|
menubar = tk.Menu(root)
|
|
|
|
|
# 定义一个空菜单单元
|
|
|
|
|
setMenu = tk.Menu(menubar, tearoff=0)
|
|
|
|
|
menubar.add_cascade(label='删除与修改', menu=setMenu)
|
|
|
|
|
setMenu.add_command(label='删除对象', command=self.delete_obj)
|
|
|
|
|
setMenu.add_command(label='删除连接线', command=self.delete_line)
|
|
|
|
|
setMenu.add_command(label='修改标签', command=self.update_tag_name)
|
|
|
|
|
# # 定义一个空菜单单元
|
|
|
|
|
setMenu = tk.Menu(menubar, tearoff=0)
|
|
|
|
|
menubar.add_cascade(label='基础配置', menu=setMenu)
|
|
|
|
|
setMenu.add_command(label='网络配置', command=self.network_config)
|
|
|
|
|
setMenu.add_command(label='路由表配置', command=self.router_table_config)
|
|
|
|
|
setMenu.add_command(label='交换表配置', command=self.mac_table_config)
|
|
|
|
|
root.config(menu=menubar)
|
|
|
|
|
# 定义一个空菜单单元
|
|
|
|
|
setMenu = tk.Menu(menubar, tearoff=0)
|
|
|
|
|
menubar.add_cascade(label='显示设置', menu=setMenu)
|
|
|
|
|
setMenu.add_command(label='显示/不显示标签', command=self.show_label)
|
|
|
|
|
setMenu.add_command(label='显示/不显示接口', command=self.show_interface)
|
|
|
|
|
setMenu.add_command(label='显示网络配置信息', command=self.show_network_config)
|
|
|
|
|
setMenu.add_command(label='显示路由/交换表信息', command=self.show_router_config)
|
|
|
|
|
# 定义一个空菜单单元
|
|
|
|
|
setMenu = tk.Menu(menubar, tearoff=0)
|
|
|
|
|
menubar.add_cascade(label='发送数据包', menu=setMenu)
|
|
|
|
|
setMenu.add_command(label='发送数据包', command=self.send_packet)
|
|
|
|
|
setMenu.add_command(label='批量发送数据包', command=self.send_packet_list)
|
|
|
|
|
|
|
|
|
|
menubar.add_command(label='清除屏幕', command=self.clear_canvas)
|
|
|
|
|
root.config(menu=menubar)
|
|
|
|
|
self.reload_data()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
|
|
root = Window()
|
|
|
|
|
root.title('网络拓扑图')
|
|
|
|
|
width = 1030
|
|
|
|
|
height = 530 # 窗口大小
|
|
|
|
|
screen_width = root.winfo_screenwidth() # winfo方法来获取当前电脑屏幕大小
|
|
|
|
|
screen_height = root.winfo_screenheight()
|
|
|
|
|
x = int((screen_width - width) / 2)
|
|
|
|
|
y = int((screen_height - height) / 2) - 40
|
|
|
|
|
size = '{}x{}+{}+{}'.format(width, height, x, y)
|
|
|
|
|
canvas = NetWorkAnalog(root, width=1030, heigh=height, bg="white")
|
|
|
|
|
canvas.place(x=0, y=0, anchor='nw')
|
|
|
|
|
root.geometry(size)
|
|
|
|
|
root.mainloop()
|
