adding stringio for security reasons

Author: paigerube14

run_kraken.py

@@ -31,11 +31,6 @@
 KUBE_BURNER_VERSION = "0.9.1"
 
 
-def publish_kraken_status(status):
-    with open("/tmp/kraken_status", "w+") as file:
-        file.write(str(status))
-
-
 # Main function
 def main(cfg):
     # Start kraken
@@ -131,8 +126,8 @@ def main(cfg):
                     port
                 )
             )
-            server.start_server(address)
-            publish_kraken_status(run_signal)
+            logging.info("Publishing kraken status at http://%s:%s" % (server_address, port))
+            server.start_server(address, run_signal)
 
         # Cluster info
         logging.info("Fetching cluster info")
@@ -179,6 +174,7 @@ def main(cfg):
 
         # Capture the start time
         start_time = int(time.time())
+        litmus_installed = False
 
         # Loop to run the chaos starts here
         while int(iteration) < iterations and run_signal != "STOP":
@@ -281,6 +277,7 @@ def main(cfg):
                                         litmus_version,
                                         litmus_namespace
                                     )
+                                litmus_installed = True
                                 common_litmus.run(
                                     scenarios_list,
                                     config,

server.py

@@ -4,6 +4,7 @@
 from http.server import HTTPServer, BaseHTTPRequestHandler
 from http.client import HTTPConnection
 
+server_status = ""
 
 class SimpleHTTPRequestHandler(BaseHTTPRequestHandler):
     """
@@ -19,8 +20,7 @@ def do_GET(self):
     def do_status(self):
         self.send_response(200)
         self.end_headers()
-        f = open("/tmp/kraken_status", "rb")
-        self.wfile.write(f.read())
+        self.wfile.write(bytes(server_status, encoding='utf8'))
         SimpleHTTPRequestHandler.requests_served += 1
 
     def do_POST(self):
@@ -34,31 +34,35 @@ def do_POST(self):
     def set_run(self):
         self.send_response(200)
         self.end_headers()
-        with open("/tmp/kraken_status", "w+") as file:
-            file.write(str("RUN"))
+        global server_status
+        server_status = 'RUN'
 
     def set_stop(self):
         self.send_response(200)
         self.end_headers()
-        with open("/tmp/kraken_status", "w+") as file:
-            file.write(str("STOP"))
+        global server_status
+        server_status = 'STOP'
 
     def set_pause(self):
         self.send_response(200)
         self.end_headers()
-        with open("/tmp/kraken_status", "w+") as file:
-            file.write(str("PAUSE"))
+        global server_status
+        server_status = 'PAUSE'
 
+def publish_kraken_status(status):
+    global server_status
+    server_status = status
 
-def start_server(address):
+def start_server(address, status):
     server = address[0]
     port = address[1]
     global httpd
     httpd = HTTPServer(address, SimpleHTTPRequestHandler)
     logging.info("Starting http server at http://%s:%s\n" % (server, port))
     try:
         _thread.start_new_thread(httpd.serve_forever, ())
-    except Exception:
+        publish_kraken_status(status)
+    except Exception as e:
         logging.error(
             "Failed to start the http server \
                       at http://%s:%s"