From 7d5361374ad0cbe64c6dbb8cce3fc8842ed6b066 Mon Sep 17 00:00:00 2001 From: lemon Date: Mon, 20 Oct 2025 17:15:37 +0200 Subject: io: reject non file files --- io.c | 35 +++++++++++++++++++---------------- main.c | 2 +- 2 files changed, 20 insertions(+), 17 deletions(-) diff --git a/io.c b/io.c index 8306fb2..6390d96 100644 --- a/io.c +++ b/io.c @@ -689,7 +689,23 @@ mapopen(const char **err, const char *path) if (fstat(fd, &stat) != 0) goto Err; - if (!S_ISREG(stat.st_mode)) { + if (S_ISREG(stat.st_mode)) { + if (stat.st_size > UINT_MAX) { + Big: + errno = EFBIG; + goto Err; + } + mapsiz = alignup(stat.st_size, pagesiz); + if ((p = mmap(NULL, mapsiz + pagesiz, PROT_READ, MAP_ANONYMOUS|MAP_PRIVATE, -1, 0)) == MAP_FAILED) + goto Err; + if (mapsiz > 0 && mmap(p, mapsiz, PROT_READ, MAP_FIXED|MAP_PRIVATE, fd, 0) == MAP_FAILED) + goto Err; + + close(fd); + f.p = p; + f.n = stat.st_size; + return f; + } else if (S_ISFIFO(stat.st_mode) || S_ISCHR(stat.st_mode)) { uint cap = 0; int ret; @@ -723,23 +739,10 @@ mapopen(const char **err, const char *path) free(p); mprotect((void *)f.p, mapsiz + pagesiz, PROT_READ); return f; + } else { + *err = "Not a file"; } - if (stat.st_size > UINT_MAX) { - Big: - errno = EFBIG; - goto Err; - } - mapsiz = alignup(stat.st_size, pagesiz); - if ((p = mmap(NULL, mapsiz + pagesiz, PROT_READ, MAP_ANONYMOUS|MAP_PRIVATE, -1, 0)) == MAP_FAILED) - goto Err; - if (mapsiz > 0 && mmap(p, mapsiz, PROT_READ, MAP_FIXED|MAP_PRIVATE, fd, 0) == MAP_FAILED) - goto Err; - - close(fd); - f.p = p; - f.n = stat.st_size; - return f; Err: if (fd >= 0) close(fd); if (!*err) *err = strerror(errno); diff --git a/main.c b/main.c index c82c94e..9449d85 100644 --- a/main.c +++ b/main.c @@ -105,7 +105,7 @@ optparse(char **args) while ((arg = *++args)) { if (*arg++ != '-' || !*arg) { assert(task.ninf < arraylength(task.inf) && "too many infiles"); - task.inf[task.ninf] = *arg ? arg-1 : "/dev/stdin"; + task.inf[task.ninf] = arg[-1] != '-' ? arg-1 : "/dev/stdin"; task.inft[task.ninf] = ft ? ft : ftdetect(arg-1); ++task.ninf; ft = IFTauto; -- cgit v1.2.3