pmdinfogen: fix resource leak of file object

Coverity flags an issue where the resources used by the FILE object for
the temporary input file are leaked. This is a very minor issue, but is
easily fixed, while also avoiding later problems where we try to close
an invalid file descriptor in the failure case.

The fix is to use "dup()" to get a new file descriptor number rather than
using the value directly from fileno. This allows us to close the file
opened with tmpfile() within in scope block, while allowing the duplicate
to pass to the outer block and be closed when the function terminates.

As a side-effect I/O in the function is therefore changed from using stdio
fread/fwrite to read/write system calls.

Coverity issue: 260399
Fixes: 0d68533617 ("pmdinfogen: allow using stdin and stdout")

Signed-off-by: Bruce Richardson <bruce.richardson@intel.com>
Acked-by: Neil Horman <nhorman@tuxdriver.com>
This commit is contained in:
Bruce Richardson 2018-02-02 12:00:58 +00:00 committed by Thomas Monjalon
parent 028e4b1dbc
commit ac7f4af813

View File

@ -50,20 +50,24 @@ static void *grab_file(const char *filename, unsigned long *size)
/* from stdin, use a temporary file to mmap */ /* from stdin, use a temporary file to mmap */
FILE *infile; FILE *infile;
char buffer[1024]; char buffer[1024];
size_t n; int n;
infile = tmpfile(); infile = tmpfile();
if (infile == NULL) { if (infile == NULL) {
perror("tmpfile"); perror("tmpfile");
return NULL; return NULL;
} }
while (!feof(stdin)) { fd = dup(fileno(infile));
n = fread(buffer, 1, sizeof(buffer), stdin); fclose(infile);
if (fwrite(buffer, 1, n, infile) != n) if (fd < 0)
return NULL;
n = read(STDIN_FILENO, buffer, sizeof(buffer));
while (n > 0) {
if (write(fd, buffer, n) != n)
goto failed; goto failed;
n = read(STDIN_FILENO, buffer, sizeof(buffer));
} }
fflush(infile);
fd = fileno(infile);
} }
if (fstat(fd, &st)) if (fstat(fd, &st))