Python实现检索指定网段内所有的数据库服务器!
Python实现检索指定网段内所有的数据库服务器!
作者:mosquito_lover1
这篇文章主要为大家详细介绍了如何使用Python实现检索指定网段内所有的数据库服务器,文中的示例代码讲解详细,感兴趣的小伙伴可以了解下。
功能介绍
图形化界面,使用tkinter实现
可以输入单个IP地址或CIDR格式的网段
支持扫描Oracle、SQL Server、MySQL、PostgreSQL和SQLite五种数据库
实时显示扫描结果,并按数据库类型分类显示
显示每种数据库类型的发现数量
可以随时停止扫描
多线程实现,不会卡住界面
不会弹出命令行窗口
界面截图

核心源码
| 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 | importtkinter as tkfromtkinter importttk, messagebox, filedialogimportthreadingimportipaddressimportsocketimportqueueimporttimeimportxml.etree.ElementTree as ETfromdatetime importdatetimeimportosimportmatplotlib.pyplot as pltfrommatplotlib.figure importFigurefrommatplotlib.backends.backend_tkagg importFigureCanvasTkAggimportmatplotlibmatplotlib.use('TkAgg')classDatabaseScanner:def__init__(self):self.root =tk.Tk()self.root.title("数据库服务器扫描工具")# 获取屏幕尺寸screen_width =self.root.winfo_screenwidth()screen_height =self.root.winfo_screenheight()# 设置窗口初始大小为屏幕的80%宽度和90%高度window_width =int(screen_width *0.8)window_height =int(screen_height *0.9)  # 增加高度比例# 计算窗口位置使其居中x =(screen_width -window_width) //2y =(screen_height -window_height) //2# 设置窗口大小和位置self.root.geometry(f"{window_width}x{window_height}+{x}+{y}")# 设置窗口最小尺寸self.root.minsize(800, 800)  # 增加最小高度# 设置主题色self.colors ={'bg': '#f0f0f0','frame_bg': '#ffffff','button': '#2196f3','button_hover': '#1976d2','success': '#4caf50','warning': '#ff9800','error': '#f44336'}# 设置样式self.style =ttk.Style()self.style.configure('Main.TLabelframe', background=self.colors['frame_bg'])self.style.configure('Main.TLabelframe.Label', background=self.colors['frame_bg'], font=('微软雅黑', 10, 'bold'))self.style.configure('Scan.TButton', background=self.colors['button'], font=('微软雅黑', 9))self.style.configure('Report.TButton', background=self.colors['success'],font=('微软雅黑', 9))# 创建数据存储变量self.scan_running =Falseself.result_queue =queue.Queue()self.db_results ={'Oracle': [],'SQL Server': [],'MySQL': [],'PostgreSQL': [],'SQLite': []}self.root.configure(bg=self.colors['bg'])self.setup_ui()defsetup_ui(self):# 创建主框架,使用grid布局main_frame =ttk.Frame(self.root)main_frame.grid(row=0, column=0, sticky="nsew")# 配置root和main_frame的grid权重self.root.grid_rowconfigure(0, weight=1)self.root.grid_columnconfigure(0, weight=1)# 调整各区域的相对大小比例main_frame.grid_rowconfigure(0, weight=0)  # 扫描设置固定高度main_frame.grid_rowconfigure(1, weight=1)  # 进度日志区域main_frame.grid_rowconfigure(2, weight=3)  # 增加图表区域的比例main_frame.grid_rowconfigure(3, weight=4)  # 增加结果区域的比例main_frame.grid_columnconfigure(0, weight=1)# 输入框框架input_frame =ttk.LabelFrame(main_frame, text="扫描设置", padding=10, style='Main.TLabelframe')input_frame.grid(row=0, column=0, sticky="ew", padx=10, pady=5)ttk.Label(input_frame, text="IP地址/网段:", background=self.colors['frame_bg'],font=('微软雅黑', 9)).grid(row=0, column=0, padx=5)self.ip_entry =ttk.Entry(input_frame, width=40, font=('微软雅黑', 9))self.ip_entry.grid(row=0, column=1, padx=5)self.ip_entry.insert(0, "192.168.1.0/24")button_frame =ttk.Frame(input_frame, style='Main.TLabelframe')button_frame.grid(row=0, column=2, padx=5)self.scan_button =ttk.Button(button_frame, text="开始扫描", command=self.start_scan, style='Scan.TButton')self.scan_button.pack(side=tk.LEFT, padx=2)self.report_button =ttk.Button(button_frame, text="生成报告", command=self.generate_report, style='Report.TButton')self.report_button.pack(side=tk.LEFT, padx=2)# 配置input_frame的列权重input_frame.columnconfigure(1, weight=1)# 进度日志框架log_frame =ttk.LabelFrame(main_frame, text="扫描进度", padding=10, style='Main.TLabelframe')log_frame.grid(row=1, column=0, sticky="ew", padx=10, pady=5)self.progress_var =tk.StringVar(value="等待开始扫描...")self.progress_label =ttk.Label(log_frame, textvariable=self.progress_var,background=self.colors['frame_bg'],font=('微软雅黑', 9))self.progress_label.pack(fill=tk.X)self.log_text =tk.Text(log_frame, height=4, font=('微软雅黑', 9))self.log_text.pack(fill=tk.BOTH, expand=True)self.log_text.configure(bg='#fafafa', fg='#333333')log_scrollbar =ttk.Scrollbar(log_frame, command=self.log_text.yview)log_scrollbar.pack(side=tk.RIGHT, fill=tk.Y)self.log_text.config(yscrollcommand=log_scrollbar.set)# 添加统计图表框架chart_frame =ttk.LabelFrame(main_frame, text="数据库统计", padding=10, style='Main.TLabelframe')chart_frame.grid(row=2, column=0, sticky="nsew", padx=10, pady=5)# 创建图表self.figure =Figure(dpi=100)  # 移除固定figsizeself.figure.patch.set_facecolor(self.colors['frame_bg'])self.chart_ax =self.figure.add_subplot(111)self.canvas =FigureCanvasTkAgg(self.figure, master=chart_frame)self.canvas.get_tk_widget().pack(fill=tk.BOTH, expand=True)# 初始化图表self.update_chart()# 结果显示框架result_frame =ttk.LabelFrame(main_frame, text="扫描结果", padding=10, style='Main.TLabelframe')result_frame.grid(row=3, column=0, sticky="nsew", padx=10, pady=5)# 创建数据库类型标签页self.notebook =ttk.Notebook(result_frame)self.notebook.pack(fill=tk.BOTH, expand=True)# 为不同数据库类型设置不同的颜色db_colors ={'Oracle': '#f44336','SQL Server': '#2196f3','MySQL': '#4caf50','PostgreSQL': '#ff9800','SQLite': '#9c27b0'}self.db_text_widgets ={}fordb_type inself.db_results.keys():frame =ttk.Frame(self.notebook, style='Main.TLabelframe')self.notebook.add(frame, text=f"{db_type} (0)")text_widget =tk.Text(frame, height=8, font=('微软雅黑', 9))text_widget.pack(fill=tk.BOTH, expand=True)text_widget.configure(bg='#fafafa', fg=db_colors[db_type])scrollbar =ttk.Scrollbar(frame, command=text_widget.yview)scrollbar.pack(side=tk.RIGHT, fill=tk.Y)text_widget.config(yscrollcommand=scrollbar.set)self.db_text_widgets[db_type] =text_widget# 绑定窗口大小改变事件self.root.bind('<Configure>', self.on_window_resize)defon_window_resize(self, event):"""当窗口大小改变时调整图表大小"""ifhasattr(self, 'canvas'):# 延迟更新以避免频繁刷新self.root.after_cancel(self._resize_job) ifhasattr(self, '_resize_job') elseNoneself._resize_job =self.root.after(100, self.update_chart)defvalidate_ip_input(self, ip_input):try:if'/'inip_input:  # CIDR格式ipaddress.ip_network(ip_input, strict=False)else:  # 单个IP地址ipaddress.ip_address(ip_input)returnTrueexceptValueError:returnFalsedefstart_scan(self):ifself.scan_running:self.scan_running =Falseself.scan_button.config(text="开始扫描")returnip_input =self.ip_entry.get().strip()ifnotself.validate_ip_input(ip_input):messagebox.showerror("错误", "请输入有效的IP地址或网段")return# 清空之前的结果fordb_type inself.db_results:self.db_results[db_type] =[]self.db_text_widgets[db_type].delete(1.0, tk.END)self.notebook.tab(list(self.db_results.keys()).index(db_type), text=f"{db_type} (0)")# 重置图表self.update_chart()self.scan_running =Trueself.scan_button.config(text="停止扫描")# 启动扫描线程scan_thread =threading.Thread(target=self.scan_network, args=(ip_input,))scan_thread.daemon =Truescan_thread.start()# 启动更新UI的线程self.root.after(100, self.update_results)defscan_network(self, ip_input):try:if'/'inip_input:  # 网段network =ipaddress.ip_network(ip_input, strict=False)ip_list =list(network.hosts())else:  # 单个IPip_list =[ipaddress.ip_address(ip_input)]total_ips =len(ip_list)forindex, ip inenumerate(ip_list, 1):ifnotself.scan_running:breakip_str =str(ip)progress =f"正在扫描: {ip_str} ({index}/{total_ips})"self.result_queue.put(("progress", progress))self.result_queue.put(("log", f"正在扫描 {ip_str}...\n"))# 检查各种数据库端口self.check_database_ports(ip_str)exceptException as e:self.result_queue.put(("error", f"扫描出错: {str(e)}"))finally:self.scan_running =Falseself.result_queue.put(("progress", "扫描完成"))self.result_queue.put(("log", "扫描已完成\n"))self.root.after(0, lambda: self.scan_button.config(text="开始扫描"))defcheck_database_ports(self, ip):# 定义数据库默认端口db_ports ={'Oracle': [1521],'SQL Server': [1433],'MySQL': [3306],'PostgreSQL': [5432],'SQLite': []  # SQLite是文件型数据库,没有网络端口}fordb_type, ports indb_ports.items():forport inports:try:sock =socket.socket(socket.AF_INET, socket.SOCK_STREAM)sock.settimeout(1)result =sock.connect_ex((ip, port))ifresult ==0:self.result_queue.put(("found", db_type, ip, port))sock.close()except:continuedefupdate_chart(self):"""更新统计图表"""self.chart_ax.clear()# 获取图表区域的大小chart_width =self.canvas.get_tk_widget().winfo_width()chart_height =self.canvas.get_tk_widget().winfo_height()# 调整图表的纵横比aspect_ratio =chart_height /chart_width# 根据窗口大小调整图表尺寸,保持合适的纵横比self.figure.set_size_inches(chart_width/100, (chart_width *aspect_ratio)/100)# 准备数据db_types =list(self.db_results.keys())counts =[len(ips) forips inself.db_results.values()]colors =['#f44336', '#2196f3', '#4caf50', '#ff9800', '#9c27b0']# 创建柱状图bars =self.chart_ax.bar(db_types, counts, color=colors)# 设置图表样式plt.rcParams['font.sans-serif'] =['Microsoft YaHei']  # 使用微软雅黑字体plt.rcParams['axes.unicode_minus'] =Falseself.chart_ax.set_ylabel('数量', fontsize=10)self.chart_ax.set_title('数据库服务器统计', fontsize=12, pad=10)# 设置Y轴范围和刻度max_count =max(counts) ifcounts else0self.chart_ax.set_ylim(0, max_count +1ifmax_count > 0else1)self.chart_ax.yaxis.set_ticks(range(0, int(max_count +2)))# 设置背景颜色和网格self.figure.patch.set_facecolor(self.colors['frame_bg'])self.chart_ax.set_facecolor('#f8f9fa')self.chart_ax.grid(True, axis='y', linestyle='--', alpha=0.3)# 美化X轴标签self.chart_ax.set_xticklabels(db_types, rotation=0, fontsize=9)# 在柱子上显示具体数值forbar inbars:height =bar.get_height()self.chart_ax.text(bar.get_x() +bar.get_width()/2., height,f'{int(height)}',ha='center', va='bottom',fontsize=10,fontweight='bold')# 调整布局,确保所有元素都能显示self.figure.tight_layout()# 刷新画布self.canvas.draw()defupdate_results(self):whilenotself.result_queue.empty():message =self.result_queue.get()ifmessage[0] =="found":_, db_type, ip, port =messageifip notinself.db_results[db_type]:self.db_results[db_type].append(ip)self.db_text_widgets[db_type].insert(tk.END, f"{ip}:{port}\n")self.notebook.tab(list(self.db_results.keys()).index(db_type),text=f"{db_type} ({len(self.db_results[db_type])})")self.result_queue.put(("log", f"发现 {db_type} 服务器: {ip}:{port}\n"))# 更新统计图表self.update_chart()elifmessage[0] =="error":messagebox.showerror("错误", message[1])elifmessage[0] =="progress":self.progress_var.set(message[1])elifmessage[0] =="log":self.log_text.insert(tk.END, message[1])self.log_text.see(tk.END)ifself.scan_running:self.root.after(100, self.update_results)defgenerate_report(self):ifnotany(self.db_results.values()):messagebox.showwarning("警告", "没有扫描结果可供生成报告!")returntry:# 创建XML报告root =ET.Element("ScanReport")ET.SubElement(root, "ScanTime").text =datetime.now().strftime("%Y-%m-%d %H:%M:%S")ET.SubElement(root, "ScanTarget").text =self.ip_entry.get()databases =ET.SubElement(root, "Databases")fordb_type, ips inself.db_results.items():db_elem =ET.SubElement(databases, "Database")ET.SubElement(db_elem, "Type").text =db_typeET.SubElement(db_elem, "Count").text =str(len(ips))servers =ET.SubElement(db_elem, "Servers")forip inips:server =ET.SubElement(servers, "Server")server.text =ip# 保存文件对话框file_path =filedialog.asksaveasfilename(defaultextension=".xml",filetypes=[("XML files", "*.xml")],initialfile=f"scan_report_{datetime.now().strftime('%Y%m%d_%H%M%S')}.xml")iffile_path:tree =ET.ElementTree(root)tree.write(file_path, encoding="utf-8", xml_declaration=True)self.result_queue.put(("log", f"扫描报告已保存至: {file_path}\n"))messagebox.showinfo("成功", "扫描报告已生成!")exceptException as e:messagebox.showerror("错误", f"生成报告时出错: {str(e)}")defrun(self):self.root.mainloop()if__name__ =="__main__":scanner =DatabaseScanner()scanner.run() | 
到此这篇关于Python实现检索指定网段内所有的数据库服务器的文章就介绍到这了。
学习资料见知识星球。
以上就是今天要分享的技巧,你学会了吗?若有什么问题,欢迎在下方留言。
快来试试吧,小琥 my21ke007。获取 1000个免费 Excel模板福利!
更多技巧, www.excelbook.cn
欢迎 加入 零售创新 知识星球,知识星球主要以数据分析、报告分享、数据工具讨论为主;
1、价值上万元的专业的PPT报告模板。
2、专业案例分析和解读笔记。
3、实用的Excel、Word、PPT技巧。
4、VIP讨论群,共享资源。
5、优惠的会员商品。
6、一次付费只需129元,即可下载本站文章涉及的文件和软件。
