aboutsummaryrefslogtreecommitdiffstatshomepage
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--src/msg.c9
-rw-r--r--src/net.c13
-rw-r--r--src/net.h5
3 files changed, 17 insertions, 10 deletions
diff --git a/src/msg.c b/src/msg.c
index d1f3b5e..d9e6c3b 100644
--- a/src/msg.c
+++ b/src/msg.c
@@ -2,6 +2,7 @@
#include "log.h"
#include "net.h"
+#include <stdint.h>
#include <stdlib.h>
#include <string.h>
@@ -94,9 +95,9 @@ int msg_from_argv(struct msg *msg, char **argv)
return msg_copy_argv(msg, argv);
}
-static size_t calc_buf_len(const struct msg *msg)
+static uint32_t calc_buf_len(const struct msg *msg)
{
- size_t len = 0;
+ uint32_t len = 0;
for (int i = 0; i < msg->argc; ++i)
len += strlen(msg->argv[i]) + 1;
return len;
@@ -151,7 +152,7 @@ int msg_send(int fd, const struct msg *msg)
{
int ret = 0;
- size_t len = calc_buf_len(msg);
+ uint32_t len = calc_buf_len(msg);
char *buf = malloc(len);
if (!buf) {
print_errno("malloc");
@@ -187,7 +188,7 @@ int msg_send_and_wait(int fd, const struct msg *request, struct msg *response)
int msg_recv(int fd, struct msg *msg)
{
void *buf;
- size_t len;
+ uint32_t len;
int ret = 0;
ret = net_recv_buf(fd, &buf, &len);
diff --git a/src/net.c b/src/net.c
index 534807c..f819af3 100644
--- a/src/net.c
+++ b/src/net.c
@@ -2,6 +2,7 @@
#include "log.h"
#include <netdb.h>
+#include <stdint.h>
#include <stdlib.h>
#include <string.h>
#include <sys/socket.h>
@@ -185,13 +186,15 @@ ssize_t net_recv_all(int fd, void *buf, size_t len)
return read_total;
}
-int net_send_buf(int fd, const void *buf, size_t len)
+int net_send_buf(int fd, const void *buf, uint32_t len)
{
int ret = 0;
+ len = htonl(len);
ret = net_send_all(fd, &len, sizeof(len));
if (ret < 0)
return ret;
+ len = ntohl(len);
ret = net_send_all(fd, buf, len);
if (ret < 0)
@@ -200,7 +203,7 @@ int net_send_buf(int fd, const void *buf, size_t len)
return ret;
}
-int net_recv_buf(int fd, void **buf, size_t *len)
+int net_recv_buf(int fd, void **buf, uint32_t *len)
{
ssize_t nb = 0;
@@ -213,6 +216,8 @@ int net_recv_buf(int fd, void **buf, size_t *len)
goto fail;
}
+ *len = ntohl(*len);
+
*buf = malloc(*len);
if (!*buf) {
print_errno("malloc");
@@ -240,7 +245,7 @@ fail:
int net_recv_static(int fd, void *buf, size_t len)
{
void *actual_buf;
- size_t actual_len;
+ uint32_t actual_len;
int ret = 0;
ret = net_recv_buf(fd, &actual_buf, &actual_len);
@@ -248,7 +253,7 @@ int net_recv_static(int fd, void *buf, size_t len)
return ret;
if (actual_len != len) {
- print_error("Expected message length: %lu, actual: %lu\n", len, actual_len);
+ print_error("Expected message length: %lu, actual: %u\n", len, actual_len);
ret = -1;
goto free_buf;
}
diff --git a/src/net.h b/src/net.h
index 125d4f3..5afe4b7 100644
--- a/src/net.h
+++ b/src/net.h
@@ -1,6 +1,7 @@
#ifndef __NET_H__
#define __NET_H__
+#include <stdint.h>
#include <stdlib.h>
#include <sys/types.h>
@@ -9,10 +10,10 @@ int net_accept(int fd);
int net_connect(const char *host, const char *port);
int net_send_all(int fd, const void *, size_t);
-int net_send_buf(int fd, const void *, size_t);
+int net_send_buf(int fd, const void *, uint32_t);
ssize_t net_recv_all(int fd, void *, size_t);
-int net_recv_buf(int fd, void **, size_t *);
+int net_recv_buf(int fd, void **, uint32_t *);
int net_recv_static(int fd, void *, size_t);
#endif