136 lines
2.7 KiB
C
136 lines
2.7 KiB
C
|
/*
|
|||
|
|
* 程序清单：内存池例程
|
|||
|
|
*
|
|||
|
|
* 这个程序会创建2个动态线程，一个静态的内存池对象，它们会试图分别从内存池中获得
|
|||
|
|
* 内存块
|
|||
|
|
*/
|
|||
|
|
#include <rtthread.h>
|
|||
|
|
#include "tc_comm.h"
|
|||
|
|
|
|||
|
|
static rt_uint8_t *ptr[48];
|
|||
|
|
static rt_uint8_t mempool[4096];
|
|||
|
|
static struct rt_mempool mp;
|
|||
|
|
|
|||
|
|
/* 指向线程控制块的指针 */
|
|||
|
|
static rt_thread_t tid1 = RT_NULL;
|
|||
|
|
static rt_thread_t tid2 = RT_NULL;
|
|||
|
|
|
|||
|
|
/* 线程1入口 */
|
|||
|
|
static void thread1_entry(void* parameter)
|
|||
|
|
{
|
|||
|
|
int i;
|
|||
|
|
char *block;
|
|||
|
|
|
|||
|
|
while(1)
|
|||
|
|
{
|
|||
|
|
for (i = 0; i < 48; i++)
|
|||
|
|
|
|||
|
|
{
|
|||
|
|
rt_kprintf("allocate No.%d\n", i);
|
|||
|
|
ptr[i] = rt_mp_alloc(&mp, RT_WAITING_FOREVER);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
block = rt_mp_alloc(&mp, RT_WAITING_FOREVER);
|
|||
|
|
rt_kprintf("allocate the block mem\n");
|
|||
|
|
rt_mp_free(block);
|
|||
|
|
block = RT_NULL;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/* 线程2入口 */
|
|||
|
|
static void thread2_entry(void *parameter)
|
|||
|
|
{
|
|||
|
|
int i;
|
|||
|
|
|
|||
|
|
while(1)
|
|||
|
|
{
|
|||
|
|
rt_kprintf("try to release block\n");
|
|||
|
|
|
|||
|
|
for (i = 0 ; i < 48; i ++)
|
|||
|
|
{
|
|||
|
|
if (ptr[i] != RT_NULL)
|
|||
|
|
{
|
|||
|
|
rt_kprintf("release block %d\n", i);
|
|||
|
|
|
|||
|
|
rt_mp_free(ptr[i]);
|
|||
|
|
|
|||
|
|
ptr[i] = RT_NULL;
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// rt_thread_delay(100);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
int mempool_simple_init()
|
|||
|
|
{
|
|||
|
|
int i;
|
|||
|
|
for (i = 0; i < 48; i ++) ptr[i] = RT_NULL;
|
|||
|
|
|
|||
|
|
rt_mp_init(&mp, "mp1", &mempool[0], sizeof(mempool), 80);
|
|||
|
|
|
|||
|
|
/* 创建线程1 */
|
|||
|
|
tid1 = rt_thread_create("t1",
|
|||
|
|
thread1_entry, RT_NULL, /* 线程入口是thread1_entry, 入口参数是RT_NULL */
|
|||
|
|
THREAD_STACK_SIZE, THREAD_PRIORITY, THREAD_TIMESLICE);
|
|||
|
|
if (tid1 != RT_NULL)
|
|||
|
|
rt_thread_startup(tid1);
|
|||
|
|
else
|
|||
|
|
tc_stat(TC_STAT_END | TC_STAT_FAILED);
|
|||
|
|
|
|||
|
|
/* 创建线程2 */
|
|||
|
|
tid2 = rt_thread_create("t2",
|
|||
|
|
thread2_entry, RT_NULL, /* 线程入口是thread2_entry, 入口参数是RT_NULL */
|
|||
|
|
THREAD_STACK_SIZE, THREAD_PRIORITY, THREAD_TIMESLICE);
|
|||
|
|
if (tid2 != RT_NULL)
|
|||
|
|
rt_thread_startup(tid2);
|
|||
|
|
else
|
|||
|
|
tc_stat(TC_STAT_END | TC_STAT_FAILED);
|
|||
|
|
|
|||
|
|
return 0;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
#ifdef RT_USING_TC
|
|||
|
|
static void _tc_cleanup()
|
|||
|
|
{
|
|||
|
|
/* 调度器上锁，上锁后，将不再切换到其他线程，仅响应中断 */
|
|||
|
|
rt_enter_critical();
|
|||
|
|
|
|||
|
|
/* 删除线程 */
|
|||
|
|
if (tid1 != RT_NULL && tid1->stat != RT_THREAD_CLOSE)
|
|||
|
|
rt_thread_delete(tid1);
|
|||
|
|
if (tid2 != RT_NULL && tid2->stat != RT_THREAD_CLOSE)
|
|||
|
|
rt_thread_delete(tid2);
|
|||
|
|
|
|||
|
|
/* 执行内存池脱离 */
|
|||
|
|
rt_mp_detach(&mp);
|
|||
|
|
|
|||
|
|
/* 调度器解锁 */
|
|||
|
|
rt_exit_critical();
|
|||
|
|
|
|||
|
|
/* 设置TestCase状态 */
|
|||
|
|
tc_done(TC_STAT_PASSED);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
int _tc_mempool_simple()
|
|||
|
|
{
|
|||
|
|
/* 设置TestCase清理回调函数 */
|
|||
|
|
tc_cleanup(_tc_cleanup);
|
|||
|
|
mempool_simple_init();
|
|||
|
|
|
|||
|
|
/* 返回TestCase运行的最长时间 */
|
|||
|
|
return 100;
|
|||
|
|
}
|
|||
|
|
/* 输出函数命令到finsh shell中 */
|
|||
|
|
FINSH_FUNCTION_EXPORT(_tc_mempool_simple, a memory pool example);
|
|||
|
|
#else
|
|||
|
|
/* 用户应用入口 */
|
|||
|
|
int rt_application_init()
|
|||
|
|
{
|
|||
|
|
mempool_simple_init();
|
|||
|
|
|
|||
|
|
return 0;
|
|||
|
|
}
|
|||
|
|
#endif
|