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
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
|
import threading
import platform
import time
from enum import Enum
import json
import queue
if platform.system() == 'Linux':
from src.linux import ogOperations
class jsonResponse():
def __init__(self):
self.jsontree = {}
def addElement(self, key, value):
self.jsontree[key] = value
def dumpMsg(self):
return json.dumps(self.jsontree)
class ogThread():
# Executing cmd thread
def execcmd(msgqueue, cmd):
msgqueue.queue.clear()
msgqueue.put(ogOperations.execCMD(cmd))
# Powering off thread
def poweroff():
time.sleep(2)
ogOperations.poweroff()
# Rebooting thread
def reboot():
ogOperations.reboot()
class ogResponses(Enum):
BAD_REQUEST=0
IN_PROGRESS=1
OK=2
class ogRest():
def __init__(self):
self.msgqueue = queue.Queue(1000)
def getResponse(self, response, jsonResp=None):
msg = ''
if response == ogResponses.BAD_REQUEST:
msg = 'HTTP/1.0 400 Bad request'
elif response == ogResponses.IN_PROGRESS:
msg = 'HTTP/1.0 202 Accepted'
elif response == ogResponses.OK:
msg = 'HTTP/1.0 200 OK'
else:
return msg
if not jsonResp == None:
msg = msg + '\nContent-Type:application/json'
msg = msg + '\nContent-Length:' + str(len(jsonResp.dumpMsg()))
msg = msg + '\n' + jsonResp.dumpMsg()
msg = msg + '\r\n\r\n'
return msg
def processOperation(self, op, URI, cmd, client):
if ("GET" in op):
if ("probe" in URI):
self.process_probe(client)
elif ("shell/output" in URI):
self.process_shellout(client)
else:
client.send(self.getResponse(ogResponses.BAD_REQUEST))
elif ("POST" in op):
if ("poweroff" in URI):
self.process_poweroff(client)
elif ("reboot" in URI):
self.process_reboot(client)
elif ("shell/run" in URI):
self.process_shellrun(client, cmd)
else:
client.send(self.getResponse(ogResponses.BAD_REQUEST))
else:
client.send(self.getResponse(ogResponses.BAD_REQUEST))
return 0
def process_reboot(self, client):
client.send(self.getResponse(ogResponses.IN_PROGRESS))
client.disconnect()
threading.Thread(target=ogThread.reboot).start()
def process_poweroff(self, client):
client.send(self.getResponse(ogResponses.IN_PROGRESS))
client.disconnect()
threading.Thread(target=ogThread.poweroff).start()
def process_probe(self, client):
client.send(self.getResponse(ogResponses.OK))
def process_shellrun(self, client, cmd):
if cmd == None:
client.send(self.getResponse(ogResponses.BAD_REQUEST))
return
try:
ogThread.execcmd(self.msgqueue, cmd)
except ValueError as err:
print(err.args[0])
client.send(self.getResponse(ogResponses.BAD_REQUEST))
return
client.send(self.getResponse(ogResponses.OK))
def process_shellout(self, client):
jsonResp = jsonResponse()
if self.msgqueue.empty():
jsonResp.addElement('out', '')
client.send(self.getResponse(ogResponses.OK, jsonResp))
else:
jsonResp.addElement('out', self.msgqueue.get())
client.send(self.getResponse(ogResponses.OK, jsonResp))
|
