PART 03 — ネットワーク構成図
セグメント・VLAN・ポート/プロトコル設計で定義すべきこと

"""
ポート開放マトリクスに基づいて TCP 疎通確認を行うスクリプト。
pip install pandas openpyxl
"""
import socket
import subprocess
import platform
import pandas as pd
from dataclasses import dataclass
from datetime import datetime


@dataclass
class PortCheck:
    """疎通確認定義"""
    src_desc: str       # 送信元の説明
    dst_host: str       # 宛先ホスト名 or IPアドレス
    dst_port: int       # 宛先ポート番号
    protocol: str       # TCP / ICMP
    description: str    # 用途説明


def check_tcp_port(host: str, port: int, timeout: float = 5.0) -> tuple[bool, str]:
    """
    TCP ポート疎通確認。
    Returns: (成功フラグ, メッセージ)
    """
    try:
        sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        sock.settimeout(timeout)
        result = sock.connect_ex((host, port))
        sock.close()
        if result == 0:
            return True, "OPEN"
        else:
            return False, f"CLOSED (errno={result})"
    except socket.gaierror as e:
        return False, f"DNS ERROR: {e}"
    except Exception as e:
        return False, f"ERROR: {e}"


def check_icmp_ping(host: str) -> tuple[bool, str]:
    """
    ICMP Ping 疎通確認。
    """
    param = "-n" if platform.system() == "Windows" else "-c"
    try:
        result = subprocess.run(
            ["ping", param, "1", "-W", "2", host],
            capture_output=True, text=True, timeout=10
        )
        if result.returncode == 0:
            return True, "REACHABLE"
        else:
            return False, "UNREACHABLE"
    except Exception as e:
        return False, f"ERROR: {e}"


def run_port_checks(checks: list[PortCheck]) -> pd.DataFrame:
    """
    疎通確認リストを実行して結果DataFrameを返す。
    """
    records = []
    for chk in checks:
        if chk.protocol.upper() == "ICMP":
            ok, msg = check_icmp_ping(chk.dst_host)
        else:
            ok, msg = check_tcp_port(chk.dst_host, chk.dst_port)

        records.append({
            "送信元":    chk.src_desc,
            "宛先ホスト": chk.dst_host,
            "ポート":     chk.dst_port if chk.protocol != "ICMP" else "-",
            "プロトコル": chk.protocol,
            "結果":      "✅ OK" if ok else "❌ NG",
            "詳細":      msg,
            "用途":      chk.description,
            "確認日時":   datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
        })
    return pd.DataFrame(records)


# ── 疎通確認定義リスト（ポートマトリクスを反映） ────────────────
checks = [
    PortCheck("本サーバー", "192.168.10.11", 443,  "TCP",  "Webサーバー HTTPS"),
    PortCheck("本サーバー", "192.168.20.21", 8080, "TCP",  "APサーバー HTTP"),
    PortCheck("本サーバー", "192.168.30.31", 5432, "TCP",  "DBサーバー PostgreSQL"),
    PortCheck("本サーバー", "192.168.30.32", 6379, "TCP",  "Redis"),
    PortCheck("本サーバー", "192.168.10.11", 0,    "ICMP", "Webサーバー Ping"),
]

if __name__ == "__main__":
    df = run_port_checks(checks)
    print(df.to_string(index=False))

    # Excelに出力
    df.to_excel("port_check_result.xlsx", index=False)
    print("\nport_check_result.xlsx に出力しました")

    # NG件数を表示
    ng_count = (df["結果"] == "❌ NG").sum()
    print(f"\n疎通確認結果: {len(df)}件中 {ng_count}件NG")