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

« back to all changes in this revision

Viewing changes to ivle/dispatch/request.py

  • Committer: William Grant
  • Date: 2010-07-27 12:09:13 UTC
  • mto: This revision was merged to the branch mainline in revision 1826.
  • Revision ID: grantw@unimelb.edu.au-20100727120913-v0kfnwxzbiwrjnue
(simple)json always returns a unicode when decoding, while cjson returned a str where possible. This makes cPickle unhappy, so convert back to a str.

Show diffs side-by-side

added added

removed removed

Lines of Context:
29
29
    import mod_python.Session
30
30
    import mod_python.Cookie
31
31
    import mod_python.util
 
32
    import mod_python.apache
 
33
 
 
34
    class PotentiallySecureFileSession(mod_python.Session.FileSession):
 
35
        """A mod_python FileSession that sets secure cookie when appropriate.
 
36
 
 
37
        A secure cookie will be set if the request itself is over HTTPS, or if
 
38
        a proxy in front has set X-Forwarded-Proto: https. Otherwise the cookie
 
39
        will be insecure.
 
40
        """
 
41
        def make_cookie(self):
 
42
            cookie = super(PotentiallySecureFileSession, self).make_cookie()
 
43
            if (self._req.is_https() or
 
44
                self._req.headers_in.get('X-Forwarded-Proto') == 'https'):
 
45
                cookie.secure = True
 
46
            return cookie
32
47
except ImportError:
33
48
    # This needs to be importable from outside Apache.
34
49
    pass
35
50
 
 
51
import os.path
 
52
 
36
53
import ivle.util
37
 
import ivle.conf
38
54
import ivle.database
39
55
from ivle.webapp.base.plugins import CookiePlugin
 
56
import ivle.webapp.security
 
57
 
40
58
 
41
59
class Request:
42
60
    """An IVLE request object. This is presented to the IVLE apps as a way of
47
65
            String. The request method (eg. 'GET', 'POST', etc)
48
66
        uri (read)
49
67
            String. The path portion of the URI.
 
68
        unparsed_uri (read)
 
69
            String. The path portion of the URI, unparsed with query string.
50
70
        app (read)
51
71
            String. Name of the application specified in the URL, or None.
52
72
        path (read)
78
98
        location (write)
79
99
            String. Response "Location" header value. Used with HTTP redirect
80
100
            responses.
81
 
        title (write)
82
 
            String. HTML page title. Used if write_html_head_foot is True, in
83
 
            the HTML title element text.
84
 
        styles (write)
85
 
            List of strings. Write a list of URLs to CSS files here, and they
86
 
            will be incorporated as <link rel="stylesheet" type="text/css">
87
 
            elements in the head, if write_html_head_foot is True.
88
 
            URLs should be relative to the IVLE root; they will be fixed up
89
 
            to be site-relative.
90
 
        scripts (write)
91
 
            List of strings. Write a list of URLs to JS files here, and they
92
 
            will be incorporated as <script type="text/javascript"> elements
93
 
            in the head, if write_html_head_foot is True.
94
 
            URLs should be relative to the IVLE root; they will be fixed up
95
 
            to be site-relative.
96
 
        scripts_init (write)
97
 
            List of strings. Write a list of JS function names, and they
98
 
            will be added as window.addListener('load', ..., false); calls
99
 
            in the head, if write_html_head_foot is True.
100
 
            This is the propper way to specify functions that need to run at 
101
 
            page load time.
102
 
        write_html_head_foot (write)
103
 
            Boolean. If True, dispatch assumes that this is an XHTML page, and
104
 
            will immediately write a full HTML head, open the body element,
105
 
            and write heading contents to the page, before any bytes are
106
 
            written. It will then write footer contents and close the body and
107
 
            html elements at the end of execution.  
108
 
 
109
 
            This value should be set to true by all applications for all HTML
110
 
            output (unless there is a good reason, eg. exec). The
111
 
            applications should therefore output HTML content assuming that
112
 
            it will be written inside the body tag. Do not write opening or
113
 
            closing <html> or <body> tags.
114
101
    """
115
102
 
116
103
    # Special code for an OK response.
120
107
 
121
108
    # HTTP status codes
122
109
 
123
 
    HTTP_CONTINUE                     = 100
124
 
    HTTP_SWITCHING_PROTOCOLS          = 101
125
 
    HTTP_PROCESSING                   = 102
126
110
    HTTP_OK                           = 200
127
 
    HTTP_CREATED                      = 201
128
 
    HTTP_ACCEPTED                     = 202
129
 
    HTTP_NON_AUTHORITATIVE            = 203
130
 
    HTTP_NO_CONTENT                   = 204
131
 
    HTTP_RESET_CONTENT                = 205
132
 
    HTTP_PARTIAL_CONTENT              = 206
133
 
    HTTP_MULTI_STATUS                 = 207
134
 
    HTTP_MULTIPLE_CHOICES             = 300
135
 
    HTTP_MOVED_PERMANENTLY            = 301
136
111
    HTTP_MOVED_TEMPORARILY            = 302
137
 
    HTTP_SEE_OTHER                    = 303
138
 
    HTTP_NOT_MODIFIED                 = 304
139
 
    HTTP_USE_PROXY                    = 305
140
 
    HTTP_TEMPORARY_REDIRECT           = 307
141
 
    HTTP_BAD_REQUEST                  = 400
142
 
    HTTP_UNAUTHORIZED                 = 401
143
 
    HTTP_PAYMENT_REQUIRED             = 402
144
112
    HTTP_FORBIDDEN                    = 403
145
113
    HTTP_NOT_FOUND                    = 404
146
 
    HTTP_METHOD_NOT_ALLOWED           = 405
147
 
    HTTP_NOT_ACCEPTABLE               = 406
148
 
    HTTP_PROXY_AUTHENTICATION_REQUIRED= 407
149
 
    HTTP_REQUEST_TIME_OUT             = 408
150
 
    HTTP_CONFLICT                     = 409
151
 
    HTTP_GONE                         = 410
152
 
    HTTP_LENGTH_REQUIRED              = 411
153
 
    HTTP_PRECONDITION_FAILED          = 412
154
 
    HTTP_REQUEST_ENTITY_TOO_LARGE     = 413
155
 
    HTTP_REQUEST_URI_TOO_LARGE        = 414
156
 
    HTTP_UNSUPPORTED_MEDIA_TYPE       = 415
157
 
    HTTP_RANGE_NOT_SATISFIABLE        = 416
158
 
    HTTP_EXPECTATION_FAILED           = 417
159
 
    HTTP_UNPROCESSABLE_ENTITY         = 422
160
 
    HTTP_LOCKED                       = 423
161
 
    HTTP_FAILED_DEPENDENCY            = 424
162
114
    HTTP_INTERNAL_SERVER_ERROR        = 500
163
 
    HTTP_NOT_IMPLEMENTED              = 501
164
 
    HTTP_BAD_GATEWAY                  = 502
165
 
    HTTP_SERVICE_UNAVAILABLE          = 503
166
 
    HTTP_GATEWAY_TIME_OUT             = 504
167
 
    HTTP_VERSION_NOT_SUPPORTED        = 505
168
 
    HTTP_VARIANT_ALSO_VARIES          = 506
169
 
    HTTP_INSUFFICIENT_STORAGE         = 507
170
 
    HTTP_NOT_EXTENDED                 = 510
171
 
 
172
 
    def __init__(self, req, write_html_head):
173
 
        """Builds an IVLE request object from a mod_python request object.
174
 
        This results in an object with all of the necessary methods and
175
 
        additional fields.
176
 
 
177
 
        req: A mod_python request object.
178
 
        write_html_head: Function which is called when writing the automatic
179
 
            HTML header. Accepts a single argument, the IVLE request object.
 
115
 
 
116
    _store = None
 
117
 
 
118
    def __init__(self, req, config):
 
119
        """Create an IVLE request from a mod_python one.
 
120
 
 
121
        @param req: A mod_python request.
 
122
        @param config: An IVLE configuration.
180
123
        """
181
124
 
182
125
        # Methods are mostly wrappers around the Apache request object
183
126
        self.apache_req = req
184
 
        self.func_write_html_head = write_html_head
 
127
        self.config = config
185
128
        self.headers_written = False
186
129
 
187
130
        # Determine if the browser used the public host name to make the
188
131
        # request (in which case we are in "public mode")
189
 
        if req.hostname == ivle.conf.public_host:
 
132
        if req.hostname == config['urls']['public_host']:
190
133
            self.publicmode = True
191
134
        else:
192
135
            self.publicmode = False
194
137
        # Inherit values for the input members
195
138
        self.method = req.method
196
139
        self.uri = req.uri
 
140
        self.unparsed_uri = req.unparsed_uri
197
141
        # Split the given path into the app (top-level dir) and sub-path
198
142
        # (after first stripping away the root directory)
199
 
        path = ivle.util.unmake_path(req.uri)
200
 
        (self.app, self.path) = (ivle.util.split_path(path))
201
 
        self.user = None
 
143
        (self.app, self.path) = (ivle.util.split_path(req.uri))
202
144
        self.hostname = req.hostname
203
145
        self.headers_in = req.headers_in
204
146
        self.headers_out = req.headers_out
205
147
 
206
 
        # Open a database connection and transaction, keep it around for users
207
 
        # of the Request object to use
208
 
        self.store = ivle.database.get_store()
209
 
 
210
148
        # Default values for the output members
211
149
        self.status = Request.HTTP_OK
212
150
        self.content_type = None        # Use Apache's default
213
151
        self.location = None
214
 
        self.title = None     # Will be set by dispatch before passing to app
215
 
        self.styles = []
216
 
        self.scripts = []
217
 
        self.scripts_init = []
218
 
        self.write_html_head_foot = False
219
152
        # In some cases we don't want the template JS (such as the username
220
153
        # and public FQDN) in the output HTML. In that case, set this to 0.
221
154
        self.write_javascript_settings = True
222
155
        self.got_common_vars = False
223
156
 
224
157
    def __del__(self):
225
 
        """Cleanup."""
226
 
        self.store.close()
 
158
        self.cleanup()
 
159
 
 
160
    def cleanup(self):
 
161
        """Cleanup."""
 
162
        if self._store is not None:
 
163
            self._store.close()
 
164
            self._store = None
 
165
 
 
166
    def commit(self):
 
167
        """Cleanup."""
 
168
        if self._store is not None:
 
169
            self._store.commit()
227
170
 
228
171
    def __writeheaders(self):
229
172
        """Writes out the HTTP and HTML headers before any real data is
230
173
        written."""
231
174
        self.headers_written = True
232
 
        
233
 
        # app is the App object for the chosen app
234
 
        try:
235
 
            app = ivle.conf.apps.app_url[self.app]
236
 
        except KeyError:
237
 
            app = None
238
175
 
239
176
        # Prepare the HTTP and HTML headers before the first write is made
240
177
        if self.content_type != None:
242
179
        self.apache_req.status = self.status
243
180
        if self.location != None:
244
181
            self.apache_req.headers_out['Location'] = self.location
245
 
        if self.write_html_head_foot:
246
 
            # Write the HTML header, pass "self" (request object)
247
 
            self.func_write_html_head(self)
248
182
 
249
183
    def ensure_headers_written(self):
250
184
        """Writes out the HTTP and HTML headers if they haven't already been
281
215
                for cookie in plugin.cookies:
282
216
                    self.add_cookie(mod_python.Cookie.Cookie(cookie, '',
283
217
                                                    expires=1, path='/'))
284
 
        self.throw_redirect(ivle.util.make_path('')) 
 
218
        self.throw_redirect(self.make_path(''))
285
219
 
286
220
 
287
221
    def flush(self):
302
236
        else:
303
237
            return self.apache_req.read(len)
304
238
 
305
 
    def throw_error(self, httpcode, message=None):
306
 
        """Writes out an HTTP error of the specified code. Raises an exception
307
 
        which is caught by the dispatch or web server, so any code following
308
 
        this call will not be executed.
309
 
 
310
 
        httpcode: An HTTP response status code. Pass a constant from the
311
 
        Request class.
312
 
        """
313
 
        raise ivle.util.IVLEError(httpcode, message)
314
 
 
315
239
    def throw_redirect(self, location):
316
240
        """Writes out an HTTP redirect to the specified URL. Raises an
317
241
        exception which is caught by the dispatch or web server, so any
331
255
        else:
332
256
            mod_python.Cookie.add_cookie(self.apache_req, cookie, value, **attributes)
333
257
 
 
258
    def make_path(self, path):
 
259
        """Prepend the IVLE URL prefix to the given path.
 
260
 
 
261
        This is used when generating URLs to send to the client.
 
262
 
 
263
        This method is DEPRECATED. We no longer support use of a prefix.
 
264
        """
 
265
        return os.path.join(self.config['urls']['root'], path)
 
266
 
334
267
    def get_session(self):
335
268
        """Returns a mod_python Session object for this request.
336
269
        Note that this is dependent on mod_python and may need to change
337
 
        interface if porting away from mod_python."""
 
270
        interface if porting away from mod_python.
 
271
 
 
272
        IMPORTANT: Call unlock() on the session as soon as you are done with
 
273
                   it! If you don't, all other requests will block!
 
274
        """
338
275
        # Cache the session object and set the timeout to 24 hours.
339
276
        if not hasattr(self, 'session'):
340
 
            self.session = mod_python.Session.FileSession(self.apache_req,
341
 
                                               timeout = 60 * 60 * 24)
 
277
            self.session = PotentiallySecureFileSession(
 
278
                self.apache_req, timeout = 60 * 60 * 24)
342
279
        return self.session
343
280
 
344
281
    def get_fieldstorage(self):
359
296
            self.got_common_vars = True
360
297
        return self.apache_req.subprocess_env
361
298
 
362
 
    @staticmethod
363
 
    def get_http_codename(code):
364
 
        """Given a HTTP error code int, returns a (name, description)
365
 
        pair, suitable for displaying to the user.
366
 
        May return (None,None) if code is unknown.
367
 
        Only lists common 4xx and 5xx codes (since this is just used
368
 
        to display throw_error error messages).
369
 
        """
 
299
    @property
 
300
    def store(self):
 
301
        # Open a database connection and transaction, keep it around for users
 
302
        # of the Request object to use.
 
303
        if self._store is None:
 
304
            self._store = ivle.database.get_store(self.config)
 
305
        return self._store
 
306
 
 
307
    @property
 
308
    def user(self):
 
309
        # Get and cache the request user, or None if it's not valid.
 
310
        # This is a property so that we don't create a store unless
 
311
        # some code actually requests the user.
370
312
        try:
371
 
            return http_codenames[code]
372
 
        except KeyError:
373
 
            return None, None
 
313
            return self._user
 
314
        except AttributeError:
 
315
            if self.publicmode:
 
316
                self._user = None
 
317
            else:
 
318
                temp_user = ivle.webapp.security.get_user_details(self)
 
319
                if temp_user and temp_user.valid:
 
320
                    self._user = temp_user
 
321
                else:
 
322
                    self._user = None
 
323
            return self._user
374
324
 
375
 
# Human strings for HTTP response codes
376
 
http_codenames = {
377
 
    Request.HTTP_BAD_REQUEST:
378
 
        ("Bad Request",
379
 
        "Your browser sent a request IVLE did not understand."),
380
 
    Request.HTTP_UNAUTHORIZED:
381
 
        ("Unauthorized",
382
 
        "You are not allowed to view this part of IVLE."),
383
 
    Request.HTTP_FORBIDDEN:
384
 
        ("Forbidden",
385
 
        "You are not allowed to view this part of IVLE."),
386
 
    Request.HTTP_NOT_FOUND:
387
 
        ("Not Found",
388
 
        "The application or file you requested does not exist."),
389
 
    Request.HTTP_METHOD_NOT_ALLOWED:
390
 
        ("Method Not Allowed",
391
 
        "Your browser is interacting with IVLE in the wrong way."
392
 
        "This is probably a bug in IVLE. "
393
 
        "Please report it to the administrators."),
394
 
    Request.HTTP_INTERNAL_SERVER_ERROR:
395
 
        ("Internal Server Error",
396
 
        "An unknown error occured in IVLE."),
397
 
    Request.HTTP_NOT_IMPLEMENTED:
398
 
        ("Not Implemented",
399
 
        "The application or file you requested has not been implemented "
400
 
        "in IVLE."),
401
 
    Request.HTTP_SERVICE_UNAVAILABLE:
402
 
        ("Service Unavailable",
403
 
        "IVLE is currently experiencing technical difficulties. "
404
 
        "Please try again later."),
405
 
}