summaryrefslogtreecommitdiffstats
path: root/extras/packetforge/flow_create.py
diff options
context:
space:
mode:
Diffstat (limited to 'extras/packetforge/flow_create.py')
-rw-r--r--extras/packetforge/flow_create.py157
1 files changed, 157 insertions, 0 deletions
diff --git a/extras/packetforge/flow_create.py b/extras/packetforge/flow_create.py
new file mode 100644
index 00000000000..fbbdec56694
--- /dev/null
+++ b/extras/packetforge/flow_create.py
@@ -0,0 +1,157 @@
+# 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 vpp_papi.vpp_papi import VPPApiClient
+import sys, getopt
+import packetforge
+import fnmatch
+import os
+
+# Get VPP json API file directory
+CLIENT_ID = "Vppclient"
+VPP_JSON_DIR = (
+ os.path.abspath("../..") + "/build-root/install-vpp-native/vpp/share/vpp/api/core"
+)
+VPP_JSON_DIR_PLUGIN = (
+ os.path.abspath("../..")
+ + "/build-root/install-vpp-native/vpp/share/vpp/api/plugins"
+)
+API_FILE_SUFFIX = "*.api.json"
+
+
+def Main(argv):
+ file_flag = False
+ operation = None
+ try:
+ opts, args = getopt.getopt(
+ argv,
+ "hf:p:a:i:I:",
+ [
+ "help=",
+ "add",
+ "del",
+ "file=",
+ "pattern=",
+ "actions=",
+ "interface=",
+ "flow-index=",
+ ],
+ )
+ except getopt.GetoptError:
+ print(
+ "flow_create.py --add|del -f <file> -p <pattern> -a <actions> -i <interface> -I <flow-index>"
+ )
+ sys.exit()
+ for opt, arg in opts:
+ if opt == "-h":
+ print(
+ "flow_create.py --add|del -f <file> -p <pattern> -a <actions> -i <interface> -I <flow-index>"
+ )
+ sys.exit()
+ elif opt == "--add":
+ operation = "add"
+ elif opt == "--del":
+ operation = "del"
+ elif opt in ("-f", "--file"):
+ actions = ""
+ json_file = arg
+ file_flag = True
+ elif opt in ("-p", "--pattern") and not file_flag:
+ pattern = arg
+ elif opt in ("-a", "--actions"):
+ actions = arg
+ elif opt in ("-i", "--interface"):
+ iface = arg
+ elif opt in ("-I", "--flow-index"):
+ flow_index = arg
+
+ if operation == None:
+ print("Error: Please choose the operation: add or del")
+ sys.exit()
+
+ if operation == "add":
+ if not file_flag:
+ result = packetforge.Forge(pattern, actions, False)
+ else:
+ result = packetforge.Forge(json_file, actions, True)
+ return result, int(iface), operation, None
+ elif operation == "del":
+ return None, int(iface), operation, int(flow_index)
+
+
+def load_json_api_files(suffix=API_FILE_SUFFIX):
+ jsonfiles = []
+ json_dir = VPP_JSON_DIR
+ for root, dirnames, filenames in os.walk(json_dir):
+ for filename in fnmatch.filter(filenames, suffix):
+ jsonfiles.append(os.path.join(json_dir, filename))
+ json_dir = VPP_JSON_DIR_PLUGIN
+ for root, dirnames, filenames in os.walk(json_dir):
+ for filename in fnmatch.filter(filenames, suffix):
+ jsonfiles.append(os.path.join(json_dir, filename))
+ if not jsonfiles:
+ raise RuntimeError("Error: no json api files found")
+ else:
+ print("load json api file done")
+ return jsonfiles
+
+
+def connect_vpp(jsonfiles):
+ vpp = VPPApiClient(apifiles=jsonfiles)
+ r = vpp.connect("CLIENT_ID")
+ print("VPP api opened with code: %s" % r)
+ return vpp
+
+
+if __name__ == "__main__":
+ # Python API need json definitions to interpret messages
+ vpp = connect_vpp(load_json_api_files())
+ print(vpp.api.show_version())
+
+ # Parse the arguments
+ my_flow, iface, operation, del_flow_index = Main(sys.argv[1:])
+
+ # set inteface states
+ vpp.api.sw_interface_set_flags(sw_if_index=iface, flags=1)
+
+ if operation == "add":
+ # add flow
+ rv = vpp.api.flow_add_v2(flow=my_flow)
+ flow_index = rv[3]
+ print(rv)
+
+ # enable added flow
+ rv = vpp.api.flow_enable(flow_index=flow_index, hw_if_index=iface)
+ ena_res = rv[2]
+ # if enable flow fail, delete added flow
+ if ena_res:
+ print("Error: enable flow failed, delete flow")
+ rv = vpp.api.flow_del(flow_index=flow_index)
+ print(rv)
+
+ elif operation == "del":
+ # disable flow
+ rv = vpp.api.flow_disable(flow_index=del_flow_index, hw_if_index=iface)
+ dis_res = rv[2]
+ if dis_res:
+ print("Error: disable flow failed")
+ sys.exit()
+ print(rv)
+
+ # delete flow
+ rv = vpp.api.flow_del(flow_index=del_flow_index)
+ print(rv)
+
+# command example:
+# python flow_create.py --add -p "mac()/ipv4(src=1.1.1.1,dst=2.2.2.2)/udp()" -a "redirect-to-queue 3" -i 1
+# python flow_create.py --del -i 1 -I 0