87 lines
2.9 KiB
C++
87 lines
2.9 KiB
C++
#pragma once
|
|
|
|
#ifdef _WIN32
|
|
#include <winsock2.h>
|
|
#include <ws2tcpip.h>
|
|
#else
|
|
#include <sys/socket.h>
|
|
#include <netinet/tcp.h>
|
|
#include <netinet/in.h>
|
|
#endif
|
|
|
|
namespace SocketOptimization {
|
|
// buffer sizes for different use cases
|
|
constexpr int WEBSOCKET_BUFFER_SIZE = 32768; // 32KB for WebSocket frames
|
|
constexpr int HTTP_BUFFER_SIZE = 65536; // 64KB for HTTP responses
|
|
constexpr int SSL_BUFFER_SIZE = 16384; // 16KB for SSL/TLS data
|
|
constexpr int SMALL_BUFFER_SIZE = 8192; // 8KB for small reads
|
|
constexpr int SOCKET_SEND_BUFFER = 256 * 1024; // 256KB send buffer
|
|
constexpr int SOCKET_RECV_BUFFER = 256 * 1024; // 256KB receive buffer
|
|
|
|
inline bool optimizeSocket(int socket_fd) {
|
|
if (socket_fd < 0) return false;
|
|
|
|
int nodelay = 1;
|
|
if (setsockopt(socket_fd, IPPROTO_TCP, TCP_NODELAY,
|
|
(const char*)&nodelay, sizeof(nodelay)) != 0) {
|
|
return false;
|
|
}
|
|
|
|
int send_buffer = SOCKET_SEND_BUFFER;
|
|
if (setsockopt(socket_fd, SOL_SOCKET, SO_SNDBUF,
|
|
(const char*)&send_buffer, sizeof(send_buffer)) != 0) {
|
|
return false;
|
|
}
|
|
|
|
int recv_buffer = SOCKET_RECV_BUFFER;
|
|
if (setsockopt(socket_fd, SOL_SOCKET, SO_RCVBUF,
|
|
(const char*)&recv_buffer, sizeof(recv_buffer)) != 0) {
|
|
return false;
|
|
}
|
|
|
|
// disable Nagle
|
|
int tcp_quickack = 1;
|
|
#ifndef _WIN32
|
|
setsockopt(socket_fd, IPPROTO_TCP, TCP_QUICKACK,
|
|
(const char*)&tcp_quickack, sizeof(tcp_quickack));
|
|
#endif
|
|
|
|
return true;
|
|
}
|
|
|
|
inline bool optimizeWebSocket(int socket_fd) {
|
|
if (!optimizeSocket(socket_fd)) return false;
|
|
|
|
int keepalive = 1;
|
|
if (setsockopt(socket_fd, SOL_SOCKET, SO_KEEPALIVE,
|
|
(const char*)&keepalive, sizeof(keepalive)) != 0) {
|
|
return false;
|
|
}
|
|
|
|
#ifndef _WIN32
|
|
// unix keep-alive
|
|
int keepidle = 30; // start probes after 30 seconds
|
|
int keepintvl = 5; // probe every 5 seconds
|
|
int keepcnt = 3; // 3 failed probes before timeout
|
|
|
|
setsockopt(socket_fd, IPPROTO_TCP, TCP_KEEPIDLE, &keepidle, sizeof(keepidle));
|
|
setsockopt(socket_fd, IPPROTO_TCP, TCP_KEEPINTVL, &keepintvl, sizeof(keepintvl));
|
|
setsockopt(socket_fd, IPPROTO_TCP, TCP_KEEPCNT, &keepcnt, sizeof(keepcnt));
|
|
#endif
|
|
|
|
return true;
|
|
}
|
|
|
|
inline bool optimizeHttp(int socket_fd) {
|
|
if (!optimizeSocket(socket_fd)) return false;
|
|
|
|
int reuse = 1;
|
|
if (setsockopt(socket_fd, SOL_SOCKET, SO_REUSEADDR,
|
|
(const char*)&reuse, sizeof(reuse)) != 0) {
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
};
|