From f5beac2afafeab1f1a788d1409b61834b124469c Mon Sep 17 00:00:00 2001 From: Lance Stout Date: Mon, 14 May 2012 23:12:54 -0700 Subject: Use SASLPrepFailure as the exception name instead of UnicodeError. --- .../features/feature_mechanisms/mechanisms.py | 3 ++- sleekxmpp/thirdparty/suelta/exceptions.py | 4 +++ sleekxmpp/thirdparty/suelta/saslprep.py | 31 ++++++++++++---------- 3 files changed, 23 insertions(+), 15 deletions(-) diff --git a/sleekxmpp/features/feature_mechanisms/mechanisms.py b/sleekxmpp/features/feature_mechanisms/mechanisms.py index cc3579e8..f47012a9 100644 --- a/sleekxmpp/features/feature_mechanisms/mechanisms.py +++ b/sleekxmpp/features/feature_mechanisms/mechanisms.py @@ -10,6 +10,7 @@ import logging from sleekxmpp.thirdparty import suelta from sleekxmpp.thirdparty.suelta.exceptions import SASLCancelled, SASLError +from sleekxmpp.thirdparty.suelta.exceptions import SASLPrepFailure from sleekxmpp.stanza import StreamFeatures from sleekxmpp.xmlstream import RestartStream, register_stanza_plugin @@ -129,7 +130,7 @@ class FeatureMechanisms(BasePlugin): except SASLError: self.attempted_mechs.add(self.mech.name) self._send_auth() - except UnicodeError as e: + except SASLPrepFailure: log.exception("A credential value did not pass SASLprep.") self.xmpp.disconnect() else: diff --git a/sleekxmpp/thirdparty/suelta/exceptions.py b/sleekxmpp/thirdparty/suelta/exceptions.py index 625cca0e..40d8bad3 100644 --- a/sleekxmpp/thirdparty/suelta/exceptions.py +++ b/sleekxmpp/thirdparty/suelta/exceptions.py @@ -29,3 +29,7 @@ class SASLCancelled(SASLError): :type sasl: `suelta.SASL` """ super(SASLCancelled, self).__init__(sasl, "User cancelled", mech) + + +class SASLPrepFailure(UnicodeError): + pass diff --git a/sleekxmpp/thirdparty/suelta/saslprep.py b/sleekxmpp/thirdparty/suelta/saslprep.py index 8022e1cd..0e72fcb1 100644 --- a/sleekxmpp/thirdparty/suelta/saslprep.py +++ b/sleekxmpp/thirdparty/suelta/saslprep.py @@ -5,6 +5,9 @@ import stringprep import unicodedata +from sleekxmpp.thirdparty.suelta.exceptions import SASLPrepFailure + + def saslprep(text, strict=True): """ Return a processed version of the given string, using the SASLPrep @@ -41,38 +44,38 @@ def saslprep(text, strict=True): if text: first_is_randal = stringprep.in_table_d1(text[0]) if first_is_randal and not stringprep.in_table_d1(text[-1]): - raise UnicodeError('Section 6.3 [end]') + raise SASLPrepFailure('Section 6.3 [end]') # Check for prohibited characters for x in range(len(text)): if strict and stringprep.in_table_a1(text[x]): - raise UnicodeError('Unassigned Codepoint') + raise SASLPrepFailure('Unassigned Codepoint') if stringprep.in_table_c12(text[x]): - raise UnicodeError('In table C.1.2') + raise SASLPrepFailure('In table C.1.2') if stringprep.in_table_c21(text[x]): - raise UnicodeError('In table C.2.1') + raise SASLPrepFailure('In table C.2.1') if stringprep.in_table_c22(text[x]): - raise UnicodeError('In table C.2.2') + raise SASLPrepFailure('In table C.2.2') if stringprep.in_table_c3(text[x]): - raise UnicodeError('In table C.3') + raise SASLPrepFailure('In table C.3') if stringprep.in_table_c4(text[x]): - raise UnicodeError('In table C.4') + raise SASLPrepFailure('In table C.4') if stringprep.in_table_c5(text[x]): - raise UnicodeError('In table C.5') + raise SASLPrepFailure('In table C.5') if stringprep.in_table_c6(text[x]): - raise UnicodeError('In table C.6') + raise SASLPrepFailure('In table C.6') if stringprep.in_table_c7(text[x]): - raise UnicodeError('In table C.7') + raise SASLPrepFailure('In table C.7') if stringprep.in_table_c8(text[x]): - raise UnicodeError('In table C.8') + raise SASLPrepFailure('In table C.8') if stringprep.in_table_c9(text[x]): - raise UnicodeError('In table C.9') + raise SASLPrepFailure('In table C.9') if x: if first_is_randal and stringprep.in_table_d2(text[x]): - raise UnicodeError('Section 6.2') + raise SASLPrepFailure('Section 6.2') if not first_is_randal and \ x != len(text) - 1 and \ stringprep.in_table_d1(text[x]): - raise UnicodeError('Section 6.3') + raise SASLPrepFailure('Section 6.3') return text -- cgit v1.2.3