dpdkCheckExternInvocation.h

/*
 * Copyright 2021 Intel Corp.
 * SPDX-FileCopyrightText: 2021 Intel Corp.
 *
 * SPDX-License-Identifier: Apache-2.0
 */
 
#ifndef BACKENDS_DPDK_DPDKCHECKEXTERNINVOCATION_H_
#define BACKENDS_DPDK_DPDKCHECKEXTERNINVOCATION_H_
 
#include <array>
 
#include "dpdkProgramStructure.h"
#include "frontends/p4/methodInstance.h"
#include "ir/ir.h"
#include "ir/visitor.h"
#include "lib/cstring.h"
#include "midend/checkExternInvocationCommon.h"
 
namespace P4 {
class TypeMap;
}  // namespace P4
 
namespace P4::DPDK {
 
using namespace P4::literals;

 
class CheckPNAExternInvocation : public P4::CheckExternInvocationCommon {
    DpdkProgramStructure *structure;
 
    enum block_t {
        MAIN_PARSER = 0,
        PRE_CONTROL,
        MAIN_CONTROL,
        MAIN_DEPARSER,
        BLOCK_COUNT,
    };
 
    void initPipeConstraints() override {
        bitvec validInMainControl;
        validInMainControl.setbit(MAIN_CONTROL);
        setPipeConstraints("send_to_port"_cs, validInMainControl);
        setPipeConstraints("mirror_packet"_cs, validInMainControl);
        // Add new constraints here
    }
 
    cstring getBlockName(int bit) override {
        static const std::array<cstring, BLOCK_COUNT> lookup = {
            "main parser"_cs, "pre control"_cs, "main control"_cs, "main deparser"_cs};
        return lookup[bit % BLOCK_COUNT];
    }
 
    const IR::P4Parser *getParser(const cstring parserName) {
        if (auto p = findContext<IR::P4Parser>()) {
            if (structure->parsers.count(parserName) != 0 &&
                structure->parsers.at(parserName)->name == p->name) {
                return p;
            }
        }
        return nullptr;
    }
 
    const IR::P4Control *getControl(const cstring controlName) {
        if (auto c = findContext<IR::P4Control>()) {
            if (structure->pipelines.count(controlName) != 0 &&
                structure->pipelines.at(controlName)->name == c->name) {
                return c;
            }
        }
        return nullptr;
    }
 
    const IR::P4Control *getDeparser(const cstring deparserName) {
        if (auto d = findContext<IR::P4Control>()) {
            if (structure->deparsers.count(deparserName) != 0 &&
                structure->deparsers.at(deparserName)->name == d->name) {
                return d;
            }
        }
        return nullptr;
    }
 
    void checkBlock(const IR::MethodCallExpression *expr, const cstring externType,
                    const cstring externName) {
        bitvec pos;
 
        LOG4("externType: " << externType << ", externName: " << externName);
 
        if (pipeConstraints.count(externType)) {
            if (auto block = getParser("MainParserT"_cs)) {
                LOG4("MainParser: " << (void *)block << " " << dbp(block));
                pos.setbit(MAIN_PARSER);
                checkPipeConstraints(externType, pos, expr, externName, block->name);
            } else if (auto block = getControl("PreControlT"_cs)) {
                LOG4("PreControl: " << (void *)block << " " << dbp(block));
                pos.setbit(PRE_CONTROL);
                checkPipeConstraints(externType, pos, expr, externName, block->name);
            } else if (auto block = getControl("MainControlT"_cs)) {
                LOG4("MainControl: " << (void *)block << " " << dbp(block));
                pos.setbit(MAIN_CONTROL);
                checkPipeConstraints(externType, pos, expr, externName, block->name);
            } else if (auto block = getDeparser("MainDeparserT"_cs)) {
                LOG4("MainDeparser: " << (void *)block << " " << dbp(block));
                pos.setbit(MAIN_DEPARSER);
                checkPipeConstraints(externType, pos, expr, externName, block->name);
            } else {
                LOG4("Other");
            }
        } else {
            LOG1("Pipe constraints not defined for: " << externType);
        }
    }
 
    void checkExtern(const P4::ExternMethod *extMethod,
                     const IR::MethodCallExpression *expr) override {
        LOG3("ExternMethod: " << extMethod << ", MethodCallExpression: " << expr);
        checkBlock(expr, extMethod->originalExternType->name, extMethod->object->getName().name);
    }
 
    void checkExtern(const P4::ExternFunction *extFunction,
                     const IR::MethodCallExpression *expr) override {
        LOG3("ExternFunction: " << extFunction << ", MethodCallExpression: " << expr);
        checkBlock(expr, expr->method->toString(), cstring::empty);
    }
 
 public:
    CheckPNAExternInvocation(P4::TypeMap *typeMap, DpdkProgramStructure *structure)
        : P4::CheckExternInvocationCommon(typeMap), structure(structure) {
        initPipeConstraints();
    }
};

class CheckExternInvocation : public Inspector {
    P4::TypeMap *typeMap;
    DpdkProgramStructure *structure;
 
 public:
    CheckExternInvocation(P4::TypeMap *typeMap, DpdkProgramStructure *structure)
        : typeMap(typeMap), structure(structure) {}
 
    bool preorder(const IR::P4Program *program) {
        if (structure->isPNA()) {
            LOG1("Checking extern invocations for PNA architecture.");
            CheckPNAExternInvocation checker(typeMap, structure);
            program->apply(checker, getChildContext());
        } else if (structure->isPSA()) {
            LOG1("Checking extern invocations for PSA architecture.");
            // Add class checking PSA constraints here.
        } else {
            LOG1("Unknown architecture: " << structure->p4arch << ", not checking constraints.");
        }
        return false;
    }
};
 
}  // namespace P4::DPDK
 
#endif /* BACKENDS_DPDK_DPDKCHECKEXTERNINVOCATION_H_ */