108 lines
2.9 KiB
C
Executable File
108 lines
2.9 KiB
C
Executable File
/*
|
||
* @Author: zhangzheng 1358745329@qq.com
|
||
* @Date: 2023-08-14 09:47:54
|
||
* @LastEditors: zhangzheng 1358745329@qq.com
|
||
* @LastEditTime: 2023-08-18 16:21:48
|
||
* @FilePath: /mkrtos-real/mkrtos_knl/inc/knl/thread.h
|
||
* @Description: 线程管理相关
|
||
*/
|
||
#pragma once
|
||
|
||
#include "kobject.h"
|
||
#include "scheduler.h"
|
||
#include "util.h"
|
||
#include "arch.h"
|
||
#include "ref.h"
|
||
#define THREAD_BLOCK_SIZE 0x400 //!< 线程块大小，栈在块的顶部
|
||
|
||
enum thread_state
|
||
{
|
||
THREAD_IDLE,
|
||
THREAD_DEAD,
|
||
THREAD_SUSPEND, //!< 只有接收和发送ipc消息时才能挂起
|
||
THREAD_READY,
|
||
};
|
||
typedef struct
|
||
{
|
||
umword_t rg0[4]; //!< r0-r3
|
||
umword_t r12;
|
||
umword_t lr;
|
||
umword_t pc;
|
||
umword_t xpsr;
|
||
} pf_s_t;
|
||
typedef struct pf
|
||
{
|
||
umword_t rg1[8]; //!< r4-r11
|
||
pf_s_t pf_s;
|
||
} pf_t;
|
||
|
||
typedef struct sp_info
|
||
{
|
||
void *user_sp; //!< 用户态的sp
|
||
void *knl_sp; //!< 内核sp
|
||
mword_t sp_type; //!< 使用的栈类型
|
||
} sp_info_t;
|
||
|
||
#define THREAD_MSG_BUG_LEN 128 //!< 默认的消息寄存器大小
|
||
#define MSG_BUF_HAS_DATA_FLAGS 0x01U //!< 已经有数据了
|
||
typedef struct msg_buf
|
||
{
|
||
void *msg; //!< buf，长度 @see THREAD_MSG_BUG_LEN
|
||
uhmword_t len; //!< 这里不是buf的大小，而是存储接收或者发送的长度
|
||
uhmword_t flags; //!< 传输标志
|
||
} msg_buf_t;
|
||
|
||
#define THREAD_MAIGC 0xdeadead
|
||
typedef struct thread
|
||
{
|
||
kobject_t kobj; //!< 内核对象节点
|
||
slist_head_t wait; //!< 用于等待队列
|
||
sched_t sche; //!< 调度节点
|
||
kobject_t *task; //!< 绑定的task
|
||
sp_info_t sp; //!< sp信息
|
||
ram_limit_t *lim; //!< 内存限制
|
||
ref_counter_t ref; //!< 引用计数
|
||
msg_buf_t msg; //!< 每个线程独有的消息缓存区
|
||
enum thread_state status; //!< 线程状态
|
||
umword_t magic; //!< maigc
|
||
} thread_t;
|
||
|
||
static inline void thread_set_msg_bug(thread_t *th, void *msg, uint16_t len)
|
||
{
|
||
th->msg.msg = msg;
|
||
th->msg.len = len;
|
||
th->msg.flags = 0;
|
||
}
|
||
|
||
static inline enum thread_state thread_get_status(thread_t *th)
|
||
{
|
||
return th->status;
|
||
}
|
||
static inline pf_t *thread_get_pf(thread_t *th)
|
||
{
|
||
uint8_t *bottom = (uint8_t *)th;
|
||
|
||
return ((pf_t *)(bottom + THREAD_BLOCK_SIZE)) - 1;
|
||
}
|
||
|
||
static inline thread_t *thread_get_current(void)
|
||
{
|
||
umword_t sp = arch_get_sp();
|
||
thread_t *th = (thread_t *)(ALIGN_DOWN(sp, THREAD_BLOCK_SIZE));
|
||
|
||
return th;
|
||
}
|
||
|
||
static inline pf_t *thread_get_current_pf(void)
|
||
{
|
||
return thread_get_pf(thread_get_current());
|
||
}
|
||
void thread_init(thread_t *th, ram_limit_t *lim);
|
||
void thread_set_exc_regs(thread_t *th, umword_t pc, umword_t user_sp, umword_t ram);
|
||
thread_t *thread_create(ram_limit_t *ram);
|
||
void thread_bind(thread_t *th, kobject_t *tk);
|
||
void thread_unbind(thread_t *th);
|
||
|
||
void thread_sched(void);
|
||
void thread_suspend(thread_t *th);
|
||
void thread_ready(thread_t *th, bool_t is_sche); |