forked from acm-uiuc/clearwaters-container
-
Notifications
You must be signed in to change notification settings - Fork 0
/
cw_server.py
59 lines (54 loc) · 1.76 KB
/
cw_server.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
from cw_libs import clearwaters_docker as cwd
from cw_libs import daemon
import socket
import sys
import json
from thread import *
HOST = '' # Symbolic name meaning all available interfaces
PORT = 8888 # Arbitrary non-privileged port
docker_client = cwd.CWDockerClient()
def doCommand(cmd):
global docker_client
if cmd == "TEST":
print(docker_client.create_container('echo hello world'))
class ClientDaemon(daemon.Daemon):
def run(self):
while True:
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
try:
s.bind((HOST, PORT))
except socket.error as msg:
print 'Bind failed. Error Code : ' + str(msg[0]) + ' Message ' + msg[1]
sys.exit(1)
s.listen(10)
def clientthread(conn):
while True:
data = conn.recv(4096)
if not data:
break
d = json.loads(data)
if not d:
conn.sendall("Not JSON!")
cmd = d.get('CMD')
doCommand(cmd)
conn.close()
while 1:
conn, addr = s.accept()
start_new_thread(clientthread ,(conn,))
s.close()
if __name__ == "__main__":
daemon = ClientDaemon('/tmp/cwclientd-daemon.pid', stdout='/home/clearwaters/cw-stdout', stderr='/home/clearwaters/cw-stderr')
if len(sys.argv) == 2:
if 'start' == sys.argv[1]:
daemon.start()
elif 'stop' == sys.argv[1]:
daemon.stop()
elif 'restart' == sys.argv[1]:
daemon.restart()
else:
print "Unknown Command"
sys.exit(2)
sys.exit(0)
else:
print "Usage: %s start|stop|restart" % sys.argv[0]
sys.exit(2)