Feat : Tailscale

This commit is contained in:
decolua
2026-04-11 11:36:33 +07:00
parent 838d9a7a04
commit ed17a8ffac
18 changed files with 1433 additions and 388 deletions
+442
View File
@@ -0,0 +1,442 @@
import fs from "fs";
import path from "path";
import os from "os";
import { execSync, spawn } from "child_process";
import { execWithPassword, executeElevatedPowerShell } from "@/mitm/dns/dnsConfig";
import { saveTailscalePid, loadTailscalePid, clearTailscalePid } from "./state.js";
const BIN_DIR = path.join(os.homedir(), ".9router", "bin");
const IS_MAC = os.platform() === "darwin";
const IS_LINUX = os.platform() === "linux";
const IS_WINDOWS = os.platform() === "win32";
const TAILSCALE_BIN = path.join(BIN_DIR, IS_WINDOWS ? "tailscale.exe" : "tailscale");
// Custom socket for userspace-networking mode (no root required)
const TAILSCALE_DIR = path.join(os.homedir(), ".9router", "tailscale");
export const TAILSCALE_SOCKET = path.join(TAILSCALE_DIR, "tailscaled.sock");
const SOCKET_FLAG = IS_WINDOWS ? [] : ["--socket", TAILSCALE_SOCKET];
// Prefer system tailscale, fallback to local bin
function getTailscaleBin() {
try {
const systemPath = execSync("which tailscale 2>/dev/null || where tailscale 2>nul", { encoding: "utf8" }).trim();
if (systemPath) return systemPath;
} catch (e) { /* not in PATH */ }
if (fs.existsSync(TAILSCALE_BIN)) return TAILSCALE_BIN;
return null;
}
export function isTailscaleInstalled() {
return getTailscaleBin() !== null;
}
/** Build tailscale CLI args with custom socket (no root needed) */
function tsArgs(...args) {
return [...SOCKET_FLAG, ...args];
}
export function isTailscaleLoggedIn() {
const bin = getTailscaleBin();
if (!bin) return false;
try {
const out = execSync(`"${bin}" ${SOCKET_FLAG.join(" ")} status --json`, {
encoding: "utf8",
env: { ...process.env, PATH: EXTENDED_PATH },
timeout: 5000
});
const json = JSON.parse(out);
// BackendState "Running" means fully logged in and connected
return json.BackendState === "Running";
} catch (e) {
return false;
}
}
export function isTailscaleRunning() {
const bin = getTailscaleBin();
if (!bin) return false;
try {
const out = execSync(`"${bin}" ${SOCKET_FLAG.join(" ")} funnel status --json 2>/dev/null`, { encoding: "utf8" });
const json = JSON.parse(out);
return Object.keys(json.AllowFunnel || {}).length > 0;
} catch (e) {
return false;
}
}
/** Get funnel URL from tailscale status */
export function getTailscaleFunnelUrl(port) {
const bin = getTailscaleBin();
if (!bin) return null;
try {
const out = execSync(`"${bin}" ${SOCKET_FLAG.join(" ")} status --json`, { encoding: "utf8" });
const json = JSON.parse(out);
const dnsName = json.Self?.DNSName?.replace(/\.$/, "");
if (dnsName) return `https://${dnsName}`;
} catch (e) { /* ignore */ }
return null;
}
/**
* Install tailscale.
* - macOS + brew: brew install tailscale (no sudo needed)
* - macOS no brew: download .pkg then sudo installer -pkg
* - Linux: fetch install.sh, pipe to sudo -S sh via stdin
* - Windows: download MSI via UAC-elevated PowerShell
*/
export async function installTailscale(sudoPassword, hostname, onProgress) {
const log = onProgress || (() => {});
if (IS_WINDOWS) await installTailscaleWindows(log);
else if (IS_MAC) await installTailscaleMac(sudoPassword, log);
else await installTailscaleLinux(sudoPassword, log);
log("Starting daemon...");
await startDaemonWithPassword(sudoPassword);
log("Logging in...");
return startLogin(hostname);
}
const EXTENDED_PATH = `/usr/local/bin:/opt/homebrew/bin:/usr/bin:/bin:${process.env.PATH || ""}`;
function hasBrew() {
try { execSync("which brew", { stdio: "ignore", env: { ...process.env, PATH: EXTENDED_PATH } }); return true; } catch { return false; }
}
async function installTailscaleMac(sudoPassword, log) {
if (hasBrew()) {
log("Installing via Homebrew...");
await new Promise((resolve, reject) => {
const child = spawn("brew", ["install", "tailscale"], {
stdio: ["ignore", "pipe", "pipe"],
env: { ...process.env, PATH: EXTENDED_PATH }
});
child.stdout.on("data", (d) => {
const line = d.toString().trim();
if (line) log(line);
});
child.stderr.on("data", (d) => {
const line = d.toString().trim();
if (line) log(line);
});
child.on("close", (c) => {
if (c === 0) resolve();
else reject(new Error(`brew install failed (code ${c})`));
});
child.on("error", reject);
});
return;
}
// No brew: download .pkg and install via sudo installer
const pkgUrl = "https://pkgs.tailscale.com/stable/tailscale-latest.pkg";
const pkgPath = path.join(os.tmpdir(), "tailscale.pkg");
log("Downloading Tailscale package...");
await new Promise((resolve, reject) => {
const child = spawn("curl", ["-fL", "--progress-bar", pkgUrl, "-o", pkgPath], {
stdio: ["ignore", "pipe", "pipe"]
});
child.stderr.on("data", (d) => {
const line = d.toString().trim();
if (line) log(line);
});
child.on("close", (c) => {
if (c === 0) resolve();
else reject(new Error("Download failed"));
});
child.on("error", reject);
});
log("Installing package...");
await new Promise((resolve, reject) => {
const child = spawn("sudo", ["-S", "installer", "-pkg", pkgPath, "-target", "/"], {
stdio: ["pipe", "pipe", "pipe"]
});
let stderr = "";
child.stderr.on("data", (d) => { stderr += d.toString(); });
child.stdout.on("data", (d) => {
const line = d.toString().trim();
if (line) log(line);
});
child.on("close", (c) => {
try { execSync(`rm -f ${pkgPath}`, { stdio: "ignore" }); } catch { /* ignore */ }
if (c === 0) resolve();
else {
const msg = (stderr.includes("incorrect password") || stderr.includes("Sorry"))
? "Wrong sudo password"
: stderr || `Exit code ${c}`;
reject(new Error(msg));
}
});
child.on("error", reject);
child.stdin.write(`${sudoPassword}\n`);
child.stdin.end();
});
}
async function installTailscaleLinux(sudoPassword, log) {
log("Downloading install script...");
return new Promise((resolve, reject) => {
const curlChild = spawn("curl", ["-fsSL", "https://tailscale.com/install.sh"], {
stdio: ["ignore", "pipe", "pipe"]
});
let scriptContent = "";
let curlErr = "";
curlChild.stdout.on("data", (d) => { scriptContent += d.toString(); });
curlChild.stderr.on("data", (d) => { curlErr += d.toString(); });
curlChild.on("exit", (code) => {
if (code !== 0) return reject(new Error(`Failed to download install script: ${curlErr}`));
log("Running install script...");
const child = spawn("sudo", ["-S", "sh"], { stdio: ["pipe", "pipe", "pipe"] });
let stderr = "";
child.stdout.on("data", (d) => {
const line = d.toString().trim();
if (line) log(line);
});
child.stderr.on("data", (d) => { stderr += d.toString(); });
child.on("close", (c) => {
if (c === 0) resolve();
else {
const msg = (stderr.includes("incorrect password") || stderr.includes("Sorry"))
? "Wrong sudo password"
: stderr || `Exit code ${c}`;
reject(new Error(msg));
}
});
child.on("error", reject);
child.stdin.write(`${sudoPassword}\n`);
child.stdin.write(scriptContent);
child.stdin.end();
});
curlChild.on("error", reject);
});
}
async function installTailscaleWindows(log) {
const msiUrl = "https://pkgs.tailscale.com/stable/tailscale-setup-latest-amd64.msi";
const msiPath = path.join(os.tmpdir(), "tailscale-setup.msi");
const psScriptPath = path.join(os.tmpdir(), `tailscale-install-${Date.now()}.ps1`);
log("Downloading Tailscale installer...");
const psScript = [
`Invoke-WebRequest -Uri '${msiUrl}' -OutFile '${msiPath}'`,
`Start-Process msiexec.exe -ArgumentList '/i','${msiPath}','/quiet','/norestart' -Wait`,
`Remove-Item '${msiPath}' -Force -ErrorAction SilentlyContinue`,
].join("\n");
fs.writeFileSync(psScriptPath, psScript, "utf8");
log("Installing (UAC prompt may appear)...");
await executeElevatedPowerShell(psScriptPath, 120000);
}
/** Start tailscaled with sudo (TUN mode required for Funnel) */
export async function startDaemonWithPassword(sudoPassword) {
if (IS_WINDOWS) return;
// Check if daemon already responds
try {
const bin = getTailscaleBin() || "tailscale";
execSync(`"${bin}" ${SOCKET_FLAG.join(" ")} status --json`, {
stdio: "ignore",
env: { ...process.env, PATH: EXTENDED_PATH },
timeout: 3000
});
return; // Already running
} catch { /* not running, start it */ }
// Ensure config dir exists
if (!fs.existsSync(TAILSCALE_DIR)) fs.mkdirSync(TAILSCALE_DIR, { recursive: true });
// tailscaled requires root for TUN (needed for Funnel)
const tailscaledBin = IS_MAC ? "/usr/local/bin/tailscaled" : "tailscaled";
const daemonCmd = `${tailscaledBin} --socket=${TAILSCALE_SOCKET} --statedir=${TAILSCALE_DIR}`;
// Start via sudo in background (nohup keeps it alive)
await execWithPassword(`nohup ${daemonCmd} > /dev/null 2>&1 &`, sudoPassword || "");
// Wait for daemon to be ready
await new Promise((r) => setTimeout(r, 3000));
}
/** Best-effort: ensure daemon running (used for login flow) */
function ensureDaemon() {
startDaemonWithPassword("").catch(() => {});
}
/**
* Run `tailscale up` and capture the auth URL for browser login.
* Resolves with { authUrl } or { alreadyLoggedIn: true }.
*/
export function startLogin(hostname) {
const bin = getTailscaleBin();
if (!bin) return Promise.reject(new Error("Tailscale not installed"));
return new Promise((resolve, reject) => {
// Ensure daemon is running (best-effort, no sudo)
ensureDaemon();
// Check if already logged in
if (isTailscaleLoggedIn()) {
resolve({ alreadyLoggedIn: true });
return;
}
// Spawn detached so process survives API request lifecycle
const args = tsArgs("up", "--accept-routes");
if (hostname) args.push(`--hostname=${hostname}`);
const child = spawn(bin, args, {
stdio: ["ignore", "pipe", "pipe"],
detached: true
});
let resolved = false;
let output = "";
const timeout = setTimeout(() => {
if (resolved) return;
resolved = true;
// Don't kill — let tailscale up keep waiting for auth
child.unref();
const url = parseAuthUrl(output);
if (url) resolve({ authUrl: url });
else reject(new Error("tailscale up timed out without auth URL"));
}, 15000);
const parseAuthUrl = (text) => {
const match = text.match(/https:\/\/login\.tailscale\.com\/a\/[a-zA-Z0-9]+/);
return match ? match[0] : null;
};
const handleData = (data) => {
output += data.toString();
const url = parseAuthUrl(output);
if (url && !resolved) {
resolved = true;
clearTimeout(timeout);
// Keep process alive — unref so it doesn't block Node exit
child.unref();
resolve({ authUrl: url });
}
};
child.stdout.on("data", handleData);
child.stderr.on("data", handleData);
child.on("error", (err) => {
if (resolved) return;
resolved = true;
clearTimeout(timeout);
reject(err);
});
child.on("exit", (code) => {
if (resolved) return;
resolved = true;
clearTimeout(timeout);
const url = parseAuthUrl(output);
if (url) resolve({ authUrl: url });
else if (isTailscaleLoggedIn()) resolve({ alreadyLoggedIn: true });
else reject(new Error(`tailscale up exited with code ${code}`));
});
});
}
/** Start tailscale funnel for the given port */
export async function startFunnel(port) {
const bin = getTailscaleBin();
if (!bin) throw new Error("Tailscale not installed");
// Reset any existing funnel
try { execSync(`"${bin}" ${SOCKET_FLAG.join(" ")} funnel --bg reset`, { stdio: "ignore" }); } catch (e) { /* ignore */ }
return new Promise((resolve, reject) => {
const child = spawn(bin, tsArgs("funnel", "--bg", `${port}`), {
stdio: ["ignore", "pipe", "pipe"]
});
let resolved = false;
let output = "";
const timeout = setTimeout(() => {
if (resolved) return;
resolved = true;
// --bg exits after setup, try status
const url = getTailscaleFunnelUrl(port);
if (url) resolve({ tunnelUrl: url });
else reject(new Error(`Tailscale funnel timed out: ${output.trim() || "no output"}`));
}, 30000);
const parseFunnelUrl = (text) =>
(text.match(/https:\/\/[a-z0-9-]+\.[a-z0-9.-]+\.ts\.net[^\s]*/i) || [])[0]?.replace(/\/$/, "") || null;
let funnelNotEnabled = false;
const handleData = (data) => {
output += data.toString();
if (output.includes("Funnel is not enabled")) funnelNotEnabled = true;
// Wait for the enable URL to arrive in a later chunk
if (funnelNotEnabled && !resolved) {
const enableMatch = output.match(/https:\/\/login\.tailscale\.com\/[^\s]+/);
if (enableMatch) {
resolved = true;
clearTimeout(timeout);
child.kill();
resolve({ funnelNotEnabled: true, enableUrl: enableMatch[0] });
return;
}
}
const url = parseFunnelUrl(output);
if (url && !resolved) {
resolved = true;
clearTimeout(timeout);
resolve({ tunnelUrl: url });
}
};
child.stdout.on("data", handleData);
child.stderr.on("data", handleData);
child.on("exit", (code) => {
if (resolved) return;
resolved = true;
clearTimeout(timeout);
const url = parseFunnelUrl(output) || getTailscaleFunnelUrl(port);
if (url) resolve({ tunnelUrl: url });
else reject(new Error(`tailscale funnel failed (code ${code}): ${output.trim()}`));
});
child.on("error", (err) => {
if (resolved) return;
resolved = true;
clearTimeout(timeout);
reject(err);
});
});
}
/** Stop tailscale funnel */
export function stopFunnel() {
const bin = getTailscaleBin();
if (!bin) return;
try { execSync(`"${bin}" ${SOCKET_FLAG.join(" ")} funnel --bg reset`, { stdio: "ignore" }); } catch (e) { /* ignore */ }
}
/** Kill tailscaled daemon (runs as root, needs sudo) */
export async function stopDaemon(sudoPassword) {
// Try non-sudo first
try { execSync("pkill -x tailscaled", { stdio: "ignore", timeout: 3000 }); } catch { /* ignore */ }
// Check if still alive
try { execSync("pgrep -x tailscaled", { stdio: "ignore", timeout: 2000 }); } catch { return; } // Dead, done
// Kill with sudo password
if (!IS_WINDOWS) {
try { await execWithPassword("pkill -x tailscaled", sudoPassword || ""); } catch { /* ignore */ }
}
// Cleanup socket
try { if (fs.existsSync(TAILSCALE_SOCKET)) fs.unlinkSync(TAILSCALE_SOCKET); } catch { /* ignore */ }
}