1 files changed, 112 insertions, 0 deletions
diff --git a/pods/papi/k8scmdrun.py b/pods/papi/k8scmdrun.py
new file mode 100644
index 00000000..2fb1a62e
--- /dev/null
+++ b/pods/papi/k8scmdrun.py
@@ -0,0 +1,112 @@
+# Copyright 2021 Spirent Communications.
+#
+# 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.
+"""
+Run commands inside the pod for post-deployment configuration
+"""
+
+import re
+import os
+from kubernetes import client, config
+from kubernetes.client.rest import ApiException
+from kubernetes.stream import stream
+
+def execute_command(api_instance, pod_info, exec_command):
+    """
+    Execute a command inside a specified pod
+    exec_command = list of strings
+    """
+    name = pod_info['name']
+    resp = None
+    try:
+        resp = api_instance.read_namespaced_pod(name=name,
+                                                namespace='default')
+    except ApiException as e:
+        if e.status != 404:
+            print("Unknown error: %s" % e)
+            exit(1)
+    if not resp:
+        print("Pod %s does not exist. Creating it..." % name)
+        return -1
+
+    # Calling exec and waiting for response
+    resp = stream(api_instance.connect_get_namespaced_pod_exec,
+                  name,
+                  'default',
+                  command=exec_command,
+                  stderr=True, stdin=False,
+                  stdout=True, tty=False)
+    print("Response: " + resp)
+    return resp
+
+def get_virtual_sockets(api_instance, podname, namespace):
+    """
+    Memif or VhostUser Sockets
+    """
+    socket_files = []
+    pinfo = {'name': podname,
+             'pod_ip':'',
+             'namespace': namespace}
+    cmd = ['cat', '/etc/podnetinfo/annotations']
+    resp = execute_command(api_instance, pinfo, cmd)
+    # Remove unnecessary elements
+    results = re.sub(r"(\\n|\"|\n|\\|\]|\{|\}|\[)", "", resp).strip()
+    # Remove unnecessary spaces
+    results = re.sub(r"\s+","", results, flags=re.UNICODE)
+    # Get the RHS values
+    output = results.split('=')
+    for out in output:
+        if 'socketfile' in out:
+            out2 = out.split(',')
+            for rout in out2:
+                if 'socketfile' in rout:
+                    print(rout[11:])
+                    socket_files.append(rout[11:])
+
+
+def get_sriov_interfaces(api_instance, podname, namespace):
+    """
+    Get SRIOV PIDs.
+    """
+    pinfo = {'name': podname,
+             'pod_ip':'',
+             'namespace': namespace}
+    cmd = ['cat', '/etc/podnetinfo/annotations']
+    response = execute_command(api_instance, pinfo, cmd)
+    # Remove unnecessary elements
+    results = re.sub(r"(\\n|\"|\n|\\|\]|\{|\}|\[)", "", response).strip()
+    # Remove unnecessary spaces
+    results = re.sub(r"\s+","", results, flags=re.UNICODE)
+    # Get the RHS values
+    output = results.split('=')
+    names = []
+    ifs = []
+    for out in output:
+        if 'interface' in out:
+            out2 = out.split(',')
+            for rout in out2:
+                if 'interface' in rout:
+                    ifs.append(rout[10:])
+                elif 'name' in rout:
+                    names.append(rout[5:])
+    res = {names[i]: ifs[i] for i in range(len(names))}
+
+def start_fowarding_app(api_instance, podname, namespace, appname):
+    """
+    Start the Forwarding Application
+    """
+    pinfo = {'name': podname,
+             'pod_ip':'',
+             'namespace': namespace}
+    cmd = [appname, '&']
+    response = execute_command(api_instance, pinfo, cmd)