mirror of https://github.com/debauchee/barrier.git
73 lines
1.2 KiB
C++
73 lines
1.2 KiB
C++
#include "CTCPListenSocket.h"
|
|
#include "CTCPSocket.h"
|
|
#include "CNetworkAddress.h"
|
|
#include "CThread.h"
|
|
#include <unistd.h>
|
|
#include <errno.h>
|
|
#include <sys/types.h>
|
|
#include <sys/socket.h>
|
|
#include <netinet/in.h>
|
|
|
|
//
|
|
// CTCPListenSocket
|
|
//
|
|
|
|
CTCPListenSocket::CTCPListenSocket()
|
|
{
|
|
m_fd = socket(PF_INET, SOCK_STREAM, 0);
|
|
if (m_fd == -1) {
|
|
throw XSocketCreate();
|
|
}
|
|
}
|
|
|
|
CTCPListenSocket::~CTCPListenSocket()
|
|
{
|
|
try {
|
|
close();
|
|
}
|
|
catch (...) {
|
|
// ignore
|
|
}
|
|
}
|
|
|
|
void CTCPListenSocket::bind(
|
|
const CNetworkAddress& addr)
|
|
{
|
|
if (::bind(m_fd, addr.getAddress(), addr.getAddressLength()) == -1) {
|
|
if (errno == EADDRINUSE) {
|
|
throw XSocketAddressInUse();
|
|
}
|
|
throw XSocketBind();
|
|
}
|
|
if (listen(m_fd, 3) == -1) {
|
|
throw XSocketBind();
|
|
}
|
|
}
|
|
|
|
ISocket* CTCPListenSocket::accept()
|
|
{
|
|
for (;;) {
|
|
struct sockaddr addr;
|
|
socklen_t addrlen = sizeof(addr);
|
|
CThread::testCancel();
|
|
int fd = ::accept(m_fd, &addr, &addrlen);
|
|
if (fd == -1) {
|
|
CThread::testCancel();
|
|
}
|
|
else {
|
|
return new CTCPSocket(fd);
|
|
}
|
|
}
|
|
}
|
|
|
|
void CTCPListenSocket::close()
|
|
{
|
|
if (m_fd == -1) {
|
|
throw XIOClosed();
|
|
}
|
|
if (::close(m_fd) == -1) {
|
|
throw XIOClose();
|
|
}
|
|
m_fd = -1;
|
|
}
|