76 lines
2.2 KiB
C++
76 lines
2.2 KiB
C++
/*
|
|
* This file is open source software, licensed to you under the terms
|
|
* of the Apache License, Version 2.0 (the "License"). See the NOTICE file
|
|
* distributed with this work for additional information regarding copyright
|
|
* ownership. You may not use this file except in compliance with the License.
|
|
*
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing,
|
|
* software distributed under the License is distributed on an
|
|
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
|
* KIND, either express or implied. See the License for the
|
|
* specific language governing permissions and limitations
|
|
* under the License.
|
|
*/
|
|
#include "core/reactor.hh"
|
|
#include "proxy.hh"
|
|
#include <utility>
|
|
|
|
namespace net {
|
|
|
|
class proxy_net_device : public qp {
|
|
private:
|
|
static constexpr size_t _send_queue_length = 128;
|
|
size_t _send_depth = 0;
|
|
unsigned _cpu;
|
|
device* _dev;
|
|
std::vector<packet> _moving;
|
|
public:
|
|
explicit proxy_net_device(unsigned cpu, device* dev);
|
|
virtual future<> send(packet p) override {
|
|
abort();
|
|
}
|
|
virtual uint32_t send(circular_buffer<packet>& p) override;
|
|
};
|
|
|
|
proxy_net_device::proxy_net_device(unsigned cpu, device* dev) :
|
|
_cpu(cpu),
|
|
_dev(dev)
|
|
{
|
|
_moving.reserve(_send_queue_length);
|
|
}
|
|
|
|
uint32_t proxy_net_device::send(circular_buffer<packet>& p)
|
|
{
|
|
if (!_moving.empty() || _send_depth == _send_queue_length) {
|
|
return 0;
|
|
}
|
|
|
|
for (size_t i = 0; !p.empty() && _send_depth < _send_queue_length; i++, _send_depth++) {
|
|
_moving.push_back(std::move(p.front()));
|
|
p.pop_front();
|
|
}
|
|
|
|
if (!_moving.empty()) {
|
|
qp* dev = &_dev->queue_for_cpu(_cpu);
|
|
auto cpu = engine().cpu_id();
|
|
smp::submit_to(_cpu, [this, dev, cpu]() mutable {
|
|
for(size_t i = 0; i < _moving.size(); i++) {
|
|
dev->proxy_send(_moving[i].free_on_cpu(cpu, [this] { _send_depth--; }));
|
|
}
|
|
}).then([this] {
|
|
_moving.clear();
|
|
});
|
|
}
|
|
|
|
return _moving.size();
|
|
}
|
|
|
|
std::unique_ptr<qp> create_proxy_net_device(unsigned master_cpu, device* dev) {
|
|
return std::make_unique<proxy_net_device>(master_cpu, dev);
|
|
}
|
|
}
|