32
32
import ivle.console
34
35
from ivle.webapp.base.rest import JSONRESTView, named_operation
35
from ivle.webapp.errors import BadRequest
37
37
# XXX: Should be RPC view, with actions in URL?
38
38
class ConsoleServiceRESTView(JSONRESTView):
39
39
'''An RPC interface to a Python console.'''
40
def get_permissions(self, user):
46
@named_operation('use')
47
41
def start(self, req, cwd=''):
48
42
working_dir = os.path.join("/home", req.user.login, cwd)
50
44
uid = req.user.unixid
53
jail_path = os.path.join(req.config['paths']['jails']['mounts'],
55
cons = ivle.console.Console(req.config, uid, jail_path, working_dir)
47
jail_path = os.path.join(ivle.conf.jail_base, req.user.login)
48
cons = ivle.console.Console(uid, jail_path, working_dir)
57
50
# Assemble the key and return it. Yes, it is double-encoded.
58
51
return {'key': cjson.encode({"host": cons.host,
60
53
"magic": cons.magic}).encode('hex')}
62
@named_operation('use')
63
def chat(self, req, key, text='', cwd='', kind="chat"):
56
def chat(self, req, key, text='', kind="chat"):
64
57
# The request *should* have the following four fields:
65
58
# key: Hex JSON dict of host and port where the console server lives,
66
59
# and the secret to use to digitally sign the communication with the
77
70
raise BadRequest("Invalid console key.")
79
jail_path = os.path.join(req.config['paths']['jails']['mounts'],
82
working_dir = os.path.join("/home", req.user.login, cwd)
72
jail_path = os.path.join(ivle.conf.jail_base, req.user.login)
73
working_dir = os.path.join("/home", req.user.login) # Within jail
83
74
uid = req.user.unixid
85
# XXX: JSONRESTView should do this for us.
86
text = text.decode('utf-8')
88
76
msg = {'cmd':kind, 'text':text}
90
78
json_response = ivle.chat.chat(host, port, msg, magic,decode=False)
97
85
response = {"terminate":
98
86
"Communication to console process lost"}
99
87
if "terminate" in response:
100
response = restart_console(req.config, uid, jail_path,
101
working_dir, response["terminate"])
88
response = restart_console(uid, jail_path, working_dir,
89
response["terminate"])
102
90
except socket.error, (enumber, estring):
103
91
if enumber == errno.ECONNREFUSED:
104
92
# Timeout: Restart the session
105
response = restart_console(req.config, uid, jail_path,
93
response = restart_console(uid, jail_path, working_dir,
107
94
"The IVLE console has timed out due to inactivity")
108
95
elif enumber == errno.ECONNRESET:
109
96
# Communication issue: Restart the session
110
response = restart_console(req.config, uid, jail_path,
97
response = restart_console(uid, jail_path, working_dir,
112
98
"Connection with the console has been reset")
114
100
# Some other error - probably serious
119
def restart_console(config, uid, jail_path, working_dir, reason):
105
def restart_console(uid, jail_path, working_dir, reason):
120
106
"""Tells the client that it must be issued a new console since the old
121
107
console is no longer availible. The client must accept the new key.
122
108
Returns the JSON response to be given to the client.
124
110
# Start a new console server console
125
cons = ivle.console.Console(config, uid, jail_path, working_dir)
111
cons = ivle.console.Console(uid, jail_path, working_dir)
127
113
# Make a JSON object to tell the browser to restart its console client
128
114
new_key = cjson.encode(