targets/ebpf/program_info.h

/*
 * SPDX-FileCopyrightText: 2022 The P4 Language Consortium
 *
 * SPDX-License-Identifier: Apache-2.0
 */
 
#ifndef BACKENDS_P4TOOLS_MODULES_TESTGEN_TARGETS_EBPF_PROGRAM_INFO_H_
#define BACKENDS_P4TOOLS_MODULES_TESTGEN_TARGETS_EBPF_PROGRAM_INFO_H_
 
#include <vector>
 
#include "ir/ir.h"
#include "lib/cstring.h"
#include "lib/ordered_map.h"
 
#include "backends/p4tools/modules/testgen/core/program_info.h"
#include "backends/p4tools/modules/testgen/lib/continuation.h"
 
namespace P4::P4Tools::P4Testgen::EBPF {
 
class EBPFProgramInfo : public ProgramInfo {
 private:
    const ordered_map<cstring, const IR::Type_Declaration *> programmableBlocks;

    static const IR::Type_Bits PARSER_ERR_BITS;

    std::vector<Continuation::Command> processDeclaration(const IR::Type_Declaration *cstrType,
                                                          size_t blockIdx) const;
 
 public:
    EBPFProgramInfo(const TestgenCompilerResult &compilerResult,
                    ordered_map<cstring, const IR::Type_Declaration *> inputBlocks);

    [[nodiscard]] const ArchSpec &getArchSpec() const override;

    [[nodiscard]] const ordered_map<cstring, const IR::Type_Declaration *> *getProgrammableBlocks()
        const;
 
    [[nodiscard]] const IR::StateVariable &getTargetInputPortVar() const override;
 
    [[nodiscard]] const IR::StateVariable &getTargetOutputPortVar() const override;
 
    [[nodiscard]] const IR::Expression *dropIsActive() const override;
 
    [[nodiscard]] const IR::Type_Bits *getParserErrorType() const override;

    static const ArchSpec ARCH_SPEC;
 
    DECLARE_TYPEINFO(EBPFProgramInfo, ProgramInfo);
};
 
}  // namespace P4::P4Tools::P4Testgen::EBPF
 
#endif /* BACKENDS_P4TOOLS_MODULES_TESTGEN_TARGETS_EBPF_PROGRAM_INFO_H_ */