merged some code form issue 1534 Session ID not changed after privilege level change, thanks kablubr

This commit is contained in:
mdipierro
2013-06-16 07:20:54 -05:00
parent 9b9e64e9ef
commit d04f2ff502
3 changed files with 111 additions and 1 deletions

View File

@@ -1 +1 @@
Version 2.5.1-stable+timestamp.2013.06.16.07.06.28
Version 2.5.1-stable+timestamp.2013.06.16.07.20.03

View File

@@ -490,6 +490,108 @@ class Session(Storage):
"""
defines the session object and the default values of its members (None)
"""
def renew(
self,
request=None,
response=None,
db=None,
tablename='web2py_session',
masterapp=None,
clear_session=False
):
return
if request is None:
request = current.request
if response is None:
response = current.response
#check if session is separate
separate = None
if response.session_id[2:3] == "/":
separate = lambda session_name: session_name[-2:]
self._unlock(response)
if not masterapp:
masterapp = request.application
# Load session data from cookie
cookies = request.cookies
# check if there is a session_id in cookies
if response.session_id_name in cookies:
response.session_id = \
cookies[response.session_id_name].value
else:
response.session_id = None
# if the session goes in file
if response.session_storage_type == 'file':
if global_settings.db_sessions is True \
or masterapp in global_settings.db_sessions:
return
client = request.client and request.client.replace(':', '.')
uuid = web2py_uuid()
response.session_id = '%s-%s' % (client, uuid)
if separate:
prefix = separate(response.session_id)
response.session_id = '%s/%s' % \
(prefix, response.session_id)
response.session_filename = \
os.path.join(up(request.folder), masterapp,
'sessions', response.session_id)
response.session_new = True
# else the session goes in db
elif response.session_storage_type == 'db':
# verify if tablename was set or used in connect
if response.session_table_name and tablename == 'web2py_session':
tablename = response.session_table_name
if global_settings.db_sessions is not True:
global_settings.db_sessions.add(masterapp)
if response.session_file:
self._close(response)
if settings.global_settings.web2py_runtime_gae:
# in principle this could work without GAE
request.tickets_db = db
tname = tablename + '_' + masterapp
if not db:
raise Exception('No database parameter passed: "db=database"')
table = db.get(tname, None)
if table is None:
raise Exception('No session to renew')
# Get session data out of the database
(record_id, unique_key) = response.session_id.split(':')
if record_id == '0':
raise Exception('record_id == 0')
# Select from database
row = db(table.id == record_id).select().first()
# Make sure the session data exists in the database
if not row or row.unique_key != unique_key:
raise Exception('No record')
unique_key = web2py_uuid()
row.update_record(unique_key=unique_key)
response.session_id = '%s:%s' % (record_id, unique_key)
response.session_db_table = table
response.session_db_record_id = record_id
response.session_db_unique_key = unique_key
rcookies = response.cookies
rcookies[response.session_id_name] = response.session_id
rcookies[response.session_id_name]['path'] = '/'
if clear_session:
self.clear()
def connect(
self,
@@ -653,6 +755,8 @@ class Session(Storage):
response.session_db_table = table
response.session_db_record_id = record_id
response.session_db_unique_key = unique_key
# keep tablename parameter for use in session renew
response.session_table_name = tablename
rcookies = response.cookies
rcookies[response.session_id_name] = response.session_id
rcookies[response.session_id_name]['path'] = '/'

View File

@@ -920,6 +920,8 @@ class Auth(object):
update_fields = ['email'],
ondelete="CASCADE",
client_side = True,
keep_session_onlogin=True,
keep_session_onlogout=False,
wiki = Settings(),
)
# ## these are messages that can be customized
@@ -1850,6 +1852,8 @@ class Auth(object):
for key,value in user.items():
if callable(value) or key=='password':
delattr(user,key)
current.session.renew(
clear_session=not self.settings.keep_session_onlogin, db=self.db)
current.session.auth = Storage(
user = user,
last_visit=current.request.now,
@@ -2236,6 +2240,8 @@ class Auth(object):
current.session.auth = None
current.session.flash = self.messages.logged_out
current.session.renew(
clear_session=not self.settings.keep_session_onlogout, db=self.db)
if not next is None:
redirect(next)