blob: 446520ae93ac7c066b1b34c391df3fc3f82cc5f3 [file] [log] [blame]
Tim Petersced69f82003-09-16 20:30:58 +00001/*
Guido van Rossumd57fd912000-03-10 22:53:23 +00002
3Unicode implementation based on original code by Fredrik Lundh,
Benjamin Peterson31616ea2011-10-01 00:11:09 -04004modified by Marc-Andre Lemburg <mal@lemburg.com>.
Guido van Rossumd57fd912000-03-10 22:53:23 +00005
Thomas Wouters477c8d52006-05-27 19:21:47 +00006Major speed upgrades to the method implementations at the Reykjavik
7NeedForSpeed sprint, by Fredrik Lundh and Andrew Dalke.
8
Guido van Rossum16b1ad92000-08-03 16:24:25 +00009Copyright (c) Corporation for National Research Initiatives.
Guido van Rossumd57fd912000-03-10 22:53:23 +000010
Fredrik Lundh0fdb90c2001-01-19 09:45:02 +000011--------------------------------------------------------------------
12The original string type implementation is:
Guido van Rossumd57fd912000-03-10 22:53:23 +000013
Benjamin Peterson29060642009-01-31 22:14:21 +000014 Copyright (c) 1999 by Secret Labs AB
15 Copyright (c) 1999 by Fredrik Lundh
Guido van Rossumd57fd912000-03-10 22:53:23 +000016
Fredrik Lundh0fdb90c2001-01-19 09:45:02 +000017By obtaining, using, and/or copying this software and/or its
18associated documentation, you agree that you have read, understood,
19and will comply with the following terms and conditions:
20
21Permission to use, copy, modify, and distribute this software and its
22associated documentation for any purpose and without fee is hereby
23granted, provided that the above copyright notice appears in all
24copies, and that both that copyright notice and this permission notice
25appear in supporting documentation, and that the name of Secret Labs
26AB or the author not be used in advertising or publicity pertaining to
27distribution of the software without specific, written prior
28permission.
29
30SECRET LABS AB AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH REGARD TO
31THIS SOFTWARE, INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY AND
32FITNESS. IN NO EVENT SHALL SECRET LABS AB OR THE AUTHOR BE LIABLE FOR
33ANY SPECIAL, INDIRECT OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
34WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
35ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT
36OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
37--------------------------------------------------------------------
38
39*/
Guido van Rossumd57fd912000-03-10 22:53:23 +000040
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000041#define PY_SSIZE_T_CLEAN
Guido van Rossumd57fd912000-03-10 22:53:23 +000042#include "Python.h"
Marc-André Lemburgd49e5b42000-06-30 14:58:20 +000043#include "ucnhash.h"
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -050044#include "bytes_methods.h"
Guido van Rossumd57fd912000-03-10 22:53:23 +000045
Martin v. Löwis6238d2b2002-06-30 15:26:10 +000046#ifdef MS_WINDOWS
Guido van Rossumb7a40ba2000-03-28 02:01:52 +000047#include <windows.h>
48#endif
Guido van Rossumfd4b9572000-04-10 13:51:10 +000049
Guido van Rossumd57fd912000-03-10 22:53:23 +000050/* Endianness switches; defaults to little endian */
51
52#ifdef WORDS_BIGENDIAN
53# define BYTEORDER_IS_BIG_ENDIAN
54#else
55# define BYTEORDER_IS_LITTLE_ENDIAN
56#endif
57
Marc-André Lemburgd4ab4a52000-06-08 17:54:00 +000058/* --- Globals ------------------------------------------------------------
59
Serhiy Storchaka05997252013-01-26 12:14:02 +020060NOTE: In the interpreter's initialization phase, some globals are currently
61 initialized dynamically as needed. In the process Unicode objects may
62 be created before the Unicode type is ready.
Marc-André Lemburgd4ab4a52000-06-08 17:54:00 +000063
64*/
Guido van Rossumd57fd912000-03-10 22:53:23 +000065
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000066
67#ifdef __cplusplus
68extern "C" {
69#endif
70
Victor Stinner8faf8212011-12-08 22:14:11 +010071/* Maximum code point of Unicode 6.0: 0x10ffff (1,114,111) */
72#define MAX_UNICODE 0x10ffff
73
Victor Stinner910337b2011-10-03 03:20:16 +020074#ifdef Py_DEBUG
Victor Stinnerbb10a1f2011-10-05 01:34:17 +020075# define _PyUnicode_CHECK(op) _PyUnicode_CheckConsistency(op, 0)
Victor Stinner910337b2011-10-03 03:20:16 +020076#else
77# define _PyUnicode_CHECK(op) PyUnicode_Check(op)
78#endif
Victor Stinnerfb5f5f22011-09-28 21:39:49 +020079
Victor Stinnere90fe6a2011-10-01 16:48:13 +020080#define _PyUnicode_UTF8(op) \
81 (((PyCompactUnicodeObject*)(op))->utf8)
82#define PyUnicode_UTF8(op) \
Victor Stinner910337b2011-10-03 03:20:16 +020083 (assert(_PyUnicode_CHECK(op)), \
Victor Stinnere90fe6a2011-10-01 16:48:13 +020084 assert(PyUnicode_IS_READY(op)), \
85 PyUnicode_IS_COMPACT_ASCII(op) ? \
86 ((char*)((PyASCIIObject*)(op) + 1)) : \
87 _PyUnicode_UTF8(op))
Victor Stinnerbc8b81b2011-09-29 19:31:34 +020088#define _PyUnicode_UTF8_LENGTH(op) \
Victor Stinnere90fe6a2011-10-01 16:48:13 +020089 (((PyCompactUnicodeObject*)(op))->utf8_length)
90#define PyUnicode_UTF8_LENGTH(op) \
Victor Stinner910337b2011-10-03 03:20:16 +020091 (assert(_PyUnicode_CHECK(op)), \
Victor Stinnere90fe6a2011-10-01 16:48:13 +020092 assert(PyUnicode_IS_READY(op)), \
93 PyUnicode_IS_COMPACT_ASCII(op) ? \
94 ((PyASCIIObject*)(op))->length : \
95 _PyUnicode_UTF8_LENGTH(op))
Victor Stinnera5f91632011-10-04 01:07:11 +020096#define _PyUnicode_WSTR(op) \
97 (((PyASCIIObject*)(op))->wstr)
98#define _PyUnicode_WSTR_LENGTH(op) \
99 (((PyCompactUnicodeObject*)(op))->wstr_length)
100#define _PyUnicode_LENGTH(op) \
101 (((PyASCIIObject *)(op))->length)
102#define _PyUnicode_STATE(op) \
103 (((PyASCIIObject *)(op))->state)
104#define _PyUnicode_HASH(op) \
105 (((PyASCIIObject *)(op))->hash)
Victor Stinner910337b2011-10-03 03:20:16 +0200106#define _PyUnicode_KIND(op) \
107 (assert(_PyUnicode_CHECK(op)), \
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200108 ((PyASCIIObject *)(op))->state.kind)
Victor Stinner910337b2011-10-03 03:20:16 +0200109#define _PyUnicode_GET_LENGTH(op) \
110 (assert(_PyUnicode_CHECK(op)), \
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200111 ((PyASCIIObject *)(op))->length)
Victor Stinnera5f91632011-10-04 01:07:11 +0200112#define _PyUnicode_DATA_ANY(op) \
113 (((PyUnicodeObject*)(op))->data.any)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200114
Victor Stinner910337b2011-10-03 03:20:16 +0200115#undef PyUnicode_READY
116#define PyUnicode_READY(op) \
117 (assert(_PyUnicode_CHECK(op)), \
118 (PyUnicode_IS_READY(op) ? \
Victor Stinnera5f91632011-10-04 01:07:11 +0200119 0 : \
Victor Stinner7931d9a2011-11-04 00:22:48 +0100120 _PyUnicode_Ready(op)))
Victor Stinner910337b2011-10-03 03:20:16 +0200121
Victor Stinnerc379ead2011-10-03 12:52:27 +0200122#define _PyUnicode_SHARE_UTF8(op) \
123 (assert(_PyUnicode_CHECK(op)), \
124 assert(!PyUnicode_IS_COMPACT_ASCII(op)), \
125 (_PyUnicode_UTF8(op) == PyUnicode_DATA(op)))
126#define _PyUnicode_SHARE_WSTR(op) \
127 (assert(_PyUnicode_CHECK(op)), \
128 (_PyUnicode_WSTR(unicode) == PyUnicode_DATA(op)))
129
Victor Stinner829c0ad2011-10-03 01:08:02 +0200130/* true if the Unicode object has an allocated UTF-8 memory block
131 (not shared with other data) */
Victor Stinner910337b2011-10-03 03:20:16 +0200132#define _PyUnicode_HAS_UTF8_MEMORY(op) \
133 (assert(_PyUnicode_CHECK(op)), \
134 (!PyUnicode_IS_COMPACT_ASCII(op) \
135 && _PyUnicode_UTF8(op) \
Victor Stinner829c0ad2011-10-03 01:08:02 +0200136 && _PyUnicode_UTF8(op) != PyUnicode_DATA(op)))
137
Victor Stinner03490912011-10-03 23:45:12 +0200138/* true if the Unicode object has an allocated wstr memory block
139 (not shared with other data) */
140#define _PyUnicode_HAS_WSTR_MEMORY(op) \
141 (assert(_PyUnicode_CHECK(op)), \
142 (_PyUnicode_WSTR(op) && \
143 (!PyUnicode_IS_READY(op) || \
144 _PyUnicode_WSTR(op) != PyUnicode_DATA(op))))
145
Victor Stinner910337b2011-10-03 03:20:16 +0200146/* Generic helper macro to convert characters of different types.
147 from_type and to_type have to be valid type names, begin and end
148 are pointers to the source characters which should be of type
149 "from_type *". to is a pointer of type "to_type *" and points to the
150 buffer where the result characters are written to. */
151#define _PyUnicode_CONVERT_BYTES(from_type, to_type, begin, end, to) \
152 do { \
Antoine Pitroue459a082011-10-11 20:58:41 +0200153 to_type *_to = (to_type *) to; \
154 const from_type *_iter = (begin); \
155 const from_type *_end = (end); \
156 Py_ssize_t n = (_end) - (_iter); \
157 const from_type *_unrolled_end = \
Antoine Pitrouca8aa4a2012-09-20 20:56:47 +0200158 _iter + _Py_SIZE_ROUND_DOWN(n, 4); \
Antoine Pitroue459a082011-10-11 20:58:41 +0200159 while (_iter < (_unrolled_end)) { \
160 _to[0] = (to_type) _iter[0]; \
161 _to[1] = (to_type) _iter[1]; \
162 _to[2] = (to_type) _iter[2]; \
163 _to[3] = (to_type) _iter[3]; \
164 _iter += 4; _to += 4; \
Victor Stinner910337b2011-10-03 03:20:16 +0200165 } \
Antoine Pitroue459a082011-10-11 20:58:41 +0200166 while (_iter < (_end)) \
167 *_to++ = (to_type) *_iter++; \
Victor Stinner910337b2011-10-03 03:20:16 +0200168 } while (0)
Victor Stinner829c0ad2011-10-03 01:08:02 +0200169
Walter Dörwald16807132007-05-25 13:52:07 +0000170/* This dictionary holds all interned unicode strings. Note that references
171 to strings in this dictionary are *not* counted in the string's ob_refcnt.
172 When the interned string reaches a refcnt of 0 the string deallocation
173 function will delete the reference from this dictionary.
174
175 Another way to look at this is that to say that the actual reference
Guido van Rossum98297ee2007-11-06 21:34:58 +0000176 count of a string is: s->ob_refcnt + (s->state ? 2 : 0)
Walter Dörwald16807132007-05-25 13:52:07 +0000177*/
Serhiy Storchaka05997252013-01-26 12:14:02 +0200178static PyObject *interned = NULL;
Walter Dörwald16807132007-05-25 13:52:07 +0000179
Marc-André Lemburg8155e0e2001-04-23 14:44:21 +0000180/* The empty Unicode object is shared to improve performance. */
Serhiy Storchaka678db842013-01-26 12:16:36 +0200181static PyObject *unicode_empty = NULL;
Serhiy Storchaka05997252013-01-26 12:14:02 +0200182
Serhiy Storchaka678db842013-01-26 12:16:36 +0200183#define _Py_INCREF_UNICODE_EMPTY() \
Serhiy Storchaka05997252013-01-26 12:14:02 +0200184 do { \
185 if (unicode_empty != NULL) \
186 Py_INCREF(unicode_empty); \
187 else { \
Serhiy Storchaka678db842013-01-26 12:16:36 +0200188 unicode_empty = PyUnicode_New(0, 0); \
189 if (unicode_empty != NULL) { \
Serhiy Storchaka05997252013-01-26 12:14:02 +0200190 Py_INCREF(unicode_empty); \
Serhiy Storchaka678db842013-01-26 12:16:36 +0200191 assert(_PyUnicode_CheckConsistency(unicode_empty, 1)); \
192 } \
Serhiy Storchaka05997252013-01-26 12:14:02 +0200193 } \
Serhiy Storchaka05997252013-01-26 12:14:02 +0200194 } while (0)
Marc-André Lemburg8155e0e2001-04-23 14:44:21 +0000195
Serhiy Storchaka678db842013-01-26 12:16:36 +0200196#define _Py_RETURN_UNICODE_EMPTY() \
197 do { \
198 _Py_INCREF_UNICODE_EMPTY(); \
199 return unicode_empty; \
200 } while (0)
Marc-André Lemburg8155e0e2001-04-23 14:44:21 +0000201
Martin v. Löwisafe55bb2011-10-09 10:38:36 +0200202/* List of static strings. */
Serhiy Storchaka678db842013-01-26 12:16:36 +0200203static _Py_Identifier *static_strings = NULL;
Martin v. Löwisafe55bb2011-10-09 10:38:36 +0200204
Marc-André Lemburg8155e0e2001-04-23 14:44:21 +0000205/* Single character Unicode strings in the Latin-1 range are being
206 shared as well. */
Serhiy Storchaka678db842013-01-26 12:16:36 +0200207static PyObject *unicode_latin1[256] = {NULL};
Marc-André Lemburg8155e0e2001-04-23 14:44:21 +0000208
Christian Heimes190d79e2008-01-30 11:58:22 +0000209/* Fast detection of the most frequent whitespace characters */
210const unsigned char _Py_ascii_whitespace[] = {
Benjamin Peterson14339b62009-01-31 16:36:08 +0000211 0, 0, 0, 0, 0, 0, 0, 0,
Florent Xicluna806d8cf2010-03-30 19:34:18 +0000212/* case 0x0009: * CHARACTER TABULATION */
Christian Heimes1a8501c2008-10-02 19:56:01 +0000213/* case 0x000A: * LINE FEED */
Florent Xicluna806d8cf2010-03-30 19:34:18 +0000214/* case 0x000B: * LINE TABULATION */
Christian Heimes1a8501c2008-10-02 19:56:01 +0000215/* case 0x000C: * FORM FEED */
216/* case 0x000D: * CARRIAGE RETURN */
Benjamin Peterson14339b62009-01-31 16:36:08 +0000217 0, 1, 1, 1, 1, 1, 0, 0,
218 0, 0, 0, 0, 0, 0, 0, 0,
Christian Heimes1a8501c2008-10-02 19:56:01 +0000219/* case 0x001C: * FILE SEPARATOR */
220/* case 0x001D: * GROUP SEPARATOR */
221/* case 0x001E: * RECORD SEPARATOR */
222/* case 0x001F: * UNIT SEPARATOR */
Benjamin Peterson14339b62009-01-31 16:36:08 +0000223 0, 0, 0, 0, 1, 1, 1, 1,
Christian Heimes1a8501c2008-10-02 19:56:01 +0000224/* case 0x0020: * SPACE */
Benjamin Peterson14339b62009-01-31 16:36:08 +0000225 1, 0, 0, 0, 0, 0, 0, 0,
226 0, 0, 0, 0, 0, 0, 0, 0,
227 0, 0, 0, 0, 0, 0, 0, 0,
228 0, 0, 0, 0, 0, 0, 0, 0,
Christian Heimes190d79e2008-01-30 11:58:22 +0000229
Benjamin Peterson14339b62009-01-31 16:36:08 +0000230 0, 0, 0, 0, 0, 0, 0, 0,
231 0, 0, 0, 0, 0, 0, 0, 0,
232 0, 0, 0, 0, 0, 0, 0, 0,
233 0, 0, 0, 0, 0, 0, 0, 0,
234 0, 0, 0, 0, 0, 0, 0, 0,
235 0, 0, 0, 0, 0, 0, 0, 0,
236 0, 0, 0, 0, 0, 0, 0, 0,
237 0, 0, 0, 0, 0, 0, 0, 0
Christian Heimes190d79e2008-01-30 11:58:22 +0000238};
239
Victor Stinner1b4f9ce2011-10-03 13:28:14 +0200240/* forward */
Victor Stinnerfe226c02011-10-03 03:52:20 +0200241static PyUnicodeObject *_PyUnicode_New(Py_ssize_t length);
Victor Stinner1b4f9ce2011-10-03 13:28:14 +0200242static PyObject* get_latin1_char(unsigned char ch);
Victor Stinner488fa492011-12-12 00:01:39 +0100243static int unicode_modifiable(PyObject *unicode);
244
Victor Stinnerfe226c02011-10-03 03:52:20 +0200245
Alexander Belopolsky40018472011-02-26 01:02:56 +0000246static PyObject *
Victor Stinnerd21b58c2013-02-26 00:15:54 +0100247_PyUnicode_FromUCS1(const Py_UCS1 *s, Py_ssize_t size);
Antoine Pitroudd4e2f02011-10-13 00:02:27 +0200248static PyObject *
249_PyUnicode_FromUCS2(const Py_UCS2 *s, Py_ssize_t size);
250static PyObject *
251_PyUnicode_FromUCS4(const Py_UCS4 *s, Py_ssize_t size);
252
253static PyObject *
Alexander Belopolsky40018472011-02-26 01:02:56 +0000254unicode_encode_call_errorhandler(const char *errors,
Martin v. Löwisdb12d452009-05-02 18:52:14 +0000255 PyObject **errorHandler,const char *encoding, const char *reason,
Martin v. Löwis23e275b2011-11-02 18:02:51 +0100256 PyObject *unicode, PyObject **exceptionObject,
Martin v. Löwisdb12d452009-05-02 18:52:14 +0000257 Py_ssize_t startpos, Py_ssize_t endpos, Py_ssize_t *newpos);
258
Alexander Belopolsky40018472011-02-26 01:02:56 +0000259static void
260raise_encode_exception(PyObject **exceptionObject,
Ezio Melotti2aa2b3b2011-09-29 00:58:57 +0300261 const char *encoding,
Martin v. Löwis9e816682011-11-02 12:45:42 +0100262 PyObject *unicode,
263 Py_ssize_t startpos, Py_ssize_t endpos,
264 const char *reason);
Victor Stinner31be90b2010-04-22 19:38:16 +0000265
Christian Heimes190d79e2008-01-30 11:58:22 +0000266/* Same for linebreaks */
267static unsigned char ascii_linebreak[] = {
Benjamin Peterson14339b62009-01-31 16:36:08 +0000268 0, 0, 0, 0, 0, 0, 0, 0,
Christian Heimes1a8501c2008-10-02 19:56:01 +0000269/* 0x000A, * LINE FEED */
Florent Xicluna806d8cf2010-03-30 19:34:18 +0000270/* 0x000B, * LINE TABULATION */
271/* 0x000C, * FORM FEED */
Christian Heimes1a8501c2008-10-02 19:56:01 +0000272/* 0x000D, * CARRIAGE RETURN */
Florent Xicluna806d8cf2010-03-30 19:34:18 +0000273 0, 0, 1, 1, 1, 1, 0, 0,
Benjamin Peterson14339b62009-01-31 16:36:08 +0000274 0, 0, 0, 0, 0, 0, 0, 0,
Christian Heimes1a8501c2008-10-02 19:56:01 +0000275/* 0x001C, * FILE SEPARATOR */
276/* 0x001D, * GROUP SEPARATOR */
277/* 0x001E, * RECORD SEPARATOR */
Benjamin Peterson14339b62009-01-31 16:36:08 +0000278 0, 0, 0, 0, 1, 1, 1, 0,
279 0, 0, 0, 0, 0, 0, 0, 0,
280 0, 0, 0, 0, 0, 0, 0, 0,
281 0, 0, 0, 0, 0, 0, 0, 0,
282 0, 0, 0, 0, 0, 0, 0, 0,
Christian Heimes190d79e2008-01-30 11:58:22 +0000283
Benjamin Peterson14339b62009-01-31 16:36:08 +0000284 0, 0, 0, 0, 0, 0, 0, 0,
285 0, 0, 0, 0, 0, 0, 0, 0,
286 0, 0, 0, 0, 0, 0, 0, 0,
287 0, 0, 0, 0, 0, 0, 0, 0,
288 0, 0, 0, 0, 0, 0, 0, 0,
289 0, 0, 0, 0, 0, 0, 0, 0,
290 0, 0, 0, 0, 0, 0, 0, 0,
291 0, 0, 0, 0, 0, 0, 0, 0
Christian Heimes190d79e2008-01-30 11:58:22 +0000292};
293
Ezio Melotti48a2f8f2011-09-29 00:18:19 +0300294/* The max unicode value is always 0x10FFFF while using the PEP-393 API.
295 This function is kept for backward compatibility with the old API. */
Martin v. Löwisce9b5a52001-06-27 06:28:56 +0000296Py_UNICODE
Marc-André Lemburg6c6bfb72001-07-20 17:39:11 +0000297PyUnicode_GetMax(void)
Martin v. Löwisce9b5a52001-06-27 06:28:56 +0000298{
Fredrik Lundh8f455852001-06-27 18:59:43 +0000299#ifdef Py_UNICODE_WIDE
Benjamin Peterson14339b62009-01-31 16:36:08 +0000300 return 0x10FFFF;
Martin v. Löwisce9b5a52001-06-27 06:28:56 +0000301#else
Benjamin Peterson14339b62009-01-31 16:36:08 +0000302 /* This is actually an illegal character, so it should
303 not be passed to unichr. */
304 return 0xFFFF;
Martin v. Löwisce9b5a52001-06-27 06:28:56 +0000305#endif
306}
307
Victor Stinner910337b2011-10-03 03:20:16 +0200308#ifdef Py_DEBUG
Victor Stinnerfb9ea8c2011-10-06 01:45:57 +0200309int
Victor Stinner7931d9a2011-11-04 00:22:48 +0100310_PyUnicode_CheckConsistency(PyObject *op, int check_content)
Victor Stinner910337b2011-10-03 03:20:16 +0200311{
312 PyASCIIObject *ascii;
313 unsigned int kind;
314
315 assert(PyUnicode_Check(op));
316
317 ascii = (PyASCIIObject *)op;
318 kind = ascii->state.kind;
319
Victor Stinnera3b334d2011-10-03 13:53:37 +0200320 if (ascii->state.ascii == 1 && ascii->state.compact == 1) {
Victor Stinner910337b2011-10-03 03:20:16 +0200321 assert(kind == PyUnicode_1BYTE_KIND);
Victor Stinner910337b2011-10-03 03:20:16 +0200322 assert(ascii->state.ready == 1);
323 }
Victor Stinnera41463c2011-10-04 01:05:08 +0200324 else {
Victor Stinner85041a52011-10-03 14:42:39 +0200325 PyCompactUnicodeObject *compact = (PyCompactUnicodeObject *)op;
Victor Stinner7f11ad42011-10-04 00:00:20 +0200326 void *data;
Victor Stinner910337b2011-10-03 03:20:16 +0200327
Victor Stinnera41463c2011-10-04 01:05:08 +0200328 if (ascii->state.compact == 1) {
329 data = compact + 1;
Victor Stinner910337b2011-10-03 03:20:16 +0200330 assert(kind == PyUnicode_1BYTE_KIND
331 || kind == PyUnicode_2BYTE_KIND
332 || kind == PyUnicode_4BYTE_KIND);
Victor Stinnera41463c2011-10-04 01:05:08 +0200333 assert(ascii->state.ascii == 0);
Victor Stinner910337b2011-10-03 03:20:16 +0200334 assert(ascii->state.ready == 1);
Victor Stinnera41463c2011-10-04 01:05:08 +0200335 assert (compact->utf8 != data);
Victor Stinnere30c0a12011-11-04 20:54:05 +0100336 }
337 else {
Victor Stinnera41463c2011-10-04 01:05:08 +0200338 PyUnicodeObject *unicode = (PyUnicodeObject *)op;
339
340 data = unicode->data.any;
341 if (kind == PyUnicode_WCHAR_KIND) {
Victor Stinnere30c0a12011-11-04 20:54:05 +0100342 assert(ascii->length == 0);
343 assert(ascii->hash == -1);
Victor Stinnera41463c2011-10-04 01:05:08 +0200344 assert(ascii->state.compact == 0);
345 assert(ascii->state.ascii == 0);
346 assert(ascii->state.ready == 0);
Victor Stinnere30c0a12011-11-04 20:54:05 +0100347 assert(ascii->state.interned == SSTATE_NOT_INTERNED);
Victor Stinnera41463c2011-10-04 01:05:08 +0200348 assert(ascii->wstr != NULL);
349 assert(data == NULL);
350 assert(compact->utf8 == NULL);
Victor Stinnera41463c2011-10-04 01:05:08 +0200351 }
352 else {
353 assert(kind == PyUnicode_1BYTE_KIND
354 || kind == PyUnicode_2BYTE_KIND
355 || kind == PyUnicode_4BYTE_KIND);
356 assert(ascii->state.compact == 0);
357 assert(ascii->state.ready == 1);
358 assert(data != NULL);
359 if (ascii->state.ascii) {
360 assert (compact->utf8 == data);
361 assert (compact->utf8_length == ascii->length);
362 }
363 else
364 assert (compact->utf8 != data);
365 }
366 }
367 if (kind != PyUnicode_WCHAR_KIND) {
Victor Stinner7f11ad42011-10-04 00:00:20 +0200368 if (
369#if SIZEOF_WCHAR_T == 2
370 kind == PyUnicode_2BYTE_KIND
371#else
372 kind == PyUnicode_4BYTE_KIND
373#endif
374 )
Victor Stinnera41463c2011-10-04 01:05:08 +0200375 {
376 assert(ascii->wstr == data);
377 assert(compact->wstr_length == ascii->length);
378 } else
379 assert(ascii->wstr != data);
Victor Stinner910337b2011-10-03 03:20:16 +0200380 }
Victor Stinnera41463c2011-10-04 01:05:08 +0200381
382 if (compact->utf8 == NULL)
383 assert(compact->utf8_length == 0);
384 if (ascii->wstr == NULL)
385 assert(compact->wstr_length == 0);
Victor Stinner910337b2011-10-03 03:20:16 +0200386 }
Victor Stinnerbb10a1f2011-10-05 01:34:17 +0200387 /* check that the best kind is used */
388 if (check_content && kind != PyUnicode_WCHAR_KIND)
389 {
390 Py_ssize_t i;
391 Py_UCS4 maxchar = 0;
Victor Stinner718fbf02012-04-26 00:39:37 +0200392 void *data;
393 Py_UCS4 ch;
394
395 data = PyUnicode_DATA(ascii);
Victor Stinnerbb10a1f2011-10-05 01:34:17 +0200396 for (i=0; i < ascii->length; i++)
397 {
Victor Stinner718fbf02012-04-26 00:39:37 +0200398 ch = PyUnicode_READ(kind, data, i);
Victor Stinnerbb10a1f2011-10-05 01:34:17 +0200399 if (ch > maxchar)
400 maxchar = ch;
401 }
402 if (kind == PyUnicode_1BYTE_KIND) {
Victor Stinner77faf692011-11-20 18:56:05 +0100403 if (ascii->state.ascii == 0) {
Victor Stinnerbb10a1f2011-10-05 01:34:17 +0200404 assert(maxchar >= 128);
Victor Stinner77faf692011-11-20 18:56:05 +0100405 assert(maxchar <= 255);
406 }
Victor Stinnerbb10a1f2011-10-05 01:34:17 +0200407 else
408 assert(maxchar < 128);
409 }
Victor Stinner77faf692011-11-20 18:56:05 +0100410 else if (kind == PyUnicode_2BYTE_KIND) {
Victor Stinnerbb10a1f2011-10-05 01:34:17 +0200411 assert(maxchar >= 0x100);
Victor Stinner77faf692011-11-20 18:56:05 +0100412 assert(maxchar <= 0xFFFF);
413 }
414 else {
Victor Stinnerbb10a1f2011-10-05 01:34:17 +0200415 assert(maxchar >= 0x10000);
Victor Stinner8faf8212011-12-08 22:14:11 +0100416 assert(maxchar <= MAX_UNICODE);
Victor Stinner77faf692011-11-20 18:56:05 +0100417 }
Victor Stinner718fbf02012-04-26 00:39:37 +0200418 assert(PyUnicode_READ(kind, data, ascii->length) == 0);
Victor Stinnerbb10a1f2011-10-05 01:34:17 +0200419 }
Benjamin Petersonccc51c12011-10-03 19:34:12 -0400420 return 1;
421}
Victor Stinner910337b2011-10-03 03:20:16 +0200422#endif
423
Victor Stinnerd3df8ab2011-11-22 01:22:34 +0100424static PyObject*
425unicode_result_wchar(PyObject *unicode)
426{
427#ifndef Py_DEBUG
428 Py_ssize_t len;
429
430 assert(Py_REFCNT(unicode) == 1);
431
432 len = _PyUnicode_WSTR_LENGTH(unicode);
433 if (len == 0) {
Victor Stinnerd3df8ab2011-11-22 01:22:34 +0100434 Py_DECREF(unicode);
Serhiy Storchaka678db842013-01-26 12:16:36 +0200435 _Py_RETURN_UNICODE_EMPTY();
Victor Stinnerd3df8ab2011-11-22 01:22:34 +0100436 }
437
438 if (len == 1) {
439 wchar_t ch = _PyUnicode_WSTR(unicode)[0];
Victor Stinnerd21b58c2013-02-26 00:15:54 +0100440 if ((Py_UCS4)ch < 256) {
Victor Stinnerd3df8ab2011-11-22 01:22:34 +0100441 PyObject *latin1_char = get_latin1_char((unsigned char)ch);
442 Py_DECREF(unicode);
443 return latin1_char;
444 }
445 }
446
447 if (_PyUnicode_Ready(unicode) < 0) {
448 Py_XDECREF(unicode);
449 return NULL;
450 }
451#else
452 /* don't make the result ready in debug mode to ensure that the caller
453 makes the string ready before using it */
454 assert(_PyUnicode_CheckConsistency(unicode, 1));
455#endif
456 return unicode;
457}
458
459static PyObject*
460unicode_result_ready(PyObject *unicode)
461{
462 Py_ssize_t length;
463
464 length = PyUnicode_GET_LENGTH(unicode);
465 if (length == 0) {
466 if (unicode != unicode_empty) {
Victor Stinnerd3df8ab2011-11-22 01:22:34 +0100467 Py_DECREF(unicode);
Serhiy Storchaka678db842013-01-26 12:16:36 +0200468 _Py_RETURN_UNICODE_EMPTY();
Victor Stinnerd3df8ab2011-11-22 01:22:34 +0100469 }
470 return unicode_empty;
471 }
472
473 if (length == 1) {
474 Py_UCS4 ch = PyUnicode_READ_CHAR(unicode, 0);
475 if (ch < 256) {
476 PyObject *latin1_char = unicode_latin1[ch];
477 if (latin1_char != NULL) {
478 if (unicode != latin1_char) {
479 Py_INCREF(latin1_char);
480 Py_DECREF(unicode);
481 }
482 return latin1_char;
483 }
484 else {
485 assert(_PyUnicode_CheckConsistency(unicode, 1));
486 Py_INCREF(unicode);
487 unicode_latin1[ch] = unicode;
488 return unicode;
489 }
490 }
491 }
492
493 assert(_PyUnicode_CheckConsistency(unicode, 1));
494 return unicode;
495}
496
497static PyObject*
498unicode_result(PyObject *unicode)
499{
500 assert(_PyUnicode_CHECK(unicode));
501 if (PyUnicode_IS_READY(unicode))
502 return unicode_result_ready(unicode);
503 else
504 return unicode_result_wchar(unicode);
505}
506
Victor Stinnerc4b49542011-12-11 22:44:26 +0100507static PyObject*
508unicode_result_unchanged(PyObject *unicode)
509{
510 if (PyUnicode_CheckExact(unicode)) {
Benjamin Petersonbac79492012-01-14 13:34:47 -0500511 if (PyUnicode_READY(unicode) == -1)
Victor Stinnerc4b49542011-12-11 22:44:26 +0100512 return NULL;
513 Py_INCREF(unicode);
514 return unicode;
515 }
516 else
517 /* Subtype -- return genuine unicode string with the same value. */
Victor Stinnerbf6e5602011-12-12 01:53:47 +0100518 return _PyUnicode_Copy(unicode);
Victor Stinnerc4b49542011-12-11 22:44:26 +0100519}
520
Victor Stinner3a50e702011-10-18 21:21:00 +0200521#ifdef HAVE_MBCS
522static OSVERSIONINFOEX winver;
523#endif
524
Thomas Wouters477c8d52006-05-27 19:21:47 +0000525/* --- Bloom Filters ----------------------------------------------------- */
526
527/* stuff to implement simple "bloom filters" for Unicode characters.
528 to keep things simple, we use a single bitmask, using the least 5
529 bits from each unicode characters as the bit index. */
530
531/* the linebreak mask is set up by Unicode_Init below */
532
Antoine Pitrouf068f942010-01-13 14:19:12 +0000533#if LONG_BIT >= 128
534#define BLOOM_WIDTH 128
535#elif LONG_BIT >= 64
536#define BLOOM_WIDTH 64
537#elif LONG_BIT >= 32
538#define BLOOM_WIDTH 32
539#else
540#error "LONG_BIT is smaller than 32"
541#endif
542
Thomas Wouters477c8d52006-05-27 19:21:47 +0000543#define BLOOM_MASK unsigned long
544
Serhiy Storchaka05997252013-01-26 12:14:02 +0200545static BLOOM_MASK bloom_linebreak = ~(BLOOM_MASK)0;
Thomas Wouters477c8d52006-05-27 19:21:47 +0000546
Antoine Pitrouf068f942010-01-13 14:19:12 +0000547#define BLOOM_ADD(mask, ch) ((mask |= (1UL << ((ch) & (BLOOM_WIDTH - 1)))))
548#define BLOOM(mask, ch) ((mask & (1UL << ((ch) & (BLOOM_WIDTH - 1)))))
Thomas Wouters477c8d52006-05-27 19:21:47 +0000549
Benjamin Peterson29060642009-01-31 22:14:21 +0000550#define BLOOM_LINEBREAK(ch) \
551 ((ch) < 128U ? ascii_linebreak[(ch)] : \
552 (BLOOM(bloom_linebreak, (ch)) && Py_UNICODE_ISLINEBREAK(ch)))
Thomas Wouters477c8d52006-05-27 19:21:47 +0000553
Alexander Belopolsky40018472011-02-26 01:02:56 +0000554Py_LOCAL_INLINE(BLOOM_MASK)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200555make_bloom_mask(int kind, void* ptr, Py_ssize_t len)
Thomas Wouters477c8d52006-05-27 19:21:47 +0000556{
557 /* calculate simple bloom-style bitmask for a given unicode string */
558
Antoine Pitrouf068f942010-01-13 14:19:12 +0000559 BLOOM_MASK mask;
Thomas Wouters477c8d52006-05-27 19:21:47 +0000560 Py_ssize_t i;
561
562 mask = 0;
563 for (i = 0; i < len; i++)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200564 BLOOM_ADD(mask, PyUnicode_READ(kind, ptr, i));
Thomas Wouters477c8d52006-05-27 19:21:47 +0000565
566 return mask;
567}
568
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200569#define BLOOM_MEMBER(mask, chr, str) \
570 (BLOOM(mask, chr) \
571 && (PyUnicode_FindChar(str, chr, 0, PyUnicode_GET_LENGTH(str), 1) >= 0))
Thomas Wouters477c8d52006-05-27 19:21:47 +0000572
Antoine Pitroudd4e2f02011-10-13 00:02:27 +0200573/* Compilation of templated routines */
574
575#include "stringlib/asciilib.h"
576#include "stringlib/fastsearch.h"
577#include "stringlib/partition.h"
578#include "stringlib/split.h"
579#include "stringlib/count.h"
580#include "stringlib/find.h"
581#include "stringlib/find_max_char.h"
582#include "stringlib/localeutil.h"
583#include "stringlib/undef.h"
584
585#include "stringlib/ucs1lib.h"
586#include "stringlib/fastsearch.h"
587#include "stringlib/partition.h"
588#include "stringlib/split.h"
589#include "stringlib/count.h"
590#include "stringlib/find.h"
591#include "stringlib/find_max_char.h"
592#include "stringlib/localeutil.h"
593#include "stringlib/undef.h"
594
595#include "stringlib/ucs2lib.h"
596#include "stringlib/fastsearch.h"
597#include "stringlib/partition.h"
598#include "stringlib/split.h"
599#include "stringlib/count.h"
600#include "stringlib/find.h"
601#include "stringlib/find_max_char.h"
602#include "stringlib/localeutil.h"
603#include "stringlib/undef.h"
604
605#include "stringlib/ucs4lib.h"
606#include "stringlib/fastsearch.h"
607#include "stringlib/partition.h"
608#include "stringlib/split.h"
609#include "stringlib/count.h"
610#include "stringlib/find.h"
611#include "stringlib/find_max_char.h"
612#include "stringlib/localeutil.h"
613#include "stringlib/undef.h"
614
Antoine Pitrouf0b934b2011-10-13 18:55:09 +0200615#include "stringlib/unicodedefs.h"
616#include "stringlib/fastsearch.h"
617#include "stringlib/count.h"
618#include "stringlib/find.h"
Antoine Pitrou0a3229d2011-11-21 20:39:13 +0100619#include "stringlib/undef.h"
Antoine Pitrouf0b934b2011-10-13 18:55:09 +0200620
Guido van Rossumd57fd912000-03-10 22:53:23 +0000621/* --- Unicode Object ----------------------------------------------------- */
622
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200623static PyObject *
Victor Stinner9310abb2011-10-05 00:59:23 +0200624fixup(PyObject *self, Py_UCS4 (*fixfct)(PyObject *s));
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200625
Antoine Pitrouf0b934b2011-10-13 18:55:09 +0200626Py_LOCAL_INLINE(Py_ssize_t) findchar(void *s, int kind,
627 Py_ssize_t size, Py_UCS4 ch,
628 int direction)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200629{
Antoine Pitrouf0b934b2011-10-13 18:55:09 +0200630 int mode = (direction == 1) ? FAST_SEARCH : FAST_RSEARCH;
631
632 switch (kind) {
633 case PyUnicode_1BYTE_KIND:
634 {
635 Py_UCS1 ch1 = (Py_UCS1) ch;
636 if (ch1 == ch)
637 return ucs1lib_fastsearch((Py_UCS1 *) s, size, &ch1, 1, 0, mode);
638 else
639 return -1;
640 }
641 case PyUnicode_2BYTE_KIND:
642 {
643 Py_UCS2 ch2 = (Py_UCS2) ch;
644 if (ch2 == ch)
645 return ucs2lib_fastsearch((Py_UCS2 *) s, size, &ch2, 1, 0, mode);
646 else
647 return -1;
648 }
649 case PyUnicode_4BYTE_KIND:
650 return ucs4lib_fastsearch((Py_UCS4 *) s, size, &ch, 1, 0, mode);
651 default:
652 assert(0);
653 return -1;
Victor Stinner9e7a1bc2011-10-13 00:18:12 +0200654 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200655}
656
Victor Stinnerfe226c02011-10-03 03:52:20 +0200657static PyObject*
658resize_compact(PyObject *unicode, Py_ssize_t length)
659{
660 Py_ssize_t char_size;
661 Py_ssize_t struct_size;
662 Py_ssize_t new_size;
663 int share_wstr;
Victor Stinner84def372011-12-11 20:04:56 +0100664 PyObject *new_unicode;
Victor Stinner79891572012-05-03 13:43:07 +0200665 assert(unicode_modifiable(unicode));
Victor Stinnerfe226c02011-10-03 03:52:20 +0200666 assert(PyUnicode_IS_READY(unicode));
Victor Stinner488fa492011-12-12 00:01:39 +0100667 assert(PyUnicode_IS_COMPACT(unicode));
668
Martin v. Löwisc47adb02011-10-07 20:55:35 +0200669 char_size = PyUnicode_KIND(unicode);
Victor Stinner488fa492011-12-12 00:01:39 +0100670 if (PyUnicode_IS_ASCII(unicode))
Victor Stinnerfe226c02011-10-03 03:52:20 +0200671 struct_size = sizeof(PyASCIIObject);
672 else
673 struct_size = sizeof(PyCompactUnicodeObject);
Victor Stinnerc379ead2011-10-03 12:52:27 +0200674 share_wstr = _PyUnicode_SHARE_WSTR(unicode);
Victor Stinnerfe226c02011-10-03 03:52:20 +0200675
Victor Stinnerfe226c02011-10-03 03:52:20 +0200676 if (length > ((PY_SSIZE_T_MAX - struct_size) / char_size - 1)) {
677 PyErr_NoMemory();
678 return NULL;
679 }
680 new_size = (struct_size + (length + 1) * char_size);
681
Victor Stinner84def372011-12-11 20:04:56 +0100682 _Py_DEC_REFTOTAL;
683 _Py_ForgetReference(unicode);
684
685 new_unicode = (PyObject *)PyObject_REALLOC((char *)unicode, new_size);
686 if (new_unicode == NULL) {
Victor Stinnerb0a82a62011-12-12 13:08:33 +0100687 _Py_NewReference(unicode);
Victor Stinnerfe226c02011-10-03 03:52:20 +0200688 PyErr_NoMemory();
689 return NULL;
690 }
Victor Stinner84def372011-12-11 20:04:56 +0100691 unicode = new_unicode;
Victor Stinnerfe226c02011-10-03 03:52:20 +0200692 _Py_NewReference(unicode);
Victor Stinner84def372011-12-11 20:04:56 +0100693
Victor Stinnerfe226c02011-10-03 03:52:20 +0200694 _PyUnicode_LENGTH(unicode) = length;
Victor Stinnerc379ead2011-10-03 12:52:27 +0200695 if (share_wstr) {
Victor Stinnerfe226c02011-10-03 03:52:20 +0200696 _PyUnicode_WSTR(unicode) = PyUnicode_DATA(unicode);
Victor Stinner488fa492011-12-12 00:01:39 +0100697 if (!PyUnicode_IS_ASCII(unicode))
Victor Stinnerc379ead2011-10-03 12:52:27 +0200698 _PyUnicode_WSTR_LENGTH(unicode) = length;
699 }
Victor Stinnerbbbac2e2013-02-07 23:12:46 +0100700 else if (_PyUnicode_HAS_WSTR_MEMORY(unicode)) {
701 PyObject_DEL(_PyUnicode_WSTR(unicode));
702 _PyUnicode_WSTR(unicode) = NULL;
703 }
Victor Stinnerfe226c02011-10-03 03:52:20 +0200704 PyUnicode_WRITE(PyUnicode_KIND(unicode), PyUnicode_DATA(unicode),
705 length, 0);
Victor Stinner79891572012-05-03 13:43:07 +0200706 assert(_PyUnicode_CheckConsistency(unicode, 0));
Victor Stinnerfe226c02011-10-03 03:52:20 +0200707 return unicode;
708}
709
Alexander Belopolsky40018472011-02-26 01:02:56 +0000710static int
Victor Stinner9db1a8b2011-10-23 20:04:37 +0200711resize_inplace(PyObject *unicode, Py_ssize_t length)
Guido van Rossumd57fd912000-03-10 22:53:23 +0000712{
Victor Stinner95663112011-10-04 01:03:50 +0200713 wchar_t *wstr;
Victor Stinner7a9105a2011-12-12 00:13:42 +0100714 Py_ssize_t new_size;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200715 assert(!PyUnicode_IS_COMPACT(unicode));
Victor Stinnerfe226c02011-10-03 03:52:20 +0200716 assert(Py_REFCNT(unicode) == 1);
Tim Petersced69f82003-09-16 20:30:58 +0000717
Victor Stinnerfe226c02011-10-03 03:52:20 +0200718 if (PyUnicode_IS_READY(unicode)) {
719 Py_ssize_t char_size;
Victor Stinner1c8d0c72011-10-03 12:11:00 +0200720 int share_wstr, share_utf8;
Victor Stinnerfe226c02011-10-03 03:52:20 +0200721 void *data;
722
723 data = _PyUnicode_DATA_ANY(unicode);
Martin v. Löwisc47adb02011-10-07 20:55:35 +0200724 char_size = PyUnicode_KIND(unicode);
Victor Stinnerc379ead2011-10-03 12:52:27 +0200725 share_wstr = _PyUnicode_SHARE_WSTR(unicode);
726 share_utf8 = _PyUnicode_SHARE_UTF8(unicode);
Victor Stinnerfe226c02011-10-03 03:52:20 +0200727
728 if (length > (PY_SSIZE_T_MAX / char_size - 1)) {
729 PyErr_NoMemory();
730 return -1;
731 }
732 new_size = (length + 1) * char_size;
733
Victor Stinner7a9105a2011-12-12 00:13:42 +0100734 if (!share_utf8 && _PyUnicode_HAS_UTF8_MEMORY(unicode))
735 {
736 PyObject_DEL(_PyUnicode_UTF8(unicode));
737 _PyUnicode_UTF8(unicode) = NULL;
738 _PyUnicode_UTF8_LENGTH(unicode) = 0;
739 }
740
Victor Stinnerfe226c02011-10-03 03:52:20 +0200741 data = (PyObject *)PyObject_REALLOC(data, new_size);
742 if (data == NULL) {
743 PyErr_NoMemory();
744 return -1;
745 }
746 _PyUnicode_DATA_ANY(unicode) = data;
Victor Stinnerc379ead2011-10-03 12:52:27 +0200747 if (share_wstr) {
Victor Stinnerfe226c02011-10-03 03:52:20 +0200748 _PyUnicode_WSTR(unicode) = data;
Victor Stinnerc379ead2011-10-03 12:52:27 +0200749 _PyUnicode_WSTR_LENGTH(unicode) = length;
750 }
751 if (share_utf8) {
Victor Stinner1c8d0c72011-10-03 12:11:00 +0200752 _PyUnicode_UTF8(unicode) = data;
Victor Stinnerc379ead2011-10-03 12:52:27 +0200753 _PyUnicode_UTF8_LENGTH(unicode) = length;
754 }
Victor Stinnerfe226c02011-10-03 03:52:20 +0200755 _PyUnicode_LENGTH(unicode) = length;
756 PyUnicode_WRITE(PyUnicode_KIND(unicode), data, length, 0);
Victor Stinner95663112011-10-04 01:03:50 +0200757 if (share_wstr || _PyUnicode_WSTR(unicode) == NULL) {
Victor Stinnerbb10a1f2011-10-05 01:34:17 +0200758 assert(_PyUnicode_CheckConsistency(unicode, 0));
Victor Stinnerfe226c02011-10-03 03:52:20 +0200759 return 0;
Victor Stinnerfe226c02011-10-03 03:52:20 +0200760 }
Victor Stinnerfe226c02011-10-03 03:52:20 +0200761 }
Victor Stinner95663112011-10-04 01:03:50 +0200762 assert(_PyUnicode_WSTR(unicode) != NULL);
763
764 /* check for integer overflow */
765 if (length > PY_SSIZE_T_MAX / sizeof(wchar_t) - 1) {
766 PyErr_NoMemory();
767 return -1;
768 }
Victor Stinner7a9105a2011-12-12 00:13:42 +0100769 new_size = sizeof(wchar_t) * (length + 1);
Victor Stinner95663112011-10-04 01:03:50 +0200770 wstr = _PyUnicode_WSTR(unicode);
Victor Stinner7a9105a2011-12-12 00:13:42 +0100771 wstr = PyObject_REALLOC(wstr, new_size);
Victor Stinner95663112011-10-04 01:03:50 +0200772 if (!wstr) {
773 PyErr_NoMemory();
774 return -1;
775 }
776 _PyUnicode_WSTR(unicode) = wstr;
777 _PyUnicode_WSTR(unicode)[length] = 0;
778 _PyUnicode_WSTR_LENGTH(unicode) = length;
Victor Stinnerbb10a1f2011-10-05 01:34:17 +0200779 assert(_PyUnicode_CheckConsistency(unicode, 0));
Guido van Rossumd57fd912000-03-10 22:53:23 +0000780 return 0;
781}
782
Victor Stinnerfe226c02011-10-03 03:52:20 +0200783static PyObject*
784resize_copy(PyObject *unicode, Py_ssize_t length)
785{
786 Py_ssize_t copy_length;
Victor Stinner7a9105a2011-12-12 00:13:42 +0100787 if (_PyUnicode_KIND(unicode) != PyUnicode_WCHAR_KIND) {
Victor Stinnerfe226c02011-10-03 03:52:20 +0200788 PyObject *copy;
Victor Stinner7a9105a2011-12-12 00:13:42 +0100789
Benjamin Petersonbac79492012-01-14 13:34:47 -0500790 if (PyUnicode_READY(unicode) == -1)
Victor Stinner7a9105a2011-12-12 00:13:42 +0100791 return NULL;
Victor Stinnerfe226c02011-10-03 03:52:20 +0200792
793 copy = PyUnicode_New(length, PyUnicode_MAX_CHAR_VALUE(unicode));
794 if (copy == NULL)
795 return NULL;
796
797 copy_length = Py_MIN(length, PyUnicode_GET_LENGTH(unicode));
Victor Stinnerd3f08822012-05-29 12:57:52 +0200798 _PyUnicode_FastCopyCharacters(copy, 0, unicode, 0, copy_length);
Victor Stinnerfe226c02011-10-03 03:52:20 +0200799 return copy;
Victor Stinner8cfcbed2011-10-03 23:19:21 +0200800 }
801 else {
Victor Stinner9db1a8b2011-10-23 20:04:37 +0200802 PyObject *w;
Victor Stinner7a9105a2011-12-12 00:13:42 +0100803
Victor Stinner9db1a8b2011-10-23 20:04:37 +0200804 w = (PyObject*)_PyUnicode_New(length);
Victor Stinnerfe226c02011-10-03 03:52:20 +0200805 if (w == NULL)
806 return NULL;
807 copy_length = _PyUnicode_WSTR_LENGTH(unicode);
808 copy_length = Py_MIN(copy_length, length);
809 Py_UNICODE_COPY(_PyUnicode_WSTR(w), _PyUnicode_WSTR(unicode),
810 copy_length);
Victor Stinner9db1a8b2011-10-23 20:04:37 +0200811 return w;
Victor Stinnerfe226c02011-10-03 03:52:20 +0200812 }
813}
814
Guido van Rossumd57fd912000-03-10 22:53:23 +0000815/* We allocate one more byte to make sure the string is
Martin v. Löwis47383402007-08-15 07:32:56 +0000816 Ux0000 terminated; some code (e.g. new_identifier)
817 relies on that.
Guido van Rossumd57fd912000-03-10 22:53:23 +0000818
819 XXX This allocator could further be enhanced by assuring that the
Benjamin Peterson29060642009-01-31 22:14:21 +0000820 free list never reduces its size below 1.
Guido van Rossumd57fd912000-03-10 22:53:23 +0000821
822*/
823
Alexander Belopolsky40018472011-02-26 01:02:56 +0000824static PyUnicodeObject *
825_PyUnicode_New(Py_ssize_t length)
Guido van Rossumd57fd912000-03-10 22:53:23 +0000826{
827 register PyUnicodeObject *unicode;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200828 size_t new_size;
Guido van Rossumd57fd912000-03-10 22:53:23 +0000829
Thomas Wouters477c8d52006-05-27 19:21:47 +0000830 /* Optimization for empty strings */
Guido van Rossumd57fd912000-03-10 22:53:23 +0000831 if (length == 0 && unicode_empty != NULL) {
832 Py_INCREF(unicode_empty);
Victor Stinnera464fc12011-10-02 20:39:30 +0200833 return (PyUnicodeObject*)unicode_empty;
Guido van Rossumd57fd912000-03-10 22:53:23 +0000834 }
835
Neal Norwitz3ce5d922008-08-24 07:08:55 +0000836 /* Ensure we won't overflow the size. */
837 if (length > ((PY_SSIZE_T_MAX / sizeof(Py_UNICODE)) - 1)) {
838 return (PyUnicodeObject *)PyErr_NoMemory();
839 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200840 if (length < 0) {
841 PyErr_SetString(PyExc_SystemError,
842 "Negative size passed to _PyUnicode_New");
843 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +0000844 }
845
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200846 unicode = PyObject_New(PyUnicodeObject, &PyUnicode_Type);
847 if (unicode == NULL)
848 return NULL;
849 new_size = sizeof(Py_UNICODE) * ((size_t)length + 1);
850 _PyUnicode_WSTR(unicode) = (Py_UNICODE*) PyObject_MALLOC(new_size);
851 if (!_PyUnicode_WSTR(unicode)) {
Victor Stinnerb0a82a62011-12-12 13:08:33 +0100852 Py_DECREF(unicode);
Benjamin Peterson29060642009-01-31 22:14:21 +0000853 PyErr_NoMemory();
Victor Stinnerb0a82a62011-12-12 13:08:33 +0100854 return NULL;
Guido van Rossum3c1bb802000-04-27 20:13:50 +0000855 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200856
Jeremy Hyltond8082792003-09-16 19:41:39 +0000857 /* Initialize the first element to guard against cases where
Tim Petersced69f82003-09-16 20:30:58 +0000858 * the caller fails before initializing str -- unicode_resize()
859 * reads str[0], and the Keep-Alive optimization can keep memory
860 * allocated for str alive across a call to unicode_dealloc(unicode).
861 * We don't want unicode_resize to read uninitialized memory in
862 * that case.
863 */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200864 _PyUnicode_WSTR(unicode)[0] = 0;
865 _PyUnicode_WSTR(unicode)[length] = 0;
866 _PyUnicode_WSTR_LENGTH(unicode) = length;
867 _PyUnicode_HASH(unicode) = -1;
868 _PyUnicode_STATE(unicode).interned = 0;
869 _PyUnicode_STATE(unicode).kind = 0;
870 _PyUnicode_STATE(unicode).compact = 0;
871 _PyUnicode_STATE(unicode).ready = 0;
872 _PyUnicode_STATE(unicode).ascii = 0;
Victor Stinnerc3c74152011-10-02 20:39:55 +0200873 _PyUnicode_DATA_ANY(unicode) = NULL;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200874 _PyUnicode_LENGTH(unicode) = 0;
Victor Stinnere90fe6a2011-10-01 16:48:13 +0200875 _PyUnicode_UTF8(unicode) = NULL;
876 _PyUnicode_UTF8_LENGTH(unicode) = 0;
Victor Stinner7931d9a2011-11-04 00:22:48 +0100877 assert(_PyUnicode_CheckConsistency((PyObject *)unicode, 0));
Guido van Rossumd57fd912000-03-10 22:53:23 +0000878 return unicode;
879}
880
Victor Stinnerf42dc442011-10-02 23:33:16 +0200881static const char*
882unicode_kind_name(PyObject *unicode)
883{
Victor Stinner42dfd712011-10-03 14:41:45 +0200884 /* don't check consistency: unicode_kind_name() is called from
885 _PyUnicode_Dump() */
Victor Stinnerf42dc442011-10-02 23:33:16 +0200886 if (!PyUnicode_IS_COMPACT(unicode))
887 {
888 if (!PyUnicode_IS_READY(unicode))
889 return "wstr";
Benjamin Petersonead6b532011-12-20 17:23:42 -0600890 switch (PyUnicode_KIND(unicode))
Victor Stinnerf42dc442011-10-02 23:33:16 +0200891 {
892 case PyUnicode_1BYTE_KIND:
Victor Stinnera3b334d2011-10-03 13:53:37 +0200893 if (PyUnicode_IS_ASCII(unicode))
Victor Stinnerf42dc442011-10-02 23:33:16 +0200894 return "legacy ascii";
895 else
896 return "legacy latin1";
897 case PyUnicode_2BYTE_KIND:
898 return "legacy UCS2";
899 case PyUnicode_4BYTE_KIND:
900 return "legacy UCS4";
901 default:
902 return "<legacy invalid kind>";
903 }
904 }
905 assert(PyUnicode_IS_READY(unicode));
Benjamin Petersonead6b532011-12-20 17:23:42 -0600906 switch (PyUnicode_KIND(unicode)) {
Victor Stinnerf42dc442011-10-02 23:33:16 +0200907 case PyUnicode_1BYTE_KIND:
Victor Stinnera3b334d2011-10-03 13:53:37 +0200908 if (PyUnicode_IS_ASCII(unicode))
Victor Stinnerf42dc442011-10-02 23:33:16 +0200909 return "ascii";
910 else
Victor Stinnera3b334d2011-10-03 13:53:37 +0200911 return "latin1";
Victor Stinnerf42dc442011-10-02 23:33:16 +0200912 case PyUnicode_2BYTE_KIND:
Victor Stinnera3b334d2011-10-03 13:53:37 +0200913 return "UCS2";
Victor Stinnerf42dc442011-10-02 23:33:16 +0200914 case PyUnicode_4BYTE_KIND:
Victor Stinnera3b334d2011-10-03 13:53:37 +0200915 return "UCS4";
Victor Stinnerf42dc442011-10-02 23:33:16 +0200916 default:
917 return "<invalid compact kind>";
918 }
919}
920
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200921#ifdef Py_DEBUG
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200922/* Functions wrapping macros for use in debugger */
923char *_PyUnicode_utf8(void *unicode){
Victor Stinnere90fe6a2011-10-01 16:48:13 +0200924 return PyUnicode_UTF8(unicode);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200925}
926
927void *_PyUnicode_compact_data(void *unicode) {
928 return _PyUnicode_COMPACT_DATA(unicode);
929}
930void *_PyUnicode_data(void *unicode){
931 printf("obj %p\n", unicode);
932 printf("compact %d\n", PyUnicode_IS_COMPACT(unicode));
933 printf("compact ascii %d\n", PyUnicode_IS_COMPACT_ASCII(unicode));
934 printf("ascii op %p\n", ((void*)((PyASCIIObject*)(unicode) + 1)));
935 printf("compact op %p\n", ((void*)((PyCompactUnicodeObject*)(unicode) + 1)));
936 printf("compact data %p\n", _PyUnicode_COMPACT_DATA(unicode));
937 return PyUnicode_DATA(unicode);
938}
Victor Stinnerfe0c1552011-10-03 02:59:31 +0200939
940void
941_PyUnicode_Dump(PyObject *op)
942{
943 PyASCIIObject *ascii = (PyASCIIObject *)op;
Victor Stinnera849a4b2011-10-03 12:12:11 +0200944 PyCompactUnicodeObject *compact = (PyCompactUnicodeObject *)op;
945 PyUnicodeObject *unicode = (PyUnicodeObject *)op;
946 void *data;
Victor Stinner0d60e872011-10-23 19:47:19 +0200947
Victor Stinnera849a4b2011-10-03 12:12:11 +0200948 if (ascii->state.compact)
Victor Stinner0d60e872011-10-23 19:47:19 +0200949 {
950 if (ascii->state.ascii)
951 data = (ascii + 1);
952 else
953 data = (compact + 1);
954 }
Victor Stinnera849a4b2011-10-03 12:12:11 +0200955 else
956 data = unicode->data.any;
Victor Stinner0d60e872011-10-23 19:47:19 +0200957 printf("%s: len=%zu, ",unicode_kind_name(op), ascii->length);
958
Victor Stinnera849a4b2011-10-03 12:12:11 +0200959 if (ascii->wstr == data)
960 printf("shared ");
961 printf("wstr=%p", ascii->wstr);
Victor Stinner0d60e872011-10-23 19:47:19 +0200962
Victor Stinnera3b334d2011-10-03 13:53:37 +0200963 if (!(ascii->state.ascii == 1 && ascii->state.compact == 1)) {
Victor Stinnera849a4b2011-10-03 12:12:11 +0200964 printf(" (%zu), ", compact->wstr_length);
965 if (!ascii->state.compact && compact->utf8 == unicode->data.any)
966 printf("shared ");
967 printf("utf8=%p (%zu)", compact->utf8, compact->utf8_length);
Victor Stinnerfe0c1552011-10-03 02:59:31 +0200968 }
Victor Stinnera849a4b2011-10-03 12:12:11 +0200969 printf(", data=%p\n", data);
Victor Stinnerfe0c1552011-10-03 02:59:31 +0200970}
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200971#endif
972
973PyObject *
974PyUnicode_New(Py_ssize_t size, Py_UCS4 maxchar)
975{
976 PyObject *obj;
977 PyCompactUnicodeObject *unicode;
978 void *data;
Victor Stinner8f825062012-04-27 13:55:39 +0200979 enum PyUnicode_Kind kind;
Victor Stinner9e9d6892011-10-04 01:02:02 +0200980 int is_sharing, is_ascii;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200981 Py_ssize_t char_size;
982 Py_ssize_t struct_size;
983
984 /* Optimization for empty strings */
985 if (size == 0 && unicode_empty != NULL) {
986 Py_INCREF(unicode_empty);
Victor Stinnera464fc12011-10-02 20:39:30 +0200987 return unicode_empty;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200988 }
989
Victor Stinner9e9d6892011-10-04 01:02:02 +0200990 is_ascii = 0;
991 is_sharing = 0;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200992 struct_size = sizeof(PyCompactUnicodeObject);
993 if (maxchar < 128) {
Victor Stinner8f825062012-04-27 13:55:39 +0200994 kind = PyUnicode_1BYTE_KIND;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200995 char_size = 1;
996 is_ascii = 1;
997 struct_size = sizeof(PyASCIIObject);
998 }
999 else if (maxchar < 256) {
Victor Stinner8f825062012-04-27 13:55:39 +02001000 kind = PyUnicode_1BYTE_KIND;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001001 char_size = 1;
1002 }
1003 else if (maxchar < 65536) {
Victor Stinner8f825062012-04-27 13:55:39 +02001004 kind = PyUnicode_2BYTE_KIND;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001005 char_size = 2;
1006 if (sizeof(wchar_t) == 2)
1007 is_sharing = 1;
1008 }
1009 else {
Victor Stinnerc9590ad2012-03-04 01:34:37 +01001010 if (maxchar > MAX_UNICODE) {
1011 PyErr_SetString(PyExc_SystemError,
1012 "invalid maximum character passed to PyUnicode_New");
1013 return NULL;
1014 }
Victor Stinner8f825062012-04-27 13:55:39 +02001015 kind = PyUnicode_4BYTE_KIND;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001016 char_size = 4;
1017 if (sizeof(wchar_t) == 4)
1018 is_sharing = 1;
1019 }
1020
1021 /* Ensure we won't overflow the size. */
1022 if (size < 0) {
1023 PyErr_SetString(PyExc_SystemError,
1024 "Negative size passed to PyUnicode_New");
1025 return NULL;
1026 }
1027 if (size > ((PY_SSIZE_T_MAX - struct_size) / char_size - 1))
1028 return PyErr_NoMemory();
1029
1030 /* Duplicated allocation code from _PyObject_New() instead of a call to
1031 * PyObject_New() so we are able to allocate space for the object and
1032 * it's data buffer.
1033 */
1034 obj = (PyObject *) PyObject_MALLOC(struct_size + (size + 1) * char_size);
1035 if (obj == NULL)
1036 return PyErr_NoMemory();
1037 obj = PyObject_INIT(obj, &PyUnicode_Type);
1038 if (obj == NULL)
1039 return NULL;
1040
1041 unicode = (PyCompactUnicodeObject *)obj;
1042 if (is_ascii)
1043 data = ((PyASCIIObject*)obj) + 1;
1044 else
1045 data = unicode + 1;
1046 _PyUnicode_LENGTH(unicode) = size;
1047 _PyUnicode_HASH(unicode) = -1;
1048 _PyUnicode_STATE(unicode).interned = 0;
Victor Stinner8f825062012-04-27 13:55:39 +02001049 _PyUnicode_STATE(unicode).kind = kind;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001050 _PyUnicode_STATE(unicode).compact = 1;
1051 _PyUnicode_STATE(unicode).ready = 1;
1052 _PyUnicode_STATE(unicode).ascii = is_ascii;
1053 if (is_ascii) {
1054 ((char*)data)[size] = 0;
1055 _PyUnicode_WSTR(unicode) = NULL;
1056 }
Victor Stinner8f825062012-04-27 13:55:39 +02001057 else if (kind == PyUnicode_1BYTE_KIND) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001058 ((char*)data)[size] = 0;
1059 _PyUnicode_WSTR(unicode) = NULL;
1060 _PyUnicode_WSTR_LENGTH(unicode) = 0;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001061 unicode->utf8 = NULL;
Victor Stinner9e9d6892011-10-04 01:02:02 +02001062 unicode->utf8_length = 0;
Victor Stinner8f825062012-04-27 13:55:39 +02001063 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001064 else {
1065 unicode->utf8 = NULL;
Victor Stinner9e9d6892011-10-04 01:02:02 +02001066 unicode->utf8_length = 0;
Victor Stinner8f825062012-04-27 13:55:39 +02001067 if (kind == PyUnicode_2BYTE_KIND)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001068 ((Py_UCS2*)data)[size] = 0;
Victor Stinner8f825062012-04-27 13:55:39 +02001069 else /* kind == PyUnicode_4BYTE_KIND */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001070 ((Py_UCS4*)data)[size] = 0;
1071 if (is_sharing) {
1072 _PyUnicode_WSTR_LENGTH(unicode) = size;
1073 _PyUnicode_WSTR(unicode) = (wchar_t *)data;
1074 }
1075 else {
1076 _PyUnicode_WSTR_LENGTH(unicode) = 0;
1077 _PyUnicode_WSTR(unicode) = NULL;
1078 }
1079 }
Victor Stinner8f825062012-04-27 13:55:39 +02001080#ifdef Py_DEBUG
1081 /* Fill the data with invalid characters to detect bugs earlier.
1082 _PyUnicode_CheckConsistency(str, 1) detects invalid characters,
1083 at least for ASCII and UCS-4 strings. U+00FF is invalid in ASCII
1084 and U+FFFFFFFF is an invalid character in Unicode 6.0. */
1085 memset(data, 0xff, size * kind);
1086#endif
Victor Stinner7931d9a2011-11-04 00:22:48 +01001087 assert(_PyUnicode_CheckConsistency((PyObject*)unicode, 0));
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001088 return obj;
1089}
1090
1091#if SIZEOF_WCHAR_T == 2
1092/* Helper function to convert a 16-bits wchar_t representation to UCS4, this
1093 will decode surrogate pairs, the other conversions are implemented as macros
Georg Brandl7597add2011-10-05 16:36:47 +02001094 for efficiency.
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001095
1096 This function assumes that unicode can hold one more code point than wstr
1097 characters for a terminating null character. */
Victor Stinnerc53be962011-10-02 21:33:54 +02001098static void
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001099unicode_convert_wchar_to_ucs4(const wchar_t *begin, const wchar_t *end,
Victor Stinner9db1a8b2011-10-23 20:04:37 +02001100 PyObject *unicode)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001101{
1102 const wchar_t *iter;
1103 Py_UCS4 *ucs4_out;
1104
Victor Stinner910337b2011-10-03 03:20:16 +02001105 assert(unicode != NULL);
1106 assert(_PyUnicode_CHECK(unicode));
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001107 assert(_PyUnicode_KIND(unicode) == PyUnicode_4BYTE_KIND);
1108 ucs4_out = PyUnicode_4BYTE_DATA(unicode);
1109
1110 for (iter = begin; iter < end; ) {
1111 assert(ucs4_out < (PyUnicode_4BYTE_DATA(unicode) +
1112 _PyUnicode_GET_LENGTH(unicode)));
Victor Stinner551ac952011-11-29 22:58:13 +01001113 if (Py_UNICODE_IS_HIGH_SURROGATE(iter[0])
1114 && (iter+1) < end
1115 && Py_UNICODE_IS_LOW_SURROGATE(iter[1]))
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001116 {
Victor Stinner551ac952011-11-29 22:58:13 +01001117 *ucs4_out++ = Py_UNICODE_JOIN_SURROGATES(iter[0], iter[1]);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001118 iter += 2;
1119 }
1120 else {
1121 *ucs4_out++ = *iter;
1122 iter++;
1123 }
1124 }
1125 assert(ucs4_out == (PyUnicode_4BYTE_DATA(unicode) +
1126 _PyUnicode_GET_LENGTH(unicode)));
1127
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001128}
1129#endif
1130
Victor Stinnercd9950f2011-10-02 00:34:53 +02001131static int
Victor Stinner488fa492011-12-12 00:01:39 +01001132unicode_check_modifiable(PyObject *unicode)
Victor Stinnercd9950f2011-10-02 00:34:53 +02001133{
Victor Stinner488fa492011-12-12 00:01:39 +01001134 if (!unicode_modifiable(unicode)) {
Victor Stinner01698042011-10-04 00:04:26 +02001135 PyErr_SetString(PyExc_SystemError,
Victor Stinner488fa492011-12-12 00:01:39 +01001136 "Cannot modify a string currently used");
Victor Stinnercd9950f2011-10-02 00:34:53 +02001137 return -1;
1138 }
Victor Stinnercd9950f2011-10-02 00:34:53 +02001139 return 0;
1140}
1141
Victor Stinnerfb9ea8c2011-10-06 01:45:57 +02001142static int
1143_copy_characters(PyObject *to, Py_ssize_t to_start,
1144 PyObject *from, Py_ssize_t from_start,
1145 Py_ssize_t how_many, int check_maxchar)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001146{
Victor Stinnera0702ab2011-09-29 14:14:38 +02001147 unsigned int from_kind, to_kind;
1148 void *from_data, *to_data;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001149
Victor Stinneree4544c2012-05-09 22:24:08 +02001150 assert(0 <= how_many);
1151 assert(0 <= from_start);
1152 assert(0 <= to_start);
Victor Stinnerfb9ea8c2011-10-06 01:45:57 +02001153 assert(PyUnicode_Check(from));
Victor Stinnerfb9ea8c2011-10-06 01:45:57 +02001154 assert(PyUnicode_IS_READY(from));
Victor Stinneree4544c2012-05-09 22:24:08 +02001155 assert(from_start + how_many <= PyUnicode_GET_LENGTH(from));
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001156
Victor Stinnerd3f08822012-05-29 12:57:52 +02001157 assert(PyUnicode_Check(to));
1158 assert(PyUnicode_IS_READY(to));
1159 assert(to_start + how_many <= PyUnicode_GET_LENGTH(to));
1160
Victor Stinnerc9d369f2012-06-16 02:22:37 +02001161 if (how_many == 0)
1162 return 0;
1163
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001164 from_kind = PyUnicode_KIND(from);
Victor Stinnera0702ab2011-09-29 14:14:38 +02001165 from_data = PyUnicode_DATA(from);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001166 to_kind = PyUnicode_KIND(to);
Victor Stinnera0702ab2011-09-29 14:14:38 +02001167 to_data = PyUnicode_DATA(to);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001168
Victor Stinnerf1852262012-06-16 16:38:26 +02001169#ifdef Py_DEBUG
1170 if (!check_maxchar
1171 && PyUnicode_MAX_CHAR_VALUE(from) > PyUnicode_MAX_CHAR_VALUE(to))
1172 {
1173 const Py_UCS4 to_maxchar = PyUnicode_MAX_CHAR_VALUE(to);
1174 Py_UCS4 ch;
1175 Py_ssize_t i;
1176 for (i=0; i < how_many; i++) {
1177 ch = PyUnicode_READ(from_kind, from_data, from_start + i);
1178 assert(ch <= to_maxchar);
1179 }
1180 }
1181#endif
1182
Victor Stinnerc9d369f2012-06-16 02:22:37 +02001183 if (from_kind == to_kind) {
Victor Stinnerf1852262012-06-16 16:38:26 +02001184 if (check_maxchar
1185 && !PyUnicode_IS_ASCII(from) && PyUnicode_IS_ASCII(to))
1186 {
Victor Stinnerc9d369f2012-06-16 02:22:37 +02001187 /* Writing Latin-1 characters into an ASCII string requires to
1188 check that all written characters are pure ASCII */
Victor Stinnerf1852262012-06-16 16:38:26 +02001189 Py_UCS4 max_char;
1190 max_char = ucs1lib_find_max_char(from_data,
1191 (Py_UCS1*)from_data + how_many);
1192 if (max_char >= 128)
1193 return -1;
Victor Stinnerc9d369f2012-06-16 02:22:37 +02001194 }
Martin v. Löwisc47adb02011-10-07 20:55:35 +02001195 Py_MEMCPY((char*)to_data + to_kind * to_start,
1196 (char*)from_data + from_kind * from_start,
1197 to_kind * how_many);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001198 }
Victor Stinnera0702ab2011-09-29 14:14:38 +02001199 else if (from_kind == PyUnicode_1BYTE_KIND
1200 && to_kind == PyUnicode_2BYTE_KIND)
Victor Stinnerbe78eaf2011-09-28 21:37:03 +02001201 {
1202 _PyUnicode_CONVERT_BYTES(
1203 Py_UCS1, Py_UCS2,
1204 PyUnicode_1BYTE_DATA(from) + from_start,
1205 PyUnicode_1BYTE_DATA(from) + from_start + how_many,
1206 PyUnicode_2BYTE_DATA(to) + to_start
1207 );
Victor Stinnerbe78eaf2011-09-28 21:37:03 +02001208 }
Victor Stinner157f83f2011-09-28 21:41:31 +02001209 else if (from_kind == PyUnicode_1BYTE_KIND
Victor Stinnerbe78eaf2011-09-28 21:37:03 +02001210 && to_kind == PyUnicode_4BYTE_KIND)
1211 {
1212 _PyUnicode_CONVERT_BYTES(
1213 Py_UCS1, Py_UCS4,
1214 PyUnicode_1BYTE_DATA(from) + from_start,
1215 PyUnicode_1BYTE_DATA(from) + from_start + how_many,
1216 PyUnicode_4BYTE_DATA(to) + to_start
1217 );
Victor Stinnerbe78eaf2011-09-28 21:37:03 +02001218 }
1219 else if (from_kind == PyUnicode_2BYTE_KIND
1220 && to_kind == PyUnicode_4BYTE_KIND)
1221 {
1222 _PyUnicode_CONVERT_BYTES(
1223 Py_UCS2, Py_UCS4,
1224 PyUnicode_2BYTE_DATA(from) + from_start,
1225 PyUnicode_2BYTE_DATA(from) + from_start + how_many,
1226 PyUnicode_4BYTE_DATA(to) + to_start
1227 );
Victor Stinnerbe78eaf2011-09-28 21:37:03 +02001228 }
Victor Stinnera0702ab2011-09-29 14:14:38 +02001229 else {
Victor Stinnerc9d369f2012-06-16 02:22:37 +02001230 assert (PyUnicode_MAX_CHAR_VALUE(from) > PyUnicode_MAX_CHAR_VALUE(to));
1231
Victor Stinnerc9d369f2012-06-16 02:22:37 +02001232 if (!check_maxchar) {
1233 if (from_kind == PyUnicode_2BYTE_KIND
1234 && to_kind == PyUnicode_1BYTE_KIND)
1235 {
1236 _PyUnicode_CONVERT_BYTES(
1237 Py_UCS2, Py_UCS1,
1238 PyUnicode_2BYTE_DATA(from) + from_start,
1239 PyUnicode_2BYTE_DATA(from) + from_start + how_many,
1240 PyUnicode_1BYTE_DATA(to) + to_start
1241 );
1242 }
1243 else if (from_kind == PyUnicode_4BYTE_KIND
1244 && to_kind == PyUnicode_1BYTE_KIND)
1245 {
1246 _PyUnicode_CONVERT_BYTES(
1247 Py_UCS4, Py_UCS1,
1248 PyUnicode_4BYTE_DATA(from) + from_start,
1249 PyUnicode_4BYTE_DATA(from) + from_start + how_many,
1250 PyUnicode_1BYTE_DATA(to) + to_start
1251 );
1252 }
1253 else if (from_kind == PyUnicode_4BYTE_KIND
1254 && to_kind == PyUnicode_2BYTE_KIND)
1255 {
1256 _PyUnicode_CONVERT_BYTES(
1257 Py_UCS4, Py_UCS2,
1258 PyUnicode_4BYTE_DATA(from) + from_start,
1259 PyUnicode_4BYTE_DATA(from) + from_start + how_many,
1260 PyUnicode_2BYTE_DATA(to) + to_start
1261 );
1262 }
1263 else {
1264 assert(0);
1265 return -1;
1266 }
1267 }
Victor Stinnerf1852262012-06-16 16:38:26 +02001268 else {
Victor Stinnera0702ab2011-09-29 14:14:38 +02001269 const Py_UCS4 to_maxchar = PyUnicode_MAX_CHAR_VALUE(to);
Victor Stinnerfb9ea8c2011-10-06 01:45:57 +02001270 Py_UCS4 ch;
Victor Stinnera0702ab2011-09-29 14:14:38 +02001271 Py_ssize_t i;
1272
Victor Stinnera0702ab2011-09-29 14:14:38 +02001273 for (i=0; i < how_many; i++) {
1274 ch = PyUnicode_READ(from_kind, from_data, from_start + i);
Victor Stinnerc9d369f2012-06-16 02:22:37 +02001275 if (ch > to_maxchar)
1276 return -1;
Victor Stinnera0702ab2011-09-29 14:14:38 +02001277 PyUnicode_WRITE(to_kind, to_data, to_start + i, ch);
1278 }
Victor Stinnera0702ab2011-09-29 14:14:38 +02001279 }
1280 }
Victor Stinnerfb9ea8c2011-10-06 01:45:57 +02001281 return 0;
1282}
1283
Victor Stinnerd3f08822012-05-29 12:57:52 +02001284void
1285_PyUnicode_FastCopyCharacters(
1286 PyObject *to, Py_ssize_t to_start,
1287 PyObject *from, Py_ssize_t from_start, Py_ssize_t how_many)
Victor Stinnerfb9ea8c2011-10-06 01:45:57 +02001288{
1289 (void)_copy_characters(to, to_start, from, from_start, how_many, 0);
1290}
1291
1292Py_ssize_t
1293PyUnicode_CopyCharacters(PyObject *to, Py_ssize_t to_start,
1294 PyObject *from, Py_ssize_t from_start,
1295 Py_ssize_t how_many)
1296{
1297 int err;
1298
1299 if (!PyUnicode_Check(from) || !PyUnicode_Check(to)) {
1300 PyErr_BadInternalCall();
1301 return -1;
1302 }
1303
Benjamin Petersonbac79492012-01-14 13:34:47 -05001304 if (PyUnicode_READY(from) == -1)
Victor Stinnerfb9ea8c2011-10-06 01:45:57 +02001305 return -1;
Benjamin Petersonbac79492012-01-14 13:34:47 -05001306 if (PyUnicode_READY(to) == -1)
Victor Stinnerfb9ea8c2011-10-06 01:45:57 +02001307 return -1;
1308
Victor Stinnerd3f08822012-05-29 12:57:52 +02001309 if (from_start < 0) {
1310 PyErr_SetString(PyExc_IndexError, "string index out of range");
1311 return -1;
1312 }
1313 if (to_start < 0) {
1314 PyErr_SetString(PyExc_IndexError, "string index out of range");
1315 return -1;
1316 }
Victor Stinnerfb9ea8c2011-10-06 01:45:57 +02001317 how_many = Py_MIN(PyUnicode_GET_LENGTH(from), how_many);
1318 if (to_start + how_many > PyUnicode_GET_LENGTH(to)) {
1319 PyErr_Format(PyExc_SystemError,
1320 "Cannot write %zi characters at %zi "
1321 "in a string of %zi characters",
1322 how_many, to_start, PyUnicode_GET_LENGTH(to));
1323 return -1;
1324 }
1325
1326 if (how_many == 0)
1327 return 0;
1328
Victor Stinner488fa492011-12-12 00:01:39 +01001329 if (unicode_check_modifiable(to))
Victor Stinnerfb9ea8c2011-10-06 01:45:57 +02001330 return -1;
1331
1332 err = _copy_characters(to, to_start, from, from_start, how_many, 1);
1333 if (err) {
1334 PyErr_Format(PyExc_SystemError,
1335 "Cannot copy %s characters "
1336 "into a string of %s characters",
1337 unicode_kind_name(from),
1338 unicode_kind_name(to));
1339 return -1;
1340 }
Victor Stinnera0702ab2011-09-29 14:14:38 +02001341 return how_many;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001342}
1343
Victor Stinner17222162011-09-28 22:15:37 +02001344/* Find the maximum code point and count the number of surrogate pairs so a
1345 correct string length can be computed before converting a string to UCS4.
1346 This function counts single surrogates as a character and not as a pair.
1347
1348 Return 0 on success, or -1 on error. */
1349static int
1350find_maxchar_surrogates(const wchar_t *begin, const wchar_t *end,
1351 Py_UCS4 *maxchar, Py_ssize_t *num_surrogates)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001352{
1353 const wchar_t *iter;
Victor Stinner8faf8212011-12-08 22:14:11 +01001354 Py_UCS4 ch;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001355
Victor Stinnerc53be962011-10-02 21:33:54 +02001356 assert(num_surrogates != NULL && maxchar != NULL);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001357 *num_surrogates = 0;
1358 *maxchar = 0;
1359
1360 for (iter = begin; iter < end; ) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001361#if SIZEOF_WCHAR_T == 2
Victor Stinnerca4f2072011-11-22 03:38:40 +01001362 if (Py_UNICODE_IS_HIGH_SURROGATE(iter[0])
1363 && (iter+1) < end
1364 && Py_UNICODE_IS_LOW_SURROGATE(iter[1]))
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001365 {
Victor Stinner8faf8212011-12-08 22:14:11 +01001366 ch = Py_UNICODE_JOIN_SURROGATES(iter[0], iter[1]);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001367 ++(*num_surrogates);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001368 iter += 2;
1369 }
1370 else
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001371#endif
Victor Stinner8faf8212011-12-08 22:14:11 +01001372 {
1373 ch = *iter;
1374 iter++;
1375 }
1376 if (ch > *maxchar) {
1377 *maxchar = ch;
1378 if (*maxchar > MAX_UNICODE) {
1379 PyErr_Format(PyExc_ValueError,
1380 "character U+%x is not in range [U+0000; U+10ffff]",
1381 ch);
1382 return -1;
1383 }
1384 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001385 }
1386 return 0;
1387}
1388
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01001389int
1390_PyUnicode_Ready(PyObject *unicode)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001391{
1392 wchar_t *end;
1393 Py_UCS4 maxchar = 0;
1394 Py_ssize_t num_surrogates;
1395#if SIZEOF_WCHAR_T == 2
1396 Py_ssize_t length_wo_surrogates;
1397#endif
1398
Georg Brandl7597add2011-10-05 16:36:47 +02001399 /* _PyUnicode_Ready() is only intended for old-style API usage where
Victor Stinnerd8f65102011-09-29 19:43:17 +02001400 strings were created using _PyObject_New() and where no canonical
1401 representation (the str field) has been set yet aka strings
1402 which are not yet ready. */
Victor Stinner910337b2011-10-03 03:20:16 +02001403 assert(_PyUnicode_CHECK(unicode));
1404 assert(_PyUnicode_KIND(unicode) == PyUnicode_WCHAR_KIND);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001405 assert(_PyUnicode_WSTR(unicode) != NULL);
Victor Stinnerc3c74152011-10-02 20:39:55 +02001406 assert(_PyUnicode_DATA_ANY(unicode) == NULL);
Victor Stinnere90fe6a2011-10-01 16:48:13 +02001407 assert(_PyUnicode_UTF8(unicode) == NULL);
Victor Stinnerd8f65102011-09-29 19:43:17 +02001408 /* Actually, it should neither be interned nor be anything else: */
1409 assert(_PyUnicode_STATE(unicode).interned == SSTATE_NOT_INTERNED);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001410
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001411 end = _PyUnicode_WSTR(unicode) + _PyUnicode_WSTR_LENGTH(unicode);
Victor Stinner17222162011-09-28 22:15:37 +02001412 if (find_maxchar_surrogates(_PyUnicode_WSTR(unicode), end,
Victor Stinnerd8f65102011-09-29 19:43:17 +02001413 &maxchar, &num_surrogates) == -1)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001414 return -1;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001415
1416 if (maxchar < 256) {
Victor Stinnerc3c74152011-10-02 20:39:55 +02001417 _PyUnicode_DATA_ANY(unicode) = PyObject_MALLOC(_PyUnicode_WSTR_LENGTH(unicode) + 1);
1418 if (!_PyUnicode_DATA_ANY(unicode)) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001419 PyErr_NoMemory();
1420 return -1;
1421 }
Victor Stinnerfb5f5f22011-09-28 21:39:49 +02001422 _PyUnicode_CONVERT_BYTES(wchar_t, unsigned char,
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001423 _PyUnicode_WSTR(unicode), end,
1424 PyUnicode_1BYTE_DATA(unicode));
1425 PyUnicode_1BYTE_DATA(unicode)[_PyUnicode_WSTR_LENGTH(unicode)] = '\0';
1426 _PyUnicode_LENGTH(unicode) = _PyUnicode_WSTR_LENGTH(unicode);
1427 _PyUnicode_STATE(unicode).kind = PyUnicode_1BYTE_KIND;
1428 if (maxchar < 128) {
Victor Stinnera3b334d2011-10-03 13:53:37 +02001429 _PyUnicode_STATE(unicode).ascii = 1;
Victor Stinnerc3c74152011-10-02 20:39:55 +02001430 _PyUnicode_UTF8(unicode) = _PyUnicode_DATA_ANY(unicode);
Victor Stinnere90fe6a2011-10-01 16:48:13 +02001431 _PyUnicode_UTF8_LENGTH(unicode) = _PyUnicode_WSTR_LENGTH(unicode);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001432 }
1433 else {
Victor Stinnera3b334d2011-10-03 13:53:37 +02001434 _PyUnicode_STATE(unicode).ascii = 0;
Victor Stinnere90fe6a2011-10-01 16:48:13 +02001435 _PyUnicode_UTF8(unicode) = NULL;
1436 _PyUnicode_UTF8_LENGTH(unicode) = 0;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001437 }
1438 PyObject_FREE(_PyUnicode_WSTR(unicode));
1439 _PyUnicode_WSTR(unicode) = NULL;
1440 _PyUnicode_WSTR_LENGTH(unicode) = 0;
1441 }
1442 /* In this case we might have to convert down from 4-byte native
1443 wchar_t to 2-byte unicode. */
1444 else if (maxchar < 65536) {
1445 assert(num_surrogates == 0 &&
1446 "FindMaxCharAndNumSurrogatePairs() messed up");
1447
Victor Stinner506f5922011-09-28 22:34:18 +02001448#if SIZEOF_WCHAR_T == 2
1449 /* We can share representations and are done. */
Victor Stinnerc3c74152011-10-02 20:39:55 +02001450 _PyUnicode_DATA_ANY(unicode) = _PyUnicode_WSTR(unicode);
Victor Stinner506f5922011-09-28 22:34:18 +02001451 PyUnicode_2BYTE_DATA(unicode)[_PyUnicode_WSTR_LENGTH(unicode)] = '\0';
1452 _PyUnicode_LENGTH(unicode) = _PyUnicode_WSTR_LENGTH(unicode);
1453 _PyUnicode_STATE(unicode).kind = PyUnicode_2BYTE_KIND;
Victor Stinnere90fe6a2011-10-01 16:48:13 +02001454 _PyUnicode_UTF8(unicode) = NULL;
1455 _PyUnicode_UTF8_LENGTH(unicode) = 0;
Victor Stinner506f5922011-09-28 22:34:18 +02001456#else
1457 /* sizeof(wchar_t) == 4 */
Victor Stinnerc3c74152011-10-02 20:39:55 +02001458 _PyUnicode_DATA_ANY(unicode) = PyObject_MALLOC(
Victor Stinner506f5922011-09-28 22:34:18 +02001459 2 * (_PyUnicode_WSTR_LENGTH(unicode) + 1));
Victor Stinnerc3c74152011-10-02 20:39:55 +02001460 if (!_PyUnicode_DATA_ANY(unicode)) {
Victor Stinner506f5922011-09-28 22:34:18 +02001461 PyErr_NoMemory();
1462 return -1;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001463 }
Victor Stinner506f5922011-09-28 22:34:18 +02001464 _PyUnicode_CONVERT_BYTES(wchar_t, Py_UCS2,
1465 _PyUnicode_WSTR(unicode), end,
1466 PyUnicode_2BYTE_DATA(unicode));
1467 PyUnicode_2BYTE_DATA(unicode)[_PyUnicode_WSTR_LENGTH(unicode)] = '\0';
1468 _PyUnicode_LENGTH(unicode) = _PyUnicode_WSTR_LENGTH(unicode);
1469 _PyUnicode_STATE(unicode).kind = PyUnicode_2BYTE_KIND;
Victor Stinnere90fe6a2011-10-01 16:48:13 +02001470 _PyUnicode_UTF8(unicode) = NULL;
1471 _PyUnicode_UTF8_LENGTH(unicode) = 0;
Victor Stinner506f5922011-09-28 22:34:18 +02001472 PyObject_FREE(_PyUnicode_WSTR(unicode));
1473 _PyUnicode_WSTR(unicode) = NULL;
1474 _PyUnicode_WSTR_LENGTH(unicode) = 0;
1475#endif
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001476 }
1477 /* maxchar exeeds 16 bit, wee need 4 bytes for unicode characters */
1478 else {
1479#if SIZEOF_WCHAR_T == 2
1480 /* in case the native representation is 2-bytes, we need to allocate a
1481 new normalized 4-byte version. */
1482 length_wo_surrogates = _PyUnicode_WSTR_LENGTH(unicode) - num_surrogates;
Victor Stinnerc3c74152011-10-02 20:39:55 +02001483 _PyUnicode_DATA_ANY(unicode) = PyObject_MALLOC(4 * (length_wo_surrogates + 1));
1484 if (!_PyUnicode_DATA_ANY(unicode)) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001485 PyErr_NoMemory();
1486 return -1;
1487 }
1488 _PyUnicode_LENGTH(unicode) = length_wo_surrogates;
1489 _PyUnicode_STATE(unicode).kind = PyUnicode_4BYTE_KIND;
Victor Stinnere90fe6a2011-10-01 16:48:13 +02001490 _PyUnicode_UTF8(unicode) = NULL;
1491 _PyUnicode_UTF8_LENGTH(unicode) = 0;
Victor Stinner126c5592011-10-03 04:17:10 +02001492 /* unicode_convert_wchar_to_ucs4() requires a ready string */
1493 _PyUnicode_STATE(unicode).ready = 1;
Victor Stinnerc53be962011-10-02 21:33:54 +02001494 unicode_convert_wchar_to_ucs4(_PyUnicode_WSTR(unicode), end, unicode);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001495 PyObject_FREE(_PyUnicode_WSTR(unicode));
1496 _PyUnicode_WSTR(unicode) = NULL;
1497 _PyUnicode_WSTR_LENGTH(unicode) = 0;
1498#else
1499 assert(num_surrogates == 0);
1500
Victor Stinnerc3c74152011-10-02 20:39:55 +02001501 _PyUnicode_DATA_ANY(unicode) = _PyUnicode_WSTR(unicode);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001502 _PyUnicode_LENGTH(unicode) = _PyUnicode_WSTR_LENGTH(unicode);
Victor Stinnere90fe6a2011-10-01 16:48:13 +02001503 _PyUnicode_UTF8(unicode) = NULL;
1504 _PyUnicode_UTF8_LENGTH(unicode) = 0;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001505 _PyUnicode_STATE(unicode).kind = PyUnicode_4BYTE_KIND;
1506#endif
1507 PyUnicode_4BYTE_DATA(unicode)[_PyUnicode_LENGTH(unicode)] = '\0';
1508 }
1509 _PyUnicode_STATE(unicode).ready = 1;
Victor Stinnerbb10a1f2011-10-05 01:34:17 +02001510 assert(_PyUnicode_CheckConsistency(unicode, 1));
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001511 return 0;
1512}
1513
Alexander Belopolsky40018472011-02-26 01:02:56 +00001514static void
Victor Stinner9db1a8b2011-10-23 20:04:37 +02001515unicode_dealloc(register PyObject *unicode)
Guido van Rossumd57fd912000-03-10 22:53:23 +00001516{
Walter Dörwald16807132007-05-25 13:52:07 +00001517 switch (PyUnicode_CHECK_INTERNED(unicode)) {
Benjamin Peterson29060642009-01-31 22:14:21 +00001518 case SSTATE_NOT_INTERNED:
1519 break;
Walter Dörwald16807132007-05-25 13:52:07 +00001520
Benjamin Peterson29060642009-01-31 22:14:21 +00001521 case SSTATE_INTERNED_MORTAL:
1522 /* revive dead object temporarily for DelItem */
1523 Py_REFCNT(unicode) = 3;
Victor Stinner7931d9a2011-11-04 00:22:48 +01001524 if (PyDict_DelItem(interned, unicode) != 0)
Benjamin Peterson29060642009-01-31 22:14:21 +00001525 Py_FatalError(
1526 "deletion of interned string failed");
1527 break;
Walter Dörwald16807132007-05-25 13:52:07 +00001528
Benjamin Peterson29060642009-01-31 22:14:21 +00001529 case SSTATE_INTERNED_IMMORTAL:
1530 Py_FatalError("Immortal interned string died.");
Walter Dörwald16807132007-05-25 13:52:07 +00001531
Benjamin Peterson29060642009-01-31 22:14:21 +00001532 default:
1533 Py_FatalError("Inconsistent interned string state.");
Walter Dörwald16807132007-05-25 13:52:07 +00001534 }
1535
Victor Stinner03490912011-10-03 23:45:12 +02001536 if (_PyUnicode_HAS_WSTR_MEMORY(unicode))
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001537 PyObject_DEL(_PyUnicode_WSTR(unicode));
Victor Stinner829c0ad2011-10-03 01:08:02 +02001538 if (_PyUnicode_HAS_UTF8_MEMORY(unicode))
Victor Stinnere90fe6a2011-10-01 16:48:13 +02001539 PyObject_DEL(_PyUnicode_UTF8(unicode));
Victor Stinnerb0a82a62011-12-12 13:08:33 +01001540 if (!PyUnicode_IS_COMPACT(unicode) && _PyUnicode_DATA_ANY(unicode))
1541 PyObject_DEL(_PyUnicode_DATA_ANY(unicode));
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001542
Victor Stinnerb0a82a62011-12-12 13:08:33 +01001543 Py_TYPE(unicode)->tp_free(unicode);
Guido van Rossumd57fd912000-03-10 22:53:23 +00001544}
1545
Victor Stinnerfb9ea8c2011-10-06 01:45:57 +02001546#ifdef Py_DEBUG
1547static int
1548unicode_is_singleton(PyObject *unicode)
1549{
1550 PyASCIIObject *ascii = (PyASCIIObject *)unicode;
1551 if (unicode == unicode_empty)
1552 return 1;
1553 if (ascii->state.kind != PyUnicode_WCHAR_KIND && ascii->length == 1)
1554 {
1555 Py_UCS4 ch = PyUnicode_READ_CHAR(unicode, 0);
1556 if (ch < 256 && unicode_latin1[ch] == unicode)
1557 return 1;
1558 }
1559 return 0;
1560}
1561#endif
1562
Alexander Belopolsky40018472011-02-26 01:02:56 +00001563static int
Victor Stinner488fa492011-12-12 00:01:39 +01001564unicode_modifiable(PyObject *unicode)
Marc-André Lemburg8155e0e2001-04-23 14:44:21 +00001565{
Victor Stinner488fa492011-12-12 00:01:39 +01001566 assert(_PyUnicode_CHECK(unicode));
Victor Stinnerfe226c02011-10-03 03:52:20 +02001567 if (Py_REFCNT(unicode) != 1)
1568 return 0;
Victor Stinner488fa492011-12-12 00:01:39 +01001569 if (_PyUnicode_HASH(unicode) != -1)
1570 return 0;
Victor Stinnerfe226c02011-10-03 03:52:20 +02001571 if (PyUnicode_CHECK_INTERNED(unicode))
1572 return 0;
Victor Stinner488fa492011-12-12 00:01:39 +01001573 if (!PyUnicode_CheckExact(unicode))
1574 return 0;
Victor Stinner77bb47b2011-10-03 20:06:05 +02001575#ifdef Py_DEBUG
Victor Stinnerfb9ea8c2011-10-06 01:45:57 +02001576 /* singleton refcount is greater than 1 */
1577 assert(!unicode_is_singleton(unicode));
Victor Stinner77bb47b2011-10-03 20:06:05 +02001578#endif
Victor Stinnerfe226c02011-10-03 03:52:20 +02001579 return 1;
1580}
Marc-André Lemburg8155e0e2001-04-23 14:44:21 +00001581
Victor Stinnerfe226c02011-10-03 03:52:20 +02001582static int
1583unicode_resize(PyObject **p_unicode, Py_ssize_t length)
1584{
1585 PyObject *unicode;
1586 Py_ssize_t old_length;
1587
1588 assert(p_unicode != NULL);
1589 unicode = *p_unicode;
1590
1591 assert(unicode != NULL);
1592 assert(PyUnicode_Check(unicode));
1593 assert(0 <= length);
1594
Victor Stinner910337b2011-10-03 03:20:16 +02001595 if (_PyUnicode_KIND(unicode) == PyUnicode_WCHAR_KIND)
Victor Stinnerfe226c02011-10-03 03:52:20 +02001596 old_length = PyUnicode_WSTR_LENGTH(unicode);
1597 else
1598 old_length = PyUnicode_GET_LENGTH(unicode);
1599 if (old_length == length)
1600 return 0;
1601
Martin v. Löwise9b11c12011-11-08 17:35:34 +01001602 if (length == 0) {
Serhiy Storchaka678db842013-01-26 12:16:36 +02001603 _Py_INCREF_UNICODE_EMPTY();
1604 if (!unicode_empty)
Benjamin Peterson29060642009-01-31 22:14:21 +00001605 return -1;
Martin v. Löwise9b11c12011-11-08 17:35:34 +01001606 Py_DECREF(*p_unicode);
1607 *p_unicode = unicode_empty;
Martin v. Löwise9b11c12011-11-08 17:35:34 +01001608 return 0;
1609 }
1610
Victor Stinner488fa492011-12-12 00:01:39 +01001611 if (!unicode_modifiable(unicode)) {
Victor Stinnerfe226c02011-10-03 03:52:20 +02001612 PyObject *copy = resize_copy(unicode, length);
1613 if (copy == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00001614 return -1;
Victor Stinnerfe226c02011-10-03 03:52:20 +02001615 Py_DECREF(*p_unicode);
1616 *p_unicode = copy;
Benjamin Peterson29060642009-01-31 22:14:21 +00001617 return 0;
Marc-André Lemburg8155e0e2001-04-23 14:44:21 +00001618 }
1619
Victor Stinnerfe226c02011-10-03 03:52:20 +02001620 if (PyUnicode_IS_COMPACT(unicode)) {
Victor Stinnerb0a82a62011-12-12 13:08:33 +01001621 PyObject *new_unicode = resize_compact(unicode, length);
1622 if (new_unicode == NULL)
Victor Stinnerfe226c02011-10-03 03:52:20 +02001623 return -1;
Victor Stinnerb0a82a62011-12-12 13:08:33 +01001624 *p_unicode = new_unicode;
Victor Stinnerfe226c02011-10-03 03:52:20 +02001625 return 0;
Benjamin Peterson4bfce8f2011-10-03 19:35:07 -04001626 }
Victor Stinner9db1a8b2011-10-23 20:04:37 +02001627 return resize_inplace(unicode, length);
Marc-André Lemburg8155e0e2001-04-23 14:44:21 +00001628}
1629
Alexander Belopolsky40018472011-02-26 01:02:56 +00001630int
Victor Stinnerfe226c02011-10-03 03:52:20 +02001631PyUnicode_Resize(PyObject **p_unicode, Py_ssize_t length)
Alexandre Vassalottiaa0e5312008-12-27 06:43:58 +00001632{
Victor Stinnerfe226c02011-10-03 03:52:20 +02001633 PyObject *unicode;
1634 if (p_unicode == NULL) {
1635 PyErr_BadInternalCall();
1636 return -1;
1637 }
1638 unicode = *p_unicode;
Martin v. Löwise9b11c12011-11-08 17:35:34 +01001639 if (unicode == NULL || !PyUnicode_Check(unicode) || length < 0)
Victor Stinnerfe226c02011-10-03 03:52:20 +02001640 {
1641 PyErr_BadInternalCall();
1642 return -1;
1643 }
1644 return unicode_resize(p_unicode, length);
Alexandre Vassalottiaa0e5312008-12-27 06:43:58 +00001645}
Marc-André Lemburg8155e0e2001-04-23 14:44:21 +00001646
Martin v. Löwise9b11c12011-11-08 17:35:34 +01001647static int
Victor Stinner1b487b42012-05-03 12:29:04 +02001648unicode_widen(PyObject **p_unicode, Py_ssize_t length,
1649 unsigned int maxchar)
Martin v. Löwise9b11c12011-11-08 17:35:34 +01001650{
1651 PyObject *result;
1652 assert(PyUnicode_IS_READY(*p_unicode));
Victor Stinner1b487b42012-05-03 12:29:04 +02001653 assert(length <= PyUnicode_GET_LENGTH(*p_unicode));
Martin v. Löwise9b11c12011-11-08 17:35:34 +01001654 if (maxchar <= PyUnicode_MAX_CHAR_VALUE(*p_unicode))
1655 return 0;
1656 result = PyUnicode_New(PyUnicode_GET_LENGTH(*p_unicode),
1657 maxchar);
1658 if (result == NULL)
1659 return -1;
Victor Stinnerd3f08822012-05-29 12:57:52 +02001660 _PyUnicode_FastCopyCharacters(result, 0, *p_unicode, 0, length);
Martin v. Löwise9b11c12011-11-08 17:35:34 +01001661 Py_DECREF(*p_unicode);
1662 *p_unicode = result;
1663 return 0;
1664}
1665
1666static int
1667unicode_putchar(PyObject **p_unicode, Py_ssize_t *pos,
1668 Py_UCS4 ch)
1669{
Victor Stinner15e9ed22012-02-22 13:36:20 +01001670 assert(ch <= MAX_UNICODE);
Victor Stinner1b487b42012-05-03 12:29:04 +02001671 if (unicode_widen(p_unicode, *pos, ch) < 0)
Martin v. Löwise9b11c12011-11-08 17:35:34 +01001672 return -1;
1673 PyUnicode_WRITE(PyUnicode_KIND(*p_unicode),
1674 PyUnicode_DATA(*p_unicode),
1675 (*pos)++, ch);
1676 return 0;
1677}
1678
Victor Stinnerc5166102012-02-22 13:55:02 +01001679/* Copy a ASCII or latin1 char* string into a Python Unicode string.
Victor Stinnerc5166102012-02-22 13:55:02 +01001680
Victor Stinnerb429d3b2012-02-22 21:22:20 +01001681 WARNING: The function doesn't copy the terminating null character and
1682 doesn't check the maximum character (may write a latin1 character in an
1683 ASCII string). */
Victor Stinner184252a2012-06-16 02:57:41 +02001684static void
1685unicode_write_cstr(PyObject *unicode, Py_ssize_t index,
1686 const char *str, Py_ssize_t len)
Victor Stinnerc5166102012-02-22 13:55:02 +01001687{
1688 enum PyUnicode_Kind kind = PyUnicode_KIND(unicode);
1689 void *data = PyUnicode_DATA(unicode);
Victor Stinner184252a2012-06-16 02:57:41 +02001690 const char *end = str + len;
Victor Stinnerc5166102012-02-22 13:55:02 +01001691
1692 switch (kind) {
1693 case PyUnicode_1BYTE_KIND: {
Victor Stinnerc5166102012-02-22 13:55:02 +01001694 assert(index + len <= PyUnicode_GET_LENGTH(unicode));
Antoine Pitrouba6bafc2012-02-22 16:41:50 +01001695 memcpy((char *) data + index, str, len);
Victor Stinner184252a2012-06-16 02:57:41 +02001696 break;
Victor Stinnerc5166102012-02-22 13:55:02 +01001697 }
1698 case PyUnicode_2BYTE_KIND: {
1699 Py_UCS2 *start = (Py_UCS2 *)data + index;
1700 Py_UCS2 *ucs2 = start;
1701 assert(index <= PyUnicode_GET_LENGTH(unicode));
1702
Victor Stinner184252a2012-06-16 02:57:41 +02001703 for (; str < end; ++ucs2, ++str)
Victor Stinnerc5166102012-02-22 13:55:02 +01001704 *ucs2 = (Py_UCS2)*str;
1705
1706 assert((ucs2 - start) <= PyUnicode_GET_LENGTH(unicode));
Victor Stinner184252a2012-06-16 02:57:41 +02001707 break;
Victor Stinnerc5166102012-02-22 13:55:02 +01001708 }
1709 default: {
1710 Py_UCS4 *start = (Py_UCS4 *)data + index;
1711 Py_UCS4 *ucs4 = start;
1712 assert(kind == PyUnicode_4BYTE_KIND);
1713 assert(index <= PyUnicode_GET_LENGTH(unicode));
1714
Victor Stinner184252a2012-06-16 02:57:41 +02001715 for (; str < end; ++ucs4, ++str)
Victor Stinnerc5166102012-02-22 13:55:02 +01001716 *ucs4 = (Py_UCS4)*str;
1717
1718 assert((ucs4 - start) <= PyUnicode_GET_LENGTH(unicode));
Victor Stinnerc5166102012-02-22 13:55:02 +01001719 }
1720 }
1721}
1722
1723
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001724static PyObject*
1725get_latin1_char(unsigned char ch)
1726{
Victor Stinnera464fc12011-10-02 20:39:30 +02001727 PyObject *unicode = unicode_latin1[ch];
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001728 if (!unicode) {
Victor Stinnera464fc12011-10-02 20:39:30 +02001729 unicode = PyUnicode_New(1, ch);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001730 if (!unicode)
1731 return NULL;
1732 PyUnicode_1BYTE_DATA(unicode)[0] = ch;
Victor Stinnerbb10a1f2011-10-05 01:34:17 +02001733 assert(_PyUnicode_CheckConsistency(unicode, 1));
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001734 unicode_latin1[ch] = unicode;
1735 }
1736 Py_INCREF(unicode);
Victor Stinnera464fc12011-10-02 20:39:30 +02001737 return unicode;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001738}
1739
Alexander Belopolsky40018472011-02-26 01:02:56 +00001740PyObject *
1741PyUnicode_FromUnicode(const Py_UNICODE *u, Py_ssize_t size)
Guido van Rossumd57fd912000-03-10 22:53:23 +00001742{
Victor Stinner9db1a8b2011-10-23 20:04:37 +02001743 PyObject *unicode;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001744 Py_UCS4 maxchar = 0;
1745 Py_ssize_t num_surrogates;
1746
1747 if (u == NULL)
1748 return (PyObject*)_PyUnicode_New(size);
Guido van Rossumd57fd912000-03-10 22:53:23 +00001749
Marc-André Lemburg8155e0e2001-04-23 14:44:21 +00001750 /* If the Unicode data is known at construction time, we can apply
1751 some optimizations which share commonly used objects. */
Marc-André Lemburg8155e0e2001-04-23 14:44:21 +00001752
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001753 /* Optimization for empty strings */
Serhiy Storchaka678db842013-01-26 12:16:36 +02001754 if (size == 0)
1755 _Py_RETURN_UNICODE_EMPTY();
Tim Petersced69f82003-09-16 20:30:58 +00001756
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001757 /* Single character Unicode objects in the Latin-1 range are
1758 shared when using this constructor */
Victor Stinnerd21b58c2013-02-26 00:15:54 +01001759 if (size == 1 && (Py_UCS4)*u < 256)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001760 return get_latin1_char((unsigned char)*u);
1761
1762 /* If not empty and not single character, copy the Unicode data
1763 into the new object */
Victor Stinnerd8f65102011-09-29 19:43:17 +02001764 if (find_maxchar_surrogates(u, u + size,
1765 &maxchar, &num_surrogates) == -1)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001766 return NULL;
1767
Victor Stinner8faf8212011-12-08 22:14:11 +01001768 unicode = PyUnicode_New(size - num_surrogates, maxchar);
Guido van Rossumd57fd912000-03-10 22:53:23 +00001769 if (!unicode)
1770 return NULL;
1771
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001772 switch (PyUnicode_KIND(unicode)) {
1773 case PyUnicode_1BYTE_KIND:
Victor Stinnerfb5f5f22011-09-28 21:39:49 +02001774 _PyUnicode_CONVERT_BYTES(Py_UNICODE, unsigned char,
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001775 u, u + size, PyUnicode_1BYTE_DATA(unicode));
1776 break;
1777 case PyUnicode_2BYTE_KIND:
1778#if Py_UNICODE_SIZE == 2
1779 Py_MEMCPY(PyUnicode_2BYTE_DATA(unicode), u, size * 2);
1780#else
Victor Stinnerfb5f5f22011-09-28 21:39:49 +02001781 _PyUnicode_CONVERT_BYTES(Py_UNICODE, Py_UCS2,
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001782 u, u + size, PyUnicode_2BYTE_DATA(unicode));
1783#endif
1784 break;
1785 case PyUnicode_4BYTE_KIND:
1786#if SIZEOF_WCHAR_T == 2
1787 /* This is the only case which has to process surrogates, thus
1788 a simple copy loop is not enough and we need a function. */
Victor Stinnerc53be962011-10-02 21:33:54 +02001789 unicode_convert_wchar_to_ucs4(u, u + size, unicode);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001790#else
1791 assert(num_surrogates == 0);
1792 Py_MEMCPY(PyUnicode_4BYTE_DATA(unicode), u, size * 4);
1793#endif
1794 break;
1795 default:
1796 assert(0 && "Impossible state");
1797 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00001798
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01001799 return unicode_result(unicode);
Guido van Rossumd57fd912000-03-10 22:53:23 +00001800}
1801
Alexander Belopolsky40018472011-02-26 01:02:56 +00001802PyObject *
1803PyUnicode_FromStringAndSize(const char *u, Py_ssize_t size)
Walter Dörwaldacaa5a12007-05-05 12:00:46 +00001804{
Benjamin Peterson14339b62009-01-31 16:36:08 +00001805 if (size < 0) {
1806 PyErr_SetString(PyExc_SystemError,
Benjamin Peterson29060642009-01-31 22:14:21 +00001807 "Negative size passed to PyUnicode_FromStringAndSize");
Benjamin Peterson14339b62009-01-31 16:36:08 +00001808 return NULL;
1809 }
Victor Stinnera1d12bb2011-12-11 21:53:09 +01001810 if (u != NULL)
1811 return PyUnicode_DecodeUTF8Stateful(u, size, NULL, NULL);
1812 else
1813 return (PyObject *)_PyUnicode_New(size);
Walter Dörwaldacaa5a12007-05-05 12:00:46 +00001814}
1815
Alexander Belopolsky40018472011-02-26 01:02:56 +00001816PyObject *
1817PyUnicode_FromString(const char *u)
Walter Dörwaldd2034312007-05-18 16:29:38 +00001818{
1819 size_t size = strlen(u);
1820 if (size > PY_SSIZE_T_MAX) {
1821 PyErr_SetString(PyExc_OverflowError, "input too long");
1822 return NULL;
1823 }
Victor Stinnera1d12bb2011-12-11 21:53:09 +01001824 return PyUnicode_DecodeUTF8Stateful(u, (Py_ssize_t)size, NULL, NULL);
Walter Dörwaldd2034312007-05-18 16:29:38 +00001825}
1826
Martin v. Löwisafe55bb2011-10-09 10:38:36 +02001827PyObject *
1828_PyUnicode_FromId(_Py_Identifier *id)
1829{
1830 if (!id->object) {
Victor Stinnerd1cd99b2012-02-07 23:05:55 +01001831 id->object = PyUnicode_DecodeUTF8Stateful(id->string,
1832 strlen(id->string),
1833 NULL, NULL);
Martin v. Löwisafe55bb2011-10-09 10:38:36 +02001834 if (!id->object)
1835 return NULL;
1836 PyUnicode_InternInPlace(&id->object);
1837 assert(!id->next);
1838 id->next = static_strings;
1839 static_strings = id;
1840 }
Martin v. Löwisafe55bb2011-10-09 10:38:36 +02001841 return id->object;
1842}
1843
1844void
1845_PyUnicode_ClearStaticStrings()
1846{
Benjamin Peterson0c270a82013-01-09 09:52:01 -06001847 _Py_Identifier *tmp, *s = static_strings;
1848 while (s) {
1849 Py_DECREF(s->object);
1850 s->object = NULL;
1851 tmp = s->next;
1852 s->next = NULL;
1853 s = tmp;
Martin v. Löwisafe55bb2011-10-09 10:38:36 +02001854 }
Benjamin Peterson0c270a82013-01-09 09:52:01 -06001855 static_strings = NULL;
Martin v. Löwisafe55bb2011-10-09 10:38:36 +02001856}
1857
Benjamin Peterson0df54292012-03-26 14:50:32 -04001858/* Internal function, doesn't check maximum character */
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01001859
Victor Stinnerd3f08822012-05-29 12:57:52 +02001860PyObject*
1861_PyUnicode_FromASCII(const char *buffer, Py_ssize_t size)
Victor Stinner702c7342011-10-05 13:50:52 +02001862{
Victor Stinnerd3f08822012-05-29 12:57:52 +02001863 const unsigned char *s = (const unsigned char *)buffer;
Victor Stinner785938e2011-12-11 20:09:03 +01001864 PyObject *unicode;
Victor Stinnere6b2d442011-12-11 21:54:30 +01001865 if (size == 1) {
Victor Stinner0617b6e2011-10-05 23:26:01 +02001866#ifdef Py_DEBUG
Victor Stinnerd21b58c2013-02-26 00:15:54 +01001867 assert((unsigned char)s[0] < 128);
Victor Stinner0617b6e2011-10-05 23:26:01 +02001868#endif
Antoine Pitrou7c46da72011-10-06 22:07:51 +02001869 return get_latin1_char(s[0]);
Victor Stinnere6b2d442011-12-11 21:54:30 +01001870 }
Victor Stinner785938e2011-12-11 20:09:03 +01001871 unicode = PyUnicode_New(size, 127);
1872 if (!unicode)
Victor Stinner702c7342011-10-05 13:50:52 +02001873 return NULL;
Victor Stinner785938e2011-12-11 20:09:03 +01001874 memcpy(PyUnicode_1BYTE_DATA(unicode), s, size);
1875 assert(_PyUnicode_CheckConsistency(unicode, 1));
1876 return unicode;
Victor Stinner702c7342011-10-05 13:50:52 +02001877}
1878
Victor Stinnerc80d6d22011-10-05 14:13:28 +02001879static Py_UCS4
1880kind_maxchar_limit(unsigned int kind)
1881{
Benjamin Petersonead6b532011-12-20 17:23:42 -06001882 switch (kind) {
Victor Stinnerc80d6d22011-10-05 14:13:28 +02001883 case PyUnicode_1BYTE_KIND:
1884 return 0x80;
1885 case PyUnicode_2BYTE_KIND:
1886 return 0x100;
1887 case PyUnicode_4BYTE_KIND:
1888 return 0x10000;
1889 default:
1890 assert(0 && "invalid kind");
Victor Stinner8faf8212011-12-08 22:14:11 +01001891 return MAX_UNICODE;
Victor Stinnerc80d6d22011-10-05 14:13:28 +02001892 }
1893}
1894
Victor Stinnere6abb482012-05-02 01:15:40 +02001895Py_LOCAL_INLINE(Py_UCS4)
1896align_maxchar(Py_UCS4 maxchar)
1897{
1898 if (maxchar <= 127)
1899 return 127;
1900 else if (maxchar <= 255)
1901 return 255;
1902 else if (maxchar <= 65535)
1903 return 65535;
1904 else
1905 return MAX_UNICODE;
1906}
1907
Victor Stinner702c7342011-10-05 13:50:52 +02001908static PyObject*
Victor Stinnerd21b58c2013-02-26 00:15:54 +01001909_PyUnicode_FromUCS1(const Py_UCS1* u, Py_ssize_t size)
Mark Dickinson081dfee2009-03-18 14:47:41 +00001910{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001911 PyObject *res;
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01001912 unsigned char max_char;
Victor Stinnerb9275c12011-10-05 14:01:42 +02001913
Serhiy Storchaka678db842013-01-26 12:16:36 +02001914 if (size == 0)
1915 _Py_RETURN_UNICODE_EMPTY();
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01001916 assert(size > 0);
Antoine Pitrou7c46da72011-10-06 22:07:51 +02001917 if (size == 1)
1918 return get_latin1_char(u[0]);
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01001919
Antoine Pitroudd4e2f02011-10-13 00:02:27 +02001920 max_char = ucs1lib_find_max_char(u, u + size);
Victor Stinnerb9275c12011-10-05 14:01:42 +02001921 res = PyUnicode_New(size, max_char);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001922 if (!res)
1923 return NULL;
1924 memcpy(PyUnicode_1BYTE_DATA(res), u, size);
Victor Stinnerbb10a1f2011-10-05 01:34:17 +02001925 assert(_PyUnicode_CheckConsistency(res, 1));
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001926 return res;
Mark Dickinson081dfee2009-03-18 14:47:41 +00001927}
1928
Victor Stinnere57b1c02011-09-28 22:20:48 +02001929static PyObject*
1930_PyUnicode_FromUCS2(const Py_UCS2 *u, Py_ssize_t size)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001931{
1932 PyObject *res;
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01001933 Py_UCS2 max_char;
Victor Stinnerb9275c12011-10-05 14:01:42 +02001934
Serhiy Storchaka678db842013-01-26 12:16:36 +02001935 if (size == 0)
1936 _Py_RETURN_UNICODE_EMPTY();
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01001937 assert(size > 0);
Victor Stinnerb6cd0142012-05-03 02:17:04 +02001938 if (size == 1) {
1939 Py_UCS4 ch = u[0];
1940 if (ch < 256)
1941 return get_latin1_char((unsigned char)ch);
1942
1943 res = PyUnicode_New(1, ch);
1944 if (res == NULL)
1945 return NULL;
1946 PyUnicode_WRITE(PyUnicode_KIND(res), PyUnicode_DATA(res), 0, ch);
1947 assert(_PyUnicode_CheckConsistency(res, 1));
1948 return res;
1949 }
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01001950
Antoine Pitroudd4e2f02011-10-13 00:02:27 +02001951 max_char = ucs2lib_find_max_char(u, u + size);
Victor Stinnerb9275c12011-10-05 14:01:42 +02001952 res = PyUnicode_New(size, max_char);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001953 if (!res)
1954 return NULL;
Victor Stinnerb9275c12011-10-05 14:01:42 +02001955 if (max_char >= 256)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001956 memcpy(PyUnicode_2BYTE_DATA(res), u, sizeof(Py_UCS2)*size);
Antoine Pitroudd4e2f02011-10-13 00:02:27 +02001957 else {
1958 _PyUnicode_CONVERT_BYTES(
1959 Py_UCS2, Py_UCS1, u, u + size, PyUnicode_1BYTE_DATA(res));
1960 }
Victor Stinnerbb10a1f2011-10-05 01:34:17 +02001961 assert(_PyUnicode_CheckConsistency(res, 1));
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001962 return res;
1963}
1964
Victor Stinnere57b1c02011-09-28 22:20:48 +02001965static PyObject*
1966_PyUnicode_FromUCS4(const Py_UCS4 *u, Py_ssize_t size)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001967{
1968 PyObject *res;
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01001969 Py_UCS4 max_char;
Victor Stinnerb9275c12011-10-05 14:01:42 +02001970
Serhiy Storchaka678db842013-01-26 12:16:36 +02001971 if (size == 0)
1972 _Py_RETURN_UNICODE_EMPTY();
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01001973 assert(size > 0);
Victor Stinnerb6cd0142012-05-03 02:17:04 +02001974 if (size == 1) {
1975 Py_UCS4 ch = u[0];
1976 if (ch < 256)
1977 return get_latin1_char((unsigned char)ch);
1978
1979 res = PyUnicode_New(1, ch);
1980 if (res == NULL)
1981 return NULL;
1982 PyUnicode_WRITE(PyUnicode_KIND(res), PyUnicode_DATA(res), 0, ch);
1983 assert(_PyUnicode_CheckConsistency(res, 1));
1984 return res;
1985 }
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01001986
Antoine Pitroudd4e2f02011-10-13 00:02:27 +02001987 max_char = ucs4lib_find_max_char(u, u + size);
Victor Stinnerb9275c12011-10-05 14:01:42 +02001988 res = PyUnicode_New(size, max_char);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001989 if (!res)
1990 return NULL;
Antoine Pitrou950468e2011-10-11 22:45:48 +02001991 if (max_char < 256)
1992 _PyUnicode_CONVERT_BYTES(Py_UCS4, Py_UCS1, u, u + size,
1993 PyUnicode_1BYTE_DATA(res));
1994 else if (max_char < 0x10000)
1995 _PyUnicode_CONVERT_BYTES(Py_UCS4, Py_UCS2, u, u + size,
1996 PyUnicode_2BYTE_DATA(res));
1997 else
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001998 memcpy(PyUnicode_4BYTE_DATA(res), u, sizeof(Py_UCS4)*size);
Victor Stinnerbb10a1f2011-10-05 01:34:17 +02001999 assert(_PyUnicode_CheckConsistency(res, 1));
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002000 return res;
2001}
2002
2003PyObject*
2004PyUnicode_FromKindAndData(int kind, const void *buffer, Py_ssize_t size)
2005{
Victor Stinnercfed46e2011-11-22 01:29:14 +01002006 if (size < 0) {
2007 PyErr_SetString(PyExc_ValueError, "size must be positive");
2008 return NULL;
2009 }
Benjamin Petersonead6b532011-12-20 17:23:42 -06002010 switch (kind) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002011 case PyUnicode_1BYTE_KIND:
Victor Stinnere57b1c02011-09-28 22:20:48 +02002012 return _PyUnicode_FromUCS1(buffer, size);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002013 case PyUnicode_2BYTE_KIND:
Victor Stinnere57b1c02011-09-28 22:20:48 +02002014 return _PyUnicode_FromUCS2(buffer, size);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002015 case PyUnicode_4BYTE_KIND:
Victor Stinnere57b1c02011-09-28 22:20:48 +02002016 return _PyUnicode_FromUCS4(buffer, size);
Victor Stinnerb9275c12011-10-05 14:01:42 +02002017 default:
Victor Stinnerb9275c12011-10-05 14:01:42 +02002018 PyErr_SetString(PyExc_SystemError, "invalid kind");
2019 return NULL;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002020 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002021}
2022
Victor Stinnerece58de2012-04-23 23:36:38 +02002023Py_UCS4
2024_PyUnicode_FindMaxChar(PyObject *unicode, Py_ssize_t start, Py_ssize_t end)
2025{
2026 enum PyUnicode_Kind kind;
2027 void *startptr, *endptr;
2028
2029 assert(PyUnicode_IS_READY(unicode));
2030 assert(0 <= start);
2031 assert(end <= PyUnicode_GET_LENGTH(unicode));
2032 assert(start <= end);
2033
2034 if (start == 0 && end == PyUnicode_GET_LENGTH(unicode))
2035 return PyUnicode_MAX_CHAR_VALUE(unicode);
2036
2037 if (start == end)
2038 return 127;
2039
Victor Stinner94d558b2012-04-27 22:26:58 +02002040 if (PyUnicode_IS_ASCII(unicode))
2041 return 127;
2042
Victor Stinnerece58de2012-04-23 23:36:38 +02002043 kind = PyUnicode_KIND(unicode);
Benjamin Petersonf3b7d862012-04-23 18:07:01 -04002044 startptr = PyUnicode_DATA(unicode);
Benjamin Petersonb9f4c9d2012-04-23 21:45:40 -04002045 endptr = (char *)startptr + end * kind;
2046 startptr = (char *)startptr + start * kind;
Benjamin Peterson2844a7a2012-04-23 18:00:25 -04002047 switch(kind) {
2048 case PyUnicode_1BYTE_KIND:
2049 return ucs1lib_find_max_char(startptr, endptr);
2050 case PyUnicode_2BYTE_KIND:
2051 return ucs2lib_find_max_char(startptr, endptr);
2052 case PyUnicode_4BYTE_KIND:
2053 return ucs4lib_find_max_char(startptr, endptr);
Victor Stinnerece58de2012-04-23 23:36:38 +02002054 default:
Benjamin Peterson2844a7a2012-04-23 18:00:25 -04002055 assert(0);
2056 return 0;
Victor Stinnerece58de2012-04-23 23:36:38 +02002057 }
2058}
2059
Victor Stinner25a4b292011-10-06 12:31:55 +02002060/* Ensure that a string uses the most efficient storage, if it is not the
2061 case: create a new string with of the right kind. Write NULL into *p_unicode
2062 on error. */
Antoine Pitrou53bb5482011-10-10 23:49:24 +02002063static void
Victor Stinner25a4b292011-10-06 12:31:55 +02002064unicode_adjust_maxchar(PyObject **p_unicode)
2065{
2066 PyObject *unicode, *copy;
2067 Py_UCS4 max_char;
Antoine Pitroudd4e2f02011-10-13 00:02:27 +02002068 Py_ssize_t len;
Victor Stinner25a4b292011-10-06 12:31:55 +02002069 unsigned int kind;
2070
2071 assert(p_unicode != NULL);
2072 unicode = *p_unicode;
2073 assert(PyUnicode_IS_READY(unicode));
2074 if (PyUnicode_IS_ASCII(unicode))
2075 return;
2076
2077 len = PyUnicode_GET_LENGTH(unicode);
2078 kind = PyUnicode_KIND(unicode);
2079 if (kind == PyUnicode_1BYTE_KIND) {
2080 const Py_UCS1 *u = PyUnicode_1BYTE_DATA(unicode);
Antoine Pitroudd4e2f02011-10-13 00:02:27 +02002081 max_char = ucs1lib_find_max_char(u, u + len);
2082 if (max_char >= 128)
2083 return;
Victor Stinner25a4b292011-10-06 12:31:55 +02002084 }
2085 else if (kind == PyUnicode_2BYTE_KIND) {
2086 const Py_UCS2 *u = PyUnicode_2BYTE_DATA(unicode);
Antoine Pitroudd4e2f02011-10-13 00:02:27 +02002087 max_char = ucs2lib_find_max_char(u, u + len);
2088 if (max_char >= 256)
2089 return;
Victor Stinner25a4b292011-10-06 12:31:55 +02002090 }
2091 else {
Antoine Pitroudd4e2f02011-10-13 00:02:27 +02002092 const Py_UCS4 *u = PyUnicode_4BYTE_DATA(unicode);
Victor Stinner25a4b292011-10-06 12:31:55 +02002093 assert(kind == PyUnicode_4BYTE_KIND);
Antoine Pitroudd4e2f02011-10-13 00:02:27 +02002094 max_char = ucs4lib_find_max_char(u, u + len);
2095 if (max_char >= 0x10000)
2096 return;
Victor Stinner25a4b292011-10-06 12:31:55 +02002097 }
Victor Stinner25a4b292011-10-06 12:31:55 +02002098 copy = PyUnicode_New(len, max_char);
Victor Stinnerca439ee2012-06-16 03:17:34 +02002099 if (copy != NULL)
2100 _PyUnicode_FastCopyCharacters(copy, 0, unicode, 0, len);
Victor Stinner25a4b292011-10-06 12:31:55 +02002101 Py_DECREF(unicode);
2102 *p_unicode = copy;
2103}
2104
Victor Stinner034f6cf2011-09-30 02:26:44 +02002105PyObject*
Victor Stinnerbf6e5602011-12-12 01:53:47 +01002106_PyUnicode_Copy(PyObject *unicode)
Victor Stinner034f6cf2011-09-30 02:26:44 +02002107{
Victor Stinner87af4f22011-11-21 23:03:47 +01002108 Py_ssize_t length;
Victor Stinnerc841e7d2011-10-01 01:34:32 +02002109 PyObject *copy;
Victor Stinnerc841e7d2011-10-01 01:34:32 +02002110
Victor Stinner034f6cf2011-09-30 02:26:44 +02002111 if (!PyUnicode_Check(unicode)) {
2112 PyErr_BadInternalCall();
2113 return NULL;
2114 }
Benjamin Petersonbac79492012-01-14 13:34:47 -05002115 if (PyUnicode_READY(unicode) == -1)
Victor Stinner034f6cf2011-09-30 02:26:44 +02002116 return NULL;
Victor Stinnerc841e7d2011-10-01 01:34:32 +02002117
Victor Stinner87af4f22011-11-21 23:03:47 +01002118 length = PyUnicode_GET_LENGTH(unicode);
2119 copy = PyUnicode_New(length, PyUnicode_MAX_CHAR_VALUE(unicode));
Victor Stinnerc841e7d2011-10-01 01:34:32 +02002120 if (!copy)
2121 return NULL;
2122 assert(PyUnicode_KIND(copy) == PyUnicode_KIND(unicode));
2123
Victor Stinner87af4f22011-11-21 23:03:47 +01002124 Py_MEMCPY(PyUnicode_DATA(copy), PyUnicode_DATA(unicode),
2125 length * PyUnicode_KIND(unicode));
Victor Stinnerbb10a1f2011-10-05 01:34:17 +02002126 assert(_PyUnicode_CheckConsistency(copy, 1));
Victor Stinnerc841e7d2011-10-01 01:34:32 +02002127 return copy;
Victor Stinner034f6cf2011-09-30 02:26:44 +02002128}
2129
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002130
Victor Stinnerbc603d12011-10-02 01:00:40 +02002131/* Widen Unicode objects to larger buffers. Don't write terminating null
2132 character. Return NULL on error. */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002133
2134void*
2135_PyUnicode_AsKind(PyObject *s, unsigned int kind)
2136{
Victor Stinnerbc603d12011-10-02 01:00:40 +02002137 Py_ssize_t len;
2138 void *result;
2139 unsigned int skind;
2140
Benjamin Petersonbac79492012-01-14 13:34:47 -05002141 if (PyUnicode_READY(s) == -1)
Victor Stinnerbc603d12011-10-02 01:00:40 +02002142 return NULL;
2143
2144 len = PyUnicode_GET_LENGTH(s);
2145 skind = PyUnicode_KIND(s);
2146 if (skind >= kind) {
Victor Stinner01698042011-10-04 00:04:26 +02002147 PyErr_SetString(PyExc_SystemError, "invalid widening attempt");
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002148 return NULL;
2149 }
Benjamin Petersonead6b532011-12-20 17:23:42 -06002150 switch (kind) {
Victor Stinnerbc603d12011-10-02 01:00:40 +02002151 case PyUnicode_2BYTE_KIND:
2152 result = PyMem_Malloc(len * sizeof(Py_UCS2));
2153 if (!result)
2154 return PyErr_NoMemory();
2155 assert(skind == PyUnicode_1BYTE_KIND);
2156 _PyUnicode_CONVERT_BYTES(
2157 Py_UCS1, Py_UCS2,
2158 PyUnicode_1BYTE_DATA(s),
2159 PyUnicode_1BYTE_DATA(s) + len,
2160 result);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002161 return result;
Victor Stinnerbc603d12011-10-02 01:00:40 +02002162 case PyUnicode_4BYTE_KIND:
2163 result = PyMem_Malloc(len * sizeof(Py_UCS4));
2164 if (!result)
2165 return PyErr_NoMemory();
2166 if (skind == PyUnicode_2BYTE_KIND) {
2167 _PyUnicode_CONVERT_BYTES(
2168 Py_UCS2, Py_UCS4,
2169 PyUnicode_2BYTE_DATA(s),
2170 PyUnicode_2BYTE_DATA(s) + len,
2171 result);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002172 }
Victor Stinnerbc603d12011-10-02 01:00:40 +02002173 else {
2174 assert(skind == PyUnicode_1BYTE_KIND);
2175 _PyUnicode_CONVERT_BYTES(
2176 Py_UCS1, Py_UCS4,
2177 PyUnicode_1BYTE_DATA(s),
2178 PyUnicode_1BYTE_DATA(s) + len,
2179 result);
2180 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002181 return result;
Victor Stinnerbc603d12011-10-02 01:00:40 +02002182 default:
2183 break;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002184 }
Victor Stinner01698042011-10-04 00:04:26 +02002185 PyErr_SetString(PyExc_SystemError, "invalid kind");
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002186 return NULL;
2187}
2188
2189static Py_UCS4*
2190as_ucs4(PyObject *string, Py_UCS4 *target, Py_ssize_t targetsize,
2191 int copy_null)
2192{
2193 int kind;
2194 void *data;
2195 Py_ssize_t len, targetlen;
2196 if (PyUnicode_READY(string) == -1)
2197 return NULL;
2198 kind = PyUnicode_KIND(string);
2199 data = PyUnicode_DATA(string);
2200 len = PyUnicode_GET_LENGTH(string);
2201 targetlen = len;
2202 if (copy_null)
2203 targetlen++;
2204 if (!target) {
2205 if (PY_SSIZE_T_MAX / sizeof(Py_UCS4) < targetlen) {
2206 PyErr_NoMemory();
2207 return NULL;
2208 }
2209 target = PyMem_Malloc(targetlen * sizeof(Py_UCS4));
2210 if (!target) {
2211 PyErr_NoMemory();
2212 return NULL;
2213 }
2214 }
2215 else {
2216 if (targetsize < targetlen) {
2217 PyErr_Format(PyExc_SystemError,
2218 "string is longer than the buffer");
2219 if (copy_null && 0 < targetsize)
2220 target[0] = 0;
2221 return NULL;
2222 }
2223 }
Antoine Pitrou950468e2011-10-11 22:45:48 +02002224 if (kind == PyUnicode_1BYTE_KIND) {
2225 Py_UCS1 *start = (Py_UCS1 *) data;
2226 _PyUnicode_CONVERT_BYTES(Py_UCS1, Py_UCS4, start, start + len, target);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002227 }
Antoine Pitrou950468e2011-10-11 22:45:48 +02002228 else if (kind == PyUnicode_2BYTE_KIND) {
2229 Py_UCS2 *start = (Py_UCS2 *) data;
2230 _PyUnicode_CONVERT_BYTES(Py_UCS2, Py_UCS4, start, start + len, target);
2231 }
2232 else {
2233 assert(kind == PyUnicode_4BYTE_KIND);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002234 Py_MEMCPY(target, data, len * sizeof(Py_UCS4));
Antoine Pitrou950468e2011-10-11 22:45:48 +02002235 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002236 if (copy_null)
2237 target[len] = 0;
2238 return target;
2239}
2240
2241Py_UCS4*
2242PyUnicode_AsUCS4(PyObject *string, Py_UCS4 *target, Py_ssize_t targetsize,
2243 int copy_null)
2244{
Antoine Pitroude20b0b2011-11-10 21:47:38 +01002245 if (target == NULL || targetsize < 0) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002246 PyErr_BadInternalCall();
2247 return NULL;
2248 }
2249 return as_ucs4(string, target, targetsize, copy_null);
2250}
2251
2252Py_UCS4*
2253PyUnicode_AsUCS4Copy(PyObject *string)
2254{
2255 return as_ucs4(string, NULL, 0, 1);
2256}
2257
2258#ifdef HAVE_WCHAR_H
Mark Dickinson081dfee2009-03-18 14:47:41 +00002259
Alexander Belopolsky40018472011-02-26 01:02:56 +00002260PyObject *
2261PyUnicode_FromWideChar(register const wchar_t *w, Py_ssize_t size)
Guido van Rossumd57fd912000-03-10 22:53:23 +00002262{
Guido van Rossumd57fd912000-03-10 22:53:23 +00002263 if (w == NULL) {
Guido van Rossumd57fd912000-03-10 22:53:23 +00002264 if (size == 0)
Serhiy Storchaka678db842013-01-26 12:16:36 +02002265 _Py_RETURN_UNICODE_EMPTY();
Benjamin Peterson29060642009-01-31 22:14:21 +00002266 PyErr_BadInternalCall();
2267 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00002268 }
2269
Martin v. Löwis790465f2008-04-05 20:41:37 +00002270 if (size == -1) {
2271 size = wcslen(w);
2272 }
2273
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002274 return PyUnicode_FromUnicode(w, size);
Guido van Rossumd57fd912000-03-10 22:53:23 +00002275}
2276
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002277#endif /* HAVE_WCHAR_H */
Mark Dickinson081dfee2009-03-18 14:47:41 +00002278
Walter Dörwald346737f2007-05-31 10:44:43 +00002279static void
Mark Dickinson6ce4a9a2009-11-16 17:00:11 +00002280makefmt(char *fmt, int longflag, int longlongflag, int size_tflag,
2281 int zeropad, int width, int precision, char c)
Walter Dörwald346737f2007-05-31 10:44:43 +00002282{
Benjamin Peterson14339b62009-01-31 16:36:08 +00002283 *fmt++ = '%';
2284 if (width) {
2285 if (zeropad)
2286 *fmt++ = '0';
2287 fmt += sprintf(fmt, "%d", width);
2288 }
2289 if (precision)
2290 fmt += sprintf(fmt, ".%d", precision);
2291 if (longflag)
2292 *fmt++ = 'l';
Mark Dickinson6ce4a9a2009-11-16 17:00:11 +00002293 else if (longlongflag) {
2294 /* longlongflag should only ever be nonzero on machines with
2295 HAVE_LONG_LONG defined */
2296#ifdef HAVE_LONG_LONG
2297 char *f = PY_FORMAT_LONG_LONG;
2298 while (*f)
2299 *fmt++ = *f++;
2300#else
2301 /* we shouldn't ever get here */
2302 assert(0);
2303 *fmt++ = 'l';
2304#endif
2305 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00002306 else if (size_tflag) {
2307 char *f = PY_FORMAT_SIZE_T;
2308 while (*f)
2309 *fmt++ = *f++;
2310 }
2311 *fmt++ = c;
2312 *fmt = '\0';
Walter Dörwald346737f2007-05-31 10:44:43 +00002313}
2314
Victor Stinner96865452011-03-01 23:44:09 +00002315/* helper for PyUnicode_FromFormatV() */
2316
2317static const char*
2318parse_format_flags(const char *f,
2319 int *p_width, int *p_precision,
2320 int *p_longflag, int *p_longlongflag, int *p_size_tflag)
2321{
2322 int width, precision, longflag, longlongflag, size_tflag;
2323
2324 /* parse the width.precision part, e.g. "%2.5s" => width=2, precision=5 */
2325 f++;
2326 width = 0;
2327 while (Py_ISDIGIT((unsigned)*f))
2328 width = (width*10) + *f++ - '0';
2329 precision = 0;
2330 if (*f == '.') {
2331 f++;
2332 while (Py_ISDIGIT((unsigned)*f))
2333 precision = (precision*10) + *f++ - '0';
2334 if (*f == '%') {
2335 /* "%.3%s" => f points to "3" */
2336 f--;
2337 }
2338 }
2339 if (*f == '\0') {
2340 /* bogus format "%.1" => go backward, f points to "1" */
2341 f--;
2342 }
2343 if (p_width != NULL)
2344 *p_width = width;
2345 if (p_precision != NULL)
2346 *p_precision = precision;
2347
2348 /* Handle %ld, %lu, %lld and %llu. */
2349 longflag = 0;
2350 longlongflag = 0;
Victor Stinnere7faec12011-03-02 00:01:53 +00002351 size_tflag = 0;
Victor Stinner96865452011-03-01 23:44:09 +00002352
2353 if (*f == 'l') {
Victor Stinner6d970f42011-03-02 00:04:25 +00002354 if (f[1] == 'd' || f[1] == 'u' || f[1] == 'i') {
Victor Stinner96865452011-03-01 23:44:09 +00002355 longflag = 1;
2356 ++f;
2357 }
2358#ifdef HAVE_LONG_LONG
2359 else if (f[1] == 'l' &&
Victor Stinner6d970f42011-03-02 00:04:25 +00002360 (f[2] == 'd' || f[2] == 'u' || f[2] == 'i')) {
Victor Stinner96865452011-03-01 23:44:09 +00002361 longlongflag = 1;
2362 f += 2;
2363 }
2364#endif
2365 }
2366 /* handle the size_t flag. */
Victor Stinner6d970f42011-03-02 00:04:25 +00002367 else if (*f == 'z' && (f[1] == 'd' || f[1] == 'u' || f[1] == 'i')) {
Victor Stinner96865452011-03-01 23:44:09 +00002368 size_tflag = 1;
2369 ++f;
2370 }
2371 if (p_longflag != NULL)
2372 *p_longflag = longflag;
2373 if (p_longlongflag != NULL)
2374 *p_longlongflag = longlongflag;
2375 if (p_size_tflag != NULL)
2376 *p_size_tflag = size_tflag;
2377 return f;
2378}
2379
Mark Dickinson6ce4a9a2009-11-16 17:00:11 +00002380/* maximum number of characters required for output of %ld. 21 characters
2381 allows for 64-bit integers (in decimal) and an optional sign. */
2382#define MAX_LONG_CHARS 21
2383/* maximum number of characters required for output of %lld.
2384 We need at most ceil(log10(256)*SIZEOF_LONG_LONG) digits,
2385 plus 1 for the sign. 53/22 is an upper bound for log10(256). */
2386#define MAX_LONG_LONG_CHARS (2 + (SIZEOF_LONG_LONG*53-1) / 22)
2387
Walter Dörwaldd2034312007-05-18 16:29:38 +00002388PyObject *
2389PyUnicode_FromFormatV(const char *format, va_list vargs)
2390{
Benjamin Peterson14339b62009-01-31 16:36:08 +00002391 va_list count;
2392 Py_ssize_t callcount = 0;
2393 PyObject **callresults = NULL;
2394 PyObject **callresult = NULL;
2395 Py_ssize_t n = 0;
2396 int width = 0;
2397 int precision = 0;
2398 int zeropad;
2399 const char* f;
Victor Stinnerfb9ea8c2011-10-06 01:45:57 +02002400 PyObject *string;
Benjamin Peterson14339b62009-01-31 16:36:08 +00002401 /* used by sprintf */
Mark Dickinson6ce4a9a2009-11-16 17:00:11 +00002402 char fmt[61]; /* should be enough for %0width.precisionlld */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002403 Py_UCS4 maxchar = 127; /* result is ASCII by default */
2404 Py_UCS4 argmaxchar;
2405 Py_ssize_t numbersize = 0;
2406 char *numberresults = NULL;
2407 char *numberresult = NULL;
2408 Py_ssize_t i;
2409 int kind;
2410 void *data;
Walter Dörwaldd2034312007-05-18 16:29:38 +00002411
Victor Stinner4a2b7a12010-08-13 14:03:48 +00002412 Py_VA_COPY(count, vargs);
Walter Dörwaldc1651a02009-05-03 22:55:55 +00002413 /* step 1: count the number of %S/%R/%A/%s format specifications
2414 * (we call PyObject_Str()/PyObject_Repr()/PyObject_ASCII()/
2415 * PyUnicode_DecodeUTF8() for these objects once during step 3 and put the
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002416 * result in an array)
Georg Brandl7597add2011-10-05 16:36:47 +02002417 * also estimate a upper bound for all the number formats in the string,
2418 * numbers will be formatted in step 3 and be kept in a '\0'-separated
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002419 * buffer before putting everything together. */
Benjamin Peterson14339b62009-01-31 16:36:08 +00002420 for (f = format; *f; f++) {
2421 if (*f == '%') {
Victor Stinner96865452011-03-01 23:44:09 +00002422 int longlongflag;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002423 /* skip width or width.precision (eg. "1.2" of "%1.2f") */
2424 f = parse_format_flags(f, &width, NULL, NULL, &longlongflag, NULL);
2425 if (*f == 's' || *f=='S' || *f=='R' || *f=='A' || *f=='V')
2426 ++callcount;
Walter Dörwaldd2034312007-05-18 16:29:38 +00002427
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002428 else if (*f == 'd' || *f=='u' || *f=='i' || *f=='x' || *f=='p') {
Mark Dickinson6ce4a9a2009-11-16 17:00:11 +00002429#ifdef HAVE_LONG_LONG
2430 if (longlongflag) {
2431 if (width < MAX_LONG_LONG_CHARS)
2432 width = MAX_LONG_LONG_CHARS;
2433 }
2434 else
2435#endif
2436 /* MAX_LONG_CHARS is enough to hold a 64-bit integer,
2437 including sign. Decimal takes the most space. This
2438 isn't enough for octal. If a width is specified we
2439 need more (which we allocate later). */
2440 if (width < MAX_LONG_CHARS)
2441 width = MAX_LONG_CHARS;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002442
2443 /* account for the size + '\0' to separate numbers
2444 inside of the numberresults buffer */
2445 numbersize += (width + 1);
2446 }
2447 }
2448 else if ((unsigned char)*f > 127) {
2449 PyErr_Format(PyExc_ValueError,
2450 "PyUnicode_FromFormatV() expects an ASCII-encoded format "
2451 "string, got a non-ASCII byte: 0x%02x",
2452 (unsigned char)*f);
2453 return NULL;
2454 }
2455 }
2456 /* step 2: allocate memory for the results of
2457 * PyObject_Str()/PyObject_Repr()/PyUnicode_DecodeUTF8() calls */
2458 if (callcount) {
2459 callresults = PyObject_Malloc(sizeof(PyObject *) * callcount);
2460 if (!callresults) {
2461 PyErr_NoMemory();
2462 return NULL;
2463 }
2464 callresult = callresults;
2465 }
2466 /* step 2.5: allocate memory for the results of formating numbers */
2467 if (numbersize) {
2468 numberresults = PyObject_Malloc(numbersize);
2469 if (!numberresults) {
2470 PyErr_NoMemory();
2471 goto fail;
2472 }
2473 numberresult = numberresults;
2474 }
2475
2476 /* step 3: format numbers and figure out how large a buffer we need */
2477 for (f = format; *f; f++) {
2478 if (*f == '%') {
2479 const char* p;
2480 int longflag;
2481 int longlongflag;
2482 int size_tflag;
2483 int numprinted;
2484
2485 p = f;
2486 zeropad = (f[1] == '0');
2487 f = parse_format_flags(f, &width, &precision,
2488 &longflag, &longlongflag, &size_tflag);
2489 switch (*f) {
2490 case 'c':
2491 {
Serhiy Storchaka8eeae212013-06-23 20:12:14 +03002492 int ordinal = va_arg(count, int);
2493 if (ordinal < 0 || ordinal > MAX_UNICODE) {
2494 PyErr_SetString(PyExc_OverflowError,
2495 "%c arg not in range(0x110000)");
2496 goto fail;
2497 }
2498 maxchar = Py_MAX(maxchar, (Py_UCS4)ordinal);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002499 n++;
2500 break;
2501 }
2502 case '%':
2503 n++;
2504 break;
2505 case 'i':
2506 case 'd':
2507 makefmt(fmt, longflag, longlongflag, size_tflag, zeropad,
2508 width, precision, *f);
2509 if (longflag)
2510 numprinted = sprintf(numberresult, fmt,
2511 va_arg(count, long));
2512#ifdef HAVE_LONG_LONG
2513 else if (longlongflag)
2514 numprinted = sprintf(numberresult, fmt,
2515 va_arg(count, PY_LONG_LONG));
2516#endif
2517 else if (size_tflag)
2518 numprinted = sprintf(numberresult, fmt,
2519 va_arg(count, Py_ssize_t));
2520 else
2521 numprinted = sprintf(numberresult, fmt,
2522 va_arg(count, int));
2523 n += numprinted;
2524 /* advance by +1 to skip over the '\0' */
2525 numberresult += (numprinted + 1);
2526 assert(*(numberresult - 1) == '\0');
2527 assert(*(numberresult - 2) != '\0');
2528 assert(numprinted >= 0);
2529 assert(numberresult <= numberresults + numbersize);
2530 break;
2531 case 'u':
2532 makefmt(fmt, longflag, longlongflag, size_tflag, zeropad,
2533 width, precision, 'u');
2534 if (longflag)
2535 numprinted = sprintf(numberresult, fmt,
2536 va_arg(count, unsigned long));
2537#ifdef HAVE_LONG_LONG
2538 else if (longlongflag)
2539 numprinted = sprintf(numberresult, fmt,
2540 va_arg(count, unsigned PY_LONG_LONG));
2541#endif
2542 else if (size_tflag)
2543 numprinted = sprintf(numberresult, fmt,
2544 va_arg(count, size_t));
2545 else
2546 numprinted = sprintf(numberresult, fmt,
2547 va_arg(count, unsigned int));
2548 n += numprinted;
2549 numberresult += (numprinted + 1);
2550 assert(*(numberresult - 1) == '\0');
2551 assert(*(numberresult - 2) != '\0');
2552 assert(numprinted >= 0);
2553 assert(numberresult <= numberresults + numbersize);
2554 break;
2555 case 'x':
2556 makefmt(fmt, 0, 0, 0, zeropad, width, precision, 'x');
2557 numprinted = sprintf(numberresult, fmt, va_arg(count, int));
2558 n += numprinted;
2559 numberresult += (numprinted + 1);
2560 assert(*(numberresult - 1) == '\0');
2561 assert(*(numberresult - 2) != '\0');
2562 assert(numprinted >= 0);
2563 assert(numberresult <= numberresults + numbersize);
2564 break;
2565 case 'p':
2566 numprinted = sprintf(numberresult, "%p", va_arg(count, void*));
2567 /* %p is ill-defined: ensure leading 0x. */
2568 if (numberresult[1] == 'X')
2569 numberresult[1] = 'x';
2570 else if (numberresult[1] != 'x') {
2571 memmove(numberresult + 2, numberresult,
2572 strlen(numberresult) + 1);
2573 numberresult[0] = '0';
2574 numberresult[1] = 'x';
2575 numprinted += 2;
2576 }
2577 n += numprinted;
2578 numberresult += (numprinted + 1);
2579 assert(*(numberresult - 1) == '\0');
2580 assert(*(numberresult - 2) != '\0');
2581 assert(numprinted >= 0);
2582 assert(numberresult <= numberresults + numbersize);
Benjamin Peterson14339b62009-01-31 16:36:08 +00002583 break;
2584 case 's':
2585 {
2586 /* UTF-8 */
Georg Brandl780b2a62009-05-05 09:19:59 +00002587 const char *s = va_arg(count, const char*);
Victor Stinnera1d12bb2011-12-11 21:53:09 +01002588 PyObject *str = PyUnicode_DecodeUTF8Stateful(s, strlen(s), "replace", NULL);
Walter Dörwaldc1651a02009-05-03 22:55:55 +00002589 if (!str)
2590 goto fail;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002591 /* since PyUnicode_DecodeUTF8 returns already flexible
2592 unicode objects, there is no need to call ready on them */
2593 argmaxchar = PyUnicode_MAX_CHAR_VALUE(str);
Benjamin Peterson7e303732013-06-10 09:19:46 -07002594 maxchar = Py_MAX(maxchar, argmaxchar);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002595 n += PyUnicode_GET_LENGTH(str);
Walter Dörwaldc1651a02009-05-03 22:55:55 +00002596 /* Remember the str and switch to the next slot */
2597 *callresult++ = str;
Benjamin Peterson14339b62009-01-31 16:36:08 +00002598 break;
2599 }
2600 case 'U':
2601 {
2602 PyObject *obj = va_arg(count, PyObject *);
Victor Stinner910337b2011-10-03 03:20:16 +02002603 assert(obj && _PyUnicode_CHECK(obj));
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002604 if (PyUnicode_READY(obj) == -1)
2605 goto fail;
2606 argmaxchar = PyUnicode_MAX_CHAR_VALUE(obj);
Benjamin Peterson7e303732013-06-10 09:19:46 -07002607 maxchar = Py_MAX(maxchar, argmaxchar);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002608 n += PyUnicode_GET_LENGTH(obj);
Benjamin Peterson14339b62009-01-31 16:36:08 +00002609 break;
2610 }
2611 case 'V':
2612 {
2613 PyObject *obj = va_arg(count, PyObject *);
2614 const char *str = va_arg(count, const char *);
Victor Stinner2512a8b2011-03-01 22:46:52 +00002615 PyObject *str_obj;
Benjamin Peterson14339b62009-01-31 16:36:08 +00002616 assert(obj || str);
Victor Stinner910337b2011-10-03 03:20:16 +02002617 assert(!obj || _PyUnicode_CHECK(obj));
Victor Stinner2512a8b2011-03-01 22:46:52 +00002618 if (obj) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002619 if (PyUnicode_READY(obj) == -1)
2620 goto fail;
2621 argmaxchar = PyUnicode_MAX_CHAR_VALUE(obj);
Benjamin Peterson7e303732013-06-10 09:19:46 -07002622 maxchar = Py_MAX(maxchar, argmaxchar);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002623 n += PyUnicode_GET_LENGTH(obj);
Victor Stinner2512a8b2011-03-01 22:46:52 +00002624 *callresult++ = NULL;
2625 }
2626 else {
Victor Stinnera1d12bb2011-12-11 21:53:09 +01002627 str_obj = PyUnicode_DecodeUTF8Stateful(str, strlen(str), "replace", NULL);
Victor Stinner2512a8b2011-03-01 22:46:52 +00002628 if (!str_obj)
2629 goto fail;
Benjamin Petersonbac79492012-01-14 13:34:47 -05002630 if (PyUnicode_READY(str_obj) == -1) {
Victor Stinnere1335c72011-10-04 20:53:03 +02002631 Py_DECREF(str_obj);
2632 goto fail;
2633 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002634 argmaxchar = PyUnicode_MAX_CHAR_VALUE(str_obj);
Benjamin Peterson7e303732013-06-10 09:19:46 -07002635 maxchar = Py_MAX(maxchar, argmaxchar);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002636 n += PyUnicode_GET_LENGTH(str_obj);
Victor Stinner2512a8b2011-03-01 22:46:52 +00002637 *callresult++ = str_obj;
2638 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00002639 break;
2640 }
2641 case 'S':
2642 {
2643 PyObject *obj = va_arg(count, PyObject *);
2644 PyObject *str;
2645 assert(obj);
2646 str = PyObject_Str(obj);
Benjamin Petersonc8d8b882012-01-14 13:37:31 -05002647 if (!str)
Benjamin Peterson14339b62009-01-31 16:36:08 +00002648 goto fail;
Benjamin Petersonc8d8b882012-01-14 13:37:31 -05002649 if (PyUnicode_READY(str) == -1) {
2650 Py_DECREF(str);
2651 goto fail;
2652 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002653 argmaxchar = PyUnicode_MAX_CHAR_VALUE(str);
Benjamin Peterson7e303732013-06-10 09:19:46 -07002654 maxchar = Py_MAX(maxchar, argmaxchar);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002655 n += PyUnicode_GET_LENGTH(str);
Benjamin Peterson14339b62009-01-31 16:36:08 +00002656 /* Remember the str and switch to the next slot */
2657 *callresult++ = str;
2658 break;
2659 }
2660 case 'R':
2661 {
2662 PyObject *obj = va_arg(count, PyObject *);
2663 PyObject *repr;
2664 assert(obj);
2665 repr = PyObject_Repr(obj);
Benjamin Petersonc8d8b882012-01-14 13:37:31 -05002666 if (!repr)
Benjamin Peterson14339b62009-01-31 16:36:08 +00002667 goto fail;
Benjamin Petersonc8d8b882012-01-14 13:37:31 -05002668 if (PyUnicode_READY(repr) == -1) {
2669 Py_DECREF(repr);
2670 goto fail;
2671 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002672 argmaxchar = PyUnicode_MAX_CHAR_VALUE(repr);
Benjamin Peterson7e303732013-06-10 09:19:46 -07002673 maxchar = Py_MAX(maxchar, argmaxchar);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002674 n += PyUnicode_GET_LENGTH(repr);
Benjamin Peterson14339b62009-01-31 16:36:08 +00002675 /* Remember the repr and switch to the next slot */
2676 *callresult++ = repr;
2677 break;
2678 }
2679 case 'A':
2680 {
2681 PyObject *obj = va_arg(count, PyObject *);
2682 PyObject *ascii;
2683 assert(obj);
2684 ascii = PyObject_ASCII(obj);
Benjamin Petersonc8d8b882012-01-14 13:37:31 -05002685 if (!ascii)
Benjamin Peterson14339b62009-01-31 16:36:08 +00002686 goto fail;
Benjamin Petersonc8d8b882012-01-14 13:37:31 -05002687 if (PyUnicode_READY(ascii) == -1) {
2688 Py_DECREF(ascii);
2689 goto fail;
2690 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002691 argmaxchar = PyUnicode_MAX_CHAR_VALUE(ascii);
Benjamin Peterson7e303732013-06-10 09:19:46 -07002692 maxchar = Py_MAX(maxchar, argmaxchar);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002693 n += PyUnicode_GET_LENGTH(ascii);
Benjamin Peterson14339b62009-01-31 16:36:08 +00002694 /* Remember the repr and switch to the next slot */
2695 *callresult++ = ascii;
2696 break;
2697 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00002698 default:
2699 /* if we stumble upon an unknown
2700 formatting code, copy the rest of
2701 the format string to the output
2702 string. (we cannot just skip the
2703 code, since there's no way to know
2704 what's in the argument list) */
2705 n += strlen(p);
2706 goto expand;
2707 }
2708 } else
2709 n++;
2710 }
Benjamin Peterson29060642009-01-31 22:14:21 +00002711 expand:
Benjamin Peterson14339b62009-01-31 16:36:08 +00002712 /* step 4: fill the buffer */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002713 /* Since we've analyzed how much space we need,
Benjamin Peterson14339b62009-01-31 16:36:08 +00002714 we don't have to resize the string.
2715 There can be no errors beyond this point. */
Victor Stinnerfb9ea8c2011-10-06 01:45:57 +02002716 string = PyUnicode_New(n, maxchar);
Benjamin Peterson14339b62009-01-31 16:36:08 +00002717 if (!string)
2718 goto fail;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002719 kind = PyUnicode_KIND(string);
2720 data = PyUnicode_DATA(string);
Benjamin Peterson14339b62009-01-31 16:36:08 +00002721 callresult = callresults;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002722 numberresult = numberresults;
Walter Dörwaldd2034312007-05-18 16:29:38 +00002723
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002724 for (i = 0, f = format; *f; f++) {
Benjamin Peterson14339b62009-01-31 16:36:08 +00002725 if (*f == '%') {
Victor Stinner96865452011-03-01 23:44:09 +00002726 const char* p;
Victor Stinner96865452011-03-01 23:44:09 +00002727
2728 p = f;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002729 f = parse_format_flags(f, NULL, NULL, NULL, NULL, NULL);
2730 /* checking for == because the last argument could be a empty
2731 string, which causes i to point to end, the assert at the end of
2732 the loop */
2733 assert(i <= PyUnicode_GET_LENGTH(string));
Walter Dörwaldd2034312007-05-18 16:29:38 +00002734
Benjamin Peterson14339b62009-01-31 16:36:08 +00002735 switch (*f) {
2736 case 'c':
Victor Stinner5ed8b2c2011-02-21 21:13:44 +00002737 {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002738 const int ordinal = va_arg(vargs, int);
2739 PyUnicode_WRITE(kind, data, i++, ordinal);
Benjamin Peterson14339b62009-01-31 16:36:08 +00002740 break;
Victor Stinner5ed8b2c2011-02-21 21:13:44 +00002741 }
Victor Stinner6d970f42011-03-02 00:04:25 +00002742 case 'i':
Benjamin Peterson14339b62009-01-31 16:36:08 +00002743 case 'd':
Benjamin Peterson14339b62009-01-31 16:36:08 +00002744 case 'u':
Benjamin Peterson14339b62009-01-31 16:36:08 +00002745 case 'x':
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002746 case 'p':
Victor Stinnerc5166102012-02-22 13:55:02 +01002747 {
Victor Stinner184252a2012-06-16 02:57:41 +02002748 Py_ssize_t len;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002749 /* unused, since we already have the result */
2750 if (*f == 'p')
2751 (void) va_arg(vargs, void *);
2752 else
2753 (void) va_arg(vargs, int);
2754 /* extract the result from numberresults and append. */
Victor Stinner184252a2012-06-16 02:57:41 +02002755 len = strlen(numberresult);
2756 unicode_write_cstr(string, i, numberresult, len);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002757 /* skip over the separating '\0' */
Victor Stinner184252a2012-06-16 02:57:41 +02002758 i += len;
2759 numberresult += len;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002760 assert(*numberresult == '\0');
2761 numberresult++;
2762 assert(numberresult <= numberresults + numbersize);
Benjamin Peterson14339b62009-01-31 16:36:08 +00002763 break;
Victor Stinnerc5166102012-02-22 13:55:02 +01002764 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00002765 case 's':
2766 {
Walter Dörwaldc1651a02009-05-03 22:55:55 +00002767 /* unused, since we already have the result */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002768 Py_ssize_t size;
Walter Dörwaldc1651a02009-05-03 22:55:55 +00002769 (void) va_arg(vargs, char *);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002770 size = PyUnicode_GET_LENGTH(*callresult);
2771 assert(PyUnicode_KIND(*callresult) <= PyUnicode_KIND(string));
Victor Stinnerd3f08822012-05-29 12:57:52 +02002772 _PyUnicode_FastCopyCharacters(string, i, *callresult, 0, size);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002773 i += size;
Walter Dörwaldc1651a02009-05-03 22:55:55 +00002774 /* We're done with the unicode()/repr() => forget it */
2775 Py_DECREF(*callresult);
2776 /* switch to next unicode()/repr() result */
2777 ++callresult;
Benjamin Peterson14339b62009-01-31 16:36:08 +00002778 break;
2779 }
2780 case 'U':
2781 {
2782 PyObject *obj = va_arg(vargs, PyObject *);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002783 Py_ssize_t size;
2784 assert(PyUnicode_KIND(obj) <= PyUnicode_KIND(string));
2785 size = PyUnicode_GET_LENGTH(obj);
Victor Stinnerd3f08822012-05-29 12:57:52 +02002786 _PyUnicode_FastCopyCharacters(string, i, obj, 0, size);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002787 i += size;
Benjamin Peterson14339b62009-01-31 16:36:08 +00002788 break;
2789 }
2790 case 'V':
2791 {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002792 Py_ssize_t size;
Benjamin Peterson14339b62009-01-31 16:36:08 +00002793 PyObject *obj = va_arg(vargs, PyObject *);
Victor Stinner2512a8b2011-03-01 22:46:52 +00002794 va_arg(vargs, const char *);
Benjamin Peterson14339b62009-01-31 16:36:08 +00002795 if (obj) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002796 size = PyUnicode_GET_LENGTH(obj);
2797 assert(PyUnicode_KIND(obj) <= PyUnicode_KIND(string));
Victor Stinnerd3f08822012-05-29 12:57:52 +02002798 _PyUnicode_FastCopyCharacters(string, i, obj, 0, size);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002799 i += size;
Benjamin Peterson14339b62009-01-31 16:36:08 +00002800 } else {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002801 size = PyUnicode_GET_LENGTH(*callresult);
2802 assert(PyUnicode_KIND(*callresult) <=
2803 PyUnicode_KIND(string));
Victor Stinnerd3f08822012-05-29 12:57:52 +02002804 _PyUnicode_FastCopyCharacters(string, i, *callresult, 0, size);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002805 i += size;
Victor Stinner2512a8b2011-03-01 22:46:52 +00002806 Py_DECREF(*callresult);
Benjamin Peterson14339b62009-01-31 16:36:08 +00002807 }
Victor Stinner2512a8b2011-03-01 22:46:52 +00002808 ++callresult;
Benjamin Peterson14339b62009-01-31 16:36:08 +00002809 break;
2810 }
2811 case 'S':
2812 case 'R':
Victor Stinner9a909002010-10-18 20:59:24 +00002813 case 'A':
Benjamin Peterson14339b62009-01-31 16:36:08 +00002814 {
Victor Stinnerfb9ea8c2011-10-06 01:45:57 +02002815 Py_ssize_t size = PyUnicode_GET_LENGTH(*callresult);
Benjamin Peterson14339b62009-01-31 16:36:08 +00002816 /* unused, since we already have the result */
2817 (void) va_arg(vargs, PyObject *);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002818 assert(PyUnicode_KIND(*callresult) <= PyUnicode_KIND(string));
Victor Stinnerd3f08822012-05-29 12:57:52 +02002819 _PyUnicode_FastCopyCharacters(string, i, *callresult, 0, size);
Victor Stinnerfb9ea8c2011-10-06 01:45:57 +02002820 i += size;
Benjamin Peterson14339b62009-01-31 16:36:08 +00002821 /* We're done with the unicode()/repr() => forget it */
2822 Py_DECREF(*callresult);
2823 /* switch to next unicode()/repr() result */
2824 ++callresult;
2825 break;
2826 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00002827 case '%':
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002828 PyUnicode_WRITE(kind, data, i++, '%');
Benjamin Peterson14339b62009-01-31 16:36:08 +00002829 break;
2830 default:
Victor Stinner184252a2012-06-16 02:57:41 +02002831 {
2832 Py_ssize_t len = strlen(p);
2833 unicode_write_cstr(string, i, p, len);
2834 i += len;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002835 assert(i == PyUnicode_GET_LENGTH(string));
Benjamin Peterson14339b62009-01-31 16:36:08 +00002836 goto end;
2837 }
Victor Stinner184252a2012-06-16 02:57:41 +02002838 }
Victor Stinner1205f272010-09-11 00:54:47 +00002839 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002840 else {
2841 assert(i < PyUnicode_GET_LENGTH(string));
2842 PyUnicode_WRITE(kind, data, i++, *f);
2843 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00002844 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002845 assert(i == PyUnicode_GET_LENGTH(string));
Walter Dörwaldd2034312007-05-18 16:29:38 +00002846
Benjamin Peterson29060642009-01-31 22:14:21 +00002847 end:
Benjamin Peterson14339b62009-01-31 16:36:08 +00002848 if (callresults)
2849 PyObject_Free(callresults);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002850 if (numberresults)
2851 PyObject_Free(numberresults);
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01002852 return unicode_result(string);
Benjamin Peterson29060642009-01-31 22:14:21 +00002853 fail:
Benjamin Peterson14339b62009-01-31 16:36:08 +00002854 if (callresults) {
2855 PyObject **callresult2 = callresults;
2856 while (callresult2 < callresult) {
Victor Stinner2512a8b2011-03-01 22:46:52 +00002857 Py_XDECREF(*callresult2);
Benjamin Peterson14339b62009-01-31 16:36:08 +00002858 ++callresult2;
2859 }
2860 PyObject_Free(callresults);
2861 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002862 if (numberresults)
2863 PyObject_Free(numberresults);
Benjamin Peterson14339b62009-01-31 16:36:08 +00002864 return NULL;
Walter Dörwaldd2034312007-05-18 16:29:38 +00002865}
2866
Walter Dörwaldd2034312007-05-18 16:29:38 +00002867PyObject *
2868PyUnicode_FromFormat(const char *format, ...)
2869{
Benjamin Peterson14339b62009-01-31 16:36:08 +00002870 PyObject* ret;
2871 va_list vargs;
Walter Dörwaldd2034312007-05-18 16:29:38 +00002872
2873#ifdef HAVE_STDARG_PROTOTYPES
Benjamin Peterson14339b62009-01-31 16:36:08 +00002874 va_start(vargs, format);
Walter Dörwaldd2034312007-05-18 16:29:38 +00002875#else
Benjamin Peterson14339b62009-01-31 16:36:08 +00002876 va_start(vargs);
Walter Dörwaldd2034312007-05-18 16:29:38 +00002877#endif
Benjamin Peterson14339b62009-01-31 16:36:08 +00002878 ret = PyUnicode_FromFormatV(format, vargs);
2879 va_end(vargs);
2880 return ret;
Walter Dörwaldd2034312007-05-18 16:29:38 +00002881}
2882
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002883#ifdef HAVE_WCHAR_H
2884
Victor Stinner5593d8a2010-10-02 11:11:27 +00002885/* Helper function for PyUnicode_AsWideChar() and PyUnicode_AsWideCharString():
2886 convert a Unicode object to a wide character string.
2887
Victor Stinnerd88d9832011-09-06 02:00:05 +02002888 - If w is NULL: return the number of wide characters (including the null
Victor Stinner5593d8a2010-10-02 11:11:27 +00002889 character) required to convert the unicode object. Ignore size argument.
2890
Victor Stinnerd88d9832011-09-06 02:00:05 +02002891 - Otherwise: return the number of wide characters (excluding the null
Victor Stinner5593d8a2010-10-02 11:11:27 +00002892 character) written into w. Write at most size wide characters (including
Victor Stinnerd88d9832011-09-06 02:00:05 +02002893 the null character). */
Victor Stinner5593d8a2010-10-02 11:11:27 +00002894static Py_ssize_t
Victor Stinner9db1a8b2011-10-23 20:04:37 +02002895unicode_aswidechar(PyObject *unicode,
Victor Stinner137c34c2010-09-29 10:25:54 +00002896 wchar_t *w,
2897 Py_ssize_t size)
2898{
Victor Stinner5593d8a2010-10-02 11:11:27 +00002899 Py_ssize_t res;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002900 const wchar_t *wstr;
2901
Victor Stinner9db1a8b2011-10-23 20:04:37 +02002902 wstr = PyUnicode_AsUnicodeAndSize(unicode, &res);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002903 if (wstr == NULL)
2904 return -1;
2905
Victor Stinner5593d8a2010-10-02 11:11:27 +00002906 if (w != NULL) {
Victor Stinner5593d8a2010-10-02 11:11:27 +00002907 if (size > res)
2908 size = res + 1;
2909 else
2910 res = size;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002911 Py_MEMCPY(w, wstr, size * sizeof(wchar_t));
Victor Stinner5593d8a2010-10-02 11:11:27 +00002912 return res;
2913 }
2914 else
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002915 return res + 1;
Victor Stinner137c34c2010-09-29 10:25:54 +00002916}
2917
2918Py_ssize_t
Martin v. Löwis4d0d4712010-12-03 20:14:31 +00002919PyUnicode_AsWideChar(PyObject *unicode,
Victor Stinner137c34c2010-09-29 10:25:54 +00002920 wchar_t *w,
2921 Py_ssize_t size)
Guido van Rossumd57fd912000-03-10 22:53:23 +00002922{
2923 if (unicode == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +00002924 PyErr_BadInternalCall();
2925 return -1;
Guido van Rossumd57fd912000-03-10 22:53:23 +00002926 }
Victor Stinner9db1a8b2011-10-23 20:04:37 +02002927 return unicode_aswidechar(unicode, w, size);
Guido van Rossumd57fd912000-03-10 22:53:23 +00002928}
2929
Victor Stinner137c34c2010-09-29 10:25:54 +00002930wchar_t*
Victor Stinnerbeb4135b2010-10-07 01:02:42 +00002931PyUnicode_AsWideCharString(PyObject *unicode,
Victor Stinner137c34c2010-09-29 10:25:54 +00002932 Py_ssize_t *size)
2933{
2934 wchar_t* buffer;
2935 Py_ssize_t buflen;
2936
2937 if (unicode == NULL) {
2938 PyErr_BadInternalCall();
2939 return NULL;
2940 }
2941
Victor Stinner9db1a8b2011-10-23 20:04:37 +02002942 buflen = unicode_aswidechar(unicode, NULL, 0);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002943 if (buflen == -1)
2944 return NULL;
Victor Stinner5593d8a2010-10-02 11:11:27 +00002945 if (PY_SSIZE_T_MAX / sizeof(wchar_t) < buflen) {
Victor Stinner137c34c2010-09-29 10:25:54 +00002946 PyErr_NoMemory();
2947 return NULL;
2948 }
2949
Victor Stinner137c34c2010-09-29 10:25:54 +00002950 buffer = PyMem_MALLOC(buflen * sizeof(wchar_t));
2951 if (buffer == NULL) {
2952 PyErr_NoMemory();
2953 return NULL;
2954 }
Victor Stinner9db1a8b2011-10-23 20:04:37 +02002955 buflen = unicode_aswidechar(unicode, buffer, buflen);
Stefan Krah8528c312012-08-19 21:52:43 +02002956 if (buflen == -1) {
2957 PyMem_FREE(buffer);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002958 return NULL;
Stefan Krah8528c312012-08-19 21:52:43 +02002959 }
Victor Stinner5593d8a2010-10-02 11:11:27 +00002960 if (size != NULL)
2961 *size = buflen;
Victor Stinner137c34c2010-09-29 10:25:54 +00002962 return buffer;
2963}
2964
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002965#endif /* HAVE_WCHAR_H */
Guido van Rossumd57fd912000-03-10 22:53:23 +00002966
Alexander Belopolsky40018472011-02-26 01:02:56 +00002967PyObject *
2968PyUnicode_FromOrdinal(int ordinal)
Marc-André Lemburgcc8764c2002-08-11 12:23:04 +00002969{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002970 PyObject *v;
Victor Stinner8faf8212011-12-08 22:14:11 +01002971 if (ordinal < 0 || ordinal > MAX_UNICODE) {
Benjamin Peterson29060642009-01-31 22:14:21 +00002972 PyErr_SetString(PyExc_ValueError,
2973 "chr() arg not in range(0x110000)");
2974 return NULL;
Marc-André Lemburgcc8764c2002-08-11 12:23:04 +00002975 }
Guido van Rossum8ac004e2007-07-15 13:00:05 +00002976
Victor Stinnerd21b58c2013-02-26 00:15:54 +01002977 if ((Py_UCS4)ordinal < 256)
2978 return get_latin1_char((unsigned char)ordinal);
Marc-André Lemburgcc8764c2002-08-11 12:23:04 +00002979
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002980 v = PyUnicode_New(1, ordinal);
2981 if (v == NULL)
2982 return NULL;
2983 PyUnicode_WRITE(PyUnicode_KIND(v), PyUnicode_DATA(v), 0, ordinal);
Victor Stinnerbb10a1f2011-10-05 01:34:17 +02002984 assert(_PyUnicode_CheckConsistency(v, 1));
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002985 return v;
Marc-André Lemburgcc8764c2002-08-11 12:23:04 +00002986}
2987
Alexander Belopolsky40018472011-02-26 01:02:56 +00002988PyObject *
2989PyUnicode_FromObject(register PyObject *obj)
Guido van Rossumd57fd912000-03-10 22:53:23 +00002990{
Guido van Rossumb8c65bc2001-10-19 02:01:31 +00002991 /* XXX Perhaps we should make this API an alias of
Benjamin Peterson29060642009-01-31 22:14:21 +00002992 PyObject_Str() instead ?! */
Guido van Rossumb8c65bc2001-10-19 02:01:31 +00002993 if (PyUnicode_CheckExact(obj)) {
Benjamin Petersonbac79492012-01-14 13:34:47 -05002994 if (PyUnicode_READY(obj) == -1)
Victor Stinnerd3a83d52011-10-01 03:09:33 +02002995 return NULL;
Benjamin Peterson29060642009-01-31 22:14:21 +00002996 Py_INCREF(obj);
2997 return obj;
Guido van Rossumb8c65bc2001-10-19 02:01:31 +00002998 }
2999 if (PyUnicode_Check(obj)) {
Benjamin Peterson29060642009-01-31 22:14:21 +00003000 /* For a Unicode subtype that's not a Unicode object,
3001 return a true Unicode object with the same data. */
Victor Stinnerbf6e5602011-12-12 01:53:47 +01003002 return _PyUnicode_Copy(obj);
Guido van Rossumb8c65bc2001-10-19 02:01:31 +00003003 }
Guido van Rossum98297ee2007-11-06 21:34:58 +00003004 PyErr_Format(PyExc_TypeError,
3005 "Can't convert '%.100s' object to str implicitly",
Christian Heimes90aa7642007-12-19 02:45:37 +00003006 Py_TYPE(obj)->tp_name);
Guido van Rossum98297ee2007-11-06 21:34:58 +00003007 return NULL;
Marc-André Lemburg5a5c81a2000-07-07 13:46:42 +00003008}
3009
Alexander Belopolsky40018472011-02-26 01:02:56 +00003010PyObject *
3011PyUnicode_FromEncodedObject(register PyObject *obj,
Ezio Melotti2aa2b3b2011-09-29 00:58:57 +03003012 const char *encoding,
3013 const char *errors)
Marc-André Lemburg5a5c81a2000-07-07 13:46:42 +00003014{
Antoine Pitroub0fa8312010-09-01 15:10:12 +00003015 Py_buffer buffer;
Marc-André Lemburg5a5c81a2000-07-07 13:46:42 +00003016 PyObject *v;
Tim Petersced69f82003-09-16 20:30:58 +00003017
Guido van Rossumd57fd912000-03-10 22:53:23 +00003018 if (obj == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +00003019 PyErr_BadInternalCall();
3020 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00003021 }
Marc-André Lemburg5a5c81a2000-07-07 13:46:42 +00003022
Antoine Pitroub0fa8312010-09-01 15:10:12 +00003023 /* Decoding bytes objects is the most common case and should be fast */
3024 if (PyBytes_Check(obj)) {
Serhiy Storchaka05997252013-01-26 12:14:02 +02003025 if (PyBytes_GET_SIZE(obj) == 0)
3026 _Py_RETURN_UNICODE_EMPTY();
3027 v = PyUnicode_Decode(
3028 PyBytes_AS_STRING(obj), PyBytes_GET_SIZE(obj),
3029 encoding, errors);
Antoine Pitroub0fa8312010-09-01 15:10:12 +00003030 return v;
3031 }
3032
Guido van Rossumb8c65bc2001-10-19 02:01:31 +00003033 if (PyUnicode_Check(obj)) {
Benjamin Peterson29060642009-01-31 22:14:21 +00003034 PyErr_SetString(PyExc_TypeError,
3035 "decoding str is not supported");
3036 return NULL;
Benjamin Peterson14339b62009-01-31 16:36:08 +00003037 }
Guido van Rossumb8c65bc2001-10-19 02:01:31 +00003038
Antoine Pitroub0fa8312010-09-01 15:10:12 +00003039 /* Retrieve a bytes buffer view through the PEP 3118 buffer interface */
3040 if (PyObject_GetBuffer(obj, &buffer, PyBUF_SIMPLE) < 0) {
3041 PyErr_Format(PyExc_TypeError,
3042 "coercing to str: need bytes, bytearray "
3043 "or buffer-like object, %.80s found",
3044 Py_TYPE(obj)->tp_name);
3045 return NULL;
Marc-André Lemburg6871f6a2001-09-20 12:53:16 +00003046 }
Tim Petersced69f82003-09-16 20:30:58 +00003047
Antoine Pitroub0fa8312010-09-01 15:10:12 +00003048 if (buffer.len == 0) {
Serhiy Storchaka05997252013-01-26 12:14:02 +02003049 PyBuffer_Release(&buffer);
3050 _Py_RETURN_UNICODE_EMPTY();
Guido van Rossumd57fd912000-03-10 22:53:23 +00003051 }
Marc-André Lemburgad7c98e2001-01-17 17:09:53 +00003052
Serhiy Storchaka05997252013-01-26 12:14:02 +02003053 v = PyUnicode_Decode((char*) buffer.buf, buffer.len, encoding, errors);
Antoine Pitroub0fa8312010-09-01 15:10:12 +00003054 PyBuffer_Release(&buffer);
Marc-André Lemburg5a5c81a2000-07-07 13:46:42 +00003055 return v;
Guido van Rossumd57fd912000-03-10 22:53:23 +00003056}
3057
Victor Stinner600d3be2010-06-10 12:00:55 +00003058/* Convert encoding to lower case and replace '_' with '-' in order to
Victor Stinner37296e82010-06-10 13:36:23 +00003059 catch e.g. UTF_8. Return 0 on error (encoding is longer than lower_len-1),
3060 1 on success. */
Victor Stinner20b654a2013-01-03 01:08:58 +01003061int
3062_Py_normalize_encoding(const char *encoding,
Victor Stinner37296e82010-06-10 13:36:23 +00003063 char *lower,
3064 size_t lower_len)
Guido van Rossumd57fd912000-03-10 22:53:23 +00003065{
Guido van Rossumdaa251c2007-10-25 23:47:33 +00003066 const char *e;
Victor Stinner600d3be2010-06-10 12:00:55 +00003067 char *l;
3068 char *l_end;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00003069
Benjamin Peterson7a6debe2011-10-15 09:25:28 -04003070 if (encoding == NULL) {
3071 strcpy(lower, "utf-8");
3072 return 1;
3073 }
Guido van Rossumdaa251c2007-10-25 23:47:33 +00003074 e = encoding;
3075 l = lower;
Victor Stinner600d3be2010-06-10 12:00:55 +00003076 l_end = &lower[lower_len - 1];
Victor Stinner37296e82010-06-10 13:36:23 +00003077 while (*e) {
3078 if (l == l_end)
3079 return 0;
David Malcolm96960882010-11-05 17:23:41 +00003080 if (Py_ISUPPER(*e)) {
3081 *l++ = Py_TOLOWER(*e++);
Guido van Rossumdaa251c2007-10-25 23:47:33 +00003082 }
3083 else if (*e == '_') {
3084 *l++ = '-';
3085 e++;
3086 }
3087 else {
3088 *l++ = *e++;
3089 }
3090 }
3091 *l = '\0';
Victor Stinner37296e82010-06-10 13:36:23 +00003092 return 1;
Victor Stinner600d3be2010-06-10 12:00:55 +00003093}
3094
Alexander Belopolsky40018472011-02-26 01:02:56 +00003095PyObject *
3096PyUnicode_Decode(const char *s,
Ezio Melotti2aa2b3b2011-09-29 00:58:57 +03003097 Py_ssize_t size,
3098 const char *encoding,
3099 const char *errors)
Victor Stinner600d3be2010-06-10 12:00:55 +00003100{
3101 PyObject *buffer = NULL, *unicode;
3102 Py_buffer info;
3103 char lower[11]; /* Enough for any encoding shortcut */
3104
Fred Drakee4315f52000-05-09 19:53:39 +00003105 /* Shortcuts for common default encodings */
Victor Stinner20b654a2013-01-03 01:08:58 +01003106 if (_Py_normalize_encoding(encoding, lower, sizeof(lower))) {
Alexander Belopolsky1d521462011-02-25 19:19:57 +00003107 if ((strcmp(lower, "utf-8") == 0) ||
3108 (strcmp(lower, "utf8") == 0))
Victor Stinnera1d12bb2011-12-11 21:53:09 +01003109 return PyUnicode_DecodeUTF8Stateful(s, size, errors, NULL);
Victor Stinner37296e82010-06-10 13:36:23 +00003110 else if ((strcmp(lower, "latin-1") == 0) ||
Alexander Belopolsky1d521462011-02-25 19:19:57 +00003111 (strcmp(lower, "latin1") == 0) ||
Victor Stinner37296e82010-06-10 13:36:23 +00003112 (strcmp(lower, "iso-8859-1") == 0))
3113 return PyUnicode_DecodeLatin1(s, size, errors);
Victor Stinner99b95382011-07-04 14:23:54 +02003114#ifdef HAVE_MBCS
Victor Stinner37296e82010-06-10 13:36:23 +00003115 else if (strcmp(lower, "mbcs") == 0)
3116 return PyUnicode_DecodeMBCS(s, size, errors);
Mark Hammond0ccda1e2003-07-01 00:13:27 +00003117#endif
Victor Stinner37296e82010-06-10 13:36:23 +00003118 else if (strcmp(lower, "ascii") == 0)
3119 return PyUnicode_DecodeASCII(s, size, errors);
3120 else if (strcmp(lower, "utf-16") == 0)
3121 return PyUnicode_DecodeUTF16(s, size, errors, 0);
3122 else if (strcmp(lower, "utf-32") == 0)
3123 return PyUnicode_DecodeUTF32(s, size, errors, 0);
3124 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00003125
3126 /* Decode via the codec registry */
Guido van Rossumbe801ac2007-10-08 03:32:34 +00003127 buffer = NULL;
Antoine Pitrouc3b39242009-01-03 16:59:18 +00003128 if (PyBuffer_FillInfo(&info, NULL, (void *)s, size, 1, PyBUF_FULL_RO) < 0)
Guido van Rossumbe801ac2007-10-08 03:32:34 +00003129 goto onError;
Antoine Pitrouee58fa42008-08-19 18:22:14 +00003130 buffer = PyMemoryView_FromBuffer(&info);
Guido van Rossumd57fd912000-03-10 22:53:23 +00003131 if (buffer == NULL)
3132 goto onError;
3133 unicode = PyCodec_Decode(buffer, encoding, errors);
3134 if (unicode == NULL)
3135 goto onError;
3136 if (!PyUnicode_Check(unicode)) {
3137 PyErr_Format(PyExc_TypeError,
Benjamin Peterson142957c2008-07-04 19:55:29 +00003138 "decoder did not return a str object (type=%.400s)",
Christian Heimes90aa7642007-12-19 02:45:37 +00003139 Py_TYPE(unicode)->tp_name);
Guido van Rossumd57fd912000-03-10 22:53:23 +00003140 Py_DECREF(unicode);
3141 goto onError;
3142 }
3143 Py_DECREF(buffer);
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01003144 return unicode_result(unicode);
Tim Petersced69f82003-09-16 20:30:58 +00003145
Benjamin Peterson29060642009-01-31 22:14:21 +00003146 onError:
Guido van Rossumd57fd912000-03-10 22:53:23 +00003147 Py_XDECREF(buffer);
3148 return NULL;
3149}
3150
Alexander Belopolsky40018472011-02-26 01:02:56 +00003151PyObject *
3152PyUnicode_AsDecodedObject(PyObject *unicode,
Ezio Melotti2aa2b3b2011-09-29 00:58:57 +03003153 const char *encoding,
3154 const char *errors)
Marc-André Lemburgd2d45982004-07-08 17:57:32 +00003155{
3156 PyObject *v;
3157
3158 if (!PyUnicode_Check(unicode)) {
3159 PyErr_BadArgument();
3160 goto onError;
3161 }
3162
3163 if (encoding == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00003164 encoding = PyUnicode_GetDefaultEncoding();
Marc-André Lemburgd2d45982004-07-08 17:57:32 +00003165
3166 /* Decode via the codec registry */
3167 v = PyCodec_Decode(unicode, encoding, errors);
3168 if (v == NULL)
3169 goto onError;
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01003170 return unicode_result(v);
Marc-André Lemburgd2d45982004-07-08 17:57:32 +00003171
Benjamin Peterson29060642009-01-31 22:14:21 +00003172 onError:
Marc-André Lemburgd2d45982004-07-08 17:57:32 +00003173 return NULL;
3174}
3175
Alexander Belopolsky40018472011-02-26 01:02:56 +00003176PyObject *
3177PyUnicode_AsDecodedUnicode(PyObject *unicode,
Ezio Melotti2aa2b3b2011-09-29 00:58:57 +03003178 const char *encoding,
3179 const char *errors)
Marc-André Lemburgb2750b52008-06-06 12:18:17 +00003180{
3181 PyObject *v;
3182
3183 if (!PyUnicode_Check(unicode)) {
3184 PyErr_BadArgument();
3185 goto onError;
3186 }
3187
3188 if (encoding == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00003189 encoding = PyUnicode_GetDefaultEncoding();
Marc-André Lemburgb2750b52008-06-06 12:18:17 +00003190
3191 /* Decode via the codec registry */
3192 v = PyCodec_Decode(unicode, encoding, errors);
3193 if (v == NULL)
3194 goto onError;
3195 if (!PyUnicode_Check(v)) {
3196 PyErr_Format(PyExc_TypeError,
Benjamin Peterson142957c2008-07-04 19:55:29 +00003197 "decoder did not return a str object (type=%.400s)",
Marc-André Lemburgb2750b52008-06-06 12:18:17 +00003198 Py_TYPE(v)->tp_name);
3199 Py_DECREF(v);
3200 goto onError;
3201 }
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01003202 return unicode_result(v);
Marc-André Lemburgb2750b52008-06-06 12:18:17 +00003203
Benjamin Peterson29060642009-01-31 22:14:21 +00003204 onError:
Marc-André Lemburgb2750b52008-06-06 12:18:17 +00003205 return NULL;
3206}
3207
Alexander Belopolsky40018472011-02-26 01:02:56 +00003208PyObject *
3209PyUnicode_Encode(const Py_UNICODE *s,
Ezio Melotti2aa2b3b2011-09-29 00:58:57 +03003210 Py_ssize_t size,
3211 const char *encoding,
3212 const char *errors)
Guido van Rossumd57fd912000-03-10 22:53:23 +00003213{
3214 PyObject *v, *unicode;
Tim Petersced69f82003-09-16 20:30:58 +00003215
Guido van Rossumd57fd912000-03-10 22:53:23 +00003216 unicode = PyUnicode_FromUnicode(s, size);
3217 if (unicode == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00003218 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00003219 v = PyUnicode_AsEncodedString(unicode, encoding, errors);
3220 Py_DECREF(unicode);
3221 return v;
3222}
3223
Alexander Belopolsky40018472011-02-26 01:02:56 +00003224PyObject *
3225PyUnicode_AsEncodedObject(PyObject *unicode,
Ezio Melotti2aa2b3b2011-09-29 00:58:57 +03003226 const char *encoding,
3227 const char *errors)
Marc-André Lemburgd2d45982004-07-08 17:57:32 +00003228{
3229 PyObject *v;
3230
3231 if (!PyUnicode_Check(unicode)) {
3232 PyErr_BadArgument();
3233 goto onError;
3234 }
3235
3236 if (encoding == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00003237 encoding = PyUnicode_GetDefaultEncoding();
Marc-André Lemburgd2d45982004-07-08 17:57:32 +00003238
3239 /* Encode via the codec registry */
3240 v = PyCodec_Encode(unicode, encoding, errors);
3241 if (v == NULL)
3242 goto onError;
3243 return v;
3244
Benjamin Peterson29060642009-01-31 22:14:21 +00003245 onError:
Marc-André Lemburgd2d45982004-07-08 17:57:32 +00003246 return NULL;
3247}
3248
Victor Stinnerf2ea71f2011-12-17 04:13:41 +01003249static size_t
3250wcstombs_errorpos(const wchar_t *wstr)
3251{
3252 size_t len;
3253#if SIZEOF_WCHAR_T == 2
3254 wchar_t buf[3];
3255#else
3256 wchar_t buf[2];
3257#endif
3258 char outbuf[MB_LEN_MAX];
3259 const wchar_t *start, *previous;
Victor Stinnerf2ea71f2011-12-17 04:13:41 +01003260
Victor Stinnerf2ea71f2011-12-17 04:13:41 +01003261#if SIZEOF_WCHAR_T == 2
3262 buf[2] = 0;
3263#else
3264 buf[1] = 0;
3265#endif
3266 start = wstr;
3267 while (*wstr != L'\0')
3268 {
3269 previous = wstr;
3270#if SIZEOF_WCHAR_T == 2
3271 if (Py_UNICODE_IS_HIGH_SURROGATE(wstr[0])
3272 && Py_UNICODE_IS_LOW_SURROGATE(wstr[1]))
3273 {
3274 buf[0] = wstr[0];
3275 buf[1] = wstr[1];
3276 wstr += 2;
3277 }
3278 else {
3279 buf[0] = *wstr;
3280 buf[1] = 0;
3281 wstr++;
3282 }
3283#else
3284 buf[0] = *wstr;
3285 wstr++;
3286#endif
3287 len = wcstombs(outbuf, buf, sizeof(outbuf));
Victor Stinner2f197072011-12-17 07:08:30 +01003288 if (len == (size_t)-1)
Victor Stinnerf2ea71f2011-12-17 04:13:41 +01003289 return previous - start;
Victor Stinnerf2ea71f2011-12-17 04:13:41 +01003290 }
3291
3292 /* failed to find the unencodable character */
Victor Stinnerf2ea71f2011-12-17 04:13:41 +01003293 return 0;
3294}
3295
Victor Stinner1b579672011-12-17 05:47:23 +01003296static int
3297locale_error_handler(const char *errors, int *surrogateescape)
3298{
3299 if (errors == NULL) {
3300 *surrogateescape = 0;
3301 return 0;
3302 }
3303
3304 if (strcmp(errors, "strict") == 0) {
3305 *surrogateescape = 0;
3306 return 0;
3307 }
3308 if (strcmp(errors, "surrogateescape") == 0) {
3309 *surrogateescape = 1;
3310 return 0;
3311 }
3312 PyErr_Format(PyExc_ValueError,
3313 "only 'strict' and 'surrogateescape' error handlers "
3314 "are supported, not '%s'",
3315 errors);
3316 return -1;
3317}
3318
Victor Stinnerf2ea71f2011-12-17 04:13:41 +01003319PyObject *
Victor Stinner1b579672011-12-17 05:47:23 +01003320PyUnicode_EncodeLocale(PyObject *unicode, const char *errors)
Victor Stinnerf2ea71f2011-12-17 04:13:41 +01003321{
3322 Py_ssize_t wlen, wlen2;
3323 wchar_t *wstr;
3324 PyObject *bytes = NULL;
3325 char *errmsg;
Victor Stinner1f33f2b2011-12-17 04:45:09 +01003326 PyObject *reason;
Victor Stinnerf2ea71f2011-12-17 04:13:41 +01003327 PyObject *exc;
3328 size_t error_pos;
Victor Stinner1b579672011-12-17 05:47:23 +01003329 int surrogateescape;
3330
3331 if (locale_error_handler(errors, &surrogateescape) < 0)
3332 return NULL;
Victor Stinnerf2ea71f2011-12-17 04:13:41 +01003333
3334 wstr = PyUnicode_AsWideCharString(unicode, &wlen);
3335 if (wstr == NULL)
3336 return NULL;
3337
3338 wlen2 = wcslen(wstr);
3339 if (wlen2 != wlen) {
3340 PyMem_Free(wstr);
3341 PyErr_SetString(PyExc_TypeError, "embedded null character");
3342 return NULL;
3343 }
3344
3345 if (surrogateescape) {
3346 /* locale encoding with surrogateescape */
3347 char *str;
3348
3349 str = _Py_wchar2char(wstr, &error_pos);
3350 if (str == NULL) {
3351 if (error_pos == (size_t)-1) {
3352 PyErr_NoMemory();
3353 PyMem_Free(wstr);
3354 return NULL;
3355 }
3356 else {
3357 goto encode_error;
3358 }
3359 }
3360 PyMem_Free(wstr);
3361
3362 bytes = PyBytes_FromString(str);
3363 PyMem_Free(str);
3364 }
3365 else {
3366 size_t len, len2;
3367
3368 len = wcstombs(NULL, wstr, 0);
3369 if (len == (size_t)-1) {
Victor Stinner2f197072011-12-17 07:08:30 +01003370 error_pos = (size_t)-1;
Victor Stinnerf2ea71f2011-12-17 04:13:41 +01003371 goto encode_error;
3372 }
3373
3374 bytes = PyBytes_FromStringAndSize(NULL, len);
3375 if (bytes == NULL) {
3376 PyMem_Free(wstr);
3377 return NULL;
3378 }
3379
3380 len2 = wcstombs(PyBytes_AS_STRING(bytes), wstr, len+1);
3381 if (len2 == (size_t)-1 || len2 > len) {
Victor Stinner2f197072011-12-17 07:08:30 +01003382 error_pos = (size_t)-1;
Victor Stinnerf2ea71f2011-12-17 04:13:41 +01003383 goto encode_error;
3384 }
3385 PyMem_Free(wstr);
3386 }
3387 return bytes;
3388
3389encode_error:
3390 errmsg = strerror(errno);
3391 assert(errmsg != NULL);
Victor Stinner2f197072011-12-17 07:08:30 +01003392
3393 if (error_pos == (size_t)-1)
3394 error_pos = wcstombs_errorpos(wstr);
3395
Victor Stinnerf2ea71f2011-12-17 04:13:41 +01003396 PyMem_Free(wstr);
3397 Py_XDECREF(bytes);
3398
Victor Stinner2f197072011-12-17 07:08:30 +01003399 if (errmsg != NULL) {
3400 size_t errlen;
3401 wstr = _Py_char2wchar(errmsg, &errlen);
3402 if (wstr != NULL) {
3403 reason = PyUnicode_FromWideChar(wstr, errlen);
3404 PyMem_Free(wstr);
3405 } else
3406 errmsg = NULL;
3407 }
3408 if (errmsg == NULL)
Victor Stinner1f33f2b2011-12-17 04:45:09 +01003409 reason = PyUnicode_FromString(
3410 "wcstombs() encountered an unencodable "
3411 "wide character");
3412 if (reason == NULL)
3413 return NULL;
3414
3415 exc = PyObject_CallFunction(PyExc_UnicodeEncodeError, "sOnnO",
3416 "locale", unicode,
3417 (Py_ssize_t)error_pos,
3418 (Py_ssize_t)(error_pos+1),
3419 reason);
3420 Py_DECREF(reason);
3421 if (exc != NULL) {
3422 PyCodec_StrictErrors(exc);
3423 Py_XDECREF(exc);
3424 }
Victor Stinnerf2ea71f2011-12-17 04:13:41 +01003425 return NULL;
3426}
3427
Victor Stinnerad158722010-10-27 00:25:46 +00003428PyObject *
3429PyUnicode_EncodeFSDefault(PyObject *unicode)
Victor Stinnerae6265f2010-05-15 16:27:27 +00003430{
Victor Stinner99b95382011-07-04 14:23:54 +02003431#ifdef HAVE_MBCS
Victor Stinnerac931b12011-11-20 18:27:03 +01003432 return PyUnicode_EncodeCodePage(CP_ACP, unicode, NULL);
Victor Stinnerad158722010-10-27 00:25:46 +00003433#elif defined(__APPLE__)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003434 return _PyUnicode_AsUTF8String(unicode, "surrogateescape");
Victor Stinnerad158722010-10-27 00:25:46 +00003435#else
Victor Stinner793b5312011-04-27 00:24:21 +02003436 PyInterpreterState *interp = PyThreadState_GET()->interp;
3437 /* Bootstrap check: if the filesystem codec is implemented in Python, we
3438 cannot use it to encode and decode filenames before it is loaded. Load
3439 the Python codec requires to encode at least its own filename. Use the C
3440 version of the locale codec until the codec registry is initialized and
3441 the Python codec is loaded.
3442
3443 Py_FileSystemDefaultEncoding is shared between all interpreters, we
3444 cannot only rely on it: check also interp->fscodec_initialized for
3445 subinterpreters. */
3446 if (Py_FileSystemDefaultEncoding && interp->fscodec_initialized) {
Victor Stinnerae6265f2010-05-15 16:27:27 +00003447 return PyUnicode_AsEncodedString(unicode,
3448 Py_FileSystemDefaultEncoding,
3449 "surrogateescape");
Victor Stinnerc39211f2010-09-29 16:35:47 +00003450 }
3451 else {
Victor Stinner1b579672011-12-17 05:47:23 +01003452 return PyUnicode_EncodeLocale(unicode, "surrogateescape");
Victor Stinnerc39211f2010-09-29 16:35:47 +00003453 }
Victor Stinnerad158722010-10-27 00:25:46 +00003454#endif
Victor Stinnerae6265f2010-05-15 16:27:27 +00003455}
3456
Alexander Belopolsky40018472011-02-26 01:02:56 +00003457PyObject *
3458PyUnicode_AsEncodedString(PyObject *unicode,
Ezio Melotti2aa2b3b2011-09-29 00:58:57 +03003459 const char *encoding,
3460 const char *errors)
Guido van Rossumd57fd912000-03-10 22:53:23 +00003461{
3462 PyObject *v;
Victor Stinner600d3be2010-06-10 12:00:55 +00003463 char lower[11]; /* Enough for any encoding shortcut */
Tim Petersced69f82003-09-16 20:30:58 +00003464
Guido van Rossumd57fd912000-03-10 22:53:23 +00003465 if (!PyUnicode_Check(unicode)) {
3466 PyErr_BadArgument();
Amaury Forgeot d'Arcf0481112008-09-05 20:48:47 +00003467 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00003468 }
Fred Drakee4315f52000-05-09 19:53:39 +00003469
Fred Drakee4315f52000-05-09 19:53:39 +00003470 /* Shortcuts for common default encodings */
Victor Stinner20b654a2013-01-03 01:08:58 +01003471 if (_Py_normalize_encoding(encoding, lower, sizeof(lower))) {
Alexander Belopolsky1d521462011-02-25 19:19:57 +00003472 if ((strcmp(lower, "utf-8") == 0) ||
3473 (strcmp(lower, "utf8") == 0))
Victor Stinnera5c68c32011-03-02 01:03:14 +00003474 {
Victor Stinner2f283c22011-03-02 01:21:46 +00003475 if (errors == NULL || strcmp(errors, "strict") == 0)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003476 return _PyUnicode_AsUTF8String(unicode, NULL);
Victor Stinner2f283c22011-03-02 01:21:46 +00003477 else
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003478 return _PyUnicode_AsUTF8String(unicode, errors);
Victor Stinnera5c68c32011-03-02 01:03:14 +00003479 }
Victor Stinner37296e82010-06-10 13:36:23 +00003480 else if ((strcmp(lower, "latin-1") == 0) ||
Alexander Belopolsky1d521462011-02-25 19:19:57 +00003481 (strcmp(lower, "latin1") == 0) ||
Victor Stinner37296e82010-06-10 13:36:23 +00003482 (strcmp(lower, "iso-8859-1") == 0))
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003483 return _PyUnicode_AsLatin1String(unicode, errors);
Victor Stinner99b95382011-07-04 14:23:54 +02003484#ifdef HAVE_MBCS
Victor Stinnerac931b12011-11-20 18:27:03 +01003485 else if (strcmp(lower, "mbcs") == 0)
3486 return PyUnicode_EncodeCodePage(CP_ACP, unicode, errors);
Mark Hammond0ccda1e2003-07-01 00:13:27 +00003487#endif
Victor Stinner37296e82010-06-10 13:36:23 +00003488 else if (strcmp(lower, "ascii") == 0)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003489 return _PyUnicode_AsASCIIString(unicode, errors);
Victor Stinner37296e82010-06-10 13:36:23 +00003490 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00003491
3492 /* Encode via the codec registry */
3493 v = PyCodec_Encode(unicode, encoding, errors);
3494 if (v == NULL)
Amaury Forgeot d'Arcf0481112008-09-05 20:48:47 +00003495 return NULL;
3496
3497 /* The normal path */
3498 if (PyBytes_Check(v))
3499 return v;
3500
3501 /* If the codec returns a buffer, raise a warning and convert to bytes */
Marc-André Lemburgb2750b52008-06-06 12:18:17 +00003502 if (PyByteArray_Check(v)) {
Victor Stinner4a2b7a12010-08-13 14:03:48 +00003503 int error;
Amaury Forgeot d'Arcf0481112008-09-05 20:48:47 +00003504 PyObject *b;
Victor Stinner4a2b7a12010-08-13 14:03:48 +00003505
3506 error = PyErr_WarnFormat(PyExc_RuntimeWarning, 1,
3507 "encoder %s returned bytearray instead of bytes",
3508 encoding);
3509 if (error) {
Amaury Forgeot d'Arcf0481112008-09-05 20:48:47 +00003510 Py_DECREF(v);
3511 return NULL;
Marc-André Lemburgb2750b52008-06-06 12:18:17 +00003512 }
Marc-André Lemburgb2750b52008-06-06 12:18:17 +00003513
Amaury Forgeot d'Arcf0481112008-09-05 20:48:47 +00003514 b = PyBytes_FromStringAndSize(PyByteArray_AS_STRING(v), Py_SIZE(v));
3515 Py_DECREF(v);
3516 return b;
3517 }
3518
3519 PyErr_Format(PyExc_TypeError,
3520 "encoder did not return a bytes object (type=%.400s)",
3521 Py_TYPE(v)->tp_name);
3522 Py_DECREF(v);
Marc-André Lemburgb2750b52008-06-06 12:18:17 +00003523 return NULL;
3524}
3525
Alexander Belopolsky40018472011-02-26 01:02:56 +00003526PyObject *
3527PyUnicode_AsEncodedUnicode(PyObject *unicode,
Ezio Melotti2aa2b3b2011-09-29 00:58:57 +03003528 const char *encoding,
3529 const char *errors)
Marc-André Lemburgb2750b52008-06-06 12:18:17 +00003530{
3531 PyObject *v;
3532
3533 if (!PyUnicode_Check(unicode)) {
3534 PyErr_BadArgument();
3535 goto onError;
3536 }
3537
3538 if (encoding == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00003539 encoding = PyUnicode_GetDefaultEncoding();
Marc-André Lemburgb2750b52008-06-06 12:18:17 +00003540
3541 /* Encode via the codec registry */
3542 v = PyCodec_Encode(unicode, encoding, errors);
3543 if (v == NULL)
3544 goto onError;
3545 if (!PyUnicode_Check(v)) {
3546 PyErr_Format(PyExc_TypeError,
Benjamin Peterson142957c2008-07-04 19:55:29 +00003547 "encoder did not return an str object (type=%.400s)",
Marc-André Lemburgb2750b52008-06-06 12:18:17 +00003548 Py_TYPE(v)->tp_name);
3549 Py_DECREF(v);
3550 goto onError;
3551 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00003552 return v;
Tim Petersced69f82003-09-16 20:30:58 +00003553
Benjamin Peterson29060642009-01-31 22:14:21 +00003554 onError:
Guido van Rossumd57fd912000-03-10 22:53:23 +00003555 return NULL;
3556}
3557
Victor Stinner2f197072011-12-17 07:08:30 +01003558static size_t
3559mbstowcs_errorpos(const char *str, size_t len)
3560{
3561#ifdef HAVE_MBRTOWC
3562 const char *start = str;
3563 mbstate_t mbs;
3564 size_t converted;
3565 wchar_t ch;
3566
3567 memset(&mbs, 0, sizeof mbs);
3568 while (len)
3569 {
3570 converted = mbrtowc(&ch, (char*)str, len, &mbs);
3571 if (converted == 0)
3572 /* Reached end of string */
3573 break;
3574 if (converted == (size_t)-1 || converted == (size_t)-2) {
3575 /* Conversion error or incomplete character */
3576 return str - start;
3577 }
3578 else {
3579 str += converted;
3580 len -= converted;
3581 }
3582 }
3583 /* failed to find the undecodable byte sequence */
3584 return 0;
3585#endif
3586 return 0;
3587}
3588
Guido van Rossum00bc0e02007-10-15 02:52:41 +00003589PyObject*
Victor Stinneraf02e1c2011-12-16 23:56:01 +01003590PyUnicode_DecodeLocaleAndSize(const char *str, Py_ssize_t len,
Victor Stinner1b579672011-12-17 05:47:23 +01003591 const char *errors)
Victor Stinneraf02e1c2011-12-16 23:56:01 +01003592{
3593 wchar_t smallbuf[256];
3594 size_t smallbuf_len = Py_ARRAY_LENGTH(smallbuf);
3595 wchar_t *wstr;
3596 size_t wlen, wlen2;
3597 PyObject *unicode;
Victor Stinner1b579672011-12-17 05:47:23 +01003598 int surrogateescape;
Victor Stinner2f197072011-12-17 07:08:30 +01003599 size_t error_pos;
3600 char *errmsg;
3601 PyObject *reason, *exc;
Victor Stinner1b579672011-12-17 05:47:23 +01003602
3603 if (locale_error_handler(errors, &surrogateescape) < 0)
3604 return NULL;
Victor Stinneraf02e1c2011-12-16 23:56:01 +01003605
3606 if (str[len] != '\0' || len != strlen(str)) {
3607 PyErr_SetString(PyExc_TypeError, "embedded null character");
3608 return NULL;
3609 }
3610
3611 if (surrogateescape)
3612 {
3613 wstr = _Py_char2wchar(str, &wlen);
3614 if (wstr == NULL) {
3615 if (wlen == (size_t)-1)
3616 PyErr_NoMemory();
3617 else
3618 PyErr_SetFromErrno(PyExc_OSError);
3619 return NULL;
3620 }
3621
3622 unicode = PyUnicode_FromWideChar(wstr, wlen);
3623 PyMem_Free(wstr);
3624 }
3625 else {
3626#ifndef HAVE_BROKEN_MBSTOWCS
3627 wlen = mbstowcs(NULL, str, 0);
3628#else
3629 wlen = len;
3630#endif
Victor Stinner2f197072011-12-17 07:08:30 +01003631 if (wlen == (size_t)-1)
3632 goto decode_error;
Victor Stinneraf02e1c2011-12-16 23:56:01 +01003633 if (wlen+1 <= smallbuf_len) {
3634 wstr = smallbuf;
3635 }
3636 else {
3637 if (wlen > PY_SSIZE_T_MAX / sizeof(wchar_t) - 1)
3638 return PyErr_NoMemory();
3639
3640 wstr = PyMem_Malloc((wlen+1) * sizeof(wchar_t));
3641 if (!wstr)
3642 return PyErr_NoMemory();
3643 }
3644
3645 /* This shouldn't fail now */
3646 wlen2 = mbstowcs(wstr, str, wlen+1);
3647 if (wlen2 == (size_t)-1) {
3648 if (wstr != smallbuf)
3649 PyMem_Free(wstr);
Victor Stinner2f197072011-12-17 07:08:30 +01003650 goto decode_error;
Victor Stinneraf02e1c2011-12-16 23:56:01 +01003651 }
3652#ifdef HAVE_BROKEN_MBSTOWCS
3653 assert(wlen2 == wlen);
3654#endif
3655 unicode = PyUnicode_FromWideChar(wstr, wlen2);
3656 if (wstr != smallbuf)
3657 PyMem_Free(wstr);
3658 }
3659 return unicode;
Victor Stinner2f197072011-12-17 07:08:30 +01003660
3661decode_error:
3662 errmsg = strerror(errno);
3663 assert(errmsg != NULL);
3664
3665 error_pos = mbstowcs_errorpos(str, len);
3666 if (errmsg != NULL) {
3667 size_t errlen;
3668 wstr = _Py_char2wchar(errmsg, &errlen);
3669 if (wstr != NULL) {
3670 reason = PyUnicode_FromWideChar(wstr, errlen);
3671 PyMem_Free(wstr);
3672 } else
3673 errmsg = NULL;
3674 }
3675 if (errmsg == NULL)
3676 reason = PyUnicode_FromString(
3677 "mbstowcs() encountered an invalid multibyte sequence");
3678 if (reason == NULL)
3679 return NULL;
3680
3681 exc = PyObject_CallFunction(PyExc_UnicodeDecodeError, "sy#nnO",
3682 "locale", str, len,
3683 (Py_ssize_t)error_pos,
3684 (Py_ssize_t)(error_pos+1),
3685 reason);
3686 Py_DECREF(reason);
3687 if (exc != NULL) {
3688 PyCodec_StrictErrors(exc);
3689 Py_XDECREF(exc);
3690 }
3691 return NULL;
Victor Stinneraf02e1c2011-12-16 23:56:01 +01003692}
3693
3694PyObject*
Victor Stinner1b579672011-12-17 05:47:23 +01003695PyUnicode_DecodeLocale(const char *str, const char *errors)
Victor Stinneraf02e1c2011-12-16 23:56:01 +01003696{
3697 Py_ssize_t size = (Py_ssize_t)strlen(str);
Victor Stinner1b579672011-12-17 05:47:23 +01003698 return PyUnicode_DecodeLocaleAndSize(str, size, errors);
Victor Stinneraf02e1c2011-12-16 23:56:01 +01003699}
3700
3701
3702PyObject*
Christian Heimes5894ba72007-11-04 11:43:14 +00003703PyUnicode_DecodeFSDefault(const char *s) {
Guido van Rossum00bc0e02007-10-15 02:52:41 +00003704 Py_ssize_t size = (Py_ssize_t)strlen(s);
Christian Heimes5894ba72007-11-04 11:43:14 +00003705 return PyUnicode_DecodeFSDefaultAndSize(s, size);
3706}
Guido van Rossum00bc0e02007-10-15 02:52:41 +00003707
Christian Heimes5894ba72007-11-04 11:43:14 +00003708PyObject*
3709PyUnicode_DecodeFSDefaultAndSize(const char *s, Py_ssize_t size)
3710{
Victor Stinner99b95382011-07-04 14:23:54 +02003711#ifdef HAVE_MBCS
Victor Stinnerad158722010-10-27 00:25:46 +00003712 return PyUnicode_DecodeMBCS(s, size, NULL);
3713#elif defined(__APPLE__)
Victor Stinnera1d12bb2011-12-11 21:53:09 +01003714 return PyUnicode_DecodeUTF8Stateful(s, size, "surrogateescape", NULL);
Victor Stinnerad158722010-10-27 00:25:46 +00003715#else
Victor Stinner793b5312011-04-27 00:24:21 +02003716 PyInterpreterState *interp = PyThreadState_GET()->interp;
3717 /* Bootstrap check: if the filesystem codec is implemented in Python, we
3718 cannot use it to encode and decode filenames before it is loaded. Load
3719 the Python codec requires to encode at least its own filename. Use the C
3720 version of the locale codec until the codec registry is initialized and
3721 the Python codec is loaded.
3722
3723 Py_FileSystemDefaultEncoding is shared between all interpreters, we
3724 cannot only rely on it: check also interp->fscodec_initialized for
3725 subinterpreters. */
3726 if (Py_FileSystemDefaultEncoding && interp->fscodec_initialized) {
Guido van Rossum00bc0e02007-10-15 02:52:41 +00003727 return PyUnicode_Decode(s, size,
3728 Py_FileSystemDefaultEncoding,
Victor Stinnerb9a20ad2010-04-30 16:37:52 +00003729 "surrogateescape");
Guido van Rossum00bc0e02007-10-15 02:52:41 +00003730 }
3731 else {
Victor Stinner1b579672011-12-17 05:47:23 +01003732 return PyUnicode_DecodeLocaleAndSize(s, size, "surrogateescape");
Guido van Rossum00bc0e02007-10-15 02:52:41 +00003733 }
Victor Stinnerad158722010-10-27 00:25:46 +00003734#endif
Guido van Rossum00bc0e02007-10-15 02:52:41 +00003735}
3736
Martin v. Löwis011e8422009-05-05 04:43:17 +00003737
3738int
Antoine Pitrou13348842012-01-29 18:36:34 +01003739_PyUnicode_HasNULChars(PyObject* s)
3740{
3741 static PyObject *nul = NULL;
3742
3743 if (nul == NULL)
3744 nul = PyUnicode_FromStringAndSize("\0", 1);
3745 if (nul == NULL)
3746 return -1;
3747 return PyUnicode_Contains(s, nul);
3748}
3749
3750
3751int
Martin v. Löwis011e8422009-05-05 04:43:17 +00003752PyUnicode_FSConverter(PyObject* arg, void* addr)
3753{
3754 PyObject *output = NULL;
3755 Py_ssize_t size;
3756 void *data;
Martin v. Löwisc15bdef2009-05-29 14:47:46 +00003757 if (arg == NULL) {
3758 Py_DECREF(*(PyObject**)addr);
3759 return 1;
3760 }
Victor Stinnerdcb24032010-04-22 12:08:36 +00003761 if (PyBytes_Check(arg)) {
Martin v. Löwis011e8422009-05-05 04:43:17 +00003762 output = arg;
3763 Py_INCREF(output);
3764 }
3765 else {
3766 arg = PyUnicode_FromObject(arg);
3767 if (!arg)
3768 return 0;
Victor Stinnerae6265f2010-05-15 16:27:27 +00003769 output = PyUnicode_EncodeFSDefault(arg);
Martin v. Löwis011e8422009-05-05 04:43:17 +00003770 Py_DECREF(arg);
3771 if (!output)
3772 return 0;
3773 if (!PyBytes_Check(output)) {
3774 Py_DECREF(output);
3775 PyErr_SetString(PyExc_TypeError, "encoder failed to return bytes");
3776 return 0;
3777 }
3778 }
Victor Stinner0ea2a462010-04-30 00:22:08 +00003779 size = PyBytes_GET_SIZE(output);
3780 data = PyBytes_AS_STRING(output);
Martin v. Löwis011e8422009-05-05 04:43:17 +00003781 if (size != strlen(data)) {
Benjamin Peterson7a6b44a2011-08-18 13:51:47 -05003782 PyErr_SetString(PyExc_TypeError, "embedded NUL character");
Martin v. Löwis011e8422009-05-05 04:43:17 +00003783 Py_DECREF(output);
3784 return 0;
3785 }
3786 *(PyObject**)addr = output;
Martin v. Löwisc15bdef2009-05-29 14:47:46 +00003787 return Py_CLEANUP_SUPPORTED;
Martin v. Löwis011e8422009-05-05 04:43:17 +00003788}
3789
3790
Victor Stinner47fcb5b2010-08-13 23:59:58 +00003791int
3792PyUnicode_FSDecoder(PyObject* arg, void* addr)
3793{
3794 PyObject *output = NULL;
Victor Stinner47fcb5b2010-08-13 23:59:58 +00003795 if (arg == NULL) {
3796 Py_DECREF(*(PyObject**)addr);
3797 return 1;
3798 }
3799 if (PyUnicode_Check(arg)) {
Benjamin Petersonbac79492012-01-14 13:34:47 -05003800 if (PyUnicode_READY(arg) == -1)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003801 return 0;
Victor Stinner47fcb5b2010-08-13 23:59:58 +00003802 output = arg;
3803 Py_INCREF(output);
3804 }
3805 else {
3806 arg = PyBytes_FromObject(arg);
3807 if (!arg)
3808 return 0;
3809 output = PyUnicode_DecodeFSDefaultAndSize(PyBytes_AS_STRING(arg),
3810 PyBytes_GET_SIZE(arg));
3811 Py_DECREF(arg);
3812 if (!output)
3813 return 0;
3814 if (!PyUnicode_Check(output)) {
3815 Py_DECREF(output);
3816 PyErr_SetString(PyExc_TypeError, "decoder failed to return unicode");
3817 return 0;
3818 }
3819 }
Benjamin Petersonbac79492012-01-14 13:34:47 -05003820 if (PyUnicode_READY(output) == -1) {
Victor Stinner065836e2011-10-27 01:56:33 +02003821 Py_DECREF(output);
3822 return 0;
3823 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003824 if (findchar(PyUnicode_DATA(output), PyUnicode_KIND(output),
Antoine Pitrouf0b934b2011-10-13 18:55:09 +02003825 PyUnicode_GET_LENGTH(output), 0, 1) >= 0) {
Victor Stinner47fcb5b2010-08-13 23:59:58 +00003826 PyErr_SetString(PyExc_TypeError, "embedded NUL character");
3827 Py_DECREF(output);
3828 return 0;
3829 }
3830 *(PyObject**)addr = output;
3831 return Py_CLEANUP_SUPPORTED;
3832}
3833
3834
Martin v. Löwis5b222132007-06-10 09:51:05 +00003835char*
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003836PyUnicode_AsUTF8AndSize(PyObject *unicode, Py_ssize_t *psize)
Martin v. Löwis5b222132007-06-10 09:51:05 +00003837{
Christian Heimesf3863112007-11-22 07:46:41 +00003838 PyObject *bytes;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003839
Neal Norwitze0a0a6e2007-08-25 01:04:21 +00003840 if (!PyUnicode_Check(unicode)) {
3841 PyErr_BadArgument();
3842 return NULL;
3843 }
Victor Stinner9db1a8b2011-10-23 20:04:37 +02003844 if (PyUnicode_READY(unicode) == -1)
Martin v. Löwis5b222132007-06-10 09:51:05 +00003845 return NULL;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003846
Victor Stinnere90fe6a2011-10-01 16:48:13 +02003847 if (PyUnicode_UTF8(unicode) == NULL) {
3848 assert(!PyUnicode_IS_COMPACT_ASCII(unicode));
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003849 bytes = _PyUnicode_AsUTF8String(unicode, "strict");
3850 if (bytes == NULL)
3851 return NULL;
Victor Stinner9db1a8b2011-10-23 20:04:37 +02003852 _PyUnicode_UTF8(unicode) = PyObject_MALLOC(PyBytes_GET_SIZE(bytes) + 1);
3853 if (_PyUnicode_UTF8(unicode) == NULL) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003854 Py_DECREF(bytes);
3855 return NULL;
3856 }
Victor Stinner9db1a8b2011-10-23 20:04:37 +02003857 _PyUnicode_UTF8_LENGTH(unicode) = PyBytes_GET_SIZE(bytes);
3858 Py_MEMCPY(_PyUnicode_UTF8(unicode),
3859 PyBytes_AS_STRING(bytes),
3860 _PyUnicode_UTF8_LENGTH(unicode) + 1);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003861 Py_DECREF(bytes);
3862 }
3863
3864 if (psize)
Victor Stinnere90fe6a2011-10-01 16:48:13 +02003865 *psize = PyUnicode_UTF8_LENGTH(unicode);
3866 return PyUnicode_UTF8(unicode);
Guido van Rossum7d1df6c2007-08-29 13:53:23 +00003867}
3868
3869char*
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003870PyUnicode_AsUTF8(PyObject *unicode)
Guido van Rossum7d1df6c2007-08-29 13:53:23 +00003871{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003872 return PyUnicode_AsUTF8AndSize(unicode, NULL);
3873}
3874
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003875Py_UNICODE *
3876PyUnicode_AsUnicodeAndSize(PyObject *unicode, Py_ssize_t *size)
3877{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003878 const unsigned char *one_byte;
3879#if SIZEOF_WCHAR_T == 4
3880 const Py_UCS2 *two_bytes;
3881#else
3882 const Py_UCS4 *four_bytes;
3883 const Py_UCS4 *ucs4_end;
3884 Py_ssize_t num_surrogates;
3885#endif
3886 wchar_t *w;
3887 wchar_t *wchar_end;
3888
3889 if (!PyUnicode_Check(unicode)) {
3890 PyErr_BadArgument();
3891 return NULL;
3892 }
Victor Stinner9db1a8b2011-10-23 20:04:37 +02003893 if (_PyUnicode_WSTR(unicode) == NULL) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003894 /* Non-ASCII compact unicode object */
Victor Stinner9db1a8b2011-10-23 20:04:37 +02003895 assert(_PyUnicode_KIND(unicode) != 0);
3896 assert(PyUnicode_IS_READY(unicode));
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003897
Victor Stinner9db1a8b2011-10-23 20:04:37 +02003898 if (PyUnicode_KIND(unicode) == PyUnicode_4BYTE_KIND) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003899#if SIZEOF_WCHAR_T == 2
Victor Stinner9db1a8b2011-10-23 20:04:37 +02003900 four_bytes = PyUnicode_4BYTE_DATA(unicode);
3901 ucs4_end = four_bytes + _PyUnicode_LENGTH(unicode);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003902 num_surrogates = 0;
3903
3904 for (; four_bytes < ucs4_end; ++four_bytes) {
3905 if (*four_bytes > 0xFFFF)
3906 ++num_surrogates;
3907 }
3908
Victor Stinner9db1a8b2011-10-23 20:04:37 +02003909 _PyUnicode_WSTR(unicode) = (wchar_t *) PyObject_MALLOC(
3910 sizeof(wchar_t) * (_PyUnicode_LENGTH(unicode) + 1 + num_surrogates));
3911 if (!_PyUnicode_WSTR(unicode)) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003912 PyErr_NoMemory();
3913 return NULL;
3914 }
Victor Stinner9db1a8b2011-10-23 20:04:37 +02003915 _PyUnicode_WSTR_LENGTH(unicode) = _PyUnicode_LENGTH(unicode) + num_surrogates;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003916
Victor Stinner9db1a8b2011-10-23 20:04:37 +02003917 w = _PyUnicode_WSTR(unicode);
3918 wchar_end = w + _PyUnicode_WSTR_LENGTH(unicode);
3919 four_bytes = PyUnicode_4BYTE_DATA(unicode);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003920 for (; four_bytes < ucs4_end; ++four_bytes, ++w) {
3921 if (*four_bytes > 0xFFFF) {
Victor Stinner8faf8212011-12-08 22:14:11 +01003922 assert(*four_bytes <= MAX_UNICODE);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003923 /* encode surrogate pair in this case */
Victor Stinner551ac952011-11-29 22:58:13 +01003924 *w++ = Py_UNICODE_HIGH_SURROGATE(*four_bytes);
3925 *w = Py_UNICODE_LOW_SURROGATE(*four_bytes);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003926 }
3927 else
3928 *w = *four_bytes;
3929
3930 if (w > wchar_end) {
3931 assert(0 && "Miscalculated string end");
3932 }
3933 }
3934 *w = 0;
3935#else
3936 /* sizeof(wchar_t) == 4 */
3937 Py_FatalError("Impossible unicode object state, wstr and str "
3938 "should share memory already.");
3939 return NULL;
3940#endif
3941 }
3942 else {
Victor Stinner9db1a8b2011-10-23 20:04:37 +02003943 _PyUnicode_WSTR(unicode) = (wchar_t *) PyObject_MALLOC(sizeof(wchar_t) *
3944 (_PyUnicode_LENGTH(unicode) + 1));
3945 if (!_PyUnicode_WSTR(unicode)) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003946 PyErr_NoMemory();
3947 return NULL;
3948 }
Victor Stinner9db1a8b2011-10-23 20:04:37 +02003949 if (!PyUnicode_IS_COMPACT_ASCII(unicode))
3950 _PyUnicode_WSTR_LENGTH(unicode) = _PyUnicode_LENGTH(unicode);
3951 w = _PyUnicode_WSTR(unicode);
3952 wchar_end = w + _PyUnicode_LENGTH(unicode);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003953
Victor Stinner9db1a8b2011-10-23 20:04:37 +02003954 if (PyUnicode_KIND(unicode) == PyUnicode_1BYTE_KIND) {
3955 one_byte = PyUnicode_1BYTE_DATA(unicode);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003956 for (; w < wchar_end; ++one_byte, ++w)
3957 *w = *one_byte;
3958 /* null-terminate the wstr */
3959 *w = 0;
3960 }
Victor Stinner9db1a8b2011-10-23 20:04:37 +02003961 else if (PyUnicode_KIND(unicode) == PyUnicode_2BYTE_KIND) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003962#if SIZEOF_WCHAR_T == 4
Victor Stinner9db1a8b2011-10-23 20:04:37 +02003963 two_bytes = PyUnicode_2BYTE_DATA(unicode);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003964 for (; w < wchar_end; ++two_bytes, ++w)
3965 *w = *two_bytes;
3966 /* null-terminate the wstr */
3967 *w = 0;
3968#else
3969 /* sizeof(wchar_t) == 2 */
Victor Stinner9db1a8b2011-10-23 20:04:37 +02003970 PyObject_FREE(_PyUnicode_WSTR(unicode));
3971 _PyUnicode_WSTR(unicode) = NULL;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003972 Py_FatalError("Impossible unicode object state, wstr "
3973 "and str should share memory already.");
3974 return NULL;
3975#endif
3976 }
3977 else {
3978 assert(0 && "This should never happen.");
3979 }
3980 }
3981 }
3982 if (size != NULL)
Victor Stinner9db1a8b2011-10-23 20:04:37 +02003983 *size = PyUnicode_WSTR_LENGTH(unicode);
3984 return _PyUnicode_WSTR(unicode);
Martin v. Löwis5b222132007-06-10 09:51:05 +00003985}
3986
Alexander Belopolsky40018472011-02-26 01:02:56 +00003987Py_UNICODE *
3988PyUnicode_AsUnicode(PyObject *unicode)
Guido van Rossumd57fd912000-03-10 22:53:23 +00003989{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003990 return PyUnicode_AsUnicodeAndSize(unicode, NULL);
Guido van Rossumd57fd912000-03-10 22:53:23 +00003991}
3992
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003993
Alexander Belopolsky40018472011-02-26 01:02:56 +00003994Py_ssize_t
3995PyUnicode_GetSize(PyObject *unicode)
Guido van Rossumd57fd912000-03-10 22:53:23 +00003996{
3997 if (!PyUnicode_Check(unicode)) {
3998 PyErr_BadArgument();
3999 goto onError;
4000 }
4001 return PyUnicode_GET_SIZE(unicode);
4002
Benjamin Peterson29060642009-01-31 22:14:21 +00004003 onError:
Guido van Rossumd57fd912000-03-10 22:53:23 +00004004 return -1;
4005}
4006
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02004007Py_ssize_t
4008PyUnicode_GetLength(PyObject *unicode)
4009{
Victor Stinner07621332012-06-16 04:53:46 +02004010 if (!PyUnicode_Check(unicode)) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02004011 PyErr_BadArgument();
4012 return -1;
4013 }
Victor Stinner07621332012-06-16 04:53:46 +02004014 if (PyUnicode_READY(unicode) == -1)
4015 return -1;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02004016 return PyUnicode_GET_LENGTH(unicode);
4017}
4018
4019Py_UCS4
4020PyUnicode_ReadChar(PyObject *unicode, Py_ssize_t index)
4021{
Victor Stinner2fe5ced2011-10-02 00:25:40 +02004022 if (!PyUnicode_Check(unicode) || PyUnicode_READY(unicode) == -1) {
4023 PyErr_BadArgument();
4024 return (Py_UCS4)-1;
4025 }
Victor Stinnerc4b49542011-12-11 22:44:26 +01004026 if (index < 0 || index >= PyUnicode_GET_LENGTH(unicode)) {
Victor Stinner2fe5ced2011-10-02 00:25:40 +02004027 PyErr_SetString(PyExc_IndexError, "string index out of range");
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02004028 return (Py_UCS4)-1;
4029 }
4030 return PyUnicode_READ_CHAR(unicode, index);
4031}
4032
4033int
4034PyUnicode_WriteChar(PyObject *unicode, Py_ssize_t index, Py_UCS4 ch)
4035{
4036 if (!PyUnicode_Check(unicode) || !PyUnicode_IS_COMPACT(unicode)) {
Victor Stinnercd9950f2011-10-02 00:34:53 +02004037 PyErr_BadArgument();
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02004038 return -1;
4039 }
Victor Stinner488fa492011-12-12 00:01:39 +01004040 assert(PyUnicode_IS_READY(unicode));
Victor Stinnerc4b49542011-12-11 22:44:26 +01004041 if (index < 0 || index >= PyUnicode_GET_LENGTH(unicode)) {
Victor Stinnercd9950f2011-10-02 00:34:53 +02004042 PyErr_SetString(PyExc_IndexError, "string index out of range");
4043 return -1;
4044 }
Victor Stinner488fa492011-12-12 00:01:39 +01004045 if (unicode_check_modifiable(unicode))
Victor Stinnercd9950f2011-10-02 00:34:53 +02004046 return -1;
Victor Stinnerc9590ad2012-03-04 01:34:37 +01004047 if (ch > PyUnicode_MAX_CHAR_VALUE(unicode)) {
4048 PyErr_SetString(PyExc_ValueError, "character out of range");
4049 return -1;
4050 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02004051 PyUnicode_WRITE(PyUnicode_KIND(unicode), PyUnicode_DATA(unicode),
4052 index, ch);
4053 return 0;
4054}
4055
Alexander Belopolsky40018472011-02-26 01:02:56 +00004056const char *
4057PyUnicode_GetDefaultEncoding(void)
Fred Drakee4315f52000-05-09 19:53:39 +00004058{
Victor Stinner42cb4622010-09-01 19:39:01 +00004059 return "utf-8";
Fred Drakee4315f52000-05-09 19:53:39 +00004060}
4061
Victor Stinner554f3f02010-06-16 23:33:54 +00004062/* create or adjust a UnicodeDecodeError */
4063static void
4064make_decode_exception(PyObject **exceptionObject,
4065 const char *encoding,
4066 const char *input, Py_ssize_t length,
4067 Py_ssize_t startpos, Py_ssize_t endpos,
4068 const char *reason)
4069{
4070 if (*exceptionObject == NULL) {
4071 *exceptionObject = PyUnicodeDecodeError_Create(
4072 encoding, input, length, startpos, endpos, reason);
4073 }
4074 else {
4075 if (PyUnicodeDecodeError_SetStart(*exceptionObject, startpos))
4076 goto onError;
4077 if (PyUnicodeDecodeError_SetEnd(*exceptionObject, endpos))
4078 goto onError;
4079 if (PyUnicodeDecodeError_SetReason(*exceptionObject, reason))
4080 goto onError;
4081 }
4082 return;
4083
4084onError:
4085 Py_DECREF(*exceptionObject);
4086 *exceptionObject = NULL;
4087}
4088
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004089/* error handling callback helper:
4090 build arguments, call the callback and check the arguments,
Fred Drakedb390c12005-10-28 14:39:47 +00004091 if no exception occurred, copy the replacement to the output
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004092 and adjust various state variables.
4093 return 0 on success, -1 on error
4094*/
4095
Alexander Belopolsky40018472011-02-26 01:02:56 +00004096static int
4097unicode_decode_call_errorhandler(const char *errors, PyObject **errorHandler,
Ezio Melotti2aa2b3b2011-09-29 00:58:57 +03004098 const char *encoding, const char *reason,
4099 const char **input, const char **inend, Py_ssize_t *startinpos,
4100 Py_ssize_t *endinpos, PyObject **exceptionObject, const char **inptr,
Martin v. Löwise9b11c12011-11-08 17:35:34 +01004101 PyObject **output, Py_ssize_t *outpos)
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004102{
Benjamin Peterson142957c2008-07-04 19:55:29 +00004103 static char *argparse = "O!n;decoding error handler must return (str, int) tuple";
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004104
4105 PyObject *restuple = NULL;
4106 PyObject *repunicode = NULL;
Victor Stinner596a6c42011-11-09 00:02:18 +01004107 Py_ssize_t outsize;
Walter Dörwalde78178e2007-07-30 13:31:40 +00004108 Py_ssize_t insize;
Martin v. Löwis18e16552006-02-15 17:27:45 +00004109 Py_ssize_t requiredsize;
4110 Py_ssize_t newpos;
Walter Dörwalde78178e2007-07-30 13:31:40 +00004111 PyObject *inputobj = NULL;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004112 int res = -1;
4113
Victor Stinner596a6c42011-11-09 00:02:18 +01004114 if (_PyUnicode_KIND(*output) != PyUnicode_WCHAR_KIND)
4115 outsize = PyUnicode_GET_LENGTH(*output);
4116 else
4117 outsize = _PyUnicode_WSTR_LENGTH(*output);
4118
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004119 if (*errorHandler == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +00004120 *errorHandler = PyCodec_LookupError(errors);
4121 if (*errorHandler == NULL)
4122 goto onError;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004123 }
4124
Victor Stinner554f3f02010-06-16 23:33:54 +00004125 make_decode_exception(exceptionObject,
4126 encoding,
4127 *input, *inend - *input,
4128 *startinpos, *endinpos,
4129 reason);
4130 if (*exceptionObject == NULL)
4131 goto onError;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004132
4133 restuple = PyObject_CallFunctionObjArgs(*errorHandler, *exceptionObject, NULL);
4134 if (restuple == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00004135 goto onError;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004136 if (!PyTuple_Check(restuple)) {
Benjamin Petersond75fcb42009-02-19 04:22:03 +00004137 PyErr_SetString(PyExc_TypeError, &argparse[4]);
Benjamin Peterson29060642009-01-31 22:14:21 +00004138 goto onError;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004139 }
4140 if (!PyArg_ParseTuple(restuple, argparse, &PyUnicode_Type, &repunicode, &newpos))
Benjamin Peterson29060642009-01-31 22:14:21 +00004141 goto onError;
Benjamin Petersonbac79492012-01-14 13:34:47 -05004142 if (PyUnicode_READY(repunicode) == -1)
Martin v. Löwise9b11c12011-11-08 17:35:34 +01004143 goto onError;
Walter Dörwalde78178e2007-07-30 13:31:40 +00004144
4145 /* Copy back the bytes variables, which might have been modified by the
4146 callback */
4147 inputobj = PyUnicodeDecodeError_GetObject(*exceptionObject);
4148 if (!inputobj)
4149 goto onError;
Christian Heimes72b710a2008-05-26 13:28:38 +00004150 if (!PyBytes_Check(inputobj)) {
Benjamin Peterson29060642009-01-31 22:14:21 +00004151 PyErr_Format(PyExc_TypeError, "exception attribute object must be bytes");
Walter Dörwalde78178e2007-07-30 13:31:40 +00004152 }
Christian Heimes72b710a2008-05-26 13:28:38 +00004153 *input = PyBytes_AS_STRING(inputobj);
4154 insize = PyBytes_GET_SIZE(inputobj);
Walter Dörwalde78178e2007-07-30 13:31:40 +00004155 *inend = *input + insize;
Walter Dörwald36f938f2007-08-10 10:11:43 +00004156 /* we can DECREF safely, as the exception has another reference,
4157 so the object won't go away. */
4158 Py_DECREF(inputobj);
Walter Dörwalde78178e2007-07-30 13:31:40 +00004159
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004160 if (newpos<0)
Benjamin Peterson29060642009-01-31 22:14:21 +00004161 newpos = insize+newpos;
Walter Dörwald2e0b18a2003-01-31 17:19:08 +00004162 if (newpos<0 || newpos>insize) {
Benjamin Peterson29060642009-01-31 22:14:21 +00004163 PyErr_Format(PyExc_IndexError, "position %zd from error handler out of bounds", newpos);
4164 goto onError;
Walter Dörwald2e0b18a2003-01-31 17:19:08 +00004165 }
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004166
Victor Stinner596a6c42011-11-09 00:02:18 +01004167 if (_PyUnicode_KIND(*output) != PyUnicode_WCHAR_KIND) {
4168 /* need more space? (at least enough for what we
4169 have+the replacement+the rest of the string (starting
4170 at the new input position), so we won't have to check space
4171 when there are no errors in the rest of the string) */
4172 Py_ssize_t replen = PyUnicode_GET_LENGTH(repunicode);
4173 requiredsize = *outpos + replen + insize-newpos;
4174 if (requiredsize > outsize) {
4175 if (requiredsize<2*outsize)
4176 requiredsize = 2*outsize;
4177 if (unicode_resize(output, requiredsize) < 0)
4178 goto onError;
4179 }
Victor Stinner1b487b42012-05-03 12:29:04 +02004180 if (unicode_widen(output, *outpos,
4181 PyUnicode_MAX_CHAR_VALUE(repunicode)) < 0)
Benjamin Peterson29060642009-01-31 22:14:21 +00004182 goto onError;
Victor Stinnerd3f08822012-05-29 12:57:52 +02004183 _PyUnicode_FastCopyCharacters(*output, *outpos, repunicode, 0, replen);
Victor Stinner596a6c42011-11-09 00:02:18 +01004184 *outpos += replen;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004185 }
Victor Stinner596a6c42011-11-09 00:02:18 +01004186 else {
4187 wchar_t *repwstr;
4188 Py_ssize_t repwlen;
4189 repwstr = PyUnicode_AsUnicodeAndSize(repunicode, &repwlen);
4190 if (repwstr == NULL)
4191 goto onError;
4192 /* need more space? (at least enough for what we
4193 have+the replacement+the rest of the string (starting
4194 at the new input position), so we won't have to check space
4195 when there are no errors in the rest of the string) */
4196 requiredsize = *outpos + repwlen + insize-newpos;
4197 if (requiredsize > outsize) {
4198 if (requiredsize < 2*outsize)
4199 requiredsize = 2*outsize;
4200 if (unicode_resize(output, requiredsize) < 0)
4201 goto onError;
4202 }
4203 wcsncpy(_PyUnicode_WSTR(*output) + *outpos, repwstr, repwlen);
4204 *outpos += repwlen;
4205 }
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004206 *endinpos = newpos;
Walter Dörwalde78178e2007-07-30 13:31:40 +00004207 *inptr = *input + newpos;
Walter Dörwalde78178e2007-07-30 13:31:40 +00004208
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004209 /* we made it! */
4210 res = 0;
4211
Benjamin Peterson29060642009-01-31 22:14:21 +00004212 onError:
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004213 Py_XDECREF(restuple);
4214 return res;
4215}
4216
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004217/* --- UTF-7 Codec -------------------------------------------------------- */
4218
Antoine Pitrou244651a2009-05-04 18:56:13 +00004219/* See RFC2152 for details. We encode conservatively and decode liberally. */
4220
4221/* Three simple macros defining base-64. */
4222
4223/* Is c a base-64 character? */
4224
4225#define IS_BASE64(c) \
4226 (((c) >= 'A' && (c) <= 'Z') || \
4227 ((c) >= 'a' && (c) <= 'z') || \
4228 ((c) >= '0' && (c) <= '9') || \
4229 (c) == '+' || (c) == '/')
4230
4231/* given that c is a base-64 character, what is its base-64 value? */
4232
4233#define FROM_BASE64(c) \
4234 (((c) >= 'A' && (c) <= 'Z') ? (c) - 'A' : \
4235 ((c) >= 'a' && (c) <= 'z') ? (c) - 'a' + 26 : \
4236 ((c) >= '0' && (c) <= '9') ? (c) - '0' + 52 : \
4237 (c) == '+' ? 62 : 63)
4238
4239/* What is the base-64 character of the bottom 6 bits of n? */
4240
4241#define TO_BASE64(n) \
4242 ("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/"[(n) & 0x3f])
4243
4244/* DECODE_DIRECT: this byte encountered in a UTF-7 string should be
4245 * decoded as itself. We are permissive on decoding; the only ASCII
4246 * byte not decoding to itself is the + which begins a base64
4247 * string. */
4248
4249#define DECODE_DIRECT(c) \
4250 ((c) <= 127 && (c) != '+')
4251
4252/* The UTF-7 encoder treats ASCII characters differently according to
4253 * whether they are Set D, Set O, Whitespace, or special (i.e. none of
4254 * the above). See RFC2152. This array identifies these different
4255 * sets:
4256 * 0 : "Set D"
4257 * alphanumeric and '(),-./:?
4258 * 1 : "Set O"
4259 * !"#$%&*;<=>@[]^_`{|}
4260 * 2 : "whitespace"
4261 * ht nl cr sp
4262 * 3 : special (must be base64 encoded)
4263 * everything else (i.e. +\~ and non-printing codes 0-8 11-12 14-31 127)
4264 */
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004265
Tim Petersced69f82003-09-16 20:30:58 +00004266static
Antoine Pitrou244651a2009-05-04 18:56:13 +00004267char utf7_category[128] = {
4268/* nul soh stx etx eot enq ack bel bs ht nl vt np cr so si */
4269 3, 3, 3, 3, 3, 3, 3, 3, 3, 2, 2, 3, 3, 2, 3, 3,
4270/* dle dc1 dc2 dc3 dc4 nak syn etb can em sub esc fs gs rs us */
4271 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3,
4272/* sp ! " # $ % & ' ( ) * + , - . / */
4273 2, 1, 1, 1, 1, 1, 1, 0, 0, 0, 1, 3, 0, 0, 0, 0,
4274/* 0 1 2 3 4 5 6 7 8 9 : ; < = > ? */
4275 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 0,
4276/* @ A B C D E F G H I J K L M N O */
4277 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
4278/* P Q R S T U V W X Y Z [ \ ] ^ _ */
4279 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 3, 1, 1, 1,
4280/* ` a b c d e f g h i j k l m n o */
4281 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
4282/* p q r s t u v w x y z { | } ~ del */
4283 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 3, 3,
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004284};
4285
Antoine Pitrou244651a2009-05-04 18:56:13 +00004286/* ENCODE_DIRECT: this character should be encoded as itself. The
4287 * answer depends on whether we are encoding set O as itself, and also
4288 * on whether we are encoding whitespace as itself. RFC2152 makes it
4289 * clear that the answers to these questions vary between
4290 * applications, so this code needs to be flexible. */
Marc-André Lemburge115ec82005-10-19 22:33:31 +00004291
Antoine Pitrou244651a2009-05-04 18:56:13 +00004292#define ENCODE_DIRECT(c, directO, directWS) \
4293 ((c) < 128 && (c) > 0 && \
4294 ((utf7_category[(c)] == 0) || \
4295 (directWS && (utf7_category[(c)] == 2)) || \
4296 (directO && (utf7_category[(c)] == 1))))
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004297
Alexander Belopolsky40018472011-02-26 01:02:56 +00004298PyObject *
4299PyUnicode_DecodeUTF7(const char *s,
Ezio Melotti2aa2b3b2011-09-29 00:58:57 +03004300 Py_ssize_t size,
4301 const char *errors)
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004302{
Christian Heimes5d14c2b2007-11-20 23:38:09 +00004303 return PyUnicode_DecodeUTF7Stateful(s, size, errors, NULL);
4304}
4305
Antoine Pitrou244651a2009-05-04 18:56:13 +00004306/* The decoder. The only state we preserve is our read position,
4307 * i.e. how many characters we have consumed. So if we end in the
4308 * middle of a shift sequence we have to back off the read position
4309 * and the output to the beginning of the sequence, otherwise we lose
4310 * all the shift state (seen bits, number of bits seen, high
4311 * surrogate). */
4312
Alexander Belopolsky40018472011-02-26 01:02:56 +00004313PyObject *
4314PyUnicode_DecodeUTF7Stateful(const char *s,
Ezio Melotti2aa2b3b2011-09-29 00:58:57 +03004315 Py_ssize_t size,
4316 const char *errors,
4317 Py_ssize_t *consumed)
Christian Heimes5d14c2b2007-11-20 23:38:09 +00004318{
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004319 const char *starts = s;
Martin v. Löwis18e16552006-02-15 17:27:45 +00004320 Py_ssize_t startinpos;
4321 Py_ssize_t endinpos;
4322 Py_ssize_t outpos;
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004323 const char *e;
Victor Stinner7931d9a2011-11-04 00:22:48 +01004324 PyObject *unicode;
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004325 const char *errmsg = "";
4326 int inShift = 0;
Martin v. Löwise9b11c12011-11-08 17:35:34 +01004327 Py_ssize_t shiftOutStart;
Antoine Pitrou244651a2009-05-04 18:56:13 +00004328 unsigned int base64bits = 0;
4329 unsigned long base64buffer = 0;
Victor Stinner24729f32011-11-10 20:31:37 +01004330 Py_UCS4 surrogate = 0;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004331 PyObject *errorHandler = NULL;
4332 PyObject *exc = NULL;
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004333
Martin v. Löwise9b11c12011-11-08 17:35:34 +01004334 /* Start off assuming it's all ASCII. Widen later as necessary. */
4335 unicode = PyUnicode_New(size, 127);
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004336 if (!unicode)
4337 return NULL;
Christian Heimes5d14c2b2007-11-20 23:38:09 +00004338 if (size == 0) {
4339 if (consumed)
4340 *consumed = 0;
Victor Stinner7931d9a2011-11-04 00:22:48 +01004341 return unicode;
Christian Heimes5d14c2b2007-11-20 23:38:09 +00004342 }
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004343
Martin v. Löwise9b11c12011-11-08 17:35:34 +01004344 shiftOutStart = outpos = 0;
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004345 e = s + size;
4346
4347 while (s < e) {
Martin v. Löwise9b11c12011-11-08 17:35:34 +01004348 Py_UCS4 ch;
Benjamin Peterson29060642009-01-31 22:14:21 +00004349 restart:
Antoine Pitrou5ffd9e92008-07-25 18:05:24 +00004350 ch = (unsigned char) *s;
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004351
Antoine Pitrou244651a2009-05-04 18:56:13 +00004352 if (inShift) { /* in a base-64 section */
4353 if (IS_BASE64(ch)) { /* consume a base-64 character */
4354 base64buffer = (base64buffer << 6) | FROM_BASE64(ch);
4355 base64bits += 6;
4356 s++;
4357 if (base64bits >= 16) {
4358 /* we have enough bits for a UTF-16 value */
Victor Stinner24729f32011-11-10 20:31:37 +01004359 Py_UCS4 outCh = (Py_UCS4)(base64buffer >> (base64bits-16));
Antoine Pitrou244651a2009-05-04 18:56:13 +00004360 base64bits -= 16;
4361 base64buffer &= (1 << base64bits) - 1; /* clear high bits */
4362 if (surrogate) {
4363 /* expecting a second surrogate */
Victor Stinner551ac952011-11-29 22:58:13 +01004364 if (Py_UNICODE_IS_LOW_SURROGATE(outCh)) {
4365 Py_UCS4 ch2 = Py_UNICODE_JOIN_SURROGATES(surrogate, outCh);
Martin v. Löwise9b11c12011-11-08 17:35:34 +01004366 if (unicode_putchar(&unicode, &outpos, ch2) < 0)
4367 goto onError;
Antoine Pitrou244651a2009-05-04 18:56:13 +00004368 surrogate = 0;
Antoine Pitrou5418ee02011-11-15 01:42:21 +01004369 continue;
Antoine Pitrou244651a2009-05-04 18:56:13 +00004370 }
4371 else {
Antoine Pitrou78edf752011-11-15 01:44:16 +01004372 if (unicode_putchar(&unicode, &outpos, surrogate) < 0)
4373 goto onError;
Antoine Pitrou244651a2009-05-04 18:56:13 +00004374 surrogate = 0;
Antoine Pitrou244651a2009-05-04 18:56:13 +00004375 }
4376 }
Victor Stinner551ac952011-11-29 22:58:13 +01004377 if (Py_UNICODE_IS_HIGH_SURROGATE(outCh)) {
Antoine Pitrou244651a2009-05-04 18:56:13 +00004378 /* first surrogate */
4379 surrogate = outCh;
4380 }
Antoine Pitrou244651a2009-05-04 18:56:13 +00004381 else {
Martin v. Löwise9b11c12011-11-08 17:35:34 +01004382 if (unicode_putchar(&unicode, &outpos, outCh) < 0)
4383 goto onError;
Antoine Pitrou244651a2009-05-04 18:56:13 +00004384 }
4385 }
4386 }
4387 else { /* now leaving a base-64 section */
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004388 inShift = 0;
4389 s++;
Antoine Pitrou244651a2009-05-04 18:56:13 +00004390 if (surrogate) {
Antoine Pitrou78edf752011-11-15 01:44:16 +01004391 if (unicode_putchar(&unicode, &outpos, surrogate) < 0)
4392 goto onError;
Antoine Pitrou5418ee02011-11-15 01:42:21 +01004393 surrogate = 0;
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004394 }
Antoine Pitrou244651a2009-05-04 18:56:13 +00004395 if (base64bits > 0) { /* left-over bits */
4396 if (base64bits >= 6) {
4397 /* We've seen at least one base-64 character */
4398 errmsg = "partial character in shift sequence";
4399 goto utf7Error;
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004400 }
Antoine Pitrou244651a2009-05-04 18:56:13 +00004401 else {
4402 /* Some bits remain; they should be zero */
4403 if (base64buffer != 0) {
4404 errmsg = "non-zero padding bits in shift sequence";
4405 goto utf7Error;
4406 }
4407 }
4408 }
4409 if (ch != '-') {
4410 /* '-' is absorbed; other terminating
4411 characters are preserved */
Martin v. Löwise9b11c12011-11-08 17:35:34 +01004412 if (unicode_putchar(&unicode, &outpos, ch) < 0)
4413 goto onError;
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004414 }
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004415 }
4416 }
4417 else if ( ch == '+' ) {
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004418 startinpos = s-starts;
Antoine Pitrou244651a2009-05-04 18:56:13 +00004419 s++; /* consume '+' */
4420 if (s < e && *s == '-') { /* '+-' encodes '+' */
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004421 s++;
Martin v. Löwise9b11c12011-11-08 17:35:34 +01004422 if (unicode_putchar(&unicode, &outpos, '+') < 0)
4423 goto onError;
Antoine Pitrou244651a2009-05-04 18:56:13 +00004424 }
4425 else { /* begin base64-encoded section */
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004426 inShift = 1;
Martin v. Löwise9b11c12011-11-08 17:35:34 +01004427 shiftOutStart = outpos;
Antoine Pitrou244651a2009-05-04 18:56:13 +00004428 base64bits = 0;
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004429 }
4430 }
Antoine Pitrou244651a2009-05-04 18:56:13 +00004431 else if (DECODE_DIRECT(ch)) { /* character decodes as itself */
Martin v. Löwise9b11c12011-11-08 17:35:34 +01004432 if (unicode_putchar(&unicode, &outpos, ch) < 0)
4433 goto onError;
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004434 s++;
4435 }
Antoine Pitrou244651a2009-05-04 18:56:13 +00004436 else {
4437 startinpos = s-starts;
4438 s++;
4439 errmsg = "unexpected special character";
4440 goto utf7Error;
4441 }
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004442 continue;
Antoine Pitrou244651a2009-05-04 18:56:13 +00004443utf7Error:
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004444 endinpos = s-starts;
4445 if (unicode_decode_call_errorhandler(
Benjamin Peterson29060642009-01-31 22:14:21 +00004446 errors, &errorHandler,
4447 "utf7", errmsg,
4448 &starts, &e, &startinpos, &endinpos, &exc, &s,
Martin v. Löwise9b11c12011-11-08 17:35:34 +01004449 &unicode, &outpos))
Benjamin Peterson29060642009-01-31 22:14:21 +00004450 goto onError;
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004451 }
4452
Antoine Pitrou244651a2009-05-04 18:56:13 +00004453 /* end of string */
4454
4455 if (inShift && !consumed) { /* in shift sequence, no more to follow */
4456 /* if we're in an inconsistent state, that's an error */
4457 if (surrogate ||
4458 (base64bits >= 6) ||
4459 (base64bits > 0 && base64buffer != 0)) {
Antoine Pitrou244651a2009-05-04 18:56:13 +00004460 endinpos = size;
4461 if (unicode_decode_call_errorhandler(
4462 errors, &errorHandler,
4463 "utf7", "unterminated shift sequence",
4464 &starts, &e, &startinpos, &endinpos, &exc, &s,
Martin v. Löwise9b11c12011-11-08 17:35:34 +01004465 &unicode, &outpos))
Antoine Pitrou244651a2009-05-04 18:56:13 +00004466 goto onError;
4467 if (s < e)
4468 goto restart;
4469 }
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004470 }
Antoine Pitrou244651a2009-05-04 18:56:13 +00004471
4472 /* return state */
Christian Heimes5d14c2b2007-11-20 23:38:09 +00004473 if (consumed) {
Antoine Pitrou244651a2009-05-04 18:56:13 +00004474 if (inShift) {
Martin v. Löwise9b11c12011-11-08 17:35:34 +01004475 outpos = shiftOutStart; /* back off output */
Christian Heimes5d14c2b2007-11-20 23:38:09 +00004476 *consumed = startinpos;
Antoine Pitrou244651a2009-05-04 18:56:13 +00004477 }
4478 else {
Christian Heimes5d14c2b2007-11-20 23:38:09 +00004479 *consumed = s-starts;
Antoine Pitrou244651a2009-05-04 18:56:13 +00004480 }
Christian Heimes5d14c2b2007-11-20 23:38:09 +00004481 }
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004482
Martin v. Löwise9b11c12011-11-08 17:35:34 +01004483 if (unicode_resize(&unicode, outpos) < 0)
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004484 goto onError;
4485
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004486 Py_XDECREF(errorHandler);
4487 Py_XDECREF(exc);
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01004488 return unicode_result(unicode);
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004489
Benjamin Peterson29060642009-01-31 22:14:21 +00004490 onError:
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004491 Py_XDECREF(errorHandler);
4492 Py_XDECREF(exc);
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004493 Py_DECREF(unicode);
4494 return NULL;
4495}
4496
4497
Alexander Belopolsky40018472011-02-26 01:02:56 +00004498PyObject *
Martin v. Löwis1db7c132011-11-10 18:24:32 +01004499_PyUnicode_EncodeUTF7(PyObject *str,
4500 int base64SetO,
4501 int base64WhiteSpace,
4502 const char *errors)
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004503{
Martin v. Löwis1db7c132011-11-10 18:24:32 +01004504 int kind;
4505 void *data;
4506 Py_ssize_t len;
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00004507 PyObject *v;
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004508 int inShift = 0;
Martin v. Löwis1db7c132011-11-10 18:24:32 +01004509 Py_ssize_t i;
Antoine Pitrou244651a2009-05-04 18:56:13 +00004510 unsigned int base64bits = 0;
4511 unsigned long base64buffer = 0;
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004512 char * out;
4513 char * start;
4514
Benjamin Petersonbac79492012-01-14 13:34:47 -05004515 if (PyUnicode_READY(str) == -1)
Martin v. Löwis1db7c132011-11-10 18:24:32 +01004516 return NULL;
4517 kind = PyUnicode_KIND(str);
4518 data = PyUnicode_DATA(str);
4519 len = PyUnicode_GET_LENGTH(str);
4520
4521 if (len == 0)
Benjamin Peterson29060642009-01-31 22:14:21 +00004522 return PyBytes_FromStringAndSize(NULL, 0);
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004523
Martin v. Löwis1db7c132011-11-10 18:24:32 +01004524 /* It might be possible to tighten this worst case */
Mark Dickinsonc04ddff2012-10-06 18:04:49 +01004525 if (len > PY_SSIZE_T_MAX / 8)
Neal Norwitz3ce5d922008-08-24 07:08:55 +00004526 return PyErr_NoMemory();
Mark Dickinsonc04ddff2012-10-06 18:04:49 +01004527 v = PyBytes_FromStringAndSize(NULL, len * 8);
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004528 if (v == NULL)
4529 return NULL;
4530
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00004531 start = out = PyBytes_AS_STRING(v);
Martin v. Löwis1db7c132011-11-10 18:24:32 +01004532 for (i = 0; i < len; ++i) {
Victor Stinner0e368262011-11-10 20:12:49 +01004533 Py_UCS4 ch = PyUnicode_READ(kind, data, i);
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004534
Antoine Pitrou244651a2009-05-04 18:56:13 +00004535 if (inShift) {
4536 if (ENCODE_DIRECT(ch, !base64SetO, !base64WhiteSpace)) {
4537 /* shifting out */
4538 if (base64bits) { /* output remaining bits */
4539 *out++ = TO_BASE64(base64buffer << (6-base64bits));
4540 base64buffer = 0;
4541 base64bits = 0;
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004542 }
4543 inShift = 0;
Antoine Pitrou244651a2009-05-04 18:56:13 +00004544 /* Characters not in the BASE64 set implicitly unshift the sequence
4545 so no '-' is required, except if the character is itself a '-' */
4546 if (IS_BASE64(ch) || ch == '-') {
4547 *out++ = '-';
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004548 }
Antoine Pitrou244651a2009-05-04 18:56:13 +00004549 *out++ = (char) ch;
4550 }
4551 else {
4552 goto encode_char;
Tim Petersced69f82003-09-16 20:30:58 +00004553 }
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004554 }
Antoine Pitrou244651a2009-05-04 18:56:13 +00004555 else { /* not in a shift sequence */
4556 if (ch == '+') {
4557 *out++ = '+';
4558 *out++ = '-';
4559 }
4560 else if (ENCODE_DIRECT(ch, !base64SetO, !base64WhiteSpace)) {
4561 *out++ = (char) ch;
4562 }
4563 else {
4564 *out++ = '+';
4565 inShift = 1;
4566 goto encode_char;
4567 }
4568 }
4569 continue;
4570encode_char:
Antoine Pitrou244651a2009-05-04 18:56:13 +00004571 if (ch >= 0x10000) {
Victor Stinner8faf8212011-12-08 22:14:11 +01004572 assert(ch <= MAX_UNICODE);
Victor Stinner0d3721d2011-11-22 03:27:53 +01004573
Antoine Pitrou244651a2009-05-04 18:56:13 +00004574 /* code first surrogate */
4575 base64bits += 16;
4576 base64buffer = (base64buffer << 16) | 0xd800 | ((ch-0x10000) >> 10);
4577 while (base64bits >= 6) {
4578 *out++ = TO_BASE64(base64buffer >> (base64bits-6));
4579 base64bits -= 6;
4580 }
4581 /* prepare second surrogate */
Victor Stinner551ac952011-11-29 22:58:13 +01004582 ch = Py_UNICODE_LOW_SURROGATE(ch);
Antoine Pitrou244651a2009-05-04 18:56:13 +00004583 }
Antoine Pitrou244651a2009-05-04 18:56:13 +00004584 base64bits += 16;
4585 base64buffer = (base64buffer << 16) | ch;
4586 while (base64bits >= 6) {
4587 *out++ = TO_BASE64(base64buffer >> (base64bits-6));
4588 base64bits -= 6;
4589 }
Hye-Shik Chang1bc09b72004-01-03 19:35:43 +00004590 }
Antoine Pitrou244651a2009-05-04 18:56:13 +00004591 if (base64bits)
4592 *out++= TO_BASE64(base64buffer << (6-base64bits) );
4593 if (inShift)
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004594 *out++ = '-';
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00004595 if (_PyBytes_Resize(&v, out - start) < 0)
4596 return NULL;
4597 return v;
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004598}
Martin v. Löwis1db7c132011-11-10 18:24:32 +01004599PyObject *
4600PyUnicode_EncodeUTF7(const Py_UNICODE *s,
4601 Py_ssize_t size,
4602 int base64SetO,
4603 int base64WhiteSpace,
4604 const char *errors)
4605{
4606 PyObject *result;
4607 PyObject *tmp = PyUnicode_FromUnicode(s, size);
4608 if (tmp == NULL)
4609 return NULL;
Victor Stinner0e368262011-11-10 20:12:49 +01004610 result = _PyUnicode_EncodeUTF7(tmp, base64SetO,
Martin v. Löwis1db7c132011-11-10 18:24:32 +01004611 base64WhiteSpace, errors);
4612 Py_DECREF(tmp);
4613 return result;
4614}
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004615
Antoine Pitrou244651a2009-05-04 18:56:13 +00004616#undef IS_BASE64
4617#undef FROM_BASE64
4618#undef TO_BASE64
4619#undef DECODE_DIRECT
4620#undef ENCODE_DIRECT
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004621
Guido van Rossumd57fd912000-03-10 22:53:23 +00004622/* --- UTF-8 Codec -------------------------------------------------------- */
4623
Alexander Belopolsky40018472011-02-26 01:02:56 +00004624PyObject *
4625PyUnicode_DecodeUTF8(const char *s,
Ezio Melotti2aa2b3b2011-09-29 00:58:57 +03004626 Py_ssize_t size,
4627 const char *errors)
Guido van Rossumd57fd912000-03-10 22:53:23 +00004628{
Walter Dörwald69652032004-09-07 20:24:22 +00004629 return PyUnicode_DecodeUTF8Stateful(s, size, errors, NULL);
4630}
4631
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004632#include "stringlib/asciilib.h"
4633#include "stringlib/codecs.h"
4634#include "stringlib/undef.h"
4635
Antoine Pitrou0a3229d2011-11-21 20:39:13 +01004636#include "stringlib/ucs1lib.h"
4637#include "stringlib/codecs.h"
4638#include "stringlib/undef.h"
4639
4640#include "stringlib/ucs2lib.h"
4641#include "stringlib/codecs.h"
4642#include "stringlib/undef.h"
4643
4644#include "stringlib/ucs4lib.h"
4645#include "stringlib/codecs.h"
4646#include "stringlib/undef.h"
4647
Antoine Pitrouab868312009-01-10 15:40:25 +00004648/* Mask to quickly check whether a C 'long' contains a
4649 non-ASCII, UTF8-encoded char. */
4650#if (SIZEOF_LONG == 8)
Mark Dickinson01ac8b62012-07-07 14:08:48 +02004651# define ASCII_CHAR_MASK 0x8080808080808080UL
Antoine Pitrouab868312009-01-10 15:40:25 +00004652#elif (SIZEOF_LONG == 4)
Mark Dickinson01ac8b62012-07-07 14:08:48 +02004653# define ASCII_CHAR_MASK 0x80808080UL
Antoine Pitrouab868312009-01-10 15:40:25 +00004654#else
4655# error C 'long' size should be either 4 or 8!
4656#endif
4657
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004658static Py_ssize_t
4659ascii_decode(const char *start, const char *end, Py_UCS1 *dest)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02004660{
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004661 const char *p = start;
Antoine Pitrouca8aa4a2012-09-20 20:56:47 +02004662 const char *aligned_end = (const char *) _Py_ALIGN_DOWN(end, SIZEOF_LONG);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02004663
Antoine Pitrou8b0e9842013-05-11 15:58:34 +02004664 /*
4665 * Issue #17237: m68k is a bit different from most architectures in
4666 * that objects do not use "natural alignment" - for example, int and
4667 * long are only aligned at 2-byte boundaries. Therefore the assert()
4668 * won't work; also, tests have shown that skipping the "optimised
4669 * version" will even speed up m68k.
4670 */
4671#if !defined(__m68k__)
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004672#if SIZEOF_LONG <= SIZEOF_VOID_P
Antoine Pitrouca8aa4a2012-09-20 20:56:47 +02004673 assert(_Py_IS_ALIGNED(dest, SIZEOF_LONG));
4674 if (_Py_IS_ALIGNED(p, SIZEOF_LONG)) {
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004675 /* Fast path, see in STRINGLIB(utf8_decode) for
4676 an explanation. */
4677 /* Help register allocation */
4678 register const char *_p = p;
4679 register Py_UCS1 * q = dest;
4680 while (_p < aligned_end) {
4681 unsigned long value = *(const unsigned long *) _p;
4682 if (value & ASCII_CHAR_MASK)
Benjamin Peterson29060642009-01-31 22:14:21 +00004683 break;
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004684 *((unsigned long *)q) = value;
4685 _p += SIZEOF_LONG;
4686 q += SIZEOF_LONG;
Benjamin Peterson14339b62009-01-31 16:36:08 +00004687 }
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004688 p = _p;
4689 while (p < end) {
4690 if ((unsigned char)*p & 0x80)
4691 break;
4692 *q++ = *p++;
Guido van Rossumd57fd912000-03-10 22:53:23 +00004693 }
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004694 return p - start;
Guido van Rossumd57fd912000-03-10 22:53:23 +00004695 }
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004696#endif
Antoine Pitrou8b0e9842013-05-11 15:58:34 +02004697#endif
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004698 while (p < end) {
4699 /* Fast path, see in STRINGLIB(utf8_decode) in stringlib/codecs.h
4700 for an explanation. */
Antoine Pitrouca8aa4a2012-09-20 20:56:47 +02004701 if (_Py_IS_ALIGNED(p, SIZEOF_LONG)) {
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004702 /* Help register allocation */
4703 register const char *_p = p;
4704 while (_p < aligned_end) {
4705 unsigned long value = *(unsigned long *) _p;
4706 if (value & ASCII_CHAR_MASK)
4707 break;
4708 _p += SIZEOF_LONG;
4709 }
4710 p = _p;
4711 if (_p == end)
4712 break;
4713 }
4714 if ((unsigned char)*p & 0x80)
4715 break;
4716 ++p;
4717 }
4718 memcpy(dest, start, p - start);
4719 return p - start;
Guido van Rossumd57fd912000-03-10 22:53:23 +00004720}
Antoine Pitrouab868312009-01-10 15:40:25 +00004721
Victor Stinner785938e2011-12-11 20:09:03 +01004722PyObject *
4723PyUnicode_DecodeUTF8Stateful(const char *s,
4724 Py_ssize_t size,
4725 const char *errors,
4726 Py_ssize_t *consumed)
4727{
Victor Stinner785938e2011-12-11 20:09:03 +01004728 PyObject *unicode;
Victor Stinner785938e2011-12-11 20:09:03 +01004729 const char *starts = s;
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004730 const char *end = s + size;
4731 Py_ssize_t outpos;
4732
4733 Py_ssize_t startinpos;
4734 Py_ssize_t endinpos;
4735 const char *errmsg = "";
4736 PyObject *errorHandler = NULL;
4737 PyObject *exc = NULL;
Victor Stinner785938e2011-12-11 20:09:03 +01004738
4739 if (size == 0) {
4740 if (consumed)
4741 *consumed = 0;
Serhiy Storchaka678db842013-01-26 12:16:36 +02004742 _Py_RETURN_UNICODE_EMPTY();
Victor Stinner785938e2011-12-11 20:09:03 +01004743 }
4744
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004745 /* ASCII is equivalent to the first 128 ordinals in Unicode. */
4746 if (size == 1 && (unsigned char)s[0] < 128) {
Victor Stinner785938e2011-12-11 20:09:03 +01004747 if (consumed)
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004748 *consumed = 1;
4749 return get_latin1_char((unsigned char)s[0]);
Victor Stinner785938e2011-12-11 20:09:03 +01004750 }
4751
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004752 unicode = PyUnicode_New(size, 127);
Victor Stinner785938e2011-12-11 20:09:03 +01004753 if (!unicode)
4754 return NULL;
Victor Stinner785938e2011-12-11 20:09:03 +01004755
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004756 outpos = ascii_decode(s, end, PyUnicode_1BYTE_DATA(unicode));
4757 s += outpos;
4758 while (s < end) {
4759 Py_UCS4 ch;
4760 int kind = PyUnicode_KIND(unicode);
4761 if (kind == PyUnicode_1BYTE_KIND) {
4762 if (PyUnicode_IS_ASCII(unicode))
4763 ch = asciilib_utf8_decode(&s, end,
4764 PyUnicode_1BYTE_DATA(unicode), &outpos);
4765 else
4766 ch = ucs1lib_utf8_decode(&s, end,
4767 PyUnicode_1BYTE_DATA(unicode), &outpos);
4768 } else if (kind == PyUnicode_2BYTE_KIND) {
4769 ch = ucs2lib_utf8_decode(&s, end,
4770 PyUnicode_2BYTE_DATA(unicode), &outpos);
4771 } else {
4772 assert(kind == PyUnicode_4BYTE_KIND);
4773 ch = ucs4lib_utf8_decode(&s, end,
4774 PyUnicode_4BYTE_DATA(unicode), &outpos);
4775 }
4776
4777 switch (ch) {
4778 case 0:
4779 if (s == end || consumed)
4780 goto End;
4781 errmsg = "unexpected end of data";
4782 startinpos = s - starts;
Ezio Melottif7ed5d12012-11-04 23:21:38 +02004783 endinpos = end - starts;
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004784 break;
4785 case 1:
4786 errmsg = "invalid start byte";
4787 startinpos = s - starts;
4788 endinpos = startinpos + 1;
4789 break;
4790 case 2:
Ezio Melottif7ed5d12012-11-04 23:21:38 +02004791 case 3:
4792 case 4:
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004793 errmsg = "invalid continuation byte";
4794 startinpos = s - starts;
Ezio Melottif7ed5d12012-11-04 23:21:38 +02004795 endinpos = startinpos + ch - 1;
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004796 break;
4797 default:
4798 if (unicode_putchar(&unicode, &outpos, ch) < 0)
4799 goto onError;
4800 continue;
4801 }
4802
4803 if (unicode_decode_call_errorhandler(
4804 errors, &errorHandler,
4805 "utf-8", errmsg,
4806 &starts, &end, &startinpos, &endinpos, &exc, &s,
4807 &unicode, &outpos))
4808 goto onError;
Victor Stinner785938e2011-12-11 20:09:03 +01004809 }
4810
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004811End:
4812 if (unicode_resize(&unicode, outpos) < 0)
4813 goto onError;
4814
4815 if (consumed)
4816 *consumed = s - starts;
4817
4818 Py_XDECREF(errorHandler);
4819 Py_XDECREF(exc);
4820 assert(_PyUnicode_CheckConsistency(unicode, 1));
4821 return unicode;
4822
4823onError:
4824 Py_XDECREF(errorHandler);
4825 Py_XDECREF(exc);
4826 Py_XDECREF(unicode);
4827 return NULL;
Victor Stinner785938e2011-12-11 20:09:03 +01004828}
4829
Victor Stinnerf933e1a2010-10-20 22:58:25 +00004830#ifdef __APPLE__
4831
4832/* Simplified UTF-8 decoder using surrogateescape error handler,
Victor Stinner27b1ca22012-12-03 12:47:59 +01004833 used to decode the command line arguments on Mac OS X.
4834
4835 Return a pointer to a newly allocated wide character string (use
4836 PyMem_Free() to free the memory), or NULL on memory allocation error. */
Victor Stinnerf933e1a2010-10-20 22:58:25 +00004837
4838wchar_t*
4839_Py_DecodeUTF8_surrogateescape(const char *s, Py_ssize_t size)
4840{
Victor Stinnerf933e1a2010-10-20 22:58:25 +00004841 const char *e;
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004842 wchar_t *unicode;
4843 Py_ssize_t outpos;
Victor Stinnerf933e1a2010-10-20 22:58:25 +00004844
4845 /* Note: size will always be longer than the resulting Unicode
4846 character count */
Victor Stinner27b1ca22012-12-03 12:47:59 +01004847 if (PY_SSIZE_T_MAX / sizeof(wchar_t) < (size + 1))
Victor Stinnerf933e1a2010-10-20 22:58:25 +00004848 return NULL;
Victor Stinnerf933e1a2010-10-20 22:58:25 +00004849 unicode = PyMem_Malloc((size + 1) * sizeof(wchar_t));
4850 if (!unicode)
4851 return NULL;
4852
4853 /* Unpack UTF-8 encoded data */
Victor Stinnerf933e1a2010-10-20 22:58:25 +00004854 e = s + size;
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004855 outpos = 0;
Victor Stinnerf933e1a2010-10-20 22:58:25 +00004856 while (s < e) {
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004857 Py_UCS4 ch;
Victor Stinnerf933e1a2010-10-20 22:58:25 +00004858#if SIZEOF_WCHAR_T == 4
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004859 ch = ucs4lib_utf8_decode(&s, e, (Py_UCS4 *)unicode, &outpos);
Victor Stinnerf933e1a2010-10-20 22:58:25 +00004860#else
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004861 ch = ucs2lib_utf8_decode(&s, e, (Py_UCS2 *)unicode, &outpos);
Victor Stinnerf933e1a2010-10-20 22:58:25 +00004862#endif
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004863 if (ch > 0xFF) {
4864#if SIZEOF_WCHAR_T == 4
4865 assert(0);
4866#else
4867 assert(Py_UNICODE_IS_SURROGATE(ch));
4868 /* compute and append the two surrogates: */
4869 unicode[outpos++] = (wchar_t)Py_UNICODE_HIGH_SURROGATE(ch);
4870 unicode[outpos++] = (wchar_t)Py_UNICODE_LOW_SURROGATE(ch);
4871#endif
Victor Stinnerf933e1a2010-10-20 22:58:25 +00004872 }
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004873 else {
4874 if (!ch && s == e)
4875 break;
4876 /* surrogateescape */
4877 unicode[outpos++] = 0xDC00 + (unsigned char)*s++;
4878 }
Victor Stinnerf933e1a2010-10-20 22:58:25 +00004879 }
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004880 unicode[outpos] = L'\0';
Victor Stinnerf933e1a2010-10-20 22:58:25 +00004881 return unicode;
4882}
4883
4884#endif /* __APPLE__ */
Antoine Pitrouab868312009-01-10 15:40:25 +00004885
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02004886/* Primary internal function which creates utf8 encoded bytes objects.
4887
4888 Allocation strategy: if the string is short, convert into a stack buffer
Tim Peters602f7402002-04-27 18:03:26 +00004889 and allocate exactly as much space needed at the end. Else allocate the
4890 maximum possible needed (4 result bytes per Unicode character), and return
4891 the excess memory at the end.
Martin v. Löwis2a7ff352002-04-21 09:59:45 +00004892*/
Tim Peters7e3d9612002-04-21 03:26:37 +00004893PyObject *
Victor Stinner7931d9a2011-11-04 00:22:48 +01004894_PyUnicode_AsUTF8String(PyObject *unicode, const char *errors)
Guido van Rossumd57fd912000-03-10 22:53:23 +00004895{
Victor Stinner6099a032011-12-18 14:22:26 +01004896 enum PyUnicode_Kind kind;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02004897 void *data;
4898 Py_ssize_t size;
Marc-André Lemburgbd3be8f2002-02-07 11:33:49 +00004899
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02004900 if (!PyUnicode_Check(unicode)) {
4901 PyErr_BadArgument();
4902 return NULL;
4903 }
4904
4905 if (PyUnicode_READY(unicode) == -1)
4906 return NULL;
4907
Victor Stinnere90fe6a2011-10-01 16:48:13 +02004908 if (PyUnicode_UTF8(unicode))
4909 return PyBytes_FromStringAndSize(PyUnicode_UTF8(unicode),
4910 PyUnicode_UTF8_LENGTH(unicode));
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02004911
4912 kind = PyUnicode_KIND(unicode);
4913 data = PyUnicode_DATA(unicode);
4914 size = PyUnicode_GET_LENGTH(unicode);
4915
Benjamin Petersonead6b532011-12-20 17:23:42 -06004916 switch (kind) {
Victor Stinner6099a032011-12-18 14:22:26 +01004917 default:
4918 assert(0);
4919 case PyUnicode_1BYTE_KIND:
4920 /* the string cannot be ASCII, or PyUnicode_UTF8() would be set */
4921 assert(!PyUnicode_IS_ASCII(unicode));
4922 return ucs1lib_utf8_encoder(unicode, data, size, errors);
4923 case PyUnicode_2BYTE_KIND:
4924 return ucs2lib_utf8_encoder(unicode, data, size, errors);
4925 case PyUnicode_4BYTE_KIND:
4926 return ucs4lib_utf8_encoder(unicode, data, size, errors);
Tim Peters602f7402002-04-27 18:03:26 +00004927 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00004928}
4929
Alexander Belopolsky40018472011-02-26 01:02:56 +00004930PyObject *
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02004931PyUnicode_EncodeUTF8(const Py_UNICODE *s,
4932 Py_ssize_t size,
4933 const char *errors)
4934{
4935 PyObject *v, *unicode;
4936
4937 unicode = PyUnicode_FromUnicode(s, size);
4938 if (unicode == NULL)
4939 return NULL;
4940 v = _PyUnicode_AsUTF8String(unicode, errors);
4941 Py_DECREF(unicode);
4942 return v;
4943}
4944
4945PyObject *
Alexander Belopolsky40018472011-02-26 01:02:56 +00004946PyUnicode_AsUTF8String(PyObject *unicode)
Guido van Rossumd57fd912000-03-10 22:53:23 +00004947{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02004948 return _PyUnicode_AsUTF8String(unicode, NULL);
Guido van Rossumd57fd912000-03-10 22:53:23 +00004949}
4950
Walter Dörwald41980ca2007-08-16 21:55:45 +00004951/* --- UTF-32 Codec ------------------------------------------------------- */
4952
4953PyObject *
4954PyUnicode_DecodeUTF32(const char *s,
Benjamin Peterson29060642009-01-31 22:14:21 +00004955 Py_ssize_t size,
4956 const char *errors,
4957 int *byteorder)
Walter Dörwald41980ca2007-08-16 21:55:45 +00004958{
4959 return PyUnicode_DecodeUTF32Stateful(s, size, errors, byteorder, NULL);
4960}
4961
4962PyObject *
4963PyUnicode_DecodeUTF32Stateful(const char *s,
Benjamin Peterson29060642009-01-31 22:14:21 +00004964 Py_ssize_t size,
4965 const char *errors,
4966 int *byteorder,
4967 Py_ssize_t *consumed)
Walter Dörwald41980ca2007-08-16 21:55:45 +00004968{
4969 const char *starts = s;
4970 Py_ssize_t startinpos;
4971 Py_ssize_t endinpos;
4972 Py_ssize_t outpos;
Victor Stinner7931d9a2011-11-04 00:22:48 +01004973 PyObject *unicode;
Mark Dickinson7db923c2010-06-12 09:10:14 +00004974 const unsigned char *q, *e;
Walter Dörwald41980ca2007-08-16 21:55:45 +00004975 int bo = 0; /* assume native ordering by default */
4976 const char *errmsg = "";
Walter Dörwald41980ca2007-08-16 21:55:45 +00004977 /* Offsets from q for retrieving bytes in the right order. */
4978#ifdef BYTEORDER_IS_LITTLE_ENDIAN
4979 int iorder[] = {0, 1, 2, 3};
4980#else
4981 int iorder[] = {3, 2, 1, 0};
4982#endif
4983 PyObject *errorHandler = NULL;
4984 PyObject *exc = NULL;
Victor Stinner313a1202010-06-11 23:56:51 +00004985
Walter Dörwald41980ca2007-08-16 21:55:45 +00004986 q = (unsigned char *)s;
4987 e = q + size;
4988
4989 if (byteorder)
4990 bo = *byteorder;
4991
4992 /* Check for BOM marks (U+FEFF) in the input and adjust current
4993 byte order setting accordingly. In native mode, the leading BOM
4994 mark is skipped, in all other modes, it is copied to the output
4995 stream as-is (giving a ZWNBSP character). */
4996 if (bo == 0) {
4997 if (size >= 4) {
4998 const Py_UCS4 bom = (q[iorder[3]] << 24) | (q[iorder[2]] << 16) |
Benjamin Peterson29060642009-01-31 22:14:21 +00004999 (q[iorder[1]] << 8) | q[iorder[0]];
Walter Dörwald41980ca2007-08-16 21:55:45 +00005000#ifdef BYTEORDER_IS_LITTLE_ENDIAN
Benjamin Peterson29060642009-01-31 22:14:21 +00005001 if (bom == 0x0000FEFF) {
5002 q += 4;
5003 bo = -1;
5004 }
5005 else if (bom == 0xFFFE0000) {
5006 q += 4;
5007 bo = 1;
5008 }
Walter Dörwald41980ca2007-08-16 21:55:45 +00005009#else
Benjamin Peterson29060642009-01-31 22:14:21 +00005010 if (bom == 0x0000FEFF) {
5011 q += 4;
5012 bo = 1;
5013 }
5014 else if (bom == 0xFFFE0000) {
5015 q += 4;
5016 bo = -1;
5017 }
Walter Dörwald41980ca2007-08-16 21:55:45 +00005018#endif
Benjamin Peterson29060642009-01-31 22:14:21 +00005019 }
Walter Dörwald41980ca2007-08-16 21:55:45 +00005020 }
5021
5022 if (bo == -1) {
5023 /* force LE */
5024 iorder[0] = 0;
5025 iorder[1] = 1;
5026 iorder[2] = 2;
5027 iorder[3] = 3;
5028 }
5029 else if (bo == 1) {
5030 /* force BE */
5031 iorder[0] = 3;
5032 iorder[1] = 2;
5033 iorder[2] = 1;
5034 iorder[3] = 0;
5035 }
5036
Antoine Pitroucc0cfd32010-06-11 21:46:32 +00005037 /* This might be one to much, because of a BOM */
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005038 unicode = PyUnicode_New((size+3)/4, 127);
Antoine Pitroucc0cfd32010-06-11 21:46:32 +00005039 if (!unicode)
5040 return NULL;
5041 if (size == 0)
Victor Stinner7931d9a2011-11-04 00:22:48 +01005042 return unicode;
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005043 outpos = 0;
Antoine Pitroucc0cfd32010-06-11 21:46:32 +00005044
Walter Dörwald41980ca2007-08-16 21:55:45 +00005045 while (q < e) {
Benjamin Peterson29060642009-01-31 22:14:21 +00005046 Py_UCS4 ch;
5047 /* remaining bytes at the end? (size should be divisible by 4) */
5048 if (e-q<4) {
5049 if (consumed)
5050 break;
5051 errmsg = "truncated data";
5052 startinpos = ((const char *)q)-starts;
5053 endinpos = ((const char *)e)-starts;
5054 goto utf32Error;
5055 /* The remaining input chars are ignored if the callback
5056 chooses to skip the input */
5057 }
5058 ch = (q[iorder[3]] << 24) | (q[iorder[2]] << 16) |
5059 (q[iorder[1]] << 8) | q[iorder[0]];
Walter Dörwald41980ca2007-08-16 21:55:45 +00005060
Benjamin Peterson29060642009-01-31 22:14:21 +00005061 if (ch >= 0x110000)
5062 {
5063 errmsg = "codepoint not in range(0x110000)";
5064 startinpos = ((const char *)q)-starts;
5065 endinpos = startinpos+4;
5066 goto utf32Error;
5067 }
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005068 if (unicode_putchar(&unicode, &outpos, ch) < 0)
5069 goto onError;
Benjamin Peterson29060642009-01-31 22:14:21 +00005070 q += 4;
5071 continue;
5072 utf32Error:
Benjamin Peterson29060642009-01-31 22:14:21 +00005073 if (unicode_decode_call_errorhandler(
5074 errors, &errorHandler,
5075 "utf32", errmsg,
5076 &starts, (const char **)&e, &startinpos, &endinpos, &exc, (const char **)&q,
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005077 &unicode, &outpos))
Benjamin Peterson29060642009-01-31 22:14:21 +00005078 goto onError;
Walter Dörwald41980ca2007-08-16 21:55:45 +00005079 }
5080
5081 if (byteorder)
5082 *byteorder = bo;
5083
5084 if (consumed)
Benjamin Peterson29060642009-01-31 22:14:21 +00005085 *consumed = (const char *)q-starts;
Walter Dörwald41980ca2007-08-16 21:55:45 +00005086
5087 /* Adjust length */
Victor Stinner16e6a802011-12-12 13:24:15 +01005088 if (unicode_resize(&unicode, outpos) < 0)
Walter Dörwald41980ca2007-08-16 21:55:45 +00005089 goto onError;
5090
5091 Py_XDECREF(errorHandler);
5092 Py_XDECREF(exc);
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01005093 return unicode_result(unicode);
Walter Dörwald41980ca2007-08-16 21:55:45 +00005094
Benjamin Peterson29060642009-01-31 22:14:21 +00005095 onError:
Walter Dörwald41980ca2007-08-16 21:55:45 +00005096 Py_DECREF(unicode);
5097 Py_XDECREF(errorHandler);
5098 Py_XDECREF(exc);
5099 return NULL;
5100}
5101
5102PyObject *
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005103_PyUnicode_EncodeUTF32(PyObject *str,
5104 const char *errors,
5105 int byteorder)
Walter Dörwald41980ca2007-08-16 21:55:45 +00005106{
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005107 int kind;
5108 void *data;
5109 Py_ssize_t len;
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00005110 PyObject *v;
Walter Dörwald41980ca2007-08-16 21:55:45 +00005111 unsigned char *p;
Mark Dickinsonc04ddff2012-10-06 18:04:49 +01005112 Py_ssize_t nsize, i;
Walter Dörwald41980ca2007-08-16 21:55:45 +00005113 /* Offsets from p for storing byte pairs in the right order. */
5114#ifdef BYTEORDER_IS_LITTLE_ENDIAN
5115 int iorder[] = {0, 1, 2, 3};
5116#else
5117 int iorder[] = {3, 2, 1, 0};
5118#endif
5119
Benjamin Peterson29060642009-01-31 22:14:21 +00005120#define STORECHAR(CH) \
5121 do { \
5122 p[iorder[3]] = ((CH) >> 24) & 0xff; \
5123 p[iorder[2]] = ((CH) >> 16) & 0xff; \
5124 p[iorder[1]] = ((CH) >> 8) & 0xff; \
5125 p[iorder[0]] = (CH) & 0xff; \
5126 p += 4; \
Walter Dörwald41980ca2007-08-16 21:55:45 +00005127 } while(0)
5128
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005129 if (!PyUnicode_Check(str)) {
5130 PyErr_BadArgument();
5131 return NULL;
5132 }
Benjamin Petersonbac79492012-01-14 13:34:47 -05005133 if (PyUnicode_READY(str) == -1)
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005134 return NULL;
5135 kind = PyUnicode_KIND(str);
5136 data = PyUnicode_DATA(str);
5137 len = PyUnicode_GET_LENGTH(str);
5138
5139 nsize = len + (byteorder == 0);
Mark Dickinsonc04ddff2012-10-06 18:04:49 +01005140 if (nsize > PY_SSIZE_T_MAX / 4)
Benjamin Peterson29060642009-01-31 22:14:21 +00005141 return PyErr_NoMemory();
Mark Dickinsonc04ddff2012-10-06 18:04:49 +01005142 v = PyBytes_FromStringAndSize(NULL, nsize * 4);
Walter Dörwald41980ca2007-08-16 21:55:45 +00005143 if (v == NULL)
5144 return NULL;
5145
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00005146 p = (unsigned char *)PyBytes_AS_STRING(v);
Walter Dörwald41980ca2007-08-16 21:55:45 +00005147 if (byteorder == 0)
Benjamin Peterson29060642009-01-31 22:14:21 +00005148 STORECHAR(0xFEFF);
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005149 if (len == 0)
Guido van Rossum98297ee2007-11-06 21:34:58 +00005150 goto done;
Walter Dörwald41980ca2007-08-16 21:55:45 +00005151
5152 if (byteorder == -1) {
5153 /* force LE */
5154 iorder[0] = 0;
5155 iorder[1] = 1;
5156 iorder[2] = 2;
5157 iorder[3] = 3;
5158 }
5159 else if (byteorder == 1) {
5160 /* force BE */
5161 iorder[0] = 3;
5162 iorder[1] = 2;
5163 iorder[2] = 1;
5164 iorder[3] = 0;
5165 }
5166
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005167 for (i = 0; i < len; i++)
5168 STORECHAR(PyUnicode_READ(kind, data, i));
Guido van Rossum98297ee2007-11-06 21:34:58 +00005169
5170 done:
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00005171 return v;
Walter Dörwald41980ca2007-08-16 21:55:45 +00005172#undef STORECHAR
5173}
5174
Alexander Belopolsky40018472011-02-26 01:02:56 +00005175PyObject *
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005176PyUnicode_EncodeUTF32(const Py_UNICODE *s,
5177 Py_ssize_t size,
5178 const char *errors,
5179 int byteorder)
5180{
5181 PyObject *result;
5182 PyObject *tmp = PyUnicode_FromUnicode(s, size);
5183 if (tmp == NULL)
5184 return NULL;
5185 result = _PyUnicode_EncodeUTF32(tmp, errors, byteorder);
5186 Py_DECREF(tmp);
5187 return result;
5188}
5189
5190PyObject *
Alexander Belopolsky40018472011-02-26 01:02:56 +00005191PyUnicode_AsUTF32String(PyObject *unicode)
Walter Dörwald41980ca2007-08-16 21:55:45 +00005192{
Victor Stinnerb960b342011-11-20 19:12:52 +01005193 return _PyUnicode_EncodeUTF32(unicode, NULL, 0);
Walter Dörwald41980ca2007-08-16 21:55:45 +00005194}
5195
Guido van Rossumd57fd912000-03-10 22:53:23 +00005196/* --- UTF-16 Codec ------------------------------------------------------- */
5197
Tim Peters772747b2001-08-09 22:21:55 +00005198PyObject *
5199PyUnicode_DecodeUTF16(const char *s,
Benjamin Peterson29060642009-01-31 22:14:21 +00005200 Py_ssize_t size,
5201 const char *errors,
5202 int *byteorder)
Guido van Rossumd57fd912000-03-10 22:53:23 +00005203{
Walter Dörwald69652032004-09-07 20:24:22 +00005204 return PyUnicode_DecodeUTF16Stateful(s, size, errors, byteorder, NULL);
5205}
5206
5207PyObject *
5208PyUnicode_DecodeUTF16Stateful(const char *s,
Benjamin Peterson29060642009-01-31 22:14:21 +00005209 Py_ssize_t size,
5210 const char *errors,
5211 int *byteorder,
5212 Py_ssize_t *consumed)
Walter Dörwald69652032004-09-07 20:24:22 +00005213{
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005214 const char *starts = s;
Martin v. Löwis18e16552006-02-15 17:27:45 +00005215 Py_ssize_t startinpos;
5216 Py_ssize_t endinpos;
5217 Py_ssize_t outpos;
Victor Stinner7931d9a2011-11-04 00:22:48 +01005218 PyObject *unicode;
Antoine Pitrou63065d72012-05-15 23:48:04 +02005219 const unsigned char *q, *e;
Tim Peters772747b2001-08-09 22:21:55 +00005220 int bo = 0; /* assume native ordering by default */
Antoine Pitrou63065d72012-05-15 23:48:04 +02005221 int native_ordering;
Marc-André Lemburg9542f482000-07-17 18:23:13 +00005222 const char *errmsg = "";
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005223 PyObject *errorHandler = NULL;
5224 PyObject *exc = NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005225
Tim Peters772747b2001-08-09 22:21:55 +00005226 q = (unsigned char *)s;
Antoine Pitrou63065d72012-05-15 23:48:04 +02005227 e = q + size;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005228
5229 if (byteorder)
Tim Peters772747b2001-08-09 22:21:55 +00005230 bo = *byteorder;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005231
Marc-André Lemburg489b56e2001-05-21 20:30:15 +00005232 /* Check for BOM marks (U+FEFF) in the input and adjust current
5233 byte order setting accordingly. In native mode, the leading BOM
5234 mark is skipped, in all other modes, it is copied to the output
5235 stream as-is (giving a ZWNBSP character). */
Antoine Pitrou63065d72012-05-15 23:48:04 +02005236 if (bo == 0 && size >= 2) {
5237 const Py_UCS4 bom = (q[1] << 8) | q[0];
5238 if (bom == 0xFEFF) {
5239 q += 2;
5240 bo = -1;
Benjamin Peterson29060642009-01-31 22:14:21 +00005241 }
Antoine Pitrou63065d72012-05-15 23:48:04 +02005242 else if (bom == 0xFFFE) {
5243 q += 2;
5244 bo = 1;
5245 }
5246 if (byteorder)
5247 *byteorder = bo;
Marc-André Lemburg489b56e2001-05-21 20:30:15 +00005248 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00005249
Antoine Pitrou63065d72012-05-15 23:48:04 +02005250 if (q == e) {
5251 if (consumed)
5252 *consumed = size;
Serhiy Storchaka678db842013-01-26 12:16:36 +02005253 _Py_RETURN_UNICODE_EMPTY();
Tim Peters772747b2001-08-09 22:21:55 +00005254 }
Antoine Pitrou63065d72012-05-15 23:48:04 +02005255
Antoine Pitrouab868312009-01-10 15:40:25 +00005256#ifdef BYTEORDER_IS_LITTLE_ENDIAN
Antoine Pitrou63065d72012-05-15 23:48:04 +02005257 native_ordering = bo <= 0;
Antoine Pitrouab868312009-01-10 15:40:25 +00005258#else
Antoine Pitrou63065d72012-05-15 23:48:04 +02005259 native_ordering = bo >= 0;
Antoine Pitrouab868312009-01-10 15:40:25 +00005260#endif
Tim Peters772747b2001-08-09 22:21:55 +00005261
Antoine Pitrou63065d72012-05-15 23:48:04 +02005262 /* Note: size will always be longer than the resulting Unicode
5263 character count */
5264 unicode = PyUnicode_New((e - q + 1) / 2, 127);
5265 if (!unicode)
5266 return NULL;
5267
5268 outpos = 0;
5269 while (1) {
5270 Py_UCS4 ch = 0;
5271 if (e - q >= 2) {
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005272 int kind = PyUnicode_KIND(unicode);
Antoine Pitrou63065d72012-05-15 23:48:04 +02005273 if (kind == PyUnicode_1BYTE_KIND) {
5274 if (PyUnicode_IS_ASCII(unicode))
5275 ch = asciilib_utf16_decode(&q, e,
5276 PyUnicode_1BYTE_DATA(unicode), &outpos,
5277 native_ordering);
5278 else
5279 ch = ucs1lib_utf16_decode(&q, e,
5280 PyUnicode_1BYTE_DATA(unicode), &outpos,
5281 native_ordering);
5282 } else if (kind == PyUnicode_2BYTE_KIND) {
5283 ch = ucs2lib_utf16_decode(&q, e,
5284 PyUnicode_2BYTE_DATA(unicode), &outpos,
5285 native_ordering);
5286 } else {
5287 assert(kind == PyUnicode_4BYTE_KIND);
5288 ch = ucs4lib_utf16_decode(&q, e,
5289 PyUnicode_4BYTE_DATA(unicode), &outpos,
5290 native_ordering);
Antoine Pitrouab868312009-01-10 15:40:25 +00005291 }
Antoine Pitrouab868312009-01-10 15:40:25 +00005292 }
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005293
Antoine Pitrou63065d72012-05-15 23:48:04 +02005294 switch (ch)
5295 {
5296 case 0:
5297 /* remaining byte at the end? (size should be even) */
5298 if (q == e || consumed)
5299 goto End;
5300 errmsg = "truncated data";
5301 startinpos = ((const char *)q) - starts;
5302 endinpos = ((const char *)e) - starts;
5303 break;
5304 /* The remaining input chars are ignored if the callback
5305 chooses to skip the input */
5306 case 1:
Serhiy Storchaka48e188e2013-01-08 23:14:24 +02005307 q -= 2;
5308 if (consumed)
Serhiy Storchakaae3b32a2013-01-08 23:40:52 +02005309 goto End;
Antoine Pitrou63065d72012-05-15 23:48:04 +02005310 errmsg = "unexpected end of data";
Serhiy Storchaka48e188e2013-01-08 23:14:24 +02005311 startinpos = ((const char *)q) - starts;
Antoine Pitrou63065d72012-05-15 23:48:04 +02005312 endinpos = ((const char *)e) - starts;
5313 break;
5314 case 2:
5315 errmsg = "illegal encoding";
5316 startinpos = ((const char *)q) - 2 - starts;
5317 endinpos = startinpos + 2;
5318 break;
5319 case 3:
5320 errmsg = "illegal UTF-16 surrogate";
5321 startinpos = ((const char *)q) - 4 - starts;
5322 endinpos = startinpos + 2;
5323 break;
5324 default:
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005325 if (unicode_putchar(&unicode, &outpos, ch) < 0)
5326 goto onError;
Benjamin Peterson29060642009-01-31 22:14:21 +00005327 continue;
5328 }
5329
Benjamin Peterson29060642009-01-31 22:14:21 +00005330 if (unicode_decode_call_errorhandler(
Antoine Pitrouab868312009-01-10 15:40:25 +00005331 errors,
5332 &errorHandler,
5333 "utf16", errmsg,
5334 &starts,
5335 (const char **)&e,
5336 &startinpos,
5337 &endinpos,
5338 &exc,
5339 (const char **)&q,
5340 &unicode,
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005341 &outpos))
Benjamin Peterson29060642009-01-31 22:14:21 +00005342 goto onError;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005343 }
5344
Antoine Pitrou63065d72012-05-15 23:48:04 +02005345End:
Walter Dörwald69652032004-09-07 20:24:22 +00005346 if (consumed)
Benjamin Peterson29060642009-01-31 22:14:21 +00005347 *consumed = (const char *)q-starts;
Walter Dörwald69652032004-09-07 20:24:22 +00005348
Guido van Rossumd57fd912000-03-10 22:53:23 +00005349 /* Adjust length */
Victor Stinner16e6a802011-12-12 13:24:15 +01005350 if (unicode_resize(&unicode, outpos) < 0)
Guido van Rossumd57fd912000-03-10 22:53:23 +00005351 goto onError;
5352
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005353 Py_XDECREF(errorHandler);
5354 Py_XDECREF(exc);
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01005355 return unicode_result(unicode);
Guido van Rossumd57fd912000-03-10 22:53:23 +00005356
Benjamin Peterson29060642009-01-31 22:14:21 +00005357 onError:
Guido van Rossumd57fd912000-03-10 22:53:23 +00005358 Py_DECREF(unicode);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005359 Py_XDECREF(errorHandler);
5360 Py_XDECREF(exc);
Guido van Rossumd57fd912000-03-10 22:53:23 +00005361 return NULL;
5362}
5363
Tim Peters772747b2001-08-09 22:21:55 +00005364PyObject *
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005365_PyUnicode_EncodeUTF16(PyObject *str,
5366 const char *errors,
5367 int byteorder)
Guido van Rossumd57fd912000-03-10 22:53:23 +00005368{
Antoine Pitrou27f6a3b2012-06-15 22:15:23 +02005369 enum PyUnicode_Kind kind;
5370 const void *data;
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005371 Py_ssize_t len;
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00005372 PyObject *v;
Antoine Pitrou27f6a3b2012-06-15 22:15:23 +02005373 unsigned short *out;
5374 Py_ssize_t bytesize;
5375 Py_ssize_t pairs;
5376#ifdef WORDS_BIGENDIAN
5377 int native_ordering = byteorder >= 0;
Tim Peters772747b2001-08-09 22:21:55 +00005378#else
Antoine Pitrou27f6a3b2012-06-15 22:15:23 +02005379 int native_ordering = byteorder <= 0;
Tim Peters772747b2001-08-09 22:21:55 +00005380#endif
5381
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005382 if (!PyUnicode_Check(str)) {
5383 PyErr_BadArgument();
5384 return NULL;
5385 }
Benjamin Petersonbac79492012-01-14 13:34:47 -05005386 if (PyUnicode_READY(str) == -1)
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005387 return NULL;
5388 kind = PyUnicode_KIND(str);
5389 data = PyUnicode_DATA(str);
5390 len = PyUnicode_GET_LENGTH(str);
Victor Stinner0e368262011-11-10 20:12:49 +01005391
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005392 pairs = 0;
Antoine Pitrou27f6a3b2012-06-15 22:15:23 +02005393 if (kind == PyUnicode_4BYTE_KIND) {
5394 const Py_UCS4 *in = (const Py_UCS4 *)data;
5395 const Py_UCS4 *end = in + len;
5396 while (in < end)
5397 if (*in++ >= 0x10000)
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005398 pairs++;
Antoine Pitrou27f6a3b2012-06-15 22:15:23 +02005399 }
5400 if (len > PY_SSIZE_T_MAX / 2 - pairs - (byteorder == 0))
Benjamin Peterson29060642009-01-31 22:14:21 +00005401 return PyErr_NoMemory();
Antoine Pitrou27f6a3b2012-06-15 22:15:23 +02005402 bytesize = (len + pairs + (byteorder == 0)) * 2;
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00005403 v = PyBytes_FromStringAndSize(NULL, bytesize);
Guido van Rossumd57fd912000-03-10 22:53:23 +00005404 if (v == NULL)
5405 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005406
Antoine Pitrou27f6a3b2012-06-15 22:15:23 +02005407 /* output buffer is 2-bytes aligned */
Antoine Pitrouca8aa4a2012-09-20 20:56:47 +02005408 assert(_Py_IS_ALIGNED(PyBytes_AS_STRING(v), 2));
Antoine Pitrou27f6a3b2012-06-15 22:15:23 +02005409 out = (unsigned short *)PyBytes_AS_STRING(v);
Guido van Rossumd57fd912000-03-10 22:53:23 +00005410 if (byteorder == 0)
Antoine Pitrou27f6a3b2012-06-15 22:15:23 +02005411 *out++ = 0xFEFF;
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005412 if (len == 0)
Guido van Rossum98297ee2007-11-06 21:34:58 +00005413 goto done;
Tim Peters772747b2001-08-09 22:21:55 +00005414
Antoine Pitrou27f6a3b2012-06-15 22:15:23 +02005415 switch (kind) {
5416 case PyUnicode_1BYTE_KIND: {
5417 ucs1lib_utf16_encode(out, (const Py_UCS1 *)data, len, native_ordering);
5418 break;
Tim Peters772747b2001-08-09 22:21:55 +00005419 }
Antoine Pitrou27f6a3b2012-06-15 22:15:23 +02005420 case PyUnicode_2BYTE_KIND: {
5421 ucs2lib_utf16_encode(out, (const Py_UCS2 *)data, len, native_ordering);
5422 break;
Tim Peters772747b2001-08-09 22:21:55 +00005423 }
Antoine Pitrou27f6a3b2012-06-15 22:15:23 +02005424 case PyUnicode_4BYTE_KIND: {
5425 ucs4lib_utf16_encode(out, (const Py_UCS4 *)data, len, native_ordering);
5426 break;
5427 }
5428 default:
5429 assert(0);
Martin v. Löwis0ba70cc2001-06-26 22:22:37 +00005430 }
Guido van Rossum98297ee2007-11-06 21:34:58 +00005431
5432 done:
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00005433 return v;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005434}
5435
Alexander Belopolsky40018472011-02-26 01:02:56 +00005436PyObject *
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005437PyUnicode_EncodeUTF16(const Py_UNICODE *s,
5438 Py_ssize_t size,
5439 const char *errors,
5440 int byteorder)
5441{
5442 PyObject *result;
5443 PyObject *tmp = PyUnicode_FromUnicode(s, size);
5444 if (tmp == NULL)
5445 return NULL;
5446 result = _PyUnicode_EncodeUTF16(tmp, errors, byteorder);
5447 Py_DECREF(tmp);
5448 return result;
5449}
5450
5451PyObject *
Alexander Belopolsky40018472011-02-26 01:02:56 +00005452PyUnicode_AsUTF16String(PyObject *unicode)
Guido van Rossumd57fd912000-03-10 22:53:23 +00005453{
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005454 return _PyUnicode_EncodeUTF16(unicode, NULL, 0);
Guido van Rossumd57fd912000-03-10 22:53:23 +00005455}
5456
5457/* --- Unicode Escape Codec ----------------------------------------------- */
5458
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02005459/* Helper function for PyUnicode_DecodeUnicodeEscape, determines
5460 if all the escapes in the string make it still a valid ASCII string.
5461 Returns -1 if any escapes were found which cause the string to
5462 pop out of ASCII range. Otherwise returns the length of the
5463 required buffer to hold the string.
5464 */
Antoine Pitrou53bb5482011-10-10 23:49:24 +02005465static Py_ssize_t
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02005466length_of_escaped_ascii_string(const char *s, Py_ssize_t size)
5467{
5468 const unsigned char *p = (const unsigned char *)s;
5469 const unsigned char *end = p + size;
5470 Py_ssize_t length = 0;
5471
5472 if (size < 0)
5473 return -1;
5474
5475 for (; p < end; ++p) {
5476 if (*p > 127) {
5477 /* Non-ASCII */
5478 return -1;
5479 }
5480 else if (*p != '\\') {
5481 /* Normal character */
5482 ++length;
5483 }
5484 else {
5485 /* Backslash-escape, check next char */
5486 ++p;
5487 /* Escape sequence reaches till end of string or
5488 non-ASCII follow-up. */
5489 if (p >= end || *p > 127)
5490 return -1;
5491 switch (*p) {
5492 case '\n':
5493 /* backslash + \n result in zero characters */
5494 break;
5495 case '\\': case '\'': case '\"':
5496 case 'b': case 'f': case 't':
5497 case 'n': case 'r': case 'v': case 'a':
5498 ++length;
5499 break;
5500 case '0': case '1': case '2': case '3':
5501 case '4': case '5': case '6': case '7':
5502 case 'x': case 'u': case 'U': case 'N':
5503 /* these do not guarantee ASCII characters */
5504 return -1;
5505 default:
5506 /* count the backslash + the other character */
5507 length += 2;
5508 }
5509 }
5510 }
5511 return length;
5512}
5513
Fredrik Lundh06d12682001-01-24 07:59:11 +00005514static _PyUnicode_Name_CAPI *ucnhash_CAPI = NULL;
Marc-André Lemburg0f774e32000-06-28 16:43:35 +00005515
Alexander Belopolsky40018472011-02-26 01:02:56 +00005516PyObject *
5517PyUnicode_DecodeUnicodeEscape(const char *s,
Ezio Melotti2aa2b3b2011-09-29 00:58:57 +03005518 Py_ssize_t size,
Victor Stinnerc17f5402011-09-29 00:16:58 +02005519 const char *errors)
Guido van Rossumd57fd912000-03-10 22:53:23 +00005520{
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005521 const char *starts = s;
Martin v. Löwis18e16552006-02-15 17:27:45 +00005522 Py_ssize_t startinpos;
5523 Py_ssize_t endinpos;
Victor Stinner7931d9a2011-11-04 00:22:48 +01005524 PyObject *v;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005525 const char *end;
Fredrik Lundhccc74732001-02-18 22:13:49 +00005526 char* message;
5527 Py_UCS4 chr = 0xffffffff; /* in case 'getcode' messes up */
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005528 PyObject *errorHandler = NULL;
5529 PyObject *exc = NULL;
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005530 Py_ssize_t len;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02005531 Py_ssize_t i;
Fredrik Lundhccc74732001-02-18 22:13:49 +00005532
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005533 len = length_of_escaped_ascii_string(s, size);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02005534
5535 /* After length_of_escaped_ascii_string() there are two alternatives,
5536 either the string is pure ASCII with named escapes like \n, etc.
5537 and we determined it's exact size (common case)
5538 or it contains \x, \u, ... escape sequences. then we create a
5539 legacy wchar string and resize it at the end of this function. */
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005540 if (len >= 0) {
5541 v = PyUnicode_New(len, 127);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02005542 if (!v)
5543 goto onError;
5544 assert(PyUnicode_KIND(v) == PyUnicode_1BYTE_KIND);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02005545 }
5546 else {
5547 /* Escaped strings will always be longer than the resulting
5548 Unicode string, so we start with size here and then reduce the
5549 length after conversion to the true value.
5550 (but if the error callback returns a long replacement string
5551 we'll have to allocate more space) */
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005552 v = PyUnicode_New(size, 127);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02005553 if (!v)
5554 goto onError;
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005555 len = size;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02005556 }
5557
Guido van Rossumd57fd912000-03-10 22:53:23 +00005558 if (size == 0)
Victor Stinner7931d9a2011-11-04 00:22:48 +01005559 return v;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02005560 i = 0;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005561 end = s + size;
Fredrik Lundhccc74732001-02-18 22:13:49 +00005562
Guido van Rossumd57fd912000-03-10 22:53:23 +00005563 while (s < end) {
5564 unsigned char c;
Victor Stinner24729f32011-11-10 20:31:37 +01005565 Py_UCS4 x;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005566 int digits;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005567
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005568 /* The only case in which i == ascii_length is a backslash
5569 followed by a newline. */
5570 assert(i <= len);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02005571
Guido van Rossumd57fd912000-03-10 22:53:23 +00005572 /* Non-escape characters are interpreted as Unicode ordinals */
5573 if (*s != '\\') {
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005574 if (unicode_putchar(&v, &i, (unsigned char) *s++) < 0)
5575 goto onError;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005576 continue;
5577 }
5578
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005579 startinpos = s-starts;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005580 /* \ - Escapes */
5581 s++;
Guido van Rossum8ce8a782007-11-01 19:42:39 +00005582 c = *s++;
5583 if (s > end)
5584 c = '\0'; /* Invalid after \ */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02005585
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005586 /* The only case in which i == ascii_length is a backslash
5587 followed by a newline. */
5588 assert(i < len || (i == len && c == '\n'));
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02005589
Guido van Rossum8ce8a782007-11-01 19:42:39 +00005590 switch (c) {
Guido van Rossumd57fd912000-03-10 22:53:23 +00005591
Benjamin Peterson29060642009-01-31 22:14:21 +00005592 /* \x escapes */
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005593#define WRITECHAR(ch) \
5594 do { \
5595 if (unicode_putchar(&v, &i, ch) < 0) \
5596 goto onError; \
5597 }while(0)
5598
Guido van Rossumd57fd912000-03-10 22:53:23 +00005599 case '\n': break;
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005600 case '\\': WRITECHAR('\\'); break;
5601 case '\'': WRITECHAR('\''); break;
5602 case '\"': WRITECHAR('\"'); break;
5603 case 'b': WRITECHAR('\b'); break;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02005604 /* FF */
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005605 case 'f': WRITECHAR('\014'); break;
5606 case 't': WRITECHAR('\t'); break;
5607 case 'n': WRITECHAR('\n'); break;
5608 case 'r': WRITECHAR('\r'); break;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02005609 /* VT */
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005610 case 'v': WRITECHAR('\013'); break;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02005611 /* BEL, not classic C */
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005612 case 'a': WRITECHAR('\007'); break;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005613
Benjamin Peterson29060642009-01-31 22:14:21 +00005614 /* \OOO (octal) escapes */
Guido van Rossumd57fd912000-03-10 22:53:23 +00005615 case '0': case '1': case '2': case '3':
5616 case '4': case '5': case '6': case '7':
Guido van Rossum0e4f6572000-05-01 21:27:20 +00005617 x = s[-1] - '0';
Guido van Rossum8ce8a782007-11-01 19:42:39 +00005618 if (s < end && '0' <= *s && *s <= '7') {
Guido van Rossum0e4f6572000-05-01 21:27:20 +00005619 x = (x<<3) + *s++ - '0';
Guido van Rossum8ce8a782007-11-01 19:42:39 +00005620 if (s < end && '0' <= *s && *s <= '7')
Guido van Rossum0e4f6572000-05-01 21:27:20 +00005621 x = (x<<3) + *s++ - '0';
Guido van Rossumd57fd912000-03-10 22:53:23 +00005622 }
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005623 WRITECHAR(x);
Guido van Rossumd57fd912000-03-10 22:53:23 +00005624 break;
5625
Benjamin Peterson29060642009-01-31 22:14:21 +00005626 /* hex escapes */
5627 /* \xXX */
Guido van Rossumd57fd912000-03-10 22:53:23 +00005628 case 'x':
Fredrik Lundhccc74732001-02-18 22:13:49 +00005629 digits = 2;
5630 message = "truncated \\xXX escape";
5631 goto hexescape;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005632
Benjamin Peterson29060642009-01-31 22:14:21 +00005633 /* \uXXXX */
Guido van Rossumd57fd912000-03-10 22:53:23 +00005634 case 'u':
Fredrik Lundhccc74732001-02-18 22:13:49 +00005635 digits = 4;
5636 message = "truncated \\uXXXX escape";
5637 goto hexescape;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005638
Benjamin Peterson29060642009-01-31 22:14:21 +00005639 /* \UXXXXXXXX */
Fredrik Lundhdf846752000-09-03 11:29:49 +00005640 case 'U':
Fredrik Lundhccc74732001-02-18 22:13:49 +00005641 digits = 8;
5642 message = "truncated \\UXXXXXXXX escape";
5643 hexescape:
5644 chr = 0;
Serhiy Storchakad6793772013-01-29 10:20:44 +02005645 if (end - s < digits) {
5646 /* count only hex digits */
5647 for (; s < end; ++s) {
5648 c = (unsigned char)*s;
5649 if (!Py_ISXDIGIT(c))
5650 goto error;
Fredrik Lundhdf846752000-09-03 11:29:49 +00005651 }
Serhiy Storchakad6793772013-01-29 10:20:44 +02005652 goto error;
5653 }
5654 for (; digits--; ++s) {
5655 c = (unsigned char)*s;
5656 if (!Py_ISXDIGIT(c))
5657 goto error;
Fredrik Lundhdf846752000-09-03 11:29:49 +00005658 chr = (chr<<4) & ~0xF;
5659 if (c >= '0' && c <= '9')
5660 chr += c - '0';
5661 else if (c >= 'a' && c <= 'f')
5662 chr += 10 + c - 'a';
5663 else
5664 chr += 10 + c - 'A';
5665 }
Jeremy Hylton504de6b2003-10-06 05:08:26 +00005666 if (chr == 0xffffffff && PyErr_Occurred())
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005667 /* _decoding_error will have already written into the
5668 target buffer. */
5669 break;
Fredrik Lundhccc74732001-02-18 22:13:49 +00005670 store:
Fredrik Lundhdf846752000-09-03 11:29:49 +00005671 /* when we get here, chr is a 32-bit unicode character */
Serhiy Storchaka24193de2013-01-29 10:28:07 +02005672 message = "illegal Unicode character";
5673 if (chr > MAX_UNICODE)
Serhiy Storchakad6793772013-01-29 10:20:44 +02005674 goto error;
Serhiy Storchaka24193de2013-01-29 10:28:07 +02005675 WRITECHAR(chr);
Fredrik Lundhccc74732001-02-18 22:13:49 +00005676 break;
5677
Benjamin Peterson29060642009-01-31 22:14:21 +00005678 /* \N{name} */
Fredrik Lundhccc74732001-02-18 22:13:49 +00005679 case 'N':
5680 message = "malformed \\N character escape";
5681 if (ucnhash_CAPI == NULL) {
5682 /* load the unicode data module */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02005683 ucnhash_CAPI = (_PyUnicode_Name_CAPI *)PyCapsule_Import(
5684 PyUnicodeData_CAPSULE_NAME, 1);
Fredrik Lundhccc74732001-02-18 22:13:49 +00005685 if (ucnhash_CAPI == NULL)
5686 goto ucnhashError;
5687 }
5688 if (*s == '{') {
5689 const char *start = s+1;
5690 /* look for the closing brace */
5691 while (*s != '}' && s < end)
5692 s++;
5693 if (s > start && s < end && *s == '}') {
5694 /* found a name. look it up in the unicode database */
5695 message = "unknown Unicode character name";
5696 s++;
Serhiy Storchaka4f5f0e52013-01-21 11:38:00 +02005697 if (s - start - 1 <= INT_MAX &&
Serhiy Storchakac35f3a92013-01-21 11:42:57 +02005698 ucnhash_CAPI->getcode(NULL, start, (int)(s-start-1),
Ezio Melotti931b8aa2011-10-21 21:57:36 +03005699 &chr, 0))
Fredrik Lundhccc74732001-02-18 22:13:49 +00005700 goto store;
5701 }
5702 }
Serhiy Storchakad6793772013-01-29 10:20:44 +02005703 goto error;
Fredrik Lundhccc74732001-02-18 22:13:49 +00005704
5705 default:
Walter Dörwald8c077222002-03-25 11:16:18 +00005706 if (s > end) {
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005707 message = "\\ at end of string";
5708 s--;
Serhiy Storchakad6793772013-01-29 10:20:44 +02005709 goto error;
Walter Dörwald8c077222002-03-25 11:16:18 +00005710 }
5711 else {
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005712 WRITECHAR('\\');
Serhiy Storchaka73e38802013-01-25 23:52:21 +02005713 WRITECHAR((unsigned char)s[-1]);
Walter Dörwald8c077222002-03-25 11:16:18 +00005714 }
Fredrik Lundhccc74732001-02-18 22:13:49 +00005715 break;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005716 }
Serhiy Storchakad6793772013-01-29 10:20:44 +02005717 continue;
5718
5719 error:
5720 endinpos = s-starts;
Serhiy Storchakad6793772013-01-29 10:20:44 +02005721 if (unicode_decode_call_errorhandler(
5722 errors, &errorHandler,
5723 "unicodeescape", message,
5724 &starts, &end, &startinpos, &endinpos, &exc, &s,
Serhiy Storchaka24193de2013-01-29 10:28:07 +02005725 &v, &i))
Serhiy Storchakad6793772013-01-29 10:20:44 +02005726 goto onError;
Serhiy Storchaka24193de2013-01-29 10:28:07 +02005727 len = PyUnicode_GET_LENGTH(v);
Serhiy Storchakad6793772013-01-29 10:20:44 +02005728 continue;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005729 }
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005730#undef WRITECHAR
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02005731
Victor Stinner16e6a802011-12-12 13:24:15 +01005732 if (unicode_resize(&v, i) < 0)
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005733 goto onError;
Walter Dörwaldd4ade082003-08-15 15:00:26 +00005734 Py_XDECREF(errorHandler);
5735 Py_XDECREF(exc);
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01005736 return unicode_result(v);
Walter Dörwald8c077222002-03-25 11:16:18 +00005737
Benjamin Peterson29060642009-01-31 22:14:21 +00005738 ucnhashError:
Fredrik Lundh06d12682001-01-24 07:59:11 +00005739 PyErr_SetString(
5740 PyExc_UnicodeError,
5741 "\\N escapes not supported (can't load unicodedata module)"
5742 );
Hye-Shik Chang4af5c8c2006-03-07 15:39:21 +00005743 Py_XDECREF(v);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005744 Py_XDECREF(errorHandler);
5745 Py_XDECREF(exc);
Fredrik Lundhf6056062001-01-20 11:15:25 +00005746 return NULL;
5747
Benjamin Peterson29060642009-01-31 22:14:21 +00005748 onError:
Guido van Rossumd57fd912000-03-10 22:53:23 +00005749 Py_XDECREF(v);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005750 Py_XDECREF(errorHandler);
5751 Py_XDECREF(exc);
Guido van Rossumd57fd912000-03-10 22:53:23 +00005752 return NULL;
5753}
5754
5755/* Return a Unicode-Escape string version of the Unicode object.
5756
5757 If quotes is true, the string is enclosed in u"" or u'' quotes as
5758 appropriate.
5759
5760*/
5761
Alexander Belopolsky40018472011-02-26 01:02:56 +00005762PyObject *
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005763PyUnicode_AsUnicodeEscapeString(PyObject *unicode)
Guido van Rossumd57fd912000-03-10 22:53:23 +00005764{
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005765 Py_ssize_t i, len;
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00005766 PyObject *repr;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005767 char *p;
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005768 int kind;
5769 void *data;
5770 Py_ssize_t expandsize = 0;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005771
Ezio Melottie7f90372012-10-05 03:33:31 +03005772 /* Initial allocation is based on the longest-possible character
Thomas Wouters89f507f2006-12-13 04:49:30 +00005773 escape.
5774
Ezio Melottie7f90372012-10-05 03:33:31 +03005775 For UCS1 strings it's '\xxx', 4 bytes per source character.
5776 For UCS2 strings it's '\uxxxx', 6 bytes per source character.
5777 For UCS4 strings it's '\U00xxxxxx', 10 bytes per source character.
Thomas Wouters89f507f2006-12-13 04:49:30 +00005778 */
5779
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005780 if (!PyUnicode_Check(unicode)) {
5781 PyErr_BadArgument();
5782 return NULL;
5783 }
Benjamin Petersonbac79492012-01-14 13:34:47 -05005784 if (PyUnicode_READY(unicode) == -1)
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005785 return NULL;
5786 len = PyUnicode_GET_LENGTH(unicode);
5787 kind = PyUnicode_KIND(unicode);
5788 data = PyUnicode_DATA(unicode);
Benjamin Petersonead6b532011-12-20 17:23:42 -06005789 switch (kind) {
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005790 case PyUnicode_1BYTE_KIND: expandsize = 4; break;
5791 case PyUnicode_2BYTE_KIND: expandsize = 6; break;
5792 case PyUnicode_4BYTE_KIND: expandsize = 10; break;
5793 }
5794
5795 if (len == 0)
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00005796 return PyBytes_FromStringAndSize(NULL, 0);
5797
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005798 if (len > (PY_SSIZE_T_MAX - 2 - 1) / expandsize)
Benjamin Peterson29060642009-01-31 22:14:21 +00005799 return PyErr_NoMemory();
Neal Norwitz3ce5d922008-08-24 07:08:55 +00005800
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00005801 repr = PyBytes_FromStringAndSize(NULL,
Benjamin Peterson29060642009-01-31 22:14:21 +00005802 2
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005803 + expandsize*len
Benjamin Peterson29060642009-01-31 22:14:21 +00005804 + 1);
Guido van Rossumd57fd912000-03-10 22:53:23 +00005805 if (repr == NULL)
5806 return NULL;
5807
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00005808 p = PyBytes_AS_STRING(repr);
Guido van Rossumd57fd912000-03-10 22:53:23 +00005809
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005810 for (i = 0; i < len; i++) {
Victor Stinner3326cb62011-11-10 20:15:25 +01005811 Py_UCS4 ch = PyUnicode_READ(kind, data, i);
Marc-André Lemburg80d1dd52001-07-25 16:05:59 +00005812
Walter Dörwald79e913e2007-05-12 11:08:06 +00005813 /* Escape backslashes */
5814 if (ch == '\\') {
Guido van Rossumd57fd912000-03-10 22:53:23 +00005815 *p++ = '\\';
5816 *p++ = (char) ch;
Walter Dörwald79e913e2007-05-12 11:08:06 +00005817 continue;
Tim Petersced69f82003-09-16 20:30:58 +00005818 }
Marc-André Lemburg80d1dd52001-07-25 16:05:59 +00005819
Martin v. Löwis0ba70cc2001-06-26 22:22:37 +00005820 /* Map 21-bit characters to '\U00xxxxxx' */
5821 else if (ch >= 0x10000) {
Victor Stinner8faf8212011-12-08 22:14:11 +01005822 assert(ch <= MAX_UNICODE);
Martin v. Löwis0ba70cc2001-06-26 22:22:37 +00005823 *p++ = '\\';
5824 *p++ = 'U';
Victor Stinnerf5cff562011-10-14 02:13:11 +02005825 *p++ = Py_hexdigits[(ch >> 28) & 0x0000000F];
5826 *p++ = Py_hexdigits[(ch >> 24) & 0x0000000F];
5827 *p++ = Py_hexdigits[(ch >> 20) & 0x0000000F];
5828 *p++ = Py_hexdigits[(ch >> 16) & 0x0000000F];
5829 *p++ = Py_hexdigits[(ch >> 12) & 0x0000000F];
5830 *p++ = Py_hexdigits[(ch >> 8) & 0x0000000F];
5831 *p++ = Py_hexdigits[(ch >> 4) & 0x0000000F];
5832 *p++ = Py_hexdigits[ch & 0x0000000F];
Benjamin Peterson29060642009-01-31 22:14:21 +00005833 continue;
Martin v. Löwis0ba70cc2001-06-26 22:22:37 +00005834 }
Marc-André Lemburg6c6bfb72001-07-20 17:39:11 +00005835
Guido van Rossumd57fd912000-03-10 22:53:23 +00005836 /* Map 16-bit characters to '\uxxxx' */
Marc-André Lemburg6c6bfb72001-07-20 17:39:11 +00005837 if (ch >= 256) {
Guido van Rossumd57fd912000-03-10 22:53:23 +00005838 *p++ = '\\';
5839 *p++ = 'u';
Victor Stinnerf5cff562011-10-14 02:13:11 +02005840 *p++ = Py_hexdigits[(ch >> 12) & 0x000F];
5841 *p++ = Py_hexdigits[(ch >> 8) & 0x000F];
5842 *p++ = Py_hexdigits[(ch >> 4) & 0x000F];
5843 *p++ = Py_hexdigits[ch & 0x000F];
Guido van Rossumd57fd912000-03-10 22:53:23 +00005844 }
Marc-André Lemburg80d1dd52001-07-25 16:05:59 +00005845
Ka-Ping Yeefa004ad2001-01-24 17:19:08 +00005846 /* Map special whitespace to '\t', \n', '\r' */
5847 else if (ch == '\t') {
5848 *p++ = '\\';
5849 *p++ = 't';
5850 }
5851 else if (ch == '\n') {
5852 *p++ = '\\';
5853 *p++ = 'n';
5854 }
5855 else if (ch == '\r') {
5856 *p++ = '\\';
5857 *p++ = 'r';
5858 }
Marc-André Lemburg80d1dd52001-07-25 16:05:59 +00005859
Ka-Ping Yeefa004ad2001-01-24 17:19:08 +00005860 /* Map non-printable US ASCII to '\xhh' */
Marc-André Lemburg11326de2001-11-28 12:56:20 +00005861 else if (ch < ' ' || ch >= 0x7F) {
Guido van Rossumd57fd912000-03-10 22:53:23 +00005862 *p++ = '\\';
Ka-Ping Yeefa004ad2001-01-24 17:19:08 +00005863 *p++ = 'x';
Victor Stinnerf5cff562011-10-14 02:13:11 +02005864 *p++ = Py_hexdigits[(ch >> 4) & 0x000F];
5865 *p++ = Py_hexdigits[ch & 0x000F];
Tim Petersced69f82003-09-16 20:30:58 +00005866 }
Marc-André Lemburg80d1dd52001-07-25 16:05:59 +00005867
Guido van Rossumd57fd912000-03-10 22:53:23 +00005868 /* Copy everything else as-is */
5869 else
5870 *p++ = (char) ch;
5871 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00005872
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00005873 assert(p - PyBytes_AS_STRING(repr) > 0);
5874 if (_PyBytes_Resize(&repr, p - PyBytes_AS_STRING(repr)) < 0)
5875 return NULL;
5876 return repr;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005877}
5878
Alexander Belopolsky40018472011-02-26 01:02:56 +00005879PyObject *
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005880PyUnicode_EncodeUnicodeEscape(const Py_UNICODE *s,
5881 Py_ssize_t size)
Guido van Rossumd57fd912000-03-10 22:53:23 +00005882{
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005883 PyObject *result;
5884 PyObject *tmp = PyUnicode_FromUnicode(s, size);
5885 if (tmp == NULL)
Guido van Rossumd57fd912000-03-10 22:53:23 +00005886 return NULL;
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005887 result = PyUnicode_AsUnicodeEscapeString(tmp);
5888 Py_DECREF(tmp);
5889 return result;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005890}
5891
5892/* --- Raw Unicode Escape Codec ------------------------------------------- */
5893
Alexander Belopolsky40018472011-02-26 01:02:56 +00005894PyObject *
5895PyUnicode_DecodeRawUnicodeEscape(const char *s,
Ezio Melotti2aa2b3b2011-09-29 00:58:57 +03005896 Py_ssize_t size,
5897 const char *errors)
Guido van Rossumd57fd912000-03-10 22:53:23 +00005898{
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005899 const char *starts = s;
Martin v. Löwis18e16552006-02-15 17:27:45 +00005900 Py_ssize_t startinpos;
5901 Py_ssize_t endinpos;
5902 Py_ssize_t outpos;
Victor Stinner7931d9a2011-11-04 00:22:48 +01005903 PyObject *v;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005904 const char *end;
5905 const char *bs;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005906 PyObject *errorHandler = NULL;
5907 PyObject *exc = NULL;
Tim Petersced69f82003-09-16 20:30:58 +00005908
Guido van Rossumd57fd912000-03-10 22:53:23 +00005909 /* Escaped strings will always be longer than the resulting
5910 Unicode string, so we start with size here and then reduce the
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005911 length after conversion to the true value. (But decoding error
5912 handler might have to resize the string) */
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005913 v = PyUnicode_New(size, 127);
Guido van Rossumd57fd912000-03-10 22:53:23 +00005914 if (v == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00005915 goto onError;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005916 if (size == 0)
Victor Stinner7931d9a2011-11-04 00:22:48 +01005917 return v;
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005918 outpos = 0;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005919 end = s + size;
5920 while (s < end) {
Benjamin Peterson29060642009-01-31 22:14:21 +00005921 unsigned char c;
5922 Py_UCS4 x;
5923 int i;
Martin v. Löwis9a3a9f72003-05-18 12:31:09 +00005924 int count;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005925
Benjamin Peterson29060642009-01-31 22:14:21 +00005926 /* Non-escape characters are interpreted as Unicode ordinals */
5927 if (*s != '\\') {
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005928 if (unicode_putchar(&v, &outpos, (unsigned char)*s++) < 0)
5929 goto onError;
Benjamin Peterson29060642009-01-31 22:14:21 +00005930 continue;
Benjamin Peterson14339b62009-01-31 16:36:08 +00005931 }
Benjamin Peterson29060642009-01-31 22:14:21 +00005932 startinpos = s-starts;
5933
5934 /* \u-escapes are only interpreted iff the number of leading
5935 backslashes if odd */
5936 bs = s;
5937 for (;s < end;) {
5938 if (*s != '\\')
5939 break;
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005940 if (unicode_putchar(&v, &outpos, (unsigned char)*s++) < 0)
5941 goto onError;
Benjamin Peterson29060642009-01-31 22:14:21 +00005942 }
5943 if (((s - bs) & 1) == 0 ||
5944 s >= end ||
5945 (*s != 'u' && *s != 'U')) {
5946 continue;
5947 }
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005948 outpos--;
Benjamin Peterson29060642009-01-31 22:14:21 +00005949 count = *s=='u' ? 4 : 8;
5950 s++;
5951
5952 /* \uXXXX with 4 hex digits, \Uxxxxxxxx with 8 */
Benjamin Peterson29060642009-01-31 22:14:21 +00005953 for (x = 0, i = 0; i < count; ++i, ++s) {
5954 c = (unsigned char)*s;
David Malcolm96960882010-11-05 17:23:41 +00005955 if (!Py_ISXDIGIT(c)) {
Benjamin Peterson29060642009-01-31 22:14:21 +00005956 endinpos = s-starts;
5957 if (unicode_decode_call_errorhandler(
5958 errors, &errorHandler,
5959 "rawunicodeescape", "truncated \\uXXXX",
5960 &starts, &end, &startinpos, &endinpos, &exc, &s,
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005961 &v, &outpos))
Benjamin Peterson29060642009-01-31 22:14:21 +00005962 goto onError;
5963 goto nextByte;
5964 }
5965 x = (x<<4) & ~0xF;
5966 if (c >= '0' && c <= '9')
5967 x += c - '0';
5968 else if (c >= 'a' && c <= 'f')
5969 x += 10 + c - 'a';
5970 else
5971 x += 10 + c - 'A';
5972 }
Victor Stinner8faf8212011-12-08 22:14:11 +01005973 if (x <= MAX_UNICODE) {
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005974 if (unicode_putchar(&v, &outpos, x) < 0)
5975 goto onError;
Christian Heimesfe337bf2008-03-23 21:54:12 +00005976 } else {
5977 endinpos = s-starts;
Martin v. Löwis9a3a9f72003-05-18 12:31:09 +00005978 if (unicode_decode_call_errorhandler(
5979 errors, &errorHandler,
5980 "rawunicodeescape", "\\Uxxxxxxxx out of range",
Benjamin Peterson29060642009-01-31 22:14:21 +00005981 &starts, &end, &startinpos, &endinpos, &exc, &s,
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005982 &v, &outpos))
Benjamin Peterson29060642009-01-31 22:14:21 +00005983 goto onError;
Martin v. Löwis9a3a9f72003-05-18 12:31:09 +00005984 }
Benjamin Peterson29060642009-01-31 22:14:21 +00005985 nextByte:
5986 ;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005987 }
Victor Stinner16e6a802011-12-12 13:24:15 +01005988 if (unicode_resize(&v, outpos) < 0)
Benjamin Peterson29060642009-01-31 22:14:21 +00005989 goto onError;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005990 Py_XDECREF(errorHandler);
5991 Py_XDECREF(exc);
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01005992 return unicode_result(v);
Tim Petersced69f82003-09-16 20:30:58 +00005993
Benjamin Peterson29060642009-01-31 22:14:21 +00005994 onError:
Guido van Rossumd57fd912000-03-10 22:53:23 +00005995 Py_XDECREF(v);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005996 Py_XDECREF(errorHandler);
5997 Py_XDECREF(exc);
Guido van Rossumd57fd912000-03-10 22:53:23 +00005998 return NULL;
5999}
6000
Martin v. Löwis1db7c132011-11-10 18:24:32 +01006001
Alexander Belopolsky40018472011-02-26 01:02:56 +00006002PyObject *
Martin v. Löwis1db7c132011-11-10 18:24:32 +01006003PyUnicode_AsRawUnicodeEscapeString(PyObject *unicode)
Guido van Rossumd57fd912000-03-10 22:53:23 +00006004{
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00006005 PyObject *repr;
Guido van Rossumd57fd912000-03-10 22:53:23 +00006006 char *p;
6007 char *q;
Martin v. Löwis1db7c132011-11-10 18:24:32 +01006008 Py_ssize_t expandsize, pos;
6009 int kind;
6010 void *data;
6011 Py_ssize_t len;
Guido van Rossumd57fd912000-03-10 22:53:23 +00006012
Martin v. Löwis1db7c132011-11-10 18:24:32 +01006013 if (!PyUnicode_Check(unicode)) {
6014 PyErr_BadArgument();
6015 return NULL;
6016 }
Benjamin Petersonbac79492012-01-14 13:34:47 -05006017 if (PyUnicode_READY(unicode) == -1)
Martin v. Löwis1db7c132011-11-10 18:24:32 +01006018 return NULL;
6019 kind = PyUnicode_KIND(unicode);
6020 data = PyUnicode_DATA(unicode);
6021 len = PyUnicode_GET_LENGTH(unicode);
Benjamin Peterson1518e872011-11-23 10:44:52 -06006022 /* 4 byte characters can take up 10 bytes, 2 byte characters can take up 6
6023 bytes, and 1 byte characters 4. */
6024 expandsize = kind * 2 + 2;
Victor Stinner0e368262011-11-10 20:12:49 +01006025
Martin v. Löwis1db7c132011-11-10 18:24:32 +01006026 if (len > PY_SSIZE_T_MAX / expandsize)
Benjamin Peterson29060642009-01-31 22:14:21 +00006027 return PyErr_NoMemory();
Benjamin Peterson14339b62009-01-31 16:36:08 +00006028
Martin v. Löwis1db7c132011-11-10 18:24:32 +01006029 repr = PyBytes_FromStringAndSize(NULL, expandsize * len);
Guido van Rossumd57fd912000-03-10 22:53:23 +00006030 if (repr == NULL)
6031 return NULL;
Martin v. Löwis1db7c132011-11-10 18:24:32 +01006032 if (len == 0)
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00006033 return repr;
Guido van Rossumd57fd912000-03-10 22:53:23 +00006034
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00006035 p = q = PyBytes_AS_STRING(repr);
Martin v. Löwis1db7c132011-11-10 18:24:32 +01006036 for (pos = 0; pos < len; pos++) {
6037 Py_UCS4 ch = PyUnicode_READ(kind, data, pos);
Benjamin Peterson29060642009-01-31 22:14:21 +00006038 /* Map 32-bit characters to '\Uxxxxxxxx' */
6039 if (ch >= 0x10000) {
Victor Stinner8faf8212011-12-08 22:14:11 +01006040 assert(ch <= MAX_UNICODE);
Martin v. Löwis9a3a9f72003-05-18 12:31:09 +00006041 *p++ = '\\';
6042 *p++ = 'U';
Victor Stinnerf5cff562011-10-14 02:13:11 +02006043 *p++ = Py_hexdigits[(ch >> 28) & 0xf];
6044 *p++ = Py_hexdigits[(ch >> 24) & 0xf];
6045 *p++ = Py_hexdigits[(ch >> 20) & 0xf];
6046 *p++ = Py_hexdigits[(ch >> 16) & 0xf];
6047 *p++ = Py_hexdigits[(ch >> 12) & 0xf];
6048 *p++ = Py_hexdigits[(ch >> 8) & 0xf];
6049 *p++ = Py_hexdigits[(ch >> 4) & 0xf];
6050 *p++ = Py_hexdigits[ch & 15];
Tim Petersced69f82003-09-16 20:30:58 +00006051 }
Benjamin Peterson29060642009-01-31 22:14:21 +00006052 /* Map 16-bit characters to '\uxxxx' */
Martin v. Löwis1db7c132011-11-10 18:24:32 +01006053 else if (ch >= 256) {
Guido van Rossumd57fd912000-03-10 22:53:23 +00006054 *p++ = '\\';
6055 *p++ = 'u';
Victor Stinnerf5cff562011-10-14 02:13:11 +02006056 *p++ = Py_hexdigits[(ch >> 12) & 0xf];
6057 *p++ = Py_hexdigits[(ch >> 8) & 0xf];
6058 *p++ = Py_hexdigits[(ch >> 4) & 0xf];
6059 *p++ = Py_hexdigits[ch & 15];
Guido van Rossumd57fd912000-03-10 22:53:23 +00006060 }
Benjamin Peterson29060642009-01-31 22:14:21 +00006061 /* Copy everything else as-is */
6062 else
Guido van Rossumd57fd912000-03-10 22:53:23 +00006063 *p++ = (char) ch;
6064 }
Guido van Rossum98297ee2007-11-06 21:34:58 +00006065
Martin v. Löwis1db7c132011-11-10 18:24:32 +01006066 assert(p > q);
6067 if (_PyBytes_Resize(&repr, p - q) < 0)
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00006068 return NULL;
6069 return repr;
Guido van Rossumd57fd912000-03-10 22:53:23 +00006070}
6071
Alexander Belopolsky40018472011-02-26 01:02:56 +00006072PyObject *
Martin v. Löwis1db7c132011-11-10 18:24:32 +01006073PyUnicode_EncodeRawUnicodeEscape(const Py_UNICODE *s,
6074 Py_ssize_t size)
Guido van Rossumd57fd912000-03-10 22:53:23 +00006075{
Martin v. Löwis1db7c132011-11-10 18:24:32 +01006076 PyObject *result;
6077 PyObject *tmp = PyUnicode_FromUnicode(s, size);
6078 if (tmp == NULL)
Walter Dörwald711005d2007-05-12 12:03:26 +00006079 return NULL;
Martin v. Löwis1db7c132011-11-10 18:24:32 +01006080 result = PyUnicode_AsRawUnicodeEscapeString(tmp);
6081 Py_DECREF(tmp);
6082 return result;
Guido van Rossumd57fd912000-03-10 22:53:23 +00006083}
6084
Walter Dörwalda47d1c02005-08-30 10:23:14 +00006085/* --- Unicode Internal Codec ------------------------------------------- */
6086
Alexander Belopolsky40018472011-02-26 01:02:56 +00006087PyObject *
6088_PyUnicode_DecodeUnicodeInternal(const char *s,
Ezio Melotti2aa2b3b2011-09-29 00:58:57 +03006089 Py_ssize_t size,
6090 const char *errors)
Walter Dörwalda47d1c02005-08-30 10:23:14 +00006091{
6092 const char *starts = s;
Martin v. Löwis18e16552006-02-15 17:27:45 +00006093 Py_ssize_t startinpos;
6094 Py_ssize_t endinpos;
6095 Py_ssize_t outpos;
Victor Stinner7931d9a2011-11-04 00:22:48 +01006096 PyObject *v;
Walter Dörwalda47d1c02005-08-30 10:23:14 +00006097 const char *end;
6098 const char *reason;
6099 PyObject *errorHandler = NULL;
6100 PyObject *exc = NULL;
6101
Victor Stinner9f4b1e92011-11-10 20:56:30 +01006102 if (PyErr_WarnEx(PyExc_DeprecationWarning,
Ezio Melotti11060a42011-11-16 09:39:10 +02006103 "unicode_internal codec has been deprecated",
Victor Stinner9f4b1e92011-11-10 20:56:30 +01006104 1))
6105 return NULL;
6106
Thomas Wouters89f507f2006-12-13 04:49:30 +00006107 /* XXX overflow detection missing */
Martin v. Löwise9b11c12011-11-08 17:35:34 +01006108 v = PyUnicode_New((size+Py_UNICODE_SIZE-1)/ Py_UNICODE_SIZE, 127);
Walter Dörwalda47d1c02005-08-30 10:23:14 +00006109 if (v == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00006110 goto onError;
Martin v. Löwise9b11c12011-11-08 17:35:34 +01006111 if (PyUnicode_GET_LENGTH(v) == 0)
Victor Stinner7931d9a2011-11-04 00:22:48 +01006112 return v;
Martin v. Löwise9b11c12011-11-08 17:35:34 +01006113 outpos = 0;
Walter Dörwalda47d1c02005-08-30 10:23:14 +00006114 end = s + size;
6115
6116 while (s < end) {
Antoine Pitrou0290c7a2011-11-11 13:29:12 +01006117 Py_UNICODE uch;
Antoine Pitrou44c6aff2011-11-11 02:59:42 +01006118 Py_UCS4 ch;
Serhiy Storchaka03ee12e2013-02-07 16:25:25 +02006119 if (end - s < Py_UNICODE_SIZE) {
Serhiy Storchaka3fd4ab32013-02-07 16:23:21 +02006120 endinpos = end-starts;
6121 reason = "truncated input";
6122 goto error;
6123 }
Antoine Pitrou44c6aff2011-11-11 02:59:42 +01006124 /* We copy the raw representation one byte at a time because the
6125 pointer may be unaligned (see test_codeccallbacks). */
Antoine Pitrou0290c7a2011-11-11 13:29:12 +01006126 ((char *) &uch)[0] = s[0];
6127 ((char *) &uch)[1] = s[1];
Antoine Pitrou44c6aff2011-11-11 02:59:42 +01006128#ifdef Py_UNICODE_WIDE
Antoine Pitrou0290c7a2011-11-11 13:29:12 +01006129 ((char *) &uch)[2] = s[2];
6130 ((char *) &uch)[3] = s[3];
Antoine Pitrou44c6aff2011-11-11 02:59:42 +01006131#endif
Antoine Pitrou0290c7a2011-11-11 13:29:12 +01006132 ch = uch;
Serhiy Storchaka3fd4ab32013-02-07 16:23:21 +02006133#ifdef Py_UNICODE_WIDE
Walter Dörwalda47d1c02005-08-30 10:23:14 +00006134 /* We have to sanity check the raw data, otherwise doom looms for
6135 some malformed UCS-4 data. */
Serhiy Storchaka03ee12e2013-02-07 16:25:25 +02006136 if (ch > 0x10ffff) {
Serhiy Storchaka3fd4ab32013-02-07 16:23:21 +02006137 endinpos = s - starts + Py_UNICODE_SIZE;
6138 reason = "illegal code point (> 0x10FFFF)";
6139 goto error;
Victor Stinner9f4b1e92011-11-10 20:56:30 +01006140 }
Serhiy Storchaka3fd4ab32013-02-07 16:23:21 +02006141#endif
Victor Stinner9f4b1e92011-11-10 20:56:30 +01006142 s += Py_UNICODE_SIZE;
6143#ifndef Py_UNICODE_WIDE
Serhiy Storchaka03ee12e2013-02-07 16:25:25 +02006144 if (Py_UNICODE_IS_HIGH_SURROGATE(ch) && end - s >= Py_UNICODE_SIZE)
Victor Stinner9f4b1e92011-11-10 20:56:30 +01006145 {
Antoine Pitrou0290c7a2011-11-11 13:29:12 +01006146 Py_UNICODE uch2;
6147 ((char *) &uch2)[0] = s[0];
6148 ((char *) &uch2)[1] = s[1];
Victor Stinner551ac952011-11-29 22:58:13 +01006149 if (Py_UNICODE_IS_LOW_SURROGATE(uch2))
Victor Stinner9f4b1e92011-11-10 20:56:30 +01006150 {
Victor Stinner551ac952011-11-29 22:58:13 +01006151 ch = Py_UNICODE_JOIN_SURROGATES(uch, uch2);
Victor Stinner9f4b1e92011-11-10 20:56:30 +01006152 s += Py_UNICODE_SIZE;
Walter Dörwalda47d1c02005-08-30 10:23:14 +00006153 }
6154 }
Victor Stinner9f4b1e92011-11-10 20:56:30 +01006155#endif
6156
6157 if (unicode_putchar(&v, &outpos, ch) < 0)
6158 goto onError;
Serhiy Storchaka3fd4ab32013-02-07 16:23:21 +02006159 continue;
6160
6161 error:
6162 startinpos = s - starts;
Serhiy Storchaka3fd4ab32013-02-07 16:23:21 +02006163 if (unicode_decode_call_errorhandler(
6164 errors, &errorHandler,
6165 "unicode_internal", reason,
6166 &starts, &end, &startinpos, &endinpos, &exc, &s,
Serhiy Storchaka03ee12e2013-02-07 16:25:25 +02006167 &v, &outpos))
Serhiy Storchaka3fd4ab32013-02-07 16:23:21 +02006168 goto onError;
Walter Dörwalda47d1c02005-08-30 10:23:14 +00006169 }
6170
Victor Stinner16e6a802011-12-12 13:24:15 +01006171 if (unicode_resize(&v, outpos) < 0)
Walter Dörwalda47d1c02005-08-30 10:23:14 +00006172 goto onError;
6173 Py_XDECREF(errorHandler);
6174 Py_XDECREF(exc);
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01006175 return unicode_result(v);
Walter Dörwalda47d1c02005-08-30 10:23:14 +00006176
Benjamin Peterson29060642009-01-31 22:14:21 +00006177 onError:
Walter Dörwalda47d1c02005-08-30 10:23:14 +00006178 Py_XDECREF(v);
6179 Py_XDECREF(errorHandler);
6180 Py_XDECREF(exc);
6181 return NULL;
6182}
6183
Guido van Rossumd57fd912000-03-10 22:53:23 +00006184/* --- Latin-1 Codec ------------------------------------------------------ */
6185
Alexander Belopolsky40018472011-02-26 01:02:56 +00006186PyObject *
6187PyUnicode_DecodeLatin1(const char *s,
Ezio Melotti2aa2b3b2011-09-29 00:58:57 +03006188 Py_ssize_t size,
6189 const char *errors)
Guido van Rossumd57fd912000-03-10 22:53:23 +00006190{
Guido van Rossumd57fd912000-03-10 22:53:23 +00006191 /* Latin-1 is equivalent to the first 256 ordinals in Unicode. */
Victor Stinnere57b1c02011-09-28 22:20:48 +02006192 return _PyUnicode_FromUCS1((unsigned char*)s, size);
Guido van Rossumd57fd912000-03-10 22:53:23 +00006193}
6194
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006195/* create or adjust a UnicodeEncodeError */
Alexander Belopolsky40018472011-02-26 01:02:56 +00006196static void
6197make_encode_exception(PyObject **exceptionObject,
Ezio Melotti2aa2b3b2011-09-29 00:58:57 +03006198 const char *encoding,
Martin v. Löwis9e816682011-11-02 12:45:42 +01006199 PyObject *unicode,
6200 Py_ssize_t startpos, Py_ssize_t endpos,
6201 const char *reason)
6202{
6203 if (*exceptionObject == NULL) {
6204 *exceptionObject = PyObject_CallFunction(
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006205 PyExc_UnicodeEncodeError, "sOnns",
Martin v. Löwis9e816682011-11-02 12:45:42 +01006206 encoding, unicode, startpos, endpos, reason);
6207 }
6208 else {
6209 if (PyUnicodeEncodeError_SetStart(*exceptionObject, startpos))
6210 goto onError;
6211 if (PyUnicodeEncodeError_SetEnd(*exceptionObject, endpos))
6212 goto onError;
6213 if (PyUnicodeEncodeError_SetReason(*exceptionObject, reason))
6214 goto onError;
6215 return;
6216 onError:
6217 Py_DECREF(*exceptionObject);
6218 *exceptionObject = NULL;
6219 }
6220}
6221
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006222/* raises a UnicodeEncodeError */
Alexander Belopolsky40018472011-02-26 01:02:56 +00006223static void
6224raise_encode_exception(PyObject **exceptionObject,
Ezio Melotti2aa2b3b2011-09-29 00:58:57 +03006225 const char *encoding,
Martin v. Löwis9e816682011-11-02 12:45:42 +01006226 PyObject *unicode,
6227 Py_ssize_t startpos, Py_ssize_t endpos,
6228 const char *reason)
6229{
Martin v. Löwis12be46c2011-11-04 19:04:15 +01006230 make_encode_exception(exceptionObject,
Martin v. Löwis9e816682011-11-02 12:45:42 +01006231 encoding, unicode, startpos, endpos, reason);
6232 if (*exceptionObject != NULL)
6233 PyCodec_StrictErrors(*exceptionObject);
6234}
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006235
6236/* error handling callback helper:
6237 build arguments, call the callback and check the arguments,
6238 put the result into newpos and return the replacement string, which
6239 has to be freed by the caller */
Alexander Belopolsky40018472011-02-26 01:02:56 +00006240static PyObject *
6241unicode_encode_call_errorhandler(const char *errors,
Ezio Melotti2aa2b3b2011-09-29 00:58:57 +03006242 PyObject **errorHandler,
6243 const char *encoding, const char *reason,
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006244 PyObject *unicode, PyObject **exceptionObject,
Ezio Melotti2aa2b3b2011-09-29 00:58:57 +03006245 Py_ssize_t startpos, Py_ssize_t endpos,
6246 Py_ssize_t *newpos)
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006247{
Martin v. Löwisdb12d452009-05-02 18:52:14 +00006248 static char *argparse = "On;encoding error handler must return (str/bytes, int) tuple";
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006249 Py_ssize_t len;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006250 PyObject *restuple;
6251 PyObject *resunicode;
6252
6253 if (*errorHandler == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +00006254 *errorHandler = PyCodec_LookupError(errors);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006255 if (*errorHandler == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00006256 return NULL;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006257 }
6258
Benjamin Petersonbac79492012-01-14 13:34:47 -05006259 if (PyUnicode_READY(unicode) == -1)
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006260 return NULL;
6261 len = PyUnicode_GET_LENGTH(unicode);
6262
Martin v. Löwis12be46c2011-11-04 19:04:15 +01006263 make_encode_exception(exceptionObject,
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006264 encoding, unicode, startpos, endpos, reason);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006265 if (*exceptionObject == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00006266 return NULL;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006267
6268 restuple = PyObject_CallFunctionObjArgs(
Benjamin Peterson29060642009-01-31 22:14:21 +00006269 *errorHandler, *exceptionObject, NULL);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006270 if (restuple == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00006271 return NULL;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006272 if (!PyTuple_Check(restuple)) {
Martin v. Löwisdb12d452009-05-02 18:52:14 +00006273 PyErr_SetString(PyExc_TypeError, &argparse[3]);
Benjamin Peterson29060642009-01-31 22:14:21 +00006274 Py_DECREF(restuple);
6275 return NULL;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006276 }
Martin v. Löwisdb12d452009-05-02 18:52:14 +00006277 if (!PyArg_ParseTuple(restuple, argparse,
Benjamin Peterson29060642009-01-31 22:14:21 +00006278 &resunicode, newpos)) {
6279 Py_DECREF(restuple);
6280 return NULL;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006281 }
Martin v. Löwisdb12d452009-05-02 18:52:14 +00006282 if (!PyUnicode_Check(resunicode) && !PyBytes_Check(resunicode)) {
6283 PyErr_SetString(PyExc_TypeError, &argparse[3]);
6284 Py_DECREF(restuple);
6285 return NULL;
6286 }
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006287 if (*newpos<0)
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006288 *newpos = len + *newpos;
6289 if (*newpos<0 || *newpos>len) {
Benjamin Peterson29060642009-01-31 22:14:21 +00006290 PyErr_Format(PyExc_IndexError, "position %zd from error handler out of bounds", *newpos);
6291 Py_DECREF(restuple);
6292 return NULL;
Walter Dörwald2e0b18a2003-01-31 17:19:08 +00006293 }
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006294 Py_INCREF(resunicode);
6295 Py_DECREF(restuple);
6296 return resunicode;
6297}
6298
Alexander Belopolsky40018472011-02-26 01:02:56 +00006299static PyObject *
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006300unicode_encode_ucs1(PyObject *unicode,
Ezio Melotti2aa2b3b2011-09-29 00:58:57 +03006301 const char *errors,
Victor Stinnerfcd96532011-11-04 00:28:50 +01006302 unsigned int limit)
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006303{
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006304 /* input state */
6305 Py_ssize_t pos=0, size;
6306 int kind;
6307 void *data;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006308 /* output object */
6309 PyObject *res;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006310 /* pointer into the output */
6311 char *str;
6312 /* current output position */
Martin v. Löwis18e16552006-02-15 17:27:45 +00006313 Py_ssize_t ressize;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006314 const char *encoding = (limit == 256) ? "latin-1" : "ascii";
6315 const char *reason = (limit == 256) ? "ordinal not in range(256)" : "ordinal not in range(128)";
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006316 PyObject *errorHandler = NULL;
6317 PyObject *exc = NULL;
6318 /* the following variable is used for caching string comparisons
6319 * -1=not initialized, 0=unknown, 1=strict, 2=replace, 3=ignore, 4=xmlcharrefreplace */
6320 int known_errorHandler = -1;
6321
Benjamin Petersonbac79492012-01-14 13:34:47 -05006322 if (PyUnicode_READY(unicode) == -1)
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006323 return NULL;
6324 size = PyUnicode_GET_LENGTH(unicode);
6325 kind = PyUnicode_KIND(unicode);
6326 data = PyUnicode_DATA(unicode);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006327 /* allocate enough for a simple encoding without
6328 replacements, if we need more, we'll resize */
Guido van Rossum98297ee2007-11-06 21:34:58 +00006329 if (size == 0)
Christian Heimes72b710a2008-05-26 13:28:38 +00006330 return PyBytes_FromStringAndSize(NULL, 0);
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00006331 res = PyBytes_FromStringAndSize(NULL, size);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006332 if (res == NULL)
Guido van Rossum98297ee2007-11-06 21:34:58 +00006333 return NULL;
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00006334 str = PyBytes_AS_STRING(res);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006335 ressize = size;
6336
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006337 while (pos < size) {
6338 Py_UCS4 c = PyUnicode_READ(kind, data, pos);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006339
Benjamin Peterson29060642009-01-31 22:14:21 +00006340 /* can we encode this? */
6341 if (c<limit) {
6342 /* no overflow check, because we know that the space is enough */
6343 *str++ = (char)c;
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006344 ++pos;
Benjamin Peterson14339b62009-01-31 16:36:08 +00006345 }
Benjamin Peterson29060642009-01-31 22:14:21 +00006346 else {
Benjamin Peterson29060642009-01-31 22:14:21 +00006347 Py_ssize_t requiredsize;
6348 PyObject *repunicode;
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006349 Py_ssize_t repsize, newpos, respos, i;
Benjamin Peterson29060642009-01-31 22:14:21 +00006350 /* startpos for collecting unencodable chars */
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006351 Py_ssize_t collstart = pos;
6352 Py_ssize_t collend = pos;
Benjamin Peterson29060642009-01-31 22:14:21 +00006353 /* find all unecodable characters */
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006354 while ((collend < size) && (PyUnicode_READ(kind, data, collend)>=limit))
Benjamin Peterson29060642009-01-31 22:14:21 +00006355 ++collend;
6356 /* cache callback name lookup (if not done yet, i.e. it's the first error) */
6357 if (known_errorHandler==-1) {
6358 if ((errors==NULL) || (!strcmp(errors, "strict")))
6359 known_errorHandler = 1;
6360 else if (!strcmp(errors, "replace"))
6361 known_errorHandler = 2;
6362 else if (!strcmp(errors, "ignore"))
6363 known_errorHandler = 3;
6364 else if (!strcmp(errors, "xmlcharrefreplace"))
6365 known_errorHandler = 4;
6366 else
6367 known_errorHandler = 0;
6368 }
6369 switch (known_errorHandler) {
6370 case 1: /* strict */
Martin v. Löwis12be46c2011-11-04 19:04:15 +01006371 raise_encode_exception(&exc, encoding, unicode, collstart, collend, reason);
Benjamin Peterson29060642009-01-31 22:14:21 +00006372 goto onError;
6373 case 2: /* replace */
6374 while (collstart++<collend)
6375 *str++ = '?'; /* fall through */
6376 case 3: /* ignore */
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006377 pos = collend;
Benjamin Peterson29060642009-01-31 22:14:21 +00006378 break;
6379 case 4: /* xmlcharrefreplace */
6380 respos = str - PyBytes_AS_STRING(res);
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006381 /* determine replacement size */
6382 for (i = collstart, repsize = 0; i < collend; ++i) {
6383 Py_UCS4 ch = PyUnicode_READ(kind, data, i);
6384 if (ch < 10)
Benjamin Peterson29060642009-01-31 22:14:21 +00006385 repsize += 2+1+1;
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006386 else if (ch < 100)
Benjamin Peterson29060642009-01-31 22:14:21 +00006387 repsize += 2+2+1;
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006388 else if (ch < 1000)
Benjamin Peterson29060642009-01-31 22:14:21 +00006389 repsize += 2+3+1;
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006390 else if (ch < 10000)
Benjamin Peterson29060642009-01-31 22:14:21 +00006391 repsize += 2+4+1;
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006392 else if (ch < 100000)
Benjamin Peterson29060642009-01-31 22:14:21 +00006393 repsize += 2+5+1;
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006394 else if (ch < 1000000)
Benjamin Peterson29060642009-01-31 22:14:21 +00006395 repsize += 2+6+1;
Victor Stinner0d3721d2011-11-22 03:27:53 +01006396 else {
Victor Stinner8faf8212011-12-08 22:14:11 +01006397 assert(ch <= MAX_UNICODE);
Benjamin Peterson29060642009-01-31 22:14:21 +00006398 repsize += 2+7+1;
Victor Stinner0d3721d2011-11-22 03:27:53 +01006399 }
Benjamin Peterson29060642009-01-31 22:14:21 +00006400 }
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006401 requiredsize = respos+repsize+(size-collend);
Benjamin Peterson29060642009-01-31 22:14:21 +00006402 if (requiredsize > ressize) {
6403 if (requiredsize<2*ressize)
6404 requiredsize = 2*ressize;
6405 if (_PyBytes_Resize(&res, requiredsize))
6406 goto onError;
6407 str = PyBytes_AS_STRING(res) + respos;
6408 ressize = requiredsize;
6409 }
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006410 /* generate replacement */
6411 for (i = collstart; i < collend; ++i) {
6412 str += sprintf(str, "&#%d;", PyUnicode_READ(kind, data, i));
Benjamin Peterson29060642009-01-31 22:14:21 +00006413 }
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006414 pos = collend;
Benjamin Peterson29060642009-01-31 22:14:21 +00006415 break;
6416 default:
6417 repunicode = unicode_encode_call_errorhandler(errors, &errorHandler,
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006418 encoding, reason, unicode, &exc,
6419 collstart, collend, &newpos);
6420 if (repunicode == NULL || (PyUnicode_Check(repunicode) &&
Benjamin Petersonbac79492012-01-14 13:34:47 -05006421 PyUnicode_READY(repunicode) == -1))
Benjamin Peterson29060642009-01-31 22:14:21 +00006422 goto onError;
Martin v. Löwis011e8422009-05-05 04:43:17 +00006423 if (PyBytes_Check(repunicode)) {
6424 /* Directly copy bytes result to output. */
6425 repsize = PyBytes_Size(repunicode);
6426 if (repsize > 1) {
6427 /* Make room for all additional bytes. */
Amaury Forgeot d'Arc84ec8d92009-06-29 22:36:49 +00006428 respos = str - PyBytes_AS_STRING(res);
Martin v. Löwis011e8422009-05-05 04:43:17 +00006429 if (_PyBytes_Resize(&res, ressize+repsize-1)) {
6430 Py_DECREF(repunicode);
6431 goto onError;
6432 }
Amaury Forgeot d'Arc84ec8d92009-06-29 22:36:49 +00006433 str = PyBytes_AS_STRING(res) + respos;
Martin v. Löwis011e8422009-05-05 04:43:17 +00006434 ressize += repsize-1;
6435 }
6436 memcpy(str, PyBytes_AsString(repunicode), repsize);
6437 str += repsize;
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006438 pos = newpos;
Martin v. Löwisdb12d452009-05-02 18:52:14 +00006439 Py_DECREF(repunicode);
Martin v. Löwis011e8422009-05-05 04:43:17 +00006440 break;
Martin v. Löwisdb12d452009-05-02 18:52:14 +00006441 }
Benjamin Peterson29060642009-01-31 22:14:21 +00006442 /* need more space? (at least enough for what we
6443 have+the replacement+the rest of the string, so
6444 we won't have to check space for encodable characters) */
6445 respos = str - PyBytes_AS_STRING(res);
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006446 repsize = PyUnicode_GET_LENGTH(repunicode);
6447 requiredsize = respos+repsize+(size-collend);
Benjamin Peterson29060642009-01-31 22:14:21 +00006448 if (requiredsize > ressize) {
6449 if (requiredsize<2*ressize)
6450 requiredsize = 2*ressize;
6451 if (_PyBytes_Resize(&res, requiredsize)) {
6452 Py_DECREF(repunicode);
6453 goto onError;
6454 }
6455 str = PyBytes_AS_STRING(res) + respos;
6456 ressize = requiredsize;
6457 }
6458 /* check if there is anything unencodable in the replacement
6459 and copy it to the output */
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006460 for (i = 0; repsize-->0; ++i, ++str) {
6461 c = PyUnicode_READ_CHAR(repunicode, i);
Benjamin Peterson29060642009-01-31 22:14:21 +00006462 if (c >= limit) {
Martin v. Löwis12be46c2011-11-04 19:04:15 +01006463 raise_encode_exception(&exc, encoding, unicode,
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006464 pos, pos+1, reason);
Benjamin Peterson29060642009-01-31 22:14:21 +00006465 Py_DECREF(repunicode);
6466 goto onError;
6467 }
6468 *str = (char)c;
6469 }
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006470 pos = newpos;
Benjamin Peterson14339b62009-01-31 16:36:08 +00006471 Py_DECREF(repunicode);
Benjamin Peterson14339b62009-01-31 16:36:08 +00006472 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00006473 }
6474 }
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00006475 /* Resize if we allocated to much */
6476 size = str - PyBytes_AS_STRING(res);
6477 if (size < ressize) { /* If this falls res will be NULL */
Alexandre Vassalottibad1b922008-12-27 09:49:09 +00006478 assert(size >= 0);
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00006479 if (_PyBytes_Resize(&res, size) < 0)
6480 goto onError;
6481 }
6482
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006483 Py_XDECREF(errorHandler);
6484 Py_XDECREF(exc);
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00006485 return res;
6486
6487 onError:
6488 Py_XDECREF(res);
6489 Py_XDECREF(errorHandler);
6490 Py_XDECREF(exc);
6491 return NULL;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006492}
6493
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006494/* Deprecated */
Alexander Belopolsky40018472011-02-26 01:02:56 +00006495PyObject *
6496PyUnicode_EncodeLatin1(const Py_UNICODE *p,
Ezio Melotti2aa2b3b2011-09-29 00:58:57 +03006497 Py_ssize_t size,
6498 const char *errors)
Guido van Rossumd57fd912000-03-10 22:53:23 +00006499{
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006500 PyObject *result;
6501 PyObject *unicode = PyUnicode_FromUnicode(p, size);
6502 if (unicode == NULL)
6503 return NULL;
6504 result = unicode_encode_ucs1(unicode, errors, 256);
6505 Py_DECREF(unicode);
6506 return result;
Guido van Rossumd57fd912000-03-10 22:53:23 +00006507}
6508
Alexander Belopolsky40018472011-02-26 01:02:56 +00006509PyObject *
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02006510_PyUnicode_AsLatin1String(PyObject *unicode, const char *errors)
Guido van Rossumd57fd912000-03-10 22:53:23 +00006511{
6512 if (!PyUnicode_Check(unicode)) {
Benjamin Peterson29060642009-01-31 22:14:21 +00006513 PyErr_BadArgument();
6514 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00006515 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02006516 if (PyUnicode_READY(unicode) == -1)
6517 return NULL;
6518 /* Fast path: if it is a one-byte string, construct
6519 bytes object directly. */
6520 if (PyUnicode_KIND(unicode) == PyUnicode_1BYTE_KIND)
6521 return PyBytes_FromStringAndSize(PyUnicode_DATA(unicode),
6522 PyUnicode_GET_LENGTH(unicode));
6523 /* Non-Latin-1 characters present. Defer to above function to
6524 raise the exception. */
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006525 return unicode_encode_ucs1(unicode, errors, 256);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02006526}
6527
6528PyObject*
6529PyUnicode_AsLatin1String(PyObject *unicode)
6530{
6531 return _PyUnicode_AsLatin1String(unicode, NULL);
Guido van Rossumd57fd912000-03-10 22:53:23 +00006532}
6533
6534/* --- 7-bit ASCII Codec -------------------------------------------------- */
6535
Alexander Belopolsky40018472011-02-26 01:02:56 +00006536PyObject *
6537PyUnicode_DecodeASCII(const char *s,
6538 Py_ssize_t size,
6539 const char *errors)
Guido van Rossumd57fd912000-03-10 22:53:23 +00006540{
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006541 const char *starts = s;
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02006542 PyObject *unicode;
Martin v. Löwise9b11c12011-11-08 17:35:34 +01006543 int kind;
6544 void *data;
Martin v. Löwis18e16552006-02-15 17:27:45 +00006545 Py_ssize_t startinpos;
6546 Py_ssize_t endinpos;
6547 Py_ssize_t outpos;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006548 const char *e;
6549 PyObject *errorHandler = NULL;
6550 PyObject *exc = NULL;
Tim Petersced69f82003-09-16 20:30:58 +00006551
Guido van Rossumd57fd912000-03-10 22:53:23 +00006552 if (size == 0)
Serhiy Storchaka678db842013-01-26 12:16:36 +02006553 _Py_RETURN_UNICODE_EMPTY();
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01006554
Guido van Rossumd57fd912000-03-10 22:53:23 +00006555 /* ASCII is equivalent to the first 128 ordinals in Unicode. */
Victor Stinner702c7342011-10-05 13:50:52 +02006556 if (size == 1 && (unsigned char)s[0] < 128)
6557 return get_latin1_char((unsigned char)s[0]);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02006558
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02006559 unicode = PyUnicode_New(size, 127);
6560 if (unicode == NULL)
Guido van Rossumd57fd912000-03-10 22:53:23 +00006561 goto onError;
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02006562
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006563 e = s + size;
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02006564 data = PyUnicode_1BYTE_DATA(unicode);
6565 outpos = ascii_decode(s, e, (Py_UCS1 *)data);
6566 if (outpos == size)
6567 return unicode;
6568
6569 s += outpos;
6570 kind = PyUnicode_1BYTE_KIND;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006571 while (s < e) {
Benjamin Peterson29060642009-01-31 22:14:21 +00006572 register unsigned char c = (unsigned char)*s;
6573 if (c < 128) {
Martin v. Löwise9b11c12011-11-08 17:35:34 +01006574 PyUnicode_WRITE(kind, data, outpos++, c);
Benjamin Peterson29060642009-01-31 22:14:21 +00006575 ++s;
6576 }
6577 else {
6578 startinpos = s-starts;
6579 endinpos = startinpos + 1;
Benjamin Peterson29060642009-01-31 22:14:21 +00006580 if (unicode_decode_call_errorhandler(
6581 errors, &errorHandler,
6582 "ascii", "ordinal not in range(128)",
6583 &starts, &e, &startinpos, &endinpos, &exc, &s,
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02006584 &unicode, &outpos))
Benjamin Peterson29060642009-01-31 22:14:21 +00006585 goto onError;
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02006586 kind = PyUnicode_KIND(unicode);
6587 data = PyUnicode_DATA(unicode);
Benjamin Peterson29060642009-01-31 22:14:21 +00006588 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00006589 }
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02006590 if (unicode_resize(&unicode, outpos) < 0)
Martin v. Löwise9b11c12011-11-08 17:35:34 +01006591 goto onError;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006592 Py_XDECREF(errorHandler);
6593 Py_XDECREF(exc);
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02006594 assert(_PyUnicode_CheckConsistency(unicode, 1));
6595 return unicode;
Tim Petersced69f82003-09-16 20:30:58 +00006596
Benjamin Peterson29060642009-01-31 22:14:21 +00006597 onError:
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02006598 Py_XDECREF(unicode);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006599 Py_XDECREF(errorHandler);
6600 Py_XDECREF(exc);
Guido van Rossumd57fd912000-03-10 22:53:23 +00006601 return NULL;
6602}
6603
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006604/* Deprecated */
Alexander Belopolsky40018472011-02-26 01:02:56 +00006605PyObject *
6606PyUnicode_EncodeASCII(const Py_UNICODE *p,
6607 Py_ssize_t size,
6608 const char *errors)
Guido van Rossumd57fd912000-03-10 22:53:23 +00006609{
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006610 PyObject *result;
6611 PyObject *unicode = PyUnicode_FromUnicode(p, size);
6612 if (unicode == NULL)
6613 return NULL;
6614 result = unicode_encode_ucs1(unicode, errors, 128);
6615 Py_DECREF(unicode);
6616 return result;
Guido van Rossumd57fd912000-03-10 22:53:23 +00006617}
6618
Alexander Belopolsky40018472011-02-26 01:02:56 +00006619PyObject *
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02006620_PyUnicode_AsASCIIString(PyObject *unicode, const char *errors)
Guido van Rossumd57fd912000-03-10 22:53:23 +00006621{
6622 if (!PyUnicode_Check(unicode)) {
Benjamin Peterson29060642009-01-31 22:14:21 +00006623 PyErr_BadArgument();
6624 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00006625 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02006626 if (PyUnicode_READY(unicode) == -1)
6627 return NULL;
6628 /* Fast path: if it is an ASCII-only string, construct bytes object
6629 directly. Else defer to above function to raise the exception. */
6630 if (PyUnicode_MAX_CHAR_VALUE(unicode) < 128)
6631 return PyBytes_FromStringAndSize(PyUnicode_DATA(unicode),
6632 PyUnicode_GET_LENGTH(unicode));
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006633 return unicode_encode_ucs1(unicode, errors, 128);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02006634}
6635
6636PyObject *
6637PyUnicode_AsASCIIString(PyObject *unicode)
6638{
6639 return _PyUnicode_AsASCIIString(unicode, NULL);
Guido van Rossumd57fd912000-03-10 22:53:23 +00006640}
6641
Victor Stinner99b95382011-07-04 14:23:54 +02006642#ifdef HAVE_MBCS
Guido van Rossum2ea3e142000-03-31 17:24:09 +00006643
Guido van Rossumb7a40ba2000-03-28 02:01:52 +00006644/* --- MBCS codecs for Windows -------------------------------------------- */
Guido van Rossum2ea3e142000-03-31 17:24:09 +00006645
Hirokazu Yamamoto35302462009-03-21 13:23:27 +00006646#if SIZEOF_INT < SIZEOF_SIZE_T
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006647#define NEED_RETRY
6648#endif
6649
Victor Stinner3a50e702011-10-18 21:21:00 +02006650#ifndef WC_ERR_INVALID_CHARS
6651# define WC_ERR_INVALID_CHARS 0x0080
6652#endif
6653
6654static char*
6655code_page_name(UINT code_page, PyObject **obj)
6656{
6657 *obj = NULL;
6658 if (code_page == CP_ACP)
6659 return "mbcs";
6660 if (code_page == CP_UTF7)
6661 return "CP_UTF7";
6662 if (code_page == CP_UTF8)
6663 return "CP_UTF8";
6664
6665 *obj = PyBytes_FromFormat("cp%u", code_page);
6666 if (*obj == NULL)
6667 return NULL;
6668 return PyBytes_AS_STRING(*obj);
6669}
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006670
Alexander Belopolsky40018472011-02-26 01:02:56 +00006671static int
Victor Stinner3a50e702011-10-18 21:21:00 +02006672is_dbcs_lead_byte(UINT code_page, const char *s, int offset)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006673{
6674 const char *curr = s + offset;
Victor Stinner3a50e702011-10-18 21:21:00 +02006675 const char *prev;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006676
Victor Stinner3a50e702011-10-18 21:21:00 +02006677 if (!IsDBCSLeadByteEx(code_page, *curr))
6678 return 0;
6679
6680 prev = CharPrevExA(code_page, s, curr, 0);
6681 if (prev == curr)
6682 return 1;
6683 /* FIXME: This code is limited to "true" double-byte encodings,
6684 as it assumes an incomplete character consists of a single
6685 byte. */
6686 if (curr - prev == 2)
6687 return 1;
6688 if (!IsDBCSLeadByteEx(code_page, *prev))
6689 return 1;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006690 return 0;
6691}
6692
Victor Stinner3a50e702011-10-18 21:21:00 +02006693static DWORD
6694decode_code_page_flags(UINT code_page)
6695{
6696 if (code_page == CP_UTF7) {
6697 /* The CP_UTF7 decoder only supports flags=0 */
6698 return 0;
6699 }
6700 else
6701 return MB_ERR_INVALID_CHARS;
6702}
6703
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006704/*
Victor Stinner3a50e702011-10-18 21:21:00 +02006705 * Decode a byte string from a Windows code page into unicode object in strict
6706 * mode.
6707 *
6708 * Returns consumed size if succeed, returns -2 on decode error, or raise a
6709 * WindowsError and returns -1 on other error.
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006710 */
Alexander Belopolsky40018472011-02-26 01:02:56 +00006711static int
Victor Stinner3a50e702011-10-18 21:21:00 +02006712decode_code_page_strict(UINT code_page,
Victor Stinner76a31a62011-11-04 00:05:13 +01006713 PyObject **v,
Victor Stinner3a50e702011-10-18 21:21:00 +02006714 const char *in,
6715 int insize)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006716{
Victor Stinner3a50e702011-10-18 21:21:00 +02006717 const DWORD flags = decode_code_page_flags(code_page);
Victor Stinner24729f32011-11-10 20:31:37 +01006718 wchar_t *out;
Victor Stinner3a50e702011-10-18 21:21:00 +02006719 DWORD outsize;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006720
6721 /* First get the size of the result */
Victor Stinner3a50e702011-10-18 21:21:00 +02006722 assert(insize > 0);
6723 outsize = MultiByteToWideChar(code_page, flags, in, insize, NULL, 0);
6724 if (outsize <= 0)
6725 goto error;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006726
6727 if (*v == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +00006728 /* Create unicode object */
Victor Stinnerab595942011-12-17 04:59:06 +01006729 /* FIXME: don't use _PyUnicode_New(), but allocate a wchar_t* buffer */
Victor Stinner76a31a62011-11-04 00:05:13 +01006730 *v = (PyObject*)_PyUnicode_New(outsize);
Benjamin Peterson29060642009-01-31 22:14:21 +00006731 if (*v == NULL)
6732 return -1;
Victor Stinner3a50e702011-10-18 21:21:00 +02006733 out = PyUnicode_AS_UNICODE(*v);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006734 }
6735 else {
Benjamin Peterson29060642009-01-31 22:14:21 +00006736 /* Extend unicode object */
Victor Stinner3a50e702011-10-18 21:21:00 +02006737 Py_ssize_t n = PyUnicode_GET_SIZE(*v);
Victor Stinner16e6a802011-12-12 13:24:15 +01006738 if (unicode_resize(v, n + outsize) < 0)
Benjamin Peterson29060642009-01-31 22:14:21 +00006739 return -1;
Victor Stinner3a50e702011-10-18 21:21:00 +02006740 out = PyUnicode_AS_UNICODE(*v) + n;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006741 }
6742
6743 /* Do the conversion */
Victor Stinner3a50e702011-10-18 21:21:00 +02006744 outsize = MultiByteToWideChar(code_page, flags, in, insize, out, outsize);
6745 if (outsize <= 0)
6746 goto error;
6747 return insize;
Victor Stinner554f3f02010-06-16 23:33:54 +00006748
Victor Stinner3a50e702011-10-18 21:21:00 +02006749error:
6750 if (GetLastError() == ERROR_NO_UNICODE_TRANSLATION)
6751 return -2;
6752 PyErr_SetFromWindowsErr(0);
Victor Stinner554f3f02010-06-16 23:33:54 +00006753 return -1;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006754}
6755
Victor Stinner3a50e702011-10-18 21:21:00 +02006756/*
6757 * Decode a byte string from a code page into unicode object with an error
6758 * handler.
6759 *
6760 * Returns consumed size if succeed, or raise a WindowsError or
6761 * UnicodeDecodeError exception and returns -1 on error.
6762 */
6763static int
6764decode_code_page_errors(UINT code_page,
Victor Stinner76a31a62011-11-04 00:05:13 +01006765 PyObject **v,
6766 const char *in, const int size,
Victor Stinner3a50e702011-10-18 21:21:00 +02006767 const char *errors)
6768{
6769 const char *startin = in;
6770 const char *endin = in + size;
6771 const DWORD flags = decode_code_page_flags(code_page);
6772 /* Ideally, we should get reason from FormatMessage. This is the Windows
6773 2000 English version of the message. */
6774 const char *reason = "No mapping for the Unicode character exists "
6775 "in the target code page.";
6776 /* each step cannot decode more than 1 character, but a character can be
6777 represented as a surrogate pair */
6778 wchar_t buffer[2], *startout, *out;
6779 int insize, outsize;
6780 PyObject *errorHandler = NULL;
6781 PyObject *exc = NULL;
6782 PyObject *encoding_obj = NULL;
6783 char *encoding;
6784 DWORD err;
6785 int ret = -1;
6786
6787 assert(size > 0);
6788
6789 encoding = code_page_name(code_page, &encoding_obj);
6790 if (encoding == NULL)
6791 return -1;
6792
6793 if (errors == NULL || strcmp(errors, "strict") == 0) {
6794 /* The last error was ERROR_NO_UNICODE_TRANSLATION, then we raise a
6795 UnicodeDecodeError. */
6796 make_decode_exception(&exc, encoding, in, size, 0, 0, reason);
6797 if (exc != NULL) {
6798 PyCodec_StrictErrors(exc);
6799 Py_CLEAR(exc);
6800 }
6801 goto error;
6802 }
6803
6804 if (*v == NULL) {
6805 /* Create unicode object */
6806 if (size > PY_SSIZE_T_MAX / (Py_ssize_t)Py_ARRAY_LENGTH(buffer)) {
6807 PyErr_NoMemory();
6808 goto error;
6809 }
Victor Stinnerab595942011-12-17 04:59:06 +01006810 /* FIXME: don't use _PyUnicode_New(), but allocate a wchar_t* buffer */
Victor Stinner76a31a62011-11-04 00:05:13 +01006811 *v = (PyObject*)_PyUnicode_New(size * Py_ARRAY_LENGTH(buffer));
Victor Stinner3a50e702011-10-18 21:21:00 +02006812 if (*v == NULL)
6813 goto error;
6814 startout = PyUnicode_AS_UNICODE(*v);
6815 }
6816 else {
6817 /* Extend unicode object */
6818 Py_ssize_t n = PyUnicode_GET_SIZE(*v);
6819 if (size > (PY_SSIZE_T_MAX - n) / (Py_ssize_t)Py_ARRAY_LENGTH(buffer)) {
6820 PyErr_NoMemory();
6821 goto error;
6822 }
Victor Stinner16e6a802011-12-12 13:24:15 +01006823 if (unicode_resize(v, n + size * Py_ARRAY_LENGTH(buffer)) < 0)
Victor Stinner3a50e702011-10-18 21:21:00 +02006824 goto error;
6825 startout = PyUnicode_AS_UNICODE(*v) + n;
6826 }
6827
6828 /* Decode the byte string character per character */
6829 out = startout;
6830 while (in < endin)
6831 {
6832 /* Decode a character */
6833 insize = 1;
6834 do
6835 {
6836 outsize = MultiByteToWideChar(code_page, flags,
6837 in, insize,
6838 buffer, Py_ARRAY_LENGTH(buffer));
6839 if (outsize > 0)
6840 break;
6841 err = GetLastError();
6842 if (err != ERROR_NO_UNICODE_TRANSLATION
6843 && err != ERROR_INSUFFICIENT_BUFFER)
6844 {
6845 PyErr_SetFromWindowsErr(0);
6846 goto error;
6847 }
6848 insize++;
6849 }
6850 /* 4=maximum length of a UTF-8 sequence */
6851 while (insize <= 4 && (in + insize) <= endin);
6852
6853 if (outsize <= 0) {
6854 Py_ssize_t startinpos, endinpos, outpos;
6855
6856 startinpos = in - startin;
6857 endinpos = startinpos + 1;
6858 outpos = out - PyUnicode_AS_UNICODE(*v);
6859 if (unicode_decode_call_errorhandler(
6860 errors, &errorHandler,
6861 encoding, reason,
6862 &startin, &endin, &startinpos, &endinpos, &exc, &in,
Victor Stinner596a6c42011-11-09 00:02:18 +01006863 v, &outpos))
Victor Stinner3a50e702011-10-18 21:21:00 +02006864 {
6865 goto error;
6866 }
Victor Stinner596a6c42011-11-09 00:02:18 +01006867 out = PyUnicode_AS_UNICODE(*v) + outpos;
Victor Stinner3a50e702011-10-18 21:21:00 +02006868 }
6869 else {
6870 in += insize;
6871 memcpy(out, buffer, outsize * sizeof(wchar_t));
6872 out += outsize;
6873 }
6874 }
6875
6876 /* write a NUL character at the end */
6877 *out = 0;
6878
6879 /* Extend unicode object */
6880 outsize = out - startout;
6881 assert(outsize <= PyUnicode_WSTR_LENGTH(*v));
Victor Stinner16e6a802011-12-12 13:24:15 +01006882 if (unicode_resize(v, outsize) < 0)
Victor Stinner3a50e702011-10-18 21:21:00 +02006883 goto error;
Victor Stinner76a31a62011-11-04 00:05:13 +01006884 ret = size;
Victor Stinner3a50e702011-10-18 21:21:00 +02006885
6886error:
6887 Py_XDECREF(encoding_obj);
6888 Py_XDECREF(errorHandler);
6889 Py_XDECREF(exc);
6890 return ret;
6891}
6892
Victor Stinner3a50e702011-10-18 21:21:00 +02006893static PyObject *
6894decode_code_page_stateful(int code_page,
Victor Stinner76a31a62011-11-04 00:05:13 +01006895 const char *s, Py_ssize_t size,
6896 const char *errors, Py_ssize_t *consumed)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006897{
Victor Stinner76a31a62011-11-04 00:05:13 +01006898 PyObject *v = NULL;
6899 int chunk_size, final, converted, done;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006900
Victor Stinner3a50e702011-10-18 21:21:00 +02006901 if (code_page < 0) {
6902 PyErr_SetString(PyExc_ValueError, "invalid code page number");
6903 return NULL;
6904 }
6905
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006906 if (consumed)
Benjamin Peterson29060642009-01-31 22:14:21 +00006907 *consumed = 0;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006908
Victor Stinner76a31a62011-11-04 00:05:13 +01006909 do
6910 {
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006911#ifdef NEED_RETRY
Victor Stinner76a31a62011-11-04 00:05:13 +01006912 if (size > INT_MAX) {
6913 chunk_size = INT_MAX;
6914 final = 0;
6915 done = 0;
6916 }
6917 else
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006918#endif
Victor Stinner76a31a62011-11-04 00:05:13 +01006919 {
6920 chunk_size = (int)size;
6921 final = (consumed == NULL);
6922 done = 1;
6923 }
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006924
Victor Stinner76a31a62011-11-04 00:05:13 +01006925 /* Skip trailing lead-byte unless 'final' is set */
6926 if (!final && is_dbcs_lead_byte(code_page, s, chunk_size - 1))
6927 --chunk_size;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006928
Victor Stinner76a31a62011-11-04 00:05:13 +01006929 if (chunk_size == 0 && done) {
6930 if (v != NULL)
6931 break;
Serhiy Storchaka678db842013-01-26 12:16:36 +02006932 _Py_RETURN_UNICODE_EMPTY();
Victor Stinner76a31a62011-11-04 00:05:13 +01006933 }
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006934
Victor Stinner76a31a62011-11-04 00:05:13 +01006935
6936 converted = decode_code_page_strict(code_page, &v,
6937 s, chunk_size);
6938 if (converted == -2)
6939 converted = decode_code_page_errors(code_page, &v,
6940 s, chunk_size,
6941 errors);
6942 assert(converted != 0);
6943
6944 if (converted < 0) {
6945 Py_XDECREF(v);
6946 return NULL;
6947 }
6948
6949 if (consumed)
6950 *consumed += converted;
6951
6952 s += converted;
6953 size -= converted;
6954 } while (!done);
Victor Stinner3a50e702011-10-18 21:21:00 +02006955
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01006956 return unicode_result(v);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006957}
6958
Alexander Belopolsky40018472011-02-26 01:02:56 +00006959PyObject *
Victor Stinner3a50e702011-10-18 21:21:00 +02006960PyUnicode_DecodeCodePageStateful(int code_page,
6961 const char *s,
6962 Py_ssize_t size,
6963 const char *errors,
6964 Py_ssize_t *consumed)
6965{
6966 return decode_code_page_stateful(code_page, s, size, errors, consumed);
6967}
6968
6969PyObject *
6970PyUnicode_DecodeMBCSStateful(const char *s,
6971 Py_ssize_t size,
6972 const char *errors,
6973 Py_ssize_t *consumed)
6974{
6975 return decode_code_page_stateful(CP_ACP, s, size, errors, consumed);
6976}
6977
6978PyObject *
Alexander Belopolsky40018472011-02-26 01:02:56 +00006979PyUnicode_DecodeMBCS(const char *s,
6980 Py_ssize_t size,
6981 const char *errors)
Guido van Rossumb7a40ba2000-03-28 02:01:52 +00006982{
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006983 return PyUnicode_DecodeMBCSStateful(s, size, errors, NULL);
6984}
6985
Victor Stinner3a50e702011-10-18 21:21:00 +02006986static DWORD
6987encode_code_page_flags(UINT code_page, const char *errors)
6988{
6989 if (code_page == CP_UTF8) {
6990 if (winver.dwMajorVersion >= 6)
6991 /* CP_UTF8 supports WC_ERR_INVALID_CHARS on Windows Vista
6992 and later */
6993 return WC_ERR_INVALID_CHARS;
6994 else
6995 /* CP_UTF8 only supports flags=0 on Windows older than Vista */
6996 return 0;
6997 }
6998 else if (code_page == CP_UTF7) {
6999 /* CP_UTF7 only supports flags=0 */
7000 return 0;
7001 }
7002 else {
7003 if (errors != NULL && strcmp(errors, "replace") == 0)
7004 return 0;
7005 else
7006 return WC_NO_BEST_FIT_CHARS;
7007 }
7008}
7009
Thomas Wouters0e3f5912006-08-11 14:57:12 +00007010/*
Victor Stinner3a50e702011-10-18 21:21:00 +02007011 * Encode a Unicode string to a Windows code page into a byte string in strict
7012 * mode.
7013 *
7014 * Returns consumed characters if succeed, returns -2 on encode error, or raise
7015 * a WindowsError and returns -1 on other error.
Thomas Wouters0e3f5912006-08-11 14:57:12 +00007016 */
Alexander Belopolsky40018472011-02-26 01:02:56 +00007017static int
Victor Stinner3a50e702011-10-18 21:21:00 +02007018encode_code_page_strict(UINT code_page, PyObject **outbytes,
Martin v. Löwis3d325192011-11-04 18:23:06 +01007019 PyObject *unicode, Py_ssize_t offset, int len,
Victor Stinner3a50e702011-10-18 21:21:00 +02007020 const char* errors)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00007021{
Victor Stinner554f3f02010-06-16 23:33:54 +00007022 BOOL usedDefaultChar = FALSE;
Victor Stinner3a50e702011-10-18 21:21:00 +02007023 BOOL *pusedDefaultChar = &usedDefaultChar;
7024 int outsize;
Victor Stinner554f3f02010-06-16 23:33:54 +00007025 PyObject *exc = NULL;
Victor Stinner24729f32011-11-10 20:31:37 +01007026 wchar_t *p;
Victor Stinner2fc507f2011-11-04 20:06:39 +01007027 Py_ssize_t size;
Victor Stinner3a50e702011-10-18 21:21:00 +02007028 const DWORD flags = encode_code_page_flags(code_page, NULL);
7029 char *out;
Victor Stinner2fc507f2011-11-04 20:06:39 +01007030 /* Create a substring so that we can get the UTF-16 representation
7031 of just the slice under consideration. */
7032 PyObject *substring;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00007033
Martin v. Löwis3d325192011-11-04 18:23:06 +01007034 assert(len > 0);
Guido van Rossumb7a40ba2000-03-28 02:01:52 +00007035
Victor Stinner3a50e702011-10-18 21:21:00 +02007036 if (code_page != CP_UTF8 && code_page != CP_UTF7)
Victor Stinner554f3f02010-06-16 23:33:54 +00007037 pusedDefaultChar = &usedDefaultChar;
Victor Stinner3a50e702011-10-18 21:21:00 +02007038 else
Victor Stinner554f3f02010-06-16 23:33:54 +00007039 pusedDefaultChar = NULL;
Victor Stinner554f3f02010-06-16 23:33:54 +00007040
Victor Stinner2fc507f2011-11-04 20:06:39 +01007041 substring = PyUnicode_Substring(unicode, offset, offset+len);
7042 if (substring == NULL)
7043 return -1;
7044 p = PyUnicode_AsUnicodeAndSize(substring, &size);
7045 if (p == NULL) {
7046 Py_DECREF(substring);
7047 return -1;
7048 }
Martin v. Löwis3d325192011-11-04 18:23:06 +01007049
Guido van Rossumb7a40ba2000-03-28 02:01:52 +00007050 /* First get the size of the result */
Victor Stinner3a50e702011-10-18 21:21:00 +02007051 outsize = WideCharToMultiByte(code_page, flags,
7052 p, size,
7053 NULL, 0,
7054 NULL, pusedDefaultChar);
7055 if (outsize <= 0)
7056 goto error;
7057 /* If we used a default char, then we failed! */
Victor Stinner2fc507f2011-11-04 20:06:39 +01007058 if (pusedDefaultChar && *pusedDefaultChar) {
7059 Py_DECREF(substring);
Victor Stinner3a50e702011-10-18 21:21:00 +02007060 return -2;
Victor Stinner2fc507f2011-11-04 20:06:39 +01007061 }
Guido van Rossumb7a40ba2000-03-28 02:01:52 +00007062
Victor Stinner3a50e702011-10-18 21:21:00 +02007063 if (*outbytes == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +00007064 /* Create string object */
Victor Stinner3a50e702011-10-18 21:21:00 +02007065 *outbytes = PyBytes_FromStringAndSize(NULL, outsize);
Victor Stinner2fc507f2011-11-04 20:06:39 +01007066 if (*outbytes == NULL) {
7067 Py_DECREF(substring);
Benjamin Peterson29060642009-01-31 22:14:21 +00007068 return -1;
Victor Stinner2fc507f2011-11-04 20:06:39 +01007069 }
Victor Stinner3a50e702011-10-18 21:21:00 +02007070 out = PyBytes_AS_STRING(*outbytes);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00007071 }
7072 else {
Benjamin Peterson29060642009-01-31 22:14:21 +00007073 /* Extend string object */
Victor Stinner3a50e702011-10-18 21:21:00 +02007074 const Py_ssize_t n = PyBytes_Size(*outbytes);
7075 if (outsize > PY_SSIZE_T_MAX - n) {
7076 PyErr_NoMemory();
Victor Stinner2fc507f2011-11-04 20:06:39 +01007077 Py_DECREF(substring);
Benjamin Peterson29060642009-01-31 22:14:21 +00007078 return -1;
Victor Stinner3a50e702011-10-18 21:21:00 +02007079 }
Victor Stinner2fc507f2011-11-04 20:06:39 +01007080 if (_PyBytes_Resize(outbytes, n + outsize) < 0) {
7081 Py_DECREF(substring);
Victor Stinner3a50e702011-10-18 21:21:00 +02007082 return -1;
Victor Stinner2fc507f2011-11-04 20:06:39 +01007083 }
Victor Stinner3a50e702011-10-18 21:21:00 +02007084 out = PyBytes_AS_STRING(*outbytes) + n;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00007085 }
7086
7087 /* Do the conversion */
Victor Stinner3a50e702011-10-18 21:21:00 +02007088 outsize = WideCharToMultiByte(code_page, flags,
7089 p, size,
7090 out, outsize,
7091 NULL, pusedDefaultChar);
Victor Stinner2fc507f2011-11-04 20:06:39 +01007092 Py_CLEAR(substring);
Victor Stinner3a50e702011-10-18 21:21:00 +02007093 if (outsize <= 0)
7094 goto error;
7095 if (pusedDefaultChar && *pusedDefaultChar)
7096 return -2;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00007097 return 0;
Victor Stinner554f3f02010-06-16 23:33:54 +00007098
Victor Stinner3a50e702011-10-18 21:21:00 +02007099error:
Victor Stinner2fc507f2011-11-04 20:06:39 +01007100 Py_XDECREF(substring);
Victor Stinner3a50e702011-10-18 21:21:00 +02007101 if (GetLastError() == ERROR_NO_UNICODE_TRANSLATION)
7102 return -2;
7103 PyErr_SetFromWindowsErr(0);
Victor Stinner554f3f02010-06-16 23:33:54 +00007104 return -1;
Guido van Rossumb7a40ba2000-03-28 02:01:52 +00007105}
7106
Victor Stinner3a50e702011-10-18 21:21:00 +02007107/*
7108 * Encode a Unicode string to a Windows code page into a byte string using a
7109 * error handler.
7110 *
7111 * Returns consumed characters if succeed, or raise a WindowsError and returns
7112 * -1 on other error.
7113 */
7114static int
7115encode_code_page_errors(UINT code_page, PyObject **outbytes,
Victor Stinner7581cef2011-11-03 22:32:33 +01007116 PyObject *unicode, Py_ssize_t unicode_offset,
Martin v. Löwis3d325192011-11-04 18:23:06 +01007117 Py_ssize_t insize, const char* errors)
Guido van Rossumb7a40ba2000-03-28 02:01:52 +00007118{
Victor Stinner3a50e702011-10-18 21:21:00 +02007119 const DWORD flags = encode_code_page_flags(code_page, errors);
Victor Stinner2fc507f2011-11-04 20:06:39 +01007120 Py_ssize_t pos = unicode_offset;
7121 Py_ssize_t endin = unicode_offset + insize;
Victor Stinner3a50e702011-10-18 21:21:00 +02007122 /* Ideally, we should get reason from FormatMessage. This is the Windows
7123 2000 English version of the message. */
7124 const char *reason = "invalid character";
7125 /* 4=maximum length of a UTF-8 sequence */
7126 char buffer[4];
7127 BOOL usedDefaultChar = FALSE, *pusedDefaultChar;
7128 Py_ssize_t outsize;
7129 char *out;
Victor Stinner3a50e702011-10-18 21:21:00 +02007130 PyObject *errorHandler = NULL;
7131 PyObject *exc = NULL;
7132 PyObject *encoding_obj = NULL;
7133 char *encoding;
Martin v. Löwis3d325192011-11-04 18:23:06 +01007134 Py_ssize_t newpos, newoutsize;
Victor Stinner3a50e702011-10-18 21:21:00 +02007135 PyObject *rep;
7136 int ret = -1;
7137
7138 assert(insize > 0);
7139
7140 encoding = code_page_name(code_page, &encoding_obj);
7141 if (encoding == NULL)
7142 return -1;
7143
7144 if (errors == NULL || strcmp(errors, "strict") == 0) {
7145 /* The last error was ERROR_NO_UNICODE_TRANSLATION,
7146 then we raise a UnicodeEncodeError. */
Martin v. Löwis12be46c2011-11-04 19:04:15 +01007147 make_encode_exception(&exc, encoding, unicode, 0, 0, reason);
Victor Stinner3a50e702011-10-18 21:21:00 +02007148 if (exc != NULL) {
7149 PyCodec_StrictErrors(exc);
7150 Py_DECREF(exc);
7151 }
7152 Py_XDECREF(encoding_obj);
7153 return -1;
7154 }
7155
7156 if (code_page != CP_UTF8 && code_page != CP_UTF7)
7157 pusedDefaultChar = &usedDefaultChar;
7158 else
7159 pusedDefaultChar = NULL;
7160
7161 if (Py_ARRAY_LENGTH(buffer) > PY_SSIZE_T_MAX / insize) {
7162 PyErr_NoMemory();
7163 goto error;
7164 }
7165 outsize = insize * Py_ARRAY_LENGTH(buffer);
7166
7167 if (*outbytes == NULL) {
7168 /* Create string object */
7169 *outbytes = PyBytes_FromStringAndSize(NULL, outsize);
7170 if (*outbytes == NULL)
7171 goto error;
7172 out = PyBytes_AS_STRING(*outbytes);
7173 }
7174 else {
7175 /* Extend string object */
7176 Py_ssize_t n = PyBytes_Size(*outbytes);
7177 if (n > PY_SSIZE_T_MAX - outsize) {
7178 PyErr_NoMemory();
7179 goto error;
7180 }
7181 if (_PyBytes_Resize(outbytes, n + outsize) < 0)
7182 goto error;
7183 out = PyBytes_AS_STRING(*outbytes) + n;
7184 }
7185
7186 /* Encode the string character per character */
Martin v. Löwis3d325192011-11-04 18:23:06 +01007187 while (pos < endin)
Victor Stinner3a50e702011-10-18 21:21:00 +02007188 {
Victor Stinner2fc507f2011-11-04 20:06:39 +01007189 Py_UCS4 ch = PyUnicode_READ_CHAR(unicode, pos);
7190 wchar_t chars[2];
7191 int charsize;
7192 if (ch < 0x10000) {
7193 chars[0] = (wchar_t)ch;
7194 charsize = 1;
7195 }
7196 else {
7197 ch -= 0x10000;
7198 chars[0] = 0xd800 + (ch >> 10);
7199 chars[1] = 0xdc00 + (ch & 0x3ff);
7200 charsize = 2;
7201 }
7202
Victor Stinner3a50e702011-10-18 21:21:00 +02007203 outsize = WideCharToMultiByte(code_page, flags,
Martin v. Löwis3d325192011-11-04 18:23:06 +01007204 chars, charsize,
Victor Stinner3a50e702011-10-18 21:21:00 +02007205 buffer, Py_ARRAY_LENGTH(buffer),
7206 NULL, pusedDefaultChar);
7207 if (outsize > 0) {
7208 if (pusedDefaultChar == NULL || !(*pusedDefaultChar))
7209 {
Martin v. Löwis3d325192011-11-04 18:23:06 +01007210 pos++;
Victor Stinner3a50e702011-10-18 21:21:00 +02007211 memcpy(out, buffer, outsize);
7212 out += outsize;
7213 continue;
7214 }
7215 }
7216 else if (GetLastError() != ERROR_NO_UNICODE_TRANSLATION) {
7217 PyErr_SetFromWindowsErr(0);
7218 goto error;
7219 }
7220
Victor Stinner3a50e702011-10-18 21:21:00 +02007221 rep = unicode_encode_call_errorhandler(
7222 errors, &errorHandler, encoding, reason,
Victor Stinner7581cef2011-11-03 22:32:33 +01007223 unicode, &exc,
Martin v. Löwis3d325192011-11-04 18:23:06 +01007224 pos, pos + 1, &newpos);
Victor Stinner3a50e702011-10-18 21:21:00 +02007225 if (rep == NULL)
7226 goto error;
Martin v. Löwis3d325192011-11-04 18:23:06 +01007227 pos = newpos;
Victor Stinner3a50e702011-10-18 21:21:00 +02007228
7229 if (PyBytes_Check(rep)) {
7230 outsize = PyBytes_GET_SIZE(rep);
7231 if (outsize != 1) {
7232 Py_ssize_t offset = out - PyBytes_AS_STRING(*outbytes);
7233 newoutsize = PyBytes_GET_SIZE(*outbytes) + (outsize - 1);
7234 if (_PyBytes_Resize(outbytes, newoutsize) < 0) {
7235 Py_DECREF(rep);
7236 goto error;
7237 }
7238 out = PyBytes_AS_STRING(*outbytes) + offset;
7239 }
7240 memcpy(out, PyBytes_AS_STRING(rep), outsize);
7241 out += outsize;
7242 }
7243 else {
7244 Py_ssize_t i;
7245 enum PyUnicode_Kind kind;
7246 void *data;
7247
Benjamin Petersonbac79492012-01-14 13:34:47 -05007248 if (PyUnicode_READY(rep) == -1) {
Victor Stinner3a50e702011-10-18 21:21:00 +02007249 Py_DECREF(rep);
7250 goto error;
7251 }
7252
7253 outsize = PyUnicode_GET_LENGTH(rep);
7254 if (outsize != 1) {
7255 Py_ssize_t offset = out - PyBytes_AS_STRING(*outbytes);
7256 newoutsize = PyBytes_GET_SIZE(*outbytes) + (outsize - 1);
7257 if (_PyBytes_Resize(outbytes, newoutsize) < 0) {
7258 Py_DECREF(rep);
7259 goto error;
7260 }
7261 out = PyBytes_AS_STRING(*outbytes) + offset;
7262 }
7263 kind = PyUnicode_KIND(rep);
7264 data = PyUnicode_DATA(rep);
7265 for (i=0; i < outsize; i++) {
7266 Py_UCS4 ch = PyUnicode_READ(kind, data, i);
7267 if (ch > 127) {
Martin v. Löwis12be46c2011-11-04 19:04:15 +01007268 raise_encode_exception(&exc,
Martin v. Löwis3d325192011-11-04 18:23:06 +01007269 encoding, unicode,
7270 pos, pos + 1,
Victor Stinner3a50e702011-10-18 21:21:00 +02007271 "unable to encode error handler result to ASCII");
7272 Py_DECREF(rep);
7273 goto error;
7274 }
7275 *out = (unsigned char)ch;
7276 out++;
7277 }
7278 }
7279 Py_DECREF(rep);
7280 }
7281 /* write a NUL byte */
7282 *out = 0;
7283 outsize = out - PyBytes_AS_STRING(*outbytes);
7284 assert(outsize <= PyBytes_GET_SIZE(*outbytes));
7285 if (_PyBytes_Resize(outbytes, outsize) < 0)
7286 goto error;
7287 ret = 0;
7288
7289error:
7290 Py_XDECREF(encoding_obj);
7291 Py_XDECREF(errorHandler);
7292 Py_XDECREF(exc);
7293 return ret;
7294}
7295
Victor Stinner3a50e702011-10-18 21:21:00 +02007296static PyObject *
7297encode_code_page(int code_page,
Victor Stinner7581cef2011-11-03 22:32:33 +01007298 PyObject *unicode,
Victor Stinner3a50e702011-10-18 21:21:00 +02007299 const char *errors)
7300{
Martin v. Löwis3d325192011-11-04 18:23:06 +01007301 Py_ssize_t len;
Victor Stinner3a50e702011-10-18 21:21:00 +02007302 PyObject *outbytes = NULL;
Victor Stinner7581cef2011-11-03 22:32:33 +01007303 Py_ssize_t offset;
Victor Stinner76a31a62011-11-04 00:05:13 +01007304 int chunk_len, ret, done;
Victor Stinner7581cef2011-11-03 22:32:33 +01007305
Benjamin Petersonbac79492012-01-14 13:34:47 -05007306 if (PyUnicode_READY(unicode) == -1)
Victor Stinner2fc507f2011-11-04 20:06:39 +01007307 return NULL;
7308 len = PyUnicode_GET_LENGTH(unicode);
Guido van Rossum03e29f12000-05-04 15:52:20 +00007309
Victor Stinner3a50e702011-10-18 21:21:00 +02007310 if (code_page < 0) {
7311 PyErr_SetString(PyExc_ValueError, "invalid code page number");
7312 return NULL;
7313 }
7314
Martin v. Löwis3d325192011-11-04 18:23:06 +01007315 if (len == 0)
Victor Stinner76a31a62011-11-04 00:05:13 +01007316 return PyBytes_FromStringAndSize(NULL, 0);
7317
Victor Stinner7581cef2011-11-03 22:32:33 +01007318 offset = 0;
7319 do
7320 {
Thomas Wouters0e3f5912006-08-11 14:57:12 +00007321#ifdef NEED_RETRY
Victor Stinner2fc507f2011-11-04 20:06:39 +01007322 /* UTF-16 encoding may double the size, so use only INT_MAX/2
Martin v. Löwis3d325192011-11-04 18:23:06 +01007323 chunks. */
7324 if (len > INT_MAX/2) {
7325 chunk_len = INT_MAX/2;
Victor Stinner76a31a62011-11-04 00:05:13 +01007326 done = 0;
7327 }
Victor Stinner7581cef2011-11-03 22:32:33 +01007328 else
Thomas Wouters0e3f5912006-08-11 14:57:12 +00007329#endif
Victor Stinner76a31a62011-11-04 00:05:13 +01007330 {
Martin v. Löwis3d325192011-11-04 18:23:06 +01007331 chunk_len = (int)len;
Victor Stinner76a31a62011-11-04 00:05:13 +01007332 done = 1;
7333 }
Victor Stinner2fc507f2011-11-04 20:06:39 +01007334
Victor Stinner76a31a62011-11-04 00:05:13 +01007335 ret = encode_code_page_strict(code_page, &outbytes,
Martin v. Löwis3d325192011-11-04 18:23:06 +01007336 unicode, offset, chunk_len,
Victor Stinner76a31a62011-11-04 00:05:13 +01007337 errors);
7338 if (ret == -2)
7339 ret = encode_code_page_errors(code_page, &outbytes,
7340 unicode, offset,
Martin v. Löwis3d325192011-11-04 18:23:06 +01007341 chunk_len, errors);
Victor Stinner7581cef2011-11-03 22:32:33 +01007342 if (ret < 0) {
7343 Py_XDECREF(outbytes);
7344 return NULL;
7345 }
Thomas Wouters0e3f5912006-08-11 14:57:12 +00007346
Victor Stinner7581cef2011-11-03 22:32:33 +01007347 offset += chunk_len;
Martin v. Löwis3d325192011-11-04 18:23:06 +01007348 len -= chunk_len;
Victor Stinner76a31a62011-11-04 00:05:13 +01007349 } while (!done);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00007350
Victor Stinner3a50e702011-10-18 21:21:00 +02007351 return outbytes;
7352}
7353
7354PyObject *
7355PyUnicode_EncodeMBCS(const Py_UNICODE *p,
7356 Py_ssize_t size,
7357 const char *errors)
7358{
Victor Stinner7581cef2011-11-03 22:32:33 +01007359 PyObject *unicode, *res;
7360 unicode = PyUnicode_FromUnicode(p, size);
7361 if (unicode == NULL)
7362 return NULL;
7363 res = encode_code_page(CP_ACP, unicode, errors);
7364 Py_DECREF(unicode);
7365 return res;
Victor Stinner3a50e702011-10-18 21:21:00 +02007366}
7367
7368PyObject *
7369PyUnicode_EncodeCodePage(int code_page,
7370 PyObject *unicode,
7371 const char *errors)
7372{
Victor Stinner7581cef2011-11-03 22:32:33 +01007373 return encode_code_page(code_page, unicode, errors);
Guido van Rossumb7a40ba2000-03-28 02:01:52 +00007374}
Guido van Rossum2ea3e142000-03-31 17:24:09 +00007375
Alexander Belopolsky40018472011-02-26 01:02:56 +00007376PyObject *
7377PyUnicode_AsMBCSString(PyObject *unicode)
Mark Hammond0ccda1e2003-07-01 00:13:27 +00007378{
7379 if (!PyUnicode_Check(unicode)) {
7380 PyErr_BadArgument();
7381 return NULL;
7382 }
Victor Stinner7581cef2011-11-03 22:32:33 +01007383 return PyUnicode_EncodeCodePage(CP_ACP, unicode, NULL);
Mark Hammond0ccda1e2003-07-01 00:13:27 +00007384}
7385
Thomas Wouters0e3f5912006-08-11 14:57:12 +00007386#undef NEED_RETRY
7387
Victor Stinner99b95382011-07-04 14:23:54 +02007388#endif /* HAVE_MBCS */
Guido van Rossumb7a40ba2000-03-28 02:01:52 +00007389
Guido van Rossumd57fd912000-03-10 22:53:23 +00007390/* --- Character Mapping Codec -------------------------------------------- */
7391
Alexander Belopolsky40018472011-02-26 01:02:56 +00007392PyObject *
7393PyUnicode_DecodeCharmap(const char *s,
7394 Py_ssize_t size,
7395 PyObject *mapping,
7396 const char *errors)
Guido van Rossumd57fd912000-03-10 22:53:23 +00007397{
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007398 const char *starts = s;
Martin v. Löwis18e16552006-02-15 17:27:45 +00007399 Py_ssize_t startinpos;
7400 Py_ssize_t endinpos;
7401 Py_ssize_t outpos;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007402 const char *e;
Victor Stinner7931d9a2011-11-04 00:22:48 +01007403 PyObject *v;
Martin v. Löwis18e16552006-02-15 17:27:45 +00007404 Py_ssize_t extrachars = 0;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007405 PyObject *errorHandler = NULL;
7406 PyObject *exc = NULL;
Tim Petersced69f82003-09-16 20:30:58 +00007407
Guido van Rossumd57fd912000-03-10 22:53:23 +00007408 /* Default to Latin-1 */
7409 if (mapping == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00007410 return PyUnicode_DecodeLatin1(s, size, errors);
Guido van Rossumd57fd912000-03-10 22:53:23 +00007411
Martin v. Löwise9b11c12011-11-08 17:35:34 +01007412 v = PyUnicode_New(size, 127);
Guido van Rossumd57fd912000-03-10 22:53:23 +00007413 if (v == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00007414 goto onError;
Guido van Rossumd57fd912000-03-10 22:53:23 +00007415 if (size == 0)
Victor Stinner7931d9a2011-11-04 00:22:48 +01007416 return v;
Martin v. Löwise9b11c12011-11-08 17:35:34 +01007417 outpos = 0;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007418 e = s + size;
Walter Dörwaldd1c1e102005-10-06 20:29:57 +00007419 if (PyUnicode_CheckExact(mapping)) {
Victor Stinnerebf3ba82011-11-10 20:30:22 +01007420 Py_ssize_t maplen;
Antoine Pitrouaaefac72012-06-16 22:48:21 +02007421 enum PyUnicode_Kind mapkind;
7422 void *mapdata;
Victor Stinnerebf3ba82011-11-10 20:30:22 +01007423 Py_UCS4 x;
7424
Benjamin Petersonbac79492012-01-14 13:34:47 -05007425 if (PyUnicode_READY(mapping) == -1)
Victor Stinnerebf3ba82011-11-10 20:30:22 +01007426 return NULL;
7427
7428 maplen = PyUnicode_GET_LENGTH(mapping);
Antoine Pitrouaaefac72012-06-16 22:48:21 +02007429 mapdata = PyUnicode_DATA(mapping);
7430 mapkind = PyUnicode_KIND(mapping);
Benjamin Peterson29060642009-01-31 22:14:21 +00007431 while (s < e) {
Antoine Pitrouaaefac72012-06-16 22:48:21 +02007432 unsigned char ch;
7433 if (mapkind == PyUnicode_2BYTE_KIND && maplen >= 256) {
7434 enum PyUnicode_Kind outkind = PyUnicode_KIND(v);
7435 if (outkind == PyUnicode_1BYTE_KIND) {
7436 void *outdata = PyUnicode_DATA(v);
7437 Py_UCS4 maxchar = PyUnicode_MAX_CHAR_VALUE(v);
7438 while (s < e) {
7439 unsigned char ch = *s;
7440 x = PyUnicode_READ(PyUnicode_2BYTE_KIND, mapdata, ch);
7441 if (x > maxchar)
7442 goto Error;
7443 PyUnicode_WRITE(PyUnicode_1BYTE_KIND, outdata, outpos++, x);
7444 ++s;
7445 }
7446 break;
7447 }
7448 else if (outkind == PyUnicode_2BYTE_KIND) {
7449 void *outdata = PyUnicode_DATA(v);
7450 while (s < e) {
7451 unsigned char ch = *s;
7452 x = PyUnicode_READ(PyUnicode_2BYTE_KIND, mapdata, ch);
7453 if (x == 0xFFFE)
7454 goto Error;
7455 PyUnicode_WRITE(PyUnicode_2BYTE_KIND, outdata, outpos++, x);
7456 ++s;
7457 }
7458 break;
7459 }
7460 }
7461 ch = *s;
Guido van Rossumd57fd912000-03-10 22:53:23 +00007462
Benjamin Peterson29060642009-01-31 22:14:21 +00007463 if (ch < maplen)
Antoine Pitrouaaefac72012-06-16 22:48:21 +02007464 x = PyUnicode_READ(mapkind, mapdata, ch);
Victor Stinnerebf3ba82011-11-10 20:30:22 +01007465 else
7466 x = 0xfffe; /* invalid value */
Antoine Pitrouaaefac72012-06-16 22:48:21 +02007467Error:
Victor Stinnerebf3ba82011-11-10 20:30:22 +01007468 if (x == 0xfffe)
7469 {
Benjamin Peterson29060642009-01-31 22:14:21 +00007470 /* undefined mapping */
Benjamin Peterson29060642009-01-31 22:14:21 +00007471 startinpos = s-starts;
7472 endinpos = startinpos+1;
7473 if (unicode_decode_call_errorhandler(
7474 errors, &errorHandler,
7475 "charmap", "character maps to <undefined>",
7476 &starts, &e, &startinpos, &endinpos, &exc, &s,
Martin v. Löwise9b11c12011-11-08 17:35:34 +01007477 &v, &outpos)) {
Benjamin Peterson29060642009-01-31 22:14:21 +00007478 goto onError;
7479 }
7480 continue;
7481 }
Victor Stinnerebf3ba82011-11-10 20:30:22 +01007482
Martin v. Löwise9b11c12011-11-08 17:35:34 +01007483 if (unicode_putchar(&v, &outpos, x) < 0)
7484 goto onError;
Benjamin Peterson29060642009-01-31 22:14:21 +00007485 ++s;
Benjamin Peterson14339b62009-01-31 16:36:08 +00007486 }
Walter Dörwaldd1c1e102005-10-06 20:29:57 +00007487 }
7488 else {
Benjamin Peterson29060642009-01-31 22:14:21 +00007489 while (s < e) {
7490 unsigned char ch = *s;
7491 PyObject *w, *x;
Walter Dörwaldd1c1e102005-10-06 20:29:57 +00007492
Benjamin Peterson29060642009-01-31 22:14:21 +00007493 /* Get mapping (char ordinal -> integer, Unicode char or None) */
7494 w = PyLong_FromLong((long)ch);
7495 if (w == NULL)
7496 goto onError;
7497 x = PyObject_GetItem(mapping, w);
7498 Py_DECREF(w);
7499 if (x == NULL) {
7500 if (PyErr_ExceptionMatches(PyExc_LookupError)) {
7501 /* No mapping found means: mapping is undefined. */
7502 PyErr_Clear();
Serhiy Storchaka4fb8cae2013-01-15 14:43:21 +02007503 goto Undefined;
Benjamin Peterson29060642009-01-31 22:14:21 +00007504 } else
7505 goto onError;
7506 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00007507
Benjamin Peterson29060642009-01-31 22:14:21 +00007508 /* Apply mapping */
Serhiy Storchaka4fb8cae2013-01-15 14:43:21 +02007509 if (x == Py_None)
7510 goto Undefined;
Benjamin Peterson29060642009-01-31 22:14:21 +00007511 if (PyLong_Check(x)) {
7512 long value = PyLong_AS_LONG(x);
Serhiy Storchaka4fb8cae2013-01-15 14:43:21 +02007513 if (value == 0xFFFE)
7514 goto Undefined;
Antoine Pitroua1f76552012-09-23 20:00:04 +02007515 if (value < 0 || value > MAX_UNICODE) {
7516 PyErr_Format(PyExc_TypeError,
7517 "character mapping must be in range(0x%lx)",
7518 (unsigned long)MAX_UNICODE + 1);
Benjamin Peterson29060642009-01-31 22:14:21 +00007519 Py_DECREF(x);
7520 goto onError;
7521 }
Serhiy Storchakaafb1cb52013-01-29 12:13:22 +02007522 if (unicode_putchar(&v, &outpos, value) < 0) {
7523 Py_DECREF(x);
Martin v. Löwise9b11c12011-11-08 17:35:34 +01007524 goto onError;
Serhiy Storchakaafb1cb52013-01-29 12:13:22 +02007525 }
Benjamin Peterson29060642009-01-31 22:14:21 +00007526 }
Benjamin Peterson29060642009-01-31 22:14:21 +00007527 else if (PyUnicode_Check(x)) {
Martin v. Löwise9b11c12011-11-08 17:35:34 +01007528 Py_ssize_t targetsize;
Benjamin Peterson14339b62009-01-31 16:36:08 +00007529
Serhiy Storchakaafb1cb52013-01-29 12:13:22 +02007530 if (PyUnicode_READY(x) == -1) {
7531 Py_DECREF(x);
Martin v. Löwise9b11c12011-11-08 17:35:34 +01007532 goto onError;
Serhiy Storchakaafb1cb52013-01-29 12:13:22 +02007533 }
Martin v. Löwise9b11c12011-11-08 17:35:34 +01007534 targetsize = PyUnicode_GET_LENGTH(x);
7535
7536 if (targetsize == 1) {
Benjamin Peterson29060642009-01-31 22:14:21 +00007537 /* 1-1 mapping */
Serhiy Storchaka45d16d92013-01-15 15:01:20 +02007538 Py_UCS4 value = PyUnicode_READ_CHAR(x, 0);
Serhiy Storchaka4fb8cae2013-01-15 14:43:21 +02007539 if (value == 0xFFFE)
7540 goto Undefined;
Serhiy Storchakaafb1cb52013-01-29 12:13:22 +02007541 if (unicode_putchar(&v, &outpos, value) < 0) {
7542 Py_DECREF(x);
Martin v. Löwise9b11c12011-11-08 17:35:34 +01007543 goto onError;
Serhiy Storchakaafb1cb52013-01-29 12:13:22 +02007544 }
Martin v. Löwise9b11c12011-11-08 17:35:34 +01007545 }
Benjamin Peterson29060642009-01-31 22:14:21 +00007546 else if (targetsize > 1) {
7547 /* 1-n mapping */
7548 if (targetsize > extrachars) {
7549 /* resize first */
Benjamin Peterson29060642009-01-31 22:14:21 +00007550 Py_ssize_t needed = (targetsize - extrachars) + \
7551 (targetsize << 2);
7552 extrachars += needed;
7553 /* XXX overflow detection missing */
Victor Stinner16e6a802011-12-12 13:24:15 +01007554 if (unicode_resize(&v,
7555 PyUnicode_GET_LENGTH(v) + needed) < 0)
7556 {
Benjamin Peterson29060642009-01-31 22:14:21 +00007557 Py_DECREF(x);
7558 goto onError;
7559 }
Benjamin Peterson29060642009-01-31 22:14:21 +00007560 }
Serhiy Storchakaafb1cb52013-01-29 12:13:22 +02007561 if (unicode_widen(&v, outpos,
7562 PyUnicode_MAX_CHAR_VALUE(x)) < 0) {
7563 Py_DECREF(x);
Martin v. Löwise9b11c12011-11-08 17:35:34 +01007564 goto onError;
Serhiy Storchakaafb1cb52013-01-29 12:13:22 +02007565 }
Martin v. Löwise9b11c12011-11-08 17:35:34 +01007566 PyUnicode_CopyCharacters(v, outpos, x, 0, targetsize);
7567 outpos += targetsize;
Benjamin Peterson29060642009-01-31 22:14:21 +00007568 extrachars -= targetsize;
7569 }
7570 /* 1-0 mapping: skip the character */
7571 }
7572 else {
7573 /* wrong return value */
7574 PyErr_SetString(PyExc_TypeError,
7575 "character mapping must return integer, None or str");
Benjamin Peterson14339b62009-01-31 16:36:08 +00007576 Py_DECREF(x);
7577 goto onError;
7578 }
Benjamin Peterson29060642009-01-31 22:14:21 +00007579 Py_DECREF(x);
7580 ++s;
Serhiy Storchaka4fb8cae2013-01-15 14:43:21 +02007581 continue;
7582Undefined:
7583 /* undefined mapping */
7584 Py_XDECREF(x);
Serhiy Storchaka4fb8cae2013-01-15 14:43:21 +02007585 startinpos = s-starts;
7586 endinpos = startinpos+1;
7587 if (unicode_decode_call_errorhandler(
7588 errors, &errorHandler,
7589 "charmap", "character maps to <undefined>",
7590 &starts, &e, &startinpos, &endinpos, &exc, &s,
Serhiy Storchaka45d16d92013-01-15 15:01:20 +02007591 &v, &outpos)) {
Serhiy Storchaka4fb8cae2013-01-15 14:43:21 +02007592 goto onError;
7593 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00007594 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00007595 }
Victor Stinner16e6a802011-12-12 13:24:15 +01007596 if (unicode_resize(&v, outpos) < 0)
Antoine Pitroua8f63c02011-11-08 18:37:16 +01007597 goto onError;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007598 Py_XDECREF(errorHandler);
7599 Py_XDECREF(exc);
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01007600 return unicode_result(v);
Tim Petersced69f82003-09-16 20:30:58 +00007601
Benjamin Peterson29060642009-01-31 22:14:21 +00007602 onError:
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007603 Py_XDECREF(errorHandler);
7604 Py_XDECREF(exc);
Guido van Rossumd57fd912000-03-10 22:53:23 +00007605 Py_XDECREF(v);
7606 return NULL;
7607}
7608
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007609/* Charmap encoding: the lookup table */
7610
Alexander Belopolsky40018472011-02-26 01:02:56 +00007611struct encoding_map {
Benjamin Peterson29060642009-01-31 22:14:21 +00007612 PyObject_HEAD
7613 unsigned char level1[32];
7614 int count2, count3;
7615 unsigned char level23[1];
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007616};
7617
7618static PyObject*
7619encoding_map_size(PyObject *obj, PyObject* args)
7620{
7621 struct encoding_map *map = (struct encoding_map*)obj;
Benjamin Peterson14339b62009-01-31 16:36:08 +00007622 return PyLong_FromLong(sizeof(*map) - 1 + 16*map->count2 +
Benjamin Peterson29060642009-01-31 22:14:21 +00007623 128*map->count3);
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007624}
7625
7626static PyMethodDef encoding_map_methods[] = {
Benjamin Peterson14339b62009-01-31 16:36:08 +00007627 {"size", encoding_map_size, METH_NOARGS,
Benjamin Peterson29060642009-01-31 22:14:21 +00007628 PyDoc_STR("Return the size (in bytes) of this object") },
7629 { 0 }
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007630};
7631
7632static void
7633encoding_map_dealloc(PyObject* o)
7634{
Benjamin Peterson14339b62009-01-31 16:36:08 +00007635 PyObject_FREE(o);
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007636}
7637
7638static PyTypeObject EncodingMapType = {
Benjamin Peterson14339b62009-01-31 16:36:08 +00007639 PyVarObject_HEAD_INIT(NULL, 0)
Benjamin Peterson29060642009-01-31 22:14:21 +00007640 "EncodingMap", /*tp_name*/
7641 sizeof(struct encoding_map), /*tp_basicsize*/
7642 0, /*tp_itemsize*/
7643 /* methods */
7644 encoding_map_dealloc, /*tp_dealloc*/
7645 0, /*tp_print*/
7646 0, /*tp_getattr*/
7647 0, /*tp_setattr*/
Mark Dickinsone94c6792009-02-02 20:36:42 +00007648 0, /*tp_reserved*/
Benjamin Peterson29060642009-01-31 22:14:21 +00007649 0, /*tp_repr*/
7650 0, /*tp_as_number*/
7651 0, /*tp_as_sequence*/
7652 0, /*tp_as_mapping*/
7653 0, /*tp_hash*/
7654 0, /*tp_call*/
7655 0, /*tp_str*/
7656 0, /*tp_getattro*/
7657 0, /*tp_setattro*/
7658 0, /*tp_as_buffer*/
7659 Py_TPFLAGS_DEFAULT, /*tp_flags*/
7660 0, /*tp_doc*/
7661 0, /*tp_traverse*/
7662 0, /*tp_clear*/
7663 0, /*tp_richcompare*/
7664 0, /*tp_weaklistoffset*/
7665 0, /*tp_iter*/
7666 0, /*tp_iternext*/
7667 encoding_map_methods, /*tp_methods*/
7668 0, /*tp_members*/
7669 0, /*tp_getset*/
7670 0, /*tp_base*/
7671 0, /*tp_dict*/
7672 0, /*tp_descr_get*/
7673 0, /*tp_descr_set*/
7674 0, /*tp_dictoffset*/
7675 0, /*tp_init*/
7676 0, /*tp_alloc*/
7677 0, /*tp_new*/
7678 0, /*tp_free*/
7679 0, /*tp_is_gc*/
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007680};
7681
7682PyObject*
7683PyUnicode_BuildEncodingMap(PyObject* string)
7684{
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007685 PyObject *result;
7686 struct encoding_map *mresult;
7687 int i;
7688 int need_dict = 0;
7689 unsigned char level1[32];
7690 unsigned char level2[512];
7691 unsigned char *mlevel1, *mlevel2, *mlevel3;
7692 int count2 = 0, count3 = 0;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02007693 int kind;
7694 void *data;
Antoine Pitrouaaefac72012-06-16 22:48:21 +02007695 Py_ssize_t length;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02007696 Py_UCS4 ch;
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007697
Antoine Pitrouaaefac72012-06-16 22:48:21 +02007698 if (!PyUnicode_Check(string) || !PyUnicode_GET_LENGTH(string)) {
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007699 PyErr_BadArgument();
7700 return NULL;
7701 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02007702 kind = PyUnicode_KIND(string);
7703 data = PyUnicode_DATA(string);
Antoine Pitrouaaefac72012-06-16 22:48:21 +02007704 length = PyUnicode_GET_LENGTH(string);
7705 length = Py_MIN(length, 256);
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007706 memset(level1, 0xFF, sizeof level1);
7707 memset(level2, 0xFF, sizeof level2);
7708
7709 /* If there isn't a one-to-one mapping of NULL to \0,
7710 or if there are non-BMP characters, we need to use
7711 a mapping dictionary. */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02007712 if (PyUnicode_READ(kind, data, 0) != 0)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007713 need_dict = 1;
Antoine Pitrouaaefac72012-06-16 22:48:21 +02007714 for (i = 1; i < length; i++) {
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007715 int l1, l2;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02007716 ch = PyUnicode_READ(kind, data, i);
7717 if (ch == 0 || ch > 0xFFFF) {
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007718 need_dict = 1;
7719 break;
7720 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02007721 if (ch == 0xFFFE)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007722 /* unmapped character */
7723 continue;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02007724 l1 = ch >> 11;
7725 l2 = ch >> 7;
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007726 if (level1[l1] == 0xFF)
7727 level1[l1] = count2++;
7728 if (level2[l2] == 0xFF)
Benjamin Peterson14339b62009-01-31 16:36:08 +00007729 level2[l2] = count3++;
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007730 }
7731
7732 if (count2 >= 0xFF || count3 >= 0xFF)
7733 need_dict = 1;
7734
7735 if (need_dict) {
7736 PyObject *result = PyDict_New();
7737 PyObject *key, *value;
7738 if (!result)
7739 return NULL;
Antoine Pitrouaaefac72012-06-16 22:48:21 +02007740 for (i = 0; i < length; i++) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02007741 key = PyLong_FromLong(PyUnicode_READ(kind, data, i));
Christian Heimes217cfd12007-12-02 14:31:20 +00007742 value = PyLong_FromLong(i);
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007743 if (!key || !value)
7744 goto failed1;
7745 if (PyDict_SetItem(result, key, value) == -1)
7746 goto failed1;
7747 Py_DECREF(key);
7748 Py_DECREF(value);
7749 }
7750 return result;
7751 failed1:
7752 Py_XDECREF(key);
7753 Py_XDECREF(value);
7754 Py_DECREF(result);
7755 return NULL;
7756 }
7757
7758 /* Create a three-level trie */
7759 result = PyObject_MALLOC(sizeof(struct encoding_map) +
7760 16*count2 + 128*count3 - 1);
7761 if (!result)
7762 return PyErr_NoMemory();
7763 PyObject_Init(result, &EncodingMapType);
7764 mresult = (struct encoding_map*)result;
7765 mresult->count2 = count2;
7766 mresult->count3 = count3;
7767 mlevel1 = mresult->level1;
7768 mlevel2 = mresult->level23;
7769 mlevel3 = mresult->level23 + 16*count2;
7770 memcpy(mlevel1, level1, 32);
7771 memset(mlevel2, 0xFF, 16*count2);
7772 memset(mlevel3, 0, 128*count3);
7773 count3 = 0;
Antoine Pitrouaaefac72012-06-16 22:48:21 +02007774 for (i = 1; i < length; i++) {
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007775 int o1, o2, o3, i2, i3;
Antoine Pitrouaaefac72012-06-16 22:48:21 +02007776 Py_UCS4 ch = PyUnicode_READ(kind, data, i);
7777 if (ch == 0xFFFE)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007778 /* unmapped character */
7779 continue;
Antoine Pitrouaaefac72012-06-16 22:48:21 +02007780 o1 = ch>>11;
7781 o2 = (ch>>7) & 0xF;
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007782 i2 = 16*mlevel1[o1] + o2;
7783 if (mlevel2[i2] == 0xFF)
7784 mlevel2[i2] = count3++;
Antoine Pitrouaaefac72012-06-16 22:48:21 +02007785 o3 = ch & 0x7F;
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007786 i3 = 128*mlevel2[i2] + o3;
7787 mlevel3[i3] = i;
7788 }
7789 return result;
7790}
7791
7792static int
Victor Stinner22168992011-11-20 17:09:18 +01007793encoding_map_lookup(Py_UCS4 c, PyObject *mapping)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007794{
7795 struct encoding_map *map = (struct encoding_map*)mapping;
7796 int l1 = c>>11;
7797 int l2 = (c>>7) & 0xF;
7798 int l3 = c & 0x7F;
7799 int i;
7800
Victor Stinner22168992011-11-20 17:09:18 +01007801 if (c > 0xFFFF)
Benjamin Peterson29060642009-01-31 22:14:21 +00007802 return -1;
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007803 if (c == 0)
7804 return 0;
7805 /* level 1*/
7806 i = map->level1[l1];
7807 if (i == 0xFF) {
7808 return -1;
7809 }
7810 /* level 2*/
7811 i = map->level23[16*i+l2];
7812 if (i == 0xFF) {
7813 return -1;
7814 }
7815 /* level 3 */
7816 i = map->level23[16*map->count2 + 128*i + l3];
7817 if (i == 0) {
7818 return -1;
7819 }
7820 return i;
7821}
7822
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007823/* Lookup the character ch in the mapping. If the character
7824 can't be found, Py_None is returned (or NULL, if another
Fred Drakedb390c12005-10-28 14:39:47 +00007825 error occurred). */
Alexander Belopolsky40018472011-02-26 01:02:56 +00007826static PyObject *
Victor Stinner22168992011-11-20 17:09:18 +01007827charmapencode_lookup(Py_UCS4 c, PyObject *mapping)
Guido van Rossumd57fd912000-03-10 22:53:23 +00007828{
Christian Heimes217cfd12007-12-02 14:31:20 +00007829 PyObject *w = PyLong_FromLong((long)c);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007830 PyObject *x;
7831
7832 if (w == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00007833 return NULL;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007834 x = PyObject_GetItem(mapping, w);
7835 Py_DECREF(w);
7836 if (x == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +00007837 if (PyErr_ExceptionMatches(PyExc_LookupError)) {
7838 /* No mapping found means: mapping is undefined. */
7839 PyErr_Clear();
7840 x = Py_None;
7841 Py_INCREF(x);
7842 return x;
7843 } else
7844 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00007845 }
Walter Dörwaldadc72742003-01-08 22:01:33 +00007846 else if (x == Py_None)
Benjamin Peterson29060642009-01-31 22:14:21 +00007847 return x;
Christian Heimes217cfd12007-12-02 14:31:20 +00007848 else if (PyLong_Check(x)) {
Benjamin Peterson29060642009-01-31 22:14:21 +00007849 long value = PyLong_AS_LONG(x);
7850 if (value < 0 || value > 255) {
7851 PyErr_SetString(PyExc_TypeError,
7852 "character mapping must be in range(256)");
7853 Py_DECREF(x);
7854 return NULL;
7855 }
7856 return x;
Guido van Rossumd57fd912000-03-10 22:53:23 +00007857 }
Christian Heimes72b710a2008-05-26 13:28:38 +00007858 else if (PyBytes_Check(x))
Benjamin Peterson29060642009-01-31 22:14:21 +00007859 return x;
Guido van Rossumd57fd912000-03-10 22:53:23 +00007860 else {
Benjamin Peterson29060642009-01-31 22:14:21 +00007861 /* wrong return value */
7862 PyErr_Format(PyExc_TypeError,
7863 "character mapping must return integer, bytes or None, not %.400s",
7864 x->ob_type->tp_name);
7865 Py_DECREF(x);
7866 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00007867 }
7868}
7869
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007870static int
Guido van Rossum98297ee2007-11-06 21:34:58 +00007871charmapencode_resize(PyObject **outobj, Py_ssize_t *outpos, Py_ssize_t requiredsize)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007872{
Benjamin Peterson14339b62009-01-31 16:36:08 +00007873 Py_ssize_t outsize = PyBytes_GET_SIZE(*outobj);
7874 /* exponentially overallocate to minimize reallocations */
7875 if (requiredsize < 2*outsize)
7876 requiredsize = 2*outsize;
7877 if (_PyBytes_Resize(outobj, requiredsize))
7878 return -1;
7879 return 0;
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007880}
7881
Benjamin Peterson14339b62009-01-31 16:36:08 +00007882typedef enum charmapencode_result {
Benjamin Peterson29060642009-01-31 22:14:21 +00007883 enc_SUCCESS, enc_FAILED, enc_EXCEPTION
Alexander Belopolsky40018472011-02-26 01:02:56 +00007884} charmapencode_result;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007885/* lookup the character, put the result in the output string and adjust
Walter Dörwald827b0552007-05-12 13:23:53 +00007886 various state variables. Resize the output bytes object if not enough
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007887 space is available. Return a new reference to the object that
7888 was put in the output buffer, or Py_None, if the mapping was undefined
7889 (in which case no character was written) or NULL, if a
Andrew M. Kuchling8294de52005-11-02 16:36:12 +00007890 reallocation error occurred. The caller must decref the result */
Alexander Belopolsky40018472011-02-26 01:02:56 +00007891static charmapencode_result
Victor Stinner22168992011-11-20 17:09:18 +01007892charmapencode_output(Py_UCS4 c, PyObject *mapping,
Alexander Belopolsky40018472011-02-26 01:02:56 +00007893 PyObject **outobj, Py_ssize_t *outpos)
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007894{
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007895 PyObject *rep;
7896 char *outstart;
Christian Heimes72b710a2008-05-26 13:28:38 +00007897 Py_ssize_t outsize = PyBytes_GET_SIZE(*outobj);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007898
Christian Heimes90aa7642007-12-19 02:45:37 +00007899 if (Py_TYPE(mapping) == &EncodingMapType) {
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007900 int res = encoding_map_lookup(c, mapping);
Benjamin Peterson29060642009-01-31 22:14:21 +00007901 Py_ssize_t requiredsize = *outpos+1;
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007902 if (res == -1)
7903 return enc_FAILED;
Benjamin Peterson29060642009-01-31 22:14:21 +00007904 if (outsize<requiredsize)
7905 if (charmapencode_resize(outobj, outpos, requiredsize))
7906 return enc_EXCEPTION;
Christian Heimes72b710a2008-05-26 13:28:38 +00007907 outstart = PyBytes_AS_STRING(*outobj);
Benjamin Peterson29060642009-01-31 22:14:21 +00007908 outstart[(*outpos)++] = (char)res;
7909 return enc_SUCCESS;
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007910 }
7911
7912 rep = charmapencode_lookup(c, mapping);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007913 if (rep==NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00007914 return enc_EXCEPTION;
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007915 else if (rep==Py_None) {
Benjamin Peterson29060642009-01-31 22:14:21 +00007916 Py_DECREF(rep);
7917 return enc_FAILED;
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007918 } else {
Benjamin Peterson29060642009-01-31 22:14:21 +00007919 if (PyLong_Check(rep)) {
7920 Py_ssize_t requiredsize = *outpos+1;
7921 if (outsize<requiredsize)
7922 if (charmapencode_resize(outobj, outpos, requiredsize)) {
7923 Py_DECREF(rep);
7924 return enc_EXCEPTION;
7925 }
Christian Heimes72b710a2008-05-26 13:28:38 +00007926 outstart = PyBytes_AS_STRING(*outobj);
Benjamin Peterson29060642009-01-31 22:14:21 +00007927 outstart[(*outpos)++] = (char)PyLong_AS_LONG(rep);
Benjamin Peterson14339b62009-01-31 16:36:08 +00007928 }
Benjamin Peterson29060642009-01-31 22:14:21 +00007929 else {
7930 const char *repchars = PyBytes_AS_STRING(rep);
7931 Py_ssize_t repsize = PyBytes_GET_SIZE(rep);
7932 Py_ssize_t requiredsize = *outpos+repsize;
7933 if (outsize<requiredsize)
7934 if (charmapencode_resize(outobj, outpos, requiredsize)) {
7935 Py_DECREF(rep);
7936 return enc_EXCEPTION;
7937 }
Christian Heimes72b710a2008-05-26 13:28:38 +00007938 outstart = PyBytes_AS_STRING(*outobj);
Benjamin Peterson29060642009-01-31 22:14:21 +00007939 memcpy(outstart + *outpos, repchars, repsize);
7940 *outpos += repsize;
7941 }
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007942 }
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007943 Py_DECREF(rep);
7944 return enc_SUCCESS;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007945}
7946
7947/* handle an error in PyUnicode_EncodeCharmap
7948 Return 0 on success, -1 on error */
Alexander Belopolsky40018472011-02-26 01:02:56 +00007949static int
7950charmap_encoding_error(
Martin v. Löwis23e275b2011-11-02 18:02:51 +01007951 PyObject *unicode, Py_ssize_t *inpos, PyObject *mapping,
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007952 PyObject **exceptionObject,
Walter Dörwalde5402fb2003-08-14 20:25:29 +00007953 int *known_errorHandler, PyObject **errorHandler, const char *errors,
Guido van Rossum98297ee2007-11-06 21:34:58 +00007954 PyObject **res, Py_ssize_t *respos)
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007955{
7956 PyObject *repunicode = NULL; /* initialize to prevent gcc warning */
Martin v. Löwis23e275b2011-11-02 18:02:51 +01007957 Py_ssize_t size, repsize;
Martin v. Löwis18e16552006-02-15 17:27:45 +00007958 Py_ssize_t newpos;
Victor Stinnerae4f7c82011-11-20 18:28:55 +01007959 enum PyUnicode_Kind kind;
7960 void *data;
7961 Py_ssize_t index;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007962 /* startpos for collecting unencodable chars */
Martin v. Löwis18e16552006-02-15 17:27:45 +00007963 Py_ssize_t collstartpos = *inpos;
7964 Py_ssize_t collendpos = *inpos+1;
7965 Py_ssize_t collpos;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007966 char *encoding = "charmap";
7967 char *reason = "character maps to <undefined>";
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007968 charmapencode_result x;
Martin v. Löwis23e275b2011-11-02 18:02:51 +01007969 Py_UCS4 ch;
Brian Curtin2787ea42011-11-02 15:09:37 -05007970 int val;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007971
Benjamin Petersonbac79492012-01-14 13:34:47 -05007972 if (PyUnicode_READY(unicode) == -1)
Martin v. Löwis23e275b2011-11-02 18:02:51 +01007973 return -1;
7974 size = PyUnicode_GET_LENGTH(unicode);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007975 /* find all unencodable characters */
7976 while (collendpos < size) {
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007977 PyObject *rep;
Christian Heimes90aa7642007-12-19 02:45:37 +00007978 if (Py_TYPE(mapping) == &EncodingMapType) {
Martin v. Löwis23e275b2011-11-02 18:02:51 +01007979 ch = PyUnicode_READ_CHAR(unicode, collendpos);
Brian Curtin2787ea42011-11-02 15:09:37 -05007980 val = encoding_map_lookup(ch, mapping);
7981 if (val != -1)
Benjamin Peterson29060642009-01-31 22:14:21 +00007982 break;
7983 ++collendpos;
7984 continue;
7985 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00007986
Martin v. Löwis23e275b2011-11-02 18:02:51 +01007987 ch = PyUnicode_READ_CHAR(unicode, collendpos);
7988 rep = charmapencode_lookup(ch, mapping);
Benjamin Peterson29060642009-01-31 22:14:21 +00007989 if (rep==NULL)
7990 return -1;
7991 else if (rep!=Py_None) {
7992 Py_DECREF(rep);
7993 break;
7994 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00007995 Py_DECREF(rep);
Benjamin Peterson29060642009-01-31 22:14:21 +00007996 ++collendpos;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007997 }
7998 /* cache callback name lookup
7999 * (if not done yet, i.e. it's the first error) */
8000 if (*known_errorHandler==-1) {
Benjamin Peterson29060642009-01-31 22:14:21 +00008001 if ((errors==NULL) || (!strcmp(errors, "strict")))
8002 *known_errorHandler = 1;
8003 else if (!strcmp(errors, "replace"))
8004 *known_errorHandler = 2;
8005 else if (!strcmp(errors, "ignore"))
8006 *known_errorHandler = 3;
8007 else if (!strcmp(errors, "xmlcharrefreplace"))
8008 *known_errorHandler = 4;
8009 else
8010 *known_errorHandler = 0;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008011 }
8012 switch (*known_errorHandler) {
Benjamin Peterson14339b62009-01-31 16:36:08 +00008013 case 1: /* strict */
Martin v. Löwis12be46c2011-11-04 19:04:15 +01008014 raise_encode_exception(exceptionObject, encoding, unicode, collstartpos, collendpos, reason);
Benjamin Peterson14339b62009-01-31 16:36:08 +00008015 return -1;
8016 case 2: /* replace */
8017 for (collpos = collstartpos; collpos<collendpos; ++collpos) {
Benjamin Peterson29060642009-01-31 22:14:21 +00008018 x = charmapencode_output('?', mapping, res, respos);
8019 if (x==enc_EXCEPTION) {
8020 return -1;
8021 }
8022 else if (x==enc_FAILED) {
Martin v. Löwis12be46c2011-11-04 19:04:15 +01008023 raise_encode_exception(exceptionObject, encoding, unicode, collstartpos, collendpos, reason);
Benjamin Peterson29060642009-01-31 22:14:21 +00008024 return -1;
8025 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00008026 }
8027 /* fall through */
8028 case 3: /* ignore */
8029 *inpos = collendpos;
8030 break;
8031 case 4: /* xmlcharrefreplace */
8032 /* generate replacement (temporarily (mis)uses p) */
8033 for (collpos = collstartpos; collpos < collendpos; ++collpos) {
Benjamin Peterson29060642009-01-31 22:14:21 +00008034 char buffer[2+29+1+1];
8035 char *cp;
Martin v. Löwis23e275b2011-11-02 18:02:51 +01008036 sprintf(buffer, "&#%d;", (int)PyUnicode_READ_CHAR(unicode, collpos));
Benjamin Peterson29060642009-01-31 22:14:21 +00008037 for (cp = buffer; *cp; ++cp) {
8038 x = charmapencode_output(*cp, mapping, res, respos);
8039 if (x==enc_EXCEPTION)
8040 return -1;
8041 else if (x==enc_FAILED) {
Martin v. Löwis12be46c2011-11-04 19:04:15 +01008042 raise_encode_exception(exceptionObject, encoding, unicode, collstartpos, collendpos, reason);
Benjamin Peterson29060642009-01-31 22:14:21 +00008043 return -1;
8044 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00008045 }
8046 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00008047 *inpos = collendpos;
8048 break;
8049 default:
8050 repunicode = unicode_encode_call_errorhandler(errors, errorHandler,
Martin v. Löwis23e275b2011-11-02 18:02:51 +01008051 encoding, reason, unicode, exceptionObject,
Benjamin Peterson29060642009-01-31 22:14:21 +00008052 collstartpos, collendpos, &newpos);
Benjamin Peterson14339b62009-01-31 16:36:08 +00008053 if (repunicode == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00008054 return -1;
Martin v. Löwis011e8422009-05-05 04:43:17 +00008055 if (PyBytes_Check(repunicode)) {
8056 /* Directly copy bytes result to output. */
8057 Py_ssize_t outsize = PyBytes_Size(*res);
8058 Py_ssize_t requiredsize;
8059 repsize = PyBytes_Size(repunicode);
8060 requiredsize = *respos + repsize;
8061 if (requiredsize > outsize)
8062 /* Make room for all additional bytes. */
8063 if (charmapencode_resize(res, respos, requiredsize)) {
8064 Py_DECREF(repunicode);
8065 return -1;
8066 }
8067 memcpy(PyBytes_AsString(*res) + *respos,
8068 PyBytes_AsString(repunicode), repsize);
8069 *respos += repsize;
8070 *inpos = newpos;
Martin v. Löwisdb12d452009-05-02 18:52:14 +00008071 Py_DECREF(repunicode);
Martin v. Löwis011e8422009-05-05 04:43:17 +00008072 break;
Martin v. Löwisdb12d452009-05-02 18:52:14 +00008073 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00008074 /* generate replacement */
Benjamin Petersonbac79492012-01-14 13:34:47 -05008075 if (PyUnicode_READY(repunicode) == -1) {
Victor Stinnerae4f7c82011-11-20 18:28:55 +01008076 Py_DECREF(repunicode);
8077 return -1;
8078 }
Victor Stinner9e30aa52011-11-21 02:49:52 +01008079 repsize = PyUnicode_GET_LENGTH(repunicode);
Victor Stinnerae4f7c82011-11-20 18:28:55 +01008080 data = PyUnicode_DATA(repunicode);
8081 kind = PyUnicode_KIND(repunicode);
8082 for (index = 0; index < repsize; index++) {
8083 Py_UCS4 repch = PyUnicode_READ(kind, data, index);
8084 x = charmapencode_output(repch, mapping, res, respos);
Benjamin Peterson29060642009-01-31 22:14:21 +00008085 if (x==enc_EXCEPTION) {
Victor Stinnerae4f7c82011-11-20 18:28:55 +01008086 Py_DECREF(repunicode);
Benjamin Peterson29060642009-01-31 22:14:21 +00008087 return -1;
8088 }
8089 else if (x==enc_FAILED) {
8090 Py_DECREF(repunicode);
Martin v. Löwis12be46c2011-11-04 19:04:15 +01008091 raise_encode_exception(exceptionObject, encoding, unicode, collstartpos, collendpos, reason);
Benjamin Peterson29060642009-01-31 22:14:21 +00008092 return -1;
8093 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00008094 }
8095 *inpos = newpos;
8096 Py_DECREF(repunicode);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008097 }
8098 return 0;
8099}
8100
Alexander Belopolsky40018472011-02-26 01:02:56 +00008101PyObject *
Martin v. Löwis23e275b2011-11-02 18:02:51 +01008102_PyUnicode_EncodeCharmap(PyObject *unicode,
8103 PyObject *mapping,
8104 const char *errors)
Guido van Rossumd57fd912000-03-10 22:53:23 +00008105{
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008106 /* output object */
8107 PyObject *res = NULL;
8108 /* current input position */
Martin v. Löwis18e16552006-02-15 17:27:45 +00008109 Py_ssize_t inpos = 0;
Martin v. Löwis23e275b2011-11-02 18:02:51 +01008110 Py_ssize_t size;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008111 /* current output position */
Martin v. Löwis18e16552006-02-15 17:27:45 +00008112 Py_ssize_t respos = 0;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008113 PyObject *errorHandler = NULL;
8114 PyObject *exc = NULL;
8115 /* the following variable is used for caching string comparisons
8116 * -1=not initialized, 0=unknown, 1=strict, 2=replace,
8117 * 3=ignore, 4=xmlcharrefreplace */
8118 int known_errorHandler = -1;
Guido van Rossumd57fd912000-03-10 22:53:23 +00008119
Benjamin Petersonbac79492012-01-14 13:34:47 -05008120 if (PyUnicode_READY(unicode) == -1)
Martin v. Löwis23e275b2011-11-02 18:02:51 +01008121 return NULL;
8122 size = PyUnicode_GET_LENGTH(unicode);
8123
Guido van Rossumd57fd912000-03-10 22:53:23 +00008124 /* Default to Latin-1 */
8125 if (mapping == NULL)
Martin v. Löwis23e275b2011-11-02 18:02:51 +01008126 return unicode_encode_ucs1(unicode, errors, 256);
Guido van Rossumd57fd912000-03-10 22:53:23 +00008127
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008128 /* allocate enough for a simple encoding without
8129 replacements, if we need more, we'll resize */
Christian Heimes72b710a2008-05-26 13:28:38 +00008130 res = PyBytes_FromStringAndSize(NULL, size);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008131 if (res == NULL)
8132 goto onError;
Marc-André Lemburgb7520772000-08-14 11:29:19 +00008133 if (size == 0)
Benjamin Peterson29060642009-01-31 22:14:21 +00008134 return res;
Guido van Rossumd57fd912000-03-10 22:53:23 +00008135
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008136 while (inpos<size) {
Martin v. Löwis23e275b2011-11-02 18:02:51 +01008137 Py_UCS4 ch = PyUnicode_READ_CHAR(unicode, inpos);
Benjamin Peterson29060642009-01-31 22:14:21 +00008138 /* try to encode it */
Martin v. Löwis23e275b2011-11-02 18:02:51 +01008139 charmapencode_result x = charmapencode_output(ch, mapping, &res, &respos);
Benjamin Peterson29060642009-01-31 22:14:21 +00008140 if (x==enc_EXCEPTION) /* error */
8141 goto onError;
8142 if (x==enc_FAILED) { /* unencodable character */
Martin v. Löwis23e275b2011-11-02 18:02:51 +01008143 if (charmap_encoding_error(unicode, &inpos, mapping,
Benjamin Peterson29060642009-01-31 22:14:21 +00008144 &exc,
8145 &known_errorHandler, &errorHandler, errors,
8146 &res, &respos)) {
8147 goto onError;
8148 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00008149 }
Benjamin Peterson29060642009-01-31 22:14:21 +00008150 else
8151 /* done with this character => adjust input position */
8152 ++inpos;
Guido van Rossumd57fd912000-03-10 22:53:23 +00008153 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00008154
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008155 /* Resize if we allocated to much */
Christian Heimes72b710a2008-05-26 13:28:38 +00008156 if (respos<PyBytes_GET_SIZE(res))
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00008157 if (_PyBytes_Resize(&res, respos) < 0)
8158 goto onError;
Guido van Rossum98297ee2007-11-06 21:34:58 +00008159
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008160 Py_XDECREF(exc);
8161 Py_XDECREF(errorHandler);
8162 return res;
8163
Benjamin Peterson29060642009-01-31 22:14:21 +00008164 onError:
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008165 Py_XDECREF(res);
8166 Py_XDECREF(exc);
8167 Py_XDECREF(errorHandler);
Guido van Rossumd57fd912000-03-10 22:53:23 +00008168 return NULL;
8169}
8170
Martin v. Löwis23e275b2011-11-02 18:02:51 +01008171/* Deprecated */
8172PyObject *
8173PyUnicode_EncodeCharmap(const Py_UNICODE *p,
8174 Py_ssize_t size,
8175 PyObject *mapping,
8176 const char *errors)
8177{
8178 PyObject *result;
8179 PyObject *unicode = PyUnicode_FromUnicode(p, size);
8180 if (unicode == NULL)
8181 return NULL;
8182 result = _PyUnicode_EncodeCharmap(unicode, mapping, errors);
8183 Py_DECREF(unicode);
Victor Stinnerfc026c92011-11-04 00:24:51 +01008184 return result;
Martin v. Löwis23e275b2011-11-02 18:02:51 +01008185}
8186
Alexander Belopolsky40018472011-02-26 01:02:56 +00008187PyObject *
8188PyUnicode_AsCharmapString(PyObject *unicode,
8189 PyObject *mapping)
Guido van Rossumd57fd912000-03-10 22:53:23 +00008190{
8191 if (!PyUnicode_Check(unicode) || mapping == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +00008192 PyErr_BadArgument();
8193 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00008194 }
Martin v. Löwis23e275b2011-11-02 18:02:51 +01008195 return _PyUnicode_EncodeCharmap(unicode, mapping, NULL);
Guido van Rossumd57fd912000-03-10 22:53:23 +00008196}
8197
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008198/* create or adjust a UnicodeTranslateError */
Alexander Belopolsky40018472011-02-26 01:02:56 +00008199static void
8200make_translate_exception(PyObject **exceptionObject,
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008201 PyObject *unicode,
Alexander Belopolsky40018472011-02-26 01:02:56 +00008202 Py_ssize_t startpos, Py_ssize_t endpos,
8203 const char *reason)
Guido van Rossumd57fd912000-03-10 22:53:23 +00008204{
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008205 if (*exceptionObject == NULL) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008206 *exceptionObject = _PyUnicodeTranslateError_Create(
8207 unicode, startpos, endpos, reason);
Guido van Rossumd57fd912000-03-10 22:53:23 +00008208 }
8209 else {
Benjamin Peterson29060642009-01-31 22:14:21 +00008210 if (PyUnicodeTranslateError_SetStart(*exceptionObject, startpos))
8211 goto onError;
8212 if (PyUnicodeTranslateError_SetEnd(*exceptionObject, endpos))
8213 goto onError;
8214 if (PyUnicodeTranslateError_SetReason(*exceptionObject, reason))
8215 goto onError;
8216 return;
8217 onError:
8218 Py_DECREF(*exceptionObject);
8219 *exceptionObject = NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00008220 }
8221}
8222
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008223/* raises a UnicodeTranslateError */
Alexander Belopolsky40018472011-02-26 01:02:56 +00008224static void
8225raise_translate_exception(PyObject **exceptionObject,
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008226 PyObject *unicode,
Alexander Belopolsky40018472011-02-26 01:02:56 +00008227 Py_ssize_t startpos, Py_ssize_t endpos,
8228 const char *reason)
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008229{
8230 make_translate_exception(exceptionObject,
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008231 unicode, startpos, endpos, reason);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008232 if (*exceptionObject != NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00008233 PyCodec_StrictErrors(*exceptionObject);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008234}
8235
8236/* error handling callback helper:
8237 build arguments, call the callback and check the arguments,
8238 put the result into newpos and return the replacement string, which
8239 has to be freed by the caller */
Alexander Belopolsky40018472011-02-26 01:02:56 +00008240static PyObject *
8241unicode_translate_call_errorhandler(const char *errors,
8242 PyObject **errorHandler,
8243 const char *reason,
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008244 PyObject *unicode, PyObject **exceptionObject,
Alexander Belopolsky40018472011-02-26 01:02:56 +00008245 Py_ssize_t startpos, Py_ssize_t endpos,
8246 Py_ssize_t *newpos)
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008247{
Benjamin Peterson142957c2008-07-04 19:55:29 +00008248 static char *argparse = "O!n;translating error handler must return (str, int) tuple";
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008249
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00008250 Py_ssize_t i_newpos;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008251 PyObject *restuple;
8252 PyObject *resunicode;
8253
8254 if (*errorHandler == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +00008255 *errorHandler = PyCodec_LookupError(errors);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008256 if (*errorHandler == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00008257 return NULL;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008258 }
8259
8260 make_translate_exception(exceptionObject,
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008261 unicode, startpos, endpos, reason);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008262 if (*exceptionObject == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00008263 return NULL;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008264
8265 restuple = PyObject_CallFunctionObjArgs(
Benjamin Peterson29060642009-01-31 22:14:21 +00008266 *errorHandler, *exceptionObject, NULL);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008267 if (restuple == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00008268 return NULL;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008269 if (!PyTuple_Check(restuple)) {
Benjamin Petersond75fcb42009-02-19 04:22:03 +00008270 PyErr_SetString(PyExc_TypeError, &argparse[4]);
Benjamin Peterson29060642009-01-31 22:14:21 +00008271 Py_DECREF(restuple);
8272 return NULL;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008273 }
8274 if (!PyArg_ParseTuple(restuple, argparse, &PyUnicode_Type,
Benjamin Peterson29060642009-01-31 22:14:21 +00008275 &resunicode, &i_newpos)) {
8276 Py_DECREF(restuple);
8277 return NULL;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008278 }
Martin v. Löwis18e16552006-02-15 17:27:45 +00008279 if (i_newpos<0)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008280 *newpos = PyUnicode_GET_LENGTH(unicode)+i_newpos;
Martin v. Löwis18e16552006-02-15 17:27:45 +00008281 else
8282 *newpos = i_newpos;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008283 if (*newpos<0 || *newpos>PyUnicode_GET_LENGTH(unicode)) {
Benjamin Peterson29060642009-01-31 22:14:21 +00008284 PyErr_Format(PyExc_IndexError, "position %zd from error handler out of bounds", *newpos);
8285 Py_DECREF(restuple);
8286 return NULL;
Walter Dörwald2e0b18a2003-01-31 17:19:08 +00008287 }
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008288 Py_INCREF(resunicode);
8289 Py_DECREF(restuple);
8290 return resunicode;
8291}
8292
8293/* Lookup the character ch in the mapping and put the result in result,
8294 which must be decrefed by the caller.
8295 Return 0 on success, -1 on error */
Alexander Belopolsky40018472011-02-26 01:02:56 +00008296static int
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008297charmaptranslate_lookup(Py_UCS4 c, PyObject *mapping, PyObject **result)
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008298{
Christian Heimes217cfd12007-12-02 14:31:20 +00008299 PyObject *w = PyLong_FromLong((long)c);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008300 PyObject *x;
8301
8302 if (w == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00008303 return -1;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008304 x = PyObject_GetItem(mapping, w);
8305 Py_DECREF(w);
8306 if (x == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +00008307 if (PyErr_ExceptionMatches(PyExc_LookupError)) {
8308 /* No mapping found means: use 1:1 mapping. */
8309 PyErr_Clear();
8310 *result = NULL;
8311 return 0;
8312 } else
8313 return -1;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008314 }
8315 else if (x == Py_None) {
Benjamin Peterson29060642009-01-31 22:14:21 +00008316 *result = x;
8317 return 0;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008318 }
Christian Heimes217cfd12007-12-02 14:31:20 +00008319 else if (PyLong_Check(x)) {
Benjamin Peterson29060642009-01-31 22:14:21 +00008320 long value = PyLong_AS_LONG(x);
8321 long max = PyUnicode_GetMax();
8322 if (value < 0 || value > max) {
8323 PyErr_Format(PyExc_TypeError,
Guido van Rossum5a2f7e602007-10-24 21:13:09 +00008324 "character mapping must be in range(0x%x)", max+1);
Benjamin Peterson29060642009-01-31 22:14:21 +00008325 Py_DECREF(x);
8326 return -1;
8327 }
8328 *result = x;
8329 return 0;
8330 }
8331 else if (PyUnicode_Check(x)) {
8332 *result = x;
8333 return 0;
8334 }
8335 else {
8336 /* wrong return value */
8337 PyErr_SetString(PyExc_TypeError,
8338 "character mapping must return integer, None or str");
Benjamin Peterson14339b62009-01-31 16:36:08 +00008339 Py_DECREF(x);
8340 return -1;
8341 }
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008342}
8343/* ensure that *outobj is at least requiredsize characters long,
Benjamin Peterson29060642009-01-31 22:14:21 +00008344 if not reallocate and adjust various state variables.
8345 Return 0 on success, -1 on error */
Alexander Belopolsky40018472011-02-26 01:02:56 +00008346static int
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008347charmaptranslate_makespace(Py_UCS4 **outobj, Py_ssize_t *psize,
Benjamin Peterson29060642009-01-31 22:14:21 +00008348 Py_ssize_t requiredsize)
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008349{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008350 Py_ssize_t oldsize = *psize;
Kristjan Valur Jonsson85634d72012-05-31 09:37:31 +00008351 Py_UCS4 *new_outobj;
Walter Dörwald4894c302003-10-24 14:25:28 +00008352 if (requiredsize > oldsize) {
Benjamin Peterson29060642009-01-31 22:14:21 +00008353 /* exponentially overallocate to minimize reallocations */
8354 if (requiredsize < 2 * oldsize)
8355 requiredsize = 2 * oldsize;
Kristjan Valur Jonsson85634d72012-05-31 09:37:31 +00008356 new_outobj = PyMem_Realloc(*outobj, requiredsize * sizeof(Py_UCS4));
8357 if (new_outobj == 0)
Benjamin Peterson29060642009-01-31 22:14:21 +00008358 return -1;
Kristjan Valur Jonsson85634d72012-05-31 09:37:31 +00008359 *outobj = new_outobj;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008360 *psize = requiredsize;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008361 }
8362 return 0;
8363}
8364/* lookup the character, put the result in the output string and adjust
8365 various state variables. Return a new reference to the object that
8366 was put in the output buffer in *result, or Py_None, if the mapping was
8367 undefined (in which case no character was written).
8368 The called must decref result.
8369 Return 0 on success, -1 on error. */
Alexander Belopolsky40018472011-02-26 01:02:56 +00008370static int
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008371charmaptranslate_output(PyObject *input, Py_ssize_t ipos,
8372 PyObject *mapping, Py_UCS4 **output,
8373 Py_ssize_t *osize, Py_ssize_t *opos,
Alexander Belopolsky40018472011-02-26 01:02:56 +00008374 PyObject **res)
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008375{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008376 Py_UCS4 curinp = PyUnicode_READ_CHAR(input, ipos);
8377 if (charmaptranslate_lookup(curinp, mapping, res))
Benjamin Peterson29060642009-01-31 22:14:21 +00008378 return -1;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008379 if (*res==NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +00008380 /* not found => default to 1:1 mapping */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008381 (*output)[(*opos)++] = curinp;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008382 }
8383 else if (*res==Py_None)
Benjamin Peterson29060642009-01-31 22:14:21 +00008384 ;
Christian Heimes217cfd12007-12-02 14:31:20 +00008385 else if (PyLong_Check(*res)) {
Benjamin Peterson29060642009-01-31 22:14:21 +00008386 /* no overflow check, because we know that the space is enough */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008387 (*output)[(*opos)++] = (Py_UCS4)PyLong_AS_LONG(*res);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008388 }
8389 else if (PyUnicode_Check(*res)) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008390 Py_ssize_t repsize;
8391 if (PyUnicode_READY(*res) == -1)
8392 return -1;
8393 repsize = PyUnicode_GET_LENGTH(*res);
Benjamin Peterson29060642009-01-31 22:14:21 +00008394 if (repsize==1) {
8395 /* no overflow check, because we know that the space is enough */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008396 (*output)[(*opos)++] = PyUnicode_READ_CHAR(*res, 0);
Benjamin Peterson29060642009-01-31 22:14:21 +00008397 }
8398 else if (repsize!=0) {
8399 /* more than one character */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008400 Py_ssize_t requiredsize = *opos +
8401 (PyUnicode_GET_LENGTH(input) - ipos) +
Benjamin Peterson29060642009-01-31 22:14:21 +00008402 repsize - 1;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008403 Py_ssize_t i;
8404 if (charmaptranslate_makespace(output, osize, requiredsize))
Benjamin Peterson29060642009-01-31 22:14:21 +00008405 return -1;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008406 for(i = 0; i < repsize; i++)
8407 (*output)[(*opos)++] = PyUnicode_READ_CHAR(*res, i);
Benjamin Peterson29060642009-01-31 22:14:21 +00008408 }
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008409 }
8410 else
Benjamin Peterson29060642009-01-31 22:14:21 +00008411 return -1;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008412 return 0;
8413}
8414
Alexander Belopolsky40018472011-02-26 01:02:56 +00008415PyObject *
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008416_PyUnicode_TranslateCharmap(PyObject *input,
8417 PyObject *mapping,
8418 const char *errors)
Guido van Rossumd57fd912000-03-10 22:53:23 +00008419{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008420 /* input object */
8421 char *idata;
8422 Py_ssize_t size, i;
8423 int kind;
8424 /* output buffer */
8425 Py_UCS4 *output = NULL;
8426 Py_ssize_t osize;
8427 PyObject *res;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008428 /* current output position */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008429 Py_ssize_t opos;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008430 char *reason = "character maps to <undefined>";
8431 PyObject *errorHandler = NULL;
8432 PyObject *exc = NULL;
8433 /* the following variable is used for caching string comparisons
8434 * -1=not initialized, 0=unknown, 1=strict, 2=replace,
8435 * 3=ignore, 4=xmlcharrefreplace */
8436 int known_errorHandler = -1;
8437
Guido van Rossumd57fd912000-03-10 22:53:23 +00008438 if (mapping == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +00008439 PyErr_BadArgument();
8440 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00008441 }
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008442
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008443 if (PyUnicode_READY(input) == -1)
8444 return NULL;
8445 idata = (char*)PyUnicode_DATA(input);
8446 kind = PyUnicode_KIND(input);
8447 size = PyUnicode_GET_LENGTH(input);
8448 i = 0;
8449
8450 if (size == 0) {
8451 Py_INCREF(input);
8452 return input;
8453 }
8454
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008455 /* allocate enough for a simple 1:1 translation without
8456 replacements, if we need more, we'll resize */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008457 osize = size;
8458 output = PyMem_Malloc(osize * sizeof(Py_UCS4));
8459 opos = 0;
8460 if (output == NULL) {
8461 PyErr_NoMemory();
Benjamin Peterson29060642009-01-31 22:14:21 +00008462 goto onError;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008463 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00008464
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008465 while (i<size) {
Benjamin Peterson29060642009-01-31 22:14:21 +00008466 /* try to encode it */
8467 PyObject *x = NULL;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008468 if (charmaptranslate_output(input, i, mapping,
8469 &output, &osize, &opos, &x)) {
Benjamin Peterson29060642009-01-31 22:14:21 +00008470 Py_XDECREF(x);
8471 goto onError;
8472 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00008473 Py_XDECREF(x);
Benjamin Peterson29060642009-01-31 22:14:21 +00008474 if (x!=Py_None) /* it worked => adjust input pointer */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008475 ++i;
Benjamin Peterson29060642009-01-31 22:14:21 +00008476 else { /* untranslatable character */
8477 PyObject *repunicode = NULL; /* initialize to prevent gcc warning */
8478 Py_ssize_t repsize;
8479 Py_ssize_t newpos;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008480 Py_ssize_t uni2;
Benjamin Peterson29060642009-01-31 22:14:21 +00008481 /* startpos for collecting untranslatable chars */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008482 Py_ssize_t collstart = i;
8483 Py_ssize_t collend = i+1;
8484 Py_ssize_t coll;
Guido van Rossumd57fd912000-03-10 22:53:23 +00008485
Benjamin Peterson29060642009-01-31 22:14:21 +00008486 /* find all untranslatable characters */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008487 while (collend < size) {
8488 if (charmaptranslate_lookup(PyUnicode_READ(kind,idata, collend), mapping, &x))
Benjamin Peterson29060642009-01-31 22:14:21 +00008489 goto onError;
8490 Py_XDECREF(x);
8491 if (x!=Py_None)
8492 break;
8493 ++collend;
8494 }
8495 /* cache callback name lookup
8496 * (if not done yet, i.e. it's the first error) */
8497 if (known_errorHandler==-1) {
8498 if ((errors==NULL) || (!strcmp(errors, "strict")))
8499 known_errorHandler = 1;
8500 else if (!strcmp(errors, "replace"))
8501 known_errorHandler = 2;
8502 else if (!strcmp(errors, "ignore"))
8503 known_errorHandler = 3;
8504 else if (!strcmp(errors, "xmlcharrefreplace"))
8505 known_errorHandler = 4;
8506 else
8507 known_errorHandler = 0;
8508 }
8509 switch (known_errorHandler) {
8510 case 1: /* strict */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008511 raise_translate_exception(&exc, input, collstart,
8512 collend, reason);
Benjamin Peterson14339b62009-01-31 16:36:08 +00008513 goto onError;
Benjamin Peterson29060642009-01-31 22:14:21 +00008514 case 2: /* replace */
8515 /* No need to check for space, this is a 1:1 replacement */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008516 for (coll = collstart; coll<collend; coll++)
8517 output[opos++] = '?';
Benjamin Peterson29060642009-01-31 22:14:21 +00008518 /* fall through */
8519 case 3: /* ignore */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008520 i = collend;
Benjamin Peterson29060642009-01-31 22:14:21 +00008521 break;
8522 case 4: /* xmlcharrefreplace */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008523 /* generate replacement (temporarily (mis)uses i) */
8524 for (i = collstart; i < collend; ++i) {
Benjamin Peterson29060642009-01-31 22:14:21 +00008525 char buffer[2+29+1+1];
8526 char *cp;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008527 sprintf(buffer, "&#%d;", PyUnicode_READ(kind, idata, i));
8528 if (charmaptranslate_makespace(&output, &osize,
8529 opos+strlen(buffer)+(size-collend)))
Benjamin Peterson29060642009-01-31 22:14:21 +00008530 goto onError;
8531 for (cp = buffer; *cp; ++cp)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008532 output[opos++] = *cp;
Benjamin Peterson29060642009-01-31 22:14:21 +00008533 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008534 i = collend;
Benjamin Peterson29060642009-01-31 22:14:21 +00008535 break;
8536 default:
8537 repunicode = unicode_translate_call_errorhandler(errors, &errorHandler,
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008538 reason, input, &exc,
8539 collstart, collend, &newpos);
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01008540 if (repunicode == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00008541 goto onError;
Benjamin Peterson9ca3ffa2012-01-01 16:04:29 -06008542 if (PyUnicode_READY(repunicode) == -1) {
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01008543 Py_DECREF(repunicode);
8544 goto onError;
8545 }
Benjamin Peterson29060642009-01-31 22:14:21 +00008546 /* generate replacement */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008547 repsize = PyUnicode_GET_LENGTH(repunicode);
8548 if (charmaptranslate_makespace(&output, &osize,
8549 opos+repsize+(size-collend))) {
Benjamin Peterson29060642009-01-31 22:14:21 +00008550 Py_DECREF(repunicode);
8551 goto onError;
8552 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008553 for (uni2 = 0; repsize-->0; ++uni2)
8554 output[opos++] = PyUnicode_READ_CHAR(repunicode, uni2);
8555 i = newpos;
Benjamin Peterson29060642009-01-31 22:14:21 +00008556 Py_DECREF(repunicode);
Benjamin Peterson14339b62009-01-31 16:36:08 +00008557 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00008558 }
8559 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008560 res = PyUnicode_FromKindAndData(PyUnicode_4BYTE_KIND, output, opos);
8561 if (!res)
8562 goto onError;
8563 PyMem_Free(output);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008564 Py_XDECREF(exc);
8565 Py_XDECREF(errorHandler);
8566 return res;
Guido van Rossumd57fd912000-03-10 22:53:23 +00008567
Benjamin Peterson29060642009-01-31 22:14:21 +00008568 onError:
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008569 PyMem_Free(output);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008570 Py_XDECREF(exc);
8571 Py_XDECREF(errorHandler);
Guido van Rossumd57fd912000-03-10 22:53:23 +00008572 return NULL;
8573}
8574
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008575/* Deprecated. Use PyUnicode_Translate instead. */
8576PyObject *
8577PyUnicode_TranslateCharmap(const Py_UNICODE *p,
8578 Py_ssize_t size,
8579 PyObject *mapping,
8580 const char *errors)
8581{
Christian Heimes5f520f42012-09-11 14:03:25 +02008582 PyObject *result;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008583 PyObject *unicode = PyUnicode_FromUnicode(p, size);
8584 if (!unicode)
8585 return NULL;
Christian Heimes5f520f42012-09-11 14:03:25 +02008586 result = _PyUnicode_TranslateCharmap(unicode, mapping, errors);
8587 Py_DECREF(unicode);
8588 return result;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008589}
8590
Alexander Belopolsky40018472011-02-26 01:02:56 +00008591PyObject *
8592PyUnicode_Translate(PyObject *str,
8593 PyObject *mapping,
8594 const char *errors)
Guido van Rossumd57fd912000-03-10 22:53:23 +00008595{
8596 PyObject *result;
Tim Petersced69f82003-09-16 20:30:58 +00008597
Guido van Rossumd57fd912000-03-10 22:53:23 +00008598 str = PyUnicode_FromObject(str);
8599 if (str == NULL)
Christian Heimes5f520f42012-09-11 14:03:25 +02008600 return NULL;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008601 result = _PyUnicode_TranslateCharmap(str, mapping, errors);
Guido van Rossumd57fd912000-03-10 22:53:23 +00008602 Py_DECREF(str);
8603 return result;
Guido van Rossumd57fd912000-03-10 22:53:23 +00008604}
Tim Petersced69f82003-09-16 20:30:58 +00008605
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008606static Py_UCS4
Victor Stinner9310abb2011-10-05 00:59:23 +02008607fix_decimal_and_space_to_ascii(PyObject *self)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008608{
8609 /* No need to call PyUnicode_READY(self) because this function is only
8610 called as a callback from fixup() which does it already. */
8611 const Py_ssize_t len = PyUnicode_GET_LENGTH(self);
8612 const int kind = PyUnicode_KIND(self);
8613 void *data = PyUnicode_DATA(self);
Victor Stinnere6abb482012-05-02 01:15:40 +02008614 Py_UCS4 maxchar = 127, ch, fixed;
Benjamin Peterson821e4cf2012-01-12 15:40:18 -05008615 int modified = 0;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008616 Py_ssize_t i;
8617
8618 for (i = 0; i < len; ++i) {
8619 ch = PyUnicode_READ(kind, data, i);
8620 fixed = 0;
8621 if (ch > 127) {
8622 if (Py_UNICODE_ISSPACE(ch))
8623 fixed = ' ';
8624 else {
8625 const int decimal = Py_UNICODE_TODECIMAL(ch);
8626 if (decimal >= 0)
8627 fixed = '0' + decimal;
8628 }
8629 if (fixed != 0) {
Benjamin Peterson821e4cf2012-01-12 15:40:18 -05008630 modified = 1;
Benjamin Peterson7e303732013-06-10 09:19:46 -07008631 maxchar = Py_MAX(maxchar, fixed);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008632 PyUnicode_WRITE(kind, data, i, fixed);
8633 }
Victor Stinnere6abb482012-05-02 01:15:40 +02008634 else
Benjamin Peterson7e303732013-06-10 09:19:46 -07008635 maxchar = Py_MAX(maxchar, ch);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008636 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008637 }
8638
Benjamin Peterson821e4cf2012-01-12 15:40:18 -05008639 return (modified) ? maxchar : 0;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008640}
8641
8642PyObject *
8643_PyUnicode_TransformDecimalAndSpaceToASCII(PyObject *unicode)
8644{
8645 if (!PyUnicode_Check(unicode)) {
8646 PyErr_BadInternalCall();
8647 return NULL;
8648 }
8649 if (PyUnicode_READY(unicode) == -1)
8650 return NULL;
8651 if (PyUnicode_MAX_CHAR_VALUE(unicode) <= 127) {
8652 /* If the string is already ASCII, just return the same string */
8653 Py_INCREF(unicode);
8654 return unicode;
8655 }
Victor Stinner9310abb2011-10-05 00:59:23 +02008656 return fixup(unicode, fix_decimal_and_space_to_ascii);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008657}
8658
Alexander Belopolsky942af5a2010-12-04 03:38:46 +00008659PyObject *
8660PyUnicode_TransformDecimalToASCII(Py_UNICODE *s,
8661 Py_ssize_t length)
8662{
Victor Stinnerf0124502011-11-21 23:12:56 +01008663 PyObject *decimal;
Alexander Belopolsky942af5a2010-12-04 03:38:46 +00008664 Py_ssize_t i;
Victor Stinnerf0124502011-11-21 23:12:56 +01008665 Py_UCS4 maxchar;
8666 enum PyUnicode_Kind kind;
8667 void *data;
8668
Victor Stinner99d7ad02012-02-22 13:37:39 +01008669 maxchar = 127;
Alexander Belopolsky942af5a2010-12-04 03:38:46 +00008670 for (i = 0; i < length; i++) {
Victor Stinnerf0124502011-11-21 23:12:56 +01008671 Py_UNICODE ch = s[i];
Alexander Belopolsky942af5a2010-12-04 03:38:46 +00008672 if (ch > 127) {
8673 int decimal = Py_UNICODE_TODECIMAL(ch);
8674 if (decimal >= 0)
Victor Stinnerf0124502011-11-21 23:12:56 +01008675 ch = '0' + decimal;
Benjamin Peterson7e303732013-06-10 09:19:46 -07008676 maxchar = Py_MAX(maxchar, ch);
Alexander Belopolsky942af5a2010-12-04 03:38:46 +00008677 }
8678 }
Victor Stinnerf0124502011-11-21 23:12:56 +01008679
8680 /* Copy to a new string */
8681 decimal = PyUnicode_New(length, maxchar);
8682 if (decimal == NULL)
8683 return decimal;
8684 kind = PyUnicode_KIND(decimal);
8685 data = PyUnicode_DATA(decimal);
8686 /* Iterate over code points */
8687 for (i = 0; i < length; i++) {
8688 Py_UNICODE ch = s[i];
8689 if (ch > 127) {
8690 int decimal = Py_UNICODE_TODECIMAL(ch);
8691 if (decimal >= 0)
8692 ch = '0' + decimal;
8693 }
8694 PyUnicode_WRITE(kind, data, i, ch);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008695 }
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01008696 return unicode_result(decimal);
Alexander Belopolsky942af5a2010-12-04 03:38:46 +00008697}
Guido van Rossum9e896b32000-04-05 20:11:21 +00008698/* --- Decimal Encoder ---------------------------------------------------- */
8699
Alexander Belopolsky40018472011-02-26 01:02:56 +00008700int
8701PyUnicode_EncodeDecimal(Py_UNICODE *s,
8702 Py_ssize_t length,
8703 char *output,
8704 const char *errors)
Guido van Rossum9e896b32000-04-05 20:11:21 +00008705{
Martin v. Löwis23e275b2011-11-02 18:02:51 +01008706 PyObject *unicode;
Victor Stinner6345be92011-11-25 20:09:01 +01008707 Py_ssize_t i;
Victor Stinner42bf7752011-11-21 22:52:58 +01008708 enum PyUnicode_Kind kind;
8709 void *data;
Guido van Rossum9e896b32000-04-05 20:11:21 +00008710
8711 if (output == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +00008712 PyErr_BadArgument();
8713 return -1;
Guido van Rossum9e896b32000-04-05 20:11:21 +00008714 }
8715
Victor Stinner42bf7752011-11-21 22:52:58 +01008716 unicode = PyUnicode_FromUnicode(s, length);
8717 if (unicode == NULL)
8718 return -1;
8719
Benjamin Petersonbac79492012-01-14 13:34:47 -05008720 if (PyUnicode_READY(unicode) == -1) {
Victor Stinner6345be92011-11-25 20:09:01 +01008721 Py_DECREF(unicode);
8722 return -1;
8723 }
Victor Stinner42bf7752011-11-21 22:52:58 +01008724 kind = PyUnicode_KIND(unicode);
8725 data = PyUnicode_DATA(unicode);
8726
Victor Stinnerb84d7232011-11-22 01:50:07 +01008727 for (i=0; i < length; ) {
Victor Stinner6345be92011-11-25 20:09:01 +01008728 PyObject *exc;
8729 Py_UCS4 ch;
Benjamin Peterson29060642009-01-31 22:14:21 +00008730 int decimal;
Victor Stinner6345be92011-11-25 20:09:01 +01008731 Py_ssize_t startpos;
8732
8733 ch = PyUnicode_READ(kind, data, i);
Tim Petersced69f82003-09-16 20:30:58 +00008734
Benjamin Peterson29060642009-01-31 22:14:21 +00008735 if (Py_UNICODE_ISSPACE(ch)) {
Benjamin Peterson14339b62009-01-31 16:36:08 +00008736 *output++ = ' ';
Victor Stinnerb84d7232011-11-22 01:50:07 +01008737 i++;
Benjamin Peterson29060642009-01-31 22:14:21 +00008738 continue;
Benjamin Peterson14339b62009-01-31 16:36:08 +00008739 }
Benjamin Peterson29060642009-01-31 22:14:21 +00008740 decimal = Py_UNICODE_TODECIMAL(ch);
8741 if (decimal >= 0) {
8742 *output++ = '0' + decimal;
Victor Stinnerb84d7232011-11-22 01:50:07 +01008743 i++;
Benjamin Peterson29060642009-01-31 22:14:21 +00008744 continue;
8745 }
8746 if (0 < ch && ch < 256) {
8747 *output++ = (char)ch;
Victor Stinnerb84d7232011-11-22 01:50:07 +01008748 i++;
Benjamin Peterson29060642009-01-31 22:14:21 +00008749 continue;
8750 }
Victor Stinner6345be92011-11-25 20:09:01 +01008751
Victor Stinner42bf7752011-11-21 22:52:58 +01008752 startpos = i;
Victor Stinner6345be92011-11-25 20:09:01 +01008753 exc = NULL;
8754 raise_encode_exception(&exc, "decimal", unicode,
8755 startpos, startpos+1,
8756 "invalid decimal Unicode string");
8757 Py_XDECREF(exc);
8758 Py_DECREF(unicode);
8759 return -1;
Guido van Rossum9e896b32000-04-05 20:11:21 +00008760 }
8761 /* 0-terminate the output string */
8762 *output++ = '\0';
Victor Stinner42bf7752011-11-21 22:52:58 +01008763 Py_DECREF(unicode);
Guido van Rossum9e896b32000-04-05 20:11:21 +00008764 return 0;
Guido van Rossum9e896b32000-04-05 20:11:21 +00008765}
8766
Guido van Rossumd57fd912000-03-10 22:53:23 +00008767/* --- Helpers ------------------------------------------------------------ */
8768
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008769static Py_ssize_t
Victor Stinner794d5672011-10-10 03:21:36 +02008770any_find_slice(int direction, PyObject* s1, PyObject* s2,
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008771 Py_ssize_t start,
8772 Py_ssize_t end)
8773{
8774 int kind1, kind2, kind;
8775 void *buf1, *buf2;
8776 Py_ssize_t len1, len2, result;
8777
8778 kind1 = PyUnicode_KIND(s1);
8779 kind2 = PyUnicode_KIND(s2);
8780 kind = kind1 > kind2 ? kind1 : kind2;
8781 buf1 = PyUnicode_DATA(s1);
8782 buf2 = PyUnicode_DATA(s2);
8783 if (kind1 != kind)
8784 buf1 = _PyUnicode_AsKind(s1, kind);
8785 if (!buf1)
8786 return -2;
8787 if (kind2 != kind)
8788 buf2 = _PyUnicode_AsKind(s2, kind);
8789 if (!buf2) {
8790 if (kind1 != kind) PyMem_Free(buf1);
8791 return -2;
8792 }
8793 len1 = PyUnicode_GET_LENGTH(s1);
8794 len2 = PyUnicode_GET_LENGTH(s2);
8795
Victor Stinner794d5672011-10-10 03:21:36 +02008796 if (direction > 0) {
Benjamin Petersonead6b532011-12-20 17:23:42 -06008797 switch (kind) {
Victor Stinner794d5672011-10-10 03:21:36 +02008798 case PyUnicode_1BYTE_KIND:
8799 if (PyUnicode_IS_ASCII(s1) && PyUnicode_IS_ASCII(s2))
8800 result = asciilib_find_slice(buf1, len1, buf2, len2, start, end);
8801 else
8802 result = ucs1lib_find_slice(buf1, len1, buf2, len2, start, end);
8803 break;
8804 case PyUnicode_2BYTE_KIND:
8805 result = ucs2lib_find_slice(buf1, len1, buf2, len2, start, end);
8806 break;
8807 case PyUnicode_4BYTE_KIND:
8808 result = ucs4lib_find_slice(buf1, len1, buf2, len2, start, end);
8809 break;
8810 default:
8811 assert(0); result = -2;
8812 }
8813 }
8814 else {
Benjamin Petersonead6b532011-12-20 17:23:42 -06008815 switch (kind) {
Victor Stinner794d5672011-10-10 03:21:36 +02008816 case PyUnicode_1BYTE_KIND:
8817 if (PyUnicode_IS_ASCII(s1) && PyUnicode_IS_ASCII(s2))
8818 result = asciilib_rfind_slice(buf1, len1, buf2, len2, start, end);
8819 else
8820 result = ucs1lib_rfind_slice(buf1, len1, buf2, len2, start, end);
8821 break;
8822 case PyUnicode_2BYTE_KIND:
8823 result = ucs2lib_rfind_slice(buf1, len1, buf2, len2, start, end);
8824 break;
8825 case PyUnicode_4BYTE_KIND:
8826 result = ucs4lib_rfind_slice(buf1, len1, buf2, len2, start, end);
8827 break;
8828 default:
8829 assert(0); result = -2;
8830 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008831 }
8832
8833 if (kind1 != kind)
8834 PyMem_Free(buf1);
8835 if (kind2 != kind)
8836 PyMem_Free(buf2);
8837
8838 return result;
8839}
8840
8841Py_ssize_t
Victor Stinner41a863c2012-02-24 00:37:51 +01008842_PyUnicode_InsertThousandsGrouping(
8843 PyObject *unicode, Py_ssize_t index,
8844 Py_ssize_t n_buffer,
8845 void *digits, Py_ssize_t n_digits,
8846 Py_ssize_t min_width,
8847 const char *grouping, PyObject *thousands_sep,
8848 Py_UCS4 *maxchar)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008849{
Victor Stinner41a863c2012-02-24 00:37:51 +01008850 unsigned int kind, thousands_sep_kind;
Antoine Pitrou842c0f12012-02-24 13:30:46 +01008851 char *data, *thousands_sep_data;
Victor Stinner41a863c2012-02-24 00:37:51 +01008852 Py_ssize_t thousands_sep_len;
8853 Py_ssize_t len;
8854
8855 if (unicode != NULL) {
8856 kind = PyUnicode_KIND(unicode);
Antoine Pitrou842c0f12012-02-24 13:30:46 +01008857 data = (char *) PyUnicode_DATA(unicode) + index * kind;
Victor Stinner41a863c2012-02-24 00:37:51 +01008858 }
8859 else {
8860 kind = PyUnicode_1BYTE_KIND;
8861 data = NULL;
8862 }
8863 thousands_sep_kind = PyUnicode_KIND(thousands_sep);
8864 thousands_sep_data = PyUnicode_DATA(thousands_sep);
8865 thousands_sep_len = PyUnicode_GET_LENGTH(thousands_sep);
8866 if (unicode != NULL && thousands_sep_kind != kind) {
Victor Stinner90f50d42012-02-24 01:44:47 +01008867 if (thousands_sep_kind < kind) {
8868 thousands_sep_data = _PyUnicode_AsKind(thousands_sep, kind);
8869 if (!thousands_sep_data)
8870 return -1;
8871 }
8872 else {
8873 data = _PyUnicode_AsKind(unicode, thousands_sep_kind);
8874 if (!data)
8875 return -1;
8876 }
Victor Stinner41a863c2012-02-24 00:37:51 +01008877 }
8878
Benjamin Petersonead6b532011-12-20 17:23:42 -06008879 switch (kind) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008880 case PyUnicode_1BYTE_KIND:
Victor Stinnerc3cec782011-10-05 21:24:08 +02008881 if (unicode != NULL && PyUnicode_IS_ASCII(unicode))
Victor Stinner41a863c2012-02-24 00:37:51 +01008882 len = asciilib_InsertThousandsGrouping(
Antoine Pitrou842c0f12012-02-24 13:30:46 +01008883 (Py_UCS1 *) data, n_buffer, (Py_UCS1 *) digits, n_digits,
Victor Stinner41a863c2012-02-24 00:37:51 +01008884 min_width, grouping,
Antoine Pitrou842c0f12012-02-24 13:30:46 +01008885 (Py_UCS1 *) thousands_sep_data, thousands_sep_len);
Victor Stinnerc3cec782011-10-05 21:24:08 +02008886 else
Victor Stinner41a863c2012-02-24 00:37:51 +01008887 len = ucs1lib_InsertThousandsGrouping(
Victor Stinnerc3cec782011-10-05 21:24:08 +02008888 (Py_UCS1*)data, n_buffer, (Py_UCS1*)digits, n_digits,
Victor Stinner41a863c2012-02-24 00:37:51 +01008889 min_width, grouping,
Antoine Pitrou842c0f12012-02-24 13:30:46 +01008890 (Py_UCS1 *) thousands_sep_data, thousands_sep_len);
Victor Stinner41a863c2012-02-24 00:37:51 +01008891 break;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008892 case PyUnicode_2BYTE_KIND:
Victor Stinner41a863c2012-02-24 00:37:51 +01008893 len = ucs2lib_InsertThousandsGrouping(
Antoine Pitrou842c0f12012-02-24 13:30:46 +01008894 (Py_UCS2 *) data, n_buffer, (Py_UCS2 *) digits, n_digits,
Victor Stinner41a863c2012-02-24 00:37:51 +01008895 min_width, grouping,
Antoine Pitrou842c0f12012-02-24 13:30:46 +01008896 (Py_UCS2 *) thousands_sep_data, thousands_sep_len);
Victor Stinner41a863c2012-02-24 00:37:51 +01008897 break;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008898 case PyUnicode_4BYTE_KIND:
Victor Stinner41a863c2012-02-24 00:37:51 +01008899 len = ucs4lib_InsertThousandsGrouping(
Antoine Pitrou842c0f12012-02-24 13:30:46 +01008900 (Py_UCS4 *) data, n_buffer, (Py_UCS4 *) digits, n_digits,
Victor Stinner41a863c2012-02-24 00:37:51 +01008901 min_width, grouping,
Antoine Pitrou842c0f12012-02-24 13:30:46 +01008902 (Py_UCS4 *) thousands_sep_data, thousands_sep_len);
Victor Stinner41a863c2012-02-24 00:37:51 +01008903 break;
8904 default:
8905 assert(0);
8906 return -1;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008907 }
Victor Stinner90f50d42012-02-24 01:44:47 +01008908 if (unicode != NULL && thousands_sep_kind != kind) {
8909 if (thousands_sep_kind < kind)
8910 PyMem_Free(thousands_sep_data);
8911 else
8912 PyMem_Free(data);
8913 }
Victor Stinner41a863c2012-02-24 00:37:51 +01008914 if (unicode == NULL) {
8915 *maxchar = 127;
8916 if (len != n_digits) {
Benjamin Peterson7e303732013-06-10 09:19:46 -07008917 *maxchar = Py_MAX(*maxchar,
Victor Stinnere6abb482012-05-02 01:15:40 +02008918 PyUnicode_MAX_CHAR_VALUE(thousands_sep));
Victor Stinner41a863c2012-02-24 00:37:51 +01008919 }
8920 }
8921 return len;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008922}
8923
8924
Thomas Wouters477c8d52006-05-27 19:21:47 +00008925/* helper macro to fixup start/end slice values */
Antoine Pitrouf2c54842010-01-13 08:07:53 +00008926#define ADJUST_INDICES(start, end, len) \
8927 if (end > len) \
8928 end = len; \
8929 else if (end < 0) { \
8930 end += len; \
8931 if (end < 0) \
8932 end = 0; \
8933 } \
8934 if (start < 0) { \
8935 start += len; \
8936 if (start < 0) \
8937 start = 0; \
8938 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00008939
Alexander Belopolsky40018472011-02-26 01:02:56 +00008940Py_ssize_t
8941PyUnicode_Count(PyObject *str,
8942 PyObject *substr,
8943 Py_ssize_t start,
8944 Py_ssize_t end)
Guido van Rossumd57fd912000-03-10 22:53:23 +00008945{
Martin v. Löwis18e16552006-02-15 17:27:45 +00008946 Py_ssize_t result;
Victor Stinner9db1a8b2011-10-23 20:04:37 +02008947 PyObject* str_obj;
8948 PyObject* sub_obj;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008949 int kind1, kind2, kind;
8950 void *buf1 = NULL, *buf2 = NULL;
8951 Py_ssize_t len1, len2;
Tim Petersced69f82003-09-16 20:30:58 +00008952
Victor Stinner9db1a8b2011-10-23 20:04:37 +02008953 str_obj = PyUnicode_FromObject(str);
Benjamin Peterson22a29702012-01-02 09:00:30 -06008954 if (!str_obj)
Benjamin Peterson29060642009-01-31 22:14:21 +00008955 return -1;
Victor Stinner9db1a8b2011-10-23 20:04:37 +02008956 sub_obj = PyUnicode_FromObject(substr);
Benjamin Peterson22a29702012-01-02 09:00:30 -06008957 if (!sub_obj) {
8958 Py_DECREF(str_obj);
8959 return -1;
8960 }
Benjamin Peterson4c13a4a2012-01-02 09:07:38 -06008961 if (PyUnicode_READY(sub_obj) == -1 || PyUnicode_READY(str_obj) == -1) {
Benjamin Peterson5e458f52012-01-02 10:12:13 -06008962 Py_DECREF(sub_obj);
Benjamin Peterson29060642009-01-31 22:14:21 +00008963 Py_DECREF(str_obj);
8964 return -1;
Guido van Rossumd57fd912000-03-10 22:53:23 +00008965 }
Tim Petersced69f82003-09-16 20:30:58 +00008966
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008967 kind1 = PyUnicode_KIND(str_obj);
8968 kind2 = PyUnicode_KIND(sub_obj);
Antoine Pitroue45c0c52012-05-12 15:49:07 +02008969 kind = kind1;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008970 buf1 = PyUnicode_DATA(str_obj);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008971 buf2 = PyUnicode_DATA(sub_obj);
Benjamin Peterson1ff2e352012-05-11 17:41:20 -05008972 if (kind2 != kind) {
Antoine Pitrou758153b2012-05-12 15:51:51 +02008973 if (kind2 > kind) {
8974 Py_DECREF(sub_obj);
8975 Py_DECREF(str_obj);
Antoine Pitroue45c0c52012-05-12 15:49:07 +02008976 return 0;
Antoine Pitrou758153b2012-05-12 15:51:51 +02008977 }
Victor Stinner7931d9a2011-11-04 00:22:48 +01008978 buf2 = _PyUnicode_AsKind(sub_obj, kind);
Benjamin Peterson1ff2e352012-05-11 17:41:20 -05008979 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008980 if (!buf2)
8981 goto onError;
8982 len1 = PyUnicode_GET_LENGTH(str_obj);
8983 len2 = PyUnicode_GET_LENGTH(sub_obj);
8984
8985 ADJUST_INDICES(start, end, len1);
Benjamin Petersonead6b532011-12-20 17:23:42 -06008986 switch (kind) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008987 case PyUnicode_1BYTE_KIND:
Victor Stinnerc3cec782011-10-05 21:24:08 +02008988 if (PyUnicode_IS_ASCII(str_obj) && PyUnicode_IS_ASCII(sub_obj))
8989 result = asciilib_count(
8990 ((Py_UCS1*)buf1) + start, end - start,
8991 buf2, len2, PY_SSIZE_T_MAX
8992 );
8993 else
8994 result = ucs1lib_count(
8995 ((Py_UCS1*)buf1) + start, end - start,
8996 buf2, len2, PY_SSIZE_T_MAX
8997 );
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008998 break;
8999 case PyUnicode_2BYTE_KIND:
9000 result = ucs2lib_count(
9001 ((Py_UCS2*)buf1) + start, end - start,
9002 buf2, len2, PY_SSIZE_T_MAX
9003 );
9004 break;
9005 case PyUnicode_4BYTE_KIND:
9006 result = ucs4lib_count(
9007 ((Py_UCS4*)buf1) + start, end - start,
9008 buf2, len2, PY_SSIZE_T_MAX
9009 );
9010 break;
9011 default:
9012 assert(0); result = 0;
9013 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00009014
9015 Py_DECREF(sub_obj);
9016 Py_DECREF(str_obj);
9017
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009018 if (kind2 != kind)
9019 PyMem_Free(buf2);
9020
Guido van Rossumd57fd912000-03-10 22:53:23 +00009021 return result;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009022 onError:
9023 Py_DECREF(sub_obj);
9024 Py_DECREF(str_obj);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009025 if (kind2 != kind && buf2)
9026 PyMem_Free(buf2);
9027 return -1;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009028}
9029
Alexander Belopolsky40018472011-02-26 01:02:56 +00009030Py_ssize_t
9031PyUnicode_Find(PyObject *str,
9032 PyObject *sub,
9033 Py_ssize_t start,
9034 Py_ssize_t end,
9035 int direction)
Guido van Rossumd57fd912000-03-10 22:53:23 +00009036{
Martin v. Löwis18e16552006-02-15 17:27:45 +00009037 Py_ssize_t result;
Tim Petersced69f82003-09-16 20:30:58 +00009038
Guido van Rossumd57fd912000-03-10 22:53:23 +00009039 str = PyUnicode_FromObject(str);
Benjamin Peterson22a29702012-01-02 09:00:30 -06009040 if (!str)
Benjamin Peterson29060642009-01-31 22:14:21 +00009041 return -2;
Thomas Wouters477c8d52006-05-27 19:21:47 +00009042 sub = PyUnicode_FromObject(sub);
Benjamin Peterson22a29702012-01-02 09:00:30 -06009043 if (!sub) {
9044 Py_DECREF(str);
9045 return -2;
9046 }
9047 if (PyUnicode_READY(sub) == -1 || PyUnicode_READY(str) == -1) {
9048 Py_DECREF(sub);
Benjamin Peterson29060642009-01-31 22:14:21 +00009049 Py_DECREF(str);
9050 return -2;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009051 }
Tim Petersced69f82003-09-16 20:30:58 +00009052
Victor Stinner794d5672011-10-10 03:21:36 +02009053 result = any_find_slice(direction,
9054 str, sub, start, end
9055 );
Thomas Wouters477c8d52006-05-27 19:21:47 +00009056
Guido van Rossumd57fd912000-03-10 22:53:23 +00009057 Py_DECREF(str);
Thomas Wouters477c8d52006-05-27 19:21:47 +00009058 Py_DECREF(sub);
9059
Guido van Rossumd57fd912000-03-10 22:53:23 +00009060 return result;
9061}
9062
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009063Py_ssize_t
9064PyUnicode_FindChar(PyObject *str, Py_UCS4 ch,
9065 Py_ssize_t start, Py_ssize_t end,
9066 int direction)
9067{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009068 int kind;
Antoine Pitrouf0b934b2011-10-13 18:55:09 +02009069 Py_ssize_t result;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009070 if (PyUnicode_READY(str) == -1)
9071 return -2;
Victor Stinner267aa242011-10-02 01:08:37 +02009072 if (start < 0 || end < 0) {
9073 PyErr_SetString(PyExc_IndexError, "string index out of range");
9074 return -2;
9075 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009076 if (end > PyUnicode_GET_LENGTH(str))
9077 end = PyUnicode_GET_LENGTH(str);
9078 kind = PyUnicode_KIND(str);
Antoine Pitrouf0b934b2011-10-13 18:55:09 +02009079 result = findchar(PyUnicode_1BYTE_DATA(str) + kind*start,
9080 kind, end-start, ch, direction);
9081 if (result == -1)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009082 return -1;
Antoine Pitrouf0b934b2011-10-13 18:55:09 +02009083 else
9084 return start + result;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009085}
9086
Alexander Belopolsky40018472011-02-26 01:02:56 +00009087static int
Victor Stinner9db1a8b2011-10-23 20:04:37 +02009088tailmatch(PyObject *self,
9089 PyObject *substring,
Alexander Belopolsky40018472011-02-26 01:02:56 +00009090 Py_ssize_t start,
9091 Py_ssize_t end,
9092 int direction)
Guido van Rossumd57fd912000-03-10 22:53:23 +00009093{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009094 int kind_self;
9095 int kind_sub;
9096 void *data_self;
9097 void *data_sub;
9098 Py_ssize_t offset;
9099 Py_ssize_t i;
9100 Py_ssize_t end_sub;
9101
9102 if (PyUnicode_READY(self) == -1 ||
9103 PyUnicode_READY(substring) == -1)
9104 return 0;
9105
9106 if (PyUnicode_GET_LENGTH(substring) == 0)
Guido van Rossumd57fd912000-03-10 22:53:23 +00009107 return 1;
9108
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009109 ADJUST_INDICES(start, end, PyUnicode_GET_LENGTH(self));
9110 end -= PyUnicode_GET_LENGTH(substring);
Guido van Rossumd57fd912000-03-10 22:53:23 +00009111 if (end < start)
Benjamin Peterson29060642009-01-31 22:14:21 +00009112 return 0;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009113
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009114 kind_self = PyUnicode_KIND(self);
9115 data_self = PyUnicode_DATA(self);
9116 kind_sub = PyUnicode_KIND(substring);
9117 data_sub = PyUnicode_DATA(substring);
9118 end_sub = PyUnicode_GET_LENGTH(substring) - 1;
9119
9120 if (direction > 0)
9121 offset = end;
9122 else
9123 offset = start;
9124
9125 if (PyUnicode_READ(kind_self, data_self, offset) ==
9126 PyUnicode_READ(kind_sub, data_sub, 0) &&
9127 PyUnicode_READ(kind_self, data_self, offset + end_sub) ==
9128 PyUnicode_READ(kind_sub, data_sub, end_sub)) {
9129 /* If both are of the same kind, memcmp is sufficient */
9130 if (kind_self == kind_sub) {
9131 return ! memcmp((char *)data_self +
Martin v. Löwisc47adb02011-10-07 20:55:35 +02009132 (offset * PyUnicode_KIND(substring)),
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009133 data_sub,
9134 PyUnicode_GET_LENGTH(substring) *
Martin v. Löwisc47adb02011-10-07 20:55:35 +02009135 PyUnicode_KIND(substring));
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009136 }
9137 /* otherwise we have to compare each character by first accesing it */
9138 else {
9139 /* We do not need to compare 0 and len(substring)-1 because
9140 the if statement above ensured already that they are equal
9141 when we end up here. */
Antoine Pitrou057119b2012-09-02 17:56:33 +02009142 /* TODO: honor direction and do a forward or backwards search */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009143 for (i = 1; i < end_sub; ++i) {
9144 if (PyUnicode_READ(kind_self, data_self, offset + i) !=
9145 PyUnicode_READ(kind_sub, data_sub, i))
9146 return 0;
9147 }
Benjamin Peterson29060642009-01-31 22:14:21 +00009148 return 1;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009149 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00009150 }
9151
9152 return 0;
9153}
9154
Alexander Belopolsky40018472011-02-26 01:02:56 +00009155Py_ssize_t
9156PyUnicode_Tailmatch(PyObject *str,
9157 PyObject *substr,
9158 Py_ssize_t start,
9159 Py_ssize_t end,
9160 int direction)
Guido van Rossumd57fd912000-03-10 22:53:23 +00009161{
Martin v. Löwis18e16552006-02-15 17:27:45 +00009162 Py_ssize_t result;
Tim Petersced69f82003-09-16 20:30:58 +00009163
Guido van Rossumd57fd912000-03-10 22:53:23 +00009164 str = PyUnicode_FromObject(str);
9165 if (str == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00009166 return -1;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009167 substr = PyUnicode_FromObject(substr);
9168 if (substr == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +00009169 Py_DECREF(str);
9170 return -1;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009171 }
Tim Petersced69f82003-09-16 20:30:58 +00009172
Victor Stinner9db1a8b2011-10-23 20:04:37 +02009173 result = tailmatch(str, substr,
Benjamin Peterson29060642009-01-31 22:14:21 +00009174 start, end, direction);
Guido van Rossumd57fd912000-03-10 22:53:23 +00009175 Py_DECREF(str);
9176 Py_DECREF(substr);
9177 return result;
9178}
9179
Guido van Rossumd57fd912000-03-10 22:53:23 +00009180/* Apply fixfct filter to the Unicode object self and return a
9181 reference to the modified object */
9182
Alexander Belopolsky40018472011-02-26 01:02:56 +00009183static PyObject *
Victor Stinner9310abb2011-10-05 00:59:23 +02009184fixup(PyObject *self,
9185 Py_UCS4 (*fixfct)(PyObject *s))
Guido van Rossumd57fd912000-03-10 22:53:23 +00009186{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009187 PyObject *u;
9188 Py_UCS4 maxchar_old, maxchar_new = 0;
Victor Stinnereaab6042011-12-11 22:22:39 +01009189 PyObject *v;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009190
Victor Stinnerbf6e5602011-12-12 01:53:47 +01009191 u = _PyUnicode_Copy(self);
Guido van Rossumd57fd912000-03-10 22:53:23 +00009192 if (u == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00009193 return NULL;
Victor Stinner87af4f22011-11-21 23:03:47 +01009194 maxchar_old = PyUnicode_MAX_CHAR_VALUE(u);
Marc-André Lemburg8155e0e2001-04-23 14:44:21 +00009195
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009196 /* fix functions return the new maximum character in a string,
9197 if the kind of the resulting unicode object does not change,
9198 everything is fine. Otherwise we need to change the string kind
9199 and re-run the fix function. */
Victor Stinner9310abb2011-10-05 00:59:23 +02009200 maxchar_new = fixfct(u);
Victor Stinnereaab6042011-12-11 22:22:39 +01009201
9202 if (maxchar_new == 0) {
9203 /* no changes */;
9204 if (PyUnicode_CheckExact(self)) {
9205 Py_DECREF(u);
9206 Py_INCREF(self);
9207 return self;
9208 }
9209 else
9210 return u;
9211 }
9212
Victor Stinnere6abb482012-05-02 01:15:40 +02009213 maxchar_new = align_maxchar(maxchar_new);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009214
Victor Stinnereaab6042011-12-11 22:22:39 +01009215 if (maxchar_new == maxchar_old)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009216 return u;
Victor Stinnereaab6042011-12-11 22:22:39 +01009217
9218 /* In case the maximum character changed, we need to
9219 convert the string to the new category. */
9220 v = PyUnicode_New(PyUnicode_GET_LENGTH(self), maxchar_new);
9221 if (v == NULL) {
9222 Py_DECREF(u);
9223 return NULL;
9224 }
9225 if (maxchar_new > maxchar_old) {
9226 /* If the maxchar increased so that the kind changed, not all
9227 characters are representable anymore and we need to fix the
9228 string again. This only happens in very few cases. */
Victor Stinnerd3f08822012-05-29 12:57:52 +02009229 _PyUnicode_FastCopyCharacters(v, 0,
9230 self, 0, PyUnicode_GET_LENGTH(self));
Victor Stinnereaab6042011-12-11 22:22:39 +01009231 maxchar_old = fixfct(v);
9232 assert(maxchar_old > 0 && maxchar_old <= maxchar_new);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009233 }
9234 else {
Victor Stinnerd3f08822012-05-29 12:57:52 +02009235 _PyUnicode_FastCopyCharacters(v, 0,
9236 u, 0, PyUnicode_GET_LENGTH(self));
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009237 }
Victor Stinnereaab6042011-12-11 22:22:39 +01009238 Py_DECREF(u);
9239 assert(_PyUnicode_CheckConsistency(v, 1));
9240 return v;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009241}
9242
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009243static PyObject *
9244ascii_upper_or_lower(PyObject *self, int lower)
Guido van Rossumd57fd912000-03-10 22:53:23 +00009245{
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009246 Py_ssize_t len = PyUnicode_GET_LENGTH(self);
9247 char *resdata, *data = PyUnicode_DATA(self);
9248 PyObject *res;
Tim Petersced69f82003-09-16 20:30:58 +00009249
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009250 res = PyUnicode_New(len, 127);
9251 if (res == NULL)
9252 return NULL;
9253 resdata = PyUnicode_DATA(res);
9254 if (lower)
9255 _Py_bytes_lower(resdata, data, len);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009256 else
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009257 _Py_bytes_upper(resdata, data, len);
9258 return res;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009259}
9260
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009261static Py_UCS4
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009262handle_capital_sigma(int kind, void *data, Py_ssize_t length, Py_ssize_t i)
Guido van Rossumd57fd912000-03-10 22:53:23 +00009263{
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009264 Py_ssize_t j;
9265 int final_sigma;
9266 Py_UCS4 c;
9267 /* U+03A3 is in the Final_Sigma context when, it is found like this:
Tim Petersced69f82003-09-16 20:30:58 +00009268
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009269 \p{cased}\p{case-ignorable}*U+03A3!(\p{case-ignorable}*\p{cased})
9270
9271 where ! is a negation and \p{xxx} is a character with property xxx.
9272 */
9273 for (j = i - 1; j >= 0; j--) {
9274 c = PyUnicode_READ(kind, data, j);
9275 if (!_PyUnicode_IsCaseIgnorable(c))
9276 break;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009277 }
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009278 final_sigma = j >= 0 && _PyUnicode_IsCased(c);
9279 if (final_sigma) {
9280 for (j = i + 1; j < length; j++) {
9281 c = PyUnicode_READ(kind, data, j);
9282 if (!_PyUnicode_IsCaseIgnorable(c))
9283 break;
9284 }
9285 final_sigma = j == length || !_PyUnicode_IsCased(c);
9286 }
9287 return (final_sigma) ? 0x3C2 : 0x3C3;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009288}
9289
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009290static int
9291lower_ucs4(int kind, void *data, Py_ssize_t length, Py_ssize_t i,
9292 Py_UCS4 c, Py_UCS4 *mapped)
Guido van Rossumd57fd912000-03-10 22:53:23 +00009293{
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009294 /* Obscure special case. */
9295 if (c == 0x3A3) {
9296 mapped[0] = handle_capital_sigma(kind, data, length, i);
9297 return 1;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009298 }
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009299 return _PyUnicode_ToLowerFull(c, mapped);
Guido van Rossumd57fd912000-03-10 22:53:23 +00009300}
9301
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009302static Py_ssize_t
9303do_capitalize(int kind, void *data, Py_ssize_t length, Py_UCS4 *res, Py_UCS4 *maxchar)
Guido van Rossumd57fd912000-03-10 22:53:23 +00009304{
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009305 Py_ssize_t i, k = 0;
9306 int n_res, j;
9307 Py_UCS4 c, mapped[3];
Tim Petersced69f82003-09-16 20:30:58 +00009308
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009309 c = PyUnicode_READ(kind, data, 0);
9310 n_res = _PyUnicode_ToUpperFull(c, mapped);
9311 for (j = 0; j < n_res; j++) {
Benjamin Peterson7e303732013-06-10 09:19:46 -07009312 *maxchar = Py_MAX(*maxchar, mapped[j]);
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009313 res[k++] = mapped[j];
Guido van Rossumd57fd912000-03-10 22:53:23 +00009314 }
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009315 for (i = 1; i < length; i++) {
9316 c = PyUnicode_READ(kind, data, i);
9317 n_res = lower_ucs4(kind, data, length, i, c, mapped);
9318 for (j = 0; j < n_res; j++) {
Benjamin Peterson7e303732013-06-10 09:19:46 -07009319 *maxchar = Py_MAX(*maxchar, mapped[j]);
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009320 res[k++] = mapped[j];
Marc-André Lemburgfde66e12001-01-29 11:14:16 +00009321 }
Marc-André Lemburgfde66e12001-01-29 11:14:16 +00009322 }
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009323 return k;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009324}
9325
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009326static Py_ssize_t
9327do_swapcase(int kind, void *data, Py_ssize_t length, Py_UCS4 *res, Py_UCS4 *maxchar) {
9328 Py_ssize_t i, k = 0;
9329
9330 for (i = 0; i < length; i++) {
9331 Py_UCS4 c = PyUnicode_READ(kind, data, i), mapped[3];
9332 int n_res, j;
9333 if (Py_UNICODE_ISUPPER(c)) {
9334 n_res = lower_ucs4(kind, data, length, i, c, mapped);
9335 }
9336 else if (Py_UNICODE_ISLOWER(c)) {
9337 n_res = _PyUnicode_ToUpperFull(c, mapped);
9338 }
9339 else {
9340 n_res = 1;
9341 mapped[0] = c;
9342 }
9343 for (j = 0; j < n_res; j++) {
Benjamin Peterson7e303732013-06-10 09:19:46 -07009344 *maxchar = Py_MAX(*maxchar, mapped[j]);
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009345 res[k++] = mapped[j];
9346 }
9347 }
9348 return k;
9349}
9350
9351static Py_ssize_t
9352do_upper_or_lower(int kind, void *data, Py_ssize_t length, Py_UCS4 *res,
9353 Py_UCS4 *maxchar, int lower)
Guido van Rossumd57fd912000-03-10 22:53:23 +00009354{
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009355 Py_ssize_t i, k = 0;
9356
9357 for (i = 0; i < length; i++) {
9358 Py_UCS4 c = PyUnicode_READ(kind, data, i), mapped[3];
9359 int n_res, j;
9360 if (lower)
9361 n_res = lower_ucs4(kind, data, length, i, c, mapped);
9362 else
9363 n_res = _PyUnicode_ToUpperFull(c, mapped);
9364 for (j = 0; j < n_res; j++) {
Benjamin Peterson7e303732013-06-10 09:19:46 -07009365 *maxchar = Py_MAX(*maxchar, mapped[j]);
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009366 res[k++] = mapped[j];
9367 }
9368 }
9369 return k;
9370}
9371
9372static Py_ssize_t
9373do_upper(int kind, void *data, Py_ssize_t length, Py_UCS4 *res, Py_UCS4 *maxchar)
9374{
9375 return do_upper_or_lower(kind, data, length, res, maxchar, 0);
9376}
9377
9378static Py_ssize_t
9379do_lower(int kind, void *data, Py_ssize_t length, Py_UCS4 *res, Py_UCS4 *maxchar)
9380{
9381 return do_upper_or_lower(kind, data, length, res, maxchar, 1);
9382}
9383
Benjamin Petersone51757f2012-01-12 21:10:29 -05009384static Py_ssize_t
Benjamin Petersond5890c82012-01-14 13:23:30 -05009385do_casefold(int kind, void *data, Py_ssize_t length, Py_UCS4 *res, Py_UCS4 *maxchar)
9386{
9387 Py_ssize_t i, k = 0;
9388
9389 for (i = 0; i < length; i++) {
9390 Py_UCS4 c = PyUnicode_READ(kind, data, i);
9391 Py_UCS4 mapped[3];
9392 int j, n_res = _PyUnicode_ToFoldedFull(c, mapped);
9393 for (j = 0; j < n_res; j++) {
Benjamin Peterson7e303732013-06-10 09:19:46 -07009394 *maxchar = Py_MAX(*maxchar, mapped[j]);
Benjamin Petersond5890c82012-01-14 13:23:30 -05009395 res[k++] = mapped[j];
9396 }
9397 }
9398 return k;
9399}
9400
9401static Py_ssize_t
Benjamin Petersone51757f2012-01-12 21:10:29 -05009402do_title(int kind, void *data, Py_ssize_t length, Py_UCS4 *res, Py_UCS4 *maxchar)
9403{
9404 Py_ssize_t i, k = 0;
9405 int previous_is_cased;
9406
9407 previous_is_cased = 0;
9408 for (i = 0; i < length; i++) {
9409 const Py_UCS4 c = PyUnicode_READ(kind, data, i);
9410 Py_UCS4 mapped[3];
9411 int n_res, j;
9412
9413 if (previous_is_cased)
9414 n_res = lower_ucs4(kind, data, length, i, c, mapped);
9415 else
9416 n_res = _PyUnicode_ToTitleFull(c, mapped);
9417
9418 for (j = 0; j < n_res; j++) {
Benjamin Peterson7e303732013-06-10 09:19:46 -07009419 *maxchar = Py_MAX(*maxchar, mapped[j]);
Benjamin Petersone51757f2012-01-12 21:10:29 -05009420 res[k++] = mapped[j];
9421 }
9422
9423 previous_is_cased = _PyUnicode_IsCased(c);
9424 }
9425 return k;
9426}
9427
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009428static PyObject *
9429case_operation(PyObject *self,
9430 Py_ssize_t (*perform)(int, void *, Py_ssize_t, Py_UCS4 *, Py_UCS4 *))
9431{
9432 PyObject *res = NULL;
9433 Py_ssize_t length, newlength = 0;
9434 int kind, outkind;
9435 void *data, *outdata;
9436 Py_UCS4 maxchar = 0, *tmp, *tmpend;
9437
Benjamin Petersoneea48462012-01-16 14:28:50 -05009438 assert(PyUnicode_IS_READY(self));
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009439
9440 kind = PyUnicode_KIND(self);
9441 data = PyUnicode_DATA(self);
9442 length = PyUnicode_GET_LENGTH(self);
9443 tmp = PyMem_MALLOC(sizeof(Py_UCS4) * 3 * length);
9444 if (tmp == NULL)
9445 return PyErr_NoMemory();
9446 newlength = perform(kind, data, length, tmp, &maxchar);
9447 res = PyUnicode_New(newlength, maxchar);
9448 if (res == NULL)
9449 goto leave;
9450 tmpend = tmp + newlength;
9451 outdata = PyUnicode_DATA(res);
9452 outkind = PyUnicode_KIND(res);
9453 switch (outkind) {
9454 case PyUnicode_1BYTE_KIND:
9455 _PyUnicode_CONVERT_BYTES(Py_UCS4, Py_UCS1, tmp, tmpend, outdata);
9456 break;
9457 case PyUnicode_2BYTE_KIND:
9458 _PyUnicode_CONVERT_BYTES(Py_UCS4, Py_UCS2, tmp, tmpend, outdata);
9459 break;
9460 case PyUnicode_4BYTE_KIND:
9461 memcpy(outdata, tmp, sizeof(Py_UCS4) * newlength);
9462 break;
9463 default:
9464 assert(0);
9465 break;
9466 }
9467 leave:
9468 PyMem_FREE(tmp);
9469 return res;
9470}
9471
Tim Peters8ce9f162004-08-27 01:49:32 +00009472PyObject *
9473PyUnicode_Join(PyObject *separator, PyObject *seq)
Guido van Rossumd57fd912000-03-10 22:53:23 +00009474{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009475 PyObject *sep = NULL;
Victor Stinnerdd077322011-10-07 17:02:31 +02009476 Py_ssize_t seplen;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009477 PyObject *res = NULL; /* the result */
Tim Peters05eba1f2004-08-27 21:32:02 +00009478 PyObject *fseq; /* PySequence_Fast(seq) */
Antoine Pitrouaf14b792008-08-07 21:50:41 +00009479 Py_ssize_t seqlen; /* len(fseq) -- number of items in sequence */
9480 PyObject **items;
Tim Peters8ce9f162004-08-27 01:49:32 +00009481 PyObject *item;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009482 Py_ssize_t sz, i, res_offset;
Victor Stinnerfb9ea8c2011-10-06 01:45:57 +02009483 Py_UCS4 maxchar;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009484 Py_UCS4 item_maxchar;
Victor Stinnerdd077322011-10-07 17:02:31 +02009485 int use_memcpy;
9486 unsigned char *res_data = NULL, *sep_data = NULL;
9487 PyObject *last_obj;
9488 unsigned int kind = 0;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009489
Tim Peters05eba1f2004-08-27 21:32:02 +00009490 fseq = PySequence_Fast(seq, "");
9491 if (fseq == NULL) {
Benjamin Peterson14339b62009-01-31 16:36:08 +00009492 return NULL;
Tim Peters8ce9f162004-08-27 01:49:32 +00009493 }
9494
Antoine Pitrouaf14b792008-08-07 21:50:41 +00009495 /* NOTE: the following code can't call back into Python code,
9496 * so we are sure that fseq won't be mutated.
Tim Peters91879ab2004-08-27 22:35:44 +00009497 */
Antoine Pitrouaf14b792008-08-07 21:50:41 +00009498
Tim Peters05eba1f2004-08-27 21:32:02 +00009499 seqlen = PySequence_Fast_GET_SIZE(fseq);
9500 /* If empty sequence, return u"". */
9501 if (seqlen == 0) {
Victor Stinnerfb9ea8c2011-10-06 01:45:57 +02009502 Py_DECREF(fseq);
Serhiy Storchaka678db842013-01-26 12:16:36 +02009503 _Py_RETURN_UNICODE_EMPTY();
Tim Peters05eba1f2004-08-27 21:32:02 +00009504 }
Victor Stinnerfb9ea8c2011-10-06 01:45:57 +02009505
Tim Peters05eba1f2004-08-27 21:32:02 +00009506 /* If singleton sequence with an exact Unicode, return that. */
Victor Stinnerdd077322011-10-07 17:02:31 +02009507 last_obj = NULL;
Victor Stinnerfb9ea8c2011-10-06 01:45:57 +02009508 items = PySequence_Fast_ITEMS(fseq);
Victor Stinneracf47b82011-10-06 12:32:37 +02009509 if (seqlen == 1) {
9510 if (PyUnicode_CheckExact(items[0])) {
9511 res = items[0];
9512 Py_INCREF(res);
9513 Py_DECREF(fseq);
9514 return res;
9515 }
Victor Stinnerdd077322011-10-07 17:02:31 +02009516 seplen = 0;
Victor Stinnerc6f0df72011-10-06 15:58:54 +02009517 maxchar = 0;
Tim Peters8ce9f162004-08-27 01:49:32 +00009518 }
Antoine Pitrouaf14b792008-08-07 21:50:41 +00009519 else {
Victor Stinneracf47b82011-10-06 12:32:37 +02009520 /* Set up sep and seplen */
9521 if (separator == NULL) {
9522 /* fall back to a blank space separator */
9523 sep = PyUnicode_FromOrdinal(' ');
9524 if (!sep)
9525 goto onError;
Victor Stinnerdd077322011-10-07 17:02:31 +02009526 seplen = 1;
Victor Stinneracf47b82011-10-06 12:32:37 +02009527 maxchar = 32;
Tim Peters05eba1f2004-08-27 21:32:02 +00009528 }
Victor Stinneracf47b82011-10-06 12:32:37 +02009529 else {
9530 if (!PyUnicode_Check(separator)) {
9531 PyErr_Format(PyExc_TypeError,
9532 "separator: expected str instance,"
9533 " %.80s found",
9534 Py_TYPE(separator)->tp_name);
9535 goto onError;
9536 }
9537 if (PyUnicode_READY(separator))
9538 goto onError;
9539 sep = separator;
9540 seplen = PyUnicode_GET_LENGTH(separator);
9541 maxchar = PyUnicode_MAX_CHAR_VALUE(separator);
9542 /* inc refcount to keep this code path symmetric with the
9543 above case of a blank separator */
9544 Py_INCREF(sep);
9545 }
Victor Stinnerdd077322011-10-07 17:02:31 +02009546 last_obj = sep;
Tim Peters05eba1f2004-08-27 21:32:02 +00009547 }
9548
Antoine Pitrouaf14b792008-08-07 21:50:41 +00009549 /* There are at least two things to join, or else we have a subclass
9550 * of str in the sequence.
9551 * Do a pre-pass to figure out the total amount of space we'll
9552 * need (sz), and see whether all argument are strings.
9553 */
9554 sz = 0;
Victor Stinnerdd077322011-10-07 17:02:31 +02009555#ifdef Py_DEBUG
9556 use_memcpy = 0;
9557#else
9558 use_memcpy = 1;
9559#endif
Antoine Pitrouaf14b792008-08-07 21:50:41 +00009560 for (i = 0; i < seqlen; i++) {
9561 const Py_ssize_t old_sz = sz;
9562 item = items[i];
Benjamin Peterson29060642009-01-31 22:14:21 +00009563 if (!PyUnicode_Check(item)) {
9564 PyErr_Format(PyExc_TypeError,
9565 "sequence item %zd: expected str instance,"
9566 " %.80s found",
9567 i, Py_TYPE(item)->tp_name);
9568 goto onError;
9569 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009570 if (PyUnicode_READY(item) == -1)
9571 goto onError;
9572 sz += PyUnicode_GET_LENGTH(item);
9573 item_maxchar = PyUnicode_MAX_CHAR_VALUE(item);
Benjamin Peterson7e303732013-06-10 09:19:46 -07009574 maxchar = Py_MAX(maxchar, item_maxchar);
Antoine Pitrouaf14b792008-08-07 21:50:41 +00009575 if (i != 0)
9576 sz += seplen;
9577 if (sz < old_sz || sz > PY_SSIZE_T_MAX) {
9578 PyErr_SetString(PyExc_OverflowError,
Benjamin Peterson29060642009-01-31 22:14:21 +00009579 "join() result is too long for a Python string");
Antoine Pitrouaf14b792008-08-07 21:50:41 +00009580 goto onError;
9581 }
Victor Stinnerdd077322011-10-07 17:02:31 +02009582 if (use_memcpy && last_obj != NULL) {
9583 if (PyUnicode_KIND(last_obj) != PyUnicode_KIND(item))
9584 use_memcpy = 0;
9585 }
9586 last_obj = item;
Antoine Pitrouaf14b792008-08-07 21:50:41 +00009587 }
Tim Petersced69f82003-09-16 20:30:58 +00009588
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009589 res = PyUnicode_New(sz, maxchar);
Antoine Pitrouaf14b792008-08-07 21:50:41 +00009590 if (res == NULL)
9591 goto onError;
Tim Peters91879ab2004-08-27 22:35:44 +00009592
Antoine Pitrouaf14b792008-08-07 21:50:41 +00009593 /* Catenate everything. */
Victor Stinnerdd077322011-10-07 17:02:31 +02009594#ifdef Py_DEBUG
9595 use_memcpy = 0;
9596#else
9597 if (use_memcpy) {
9598 res_data = PyUnicode_1BYTE_DATA(res);
9599 kind = PyUnicode_KIND(res);
9600 if (seplen != 0)
9601 sep_data = PyUnicode_1BYTE_DATA(sep);
9602 }
9603#endif
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009604 for (i = 0, res_offset = 0; i < seqlen; ++i) {
Victor Stinnerfb9ea8c2011-10-06 01:45:57 +02009605 Py_ssize_t itemlen;
Antoine Pitrouaf14b792008-08-07 21:50:41 +00009606 item = items[i];
Benjamin Peterson29060642009-01-31 22:14:21 +00009607 /* Copy item, and maybe the separator. */
Victor Stinner9ce5a832011-10-03 23:36:02 +02009608 if (i && seplen != 0) {
Victor Stinnerdd077322011-10-07 17:02:31 +02009609 if (use_memcpy) {
9610 Py_MEMCPY(res_data,
9611 sep_data,
Martin v. Löwisc47adb02011-10-07 20:55:35 +02009612 kind * seplen);
9613 res_data += kind * seplen;
Victor Stinnerdd077322011-10-07 17:02:31 +02009614 }
9615 else {
Victor Stinnerd3f08822012-05-29 12:57:52 +02009616 _PyUnicode_FastCopyCharacters(res, res_offset, sep, 0, seplen);
Victor Stinnerdd077322011-10-07 17:02:31 +02009617 res_offset += seplen;
9618 }
Benjamin Peterson29060642009-01-31 22:14:21 +00009619 }
Victor Stinner9ce5a832011-10-03 23:36:02 +02009620 itemlen = PyUnicode_GET_LENGTH(item);
9621 if (itemlen != 0) {
Victor Stinnerdd077322011-10-07 17:02:31 +02009622 if (use_memcpy) {
9623 Py_MEMCPY(res_data,
9624 PyUnicode_DATA(item),
Martin v. Löwisc47adb02011-10-07 20:55:35 +02009625 kind * itemlen);
9626 res_data += kind * itemlen;
Victor Stinnerdd077322011-10-07 17:02:31 +02009627 }
9628 else {
Victor Stinnerd3f08822012-05-29 12:57:52 +02009629 _PyUnicode_FastCopyCharacters(res, res_offset, item, 0, itemlen);
Victor Stinnerdd077322011-10-07 17:02:31 +02009630 res_offset += itemlen;
9631 }
Victor Stinner9ce5a832011-10-03 23:36:02 +02009632 }
Tim Peters05eba1f2004-08-27 21:32:02 +00009633 }
Victor Stinnerdd077322011-10-07 17:02:31 +02009634 if (use_memcpy)
9635 assert(res_data == PyUnicode_1BYTE_DATA(res)
Martin v. Löwisc47adb02011-10-07 20:55:35 +02009636 + kind * PyUnicode_GET_LENGTH(res));
Victor Stinnerdd077322011-10-07 17:02:31 +02009637 else
9638 assert(res_offset == PyUnicode_GET_LENGTH(res));
Tim Peters8ce9f162004-08-27 01:49:32 +00009639
Tim Peters05eba1f2004-08-27 21:32:02 +00009640 Py_DECREF(fseq);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009641 Py_XDECREF(sep);
Victor Stinnerbb10a1f2011-10-05 01:34:17 +02009642 assert(_PyUnicode_CheckConsistency(res, 1));
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009643 return res;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009644
Benjamin Peterson29060642009-01-31 22:14:21 +00009645 onError:
Tim Peters05eba1f2004-08-27 21:32:02 +00009646 Py_DECREF(fseq);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009647 Py_XDECREF(sep);
Tim Peters8ce9f162004-08-27 01:49:32 +00009648 Py_XDECREF(res);
Guido van Rossumd57fd912000-03-10 22:53:23 +00009649 return NULL;
9650}
9651
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009652#define FILL(kind, data, value, start, length) \
9653 do { \
9654 Py_ssize_t i_ = 0; \
9655 assert(kind != PyUnicode_WCHAR_KIND); \
9656 switch ((kind)) { \
9657 case PyUnicode_1BYTE_KIND: { \
9658 unsigned char * to_ = (unsigned char *)((data)) + (start); \
Victor Stinnerf2c76aa2012-05-03 13:10:40 +02009659 memset(to_, (unsigned char)value, (length)); \
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009660 break; \
9661 } \
9662 case PyUnicode_2BYTE_KIND: { \
9663 Py_UCS2 * to_ = (Py_UCS2 *)((data)) + (start); \
9664 for (; i_ < (length); ++i_, ++to_) *to_ = (value); \
9665 break; \
9666 } \
Benjamin Petersone157cf12012-01-01 15:56:20 -06009667 case PyUnicode_4BYTE_KIND: { \
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009668 Py_UCS4 * to_ = (Py_UCS4 *)((data)) + (start); \
9669 for (; i_ < (length); ++i_, ++to_) *to_ = (value); \
9670 break; \
Benjamin Petersone157cf12012-01-01 15:56:20 -06009671 default: assert(0); \
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009672 } \
9673 } \
9674 } while (0)
9675
Victor Stinnerd3f08822012-05-29 12:57:52 +02009676void
9677_PyUnicode_FastFill(PyObject *unicode, Py_ssize_t start, Py_ssize_t length,
9678 Py_UCS4 fill_char)
9679{
9680 const enum PyUnicode_Kind kind = PyUnicode_KIND(unicode);
9681 const void *data = PyUnicode_DATA(unicode);
9682 assert(PyUnicode_IS_READY(unicode));
9683 assert(unicode_modifiable(unicode));
9684 assert(fill_char <= PyUnicode_MAX_CHAR_VALUE(unicode));
9685 assert(start >= 0);
9686 assert(start + length <= PyUnicode_GET_LENGTH(unicode));
9687 FILL(kind, data, fill_char, start, length);
9688}
9689
Victor Stinner3fe55312012-01-04 00:33:50 +01009690Py_ssize_t
9691PyUnicode_Fill(PyObject *unicode, Py_ssize_t start, Py_ssize_t length,
9692 Py_UCS4 fill_char)
9693{
9694 Py_ssize_t maxlen;
Victor Stinner3fe55312012-01-04 00:33:50 +01009695
9696 if (!PyUnicode_Check(unicode)) {
9697 PyErr_BadInternalCall();
9698 return -1;
9699 }
9700 if (PyUnicode_READY(unicode) == -1)
9701 return -1;
9702 if (unicode_check_modifiable(unicode))
9703 return -1;
9704
Victor Stinnerd3f08822012-05-29 12:57:52 +02009705 if (start < 0) {
9706 PyErr_SetString(PyExc_IndexError, "string index out of range");
9707 return -1;
9708 }
Victor Stinner3fe55312012-01-04 00:33:50 +01009709 if (fill_char > PyUnicode_MAX_CHAR_VALUE(unicode)) {
9710 PyErr_SetString(PyExc_ValueError,
9711 "fill character is bigger than "
9712 "the string maximum character");
9713 return -1;
9714 }
9715
9716 maxlen = PyUnicode_GET_LENGTH(unicode) - start;
9717 length = Py_MIN(maxlen, length);
9718 if (length <= 0)
9719 return 0;
9720
Victor Stinnerd3f08822012-05-29 12:57:52 +02009721 _PyUnicode_FastFill(unicode, start, length, fill_char);
Victor Stinner3fe55312012-01-04 00:33:50 +01009722 return length;
9723}
9724
Victor Stinner9310abb2011-10-05 00:59:23 +02009725static PyObject *
9726pad(PyObject *self,
Alexander Belopolsky40018472011-02-26 01:02:56 +00009727 Py_ssize_t left,
9728 Py_ssize_t right,
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009729 Py_UCS4 fill)
Guido van Rossumd57fd912000-03-10 22:53:23 +00009730{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009731 PyObject *u;
9732 Py_UCS4 maxchar;
Victor Stinner6c7a52a2011-09-28 21:39:17 +02009733 int kind;
9734 void *data;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009735
9736 if (left < 0)
9737 left = 0;
9738 if (right < 0)
9739 right = 0;
9740
Victor Stinnerc4b49542011-12-11 22:44:26 +01009741 if (left == 0 && right == 0)
9742 return unicode_result_unchanged(self);
Guido van Rossumd57fd912000-03-10 22:53:23 +00009743
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009744 if (left > PY_SSIZE_T_MAX - _PyUnicode_LENGTH(self) ||
9745 right > PY_SSIZE_T_MAX - (left + _PyUnicode_LENGTH(self))) {
Neal Norwitz3ce5d922008-08-24 07:08:55 +00009746 PyErr_SetString(PyExc_OverflowError, "padded string is too long");
9747 return NULL;
9748 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009749 maxchar = PyUnicode_MAX_CHAR_VALUE(self);
Benjamin Peterson7e303732013-06-10 09:19:46 -07009750 maxchar = Py_MAX(maxchar, fill);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009751 u = PyUnicode_New(left + _PyUnicode_LENGTH(self) + right, maxchar);
Victor Stinner6c7a52a2011-09-28 21:39:17 +02009752 if (!u)
9753 return NULL;
9754
9755 kind = PyUnicode_KIND(u);
9756 data = PyUnicode_DATA(u);
9757 if (left)
9758 FILL(kind, data, fill, 0, left);
9759 if (right)
9760 FILL(kind, data, fill, left + _PyUnicode_LENGTH(self), right);
Victor Stinnerd3f08822012-05-29 12:57:52 +02009761 _PyUnicode_FastCopyCharacters(u, left, self, 0, _PyUnicode_LENGTH(self));
Victor Stinnerbb10a1f2011-10-05 01:34:17 +02009762 assert(_PyUnicode_CheckConsistency(u, 1));
9763 return u;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009764}
9765
Alexander Belopolsky40018472011-02-26 01:02:56 +00009766PyObject *
9767PyUnicode_Splitlines(PyObject *string, int keepends)
Guido van Rossumd57fd912000-03-10 22:53:23 +00009768{
Guido van Rossumd57fd912000-03-10 22:53:23 +00009769 PyObject *list;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009770
9771 string = PyUnicode_FromObject(string);
Benjamin Peterson22a29702012-01-02 09:00:30 -06009772 if (string == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00009773 return NULL;
Benjamin Peterson22a29702012-01-02 09:00:30 -06009774 if (PyUnicode_READY(string) == -1) {
9775 Py_DECREF(string);
9776 return NULL;
9777 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00009778
Benjamin Petersonead6b532011-12-20 17:23:42 -06009779 switch (PyUnicode_KIND(string)) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009780 case PyUnicode_1BYTE_KIND:
Victor Stinnerc3cec782011-10-05 21:24:08 +02009781 if (PyUnicode_IS_ASCII(string))
9782 list = asciilib_splitlines(
Victor Stinner7931d9a2011-11-04 00:22:48 +01009783 string, PyUnicode_1BYTE_DATA(string),
Victor Stinnerc3cec782011-10-05 21:24:08 +02009784 PyUnicode_GET_LENGTH(string), keepends);
9785 else
9786 list = ucs1lib_splitlines(
Victor Stinner7931d9a2011-11-04 00:22:48 +01009787 string, PyUnicode_1BYTE_DATA(string),
Victor Stinnerc3cec782011-10-05 21:24:08 +02009788 PyUnicode_GET_LENGTH(string), keepends);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009789 break;
9790 case PyUnicode_2BYTE_KIND:
9791 list = ucs2lib_splitlines(
Victor Stinner7931d9a2011-11-04 00:22:48 +01009792 string, PyUnicode_2BYTE_DATA(string),
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009793 PyUnicode_GET_LENGTH(string), keepends);
9794 break;
9795 case PyUnicode_4BYTE_KIND:
9796 list = ucs4lib_splitlines(
Victor Stinner7931d9a2011-11-04 00:22:48 +01009797 string, PyUnicode_4BYTE_DATA(string),
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009798 PyUnicode_GET_LENGTH(string), keepends);
9799 break;
9800 default:
9801 assert(0);
9802 list = 0;
9803 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00009804 Py_DECREF(string);
9805 return list;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009806}
9807
Alexander Belopolsky40018472011-02-26 01:02:56 +00009808static PyObject *
Victor Stinner9310abb2011-10-05 00:59:23 +02009809split(PyObject *self,
9810 PyObject *substring,
Alexander Belopolsky40018472011-02-26 01:02:56 +00009811 Py_ssize_t maxcount)
Guido van Rossumd57fd912000-03-10 22:53:23 +00009812{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009813 int kind1, kind2, kind;
9814 void *buf1, *buf2;
9815 Py_ssize_t len1, len2;
9816 PyObject* out;
9817
Guido van Rossumd57fd912000-03-10 22:53:23 +00009818 if (maxcount < 0)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00009819 maxcount = PY_SSIZE_T_MAX;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009820
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009821 if (PyUnicode_READY(self) == -1)
9822 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009823
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009824 if (substring == NULL)
Benjamin Petersonead6b532011-12-20 17:23:42 -06009825 switch (PyUnicode_KIND(self)) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009826 case PyUnicode_1BYTE_KIND:
Victor Stinnerc3cec782011-10-05 21:24:08 +02009827 if (PyUnicode_IS_ASCII(self))
9828 return asciilib_split_whitespace(
Victor Stinner7931d9a2011-11-04 00:22:48 +01009829 self, PyUnicode_1BYTE_DATA(self),
Victor Stinnerc3cec782011-10-05 21:24:08 +02009830 PyUnicode_GET_LENGTH(self), maxcount
9831 );
9832 else
9833 return ucs1lib_split_whitespace(
Victor Stinner7931d9a2011-11-04 00:22:48 +01009834 self, PyUnicode_1BYTE_DATA(self),
Victor Stinnerc3cec782011-10-05 21:24:08 +02009835 PyUnicode_GET_LENGTH(self), maxcount
9836 );
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009837 case PyUnicode_2BYTE_KIND:
9838 return ucs2lib_split_whitespace(
Victor Stinner7931d9a2011-11-04 00:22:48 +01009839 self, PyUnicode_2BYTE_DATA(self),
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009840 PyUnicode_GET_LENGTH(self), maxcount
9841 );
9842 case PyUnicode_4BYTE_KIND:
9843 return ucs4lib_split_whitespace(
Victor Stinner7931d9a2011-11-04 00:22:48 +01009844 self, PyUnicode_4BYTE_DATA(self),
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009845 PyUnicode_GET_LENGTH(self), maxcount
9846 );
9847 default:
9848 assert(0);
9849 return NULL;
9850 }
9851
9852 if (PyUnicode_READY(substring) == -1)
9853 return NULL;
9854
9855 kind1 = PyUnicode_KIND(self);
9856 kind2 = PyUnicode_KIND(substring);
9857 kind = kind1 > kind2 ? kind1 : kind2;
9858 buf1 = PyUnicode_DATA(self);
9859 buf2 = PyUnicode_DATA(substring);
9860 if (kind1 != kind)
Victor Stinner7931d9a2011-11-04 00:22:48 +01009861 buf1 = _PyUnicode_AsKind(self, kind);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009862 if (!buf1)
9863 return NULL;
9864 if (kind2 != kind)
Victor Stinner7931d9a2011-11-04 00:22:48 +01009865 buf2 = _PyUnicode_AsKind(substring, kind);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009866 if (!buf2) {
9867 if (kind1 != kind) PyMem_Free(buf1);
9868 return NULL;
9869 }
9870 len1 = PyUnicode_GET_LENGTH(self);
9871 len2 = PyUnicode_GET_LENGTH(substring);
9872
Benjamin Petersonead6b532011-12-20 17:23:42 -06009873 switch (kind) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009874 case PyUnicode_1BYTE_KIND:
Victor Stinnerc3cec782011-10-05 21:24:08 +02009875 if (PyUnicode_IS_ASCII(self) && PyUnicode_IS_ASCII(substring))
9876 out = asciilib_split(
Victor Stinner7931d9a2011-11-04 00:22:48 +01009877 self, buf1, len1, buf2, len2, maxcount);
Victor Stinnerc3cec782011-10-05 21:24:08 +02009878 else
9879 out = ucs1lib_split(
Victor Stinner7931d9a2011-11-04 00:22:48 +01009880 self, buf1, len1, buf2, len2, maxcount);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009881 break;
9882 case PyUnicode_2BYTE_KIND:
9883 out = ucs2lib_split(
Victor Stinner7931d9a2011-11-04 00:22:48 +01009884 self, buf1, len1, buf2, len2, maxcount);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009885 break;
9886 case PyUnicode_4BYTE_KIND:
9887 out = ucs4lib_split(
Victor Stinner7931d9a2011-11-04 00:22:48 +01009888 self, buf1, len1, buf2, len2, maxcount);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009889 break;
9890 default:
9891 out = NULL;
9892 }
9893 if (kind1 != kind)
9894 PyMem_Free(buf1);
9895 if (kind2 != kind)
9896 PyMem_Free(buf2);
9897 return out;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009898}
9899
Alexander Belopolsky40018472011-02-26 01:02:56 +00009900static PyObject *
Victor Stinner9310abb2011-10-05 00:59:23 +02009901rsplit(PyObject *self,
9902 PyObject *substring,
Alexander Belopolsky40018472011-02-26 01:02:56 +00009903 Py_ssize_t maxcount)
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +00009904{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009905 int kind1, kind2, kind;
9906 void *buf1, *buf2;
9907 Py_ssize_t len1, len2;
9908 PyObject* out;
9909
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +00009910 if (maxcount < 0)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00009911 maxcount = PY_SSIZE_T_MAX;
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +00009912
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009913 if (PyUnicode_READY(self) == -1)
9914 return NULL;
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +00009915
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009916 if (substring == NULL)
Benjamin Petersonead6b532011-12-20 17:23:42 -06009917 switch (PyUnicode_KIND(self)) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009918 case PyUnicode_1BYTE_KIND:
Victor Stinnerc3cec782011-10-05 21:24:08 +02009919 if (PyUnicode_IS_ASCII(self))
9920 return asciilib_rsplit_whitespace(
Victor Stinner7931d9a2011-11-04 00:22:48 +01009921 self, PyUnicode_1BYTE_DATA(self),
Victor Stinnerc3cec782011-10-05 21:24:08 +02009922 PyUnicode_GET_LENGTH(self), maxcount
9923 );
9924 else
9925 return ucs1lib_rsplit_whitespace(
Victor Stinner7931d9a2011-11-04 00:22:48 +01009926 self, PyUnicode_1BYTE_DATA(self),
Victor Stinnerc3cec782011-10-05 21:24:08 +02009927 PyUnicode_GET_LENGTH(self), maxcount
9928 );
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009929 case PyUnicode_2BYTE_KIND:
9930 return ucs2lib_rsplit_whitespace(
Victor Stinner7931d9a2011-11-04 00:22:48 +01009931 self, PyUnicode_2BYTE_DATA(self),
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009932 PyUnicode_GET_LENGTH(self), maxcount
9933 );
9934 case PyUnicode_4BYTE_KIND:
9935 return ucs4lib_rsplit_whitespace(
Victor Stinner7931d9a2011-11-04 00:22:48 +01009936 self, PyUnicode_4BYTE_DATA(self),
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009937 PyUnicode_GET_LENGTH(self), maxcount
9938 );
9939 default:
9940 assert(0);
9941 return NULL;
9942 }
9943
9944 if (PyUnicode_READY(substring) == -1)
9945 return NULL;
9946
9947 kind1 = PyUnicode_KIND(self);
9948 kind2 = PyUnicode_KIND(substring);
9949 kind = kind1 > kind2 ? kind1 : kind2;
9950 buf1 = PyUnicode_DATA(self);
9951 buf2 = PyUnicode_DATA(substring);
9952 if (kind1 != kind)
Victor Stinner7931d9a2011-11-04 00:22:48 +01009953 buf1 = _PyUnicode_AsKind(self, kind);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009954 if (!buf1)
9955 return NULL;
9956 if (kind2 != kind)
Victor Stinner7931d9a2011-11-04 00:22:48 +01009957 buf2 = _PyUnicode_AsKind(substring, kind);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009958 if (!buf2) {
9959 if (kind1 != kind) PyMem_Free(buf1);
9960 return NULL;
9961 }
9962 len1 = PyUnicode_GET_LENGTH(self);
9963 len2 = PyUnicode_GET_LENGTH(substring);
9964
Benjamin Petersonead6b532011-12-20 17:23:42 -06009965 switch (kind) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009966 case PyUnicode_1BYTE_KIND:
Victor Stinnerc3cec782011-10-05 21:24:08 +02009967 if (PyUnicode_IS_ASCII(self) && PyUnicode_IS_ASCII(substring))
9968 out = asciilib_rsplit(
Victor Stinner7931d9a2011-11-04 00:22:48 +01009969 self, buf1, len1, buf2, len2, maxcount);
Victor Stinnerc3cec782011-10-05 21:24:08 +02009970 else
9971 out = ucs1lib_rsplit(
Victor Stinner7931d9a2011-11-04 00:22:48 +01009972 self, buf1, len1, buf2, len2, maxcount);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009973 break;
9974 case PyUnicode_2BYTE_KIND:
9975 out = ucs2lib_rsplit(
Victor Stinner7931d9a2011-11-04 00:22:48 +01009976 self, buf1, len1, buf2, len2, maxcount);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009977 break;
9978 case PyUnicode_4BYTE_KIND:
9979 out = ucs4lib_rsplit(
Victor Stinner7931d9a2011-11-04 00:22:48 +01009980 self, buf1, len1, buf2, len2, maxcount);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009981 break;
9982 default:
9983 out = NULL;
9984 }
9985 if (kind1 != kind)
9986 PyMem_Free(buf1);
9987 if (kind2 != kind)
9988 PyMem_Free(buf2);
9989 return out;
9990}
9991
9992static Py_ssize_t
Victor Stinnerc3cec782011-10-05 21:24:08 +02009993anylib_find(int kind, PyObject *str1, void *buf1, Py_ssize_t len1,
9994 PyObject *str2, void *buf2, Py_ssize_t len2, Py_ssize_t offset)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009995{
Benjamin Petersonead6b532011-12-20 17:23:42 -06009996 switch (kind) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009997 case PyUnicode_1BYTE_KIND:
Victor Stinnerc3cec782011-10-05 21:24:08 +02009998 if (PyUnicode_IS_ASCII(str1) && PyUnicode_IS_ASCII(str2))
9999 return asciilib_find(buf1, len1, buf2, len2, offset);
10000 else
10001 return ucs1lib_find(buf1, len1, buf2, len2, offset);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010002 case PyUnicode_2BYTE_KIND:
10003 return ucs2lib_find(buf1, len1, buf2, len2, offset);
10004 case PyUnicode_4BYTE_KIND:
10005 return ucs4lib_find(buf1, len1, buf2, len2, offset);
10006 }
10007 assert(0);
10008 return -1;
10009}
10010
10011static Py_ssize_t
Victor Stinnerc3cec782011-10-05 21:24:08 +020010012anylib_count(int kind, PyObject *sstr, void* sbuf, Py_ssize_t slen,
10013 PyObject *str1, void *buf1, Py_ssize_t len1, Py_ssize_t maxcount)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010014{
Benjamin Petersonc0b95d12011-12-20 17:24:05 -060010015 switch (kind) {
10016 case PyUnicode_1BYTE_KIND:
10017 if (PyUnicode_IS_ASCII(sstr) && PyUnicode_IS_ASCII(str1))
10018 return asciilib_count(sbuf, slen, buf1, len1, maxcount);
10019 else
10020 return ucs1lib_count(sbuf, slen, buf1, len1, maxcount);
10021 case PyUnicode_2BYTE_KIND:
10022 return ucs2lib_count(sbuf, slen, buf1, len1, maxcount);
10023 case PyUnicode_4BYTE_KIND:
10024 return ucs4lib_count(sbuf, slen, buf1, len1, maxcount);
10025 }
10026 assert(0);
10027 return 0;
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +000010028}
10029
Alexander Belopolsky40018472011-02-26 01:02:56 +000010030static PyObject *
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010031replace(PyObject *self, PyObject *str1,
10032 PyObject *str2, Py_ssize_t maxcount)
Guido van Rossumd57fd912000-03-10 22:53:23 +000010033{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010034 PyObject *u;
10035 char *sbuf = PyUnicode_DATA(self);
10036 char *buf1 = PyUnicode_DATA(str1);
10037 char *buf2 = PyUnicode_DATA(str2);
10038 int srelease = 0, release1 = 0, release2 = 0;
10039 int skind = PyUnicode_KIND(self);
10040 int kind1 = PyUnicode_KIND(str1);
10041 int kind2 = PyUnicode_KIND(str2);
10042 Py_ssize_t slen = PyUnicode_GET_LENGTH(self);
10043 Py_ssize_t len1 = PyUnicode_GET_LENGTH(str1);
10044 Py_ssize_t len2 = PyUnicode_GET_LENGTH(str2);
Victor Stinner49a0a212011-10-12 23:46:10 +020010045 int mayshrink;
10046 Py_UCS4 maxchar, maxchar_str2;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010047
10048 if (maxcount < 0)
Benjamin Peterson29060642009-01-31 22:14:21 +000010049 maxcount = PY_SSIZE_T_MAX;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010050 else if (maxcount == 0 || slen == 0)
Antoine Pitrouf2c54842010-01-13 08:07:53 +000010051 goto nothing;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010052
Victor Stinner59de0ee2011-10-07 10:01:28 +020010053 if (str1 == str2)
10054 goto nothing;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010055 if (skind < kind1)
10056 /* substring too wide to be present */
10057 goto nothing;
10058
Victor Stinner49a0a212011-10-12 23:46:10 +020010059 maxchar = PyUnicode_MAX_CHAR_VALUE(self);
10060 maxchar_str2 = PyUnicode_MAX_CHAR_VALUE(str2);
10061 /* Replacing str1 with str2 may cause a maxchar reduction in the
10062 result string. */
10063 mayshrink = (maxchar_str2 < maxchar);
Benjamin Peterson7e303732013-06-10 09:19:46 -070010064 maxchar = Py_MAX(maxchar, maxchar_str2);
Victor Stinner49a0a212011-10-12 23:46:10 +020010065
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010066 if (len1 == len2) {
Thomas Wouters477c8d52006-05-27 19:21:47 +000010067 /* same length */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010068 if (len1 == 0)
Antoine Pitrouf2c54842010-01-13 08:07:53 +000010069 goto nothing;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010070 if (len1 == 1) {
Thomas Wouters477c8d52006-05-27 19:21:47 +000010071 /* replace characters */
Victor Stinner49a0a212011-10-12 23:46:10 +020010072 Py_UCS4 u1, u2;
10073 int rkind;
Victor Stinnerf6441102011-12-18 02:43:08 +010010074 Py_ssize_t index, pos;
10075 char *src;
10076
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010077 u1 = PyUnicode_READ_CHAR(str1, 0);
Victor Stinnerf6441102011-12-18 02:43:08 +010010078 pos = findchar(sbuf, PyUnicode_KIND(self), slen, u1, 1);
10079 if (pos < 0)
Thomas Wouters477c8d52006-05-27 19:21:47 +000010080 goto nothing;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010081 u2 = PyUnicode_READ_CHAR(str2, 0);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010082 u = PyUnicode_New(slen, maxchar);
Thomas Wouters477c8d52006-05-27 19:21:47 +000010083 if (!u)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010084 goto error;
Victor Stinnerd3f08822012-05-29 12:57:52 +020010085 _PyUnicode_FastCopyCharacters(u, 0, self, 0, slen);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010086 rkind = PyUnicode_KIND(u);
Victor Stinnerf6441102011-12-18 02:43:08 +010010087
10088 PyUnicode_WRITE(rkind, PyUnicode_DATA(u), pos, u2);
10089 index = 0;
10090 src = sbuf;
10091 while (--maxcount)
10092 {
10093 pos++;
10094 src += pos * PyUnicode_KIND(self);
10095 slen -= pos;
10096 index += pos;
10097 pos = findchar(src, PyUnicode_KIND(self), slen, u1, 1);
10098 if (pos < 0)
10099 break;
10100 PyUnicode_WRITE(rkind, PyUnicode_DATA(u), index + pos, u2);
10101 }
Victor Stinner49a0a212011-10-12 23:46:10 +020010102 }
10103 else {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010104 int rkind = skind;
10105 char *res;
Victor Stinnerf6441102011-12-18 02:43:08 +010010106 Py_ssize_t i;
Victor Stinner25a4b292011-10-06 12:31:55 +020010107
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010108 if (kind1 < rkind) {
10109 /* widen substring */
10110 buf1 = _PyUnicode_AsKind(str1, rkind);
10111 if (!buf1) goto error;
10112 release1 = 1;
10113 }
Victor Stinnerc3cec782011-10-05 21:24:08 +020010114 i = anylib_find(rkind, self, sbuf, slen, str1, buf1, len1, 0);
Thomas Wouters477c8d52006-05-27 19:21:47 +000010115 if (i < 0)
10116 goto nothing;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010117 if (rkind > kind2) {
10118 /* widen replacement */
10119 buf2 = _PyUnicode_AsKind(str2, rkind);
10120 if (!buf2) goto error;
10121 release2 = 1;
10122 }
10123 else if (rkind < kind2) {
10124 /* widen self and buf1 */
10125 rkind = kind2;
10126 if (release1) PyMem_Free(buf1);
Antoine Pitrou6d5ad222012-11-17 23:28:17 +010010127 release1 = 0;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010128 sbuf = _PyUnicode_AsKind(self, rkind);
10129 if (!sbuf) goto error;
10130 srelease = 1;
10131 buf1 = _PyUnicode_AsKind(str1, rkind);
10132 if (!buf1) goto error;
10133 release1 = 1;
10134 }
Victor Stinner49a0a212011-10-12 23:46:10 +020010135 u = PyUnicode_New(slen, maxchar);
10136 if (!u)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010137 goto error;
Victor Stinner49a0a212011-10-12 23:46:10 +020010138 assert(PyUnicode_KIND(u) == rkind);
10139 res = PyUnicode_DATA(u);
Victor Stinner25a4b292011-10-06 12:31:55 +020010140
Martin v. Löwisc47adb02011-10-07 20:55:35 +020010141 memcpy(res, sbuf, rkind * slen);
Antoine Pitrouf2c54842010-01-13 08:07:53 +000010142 /* change everything in-place, starting with this one */
Martin v. Löwisc47adb02011-10-07 20:55:35 +020010143 memcpy(res + rkind * i,
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010144 buf2,
Martin v. Löwisc47adb02011-10-07 20:55:35 +020010145 rkind * len2);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010146 i += len1;
Antoine Pitrouf2c54842010-01-13 08:07:53 +000010147
10148 while ( --maxcount > 0) {
Victor Stinnerc3cec782011-10-05 21:24:08 +020010149 i = anylib_find(rkind, self,
Martin v. Löwisc47adb02011-10-07 20:55:35 +020010150 sbuf+rkind*i, slen-i,
Victor Stinnerc3cec782011-10-05 21:24:08 +020010151 str1, buf1, len1, i);
Antoine Pitrouf2c54842010-01-13 08:07:53 +000010152 if (i == -1)
10153 break;
Martin v. Löwisc47adb02011-10-07 20:55:35 +020010154 memcpy(res + rkind * i,
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010155 buf2,
Martin v. Löwisc47adb02011-10-07 20:55:35 +020010156 rkind * len2);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010157 i += len1;
Antoine Pitrouf2c54842010-01-13 08:07:53 +000010158 }
Guido van Rossumd57fd912000-03-10 22:53:23 +000010159 }
Victor Stinner49a0a212011-10-12 23:46:10 +020010160 }
10161 else {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010162 Py_ssize_t n, i, j, ires;
Mark Dickinsonc04ddff2012-10-06 18:04:49 +010010163 Py_ssize_t new_size;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010164 int rkind = skind;
10165 char *res;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010166
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010167 if (kind1 < rkind) {
Victor Stinner49a0a212011-10-12 23:46:10 +020010168 /* widen substring */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010169 buf1 = _PyUnicode_AsKind(str1, rkind);
10170 if (!buf1) goto error;
10171 release1 = 1;
10172 }
Victor Stinnerc3cec782011-10-05 21:24:08 +020010173 n = anylib_count(rkind, self, sbuf, slen, str1, buf1, len1, maxcount);
Thomas Wouters477c8d52006-05-27 19:21:47 +000010174 if (n == 0)
10175 goto nothing;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010176 if (kind2 < rkind) {
Victor Stinner49a0a212011-10-12 23:46:10 +020010177 /* widen replacement */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010178 buf2 = _PyUnicode_AsKind(str2, rkind);
10179 if (!buf2) goto error;
10180 release2 = 1;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010181 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010182 else if (kind2 > rkind) {
Victor Stinner49a0a212011-10-12 23:46:10 +020010183 /* widen self and buf1 */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010184 rkind = kind2;
10185 sbuf = _PyUnicode_AsKind(self, rkind);
10186 if (!sbuf) goto error;
10187 srelease = 1;
10188 if (release1) PyMem_Free(buf1);
Antoine Pitrou6d5ad222012-11-17 23:28:17 +010010189 release1 = 0;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010190 buf1 = _PyUnicode_AsKind(str1, rkind);
10191 if (!buf1) goto error;
10192 release1 = 1;
10193 }
10194 /* new_size = PyUnicode_GET_LENGTH(self) + n * (PyUnicode_GET_LENGTH(str2) -
10195 PyUnicode_GET_LENGTH(str1))); */
Mark Dickinsonc04ddff2012-10-06 18:04:49 +010010196 if (len2 > len1 && len2 - len1 > (PY_SSIZE_T_MAX - slen) / n) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010197 PyErr_SetString(PyExc_OverflowError,
10198 "replace string is too long");
10199 goto error;
10200 }
Mark Dickinsonc04ddff2012-10-06 18:04:49 +010010201 new_size = slen + n * (len2 - len1);
Victor Stinner49a0a212011-10-12 23:46:10 +020010202 if (new_size == 0) {
Serhiy Storchaka678db842013-01-26 12:16:36 +020010203 _Py_INCREF_UNICODE_EMPTY();
10204 if (!unicode_empty)
10205 goto error;
Victor Stinner49a0a212011-10-12 23:46:10 +020010206 u = unicode_empty;
10207 goto done;
10208 }
Mark Dickinsonc04ddff2012-10-06 18:04:49 +010010209 if (new_size > (PY_SSIZE_T_MAX >> (rkind-1))) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010210 PyErr_SetString(PyExc_OverflowError,
10211 "replace string is too long");
10212 goto error;
10213 }
Victor Stinner49a0a212011-10-12 23:46:10 +020010214 u = PyUnicode_New(new_size, maxchar);
10215 if (!u)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010216 goto error;
Victor Stinner49a0a212011-10-12 23:46:10 +020010217 assert(PyUnicode_KIND(u) == rkind);
10218 res = PyUnicode_DATA(u);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010219 ires = i = 0;
10220 if (len1 > 0) {
Thomas Wouters477c8d52006-05-27 19:21:47 +000010221 while (n-- > 0) {
10222 /* look for next match */
Victor Stinnerc3cec782011-10-05 21:24:08 +020010223 j = anylib_find(rkind, self,
Martin v. Löwisc47adb02011-10-07 20:55:35 +020010224 sbuf + rkind * i, slen-i,
Victor Stinnerc3cec782011-10-05 21:24:08 +020010225 str1, buf1, len1, i);
Antoine Pitrouf2c54842010-01-13 08:07:53 +000010226 if (j == -1)
10227 break;
10228 else if (j > i) {
Thomas Wouters477c8d52006-05-27 19:21:47 +000010229 /* copy unchanged part [i:j] */
Martin v. Löwisc47adb02011-10-07 20:55:35 +020010230 memcpy(res + rkind * ires,
10231 sbuf + rkind * i,
10232 rkind * (j-i));
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010233 ires += j - i;
Thomas Wouters477c8d52006-05-27 19:21:47 +000010234 }
10235 /* copy substitution string */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010236 if (len2 > 0) {
Martin v. Löwisc47adb02011-10-07 20:55:35 +020010237 memcpy(res + rkind * ires,
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010238 buf2,
Martin v. Löwisc47adb02011-10-07 20:55:35 +020010239 rkind * len2);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010240 ires += len2;
Thomas Wouters477c8d52006-05-27 19:21:47 +000010241 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010242 i = j + len1;
Thomas Wouters477c8d52006-05-27 19:21:47 +000010243 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010244 if (i < slen)
Thomas Wouters477c8d52006-05-27 19:21:47 +000010245 /* copy tail [i:] */
Martin v. Löwisc47adb02011-10-07 20:55:35 +020010246 memcpy(res + rkind * ires,
10247 sbuf + rkind * i,
10248 rkind * (slen-i));
Victor Stinner49a0a212011-10-12 23:46:10 +020010249 }
10250 else {
Thomas Wouters477c8d52006-05-27 19:21:47 +000010251 /* interleave */
10252 while (n > 0) {
Martin v. Löwisc47adb02011-10-07 20:55:35 +020010253 memcpy(res + rkind * ires,
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010254 buf2,
Martin v. Löwisc47adb02011-10-07 20:55:35 +020010255 rkind * len2);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010256 ires += len2;
Thomas Wouters477c8d52006-05-27 19:21:47 +000010257 if (--n <= 0)
10258 break;
Martin v. Löwisc47adb02011-10-07 20:55:35 +020010259 memcpy(res + rkind * ires,
10260 sbuf + rkind * i,
10261 rkind);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010262 ires++;
10263 i++;
Thomas Wouters477c8d52006-05-27 19:21:47 +000010264 }
Martin v. Löwisc47adb02011-10-07 20:55:35 +020010265 memcpy(res + rkind * ires,
10266 sbuf + rkind * i,
10267 rkind * (slen-i));
Thomas Wouters477c8d52006-05-27 19:21:47 +000010268 }
Victor Stinner49a0a212011-10-12 23:46:10 +020010269 }
10270
10271 if (mayshrink) {
Victor Stinner25a4b292011-10-06 12:31:55 +020010272 unicode_adjust_maxchar(&u);
10273 if (u == NULL)
10274 goto error;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010275 }
Victor Stinner49a0a212011-10-12 23:46:10 +020010276
10277 done:
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010278 if (srelease)
10279 PyMem_FREE(sbuf);
10280 if (release1)
10281 PyMem_FREE(buf1);
10282 if (release2)
10283 PyMem_FREE(buf2);
Victor Stinnerbb10a1f2011-10-05 01:34:17 +020010284 assert(_PyUnicode_CheckConsistency(u, 1));
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010285 return u;
Thomas Wouters477c8d52006-05-27 19:21:47 +000010286
Benjamin Peterson29060642009-01-31 22:14:21 +000010287 nothing:
Thomas Wouters477c8d52006-05-27 19:21:47 +000010288 /* nothing to replace; return original string (when possible) */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010289 if (srelease)
10290 PyMem_FREE(sbuf);
10291 if (release1)
10292 PyMem_FREE(buf1);
10293 if (release2)
10294 PyMem_FREE(buf2);
Victor Stinnerc4b49542011-12-11 22:44:26 +010010295 return unicode_result_unchanged(self);
10296
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010297 error:
10298 if (srelease && sbuf)
10299 PyMem_FREE(sbuf);
10300 if (release1 && buf1)
10301 PyMem_FREE(buf1);
10302 if (release2 && buf2)
10303 PyMem_FREE(buf2);
10304 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010305}
10306
10307/* --- Unicode Object Methods --------------------------------------------- */
10308
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000010309PyDoc_STRVAR(title__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000010310 "S.title() -> str\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000010311\n\
10312Return a titlecased version of S, i.e. words start with title case\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000010313characters, all remaining cased characters have lower case.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000010314
10315static PyObject*
Victor Stinner9310abb2011-10-05 00:59:23 +020010316unicode_title(PyObject *self)
Guido van Rossumd57fd912000-03-10 22:53:23 +000010317{
Benjamin Petersoneea48462012-01-16 14:28:50 -050010318 if (PyUnicode_READY(self) == -1)
10319 return NULL;
Victor Stinnerb0800dc2012-02-25 00:47:08 +010010320 return case_operation(self, do_title);
Guido van Rossumd57fd912000-03-10 22:53:23 +000010321}
10322
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000010323PyDoc_STRVAR(capitalize__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000010324 "S.capitalize() -> str\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000010325\n\
10326Return a capitalized version of S, i.e. make the first character\n\
Senthil Kumarane51ee8a2010-07-05 12:00:56 +000010327have upper case and the rest lower case.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000010328
10329static PyObject*
Victor Stinner9310abb2011-10-05 00:59:23 +020010330unicode_capitalize(PyObject *self)
Guido van Rossumd57fd912000-03-10 22:53:23 +000010331{
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -050010332 if (PyUnicode_READY(self) == -1)
10333 return NULL;
10334 if (PyUnicode_GET_LENGTH(self) == 0)
10335 return unicode_result_unchanged(self);
Victor Stinnerb0800dc2012-02-25 00:47:08 +010010336 return case_operation(self, do_capitalize);
Guido van Rossumd57fd912000-03-10 22:53:23 +000010337}
10338
Benjamin Petersond5890c82012-01-14 13:23:30 -050010339PyDoc_STRVAR(casefold__doc__,
10340 "S.casefold() -> str\n\
10341\n\
10342Return a version of S suitable for caseless comparisons.");
10343
10344static PyObject *
10345unicode_casefold(PyObject *self)
10346{
10347 if (PyUnicode_READY(self) == -1)
10348 return NULL;
10349 if (PyUnicode_IS_ASCII(self))
10350 return ascii_upper_or_lower(self, 1);
Victor Stinnerb0800dc2012-02-25 00:47:08 +010010351 return case_operation(self, do_casefold);
Benjamin Petersond5890c82012-01-14 13:23:30 -050010352}
10353
10354
Raymond Hettinger4f8f9762003-11-26 08:21:35 +000010355/* Argument converter. Coerces to a single unicode character */
10356
10357static int
10358convert_uc(PyObject *obj, void *addr)
10359{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010360 Py_UCS4 *fillcharloc = (Py_UCS4 *)addr;
Benjamin Peterson14339b62009-01-31 16:36:08 +000010361 PyObject *uniobj;
Raymond Hettinger4f8f9762003-11-26 08:21:35 +000010362
Benjamin Peterson14339b62009-01-31 16:36:08 +000010363 uniobj = PyUnicode_FromObject(obj);
10364 if (uniobj == NULL) {
10365 PyErr_SetString(PyExc_TypeError,
Benjamin Peterson29060642009-01-31 22:14:21 +000010366 "The fill character cannot be converted to Unicode");
Benjamin Peterson14339b62009-01-31 16:36:08 +000010367 return 0;
10368 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010369 if (PyUnicode_GET_LENGTH(uniobj) != 1) {
Benjamin Peterson14339b62009-01-31 16:36:08 +000010370 PyErr_SetString(PyExc_TypeError,
Benjamin Peterson29060642009-01-31 22:14:21 +000010371 "The fill character must be exactly one character long");
Benjamin Peterson14339b62009-01-31 16:36:08 +000010372 Py_DECREF(uniobj);
10373 return 0;
10374 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010375 *fillcharloc = PyUnicode_READ_CHAR(uniobj, 0);
Benjamin Peterson14339b62009-01-31 16:36:08 +000010376 Py_DECREF(uniobj);
10377 return 1;
Raymond Hettinger4f8f9762003-11-26 08:21:35 +000010378}
10379
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000010380PyDoc_STRVAR(center__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000010381 "S.center(width[, fillchar]) -> str\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000010382\n\
Benjamin Peterson142957c2008-07-04 19:55:29 +000010383Return S centered in a string of length width. Padding is\n\
Raymond Hettinger4f8f9762003-11-26 08:21:35 +000010384done using the specified fill character (default is a space)");
Guido van Rossumd57fd912000-03-10 22:53:23 +000010385
10386static PyObject *
Victor Stinner9310abb2011-10-05 00:59:23 +020010387unicode_center(PyObject *self, PyObject *args)
Guido van Rossumd57fd912000-03-10 22:53:23 +000010388{
Martin v. Löwis18e16552006-02-15 17:27:45 +000010389 Py_ssize_t marg, left;
10390 Py_ssize_t width;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010391 Py_UCS4 fillchar = ' ';
10392
Victor Stinnere9a29352011-10-01 02:14:59 +020010393 if (!PyArg_ParseTuple(args, "n|O&:center", &width, convert_uc, &fillchar))
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010394 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010395
Benjamin Petersonbac79492012-01-14 13:34:47 -050010396 if (PyUnicode_READY(self) == -1)
Guido van Rossumd57fd912000-03-10 22:53:23 +000010397 return NULL;
10398
Victor Stinnerc4b49542011-12-11 22:44:26 +010010399 if (PyUnicode_GET_LENGTH(self) >= width)
10400 return unicode_result_unchanged(self);
Guido van Rossumd57fd912000-03-10 22:53:23 +000010401
Victor Stinnerc4b49542011-12-11 22:44:26 +010010402 marg = width - PyUnicode_GET_LENGTH(self);
Guido van Rossumd57fd912000-03-10 22:53:23 +000010403 left = marg / 2 + (marg & width & 1);
10404
Victor Stinner9310abb2011-10-05 00:59:23 +020010405 return pad(self, left, marg - left, fillchar);
Guido van Rossumd57fd912000-03-10 22:53:23 +000010406}
10407
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010408/* This function assumes that str1 and str2 are readied by the caller. */
10409
Marc-André Lemburge5034372000-08-08 08:04:29 +000010410static int
Victor Stinner9db1a8b2011-10-23 20:04:37 +020010411unicode_compare(PyObject *str1, PyObject *str2)
Marc-André Lemburge5034372000-08-08 08:04:29 +000010412{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010413 int kind1, kind2;
10414 void *data1, *data2;
10415 Py_ssize_t len1, len2, i;
Marc-André Lemburge5034372000-08-08 08:04:29 +000010416
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010417 kind1 = PyUnicode_KIND(str1);
10418 kind2 = PyUnicode_KIND(str2);
10419 data1 = PyUnicode_DATA(str1);
10420 data2 = PyUnicode_DATA(str2);
10421 len1 = PyUnicode_GET_LENGTH(str1);
10422 len2 = PyUnicode_GET_LENGTH(str2);
Marc-André Lemburge5034372000-08-08 08:04:29 +000010423
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010424 for (i = 0; i < len1 && i < len2; ++i) {
10425 Py_UCS4 c1, c2;
10426 c1 = PyUnicode_READ(kind1, data1, i);
10427 c2 = PyUnicode_READ(kind2, data2, i);
Fredrik Lundh45714e92001-06-26 16:39:36 +000010428
10429 if (c1 != c2)
10430 return (c1 < c2) ? -1 : 1;
Marc-André Lemburge5034372000-08-08 08:04:29 +000010431 }
10432
10433 return (len1 < len2) ? -1 : (len1 != len2);
10434}
10435
Alexander Belopolsky40018472011-02-26 01:02:56 +000010436int
10437PyUnicode_Compare(PyObject *left, PyObject *right)
Guido van Rossumd57fd912000-03-10 22:53:23 +000010438{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010439 if (PyUnicode_Check(left) && PyUnicode_Check(right)) {
10440 if (PyUnicode_READY(left) == -1 ||
10441 PyUnicode_READY(right) == -1)
10442 return -1;
Victor Stinner9db1a8b2011-10-23 20:04:37 +020010443 return unicode_compare(left, right);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010444 }
Guido van Rossum09dc34f2007-05-04 04:17:33 +000010445 PyErr_Format(PyExc_TypeError,
10446 "Can't compare %.100s and %.100s",
10447 left->ob_type->tp_name,
10448 right->ob_type->tp_name);
Guido van Rossumd57fd912000-03-10 22:53:23 +000010449 return -1;
10450}
10451
Martin v. Löwis5b222132007-06-10 09:51:05 +000010452int
10453PyUnicode_CompareWithASCIIString(PyObject* uni, const char* str)
10454{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010455 Py_ssize_t i;
10456 int kind;
10457 void *data;
10458 Py_UCS4 chr;
10459
Victor Stinner910337b2011-10-03 03:20:16 +020010460 assert(_PyUnicode_CHECK(uni));
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010461 if (PyUnicode_READY(uni) == -1)
10462 return -1;
10463 kind = PyUnicode_KIND(uni);
10464 data = PyUnicode_DATA(uni);
Martin v. Löwis5b222132007-06-10 09:51:05 +000010465 /* Compare Unicode string and source character set string */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010466 for (i = 0; (chr = PyUnicode_READ(kind, data, i)) && str[i]; i++)
10467 if (chr != str[i])
10468 return (chr < (unsigned char)(str[i])) ? -1 : 1;
Benjamin Peterson8667a9b2010-01-09 21:45:28 +000010469 /* This check keeps Python strings that end in '\0' from comparing equal
10470 to C strings identical up to that point. */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010471 if (PyUnicode_GET_LENGTH(uni) != i || chr)
Benjamin Peterson29060642009-01-31 22:14:21 +000010472 return 1; /* uni is longer */
Martin v. Löwis5b222132007-06-10 09:51:05 +000010473 if (str[i])
Benjamin Peterson29060642009-01-31 22:14:21 +000010474 return -1; /* str is longer */
Martin v. Löwis5b222132007-06-10 09:51:05 +000010475 return 0;
10476}
10477
Antoine Pitrou51f3ef92008-12-20 13:14:23 +000010478
Benjamin Peterson29060642009-01-31 22:14:21 +000010479#define TEST_COND(cond) \
Benjamin Peterson14339b62009-01-31 16:36:08 +000010480 ((cond) ? Py_True : Py_False)
Antoine Pitrou51f3ef92008-12-20 13:14:23 +000010481
Alexander Belopolsky40018472011-02-26 01:02:56 +000010482PyObject *
10483PyUnicode_RichCompare(PyObject *left, PyObject *right, int op)
Thomas Wouters00ee7ba2006-08-21 19:07:27 +000010484{
10485 int result;
Benjamin Peterson14339b62009-01-31 16:36:08 +000010486
Antoine Pitrou51f3ef92008-12-20 13:14:23 +000010487 if (PyUnicode_Check(left) && PyUnicode_Check(right)) {
10488 PyObject *v;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010489 if (PyUnicode_READY(left) == -1 ||
10490 PyUnicode_READY(right) == -1)
10491 return NULL;
10492 if (PyUnicode_GET_LENGTH(left) != PyUnicode_GET_LENGTH(right) ||
10493 PyUnicode_KIND(left) != PyUnicode_KIND(right)) {
Antoine Pitrou51f3ef92008-12-20 13:14:23 +000010494 if (op == Py_EQ) {
10495 Py_INCREF(Py_False);
10496 return Py_False;
10497 }
10498 if (op == Py_NE) {
10499 Py_INCREF(Py_True);
10500 return Py_True;
10501 }
10502 }
10503 if (left == right)
10504 result = 0;
10505 else
Victor Stinner9db1a8b2011-10-23 20:04:37 +020010506 result = unicode_compare(left, right);
Benjamin Peterson14339b62009-01-31 16:36:08 +000010507
Antoine Pitrou51f3ef92008-12-20 13:14:23 +000010508 /* Convert the return value to a Boolean */
10509 switch (op) {
10510 case Py_EQ:
10511 v = TEST_COND(result == 0);
10512 break;
10513 case Py_NE:
10514 v = TEST_COND(result != 0);
10515 break;
10516 case Py_LE:
10517 v = TEST_COND(result <= 0);
10518 break;
10519 case Py_GE:
10520 v = TEST_COND(result >= 0);
10521 break;
10522 case Py_LT:
10523 v = TEST_COND(result == -1);
10524 break;
10525 case Py_GT:
10526 v = TEST_COND(result == 1);
10527 break;
10528 default:
10529 PyErr_BadArgument();
10530 return NULL;
10531 }
10532 Py_INCREF(v);
10533 return v;
Thomas Wouters00ee7ba2006-08-21 19:07:27 +000010534 }
Benjamin Peterson14339b62009-01-31 16:36:08 +000010535
Brian Curtindfc80e32011-08-10 20:28:54 -050010536 Py_RETURN_NOTIMPLEMENTED;
Thomas Wouters00ee7ba2006-08-21 19:07:27 +000010537}
10538
Alexander Belopolsky40018472011-02-26 01:02:56 +000010539int
10540PyUnicode_Contains(PyObject *container, PyObject *element)
Guido van Rossum403d68b2000-03-13 15:55:09 +000010541{
Thomas Wouters477c8d52006-05-27 19:21:47 +000010542 PyObject *str, *sub;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010543 int kind1, kind2, kind;
10544 void *buf1, *buf2;
10545 Py_ssize_t len1, len2;
Martin v. Löwis18e16552006-02-15 17:27:45 +000010546 int result;
Guido van Rossum403d68b2000-03-13 15:55:09 +000010547
10548 /* Coerce the two arguments */
Thomas Wouters477c8d52006-05-27 19:21:47 +000010549 sub = PyUnicode_FromObject(element);
10550 if (!sub) {
Benjamin Peterson29060642009-01-31 22:14:21 +000010551 PyErr_Format(PyExc_TypeError,
10552 "'in <string>' requires string as left operand, not %s",
10553 element->ob_type->tp_name);
Thomas Wouters477c8d52006-05-27 19:21:47 +000010554 return -1;
Guido van Rossum403d68b2000-03-13 15:55:09 +000010555 }
10556
Thomas Wouters477c8d52006-05-27 19:21:47 +000010557 str = PyUnicode_FromObject(container);
Benjamin Peterson22a29702012-01-02 09:00:30 -060010558 if (!str) {
Thomas Wouters477c8d52006-05-27 19:21:47 +000010559 Py_DECREF(sub);
10560 return -1;
10561 }
Benjamin Peterson22a29702012-01-02 09:00:30 -060010562 if (PyUnicode_READY(sub) == -1 || PyUnicode_READY(str) == -1) {
10563 Py_DECREF(sub);
10564 Py_DECREF(str);
10565 }
Thomas Wouters477c8d52006-05-27 19:21:47 +000010566
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010567 kind1 = PyUnicode_KIND(str);
10568 kind2 = PyUnicode_KIND(sub);
Benjamin Peterson1ff2e352012-05-11 17:41:20 -050010569 kind = kind1;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010570 buf1 = PyUnicode_DATA(str);
10571 buf2 = PyUnicode_DATA(sub);
Benjamin Peterson1ff2e352012-05-11 17:41:20 -050010572 if (kind2 != kind) {
Antoine Pitrou758153b2012-05-12 15:51:51 +020010573 if (kind2 > kind) {
10574 Py_DECREF(sub);
10575 Py_DECREF(str);
Benjamin Peterson1ff2e352012-05-11 17:41:20 -050010576 return 0;
Antoine Pitrou758153b2012-05-12 15:51:51 +020010577 }
Victor Stinner7931d9a2011-11-04 00:22:48 +010010578 buf2 = _PyUnicode_AsKind(sub, kind);
Benjamin Peterson1ff2e352012-05-11 17:41:20 -050010579 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010580 if (!buf2) {
10581 Py_DECREF(sub);
Benjamin Peterson1ff2e352012-05-11 17:41:20 -050010582 Py_DECREF(str);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010583 return -1;
10584 }
10585 len1 = PyUnicode_GET_LENGTH(str);
10586 len2 = PyUnicode_GET_LENGTH(sub);
10587
Benjamin Petersonead6b532011-12-20 17:23:42 -060010588 switch (kind) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010589 case PyUnicode_1BYTE_KIND:
10590 result = ucs1lib_find(buf1, len1, buf2, len2, 0) != -1;
10591 break;
10592 case PyUnicode_2BYTE_KIND:
10593 result = ucs2lib_find(buf1, len1, buf2, len2, 0) != -1;
10594 break;
10595 case PyUnicode_4BYTE_KIND:
10596 result = ucs4lib_find(buf1, len1, buf2, len2, 0) != -1;
10597 break;
10598 default:
10599 result = -1;
10600 assert(0);
10601 }
Thomas Wouters477c8d52006-05-27 19:21:47 +000010602
10603 Py_DECREF(str);
10604 Py_DECREF(sub);
10605
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010606 if (kind2 != kind)
10607 PyMem_Free(buf2);
10608
Guido van Rossum403d68b2000-03-13 15:55:09 +000010609 return result;
Guido van Rossum403d68b2000-03-13 15:55:09 +000010610}
10611
Guido van Rossumd57fd912000-03-10 22:53:23 +000010612/* Concat to string or Unicode object giving a new Unicode object. */
10613
Alexander Belopolsky40018472011-02-26 01:02:56 +000010614PyObject *
10615PyUnicode_Concat(PyObject *left, PyObject *right)
Guido van Rossumd57fd912000-03-10 22:53:23 +000010616{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010617 PyObject *u = NULL, *v = NULL, *w;
Victor Stinner127226b2011-10-13 01:12:34 +020010618 Py_UCS4 maxchar, maxchar2;
Victor Stinner488fa492011-12-12 00:01:39 +010010619 Py_ssize_t u_len, v_len, new_len;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010620
10621 /* Coerce the two arguments */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010622 u = PyUnicode_FromObject(left);
Guido van Rossumd57fd912000-03-10 22:53:23 +000010623 if (u == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +000010624 goto onError;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010625 v = PyUnicode_FromObject(right);
Guido van Rossumd57fd912000-03-10 22:53:23 +000010626 if (v == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +000010627 goto onError;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010628
10629 /* Shortcuts */
Victor Stinnera464fc12011-10-02 20:39:30 +020010630 if (v == unicode_empty) {
Benjamin Peterson29060642009-01-31 22:14:21 +000010631 Py_DECREF(v);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010632 return u;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010633 }
Victor Stinnera464fc12011-10-02 20:39:30 +020010634 if (u == unicode_empty) {
Benjamin Peterson29060642009-01-31 22:14:21 +000010635 Py_DECREF(u);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010636 return v;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010637 }
10638
Victor Stinner488fa492011-12-12 00:01:39 +010010639 u_len = PyUnicode_GET_LENGTH(u);
10640 v_len = PyUnicode_GET_LENGTH(v);
10641 if (u_len > PY_SSIZE_T_MAX - v_len) {
10642 PyErr_SetString(PyExc_OverflowError,
10643 "strings are too large to concat");
10644 goto onError;
10645 }
10646 new_len = u_len + v_len;
10647
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010648 maxchar = PyUnicode_MAX_CHAR_VALUE(u);
Victor Stinner127226b2011-10-13 01:12:34 +020010649 maxchar2 = PyUnicode_MAX_CHAR_VALUE(v);
Benjamin Peterson7e303732013-06-10 09:19:46 -070010650 maxchar = Py_MAX(maxchar, maxchar2);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010651
Guido van Rossumd57fd912000-03-10 22:53:23 +000010652 /* Concat the two Unicode strings */
Victor Stinner488fa492011-12-12 00:01:39 +010010653 w = PyUnicode_New(new_len, maxchar);
Guido van Rossumd57fd912000-03-10 22:53:23 +000010654 if (w == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +000010655 goto onError;
Victor Stinnerd3f08822012-05-29 12:57:52 +020010656 _PyUnicode_FastCopyCharacters(w, 0, u, 0, u_len);
10657 _PyUnicode_FastCopyCharacters(w, u_len, v, 0, v_len);
Guido van Rossumd57fd912000-03-10 22:53:23 +000010658 Py_DECREF(u);
10659 Py_DECREF(v);
Victor Stinnerbb10a1f2011-10-05 01:34:17 +020010660 assert(_PyUnicode_CheckConsistency(w, 1));
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010661 return w;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010662
Benjamin Peterson29060642009-01-31 22:14:21 +000010663 onError:
Guido van Rossumd57fd912000-03-10 22:53:23 +000010664 Py_XDECREF(u);
10665 Py_XDECREF(v);
10666 return NULL;
10667}
10668
Walter Dörwald1ab83302007-05-18 17:15:44 +000010669void
Victor Stinner23e56682011-10-03 03:54:37 +020010670PyUnicode_Append(PyObject **p_left, PyObject *right)
Walter Dörwald1ab83302007-05-18 17:15:44 +000010671{
Victor Stinner23e56682011-10-03 03:54:37 +020010672 PyObject *left, *res;
Victor Stinner488fa492011-12-12 00:01:39 +010010673 Py_UCS4 maxchar, maxchar2;
10674 Py_ssize_t left_len, right_len, new_len;
Victor Stinner23e56682011-10-03 03:54:37 +020010675
10676 if (p_left == NULL) {
10677 if (!PyErr_Occurred())
10678 PyErr_BadInternalCall();
Benjamin Peterson14339b62009-01-31 16:36:08 +000010679 return;
10680 }
Victor Stinner23e56682011-10-03 03:54:37 +020010681 left = *p_left;
Serhiy Storchaka6c83e732013-01-04 12:39:34 +020010682 if (right == NULL || left == NULL || !PyUnicode_Check(left)) {
Victor Stinner23e56682011-10-03 03:54:37 +020010683 if (!PyErr_Occurred())
10684 PyErr_BadInternalCall();
10685 goto error;
10686 }
10687
Benjamin Petersonbac79492012-01-14 13:34:47 -050010688 if (PyUnicode_READY(left) == -1)
Victor Stinnere1335c72011-10-04 20:53:03 +020010689 goto error;
Benjamin Petersonbac79492012-01-14 13:34:47 -050010690 if (PyUnicode_READY(right) == -1)
Victor Stinnere1335c72011-10-04 20:53:03 +020010691 goto error;
10692
Victor Stinner488fa492011-12-12 00:01:39 +010010693 /* Shortcuts */
10694 if (left == unicode_empty) {
10695 Py_DECREF(left);
10696 Py_INCREF(right);
10697 *p_left = right;
10698 return;
10699 }
10700 if (right == unicode_empty)
10701 return;
10702
10703 left_len = PyUnicode_GET_LENGTH(left);
10704 right_len = PyUnicode_GET_LENGTH(right);
10705 if (left_len > PY_SSIZE_T_MAX - right_len) {
10706 PyErr_SetString(PyExc_OverflowError,
10707 "strings are too large to concat");
10708 goto error;
10709 }
10710 new_len = left_len + right_len;
10711
10712 if (unicode_modifiable(left)
10713 && PyUnicode_CheckExact(right)
10714 && PyUnicode_KIND(right) <= PyUnicode_KIND(left)
Victor Stinnerb0923652011-10-04 01:17:31 +020010715 /* Don't resize for ascii += latin1. Convert ascii to latin1 requires
10716 to change the structure size, but characters are stored just after
Georg Brandl7597add2011-10-05 16:36:47 +020010717 the structure, and so it requires to move all characters which is
Victor Stinnerb0923652011-10-04 01:17:31 +020010718 not so different than duplicating the string. */
Victor Stinner488fa492011-12-12 00:01:39 +010010719 && !(PyUnicode_IS_ASCII(left) && !PyUnicode_IS_ASCII(right)))
10720 {
10721 /* append inplace */
10722 if (unicode_resize(p_left, new_len) != 0) {
10723 /* XXX if _PyUnicode_Resize() fails, 'left' has been
10724 * deallocated so it cannot be put back into
10725 * 'variable'. The MemoryError is raised when there
10726 * is no value in 'variable', which might (very
10727 * remotely) be a cause of incompatibilities.
10728 */
10729 goto error;
Victor Stinner23e56682011-10-03 03:54:37 +020010730 }
Victor Stinner488fa492011-12-12 00:01:39 +010010731 /* copy 'right' into the newly allocated area of 'left' */
Victor Stinnerd3f08822012-05-29 12:57:52 +020010732 _PyUnicode_FastCopyCharacters(*p_left, left_len, right, 0, right_len);
Victor Stinner23e56682011-10-03 03:54:37 +020010733 }
Victor Stinner488fa492011-12-12 00:01:39 +010010734 else {
10735 maxchar = PyUnicode_MAX_CHAR_VALUE(left);
10736 maxchar2 = PyUnicode_MAX_CHAR_VALUE(right);
Benjamin Peterson7e303732013-06-10 09:19:46 -070010737 maxchar = Py_MAX(maxchar, maxchar2);
Victor Stinner23e56682011-10-03 03:54:37 +020010738
Victor Stinner488fa492011-12-12 00:01:39 +010010739 /* Concat the two Unicode strings */
10740 res = PyUnicode_New(new_len, maxchar);
10741 if (res == NULL)
10742 goto error;
Victor Stinnerd3f08822012-05-29 12:57:52 +020010743 _PyUnicode_FastCopyCharacters(res, 0, left, 0, left_len);
10744 _PyUnicode_FastCopyCharacters(res, left_len, right, 0, right_len);
Victor Stinner488fa492011-12-12 00:01:39 +010010745 Py_DECREF(left);
10746 *p_left = res;
10747 }
10748 assert(_PyUnicode_CheckConsistency(*p_left, 1));
Victor Stinner23e56682011-10-03 03:54:37 +020010749 return;
10750
10751error:
Victor Stinner488fa492011-12-12 00:01:39 +010010752 Py_CLEAR(*p_left);
Walter Dörwald1ab83302007-05-18 17:15:44 +000010753}
10754
10755void
10756PyUnicode_AppendAndDel(PyObject **pleft, PyObject *right)
10757{
Benjamin Peterson14339b62009-01-31 16:36:08 +000010758 PyUnicode_Append(pleft, right);
10759 Py_XDECREF(right);
Walter Dörwald1ab83302007-05-18 17:15:44 +000010760}
10761
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000010762PyDoc_STRVAR(count__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000010763 "S.count(sub[, start[, end]]) -> int\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000010764\n\
Thomas Wouters477c8d52006-05-27 19:21:47 +000010765Return the number of non-overlapping occurrences of substring sub in\n\
Benjamin Peterson142957c2008-07-04 19:55:29 +000010766string S[start:end]. Optional arguments start and end are\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000010767interpreted as in slice notation.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000010768
10769static PyObject *
Victor Stinner9db1a8b2011-10-23 20:04:37 +020010770unicode_count(PyObject *self, PyObject *args)
Guido van Rossumd57fd912000-03-10 22:53:23 +000010771{
Victor Stinner9db1a8b2011-10-23 20:04:37 +020010772 PyObject *substring;
Martin v. Löwis18e16552006-02-15 17:27:45 +000010773 Py_ssize_t start = 0;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000010774 Py_ssize_t end = PY_SSIZE_T_MAX;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010775 PyObject *result;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010776 int kind1, kind2, kind;
10777 void *buf1, *buf2;
10778 Py_ssize_t len1, len2, iresult;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010779
Jesus Ceaac451502011-04-20 17:09:23 +020010780 if (!stringlib_parse_args_finds_unicode("count", args, &substring,
10781 &start, &end))
Benjamin Peterson29060642009-01-31 22:14:21 +000010782 return NULL;
Tim Petersced69f82003-09-16 20:30:58 +000010783
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010784 kind1 = PyUnicode_KIND(self);
10785 kind2 = PyUnicode_KIND(substring);
Benjamin Petersonb63f49f2012-05-03 18:31:07 -040010786 if (kind2 > kind1)
10787 return PyLong_FromLong(0);
10788 kind = kind1;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010789 buf1 = PyUnicode_DATA(self);
10790 buf2 = PyUnicode_DATA(substring);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010791 if (kind2 != kind)
Victor Stinner7931d9a2011-11-04 00:22:48 +010010792 buf2 = _PyUnicode_AsKind(substring, kind);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010793 if (!buf2) {
10794 Py_DECREF(substring);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010795 return NULL;
10796 }
10797 len1 = PyUnicode_GET_LENGTH(self);
10798 len2 = PyUnicode_GET_LENGTH(substring);
10799
10800 ADJUST_INDICES(start, end, len1);
Benjamin Petersonead6b532011-12-20 17:23:42 -060010801 switch (kind) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010802 case PyUnicode_1BYTE_KIND:
10803 iresult = ucs1lib_count(
10804 ((Py_UCS1*)buf1) + start, end - start,
10805 buf2, len2, PY_SSIZE_T_MAX
10806 );
10807 break;
10808 case PyUnicode_2BYTE_KIND:
10809 iresult = ucs2lib_count(
10810 ((Py_UCS2*)buf1) + start, end - start,
10811 buf2, len2, PY_SSIZE_T_MAX
10812 );
10813 break;
10814 case PyUnicode_4BYTE_KIND:
10815 iresult = ucs4lib_count(
10816 ((Py_UCS4*)buf1) + start, end - start,
10817 buf2, len2, PY_SSIZE_T_MAX
10818 );
10819 break;
10820 default:
10821 assert(0); iresult = 0;
10822 }
10823
10824 result = PyLong_FromSsize_t(iresult);
10825
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010826 if (kind2 != kind)
10827 PyMem_Free(buf2);
Guido van Rossumd57fd912000-03-10 22:53:23 +000010828
10829 Py_DECREF(substring);
Thomas Wouters477c8d52006-05-27 19:21:47 +000010830
Guido van Rossumd57fd912000-03-10 22:53:23 +000010831 return result;
10832}
10833
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000010834PyDoc_STRVAR(encode__doc__,
Victor Stinnerc911bbf2010-11-07 19:04:46 +000010835 "S.encode(encoding='utf-8', errors='strict') -> bytes\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000010836\n\
Victor Stinnere14e2122010-11-07 18:41:46 +000010837Encode S using the codec registered for encoding. Default encoding\n\
10838is 'utf-8'. errors may be given to set a different error\n\
Fred Drakee4315f52000-05-09 19:53:39 +000010839handling scheme. Default is 'strict' meaning that encoding errors raise\n\
Walter Dörwald3aeb6322002-09-02 13:14:32 +000010840a UnicodeEncodeError. Other possible values are 'ignore', 'replace' and\n\
10841'xmlcharrefreplace' as well as any other name registered with\n\
10842codecs.register_error that can handle UnicodeEncodeErrors.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000010843
10844static PyObject *
Victor Stinner9db1a8b2011-10-23 20:04:37 +020010845unicode_encode(PyObject *self, PyObject *args, PyObject *kwargs)
Guido van Rossumd57fd912000-03-10 22:53:23 +000010846{
Benjamin Peterson308d6372009-09-18 21:42:35 +000010847 static char *kwlist[] = {"encoding", "errors", 0};
Guido van Rossumd57fd912000-03-10 22:53:23 +000010848 char *encoding = NULL;
10849 char *errors = NULL;
Guido van Rossum35d94282007-08-27 18:20:11 +000010850
Benjamin Peterson308d6372009-09-18 21:42:35 +000010851 if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|ss:encode",
10852 kwlist, &encoding, &errors))
Guido van Rossumd57fd912000-03-10 22:53:23 +000010853 return NULL;
Victor Stinner9db1a8b2011-10-23 20:04:37 +020010854 return PyUnicode_AsEncodedString(self, encoding, errors);
Marc-André Lemburgd2d45982004-07-08 17:57:32 +000010855}
10856
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000010857PyDoc_STRVAR(expandtabs__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000010858 "S.expandtabs([tabsize]) -> str\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000010859\n\
10860Return a copy of S where all tab characters are expanded using spaces.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000010861If tabsize is not given, a tab size of 8 characters is assumed.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000010862
10863static PyObject*
Victor Stinner9db1a8b2011-10-23 20:04:37 +020010864unicode_expandtabs(PyObject *self, PyObject *args)
Guido van Rossumd57fd912000-03-10 22:53:23 +000010865{
Antoine Pitroue71d5742011-10-04 15:55:09 +020010866 Py_ssize_t i, j, line_pos, src_len, incr;
10867 Py_UCS4 ch;
10868 PyObject *u;
10869 void *src_data, *dest_data;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010870 int tabsize = 8;
Antoine Pitroue71d5742011-10-04 15:55:09 +020010871 int kind;
Antoine Pitroue19aa382011-10-04 16:04:01 +020010872 int found;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010873
10874 if (!PyArg_ParseTuple(args, "|i:expandtabs", &tabsize))
Benjamin Peterson29060642009-01-31 22:14:21 +000010875 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010876
Antoine Pitrou22425222011-10-04 19:10:51 +020010877 if (PyUnicode_READY(self) == -1)
10878 return NULL;
10879
Thomas Wouters7e474022000-07-16 12:04:32 +000010880 /* First pass: determine size of output string */
Antoine Pitroue71d5742011-10-04 15:55:09 +020010881 src_len = PyUnicode_GET_LENGTH(self);
10882 i = j = line_pos = 0;
10883 kind = PyUnicode_KIND(self);
10884 src_data = PyUnicode_DATA(self);
Antoine Pitroue19aa382011-10-04 16:04:01 +020010885 found = 0;
Antoine Pitroue71d5742011-10-04 15:55:09 +020010886 for (; i < src_len; i++) {
10887 ch = PyUnicode_READ(kind, src_data, i);
10888 if (ch == '\t') {
Antoine Pitroue19aa382011-10-04 16:04:01 +020010889 found = 1;
Benjamin Peterson29060642009-01-31 22:14:21 +000010890 if (tabsize > 0) {
Antoine Pitroue71d5742011-10-04 15:55:09 +020010891 incr = tabsize - (line_pos % tabsize); /* cannot overflow */
Benjamin Peterson29060642009-01-31 22:14:21 +000010892 if (j > PY_SSIZE_T_MAX - incr)
Antoine Pitroue71d5742011-10-04 15:55:09 +020010893 goto overflow;
10894 line_pos += incr;
Benjamin Peterson29060642009-01-31 22:14:21 +000010895 j += incr;
Christian Heimesdd15f6c2008-03-16 00:07:10 +000010896 }
Benjamin Peterson29060642009-01-31 22:14:21 +000010897 }
Guido van Rossumd57fd912000-03-10 22:53:23 +000010898 else {
Benjamin Peterson29060642009-01-31 22:14:21 +000010899 if (j > PY_SSIZE_T_MAX - 1)
Antoine Pitroue71d5742011-10-04 15:55:09 +020010900 goto overflow;
10901 line_pos++;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010902 j++;
Antoine Pitroue71d5742011-10-04 15:55:09 +020010903 if (ch == '\n' || ch == '\r')
10904 line_pos = 0;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010905 }
Antoine Pitroue71d5742011-10-04 15:55:09 +020010906 }
Victor Stinnerc4b49542011-12-11 22:44:26 +010010907 if (!found)
10908 return unicode_result_unchanged(self);
Guido van Rossumcd16bf62007-06-13 18:07:49 +000010909
Guido van Rossumd57fd912000-03-10 22:53:23 +000010910 /* Second pass: create output string and fill it */
Antoine Pitroue71d5742011-10-04 15:55:09 +020010911 u = PyUnicode_New(j, PyUnicode_MAX_CHAR_VALUE(self));
Guido van Rossumd57fd912000-03-10 22:53:23 +000010912 if (!u)
10913 return NULL;
Antoine Pitroue71d5742011-10-04 15:55:09 +020010914 dest_data = PyUnicode_DATA(u);
Guido van Rossumd57fd912000-03-10 22:53:23 +000010915
Antoine Pitroue71d5742011-10-04 15:55:09 +020010916 i = j = line_pos = 0;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010917
Antoine Pitroue71d5742011-10-04 15:55:09 +020010918 for (; i < src_len; i++) {
10919 ch = PyUnicode_READ(kind, src_data, i);
10920 if (ch == '\t') {
Benjamin Peterson29060642009-01-31 22:14:21 +000010921 if (tabsize > 0) {
Antoine Pitroue71d5742011-10-04 15:55:09 +020010922 incr = tabsize - (line_pos % tabsize);
10923 line_pos += incr;
Victor Stinnerda79e632012-02-22 13:37:04 +010010924 FILL(kind, dest_data, ' ', j, incr);
10925 j += incr;
Benjamin Peterson29060642009-01-31 22:14:21 +000010926 }
Benjamin Peterson14339b62009-01-31 16:36:08 +000010927 }
Benjamin Peterson29060642009-01-31 22:14:21 +000010928 else {
Antoine Pitroue71d5742011-10-04 15:55:09 +020010929 line_pos++;
10930 PyUnicode_WRITE(kind, dest_data, j, ch);
Christian Heimesdd15f6c2008-03-16 00:07:10 +000010931 j++;
Antoine Pitroue71d5742011-10-04 15:55:09 +020010932 if (ch == '\n' || ch == '\r')
10933 line_pos = 0;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010934 }
Antoine Pitroue71d5742011-10-04 15:55:09 +020010935 }
10936 assert (j == PyUnicode_GET_LENGTH(u));
Victor Stinnerd3df8ab2011-11-22 01:22:34 +010010937 return unicode_result(u);
Christian Heimesdd15f6c2008-03-16 00:07:10 +000010938
Antoine Pitroue71d5742011-10-04 15:55:09 +020010939 overflow:
Christian Heimesdd15f6c2008-03-16 00:07:10 +000010940 PyErr_SetString(PyExc_OverflowError, "new string is too long");
10941 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010942}
10943
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000010944PyDoc_STRVAR(find__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000010945 "S.find(sub[, start[, end]]) -> int\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000010946\n\
10947Return the lowest index in S where substring sub is found,\n\
Senthil Kumaran53516a82011-07-27 23:33:54 +080010948such that sub is contained within S[start:end]. Optional\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000010949arguments start and end are interpreted as in slice notation.\n\
10950\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000010951Return -1 on failure.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000010952
10953static PyObject *
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010954unicode_find(PyObject *self, PyObject *args)
Guido van Rossumd57fd912000-03-10 22:53:23 +000010955{
Victor Stinner9db1a8b2011-10-23 20:04:37 +020010956 PyObject *substring;
Christian Heimes9cd17752007-11-18 19:35:23 +000010957 Py_ssize_t start;
10958 Py_ssize_t end;
Thomas Wouters477c8d52006-05-27 19:21:47 +000010959 Py_ssize_t result;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010960
Jesus Ceaac451502011-04-20 17:09:23 +020010961 if (!stringlib_parse_args_finds_unicode("find", args, &substring,
10962 &start, &end))
Guido van Rossumd57fd912000-03-10 22:53:23 +000010963 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010964
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010965 if (PyUnicode_READY(self) == -1)
10966 return NULL;
10967 if (PyUnicode_READY(substring) == -1)
10968 return NULL;
10969
Victor Stinner7931d9a2011-11-04 00:22:48 +010010970 result = any_find_slice(1, self, substring, start, end);
Guido van Rossumd57fd912000-03-10 22:53:23 +000010971
10972 Py_DECREF(substring);
Thomas Wouters477c8d52006-05-27 19:21:47 +000010973
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010974 if (result == -2)
10975 return NULL;
10976
Christian Heimes217cfd12007-12-02 14:31:20 +000010977 return PyLong_FromSsize_t(result);
Guido van Rossumd57fd912000-03-10 22:53:23 +000010978}
10979
10980static PyObject *
Victor Stinner2fe5ced2011-10-02 00:25:40 +020010981unicode_getitem(PyObject *self, Py_ssize_t index)
Guido van Rossumd57fd912000-03-10 22:53:23 +000010982{
Victor Stinnerb6cd0142012-05-03 02:17:04 +020010983 void *data;
10984 enum PyUnicode_Kind kind;
10985 Py_UCS4 ch;
10986 PyObject *res;
10987
10988 if (!PyUnicode_Check(self) || PyUnicode_READY(self) == -1) {
10989 PyErr_BadArgument();
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010990 return NULL;
Victor Stinnerb6cd0142012-05-03 02:17:04 +020010991 }
10992 if (index < 0 || index >= PyUnicode_GET_LENGTH(self)) {
10993 PyErr_SetString(PyExc_IndexError, "string index out of range");
10994 return NULL;
10995 }
10996 kind = PyUnicode_KIND(self);
10997 data = PyUnicode_DATA(self);
10998 ch = PyUnicode_READ(kind, data, index);
10999 if (ch < 256)
11000 return get_latin1_char(ch);
11001
11002 res = PyUnicode_New(1, ch);
11003 if (res == NULL)
11004 return NULL;
11005 kind = PyUnicode_KIND(res);
11006 data = PyUnicode_DATA(res);
11007 PyUnicode_WRITE(kind, data, 0, ch);
11008 assert(_PyUnicode_CheckConsistency(res, 1));
11009 return res;
Guido van Rossumd57fd912000-03-10 22:53:23 +000011010}
11011
Guido van Rossumc2504932007-09-18 19:42:40 +000011012/* Believe it or not, this produces the same value for ASCII strings
Mark Dickinson57e683e2011-09-24 18:18:40 +010011013 as bytes_hash(). */
Benjamin Peterson8f67d082010-10-17 20:54:53 +000011014static Py_hash_t
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011015unicode_hash(PyObject *self)
Guido van Rossumd57fd912000-03-10 22:53:23 +000011016{
Guido van Rossumc2504932007-09-18 19:42:40 +000011017 Py_ssize_t len;
Gregory P. Smith27cbcd62012-12-10 18:15:46 -080011018 Py_uhash_t x; /* Unsigned for defined overflow behavior. */
Guido van Rossumc2504932007-09-18 19:42:40 +000011019
Benjamin Petersonf6622c82012-04-09 14:53:07 -040011020#ifdef Py_DEBUG
Benjamin Peterson69e97272012-02-21 11:08:50 -050011021 assert(_Py_HashSecret_Initialized);
Benjamin Petersonf6622c82012-04-09 14:53:07 -040011022#endif
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011023 if (_PyUnicode_HASH(self) != -1)
11024 return _PyUnicode_HASH(self);
11025 if (PyUnicode_READY(self) == -1)
11026 return -1;
11027 len = PyUnicode_GET_LENGTH(self);
Georg Brandl16fa2a12012-02-21 00:50:13 +010011028 /*
11029 We make the hash of the empty string be 0, rather than using
11030 (prefix ^ suffix), since this slightly obfuscates the hash secret
11031 */
11032 if (len == 0) {
11033 _PyUnicode_HASH(self) = 0;
11034 return 0;
11035 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011036
11037 /* The hash function as a macro, gets expanded three times below. */
Georg Brandl2fb477c2012-02-21 00:33:36 +010011038#define HASH(P) \
11039 x ^= (Py_uhash_t) *P << 7; \
11040 while (--len >= 0) \
11041 x = (_PyHASH_MULTIPLIER * x) ^ (Py_uhash_t) *P++; \
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011042
Georg Brandl2fb477c2012-02-21 00:33:36 +010011043 x = (Py_uhash_t) _Py_HashSecret.prefix;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011044 switch (PyUnicode_KIND(self)) {
11045 case PyUnicode_1BYTE_KIND: {
11046 const unsigned char *c = PyUnicode_1BYTE_DATA(self);
11047 HASH(c);
11048 break;
11049 }
11050 case PyUnicode_2BYTE_KIND: {
11051 const Py_UCS2 *s = PyUnicode_2BYTE_DATA(self);
11052 HASH(s);
11053 break;
11054 }
11055 default: {
11056 Py_UCS4 *l;
11057 assert(PyUnicode_KIND(self) == PyUnicode_4BYTE_KIND &&
11058 "Impossible switch case in unicode_hash");
11059 l = PyUnicode_4BYTE_DATA(self);
11060 HASH(l);
11061 break;
11062 }
11063 }
Georg Brandl2fb477c2012-02-21 00:33:36 +010011064 x ^= (Py_uhash_t) PyUnicode_GET_LENGTH(self);
11065 x ^= (Py_uhash_t) _Py_HashSecret.suffix;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011066
Guido van Rossumc2504932007-09-18 19:42:40 +000011067 if (x == -1)
11068 x = -2;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011069 _PyUnicode_HASH(self) = x;
Guido van Rossumc2504932007-09-18 19:42:40 +000011070 return x;
Guido van Rossumd57fd912000-03-10 22:53:23 +000011071}
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011072#undef HASH
Guido van Rossumd57fd912000-03-10 22:53:23 +000011073
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011074PyDoc_STRVAR(index__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000011075 "S.index(sub[, start[, end]]) -> int\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000011076\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011077Like S.find() but raise ValueError when the substring is not found.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000011078
11079static PyObject *
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011080unicode_index(PyObject *self, PyObject *args)
Guido van Rossumd57fd912000-03-10 22:53:23 +000011081{
Martin v. Löwis18e16552006-02-15 17:27:45 +000011082 Py_ssize_t result;
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011083 PyObject *substring;
Christian Heimes9cd17752007-11-18 19:35:23 +000011084 Py_ssize_t start;
11085 Py_ssize_t end;
Guido van Rossumd57fd912000-03-10 22:53:23 +000011086
Jesus Ceaac451502011-04-20 17:09:23 +020011087 if (!stringlib_parse_args_finds_unicode("index", args, &substring,
11088 &start, &end))
Guido van Rossumd57fd912000-03-10 22:53:23 +000011089 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +000011090
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011091 if (PyUnicode_READY(self) == -1)
11092 return NULL;
11093 if (PyUnicode_READY(substring) == -1)
11094 return NULL;
11095
Victor Stinner7931d9a2011-11-04 00:22:48 +010011096 result = any_find_slice(1, self, substring, start, end);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011097
11098 Py_DECREF(substring);
Thomas Wouters477c8d52006-05-27 19:21:47 +000011099
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011100 if (result == -2)
11101 return NULL;
11102
Guido van Rossumd57fd912000-03-10 22:53:23 +000011103 if (result < 0) {
11104 PyErr_SetString(PyExc_ValueError, "substring not found");
11105 return NULL;
11106 }
Thomas Wouters477c8d52006-05-27 19:21:47 +000011107
Christian Heimes217cfd12007-12-02 14:31:20 +000011108 return PyLong_FromSsize_t(result);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011109}
11110
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011111PyDoc_STRVAR(islower__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000011112 "S.islower() -> bool\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000011113\n\
Guido van Rossum77f6a652002-04-03 22:41:51 +000011114Return True if all cased characters in S are lowercase and there is\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011115at least one cased character in S, False otherwise.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000011116
11117static PyObject*
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011118unicode_islower(PyObject *self)
Guido van Rossumd57fd912000-03-10 22:53:23 +000011119{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011120 Py_ssize_t i, length;
11121 int kind;
11122 void *data;
Guido van Rossumd57fd912000-03-10 22:53:23 +000011123 int cased;
11124
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011125 if (PyUnicode_READY(self) == -1)
11126 return NULL;
11127 length = PyUnicode_GET_LENGTH(self);
11128 kind = PyUnicode_KIND(self);
11129 data = PyUnicode_DATA(self);
11130
Guido van Rossumd57fd912000-03-10 22:53:23 +000011131 /* Shortcut for single character strings */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011132 if (length == 1)
11133 return PyBool_FromLong(
11134 Py_UNICODE_ISLOWER(PyUnicode_READ(kind, data, 0)));
Guido van Rossumd57fd912000-03-10 22:53:23 +000011135
Marc-André Lemburg60bc8092000-06-14 09:18:32 +000011136 /* Special case for empty strings */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011137 if (length == 0)
Benjamin Peterson29060642009-01-31 22:14:21 +000011138 return PyBool_FromLong(0);
Marc-André Lemburg60bc8092000-06-14 09:18:32 +000011139
Guido van Rossumd57fd912000-03-10 22:53:23 +000011140 cased = 0;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011141 for (i = 0; i < length; i++) {
11142 const Py_UCS4 ch = PyUnicode_READ(kind, data, i);
Tim Petersced69f82003-09-16 20:30:58 +000011143
Benjamin Peterson29060642009-01-31 22:14:21 +000011144 if (Py_UNICODE_ISUPPER(ch) || Py_UNICODE_ISTITLE(ch))
11145 return PyBool_FromLong(0);
11146 else if (!cased && Py_UNICODE_ISLOWER(ch))
11147 cased = 1;
Guido van Rossumd57fd912000-03-10 22:53:23 +000011148 }
Guido van Rossum77f6a652002-04-03 22:41:51 +000011149 return PyBool_FromLong(cased);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011150}
11151
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011152PyDoc_STRVAR(isupper__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000011153 "S.isupper() -> bool\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000011154\n\
Martin v. Löwis6828e182003-10-18 09:55:08 +000011155Return True if all cased characters in S are uppercase and there is\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011156at least one cased character in S, False otherwise.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000011157
11158static PyObject*
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011159unicode_isupper(PyObject *self)
Guido van Rossumd57fd912000-03-10 22:53:23 +000011160{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011161 Py_ssize_t i, length;
11162 int kind;
11163 void *data;
Guido van Rossumd57fd912000-03-10 22:53:23 +000011164 int cased;
11165
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011166 if (PyUnicode_READY(self) == -1)
11167 return NULL;
11168 length = PyUnicode_GET_LENGTH(self);
11169 kind = PyUnicode_KIND(self);
11170 data = PyUnicode_DATA(self);
11171
Guido van Rossumd57fd912000-03-10 22:53:23 +000011172 /* Shortcut for single character strings */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011173 if (length == 1)
11174 return PyBool_FromLong(
11175 Py_UNICODE_ISUPPER(PyUnicode_READ(kind, data, 0)) != 0);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011176
Marc-André Lemburg60bc8092000-06-14 09:18:32 +000011177 /* Special case for empty strings */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011178 if (length == 0)
Benjamin Peterson29060642009-01-31 22:14:21 +000011179 return PyBool_FromLong(0);
Marc-André Lemburg60bc8092000-06-14 09:18:32 +000011180
Guido van Rossumd57fd912000-03-10 22:53:23 +000011181 cased = 0;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011182 for (i = 0; i < length; i++) {
11183 const Py_UCS4 ch = PyUnicode_READ(kind, data, i);
Tim Petersced69f82003-09-16 20:30:58 +000011184
Benjamin Peterson29060642009-01-31 22:14:21 +000011185 if (Py_UNICODE_ISLOWER(ch) || Py_UNICODE_ISTITLE(ch))
11186 return PyBool_FromLong(0);
11187 else if (!cased && Py_UNICODE_ISUPPER(ch))
11188 cased = 1;
Guido van Rossumd57fd912000-03-10 22:53:23 +000011189 }
Guido van Rossum77f6a652002-04-03 22:41:51 +000011190 return PyBool_FromLong(cased);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011191}
11192
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011193PyDoc_STRVAR(istitle__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000011194 "S.istitle() -> bool\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000011195\n\
Martin v. Löwis6828e182003-10-18 09:55:08 +000011196Return True if S is a titlecased string and there is at least one\n\
11197character in S, i.e. upper- and titlecase characters may only\n\
11198follow uncased characters and lowercase characters only cased ones.\n\
11199Return False otherwise.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000011200
11201static PyObject*
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011202unicode_istitle(PyObject *self)
Guido van Rossumd57fd912000-03-10 22:53:23 +000011203{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011204 Py_ssize_t i, length;
11205 int kind;
11206 void *data;
Guido van Rossumd57fd912000-03-10 22:53:23 +000011207 int cased, previous_is_cased;
11208
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011209 if (PyUnicode_READY(self) == -1)
11210 return NULL;
11211 length = PyUnicode_GET_LENGTH(self);
11212 kind = PyUnicode_KIND(self);
11213 data = PyUnicode_DATA(self);
11214
Guido van Rossumd57fd912000-03-10 22:53:23 +000011215 /* Shortcut for single character strings */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011216 if (length == 1) {
11217 Py_UCS4 ch = PyUnicode_READ(kind, data, 0);
11218 return PyBool_FromLong((Py_UNICODE_ISTITLE(ch) != 0) ||
11219 (Py_UNICODE_ISUPPER(ch) != 0));
11220 }
Guido van Rossumd57fd912000-03-10 22:53:23 +000011221
Marc-André Lemburg60bc8092000-06-14 09:18:32 +000011222 /* Special case for empty strings */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011223 if (length == 0)
Benjamin Peterson29060642009-01-31 22:14:21 +000011224 return PyBool_FromLong(0);
Marc-André Lemburg60bc8092000-06-14 09:18:32 +000011225
Guido van Rossumd57fd912000-03-10 22:53:23 +000011226 cased = 0;
11227 previous_is_cased = 0;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011228 for (i = 0; i < length; i++) {
11229 const Py_UCS4 ch = PyUnicode_READ(kind, data, i);
Tim Petersced69f82003-09-16 20:30:58 +000011230
Benjamin Peterson29060642009-01-31 22:14:21 +000011231 if (Py_UNICODE_ISUPPER(ch) || Py_UNICODE_ISTITLE(ch)) {
11232 if (previous_is_cased)
11233 return PyBool_FromLong(0);
11234 previous_is_cased = 1;
11235 cased = 1;
11236 }
11237 else if (Py_UNICODE_ISLOWER(ch)) {
11238 if (!previous_is_cased)
11239 return PyBool_FromLong(0);
11240 previous_is_cased = 1;
11241 cased = 1;
11242 }
11243 else
11244 previous_is_cased = 0;
Guido van Rossumd57fd912000-03-10 22:53:23 +000011245 }
Guido van Rossum77f6a652002-04-03 22:41:51 +000011246 return PyBool_FromLong(cased);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011247}
11248
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011249PyDoc_STRVAR(isspace__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000011250 "S.isspace() -> bool\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000011251\n\
Martin v. Löwis6828e182003-10-18 09:55:08 +000011252Return True if all characters in S are whitespace\n\
11253and there is at least one character in S, False otherwise.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000011254
11255static PyObject*
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011256unicode_isspace(PyObject *self)
Guido van Rossumd57fd912000-03-10 22:53:23 +000011257{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011258 Py_ssize_t i, length;
11259 int kind;
11260 void *data;
11261
11262 if (PyUnicode_READY(self) == -1)
11263 return NULL;
11264 length = PyUnicode_GET_LENGTH(self);
11265 kind = PyUnicode_KIND(self);
11266 data = PyUnicode_DATA(self);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011267
Guido van Rossumd57fd912000-03-10 22:53:23 +000011268 /* Shortcut for single character strings */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011269 if (length == 1)
11270 return PyBool_FromLong(
11271 Py_UNICODE_ISSPACE(PyUnicode_READ(kind, data, 0)));
Guido van Rossumd57fd912000-03-10 22:53:23 +000011272
Marc-André Lemburg60bc8092000-06-14 09:18:32 +000011273 /* Special case for empty strings */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011274 if (length == 0)
Benjamin Peterson29060642009-01-31 22:14:21 +000011275 return PyBool_FromLong(0);
Marc-André Lemburg60bc8092000-06-14 09:18:32 +000011276
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011277 for (i = 0; i < length; i++) {
11278 const Py_UCS4 ch = PyUnicode_READ(kind, data, i);
Ezio Melotti93e7afc2011-08-22 14:08:38 +030011279 if (!Py_UNICODE_ISSPACE(ch))
Benjamin Peterson29060642009-01-31 22:14:21 +000011280 return PyBool_FromLong(0);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011281 }
Guido van Rossum77f6a652002-04-03 22:41:51 +000011282 return PyBool_FromLong(1);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011283}
11284
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011285PyDoc_STRVAR(isalpha__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000011286 "S.isalpha() -> bool\n\
Marc-André Lemburga7acf422000-07-05 09:49:44 +000011287\n\
Martin v. Löwis6828e182003-10-18 09:55:08 +000011288Return True if all characters in S are alphabetic\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011289and there is at least one character in S, False otherwise.");
Marc-André Lemburga7acf422000-07-05 09:49:44 +000011290
11291static PyObject*
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011292unicode_isalpha(PyObject *self)
Marc-André Lemburga7acf422000-07-05 09:49:44 +000011293{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011294 Py_ssize_t i, length;
11295 int kind;
11296 void *data;
11297
11298 if (PyUnicode_READY(self) == -1)
11299 return NULL;
11300 length = PyUnicode_GET_LENGTH(self);
11301 kind = PyUnicode_KIND(self);
11302 data = PyUnicode_DATA(self);
Marc-André Lemburga7acf422000-07-05 09:49:44 +000011303
Marc-André Lemburga7acf422000-07-05 09:49:44 +000011304 /* Shortcut for single character strings */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011305 if (length == 1)
11306 return PyBool_FromLong(
11307 Py_UNICODE_ISALPHA(PyUnicode_READ(kind, data, 0)));
Marc-André Lemburga7acf422000-07-05 09:49:44 +000011308
11309 /* Special case for empty strings */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011310 if (length == 0)
Benjamin Peterson29060642009-01-31 22:14:21 +000011311 return PyBool_FromLong(0);
Marc-André Lemburga7acf422000-07-05 09:49:44 +000011312
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011313 for (i = 0; i < length; i++) {
11314 if (!Py_UNICODE_ISALPHA(PyUnicode_READ(kind, data, i)))
Benjamin Peterson29060642009-01-31 22:14:21 +000011315 return PyBool_FromLong(0);
Marc-André Lemburga7acf422000-07-05 09:49:44 +000011316 }
Guido van Rossum77f6a652002-04-03 22:41:51 +000011317 return PyBool_FromLong(1);
Marc-André Lemburga7acf422000-07-05 09:49:44 +000011318}
11319
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011320PyDoc_STRVAR(isalnum__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000011321 "S.isalnum() -> bool\n\
Marc-André Lemburga7acf422000-07-05 09:49:44 +000011322\n\
Martin v. Löwis6828e182003-10-18 09:55:08 +000011323Return True if all characters in S are alphanumeric\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011324and there is at least one character in S, False otherwise.");
Marc-André Lemburga7acf422000-07-05 09:49:44 +000011325
11326static PyObject*
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011327unicode_isalnum(PyObject *self)
Marc-André Lemburga7acf422000-07-05 09:49:44 +000011328{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011329 int kind;
11330 void *data;
11331 Py_ssize_t len, i;
11332
11333 if (PyUnicode_READY(self) == -1)
11334 return NULL;
11335
11336 kind = PyUnicode_KIND(self);
11337 data = PyUnicode_DATA(self);
11338 len = PyUnicode_GET_LENGTH(self);
Marc-André Lemburga7acf422000-07-05 09:49:44 +000011339
Marc-André Lemburga7acf422000-07-05 09:49:44 +000011340 /* Shortcut for single character strings */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011341 if (len == 1) {
11342 const Py_UCS4 ch = PyUnicode_READ(kind, data, 0);
11343 return PyBool_FromLong(Py_UNICODE_ISALNUM(ch));
11344 }
Marc-André Lemburga7acf422000-07-05 09:49:44 +000011345
11346 /* Special case for empty strings */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011347 if (len == 0)
Benjamin Peterson29060642009-01-31 22:14:21 +000011348 return PyBool_FromLong(0);
Marc-André Lemburga7acf422000-07-05 09:49:44 +000011349
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011350 for (i = 0; i < len; i++) {
11351 const Py_UCS4 ch = PyUnicode_READ(kind, data, i);
Ezio Melotti93e7afc2011-08-22 14:08:38 +030011352 if (!Py_UNICODE_ISALNUM(ch))
Benjamin Peterson29060642009-01-31 22:14:21 +000011353 return PyBool_FromLong(0);
Marc-André Lemburga7acf422000-07-05 09:49:44 +000011354 }
Guido van Rossum77f6a652002-04-03 22:41:51 +000011355 return PyBool_FromLong(1);
Marc-André Lemburga7acf422000-07-05 09:49:44 +000011356}
11357
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011358PyDoc_STRVAR(isdecimal__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000011359 "S.isdecimal() -> bool\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000011360\n\
Guido van Rossum77f6a652002-04-03 22:41:51 +000011361Return True if there are only decimal characters in S,\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011362False otherwise.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000011363
11364static PyObject*
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011365unicode_isdecimal(PyObject *self)
Guido van Rossumd57fd912000-03-10 22:53:23 +000011366{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011367 Py_ssize_t i, length;
11368 int kind;
11369 void *data;
11370
11371 if (PyUnicode_READY(self) == -1)
11372 return NULL;
11373 length = PyUnicode_GET_LENGTH(self);
11374 kind = PyUnicode_KIND(self);
11375 data = PyUnicode_DATA(self);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011376
Guido van Rossumd57fd912000-03-10 22:53:23 +000011377 /* Shortcut for single character strings */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011378 if (length == 1)
11379 return PyBool_FromLong(
11380 Py_UNICODE_ISDECIMAL(PyUnicode_READ(kind, data, 0)));
Guido van Rossumd57fd912000-03-10 22:53:23 +000011381
Marc-André Lemburg60bc8092000-06-14 09:18:32 +000011382 /* Special case for empty strings */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011383 if (length == 0)
Benjamin Peterson29060642009-01-31 22:14:21 +000011384 return PyBool_FromLong(0);
Marc-André Lemburg60bc8092000-06-14 09:18:32 +000011385
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011386 for (i = 0; i < length; i++) {
11387 if (!Py_UNICODE_ISDECIMAL(PyUnicode_READ(kind, data, i)))
Benjamin Peterson29060642009-01-31 22:14:21 +000011388 return PyBool_FromLong(0);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011389 }
Guido van Rossum77f6a652002-04-03 22:41:51 +000011390 return PyBool_FromLong(1);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011391}
11392
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011393PyDoc_STRVAR(isdigit__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000011394 "S.isdigit() -> bool\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000011395\n\
Martin v. Löwis6828e182003-10-18 09:55:08 +000011396Return True if all characters in S are digits\n\
11397and there is at least one character in S, False otherwise.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000011398
11399static PyObject*
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011400unicode_isdigit(PyObject *self)
Guido van Rossumd57fd912000-03-10 22:53:23 +000011401{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011402 Py_ssize_t i, length;
11403 int kind;
11404 void *data;
11405
11406 if (PyUnicode_READY(self) == -1)
11407 return NULL;
11408 length = PyUnicode_GET_LENGTH(self);
11409 kind = PyUnicode_KIND(self);
11410 data = PyUnicode_DATA(self);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011411
Guido van Rossumd57fd912000-03-10 22:53:23 +000011412 /* Shortcut for single character strings */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011413 if (length == 1) {
11414 const Py_UCS4 ch = PyUnicode_READ(kind, data, 0);
11415 return PyBool_FromLong(Py_UNICODE_ISDIGIT(ch));
11416 }
Guido van Rossumd57fd912000-03-10 22:53:23 +000011417
Marc-André Lemburg60bc8092000-06-14 09:18:32 +000011418 /* Special case for empty strings */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011419 if (length == 0)
Benjamin Peterson29060642009-01-31 22:14:21 +000011420 return PyBool_FromLong(0);
Marc-André Lemburg60bc8092000-06-14 09:18:32 +000011421
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011422 for (i = 0; i < length; i++) {
11423 if (!Py_UNICODE_ISDIGIT(PyUnicode_READ(kind, data, i)))
Benjamin Peterson29060642009-01-31 22:14:21 +000011424 return PyBool_FromLong(0);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011425 }
Guido van Rossum77f6a652002-04-03 22:41:51 +000011426 return PyBool_FromLong(1);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011427}
11428
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011429PyDoc_STRVAR(isnumeric__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000011430 "S.isnumeric() -> bool\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000011431\n\
Guido van Rossum77f6a652002-04-03 22:41:51 +000011432Return True if there are only numeric characters in S,\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011433False otherwise.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000011434
11435static PyObject*
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011436unicode_isnumeric(PyObject *self)
Guido van Rossumd57fd912000-03-10 22:53:23 +000011437{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011438 Py_ssize_t i, length;
11439 int kind;
11440 void *data;
11441
11442 if (PyUnicode_READY(self) == -1)
11443 return NULL;
11444 length = PyUnicode_GET_LENGTH(self);
11445 kind = PyUnicode_KIND(self);
11446 data = PyUnicode_DATA(self);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011447
Guido van Rossumd57fd912000-03-10 22:53:23 +000011448 /* Shortcut for single character strings */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011449 if (length == 1)
11450 return PyBool_FromLong(
11451 Py_UNICODE_ISNUMERIC(PyUnicode_READ(kind, data, 0)));
Guido van Rossumd57fd912000-03-10 22:53:23 +000011452
Marc-André Lemburg60bc8092000-06-14 09:18:32 +000011453 /* Special case for empty strings */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011454 if (length == 0)
Benjamin Peterson29060642009-01-31 22:14:21 +000011455 return PyBool_FromLong(0);
Marc-André Lemburg60bc8092000-06-14 09:18:32 +000011456
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011457 for (i = 0; i < length; i++) {
11458 if (!Py_UNICODE_ISNUMERIC(PyUnicode_READ(kind, data, i)))
Benjamin Peterson29060642009-01-31 22:14:21 +000011459 return PyBool_FromLong(0);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011460 }
Guido van Rossum77f6a652002-04-03 22:41:51 +000011461 return PyBool_FromLong(1);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011462}
11463
Martin v. Löwis47383402007-08-15 07:32:56 +000011464int
11465PyUnicode_IsIdentifier(PyObject *self)
11466{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011467 int kind;
11468 void *data;
11469 Py_ssize_t i;
Ezio Melotti93e7afc2011-08-22 14:08:38 +030011470 Py_UCS4 first;
Martin v. Löwis47383402007-08-15 07:32:56 +000011471
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011472 if (PyUnicode_READY(self) == -1) {
11473 Py_FatalError("identifier not ready");
Benjamin Peterson29060642009-01-31 22:14:21 +000011474 return 0;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011475 }
11476
11477 /* Special case for empty strings */
11478 if (PyUnicode_GET_LENGTH(self) == 0)
11479 return 0;
11480 kind = PyUnicode_KIND(self);
11481 data = PyUnicode_DATA(self);
Martin v. Löwis47383402007-08-15 07:32:56 +000011482
11483 /* PEP 3131 says that the first character must be in
11484 XID_Start and subsequent characters in XID_Continue,
11485 and for the ASCII range, the 2.x rules apply (i.e
Benjamin Peterson14339b62009-01-31 16:36:08 +000011486 start with letters and underscore, continue with
Martin v. Löwis47383402007-08-15 07:32:56 +000011487 letters, digits, underscore). However, given the current
11488 definition of XID_Start and XID_Continue, it is sufficient
11489 to check just for these, except that _ must be allowed
11490 as starting an identifier. */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011491 first = PyUnicode_READ(kind, data, 0);
Benjamin Petersonf413b802011-08-12 22:17:18 -050011492 if (!_PyUnicode_IsXidStart(first) && first != 0x5F /* LOW LINE */)
Martin v. Löwis47383402007-08-15 07:32:56 +000011493 return 0;
11494
Benjamin Peterson9c6e6a02011-09-28 08:09:05 -040011495 for (i = 1; i < PyUnicode_GET_LENGTH(self); i++)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011496 if (!_PyUnicode_IsXidContinue(PyUnicode_READ(kind, data, i)))
Benjamin Peterson29060642009-01-31 22:14:21 +000011497 return 0;
Martin v. Löwis47383402007-08-15 07:32:56 +000011498 return 1;
11499}
11500
11501PyDoc_STRVAR(isidentifier__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000011502 "S.isidentifier() -> bool\n\
Martin v. Löwis47383402007-08-15 07:32:56 +000011503\n\
11504Return True if S is a valid identifier according\n\
Raymond Hettinger378170d2013-03-23 08:21:12 -070011505to the language definition.\n\
11506\n\
11507Use keyword.iskeyword() to test for reserved identifiers\n\
11508such as \"def\" and \"class\".\n");
Martin v. Löwis47383402007-08-15 07:32:56 +000011509
11510static PyObject*
11511unicode_isidentifier(PyObject *self)
11512{
11513 return PyBool_FromLong(PyUnicode_IsIdentifier(self));
11514}
11515
Georg Brandl559e5d72008-06-11 18:37:52 +000011516PyDoc_STRVAR(isprintable__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000011517 "S.isprintable() -> bool\n\
Georg Brandl559e5d72008-06-11 18:37:52 +000011518\n\
11519Return True if all characters in S are considered\n\
11520printable in repr() or S is empty, False otherwise.");
11521
11522static PyObject*
11523unicode_isprintable(PyObject *self)
11524{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011525 Py_ssize_t i, length;
11526 int kind;
11527 void *data;
11528
11529 if (PyUnicode_READY(self) == -1)
11530 return NULL;
11531 length = PyUnicode_GET_LENGTH(self);
11532 kind = PyUnicode_KIND(self);
11533 data = PyUnicode_DATA(self);
Georg Brandl559e5d72008-06-11 18:37:52 +000011534
11535 /* Shortcut for single character strings */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011536 if (length == 1)
11537 return PyBool_FromLong(
11538 Py_UNICODE_ISPRINTABLE(PyUnicode_READ(kind, data, 0)));
Georg Brandl559e5d72008-06-11 18:37:52 +000011539
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011540 for (i = 0; i < length; i++) {
11541 if (!Py_UNICODE_ISPRINTABLE(PyUnicode_READ(kind, data, i))) {
Georg Brandl559e5d72008-06-11 18:37:52 +000011542 Py_RETURN_FALSE;
11543 }
11544 }
11545 Py_RETURN_TRUE;
11546}
11547
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011548PyDoc_STRVAR(join__doc__,
Georg Brandl495f7b52009-10-27 15:28:25 +000011549 "S.join(iterable) -> str\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000011550\n\
11551Return a string which is the concatenation of the strings in the\n\
Georg Brandl495f7b52009-10-27 15:28:25 +000011552iterable. The separator between elements is S.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000011553
11554static PyObject*
Martin v. Löwise3eb1f22001-08-16 13:15:00 +000011555unicode_join(PyObject *self, PyObject *data)
Guido van Rossumd57fd912000-03-10 22:53:23 +000011556{
Martin v. Löwise3eb1f22001-08-16 13:15:00 +000011557 return PyUnicode_Join(self, data);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011558}
11559
Martin v. Löwis18e16552006-02-15 17:27:45 +000011560static Py_ssize_t
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011561unicode_length(PyObject *self)
Guido van Rossumd57fd912000-03-10 22:53:23 +000011562{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011563 if (PyUnicode_READY(self) == -1)
11564 return -1;
11565 return PyUnicode_GET_LENGTH(self);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011566}
11567
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011568PyDoc_STRVAR(ljust__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000011569 "S.ljust(width[, fillchar]) -> str\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000011570\n\
Benjamin Petersonf10a79a2008-10-11 00:49:57 +000011571Return S left-justified in a Unicode string of length width. Padding is\n\
Raymond Hettinger4f8f9762003-11-26 08:21:35 +000011572done using the specified fill character (default is a space).");
Guido van Rossumd57fd912000-03-10 22:53:23 +000011573
11574static PyObject *
Victor Stinner9310abb2011-10-05 00:59:23 +020011575unicode_ljust(PyObject *self, PyObject *args)
Guido van Rossumd57fd912000-03-10 22:53:23 +000011576{
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000011577 Py_ssize_t width;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011578 Py_UCS4 fillchar = ' ';
11579
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000011580 if (!PyArg_ParseTuple(args, "n|O&:ljust", &width, convert_uc, &fillchar))
Guido van Rossumd57fd912000-03-10 22:53:23 +000011581 return NULL;
11582
Benjamin Petersonbac79492012-01-14 13:34:47 -050011583 if (PyUnicode_READY(self) == -1)
Victor Stinnerc4b49542011-12-11 22:44:26 +010011584 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +000011585
Victor Stinnerc4b49542011-12-11 22:44:26 +010011586 if (PyUnicode_GET_LENGTH(self) >= width)
11587 return unicode_result_unchanged(self);
11588
11589 return pad(self, 0, width - PyUnicode_GET_LENGTH(self), fillchar);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011590}
11591
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011592PyDoc_STRVAR(lower__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000011593 "S.lower() -> str\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000011594\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011595Return a copy of the string S converted to lowercase.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000011596
11597static PyObject*
Victor Stinner9310abb2011-10-05 00:59:23 +020011598unicode_lower(PyObject *self)
Guido van Rossumd57fd912000-03-10 22:53:23 +000011599{
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -050011600 if (PyUnicode_READY(self) == -1)
11601 return NULL;
11602 if (PyUnicode_IS_ASCII(self))
11603 return ascii_upper_or_lower(self, 1);
Victor Stinnerb0800dc2012-02-25 00:47:08 +010011604 return case_operation(self, do_lower);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011605}
11606
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011607#define LEFTSTRIP 0
11608#define RIGHTSTRIP 1
11609#define BOTHSTRIP 2
11610
11611/* Arrays indexed by above */
11612static const char *stripformat[] = {"|O:lstrip", "|O:rstrip", "|O:strip"};
11613
11614#define STRIPNAME(i) (stripformat[i]+3)
11615
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011616/* externally visible for str.strip(unicode) */
11617PyObject *
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011618_PyUnicode_XStrip(PyObject *self, int striptype, PyObject *sepobj)
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011619{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011620 void *data;
11621 int kind;
11622 Py_ssize_t i, j, len;
11623 BLOOM_MASK sepmask;
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011624
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011625 if (PyUnicode_READY(self) == -1 || PyUnicode_READY(sepobj) == -1)
11626 return NULL;
11627
11628 kind = PyUnicode_KIND(self);
11629 data = PyUnicode_DATA(self);
11630 len = PyUnicode_GET_LENGTH(self);
11631 sepmask = make_bloom_mask(PyUnicode_KIND(sepobj),
11632 PyUnicode_DATA(sepobj),
11633 PyUnicode_GET_LENGTH(sepobj));
Thomas Wouters477c8d52006-05-27 19:21:47 +000011634
Benjamin Peterson14339b62009-01-31 16:36:08 +000011635 i = 0;
11636 if (striptype != RIGHTSTRIP) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011637 while (i < len &&
11638 BLOOM_MEMBER(sepmask, PyUnicode_READ(kind, data, i), sepobj)) {
Benjamin Peterson29060642009-01-31 22:14:21 +000011639 i++;
11640 }
Benjamin Peterson14339b62009-01-31 16:36:08 +000011641 }
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011642
Benjamin Peterson14339b62009-01-31 16:36:08 +000011643 j = len;
11644 if (striptype != LEFTSTRIP) {
Benjamin Peterson29060642009-01-31 22:14:21 +000011645 do {
11646 j--;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011647 } while (j >= i &&
11648 BLOOM_MEMBER(sepmask, PyUnicode_READ(kind, data, j), sepobj));
Benjamin Peterson29060642009-01-31 22:14:21 +000011649 j++;
Benjamin Peterson14339b62009-01-31 16:36:08 +000011650 }
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011651
Victor Stinner7931d9a2011-11-04 00:22:48 +010011652 return PyUnicode_Substring(self, i, j);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011653}
11654
11655PyObject*
11656PyUnicode_Substring(PyObject *self, Py_ssize_t start, Py_ssize_t end)
11657{
11658 unsigned char *data;
11659 int kind;
Victor Stinner12bab6d2011-10-01 01:53:49 +020011660 Py_ssize_t length;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011661
Victor Stinnerde636f32011-10-01 03:55:54 +020011662 if (PyUnicode_READY(self) == -1)
11663 return NULL;
11664
Victor Stinner684d5fd2012-05-03 02:32:34 +020011665 length = PyUnicode_GET_LENGTH(self);
11666 end = Py_MIN(end, length);
Victor Stinnerde636f32011-10-01 03:55:54 +020011667
Victor Stinner684d5fd2012-05-03 02:32:34 +020011668 if (start == 0 && end == length)
Victor Stinnerc4b49542011-12-11 22:44:26 +010011669 return unicode_result_unchanged(self);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011670
Victor Stinnerde636f32011-10-01 03:55:54 +020011671 if (start < 0 || end < 0) {
Victor Stinner12bab6d2011-10-01 01:53:49 +020011672 PyErr_SetString(PyExc_IndexError, "string index out of range");
11673 return NULL;
11674 }
Serhiy Storchaka678db842013-01-26 12:16:36 +020011675 if (start >= length || end < start)
11676 _Py_RETURN_UNICODE_EMPTY();
Victor Stinner12bab6d2011-10-01 01:53:49 +020011677
Victor Stinner684d5fd2012-05-03 02:32:34 +020011678 length = end - start;
Victor Stinnerb9275c12011-10-05 14:01:42 +020011679 if (PyUnicode_IS_ASCII(self)) {
Victor Stinnerb9275c12011-10-05 14:01:42 +020011680 data = PyUnicode_1BYTE_DATA(self);
Victor Stinnerd3f08822012-05-29 12:57:52 +020011681 return _PyUnicode_FromASCII((char*)(data + start), length);
Victor Stinnerb9275c12011-10-05 14:01:42 +020011682 }
11683 else {
11684 kind = PyUnicode_KIND(self);
11685 data = PyUnicode_1BYTE_DATA(self);
11686 return PyUnicode_FromKindAndData(kind,
Martin v. Löwisc47adb02011-10-07 20:55:35 +020011687 data + kind * start,
Victor Stinnerb9275c12011-10-05 14:01:42 +020011688 length);
11689 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011690}
Guido van Rossumd57fd912000-03-10 22:53:23 +000011691
11692static PyObject *
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011693do_strip(PyObject *self, int striptype)
Guido van Rossumd57fd912000-03-10 22:53:23 +000011694{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011695 int kind;
11696 void *data;
11697 Py_ssize_t len, i, j;
11698
11699 if (PyUnicode_READY(self) == -1)
11700 return NULL;
11701
11702 kind = PyUnicode_KIND(self);
11703 data = PyUnicode_DATA(self);
11704 len = PyUnicode_GET_LENGTH(self);
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011705
Benjamin Peterson14339b62009-01-31 16:36:08 +000011706 i = 0;
11707 if (striptype != RIGHTSTRIP) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011708 while (i < len && Py_UNICODE_ISSPACE(PyUnicode_READ(kind, data, i))) {
Benjamin Peterson14339b62009-01-31 16:36:08 +000011709 i++;
11710 }
11711 }
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011712
Benjamin Peterson14339b62009-01-31 16:36:08 +000011713 j = len;
11714 if (striptype != LEFTSTRIP) {
11715 do {
11716 j--;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011717 } while (j >= i && Py_UNICODE_ISSPACE(PyUnicode_READ(kind, data, j)));
Benjamin Peterson14339b62009-01-31 16:36:08 +000011718 j++;
11719 }
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011720
Victor Stinner7931d9a2011-11-04 00:22:48 +010011721 return PyUnicode_Substring(self, i, j);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011722}
11723
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011724
11725static PyObject *
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011726do_argstrip(PyObject *self, int striptype, PyObject *args)
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011727{
Benjamin Peterson14339b62009-01-31 16:36:08 +000011728 PyObject *sep = NULL;
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011729
Benjamin Peterson14339b62009-01-31 16:36:08 +000011730 if (!PyArg_ParseTuple(args, (char *)stripformat[striptype], &sep))
11731 return NULL;
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011732
Benjamin Peterson14339b62009-01-31 16:36:08 +000011733 if (sep != NULL && sep != Py_None) {
11734 if (PyUnicode_Check(sep))
11735 return _PyUnicode_XStrip(self, striptype, sep);
11736 else {
11737 PyErr_Format(PyExc_TypeError,
Benjamin Peterson29060642009-01-31 22:14:21 +000011738 "%s arg must be None or str",
11739 STRIPNAME(striptype));
Benjamin Peterson14339b62009-01-31 16:36:08 +000011740 return NULL;
11741 }
11742 }
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011743
Benjamin Peterson14339b62009-01-31 16:36:08 +000011744 return do_strip(self, striptype);
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011745}
11746
11747
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011748PyDoc_STRVAR(strip__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000011749 "S.strip([chars]) -> str\n\
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011750\n\
11751Return a copy of the string S with leading and trailing\n\
11752whitespace removed.\n\
Benjamin Peterson142957c2008-07-04 19:55:29 +000011753If chars is given and not None, remove characters in chars instead.");
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011754
11755static PyObject *
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011756unicode_strip(PyObject *self, PyObject *args)
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011757{
Benjamin Peterson14339b62009-01-31 16:36:08 +000011758 if (PyTuple_GET_SIZE(args) == 0)
11759 return do_strip(self, BOTHSTRIP); /* Common case */
11760 else
11761 return do_argstrip(self, BOTHSTRIP, args);
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011762}
11763
11764
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011765PyDoc_STRVAR(lstrip__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000011766 "S.lstrip([chars]) -> str\n\
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011767\n\
11768Return a copy of the string S with leading whitespace removed.\n\
Benjamin Peterson142957c2008-07-04 19:55:29 +000011769If chars is given and not None, remove characters in chars instead.");
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011770
11771static PyObject *
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011772unicode_lstrip(PyObject *self, PyObject *args)
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011773{
Benjamin Peterson14339b62009-01-31 16:36:08 +000011774 if (PyTuple_GET_SIZE(args) == 0)
11775 return do_strip(self, LEFTSTRIP); /* Common case */
11776 else
11777 return do_argstrip(self, LEFTSTRIP, args);
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011778}
11779
11780
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011781PyDoc_STRVAR(rstrip__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000011782 "S.rstrip([chars]) -> str\n\
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011783\n\
11784Return a copy of the string S with trailing whitespace removed.\n\
Benjamin Peterson142957c2008-07-04 19:55:29 +000011785If chars is given and not None, remove characters in chars instead.");
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011786
11787static PyObject *
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011788unicode_rstrip(PyObject *self, PyObject *args)
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011789{
Benjamin Peterson14339b62009-01-31 16:36:08 +000011790 if (PyTuple_GET_SIZE(args) == 0)
11791 return do_strip(self, RIGHTSTRIP); /* Common case */
11792 else
11793 return do_argstrip(self, RIGHTSTRIP, args);
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011794}
11795
11796
Guido van Rossumd57fd912000-03-10 22:53:23 +000011797static PyObject*
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011798unicode_repeat(PyObject *str, Py_ssize_t len)
Guido van Rossumd57fd912000-03-10 22:53:23 +000011799{
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011800 PyObject *u;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011801 Py_ssize_t nchars, n;
Guido van Rossumd57fd912000-03-10 22:53:23 +000011802
Serhiy Storchaka05997252013-01-26 12:14:02 +020011803 if (len < 1)
11804 _Py_RETURN_UNICODE_EMPTY();
Guido van Rossumd57fd912000-03-10 22:53:23 +000011805
Victor Stinnerc4b49542011-12-11 22:44:26 +010011806 /* no repeat, return original string */
11807 if (len == 1)
11808 return unicode_result_unchanged(str);
Tim Peters8f422462000-09-09 06:13:41 +000011809
Benjamin Petersonbac79492012-01-14 13:34:47 -050011810 if (PyUnicode_READY(str) == -1)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011811 return NULL;
11812
Victor Stinnerc759f3e2011-10-01 03:09:58 +020011813 if (PyUnicode_GET_LENGTH(str) > PY_SSIZE_T_MAX / len) {
Victor Stinner67ca64c2011-10-01 02:47:29 +020011814 PyErr_SetString(PyExc_OverflowError,
11815 "repeated string is too long");
11816 return NULL;
11817 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011818 nchars = len * PyUnicode_GET_LENGTH(str);
Victor Stinner67ca64c2011-10-01 02:47:29 +020011819
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011820 u = PyUnicode_New(nchars, PyUnicode_MAX_CHAR_VALUE(str));
Guido van Rossumd57fd912000-03-10 22:53:23 +000011821 if (!u)
11822 return NULL;
Victor Stinner67ca64c2011-10-01 02:47:29 +020011823 assert(PyUnicode_KIND(u) == PyUnicode_KIND(str));
Guido van Rossumd57fd912000-03-10 22:53:23 +000011824
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011825 if (PyUnicode_GET_LENGTH(str) == 1) {
11826 const int kind = PyUnicode_KIND(str);
11827 const Py_UCS4 fill_char = PyUnicode_READ(kind, PyUnicode_DATA(str), 0);
Victor Stinner73f53b52011-12-18 03:26:31 +010011828 if (kind == PyUnicode_1BYTE_KIND) {
11829 void *to = PyUnicode_DATA(u);
Victor Stinner67ca64c2011-10-01 02:47:29 +020011830 memset(to, (unsigned char)fill_char, len);
Victor Stinner73f53b52011-12-18 03:26:31 +010011831 }
11832 else if (kind == PyUnicode_2BYTE_KIND) {
11833 Py_UCS2 *ucs2 = PyUnicode_2BYTE_DATA(u);
Victor Stinner67ca64c2011-10-01 02:47:29 +020011834 for (n = 0; n < len; ++n)
Victor Stinner73f53b52011-12-18 03:26:31 +010011835 ucs2[n] = fill_char;
11836 } else {
11837 Py_UCS4 *ucs4 = PyUnicode_4BYTE_DATA(u);
11838 assert(kind == PyUnicode_4BYTE_KIND);
11839 for (n = 0; n < len; ++n)
11840 ucs4[n] = fill_char;
Victor Stinner67ca64c2011-10-01 02:47:29 +020011841 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011842 }
11843 else {
11844 /* number of characters copied this far */
11845 Py_ssize_t done = PyUnicode_GET_LENGTH(str);
Martin v. Löwisc47adb02011-10-07 20:55:35 +020011846 const Py_ssize_t char_size = PyUnicode_KIND(str);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011847 char *to = (char *) PyUnicode_DATA(u);
11848 Py_MEMCPY(to, PyUnicode_DATA(str),
11849 PyUnicode_GET_LENGTH(str) * char_size);
Benjamin Peterson29060642009-01-31 22:14:21 +000011850 while (done < nchars) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011851 n = (done <= nchars-done) ? done : nchars-done;
11852 Py_MEMCPY(to + (done * char_size), to, n * char_size);
Thomas Wouters477c8d52006-05-27 19:21:47 +000011853 done += n;
Benjamin Peterson29060642009-01-31 22:14:21 +000011854 }
Guido van Rossumd57fd912000-03-10 22:53:23 +000011855 }
11856
Victor Stinnerbb10a1f2011-10-05 01:34:17 +020011857 assert(_PyUnicode_CheckConsistency(u, 1));
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011858 return u;
Guido van Rossumd57fd912000-03-10 22:53:23 +000011859}
11860
Alexander Belopolsky40018472011-02-26 01:02:56 +000011861PyObject *
11862PyUnicode_Replace(PyObject *obj,
11863 PyObject *subobj,
11864 PyObject *replobj,
11865 Py_ssize_t maxcount)
Guido van Rossumd57fd912000-03-10 22:53:23 +000011866{
11867 PyObject *self;
11868 PyObject *str1;
11869 PyObject *str2;
11870 PyObject *result;
11871
11872 self = PyUnicode_FromObject(obj);
Benjamin Peterson22a29702012-01-02 09:00:30 -060011873 if (self == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +000011874 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +000011875 str1 = PyUnicode_FromObject(subobj);
Benjamin Peterson22a29702012-01-02 09:00:30 -060011876 if (str1 == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +000011877 Py_DECREF(self);
11878 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +000011879 }
11880 str2 = PyUnicode_FromObject(replobj);
Benjamin Peterson22a29702012-01-02 09:00:30 -060011881 if (str2 == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +000011882 Py_DECREF(self);
11883 Py_DECREF(str1);
11884 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +000011885 }
Benjamin Peterson22a29702012-01-02 09:00:30 -060011886 if (PyUnicode_READY(self) == -1 ||
11887 PyUnicode_READY(str1) == -1 ||
11888 PyUnicode_READY(str2) == -1)
11889 result = NULL;
11890 else
11891 result = replace(self, str1, str2, maxcount);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011892 Py_DECREF(self);
11893 Py_DECREF(str1);
11894 Py_DECREF(str2);
11895 return result;
11896}
11897
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011898PyDoc_STRVAR(replace__doc__,
Ezio Melottic1897e72010-06-26 18:50:39 +000011899 "S.replace(old, new[, count]) -> str\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000011900\n\
11901Return a copy of S with all occurrences of substring\n\
Georg Brandlf08a9dd2008-06-10 16:57:31 +000011902old replaced by new. If the optional argument count is\n\
11903given, only the first count occurrences are replaced.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000011904
11905static PyObject*
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011906unicode_replace(PyObject *self, PyObject *args)
Guido van Rossumd57fd912000-03-10 22:53:23 +000011907{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011908 PyObject *str1;
11909 PyObject *str2;
Martin v. Löwis18e16552006-02-15 17:27:45 +000011910 Py_ssize_t maxcount = -1;
Guido van Rossumd57fd912000-03-10 22:53:23 +000011911 PyObject *result;
11912
Martin v. Löwis18e16552006-02-15 17:27:45 +000011913 if (!PyArg_ParseTuple(args, "OO|n:replace", &str1, &str2, &maxcount))
Guido van Rossumd57fd912000-03-10 22:53:23 +000011914 return NULL;
Benjamin Peterson22a29702012-01-02 09:00:30 -060011915 if (PyUnicode_READY(self) == -1)
Benjamin Peterson29060642009-01-31 22:14:21 +000011916 return NULL;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011917 str1 = PyUnicode_FromObject(str1);
Benjamin Peterson22a29702012-01-02 09:00:30 -060011918 if (str1 == NULL)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011919 return NULL;
11920 str2 = PyUnicode_FromObject(str2);
Benjamin Peterson22a29702012-01-02 09:00:30 -060011921 if (str2 == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +000011922 Py_DECREF(str1);
11923 return NULL;
Walter Dörwaldf6b56ae2003-02-09 23:42:56 +000011924 }
Benjamin Peterson22a29702012-01-02 09:00:30 -060011925 if (PyUnicode_READY(str1) == -1 || PyUnicode_READY(str2) == -1)
11926 result = NULL;
11927 else
11928 result = replace(self, str1, str2, maxcount);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011929
11930 Py_DECREF(str1);
11931 Py_DECREF(str2);
11932 return result;
11933}
11934
Alexander Belopolsky40018472011-02-26 01:02:56 +000011935static PyObject *
11936unicode_repr(PyObject *unicode)
Guido van Rossumd57fd912000-03-10 22:53:23 +000011937{
Walter Dörwald79e913e2007-05-12 11:08:06 +000011938 PyObject *repr;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011939 Py_ssize_t isize;
11940 Py_ssize_t osize, squote, dquote, i, o;
11941 Py_UCS4 max, quote;
11942 int ikind, okind;
11943 void *idata, *odata;
Walter Dörwald79e913e2007-05-12 11:08:06 +000011944
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011945 if (PyUnicode_READY(unicode) == -1)
Walter Dörwald79e913e2007-05-12 11:08:06 +000011946 return NULL;
11947
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011948 isize = PyUnicode_GET_LENGTH(unicode);
11949 idata = PyUnicode_DATA(unicode);
Walter Dörwald79e913e2007-05-12 11:08:06 +000011950
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011951 /* Compute length of output, quote characters, and
11952 maximum character */
11953 osize = 2; /* quotes */
11954 max = 127;
11955 squote = dquote = 0;
11956 ikind = PyUnicode_KIND(unicode);
11957 for (i = 0; i < isize; i++) {
11958 Py_UCS4 ch = PyUnicode_READ(ikind, idata, i);
11959 switch (ch) {
11960 case '\'': squote++; osize++; break;
11961 case '"': dquote++; osize++; break;
11962 case '\\': case '\t': case '\r': case '\n':
11963 osize += 2; break;
11964 default:
11965 /* Fast-path ASCII */
11966 if (ch < ' ' || ch == 0x7f)
11967 osize += 4; /* \xHH */
11968 else if (ch < 0x7f)
11969 osize++;
11970 else if (Py_UNICODE_ISPRINTABLE(ch)) {
11971 osize++;
11972 max = ch > max ? ch : max;
11973 }
11974 else if (ch < 0x100)
11975 osize += 4; /* \xHH */
11976 else if (ch < 0x10000)
11977 osize += 6; /* \uHHHH */
11978 else
11979 osize += 10; /* \uHHHHHHHH */
11980 }
11981 }
11982
11983 quote = '\'';
11984 if (squote) {
11985 if (dquote)
11986 /* Both squote and dquote present. Use squote,
11987 and escape them */
11988 osize += squote;
11989 else
11990 quote = '"';
11991 }
11992
11993 repr = PyUnicode_New(osize, max);
11994 if (repr == NULL)
11995 return NULL;
11996 okind = PyUnicode_KIND(repr);
11997 odata = PyUnicode_DATA(repr);
11998
11999 PyUnicode_WRITE(okind, odata, 0, quote);
12000 PyUnicode_WRITE(okind, odata, osize-1, quote);
12001
12002 for (i = 0, o = 1; i < isize; i++) {
12003 Py_UCS4 ch = PyUnicode_READ(ikind, idata, i);
Walter Dörwald79e913e2007-05-12 11:08:06 +000012004
12005 /* Escape quotes and backslashes */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012006 if ((ch == quote) || (ch == '\\')) {
12007 PyUnicode_WRITE(okind, odata, o++, '\\');
12008 PyUnicode_WRITE(okind, odata, o++, ch);
Walter Dörwald79e913e2007-05-12 11:08:06 +000012009 continue;
12010 }
12011
Benjamin Peterson29060642009-01-31 22:14:21 +000012012 /* Map special whitespace to '\t', \n', '\r' */
Georg Brandl559e5d72008-06-11 18:37:52 +000012013 if (ch == '\t') {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012014 PyUnicode_WRITE(okind, odata, o++, '\\');
12015 PyUnicode_WRITE(okind, odata, o++, 't');
Walter Dörwald79e913e2007-05-12 11:08:06 +000012016 }
12017 else if (ch == '\n') {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012018 PyUnicode_WRITE(okind, odata, o++, '\\');
12019 PyUnicode_WRITE(okind, odata, o++, 'n');
Walter Dörwald79e913e2007-05-12 11:08:06 +000012020 }
12021 else if (ch == '\r') {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012022 PyUnicode_WRITE(okind, odata, o++, '\\');
12023 PyUnicode_WRITE(okind, odata, o++, 'r');
Walter Dörwald79e913e2007-05-12 11:08:06 +000012024 }
12025
12026 /* Map non-printable US ASCII to '\xhh' */
Georg Brandl559e5d72008-06-11 18:37:52 +000012027 else if (ch < ' ' || ch == 0x7F) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012028 PyUnicode_WRITE(okind, odata, o++, '\\');
12029 PyUnicode_WRITE(okind, odata, o++, 'x');
Victor Stinnerf5cff562011-10-14 02:13:11 +020012030 PyUnicode_WRITE(okind, odata, o++, Py_hexdigits[(ch >> 4) & 0x000F]);
12031 PyUnicode_WRITE(okind, odata, o++, Py_hexdigits[ch & 0x000F]);
Walter Dörwald79e913e2007-05-12 11:08:06 +000012032 }
12033
Georg Brandl559e5d72008-06-11 18:37:52 +000012034 /* Copy ASCII characters as-is */
12035 else if (ch < 0x7F) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012036 PyUnicode_WRITE(okind, odata, o++, ch);
Georg Brandl559e5d72008-06-11 18:37:52 +000012037 }
12038
Benjamin Peterson29060642009-01-31 22:14:21 +000012039 /* Non-ASCII characters */
Georg Brandl559e5d72008-06-11 18:37:52 +000012040 else {
Benjamin Peterson14339b62009-01-31 16:36:08 +000012041 /* Map Unicode whitespace and control characters
Georg Brandl559e5d72008-06-11 18:37:52 +000012042 (categories Z* and C* except ASCII space)
12043 */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012044 if (!Py_UNICODE_ISPRINTABLE(ch)) {
Kristján Valur Jónsson55e5dc82012-06-06 21:58:08 +000012045 PyUnicode_WRITE(okind, odata, o++, '\\');
Georg Brandl559e5d72008-06-11 18:37:52 +000012046 /* Map 8-bit characters to '\xhh' */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012047 if (ch <= 0xff) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012048 PyUnicode_WRITE(okind, odata, o++, 'x');
Victor Stinnerf5cff562011-10-14 02:13:11 +020012049 PyUnicode_WRITE(okind, odata, o++, Py_hexdigits[(ch >> 4) & 0x000F]);
12050 PyUnicode_WRITE(okind, odata, o++, Py_hexdigits[ch & 0x000F]);
Georg Brandl559e5d72008-06-11 18:37:52 +000012051 }
Kristján Valur Jónsson55e5dc82012-06-06 21:58:08 +000012052 /* Map 16-bit characters to '\uxxxx' */
12053 else if (ch <= 0xffff) {
12054 PyUnicode_WRITE(okind, odata, o++, 'u');
Victor Stinnerf5cff562011-10-14 02:13:11 +020012055 PyUnicode_WRITE(okind, odata, o++, Py_hexdigits[(ch >> 12) & 0xF]);
12056 PyUnicode_WRITE(okind, odata, o++, Py_hexdigits[(ch >> 8) & 0xF]);
12057 PyUnicode_WRITE(okind, odata, o++, Py_hexdigits[(ch >> 4) & 0xF]);
12058 PyUnicode_WRITE(okind, odata, o++, Py_hexdigits[ch & 0xF]);
Georg Brandl559e5d72008-06-11 18:37:52 +000012059 }
Kristján Valur Jónsson55e5dc82012-06-06 21:58:08 +000012060 /* Map 21-bit characters to '\U00xxxxxx' */
Georg Brandl559e5d72008-06-11 18:37:52 +000012061 else {
Kristján Valur Jónsson55e5dc82012-06-06 21:58:08 +000012062 PyUnicode_WRITE(okind, odata, o++, 'U');
12063 PyUnicode_WRITE(okind, odata, o++, Py_hexdigits[(ch >> 28) & 0xF]);
12064 PyUnicode_WRITE(okind, odata, o++, Py_hexdigits[(ch >> 24) & 0xF]);
12065 PyUnicode_WRITE(okind, odata, o++, Py_hexdigits[(ch >> 20) & 0xF]);
12066 PyUnicode_WRITE(okind, odata, o++, Py_hexdigits[(ch >> 16) & 0xF]);
Victor Stinnerf5cff562011-10-14 02:13:11 +020012067 PyUnicode_WRITE(okind, odata, o++, Py_hexdigits[(ch >> 12) & 0xF]);
12068 PyUnicode_WRITE(okind, odata, o++, Py_hexdigits[(ch >> 8) & 0xF]);
12069 PyUnicode_WRITE(okind, odata, o++, Py_hexdigits[(ch >> 4) & 0xF]);
12070 PyUnicode_WRITE(okind, odata, o++, Py_hexdigits[ch & 0xF]);
Georg Brandl559e5d72008-06-11 18:37:52 +000012071 }
12072 }
12073 /* Copy characters as-is */
12074 else {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012075 PyUnicode_WRITE(okind, odata, o++, ch);
Georg Brandl559e5d72008-06-11 18:37:52 +000012076 }
12077 }
Walter Dörwald79e913e2007-05-12 11:08:06 +000012078 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012079 /* Closing quote already added at the beginning */
Victor Stinner05d11892011-10-06 01:13:58 +020012080 assert(_PyUnicode_CheckConsistency(repr, 1));
Walter Dörwald79e913e2007-05-12 11:08:06 +000012081 return repr;
Guido van Rossumd57fd912000-03-10 22:53:23 +000012082}
12083
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000012084PyDoc_STRVAR(rfind__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000012085 "S.rfind(sub[, start[, end]]) -> int\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000012086\n\
12087Return the highest index in S where substring sub is found,\n\
Senthil Kumaran53516a82011-07-27 23:33:54 +080012088such that sub is contained within S[start:end]. Optional\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000012089arguments start and end are interpreted as in slice notation.\n\
12090\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000012091Return -1 on failure.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000012092
12093static PyObject *
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012094unicode_rfind(PyObject *self, PyObject *args)
Guido van Rossumd57fd912000-03-10 22:53:23 +000012095{
Victor Stinner9db1a8b2011-10-23 20:04:37 +020012096 PyObject *substring;
Christian Heimes9cd17752007-11-18 19:35:23 +000012097 Py_ssize_t start;
12098 Py_ssize_t end;
Thomas Wouters477c8d52006-05-27 19:21:47 +000012099 Py_ssize_t result;
Guido van Rossumd57fd912000-03-10 22:53:23 +000012100
Jesus Ceaac451502011-04-20 17:09:23 +020012101 if (!stringlib_parse_args_finds_unicode("rfind", args, &substring,
12102 &start, &end))
Benjamin Peterson14339b62009-01-31 16:36:08 +000012103 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +000012104
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012105 if (PyUnicode_READY(self) == -1)
12106 return NULL;
12107 if (PyUnicode_READY(substring) == -1)
12108 return NULL;
12109
Victor Stinner7931d9a2011-11-04 00:22:48 +010012110 result = any_find_slice(-1, self, substring, start, end);
Guido van Rossumd57fd912000-03-10 22:53:23 +000012111
12112 Py_DECREF(substring);
Thomas Wouters477c8d52006-05-27 19:21:47 +000012113
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012114 if (result == -2)
12115 return NULL;
12116
Christian Heimes217cfd12007-12-02 14:31:20 +000012117 return PyLong_FromSsize_t(result);
Guido van Rossumd57fd912000-03-10 22:53:23 +000012118}
12119
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000012120PyDoc_STRVAR(rindex__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000012121 "S.rindex(sub[, start[, end]]) -> int\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000012122\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000012123Like S.rfind() but raise ValueError when the substring is not found.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000012124
12125static PyObject *
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012126unicode_rindex(PyObject *self, PyObject *args)
Guido van Rossumd57fd912000-03-10 22:53:23 +000012127{
Victor Stinner9db1a8b2011-10-23 20:04:37 +020012128 PyObject *substring;
Christian Heimes9cd17752007-11-18 19:35:23 +000012129 Py_ssize_t start;
12130 Py_ssize_t end;
Thomas Wouters477c8d52006-05-27 19:21:47 +000012131 Py_ssize_t result;
Guido van Rossumd57fd912000-03-10 22:53:23 +000012132
Jesus Ceaac451502011-04-20 17:09:23 +020012133 if (!stringlib_parse_args_finds_unicode("rindex", args, &substring,
12134 &start, &end))
Benjamin Peterson14339b62009-01-31 16:36:08 +000012135 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +000012136
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012137 if (PyUnicode_READY(self) == -1)
12138 return NULL;
12139 if (PyUnicode_READY(substring) == -1)
12140 return NULL;
12141
Victor Stinner7931d9a2011-11-04 00:22:48 +010012142 result = any_find_slice(-1, self, substring, start, end);
Guido van Rossumd57fd912000-03-10 22:53:23 +000012143
12144 Py_DECREF(substring);
Thomas Wouters477c8d52006-05-27 19:21:47 +000012145
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012146 if (result == -2)
12147 return NULL;
12148
Guido van Rossumd57fd912000-03-10 22:53:23 +000012149 if (result < 0) {
12150 PyErr_SetString(PyExc_ValueError, "substring not found");
12151 return NULL;
12152 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012153
Christian Heimes217cfd12007-12-02 14:31:20 +000012154 return PyLong_FromSsize_t(result);
Guido van Rossumd57fd912000-03-10 22:53:23 +000012155}
12156
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000012157PyDoc_STRVAR(rjust__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000012158 "S.rjust(width[, fillchar]) -> str\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000012159\n\
Benjamin Petersonf10a79a2008-10-11 00:49:57 +000012160Return S right-justified in a string of length width. Padding is\n\
Raymond Hettinger4f8f9762003-11-26 08:21:35 +000012161done using the specified fill character (default is a space).");
Guido van Rossumd57fd912000-03-10 22:53:23 +000012162
12163static PyObject *
Victor Stinner9310abb2011-10-05 00:59:23 +020012164unicode_rjust(PyObject *self, PyObject *args)
Guido van Rossumd57fd912000-03-10 22:53:23 +000012165{
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000012166 Py_ssize_t width;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012167 Py_UCS4 fillchar = ' ';
12168
Victor Stinnere9a29352011-10-01 02:14:59 +020012169 if (!PyArg_ParseTuple(args, "n|O&:rjust", &width, convert_uc, &fillchar))
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012170 return NULL;
Raymond Hettinger4f8f9762003-11-26 08:21:35 +000012171
Benjamin Petersonbac79492012-01-14 13:34:47 -050012172 if (PyUnicode_READY(self) == -1)
Guido van Rossumd57fd912000-03-10 22:53:23 +000012173 return NULL;
12174
Victor Stinnerc4b49542011-12-11 22:44:26 +010012175 if (PyUnicode_GET_LENGTH(self) >= width)
12176 return unicode_result_unchanged(self);
Guido van Rossumd57fd912000-03-10 22:53:23 +000012177
Victor Stinnerc4b49542011-12-11 22:44:26 +010012178 return pad(self, width - PyUnicode_GET_LENGTH(self), 0, fillchar);
Guido van Rossumd57fd912000-03-10 22:53:23 +000012179}
12180
Alexander Belopolsky40018472011-02-26 01:02:56 +000012181PyObject *
12182PyUnicode_Split(PyObject *s, PyObject *sep, Py_ssize_t maxsplit)
Guido van Rossumd57fd912000-03-10 22:53:23 +000012183{
12184 PyObject *result;
Tim Petersced69f82003-09-16 20:30:58 +000012185
Guido van Rossumd57fd912000-03-10 22:53:23 +000012186 s = PyUnicode_FromObject(s);
12187 if (s == NULL)
Benjamin Peterson14339b62009-01-31 16:36:08 +000012188 return NULL;
Benjamin Peterson29060642009-01-31 22:14:21 +000012189 if (sep != NULL) {
12190 sep = PyUnicode_FromObject(sep);
12191 if (sep == NULL) {
12192 Py_DECREF(s);
12193 return NULL;
12194 }
Guido van Rossumd57fd912000-03-10 22:53:23 +000012195 }
12196
Victor Stinner9310abb2011-10-05 00:59:23 +020012197 result = split(s, sep, maxsplit);
Guido van Rossumd57fd912000-03-10 22:53:23 +000012198
12199 Py_DECREF(s);
12200 Py_XDECREF(sep);
12201 return result;
12202}
12203
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000012204PyDoc_STRVAR(split__doc__,
Ezio Melotticda6b6d2012-02-26 09:39:55 +020012205 "S.split(sep=None, maxsplit=-1) -> list of strings\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000012206\n\
12207Return a list of the words in S, using sep as the\n\
12208delimiter string. If maxsplit is given, at most maxsplit\n\
Alexandre Vassalotti5f8ced22008-05-16 00:03:33 +000012209splits are done. If sep is not specified or is None, any\n\
Alexandre Vassalotti8ae3e052008-05-16 00:41:41 +000012210whitespace string is a separator and empty strings are\n\
12211removed from the result.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000012212
12213static PyObject*
Ezio Melotticda6b6d2012-02-26 09:39:55 +020012214unicode_split(PyObject *self, PyObject *args, PyObject *kwds)
Guido van Rossumd57fd912000-03-10 22:53:23 +000012215{
Ezio Melotticda6b6d2012-02-26 09:39:55 +020012216 static char *kwlist[] = {"sep", "maxsplit", 0};
Guido van Rossumd57fd912000-03-10 22:53:23 +000012217 PyObject *substring = Py_None;
Martin v. Löwis18e16552006-02-15 17:27:45 +000012218 Py_ssize_t maxcount = -1;
Guido van Rossumd57fd912000-03-10 22:53:23 +000012219
Ezio Melotticda6b6d2012-02-26 09:39:55 +020012220 if (!PyArg_ParseTupleAndKeywords(args, kwds, "|On:split",
12221 kwlist, &substring, &maxcount))
Guido van Rossumd57fd912000-03-10 22:53:23 +000012222 return NULL;
12223
12224 if (substring == Py_None)
Benjamin Peterson29060642009-01-31 22:14:21 +000012225 return split(self, NULL, maxcount);
Guido van Rossumd57fd912000-03-10 22:53:23 +000012226 else if (PyUnicode_Check(substring))
Victor Stinner9310abb2011-10-05 00:59:23 +020012227 return split(self, substring, maxcount);
Guido van Rossumd57fd912000-03-10 22:53:23 +000012228 else
Victor Stinner7931d9a2011-11-04 00:22:48 +010012229 return PyUnicode_Split(self, substring, maxcount);
Guido van Rossumd57fd912000-03-10 22:53:23 +000012230}
12231
Thomas Wouters477c8d52006-05-27 19:21:47 +000012232PyObject *
12233PyUnicode_Partition(PyObject *str_in, PyObject *sep_in)
12234{
12235 PyObject* str_obj;
12236 PyObject* sep_obj;
12237 PyObject* out;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012238 int kind1, kind2, kind;
12239 void *buf1 = NULL, *buf2 = NULL;
12240 Py_ssize_t len1, len2;
Thomas Wouters477c8d52006-05-27 19:21:47 +000012241
12242 str_obj = PyUnicode_FromObject(str_in);
Benjamin Peterson22a29702012-01-02 09:00:30 -060012243 if (!str_obj)
Benjamin Peterson29060642009-01-31 22:14:21 +000012244 return NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +000012245 sep_obj = PyUnicode_FromObject(sep_in);
Benjamin Peterson22a29702012-01-02 09:00:30 -060012246 if (!sep_obj) {
12247 Py_DECREF(str_obj);
12248 return NULL;
12249 }
12250 if (PyUnicode_READY(sep_obj) == -1 || PyUnicode_READY(str_obj) == -1) {
12251 Py_DECREF(sep_obj);
Thomas Wouters477c8d52006-05-27 19:21:47 +000012252 Py_DECREF(str_obj);
12253 return NULL;
12254 }
12255
Victor Stinner14f8f022011-10-05 20:58:25 +020012256 kind1 = PyUnicode_KIND(str_obj);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012257 kind2 = PyUnicode_KIND(sep_obj);
Victor Stinner14f8f022011-10-05 20:58:25 +020012258 kind = Py_MAX(kind1, kind2);
12259 buf1 = PyUnicode_DATA(str_obj);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012260 if (kind1 != kind)
Victor Stinner14f8f022011-10-05 20:58:25 +020012261 buf1 = _PyUnicode_AsKind(str_obj, kind);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012262 if (!buf1)
12263 goto onError;
12264 buf2 = PyUnicode_DATA(sep_obj);
12265 if (kind2 != kind)
12266 buf2 = _PyUnicode_AsKind(sep_obj, kind);
12267 if (!buf2)
12268 goto onError;
12269 len1 = PyUnicode_GET_LENGTH(str_obj);
12270 len2 = PyUnicode_GET_LENGTH(sep_obj);
12271
Benjamin Petersonead6b532011-12-20 17:23:42 -060012272 switch (PyUnicode_KIND(str_obj)) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012273 case PyUnicode_1BYTE_KIND:
Victor Stinnerc3cec782011-10-05 21:24:08 +020012274 if (PyUnicode_IS_ASCII(str_obj) && PyUnicode_IS_ASCII(sep_obj))
12275 out = asciilib_partition(str_obj, buf1, len1, sep_obj, buf2, len2);
12276 else
12277 out = ucs1lib_partition(str_obj, buf1, len1, sep_obj, buf2, len2);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012278 break;
12279 case PyUnicode_2BYTE_KIND:
12280 out = ucs2lib_partition(str_obj, buf1, len1, sep_obj, buf2, len2);
12281 break;
12282 case PyUnicode_4BYTE_KIND:
12283 out = ucs4lib_partition(str_obj, buf1, len1, sep_obj, buf2, len2);
12284 break;
12285 default:
12286 assert(0);
12287 out = 0;
12288 }
Thomas Wouters477c8d52006-05-27 19:21:47 +000012289
12290 Py_DECREF(sep_obj);
12291 Py_DECREF(str_obj);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012292 if (kind1 != kind)
12293 PyMem_Free(buf1);
12294 if (kind2 != kind)
12295 PyMem_Free(buf2);
Thomas Wouters477c8d52006-05-27 19:21:47 +000012296
12297 return out;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012298 onError:
12299 Py_DECREF(sep_obj);
12300 Py_DECREF(str_obj);
12301 if (kind1 != kind && buf1)
12302 PyMem_Free(buf1);
12303 if (kind2 != kind && buf2)
12304 PyMem_Free(buf2);
12305 return NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +000012306}
12307
12308
12309PyObject *
12310PyUnicode_RPartition(PyObject *str_in, PyObject *sep_in)
12311{
12312 PyObject* str_obj;
12313 PyObject* sep_obj;
12314 PyObject* out;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012315 int kind1, kind2, kind;
12316 void *buf1 = NULL, *buf2 = NULL;
12317 Py_ssize_t len1, len2;
Thomas Wouters477c8d52006-05-27 19:21:47 +000012318
12319 str_obj = PyUnicode_FromObject(str_in);
12320 if (!str_obj)
Benjamin Peterson29060642009-01-31 22:14:21 +000012321 return NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +000012322 sep_obj = PyUnicode_FromObject(sep_in);
12323 if (!sep_obj) {
12324 Py_DECREF(str_obj);
12325 return NULL;
12326 }
12327
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012328 kind1 = PyUnicode_KIND(str_in);
12329 kind2 = PyUnicode_KIND(sep_obj);
Georg Brandl4cb0de22011-09-28 21:49:49 +020012330 kind = Py_MAX(kind1, kind2);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012331 buf1 = PyUnicode_DATA(str_in);
12332 if (kind1 != kind)
12333 buf1 = _PyUnicode_AsKind(str_in, kind);
12334 if (!buf1)
12335 goto onError;
12336 buf2 = PyUnicode_DATA(sep_obj);
12337 if (kind2 != kind)
12338 buf2 = _PyUnicode_AsKind(sep_obj, kind);
12339 if (!buf2)
12340 goto onError;
12341 len1 = PyUnicode_GET_LENGTH(str_obj);
12342 len2 = PyUnicode_GET_LENGTH(sep_obj);
12343
Benjamin Petersonead6b532011-12-20 17:23:42 -060012344 switch (PyUnicode_KIND(str_in)) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012345 case PyUnicode_1BYTE_KIND:
Victor Stinnerc3cec782011-10-05 21:24:08 +020012346 if (PyUnicode_IS_ASCII(str_obj) && PyUnicode_IS_ASCII(sep_obj))
12347 out = asciilib_rpartition(str_obj, buf1, len1, sep_obj, buf2, len2);
12348 else
12349 out = ucs1lib_rpartition(str_obj, buf1, len1, sep_obj, buf2, len2);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012350 break;
12351 case PyUnicode_2BYTE_KIND:
12352 out = ucs2lib_rpartition(str_obj, buf1, len1, sep_obj, buf2, len2);
12353 break;
12354 case PyUnicode_4BYTE_KIND:
12355 out = ucs4lib_rpartition(str_obj, buf1, len1, sep_obj, buf2, len2);
12356 break;
12357 default:
12358 assert(0);
12359 out = 0;
12360 }
Thomas Wouters477c8d52006-05-27 19:21:47 +000012361
12362 Py_DECREF(sep_obj);
12363 Py_DECREF(str_obj);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012364 if (kind1 != kind)
12365 PyMem_Free(buf1);
12366 if (kind2 != kind)
12367 PyMem_Free(buf2);
Thomas Wouters477c8d52006-05-27 19:21:47 +000012368
12369 return out;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012370 onError:
12371 Py_DECREF(sep_obj);
12372 Py_DECREF(str_obj);
12373 if (kind1 != kind && buf1)
12374 PyMem_Free(buf1);
12375 if (kind2 != kind && buf2)
12376 PyMem_Free(buf2);
12377 return NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +000012378}
12379
12380PyDoc_STRVAR(partition__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000012381 "S.partition(sep) -> (head, sep, tail)\n\
Thomas Wouters477c8d52006-05-27 19:21:47 +000012382\n\
Georg Brandl17cb8a82008-05-30 08:20:09 +000012383Search for the separator sep in S, and return the part before it,\n\
Thomas Wouters477c8d52006-05-27 19:21:47 +000012384the separator itself, and the part after it. If the separator is not\n\
Benjamin Petersonf10a79a2008-10-11 00:49:57 +000012385found, return S and two empty strings.");
Thomas Wouters477c8d52006-05-27 19:21:47 +000012386
12387static PyObject*
Victor Stinner9310abb2011-10-05 00:59:23 +020012388unicode_partition(PyObject *self, PyObject *separator)
Thomas Wouters477c8d52006-05-27 19:21:47 +000012389{
Victor Stinner9310abb2011-10-05 00:59:23 +020012390 return PyUnicode_Partition(self, separator);
Thomas Wouters477c8d52006-05-27 19:21:47 +000012391}
12392
12393PyDoc_STRVAR(rpartition__doc__,
Ezio Melotti5b2b2422010-01-25 11:58:28 +000012394 "S.rpartition(sep) -> (head, sep, tail)\n\
Thomas Wouters477c8d52006-05-27 19:21:47 +000012395\n\
Georg Brandl17cb8a82008-05-30 08:20:09 +000012396Search for the separator sep in S, starting at the end of S, and return\n\
Thomas Wouters477c8d52006-05-27 19:21:47 +000012397the part before it, the separator itself, and the part after it. If the\n\
Benjamin Petersonf10a79a2008-10-11 00:49:57 +000012398separator is not found, return two empty strings and S.");
Thomas Wouters477c8d52006-05-27 19:21:47 +000012399
12400static PyObject*
Victor Stinner9310abb2011-10-05 00:59:23 +020012401unicode_rpartition(PyObject *self, PyObject *separator)
Thomas Wouters477c8d52006-05-27 19:21:47 +000012402{
Victor Stinner9310abb2011-10-05 00:59:23 +020012403 return PyUnicode_RPartition(self, separator);
Thomas Wouters477c8d52006-05-27 19:21:47 +000012404}
12405
Alexander Belopolsky40018472011-02-26 01:02:56 +000012406PyObject *
12407PyUnicode_RSplit(PyObject *s, PyObject *sep, Py_ssize_t maxsplit)
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +000012408{
12409 PyObject *result;
Benjamin Peterson14339b62009-01-31 16:36:08 +000012410
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +000012411 s = PyUnicode_FromObject(s);
12412 if (s == NULL)
Benjamin Peterson14339b62009-01-31 16:36:08 +000012413 return NULL;
Benjamin Peterson29060642009-01-31 22:14:21 +000012414 if (sep != NULL) {
12415 sep = PyUnicode_FromObject(sep);
12416 if (sep == NULL) {
12417 Py_DECREF(s);
12418 return NULL;
12419 }
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +000012420 }
12421
Victor Stinner9310abb2011-10-05 00:59:23 +020012422 result = rsplit(s, sep, maxsplit);
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +000012423
12424 Py_DECREF(s);
12425 Py_XDECREF(sep);
12426 return result;
12427}
12428
12429PyDoc_STRVAR(rsplit__doc__,
Ezio Melotticda6b6d2012-02-26 09:39:55 +020012430 "S.rsplit(sep=None, maxsplit=-1) -> list of strings\n\
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +000012431\n\
12432Return a list of the words in S, using sep as the\n\
12433delimiter string, starting at the end of the string and\n\
12434working to the front. If maxsplit is given, at most maxsplit\n\
12435splits are done. If sep is not specified, any whitespace string\n\
12436is a separator.");
12437
12438static PyObject*
Ezio Melotticda6b6d2012-02-26 09:39:55 +020012439unicode_rsplit(PyObject *self, PyObject *args, PyObject *kwds)
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +000012440{
Ezio Melotticda6b6d2012-02-26 09:39:55 +020012441 static char *kwlist[] = {"sep", "maxsplit", 0};
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +000012442 PyObject *substring = Py_None;
Martin v. Löwis18e16552006-02-15 17:27:45 +000012443 Py_ssize_t maxcount = -1;
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +000012444
Ezio Melotticda6b6d2012-02-26 09:39:55 +020012445 if (!PyArg_ParseTupleAndKeywords(args, kwds, "|On:rsplit",
12446 kwlist, &substring, &maxcount))
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +000012447 return NULL;
12448
12449 if (substring == Py_None)
Benjamin Peterson29060642009-01-31 22:14:21 +000012450 return rsplit(self, NULL, maxcount);
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +000012451 else if (PyUnicode_Check(substring))
Victor Stinner9310abb2011-10-05 00:59:23 +020012452 return rsplit(self, substring, maxcount);
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +000012453 else
Victor Stinner9310abb2011-10-05 00:59:23 +020012454 return PyUnicode_RSplit(self, substring, maxcount);
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +000012455}
12456
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000012457PyDoc_STRVAR(splitlines__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000012458 "S.splitlines([keepends]) -> list of strings\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000012459\n\
12460Return a list of the lines in S, breaking at line boundaries.\n\
Guido van Rossum86662912000-04-11 15:38:46 +000012461Line breaks are not included in the resulting list unless keepends\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000012462is given and true.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000012463
12464static PyObject*
Victor Stinner9db1a8b2011-10-23 20:04:37 +020012465unicode_splitlines(PyObject *self, PyObject *args, PyObject *kwds)
Guido van Rossumd57fd912000-03-10 22:53:23 +000012466{
Mark Dickinson0d5f6ad2011-09-24 09:14:39 +010012467 static char *kwlist[] = {"keepends", 0};
Guido van Rossum86662912000-04-11 15:38:46 +000012468 int keepends = 0;
Guido van Rossumd57fd912000-03-10 22:53:23 +000012469
Mark Dickinson0d5f6ad2011-09-24 09:14:39 +010012470 if (!PyArg_ParseTupleAndKeywords(args, kwds, "|i:splitlines",
12471 kwlist, &keepends))
Guido van Rossumd57fd912000-03-10 22:53:23 +000012472 return NULL;
12473
Victor Stinner9db1a8b2011-10-23 20:04:37 +020012474 return PyUnicode_Splitlines(self, keepends);
Guido van Rossumd57fd912000-03-10 22:53:23 +000012475}
12476
12477static
Guido van Rossumf15a29f2007-05-04 00:41:39 +000012478PyObject *unicode_str(PyObject *self)
Guido van Rossumd57fd912000-03-10 22:53:23 +000012479{
Victor Stinnerc4b49542011-12-11 22:44:26 +010012480 return unicode_result_unchanged(self);
Guido van Rossumd57fd912000-03-10 22:53:23 +000012481}
12482
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000012483PyDoc_STRVAR(swapcase__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000012484 "S.swapcase() -> str\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000012485\n\
12486Return a copy of S with uppercase characters converted to lowercase\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000012487and vice versa.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000012488
12489static PyObject*
Victor Stinner9310abb2011-10-05 00:59:23 +020012490unicode_swapcase(PyObject *self)
Guido van Rossumd57fd912000-03-10 22:53:23 +000012491{
Benjamin Petersoneea48462012-01-16 14:28:50 -050012492 if (PyUnicode_READY(self) == -1)
12493 return NULL;
Victor Stinnerb0800dc2012-02-25 00:47:08 +010012494 return case_operation(self, do_swapcase);
Guido van Rossumd57fd912000-03-10 22:53:23 +000012495}
12496
Georg Brandlceee0772007-11-27 23:48:05 +000012497PyDoc_STRVAR(maketrans__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000012498 "str.maketrans(x[, y[, z]]) -> dict (static method)\n\
Georg Brandlceee0772007-11-27 23:48:05 +000012499\n\
12500Return a translation table usable for str.translate().\n\
12501If there is only one argument, it must be a dictionary mapping Unicode\n\
12502ordinals (integers) or characters to Unicode ordinals, strings or None.\n\
Benjamin Peterson142957c2008-07-04 19:55:29 +000012503Character keys will be then converted to ordinals.\n\
Georg Brandlceee0772007-11-27 23:48:05 +000012504If there are two arguments, they must be strings of equal length, and\n\
12505in the resulting dictionary, each character in x will be mapped to the\n\
12506character at the same position in y. If there is a third argument, it\n\
12507must be a string, whose characters will be mapped to None in the result.");
12508
12509static PyObject*
Victor Stinner9db1a8b2011-10-23 20:04:37 +020012510unicode_maketrans(PyObject *null, PyObject *args)
Georg Brandlceee0772007-11-27 23:48:05 +000012511{
12512 PyObject *x, *y = NULL, *z = NULL;
12513 PyObject *new = NULL, *key, *value;
12514 Py_ssize_t i = 0;
12515 int res;
Benjamin Peterson14339b62009-01-31 16:36:08 +000012516
Georg Brandlceee0772007-11-27 23:48:05 +000012517 if (!PyArg_ParseTuple(args, "O|UU:maketrans", &x, &y, &z))
12518 return NULL;
12519 new = PyDict_New();
12520 if (!new)
12521 return NULL;
12522 if (y != NULL) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012523 int x_kind, y_kind, z_kind;
12524 void *x_data, *y_data, *z_data;
12525
Georg Brandlceee0772007-11-27 23:48:05 +000012526 /* x must be a string too, of equal length */
Georg Brandlceee0772007-11-27 23:48:05 +000012527 if (!PyUnicode_Check(x)) {
12528 PyErr_SetString(PyExc_TypeError, "first maketrans argument must "
12529 "be a string if there is a second argument");
12530 goto err;
12531 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012532 if (PyUnicode_GET_LENGTH(x) != PyUnicode_GET_LENGTH(y)) {
Georg Brandlceee0772007-11-27 23:48:05 +000012533 PyErr_SetString(PyExc_ValueError, "the first two maketrans "
12534 "arguments must have equal length");
12535 goto err;
12536 }
12537 /* create entries for translating chars in x to those in y */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012538 x_kind = PyUnicode_KIND(x);
12539 y_kind = PyUnicode_KIND(y);
12540 x_data = PyUnicode_DATA(x);
12541 y_data = PyUnicode_DATA(y);
12542 for (i = 0; i < PyUnicode_GET_LENGTH(x); i++) {
12543 key = PyLong_FromLong(PyUnicode_READ(x_kind, x_data, i));
Benjamin Peterson53aa1d72011-12-20 13:29:45 -060012544 if (!key)
Georg Brandlceee0772007-11-27 23:48:05 +000012545 goto err;
Benjamin Peterson822c7902011-12-20 13:32:50 -060012546 value = PyLong_FromLong(PyUnicode_READ(y_kind, y_data, i));
Benjamin Peterson53aa1d72011-12-20 13:29:45 -060012547 if (!value) {
12548 Py_DECREF(key);
12549 goto err;
12550 }
Georg Brandlceee0772007-11-27 23:48:05 +000012551 res = PyDict_SetItem(new, key, value);
12552 Py_DECREF(key);
12553 Py_DECREF(value);
12554 if (res < 0)
12555 goto err;
12556 }
12557 /* create entries for deleting chars in z */
12558 if (z != NULL) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012559 z_kind = PyUnicode_KIND(z);
12560 z_data = PyUnicode_DATA(z);
Victor Stinnerc4f281e2011-10-11 22:11:42 +020012561 for (i = 0; i < PyUnicode_GET_LENGTH(z); i++) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012562 key = PyLong_FromLong(PyUnicode_READ(z_kind, z_data, i));
Georg Brandlceee0772007-11-27 23:48:05 +000012563 if (!key)
12564 goto err;
12565 res = PyDict_SetItem(new, key, Py_None);
12566 Py_DECREF(key);
12567 if (res < 0)
12568 goto err;
12569 }
12570 }
12571 } else {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012572 int kind;
12573 void *data;
12574
Georg Brandlceee0772007-11-27 23:48:05 +000012575 /* x must be a dict */
Raymond Hettinger3ad05762009-05-29 22:11:22 +000012576 if (!PyDict_CheckExact(x)) {
Georg Brandlceee0772007-11-27 23:48:05 +000012577 PyErr_SetString(PyExc_TypeError, "if you give only one argument "
12578 "to maketrans it must be a dict");
12579 goto err;
12580 }
12581 /* copy entries into the new dict, converting string keys to int keys */
12582 while (PyDict_Next(x, &i, &key, &value)) {
12583 if (PyUnicode_Check(key)) {
12584 /* convert string keys to integer keys */
12585 PyObject *newkey;
Victor Stinnerc4f281e2011-10-11 22:11:42 +020012586 if (PyUnicode_GET_LENGTH(key) != 1) {
Georg Brandlceee0772007-11-27 23:48:05 +000012587 PyErr_SetString(PyExc_ValueError, "string keys in translate "
12588 "table must be of length 1");
12589 goto err;
12590 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012591 kind = PyUnicode_KIND(key);
12592 data = PyUnicode_DATA(key);
12593 newkey = PyLong_FromLong(PyUnicode_READ(kind, data, 0));
Georg Brandlceee0772007-11-27 23:48:05 +000012594 if (!newkey)
12595 goto err;
12596 res = PyDict_SetItem(new, newkey, value);
12597 Py_DECREF(newkey);
12598 if (res < 0)
12599 goto err;
Christian Heimes217cfd12007-12-02 14:31:20 +000012600 } else if (PyLong_Check(key)) {
Georg Brandlceee0772007-11-27 23:48:05 +000012601 /* just keep integer keys */
12602 if (PyDict_SetItem(new, key, value) < 0)
12603 goto err;
12604 } else {
12605 PyErr_SetString(PyExc_TypeError, "keys in translate table must "
12606 "be strings or integers");
12607 goto err;
12608 }
12609 }
12610 }
12611 return new;
12612 err:
12613 Py_DECREF(new);
12614 return NULL;
12615}
12616
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000012617PyDoc_STRVAR(translate__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000012618 "S.translate(table) -> str\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000012619\n\
12620Return a copy of the string S, where all characters have been mapped\n\
12621through the given translation table, which must be a mapping of\n\
Benjamin Peterson142957c2008-07-04 19:55:29 +000012622Unicode ordinals to Unicode ordinals, strings, or None.\n\
Walter Dörwald5c1ee172002-09-04 20:31:32 +000012623Unmapped characters are left untouched. Characters mapped to None\n\
12624are deleted.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000012625
12626static PyObject*
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012627unicode_translate(PyObject *self, PyObject *table)
Guido van Rossumd57fd912000-03-10 22:53:23 +000012628{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012629 return _PyUnicode_TranslateCharmap(self, table, "ignore");
Guido van Rossumd57fd912000-03-10 22:53:23 +000012630}
12631
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000012632PyDoc_STRVAR(upper__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000012633 "S.upper() -> str\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000012634\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000012635Return a copy of S converted to uppercase.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000012636
12637static PyObject*
Victor Stinner9310abb2011-10-05 00:59:23 +020012638unicode_upper(PyObject *self)
Guido van Rossumd57fd912000-03-10 22:53:23 +000012639{
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -050012640 if (PyUnicode_READY(self) == -1)
12641 return NULL;
12642 if (PyUnicode_IS_ASCII(self))
12643 return ascii_upper_or_lower(self, 0);
Victor Stinnerb0800dc2012-02-25 00:47:08 +010012644 return case_operation(self, do_upper);
Guido van Rossumd57fd912000-03-10 22:53:23 +000012645}
12646
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000012647PyDoc_STRVAR(zfill__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000012648 "S.zfill(width) -> str\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000012649\n\
Benjamin Peterson9aa42992008-09-10 21:57:34 +000012650Pad a numeric string S with zeros on the left, to fill a field\n\
12651of the specified width. The string S is never truncated.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000012652
12653static PyObject *
Victor Stinner9310abb2011-10-05 00:59:23 +020012654unicode_zfill(PyObject *self, PyObject *args)
Guido van Rossumd57fd912000-03-10 22:53:23 +000012655{
Martin v. Löwis18e16552006-02-15 17:27:45 +000012656 Py_ssize_t fill;
Victor Stinner9310abb2011-10-05 00:59:23 +020012657 PyObject *u;
Martin v. Löwis18e16552006-02-15 17:27:45 +000012658 Py_ssize_t width;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012659 int kind;
12660 void *data;
12661 Py_UCS4 chr;
12662
Martin v. Löwis18e16552006-02-15 17:27:45 +000012663 if (!PyArg_ParseTuple(args, "n:zfill", &width))
Guido van Rossumd57fd912000-03-10 22:53:23 +000012664 return NULL;
12665
Benjamin Petersonbac79492012-01-14 13:34:47 -050012666 if (PyUnicode_READY(self) == -1)
Victor Stinnerc4b49542011-12-11 22:44:26 +010012667 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +000012668
Victor Stinnerc4b49542011-12-11 22:44:26 +010012669 if (PyUnicode_GET_LENGTH(self) >= width)
12670 return unicode_result_unchanged(self);
12671
12672 fill = width - PyUnicode_GET_LENGTH(self);
Guido van Rossumd57fd912000-03-10 22:53:23 +000012673
12674 u = pad(self, fill, 0, '0');
12675
Walter Dörwald068325e2002-04-15 13:36:47 +000012676 if (u == NULL)
12677 return NULL;
12678
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012679 kind = PyUnicode_KIND(u);
12680 data = PyUnicode_DATA(u);
12681 chr = PyUnicode_READ(kind, data, fill);
12682
12683 if (chr == '+' || chr == '-') {
Guido van Rossumd57fd912000-03-10 22:53:23 +000012684 /* move sign to beginning of string */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012685 PyUnicode_WRITE(kind, data, 0, chr);
12686 PyUnicode_WRITE(kind, data, fill, '0');
Guido van Rossumd57fd912000-03-10 22:53:23 +000012687 }
12688
Victor Stinnerbb10a1f2011-10-05 01:34:17 +020012689 assert(_PyUnicode_CheckConsistency(u, 1));
Victor Stinner7931d9a2011-11-04 00:22:48 +010012690 return u;
Guido van Rossumd57fd912000-03-10 22:53:23 +000012691}
Guido van Rossumd57fd912000-03-10 22:53:23 +000012692
12693#if 0
Alexander Belopolsky942af5a2010-12-04 03:38:46 +000012694static PyObject *
12695unicode__decimal2ascii(PyObject *self)
12696{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012697 return PyUnicode_TransformDecimalAndSpaceToASCII(self);
Alexander Belopolsky942af5a2010-12-04 03:38:46 +000012698}
Guido van Rossumd57fd912000-03-10 22:53:23 +000012699#endif
12700
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000012701PyDoc_STRVAR(startswith__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000012702 "S.startswith(prefix[, start[, end]]) -> bool\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000012703\n\
Guido van Rossuma7132182003-04-09 19:32:45 +000012704Return True if S starts with the specified prefix, False otherwise.\n\
12705With optional start, test S beginning at that position.\n\
Thomas Wouters0e3f5912006-08-11 14:57:12 +000012706With optional end, stop comparing S at that position.\n\
12707prefix can also be a tuple of strings to try.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000012708
12709static PyObject *
Victor Stinner9db1a8b2011-10-23 20:04:37 +020012710unicode_startswith(PyObject *self,
Benjamin Peterson29060642009-01-31 22:14:21 +000012711 PyObject *args)
Guido van Rossumd57fd912000-03-10 22:53:23 +000012712{
Thomas Wouters0e3f5912006-08-11 14:57:12 +000012713 PyObject *subobj;
Victor Stinner9db1a8b2011-10-23 20:04:37 +020012714 PyObject *substring;
Martin v. Löwis18e16552006-02-15 17:27:45 +000012715 Py_ssize_t start = 0;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000012716 Py_ssize_t end = PY_SSIZE_T_MAX;
Thomas Wouters0e3f5912006-08-11 14:57:12 +000012717 int result;
Guido van Rossumd57fd912000-03-10 22:53:23 +000012718
Jesus Ceaac451502011-04-20 17:09:23 +020012719 if (!stringlib_parse_args_finds("startswith", args, &subobj, &start, &end))
Benjamin Peterson29060642009-01-31 22:14:21 +000012720 return NULL;
Thomas Wouters0e3f5912006-08-11 14:57:12 +000012721 if (PyTuple_Check(subobj)) {
12722 Py_ssize_t i;
12723 for (i = 0; i < PyTuple_GET_SIZE(subobj); i++) {
Victor Stinner9db1a8b2011-10-23 20:04:37 +020012724 substring = PyUnicode_FromObject(PyTuple_GET_ITEM(subobj, i));
Thomas Wouters0e3f5912006-08-11 14:57:12 +000012725 if (substring == NULL)
12726 return NULL;
12727 result = tailmatch(self, substring, start, end, -1);
12728 Py_DECREF(substring);
12729 if (result) {
12730 Py_RETURN_TRUE;
12731 }
12732 }
12733 /* nothing matched */
12734 Py_RETURN_FALSE;
12735 }
Victor Stinner9db1a8b2011-10-23 20:04:37 +020012736 substring = PyUnicode_FromObject(subobj);
Ezio Melottiba42fd52011-04-26 06:09:45 +030012737 if (substring == NULL) {
12738 if (PyErr_ExceptionMatches(PyExc_TypeError))
12739 PyErr_Format(PyExc_TypeError, "startswith first arg must be str or "
12740 "a tuple of str, not %s", Py_TYPE(subobj)->tp_name);
Benjamin Peterson29060642009-01-31 22:14:21 +000012741 return NULL;
Ezio Melottiba42fd52011-04-26 06:09:45 +030012742 }
Thomas Wouters0e3f5912006-08-11 14:57:12 +000012743 result = tailmatch(self, substring, start, end, -1);
Guido van Rossumd57fd912000-03-10 22:53:23 +000012744 Py_DECREF(substring);
Thomas Wouters0e3f5912006-08-11 14:57:12 +000012745 return PyBool_FromLong(result);
Guido van Rossumd57fd912000-03-10 22:53:23 +000012746}
12747
12748
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000012749PyDoc_STRVAR(endswith__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000012750 "S.endswith(suffix[, start[, end]]) -> bool\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000012751\n\
Guido van Rossuma7132182003-04-09 19:32:45 +000012752Return True if S ends with the specified suffix, False otherwise.\n\
12753With optional start, test S beginning at that position.\n\
Thomas Wouters0e3f5912006-08-11 14:57:12 +000012754With optional end, stop comparing S at that position.\n\
12755suffix can also be a tuple of strings to try.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000012756
12757static PyObject *
Victor Stinner9db1a8b2011-10-23 20:04:37 +020012758unicode_endswith(PyObject *self,
Benjamin Peterson29060642009-01-31 22:14:21 +000012759 PyObject *args)
Guido van Rossumd57fd912000-03-10 22:53:23 +000012760{
Thomas Wouters0e3f5912006-08-11 14:57:12 +000012761 PyObject *subobj;
Victor Stinner9db1a8b2011-10-23 20:04:37 +020012762 PyObject *substring;
Martin v. Löwis18e16552006-02-15 17:27:45 +000012763 Py_ssize_t start = 0;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000012764 Py_ssize_t end = PY_SSIZE_T_MAX;
Thomas Wouters0e3f5912006-08-11 14:57:12 +000012765 int result;
Guido van Rossumd57fd912000-03-10 22:53:23 +000012766
Jesus Ceaac451502011-04-20 17:09:23 +020012767 if (!stringlib_parse_args_finds("endswith", args, &subobj, &start, &end))
Benjamin Peterson29060642009-01-31 22:14:21 +000012768 return NULL;
Thomas Wouters0e3f5912006-08-11 14:57:12 +000012769 if (PyTuple_Check(subobj)) {
12770 Py_ssize_t i;
12771 for (i = 0; i < PyTuple_GET_SIZE(subobj); i++) {
Victor Stinner9db1a8b2011-10-23 20:04:37 +020012772 substring = PyUnicode_FromObject(
Benjamin Peterson29060642009-01-31 22:14:21 +000012773 PyTuple_GET_ITEM(subobj, i));
Thomas Wouters0e3f5912006-08-11 14:57:12 +000012774 if (substring == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +000012775 return NULL;
Thomas Wouters0e3f5912006-08-11 14:57:12 +000012776 result = tailmatch(self, substring, start, end, +1);
12777 Py_DECREF(substring);
12778 if (result) {
12779 Py_RETURN_TRUE;
12780 }
12781 }
12782 Py_RETURN_FALSE;
12783 }
Victor Stinner9db1a8b2011-10-23 20:04:37 +020012784 substring = PyUnicode_FromObject(subobj);
Ezio Melottiba42fd52011-04-26 06:09:45 +030012785 if (substring == NULL) {
12786 if (PyErr_ExceptionMatches(PyExc_TypeError))
12787 PyErr_Format(PyExc_TypeError, "endswith first arg must be str or "
12788 "a tuple of str, not %s", Py_TYPE(subobj)->tp_name);
Benjamin Peterson29060642009-01-31 22:14:21 +000012789 return NULL;
Ezio Melottiba42fd52011-04-26 06:09:45 +030012790 }
Thomas Wouters0e3f5912006-08-11 14:57:12 +000012791 result = tailmatch(self, substring, start, end, +1);
Guido van Rossumd57fd912000-03-10 22:53:23 +000012792 Py_DECREF(substring);
Thomas Wouters0e3f5912006-08-11 14:57:12 +000012793 return PyBool_FromLong(result);
Guido van Rossumd57fd912000-03-10 22:53:23 +000012794}
12795
Victor Stinner202fdca2012-05-07 12:47:02 +020012796Py_LOCAL_INLINE(void)
Victor Stinner3b1a74a2012-05-09 22:25:00 +020012797_PyUnicodeWriter_Update(_PyUnicodeWriter *writer)
Victor Stinner202fdca2012-05-07 12:47:02 +020012798{
Victor Stinnerd3f08822012-05-29 12:57:52 +020012799 writer->size = PyUnicode_GET_LENGTH(writer->buffer);
Victor Stinner202fdca2012-05-07 12:47:02 +020012800 writer->maxchar = PyUnicode_MAX_CHAR_VALUE(writer->buffer);
12801 writer->data = PyUnicode_DATA(writer->buffer);
12802 writer->kind = PyUnicode_KIND(writer->buffer);
12803}
12804
Victor Stinnerd3f08822012-05-29 12:57:52 +020012805void
12806_PyUnicodeWriter_Init(_PyUnicodeWriter *writer, Py_ssize_t min_length)
Victor Stinner202fdca2012-05-07 12:47:02 +020012807{
Victor Stinnerd3f08822012-05-29 12:57:52 +020012808 memset(writer, 0, sizeof(*writer));
12809#ifdef Py_DEBUG
12810 writer->kind = 5; /* invalid kind */
12811#endif
12812 writer->min_length = Py_MAX(min_length, 100);
Victor Stinnerd7b7c742012-06-04 22:52:12 +020012813 writer->overallocate = (min_length > 0);
Victor Stinner202fdca2012-05-07 12:47:02 +020012814}
12815
Victor Stinnerd3f08822012-05-29 12:57:52 +020012816int
12817_PyUnicodeWriter_PrepareInternal(_PyUnicodeWriter *writer,
12818 Py_ssize_t length, Py_UCS4 maxchar)
Victor Stinner202fdca2012-05-07 12:47:02 +020012819{
12820 Py_ssize_t newlen;
12821 PyObject *newbuffer;
12822
Victor Stinnerd3f08822012-05-29 12:57:52 +020012823 assert(length > 0);
12824
Victor Stinner202fdca2012-05-07 12:47:02 +020012825 if (length > PY_SSIZE_T_MAX - writer->pos) {
12826 PyErr_NoMemory();
12827 return -1;
12828 }
12829 newlen = writer->pos + length;
12830
Victor Stinnerd3f08822012-05-29 12:57:52 +020012831 if (writer->buffer == NULL) {
Victor Stinnerd7b7c742012-06-04 22:52:12 +020012832 if (writer->overallocate) {
Victor Stinnerd3f08822012-05-29 12:57:52 +020012833 /* overallocate 25% to limit the number of resize */
12834 if (newlen <= (PY_SSIZE_T_MAX - newlen / 4))
12835 newlen += newlen / 4;
12836 if (newlen < writer->min_length)
12837 newlen = writer->min_length;
12838 }
12839 writer->buffer = PyUnicode_New(newlen, maxchar);
12840 if (writer->buffer == NULL)
12841 return -1;
12842 _PyUnicodeWriter_Update(writer);
12843 return 0;
12844 }
Victor Stinner202fdca2012-05-07 12:47:02 +020012845
Victor Stinnerd3f08822012-05-29 12:57:52 +020012846 if (newlen > writer->size) {
Victor Stinnerd7b7c742012-06-04 22:52:12 +020012847 if (writer->overallocate) {
Victor Stinnerd3f08822012-05-29 12:57:52 +020012848 /* overallocate 25% to limit the number of resize */
12849 if (newlen <= (PY_SSIZE_T_MAX - newlen / 4))
12850 newlen += newlen / 4;
12851 if (newlen < writer->min_length)
12852 newlen = writer->min_length;
12853 }
12854
Victor Stinnerd7b7c742012-06-04 22:52:12 +020012855 if (maxchar > writer->maxchar || writer->readonly) {
Victor Stinner202fdca2012-05-07 12:47:02 +020012856 /* resize + widen */
12857 newbuffer = PyUnicode_New(newlen, maxchar);
12858 if (newbuffer == NULL)
12859 return -1;
Victor Stinnerd3f08822012-05-29 12:57:52 +020012860 _PyUnicode_FastCopyCharacters(newbuffer, 0,
12861 writer->buffer, 0, writer->pos);
Victor Stinner202fdca2012-05-07 12:47:02 +020012862 Py_DECREF(writer->buffer);
Victor Stinnerd7b7c742012-06-04 22:52:12 +020012863 writer->readonly = 0;
Victor Stinner202fdca2012-05-07 12:47:02 +020012864 }
12865 else {
12866 newbuffer = resize_compact(writer->buffer, newlen);
12867 if (newbuffer == NULL)
12868 return -1;
12869 }
12870 writer->buffer = newbuffer;
Victor Stinner3b1a74a2012-05-09 22:25:00 +020012871 _PyUnicodeWriter_Update(writer);
Victor Stinner202fdca2012-05-07 12:47:02 +020012872 }
12873 else if (maxchar > writer->maxchar) {
Victor Stinnerd7b7c742012-06-04 22:52:12 +020012874 assert(!writer->readonly);
Victor Stinnerd3f08822012-05-29 12:57:52 +020012875 newbuffer = PyUnicode_New(writer->size, maxchar);
12876 if (newbuffer == NULL)
Victor Stinner202fdca2012-05-07 12:47:02 +020012877 return -1;
Victor Stinnerd3f08822012-05-29 12:57:52 +020012878 _PyUnicode_FastCopyCharacters(newbuffer, 0,
12879 writer->buffer, 0, writer->pos);
12880 Py_DECREF(writer->buffer);
12881 writer->buffer = newbuffer;
Victor Stinner3b1a74a2012-05-09 22:25:00 +020012882 _PyUnicodeWriter_Update(writer);
Victor Stinner202fdca2012-05-07 12:47:02 +020012883 }
12884 return 0;
12885}
12886
Victor Stinnerd3f08822012-05-29 12:57:52 +020012887int
12888_PyUnicodeWriter_WriteStr(_PyUnicodeWriter *writer, PyObject *str)
12889{
12890 Py_UCS4 maxchar;
12891 Py_ssize_t len;
12892
12893 if (PyUnicode_READY(str) == -1)
12894 return -1;
12895 len = PyUnicode_GET_LENGTH(str);
12896 if (len == 0)
12897 return 0;
12898 maxchar = PyUnicode_MAX_CHAR_VALUE(str);
12899 if (maxchar > writer->maxchar || len > writer->size - writer->pos) {
Victor Stinnerd7b7c742012-06-04 22:52:12 +020012900 if (writer->buffer == NULL && !writer->overallocate) {
Victor Stinnerd3f08822012-05-29 12:57:52 +020012901 Py_INCREF(str);
12902 writer->buffer = str;
12903 _PyUnicodeWriter_Update(writer);
Victor Stinnerd7b7c742012-06-04 22:52:12 +020012904 writer->readonly = 1;
Victor Stinnerd3f08822012-05-29 12:57:52 +020012905 writer->size = 0;
12906 writer->pos += len;
12907 return 0;
12908 }
12909 if (_PyUnicodeWriter_PrepareInternal(writer, len, maxchar) == -1)
12910 return -1;
12911 }
12912 _PyUnicode_FastCopyCharacters(writer->buffer, writer->pos,
12913 str, 0, len);
12914 writer->pos += len;
12915 return 0;
12916}
12917
12918PyObject *
Victor Stinner3b1a74a2012-05-09 22:25:00 +020012919_PyUnicodeWriter_Finish(_PyUnicodeWriter *writer)
Victor Stinner202fdca2012-05-07 12:47:02 +020012920{
Victor Stinnerd3f08822012-05-29 12:57:52 +020012921 if (writer->pos == 0) {
12922 Py_XDECREF(writer->buffer);
Serhiy Storchaka678db842013-01-26 12:16:36 +020012923 _Py_RETURN_UNICODE_EMPTY();
Victor Stinnerd3f08822012-05-29 12:57:52 +020012924 }
Victor Stinnerd7b7c742012-06-04 22:52:12 +020012925 if (writer->readonly) {
Victor Stinnerd3f08822012-05-29 12:57:52 +020012926 assert(PyUnicode_GET_LENGTH(writer->buffer) == writer->pos);
12927 return writer->buffer;
12928 }
12929 if (PyUnicode_GET_LENGTH(writer->buffer) != writer->pos) {
12930 PyObject *newbuffer;
12931 newbuffer = resize_compact(writer->buffer, writer->pos);
12932 if (newbuffer == NULL) {
12933 Py_DECREF(writer->buffer);
12934 return NULL;
12935 }
12936 writer->buffer = newbuffer;
Victor Stinner202fdca2012-05-07 12:47:02 +020012937 }
Victor Stinnerf59c28c2012-05-09 03:24:14 +020012938 assert(_PyUnicode_CheckConsistency(writer->buffer, 1));
Victor Stinner2cb16aa2013-03-06 19:28:37 +010012939 return unicode_result_ready(writer->buffer);
Victor Stinner202fdca2012-05-07 12:47:02 +020012940}
12941
Victor Stinnerd3f08822012-05-29 12:57:52 +020012942void
Victor Stinner3b1a74a2012-05-09 22:25:00 +020012943_PyUnicodeWriter_Dealloc(_PyUnicodeWriter *writer)
Victor Stinner202fdca2012-05-07 12:47:02 +020012944{
12945 Py_CLEAR(writer->buffer);
12946}
12947
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012948#include "stringlib/unicode_format.h"
Eric Smith8c663262007-08-25 02:26:07 +000012949
12950PyDoc_STRVAR(format__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000012951 "S.format(*args, **kwargs) -> str\n\
Eric Smith8c663262007-08-25 02:26:07 +000012952\n\
Eric Smith51d2fd92010-11-06 19:27:37 +000012953Return a formatted version of S, using substitutions from args and kwargs.\n\
12954The substitutions are identified by braces ('{' and '}').");
Eric Smith8c663262007-08-25 02:26:07 +000012955
Eric Smith27bbca62010-11-04 17:06:58 +000012956PyDoc_STRVAR(format_map__doc__,
12957 "S.format_map(mapping) -> str\n\
12958\n\
Eric Smith51d2fd92010-11-06 19:27:37 +000012959Return a formatted version of S, using substitutions from mapping.\n\
12960The substitutions are identified by braces ('{' and '}').");
Eric Smith27bbca62010-11-04 17:06:58 +000012961
Eric Smith4a7d76d2008-05-30 18:10:19 +000012962static PyObject *
12963unicode__format__(PyObject* self, PyObject* args)
12964{
Victor Stinnerd3f08822012-05-29 12:57:52 +020012965 PyObject *format_spec;
12966 _PyUnicodeWriter writer;
12967 int ret;
Eric Smith4a7d76d2008-05-30 18:10:19 +000012968
12969 if (!PyArg_ParseTuple(args, "U:__format__", &format_spec))
12970 return NULL;
12971
Victor Stinnerd3f08822012-05-29 12:57:52 +020012972 if (PyUnicode_READY(self) == -1)
12973 return NULL;
12974 _PyUnicodeWriter_Init(&writer, 0);
12975 ret = _PyUnicode_FormatAdvancedWriter(&writer,
12976 self, format_spec, 0,
12977 PyUnicode_GET_LENGTH(format_spec));
12978 if (ret == -1) {
12979 _PyUnicodeWriter_Dealloc(&writer);
12980 return NULL;
12981 }
12982 return _PyUnicodeWriter_Finish(&writer);
Eric Smith4a7d76d2008-05-30 18:10:19 +000012983}
12984
Eric Smith8c663262007-08-25 02:26:07 +000012985PyDoc_STRVAR(p_format__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000012986 "S.__format__(format_spec) -> str\n\
Eric Smith8c663262007-08-25 02:26:07 +000012987\n\
Eric Smith51d2fd92010-11-06 19:27:37 +000012988Return a formatted version of S as described by format_spec.");
Eric Smith8c663262007-08-25 02:26:07 +000012989
12990static PyObject *
Victor Stinner9db1a8b2011-10-23 20:04:37 +020012991unicode__sizeof__(PyObject *v)
Georg Brandlc28e1fa2008-06-10 19:20:26 +000012992{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012993 Py_ssize_t size;
12994
12995 /* If it's a compact object, account for base structure +
12996 character data. */
12997 if (PyUnicode_IS_COMPACT_ASCII(v))
12998 size = sizeof(PyASCIIObject) + PyUnicode_GET_LENGTH(v) + 1;
12999 else if (PyUnicode_IS_COMPACT(v))
13000 size = sizeof(PyCompactUnicodeObject) +
Martin v. Löwisc47adb02011-10-07 20:55:35 +020013001 (PyUnicode_GET_LENGTH(v) + 1) * PyUnicode_KIND(v);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013002 else {
13003 /* If it is a two-block object, account for base object, and
13004 for character block if present. */
13005 size = sizeof(PyUnicodeObject);
Victor Stinnerc3c74152011-10-02 20:39:55 +020013006 if (_PyUnicode_DATA_ANY(v))
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013007 size += (PyUnicode_GET_LENGTH(v) + 1) *
Martin v. Löwisc47adb02011-10-07 20:55:35 +020013008 PyUnicode_KIND(v);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013009 }
13010 /* If the wstr pointer is present, account for it unless it is shared
Victor Stinnera3be6132011-10-03 02:16:37 +020013011 with the data pointer. Check if the data is not shared. */
Victor Stinner03490912011-10-03 23:45:12 +020013012 if (_PyUnicode_HAS_WSTR_MEMORY(v))
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013013 size += (PyUnicode_WSTR_LENGTH(v) + 1) * sizeof(wchar_t);
Victor Stinner829c0ad2011-10-03 01:08:02 +020013014 if (_PyUnicode_HAS_UTF8_MEMORY(v))
Victor Stinnere90fe6a2011-10-01 16:48:13 +020013015 size += PyUnicode_UTF8_LENGTH(v) + 1;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013016
13017 return PyLong_FromSsize_t(size);
Georg Brandlc28e1fa2008-06-10 19:20:26 +000013018}
13019
13020PyDoc_STRVAR(sizeof__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000013021 "S.__sizeof__() -> size of S in memory, in bytes");
Georg Brandlc28e1fa2008-06-10 19:20:26 +000013022
13023static PyObject *
Victor Stinner034f6cf2011-09-30 02:26:44 +020013024unicode_getnewargs(PyObject *v)
Guido van Rossum5d9113d2003-01-29 17:58:45 +000013025{
Victor Stinnerbf6e5602011-12-12 01:53:47 +010013026 PyObject *copy = _PyUnicode_Copy(v);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013027 if (!copy)
13028 return NULL;
13029 return Py_BuildValue("(N)", copy);
Guido van Rossum5d9113d2003-01-29 17:58:45 +000013030}
13031
Guido van Rossumd57fd912000-03-10 22:53:23 +000013032static PyMethodDef unicode_methods[] = {
Benjamin Peterson28a4dce2010-12-12 01:33:04 +000013033 {"encode", (PyCFunction) unicode_encode, METH_VARARGS | METH_KEYWORDS, encode__doc__},
Martin v. Löwise3eb1f22001-08-16 13:15:00 +000013034 {"replace", (PyCFunction) unicode_replace, METH_VARARGS, replace__doc__},
Ezio Melotticda6b6d2012-02-26 09:39:55 +020013035 {"split", (PyCFunction) unicode_split, METH_VARARGS | METH_KEYWORDS, split__doc__},
13036 {"rsplit", (PyCFunction) unicode_rsplit, METH_VARARGS | METH_KEYWORDS, rsplit__doc__},
Martin v. Löwise3eb1f22001-08-16 13:15:00 +000013037 {"join", (PyCFunction) unicode_join, METH_O, join__doc__},
13038 {"capitalize", (PyCFunction) unicode_capitalize, METH_NOARGS, capitalize__doc__},
Benjamin Petersond5890c82012-01-14 13:23:30 -050013039 {"casefold", (PyCFunction) unicode_casefold, METH_NOARGS, casefold__doc__},
Martin v. Löwise3eb1f22001-08-16 13:15:00 +000013040 {"title", (PyCFunction) unicode_title, METH_NOARGS, title__doc__},
13041 {"center", (PyCFunction) unicode_center, METH_VARARGS, center__doc__},
13042 {"count", (PyCFunction) unicode_count, METH_VARARGS, count__doc__},
13043 {"expandtabs", (PyCFunction) unicode_expandtabs, METH_VARARGS, expandtabs__doc__},
13044 {"find", (PyCFunction) unicode_find, METH_VARARGS, find__doc__},
Thomas Wouters477c8d52006-05-27 19:21:47 +000013045 {"partition", (PyCFunction) unicode_partition, METH_O, partition__doc__},
Martin v. Löwise3eb1f22001-08-16 13:15:00 +000013046 {"index", (PyCFunction) unicode_index, METH_VARARGS, index__doc__},
13047 {"ljust", (PyCFunction) unicode_ljust, METH_VARARGS, ljust__doc__},
13048 {"lower", (PyCFunction) unicode_lower, METH_NOARGS, lower__doc__},
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000013049 {"lstrip", (PyCFunction) unicode_lstrip, METH_VARARGS, lstrip__doc__},
Martin v. Löwise3eb1f22001-08-16 13:15:00 +000013050 {"rfind", (PyCFunction) unicode_rfind, METH_VARARGS, rfind__doc__},
13051 {"rindex", (PyCFunction) unicode_rindex, METH_VARARGS, rindex__doc__},
13052 {"rjust", (PyCFunction) unicode_rjust, METH_VARARGS, rjust__doc__},
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000013053 {"rstrip", (PyCFunction) unicode_rstrip, METH_VARARGS, rstrip__doc__},
Thomas Wouters477c8d52006-05-27 19:21:47 +000013054 {"rpartition", (PyCFunction) unicode_rpartition, METH_O, rpartition__doc__},
Mark Dickinson0d5f6ad2011-09-24 09:14:39 +010013055 {"splitlines", (PyCFunction) unicode_splitlines, METH_VARARGS | METH_KEYWORDS, splitlines__doc__},
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000013056 {"strip", (PyCFunction) unicode_strip, METH_VARARGS, strip__doc__},
Martin v. Löwise3eb1f22001-08-16 13:15:00 +000013057 {"swapcase", (PyCFunction) unicode_swapcase, METH_NOARGS, swapcase__doc__},
13058 {"translate", (PyCFunction) unicode_translate, METH_O, translate__doc__},
13059 {"upper", (PyCFunction) unicode_upper, METH_NOARGS, upper__doc__},
13060 {"startswith", (PyCFunction) unicode_startswith, METH_VARARGS, startswith__doc__},
13061 {"endswith", (PyCFunction) unicode_endswith, METH_VARARGS, endswith__doc__},
13062 {"islower", (PyCFunction) unicode_islower, METH_NOARGS, islower__doc__},
13063 {"isupper", (PyCFunction) unicode_isupper, METH_NOARGS, isupper__doc__},
13064 {"istitle", (PyCFunction) unicode_istitle, METH_NOARGS, istitle__doc__},
13065 {"isspace", (PyCFunction) unicode_isspace, METH_NOARGS, isspace__doc__},
13066 {"isdecimal", (PyCFunction) unicode_isdecimal, METH_NOARGS, isdecimal__doc__},
13067 {"isdigit", (PyCFunction) unicode_isdigit, METH_NOARGS, isdigit__doc__},
13068 {"isnumeric", (PyCFunction) unicode_isnumeric, METH_NOARGS, isnumeric__doc__},
13069 {"isalpha", (PyCFunction) unicode_isalpha, METH_NOARGS, isalpha__doc__},
13070 {"isalnum", (PyCFunction) unicode_isalnum, METH_NOARGS, isalnum__doc__},
Martin v. Löwis47383402007-08-15 07:32:56 +000013071 {"isidentifier", (PyCFunction) unicode_isidentifier, METH_NOARGS, isidentifier__doc__},
Georg Brandl559e5d72008-06-11 18:37:52 +000013072 {"isprintable", (PyCFunction) unicode_isprintable, METH_NOARGS, isprintable__doc__},
Martin v. Löwise3eb1f22001-08-16 13:15:00 +000013073 {"zfill", (PyCFunction) unicode_zfill, METH_VARARGS, zfill__doc__},
Eric Smith9cd1e092007-08-31 18:39:38 +000013074 {"format", (PyCFunction) do_string_format, METH_VARARGS | METH_KEYWORDS, format__doc__},
Eric Smith27bbca62010-11-04 17:06:58 +000013075 {"format_map", (PyCFunction) do_string_format_map, METH_O, format_map__doc__},
Eric Smith4a7d76d2008-05-30 18:10:19 +000013076 {"__format__", (PyCFunction) unicode__format__, METH_VARARGS, p_format__doc__},
Georg Brandlceee0772007-11-27 23:48:05 +000013077 {"maketrans", (PyCFunction) unicode_maketrans,
13078 METH_VARARGS | METH_STATIC, maketrans__doc__},
Georg Brandlc28e1fa2008-06-10 19:20:26 +000013079 {"__sizeof__", (PyCFunction) unicode__sizeof__, METH_NOARGS, sizeof__doc__},
Walter Dörwald068325e2002-04-15 13:36:47 +000013080#if 0
Alexander Belopolsky942af5a2010-12-04 03:38:46 +000013081 /* These methods are just used for debugging the implementation. */
Alexander Belopolsky942af5a2010-12-04 03:38:46 +000013082 {"_decimal2ascii", (PyCFunction) unicode__decimal2ascii, METH_NOARGS},
Guido van Rossumd57fd912000-03-10 22:53:23 +000013083#endif
13084
Benjamin Peterson14339b62009-01-31 16:36:08 +000013085 {"__getnewargs__", (PyCFunction)unicode_getnewargs, METH_NOARGS},
Guido van Rossumd57fd912000-03-10 22:53:23 +000013086 {NULL, NULL}
13087};
13088
Neil Schemenauerce30bc92002-11-18 16:10:18 +000013089static PyObject *
13090unicode_mod(PyObject *v, PyObject *w)
13091{
Brian Curtindfc80e32011-08-10 20:28:54 -050013092 if (!PyUnicode_Check(v))
13093 Py_RETURN_NOTIMPLEMENTED;
Benjamin Peterson29060642009-01-31 22:14:21 +000013094 return PyUnicode_Format(v, w);
Neil Schemenauerce30bc92002-11-18 16:10:18 +000013095}
13096
13097static PyNumberMethods unicode_as_number = {
Benjamin Peterson14339b62009-01-31 16:36:08 +000013098 0, /*nb_add*/
13099 0, /*nb_subtract*/
13100 0, /*nb_multiply*/
13101 unicode_mod, /*nb_remainder*/
Neil Schemenauerce30bc92002-11-18 16:10:18 +000013102};
13103
Guido van Rossumd57fd912000-03-10 22:53:23 +000013104static PySequenceMethods unicode_as_sequence = {
Benjamin Peterson14339b62009-01-31 16:36:08 +000013105 (lenfunc) unicode_length, /* sq_length */
13106 PyUnicode_Concat, /* sq_concat */
13107 (ssizeargfunc) unicode_repeat, /* sq_repeat */
13108 (ssizeargfunc) unicode_getitem, /* sq_item */
13109 0, /* sq_slice */
13110 0, /* sq_ass_item */
13111 0, /* sq_ass_slice */
13112 PyUnicode_Contains, /* sq_contains */
Guido van Rossumd57fd912000-03-10 22:53:23 +000013113};
13114
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +000013115static PyObject*
Victor Stinner9db1a8b2011-10-23 20:04:37 +020013116unicode_subscript(PyObject* self, PyObject* item)
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +000013117{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013118 if (PyUnicode_READY(self) == -1)
13119 return NULL;
13120
Thomas Wouters00ee7ba2006-08-21 19:07:27 +000013121 if (PyIndex_Check(item)) {
13122 Py_ssize_t i = PyNumber_AsSsize_t(item, PyExc_IndexError);
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +000013123 if (i == -1 && PyErr_Occurred())
13124 return NULL;
13125 if (i < 0)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013126 i += PyUnicode_GET_LENGTH(self);
Victor Stinner9db1a8b2011-10-23 20:04:37 +020013127 return unicode_getitem(self, i);
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +000013128 } else if (PySlice_Check(item)) {
Martin v. Löwis18e16552006-02-15 17:27:45 +000013129 Py_ssize_t start, stop, step, slicelength, cur, i;
Antoine Pitrou7aec4012011-10-04 19:08:01 +020013130 PyObject *result;
13131 void *src_data, *dest_data;
Antoine Pitrou875f29b2011-10-04 20:00:49 +020013132 int src_kind, dest_kind;
Victor Stinnerc80d6d22011-10-05 14:13:28 +020013133 Py_UCS4 ch, max_char, kind_limit;
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +000013134
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013135 if (PySlice_GetIndicesEx(item, PyUnicode_GET_LENGTH(self),
Benjamin Peterson29060642009-01-31 22:14:21 +000013136 &start, &stop, &step, &slicelength) < 0) {
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +000013137 return NULL;
13138 }
13139
13140 if (slicelength <= 0) {
Serhiy Storchaka678db842013-01-26 12:16:36 +020013141 _Py_RETURN_UNICODE_EMPTY();
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013142 } else if (start == 0 && step == 1 &&
Victor Stinnerc4b49542011-12-11 22:44:26 +010013143 slicelength == PyUnicode_GET_LENGTH(self)) {
13144 return unicode_result_unchanged(self);
Thomas Woutersed03b412007-08-28 21:37:11 +000013145 } else if (step == 1) {
Victor Stinner9db1a8b2011-10-23 20:04:37 +020013146 return PyUnicode_Substring(self,
Victor Stinner12bab6d2011-10-01 01:53:49 +020013147 start, start + slicelength);
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +000013148 }
Antoine Pitrou875f29b2011-10-04 20:00:49 +020013149 /* General case */
Antoine Pitrou875f29b2011-10-04 20:00:49 +020013150 src_kind = PyUnicode_KIND(self);
13151 src_data = PyUnicode_DATA(self);
Victor Stinner55c99112011-10-13 01:17:06 +020013152 if (!PyUnicode_IS_ASCII(self)) {
13153 kind_limit = kind_maxchar_limit(src_kind);
13154 max_char = 0;
13155 for (cur = start, i = 0; i < slicelength; cur += step, i++) {
13156 ch = PyUnicode_READ(src_kind, src_data, cur);
13157 if (ch > max_char) {
13158 max_char = ch;
13159 if (max_char >= kind_limit)
13160 break;
13161 }
Victor Stinnerc80d6d22011-10-05 14:13:28 +020013162 }
Antoine Pitrou875f29b2011-10-04 20:00:49 +020013163 }
Victor Stinner55c99112011-10-13 01:17:06 +020013164 else
13165 max_char = 127;
Antoine Pitrou875f29b2011-10-04 20:00:49 +020013166 result = PyUnicode_New(slicelength, max_char);
Antoine Pitrou7aec4012011-10-04 19:08:01 +020013167 if (result == NULL)
13168 return NULL;
Antoine Pitrou875f29b2011-10-04 20:00:49 +020013169 dest_kind = PyUnicode_KIND(result);
Antoine Pitrou7aec4012011-10-04 19:08:01 +020013170 dest_data = PyUnicode_DATA(result);
13171
13172 for (cur = start, i = 0; i < slicelength; cur += step, i++) {
Antoine Pitrou875f29b2011-10-04 20:00:49 +020013173 Py_UCS4 ch = PyUnicode_READ(src_kind, src_data, cur);
13174 PyUnicode_WRITE(dest_kind, dest_data, i, ch);
Antoine Pitrou7aec4012011-10-04 19:08:01 +020013175 }
Victor Stinnerbb10a1f2011-10-05 01:34:17 +020013176 assert(_PyUnicode_CheckConsistency(result, 1));
Antoine Pitrou7aec4012011-10-04 19:08:01 +020013177 return result;
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +000013178 } else {
13179 PyErr_SetString(PyExc_TypeError, "string indices must be integers");
13180 return NULL;
13181 }
13182}
13183
13184static PyMappingMethods unicode_as_mapping = {
Benjamin Peterson14339b62009-01-31 16:36:08 +000013185 (lenfunc)unicode_length, /* mp_length */
13186 (binaryfunc)unicode_subscript, /* mp_subscript */
13187 (objobjargproc)0, /* mp_ass_subscript */
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +000013188};
13189
Guido van Rossumd57fd912000-03-10 22:53:23 +000013190
Guido van Rossumd57fd912000-03-10 22:53:23 +000013191/* Helpers for PyUnicode_Format() */
13192
13193static PyObject *
Martin v. Löwis18e16552006-02-15 17:27:45 +000013194getnextarg(PyObject *args, Py_ssize_t arglen, Py_ssize_t *p_argidx)
Guido van Rossumd57fd912000-03-10 22:53:23 +000013195{
Martin v. Löwis18e16552006-02-15 17:27:45 +000013196 Py_ssize_t argidx = *p_argidx;
Guido van Rossumd57fd912000-03-10 22:53:23 +000013197 if (argidx < arglen) {
Benjamin Peterson29060642009-01-31 22:14:21 +000013198 (*p_argidx)++;
13199 if (arglen < 0)
13200 return args;
13201 else
13202 return PyTuple_GetItem(args, argidx);
Guido van Rossumd57fd912000-03-10 22:53:23 +000013203 }
13204 PyErr_SetString(PyExc_TypeError,
Benjamin Peterson29060642009-01-31 22:14:21 +000013205 "not enough arguments for format string");
Guido van Rossumd57fd912000-03-10 22:53:23 +000013206 return NULL;
13207}
13208
Mark Dickinsonf489caf2009-05-01 11:42:00 +000013209/* Returns a new reference to a PyUnicode object, or NULL on failure. */
Guido van Rossumd57fd912000-03-10 22:53:23 +000013210
Victor Stinnerd3f08822012-05-29 12:57:52 +020013211static int
13212formatfloat(PyObject *v, int flags, int prec, int type,
13213 PyObject **p_output, _PyUnicodeWriter *writer)
Guido van Rossumd57fd912000-03-10 22:53:23 +000013214{
Mark Dickinsonf489caf2009-05-01 11:42:00 +000013215 char *p;
Guido van Rossumd57fd912000-03-10 22:53:23 +000013216 double x;
Victor Stinnerd3f08822012-05-29 12:57:52 +020013217 Py_ssize_t len;
Tim Petersced69f82003-09-16 20:30:58 +000013218
Guido van Rossumd57fd912000-03-10 22:53:23 +000013219 x = PyFloat_AsDouble(v);
13220 if (x == -1.0 && PyErr_Occurred())
Victor Stinnerd3f08822012-05-29 12:57:52 +020013221 return -1;
Mark Dickinsonf489caf2009-05-01 11:42:00 +000013222
Guido van Rossumd57fd912000-03-10 22:53:23 +000013223 if (prec < 0)
Benjamin Peterson29060642009-01-31 22:14:21 +000013224 prec = 6;
Eric Smith0923d1d2009-04-16 20:16:10 +000013225
Eric Smith0923d1d2009-04-16 20:16:10 +000013226 p = PyOS_double_to_string(x, type, prec,
13227 (flags & F_ALT) ? Py_DTSF_ALT : 0, NULL);
Mark Dickinsonf489caf2009-05-01 11:42:00 +000013228 if (p == NULL)
Victor Stinnerd3f08822012-05-29 12:57:52 +020013229 return -1;
13230 len = strlen(p);
13231 if (writer) {
Christian Heimesf4f99392012-09-10 11:48:41 +020013232 if (_PyUnicodeWriter_Prepare(writer, len, 127) == -1) {
13233 PyMem_Free(p);
Victor Stinnerd3f08822012-05-29 12:57:52 +020013234 return -1;
Christian Heimesf4f99392012-09-10 11:48:41 +020013235 }
Victor Stinner184252a2012-06-16 02:57:41 +020013236 unicode_write_cstr(writer->buffer, writer->pos, p, len);
Victor Stinnerd3f08822012-05-29 12:57:52 +020013237 writer->pos += len;
13238 }
13239 else
13240 *p_output = _PyUnicode_FromASCII(p, len);
Eric Smith0923d1d2009-04-16 20:16:10 +000013241 PyMem_Free(p);
Victor Stinnerd3f08822012-05-29 12:57:52 +020013242 return 0;
Guido van Rossumd57fd912000-03-10 22:53:23 +000013243}
13244
Victor Stinnerd0880d52012-04-27 23:40:13 +020013245/* formatlong() emulates the format codes d, u, o, x and X, and
13246 * the F_ALT flag, for Python's long (unbounded) ints. It's not used for
13247 * Python's regular ints.
13248 * Return value: a new PyUnicodeObject*, or NULL if error.
13249 * The output string is of the form
13250 * "-"? ("0x" | "0X")? digit+
13251 * "0x"/"0X" are present only for x and X conversions, with F_ALT
13252 * set in flags. The case of hex digits will be correct,
13253 * There will be at least prec digits, zero-filled on the left if
13254 * necessary to get that many.
13255 * val object to be converted
13256 * flags bitmask of format flags; only F_ALT is looked at
13257 * prec minimum number of digits; 0-fill on left if needed
13258 * type a character in [duoxX]; u acts the same as d
13259 *
13260 * CAUTION: o, x and X conversions on regular ints can never
13261 * produce a '-' sign, but can for Python's unbounded ints.
13262 */
Tim Peters38fd5b62000-09-21 05:43:11 +000013263static PyObject*
13264formatlong(PyObject *val, int flags, int prec, int type)
13265{
Victor Stinnerd0880d52012-04-27 23:40:13 +020013266 PyObject *result = NULL;
Benjamin Peterson14339b62009-01-31 16:36:08 +000013267 char *buf;
Victor Stinnerd0880d52012-04-27 23:40:13 +020013268 Py_ssize_t i;
13269 int sign; /* 1 if '-', else 0 */
13270 int len; /* number of characters */
13271 Py_ssize_t llen;
13272 int numdigits; /* len == numnondigits + numdigits */
13273 int numnondigits = 0;
Tim Peters38fd5b62000-09-21 05:43:11 +000013274
Victor Stinnerd0880d52012-04-27 23:40:13 +020013275 /* Avoid exceeding SSIZE_T_MAX */
13276 if (prec > INT_MAX-3) {
13277 PyErr_SetString(PyExc_OverflowError,
13278 "precision too large");
Benjamin Peterson14339b62009-01-31 16:36:08 +000013279 return NULL;
Victor Stinnerd0880d52012-04-27 23:40:13 +020013280 }
13281
13282 assert(PyLong_Check(val));
13283
13284 switch (type) {
13285 case 'd':
13286 case 'u':
13287 /* Special-case boolean: we want 0/1 */
Victor Stinnerb11d91d2012-04-28 00:25:34 +020013288 if (PyBool_Check(val))
13289 result = PyNumber_ToBase(val, 10);
13290 else
13291 result = Py_TYPE(val)->tp_str(val);
Victor Stinnerd0880d52012-04-27 23:40:13 +020013292 break;
13293 case 'o':
13294 numnondigits = 2;
13295 result = PyNumber_ToBase(val, 8);
13296 break;
13297 case 'x':
13298 case 'X':
13299 numnondigits = 2;
13300 result = PyNumber_ToBase(val, 16);
13301 break;
13302 default:
13303 assert(!"'type' not in [duoxX]");
13304 }
13305 if (!result)
13306 return NULL;
13307
13308 assert(unicode_modifiable(result));
13309 assert(PyUnicode_IS_READY(result));
13310 assert(PyUnicode_IS_ASCII(result));
13311
13312 /* To modify the string in-place, there can only be one reference. */
13313 if (Py_REFCNT(result) != 1) {
13314 PyErr_BadInternalCall();
13315 return NULL;
13316 }
13317 buf = PyUnicode_DATA(result);
13318 llen = PyUnicode_GET_LENGTH(result);
13319 if (llen > INT_MAX) {
13320 PyErr_SetString(PyExc_ValueError,
13321 "string too large in _PyBytes_FormatLong");
13322 return NULL;
13323 }
13324 len = (int)llen;
13325 sign = buf[0] == '-';
13326 numnondigits += sign;
13327 numdigits = len - numnondigits;
13328 assert(numdigits > 0);
13329
13330 /* Get rid of base marker unless F_ALT */
13331 if (((flags & F_ALT) == 0 &&
13332 (type == 'o' || type == 'x' || type == 'X'))) {
13333 assert(buf[sign] == '0');
13334 assert(buf[sign+1] == 'x' || buf[sign+1] == 'X' ||
13335 buf[sign+1] == 'o');
13336 numnondigits -= 2;
13337 buf += 2;
13338 len -= 2;
13339 if (sign)
13340 buf[0] = '-';
13341 assert(len == numnondigits + numdigits);
13342 assert(numdigits > 0);
13343 }
13344
13345 /* Fill with leading zeroes to meet minimum width. */
13346 if (prec > numdigits) {
13347 PyObject *r1 = PyBytes_FromStringAndSize(NULL,
13348 numnondigits + prec);
13349 char *b1;
13350 if (!r1) {
13351 Py_DECREF(result);
13352 return NULL;
13353 }
13354 b1 = PyBytes_AS_STRING(r1);
13355 for (i = 0; i < numnondigits; ++i)
13356 *b1++ = *buf++;
13357 for (i = 0; i < prec - numdigits; i++)
13358 *b1++ = '0';
13359 for (i = 0; i < numdigits; i++)
13360 *b1++ = *buf++;
13361 *b1 = '\0';
13362 Py_DECREF(result);
13363 result = r1;
13364 buf = PyBytes_AS_STRING(result);
13365 len = numnondigits + prec;
13366 }
13367
13368 /* Fix up case for hex conversions. */
13369 if (type == 'X') {
13370 /* Need to convert all lower case letters to upper case.
13371 and need to convert 0x to 0X (and -0x to -0X). */
13372 for (i = 0; i < len; i++)
13373 if (buf[i] >= 'a' && buf[i] <= 'x')
13374 buf[i] -= 'a'-'A';
13375 }
13376 if (!PyUnicode_Check(result) || len != PyUnicode_GET_LENGTH(result)) {
13377 PyObject *unicode;
Victor Stinnerd3f08822012-05-29 12:57:52 +020013378 unicode = _PyUnicode_FromASCII(buf, len);
Victor Stinnerd0880d52012-04-27 23:40:13 +020013379 Py_DECREF(result);
13380 result = unicode;
13381 }
Benjamin Peterson14339b62009-01-31 16:36:08 +000013382 return result;
Tim Peters38fd5b62000-09-21 05:43:11 +000013383}
13384
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013385static Py_UCS4
13386formatchar(PyObject *v)
Guido van Rossumd57fd912000-03-10 22:53:23 +000013387{
Amaury Forgeot d'Arca4db6862008-07-04 21:26:43 +000013388 /* presume that the buffer is at least 3 characters long */
Marc-André Lemburgd4ab4a52000-06-08 17:54:00 +000013389 if (PyUnicode_Check(v)) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013390 if (PyUnicode_GET_LENGTH(v) == 1) {
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013391 return PyUnicode_READ_CHAR(v, 0);
Benjamin Peterson29060642009-01-31 22:14:21 +000013392 }
Benjamin Peterson29060642009-01-31 22:14:21 +000013393 goto onError;
13394 }
13395 else {
13396 /* Integer input truncated to a character */
13397 long x;
13398 x = PyLong_AsLong(v);
13399 if (x == -1 && PyErr_Occurred())
13400 goto onError;
13401
Victor Stinner8faf8212011-12-08 22:14:11 +010013402 if (x < 0 || x > MAX_UNICODE) {
Benjamin Peterson29060642009-01-31 22:14:21 +000013403 PyErr_SetString(PyExc_OverflowError,
13404 "%c arg not in range(0x110000)");
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013405 return (Py_UCS4) -1;
Benjamin Peterson29060642009-01-31 22:14:21 +000013406 }
13407
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013408 return (Py_UCS4) x;
Benjamin Peterson14339b62009-01-31 16:36:08 +000013409 }
Amaury Forgeot d'Arca4db6862008-07-04 21:26:43 +000013410
Benjamin Peterson29060642009-01-31 22:14:21 +000013411 onError:
Marc-André Lemburgd4ab4a52000-06-08 17:54:00 +000013412 PyErr_SetString(PyExc_TypeError,
Benjamin Peterson29060642009-01-31 22:14:21 +000013413 "%c requires int or char");
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013414 return (Py_UCS4) -1;
Guido van Rossumd57fd912000-03-10 22:53:23 +000013415}
13416
Alexander Belopolsky40018472011-02-26 01:02:56 +000013417PyObject *
13418PyUnicode_Format(PyObject *format, PyObject *args)
Guido van Rossumd57fd912000-03-10 22:53:23 +000013419{
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013420 Py_ssize_t fmtcnt, fmtpos, arglen, argidx;
Guido van Rossumd57fd912000-03-10 22:53:23 +000013421 int args_owned = 0;
Guido van Rossumd57fd912000-03-10 22:53:23 +000013422 PyObject *dict = NULL;
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013423 PyObject *temp = NULL;
13424 PyObject *second = NULL;
Victor Stinner9db1a8b2011-10-23 20:04:37 +020013425 PyObject *uformat;
Victor Stinnerf2c76aa2012-05-03 13:10:40 +020013426 void *fmt;
13427 enum PyUnicode_Kind kind, fmtkind;
Victor Stinner3b1a74a2012-05-09 22:25:00 +020013428 _PyUnicodeWriter writer;
Victor Stinneree4544c2012-05-09 22:24:08 +020013429 Py_ssize_t sublen;
13430 Py_UCS4 maxchar;
Tim Petersced69f82003-09-16 20:30:58 +000013431
Guido van Rossumd57fd912000-03-10 22:53:23 +000013432 if (format == NULL || args == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +000013433 PyErr_BadInternalCall();
13434 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +000013435 }
Victor Stinner9db1a8b2011-10-23 20:04:37 +020013436 uformat = PyUnicode_FromObject(format);
Benjamin Peterson22a29702012-01-02 09:00:30 -060013437 if (uformat == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +000013438 return NULL;
Victor Stinner19294072012-10-05 00:09:33 +020013439 if (PyUnicode_READY(uformat) == -1) {
Benjamin Peterson22a29702012-01-02 09:00:30 -060013440 Py_DECREF(uformat);
Victor Stinner19294072012-10-05 00:09:33 +020013441 return NULL;
13442 }
Victor Stinnerf2c76aa2012-05-03 13:10:40 +020013443
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013444 fmt = PyUnicode_DATA(uformat);
13445 fmtkind = PyUnicode_KIND(uformat);
13446 fmtcnt = PyUnicode_GET_LENGTH(uformat);
13447 fmtpos = 0;
Guido van Rossumd57fd912000-03-10 22:53:23 +000013448
Victor Stinnerd3f08822012-05-29 12:57:52 +020013449 _PyUnicodeWriter_Init(&writer, fmtcnt + 100);
Victor Stinnerf2c76aa2012-05-03 13:10:40 +020013450
Guido van Rossumd57fd912000-03-10 22:53:23 +000013451 if (PyTuple_Check(args)) {
Benjamin Peterson29060642009-01-31 22:14:21 +000013452 arglen = PyTuple_Size(args);
13453 argidx = 0;
Guido van Rossumd57fd912000-03-10 22:53:23 +000013454 }
13455 else {
Benjamin Peterson29060642009-01-31 22:14:21 +000013456 arglen = -1;
13457 argidx = -2;
Guido van Rossumd57fd912000-03-10 22:53:23 +000013458 }
Benjamin Peterson28a6cfa2012-08-28 17:55:35 -040013459 if (PyMapping_Check(args) && !PyTuple_Check(args) && !PyUnicode_Check(args))
Benjamin Peterson29060642009-01-31 22:14:21 +000013460 dict = args;
Guido van Rossumd57fd912000-03-10 22:53:23 +000013461
13462 while (--fmtcnt >= 0) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013463 if (PyUnicode_READ(fmtkind, fmt, fmtpos) != '%') {
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013464 Py_ssize_t nonfmtpos;
13465 nonfmtpos = fmtpos++;
13466 while (fmtcnt >= 0 &&
13467 PyUnicode_READ(fmtkind, fmt, fmtpos) != '%') {
13468 fmtpos++;
13469 fmtcnt--;
Benjamin Peterson14339b62009-01-31 16:36:08 +000013470 }
Victor Stinnerf2c76aa2012-05-03 13:10:40 +020013471 if (fmtcnt < 0)
13472 fmtpos--;
Victor Stinneree4544c2012-05-09 22:24:08 +020013473 sublen = fmtpos - nonfmtpos;
13474 maxchar = _PyUnicode_FindMaxChar(uformat,
13475 nonfmtpos, nonfmtpos + sublen);
Victor Stinner3b1a74a2012-05-09 22:25:00 +020013476 if (_PyUnicodeWriter_Prepare(&writer, sublen, maxchar) == -1)
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013477 goto onError;
Victor Stinneree4544c2012-05-09 22:24:08 +020013478
Victor Stinnerd3f08822012-05-29 12:57:52 +020013479 _PyUnicode_FastCopyCharacters(writer.buffer, writer.pos,
13480 uformat, nonfmtpos, sublen);
Victor Stinneree4544c2012-05-09 22:24:08 +020013481 writer.pos += sublen;
Benjamin Peterson14339b62009-01-31 16:36:08 +000013482 }
13483 else {
Benjamin Peterson29060642009-01-31 22:14:21 +000013484 /* Got a format specifier */
13485 int flags = 0;
13486 Py_ssize_t width = -1;
13487 int prec = -1;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013488 Py_UCS4 c = '\0';
Victor Stinnerf2c76aa2012-05-03 13:10:40 +020013489 Py_UCS4 fill;
13490 int sign;
13491 Py_UCS4 signchar;
Benjamin Peterson29060642009-01-31 22:14:21 +000013492 int isnumok;
13493 PyObject *v = NULL;
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013494 void *pbuf = NULL;
13495 Py_ssize_t pindex, len;
Victor Stinneree4544c2012-05-09 22:24:08 +020013496 Py_UCS4 bufmaxchar;
13497 Py_ssize_t buflen;
Guido van Rossumd57fd912000-03-10 22:53:23 +000013498
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013499 fmtpos++;
Victor Stinner438106b2012-05-02 00:41:57 +020013500 c = PyUnicode_READ(fmtkind, fmt, fmtpos);
13501 if (c == '(') {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013502 Py_ssize_t keystart;
Benjamin Peterson29060642009-01-31 22:14:21 +000013503 Py_ssize_t keylen;
13504 PyObject *key;
13505 int pcount = 1;
Christian Heimesa612dc02008-02-24 13:08:18 +000013506
Benjamin Peterson29060642009-01-31 22:14:21 +000013507 if (dict == NULL) {
13508 PyErr_SetString(PyExc_TypeError,
13509 "format requires a mapping");
13510 goto onError;
13511 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013512 ++fmtpos;
Benjamin Peterson29060642009-01-31 22:14:21 +000013513 --fmtcnt;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013514 keystart = fmtpos;
Benjamin Peterson29060642009-01-31 22:14:21 +000013515 /* Skip over balanced parentheses */
13516 while (pcount > 0 && --fmtcnt >= 0) {
Victor Stinnerbff7c962012-05-03 01:44:59 +020013517 c = PyUnicode_READ(fmtkind, fmt, fmtpos);
13518 if (c == ')')
Benjamin Peterson29060642009-01-31 22:14:21 +000013519 --pcount;
Victor Stinnerbff7c962012-05-03 01:44:59 +020013520 else if (c == '(')
Benjamin Peterson29060642009-01-31 22:14:21 +000013521 ++pcount;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013522 fmtpos++;
Benjamin Peterson29060642009-01-31 22:14:21 +000013523 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013524 keylen = fmtpos - keystart - 1;
Benjamin Peterson29060642009-01-31 22:14:21 +000013525 if (fmtcnt < 0 || pcount > 0) {
13526 PyErr_SetString(PyExc_ValueError,
13527 "incomplete format key");
13528 goto onError;
13529 }
Victor Stinner7931d9a2011-11-04 00:22:48 +010013530 key = PyUnicode_Substring(uformat,
Victor Stinner12bab6d2011-10-01 01:53:49 +020013531 keystart, keystart + keylen);
Benjamin Peterson29060642009-01-31 22:14:21 +000013532 if (key == NULL)
13533 goto onError;
13534 if (args_owned) {
13535 Py_DECREF(args);
13536 args_owned = 0;
13537 }
13538 args = PyObject_GetItem(dict, key);
13539 Py_DECREF(key);
13540 if (args == NULL) {
13541 goto onError;
13542 }
13543 args_owned = 1;
13544 arglen = -1;
13545 argidx = -2;
Benjamin Peterson14339b62009-01-31 16:36:08 +000013546 }
Benjamin Peterson29060642009-01-31 22:14:21 +000013547 while (--fmtcnt >= 0) {
Victor Stinner438106b2012-05-02 00:41:57 +020013548 c = PyUnicode_READ(fmtkind, fmt, fmtpos++);
13549 switch (c) {
Benjamin Peterson29060642009-01-31 22:14:21 +000013550 case '-': flags |= F_LJUST; continue;
13551 case '+': flags |= F_SIGN; continue;
13552 case ' ': flags |= F_BLANK; continue;
13553 case '#': flags |= F_ALT; continue;
13554 case '0': flags |= F_ZERO; continue;
13555 }
13556 break;
Benjamin Peterson14339b62009-01-31 16:36:08 +000013557 }
Benjamin Peterson29060642009-01-31 22:14:21 +000013558 if (c == '*') {
13559 v = getnextarg(args, arglen, &argidx);
13560 if (v == NULL)
13561 goto onError;
13562 if (!PyLong_Check(v)) {
13563 PyErr_SetString(PyExc_TypeError,
13564 "* wants int");
13565 goto onError;
13566 }
Serhiy Storchaka441d30f2013-01-19 12:26:26 +020013567 width = PyLong_AsSsize_t(v);
Benjamin Peterson29060642009-01-31 22:14:21 +000013568 if (width == -1 && PyErr_Occurred())
13569 goto onError;
13570 if (width < 0) {
13571 flags |= F_LJUST;
13572 width = -width;
13573 }
13574 if (--fmtcnt >= 0)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013575 c = PyUnicode_READ(fmtkind, fmt, fmtpos++);
Benjamin Peterson29060642009-01-31 22:14:21 +000013576 }
13577 else if (c >= '0' && c <= '9') {
13578 width = c - '0';
13579 while (--fmtcnt >= 0) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013580 c = PyUnicode_READ(fmtkind, fmt, fmtpos++);
Benjamin Peterson29060642009-01-31 22:14:21 +000013581 if (c < '0' || c > '9')
13582 break;
Martin v. Löwisb05c0732012-05-15 13:45:49 +020013583 /* Since c is unsigned, the RHS would end up as unsigned,
13584 mixing signed and unsigned comparison. Since c is between
13585 '0' and '9', casting to int is safe. */
13586 if (width > (PY_SSIZE_T_MAX - ((int)c - '0')) / 10) {
Benjamin Peterson29060642009-01-31 22:14:21 +000013587 PyErr_SetString(PyExc_ValueError,
13588 "width too big");
Benjamin Peterson14339b62009-01-31 16:36:08 +000013589 goto onError;
Benjamin Peterson29060642009-01-31 22:14:21 +000013590 }
13591 width = width*10 + (c - '0');
13592 }
13593 }
13594 if (c == '.') {
13595 prec = 0;
13596 if (--fmtcnt >= 0)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013597 c = PyUnicode_READ(fmtkind, fmt, fmtpos++);
Benjamin Peterson29060642009-01-31 22:14:21 +000013598 if (c == '*') {
13599 v = getnextarg(args, arglen, &argidx);
13600 if (v == NULL)
13601 goto onError;
13602 if (!PyLong_Check(v)) {
13603 PyErr_SetString(PyExc_TypeError,
13604 "* wants int");
13605 goto onError;
13606 }
Serhiy Storchaka441d30f2013-01-19 12:26:26 +020013607 prec = _PyLong_AsInt(v);
Benjamin Peterson29060642009-01-31 22:14:21 +000013608 if (prec == -1 && PyErr_Occurred())
13609 goto onError;
13610 if (prec < 0)
13611 prec = 0;
13612 if (--fmtcnt >= 0)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013613 c = PyUnicode_READ(fmtkind, fmt, fmtpos++);
Benjamin Peterson29060642009-01-31 22:14:21 +000013614 }
13615 else if (c >= '0' && c <= '9') {
13616 prec = c - '0';
13617 while (--fmtcnt >= 0) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013618 c = PyUnicode_READ(fmtkind, fmt, fmtpos++);
Benjamin Peterson29060642009-01-31 22:14:21 +000013619 if (c < '0' || c > '9')
13620 break;
Martin v. Löwisb05c0732012-05-15 13:45:49 +020013621 if (prec > (INT_MAX - ((int)c - '0')) / 10) {
Benjamin Peterson29060642009-01-31 22:14:21 +000013622 PyErr_SetString(PyExc_ValueError,
13623 "prec too big");
13624 goto onError;
13625 }
13626 prec = prec*10 + (c - '0');
13627 }
13628 }
13629 } /* prec */
13630 if (fmtcnt >= 0) {
13631 if (c == 'h' || c == 'l' || c == 'L') {
13632 if (--fmtcnt >= 0)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013633 c = PyUnicode_READ(fmtkind, fmt, fmtpos++);
Benjamin Peterson29060642009-01-31 22:14:21 +000013634 }
13635 }
13636 if (fmtcnt < 0) {
13637 PyErr_SetString(PyExc_ValueError,
13638 "incomplete format");
13639 goto onError;
13640 }
Victor Stinnerd3f08822012-05-29 12:57:52 +020013641 if (fmtcnt == 0)
Victor Stinnerd7b7c742012-06-04 22:52:12 +020013642 writer.overallocate = 0;
Victor Stinneraff3cc62012-04-30 05:19:21 +020013643
13644 if (c == '%') {
Victor Stinner3b1a74a2012-05-09 22:25:00 +020013645 if (_PyUnicodeWriter_Prepare(&writer, 1, '%') == -1)
Victor Stinnerf2c76aa2012-05-03 13:10:40 +020013646 goto onError;
Victor Stinneree4544c2012-05-09 22:24:08 +020013647 PyUnicode_WRITE(writer.kind, writer.data, writer.pos, '%');
13648 writer.pos += 1;
Victor Stinneraff3cc62012-04-30 05:19:21 +020013649 continue;
Benjamin Peterson29060642009-01-31 22:14:21 +000013650 }
Victor Stinneraff3cc62012-04-30 05:19:21 +020013651
Victor Stinneraff3cc62012-04-30 05:19:21 +020013652 v = getnextarg(args, arglen, &argidx);
13653 if (v == NULL)
13654 goto onError;
13655
Benjamin Peterson29060642009-01-31 22:14:21 +000013656 sign = 0;
Victor Stinnerf2c76aa2012-05-03 13:10:40 +020013657 signchar = '\0';
Benjamin Peterson29060642009-01-31 22:14:21 +000013658 fill = ' ';
13659 switch (c) {
13660
Benjamin Peterson29060642009-01-31 22:14:21 +000013661 case 's':
13662 case 'r':
13663 case 'a':
Victor Stinnerd3f08822012-05-29 12:57:52 +020013664 if (PyLong_CheckExact(v) && width == -1 && prec == -1) {
13665 /* Fast path */
13666 if (_PyLong_FormatWriter(&writer, v, 10, flags & F_ALT) == -1)
13667 goto onError;
13668 goto nextarg;
13669 }
13670
Victor Stinner808fc0a2010-03-22 12:50:40 +000013671 if (PyUnicode_CheckExact(v) && c == 's') {
Benjamin Peterson29060642009-01-31 22:14:21 +000013672 temp = v;
13673 Py_INCREF(temp);
Benjamin Peterson14339b62009-01-31 16:36:08 +000013674 }
13675 else {
Benjamin Peterson29060642009-01-31 22:14:21 +000013676 if (c == 's')
13677 temp = PyObject_Str(v);
13678 else if (c == 'r')
13679 temp = PyObject_Repr(v);
13680 else
13681 temp = PyObject_ASCII(v);
Benjamin Peterson29060642009-01-31 22:14:21 +000013682 }
Benjamin Peterson29060642009-01-31 22:14:21 +000013683 break;
13684
13685 case 'i':
13686 case 'd':
13687 case 'u':
13688 case 'o':
13689 case 'x':
13690 case 'X':
Victor Stinnerd3f08822012-05-29 12:57:52 +020013691 if (PyLong_CheckExact(v)
13692 && width == -1 && prec == -1
13693 && !(flags & (F_SIGN | F_BLANK)))
13694 {
13695 /* Fast path */
13696 switch(c)
13697 {
13698 case 'd':
13699 case 'i':
13700 case 'u':
13701 if (_PyLong_FormatWriter(&writer, v, 10, flags & F_ALT) == -1)
13702 goto onError;
13703 goto nextarg;
13704 case 'x':
13705 if (_PyLong_FormatWriter(&writer, v, 16, flags & F_ALT) == -1)
13706 goto onError;
13707 goto nextarg;
13708 case 'o':
13709 if (_PyLong_FormatWriter(&writer, v, 8, flags & F_ALT) == -1)
13710 goto onError;
13711 goto nextarg;
13712 default:
13713 break;
13714 }
13715 }
13716
Benjamin Peterson29060642009-01-31 22:14:21 +000013717 isnumok = 0;
13718 if (PyNumber_Check(v)) {
13719 PyObject *iobj=NULL;
13720
13721 if (PyLong_Check(v)) {
13722 iobj = v;
13723 Py_INCREF(iobj);
13724 }
13725 else {
13726 iobj = PyNumber_Long(v);
13727 }
13728 if (iobj!=NULL) {
13729 if (PyLong_Check(iobj)) {
13730 isnumok = 1;
Victor Stinneraff3cc62012-04-30 05:19:21 +020013731 sign = 1;
Senthil Kumaran9ebe08d2011-07-03 21:03:16 -070013732 temp = formatlong(iobj, flags, prec, (c == 'i'? 'd': c));
Benjamin Peterson29060642009-01-31 22:14:21 +000013733 Py_DECREF(iobj);
Benjamin Peterson29060642009-01-31 22:14:21 +000013734 }
13735 else {
13736 Py_DECREF(iobj);
13737 }
13738 }
13739 }
13740 if (!isnumok) {
13741 PyErr_Format(PyExc_TypeError,
13742 "%%%c format: a number is required, "
13743 "not %.200s", (char)c, Py_TYPE(v)->tp_name);
13744 goto onError;
13745 }
Victor Stinnerf2c76aa2012-05-03 13:10:40 +020013746 if (flags & F_ZERO)
Benjamin Peterson29060642009-01-31 22:14:21 +000013747 fill = '0';
13748 break;
13749
13750 case 'e':
13751 case 'E':
13752 case 'f':
13753 case 'F':
13754 case 'g':
13755 case 'G':
Victor Stinnerd3f08822012-05-29 12:57:52 +020013756 if (width == -1 && prec == -1
13757 && !(flags & (F_SIGN | F_BLANK)))
13758 {
13759 /* Fast path */
13760 if (formatfloat(v, flags, prec, c, NULL, &writer) == -1)
13761 goto onError;
13762 goto nextarg;
13763 }
13764
Benjamin Peterson29060642009-01-31 22:14:21 +000013765 sign = 1;
Victor Stinnerf2c76aa2012-05-03 13:10:40 +020013766 if (flags & F_ZERO)
Benjamin Peterson29060642009-01-31 22:14:21 +000013767 fill = '0';
Victor Stinnerd3f08822012-05-29 12:57:52 +020013768 if (formatfloat(v, flags, prec, c, &temp, NULL) == -1)
13769 temp = NULL;
Benjamin Peterson29060642009-01-31 22:14:21 +000013770 break;
13771
13772 case 'c':
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013773 {
13774 Py_UCS4 ch = formatchar(v);
13775 if (ch == (Py_UCS4) -1)
Benjamin Peterson29060642009-01-31 22:14:21 +000013776 goto onError;
Victor Stinnerd3f08822012-05-29 12:57:52 +020013777 if (width == -1 && prec == -1) {
13778 /* Fast path */
13779 if (_PyUnicodeWriter_Prepare(&writer, 1, ch) == -1)
13780 goto onError;
13781 PyUnicode_WRITE(writer.kind, writer.data, writer.pos, ch);
13782 writer.pos += 1;
13783 goto nextarg;
13784 }
Victor Stinnerb5c3ea32012-05-02 00:29:36 +020013785 temp = PyUnicode_FromOrdinal(ch);
Benjamin Peterson29060642009-01-31 22:14:21 +000013786 break;
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013787 }
Benjamin Peterson29060642009-01-31 22:14:21 +000013788
13789 default:
13790 PyErr_Format(PyExc_ValueError,
13791 "unsupported format character '%c' (0x%x) "
13792 "at index %zd",
13793 (31<=c && c<=126) ? (char)c : '?',
13794 (int)c,
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013795 fmtpos - 1);
Benjamin Peterson29060642009-01-31 22:14:21 +000013796 goto onError;
13797 }
Victor Stinneraff3cc62012-04-30 05:19:21 +020013798 if (temp == NULL)
13799 goto onError;
13800 assert (PyUnicode_Check(temp));
Victor Stinnerd3f08822012-05-29 12:57:52 +020013801
13802 if (width == -1 && prec == -1
13803 && !(flags & (F_SIGN | F_BLANK)))
13804 {
13805 /* Fast path */
13806 if (_PyUnicodeWriter_WriteStr(&writer, temp) == -1)
13807 goto onError;
13808 goto nextarg;
13809 }
13810
Victor Stinneraff3cc62012-04-30 05:19:21 +020013811 if (PyUnicode_READY(temp) == -1) {
13812 Py_CLEAR(temp);
13813 goto onError;
13814 }
13815 kind = PyUnicode_KIND(temp);
13816 pbuf = PyUnicode_DATA(temp);
13817 len = PyUnicode_GET_LENGTH(temp);
13818
13819 if (c == 's' || c == 'r' || c == 'a') {
13820 if (prec >= 0 && len > prec)
13821 len = prec;
13822 }
13823
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013824 /* pbuf is initialized here. */
13825 pindex = 0;
Benjamin Peterson29060642009-01-31 22:14:21 +000013826 if (sign) {
Victor Stinnerf2c76aa2012-05-03 13:10:40 +020013827 Py_UCS4 ch = PyUnicode_READ(kind, pbuf, pindex);
13828 if (ch == '-' || ch == '+') {
13829 signchar = ch;
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013830 len--;
13831 pindex++;
Benjamin Peterson29060642009-01-31 22:14:21 +000013832 }
13833 else if (flags & F_SIGN)
Victor Stinnerf2c76aa2012-05-03 13:10:40 +020013834 signchar = '+';
Benjamin Peterson29060642009-01-31 22:14:21 +000013835 else if (flags & F_BLANK)
Victor Stinnerf2c76aa2012-05-03 13:10:40 +020013836 signchar = ' ';
Benjamin Peterson29060642009-01-31 22:14:21 +000013837 else
13838 sign = 0;
13839 }
13840 if (width < len)
13841 width = len;
Victor Stinneree4544c2012-05-09 22:24:08 +020013842
13843 /* Compute the length and maximum character of the
13844 written characters */
13845 bufmaxchar = 127;
13846 if (!(flags & F_LJUST)) {
13847 if (sign) {
13848 if ((width-1) > len)
Benjamin Peterson7e303732013-06-10 09:19:46 -070013849 bufmaxchar = Py_MAX(bufmaxchar, fill);
Victor Stinneree4544c2012-05-09 22:24:08 +020013850 }
13851 else {
13852 if (width > len)
Benjamin Peterson7e303732013-06-10 09:19:46 -070013853 bufmaxchar = Py_MAX(bufmaxchar, fill);
Victor Stinneree4544c2012-05-09 22:24:08 +020013854 }
13855 }
13856 maxchar = _PyUnicode_FindMaxChar(temp, 0, pindex+len);
Benjamin Peterson7e303732013-06-10 09:19:46 -070013857 bufmaxchar = Py_MAX(bufmaxchar, maxchar);
Victor Stinneree4544c2012-05-09 22:24:08 +020013858
13859 buflen = width;
13860 if (sign && len == width)
13861 buflen++;
13862
Victor Stinner3b1a74a2012-05-09 22:25:00 +020013863 if (_PyUnicodeWriter_Prepare(&writer, buflen, bufmaxchar) == -1)
Victor Stinneree4544c2012-05-09 22:24:08 +020013864 goto onError;
13865
13866 /* Write characters */
Benjamin Peterson29060642009-01-31 22:14:21 +000013867 if (sign) {
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013868 if (fill != ' ') {
Victor Stinneree4544c2012-05-09 22:24:08 +020013869 PyUnicode_WRITE(writer.kind, writer.data, writer.pos, signchar);
13870 writer.pos += 1;
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013871 }
Benjamin Peterson29060642009-01-31 22:14:21 +000013872 if (width > len)
13873 width--;
13874 }
13875 if ((flags & F_ALT) && (c == 'x' || c == 'X' || c == 'o')) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013876 assert(PyUnicode_READ(kind, pbuf, pindex) == '0');
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013877 assert(PyUnicode_READ(kind, pbuf, pindex + 1) == c);
Benjamin Peterson29060642009-01-31 22:14:21 +000013878 if (fill != ' ') {
Victor Stinnerf2c76aa2012-05-03 13:10:40 +020013879 PyUnicode_WRITE(writer.kind, writer.data, writer.pos, '0');
13880 PyUnicode_WRITE(writer.kind, writer.data, writer.pos+1, c);
13881 writer.pos += 2;
13882 pindex += 2;
Benjamin Peterson29060642009-01-31 22:14:21 +000013883 }
Benjamin Peterson29060642009-01-31 22:14:21 +000013884 width -= 2;
13885 if (width < 0)
13886 width = 0;
13887 len -= 2;
13888 }
13889 if (width > len && !(flags & F_LJUST)) {
Victor Stinnerf2c76aa2012-05-03 13:10:40 +020013890 sublen = width - len;
Victor Stinnerf2c76aa2012-05-03 13:10:40 +020013891 FILL(writer.kind, writer.data, fill, writer.pos, sublen);
13892 writer.pos += sublen;
Antoine Pitrou978b9d22011-10-07 12:35:48 +020013893 width = len;
Benjamin Peterson29060642009-01-31 22:14:21 +000013894 }
13895 if (fill == ' ') {
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013896 if (sign) {
Victor Stinneree4544c2012-05-09 22:24:08 +020013897 PyUnicode_WRITE(writer.kind, writer.data, writer.pos, signchar);
13898 writer.pos += 1;
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013899 }
Benjamin Peterson29060642009-01-31 22:14:21 +000013900 if ((flags & F_ALT) && (c == 'x' || c == 'X' || c == 'o')) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013901 assert(PyUnicode_READ(kind, pbuf, pindex) == '0');
13902 assert(PyUnicode_READ(kind, pbuf, pindex+1) == c);
Victor Stinnerf2c76aa2012-05-03 13:10:40 +020013903 PyUnicode_WRITE(writer.kind, writer.data, writer.pos, '0');
13904 PyUnicode_WRITE(writer.kind, writer.data, writer.pos+1, c);
13905 writer.pos += 2;
Victor Stinnerf2c76aa2012-05-03 13:10:40 +020013906 pindex += 2;
Benjamin Peterson14339b62009-01-31 16:36:08 +000013907 }
13908 }
Victor Stinnerf2c76aa2012-05-03 13:10:40 +020013909
Victor Stinnerc9d369f2012-06-16 02:22:37 +020013910 if (len) {
13911 _PyUnicode_FastCopyCharacters(writer.buffer, writer.pos,
13912 temp, pindex, len);
13913 writer.pos += len;
13914 }
Victor Stinnerf2c76aa2012-05-03 13:10:40 +020013915 if (width > len) {
Victor Stinneree4544c2012-05-09 22:24:08 +020013916 sublen = width - len;
Victor Stinnerf2c76aa2012-05-03 13:10:40 +020013917 FILL(writer.kind, writer.data, ' ', writer.pos, sublen);
13918 writer.pos += sublen;
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013919 }
Victor Stinneree4544c2012-05-09 22:24:08 +020013920
Victor Stinnerd3f08822012-05-29 12:57:52 +020013921nextarg:
Benjamin Peterson29060642009-01-31 22:14:21 +000013922 if (dict && (argidx < arglen) && c != '%') {
13923 PyErr_SetString(PyExc_TypeError,
13924 "not all arguments converted during string formatting");
Benjamin Peterson29060642009-01-31 22:14:21 +000013925 goto onError;
13926 }
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013927 Py_CLEAR(temp);
Benjamin Peterson29060642009-01-31 22:14:21 +000013928 } /* '%' */
Guido van Rossumd57fd912000-03-10 22:53:23 +000013929 } /* until end */
13930 if (argidx < arglen && !dict) {
Benjamin Peterson29060642009-01-31 22:14:21 +000013931 PyErr_SetString(PyExc_TypeError,
13932 "not all arguments converted during string formatting");
13933 goto onError;
Guido van Rossumd57fd912000-03-10 22:53:23 +000013934 }
13935
13936 if (args_owned) {
Benjamin Peterson29060642009-01-31 22:14:21 +000013937 Py_DECREF(args);
Guido van Rossumd57fd912000-03-10 22:53:23 +000013938 }
13939 Py_DECREF(uformat);
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013940 Py_XDECREF(temp);
13941 Py_XDECREF(second);
Victor Stinner3b1a74a2012-05-09 22:25:00 +020013942 return _PyUnicodeWriter_Finish(&writer);
Guido van Rossumd57fd912000-03-10 22:53:23 +000013943
Benjamin Peterson29060642009-01-31 22:14:21 +000013944 onError:
Guido van Rossumd57fd912000-03-10 22:53:23 +000013945 Py_DECREF(uformat);
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013946 Py_XDECREF(temp);
13947 Py_XDECREF(second);
Victor Stinner3b1a74a2012-05-09 22:25:00 +020013948 _PyUnicodeWriter_Dealloc(&writer);
Guido van Rossumd57fd912000-03-10 22:53:23 +000013949 if (args_owned) {
Benjamin Peterson29060642009-01-31 22:14:21 +000013950 Py_DECREF(args);
Guido van Rossumd57fd912000-03-10 22:53:23 +000013951 }
13952 return NULL;
13953}
13954
Jeremy Hylton938ace62002-07-17 16:30:39 +000013955static PyObject *
Guido van Rossume023fe02001-08-30 03:12:59 +000013956unicode_subtype_new(PyTypeObject *type, PyObject *args, PyObject *kwds);
13957
Tim Peters6d6c1a32001-08-02 04:15:00 +000013958static PyObject *
13959unicode_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
13960{
Benjamin Peterson29060642009-01-31 22:14:21 +000013961 PyObject *x = NULL;
Benjamin Peterson14339b62009-01-31 16:36:08 +000013962 static char *kwlist[] = {"object", "encoding", "errors", 0};
13963 char *encoding = NULL;
13964 char *errors = NULL;
Tim Peters6d6c1a32001-08-02 04:15:00 +000013965
Benjamin Peterson14339b62009-01-31 16:36:08 +000013966 if (type != &PyUnicode_Type)
13967 return unicode_subtype_new(type, args, kwds);
13968 if (!PyArg_ParseTupleAndKeywords(args, kwds, "|Oss:str",
Benjamin Peterson29060642009-01-31 22:14:21 +000013969 kwlist, &x, &encoding, &errors))
Benjamin Peterson14339b62009-01-31 16:36:08 +000013970 return NULL;
13971 if (x == NULL)
Serhiy Storchaka678db842013-01-26 12:16:36 +020013972 _Py_RETURN_UNICODE_EMPTY();
Benjamin Peterson14339b62009-01-31 16:36:08 +000013973 if (encoding == NULL && errors == NULL)
13974 return PyObject_Str(x);
13975 else
Benjamin Peterson29060642009-01-31 22:14:21 +000013976 return PyUnicode_FromEncodedObject(x, encoding, errors);
Tim Peters6d6c1a32001-08-02 04:15:00 +000013977}
13978
Guido van Rossume023fe02001-08-30 03:12:59 +000013979static PyObject *
13980unicode_subtype_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
13981{
Victor Stinner9db1a8b2011-10-23 20:04:37 +020013982 PyObject *unicode, *self;
Victor Stinner07ac3eb2011-10-01 16:16:43 +020013983 Py_ssize_t length, char_size;
13984 int share_wstr, share_utf8;
13985 unsigned int kind;
13986 void *data;
Guido van Rossume023fe02001-08-30 03:12:59 +000013987
Benjamin Peterson14339b62009-01-31 16:36:08 +000013988 assert(PyType_IsSubtype(type, &PyUnicode_Type));
Victor Stinner07ac3eb2011-10-01 16:16:43 +020013989
Victor Stinner9db1a8b2011-10-23 20:04:37 +020013990 unicode = unicode_new(&PyUnicode_Type, args, kwds);
Victor Stinner07ac3eb2011-10-01 16:16:43 +020013991 if (unicode == NULL)
Benjamin Peterson14339b62009-01-31 16:36:08 +000013992 return NULL;
Victor Stinner910337b2011-10-03 03:20:16 +020013993 assert(_PyUnicode_CHECK(unicode));
Benjamin Petersonbac79492012-01-14 13:34:47 -050013994 if (PyUnicode_READY(unicode) == -1) {
Benjamin Peterson22a29702012-01-02 09:00:30 -060013995 Py_DECREF(unicode);
Victor Stinner07ac3eb2011-10-01 16:16:43 +020013996 return NULL;
Benjamin Peterson22a29702012-01-02 09:00:30 -060013997 }
Victor Stinner07ac3eb2011-10-01 16:16:43 +020013998
Victor Stinner9db1a8b2011-10-23 20:04:37 +020013999 self = type->tp_alloc(type, 0);
Victor Stinner07ac3eb2011-10-01 16:16:43 +020014000 if (self == NULL) {
14001 Py_DECREF(unicode);
Benjamin Peterson14339b62009-01-31 16:36:08 +000014002 return NULL;
14003 }
Victor Stinner07ac3eb2011-10-01 16:16:43 +020014004 kind = PyUnicode_KIND(unicode);
14005 length = PyUnicode_GET_LENGTH(unicode);
14006
14007 _PyUnicode_LENGTH(self) = length;
Victor Stinnerfb9ea8c2011-10-06 01:45:57 +020014008#ifdef Py_DEBUG
14009 _PyUnicode_HASH(self) = -1;
14010#else
Victor Stinner07ac3eb2011-10-01 16:16:43 +020014011 _PyUnicode_HASH(self) = _PyUnicode_HASH(unicode);
Victor Stinnerfb9ea8c2011-10-06 01:45:57 +020014012#endif
Victor Stinner07ac3eb2011-10-01 16:16:43 +020014013 _PyUnicode_STATE(self).interned = 0;
14014 _PyUnicode_STATE(self).kind = kind;
14015 _PyUnicode_STATE(self).compact = 0;
Victor Stinner3cf46372011-10-03 14:42:15 +020014016 _PyUnicode_STATE(self).ascii = _PyUnicode_STATE(unicode).ascii;
Victor Stinner07ac3eb2011-10-01 16:16:43 +020014017 _PyUnicode_STATE(self).ready = 1;
14018 _PyUnicode_WSTR(self) = NULL;
14019 _PyUnicode_UTF8_LENGTH(self) = 0;
14020 _PyUnicode_UTF8(self) = NULL;
14021 _PyUnicode_WSTR_LENGTH(self) = 0;
Victor Stinnerc3c74152011-10-02 20:39:55 +020014022 _PyUnicode_DATA_ANY(self) = NULL;
Victor Stinner07ac3eb2011-10-01 16:16:43 +020014023
14024 share_utf8 = 0;
14025 share_wstr = 0;
14026 if (kind == PyUnicode_1BYTE_KIND) {
14027 char_size = 1;
14028 if (PyUnicode_MAX_CHAR_VALUE(unicode) < 128)
14029 share_utf8 = 1;
14030 }
14031 else if (kind == PyUnicode_2BYTE_KIND) {
14032 char_size = 2;
14033 if (sizeof(wchar_t) == 2)
14034 share_wstr = 1;
14035 }
14036 else {
14037 assert(kind == PyUnicode_4BYTE_KIND);
14038 char_size = 4;
14039 if (sizeof(wchar_t) == 4)
14040 share_wstr = 1;
14041 }
14042
14043 /* Ensure we won't overflow the length. */
14044 if (length > (PY_SSIZE_T_MAX / char_size - 1)) {
14045 PyErr_NoMemory();
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020014046 goto onError;
Benjamin Peterson14339b62009-01-31 16:36:08 +000014047 }
Victor Stinner07ac3eb2011-10-01 16:16:43 +020014048 data = PyObject_MALLOC((length + 1) * char_size);
14049 if (data == NULL) {
14050 PyErr_NoMemory();
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020014051 goto onError;
14052 }
14053
Victor Stinnerc3c74152011-10-02 20:39:55 +020014054 _PyUnicode_DATA_ANY(self) = data;
Victor Stinner07ac3eb2011-10-01 16:16:43 +020014055 if (share_utf8) {
14056 _PyUnicode_UTF8_LENGTH(self) = length;
14057 _PyUnicode_UTF8(self) = data;
14058 }
14059 if (share_wstr) {
14060 _PyUnicode_WSTR_LENGTH(self) = length;
14061 _PyUnicode_WSTR(self) = (wchar_t *)data;
14062 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020014063
Victor Stinner07ac3eb2011-10-01 16:16:43 +020014064 Py_MEMCPY(data, PyUnicode_DATA(unicode),
Martin v. Löwisc47adb02011-10-07 20:55:35 +020014065 kind * (length + 1));
Victor Stinnerbb10a1f2011-10-05 01:34:17 +020014066 assert(_PyUnicode_CheckConsistency(self, 1));
Victor Stinnerfb9ea8c2011-10-06 01:45:57 +020014067#ifdef Py_DEBUG
14068 _PyUnicode_HASH(self) = _PyUnicode_HASH(unicode);
14069#endif
Victor Stinnerdd18d3a2011-10-22 11:08:10 +020014070 Py_DECREF(unicode);
Victor Stinner7931d9a2011-11-04 00:22:48 +010014071 return self;
Victor Stinner07ac3eb2011-10-01 16:16:43 +020014072
14073onError:
14074 Py_DECREF(unicode);
14075 Py_DECREF(self);
14076 return NULL;
Guido van Rossume023fe02001-08-30 03:12:59 +000014077}
14078
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000014079PyDoc_STRVAR(unicode_doc,
Chris Jerdonek83fe2e12012-10-07 14:48:36 -070014080"str(object='') -> str\n\
14081str(bytes_or_buffer[, encoding[, errors]]) -> str\n\
Tim Peters6d6c1a32001-08-02 04:15:00 +000014082\n\
Nick Coghlan573b1fd2012-08-16 14:13:07 +100014083Create a new string object from the given object. If encoding or\n\
14084errors is specified, then the object must expose a data buffer\n\
14085that will be decoded using the given encoding and error handler.\n\
14086Otherwise, returns the result of object.__str__() (if defined)\n\
14087or repr(object).\n\
14088encoding defaults to sys.getdefaultencoding().\n\
14089errors defaults to 'strict'.");
Tim Peters6d6c1a32001-08-02 04:15:00 +000014090
Guido van Rossum50e9fb92006-08-17 05:42:55 +000014091static PyObject *unicode_iter(PyObject *seq);
14092
Guido van Rossumd57fd912000-03-10 22:53:23 +000014093PyTypeObject PyUnicode_Type = {
Martin v. Löwis9f2e3462007-07-21 17:22:18 +000014094 PyVarObject_HEAD_INIT(&PyType_Type, 0)
Benjamin Peterson14339b62009-01-31 16:36:08 +000014095 "str", /* tp_name */
14096 sizeof(PyUnicodeObject), /* tp_size */
14097 0, /* tp_itemsize */
Guido van Rossumd57fd912000-03-10 22:53:23 +000014098 /* Slots */
Benjamin Peterson14339b62009-01-31 16:36:08 +000014099 (destructor)unicode_dealloc, /* tp_dealloc */
14100 0, /* tp_print */
14101 0, /* tp_getattr */
14102 0, /* tp_setattr */
Mark Dickinsone94c6792009-02-02 20:36:42 +000014103 0, /* tp_reserved */
Benjamin Peterson14339b62009-01-31 16:36:08 +000014104 unicode_repr, /* tp_repr */
14105 &unicode_as_number, /* tp_as_number */
14106 &unicode_as_sequence, /* tp_as_sequence */
14107 &unicode_as_mapping, /* tp_as_mapping */
14108 (hashfunc) unicode_hash, /* tp_hash*/
14109 0, /* tp_call*/
14110 (reprfunc) unicode_str, /* tp_str */
14111 PyObject_GenericGetAttr, /* tp_getattro */
14112 0, /* tp_setattro */
14113 0, /* tp_as_buffer */
14114 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE |
Benjamin Peterson29060642009-01-31 22:14:21 +000014115 Py_TPFLAGS_UNICODE_SUBCLASS, /* tp_flags */
Benjamin Peterson14339b62009-01-31 16:36:08 +000014116 unicode_doc, /* tp_doc */
14117 0, /* tp_traverse */
14118 0, /* tp_clear */
14119 PyUnicode_RichCompare, /* tp_richcompare */
14120 0, /* tp_weaklistoffset */
14121 unicode_iter, /* tp_iter */
14122 0, /* tp_iternext */
14123 unicode_methods, /* tp_methods */
14124 0, /* tp_members */
14125 0, /* tp_getset */
14126 &PyBaseObject_Type, /* tp_base */
14127 0, /* tp_dict */
14128 0, /* tp_descr_get */
14129 0, /* tp_descr_set */
14130 0, /* tp_dictoffset */
14131 0, /* tp_init */
14132 0, /* tp_alloc */
14133 unicode_new, /* tp_new */
14134 PyObject_Del, /* tp_free */
Guido van Rossumd57fd912000-03-10 22:53:23 +000014135};
14136
14137/* Initialize the Unicode implementation */
14138
Victor Stinner3a50e702011-10-18 21:21:00 +020014139int _PyUnicode_Init(void)
Guido van Rossumd57fd912000-03-10 22:53:23 +000014140{
Thomas Wouters477c8d52006-05-27 19:21:47 +000014141 /* XXX - move this array to unicodectype.c ? */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020014142 Py_UCS2 linebreak[] = {
Thomas Wouters477c8d52006-05-27 19:21:47 +000014143 0x000A, /* LINE FEED */
14144 0x000D, /* CARRIAGE RETURN */
14145 0x001C, /* FILE SEPARATOR */
14146 0x001D, /* GROUP SEPARATOR */
14147 0x001E, /* RECORD SEPARATOR */
14148 0x0085, /* NEXT LINE */
14149 0x2028, /* LINE SEPARATOR */
14150 0x2029, /* PARAGRAPH SEPARATOR */
14151 };
14152
Fred Drakee4315f52000-05-09 19:53:39 +000014153 /* Init the implementation */
Serhiy Storchaka678db842013-01-26 12:16:36 +020014154 _Py_INCREF_UNICODE_EMPTY();
Thomas Wouters0e3f5912006-08-11 14:57:12 +000014155 if (!unicode_empty)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020014156 Py_FatalError("Can't create empty string");
Serhiy Storchaka678db842013-01-26 12:16:36 +020014157 Py_DECREF(unicode_empty);
Thomas Wouters0e3f5912006-08-11 14:57:12 +000014158
Guido van Rossumcacfc072002-05-24 19:01:59 +000014159 if (PyType_Ready(&PyUnicode_Type) < 0)
Benjamin Peterson29060642009-01-31 22:14:21 +000014160 Py_FatalError("Can't initialize 'unicode'");
Thomas Wouters477c8d52006-05-27 19:21:47 +000014161
14162 /* initialize the linebreak bloom filter */
14163 bloom_linebreak = make_bloom_mask(
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020014164 PyUnicode_2BYTE_KIND, linebreak,
Victor Stinner63941882011-09-29 00:42:28 +020014165 Py_ARRAY_LENGTH(linebreak));
Thomas Wouters0e3f5912006-08-11 14:57:12 +000014166
Christian Heimes26532f72013-07-20 14:57:16 +020014167 if (PyType_Ready(&EncodingMapType) < 0)
14168 Py_FatalError("Can't initialize encoding map type");
Victor Stinner3a50e702011-10-18 21:21:00 +020014169
Benjamin Petersonc4311282012-10-30 23:21:10 -040014170 if (PyType_Ready(&PyFieldNameIter_Type) < 0)
14171 Py_FatalError("Can't initialize field name iterator type");
14172
14173 if (PyType_Ready(&PyFormatterIter_Type) < 0)
14174 Py_FatalError("Can't initialize formatter iter type");
Benjamin Petersone8ea97f2012-10-30 23:27:52 -040014175
Victor Stinner3a50e702011-10-18 21:21:00 +020014176#ifdef HAVE_MBCS
14177 winver.dwOSVersionInfoSize = sizeof(winver);
14178 if (!GetVersionEx((OSVERSIONINFO*)&winver)) {
14179 PyErr_SetFromWindowsErr(0);
14180 return -1;
14181 }
14182#endif
14183 return 0;
Guido van Rossumd57fd912000-03-10 22:53:23 +000014184}
14185
14186/* Finalize the Unicode implementation */
14187
Christian Heimesa156e092008-02-16 07:38:31 +000014188int
14189PyUnicode_ClearFreeList(void)
14190{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020014191 return 0;
Christian Heimesa156e092008-02-16 07:38:31 +000014192}
14193
Guido van Rossumd57fd912000-03-10 22:53:23 +000014194void
Thomas Wouters78890102000-07-22 19:25:51 +000014195_PyUnicode_Fini(void)
Guido van Rossumd57fd912000-03-10 22:53:23 +000014196{
Marc-André Lemburg8155e0e2001-04-23 14:44:21 +000014197 int i;
Guido van Rossumd57fd912000-03-10 22:53:23 +000014198
Serhiy Storchaka05997252013-01-26 12:14:02 +020014199 Py_CLEAR(unicode_empty);
Barry Warsaw5b4c2282000-10-03 20:45:26 +000014200
Serhiy Storchaka05997252013-01-26 12:14:02 +020014201 for (i = 0; i < 256; i++)
14202 Py_CLEAR(unicode_latin1[i]);
Martin v. Löwisafe55bb2011-10-09 10:38:36 +020014203 _PyUnicode_ClearStaticStrings();
Christian Heimesa156e092008-02-16 07:38:31 +000014204 (void)PyUnicode_ClearFreeList();
Guido van Rossumd57fd912000-03-10 22:53:23 +000014205}
Martin v. Löwis9a3a9f72003-05-18 12:31:09 +000014206
Walter Dörwald16807132007-05-25 13:52:07 +000014207void
14208PyUnicode_InternInPlace(PyObject **p)
14209{
Victor Stinner9db1a8b2011-10-23 20:04:37 +020014210 register PyObject *s = *p;
Benjamin Peterson14339b62009-01-31 16:36:08 +000014211 PyObject *t;
Victor Stinner4fae54c2011-10-03 02:01:52 +020014212#ifdef Py_DEBUG
14213 assert(s != NULL);
14214 assert(_PyUnicode_CHECK(s));
14215#else
Benjamin Peterson14339b62009-01-31 16:36:08 +000014216 if (s == NULL || !PyUnicode_Check(s))
Victor Stinner4fae54c2011-10-03 02:01:52 +020014217 return;
14218#endif
Benjamin Peterson14339b62009-01-31 16:36:08 +000014219 /* If it's a subclass, we don't really know what putting
14220 it in the interned dict might do. */
14221 if (!PyUnicode_CheckExact(s))
14222 return;
14223 if (PyUnicode_CHECK_INTERNED(s))
14224 return;
14225 if (interned == NULL) {
14226 interned = PyDict_New();
14227 if (interned == NULL) {
14228 PyErr_Clear(); /* Don't leave an exception */
14229 return;
14230 }
14231 }
14232 /* It might be that the GetItem call fails even
14233 though the key is present in the dictionary,
14234 namely when this happens during a stack overflow. */
14235 Py_ALLOW_RECURSION
Victor Stinner7931d9a2011-11-04 00:22:48 +010014236 t = PyDict_GetItem(interned, s);
Benjamin Peterson14339b62009-01-31 16:36:08 +000014237 Py_END_ALLOW_RECURSION
Martin v. Löwis5b222132007-06-10 09:51:05 +000014238
Benjamin Peterson29060642009-01-31 22:14:21 +000014239 if (t) {
14240 Py_INCREF(t);
14241 Py_DECREF(*p);
14242 *p = t;
14243 return;
14244 }
Walter Dörwald16807132007-05-25 13:52:07 +000014245
Benjamin Peterson14339b62009-01-31 16:36:08 +000014246 PyThreadState_GET()->recursion_critical = 1;
Victor Stinner7931d9a2011-11-04 00:22:48 +010014247 if (PyDict_SetItem(interned, s, s) < 0) {
Benjamin Peterson14339b62009-01-31 16:36:08 +000014248 PyErr_Clear();
14249 PyThreadState_GET()->recursion_critical = 0;
14250 return;
14251 }
14252 PyThreadState_GET()->recursion_critical = 0;
14253 /* The two references in interned are not counted by refcnt.
14254 The deallocator will take care of this */
14255 Py_REFCNT(s) -= 2;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020014256 _PyUnicode_STATE(s).interned = SSTATE_INTERNED_MORTAL;
Walter Dörwald16807132007-05-25 13:52:07 +000014257}
14258
14259void
14260PyUnicode_InternImmortal(PyObject **p)
14261{
Benjamin Peterson14339b62009-01-31 16:36:08 +000014262 PyUnicode_InternInPlace(p);
14263 if (PyUnicode_CHECK_INTERNED(*p) != SSTATE_INTERNED_IMMORTAL) {
Victor Stinneraf9e4b82011-10-23 20:07:00 +020014264 _PyUnicode_STATE(*p).interned = SSTATE_INTERNED_IMMORTAL;
Benjamin Peterson14339b62009-01-31 16:36:08 +000014265 Py_INCREF(*p);
14266 }
Walter Dörwald16807132007-05-25 13:52:07 +000014267}
14268
14269PyObject *
14270PyUnicode_InternFromString(const char *cp)
14271{
Benjamin Peterson14339b62009-01-31 16:36:08 +000014272 PyObject *s = PyUnicode_FromString(cp);
14273 if (s == NULL)
14274 return NULL;
14275 PyUnicode_InternInPlace(&s);
14276 return s;
Walter Dörwald16807132007-05-25 13:52:07 +000014277}
14278
Alexander Belopolsky40018472011-02-26 01:02:56 +000014279void
14280_Py_ReleaseInternedUnicodeStrings(void)
Walter Dörwald16807132007-05-25 13:52:07 +000014281{
Benjamin Peterson14339b62009-01-31 16:36:08 +000014282 PyObject *keys;
Victor Stinner9db1a8b2011-10-23 20:04:37 +020014283 PyObject *s;
Benjamin Peterson14339b62009-01-31 16:36:08 +000014284 Py_ssize_t i, n;
14285 Py_ssize_t immortal_size = 0, mortal_size = 0;
Walter Dörwald16807132007-05-25 13:52:07 +000014286
Benjamin Peterson14339b62009-01-31 16:36:08 +000014287 if (interned == NULL || !PyDict_Check(interned))
14288 return;
14289 keys = PyDict_Keys(interned);
14290 if (keys == NULL || !PyList_Check(keys)) {
14291 PyErr_Clear();
14292 return;
14293 }
Walter Dörwald16807132007-05-25 13:52:07 +000014294
Benjamin Peterson14339b62009-01-31 16:36:08 +000014295 /* Since _Py_ReleaseInternedUnicodeStrings() is intended to help a leak
14296 detector, interned unicode strings are not forcibly deallocated;
14297 rather, we give them their stolen references back, and then clear
14298 and DECREF the interned dict. */
Walter Dörwald16807132007-05-25 13:52:07 +000014299
Benjamin Peterson14339b62009-01-31 16:36:08 +000014300 n = PyList_GET_SIZE(keys);
14301 fprintf(stderr, "releasing %" PY_FORMAT_SIZE_T "d interned strings\n",
Benjamin Peterson29060642009-01-31 22:14:21 +000014302 n);
Benjamin Peterson14339b62009-01-31 16:36:08 +000014303 for (i = 0; i < n; i++) {
Victor Stinner9db1a8b2011-10-23 20:04:37 +020014304 s = PyList_GET_ITEM(keys, i);
Victor Stinner6b56a7f2011-10-04 20:04:52 +020014305 if (PyUnicode_READY(s) == -1) {
14306 assert(0 && "could not ready string");
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020014307 fprintf(stderr, "could not ready string\n");
Victor Stinner6b56a7f2011-10-04 20:04:52 +020014308 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020014309 switch (PyUnicode_CHECK_INTERNED(s)) {
Benjamin Peterson14339b62009-01-31 16:36:08 +000014310 case SSTATE_NOT_INTERNED:
14311 /* XXX Shouldn't happen */
14312 break;
14313 case SSTATE_INTERNED_IMMORTAL:
14314 Py_REFCNT(s) += 1;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020014315 immortal_size += PyUnicode_GET_LENGTH(s);
Benjamin Peterson14339b62009-01-31 16:36:08 +000014316 break;
14317 case SSTATE_INTERNED_MORTAL:
14318 Py_REFCNT(s) += 2;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020014319 mortal_size += PyUnicode_GET_LENGTH(s);
Benjamin Peterson14339b62009-01-31 16:36:08 +000014320 break;
14321 default:
14322 Py_FatalError("Inconsistent interned string state.");
14323 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020014324 _PyUnicode_STATE(s).interned = SSTATE_NOT_INTERNED;
Benjamin Peterson14339b62009-01-31 16:36:08 +000014325 }
14326 fprintf(stderr, "total size of all interned strings: "
14327 "%" PY_FORMAT_SIZE_T "d/%" PY_FORMAT_SIZE_T "d "
14328 "mortal/immortal\n", mortal_size, immortal_size);
14329 Py_DECREF(keys);
14330 PyDict_Clear(interned);
Serhiy Storchaka05997252013-01-26 12:14:02 +020014331 Py_CLEAR(interned);
Walter Dörwald16807132007-05-25 13:52:07 +000014332}
Guido van Rossum50e9fb92006-08-17 05:42:55 +000014333
14334
14335/********************* Unicode Iterator **************************/
14336
14337typedef struct {
Benjamin Peterson14339b62009-01-31 16:36:08 +000014338 PyObject_HEAD
14339 Py_ssize_t it_index;
Victor Stinner9db1a8b2011-10-23 20:04:37 +020014340 PyObject *it_seq; /* Set to NULL when iterator is exhausted */
Guido van Rossum50e9fb92006-08-17 05:42:55 +000014341} unicodeiterobject;
14342
14343static void
14344unicodeiter_dealloc(unicodeiterobject *it)
14345{
Benjamin Peterson14339b62009-01-31 16:36:08 +000014346 _PyObject_GC_UNTRACK(it);
14347 Py_XDECREF(it->it_seq);
14348 PyObject_GC_Del(it);
Guido van Rossum50e9fb92006-08-17 05:42:55 +000014349}
14350
14351static int
14352unicodeiter_traverse(unicodeiterobject *it, visitproc visit, void *arg)
14353{
Benjamin Peterson14339b62009-01-31 16:36:08 +000014354 Py_VISIT(it->it_seq);
14355 return 0;
Guido van Rossum50e9fb92006-08-17 05:42:55 +000014356}
14357
14358static PyObject *
14359unicodeiter_next(unicodeiterobject *it)
14360{
Victor Stinner9db1a8b2011-10-23 20:04:37 +020014361 PyObject *seq, *item;
Guido van Rossum50e9fb92006-08-17 05:42:55 +000014362
Benjamin Peterson14339b62009-01-31 16:36:08 +000014363 assert(it != NULL);
14364 seq = it->it_seq;
14365 if (seq == NULL)
14366 return NULL;
Victor Stinner910337b2011-10-03 03:20:16 +020014367 assert(_PyUnicode_CHECK(seq));
Guido van Rossum50e9fb92006-08-17 05:42:55 +000014368
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020014369 if (it->it_index < PyUnicode_GET_LENGTH(seq)) {
14370 int kind = PyUnicode_KIND(seq);
14371 void *data = PyUnicode_DATA(seq);
14372 Py_UCS4 chr = PyUnicode_READ(kind, data, it->it_index);
14373 item = PyUnicode_FromOrdinal(chr);
Benjamin Peterson14339b62009-01-31 16:36:08 +000014374 if (item != NULL)
14375 ++it->it_index;
14376 return item;
14377 }
Guido van Rossum50e9fb92006-08-17 05:42:55 +000014378
Benjamin Peterson14339b62009-01-31 16:36:08 +000014379 Py_DECREF(seq);
14380 it->it_seq = NULL;
14381 return NULL;
Guido van Rossum50e9fb92006-08-17 05:42:55 +000014382}
14383
14384static PyObject *
14385unicodeiter_len(unicodeiterobject *it)
14386{
Benjamin Peterson14339b62009-01-31 16:36:08 +000014387 Py_ssize_t len = 0;
14388 if (it->it_seq)
Victor Stinnerc4f281e2011-10-11 22:11:42 +020014389 len = PyUnicode_GET_LENGTH(it->it_seq) - it->it_index;
Benjamin Peterson14339b62009-01-31 16:36:08 +000014390 return PyLong_FromSsize_t(len);
Guido van Rossum50e9fb92006-08-17 05:42:55 +000014391}
14392
14393PyDoc_STRVAR(length_hint_doc, "Private method returning an estimate of len(list(it)).");
14394
Kristján Valur Jónsson31668b82012-04-03 10:49:41 +000014395static PyObject *
14396unicodeiter_reduce(unicodeiterobject *it)
14397{
14398 if (it->it_seq != NULL) {
Antoine Pitroua7013882012-04-05 00:04:20 +020014399 return Py_BuildValue("N(O)n", _PyObject_GetBuiltin("iter"),
Kristján Valur Jónsson31668b82012-04-03 10:49:41 +000014400 it->it_seq, it->it_index);
14401 } else {
14402 PyObject *u = PyUnicode_FromUnicode(NULL, 0);
14403 if (u == NULL)
14404 return NULL;
Antoine Pitroua7013882012-04-05 00:04:20 +020014405 return Py_BuildValue("N(N)", _PyObject_GetBuiltin("iter"), u);
Kristján Valur Jónsson31668b82012-04-03 10:49:41 +000014406 }
14407}
14408
14409PyDoc_STRVAR(reduce_doc, "Return state information for pickling.");
14410
14411static PyObject *
14412unicodeiter_setstate(unicodeiterobject *it, PyObject *state)
14413{
14414 Py_ssize_t index = PyLong_AsSsize_t(state);
14415 if (index == -1 && PyErr_Occurred())
14416 return NULL;
14417 if (index < 0)
14418 index = 0;
14419 it->it_index = index;
14420 Py_RETURN_NONE;
14421}
14422
14423PyDoc_STRVAR(setstate_doc, "Set state information for unpickling.");
14424
Guido van Rossum50e9fb92006-08-17 05:42:55 +000014425static PyMethodDef unicodeiter_methods[] = {
Benjamin Peterson14339b62009-01-31 16:36:08 +000014426 {"__length_hint__", (PyCFunction)unicodeiter_len, METH_NOARGS,
Benjamin Peterson29060642009-01-31 22:14:21 +000014427 length_hint_doc},
Kristján Valur Jónsson31668b82012-04-03 10:49:41 +000014428 {"__reduce__", (PyCFunction)unicodeiter_reduce, METH_NOARGS,
14429 reduce_doc},
14430 {"__setstate__", (PyCFunction)unicodeiter_setstate, METH_O,
14431 setstate_doc},
Benjamin Peterson14339b62009-01-31 16:36:08 +000014432 {NULL, NULL} /* sentinel */
Guido van Rossum50e9fb92006-08-17 05:42:55 +000014433};
14434
14435PyTypeObject PyUnicodeIter_Type = {
Benjamin Peterson14339b62009-01-31 16:36:08 +000014436 PyVarObject_HEAD_INIT(&PyType_Type, 0)
14437 "str_iterator", /* tp_name */
14438 sizeof(unicodeiterobject), /* tp_basicsize */
14439 0, /* tp_itemsize */
14440 /* methods */
14441 (destructor)unicodeiter_dealloc, /* tp_dealloc */
14442 0, /* tp_print */
14443 0, /* tp_getattr */
14444 0, /* tp_setattr */
Mark Dickinsone94c6792009-02-02 20:36:42 +000014445 0, /* tp_reserved */
Benjamin Peterson14339b62009-01-31 16:36:08 +000014446 0, /* tp_repr */
14447 0, /* tp_as_number */
14448 0, /* tp_as_sequence */
14449 0, /* tp_as_mapping */
14450 0, /* tp_hash */
14451 0, /* tp_call */
14452 0, /* tp_str */
14453 PyObject_GenericGetAttr, /* tp_getattro */
14454 0, /* tp_setattro */
14455 0, /* tp_as_buffer */
14456 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC,/* tp_flags */
14457 0, /* tp_doc */
14458 (traverseproc)unicodeiter_traverse, /* tp_traverse */
14459 0, /* tp_clear */
14460 0, /* tp_richcompare */
14461 0, /* tp_weaklistoffset */
14462 PyObject_SelfIter, /* tp_iter */
14463 (iternextfunc)unicodeiter_next, /* tp_iternext */
14464 unicodeiter_methods, /* tp_methods */
14465 0,
Guido van Rossum50e9fb92006-08-17 05:42:55 +000014466};
14467
14468static PyObject *
14469unicode_iter(PyObject *seq)
14470{
Benjamin Peterson14339b62009-01-31 16:36:08 +000014471 unicodeiterobject *it;
Guido van Rossum50e9fb92006-08-17 05:42:55 +000014472
Benjamin Peterson14339b62009-01-31 16:36:08 +000014473 if (!PyUnicode_Check(seq)) {
14474 PyErr_BadInternalCall();
14475 return NULL;
14476 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020014477 if (PyUnicode_READY(seq) == -1)
14478 return NULL;
Benjamin Peterson14339b62009-01-31 16:36:08 +000014479 it = PyObject_GC_New(unicodeiterobject, &PyUnicodeIter_Type);
14480 if (it == NULL)
14481 return NULL;
14482 it->it_index = 0;
14483 Py_INCREF(seq);
Victor Stinner9db1a8b2011-10-23 20:04:37 +020014484 it->it_seq = seq;
Benjamin Peterson14339b62009-01-31 16:36:08 +000014485 _PyObject_GC_TRACK(it);
14486 return (PyObject *)it;
Guido van Rossum50e9fb92006-08-17 05:42:55 +000014487}
14488
Martin v. Löwis0d3072e2011-10-31 08:40:56 +010014489
14490size_t
14491Py_UNICODE_strlen(const Py_UNICODE *u)
14492{
14493 int res = 0;
14494 while(*u++)
14495 res++;
14496 return res;
14497}
14498
14499Py_UNICODE*
14500Py_UNICODE_strcpy(Py_UNICODE *s1, const Py_UNICODE *s2)
14501{
14502 Py_UNICODE *u = s1;
14503 while ((*u++ = *s2++));
14504 return s1;
14505}
14506
14507Py_UNICODE*
14508Py_UNICODE_strncpy(Py_UNICODE *s1, const Py_UNICODE *s2, size_t n)
14509{
14510 Py_UNICODE *u = s1;
14511 while ((*u++ = *s2++))
14512 if (n-- == 0)
14513 break;
14514 return s1;
14515}
14516
14517Py_UNICODE*
14518Py_UNICODE_strcat(Py_UNICODE *s1, const Py_UNICODE *s2)
14519{
14520 Py_UNICODE *u1 = s1;
14521 u1 += Py_UNICODE_strlen(u1);
14522 Py_UNICODE_strcpy(u1, s2);
14523 return s1;
14524}
14525
14526int
14527Py_UNICODE_strcmp(const Py_UNICODE *s1, const Py_UNICODE *s2)
14528{
14529 while (*s1 && *s2 && *s1 == *s2)
14530 s1++, s2++;
14531 if (*s1 && *s2)
14532 return (*s1 < *s2) ? -1 : +1;
14533 if (*s1)
14534 return 1;
14535 if (*s2)
14536 return -1;
14537 return 0;
14538}
14539
14540int
14541Py_UNICODE_strncmp(const Py_UNICODE *s1, const Py_UNICODE *s2, size_t n)
14542{
14543 register Py_UNICODE u1, u2;
14544 for (; n != 0; n--) {
14545 u1 = *s1;
14546 u2 = *s2;
14547 if (u1 != u2)
14548 return (u1 < u2) ? -1 : +1;
14549 if (u1 == '\0')
14550 return 0;
14551 s1++;
14552 s2++;
14553 }
14554 return 0;
14555}
14556
14557Py_UNICODE*
14558Py_UNICODE_strchr(const Py_UNICODE *s, Py_UNICODE c)
14559{
14560 const Py_UNICODE *p;
14561 for (p = s; *p; p++)
14562 if (*p == c)
14563 return (Py_UNICODE*)p;
14564 return NULL;
14565}
14566
14567Py_UNICODE*
14568Py_UNICODE_strrchr(const Py_UNICODE *s, Py_UNICODE c)
14569{
14570 const Py_UNICODE *p;
14571 p = s + Py_UNICODE_strlen(s);
14572 while (p != s) {
14573 p--;
14574 if (*p == c)
14575 return (Py_UNICODE*)p;
14576 }
14577 return NULL;
14578}
Victor Stinner331ea922010-08-10 16:37:20 +000014579
Victor Stinner71133ff2010-09-01 23:43:53 +000014580Py_UNICODE*
Victor Stinner9db1a8b2011-10-23 20:04:37 +020014581PyUnicode_AsUnicodeCopy(PyObject *unicode)
Victor Stinner71133ff2010-09-01 23:43:53 +000014582{
Victor Stinner577db2c2011-10-11 22:12:48 +020014583 Py_UNICODE *u, *copy;
Victor Stinner57ffa9d2011-10-23 20:10:08 +020014584 Py_ssize_t len, size;
Victor Stinner71133ff2010-09-01 23:43:53 +000014585
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020014586 if (!PyUnicode_Check(unicode)) {
14587 PyErr_BadArgument();
14588 return NULL;
14589 }
Victor Stinner57ffa9d2011-10-23 20:10:08 +020014590 u = PyUnicode_AsUnicodeAndSize(unicode, &len);
Victor Stinner577db2c2011-10-11 22:12:48 +020014591 if (u == NULL)
14592 return NULL;
Victor Stinner71133ff2010-09-01 23:43:53 +000014593 /* Ensure we won't overflow the size. */
Victor Stinner57ffa9d2011-10-23 20:10:08 +020014594 if (len > ((PY_SSIZE_T_MAX / sizeof(Py_UNICODE)) - 1)) {
Victor Stinner71133ff2010-09-01 23:43:53 +000014595 PyErr_NoMemory();
14596 return NULL;
14597 }
Victor Stinner57ffa9d2011-10-23 20:10:08 +020014598 size = len + 1; /* copy the null character */
Victor Stinner71133ff2010-09-01 23:43:53 +000014599 size *= sizeof(Py_UNICODE);
14600 copy = PyMem_Malloc(size);
14601 if (copy == NULL) {
14602 PyErr_NoMemory();
14603 return NULL;
14604 }
Victor Stinner577db2c2011-10-11 22:12:48 +020014605 memcpy(copy, u, size);
Victor Stinner71133ff2010-09-01 23:43:53 +000014606 return copy;
14607}
Martin v. Löwis5b222132007-06-10 09:51:05 +000014608
Georg Brandl66c221e2010-10-14 07:04:07 +000014609/* A _string module, to export formatter_parser and formatter_field_name_split
14610 to the string.Formatter class implemented in Python. */
14611
14612static PyMethodDef _string_methods[] = {
14613 {"formatter_field_name_split", (PyCFunction) formatter_field_name_split,
14614 METH_O, PyDoc_STR("split the argument as a field name")},
14615 {"formatter_parser", (PyCFunction) formatter_parser,
14616 METH_O, PyDoc_STR("parse the argument as a format string")},
14617 {NULL, NULL}
14618};
14619
14620static struct PyModuleDef _string_module = {
14621 PyModuleDef_HEAD_INIT,
14622 "_string",
14623 PyDoc_STR("string helper module"),
14624 0,
14625 _string_methods,
14626 NULL,
14627 NULL,
14628 NULL,
14629 NULL
14630};
14631
14632PyMODINIT_FUNC
14633PyInit__string(void)
14634{
14635 return PyModule_Create(&_string_module);
14636}
14637
14638
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000014639#ifdef __cplusplus
14640}
14641#endif