summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--src/ipcpd/ipcp.c4
-rw-r--r--src/irmd/main.c18
-rw-r--r--src/lib/ipcp.c22
-rw-r--r--src/lib/sockets.c21
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);