Try to read a little more

This commit is contained in:
angt
2016-02-03 19:49:42 +01:00
parent bfcfd3263e
commit 605b68dad5

82
mud.c
View File

@@ -396,56 +396,58 @@ int mud_pull (struct mud *mud)
struct sock *sock; struct sock *sock;
for (sock = mud->sock; sock; sock = sock->next) { for (sock = mud->sock; sock; sock = sock->next) {
unsigned char next = mud->rx.end+1; for (int i = 0; i < 16; i++) {
unsigned char next = mud->rx.end+1;
if (mud->rx.start == next) if (mud->rx.start == next)
return 0; return 0;
struct packet *packet = &mud->rx.packet[mud->rx.end]; struct packet *packet = &mud->rx.packet[mud->rx.end];
struct sockaddr_storage addr; struct sockaddr_storage addr;
socklen_t addrlen = sizeof(addr); socklen_t addrlen = sizeof(addr);
ssize_t ret = recvfrom(sock->fd, packet->data, sizeof(packet->data), ssize_t ret = recvfrom(sock->fd, packet->data, sizeof(packet->data),
0, (struct sockaddr *)&addr, &addrlen); 0, (struct sockaddr *)&addr, &addrlen);
if (ret<=0) if (ret <= 0)
continue; break;
struct path *path = mud_new_path(mud, sock->fd, &addr, addrlen); struct path *path = mud_new_path(mud, sock->fd, &addr, addrlen);
if (!path) if (!path)
return -1; return -1;
uint32_t send_now = mud_read32(packet->data); uint32_t send_now = mud_read32(packet->data);
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->dt = mud_read32(&packet->data[8]);
path->rtt = now-send_now; path->rtt = now-send_now;
continue; continue;
}
if (path->recv_count == 256) {
unsigned char reply[3*4];
uint32_t dt = (now-path->recv_time)>>8;
path->recv_count = 0;
path->recv_time = now;
memset(reply, 0, 4);
memcpy(&reply[4], packet->data, 4);
mud_write32(&reply[8], dt);
mud_send_path(path, reply, sizeof(reply));
} else {
path->recv_count++;
}
packet->size = ret;
// packet->time = now;
mud->rx.end = next;
} }
if (path->recv_count == 256) {
unsigned char reply[3*4];
uint32_t dt = (now-path->recv_time)>>8;
path->recv_count = 0;
path->recv_time = now;
memset(reply, 0, 4);
memcpy(&reply[4], packet->data, 4);
mud_write32(&reply[8], dt);
mud_send_path(path, reply, sizeof(reply));
} else {
path->recv_count++;
}
packet->size = ret;
// packet->time = now;
mud->rx.end = next;
} }
return 0; return 0;