Skip to content

Commit 21bab87

Browse files
committed
Use setSocketOptionAndWarnOnError instead of ur_setsockopt directly
1 parent c73d273 commit 21bab87

2 files changed

Lines changed: 10 additions & 8 deletions

File tree

src/comm/tcp_server.cpp

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -70,11 +70,11 @@ void TCPServer::init()
7070
{
7171
throw makeSocketError("Failed to create socket endpoint");
7272
}
73-
int flag = 1;
73+
constexpr int flag = 1;
7474
#ifndef _WIN32
75-
ur_setsockopt(listen_fd_, SOL_SOCKET, SO_REUSEADDR, &flag, sizeof(int));
75+
setSocketOptionAndWarnOnError(listen_fd_, SOL_SOCKET, SO_REUSEADDR, &flag, sizeof(flag), "SO_REUSEADDR");
7676
#endif
77-
ur_setsockopt(listen_fd_, SOL_SOCKET, SO_KEEPALIVE, &flag, sizeof(int));
77+
setSocketOptionAndWarnOnError(listen_fd_, SOL_SOCKET, SO_KEEPALIVE, &flag, sizeof(flag), "SO_KEEPALIVE");
7878

7979
URCL_LOG_DEBUG("Created socket with FD %d", (int)listen_fd_);
8080

src/comm/tcp_socket.cpp

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,7 @@
2424
#include <cstring>
2525
#include <sstream>
2626
#include <thread>
27+
#include "ur_client_library/comm/socket_t.h"
2728

2829
#ifndef _WIN32
2930
# include <arpa/inet.h>
@@ -52,22 +53,23 @@ TCPSocket::~TCPSocket()
5253

5354
void TCPSocket::setupOptions()
5455
{
55-
int flag = 1;
56-
ur_setsockopt(socket_fd_, IPPROTO_TCP, TCP_NODELAY, &flag, sizeof(int));
56+
constexpr int flag = 1;
57+
setSocketOptionAndWarnOnError(socket_fd_, IPPROTO_TCP, TCP_NODELAY, &flag, sizeof(flag), "TCP_NODELAY");
5758

5859
// macOS does not have TCP_QUICKACK
5960
#ifdef TCP_QUICKACK
60-
ur_setsockopt(socket_fd_, IPPROTO_TCP, TCP_QUICKACK, &flag, sizeof(int));
61+
setSocketOptionAndWarnOnError(socket_fd_, IPPROTO_TCP, TCP_QUICKACK, &flag, sizeof(flag), "TCP_QUICKACK");
6162
#endif
6263

6364
if (recv_timeout_ != nullptr)
6465
{
6566
#ifdef _WIN32
6667
DWORD value = recv_timeout_->tv_sec * 1000;
6768
value += recv_timeout_->tv_usec / 1000;
68-
ur_setsockopt(socket_fd_, SOL_SOCKET, SO_RCVTIMEO, &value, sizeof(value));
69+
setSocketOptionAndWarnOnError(socket_fd_, SOL_SOCKET, SO_RCVTIMEO, &value, sizeof(value), "SO_RCVTIMEO");
6970
#else
70-
ur_setsockopt(socket_fd_, SOL_SOCKET, SO_RCVTIMEO, recv_timeout_.get(), sizeof(timeval));
71+
setSocketOptionAndWarnOnError(socket_fd_, SOL_SOCKET, SO_RCVTIMEO, recv_timeout_.get(), sizeof(timeval),
72+
"SO_RCVTIMEO");
7173
#endif
7274
}
7375
}

0 commit comments

Comments
 (0)