sched_ext 介绍

// sched_ext 的主要 BPF 回调（struct sched_ext_ops 定义）
struct sched_ext_ops {
    // ===== 任务管理 =====
    int (*init)(void);                              // BPF 程序初始化
    void (*exit)(struct scx_exit_info *ei);         // BPF 程序退出
    s32 (*init_task)(struct task_struct *p,         // 任务初始化（选择默认 CPU/调度域）
                     struct scx_init_task_args *args);
    void (*exit_task)(struct task_struct *p);       // 任务退出清理

    // ===== 核心调度 =====
    s32 (*select_cpu)(struct task_struct *p,        // 选择 CPU（热路径）
                      s32 prev_cpu, u64 wake_flags);
    void (*enqueue)(struct task_struct *p,          // 任务进入就绪队列
                    u64 enq_flags);
    void (*dequeue)(struct task_struct *p,          // 任务移出就绪队列
                    u64 deq_flags);
    void (*dispatch)(s32 cpu, struct task_struct *p); // CPU 空闲时获取任务
    void (*running)(struct task_struct *p);         // 任务开始运行
    void (*stopping)(struct task_struct *p,         // 任务停止运行
                     bool runnable);

    // ===== 时间片与周期 =====
    void (*tick)(struct task_struct *p);            // 周期性 tick（时间片管理）

    // ===== CPU 状态 =====
    void (*cpu_online)(s32 cpu);
    void (*cpu_offline)(s32 cpu);

    // ===== 权重控制 =====
    void (*set_weight)(struct task_struct *p, u32 weight);  // nice 值变更
    void (*set_cpumask)(struct task_struct *p,      // CPU 亲和性变更
                        const struct cpumask *cpumask);

    // ===== 可选标志 =====
    u64 flags;  // SCX_OPS_* 标志控制行为
};

// 任务维度的调度元数据
struct task_ctx {
    u64 deadline_ns;
    u64 budget_ns;
    u32 priority;
    bool pinned_cpu;
};

struct {
    __uint(type, BPF_MAP_TYPE_TASK_STORAGE);
    __uint(map_flags, BPF_F_NO_PREALLOC);
    __type(key, int);
    __type(value, struct task_ctx);
} task_data SEC(".maps");

// per-CPU 调度队列
struct {
    __uint(type, BPF_MAP_TYPE_QUEUE);
    __type(value, struct task_struct *);
    __uint(max_entries, 1024);
} dispatch_queue SEC(".maps");

// 在 BPF arena 中分配自定义数据结构
struct sched_node {
    struct sched_node *next;
    struct sched_node *prev;  // 双向链表
    struct task_struct *task;
    u64 slice_start;
    u32 vruntime;
};

// Arena 内部分配
struct sched_node *node = bpf_arena_alloc(sizeof(*node));

// 任务入队 -> 按简单 FIFO 顺序
void BPF_STRUCT_OPS(simple_enqueue, struct task_struct *p, u64 enq_flags)
{
    if (enq_flags & SCX_ENQ_LOCAL) {
        // 本地 CPU 唤醒 -> 直接入本地队列
        scx_bpf_dispatch(p, SCX_DSQ_LOCAL, SCX_SLICE_DFL, enq_flags);
        return;
    }

    // 否则入全局 DSQ（任何 CPU 都可消费）
    scx_bpf_dispatch(p, SCX_DSQ_GLOBAL, SCX_SLICE_DFL, enq_flags);
}

// CPU 空闲 -> 从全局或本地 DSQ 取任务
void BPF_STRUCT_OPS(simple_dispatch, s32 cpu, struct task_struct *p)
{
    // 默认行为：从 SCX_DSQ_LOCAL（本地）或 SCX_DSQ_GLOBAL（全局）取任务
    // 这里使用默认实现即可
}