1 files changed, 61 insertions, 0 deletions
diff --git a/VNFs/DPPD-PROX/helper-scripts/dpi/proxdpisut.py b/VNFs/DPPD-PROX/helper-scripts/dpi/proxdpisut.py
new file mode 100644
index 00000000..aae900b0
--- /dev/null
+++ b/VNFs/DPPD-PROX/helper-scripts/dpi/proxdpisut.py
@@ -0,0 +1,61 @@
+#!/bin/env python
+
+##
+## Copyright (c) 2010-2017 Intel Corporation
+##
+## 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 prox import *
+from remotesystem import *
+from time import *
+from decimal import *
+
+class ProxDpiSut(Prox):
+    def __init__(self, ts, coreCount):
+        super(ProxDpiSut, self).__init__(ts)
+
+        self._setDefaultArguments();
+        self._setDpiCoreCount(coreCount);
+
+    def _setDefaultArguments(self):
+        self.addArgument("-e");
+        self.addArgument("-t");
+        self.addArgument("-k");
+        self.addArgument("-d");
+        self.addArgument("-r 0.01");
+
+    def _setDpiCoreCount(self, count):
+        self.addArgument("-q dpi_core_count=" + str(count))
+
+    def _querySetup2(self):
+        self._query_cores();
+
+    def _query_cores(self):
+        print "querying cores"
+        self._wk = self._get_core_list("$wk");
+
+    def _get_core_list(self, var):
+        ret = []
+        result = self._send("echo " + var)._recv();
+        for e in result.split(","):
+            ret += [e];
+        return ret;
+
+    def getTsc(self):
+        cmd = "stats task.core(%s).task(0).tsc" % self._wk[-1]
+        res = int(self._send(cmd)._recv());
+        if (res == 0):
+            return self._getTsc();
+        else:
+            return res;