stdio.c 1.6 KB
Newer Older
B
bernard 已提交
1
/*
2
 * Copyright (c) 2006-2021, RT-Thread Development Team
B
bernard 已提交
3
 *
4
 * SPDX-License-Identifier: Apache-2.0
B
bernard 已提交
5 6 7 8 9 10 11
 *
 * Change Logs:
 * Date           Author       Notes
 * 2017/10/15     bernard      the first version
 */
#include <stdio.h>
#include <stdlib.h>
B
Bernard Xiong 已提交
12
#include <fcntl.h>
B
bernard 已提交
13 14 15 16 17
#include <rtthread.h>
#include "libc.h"

#define STDIO_DEVICE_NAME_MAX   32

18
int _EXFUN(fileno, (FILE *));
B
Bernard Xiong 已提交
19

B
bernard 已提交
20 21 22 23 24 25 26 27 28 29 30 31 32
static FILE* std_console = NULL;

int libc_stdio_set_console(const char* device_name, int mode)
{
    FILE *fp;
    char name[STDIO_DEVICE_NAME_MAX];
    char *file_mode;

    snprintf(name, sizeof(name) - 1, "/dev/%s", device_name);
    name[STDIO_DEVICE_NAME_MAX - 1] = '\0';

    if (mode == O_RDWR) file_mode = "r+";
    else if (mode == O_WRONLY) file_mode = "wb";
B
Bernard Xiong 已提交
33
    else file_mode = "rb";
B
bernard 已提交
34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50

    fp = fopen(name, file_mode);
    if (fp)
    {
        setvbuf(fp, NULL, _IONBF, 0);

        if (std_console)
        {
            fclose(std_console);
            std_console = NULL;
        }
        std_console = fp;

        if (mode == O_RDWR)
        {
            _GLOBAL_REENT->_stdin  = std_console;
        }
51
        else
B
bernard 已提交
52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69
        {
            _GLOBAL_REENT->_stdin  = NULL;
        }

        if (mode == O_RDONLY)
        {
            _GLOBAL_REENT->_stdout = NULL;
            _GLOBAL_REENT->_stderr = NULL;
        }
        else
        {
            _GLOBAL_REENT->_stdout = std_console;
            _GLOBAL_REENT->_stderr = std_console;
        }

        _GLOBAL_REENT->__sdidinit = 1;
    }

B
BernardXiong 已提交
70 71 72
    if (std_console) return fileno(std_console);

    return -1;
B
bernard 已提交
73
}
armink_ztl's avatar
armink_ztl 已提交
74 75 76 77 78 79 80

int libc_stdio_get_console(void) {
    if (std_console)
        return fileno(std_console);
    else
        return -1;
}