natto1784/matar
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

utils/tcp_server: set TCP_NODELAY

Browse Source 
Signed-off-by: Amneesh Singh <natto@weirdnatto.in>
This commit is contained in:
Amneesh Singh
2024-06-20 06:12:42 +05:30
parent 1c96f418eb
commit f510a54d40
2 changed files with 14 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
src/util/tcp_server.cc
View File

@@ -1,4 +1,5 @@
#include "tcp_server.hh" #include "tcp_server.hh"
#include <netinet/tcp.h>
#include <cstring> #include <cstring>
#include <format> #include <format>
@@ -35,13 +36,22 @@ TcpServer::run() {
void void
TcpServer::start(uint port) { TcpServer::start(uint port) {
int opts = 0;
server_fd = socket(PF_INET, SOCK_STREAM, 0); server_fd = socket(PF_INET, SOCK_STREAM, 0);
if (server_fd == -1) { if (server_fd == -1) {
throw std::runtime_error("creating socket failed"); throw std::runtime_error("creating socket failed");
} }
int option = 1; int option = 1;
setsockopt(server_fd, SOL_SOCKET, SO_REUSEADDR, &option, sizeof(option));
opts +=
setsockopt(server_fd, SOL_SOCKET, SO_REUSEADDR, &option, sizeof(option));
opts +=
setsockopt(server_fd, IPPROTO_TCP, TCP_NODELAY, &option, sizeof(option));
if (opts != 0) {
throw std::runtime_error("failed to set socket opts");
}
std::memset(&server_addr, 0, sizeof(server_addr)); std::memset(&server_addr, 0, sizeof(server_addr));
server_addr.sin_family = PF_INET; server_addr.sin_family = PF_INET;
@@ -68,9 +78,8 @@ TcpServer::send(std::string msg) {
std::string std::string
TcpServer::receive(uint length) { TcpServer::receive(uint length) {
char msg[MAX_PACKET_SIZE];
ssize_t num_bytes = recv(client_fd, msg, length, 0); ssize_t num_bytes = recv(client_fd, msg, length, 0);
msg[length] = '\0'; msg[num_bytes] = '\0';
if (num_bytes < 0) { if (num_bytes < 0) {
throw std::runtime_error( throw std::runtime_error(
std::format("failed to receive messages: {}\n", strerror(errno))); std::format("failed to receive messages: {}\n", strerror(errno)));

2
src/util/tcp_server.hh
View File

@@ -14,10 +14,12 @@ class TcpServer {
void start(uint port); void start(uint port);
void send(std::string msg); void send(std::string msg);
std::string receive(uint length); std::string receive(uint length);
std::string receive_all() { return receive(MAX_PACKET_SIZE); };
bool client_waiting(); bool client_waiting();
private: private:
static constexpr uint MAX_PACKET_SIZE = 4096; static constexpr uint MAX_PACKET_SIZE = 4096;
char msg[MAX_PACKET_SIZE];
int server_fd; int server_fd;
int client_fd; int client_fd;
sockaddr_in server_addr; sockaddr_in server_addr;