diff options
Diffstat (limited to 'misc-utils')
| -rw-r--r-- | misc-utils/docfdisk.c | 6 | ||||
| -rw-r--r-- | misc-utils/ftl_check.c | 2 |
2 files changed, 6 insertions, 2 deletions
diff --git a/misc-utils/docfdisk.c b/misc-utils/docfdisk.c index 9956de5..b363662 100644 --- a/misc-utils/docfdisk.c +++ b/misc-utils/docfdisk.c @@ -275,7 +275,11 @@ int main(int argc, char **argv) show_header(mhoffs); printf("\nReady to update device. Type 'yes' to proceed, anything else to abort: "); - fgets(line, sizeof(line), stdin); + if (!fgets(line, sizeof(line), stdin)) { + printf("Failed to retrieve input chars!\n"); + return 1; + } + if (strcmp("yes\n", line)) return 0; printf("Updating MediaHeader...\n"); diff --git a/misc-utils/ftl_check.c b/misc-utils/ftl_check.c index d7d2e8b..970d968 100644 --- a/misc-utils/ftl_check.c +++ b/misc-utils/ftl_check.c @@ -95,7 +95,7 @@ static void check_partition(int fd) perror("seek failed"); break; } - read(fd, &hdr, sizeof(hdr)); + read_nocheck(fd, &hdr, sizeof(hdr)); if ((le32_to_cpu(hdr.FormattedSize) > 0) && (le32_to_cpu(hdr.FormattedSize) <= mtd.size) && (le16_to_cpu(hdr.NumEraseUnits) > 0) && |