mirror of
https://github.com/Mikaela/Limnoria.git
synced 2024-11-26 20:59:27 +01:00
ircdb: Deduplicate hostmasks in IrcUser.addAuth
If addAuth is called with a hostmask that already exists (exactly, not just equivalent patterns) in self.auth, remove the entry for the old authentication. This essentially updates the timeout for the hostmask. Signed-off-by: James McCoy <jamessan@users.sourceforge.net>
This commit is contained in:
parent
4bc549c1ed
commit
6361b1e856
15
src/ircdb.py
15
src/ircdb.py
@ -1,6 +1,6 @@
|
|||||||
###
|
###
|
||||||
# Copyright (c) 2002-2009, Jeremiah Fincher
|
# Copyright (c) 2002-2009, Jeremiah Fincher
|
||||||
# Copyright (c) 2009, James McCoy
|
# Copyright (c) 2009,2013, James McCoy
|
||||||
# All rights reserved.
|
# All rights reserved.
|
||||||
#
|
#
|
||||||
# Redistribution and use in source and binary forms, with or without
|
# Redistribution and use in source and binary forms, with or without
|
||||||
@ -291,9 +291,20 @@ class IrcUser(object):
|
|||||||
self.hostmasks.remove(hostmask)
|
self.hostmasks.remove(hostmask)
|
||||||
|
|
||||||
def addAuth(self, hostmask):
|
def addAuth(self, hostmask):
|
||||||
"""Sets a user's authenticated hostmask. This times out in 1 hour."""
|
"""Sets a user's authenticated hostmask. This times out according to
|
||||||
|
conf.supybot.timeoutIdentification. If hostmask exactly matches an
|
||||||
|
existing, known hostmask, the previous entry is removed."""
|
||||||
if self.checkHostmask(hostmask, useAuth=False) or not self.secure:
|
if self.checkHostmask(hostmask, useAuth=False) or not self.secure:
|
||||||
self.auth.append((time.time(), hostmask))
|
self.auth.append((time.time(), hostmask))
|
||||||
|
knownHostmasks = set()
|
||||||
|
def uniqueHostmask(auth):
|
||||||
|
(_, mask) = auth
|
||||||
|
if mask not in knownHostmasks:
|
||||||
|
knownHostmasks.add(mask)
|
||||||
|
return True
|
||||||
|
return False
|
||||||
|
uniqued = filter(uniqueHostmask, reversed(self.auth))
|
||||||
|
self.auth = list(reversed(uniqued))
|
||||||
else:
|
else:
|
||||||
raise ValueError, 'secure flag set, unmatched hostmask'
|
raise ValueError, 'secure flag set, unmatched hostmask'
|
||||||
|
|
||||||
|
@ -260,6 +260,9 @@ class IrcUserTestCase(IrcdbTestCase):
|
|||||||
u = ircdb.IrcUser()
|
u = ircdb.IrcUser()
|
||||||
u.addAuth('foo!bar@baz')
|
u.addAuth('foo!bar@baz')
|
||||||
self.failUnless(u.checkHostmask('foo!bar@baz'))
|
self.failUnless(u.checkHostmask('foo!bar@baz'))
|
||||||
|
u.addAuth('foo!bar@baz')
|
||||||
|
self.failUnless(u.checkHostmask('foo!bar@baz'))
|
||||||
|
self.failUnless(len(u.auth) == 1)
|
||||||
u.addAuth('boo!far@fizz')
|
u.addAuth('boo!far@fizz')
|
||||||
self.failUnless(u.checkHostmask('boo!far@fizz'))
|
self.failUnless(u.checkHostmask('boo!far@fizz'))
|
||||||
time.sleep(2.1)
|
time.sleep(2.1)
|
||||||
|
Loading…
Reference in New Issue
Block a user