Store map from sub to state.

Author: rohe

doc/index.rst

@@ -1,3 +1,4 @@
+Katt.smuts-1
 .. oidccli documentation master file, created by
    sphinx-quickstart on Sat Dec 30 17:32:48 2017.
    You can adapt this file completely to your liking, but it should at least

src/oidcservice/oidc/access_token.py

@@ -63,6 +63,8 @@ def update_service_context(self, resp, state='', **kwargs):
             except KeyError:
                 raise ValueError('Invalid nonce value')
 
+            self.store_sub2state(_idt['sub'], state)
+
         if 'expires_in' in resp:
             resp['__expires_at'] = time_sans_frac() + int(
                 resp['expires_in'])

src/oidcservice/oidc/authorization.py

@@ -59,6 +59,8 @@ def update_service_context(self, resp, state='', **kwargs):
             except KeyError:
                 raise ValueError('Missing nonce value')
 
+            self.store_sub2state(_idt['sub'], state)
+
         if 'expires_in' in resp:
             resp['__expires_at'] = time_sans_frac() + int(resp['expires_in'])
         self.store_item(resp.to_json(), 'auth_response', state)