90 lines
1.7 KiB
C++
90 lines
1.7 KiB
C++
#include "tcp_connection.h"
|
|
|
|
#include <sys/types.h>
|
|
#include <sys/socket.h>
|
|
#include <netinet/in.h>
|
|
#include <unistd.h>
|
|
#include <string.h>
|
|
#include <netdb.h>
|
|
|
|
|
|
namespace
|
|
{
|
|
|
|
static void set_port(struct sockaddr_storage *addr, uint16_t port)
|
|
{
|
|
if (addr->ss_family == AF_INET)
|
|
{
|
|
struct sockaddr_in *ip = (struct sockaddr_in*)addr;
|
|
ip->sin_port = htons(port);
|
|
}
|
|
else if (addr->ss_family == AF_INET6)
|
|
{
|
|
struct sockaddr_in6 *ip = (struct sockaddr_in6*)addr;
|
|
ip->sin6_port = htons(port);
|
|
}
|
|
}
|
|
|
|
int open_network_socket(struct sockaddr_storage *addr, const char *host, uint16_t port)
|
|
{
|
|
struct addrinfo *ai = NULL, hint = {};
|
|
int so = -1;
|
|
hint.ai_socktype = SOCK_STREAM;
|
|
hint.ai_family = AF_UNSPEC;
|
|
hint.ai_flags = AI_ALL;
|
|
|
|
/* Take the first one */
|
|
if (getaddrinfo(host, NULL, &hint, &ai) == 0 && ai)
|
|
{
|
|
if ((so = socket(ai->ai_family, SOCK_STREAM, 0)) != -1)
|
|
{
|
|
memcpy(addr, ai->ai_addr, ai->ai_addrlen);
|
|
set_port(addr, port);
|
|
freeaddrinfo(ai);
|
|
}
|
|
}
|
|
|
|
return so;
|
|
}
|
|
|
|
}
|
|
|
|
namespace tcp
|
|
{
|
|
|
|
Connection::~Connection()
|
|
{
|
|
if (m_so != -1)
|
|
{
|
|
close(m_so);
|
|
}
|
|
}
|
|
|
|
bool Connection::connect(const char* host, uint16_t port)
|
|
{
|
|
struct sockaddr_storage addr;
|
|
|
|
if ((m_so = open_network_socket(&addr, host, port)) != -1)
|
|
{
|
|
if (::connect(m_so, (struct sockaddr*)&addr, sizeof(addr)) != 0)
|
|
{
|
|
close(m_so);
|
|
m_so = -1;
|
|
}
|
|
}
|
|
|
|
return m_so != -1;
|
|
}
|
|
|
|
int Connection::write(void* buf, size_t size)
|
|
{
|
|
return ::write(m_so, buf, size);
|
|
}
|
|
|
|
int Connection::read(void* buf, size_t size)
|
|
{
|
|
return ::read(m_so, buf, size);
|
|
}
|
|
|
|
}
|