1131105-區網掃描工具練習

1131104-wakeonlan-test02-14-01

-匯出CSV

-使用繁體中文，想在以下程式中的"清除結果"的右邊新一個按鈕"匯出掃描資料"，功能是把掃描完後資料全部匯出，匯出可以選擇要存放的位置，要如何修正程式？ 

import tkinter as tk

from tkinter import ttk

import subprocess

import threading

import re

import socket

from wakeonlan import send_magic_packet

import ipaddress

import logging

from concurrent.futures import ThreadPoolExecutor

from tkinter import messagebox, filedialog

import threading

import csv

# 全局變量來跟踪進度

scan_count = 0  # 記錄已完成的掃描數量

total_count = 254  # 總共需要掃描的 IP 數量（從 1 到 254）

# 創建自定義的日誌處理器，將日誌輸出到 Text 小部件

class TkinterLogHandler(logging.Handler):

    def __init__(self, text_widget, auto_scroll=True):

        super().__init__()

        self.text_widget = text_widget

        self.auto_scroll = auto_scroll

    def emit(self, record):

        msg = self.format(record)

        self.text_widget.insert(tk.END, msg + '\n')

        if self.auto_scroll:

            self.text_widget.yview(tk.END)  # 自動滾動到底部

# 設定日誌

logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')

# 初始化 UI 和設定

root = tk.Tk()

root.title("區域網路掃描與遠端開機工具")

root.geometry("1000x600")

# 更新 Treeview 欄位，添加「電腦名稱」

tree = ttk.Treeview(root, columns=("序號", "IP Address", "MAC Address", "Computer Name", "Status"), show="headings")

tree.heading("序號", text="序號")  # 新增序號欄位的標題

tree.heading("IP Address", text="IP 位址")

tree.heading("MAC Address", text="MAC 位址")

tree.heading("Computer Name", text="電腦名稱")  # 新增的欄位

tree.heading("Status", text="狀態")

# 設定所有欄位的資料置中

tree.column("序號", anchor="center")

tree.column("IP Address", anchor="center")

tree.column("MAC Address", anchor="center")

tree.column("Computer Name", anchor="center")

tree.column("Status", anchor="center")

tree.pack(fill=tk.BOTH, expand=True)

# 儲存掃描結果的列表

scan_results = []

# 上鎖機制確保執行緒安全

lock = threading.Lock()

# 獲取本機 IP 網段（預設網段）

def get_local_ip_prefix():

    try:

        local_ip = socket.gethostbyname(socket.gethostname())  # 獲取本機的 IP 位址

        return '.'.join(local_ip.split('.')[:3]) + '.'  # 回傳網段（例如 192.168.1.）

    except Exception as e:

        logging.error(f"無法獲取本機 IP 位址: {e}")

        return "192.168.1."  # 預設網段

def get_computer_name(ip):

    # 嘗試通過 nslookup 獲取電腦名稱

    try:

        result = subprocess.run(["nslookup", ip], capture_output=True, text=True)

        for line in result.stdout.splitlines():

            if "name =" in line:

                return line.split("=")[-1].strip()

    except Exception as e:

        logging.error(f"獲取電腦名稱時發生錯誤: {e}")

    return "N/A"  # 確保返回有效值

def arp_scan(net_prefix):

    # 執行 ARP 命令來獲取局域網中的設備

    output = subprocess.run(["arp", "-a"], capture_output=True, text=True)

    for line in output.stdout.splitlines():

        # 使用正則表達式來匹配 IP 和 MAC 地址

        ip_match = re.search(r"\d+\.\d+\.\d+\.\d+", line)

        mac_match = re.search(r"([0-9A-Fa-f]{2}[:-]){5}([0-9A-Fa-f]{2})", line)

        if ip_match and mac_match:

            ip = ip_match.group(0)

            mac = mac_match.group(0)

            computer_name = get_computer_name(ip)  # 獲取電腦名稱

            status = "On"  # 假設在 ARP 表中找到的設備處於開啟狀態

            with lock:

                update_scan_results(ip, mac, computer_name, status)

def port_scan(ip, port=80):

    # 檢查 IP 是否為有效的 IPv4 格式

    try:

        ipaddress.IPv4Address(ip)  # 如果 IP 無效，會拋出 ValueError

    except ValueError:

        logging.error(f"無效的 IP 地址：{ip}")

        return False

    # 嘗試連接到指定 IP 的特定端口，以檢查是否存在

    try:

        with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as sock:

            sock.settimeout(0.5)  # 設置超時時間

            if sock.connect_ex((ip, port)) == 0:

                return True

    except socket.error as e:

        logging.error(f"端口掃描錯誤（{ip}）：{e}")

    return False

def threaded_port_scan(ip):

    # 執行執行緒中的端口掃描並更新結果

    if port_scan(ip, port=80):

        mac_address = "N/A"  # 如果無法通過 ARP 獲取 MAC，可以留空或設為 N/A

        computer_name = get_computer_name(ip)  # 獲取電腦名稱

        status = "On"

        with lock:

            update_scan_results(ip, mac_address, computer_name, status)

        update_tree()

def update_scan_results(ip, mac, computer_name, status):

    """更新掃描結果，處理重複的 IP 和無效的 MAC"""

    # 檢查 scan_results 是否已經存在該 IP，若有則檢查 MAC 是否為 "N/A"

    for idx, (existing_ip, existing_mac, existing_computer_name, existing_status) in enumerate(scan_results):

        if existing_ip == ip:

            # 如果已有的 MAC 為 "N/A"，且當前 MAC 地址有效，則更新 MAC

            if existing_mac == "N/A" and mac != "N/A":

                scan_results[idx] = (ip, mac, computer_name, status)

            return  # 找到並處理過就退出

    # 若是新 IP 或無重複的情況，直接加入

    scan_results.append((ip, mac, computer_name, status))

def update_tree(data=None):

    if data is None:

        data = scan_results

    else:

        data = sorted(data, key=lambda x: ipaddress.IPv4Address(x[0]))

    def update_ui():

        # 清空現有顯示

        tree.delete(*tree.get_children())

        for index, (ip, mac_address, computer_name, status) in enumerate(data):

            tree.insert("", tk.END, values=(index + 1, ip, mac_address, computer_name, status))  # 加入序號

    root.after(0, update_ui)

# 檢查網段格式是否有效

def is_valid_network_prefix(prefix):

    """

    檢查網段格式是否有效，可以是單純的 x.x.x. 或 CIDR 格式 x.x.x.x/xx。

    例如: 192.168.1. 或 192.168.1.0/24

    """

    # 如果網段以 "." 結尾，將其補充為最後一段 0

    if prefix.endswith('.'):

        prefix = prefix + '0'

    # 正則表達式匹配 x.x.x.x 或 x.x.x.x/xx（CIDR 格式）

    network_regex = r'^(\d{1,3}\.){3}\d{1,3}(/(\d|[12][0-9]|3[0-2]))?$'

    # 檢查網段格式是否符合正則表達式

    if re.match(network_regex, prefix):

        # 檢查每個 IP 部分是否在 0 到 255 之間

        try:

            parts = prefix.split('/')

            # 檢查 IP 地址部分

            ip_parts = parts[0].split('.')

            if len(ip_parts) == 4 and all(0 <= int(part) <= 255 for part in ip_parts):

                # 如果有 CIDR 部分，檢查是否是 0-32 之間的數字

                if len(parts) > 1:

                    subnet_mask = int(parts[1])

                    if 0 <= subnet_mask <= 32:

                        return True

                else:

                    return True  # 沒有 CIDR 部分也有效

        except ValueError:

            return False

    return False

# 更新進度條的函數

def update_progress_bar(value):

    progress_bar['value'] = value

    root.update_idletasks()

# 使用 ThreadPoolExecutor 進行多執行緒掃描

def scan_network():

    global scan_count

    net_prefix = network_entry.get()  # 從使用者輸入的 Entry 中獲取網段

    if not net_prefix or not is_valid_network_prefix(net_prefix):

        logging.error("請先輸入有效的網段！")

        messagebox.showerror("錯誤", "請先輸入有效的網段！")

        return

    tree.delete(*tree.get_children())  # 清空 Treeview 資料

    scan_results.clear()  # 清除之前的掃描結果

    scan_count = 0  # 重置掃描進度

    # 執行 ARP 掃描

    arp_scan(net_prefix)

    update_tree()  # 顯示 ARP 掃描結果

    # 端口掃描：為每個 IP 啟動執行緒來掃描

    with ThreadPoolExecutor(max_workers=20) as executor:  # 設定最大工作執行緒數

        future_to_ip = {executor.submit(threaded_port_scan, f"{net_prefix}{i}"): i for i in range(1, 255)}

        for future in future_to_ip:

            future.result()  # 確保每個任務的結果都被獲得

            # 更新進度條

            with lock:

                scan_count += 1  # 每完成一個掃描，進行一次累加

                progress = (scan_count / total_count) * 100

                root.after(0, update_progress_bar, progress)  # 使用 after 將更新交給主執行緒

    # 確保最終結果更新

    logging.info(f"掃描完成，共找到 {len(scan_results)} 台設備")

    update_tree(sorted(scan_results, key=lambda x: ipaddress.IPv4Address(x[0])))

# 發送遠端開機封包

def wake_on_lan():

    selected_item = tree.selection()

    if selected_item:

        mac_address = tree.item(selected_item)["values"][1]  # 確保正確取用 MAC 位址

        if mac_address != "N/A":

            send_magic_packet(mac_address)

            logging.info(f"發送遠端開機封包至 MAC: {mac_address}")

        else:

            logging.warning("選取的電腦沒有有效的 MAC 位址。")

# 匯出掃描結果到 CSV 檔案的函數

def export_scan_results():

    # 讓使用者選擇匯出位置

    file_path = filedialog.asksaveasfilename(defaultextension=".csv", filetypes=[("CSV Files", "*.csv")])

    if not file_path:

        return  # 如果沒有選擇檔案，則直接返回

    try:

        # 打開 CSV 檔案進行寫入

        with open(file_path, mode='w', newline='') as file:

            writer = csv.writer(file)

            # 寫入表頭

            writer.writerow(["序號", "IP Address", "MAC Address", "Computer Name", "Status"])

            # 寫入掃描結果

            for idx, (ip, mac_address, computer_name, status) in enumerate(scan_results):

                writer.writerow([idx + 1, ip, mac_address, computer_name, status])

        logging.info(f"掃描結果已成功匯出到 {file_path}")

        messagebox.showinfo("匯出成功", f"掃描結果已成功匯出到 {file_path}")

    except Exception as e:

        logging.error(f"匯出掃描結果時發生錯誤: {e}")

        messagebox.showerror("匯出失敗", f"匯出掃描結果時發生錯誤: {e}")

# 用來讓使用者輸入網段的輸入框和標籤

network_label = tk.Label(root, text="輸入掃描網段 (例如: 192.168.1.)")

network_label.pack(padx=10, pady=5)

# 預設為本機的網段

local_ip_prefix = get_local_ip_prefix()

network_entry = tk.Entry(root, width=20)

network_entry.pack(padx=10, pady=5)

network_entry.insert(0, local_ip_prefix)  # 預設顯示本機 IP 的網段

# 顯示提示文字

alert_label = tk.Label(root, text="", fg="red")

alert_label.pack(padx=10, pady=10)

# 創建一個 Text 小部件來顯示日誌訊息

log_text = tk.Text(root, height=3, width=80)

log_text.pack(side=tk.RIGHT,padx=5, pady=5)

# 設置自定義的日誌處理器

log_handler = TkinterLogHandler(log_text)

logging.getLogger().addHandler(log_handler)

# 創建掃描按鈕

scan_button = tk.Button(root, text="掃描區網", command=lambda: threading.Thread(target=scan_network).start())

scan_button.pack(side=tk.LEFT, padx=10, pady=10)

# 創建遠端開機按鈕

wake_button = tk.Button(root, text="遠端開機", command=wake_on_lan)

wake_button.pack(side=tk.LEFT, padx=10, pady=10)

# 創建清除結果按鈕

def clear_results():

    tree.delete(*tree.get_children())

    scan_results.clear()

clear_button = tk.Button(root, text="清除結果", command=clear_results)

clear_button.pack(side=tk.LEFT, padx=10, pady=10)

# 創建匯出掃描資料的按鈕

export_button = tk.Button(root, text="匯出掃描資料", command=export_scan_results)

export_button.pack(side=tk.LEFT, padx=10, pady=10)

# 創建進度條

progress_bar = ttk.Progressbar(root, orient="horizontal", length=200, mode="determinate")

progress_bar.pack(side=tk.BOTTOM, padx=10, pady=10)

root.mainloop()

1131104-wakeonlan-test02-13-01

-進度條快

-會有IP錯誤發生

如果想在"清除結果"的右邊新一個按鈕"匯出掃描資料"，功能是把掃描完後資料全部匯出，要如何修正程式？ 

import tkinter as tk

from tkinter import ttk

import subprocess

import threading

import re

import socket

from wakeonlan import send_magic_packet

import ipaddress

import logging

from concurrent.futures import ThreadPoolExecutor

from tkinter import messagebox

import threading

# 全局變量來跟踪進度

scan_count = 0  # 記錄已完成的掃描數量

total_count = 254  # 總共需要掃描的 IP 數量（從 1 到 254）

# 創建自定義的日誌處理器，將日誌輸出到 Text 小部件

class TkinterLogHandler(logging.Handler):

    def __init__(self, text_widget, auto_scroll=True):

        super().__init__()

        self.text_widget = text_widget

        self.auto_scroll = auto_scroll

    def emit(self, record):

        msg = self.format(record)

        self.text_widget.insert(tk.END, msg + '\n')

        if self.auto_scroll:

            self.text_widget.yview(tk.END)  # 自動滾動到底部

# 設定日誌

logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')

# 初始化 UI 和設定

root = tk.Tk()

root.title("區域網路掃描與遠端開機工具")

root.geometry("1000x600")

# 更新 Treeview 欄位，添加「電腦名稱」

tree = ttk.Treeview(root, columns=("序號", "IP Address", "MAC Address", "Computer Name", "Status"), show="headings")

tree.heading("序號", text="序號")  # 新增序號欄位的標題

tree.heading("IP Address", text="IP 位址")

tree.heading("MAC Address", text="MAC 位址")

tree.heading("Computer Name", text="電腦名稱")  # 新增的欄位

tree.heading("Status", text="狀態")

# 設定所有欄位的資料置中

tree.column("序號", anchor="center")

tree.column("IP Address", anchor="center")

tree.column("MAC Address", anchor="center")

tree.column("Computer Name", anchor="center")

tree.column("Status", anchor="center")

tree.pack(fill=tk.BOTH, expand=True)

# 儲存掃描結果的列表

scan_results = []

# 上鎖機制確保執行緒安全

lock = threading.Lock()

# 獲取本機 IP 網段（預設網段）

def get_local_ip_prefix():

    try:

        local_ip = socket.gethostbyname(socket.gethostname())  # 獲取本機的 IP 位址

        return '.'.join(local_ip.split('.')[:3]) + '.'  # 回傳網段（例如 192.168.1.）

    except Exception as e:

        logging.error(f"無法獲取本機 IP 位址: {e}")

        return "192.168.1."  # 預設網段

def get_computer_name(ip):

    # 嘗試通過 nslookup 獲取電腦名稱

    try:

        result = subprocess.run(["nslookup", ip], capture_output=True, text=True)

        for line in result.stdout.splitlines():

            if "name =" in line:

                return line.split("=")[-1].strip()

    except Exception as e:

        logging.error(f"獲取電腦名稱時發生錯誤: {e}")

    return "N/A"  # 確保返回有效值

def arp_scan(net_prefix):

    # 執行 ARP 命令來獲取局域網中的設備

    output = subprocess.run(["arp", "-a"], capture_output=True, text=True)

    for line in output.stdout.splitlines():

        # 使用正則表達式來匹配 IP 和 MAC 地址

        ip_match = re.search(r"\d+\.\d+\.\d+\.\d+", line)

        mac_match = re.search(r"([0-9A-Fa-f]{2}[:-]){5}([0-9A-Fa-f]{2})", line)

        if ip_match and mac_match:

            ip = ip_match.group(0)

            mac = mac_match.group(0)

            computer_name = get_computer_name(ip)  # 獲取電腦名稱

            status = "On"  # 假設在 ARP 表中找到的設備處於開啟狀態

            with lock:

                update_scan_results(ip, mac, computer_name, status)

def port_scan(ip, port=80):

    # 檢查 IP 是否為有效的 IPv4 格式

    try:

        ipaddress.IPv4Address(ip)  # 如果 IP 無效，會拋出 ValueError

    except ValueError:

        logging.error(f"無效的 IP 地址：{ip}")

        return False

    # 嘗試連接到指定 IP 的特定端口，以檢查是否存在

    try:

        with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as sock:

            sock.settimeout(0.5)  # 設置超時時間

            if sock.connect_ex((ip, port)) == 0:

                return True

    except socket.error as e:

        logging.error(f"端口掃描錯誤（{ip}）：{e}")

    return False

def threaded_port_scan(ip):

    # 執行執行緒中的端口掃描並更新結果

    if port_scan(ip, port=80):

        mac_address = "N/A"  # 如果無法通過 ARP 獲取 MAC，可以留空或設為 N/A

        computer_name = get_computer_name(ip)  # 獲取電腦名稱

        status = "On"

        with lock:

            update_scan_results(ip, mac_address, computer_name, status)

        update_tree()

def update_scan_results(ip, mac, computer_name, status):

    """更新掃描結果，處理重複的 IP 和無效的 MAC"""

    # 檢查 scan_results 是否已經存在該 IP，若有則檢查 MAC 是否為 "N/A"

    for idx, (existing_ip, existing_mac, existing_computer_name, existing_status) in enumerate(scan_results):

        if existing_ip == ip:

            # 如果已有的 MAC 為 "N/A"，且當前 MAC 地址有效，則更新 MAC

            if existing_mac == "N/A" and mac != "N/A":

                scan_results[idx] = (ip, mac, computer_name, status)

            return  # 找到並處理過就退出

    # 若是新 IP 或無重複的情況，直接加入

    scan_results.append((ip, mac, computer_name, status))

def update_tree(data=None):

    if data is None:

        data = scan_results

    else:

        data = sorted(data, key=lambda x: ipaddress.IPv4Address(x[0]))

    def update_ui():

        # 清空現有顯示

        tree.delete(*tree.get_children())

        for index, (ip, mac_address, computer_name, status) in enumerate(data):

            tree.insert("", tk.END, values=(index + 1, ip, mac_address, computer_name, status))  # 加入序號

    root.after(0, update_ui)

# 檢查網段格式是否有效

def is_valid_network_prefix(prefix):

    """

    檢查網段格式是否有效，可以是單純的 x.x.x. 或 CIDR 格式 x.x.x.x/xx。

    例如: 192.168.1. 或 192.168.1.0/24

    """

    # 正則表達式匹配 x.x.x.x 或 x.x.x.x/xx（CIDR 格式）

    network_regex = r'^(\d{1,3}\.){3}\d{1,3}(/(\d|[12][0-9]|3[0-2]))?$'

    # 檢查網段格式是否符合正則表達式

    if re.match(network_regex, prefix):

        # 檢查每個 IP 部分是否在 0 到 255 之間

        try:

            parts = prefix.split('/')

            # 檢查 IP 地址部分

            ip_parts = parts[0].split('.')

            if len(ip_parts) == 4 and all(0 <= int(part) <= 255 for part in ip_parts):

                # 如果有 CIDR 部分，檢查是否是 0-32 之間的數字

                if len(parts) > 1:

                    subnet_mask = int(parts[1])

                    if 0 <= subnet_mask <= 32:

                        return True

                else:

                    return True  # 沒有 CIDR 部分也有效

        except ValueError:

            return False

    return False

# 更新進度條的函數

def update_progress_bar(value):

    progress_bar['value'] = value

    root.update_idletasks()

# 使用 ThreadPoolExecutor 進行多執行緒掃描

def scan_network():

    global scan_count

    net_prefix = network_entry.get()  # 從使用者輸入的 Entry 中獲取網段

    if not net_prefix or not is_valid_network_prefix(net_prefix):

        logging.error("請先輸入有效的網段！")

        messagebox.showerror("錯誤", "請先輸入有效的網段！")

        return

    tree.delete(*tree.get_children())  # 清空 Treeview 資料

    scan_results.clear()  # 清除之前的掃描結果

    scan_count = 0  # 重置掃描進度

    # 執行 ARP 掃描

    arp_scan(net_prefix)

    update_tree()  # 顯示 ARP 掃描結果

    # 端口掃描：為每個 IP 啟動執行緒來掃描

    with ThreadPoolExecutor(max_workers=20) as executor:  # 設定最大工作執行緒數

        future_to_ip = {executor.submit(threaded_port_scan, f"{net_prefix}{i}"): i for i in range(1, 255)}

        for future in future_to_ip:

            future.result()  # 確保每個任務的結果都被獲得

            # 更新進度條

            with lock:

                scan_count += 1  # 每完成一個掃描，進行一次累加

                progress = (scan_count / total_count) * 100

                root.after(0, update_progress_bar, progress)  # 使用 after 將更新交給主執行緒

    # 確保最終結果更新

    logging.info(f"掃描完成，共找到 {len(scan_results)} 台設備")

    update_tree(sorted(scan_results, key=lambda x: ipaddress.IPv4Address(x[0])))

# 發送遠端開機封包

def wake_on_lan():

    selected_item = tree.selection()

    if selected_item:

        mac_address = tree.item(selected_item)["values"][1]  # 確保正確取用 MAC 位址

        if mac_address != "N/A":

            send_magic_packet(mac_address)

            logging.info(f"發送遠端開機封包至 MAC: {mac_address}")

        else:

            logging.warning("選取的電腦沒有有效的 MAC 位址。")

# 用來讓使用者輸入網段的輸入框和標籤

network_label = tk.Label(root, text="輸入掃描網段 (例如: 192.168.1.)")

network_label.pack(padx=10, pady=5)

# 預設為本機的網段

local_ip_prefix = get_local_ip_prefix()

network_entry = tk.Entry(root, width=20)

network_entry.pack(padx=10, pady=5)

network_entry.insert(0, local_ip_prefix)  # 預設顯示本機 IP 的網段

# 顯示提示文字

alert_label = tk.Label(root, text="", fg="red")

alert_label.pack(padx=10, pady=10)

# 創建一個 Text 小部件來顯示日誌訊息

log_text = tk.Text(root, height=3, width=80)

log_text.pack(side=tk.RIGHT,padx=5, pady=5)

# 設置自定義的日誌處理器

log_handler = TkinterLogHandler(log_text)

logging.getLogger().addHandler(log_handler)

# 創建掃描按鈕

scan_button = tk.Button(root, text="掃描區網", command=lambda: threading.Thread(target=scan_network).start())

scan_button.pack(side=tk.LEFT, padx=10, pady=10)

# 創建遠端開機按鈕

wake_button = tk.Button(root, text="遠端開機", command=wake_on_lan)

wake_button.pack(side=tk.LEFT, padx=10, pady=10)

# 創建清除結果按鈕

def clear_results():

    tree.delete(*tree.get_children())

    scan_results.clear()

clear_button = tk.Button(root, text="清除結果", command=clear_results)

clear_button.pack(side=tk.LEFT, padx=10, pady=10)

# 創建進度條

progress_bar = ttk.Progressbar(root, orient="horizontal", length=200, mode="determinate")

progress_bar.pack(side=tk.BOTTOM, padx=10, pady=10)

root.mainloop()

1131104-wakeonlan-test02-13

-進度條慢

-IP不會有錯誤發生

import tkinter as tk

from tkinter import ttk

import subprocess

import threading

import re

import socket

from wakeonlan import send_magic_packet

import ipaddress

import logging

from concurrent.futures import ThreadPoolExecutor

from tkinter import messagebox

import threading

# 全局變量來跟踪進度

scan_count = 0  # 記錄已完成的掃描數量

total_count = 254  # 總共需要掃描的 IP 數量（從 1 到 254）

# 創建自定義的日誌處理器，將日誌輸出到 Text 小部件

class TkinterLogHandler(logging.Handler):

    def __init__(self, text_widget, auto_scroll=True):

        super().__init__()

        self.text_widget = text_widget

        self.auto_scroll = auto_scroll

    def emit(self, record):

        msg = self.format(record)

        self.text_widget.insert(tk.END, msg + '\n')

        if self.auto_scroll:

            self.text_widget.yview(tk.END)  # 自動滾動到底部

# 設定日誌

logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')

# 初始化 UI 和設定

root = tk.Tk()

root.title("區域網路掃描與遠端開機工具")

root.geometry("1000x600")

# 更新 Treeview 欄位，添加「電腦名稱」

tree = ttk.Treeview(root, columns=("序號", "IP Address", "MAC Address", "Computer Name", "Status"), show="headings")

tree.heading("序號", text="序號")  # 新增序號欄位的標題

tree.heading("IP Address", text="IP 位址")

tree.heading("MAC Address", text="MAC 位址")

tree.heading("Computer Name", text="電腦名稱")  # 新增的欄位

tree.heading("Status", text="狀態")

# 設定所有欄位的資料置中

tree.column("序號", anchor="center")

tree.column("IP Address", anchor="center")

tree.column("MAC Address", anchor="center")

tree.column("Computer Name", anchor="center")

tree.column("Status", anchor="center")

tree.pack(fill=tk.BOTH, expand=True)

# 儲存掃描結果的列表

scan_results = []

# 上鎖機制確保執行緒安全

lock = threading.Lock()

# 獲取本機 IP 網段（預設網段）

def get_local_ip_prefix():

    try:

        local_ip = socket.gethostbyname(socket.gethostname())  # 獲取本機的 IP 位址

        return '.'.join(local_ip.split('.')[:3]) + '.'  # 回傳網段（例如 192.168.1.）

    except Exception as e:

        logging.error(f"無法獲取本機 IP 位址: {e}")

        return "192.168.1."  # 預設網段

def get_computer_name(ip):

    # 嘗試通過 nslookup 獲取電腦名稱

    try:

        result = subprocess.run(["nslookup", ip], capture_output=True, text=True)

        for line in result.stdout.splitlines():

            if "name =" in line:

                return line.split("=")[-1].strip()

    except Exception as e:

        logging.error(f"獲取電腦名稱時發生錯誤: {e}")

    return "N/A"  # 確保返回有效值

def arp_scan(net_prefix):

    # 執行 ARP 命令來獲取局域網中的設備

    output = subprocess.run(["arp", "-a"], capture_output=True, text=True)

    for line in output.stdout.splitlines():

        # 使用正則表達式來匹配 IP 和 MAC 地址

        ip_match = re.search(r"\d+\.\d+\.\d+\.\d+", line)

        mac_match = re.search(r"([0-9A-Fa-f]{2}[:-]){5}([0-9A-Fa-f]{2})", line)

        if ip_match and mac_match:

            ip = ip_match.group(0)

            mac = mac_match.group(0)

            computer_name = get_computer_name(ip)  # 獲取電腦名稱

            status = "On"  # 假設在 ARP 表中找到的設備處於開啟狀態

            with lock:

                update_scan_results(ip, mac, computer_name, status)

def port_scan(ip, port=80):

    # 檢查 IP 是否為有效的 IPv4 格式

    try:

        ipaddress.IPv4Address(ip)  # 如果 IP 無效，會拋出 ValueError

    except ValueError:

        logging.error(f"無效的 IP 地址：{ip}")

        return False

    # 嘗試連接到指定 IP 的特定端口，以檢查是否存在

    try:

        with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as sock:

            sock.settimeout(0.5)  # 設置超時時間

            if sock.connect_ex((ip, port)) == 0:

                return True

    except socket.error as e:

        logging.error(f"端口掃描錯誤（{ip}）：{e}")

    return False

def threaded_port_scan(ip):

    # 執行執行緒中的端口掃描並更新結果

    if port_scan(ip, port=80):

        mac_address = "N/A"  # 如果無法通過 ARP 獲取 MAC，可以留空或設為 N/A

        computer_name = get_computer_name(ip)  # 獲取電腦名稱

        status = "On"

        with lock:

            update_scan_results(ip, mac_address, computer_name, status)

        update_tree()

def update_scan_results(ip, mac, computer_name, status):

    """更新掃描結果，處理重複的 IP 和無效的 MAC"""

    # 檢查 scan_results 是否已經存在該 IP，若有則檢查 MAC 是否為 "N/A"

    for idx, (existing_ip, existing_mac, existing_computer_name, existing_status) in enumerate(scan_results):

        if existing_ip == ip:

            # 如果已有的 MAC 為 "N/A"，且當前 MAC 地址有效，則更新 MAC

            if existing_mac == "N/A" and mac != "N/A":

                scan_results[idx] = (ip, mac, computer_name, status)

            return  # 找到並處理過就退出

    # 若是新 IP 或無重複的情況，直接加入

    scan_results.append((ip, mac, computer_name, status))

def update_tree(data=None):

    if data is None:

        data = scan_results

    else:

        data = sorted(data, key=lambda x: ipaddress.IPv4Address(x[0]))

    def update_ui():

        # 清空現有顯示

        tree.delete(*tree.get_children())

        for index, (ip, mac_address, computer_name, status) in enumerate(data):

            tree.insert("", tk.END, values=(index + 1, ip, mac_address, computer_name, status))  # 加入序號

    root.after(0, update_ui)

# 檢查網段格式是否有效

def is_valid_network_prefix(prefix):

    """

    檢查網段格式是否有效，可以是單純的 x.x.x. 或 CIDR 格式 x.x.x.x/xx。

    例如: 192.168.1. 或 192.168.1.0/24

    """

    # 如果網段以 "." 結尾，將其補充為最後一段 0

    if prefix.endswith('.'):

        prefix = prefix + '0'

    # 正則表達式匹配 x.x.x.x 或 x.x.x.x/xx（CIDR 格式）

    network_regex = r'^(\d{1,3}\.){3}\d{1,3}(/(\d|[12][0-9]|3[0-2]))?$'

    # 檢查網段格式是否符合正則表達式

    if re.match(network_regex, prefix):

        # 檢查每個 IP 部分是否在 0 到 255 之間

        try:

            parts = prefix.split('/')

            # 檢查 IP 地址部分

            ip_parts = parts[0].split('.')

            if len(ip_parts) == 4 and all(0 <= int(part) <= 255 for part in ip_parts):

                # 如果有 CIDR 部分，檢查是否是 0-32 之間的數字

                if len(parts) > 1:

                    subnet_mask = int(parts[1])

                    if 0 <= subnet_mask <= 32:

                        return True

                else:

                    return True  # 沒有 CIDR 部分也有效

        except ValueError:

            return False

    return False

# 更新進度條的函數

def update_progress_bar(value):

    progress_bar['value'] = value

    root.update_idletasks()

# 使用 ThreadPoolExecutor 進行多執行緒掃描

def scan_network():

    global scan_count

    net_prefix = network_entry.get()  # 從使用者輸入的 Entry 中獲取網段

    if not net_prefix or not is_valid_network_prefix(net_prefix):

        logging.error("請先輸入有效的網段！")

        messagebox.showerror("錯誤", "請先輸入有效的網段！")

        return

    tree.delete(*tree.get_children())  # 清空 Treeview 資料

    scan_results.clear()  # 清除之前的掃描結果

    scan_count = 0  # 重置掃描進度

    # 執行 ARP 掃描

    arp_scan(net_prefix)

    update_tree()  # 顯示 ARP 掃描結果

    # 端口掃描：為每個 IP 啟動執行緒來掃描

    with ThreadPoolExecutor(max_workers=20) as executor:  # 設定最大工作執行緒數

        future_to_ip = {executor.submit(threaded_port_scan, f"{net_prefix}{i}"): i for i in range(1, 255)}

        for future in future_to_ip:

            future.result()  # 確保每個任務的結果都被獲得

            # 更新進度條

            with lock:

                scan_count += 1  # 每完成一個掃描，進行一次累加

                progress = (scan_count / total_count) * 100

                root.after(0, update_progress_bar, progress)  # 使用 after 將更新交給主執行緒

    # 確保最終結果更新

    logging.info(f"掃描完成，共找到 {len(scan_results)} 台設備")

    update_tree(sorted(scan_results, key=lambda x: ipaddress.IPv4Address(x[0])))

# 發送遠端開機封包

def wake_on_lan():

    selected_item = tree.selection()

    if selected_item:

        mac_address = tree.item(selected_item)["values"][1]  # 確保正確取用 MAC 位址

        if mac_address != "N/A":

            send_magic_packet(mac_address)

            logging.info(f"發送遠端開機封包至 MAC: {mac_address}")

        else:

            logging.warning("選取的電腦沒有有效的 MAC 位址。")

# 用來讓使用者輸入網段的輸入框和標籤

network_label = tk.Label(root, text="輸入掃描網段 (例如: 192.168.1.)")

network_label.pack(padx=10, pady=5)

# 預設為本機的網段

local_ip_prefix = get_local_ip_prefix()

network_entry = tk.Entry(root, width=20)

network_entry.pack(padx=10, pady=5)

network_entry.insert(0, local_ip_prefix)  # 預設顯示本機 IP 的網段

# 顯示提示文字

alert_label = tk.Label(root, text="", fg="red")

alert_label.pack(padx=10, pady=10)

# 創建一個 Text 小部件來顯示日誌訊息

log_text = tk.Text(root, height=3, width=80)

log_text.pack(side=tk.RIGHT,padx=5, pady=5)

# 設置自定義的日誌處理器

log_handler = TkinterLogHandler(log_text)

logging.getLogger().addHandler(log_handler)

# 創建掃描按鈕

scan_button = tk.Button(root, text="掃描區網", command=lambda: threading.Thread(target=scan_network).start())

scan_button.pack(side=tk.LEFT, padx=10, pady=10)

# 創建遠端開機按鈕

wake_button = tk.Button(root, text="遠端開機", command=wake_on_lan)

wake_button.pack(side=tk.LEFT, padx=10, pady=10)

# 創建清除結果按鈕

def clear_results():

    tree.delete(*tree.get_children())

    scan_results.clear()

clear_button = tk.Button(root, text="清除結果", command=clear_results)

clear_button.pack(side=tk.LEFT, padx=10, pady=10)

# 創建進度條

progress_bar = ttk.Progressbar(root, orient="horizontal", length=200, mode="determinate")

progress_bar.pack(side=tk.BOTTOM, padx=10, pady=10)

root.mainloop()

1131104-wakeonlan-test02-12

-IP輸入還是有問題

關於 is_valid_network_prefix 函數的正則表達式和格式判斷，您可以進一步提升網段檢查的準確性。由於現在的判斷方式僅是將 IP 地址分割並檢查前三段的數值是否有效，但它沒有考慮到一些更細緻的情況（例如，CIDR 格式，或是使用不同子網掩碼的情況）。因此，根據您的需求，以下是一些改進的方式：

1. 允許檢查 CIDR 格式的網段

CIDR 格式的網段通常是以 x.x.x.x/24 這種形式呈現，其中 /24 表示子網掩碼。您可以對這種格式進行檢查。

2. 檢查 IP 地址是否有效

首先，我們仍然需要檢查 IP 地址是否符合正確的格式（x.x.x.x），然後如果有 /，我們需要確保後面的數字在 0 到 32 之間。

3. 修正函數以使用正則表達式驗證

您可以使用正則表達式來檢查網段格式是否符合標準。下面是修正後的 is_valid_network_prefix 函數，加入了對 CIDR 格式的支持以及更詳細的檢查：

修正後的 is_valid_network_prefix 函數：

python

複製程式碼

import re

import ipaddress

def is_valid_network_prefix(prefix):

    """

    檢查網段格式是否有效，可以是單純的 x.x.x. 或 CIDR 格式 x.x.x.x/xx。

    例如: 192.168.1. 或 192.168.1.0/24

    """

    # 正則表達式匹配 x.x.x.x 或 x.x.x.x/xx（CIDR 格式）

    network_regex = r'^(\d{1,3}\.){3}\d{1,3}(/(\d|[12][0-9]|3[0-2]))?$'

    # 檢查網段格式是否符合正則表達式

    if re.match(network_regex, prefix):

        # 檢查每個 IP 部分是否在 0 到 255 之間

        try:

            parts = prefix.split('/')

            # 檢查 IP 地址部分

            ip_parts = parts[0].split('.')

            if len(ip_parts) == 4 and all(0 <= int(part) <= 255 for part in ip_parts):

                # 如果有 CIDR 部分，檢查是否是 0-32 之間的數字

                if len(parts) > 1:

                    subnet_mask = int(parts[1])

                    if 0 <= subnet_mask <= 32:

                        return True

                else:

                    return True  # 沒有 CIDR 部分也有效

        except ValueError:

            return False

    return False

解釋：

1. 正則表達式：

   • ^(\d{1,3}\.){3}\d{1,3}(/(\d|[12][0-9]|3[0-2]))?$

     • ^ 和 $ 表示字符串的開始和結束，保證正確的匹配。

     • (\d{1,3}\.){3} 匹配前三個數字段並且後面接上 .（例如 192.168.1.）。

     • \d{1,3} 用來匹配每個段的數字，允許 1 到 3 位數字。

     • (/(\d|[12][0-9]|3[0-2]))? 匹配可選的 / 以及子網掩碼，子網掩碼必須是 0 到 32 之間的數字。

2. 解析 IP 地址：

   • 先檢查是否匹配正則表達式。

   • 如果匹配，則將網段部分以 / 分割，獲得 IP 地址和子網掩碼部分。

   • 檢查 IP 地址的每一段是否在 0 到 255 之間，這是確保 IP 地址有效的基本條件。

3. CIDR 子網掩碼的檢查：

   • 如果網段包含 /（即 CIDR 格式），則檢查子網掩碼是否在 0 到 32 之間。

4. 返回值：

   • 如果所有檢查通過，則返回 True，表示網段有效；否則返回 False。

測試：

這樣的 is_valid_network_prefix 函數可以處理多種情況，包括：

• 只包含 IP 地址：192.168.1.

• 包含 CIDR 子網掩碼：192.168.1.0/24

• 其他無效的格式：192.168.1.500（無效 IP 地址），192.168.1./24（無效格式）

範例：

python

複製程式碼

print(is_valid_network_prefix("192.168.1."))  # True

print(is_valid_network_prefix("192.168.1.0/24"))  # True

print(is_valid_network_prefix("192.168.1.500"))  # False

print(is_valid_network_prefix("192.168.1.0/33"))  # False

print(is_valid_network_prefix("10.0.0.0/8"))  # True

結論：

這樣的修正可以讓 is_valid_network_prefix 函數不僅能夠處理簡單的 IP 網段，還能支持 CIDR 格式的網段，並進行更嚴格的驗證。這樣不僅提高了程式的靈活性，也能避免用戶輸入錯誤的網段格式，提升使用體驗。

import tkinter as tk

from tkinter import ttk

import subprocess

import threading

import re

import socket

from wakeonlan import send_magic_packet

import ipaddress

import logging

from concurrent.futures import ThreadPoolExecutor

from tkinter import messagebox

import threading

# 全局變量來跟踪進度

scan_count = 0  # 記錄已完成的掃描數量

total_count = 254  # 總共需要掃描的 IP 數量（從 1 到 254）

# 創建自定義的日誌處理器，將日誌輸出到 Text 小部件

class TkinterLogHandler(logging.Handler):

    def __init__(self, text_widget, auto_scroll=True):

        super().__init__()

        self.text_widget = text_widget

        self.auto_scroll = auto_scroll

    def emit(self, record):

        msg = self.format(record)

        self.text_widget.insert(tk.END, msg + '\n')

        if self.auto_scroll:

            self.text_widget.yview(tk.END)  # 自動滾動到底部

# 設定日誌

logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')

# 初始化 UI 和設定

root = tk.Tk()

root.title("區域網路掃描與遠端開機工具")

root.geometry("1000x600")

# 更新 Treeview 欄位，添加「電腦名稱」

tree = ttk.Treeview(root, columns=("序號", "IP Address", "MAC Address", "Computer Name", "Status"), show="headings")

tree.heading("序號", text="序號")  # 新增序號欄位的標題

tree.heading("IP Address", text="IP 位址")

tree.heading("MAC Address", text="MAC 位址")

tree.heading("Computer Name", text="電腦名稱")  # 新增的欄位

tree.heading("Status", text="狀態")

# 設定所有欄位的資料置中

tree.column("序號", anchor="center")

tree.column("IP Address", anchor="center")

tree.column("MAC Address", anchor="center")

tree.column("Computer Name", anchor="center")

tree.column("Status", anchor="center")

tree.pack(fill=tk.BOTH, expand=True)

# 儲存掃描結果的列表

scan_results = []

# 上鎖機制確保執行緒安全

lock = threading.Lock()

# 獲取本機 IP 網段（預設網段）