← Back to branch summary

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

« back to all changes in this revision

Viewing changes to services/serveservice

  • Committer: William Grant
  • Date: 2010-07-28 05:06:15 UTC
  • Revision ID: grantw@unimelb.edu.au-20100728050615-uwbxn9frla3pdw8m
Encode content_type when downloading files. cjson made us write bad code.

Show diffs side-by-side

added added

removed removed

Lines of Context:
services/serveservice
17
17
# along with this program; if not, write to the Free Software
18
18
# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
19
19
 
20
 
# Script: serveservice
21
 
# Author: Thomas Conway
22
 
# Date:   6/3/2007
23
 
 
24
 
# A CGI script for serving files.
 
20
# Author: Thomas Conway, Will Grant
25
21
 
26
22
import mimetypes
27
23
import os
 
24
import sys
28
25
import StringIO
29
 
import urlparse
30
 
 
31
 
from ivle import (cgirequest, studpath)
 
26
from optparse import OptionParser
 
27
 
 
28
try:
 
29
    import json
 
30
except ImportError:
 
31
    import simplejson as json
 
32
 
32
33
from ivle import zip as zipmod
33
 
 
34
 
req = cgirequest.CGIRequest()
35
 
req.install_error_handler()
36
 
 
37
 
# Work out the parts of the URL
38
 
url = urlparse.urlparse(req.path)
39
 
querystr = url[4]
40
 
urlpath = url[2]
41
 
filename = studpath.url_to_jailpaths(urlpath)[2]
 
34
import ivle.conf.app.server
 
35
import ivle.mimetypes
 
36
 
 
37
def determine_file_type(filename):
 
38
    filetype = mimetypes.guess_type(filename)[0]
 
39
    if filetype is None:
 
40
         filetype = ivle.mimetypes.DEFAULT_MIMETYPE
 
41
    return filetype
 
42
 
 
43
def throw_error(message, extra={}):
 
44
    error = {'error': message}
 
45
    error.update(extra)
 
46
    print json.dumps(error)
 
47
    sys.exit(0)
 
48
 
 
49
parser = OptionParser()
 
50
parser.add_option('-d', '--download', dest='download', action='store_true',
 
51
                  help='force download, not execution, of the paths')
 
52
(options, args) = parser.parse_args()
 
53
 
 
54
# Detect download mode. Download mode zips any multiple selection,
 
55
# and does not execute CGI scripts.
 
56
if options.download:
 
57
    download = True
 
58
    # paths is filled later.
 
59
else:
 
60
    download = False
 
61
    assert len(args) == 1
42
62
 
43
63
default_mimetype = "application/octet-stream"
44
64
zip_mimetype = "application/zip"
46
66
zipmode = False
47
67
zipbasepath = None
48
68
zipfilename = None
49
 
path = None
50
69
 
51
 
# If any "path=" variables have been supplied, bring these into a list and
52
 
# make a zip file instead.
53
 
fields = req.get_fieldstorage()
54
 
paths = fields.getlist("path")
55
 
if len(paths) > 0:
 
70
# If multiple paths have been specified, zip them up.
 
71
if len(args) > 1:
 
72
    # Mangle the paths - we want the basename of their dirname in front.
 
73
    paths = []
 
74
    dir = os.path.dirname(args[0])
 
75
    for path in args:
 
76
        assert os.path.dirname(path) == dir
 
77
        paths.append(os.path.join(os.path.basename(dir),
 
78
                                  os.path.basename(path)))
 
79
    dir = os.path.dirname(dir)
56
80
    zipmode = True
57
 
    zipbasepath = filename
 
81
    zipbasepath = dir
58
82
    zipfilename = os.path.basename(zipbasepath)
59
 
    #for i in range(0, len(paths)):
60
 
        #paths[i] = paths[i].value
61
83
else:
62
 
    if filename is None:
63
 
        req.throw_error(req.HTTP_NOT_FOUND,
64
 
            "The path specified is invalid.")
65
 
    elif not os.access(filename, os.R_OK):
66
 
        req.throw_error(req.HTTP_NOT_FOUND,
67
 
            "The specified file (%s) does not exist." % urlpath)
68
 
     # If it's a directory, serve as a zip file
 
84
    paths = args
 
85
    filename = paths[0]
 
86
    if not os.access(filename, os.F_OK):
 
87
        # The given path doesn't exist. CGI lets us backtrack and put the path
 
88
        # elements through which we pass into PATH_INFO, so we try that.
 
89
        while not os.access(filename, os.F_OK):
 
90
            filename, path_info_frag = os.path.split(filename)
 
91
 
 
92
        # We now have a file that exists, but is it something that we're allowed
 
93
        # to execute? If not, we should 404 anyway.
 
94
        if determine_file_type(filename) not in ivle.conf.app.server.interpreters:
 
95
            throw_error('not-found')
 
96
 
 
97
    # If it's a directory, serve as a zip file
69
98
    if os.path.isdir(filename):
 
99
        if not download:
 
100
            # Not giving a directory listing - this is visible to everyone.
 
101
            throw_error('is-directory')
70
102
        zipmode = True
71
103
        # Zip it from the perspective of its own parent.
72
104
        # That way it will be a directory in the top level of the zip
79
111
        else:
80
112
            zipbasepath = splitpath[0]
81
113
            paths = [splitpath[1]]
82
 
        zipfilename = paths[0]
 
114
        zipfilename = filename
 
115
    else:
 
116
        if not download and \
 
117
           determine_file_type(filename) in ivle.conf.app.server.interpreters:
 
118
            throw_error('is-executable', {'path': filename.decode('utf-8')})
 
119
 
 
120
        if not download and (
 
121
            (ivle.conf.app.server.blacklist_served_filetypes and \
 
122
                determine_file_type(filename) in \
 
123
                ivle.conf.app.server.served_filetypes_blacklist) or \
 
124
            (ivle.conf.app.server.served_filetypes_whitelist and \
 
125
                determine_file_type(filename) not in \
 
126
                ivle.conf.app.server.served_filetypes_whitelist)):
 
127
            throw_error('forbidden')
83
128
 
84
129
if zipmode:
85
 
    req.content_type = zip_mimetype
86
130
    # zipfilename is some filename. Strip trailing slash or extension,
87
131
    # and add ".zip".
88
132
    if zipfilename == '':
92
136
    elif '.' in zipfilename:
93
137
        zipfilename = zipfilename[:zipfilename.rindex('.')]
94
138
    zipfilename += ".zip"
95
 
    req.headers_out["Content-Disposition"] = ("attachment; filename=" +   
96
 
        zipfilename)
 
139
    #req.headers_out["Content-Disposition"] = ("attachment; filename=" +   
 
140
    #    zipfilename) # TODO
97
141
    zipfile = StringIO.StringIO()
98
142
    zipmod.make_zip(zipbasepath, paths, zipfile)
99
 
        
100
 
    req.write(zipfile.getvalue())
 
143
 
 
144
    print json.dumps({'type': zip_mimetype,
 
145
                      'name': zipfilename.decode('utf-8')})
 
146
 
 
147
    stream = zipfile
 
148
    stream.seek(0)
101
149
else:
102
 
    #req.content_type = default_mimetype
103
 
    req.sendfile(filename)
 
150
 
 
151
    print json.dumps({'type': determine_file_type(filename),
 
152
                      'name': os.path.basename(filename).decode('utf-8')})
 
153
    stream = open(filename)
 
154
 
 
155
next = stream.read(1024)
 
156
while next:
 
157
    sys.stdout.write(next)
 
158
    next = stream.read(1024)