ebpfPsaParser.h

/*
 * SPDX-FileCopyrightText: 2022 Open Networking Foundation
 * SPDX-FileCopyrightText: 2022 Orange
 *
 * SPDX-License-Identifier: Apache-2.0
 */
#ifndef BACKENDS_EBPF_PSA_EBPFPSAPARSER_H_
#define BACKENDS_EBPF_PSA_EBPFPSAPARSER_H_
 
#include "backends/ebpf/ebpfParser.h"
#include "backends/ebpf/ebpfType.h"
#include "backends/ebpf/psa/ebpfPsaTable.h"
#include "backends/ebpf/psa/externs/ebpfPsaChecksum.h"
 
namespace P4::EBPF {
 
class EBPFPsaParser;
 
class PsaStateTranslationVisitor : public StateTranslationVisitor {
 public:
    EBPFPsaParser *parser;
 
    explicit PsaStateTranslationVisitor(P4::ReferenceMap *refMap, P4::TypeMap *typeMap,
                                        EBPFPsaParser *prsr)
        : StateTranslationVisitor(refMap, typeMap), parser(prsr) {}
 
    void processMethod(const P4::ExternMethod *ext) override;
};
 
class EBPFPsaParser : public EBPFParser {
 public:
    std::map<cstring, EBPFChecksumPSA *> checksums;
    const IR::Parameter *inputMetadata;
 
    EBPFPsaParser(const EBPFProgram *program, const IR::ParserBlock *block,
                  const P4::TypeMap *typeMap);
 
    void emit(CodeBuilder *builder) override;
    void emitParserInputMetadata(CodeBuilder *builder);
    void emitDeclaration(CodeBuilder *builder, const IR::Declaration *decl) override;
    void emitRejectState(CodeBuilder *builder) override;
 
    EBPFChecksumPSA *getChecksum(cstring name) const {
        auto result = ::P4::get(checksums, name);
        BUG_CHECK(result != nullptr, "No checksum named %1%", name);
        return result;
    }
 
    DECLARE_TYPEINFO(EBPFPsaParser, EBPFParser);
};
 
}  // namespace P4::EBPF
 
#endif /* BACKENDS_EBPF_PSA_EBPFPSAPARSER_H_ */