~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-02-26 01:09:49 UTC
  • Revision ID: grantw@unimelb.edu.au-20100226010949-xka8c9s6y7aq4id1
Scroll to the end of the console output after *every* output, not just some. This removes some artifiacts resulting from scrolling a fraction of a second too late.

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
1
# IVLE - Informatics Virtual Learning Environment
 
2
# Copyright (C) 2007-2009 The University of Melbourne
 
3
#
 
4
# This program is free software; you can redistribute it and/or modify
 
5
# it under the terms of the GNU General Public License as published by
 
6
# the Free Software Foundation; either version 2 of the License, or
 
7
# (at your option) any later version.
 
8
#
 
9
# This program is distributed in the hope that it will be useful,
 
10
# but WITHOUT ANY WARRANTY; without even the implied warranty of
 
11
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
12
# GNU General Public License for more details.
 
13
#
 
14
# You should have received a copy of the GNU General Public License
 
15
# along with this program; if not, write to the Free Software
 
16
# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
 
17
 
 
18
# Author: Matt Giuca
 
19
 
 
20
"""
 
21
IVLE Request Object
 
22
 
 
23
Builds an IVLE request object from a mod_python request object.
 
24
See design notes/apps/dispatch.txt for a full specification of this request
 
25
object.
 
26
"""
 
27
 
 
28
try:
 
29
    import mod_python.Session
 
30
    import mod_python.Cookie
 
31
    import mod_python.util
 
32
    import mod_python.apache
 
33
except ImportError:
 
34
    # This needs to be importable from outside Apache.
 
35
    pass
 
36
 
 
37
import os.path
 
38
 
 
39
import ivle.util
 
40
import ivle.database
 
41
from ivle.webapp.base.plugins import CookiePlugin
 
42
import ivle.webapp.security
 
43
 
 
44
 
 
45
class PotentiallySecureFileSession(mod_python.Session.FileSession):
 
46
    """A mod_python FileSession that sets secure cookie when appropriate.
 
47
 
 
48
    A secure cookie will be set if the request itself is over HTTPS, or if
 
49
    a proxy in front has set X-Forwarded-Proto: https. Otherwise the cookie
 
50
    will be insecure.
 
51
    """
 
52
    def make_cookie(self):
 
53
        cookie = super(PotentiallySecureFileSession, self).make_cookie()
 
54
        if (self._req.is_https() or
 
55
            self._req.headers_in.get('X-Forwarded-Proto') == 'https'):
 
56
            cookie.secure = True
 
57
        return cookie
 
58
 
 
59
 
 
60
class Request:
 
61
    """An IVLE request object. This is presented to the IVLE apps as a way of
 
62
    interacting with the web server and the dispatcher.
 
63
 
 
64
    Request object attributes:
 
65
        method (read)
 
66
            String. The request method (eg. 'GET', 'POST', etc)
 
67
        uri (read)
 
68
            String. The path portion of the URI.
 
69
        app (read)
 
70
            String. Name of the application specified in the URL, or None.
 
71
        path (read)
 
72
            String. The path specified in the URL *not including* the
 
73
            application or the IVLE location prefix. eg. a URL of
 
74
            "/ivle/files/joe/myfiles" has a path of "joe/myfiles".
 
75
        user (read)
 
76
            User object. Details of the user who is currently logged in, or
 
77
            None.
 
78
        store (read)
 
79
            storm.store.Store instance. Holds a database transaction open,
 
80
            which is available for the entire lifetime of the request.
 
81
        hostname (read)
 
82
            String. Hostname the server is running on.
 
83
        headers_in (read)
 
84
            Table object representing headers sent by the client.
 
85
        headers_out (read, can be written to)
 
86
            Table object representing headers to be sent to the client.
 
87
        publicmode (read)
 
88
            Bool. True if the request came for the "public host" as
 
89
            configured in conf.py. Note that public mode requests do not
 
90
            have an app (app is set to None).
 
91
 
 
92
        status (write)
 
93
            Int. Response status number. Use one of the status codes defined
 
94
            in class Request.
 
95
        content_type (write)
 
96
            String. The Content-Type (mime type) header value.
 
97
        location (write)
 
98
            String. Response "Location" header value. Used with HTTP redirect
 
99
            responses.
 
100
    """
 
101
 
 
102
    # Special code for an OK response.
 
103
    # Do not use HTTP_OK; for some reason Apache produces an "OK" error
 
104
    # message if you do that.
 
105
    OK  = 0
 
106
 
 
107
    # HTTP status codes
 
108
 
 
109
    HTTP_OK                           = 200
 
110
    HTTP_MOVED_TEMPORARILY            = 302
 
111
    HTTP_FORBIDDEN                    = 403
 
112
    HTTP_NOT_FOUND                    = 404
 
113
    HTTP_INTERNAL_SERVER_ERROR        = 500
 
114
 
 
115
    _store = None
 
116
 
 
117
    def __init__(self, req, config):
 
118
        """Create an IVLE request from a mod_python one.
 
119
 
 
120
        @param req: A mod_python request.
 
121
        @param config: An IVLE configuration.
 
122
        """
 
123
 
 
124
        # Methods are mostly wrappers around the Apache request object
 
125
        self.apache_req = req
 
126
        self.config = config
 
127
        self.headers_written = False
 
128
 
 
129
        # Determine if the browser used the public host name to make the
 
130
        # request (in which case we are in "public mode")
 
131
        if req.hostname == config['urls']['public_host']:
 
132
            self.publicmode = True
 
133
        else:
 
134
            self.publicmode = False
 
135
 
 
136
        # Inherit values for the input members
 
137
        self.method = req.method
 
138
        self.uri = req.uri
 
139
        # Split the given path into the app (top-level dir) and sub-path
 
140
        # (after first stripping away the root directory)
 
141
        (self.app, self.path) = (ivle.util.split_path(req.uri))
 
142
        self.hostname = req.hostname
 
143
        self.headers_in = req.headers_in
 
144
        self.headers_out = req.headers_out
 
145
 
 
146
        # Default values for the output members
 
147
        self.status = Request.HTTP_OK
 
148
        self.content_type = None        # Use Apache's default
 
149
        self.location = None
 
150
        # In some cases we don't want the template JS (such as the username
 
151
        # and public FQDN) in the output HTML. In that case, set this to 0.
 
152
        self.write_javascript_settings = True
 
153
        self.got_common_vars = False
 
154
 
 
155
    def __del__(self):
 
156
        self.cleanup()
 
157
 
 
158
    def cleanup(self):
 
159
        """Cleanup."""
 
160
        if self._store is not None:
 
161
            self._store.close()
 
162
            self._store = None
 
163
 
 
164
    def commit(self):
 
165
        """Cleanup."""
 
166
        if self._store is not None:
 
167
            self._store.commit()
 
168
 
 
169
    def __writeheaders(self):
 
170
        """Writes out the HTTP and HTML headers before any real data is
 
171
        written."""
 
172
        self.headers_written = True
 
173
 
 
174
        # Prepare the HTTP and HTML headers before the first write is made
 
175
        if self.content_type != None:
 
176
            self.apache_req.content_type = self.content_type
 
177
        self.apache_req.status = self.status
 
178
        if self.location != None:
 
179
            self.apache_req.headers_out['Location'] = self.location
 
180
 
 
181
    def ensure_headers_written(self):
 
182
        """Writes out the HTTP and HTML headers if they haven't already been
 
183
        written."""
 
184
        if not self.headers_written:
 
185
            self.__writeheaders()
 
186
 
 
187
    def write(self, string, flush=1):
 
188
        """Writes string directly to the client, then flushes the buffer,
 
189
        unless flush is 0."""
 
190
 
 
191
        if not self.headers_written:
 
192
            self.__writeheaders()
 
193
        if isinstance(string, unicode):
 
194
            # Encode unicode strings as UTF-8
 
195
            # (Otherwise cannot handle being written to a bytestream)
 
196
            self.apache_req.write(string.encode('utf8'), flush)
 
197
        else:
 
198
            # 8-bit clean strings just get written directly.
 
199
            # This includes binary strings.
 
200
            self.apache_req.write(string, flush)
 
201
 
 
202
    def logout(self):
 
203
        """Log out the current user by destroying the session state.
 
204
        Then redirect to the top-level IVLE page."""
 
205
        if hasattr(self, 'session'):
 
206
            self.session.invalidate()
 
207
            self.session.delete()
 
208
            # Invalidates all IVLE cookies
 
209
            all_cookies = mod_python.Cookie.get_cookies(self)
 
210
 
 
211
            # Create cookies for plugins that might request them.
 
212
            for plugin in self.config.plugin_index[CookiePlugin]:
 
213
                for cookie in plugin.cookies:
 
214
                    self.add_cookie(mod_python.Cookie.Cookie(cookie, '',
 
215
                                                    expires=1, path='/'))
 
216
        self.throw_redirect(self.make_path(''))
 
217
 
 
218
 
 
219
    def flush(self):
 
220
        """Flushes the output buffer."""
 
221
        self.apache_req.flush()
 
222
 
 
223
    def sendfile(self, filename):
 
224
        """Sends the named file directly to the client."""
 
225
        if not self.headers_written:
 
226
            self.__writeheaders()
 
227
        self.apache_req.sendfile(filename)
 
228
 
 
229
    def read(self, len=None):
 
230
        """Reads at most len bytes directly from the client. (See mod_python
 
231
        Request.read)."""
 
232
        if len is None:
 
233
            return self.apache_req.read()
 
234
        else:
 
235
            return self.apache_req.read(len)
 
236
 
 
237
    def throw_redirect(self, location):
 
238
        """Writes out an HTTP redirect to the specified URL. Raises an
 
239
        exception which is caught by the dispatch or web server, so any
 
240
        code following this call will not be executed.
 
241
 
 
242
        httpcode: An HTTP response status code. Pass a constant from the
 
243
        Request class.
 
244
        """
 
245
        # Note: location may be a unicode, but it MUST only have ASCII
 
246
        # characters (non-ascii characters should be URL-encoded).
 
247
        mod_python.util.redirect(self.apache_req, location.encode("ascii"))
 
248
 
 
249
    def add_cookie(self, cookie, value=None, **attributes):
 
250
        """Inserts a cookie into this request object's headers."""
 
251
        if value is None:
 
252
            mod_python.Cookie.add_cookie(self.apache_req, cookie)
 
253
        else:
 
254
            mod_python.Cookie.add_cookie(self.apache_req, cookie, value, **attributes)
 
255
 
 
256
    def make_path(self, path):
 
257
        """Prepend the IVLE URL prefix to the given path.
 
258
 
 
259
        This is used when generating URLs to send to the client.
 
260
 
 
261
        This method is DEPRECATED. We no longer support use of a prefix.
 
262
        """
 
263
        return os.path.join(self.config['urls']['root'], path)
 
264
 
 
265
    def get_session(self):
 
266
        """Returns a mod_python Session object for this request.
 
267
        Note that this is dependent on mod_python and may need to change
 
268
        interface if porting away from mod_python.
 
269
 
 
270
        IMPORTANT: Call unlock() on the session as soon as you are done with
 
271
                   it! If you don't, all other requests will block!
 
272
        """
 
273
        # Cache the session object and set the timeout to 24 hours.
 
274
        if not hasattr(self, 'session'):
 
275
            self.session = PotentiallySecureFileSession(
 
276
                self.apache_req, timeout = 60 * 60 * 24)
 
277
        return self.session
 
278
 
 
279
    def get_fieldstorage(self):
 
280
        """Returns a mod_python FieldStorage object for this request.
 
281
        Note that this is dependent on mod_python and may need to change
 
282
        interface if porting away from mod_python."""
 
283
        # Cache the fieldstorage object
 
284
        if not hasattr(self, 'fields'):
 
285
            self.fields = mod_python.util.FieldStorage(self.apache_req)
 
286
        return self.fields
 
287
 
 
288
    def get_cgi_environ(self):
 
289
        """Returns the CGI environment emulation for this request. (Calls
 
290
        add_common_vars). The environment is returned as a mapping
 
291
        compatible with os.environ."""
 
292
        if not self.got_common_vars:
 
293
            self.apache_req.add_common_vars()
 
294
            self.got_common_vars = True
 
295
        return self.apache_req.subprocess_env
 
296
 
 
297
    @property
 
298
    def store(self):
 
299
        # Open a database connection and transaction, keep it around for users
 
300
        # of the Request object to use.
 
301
        if self._store is None:
 
302
            self._store = ivle.database.get_store(self.config)
 
303
        return self._store
 
304
 
 
305
    @property
 
306
    def user(self):
 
307
        # Get and cache the request user, or None if it's not valid.
 
308
        # This is a property so that we don't create a store unless
 
309
        # some code actually requests the user.
 
310
        try:
 
311
            return self._user
 
312
        except AttributeError:
 
313
            if self.publicmode:
 
314
                self._user = None
 
315
            else:
 
316
                temp_user = ivle.webapp.security.get_user_details(self)
 
317
                if temp_user and temp_user.valid:
 
318
                    self._user = temp_user
 
319
                else:
 
320
                    self._user = None
 
321
            return self._user
 
322