[ptftest] Add p4_dash_utils

dash-pipeline/dockerfiles/Dockerfile.saithrift-client-bldr

@@ -15,7 +15,7 @@ ADD requirements.txt /tests/
 RUN apt update && apt install -y python3 python3-pip sudo && \
     sudo update-alternatives --install /usr/bin/python python /usr/bin/python3.8 1 && \
     sudo python3 -m pip install -r /tests/requirements.txt && \
-    sudo pip3 install scapy pysubnettree
+    sudo pip3 install scapy pysubnettree p4runtime
 
 WORKDIR /
 

test/test-cases/functional/ptf/p4_dash_utils.py

@@ -0,0 +1,115 @@
+import grpc
+from p4.v1 import p4runtime_pb2
+from p4.v1 import p4runtime_pb2_grpc
+
+class P4info():
+    def __init__(self, stub):
+        self.config = P4info.get_pipeline_config(stub)
+
+    @staticmethod
+    def get_pipeline_config(stub):
+        try:
+            req = p4runtime_pb2.GetForwardingPipelineConfigRequest()
+            req.device_id = 0
+            req.response_type = p4runtime_pb2.GetForwardingPipelineConfigRequest.ResponseType.P4INFO_AND_COOKIE
+            return stub.GetForwardingPipelineConfig(req).config.p4info
+        except Exception as e:
+            print(f'gRPC error: str({e})')
+            return None
+
+    def get_table(self, name):
+        for table in self.config.tables:
+            if table.preamble.name == name:
+                return table
+
+        return None
+
+    def get_action(self, name):
+        for action in self.config.actions:
+            if action.preamble.name == name:
+                return action
+
+        return None
+
+
+def toggle_flow_lookup(enabled):
+    channel = grpc.insecure_channel('localhost:9559')
+    stub = p4runtime_pb2_grpc.P4RuntimeStub(channel)
+
+    p4info = P4info(stub)
+    internal_config = p4info.get_table("dash_ingress.dash_lookup_stage.pre_pipeline_stage.internal_config")
+
+    entry = p4runtime_pb2.TableEntry()
+    entry.table_id = internal_config.preamble.id
+    entry.priority = 1
+
+    match = entry.match.add()
+    match.field_id = 1
+    match.ternary.value = b'\x00'
+    match.ternary.mask = b'\xff'
+
+    req = p4runtime_pb2.ReadRequest()
+    req.device_id = 0
+    entity = req.entities.add()
+    entity.table_entry.CopyFrom(entry)
+    for response in stub.Read(req):
+        if not response.entities:
+            continue
+        entry = response.entities[0].table_entry
+        param = entry.action.action.params[2] #flow_enabled
+        if param.value[0] == enabled:
+            return # none of change
+
+        param.value = b'\x01' if enabled else b'\x00'
+        req = p4runtime_pb2.WriteRequest()
+        req.device_id = 0
+        update = req.updates.add()
+        update.type = p4runtime_pb2.Update.MODIFY
+        update.entity.table_entry.CopyFrom(entry)
+        stub.Write(req)
+        return
+
+    # Add one entry
+    set_internal_config = p4info.get_action("dash_ingress.dash_lookup_stage.pre_pipeline_stage.set_internal_config")
+    entry.action.action.action_id = set_internal_config.preamble.id
+    action = entry.action.action
+    param = action.params.add()
+    param.param_id = 1
+    param.value = b'\x00\x00\x00\x00\x00\x00'
+    param = action.params.add()
+    param.param_id = 2
+    param.value = b'\x00\x00\x00\x00\x00\x00'
+    param = action.params.add()
+    param.param_id = 3
+    param.value = b'\x01' if enabled else b'\x00'
+
+    req = p4runtime_pb2.WriteRequest()
+    req.device_id = 0
+    update = req.updates.add()
+    update.type = p4runtime_pb2.Update.INSERT
+    update.entity.table_entry.CopyFrom(entry)
+    stub.Write(req)
+
+
+def use_flow(cls):
+    _setUp = getattr(cls, "setUp", None)
+    _tearDown = getattr(cls, "tearDown", None)
+
+    def setUp(self, *args, **kwargs):
+        if _setUp is not None:
+            _setUp(self, *args, **kwargs)
+        print(f'*** Enable Flow lookup')
+        toggle_flow_lookup(True)
+        return
+
+    def tearDown(self, *args, **kwargs):
+        print(f'*** Disable Flow lookup')
+        toggle_flow_lookup(False)
+        if _tearDown is not None:
+            _tearDown(self, *args, **kwargs)
+        return
+
+    setattr(cls, "setUp", setUp)
+    setattr(cls, "tearDown", tearDown)
+    return cls
+