1 files changed, 89 insertions, 0 deletions
diff --git a/libtransport/src/io_modules/forwarder/forwarder_module.cc b/libtransport/src/io_modules/forwarder/forwarder_module.cc
new file mode 100644
index 000000000..ca9466f01
--- /dev/null
+++ b/libtransport/src/io_modules/forwarder/forwarder_module.cc
@@ -0,0 +1,89 @@
+/*
+ * Copyright (c) 2021 Cisco and/or its affiliates.
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * 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 <glog/logging.h>
+#include <hicn/transport/errors/not_implemented_exception.h>
+#include <io_modules/forwarder/forwarder_module.h>
+
+namespace transport {
+
+namespace core {
+
+ForwarderModule::ForwarderModule()
+    : IoModule(),
+      name_(""),
+      connector_id_(Connector::invalid_connector),
+      forwarder_ptr_(ForwarderGlobal::getInstance().getReference()),
+      forwarder_(*forwarder_ptr_) {}
+
+ForwarderModule::~ForwarderModule() {}
+
+bool ForwarderModule::isConnected() { return true; }
+
+void ForwarderModule::send(Packet &packet) {
+  IoModule::send(packet);
+  forwarder_.send(packet, connector_id_);
+  DLOG_IF(INFO, VLOG_IS_ON(3))
+      << "Sending from " << connector_id_ << " to " << 1 - connector_id_;
+}
+
+void ForwarderModule::send(const utils::MemBuf::Ptr &buffer) {
+  // not supported
+  throw errors::NotImplementedException();
+}
+
+void ForwarderModule::registerRoute(const Prefix &prefix) {
+  // For the moment we route packets from one socket to the other.
+  // Next step will be to introduce a FIB
+  return;
+}
+
+void ForwarderModule::closeConnection() {
+  forwarder_.deleteConnector(connector_id_);
+}
+
+void ForwarderModule::init(Connector::PacketReceivedCallback &&receive_callback,
+                           Connector::PacketSentCallback &&sent_callback,
+                           Connector::OnCloseCallback &&close_callback,
+                           Connector::OnReconnectCallback &&reconnect_callback,
+                           asio::io_service &io_service,
+                           const std::string &app_name) {
+  connector_id_ = forwarder_.registerLocalConnector(
+      io_service, std::move(receive_callback), std::move(sent_callback),
+      std::move(close_callback), std::move(reconnect_callback));
+  name_ = app_name;
+}
+
+void ForwarderModule::processControlMessageReply(utils::MemBuf &packet_buffer) {
+  return;
+}
+
+void ForwarderModule::connect(bool is_consumer) {
+  forwarder_.getConnector(connector_id_)
+      ->setRole(is_consumer ? Connector::Role::CONSUMER
+                            : Connector::Role::PRODUCER);
+}
+
+std::uint32_t ForwarderModule::getMtu() { return interface_mtu; }
+
+bool ForwarderModule::isControlMessage(utils::MemBuf &packet_buffer) {
+  return false;
+}
+
+extern "C" IoModule *create_module(void) { return new ForwarderModule(); }
+
+}  // namespace core
+
+}  // namespace transport