From f5beac2afafeab1f1a788d1409b61834b124469c Mon Sep 17 00:00:00 2001
From: Lance Stout <lancestout@gmail.com>
Date: Mon, 14 May 2012 23:12:54 -0700
Subject: [PATCH] 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