Do flow-specific setsockopt() before accept()

This commit is contained in:
Ian Gulliver
2016-02-29 22:17:37 -08:00
parent a093b8a1b6
commit b6a582e95a
8 changed files with 43 additions and 32 deletions

View File

@@ -7,23 +7,19 @@
#include "socket.h"
void socket_pre_bind_init(int fd) {
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_bound_init(int fd) {
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)));
// These options are inherited through accept()
// Verified by flamingcow on kernel 4.2.0
socket_connected_init(fd);
}
void socket_connected_init(int fd) {
void socket_connected(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)));
@@ -35,25 +31,30 @@ void socket_connected_init(int fd) {
assert(!setsockopt(fd, IPPROTO_TCP, TCP_KEEPCNT, &optval, sizeof(optval)));
}
void socket_send_init(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_send(int fd) {
int optval = 128; // Lowest value that the kernel will accept
res = setsockopt(fd, SOL_SOCKET, SO_RCVBUF, &optval, sizeof(optval));
assert(res == 0 || (res == -1 && errno == ENOTSOCK));
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
res = setsockopt(fd, IPPROTO_TCP, TCP_WINDOW_CLAMP, &optval, sizeof(optval));
assert(res == 0 || (res == -1 && errno == ENOTSOCK));
assert(res == 0);
optval = 60000; // 60s
res = setsockopt(fd, IPPROTO_TCP, TCP_USER_TIMEOUT, &optval, sizeof(optval));
assert(res == 0);
}
void socket_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_receive_init(int fd) {
void socket_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));