1 files changed, 134 insertions, 0 deletions
diff --git a/libtransport/src/protocols/production_protocol.h b/libtransport/src/protocols/production_protocol.h
new file mode 100644
index 000000000..09718631f
--- /dev/null
+++ b/libtransport/src/protocols/production_protocol.h
@@ -0,0 +1,134 @@
+/*
+ * 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.
+ */
+
+#pragma once
+
+#include <hicn/transport/interfaces/callbacks.h>
+#include <hicn/transport/interfaces/socket_producer.h>
+#include <hicn/transport/interfaces/statistics.h>
+#include <hicn/transport/utils/object_pool.h>
+#include <implementation/socket.h>
+#include <protocols/fec_base.h>
+#include <protocols/fec_utils.h>
+#include <protocols/protocol.h>
+#include <utils/content_store.h>
+
+#include <atomic>
+#include <thread>
+
+namespace transport {
+
+namespace protocol {
+
+using namespace core;
+
+class ProductionProtocol
+    : public Protocol,
+      public std::enable_shared_from_this<ProductionProtocol> {
+ public:
+  ProductionProtocol(implementation::ProducerSocket *icn_socket);
+  virtual ~ProductionProtocol();
+
+  virtual int start();
+  using Protocol::stop;
+
+  virtual void setProducerParam(){};
+
+  virtual void produce(ContentObject &content_object);
+  virtual void sendMapme();
+  virtual uint32_t produceStream(const Name &content_name,
+                                 std::unique_ptr<utils::MemBuf> &&buffer,
+                                 bool is_last = true,
+                                 uint32_t start_offset = 0) = 0;
+  virtual uint32_t produceStream(const Name &content_name,
+                                 const uint8_t *buffer, size_t buffer_size,
+                                 bool is_last = true,
+                                 uint32_t start_offset = 0) = 0;
+  virtual uint32_t produceDatagram(const Name &content_name,
+                                   std::unique_ptr<utils::MemBuf> &&buffer) = 0;
+  virtual uint32_t produceDatagram(const Name &content_name,
+                                   const uint8_t *buffer,
+                                   size_t buffer_size) = 0;
+
+  void setOutputBufferSize(std::size_t size) { output_buffer_.setLimit(size); }
+  std::size_t getOutputBufferSize() { return output_buffer_.getLimit(); }
+
+ protected:
+  // Producer callback
+  virtual void onInterest(core::Interest &i) override = 0;
+  virtual void onError(const std::error_code &ec) override;
+
+  template <typename FECHandler, typename AllocatorHandler>
+  void enableFEC(FECHandler &&fec_handler,
+                 AllocatorHandler &&allocator_handler) {
+    if (!fec_encoder_) {
+      // Try to get FEC from environment
+      const char *fec_str = std::getenv("TRANSPORT_FEC_TYPE");
+      if (fec_str && (fec_type_ == fec::FECType::UNKNOWN)) {
+        LOG(INFO) << "Using FEC " << fec_str;
+        fec_type_ = fec::FECUtils::fecTypeFromString(fec_str);
+        CHECK(fec_type_ != fec::FECType::UNKNOWN);
+      }
+
+      if (fec_type_ == fec::FECType::UNKNOWN) {
+        return;
+      }
+
+      fec_encoder_ = fec::FECUtils::getEncoder(fec_type_, 1);
+      fec_encoder_->setFECCallback(std::forward<FECHandler>(fec_handler));
+      fec_encoder_->setBufferCallback(
+          std::forward<AllocatorHandler>(allocator_handler));
+    }
+  }
+
+ protected:
+  implementation::ProducerSocket *socket_;
+
+  // Thread pool responsible for IO operations (send data / receive interests)
+  std::vector<utils::EventThread> io_threads_;
+  interface::ProductionStatistics *stats_;
+  std::unique_ptr<fec::ProducerFEC> fec_encoder_;
+
+  // Callbacks
+  interface::ProducerInterestCallback *on_interest_input_;
+  interface::ProducerInterestCallback *on_interest_dropped_input_buffer_;
+  interface::ProducerInterestCallback *on_interest_inserted_input_buffer_;
+  interface::ProducerInterestCallback *on_interest_satisfied_output_buffer_;
+  interface::ProducerInterestCallback *on_interest_process_;
+
+  interface::ProducerContentObjectCallback *on_new_segment_;
+  interface::ProducerContentObjectCallback *on_content_object_to_sign_;
+  interface::ProducerContentObjectCallback *on_content_object_in_output_buffer_;
+  interface::ProducerContentObjectCallback *on_content_object_output_;
+  interface::ProducerContentObjectCallback
+      *on_content_object_evicted_from_output_buffer_;
+
+  interface::ProducerContentCallback *on_content_produced_;
+
+  interface::ProducerSocket::Callback *producer_callback_;
+
+  // Output buffer
+  utils::ContentStore output_buffer_;
+
+  // Signature and manifest
+  std::shared_ptr<auth::Signer> signer_;
+  uint32_t manifest_max_capacity_;
+
+  bool is_async_;
+  fec::FECType fec_type_;
+};
+
+}  // end namespace protocol
+}  // end namespace transport