#include "cafe/nsysnet.h"

#include "socket.h"

Socket::Socket() : sock(-1) {}

Socket::~Socket() {
	if (sock >= 0) {
		socketclose(sock);
	}
}

bool Socket::init(Type type) {
	if (type == TCP) {
		sock = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
	}
	else {
		sock = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
	}
	return sock >= 0;
}

bool Socket::close() {
	if (sock >= 0) {
		int result = socketclose(sock);
		sock = -1;
		return result == 0;
	}
	return true;
}


bool Client::sendall(const void *data, size_t length) {
	size_t sent = 0;
	while (sent < length) {
		int num = send(sock, data, length - sent, 0);
		if (num < 0) {
			close();
			return false;
		}
		
		sent += num;
		data = (const char *)data + num;
	}
	return true;
}

bool Client::recvall(void *data, size_t length) {
	size_t received = 0;
	while (received < length) {
		int num = recv(sock, data, length - received, 0);
		if (num <= 0) {
			close();
			return false;
		}
		
		received += num;
		data = (char *)data + num;
	}
	return true;
}


bool Server::bind(int port) {
	uint32_t reuseaddr = 1;
	int result = setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, &reuseaddr, 4);
	if (result < 0) {
		close();
		return false;
	}
		
	sockaddr serverAddr = {0};
	serverAddr.family = AF_INET;
	serverAddr.port = port;
	serverAddr.addr = 0;
	
	result = ::bind(sock, &serverAddr, 16);
	if (result < 0) {
		close();
		return false;
	}
	return true;
}

bool Server::accept(Client *client) {
	int result = listen(sock, 1);
	if (result < 0) {
		close();
		return false;
	}
	
	int fd = ::accept(sock, 0, 0);
	if (fd < 0) {
		close();
		return false;
	}
	
	client->sock = fd;
	return true;
}