sys/shell: correctly detect and handle long lines
The numeric value for EOF is -1. This caused the shell to return the same code when EOF was encountered and when the line lenght was exceeded. Additionally, if the line length is exceeded, the correct behaviour is to consume the remaining characters until the end of the line, to prevent the following line from containing (potentially dangerous) garbage. Co-authored-by: Hendrik van Essen <hendrik.ve@fu-berlin.de>
This commit is contained in:
parent
bcdb693bd3
commit
2810637306
@ -1,5 +1,5 @@
|
||||
/*
|
||||
* Copyright (C) 2009, Freie Universitaet Berlin (FUB).
|
||||
* Copyright (C) 2009, 2020 Freie Universität Berlin
|
||||
* Copyright (C) 2013, INRIA.
|
||||
* Copyright (C) 2015 Kaspar Schleiser <kaspar@schleiser.de>
|
||||
*
|
||||
@ -21,6 +21,8 @@
|
||||
*
|
||||
* @author Kaspar Schleiser <kaspar@schleiser.de>
|
||||
* @author René Kijewski <rene.kijewski@fu-berlin.de>
|
||||
* @author Juan Carrano <j.carrano@fu-berlin.de>
|
||||
* @author Hendrik van Essen <hendrik.ve@fu-berlin.de>
|
||||
*
|
||||
* @}
|
||||
*/
|
||||
@ -29,6 +31,10 @@
|
||||
#include <stdio.h>
|
||||
#include <stdint.h>
|
||||
#include <stdlib.h>
|
||||
#include <stdbool.h>
|
||||
#include <assert.h>
|
||||
#include <errno.h>
|
||||
|
||||
#include "shell.h"
|
||||
#include "shell_commands.h"
|
||||
|
||||
@ -227,14 +233,36 @@ static void handle_input_line(const shell_command_t *command_list, char *line)
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* @brief Read a single line from standard input into a buffer.
|
||||
*
|
||||
* In addition to copying characters, this routine echoes the line back to
|
||||
* stdout and also supports primitive line editing.
|
||||
*
|
||||
* If the input line is too long, the input will still be consumed until the end
|
||||
* to prevent the next line from containing garbage.
|
||||
*
|
||||
* @param buf Buffer where the input will be placed.
|
||||
* @param size Size of the buffer. The maximum line length will be one less
|
||||
* than size, to accommodate for the null terminator.
|
||||
* The minimum buffer size is 1.
|
||||
*
|
||||
* @return length of the read line, excluding the terminator, if reading was
|
||||
* successful.
|
||||
* @return EOF, if the end of the input stream was reached.
|
||||
* @return ENOBUFS if the buffer size was exceeded.
|
||||
*/
|
||||
static int readline(char *buf, size_t size)
|
||||
{
|
||||
char *line_buf_ptr = buf;
|
||||
int curr_pos = 0;
|
||||
bool length_exceeded = false;
|
||||
|
||||
assert((size_t) size > 0);
|
||||
|
||||
while (1) {
|
||||
if ((line_buf_ptr - buf) >= ((int) size) - 1) {
|
||||
return -1;
|
||||
}
|
||||
/* At the start of the loop, cur_pos should point inside of
|
||||
* buf. This ensures the terminator can always fit. */
|
||||
assert((size_t) curr_pos < size);
|
||||
|
||||
int c = getchar();
|
||||
if (c < 0) {
|
||||
@ -246,23 +274,32 @@ static int readline(char *buf, size_t size)
|
||||
/* DOS newlines are handled like hitting enter twice, but empty lines are ignored. */
|
||||
/* Ctrl-C cancels the current line. */
|
||||
if (c == '\r' || c == '\n' || c == ETX) {
|
||||
*line_buf_ptr = '\0';
|
||||
if (c == ETX) {
|
||||
curr_pos = 0;
|
||||
length_exceeded = false;
|
||||
}
|
||||
|
||||
buf[curr_pos] = '\0';
|
||||
#ifndef SHELL_NO_ECHO
|
||||
_putchar('\r');
|
||||
_putchar('\n');
|
||||
#endif
|
||||
|
||||
/* return 1 if line is empty, 0 otherwise */
|
||||
return c == ETX || line_buf_ptr == buf;
|
||||
return (length_exceeded) ? -ENOBUFS : curr_pos;
|
||||
}
|
||||
|
||||
/* QEMU uses 0x7f (DEL) as backspace, while 0x08 (BS) is for most terminals */
|
||||
else if (c == 0x08 || c == 0x7f) {
|
||||
if (line_buf_ptr == buf) {
|
||||
if (c == 0x08 || c == 0x7f) {
|
||||
if (curr_pos == 0) {
|
||||
/* The line is empty. */
|
||||
continue;
|
||||
}
|
||||
|
||||
*--line_buf_ptr = '\0';
|
||||
/* after we dropped characters don't edit the line, yet keep the
|
||||
* visual effects */
|
||||
if (!length_exceeded) {
|
||||
buf[--curr_pos] = '\0';
|
||||
}
|
||||
/* white-tape the character */
|
||||
#ifndef SHELL_NO_ECHO
|
||||
_putchar('\b');
|
||||
@ -271,7 +308,13 @@ static int readline(char *buf, size_t size)
|
||||
#endif
|
||||
}
|
||||
else {
|
||||
*line_buf_ptr++ = c;
|
||||
/* Always consume characters, but do not not always store them */
|
||||
if ((size_t) curr_pos < size - 1) {
|
||||
buf[curr_pos++] = c;
|
||||
}
|
||||
else {
|
||||
length_exceeded = true;
|
||||
}
|
||||
#ifndef SHELL_NO_ECHO
|
||||
_putchar(c);
|
||||
#endif
|
||||
@ -298,12 +341,18 @@ void shell_run_once(const shell_command_t *shell_commands,
|
||||
while (1) {
|
||||
int res = readline(line_buf, len);
|
||||
|
||||
if (res == EOF) {
|
||||
break;
|
||||
}
|
||||
switch (res) {
|
||||
|
||||
if (!res) {
|
||||
case EOF:
|
||||
return;
|
||||
|
||||
case -ENOBUFS:
|
||||
puts("shell: maximum line length exceeded");
|
||||
break;
|
||||
|
||||
default:
|
||||
handle_input_line(shell_commands, line_buf);
|
||||
break;
|
||||
}
|
||||
|
||||
print_prompt();
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user