control-plane/bfruntime.h

/*
 * Copyright 2021 Intel Corporation
 * SPDX-FileCopyrightText: 2021 Intel Corporation
 *
 * SPDX-License-Identifier: Apache-2.0
 */
#ifndef CONTROL_PLANE_BFRUNTIME_H_
#define CONTROL_PLANE_BFRUNTIME_H_
 
#include <algorithm>
#include <iosfwd>
#include <iterator>
#include <optional>
#include <ostream>
#include <string>
 
#include "control-plane/p4RuntimeSerializer.h"
#include "lib/error.h"
#include "lib/error_catalog.h"
#include "lib/json.h"
#pragma GCC diagnostic push
#pragma GCC diagnostic ignored "-Wunused-parameter"
#pragma GCC diagnostic ignored "-Wpedantic"
#include "p4/config/v1/p4info.pb.h"
#pragma GCC diagnostic pop
 
namespace p4configv1 = ::p4::config::v1;
 
namespace P4 {
struct P4RuntimeAPI;
}  // namespace P4
 
namespace P4 {
 
namespace BFRT {
 
using P4Id = uint32_t;
 
// Helpers
template <typename T, typename R>
static inline constexpr P4Id makeBFRuntimeId(T base, R prefix) {
    return static_cast<P4Id>((base & 0xffffff) | (prefix << 24));
}
 
static inline constexpr P4Id getIdPrefix(P4Id id) { return ((id >> 24) & 0xff); }
 
static inline Util::JsonObject *findJsonTable(Util::JsonArray *tablesJson, cstring tblName) {
    for (auto *t : *tablesJson) {
        auto *tblObj = t->to<Util::JsonObject>();
        auto tName = tblObj->getAs<Util::JsonValue>("name")->getString();
        if (tName == tblName) {
            return tblObj;
        }
    }
    return nullptr;
}
 
static inline Util::JsonObject *transformAnnotation(const cstring &annotation) {
    auto *annotationJson = new Util::JsonObject();
    // TODO(antonin): annotation string will need to be parsed so we can have it
    // in key/value format here.
    annotationJson->emplace("name", annotation.escapeJson());
    return annotationJson;
}
 
template <typename It>
static inline Util::JsonArray *transformAnnotations(const It &first, const It &last) {
    auto *annotations = new Util::JsonArray();
    for (auto it = first; it != last; it++) annotations->append(transformAnnotation(*it));
    return annotations;
}
 
static inline Util::JsonArray *transformAnnotations(const p4configv1::Preamble &pre) {
    return transformAnnotations(pre.annotations().begin(), pre.annotations().end());
}

template <typename T>
static inline bool isOfType(P4Id id, T prefix) {
    return getIdPrefix(id) == static_cast<P4Id>(prefix);
}
 
namespace Standard {
 
template <typename It>
static inline auto findP4InfoObject(const It &first, const It &last, P4Id objectId) -> const
    typename std::iterator_traits<It>::value_type * {
    using T = typename std::iterator_traits<It>::value_type;
    auto desiredObject = std::find_if(
        first, last, [&](const T &object) { return object.preamble().id() == objectId; });
    if (desiredObject == last) return nullptr;
    return &*desiredObject;
}
 
static inline const p4configv1::Table *findTable(const p4configv1::P4Info &p4info, P4Id tableId) {
    const auto &tables = p4info.tables();
    return findP4InfoObject(tables.begin(), tables.end(), tableId);
}
 
static inline const p4configv1::Action *findAction(const p4configv1::P4Info &p4info,
                                                   P4Id actionId) {
    const auto &actions = p4info.actions();
    return Standard::findP4InfoObject(actions.begin(), actions.end(), actionId);
}
 
static inline const p4configv1::ActionProfile *findActionProf(const p4configv1::P4Info &p4info,
                                                              P4Id actionProfId) {
    const auto &actionProfs = p4info.action_profiles();
    return findP4InfoObject(actionProfs.begin(), actionProfs.end(), actionProfId);
}
 
static inline const p4configv1::DirectCounter *findDirectCounter(const p4configv1::P4Info &p4info,
                                                                 P4Id counterId) {
    const auto &counters = p4info.direct_counters();
    return findP4InfoObject(counters.begin(), counters.end(), counterId);
}
 
static inline const p4configv1::DirectMeter *findDirectMeter(const p4configv1::P4Info &p4info,
                                                             P4Id meterId) {
    const auto &meters = p4info.direct_meters();
    return findP4InfoObject(meters.begin(), meters.end(), meterId);
}
 
}  // namespace Standard
 
static inline Util::JsonObject *makeType(cstring type) {
    auto *typeObj = new Util::JsonObject();
    typeObj->emplace("type", type);
    return typeObj;
}
 
template <typename T, typename std::enable_if<std::is_integral<T>::value, int>::type = 0>
static inline Util::JsonObject *makeType(cstring type, T defaultValue) {
    auto *typeObj = new Util::JsonObject();
    typeObj->emplace("type", type);
    typeObj->emplace("default_value", defaultValue);
    return typeObj;
}
 
static inline Util::JsonObject *makeTypeBool(std::optional<bool> defaultValue = std::nullopt) {
    auto *typeObj = new Util::JsonObject();
    typeObj->emplace("type", "bool");
    if (defaultValue != std::nullopt) typeObj->emplace("default_value", *defaultValue);
    return typeObj;
}
 
static inline Util::JsonObject *makeTypeBytes(int width,
                                              std::optional<int64_t> defaultValue = std::nullopt) {
    auto *typeObj = new Util::JsonObject();
    typeObj->emplace("type", "bytes");
    typeObj->emplace("width", width);
    if (defaultValue != std::nullopt) typeObj->emplace("default_value", *defaultValue);
    return typeObj;
}
 
static inline Util::JsonObject *makeTypeEnum(const std::vector<cstring> &choices,
                                             std::optional<cstring> defaultValue = std::nullopt) {
    auto *typeObj = new Util::JsonObject();
    typeObj->emplace("type", "string");
    auto *choicesArray = new Util::JsonArray();
    for (auto choice : choices) choicesArray->append(choice);
    typeObj->emplace("choices", choicesArray);
    if (defaultValue != std::nullopt) typeObj->emplace("default_value", *defaultValue);
    return typeObj;
}
 
static inline void addSingleton(Util::JsonArray *dataJson, Util::JsonObject *dataField,
                                bool mandatory, bool readOnly) {
    auto *singletonJson = new Util::JsonObject();
    singletonJson->emplace("mandatory", mandatory);
    singletonJson->emplace("read_only", readOnly);
    singletonJson->emplace("singleton", dataField);
    dataJson->append(singletonJson);
}
 
static inline void addOneOf(Util::JsonArray *dataJson, Util::JsonArray *choicesJson, bool mandatory,
                            bool readOnly) {
    auto *oneOfJson = new Util::JsonObject();
    oneOfJson->emplace("mandatory", mandatory);
    oneOfJson->emplace("read_only", readOnly);
    oneOfJson->emplace("oneof", choicesJson);
    dataJson->append(oneOfJson);
}
 
// FIXME: Switch to std::string_view
static inline std::optional<cstring> transformMatchType(
    p4configv1::MatchField_MatchType matchType) {
    switch (matchType) {
        case p4configv1::MatchField_MatchType_UNSPECIFIED:
            return std::nullopt;
        case p4configv1::MatchField_MatchType_EXACT:
            return "Exact"_cs;
        case p4configv1::MatchField_MatchType_LPM:
            return "LPM"_cs;
        case p4configv1::MatchField_MatchType_TERNARY:
            return "Ternary"_cs;
        case p4configv1::MatchField_MatchType_RANGE:
            return "Range"_cs;
        case p4configv1::MatchField_MatchType_OPTIONAL:
            return "Optional"_cs;
        default:
            return std::nullopt;
    }
}
 
// FIXME: Switch to std::string_view
static inline std::optional<cstring> transformOtherMatchType(std::string matchType) {
    if (matchType == "atcam_partition_index") return "ATCAM"_cs;
    if (matchType == "dleft_hash") return "DLEFT_HASH"_cs;
 
    return std::nullopt;
}
 
template <typename It>
static std::vector<P4Id> collectTableIds(const p4configv1::P4Info &p4info, const It &first,
                                         const It &last) {
    std::vector<P4Id> tableIds;
    for (auto it = first; it != last; it++) {
        auto *table = Standard::findTable(p4info, *it);
        if (table == nullptr) {
            ::P4::error(ErrorType::ERR_INVALID, "Invalid table id '%1%'", *it);
            continue;
        }
        tableIds.push_back(*it);
    }
    return tableIds;
}

class TypeSpecParser {
 public:
    struct Field {
        cstring name;
        P4Id id;
        Util::JsonObject *type;
    };
 
    using Fields = std::vector<Field>;
    using iterator = Fields::iterator;
    using const_iterator = Fields::const_iterator;
 
    static TypeSpecParser make(const p4configv1::P4Info &p4info,
                               const p4configv1::P4DataTypeSpec &typeSpec, cstring instanceType,
                               cstring instanceName,
                               const std::vector<cstring> *fieldNames = nullptr,
                               cstring prefix = cstring::empty, cstring suffix = cstring::empty,
                               P4Id idOffset = 1);
 
    iterator begin() { return fields.begin(); }
    const_iterator cbegin() { return fields.cbegin(); }
    iterator end() { return fields.end(); }
    const_iterator cend() { return fields.cend(); }
    size_t size() { return fields.size(); }
 
 private:
    explicit TypeSpecParser(Fields &&fields) : fields(std::move(fields)) {}
 
    Fields fields;
};
 
class BFRuntimeGenerator {
 public:
    virtual ~BFRuntimeGenerator() = default;
    explicit BFRuntimeGenerator(const p4configv1::P4Info &p4info) : p4info(p4info) {}

    virtual const Util::JsonObject *genSchema() const;

    void serializeBFRuntimeSchema(std::ostream *destination);
 
 protected:
    // To avoid potential clashes with P4 names, we prefix the names of "fixed"
    // data field with a '$'. For example, for TD_DATA_ACTION_MEMBER_ID, we
    // use the name $ACTION_MEMBER_ID.
    enum TDDataFieldIds : P4Id {
        // ids for fixed data fields must not collide with the auto-generated
        // ids for P4 fields (e.g. match key fields).
        TD_DATA_START = (1 << 16),
 
        TD_DATA_MATCH_PRIORITY,
 
        TD_DATA_ACTION,
        TD_DATA_ACTION_MEMBER_ID,
        TD_DATA_SELECTOR_GROUP_ID,
        TD_DATA_ACTION_MEMBER_STATUS,
        TD_DATA_MAX_GROUP_SIZE,
 
        TD_DATA_ENTRY_TTL,
        TD_DATA_ENTRY_HIT_STATE,
 
        TD_DATA_METER_SPEC_CIR_KBPS,
        TD_DATA_METER_SPEC_PIR_KBPS,
        TD_DATA_METER_SPEC_CBS_KBITS,
        TD_DATA_METER_SPEC_PBS_KBITS,
 
        TD_DATA_METER_SPEC_CIR_PPS,
        TD_DATA_METER_SPEC_PIR_PPS,
        TD_DATA_METER_SPEC_CBS_PKTS,
        TD_DATA_METER_SPEC_PBS_PKTS,
 
        TD_DATA_COUNTER_SPEC_BYTES,
        TD_DATA_COUNTER_SPEC_PKTS,
 
        TD_DATA_METER_INDEX,
        TD_DATA_COUNTER_INDEX,
        TD_DATA_REGISTER_INDEX,
 
        TD_DATA_END,
    };

    struct Counter {
        enum Unit { UNSPECIFIED = 0, BYTES = 1, PACKETS = 2, BOTH = 3 };
        std::string name;
        P4Id id;
        int64_t size;
        Unit unit;
        Util::JsonArray *annotations;
 
        static std::optional<Counter> from(const p4configv1::Counter &counterInstance);
        static std::optional<Counter> fromDirect(const p4configv1::DirectCounter &counterInstance);
    };

    struct Meter {
        enum Unit { UNSPECIFIED = 0, BYTES = 1, PACKETS = 2 };
        enum Type { COLOR_UNAWARE = 0, COLOR_AWARE = 1 };
        std::string name;
        P4Id id;
        int64_t size;
        Unit unit;
        Util::JsonArray *annotations;
 
        static std::optional<Meter> from(const p4configv1::Meter &meterInstance);
        static std::optional<Meter> fromDirect(const p4configv1::DirectMeter &meterInstance);
    };

    struct ActionProf {
        std::string name;
        P4Id id;
        int64_t size;
        std::vector<P4Id> tableIds;
        Util::JsonArray *annotations;
        static P4Id makeActProfId(P4Id implementationId);
        static std::optional<ActionProf> from(const p4configv1::P4Info &p4info,
                                              const p4configv1::ActionProfile &actionProfile);
    };

    struct Digest {
        std::string name;
        P4Id id;
        p4configv1::P4DataTypeSpec typeSpec;
        Util::JsonArray *annotations;
 
        static std::optional<Digest> from(const p4configv1::Digest &digest);
    };

    struct Register {
        std::string name;
        std::string dataFieldName;
        P4Id id;
        int64_t size;
        p4configv1::P4DataTypeSpec typeSpec;
        Util::JsonArray *annotations;
 
        static std::optional<Register> from(const p4configv1::Register &regInstance);
    };
 
    void addMatchTables(Util::JsonArray *tablesJson) const;
    virtual bool addMatchTypePriority(std::optional<cstring> &matchType) const;
    virtual void addConstTableAttr(Util::JsonArray *attrJson) const;
    virtual void addActionProfs(Util::JsonArray *tablesJson) const;
    virtual bool addActionProfIds(const p4configv1::Table &table,
                                  Util::JsonObject *tableJson) const;
    void addCounters(Util::JsonArray *tablesJson) const;
    void addMeters(Util::JsonArray *tablesJson) const;
    void addRegisters(Util::JsonArray *tablesJson) const;
 
    void addCounterCommon(Util::JsonArray *tablesJson, const Counter &counter) const;
    void addMeterCommon(Util::JsonArray *tablesJson, const Meter &meter) const;
    void addRegisterCommon(Util::JsonArray *tablesJson, const Register &reg) const;
    void addActionProfCommon(Util::JsonArray *tablesJson, const ActionProf &actionProf) const;
    void addLearnFilters(Util::JsonArray *learnFiltersJson) const;
    void addLearnFilterCommon(Util::JsonArray *learnFiltersJson, const Digest &digest) const;
    virtual void addDirectResources(const p4configv1::Table &table, Util::JsonArray *dataJson,
                                    Util::JsonArray *operationsJson,
                                    Util::JsonArray *attributesJson,
                                    P4Id maxActionParamId = 0) const;
 
    virtual std::optional<bool> actProfHasSelector(P4Id actProfId) const;
    Util::JsonArray *makeActionSpecs(const p4configv1::Table &table,
                                     P4Id *maxActionParamId = nullptr) const;
    virtual std::optional<Counter> getDirectCounter(P4Id counterId) const;
    virtual std::optional<Meter> getDirectMeter(P4Id meterId) const;

    void transformTypeSpecToDataFields(Util::JsonArray *fieldsJson,
                                       const p4configv1::P4DataTypeSpec &typeSpec,
                                       cstring instanceType, cstring instanceName,
                                       const std::vector<cstring> *fieldNames = nullptr,
                                       cstring prefix = cstring::empty,
                                       cstring suffix = cstring::empty, P4Id idOffset = 1) const;
 
    static void addMeterDataFields(Util::JsonArray *dataJson, const Meter &meter);
    static Util::JsonObject *makeCommonDataField(P4Id id, cstring name, Util::JsonObject *type,
                                                 bool repeated,
                                                 Util::JsonArray *annotations = nullptr);
 
    static Util::JsonObject *makeContainerDataField(P4Id id, cstring name, Util::JsonArray *items,
                                                    bool repeated,
                                                    Util::JsonArray *annotations = nullptr);
 
    static void addActionDataField(Util::JsonArray *dataJson, P4Id id, const std::string &name,
                                   bool mandatory, bool read_only, Util::JsonObject *type,
                                   Util::JsonArray *annotations = nullptr);
 
    static void addKeyField(Util::JsonArray *dataJson, P4Id id, cstring name, bool mandatory,
                            cstring matchType, Util::JsonObject *type,
                            Util::JsonArray *annotations = nullptr);
 
    static void addCounterDataFields(Util::JsonArray *dataJson, const Counter &counter);
 
    static Util::JsonObject *initTableJson(const std::string &name, P4Id id, cstring tableType,
                                           int64_t size, Util::JsonArray *annotations = nullptr);
 
    static void addToDependsOn(Util::JsonObject *tableJson, P4Id id);

    void addRegisterDataFields(Util::JsonArray *dataJson, const Register &register_,
                               P4Id idOffset = 1) const;
 
    const p4configv1::P4Info &p4info;
};
 
}  // namespace BFRT
 
}  // namespace P4
 
#endif /* CONTROL_PLANE_BFRUNTIME_H_ */