mirror of
https://github.com/DeusData/codebase-memory-mcp.git
synced 2026-11-11 09:25:35 +08:00
Daemon lifecycle and Windows correctness, verified on a real Windows 11 ARM64 VM through the maintained test-infrastructure/vm drivers, plus the macOS and Linux arm64 suites and the container lint gate. Daemon lifecycle: - daemon start/stop/status subcommands. `daemon start` launches a PERMANENT daemon (spawn shape is byte-exact argv; survives idle periods and session ends) and reports an already-active daemon instead of failing. Permanence is honored at every stop latch: last-committed-client disconnect, host initial-client window, coordinator release, and application final-session close — a permanent daemon also keeps admitting new sessions after its last one closes. - daemon stop refuses while sessions are active and lists the blocking peers (pid/role) that must finish first; an idle daemon drains through the activation-shutdown machinery with the ACK ordered after connection interrupts. A second stop is idempotent. The wire ops are no-cohort first-frame requests with peer fingerprint authentication, so stop/status never conflict with an exact-build admission gate. - One-shot CLI commands now execute through the daemon (index workers keep their local supervised path). A cold CLI run that had to spawn a temporary daemon prints a hint that `daemon start` removes the per-command startup tax; a warm daemon is recycled silently. - Hooks are connect-only fail-open: with no daemon present the hook emits a visible, rate-limited notice (Claude-dialect systemMessage plus stderr for other dialects) and always exits 0 — augmentation is never allowed to block the caller's tool use. - Version skew: a newer-build client automatically drains an older-build permanent daemon (strict semantic-version triples only; dev builds never auto-drain) and the build-conflict message names `cbm daemon stop` as the manual escape hatch. Windows IPC/runtime (real-VM verified): - ipc(win): persistent pending overlapped ConnectNamedPipe. The accept path used to destroy its listening pipe instance on every 20 ms poll timeout; a client attaching in the teardown window was severed or left on an orphaned pipe object whose HELLO no server handle could ever read, absorbing the connect until the client's own timeout expired. The pending connect now survives poll timeouts and nothing is destroyed while a client could be attaching. - ipc(win): drain-before-close for final responses. Closing a named-pipe server handle can discard a just-sent response before the peer reads it (POSIX stream sockets never lose buffered data on close). A bounded cbm_daemon_ipc_connection_drain (read-until-peer-EOF; no-op on POSIX, immediate on interrupted connections) now precedes close in runtime_worker_finish and runtime_reject_inline, so hello-conflict, capacity and disconnect acknowledgements reliably reach the peer. - runtime: CLOSE_INTENT wire frame. A Windows named-pipe client has no transport half-close, so close_begin now announces departure with an explicit frame (ordered after APPLICATION_CANCEL, before the local interrupt); the server releases the client's admission on receipt instead of waiting for the handle to close. Admission-drop timing is now identical to POSIX shutdown() semantics on every platform. - runtime(win): client close cancellation. close_begin serializes with request publication under the send lock, best-effort sends the active token's APPLICATION_CANCEL frame, then interrupts local I/O; the server cancels MCP/subprocess work promptly. Contract tests accept both correct outcomes (interrupted transport or decoded CANCELLED). - runtime: activation acknowledgement ordering. The activation ACK is the requester's license to act on "snapshotted and draining", so every connection interrupt is now initiated before the ACK is sent; a session could previously get one more request serviced after the requester observed the ACK. - service(win): deadline-bounded private-file prepare. The conflict-log prepare retry loop (100 x Sleep(2), which rounds up to the ~16 ms timer granularity) burned ~1.6 s against permanently obstructed paths, stalling hello rejections past the client's timeout. The retry budget is now a 250 ms deadline; transient share collisions still retry. - subprocess(win): cmd.exe /C payload encoder quotes metacharacters correctly (root cause of the git-on-Windows failure cluster). - watcher: SHA-256 buffer sizing (CBM_SZ_64 -> CBM_SZ_128) and a native Windows stop/unwatch cancellation test with exact-image verification. - httpd: send_all writes in bounded 64 KiB slices. A single giant nonblocking send() on Windows is absorbed wholesale into AFD kernel buffering regardless of SO_SNDBUF, so send deadlines and interrupts could never engage against a slow peer (and the full payload was pinned in nonpaged pool). Slicing restores a deterministic backpressure point; a test hook pins SO_SNDBUF for the deadline and interrupt tests. - ui/http: shutdown lifecycle — interrupt checks, response-wide send deadline, explicit connection states, refusal to free a server while a listener-owned connection is active. Windows long-path support: - Central path-aware wide conversion (canonicalize via GetFullPathNameW and prepend the extended-length prefix for absolute paths >=240) at the compat chokepoints (cbm_fopen/compat_fs/mkstemp/mkdtemp), sqlite store opens, and the daemon build-fingerprint/log paths. Deep managed installs (a 64-hex generation directory routinely exceeds MAX_PATH) now index, stage and activate correctly. - activation transaction: its own file APIs and the component-walking ancestry validators now operate in the extended-length namespace; the launcher path is canonicalized (and prefixed when deep) once at entry so every downstream exact-string comparison stays form-consistent. - Executable self-resolution uses the wide APIs (GetModuleFileNameW, GetFileAttributesW) so non-ASCII install paths survive argv[0] resolution. Windows launcher install/uninstall transaction: - FileRenameInfoEx names are NUL-terminated in an over-allocated buffer. FileNameLength governs per the contract, but filter drivers read FileName as NUL-terminated and appended adjacent heap bytes to created names — a flaky, garbage-suffixed rename target. Both the CLI and the launcher rename helpers are fixed. - Uninstall retires state via rename-aside (.cbm -> .cbm-retired-v1-<tag>-<pid>) with the retired tag shortened to 16 hex chars so the bare rename target stays under the FileRenameInfoEx NT-conversion ceiling at guard depths; 64 bits still uniquely identify the generation. - When the running launcher's mapped generation backings pin .cbm against rename, the backings are relocated to activation-<pid>-N .retired tombstones beside the install (a mapped image may be renamed, never deleted; the launcher's liveness-guarded sweep reclaims stale tombstones). Every relocation is recorded, and a FAILED uninstall reverses the moves after restoring .cbm — via MoveFileExW with extended-length paths on both arguments, since the deep generation target is beyond the handle-based rename's bare-path reach — so a restored install keeps its generation backings and stays runnable. - After a committed uninstall the retired tree's backings are relocated out so the tree is shallow enough for the detached cleanup's rd, and the cleanup's working directory strips the extended-length prefix (CreateProcessW lpCurrentDirectory silently ignores prefixed paths). - Files created under Administrators-default-owner directories (CopyFileW destinations, CREATE_NEW tombstones, probe directories) are explicitly owner-stamped so the exact-owner validators hold on runner images; guard fixtures stamp hand-built trees the same way. Diagnostics, tests and infra: - diagnostics: discovery is now an always-delivered JSON control record (new cbm_log_control) that survives CBM_LOG_LEVEL suppression and paths containing spaces; placement honors $TMPDIR with /tmp fallback via a diagnostics-local helper; the soak parser reads the JSON record; documented in docs/CONFIGURATION.md. Red-first coverage for suppressed log levels, TMPDIR-with-spaces, and native Windows output-contract assertions. - tests(win): daemon_ipc/daemon_frontend fixtures now build endpoint parents with production-shaped ancestry (LocalAppData on Windows, via th_secure_runtime_parent_new) — the runtime ancestry validation correctly refuses temp roots whose ancestors grant mutation rights to Authenticated Users (C:/msys64/tmp, GitHub-runner work dirs) — and drive the documented startup-owner publication flow before reading generation-bound endpoint addresses. This turns the 26 Windows failures previously visible in CI's full-test job green without weakening any validation. - tests(win): the launcher guard covers the full permanent-launcher contract including failed-uninstall restore and immediate reinstall after uninstall; new daemon lifecycle and reworked hook-augment guards run the start/recycle/stop flow end to end. - tests: CBM_SKIP_PERF is now actually consumed by the test runner (it was set by CI but never read, so perf suites ran everywhere); four throughput/bench suites are classified as perf, the heavy store_arch suite moved to the slow-timeout tier, and two wall-clock-sensitive assertions were rewritten as invariant checks with coarse hang-detector backstops. - build/test infra: build-dir safety contract, UI dev-proxy security contract, soak daemon-recovery contract, path-safety helper, the Windows VM worktree-sync contract wired into scripts/test.sh, and vm/win.sh guards building its clean embedded-UI product in an isolated BUILD_DIR so it cannot clobber the incremental test build. provision-windows.sh now installs Node.js for the guards UI build. Signed-off-by: Martin Vogel <martin.vogel.tech@gmail.com>
996 lines
38 KiB
C
996 lines
38 KiB
C
/*
|
|
* bootstrap.c — Mandatory per-account daemon startup policy.
|
|
*/
|
|
#include "daemon/bootstrap.h"
|
|
|
|
#include "daemon/ipc.h"
|
|
#include "daemon/service.h"
|
|
#include "foundation/compat.h"
|
|
#include "foundation/platform.h"
|
|
|
|
#include <limits.h>
|
|
#include <stdint.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
|
|
#ifdef _WIN32
|
|
#ifndef WIN32_LEAN_AND_MEAN
|
|
#define WIN32_LEAN_AND_MEAN
|
|
#endif
|
|
#include "foundation/subprocess.h"
|
|
#include "foundation/win_utf8.h"
|
|
#include <windows.h>
|
|
#else
|
|
#include <errno.h>
|
|
#include <fcntl.h>
|
|
#include <signal.h>
|
|
#include <sys/stat.h>
|
|
#include <sys/types.h>
|
|
#include <sys/wait.h>
|
|
#include <unistd.h>
|
|
#ifdef __APPLE__
|
|
#include <pthread.h>
|
|
#include <spawn.h>
|
|
extern char **environ;
|
|
#endif
|
|
#endif
|
|
|
|
enum {
|
|
BOOTSTRAP_RETRY_NS = 1000000,
|
|
BOOTSTRAP_COORDINATION_CLEANUP_MS = 500,
|
|
BOOTSTRAP_PATH_CAP = 4096,
|
|
};
|
|
|
|
static bool bootstrap_arg_is(const char *arg, const char *expected) {
|
|
return arg && expected && strcmp(arg, expected) == 0;
|
|
}
|
|
|
|
static int bootstrap_find_arg(int argc, char *const argv[], const char *expected) {
|
|
for (int i = 1; argv && i < argc; i++) {
|
|
if (bootstrap_arg_is(argv[i], expected)) {
|
|
return i;
|
|
}
|
|
}
|
|
return -1;
|
|
}
|
|
|
|
static bool bootstrap_has_help_after(int argc, char *const argv[], int start) {
|
|
for (int i = start; argv && i < argc; i++) {
|
|
if (bootstrap_arg_is(argv[i], "--help") || bootstrap_arg_is(argv[i], "-h")) {
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
static bool bootstrap_worker_fingerprint_valid(const char *fingerprint) {
|
|
if (!fingerprint || strlen(fingerprint) != 64U) {
|
|
return false;
|
|
}
|
|
for (size_t i = 0; i < 64U; i++) {
|
|
char ch = fingerprint[i];
|
|
if (!((ch >= '0' && ch <= '9') || (ch >= 'a' && ch <= 'f'))) {
|
|
return false;
|
|
}
|
|
}
|
|
return true;
|
|
}
|
|
|
|
static bool bootstrap_worker_budget_valid(const char *text) {
|
|
if (!text || !text[0]) {
|
|
return false;
|
|
}
|
|
size_t value = 0;
|
|
for (const unsigned char *cursor = (const unsigned char *)text; *cursor; cursor++) {
|
|
if (*cursor < '0' || *cursor > '9') {
|
|
return false;
|
|
}
|
|
size_t digit = (size_t)(*cursor - '0');
|
|
if (value > (SIZE_MAX - digit) / 10U) {
|
|
return false;
|
|
}
|
|
value = value * 10U + digit;
|
|
}
|
|
return value > 0;
|
|
}
|
|
|
|
/* Keep the bootstrap role boundary exact and fail closed before any client or
|
|
* worker state is initialized. index_supervisor owns the matching builder and
|
|
* performs the captured-build comparison after this syntax-only classification. */
|
|
static bool bootstrap_worker_argv_exact(int argc, char *const argv[]) {
|
|
if (argc < 9 || !argv || !bootstrap_arg_is(argv[1], "cli") ||
|
|
!bootstrap_arg_is(argv[2], "--index-worker") ||
|
|
!bootstrap_arg_is(argv[3], "--index-worker-build") ||
|
|
!bootstrap_worker_fingerprint_valid(argv[4]) ||
|
|
!bootstrap_arg_is(argv[5], "index_repository") || !argv[6] || !argv[6][0] ||
|
|
!bootstrap_arg_is(argv[7], "--response-out") || !argv[8] || !argv[8][0]) {
|
|
return false;
|
|
}
|
|
int next = 9;
|
|
if (next < argc && bootstrap_arg_is(argv[next], "--index-worker-memory-budget-bytes")) {
|
|
if (next + 1 >= argc || !bootstrap_worker_budget_valid(argv[next + 1])) {
|
|
return false;
|
|
}
|
|
next += 2;
|
|
}
|
|
if (next < argc && bootstrap_arg_is(argv[next], "--index-worker-single-thread")) {
|
|
next++;
|
|
}
|
|
if (next < argc && bootstrap_arg_is(argv[next], "--index-worker-marker")) {
|
|
if (next + 1 >= argc || !argv[next + 1] || !argv[next + 1][0]) {
|
|
return false;
|
|
}
|
|
next += 2;
|
|
}
|
|
if (next < argc && bootstrap_arg_is(argv[next], "--index-worker-quarantine")) {
|
|
if (next + 1 >= argc || !argv[next + 1] || !argv[next + 1][0]) {
|
|
return false;
|
|
}
|
|
next += 2;
|
|
}
|
|
return next == argc;
|
|
}
|
|
|
|
cbm_daemon_process_role_t cbm_daemon_process_role(int argc, char *const argv[]) {
|
|
if (argc <= 0 || !argv || !argv[0] || argv[0][0] == '\0') {
|
|
return CBM_DAEMON_PROCESS_INVALID;
|
|
}
|
|
|
|
int daemon_arg = bootstrap_find_arg(argc, argv, CBM_DAEMON_INTERNAL_ARG);
|
|
if (daemon_arg >= 0) {
|
|
/* Byte-exact daemon-role grammar, deliberately unforgiving: the bare
|
|
* internal marker, or the marker followed by exactly the permanent
|
|
* flag. Every other shape — reordered, repeated, or extended — is
|
|
* INVALID, so argv smuggling cannot reach the daemon role. */
|
|
if (argc == 2 && daemon_arg == 1) {
|
|
return CBM_DAEMON_PROCESS_DAEMON;
|
|
}
|
|
if (argc == 3 && daemon_arg == 1 && bootstrap_arg_is(argv[2], CBM_DAEMON_PERMANENT_ARG)) {
|
|
return CBM_DAEMON_PROCESS_DAEMON;
|
|
}
|
|
return CBM_DAEMON_PROCESS_INVALID;
|
|
}
|
|
|
|
int worker_arg = bootstrap_find_arg(argc, argv, "--index-worker");
|
|
if (worker_arg >= 0) {
|
|
return bootstrap_worker_argv_exact(argc, argv) ? CBM_DAEMON_PROCESS_WORKER
|
|
: CBM_DAEMON_PROCESS_INVALID;
|
|
}
|
|
|
|
static const char *const stateless_commands[] = {
|
|
"install",
|
|
"uninstall",
|
|
"update",
|
|
};
|
|
/* Stop at the first top-level mode token. Tool names, flag values, and JSON
|
|
* following `cli` are opaque user input: a search query named "install"
|
|
* or containing "--version" must never bypass the mandatory daemon. */
|
|
for (int arg = 1; arg < argc; arg++) {
|
|
if (bootstrap_arg_is(argv[arg], "cli")) {
|
|
if (bootstrap_has_help_after(argc, argv, arg + 1)) {
|
|
return CBM_DAEMON_PROCESS_STATELESS;
|
|
}
|
|
return CBM_DAEMON_PROCESS_LOCAL_CLI;
|
|
}
|
|
if (bootstrap_arg_is(argv[arg], "hook-augment")) {
|
|
return CBM_DAEMON_PROCESS_HOOK_CLIENT;
|
|
}
|
|
if (bootstrap_arg_is(argv[arg], "config")) {
|
|
return bootstrap_has_help_after(argc, argv, arg + 1) ? CBM_DAEMON_PROCESS_STATELESS
|
|
: CBM_DAEMON_PROCESS_LOCAL_CLI;
|
|
}
|
|
/* Placed after the `cli` check on purpose: `cbm cli search "daemon
|
|
* start"` is opaque tool input and must stay LOCAL_CLI. */
|
|
if (bootstrap_arg_is(argv[arg], "daemon")) {
|
|
return bootstrap_has_help_after(argc, argv, arg + 1) ? CBM_DAEMON_PROCESS_STATELESS
|
|
: CBM_DAEMON_PROCESS_DAEMON_CTL;
|
|
}
|
|
if (bootstrap_arg_is(argv[arg], "--version") || bootstrap_arg_is(argv[arg], "--help") ||
|
|
bootstrap_arg_is(argv[arg], "-h")) {
|
|
return CBM_DAEMON_PROCESS_STATELESS;
|
|
}
|
|
for (size_t command = 0;
|
|
command < sizeof(stateless_commands) / sizeof(stateless_commands[0]); command++) {
|
|
if (bootstrap_arg_is(argv[arg], stateless_commands[command])) {
|
|
return CBM_DAEMON_PROCESS_STATELESS;
|
|
}
|
|
}
|
|
}
|
|
return CBM_DAEMON_PROCESS_MCP_CLIENT;
|
|
}
|
|
|
|
bool cbm_daemon_process_role_requires_client(cbm_daemon_process_role_t role) {
|
|
return role == CBM_DAEMON_PROCESS_MCP_CLIENT || role == CBM_DAEMON_PROCESS_HOOK_CLIENT;
|
|
}
|
|
|
|
cbm_daemon_ipc_endpoint_t *cbm_daemon_bootstrap_endpoint_new(const char *runtime_parent) {
|
|
char key[CBM_DAEMON_KEY_SIZE];
|
|
if (!cbm_daemon_rendezvous_key(key)) {
|
|
return NULL;
|
|
}
|
|
return cbm_daemon_ipc_endpoint_new(key, runtime_parent);
|
|
}
|
|
|
|
bool cbm_daemon_bootstrap_launch_spec_init(const char *executable_path,
|
|
cbm_daemon_bootstrap_launch_spec_t *spec_out) {
|
|
if (!executable_path || executable_path[0] == '\0' || !spec_out) {
|
|
return false;
|
|
}
|
|
memset(spec_out, 0, sizeof(*spec_out));
|
|
spec_out->executable_path = executable_path;
|
|
spec_out->argv[0] = executable_path;
|
|
spec_out->argv[1] = CBM_DAEMON_INTERNAL_ARG;
|
|
spec_out->argc = 2U;
|
|
spec_out->detached = true;
|
|
spec_out->inherit_standard_handles = false;
|
|
spec_out->use_shell = false;
|
|
return true;
|
|
}
|
|
|
|
bool cbm_daemon_bootstrap_launch_spec_init_permanent(const char *executable_path,
|
|
cbm_daemon_bootstrap_launch_spec_t *spec_out) {
|
|
if (!cbm_daemon_bootstrap_launch_spec_init(executable_path, spec_out)) {
|
|
return false;
|
|
}
|
|
spec_out->argv[2] = CBM_DAEMON_PERMANENT_ARG;
|
|
spec_out->argc = 3U;
|
|
return true;
|
|
}
|
|
|
|
static uint64_t bootstrap_deadline_after(uint32_t timeout_ms) {
|
|
uint64_t now = cbm_now_ms();
|
|
return UINT64_MAX - now < timeout_ms ? UINT64_MAX : now + (uint64_t)timeout_ms;
|
|
}
|
|
|
|
static _Noreturn void bootstrap_cleanup_fail_stop(const char *component) {
|
|
(void)fprintf(stderr,
|
|
"codebase-memory-mcp: coordination cleanup failed (%s); "
|
|
"terminating so the OS releases retained claims\n",
|
|
component ? component : "unknown");
|
|
(void)fflush(stderr);
|
|
#ifdef _WIN32
|
|
(void)TerminateProcess(GetCurrentProcess(), EXIT_FAILURE);
|
|
abort();
|
|
#else
|
|
_exit(EXIT_FAILURE);
|
|
#endif
|
|
}
|
|
|
|
static void bootstrap_pause(uint64_t deadline) {
|
|
uint64_t now = cbm_now_ms();
|
|
if (now >= deadline) {
|
|
return;
|
|
}
|
|
uint64_t remaining_ms = deadline - now;
|
|
struct timespec pause = {
|
|
.tv_sec = 0,
|
|
.tv_nsec = remaining_ms > 1 ? BOOTSTRAP_RETRY_NS : (long)(remaining_ms * 1000000ULL),
|
|
};
|
|
(void)cbm_nanosleep(&pause, NULL);
|
|
}
|
|
|
|
static void bootstrap_startup_lock_release_complete(const cbm_daemon_bootstrap_ops_t *ops,
|
|
cbm_daemon_bootstrap_lock_t *lock_io) {
|
|
uint64_t deadline = bootstrap_deadline_after(BOOTSTRAP_COORDINATION_CLEANUP_MS);
|
|
while (ops && ops->startup_lock_release && lock_io && *lock_io) {
|
|
(void)ops->startup_lock_release(ops->context, lock_io);
|
|
if (!*lock_io) {
|
|
return;
|
|
}
|
|
if (cbm_now_ms() >= deadline) {
|
|
bootstrap_cleanup_fail_stop("startup_lock_cleanup");
|
|
}
|
|
cbm_usleep(1000);
|
|
}
|
|
}
|
|
|
|
static void bootstrap_result_reset(cbm_daemon_bootstrap_result_t *result,
|
|
cbm_daemon_process_role_t role) {
|
|
memset(result, 0, sizeof(*result));
|
|
result->status = cbm_daemon_process_role_requires_client(role) ? CBM_DAEMON_BOOTSTRAP_FAILED
|
|
: CBM_DAEMON_BOOTSTRAP_BYPASSED;
|
|
}
|
|
|
|
static cbm_daemon_bootstrap_status_t bootstrap_finish_probe(
|
|
cbm_daemon_bootstrap_probe_status_t probe, cbm_daemon_runtime_client_t *client,
|
|
const cbm_daemon_runtime_connect_result_t *connect_result,
|
|
const cbm_daemon_bootstrap_ops_t *ops, cbm_daemon_bootstrap_result_t *result) {
|
|
if (connect_result) {
|
|
result->connect_result = *connect_result;
|
|
}
|
|
result->client = client;
|
|
if (probe == CBM_DAEMON_BOOTSTRAP_PROBE_CONNECTED && client) {
|
|
result->status = CBM_DAEMON_BOOTSTRAP_CONNECTED;
|
|
return result->status;
|
|
}
|
|
result->client = NULL;
|
|
if (probe == CBM_DAEMON_BOOTSTRAP_PROBE_CONFLICT) {
|
|
result->status = CBM_DAEMON_BOOTSTRAP_CONFLICT;
|
|
(void)snprintf(result->message, sizeof(result->message), "%s",
|
|
connect_result && connect_result->message[0]
|
|
? connect_result->message
|
|
: "CBM could not start because a conflicting CBM process is active; "
|
|
"close all CBM sessions and commands, then retry. If a permanent "
|
|
"daemon from another build is running, `codebase-memory-mcp daemon "
|
|
"stop` retires it");
|
|
if (ops->visible_diagnostic) {
|
|
ops->visible_diagnostic(ops->context, result->message);
|
|
}
|
|
return result->status;
|
|
}
|
|
return CBM_DAEMON_BOOTSTRAP_FAILED;
|
|
}
|
|
|
|
static cbm_daemon_bootstrap_probe_status_t bootstrap_probe(
|
|
const cbm_daemon_bootstrap_config_t *config, const cbm_daemon_bootstrap_ops_t *ops,
|
|
cbm_daemon_runtime_client_t **client_out, cbm_daemon_runtime_connect_result_t *connect_result) {
|
|
memset(connect_result, 0, sizeof(*connect_result));
|
|
*client_out = NULL;
|
|
return ops->probe(ops->context, config->endpoint, config->identity, config->connect_timeout_ms,
|
|
client_out, connect_result);
|
|
}
|
|
|
|
static bool bootstrap_probe_is_finishable(cbm_daemon_bootstrap_probe_status_t probe) {
|
|
return probe == CBM_DAEMON_BOOTSTRAP_PROBE_CONNECTED ||
|
|
probe == CBM_DAEMON_BOOTSTRAP_PROBE_CONFLICT;
|
|
}
|
|
|
|
static bool bootstrap_probe_is_waitable(cbm_daemon_bootstrap_probe_status_t probe) {
|
|
return probe == CBM_DAEMON_BOOTSTRAP_PROBE_UNAVAILABLE ||
|
|
probe == CBM_DAEMON_BOOTSTRAP_PROBE_RESERVED ||
|
|
probe == CBM_DAEMON_BOOTSTRAP_PROBE_TERMINAL;
|
|
}
|
|
|
|
static bool bootstrap_config_valid(const cbm_daemon_bootstrap_config_t *config,
|
|
const cbm_daemon_bootstrap_ops_t *ops) {
|
|
return config && ops && config->endpoint && config->identity && config->executable_path &&
|
|
config->executable_path[0] && config->connect_timeout_ms > 0 &&
|
|
config->startup_timeout_ms > 0 && ops->cohort_acquire && ops->cohort_release &&
|
|
ops->probe && ops->startup_lock_try_acquire && ops->startup_lock_prepare_handoff &&
|
|
ops->startup_lock_release && ops->spawn_daemon;
|
|
}
|
|
|
|
cbm_daemon_bootstrap_status_t cbm_daemon_bootstrap_execute_with_ops(
|
|
const cbm_daemon_bootstrap_config_t *config, const cbm_daemon_bootstrap_ops_t *ops,
|
|
cbm_daemon_bootstrap_result_t *result_out) {
|
|
if (!result_out) {
|
|
return CBM_DAEMON_BOOTSTRAP_FAILED;
|
|
}
|
|
cbm_daemon_process_role_t role = config ? config->role : CBM_DAEMON_PROCESS_INVALID;
|
|
bootstrap_result_reset(result_out, role);
|
|
if (!cbm_daemon_process_role_requires_client(role)) {
|
|
return role == CBM_DAEMON_PROCESS_INVALID ? CBM_DAEMON_BOOTSTRAP_FAILED
|
|
: CBM_DAEMON_BOOTSTRAP_BYPASSED;
|
|
}
|
|
if (!bootstrap_config_valid(config, ops)) {
|
|
return CBM_DAEMON_BOOTSTRAP_FAILED;
|
|
}
|
|
|
|
uint64_t deadline = bootstrap_deadline_after(config->startup_timeout_ms);
|
|
cbm_daemon_bootstrap_cohort_t cohort = NULL;
|
|
cbm_daemon_conflict_t cohort_conflict;
|
|
cbm_version_cohort_status_t cohort_status = ops->cohort_acquire(
|
|
ops->context, config->endpoint, config->identity, deadline, &cohort, &cohort_conflict);
|
|
if (cohort_status != CBM_VERSION_COHORT_OK) {
|
|
result_out->status = cohort_status == CBM_VERSION_COHORT_CONFLICT
|
|
? CBM_DAEMON_BOOTSTRAP_CONFLICT
|
|
: CBM_DAEMON_BOOTSTRAP_FAILED;
|
|
bool formatted = cohort_status == CBM_VERSION_COHORT_CONFLICT &&
|
|
cbm_daemon_conflict_format(&cohort_conflict, result_out->message,
|
|
sizeof(result_out->message));
|
|
if (!formatted) {
|
|
const char *reason = cohort_status == CBM_VERSION_COHORT_BUSY
|
|
? "another CBM activation is in progress"
|
|
: "exact-build admission could not be verified";
|
|
(void)snprintf(result_out->message, sizeof(result_out->message),
|
|
"CBM daemon could not start: %s", reason);
|
|
}
|
|
if (ops->visible_diagnostic) {
|
|
ops->visible_diagnostic(ops->context, result_out->message);
|
|
}
|
|
if (cohort) {
|
|
ops->cohort_release(ops->context, cohort);
|
|
}
|
|
return result_out->status;
|
|
}
|
|
|
|
cbm_daemon_runtime_client_t *client = NULL;
|
|
cbm_daemon_runtime_connect_result_t connect_result;
|
|
cbm_daemon_bootstrap_probe_status_t probe =
|
|
bootstrap_probe(config, ops, &client, &connect_result);
|
|
if (bootstrap_probe_is_finishable(probe)) {
|
|
cbm_daemon_bootstrap_status_t status =
|
|
bootstrap_finish_probe(probe, client, &connect_result, ops, result_out);
|
|
ops->cohort_release(ops->context, cohort);
|
|
return status;
|
|
}
|
|
if (!bootstrap_probe_is_waitable(probe)) {
|
|
probe = CBM_DAEMON_BOOTSTRAP_PROBE_ERROR;
|
|
}
|
|
|
|
cbm_daemon_bootstrap_lock_t startup_lock = NULL;
|
|
bool lock_acquired = false;
|
|
bool generation_observed = probe == CBM_DAEMON_BOOTSTRAP_PROBE_RESERVED ||
|
|
probe == CBM_DAEMON_BOOTSTRAP_PROBE_TERMINAL;
|
|
while (cbm_now_ms() < deadline) {
|
|
if (!bootstrap_probe_is_waitable(probe)) {
|
|
break;
|
|
}
|
|
if (probe == CBM_DAEMON_BOOTSTRAP_PROBE_RESERVED ||
|
|
probe == CBM_DAEMON_BOOTSTRAP_PROBE_TERMINAL) {
|
|
/* A live or stopping generation owns the transition for now. Its
|
|
* disappearance is not sticky: after observing true absence, the
|
|
* same bootstrap attempt may serialize and become the next first
|
|
* client. The startup lock and the re-probe below prevent two
|
|
* replacements from being launched. */
|
|
generation_observed = true;
|
|
bootstrap_pause(deadline);
|
|
probe = bootstrap_probe(config, ops, &client, &connect_result);
|
|
continue;
|
|
}
|
|
|
|
int lock_status =
|
|
ops->startup_lock_try_acquire(ops->context, config->endpoint, &startup_lock);
|
|
if (lock_status < 0) {
|
|
probe = CBM_DAEMON_BOOTSTRAP_PROBE_ERROR;
|
|
break;
|
|
}
|
|
if (lock_status == 0) {
|
|
bootstrap_pause(deadline);
|
|
probe = bootstrap_probe(config, ops, &client, &connect_result);
|
|
continue;
|
|
}
|
|
if (lock_status != 1 || !startup_lock) {
|
|
probe = CBM_DAEMON_BOOTSTRAP_PROBE_ERROR;
|
|
break;
|
|
}
|
|
|
|
lock_acquired = true;
|
|
probe = bootstrap_probe(config, ops, &client, &connect_result);
|
|
if (probe == CBM_DAEMON_BOOTSTRAP_PROBE_RESERVED ||
|
|
probe == CBM_DAEMON_BOOTSTRAP_PROBE_TERMINAL) {
|
|
generation_observed = true;
|
|
bootstrap_startup_lock_release_complete(ops, &startup_lock);
|
|
lock_acquired = false;
|
|
continue;
|
|
}
|
|
if (bootstrap_probe_is_finishable(probe) || probe == CBM_DAEMON_BOOTSTRAP_PROBE_ERROR) {
|
|
break;
|
|
}
|
|
if (probe != CBM_DAEMON_BOOTSTRAP_PROBE_UNAVAILABLE) {
|
|
probe = CBM_DAEMON_BOOTSTRAP_PROBE_ERROR;
|
|
break;
|
|
}
|
|
|
|
cbm_daemon_bootstrap_launch_spec_t spec;
|
|
bool spec_ready =
|
|
config->spawn_permanent
|
|
? cbm_daemon_bootstrap_launch_spec_init_permanent(config->executable_path, &spec)
|
|
: cbm_daemon_bootstrap_launch_spec_init(config->executable_path, &spec);
|
|
if (!spec_ready || !ops->startup_lock_prepare_handoff(ops->context, startup_lock) ||
|
|
!ops->spawn_daemon(ops->context, &spec)) {
|
|
probe = CBM_DAEMON_BOOTSTRAP_PROBE_ERROR;
|
|
break;
|
|
}
|
|
result_out->daemon_spawned = true;
|
|
|
|
/* Keep startup ownership only until the child becomes observable.
|
|
* Windows participant teardown reacquires the startup transition;
|
|
* retaining it while probing an observable generation can deadlock the
|
|
* bootstrap against a daemon that is trying to cleanly stand down. */
|
|
do {
|
|
bootstrap_pause(deadline);
|
|
probe = bootstrap_probe(config, ops, &client, &connect_result);
|
|
if (probe == CBM_DAEMON_BOOTSTRAP_PROBE_RESERVED ||
|
|
probe == CBM_DAEMON_BOOTSTRAP_PROBE_TERMINAL) {
|
|
generation_observed = true;
|
|
bootstrap_startup_lock_release_complete(ops, &startup_lock);
|
|
lock_acquired = false;
|
|
break;
|
|
}
|
|
if (!bootstrap_probe_is_waitable(probe)) {
|
|
break;
|
|
}
|
|
} while (cbm_now_ms() < deadline);
|
|
if (!lock_acquired) {
|
|
continue;
|
|
}
|
|
break;
|
|
}
|
|
|
|
if (lock_acquired) {
|
|
bootstrap_startup_lock_release_complete(ops, &startup_lock);
|
|
}
|
|
if (bootstrap_probe_is_finishable(probe)) {
|
|
cbm_daemon_bootstrap_status_t status =
|
|
bootstrap_finish_probe(probe, client, &connect_result, ops, result_out);
|
|
ops->cohort_release(ops->context, cohort);
|
|
return status;
|
|
}
|
|
|
|
result_out->status = CBM_DAEMON_BOOTSTRAP_FAILED;
|
|
if (generation_observed) {
|
|
(void)snprintf(result_out->message, sizeof(result_out->message),
|
|
"CBM daemon is active or starting but could not accept this client "
|
|
"within %u ms",
|
|
config->startup_timeout_ms);
|
|
} else {
|
|
(void)snprintf(result_out->message, sizeof(result_out->message),
|
|
"CBM daemon could not start within %u ms", config->startup_timeout_ms);
|
|
}
|
|
if (ops->visible_diagnostic) {
|
|
ops->visible_diagnostic(ops->context, result_out->message);
|
|
}
|
|
ops->cohort_release(ops->context, cohort);
|
|
return result_out->status;
|
|
}
|
|
|
|
cbm_daemon_bootstrap_probe_status_t cbm_daemon_bootstrap_classify_failed_connect(
|
|
const cbm_daemon_runtime_connect_result_t *connect_result, int lifetime_status) {
|
|
if (!connect_result) {
|
|
return CBM_DAEMON_BOOTSTRAP_PROBE_ERROR;
|
|
}
|
|
if (connect_result->status == CBM_DAEMON_RUNTIME_CONNECT_CONFLICT) {
|
|
return CBM_DAEMON_BOOTSTRAP_PROBE_CONFLICT;
|
|
}
|
|
if (connect_result->status == CBM_DAEMON_RUNTIME_CONNECT_REJECTED) {
|
|
if (strstr(connect_result->message, "stopping") ||
|
|
strstr(connect_result->message, "shutting down")) {
|
|
return CBM_DAEMON_BOOTSTRAP_PROBE_TERMINAL;
|
|
}
|
|
/* Capacity, admission, and other protocol-level rejections prove an
|
|
* existing generation answered. Never reinterpret them as absence. */
|
|
return CBM_DAEMON_BOOTSTRAP_PROBE_RESERVED;
|
|
}
|
|
if (lifetime_status == 1) {
|
|
return CBM_DAEMON_BOOTSTRAP_PROBE_RESERVED;
|
|
}
|
|
if (lifetime_status != 0) {
|
|
return CBM_DAEMON_BOOTSTRAP_PROBE_ERROR;
|
|
}
|
|
return connect_result->status == CBM_DAEMON_RUNTIME_CONNECT_ERROR
|
|
? CBM_DAEMON_BOOTSTRAP_PROBE_UNAVAILABLE
|
|
: CBM_DAEMON_BOOTSTRAP_PROBE_ERROR;
|
|
}
|
|
|
|
typedef struct bootstrap_production_cohort {
|
|
cbm_version_cohort_manager_t *manager;
|
|
cbm_version_cohort_lease_t *lease;
|
|
} bootstrap_production_cohort_t;
|
|
|
|
typedef struct {
|
|
bootstrap_production_cohort_t *cohort;
|
|
#ifdef _WIN32
|
|
DWORD spawn_error;
|
|
#elif defined(__APPLE__)
|
|
int spawn_error;
|
|
#endif
|
|
} bootstrap_production_context_t;
|
|
|
|
static cbm_daemon_bootstrap_probe_status_t bootstrap_production_probe(
|
|
void *context, const cbm_daemon_ipc_endpoint_t *endpoint,
|
|
const cbm_daemon_build_identity_t *identity, uint32_t timeout_ms,
|
|
cbm_daemon_runtime_client_t **client_out, cbm_daemon_runtime_connect_result_t *result_out) {
|
|
bootstrap_production_context_t *production = context;
|
|
if (!production || !production->cohort || !production->cohort->manager) {
|
|
return CBM_DAEMON_BOOTSTRAP_PROBE_ERROR;
|
|
}
|
|
cbm_version_cohort_daemon_presence_t claim =
|
|
cbm_version_cohort_daemon_claim_presence(production->cohort->manager);
|
|
if (claim == CBM_VERSION_COHORT_DAEMON_ABSENT) {
|
|
int lifetime = cbm_daemon_ipc_lifetime_reservation_probe(endpoint);
|
|
if (lifetime == 0) {
|
|
/* Do not spend the per-connect timeout polling a generation that
|
|
* both independent ownership signals prove absent. The startup
|
|
* lock and its mandatory re-probe serialize a concurrent launch. */
|
|
return CBM_DAEMON_BOOTSTRAP_PROBE_UNAVAILABLE;
|
|
}
|
|
if (lifetime != 1) {
|
|
return CBM_DAEMON_BOOTSTRAP_PROBE_ERROR;
|
|
}
|
|
} else if (claim != CBM_VERSION_COHORT_DAEMON_COORDINATED) {
|
|
return CBM_DAEMON_BOOTSTRAP_PROBE_ERROR;
|
|
}
|
|
|
|
*client_out = cbm_daemon_runtime_client_connect(endpoint, identity, timeout_ms, result_out);
|
|
if (*client_out) {
|
|
return CBM_DAEMON_BOOTSTRAP_PROBE_CONNECTED;
|
|
}
|
|
|
|
/* Ownership may turn over while the connection attempt is in flight.
|
|
* Re-observe both signals so disappearance is not sticky and a live or
|
|
* cleaning-up generation is never mistaken for absence. */
|
|
claim = cbm_version_cohort_daemon_claim_presence(production->cohort->manager);
|
|
if (claim == CBM_VERSION_COHORT_DAEMON_COORDINATED) {
|
|
return cbm_daemon_bootstrap_classify_failed_connect(result_out, 1);
|
|
}
|
|
if (claim != CBM_VERSION_COHORT_DAEMON_ABSENT) {
|
|
return CBM_DAEMON_BOOTSTRAP_PROBE_ERROR;
|
|
}
|
|
int lifetime_status = cbm_daemon_ipc_lifetime_reservation_probe(endpoint);
|
|
return cbm_daemon_bootstrap_classify_failed_connect(result_out, lifetime_status);
|
|
}
|
|
|
|
static cbm_version_cohort_status_t bootstrap_production_cohort_acquire(
|
|
void *context, const cbm_daemon_ipc_endpoint_t *endpoint,
|
|
const cbm_daemon_build_identity_t *identity, uint64_t deadline_ms,
|
|
cbm_daemon_bootstrap_cohort_t *cohort_out, cbm_daemon_conflict_t *conflict_out) {
|
|
*cohort_out = NULL;
|
|
bootstrap_production_cohort_t *cohort = calloc(1, sizeof(*cohort));
|
|
if (cohort) {
|
|
cohort->manager = cbm_version_cohort_manager_new(endpoint);
|
|
}
|
|
if (!cohort || !cohort->manager) {
|
|
free(cohort);
|
|
return CBM_VERSION_COHORT_IO;
|
|
}
|
|
cbm_version_cohort_status_t status = cbm_version_cohort_acquire(
|
|
cohort->manager, identity, deadline_ms, &cohort->lease, conflict_out);
|
|
if (status == CBM_VERSION_COHORT_CONFLICT) {
|
|
(void)cbm_version_cohort_log_conflict(conflict_out);
|
|
}
|
|
if (status == CBM_VERSION_COHORT_OK || cohort->lease) {
|
|
*cohort_out = cohort;
|
|
if (context) {
|
|
bootstrap_production_context_t *production = context;
|
|
production->cohort = cohort;
|
|
}
|
|
return status;
|
|
}
|
|
uint64_t cleanup_deadline = bootstrap_deadline_after(BOOTSTRAP_COORDINATION_CLEANUP_MS);
|
|
cbm_private_file_lock_status_t cleanup = CBM_PRIVATE_FILE_LOCK_OK;
|
|
while (cohort->manager) {
|
|
cleanup = cbm_version_cohort_manager_free(&cohort->manager);
|
|
if (!cohort->manager) {
|
|
break;
|
|
}
|
|
if (cbm_now_ms() >= cleanup_deadline) {
|
|
bootstrap_cleanup_fail_stop("cohort_manager_cleanup");
|
|
}
|
|
cbm_usleep(1000);
|
|
}
|
|
free(cohort);
|
|
return cleanup == CBM_PRIVATE_FILE_LOCK_OK ? status : CBM_VERSION_COHORT_IO;
|
|
}
|
|
|
|
static void bootstrap_production_cohort_release(void *context,
|
|
cbm_daemon_bootstrap_cohort_t opaque) {
|
|
bootstrap_production_context_t *production = context;
|
|
bootstrap_production_cohort_t *cohort = opaque;
|
|
if (!cohort) {
|
|
return;
|
|
}
|
|
uint64_t cleanup_deadline = bootstrap_deadline_after(BOOTSTRAP_COORDINATION_CLEANUP_MS);
|
|
while (cohort->lease) {
|
|
(void)cbm_version_cohort_lease_release(&cohort->lease);
|
|
if (!cohort->lease) {
|
|
break;
|
|
}
|
|
if (cbm_now_ms() >= cleanup_deadline) {
|
|
bootstrap_cleanup_fail_stop("cohort_lease_cleanup");
|
|
}
|
|
cbm_usleep(1000);
|
|
}
|
|
cleanup_deadline = bootstrap_deadline_after(BOOTSTRAP_COORDINATION_CLEANUP_MS);
|
|
while (cohort->manager) {
|
|
(void)cbm_version_cohort_manager_free(&cohort->manager);
|
|
if (!cohort->manager) {
|
|
break;
|
|
}
|
|
if (cbm_now_ms() >= cleanup_deadline) {
|
|
bootstrap_cleanup_fail_stop("cohort_manager_cleanup");
|
|
}
|
|
cbm_usleep(1000);
|
|
}
|
|
if (production && production->cohort == cohort) {
|
|
production->cohort = NULL;
|
|
}
|
|
free(cohort);
|
|
}
|
|
|
|
static int bootstrap_production_lock(void *context, const cbm_daemon_ipc_endpoint_t *endpoint,
|
|
cbm_daemon_bootstrap_lock_t *lock_out) {
|
|
(void)context;
|
|
cbm_daemon_ipc_startup_lock_t *lock = NULL;
|
|
int status = cbm_daemon_ipc_startup_lock_try_acquire(endpoint, &lock);
|
|
*lock_out = lock;
|
|
return status;
|
|
}
|
|
|
|
static bool bootstrap_production_unlock(void *context, cbm_daemon_bootstrap_lock_t *lock_io) {
|
|
(void)context;
|
|
if (!lock_io) {
|
|
return false;
|
|
}
|
|
cbm_daemon_ipc_startup_lock_t *lock = *lock_io;
|
|
bool released = cbm_daemon_ipc_startup_lock_release(&lock);
|
|
*lock_io = lock;
|
|
return released;
|
|
}
|
|
|
|
static bool bootstrap_production_handoff(void *context, cbm_daemon_bootstrap_lock_t lock) {
|
|
(void)context;
|
|
return cbm_daemon_ipc_startup_lock_prepare_handoff((cbm_daemon_ipc_startup_lock_t *)lock);
|
|
}
|
|
|
|
#ifdef _WIN32
|
|
static bool bootstrap_production_spawn(void *context,
|
|
const cbm_daemon_bootstrap_launch_spec_t *spec) {
|
|
bootstrap_production_context_t *production = context;
|
|
if (production) {
|
|
production->spawn_error = ERROR_SUCCESS;
|
|
}
|
|
if (!spec || !spec->detached || spec->inherit_standard_handles || spec->use_shell) {
|
|
if (production) {
|
|
production->spawn_error = ERROR_INVALID_PARAMETER;
|
|
}
|
|
return false;
|
|
}
|
|
char command_line[BOOTSTRAP_PATH_CAP * 2];
|
|
if (!cbm_build_win_cmdline(command_line, sizeof(command_line), spec->argv)) {
|
|
if (production) {
|
|
production->spawn_error = ERROR_INVALID_PARAMETER;
|
|
}
|
|
return false;
|
|
}
|
|
wchar_t *application = cbm_utf8_to_wide(spec->executable_path);
|
|
wchar_t *command = cbm_utf8_to_wide(command_line);
|
|
if (!application || !command) {
|
|
if (production) {
|
|
production->spawn_error = ERROR_NOT_ENOUGH_MEMORY;
|
|
}
|
|
free(application);
|
|
free(command);
|
|
return false;
|
|
}
|
|
STARTUPINFOW startup;
|
|
PROCESS_INFORMATION child;
|
|
ZeroMemory(&startup, sizeof(startup));
|
|
ZeroMemory(&child, sizeof(child));
|
|
startup.cb = sizeof(startup);
|
|
DWORD flags = DETACHED_PROCESS | CREATE_NEW_PROCESS_GROUP | CREATE_NO_WINDOW;
|
|
/* A managed frontend payload is intentionally contained in the permanent
|
|
* launcher's kill-on-close job. The account daemon outlives that one
|
|
* frontend and therefore uses breakaway when the containing job explicitly
|
|
* permits it. Do not request breakaway from an unrelated restrictive job:
|
|
* CreateProcess would fail and regress portable/package-manager payloads. */
|
|
BOOL in_job = FALSE;
|
|
JOBOBJECT_EXTENDED_LIMIT_INFORMATION job_limits;
|
|
memset(&job_limits, 0, sizeof(job_limits));
|
|
if (IsProcessInJob(GetCurrentProcess(), NULL, &in_job) && in_job &&
|
|
QueryInformationJobObject(NULL, JobObjectExtendedLimitInformation, &job_limits,
|
|
sizeof(job_limits), NULL) &&
|
|
(job_limits.BasicLimitInformation.LimitFlags &
|
|
(JOB_OBJECT_LIMIT_BREAKAWAY_OK | JOB_OBJECT_LIMIT_SILENT_BREAKAWAY_OK)) != 0) {
|
|
flags |= CREATE_BREAKAWAY_FROM_JOB;
|
|
}
|
|
BOOL created = CreateProcessW(application, command, NULL, NULL, FALSE, flags, NULL, NULL,
|
|
&startup, &child);
|
|
DWORD spawn_error = created ? ERROR_SUCCESS : GetLastError();
|
|
free(application);
|
|
free(command);
|
|
if (production) {
|
|
production->spawn_error = spawn_error;
|
|
}
|
|
if (!created) {
|
|
return false;
|
|
}
|
|
(void)CloseHandle(child.hThread);
|
|
(void)CloseHandle(child.hProcess);
|
|
return true;
|
|
}
|
|
#elif defined(__APPLE__)
|
|
static bool bootstrap_darwin_spawn_state_init(posix_spawn_file_actions_t *actions,
|
|
posix_spawnattr_t *attributes) {
|
|
if (posix_spawn_file_actions_init(actions) != 0) {
|
|
return false;
|
|
}
|
|
bool actions_ready =
|
|
posix_spawn_file_actions_addopen(actions, STDIN_FILENO, "/dev/null", O_RDWR, 0) == 0 &&
|
|
posix_spawn_file_actions_addopen(actions, STDOUT_FILENO, "/dev/null", O_RDWR, 0) == 0 &&
|
|
posix_spawn_file_actions_addopen(actions, STDERR_FILENO, "/dev/null", O_RDWR, 0) == 0;
|
|
if (!actions_ready) {
|
|
(void)posix_spawn_file_actions_destroy(actions);
|
|
return false;
|
|
}
|
|
if (posix_spawnattr_init(attributes) != 0) {
|
|
(void)posix_spawn_file_actions_destroy(actions);
|
|
return false;
|
|
}
|
|
sigset_t empty;
|
|
(void)sigemptyset(&empty);
|
|
short flags = POSIX_SPAWN_SETSIGMASK | POSIX_SPAWN_SETSID | POSIX_SPAWN_CLOEXEC_DEFAULT;
|
|
if (posix_spawnattr_setsigmask(attributes, &empty) != 0 ||
|
|
posix_spawnattr_setflags(attributes, flags) != 0) {
|
|
(void)posix_spawnattr_destroy(attributes);
|
|
(void)posix_spawn_file_actions_destroy(actions);
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
typedef struct {
|
|
pid_t pid;
|
|
} bootstrap_darwin_reaper_t;
|
|
|
|
static void *bootstrap_darwin_reap(void *opaque) {
|
|
bootstrap_darwin_reaper_t *reaper = opaque;
|
|
int status = 0;
|
|
pid_t waited;
|
|
do {
|
|
waited = waitpid(reaper->pid, &status, 0);
|
|
} while (waited < 0 && errno == EINTR);
|
|
free(reaper);
|
|
return NULL;
|
|
}
|
|
|
|
static bool bootstrap_darwin_reaper_start(pid_t daemon) {
|
|
bootstrap_darwin_reaper_t *reaper = malloc(sizeof(*reaper));
|
|
if (!reaper) {
|
|
return false;
|
|
}
|
|
reaper->pid = daemon;
|
|
pthread_attr_t attributes;
|
|
bool attributes_ready = pthread_attr_init(&attributes) == 0;
|
|
bool detached =
|
|
attributes_ready && pthread_attr_setdetachstate(&attributes, PTHREAD_CREATE_DETACHED) == 0;
|
|
pthread_t thread;
|
|
int thread_status =
|
|
detached ? pthread_create(&thread, &attributes, bootstrap_darwin_reap, reaper) : -1;
|
|
if (attributes_ready) {
|
|
(void)pthread_attr_destroy(&attributes);
|
|
}
|
|
if (thread_status != 0) {
|
|
free(reaper);
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
static bool bootstrap_production_spawn(void *context,
|
|
const cbm_daemon_bootstrap_launch_spec_t *spec) {
|
|
bootstrap_production_context_t *production = context;
|
|
if (production) {
|
|
production->spawn_error = 0;
|
|
}
|
|
if (!spec || !spec->detached || spec->inherit_standard_handles || spec->use_shell) {
|
|
return false;
|
|
}
|
|
posix_spawn_file_actions_t actions;
|
|
posix_spawnattr_t attributes;
|
|
if (!bootstrap_darwin_spawn_state_init(&actions, &attributes)) {
|
|
return false;
|
|
}
|
|
/* Calling library-heavy process creation between fork() and exec is not a
|
|
* safe boundary once another thread may own libc state. Darwin's spawn
|
|
* primitive can create the detached session and close inherited FDs in
|
|
* one operation, so launch it from the original process and retain only a
|
|
* tiny detached waiter to prevent a crashed daemon from becoming a
|
|
* long-lived zombie owned by the first frontend. */
|
|
pid_t daemon = 0;
|
|
int spawn_status = posix_spawn(&daemon, spec->executable_path, &actions, &attributes,
|
|
(char *const *)spec->argv, environ);
|
|
(void)posix_spawnattr_destroy(&attributes);
|
|
(void)posix_spawn_file_actions_destroy(&actions);
|
|
if (production) {
|
|
production->spawn_error = spawn_status;
|
|
}
|
|
if (spawn_status != 0) {
|
|
return false;
|
|
}
|
|
if (!bootstrap_darwin_reaper_start(daemon)) {
|
|
(void)kill(daemon, SIGKILL);
|
|
int status = 0;
|
|
while (waitpid(daemon, &status, 0) < 0 && errno == EINTR) {}
|
|
if (production) {
|
|
production->spawn_error = EAGAIN;
|
|
}
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
#else
|
|
static void bootstrap_child_close_fds(void) {
|
|
long open_max = sysconf(_SC_OPEN_MAX);
|
|
if (open_max < 0 || open_max > 1048576L) {
|
|
open_max = 65536L;
|
|
}
|
|
for (int fd = 3; fd < open_max; fd++) {
|
|
(void)close(fd);
|
|
}
|
|
}
|
|
|
|
static void bootstrap_daemon_grandchild(const cbm_daemon_bootstrap_launch_spec_t *spec) {
|
|
(void)umask(077);
|
|
sigset_t empty;
|
|
(void)sigemptyset(&empty);
|
|
(void)sigprocmask(SIG_SETMASK, &empty, NULL);
|
|
int null_fd = open("/dev/null", O_RDWR);
|
|
if (null_fd < 0 || dup2(null_fd, STDIN_FILENO) < 0 || dup2(null_fd, STDOUT_FILENO) < 0 ||
|
|
dup2(null_fd, STDERR_FILENO) < 0) {
|
|
_exit(127);
|
|
}
|
|
if (null_fd > STDERR_FILENO) {
|
|
(void)close(null_fd);
|
|
}
|
|
bootstrap_child_close_fds();
|
|
execv(spec->executable_path, (char *const *)spec->argv);
|
|
_exit(127);
|
|
}
|
|
|
|
static bool bootstrap_production_spawn(void *context,
|
|
const cbm_daemon_bootstrap_launch_spec_t *spec) {
|
|
(void)context;
|
|
if (!spec || !spec->detached || spec->inherit_standard_handles || spec->use_shell) {
|
|
return false;
|
|
}
|
|
pid_t first = fork();
|
|
if (first < 0) {
|
|
return false;
|
|
}
|
|
if (first == 0) {
|
|
if (setsid() < 0) {
|
|
_exit(127);
|
|
}
|
|
pid_t daemon = fork();
|
|
if (daemon < 0) {
|
|
_exit(127);
|
|
}
|
|
if (daemon > 0) {
|
|
_exit(0);
|
|
}
|
|
bootstrap_daemon_grandchild(spec);
|
|
}
|
|
|
|
int status = 0;
|
|
pid_t waited;
|
|
do {
|
|
waited = waitpid(first, &status, 0);
|
|
} while (waited < 0 && errno == EINTR);
|
|
return waited == first && WIFEXITED(status) && WEXITSTATUS(status) == 0;
|
|
}
|
|
#endif
|
|
|
|
static void bootstrap_production_diagnostic(void *context, const char *message) {
|
|
bootstrap_production_context_t *production = context;
|
|
#ifdef _WIN32
|
|
if (production && production->spawn_error != ERROR_SUCCESS) {
|
|
(void)fprintf(stderr, "codebase-memory-mcp: %s (daemon launch error %lu)\n",
|
|
message ? message : "daemon startup failed",
|
|
(unsigned long)production->spawn_error);
|
|
(void)fflush(stderr);
|
|
return;
|
|
}
|
|
#elif defined(__APPLE__)
|
|
if (production && production->spawn_error != 0) {
|
|
(void)fprintf(stderr, "codebase-memory-mcp: %s (daemon launch: %s)\n",
|
|
message ? message : "daemon startup failed",
|
|
strerror(production->spawn_error));
|
|
(void)fflush(stderr);
|
|
return;
|
|
}
|
|
#else
|
|
(void)production;
|
|
#endif
|
|
(void)fprintf(stderr, "codebase-memory-mcp: %s\n", message ? message : "daemon startup failed");
|
|
(void)fflush(stderr);
|
|
}
|
|
|
|
cbm_daemon_bootstrap_status_t cbm_daemon_bootstrap_execute(
|
|
const cbm_daemon_bootstrap_config_t *config, cbm_daemon_bootstrap_result_t *result_out) {
|
|
bootstrap_production_context_t context = {0};
|
|
const cbm_daemon_bootstrap_ops_t ops = {
|
|
.context = &context,
|
|
.cohort_acquire = bootstrap_production_cohort_acquire,
|
|
.cohort_release = bootstrap_production_cohort_release,
|
|
.probe = bootstrap_production_probe,
|
|
.startup_lock_try_acquire = bootstrap_production_lock,
|
|
.startup_lock_prepare_handoff = bootstrap_production_handoff,
|
|
.startup_lock_release = bootstrap_production_unlock,
|
|
.spawn_daemon = bootstrap_production_spawn,
|
|
.visible_diagnostic = bootstrap_production_diagnostic,
|
|
};
|
|
return cbm_daemon_bootstrap_execute_with_ops(config, &ops, result_out);
|
|
}
|