nhyatt/putty-source
1
0
Fork 0
mirror of https://git.tartarus.org/simon/putty.git synced 2025-01-25 09:12:24 +00:00
Code Issues Projects Releases Wiki Activity

cmdgen: fix segfault on failing to open the output file.

Browse Source 
D'oh - simply forgot to check the return value of fopen for NULL.
This commit is contained in:
Simon Tatham 2018-10-15 22:26:20 +01:00
parent 35a4283615
commit b9bfc81531
1 changed files with 14 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
cmdgen.c
View File

@ -959,10 +959,15 @@ int main(int argc, char **argv)
{ {
FILE *fp; FILE *fp;
if (outfile) if (outfile) {
fp = f_open(outfilename, "w", FALSE); fp = f_open(outfilename, "w", FALSE);
else if (!fp) {
fprintf(stderr, "unable to open output file\n");
exit(1);
}
} else {
fp = stdout; fp = stdout;
}
if (sshver == 1) { if (sshver == 1) {
ssh1_write_pubkey(fp, ssh1key); ssh1_write_pubkey(fp, ssh1key);
@ -1003,10 +1008,15 @@ int main(int argc, char **argv)
} }
} }
if (outfile) if (outfile) {
fp = f_open(outfilename, "w", FALSE); fp = f_open(outfilename, "w", FALSE);
else if (!fp) {
fprintf(stderr, "unable to open output file\n");
exit(1);
}
} else {
fp = stdout; fp = stdout;
}
fprintf(fp, "%s\n", fingerprint); fprintf(fp, "%s\n", fingerprint);
if (outfile) if (outfile)
fclose(fp); fclose(fp);