#include <dragonnet/recv.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>

void dragonnet_recv_raw(DragonnetPeer *p, void *buf, size_t n)
{
	pthread_rwlock_rdlock(&p->mu);
	int sock = p->sock;
	pthread_rwlock_unlock(&p->mu);

	ssize_t len = recv(sock, buf, n, MSG_WAITALL);
	if (len < 0) {
		perror("recv");
		dragonnet_peer_delete(p);
		return;
	}

	// Connection closed
	if (len == 0) {
		pthread_rwlock_wrlock(&p->mu);

		close(p->sock);
		p->sock = -1;
		p->state++;

		pthread_rwlock_unlock(&p->mu);
	}
}

void dragonnet_read_raw(u8 **buf, size_t *n, void *data, size_t len)
{
	memcpy(data, *buf, len);
	memcpy(*buf, &((*buf)[len]), -len + *n);

	*buf = realloc(*buf, -len + *n);
	*n -= len;
}