You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
253 lines
5.3 KiB
C++
253 lines
5.3 KiB
C++
5 years ago
|
// CrossSockets
|
||
|
// Copyright Jeroen Saey
|
||
|
// Created 27-01-2013
|
||
|
// CrossSocket.cpp
|
||
|
|
||
|
#include "CrossSocket.h"
|
||
|
|
||
|
#ifndef __WIN32__
|
||
|
#include <netdb.h>
|
||
|
#include <arpa/inet.h>
|
||
|
#include <fcntl.h>
|
||
|
#include <sys/select.h>
|
||
|
#include <sys/time.h>
|
||
|
#else
|
||
|
#define F_SETFL FIONBIO
|
||
|
#define O_NONBLOCK 1
|
||
|
|
||
|
extern int close(int fd);
|
||
|
extern int fcntl(int fd, int cmd, long arg);
|
||
|
#endif
|
||
|
|
||
|
using namespace std;
|
||
|
|
||
|
CrossSocket::CrossSocket(blockTypeEnum block)
|
||
|
{
|
||
|
_blockingMode = block;
|
||
|
}
|
||
|
|
||
|
CrossSocket::~CrossSocket()
|
||
|
{
|
||
|
|
||
|
}
|
||
|
|
||
|
void CrossSocket::getSocket()
|
||
|
{
|
||
|
if (_socketHandle < 0)
|
||
|
{
|
||
|
_socketHandle = socket(PF_INET, SOCK_STREAM, 0);
|
||
|
|
||
|
if (_blockingMode == nonblocking)
|
||
|
{
|
||
|
fcntl(_socketHandle, F_SETFL, O_NONBLOCK);
|
||
|
}
|
||
|
|
||
|
//reset state
|
||
|
reset();
|
||
|
}
|
||
|
}
|
||
|
|
||
|
BaseCrossSocket* CrossSocket::create(int socketdescriptor, CrossSocketErrors *error)
|
||
|
{
|
||
|
CrossSocket* remoteClass;
|
||
|
|
||
|
remoteClass = new CrossSocket(_blockingMode);
|
||
|
remoteClass->_socketHandle = socketdescriptor;
|
||
|
|
||
|
noError(error);
|
||
|
return remoteClass;
|
||
|
}
|
||
|
|
||
|
bool CrossSocket::bind(int port, CrossSocketErrors *error)
|
||
|
{
|
||
|
return bind(port, "", error);
|
||
|
}
|
||
|
|
||
|
bool CrossSocket::bind(int port, string host, CrossSocketErrors *error)
|
||
|
{
|
||
|
hostent *hostEntry;
|
||
|
in_addr networkAddress;
|
||
|
|
||
|
if (host.size() > 0 )
|
||
|
{
|
||
|
// Bind to a specific address
|
||
|
|
||
|
if ((hostEntry = gethostbyname(host.data())) == NULL)
|
||
|
{
|
||
|
setError(error, fatal, "CrossSocket::bind() - Can't get host by name");
|
||
|
return false;
|
||
|
}
|
||
|
|
||
|
networkAddress = *((in_addr *)hostEntry->h_addr);
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
// Bind to any address
|
||
|
networkAddress.s_addr = INADDR_ANY;
|
||
|
}
|
||
|
|
||
|
getSocket();
|
||
|
|
||
|
sockaddr_in serverAddress;
|
||
|
|
||
|
memset(&serverAddress, 0, sizeof(serverAddress));
|
||
|
serverAddress.sin_family = AF_INET;
|
||
|
serverAddress.sin_port = htons(port);
|
||
|
serverAddress.sin_addr.s_addr = networkAddress.s_addr;
|
||
|
|
||
|
if (::bind(_socketHandle, (sockaddr *)&serverAddress, sizeof(serverAddress)) == -1)
|
||
|
{
|
||
|
handleError(error, "CrossSocket::bind() error: ");
|
||
|
return false;
|
||
|
}
|
||
|
|
||
|
noError(error);
|
||
|
return true;
|
||
|
}
|
||
|
|
||
|
bool CrossSocket::connect(int port, string hostname, CrossSocketErrors *error)
|
||
|
{
|
||
|
getSocket();
|
||
|
|
||
|
hostent *host;
|
||
|
|
||
|
if ((host = gethostbyname(hostname.data())) == NULL )
|
||
|
{
|
||
|
setError(error, fatal, "CrossSocket::connect() - Can't get host by name");
|
||
|
return false;
|
||
|
}
|
||
|
|
||
|
sockaddr_in clientAddress;
|
||
|
|
||
|
memset(&clientAddress, 0, sizeof(clientAddress));
|
||
|
clientAddress.sin_family = AF_INET;
|
||
|
clientAddress.sin_port = htons(port);
|
||
|
clientAddress.sin_addr = *((in_addr *)host->h_addr);
|
||
|
|
||
|
if (::connect(_socketHandle, (sockaddr *)&clientAddress, sizeof(clientAddress)) == -1)
|
||
|
{
|
||
|
handleError(error, "CrossSocket::connect() error: ");
|
||
|
return false;
|
||
|
}
|
||
|
|
||
|
noError(error);
|
||
|
return true;
|
||
|
}
|
||
|
|
||
|
|
||
|
string CrossSocket::getClientAddress(CrossSocketErrors *error)
|
||
|
{
|
||
|
sockaddr_in clientAddress;
|
||
|
|
||
|
if (!getClientHost((sockaddr *)&clientAddress, error))
|
||
|
{
|
||
|
return "";
|
||
|
}
|
||
|
|
||
|
char *addressPointer;
|
||
|
|
||
|
if ((addressPointer = inet_ntoa(clientAddress.sin_addr)) == NULL )
|
||
|
{
|
||
|
setError(error, fatal, "CrossSocket::getClientHostName() - Can't get client address");
|
||
|
return "";
|
||
|
}
|
||
|
|
||
|
noError(error);
|
||
|
return string (addressPointer);
|
||
|
}
|
||
|
|
||
|
int CrossSocket::getClientPort(CrossSocketErrors *error)
|
||
|
{
|
||
|
sockaddr_in clientAddress;
|
||
|
|
||
|
if (!getClientHost((sockaddr *)&clientAddress, error))
|
||
|
{
|
||
|
return -1;
|
||
|
}
|
||
|
|
||
|
noError(error);
|
||
|
|
||
|
return ntohs(clientAddress.sin_port);
|
||
|
}
|
||
|
|
||
|
string CrossSocket::getClientHostName(CrossSocketErrors *error)
|
||
|
{
|
||
|
string name = getClientAddress(error);
|
||
|
|
||
|
if (name.size() < 1)
|
||
|
{
|
||
|
return "";
|
||
|
}
|
||
|
|
||
|
hostent *host;
|
||
|
|
||
|
if ((host = gethostbyname(name.data())) == NULL )
|
||
|
{
|
||
|
setError(error, fatal, "CrossSocket::getClientHostName() - Can't get client by address");
|
||
|
return "";
|
||
|
}
|
||
|
|
||
|
noError(error);
|
||
|
return string(host->h_name);;
|
||
|
}
|
||
|
|
||
|
string CrossSocket::getServerAddress(CrossSocketErrors *error)
|
||
|
{
|
||
|
//We need to get the real address, so we must
|
||
|
//first get this computers host name and then
|
||
|
//translate that into an address!
|
||
|
|
||
|
string name = getServerHostName(error);
|
||
|
if (name.size() < 1)
|
||
|
{
|
||
|
return "";
|
||
|
}
|
||
|
|
||
|
hostent *host;
|
||
|
|
||
|
if ((host = gethostbyname(name.data())) == NULL )
|
||
|
{
|
||
|
setError(error, fatal, "CrossSocket::getServerAddress() - Can't get host by name");
|
||
|
return "";
|
||
|
}
|
||
|
|
||
|
char *addressPointer;
|
||
|
|
||
|
if ((addressPointer = inet_ntoa(*((in_addr *)host->h_addr))) == NULL)
|
||
|
{
|
||
|
setError(error, fatal, "CrossSocket::getServerAddress() - Can't get host address");
|
||
|
return "";
|
||
|
}
|
||
|
return string(addressPointer);
|
||
|
}
|
||
|
|
||
|
int CrossSocket::getServerPort(CrossSocketErrors *error)
|
||
|
{
|
||
|
sockaddr_in address;
|
||
|
|
||
|
if (!getServerHost((sockaddr *)&address, error))
|
||
|
{
|
||
|
return -1;
|
||
|
}
|
||
|
|
||
|
noError(error);
|
||
|
|
||
|
return ntohs(address.sin_port);
|
||
|
}
|
||
|
|
||
|
string CrossSocket::getServerHostName(CrossSocketErrors *error)
|
||
|
{
|
||
|
char buffer[256];
|
||
|
|
||
|
if (gethostname(buffer, 256) != 0)
|
||
|
{
|
||
|
handleError(error, "CrossSocket::gethostname() error: ");
|
||
|
return "";
|
||
|
}
|
||
|
|
||
|
string message(buffer);
|
||
|
|
||
|
noError(error);
|
||
|
return message;
|
||
|
}
|