Cool_Tools/ImHex
1
0
Fork 0
mirror of synced 2025-01-18 00:56:49 +01:00
Code Releases Wiki Activity

sys: Fix sockets compiling on Unix

Browse Source
This commit is contained in:
WerWolv 2021-12-07 23:09:30 +01:00
parent cc5a437573
commit c55146a78c
2 changed files with 26 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
plugins/libimhex/include/hex/helpers/socket.hpp
View File

@ -7,10 +7,17 @@
#if defined(OS_WINDOWS)
#include <winsock2.h>
#include <ws2tcpip.h>
#include <windows.h>
#define SOCKET_NONE INVALID_SOCKET
#else
#include <unistd.h>
#include <sys/sock.h>
#include <sys/socket.h>
#include <netinet/ip.h>
#include <arpa/inet.h>
#define SOCKET_NONE -1
#endif
namespace hex {
@ -39,9 +46,9 @@ namespace hex {
private:
bool m_connected = false;
#if defined(OS_WINDOWS)
SOCKET m_socket = INVALID_SOCKET;
SOCKET m_socket = SOCKET_NONE;
#else
int m_socket = -1;
int m_socket = SOCKET_NONE;
#endif
};

24
plugins/libimhex/source/helpers/socket.cpp
View File

@ -3,8 +3,6 @@
#include <hex/helpers/utils.hpp>
#include <hex/helpers/logger.hpp>
#include <ws2tcpip.h>
namespace hex {
Socket::Socket(const std::string &address, u16 port) {
@ -26,7 +24,7 @@ namespace hex {
this->m_socket = other.m_socket;
this->m_connected = other.m_connected;
other.m_socket = INVALID_SOCKET;
other.m_socket = SOCKET_NONE;
}
Socket::~Socket() {
@ -74,21 +72,31 @@ namespace hex {
void Socket::connect(const std::string &address, u16 port) {
this->m_socket = ::socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
if (this->m_socket == INVALID_SOCKET)
if (this->m_socket == SOCKET_NONE)
return;
sockaddr_in client = { 0 };
client.sin_family = AF_INET;
client.sin_addr.S_un.S_addr = ::inet_addr(address.c_str());
client.sin_port = ::htons(port);
this->m_connected = ::connect(this->m_socket, reinterpret_cast<SOCKADDR*>(&client), sizeof(client)) != SOCKET_ERROR;
#if defined(OS_WINDOWS)
client.sin_addr.S_un.S_addr = ::inet_addr(address.c_str());
#else
client.sin_addr.s_addr = ::inet_addr(address.c_str());
#endif
this->m_connected = ::connect(this->m_socket, reinterpret_cast<sockaddr*>(&client), sizeof(client)) == 0;
}
void Socket::disconnect() {
if (this->m_socket != INVALID_SOCKET)
closesocket(this->m_socket);
if (this->m_socket != SOCKET_NONE) {
#if defined(OS_WINDOWS)
closesocket(this->m_socket);
#else
close(this->m_socket);
#endif
}
this->m_connected = false;
}