newkey(8): fix 'tmpname' memory leak (always) and input file descriptor leak

when output file cannot be opened

PR:		201732
Reported by:	David Binderman <dcb314@hotmail.com>
MFC after:	1 week
This commit is contained in:
Andriy Voskoboinyk 2019-02-08 14:31:44 +00:00
parent 00c4df53b6
commit 245a7edb05
Notes: svn2git 2020-12-20 02:59:44 +00:00
svn path=/head/; revision=343909

View file

@ -266,11 +266,14 @@ localupdate(char *name, char *filename, u_int op, u_int keylen __unused,
sprintf(tmpname, "%s.tmp", filename);
rf = fopen(filename, "r");
if (rf == NULL) {
return (ERR_READ);
err = ERR_READ;
goto cleanup;
}
wf = fopen(tmpname, "w");
if (wf == NULL) {
return (ERR_WRITE);
fclose(rf);
err = ERR_WRITE;
goto cleanup;
}
err = -1;
while (fgets(line, sizeof (line), rf)) {
@ -310,13 +313,18 @@ localupdate(char *name, char *filename, u_int op, u_int keylen __unused,
fclose(rf);
if (err == 0) {
if (rename(tmpname, filename) < 0) {
return (ERR_DBASE);
err = ERR_DBASE;
goto cleanup;
}
} else {
if (unlink(tmpname) < 0) {
return (ERR_DBASE);
err = ERR_DBASE;
goto cleanup;
}
}
cleanup:
free(tmpname);
return (err);
}