Update server.py

DermotOHalloran · May 23, 2017 · e9437bc · e9437bc
1 parent 7f64a56
commit e9437bc
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/server/server.py b/server/server.py
@@ -1,5 +1,16 @@
 import http.server
 import socketserver
+
+#set the process name to "chaos_server" so we can easily kill it with "pkill chaos_server"
+def set_proc_name(newname):
+    from ctypes import cdll, byref, create_string_buffer
+    libc = cdll.LoadLibrary('libc.so.6')
+    buff = create_string_buffer(len(newname)+1)
+    buff.value = newname
+    libc.prctl(15, byref(buff), 0, 0, 0)
+set_proc_name("chaos_server")
+
+#start server on port 80
 PORT = 80
 Handler = http.server.SimpleHTTPRequestHandler
 httpd = socketserver.TCPServer(("", PORT), Handler)