From 7e76095180a5d64821e21900b85dbd8829e2d90d Mon Sep 17 00:00:00 2001 From: Rich Felker Date: Tue, 27 Sep 2011 17:00:38 -0400 Subject: improve error handling in mntent stuff... not sure if this is correct/ideal. it needs further attention. --- src/linux/mntent.c | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/src/linux/mntent.c b/src/linux/mntent.c index 48c85bd6..f1516252 100644 --- a/src/linux/mntent.c +++ b/src/linux/mntent.c @@ -1,6 +1,7 @@ #include #include #include +#include FILE *setmntent(const char *name, const char *mode) { @@ -22,7 +23,13 @@ struct mntent *getmntent_r(FILE *f, struct mntent *mnt, char *linebuf, int bufle do { fgets(linebuf, buflen, f); - if (feof(f)) return NULL; + if (feof(f) || ferror(f)) return 0; + if (!strchr(linebuf, '\n')) { + if (fseeko(f, -(off_t)strlen(linebuf), SEEK_CUR)) + fscanf(f, "%*[^\n]%*[\n]"); + errno = ERANGE; + return 0; + } cnt = sscanf(linebuf, " %n%*s%n %n%*s%n %n%*s%n %n%*s%n %d %d", n, n+1, n+2, n+3, n+4, n+5, n+6, n+7, &mnt->mnt_freq, &mnt->mnt_passno); @@ -50,7 +57,7 @@ struct mntent *getmntent(FILE *f) int addmntent(FILE *f, const struct mntent *mnt) { - fseek(f, 0, SEEK_END); + if (fseek(f, 0, SEEK_END)) return 1; return fprintf(f, "%s\t%s\t%s\t%s\t%d\t%d\n", mnt->mnt_fsname, mnt->mnt_dir, mnt->mnt_type, mnt->mnt_opts, mnt->mnt_freq, mnt->mnt_passno) < 0; -- cgit v1.2.1