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, config):
40
def get_permissions(self, user):
41
41
if user is not None:
42
42
return set(['use'])
53
53
jail_path = os.path.join(req.config['paths']['jails']['mounts'],
55
cons = ivle.console.Console(req.config, uid, jail_path, working_dir)
55
cons = ivle.console.Console(uid, jail_path, working_dir)
57
57
# Assemble the key and return it. Yes, it is double-encoded.
58
58
return {'key': cjson.encode({"host": cons.host,
60
60
"magic": cons.magic}).encode('hex')}
62
62
@named_operation('use')
63
def chat(self, req, key, text='', cwd='', kind="chat"):
63
def chat(self, req, key, text='', kind="chat"):
64
64
# The request *should* have the following four fields:
65
65
# key: Hex JSON dict of host and port where the console server lives,
66
66
# and the secret to use to digitally sign the communication with the
79
79
jail_path = os.path.join(req.config['paths']['jails']['mounts'],
82
working_dir = os.path.join("/home", req.user.login, cwd)
81
working_dir = os.path.join("/home", req.user.login) # Within jail
83
82
uid = req.user.unixid
85
84
# XXX: JSONRESTView should do this for us.
88
87
msg = {'cmd':kind, 'text':text}
89
json_response = ivle.chat.chat(host, port, msg, magic,decode=False)
91
# Snoop the response from python-console to check that it's valid
91
json_response = ivle.chat.chat(host, port, msg, magic,decode=False)
92
# Snoop the response from python-console to check that it's valid
93
93
response = cjson.decode(json_response)
94
except (cjson.DecodeError, ivle.chat.ProtocolError):
94
except cjson.DecodeError:
95
95
# Could not decode the reply from the python-console server
96
96
response = {"terminate":
97
97
"Communication to console process lost"}
98
98
if "terminate" in response:
99
response = restart_console(req.config, uid, jail_path,
100
working_dir, response["terminate"])
99
response = restart_console(uid, jail_path, working_dir,
100
response["terminate"])
101
101
except socket.error, (enumber, estring):
102
102
if enumber == errno.ECONNREFUSED:
103
103
# Timeout: Restart the session
104
response = restart_console(req.config, uid, jail_path,
104
response = restart_console(uid, jail_path, working_dir,
106
105
"The IVLE console has timed out due to inactivity")
107
106
elif enumber == errno.ECONNRESET:
108
107
# Communication issue: Restart the session
109
response = restart_console(req.config, uid, jail_path,
108
response = restart_console(uid, jail_path, working_dir,
111
109
"Connection with the console has been reset")
113
111
# Some other error - probably serious
118
def restart_console(config, uid, jail_path, working_dir, reason):
116
def restart_console(uid, jail_path, working_dir, reason):
119
117
"""Tells the client that it must be issued a new console since the old
120
118
console is no longer availible. The client must accept the new key.
121
119
Returns the JSON response to be given to the client.
123
121
# Start a new console server console
124
cons = ivle.console.Console(config, uid, jail_path, working_dir)
122
cons = ivle.console.Console(uid, jail_path, working_dir)
126
124
# Make a JSON object to tell the browser to restart its console client
127
125
new_key = cjson.encode(