# Note: # This *is* now explicitly RPython. # Please make sure not to break this. """ _codecs -- Provides access to the codec registry and the builtin codecs. This module should never be imported directly. The standard library module "codecs" wraps this builtin module for use within Python. The codec registry is accessible via: register(search_function) -> None lookup(encoding) -> (encoder, decoder, stream_reader, stream_writer) The builtin Unicode codecs use the following interface: _encode(Unicode_object[,errors='strict']) -> (string object, bytes consumed) _decode(char_buffer_obj[,errors='strict']) -> (Unicode object, bytes consumed) _encode() interfaces also accept non-Unicode object as input. The objects are then converted to Unicode using PyUnicode_FromObject() prior to applying the conversion. These s are available: utf_8, unicode_escape, raw_unicode_escape, unicode_internal, latin_1, ascii (7-bit), mbcs (on win32). Written by Marc-Andre Lemburg (mal@lemburg.com). Copyright (c) Corporation for National Research Initiatives. From PyPy v1.0.0 """ #from unicodecodec import * __all__ = ['register', 'lookup', 'lookup_error', 'register_error', 'encode', 'decode', 'latin_1_encode', 'mbcs_decode', 'readbuffer_encode', 'escape_encode', 'utf_8_decode', 'raw_unicode_escape_decode', 'utf_7_decode', 'unicode_escape_encode', 'latin_1_decode', 'utf_16_decode', 'unicode_escape_decode', 'ascii_decode', 'charmap_encode', 'charmap_build', 'unicode_internal_encode', 'unicode_internal_decode', 'utf_16_ex_decode', 'escape_decode', 'charmap_decode', 'utf_7_encode', 'mbcs_encode', 'ascii_encode', 'utf_16_encode', 'raw_unicode_escape_encode', 'utf_8_encode', 'utf_16_le_encode', 'utf_16_be_encode', 'utf_16_le_decode', 'utf_16_be_decode',] import sys #/* --- Registry ----------------------------------------------------------- */ codec_search_path = [] codec_search_cache = {} codec_error_registry = {} codec_need_encodings = [True] def register( search_function ): """register(search_function) Register a codec search function. Search functions are expected to take one argument, the encoding name in all lower case letters, and return a tuple of functions (encoder, decoder, stream_reader, stream_writer). """ if not callable(search_function): raise TypeError("argument must be callable") codec_search_path.append(search_function) def lookup(encoding): """lookup(encoding) -> (encoder, decoder, stream_reader, stream_writer) Looks up a codec tuple in the Python codec registry and returns a tuple of functions. """ if not isinstance(encoding, str): raise TypeError("Encoding must be a string") normalized_encoding = encoding.replace(" ", "-").lower() result = codec_search_cache.get(normalized_encoding, None) if not result: if codec_need_encodings: import encodings if len(codec_search_path) == 0: raise LookupError("no codec search functions registered: can't find encoding") del codec_need_encodings[:] for search in codec_search_path: result = search(normalized_encoding) if result: codec_search_cache[normalized_encoding] = result return result if not result: raise LookupError("unknown encoding: %s" % encoding) return result def encode(v, encoding=None, errors='strict'): """encode(obj, [encoding[,errors]]) -> object Encodes obj using the codec registered for encoding. encoding defaults to the default encoding. errors may be given to set a different error handling scheme. Default is 'strict' meaning that encoding errors raise a ValueError. Other possible values are 'ignore', 'replace' and 'xmlcharrefreplace' as well as any other name registered with codecs.register_error that can handle ValueErrors. """ if encoding == None: encoding = sys.getdefaultencoding() if not isinstance(encoding, str): raise TypeError("Encoding must be a string") if not isinstance(errors, str): raise TypeError("Errors must be a string") codec = lookup(encoding) res = codec[0](v, errors) if not isinstance(res, tuple) or len(res) != 2: raise TypeError("encoder must return a tuple (object, integer)") return res[0] def decode(obj, encoding=None, errors='strict'): """decode(obj, [encoding[,errors]]) -> object Decodes obj using the codec registered for encoding. encoding defaults to the default encoding. errors may be given to set a different error handling scheme. Default is 'strict' meaning that encoding errors raise a ValueError. Other possible values are 'ignore' and 'replace' as well as any other name registerd with codecs.register_error that is able to handle ValueErrors. """ if encoding == None: encoding = sys.getdefaultencoding() if not isinstance(encoding, str): raise TypeError("Encoding must be a string") if not isinstance(errors, str): raise TypeError("Errors must be a string") codec = lookup(encoding) res = codec[1](obj, errors) if not isinstance(res, tuple) or len(res) != 2: raise TypeError("encoder must return a tuple (object, integer)") return res[0] def latin_1_encode( obj, errors='strict'): """None """ res = PyUnicode_EncodeLatin1(obj, len(obj), errors) res = bytes(res) return res, len(res) # XXX MBCS codec might involve ctypes ? def mbcs_decode(): """None """ pass def readbuffer_encode( obj, errors='strict'): """None """ res = str(obj) return res, len(res) def escape_encode( obj, errors='strict'): """None """ s = repr(obj) v = s[1:-1] return v, len(v) def utf_8_decode( data, errors='strict', final=False): """None """ consumed = len(data) if final: consumed = 0 res, consumed = PyUnicode_DecodeUTF8Stateful(data, len(data), errors, final) res = ''.join(res) return res, consumed def raw_unicode_escape_decode( data, errors='strict'): """None """ res = PyUnicode_DecodeRawUnicodeEscape(data, len(data), errors) res = ''.join(res) return res, len(res) def utf_7_decode( data, errors='strict'): """None """ res = PyUnicode_DecodeUTF7(data, len(data), errors) res = ''.join(res) return res, len(res) def unicode_escape_encode( obj, errors='strict'): """None """ res = unicodeescape_string(obj, len(obj), 0) res = bytes(res) return res, len(res) def latin_1_decode( data, errors='strict'): """None """ res = PyUnicode_DecodeLatin1(data, len(data), errors) res = ''.join(res) return res, len(res) def utf_16_decode( data, errors='strict', final=False): """None """ consumed = len(data) if final: consumed = 0 res, consumed, byteorder = PyUnicode_DecodeUTF16Stateful(data, len(data), errors, 'native', final) res = ''.join(res) return res, consumed def unicode_escape_decode( data, errors='strict'): """None """ res = PyUnicode_DecodeUnicodeEscape(data, len(data), errors) res = ''.join(res) return res, len(res) def ascii_decode( data, errors='strict'): """None """ res = PyUnicode_DecodeASCII(data, len(data), errors) res = ''.join(res) return res, len(res) def charmap_encode(obj, errors='strict', mapping='latin-1'): """None """ res = PyUnicode_EncodeCharmap(obj, len(obj), mapping, errors) res = bytes(res) return res, len(res) def charmap_build(s): return {ord(c): i for i, c in enumerate(s)} if sys.maxunicode == 65535: unicode_bytes = 2 else: unicode_bytes = 4 def unicode_internal_encode( obj, errors='strict'): """None """ if type(obj) == str: p = bytearray() t = [ord(x) for x in obj] for i in t: b = bytearray() for j in range(unicode_bytes): b.append(i%256) i >>= 8 if sys.byteorder == "big": b.reverse() p += b res = bytes(p) return res, len(res) else: res = "You can do better than this" # XXX make this right return res, len(res) def unicode_internal_decode( unistr, errors='strict'): """None """ if type(unistr) == str: return unistr, len(unistr) else: p = [] i = 0 if sys.byteorder == "big": start = unicode_bytes - 1 stop = -1 step = -1 else: start = 0 stop = unicode_bytes step = 1 while i < len(unistr)-unicode_bytes+1: t = 0 h = 0 for j in range(start, stop, step): t += ord(unistr[i+j])<<(h*8) h += 1 i += unicode_bytes p += chr(t) res = ''.join(p) return res, len(res) def utf_16_ex_decode( data, errors='strict', byteorder=0, final=0): """None """ if byteorder == 0: bm = 'native' elif byteorder == -1: bm = 'little' else: bm = 'big' consumed = len(data) if final: consumed = 0 res, consumed, byteorder = PyUnicode_DecodeUTF16Stateful(data, len(data), errors, bm, final) res = ''.join(res) return res, consumed, byteorder # XXX needs error messages when the input is invalid def escape_decode(data, errors='strict'): """None """ l = len(data) i = 0 res = [] while i < l: if data[i] == '\\': i += 1 if i >= l: raise ValueError("Trailing \\ in string") else: if data[i] == '\\': res += '\\' elif data[i] == 'n': res += '\n' elif data[i] == 't': res += '\t' elif data[i] == 'r': res += '\r' elif data[i] == 'b': res += '\b' elif data[i] == '\'': res += '\'' elif data[i] == '\"': res += '\"' elif data[i] == 'f': res += '\f' elif data[i] == 'a': res += '\a' elif data[i] == 'v': res += '\v' elif '0' <= data[i] <= '9': # emulate a strange wrap-around behavior of CPython: # \400 is the same as \000 because 0400 == 256 octal = data[i:i+3] res += chr(int(octal, 8) & 0xFF) i += 2 elif data[i] == 'x': hexa = data[i+1:i+3] res += chr(int(hexa, 16)) i += 2 else: res += data[i] i += 1 res = ''.join(res) return res, len(res) def charmap_decode( data, errors='strict', mapping=None): """None """ res = PyUnicode_DecodeCharmap(data, len(data), mapping, errors) res = ''.join(res) return res, len(res) def utf_7_encode( obj, errors='strict'): """None """ res = PyUnicode_EncodeUTF7(obj, len(obj), 0, 0, errors) res = bytes(res) return res, len(res) def mbcs_encode( obj, errors='strict'): """None """ pass ## return (PyUnicode_EncodeMBCS( ## (obj), ## len(obj), ## errors), ## len(obj)) def ascii_encode( obj, errors='strict'): """None """ res = PyUnicode_EncodeASCII(obj, len(obj), errors) res = bytes(res) return res, len(res) def utf_16_encode( obj, errors='strict'): """None """ res = PyUnicode_EncodeUTF16(obj, len(obj), errors, 'native') res = bytes(res) return res, len(res) def raw_unicode_escape_encode( obj, errors='strict'): """None """ res = PyUnicode_EncodeRawUnicodeEscape(obj, len(obj)) res = bytes(res) return res, len(res) def utf_8_encode( obj, errors='strict'): """None """ res = PyUnicode_EncodeUTF8(obj, len(obj), errors) res = bytes(res) return res, len(res) def utf_16_le_encode( obj, errors='strict'): """None """ res = PyUnicode_EncodeUTF16(obj, len(obj), errors, 'little') res = bytes(res) return res, len(res) def utf_16_be_encode( obj, errors='strict'): """None """ res = PyUnicode_EncodeUTF16(obj, len(obj), errors, 'big') res = bytes(res) return res, len(res) def utf_16_le_decode( data, errors='strict', byteorder=0, final = 0): """None """ consumed = len(data) if final: consumed = 0 res, consumed, byteorder = PyUnicode_DecodeUTF16Stateful(data, len(data), errors, 'little', final) res = ''.join(res) return res, consumed def utf_16_be_decode( data, errors='strict', byteorder=0, final = 0): """None """ consumed = len(data) if final: consumed = 0 res, consumed, byteorder = PyUnicode_DecodeUTF16Stateful(data, len(data), errors, 'big', final) res = ''.join(res) return res, consumed class _PretendCFunction: def __init__(self, f): self.__f = f @property def _f(self): return self.__dict__['__f'] def __repr__(self): return repr(self._f) def __str__(self): return str(self._f) def __call__(self, *args, **kwargs): return self._f(*args, **kwargs) _mod = globals() for exp in __all__: if exp.endswith('_encode') or exp.endswith('_decode'): # encodings expect most of these to be builtin functions, so we pretend they are _mod[exp] = _PretendCFunction(_mod[exp]) del _mod, exp def strict_errors(exc): if isinstance(exc, Exception): raise exc else: raise TypeError("codec must pass exception instance") def ignore_errors(exc): if isinstance(exc, UnicodeEncodeError): return '', exc.end elif isinstance(exc, (UnicodeDecodeError, UnicodeTranslateError)): return '', exc.end else: raise TypeError("don't know how to handle %.400s in error callback"%exc) Py_UNICODE_REPLACEMENT_CHARACTER = "\ufffd" def replace_errors(exc): if isinstance(exc, UnicodeEncodeError): return '?'*(exc.end-exc.start), exc.end elif isinstance(exc, (UnicodeTranslateError, UnicodeDecodeError)): return Py_UNICODE_REPLACEMENT_CHARACTER*(exc.end-exc.start), exc.end else: raise TypeError("don't know how to handle %.400s in error callback"%exc) def xmlcharrefreplace_errors(exc): if isinstance(exc, UnicodeEncodeError): res = [] for ch in exc.object[exc.start:exc.end]: res += '&#' res += str(ord(ch)) res += ';' return ''.join(res), exc.end else: raise TypeError("don't know how to handle %.400s in error callback"%type(exc)) def backslashreplace_errors(exc): if isinstance(exc, UnicodeEncodeError): p = [] for c in exc.object[exc.start:exc.end]: p += '\\' oc = ord(c) if (oc >= 0x00010000): p += 'U' p += "%.8x" % ord(c) elif (oc >= 0x100): p += 'u' p += "%.4x" % ord(c) else: p += 'x' p += "%.2x" % ord(c) return ''.join(p), exc.end else: raise TypeError("don't know how to handle %.400s in error callback"%type(exc)) # ---------------------------------------------------------------------- ##import sys ##""" Python implementation of CPythons builtin unicode codecs. ## ## Generally the functions in this module take a list of characters an returns ## a list of characters. ## ## For use in the PyPy project""" ## indicate whether a UTF-7 character is special i.e. cannot be directly ## encoded: ## 0 - not special ## 1 - special ## 2 - whitespace (optional) ## 3 - RFC2152 Set O (optional) utf7_special = [ 1, 1, 1, 1, 1, 1, 1, 1, 1, 2, 2, 1, 1, 2, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 2, 3, 3, 3, 3, 3, 3, 0, 0, 0, 3, 1, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 3, 3, 3, 3, 0, 3, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 3, 1, 3, 3, 3, 3, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 3, 3, 3, 1, 1, ] unicode_latin1 = [None]*256 def lookup_error(errors): """lookup_error(errors) -> handler Return the error handler for the specified error handling name or raise a LookupError, if no handler exists under this name. """ try: err_handler = codec_error_registry[errors] except KeyError: raise LookupError("unknown error handler name %s"%errors) return err_handler def register_error(errors, handler): """register_error(errors, handler) Register the specified error handler under the name errors. handler must be a callable object, that will be called with an exception instance containing information about the location of the encoding/decoding error and must return a (replacement, new position) tuple. """ if callable(handler): codec_error_registry[errors] = handler else: raise TypeError("handler must be callable") register_error("strict", strict_errors) register_error("ignore", ignore_errors) register_error("replace", replace_errors) register_error("xmlcharrefreplace", xmlcharrefreplace_errors) register_error("backslashreplace", backslashreplace_errors) def SPECIAL(c, encodeO, encodeWS): c = ord(c) return (c>127 or utf7_special[c] == 1) or \ (encodeWS and (utf7_special[(c)] == 2)) or \ (encodeO and (utf7_special[(c)] == 3)) def B64(n): return ("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/"[(n) & 0x3f]) def B64CHAR(c): return (c.isalnum() or (c) == '+' or (c) == '/') def UB64(c): if (c) == '+' : return 62 elif (c) == '/': return 63 elif (c) >= 'a': return ord(c) - 71 elif (c) >= 'A': return ord(c) - 65 else: return ord(c) + 4 def ENCODE( ch, bits) : out = [] while (bits >= 6): out += B64(ch >> (bits-6)) bits -= 6 return out, bits def PyUnicode_DecodeUTF7(s, size, errors): starts = s errmsg = "" inShift = 0 bitsleft = 0 charsleft = 0 surrogate = 0 p = [] errorHandler = None exc = None if (size == 0): return str('') i = 0 while i < size: ch = s[i] if (inShift): if ((ch == '-') or not B64CHAR(ch)): inShift = 0 i += 1 while (bitsleft >= 16): outCh = ((charsleft) >> (bitsleft-16)) & 0xffff bitsleft -= 16 if (surrogate): ## We have already generated an error for the high surrogate ## so let's not bother seeing if the low surrogate is correct or not surrogate = 0 elif (0xDC00 <= (outCh) and (outCh) <= 0xDFFF): ## This is a surrogate pair. Unfortunately we can't represent ## it in a 16-bit character surrogate = 1 msg = "code pairs are not supported" out, x = unicode_call_errorhandler(errors, 'utf-7', msg, s, i-1, i) p += out bitsleft = 0 break else: p += chr(outCh ) #p += out if (bitsleft >= 6): ## /* The shift sequence has a partial character in it. If ## bitsleft < 6 then we could just classify it as padding ## but that is not the case here */ msg = "partial character in shift sequence" out, x = unicode_call_errorhandler(errors, 'utf-7', msg, s, i-1, i) ## /* According to RFC2152 the remaining bits should be zero. We ## choose to signal an error/insert a replacement character ## here so indicate the potential of a misencoded character. */ ## /* On x86, a << b == a << (b%32) so make sure that bitsleft != 0 */ ## if (bitsleft and (charsleft << (sizeof(charsleft) * 8 - bitsleft))): ## raise UnicodeDecodeError, "non-zero padding bits in shift sequence" if (ch == '-') : if ((i < size) and (s[i] == '-')) : p += '-' inShift = 1 elif SPECIAL(ch, 0, 0) : raise UnicodeDecodeError("unexpected special character") else: p += ch else: charsleft = (charsleft << 6) | UB64(ch) bitsleft += 6 i += 1 ## /* p, charsleft, bitsleft, surrogate = */ DECODE(p, charsleft, bitsleft, surrogate); elif ( ch == '+' ): startinpos = i i += 1 if (i 0 else: out += chr(ord(ch)) else: if (not SPECIAL(ch, encodeSetO, encodeWhiteSpace)): out += B64((charsleft) << (6-bitsleft)) charsleft = 0 bitsleft = 0 ## /* Characters not in the BASE64 set implicitly unshift the sequence ## so no '-' is required, except if the character is itself a '-' */ if (B64CHAR(ch) or ch == '-'): out += '-' inShift = False out += chr(ord(ch)) else: bitsleft += 16 charsleft = (((charsleft) << 16) | ord(ch)) p, bitsleft = ENCODE(charsleft, bitsleft) out += p ## /* If the next character is special then we dont' need to terminate ## the shift sequence. If the next character is not a BASE64 character ## or '-' then the shift sequence will be terminated implicitly and we ## don't have to insert a '-'. */ if (bitsleft == 0): if (i + 1 < size): ch2 = s[i+1] if (SPECIAL(ch2, encodeSetO, encodeWhiteSpace)): pass elif (B64CHAR(ch2) or ch2 == '-'): out += '-' inShift = False else: inShift = False else: out += '-' inShift = False i += 1 if (bitsleft): out += B64(charsleft << (6-bitsleft) ) out += '-' return out unicode_empty = '' def unicodeescape_string(s, size, quotes): p = [] if (quotes) : p += 'u' if (s.find('\'') != -1 and s.find('"') == -1): p += '"' else: p += '\'' pos = 0 while (pos < size): ch = s[pos] #/* Escape quotes */ if (quotes and (ch == p[1] or ch == '\\')): p += '\\' p += chr(ord(ch)) pos += 1 continue #ifdef Py_UNICODE_WIDE #/* Map 21-bit characters to '\U00xxxxxx' */ elif (ord(ch) >= 0x10000): p += '\\' p += 'U' p += '%08x' % ord(ch) pos += 1 continue #endif #/* Map UTF-16 surrogate pairs to Unicode \UXXXXXXXX escapes */ elif (ord(ch) >= 0xD800 and ord(ch) < 0xDC00): pos += 1 ch2 = s[pos] if (ord(ch2) >= 0xDC00 and ord(ch2) <= 0xDFFF): ucs = (((ord(ch) & 0x03FF) << 10) | (ord(ch2) & 0x03FF)) + 0x00010000 p += '\\' p += 'U' p += '%08x' % ucs pos += 1 continue #/* Fall through: isolated surrogates are copied as-is */ pos -= 1 #/* Map 16-bit characters to '\uxxxx' */ if (ord(ch) >= 256): p += '\\' p += 'u' p += '%04x' % ord(ch) #/* Map special whitespace to '\t', \n', '\r' */ elif (ch == '\t'): p += '\\' p += 't' elif (ch == '\n'): p += '\\' p += 'n' elif (ch == '\r'): p += '\\' p += 'r' #/* Map non-printable US ASCII to '\xhh' */ elif (ch < ' ' or ch >= 0x7F) : p += '\\' p += 'x' p += '%02x' % ord(ch) #/* Copy everything else as-is */ else: p += chr(ord(ch)) pos += 1 if (quotes): p += p[1] return p def PyUnicode_DecodeASCII(s, size, errors): # /* ASCII is equivalent to the first 128 ordinals in Unicode. */ if (size == 1 and ord(s) < 128) : return [chr(ord(s))] if (size == 0): return [''] #unicode('') p = [] pos = 0 while pos < len(s): c = s[pos] if c < 128: p += chr(c) pos += 1 else: res = unicode_call_errorhandler( errors, "ascii", "ordinal not in range(128)", s, pos, pos+1) p += res[0] pos = res[1] return p def PyUnicode_EncodeASCII(p, size, errors): return unicode_encode_ucs1(p, size, errors, 128) def PyUnicode_AsASCIIString(unistr): if not type(unistr) == str: raise TypeError return PyUnicode_EncodeASCII(str(unistr), len(str), None) def PyUnicode_DecodeUTF16Stateful(s, size, errors, byteorder='native', final=True): bo = 0 #/* assume native ordering by default */ consumed = 0 errmsg = "" if sys.byteorder == 'little': ihi = 1 ilo = 0 else: ihi = 0 ilo = 1 #/* Unpack UTF-16 encoded data */ ## /* Check for BOM marks (U+FEFF) in the input and adjust current ## byte order setting accordingly. In native mode, the leading BOM ## mark is skipped, in all other modes, it is copied to the output ## stream as-is (giving a ZWNBSP character). */ q = 0 p = [] if byteorder == 'native': if (size >= 2): bom = (s[ihi] << 8) | s[ilo] #ifdef BYTEORDER_IS_LITTLE_ENDIAN if sys.byteorder == 'little': if (bom == 0xFEFF): q += 2 bo = -1 elif bom == 0xFFFE: q += 2 bo = 1 else: if bom == 0xFEFF: q += 2 bo = 1 elif bom == 0xFFFE: q += 2 bo = -1 elif byteorder == 'little': bo = -1 else: bo = 1 if (size == 0): return [''], 0, bo if (bo == -1): #/* force LE */ ihi = 1 ilo = 0 elif (bo == 1): #/* force BE */ ihi = 0 ilo = 1 while (q < len(s)): #/* remaining bytes at the end? (size should be even) */ if (len(s)-q<2): if not final: break errmsg = "truncated data" startinpos = q endinpos = len(s) unicode_call_errorhandler(errors, 'utf-16', errmsg, s, startinpos, endinpos, True) # /* The remaining input chars are ignored if the callback ## chooses to skip the input */ ch = (s[q+ihi] << 8) | s[q+ilo] q += 2 if (ch < 0xD800 or ch > 0xDFFF): p.append(chr(ch)) continue #/* UTF-16 code pair: */ if (q >= len(s)): errmsg = "unexpected end of data" startinpos = q-2 endinpos = len(s) unicode_call_errorhandler(errors, 'utf-16', errmsg, s, startinpos, endinpos, True) if (0xD800 <= ch and ch <= 0xDBFF): ch2 = (s[q+ihi] << 8) | s[q+ilo] q += 2 if (0xDC00 <= ch2 and ch2 <= 0xDFFF): #ifndef Py_UNICODE_WIDE if sys.maxunicode < 65536: p += [chr(ch), chr(ch2)] else: p.append(chr((((ch & 0x3FF)<<10) | (ch2 & 0x3FF)) + 0x10000)) #endif continue else: errmsg = "illegal UTF-16 surrogate" startinpos = q-4 endinpos = startinpos+2 unicode_call_errorhandler(errors, 'utf-16', errmsg, s, startinpos, endinpos, True) errmsg = "illegal encoding" startinpos = q-2 endinpos = startinpos+2 unicode_call_errorhandler(errors, 'utf-16', errmsg, s, startinpos, endinpos, True) return p, q, bo # moved out of local scope, especially because it didn't # have any nested variables. def STORECHAR(CH, byteorder): hi = (CH >> 8) & 0xff lo = CH & 0xff if byteorder == 'little': return [lo, hi] else: return [hi, lo] def PyUnicode_EncodeUTF16(s, size, errors, byteorder='little'): # /* Offsets from p for storing byte pairs in the right order. */ p = [] bom = sys.byteorder if (byteorder == 'native'): bom = sys.byteorder p += STORECHAR(0xFEFF, bom) if (size == 0): return "" if (byteorder == 'little' ): bom = 'little' elif (byteorder == 'big'): bom = 'big' for c in s: ch = ord(c) ch2 = 0 if (ch >= 0x10000) : ch2 = 0xDC00 | ((ch-0x10000) & 0x3FF) ch = 0xD800 | ((ch-0x10000) >> 10) p += STORECHAR(ch, bom) if (ch2): p += STORECHAR(ch2, bom) return p def PyUnicode_DecodeMBCS(s, size, errors): pass def PyUnicode_EncodeMBCS(p, size, errors): pass def unicode_call_errorhandler(errors, encoding, reason, input, startinpos, endinpos, decode=True): errorHandler = lookup_error(errors) if decode: exceptionObject = UnicodeDecodeError(encoding, input, startinpos, endinpos, reason) else: exceptionObject = UnicodeEncodeError(encoding, input, startinpos, endinpos, reason) res = errorHandler(exceptionObject) if isinstance(res, tuple) and isinstance(res[0], str) and isinstance(res[1], int): newpos = res[1] if (newpos < 0): newpos = len(input) + newpos if newpos < 0 or newpos > len(input): raise IndexError( "position %d from error handler out of bounds" % newpos) return res[0], newpos else: raise TypeError("encoding error handler must return (unicode, int) tuple, not %s" % repr(res)) def PyUnicode_DecodeUTF8(s, size, errors): return PyUnicode_DecodeUTF8Stateful(s, size, errors, False) ## /* Map UTF-8 encoded prefix byte to sequence length. zero means ## illegal prefix. see RFC 2279 for details */ utf8_code_length = [ 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 4, 4, 4, 4, 4, 4, 4, 4, 5, 5, 5, 5, 6, 6, 0, 0 ] def PyUnicode_DecodeUTF8Stateful(s, size, errors, final): consumed = 0 if (size == 0): if not final: consumed = 0 return '', consumed p = [] pos = 0 while pos < size: ch = s[pos] if ch < 0x80: p += chr(ch) pos += 1 continue n = utf8_code_length[ch] startinpos = pos if (startinpos + n > size): if not final: break else: errmsg = "unexpected end of data" endinpos = size res = unicode_call_errorhandler( errors, "utf8", errmsg, s, startinpos, endinpos) p += res[0] pos = res[1] if n == 0: errmsg = "unexpected code byte" endinpos = startinpos+1 res = unicode_call_errorhandler( errors, "utf8", errmsg, s, startinpos, endinpos) p += res[0] pos = res[1] elif n == 1: errmsg = "internal error" endinpos = startinpos+1 res = unicode_call_errorhandler( errors, "utf8", errmsg, s, startinpos, endinpos) p += res[0] pos = res[1] elif n == 2: if ((s[pos+1] & 0xc0) != 0x80): errmsg = "invalid data" endinpos = startinpos+2 res = unicode_call_errorhandler( errors, "utf8", errmsg, s, startinpos, endinpos) p += res[0] pos = res[1] else: c = ((s[pos] & 0x1f) << 6) + (s[pos+1] & 0x3f) if c < 0x80: errmsg = "illegal encoding" endinpos = startinpos+2 res = unicode_call_errorhandler( errors, "utf8", errmsg, s, startinpos, endinpos) p += res[0] pos = res[1] else: p += chr(c) pos += n #break elif n == 3: if ((s[pos+1] & 0xc0) != 0x80 or (s[pos+2] & 0xc0) != 0x80): errmsg = "invalid data" endinpos = startinpos+3 res = unicode_call_errorhandler( errors, "utf8", errmsg, s, startinpos, endinpos) p += res[0] pos = res[1] else: c = ((s[pos] & 0x0f) << 12) + \ ((s[pos+1] & 0x3f) << 6) +\ (s[pos+2] & 0x3f) ## /* Note: UTF-8 encodings of surrogates are considered ## legal UTF-8 sequences; ## ## XXX For wide builds (UCS-4) we should probably try ## to recombine the surrogates into a single code ## unit. ## */ if c < 0x0800: errmsg = "illegal encoding" endinpos = startinpos+3 res = unicode_call_errorhandler( errors, "utf8", errmsg, s, startinpos, endinpos) p += res[0] pos = res[1] else: p += chr(c) pos += n elif n == 4: ## case 4: if ((s[pos+1] & 0xc0) != 0x80 or (s[pos+2] & 0xc0) != 0x80 or (s[pos+3] & 0xc0) != 0x80): errmsg = "invalid data" startinpos = pos endinpos = startinpos+4 res = unicode_call_errorhandler( errors, "utf8", errmsg, s, startinpos, endinpos) p += res[0] pos = res[1] else: c = ((s[pos+0] & 0x7) << 18) + ((s[pos+1] & 0x3f) << 12) +\ ((s[pos+2] & 0x3f) << 6) + (s[pos+3] & 0x3f) #/* validate and convert to UTF-16 */ if ((c < 0x10000) or (c > 0x10ffff)): #/* minimum value allowed for 4 byte encoding */ #/* maximum value allowed for UTF-16 */ errmsg = "illegal encoding" startinpos = pos endinpos = startinpos+4 res = unicode_call_errorhandler( errors, "utf8", errmsg, s, startinpos, endinpos) p += res[0] pos = res[1] else: #ifdef Py_UNICODE_WIDE if c < sys.maxunicode: p += chr(c) pos += n else: ## /* compute and append the two surrogates: */ ## /* translate from 10000..10FFFF to 0..FFFF */ c -= 0x10000 #/* high surrogate = top 10 bits added to D800 */ p += chr(0xD800 + (c >> 10)) #/* low surrogate = bottom 10 bits added to DC00 */ p += chr(0xDC00 + (c & 0x03FF)) pos += n else: ## default: ## /* Other sizes are only needed for UCS-4 */ errmsg = "unsupported Unicode code range" startinpos = pos endinpos = startinpos+n res = unicode_call_errorhandler( errors, "utf8", errmsg, s, startinpos, endinpos) p += res[0] pos = res[1] #continue if not final: consumed = pos return p, pos # consumed def PyUnicode_EncodeUTF8(s, size, errors): #assert(s != None) assert(size >= 0) p = bytearray() i = 0 while i < size: ch = s[i] i += 1 if (ord(ch) < 0x80): ## /* Encode ASCII */ p.append(ord(ch)) elif (ord(ch) < 0x0800) : ## /* Encode Latin-1 */ p.append(0xc0 | (ord(ch) >> 6)) p.append(0x80 | (ord(ch) & 0x3f)) else: ## /* Encode UCS2 Unicode ordinals */ if (ord(ch) < 0x10000): ## /* Special case: check for high surrogate */ if (0xD800 <= ord(ch) and ord(ch) <= 0xDBFF and i != size) : ch2 = s[i] ## /* Check for low surrogate and combine the two to ## form a UCS4 value */ if (0xDC00 <= ord(ch2) and ord(ch2) <= 0xDFFF) : ch3 = ((ord(ch) - 0xD800) << 10 | (ord(ch2) - 0xDC00)) + 0x10000 i += 1 p += encodeUCS4(ch3) continue ## /* Fall through: handles isolated high surrogates */ p.append(0xe0 | (ord(ch) >> 12)) p.append(0x80 | ((ord(ch) >> 6) & 0x3f)) p.append(0x80 | (ord(ch) & 0x3f)) continue else: p += encodeUCS4(ord(ch)) return p def encodeUCS4(ch): ## /* Encode UCS4 Unicode ordinals */ p = bytearray() p.append(0xf0 | (ch >> 18)) p.append(0x80 | ((ch >> 12) & 0x3f)) p.append(0x80 | ((ch >> 6) & 0x3f)) p.append(0x80 | (ch & 0x3f)) return p #/* --- Latin-1 Codec ------------------------------------------------------ */ def PyUnicode_DecodeLatin1(s, size, errors): #/* Latin-1 is equivalent to the first 256 ordinals in Unicode. */ ## if (size == 1): ## return [PyUnicode_FromUnicode(s, 1)] pos = 0 p = [] while (pos < size): p += chr(s[pos]) pos += 1 return p def unicode_encode_ucs1(p, size, errors, limit): if limit == 256: reason = "ordinal not in range(256)" encoding = "latin-1" else: reason = "ordinal not in range(128)" encoding = "ascii" if (size == 0): return [] res = bytearray() pos = 0 while pos < len(p): #for ch in p: ch = p[pos] if ord(ch) < limit: res.append(ord(ch)) pos += 1 else: #/* startpos for collecting unencodable chars */ collstart = pos collend = pos+1 while collend < len(p) and ord(p[collend]) >= limit: collend += 1 x = unicode_call_errorhandler(errors, encoding, reason, p, collstart, collend, False) res += x[0].encode() pos = x[1] return res def PyUnicode_EncodeLatin1(p, size, errors): res = unicode_encode_ucs1(p, size, errors, 256) return res hexdigits = [hex(i)[-1] for i in range(16)]+[hex(i)[-1].upper() for i in range(10, 16)] def hexescape(s, pos, digits, message, errors): chr = 0 p = [] if (pos+digits>len(s)): message = "end of string in escape sequence" x = unicode_call_errorhandler(errors, "unicodeescape", message, s, pos-2, len(s)) p += x[0] pos = x[1] else: try: chr = int(s[pos:pos+digits], 16) except ValueError: endinpos = pos while s[endinpos] in hexdigits: endinpos += 1 x = unicode_call_errorhandler(errors, "unicodeescape", message, s, pos-2, endinpos+1) p += x[0] pos = x[1] #/* when we get here, chr is a 32-bit unicode character */ else: if chr <= sys.maxunicode: p += chr(chr) pos += digits elif (chr <= 0x10ffff): chr -= 0x10000 p += chr(0xD800 + (chr >> 10)) p += chr(0xDC00 + (chr & 0x03FF)) pos += digits else: message = "illegal Unicode character" x = unicode_call_errorhandler(errors, "unicodeescape", message, s, pos-2, pos+1) p += x[0] pos = x[1] res = p return res, pos def PyUnicode_DecodeUnicodeEscape(s, size, errors): if (size == 0): return '' p = [] pos = 0 while (pos < size): ## /* Non-escape characters are interpreted as Unicode ordinals */ if (chr(s[pos]) != '\\') : p += chr(s[pos]) pos += 1 continue ## /* \ - Escapes */ else: pos += 1 if pos >= len(s): errmessage = "\\ at end of string" unicode_call_errorhandler(errors, "unicodeescape", errmessage, s, pos-1, size) ch = chr(s[pos]) pos += 1 ## /* \x escapes */ if ch == '\\' : p += '\\' elif ch == '\'': p += '\'' elif ch == '\"': p += '\"' elif ch == 'b' : p += '\b' elif ch == 'f' : p += '\014' #/* FF */ elif ch == 't' : p += '\t' elif ch == 'n' : p += '\n' elif ch == 'r' : p += '\r' elif ch == 'v': p += '\013' #break; /* VT */ elif ch == 'a': p += '\007' # break; /* BEL, not classic C */ elif '0' <= ch <= '7': x = ord(ch) - ord('0') if pos < size: ch = s[pos] if '0' <= ch <= '7': pos += 1 x = (x<<3) + ord(ch) - ord('0') if pos < size: ch = s[pos] if '0' <= ch <= '7': pos += 1 x = (x<<3) + ord(ch) - ord('0') p += chr(x) ## /* hex escapes */ ## /* \xXX */ elif ch == 'x': digits = 2 message = "truncated \\xXX escape" x = hexescape(s, pos, digits, message, errors) p += x[0] pos = x[1] # /* \uXXXX */ elif ch == 'u': digits = 4 message = "truncated \\uXXXX escape" x = hexescape(s, pos, digits, message, errors) p += x[0] pos = x[1] # /* \UXXXXXXXX */ elif ch == 'U': digits = 8 message = "truncated \\UXXXXXXXX escape" x = hexescape(s, pos, digits, message, errors) p += x[0] pos = x[1] ## /* \N{name} */ elif ch == 'N': message = "malformed \\N character escape" # pos += 1 look = pos try: import unicodedata except ImportError: message = "\\N escapes not supported (can't load unicodedata module)" unicode_call_errorhandler(errors, "unicodeescape", message, s, pos-1, size) if look < size and chr(s[look]) == '{': #/* look for the closing brace */ while (look < size and chr(s[look]) != '}'): look += 1 if (look > pos+1 and look < size and chr(s[look]) == '}'): #/* found a name. look it up in the unicode database */ message = "unknown Unicode character name" st = s[pos+1:look] try: chr_codec = unicodedata.lookup("%s" % st) except LookupError as e: x = unicode_call_errorhandler(errors, "unicodeescape", message, s, pos-1, look+1) else: x = chr_codec, look + 1 p += x[0] pos = x[1] else: x = unicode_call_errorhandler(errors, "unicodeescape", message, s, pos-1, look+1) else: x = unicode_call_errorhandler(errors, "unicodeescape", message, s, pos-1, look+1) else: p += '\\' p += ch return p def PyUnicode_EncodeRawUnicodeEscape(s, size): if (size == 0): return '' p = bytearray() for ch in s: # /* Map 32-bit characters to '\Uxxxxxxxx' */ if (ord(ch) >= 0x10000): p += '\\' p += 'U' p += '%08x' % (ord(ch)) elif (ord(ch) >= 256) : # /* Map 16-bit characters to '\uxxxx' */ p += b'\\u%04x' % (ord(ch)) # /* Copy everything else as-is */ else: p.append(ord(ch)) #p += '\0' return p def charmapencode_output(c, mapping): rep = mapping[c] if isinstance(rep, int) or isinstance(rep, int): if rep < 256: return rep else: raise TypeError("character mapping must be in range(256)") elif isinstance(rep, str): return ord(rep) elif rep == None: raise KeyError("character maps to ") else: raise TypeError("character mapping must return integer, None or str") def PyUnicode_EncodeCharmap(p, size, mapping='latin-1', errors='strict'): ## /* the following variable is used for caching string comparisons ## * -1=not initialized, 0=unknown, 1=strict, 2=replace, ## * 3=ignore, 4=xmlcharrefreplace */ # /* Default to Latin-1 */ if mapping == 'latin-1': return PyUnicode_EncodeLatin1(p, size, errors) if (size == 0): return '' inpos = 0 res = [] while (inpos", p, inpos, inpos+1, False) try: res += [charmapencode_output(ord(y), mapping) for y in x[0]] except KeyError: raise UnicodeEncodeError("charmap", p, inpos, inpos+1, "character maps to ") inpos += 1 return res def PyUnicode_DecodeCharmap(s, size, mapping, errors): ## /* Default to Latin-1 */ if (mapping == None): return PyUnicode_DecodeLatin1(s, size, errors) if (size == 0): return '' p = [] inpos = 0 while (inpos< len(s)): #/* Get mapping (char ordinal -> integer, Unicode char or None) */ ch = s[inpos] try: x = mapping[ch] if isinstance(x, int): if x < 65536: p += chr(x) else: raise TypeError("character mapping must be in range(65536)") elif isinstance(x, str): p += x elif not x: raise KeyError else: raise TypeError except KeyError: x = unicode_call_errorhandler(errors, "charmap", "character maps to ", s, inpos, inpos+1) p += x[0] inpos += 1 return p def PyUnicode_DecodeRawUnicodeEscape(s, size, errors): if (size == 0): return '' pos = 0 p = [] while (pos < len(s)): ch = s[pos] #/* Non-escape characters are interpreted as Unicode ordinals */ if (ch != ord('\\')): p.append(chr(ch)) pos += 1 continue startinpos = pos ## /* \u-escapes are only interpreted iff the number of leading ## backslashes is odd */ bs = pos while pos < size: if (s[pos] != ord('\\')): break p.append(chr(s[pos])) pos += 1 if (((pos - bs) & 1) == 0 or pos >= size or (s[pos] != ord('u') and s[pos] != ord('U'))) : p.append(chr(s[pos])) pos += 1 continue p.pop(-1) if s[pos] == ord('u'): count = 4 else: count = 8 pos += 1 #/* \uXXXX with 4 hex digits, \Uxxxxxxxx with 8 */ x = 0 try: x = int(s[pos:pos+count], 16) except ValueError: res = unicode_call_errorhandler( errors, "rawunicodeescape", "truncated \\uXXXX", s, size, pos, pos+count) p.append(res[0]) pos = res[1] else: #ifndef Py_UNICODE_WIDE if sys.maxunicode > 0xffff: if (x > sys.maxunicode): res = unicode_call_errorhandler( errors, "rawunicodeescape", "\\Uxxxxxxxx out of range", s, size, pos, pos+1) pos = res[1] p.append(res[0]) else: p.append(chr(x)) pos += count else: if (x > 0x10000): res = unicode_call_errorhandler( errors, "rawunicodeescape", "\\Uxxxxxxxx out of range", s, size, pos, pos+1) pos = res[1] p.append(res[0]) #endif else: p.append(chr(x)) pos += count return p