Restrict next url parameter
[astakos] / snf-astakos-app / astakos / im / target / redirect.py
1 # Copyright 2011-2012 GRNET S.A. All rights reserved.
2 #
3 # Redistribution and use in source and binary forms, with or
4 # without modification, are permitted provided that the following
5 # conditions are met:
6 #
7 #   1. Redistributions of source code must retain the above
8 #      copyright notice, this list of conditions and the following
9 #      disclaimer.
10 #
11 #   2. Redistributions in binary form must reproduce the above
12 #      copyright notice, this list of conditions and the following
13 #      disclaimer in the documentation and/or other materials
14 #      provided with the distribution.
15 #
16 # THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS
17 # OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
18 # WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
19 # PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR
20 # CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
21 # SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
22 # LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
23 # USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
24 # AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25 # LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
26 # ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
27 # POSSIBILITY OF SUCH DAMAGE.
28 #
29 # The views and conclusions contained in the software and
30 # documentation are those of the authors and should not be
31 # interpreted as representing official policies, either expressed
32 # or implied, of GRNET S.A.
33
34 from django.core.urlresolvers import reverse
35 from django.shortcuts import redirect
36 from django.utils.translation import ugettext as _
37 from django.contrib import messages
38 from django.utils.http import urlencode
39 from django.contrib.auth import authenticate
40 from django.http import (
41     HttpResponse, HttpResponseBadRequest, HttpResponseForbidden
42 )
43 from django.core.exceptions import ValidationError
44 from django.views.decorators.http import require_http_methods
45
46 from urllib import quote
47 from urlparse import urlunsplit, urlsplit, urlparse, parse_qsl
48
49 from astakos.im.settings import COOKIE_NAME, COOKIE_DOMAIN
50 from astakos.im.util import set_cookie, restrict_next
51 from astakos.im.functions import login as auth_login, logout
52
53 import logging
54
55 logger = logging.getLogger(__name__)
56
57 @require_http_methods(["GET", "POST"])
58 def login(request):
59     """
60     If there is no ``next`` request parameter redirects to astakos index page
61     displaying an error message.
62     If the request user is authenticated and has signed the approval terms,
63     redirects to `next` request parameter. If not, redirects to approval terms
64     in order to return back here after agreeing with the terms.
65     Otherwise, redirects to login in order to return back here after successful login.
66     """
67     next = request.GET.get('next')
68     if not next:
69         return HttpResponseBadRequest(_('No next parameter'))
70     if not restrict_next(
71         next, domain=COOKIE_DOMAIN, allowed_schemes=('pithos',)
72     ):
73         return HttpResponseForbidden(_('Not allowed next parameter'))
74     force = request.GET.get('force', None)
75     response = HttpResponse()
76     if force == '':
77         logout(request)
78         response.delete_cookie(COOKIE_NAME, path='/', domain=COOKIE_DOMAIN)
79     if request.user.is_authenticated():
80         # if user has not signed the approval terms
81         # redirect to approval terms with next the request path
82         if not request.user.signed_terms():
83             # first build next parameter
84             parts = list(urlsplit(request.build_absolute_uri()))
85             params = dict(parse_qsl(parts[3], keep_blank_values=True))
86             # delete force parameter
87             parts[3] = urlencode(params)
88             next = urlunsplit(parts)
89             
90             # build url location
91             parts[2] = reverse('latest_terms')
92             params = {'next':next}
93             parts[3] = urlencode(params)
94             url = urlunsplit(parts)
95             response['Location'] = url
96             response.status_code = 302
97             return response
98         renew = request.GET.get('renew', None)
99         if renew == '':
100             request.user.renew_token()
101             try:
102                 request.user.save()
103             except ValidationError, e:
104                 return HttpResponseBadRequest(e)
105             # authenticate before login
106             user = authenticate(email=request.user.email, auth_token=request.user.auth_token)
107             auth_login(request, user)
108             set_cookie(response, user)
109             logger.info('Token reset for %s' % request.user.email)
110         parts = list(urlsplit(next))
111         parts[3] = urlencode({'user': request.user.email, 'token': request.user.auth_token})
112         url = urlunsplit(parts)
113         response['Location'] = url
114         response.status_code = 302
115         return response
116     else:
117         # redirect to login with next the request path
118         
119         # first build next parameter
120         parts = list(urlsplit(request.build_absolute_uri()))
121         params = dict(parse_qsl(parts[3], keep_blank_values=True))
122         # delete force parameter
123         if 'force' in params:
124             del params['force']
125         parts[3] = urlencode(params)
126         next = urlunsplit(parts)
127         
128         # build url location
129         parts[2] = reverse('astakos.im.views.index')
130         params = {'next':next}
131         parts[3] = urlencode(params)
132         url = urlunsplit(parts)
133         response['Location'] = url
134         response.status_code = 302
135         return response