ports/iperf3
ports/iperf3
1
1
Форкнуть 0
Код Релизы Активность

Clean up some memory leaks in iperf_new_stream error handling.

Просмотр исходного кода 
Submitted by:	Susant Sahani <ssahani@redhat.com>
Этот коммит содержится в:
Bruce A. Mah 2014-01-13 09:27:42 -08:00
родитель 0129f08391
Коммит 55a453c2a1
1 изменённых файлов: 17 добавлений и 1 удалений
Показать статистику Скачать Patch файл Скачать Diff файл Показать все файлы Свернуть все файлы

18
src/iperf_api.c
Просмотреть файл

@ -2162,19 +2162,27 @@ iperf_new_stream(struct iperf_test *test, int s)
sp->buffer_fd = mkstemp(template); sp->buffer_fd = mkstemp(template);
if (sp->buffer_fd == -1) { if (sp->buffer_fd == -1) {
i_errno = IECREATESTREAM; i_errno = IECREATESTREAM;
free(sp->result);
free(sp);
return NULL; return NULL;
} }
if (unlink(template) < 0) { if (unlink(template) < 0) {
i_errno = IECREATESTREAM; i_errno = IECREATESTREAM;
free(sp->result);
free(sp);
return NULL; return NULL;
} }
if (ftruncate(sp->buffer_fd, test->settings->blksize) < 0) { if (ftruncate(sp->buffer_fd, test->settings->blksize) < 0) {
i_errno = IECREATESTREAM; i_errno = IECREATESTREAM;
free(sp->result);
free(sp);
return NULL; return NULL;
} }
sp->buffer = (char *) mmap(NULL, test->settings->blksize, PROT_READ|PROT_WRITE, MAP_PRIVATE, sp->buffer_fd, 0); sp->buffer = (char *) mmap(NULL, test->settings->blksize, PROT_READ|PROT_WRITE, MAP_PRIVATE, sp->buffer_fd, 0);
if (sp->buffer == MAP_FAILED) { if (sp->buffer == MAP_FAILED) {
i_errno = IECREATESTREAM; i_errno = IECREATESTREAM;
free(sp->result);
free(sp);
return NULL; return NULL;
} }
srandom(time(NULL)); srandom(time(NULL));
@ -2191,6 +2199,9 @@ iperf_new_stream(struct iperf_test *test, int s)
sp->diskfile_fd = open(test->diskfile_name, test->sender ? O_RDONLY : (O_WRONLY|O_CREAT|O_TRUNC)); sp->diskfile_fd = open(test->diskfile_name, test->sender ? O_RDONLY : (O_WRONLY|O_CREAT|O_TRUNC));
if (sp->diskfile_fd == -1) { if (sp->diskfile_fd == -1) {
i_errno = IEFILE; i_errno = IEFILE;
munmap(sp->buffer, sp->test->settings->blksize);
free(sp->result);
free(sp);
return NULL; return NULL;
} }
sp->snd2 = sp->snd; sp->snd2 = sp->snd;
@ -2201,8 +2212,13 @@ iperf_new_stream(struct iperf_test *test, int s)
sp->diskfile_fd = -1; sp->diskfile_fd = -1;
/* Initialize stream */ /* Initialize stream */
if (iperf_init_stream(sp, test) < 0) if (iperf_init_stream(sp, test) < 0) {
close(sp->buffer_fd);
munmap(sp->buffer, sp->test->settings->blksize);
free(sp->result);
free(sp);
return NULL; return NULL;
}
iperf_add_stream(test, sp); iperf_add_stream(test, sp);
return sp; return sp;