136 lines
2.9 KiB
C
136 lines
2.9 KiB
C
#include "tools/socket.h"
|
|
|
|
#include <string.h>
|
|
#include <arpa/inet.h>
|
|
#include <unistd.h>
|
|
#include <errno.h>
|
|
#include <fcntl.h>
|
|
|
|
cTBSocket::cTBSocket(int Type) {
|
|
memset(&m_LocalAddr, 0, sizeof(m_LocalAddr));
|
|
memset(&m_RemoteAddr, 0, sizeof(m_RemoteAddr));
|
|
m_Type = Type;
|
|
}
|
|
|
|
cTBSocket::~cTBSocket() {
|
|
if (IsOpen()) Close();
|
|
}
|
|
|
|
bool cTBSocket::Connect(const cTBString &Host, unsigned int Port) {
|
|
socklen_t len;
|
|
int socket;
|
|
|
|
if (IsOpen()) Close();
|
|
|
|
if ((socket = ::socket(PF_INET, m_Type, IPPROTO_IP)) == -1)
|
|
return false;
|
|
|
|
m_LocalAddr.sin_family = AF_INET;
|
|
m_LocalAddr.sin_port = 0;
|
|
m_LocalAddr.sin_addr.s_addr = INADDR_ANY;
|
|
if (::bind(socket, (struct sockaddr*)&m_LocalAddr, sizeof(m_LocalAddr))
|
|
== -1)
|
|
return false;
|
|
|
|
m_RemoteAddr.sin_family = AF_INET;
|
|
m_RemoteAddr.sin_port = htons(Port);
|
|
m_RemoteAddr.sin_addr.s_addr = inet_addr(Host);
|
|
if (::connect(socket, (struct sockaddr*)&m_RemoteAddr,
|
|
sizeof(m_RemoteAddr)) == -1)
|
|
return false;
|
|
|
|
len = sizeof(struct sockaddr_in);
|
|
if (::getpeername(socket, (struct sockaddr*)&m_RemoteAddr, &len) == -1)
|
|
return false;
|
|
|
|
len = sizeof(struct sockaddr_in);
|
|
if (::getsockname(socket, (struct sockaddr*)&m_LocalAddr, &len) == -1)
|
|
return false;
|
|
|
|
return cTBSource::Open(socket);
|
|
}
|
|
|
|
bool cTBSocket::Listen(const char *Ip, unsigned int Port, int BackLog) {
|
|
int val;
|
|
socklen_t len;
|
|
int socket;
|
|
|
|
if (IsOpen()) Close();
|
|
|
|
if ((socket = ::socket(PF_INET, m_Type, IPPROTO_IP)) == -1)
|
|
return false;
|
|
|
|
val = 1;
|
|
if (::setsockopt(socket, SOL_SOCKET, SO_REUSEADDR, &val, sizeof(val)) == -1)
|
|
return false;
|
|
|
|
m_LocalAddr.sin_family = AF_INET;
|
|
m_LocalAddr.sin_port = htons(Port);
|
|
m_LocalAddr.sin_addr.s_addr = inet_addr(Ip);
|
|
if (::bind(socket, (struct sockaddr*)&m_LocalAddr, sizeof(m_LocalAddr))
|
|
== -1)
|
|
return false;
|
|
|
|
len = sizeof(struct sockaddr_in);
|
|
if (::getsockname(socket, (struct sockaddr*)&m_LocalAddr, &len) == -1)
|
|
return false;
|
|
|
|
if (m_Type == SOCK_STREAM && ::listen(socket, BackLog) == -1)
|
|
return false;
|
|
|
|
if (!cTBSource::Open(socket))
|
|
return false;
|
|
|
|
return true;
|
|
}
|
|
|
|
bool cTBSocket::Accept(const cTBSocket &Listener) {
|
|
socklen_t addrlen;
|
|
int socket;
|
|
|
|
if (IsOpen()) Close();
|
|
|
|
addrlen = sizeof(struct sockaddr_in);
|
|
if ((socket = ::accept(Listener, (struct sockaddr*)&m_RemoteAddr,
|
|
&addrlen)) == -1)
|
|
return false;
|
|
|
|
addrlen = sizeof(struct sockaddr_in);
|
|
if (::getsockname(socket, (struct sockaddr*)&m_LocalAddr, &addrlen) == -1)
|
|
return false;
|
|
|
|
if (!cTBSource::Open(socket))
|
|
return false;
|
|
|
|
return true;
|
|
}
|
|
|
|
RETURNS(cTBSocket, cTBSocket::Accept(void) const, ret)
|
|
ret.Accept(*this);
|
|
RETURN(ret)
|
|
|
|
bool cTBSocket::Close(void) {
|
|
bool ret = true;
|
|
|
|
if (!IsOpen())
|
|
ERRNUL(EBADF);
|
|
|
|
if (::close(*this) == -1)
|
|
ret = false;
|
|
|
|
if (!cTBSource::Close())
|
|
ret = false;
|
|
|
|
memset(&m_LocalAddr, 0, sizeof(m_LocalAddr));
|
|
memset(&m_RemoteAddr, 0, sizeof(m_RemoteAddr));
|
|
|
|
return ret;
|
|
}
|
|
|
|
bool cTBSocket::Shutdown(int how) {
|
|
if (!IsOpen())
|
|
ERRNUL(EBADF);
|
|
|
|
return ::shutdown(*this, how) != -1;
|
|
}
|