防犯・見回りのモニタリングシステムをラズパイで作る(10)－Wireguardコンフィグ生成プログラム

import os
import subprocess
from tkinter import *
from tkinter import messagebox

# 鍵生成用の関数
def generate_keys(name):
    private_key = subprocess.check_output("wg genkey", shell=True).decode().strip()
    public_key = subprocess.check_output(f"echo {private_key} | wg pubkey", shell=True).decode().strip()
    with open(f"{name}_privatekey", "w") as f:
        f.write(private_key)
    with open(f"{name}_publickey", "w") as f:
        f.write(public_key)
    return private_key, public_key

# 設定ファイル生成処理
def generate_configs():
    try:
        server_ip = entry_server_ip.get()
        server_port = entry_port.get() or "51820"
        allowed_ips = entry_allowed_ips.get()

        raspberry_prefix = entry_raspberry_prefix.get()
        raspberry_count = int(entry_raspberry_count.get() or "0")

        windows_prefix = entry_windows_prefix.get()
        windows_count = int(entry_windows_count.get() or "0")

        smartphone_prefix = entry_smartphone_prefix.get()
        smartphone_count = int(entry_smartphone_count.get() or "0")

        total_clients = raspberry_count + windows_count + smartphone_count

        # サーバー鍵生成
        server_priv, server_pub = generate_keys("server")

        # クライアントIPベース
        client_ip_base = "10.0.0."
        current_ip_suffix = 2

        peer_entries = ""
        client_entries = []

        # クライアント生成ヘルパー
        def create_client_configs(prefix, count):
            nonlocal current_ip_suffix, peer_entries
            for i in range(count):
                name = f"{prefix}{i+1}"
                priv, pub = generate_keys(name)
                client_ip = f"{client_ip_base}{current_ip_suffix}/32"
                client_ip_no_mask = f"{client_ip_base}{current_ip_suffix}"
                current_ip_suffix += 1

                # クライアント用conf
                client_conf = f"""[Interface]
PrivateKey = {priv}
Address = {client_ip}
DNS = 1.1.1.1

[Peer]
PublicKey = {server_pub}
Endpoint = {server_ip}:{server_port}
AllowedIPs = {allowed_ips}
PersistentKeepalive = 25
"""
                with open(f"{name}.conf", "w") as f:
                    f.write(client_conf)

                # サーバー用Peer追記
                peer_entries += f"""
[Peer]
PublicKey = {pub}
AllowedIPs = {client_ip}
"""
        
        # 各クライアント生成
        create_client_configs(raspberry_prefix, raspberry_count)
        create_client_configs(windows_prefix, windows_count)
        create_client_configs(smartphone_prefix, smartphone_count)

        # サーバー設定ファイル生成
        server_conf = f"""[Interface]
PrivateKey = {server_priv}
Address = 10.0.0.1/24
ListenPort = {server_port}
PostUp = iptables -A FORWARD -i %i -j ACCEPT; iptables -t nat -A POSTROUTING -o eth0 -j MASQUERADE
PostDown = iptables -D FORWARD -i %i -j ACCEPT; iptables -t nat -D POSTROUTING -o eth0 -j MASQUERADE
{peer_entries}
"""

        with open("wg0.conf", "w") as f:
            f.write(server_conf)

        messagebox.showinfo("完了", "すべてのコンフィグファイルを生成しました。")

    except Exception as e:
        messagebox.showerror("エラー", str(e))

# GUI作成
root = Tk()
root.title("WireGuard 設定ファイル生成")

Label(root, text="WireGuardサーバのグローバルIPアドレス").grid(row=0, sticky=W)
entry_server_ip = Entry(root, width=30)
entry_server_ip.grid(row=0, column=1)

Label(root, text="WireGuard通信ポート番号（デフォルト51820）").grid(row=1, sticky=W)
entry_port = Entry(root, width=30)
entry_port.insert(0, "51820")
entry_port.grid(row=1, column=1)

Label(root, text="ラズパイ クライアントファイル名接頭辞").grid(row=2, sticky=W)
entry_raspberry_prefix = Entry(root, width=30)
entry_raspberry_prefix.grid(row=2, column=1)
Label(root, text="ラズパイ クライアント数").grid(row=2, column=2)
entry_raspberry_count = Entry(root, width=5)
entry_raspberry_count.grid(row=2, column=3)

Label(root, text="Windows クライアントファイル名接頭辞").grid(row=3, sticky=W)
entry_windows_prefix = Entry(root, width=30)
entry_windows_prefix.grid(row=3, column=1)
Label(root, text="Windows クライアント数").grid(row=3, column=2)
entry_windows_count = Entry(root, width=5)
entry_windows_count.grid(row=3, column=3)

Label(root, text="スマホ クライアントファイル名接頭辞").grid(row=4, sticky=W)
entry_smartphone_prefix = Entry(root, width=30)
entry_smartphone_prefix.grid(row=4, column=1)
Label(root, text="スマホ クライアント数").grid(row=4, column=2)
entry_smartphone_count = Entry(root, width=5)
entry_smartphone_count.grid(row=4, column=3)

Label(root, text="クライアント用 AllowedIPs（例：192.168.0.0/24）").grid(row=5, sticky=W)
entry_allowed_ips = Entry(root, width=30)
entry_allowed_ips.grid(row=5, column=1)

Button(root, text="コンフィグ生成", command=generate_configs).grid(row=6, column=1, pady=10)

root.mainloop()