101 lines
2.0 KiB
C
101 lines
2.0 KiB
C
|
/**
|
|||
|
|
* @file semaphore.h
|
|||
|
|
* @author fslngjin (lonjin@RinGoTek.cn)
|
|||
|
|
* @brief 信号量
|
|||
|
|
* @version 0.1
|
|||
|
|
* @date 2022-04-12
|
|||
|
|
*
|
|||
|
|
* @copyright Copyright (c) 2022
|
|||
|
|
*
|
|||
|
|
*/
|
|||
|
|
|
|||
|
|
#pragma once
|
|||
|
|
#include <process/atomic.h>
|
|||
|
|
|
|||
|
|
#include <process/process.h>
|
|||
|
|
#include <sched/sched.h>
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 信号量的等待队列
|
|||
|
|
*
|
|||
|
|
*/
|
|||
|
|
typedef struct
|
|||
|
|
{
|
|||
|
|
struct List wait_list;
|
|||
|
|
struct process_control_block *pcb;
|
|||
|
|
} wait_queue_node_t;
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 初始化信号量的等待队列
|
|||
|
|
*
|
|||
|
|
* @param wait_queue 等待队列
|
|||
|
|
* @param pcb pcb
|
|||
|
|
*/
|
|||
|
|
void wait_queue_init(wait_queue_node_t *wait_queue, struct process_control_block *pcb)
|
|||
|
|
{
|
|||
|
|
list_init(&wait_queue->wait_list);
|
|||
|
|
wait_queue->pcb = pcb;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 信号量的结构体
|
|||
|
|
*
|
|||
|
|
*/
|
|||
|
|
typedef struct
|
|||
|
|
{
|
|||
|
|
atomic_t counter;
|
|||
|
|
wait_queue_node_t wait_queue;
|
|||
|
|
} semaphore_t;
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 初始化信号量
|
|||
|
|
*
|
|||
|
|
* @param sema 信号量对象
|
|||
|
|
* @param count 信号量的初始值
|
|||
|
|
*/
|
|||
|
|
void semaphore_init(semaphore_t *sema, ul count)
|
|||
|
|
{
|
|||
|
|
atomic_set(&sema->counter, count);
|
|||
|
|
wait_queue_init(&sema->wait_queue, NULL);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 信号量down
|
|||
|
|
*
|
|||
|
|
* @param sema
|
|||
|
|
*/
|
|||
|
|
void semaphore_down(semaphore_t *sema)
|
|||
|
|
{
|
|||
|
|
if (atomic_read(&sema->counter) > 0) // 信号量大于0，资源充足
|
|||
|
|
atomic_dec(&sema->counter);
|
|||
|
|
else // 资源不足，进程休眠
|
|||
|
|
{
|
|||
|
|
// 将当前进程加入信号量的等待队列
|
|||
|
|
wait_queue_node_t wait;
|
|||
|
|
wait_queue_init(&wait, current_pcb);
|
|||
|
|
|
|||
|
|
current_pcb->state = PROC_UNINTERRUPTIBLE;
|
|||
|
|
|
|||
|
|
list_append(&sema->wait_queue.wait_list, &wait.wait_list);
|
|||
|
|
|
|||
|
|
// 执行调度
|
|||
|
|
sched_cfs();
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
void semaphore_up(semaphore_t *sema)
|
|||
|
|
{
|
|||
|
|
if (list_empty(&sema->wait_queue.wait_list)) // 没有进程在等待资源
|
|||
|
|
{
|
|||
|
|
atomic_inc(&sema->counter);
|
|||
|
|
}
|
|||
|
|
else // 有进程在等待资源，唤醒进程
|
|||
|
|
{
|
|||
|
|
|
|||
|
|
wait_queue_node_t *wq = container_of(list_next(&sema->wait_queue.wait_list), wait_queue_node_t, wait_list);
|
|||
|
|
list_del(&wq->wait_list);
|
|||
|
|
|
|||
|
|
wq->pcb->state = PROC_RUNNING;
|
|||
|
|
sched_cfs_enqueue(wq->pcb);
|
|||
|
|
}
|
|||
|
|
}
|