scylladb/net/net.cc

/*
 * Copyright (C) 2014 Cloudius Systems, Ltd.
 *
 */

#include "net.hh"
#include <utility>

using std::move;

namespace net {

l3_protocol::l3_protocol(interface* netif, uint16_t proto_num)
    : _netif(netif), _proto_num(proto_num) {
    _netif->register_l3(proto_num, 1000);
}

future<packet, ethernet_address> l3_protocol::receive() {
    return _netif->receive(_proto_num);
};

future<> l3_protocol::send(ethernet_address to, packet p) {
    return _netif->send(_proto_num, to, std::move(p));
}

future<packet, ethernet_address> interface::receive(uint16_t proto_num) {
    auto i = _proto_map.find(proto_num);
    assert(i != _proto_map.end());
    auto& q = i->second;
    return q.not_empty().then([&q] {
        auto x = q.pop();
        return make_ready_future<packet, ethernet_address>(std::move(std::get<0>(x)), std::move(std::get<1>(x)));
    });
}

interface::interface(std::unique_ptr<device> dev)
    : _dev(std::move(dev)), _hw_address(_dev->hw_address()) {
}

void interface::register_l3(uint16_t proto_num, size_t queue_length) {
    _proto_map.emplace(std::piecewise_construct,
                std::make_tuple(proto_num),
                std::make_tuple(queue_length));
}

void interface::run() {
    _dev->receive().then([this] (packet p) {
        auto eh = p.get_header<eth_hdr>();
        if (eh) {
            ntoh(*eh);
            auto i = _proto_map.find(eh->eth_proto);
            if (i != _proto_map.end() && !i->second.full()) {
                auto from = eh->src_mac;
                p.trim_front(sizeof(*eh));
                i->second.push(std::make_tuple(std::move(p), from));
            } else {
                print("dropping packet: no handler for protcol 0x%x\n", eh->eth_proto);
            }
        }
        run();
    });
}

future<> interface::send(uint16_t proto_num, ethernet_address to, packet p) {
    auto eh = p.prepend_header<eth_hdr>();
    eh->dst_mac = to;
    eh->src_mac = _hw_address;
    eh->eth_proto = proto_num;
    hton(*eh);
    return _dev->send(std::move(p));
}

}