#include <assert.h>
#include <errno.h>
#include <netinet/in.h>
#include <netinet/tcp.h>
#include <sys/socket.h>
#include <sys/stat.h>
#include <sys/types.h>

#include "socket.h"

void socket_pre_bind(int fd) {
	// Called by transport code; safe to assume that fd is a socket
	int optval = 1;
	assert(!setsockopt(fd, SOL_SOCKET, SO_REUSEPORT, &optval, sizeof(optval)));
}

void socket_pre_listen(int fd) {
	// Called by transport code; safe to assume that fd is a socket
	int qlen = 5;
	assert(!setsockopt(fd, SOL_TCP, TCP_FASTOPEN, &qlen, sizeof(qlen)));
}

void socket_ready(int fd) {
	// Called by transport code; safe to assume that fd is a socket
	int optval = 1;
	assert(!setsockopt(fd, SOL_SOCKET, SO_KEEPALIVE, &optval, sizeof(optval)));
	optval = 30;
	assert(!setsockopt(fd, IPPROTO_TCP, TCP_KEEPIDLE, &optval, sizeof(optval)));
	optval = 10;
	assert(!setsockopt(fd, IPPROTO_TCP, TCP_KEEPINTVL, &optval, sizeof(optval)));
	optval = 3;
	assert(!setsockopt(fd, IPPROTO_TCP, TCP_KEEPCNT, &optval, sizeof(optval)));

	optval = 60000; // 60s
	assert(!setsockopt(fd, IPPROTO_TCP, TCP_USER_TIMEOUT, &optval, sizeof(optval)));
}

void socket_ready_send(int fd) {
	int optval = 128; // Lowest value that the kernel will accept
	int res = setsockopt(fd, SOL_SOCKET, SO_RCVBUF, &optval, sizeof(optval));
	if (res == -1 && errno == ENOTSOCK) {
		return;
	}
	assert(res == 0);

	optval = 128; // Lowest value that the kernel will accept
	assert(!setsockopt(fd, IPPROTO_TCP, TCP_WINDOW_CLAMP, &optval, sizeof(optval)));
}

void socket_connected_send(int fd) {
	// Called by data flow code; NOT safe to assume that fd is a socket
	int res = shutdown(fd, SHUT_RD);
	assert(res == 0 || (res == -1 && errno == ENOTSOCK));
}

void socket_connected_receive(int fd) {
	// Called by data flow code; NOT safe to assume that fd is a socket
	int res = shutdown(fd, SHUT_WR);
	assert(res == 0 || (res == -1 && errno == ENOTSOCK));
}