1 files changed, 155 insertions, 0 deletions
diff --git a/libtransport/src/protocols/fec_utils.h b/libtransport/src/protocols/fec_utils.h
new file mode 100644
index 000000000..d70ff1c09
--- /dev/null
+++ b/libtransport/src/protocols/fec_utils.h
@@ -0,0 +1,155 @@
+/*
+ * 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/config.h>
+#include <hicn/transport/core/content_object.h>
+#include <hicn/transport/errors/not_implemented_exception.h>
+#include <protocols/fec/rs.h>
+
+#if ENABLE_RELY
+#include <protocols/fec/rely.h>
+#endif
+
+#include <functional>
+
+namespace transport {
+namespace protocol {
+
+namespace fec {
+
+#if ENABLE_RELY
+#define foreach_fec_type foreach_rs_fec_type foreach_rely_fec_type
+#else
+#define foreach_fec_type foreach_rs_fec_type
+#endif
+
+#define ENUM_FROM_MACRO(name, k, n) name##_K##k##_N##n
+#define ENUM_FROM_MACRO_STR(name, k, n) #name "_K" #k "_N" #n
+
+enum class FECType : uint8_t {
+#define _(name, k, n) ENUM_FROM_MACRO(name, k, n),
+  foreach_fec_type
+#undef _
+      UNKNOWN
+};
+
+#define ENUM_FROM_MACRO2(name, k, n) FECType::ENUM_FROM_MACRO(name, k, n)
+
+class FECUtils {
+ public:
+  static FECType fecTypeFromString(const char *fec_type) {
+#define _(name, k, n)                                            \
+  do {                                                           \
+    if (strncmp(fec_type, ENUM_FROM_MACRO_STR(name, k, n),       \
+                strlen(ENUM_FROM_MACRO_STR(name, k, n))) == 0) { \
+      return ENUM_FROM_MACRO2(name, k, n);                       \
+    }                                                            \
+  } while (0);
+    foreach_fec_type
+#undef _
+
+        return FECType::UNKNOWN;
+  }
+
+  static bool isFec(FECType fec_type, uint32_t index, uint32_t seq_offset = 0) {
+    switch (fec_type) {
+#define _(name, k, n)                \
+  case ENUM_FROM_MACRO2(name, k, n): \
+    return FecInfo<Code<k, n>>::isFec(index - (seq_offset % n));
+
+      foreach_fec_type
+#undef _
+          default : return false;
+    }
+  }
+
+  static uint32_t nextSource(FECType fec_type, uint32_t index,
+                             uint32_t seq_offset = 0) {
+    switch (fec_type) {
+#define _(name, k, n)                \
+  case ENUM_FROM_MACRO2(name, k, n): \
+    return FecInfo<Code<k, n>>::nextSource(index) + (seq_offset % n);
+
+      foreach_fec_type
+#undef _
+          default : throw std::runtime_error("Unknown fec type");
+    }
+  }
+
+  static uint32_t getSourceSymbols(FECType fec_type) {
+    switch (fec_type) {
+#define _(name, k, n)                \
+  case ENUM_FROM_MACRO2(name, k, n): \
+    return k;
+      foreach_fec_type
+#undef _
+          default : throw std::runtime_error("Unknown fec type");
+    }
+  }
+
+  static uint32_t getBlockSymbols(FECType fec_type) {
+    switch (fec_type) {
+#define _(name, k, n)                \
+  case ENUM_FROM_MACRO2(name, k, n): \
+    return n;
+      foreach_fec_type
+#undef _
+          default : throw std::runtime_error("Unknown fec type");
+    }
+  }
+
+  static std::unique_ptr<ProducerFEC> getEncoder(FECType fec_type,
+                                                 uint32_t seq_offset = 0) {
+    return factoryEncoder(fec_type, seq_offset);
+  }
+
+  static std::unique_ptr<ConsumerFEC> getDecoder(FECType fec_type,
+                                                 uint32_t seq_offset = 0) {
+    return factoryDencoder(fec_type, seq_offset);
+  }
+
+ private:
+  static std::unique_ptr<ProducerFEC> factoryEncoder(FECType fec_type,
+                                                     uint32_t seq_offset) {
+    switch (fec_type) {
+#define _(name, k, n)                \
+  case ENUM_FROM_MACRO2(name, k, n): \
+    return std::make_unique<name##Encoder>(k, n, seq_offset);
+
+      foreach_fec_type
+#undef _
+          default : throw std::runtime_error("Unknown fec type");
+    }
+  }
+
+  static std::unique_ptr<ConsumerFEC> factoryDencoder(FECType fec_type,
+                                                      uint32_t seq_offset) {
+    switch (fec_type) {
+#define _(name, k, n)                \
+  case ENUM_FROM_MACRO2(name, k, n): \
+    return std::make_unique<name##Decoder>(k, n, seq_offset);
+
+      foreach_fec_type
+#undef _
+          default : throw std::runtime_error("Unknown fec type");
+    }
+  }
+};  // namespace fec
+
+}  // namespace fec
+}  // namespace protocol
+}  // namespace transport