Files
WGSecure/app/service/helper_service.py
T
tuxgyverandClaude Opus 5 e520e30716 feat(update): mise à jour servie par le serveur, depuis l'intérieur du tunnel
Les binaires ne peuvent pas venir du dépôt Gitea : il est privé, et un jeton
embarqué dans une application distribuée est un jeton donné à tous ses
utilisateurs. Ils viennent donc du serveur WGSecure, où le contrôle d'accès
existe déjà — l'API n'est joignable que par le tunnel, et on y parvient parce
que sa clé WireGuard est déclarée. Aucun secret à embarquer.

Le serveur expose GET /api/version, qui rend la version publiée et l'empreinte
de chaque artefact, et /api/version/download/{clé} qui les sert. Le manifeste
est déposé par `make publish-updates`, avec les condensats calculés une fois à
la publication : les recalculer à chaque appel bloquerait l'API sur 125 Mo.
`ANNOUNCE=` permet d'annoncer les artefacts sous un autre numéro, pour exercer
la chaîne sans compiler une seconde version.

Rien ne s'installe sans un clic. La vérification est automatique — au montage
du tunnel puis toutes les demi-heures, car ne la faire qu'au montage laissait
un serveur momentanément injoignable annuler toute proposition pour la session
— mais le téléchargement attend le bouton. Cette application monte un VPN, et
une version défaillante qui se propagerait seule couperait l'accès d'un parc
entier sans que personne ne l'ait demandé.

L'artefact est vérifié contre son empreinte SHA-256 avant d'être mis en place,
et le fichier temporaire est détruit à la moindre anomalie. Le canal est déjà
authentifié par WireGuard ; le condensat couvre ce qu'il ne couvre pas — un
téléchargement tronqué, un disque plein, un artefact mal publié.

Sous Linux, le binaire est remplacé par renommage, sur le même système de
fichiers que sa destination. Un déplacement depuis /tmp se rabattait sur une
copie, donc sur une écriture dans l'exécutable en cours, que le noyau refuse
(ETXTBSY). Un renommage ne touche qu'une entrée de répertoire : l'ancien inode
reste vivant pour le processus, qui continue jusqu'à sa fermeture — et
l'application propose désormais de redémarrer plutôt que de le laisser deviner.

Sous Windows, l'installation passe par le service, qui tourne en LocalSystem et
écrit dans Program Files sans invite d'élévation. La commande `apply_update` ne
prend aucun paramètre : le service lit le manifeste, télécharge et vérifie
lui-même. Lui passer un fichier déjà téléchargé aurait donné à tout compte du
poste — le pipe est ouvert aux utilisateurs interactifs — le moyen de faire
exécuter ce qu'il veut avec les privilèges du système.

L'installeur orchestre le remplacement, ce qu'il ne faisait pas : les deux
exécutables à remplacer tournent au moment de la mise à jour, et Windows
verrouille l'image d'un processus vivant. Il ferme donc l'application par le
gestionnaire de redémarrage, arrête les tunnels et le service avant la copie —
en attendant la libération réelle du fichier, `sc stop` rendant la main avant
la fin de l'arrêt — et garde `restartreplace` en secours. Le service, lui,
lance l'installeur détaché et ne l'attend pas : il lui demande de se remplacer
lui-même.

Durcissement du service au passage. Le premier appelant devient propriétaire et
son SID est retenu ; ensuite seuls ce compte et les administrateurs sont
servis. Le pipe étant ouvert à tout utilisateur interactif, n'importe quel
compte du poste pouvait jusqu'ici couper le tunnel d'un autre ou poser une
règle NRPT valable pour toute la machine. Le contenu d'un `.conf` est également
vérifié — seul son chemin l'était — et les directives exécutables y sont
refusées.

Corrige enfin un défaut de compilation : Linux et Windows partageaient le
répertoire de travail de PyInstaller, que `--clean` vide au démarrage. Lancées
à la suite, les deux cibles effaçaient mutuellement leurs fichiers
intermédiaires, produisant un .exe gonflé de 44 Mo et un binaire Linux tronqué
dont l'archive ne se décompressait plus.

Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>
2026-09-08 14:33:52 +02:00

648 lines
27 KiB
Python

"""Service Windows privilégié pour WGSecure.
Installe, démarre, arrête et désinstalle le tunnel WireGuard, pose/retire les règles NRPT
(split-DNS) et lit l'état du tunnel (`wg show`) sur demande de l'app — non
privilégiée — pour éviter une invite d'authentification à chaque connexion
et déconnexion. Tourne en LocalSystem.
Exécutable séparé (`wgsecure-helper.exe`, build PyInstaller dédié) : jamais
importé par l'app principale (`wgsecure.exe`), qui lui parle uniquement via
le pipe nommé (`app/utils/helper_ipc.py`, ctypes pur, sans dépendance à
pywin32).
Invariant de sécurité central : le pipe reste accessible à tout process
d'un utilisateur local, pas seulement WGSecure — donc SEPT commandes fixes
seulement, jamais de script/commande arbitraire. Chaque commande valide
strictement ses paramètres avant d'exécuter quoi que ce soit.
"""
from __future__ import annotations
import ipaddress
import hashlib
import json
import logging
import os
import re
import subprocess
import sys
import tempfile
import urllib.error
import urllib.request
from logging.handlers import RotatingFileHandler
import pywintypes
import servicemanager
import win32api
import win32con
import win32event
import win32file
import win32pipe
import win32security
import win32service
import win32serviceutil
from app.core.dns import _nrpt_add_split_dns_ps, _nrpt_remove_split_dns_ps
from app.utils.platform_utils import wg_exe, wireguard_exe
_PIPE_NAME = r"\\.\pipe\WGSecureHelper"
# SYSTEM + Administrateurs : contrôle total. Utilisateurs interactifs
# locaux (IU) : lecture/écriture seulement (se connecter, envoyer une
# requête, lire la réponse) — jamais anonyme, jamais réseau (un pipe nommé
# n'est de toute façon jamais accessible à distance sans partage explicite,
# absent ici).
_PIPE_SDDL = "D:(A;;GA;;;SY)(A;;GA;;;BA)(A;;GRGW;;;IU)"
_INTERFACE_NAME_RE = re.compile(r"^[A-Za-z0-9_-]{1,15}$")
_RULE_NAME_RE = re.compile(r"^WGSecure-SplitDNS-[A-Za-z0-9_-]+$")
_HOSTNAME_LABEL = r"[A-Za-z0-9]([A-Za-z0-9-]{0,61}[A-Za-z0-9])?"
_HOSTNAME_RE = re.compile(rf"^{_HOSTNAME_LABEL}(\.{_HOSTNAME_LABEL})*$")
_MAX_DOMAINS = 20
_LOG_DIR = os.path.join(os.environ.get("PROGRAMDATA", r"C:\ProgramData"), "WGSecure")
_LOG_FILE = os.path.join(_LOG_DIR, "helper.log")
# SID du compte autorisé à piloter le tunnel, retenu au premier usage. Le pipe
# est ouvert à tout utilisateur interactif : sans cet ancrage, n'importe quel
# compte du poste pouvait arrêter le tunnel d'un autre, installer le sien, ou
# poser une règle DNS valable pour toute la machine.
_OWNER_FILE = os.path.join(_LOG_DIR, "owner.sid")
# Directives refusées dans un `.conf` : elles font exécuter des commandes par
# le service, donc en LocalSystem. `wireguard.exe` les ignore sous Windows,
# mais le fichier n'a aucune raison d'en porter — les refuser coûte une ligne
# et ferme la porte si un jour il les honore.
_CONF_FORBIDDEN = ("postup", "predown", "postdown", "preup")
# Un `.conf` WireGuard tient en quelques centaines d'octets ; au-delà, ce
# n'est pas une configuration.
_CONF_MAX_BYTES = 64 * 1024
def _get_logger() -> logging.Logger:
os.makedirs(_LOG_DIR, exist_ok=True)
logger = logging.getLogger("wgsecure-helper")
if not logger.handlers:
logger.setLevel(logging.INFO)
handler = RotatingFileHandler(_LOG_FILE, maxBytes=1_000_000,
backupCount=3, encoding="utf-8")
handler.setFormatter(logging.Formatter(
"%(asctime)s [%(levelname)s] %(message)s", "%Y-%m-%d %H:%M:%S"))
logger.addHandler(handler)
logger.propagate = False
return logger
class ValidationError(Exception):
pass
def _caller_sid(handle) -> str:
"""SID de l'appelant du pipe, sous forme textuelle.
Lève plutôt que de rendre une valeur par défaut : une identité qu'on ne
peut pas établir ne doit jamais valoir autorisation.
"""
try:
win32security.ImpersonateNamedPipeClient(handle)
except pywintypes.error as e:
raise ValidationError(f"Identité de l'appelant non vérifiable : {e}") from None
try:
token = win32security.OpenThreadToken(
win32api.GetCurrentThread(), win32con.TOKEN_QUERY, True)
sid, _ = win32security.GetTokenInformation(token, win32security.TokenUser)
return win32security.ConvertSidToStringSid(sid)
except pywintypes.error as e:
raise ValidationError(f"Identité de l'appelant illisible : {e}") from None
finally:
win32security.RevertToSelf()
def _caller_is_admin(handle) -> bool:
"""L'appelant est-il administrateur local ? Faux si indéterminable."""
try:
win32security.ImpersonateNamedPipeClient(handle)
except pywintypes.error:
return False
try:
admins = win32security.CreateWellKnownSid(win32security.WinBuiltinAdministratorsSid)
token = win32security.OpenThreadToken(
win32api.GetCurrentThread(), win32con.TOKEN_QUERY, True)
for sid, attrs in win32security.GetTokenInformation(token, win32security.TokenGroups):
if sid == admins and attrs & win32security.SE_GROUP_ENABLED:
return True
return False
except pywintypes.error:
return False
finally:
win32security.RevertToSelf()
def _authorize(handle, logger: logging.Logger) -> str:
"""Autorise une commande qui modifie l'état de la machine.
Le premier appelant devient le propriétaire, et son SID est retenu : sur
un poste où une seule personne configure le VPN — le cas courant — cela se
fait sans rien demander. Ensuite, seuls ce compte et les administrateurs
sont servis.
Ce n'est pas une barrière contre le propriétaire lui-même, qui reste libre
d'installer le tunnel qu'il veut : c'en est une contre *les autres comptes
du poste*, qui pouvaient jusqu'ici couper son tunnel ou détourner la
résolution de noms de toute la machine sans être administrateurs.
"""
sid = _caller_sid(handle)
try:
owner = open(_OWNER_FILE, encoding="utf-8").read().strip()
except OSError:
owner = ""
if not owner:
try:
os.makedirs(_LOG_DIR, exist_ok=True)
with open(_OWNER_FILE, "w", encoding="utf-8") as f:
f.write(sid)
logger.info("Propriétaire du helper enregistré : %s", sid)
except OSError as e:
raise ValidationError(f"Propriétaire non enregistrable : {e}") from None
return sid
if sid != owner and not _caller_is_admin(handle):
logger.warning("Commande refusée : %s n'est ni le propriétaire (%s) "
"ni administrateur", sid, owner)
raise ValidationError(
"Ce compte n'est pas autorisé à piloter le tunnel WGSecure.")
return sid
def _validate_conf_content(path: str) -> None:
"""Refuse un `.conf` qui porterait des directives exécutables.
Le chemin était vérifié, jamais le contenu : un utilisateur pouvait faire
installer par LocalSystem un tunnel dont il choisissait l'endpoint, les
clés et les routes. Le contrôle reste modeste — il ne dit pas *quel*
serveur est légitime, information que ce service n'a pas — mais il barre
l'exécution de commandes, qui est le pas de trop.
"""
try:
size = os.path.getsize(path)
except OSError as e:
raise ValidationError(f"Configuration illisible : {e}") from None
if size > _CONF_MAX_BYTES:
raise ValidationError(f"Configuration anormalement volumineuse ({size} octets)")
try:
with open(path, encoding="utf-8", errors="replace") as f:
lignes = f.read().splitlines()
except OSError as e:
raise ValidationError(f"Configuration illisible : {e}") from None
for ligne in lignes:
cle = ligne.split("=", 1)[0].strip().lower()
if cle in _CONF_FORBIDDEN:
raise ValidationError(
f"Directive « {cle} » interdite dans une configuration "
f"installée par le service : elle ferait exécuter des "
f"commandes avec les privilèges du système.")
if not any(l.strip().lower() == "[interface]" for l in lignes):
raise ValidationError("Configuration sans section [Interface]")
def _validate_interface_name(name) -> str:
if not isinstance(name, str) or not _INTERFACE_NAME_RE.match(name):
raise ValidationError(f"Nom d'interface invalide : {name!r}")
return name
def _validate_rule_name(name) -> str:
if not isinstance(name, str) or not _RULE_NAME_RE.match(name):
raise ValidationError(f"Nom de règle NRPT invalide : {name!r}")
return name
def _validate_domains(domains) -> list[str]:
if not isinstance(domains, list) or not domains or len(domains) > _MAX_DOMAINS:
raise ValidationError("Liste de domaines invalide")
out = []
for d in domains:
if not isinstance(d, str) or not _HOSTNAME_RE.match(d):
raise ValidationError(f"Domaine invalide : {d!r}")
out.append(d)
return out
def _validate_server(server) -> str:
if not isinstance(server, str):
raise ValidationError("Serveur invalide")
try:
ipaddress.ip_address(server)
except ValueError:
raise ValidationError(f"Adresse IP invalide : {server!r}") from None
return server
def _validate_config_path(config_path, name: str, handle) -> str:
"""Valide le .conf que l'appelant demande d'installer.
Le chemin ne peut pas être comparé à un dossier fixe : l'app écrit sa
config dans le `%APPDATA%` de *l'utilisateur* (`get_client_config_path`
dans app/core/wireguard.py), or ce service tourne en LocalSystem — son
propre `%APPDATA%` est celui de SYSTEM, pas celui de l'appelant, et il
n'a aucun moyen fiable de recalculer celui-ci.
Le contrôle réel est donc l'usurpation d'identité du client du pipe :
le fichier doit être ouvrable *par l'appelant lui-même*. Un utilisateur
ne peut ainsi jamais faire lire à SYSTEM un fichier auquel il n'a pas
déjà accès. S'y ajoutent des contrôles structurels bon marché : chemin
absolu, sans remontée de répertoire, nommé exactement `<name>.conf`
dans un dossier `WGSecure`.
"""
if not isinstance(config_path, str) or not config_path:
raise ValidationError("Chemin de configuration invalide")
path = os.path.abspath(config_path)
if ".." in path.replace("/", "\\").split("\\"):
raise ValidationError(f"Chemin de configuration suspect : {config_path!r}")
if os.path.basename(path).lower() != f"{name}.conf".lower():
raise ValidationError(
f"Le fichier doit s'appeler {name}.conf, reçu : {os.path.basename(path)!r}")
if os.path.basename(os.path.dirname(path)).lower() != "wgsecure":
raise ValidationError(
"Le fichier doit se trouver dans un dossier WGSecure, reçu : "
f"{os.path.dirname(path)!r}")
# Ouverture sous l'identité de l'appelant : échoue si le fichier ne lui
# appartient pas / ne lui est pas lisible. Fail closed — si
# l'usurpation elle-même échoue, on refuse.
try:
win32security.ImpersonateNamedPipeClient(handle)
except pywintypes.error as e:
raise ValidationError(f"Identité de l'appelant non vérifiable : {e}") from None
try:
with open(path, "rb"):
pass
except OSError as e:
raise ValidationError(
f"Fichier illisible pour l'appelant : {config_path!r} ({e})") from None
finally:
win32security.RevertToSelf()
return path
def _run(cmd: list[str], timeout: int = 30) -> tuple[int, str]:
"""Exécution directe : le service tourne déjà en SYSTEM, aucune
élévation à demander (contrairement à platform_utils.run_privileged,
conçu pour un process non privilégié)."""
try:
result = subprocess.run(cmd, capture_output=True, timeout=timeout,
encoding="utf-8", errors="replace")
return result.returncode, (result.stderr.strip() or result.stdout.strip())
except subprocess.TimeoutExpired:
return -1, "Timeout"
except OSError as e:
return -1, str(e)
def _handle_install_tunnel(req: dict, handle, logger) -> dict:
_authorize(handle, logger)
name = _validate_interface_name(req.get("name"))
config_path = _validate_config_path(req.get("config_path"), name, handle)
_validate_conf_content(config_path)
code, msg = _run([wireguard_exe(), "/installtunnelservice", config_path], timeout=90)
return {"ok": code == 0, "message": msg}
def _handle_uninstall_tunnel(req: dict, handle, logger) -> dict:
_authorize(handle, logger)
name = _validate_interface_name(req.get("name"))
code, msg = _run([wireguard_exe(), "/uninstalltunnelservice", name], timeout=60)
return {"ok": code == 0, "message": msg}
def _handle_start_tunnel_service(req: dict, handle, logger) -> dict:
"""Démarre le service d'un tunnel déjà installé.
Complète install/uninstall : dans le fonctionnement courant le tunnel
est installé une fois, puis seulement démarré et arrêté. Le nom est
validé, et le préfixe « WireGuardTunnel$ » est ajouté ici — l'appelant
ne choisit donc jamais quel service piloter.
"""
_authorize(handle, logger)
name = _validate_interface_name(req.get("name"))
code, msg = _run(["sc.exe", "start", f"WireGuardTunnel${name}"], timeout=45)
return {"ok": code == 0, "message": msg}
def _handle_stop_tunnel_service(req: dict, handle, logger) -> dict:
_authorize(handle, logger)
name = _validate_interface_name(req.get("name"))
code, msg = _run(["sc.exe", "stop", f"WireGuardTunnel${name}"], timeout=45)
return {"ok": code == 0, "message": msg}
def _handle_set_split_dns(req: dict, handle, logger) -> dict:
_authorize(handle, logger)
server = _validate_server(req.get("server"))
domains = _validate_domains(req.get("domains"))
rule_name = _validate_rule_name(req.get("rule_name"))
script = _nrpt_add_split_dns_ps(server, domains, rule_name)
code, msg = _run(["powershell", "-NoProfile", "-NonInteractive", "-Command", script], timeout=20)
return {"ok": code == 0, "message": msg}
def _handle_clear_split_dns(req: dict, handle, logger) -> dict:
_authorize(handle, logger)
rule_name = _validate_rule_name(req.get("rule_name"))
script = _nrpt_remove_split_dns_ps(rule_name)
code, msg = _run(["powershell", "-NoProfile", "-NonInteractive", "-Command", script], timeout=20)
return {"ok": code == 0, "message": msg}
def _handle_get_status(req: dict, handle, logger) -> dict:
"""Lecture seule : "message" porte la sortie de `wg show`, pas une
erreur — "ok" reflète juste si la commande a pu s'exécuter.
`latest_handshakes` : ajoute le sous-argument `latest-handshakes`
(utilisé par `last_handshake_age` pour un horodatage exploitable,
plutôt que le texte humain « il y a X secondes » de `wg show` simple).
"""
name = _validate_interface_name(req.get("name"))
cmd = [wg_exe(), "show", name]
if req.get("latest_handshakes"):
cmd.append("latest-handshakes")
code, msg = _run(cmd, timeout=10)
return {"ok": code == 0, "message": msg}
def _handle_apply_update(req: dict, handle, logger) -> dict:
"""Installe la mise à jour publiée par le serveur. Sans aucun paramètre.
L'appelant ne fournit ni chemin, ni binaire, ni empreinte : il déclenche,
rien de plus. C'est la seule forme acceptable ici. Une commande qui
installerait un fichier désigné par l'appelant donnerait à tout compte du
poste — le pipe est ouvert aux utilisateurs interactifs — le moyen de
faire exécuter ce qu'il veut avec les privilèges du système. Le service
lit donc lui-même le manifeste, télécharge lui-même, et vérifie lui-même.
La confiance restante va au serveur WGSecure, joignable par le seul
tunnel : un serveur compromis diffuserait un binaire exécuté en
LocalSystem. C'est le propre de toute mise à jour automatique ; seule une
signature vérifiée contre une clé publique embarquée ici la lèverait.
"""
_authorize(handle, logger)
base = (req.get("api_base") or "").strip()
# L'URL vient de l'appelant, mais elle est bornée : uniquement http(s),
# et seule l'adresse du serveur dans le tunnel a un manifeste à servir.
# Ce qui compte est que ni le fichier ni son empreinte n'en dépendent —
# les deux viennent de la même source, lue par le service.
if not base.startswith(("http://", "https://")):
raise ValidationError("Adresse du serveur de mise à jour invalide")
try:
with urllib.request.urlopen(f"{base}/api/version", timeout=10) as r:
manifest = json.loads(r.read().decode("utf-8"))
except (urllib.error.URLError, OSError, ValueError) as e:
return {"ok": False, "message": f"Manifeste illisible : {e}"}
entry = next((a for a in manifest.get("artifacts", [])
if a.get("key") == "windows-setup"), None)
if not entry or not entry.get("sha256"):
return {"ok": False, "message": "Aucun installeur Windows publié."}
digest = hashlib.sha256()
fd, tmp = tempfile.mkstemp(prefix="wgsecure-update-", suffix=".exe",
dir=_LOG_DIR)
try:
url = f"{base}/api/version/download/windows-setup"
with urllib.request.urlopen(url, timeout=600) as r, os.fdopen(fd, "wb") as out:
while chunk := r.read(262144):
out.write(chunk)
digest.update(chunk)
if digest.hexdigest() != entry["sha256"].lower():
os.unlink(tmp)
logger.warning("Mise à jour refusée : empreinte non conforme")
return {"ok": False,
"message": "Empreinte non conforme — rien n'a été installé."}
except (urllib.error.URLError, OSError, ValueError) as e:
try:
os.unlink(tmp)
except OSError:
pass
return {"ok": False, "message": f"Téléchargement impossible : {e}"}
# L'installeur est lancé *détaché*, et surtout pas attendu : son travail
# consiste à arrêter ce service et à remplacer son binaire. L'attendre
# reviendrait à demander à un processus de survivre à sa propre mise à
# mort — `subprocess.run` ne rendrait jamais la main, et Windows refuse de
# toute façon d'écraser l'image d'un processus vivant.
#
# DETACHED_PROCESS le sort de la console du service, CREATE_NEW_PROCESS_GROUP
# de son groupe, et CREATE_BREAKAWAY_FROM_JOB de son job éventuel : sans
# cela, l'arrêt du service emporterait l'installeur avec lui.
#
# Le fichier est déposé sous %PROGRAMDATA%\WGSecure, écrit par le seul
# LocalSystem : un compte local ne peut pas le remplacer entre la
# vérification et l'exécution.
DETACHED_PROCESS = 0x00000008
CREATE_NEW_PROCESS_GROUP = 0x00000200
CREATE_BREAKAWAY_FROM_JOB = 0x01000000
flags = DETACHED_PROCESS | CREATE_NEW_PROCESS_GROUP | CREATE_BREAKAWAY_FROM_JOB
try:
subprocess.Popen(
[tmp, "/VERYSILENT", "/SUPPRESSMSGBOXES", "/NORESTART", "/NOCANCEL"],
creationflags=flags, close_fds=True,
stdin=subprocess.DEVNULL, stdout=subprocess.DEVNULL,
stderr=subprocess.DEVNULL)
except OSError as e:
try:
os.unlink(tmp)
except OSError:
pass
return {"ok": False, "message": f"Installeur non lançable : {e}"}
# Volontairement pas de `unlink` : le fichier est encore en cours
# d'exécution, et ce service va être arrêté avant d'avoir pu le nettoyer.
# L'installeur suivant écrasera ce reliquat, nommé de façon unique.
logger.info("Installeur %s lancé, arrêt du service imminent",
manifest.get("version"))
return {"ok": True,
"message": f"Version {manifest.get('version')} en cours "
f"d'installation. WGSecure va se fermer, le tunnel "
f"sera coupé, puis l'application redémarrera à jour."}
_HANDLERS = {
"install_tunnel": _handle_install_tunnel,
"uninstall_tunnel": _handle_uninstall_tunnel,
"start_tunnel_service": _handle_start_tunnel_service,
"stop_tunnel_service": _handle_stop_tunnel_service,
"set_split_dns": _handle_set_split_dns,
"clear_split_dns": _handle_clear_split_dns,
"get_status": _handle_get_status,
"apply_update": _handle_apply_update,
}
def _create_pipe_instance():
sd = win32security.ConvertStringSecurityDescriptorToSecurityDescriptor(
_PIPE_SDDL, win32security.SDDL_REVISION_1)
sa = win32security.SECURITY_ATTRIBUTES()
sa.SECURITY_DESCRIPTOR = sd
return win32pipe.CreateNamedPipe(
_PIPE_NAME,
win32pipe.PIPE_ACCESS_DUPLEX,
win32pipe.PIPE_TYPE_BYTE | win32pipe.PIPE_READMODE_BYTE | win32pipe.PIPE_WAIT,
win32pipe.PIPE_UNLIMITED_INSTANCES,
65536, 65536, 0, sa,
)
def _handle_connection(handle, logger: logging.Logger) -> None:
resp: dict
try:
# Un seul message par connexion (le client ferme après avoir lu la
# réponse) : bornée à 64 Ko pour ne jamais bloquer sur un client
# qui n'enverrait jamais de fin de ligne.
data = b""
for _ in range(16):
_, chunk = win32file.ReadFile(handle, 4096)
data += chunk
if b"\n" in data or not chunk:
break
line = data.split(b"\n", 1)[0]
req = json.loads(line.decode("utf-8"))
cmd = req.get("cmd") if isinstance(req, dict) else None
handler = _HANDLERS.get(cmd)
if handler is None:
resp = {"ok": False, "message": f"Commande inconnue : {cmd!r}"}
else:
try:
resp = handler(req, handle, logger)
except ValidationError as e:
resp = {"ok": False, "message": str(e)}
logger.info("%s -> ok=%s", cmd, resp.get("ok"))
except Exception as e:
logger.exception("Requête invalide ou erreur pipe")
resp = {"ok": False, "message": f"Erreur interne : {e}"}
try:
win32file.WriteFile(handle, (json.dumps(resp) + "\n").encode("utf-8"))
except Exception:
pass
class WGSecureHelperService(win32serviceutil.ServiceFramework):
_svc_name_ = "WGSecureHelper"
_svc_display_name_ = "WGSecure Helper"
_svc_description_ = (
"Applique pour WGSecure, sans invite d'authentification répétée, "
"l'installation du tunnel WireGuard et les règles DNS (split-DNS) "
"qui exigent normalement des droits administrateur."
)
def __init__(self, args):
win32serviceutil.ServiceFramework.__init__(self, args)
self._stop_event = win32event.CreateEvent(None, 0, 0, None)
self._running = True
def SvcStop(self):
self.ReportServiceStatus(win32service.SERVICE_STOP_PENDING)
self._running = False
# Débloque un ConnectNamedPipe() en attente en se connectant à son
# propre pipe — plus simple et plus robuste qu'un I/O recouvrant
# pour un service à trafic très faible (quelques appels par
# connexion/déconnexion de tunnel).
try:
h = win32file.CreateFile(
_PIPE_NAME, win32file.GENERIC_READ, 0, None,
win32file.OPEN_EXISTING, 0, None)
win32file.CloseHandle(h)
except pywintypes.error:
pass
win32event.SetEvent(self._stop_event)
def SvcDoRun(self):
servicemanager.LogMsg(
servicemanager.EVENTLOG_INFORMATION_TYPE,
servicemanager.PYS_SERVICE_STARTED,
(self._svc_name_, ""),
)
# La création du journal fichier est elle-même faillible (dossier
# non inscriptible…) : sans ce garde-fou, un échec ici laisserait
# zéro trace nulle part. Le journal d'événements Windows sert donc
# de canal de dernier recours (Observateur d'événements →
# Applications, source « WGSecureHelper »).
try:
logger = _get_logger()
except Exception as e:
servicemanager.LogErrorMsg(
f"WGSecureHelper : journal fichier impossible ({e})")
raise
logger.info("Service démarré")
try:
self._serve_forever(logger)
except Exception:
# Filet de sécurité final : si malgré tout ce qui suit une
# exception s'échappe encore, elle laisse au moins une trace
# exploitable ici plutôt qu'un service qui disparaît en
# silence sans rien dans le log.
logger.exception("Arrêt inattendu de la boucle principale")
servicemanager.LogErrorMsg(
"WGSecureHelper : arrêt inattendu, cf. helper.log")
finally:
logger.info("Service arrêté")
def _serve_forever(self, logger: logging.Logger) -> None:
# Chaque étape est individuellement protégée et ne peut donc jamais
# faire sortir la boucle sur une exception non prévue : un service
# censé tourner indéfiniment sans surveillance ne doit jamais
# s'arrêter à cause d'un incident sur UNE requête. Une version
# antérieure ne capturait que pywintypes.error (pas Exception en
# général) et ne protégeait pas CloseHandle individuellement dans
# le `finally` : une erreur non prévue là pouvait remonter et
# arrêter tout le service — expliquant un premier échange réussi
# (juste après le démarrage du service) puis silence total ensuite.
ERROR_PIPE_CONNECTED = 535
while self._running:
handle = None
try:
handle = _create_pipe_instance()
try:
win32pipe.ConnectNamedPipe(handle, None)
except pywintypes.error as e:
# Le client s'est connecté entre CreateNamedPipe et
# ConnectNamedPipe : cas normal (pas une erreur), pas
# rare vu la fréquence des appels — sans ce cas
# particulier, la requête était silencieusement ratée
# à chaque fois qu'il se produisait.
if e.winerror != ERROR_PIPE_CONNECTED:
raise
if not self._running:
break
_handle_connection(handle, logger)
except Exception:
logger.exception("Erreur sur une connexion cliente")
finally:
if handle:
try:
win32pipe.DisconnectNamedPipe(handle)
except Exception:
pass
try:
win32file.CloseHandle(handle)
except Exception:
pass
if __name__ == "__main__":
if len(sys.argv) == 1:
# Lancé par le gestionnaire de services (SCM), qui n'ajoute aucune
# commande à la ligne d'appel : il faut entrer explicitement dans le
# dispatcher. `HandleCommandLine()` seul, dans ce cas, se contente
# d'afficher son aide et de sortir — le service démarrait donc puis
# s'arrêtait aussitôt, sans jamais exécuter SvcDoRun (d'où l'absence
# totale de fichier journal et un pipe qui n'existait jamais, l'app
# retombant alors systématiquement sur l'élévation UAC).
servicemanager.Initialize()
servicemanager.PrepareToHostSingle(WGSecureHelperService)
servicemanager.StartServiceCtrlDispatcher()
else:
# Invocation manuelle : install / remove / start / stop / debug…
win32serviceutil.HandleCommandLine(WGSecureHelperService)