From 2e4900d850a79c3ae2db418af61ac9704a99abd6 Mon Sep 17 00:00:00 2001
From: Aleksandra Obeso Duque <aleksandra.obeso.duque@ericsson.com>
Date: Thu, 5 May 2022 19:50:16 +0000
Subject: [PATCH] fixing stress-ng timeout less than 1s

---
 generator/src/pkg/service/util.go |  2 +-
 model/Dockerfile                  | 25 ++++++++++++++++++++++++-
 model/restful/utils/task.py       |  6 +++---
 3 files changed, 28 insertions(+), 5 deletions(-)

diff --git a/generator/src/pkg/service/util.go b/generator/src/pkg/service/util.go
index 5f3fc057..0841de4f 100644
--- a/generator/src/pkg/service/util.go
+++ b/generator/src/pkg/service/util.go
@@ -50,7 +50,7 @@ const (
 	EpExecModeDefault       = "sequential"
 	EpNwResponseSizeDefault = 512
 
-	EpExecTimeDefault = "1s"
+	EpExecTimeDefault = "0.1s"
 	EpMethodDefault   = "all"
 	EpWorkersDefault  = 1
 	EpLoadDefault     = "5%"
diff --git a/model/Dockerfile b/model/Dockerfile
index 7c460a8d..3dbfea8b 100644
--- a/model/Dockerfile
+++ b/model/Dockerfile
@@ -18,13 +18,36 @@ FROM python:3.8.0-slim
 
 RUN mkdir -p /usr/src/app
 RUN apt update
+RUN apt upgrade -y
 RUN apt install -y jq \
     wget \
-    stress-ng \
     2to3
 
+RUN apt install -y libbsd-dev \
+    libcap-dev \
+    libipsec-mb-dev \
+    libjudy-dev \
+    libkeyutils-dev \
+    libsctp-dev \
+    libatomic1 \
+    zlib1g-dev \
+    libkmod-dev \
+    libxxhash-dev \
+    git \
+    build-essential
+
 WORKDIR /usr/src/app
 
+RUN git clone https://github.com/alekodu/stress-ng.git &&\
+    cd stress-ng/ &&\
+    git checkout cloudsim &&\
+    make clean &&\
+    make &&\
+    mv stress-ng my-stress-ng &&\
+    cp my-stress-ng /usr/src/app &&\
+    cd .. &&\
+    rm -R stress-ng/
+
 ADD ./requirements.txt /usr/src/app/requirements.txt
 
 RUN pip install --upgrade pip
diff --git a/model/restful/utils/task.py b/model/restful/utils/task.py
index b8f2f468..17435f11 100644
--- a/model/restful/utils/task.py
+++ b/model/restful/utils/task.py
@@ -110,15 +110,15 @@ def run_task(service_name, service_endpoint):
 
 def execute_cpu_bounded_task(conf):
     if len(conf["cpu_affinity"]) > 0:
-        res = subprocess.run(['stress-ng --class cpu --cpu %s --cpu-method %s --taskset %s --cpu-load %s --timeout %s --metrics-brief' % (conf["workers"], conf["method"], ",".join(str(cpu_id) for cpu_id in conf["cpu_affinity"]), conf["cpu_load"], conf["execution_time"])], capture_output=True, shell=True)
+        res = subprocess.run(['/usr/src/app/my-stress-ng --class cpu --cpu %s --cpu-method %s --taskset %s --cpu-load %s --timeout %s --metrics-brief' % (conf["workers"], conf["method"], ",".join(str(cpu_id) for cpu_id in conf["cpu_affinity"]), conf["cpu_load"], conf["execution_time"])], capture_output=True, shell=True)
     else:
-        res = subprocess.run(['stress-ng --class cpu --cpu %s --cpu-method %s --cpu-load %s --timeout %s --metrics-brief' % (conf["workers"], conf["method"], conf["cpu_load"], conf["execution_time"])], capture_output=True, shell=True)
+        res = subprocess.run(['/usr/src/app/my-stress-ng --class cpu --cpu %s --cpu-method %s --cpu-load %s --timeout %s --metrics-brief' % (conf["workers"], conf["method"], conf["cpu_load"], conf["execution_time"])], capture_output=True, shell=True)
 
     return res.stderr.decode("utf-8"), "cpu"
 
 
 def execute_memory_bounded_task(conf):
-    res = subprocess.run(['stress-ng --class memory --vm %s --vm-method %s --vm-bytes %s --timeout %s --metrics-brief' % (conf["workers"], conf["method"], conf["bytes_load"], conf["execution_time"])], capture_output=True, shell=True)
+    res = subprocess.run(['/usr/src/app/my-stress-ng --class memory --vm %s --vm-method %s --vm-bytes %s --timeout %s --metrics-brief' % (conf["workers"], conf["method"], conf["bytes_load"], conf["execution_time"])], capture_output=True, shell=True)
 
     return res.stderr.decode("utf-8"), "memory"