summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--package/wprobe/src/user/wprobe-lib.c6
1 files changed, 5 insertions, 1 deletions
diff --git a/package/wprobe/src/user/wprobe-lib.c b/package/wprobe/src/user/wprobe-lib.c
index 7ba67cf..7a5460b 100644
--- a/package/wprobe/src/user/wprobe-lib.c
+++ b/package/wprobe/src/user/wprobe-lib.c
@@ -560,7 +560,9 @@ wprobe_msg_to_network(int socket, struct nl_msg *msg)
mhdr.status = WPROBE_MSG_DATA;
mhdr.len = buflen;
wprobe_swap_msg_hdr(&mhdr);
- write(socket, &mhdr, sizeof(mhdr));
+ ret = write(socket, &mhdr, sizeof(mhdr));
+ if (ret < 0)
+ goto out;
memcpy(buf, nlh, buflen);
nlh = buf;
@@ -572,6 +574,8 @@ wprobe_msg_to_network(int socket, struct nl_msg *msg)
swap_genlmsghdr(gnlh);
swap_nlmsghdr(nlh);
ret = write(socket, buf, buflen);
+
+out:
free(buf);
return ret;