NFQueue.hpp

Go to the documentation of this file.

#pragma once
#include "bsc/model/Connection.hpp"
#include <atomic>
#include <cstdint>
#include <functional>
#include <memory>
#include <optional>
#include <string>
 
namespace bsc::netfilter {
 
enum class Verdict : std::uint32_t {
    Drop = 0,
    Accept = 1,
    Queue = 3,
    Repeat = 4,
};
 
struct PacketInfo {
    std::uint32_t id{0};
    std::uint32_t mark{0};
    std::uint32_t uid{0xffffffffu};  // 0xffffffff = &quot;non disponibile&quot;, 0 = root valido
    std::uint32_t in_dev{0};
    std::uint32_t out_dev{0};
    std::uint16_t queue_id{0};
    bsc::model::Connection connection;
};
 
struct Decision {
    Verdict verdict{Verdict::Accept};
    std::uint32_t mark{0};
    bool set_mark{false};
};
 
class NFQueue {
  public:
    using Handler = std::function<Decision(const PacketInfo&)>;
 
    NFQueue();
    ~NFQueue();
 
    bool open(std::uint16_t queue_id, Handler handler);
    void close();
    bool running() const noexcept { return running_; }
 
  private:
    class Impl;
    std::unique_ptr<Impl> impl_;
    std::atomic_bool running_{false};
};
 
bool available();
 
} // namespace bsc::netfilter