1 files changed, 92 insertions, 0 deletions
diff --git a/extras/packetforge/Edge.py b/extras/packetforge/Edge.py
new file mode 100644
index 00000000000..08e6c9d9ad3
--- /dev/null
+++ b/extras/packetforge/Edge.py
@@ -0,0 +1,92 @@
+# Copyright (c) 2022 Intel and/or its affiliates.
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at:
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from EdgeAction import *
+import json
+
+
+class Edge:
+    def __init__(self):
+        self.JSON = None
+        self.Start = None
+        self.End = None
+        self.actionList = []
+
+    def Create(jsonfile):
+        f = open(jsonfile, "r", encoding="utf-8")
+        token = json.load(f)
+
+        if token == None:
+            return None
+
+        if token["type"] != "edge":
+            return None
+
+        edgeList = []
+
+        startNodes = token["start"]
+        endNodes = token["end"]
+
+        if startNodes == None or endNodes == None:
+            return None
+
+        startTokens = startNodes.split(",")
+        endTokens = endNodes.split(",")
+
+        for start in startTokens:
+            for end in endTokens:
+
+                edge = Edge()
+
+                edge.Start = start
+                edge.End = end
+
+                if "actions" in token:
+                    for at in token["actions"]:
+                        action = EdgeAction.Create(at)
+                        if not action:
+                            return None
+
+                        edge.actionList.append(action)
+
+                edge.JSON = jsonfile
+                edgeList.append(edge)
+
+        return edgeList
+
+    def Apply(self, first, second):
+        exp = []
+
+        for i in range(len(self.actionList)):
+            act = self.actionList[i]
+
+            if act.FromStartObject == True:
+                exp.append(first.GetValue(act.FromExpression))
+            elif act.FromStartObject == False:
+                exp.append(second.GetValue(act.FromExpression))
+            else:
+                exp.append(act.FromExpression)
+
+        for i in range(len(exp)):
+            act = self.actionList[i]
+
+            if act.ToStartObject:
+                first.SetFieldAuto(act.ToExpression, exp[i])
+            else:
+                second.SetFieldAuto(act.ToExpression, exp[i])
+
+    def Actions(self):
+        return self.actionList
+
+    def Name(self):
+        return self.Start + "_" + self.End