v3.2.2

UNIT_TESTS/c_plugin_test/forkrun_plugin.h

@@ -0,0 +1,36 @@
+// forkrun_plugin.h
+#ifndef FORKRUN_PLUGIN_H
+#define FORKRUN_PLUGIN_H
+
+#include <stdint.h>
+
+// Opt-in flag: define this in exactly ONE of your C files to receive the context
+// int forkrun_use_ctx = 1;
+
+struct forkrun_ctx {
+    uint64_t batch_index;       // global batch sequence number
+    uint64_t batch_offset;      // byte offset in input stream
+    uint64_t batch_byte_length; // length of current batch in bytes
+    uint32_t version;           // struct version, currently 1
+    uint32_t worker_id;         // internal worker ID
+    uint32_t node_id;           // NUMA node ID
+    uint32_t num_kills;         // retry count for this batch (if failure recovery active)
+    uint32_t numa_major;        // NUMA major sequence (0 if not NUMA)
+    uint32_t numa_minor;        // NUMA minor sequence (0 if not NUMA)
+    int32_t  fd_in;             // input memfd file descriptor
+    char     delimiter;         // batch delimiter character
+    uint8_t  cfg_state[3];      // global configuration state
+};
+
+
+/* === FUNCTION DEFINITION TEMPLATES ===
+ *
+ * 1. Standard Fast Path (2-arg):
+ *    int my_func(int argc, char **argv) { ... }
+ *
+ * 2. Context-Aware Path (3-arg):
+ *    int forkrun_use_ctx = 1;
+ *    int my_func(int argc, char **argv, const struct forkrun_ctx *ctx) { ... }
+ */
+
+#endif // FORKRUN_PLUGIN_H

UNIT_TESTS/c_plugin_test/plugin_ctx_math.c

@@ -0,0 +1,9 @@
+#include <stdio.h>
+#include "forkrun_plugin.h"
+int forkrun_use_ctx = 1;
+
+int plugin_ctx_math(int argc, char **argv, void *ctx_ptr) {
+    struct forkrun_ctx *ctx = (struct forkrun_ctx *)ctx_ptr;
+    printf("%lu\n", ctx->batch_byte_length);
+    return 0;
+}

UNIT_TESTS/c_plugin_test/plugin_echo.c

@@ -0,0 +1,5 @@
+#include <stdio.h>
+int plugin_echo(int argc, char **argv) {
+    for (int i = 0; i < argc; i++) printf("%s\n", argv[i]);
+    return 0;
+}

UNIT_TESTS/c_plugin_test/plugin_poison.c

@@ -0,0 +1,16 @@
+#include <stdio.h>
+#include "forkrun_plugin.h"
+int forkrun_use_ctx = 1;
+
+int plugin_poison(int argc, char **argv, void *ctx_ptr) {
+    struct forkrun_ctx *ctx = (struct forkrun_ctx *)ctx_ptr;
+
+    // Simulate a segfault/crash on batch index 7, ONLY on the first attempt
+    if (ctx->batch_index == 7 && ctx->num_kills == 0) {
+        return 1; // Trigger failure!
+    }
+
+    // On retry (num_kills > 0) or normal batches, process normally
+    for (int i = 0; i < argc; i++) printf("%s\n", argv[i]);
+    return 0;
+}

UNIT_TESTS/c_plugin_test/test_basic.c

@@ -0,0 +1,9 @@
+#include <stdio.h>
+#include <string.h>
+
+int test_basic(int argc, char **argv) {
+    size_t total = 0;
+    for (int i = 0; i < argc; i++) total += strlen(argv[i]);
+    printf("BASIC: batch_size=%d total_chars=%zu\n", argc, total);
+    return 0;
+}

UNIT_TESTS/c_plugin_test/test_ctx_header.c

@@ -0,0 +1,13 @@
+#include <stdio.h>
+#include "forkrun_plugin.h"
+
+int forkrun_use_ctx = 1;
+
+int test_ctx_header(int argc, char **argv, const struct forkrun_ctx *ctx) {
+    if (ctx && ctx->version >= 1) {
+        printf("CTX_HEADER: batch=%lu worker=%u node=%u retries=%u bytes=%lu\n",
+               ctx->batch_index, ctx->worker_id, ctx->node_id,
+               ctx->num_kills, ctx->batch_byte_length);
+    }
+    return 0;
+}

UNIT_TESTS/c_plugin_test/test_ctx_naked.c

@@ -0,0 +1,28 @@
+#include <stdio.h>
+#include <stdint.h>
+
+int forkrun_use_ctx = 1;
+
+struct forkrun_ctx {
+    uint64_t batch_index;
+    uint64_t batch_offset;
+    uint64_t batch_byte_length;
+    uint32_t version;
+    uint32_t worker_id;
+    uint32_t node_id;
+    uint32_t num_kills;
+    uint32_t numa_major;
+    uint32_t numa_minor;
+    int32_t  fd_in;
+    char     delimiter;
+    char     _pad[3];
+};
+
+int test_ctx_naked(int argc, char **argv, const struct forkrun_ctx *ctx) {
+    if (ctx && ctx->version >= 1) {
+        printf("CTX_NAKED: batch=%lu worker=%u node=%u retries=%u bytes=%lu\n",
+               ctx->batch_index, ctx->worker_id, ctx->node_id, ctx->num_kills,
+               ctx->batch_byte_length);
+    }
+    return 0;
+}