a1ee97c222
Default to listening on both IPv44 and IPv6 for public interfaces. Change-Id: Ib04e3bf65e7dcf2a798d381297b15ee9c56e9259
113 lines
3.2 KiB
C++
113 lines
3.2 KiB
C++
/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4; fill-column: 100 -*- */
|
|
/*
|
|
* This file is part of the LibreOffice project.
|
|
*
|
|
* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
*/
|
|
|
|
#ifndef INCLUDED_SERVERSOCKET_HPP
|
|
#define INCLUDED_SERVERSOCKET_HPP
|
|
|
|
#include "memory"
|
|
|
|
#include "Socket.hpp"
|
|
#include "Log.hpp"
|
|
|
|
#include <Poco/Net/SocketAddress.h>
|
|
|
|
class SocketFactory
|
|
{
|
|
public:
|
|
virtual std::shared_ptr<Socket> create(const int fd) = 0;
|
|
};
|
|
|
|
/// A non-blocking, streaming socket.
|
|
class ServerSocket : public Socket
|
|
{
|
|
public:
|
|
ServerSocket(Socket::Type type, SocketPoll& clientPoller, std::shared_ptr<SocketFactory> sockFactory) :
|
|
Socket(type),
|
|
_type(type),
|
|
_clientPoller(clientPoller),
|
|
_sockFactory(std::move(sockFactory))
|
|
{
|
|
}
|
|
|
|
enum Type { Local, Public };
|
|
|
|
/// Binds to a local address (Servers only).
|
|
/// Does not retry on error.
|
|
/// Returns true only on success.
|
|
bool bind(Type type, int port);
|
|
|
|
/// Listen to incoming connections (Servers only).
|
|
/// Does not retry on error.
|
|
/// Returns true on success only.
|
|
bool listen(const int backlog = 64)
|
|
{
|
|
const int rc = ::listen(getFD(), backlog);
|
|
|
|
if (rc)
|
|
LOG_SYS("Failed to listen");
|
|
return rc == 0;
|
|
}
|
|
|
|
/// Accepts an incoming connection (Servers only).
|
|
/// Does not retry on error.
|
|
/// Returns a valid Socket shared_ptr on success only.
|
|
std::shared_ptr<Socket> accept()
|
|
{
|
|
// Accept a connection (if any) and set it to non-blocking.
|
|
// We don't care about the client's address, so ignored.
|
|
const int rc = ::accept4(getFD(), nullptr, nullptr, SOCK_NONBLOCK);
|
|
LOG_DBG("Accepted socket #" << rc << ", creating socket object.");
|
|
try
|
|
{
|
|
// Create a socket object using the factory.
|
|
return rc != -1 ? _sockFactory->create(rc) : std::shared_ptr<Socket>(nullptr);
|
|
}
|
|
catch (const std::exception& ex)
|
|
{
|
|
LOG_SYS("Failed to create client socket #" << rc << ". Error: " << ex.what());
|
|
}
|
|
|
|
return nullptr;
|
|
}
|
|
|
|
int getPollEvents(std::chrono::steady_clock::time_point /* now */,
|
|
int & /* timeoutMaxMs */) override
|
|
{
|
|
return POLLIN;
|
|
}
|
|
|
|
void dumpState(std::ostream& os) override;
|
|
|
|
void handlePoll(SocketDisposition &,
|
|
std::chrono::steady_clock::time_point /* now */,
|
|
int events) override
|
|
{
|
|
if (events & POLLIN)
|
|
{
|
|
std::shared_ptr<Socket> clientSocket = accept();
|
|
if (!clientSocket)
|
|
{
|
|
const std::string msg = "Failed to accept. (errno: ";
|
|
throw std::runtime_error(msg + std::strerror(errno) + ")");
|
|
}
|
|
|
|
LOG_DBG("Accepted client #" << clientSocket->getFD());
|
|
_clientPoller.insertNewSocket(clientSocket);
|
|
}
|
|
}
|
|
|
|
private:
|
|
Socket::Type _type;
|
|
SocketPoll& _clientPoller;
|
|
std::shared_ptr<SocketFactory> _sockFactory;
|
|
};
|
|
|
|
#endif
|
|
|
|
/* vim:set shiftwidth=4 softtabstop=4 expandtab: */
|