void ServerPool::newTcpConnection(qt_socket_fd_t socket)
{
// Ignore connections from an SSL server for now, these are only handled
// by HttpServer which overrides newTcpConnection
PrivTcpServer *server = dynamic_cast<PrivTcpServer *>(QObject::sender());
if (!server || server->GetServerType() == kSSLServer)
return;
QTcpSocket *qsock = new QTcpSocket(this);
if (qsock->setSocketDescriptor(socket))
{
emit newConnection(qsock);
}
else
delete qsock;
}
void WebSocketServer::newTcpConnection(qt_socket_fd_t socket)
{
PoolServerType type = kTCPServer;
PrivTcpServer *server = dynamic_cast<PrivTcpServer *>(QObject::sender());
if (server)
type = server->GetServerType();
m_threadPool.startReserved(
new WebSocketWorkerThread(*this, socket, type
#ifndef QT_NO_OPENSSL
, m_sslConfig
#endif
),
QString("WebSocketServer%1").arg(socket));
}
bool ServerPool::listen(QList<QHostAddress> addrs, quint16 port,
bool requireall)
{
m_port = port;
QList<QHostAddress>::const_iterator it;
for (it = addrs.begin(); it != addrs.end(); ++it)
{
PrivTcpServer *server = new PrivTcpServer(this);
server->setProxy(m_proxy);
server->setMaxPendingConnections(m_maxPendingConn);
connect(server, SIGNAL(newConnection(int)),
this, SLOT(newTcpConnection(int)));
if (server->listen(*it, m_port))
{
LOG(VB_GENERAL, LOG_INFO, QString("Listening on TCP %1:%2")
.arg(PRETTYIP(it)).arg(port));
m_tcpServers.append(server);
if (m_port == 0)
m_port = server->serverPort();
}
else if (requireall)
{
LOG(VB_GENERAL, LOG_ERR, QString("Failed listening on TCP %1:%2")
.arg(PRETTYIP(it)).arg(port));
close();
server->disconnect();
server->deleteLater();
return false;
}
else
{
LOG(VB_GENERAL, LOG_WARNING, QString("Failed listening on TCP %1:%2")
.arg(PRETTYIP(it)).arg(port));
server->disconnect();
server->deleteLater();
}
}
if (m_tcpServers.size() == 0)
return false;
m_listening = true;
return true;
}
void ServerPool::close(void)
{
PrivTcpServer *server;
while (!m_tcpServers.isEmpty())
{
server = m_tcpServers.takeLast();
server->disconnect();
server->close();
server->deleteLater();
}
PrivUdpSocket *socket;
while (!m_udpSockets.isEmpty())
{
socket = m_udpSockets.takeLast();
socket->disconnect();
socket->close();
socket->deleteLater();
}
m_lastUdpSocket = NULL;
m_listening = false;
}
bool ServerPool::listen(QList<QHostAddress> addrs, quint16 port,
bool requireall, PoolServerType servertype)
{
m_port = port;
QList<QHostAddress>::const_iterator it;
for (it = addrs.begin(); it != addrs.end(); ++it)
{
PrivTcpServer *server = new PrivTcpServer(this, servertype);
#if (QT_VERSION >= 0x050000)
connect(server, &PrivTcpServer::newConnection,
this, &ServerPool::newTcpConnection);
#else
connect(server, SIGNAL(newConnection(qt_socket_fd_t)),
this, SLOT(newTcpConnection(qt_socket_fd_t)));
#endif
server->setProxy(m_proxy);
server->setMaxPendingConnections(m_maxPendingConn);
if (server->listen(*it, m_port))
{
LOG(VB_GENERAL, LOG_INFO, QString("Listening on TCP %1:%2")
.arg(PRETTYIP(it)).arg(port));
if (servertype == kTCPServer)
m_tcpServers.append(server);
else
if (m_port == 0)
m_port = server->serverPort();
}
else
{
LOG(VB_GENERAL, LOG_ERR,
QString("Failed listening on TCP %1:%2 - Error %3: %4")
.arg(PRETTYIP(it))
.arg(port)
.arg(server->serverError())
.arg(server->errorString()));
server->disconnect();
server->deleteLater();
if (server->serverError() == QAbstractSocket::HostNotFoundError)
{
LOG(VB_GENERAL, LOG_ERR,
QString("Address %1 no longer exists - ignoring")
.arg(PRETTYIP(it)));
continue;
}
if (requireall)
{
close();
return false;
}
}
}
if (m_tcpServers.size() == 0)
return false;
m_listening = true;
return true;
}