1 /*
2  * Copyright (c) 2006-2024 RT-Thread Development Team
3  *
4  * SPDX-License-Identifier: Apache-2.0
5  *
6  * Change Logs:
7  * Date           Author       Notes
8  * 2024-11-20     zhujiale     the first version
9  */
10 #include <rtthread.h>
11 #include <rtdevice.h>
12 #include "utest.h"
13 
14 static struct rt_serial_device* _serial0;
15 static int cnt;
16 
17 #define __REG32(x) (*((volatile unsigned int*)((rt_ubase_t)x)))
18 #define UART_FR(base)   __REG32(base + 0x18)
19 #define UART_DR(base)   __REG32(base + 0x00)
20 #define UARTFR_TXFF     0x20
utest_upper_run(struct rt_serial_device * serial,char ch,void * data)21 static rt_err_t utest_upper_run(struct rt_serial_device* serial, char ch, void* data)
22 {
23     uassert_true(ch == 'a');
24     cnt++;
25     return RT_EOK;
26 }
27 
utest_getc(struct rt_serial_device * serial)28 static int utest_getc(struct rt_serial_device* serial)
29 {
30     static rt_uint8_t num = 0;
31     if (num == 10)
32         return -1;
33     num++;
34     return 'a';
35 }
36 
37 struct hw_uart_device
38 {
39     rt_size_t hw_base;
40     rt_size_t irqno;
41 };
42 
uart_putc(struct rt_serial_device * serial,char c)43 static int uart_putc(struct rt_serial_device* serial, char c)
44 {
45     struct hw_uart_device* uart;
46 
47     RT_ASSERT(serial != RT_NULL);
48     uart = (struct hw_uart_device*)serial->parent.user_data;
49 
50     while (UART_FR(uart->hw_base) & UARTFR_TXFF);
51     UART_DR(uart->hw_base) = c;
52 
53     return 1;
54 }
55 
56 static const struct rt_uart_ops _utest_ops =
57 {
58     RT_NULL,
59     RT_NULL,
60     uart_putc,
61     utest_getc,
62 };
63 
utest_lower_run_test2(struct rt_serial_device * serial,char ch,void * data)64 static rt_err_t utest_lower_run_test2(struct rt_serial_device* serial, char ch, void* data)
65 {
66     static rt_uint8_t num = 0;
67     num++;
68     uassert_true(ch == ('a' + num));
69     return RT_EOK;
70 }
71 
utest_getc_2(struct rt_serial_device * serial)72 static int utest_getc_2(struct rt_serial_device* serial)
73 {
74     static rt_uint8_t num = 0;
75     if (num == 20)
76         return -1;
77     num++;
78     return 'a' + num;
79 }
80 
81 static const struct rt_uart_ops _utest_ops2 =
82 {
83     RT_NULL,
84     RT_NULL,
85     uart_putc,
86     utest_getc_2,
87 };
bypass_upper_001(void)88 static void bypass_upper_001(void)
89 {
90     const struct rt_uart_ops* tmp = _serial0->ops;
91     _serial0->ops = &_utest_ops;
92     rt_bypass_upper_register(_serial0, "utest", RT_BYPASS_LEVEL_1, utest_upper_run, RT_NULL);
93     rt_hw_serial_isr(_serial0, RT_SERIAL_EVENT_RX_IND);
94     uassert_true(cnt == 10);
95     _serial0->ops = tmp;
96     rt_bypass_upper_unregister(_serial0, RT_BYPASS_LEVEL_1);
97 }
98 
bypass_upper_002(void)99 static void bypass_upper_002(void)
100 {
101     const struct rt_uart_ops* tmp = _serial0->ops;
102     _serial0->ops = &_utest_ops2;
103     rt_bypass_upper_register(_serial0, "utest", RT_BYPASS_MAX_LEVEL, utest_lower_run_test2, RT_NULL);
104 
105     rt_hw_serial_isr(_serial0, RT_SERIAL_EVENT_RX_IND);
106     rt_thread_mdelay(100);
107     uassert_true(cnt == 10);
108     _serial0->ops = tmp;
109     rt_bypass_upper_unregister(_serial0, RT_BYPASS_MAX_LEVEL);
110 }
111 
utest_tc_init(void)112 static rt_err_t utest_tc_init(void)
113 {
114     _serial0 = (struct rt_serial_device*)rt_console_get_device();
115     return RT_EOK;
116 }
117 
utest_tc_cleanup(void)118 static rt_err_t utest_tc_cleanup(void)
119 {
120     return RT_EOK;
121 }
122 
_testcase(void)123 static void _testcase(void)
124 {
125     UTEST_UNIT_RUN(bypass_upper_001);
126     UTEST_UNIT_RUN(bypass_upper_002);
127 }
128 
129 UTEST_TC_EXPORT(_testcase, "testcase.bypass.upper.001", utest_tc_init, utest_tc_cleanup, 10);
130