aboutsummaryrefslogtreecommitdiffstats
path: root/lib/python2.7/site-packages/Twisted-12.2.0-py2.7-linux-x86_64.egg/twisted/words/protocols/jabber/xmpp_stringprep.py
blob: 752741218aa40740489bc61c0b8bf551aedf3e22 (plain)
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
# -*- test-case-name: twisted.words.test.test_jabberxmppstringprep -*-
#
# Copyright (c) Twisted Matrix Laboratories.
# See LICENSE for details.

import sys, warnings
from zope.interface import Interface, implements

if sys.version_info < (2,3,2):
    import re

    class IDNA:
        dots = re.compile(u"[\u002E\u3002\uFF0E\uFF61]")
        def nameprep(self, label):
            return label.lower()

    idna = IDNA()

    crippled = True

    warnings.warn("Accented and non-Western Jabber IDs will not be properly "
                  "case-folded with this version of Python, resulting in "
                  "incorrect protocol-level behavior.  It is strongly "
                  "recommended you upgrade to Python 2.3.2 or newer if you "
                  "intend to use Twisted's Jabber support.")

else:
    import stringprep
    # We require Unicode version 3.2. Python 2.5 and later provide this as
    # a separate object. Before that the unicodedata module uses 3.2.
    try:
        from unicodedata import ucd_3_2_0 as unicodedata
    except:
        import unicodedata
    from encodings import idna

    crippled = False

del sys, warnings

class ILookupTable(Interface):
    """ Interface for character lookup classes. """

    def lookup(c):
        """ Return whether character is in this table. """

class IMappingTable(Interface):
    """ Interface for character mapping classes. """

    def map(c):
        """ Return mapping for character. """

class LookupTableFromFunction:

    implements(ILookupTable)

    def __init__(self, in_table_function):
        self.lookup = in_table_function

class LookupTable:

    implements(ILookupTable)

    def __init__(self, table):
        self._table = table

    def lookup(self, c):
        return c in self._table

class MappingTableFromFunction:

    implements(IMappingTable)

    def __init__(self, map_table_function):
        self.map = map_table_function

class EmptyMappingTable:

    implements(IMappingTable)

    def __init__(self, in_table_function):
        self._in_table_function = in_table_function

    def map(self, c):
        if self._in_table_function(c):
            return None
        else:
            return c

class Profile:
    def __init__(self, mappings=[],  normalize=True, prohibiteds=[],
                       check_unassigneds=True, check_bidi=True):
        self.mappings = mappings
        self.normalize = normalize
        self.prohibiteds = prohibiteds
        self.do_check_unassigneds = check_unassigneds
        self.do_check_bidi = check_bidi

    def prepare(self, string):
        result = self.map(string)
        if self.normalize:
            result = unicodedata.normalize("NFKC", result)
        self.check_prohibiteds(result)
        if self.do_check_unassigneds:
            self.check_unassigneds(result)
        if self.do_check_bidi:
            self.check_bidirectionals(result)
        return result

    def map(self, string):
        result = []

        for c in string:
            result_c = c

            for mapping in self.mappings:
                result_c = mapping.map(c)
                if result_c != c:
                    break

            if result_c is not None:
                result.append(result_c)

        return u"".join(result)

    def check_prohibiteds(self, string):
        for c in string:
            for table in self.prohibiteds:
                if table.lookup(c):
                    raise UnicodeError, "Invalid character %s" % repr(c)

    def check_unassigneds(self, string):
        for c in string:
            if stringprep.in_table_a1(c):
                raise UnicodeError, "Unassigned code point %s" % repr(c)

    def check_bidirectionals(self, string):
        found_LCat = False
        found_RandALCat = False

        for c in string:
            if stringprep.in_table_d1(c):
                found_RandALCat = True
            if stringprep.in_table_d2(c):
                found_LCat = True

        if found_LCat and found_RandALCat:
            raise UnicodeError, "Violation of BIDI Requirement 2"

        if found_RandALCat and not (stringprep.in_table_d1(string[0]) and
                                    stringprep.in_table_d1(string[-1])):
            raise UnicodeError, "Violation of BIDI Requirement 3"


class NamePrep:
    """ Implements preparation of internationalized domain names.

    This class implements preparing internationalized domain names using the
    rules defined in RFC 3491, section 4 (Conversion operations).

    We do not perform step 4 since we deal with unicode representations of
    domain names and do not convert from or to ASCII representations using
    punycode encoding. When such a conversion is needed, the C{idna} standard
    library provides the C{ToUnicode()} and C{ToASCII()} functions. Note that
    C{idna} itself assumes UseSTD3ASCIIRules to be false.

    The following steps are performed by C{prepare()}:

      - Split the domain name in labels at the dots (RFC 3490, 3.1)
      - Apply nameprep proper on each label (RFC 3491)
      - Enforce the restrictions on ASCII characters in host names by
        assuming STD3ASCIIRules to be true. (STD 3)
      - Rejoin the labels using the label separator U+002E (full stop).

    """

    # Prohibited characters.
    prohibiteds = [unichr(n) for n in range(0x00, 0x2c + 1) +
                                       range(0x2e, 0x2f + 1) +
                                       range(0x3a, 0x40 + 1) +
                                       range(0x5b, 0x60 + 1) +
                                       range(0x7b, 0x7f + 1) ]

    def prepare(self, string):
        result = []

        labels = idna.dots.split(string)

        if labels and len(labels[-1]) == 0:
            trailing_dot = '.'
            del labels[-1]
        else:
            trailing_dot = ''

        for label in labels:
            result.append(self.nameprep(label))

        return ".".join(result) + trailing_dot

    def check_prohibiteds(self, string):
        for c in string:
           if c in self.prohibiteds:
               raise UnicodeError, "Invalid character %s" % repr(c)

    def nameprep(self, label):
        label = idna.nameprep(label)
        self.check_prohibiteds(label)
        if label[0] == '-':
            raise UnicodeError, "Invalid leading hyphen-minus"
        if label[-1] == '-':
            raise UnicodeError, "Invalid trailing hyphen-minus"
        return label

if crippled:
    case_map = MappingTableFromFunction(lambda c: c.lower())
    nodeprep = Profile(mappings=[case_map],
                       normalize=False,
                       prohibiteds=[LookupTable([u' ', u'"', u'&', u"'", u'/',
                                                 u':', u'<', u'>', u'@'])],
                       check_unassigneds=False,
                       check_bidi=False)

    resourceprep = Profile(normalize=False,
                           check_unassigneds=False,
                           check_bidi=False)

else:
    C_11 = LookupTableFromFunction(stringprep.in_table_c11)
    C_12 = LookupTableFromFunction(stringprep.in_table_c12)
    C_21 = LookupTableFromFunction(stringprep.in_table_c21)
    C_22 = LookupTableFromFunction(stringprep.in_table_c22)
    C_3 = LookupTableFromFunction(stringprep.in_table_c3)
    C_4 = LookupTableFromFunction(stringprep.in_table_c4)
    C_5 = LookupTableFromFunction(stringprep.in_table_c5)
    C_6 = LookupTableFromFunction(stringprep.in_table_c6)
    C_7 = LookupTableFromFunction(stringprep.in_table_c7)
    C_8 = LookupTableFromFunction(stringprep.in_table_c8)
    C_9 = LookupTableFromFunction(stringprep.in_table_c9)

    B_1 = EmptyMappingTable(stringprep.in_table_b1)
    B_2 = MappingTableFromFunction(stringprep.map_table_b2)

    nodeprep = Profile(mappings=[B_1, B_2],
                       prohibiteds=[C_11, C_12, C_21, C_22,
                                    C_3, C_4, C_5, C_6, C_7, C_8, C_9,
                                    LookupTable([u'"', u'&', u"'", u'/',
                                                 u':', u'<', u'>', u'@'])])

    resourceprep = Profile(mappings=[B_1,],
                           prohibiteds=[C_12, C_21, C_22,
                                        C_3, C_4, C_5, C_6, C_7, C_8, C_9])

nameprep = NamePrep()