dpdk/control-plane/bfruntime_arch_handler.h

/*
 * Copyright 2021 Intel Corporation
 * SPDX-FileCopyrightText: 2021 Intel Corporation
 *
 * SPDX-License-Identifier: Apache-2.0
 */
#ifndef BACKENDS_DPDK_CONTROL_PLANE_BFRUNTIME_ARCH_HANDLER_H_
#define BACKENDS_DPDK_CONTROL_PLANE_BFRUNTIME_ARCH_HANDLER_H_
 
#include <iostream>
#include <optional>
#include <set>
#include <unordered_map>
#include <vector>
 
#pragma GCC diagnostic push
#pragma GCC diagnostic ignored "-Wunused-parameter"
#pragma GCC diagnostic ignored "-Wpedantic"
#include "backends/dpdk/p4/config/p4info.pb.h"
#pragma GCC diagnostic pop
 
#include "control-plane/bfruntime.h"
#include "control-plane/p4RuntimeArchHandler.h"
#include "control-plane/p4RuntimeArchStandard.h"
#include "control-plane/p4RuntimeSerializer.h"
#include "control-plane/typeSpecConverter.h"
#include "frontends/common/resolveReferences/referenceMap.h"
#include "frontends/p4/externInstance.h"
#include "frontends/p4/methodInstance.h"
#include "frontends/p4/typeMap.h"
#include "midend/eliminateTypedefs.h"
 
using ::P4::ReferenceMap;
using ::P4::TypeMap;
using ::P4::ControlPlaneAPI::Helpers::getExternInstanceFromProperty;
using ::P4::ControlPlaneAPI::Helpers::isExternPropertyConstructedInPlace;
 
namespace p4configv1 = ::p4::config::v1;
 
namespace P4 {

namespace ControlPlaneAPI {

namespace Standard {
 
cstring prefix(cstring p, cstring str) { return p.isNullOrEmpty() ? str : p + "." + str; }

class SymbolTypeDPDK final : public SymbolType {
 public:
    SymbolTypeDPDK() = delete;
 
    static P4RuntimeSymbolType P4RT_ACTION_SELECTOR() {
        return P4RuntimeSymbolType::make(dpdk::P4Ids::ACTION_SELECTOR);
    }
};

struct ActionSelector {
    const cstring name;  // The fully qualified external name of this action selector.
    const int64_t size;  // TODO(hanw): size does not make sense with new ActionSelector P4 extern
    const int64_t maxGroupSize;
    const int64_t numGroups;
    const IR::IAnnotated *annotations;  // If non-null, any annotations applied to this action
                                        // profile declaration.
 
    static constexpr int64_t defaultMaxGroupSize = 120;
 
    p4rt_id_t getId(const P4RuntimeSymbolTableIface &symbols) const {
        return symbols.getId(SymbolTypeDPDK::P4RT_ACTION_SELECTOR(), name + "_sel");
    }
};
 
template <Arch arch>
class BFRuntimeArchHandler : public P4RuntimeArchHandlerCommon<arch> {
 protected:
    std::unordered_map<const IR::Block *, cstring> blockNamePrefixMap;
 
 public:
    template <typename Func>
    void forAllPipeBlocks(const IR::ToplevelBlock *evaluatedProgram, Func function) {
        auto main = evaluatedProgram->getMain();
        if (!main)
            ::P4::error(ErrorType::ERR_NOT_FOUND, "Program does not contain a `main` module");
        auto cparams = main->getConstructorParameters();
        int index = -1;
        for (auto param : main->constantValue) {
            index++;
            if (!param.second) continue;
            auto pipe = param.second;
            if (!pipe->is<IR::PackageBlock>()) continue;
            auto idxParam = cparams->getParameter(index);
            auto pipeName = idxParam->name;
            function(pipeName, pipe->to<IR::PackageBlock>());
        }
    }
 
    using ArchCounterExtern = CounterExtern<arch>;
    using CounterTraits = Helpers::CounterlikeTraits<ArchCounterExtern>;
    using ArchMeterExtern = MeterExtern<arch>;
    using MeterTraits = Helpers::CounterlikeTraits<ArchMeterExtern>;
 
    using Counter = p4configv1::Counter;
    using Meter = p4configv1::Meter;
    using CounterSpec = p4configv1::CounterSpec;
    using MeterSpec = p4configv1::MeterSpec;
 
    BFRuntimeArchHandler(ReferenceMap *refMap, TypeMap *typeMap,
                         const IR::ToplevelBlock *evaluatedProgram)
        : P4RuntimeArchHandlerCommon<arch>(refMap, typeMap, evaluatedProgram) {
        // Create a map of all blocks to their pipe names. This map will
        // be used during collect and post processing to prefix
        // table/extern instances wherever applicable with a fully qualified
        // name. This distinction is necessary when the driver looks up
        // context.json across multiple pipes for the table name
        forAllPipeBlocks(evaluatedProgram, [&](cstring pipeName, const IR::PackageBlock *pkg) {
            Helpers::forAllEvaluatedBlocks(pkg, [&](const IR::Block *block) {
                auto decl = pkg->node->to<IR::Declaration_Instance>();
                cstring blockNamePrefix = pipeName;
                if (decl) blockNamePrefix = decl->controlPlaneName();
                blockNamePrefixMap[block] = blockNamePrefix;
            });
        });
    }
 
    cstring getBlockNamePrefix(const IR::Block *blk) {
        if (blockNamePrefixMap.count(blk) > 0) return blockNamePrefixMap[blk];
        return "pipe"_cs;
    }
 
    static p4configv1::Extern *getP4InfoExtern(P4RuntimeSymbolType typeId, cstring typeName,
                                               p4configv1::P4Info *p4info) {
        for (auto &externType : *p4info->mutable_externs()) {
            if (externType.extern_type_id() == static_cast<p4rt_id_t>(typeId)) return &externType;
        }
        auto *externType = p4info->add_externs();
        externType->set_extern_type_id(static_cast<p4rt_id_t>(typeId));
        externType->set_extern_type_name(typeName);
        return externType;
    }
 
    static void addP4InfoExternInstance(const P4RuntimeSymbolTableIface &symbols,
                                        P4RuntimeSymbolType typeId, cstring typeName, cstring name,
                                        const IR::IAnnotated *annotations,
                                        const ::google::protobuf::Message &message,
                                        p4configv1::P4Info *p4info,
                                        cstring pipeName = cstring::empty) {
        auto *externType = getP4InfoExtern(typeId, typeName, p4info);
        auto *externInstance = externType->add_instances();
        auto *pre = externInstance->mutable_preamble();
        pre->set_id(symbols.getId(typeId, name));
        pre->set_name(prefix(pipeName, name));
        pre->set_alias(symbols.getAlias(name));
        Helpers::addAnnotations(pre, annotations);
        Helpers::addDocumentation(pre, annotations);
        externInstance->mutable_info()->PackFrom(message);
    }
 
    std::optional<ActionSelector> getActionSelector(const IR::ExternBlock *instance) {
        auto actionSelDecl = instance->node->to<IR::IDeclaration>();
        // to be deleted, used to support deprecated ActionSelector constructor.
        auto size = instance->getParameterValue("size"_cs);
        BUG_CHECK(size->is<IR::Constant>(), "Non-constant size");
        return ActionSelector{actionSelDecl->controlPlaneName(), size->to<IR::Constant>()->asInt(),
                              ActionSelector::defaultMaxGroupSize,
                              size->to<IR::Constant>()->asInt(),
                              actionSelDecl->to<IR::IAnnotated>()};
    }
 
    void addActionSelector(const P4RuntimeSymbolTableIface &symbols, p4configv1::P4Info *p4Info,
                           const ActionSelector &actionSelector,
                           cstring pipeName = cstring::empty) {
        ::dpdk::ActionSelector selector;
        selector.set_max_group_size(actionSelector.maxGroupSize);
        selector.set_num_groups(actionSelector.numGroups);
        p4configv1::ActionProfile profile;
        profile.set_size(actionSelector.size);
        auto tablesIt = this->actionProfilesRefs.find(actionSelector.name);
        if (tablesIt != this->actionProfilesRefs.end()) {
            for (const auto &table : tablesIt->second) {
                profile.add_table_ids(symbols.getId(P4RuntimeSymbolType::P4RT_TABLE(), table));
                selector.add_table_ids(symbols.getId(P4RuntimeSymbolType::P4RT_TABLE(), table));
            }
        }
        // We use the ActionSelector name for the action profile, and add a "_sel" suffix for
        // the action selector.
        cstring profileName = actionSelector.name;
        selector.set_action_profile_id(
            symbols.getId(SymbolType::P4RT_ACTION_PROFILE(), profileName));
        cstring selectorName = profileName + "_sel";
        addP4InfoExternInstance(symbols, SymbolTypeDPDK::P4RT_ACTION_SELECTOR(),
                                "ActionSelector"_cs, selectorName, actionSelector.annotations,
                                selector, p4Info, pipeName);
    }
 
    void collectExternInstance(P4RuntimeSymbolTableIface *symbols,
                               const IR::ExternBlock *externBlock) override {
        P4RuntimeArchHandlerCommon<arch>::collectExternInstance(symbols, externBlock);
 
        auto decl = externBlock->node->to<IR::IDeclaration>();
        if (decl == nullptr) return;
        if (externBlock->type->name == "Digest") {
            symbols->add(SymbolType::P4RT_DIGEST(), decl);
        } else if (externBlock->type->name == ActionSelectorTraits<arch>::typeName()) {
            auto selName = decl->controlPlaneName() + "_sel";
            auto profName = decl->controlPlaneName();
            symbols->add(SymbolTypeDPDK::P4RT_ACTION_SELECTOR(), selName);
            symbols->add(SymbolType::P4RT_ACTION_PROFILE(), profName);
        }
    }
 
    void addTableProperties(const P4RuntimeSymbolTableIface &symbols, p4configv1::P4Info *p4info,
                            p4configv1::Table *table, const IR::TableBlock *tableBlock) override {
        P4RuntimeArchHandlerCommon<arch>::addTableProperties(symbols, p4info, table, tableBlock);
 
        auto tableDeclaration = tableBlock->container;
        bool supportsTimeout = getSupportsTimeout(tableDeclaration);
        if (supportsTimeout) {
            table->set_idle_timeout_behavior(p4configv1::Table::NOTIFY_CONTROL);
        } else {
            table->set_idle_timeout_behavior(p4configv1::Table::NO_TIMEOUT);
        }
 
        // add pipe name prefix to the table names
        auto pipeName = getBlockNamePrefix(tableBlock);
        auto *pre = table->mutable_preamble();
        if (pre->name() == tableDeclaration->controlPlaneName())
            pre->set_name(prefix(pipeName, pre->name()));
    }
 
    void addExternInstance(const P4RuntimeSymbolTableIface &symbols, p4configv1::P4Info *p4info,
                           const IR::ExternBlock *externBlock) override {
        P4RuntimeArchHandlerCommon<arch>::addExternInstance(symbols, p4info, externBlock);
 
        auto decl = externBlock->node->to<IR::Declaration_Instance>();
        if (decl == nullptr) return;
 
        // DPDK control plane software requires pipe name to be prefixed to the
        // table and extern names
        cstring pipeName = getBlockNamePrefix(externBlock);
 
        auto p4RtTypeInfo = p4info->mutable_type_info();
        if (externBlock->type->name == "Digest") {
            auto digest = getDigest(decl, p4RtTypeInfo);
            if (digest) this->addDigest(symbols, p4info, *digest);
        } else if (externBlock->type->name == "ActionSelector") {
            auto actionSelector = getActionSelector(externBlock);
            if (actionSelector) addActionSelector(symbols, p4info, *actionSelector, pipeName);
            for (auto &extType : *p4info->mutable_action_profiles()) {
                auto *pre = extType.mutable_preamble();
                if (pre->name() == decl->controlPlaneName()) {
                    pre->set_name(prefix(pipeName, pre->name()));
                    break;
                }
            }
        } else if (externBlock->type->name == "ActionProfile") {
            for (auto &extType : *p4info->mutable_action_profiles()) {
                auto *pre = extType.mutable_preamble();
                if (pre->name() == decl->controlPlaneName()) {
                    pre->set_name(prefix(pipeName, pre->name()));
                    break;
                }
            }
        } else if (externBlock->type->name == "Meter") {
            for (auto &extType : *p4info->mutable_meters()) {
                auto *pre = extType.mutable_preamble();
                if (pre->name() == decl->controlPlaneName()) {
                    pre->set_name(prefix(pipeName, pre->name()));
                    break;
                }
            }
        } else if (externBlock->type->name == "Counter") {
            for (auto &extType : *p4info->mutable_counters()) {
                auto *pre = extType.mutable_preamble();
                if (pre->name() == decl->controlPlaneName()) {
                    pre->set_name(prefix(pipeName, pre->name()));
                    break;
                }
            }
        } else if (externBlock->type->name == "Register") {
            for (auto &extType : *p4info->mutable_registers()) {
                auto *pre = extType.mutable_preamble();
                if (pre->name() == decl->controlPlaneName()) {
                    pre->set_name(prefix(pipeName, pre->name()));
                    break;
                }
            }
        }
    }

    std::optional<Digest> getDigest(const IR::Declaration_Instance *decl,
                                    p4configv1::P4TypeInfo *p4RtTypeInfo) {
        BUG_CHECK(decl->type->is<IR::Type_Specialized>(), "%1%: expected Type_Specialized",
                  decl->type);
        auto type = decl->type->to<IR::Type_Specialized>();
        BUG_CHECK(type->arguments->size() == 1, "%1%: expected one type argument", decl);
        auto typeArg = type->arguments->at(0);
        auto typeSpec =
            TypeSpecConverter::convert(this->refMap, this->typeMap, typeArg, p4RtTypeInfo);
        BUG_CHECK(typeSpec != nullptr,
                  "P4 type %1% could not be converted to P4Info P4DataTypeSpec");
 
        return Digest{decl->controlPlaneName(), typeSpec, decl->to<IR::IAnnotated>()};
    }

    static bool getSupportsTimeout(const IR::P4Table *table) {
        auto timeout = table->properties->getProperty("psa_idle_timeout");
 
        if (timeout == nullptr) return false;
 
        if (auto exprValue = timeout->value->to<IR::ExpressionValue>()) {
            if (auto expr = exprValue->expression) {
                if (auto member = expr->to<IR::Member>()) {
                    if (member->member == "NOTIFY_CONTROL") {
                        return true;
                    } else if (member->member == "NO_TIMEOUT") {
                        return false;
                    }
                } else if (expr->is<IR::PathExpression>()) {
                    ::P4::error(ErrorType::ERR_UNEXPECTED,
                                "Unresolved value %1% for psa_idle_timeout "
                                "property on table %2%. Must be a constant and one of "
                                "{ NOTIFY_CONTROL, NO_TIMEOUT }",
                                timeout, table);
                    return false;
                }
            }
        }
 
        ::P4::error(ErrorType::ERR_UNEXPECTED,
                    "Unexpected value %1% for psa_idle_timeout "
                    "property on table %2%. Supported values are "
                    "{ NOTIFY_CONTROL, NO_TIMEOUT }",
                    timeout, table);
        return false;
    }
};
 
class BFRuntimeArchHandlerPSA final : public BFRuntimeArchHandler<Arch::PSA> {
 public:
    BFRuntimeArchHandlerPSA(ReferenceMap *refMap, TypeMap *typeMap,
                            const IR::ToplevelBlock *evaluatedProgram)
        : BFRuntimeArchHandler(refMap, typeMap, evaluatedProgram) {}
};
 
class BFRuntimeArchHandlerPNA final : public BFRuntimeArchHandler<Arch::PNA> {
 public:
    BFRuntimeArchHandlerPNA(ReferenceMap *refMap, TypeMap *typeMap,
                            const IR::ToplevelBlock *evaluatedProgram)
        : BFRuntimeArchHandler(refMap, typeMap, evaluatedProgram) {}
};

struct PSAArchHandlerBuilderForDPDK : public P4::ControlPlaneAPI::P4RuntimeArchHandlerBuilderIface {
    P4::ControlPlaneAPI::P4RuntimeArchHandlerIface *operator()(
        ReferenceMap *refMap, TypeMap *typeMap,
        const IR::ToplevelBlock *evaluatedProgram) const override {
        return new P4::ControlPlaneAPI::Standard::BFRuntimeArchHandlerPSA(refMap, typeMap,
                                                                          evaluatedProgram);
    }
};

struct PNAArchHandlerBuilderForDPDK : public P4::ControlPlaneAPI::P4RuntimeArchHandlerBuilderIface {
    P4::ControlPlaneAPI::P4RuntimeArchHandlerIface *operator()(
        ReferenceMap *refMap, TypeMap *typeMap,
        const IR::ToplevelBlock *evaluatedProgram) const override {
        return new P4::ControlPlaneAPI::Standard::BFRuntimeArchHandlerPNA(refMap, typeMap,
                                                                          evaluatedProgram);
    }
};
 
}  // namespace Standard
 
}  // namespace ControlPlaneAPI
 /* end group control_plane */
}  // namespace P4
 
#endif /* BACKENDS_DPDK_CONTROL_PLANE_BFRUNTIME_ARCH_HANDLER_H_ */