← Back to branch summary

~azzar1/unity/add-show-desktop-key

« back to all changes in this revision

Viewing changes to ivle/interpret.py

  • Committer: Matt Giuca
  • Date: 2010-02-25 09:34:17 UTC
  • Revision ID: matt.giuca@gmail.com-20100225093417-02z6stnx9biqpbol
Exercise display: Shows a warning if the worksheet cutoff has passed for this subject, that it will not count towards your marks.

Show diffs side-by-side

added added

removed removed

Lines of Context:
ivle/interpret.py
23
23
 
24
24
import ivle
25
25
from ivle import studpath
26
 
from ivle.util import IVLEJailError, split_path
 
26
from ivle.util import IVLEError, IVLEJailError, split_path
27
27
 
28
28
import functools
29
29
 
37
37
# working on smaller output
38
38
 
39
39
CGI_BLOCK_SIZE = 65535
40
 
PATH = "/usr/local/bin:/usr/bin:/bin"
41
40
 
42
41
def interpret_file(req, owner, jail_dir, filename, interpreter, gentle=True):
43
42
    """Serves a file by interpreting it using one of IVLE's builtin
74
73
    # (Note that paths "relative" to the jail actually begin with a '/' as
75
74
    # they are absolute in the jailspace)
76
75
 
77
 
    return interpreter(owner, jail_dir, working_dir, filename_abs, req,
 
76
    return interpreter(owner.unixid, jail_dir, working_dir, filename_abs, req,
78
77
                       gentle)
79
78
 
80
79
class CGIFlags:
89
88
        self.linebuf = ""
90
89
        self.headers = {}       # Header names : values
91
90
 
92
 
def execute_cgi(interpreter, owner, jail_dir, working_dir, script_path,
 
91
def execute_cgi(interpreter, uid, jail_dir, working_dir, script_path,
93
92
                req, gentle):
94
93
    """
95
94
    trampoline: Full path on the local system to the CGI wrapper program
96
95
        being executed.
97
 
    owner: User object of the owner of the file.
 
96
    uid: User ID of the owner of the file.
98
97
    jail_dir: Absolute path of owner's jail directory.
99
98
    working_dir: Directory containing the script file relative to owner's
100
99
        jail.
130
129
        f.seek(0)       # Rewind, for reading
131
130
 
132
131
    # Set up the environment
133
 
    environ = cgi_environ(req, script_path, owner)
 
132
    # This automatically asks mod_python to load up the CGI variables into the
 
133
    # environment (which is a good first approximation)
 
134
    old_env = os.environ.copy()
 
135
    for k in os.environ.keys():
 
136
        del os.environ[k]
 
137
    for (k,v) in req.get_cgi_environ().items():
 
138
        os.environ[k] = v
 
139
    fixup_environ(req, script_path)
134
140
 
135
141
    # usage: tramp uid jail_dir working_dir script_path
136
 
    cmd_line = [trampoline, str(owner.unixid),
137
 
            req.config['paths']['jails']['mounts'],
138
 
            req.config['paths']['jails']['src'],
139
 
            req.config['paths']['jails']['template'],
140
 
            jail_dir, working_dir, interpreter, script_path]
 
142
    cmd_line = [trampoline, str(uid), req.config['paths']['jails']['mounts'],
 
143
         req.config['paths']['jails']['src'],
 
144
         req.config['paths']['jails']['template'],
 
145
         jail_dir, working_dir, interpreter, script_path]
141
146
    # Popen doesn't like unicode strings. It hateses them.
142
147
    cmd_line = [(s.encode('utf-8') if isinstance(s, unicode) else s)
143
148
                for s in cmd_line]
144
149
    pid = subprocess.Popen(cmd_line,
145
150
        stdin=f, stdout=subprocess.PIPE, stderr=subprocess.STDOUT,
146
 
        cwd=tramp_dir, env=environ)
 
151
        cwd=tramp_dir)
 
152
 
 
153
    # Restore the environment
 
154
    for k in os.environ.keys():
 
155
        del os.environ[k]
 
156
    for (k,v) in old_env.items():
 
157
        os.environ[k] = v
147
158
 
148
159
    # We don't want any output! Bail out after the process terminates.
149
160
    if noop:
218
229
        # Is this an internal IVLE error condition?
219
230
        hs = cgiflags.headers
220
231
        if 'X-IVLE-Error-Type' in hs:
221
 
            try:
222
 
                raise IVLEJailError(hs['X-IVLE-Error-Type'],
223
 
                                    hs['X-IVLE-Error-Message'],
224
 
                                    hs['X-IVLE-Error-Info'])
225
 
            except KeyError:
226
 
                raise AssertionError("Bad error headers written by CGI.")
 
232
            t = hs['X-IVLE-Error-Type']
 
233
            if t == IVLEError.__name__:
 
234
                raise IVLEError(int(hs['X-IVLE-Error-Code']),
 
235
                                hs['X-IVLE-Error-Message'])
 
236
            else:
 
237
                try:
 
238
                    raise IVLEJailError(hs['X-IVLE-Error-Type'],
 
239
                                        hs['X-IVLE-Error-Message'],
 
240
                                        hs['X-IVLE-Error-Info'])
 
241
                except KeyError:
 
242
                    raise IVLEError(500, 'bad error headers written by CGI')
227
243
 
228
244
        # Check to make sure the required headers were written
229
245
        if cgiflags.wrote_html_warning or not cgiflags.gentle:
345
361
    # python-server-page
346
362
}
347
363
 
348
 
def cgi_environ(req, script_path, user):
349
 
    """Gets CGI variables from apache and makes a few changes for security and 
350
 
    correctness.
 
364
def fixup_environ(req, script_path):
 
365
    """Assuming os.environ has been written with the CGI variables from
 
366
    apache, make a few changes for security and correctness.
351
367
 
352
368
    Does not modify req, only reads it.
353
369
    """
354
 
    env = {}
 
370
    env = os.environ
355
371
    # Comments here are on the heavy side, explained carefully for security
356
372
    # reasons. Please read carefully before making changes.
357
 
    
358
 
    # This automatically asks mod_python to load up the CGI variables into the
359
 
    # environment (which is a good first approximation)
360
 
    for (k,v) in req.get_cgi_environ().items():
361
 
        env[k] = v
362
373
 
363
374
    # Remove DOCUMENT_ROOT and SCRIPT_FILENAME. Not part of CGI spec and
364
375
    # exposes unnecessary details about server.
407
418
    env['SERVER_SOFTWARE'] = "IVLE/" + ivle.__version__
408
419
 
409
420
    # Additional environment variables
410
 
    username = user.login
 
421
    username = split_path(req.path)[0]
411
422
    env['HOME'] = os.path.join('/home', username)
412
423
 
413
 
    return env
414
 
 
415
424
class ExecutionError(Exception):
416
425
    pass
417
426
 
435
444
                for s in cmd_line]
436
445
    proc = subprocess.Popen(cmd_line,
437
446
        stdin=subprocess.PIPE, stdout=subprocess.PIPE,
438
 
        stderr=subprocess.PIPE, cwd=tramp_dir, close_fds=True,
439
 
        env={'HOME': os.path.join('/home', user.login),
440
 
             'PATH': PATH,
441
 
             'USER': user.login,
442
 
             'LOGNAME': user.login})
 
447
        stderr=subprocess.PIPE, cwd=tramp_dir, close_fds=True)
443
448
 
444
449
    (stdout, stderr) = proc.communicate()
445
450
    exitcode = proc.returncode
446
451
 
447
452
    if exitcode != 0:
448
 
        raise ExecutionError('subprocess ended with code %d, stderr: "%s"' %
449
 
                             (exitcode, stderr))
 
453
        raise ExecutionError('subprocess ended with code %d, stderr %s' %
 
454
                             (exitcode, proc.stderr.read()))
450
455
    return (stdout, stderr)