78 lines
1.9 KiB
C
78 lines
1.9 KiB
C
|
#include "msg.h"
|
||
|
|
#include "hosts.h"
|
||
|
|
#include <poll.h>
|
||
|
|
#include <stdint.h>
|
||
|
|
#include <stdio.h>
|
||
|
|
#include <stdlib.h>
|
||
|
|
#include <string.h>
|
||
|
|
|
||
|
|
fp *packet_to_fp(packet *p) {
|
||
|
|
fp *f = malloc(sizeof(fp));
|
||
|
|
f->val = be64toh(((uint64_t *)p->data)[0]);
|
||
|
|
return f;
|
||
|
|
}
|
||
|
|
|
||
|
|
packet *fp_to_packet(fp *elem) {
|
||
|
|
uint64_t *buf = malloc(sizeof(char) * 8);
|
||
|
|
buf[0] = htobe64(elem->val);
|
||
|
|
packet *p = malloc(sizeof(packet));
|
||
|
|
p->data = (uint8_t *)buf;
|
||
|
|
p->len = 8;
|
||
|
|
return p;
|
||
|
|
}
|
||
|
|
|
||
|
|
packet *to_packet(uint8_t *data, uint8_t len) {
|
||
|
|
packet *p = malloc(sizeof(packet));
|
||
|
|
p->data = data;
|
||
|
|
p->len = len;
|
||
|
|
return p;
|
||
|
|
}
|
||
|
|
|
||
|
|
void broadcast(struct host **conns, int num_conns, packet *data) {
|
||
|
|
|
||
|
|
char *transmit = malloc(sizeof(char) * data->len + 1);
|
||
|
|
transmit[0] = (char)data->len;
|
||
|
|
memcpy((transmit + 1), data->data, data->len);
|
||
|
|
for (int i = 0; i < num_conns; i++) {
|
||
|
|
sendto(conns[i]->sockfd, transmit, data->len + 1, 0,
|
||
|
|
(const struct sockaddr *)&conns[i]->conn, sizeof(conns[i]->conn));
|
||
|
|
}
|
||
|
|
free(transmit);
|
||
|
|
}
|
||
|
|
|
||
|
|
packet *recv_broadcasts(struct host **conns, int num_conns) {
|
||
|
|
|
||
|
|
packet *packets = malloc(sizeof(packet) * num_conns);
|
||
|
|
|
||
|
|
for (int i = 0; i < num_conns; i++) {
|
||
|
|
packets[i].data = NULL;
|
||
|
|
packets[i].len = 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
for (int i = 0; i < num_conns; i++) {
|
||
|
|
uint8_t *buf = malloc(sizeof(uint8_t) * 64);
|
||
|
|
socklen_t len = sizeof(conns[i]->conn);
|
||
|
|
recvfrom(conns[i]->sockfd, buf, 64, 0, (struct sockaddr *)&(conns[i]->conn),
|
||
|
|
&len);
|
||
|
|
uint8_t size = buf[0];
|
||
|
|
packets[i].data = (buf + 1);
|
||
|
|
packets[i].len = size;
|
||
|
|
}
|
||
|
|
|
||
|
|
return packets;
|
||
|
|
}
|
||
|
|
|
||
|
|
packet *recv_single_broadcast(struct host *con) {
|
||
|
|
packet *p = malloc(sizeof(packet));
|
||
|
|
p->data = NULL;
|
||
|
|
p->len = 0;
|
||
|
|
uint8_t *buf = malloc(sizeof(uint8_t) * 64);
|
||
|
|
socklen_t len = sizeof(con->conn);
|
||
|
|
recvfrom(con->sockfd, buf, 64, 0, (struct sockaddr *)&(con->conn), &len);
|
||
|
|
uint8_t size = buf[0];
|
||
|
|
p->data = (buf + 1);
|
||
|
|
p->len = size;
|
||
|
|
|
||
|
|
return p;
|
||
|
|
}
|