Take size of table from P4 code.

p4lang · mihaibudiu · Mar 4, 2020 · Apr 8, 2019 · Apr 8, 2019 · Apr 13, 2019
commit dbb237ea449b0a8f4d3dc9fb981470346088d5ce
diff --git a/backends/ubpf/ubpfTable.cpp b/backends/ubpf/ubpfTable.cpp
@@ -376,6 +376,21 @@ namespace UBPF {
 
         keyGenerator = table->container->getKey();
         actionList = table->container->getActionList();
+
+        setTableSize(table);
+    }
+
+    void UBPFTable::setTableSize(const IR::TableBlock *table) {
+        auto properties = table->container->properties->properties;
+        size = std::numeric_limits<int>::max();// Default value
+        for (auto property : properties) {
+            if (property->name.name == table->container->properties->sizePropertyName && property->value->is<IR::ExpressionValue>()) {
+                auto pExpressionValue = property->value->to<IR::ExpressionValue>();
+                auto pConstant = pExpressionValue->expression->to<IR::Constant>();
+                this->size = pConstant->asInt();
+                break;
+            }
+        }
     }
 
     void UBPFTable::emitKeyType(EBPF::CodeBuilder *builder) {

diff --git a/backends/ubpf/ubpfTable.h b/backends/ubpf/ubpfTable.h
@@ -48,12 +48,12 @@ namespace UBPF {
             keyTypeName = program->refMap->newName(instanceName + "_key");
             valueTypeName = program->refMap->newName(instanceName + "_value");
             dataMapName = instanceName;
-            //TODO: determine from where take this value for simple table
-            size = 1024;
         }
     };
 
     class UBPFTable final : public UBPFTableBase {
+    private:
+        void setTableSize(const IR::TableBlock *table);
     public:
         const IR::Key *keyGenerator;
         const IR::ActionList *actionList;