mkrtos-real/mkrtos_user/lib/letter-shell/demo/mkrtos/shell_port.c

/**
 * @file shell_port.c
 * @author Letter (NevermindZZT@gmail.com)
 * @brief
 * @version 0.1
 * @date 2019-02-22
 *
 * @copyright (c) 2019 Letter
 *
 */

#include "shell.h"
#include "shell_ext.h"
#include "shell_fs.h"
#include "shell_passthrough.h"
#include "shell_secure_user.h"
#include "log.h"
// #include "telnetd.h"
#include <stdio.h>
#include <dirent.h>
#include <unistd.h>
#include <stddef.h>
#include <string.h>
#include <sys/time.h>
#include <pthread.h>
#include <fcntl.h>
#include <sys/types.h>
#include <termios.h>
#include <sys/stat.h>
#include "cons_cli.h"
#include "u_sleep.h"
#include "u_sema.h"
#include "u_task.h"
static Shell shell;
static ShellFs shellFs;
static char shellBuffer[512];
// static char shellPathBuffer[128] = "/";

/**
 * @brief 用户shell写
 *
 * @param data 数据
 * @param len 数据长度
 *
 * @return unsigned short 写入实际长度
 */
signed short userShellWrite(char *data, unsigned short len)
{    return write(STDOUT_FILENO, data, len);
}

/**
 * @brief 用户shell读
 *
 * @param data 数据
 * @param len 数据长度
 *
 * @return unsigned short 读取实际长度
 */
signed short userShellRead(char *data, unsigned short len)
{
    int rlen;

    rlen = read(STDIN_FILENO, data, len);
    return rlen;
}
/**
 * @brief 列出文件
 *
 * @param path 路径
 * @param buffer 结果缓冲
 * @param maxLen 最大缓冲长度
 * @return size_t 0
 */
size_t userShellListDir(char *path, char *buffer, size_t maxLen)
{
    DIR *dir;
    struct dirent *ptr;
    int i;

    if (!path)
    {
        return 0;
    }
    dir = opendir(path);
    if (!dir)
    {
        return 0;
    }
    memset(buffer, 0, maxLen);
    while ((ptr = readdir(dir)) != NULL)
    {
        strcat(buffer, ptr->d_name);
        strcat(buffer, "\t");
    }
    closedir(dir);
    return 0;
}
static struct termios old_settings;
static struct termios new_settings;

void tty_set_raw_mode(void)
{
    new_settings = old_settings;
    new_settings.c_lflag &= ~(ICANON | ECHO);   // 禁用规范模式和回显
    new_settings.c_cc[VMIN] = 1;                // 读取的最小字符数
    new_settings.c_cc[VTIME] = 0;               // 读取的超时时间（以10ms为单位）
    tcsetattr(STDIN_FILENO, TCSANOW, &new_settings);
}
void tty_set_normal_mode(void)
{
    tcsetattr(STDIN_FILENO, TCSANOW, &old_settings);
}
/**
 * @brief 用户shell初始化
 *
 */
void userShellInit(void)
{
    tcgetattr(STDIN_FILENO, &old_settings);
    tty_set_raw_mode();
    task_set_obj_name(TASK_THIS, TASK_THIS, "tk_shell");
    task_set_obj_name(TASK_THIS, THREAD_MAIN, "th_shell");
    // shellFs.getcwd = getcwd;
    // shellFs.chdir = chdir;
    // shellFs.listdir = userShellListDir;
    // shellFsInit(&shellFs, shellPathBuffer, sizeof(shellPathBuffer));
    shell.write = userShellWrite;
    shell.read = userShellRead;
    // shellSetPath(&shell, shellPathBuffer);
    shellInit(&shell, shellBuffer, sizeof(shellBuffer));
    // shellCompanionAdd(&shell, SHELL_COMPANION_ID_FS, &shellFs);

    while (1)
    {
        shellTask(&shell);
    }
}