@traffic-twin/sdk 0.0.0-stage → 0.1.0

package/dist/cjs/agent.js

@@ -0,0 +1,927 @@
+"use strict";
+// The agent: Fleet lifecycle (policy.md 4, fleet-api), capture gate, asynchronous
+// sanitization and export. Fail-closed everywhere: any doubt means "do not capture".
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.Agent = void 0;
+exports.serviceNameFromEnv = serviceNameFromEnv;
+exports.optionsFromEnv = optionsFromEnv;
+const node_os_1 = require("node:os");
+const node_fs_1 = require("node:fs");
+const node_crypto_1 = require("node:crypto");
+const context_js_1 = require("./capture/context.js");
+const inbound_js_1 = require("./capture/inbound.js");
+const outbound_http_js_1 = require("./capture/outbound-http.js");
+const outbound_undici_js_1 = require("./capture/outbound-undici.js");
+const pipeline_js_1 = require("./export/pipeline.js");
+const pool_js_1 = require("./worker/pool.js");
+const node_os_2 = require("node:os");
+const exporter_js_1 = require("./export/exporter.js");
+const client_js_1 = require("./fleet/client.js");
+const store_js_1 = require("./fleet/store.js");
+const log_js_1 = require("./log.js");
+const json_js_1 = require("./sanitizer/json.js");
+const policy_js_1 = require("./sanitizer/policy.js");
+const sanitizer_js_1 = require("./sanitizer/sanitizer.js");
+const utf8_js_1 = require("./sanitizer/utf8.js");
+const version_js_1 = require("./version.js");
+const PENDING_MAX = 2000;
+const PENDING_MAX_BYTES = 64 << 20;
+const DRAIN_BUDGET_MS = 2;
+const WORKER_BATCH = 256;
+const WORKER_BATCH_MS = 8;
+const WORKER_MIN_BP = 200;
+const RAW_PATH_SVC = /^[a-z0-9][a-z0-9._-]{0,62}$/;
+function serviceNameFromEnv(env) {
+    const raw = env.TT_SERVICE_NAME || env.OTEL_SERVICE_NAME;
+    if (!raw)
+        return undefined;
+    const s = (0, utf8_js_1.asciiLower)(raw).replace(/[^a-z0-9._-]/g, '-').replace(/^[^a-z0-9]+/, '').slice(0, 63);
+    return RAW_PATH_SVC.test(s) ? s : undefined;
+}
+function optionsFromEnv(env = process.env) {
+    return {
+        enrollUrl: env.TT_ENROLL_URL,
+        enrollToken: env.TT_ENROLL_TOKEN,
+        hmacKey: env.TT_HMAC_KEY,
+        hmacKeyFile: env.TT_HMAC_KEY_FILE,
+        serviceName: serviceNameFromEnv(env),
+        stateDir: env.TT_STATE_DIR,
+        instance: env.TT_INSTANCE,
+        useCsr: env.TT_ENROLL_CSR !== '0',
+        debug: /^(1|true|yes|debug)$/i.test(env.TT_DEBUG ?? ''),
+        heartbeatIntervalMs: msEnv(env.TT_HEARTBEAT_INTERVAL_MS),
+        policyPollIntervalMs: msEnv(env.TT_POLICY_POLL_INTERVAL_MS),
+        exportFlushMs: msEnv(env.TT_EXPORT_FLUSH_MS),
+        sanitizerThreads: /^[0-9]{1,2}$/.test(env.TT_SANITIZER_THREADS ?? '') ? Number(env.TT_SANITIZER_THREADS) : undefined,
+    };
+}
+function msEnv(v) {
+    if (!v || !/^[0-9]{1,9}$/.test(v))
+        return undefined;
+    return Math.max(100, Number(v));
+}
+class Agent {
+    opts;
+    counters = {
+        exchanges_seen: 0,
+        exchanges_sampled: 0,
+        events_sent: 0,
+        events_dropped_queue: 0,
+        events_rejected: 0,
+        send_errors: 0,
+        sanitizer_errors: 0,
+    };
+    state = 'starting';
+    san = null;
+    policy = null;
+    policyKid = '';
+    rejected = null;
+    ring = null;
+    keyState = null;
+    heartbeatOk = false;
+    lastContact = 0;
+    persisted = {};
+    overrideBp = null;
+    decisionNames = new Set(['cookie', 'baggage']);
+    overhead = new Float64Array(1024);
+    overheadN = 0;
+    lastErrors = [];
+    pending = [];
+    pendingBytes = 0;
+    drainScheduled = false;
+    pool = null;
+    /** The last drain handed its work to the workers: the next one is batched on a timer. */
+    workerMode = false;
+    inflight = 0;
+    inflightBytes = 0;
+    uninstall = [];
+    timers = [];
+    stopped = false;
+    store = null;
+    enrollment = null;
+    keyPem = '';
+    httpsAgent = null;
+    fleet = null;
+    exporter = null;
+    etag;
+    // A heartbeat said a newer policy is current (a normal policy change): capture pauses until it
+    // is fetched, and only a failed fetch is an error (policy.md 4).
+    switching = false;
+    // Hashes of the last policies accepted: a response that brings one of them again (a poll that
+    // started before the change) is stale, not a replay.
+    acceptedHashes = [];
+    polling = null;
+    pollAgain = false;
+    constructor(opts) {
+        this.opts = opts;
+        if (opts.debug)
+            (0, log_js_1.setDebug)(true);
+    }
+    // --- CaptureHost (hot path) --------------------------------------------------------
+    active() {
+        return this.san;
+    }
+    serviceName() {
+        return this.enrollment?.service_name ?? '';
+    }
+    decisionHeaders() {
+        return this.decisionNames;
+    }
+    propagates() {
+        const p = this.policy;
+        return p !== null && p.propagateTo.size > 0;
+    }
+    mayPropagate(urlBase) {
+        const p = this.policy;
+        if (!p || p.propagateTo.size === 0)
+            return false;
+        if (this.propMemoFor !== p) {
+            this.propMemoFor = p;
+            this.propMemo.clear();
+        }
+        let r = this.propMemo.get(urlBase);
+        if (r === undefined) {
+            r = this.mayPropagateUncached(p, urlBase);
+            if (this.propMemo.size >= 1024)
+                this.propMemo.clear();
+            this.propMemo.set(urlBase, r);
+        }
+        return r;
+    }
+    propMemoFor = null;
+    propMemo = new Map();
+    mayPropagateUncached(p, urlBase) {
+        const u = (0, sanitizer_js_1.normalizeUrlBase)(urlBase);
+        // propagate_to and bases hold names, never IPv6 literals (v1.5).
+        if (u.invalid || u.ip6)
+            return false;
+        const host = u.hostport.replace(/:[0-9]+$/, '');
+        // D-M8: never toward a host that matches a policy base (external service).
+        if (p.baseHosts.has(u.hostport) || p.baseHosts.has(host))
+            return false;
+        return p.propagateTo.has(u.hostport) || p.propagateTo.has(host);
+    }
+    noteSeen() {
+        this.counters.exchanges_seen++;
+    }
+    noteSampled() {
+        this.counters.exchanges_sampled++;
+    }
+    noteOverhead(ns) {
+        this.overhead[this.overheadN++ % this.overhead.length] = ns;
+    }
+    noteError(code) {
+        this.lastErrors.push({ code: code.slice(0, 64), at: new Date().toISOString() });
+        if (this.lastErrors.length > 20)
+            this.lastErrors.shift();
+        (0, log_js_1.debug)('error', code);
+    }
+    submit(raw) {
+        if (this.san === null || this.pending.length + this.inflight >= PENDING_MAX || this.pendingBytes + this.inflightBytes + raw.heldBytes > PENDING_MAX_BYTES) {
+            this.counters.events_dropped_queue++;
+            return;
+        }
+        this.pending.push(raw);
+        this.pendingBytes += raw.heldBytes;
+        if (!this.drainScheduled) {
+            this.drainScheduled = true;
+            this.scheduleDrain();
+        }
+    }
+    // --- asynchronous sanitization ------------------------------------------------------
+    threads() {
+        const t = this.opts.sanitizerThreads;
+        if (typeof t === 'number')
+            return t;
+        return Math.min(2, Math.max(1, (0, node_os_2.availableParallelism)() - 1));
+    }
+    /** Worker threads pay off only when many exchanges are sampled (high rate). */
+    wantWorkers() {
+        const san = this.san;
+        if (!san || this.threads() <= 0)
+            return false;
+        if (san.rateBp() < (this.opts.workerMinBp ?? WORKER_MIN_BP))
+            return false;
+        if (!this.pool) {
+            this.pool = new pool_js_1.SanitizerPool({
+                size: this.threads(),
+                onOut: (b) => this.onWorkerOut(b),
+                onLost: (b) => {
+                    this.inflight -= b.n;
+                    this.inflightBytes -= b.bytes;
+                    this.counters.events_dropped_queue += b.n;
+                    this.noteError('sanitizer_worker_lost');
+                },
+            });
+        }
+        this.pool.warm();
+        return this.pool.usable;
+    }
+    onWorkerOut(b) {
+        this.inflight -= b.n;
+        this.inflightBytes -= b.bytes;
+        if (b.stale) {
+            this.counters.events_dropped_queue += b.n;
+            return;
+        }
+        const ex = this.exporter;
+        if (ex)
+            for (const record of b.recs)
+                ex.enqueue({ json: '', record, size: record.length, tsMs: 0, policyHash: '' });
+        for (const code of b.errs) {
+            this.counters.sanitizer_errors++;
+            this.noteError(code);
+        }
+    }
+    /**
+     * Inline sanitization runs right after the response (setImmediate). With worker threads
+     * the exchanges are collected for a few milliseconds and posted as one batch: one message
+     * and one thread wake-up for many exchanges instead of one each.
+     */
+    scheduleDrain() {
+        if (this.workerMode) {
+            const t = setTimeout(() => this.drain(), WORKER_BATCH_MS);
+            t.unref();
+        }
+        else {
+            setImmediate(() => this.drain());
+        }
+    }
+    drain() {
+        this.drainScheduled = false;
+        this.workerMode = false;
+        if (this.pending.length > 0 && this.wantWorkers() && this.drainToWorkers()) {
+            this.workerMode = true;
+            if (this.pending.length > 0 && !this.drainScheduled) {
+                this.drainScheduled = true;
+                this.scheduleDrain();
+            }
+            return;
+        }
+        const t0 = performance.now();
+        while (this.pending.length > 0) {
+            const raw = this.pending.shift();
+            this.pendingBytes -= raw.heldBytes;
+            this.process(raw);
+            if (performance.now() - t0 > DRAIN_BUDGET_MS)
+                break;
+        }
+        if (this.pending.length > 0 && !this.drainScheduled) {
+            this.drainScheduled = true;
+            this.scheduleDrain();
+        }
+    }
+    /** Posts up to WORKER_BATCH pending exchanges to a worker. False when none could be posted. */
+    drainToWorkers() {
+        const pool = this.pool;
+        const enr = this.enrollment;
+        if (!enr || !this.exporter)
+            return false;
+        const items = [];
+        const taken = [];
+        let bytes = 0;
+        while (this.pending.length > 0 && taken.length < WORKER_BATCH) {
+            const raw = this.pending[0];
+            if (!pool.canHandle(raw.sanitizer)) {
+                // An exchange captured under a policy the workers cannot rebuild: sanitize it here.
+                if (taken.length > 0)
+                    break;
+                return false;
+            }
+            this.pending.shift();
+            this.pendingBytes -= raw.heldBytes;
+            bytes += raw.heldBytes;
+            taken.push(raw);
+            items.push({ san: raw.sanitizer, job: { gen: 0, decision: raw.decision, meta: metaOf(raw), wire: raw.wire } });
+        }
+        if (items.length === 0)
+            return false;
+        if (pool.post(items, bytes, enr.source_id)) {
+            this.inflight += items.length;
+            this.inflightBytes += bytes;
+            return true;
+        }
+        // No worker could take them: put them back (in order) and sanitize inline.
+        this.pending.unshift(...taken);
+        this.pendingBytes += bytes;
+        return false;
+    }
+    process(raw) {
+        if (!this.exporter || !this.enrollment || this.san === null)
+            return;
+        const r = (0, pipeline_js_1.sanitizeToRecord)(raw.sanitizer, raw.wire, raw.decision, metaOf(raw), this.enrollment.source_id, Date.now());
+        if (r === null)
+            return;
+        if ('err' in r) {
+            this.counters.sanitizer_errors++;
+            this.noteError(r.err);
+            return;
+        }
+        this.exporter.enqueue({ json: '', record: r.record, size: r.record.length, tsMs: raw.tsMs, policyHash: raw.sanitizer.policy.hash });
+    }
+    // --- lifecycle --------------------------------------------------------------------
+    /** Installs the hooks and runs the first Fleet cycle. Never throws. */
+    async start() {
+        try {
+            (0, context_js_1.loadOtelApi)();
+            this.uninstall.push((0, inbound_js_1.installInbound)(this), (0, outbound_http_js_1.installOutboundHttp)(this), (0, outbound_undici_js_1.installOutboundUndici)(this));
+            const o = this.opts;
+            const svc = o.serviceName;
+            if (!o.enrollUrl || !svc) {
+                this.noteError('not_configured');
+                return;
+            }
+            this.store = new store_js_1.StateStore(o.stateDir ?? (0, store_js_1.defaultStateDir)(o.enrollUrl, svc));
+            this.store.onUntrusted = () => this.noteError('state_untrusted');
+            this.persisted = this.store.readJson('state.json') ?? {};
+            this.loadKeys();
+            const saved = this.store.readJson('enrollment.json');
+            const savedKey = this.store.readText('client.key');
+            if (saved && savedKey && saved.enroll_url === o.enrollUrl) {
+                this.enrollment = saved;
+                this.keyPem = savedKey;
+            }
+            else {
+                await this.enrollWithRetry();
+                if (!this.enrollment)
+                    return;
+            }
+            this.connect();
+            // Re-verify the cached policy (its issued_seq may equal the last accepted one).
+            const cached = this.store.readText('policy.envelope.json');
+            if (cached !== null) {
+                try {
+                    this.tryAccept((0, json_js_1.nodeToPlain)((0, json_js_1.parseJson)(Buffer.from(cached, 'utf8')), true), true, cached);
+                }
+                catch {
+                    this.noteError('policy_cache_invalid');
+                }
+            }
+            await this.heartbeat();
+            await this.pollPolicy();
+            this.schedule();
+        }
+        catch (e) {
+            this.noteError(e instanceof client_js_1.FleetError ? e.code : 'start_failed');
+        }
+        this.recompute();
+    }
+    async stop() {
+        this.stopped = true;
+        for (const t of this.timers)
+            clearTimeout(t);
+        this.timers = [];
+        for (const u of this.uninstall)
+            u();
+        this.uninstall = [];
+        this.san = null;
+        this.pool?.terminate();
+        this.exporter?.stop();
+        this.httpsAgent?.destroy();
+    }
+    /** Sanitizes what is pending and sends everything queued (tests, graceful shutdown). */
+    async flush() {
+        while (this.pending.length > 0) {
+            const raw = this.pending.shift();
+            this.pendingBytes -= raw.heldBytes;
+            this.process(raw);
+        }
+        await this.pool?.sync();
+        await this.exporter?.flush();
+    }
+    status() {
+        return { state: this.state, counters: { ...this.counters }, queue: this.exporter?.size ?? { events: 0, bytes: 0 } };
+    }
+    async enrollWithRetry() {
+        const o = this.opts;
+        if (!o.enrollToken) {
+            this.noteError('enroll_token_missing');
+            return;
+        }
+        let delay = 2000;
+        for (let attempt = 0; attempt < 8 && !this.stopped; attempt++) {
+            try {
+                const { enrollment, keyPem } = await (0, client_js_1.enroll)({
+                    enrollUrl: o.enrollUrl,
+                    token: o.enrollToken,
+                    useCsr: o.useCsr !== false,
+                    source: {
+                        kind: version_js_1.SOURCE_KIND,
+                        version: version_js_1.SDK_VERSION,
+                        service_name: o.serviceName,
+                        instance: (o.instance ?? (0, node_os_1.hostname)()).slice(0, 128),
+                        runtime: { lang: 'node', lang_version: process.versions.node, os: process.platform, arch: process.arch },
+                        hmac_kids: this.ring ? this.ring.order.slice(0, 8) : undefined,
+                    },
+                });
+                this.enrollment = enrollment;
+                this.keyPem = keyPem;
+                this.store.writeText('client.key', keyPem);
+                this.store.writeJson('enrollment.json', enrollment);
+                return;
+            }
+            catch (e) {
+                const code = e instanceof client_js_1.FleetError ? e.code : 'enroll_failed';
+                this.noteError(code);
+                // Token or identity errors will not fix themselves.
+                if (e instanceof client_js_1.FleetError && e.status !== undefined && e.status < 500)
+                    return;
+                if (/pin|root|token|chain|bad_/.test(code))
+                    return;
+            }
+            await new Promise((r) => {
+                const t = setTimeout(r, delay);
+                t.unref();
+            });
+            delay = Math.min(delay * 2, 300_000);
+        }
+    }
+    connect() {
+        const e = this.enrollment;
+        this.httpsAgent?.destroy();
+        this.httpsAgent = (0, client_js_1.mtlsAgent)(e, this.keyPem);
+        if (this.fleet)
+            this.fleet.setAgent(this.httpsAgent);
+        else
+            this.fleet = new client_js_1.FleetClient(e, this.httpsAgent);
+        if (this.exporter)
+            this.exporter.setTarget(e.intake_url, this.httpsAgent);
+        else {
+            this.exporter = new exporter_js_1.Exporter({
+                url: e.intake_url,
+                agent: this.httpsAgent,
+                resource: { serviceName: e.service_name, sourceId: e.source_id },
+                counters: this.counters,
+                onError: (c) => this.noteError(c),
+                flushMs: this.opts.exportFlushMs,
+            });
+        }
+    }
+    schedule() {
+        const e = this.enrollment;
+        const hb = this.opts.heartbeatIntervalMs ?? Math.max(5, e.heartbeat_interval_s) * 1000;
+        const pp = this.opts.policyPollIntervalMs ?? Math.max(5, e.policy_poll_interval_s) * 1000;
+        const loop = (ms, fn) => {
+            const tick = () => {
+                if (this.stopped)
+                    return;
+                void fn().finally(() => {
+                    if (this.stopped)
+                        return;
+                    const t = setTimeout(tick, ms);
+                    t.unref();
+                    this.timers.push(t);
+                    if (this.timers.length > 16)
+                        this.timers = this.timers.slice(-4);
+                });
+            };
+            const t = setTimeout(tick, ms);
+            t.unref();
+            this.timers.push(t);
+        };
+        loop(hb, () => this.heartbeat());
+        loop(pp, () => this.pollPolicy());
+    }
+    // --- keys -------------------------------------------------------------------------
+    loadKeys() {
+        let text = this.opts.hmacKey;
+        let why = 'not_set';
+        if (text === undefined && this.opts.hmacKeyFile) {
+            try {
+                text = (0, node_fs_1.readFileSync)(this.opts.hmacKeyFile, 'utf8');
+            }
+            catch {
+                text = undefined;
+                why = 'unreadable';
+            }
+        }
+        try {
+            this.ring = (0, policy_js_1.parseKeyRing)(text);
+            this.keyWarned = '';
+        }
+        catch {
+            this.ring = null;
+            if (text !== undefined)
+                why = 'malformed';
+            this.warnKeyMissing(why);
+        }
+    }
+    /** Said once per cause, whatever TT_DEBUG: without the key nothing is captured. Never prints key material or paths. */
+    keyWarned = '';
+    warnKeyMissing(why) {
+        if (this.keyWarned === why)
+            return;
+        this.keyWarned = why;
+        const cause = why === 'unreadable'
+            ? 'the file in TT_HMAC_KEY_FILE cannot be read'
+            : why === 'malformed'
+                ? 'the HMAC key is malformed (want one line <kid>:<base64url of 32+ bytes>, e.g. k1:...)'
+                : 'neither TT_HMAC_KEY_FILE nor TT_HMAC_KEY is set';
+        (0, log_js_1.warn)('key_missing: ' + cause + '. The SDK does not capture anything until the key is there (the application is not affected). ' +
+            'The key file is installation-wide and comes from your platform team: same kid as the policy (hmac.kid, e.g. k1) and the same key keyguard holds. ' +
+            'Console: Help > "Chiavi HMAC".');
+    }
+    signingKeys() {
+        const revoked = new Set(this.persisted.revoked_kids ?? []);
+        const m = new Map();
+        for (const k of [...(this.enrollment?.policy_signing_keys ?? []), ...(this.persisted.added_keys ?? [])]) {
+            if (revoked.has(k.kid) || m.has(k.kid))
+                continue;
+            const pk = (0, policy_js_1.ed25519PublicKey)(k.public_key);
+            if (pk)
+                m.set(k.kid, pk);
+        }
+        return m;
+    }
+    persist() {
+        try {
+            this.store?.writeJson('state.json', this.persisted);
+        }
+        catch {
+            this.noteError('state_write_failed');
+        }
+    }
+    // --- policy (policy.md 4) -----------------------------------------------------------
+    tryAccept(envelope, fromCache, rawText) {
+        try {
+            const v = (0, policy_js_1.verifyEnvelope)(envelope, this.signingKeys(), true);
+            const p = new policy_js_1.Policy(v.body);
+            p.hash = v.hash;
+            p.bodyBytes = v.bodyBytes;
+            if (this.policy && this.policy.hash === p.hash && !fromCache)
+                return;
+            if (!fromCache && this.acceptedHashes.includes(p.hash) && p.issuedSeq <= (this.persisted.last_issued_seq ?? 0)) {
+                // An answer that started before a newer policy was accepted (two polls crossing).
+                (0, log_js_1.debug)('info', 'policy_stale_response: issued_seq ' + p.issuedSeq + ' already superseded, kept the newer policy');
+                return;
+            }
+            (0, policy_js_1.checkAcceptance)(p, {
+                group: this.enrollment.group,
+                now: Date.now(),
+                lastIssuedSeq: this.persisted.last_issued_seq ?? 0,
+                minIssuedSeq: this.persisted.min_issued_seq ?? 0,
+                allowEqualSeq: fromCache,
+            });
+            if (this.switching)
+                (0, log_js_1.debug)('info', 'policy_changed: now issued_seq ' + p.issuedSeq);
+            this.policy = p;
+            this.policyKid = String(envelope.signature?.kid ?? '');
+            this.rejected = null;
+            this.switching = false;
+            this.acceptedHashes = [...this.acceptedHashes.filter((h) => h !== p.hash), p.hash].slice(-4);
+            this.persisted.last_issued_seq = p.issuedSeq;
+            this.persist();
+            // The cache holds the signed envelope exactly as received; it is re-verified at start.
+            if (!fromCache)
+                this.store?.writeText('policy.envelope.json', rawText);
+            const names = new Set(['cookie', 'baggage']);
+            if (p.subject?.from === 'header')
+                names.add((0, utf8_js_1.asciiLower)(p.subject.name));
+            this.decisionNames = names;
+        }
+        catch (e) {
+            const code = e instanceof policy_js_1.Refused ? e.code : 'policy_rejected';
+            // hardening.md 2.2: a policy refused for the safety subset is reported as policy_unsafe.
+            this.noteError(e instanceof policy_js_1.Refused && e.detail.startsWith(policy_js_1.UNSAFE_DETAIL) ? 'policy_unsafe' : code);
+            // Keep the previous policy if it is still valid; otherwise report the rejection.
+            if (!this.policy)
+                this.rejected = code === 'expired_policy' ? 'expired_policy' : 'policy_rejected';
+        }
+        this.recompute();
+    }
+    // One poll at a time: a heartbeat that sees a new policy and the periodic poll would otherwise
+    // race, and the older answer could arrive last.
+    pollPolicy() {
+        if (this.polling) {
+            this.pollAgain = true;
+            return this.polling;
+        }
+        this.polling = (async () => {
+            try {
+                do {
+                    this.pollAgain = false;
+                    await this.pollOnce();
+                } while (this.pollAgain && !this.stopped);
+            }
+            finally {
+                this.polling = null;
+            }
+        })();
+        return this.polling;
+    }
+    async pollOnce() {
+        if (!this.fleet || this.stopped)
+            return;
+        try {
+            const r = await this.fleet.getPolicy(this.policy ? this.etag : undefined);
+            if (r.status === 304) {
+                this.lastContact = Date.now();
+            }
+            else if (r.status === 200 && r.body) {
+                this.lastContact = Date.now();
+                let env;
+                try {
+                    env = (0, json_js_1.nodeToPlain)((0, json_js_1.parseJson)(r.body), true);
+                }
+                catch {
+                    this.noteError('policy_json');
+                    if (!this.policy)
+                        this.rejected = 'policy_rejected';
+                    this.recompute();
+                    return;
+                }
+                this.tryAccept(env, false, r.body.toString('utf8'));
+                if (this.policy && r.etag)
+                    this.etag = r.etag;
+            }
+            else if (r.status === 404) {
+                this.lastContact = Date.now();
+                this.policy = null;
+                this.rejected = 'no_policy';
+            }
+            else if (r.status === 403) {
+                this.revoke();
+            }
+            else {
+                this.noteError('policy_http_' + r.status);
+            }
+        }
+        catch {
+            this.noteError('policy_network');
+        }
+        if (this.switching && !this.policy) {
+            // The newer policy announced by the heartbeat could not be fetched or accepted.
+            this.switching = false;
+            if (this.rejected === 'starting') {
+                this.rejected = 'policy_rejected';
+                this.noteError('policy_below_min_issued_seq');
+            }
+        }
+        this.recompute();
+    }
+    // --- heartbeat --------------------------------------------------------------------
+    overheadUs() {
+        const n = Math.min(this.overheadN, this.overhead.length);
+        if (n === 0)
+            return { p50: 0, p99: 0 };
+        const a = Array.from(this.overhead.subarray(0, n)).sort((x, y) => x - y);
+        const q = (f) => Math.round(a[Math.min(n - 1, Math.floor(f * n))] / 1000);
+        return { p50: q(0.5), p99: q(0.99) };
+    }
+    async heartbeat() {
+        if (!this.fleet || this.stopped)
+            return;
+        this.loadKeys();
+        this.recompute();
+        const p = this.policy;
+        const body = {
+            source_id: this.enrollment.source_id,
+            state: this.state,
+            ...(p ? { policy: { id: p.policyId, version: p.version, issued_seq: p.issuedSeq, hash: p.hash } } : {}),
+            counters: { ...this.counters },
+            overhead_us: this.overheadUs(),
+            queue: this.exporter?.size ?? { events: 0, bytes: 0 },
+            last_errors: this.lastErrors.slice(-20),
+            runtime: { lang: 'node', lang_version: process.versions.node },
+            // The policy signing kids trusted now: tt-core waits for a rotation to reach the source
+            // before it lets the old key be revoked.
+            signing_kids: [...this.signingKeys().keys()].sort(),
+        };
+        try {
+            const r = await this.fleet.heartbeat(body);
+            if (r.status === 403) {
+                this.revoke();
+            }
+            else if (r.status === 200 && r.json) {
+                this.applyHeartbeat(r.json);
+            }
+            else {
+                this.noteError('heartbeat_http_' + r.status);
+            }
+        }
+        catch {
+            this.noteError('heartbeat_network');
+        }
+        this.recompute();
+    }
+    applyHeartbeat(j) {
+        this.heartbeatOk = true;
+        this.lastContact = Date.now();
+        const kill = j.kill === true;
+        if (kill !== (this.persisted.kill === true)) {
+            this.persisted.kill = kill;
+            this.persist();
+        }
+        if (kill)
+            this.dropQueues();
+        if (typeof j.min_issued_seq === 'number' && Number.isInteger(j.min_issued_seq) && j.min_issued_seq >= 0) {
+            if (j.min_issued_seq !== this.persisted.min_issued_seq) {
+                this.persisted.min_issued_seq = j.min_issued_seq;
+                this.persist();
+            }
+            if (this.policy && this.policy.issuedSeq < j.min_issued_seq) {
+                const cur = j.policy_current;
+                const newer = !!cur && typeof cur.hash === 'string' && cur.hash !== this.policy.hash &&
+                    typeof cur.issued_seq === 'number' && cur.issued_seq >= j.min_issued_seq;
+                // The policy in use is superseded for good: capture stops at once (fail-closed).
+                this.policy = null;
+                if (newer) {
+                    // A normal policy change: the newer policy is fetched right away (policy_current below).
+                    this.switching = true;
+                    this.rejected = 'starting';
+                    (0, log_js_1.debug)('info', 'policy_superseded: fetching issued_seq ' + String(cur.issued_seq));
+                }
+                else {
+                    this.rejected = 'policy_rejected';
+                    this.noteError('policy_below_min_issued_seq');
+                }
+            }
+        }
+        const o = j.sampling_override_bp;
+        this.overrideBp = typeof o === 'number' && o >= 0 && o <= 10000 ? o : null;
+        // Revocations first: they only remove trust, and a key endorsed by a kid revoked in this
+        // very response must not be pinned (RT3-KEY-3).
+        if (Array.isArray(j.revoke_signing_keys)) {
+            const rk = new Set(this.persisted.revoked_kids ?? []);
+            const before = rk.size;
+            for (const k of j.revoke_signing_keys)
+                if (typeof k === 'string')
+                    rk.add(k);
+            if (rk.size !== before) {
+                this.persisted.revoked_kids = [...rk].sort();
+                this.persist();
+            }
+            if (this.policy && rk.has(this.policyKid)) {
+                this.policy = null;
+                this.rejected = 'policy_rejected';
+                this.noteError('policy_key_revoked');
+            }
+        }
+        if (Array.isArray(j.add_signing_keys))
+            this.addSigningKeys(j.add_signing_keys);
+        const cur = j.policy_current;
+        if (cur && typeof cur.hash === 'string' && cur.hash !== this.policy?.hash)
+            setImmediate(() => void this.pollPolicy());
+        if (typeof j.renew_before === 'string' && Date.now() >= Date.parse(j.renew_before))
+            setImmediate(() => void this.renew());
+    }
+    // tt-core sends every endorsed key at every heartbeat, an endorser before the keys it
+    // endorses: the list is processed in order, so a key pinned earlier in it can endorse a later
+    // one. A revoked kid is never pinned again; a kid already pinned with the same key is skipped
+    // silently, one pinned with another key is refused.
+    addSigningKeys(list) {
+        const revoked = new Set(this.persisted.revoked_kids ?? []);
+        const known = this.signingKeys();
+        const pinned = new Map();
+        for (const k of [...(this.enrollment?.policy_signing_keys ?? []), ...(this.persisted.added_keys ?? [])]) {
+            if (revoked.has(k.kid) || pinned.has(k.kid))
+                continue;
+            const raw = (0, policy_js_1.b64urlDecodeStrict)(k.public_key);
+            if (raw)
+                pinned.set(k.kid, raw);
+        }
+        const added = [...(this.persisted.added_keys ?? [])];
+        let changed = false;
+        for (const item of list) {
+            const k = (item && typeof item === 'object' ? item : {});
+            // Canonical codes (docs/spec/policy.md §3.1): malformed -> rejected; a kid pinned to
+            // other bytes -> conflict; unknown/revoked endorser -> endorser_unknown; bad signature -> endorsement_invalid.
+            if (typeof k.kid !== 'string' || !/^[A-Za-z0-9._-]{1,64}$/.test(k.kid)) {
+                this.noteError('signing_key_rejected');
+                continue;
+            }
+            if (revoked.has(k.kid))
+                continue;
+            if (k.alg !== 'Ed25519' || typeof k.public_key !== 'string' || typeof k.endorsed_by !== 'string' || typeof k.endorsement !== 'string') {
+                this.noteError('signing_key_rejected');
+                continue;
+            }
+            const raw = (0, policy_js_1.b64urlDecodeStrict)(k.public_key);
+            const sig = (0, policy_js_1.b64urlDecodeStrict)(k.endorsement);
+            const pk = raw && raw.length === 32 ? (0, policy_js_1.ed25519PublicKey)(k.public_key) : undefined;
+            if (!raw || raw.length !== 32 || !sig || sig.length !== 64 || !pk) {
+                this.noteError('signing_key_rejected');
+                continue;
+            }
+            const have = pinned.get(k.kid);
+            if (have) {
+                if (!have.equals(raw))
+                    this.noteError('signing_key_conflict');
+                continue;
+            }
+            const endorser = revoked.has(k.endorsed_by) ? undefined : known.get(k.endorsed_by);
+            if (!endorser) {
+                this.noteError('signing_key_endorser_unknown');
+                continue;
+            }
+            const msg = Buffer.concat([Buffer.from('tt-fleet-key/v1\0' + k.kid + '\0', 'latin1'), raw]);
+            let ok = false;
+            try {
+                ok = (0, node_crypto_1.verify)(null, msg, endorser, sig);
+            }
+            catch {
+                ok = false;
+            }
+            if (!ok) {
+                this.noteError('signing_key_endorsement_invalid');
+                continue;
+            }
+            added.push({ kid: k.kid, alg: 'Ed25519', public_key: k.public_key });
+            known.set(k.kid, pk);
+            pinned.set(k.kid, raw);
+            changed = true;
+        }
+        if (changed) {
+            this.persisted.added_keys = added;
+            this.persist();
+        }
+    }
+    async renew() {
+        if (!this.fleet || !this.enrollment)
+            return;
+        try {
+            const r = await this.fleet.renew(this.opts.useCsr !== false, this.enrollment.service_name);
+            if (!r || !r.keyPem) {
+                this.noteError('renew_failed');
+                return;
+            }
+            this.enrollment.client_cert = r.cert;
+            if (r.notAfter)
+                this.enrollment.not_after = r.notAfter;
+            this.keyPem = r.keyPem;
+            this.store?.writeText('client.key', r.keyPem);
+            this.store?.writeJson('enrollment.json', this.enrollment);
+            this.connect();
+        }
+        catch {
+            this.noteError('renew_network');
+        }
+    }
+    revoke() {
+        this.persisted.revoked = true;
+        this.persist();
+        this.dropQueues();
+        this.noteError('source_revoked');
+    }
+    dropQueues() {
+        this.counters.events_dropped_queue += this.pending.length;
+        this.pending = [];
+        this.pendingBytes = 0;
+        this.pool?.discard();
+        this.exporter?.clear();
+    }
+    /** Recomputes the state and the capture gate. */
+    recompute() {
+        const now = Date.now();
+        const p = this.policy;
+        const maxAgeMs = (this.enrollment?.policy_max_age_s ?? 86400) * 1000;
+        let st;
+        let key = null;
+        if (this.persisted.kill || this.persisted.revoked)
+            st = 'killed';
+        // A missing key is said first: with or without a policy nothing is captured until it is there.
+        else if (!this.ring)
+            st = 'key_missing';
+        else if (!p)
+            st = this.rejected ?? 'no_policy';
+        else if (!(now < p.notAfter) || (this.lastContact > 0 && now - this.lastContact > maxAgeMs))
+            st = 'expired_policy';
+        else {
+            try {
+                key = (0, policy_js_1.selectKey)(this.ring, p);
+                st = this.heartbeatOk ? 'capturing' : 'starting';
+            }
+            catch {
+                st = 'key_mismatch';
+            }
+        }
+        // Capture only after a heartbeat in this process (kill switch and min_issued_seq known).
+        if (st === 'capturing' && key && p && this.lastContact > 0) {
+            const cur = this.san;
+            if (!cur || cur.policy !== p || cur.key.kid !== key.kid || cur.key.key !== key.key || cur.options.rateOverrideBp !== this.overrideBp) {
+                this.san = new sanitizer_js_1.Sanitizer(p, key, { rateOverrideBp: this.overrideBp });
+            }
+        }
+        else {
+            if (st === 'capturing')
+                st = 'starting';
+            this.san = null;
+        }
+        if (st !== this.state)
+            (0, log_js_1.debug)('state', st);
+        this.state = st;
+    }
+}
+exports.Agent = Agent;
+function metaOf(raw) {
+    return {
+        dir: raw.dir,
+        tsMs: raw.tsMs,
+        latencyMs: raw.latencyMs,
+        status: raw.status,
+        netError: raw.netError,
+        traceId: raw.traceId,
+        spanId: raw.spanId,
+        parentSpanId: raw.parentSpanId,
+    };
+}