UnixServer.hpp

// SPDX-FileCopyrightText: 2025 Vincent Leroy
// SPDX-License-Identifier: MIT
//
// This file is part of dfx.
//
// Licensed under the MIT License. See the LICENSE file in the project root
// for full license information.
 
#pragma once
 
// Standard includes
#include <chrono>
#include <mutex>
#include <unordered_map>
 
// Project includes
#include "UnixSession.hpp"
#include <dfx-fdwatch/Events.hpp>
#include <dfx-fdwatch/PollerFd.hpp>
#include <dfx-fdwatch/Timer.hpp>
#include <dfx-utilities/CopyMoveControl.hpp>
#include <dfx-utilities/FileSystem.hpp>
 
namespace dfx::FdWatch
{
class Poller;
} // !namespace dfx::FdWatch
 
namespace dfx::Utils
{
class SystemConfig;
} // !namespace dfx::Utils
 
namespace dfx::Server
{
class UnixRouter;

class UnixServer
{
public:
    struct Options
    {
        fs::path socket;         
        uint32_t permission = 0; 
        uint32_t backlog = 0;    

        std::chrono::milliseconds inactivityTimeout{0};

        std::chrono::milliseconds timeoutAccuracy{0};

        uint64_t errCountBeforeClose = 0;

        uint64_t consecutiveErrCountBeforeClose = 0;
    };
 
public:
    UnixServer(Utils::SystemConfig & sysConfig, FdWatch::Poller & poller, UnixRouter & router, std::string const & defaultSocketPath = "/var/run/dfx.sock");

    DISABLE_COPY_AND_MOVE(UnixServer);

    ~UnixServer();

    void listen();

    void stop() noexcept;

    bool isListening() const noexcept { return _serverSocket.isRegistered(); }

    void terminateSession(UnixSessionPtr session);
 
private:
    void _onTimerTimeout();
    void _serverSocketReadyRead(FdWatch::EventTriggers events);
    void _onSessionReadyRead(UnixSessionPtr session, FdWatch::EventTriggers events);
 
private:
    FdWatch::Poller & _poller;
    UnixRouter & _router;
    Options _options;
    bool _isInit = false;
    FdWatch::PollerFd _serverSocket;
    FdWatch::Timer _timer;
 
private:
    std::mutex _sessionMutex;
    std::unordered_map<FdWatch::BorrowedFd, UnixSessionPtr> _sessions;
};
} // !namespace dfx::Server