[fs_cli] Fix unchecked tainted data.

This commit is contained in:
Andrey Volk 2019-11-20 11:38:27 +04:00
parent bafe238b72
commit f664600df4
1 changed files with 4 additions and 1 deletions

View File

@ -1002,10 +1002,13 @@ static const char *basic_gets(int *cnt)
for (x = 0; x < (sizeof(command_buf) - 1); x++) { for (x = 0; x < (sizeof(command_buf) - 1); x++) {
int c = getchar(); int c = getchar();
if (c < 0) { if (c < 0) {
size_t command_buf_len;
if (fgets(command_buf, sizeof(command_buf) - 1, stdin) != command_buf) { if (fgets(command_buf, sizeof(command_buf) - 1, stdin) != command_buf) {
break; break;
}
if ((command_buf_len = strlen(command_buf)) > 0) {
command_buf[command_buf_len - 1] = '\0'; /* remove endline */
} }
command_buf[strlen(command_buf)-1] = '\0'; /* remove endline */
break; break;
} }
command_buf[x] = (char) c; command_buf[x] = (char) c;