77 lines
2.7 KiB
C
77 lines
2.7 KiB
C
|
#pragma once
|
|||
|
|
|
|||
|
|
#include <common/numa.h>
|
|||
|
|
#include <process/proc-types.h>
|
|||
|
|
#include <common/err.h>
|
|||
|
|
#include <process/process.h>
|
|||
|
|
|
|||
|
|
struct process_control_block *kthread_create_on_node(int (*thread_fn)(void *data),
|
|||
|
|
void *data,
|
|||
|
|
int node,
|
|||
|
|
const char name_fmt[], ...);
|
|||
|
|
/**
|
|||
|
|
* @brief 在当前结点上创建一个内核线程
|
|||
|
|
*
|
|||
|
|
* @param thread_fn 该内核线程要执行的函数
|
|||
|
|
* @param data 传递给 thread_fn 的参数数据
|
|||
|
|
* @param name_fmt printf-style format string for the thread name
|
|||
|
|
* @param arg name_fmt的参数
|
|||
|
|
*
|
|||
|
|
* 请注意，该宏会创建一个内核线程，并将其设置为停止状态
|
|||
|
|
*/
|
|||
|
|
#define kthread_create(thread_fn, data, name_fmt, arg...) \
|
|||
|
|
kthread_create_on_node(thread_fn, data, NUMA_NO_NODE, name_fmt, ##arg)
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 创建内核线程，并将其唤醒
|
|||
|
|
*
|
|||
|
|
* @param thread_fn 该内核线程要执行的函数
|
|||
|
|
* @param data 传递给 thread_fn 的参数数据
|
|||
|
|
* @param name_fmt printf-style format string for the thread name
|
|||
|
|
* @param arg name_fmt的参数
|
|||
|
|
*/
|
|||
|
|
#define kthread_run(thread_fn, data, name_fmt, ...) \
|
|||
|
|
({ \
|
|||
|
|
struct process_control_block *__kt = kthread_create(thread_fn, data, name_fmt, ##__VA_ARGS__); \
|
|||
|
|
if (!IS_ERR(__kt)) \
|
|||
|
|
process_wakeup(__kt); \
|
|||
|
|
__kt; \
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 向kthread发送停止信号，请求其结束
|
|||
|
|
*
|
|||
|
|
* @param pcb 内核线程的pcb
|
|||
|
|
* @return int 错误码
|
|||
|
|
*/
|
|||
|
|
int kthread_stop(struct process_control_block * pcb);
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 内核线程调用该函数，检查自身的标志位，判断自己是否应该执行完任务后退出
|
|||
|
|
*
|
|||
|
|
* @return true 内核线程应该退出
|
|||
|
|
* @return false 无需退出
|
|||
|
|
*/
|
|||
|
|
bool kthread_should_stop(void);
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 让当前内核线程退出，并返回result参数给kthread_stop()函数
|
|||
|
|
*
|
|||
|
|
* @param result 返回值
|
|||
|
|
*/
|
|||
|
|
void kthread_exit(long result);
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 初始化kthread机制(只应被process_init调用)
|
|||
|
|
*
|
|||
|
|
* @return int 错误码
|
|||
|
|
*/
|
|||
|
|
int kthread_mechanism_init();
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 设置pcb中的worker_private字段（只应被设置一次）
|
|||
|
|
*
|
|||
|
|
* @param pcb pcb
|
|||
|
|
* @return bool 成功或失败
|
|||
|
|
*/
|
|||
|
|
bool kthread_set_worker_private(struct process_control_block *pcb);
|