893a877b76
There have been some issues in Python 3 where elements of the id_token were left when encoding the token. Cause was incorrect encoding logic in pyjwkest. Version 1.0.3 has improved encoding handling.
98 lines
2.4 KiB
Python
98 lines
2.4 KiB
Python
from datetime import timedelta
|
|
import time
|
|
import uuid
|
|
|
|
from Crypto.PublicKey.RSA import importKey
|
|
from django.utils import timezone
|
|
from hashlib import md5
|
|
from jwkest.jwk import RSAKey
|
|
from jwkest.jws import JWS
|
|
|
|
from oidc_provider.lib.utils.common import get_issuer, get_rsa_key
|
|
from oidc_provider.models import *
|
|
from oidc_provider import settings
|
|
|
|
|
|
def create_id_token(user, aud, nonce):
|
|
"""
|
|
Receives a user object and aud (audience).
|
|
Then creates the id_token dictionary.
|
|
See: http://openid.net/specs/openid-connect-core-1_0.html#IDToken
|
|
|
|
Return a dic.
|
|
"""
|
|
sub = settings.get('OIDC_IDTOKEN_SUB_GENERATOR')(user=user)
|
|
|
|
expires_in = settings.get('OIDC_IDTOKEN_EXPIRE')
|
|
|
|
# Convert datetimes into timestamps.
|
|
now = timezone.now()
|
|
iat_time = int(time.mktime(now.timetuple()))
|
|
exp_time = int(time.mktime((now + timedelta(seconds=expires_in)).timetuple()))
|
|
user_auth_time = user.last_login or user.date_joined
|
|
auth_time = int(time.mktime(user_auth_time.timetuple()))
|
|
|
|
dic = {
|
|
'iss': get_issuer(),
|
|
'sub': sub,
|
|
'aud': str(aud),
|
|
'exp': exp_time,
|
|
'iat': iat_time,
|
|
'auth_time': auth_time,
|
|
}
|
|
|
|
if nonce:
|
|
dic['nonce'] = str(nonce)
|
|
|
|
return dic
|
|
|
|
|
|
def encode_id_token(payload):
|
|
"""
|
|
Represent the ID Token as a JSON Web Token (JWT).
|
|
|
|
Return a hash.
|
|
"""
|
|
key_string = get_rsa_key().encode('utf-8')
|
|
keys = [ RSAKey(key=importKey(key_string), kid=md5(key_string).hexdigest()) ]
|
|
_jws = JWS(payload, alg='RS256')
|
|
return _jws.sign_compact(keys)
|
|
|
|
|
|
def create_token(user, client, id_token_dic, scope):
|
|
"""
|
|
Create and populate a Token object.
|
|
|
|
Return a Token object.
|
|
"""
|
|
token = Token()
|
|
token.user = user
|
|
token.client = client
|
|
token.access_token = uuid.uuid4().hex
|
|
|
|
token.id_token = id_token_dic
|
|
|
|
token.refresh_token = uuid.uuid4().hex
|
|
token.expires_at = timezone.now() + timedelta(
|
|
seconds=settings.get('OIDC_TOKEN_EXPIRE'))
|
|
token.scope = scope
|
|
|
|
return token
|
|
|
|
|
|
def create_code(user, client, scope, nonce):
|
|
"""
|
|
Create and populate a Code object.
|
|
|
|
Return a Code object.
|
|
"""
|
|
code = Code()
|
|
code.user = user
|
|
code.client = client
|
|
code.code = uuid.uuid4().hex
|
|
code.expires_at = timezone.now() + timedelta(
|
|
seconds=settings.get('OIDC_CODE_EXPIRE'))
|
|
code.scope = scope
|
|
code.nonce = nonce
|
|
|
|
return code
|