miegl/mud
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Code cleanup

Browse Source
This commit is contained in:
angt
2016-02-03 21:18:00 +01:00
parent 605b68dad5
commit 0596afef7e
2 changed files with 57 additions and 44 deletions
Download Patch File Download Diff File Expand all files Collapse all files

85
mud.c
View File

@@ -14,17 +14,19 @@
#define MUD_PKT_SIZE (2048u) #define MUD_PKT_SIZE (2048u)
struct path_info {
uint32_t dt;
uint32_t time;
unsigned count;
};
struct path { struct path {
int fd; int fd;
struct sockaddr_storage addr; struct sockaddr_storage addr;
socklen_t addrlen; socklen_t addrlen;
uint32_t rtt; uint32_t rtt;
uint32_t dt; struct path_info recv;
uint32_t send_dt; struct path_info send;
uint32_t recv_time;
unsigned recv_count;
uint32_t send_time;
unsigned send_count;
struct path *next; struct path *next;
}; };
@@ -340,6 +342,11 @@ int mud_encrypt (struct mud *mud, uint32_t nonce,
if (!src_size) if (!src_size)
return 0; return 0;
size_t size = src_size+4+crypto_aead_aes256gcm_ABYTES;
if (size > dst_size)
return 0;
unsigned char npub[crypto_aead_aes256gcm_NPUBBYTES] = {0}; unsigned char npub[crypto_aead_aes256gcm_NPUBBYTES] = {0};
mud_write32(npub, nonce); mud_write32(npub, nonce);
@@ -354,17 +361,22 @@ int mud_encrypt (struct mud *mud, uint32_t nonce,
memcpy(dst, npub, 4); memcpy(dst, npub, 4);
return src_size+4+crypto_aead_aes256gcm_ABYTES; return size;
} }
static static
size_t mud_decrypt (struct mud *mud, uint32_t *nonce, int mud_decrypt (struct mud *mud, uint32_t *nonce,
unsigned char *dst, size_t dst_size, unsigned char *dst, size_t dst_size,
const unsigned char *src, size_t src_size) const unsigned char *src, size_t src_size)
{ {
if (!src_size) if (!src_size)
return 0; return 0;
size_t size = src_size-4-crypto_aead_aes256gcm_ABYTES;
if (size > dst_size)
return 0;
unsigned char npub[crypto_aead_aes256gcm_NPUBBYTES] = {0}; unsigned char npub[crypto_aead_aes256gcm_NPUBBYTES] = {0};
memcpy(npub, src, 4); memcpy(npub, src, 4);
@@ -376,12 +388,12 @@ size_t mud_decrypt (struct mud *mud, uint32_t *nonce,
NULL, 0, NULL, 0,
npub, npub,
(const crypto_aead_aes256gcm_state *)&mud->crypto.key)) (const crypto_aead_aes256gcm_state *)&mud->crypto.key))
return 0; return -1;
if (nonce) if (nonce)
*nonce = mud_read32(src); *nonce = mud_read32(src);
return src_size-4-crypto_aead_aes256gcm_ABYTES; return size;
} }
int mud_pull (struct mud *mud) int mud_pull (struct mud *mud)
@@ -422,17 +434,17 @@ int mud_pull (struct mud *mud)
if (!send_now) { if (!send_now) {
send_now = mud_read32(&packet->data[4]); send_now = mud_read32(&packet->data[4]);
path->dt = mud_read32(&packet->data[8]); path->recv.dt = mud_read32(&packet->data[8]);
path->rtt = now-send_now; path->rtt = now-send_now;
continue; continue;
} }
if (path->recv_count == 256) { if (path->recv.count == 256) {
unsigned char reply[3*4]; unsigned char reply[3*4];
uint32_t dt = (now-path->recv_time)>>8; uint32_t dt = (now-path->recv.time)>>8;
path->recv_count = 0; path->recv.count = 0;
path->recv_time = now; path->recv.time = now;
memset(reply, 0, 4); memset(reply, 0, 4);
memcpy(&reply[4], packet->data, 4); memcpy(&reply[4], packet->data, 4);
@@ -440,7 +452,7 @@ int mud_pull (struct mud *mud)
mud_send_path(path, reply, sizeof(reply)); mud_send_path(path, reply, sizeof(reply));
} else { } else {
path->recv_count++; path->recv.count++;
} }
packet->size = ret; packet->size = ret;
@@ -453,13 +465,8 @@ int mud_pull (struct mud *mud)
return 0; return 0;
} }
ssize_t mud_recv (struct mud *mud, void *data, size_t size) int mud_recv (struct mud *mud, void *data, size_t size)
{ {
if (size+4+crypto_aead_aes256gcm_ABYTES < MUD_PKT_SIZE) {
errno = EMSGSIZE;
return -1;
}
if (mud->rx.start == mud->rx.end) { if (mud->rx.start == mud->rx.end) {
errno = EAGAIN; errno = EAGAIN;
return -1; return -1;
@@ -467,12 +474,16 @@ ssize_t mud_recv (struct mud *mud, void *data, size_t size)
struct packet *packet = &mud->rx.packet[mud->rx.start]; struct packet *packet = &mud->rx.packet[mud->rx.start];
ssize_t ret = mud_decrypt(mud, NULL, int ret = mud_decrypt(mud, NULL, data, size,
data, size,
packet->data, packet->size); packet->data, packet->size);
mud->rx.start++; mud->rx.start++;
if (ret == -1) {
errno = EINVAL;
return -1;
}
return ret; return ret;
} }
@@ -502,25 +513,20 @@ int mud_push (struct mud *mud)
if (ret != packet->size) if (ret != packet->size)
continue; continue;
if (path->send_count == 256) { if (path->send.count == 256) {
path->send_count = 0; path->send.count = 0;
path->send_dt = (now-path->send_time)>>8; path->send.dt = (now-path->send.time)>>8;
path->send_time = now; path->send.time = now;
} else { } else {
path->send_count++; path->send.count++;
} }
} }
return 0; return 0;
} }
ssize_t mud_send (struct mud *mud, const void *data, size_t size) int mud_send (struct mud *mud, const void *data, size_t size)
{ {
if (size+4+crypto_aead_aes256gcm_ABYTES > MUD_PKT_SIZE) {
errno = EMSGSIZE;
return -1;
}
uint32_t now = mud_now(); uint32_t now = mud_now();
if (!now) { if (!now) {
@@ -537,10 +543,17 @@ ssize_t mud_send (struct mud *mud, const void *data, size_t size)
struct packet *packet = &mud->tx.packet[mud->tx.end]; struct packet *packet = &mud->tx.packet[mud->tx.end];
packet->size = mud_encrypt(mud, now, int ret = mud_encrypt(mud, now,
packet->data, sizeof(packet->data), packet->data, sizeof(packet->data),
data, size); data, size);
if (!ret) {
errno = EMSGSIZE;
return -1;
}
packet->size = ret;
// packet->time = now; // packet->time = now;
mud->tx.end = next; mud->tx.end = next;

6
mud.h
View File

@@ -1,6 +1,6 @@
#pragma once #pragma once
#include <unistd.h> #include <stddef.h>
struct mud; struct mud;
@@ -13,5 +13,5 @@ int mud_peer (struct mud *, const char *, const char *);
int mud_pull (struct mud *); int mud_pull (struct mud *);
int mud_push (struct mud *); int mud_push (struct mud *);
ssize_t mud_recv (struct mud *, void *, size_t); int mud_recv (struct mud *, void *, size_t);
ssize_t mud_send (struct mud *, const void *, size_t); int mud_send (struct mud *, const void *, size_t);