ipsilon/providers/openid/store.py

   1 #!/usr/bin/python
   2 #
   3 # Copyright (C) 2014  Ipsilon project Contributors, for licensee see COPYING
   4
   5 from ipsilon.util.data import Store
   6
   7 from openid import oidutil
   8 from openid.association import Association
   9 from openid.store.nonce import SKEW as NonceSKEW
  10 from openid.store.interface import OpenIDStore as OpenIDStoreInterface
  11
  12 from time import time
  13
  14
  15 class OpenIDStore(Store, OpenIDStoreInterface):
  16     def __init__(self, database_url):
  17         Store.__init__(self, database_url=database_url)
  18
  19     def storeAssociation(self, server_url, assoc):
  20         iden = '%s-%s' % (server_url, assoc.handle)
  21         datum = {'secret': oidutil.toBase64(assoc.secret),
  22                  'issued': str(assoc.issued),
  23                  'lifetime': str(assoc.lifetime),
  24                  'assoc_type': assoc.assoc_type}
  25
  26         data = {iden: datum}
  27         self.save_unique_data('association', data)
  28
  29     def getAssociation(self, server_url, handle=None):
  30         iden = '%s-%s' % (server_url, handle)
  31         data = self.get_unique_data('association', iden)
  32
  33         if len(data) < 1:
  34             return None
  35
  36         datum = data[iden]
  37         assoc = Association(handle,
  38                             oidutil.fromBase64(datum['secret']),
  39                             int(datum['issued']),
  40                             int(datum['lifetime']),
  41                             datum['assoc_type'])
  42
  43         if assoc.expiresIn == 0:
  44             self.del_unique_data('association', iden)
  45             return None
  46
  47         return assoc
  48
  49     def removeAssociation(self, server_url, handle):
  50         iden = '%s-%s' % (server_url, handle)
  51         self.del_unique_data('association', iden)
  52
  53     def useNonce(self, server_url, timestamp, salt):
  54         if abs(timestamp - time()) > NonceSKEW:
  55             return False
  56
  57         iden = '%s-%s-%s' % (server_url, timestamp, salt)
  58         data = self.get_unique_data('nonce', iden)
  59
  60         if len(data) > 0:
  61             # This server_url, timestamp, salt combination is already seen
  62             return False
  63
  64         datum = {'timestamp': timestamp}
  65         data = {iden: datum}
  66         self.save_unique_data('nonce', data)
  67
  68         return True
  69
  70     def cleanupNonces(self):
  71         nonces = self.get_unique_data('nonce')
  72         for iden in nonces:
  73             if nonces[iden]['timestamp'] < (time() - NonceSKEW):
  74                 self.del_unique_data('nonce', iden)
  75
  76     def cleanupAssociations(self):
  77         assocs = self.get_unique_data('association')
  78         for iden in assocs:
  79             if ((int(assocs[iden]['issued']) + int(assocs[iden]['lifetime']))
  80                     < time()):
  81                 self.del_unique_data('association', iden)