Fix 'uploaded' file descriptor leak

This commit is contained in:
Yang Tse 2009-12-03 20:37:49 +00:00
Родитель 636d2fe00a
Коммит d14bf09ab8
1 изменённых файлов: 10 добавлений и 0 удалений

Просмотреть файл

@ -664,6 +664,8 @@ int main(int argc, char **argv)
struct testcase test; struct testcase test;
int result = 0; int result = 0;
memset(&test, 0, sizeof(test));
while(argc>arg) { while(argc>arg) {
if(!strcmp("--version", argv[arg])) { if(!strcmp("--version", argv[arg])) {
printf("tftpd IPv4%s\n", printf("tftpd IPv4%s\n",
@ -810,6 +812,11 @@ int main(int argc, char **argv)
sclose(peer); sclose(peer);
peer = CURL_SOCKET_BAD; peer = CURL_SOCKET_BAD;
if(test.ofile > 0) {
close(test.ofile);
test.ofile = 0;
}
if(got_exit_signal) if(got_exit_signal)
break; break;
@ -824,6 +831,9 @@ int main(int argc, char **argv)
tftpd_cleanup: tftpd_cleanup:
if(test.ofile > 0)
close(test.ofile);
if((peer != sock) && (peer != CURL_SOCKET_BAD)) if((peer != sock) && (peer != CURL_SOCKET_BAD))
sclose(peer); sclose(peer);