Do flow-specific setsockopt() before accept()
This commit is contained in:
@@ -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));
|
||||
|
||||
Reference in New Issue
Block a user