From 709d8791ceadae2c2a0cd3ef9ebdf56e4d7dd27e Mon Sep 17 00:00:00 2001 From: Erik Hovland Date: Tue, 30 Sep 2008 14:05:32 -0700 Subject: [PATCH] Protect against resource leaks. Signed-off-by: Erik Hovland Signed-off-by: Dan Dennedy --- src/dispatch.c | 11 ++++++++--- src/fw.c | 4 +++- 2 files changed, 11 insertions(+), 4 deletions(-) diff --git a/src/dispatch.c b/src/dispatch.c index 6786a52..567fbbc 100644 --- a/src/dispatch.c +++ b/src/dispatch.c @@ -48,7 +48,10 @@ raw1394handle_t raw1394_new_handle(void) else if (handle) { handle->is_fw = 1; handle->mode.fw = fw_handle; - } + } else if (fw_handle) + fw_destroy_handle(fw_handle); + else if (ieee1394_handle) + ieee1394_destroy_handle(ieee1394_handle); } return handle; } @@ -76,14 +79,16 @@ raw1394handle_t raw1394_new_handle_on_port(int port) if (handle) { handle->is_fw = 0; handle->mode.ieee1394 = ieee1394_handle; - } + } else + ieee1394_destroy_handle(ieee1394_handle); } else if (fw_handle = fw_new_handle_on_port(port)) { handle = (raw1394handle_t) malloc(sizeof(struct raw1394_handle)); if (handle) { handle->is_fw = 1; handle->mode.fw = fw_handle; - } + } else + fw_destroy_handle(fw_handle); } return handle; } diff --git a/src/fw.c b/src/fw.c index 3c61385..a229d1a 100644 --- a/src/fw.c +++ b/src/fw.c @@ -501,8 +501,10 @@ fw_handle_t fw_new_handle_on_port(int port) if (handle == NULL) return NULL; - if (fw_set_port(handle, port) < 0) + if (fw_set_port(handle, port) < 0) { + fw_destroy_handle(handle); return NULL; + } return handle; }