msgbox.c

Rabbit 32Bit RTOS源代码

/*********************************************************************
*                   Copyright (c) 2011-2012,李士伟
*                     All rights reserved.
*文 件 名：msgbox.c
*描    述：消息邮箱源文件
*当前版本：V1.00
*作    者：李士伟
*创建日期：2011.12.15
**********************************************************************/
#include <kernel\msgbox.h>
#include <kernel\flag.h>
#include <kernel\sched.h>
#include <kernel\kd.h>
#include <kernel\asm.h>

/*********************************************************************
*函 数 名：API_InitMsgBox
*描    述：初始化消息邮箱
*输入参数：msgbox: 邮箱指针
           msg:    消息指针
*输出参数：无
*返 回 值：无
*注    意：
**********************************************************************/
void API_InitMsgBox(OS_MSGBOX *msgbox, void *msg)
{
    OS_ENTER_CRITICAL();
    msgbox->MsgPtr  = msg;
    msgbox->WaitQ   = NULL;
    OS_EXIT_CRITICAL();
}

/*********************************************************************
*函 数 名：API_MsgBoxPend
*描    述：请求消息邮箱
*输入参数：msgbox:  邮箱指针
           timeout: 等待时间，timeout=0，直到获得消息返回     
*输出参数：err:     错误标志
*返 回 值：消息指针
*注    意：
**********************************************************************/
void *API_MsgBoxPend(OS_MSGBOX *msgbox, INT32U timeout, INT32U *err)
{
    void *msg;

    OS_ENTER_CRITICAL();
    if ((msgbox == NULL) || (OSIntNesting > 0))
    {
        OS_EXIT_CRITICAL();
        *err = OS_ERR;
        return NULL;
    }

    msg = msgbox->MsgPtr;
    if (msg != NULL)  /* 邮箱有消息 */
    {
        msgbox->MsgPtr = NULL;  /* 清空邮箱 */

        OS_EXIT_CRITICAL();
        *err = OS_NO_ERR;
        return msg;  /* 返回邮箱消息指针 */
    }

    /* 等待邮箱消息 */
    OSTCBCur->Delay = timeout;  /* 设置超时等待时间 */
    OS_EnterWaitQHead(&(msgbox->WaitQ), OSTCBCur);  /* 加入消息邮箱的等待队列 */
    OS_Sched();

    if (OSTCBCur->MsgPtr != NULL)  /* 获得消息 */
    {
        msg = OSTCBCur->MsgPtr;   /* 获得消息指针 */
        OSTCBCur->MsgPtr = NULL;  /* 清除TCB消息指针，为下次等待消息准备 */
        OSTCBCur->Delay = 0;  /* 清除等待时限 */
        OS_EXIT_CRITICAL();
        *err = OS_NO_ERR;
        return msg;
    }

    /* 超时，未获得消息 */
    OS_EXIT_CRITICAL();
    *err = OS_OVERTIME;
    return NULL;
}

/*********************************************************************
*函 数 名：OS_MsgBoxPost
*描    述：发送消息
*输入参数：msgbox:  邮箱指针
           msg:     消息指针
           method:  发送方式
                   0=OS_BASE_PRIO  发送给一个等待该事件的高优先级任务
                   1=OS_BASE_BROADCAST 发送所有等待该事件的任务
*输出参数：无
*返 回 值：OS_MSGBOX_INVALID   邮箱不存在
           OS_MSGBOX_FULL      邮箱已满
           OS_MSGBOX_SAVE      没有任务在等待消息,消息存入邮箱
           OS_MSGBOX_SEND      有任务在等待消息，消息发送给任务
*注    意：
**********************************************************************/
INT32U API_MsgBoxPost(OS_MSGBOX *msgbox, void *msg, INT32U method)
{
    INT32U prio = 0;
    OS_TCB *ptcb;
    OS_TCB *wptcb=NULL;

    if (msgbox == NULL)
    {
        return OS_MSGBOX_INVALID;
    }

    OS_ENTER_CRITICAL();
    
    ptcb  = msgbox->WaitQ;
    wptcb = ptcb;
    if (ptcb != NULL)  /* 有任务在等待消息 */
    {
        prio = ptcb->Prio;
    }
    else  /* 没有任务在等待消息 */
    {
        if (msgbox->MsgPtr != NULL)  /* 邮箱存在消息, 不保存消息 */
        {
            OS_EXIT_CRITICAL();
            return OS_MSGBOX_FULL;
        }
        else  /* 邮箱不存在消息，将消息存入邮箱 */
        {
            msgbox->MsgPtr = msg;
            OS_EXIT_CRITICAL();
            return OS_MSGBOX_SAVE;
        }
    }

    /* 基于优先级发送消息给一个高优先级任务 */
    if (method == OS_BASE_PRIO)  
    {
        while (ptcb != NULL)  /* 从等待队列查询一个高优先级任务 */
        {
            if (prio <= ptcb->Prio)
            {
                prio = ptcb->Prio;
                wptcb= ptcb;
            }
            ptcb = ptcb->QNext;
        } 

        wptcb->MsgPtr  = msg;  /* 将消息指针存入TCB中 */
        OS_EnterRunQ(wptcb);  /* 加入运行队列 */

    }
    /* 广播方式唤醒所有等待该消息邮箱的任务 */
    else if (method == OS_BASE_BROADCAST)  
    {
        while (ptcb != NULL)
        {
            ptcb->MsgPtr = msg;  /* 将消息指针存入TCB中 */
            wptcb = ptcb->QNext;
            OS_EnterRunQ(ptcb);  /* 加入运行队列 */
            ptcb = wptcb;
        }
    }

    OS_EXIT_CRITICAL();
    OS_Sched();
    return OS_MSGBOX_SEND;
}

/*********************************************************************
*函 数 名：API_MsgBoxAccept
*描    述：无等待获取邮箱消息
*输入参数：msgbox: 邮箱指针
*输出参数：无
*返 回 值：消息指针
*注    意：
**********************************************************************/
void *API_MsgBoxAccept(OS_MSGBOX *msgbox)
{
    void *msg;

    if (msgbox == NULL)
    {
        return NULL;
    }

    OS_ENTER_CRITICAL();
    msg = msgbox->MsgPtr;
    msgbox->MsgPtr = NULL;
    OS_EXIT_CRITICAL();

    return msg;
}