@kingsoftcloud/agentkit-harness-node 0.0.0-stage → 0.0.2

package/src/native/local-process-launcher.c

@@ -0,0 +1,183 @@
+/* Trusted POSIX exec substrate for the Node adapter. No workspace imports,
+ * shell parsing, policy decisions or scheduling. Build before distribution;
+ * never compile at install time. See local-process-manifest.json. */
+#define _POSIX_C_SOURCE 200809L
+#define _DARWIN_C_SOURCE 1
+#define _DEFAULT_SOURCE 1
+#include <errno.h>
+#include <inttypes.h>
+#if defined(__linux__)
+#include <dirent.h>
+#include <limits.h>
+#endif
+#include <poll.h>
+#include <signal.h>
+#include <stdint.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <sys/resource.h>
+#include <sys/types.h>
+#include <unistd.h>
+
+#define CONTRACT "agentkit.local-process-control/v1"
+#define MAX_EXACT UINT64_C(9007199254740991)
+struct limit { const char *name; int resource; uint64_t requested; rlim_t effective; int status; };
+static int number(const char *text, uint64_t *value) {
+    if (!text || !*text || *text == '0') return -1;
+    uint64_t n = 0;
+    for (const unsigned char *p = (const unsigned char *)text; *p; ++p) {
+        if (*p < '0' || *p > '9' || n > (MAX_EXACT - (*p - '0')) / 10) return -1;
+        n = n * 10 + (*p - '0');
+    }
+    *value = n; return 0;
+}
+static int write_all(int fd, const char *text, size_t length) {
+    while (length) {
+        ssize_t n = write(fd, text, length);
+        if (n < 0 && errno == EINTR) continue;
+        if (n <= 0) return -1;
+        text += n; length -= (size_t)n;
+    }
+    return 0;
+}
+#if defined(__linux__)
+/* RLIMIT_NPROC is scoped to the real user on Linux, so a shared runner can
+ * already be at the requested limit before this command starts. Count only
+ * this process group instead; the watcher itself is excluded from the cap. */
+static int group_processes(pid_t group) {
+    DIR *directory = opendir("/proc");
+    if (!directory) return -1;
+    int count = 0;
+    struct dirent *entry;
+    while ((entry = readdir(directory)) != NULL) {
+        char *end = NULL;
+        errno = 0;
+        long number = strtol(entry->d_name, &end, 10);
+        if (errno || end == entry->d_name || *end || number < 1 || number > INT_MAX) continue;
+        char path[64];
+        int length = snprintf(path, sizeof path, "/proc/%ld/stat", number);
+        if (length <= 0 || (size_t)length >= sizeof path) continue;
+        FILE *file = fopen(path, "r");
+        if (!file) continue;
+        char line[4096];
+        if (fgets(line, sizeof line, file)) {
+            char *closing = strrchr(line, ')');
+            char state;
+            long parent = 0, process_group = 0;
+            if (closing && sscanf(closing + 1, " %c %ld %ld", &state, &parent, &process_group) == 3
+                && (pid_t)process_group == group) count++;
+        }
+        fclose(file);
+    }
+    closedir(directory);
+    return count;
+}
+#endif
+static void watch_owner(int owner, int report, pid_t group, uint64_t max_processes) {
+#if !defined(__linux__)
+    (void)max_processes;
+#endif
+    close(report);
+    close(STDIN_FILENO); close(STDOUT_FILENO); close(STDERR_FILENO);
+    /* Hold the owned group through the caller's TERM->KILL cleanup window. */
+    signal(SIGTERM, SIG_IGN);
+    char byte;
+    for (;;) {
+#if defined(__linux__)
+        int count = group_processes(group);
+        if (count < 0 || (count > 1 && (uint64_t)(count - 1) > max_processes)) {
+            kill(-group, SIGKILL);
+            _exit(0);
+        }
+        struct pollfd item = { .fd = owner, .events = POLLIN | POLLHUP };
+        int ready = poll(&item, 1, 100);
+        if (ready < 0 && errno == EINTR) continue;
+        if (ready < 0) break;
+        if (ready == 0) continue;
+#else
+        ssize_t n = read(owner, &byte, 1);
+#endif
+#if defined(__linux__)
+        ssize_t n = read(owner, &byte, 1);
+#endif
+        if (n > 0) continue;
+        if (n < 0 && errno == EINTR) continue;
+        if (n < 0 && (errno == EAGAIN || errno == EWOULDBLOCK)) {
+            struct pollfd item = { .fd = owner, .events = POLLIN | POLLHUP };
+            if (poll(&item, 1, -1) < 0 && errno != EINTR) break;
+            continue;
+        }
+        break;
+    }
+    kill(-group, SIGKILL);
+    _exit(0);
+}
+int main(int argc, char **argv) {
+    /* cpu, address bytes, process count, open files, file bytes, report fd,
+     * owner fd, --, executable, ... . All numbers are strict decimal data. */
+    if (argc < 10 || strcmp(argv[8], "--") != 0) return 125;
+    uint64_t values[7];
+    for (int i = 0; i < 7; ++i) if (number(argv[i + 1], &values[i]) < 0) return 125;
+    if (values[5] < 3 || values[5] > 1024 || values[6] < 3 || values[6] > 1024 || values[5] == values[6]) return 125;
+    int report = (int)values[5], owner = (int)values[6];
+    if (getpgrp() != getpid()) return 125;
+    signal(SIGPIPE, SIG_IGN);
+    struct limit limits[5] = {
+        { "cpuSeconds", RLIMIT_CPU, values[0], 0, 0 },
+        { "addressSpaceBytes", RLIMIT_AS, values[1], 0, 0 },
+        { "maxProcesses", RLIMIT_NPROC, values[2], 0, 0 },
+        { "maxOpenFiles", RLIMIT_NOFILE, values[3], 0, 0 },
+        { "maxFileBytes", RLIMIT_FSIZE, values[4], 0, 0 },
+    };
+    /* Fork the owner watchdog before applying the child resource limits.
+     * On Linux its process-group counter replaces RLIMIT_NPROC, whose
+     * per-user scope is unsafe on shared runners. */
+    pid_t group = getpgrp();
+    pid_t watcher = fork();
+    if (watcher == 0) watch_owner(owner, report, group, values[2]);
+    int failed = watcher < 0, unsupported = 0;
+    for (int i = 0; i < 5; ++i) {
+        struct limit *item = &limits[i];
+#if defined(__APPLE__)
+        /* Match the reference's declared Darwin exceptions exactly. */
+        if (i == 1 || i == 2) { item->status = 2; unsupported++; continue; }
+#elif defined(__linux__)
+        /* RLIMIT_NPROC is per-user and cannot safely cap one command on a
+         * shared host. The owner watchdog enforces a process-group bound. */
+        if (i == 2) { item->effective = item->requested; item->status = 0; continue; }
+#endif
+        struct rlimit before, after;
+        if (getrlimit(item->resource, &before) < 0) { item->status = 1; failed++; continue; }
+        item->effective = (rlim_t)item->requested;
+        if ((uint64_t)item->effective != item->requested) { item->status = 1; failed++; continue; }
+        if (before.rlim_max != RLIM_INFINITY && before.rlim_max < item->effective) item->effective = before.rlim_max;
+        struct rlimit desired = { .rlim_cur = item->effective, .rlim_max = item->effective };
+        if (setrlimit(item->resource, &desired) < 0 || getrlimit(item->resource, &after) < 0
+            || after.rlim_cur != item->effective || after.rlim_max != item->effective) { item->status = 1; failed++; }
+    }
+    char buffer[4096];
+    int at = snprintf(buffer, sizeof buffer, "{\"schemaVersion\":\"%s\",\"status\":\"%s\",\"watcherPid\":%ld,\"applied\":{",
+        CONTRACT, failed ? "failed" : unsupported ? "partial" : "applied", (long)watcher);
+    int count = 0;
+    for (int i = 0; i < 5; ++i) if (limits[i].status == 0)
+        at += snprintf(buffer + at, sizeof buffer - (size_t)at, "%s\"%s\":%" PRIu64,
+            count++ ? "," : "", limits[i].name, (uint64_t)limits[i].effective);
+    at += snprintf(buffer + at, sizeof buffer - (size_t)at, "},\"unsupported\":["); count = 0;
+    for (int i = 0; i < 5; ++i) if (limits[i].status == 2)
+        at += snprintf(buffer + at, sizeof buffer - (size_t)at, "%s\"%s\"", count++ ? "," : "", limits[i].name);
+    at += snprintf(buffer + at, sizeof buffer - (size_t)at, "],\"failed\":["); count = 0;
+    for (int i = 0; i < 5; ++i) if (limits[i].status == 1)
+        at += snprintf(buffer + at, sizeof buffer - (size_t)at, "%s\"%s\"", count++ ? "," : "", limits[i].name);
+    if (watcher < 0 && !count && failed) at += snprintf(buffer + at, sizeof buffer - (size_t)at, "\"owner_watchdog\"");
+    at += snprintf(buffer + at, sizeof buffer - (size_t)at, "]}\n");
+    if (at <= 0 || (size_t)at >= sizeof buffer || write_all(report, buffer, (size_t)at) < 0) return 125;
+    close(report); close(owner);
+    if (failed) return 125;
+    /* SIGPIPE was changed only for the trusted report write. Restore default
+     * before executing a program; its own signal behavior remains its choice. */
+    signal(SIGPIPE, SIG_DFL);
+    execvp(argv[9], &argv[9]);
+    fputs("local-process launcher exec failed\n", stderr);
+    return 126;
+}

package/src/native/local-process-licenses/GCC-COPYING.RUNTIME

@@ -0,0 +1,72 @@
+GCC RUNTIME LIBRARY EXCEPTION
+
+Version 3.1, 31 March 2009
+
+Copyright (C) 2009 Free Software Foundation, Inc. <http://fsf.org/>
+
+Everyone is permitted to copy and distribute verbatim copies of this
+license document, but changing it is not allowed.
+
+This GCC Runtime Library Exception ("Exception") is an additional
+permission under section 7 of the GNU General Public License, version
+3 ("GPLv3"). It applies to a given file (the "Runtime Library") that
+bears a notice placed by the copyright holder of the file stating that
+the file is governed by GPLv3 along with this Exception.
+
+When you use GCC to compile a program, GCC may combine portions of
+certain GCC header files and runtime libraries with the compiled
+program. The purpose of this Exception is to allow compilation of
+non-GPL (including proprietary) programs to use, in this way, the
+header files and runtime libraries covered by this Exception.
+
+0. Definitions.
+
+A file is an "Independent Module" if it either requires the Runtime
+Library for execution after a Compilation Process, or makes use of an
+interface provided by the Runtime Library, but is not otherwise based
+on the Runtime Library.
+
+"GCC" means a version of the GNU Compiler Collection, with or without
+modifications, governed by version 3 (or a specified later version) of
+the GNU General Public License (GPL) with the option of using any
+subsequent versions published by the FSF.
+
+"GPL-compatible Software" is software whose conditions of propagation,
+modification and use would permit combination with GCC in accord with
+the license of GCC.
+
+"Target Code" refers to output from any compiler for a real or virtual
+target processor architecture, in executable form or suitable for
+input to an assembler, loader, linker and/or execution
+phase. Notwithstanding that, Target Code does not include data in any
+format that is used as a compiler intermediate representation, or used
+for producing a compiler intermediate representation.
+
+The "Compilation Process" transforms code entirely represented in
+non-intermediate languages designed for human-written code, and/or in
+Java Virtual Machine byte code, into Target Code. Thus, for example,
+use of source code generators and preprocessors need not be considered
+part of the Compilation Process, since the Compilation Process can be
+understood as starting with the output of the generators or
+preprocessors.
+
+A Compilation Process is "Eligible" if it is done using GCC, alone or
+with other GPL-compatible software, or if it is done without using any
+work based on GCC. For example, using non-GPL-compatible Software to
+optimize any GCC intermediate representations would not qualify as an
+Eligible Compilation Process.
+
+1. Grant of Additional Permission.
+
+You have permission to propagate a work of Target Code formed by
+combining the Runtime Library with Independent Modules, even if such
+propagation would otherwise violate the terms of GPLv3, provided that
+all Target Code was generated by Eligible Compilation Processes. You
+may then convey such a combination under terms of your choice,
+consistent with the licensing of the Independent Modules.
+
+2. No Weakening of GCC Copyleft.
+
+The availability of this Exception does not imply any general
+presumption that third-party software is unaffected by the copyleft
+requirements of the license of GCC.