diff options
| -rw-r--r-- | src/ipcpd/ipcp.c | 4 | ||||
| -rw-r--r-- | src/irmd/main.c | 18 | ||||
| -rw-r--r-- | src/lib/ipcp.c | 22 | ||||
| -rw-r--r-- | src/lib/sockets.c | 21 |
4 files changed, 20 insertions, 45 deletions
diff --git a/src/ipcpd/ipcp.c b/src/ipcpd/ipcp.c index 839a5a9b..a9f2b77b 100644 --- a/src/ipcpd/ipcp.c +++ b/src/ipcpd/ipcp.c @@ -256,7 +256,7 @@ void * ipcp_main_loop(void * o) } pthread_cleanup_pop(true); - + pthread_cleanup_pop(false); buffer.len = ipcp_msg__get_packed_size(&ret_msg); if (buffer.len == 0) { @@ -280,7 +280,7 @@ void * ipcp_main_loop(void * o) } free(buffer.data); - pthread_cleanup_pop(true); + close(lsockfd); } diff --git a/src/irmd/main.c b/src/irmd/main.c index c7c1566d..58535e27 100644 --- a/src/irmd/main.c +++ b/src/irmd/main.c @@ -1431,6 +1431,11 @@ static void clean_msg(void * msg) irm_msg__free_unpacked(msg, NULL); } +static void close_ptr(void * o) +{ + close(*((int *) o)); +} + void * mainloop() { uint8_t buf[IRM_MSG_BUF_SIZE]; @@ -1465,6 +1470,7 @@ void * mainloop() continue; } + pthread_cleanup_push(close_ptr, &cli_sockfd); pthread_cleanup_push(clean_msg, (void *) msg); switch (msg->code) { @@ -1595,6 +1601,7 @@ void * mainloop() } pthread_cleanup_pop(true); + pthread_cleanup_pop(false); buffer.len = irm_msg__get_packed_size(&ret_msg); if (buffer.len == 0) { @@ -1615,17 +1622,12 @@ void * mainloop() irm_msg__pack(&ret_msg, buffer.data); - if (write(cli_sockfd, buffer.data, buffer.len) == -1) { - free(buffer.data); - if (apis != NULL) - free(apis); - close(cli_sockfd); - continue; - } - if (apis != NULL) free(apis); + if (write(cli_sockfd, buffer.data, buffer.len) == -1) + LOG_ERR("Failed to send reply message."); + free(buffer.data); close(cli_sockfd); } diff --git a/src/lib/ipcp.c b/src/lib/ipcp.c index 76098691..141db43d 100644 --- a/src/lib/ipcp.c +++ b/src/lib/ipcp.c @@ -81,25 +81,11 @@ static ipcp_msg_t * send_recv_ipcp_msg(pid_t api, ipcp_msg__pack(msg, buf.data); - if (write(sockfd, buf.data, buf.len) == -1) { - free(buf.data); - close(sockfd); - return NULL; - } - - count = read(sockfd, buf.data, IPCP_MSG_BUF_SIZE); - if (count <= 0) { - free(buf.data); - close(sockfd); - return NULL; - } + if (write(sockfd, buf.data, buf.len) != -1) + count = read(sockfd, buf.data, IPCP_MSG_BUF_SIZE); - recv_msg = ipcp_msg__unpack(NULL, count, buf.data); - if (recv_msg == NULL) { - free(buf.data); - close(sockfd); - return NULL; - } + if (count > 0) + recv_msg = ipcp_msg__unpack(NULL, count, buf.data); pthread_cleanup_pop(true); pthread_cleanup_pop(true); diff --git a/src/lib/sockets.c b/src/lib/sockets.c index 5d861cdf..0d93c0a9 100644 --- a/src/lib/sockets.c +++ b/src/lib/sockets.c @@ -134,25 +134,12 @@ irm_msg_t * send_recv_irm_msg(irm_msg_t * msg) irm_msg__pack(msg, buf.data); - if (write(sockfd, buf.data, buf.len) == -1) { - free(buf.data); - close(sockfd); - return NULL; - } + if (write(sockfd, buf.data, buf.len) != -1) + count = read(sockfd, buf.data, IRM_MSG_BUF_SIZE); - count = read(sockfd, buf.data, IRM_MSG_BUF_SIZE); - if (count <= 0) { - free(buf.data); - close(sockfd); - return NULL; - } - recv_msg = irm_msg__unpack(NULL, count, buf.data); - if (recv_msg == NULL) { - free(buf.data); - close(sockfd); - return NULL; - } + if (count > 0) + recv_msg = irm_msg__unpack(NULL, count, buf.data); pthread_cleanup_pop(true); pthread_cleanup_pop(true); |