1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
|
# Copyright (C) 2015 Ipsilon project Contributors, for license see COPYING
from ipsilon.util.log import Log
from ipsilon.util.data import SAML2SessionStore
LOGGED_IN = 1
INIT_LOGOUT = 2
LOGGING_OUT = 4
LOGGED_OUT = 8
class SAMLSession(Log):
"""
A SAML login session.
uuidval - Unique ID stored in the database
session_id - ID of the login session
provider_id - ID of the SP
user - the login name of the user that owns the session
login_session - the Login session object
logoutstate - an integer constant representing where in the
logout process this request is
relaystate - where the user will be redirected when logout is
complete
request_id - the logout request ID if initiated from IdP. The
logout response will include an InResponseTo value
which matches this.
logout_request - the Logout request object
"""
def __init__(self, uuidval, session_id, provider_id, user,
login_session, logoutstate=None, relaystate=None,
logout_request=None, request_id=None):
self.uuidval = uuidval
self.session_id = session_id
self.provider_id = provider_id
self.user = user
self.login_session = login_session
self.logoutstate = logoutstate
self.relaystate = relaystate
self.request_id = request_id
self.logout_request = logout_request
def set_logoutstate(self, relaystate=None, request=None, request_id=None):
"""
Update attributes needed to determine the state of the session for
logout.
The database is not updated when these are set. It is expected that
this is called prior to start_logout()
"""
if relaystate:
self.relaystate = relaystate
if request:
self.logout_request = request
if request_id:
self.request_id = request_id
def dump(self):
self.debug('session_id %s' % self.session_id)
self.debug('provider_id %s' % self.provider_id)
self.debug('login session %s' % self.login_session)
self.debug('logoutstate %s' % self.logoutstate)
def convert(self):
"""
Convert this object into something suitable to store in the
data backend.
"""
data = dict()
data['session_id'] = self.session_id
data['provider_id'] = self.provider_id
data['user'] = self.user
data['login_session'] = self.login_session
data['logoutstate'] = self.logoutstate
data['relaystate'] = self.relaystate
data['logout_request'] = self.logout_request
data['request_id'] = self.request_id
return {self.uuidval: data}
class SAMLSessionFactory(Log):
"""
Access SAML session information.
The sessions are stored via the data backend.
When a user logs in, add_session() is called and a new SAMLSession
created and added to the table.
When a user logs out, the next login session is found and moved to
sessions_logging_out. remove_session() will look in both when trying
to remove a session.
Returns a SAMLSession object representing the new session.
"""
def __init__(self):
self._ss = SAML2SessionStore()
self.user = None
def _data_to_samlsession(self, uuidval, data):
"""
Convert data from the data backend to a SAMLSession object.
"""
return SAMLSession(uuidval,
data.get('session_id'),
data.get('provider_id'),
data.get('user'),
data.get('login_session'),
data.get('logoutstate'),
data.get('relaystate'),
data.get('logout_request'),
data.get('request_id'))
def add_session(self, session_id, provider_id, user, login_session,
request_id=None):
"""
Add a new login session to the table.
"""
self.user = user
data = {'session_id': session_id,
'provider_id': provider_id,
'user': user,
'login_session': login_session,
'logoutstate': LOGGED_IN}
if request_id:
data['request_id'] = request_id
uuidval = self._ss.new_session(data)
return SAMLSession(uuidval, session_id, provider_id, user,
login_session, LOGGED_IN,
request_id=request_id)
def get_session_by_id(self, session_id):
"""
Retrieve a session by session ID
"""
uuidval, data = self._ss.get_session(session_id=session_id)
if uuidval is None:
return None
return self._data_to_samlsession(uuidval, data)
def get_session_id_by_provider_id(self, provider_id):
"""
Return a tuple of logged-in session IDs by provider_id
"""
candidates = self._ss.get_user_sessions(self.user)
session_ids = []
for c in candidates:
key = c.keys()[0]
if c[key].get('provider_id') == provider_id:
samlsession = self._data_to_samlsession(key, c[key])
session_ids.append(samlsession.session_id.encode('utf-8'))
return tuple(session_ids)
def get_session_by_request_id(self, request_id):
"""
Retrieve a session by logout request ID
"""
uuidval, data = self._ss.get_session(request_id=request_id)
if uuidval is None:
return None
return self._data_to_samlsession(uuidval, data)
def remove_session(self, samlsession):
return self._ss.remove_session(samlsession.uuidval)
def remove_session_by_session_id(self, session_id):
session = self.get_session_by_id(session_id)
return self._ss.remove_session(session.uuidval)
def start_logout(self, samlsession, relaystate=None, initial=True):
"""
Move a session into the logging_out state
samlsession: the SAMLSession object to start logging out
relaystate: URL to redirect user to when logout is completed
initial: boolean to indicate if this session started logout.
Only the initial session's relaystate is used.
No return value
"""
if initial:
samlsession.logoutstate = INIT_LOGOUT
else:
samlsession.logoutstate = LOGGING_OUT
if relaystate:
samlsession.relaystate = relaystate
datum = samlsession.convert()
self._ss.update_session(datum)
def get_next_logout(self, peek=False):
"""
Get the next session in the logged-in state and move
it to the logging_out state. Return the session that is
found.
:param peek: for IdP-initiated logout we can't remove the
session otherwise when the request comes back
in the user won't be seen as being logged-on.
Return None if no more sessions in LOGGED_IN state.
"""
candidates = self._ss.get_user_sessions(self.user)
for c in candidates:
key = c.keys()[0]
if int(c[key].get('logoutstate', 0)) == LOGGED_IN:
samlsession = self._data_to_samlsession(key, c[key])
self.start_logout(samlsession, initial=False)
return samlsession
return None
def get_initial_logout(self):
"""
Get the initial logout request.
Return None if no sessions in INIT_LOGOUT state.
"""
candidates = self._ss.get_user_sessions(self.user)
# FIXME: what does it mean if there are multiple in init? We
# just return the first one for now. How do we know
# it's the "right" one if multiple logouts are started
# at the same time from different SPs?
for c in candidates:
key = c.keys()[0]
if int(c[key].get('logoutstate', 0)) == INIT_LOGOUT:
samlsession = self._data_to_samlsession(key, c[key])
return samlsession
return None
def wipe_data(self):
self._ss.wipe_data()
def dump(self):
"""
Dump all sessions to debug log
"""
candidates = self._ss.get_user_sessions(self.user)
count = 0
for c in candidates:
key = c.keys()[0]
samlsession = self._data_to_samlsession(key, c[key])
self.debug('session %d: %s' % (count, samlsession.convert()))
count += 1
if __name__ == '__main__':
import cherrypy
provider1 = "http://127.0.0.10/saml2"
provider2 = "http://127.0.0.11/saml2"
# temporary database location for testing
cherrypy.config['saml2.sessions.db'] = '/tmp/saml2sessions.sqlite'
factory = SAMLSessionFactory()
factory.wipe_data()
sess1 = factory.add_session('_123456', provider1, "admin", "<Login/>")
sess2 = factory.add_session('_789012', provider2, "testuser", "<Login/>")
# Test finding sessions by provider
ids = factory.get_session_id_by_provider_id(provider2)
assert(len(ids) == 1)
sess3 = factory.add_session('_345678', provider2, "testuser", "<Login/>")
ids = factory.get_session_id_by_provider_id(provider2)
assert(len(ids) == 2)
# Test finding sessions by session ID
test1 = factory.get_session_by_id('_123456')
assert(test1.user == 'admin')
assert(test1.provider_id == provider1)
# Log out and remove the first session
test1.set_logoutstate('http://www.example.com/idp')
factory.start_logout(test1, initial=True)
test1 = factory.get_session_by_id('_123456')
assert(test1.relaystate == 'http://www.example.com/idp')
factory.remove_session_by_session_id('_123456')
# Make sure it is gone from the db
test1 = factory.get_session_by_id('_123456')
assert(test1 is None)
test2 = factory.get_session_by_id('_789012')
factory.start_logout(test2, initial=True)
test3 = factory.get_next_logout()
assert(test3.session_id == '_345678')
test4 = factory.get_initial_logout()
assert(test4.session_id == '_789012')
# Even though we've started logout, make sure we can still find
# all sessions for a provider.
ids = factory.get_session_id_by_provider_id(provider2)
assert(len(ids) == 2)
|