TcpSink.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 <memory>
 
// Project includes
#include "Sink.hpp"
#include <dfx-server/TcpServerForSession.hpp>
 
namespace dfx::Pcapng
{
class WriteOnlySession : public Server::TcpSession
{
public:
    using Server::TcpSession::TcpSession;
 
protected:
    void handleMessage(std::vector<uint8_t>) override {}
};

class TcpSink : public Sink
{
public:
    using Server    = Server::TcpServerForSession<WriteOnlySession>;
    using ServerPtr = std::unique_ptr<Server>;
 
public:
    TcpSink(std::string host = "localhost", uint16_t port = 19'000);
 
public:
    void init(std::vector<uint8_t> initialHeader) override;

    void write(std::vector<uint8_t> data) override;
 
private:
    Server::Options _serverOptions;
    ServerPtr _server;
};
} // !namespace dfx::Pcapng