router.hh

Go to the documentation of this file.

#ifndef SPONGE_LIBSPONGE_ROUTER_HH
#define SPONGE_LIBSPONGE_ROUTER_HH
 
#include "network_interface.hh"
 
#include <optional>
#include <queue>
 
class AsyncNetworkInterface : public NetworkInterface {
    std::queue<InternetDatagram> _datagrams_out{};
 
  public:
    using NetworkInterface::NetworkInterface;
 
    AsyncNetworkInterface(NetworkInterface &&interface) : NetworkInterface(interface) {}
 
 
    void recv_frame(const EthernetFrame &frame) {
        auto optional_dgram = NetworkInterface::recv_frame(frame);
        if (optional_dgram.has_value()) {
            _datagrams_out.push(std::move(optional_dgram.value()));
        }
    };
 
    std::queue<InternetDatagram> &datagrams_out() { return _datagrams_out; }
};
 
class Router {
    std::vector<AsyncNetworkInterface> _interfaces{};
 
    void route_one_datagram(InternetDatagram &dgram);
 
  public:
    size_t add_interface(AsyncNetworkInterface &&interface) {
        _interfaces.push_back(std::move(interface));
        return _interfaces.size() - 1;
    }
 
    AsyncNetworkInterface &interface(const size_t N) { return _interfaces.at(N); }
 
    void add_route(const uint32_t route_prefix,
                   const uint8_t prefix_length,
                   const std::optional<Address> next_hop,
                   const size_t interface_num);
 
    void route();
};
 
#endif  // SPONGE_LIBSPONGE_ROUTER_HH