From 4f3dc0c8018d8ca6ff31429043d64ccbdae8b956 Mon Sep 17 00:00:00 2001 From: Sebastian Annies Date: Sun, 9 Oct 2011 23:22:31 +0200 Subject: [PATCH] first try to implement proxy tickets * already issueing pgt and pt * not sure what happens when proxies are chained --- AUTHORS.txt | 1 + MANIFEST.in | 1 + cas_provider/forms.py | 2 +- cas_provider/models.py | 46 +++++++++++-- cas_provider/tests.py | 124 ++++++++++++++++++++++++---------- cas_provider/urls.py | 2 + cas_provider/views.py | 149 ++++++++++++++++++++++++++++++++++------- setup.py | 5 ++ 8 files changed, 262 insertions(+), 68 deletions(-) diff --git a/AUTHORS.txt b/AUTHORS.txt index af9d445..9900ace 100644 --- a/AUTHORS.txt +++ b/AUTHORS.txt @@ -3,3 +3,4 @@ Alex Kamedov Łukasz Rekucki Marek Stepniowski Fred Wenzel +Sebastian Annies diff --git a/MANIFEST.in b/MANIFEST.in index 775b4b5..a87457c 100644 --- a/MANIFEST.in +++ b/MANIFEST.in @@ -1,4 +1,5 @@ recursive-include cas_provider/templates *.html +recursive-include cas_provider/fixtures * include AUTHORS.txt include README.rst include LICENSE \ No newline at end of file diff --git a/cas_provider/forms.py b/cas_provider/forms.py index 731e7da..ba6aff0 100644 --- a/cas_provider/forms.py +++ b/cas_provider/forms.py @@ -26,7 +26,7 @@ class LoginForm(AuthenticationForm): return ticket def clean(self): - super(LoginForm, self).clean(self) + AuthenticationForm.clean(self) self.cleaned_data.get('lt').delete() return self.cleaned_data diff --git a/cas_provider/models.py b/cas_provider/models.py index 2075c2e..cb14c35 100644 --- a/cas_provider/models.py +++ b/cas_provider/models.py @@ -12,10 +12,8 @@ else: # Python <2.6 compatibility from cgi import parse_qs - __all__ = ['ServiceTicket', 'LoginTicket'] - class BaseTicket(models.Model): ticket = models.CharField(_('ticket'), max_length=32) created = models.DateTimeField(_('created'), auto_now=True) @@ -50,17 +48,53 @@ class ServiceTicket(BaseTicket): parsed = urlparse.urlparse(self.service) query = parse_qs(parsed.query) query['ticket'] = [self.ticket] - query = [ ((k, v) if len(v) > 1 else (k, v[0])) for k, v in query.iteritems()] + query = [((k, v) if len(v) > 1 else (k, v[0])) for k, v in query.iteritems()] parsed = urlparse.ParseResult(parsed.scheme, parsed.netloc, - parsed.path, parsed.params, - urllib.urlencode(query), parsed.fragment) + parsed.path, parsed.params, + urllib.urlencode(query), parsed.fragment) return parsed.geturl() class LoginTicket(BaseTicket): - prefix = 'LT' class Meta: verbose_name = _('Login Ticket') verbose_name_plural = _('Login Tickets') + + +class ProxyGrantingTicket(BaseTicket): + serviceTicket = models.ForeignKey(ServiceTicket, null=True) + pgtiou = models.CharField(max_length=256, verbose_name=_('PGTiou')) + prefix = 'PGT' + + def __init__(self, *args, **kwargs): + if 'pgtiou' not in kwargs: + kwargs['pgtiou'] = u"PGTIOU-%s" % (''.join(Random().sample(string.ascii_letters + string.digits, 50))) + super(ProxyGrantingTicket, self).__init__(*args, **kwargs) + + class Meta: + verbose_name = _('Proxy Granting Ticket') + verbose_name_plural = _('Proxy Granting Tickets') + + +class ProxyTicket(ServiceTicket): + proxyGrantingTicket = models.ForeignKey(ProxyGrantingTicket, verbose_name=_('Proxy Granting Ticket')) + + prefix = 'PT' + + class Meta: + verbose_name = _('Proxy Ticket') + verbose_name_plural = _('Proxy Tickets') + + +class ProxyGrantingTicketIOU(BaseTicket): + proxyGrantingTicket = models.ForeignKey(ProxyGrantingTicket, verbose_name=_('Proxy Granting Ticket')) + + prefix = 'PGTIOU' + + class Meta: + verbose_name = _('Proxy Granting Ticket IOU') + verbose_name_plural = _('Proxy Granting Tickets IOU') + + diff --git a/cas_provider/tests.py b/cas_provider/tests.py index 4b486de..2165fee 100644 --- a/cas_provider/tests.py +++ b/cas_provider/tests.py @@ -1,21 +1,72 @@ +import StringIO +import urllib2 +from xml import etree +from xml.etree import ElementTree +from cas_provider.attribute_formatters import CAS, NSMAP from cas_provider.models import ServiceTicket -from cas_provider.views import _cas2_sucess_response, _cas2_error_response, \ - INVALID_TICKET +from cas_provider.views import _cas2_sucess_response, INVALID_TICKET, _cas2_error_response from django.contrib.auth.models import User from django.core.urlresolvers import reverse from django.test import TestCase from urlparse import urlparse from django.conf import settings +import cas_provider + +def dummy_urlopen(url): + pass class ViewsTest(TestCase): - fixtures = ['cas_users.json', ] + fixtures = ['cas_users', ] def setUp(self): self.service = 'http://example.com/' + def test_successful_login_with_proxy(self): + urllib2.urlopen = dummy_urlopen # monkey patching urllib2.urlopen so that the testcase doesnt really opens a url + proxyTarget = "http://my.sweet.service" + + response = self._login_user('root', '123') + response = self._validate_cas2(response, True, proxyTarget ) + + # Test: I'm acting as the service that will call another service + # Step 1: Get the proxy granting ticket + responseXml = ElementTree.parse(StringIO.StringIO(response.content)) + auth_success = responseXml.find(CAS + 'authenticationSuccess', namespaces=NSMAP) + pgt = auth_success.find(CAS + "proxyGrantingTicket", namespaces=NSMAP) + user = auth_success.find(CAS + "user", namespaces=NSMAP) + self.assertEqual('root', user.text) + self.assertIsNotNone(pgt.text) + self.assertTrue(pgt.text.startswith('PGTIOU')) + + #Step 2: Get the actual proxy ticket + proxyTicketResponse = self.client.get(reverse('proxy'), {'targetService': proxyTarget, 'pgt': pgt.text}, follow=False) + proxyTicketResponseXml = ElementTree.parse(StringIO.StringIO(proxyTicketResponse.content)) + self.assertIsNotNone(proxyTicketResponseXml.find(CAS + "proxySuccess", namespaces=NSMAP)) + self.assertIsNotNone(proxyTicketResponseXml.find(CAS + "proxySuccess/cas:proxyTicket", namespaces=NSMAP)) + proxyTicket = proxyTicketResponseXml.find(CAS + "proxySuccess/cas:proxyTicket", namespaces=NSMAP); + + #Step 3: I have the proxy ticket I can talk to some other backend service as the currently logged in user! + proxyValidateResponse = self.client.get(reverse('cas_proxy_validate'), {'ticket': proxyTicket.text, 'service': proxyTarget, 'pgtUrl': None}) + proxyValidateResponseXml = ElementTree.parse(StringIO.StringIO(proxyValidateResponse.content)) + + auth_success_2 = proxyValidateResponseXml.find(CAS + 'authenticationSuccess', namespaces=NSMAP) + user_2 = auth_success.find(CAS + "user", namespaces=NSMAP) + proxies = auth_success.find(CAS + "proxies") + self.assertIsNotNone(auth_success_2) + self.assertEqual(user.text, user_2.text) + self.assertIsNotNone(proxies) + + + def test_successful_proxy_chaining(self): + self.assertFalse(True) + + def test_successful_service_not_matching_in_request_to_proxy(self): + self.assertFalse(True) + + def test_succeessful_login(self): response = self._login_user('root', '123') self._validate_cas1(response, True) @@ -73,47 +124,47 @@ class ViewsTest(TestCase): response = self._validate_cas2(response, True) self.assertEqual(response.content, '''''' - '''''' - '''editor''' - '''''' - '''Jasig''' - '''editor''' - '''True''' - '''True''' - '''editor@exapmle.com''' - '''''' - '''''' - '''''') + '''''' + '''editor''' + '''''' + '''Jasig''' + '''editor''' + '''True''' + '''True''' + '''editor@exapmle.com''' + '''''' + '''''' + '''''') self._cas_logout() response = self._login_user('editor', '123') settings.CAS_CUSTOM_ATTRIBUTES_FORMATER = 'cas_provider.attribute_formatters.ruby_cas' response = self._validate_cas2(response, True) self.assertEqual(response.content, '''''' - '''''' - '''editor''' - '''RubyCAS''' - '''editor''' - '''True''' - '''True''' - '''editor@exapmle.com''' - '''''' - '''''') + '''''' + '''editor''' + '''RubyCAS''' + '''editor''' + '''True''' + '''True''' + '''editor@exapmle.com''' + '''''' + '''''') self._cas_logout() response = self._login_user('editor', '123') settings.CAS_CUSTOM_ATTRIBUTES_FORMATER = 'cas_provider.attribute_formatters.name_value' response = self._validate_cas2(response, True) self.assertEqual(response.content, '''''' - '''''' - '''editor''' - '''''' - '''''' - '''''' - '''''' - '''''' - '''''' - '''''') + '''''' + '''editor''' + '''''' + '''''' + '''''' + '''''' + '''''' + '''''' + '''''') def test_cas2_fail_validate(self): @@ -167,14 +218,16 @@ class ViewsTest(TestCase): self.assertEqual(response.content, u'no\n\n') - def _validate_cas2(self, response, is_correct=True): + def _validate_cas2(self, response, is_correct=True, pgtUrl = None): if is_correct: self.assertEqual(response.status_code, 302) self.assertTrue(response.has_header('location')) location = urlparse(response['location']) ticket = location.query.split('=')[1] - - response = self.client.get(reverse('cas_service_validate'), {'ticket': ticket, 'service': self.service}, follow=False) + if pgtUrl: + response = self.client.get(reverse('cas_service_validate'), {'ticket': ticket, 'service': self.service, 'pgtUrl': pgtUrl}, follow=False) + else: + response = self.client.get(reverse('cas_service_validate'), {'ticket': ticket, 'service': self.service}, follow=False) self.assertEqual(response.status_code, 200) else: self.assertEqual(response.status_code, 200) @@ -185,7 +238,6 @@ class ViewsTest(TestCase): self.assertEqual(response.content, _cas2_error_response(INVALID_TICKET).content) return response - class ModelsTestCase(TestCase): fixtures = ['cas_users.json', ] diff --git a/cas_provider/urls.py b/cas_provider/urls.py index 16a6744..37da0af 100644 --- a/cas_provider/urls.py +++ b/cas_provider/urls.py @@ -4,6 +4,8 @@ from django.conf.urls.defaults import patterns, url urlpatterns = patterns('cas_provider.views', url(r'^login/?$', 'login', name='cas_login'), url(r'^validate/?$', 'validate', name='cas_validate'), + url(r'^proxy/?$', 'proxy', name='proxy'), url(r'^serviceValidate/?$', 'service_validate', name='cas_service_validate'), + url(r'^proxyValidate/?$', 'proxy_validate', name='cas_proxy_validate'), url(r'^logout/?$', 'logout', name='cas_logout'), ) diff --git a/cas_provider/views.py b/cas_provider/views.py index 7a3e690..717aa6f 100644 --- a/cas_provider/views.py +++ b/cas_provider/views.py @@ -1,16 +1,21 @@ +from lxml import etree +from urllib import urlencode +import urllib2 +import urlparse from django.conf import settings from django.contrib.auth import login as auth_login, logout as auth_logout from django.core.urlresolvers import get_callable from django.http import HttpResponse, HttpResponseRedirect from django.shortcuts import render_to_response from django.template import RequestContext +from cas_provider.attribute_formatters import NSMAP, CAS +from cas_provider.models import ProxyGrantingTicket, ProxyTicket from forms import LoginForm from models import ServiceTicket, LoginTicket __all__ = ['login', 'validate', 'logout', 'service_validate'] - INVALID_TICKET = 'INVALID_TICKET' INVALID_SERVICE = 'INVALID_SERVICE' INVALID_REQUEST = 'INVALID_REQUEST' @@ -21,13 +26,13 @@ ERROR_MESSAGES = ( (INVALID_SERVICE, u'Service is invalid'), (INVALID_REQUEST, u'Not all required parameters were sent.'), (INTERNAL_ERROR, u'An internal error occurred during ticket validation'), -) + ) -def login(request, template_name='cas/login.html', \ - success_redirect=settings.LOGIN_REDIRECT_URL, - warn_template_name='cas/warn.html', - form_class=LoginForm): +def login(request, template_name='cas/login.html',\ + success_redirect=settings.LOGIN_REDIRECT_URL, + warn_template_name='cas/warn.html', + form_class=LoginForm): service = request.GET.get('service', None) if request.user.is_authenticated(): if service is not None: @@ -60,7 +65,7 @@ def login(request, template_name='cas/login.html', \ return render_to_response(template_name, { 'form': form, 'errors': form.get_errors() if hasattr(form, 'get_errors') else None, - }, context_instance=RequestContext(request)) + }, context_instance=RequestContext(request)) def validate(request): @@ -75,28 +80,40 @@ def validate(request): ticket = ServiceTicket.objects.get(ticket=ticket_string) assert ticket.service == service username = ticket.user.username - ticket.delete() return HttpResponse("yes\n%s\n" % username) except: pass return HttpResponse("no\n\n") -def logout(request, template_name='cas/logout.html', \ - auto_redirect=settings.CAS_AUTO_REDIRECT_AFTER_LOGOUT): +def logout(request, template_name='cas/logout.html', + auto_redirect=settings.CAS_AUTO_REDIRECT_AFTER_LOGOUT): url = request.GET.get('url', None) if request.user.is_authenticated(): + for ticket in ServiceTicket.objects.filter(user = request.user): + ticket.delete() auth_logout(request) if url and auto_redirect: return HttpResponseRedirect(url) - return render_to_response(template_name, {'url': url}, \ + return render_to_response(template_name, {'url': url},\ context_instance=RequestContext(request)) +def proxy(request): + targetService = request.GET['targetService'] + pgtiou = request.GET['pgt'] -def service_validate(request): - """Validate ticket via CAS v.2 protocol""" - service = request.GET.get('service', None) - ticket_string = request.GET.get('ticket', None) + try: + proxyGrantingTicket = ProxyGrantingTicket.objects.get(pgtiou=pgtiou) + except ProxyGrantingTicket.DoesNotExist: + return _cas2_error_response(INVALID_TICKET) + + pt = ProxyTicket.objects.create(proxyGrantingTicket = proxyGrantingTicket, + user=proxyGrantingTicket.serviceTicket.user, + service=targetService) + return _cas2_proxy_success(pt.ticket) + + +def ticket_validate(service, ticket_string, pgtUrl): if service is None or ticket_string is None: return _cas2_error_response(INVALID_REQUEST) @@ -106,31 +123,99 @@ def service_validate(request): return _cas2_error_response(INVALID_TICKET) if ticket.service != service: - ticket.delete() return _cas2_error_response(INVALID_SERVICE) + + pgtIouId = None + proxies = () + if pgtUrl is not None: + pgt = generate_proxy_granting_ticket(pgtUrl, ticket) + if pgt: + pgtIouId = pgt.pgtiou + + while pgt: + proxies += (pgt.serviceTicket.service,) + pgt = pgt.serviceTicket.proxyGrantingTicket if hasattr(pgt.serviceTicket, 'proxyGrantingTicket') else None + + user = ticket.user - ticket.delete() - return _cas2_sucess_response(user) + return _cas2_sucess_response(user, pgtIouId, proxies) + + +def service_validate(request): + """Validate ticket via CAS v.2 protocol""" + service = request.GET.get('service', None) + ticket_string = request.GET.get('ticket', None) + pgtUrl = request.GET.get('pgtUrl', None) + if ticket_string.startswith('PT-'): + return _cas2_error_response(INVALID_TICKET, "serviceValidate cannot verify proxy tickets") + else: + return ticket_validate(service, ticket_string, pgtUrl) -def _cas2_error_response(code): +def proxy_validate(request): + """Validate ticket via CAS v.2 protocol""" + service = request.GET.get('service', None) + ticket_string = request.GET.get('ticket', None) + pgtUrl = request.GET.get('pgtUrl', None) + return ticket_validate(service, ticket_string, pgtUrl) + +def generate_proxy_granting_ticket(pgt_url, ticket): + proxy_callback_good_status = (200, 202, 301, 302, 304) + uri = list(urlparse.urlsplit(pgt_url)) + + pgt = ProxyGrantingTicket() + pgt.serviceTicket = ticket + + if hasattr(ticket, 'proxyGrantingTicket'): + # here we got a proxy ticket! tata! + pgt.pgt = ticket.proxyGrantingTicket + + params = {'pgtId': pgt.ticket, 'pgtIou': pgt.pgtiou} + + query = dict(urlparse.parse_qsl(uri[4])) + query.update(params) + + uri[4] = urlencode(query) + + + try: + response = urllib2.urlopen(urlparse.urlunsplit(uri)) + except urllib2.HTTPError, e: + if not e.code in proxy_callback_good_status: + return + except urllib2.URLError, e: + return + + pgt.save() + return pgt + + +def _cas2_proxy_success(pt): + return HttpResponse(proxy_success(pt)) + +def _cas2_sucess_response(user, pgt = None, proxies = None): + return HttpResponse(auth_success_response(user, pgt, proxies), mimetype='text/xml') + +def _cas2_error_response(code, message = None): return HttpResponse(u'''' %(message)s ''' % { 'code': code, - 'message': dict(ERROR_MESSAGES).get(code) + 'message': message if message else dict(ERROR_MESSAGES).get(code) }, mimetype='text/xml') +def proxy_success(pt): + response = etree.Element(CAS + 'serviceResponse', nsmap=NSMAP) + proxySuccess = etree.SubElement(response, CAS + 'proxySuccess') + proxyTicket = etree.SubElement(proxySuccess, CAS + 'proxyTicket') + proxyTicket.text = pt + return unicode(etree.tostring(response, encoding='utf-8'), 'utf-8') -def _cas2_sucess_response(user): - return HttpResponse(auth_success_response(user), mimetype='text/xml') - +def auth_success_response(user, pgt, proxies): -def auth_success_response(user): - from attribute_formatters import CAS, NSMAP, etree response = etree.Element(CAS + 'serviceResponse', nsmap=NSMAP) auth_success = etree.SubElement(response, CAS + 'authenticationSuccess') @@ -143,4 +228,18 @@ def auth_success_response(user): if len(attrs) > 0: formater = get_callable(settings.CAS_CUSTOM_ATTRIBUTES_FORMATER) formater(auth_success, attrs) + + + if pgt: + pgtElement = etree.SubElement(auth_success, CAS + 'proxyGrantingTicket') + pgtElement.text = pgt + + if proxies: + proxiesElement = etree.SubElement(auth_success , CAS + "proxies") + for proxy in proxies: + proxyElement = etree.SubElement(proxiesElement, CAS + "proxy") + proxyElement.text = proxy + + + return unicode(etree.tostring(response, encoding='utf-8'), 'utf-8') diff --git a/setup.py b/setup.py index 06e5d4a..bf85f5d 100644 --- a/setup.py +++ b/setup.py @@ -12,4 +12,9 @@ setup( include_package_data=True, zip_safe=False, install_requires=['setuptools'], + classifiers = [ + "Development Status :: 3 - Alpha", + "Framework :: Django", + "License :: OSI Approved :: MIT License", + ] ) -- 2.20.1