whichSystem.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

from pwn import text, log
import re
import subprocess
import sys

# ─────────────────────────────────────────
#  Constantes
# ─────────────────────────────────────────
TTL_LINUX:   int = 64
TTL_WINDOWS: int = 128
TTL_PATTERN: str = r"ttl=(\d+)"


def _resolve_os(ttl: int) -> str:
    """Devuelve el nombre del SO según el valor TTL."""
    if ttl <= TTL_LINUX:
        return "Linux"
    elif ttl <= TTL_WINDOWS:
        return "Windows"
    return "Desconocido"


def system_detection(hostname: str) -> None:
    """
    Lanza un ping al host y determina el SO basándose en el TTL.

    Args:
        hostname: IP o nombre de host objetivo.

    Raises:
        SystemExit: Si el host es inalcanzable o el ping falla.
    """
    try:
        output: str = subprocess.check_output(
            ["ping", "-c", "1", hostname],
            text=True,
            timeout=1,
            stderr=subprocess.DEVNULL,
        )
    except subprocess.TimeoutExpired:
        log.failure(text.yellow(f"Host {hostname} inalcanzable (timeout)"))
        sys.exit(1)
    except subprocess.CalledProcessError:
        log.failure(text.yellow(f"Host {hostname} inalcanzable (ping fallido)"))
        sys.exit(1)
    except FileNotFoundError:
        log.failure(text.yellow("Comando 'ping' no encontrado en el sistema"))
        sys.exit(1)

    match = re.search(TTL_PATTERN, output, re.IGNORECASE)
    if not match:
        log.failure(text.yellow(f"No se pudo extraer el TTL de la respuesta de {hostname}"))
        sys.exit(1)

    ttl_value: int = int(match.group(1))
    os_name: str  = _resolve_os(ttl_value)

    print("")
    log.info(text.yellow(f"Operative System: {hostname} →  {os_name}  (TTL={ttl_value})"))


def _parse_args() -> str:
    """Valida y devuelve el argumento IP/hostname desde la línea de comandos."""
    if len(sys.argv) < 2:
        print("")
        log.info(text.yellow(f"Uso: python3 {sys.argv[0]} <Target Ip>"))
        sys.exit(1)
    return sys.argv[1]


if __name__ == "__main__":
    target_ip: str = _parse_args()
    system_detection(target_ip)