[Lldb-commits] [lldb] deeafea - [lldb][NFC] Move few static helpers to the class Socket (#106640)
via lldb-commits
lldb-commits at lists.llvm.org
Wed Sep 4 02:38:58 PDT 2024
Author: Dmitry Vasilyev
Date: 2024-09-04T13:38:55+04:00
New Revision: deeafeab815ddfe7b507e9f79fe8f992265a9f3b
URL: https://github.com/llvm/llvm-project/commit/deeafeab815ddfe7b507e9f79fe8f992265a9f3b
DIFF: https://github.com/llvm/llvm-project/commit/deeafeab815ddfe7b507e9f79fe8f992265a9f3b.diff
LOG: [lldb][NFC] Move few static helpers to the class Socket (#106640)
Fixed a typo in Socket::SetOption().
Added:
Modified:
lldb/include/lldb/Host/Socket.h
lldb/source/Host/common/Socket.cpp
lldb/source/Host/common/TCPSocket.cpp
Removed:
################################################################################
diff --git a/lldb/include/lldb/Host/Socket.h b/lldb/include/lldb/Host/Socket.h
index 304a91bdf6741b..764a048976eb41 100644
--- a/lldb/include/lldb/Host/Socket.h
+++ b/lldb/include/lldb/Host/Socket.h
@@ -112,8 +112,17 @@ class Socket : public IOObject {
static llvm::Expected<std::unique_ptr<UDPSocket>>
UdpConnect(llvm::StringRef host_and_port, bool child_processes_inherit);
- int GetOption(int level, int option_name, int &option_value);
- int SetOption(int level, int option_name, int option_value);
+ static int GetOption(NativeSocket sockfd, int level, int option_name,
+ int &option_value);
+ int GetOption(int level, int option_name, int &option_value) {
+ return GetOption(m_socket, level, option_name, option_value);
+ };
+
+ static int SetOption(NativeSocket sockfd, int level, int option_name,
+ int option_value);
+ int SetOption(int level, int option_name, int option_value) {
+ return SetOption(m_socket, level, option_name, option_value);
+ };
NativeSocket GetNativeSocket() const { return m_socket; }
SocketProtocol GetSocketProtocol() const { return m_protocol; }
@@ -138,6 +147,8 @@ class Socket : public IOObject {
virtual size_t Send(const void *buf, const size_t num_bytes);
+ static int CloseSocket(NativeSocket sockfd);
+ static Status GetLastError();
static void SetLastError(Status &error);
static NativeSocket CreateSocket(const int domain, const int type,
const int protocol,
diff --git a/lldb/source/Host/common/Socket.cpp b/lldb/source/Host/common/Socket.cpp
index 1a506aa95b2465..1a63571b94c6f1 100644
--- a/lldb/source/Host/common/Socket.cpp
+++ b/lldb/source/Host/common/Socket.cpp
@@ -386,11 +386,7 @@ Status Socket::Close() {
LLDB_LOGF(log, "%p Socket::Close (fd = %" PRIu64 ")",
static_cast<void *>(this), static_cast<uint64_t>(m_socket));
-#if defined(_WIN32)
- bool success = closesocket(m_socket) == 0;
-#else
- bool success = ::close(m_socket) == 0;
-#endif
+ bool success = CloseSocket(m_socket) == 0;
// A reference to a FD was passed in, set it to an invalid value
m_socket = kInvalidSocketValue;
if (!success) {
@@ -400,18 +396,20 @@ Status Socket::Close() {
return error;
}
-int Socket::GetOption(int level, int option_name, int &option_value) {
+int Socket::GetOption(NativeSocket sockfd, int level, int option_name,
+ int &option_value) {
get_socket_option_arg_type option_value_p =
reinterpret_cast<get_socket_option_arg_type>(&option_value);
socklen_t option_value_size = sizeof(int);
- return ::getsockopt(m_socket, level, option_name, option_value_p,
+ return ::getsockopt(sockfd, level, option_name, option_value_p,
&option_value_size);
}
-int Socket::SetOption(int level, int option_name, int option_value) {
+int Socket::SetOption(NativeSocket sockfd, int level, int option_name,
+ int option_value) {
set_socket_option_arg_type option_value_p =
- reinterpret_cast<get_socket_option_arg_type>(&option_value);
- return ::setsockopt(m_socket, level, option_name, option_value_p,
+ reinterpret_cast<set_socket_option_arg_type>(&option_value);
+ return ::setsockopt(sockfd, level, option_name, option_value_p,
sizeof(option_value));
}
@@ -427,6 +425,24 @@ void Socket::SetLastError(Status &error) {
#endif
}
+Status Socket::GetLastError() {
+ std::error_code EC;
+#ifdef _WIN32
+ EC = llvm::mapWindowsError(WSAGetLastError());
+#else
+ EC = std::error_code(errno, std::generic_category());
+#endif
+ return EC;
+}
+
+int Socket::CloseSocket(NativeSocket sockfd) {
+#ifdef _WIN32
+ return ::closesocket(sockfd);
+#else
+ return ::close(sockfd);
+#endif
+}
+
NativeSocket Socket::CreateSocket(const int domain, const int type,
const int protocol,
bool child_processes_inherit, Status &error) {
diff --git a/lldb/source/Host/common/TCPSocket.cpp b/lldb/source/Host/common/TCPSocket.cpp
index fc005814308d90..4f1518ef697ff0 100644
--- a/lldb/source/Host/common/TCPSocket.cpp
+++ b/lldb/source/Host/common/TCPSocket.cpp
@@ -33,28 +33,9 @@
#include <winsock2.h>
#endif
-#ifdef _WIN32
-#define CLOSE_SOCKET closesocket
-typedef const char *set_socket_option_arg_type;
-#else
-#include <unistd.h>
-#define CLOSE_SOCKET ::close
-typedef const void *set_socket_option_arg_type;
-#endif
-
using namespace lldb;
using namespace lldb_private;
-static Status GetLastSocketError() {
- std::error_code EC;
-#ifdef _WIN32
- EC = llvm::mapWindowsError(WSAGetLastError());
-#else
- EC = std::error_code(errno, std::generic_category());
-#endif
- return EC;
-}
-
static const int kType = SOCK_STREAM;
TCPSocket::TCPSocket(bool should_close, bool child_processes_inherit)
@@ -208,12 +189,8 @@ Status TCPSocket::Listen(llvm::StringRef name, int backlog) {
continue;
// enable local address reuse
- int option_value = 1;
- set_socket_option_arg_type option_value_p =
- reinterpret_cast<set_socket_option_arg_type>(&option_value);
- if (::setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, option_value_p,
- sizeof(option_value)) == -1) {
- CLOSE_SOCKET(fd);
+ if (SetOption(fd, SOL_SOCKET, SO_REUSEADDR, 1) == -1) {
+ CloseSocket(fd);
continue;
}
@@ -229,8 +206,8 @@ Status TCPSocket::Listen(llvm::StringRef name, int backlog) {
err = ::listen(fd, backlog);
if (err == -1) {
- error = GetLastSocketError();
- CLOSE_SOCKET(fd);
+ error = GetLastError();
+ CloseSocket(fd);
continue;
}
@@ -251,7 +228,7 @@ Status TCPSocket::Listen(llvm::StringRef name, int backlog) {
void TCPSocket::CloseListenSockets() {
for (auto socket : m_listen_sockets)
- CLOSE_SOCKET(socket.first);
+ CloseSocket(socket.first);
m_listen_sockets.clear();
}
@@ -280,7 +257,7 @@ llvm::Expected<std::vector<MainLoopBase::ReadHandleUP>> TCPSocket::Accept(
const lldb_private::SocketAddress &AddrIn = m_listen_sockets[fd];
if (!AddrIn.IsAnyAddr() && AcceptAddr != AddrIn) {
- CLOSE_SOCKET(sock);
+ CloseSocket(sock);
LLDB_LOG(log, "rejecting incoming connection from {0} (expecting {1})",
AcceptAddr.GetIPAddress(), AddrIn.GetIPAddress());
return;
More information about the lldb-commits
mailing list