summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--src/fastd.c30
-rw-r--r--src/fastd.h2
-rw-r--r--src/packet.h3
3 files changed, 16 insertions, 19 deletions
diff --git a/src/fastd.c b/src/fastd.c
index bd9c13c..dadcd17 100644
--- a/src/fastd.c
+++ b/src/fastd.c
@@ -540,25 +540,14 @@ static void handle_tun(fastd_context *ctx) {
}
static void handle_socket(fastd_context *ctx, int sockfd) {
- size_t max_len = methods_max_packet_size(ctx);
- fastd_buffer buffer = fastd_buffer_alloc(max_len, methods_min_decrypt_head_space(ctx), 0);
- uint8_t packet_type;
+ size_t max_len = PACKET_TYPE_LEN + methods_max_packet_size(ctx);
+ fastd_buffer buffer = fastd_buffer_alloc(max_len, ALIGN8(methods_min_decrypt_head_space(ctx)), 0);
+ uint8_t *packet_type;
- struct iovec iov[2] = {
- { .iov_base = &packet_type, .iov_len = 1 },
- { .iov_base = buffer.data, .iov_len = max_len }
- };
fastd_peer_address recvaddr;
+ socklen_t recvaddrlen = sizeof(recvaddr);
- struct msghdr msg;
- memset(&msg, 0, sizeof(msg));
-
- msg.msg_name = &recvaddr;
- msg.msg_namelen = sizeof(recvaddr);
- msg.msg_iov = iov;
- msg.msg_iovlen = 2;
-
- ssize_t len = recvmsg(sockfd, &msg, 0);
+ ssize_t len = recvfrom(sockfd, buffer.data, buffer.len, 0, (struct sockaddr*)&recvaddr, &recvaddrlen);
if (len < 0) {
if (errno != EINTR)
pr_warn(ctx, "recvfrom: %s", strerror(errno));
@@ -567,7 +556,10 @@ static void handle_socket(fastd_context *ctx, int sockfd) {
return;
}
- buffer.len = len - 1;
+ packet_type = buffer.data;
+ buffer.len = len;
+
+ fastd_buffer_push_head(&buffer, 1);
fastd_peer *peer;
for (peer = ctx->peers; peer; peer = peer->next) {
@@ -576,7 +568,7 @@ static void handle_socket(fastd_context *ctx, int sockfd) {
}
if (peer) {
- switch (packet_type) {
+ switch (*packet_type) {
case PACKET_DATA:
ctx->conf->protocol->handle_recv(ctx, peer, buffer);
break;
@@ -592,7 +584,7 @@ static void handle_socket(fastd_context *ctx, int sockfd) {
else if(ctx->conf->n_floating || ctx->conf->n_dynamic ||
(recvaddr.sa.sa_family == AF_INET && ctx->conf->n_dynamic_v4) ||
(recvaddr.sa.sa_family == AF_INET6 && ctx->conf->n_dynamic_v6)) {
- switch (packet_type) {
+ switch (*packet_type) {
case PACKET_DATA:
fastd_buffer_free(buffer);
ctx->conf->protocol->handshake_init(ctx, &recvaddr, NULL);
diff --git a/src/fastd.h b/src/fastd.h
index f68ea08..12702e7 100644
--- a/src/fastd.h
+++ b/src/fastd.h
@@ -349,6 +349,8 @@ static inline void fastd_string_stack_free(fastd_string_stack *str) {
}
}
+#define ALIGN8(l) (((l+7)/8)*8)
+
static inline bool timespec_after(const struct timespec *tp1, const struct timespec *tp2) {
return (tp1->tv_sec > tp2->tv_sec ||
(tp1->tv_sec == tp2->tv_sec && tp1->tv_nsec > tp2->tv_nsec));
diff --git a/src/packet.h b/src/packet.h
index 210955d..16c37ce 100644
--- a/src/packet.h
+++ b/src/packet.h
@@ -30,6 +30,9 @@
#include <stdint.h>
+#define PACKET_TYPE_LEN 1
+
+
typedef enum _fastd_packet_type {
PACKET_UNKNOWN = 0,
PACKET_HANDSHAKE,