← Back to branch summary

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

« back to all changes in this revision

Viewing changes to services/serveservice

  • Committer: matt.giuca
  • Date: 2009-01-12 00:34:59 UTC
  • Revision ID: svn-v3-trunk0:2b9c9e99-6f39-0410-b283-7f802c844ae2:trunk:1073
usrmgt-server: When creating svn repo, now explicitly asks for the exception
    to be thrown up, rather than ignored.

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