xtc_async.h

/*-
 * Copyright (c) 2026, The XTC Project
 * Use of this source code is governed by the ISC License.
 *
 * src/inc/xtc_async.h
 *      The L2 coroutine surface.  Stackful fibers via ucontext (M4),
 *      protothreads (always available), and the macros that let user
 *      code write sync-looking async logic.
 *
 *      See M4_CLAIMS.md.
 */
 
#ifndef XTC_ASYNC_H
#define XTC_ASYNC_H
 
#include "xtc_export.h"
 
#include <stdint.h>
 
#include "xtc_loop.h"
 
/*
 * Coroutine entry function.  The argument is the user's `arg`; the
 * return value is recovered by the awaiter via xtc_await().
 *
 * The signature uses intptr_t so any pointer or fixed-width integer
 * fits without per-call allocation.  Wider returns can be done by
 * setting an out-parameter inside the user struct passed via arg.
 */
typedef intptr_t (*xtc_coro_fn)(void *arg);
 
/*
 * PUBLIC: int      xtc_async __P((xtc_loop_t *, xtc_coro_fn, void *, xtc_task_t **));
 * PUBLIC: int      xtc_await __P((xtc_task_t *, intptr_t *));
 * PUBLIC: void     xtc_yield __P((void));
 * PUBLIC: void     xtc_yield_set_budget __P((xtc_loop_t *, int64_t));
 * PUBLIC: int      xtc_yield_check __P((void));
 * PUBLIC: int      xtc_yield_if_due __P((void));
 * PUBLIC: uint64_t xtc_yield_due_count __P((const xtc_loop_t *));
 * PUBLIC: size_t   xtc_stack_size __P((void));
 * PUBLIC: int      xtc_set_stack_size __P((size_t));
 */
 
/*
 * xtc_async --
 *      Spawn fn(arg) as a stackful coroutine.  Returns immediately
 *      with the task handle in *out_task.  The fiber starts in the
 *      scheduled state and runs on the next loop step.
 */
XTC_API int      xtc_async(xtc_loop_t *loop, xtc_coro_fn fn, void *arg,
                           xtc_task_t **out_task);
 
/*
 * xtc_await --
 *      Wait for `t` to complete and recover its return value into
 *      *result.  Must be called from inside a coroutine spawned via
 *      xtc_async (or from the main thread before xtc_loop_run).
 *
 *      Returns XTC_OK on success.
 */
XTC_API int      xtc_await(xtc_task_t *t, intptr_t *result);
 
/*
 * xtc_yield --
 *      From inside a coroutine, return control to the loop.  The next
 *      loop step resumes at the line after the call.
 */
XTC_API void     xtc_yield(void);
 
/*
 * Cooperative yield watchdog.  xtc has no forcible preemption, so a
 * long compute loop must cooperate.  Set a per-loop time budget with
 * xtc_yield_set_budget (ns; 0 disables, the default), then in the
 * compute loop call xtc_yield_if_due() -- it yields when the current
 * run quantum has exceeded the budget.  xtc_yield_check() is the
 * queryable form (1 == over budget) for embedders that want to react
 * differently (e.g. fire an abort token -> xtc_svr_call_abortable).
 * xtc_yield_due_count() reports how many times a task went over
 * budget on the loop (telemetry).  All are no-ops / 0 off a loop.
 */
XTC_API void     xtc_yield_set_budget(xtc_loop_t *loop, int64_t budget_ns);
XTC_API int      xtc_yield_check(void);
XTC_API int      xtc_yield_if_due(void);
XTC_API uint64_t xtc_yield_due_count(const xtc_loop_t *loop);
 
/*
 * Default fiber stack size in bytes.  Configurable per process via
 * xtc_set_stack_size().  M4 default: 64 KiB.
 */
XTC_API size_t   xtc_stack_size(void);
XTC_API int      xtc_set_stack_size(size_t bytes);
 
/*
 * Stack-memory reclamation on park (Lever S1, M_PREEMPTION section 8).
 *
 * A parked stackful fiber commits only the pages it touched but its
 * mmap'd stack RESERVES the full configured size.  With reclaim ON, a
 * fiber that parks (xtc_yield / recv / a latch or timer wait) returns
 * the UNUSED tail of its stack -- the region above its current stack
 * pointer, beyond a small live margin -- to the OS with
 * madvise(MADV_DONTNEED); it faults back as zero-fill on resume.  This
 * cuts the per-parked-fiber RAM floor for the many-idle-connection case
 * (measure with bench/bench_mem_per_task).  Predictable: no relocation,
 * no segmented-stack thrash.
 *
 * Portable: a no-op where madvise/MADV_DONTNEED is unavailable
 * (Windows) and on the Win32-fiber substrate (whose stacks the OS
 * owns).  OFF by default (the cooperative fast path is unchanged); a
 * process-wide opt-in.  keep_bytes is the live margin left mapped below
 * the saved SP (0 selects a sensible default of one page); the reclaim
 * only fires when the reclaimable tail exceeds a page, to avoid fault
 * churn on shallow, frequently-woken fibers.
 *
 * PUBLIC: int  xtc_stack_reclaim_enable __P((size_t));
 * PUBLIC: void xtc_stack_reclaim_disable __P((void));
 * PUBLIC: int  xtc_stack_reclaim_enabled __P((void));
 * PUBLIC: uint64_t xtc_stack_reclaim_count __P((void));
 */
XTC_API int      xtc_stack_reclaim_enable(size_t keep_bytes);
XTC_API void     xtc_stack_reclaim_disable(void);
XTC_API int      xtc_stack_reclaim_enabled(void);
XTC_API uint64_t xtc_stack_reclaim_count(void);
 
/*
 * XTC_COOP_REGION { ... } --
 *      A block that is guaranteed to run to completion without the
 *      scheduler interleaving another task between its statements.
 *      M4 implementation: the macro is a documentation marker only,
 *      because the M4 scheduler is single-threaded and never
 *      preempts a running coroutine outside an explicit yield point.
 *      M5 (multi-loop) hardens this by setting a per-task "do not
 *      steal" flag for the duration of the block.
 */
#define XTC_COOP_REGION  /* see xtc_async(3) */
 
/*
 * Protothread macros (constrained-platform fallback).  Bodies of
 * these functions cannot use stack-resident locals; lift them into a
 * state struct or static.  Documented in xtc_async(3).
 */
typedef struct xtc_pt {
        unsigned short lc;     /* local-continuation cookie */
} xtc_pt_t;
 
#define XTC_PT_THREAD(...)         char __VA_ARGS__
#define XTC_PT_INIT(pt)            ((pt)->lc = 0)
#define XTC_PT_BEGIN(pt)           switch ((pt)->lc) { case 0:
#define XTC_PT_END(pt)             } (pt)->lc = 0; return 2 /* DONE */
#define XTC_PT_YIELD(pt)           do {                                    \
        (pt)->lc = __LINE__; return 0 /* PENDING */;                       \
        case __LINE__:; } while (0)
#define XTC_PT_WAIT_UNTIL(pt, c)   do {                                    \
        (pt)->lc = __LINE__; case __LINE__:                                \
        if (!(c)) return 0; } while (0)
 
#define XTC_PT_DONE     2
#define XTC_PT_YIELDED  0
 
#endif /* XTC_ASYNC_H */