net: add struct buf

1 files changed, 24 insertions, 17 deletions

diff --git a/src/msg.c b/src/msg.c
index df8e14e..5e18681 100644
--- a/src/msg.c
+++ b/src/msg.c
@@ -102,7 +102,7 @@ int msg_from_argv(struct msg *msg, char **argv)
 	return msg_copy_argv(msg, argv);
 }
 
-static uint32_t calc_buf_len(const struct msg *msg)
+static uint32_t calc_buf_size(const struct msg *msg)
 {
 	uint32_t len = 0;
 	for (int i = 0; i < msg->argc; ++i)
@@ -157,22 +157,30 @@ free:
 
 int msg_send(int fd, const struct msg *msg)
 {
+	struct buf *buf;
 	int ret = 0;
 
-	uint32_t len = calc_buf_len(msg);
-	char *buf = malloc(len);
-	if (!buf) {
+	uint32_t size = calc_buf_size(msg);
+	char *data = malloc(size);
+	if (!data) {
 		log_errno("malloc");
 		return -1;
 	}
-	argv_pack(buf, msg);
+	argv_pack(data, msg);
 
-	ret = net_send_buf(fd, buf, len);
+	ret = buf_create(&buf, data, size);
 	if (ret < 0)
-		goto free_buf;
+		goto free_data;
 
-free_buf:
-	free(buf);
+	ret = net_send_buf(fd, buf);
+	if (ret < 0)
+		goto destroy_buf;
+
+destroy_buf:
+	buf_destroy(buf);
+
+free_data:
+	free(data);
 
 	return ret;
 }
@@ -194,22 +202,21 @@ int msg_send_and_wait(int fd, const struct msg *request, struct msg *response)
 
 int msg_recv(int fd, struct msg *msg)
 {
-	void *buf;
-	uint32_t len;
+	struct buf *buf;
 	int ret = 0;
 
-	ret = net_recv_buf(fd, &buf, &len);
+	ret = net_recv_buf(fd, &buf);
 	if (ret < 0)
 		return ret;
 
-	msg->argc = calc_argv_len(buf, len);
+	msg->argc = calc_argv_len(buf_get_data(buf), buf_get_size(buf));
 
-	ret = argv_unpack(msg, buf);
+	ret = argv_unpack(msg, buf_get_data(buf));
 	if (ret < 0)
-		goto free_buf;
+		goto destroy_buf;
 
-free_buf:
-	free(buf);
+destroy_buf:
+	buf_destroy(buf);
 
 	return ret;
 }