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
20
# Author: Thomas Conway, Will Grant
20
# Script: serveservice
21
# Author: Thomas Conway
24
# A CGI script for serving files.
26
from optparse import OptionParser
31
import simplejson as json
31
from ivle import (cgirequest, studpath)
33
32
from ivle import zip as zipmod
34
import ivle.conf.app.server
37
def determine_file_type(filename):
38
filetype = mimetypes.guess_type(filename)[0]
40
filetype = ivle.mimetypes.DEFAULT_MIMETYPE
43
def throw_error(message, extra={}):
44
error = {'error': message}
46
print json.dumps(error)
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()
54
# Detect download mode. Download mode zips any multiple selection,
55
# and does not execute CGI scripts.
58
# paths is filled later.
34
req = cgirequest.CGIRequest()
35
req.install_error_handler()
37
# Work out the parts of the URL
38
url = urlparse.urlparse(req.path)
41
filename = studpath.url_to_jailpaths(urlpath)[2]
63
43
default_mimetype = "application/octet-stream"
64
44
zip_mimetype = "application/zip"
70
# If multiple paths have been specified, zip them up.
72
# Mangle the paths - we want the basename of their dirname in front.
74
dir = os.path.dirname(args[0])
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)
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")
57
zipbasepath = filename
82
58
zipfilename = os.path.basename(zipbasepath)
59
#for i in range(0, len(paths)):
60
#paths[i] = paths[i].value
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)
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')
97
# If it's a directory, serve as a zip file
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
98
69
if os.path.isdir(filename):
100
# Not giving a directory listing - this is visible to everyone.
101
throw_error('is-directory')
103
71
# Zip it from the perspective of its own parent.
104
72
# That way it will be a directory in the top level of the zip
112
80
zipbasepath = splitpath[0]
113
81
paths = [splitpath[1]]
114
zipfilename = filename
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')})
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')
82
zipfilename = paths[0]
85
req.content_type = zip_mimetype
130
86
# zipfilename is some filename. Strip trailing slash or extension,
132
88
if zipfilename == '':
136
92
elif '.' in zipfilename:
137
93
zipfilename = zipfilename[:zipfilename.rindex('.')]
138
94
zipfilename += ".zip"
139
#req.headers_out["Content-Disposition"] = ("attachment; filename=" +
140
# zipfilename) # TODO
95
req.headers_out["Content-Disposition"] = ("attachment; filename=" +
141
97
zipfile = StringIO.StringIO()
142
98
zipmod.make_zip(zipbasepath, paths, zipfile)
144
print json.dumps({'type': zip_mimetype,
145
'name': zipfilename.decode('utf-8')})
100
req.write(zipfile.getvalue())
151
print json.dumps({'type': determine_file_type(filename),
152
'name': os.path.basename(filename).decode('utf-8')})
153
stream = open(filename)
155
next = stream.read(1024)
157
sys.stdout.write(next)
158
next = stream.read(1024)
102
#req.content_type = default_mimetype
103
req.sendfile(filename)