Synnefo

from traceback import format_exc

from time import time, mktime

from django.conf import settings

from django.http import HttpResponse

from django.utils import simplejson as json

from pithos.im.faults import BadRequest, Unauthorized, ServiceUnavailable

from pithos.im.models import User

import datetime

def render_fault(request, fault):

    if settings.DEBUG or settings.TEST:

        fault.details = format_exc(fault)

    request.serialization = 'text'

    data = '\n'.join((fault.message, fault.details)) + '\n'

    response = HttpResponse(data, status=fault.code)

    return response

def update_response_headers(response):

    response['Content-Type'] = 'application/json; charset=UTF-8'

    response['Content-Length'] = len(response.content)

def authenticate(request):

    # Normal Response Codes: 204

    # Error Response Codes: serviceUnavailable (503)

    #                       badRequest (400)

    #                       unauthorised (401)

    try:

        if request.method != 'GET':

            raise BadRequest('Method not allowed.')

        x_auth_token = request.META.get('HTTP_X_AUTH_TOKEN')

        if not x_auth_token:

            return render_fault(request, BadRequest('Missing X-Auth-Token'))

        try:

            user = User.objects.get(auth_token=x_auth_token)

        except User.DoesNotExist, e:

            return render_fault(request, Unauthorized('Invalid X-Auth-Token')) 

        # Check if the is active.

        if user.state != 'ACTIVE':

            return render_fault(request, Unauthorized('User inactive'))

        # Check if the token has expired.

        if (time() - mktime(user.auth_token_expires.timetuple())) > 0:

            return render_fault(request, Unauthorized('Authentication expired'))

        response = HttpResponse()

        response.status=204

        user_info = user.__dict__

        for k,v in user_info.items():

            if isinstance(v,  datetime.datetime):

                user_info[k] = v.strftime('%a, %d-%b-%Y %H:%M:%S %Z')

        user_info.pop('_state')

        response.content = json.dumps(user_info)

        update_response_headers(response)

        return response

    except BaseException, e:

        fault = ServiceUnavailable('Unexpected error')

        return render_fault(request, fault)

# Copyright 2011 GRNET S.A. All rights reserved.

# 

# Redistribution and use in source and binary forms, with or

# without modification, are permitted provided that the following

# conditions are met:

# 

#   1. Redistributions of source code must retain the above

#      copyright notice, this list of conditions and the following

#      disclaimer.

# 

#   2. Redistributions in binary form must reproduce the above

#      copyright notice, this list of conditions and the following

#      disclaimer in the documentation and/or other materials

#      provided with the distribution.

# 

# THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS

# OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED

# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR

# PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR

# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,

# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT

# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF

# USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED

# AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT

# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN

# ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE

# POSSIBILITY OF SUCH DAMAGE.

# 

# The views and conclusions contained in the software and

# documentation are those of the authors and should not be

# interpreted as representing official policies, either expressed

# or implied, of GRNET S.A.

def camelCase(s):

    return s[0].lower() + s[1:]

class Fault(Exception):

    def __init__(self, message='', details='', name=''):

        Exception.__init__(self, message, details, name)

        self.message = message

        self.details = details

        self.name = name or camelCase(self.__class__.__name__)

class BadRequest(Fault):

    code = 400

class Unauthorized(Fault):

    code = 401

class ServiceUnavailable(Fault):

    code = 503

urlpatterns += patterns('pithos.im.api',

    (r'^authenticate/?$', 'authenticate')

)

def _handle_response(response, verbose=False, debug=False):

    return response.status, headers, data

def authenticate(authentication_host, token):

    con = HTTPConnection(authentication_host)

    kwargs = {}

    kwargs['headers'] = {}

    kwargs['headers']['X-Auth-Token'] = token

    kwargs['headers']['Content-Length'] = 0

    path = '/im/authenticate'

    con.request('GET', path, **kwargs)

    response = con.getresponse()

    return _handle_response(response)

from django.conf import settings

from django.utils import simplejson as json

from pithos.lib.client import authenticate, Fault

    if not token:

        return None

    host = settings.AUTHENTICATION_HOST

        status, headers, user = authenticate(host, token)

        return json.loads(user)

    except Fault, f:

        try:

            host, port = settings.AUTHENTICATION_HOST.split(':')

            if request.META['SERVER_NAME'] and request.META['SERVER_PORT'] == port:

                #bypass authentication

                #if it is an authentication request

                return

            request.user = None

            request.user_uniq = None

            # Try to find token in a parameter, in a request header, or in a cookie.

            user = get_user_from_token(request.GET.get('X-Auth-Token'))

            if not user:

                user = get_user_from_token(request.META.get('HTTP_X_AUTH_TOKEN'))

                # Back from an im login target.

                if request.GET.get('user', None):

                    token = request.GET.get('token', None)

                    if token:

                        request.set_auth_cookie = True

                if not user:

                    cookie_value = unquote(request.COOKIES.get('_pithos2_a', ''))

                    if cookie_value and '|' in cookie_value:

                        token = cookie_value.split('|', 1)[1]

                        user = get_user_from_token(token)

            if not user:

                return

            request.user = user

            request.user_uniq = user['uniq']

        except Exception, e:

            print e

#where astakos is hosted

AUTHENTICATION_HOST = '127.0.0.1:10000'