JT-DT-YD4N02B_4G_RTT_MRS/packages/rt_kprintf_threadsafe/rt_kprintf.c

125 lines
3.9 KiB
C

/*
* @Author : stark1898y 1658608470@qq.com
* @Date : 2024-06-14 13:21:14
* @LastEditors : stark1898y 1658608470@qq.com
* @LastEditTime : 2024-06-14 13:21:14
* @FilePath : \wuxi_alarm_ch32v303rct6_rtt\common\rt_kprintf_threadsafe\rt_kprintf.c
* @Description :
*
* Copyright (c) 2024 by yzy, All Rights Reserved.
*/
/*
* Copyright (c) 2021, Meco Jianting Man <jiantingman@foxmail.com>
*
* SPDX-License-Identifier: Apache-2.0
*
* Change Logs:
* Date Author Notes
* 2020-12-15 Meco Man first version
*/
#include <rtthread.h>
#include <rthw.h>
#ifndef RT_USING_CONSOLE
#error "Please enable console"
#endif
/**
* This function will print a formatted string on system console (thread safe)
*
* @param fmt the format
*/
#if (RTTHREAD_VERSION >= 40100) || (RTTHREAD_VERSION < 40000 && RTTHREAD_VERSION >= 30106)
int rt_kprintf(const char *fmt, ...)
#else
void rt_kprintf(const char *fmt, ...)
#endif
{
va_list args;
rt_size_t length;
rt_device_t console_dev;
static char rt_console_buf[RT_CONSOLEBUF_SIZE];
#if defined RT_USING_MUTEX || defined RT_USING_SEMAPHORE
static unsigned char kprintf_init_flag = RT_FALSE;
#if defined RT_USING_MUTEX
static struct rt_mutex kprintf_mutex;
#elif defined RT_USING_SEMAPHORE
static struct rt_semaphore kprintf_sem;
#endif /*defined RT_USING_MUTEX*/
#endif /*defined RT_USING_MUTEX || defined RT_USING_SEMAPHORE*/
if(rt_interrupt_get_nest() == 0u && rt_thread_self() != RT_NULL)
{
#if defined RT_USING_MUTEX || defined RT_USING_SEMAPHORE
if(kprintf_init_flag == RT_FALSE)
{
#if defined RT_USING_MUTEX
rt_mutex_init(&kprintf_mutex, "kprintf", RT_IPC_FLAG_FIFO);
#elif defined RT_USING_SEMAPHORE
rt_sem_init(&kprintf_sem, "kprintf", 1, RT_IPC_FLAG_FIFO);
#endif /*defined RT_USING_MUTEX*/
kprintf_init_flag = RT_TRUE;
}
#if defined RT_USING_MUTEX
rt_mutex_take(&kprintf_mutex, RT_WAITING_FOREVER);
#elif defined RT_USING_SEMAPHORE
rt_sem_take(&kprintf_sem, RT_WAITING_FOREVER);
#endif /*defined RT_USING_MUTEX*/
#else
rt_enter_critical();
#endif /*defined RT_USING_MUTEX || defined RT_USING_SEMAPHORE*/
}
console_dev = rt_console_get_device();
va_start(args, fmt);
/* the return value of vsnprintf is the number of bytes that would be
* written to buffer had if the size of the buffer been sufficiently
* large excluding the terminating null byte. If the output string
* would be larger than the rt_console_buf, we have to adjust the output
* length. */
length = rt_vsnprintf(rt_console_buf, sizeof(rt_console_buf) - 1, fmt, args);
if (length > RT_CONSOLEBUF_SIZE - 1)
length = RT_CONSOLEBUF_SIZE - 1;
#ifdef RT_USING_DEVICE
if (console_dev == RT_NULL)
{
rt_hw_console_output(rt_console_buf);
}
else
{
rt_uint16_t old_flag = console_dev->open_flag;
console_dev->open_flag |= RT_DEVICE_FLAG_STREAM;
rt_device_write(console_dev, 0, rt_console_buf, length);
console_dev->open_flag = old_flag;
}
#else
rt_hw_console_output(rt_console_buf);
#endif /*RT_USING_DEVICE*/
va_end(args);
if(rt_interrupt_get_nest() == 0u &&
#if defined RT_USING_MUTEX || defined RT_USING_SEMAPHORE
kprintf_init_flag == RT_TRUE &&
#endif
rt_thread_self() != RT_NULL)
{
#if defined RT_USING_MUTEX || defined RT_USING_SEMAPHORE
#if defined RT_USING_MUTEX
rt_mutex_release(&kprintf_mutex);
#elif defined RT_USING_SEMAPHORE
rt_sem_release(&kprintf_sem);
#endif /*defined RT_USING_MUTEX*/
#else
rt_exit_critical();
#endif /*defined RT_USING_MUTEX || defined RT_USING_SEMAPHORE*/
}
#if (RTTHREAD_VERSION >= 40100) || (RTTHREAD_VERSION < 40000 && RTTHREAD_VERSION >= 30106)
return length; /* the total number of printed characters */
#endif
}