Kernel

关于RTT的消息队列

发布于 2018-05-03 18:56:18 浏览：4827 订阅该版

10 个回答

还没想好 2018-05-04

这家伙很懒，什么也没写！

coolouba 2018-05-04

这家伙很懒，什么也没写！

还没想好 2018-05-04

这家伙很懒，什么也没写！

coolouba 2018-05-04

这家伙很懒，什么也没写！

还没想好 2018-05-05

这家伙很懒，什么也没写！

```
#include <rtthread.h>

struct msg_queue
{
    rt_mailbox_t mb;
    rt_mp_t  mp;
    rt_size_t msg_size;
};
typedef struct msg_queue *msg_queue_t;

struct msg_des
{
    rt_size_t size;
    void *data;
};

msg_queue_t msg_queue_create(const char* name, rt_size_t msg_size, rt_size_t max_msgs, rt_uint8_t flag)
{
    msg_queue_t mq;
    
    if (msg_size == 0 || max_msgs == 0)
    {
        return RT_NULL;
    }

mq = rt_malloc(sizeof(struct msg_queue));
    if(mq == RT_NULL)
    {
        //TODO:log
        return RT_NULL;
    }
    rt_memset(mq, 0, sizeof(struct msg_queue));
    
    mq->mp = rt_mp_create(name, max_msgs, msg_size + sizeof(struct msg_des));
    if (mq->mp == RT_NULL)
    {
        rt_free(mq);
        return RT_NULL;
    }
    
    mq->mb = rt_mb_create(name, max_msgs, flag);
    if (mq->mb  == RT_NULL)
    {
        rt_mp_delete(mq->mp);
        rt_free(mq);
        return RT_NULL;
    }
    mq->msg_size = msg_size;

return mq;
}

rt_err_t msg_queue_send(msg_queue_t mq, void* buffer, rt_size_t size)
{
    struct msg_des *msg = RT_NULL;

if ((mq == RT_NULL) || (size == 0))
    {
        return -RT_ERROR;
    }

if (size > mq->msg_size)
    {
        size = mq->msg_size;
    }

msg = rt_mp_alloc(mq->mp, 0);
    if (msg == RT_NULL)
    {
        return -RT_ERROR;
    }

msg->data = (void *)(((rt_uint8_t *)msg) + sizeof(struct msg_des));
    rt_memcpy(msg->data, buffer, size);
    msg->size = size;

if (rt_mb_send(mq->mb, (rt_uint32_t)msg) != RT_EOK)
    {
        rt_mp_free(msg);
        return -RT_ERROR;
    }

return RT_EOK;
}

rt_err_t msg_queue_recv(msg_queue_t mq, void* buffer, rt_size_t size, rt_int32_t timeout)
{
    struct msg_des *msg = RT_NULL;

if ((mq == RT_NULL) || (size == 0))
    {
        return -RT_ERROR;
    }
    
    if (rt_mb_recv(mq->mb, (rt_uint32_t *)(&msg), timeout) != RT_EOK)
    {
        return -RT_ERROR;
    }

if (msg == RT_NULL)
    {
        return -RT_ERROR;
    }
    
    if (size > msg->size)
    {
        size =  msg->size;
    }
    rt_memcpy(buffer, msg->data, size);
    rt_mp_free(msg);
    
    return RT_EOK;
}

rt_err_t msg_queue_delete(msg_queue_t mq)
{
    if (mq == RT_NULL)
    {
        return -RT_ERROR;
    }
    
    rt_mp_delete(mq->mp);
    rt_mb_delete(mq->mb);
    
    rt_free(mq);
    
    return RT_EOK;
}

```