[utest]add mutex_pi_tc

* add mutex_pi
* fix assertion error when thread wakeup while waiting mutex

2 files changed, 354 insertions, 1 deletions

diff --git a/examples/utest/testcases/kernel/SConscript b/examples/utest/testcases/kernel/SConscript
index f6270a5490..820f1ae042 100644
--- a/examples/utest/testcases/kernel/SConscript
+++ b/examples/utest/testcases/kernel/SConscript
@@ -33,7 +33,7 @@ if GetDepend(['UTEST_SIGNAL_TC']):
     src += ['signal_tc.c']
 
 if GetDepend(['UTEST_MUTEX_TC']):
-    src += ['mutex_tc.c']
+    src += ['mutex_tc.c', 'mutex_pi_tc.c']
 
 if GetDepend(['UTEST_MAILBOX_TC']):
     src += ['mailbox_tc.c']
diff --git a/examples/utest/testcases/kernel/mutex_pi_tc.c b/examples/utest/testcases/kernel/mutex_pi_tc.c
new file mode 100644
index 0000000000..b49e26e22f
--- /dev/null
+++ b/examples/utest/testcases/kernel/mutex_pi_tc.c
@@ -0,0 +1,353 @@
+/*
+ * Copyright (c) 2006-2019, RT-Thread Development Team
+ *
+ * SPDX-License-Identifier: Apache-2.0
+ *
+ * Change Logs:
+ * Date           Author       Notes
+ */
+#define __RT_IPC_SOURCE__
+
+#include <rtthread.h>
+#include <stdlib.h>
+#include "utest.h"
+
+#ifdef ARCH_CPU_64BIT
+#define THREAD_STACKSIZE 8192
+#else
+#define THREAD_STACKSIZE 4096
+#endif
+
+#define MUTEX_NUM 3
+#define THREAD_NUM 5
+
+static struct rt_mutex _mutex[MUTEX_NUM];
+static volatile int _sync_flag;
+
+static void test_thread_entry(void *para)
+{
+    while (!_sync_flag)
+    {
+        rt_thread_delay(1);
+    }
+
+    rt_ubase_t thread_id = (rt_ubase_t)para;
+    rt_err_t ret;
+    rt_thread_mdelay(50 + thread_id * 100);
+    ret = rt_mutex_take(&_mutex[thread_id % MUTEX_NUM], RT_WAITING_FOREVER);
+    uassert_true(ret == RT_EOK);
+    uassert_true(RT_SCHED_PRIV(rt_thread_self()).current_priority == RT_SCHED_PRIV(rt_thread_self()).init_priority);
+
+    if (thread_id == 1)
+    {
+        rt_thread_mdelay(100); // wait for main thread re-get _mutex[1]
+        uassert_true(RT_SCHED_PRIV(rt_thread_self()).current_priority == 8);
+    }
+
+    ret = rt_mutex_release(&_mutex[thread_id % MUTEX_NUM]);
+    uassert_true(ret == RT_EOK);
+    uassert_true(RT_SCHED_PRIV(rt_thread_self()).current_priority == RT_SCHED_PRIV(rt_thread_self()).init_priority);
+
+    _sync_flag ++;
+}
+
+static void test_main_thread_entry(void *para)
+{
+    while (!_sync_flag)
+    {
+        rt_thread_delay(1);
+    }
+
+    rt_err_t ret;
+
+    ret = rt_mutex_take(&_mutex[0], RT_WAITING_FOREVER);
+    uassert_true(ret == RT_EOK);
+    uassert_true(RT_SCHED_PRIV(rt_thread_self()).current_priority == 12);
+    rt_thread_mdelay(100);         // wait for t0 take mutex0
+    uassert_true(RT_SCHED_PRIV(rt_thread_self()).current_priority == 12);
+
+    ret = rt_mutex_take(&_mutex[1], RT_WAITING_FOREVER);
+    uassert_true(ret == RT_EOK);
+    uassert_true(RT_SCHED_PRIV(rt_thread_self()).current_priority == 12);
+    rt_thread_mdelay(100);         // wait for t1 take mutex1
+    uassert_true(RT_SCHED_PRIV(rt_thread_self()).current_priority == 9);
+
+    ret = rt_mutex_take(&_mutex[2], RT_WAITING_FOREVER);
+    uassert_true(ret == RT_EOK);
+    uassert_true(RT_SCHED_PRIV(rt_thread_self()).current_priority == 9);
+    rt_thread_mdelay(100);         // wait for t2 take mutex2
+    uassert_true(RT_SCHED_PRIV(rt_thread_self()).current_priority == 8);
+
+    rt_thread_mdelay(100);         // wait for t3 take mutex0
+    uassert_true(RT_SCHED_PRIV(rt_thread_self()).current_priority == 7);
+
+    rt_thread_mdelay(100);         // wait for t4 take mutex1
+    uassert_true(RT_SCHED_PRIV(rt_thread_self()).current_priority == 7);
+
+    rt_thread_mdelay(100);
+    rt_mutex_release(&_mutex[0]);   // give _mutex0 to t3
+    uassert_true(RT_SCHED_PRIV(rt_thread_self()).current_priority == 8);
+
+    rt_thread_mdelay(100);
+    rt_mutex_release(&_mutex[1]);   // give _mutex1 to t1
+    uassert_true(RT_SCHED_PRIV(rt_thread_self()).current_priority == 8);
+
+    rt_thread_mdelay(50);
+    rt_mutex_take(&_mutex[1], RT_WAITING_FOREVER);   // re-get _mutex1, which is hold by t1
+    uassert_true(RT_SCHED_PRIV(rt_thread_self()).current_priority == 8);
+    rt_mutex_release(&_mutex[1]);   // give _mutex1 to thread t1
+    uassert_true(RT_SCHED_PRIV(rt_thread_self()).current_priority == 8);
+
+    rt_thread_mdelay(100);
+    rt_mutex_release(&_mutex[2]);
+    uassert_true(RT_SCHED_PRIV(rt_thread_self()).current_priority == 12);
+
+    _sync_flag ++;
+}
+
+static void test_mutex_pi(void)
+{
+    rt_thread_t t_main;
+    rt_thread_t t[THREAD_NUM];
+    rt_uint8_t prio[THREAD_NUM] = {13, 9, 8, 7, 11}; // prio of threads
+
+    for (int i = 0; i < MUTEX_NUM; i++)
+    {
+        rt_mutex_init(&_mutex[i], "test1", 0);
+    }
+
+    _sync_flag  = 0;
+
+    t_main = rt_thread_create("t_main", test_main_thread_entry, RT_NULL, THREAD_STACKSIZE, 12, 10000);
+    uassert_true(t_main != RT_NULL);
+    rt_thread_startup(t_main);
+
+    for (rt_ubase_t i = 0; i < THREAD_NUM; i++)
+    {
+        t[i] = rt_thread_create("t", test_thread_entry, (void *)i, THREAD_STACKSIZE, prio[i], 10000);
+        uassert_true(t[i] != RT_NULL);
+        rt_thread_startup(t[i]);
+    }
+
+    _sync_flag = 1;
+
+    while (_sync_flag != THREAD_NUM + 1 + 1)
+    {
+        rt_thread_mdelay(100);
+    }
+
+    for (int i = 0; i < MUTEX_NUM; i++)
+    {
+        rt_mutex_detach(&_mutex[i]);
+    }
+}
+
+static struct rt_mutex _timeout_mutex;
+
+static void test_main_timeout_entry(void *para)
+{
+    rt_err_t ret;
+
+    ret = rt_mutex_take(&_timeout_mutex, RT_WAITING_FOREVER);
+    uassert_true(ret == -RT_EOK);
+    rt_thread_mdelay(100);
+    uassert_true(RT_SCHED_PRIV(rt_thread_self()).current_priority == 10);
+    rt_thread_mdelay(100);
+    uassert_true(RT_SCHED_PRIV(rt_thread_self()).current_priority == 12);
+    rt_mutex_release(&_timeout_mutex);
+    _sync_flag ++;
+}
+
+static void test_timeout_entry(void *para)
+{
+    rt_err_t ret;
+
+    rt_thread_mdelay(50);
+    ret = rt_mutex_take(&_timeout_mutex, rt_tick_from_millisecond(100));
+    uassert_true(ret == -RT_ETIMEOUT);
+    _sync_flag ++;
+}
+
+static void test_mutex_pi_timeout(void)
+{
+    _sync_flag = 0;
+
+    rt_mutex_init(&_timeout_mutex, "_timeout_mutex", 0);
+
+    rt_thread_t t1 = rt_thread_create("t1", test_main_timeout_entry, RT_NULL, THREAD_STACKSIZE, 12, 10000);
+    uassert_true(t1 != RT_NULL);
+    rt_thread_startup(t1);
+
+    rt_thread_t t2 = rt_thread_create("t2", test_timeout_entry, (void *)t1, THREAD_STACKSIZE, 10, 10000);
+    uassert_true(t2 != RT_NULL);
+    rt_thread_startup(t2);
+
+    while (_sync_flag != 2)
+    {
+        rt_thread_mdelay(100);
+    }
+
+    rt_mutex_detach(&_timeout_mutex);
+}
+
+#define TC_THREAD_NUM 4
+#define TC_MUTEX_NUM TC_THREAD_NUM
+static rt_thread_t t[TC_THREAD_NUM], t_hi_prio;
+static struct rt_mutex m[TC_MUTEX_NUM];
+
+static void test_recursive_mutex_depend_entry(void *para)
+{
+    rt_ubase_t id = (rt_ubase_t)para;
+
+    rt_mutex_take(&m[id], RT_WAITING_FOREVER);
+
+    rt_thread_mdelay(50);
+
+    if (id != 0)
+    {
+        rt_mutex_take(&m[id - 1], RT_WAITING_FOREVER);
+    }
+
+    if (id == 0)
+    {
+        rt_thread_mdelay(250);
+        rt_mutex_release(&m[id]);
+    }
+    else
+    {
+        rt_mutex_release(&m[id - 1]);
+        rt_mutex_release(&m[id]);
+    }
+    _sync_flag ++;
+}
+
+static void test_recursive_mutex_depend_hi_pri_entry(void *para)
+{
+    rt_thread_mdelay(100);
+    rt_err_t err = rt_mutex_take(&m[TC_MUTEX_NUM - 1], rt_tick_from_millisecond(100));
+    uassert_true(err == -RT_ETIMEOUT);
+    _sync_flag ++;
+}
+
+static void test_mutex_pi_recursive_prio_update(void)
+{
+    _sync_flag = 0;
+
+    for (int i = 0; i < TC_MUTEX_NUM; i++)
+    {
+        rt_mutex_init(&m[i], "test", 0);
+    }
+
+    for (rt_ubase_t i = 0; i < TC_THREAD_NUM; i++)
+    {
+        t[i] = rt_thread_create("t", test_recursive_mutex_depend_entry, (void *)i, THREAD_STACKSIZE, 10, 10000);
+        rt_thread_startup(t[i]);
+    }
+    t_hi_prio = rt_thread_create("t", test_recursive_mutex_depend_hi_pri_entry, (void *)RT_NULL, THREAD_STACKSIZE, 3, 10000);
+    rt_thread_startup(t_hi_prio);
+
+    rt_thread_mdelay(150);
+
+    for (int i = 0; i < TC_THREAD_NUM; i++)
+    {
+        uassert_true(RT_SCHED_PRIV(t[i]).current_priority == 3);
+    }
+
+    rt_thread_mdelay(100);
+
+    for (int i = 0; i < TC_THREAD_NUM; i++)
+    {
+        uassert_true(RT_SCHED_PRIV(t[i]).current_priority == 10);
+    }
+
+    while (_sync_flag != TC_THREAD_NUM + 1)
+    {
+        rt_thread_mdelay(100);
+    }
+
+    for (int i = 0; i < TC_MUTEX_NUM; i++)
+    {
+        rt_mutex_detach(&m[i]);
+    }
+    _sync_flag ++;
+}
+
+static void test_mutex_waiter_to_wakeup_entry(void *para)
+{
+    rt_thread_mdelay(100);
+    rt_err_t err = rt_mutex_take(&m[TC_MUTEX_NUM - 1], RT_WAITING_FOREVER);
+    uassert_true(err == -RT_EINTR);
+    _sync_flag ++;
+}
+
+static void wakeup_func(void *para)
+{
+    rt_thread_resume(t_hi_prio);
+}
+static void test_mutex_pi_wakeup_mutex_waiter(void)
+{
+    struct rt_timer wakeup_timer;
+
+    _sync_flag = 0;
+
+    for (int i = 0; i < TC_MUTEX_NUM; i++)
+    {
+        rt_mutex_init(&m[i], "test", 0);
+    }
+
+    for (rt_ubase_t i = 0; i < TC_THREAD_NUM; i++)
+    {
+        t[i] = rt_thread_create("t", test_recursive_mutex_depend_entry, (void *)i, THREAD_STACKSIZE, 10, 10000);
+        rt_thread_startup(t[i]);
+    }
+    t_hi_prio = rt_thread_create("t", test_mutex_waiter_to_wakeup_entry, (void *)RT_NULL, THREAD_STACKSIZE, 3, 10000);
+    rt_thread_startup(t_hi_prio);
+
+    rt_timer_init(&wakeup_timer, "wakeup_timer", wakeup_func, RT_NULL, rt_tick_from_millisecond(200), RT_TIMER_FLAG_ONE_SHOT);
+    rt_timer_start(&wakeup_timer);
+    rt_thread_mdelay(150);
+
+    for (int i = 0; i < TC_THREAD_NUM; i++)
+    {
+        uassert_true(RT_SCHED_PRIV(t[i]).current_priority == 3);
+    }
+
+    rt_thread_mdelay(100);
+
+    for (int i = 0; i < TC_THREAD_NUM; i++)
+    {
+        uassert_true(RT_SCHED_PRIV(t[i]).current_priority == 10);
+    }
+
+    while (_sync_flag != TC_THREAD_NUM + 1)
+    {
+        rt_thread_mdelay(100);
+    }
+
+    for (int i = 0; i < TC_MUTEX_NUM; i++)
+    {
+        rt_mutex_detach(&m[i]);
+    }
+    rt_timer_detach(&wakeup_timer);
+}
+
+static rt_err_t utest_tc_init(void)
+{
+    return RT_EOK;
+}
+
+static rt_err_t utest_tc_cleanup(void)
+{
+    return RT_EOK;
+}
+
+static void testcase(void)
+{
+    UTEST_UNIT_RUN(test_mutex_pi);
+    UTEST_UNIT_RUN(test_mutex_pi_recursive_prio_update);
+    UTEST_UNIT_RUN(test_mutex_pi_timeout);
+    UTEST_UNIT_RUN(test_mutex_pi_wakeup_mutex_waiter);
+}
+UTEST_TC_EXPORT(testcase, "testcases.kernel.mutex_pi_tc", utest_tc_init, utest_tc_cleanup, 1000);
+
+/********************* end of file ************************/