ubpfTable.h

/*
 * Copyright 2019 Orange
 * SPDX-FileCopyrightText: 2019 Orange
 *
 * SPDX-License-Identifier: Apache-2.0
 */
 
#ifndef BACKENDS_UBPF_UBPFTABLE_H_
#define BACKENDS_UBPF_UBPFTABLE_H_
 
#include "backends/ebpf/ebpfObject.h"
#include "frontends/p4/methodInstance.h"
#include "ubpfProgram.h"
 
namespace P4::UBPF {
 
class UBPFTableBase : public EBPF::EBPFObject {
 public:
    const UBPFProgram *program;
 
    cstring instanceName;
    cstring keyTypeName;
    cstring valueTypeName;
    const IR::Type *keyType{};
    const IR::Type *valueType{};
    cstring dataMapName;
    size_t size{};
    EBPF::CodeGenInspector *codeGen;
 
    void emitInstance(EBPF::CodeBuilder *pBuilder, EBPF::TableKind tableKind);
 
 protected:
    UBPFTableBase(const UBPFProgram *program, cstring instanceName, EBPF::CodeGenInspector *codeGen)
        : program(program), instanceName(instanceName), codeGen(codeGen) {
        CHECK_NULL(codeGen);
        CHECK_NULL(program);
        keyTypeName = program->refMap->newName(instanceName + "_key");
        valueTypeName = program->refMap->newName(instanceName + "_value");
        dataMapName = instanceName;
    }
};
 
class UBPFTable final : public UBPFTableBase {
 private:
    void setTableSize(const IR::TableBlock *table);
    void setTableKind();
 
 public:
    const IR::Key *keyGenerator;
    const IR::ActionList *actionList;
    const IR::TableBlock *table;
    EBPF::TableKind tableKind;
    cstring defaultActionMapName;
    cstring actionEnumName;
    cstring noActionName;
    std::map<const IR::KeyElement *, cstring> keyFieldNames;
    std::map<const IR::KeyElement *, EBPF::EBPFType *> keyTypes;
 
    UBPFTable(const UBPFProgram *program, const IR::TableBlock *table,
              EBPF::CodeGenInspector *codeGen);
 
    cstring generateActionName(const IR::P4Action *action);
    void emitInstance(EBPF::CodeBuilder *pBuilder);
    void emitTypes(EBPF::CodeBuilder *builder);
    void emitActionArguments(EBPF::CodeBuilder *builder, const IR::P4Action *action, cstring name);
    void emitKeyType(EBPF::CodeBuilder *builder);
    void emitValueType(EBPF::CodeBuilder *builder);
    void emitKey(EBPF::CodeBuilder *builder, cstring keyName);
    void emitAction(EBPF::CodeBuilder *builder, cstring valueName);
    void emitInitializer(EBPF::CodeBuilder *builder);
};
 
}  // namespace P4::UBPF
 
#endif  // BACKENDS_UBPF_UBPFTABLE_H_