blob: 61f743ebda60bea74240f9c402be00c1da5952d9 [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
60 The globals are initialized by the _PyUnicode_Init() API and should
61 not be used before calling that API.
62
63*/
Guido van Rossumd57fd912000-03-10 22:53:23 +000064
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000065
66#ifdef __cplusplus
67extern "C" {
68#endif
69
Victor Stinner8faf8212011-12-08 22:14:11 +010070/* Maximum code point of Unicode 6.0: 0x10ffff (1,114,111) */
71#define MAX_UNICODE 0x10ffff
72
Victor Stinner910337b2011-10-03 03:20:16 +020073#ifdef Py_DEBUG
Victor Stinnerbb10a1f2011-10-05 01:34:17 +020074# define _PyUnicode_CHECK(op) _PyUnicode_CheckConsistency(op, 0)
Victor Stinner910337b2011-10-03 03:20:16 +020075#else
76# define _PyUnicode_CHECK(op) PyUnicode_Check(op)
77#endif
Victor Stinnerfb5f5f22011-09-28 21:39:49 +020078
Victor Stinnere90fe6a2011-10-01 16:48:13 +020079#define _PyUnicode_UTF8(op) \
80 (((PyCompactUnicodeObject*)(op))->utf8)
81#define PyUnicode_UTF8(op) \
Victor Stinner910337b2011-10-03 03:20:16 +020082 (assert(_PyUnicode_CHECK(op)), \
Victor Stinnere90fe6a2011-10-01 16:48:13 +020083 assert(PyUnicode_IS_READY(op)), \
84 PyUnicode_IS_COMPACT_ASCII(op) ? \
85 ((char*)((PyASCIIObject*)(op) + 1)) : \
86 _PyUnicode_UTF8(op))
Victor Stinnerbc8b81b2011-09-29 19:31:34 +020087#define _PyUnicode_UTF8_LENGTH(op) \
Victor Stinnere90fe6a2011-10-01 16:48:13 +020088 (((PyCompactUnicodeObject*)(op))->utf8_length)
89#define PyUnicode_UTF8_LENGTH(op) \
Victor Stinner910337b2011-10-03 03:20:16 +020090 (assert(_PyUnicode_CHECK(op)), \
Victor Stinnere90fe6a2011-10-01 16:48:13 +020091 assert(PyUnicode_IS_READY(op)), \
92 PyUnicode_IS_COMPACT_ASCII(op) ? \
93 ((PyASCIIObject*)(op))->length : \
94 _PyUnicode_UTF8_LENGTH(op))
Victor Stinnera5f91632011-10-04 01:07:11 +020095#define _PyUnicode_WSTR(op) \
96 (((PyASCIIObject*)(op))->wstr)
97#define _PyUnicode_WSTR_LENGTH(op) \
98 (((PyCompactUnicodeObject*)(op))->wstr_length)
99#define _PyUnicode_LENGTH(op) \
100 (((PyASCIIObject *)(op))->length)
101#define _PyUnicode_STATE(op) \
102 (((PyASCIIObject *)(op))->state)
103#define _PyUnicode_HASH(op) \
104 (((PyASCIIObject *)(op))->hash)
Victor Stinner910337b2011-10-03 03:20:16 +0200105#define _PyUnicode_KIND(op) \
106 (assert(_PyUnicode_CHECK(op)), \
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200107 ((PyASCIIObject *)(op))->state.kind)
Victor Stinner910337b2011-10-03 03:20:16 +0200108#define _PyUnicode_GET_LENGTH(op) \
109 (assert(_PyUnicode_CHECK(op)), \
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200110 ((PyASCIIObject *)(op))->length)
Victor Stinnera5f91632011-10-04 01:07:11 +0200111#define _PyUnicode_DATA_ANY(op) \
112 (((PyUnicodeObject*)(op))->data.any)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200113
Victor Stinnere6abb482012-05-02 01:15:40 +0200114/* Optimized version of Py_MAX() to compute the maximum character:
115 use it when your are computing the second argument of PyUnicode_New() */
116#define MAX_MAXCHAR(maxchar1, maxchar2) \
117 ((maxchar1) | (maxchar2))
118
Victor Stinner910337b2011-10-03 03:20:16 +0200119#undef PyUnicode_READY
120#define PyUnicode_READY(op) \
121 (assert(_PyUnicode_CHECK(op)), \
122 (PyUnicode_IS_READY(op) ? \
Victor Stinnera5f91632011-10-04 01:07:11 +0200123 0 : \
Victor Stinner7931d9a2011-11-04 00:22:48 +0100124 _PyUnicode_Ready(op)))
Victor Stinner910337b2011-10-03 03:20:16 +0200125
Victor Stinnerc379ead2011-10-03 12:52:27 +0200126#define _PyUnicode_SHARE_UTF8(op) \
127 (assert(_PyUnicode_CHECK(op)), \
128 assert(!PyUnicode_IS_COMPACT_ASCII(op)), \
129 (_PyUnicode_UTF8(op) == PyUnicode_DATA(op)))
130#define _PyUnicode_SHARE_WSTR(op) \
131 (assert(_PyUnicode_CHECK(op)), \
132 (_PyUnicode_WSTR(unicode) == PyUnicode_DATA(op)))
133
Victor Stinner829c0ad2011-10-03 01:08:02 +0200134/* true if the Unicode object has an allocated UTF-8 memory block
135 (not shared with other data) */
Victor Stinner910337b2011-10-03 03:20:16 +0200136#define _PyUnicode_HAS_UTF8_MEMORY(op) \
137 (assert(_PyUnicode_CHECK(op)), \
138 (!PyUnicode_IS_COMPACT_ASCII(op) \
139 && _PyUnicode_UTF8(op) \
Victor Stinner829c0ad2011-10-03 01:08:02 +0200140 && _PyUnicode_UTF8(op) != PyUnicode_DATA(op)))
141
Victor Stinner03490912011-10-03 23:45:12 +0200142/* true if the Unicode object has an allocated wstr memory block
143 (not shared with other data) */
144#define _PyUnicode_HAS_WSTR_MEMORY(op) \
145 (assert(_PyUnicode_CHECK(op)), \
146 (_PyUnicode_WSTR(op) && \
147 (!PyUnicode_IS_READY(op) || \
148 _PyUnicode_WSTR(op) != PyUnicode_DATA(op))))
149
Victor Stinner910337b2011-10-03 03:20:16 +0200150/* Generic helper macro to convert characters of different types.
151 from_type and to_type have to be valid type names, begin and end
152 are pointers to the source characters which should be of type
153 "from_type *". to is a pointer of type "to_type *" and points to the
154 buffer where the result characters are written to. */
155#define _PyUnicode_CONVERT_BYTES(from_type, to_type, begin, end, to) \
156 do { \
Antoine Pitroue459a082011-10-11 20:58:41 +0200157 to_type *_to = (to_type *) to; \
158 const from_type *_iter = (begin); \
159 const from_type *_end = (end); \
160 Py_ssize_t n = (_end) - (_iter); \
161 const from_type *_unrolled_end = \
162 _iter + (n & ~ (Py_ssize_t) 3); \
163 while (_iter < (_unrolled_end)) { \
164 _to[0] = (to_type) _iter[0]; \
165 _to[1] = (to_type) _iter[1]; \
166 _to[2] = (to_type) _iter[2]; \
167 _to[3] = (to_type) _iter[3]; \
168 _iter += 4; _to += 4; \
Victor Stinner910337b2011-10-03 03:20:16 +0200169 } \
Antoine Pitroue459a082011-10-11 20:58:41 +0200170 while (_iter < (_end)) \
171 *_to++ = (to_type) *_iter++; \
Victor Stinner910337b2011-10-03 03:20:16 +0200172 } while (0)
Victor Stinner829c0ad2011-10-03 01:08:02 +0200173
Walter Dörwald16807132007-05-25 13:52:07 +0000174/* This dictionary holds all interned unicode strings. Note that references
175 to strings in this dictionary are *not* counted in the string's ob_refcnt.
176 When the interned string reaches a refcnt of 0 the string deallocation
177 function will delete the reference from this dictionary.
178
179 Another way to look at this is that to say that the actual reference
Guido van Rossum98297ee2007-11-06 21:34:58 +0000180 count of a string is: s->ob_refcnt + (s->state ? 2 : 0)
Walter Dörwald16807132007-05-25 13:52:07 +0000181*/
182static PyObject *interned;
183
Marc-André Lemburg8155e0e2001-04-23 14:44:21 +0000184/* The empty Unicode object is shared to improve performance. */
Victor Stinnera464fc12011-10-02 20:39:30 +0200185static PyObject *unicode_empty;
Marc-André Lemburg8155e0e2001-04-23 14:44:21 +0000186
Martin v. Löwisafe55bb2011-10-09 10:38:36 +0200187/* List of static strings. */
188static _Py_Identifier *static_strings;
189
Marc-André Lemburg8155e0e2001-04-23 14:44:21 +0000190/* Single character Unicode strings in the Latin-1 range are being
191 shared as well. */
Victor Stinnera464fc12011-10-02 20:39:30 +0200192static PyObject *unicode_latin1[256];
Marc-André Lemburg8155e0e2001-04-23 14:44:21 +0000193
Christian Heimes190d79e2008-01-30 11:58:22 +0000194/* Fast detection of the most frequent whitespace characters */
195const unsigned char _Py_ascii_whitespace[] = {
Benjamin Peterson14339b62009-01-31 16:36:08 +0000196 0, 0, 0, 0, 0, 0, 0, 0,
Florent Xicluna806d8cf2010-03-30 19:34:18 +0000197/* case 0x0009: * CHARACTER TABULATION */
Christian Heimes1a8501c2008-10-02 19:56:01 +0000198/* case 0x000A: * LINE FEED */
Florent Xicluna806d8cf2010-03-30 19:34:18 +0000199/* case 0x000B: * LINE TABULATION */
Christian Heimes1a8501c2008-10-02 19:56:01 +0000200/* case 0x000C: * FORM FEED */
201/* case 0x000D: * CARRIAGE RETURN */
Benjamin Peterson14339b62009-01-31 16:36:08 +0000202 0, 1, 1, 1, 1, 1, 0, 0,
203 0, 0, 0, 0, 0, 0, 0, 0,
Christian Heimes1a8501c2008-10-02 19:56:01 +0000204/* case 0x001C: * FILE SEPARATOR */
205/* case 0x001D: * GROUP SEPARATOR */
206/* case 0x001E: * RECORD SEPARATOR */
207/* case 0x001F: * UNIT SEPARATOR */
Benjamin Peterson14339b62009-01-31 16:36:08 +0000208 0, 0, 0, 0, 1, 1, 1, 1,
Christian Heimes1a8501c2008-10-02 19:56:01 +0000209/* case 0x0020: * SPACE */
Benjamin Peterson14339b62009-01-31 16:36:08 +0000210 1, 0, 0, 0, 0, 0, 0, 0,
211 0, 0, 0, 0, 0, 0, 0, 0,
212 0, 0, 0, 0, 0, 0, 0, 0,
213 0, 0, 0, 0, 0, 0, 0, 0,
Christian Heimes190d79e2008-01-30 11:58:22 +0000214
Benjamin Peterson14339b62009-01-31 16:36:08 +0000215 0, 0, 0, 0, 0, 0, 0, 0,
216 0, 0, 0, 0, 0, 0, 0, 0,
217 0, 0, 0, 0, 0, 0, 0, 0,
218 0, 0, 0, 0, 0, 0, 0, 0,
219 0, 0, 0, 0, 0, 0, 0, 0,
220 0, 0, 0, 0, 0, 0, 0, 0,
221 0, 0, 0, 0, 0, 0, 0, 0,
222 0, 0, 0, 0, 0, 0, 0, 0
Christian Heimes190d79e2008-01-30 11:58:22 +0000223};
224
Victor Stinner1b4f9ce2011-10-03 13:28:14 +0200225/* forward */
Victor Stinnerfe226c02011-10-03 03:52:20 +0200226static PyUnicodeObject *_PyUnicode_New(Py_ssize_t length);
Victor Stinner1b4f9ce2011-10-03 13:28:14 +0200227static PyObject* get_latin1_char(unsigned char ch);
Victor Stinner488fa492011-12-12 00:01:39 +0100228static int unicode_modifiable(PyObject *unicode);
229
Victor Stinnerfe226c02011-10-03 03:52:20 +0200230
Alexander Belopolsky40018472011-02-26 01:02:56 +0000231static PyObject *
Antoine Pitroudd4e2f02011-10-13 00:02:27 +0200232_PyUnicode_FromUCS1(const unsigned char *s, Py_ssize_t size);
233static PyObject *
234_PyUnicode_FromUCS2(const Py_UCS2 *s, Py_ssize_t size);
235static PyObject *
236_PyUnicode_FromUCS4(const Py_UCS4 *s, Py_ssize_t size);
237
238static PyObject *
Alexander Belopolsky40018472011-02-26 01:02:56 +0000239unicode_encode_call_errorhandler(const char *errors,
Martin v. Löwisdb12d452009-05-02 18:52:14 +0000240 PyObject **errorHandler,const char *encoding, const char *reason,
Martin v. Löwis23e275b2011-11-02 18:02:51 +0100241 PyObject *unicode, PyObject **exceptionObject,
Martin v. Löwisdb12d452009-05-02 18:52:14 +0000242 Py_ssize_t startpos, Py_ssize_t endpos, Py_ssize_t *newpos);
243
Alexander Belopolsky40018472011-02-26 01:02:56 +0000244static void
245raise_encode_exception(PyObject **exceptionObject,
Ezio Melotti2aa2b3b2011-09-29 00:58:57 +0300246 const char *encoding,
Martin v. Löwis9e816682011-11-02 12:45:42 +0100247 PyObject *unicode,
248 Py_ssize_t startpos, Py_ssize_t endpos,
249 const char *reason);
Victor Stinner31be90b2010-04-22 19:38:16 +0000250
Christian Heimes190d79e2008-01-30 11:58:22 +0000251/* Same for linebreaks */
252static unsigned char ascii_linebreak[] = {
Benjamin Peterson14339b62009-01-31 16:36:08 +0000253 0, 0, 0, 0, 0, 0, 0, 0,
Christian Heimes1a8501c2008-10-02 19:56:01 +0000254/* 0x000A, * LINE FEED */
Florent Xicluna806d8cf2010-03-30 19:34:18 +0000255/* 0x000B, * LINE TABULATION */
256/* 0x000C, * FORM FEED */
Christian Heimes1a8501c2008-10-02 19:56:01 +0000257/* 0x000D, * CARRIAGE RETURN */
Florent Xicluna806d8cf2010-03-30 19:34:18 +0000258 0, 0, 1, 1, 1, 1, 0, 0,
Benjamin Peterson14339b62009-01-31 16:36:08 +0000259 0, 0, 0, 0, 0, 0, 0, 0,
Christian Heimes1a8501c2008-10-02 19:56:01 +0000260/* 0x001C, * FILE SEPARATOR */
261/* 0x001D, * GROUP SEPARATOR */
262/* 0x001E, * RECORD SEPARATOR */
Benjamin Peterson14339b62009-01-31 16:36:08 +0000263 0, 0, 0, 0, 1, 1, 1, 0,
264 0, 0, 0, 0, 0, 0, 0, 0,
265 0, 0, 0, 0, 0, 0, 0, 0,
266 0, 0, 0, 0, 0, 0, 0, 0,
267 0, 0, 0, 0, 0, 0, 0, 0,
Christian Heimes190d79e2008-01-30 11:58:22 +0000268
Benjamin Peterson14339b62009-01-31 16:36:08 +0000269 0, 0, 0, 0, 0, 0, 0, 0,
270 0, 0, 0, 0, 0, 0, 0, 0,
271 0, 0, 0, 0, 0, 0, 0, 0,
272 0, 0, 0, 0, 0, 0, 0, 0,
273 0, 0, 0, 0, 0, 0, 0, 0,
274 0, 0, 0, 0, 0, 0, 0, 0,
275 0, 0, 0, 0, 0, 0, 0, 0,
276 0, 0, 0, 0, 0, 0, 0, 0
Christian Heimes190d79e2008-01-30 11:58:22 +0000277};
278
Ezio Melotti48a2f8f2011-09-29 00:18:19 +0300279/* The max unicode value is always 0x10FFFF while using the PEP-393 API.
280 This function is kept for backward compatibility with the old API. */
Martin v. Löwisce9b5a52001-06-27 06:28:56 +0000281Py_UNICODE
Marc-André Lemburg6c6bfb72001-07-20 17:39:11 +0000282PyUnicode_GetMax(void)
Martin v. Löwisce9b5a52001-06-27 06:28:56 +0000283{
Fredrik Lundh8f455852001-06-27 18:59:43 +0000284#ifdef Py_UNICODE_WIDE
Benjamin Peterson14339b62009-01-31 16:36:08 +0000285 return 0x10FFFF;
Martin v. Löwisce9b5a52001-06-27 06:28:56 +0000286#else
Benjamin Peterson14339b62009-01-31 16:36:08 +0000287 /* This is actually an illegal character, so it should
288 not be passed to unichr. */
289 return 0xFFFF;
Martin v. Löwisce9b5a52001-06-27 06:28:56 +0000290#endif
291}
292
Victor Stinner910337b2011-10-03 03:20:16 +0200293#ifdef Py_DEBUG
Victor Stinnerfb9ea8c2011-10-06 01:45:57 +0200294int
Victor Stinner7931d9a2011-11-04 00:22:48 +0100295_PyUnicode_CheckConsistency(PyObject *op, int check_content)
Victor Stinner910337b2011-10-03 03:20:16 +0200296{
297 PyASCIIObject *ascii;
298 unsigned int kind;
299
300 assert(PyUnicode_Check(op));
301
302 ascii = (PyASCIIObject *)op;
303 kind = ascii->state.kind;
304
Victor Stinnera3b334d2011-10-03 13:53:37 +0200305 if (ascii->state.ascii == 1 && ascii->state.compact == 1) {
Victor Stinner910337b2011-10-03 03:20:16 +0200306 assert(kind == PyUnicode_1BYTE_KIND);
Victor Stinner910337b2011-10-03 03:20:16 +0200307 assert(ascii->state.ready == 1);
308 }
Victor Stinnera41463c2011-10-04 01:05:08 +0200309 else {
Victor Stinner85041a52011-10-03 14:42:39 +0200310 PyCompactUnicodeObject *compact = (PyCompactUnicodeObject *)op;
Victor Stinner7f11ad42011-10-04 00:00:20 +0200311 void *data;
Victor Stinner910337b2011-10-03 03:20:16 +0200312
Victor Stinnera41463c2011-10-04 01:05:08 +0200313 if (ascii->state.compact == 1) {
314 data = compact + 1;
Victor Stinner910337b2011-10-03 03:20:16 +0200315 assert(kind == PyUnicode_1BYTE_KIND
316 || kind == PyUnicode_2BYTE_KIND
317 || kind == PyUnicode_4BYTE_KIND);
Victor Stinnera41463c2011-10-04 01:05:08 +0200318 assert(ascii->state.ascii == 0);
Victor Stinner910337b2011-10-03 03:20:16 +0200319 assert(ascii->state.ready == 1);
Victor Stinnera41463c2011-10-04 01:05:08 +0200320 assert (compact->utf8 != data);
Victor Stinnere30c0a12011-11-04 20:54:05 +0100321 }
322 else {
Victor Stinnera41463c2011-10-04 01:05:08 +0200323 PyUnicodeObject *unicode = (PyUnicodeObject *)op;
324
325 data = unicode->data.any;
326 if (kind == PyUnicode_WCHAR_KIND) {
Victor Stinnere30c0a12011-11-04 20:54:05 +0100327 assert(ascii->length == 0);
328 assert(ascii->hash == -1);
Victor Stinnera41463c2011-10-04 01:05:08 +0200329 assert(ascii->state.compact == 0);
330 assert(ascii->state.ascii == 0);
331 assert(ascii->state.ready == 0);
Victor Stinnere30c0a12011-11-04 20:54:05 +0100332 assert(ascii->state.interned == SSTATE_NOT_INTERNED);
Victor Stinnera41463c2011-10-04 01:05:08 +0200333 assert(ascii->wstr != NULL);
334 assert(data == NULL);
335 assert(compact->utf8 == NULL);
Victor Stinnera41463c2011-10-04 01:05:08 +0200336 }
337 else {
338 assert(kind == PyUnicode_1BYTE_KIND
339 || kind == PyUnicode_2BYTE_KIND
340 || kind == PyUnicode_4BYTE_KIND);
341 assert(ascii->state.compact == 0);
342 assert(ascii->state.ready == 1);
343 assert(data != NULL);
344 if (ascii->state.ascii) {
345 assert (compact->utf8 == data);
346 assert (compact->utf8_length == ascii->length);
347 }
348 else
349 assert (compact->utf8 != data);
350 }
351 }
352 if (kind != PyUnicode_WCHAR_KIND) {
Victor Stinner7f11ad42011-10-04 00:00:20 +0200353 if (
354#if SIZEOF_WCHAR_T == 2
355 kind == PyUnicode_2BYTE_KIND
356#else
357 kind == PyUnicode_4BYTE_KIND
358#endif
359 )
Victor Stinnera41463c2011-10-04 01:05:08 +0200360 {
361 assert(ascii->wstr == data);
362 assert(compact->wstr_length == ascii->length);
363 } else
364 assert(ascii->wstr != data);
Victor Stinner910337b2011-10-03 03:20:16 +0200365 }
Victor Stinnera41463c2011-10-04 01:05:08 +0200366
367 if (compact->utf8 == NULL)
368 assert(compact->utf8_length == 0);
369 if (ascii->wstr == NULL)
370 assert(compact->wstr_length == 0);
Victor Stinner910337b2011-10-03 03:20:16 +0200371 }
Victor Stinnerbb10a1f2011-10-05 01:34:17 +0200372 /* check that the best kind is used */
373 if (check_content && kind != PyUnicode_WCHAR_KIND)
374 {
375 Py_ssize_t i;
376 Py_UCS4 maxchar = 0;
Victor Stinner718fbf02012-04-26 00:39:37 +0200377 void *data;
378 Py_UCS4 ch;
379
380 data = PyUnicode_DATA(ascii);
Victor Stinnerbb10a1f2011-10-05 01:34:17 +0200381 for (i=0; i < ascii->length; i++)
382 {
Victor Stinner718fbf02012-04-26 00:39:37 +0200383 ch = PyUnicode_READ(kind, data, i);
Victor Stinnerbb10a1f2011-10-05 01:34:17 +0200384 if (ch > maxchar)
385 maxchar = ch;
386 }
387 if (kind == PyUnicode_1BYTE_KIND) {
Victor Stinner77faf692011-11-20 18:56:05 +0100388 if (ascii->state.ascii == 0) {
Victor Stinnerbb10a1f2011-10-05 01:34:17 +0200389 assert(maxchar >= 128);
Victor Stinner77faf692011-11-20 18:56:05 +0100390 assert(maxchar <= 255);
391 }
Victor Stinnerbb10a1f2011-10-05 01:34:17 +0200392 else
393 assert(maxchar < 128);
394 }
Victor Stinner77faf692011-11-20 18:56:05 +0100395 else if (kind == PyUnicode_2BYTE_KIND) {
Victor Stinnerbb10a1f2011-10-05 01:34:17 +0200396 assert(maxchar >= 0x100);
Victor Stinner77faf692011-11-20 18:56:05 +0100397 assert(maxchar <= 0xFFFF);
398 }
399 else {
Victor Stinnerbb10a1f2011-10-05 01:34:17 +0200400 assert(maxchar >= 0x10000);
Victor Stinner8faf8212011-12-08 22:14:11 +0100401 assert(maxchar <= MAX_UNICODE);
Victor Stinner77faf692011-11-20 18:56:05 +0100402 }
Victor Stinner718fbf02012-04-26 00:39:37 +0200403 assert(PyUnicode_READ(kind, data, ascii->length) == 0);
Victor Stinnerbb10a1f2011-10-05 01:34:17 +0200404 }
Benjamin Petersonccc51c12011-10-03 19:34:12 -0400405 return 1;
406}
Victor Stinner910337b2011-10-03 03:20:16 +0200407#endif
408
Victor Stinnerd3df8ab2011-11-22 01:22:34 +0100409static PyObject*
410unicode_result_wchar(PyObject *unicode)
411{
412#ifndef Py_DEBUG
413 Py_ssize_t len;
414
415 assert(Py_REFCNT(unicode) == 1);
416
417 len = _PyUnicode_WSTR_LENGTH(unicode);
418 if (len == 0) {
419 Py_INCREF(unicode_empty);
420 Py_DECREF(unicode);
421 return unicode_empty;
422 }
423
424 if (len == 1) {
425 wchar_t ch = _PyUnicode_WSTR(unicode)[0];
426 if (ch < 256) {
427 PyObject *latin1_char = get_latin1_char((unsigned char)ch);
428 Py_DECREF(unicode);
429 return latin1_char;
430 }
431 }
432
433 if (_PyUnicode_Ready(unicode) < 0) {
434 Py_XDECREF(unicode);
435 return NULL;
436 }
437#else
438 /* don't make the result ready in debug mode to ensure that the caller
439 makes the string ready before using it */
440 assert(_PyUnicode_CheckConsistency(unicode, 1));
441#endif
442 return unicode;
443}
444
445static PyObject*
446unicode_result_ready(PyObject *unicode)
447{
448 Py_ssize_t length;
449
450 length = PyUnicode_GET_LENGTH(unicode);
451 if (length == 0) {
452 if (unicode != unicode_empty) {
453 Py_INCREF(unicode_empty);
454 Py_DECREF(unicode);
455 }
456 return unicode_empty;
457 }
458
459 if (length == 1) {
460 Py_UCS4 ch = PyUnicode_READ_CHAR(unicode, 0);
461 if (ch < 256) {
462 PyObject *latin1_char = unicode_latin1[ch];
463 if (latin1_char != NULL) {
464 if (unicode != latin1_char) {
465 Py_INCREF(latin1_char);
466 Py_DECREF(unicode);
467 }
468 return latin1_char;
469 }
470 else {
471 assert(_PyUnicode_CheckConsistency(unicode, 1));
472 Py_INCREF(unicode);
473 unicode_latin1[ch] = unicode;
474 return unicode;
475 }
476 }
477 }
478
479 assert(_PyUnicode_CheckConsistency(unicode, 1));
480 return unicode;
481}
482
483static PyObject*
484unicode_result(PyObject *unicode)
485{
486 assert(_PyUnicode_CHECK(unicode));
487 if (PyUnicode_IS_READY(unicode))
488 return unicode_result_ready(unicode);
489 else
490 return unicode_result_wchar(unicode);
491}
492
Victor Stinnerc4b49542011-12-11 22:44:26 +0100493static PyObject*
494unicode_result_unchanged(PyObject *unicode)
495{
496 if (PyUnicode_CheckExact(unicode)) {
Benjamin Petersonbac79492012-01-14 13:34:47 -0500497 if (PyUnicode_READY(unicode) == -1)
Victor Stinnerc4b49542011-12-11 22:44:26 +0100498 return NULL;
499 Py_INCREF(unicode);
500 return unicode;
501 }
502 else
503 /* Subtype -- return genuine unicode string with the same value. */
Victor Stinnerbf6e5602011-12-12 01:53:47 +0100504 return _PyUnicode_Copy(unicode);
Victor Stinnerc4b49542011-12-11 22:44:26 +0100505}
506
Victor Stinner3a50e702011-10-18 21:21:00 +0200507#ifdef HAVE_MBCS
508static OSVERSIONINFOEX winver;
509#endif
510
Thomas Wouters477c8d52006-05-27 19:21:47 +0000511/* --- Bloom Filters ----------------------------------------------------- */
512
513/* stuff to implement simple "bloom filters" for Unicode characters.
514 to keep things simple, we use a single bitmask, using the least 5
515 bits from each unicode characters as the bit index. */
516
517/* the linebreak mask is set up by Unicode_Init below */
518
Antoine Pitrouf068f942010-01-13 14:19:12 +0000519#if LONG_BIT >= 128
520#define BLOOM_WIDTH 128
521#elif LONG_BIT >= 64
522#define BLOOM_WIDTH 64
523#elif LONG_BIT >= 32
524#define BLOOM_WIDTH 32
525#else
526#error "LONG_BIT is smaller than 32"
527#endif
528
Thomas Wouters477c8d52006-05-27 19:21:47 +0000529#define BLOOM_MASK unsigned long
530
531static BLOOM_MASK bloom_linebreak;
532
Antoine Pitrouf068f942010-01-13 14:19:12 +0000533#define BLOOM_ADD(mask, ch) ((mask |= (1UL << ((ch) & (BLOOM_WIDTH - 1)))))
534#define BLOOM(mask, ch) ((mask & (1UL << ((ch) & (BLOOM_WIDTH - 1)))))
Thomas Wouters477c8d52006-05-27 19:21:47 +0000535
Benjamin Peterson29060642009-01-31 22:14:21 +0000536#define BLOOM_LINEBREAK(ch) \
537 ((ch) < 128U ? ascii_linebreak[(ch)] : \
538 (BLOOM(bloom_linebreak, (ch)) && Py_UNICODE_ISLINEBREAK(ch)))
Thomas Wouters477c8d52006-05-27 19:21:47 +0000539
Alexander Belopolsky40018472011-02-26 01:02:56 +0000540Py_LOCAL_INLINE(BLOOM_MASK)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200541make_bloom_mask(int kind, void* ptr, Py_ssize_t len)
Thomas Wouters477c8d52006-05-27 19:21:47 +0000542{
543 /* calculate simple bloom-style bitmask for a given unicode string */
544
Antoine Pitrouf068f942010-01-13 14:19:12 +0000545 BLOOM_MASK mask;
Thomas Wouters477c8d52006-05-27 19:21:47 +0000546 Py_ssize_t i;
547
548 mask = 0;
549 for (i = 0; i < len; i++)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200550 BLOOM_ADD(mask, PyUnicode_READ(kind, ptr, i));
Thomas Wouters477c8d52006-05-27 19:21:47 +0000551
552 return mask;
553}
554
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200555#define BLOOM_MEMBER(mask, chr, str) \
556 (BLOOM(mask, chr) \
557 && (PyUnicode_FindChar(str, chr, 0, PyUnicode_GET_LENGTH(str), 1) >= 0))
Thomas Wouters477c8d52006-05-27 19:21:47 +0000558
Antoine Pitroudd4e2f02011-10-13 00:02:27 +0200559/* Compilation of templated routines */
560
561#include "stringlib/asciilib.h"
562#include "stringlib/fastsearch.h"
563#include "stringlib/partition.h"
564#include "stringlib/split.h"
565#include "stringlib/count.h"
566#include "stringlib/find.h"
567#include "stringlib/find_max_char.h"
568#include "stringlib/localeutil.h"
569#include "stringlib/undef.h"
570
571#include "stringlib/ucs1lib.h"
572#include "stringlib/fastsearch.h"
573#include "stringlib/partition.h"
574#include "stringlib/split.h"
575#include "stringlib/count.h"
576#include "stringlib/find.h"
577#include "stringlib/find_max_char.h"
578#include "stringlib/localeutil.h"
579#include "stringlib/undef.h"
580
581#include "stringlib/ucs2lib.h"
582#include "stringlib/fastsearch.h"
583#include "stringlib/partition.h"
584#include "stringlib/split.h"
585#include "stringlib/count.h"
586#include "stringlib/find.h"
587#include "stringlib/find_max_char.h"
588#include "stringlib/localeutil.h"
589#include "stringlib/undef.h"
590
591#include "stringlib/ucs4lib.h"
592#include "stringlib/fastsearch.h"
593#include "stringlib/partition.h"
594#include "stringlib/split.h"
595#include "stringlib/count.h"
596#include "stringlib/find.h"
597#include "stringlib/find_max_char.h"
598#include "stringlib/localeutil.h"
599#include "stringlib/undef.h"
600
Antoine Pitrouf0b934b2011-10-13 18:55:09 +0200601#include "stringlib/unicodedefs.h"
602#include "stringlib/fastsearch.h"
603#include "stringlib/count.h"
604#include "stringlib/find.h"
Antoine Pitrou0a3229d2011-11-21 20:39:13 +0100605#include "stringlib/undef.h"
Antoine Pitrouf0b934b2011-10-13 18:55:09 +0200606
Guido van Rossumd57fd912000-03-10 22:53:23 +0000607/* --- Unicode Object ----------------------------------------------------- */
608
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200609static PyObject *
Victor Stinner9310abb2011-10-05 00:59:23 +0200610fixup(PyObject *self, Py_UCS4 (*fixfct)(PyObject *s));
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200611
Antoine Pitrouf0b934b2011-10-13 18:55:09 +0200612Py_LOCAL_INLINE(Py_ssize_t) findchar(void *s, int kind,
613 Py_ssize_t size, Py_UCS4 ch,
614 int direction)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200615{
Antoine Pitrouf0b934b2011-10-13 18:55:09 +0200616 int mode = (direction == 1) ? FAST_SEARCH : FAST_RSEARCH;
617
618 switch (kind) {
619 case PyUnicode_1BYTE_KIND:
620 {
621 Py_UCS1 ch1 = (Py_UCS1) ch;
622 if (ch1 == ch)
623 return ucs1lib_fastsearch((Py_UCS1 *) s, size, &ch1, 1, 0, mode);
624 else
625 return -1;
626 }
627 case PyUnicode_2BYTE_KIND:
628 {
629 Py_UCS2 ch2 = (Py_UCS2) ch;
630 if (ch2 == ch)
631 return ucs2lib_fastsearch((Py_UCS2 *) s, size, &ch2, 1, 0, mode);
632 else
633 return -1;
634 }
635 case PyUnicode_4BYTE_KIND:
636 return ucs4lib_fastsearch((Py_UCS4 *) s, size, &ch, 1, 0, mode);
637 default:
638 assert(0);
639 return -1;
Victor Stinner9e7a1bc2011-10-13 00:18:12 +0200640 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200641}
642
Victor Stinnerfe226c02011-10-03 03:52:20 +0200643static PyObject*
644resize_compact(PyObject *unicode, Py_ssize_t length)
645{
646 Py_ssize_t char_size;
647 Py_ssize_t struct_size;
648 Py_ssize_t new_size;
649 int share_wstr;
Victor Stinner84def372011-12-11 20:04:56 +0100650 PyObject *new_unicode;
Victor Stinner79891572012-05-03 13:43:07 +0200651 assert(unicode_modifiable(unicode));
Victor Stinnerfe226c02011-10-03 03:52:20 +0200652 assert(PyUnicode_IS_READY(unicode));
Victor Stinner488fa492011-12-12 00:01:39 +0100653 assert(PyUnicode_IS_COMPACT(unicode));
654
Martin v. Löwisc47adb02011-10-07 20:55:35 +0200655 char_size = PyUnicode_KIND(unicode);
Victor Stinner488fa492011-12-12 00:01:39 +0100656 if (PyUnicode_IS_ASCII(unicode))
Victor Stinnerfe226c02011-10-03 03:52:20 +0200657 struct_size = sizeof(PyASCIIObject);
658 else
659 struct_size = sizeof(PyCompactUnicodeObject);
Victor Stinnerc379ead2011-10-03 12:52:27 +0200660 share_wstr = _PyUnicode_SHARE_WSTR(unicode);
Victor Stinnerfe226c02011-10-03 03:52:20 +0200661
Victor Stinnerfe226c02011-10-03 03:52:20 +0200662 if (length > ((PY_SSIZE_T_MAX - struct_size) / char_size - 1)) {
663 PyErr_NoMemory();
664 return NULL;
665 }
666 new_size = (struct_size + (length + 1) * char_size);
667
Victor Stinner84def372011-12-11 20:04:56 +0100668 _Py_DEC_REFTOTAL;
669 _Py_ForgetReference(unicode);
670
671 new_unicode = (PyObject *)PyObject_REALLOC((char *)unicode, new_size);
672 if (new_unicode == NULL) {
Victor Stinnerb0a82a62011-12-12 13:08:33 +0100673 _Py_NewReference(unicode);
Victor Stinnerfe226c02011-10-03 03:52:20 +0200674 PyErr_NoMemory();
675 return NULL;
676 }
Victor Stinner84def372011-12-11 20:04:56 +0100677 unicode = new_unicode;
Victor Stinnerfe226c02011-10-03 03:52:20 +0200678 _Py_NewReference(unicode);
Victor Stinner84def372011-12-11 20:04:56 +0100679
Victor Stinnerfe226c02011-10-03 03:52:20 +0200680 _PyUnicode_LENGTH(unicode) = length;
Victor Stinnerc379ead2011-10-03 12:52:27 +0200681 if (share_wstr) {
Victor Stinnerfe226c02011-10-03 03:52:20 +0200682 _PyUnicode_WSTR(unicode) = PyUnicode_DATA(unicode);
Victor Stinner488fa492011-12-12 00:01:39 +0100683 if (!PyUnicode_IS_ASCII(unicode))
Victor Stinnerc379ead2011-10-03 12:52:27 +0200684 _PyUnicode_WSTR_LENGTH(unicode) = length;
685 }
Victor Stinnerfe226c02011-10-03 03:52:20 +0200686 PyUnicode_WRITE(PyUnicode_KIND(unicode), PyUnicode_DATA(unicode),
687 length, 0);
Victor Stinner79891572012-05-03 13:43:07 +0200688 assert(_PyUnicode_CheckConsistency(unicode, 0));
Victor Stinnerfe226c02011-10-03 03:52:20 +0200689 return unicode;
690}
691
Alexander Belopolsky40018472011-02-26 01:02:56 +0000692static int
Victor Stinner9db1a8b2011-10-23 20:04:37 +0200693resize_inplace(PyObject *unicode, Py_ssize_t length)
Guido van Rossumd57fd912000-03-10 22:53:23 +0000694{
Victor Stinner95663112011-10-04 01:03:50 +0200695 wchar_t *wstr;
Victor Stinner7a9105a2011-12-12 00:13:42 +0100696 Py_ssize_t new_size;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200697 assert(!PyUnicode_IS_COMPACT(unicode));
Victor Stinnerfe226c02011-10-03 03:52:20 +0200698 assert(Py_REFCNT(unicode) == 1);
Tim Petersced69f82003-09-16 20:30:58 +0000699
Victor Stinnerfe226c02011-10-03 03:52:20 +0200700 if (PyUnicode_IS_READY(unicode)) {
701 Py_ssize_t char_size;
Victor Stinner1c8d0c72011-10-03 12:11:00 +0200702 int share_wstr, share_utf8;
Victor Stinnerfe226c02011-10-03 03:52:20 +0200703 void *data;
704
705 data = _PyUnicode_DATA_ANY(unicode);
Martin v. Löwisc47adb02011-10-07 20:55:35 +0200706 char_size = PyUnicode_KIND(unicode);
Victor Stinnerc379ead2011-10-03 12:52:27 +0200707 share_wstr = _PyUnicode_SHARE_WSTR(unicode);
708 share_utf8 = _PyUnicode_SHARE_UTF8(unicode);
Victor Stinnerfe226c02011-10-03 03:52:20 +0200709
710 if (length > (PY_SSIZE_T_MAX / char_size - 1)) {
711 PyErr_NoMemory();
712 return -1;
713 }
714 new_size = (length + 1) * char_size;
715
Victor Stinner7a9105a2011-12-12 00:13:42 +0100716 if (!share_utf8 && _PyUnicode_HAS_UTF8_MEMORY(unicode))
717 {
718 PyObject_DEL(_PyUnicode_UTF8(unicode));
719 _PyUnicode_UTF8(unicode) = NULL;
720 _PyUnicode_UTF8_LENGTH(unicode) = 0;
721 }
722
Victor Stinnerfe226c02011-10-03 03:52:20 +0200723 data = (PyObject *)PyObject_REALLOC(data, new_size);
724 if (data == NULL) {
725 PyErr_NoMemory();
726 return -1;
727 }
728 _PyUnicode_DATA_ANY(unicode) = data;
Victor Stinnerc379ead2011-10-03 12:52:27 +0200729 if (share_wstr) {
Victor Stinnerfe226c02011-10-03 03:52:20 +0200730 _PyUnicode_WSTR(unicode) = data;
Victor Stinnerc379ead2011-10-03 12:52:27 +0200731 _PyUnicode_WSTR_LENGTH(unicode) = length;
732 }
733 if (share_utf8) {
Victor Stinner1c8d0c72011-10-03 12:11:00 +0200734 _PyUnicode_UTF8(unicode) = data;
Victor Stinnerc379ead2011-10-03 12:52:27 +0200735 _PyUnicode_UTF8_LENGTH(unicode) = length;
736 }
Victor Stinnerfe226c02011-10-03 03:52:20 +0200737 _PyUnicode_LENGTH(unicode) = length;
738 PyUnicode_WRITE(PyUnicode_KIND(unicode), data, length, 0);
Victor Stinner95663112011-10-04 01:03:50 +0200739 if (share_wstr || _PyUnicode_WSTR(unicode) == NULL) {
Victor Stinnerbb10a1f2011-10-05 01:34:17 +0200740 assert(_PyUnicode_CheckConsistency(unicode, 0));
Victor Stinnerfe226c02011-10-03 03:52:20 +0200741 return 0;
Victor Stinnerfe226c02011-10-03 03:52:20 +0200742 }
Victor Stinnerfe226c02011-10-03 03:52:20 +0200743 }
Victor Stinner95663112011-10-04 01:03:50 +0200744 assert(_PyUnicode_WSTR(unicode) != NULL);
745
746 /* check for integer overflow */
747 if (length > PY_SSIZE_T_MAX / sizeof(wchar_t) - 1) {
748 PyErr_NoMemory();
749 return -1;
750 }
Victor Stinner7a9105a2011-12-12 00:13:42 +0100751 new_size = sizeof(wchar_t) * (length + 1);
Victor Stinner95663112011-10-04 01:03:50 +0200752 wstr = _PyUnicode_WSTR(unicode);
Victor Stinner7a9105a2011-12-12 00:13:42 +0100753 wstr = PyObject_REALLOC(wstr, new_size);
Victor Stinner95663112011-10-04 01:03:50 +0200754 if (!wstr) {
755 PyErr_NoMemory();
756 return -1;
757 }
758 _PyUnicode_WSTR(unicode) = wstr;
759 _PyUnicode_WSTR(unicode)[length] = 0;
760 _PyUnicode_WSTR_LENGTH(unicode) = length;
Victor Stinnerbb10a1f2011-10-05 01:34:17 +0200761 assert(_PyUnicode_CheckConsistency(unicode, 0));
Guido van Rossumd57fd912000-03-10 22:53:23 +0000762 return 0;
763}
764
Victor Stinnerfe226c02011-10-03 03:52:20 +0200765static PyObject*
766resize_copy(PyObject *unicode, Py_ssize_t length)
767{
768 Py_ssize_t copy_length;
Victor Stinner7a9105a2011-12-12 00:13:42 +0100769 if (_PyUnicode_KIND(unicode) != PyUnicode_WCHAR_KIND) {
Victor Stinnerfe226c02011-10-03 03:52:20 +0200770 PyObject *copy;
Victor Stinner7a9105a2011-12-12 00:13:42 +0100771
Benjamin Petersonbac79492012-01-14 13:34:47 -0500772 if (PyUnicode_READY(unicode) == -1)
Victor Stinner7a9105a2011-12-12 00:13:42 +0100773 return NULL;
Victor Stinnerfe226c02011-10-03 03:52:20 +0200774
775 copy = PyUnicode_New(length, PyUnicode_MAX_CHAR_VALUE(unicode));
776 if (copy == NULL)
777 return NULL;
778
779 copy_length = Py_MIN(length, PyUnicode_GET_LENGTH(unicode));
Victor Stinnerd3f08822012-05-29 12:57:52 +0200780 _PyUnicode_FastCopyCharacters(copy, 0, unicode, 0, copy_length);
Victor Stinnerfe226c02011-10-03 03:52:20 +0200781 return copy;
Victor Stinner8cfcbed2011-10-03 23:19:21 +0200782 }
783 else {
Victor Stinner9db1a8b2011-10-23 20:04:37 +0200784 PyObject *w;
Victor Stinner7a9105a2011-12-12 00:13:42 +0100785
Victor Stinner9db1a8b2011-10-23 20:04:37 +0200786 w = (PyObject*)_PyUnicode_New(length);
Victor Stinnerfe226c02011-10-03 03:52:20 +0200787 if (w == NULL)
788 return NULL;
789 copy_length = _PyUnicode_WSTR_LENGTH(unicode);
790 copy_length = Py_MIN(copy_length, length);
791 Py_UNICODE_COPY(_PyUnicode_WSTR(w), _PyUnicode_WSTR(unicode),
792 copy_length);
Victor Stinner9db1a8b2011-10-23 20:04:37 +0200793 return w;
Victor Stinnerfe226c02011-10-03 03:52:20 +0200794 }
795}
796
Guido van Rossumd57fd912000-03-10 22:53:23 +0000797/* We allocate one more byte to make sure the string is
Martin v. Löwis47383402007-08-15 07:32:56 +0000798 Ux0000 terminated; some code (e.g. new_identifier)
799 relies on that.
Guido van Rossumd57fd912000-03-10 22:53:23 +0000800
801 XXX This allocator could further be enhanced by assuring that the
Benjamin Peterson29060642009-01-31 22:14:21 +0000802 free list never reduces its size below 1.
Guido van Rossumd57fd912000-03-10 22:53:23 +0000803
804*/
805
Alexander Belopolsky40018472011-02-26 01:02:56 +0000806static PyUnicodeObject *
807_PyUnicode_New(Py_ssize_t length)
Guido van Rossumd57fd912000-03-10 22:53:23 +0000808{
809 register PyUnicodeObject *unicode;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200810 size_t new_size;
Guido van Rossumd57fd912000-03-10 22:53:23 +0000811
Thomas Wouters477c8d52006-05-27 19:21:47 +0000812 /* Optimization for empty strings */
Guido van Rossumd57fd912000-03-10 22:53:23 +0000813 if (length == 0 && unicode_empty != NULL) {
814 Py_INCREF(unicode_empty);
Victor Stinnera464fc12011-10-02 20:39:30 +0200815 return (PyUnicodeObject*)unicode_empty;
Guido van Rossumd57fd912000-03-10 22:53:23 +0000816 }
817
Neal Norwitz3ce5d922008-08-24 07:08:55 +0000818 /* Ensure we won't overflow the size. */
819 if (length > ((PY_SSIZE_T_MAX / sizeof(Py_UNICODE)) - 1)) {
820 return (PyUnicodeObject *)PyErr_NoMemory();
821 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200822 if (length < 0) {
823 PyErr_SetString(PyExc_SystemError,
824 "Negative size passed to _PyUnicode_New");
825 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +0000826 }
827
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200828 unicode = PyObject_New(PyUnicodeObject, &PyUnicode_Type);
829 if (unicode == NULL)
830 return NULL;
831 new_size = sizeof(Py_UNICODE) * ((size_t)length + 1);
832 _PyUnicode_WSTR(unicode) = (Py_UNICODE*) PyObject_MALLOC(new_size);
833 if (!_PyUnicode_WSTR(unicode)) {
Victor Stinnerb0a82a62011-12-12 13:08:33 +0100834 Py_DECREF(unicode);
Benjamin Peterson29060642009-01-31 22:14:21 +0000835 PyErr_NoMemory();
Victor Stinnerb0a82a62011-12-12 13:08:33 +0100836 return NULL;
Guido van Rossum3c1bb802000-04-27 20:13:50 +0000837 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200838
Jeremy Hyltond8082792003-09-16 19:41:39 +0000839 /* Initialize the first element to guard against cases where
Tim Petersced69f82003-09-16 20:30:58 +0000840 * the caller fails before initializing str -- unicode_resize()
841 * reads str[0], and the Keep-Alive optimization can keep memory
842 * allocated for str alive across a call to unicode_dealloc(unicode).
843 * We don't want unicode_resize to read uninitialized memory in
844 * that case.
845 */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200846 _PyUnicode_WSTR(unicode)[0] = 0;
847 _PyUnicode_WSTR(unicode)[length] = 0;
848 _PyUnicode_WSTR_LENGTH(unicode) = length;
849 _PyUnicode_HASH(unicode) = -1;
850 _PyUnicode_STATE(unicode).interned = 0;
851 _PyUnicode_STATE(unicode).kind = 0;
852 _PyUnicode_STATE(unicode).compact = 0;
853 _PyUnicode_STATE(unicode).ready = 0;
854 _PyUnicode_STATE(unicode).ascii = 0;
Victor Stinnerc3c74152011-10-02 20:39:55 +0200855 _PyUnicode_DATA_ANY(unicode) = NULL;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200856 _PyUnicode_LENGTH(unicode) = 0;
Victor Stinnere90fe6a2011-10-01 16:48:13 +0200857 _PyUnicode_UTF8(unicode) = NULL;
858 _PyUnicode_UTF8_LENGTH(unicode) = 0;
Victor Stinner7931d9a2011-11-04 00:22:48 +0100859 assert(_PyUnicode_CheckConsistency((PyObject *)unicode, 0));
Guido van Rossumd57fd912000-03-10 22:53:23 +0000860 return unicode;
861}
862
Victor Stinnerf42dc442011-10-02 23:33:16 +0200863static const char*
864unicode_kind_name(PyObject *unicode)
865{
Victor Stinner42dfd712011-10-03 14:41:45 +0200866 /* don't check consistency: unicode_kind_name() is called from
867 _PyUnicode_Dump() */
Victor Stinnerf42dc442011-10-02 23:33:16 +0200868 if (!PyUnicode_IS_COMPACT(unicode))
869 {
870 if (!PyUnicode_IS_READY(unicode))
871 return "wstr";
Benjamin Petersonead6b532011-12-20 17:23:42 -0600872 switch (PyUnicode_KIND(unicode))
Victor Stinnerf42dc442011-10-02 23:33:16 +0200873 {
874 case PyUnicode_1BYTE_KIND:
Victor Stinnera3b334d2011-10-03 13:53:37 +0200875 if (PyUnicode_IS_ASCII(unicode))
Victor Stinnerf42dc442011-10-02 23:33:16 +0200876 return "legacy ascii";
877 else
878 return "legacy latin1";
879 case PyUnicode_2BYTE_KIND:
880 return "legacy UCS2";
881 case PyUnicode_4BYTE_KIND:
882 return "legacy UCS4";
883 default:
884 return "<legacy invalid kind>";
885 }
886 }
887 assert(PyUnicode_IS_READY(unicode));
Benjamin Petersonead6b532011-12-20 17:23:42 -0600888 switch (PyUnicode_KIND(unicode)) {
Victor Stinnerf42dc442011-10-02 23:33:16 +0200889 case PyUnicode_1BYTE_KIND:
Victor Stinnera3b334d2011-10-03 13:53:37 +0200890 if (PyUnicode_IS_ASCII(unicode))
Victor Stinnerf42dc442011-10-02 23:33:16 +0200891 return "ascii";
892 else
Victor Stinnera3b334d2011-10-03 13:53:37 +0200893 return "latin1";
Victor Stinnerf42dc442011-10-02 23:33:16 +0200894 case PyUnicode_2BYTE_KIND:
Victor Stinnera3b334d2011-10-03 13:53:37 +0200895 return "UCS2";
Victor Stinnerf42dc442011-10-02 23:33:16 +0200896 case PyUnicode_4BYTE_KIND:
Victor Stinnera3b334d2011-10-03 13:53:37 +0200897 return "UCS4";
Victor Stinnerf42dc442011-10-02 23:33:16 +0200898 default:
899 return "<invalid compact kind>";
900 }
901}
902
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200903#ifdef Py_DEBUG
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200904/* Functions wrapping macros for use in debugger */
905char *_PyUnicode_utf8(void *unicode){
Victor Stinnere90fe6a2011-10-01 16:48:13 +0200906 return PyUnicode_UTF8(unicode);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200907}
908
909void *_PyUnicode_compact_data(void *unicode) {
910 return _PyUnicode_COMPACT_DATA(unicode);
911}
912void *_PyUnicode_data(void *unicode){
913 printf("obj %p\n", unicode);
914 printf("compact %d\n", PyUnicode_IS_COMPACT(unicode));
915 printf("compact ascii %d\n", PyUnicode_IS_COMPACT_ASCII(unicode));
916 printf("ascii op %p\n", ((void*)((PyASCIIObject*)(unicode) + 1)));
917 printf("compact op %p\n", ((void*)((PyCompactUnicodeObject*)(unicode) + 1)));
918 printf("compact data %p\n", _PyUnicode_COMPACT_DATA(unicode));
919 return PyUnicode_DATA(unicode);
920}
Victor Stinnerfe0c1552011-10-03 02:59:31 +0200921
922void
923_PyUnicode_Dump(PyObject *op)
924{
925 PyASCIIObject *ascii = (PyASCIIObject *)op;
Victor Stinnera849a4b2011-10-03 12:12:11 +0200926 PyCompactUnicodeObject *compact = (PyCompactUnicodeObject *)op;
927 PyUnicodeObject *unicode = (PyUnicodeObject *)op;
928 void *data;
Victor Stinner0d60e872011-10-23 19:47:19 +0200929
Victor Stinnera849a4b2011-10-03 12:12:11 +0200930 if (ascii->state.compact)
Victor Stinner0d60e872011-10-23 19:47:19 +0200931 {
932 if (ascii->state.ascii)
933 data = (ascii + 1);
934 else
935 data = (compact + 1);
936 }
Victor Stinnera849a4b2011-10-03 12:12:11 +0200937 else
938 data = unicode->data.any;
Victor Stinner0d60e872011-10-23 19:47:19 +0200939 printf("%s: len=%zu, ",unicode_kind_name(op), ascii->length);
940
Victor Stinnera849a4b2011-10-03 12:12:11 +0200941 if (ascii->wstr == data)
942 printf("shared ");
943 printf("wstr=%p", ascii->wstr);
Victor Stinner0d60e872011-10-23 19:47:19 +0200944
Victor Stinnera3b334d2011-10-03 13:53:37 +0200945 if (!(ascii->state.ascii == 1 && ascii->state.compact == 1)) {
Victor Stinnera849a4b2011-10-03 12:12:11 +0200946 printf(" (%zu), ", compact->wstr_length);
947 if (!ascii->state.compact && compact->utf8 == unicode->data.any)
948 printf("shared ");
949 printf("utf8=%p (%zu)", compact->utf8, compact->utf8_length);
Victor Stinnerfe0c1552011-10-03 02:59:31 +0200950 }
Victor Stinnera849a4b2011-10-03 12:12:11 +0200951 printf(", data=%p\n", data);
Victor Stinnerfe0c1552011-10-03 02:59:31 +0200952}
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200953#endif
954
955PyObject *
956PyUnicode_New(Py_ssize_t size, Py_UCS4 maxchar)
957{
958 PyObject *obj;
959 PyCompactUnicodeObject *unicode;
960 void *data;
Victor Stinner8f825062012-04-27 13:55:39 +0200961 enum PyUnicode_Kind kind;
Victor Stinner9e9d6892011-10-04 01:02:02 +0200962 int is_sharing, is_ascii;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200963 Py_ssize_t char_size;
964 Py_ssize_t struct_size;
965
966 /* Optimization for empty strings */
967 if (size == 0 && unicode_empty != NULL) {
968 Py_INCREF(unicode_empty);
Victor Stinnera464fc12011-10-02 20:39:30 +0200969 return unicode_empty;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200970 }
971
Victor Stinner9e9d6892011-10-04 01:02:02 +0200972 is_ascii = 0;
973 is_sharing = 0;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200974 struct_size = sizeof(PyCompactUnicodeObject);
975 if (maxchar < 128) {
Victor Stinner8f825062012-04-27 13:55:39 +0200976 kind = PyUnicode_1BYTE_KIND;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200977 char_size = 1;
978 is_ascii = 1;
979 struct_size = sizeof(PyASCIIObject);
980 }
981 else if (maxchar < 256) {
Victor Stinner8f825062012-04-27 13:55:39 +0200982 kind = PyUnicode_1BYTE_KIND;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200983 char_size = 1;
984 }
985 else if (maxchar < 65536) {
Victor Stinner8f825062012-04-27 13:55:39 +0200986 kind = PyUnicode_2BYTE_KIND;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200987 char_size = 2;
988 if (sizeof(wchar_t) == 2)
989 is_sharing = 1;
990 }
991 else {
Victor Stinnerc9590ad2012-03-04 01:34:37 +0100992 if (maxchar > MAX_UNICODE) {
993 PyErr_SetString(PyExc_SystemError,
994 "invalid maximum character passed to PyUnicode_New");
995 return NULL;
996 }
Victor Stinner8f825062012-04-27 13:55:39 +0200997 kind = PyUnicode_4BYTE_KIND;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +0200998 char_size = 4;
999 if (sizeof(wchar_t) == 4)
1000 is_sharing = 1;
1001 }
1002
1003 /* Ensure we won't overflow the size. */
1004 if (size < 0) {
1005 PyErr_SetString(PyExc_SystemError,
1006 "Negative size passed to PyUnicode_New");
1007 return NULL;
1008 }
1009 if (size > ((PY_SSIZE_T_MAX - struct_size) / char_size - 1))
1010 return PyErr_NoMemory();
1011
1012 /* Duplicated allocation code from _PyObject_New() instead of a call to
1013 * PyObject_New() so we are able to allocate space for the object and
1014 * it's data buffer.
1015 */
1016 obj = (PyObject *) PyObject_MALLOC(struct_size + (size + 1) * char_size);
1017 if (obj == NULL)
1018 return PyErr_NoMemory();
1019 obj = PyObject_INIT(obj, &PyUnicode_Type);
1020 if (obj == NULL)
1021 return NULL;
1022
1023 unicode = (PyCompactUnicodeObject *)obj;
1024 if (is_ascii)
1025 data = ((PyASCIIObject*)obj) + 1;
1026 else
1027 data = unicode + 1;
1028 _PyUnicode_LENGTH(unicode) = size;
1029 _PyUnicode_HASH(unicode) = -1;
1030 _PyUnicode_STATE(unicode).interned = 0;
Victor Stinner8f825062012-04-27 13:55:39 +02001031 _PyUnicode_STATE(unicode).kind = kind;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001032 _PyUnicode_STATE(unicode).compact = 1;
1033 _PyUnicode_STATE(unicode).ready = 1;
1034 _PyUnicode_STATE(unicode).ascii = is_ascii;
1035 if (is_ascii) {
1036 ((char*)data)[size] = 0;
1037 _PyUnicode_WSTR(unicode) = NULL;
1038 }
Victor Stinner8f825062012-04-27 13:55:39 +02001039 else if (kind == PyUnicode_1BYTE_KIND) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001040 ((char*)data)[size] = 0;
1041 _PyUnicode_WSTR(unicode) = NULL;
1042 _PyUnicode_WSTR_LENGTH(unicode) = 0;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001043 unicode->utf8 = NULL;
Victor Stinner9e9d6892011-10-04 01:02:02 +02001044 unicode->utf8_length = 0;
Victor Stinner8f825062012-04-27 13:55:39 +02001045 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001046 else {
1047 unicode->utf8 = NULL;
Victor Stinner9e9d6892011-10-04 01:02:02 +02001048 unicode->utf8_length = 0;
Victor Stinner8f825062012-04-27 13:55:39 +02001049 if (kind == PyUnicode_2BYTE_KIND)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001050 ((Py_UCS2*)data)[size] = 0;
Victor Stinner8f825062012-04-27 13:55:39 +02001051 else /* kind == PyUnicode_4BYTE_KIND */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001052 ((Py_UCS4*)data)[size] = 0;
1053 if (is_sharing) {
1054 _PyUnicode_WSTR_LENGTH(unicode) = size;
1055 _PyUnicode_WSTR(unicode) = (wchar_t *)data;
1056 }
1057 else {
1058 _PyUnicode_WSTR_LENGTH(unicode) = 0;
1059 _PyUnicode_WSTR(unicode) = NULL;
1060 }
1061 }
Victor Stinner8f825062012-04-27 13:55:39 +02001062#ifdef Py_DEBUG
1063 /* Fill the data with invalid characters to detect bugs earlier.
1064 _PyUnicode_CheckConsistency(str, 1) detects invalid characters,
1065 at least for ASCII and UCS-4 strings. U+00FF is invalid in ASCII
1066 and U+FFFFFFFF is an invalid character in Unicode 6.0. */
1067 memset(data, 0xff, size * kind);
1068#endif
Victor Stinner7931d9a2011-11-04 00:22:48 +01001069 assert(_PyUnicode_CheckConsistency((PyObject*)unicode, 0));
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001070 return obj;
1071}
1072
1073#if SIZEOF_WCHAR_T == 2
1074/* Helper function to convert a 16-bits wchar_t representation to UCS4, this
1075 will decode surrogate pairs, the other conversions are implemented as macros
Georg Brandl7597add2011-10-05 16:36:47 +02001076 for efficiency.
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001077
1078 This function assumes that unicode can hold one more code point than wstr
1079 characters for a terminating null character. */
Victor Stinnerc53be962011-10-02 21:33:54 +02001080static void
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001081unicode_convert_wchar_to_ucs4(const wchar_t *begin, const wchar_t *end,
Victor Stinner9db1a8b2011-10-23 20:04:37 +02001082 PyObject *unicode)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001083{
1084 const wchar_t *iter;
1085 Py_UCS4 *ucs4_out;
1086
Victor Stinner910337b2011-10-03 03:20:16 +02001087 assert(unicode != NULL);
1088 assert(_PyUnicode_CHECK(unicode));
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001089 assert(_PyUnicode_KIND(unicode) == PyUnicode_4BYTE_KIND);
1090 ucs4_out = PyUnicode_4BYTE_DATA(unicode);
1091
1092 for (iter = begin; iter < end; ) {
1093 assert(ucs4_out < (PyUnicode_4BYTE_DATA(unicode) +
1094 _PyUnicode_GET_LENGTH(unicode)));
Victor Stinner551ac952011-11-29 22:58:13 +01001095 if (Py_UNICODE_IS_HIGH_SURROGATE(iter[0])
1096 && (iter+1) < end
1097 && Py_UNICODE_IS_LOW_SURROGATE(iter[1]))
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001098 {
Victor Stinner551ac952011-11-29 22:58:13 +01001099 *ucs4_out++ = Py_UNICODE_JOIN_SURROGATES(iter[0], iter[1]);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001100 iter += 2;
1101 }
1102 else {
1103 *ucs4_out++ = *iter;
1104 iter++;
1105 }
1106 }
1107 assert(ucs4_out == (PyUnicode_4BYTE_DATA(unicode) +
1108 _PyUnicode_GET_LENGTH(unicode)));
1109
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001110}
1111#endif
1112
Victor Stinnercd9950f2011-10-02 00:34:53 +02001113static int
Victor Stinner488fa492011-12-12 00:01:39 +01001114unicode_check_modifiable(PyObject *unicode)
Victor Stinnercd9950f2011-10-02 00:34:53 +02001115{
Victor Stinner488fa492011-12-12 00:01:39 +01001116 if (!unicode_modifiable(unicode)) {
Victor Stinner01698042011-10-04 00:04:26 +02001117 PyErr_SetString(PyExc_SystemError,
Victor Stinner488fa492011-12-12 00:01:39 +01001118 "Cannot modify a string currently used");
Victor Stinnercd9950f2011-10-02 00:34:53 +02001119 return -1;
1120 }
Victor Stinnercd9950f2011-10-02 00:34:53 +02001121 return 0;
1122}
1123
Victor Stinnerfb9ea8c2011-10-06 01:45:57 +02001124static int
1125_copy_characters(PyObject *to, Py_ssize_t to_start,
1126 PyObject *from, Py_ssize_t from_start,
1127 Py_ssize_t how_many, int check_maxchar)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001128{
Victor Stinnera0702ab2011-09-29 14:14:38 +02001129 unsigned int from_kind, to_kind;
1130 void *from_data, *to_data;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001131
Victor Stinneree4544c2012-05-09 22:24:08 +02001132 assert(0 <= how_many);
1133 assert(0 <= from_start);
1134 assert(0 <= to_start);
Victor Stinnerfb9ea8c2011-10-06 01:45:57 +02001135 assert(PyUnicode_Check(from));
Victor Stinnerfb9ea8c2011-10-06 01:45:57 +02001136 assert(PyUnicode_IS_READY(from));
Victor Stinneree4544c2012-05-09 22:24:08 +02001137 assert(from_start + how_many <= PyUnicode_GET_LENGTH(from));
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001138
Victor Stinnerd3f08822012-05-29 12:57:52 +02001139 assert(PyUnicode_Check(to));
1140 assert(PyUnicode_IS_READY(to));
1141 assert(to_start + how_many <= PyUnicode_GET_LENGTH(to));
1142
Victor Stinnerc9d369f2012-06-16 02:22:37 +02001143 if (how_many == 0)
1144 return 0;
1145
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001146 from_kind = PyUnicode_KIND(from);
Victor Stinnera0702ab2011-09-29 14:14:38 +02001147 from_data = PyUnicode_DATA(from);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001148 to_kind = PyUnicode_KIND(to);
Victor Stinnera0702ab2011-09-29 14:14:38 +02001149 to_data = PyUnicode_DATA(to);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001150
Victor Stinnerf1852262012-06-16 16:38:26 +02001151#ifdef Py_DEBUG
1152 if (!check_maxchar
1153 && PyUnicode_MAX_CHAR_VALUE(from) > PyUnicode_MAX_CHAR_VALUE(to))
1154 {
1155 const Py_UCS4 to_maxchar = PyUnicode_MAX_CHAR_VALUE(to);
1156 Py_UCS4 ch;
1157 Py_ssize_t i;
1158 for (i=0; i < how_many; i++) {
1159 ch = PyUnicode_READ(from_kind, from_data, from_start + i);
1160 assert(ch <= to_maxchar);
1161 }
1162 }
1163#endif
1164
Victor Stinnerc9d369f2012-06-16 02:22:37 +02001165 if (from_kind == to_kind) {
Victor Stinnerf1852262012-06-16 16:38:26 +02001166 if (check_maxchar
1167 && !PyUnicode_IS_ASCII(from) && PyUnicode_IS_ASCII(to))
1168 {
Victor Stinnerc9d369f2012-06-16 02:22:37 +02001169 /* Writing Latin-1 characters into an ASCII string requires to
1170 check that all written characters are pure ASCII */
Victor Stinnerf1852262012-06-16 16:38:26 +02001171 Py_UCS4 max_char;
1172 max_char = ucs1lib_find_max_char(from_data,
1173 (Py_UCS1*)from_data + how_many);
1174 if (max_char >= 128)
1175 return -1;
Victor Stinnerc9d369f2012-06-16 02:22:37 +02001176 }
Martin v. Löwisc47adb02011-10-07 20:55:35 +02001177 Py_MEMCPY((char*)to_data + to_kind * to_start,
1178 (char*)from_data + from_kind * from_start,
1179 to_kind * how_many);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001180 }
Victor Stinnera0702ab2011-09-29 14:14:38 +02001181 else if (from_kind == PyUnicode_1BYTE_KIND
1182 && to_kind == PyUnicode_2BYTE_KIND)
Victor Stinnerbe78eaf2011-09-28 21:37:03 +02001183 {
1184 _PyUnicode_CONVERT_BYTES(
1185 Py_UCS1, Py_UCS2,
1186 PyUnicode_1BYTE_DATA(from) + from_start,
1187 PyUnicode_1BYTE_DATA(from) + from_start + how_many,
1188 PyUnicode_2BYTE_DATA(to) + to_start
1189 );
Victor Stinnerbe78eaf2011-09-28 21:37:03 +02001190 }
Victor Stinner157f83f2011-09-28 21:41:31 +02001191 else if (from_kind == PyUnicode_1BYTE_KIND
Victor Stinnerbe78eaf2011-09-28 21:37:03 +02001192 && to_kind == PyUnicode_4BYTE_KIND)
1193 {
1194 _PyUnicode_CONVERT_BYTES(
1195 Py_UCS1, Py_UCS4,
1196 PyUnicode_1BYTE_DATA(from) + from_start,
1197 PyUnicode_1BYTE_DATA(from) + from_start + how_many,
1198 PyUnicode_4BYTE_DATA(to) + to_start
1199 );
Victor Stinnerbe78eaf2011-09-28 21:37:03 +02001200 }
1201 else if (from_kind == PyUnicode_2BYTE_KIND
1202 && to_kind == PyUnicode_4BYTE_KIND)
1203 {
1204 _PyUnicode_CONVERT_BYTES(
1205 Py_UCS2, Py_UCS4,
1206 PyUnicode_2BYTE_DATA(from) + from_start,
1207 PyUnicode_2BYTE_DATA(from) + from_start + how_many,
1208 PyUnicode_4BYTE_DATA(to) + to_start
1209 );
Victor Stinnerbe78eaf2011-09-28 21:37:03 +02001210 }
Victor Stinnera0702ab2011-09-29 14:14:38 +02001211 else {
Victor Stinnerc9d369f2012-06-16 02:22:37 +02001212 assert (PyUnicode_MAX_CHAR_VALUE(from) > PyUnicode_MAX_CHAR_VALUE(to));
1213
Victor Stinnerc9d369f2012-06-16 02:22:37 +02001214 if (!check_maxchar) {
1215 if (from_kind == PyUnicode_2BYTE_KIND
1216 && to_kind == PyUnicode_1BYTE_KIND)
1217 {
1218 _PyUnicode_CONVERT_BYTES(
1219 Py_UCS2, Py_UCS1,
1220 PyUnicode_2BYTE_DATA(from) + from_start,
1221 PyUnicode_2BYTE_DATA(from) + from_start + how_many,
1222 PyUnicode_1BYTE_DATA(to) + to_start
1223 );
1224 }
1225 else if (from_kind == PyUnicode_4BYTE_KIND
1226 && to_kind == PyUnicode_1BYTE_KIND)
1227 {
1228 _PyUnicode_CONVERT_BYTES(
1229 Py_UCS4, Py_UCS1,
1230 PyUnicode_4BYTE_DATA(from) + from_start,
1231 PyUnicode_4BYTE_DATA(from) + from_start + how_many,
1232 PyUnicode_1BYTE_DATA(to) + to_start
1233 );
1234 }
1235 else if (from_kind == PyUnicode_4BYTE_KIND
1236 && to_kind == PyUnicode_2BYTE_KIND)
1237 {
1238 _PyUnicode_CONVERT_BYTES(
1239 Py_UCS4, Py_UCS2,
1240 PyUnicode_4BYTE_DATA(from) + from_start,
1241 PyUnicode_4BYTE_DATA(from) + from_start + how_many,
1242 PyUnicode_2BYTE_DATA(to) + to_start
1243 );
1244 }
1245 else {
1246 assert(0);
1247 return -1;
1248 }
1249 }
Victor Stinnerf1852262012-06-16 16:38:26 +02001250 else {
Victor Stinnera0702ab2011-09-29 14:14:38 +02001251 const Py_UCS4 to_maxchar = PyUnicode_MAX_CHAR_VALUE(to);
Victor Stinnerfb9ea8c2011-10-06 01:45:57 +02001252 Py_UCS4 ch;
Victor Stinnera0702ab2011-09-29 14:14:38 +02001253 Py_ssize_t i;
1254
Victor Stinnera0702ab2011-09-29 14:14:38 +02001255 for (i=0; i < how_many; i++) {
1256 ch = PyUnicode_READ(from_kind, from_data, from_start + i);
Victor Stinnerc9d369f2012-06-16 02:22:37 +02001257 if (ch > to_maxchar)
1258 return -1;
Victor Stinnera0702ab2011-09-29 14:14:38 +02001259 PyUnicode_WRITE(to_kind, to_data, to_start + i, ch);
1260 }
Victor Stinnera0702ab2011-09-29 14:14:38 +02001261 }
1262 }
Victor Stinnerfb9ea8c2011-10-06 01:45:57 +02001263 return 0;
1264}
1265
Victor Stinnerd3f08822012-05-29 12:57:52 +02001266void
1267_PyUnicode_FastCopyCharacters(
1268 PyObject *to, Py_ssize_t to_start,
1269 PyObject *from, Py_ssize_t from_start, Py_ssize_t how_many)
Victor Stinnerfb9ea8c2011-10-06 01:45:57 +02001270{
1271 (void)_copy_characters(to, to_start, from, from_start, how_many, 0);
1272}
1273
1274Py_ssize_t
1275PyUnicode_CopyCharacters(PyObject *to, Py_ssize_t to_start,
1276 PyObject *from, Py_ssize_t from_start,
1277 Py_ssize_t how_many)
1278{
1279 int err;
1280
1281 if (!PyUnicode_Check(from) || !PyUnicode_Check(to)) {
1282 PyErr_BadInternalCall();
1283 return -1;
1284 }
1285
Benjamin Petersonbac79492012-01-14 13:34:47 -05001286 if (PyUnicode_READY(from) == -1)
Victor Stinnerfb9ea8c2011-10-06 01:45:57 +02001287 return -1;
Benjamin Petersonbac79492012-01-14 13:34:47 -05001288 if (PyUnicode_READY(to) == -1)
Victor Stinnerfb9ea8c2011-10-06 01:45:57 +02001289 return -1;
1290
Victor Stinnerd3f08822012-05-29 12:57:52 +02001291 if (from_start < 0) {
1292 PyErr_SetString(PyExc_IndexError, "string index out of range");
1293 return -1;
1294 }
1295 if (to_start < 0) {
1296 PyErr_SetString(PyExc_IndexError, "string index out of range");
1297 return -1;
1298 }
Victor Stinnerfb9ea8c2011-10-06 01:45:57 +02001299 how_many = Py_MIN(PyUnicode_GET_LENGTH(from), how_many);
1300 if (to_start + how_many > PyUnicode_GET_LENGTH(to)) {
1301 PyErr_Format(PyExc_SystemError,
1302 "Cannot write %zi characters at %zi "
1303 "in a string of %zi characters",
1304 how_many, to_start, PyUnicode_GET_LENGTH(to));
1305 return -1;
1306 }
1307
1308 if (how_many == 0)
1309 return 0;
1310
Victor Stinner488fa492011-12-12 00:01:39 +01001311 if (unicode_check_modifiable(to))
Victor Stinnerfb9ea8c2011-10-06 01:45:57 +02001312 return -1;
1313
1314 err = _copy_characters(to, to_start, from, from_start, how_many, 1);
1315 if (err) {
1316 PyErr_Format(PyExc_SystemError,
1317 "Cannot copy %s characters "
1318 "into a string of %s characters",
1319 unicode_kind_name(from),
1320 unicode_kind_name(to));
1321 return -1;
1322 }
Victor Stinnera0702ab2011-09-29 14:14:38 +02001323 return how_many;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001324}
1325
Victor Stinner17222162011-09-28 22:15:37 +02001326/* Find the maximum code point and count the number of surrogate pairs so a
1327 correct string length can be computed before converting a string to UCS4.
1328 This function counts single surrogates as a character and not as a pair.
1329
1330 Return 0 on success, or -1 on error. */
1331static int
1332find_maxchar_surrogates(const wchar_t *begin, const wchar_t *end,
1333 Py_UCS4 *maxchar, Py_ssize_t *num_surrogates)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001334{
1335 const wchar_t *iter;
Victor Stinner8faf8212011-12-08 22:14:11 +01001336 Py_UCS4 ch;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001337
Victor Stinnerc53be962011-10-02 21:33:54 +02001338 assert(num_surrogates != NULL && maxchar != NULL);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001339 *num_surrogates = 0;
1340 *maxchar = 0;
1341
1342 for (iter = begin; iter < end; ) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001343#if SIZEOF_WCHAR_T == 2
Victor Stinnerca4f2072011-11-22 03:38:40 +01001344 if (Py_UNICODE_IS_HIGH_SURROGATE(iter[0])
1345 && (iter+1) < end
1346 && Py_UNICODE_IS_LOW_SURROGATE(iter[1]))
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001347 {
Victor Stinner8faf8212011-12-08 22:14:11 +01001348 ch = Py_UNICODE_JOIN_SURROGATES(iter[0], iter[1]);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001349 ++(*num_surrogates);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001350 iter += 2;
1351 }
1352 else
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001353#endif
Victor Stinner8faf8212011-12-08 22:14:11 +01001354 {
1355 ch = *iter;
1356 iter++;
1357 }
1358 if (ch > *maxchar) {
1359 *maxchar = ch;
1360 if (*maxchar > MAX_UNICODE) {
1361 PyErr_Format(PyExc_ValueError,
1362 "character U+%x is not in range [U+0000; U+10ffff]",
1363 ch);
1364 return -1;
1365 }
1366 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001367 }
1368 return 0;
1369}
1370
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01001371int
1372_PyUnicode_Ready(PyObject *unicode)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001373{
1374 wchar_t *end;
1375 Py_UCS4 maxchar = 0;
1376 Py_ssize_t num_surrogates;
1377#if SIZEOF_WCHAR_T == 2
1378 Py_ssize_t length_wo_surrogates;
1379#endif
1380
Georg Brandl7597add2011-10-05 16:36:47 +02001381 /* _PyUnicode_Ready() is only intended for old-style API usage where
Victor Stinnerd8f65102011-09-29 19:43:17 +02001382 strings were created using _PyObject_New() and where no canonical
1383 representation (the str field) has been set yet aka strings
1384 which are not yet ready. */
Victor Stinner910337b2011-10-03 03:20:16 +02001385 assert(_PyUnicode_CHECK(unicode));
1386 assert(_PyUnicode_KIND(unicode) == PyUnicode_WCHAR_KIND);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001387 assert(_PyUnicode_WSTR(unicode) != NULL);
Victor Stinnerc3c74152011-10-02 20:39:55 +02001388 assert(_PyUnicode_DATA_ANY(unicode) == NULL);
Victor Stinnere90fe6a2011-10-01 16:48:13 +02001389 assert(_PyUnicode_UTF8(unicode) == NULL);
Victor Stinnerd8f65102011-09-29 19:43:17 +02001390 /* Actually, it should neither be interned nor be anything else: */
1391 assert(_PyUnicode_STATE(unicode).interned == SSTATE_NOT_INTERNED);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001392
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001393 end = _PyUnicode_WSTR(unicode) + _PyUnicode_WSTR_LENGTH(unicode);
Victor Stinner17222162011-09-28 22:15:37 +02001394 if (find_maxchar_surrogates(_PyUnicode_WSTR(unicode), end,
Victor Stinnerd8f65102011-09-29 19:43:17 +02001395 &maxchar, &num_surrogates) == -1)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001396 return -1;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001397
1398 if (maxchar < 256) {
Victor Stinnerc3c74152011-10-02 20:39:55 +02001399 _PyUnicode_DATA_ANY(unicode) = PyObject_MALLOC(_PyUnicode_WSTR_LENGTH(unicode) + 1);
1400 if (!_PyUnicode_DATA_ANY(unicode)) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001401 PyErr_NoMemory();
1402 return -1;
1403 }
Victor Stinnerfb5f5f22011-09-28 21:39:49 +02001404 _PyUnicode_CONVERT_BYTES(wchar_t, unsigned char,
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001405 _PyUnicode_WSTR(unicode), end,
1406 PyUnicode_1BYTE_DATA(unicode));
1407 PyUnicode_1BYTE_DATA(unicode)[_PyUnicode_WSTR_LENGTH(unicode)] = '\0';
1408 _PyUnicode_LENGTH(unicode) = _PyUnicode_WSTR_LENGTH(unicode);
1409 _PyUnicode_STATE(unicode).kind = PyUnicode_1BYTE_KIND;
1410 if (maxchar < 128) {
Victor Stinnera3b334d2011-10-03 13:53:37 +02001411 _PyUnicode_STATE(unicode).ascii = 1;
Victor Stinnerc3c74152011-10-02 20:39:55 +02001412 _PyUnicode_UTF8(unicode) = _PyUnicode_DATA_ANY(unicode);
Victor Stinnere90fe6a2011-10-01 16:48:13 +02001413 _PyUnicode_UTF8_LENGTH(unicode) = _PyUnicode_WSTR_LENGTH(unicode);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001414 }
1415 else {
Victor Stinnera3b334d2011-10-03 13:53:37 +02001416 _PyUnicode_STATE(unicode).ascii = 0;
Victor Stinnere90fe6a2011-10-01 16:48:13 +02001417 _PyUnicode_UTF8(unicode) = NULL;
1418 _PyUnicode_UTF8_LENGTH(unicode) = 0;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001419 }
1420 PyObject_FREE(_PyUnicode_WSTR(unicode));
1421 _PyUnicode_WSTR(unicode) = NULL;
1422 _PyUnicode_WSTR_LENGTH(unicode) = 0;
1423 }
1424 /* In this case we might have to convert down from 4-byte native
1425 wchar_t to 2-byte unicode. */
1426 else if (maxchar < 65536) {
1427 assert(num_surrogates == 0 &&
1428 "FindMaxCharAndNumSurrogatePairs() messed up");
1429
Victor Stinner506f5922011-09-28 22:34:18 +02001430#if SIZEOF_WCHAR_T == 2
1431 /* We can share representations and are done. */
Victor Stinnerc3c74152011-10-02 20:39:55 +02001432 _PyUnicode_DATA_ANY(unicode) = _PyUnicode_WSTR(unicode);
Victor Stinner506f5922011-09-28 22:34:18 +02001433 PyUnicode_2BYTE_DATA(unicode)[_PyUnicode_WSTR_LENGTH(unicode)] = '\0';
1434 _PyUnicode_LENGTH(unicode) = _PyUnicode_WSTR_LENGTH(unicode);
1435 _PyUnicode_STATE(unicode).kind = PyUnicode_2BYTE_KIND;
Victor Stinnere90fe6a2011-10-01 16:48:13 +02001436 _PyUnicode_UTF8(unicode) = NULL;
1437 _PyUnicode_UTF8_LENGTH(unicode) = 0;
Victor Stinner506f5922011-09-28 22:34:18 +02001438#else
1439 /* sizeof(wchar_t) == 4 */
Victor Stinnerc3c74152011-10-02 20:39:55 +02001440 _PyUnicode_DATA_ANY(unicode) = PyObject_MALLOC(
Victor Stinner506f5922011-09-28 22:34:18 +02001441 2 * (_PyUnicode_WSTR_LENGTH(unicode) + 1));
Victor Stinnerc3c74152011-10-02 20:39:55 +02001442 if (!_PyUnicode_DATA_ANY(unicode)) {
Victor Stinner506f5922011-09-28 22:34:18 +02001443 PyErr_NoMemory();
1444 return -1;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001445 }
Victor Stinner506f5922011-09-28 22:34:18 +02001446 _PyUnicode_CONVERT_BYTES(wchar_t, Py_UCS2,
1447 _PyUnicode_WSTR(unicode), end,
1448 PyUnicode_2BYTE_DATA(unicode));
1449 PyUnicode_2BYTE_DATA(unicode)[_PyUnicode_WSTR_LENGTH(unicode)] = '\0';
1450 _PyUnicode_LENGTH(unicode) = _PyUnicode_WSTR_LENGTH(unicode);
1451 _PyUnicode_STATE(unicode).kind = PyUnicode_2BYTE_KIND;
Victor Stinnere90fe6a2011-10-01 16:48:13 +02001452 _PyUnicode_UTF8(unicode) = NULL;
1453 _PyUnicode_UTF8_LENGTH(unicode) = 0;
Victor Stinner506f5922011-09-28 22:34:18 +02001454 PyObject_FREE(_PyUnicode_WSTR(unicode));
1455 _PyUnicode_WSTR(unicode) = NULL;
1456 _PyUnicode_WSTR_LENGTH(unicode) = 0;
1457#endif
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001458 }
1459 /* maxchar exeeds 16 bit, wee need 4 bytes for unicode characters */
1460 else {
1461#if SIZEOF_WCHAR_T == 2
1462 /* in case the native representation is 2-bytes, we need to allocate a
1463 new normalized 4-byte version. */
1464 length_wo_surrogates = _PyUnicode_WSTR_LENGTH(unicode) - num_surrogates;
Victor Stinnerc3c74152011-10-02 20:39:55 +02001465 _PyUnicode_DATA_ANY(unicode) = PyObject_MALLOC(4 * (length_wo_surrogates + 1));
1466 if (!_PyUnicode_DATA_ANY(unicode)) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001467 PyErr_NoMemory();
1468 return -1;
1469 }
1470 _PyUnicode_LENGTH(unicode) = length_wo_surrogates;
1471 _PyUnicode_STATE(unicode).kind = PyUnicode_4BYTE_KIND;
Victor Stinnere90fe6a2011-10-01 16:48:13 +02001472 _PyUnicode_UTF8(unicode) = NULL;
1473 _PyUnicode_UTF8_LENGTH(unicode) = 0;
Victor Stinner126c5592011-10-03 04:17:10 +02001474 /* unicode_convert_wchar_to_ucs4() requires a ready string */
1475 _PyUnicode_STATE(unicode).ready = 1;
Victor Stinnerc53be962011-10-02 21:33:54 +02001476 unicode_convert_wchar_to_ucs4(_PyUnicode_WSTR(unicode), end, unicode);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001477 PyObject_FREE(_PyUnicode_WSTR(unicode));
1478 _PyUnicode_WSTR(unicode) = NULL;
1479 _PyUnicode_WSTR_LENGTH(unicode) = 0;
1480#else
1481 assert(num_surrogates == 0);
1482
Victor Stinnerc3c74152011-10-02 20:39:55 +02001483 _PyUnicode_DATA_ANY(unicode) = _PyUnicode_WSTR(unicode);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001484 _PyUnicode_LENGTH(unicode) = _PyUnicode_WSTR_LENGTH(unicode);
Victor Stinnere90fe6a2011-10-01 16:48:13 +02001485 _PyUnicode_UTF8(unicode) = NULL;
1486 _PyUnicode_UTF8_LENGTH(unicode) = 0;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001487 _PyUnicode_STATE(unicode).kind = PyUnicode_4BYTE_KIND;
1488#endif
1489 PyUnicode_4BYTE_DATA(unicode)[_PyUnicode_LENGTH(unicode)] = '\0';
1490 }
1491 _PyUnicode_STATE(unicode).ready = 1;
Victor Stinnerbb10a1f2011-10-05 01:34:17 +02001492 assert(_PyUnicode_CheckConsistency(unicode, 1));
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001493 return 0;
1494}
1495
Alexander Belopolsky40018472011-02-26 01:02:56 +00001496static void
Victor Stinner9db1a8b2011-10-23 20:04:37 +02001497unicode_dealloc(register PyObject *unicode)
Guido van Rossumd57fd912000-03-10 22:53:23 +00001498{
Walter Dörwald16807132007-05-25 13:52:07 +00001499 switch (PyUnicode_CHECK_INTERNED(unicode)) {
Benjamin Peterson29060642009-01-31 22:14:21 +00001500 case SSTATE_NOT_INTERNED:
1501 break;
Walter Dörwald16807132007-05-25 13:52:07 +00001502
Benjamin Peterson29060642009-01-31 22:14:21 +00001503 case SSTATE_INTERNED_MORTAL:
1504 /* revive dead object temporarily for DelItem */
1505 Py_REFCNT(unicode) = 3;
Victor Stinner7931d9a2011-11-04 00:22:48 +01001506 if (PyDict_DelItem(interned, unicode) != 0)
Benjamin Peterson29060642009-01-31 22:14:21 +00001507 Py_FatalError(
1508 "deletion of interned string failed");
1509 break;
Walter Dörwald16807132007-05-25 13:52:07 +00001510
Benjamin Peterson29060642009-01-31 22:14:21 +00001511 case SSTATE_INTERNED_IMMORTAL:
1512 Py_FatalError("Immortal interned string died.");
Walter Dörwald16807132007-05-25 13:52:07 +00001513
Benjamin Peterson29060642009-01-31 22:14:21 +00001514 default:
1515 Py_FatalError("Inconsistent interned string state.");
Walter Dörwald16807132007-05-25 13:52:07 +00001516 }
1517
Victor Stinner03490912011-10-03 23:45:12 +02001518 if (_PyUnicode_HAS_WSTR_MEMORY(unicode))
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001519 PyObject_DEL(_PyUnicode_WSTR(unicode));
Victor Stinner829c0ad2011-10-03 01:08:02 +02001520 if (_PyUnicode_HAS_UTF8_MEMORY(unicode))
Victor Stinnere90fe6a2011-10-01 16:48:13 +02001521 PyObject_DEL(_PyUnicode_UTF8(unicode));
Victor Stinnerb0a82a62011-12-12 13:08:33 +01001522 if (!PyUnicode_IS_COMPACT(unicode) && _PyUnicode_DATA_ANY(unicode))
1523 PyObject_DEL(_PyUnicode_DATA_ANY(unicode));
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001524
Victor Stinnerb0a82a62011-12-12 13:08:33 +01001525 Py_TYPE(unicode)->tp_free(unicode);
Guido van Rossumd57fd912000-03-10 22:53:23 +00001526}
1527
Victor Stinnerfb9ea8c2011-10-06 01:45:57 +02001528#ifdef Py_DEBUG
1529static int
1530unicode_is_singleton(PyObject *unicode)
1531{
1532 PyASCIIObject *ascii = (PyASCIIObject *)unicode;
1533 if (unicode == unicode_empty)
1534 return 1;
1535 if (ascii->state.kind != PyUnicode_WCHAR_KIND && ascii->length == 1)
1536 {
1537 Py_UCS4 ch = PyUnicode_READ_CHAR(unicode, 0);
1538 if (ch < 256 && unicode_latin1[ch] == unicode)
1539 return 1;
1540 }
1541 return 0;
1542}
1543#endif
1544
Alexander Belopolsky40018472011-02-26 01:02:56 +00001545static int
Victor Stinner488fa492011-12-12 00:01:39 +01001546unicode_modifiable(PyObject *unicode)
Marc-André Lemburg8155e0e2001-04-23 14:44:21 +00001547{
Victor Stinner488fa492011-12-12 00:01:39 +01001548 assert(_PyUnicode_CHECK(unicode));
Victor Stinnerfe226c02011-10-03 03:52:20 +02001549 if (Py_REFCNT(unicode) != 1)
1550 return 0;
Victor Stinner488fa492011-12-12 00:01:39 +01001551 if (_PyUnicode_HASH(unicode) != -1)
1552 return 0;
Victor Stinnerfe226c02011-10-03 03:52:20 +02001553 if (PyUnicode_CHECK_INTERNED(unicode))
1554 return 0;
Victor Stinner488fa492011-12-12 00:01:39 +01001555 if (!PyUnicode_CheckExact(unicode))
1556 return 0;
Victor Stinner77bb47b2011-10-03 20:06:05 +02001557#ifdef Py_DEBUG
Victor Stinnerfb9ea8c2011-10-06 01:45:57 +02001558 /* singleton refcount is greater than 1 */
1559 assert(!unicode_is_singleton(unicode));
Victor Stinner77bb47b2011-10-03 20:06:05 +02001560#endif
Victor Stinnerfe226c02011-10-03 03:52:20 +02001561 return 1;
1562}
Marc-André Lemburg8155e0e2001-04-23 14:44:21 +00001563
Victor Stinnerfe226c02011-10-03 03:52:20 +02001564static int
1565unicode_resize(PyObject **p_unicode, Py_ssize_t length)
1566{
1567 PyObject *unicode;
1568 Py_ssize_t old_length;
1569
1570 assert(p_unicode != NULL);
1571 unicode = *p_unicode;
1572
1573 assert(unicode != NULL);
1574 assert(PyUnicode_Check(unicode));
1575 assert(0 <= length);
1576
Victor Stinner910337b2011-10-03 03:20:16 +02001577 if (_PyUnicode_KIND(unicode) == PyUnicode_WCHAR_KIND)
Victor Stinnerfe226c02011-10-03 03:52:20 +02001578 old_length = PyUnicode_WSTR_LENGTH(unicode);
1579 else
1580 old_length = PyUnicode_GET_LENGTH(unicode);
1581 if (old_length == length)
1582 return 0;
1583
Martin v. Löwise9b11c12011-11-08 17:35:34 +01001584 if (length == 0) {
1585 Py_DECREF(*p_unicode);
1586 *p_unicode = unicode_empty;
1587 Py_INCREF(*p_unicode);
1588 return 0;
1589 }
1590
Victor Stinner488fa492011-12-12 00:01:39 +01001591 if (!unicode_modifiable(unicode)) {
Victor Stinnerfe226c02011-10-03 03:52:20 +02001592 PyObject *copy = resize_copy(unicode, length);
1593 if (copy == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00001594 return -1;
Victor Stinnerfe226c02011-10-03 03:52:20 +02001595 Py_DECREF(*p_unicode);
1596 *p_unicode = copy;
Benjamin Peterson29060642009-01-31 22:14:21 +00001597 return 0;
Marc-André Lemburg8155e0e2001-04-23 14:44:21 +00001598 }
1599
Victor Stinnerfe226c02011-10-03 03:52:20 +02001600 if (PyUnicode_IS_COMPACT(unicode)) {
Victor Stinnerb0a82a62011-12-12 13:08:33 +01001601 PyObject *new_unicode = resize_compact(unicode, length);
1602 if (new_unicode == NULL)
Victor Stinnerfe226c02011-10-03 03:52:20 +02001603 return -1;
Victor Stinnerb0a82a62011-12-12 13:08:33 +01001604 *p_unicode = new_unicode;
Victor Stinnerfe226c02011-10-03 03:52:20 +02001605 return 0;
Benjamin Peterson4bfce8f2011-10-03 19:35:07 -04001606 }
Victor Stinner9db1a8b2011-10-23 20:04:37 +02001607 return resize_inplace(unicode, length);
Marc-André Lemburg8155e0e2001-04-23 14:44:21 +00001608}
1609
Alexander Belopolsky40018472011-02-26 01:02:56 +00001610int
Victor Stinnerfe226c02011-10-03 03:52:20 +02001611PyUnicode_Resize(PyObject **p_unicode, Py_ssize_t length)
Alexandre Vassalottiaa0e5312008-12-27 06:43:58 +00001612{
Victor Stinnerfe226c02011-10-03 03:52:20 +02001613 PyObject *unicode;
1614 if (p_unicode == NULL) {
1615 PyErr_BadInternalCall();
1616 return -1;
1617 }
1618 unicode = *p_unicode;
Martin v. Löwise9b11c12011-11-08 17:35:34 +01001619 if (unicode == NULL || !PyUnicode_Check(unicode) || length < 0)
Victor Stinnerfe226c02011-10-03 03:52:20 +02001620 {
1621 PyErr_BadInternalCall();
1622 return -1;
1623 }
1624 return unicode_resize(p_unicode, length);
Alexandre Vassalottiaa0e5312008-12-27 06:43:58 +00001625}
Marc-André Lemburg8155e0e2001-04-23 14:44:21 +00001626
Martin v. Löwise9b11c12011-11-08 17:35:34 +01001627static int
Victor Stinner1b487b42012-05-03 12:29:04 +02001628unicode_widen(PyObject **p_unicode, Py_ssize_t length,
1629 unsigned int maxchar)
Martin v. Löwise9b11c12011-11-08 17:35:34 +01001630{
1631 PyObject *result;
1632 assert(PyUnicode_IS_READY(*p_unicode));
Victor Stinner1b487b42012-05-03 12:29:04 +02001633 assert(length <= PyUnicode_GET_LENGTH(*p_unicode));
Martin v. Löwise9b11c12011-11-08 17:35:34 +01001634 if (maxchar <= PyUnicode_MAX_CHAR_VALUE(*p_unicode))
1635 return 0;
1636 result = PyUnicode_New(PyUnicode_GET_LENGTH(*p_unicode),
1637 maxchar);
1638 if (result == NULL)
1639 return -1;
Victor Stinnerd3f08822012-05-29 12:57:52 +02001640 _PyUnicode_FastCopyCharacters(result, 0, *p_unicode, 0, length);
Martin v. Löwise9b11c12011-11-08 17:35:34 +01001641 Py_DECREF(*p_unicode);
1642 *p_unicode = result;
1643 return 0;
1644}
1645
1646static int
1647unicode_putchar(PyObject **p_unicode, Py_ssize_t *pos,
1648 Py_UCS4 ch)
1649{
Victor Stinner15e9ed22012-02-22 13:36:20 +01001650 assert(ch <= MAX_UNICODE);
Victor Stinner1b487b42012-05-03 12:29:04 +02001651 if (unicode_widen(p_unicode, *pos, ch) < 0)
Martin v. Löwise9b11c12011-11-08 17:35:34 +01001652 return -1;
1653 PyUnicode_WRITE(PyUnicode_KIND(*p_unicode),
1654 PyUnicode_DATA(*p_unicode),
1655 (*pos)++, ch);
1656 return 0;
1657}
1658
Victor Stinnerc5166102012-02-22 13:55:02 +01001659/* Copy a ASCII or latin1 char* string into a Python Unicode string.
Victor Stinnerc5166102012-02-22 13:55:02 +01001660
Victor Stinnerb429d3b2012-02-22 21:22:20 +01001661 WARNING: The function doesn't copy the terminating null character and
1662 doesn't check the maximum character (may write a latin1 character in an
1663 ASCII string). */
Victor Stinner184252a2012-06-16 02:57:41 +02001664static void
1665unicode_write_cstr(PyObject *unicode, Py_ssize_t index,
1666 const char *str, Py_ssize_t len)
Victor Stinnerc5166102012-02-22 13:55:02 +01001667{
1668 enum PyUnicode_Kind kind = PyUnicode_KIND(unicode);
1669 void *data = PyUnicode_DATA(unicode);
Victor Stinner184252a2012-06-16 02:57:41 +02001670 const char *end = str + len;
Victor Stinnerc5166102012-02-22 13:55:02 +01001671
1672 switch (kind) {
1673 case PyUnicode_1BYTE_KIND: {
Victor Stinnerc5166102012-02-22 13:55:02 +01001674 assert(index + len <= PyUnicode_GET_LENGTH(unicode));
Antoine Pitrouba6bafc2012-02-22 16:41:50 +01001675 memcpy((char *) data + index, str, len);
Victor Stinner184252a2012-06-16 02:57:41 +02001676 break;
Victor Stinnerc5166102012-02-22 13:55:02 +01001677 }
1678 case PyUnicode_2BYTE_KIND: {
1679 Py_UCS2 *start = (Py_UCS2 *)data + index;
1680 Py_UCS2 *ucs2 = start;
1681 assert(index <= PyUnicode_GET_LENGTH(unicode));
1682
Victor Stinner184252a2012-06-16 02:57:41 +02001683 for (; str < end; ++ucs2, ++str)
Victor Stinnerc5166102012-02-22 13:55:02 +01001684 *ucs2 = (Py_UCS2)*str;
1685
1686 assert((ucs2 - start) <= PyUnicode_GET_LENGTH(unicode));
Victor Stinner184252a2012-06-16 02:57:41 +02001687 break;
Victor Stinnerc5166102012-02-22 13:55:02 +01001688 }
1689 default: {
1690 Py_UCS4 *start = (Py_UCS4 *)data + index;
1691 Py_UCS4 *ucs4 = start;
1692 assert(kind == PyUnicode_4BYTE_KIND);
1693 assert(index <= PyUnicode_GET_LENGTH(unicode));
1694
Victor Stinner184252a2012-06-16 02:57:41 +02001695 for (; str < end; ++ucs4, ++str)
Victor Stinnerc5166102012-02-22 13:55:02 +01001696 *ucs4 = (Py_UCS4)*str;
1697
1698 assert((ucs4 - start) <= PyUnicode_GET_LENGTH(unicode));
Victor Stinnerc5166102012-02-22 13:55:02 +01001699 }
1700 }
1701}
1702
1703
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001704static PyObject*
1705get_latin1_char(unsigned char ch)
1706{
Victor Stinnera464fc12011-10-02 20:39:30 +02001707 PyObject *unicode = unicode_latin1[ch];
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001708 if (!unicode) {
Victor Stinnera464fc12011-10-02 20:39:30 +02001709 unicode = PyUnicode_New(1, ch);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001710 if (!unicode)
1711 return NULL;
1712 PyUnicode_1BYTE_DATA(unicode)[0] = ch;
Victor Stinnerbb10a1f2011-10-05 01:34:17 +02001713 assert(_PyUnicode_CheckConsistency(unicode, 1));
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001714 unicode_latin1[ch] = unicode;
1715 }
1716 Py_INCREF(unicode);
Victor Stinnera464fc12011-10-02 20:39:30 +02001717 return unicode;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001718}
1719
Alexander Belopolsky40018472011-02-26 01:02:56 +00001720PyObject *
1721PyUnicode_FromUnicode(const Py_UNICODE *u, Py_ssize_t size)
Guido van Rossumd57fd912000-03-10 22:53:23 +00001722{
Victor Stinner9db1a8b2011-10-23 20:04:37 +02001723 PyObject *unicode;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001724 Py_UCS4 maxchar = 0;
1725 Py_ssize_t num_surrogates;
1726
1727 if (u == NULL)
1728 return (PyObject*)_PyUnicode_New(size);
Guido van Rossumd57fd912000-03-10 22:53:23 +00001729
Marc-André Lemburg8155e0e2001-04-23 14:44:21 +00001730 /* If the Unicode data is known at construction time, we can apply
1731 some optimizations which share commonly used objects. */
Marc-André Lemburg8155e0e2001-04-23 14:44:21 +00001732
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001733 /* Optimization for empty strings */
1734 if (size == 0 && unicode_empty != NULL) {
1735 Py_INCREF(unicode_empty);
Victor Stinnera464fc12011-10-02 20:39:30 +02001736 return unicode_empty;
Marc-André Lemburg8155e0e2001-04-23 14:44:21 +00001737 }
Tim Petersced69f82003-09-16 20:30:58 +00001738
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001739 /* Single character Unicode objects in the Latin-1 range are
1740 shared when using this constructor */
1741 if (size == 1 && *u < 256)
1742 return get_latin1_char((unsigned char)*u);
1743
1744 /* If not empty and not single character, copy the Unicode data
1745 into the new object */
Victor Stinnerd8f65102011-09-29 19:43:17 +02001746 if (find_maxchar_surrogates(u, u + size,
1747 &maxchar, &num_surrogates) == -1)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001748 return NULL;
1749
Victor Stinner8faf8212011-12-08 22:14:11 +01001750 unicode = PyUnicode_New(size - num_surrogates, maxchar);
Guido van Rossumd57fd912000-03-10 22:53:23 +00001751 if (!unicode)
1752 return NULL;
1753
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001754 switch (PyUnicode_KIND(unicode)) {
1755 case PyUnicode_1BYTE_KIND:
Victor Stinnerfb5f5f22011-09-28 21:39:49 +02001756 _PyUnicode_CONVERT_BYTES(Py_UNICODE, unsigned char,
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001757 u, u + size, PyUnicode_1BYTE_DATA(unicode));
1758 break;
1759 case PyUnicode_2BYTE_KIND:
1760#if Py_UNICODE_SIZE == 2
1761 Py_MEMCPY(PyUnicode_2BYTE_DATA(unicode), u, size * 2);
1762#else
Victor Stinnerfb5f5f22011-09-28 21:39:49 +02001763 _PyUnicode_CONVERT_BYTES(Py_UNICODE, Py_UCS2,
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001764 u, u + size, PyUnicode_2BYTE_DATA(unicode));
1765#endif
1766 break;
1767 case PyUnicode_4BYTE_KIND:
1768#if SIZEOF_WCHAR_T == 2
1769 /* This is the only case which has to process surrogates, thus
1770 a simple copy loop is not enough and we need a function. */
Victor Stinnerc53be962011-10-02 21:33:54 +02001771 unicode_convert_wchar_to_ucs4(u, u + size, unicode);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001772#else
1773 assert(num_surrogates == 0);
1774 Py_MEMCPY(PyUnicode_4BYTE_DATA(unicode), u, size * 4);
1775#endif
1776 break;
1777 default:
1778 assert(0 && "Impossible state");
1779 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00001780
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01001781 return unicode_result(unicode);
Guido van Rossumd57fd912000-03-10 22:53:23 +00001782}
1783
Alexander Belopolsky40018472011-02-26 01:02:56 +00001784PyObject *
1785PyUnicode_FromStringAndSize(const char *u, Py_ssize_t size)
Walter Dörwaldacaa5a12007-05-05 12:00:46 +00001786{
Benjamin Peterson14339b62009-01-31 16:36:08 +00001787 if (size < 0) {
1788 PyErr_SetString(PyExc_SystemError,
Benjamin Peterson29060642009-01-31 22:14:21 +00001789 "Negative size passed to PyUnicode_FromStringAndSize");
Benjamin Peterson14339b62009-01-31 16:36:08 +00001790 return NULL;
1791 }
Victor Stinnera1d12bb2011-12-11 21:53:09 +01001792 if (u != NULL)
1793 return PyUnicode_DecodeUTF8Stateful(u, size, NULL, NULL);
1794 else
1795 return (PyObject *)_PyUnicode_New(size);
Walter Dörwaldacaa5a12007-05-05 12:00:46 +00001796}
1797
Alexander Belopolsky40018472011-02-26 01:02:56 +00001798PyObject *
1799PyUnicode_FromString(const char *u)
Walter Dörwaldd2034312007-05-18 16:29:38 +00001800{
1801 size_t size = strlen(u);
1802 if (size > PY_SSIZE_T_MAX) {
1803 PyErr_SetString(PyExc_OverflowError, "input too long");
1804 return NULL;
1805 }
Victor Stinnera1d12bb2011-12-11 21:53:09 +01001806 return PyUnicode_DecodeUTF8Stateful(u, (Py_ssize_t)size, NULL, NULL);
Walter Dörwaldd2034312007-05-18 16:29:38 +00001807}
1808
Martin v. Löwisafe55bb2011-10-09 10:38:36 +02001809PyObject *
1810_PyUnicode_FromId(_Py_Identifier *id)
1811{
1812 if (!id->object) {
Victor Stinnerd1cd99b2012-02-07 23:05:55 +01001813 id->object = PyUnicode_DecodeUTF8Stateful(id->string,
1814 strlen(id->string),
1815 NULL, NULL);
Martin v. Löwisafe55bb2011-10-09 10:38:36 +02001816 if (!id->object)
1817 return NULL;
1818 PyUnicode_InternInPlace(&id->object);
1819 assert(!id->next);
1820 id->next = static_strings;
1821 static_strings = id;
1822 }
Martin v. Löwisafe55bb2011-10-09 10:38:36 +02001823 return id->object;
1824}
1825
1826void
1827_PyUnicode_ClearStaticStrings()
1828{
1829 _Py_Identifier *i;
1830 for (i = static_strings; i; i = i->next) {
1831 Py_DECREF(i->object);
1832 i->object = NULL;
1833 i->next = NULL;
1834 }
1835}
1836
Benjamin Peterson0df54292012-03-26 14:50:32 -04001837/* Internal function, doesn't check maximum character */
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01001838
Victor Stinnerd3f08822012-05-29 12:57:52 +02001839PyObject*
1840_PyUnicode_FromASCII(const char *buffer, Py_ssize_t size)
Victor Stinner702c7342011-10-05 13:50:52 +02001841{
Victor Stinnerd3f08822012-05-29 12:57:52 +02001842 const unsigned char *s = (const unsigned char *)buffer;
Victor Stinner785938e2011-12-11 20:09:03 +01001843 PyObject *unicode;
Victor Stinnere6b2d442011-12-11 21:54:30 +01001844 if (size == 1) {
Victor Stinner0617b6e2011-10-05 23:26:01 +02001845#ifdef Py_DEBUG
Victor Stinnere6b2d442011-12-11 21:54:30 +01001846 assert(s[0] < 128);
Victor Stinner0617b6e2011-10-05 23:26:01 +02001847#endif
Antoine Pitrou7c46da72011-10-06 22:07:51 +02001848 return get_latin1_char(s[0]);
Victor Stinnere6b2d442011-12-11 21:54:30 +01001849 }
Victor Stinner785938e2011-12-11 20:09:03 +01001850 unicode = PyUnicode_New(size, 127);
1851 if (!unicode)
Victor Stinner702c7342011-10-05 13:50:52 +02001852 return NULL;
Victor Stinner785938e2011-12-11 20:09:03 +01001853 memcpy(PyUnicode_1BYTE_DATA(unicode), s, size);
1854 assert(_PyUnicode_CheckConsistency(unicode, 1));
1855 return unicode;
Victor Stinner702c7342011-10-05 13:50:52 +02001856}
1857
Victor Stinnerc80d6d22011-10-05 14:13:28 +02001858static Py_UCS4
1859kind_maxchar_limit(unsigned int kind)
1860{
Benjamin Petersonead6b532011-12-20 17:23:42 -06001861 switch (kind) {
Victor Stinnerc80d6d22011-10-05 14:13:28 +02001862 case PyUnicode_1BYTE_KIND:
1863 return 0x80;
1864 case PyUnicode_2BYTE_KIND:
1865 return 0x100;
1866 case PyUnicode_4BYTE_KIND:
1867 return 0x10000;
1868 default:
1869 assert(0 && "invalid kind");
Victor Stinner8faf8212011-12-08 22:14:11 +01001870 return MAX_UNICODE;
Victor Stinnerc80d6d22011-10-05 14:13:28 +02001871 }
1872}
1873
Victor Stinnere6abb482012-05-02 01:15:40 +02001874Py_LOCAL_INLINE(Py_UCS4)
1875align_maxchar(Py_UCS4 maxchar)
1876{
1877 if (maxchar <= 127)
1878 return 127;
1879 else if (maxchar <= 255)
1880 return 255;
1881 else if (maxchar <= 65535)
1882 return 65535;
1883 else
1884 return MAX_UNICODE;
1885}
1886
Victor Stinner702c7342011-10-05 13:50:52 +02001887static PyObject*
Victor Stinnere57b1c02011-09-28 22:20:48 +02001888_PyUnicode_FromUCS1(const unsigned char* u, Py_ssize_t size)
Mark Dickinson081dfee2009-03-18 14:47:41 +00001889{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001890 PyObject *res;
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01001891 unsigned char max_char;
Victor Stinnerb9275c12011-10-05 14:01:42 +02001892
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01001893 if (size == 0) {
1894 Py_INCREF(unicode_empty);
1895 return unicode_empty;
1896 }
1897 assert(size > 0);
Antoine Pitrou7c46da72011-10-06 22:07:51 +02001898 if (size == 1)
1899 return get_latin1_char(u[0]);
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01001900
Antoine Pitroudd4e2f02011-10-13 00:02:27 +02001901 max_char = ucs1lib_find_max_char(u, u + size);
Victor Stinnerb9275c12011-10-05 14:01:42 +02001902 res = PyUnicode_New(size, max_char);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001903 if (!res)
1904 return NULL;
1905 memcpy(PyUnicode_1BYTE_DATA(res), u, size);
Victor Stinnerbb10a1f2011-10-05 01:34:17 +02001906 assert(_PyUnicode_CheckConsistency(res, 1));
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001907 return res;
Mark Dickinson081dfee2009-03-18 14:47:41 +00001908}
1909
Victor Stinnere57b1c02011-09-28 22:20:48 +02001910static PyObject*
1911_PyUnicode_FromUCS2(const Py_UCS2 *u, Py_ssize_t size)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001912{
1913 PyObject *res;
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01001914 Py_UCS2 max_char;
Victor Stinnerb9275c12011-10-05 14:01:42 +02001915
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01001916 if (size == 0) {
1917 Py_INCREF(unicode_empty);
1918 return unicode_empty;
1919 }
1920 assert(size > 0);
Victor Stinnerb6cd0142012-05-03 02:17:04 +02001921 if (size == 1) {
1922 Py_UCS4 ch = u[0];
1923 if (ch < 256)
1924 return get_latin1_char((unsigned char)ch);
1925
1926 res = PyUnicode_New(1, ch);
1927 if (res == NULL)
1928 return NULL;
1929 PyUnicode_WRITE(PyUnicode_KIND(res), PyUnicode_DATA(res), 0, ch);
1930 assert(_PyUnicode_CheckConsistency(res, 1));
1931 return res;
1932 }
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01001933
Antoine Pitroudd4e2f02011-10-13 00:02:27 +02001934 max_char = ucs2lib_find_max_char(u, u + size);
Victor Stinnerb9275c12011-10-05 14:01:42 +02001935 res = PyUnicode_New(size, max_char);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001936 if (!res)
1937 return NULL;
Victor Stinnerb9275c12011-10-05 14:01:42 +02001938 if (max_char >= 256)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001939 memcpy(PyUnicode_2BYTE_DATA(res), u, sizeof(Py_UCS2)*size);
Antoine Pitroudd4e2f02011-10-13 00:02:27 +02001940 else {
1941 _PyUnicode_CONVERT_BYTES(
1942 Py_UCS2, Py_UCS1, u, u + size, PyUnicode_1BYTE_DATA(res));
1943 }
Victor Stinnerbb10a1f2011-10-05 01:34:17 +02001944 assert(_PyUnicode_CheckConsistency(res, 1));
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001945 return res;
1946}
1947
Victor Stinnere57b1c02011-09-28 22:20:48 +02001948static PyObject*
1949_PyUnicode_FromUCS4(const Py_UCS4 *u, Py_ssize_t size)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001950{
1951 PyObject *res;
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01001952 Py_UCS4 max_char;
Victor Stinnerb9275c12011-10-05 14:01:42 +02001953
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01001954 if (size == 0) {
1955 Py_INCREF(unicode_empty);
1956 return unicode_empty;
1957 }
1958 assert(size > 0);
Victor Stinnerb6cd0142012-05-03 02:17:04 +02001959 if (size == 1) {
1960 Py_UCS4 ch = u[0];
1961 if (ch < 256)
1962 return get_latin1_char((unsigned char)ch);
1963
1964 res = PyUnicode_New(1, ch);
1965 if (res == NULL)
1966 return NULL;
1967 PyUnicode_WRITE(PyUnicode_KIND(res), PyUnicode_DATA(res), 0, ch);
1968 assert(_PyUnicode_CheckConsistency(res, 1));
1969 return res;
1970 }
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01001971
Antoine Pitroudd4e2f02011-10-13 00:02:27 +02001972 max_char = ucs4lib_find_max_char(u, u + size);
Victor Stinnerb9275c12011-10-05 14:01:42 +02001973 res = PyUnicode_New(size, max_char);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001974 if (!res)
1975 return NULL;
Antoine Pitrou950468e2011-10-11 22:45:48 +02001976 if (max_char < 256)
1977 _PyUnicode_CONVERT_BYTES(Py_UCS4, Py_UCS1, u, u + size,
1978 PyUnicode_1BYTE_DATA(res));
1979 else if (max_char < 0x10000)
1980 _PyUnicode_CONVERT_BYTES(Py_UCS4, Py_UCS2, u, u + size,
1981 PyUnicode_2BYTE_DATA(res));
1982 else
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001983 memcpy(PyUnicode_4BYTE_DATA(res), u, sizeof(Py_UCS4)*size);
Victor Stinnerbb10a1f2011-10-05 01:34:17 +02001984 assert(_PyUnicode_CheckConsistency(res, 1));
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001985 return res;
1986}
1987
1988PyObject*
1989PyUnicode_FromKindAndData(int kind, const void *buffer, Py_ssize_t size)
1990{
Victor Stinnercfed46e2011-11-22 01:29:14 +01001991 if (size < 0) {
1992 PyErr_SetString(PyExc_ValueError, "size must be positive");
1993 return NULL;
1994 }
Benjamin Petersonead6b532011-12-20 17:23:42 -06001995 switch (kind) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001996 case PyUnicode_1BYTE_KIND:
Victor Stinnere57b1c02011-09-28 22:20:48 +02001997 return _PyUnicode_FromUCS1(buffer, size);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02001998 case PyUnicode_2BYTE_KIND:
Victor Stinnere57b1c02011-09-28 22:20:48 +02001999 return _PyUnicode_FromUCS2(buffer, size);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002000 case PyUnicode_4BYTE_KIND:
Victor Stinnere57b1c02011-09-28 22:20:48 +02002001 return _PyUnicode_FromUCS4(buffer, size);
Victor Stinnerb9275c12011-10-05 14:01:42 +02002002 default:
Victor Stinnerb9275c12011-10-05 14:01:42 +02002003 PyErr_SetString(PyExc_SystemError, "invalid kind");
2004 return NULL;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002005 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002006}
2007
Victor Stinnerece58de2012-04-23 23:36:38 +02002008Py_UCS4
2009_PyUnicode_FindMaxChar(PyObject *unicode, Py_ssize_t start, Py_ssize_t end)
2010{
2011 enum PyUnicode_Kind kind;
2012 void *startptr, *endptr;
2013
2014 assert(PyUnicode_IS_READY(unicode));
2015 assert(0 <= start);
2016 assert(end <= PyUnicode_GET_LENGTH(unicode));
2017 assert(start <= end);
2018
2019 if (start == 0 && end == PyUnicode_GET_LENGTH(unicode))
2020 return PyUnicode_MAX_CHAR_VALUE(unicode);
2021
2022 if (start == end)
2023 return 127;
2024
Victor Stinner94d558b2012-04-27 22:26:58 +02002025 if (PyUnicode_IS_ASCII(unicode))
2026 return 127;
2027
Victor Stinnerece58de2012-04-23 23:36:38 +02002028 kind = PyUnicode_KIND(unicode);
Benjamin Petersonf3b7d862012-04-23 18:07:01 -04002029 startptr = PyUnicode_DATA(unicode);
Benjamin Petersonb9f4c9d2012-04-23 21:45:40 -04002030 endptr = (char *)startptr + end * kind;
2031 startptr = (char *)startptr + start * kind;
Benjamin Peterson2844a7a2012-04-23 18:00:25 -04002032 switch(kind) {
2033 case PyUnicode_1BYTE_KIND:
2034 return ucs1lib_find_max_char(startptr, endptr);
2035 case PyUnicode_2BYTE_KIND:
2036 return ucs2lib_find_max_char(startptr, endptr);
2037 case PyUnicode_4BYTE_KIND:
2038 return ucs4lib_find_max_char(startptr, endptr);
Victor Stinnerece58de2012-04-23 23:36:38 +02002039 default:
Benjamin Peterson2844a7a2012-04-23 18:00:25 -04002040 assert(0);
2041 return 0;
Victor Stinnerece58de2012-04-23 23:36:38 +02002042 }
2043}
2044
Victor Stinner25a4b292011-10-06 12:31:55 +02002045/* Ensure that a string uses the most efficient storage, if it is not the
2046 case: create a new string with of the right kind. Write NULL into *p_unicode
2047 on error. */
Antoine Pitrou53bb5482011-10-10 23:49:24 +02002048static void
Victor Stinner25a4b292011-10-06 12:31:55 +02002049unicode_adjust_maxchar(PyObject **p_unicode)
2050{
2051 PyObject *unicode, *copy;
2052 Py_UCS4 max_char;
Antoine Pitroudd4e2f02011-10-13 00:02:27 +02002053 Py_ssize_t len;
Victor Stinner25a4b292011-10-06 12:31:55 +02002054 unsigned int kind;
2055
2056 assert(p_unicode != NULL);
2057 unicode = *p_unicode;
2058 assert(PyUnicode_IS_READY(unicode));
2059 if (PyUnicode_IS_ASCII(unicode))
2060 return;
2061
2062 len = PyUnicode_GET_LENGTH(unicode);
2063 kind = PyUnicode_KIND(unicode);
2064 if (kind == PyUnicode_1BYTE_KIND) {
2065 const Py_UCS1 *u = PyUnicode_1BYTE_DATA(unicode);
Antoine Pitroudd4e2f02011-10-13 00:02:27 +02002066 max_char = ucs1lib_find_max_char(u, u + len);
2067 if (max_char >= 128)
2068 return;
Victor Stinner25a4b292011-10-06 12:31:55 +02002069 }
2070 else if (kind == PyUnicode_2BYTE_KIND) {
2071 const Py_UCS2 *u = PyUnicode_2BYTE_DATA(unicode);
Antoine Pitroudd4e2f02011-10-13 00:02:27 +02002072 max_char = ucs2lib_find_max_char(u, u + len);
2073 if (max_char >= 256)
2074 return;
Victor Stinner25a4b292011-10-06 12:31:55 +02002075 }
2076 else {
Antoine Pitroudd4e2f02011-10-13 00:02:27 +02002077 const Py_UCS4 *u = PyUnicode_4BYTE_DATA(unicode);
Victor Stinner25a4b292011-10-06 12:31:55 +02002078 assert(kind == PyUnicode_4BYTE_KIND);
Antoine Pitroudd4e2f02011-10-13 00:02:27 +02002079 max_char = ucs4lib_find_max_char(u, u + len);
2080 if (max_char >= 0x10000)
2081 return;
Victor Stinner25a4b292011-10-06 12:31:55 +02002082 }
Victor Stinner25a4b292011-10-06 12:31:55 +02002083 copy = PyUnicode_New(len, max_char);
Victor Stinnerca439ee2012-06-16 03:17:34 +02002084 if (copy != NULL)
2085 _PyUnicode_FastCopyCharacters(copy, 0, unicode, 0, len);
Victor Stinner25a4b292011-10-06 12:31:55 +02002086 Py_DECREF(unicode);
2087 *p_unicode = copy;
2088}
2089
Victor Stinner034f6cf2011-09-30 02:26:44 +02002090PyObject*
Victor Stinnerbf6e5602011-12-12 01:53:47 +01002091_PyUnicode_Copy(PyObject *unicode)
Victor Stinner034f6cf2011-09-30 02:26:44 +02002092{
Victor Stinner87af4f22011-11-21 23:03:47 +01002093 Py_ssize_t length;
Victor Stinnerc841e7d2011-10-01 01:34:32 +02002094 PyObject *copy;
Victor Stinnerc841e7d2011-10-01 01:34:32 +02002095
Victor Stinner034f6cf2011-09-30 02:26:44 +02002096 if (!PyUnicode_Check(unicode)) {
2097 PyErr_BadInternalCall();
2098 return NULL;
2099 }
Benjamin Petersonbac79492012-01-14 13:34:47 -05002100 if (PyUnicode_READY(unicode) == -1)
Victor Stinner034f6cf2011-09-30 02:26:44 +02002101 return NULL;
Victor Stinnerc841e7d2011-10-01 01:34:32 +02002102
Victor Stinner87af4f22011-11-21 23:03:47 +01002103 length = PyUnicode_GET_LENGTH(unicode);
2104 copy = PyUnicode_New(length, PyUnicode_MAX_CHAR_VALUE(unicode));
Victor Stinnerc841e7d2011-10-01 01:34:32 +02002105 if (!copy)
2106 return NULL;
2107 assert(PyUnicode_KIND(copy) == PyUnicode_KIND(unicode));
2108
Victor Stinner87af4f22011-11-21 23:03:47 +01002109 Py_MEMCPY(PyUnicode_DATA(copy), PyUnicode_DATA(unicode),
2110 length * PyUnicode_KIND(unicode));
Victor Stinnerbb10a1f2011-10-05 01:34:17 +02002111 assert(_PyUnicode_CheckConsistency(copy, 1));
Victor Stinnerc841e7d2011-10-01 01:34:32 +02002112 return copy;
Victor Stinner034f6cf2011-09-30 02:26:44 +02002113}
2114
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002115
Victor Stinnerbc603d12011-10-02 01:00:40 +02002116/* Widen Unicode objects to larger buffers. Don't write terminating null
2117 character. Return NULL on error. */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002118
2119void*
2120_PyUnicode_AsKind(PyObject *s, unsigned int kind)
2121{
Victor Stinnerbc603d12011-10-02 01:00:40 +02002122 Py_ssize_t len;
2123 void *result;
2124 unsigned int skind;
2125
Benjamin Petersonbac79492012-01-14 13:34:47 -05002126 if (PyUnicode_READY(s) == -1)
Victor Stinnerbc603d12011-10-02 01:00:40 +02002127 return NULL;
2128
2129 len = PyUnicode_GET_LENGTH(s);
2130 skind = PyUnicode_KIND(s);
2131 if (skind >= kind) {
Victor Stinner01698042011-10-04 00:04:26 +02002132 PyErr_SetString(PyExc_SystemError, "invalid widening attempt");
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002133 return NULL;
2134 }
Benjamin Petersonead6b532011-12-20 17:23:42 -06002135 switch (kind) {
Victor Stinnerbc603d12011-10-02 01:00:40 +02002136 case PyUnicode_2BYTE_KIND:
2137 result = PyMem_Malloc(len * sizeof(Py_UCS2));
2138 if (!result)
2139 return PyErr_NoMemory();
2140 assert(skind == PyUnicode_1BYTE_KIND);
2141 _PyUnicode_CONVERT_BYTES(
2142 Py_UCS1, Py_UCS2,
2143 PyUnicode_1BYTE_DATA(s),
2144 PyUnicode_1BYTE_DATA(s) + len,
2145 result);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002146 return result;
Victor Stinnerbc603d12011-10-02 01:00:40 +02002147 case PyUnicode_4BYTE_KIND:
2148 result = PyMem_Malloc(len * sizeof(Py_UCS4));
2149 if (!result)
2150 return PyErr_NoMemory();
2151 if (skind == PyUnicode_2BYTE_KIND) {
2152 _PyUnicode_CONVERT_BYTES(
2153 Py_UCS2, Py_UCS4,
2154 PyUnicode_2BYTE_DATA(s),
2155 PyUnicode_2BYTE_DATA(s) + len,
2156 result);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002157 }
Victor Stinnerbc603d12011-10-02 01:00:40 +02002158 else {
2159 assert(skind == PyUnicode_1BYTE_KIND);
2160 _PyUnicode_CONVERT_BYTES(
2161 Py_UCS1, Py_UCS4,
2162 PyUnicode_1BYTE_DATA(s),
2163 PyUnicode_1BYTE_DATA(s) + len,
2164 result);
2165 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002166 return result;
Victor Stinnerbc603d12011-10-02 01:00:40 +02002167 default:
2168 break;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002169 }
Victor Stinner01698042011-10-04 00:04:26 +02002170 PyErr_SetString(PyExc_SystemError, "invalid kind");
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002171 return NULL;
2172}
2173
2174static Py_UCS4*
2175as_ucs4(PyObject *string, Py_UCS4 *target, Py_ssize_t targetsize,
2176 int copy_null)
2177{
2178 int kind;
2179 void *data;
2180 Py_ssize_t len, targetlen;
2181 if (PyUnicode_READY(string) == -1)
2182 return NULL;
2183 kind = PyUnicode_KIND(string);
2184 data = PyUnicode_DATA(string);
2185 len = PyUnicode_GET_LENGTH(string);
2186 targetlen = len;
2187 if (copy_null)
2188 targetlen++;
2189 if (!target) {
2190 if (PY_SSIZE_T_MAX / sizeof(Py_UCS4) < targetlen) {
2191 PyErr_NoMemory();
2192 return NULL;
2193 }
2194 target = PyMem_Malloc(targetlen * sizeof(Py_UCS4));
2195 if (!target) {
2196 PyErr_NoMemory();
2197 return NULL;
2198 }
2199 }
2200 else {
2201 if (targetsize < targetlen) {
2202 PyErr_Format(PyExc_SystemError,
2203 "string is longer than the buffer");
2204 if (copy_null && 0 < targetsize)
2205 target[0] = 0;
2206 return NULL;
2207 }
2208 }
Antoine Pitrou950468e2011-10-11 22:45:48 +02002209 if (kind == PyUnicode_1BYTE_KIND) {
2210 Py_UCS1 *start = (Py_UCS1 *) data;
2211 _PyUnicode_CONVERT_BYTES(Py_UCS1, Py_UCS4, start, start + len, target);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002212 }
Antoine Pitrou950468e2011-10-11 22:45:48 +02002213 else if (kind == PyUnicode_2BYTE_KIND) {
2214 Py_UCS2 *start = (Py_UCS2 *) data;
2215 _PyUnicode_CONVERT_BYTES(Py_UCS2, Py_UCS4, start, start + len, target);
2216 }
2217 else {
2218 assert(kind == PyUnicode_4BYTE_KIND);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002219 Py_MEMCPY(target, data, len * sizeof(Py_UCS4));
Antoine Pitrou950468e2011-10-11 22:45:48 +02002220 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002221 if (copy_null)
2222 target[len] = 0;
2223 return target;
2224}
2225
2226Py_UCS4*
2227PyUnicode_AsUCS4(PyObject *string, Py_UCS4 *target, Py_ssize_t targetsize,
2228 int copy_null)
2229{
Antoine Pitroude20b0b2011-11-10 21:47:38 +01002230 if (target == NULL || targetsize < 0) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002231 PyErr_BadInternalCall();
2232 return NULL;
2233 }
2234 return as_ucs4(string, target, targetsize, copy_null);
2235}
2236
2237Py_UCS4*
2238PyUnicode_AsUCS4Copy(PyObject *string)
2239{
2240 return as_ucs4(string, NULL, 0, 1);
2241}
2242
2243#ifdef HAVE_WCHAR_H
Mark Dickinson081dfee2009-03-18 14:47:41 +00002244
Alexander Belopolsky40018472011-02-26 01:02:56 +00002245PyObject *
2246PyUnicode_FromWideChar(register const wchar_t *w, Py_ssize_t size)
Guido van Rossumd57fd912000-03-10 22:53:23 +00002247{
Guido van Rossumd57fd912000-03-10 22:53:23 +00002248 if (w == NULL) {
Victor Stinner382955f2011-12-11 21:44:00 +01002249 if (size == 0) {
2250 Py_INCREF(unicode_empty);
2251 return unicode_empty;
2252 }
Benjamin Peterson29060642009-01-31 22:14:21 +00002253 PyErr_BadInternalCall();
2254 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00002255 }
2256
Martin v. Löwis790465f2008-04-05 20:41:37 +00002257 if (size == -1) {
2258 size = wcslen(w);
2259 }
2260
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002261 return PyUnicode_FromUnicode(w, size);
Guido van Rossumd57fd912000-03-10 22:53:23 +00002262}
2263
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002264#endif /* HAVE_WCHAR_H */
Mark Dickinson081dfee2009-03-18 14:47:41 +00002265
Walter Dörwald346737f2007-05-31 10:44:43 +00002266static void
Mark Dickinson6ce4a9a2009-11-16 17:00:11 +00002267makefmt(char *fmt, int longflag, int longlongflag, int size_tflag,
2268 int zeropad, int width, int precision, char c)
Walter Dörwald346737f2007-05-31 10:44:43 +00002269{
Benjamin Peterson14339b62009-01-31 16:36:08 +00002270 *fmt++ = '%';
2271 if (width) {
2272 if (zeropad)
2273 *fmt++ = '0';
2274 fmt += sprintf(fmt, "%d", width);
2275 }
2276 if (precision)
2277 fmt += sprintf(fmt, ".%d", precision);
2278 if (longflag)
2279 *fmt++ = 'l';
Mark Dickinson6ce4a9a2009-11-16 17:00:11 +00002280 else if (longlongflag) {
2281 /* longlongflag should only ever be nonzero on machines with
2282 HAVE_LONG_LONG defined */
2283#ifdef HAVE_LONG_LONG
2284 char *f = PY_FORMAT_LONG_LONG;
2285 while (*f)
2286 *fmt++ = *f++;
2287#else
2288 /* we shouldn't ever get here */
2289 assert(0);
2290 *fmt++ = 'l';
2291#endif
2292 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00002293 else if (size_tflag) {
2294 char *f = PY_FORMAT_SIZE_T;
2295 while (*f)
2296 *fmt++ = *f++;
2297 }
2298 *fmt++ = c;
2299 *fmt = '\0';
Walter Dörwald346737f2007-05-31 10:44:43 +00002300}
2301
Victor Stinner96865452011-03-01 23:44:09 +00002302/* helper for PyUnicode_FromFormatV() */
2303
2304static const char*
2305parse_format_flags(const char *f,
2306 int *p_width, int *p_precision,
2307 int *p_longflag, int *p_longlongflag, int *p_size_tflag)
2308{
2309 int width, precision, longflag, longlongflag, size_tflag;
2310
2311 /* parse the width.precision part, e.g. "%2.5s" => width=2, precision=5 */
2312 f++;
2313 width = 0;
2314 while (Py_ISDIGIT((unsigned)*f))
2315 width = (width*10) + *f++ - '0';
2316 precision = 0;
2317 if (*f == '.') {
2318 f++;
2319 while (Py_ISDIGIT((unsigned)*f))
2320 precision = (precision*10) + *f++ - '0';
2321 if (*f == '%') {
2322 /* "%.3%s" => f points to "3" */
2323 f--;
2324 }
2325 }
2326 if (*f == '\0') {
2327 /* bogus format "%.1" => go backward, f points to "1" */
2328 f--;
2329 }
2330 if (p_width != NULL)
2331 *p_width = width;
2332 if (p_precision != NULL)
2333 *p_precision = precision;
2334
2335 /* Handle %ld, %lu, %lld and %llu. */
2336 longflag = 0;
2337 longlongflag = 0;
Victor Stinnere7faec12011-03-02 00:01:53 +00002338 size_tflag = 0;
Victor Stinner96865452011-03-01 23:44:09 +00002339
2340 if (*f == 'l') {
Victor Stinner6d970f42011-03-02 00:04:25 +00002341 if (f[1] == 'd' || f[1] == 'u' || f[1] == 'i') {
Victor Stinner96865452011-03-01 23:44:09 +00002342 longflag = 1;
2343 ++f;
2344 }
2345#ifdef HAVE_LONG_LONG
2346 else if (f[1] == 'l' &&
Victor Stinner6d970f42011-03-02 00:04:25 +00002347 (f[2] == 'd' || f[2] == 'u' || f[2] == 'i')) {
Victor Stinner96865452011-03-01 23:44:09 +00002348 longlongflag = 1;
2349 f += 2;
2350 }
2351#endif
2352 }
2353 /* handle the size_t flag. */
Victor Stinner6d970f42011-03-02 00:04:25 +00002354 else if (*f == 'z' && (f[1] == 'd' || f[1] == 'u' || f[1] == 'i')) {
Victor Stinner96865452011-03-01 23:44:09 +00002355 size_tflag = 1;
2356 ++f;
2357 }
2358 if (p_longflag != NULL)
2359 *p_longflag = longflag;
2360 if (p_longlongflag != NULL)
2361 *p_longlongflag = longlongflag;
2362 if (p_size_tflag != NULL)
2363 *p_size_tflag = size_tflag;
2364 return f;
2365}
2366
Mark Dickinson6ce4a9a2009-11-16 17:00:11 +00002367/* maximum number of characters required for output of %ld. 21 characters
2368 allows for 64-bit integers (in decimal) and an optional sign. */
2369#define MAX_LONG_CHARS 21
2370/* maximum number of characters required for output of %lld.
2371 We need at most ceil(log10(256)*SIZEOF_LONG_LONG) digits,
2372 plus 1 for the sign. 53/22 is an upper bound for log10(256). */
2373#define MAX_LONG_LONG_CHARS (2 + (SIZEOF_LONG_LONG*53-1) / 22)
2374
Walter Dörwaldd2034312007-05-18 16:29:38 +00002375PyObject *
2376PyUnicode_FromFormatV(const char *format, va_list vargs)
2377{
Benjamin Peterson14339b62009-01-31 16:36:08 +00002378 va_list count;
2379 Py_ssize_t callcount = 0;
2380 PyObject **callresults = NULL;
2381 PyObject **callresult = NULL;
2382 Py_ssize_t n = 0;
2383 int width = 0;
2384 int precision = 0;
2385 int zeropad;
2386 const char* f;
Victor Stinnerfb9ea8c2011-10-06 01:45:57 +02002387 PyObject *string;
Benjamin Peterson14339b62009-01-31 16:36:08 +00002388 /* used by sprintf */
Mark Dickinson6ce4a9a2009-11-16 17:00:11 +00002389 char fmt[61]; /* should be enough for %0width.precisionlld */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002390 Py_UCS4 maxchar = 127; /* result is ASCII by default */
2391 Py_UCS4 argmaxchar;
2392 Py_ssize_t numbersize = 0;
2393 char *numberresults = NULL;
2394 char *numberresult = NULL;
2395 Py_ssize_t i;
2396 int kind;
2397 void *data;
Walter Dörwaldd2034312007-05-18 16:29:38 +00002398
Victor Stinner4a2b7a12010-08-13 14:03:48 +00002399 Py_VA_COPY(count, vargs);
Walter Dörwaldc1651a02009-05-03 22:55:55 +00002400 /* step 1: count the number of %S/%R/%A/%s format specifications
2401 * (we call PyObject_Str()/PyObject_Repr()/PyObject_ASCII()/
2402 * PyUnicode_DecodeUTF8() for these objects once during step 3 and put the
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002403 * result in an array)
Georg Brandl7597add2011-10-05 16:36:47 +02002404 * also estimate a upper bound for all the number formats in the string,
2405 * numbers will be formatted in step 3 and be kept in a '\0'-separated
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002406 * buffer before putting everything together. */
Benjamin Peterson14339b62009-01-31 16:36:08 +00002407 for (f = format; *f; f++) {
2408 if (*f == '%') {
Victor Stinner96865452011-03-01 23:44:09 +00002409 int longlongflag;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002410 /* skip width or width.precision (eg. "1.2" of "%1.2f") */
2411 f = parse_format_flags(f, &width, NULL, NULL, &longlongflag, NULL);
2412 if (*f == 's' || *f=='S' || *f=='R' || *f=='A' || *f=='V')
2413 ++callcount;
Walter Dörwaldd2034312007-05-18 16:29:38 +00002414
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002415 else if (*f == 'd' || *f=='u' || *f=='i' || *f=='x' || *f=='p') {
Mark Dickinson6ce4a9a2009-11-16 17:00:11 +00002416#ifdef HAVE_LONG_LONG
2417 if (longlongflag) {
2418 if (width < MAX_LONG_LONG_CHARS)
2419 width = MAX_LONG_LONG_CHARS;
2420 }
2421 else
2422#endif
2423 /* MAX_LONG_CHARS is enough to hold a 64-bit integer,
2424 including sign. Decimal takes the most space. This
2425 isn't enough for octal. If a width is specified we
2426 need more (which we allocate later). */
2427 if (width < MAX_LONG_CHARS)
2428 width = MAX_LONG_CHARS;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002429
2430 /* account for the size + '\0' to separate numbers
2431 inside of the numberresults buffer */
2432 numbersize += (width + 1);
2433 }
2434 }
2435 else if ((unsigned char)*f > 127) {
2436 PyErr_Format(PyExc_ValueError,
2437 "PyUnicode_FromFormatV() expects an ASCII-encoded format "
2438 "string, got a non-ASCII byte: 0x%02x",
2439 (unsigned char)*f);
2440 return NULL;
2441 }
2442 }
2443 /* step 2: allocate memory for the results of
2444 * PyObject_Str()/PyObject_Repr()/PyUnicode_DecodeUTF8() calls */
2445 if (callcount) {
2446 callresults = PyObject_Malloc(sizeof(PyObject *) * callcount);
2447 if (!callresults) {
2448 PyErr_NoMemory();
2449 return NULL;
2450 }
2451 callresult = callresults;
2452 }
2453 /* step 2.5: allocate memory for the results of formating numbers */
2454 if (numbersize) {
2455 numberresults = PyObject_Malloc(numbersize);
2456 if (!numberresults) {
2457 PyErr_NoMemory();
2458 goto fail;
2459 }
2460 numberresult = numberresults;
2461 }
2462
2463 /* step 3: format numbers and figure out how large a buffer we need */
2464 for (f = format; *f; f++) {
2465 if (*f == '%') {
2466 const char* p;
2467 int longflag;
2468 int longlongflag;
2469 int size_tflag;
2470 int numprinted;
2471
2472 p = f;
2473 zeropad = (f[1] == '0');
2474 f = parse_format_flags(f, &width, &precision,
2475 &longflag, &longlongflag, &size_tflag);
2476 switch (*f) {
2477 case 'c':
2478 {
2479 Py_UCS4 ordinal = va_arg(count, int);
Victor Stinnere6abb482012-05-02 01:15:40 +02002480 maxchar = MAX_MAXCHAR(maxchar, ordinal);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002481 n++;
2482 break;
2483 }
2484 case '%':
2485 n++;
2486 break;
2487 case 'i':
2488 case 'd':
2489 makefmt(fmt, longflag, longlongflag, size_tflag, zeropad,
2490 width, precision, *f);
2491 if (longflag)
2492 numprinted = sprintf(numberresult, fmt,
2493 va_arg(count, long));
2494#ifdef HAVE_LONG_LONG
2495 else if (longlongflag)
2496 numprinted = sprintf(numberresult, fmt,
2497 va_arg(count, PY_LONG_LONG));
2498#endif
2499 else if (size_tflag)
2500 numprinted = sprintf(numberresult, fmt,
2501 va_arg(count, Py_ssize_t));
2502 else
2503 numprinted = sprintf(numberresult, fmt,
2504 va_arg(count, int));
2505 n += numprinted;
2506 /* advance by +1 to skip over the '\0' */
2507 numberresult += (numprinted + 1);
2508 assert(*(numberresult - 1) == '\0');
2509 assert(*(numberresult - 2) != '\0');
2510 assert(numprinted >= 0);
2511 assert(numberresult <= numberresults + numbersize);
2512 break;
2513 case 'u':
2514 makefmt(fmt, longflag, longlongflag, size_tflag, zeropad,
2515 width, precision, 'u');
2516 if (longflag)
2517 numprinted = sprintf(numberresult, fmt,
2518 va_arg(count, unsigned long));
2519#ifdef HAVE_LONG_LONG
2520 else if (longlongflag)
2521 numprinted = sprintf(numberresult, fmt,
2522 va_arg(count, unsigned PY_LONG_LONG));
2523#endif
2524 else if (size_tflag)
2525 numprinted = sprintf(numberresult, fmt,
2526 va_arg(count, size_t));
2527 else
2528 numprinted = sprintf(numberresult, fmt,
2529 va_arg(count, unsigned int));
2530 n += numprinted;
2531 numberresult += (numprinted + 1);
2532 assert(*(numberresult - 1) == '\0');
2533 assert(*(numberresult - 2) != '\0');
2534 assert(numprinted >= 0);
2535 assert(numberresult <= numberresults + numbersize);
2536 break;
2537 case 'x':
2538 makefmt(fmt, 0, 0, 0, zeropad, width, precision, 'x');
2539 numprinted = sprintf(numberresult, fmt, va_arg(count, int));
2540 n += numprinted;
2541 numberresult += (numprinted + 1);
2542 assert(*(numberresult - 1) == '\0');
2543 assert(*(numberresult - 2) != '\0');
2544 assert(numprinted >= 0);
2545 assert(numberresult <= numberresults + numbersize);
2546 break;
2547 case 'p':
2548 numprinted = sprintf(numberresult, "%p", va_arg(count, void*));
2549 /* %p is ill-defined: ensure leading 0x. */
2550 if (numberresult[1] == 'X')
2551 numberresult[1] = 'x';
2552 else if (numberresult[1] != 'x') {
2553 memmove(numberresult + 2, numberresult,
2554 strlen(numberresult) + 1);
2555 numberresult[0] = '0';
2556 numberresult[1] = 'x';
2557 numprinted += 2;
2558 }
2559 n += numprinted;
2560 numberresult += (numprinted + 1);
2561 assert(*(numberresult - 1) == '\0');
2562 assert(*(numberresult - 2) != '\0');
2563 assert(numprinted >= 0);
2564 assert(numberresult <= numberresults + numbersize);
Benjamin Peterson14339b62009-01-31 16:36:08 +00002565 break;
2566 case 's':
2567 {
2568 /* UTF-8 */
Georg Brandl780b2a62009-05-05 09:19:59 +00002569 const char *s = va_arg(count, const char*);
Victor Stinnera1d12bb2011-12-11 21:53:09 +01002570 PyObject *str = PyUnicode_DecodeUTF8Stateful(s, strlen(s), "replace", NULL);
Walter Dörwaldc1651a02009-05-03 22:55:55 +00002571 if (!str)
2572 goto fail;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002573 /* since PyUnicode_DecodeUTF8 returns already flexible
2574 unicode objects, there is no need to call ready on them */
2575 argmaxchar = PyUnicode_MAX_CHAR_VALUE(str);
Victor Stinnere6abb482012-05-02 01:15:40 +02002576 maxchar = MAX_MAXCHAR(maxchar, argmaxchar);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002577 n += PyUnicode_GET_LENGTH(str);
Walter Dörwaldc1651a02009-05-03 22:55:55 +00002578 /* Remember the str and switch to the next slot */
2579 *callresult++ = str;
Benjamin Peterson14339b62009-01-31 16:36:08 +00002580 break;
2581 }
2582 case 'U':
2583 {
2584 PyObject *obj = va_arg(count, PyObject *);
Victor Stinner910337b2011-10-03 03:20:16 +02002585 assert(obj && _PyUnicode_CHECK(obj));
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002586 if (PyUnicode_READY(obj) == -1)
2587 goto fail;
2588 argmaxchar = PyUnicode_MAX_CHAR_VALUE(obj);
Victor Stinnere6abb482012-05-02 01:15:40 +02002589 maxchar = MAX_MAXCHAR(maxchar, argmaxchar);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002590 n += PyUnicode_GET_LENGTH(obj);
Benjamin Peterson14339b62009-01-31 16:36:08 +00002591 break;
2592 }
2593 case 'V':
2594 {
2595 PyObject *obj = va_arg(count, PyObject *);
2596 const char *str = va_arg(count, const char *);
Victor Stinner2512a8b2011-03-01 22:46:52 +00002597 PyObject *str_obj;
Benjamin Peterson14339b62009-01-31 16:36:08 +00002598 assert(obj || str);
Victor Stinner910337b2011-10-03 03:20:16 +02002599 assert(!obj || _PyUnicode_CHECK(obj));
Victor Stinner2512a8b2011-03-01 22:46:52 +00002600 if (obj) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002601 if (PyUnicode_READY(obj) == -1)
2602 goto fail;
2603 argmaxchar = PyUnicode_MAX_CHAR_VALUE(obj);
Victor Stinnere6abb482012-05-02 01:15:40 +02002604 maxchar = MAX_MAXCHAR(maxchar, argmaxchar);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002605 n += PyUnicode_GET_LENGTH(obj);
Victor Stinner2512a8b2011-03-01 22:46:52 +00002606 *callresult++ = NULL;
2607 }
2608 else {
Victor Stinnera1d12bb2011-12-11 21:53:09 +01002609 str_obj = PyUnicode_DecodeUTF8Stateful(str, strlen(str), "replace", NULL);
Victor Stinner2512a8b2011-03-01 22:46:52 +00002610 if (!str_obj)
2611 goto fail;
Benjamin Petersonbac79492012-01-14 13:34:47 -05002612 if (PyUnicode_READY(str_obj) == -1) {
Victor Stinnere1335c72011-10-04 20:53:03 +02002613 Py_DECREF(str_obj);
2614 goto fail;
2615 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002616 argmaxchar = PyUnicode_MAX_CHAR_VALUE(str_obj);
Victor Stinnere6abb482012-05-02 01:15:40 +02002617 maxchar = MAX_MAXCHAR(maxchar, argmaxchar);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002618 n += PyUnicode_GET_LENGTH(str_obj);
Victor Stinner2512a8b2011-03-01 22:46:52 +00002619 *callresult++ = str_obj;
2620 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00002621 break;
2622 }
2623 case 'S':
2624 {
2625 PyObject *obj = va_arg(count, PyObject *);
2626 PyObject *str;
2627 assert(obj);
2628 str = PyObject_Str(obj);
Benjamin Petersonc8d8b882012-01-14 13:37:31 -05002629 if (!str)
Benjamin Peterson14339b62009-01-31 16:36:08 +00002630 goto fail;
Benjamin Petersonc8d8b882012-01-14 13:37:31 -05002631 if (PyUnicode_READY(str) == -1) {
2632 Py_DECREF(str);
2633 goto fail;
2634 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002635 argmaxchar = PyUnicode_MAX_CHAR_VALUE(str);
Victor Stinnere6abb482012-05-02 01:15:40 +02002636 maxchar = MAX_MAXCHAR(maxchar, argmaxchar);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002637 n += PyUnicode_GET_LENGTH(str);
Benjamin Peterson14339b62009-01-31 16:36:08 +00002638 /* Remember the str and switch to the next slot */
2639 *callresult++ = str;
2640 break;
2641 }
2642 case 'R':
2643 {
2644 PyObject *obj = va_arg(count, PyObject *);
2645 PyObject *repr;
2646 assert(obj);
2647 repr = PyObject_Repr(obj);
Benjamin Petersonc8d8b882012-01-14 13:37:31 -05002648 if (!repr)
Benjamin Peterson14339b62009-01-31 16:36:08 +00002649 goto fail;
Benjamin Petersonc8d8b882012-01-14 13:37:31 -05002650 if (PyUnicode_READY(repr) == -1) {
2651 Py_DECREF(repr);
2652 goto fail;
2653 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002654 argmaxchar = PyUnicode_MAX_CHAR_VALUE(repr);
Victor Stinnere6abb482012-05-02 01:15:40 +02002655 maxchar = MAX_MAXCHAR(maxchar, argmaxchar);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002656 n += PyUnicode_GET_LENGTH(repr);
Benjamin Peterson14339b62009-01-31 16:36:08 +00002657 /* Remember the repr and switch to the next slot */
2658 *callresult++ = repr;
2659 break;
2660 }
2661 case 'A':
2662 {
2663 PyObject *obj = va_arg(count, PyObject *);
2664 PyObject *ascii;
2665 assert(obj);
2666 ascii = PyObject_ASCII(obj);
Benjamin Petersonc8d8b882012-01-14 13:37:31 -05002667 if (!ascii)
Benjamin Peterson14339b62009-01-31 16:36:08 +00002668 goto fail;
Benjamin Petersonc8d8b882012-01-14 13:37:31 -05002669 if (PyUnicode_READY(ascii) == -1) {
2670 Py_DECREF(ascii);
2671 goto fail;
2672 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002673 argmaxchar = PyUnicode_MAX_CHAR_VALUE(ascii);
Victor Stinnere6abb482012-05-02 01:15:40 +02002674 maxchar = MAX_MAXCHAR(maxchar, argmaxchar);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002675 n += PyUnicode_GET_LENGTH(ascii);
Benjamin Peterson14339b62009-01-31 16:36:08 +00002676 /* Remember the repr and switch to the next slot */
2677 *callresult++ = ascii;
2678 break;
2679 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00002680 default:
2681 /* if we stumble upon an unknown
2682 formatting code, copy the rest of
2683 the format string to the output
2684 string. (we cannot just skip the
2685 code, since there's no way to know
2686 what's in the argument list) */
2687 n += strlen(p);
2688 goto expand;
2689 }
2690 } else
2691 n++;
2692 }
Benjamin Peterson29060642009-01-31 22:14:21 +00002693 expand:
Benjamin Peterson14339b62009-01-31 16:36:08 +00002694 /* step 4: fill the buffer */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002695 /* Since we've analyzed how much space we need,
Benjamin Peterson14339b62009-01-31 16:36:08 +00002696 we don't have to resize the string.
2697 There can be no errors beyond this point. */
Victor Stinnerfb9ea8c2011-10-06 01:45:57 +02002698 string = PyUnicode_New(n, maxchar);
Benjamin Peterson14339b62009-01-31 16:36:08 +00002699 if (!string)
2700 goto fail;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002701 kind = PyUnicode_KIND(string);
2702 data = PyUnicode_DATA(string);
Benjamin Peterson14339b62009-01-31 16:36:08 +00002703 callresult = callresults;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002704 numberresult = numberresults;
Walter Dörwaldd2034312007-05-18 16:29:38 +00002705
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002706 for (i = 0, f = format; *f; f++) {
Benjamin Peterson14339b62009-01-31 16:36:08 +00002707 if (*f == '%') {
Victor Stinner96865452011-03-01 23:44:09 +00002708 const char* p;
Victor Stinner96865452011-03-01 23:44:09 +00002709
2710 p = f;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002711 f = parse_format_flags(f, NULL, NULL, NULL, NULL, NULL);
2712 /* checking for == because the last argument could be a empty
2713 string, which causes i to point to end, the assert at the end of
2714 the loop */
2715 assert(i <= PyUnicode_GET_LENGTH(string));
Walter Dörwaldd2034312007-05-18 16:29:38 +00002716
Benjamin Peterson14339b62009-01-31 16:36:08 +00002717 switch (*f) {
2718 case 'c':
Victor Stinner5ed8b2c2011-02-21 21:13:44 +00002719 {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002720 const int ordinal = va_arg(vargs, int);
2721 PyUnicode_WRITE(kind, data, i++, ordinal);
Benjamin Peterson14339b62009-01-31 16:36:08 +00002722 break;
Victor Stinner5ed8b2c2011-02-21 21:13:44 +00002723 }
Victor Stinner6d970f42011-03-02 00:04:25 +00002724 case 'i':
Benjamin Peterson14339b62009-01-31 16:36:08 +00002725 case 'd':
Benjamin Peterson14339b62009-01-31 16:36:08 +00002726 case 'u':
Benjamin Peterson14339b62009-01-31 16:36:08 +00002727 case 'x':
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002728 case 'p':
Victor Stinnerc5166102012-02-22 13:55:02 +01002729 {
Victor Stinner184252a2012-06-16 02:57:41 +02002730 Py_ssize_t len;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002731 /* unused, since we already have the result */
2732 if (*f == 'p')
2733 (void) va_arg(vargs, void *);
2734 else
2735 (void) va_arg(vargs, int);
2736 /* extract the result from numberresults and append. */
Victor Stinner184252a2012-06-16 02:57:41 +02002737 len = strlen(numberresult);
2738 unicode_write_cstr(string, i, numberresult, len);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002739 /* skip over the separating '\0' */
Victor Stinner184252a2012-06-16 02:57:41 +02002740 i += len;
2741 numberresult += len;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002742 assert(*numberresult == '\0');
2743 numberresult++;
2744 assert(numberresult <= numberresults + numbersize);
Benjamin Peterson14339b62009-01-31 16:36:08 +00002745 break;
Victor Stinnerc5166102012-02-22 13:55:02 +01002746 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00002747 case 's':
2748 {
Walter Dörwaldc1651a02009-05-03 22:55:55 +00002749 /* unused, since we already have the result */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002750 Py_ssize_t size;
Walter Dörwaldc1651a02009-05-03 22:55:55 +00002751 (void) va_arg(vargs, char *);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002752 size = PyUnicode_GET_LENGTH(*callresult);
2753 assert(PyUnicode_KIND(*callresult) <= PyUnicode_KIND(string));
Victor Stinnerd3f08822012-05-29 12:57:52 +02002754 _PyUnicode_FastCopyCharacters(string, i, *callresult, 0, size);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002755 i += size;
Walter Dörwaldc1651a02009-05-03 22:55:55 +00002756 /* We're done with the unicode()/repr() => forget it */
2757 Py_DECREF(*callresult);
2758 /* switch to next unicode()/repr() result */
2759 ++callresult;
Benjamin Peterson14339b62009-01-31 16:36:08 +00002760 break;
2761 }
2762 case 'U':
2763 {
2764 PyObject *obj = va_arg(vargs, PyObject *);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002765 Py_ssize_t size;
2766 assert(PyUnicode_KIND(obj) <= PyUnicode_KIND(string));
2767 size = PyUnicode_GET_LENGTH(obj);
Victor Stinnerd3f08822012-05-29 12:57:52 +02002768 _PyUnicode_FastCopyCharacters(string, i, obj, 0, size);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002769 i += size;
Benjamin Peterson14339b62009-01-31 16:36:08 +00002770 break;
2771 }
2772 case 'V':
2773 {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002774 Py_ssize_t size;
Benjamin Peterson14339b62009-01-31 16:36:08 +00002775 PyObject *obj = va_arg(vargs, PyObject *);
Victor Stinner2512a8b2011-03-01 22:46:52 +00002776 va_arg(vargs, const char *);
Benjamin Peterson14339b62009-01-31 16:36:08 +00002777 if (obj) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002778 size = PyUnicode_GET_LENGTH(obj);
2779 assert(PyUnicode_KIND(obj) <= PyUnicode_KIND(string));
Victor Stinnerd3f08822012-05-29 12:57:52 +02002780 _PyUnicode_FastCopyCharacters(string, i, obj, 0, size);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002781 i += size;
Benjamin Peterson14339b62009-01-31 16:36:08 +00002782 } else {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002783 size = PyUnicode_GET_LENGTH(*callresult);
2784 assert(PyUnicode_KIND(*callresult) <=
2785 PyUnicode_KIND(string));
Victor Stinnerd3f08822012-05-29 12:57:52 +02002786 _PyUnicode_FastCopyCharacters(string, i, *callresult, 0, size);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002787 i += size;
Victor Stinner2512a8b2011-03-01 22:46:52 +00002788 Py_DECREF(*callresult);
Benjamin Peterson14339b62009-01-31 16:36:08 +00002789 }
Victor Stinner2512a8b2011-03-01 22:46:52 +00002790 ++callresult;
Benjamin Peterson14339b62009-01-31 16:36:08 +00002791 break;
2792 }
2793 case 'S':
2794 case 'R':
Victor Stinner9a909002010-10-18 20:59:24 +00002795 case 'A':
Benjamin Peterson14339b62009-01-31 16:36:08 +00002796 {
Victor Stinnerfb9ea8c2011-10-06 01:45:57 +02002797 Py_ssize_t size = PyUnicode_GET_LENGTH(*callresult);
Benjamin Peterson14339b62009-01-31 16:36:08 +00002798 /* unused, since we already have the result */
2799 (void) va_arg(vargs, PyObject *);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002800 assert(PyUnicode_KIND(*callresult) <= PyUnicode_KIND(string));
Victor Stinnerd3f08822012-05-29 12:57:52 +02002801 _PyUnicode_FastCopyCharacters(string, i, *callresult, 0, size);
Victor Stinnerfb9ea8c2011-10-06 01:45:57 +02002802 i += size;
Benjamin Peterson14339b62009-01-31 16:36:08 +00002803 /* We're done with the unicode()/repr() => forget it */
2804 Py_DECREF(*callresult);
2805 /* switch to next unicode()/repr() result */
2806 ++callresult;
2807 break;
2808 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00002809 case '%':
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002810 PyUnicode_WRITE(kind, data, i++, '%');
Benjamin Peterson14339b62009-01-31 16:36:08 +00002811 break;
2812 default:
Victor Stinner184252a2012-06-16 02:57:41 +02002813 {
2814 Py_ssize_t len = strlen(p);
2815 unicode_write_cstr(string, i, p, len);
2816 i += len;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002817 assert(i == PyUnicode_GET_LENGTH(string));
Benjamin Peterson14339b62009-01-31 16:36:08 +00002818 goto end;
2819 }
Victor Stinner184252a2012-06-16 02:57:41 +02002820 }
Victor Stinner1205f272010-09-11 00:54:47 +00002821 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002822 else {
2823 assert(i < PyUnicode_GET_LENGTH(string));
2824 PyUnicode_WRITE(kind, data, i++, *f);
2825 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00002826 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002827 assert(i == PyUnicode_GET_LENGTH(string));
Walter Dörwaldd2034312007-05-18 16:29:38 +00002828
Benjamin Peterson29060642009-01-31 22:14:21 +00002829 end:
Benjamin Peterson14339b62009-01-31 16:36:08 +00002830 if (callresults)
2831 PyObject_Free(callresults);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002832 if (numberresults)
2833 PyObject_Free(numberresults);
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01002834 return unicode_result(string);
Benjamin Peterson29060642009-01-31 22:14:21 +00002835 fail:
Benjamin Peterson14339b62009-01-31 16:36:08 +00002836 if (callresults) {
2837 PyObject **callresult2 = callresults;
2838 while (callresult2 < callresult) {
Victor Stinner2512a8b2011-03-01 22:46:52 +00002839 Py_XDECREF(*callresult2);
Benjamin Peterson14339b62009-01-31 16:36:08 +00002840 ++callresult2;
2841 }
2842 PyObject_Free(callresults);
2843 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002844 if (numberresults)
2845 PyObject_Free(numberresults);
Benjamin Peterson14339b62009-01-31 16:36:08 +00002846 return NULL;
Walter Dörwaldd2034312007-05-18 16:29:38 +00002847}
2848
Walter Dörwaldd2034312007-05-18 16:29:38 +00002849PyObject *
2850PyUnicode_FromFormat(const char *format, ...)
2851{
Benjamin Peterson14339b62009-01-31 16:36:08 +00002852 PyObject* ret;
2853 va_list vargs;
Walter Dörwaldd2034312007-05-18 16:29:38 +00002854
2855#ifdef HAVE_STDARG_PROTOTYPES
Benjamin Peterson14339b62009-01-31 16:36:08 +00002856 va_start(vargs, format);
Walter Dörwaldd2034312007-05-18 16:29:38 +00002857#else
Benjamin Peterson14339b62009-01-31 16:36:08 +00002858 va_start(vargs);
Walter Dörwaldd2034312007-05-18 16:29:38 +00002859#endif
Benjamin Peterson14339b62009-01-31 16:36:08 +00002860 ret = PyUnicode_FromFormatV(format, vargs);
2861 va_end(vargs);
2862 return ret;
Walter Dörwaldd2034312007-05-18 16:29:38 +00002863}
2864
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002865#ifdef HAVE_WCHAR_H
2866
Victor Stinner5593d8a2010-10-02 11:11:27 +00002867/* Helper function for PyUnicode_AsWideChar() and PyUnicode_AsWideCharString():
2868 convert a Unicode object to a wide character string.
2869
Victor Stinnerd88d9832011-09-06 02:00:05 +02002870 - If w is NULL: return the number of wide characters (including the null
Victor Stinner5593d8a2010-10-02 11:11:27 +00002871 character) required to convert the unicode object. Ignore size argument.
2872
Victor Stinnerd88d9832011-09-06 02:00:05 +02002873 - Otherwise: return the number of wide characters (excluding the null
Victor Stinner5593d8a2010-10-02 11:11:27 +00002874 character) written into w. Write at most size wide characters (including
Victor Stinnerd88d9832011-09-06 02:00:05 +02002875 the null character). */
Victor Stinner5593d8a2010-10-02 11:11:27 +00002876static Py_ssize_t
Victor Stinner9db1a8b2011-10-23 20:04:37 +02002877unicode_aswidechar(PyObject *unicode,
Victor Stinner137c34c2010-09-29 10:25:54 +00002878 wchar_t *w,
2879 Py_ssize_t size)
2880{
Victor Stinner5593d8a2010-10-02 11:11:27 +00002881 Py_ssize_t res;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002882 const wchar_t *wstr;
2883
Victor Stinner9db1a8b2011-10-23 20:04:37 +02002884 wstr = PyUnicode_AsUnicodeAndSize(unicode, &res);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002885 if (wstr == NULL)
2886 return -1;
2887
Victor Stinner5593d8a2010-10-02 11:11:27 +00002888 if (w != NULL) {
Victor Stinner5593d8a2010-10-02 11:11:27 +00002889 if (size > res)
2890 size = res + 1;
2891 else
2892 res = size;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002893 Py_MEMCPY(w, wstr, size * sizeof(wchar_t));
Victor Stinner5593d8a2010-10-02 11:11:27 +00002894 return res;
2895 }
2896 else
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002897 return res + 1;
Victor Stinner137c34c2010-09-29 10:25:54 +00002898}
2899
2900Py_ssize_t
Martin v. Löwis4d0d4712010-12-03 20:14:31 +00002901PyUnicode_AsWideChar(PyObject *unicode,
Victor Stinner137c34c2010-09-29 10:25:54 +00002902 wchar_t *w,
2903 Py_ssize_t size)
Guido van Rossumd57fd912000-03-10 22:53:23 +00002904{
2905 if (unicode == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +00002906 PyErr_BadInternalCall();
2907 return -1;
Guido van Rossumd57fd912000-03-10 22:53:23 +00002908 }
Victor Stinner9db1a8b2011-10-23 20:04:37 +02002909 return unicode_aswidechar(unicode, w, size);
Guido van Rossumd57fd912000-03-10 22:53:23 +00002910}
2911
Victor Stinner137c34c2010-09-29 10:25:54 +00002912wchar_t*
Victor Stinnerbeb4135b2010-10-07 01:02:42 +00002913PyUnicode_AsWideCharString(PyObject *unicode,
Victor Stinner137c34c2010-09-29 10:25:54 +00002914 Py_ssize_t *size)
2915{
2916 wchar_t* buffer;
2917 Py_ssize_t buflen;
2918
2919 if (unicode == NULL) {
2920 PyErr_BadInternalCall();
2921 return NULL;
2922 }
2923
Victor Stinner9db1a8b2011-10-23 20:04:37 +02002924 buflen = unicode_aswidechar(unicode, NULL, 0);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002925 if (buflen == -1)
2926 return NULL;
Victor Stinner5593d8a2010-10-02 11:11:27 +00002927 if (PY_SSIZE_T_MAX / sizeof(wchar_t) < buflen) {
Victor Stinner137c34c2010-09-29 10:25:54 +00002928 PyErr_NoMemory();
2929 return NULL;
2930 }
2931
Victor Stinner137c34c2010-09-29 10:25:54 +00002932 buffer = PyMem_MALLOC(buflen * sizeof(wchar_t));
2933 if (buffer == NULL) {
2934 PyErr_NoMemory();
2935 return NULL;
2936 }
Victor Stinner9db1a8b2011-10-23 20:04:37 +02002937 buflen = unicode_aswidechar(unicode, buffer, buflen);
Stefan Krah8528c312012-08-19 21:52:43 +02002938 if (buflen == -1) {
2939 PyMem_FREE(buffer);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002940 return NULL;
Stefan Krah8528c312012-08-19 21:52:43 +02002941 }
Victor Stinner5593d8a2010-10-02 11:11:27 +00002942 if (size != NULL)
2943 *size = buflen;
Victor Stinner137c34c2010-09-29 10:25:54 +00002944 return buffer;
2945}
2946
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002947#endif /* HAVE_WCHAR_H */
Guido van Rossumd57fd912000-03-10 22:53:23 +00002948
Alexander Belopolsky40018472011-02-26 01:02:56 +00002949PyObject *
2950PyUnicode_FromOrdinal(int ordinal)
Marc-André Lemburgcc8764c2002-08-11 12:23:04 +00002951{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002952 PyObject *v;
Victor Stinner8faf8212011-12-08 22:14:11 +01002953 if (ordinal < 0 || ordinal > MAX_UNICODE) {
Benjamin Peterson29060642009-01-31 22:14:21 +00002954 PyErr_SetString(PyExc_ValueError,
2955 "chr() arg not in range(0x110000)");
2956 return NULL;
Marc-André Lemburgcc8764c2002-08-11 12:23:04 +00002957 }
Guido van Rossum8ac004e2007-07-15 13:00:05 +00002958
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002959 if (ordinal < 256)
2960 return get_latin1_char(ordinal);
Marc-André Lemburgcc8764c2002-08-11 12:23:04 +00002961
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002962 v = PyUnicode_New(1, ordinal);
2963 if (v == NULL)
2964 return NULL;
2965 PyUnicode_WRITE(PyUnicode_KIND(v), PyUnicode_DATA(v), 0, ordinal);
Victor Stinnerbb10a1f2011-10-05 01:34:17 +02002966 assert(_PyUnicode_CheckConsistency(v, 1));
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002967 return v;
Marc-André Lemburgcc8764c2002-08-11 12:23:04 +00002968}
2969
Alexander Belopolsky40018472011-02-26 01:02:56 +00002970PyObject *
2971PyUnicode_FromObject(register PyObject *obj)
Guido van Rossumd57fd912000-03-10 22:53:23 +00002972{
Guido van Rossumb8c65bc2001-10-19 02:01:31 +00002973 /* XXX Perhaps we should make this API an alias of
Benjamin Peterson29060642009-01-31 22:14:21 +00002974 PyObject_Str() instead ?! */
Guido van Rossumb8c65bc2001-10-19 02:01:31 +00002975 if (PyUnicode_CheckExact(obj)) {
Benjamin Petersonbac79492012-01-14 13:34:47 -05002976 if (PyUnicode_READY(obj) == -1)
Victor Stinnerd3a83d52011-10-01 03:09:33 +02002977 return NULL;
Benjamin Peterson29060642009-01-31 22:14:21 +00002978 Py_INCREF(obj);
2979 return obj;
Guido van Rossumb8c65bc2001-10-19 02:01:31 +00002980 }
2981 if (PyUnicode_Check(obj)) {
Benjamin Peterson29060642009-01-31 22:14:21 +00002982 /* For a Unicode subtype that's not a Unicode object,
2983 return a true Unicode object with the same data. */
Victor Stinnerbf6e5602011-12-12 01:53:47 +01002984 return _PyUnicode_Copy(obj);
Guido van Rossumb8c65bc2001-10-19 02:01:31 +00002985 }
Guido van Rossum98297ee2007-11-06 21:34:58 +00002986 PyErr_Format(PyExc_TypeError,
2987 "Can't convert '%.100s' object to str implicitly",
Christian Heimes90aa7642007-12-19 02:45:37 +00002988 Py_TYPE(obj)->tp_name);
Guido van Rossum98297ee2007-11-06 21:34:58 +00002989 return NULL;
Marc-André Lemburg5a5c81a2000-07-07 13:46:42 +00002990}
2991
Alexander Belopolsky40018472011-02-26 01:02:56 +00002992PyObject *
2993PyUnicode_FromEncodedObject(register PyObject *obj,
Ezio Melotti2aa2b3b2011-09-29 00:58:57 +03002994 const char *encoding,
2995 const char *errors)
Marc-André Lemburg5a5c81a2000-07-07 13:46:42 +00002996{
Antoine Pitroub0fa8312010-09-01 15:10:12 +00002997 Py_buffer buffer;
Marc-André Lemburg5a5c81a2000-07-07 13:46:42 +00002998 PyObject *v;
Tim Petersced69f82003-09-16 20:30:58 +00002999
Guido van Rossumd57fd912000-03-10 22:53:23 +00003000 if (obj == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +00003001 PyErr_BadInternalCall();
3002 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00003003 }
Marc-André Lemburg5a5c81a2000-07-07 13:46:42 +00003004
Antoine Pitroub0fa8312010-09-01 15:10:12 +00003005 /* Decoding bytes objects is the most common case and should be fast */
3006 if (PyBytes_Check(obj)) {
3007 if (PyBytes_GET_SIZE(obj) == 0) {
3008 Py_INCREF(unicode_empty);
Victor Stinnera464fc12011-10-02 20:39:30 +02003009 v = unicode_empty;
Antoine Pitroub0fa8312010-09-01 15:10:12 +00003010 }
3011 else {
3012 v = PyUnicode_Decode(
3013 PyBytes_AS_STRING(obj), PyBytes_GET_SIZE(obj),
3014 encoding, errors);
3015 }
3016 return v;
3017 }
3018
Guido van Rossumb8c65bc2001-10-19 02:01:31 +00003019 if (PyUnicode_Check(obj)) {
Benjamin Peterson29060642009-01-31 22:14:21 +00003020 PyErr_SetString(PyExc_TypeError,
3021 "decoding str is not supported");
3022 return NULL;
Benjamin Peterson14339b62009-01-31 16:36:08 +00003023 }
Guido van Rossumb8c65bc2001-10-19 02:01:31 +00003024
Antoine Pitroub0fa8312010-09-01 15:10:12 +00003025 /* Retrieve a bytes buffer view through the PEP 3118 buffer interface */
3026 if (PyObject_GetBuffer(obj, &buffer, PyBUF_SIMPLE) < 0) {
3027 PyErr_Format(PyExc_TypeError,
3028 "coercing to str: need bytes, bytearray "
3029 "or buffer-like object, %.80s found",
3030 Py_TYPE(obj)->tp_name);
3031 return NULL;
Marc-André Lemburg6871f6a2001-09-20 12:53:16 +00003032 }
Tim Petersced69f82003-09-16 20:30:58 +00003033
Antoine Pitroub0fa8312010-09-01 15:10:12 +00003034 if (buffer.len == 0) {
Benjamin Peterson29060642009-01-31 22:14:21 +00003035 Py_INCREF(unicode_empty);
Victor Stinnera464fc12011-10-02 20:39:30 +02003036 v = unicode_empty;
Guido van Rossumd57fd912000-03-10 22:53:23 +00003037 }
Tim Petersced69f82003-09-16 20:30:58 +00003038 else
Antoine Pitroub0fa8312010-09-01 15:10:12 +00003039 v = PyUnicode_Decode((char*) buffer.buf, buffer.len, encoding, errors);
Marc-André Lemburgad7c98e2001-01-17 17:09:53 +00003040
Antoine Pitroub0fa8312010-09-01 15:10:12 +00003041 PyBuffer_Release(&buffer);
Marc-André Lemburg5a5c81a2000-07-07 13:46:42 +00003042 return v;
Guido van Rossumd57fd912000-03-10 22:53:23 +00003043}
3044
Victor Stinner600d3be2010-06-10 12:00:55 +00003045/* Convert encoding to lower case and replace '_' with '-' in order to
Victor Stinner37296e82010-06-10 13:36:23 +00003046 catch e.g. UTF_8. Return 0 on error (encoding is longer than lower_len-1),
3047 1 on success. */
3048static int
3049normalize_encoding(const char *encoding,
3050 char *lower,
3051 size_t lower_len)
Guido van Rossumd57fd912000-03-10 22:53:23 +00003052{
Guido van Rossumdaa251c2007-10-25 23:47:33 +00003053 const char *e;
Victor Stinner600d3be2010-06-10 12:00:55 +00003054 char *l;
3055 char *l_end;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00003056
Benjamin Peterson7a6debe2011-10-15 09:25:28 -04003057 if (encoding == NULL) {
3058 strcpy(lower, "utf-8");
3059 return 1;
3060 }
Guido van Rossumdaa251c2007-10-25 23:47:33 +00003061 e = encoding;
3062 l = lower;
Victor Stinner600d3be2010-06-10 12:00:55 +00003063 l_end = &lower[lower_len - 1];
Victor Stinner37296e82010-06-10 13:36:23 +00003064 while (*e) {
3065 if (l == l_end)
3066 return 0;
David Malcolm96960882010-11-05 17:23:41 +00003067 if (Py_ISUPPER(*e)) {
3068 *l++ = Py_TOLOWER(*e++);
Guido van Rossumdaa251c2007-10-25 23:47:33 +00003069 }
3070 else if (*e == '_') {
3071 *l++ = '-';
3072 e++;
3073 }
3074 else {
3075 *l++ = *e++;
3076 }
3077 }
3078 *l = '\0';
Victor Stinner37296e82010-06-10 13:36:23 +00003079 return 1;
Victor Stinner600d3be2010-06-10 12:00:55 +00003080}
3081
Alexander Belopolsky40018472011-02-26 01:02:56 +00003082PyObject *
3083PyUnicode_Decode(const char *s,
Ezio Melotti2aa2b3b2011-09-29 00:58:57 +03003084 Py_ssize_t size,
3085 const char *encoding,
3086 const char *errors)
Victor Stinner600d3be2010-06-10 12:00:55 +00003087{
3088 PyObject *buffer = NULL, *unicode;
3089 Py_buffer info;
3090 char lower[11]; /* Enough for any encoding shortcut */
3091
Fred Drakee4315f52000-05-09 19:53:39 +00003092 /* Shortcuts for common default encodings */
Victor Stinner37296e82010-06-10 13:36:23 +00003093 if (normalize_encoding(encoding, lower, sizeof(lower))) {
Alexander Belopolsky1d521462011-02-25 19:19:57 +00003094 if ((strcmp(lower, "utf-8") == 0) ||
3095 (strcmp(lower, "utf8") == 0))
Victor Stinnera1d12bb2011-12-11 21:53:09 +01003096 return PyUnicode_DecodeUTF8Stateful(s, size, errors, NULL);
Victor Stinner37296e82010-06-10 13:36:23 +00003097 else if ((strcmp(lower, "latin-1") == 0) ||
Alexander Belopolsky1d521462011-02-25 19:19:57 +00003098 (strcmp(lower, "latin1") == 0) ||
Victor Stinner37296e82010-06-10 13:36:23 +00003099 (strcmp(lower, "iso-8859-1") == 0))
3100 return PyUnicode_DecodeLatin1(s, size, errors);
Victor Stinner99b95382011-07-04 14:23:54 +02003101#ifdef HAVE_MBCS
Victor Stinner37296e82010-06-10 13:36:23 +00003102 else if (strcmp(lower, "mbcs") == 0)
3103 return PyUnicode_DecodeMBCS(s, size, errors);
Mark Hammond0ccda1e2003-07-01 00:13:27 +00003104#endif
Victor Stinner37296e82010-06-10 13:36:23 +00003105 else if (strcmp(lower, "ascii") == 0)
3106 return PyUnicode_DecodeASCII(s, size, errors);
3107 else if (strcmp(lower, "utf-16") == 0)
3108 return PyUnicode_DecodeUTF16(s, size, errors, 0);
3109 else if (strcmp(lower, "utf-32") == 0)
3110 return PyUnicode_DecodeUTF32(s, size, errors, 0);
3111 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00003112
3113 /* Decode via the codec registry */
Guido van Rossumbe801ac2007-10-08 03:32:34 +00003114 buffer = NULL;
Antoine Pitrouc3b39242009-01-03 16:59:18 +00003115 if (PyBuffer_FillInfo(&info, NULL, (void *)s, size, 1, PyBUF_FULL_RO) < 0)
Guido van Rossumbe801ac2007-10-08 03:32:34 +00003116 goto onError;
Antoine Pitrouee58fa42008-08-19 18:22:14 +00003117 buffer = PyMemoryView_FromBuffer(&info);
Guido van Rossumd57fd912000-03-10 22:53:23 +00003118 if (buffer == NULL)
3119 goto onError;
3120 unicode = PyCodec_Decode(buffer, encoding, errors);
3121 if (unicode == NULL)
3122 goto onError;
3123 if (!PyUnicode_Check(unicode)) {
3124 PyErr_Format(PyExc_TypeError,
Benjamin Peterson142957c2008-07-04 19:55:29 +00003125 "decoder did not return a str object (type=%.400s)",
Christian Heimes90aa7642007-12-19 02:45:37 +00003126 Py_TYPE(unicode)->tp_name);
Guido van Rossumd57fd912000-03-10 22:53:23 +00003127 Py_DECREF(unicode);
3128 goto onError;
3129 }
3130 Py_DECREF(buffer);
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01003131 return unicode_result(unicode);
Tim Petersced69f82003-09-16 20:30:58 +00003132
Benjamin Peterson29060642009-01-31 22:14:21 +00003133 onError:
Guido van Rossumd57fd912000-03-10 22:53:23 +00003134 Py_XDECREF(buffer);
3135 return NULL;
3136}
3137
Alexander Belopolsky40018472011-02-26 01:02:56 +00003138PyObject *
3139PyUnicode_AsDecodedObject(PyObject *unicode,
Ezio Melotti2aa2b3b2011-09-29 00:58:57 +03003140 const char *encoding,
3141 const char *errors)
Marc-André Lemburgd2d45982004-07-08 17:57:32 +00003142{
3143 PyObject *v;
3144
3145 if (!PyUnicode_Check(unicode)) {
3146 PyErr_BadArgument();
3147 goto onError;
3148 }
3149
3150 if (encoding == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00003151 encoding = PyUnicode_GetDefaultEncoding();
Marc-André Lemburgd2d45982004-07-08 17:57:32 +00003152
3153 /* Decode via the codec registry */
3154 v = PyCodec_Decode(unicode, encoding, errors);
3155 if (v == NULL)
3156 goto onError;
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01003157 return unicode_result(v);
Marc-André Lemburgd2d45982004-07-08 17:57:32 +00003158
Benjamin Peterson29060642009-01-31 22:14:21 +00003159 onError:
Marc-André Lemburgd2d45982004-07-08 17:57:32 +00003160 return NULL;
3161}
3162
Alexander Belopolsky40018472011-02-26 01:02:56 +00003163PyObject *
3164PyUnicode_AsDecodedUnicode(PyObject *unicode,
Ezio Melotti2aa2b3b2011-09-29 00:58:57 +03003165 const char *encoding,
3166 const char *errors)
Marc-André Lemburgb2750b52008-06-06 12:18:17 +00003167{
3168 PyObject *v;
3169
3170 if (!PyUnicode_Check(unicode)) {
3171 PyErr_BadArgument();
3172 goto onError;
3173 }
3174
3175 if (encoding == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00003176 encoding = PyUnicode_GetDefaultEncoding();
Marc-André Lemburgb2750b52008-06-06 12:18:17 +00003177
3178 /* Decode via the codec registry */
3179 v = PyCodec_Decode(unicode, encoding, errors);
3180 if (v == NULL)
3181 goto onError;
3182 if (!PyUnicode_Check(v)) {
3183 PyErr_Format(PyExc_TypeError,
Benjamin Peterson142957c2008-07-04 19:55:29 +00003184 "decoder did not return a str object (type=%.400s)",
Marc-André Lemburgb2750b52008-06-06 12:18:17 +00003185 Py_TYPE(v)->tp_name);
3186 Py_DECREF(v);
3187 goto onError;
3188 }
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01003189 return unicode_result(v);
Marc-André Lemburgb2750b52008-06-06 12:18:17 +00003190
Benjamin Peterson29060642009-01-31 22:14:21 +00003191 onError:
Marc-André Lemburgb2750b52008-06-06 12:18:17 +00003192 return NULL;
3193}
3194
Alexander Belopolsky40018472011-02-26 01:02:56 +00003195PyObject *
3196PyUnicode_Encode(const Py_UNICODE *s,
Ezio Melotti2aa2b3b2011-09-29 00:58:57 +03003197 Py_ssize_t size,
3198 const char *encoding,
3199 const char *errors)
Guido van Rossumd57fd912000-03-10 22:53:23 +00003200{
3201 PyObject *v, *unicode;
Tim Petersced69f82003-09-16 20:30:58 +00003202
Guido van Rossumd57fd912000-03-10 22:53:23 +00003203 unicode = PyUnicode_FromUnicode(s, size);
3204 if (unicode == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00003205 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00003206 v = PyUnicode_AsEncodedString(unicode, encoding, errors);
3207 Py_DECREF(unicode);
3208 return v;
3209}
3210
Alexander Belopolsky40018472011-02-26 01:02:56 +00003211PyObject *
3212PyUnicode_AsEncodedObject(PyObject *unicode,
Ezio Melotti2aa2b3b2011-09-29 00:58:57 +03003213 const char *encoding,
3214 const char *errors)
Marc-André Lemburgd2d45982004-07-08 17:57:32 +00003215{
3216 PyObject *v;
3217
3218 if (!PyUnicode_Check(unicode)) {
3219 PyErr_BadArgument();
3220 goto onError;
3221 }
3222
3223 if (encoding == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00003224 encoding = PyUnicode_GetDefaultEncoding();
Marc-André Lemburgd2d45982004-07-08 17:57:32 +00003225
3226 /* Encode via the codec registry */
3227 v = PyCodec_Encode(unicode, encoding, errors);
3228 if (v == NULL)
3229 goto onError;
3230 return v;
3231
Benjamin Peterson29060642009-01-31 22:14:21 +00003232 onError:
Marc-André Lemburgd2d45982004-07-08 17:57:32 +00003233 return NULL;
3234}
3235
Victor Stinnerf2ea71f2011-12-17 04:13:41 +01003236static size_t
3237wcstombs_errorpos(const wchar_t *wstr)
3238{
3239 size_t len;
3240#if SIZEOF_WCHAR_T == 2
3241 wchar_t buf[3];
3242#else
3243 wchar_t buf[2];
3244#endif
3245 char outbuf[MB_LEN_MAX];
3246 const wchar_t *start, *previous;
Victor Stinnerf2ea71f2011-12-17 04:13:41 +01003247
Victor Stinnerf2ea71f2011-12-17 04:13:41 +01003248#if SIZEOF_WCHAR_T == 2
3249 buf[2] = 0;
3250#else
3251 buf[1] = 0;
3252#endif
3253 start = wstr;
3254 while (*wstr != L'\0')
3255 {
3256 previous = wstr;
3257#if SIZEOF_WCHAR_T == 2
3258 if (Py_UNICODE_IS_HIGH_SURROGATE(wstr[0])
3259 && Py_UNICODE_IS_LOW_SURROGATE(wstr[1]))
3260 {
3261 buf[0] = wstr[0];
3262 buf[1] = wstr[1];
3263 wstr += 2;
3264 }
3265 else {
3266 buf[0] = *wstr;
3267 buf[1] = 0;
3268 wstr++;
3269 }
3270#else
3271 buf[0] = *wstr;
3272 wstr++;
3273#endif
3274 len = wcstombs(outbuf, buf, sizeof(outbuf));
Victor Stinner2f197072011-12-17 07:08:30 +01003275 if (len == (size_t)-1)
Victor Stinnerf2ea71f2011-12-17 04:13:41 +01003276 return previous - start;
Victor Stinnerf2ea71f2011-12-17 04:13:41 +01003277 }
3278
3279 /* failed to find the unencodable character */
Victor Stinnerf2ea71f2011-12-17 04:13:41 +01003280 return 0;
3281}
3282
Victor Stinner1b579672011-12-17 05:47:23 +01003283static int
3284locale_error_handler(const char *errors, int *surrogateescape)
3285{
3286 if (errors == NULL) {
3287 *surrogateescape = 0;
3288 return 0;
3289 }
3290
3291 if (strcmp(errors, "strict") == 0) {
3292 *surrogateescape = 0;
3293 return 0;
3294 }
3295 if (strcmp(errors, "surrogateescape") == 0) {
3296 *surrogateescape = 1;
3297 return 0;
3298 }
3299 PyErr_Format(PyExc_ValueError,
3300 "only 'strict' and 'surrogateescape' error handlers "
3301 "are supported, not '%s'",
3302 errors);
3303 return -1;
3304}
3305
Victor Stinnerf2ea71f2011-12-17 04:13:41 +01003306PyObject *
Victor Stinner1b579672011-12-17 05:47:23 +01003307PyUnicode_EncodeLocale(PyObject *unicode, const char *errors)
Victor Stinnerf2ea71f2011-12-17 04:13:41 +01003308{
3309 Py_ssize_t wlen, wlen2;
3310 wchar_t *wstr;
3311 PyObject *bytes = NULL;
3312 char *errmsg;
Victor Stinner1f33f2b2011-12-17 04:45:09 +01003313 PyObject *reason;
Victor Stinnerf2ea71f2011-12-17 04:13:41 +01003314 PyObject *exc;
3315 size_t error_pos;
Victor Stinner1b579672011-12-17 05:47:23 +01003316 int surrogateescape;
3317
3318 if (locale_error_handler(errors, &surrogateescape) < 0)
3319 return NULL;
Victor Stinnerf2ea71f2011-12-17 04:13:41 +01003320
3321 wstr = PyUnicode_AsWideCharString(unicode, &wlen);
3322 if (wstr == NULL)
3323 return NULL;
3324
3325 wlen2 = wcslen(wstr);
3326 if (wlen2 != wlen) {
3327 PyMem_Free(wstr);
3328 PyErr_SetString(PyExc_TypeError, "embedded null character");
3329 return NULL;
3330 }
3331
3332 if (surrogateescape) {
3333 /* locale encoding with surrogateescape */
3334 char *str;
3335
3336 str = _Py_wchar2char(wstr, &error_pos);
3337 if (str == NULL) {
3338 if (error_pos == (size_t)-1) {
3339 PyErr_NoMemory();
3340 PyMem_Free(wstr);
3341 return NULL;
3342 }
3343 else {
3344 goto encode_error;
3345 }
3346 }
3347 PyMem_Free(wstr);
3348
3349 bytes = PyBytes_FromString(str);
3350 PyMem_Free(str);
3351 }
3352 else {
3353 size_t len, len2;
3354
3355 len = wcstombs(NULL, wstr, 0);
3356 if (len == (size_t)-1) {
Victor Stinner2f197072011-12-17 07:08:30 +01003357 error_pos = (size_t)-1;
Victor Stinnerf2ea71f2011-12-17 04:13:41 +01003358 goto encode_error;
3359 }
3360
3361 bytes = PyBytes_FromStringAndSize(NULL, len);
3362 if (bytes == NULL) {
3363 PyMem_Free(wstr);
3364 return NULL;
3365 }
3366
3367 len2 = wcstombs(PyBytes_AS_STRING(bytes), wstr, len+1);
3368 if (len2 == (size_t)-1 || len2 > len) {
Victor Stinner2f197072011-12-17 07:08:30 +01003369 error_pos = (size_t)-1;
Victor Stinnerf2ea71f2011-12-17 04:13:41 +01003370 goto encode_error;
3371 }
3372 PyMem_Free(wstr);
3373 }
3374 return bytes;
3375
3376encode_error:
3377 errmsg = strerror(errno);
3378 assert(errmsg != NULL);
Victor Stinner2f197072011-12-17 07:08:30 +01003379
3380 if (error_pos == (size_t)-1)
3381 error_pos = wcstombs_errorpos(wstr);
3382
Victor Stinnerf2ea71f2011-12-17 04:13:41 +01003383 PyMem_Free(wstr);
3384 Py_XDECREF(bytes);
3385
Victor Stinner2f197072011-12-17 07:08:30 +01003386 if (errmsg != NULL) {
3387 size_t errlen;
3388 wstr = _Py_char2wchar(errmsg, &errlen);
3389 if (wstr != NULL) {
3390 reason = PyUnicode_FromWideChar(wstr, errlen);
3391 PyMem_Free(wstr);
3392 } else
3393 errmsg = NULL;
3394 }
3395 if (errmsg == NULL)
Victor Stinner1f33f2b2011-12-17 04:45:09 +01003396 reason = PyUnicode_FromString(
3397 "wcstombs() encountered an unencodable "
3398 "wide character");
3399 if (reason == NULL)
3400 return NULL;
3401
3402 exc = PyObject_CallFunction(PyExc_UnicodeEncodeError, "sOnnO",
3403 "locale", unicode,
3404 (Py_ssize_t)error_pos,
3405 (Py_ssize_t)(error_pos+1),
3406 reason);
3407 Py_DECREF(reason);
3408 if (exc != NULL) {
3409 PyCodec_StrictErrors(exc);
3410 Py_XDECREF(exc);
3411 }
Victor Stinnerf2ea71f2011-12-17 04:13:41 +01003412 return NULL;
3413}
3414
Victor Stinnerad158722010-10-27 00:25:46 +00003415PyObject *
3416PyUnicode_EncodeFSDefault(PyObject *unicode)
Victor Stinnerae6265f2010-05-15 16:27:27 +00003417{
Victor Stinner99b95382011-07-04 14:23:54 +02003418#ifdef HAVE_MBCS
Victor Stinnerac931b12011-11-20 18:27:03 +01003419 return PyUnicode_EncodeCodePage(CP_ACP, unicode, NULL);
Victor Stinnerad158722010-10-27 00:25:46 +00003420#elif defined(__APPLE__)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003421 return _PyUnicode_AsUTF8String(unicode, "surrogateescape");
Victor Stinnerad158722010-10-27 00:25:46 +00003422#else
Victor Stinner793b5312011-04-27 00:24:21 +02003423 PyInterpreterState *interp = PyThreadState_GET()->interp;
3424 /* Bootstrap check: if the filesystem codec is implemented in Python, we
3425 cannot use it to encode and decode filenames before it is loaded. Load
3426 the Python codec requires to encode at least its own filename. Use the C
3427 version of the locale codec until the codec registry is initialized and
3428 the Python codec is loaded.
3429
3430 Py_FileSystemDefaultEncoding is shared between all interpreters, we
3431 cannot only rely on it: check also interp->fscodec_initialized for
3432 subinterpreters. */
3433 if (Py_FileSystemDefaultEncoding && interp->fscodec_initialized) {
Victor Stinnerae6265f2010-05-15 16:27:27 +00003434 return PyUnicode_AsEncodedString(unicode,
3435 Py_FileSystemDefaultEncoding,
3436 "surrogateescape");
Victor Stinnerc39211f2010-09-29 16:35:47 +00003437 }
3438 else {
Victor Stinner1b579672011-12-17 05:47:23 +01003439 return PyUnicode_EncodeLocale(unicode, "surrogateescape");
Victor Stinnerc39211f2010-09-29 16:35:47 +00003440 }
Victor Stinnerad158722010-10-27 00:25:46 +00003441#endif
Victor Stinnerae6265f2010-05-15 16:27:27 +00003442}
3443
Alexander Belopolsky40018472011-02-26 01:02:56 +00003444PyObject *
3445PyUnicode_AsEncodedString(PyObject *unicode,
Ezio Melotti2aa2b3b2011-09-29 00:58:57 +03003446 const char *encoding,
3447 const char *errors)
Guido van Rossumd57fd912000-03-10 22:53:23 +00003448{
3449 PyObject *v;
Victor Stinner600d3be2010-06-10 12:00:55 +00003450 char lower[11]; /* Enough for any encoding shortcut */
Tim Petersced69f82003-09-16 20:30:58 +00003451
Guido van Rossumd57fd912000-03-10 22:53:23 +00003452 if (!PyUnicode_Check(unicode)) {
3453 PyErr_BadArgument();
Amaury Forgeot d'Arcf0481112008-09-05 20:48:47 +00003454 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00003455 }
Fred Drakee4315f52000-05-09 19:53:39 +00003456
Fred Drakee4315f52000-05-09 19:53:39 +00003457 /* Shortcuts for common default encodings */
Victor Stinner37296e82010-06-10 13:36:23 +00003458 if (normalize_encoding(encoding, lower, sizeof(lower))) {
Alexander Belopolsky1d521462011-02-25 19:19:57 +00003459 if ((strcmp(lower, "utf-8") == 0) ||
3460 (strcmp(lower, "utf8") == 0))
Victor Stinnera5c68c32011-03-02 01:03:14 +00003461 {
Victor Stinner2f283c22011-03-02 01:21:46 +00003462 if (errors == NULL || strcmp(errors, "strict") == 0)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003463 return _PyUnicode_AsUTF8String(unicode, NULL);
Victor Stinner2f283c22011-03-02 01:21:46 +00003464 else
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003465 return _PyUnicode_AsUTF8String(unicode, errors);
Victor Stinnera5c68c32011-03-02 01:03:14 +00003466 }
Victor Stinner37296e82010-06-10 13:36:23 +00003467 else if ((strcmp(lower, "latin-1") == 0) ||
Alexander Belopolsky1d521462011-02-25 19:19:57 +00003468 (strcmp(lower, "latin1") == 0) ||
Victor Stinner37296e82010-06-10 13:36:23 +00003469 (strcmp(lower, "iso-8859-1") == 0))
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003470 return _PyUnicode_AsLatin1String(unicode, errors);
Victor Stinner99b95382011-07-04 14:23:54 +02003471#ifdef HAVE_MBCS
Victor Stinnerac931b12011-11-20 18:27:03 +01003472 else if (strcmp(lower, "mbcs") == 0)
3473 return PyUnicode_EncodeCodePage(CP_ACP, unicode, errors);
Mark Hammond0ccda1e2003-07-01 00:13:27 +00003474#endif
Victor Stinner37296e82010-06-10 13:36:23 +00003475 else if (strcmp(lower, "ascii") == 0)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003476 return _PyUnicode_AsASCIIString(unicode, errors);
Victor Stinner37296e82010-06-10 13:36:23 +00003477 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00003478
3479 /* Encode via the codec registry */
3480 v = PyCodec_Encode(unicode, encoding, errors);
3481 if (v == NULL)
Amaury Forgeot d'Arcf0481112008-09-05 20:48:47 +00003482 return NULL;
3483
3484 /* The normal path */
3485 if (PyBytes_Check(v))
3486 return v;
3487
3488 /* If the codec returns a buffer, raise a warning and convert to bytes */
Marc-André Lemburgb2750b52008-06-06 12:18:17 +00003489 if (PyByteArray_Check(v)) {
Victor Stinner4a2b7a12010-08-13 14:03:48 +00003490 int error;
Amaury Forgeot d'Arcf0481112008-09-05 20:48:47 +00003491 PyObject *b;
Victor Stinner4a2b7a12010-08-13 14:03:48 +00003492
3493 error = PyErr_WarnFormat(PyExc_RuntimeWarning, 1,
3494 "encoder %s returned bytearray instead of bytes",
3495 encoding);
3496 if (error) {
Amaury Forgeot d'Arcf0481112008-09-05 20:48:47 +00003497 Py_DECREF(v);
3498 return NULL;
Marc-André Lemburgb2750b52008-06-06 12:18:17 +00003499 }
Marc-André Lemburgb2750b52008-06-06 12:18:17 +00003500
Amaury Forgeot d'Arcf0481112008-09-05 20:48:47 +00003501 b = PyBytes_FromStringAndSize(PyByteArray_AS_STRING(v), Py_SIZE(v));
3502 Py_DECREF(v);
3503 return b;
3504 }
3505
3506 PyErr_Format(PyExc_TypeError,
3507 "encoder did not return a bytes object (type=%.400s)",
3508 Py_TYPE(v)->tp_name);
3509 Py_DECREF(v);
Marc-André Lemburgb2750b52008-06-06 12:18:17 +00003510 return NULL;
3511}
3512
Alexander Belopolsky40018472011-02-26 01:02:56 +00003513PyObject *
3514PyUnicode_AsEncodedUnicode(PyObject *unicode,
Ezio Melotti2aa2b3b2011-09-29 00:58:57 +03003515 const char *encoding,
3516 const char *errors)
Marc-André Lemburgb2750b52008-06-06 12:18:17 +00003517{
3518 PyObject *v;
3519
3520 if (!PyUnicode_Check(unicode)) {
3521 PyErr_BadArgument();
3522 goto onError;
3523 }
3524
3525 if (encoding == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00003526 encoding = PyUnicode_GetDefaultEncoding();
Marc-André Lemburgb2750b52008-06-06 12:18:17 +00003527
3528 /* Encode via the codec registry */
3529 v = PyCodec_Encode(unicode, encoding, errors);
3530 if (v == NULL)
3531 goto onError;
3532 if (!PyUnicode_Check(v)) {
3533 PyErr_Format(PyExc_TypeError,
Benjamin Peterson142957c2008-07-04 19:55:29 +00003534 "encoder did not return an str object (type=%.400s)",
Marc-André Lemburgb2750b52008-06-06 12:18:17 +00003535 Py_TYPE(v)->tp_name);
3536 Py_DECREF(v);
3537 goto onError;
3538 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00003539 return v;
Tim Petersced69f82003-09-16 20:30:58 +00003540
Benjamin Peterson29060642009-01-31 22:14:21 +00003541 onError:
Guido van Rossumd57fd912000-03-10 22:53:23 +00003542 return NULL;
3543}
3544
Victor Stinner2f197072011-12-17 07:08:30 +01003545static size_t
3546mbstowcs_errorpos(const char *str, size_t len)
3547{
3548#ifdef HAVE_MBRTOWC
3549 const char *start = str;
3550 mbstate_t mbs;
3551 size_t converted;
3552 wchar_t ch;
3553
3554 memset(&mbs, 0, sizeof mbs);
3555 while (len)
3556 {
3557 converted = mbrtowc(&ch, (char*)str, len, &mbs);
3558 if (converted == 0)
3559 /* Reached end of string */
3560 break;
3561 if (converted == (size_t)-1 || converted == (size_t)-2) {
3562 /* Conversion error or incomplete character */
3563 return str - start;
3564 }
3565 else {
3566 str += converted;
3567 len -= converted;
3568 }
3569 }
3570 /* failed to find the undecodable byte sequence */
3571 return 0;
3572#endif
3573 return 0;
3574}
3575
Guido van Rossum00bc0e02007-10-15 02:52:41 +00003576PyObject*
Victor Stinneraf02e1c2011-12-16 23:56:01 +01003577PyUnicode_DecodeLocaleAndSize(const char *str, Py_ssize_t len,
Victor Stinner1b579672011-12-17 05:47:23 +01003578 const char *errors)
Victor Stinneraf02e1c2011-12-16 23:56:01 +01003579{
3580 wchar_t smallbuf[256];
3581 size_t smallbuf_len = Py_ARRAY_LENGTH(smallbuf);
3582 wchar_t *wstr;
3583 size_t wlen, wlen2;
3584 PyObject *unicode;
Victor Stinner1b579672011-12-17 05:47:23 +01003585 int surrogateescape;
Victor Stinner2f197072011-12-17 07:08:30 +01003586 size_t error_pos;
3587 char *errmsg;
3588 PyObject *reason, *exc;
Victor Stinner1b579672011-12-17 05:47:23 +01003589
3590 if (locale_error_handler(errors, &surrogateescape) < 0)
3591 return NULL;
Victor Stinneraf02e1c2011-12-16 23:56:01 +01003592
3593 if (str[len] != '\0' || len != strlen(str)) {
3594 PyErr_SetString(PyExc_TypeError, "embedded null character");
3595 return NULL;
3596 }
3597
3598 if (surrogateescape)
3599 {
3600 wstr = _Py_char2wchar(str, &wlen);
3601 if (wstr == NULL) {
3602 if (wlen == (size_t)-1)
3603 PyErr_NoMemory();
3604 else
3605 PyErr_SetFromErrno(PyExc_OSError);
3606 return NULL;
3607 }
3608
3609 unicode = PyUnicode_FromWideChar(wstr, wlen);
3610 PyMem_Free(wstr);
3611 }
3612 else {
3613#ifndef HAVE_BROKEN_MBSTOWCS
3614 wlen = mbstowcs(NULL, str, 0);
3615#else
3616 wlen = len;
3617#endif
Victor Stinner2f197072011-12-17 07:08:30 +01003618 if (wlen == (size_t)-1)
3619 goto decode_error;
Victor Stinneraf02e1c2011-12-16 23:56:01 +01003620 if (wlen+1 <= smallbuf_len) {
3621 wstr = smallbuf;
3622 }
3623 else {
3624 if (wlen > PY_SSIZE_T_MAX / sizeof(wchar_t) - 1)
3625 return PyErr_NoMemory();
3626
3627 wstr = PyMem_Malloc((wlen+1) * sizeof(wchar_t));
3628 if (!wstr)
3629 return PyErr_NoMemory();
3630 }
3631
3632 /* This shouldn't fail now */
3633 wlen2 = mbstowcs(wstr, str, wlen+1);
3634 if (wlen2 == (size_t)-1) {
3635 if (wstr != smallbuf)
3636 PyMem_Free(wstr);
Victor Stinner2f197072011-12-17 07:08:30 +01003637 goto decode_error;
Victor Stinneraf02e1c2011-12-16 23:56:01 +01003638 }
3639#ifdef HAVE_BROKEN_MBSTOWCS
3640 assert(wlen2 == wlen);
3641#endif
3642 unicode = PyUnicode_FromWideChar(wstr, wlen2);
3643 if (wstr != smallbuf)
3644 PyMem_Free(wstr);
3645 }
3646 return unicode;
Victor Stinner2f197072011-12-17 07:08:30 +01003647
3648decode_error:
3649 errmsg = strerror(errno);
3650 assert(errmsg != NULL);
3651
3652 error_pos = mbstowcs_errorpos(str, len);
3653 if (errmsg != NULL) {
3654 size_t errlen;
3655 wstr = _Py_char2wchar(errmsg, &errlen);
3656 if (wstr != NULL) {
3657 reason = PyUnicode_FromWideChar(wstr, errlen);
3658 PyMem_Free(wstr);
3659 } else
3660 errmsg = NULL;
3661 }
3662 if (errmsg == NULL)
3663 reason = PyUnicode_FromString(
3664 "mbstowcs() encountered an invalid multibyte sequence");
3665 if (reason == NULL)
3666 return NULL;
3667
3668 exc = PyObject_CallFunction(PyExc_UnicodeDecodeError, "sy#nnO",
3669 "locale", str, len,
3670 (Py_ssize_t)error_pos,
3671 (Py_ssize_t)(error_pos+1),
3672 reason);
3673 Py_DECREF(reason);
3674 if (exc != NULL) {
3675 PyCodec_StrictErrors(exc);
3676 Py_XDECREF(exc);
3677 }
3678 return NULL;
Victor Stinneraf02e1c2011-12-16 23:56:01 +01003679}
3680
3681PyObject*
Victor Stinner1b579672011-12-17 05:47:23 +01003682PyUnicode_DecodeLocale(const char *str, const char *errors)
Victor Stinneraf02e1c2011-12-16 23:56:01 +01003683{
3684 Py_ssize_t size = (Py_ssize_t)strlen(str);
Victor Stinner1b579672011-12-17 05:47:23 +01003685 return PyUnicode_DecodeLocaleAndSize(str, size, errors);
Victor Stinneraf02e1c2011-12-16 23:56:01 +01003686}
3687
3688
3689PyObject*
Christian Heimes5894ba72007-11-04 11:43:14 +00003690PyUnicode_DecodeFSDefault(const char *s) {
Guido van Rossum00bc0e02007-10-15 02:52:41 +00003691 Py_ssize_t size = (Py_ssize_t)strlen(s);
Christian Heimes5894ba72007-11-04 11:43:14 +00003692 return PyUnicode_DecodeFSDefaultAndSize(s, size);
3693}
Guido van Rossum00bc0e02007-10-15 02:52:41 +00003694
Christian Heimes5894ba72007-11-04 11:43:14 +00003695PyObject*
3696PyUnicode_DecodeFSDefaultAndSize(const char *s, Py_ssize_t size)
3697{
Victor Stinner99b95382011-07-04 14:23:54 +02003698#ifdef HAVE_MBCS
Victor Stinnerad158722010-10-27 00:25:46 +00003699 return PyUnicode_DecodeMBCS(s, size, NULL);
3700#elif defined(__APPLE__)
Victor Stinnera1d12bb2011-12-11 21:53:09 +01003701 return PyUnicode_DecodeUTF8Stateful(s, size, "surrogateescape", NULL);
Victor Stinnerad158722010-10-27 00:25:46 +00003702#else
Victor Stinner793b5312011-04-27 00:24:21 +02003703 PyInterpreterState *interp = PyThreadState_GET()->interp;
3704 /* Bootstrap check: if the filesystem codec is implemented in Python, we
3705 cannot use it to encode and decode filenames before it is loaded. Load
3706 the Python codec requires to encode at least its own filename. Use the C
3707 version of the locale codec until the codec registry is initialized and
3708 the Python codec is loaded.
3709
3710 Py_FileSystemDefaultEncoding is shared between all interpreters, we
3711 cannot only rely on it: check also interp->fscodec_initialized for
3712 subinterpreters. */
3713 if (Py_FileSystemDefaultEncoding && interp->fscodec_initialized) {
Guido van Rossum00bc0e02007-10-15 02:52:41 +00003714 return PyUnicode_Decode(s, size,
3715 Py_FileSystemDefaultEncoding,
Victor Stinnerb9a20ad2010-04-30 16:37:52 +00003716 "surrogateescape");
Guido van Rossum00bc0e02007-10-15 02:52:41 +00003717 }
3718 else {
Victor Stinner1b579672011-12-17 05:47:23 +01003719 return PyUnicode_DecodeLocaleAndSize(s, size, "surrogateescape");
Guido van Rossum00bc0e02007-10-15 02:52:41 +00003720 }
Victor Stinnerad158722010-10-27 00:25:46 +00003721#endif
Guido van Rossum00bc0e02007-10-15 02:52:41 +00003722}
3723
Martin v. Löwis011e8422009-05-05 04:43:17 +00003724
3725int
Antoine Pitrou13348842012-01-29 18:36:34 +01003726_PyUnicode_HasNULChars(PyObject* s)
3727{
3728 static PyObject *nul = NULL;
3729
3730 if (nul == NULL)
3731 nul = PyUnicode_FromStringAndSize("\0", 1);
3732 if (nul == NULL)
3733 return -1;
3734 return PyUnicode_Contains(s, nul);
3735}
3736
3737
3738int
Martin v. Löwis011e8422009-05-05 04:43:17 +00003739PyUnicode_FSConverter(PyObject* arg, void* addr)
3740{
3741 PyObject *output = NULL;
3742 Py_ssize_t size;
3743 void *data;
Martin v. Löwisc15bdef2009-05-29 14:47:46 +00003744 if (arg == NULL) {
3745 Py_DECREF(*(PyObject**)addr);
3746 return 1;
3747 }
Victor Stinnerdcb24032010-04-22 12:08:36 +00003748 if (PyBytes_Check(arg)) {
Martin v. Löwis011e8422009-05-05 04:43:17 +00003749 output = arg;
3750 Py_INCREF(output);
3751 }
3752 else {
3753 arg = PyUnicode_FromObject(arg);
3754 if (!arg)
3755 return 0;
Victor Stinnerae6265f2010-05-15 16:27:27 +00003756 output = PyUnicode_EncodeFSDefault(arg);
Martin v. Löwis011e8422009-05-05 04:43:17 +00003757 Py_DECREF(arg);
3758 if (!output)
3759 return 0;
3760 if (!PyBytes_Check(output)) {
3761 Py_DECREF(output);
3762 PyErr_SetString(PyExc_TypeError, "encoder failed to return bytes");
3763 return 0;
3764 }
3765 }
Victor Stinner0ea2a462010-04-30 00:22:08 +00003766 size = PyBytes_GET_SIZE(output);
3767 data = PyBytes_AS_STRING(output);
Martin v. Löwis011e8422009-05-05 04:43:17 +00003768 if (size != strlen(data)) {
Benjamin Peterson7a6b44a2011-08-18 13:51:47 -05003769 PyErr_SetString(PyExc_TypeError, "embedded NUL character");
Martin v. Löwis011e8422009-05-05 04:43:17 +00003770 Py_DECREF(output);
3771 return 0;
3772 }
3773 *(PyObject**)addr = output;
Martin v. Löwisc15bdef2009-05-29 14:47:46 +00003774 return Py_CLEANUP_SUPPORTED;
Martin v. Löwis011e8422009-05-05 04:43:17 +00003775}
3776
3777
Victor Stinner47fcb5b2010-08-13 23:59:58 +00003778int
3779PyUnicode_FSDecoder(PyObject* arg, void* addr)
3780{
3781 PyObject *output = NULL;
Victor Stinner47fcb5b2010-08-13 23:59:58 +00003782 if (arg == NULL) {
3783 Py_DECREF(*(PyObject**)addr);
3784 return 1;
3785 }
3786 if (PyUnicode_Check(arg)) {
Benjamin Petersonbac79492012-01-14 13:34:47 -05003787 if (PyUnicode_READY(arg) == -1)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003788 return 0;
Victor Stinner47fcb5b2010-08-13 23:59:58 +00003789 output = arg;
3790 Py_INCREF(output);
3791 }
3792 else {
3793 arg = PyBytes_FromObject(arg);
3794 if (!arg)
3795 return 0;
3796 output = PyUnicode_DecodeFSDefaultAndSize(PyBytes_AS_STRING(arg),
3797 PyBytes_GET_SIZE(arg));
3798 Py_DECREF(arg);
3799 if (!output)
3800 return 0;
3801 if (!PyUnicode_Check(output)) {
3802 Py_DECREF(output);
3803 PyErr_SetString(PyExc_TypeError, "decoder failed to return unicode");
3804 return 0;
3805 }
3806 }
Benjamin Petersonbac79492012-01-14 13:34:47 -05003807 if (PyUnicode_READY(output) == -1) {
Victor Stinner065836e2011-10-27 01:56:33 +02003808 Py_DECREF(output);
3809 return 0;
3810 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003811 if (findchar(PyUnicode_DATA(output), PyUnicode_KIND(output),
Antoine Pitrouf0b934b2011-10-13 18:55:09 +02003812 PyUnicode_GET_LENGTH(output), 0, 1) >= 0) {
Victor Stinner47fcb5b2010-08-13 23:59:58 +00003813 PyErr_SetString(PyExc_TypeError, "embedded NUL character");
3814 Py_DECREF(output);
3815 return 0;
3816 }
3817 *(PyObject**)addr = output;
3818 return Py_CLEANUP_SUPPORTED;
3819}
3820
3821
Martin v. Löwis5b222132007-06-10 09:51:05 +00003822char*
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003823PyUnicode_AsUTF8AndSize(PyObject *unicode, Py_ssize_t *psize)
Martin v. Löwis5b222132007-06-10 09:51:05 +00003824{
Christian Heimesf3863112007-11-22 07:46:41 +00003825 PyObject *bytes;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003826
Neal Norwitze0a0a6e2007-08-25 01:04:21 +00003827 if (!PyUnicode_Check(unicode)) {
3828 PyErr_BadArgument();
3829 return NULL;
3830 }
Victor Stinner9db1a8b2011-10-23 20:04:37 +02003831 if (PyUnicode_READY(unicode) == -1)
Martin v. Löwis5b222132007-06-10 09:51:05 +00003832 return NULL;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003833
Victor Stinnere90fe6a2011-10-01 16:48:13 +02003834 if (PyUnicode_UTF8(unicode) == NULL) {
3835 assert(!PyUnicode_IS_COMPACT_ASCII(unicode));
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003836 bytes = _PyUnicode_AsUTF8String(unicode, "strict");
3837 if (bytes == NULL)
3838 return NULL;
Victor Stinner9db1a8b2011-10-23 20:04:37 +02003839 _PyUnicode_UTF8(unicode) = PyObject_MALLOC(PyBytes_GET_SIZE(bytes) + 1);
3840 if (_PyUnicode_UTF8(unicode) == NULL) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003841 Py_DECREF(bytes);
3842 return NULL;
3843 }
Victor Stinner9db1a8b2011-10-23 20:04:37 +02003844 _PyUnicode_UTF8_LENGTH(unicode) = PyBytes_GET_SIZE(bytes);
3845 Py_MEMCPY(_PyUnicode_UTF8(unicode),
3846 PyBytes_AS_STRING(bytes),
3847 _PyUnicode_UTF8_LENGTH(unicode) + 1);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003848 Py_DECREF(bytes);
3849 }
3850
3851 if (psize)
Victor Stinnere90fe6a2011-10-01 16:48:13 +02003852 *psize = PyUnicode_UTF8_LENGTH(unicode);
3853 return PyUnicode_UTF8(unicode);
Guido van Rossum7d1df6c2007-08-29 13:53:23 +00003854}
3855
3856char*
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003857PyUnicode_AsUTF8(PyObject *unicode)
Guido van Rossum7d1df6c2007-08-29 13:53:23 +00003858{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003859 return PyUnicode_AsUTF8AndSize(unicode, NULL);
3860}
3861
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003862Py_UNICODE *
3863PyUnicode_AsUnicodeAndSize(PyObject *unicode, Py_ssize_t *size)
3864{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003865 const unsigned char *one_byte;
3866#if SIZEOF_WCHAR_T == 4
3867 const Py_UCS2 *two_bytes;
3868#else
3869 const Py_UCS4 *four_bytes;
3870 const Py_UCS4 *ucs4_end;
3871 Py_ssize_t num_surrogates;
3872#endif
3873 wchar_t *w;
3874 wchar_t *wchar_end;
3875
3876 if (!PyUnicode_Check(unicode)) {
3877 PyErr_BadArgument();
3878 return NULL;
3879 }
Victor Stinner9db1a8b2011-10-23 20:04:37 +02003880 if (_PyUnicode_WSTR(unicode) == NULL) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003881 /* Non-ASCII compact unicode object */
Victor Stinner9db1a8b2011-10-23 20:04:37 +02003882 assert(_PyUnicode_KIND(unicode) != 0);
3883 assert(PyUnicode_IS_READY(unicode));
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003884
Victor Stinner9db1a8b2011-10-23 20:04:37 +02003885 if (PyUnicode_KIND(unicode) == PyUnicode_4BYTE_KIND) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003886#if SIZEOF_WCHAR_T == 2
Victor Stinner9db1a8b2011-10-23 20:04:37 +02003887 four_bytes = PyUnicode_4BYTE_DATA(unicode);
3888 ucs4_end = four_bytes + _PyUnicode_LENGTH(unicode);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003889 num_surrogates = 0;
3890
3891 for (; four_bytes < ucs4_end; ++four_bytes) {
3892 if (*four_bytes > 0xFFFF)
3893 ++num_surrogates;
3894 }
3895
Victor Stinner9db1a8b2011-10-23 20:04:37 +02003896 _PyUnicode_WSTR(unicode) = (wchar_t *) PyObject_MALLOC(
3897 sizeof(wchar_t) * (_PyUnicode_LENGTH(unicode) + 1 + num_surrogates));
3898 if (!_PyUnicode_WSTR(unicode)) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003899 PyErr_NoMemory();
3900 return NULL;
3901 }
Victor Stinner9db1a8b2011-10-23 20:04:37 +02003902 _PyUnicode_WSTR_LENGTH(unicode) = _PyUnicode_LENGTH(unicode) + num_surrogates;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003903
Victor Stinner9db1a8b2011-10-23 20:04:37 +02003904 w = _PyUnicode_WSTR(unicode);
3905 wchar_end = w + _PyUnicode_WSTR_LENGTH(unicode);
3906 four_bytes = PyUnicode_4BYTE_DATA(unicode);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003907 for (; four_bytes < ucs4_end; ++four_bytes, ++w) {
3908 if (*four_bytes > 0xFFFF) {
Victor Stinner8faf8212011-12-08 22:14:11 +01003909 assert(*four_bytes <= MAX_UNICODE);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003910 /* encode surrogate pair in this case */
Victor Stinner551ac952011-11-29 22:58:13 +01003911 *w++ = Py_UNICODE_HIGH_SURROGATE(*four_bytes);
3912 *w = Py_UNICODE_LOW_SURROGATE(*four_bytes);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003913 }
3914 else
3915 *w = *four_bytes;
3916
3917 if (w > wchar_end) {
3918 assert(0 && "Miscalculated string end");
3919 }
3920 }
3921 *w = 0;
3922#else
3923 /* sizeof(wchar_t) == 4 */
3924 Py_FatalError("Impossible unicode object state, wstr and str "
3925 "should share memory already.");
3926 return NULL;
3927#endif
3928 }
3929 else {
Victor Stinner9db1a8b2011-10-23 20:04:37 +02003930 _PyUnicode_WSTR(unicode) = (wchar_t *) PyObject_MALLOC(sizeof(wchar_t) *
3931 (_PyUnicode_LENGTH(unicode) + 1));
3932 if (!_PyUnicode_WSTR(unicode)) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003933 PyErr_NoMemory();
3934 return NULL;
3935 }
Victor Stinner9db1a8b2011-10-23 20:04:37 +02003936 if (!PyUnicode_IS_COMPACT_ASCII(unicode))
3937 _PyUnicode_WSTR_LENGTH(unicode) = _PyUnicode_LENGTH(unicode);
3938 w = _PyUnicode_WSTR(unicode);
3939 wchar_end = w + _PyUnicode_LENGTH(unicode);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003940
Victor Stinner9db1a8b2011-10-23 20:04:37 +02003941 if (PyUnicode_KIND(unicode) == PyUnicode_1BYTE_KIND) {
3942 one_byte = PyUnicode_1BYTE_DATA(unicode);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003943 for (; w < wchar_end; ++one_byte, ++w)
3944 *w = *one_byte;
3945 /* null-terminate the wstr */
3946 *w = 0;
3947 }
Victor Stinner9db1a8b2011-10-23 20:04:37 +02003948 else if (PyUnicode_KIND(unicode) == PyUnicode_2BYTE_KIND) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003949#if SIZEOF_WCHAR_T == 4
Victor Stinner9db1a8b2011-10-23 20:04:37 +02003950 two_bytes = PyUnicode_2BYTE_DATA(unicode);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003951 for (; w < wchar_end; ++two_bytes, ++w)
3952 *w = *two_bytes;
3953 /* null-terminate the wstr */
3954 *w = 0;
3955#else
3956 /* sizeof(wchar_t) == 2 */
Victor Stinner9db1a8b2011-10-23 20:04:37 +02003957 PyObject_FREE(_PyUnicode_WSTR(unicode));
3958 _PyUnicode_WSTR(unicode) = NULL;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003959 Py_FatalError("Impossible unicode object state, wstr "
3960 "and str should share memory already.");
3961 return NULL;
3962#endif
3963 }
3964 else {
3965 assert(0 && "This should never happen.");
3966 }
3967 }
3968 }
3969 if (size != NULL)
Victor Stinner9db1a8b2011-10-23 20:04:37 +02003970 *size = PyUnicode_WSTR_LENGTH(unicode);
3971 return _PyUnicode_WSTR(unicode);
Martin v. Löwis5b222132007-06-10 09:51:05 +00003972}
3973
Alexander Belopolsky40018472011-02-26 01:02:56 +00003974Py_UNICODE *
3975PyUnicode_AsUnicode(PyObject *unicode)
Guido van Rossumd57fd912000-03-10 22:53:23 +00003976{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003977 return PyUnicode_AsUnicodeAndSize(unicode, NULL);
Guido van Rossumd57fd912000-03-10 22:53:23 +00003978}
3979
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003980
Alexander Belopolsky40018472011-02-26 01:02:56 +00003981Py_ssize_t
3982PyUnicode_GetSize(PyObject *unicode)
Guido van Rossumd57fd912000-03-10 22:53:23 +00003983{
3984 if (!PyUnicode_Check(unicode)) {
3985 PyErr_BadArgument();
3986 goto onError;
3987 }
3988 return PyUnicode_GET_SIZE(unicode);
3989
Benjamin Peterson29060642009-01-31 22:14:21 +00003990 onError:
Guido van Rossumd57fd912000-03-10 22:53:23 +00003991 return -1;
3992}
3993
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003994Py_ssize_t
3995PyUnicode_GetLength(PyObject *unicode)
3996{
Victor Stinner07621332012-06-16 04:53:46 +02003997 if (!PyUnicode_Check(unicode)) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02003998 PyErr_BadArgument();
3999 return -1;
4000 }
Victor Stinner07621332012-06-16 04:53:46 +02004001 if (PyUnicode_READY(unicode) == -1)
4002 return -1;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02004003 return PyUnicode_GET_LENGTH(unicode);
4004}
4005
4006Py_UCS4
4007PyUnicode_ReadChar(PyObject *unicode, Py_ssize_t index)
4008{
Victor Stinner2fe5ced2011-10-02 00:25:40 +02004009 if (!PyUnicode_Check(unicode) || PyUnicode_READY(unicode) == -1) {
4010 PyErr_BadArgument();
4011 return (Py_UCS4)-1;
4012 }
Victor Stinnerc4b49542011-12-11 22:44:26 +01004013 if (index < 0 || index >= PyUnicode_GET_LENGTH(unicode)) {
Victor Stinner2fe5ced2011-10-02 00:25:40 +02004014 PyErr_SetString(PyExc_IndexError, "string index out of range");
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02004015 return (Py_UCS4)-1;
4016 }
4017 return PyUnicode_READ_CHAR(unicode, index);
4018}
4019
4020int
4021PyUnicode_WriteChar(PyObject *unicode, Py_ssize_t index, Py_UCS4 ch)
4022{
4023 if (!PyUnicode_Check(unicode) || !PyUnicode_IS_COMPACT(unicode)) {
Victor Stinnercd9950f2011-10-02 00:34:53 +02004024 PyErr_BadArgument();
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02004025 return -1;
4026 }
Victor Stinner488fa492011-12-12 00:01:39 +01004027 assert(PyUnicode_IS_READY(unicode));
Victor Stinnerc4b49542011-12-11 22:44:26 +01004028 if (index < 0 || index >= PyUnicode_GET_LENGTH(unicode)) {
Victor Stinnercd9950f2011-10-02 00:34:53 +02004029 PyErr_SetString(PyExc_IndexError, "string index out of range");
4030 return -1;
4031 }
Victor Stinner488fa492011-12-12 00:01:39 +01004032 if (unicode_check_modifiable(unicode))
Victor Stinnercd9950f2011-10-02 00:34:53 +02004033 return -1;
Victor Stinnerc9590ad2012-03-04 01:34:37 +01004034 if (ch > PyUnicode_MAX_CHAR_VALUE(unicode)) {
4035 PyErr_SetString(PyExc_ValueError, "character out of range");
4036 return -1;
4037 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02004038 PyUnicode_WRITE(PyUnicode_KIND(unicode), PyUnicode_DATA(unicode),
4039 index, ch);
4040 return 0;
4041}
4042
Alexander Belopolsky40018472011-02-26 01:02:56 +00004043const char *
4044PyUnicode_GetDefaultEncoding(void)
Fred Drakee4315f52000-05-09 19:53:39 +00004045{
Victor Stinner42cb4622010-09-01 19:39:01 +00004046 return "utf-8";
Fred Drakee4315f52000-05-09 19:53:39 +00004047}
4048
Victor Stinner554f3f02010-06-16 23:33:54 +00004049/* create or adjust a UnicodeDecodeError */
4050static void
4051make_decode_exception(PyObject **exceptionObject,
4052 const char *encoding,
4053 const char *input, Py_ssize_t length,
4054 Py_ssize_t startpos, Py_ssize_t endpos,
4055 const char *reason)
4056{
4057 if (*exceptionObject == NULL) {
4058 *exceptionObject = PyUnicodeDecodeError_Create(
4059 encoding, input, length, startpos, endpos, reason);
4060 }
4061 else {
4062 if (PyUnicodeDecodeError_SetStart(*exceptionObject, startpos))
4063 goto onError;
4064 if (PyUnicodeDecodeError_SetEnd(*exceptionObject, endpos))
4065 goto onError;
4066 if (PyUnicodeDecodeError_SetReason(*exceptionObject, reason))
4067 goto onError;
4068 }
4069 return;
4070
4071onError:
4072 Py_DECREF(*exceptionObject);
4073 *exceptionObject = NULL;
4074}
4075
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004076/* error handling callback helper:
4077 build arguments, call the callback and check the arguments,
Fred Drakedb390c12005-10-28 14:39:47 +00004078 if no exception occurred, copy the replacement to the output
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004079 and adjust various state variables.
4080 return 0 on success, -1 on error
4081*/
4082
Alexander Belopolsky40018472011-02-26 01:02:56 +00004083static int
4084unicode_decode_call_errorhandler(const char *errors, PyObject **errorHandler,
Ezio Melotti2aa2b3b2011-09-29 00:58:57 +03004085 const char *encoding, const char *reason,
4086 const char **input, const char **inend, Py_ssize_t *startinpos,
4087 Py_ssize_t *endinpos, PyObject **exceptionObject, const char **inptr,
Martin v. Löwise9b11c12011-11-08 17:35:34 +01004088 PyObject **output, Py_ssize_t *outpos)
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004089{
Benjamin Peterson142957c2008-07-04 19:55:29 +00004090 static char *argparse = "O!n;decoding error handler must return (str, int) tuple";
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004091
4092 PyObject *restuple = NULL;
4093 PyObject *repunicode = NULL;
Victor Stinner596a6c42011-11-09 00:02:18 +01004094 Py_ssize_t outsize;
Walter Dörwalde78178e2007-07-30 13:31:40 +00004095 Py_ssize_t insize;
Martin v. Löwis18e16552006-02-15 17:27:45 +00004096 Py_ssize_t requiredsize;
4097 Py_ssize_t newpos;
Walter Dörwalde78178e2007-07-30 13:31:40 +00004098 PyObject *inputobj = NULL;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004099 int res = -1;
4100
Victor Stinner596a6c42011-11-09 00:02:18 +01004101 if (_PyUnicode_KIND(*output) != PyUnicode_WCHAR_KIND)
4102 outsize = PyUnicode_GET_LENGTH(*output);
4103 else
4104 outsize = _PyUnicode_WSTR_LENGTH(*output);
4105
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004106 if (*errorHandler == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +00004107 *errorHandler = PyCodec_LookupError(errors);
4108 if (*errorHandler == NULL)
4109 goto onError;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004110 }
4111
Victor Stinner554f3f02010-06-16 23:33:54 +00004112 make_decode_exception(exceptionObject,
4113 encoding,
4114 *input, *inend - *input,
4115 *startinpos, *endinpos,
4116 reason);
4117 if (*exceptionObject == NULL)
4118 goto onError;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004119
4120 restuple = PyObject_CallFunctionObjArgs(*errorHandler, *exceptionObject, NULL);
4121 if (restuple == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00004122 goto onError;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004123 if (!PyTuple_Check(restuple)) {
Benjamin Petersond75fcb42009-02-19 04:22:03 +00004124 PyErr_SetString(PyExc_TypeError, &argparse[4]);
Benjamin Peterson29060642009-01-31 22:14:21 +00004125 goto onError;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004126 }
4127 if (!PyArg_ParseTuple(restuple, argparse, &PyUnicode_Type, &repunicode, &newpos))
Benjamin Peterson29060642009-01-31 22:14:21 +00004128 goto onError;
Benjamin Petersonbac79492012-01-14 13:34:47 -05004129 if (PyUnicode_READY(repunicode) == -1)
Martin v. Löwise9b11c12011-11-08 17:35:34 +01004130 goto onError;
Walter Dörwalde78178e2007-07-30 13:31:40 +00004131
4132 /* Copy back the bytes variables, which might have been modified by the
4133 callback */
4134 inputobj = PyUnicodeDecodeError_GetObject(*exceptionObject);
4135 if (!inputobj)
4136 goto onError;
Christian Heimes72b710a2008-05-26 13:28:38 +00004137 if (!PyBytes_Check(inputobj)) {
Benjamin Peterson29060642009-01-31 22:14:21 +00004138 PyErr_Format(PyExc_TypeError, "exception attribute object must be bytes");
Walter Dörwalde78178e2007-07-30 13:31:40 +00004139 }
Christian Heimes72b710a2008-05-26 13:28:38 +00004140 *input = PyBytes_AS_STRING(inputobj);
4141 insize = PyBytes_GET_SIZE(inputobj);
Walter Dörwalde78178e2007-07-30 13:31:40 +00004142 *inend = *input + insize;
Walter Dörwald36f938f2007-08-10 10:11:43 +00004143 /* we can DECREF safely, as the exception has another reference,
4144 so the object won't go away. */
4145 Py_DECREF(inputobj);
Walter Dörwalde78178e2007-07-30 13:31:40 +00004146
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004147 if (newpos<0)
Benjamin Peterson29060642009-01-31 22:14:21 +00004148 newpos = insize+newpos;
Walter Dörwald2e0b18a2003-01-31 17:19:08 +00004149 if (newpos<0 || newpos>insize) {
Benjamin Peterson29060642009-01-31 22:14:21 +00004150 PyErr_Format(PyExc_IndexError, "position %zd from error handler out of bounds", newpos);
4151 goto onError;
Walter Dörwald2e0b18a2003-01-31 17:19:08 +00004152 }
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004153
Victor Stinner596a6c42011-11-09 00:02:18 +01004154 if (_PyUnicode_KIND(*output) != PyUnicode_WCHAR_KIND) {
4155 /* need more space? (at least enough for what we
4156 have+the replacement+the rest of the string (starting
4157 at the new input position), so we won't have to check space
4158 when there are no errors in the rest of the string) */
4159 Py_ssize_t replen = PyUnicode_GET_LENGTH(repunicode);
4160 requiredsize = *outpos + replen + insize-newpos;
4161 if (requiredsize > outsize) {
4162 if (requiredsize<2*outsize)
4163 requiredsize = 2*outsize;
4164 if (unicode_resize(output, requiredsize) < 0)
4165 goto onError;
4166 }
Victor Stinner1b487b42012-05-03 12:29:04 +02004167 if (unicode_widen(output, *outpos,
4168 PyUnicode_MAX_CHAR_VALUE(repunicode)) < 0)
Benjamin Peterson29060642009-01-31 22:14:21 +00004169 goto onError;
Victor Stinnerd3f08822012-05-29 12:57:52 +02004170 _PyUnicode_FastCopyCharacters(*output, *outpos, repunicode, 0, replen);
Victor Stinner596a6c42011-11-09 00:02:18 +01004171 *outpos += replen;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004172 }
Victor Stinner596a6c42011-11-09 00:02:18 +01004173 else {
4174 wchar_t *repwstr;
4175 Py_ssize_t repwlen;
4176 repwstr = PyUnicode_AsUnicodeAndSize(repunicode, &repwlen);
4177 if (repwstr == NULL)
4178 goto onError;
4179 /* need more space? (at least enough for what we
4180 have+the replacement+the rest of the string (starting
4181 at the new input position), so we won't have to check space
4182 when there are no errors in the rest of the string) */
4183 requiredsize = *outpos + repwlen + insize-newpos;
4184 if (requiredsize > outsize) {
4185 if (requiredsize < 2*outsize)
4186 requiredsize = 2*outsize;
4187 if (unicode_resize(output, requiredsize) < 0)
4188 goto onError;
4189 }
4190 wcsncpy(_PyUnicode_WSTR(*output) + *outpos, repwstr, repwlen);
4191 *outpos += repwlen;
4192 }
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004193 *endinpos = newpos;
Walter Dörwalde78178e2007-07-30 13:31:40 +00004194 *inptr = *input + newpos;
Walter Dörwalde78178e2007-07-30 13:31:40 +00004195
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004196 /* we made it! */
4197 res = 0;
4198
Benjamin Peterson29060642009-01-31 22:14:21 +00004199 onError:
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004200 Py_XDECREF(restuple);
4201 return res;
4202}
4203
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004204/* --- UTF-7 Codec -------------------------------------------------------- */
4205
Antoine Pitrou244651a2009-05-04 18:56:13 +00004206/* See RFC2152 for details. We encode conservatively and decode liberally. */
4207
4208/* Three simple macros defining base-64. */
4209
4210/* Is c a base-64 character? */
4211
4212#define IS_BASE64(c) \
4213 (((c) >= 'A' && (c) <= 'Z') || \
4214 ((c) >= 'a' && (c) <= 'z') || \
4215 ((c) >= '0' && (c) <= '9') || \
4216 (c) == '+' || (c) == '/')
4217
4218/* given that c is a base-64 character, what is its base-64 value? */
4219
4220#define FROM_BASE64(c) \
4221 (((c) >= 'A' && (c) <= 'Z') ? (c) - 'A' : \
4222 ((c) >= 'a' && (c) <= 'z') ? (c) - 'a' + 26 : \
4223 ((c) >= '0' && (c) <= '9') ? (c) - '0' + 52 : \
4224 (c) == '+' ? 62 : 63)
4225
4226/* What is the base-64 character of the bottom 6 bits of n? */
4227
4228#define TO_BASE64(n) \
4229 ("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/"[(n) & 0x3f])
4230
4231/* DECODE_DIRECT: this byte encountered in a UTF-7 string should be
4232 * decoded as itself. We are permissive on decoding; the only ASCII
4233 * byte not decoding to itself is the + which begins a base64
4234 * string. */
4235
4236#define DECODE_DIRECT(c) \
4237 ((c) <= 127 && (c) != '+')
4238
4239/* The UTF-7 encoder treats ASCII characters differently according to
4240 * whether they are Set D, Set O, Whitespace, or special (i.e. none of
4241 * the above). See RFC2152. This array identifies these different
4242 * sets:
4243 * 0 : "Set D"
4244 * alphanumeric and '(),-./:?
4245 * 1 : "Set O"
4246 * !"#$%&*;<=>@[]^_`{|}
4247 * 2 : "whitespace"
4248 * ht nl cr sp
4249 * 3 : special (must be base64 encoded)
4250 * everything else (i.e. +\~ and non-printing codes 0-8 11-12 14-31 127)
4251 */
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004252
Tim Petersced69f82003-09-16 20:30:58 +00004253static
Antoine Pitrou244651a2009-05-04 18:56:13 +00004254char utf7_category[128] = {
4255/* nul soh stx etx eot enq ack bel bs ht nl vt np cr so si */
4256 3, 3, 3, 3, 3, 3, 3, 3, 3, 2, 2, 3, 3, 2, 3, 3,
4257/* dle dc1 dc2 dc3 dc4 nak syn etb can em sub esc fs gs rs us */
4258 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3,
4259/* sp ! " # $ % & ' ( ) * + , - . / */
4260 2, 1, 1, 1, 1, 1, 1, 0, 0, 0, 1, 3, 0, 0, 0, 0,
4261/* 0 1 2 3 4 5 6 7 8 9 : ; < = > ? */
4262 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 0,
4263/* @ A B C D E F G H I J K L M N O */
4264 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
4265/* P Q R S T U V W X Y Z [ \ ] ^ _ */
4266 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 3, 1, 1, 1,
4267/* ` a b c d e f g h i j k l m n o */
4268 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
4269/* p q r s t u v w x y z { | } ~ del */
4270 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 3, 3,
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004271};
4272
Antoine Pitrou244651a2009-05-04 18:56:13 +00004273/* ENCODE_DIRECT: this character should be encoded as itself. The
4274 * answer depends on whether we are encoding set O as itself, and also
4275 * on whether we are encoding whitespace as itself. RFC2152 makes it
4276 * clear that the answers to these questions vary between
4277 * applications, so this code needs to be flexible. */
Marc-André Lemburge115ec82005-10-19 22:33:31 +00004278
Antoine Pitrou244651a2009-05-04 18:56:13 +00004279#define ENCODE_DIRECT(c, directO, directWS) \
4280 ((c) < 128 && (c) > 0 && \
4281 ((utf7_category[(c)] == 0) || \
4282 (directWS && (utf7_category[(c)] == 2)) || \
4283 (directO && (utf7_category[(c)] == 1))))
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004284
Alexander Belopolsky40018472011-02-26 01:02:56 +00004285PyObject *
4286PyUnicode_DecodeUTF7(const char *s,
Ezio Melotti2aa2b3b2011-09-29 00:58:57 +03004287 Py_ssize_t size,
4288 const char *errors)
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004289{
Christian Heimes5d14c2b2007-11-20 23:38:09 +00004290 return PyUnicode_DecodeUTF7Stateful(s, size, errors, NULL);
4291}
4292
Antoine Pitrou244651a2009-05-04 18:56:13 +00004293/* The decoder. The only state we preserve is our read position,
4294 * i.e. how many characters we have consumed. So if we end in the
4295 * middle of a shift sequence we have to back off the read position
4296 * and the output to the beginning of the sequence, otherwise we lose
4297 * all the shift state (seen bits, number of bits seen, high
4298 * surrogate). */
4299
Alexander Belopolsky40018472011-02-26 01:02:56 +00004300PyObject *
4301PyUnicode_DecodeUTF7Stateful(const char *s,
Ezio Melotti2aa2b3b2011-09-29 00:58:57 +03004302 Py_ssize_t size,
4303 const char *errors,
4304 Py_ssize_t *consumed)
Christian Heimes5d14c2b2007-11-20 23:38:09 +00004305{
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004306 const char *starts = s;
Martin v. Löwis18e16552006-02-15 17:27:45 +00004307 Py_ssize_t startinpos;
4308 Py_ssize_t endinpos;
4309 Py_ssize_t outpos;
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004310 const char *e;
Victor Stinner7931d9a2011-11-04 00:22:48 +01004311 PyObject *unicode;
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004312 const char *errmsg = "";
4313 int inShift = 0;
Martin v. Löwise9b11c12011-11-08 17:35:34 +01004314 Py_ssize_t shiftOutStart;
Antoine Pitrou244651a2009-05-04 18:56:13 +00004315 unsigned int base64bits = 0;
4316 unsigned long base64buffer = 0;
Victor Stinner24729f32011-11-10 20:31:37 +01004317 Py_UCS4 surrogate = 0;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004318 PyObject *errorHandler = NULL;
4319 PyObject *exc = NULL;
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004320
Martin v. Löwise9b11c12011-11-08 17:35:34 +01004321 /* Start off assuming it's all ASCII. Widen later as necessary. */
4322 unicode = PyUnicode_New(size, 127);
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004323 if (!unicode)
4324 return NULL;
Christian Heimes5d14c2b2007-11-20 23:38:09 +00004325 if (size == 0) {
4326 if (consumed)
4327 *consumed = 0;
Victor Stinner7931d9a2011-11-04 00:22:48 +01004328 return unicode;
Christian Heimes5d14c2b2007-11-20 23:38:09 +00004329 }
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004330
Martin v. Löwise9b11c12011-11-08 17:35:34 +01004331 shiftOutStart = outpos = 0;
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004332 e = s + size;
4333
4334 while (s < e) {
Martin v. Löwise9b11c12011-11-08 17:35:34 +01004335 Py_UCS4 ch;
Benjamin Peterson29060642009-01-31 22:14:21 +00004336 restart:
Antoine Pitrou5ffd9e92008-07-25 18:05:24 +00004337 ch = (unsigned char) *s;
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004338
Antoine Pitrou244651a2009-05-04 18:56:13 +00004339 if (inShift) { /* in a base-64 section */
4340 if (IS_BASE64(ch)) { /* consume a base-64 character */
4341 base64buffer = (base64buffer << 6) | FROM_BASE64(ch);
4342 base64bits += 6;
4343 s++;
4344 if (base64bits >= 16) {
4345 /* we have enough bits for a UTF-16 value */
Victor Stinner24729f32011-11-10 20:31:37 +01004346 Py_UCS4 outCh = (Py_UCS4)(base64buffer >> (base64bits-16));
Antoine Pitrou244651a2009-05-04 18:56:13 +00004347 base64bits -= 16;
4348 base64buffer &= (1 << base64bits) - 1; /* clear high bits */
4349 if (surrogate) {
4350 /* expecting a second surrogate */
Victor Stinner551ac952011-11-29 22:58:13 +01004351 if (Py_UNICODE_IS_LOW_SURROGATE(outCh)) {
4352 Py_UCS4 ch2 = Py_UNICODE_JOIN_SURROGATES(surrogate, outCh);
Martin v. Löwise9b11c12011-11-08 17:35:34 +01004353 if (unicode_putchar(&unicode, &outpos, ch2) < 0)
4354 goto onError;
Antoine Pitrou244651a2009-05-04 18:56:13 +00004355 surrogate = 0;
Antoine Pitrou5418ee02011-11-15 01:42:21 +01004356 continue;
Antoine Pitrou244651a2009-05-04 18:56:13 +00004357 }
4358 else {
Antoine Pitrou78edf752011-11-15 01:44:16 +01004359 if (unicode_putchar(&unicode, &outpos, surrogate) < 0)
4360 goto onError;
Antoine Pitrou244651a2009-05-04 18:56:13 +00004361 surrogate = 0;
Antoine Pitrou244651a2009-05-04 18:56:13 +00004362 }
4363 }
Victor Stinner551ac952011-11-29 22:58:13 +01004364 if (Py_UNICODE_IS_HIGH_SURROGATE(outCh)) {
Antoine Pitrou244651a2009-05-04 18:56:13 +00004365 /* first surrogate */
4366 surrogate = outCh;
4367 }
Antoine Pitrou244651a2009-05-04 18:56:13 +00004368 else {
Martin v. Löwise9b11c12011-11-08 17:35:34 +01004369 if (unicode_putchar(&unicode, &outpos, outCh) < 0)
4370 goto onError;
Antoine Pitrou244651a2009-05-04 18:56:13 +00004371 }
4372 }
4373 }
4374 else { /* now leaving a base-64 section */
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004375 inShift = 0;
4376 s++;
Antoine Pitrou244651a2009-05-04 18:56:13 +00004377 if (surrogate) {
Antoine Pitrou78edf752011-11-15 01:44:16 +01004378 if (unicode_putchar(&unicode, &outpos, surrogate) < 0)
4379 goto onError;
Antoine Pitrou5418ee02011-11-15 01:42:21 +01004380 surrogate = 0;
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004381 }
Antoine Pitrou244651a2009-05-04 18:56:13 +00004382 if (base64bits > 0) { /* left-over bits */
4383 if (base64bits >= 6) {
4384 /* We've seen at least one base-64 character */
4385 errmsg = "partial character in shift sequence";
4386 goto utf7Error;
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004387 }
Antoine Pitrou244651a2009-05-04 18:56:13 +00004388 else {
4389 /* Some bits remain; they should be zero */
4390 if (base64buffer != 0) {
4391 errmsg = "non-zero padding bits in shift sequence";
4392 goto utf7Error;
4393 }
4394 }
4395 }
4396 if (ch != '-') {
4397 /* '-' is absorbed; other terminating
4398 characters are preserved */
Martin v. Löwise9b11c12011-11-08 17:35:34 +01004399 if (unicode_putchar(&unicode, &outpos, ch) < 0)
4400 goto onError;
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004401 }
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004402 }
4403 }
4404 else if ( ch == '+' ) {
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004405 startinpos = s-starts;
Antoine Pitrou244651a2009-05-04 18:56:13 +00004406 s++; /* consume '+' */
4407 if (s < e && *s == '-') { /* '+-' encodes '+' */
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004408 s++;
Martin v. Löwise9b11c12011-11-08 17:35:34 +01004409 if (unicode_putchar(&unicode, &outpos, '+') < 0)
4410 goto onError;
Antoine Pitrou244651a2009-05-04 18:56:13 +00004411 }
4412 else { /* begin base64-encoded section */
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004413 inShift = 1;
Martin v. Löwise9b11c12011-11-08 17:35:34 +01004414 shiftOutStart = outpos;
Antoine Pitrou244651a2009-05-04 18:56:13 +00004415 base64bits = 0;
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004416 }
4417 }
Antoine Pitrou244651a2009-05-04 18:56:13 +00004418 else if (DECODE_DIRECT(ch)) { /* character decodes as itself */
Martin v. Löwise9b11c12011-11-08 17:35:34 +01004419 if (unicode_putchar(&unicode, &outpos, ch) < 0)
4420 goto onError;
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004421 s++;
4422 }
Antoine Pitrou244651a2009-05-04 18:56:13 +00004423 else {
4424 startinpos = s-starts;
4425 s++;
4426 errmsg = "unexpected special character";
4427 goto utf7Error;
4428 }
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004429 continue;
Antoine Pitrou244651a2009-05-04 18:56:13 +00004430utf7Error:
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004431 endinpos = s-starts;
4432 if (unicode_decode_call_errorhandler(
Benjamin Peterson29060642009-01-31 22:14:21 +00004433 errors, &errorHandler,
4434 "utf7", errmsg,
4435 &starts, &e, &startinpos, &endinpos, &exc, &s,
Martin v. Löwise9b11c12011-11-08 17:35:34 +01004436 &unicode, &outpos))
Benjamin Peterson29060642009-01-31 22:14:21 +00004437 goto onError;
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004438 }
4439
Antoine Pitrou244651a2009-05-04 18:56:13 +00004440 /* end of string */
4441
4442 if (inShift && !consumed) { /* in shift sequence, no more to follow */
4443 /* if we're in an inconsistent state, that's an error */
4444 if (surrogate ||
4445 (base64bits >= 6) ||
4446 (base64bits > 0 && base64buffer != 0)) {
Antoine Pitrou244651a2009-05-04 18:56:13 +00004447 endinpos = size;
4448 if (unicode_decode_call_errorhandler(
4449 errors, &errorHandler,
4450 "utf7", "unterminated shift sequence",
4451 &starts, &e, &startinpos, &endinpos, &exc, &s,
Martin v. Löwise9b11c12011-11-08 17:35:34 +01004452 &unicode, &outpos))
Antoine Pitrou244651a2009-05-04 18:56:13 +00004453 goto onError;
4454 if (s < e)
4455 goto restart;
4456 }
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004457 }
Antoine Pitrou244651a2009-05-04 18:56:13 +00004458
4459 /* return state */
Christian Heimes5d14c2b2007-11-20 23:38:09 +00004460 if (consumed) {
Antoine Pitrou244651a2009-05-04 18:56:13 +00004461 if (inShift) {
Martin v. Löwise9b11c12011-11-08 17:35:34 +01004462 outpos = shiftOutStart; /* back off output */
Christian Heimes5d14c2b2007-11-20 23:38:09 +00004463 *consumed = startinpos;
Antoine Pitrou244651a2009-05-04 18:56:13 +00004464 }
4465 else {
Christian Heimes5d14c2b2007-11-20 23:38:09 +00004466 *consumed = s-starts;
Antoine Pitrou244651a2009-05-04 18:56:13 +00004467 }
Christian Heimes5d14c2b2007-11-20 23:38:09 +00004468 }
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004469
Martin v. Löwise9b11c12011-11-08 17:35:34 +01004470 if (unicode_resize(&unicode, outpos) < 0)
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004471 goto onError;
4472
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004473 Py_XDECREF(errorHandler);
4474 Py_XDECREF(exc);
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01004475 return unicode_result(unicode);
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004476
Benjamin Peterson29060642009-01-31 22:14:21 +00004477 onError:
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004478 Py_XDECREF(errorHandler);
4479 Py_XDECREF(exc);
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004480 Py_DECREF(unicode);
4481 return NULL;
4482}
4483
4484
Alexander Belopolsky40018472011-02-26 01:02:56 +00004485PyObject *
Martin v. Löwis1db7c132011-11-10 18:24:32 +01004486_PyUnicode_EncodeUTF7(PyObject *str,
4487 int base64SetO,
4488 int base64WhiteSpace,
4489 const char *errors)
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004490{
Martin v. Löwis1db7c132011-11-10 18:24:32 +01004491 int kind;
4492 void *data;
4493 Py_ssize_t len;
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00004494 PyObject *v;
Martin v. Löwis1db7c132011-11-10 18:24:32 +01004495 Py_ssize_t allocated;
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004496 int inShift = 0;
Martin v. Löwis1db7c132011-11-10 18:24:32 +01004497 Py_ssize_t i;
Antoine Pitrou244651a2009-05-04 18:56:13 +00004498 unsigned int base64bits = 0;
4499 unsigned long base64buffer = 0;
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004500 char * out;
4501 char * start;
4502
Benjamin Petersonbac79492012-01-14 13:34:47 -05004503 if (PyUnicode_READY(str) == -1)
Martin v. Löwis1db7c132011-11-10 18:24:32 +01004504 return NULL;
4505 kind = PyUnicode_KIND(str);
4506 data = PyUnicode_DATA(str);
4507 len = PyUnicode_GET_LENGTH(str);
4508
4509 if (len == 0)
Benjamin Peterson29060642009-01-31 22:14:21 +00004510 return PyBytes_FromStringAndSize(NULL, 0);
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004511
Martin v. Löwis1db7c132011-11-10 18:24:32 +01004512 /* It might be possible to tighten this worst case */
4513 allocated = 8 * len;
4514 if (allocated / 8 != len)
Neal Norwitz3ce5d922008-08-24 07:08:55 +00004515 return PyErr_NoMemory();
4516
Antoine Pitrou244651a2009-05-04 18:56:13 +00004517 v = PyBytes_FromStringAndSize(NULL, allocated);
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004518 if (v == NULL)
4519 return NULL;
4520
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00004521 start = out = PyBytes_AS_STRING(v);
Martin v. Löwis1db7c132011-11-10 18:24:32 +01004522 for (i = 0; i < len; ++i) {
Victor Stinner0e368262011-11-10 20:12:49 +01004523 Py_UCS4 ch = PyUnicode_READ(kind, data, i);
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004524
Antoine Pitrou244651a2009-05-04 18:56:13 +00004525 if (inShift) {
4526 if (ENCODE_DIRECT(ch, !base64SetO, !base64WhiteSpace)) {
4527 /* shifting out */
4528 if (base64bits) { /* output remaining bits */
4529 *out++ = TO_BASE64(base64buffer << (6-base64bits));
4530 base64buffer = 0;
4531 base64bits = 0;
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004532 }
4533 inShift = 0;
Antoine Pitrou244651a2009-05-04 18:56:13 +00004534 /* Characters not in the BASE64 set implicitly unshift the sequence
4535 so no '-' is required, except if the character is itself a '-' */
4536 if (IS_BASE64(ch) || ch == '-') {
4537 *out++ = '-';
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004538 }
Antoine Pitrou244651a2009-05-04 18:56:13 +00004539 *out++ = (char) ch;
4540 }
4541 else {
4542 goto encode_char;
Tim Petersced69f82003-09-16 20:30:58 +00004543 }
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004544 }
Antoine Pitrou244651a2009-05-04 18:56:13 +00004545 else { /* not in a shift sequence */
4546 if (ch == '+') {
4547 *out++ = '+';
4548 *out++ = '-';
4549 }
4550 else if (ENCODE_DIRECT(ch, !base64SetO, !base64WhiteSpace)) {
4551 *out++ = (char) ch;
4552 }
4553 else {
4554 *out++ = '+';
4555 inShift = 1;
4556 goto encode_char;
4557 }
4558 }
4559 continue;
4560encode_char:
Antoine Pitrou244651a2009-05-04 18:56:13 +00004561 if (ch >= 0x10000) {
Victor Stinner8faf8212011-12-08 22:14:11 +01004562 assert(ch <= MAX_UNICODE);
Victor Stinner0d3721d2011-11-22 03:27:53 +01004563
Antoine Pitrou244651a2009-05-04 18:56:13 +00004564 /* code first surrogate */
4565 base64bits += 16;
4566 base64buffer = (base64buffer << 16) | 0xd800 | ((ch-0x10000) >> 10);
4567 while (base64bits >= 6) {
4568 *out++ = TO_BASE64(base64buffer >> (base64bits-6));
4569 base64bits -= 6;
4570 }
4571 /* prepare second surrogate */
Victor Stinner551ac952011-11-29 22:58:13 +01004572 ch = Py_UNICODE_LOW_SURROGATE(ch);
Antoine Pitrou244651a2009-05-04 18:56:13 +00004573 }
Antoine Pitrou244651a2009-05-04 18:56:13 +00004574 base64bits += 16;
4575 base64buffer = (base64buffer << 16) | ch;
4576 while (base64bits >= 6) {
4577 *out++ = TO_BASE64(base64buffer >> (base64bits-6));
4578 base64bits -= 6;
4579 }
Hye-Shik Chang1bc09b72004-01-03 19:35:43 +00004580 }
Antoine Pitrou244651a2009-05-04 18:56:13 +00004581 if (base64bits)
4582 *out++= TO_BASE64(base64buffer << (6-base64bits) );
4583 if (inShift)
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004584 *out++ = '-';
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00004585 if (_PyBytes_Resize(&v, out - start) < 0)
4586 return NULL;
4587 return v;
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004588}
Martin v. Löwis1db7c132011-11-10 18:24:32 +01004589PyObject *
4590PyUnicode_EncodeUTF7(const Py_UNICODE *s,
4591 Py_ssize_t size,
4592 int base64SetO,
4593 int base64WhiteSpace,
4594 const char *errors)
4595{
4596 PyObject *result;
4597 PyObject *tmp = PyUnicode_FromUnicode(s, size);
4598 if (tmp == NULL)
4599 return NULL;
Victor Stinner0e368262011-11-10 20:12:49 +01004600 result = _PyUnicode_EncodeUTF7(tmp, base64SetO,
Martin v. Löwis1db7c132011-11-10 18:24:32 +01004601 base64WhiteSpace, errors);
4602 Py_DECREF(tmp);
4603 return result;
4604}
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004605
Antoine Pitrou244651a2009-05-04 18:56:13 +00004606#undef IS_BASE64
4607#undef FROM_BASE64
4608#undef TO_BASE64
4609#undef DECODE_DIRECT
4610#undef ENCODE_DIRECT
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00004611
Guido van Rossumd57fd912000-03-10 22:53:23 +00004612/* --- UTF-8 Codec -------------------------------------------------------- */
4613
Alexander Belopolsky40018472011-02-26 01:02:56 +00004614PyObject *
4615PyUnicode_DecodeUTF8(const char *s,
Ezio Melotti2aa2b3b2011-09-29 00:58:57 +03004616 Py_ssize_t size,
4617 const char *errors)
Guido van Rossumd57fd912000-03-10 22:53:23 +00004618{
Walter Dörwald69652032004-09-07 20:24:22 +00004619 return PyUnicode_DecodeUTF8Stateful(s, size, errors, NULL);
4620}
4621
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004622#include "stringlib/asciilib.h"
4623#include "stringlib/codecs.h"
4624#include "stringlib/undef.h"
4625
Antoine Pitrou0a3229d2011-11-21 20:39:13 +01004626#include "stringlib/ucs1lib.h"
4627#include "stringlib/codecs.h"
4628#include "stringlib/undef.h"
4629
4630#include "stringlib/ucs2lib.h"
4631#include "stringlib/codecs.h"
4632#include "stringlib/undef.h"
4633
4634#include "stringlib/ucs4lib.h"
4635#include "stringlib/codecs.h"
4636#include "stringlib/undef.h"
4637
Antoine Pitrouab868312009-01-10 15:40:25 +00004638/* Mask to check or force alignment of a pointer to C 'long' boundaries */
4639#define LONG_PTR_MASK (size_t) (SIZEOF_LONG - 1)
4640
4641/* Mask to quickly check whether a C 'long' contains a
4642 non-ASCII, UTF8-encoded char. */
4643#if (SIZEOF_LONG == 8)
Mark Dickinson01ac8b62012-07-07 14:08:48 +02004644# define ASCII_CHAR_MASK 0x8080808080808080UL
Antoine Pitrouab868312009-01-10 15:40:25 +00004645#elif (SIZEOF_LONG == 4)
Mark Dickinson01ac8b62012-07-07 14:08:48 +02004646# define ASCII_CHAR_MASK 0x80808080UL
Antoine Pitrouab868312009-01-10 15:40:25 +00004647#else
4648# error C 'long' size should be either 4 or 8!
4649#endif
4650
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004651static Py_ssize_t
4652ascii_decode(const char *start, const char *end, Py_UCS1 *dest)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02004653{
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004654 const char *p = start;
4655 const char *aligned_end = (const char *) ((size_t) end & ~LONG_PTR_MASK);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02004656
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004657#if SIZEOF_LONG <= SIZEOF_VOID_P
4658 assert(!((size_t) dest & LONG_PTR_MASK));
4659 if (!((size_t) p & LONG_PTR_MASK)) {
4660 /* Fast path, see in STRINGLIB(utf8_decode) for
4661 an explanation. */
4662 /* Help register allocation */
4663 register const char *_p = p;
4664 register Py_UCS1 * q = dest;
4665 while (_p < aligned_end) {
4666 unsigned long value = *(const unsigned long *) _p;
4667 if (value & ASCII_CHAR_MASK)
Benjamin Peterson29060642009-01-31 22:14:21 +00004668 break;
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004669 *((unsigned long *)q) = value;
4670 _p += SIZEOF_LONG;
4671 q += SIZEOF_LONG;
Benjamin Peterson14339b62009-01-31 16:36:08 +00004672 }
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004673 p = _p;
4674 while (p < end) {
4675 if ((unsigned char)*p & 0x80)
4676 break;
4677 *q++ = *p++;
Guido van Rossumd57fd912000-03-10 22:53:23 +00004678 }
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004679 return p - start;
Guido van Rossumd57fd912000-03-10 22:53:23 +00004680 }
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004681#endif
4682 while (p < end) {
4683 /* Fast path, see in STRINGLIB(utf8_decode) in stringlib/codecs.h
4684 for an explanation. */
4685 if (!((size_t) p & LONG_PTR_MASK)) {
4686 /* Help register allocation */
4687 register const char *_p = p;
4688 while (_p < aligned_end) {
4689 unsigned long value = *(unsigned long *) _p;
4690 if (value & ASCII_CHAR_MASK)
4691 break;
4692 _p += SIZEOF_LONG;
4693 }
4694 p = _p;
4695 if (_p == end)
4696 break;
4697 }
4698 if ((unsigned char)*p & 0x80)
4699 break;
4700 ++p;
4701 }
4702 memcpy(dest, start, p - start);
4703 return p - start;
Guido van Rossumd57fd912000-03-10 22:53:23 +00004704}
Antoine Pitrouab868312009-01-10 15:40:25 +00004705
Victor Stinner785938e2011-12-11 20:09:03 +01004706PyObject *
4707PyUnicode_DecodeUTF8Stateful(const char *s,
4708 Py_ssize_t size,
4709 const char *errors,
4710 Py_ssize_t *consumed)
4711{
Victor Stinner785938e2011-12-11 20:09:03 +01004712 PyObject *unicode;
Victor Stinner785938e2011-12-11 20:09:03 +01004713 const char *starts = s;
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004714 const char *end = s + size;
4715 Py_ssize_t outpos;
4716
4717 Py_ssize_t startinpos;
4718 Py_ssize_t endinpos;
4719 const char *errmsg = "";
4720 PyObject *errorHandler = NULL;
4721 PyObject *exc = NULL;
Victor Stinner785938e2011-12-11 20:09:03 +01004722
4723 if (size == 0) {
4724 if (consumed)
4725 *consumed = 0;
Victor Stinner382955f2011-12-11 21:44:00 +01004726 Py_INCREF(unicode_empty);
4727 return unicode_empty;
Victor Stinner785938e2011-12-11 20:09:03 +01004728 }
4729
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004730 /* ASCII is equivalent to the first 128 ordinals in Unicode. */
4731 if (size == 1 && (unsigned char)s[0] < 128) {
Victor Stinner785938e2011-12-11 20:09:03 +01004732 if (consumed)
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004733 *consumed = 1;
4734 return get_latin1_char((unsigned char)s[0]);
Victor Stinner785938e2011-12-11 20:09:03 +01004735 }
4736
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004737 unicode = PyUnicode_New(size, 127);
Victor Stinner785938e2011-12-11 20:09:03 +01004738 if (!unicode)
4739 return NULL;
Victor Stinner785938e2011-12-11 20:09:03 +01004740
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004741 outpos = ascii_decode(s, end, PyUnicode_1BYTE_DATA(unicode));
4742 s += outpos;
4743 while (s < end) {
4744 Py_UCS4 ch;
4745 int kind = PyUnicode_KIND(unicode);
4746 if (kind == PyUnicode_1BYTE_KIND) {
4747 if (PyUnicode_IS_ASCII(unicode))
4748 ch = asciilib_utf8_decode(&s, end,
4749 PyUnicode_1BYTE_DATA(unicode), &outpos);
4750 else
4751 ch = ucs1lib_utf8_decode(&s, end,
4752 PyUnicode_1BYTE_DATA(unicode), &outpos);
4753 } else if (kind == PyUnicode_2BYTE_KIND) {
4754 ch = ucs2lib_utf8_decode(&s, end,
4755 PyUnicode_2BYTE_DATA(unicode), &outpos);
4756 } else {
4757 assert(kind == PyUnicode_4BYTE_KIND);
4758 ch = ucs4lib_utf8_decode(&s, end,
4759 PyUnicode_4BYTE_DATA(unicode), &outpos);
4760 }
4761
4762 switch (ch) {
4763 case 0:
4764 if (s == end || consumed)
4765 goto End;
4766 errmsg = "unexpected end of data";
4767 startinpos = s - starts;
4768 endinpos = startinpos + 1;
4769 while (endinpos < size && (starts[endinpos] & 0xC0) == 0x80)
4770 endinpos++;
4771 break;
4772 case 1:
4773 errmsg = "invalid start byte";
4774 startinpos = s - starts;
4775 endinpos = startinpos + 1;
4776 break;
4777 case 2:
4778 errmsg = "invalid continuation byte";
4779 startinpos = s - starts;
4780 endinpos = startinpos + 1;
4781 while (endinpos < size && (starts[endinpos] & 0xC0) == 0x80)
4782 endinpos++;
4783 break;
4784 default:
4785 if (unicode_putchar(&unicode, &outpos, ch) < 0)
4786 goto onError;
4787 continue;
4788 }
4789
4790 if (unicode_decode_call_errorhandler(
4791 errors, &errorHandler,
4792 "utf-8", errmsg,
4793 &starts, &end, &startinpos, &endinpos, &exc, &s,
4794 &unicode, &outpos))
4795 goto onError;
Victor Stinner785938e2011-12-11 20:09:03 +01004796 }
4797
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004798End:
4799 if (unicode_resize(&unicode, outpos) < 0)
4800 goto onError;
4801
4802 if (consumed)
4803 *consumed = s - starts;
4804
4805 Py_XDECREF(errorHandler);
4806 Py_XDECREF(exc);
4807 assert(_PyUnicode_CheckConsistency(unicode, 1));
4808 return unicode;
4809
4810onError:
4811 Py_XDECREF(errorHandler);
4812 Py_XDECREF(exc);
4813 Py_XDECREF(unicode);
4814 return NULL;
Victor Stinner785938e2011-12-11 20:09:03 +01004815}
4816
Victor Stinnerf933e1a2010-10-20 22:58:25 +00004817#ifdef __APPLE__
4818
4819/* Simplified UTF-8 decoder using surrogateescape error handler,
4820 used to decode the command line arguments on Mac OS X. */
4821
4822wchar_t*
4823_Py_DecodeUTF8_surrogateescape(const char *s, Py_ssize_t size)
4824{
Victor Stinnerf933e1a2010-10-20 22:58:25 +00004825 const char *e;
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004826 wchar_t *unicode;
4827 Py_ssize_t outpos;
Victor Stinnerf933e1a2010-10-20 22:58:25 +00004828
4829 /* Note: size will always be longer than the resulting Unicode
4830 character count */
4831 if (PY_SSIZE_T_MAX / sizeof(wchar_t) < (size + 1)) {
4832 PyErr_NoMemory();
4833 return NULL;
4834 }
4835 unicode = PyMem_Malloc((size + 1) * sizeof(wchar_t));
4836 if (!unicode)
4837 return NULL;
4838
4839 /* Unpack UTF-8 encoded data */
Victor Stinnerf933e1a2010-10-20 22:58:25 +00004840 e = s + size;
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004841 outpos = 0;
Victor Stinnerf933e1a2010-10-20 22:58:25 +00004842 while (s < e) {
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004843 Py_UCS4 ch;
Victor Stinnerf933e1a2010-10-20 22:58:25 +00004844#if SIZEOF_WCHAR_T == 4
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004845 ch = ucs4lib_utf8_decode(&s, e, (Py_UCS4 *)unicode, &outpos);
Victor Stinnerf933e1a2010-10-20 22:58:25 +00004846#else
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004847 ch = ucs2lib_utf8_decode(&s, e, (Py_UCS2 *)unicode, &outpos);
Victor Stinnerf933e1a2010-10-20 22:58:25 +00004848#endif
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004849 if (ch > 0xFF) {
4850#if SIZEOF_WCHAR_T == 4
4851 assert(0);
4852#else
4853 assert(Py_UNICODE_IS_SURROGATE(ch));
4854 /* compute and append the two surrogates: */
4855 unicode[outpos++] = (wchar_t)Py_UNICODE_HIGH_SURROGATE(ch);
4856 unicode[outpos++] = (wchar_t)Py_UNICODE_LOW_SURROGATE(ch);
4857#endif
Victor Stinnerf933e1a2010-10-20 22:58:25 +00004858 }
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004859 else {
4860 if (!ch && s == e)
4861 break;
4862 /* surrogateescape */
4863 unicode[outpos++] = 0xDC00 + (unsigned char)*s++;
4864 }
Victor Stinnerf933e1a2010-10-20 22:58:25 +00004865 }
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004866 unicode[outpos] = L'\0';
Victor Stinnerf933e1a2010-10-20 22:58:25 +00004867 return unicode;
4868}
4869
4870#endif /* __APPLE__ */
Antoine Pitrouab868312009-01-10 15:40:25 +00004871
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02004872/* Primary internal function which creates utf8 encoded bytes objects.
4873
4874 Allocation strategy: if the string is short, convert into a stack buffer
Tim Peters602f7402002-04-27 18:03:26 +00004875 and allocate exactly as much space needed at the end. Else allocate the
4876 maximum possible needed (4 result bytes per Unicode character), and return
4877 the excess memory at the end.
Martin v. Löwis2a7ff352002-04-21 09:59:45 +00004878*/
Tim Peters7e3d9612002-04-21 03:26:37 +00004879PyObject *
Victor Stinner7931d9a2011-11-04 00:22:48 +01004880_PyUnicode_AsUTF8String(PyObject *unicode, const char *errors)
Guido van Rossumd57fd912000-03-10 22:53:23 +00004881{
Victor Stinner6099a032011-12-18 14:22:26 +01004882 enum PyUnicode_Kind kind;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02004883 void *data;
4884 Py_ssize_t size;
Marc-André Lemburgbd3be8f2002-02-07 11:33:49 +00004885
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02004886 if (!PyUnicode_Check(unicode)) {
4887 PyErr_BadArgument();
4888 return NULL;
4889 }
4890
4891 if (PyUnicode_READY(unicode) == -1)
4892 return NULL;
4893
Victor Stinnere90fe6a2011-10-01 16:48:13 +02004894 if (PyUnicode_UTF8(unicode))
4895 return PyBytes_FromStringAndSize(PyUnicode_UTF8(unicode),
4896 PyUnicode_UTF8_LENGTH(unicode));
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02004897
4898 kind = PyUnicode_KIND(unicode);
4899 data = PyUnicode_DATA(unicode);
4900 size = PyUnicode_GET_LENGTH(unicode);
4901
Benjamin Petersonead6b532011-12-20 17:23:42 -06004902 switch (kind) {
Victor Stinner6099a032011-12-18 14:22:26 +01004903 default:
4904 assert(0);
4905 case PyUnicode_1BYTE_KIND:
4906 /* the string cannot be ASCII, or PyUnicode_UTF8() would be set */
4907 assert(!PyUnicode_IS_ASCII(unicode));
4908 return ucs1lib_utf8_encoder(unicode, data, size, errors);
4909 case PyUnicode_2BYTE_KIND:
4910 return ucs2lib_utf8_encoder(unicode, data, size, errors);
4911 case PyUnicode_4BYTE_KIND:
4912 return ucs4lib_utf8_encoder(unicode, data, size, errors);
Tim Peters602f7402002-04-27 18:03:26 +00004913 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00004914}
4915
Alexander Belopolsky40018472011-02-26 01:02:56 +00004916PyObject *
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02004917PyUnicode_EncodeUTF8(const Py_UNICODE *s,
4918 Py_ssize_t size,
4919 const char *errors)
4920{
4921 PyObject *v, *unicode;
4922
4923 unicode = PyUnicode_FromUnicode(s, size);
4924 if (unicode == NULL)
4925 return NULL;
4926 v = _PyUnicode_AsUTF8String(unicode, errors);
4927 Py_DECREF(unicode);
4928 return v;
4929}
4930
4931PyObject *
Alexander Belopolsky40018472011-02-26 01:02:56 +00004932PyUnicode_AsUTF8String(PyObject *unicode)
Guido van Rossumd57fd912000-03-10 22:53:23 +00004933{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02004934 return _PyUnicode_AsUTF8String(unicode, NULL);
Guido van Rossumd57fd912000-03-10 22:53:23 +00004935}
4936
Walter Dörwald41980ca2007-08-16 21:55:45 +00004937/* --- UTF-32 Codec ------------------------------------------------------- */
4938
4939PyObject *
4940PyUnicode_DecodeUTF32(const char *s,
Benjamin Peterson29060642009-01-31 22:14:21 +00004941 Py_ssize_t size,
4942 const char *errors,
4943 int *byteorder)
Walter Dörwald41980ca2007-08-16 21:55:45 +00004944{
4945 return PyUnicode_DecodeUTF32Stateful(s, size, errors, byteorder, NULL);
4946}
4947
4948PyObject *
4949PyUnicode_DecodeUTF32Stateful(const char *s,
Benjamin Peterson29060642009-01-31 22:14:21 +00004950 Py_ssize_t size,
4951 const char *errors,
4952 int *byteorder,
4953 Py_ssize_t *consumed)
Walter Dörwald41980ca2007-08-16 21:55:45 +00004954{
4955 const char *starts = s;
4956 Py_ssize_t startinpos;
4957 Py_ssize_t endinpos;
4958 Py_ssize_t outpos;
Victor Stinner7931d9a2011-11-04 00:22:48 +01004959 PyObject *unicode;
Mark Dickinson7db923c2010-06-12 09:10:14 +00004960 const unsigned char *q, *e;
Walter Dörwald41980ca2007-08-16 21:55:45 +00004961 int bo = 0; /* assume native ordering by default */
4962 const char *errmsg = "";
Walter Dörwald41980ca2007-08-16 21:55:45 +00004963 /* Offsets from q for retrieving bytes in the right order. */
4964#ifdef BYTEORDER_IS_LITTLE_ENDIAN
4965 int iorder[] = {0, 1, 2, 3};
4966#else
4967 int iorder[] = {3, 2, 1, 0};
4968#endif
4969 PyObject *errorHandler = NULL;
4970 PyObject *exc = NULL;
Victor Stinner313a1202010-06-11 23:56:51 +00004971
Walter Dörwald41980ca2007-08-16 21:55:45 +00004972 q = (unsigned char *)s;
4973 e = q + size;
4974
4975 if (byteorder)
4976 bo = *byteorder;
4977
4978 /* Check for BOM marks (U+FEFF) in the input and adjust current
4979 byte order setting accordingly. In native mode, the leading BOM
4980 mark is skipped, in all other modes, it is copied to the output
4981 stream as-is (giving a ZWNBSP character). */
4982 if (bo == 0) {
4983 if (size >= 4) {
4984 const Py_UCS4 bom = (q[iorder[3]] << 24) | (q[iorder[2]] << 16) |
Benjamin Peterson29060642009-01-31 22:14:21 +00004985 (q[iorder[1]] << 8) | q[iorder[0]];
Walter Dörwald41980ca2007-08-16 21:55:45 +00004986#ifdef BYTEORDER_IS_LITTLE_ENDIAN
Benjamin Peterson29060642009-01-31 22:14:21 +00004987 if (bom == 0x0000FEFF) {
4988 q += 4;
4989 bo = -1;
4990 }
4991 else if (bom == 0xFFFE0000) {
4992 q += 4;
4993 bo = 1;
4994 }
Walter Dörwald41980ca2007-08-16 21:55:45 +00004995#else
Benjamin Peterson29060642009-01-31 22:14:21 +00004996 if (bom == 0x0000FEFF) {
4997 q += 4;
4998 bo = 1;
4999 }
5000 else if (bom == 0xFFFE0000) {
5001 q += 4;
5002 bo = -1;
5003 }
Walter Dörwald41980ca2007-08-16 21:55:45 +00005004#endif
Benjamin Peterson29060642009-01-31 22:14:21 +00005005 }
Walter Dörwald41980ca2007-08-16 21:55:45 +00005006 }
5007
5008 if (bo == -1) {
5009 /* force LE */
5010 iorder[0] = 0;
5011 iorder[1] = 1;
5012 iorder[2] = 2;
5013 iorder[3] = 3;
5014 }
5015 else if (bo == 1) {
5016 /* force BE */
5017 iorder[0] = 3;
5018 iorder[1] = 2;
5019 iorder[2] = 1;
5020 iorder[3] = 0;
5021 }
5022
Antoine Pitroucc0cfd32010-06-11 21:46:32 +00005023 /* This might be one to much, because of a BOM */
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005024 unicode = PyUnicode_New((size+3)/4, 127);
Antoine Pitroucc0cfd32010-06-11 21:46:32 +00005025 if (!unicode)
5026 return NULL;
5027 if (size == 0)
Victor Stinner7931d9a2011-11-04 00:22:48 +01005028 return unicode;
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005029 outpos = 0;
Antoine Pitroucc0cfd32010-06-11 21:46:32 +00005030
Walter Dörwald41980ca2007-08-16 21:55:45 +00005031 while (q < e) {
Benjamin Peterson29060642009-01-31 22:14:21 +00005032 Py_UCS4 ch;
5033 /* remaining bytes at the end? (size should be divisible by 4) */
5034 if (e-q<4) {
5035 if (consumed)
5036 break;
5037 errmsg = "truncated data";
5038 startinpos = ((const char *)q)-starts;
5039 endinpos = ((const char *)e)-starts;
5040 goto utf32Error;
5041 /* The remaining input chars are ignored if the callback
5042 chooses to skip the input */
5043 }
5044 ch = (q[iorder[3]] << 24) | (q[iorder[2]] << 16) |
5045 (q[iorder[1]] << 8) | q[iorder[0]];
Walter Dörwald41980ca2007-08-16 21:55:45 +00005046
Benjamin Peterson29060642009-01-31 22:14:21 +00005047 if (ch >= 0x110000)
5048 {
5049 errmsg = "codepoint not in range(0x110000)";
5050 startinpos = ((const char *)q)-starts;
5051 endinpos = startinpos+4;
5052 goto utf32Error;
5053 }
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005054 if (unicode_putchar(&unicode, &outpos, ch) < 0)
5055 goto onError;
Benjamin Peterson29060642009-01-31 22:14:21 +00005056 q += 4;
5057 continue;
5058 utf32Error:
Benjamin Peterson29060642009-01-31 22:14:21 +00005059 if (unicode_decode_call_errorhandler(
5060 errors, &errorHandler,
5061 "utf32", errmsg,
5062 &starts, (const char **)&e, &startinpos, &endinpos, &exc, (const char **)&q,
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005063 &unicode, &outpos))
Benjamin Peterson29060642009-01-31 22:14:21 +00005064 goto onError;
Walter Dörwald41980ca2007-08-16 21:55:45 +00005065 }
5066
5067 if (byteorder)
5068 *byteorder = bo;
5069
5070 if (consumed)
Benjamin Peterson29060642009-01-31 22:14:21 +00005071 *consumed = (const char *)q-starts;
Walter Dörwald41980ca2007-08-16 21:55:45 +00005072
5073 /* Adjust length */
Victor Stinner16e6a802011-12-12 13:24:15 +01005074 if (unicode_resize(&unicode, outpos) < 0)
Walter Dörwald41980ca2007-08-16 21:55:45 +00005075 goto onError;
5076
5077 Py_XDECREF(errorHandler);
5078 Py_XDECREF(exc);
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01005079 return unicode_result(unicode);
Walter Dörwald41980ca2007-08-16 21:55:45 +00005080
Benjamin Peterson29060642009-01-31 22:14:21 +00005081 onError:
Walter Dörwald41980ca2007-08-16 21:55:45 +00005082 Py_DECREF(unicode);
5083 Py_XDECREF(errorHandler);
5084 Py_XDECREF(exc);
5085 return NULL;
5086}
5087
5088PyObject *
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005089_PyUnicode_EncodeUTF32(PyObject *str,
5090 const char *errors,
5091 int byteorder)
Walter Dörwald41980ca2007-08-16 21:55:45 +00005092{
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005093 int kind;
5094 void *data;
5095 Py_ssize_t len;
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00005096 PyObject *v;
Walter Dörwald41980ca2007-08-16 21:55:45 +00005097 unsigned char *p;
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005098 Py_ssize_t nsize, bytesize, i;
Walter Dörwald41980ca2007-08-16 21:55:45 +00005099 /* Offsets from p for storing byte pairs in the right order. */
5100#ifdef BYTEORDER_IS_LITTLE_ENDIAN
5101 int iorder[] = {0, 1, 2, 3};
5102#else
5103 int iorder[] = {3, 2, 1, 0};
5104#endif
5105
Benjamin Peterson29060642009-01-31 22:14:21 +00005106#define STORECHAR(CH) \
5107 do { \
5108 p[iorder[3]] = ((CH) >> 24) & 0xff; \
5109 p[iorder[2]] = ((CH) >> 16) & 0xff; \
5110 p[iorder[1]] = ((CH) >> 8) & 0xff; \
5111 p[iorder[0]] = (CH) & 0xff; \
5112 p += 4; \
Walter Dörwald41980ca2007-08-16 21:55:45 +00005113 } while(0)
5114
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005115 if (!PyUnicode_Check(str)) {
5116 PyErr_BadArgument();
5117 return NULL;
5118 }
Benjamin Petersonbac79492012-01-14 13:34:47 -05005119 if (PyUnicode_READY(str) == -1)
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005120 return NULL;
5121 kind = PyUnicode_KIND(str);
5122 data = PyUnicode_DATA(str);
5123 len = PyUnicode_GET_LENGTH(str);
5124
5125 nsize = len + (byteorder == 0);
Neal Norwitz3ce5d922008-08-24 07:08:55 +00005126 bytesize = nsize * 4;
5127 if (bytesize / 4 != nsize)
Benjamin Peterson29060642009-01-31 22:14:21 +00005128 return PyErr_NoMemory();
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00005129 v = PyBytes_FromStringAndSize(NULL, bytesize);
Walter Dörwald41980ca2007-08-16 21:55:45 +00005130 if (v == NULL)
5131 return NULL;
5132
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00005133 p = (unsigned char *)PyBytes_AS_STRING(v);
Walter Dörwald41980ca2007-08-16 21:55:45 +00005134 if (byteorder == 0)
Benjamin Peterson29060642009-01-31 22:14:21 +00005135 STORECHAR(0xFEFF);
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005136 if (len == 0)
Guido van Rossum98297ee2007-11-06 21:34:58 +00005137 goto done;
Walter Dörwald41980ca2007-08-16 21:55:45 +00005138
5139 if (byteorder == -1) {
5140 /* force LE */
5141 iorder[0] = 0;
5142 iorder[1] = 1;
5143 iorder[2] = 2;
5144 iorder[3] = 3;
5145 }
5146 else if (byteorder == 1) {
5147 /* force BE */
5148 iorder[0] = 3;
5149 iorder[1] = 2;
5150 iorder[2] = 1;
5151 iorder[3] = 0;
5152 }
5153
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005154 for (i = 0; i < len; i++)
5155 STORECHAR(PyUnicode_READ(kind, data, i));
Guido van Rossum98297ee2007-11-06 21:34:58 +00005156
5157 done:
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00005158 return v;
Walter Dörwald41980ca2007-08-16 21:55:45 +00005159#undef STORECHAR
5160}
5161
Alexander Belopolsky40018472011-02-26 01:02:56 +00005162PyObject *
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005163PyUnicode_EncodeUTF32(const Py_UNICODE *s,
5164 Py_ssize_t size,
5165 const char *errors,
5166 int byteorder)
5167{
5168 PyObject *result;
5169 PyObject *tmp = PyUnicode_FromUnicode(s, size);
5170 if (tmp == NULL)
5171 return NULL;
5172 result = _PyUnicode_EncodeUTF32(tmp, errors, byteorder);
5173 Py_DECREF(tmp);
5174 return result;
5175}
5176
5177PyObject *
Alexander Belopolsky40018472011-02-26 01:02:56 +00005178PyUnicode_AsUTF32String(PyObject *unicode)
Walter Dörwald41980ca2007-08-16 21:55:45 +00005179{
Victor Stinnerb960b342011-11-20 19:12:52 +01005180 return _PyUnicode_EncodeUTF32(unicode, NULL, 0);
Walter Dörwald41980ca2007-08-16 21:55:45 +00005181}
5182
Guido van Rossumd57fd912000-03-10 22:53:23 +00005183/* --- UTF-16 Codec ------------------------------------------------------- */
5184
Tim Peters772747b2001-08-09 22:21:55 +00005185PyObject *
5186PyUnicode_DecodeUTF16(const char *s,
Benjamin Peterson29060642009-01-31 22:14:21 +00005187 Py_ssize_t size,
5188 const char *errors,
5189 int *byteorder)
Guido van Rossumd57fd912000-03-10 22:53:23 +00005190{
Walter Dörwald69652032004-09-07 20:24:22 +00005191 return PyUnicode_DecodeUTF16Stateful(s, size, errors, byteorder, NULL);
5192}
5193
5194PyObject *
5195PyUnicode_DecodeUTF16Stateful(const char *s,
Benjamin Peterson29060642009-01-31 22:14:21 +00005196 Py_ssize_t size,
5197 const char *errors,
5198 int *byteorder,
5199 Py_ssize_t *consumed)
Walter Dörwald69652032004-09-07 20:24:22 +00005200{
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005201 const char *starts = s;
Martin v. Löwis18e16552006-02-15 17:27:45 +00005202 Py_ssize_t startinpos;
5203 Py_ssize_t endinpos;
5204 Py_ssize_t outpos;
Victor Stinner7931d9a2011-11-04 00:22:48 +01005205 PyObject *unicode;
Antoine Pitrou63065d72012-05-15 23:48:04 +02005206 const unsigned char *q, *e;
Tim Peters772747b2001-08-09 22:21:55 +00005207 int bo = 0; /* assume native ordering by default */
Antoine Pitrou63065d72012-05-15 23:48:04 +02005208 int native_ordering;
Marc-André Lemburg9542f482000-07-17 18:23:13 +00005209 const char *errmsg = "";
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005210 PyObject *errorHandler = NULL;
5211 PyObject *exc = NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005212
Tim Peters772747b2001-08-09 22:21:55 +00005213 q = (unsigned char *)s;
Antoine Pitrou63065d72012-05-15 23:48:04 +02005214 e = q + size;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005215
5216 if (byteorder)
Tim Peters772747b2001-08-09 22:21:55 +00005217 bo = *byteorder;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005218
Marc-André Lemburg489b56e2001-05-21 20:30:15 +00005219 /* Check for BOM marks (U+FEFF) in the input and adjust current
5220 byte order setting accordingly. In native mode, the leading BOM
5221 mark is skipped, in all other modes, it is copied to the output
5222 stream as-is (giving a ZWNBSP character). */
Antoine Pitrou63065d72012-05-15 23:48:04 +02005223 if (bo == 0 && size >= 2) {
5224 const Py_UCS4 bom = (q[1] << 8) | q[0];
5225 if (bom == 0xFEFF) {
5226 q += 2;
5227 bo = -1;
Benjamin Peterson29060642009-01-31 22:14:21 +00005228 }
Antoine Pitrou63065d72012-05-15 23:48:04 +02005229 else if (bom == 0xFFFE) {
5230 q += 2;
5231 bo = 1;
5232 }
5233 if (byteorder)
5234 *byteorder = bo;
Marc-André Lemburg489b56e2001-05-21 20:30:15 +00005235 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00005236
Antoine Pitrou63065d72012-05-15 23:48:04 +02005237 if (q == e) {
5238 if (consumed)
5239 *consumed = size;
5240 Py_INCREF(unicode_empty);
5241 return unicode_empty;
Tim Peters772747b2001-08-09 22:21:55 +00005242 }
Antoine Pitrou63065d72012-05-15 23:48:04 +02005243
Antoine Pitrouab868312009-01-10 15:40:25 +00005244#ifdef BYTEORDER_IS_LITTLE_ENDIAN
Antoine Pitrou63065d72012-05-15 23:48:04 +02005245 native_ordering = bo <= 0;
Antoine Pitrouab868312009-01-10 15:40:25 +00005246#else
Antoine Pitrou63065d72012-05-15 23:48:04 +02005247 native_ordering = bo >= 0;
Antoine Pitrouab868312009-01-10 15:40:25 +00005248#endif
Tim Peters772747b2001-08-09 22:21:55 +00005249
Antoine Pitrou63065d72012-05-15 23:48:04 +02005250 /* Note: size will always be longer than the resulting Unicode
5251 character count */
5252 unicode = PyUnicode_New((e - q + 1) / 2, 127);
5253 if (!unicode)
5254 return NULL;
5255
5256 outpos = 0;
5257 while (1) {
5258 Py_UCS4 ch = 0;
5259 if (e - q >= 2) {
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005260 int kind = PyUnicode_KIND(unicode);
Antoine Pitrou63065d72012-05-15 23:48:04 +02005261 if (kind == PyUnicode_1BYTE_KIND) {
5262 if (PyUnicode_IS_ASCII(unicode))
5263 ch = asciilib_utf16_decode(&q, e,
5264 PyUnicode_1BYTE_DATA(unicode), &outpos,
5265 native_ordering);
5266 else
5267 ch = ucs1lib_utf16_decode(&q, e,
5268 PyUnicode_1BYTE_DATA(unicode), &outpos,
5269 native_ordering);
5270 } else if (kind == PyUnicode_2BYTE_KIND) {
5271 ch = ucs2lib_utf16_decode(&q, e,
5272 PyUnicode_2BYTE_DATA(unicode), &outpos,
5273 native_ordering);
5274 } else {
5275 assert(kind == PyUnicode_4BYTE_KIND);
5276 ch = ucs4lib_utf16_decode(&q, e,
5277 PyUnicode_4BYTE_DATA(unicode), &outpos,
5278 native_ordering);
Antoine Pitrouab868312009-01-10 15:40:25 +00005279 }
Antoine Pitrouab868312009-01-10 15:40:25 +00005280 }
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005281
Antoine Pitrou63065d72012-05-15 23:48:04 +02005282 switch (ch)
5283 {
5284 case 0:
5285 /* remaining byte at the end? (size should be even) */
5286 if (q == e || consumed)
5287 goto End;
5288 errmsg = "truncated data";
5289 startinpos = ((const char *)q) - starts;
5290 endinpos = ((const char *)e) - starts;
5291 break;
5292 /* The remaining input chars are ignored if the callback
5293 chooses to skip the input */
5294 case 1:
5295 errmsg = "unexpected end of data";
5296 startinpos = ((const char *)q) - 2 - starts;
5297 endinpos = ((const char *)e) - starts;
5298 break;
5299 case 2:
5300 errmsg = "illegal encoding";
5301 startinpos = ((const char *)q) - 2 - starts;
5302 endinpos = startinpos + 2;
5303 break;
5304 case 3:
5305 errmsg = "illegal UTF-16 surrogate";
5306 startinpos = ((const char *)q) - 4 - starts;
5307 endinpos = startinpos + 2;
5308 break;
5309 default:
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005310 if (unicode_putchar(&unicode, &outpos, ch) < 0)
5311 goto onError;
Benjamin Peterson29060642009-01-31 22:14:21 +00005312 continue;
5313 }
5314
Benjamin Peterson29060642009-01-31 22:14:21 +00005315 if (unicode_decode_call_errorhandler(
Antoine Pitrouab868312009-01-10 15:40:25 +00005316 errors,
5317 &errorHandler,
5318 "utf16", errmsg,
5319 &starts,
5320 (const char **)&e,
5321 &startinpos,
5322 &endinpos,
5323 &exc,
5324 (const char **)&q,
5325 &unicode,
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005326 &outpos))
Benjamin Peterson29060642009-01-31 22:14:21 +00005327 goto onError;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005328 }
5329
Antoine Pitrou63065d72012-05-15 23:48:04 +02005330End:
Walter Dörwald69652032004-09-07 20:24:22 +00005331 if (consumed)
Benjamin Peterson29060642009-01-31 22:14:21 +00005332 *consumed = (const char *)q-starts;
Walter Dörwald69652032004-09-07 20:24:22 +00005333
Guido van Rossumd57fd912000-03-10 22:53:23 +00005334 /* Adjust length */
Victor Stinner16e6a802011-12-12 13:24:15 +01005335 if (unicode_resize(&unicode, outpos) < 0)
Guido van Rossumd57fd912000-03-10 22:53:23 +00005336 goto onError;
5337
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005338 Py_XDECREF(errorHandler);
5339 Py_XDECREF(exc);
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01005340 return unicode_result(unicode);
Guido van Rossumd57fd912000-03-10 22:53:23 +00005341
Benjamin Peterson29060642009-01-31 22:14:21 +00005342 onError:
Guido van Rossumd57fd912000-03-10 22:53:23 +00005343 Py_DECREF(unicode);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005344 Py_XDECREF(errorHandler);
5345 Py_XDECREF(exc);
Guido van Rossumd57fd912000-03-10 22:53:23 +00005346 return NULL;
5347}
5348
Tim Peters772747b2001-08-09 22:21:55 +00005349PyObject *
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005350_PyUnicode_EncodeUTF16(PyObject *str,
5351 const char *errors,
5352 int byteorder)
Guido van Rossumd57fd912000-03-10 22:53:23 +00005353{
Antoine Pitrou27f6a3b2012-06-15 22:15:23 +02005354 enum PyUnicode_Kind kind;
5355 const void *data;
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005356 Py_ssize_t len;
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00005357 PyObject *v;
Antoine Pitrou27f6a3b2012-06-15 22:15:23 +02005358 unsigned short *out;
5359 Py_ssize_t bytesize;
5360 Py_ssize_t pairs;
5361#ifdef WORDS_BIGENDIAN
5362 int native_ordering = byteorder >= 0;
Tim Peters772747b2001-08-09 22:21:55 +00005363#else
Antoine Pitrou27f6a3b2012-06-15 22:15:23 +02005364 int native_ordering = byteorder <= 0;
Tim Peters772747b2001-08-09 22:21:55 +00005365#endif
5366
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005367 if (!PyUnicode_Check(str)) {
5368 PyErr_BadArgument();
5369 return NULL;
5370 }
Benjamin Petersonbac79492012-01-14 13:34:47 -05005371 if (PyUnicode_READY(str) == -1)
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005372 return NULL;
5373 kind = PyUnicode_KIND(str);
5374 data = PyUnicode_DATA(str);
5375 len = PyUnicode_GET_LENGTH(str);
Victor Stinner0e368262011-11-10 20:12:49 +01005376
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005377 pairs = 0;
Antoine Pitrou27f6a3b2012-06-15 22:15:23 +02005378 if (kind == PyUnicode_4BYTE_KIND) {
5379 const Py_UCS4 *in = (const Py_UCS4 *)data;
5380 const Py_UCS4 *end = in + len;
5381 while (in < end)
5382 if (*in++ >= 0x10000)
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005383 pairs++;
Antoine Pitrou27f6a3b2012-06-15 22:15:23 +02005384 }
5385 if (len > PY_SSIZE_T_MAX / 2 - pairs - (byteorder == 0))
Benjamin Peterson29060642009-01-31 22:14:21 +00005386 return PyErr_NoMemory();
Antoine Pitrou27f6a3b2012-06-15 22:15:23 +02005387 bytesize = (len + pairs + (byteorder == 0)) * 2;
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00005388 v = PyBytes_FromStringAndSize(NULL, bytesize);
Guido van Rossumd57fd912000-03-10 22:53:23 +00005389 if (v == NULL)
5390 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005391
Antoine Pitrou27f6a3b2012-06-15 22:15:23 +02005392 /* output buffer is 2-bytes aligned */
5393 assert(((Py_uintptr_t)PyBytes_AS_STRING(v) & 1) == 0);
5394 out = (unsigned short *)PyBytes_AS_STRING(v);
Guido van Rossumd57fd912000-03-10 22:53:23 +00005395 if (byteorder == 0)
Antoine Pitrou27f6a3b2012-06-15 22:15:23 +02005396 *out++ = 0xFEFF;
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005397 if (len == 0)
Guido van Rossum98297ee2007-11-06 21:34:58 +00005398 goto done;
Tim Peters772747b2001-08-09 22:21:55 +00005399
Antoine Pitrou27f6a3b2012-06-15 22:15:23 +02005400 switch (kind) {
5401 case PyUnicode_1BYTE_KIND: {
5402 ucs1lib_utf16_encode(out, (const Py_UCS1 *)data, len, native_ordering);
5403 break;
Tim Peters772747b2001-08-09 22:21:55 +00005404 }
Antoine Pitrou27f6a3b2012-06-15 22:15:23 +02005405 case PyUnicode_2BYTE_KIND: {
5406 ucs2lib_utf16_encode(out, (const Py_UCS2 *)data, len, native_ordering);
5407 break;
Tim Peters772747b2001-08-09 22:21:55 +00005408 }
Antoine Pitrou27f6a3b2012-06-15 22:15:23 +02005409 case PyUnicode_4BYTE_KIND: {
5410 ucs4lib_utf16_encode(out, (const Py_UCS4 *)data, len, native_ordering);
5411 break;
5412 }
5413 default:
5414 assert(0);
Martin v. Löwis0ba70cc2001-06-26 22:22:37 +00005415 }
Guido van Rossum98297ee2007-11-06 21:34:58 +00005416
5417 done:
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00005418 return v;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005419}
5420
Alexander Belopolsky40018472011-02-26 01:02:56 +00005421PyObject *
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005422PyUnicode_EncodeUTF16(const Py_UNICODE *s,
5423 Py_ssize_t size,
5424 const char *errors,
5425 int byteorder)
5426{
5427 PyObject *result;
5428 PyObject *tmp = PyUnicode_FromUnicode(s, size);
5429 if (tmp == NULL)
5430 return NULL;
5431 result = _PyUnicode_EncodeUTF16(tmp, errors, byteorder);
5432 Py_DECREF(tmp);
5433 return result;
5434}
5435
5436PyObject *
Alexander Belopolsky40018472011-02-26 01:02:56 +00005437PyUnicode_AsUTF16String(PyObject *unicode)
Guido van Rossumd57fd912000-03-10 22:53:23 +00005438{
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005439 return _PyUnicode_EncodeUTF16(unicode, NULL, 0);
Guido van Rossumd57fd912000-03-10 22:53:23 +00005440}
5441
5442/* --- Unicode Escape Codec ----------------------------------------------- */
5443
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02005444/* Helper function for PyUnicode_DecodeUnicodeEscape, determines
5445 if all the escapes in the string make it still a valid ASCII string.
5446 Returns -1 if any escapes were found which cause the string to
5447 pop out of ASCII range. Otherwise returns the length of the
5448 required buffer to hold the string.
5449 */
Antoine Pitrou53bb5482011-10-10 23:49:24 +02005450static Py_ssize_t
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02005451length_of_escaped_ascii_string(const char *s, Py_ssize_t size)
5452{
5453 const unsigned char *p = (const unsigned char *)s;
5454 const unsigned char *end = p + size;
5455 Py_ssize_t length = 0;
5456
5457 if (size < 0)
5458 return -1;
5459
5460 for (; p < end; ++p) {
5461 if (*p > 127) {
5462 /* Non-ASCII */
5463 return -1;
5464 }
5465 else if (*p != '\\') {
5466 /* Normal character */
5467 ++length;
5468 }
5469 else {
5470 /* Backslash-escape, check next char */
5471 ++p;
5472 /* Escape sequence reaches till end of string or
5473 non-ASCII follow-up. */
5474 if (p >= end || *p > 127)
5475 return -1;
5476 switch (*p) {
5477 case '\n':
5478 /* backslash + \n result in zero characters */
5479 break;
5480 case '\\': case '\'': case '\"':
5481 case 'b': case 'f': case 't':
5482 case 'n': case 'r': case 'v': case 'a':
5483 ++length;
5484 break;
5485 case '0': case '1': case '2': case '3':
5486 case '4': case '5': case '6': case '7':
5487 case 'x': case 'u': case 'U': case 'N':
5488 /* these do not guarantee ASCII characters */
5489 return -1;
5490 default:
5491 /* count the backslash + the other character */
5492 length += 2;
5493 }
5494 }
5495 }
5496 return length;
5497}
5498
Fredrik Lundh06d12682001-01-24 07:59:11 +00005499static _PyUnicode_Name_CAPI *ucnhash_CAPI = NULL;
Marc-André Lemburg0f774e32000-06-28 16:43:35 +00005500
Alexander Belopolsky40018472011-02-26 01:02:56 +00005501PyObject *
5502PyUnicode_DecodeUnicodeEscape(const char *s,
Ezio Melotti2aa2b3b2011-09-29 00:58:57 +03005503 Py_ssize_t size,
Victor Stinnerc17f5402011-09-29 00:16:58 +02005504 const char *errors)
Guido van Rossumd57fd912000-03-10 22:53:23 +00005505{
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005506 const char *starts = s;
Martin v. Löwis18e16552006-02-15 17:27:45 +00005507 Py_ssize_t startinpos;
5508 Py_ssize_t endinpos;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02005509 int j;
Victor Stinner7931d9a2011-11-04 00:22:48 +01005510 PyObject *v;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005511 const char *end;
Fredrik Lundhccc74732001-02-18 22:13:49 +00005512 char* message;
5513 Py_UCS4 chr = 0xffffffff; /* in case 'getcode' messes up */
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005514 PyObject *errorHandler = NULL;
5515 PyObject *exc = NULL;
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005516 Py_ssize_t len;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02005517 Py_ssize_t i;
Fredrik Lundhccc74732001-02-18 22:13:49 +00005518
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005519 len = length_of_escaped_ascii_string(s, size);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02005520
5521 /* After length_of_escaped_ascii_string() there are two alternatives,
5522 either the string is pure ASCII with named escapes like \n, etc.
5523 and we determined it's exact size (common case)
5524 or it contains \x, \u, ... escape sequences. then we create a
5525 legacy wchar string and resize it at the end of this function. */
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005526 if (len >= 0) {
5527 v = PyUnicode_New(len, 127);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02005528 if (!v)
5529 goto onError;
5530 assert(PyUnicode_KIND(v) == PyUnicode_1BYTE_KIND);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02005531 }
5532 else {
5533 /* Escaped strings will always be longer than the resulting
5534 Unicode string, so we start with size here and then reduce the
5535 length after conversion to the true value.
5536 (but if the error callback returns a long replacement string
5537 we'll have to allocate more space) */
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005538 v = PyUnicode_New(size, 127);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02005539 if (!v)
5540 goto onError;
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005541 len = size;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02005542 }
5543
Guido van Rossumd57fd912000-03-10 22:53:23 +00005544 if (size == 0)
Victor Stinner7931d9a2011-11-04 00:22:48 +01005545 return v;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02005546 i = 0;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005547 end = s + size;
Fredrik Lundhccc74732001-02-18 22:13:49 +00005548
Guido van Rossumd57fd912000-03-10 22:53:23 +00005549 while (s < end) {
5550 unsigned char c;
Victor Stinner24729f32011-11-10 20:31:37 +01005551 Py_UCS4 x;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005552 int digits;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005553
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005554 /* The only case in which i == ascii_length is a backslash
5555 followed by a newline. */
5556 assert(i <= len);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02005557
Guido van Rossumd57fd912000-03-10 22:53:23 +00005558 /* Non-escape characters are interpreted as Unicode ordinals */
5559 if (*s != '\\') {
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005560 if (unicode_putchar(&v, &i, (unsigned char) *s++) < 0)
5561 goto onError;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005562 continue;
5563 }
5564
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005565 startinpos = s-starts;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005566 /* \ - Escapes */
5567 s++;
Guido van Rossum8ce8a782007-11-01 19:42:39 +00005568 c = *s++;
5569 if (s > end)
5570 c = '\0'; /* Invalid after \ */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02005571
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005572 /* The only case in which i == ascii_length is a backslash
5573 followed by a newline. */
5574 assert(i < len || (i == len && c == '\n'));
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02005575
Guido van Rossum8ce8a782007-11-01 19:42:39 +00005576 switch (c) {
Guido van Rossumd57fd912000-03-10 22:53:23 +00005577
Benjamin Peterson29060642009-01-31 22:14:21 +00005578 /* \x escapes */
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005579#define WRITECHAR(ch) \
5580 do { \
5581 if (unicode_putchar(&v, &i, ch) < 0) \
5582 goto onError; \
5583 }while(0)
5584
Guido van Rossumd57fd912000-03-10 22:53:23 +00005585 case '\n': break;
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005586 case '\\': WRITECHAR('\\'); break;
5587 case '\'': WRITECHAR('\''); break;
5588 case '\"': WRITECHAR('\"'); break;
5589 case 'b': WRITECHAR('\b'); break;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02005590 /* FF */
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005591 case 'f': WRITECHAR('\014'); break;
5592 case 't': WRITECHAR('\t'); break;
5593 case 'n': WRITECHAR('\n'); break;
5594 case 'r': WRITECHAR('\r'); break;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02005595 /* VT */
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005596 case 'v': WRITECHAR('\013'); break;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02005597 /* BEL, not classic C */
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005598 case 'a': WRITECHAR('\007'); break;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005599
Benjamin Peterson29060642009-01-31 22:14:21 +00005600 /* \OOO (octal) escapes */
Guido van Rossumd57fd912000-03-10 22:53:23 +00005601 case '0': case '1': case '2': case '3':
5602 case '4': case '5': case '6': case '7':
Guido van Rossum0e4f6572000-05-01 21:27:20 +00005603 x = s[-1] - '0';
Guido van Rossum8ce8a782007-11-01 19:42:39 +00005604 if (s < end && '0' <= *s && *s <= '7') {
Guido van Rossum0e4f6572000-05-01 21:27:20 +00005605 x = (x<<3) + *s++ - '0';
Guido van Rossum8ce8a782007-11-01 19:42:39 +00005606 if (s < end && '0' <= *s && *s <= '7')
Guido van Rossum0e4f6572000-05-01 21:27:20 +00005607 x = (x<<3) + *s++ - '0';
Guido van Rossumd57fd912000-03-10 22:53:23 +00005608 }
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005609 WRITECHAR(x);
Guido van Rossumd57fd912000-03-10 22:53:23 +00005610 break;
5611
Benjamin Peterson29060642009-01-31 22:14:21 +00005612 /* hex escapes */
5613 /* \xXX */
Guido van Rossumd57fd912000-03-10 22:53:23 +00005614 case 'x':
Fredrik Lundhccc74732001-02-18 22:13:49 +00005615 digits = 2;
5616 message = "truncated \\xXX escape";
5617 goto hexescape;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005618
Benjamin Peterson29060642009-01-31 22:14:21 +00005619 /* \uXXXX */
Guido van Rossumd57fd912000-03-10 22:53:23 +00005620 case 'u':
Fredrik Lundhccc74732001-02-18 22:13:49 +00005621 digits = 4;
5622 message = "truncated \\uXXXX escape";
5623 goto hexescape;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005624
Benjamin Peterson29060642009-01-31 22:14:21 +00005625 /* \UXXXXXXXX */
Fredrik Lundhdf846752000-09-03 11:29:49 +00005626 case 'U':
Fredrik Lundhccc74732001-02-18 22:13:49 +00005627 digits = 8;
5628 message = "truncated \\UXXXXXXXX escape";
5629 hexescape:
5630 chr = 0;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005631 if (s+digits>end) {
5632 endinpos = size;
5633 if (unicode_decode_call_errorhandler(
Benjamin Peterson29060642009-01-31 22:14:21 +00005634 errors, &errorHandler,
5635 "unicodeescape", "end of string in escape sequence",
5636 &starts, &end, &startinpos, &endinpos, &exc, &s,
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005637 &v, &i))
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005638 goto onError;
5639 goto nextByte;
5640 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02005641 for (j = 0; j < digits; ++j) {
5642 c = (unsigned char) s[j];
David Malcolm96960882010-11-05 17:23:41 +00005643 if (!Py_ISXDIGIT(c)) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02005644 endinpos = (s+j+1)-starts;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005645 if (unicode_decode_call_errorhandler(
Benjamin Peterson29060642009-01-31 22:14:21 +00005646 errors, &errorHandler,
5647 "unicodeescape", message,
5648 &starts, &end, &startinpos, &endinpos, &exc, &s,
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005649 &v, &i))
Fredrik Lundhdf846752000-09-03 11:29:49 +00005650 goto onError;
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005651 len = PyUnicode_GET_LENGTH(v);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005652 goto nextByte;
Fredrik Lundhdf846752000-09-03 11:29:49 +00005653 }
5654 chr = (chr<<4) & ~0xF;
5655 if (c >= '0' && c <= '9')
5656 chr += c - '0';
5657 else if (c >= 'a' && c <= 'f')
5658 chr += 10 + c - 'a';
5659 else
5660 chr += 10 + c - 'A';
5661 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02005662 s += j;
Jeremy Hylton504de6b2003-10-06 05:08:26 +00005663 if (chr == 0xffffffff && PyErr_Occurred())
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005664 /* _decoding_error will have already written into the
5665 target buffer. */
5666 break;
Fredrik Lundhccc74732001-02-18 22:13:49 +00005667 store:
Fredrik Lundhdf846752000-09-03 11:29:49 +00005668 /* when we get here, chr is a 32-bit unicode character */
Victor Stinner8faf8212011-12-08 22:14:11 +01005669 if (chr <= MAX_UNICODE) {
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005670 WRITECHAR(chr);
Fredrik Lundhdf846752000-09-03 11:29:49 +00005671 } else {
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005672 endinpos = s-starts;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005673 if (unicode_decode_call_errorhandler(
Benjamin Peterson29060642009-01-31 22:14:21 +00005674 errors, &errorHandler,
5675 "unicodeescape", "illegal Unicode character",
5676 &starts, &end, &startinpos, &endinpos, &exc, &s,
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005677 &v, &i))
Fredrik Lundhdf846752000-09-03 11:29:49 +00005678 goto onError;
5679 }
Fredrik Lundhccc74732001-02-18 22:13:49 +00005680 break;
5681
Benjamin Peterson29060642009-01-31 22:14:21 +00005682 /* \N{name} */
Fredrik Lundhccc74732001-02-18 22:13:49 +00005683 case 'N':
5684 message = "malformed \\N character escape";
5685 if (ucnhash_CAPI == NULL) {
5686 /* load the unicode data module */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02005687 ucnhash_CAPI = (_PyUnicode_Name_CAPI *)PyCapsule_Import(
5688 PyUnicodeData_CAPSULE_NAME, 1);
Fredrik Lundhccc74732001-02-18 22:13:49 +00005689 if (ucnhash_CAPI == NULL)
5690 goto ucnhashError;
5691 }
5692 if (*s == '{') {
5693 const char *start = s+1;
5694 /* look for the closing brace */
5695 while (*s != '}' && s < end)
5696 s++;
5697 if (s > start && s < end && *s == '}') {
5698 /* found a name. look it up in the unicode database */
5699 message = "unknown Unicode character name";
5700 s++;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02005701 if (ucnhash_CAPI->getcode(NULL, start, (int)(s-start-1),
Ezio Melotti931b8aa2011-10-21 21:57:36 +03005702 &chr, 0))
Fredrik Lundhccc74732001-02-18 22:13:49 +00005703 goto store;
5704 }
5705 }
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005706 endinpos = s-starts;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005707 if (unicode_decode_call_errorhandler(
Benjamin Peterson29060642009-01-31 22:14:21 +00005708 errors, &errorHandler,
5709 "unicodeescape", message,
5710 &starts, &end, &startinpos, &endinpos, &exc, &s,
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005711 &v, &i))
Fredrik Lundhccc74732001-02-18 22:13:49 +00005712 goto onError;
Fredrik Lundhccc74732001-02-18 22:13:49 +00005713 break;
5714
5715 default:
Walter Dörwald8c077222002-03-25 11:16:18 +00005716 if (s > end) {
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005717 message = "\\ at end of string";
5718 s--;
5719 endinpos = s-starts;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005720 if (unicode_decode_call_errorhandler(
Benjamin Peterson29060642009-01-31 22:14:21 +00005721 errors, &errorHandler,
5722 "unicodeescape", message,
5723 &starts, &end, &startinpos, &endinpos, &exc, &s,
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005724 &v, &i))
Walter Dörwald8c077222002-03-25 11:16:18 +00005725 goto onError;
5726 }
5727 else {
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005728 WRITECHAR('\\');
5729 WRITECHAR(s[-1]);
Walter Dörwald8c077222002-03-25 11:16:18 +00005730 }
Fredrik Lundhccc74732001-02-18 22:13:49 +00005731 break;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005732 }
Benjamin Peterson29060642009-01-31 22:14:21 +00005733 nextByte:
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005734 ;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005735 }
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005736#undef WRITECHAR
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02005737
Victor Stinner16e6a802011-12-12 13:24:15 +01005738 if (unicode_resize(&v, i) < 0)
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005739 goto onError;
Walter Dörwaldd4ade082003-08-15 15:00:26 +00005740 Py_XDECREF(errorHandler);
5741 Py_XDECREF(exc);
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01005742 return unicode_result(v);
Walter Dörwald8c077222002-03-25 11:16:18 +00005743
Benjamin Peterson29060642009-01-31 22:14:21 +00005744 ucnhashError:
Fredrik Lundh06d12682001-01-24 07:59:11 +00005745 PyErr_SetString(
5746 PyExc_UnicodeError,
5747 "\\N escapes not supported (can't load unicodedata module)"
5748 );
Hye-Shik Chang4af5c8c2006-03-07 15:39:21 +00005749 Py_XDECREF(v);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005750 Py_XDECREF(errorHandler);
5751 Py_XDECREF(exc);
Fredrik Lundhf6056062001-01-20 11:15:25 +00005752 return NULL;
5753
Benjamin Peterson29060642009-01-31 22:14:21 +00005754 onError:
Guido van Rossumd57fd912000-03-10 22:53:23 +00005755 Py_XDECREF(v);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005756 Py_XDECREF(errorHandler);
5757 Py_XDECREF(exc);
Guido van Rossumd57fd912000-03-10 22:53:23 +00005758 return NULL;
5759}
5760
5761/* Return a Unicode-Escape string version of the Unicode object.
5762
5763 If quotes is true, the string is enclosed in u"" or u'' quotes as
5764 appropriate.
5765
5766*/
5767
Alexander Belopolsky40018472011-02-26 01:02:56 +00005768PyObject *
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005769PyUnicode_AsUnicodeEscapeString(PyObject *unicode)
Guido van Rossumd57fd912000-03-10 22:53:23 +00005770{
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005771 Py_ssize_t i, len;
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00005772 PyObject *repr;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005773 char *p;
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005774 int kind;
5775 void *data;
5776 Py_ssize_t expandsize = 0;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005777
Thomas Wouters89f507f2006-12-13 04:49:30 +00005778 /* Initial allocation is based on the longest-possible unichr
5779 escape.
5780
5781 In wide (UTF-32) builds '\U00xxxxxx' is 10 chars per source
5782 unichr, so in this case it's the longest unichr escape. In
5783 narrow (UTF-16) builds this is five chars per source unichr
5784 since there are two unichrs in the surrogate pair, so in narrow
5785 (UTF-16) builds it's not the longest unichr escape.
5786
5787 In wide or narrow builds '\uxxxx' is 6 chars per source unichr,
5788 so in the narrow (UTF-16) build case it's the longest unichr
5789 escape.
5790 */
5791
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005792 if (!PyUnicode_Check(unicode)) {
5793 PyErr_BadArgument();
5794 return NULL;
5795 }
Benjamin Petersonbac79492012-01-14 13:34:47 -05005796 if (PyUnicode_READY(unicode) == -1)
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005797 return NULL;
5798 len = PyUnicode_GET_LENGTH(unicode);
5799 kind = PyUnicode_KIND(unicode);
5800 data = PyUnicode_DATA(unicode);
Benjamin Petersonead6b532011-12-20 17:23:42 -06005801 switch (kind) {
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005802 case PyUnicode_1BYTE_KIND: expandsize = 4; break;
5803 case PyUnicode_2BYTE_KIND: expandsize = 6; break;
5804 case PyUnicode_4BYTE_KIND: expandsize = 10; break;
5805 }
5806
5807 if (len == 0)
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00005808 return PyBytes_FromStringAndSize(NULL, 0);
5809
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005810 if (len > (PY_SSIZE_T_MAX - 2 - 1) / expandsize)
Benjamin Peterson29060642009-01-31 22:14:21 +00005811 return PyErr_NoMemory();
Neal Norwitz3ce5d922008-08-24 07:08:55 +00005812
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00005813 repr = PyBytes_FromStringAndSize(NULL,
Benjamin Peterson29060642009-01-31 22:14:21 +00005814 2
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005815 + expandsize*len
Benjamin Peterson29060642009-01-31 22:14:21 +00005816 + 1);
Guido van Rossumd57fd912000-03-10 22:53:23 +00005817 if (repr == NULL)
5818 return NULL;
5819
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00005820 p = PyBytes_AS_STRING(repr);
Guido van Rossumd57fd912000-03-10 22:53:23 +00005821
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005822 for (i = 0; i < len; i++) {
Victor Stinner3326cb62011-11-10 20:15:25 +01005823 Py_UCS4 ch = PyUnicode_READ(kind, data, i);
Marc-André Lemburg80d1dd52001-07-25 16:05:59 +00005824
Walter Dörwald79e913e2007-05-12 11:08:06 +00005825 /* Escape backslashes */
5826 if (ch == '\\') {
Guido van Rossumd57fd912000-03-10 22:53:23 +00005827 *p++ = '\\';
5828 *p++ = (char) ch;
Walter Dörwald79e913e2007-05-12 11:08:06 +00005829 continue;
Tim Petersced69f82003-09-16 20:30:58 +00005830 }
Marc-André Lemburg80d1dd52001-07-25 16:05:59 +00005831
Martin v. Löwis0ba70cc2001-06-26 22:22:37 +00005832 /* Map 21-bit characters to '\U00xxxxxx' */
5833 else if (ch >= 0x10000) {
Victor Stinner8faf8212011-12-08 22:14:11 +01005834 assert(ch <= MAX_UNICODE);
Martin v. Löwis0ba70cc2001-06-26 22:22:37 +00005835 *p++ = '\\';
5836 *p++ = 'U';
Victor Stinnerf5cff562011-10-14 02:13:11 +02005837 *p++ = Py_hexdigits[(ch >> 28) & 0x0000000F];
5838 *p++ = Py_hexdigits[(ch >> 24) & 0x0000000F];
5839 *p++ = Py_hexdigits[(ch >> 20) & 0x0000000F];
5840 *p++ = Py_hexdigits[(ch >> 16) & 0x0000000F];
5841 *p++ = Py_hexdigits[(ch >> 12) & 0x0000000F];
5842 *p++ = Py_hexdigits[(ch >> 8) & 0x0000000F];
5843 *p++ = Py_hexdigits[(ch >> 4) & 0x0000000F];
5844 *p++ = Py_hexdigits[ch & 0x0000000F];
Benjamin Peterson29060642009-01-31 22:14:21 +00005845 continue;
Martin v. Löwis0ba70cc2001-06-26 22:22:37 +00005846 }
Marc-André Lemburg6c6bfb72001-07-20 17:39:11 +00005847
Guido van Rossumd57fd912000-03-10 22:53:23 +00005848 /* Map 16-bit characters to '\uxxxx' */
Marc-André Lemburg6c6bfb72001-07-20 17:39:11 +00005849 if (ch >= 256) {
Guido van Rossumd57fd912000-03-10 22:53:23 +00005850 *p++ = '\\';
5851 *p++ = 'u';
Victor Stinnerf5cff562011-10-14 02:13:11 +02005852 *p++ = Py_hexdigits[(ch >> 12) & 0x000F];
5853 *p++ = Py_hexdigits[(ch >> 8) & 0x000F];
5854 *p++ = Py_hexdigits[(ch >> 4) & 0x000F];
5855 *p++ = Py_hexdigits[ch & 0x000F];
Guido van Rossumd57fd912000-03-10 22:53:23 +00005856 }
Marc-André Lemburg80d1dd52001-07-25 16:05:59 +00005857
Ka-Ping Yeefa004ad2001-01-24 17:19:08 +00005858 /* Map special whitespace to '\t', \n', '\r' */
5859 else if (ch == '\t') {
5860 *p++ = '\\';
5861 *p++ = 't';
5862 }
5863 else if (ch == '\n') {
5864 *p++ = '\\';
5865 *p++ = 'n';
5866 }
5867 else if (ch == '\r') {
5868 *p++ = '\\';
5869 *p++ = 'r';
5870 }
Marc-André Lemburg80d1dd52001-07-25 16:05:59 +00005871
Ka-Ping Yeefa004ad2001-01-24 17:19:08 +00005872 /* Map non-printable US ASCII to '\xhh' */
Marc-André Lemburg11326de2001-11-28 12:56:20 +00005873 else if (ch < ' ' || ch >= 0x7F) {
Guido van Rossumd57fd912000-03-10 22:53:23 +00005874 *p++ = '\\';
Ka-Ping Yeefa004ad2001-01-24 17:19:08 +00005875 *p++ = 'x';
Victor Stinnerf5cff562011-10-14 02:13:11 +02005876 *p++ = Py_hexdigits[(ch >> 4) & 0x000F];
5877 *p++ = Py_hexdigits[ch & 0x000F];
Tim Petersced69f82003-09-16 20:30:58 +00005878 }
Marc-André Lemburg80d1dd52001-07-25 16:05:59 +00005879
Guido van Rossumd57fd912000-03-10 22:53:23 +00005880 /* Copy everything else as-is */
5881 else
5882 *p++ = (char) ch;
5883 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00005884
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00005885 assert(p - PyBytes_AS_STRING(repr) > 0);
5886 if (_PyBytes_Resize(&repr, p - PyBytes_AS_STRING(repr)) < 0)
5887 return NULL;
5888 return repr;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005889}
5890
Alexander Belopolsky40018472011-02-26 01:02:56 +00005891PyObject *
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005892PyUnicode_EncodeUnicodeEscape(const Py_UNICODE *s,
5893 Py_ssize_t size)
Guido van Rossumd57fd912000-03-10 22:53:23 +00005894{
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005895 PyObject *result;
5896 PyObject *tmp = PyUnicode_FromUnicode(s, size);
5897 if (tmp == NULL)
Guido van Rossumd57fd912000-03-10 22:53:23 +00005898 return NULL;
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005899 result = PyUnicode_AsUnicodeEscapeString(tmp);
5900 Py_DECREF(tmp);
5901 return result;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005902}
5903
5904/* --- Raw Unicode Escape Codec ------------------------------------------- */
5905
Alexander Belopolsky40018472011-02-26 01:02:56 +00005906PyObject *
5907PyUnicode_DecodeRawUnicodeEscape(const char *s,
Ezio Melotti2aa2b3b2011-09-29 00:58:57 +03005908 Py_ssize_t size,
5909 const char *errors)
Guido van Rossumd57fd912000-03-10 22:53:23 +00005910{
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005911 const char *starts = s;
Martin v. Löwis18e16552006-02-15 17:27:45 +00005912 Py_ssize_t startinpos;
5913 Py_ssize_t endinpos;
5914 Py_ssize_t outpos;
Victor Stinner7931d9a2011-11-04 00:22:48 +01005915 PyObject *v;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005916 const char *end;
5917 const char *bs;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005918 PyObject *errorHandler = NULL;
5919 PyObject *exc = NULL;
Tim Petersced69f82003-09-16 20:30:58 +00005920
Guido van Rossumd57fd912000-03-10 22:53:23 +00005921 /* Escaped strings will always be longer than the resulting
5922 Unicode string, so we start with size here and then reduce the
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005923 length after conversion to the true value. (But decoding error
5924 handler might have to resize the string) */
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005925 v = PyUnicode_New(size, 127);
Guido van Rossumd57fd912000-03-10 22:53:23 +00005926 if (v == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00005927 goto onError;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005928 if (size == 0)
Victor Stinner7931d9a2011-11-04 00:22:48 +01005929 return v;
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005930 outpos = 0;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005931 end = s + size;
5932 while (s < end) {
Benjamin Peterson29060642009-01-31 22:14:21 +00005933 unsigned char c;
5934 Py_UCS4 x;
5935 int i;
Martin v. Löwis9a3a9f72003-05-18 12:31:09 +00005936 int count;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005937
Benjamin Peterson29060642009-01-31 22:14:21 +00005938 /* Non-escape characters are interpreted as Unicode ordinals */
5939 if (*s != '\\') {
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 continue;
Benjamin Peterson14339b62009-01-31 16:36:08 +00005943 }
Benjamin Peterson29060642009-01-31 22:14:21 +00005944 startinpos = s-starts;
5945
5946 /* \u-escapes are only interpreted iff the number of leading
5947 backslashes if odd */
5948 bs = s;
5949 for (;s < end;) {
5950 if (*s != '\\')
5951 break;
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005952 if (unicode_putchar(&v, &outpos, (unsigned char)*s++) < 0)
5953 goto onError;
Benjamin Peterson29060642009-01-31 22:14:21 +00005954 }
5955 if (((s - bs) & 1) == 0 ||
5956 s >= end ||
5957 (*s != 'u' && *s != 'U')) {
5958 continue;
5959 }
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005960 outpos--;
Benjamin Peterson29060642009-01-31 22:14:21 +00005961 count = *s=='u' ? 4 : 8;
5962 s++;
5963
5964 /* \uXXXX with 4 hex digits, \Uxxxxxxxx with 8 */
Benjamin Peterson29060642009-01-31 22:14:21 +00005965 for (x = 0, i = 0; i < count; ++i, ++s) {
5966 c = (unsigned char)*s;
David Malcolm96960882010-11-05 17:23:41 +00005967 if (!Py_ISXDIGIT(c)) {
Benjamin Peterson29060642009-01-31 22:14:21 +00005968 endinpos = s-starts;
5969 if (unicode_decode_call_errorhandler(
5970 errors, &errorHandler,
5971 "rawunicodeescape", "truncated \\uXXXX",
5972 &starts, &end, &startinpos, &endinpos, &exc, &s,
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005973 &v, &outpos))
Benjamin Peterson29060642009-01-31 22:14:21 +00005974 goto onError;
5975 goto nextByte;
5976 }
5977 x = (x<<4) & ~0xF;
5978 if (c >= '0' && c <= '9')
5979 x += c - '0';
5980 else if (c >= 'a' && c <= 'f')
5981 x += 10 + c - 'a';
5982 else
5983 x += 10 + c - 'A';
5984 }
Victor Stinner8faf8212011-12-08 22:14:11 +01005985 if (x <= MAX_UNICODE) {
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005986 if (unicode_putchar(&v, &outpos, x) < 0)
5987 goto onError;
Christian Heimesfe337bf2008-03-23 21:54:12 +00005988 } else {
5989 endinpos = s-starts;
Martin v. Löwis9a3a9f72003-05-18 12:31:09 +00005990 if (unicode_decode_call_errorhandler(
5991 errors, &errorHandler,
5992 "rawunicodeescape", "\\Uxxxxxxxx out of range",
Benjamin Peterson29060642009-01-31 22:14:21 +00005993 &starts, &end, &startinpos, &endinpos, &exc, &s,
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005994 &v, &outpos))
Benjamin Peterson29060642009-01-31 22:14:21 +00005995 goto onError;
Martin v. Löwis9a3a9f72003-05-18 12:31:09 +00005996 }
Benjamin Peterson29060642009-01-31 22:14:21 +00005997 nextByte:
5998 ;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005999 }
Victor Stinner16e6a802011-12-12 13:24:15 +01006000 if (unicode_resize(&v, outpos) < 0)
Benjamin Peterson29060642009-01-31 22:14:21 +00006001 goto onError;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006002 Py_XDECREF(errorHandler);
6003 Py_XDECREF(exc);
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01006004 return unicode_result(v);
Tim Petersced69f82003-09-16 20:30:58 +00006005
Benjamin Peterson29060642009-01-31 22:14:21 +00006006 onError:
Guido van Rossumd57fd912000-03-10 22:53:23 +00006007 Py_XDECREF(v);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006008 Py_XDECREF(errorHandler);
6009 Py_XDECREF(exc);
Guido van Rossumd57fd912000-03-10 22:53:23 +00006010 return NULL;
6011}
6012
Martin v. Löwis1db7c132011-11-10 18:24:32 +01006013
Alexander Belopolsky40018472011-02-26 01:02:56 +00006014PyObject *
Martin v. Löwis1db7c132011-11-10 18:24:32 +01006015PyUnicode_AsRawUnicodeEscapeString(PyObject *unicode)
Guido van Rossumd57fd912000-03-10 22:53:23 +00006016{
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00006017 PyObject *repr;
Guido van Rossumd57fd912000-03-10 22:53:23 +00006018 char *p;
6019 char *q;
Martin v. Löwis1db7c132011-11-10 18:24:32 +01006020 Py_ssize_t expandsize, pos;
6021 int kind;
6022 void *data;
6023 Py_ssize_t len;
Guido van Rossumd57fd912000-03-10 22:53:23 +00006024
Martin v. Löwis1db7c132011-11-10 18:24:32 +01006025 if (!PyUnicode_Check(unicode)) {
6026 PyErr_BadArgument();
6027 return NULL;
6028 }
Benjamin Petersonbac79492012-01-14 13:34:47 -05006029 if (PyUnicode_READY(unicode) == -1)
Martin v. Löwis1db7c132011-11-10 18:24:32 +01006030 return NULL;
6031 kind = PyUnicode_KIND(unicode);
6032 data = PyUnicode_DATA(unicode);
6033 len = PyUnicode_GET_LENGTH(unicode);
Benjamin Peterson1518e872011-11-23 10:44:52 -06006034 /* 4 byte characters can take up 10 bytes, 2 byte characters can take up 6
6035 bytes, and 1 byte characters 4. */
6036 expandsize = kind * 2 + 2;
Victor Stinner0e368262011-11-10 20:12:49 +01006037
Martin v. Löwis1db7c132011-11-10 18:24:32 +01006038 if (len > PY_SSIZE_T_MAX / expandsize)
Benjamin Peterson29060642009-01-31 22:14:21 +00006039 return PyErr_NoMemory();
Benjamin Peterson14339b62009-01-31 16:36:08 +00006040
Martin v. Löwis1db7c132011-11-10 18:24:32 +01006041 repr = PyBytes_FromStringAndSize(NULL, expandsize * len);
Guido van Rossumd57fd912000-03-10 22:53:23 +00006042 if (repr == NULL)
6043 return NULL;
Martin v. Löwis1db7c132011-11-10 18:24:32 +01006044 if (len == 0)
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00006045 return repr;
Guido van Rossumd57fd912000-03-10 22:53:23 +00006046
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00006047 p = q = PyBytes_AS_STRING(repr);
Martin v. Löwis1db7c132011-11-10 18:24:32 +01006048 for (pos = 0; pos < len; pos++) {
6049 Py_UCS4 ch = PyUnicode_READ(kind, data, pos);
Benjamin Peterson29060642009-01-31 22:14:21 +00006050 /* Map 32-bit characters to '\Uxxxxxxxx' */
6051 if (ch >= 0x10000) {
Victor Stinner8faf8212011-12-08 22:14:11 +01006052 assert(ch <= MAX_UNICODE);
Martin v. Löwis9a3a9f72003-05-18 12:31:09 +00006053 *p++ = '\\';
6054 *p++ = 'U';
Victor Stinnerf5cff562011-10-14 02:13:11 +02006055 *p++ = Py_hexdigits[(ch >> 28) & 0xf];
6056 *p++ = Py_hexdigits[(ch >> 24) & 0xf];
6057 *p++ = Py_hexdigits[(ch >> 20) & 0xf];
6058 *p++ = Py_hexdigits[(ch >> 16) & 0xf];
6059 *p++ = Py_hexdigits[(ch >> 12) & 0xf];
6060 *p++ = Py_hexdigits[(ch >> 8) & 0xf];
6061 *p++ = Py_hexdigits[(ch >> 4) & 0xf];
6062 *p++ = Py_hexdigits[ch & 15];
Tim Petersced69f82003-09-16 20:30:58 +00006063 }
Benjamin Peterson29060642009-01-31 22:14:21 +00006064 /* Map 16-bit characters to '\uxxxx' */
Martin v. Löwis1db7c132011-11-10 18:24:32 +01006065 else if (ch >= 256) {
Guido van Rossumd57fd912000-03-10 22:53:23 +00006066 *p++ = '\\';
6067 *p++ = 'u';
Victor Stinnerf5cff562011-10-14 02:13:11 +02006068 *p++ = Py_hexdigits[(ch >> 12) & 0xf];
6069 *p++ = Py_hexdigits[(ch >> 8) & 0xf];
6070 *p++ = Py_hexdigits[(ch >> 4) & 0xf];
6071 *p++ = Py_hexdigits[ch & 15];
Guido van Rossumd57fd912000-03-10 22:53:23 +00006072 }
Benjamin Peterson29060642009-01-31 22:14:21 +00006073 /* Copy everything else as-is */
6074 else
Guido van Rossumd57fd912000-03-10 22:53:23 +00006075 *p++ = (char) ch;
6076 }
Guido van Rossum98297ee2007-11-06 21:34:58 +00006077
Martin v. Löwis1db7c132011-11-10 18:24:32 +01006078 assert(p > q);
6079 if (_PyBytes_Resize(&repr, p - q) < 0)
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00006080 return NULL;
6081 return repr;
Guido van Rossumd57fd912000-03-10 22:53:23 +00006082}
6083
Alexander Belopolsky40018472011-02-26 01:02:56 +00006084PyObject *
Martin v. Löwis1db7c132011-11-10 18:24:32 +01006085PyUnicode_EncodeRawUnicodeEscape(const Py_UNICODE *s,
6086 Py_ssize_t size)
Guido van Rossumd57fd912000-03-10 22:53:23 +00006087{
Martin v. Löwis1db7c132011-11-10 18:24:32 +01006088 PyObject *result;
6089 PyObject *tmp = PyUnicode_FromUnicode(s, size);
6090 if (tmp == NULL)
Walter Dörwald711005d2007-05-12 12:03:26 +00006091 return NULL;
Martin v. Löwis1db7c132011-11-10 18:24:32 +01006092 result = PyUnicode_AsRawUnicodeEscapeString(tmp);
6093 Py_DECREF(tmp);
6094 return result;
Guido van Rossumd57fd912000-03-10 22:53:23 +00006095}
6096
Walter Dörwalda47d1c02005-08-30 10:23:14 +00006097/* --- Unicode Internal Codec ------------------------------------------- */
6098
Alexander Belopolsky40018472011-02-26 01:02:56 +00006099PyObject *
6100_PyUnicode_DecodeUnicodeInternal(const char *s,
Ezio Melotti2aa2b3b2011-09-29 00:58:57 +03006101 Py_ssize_t size,
6102 const char *errors)
Walter Dörwalda47d1c02005-08-30 10:23:14 +00006103{
6104 const char *starts = s;
Martin v. Löwis18e16552006-02-15 17:27:45 +00006105 Py_ssize_t startinpos;
6106 Py_ssize_t endinpos;
6107 Py_ssize_t outpos;
Victor Stinner7931d9a2011-11-04 00:22:48 +01006108 PyObject *v;
Walter Dörwalda47d1c02005-08-30 10:23:14 +00006109 const char *end;
6110 const char *reason;
6111 PyObject *errorHandler = NULL;
6112 PyObject *exc = NULL;
6113
Victor Stinner9f4b1e92011-11-10 20:56:30 +01006114 if (PyErr_WarnEx(PyExc_DeprecationWarning,
Ezio Melotti11060a42011-11-16 09:39:10 +02006115 "unicode_internal codec has been deprecated",
Victor Stinner9f4b1e92011-11-10 20:56:30 +01006116 1))
6117 return NULL;
6118
Thomas Wouters89f507f2006-12-13 04:49:30 +00006119 /* XXX overflow detection missing */
Martin v. Löwise9b11c12011-11-08 17:35:34 +01006120 v = PyUnicode_New((size+Py_UNICODE_SIZE-1)/ Py_UNICODE_SIZE, 127);
Walter Dörwalda47d1c02005-08-30 10:23:14 +00006121 if (v == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00006122 goto onError;
Martin v. Löwise9b11c12011-11-08 17:35:34 +01006123 if (PyUnicode_GET_LENGTH(v) == 0)
Victor Stinner7931d9a2011-11-04 00:22:48 +01006124 return v;
Martin v. Löwise9b11c12011-11-08 17:35:34 +01006125 outpos = 0;
Walter Dörwalda47d1c02005-08-30 10:23:14 +00006126 end = s + size;
6127
6128 while (s < end) {
Antoine Pitrou0290c7a2011-11-11 13:29:12 +01006129 Py_UNICODE uch;
Antoine Pitrou44c6aff2011-11-11 02:59:42 +01006130 Py_UCS4 ch;
6131 /* We copy the raw representation one byte at a time because the
6132 pointer may be unaligned (see test_codeccallbacks). */
Antoine Pitrou0290c7a2011-11-11 13:29:12 +01006133 ((char *) &uch)[0] = s[0];
6134 ((char *) &uch)[1] = s[1];
Antoine Pitrou44c6aff2011-11-11 02:59:42 +01006135#ifdef Py_UNICODE_WIDE
Antoine Pitrou0290c7a2011-11-11 13:29:12 +01006136 ((char *) &uch)[2] = s[2];
6137 ((char *) &uch)[3] = s[3];
Antoine Pitrou44c6aff2011-11-11 02:59:42 +01006138#endif
Antoine Pitrou0290c7a2011-11-11 13:29:12 +01006139 ch = uch;
6140
Walter Dörwalda47d1c02005-08-30 10:23:14 +00006141 /* We have to sanity check the raw data, otherwise doom looms for
6142 some malformed UCS-4 data. */
6143 if (
Benjamin Peterson29060642009-01-31 22:14:21 +00006144#ifdef Py_UNICODE_WIDE
Martin v. Löwise9b11c12011-11-08 17:35:34 +01006145 ch > 0x10ffff ||
Benjamin Peterson29060642009-01-31 22:14:21 +00006146#endif
Walter Dörwalda47d1c02005-08-30 10:23:14 +00006147 end-s < Py_UNICODE_SIZE
6148 )
Benjamin Peterson29060642009-01-31 22:14:21 +00006149 {
Walter Dörwalda47d1c02005-08-30 10:23:14 +00006150 startinpos = s - starts;
6151 if (end-s < Py_UNICODE_SIZE) {
6152 endinpos = end-starts;
6153 reason = "truncated input";
6154 }
6155 else {
6156 endinpos = s - starts + Py_UNICODE_SIZE;
6157 reason = "illegal code point (> 0x10FFFF)";
6158 }
Walter Dörwalda47d1c02005-08-30 10:23:14 +00006159 if (unicode_decode_call_errorhandler(
6160 errors, &errorHandler,
6161 "unicode_internal", reason,
Walter Dörwalde78178e2007-07-30 13:31:40 +00006162 &starts, &end, &startinpos, &endinpos, &exc, &s,
Victor Stinner9f4b1e92011-11-10 20:56:30 +01006163 &v, &outpos))
Walter Dörwalda47d1c02005-08-30 10:23:14 +00006164 goto onError;
Victor Stinner9f4b1e92011-11-10 20:56:30 +01006165 continue;
6166 }
6167
6168 s += Py_UNICODE_SIZE;
6169#ifndef Py_UNICODE_WIDE
Victor Stinner551ac952011-11-29 22:58:13 +01006170 if (Py_UNICODE_IS_HIGH_SURROGATE(ch) && s < end)
Victor Stinner9f4b1e92011-11-10 20:56:30 +01006171 {
Antoine Pitrou0290c7a2011-11-11 13:29:12 +01006172 Py_UNICODE uch2;
6173 ((char *) &uch2)[0] = s[0];
6174 ((char *) &uch2)[1] = s[1];
Victor Stinner551ac952011-11-29 22:58:13 +01006175 if (Py_UNICODE_IS_LOW_SURROGATE(uch2))
Victor Stinner9f4b1e92011-11-10 20:56:30 +01006176 {
Victor Stinner551ac952011-11-29 22:58:13 +01006177 ch = Py_UNICODE_JOIN_SURROGATES(uch, uch2);
Victor Stinner9f4b1e92011-11-10 20:56:30 +01006178 s += Py_UNICODE_SIZE;
Walter Dörwalda47d1c02005-08-30 10:23:14 +00006179 }
6180 }
Victor Stinner9f4b1e92011-11-10 20:56:30 +01006181#endif
6182
6183 if (unicode_putchar(&v, &outpos, ch) < 0)
6184 goto onError;
Walter Dörwalda47d1c02005-08-30 10:23:14 +00006185 }
6186
Victor Stinner16e6a802011-12-12 13:24:15 +01006187 if (unicode_resize(&v, outpos) < 0)
Walter Dörwalda47d1c02005-08-30 10:23:14 +00006188 goto onError;
6189 Py_XDECREF(errorHandler);
6190 Py_XDECREF(exc);
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01006191 return unicode_result(v);
Walter Dörwalda47d1c02005-08-30 10:23:14 +00006192
Benjamin Peterson29060642009-01-31 22:14:21 +00006193 onError:
Walter Dörwalda47d1c02005-08-30 10:23:14 +00006194 Py_XDECREF(v);
6195 Py_XDECREF(errorHandler);
6196 Py_XDECREF(exc);
6197 return NULL;
6198}
6199
Guido van Rossumd57fd912000-03-10 22:53:23 +00006200/* --- Latin-1 Codec ------------------------------------------------------ */
6201
Alexander Belopolsky40018472011-02-26 01:02:56 +00006202PyObject *
6203PyUnicode_DecodeLatin1(const char *s,
Ezio Melotti2aa2b3b2011-09-29 00:58:57 +03006204 Py_ssize_t size,
6205 const char *errors)
Guido van Rossumd57fd912000-03-10 22:53:23 +00006206{
Guido van Rossumd57fd912000-03-10 22:53:23 +00006207 /* Latin-1 is equivalent to the first 256 ordinals in Unicode. */
Victor Stinnere57b1c02011-09-28 22:20:48 +02006208 return _PyUnicode_FromUCS1((unsigned char*)s, size);
Guido van Rossumd57fd912000-03-10 22:53:23 +00006209}
6210
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006211/* create or adjust a UnicodeEncodeError */
Alexander Belopolsky40018472011-02-26 01:02:56 +00006212static void
6213make_encode_exception(PyObject **exceptionObject,
Ezio Melotti2aa2b3b2011-09-29 00:58:57 +03006214 const char *encoding,
Martin v. Löwis9e816682011-11-02 12:45:42 +01006215 PyObject *unicode,
6216 Py_ssize_t startpos, Py_ssize_t endpos,
6217 const char *reason)
6218{
6219 if (*exceptionObject == NULL) {
6220 *exceptionObject = PyObject_CallFunction(
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006221 PyExc_UnicodeEncodeError, "sOnns",
Martin v. Löwis9e816682011-11-02 12:45:42 +01006222 encoding, unicode, startpos, endpos, reason);
6223 }
6224 else {
6225 if (PyUnicodeEncodeError_SetStart(*exceptionObject, startpos))
6226 goto onError;
6227 if (PyUnicodeEncodeError_SetEnd(*exceptionObject, endpos))
6228 goto onError;
6229 if (PyUnicodeEncodeError_SetReason(*exceptionObject, reason))
6230 goto onError;
6231 return;
6232 onError:
6233 Py_DECREF(*exceptionObject);
6234 *exceptionObject = NULL;
6235 }
6236}
6237
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006238/* raises a UnicodeEncodeError */
Alexander Belopolsky40018472011-02-26 01:02:56 +00006239static void
6240raise_encode_exception(PyObject **exceptionObject,
Ezio Melotti2aa2b3b2011-09-29 00:58:57 +03006241 const char *encoding,
Martin v. Löwis9e816682011-11-02 12:45:42 +01006242 PyObject *unicode,
6243 Py_ssize_t startpos, Py_ssize_t endpos,
6244 const char *reason)
6245{
Martin v. Löwis12be46c2011-11-04 19:04:15 +01006246 make_encode_exception(exceptionObject,
Martin v. Löwis9e816682011-11-02 12:45:42 +01006247 encoding, unicode, startpos, endpos, reason);
6248 if (*exceptionObject != NULL)
6249 PyCodec_StrictErrors(*exceptionObject);
6250}
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006251
6252/* error handling callback helper:
6253 build arguments, call the callback and check the arguments,
6254 put the result into newpos and return the replacement string, which
6255 has to be freed by the caller */
Alexander Belopolsky40018472011-02-26 01:02:56 +00006256static PyObject *
6257unicode_encode_call_errorhandler(const char *errors,
Ezio Melotti2aa2b3b2011-09-29 00:58:57 +03006258 PyObject **errorHandler,
6259 const char *encoding, const char *reason,
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006260 PyObject *unicode, PyObject **exceptionObject,
Ezio Melotti2aa2b3b2011-09-29 00:58:57 +03006261 Py_ssize_t startpos, Py_ssize_t endpos,
6262 Py_ssize_t *newpos)
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006263{
Martin v. Löwisdb12d452009-05-02 18:52:14 +00006264 static char *argparse = "On;encoding error handler must return (str/bytes, int) tuple";
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006265 Py_ssize_t len;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006266 PyObject *restuple;
6267 PyObject *resunicode;
6268
6269 if (*errorHandler == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +00006270 *errorHandler = PyCodec_LookupError(errors);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006271 if (*errorHandler == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00006272 return NULL;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006273 }
6274
Benjamin Petersonbac79492012-01-14 13:34:47 -05006275 if (PyUnicode_READY(unicode) == -1)
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006276 return NULL;
6277 len = PyUnicode_GET_LENGTH(unicode);
6278
Martin v. Löwis12be46c2011-11-04 19:04:15 +01006279 make_encode_exception(exceptionObject,
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006280 encoding, unicode, startpos, endpos, reason);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006281 if (*exceptionObject == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00006282 return NULL;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006283
6284 restuple = PyObject_CallFunctionObjArgs(
Benjamin Peterson29060642009-01-31 22:14:21 +00006285 *errorHandler, *exceptionObject, NULL);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006286 if (restuple == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00006287 return NULL;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006288 if (!PyTuple_Check(restuple)) {
Martin v. Löwisdb12d452009-05-02 18:52:14 +00006289 PyErr_SetString(PyExc_TypeError, &argparse[3]);
Benjamin Peterson29060642009-01-31 22:14:21 +00006290 Py_DECREF(restuple);
6291 return NULL;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006292 }
Martin v. Löwisdb12d452009-05-02 18:52:14 +00006293 if (!PyArg_ParseTuple(restuple, argparse,
Benjamin Peterson29060642009-01-31 22:14:21 +00006294 &resunicode, newpos)) {
6295 Py_DECREF(restuple);
6296 return NULL;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006297 }
Martin v. Löwisdb12d452009-05-02 18:52:14 +00006298 if (!PyUnicode_Check(resunicode) && !PyBytes_Check(resunicode)) {
6299 PyErr_SetString(PyExc_TypeError, &argparse[3]);
6300 Py_DECREF(restuple);
6301 return NULL;
6302 }
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006303 if (*newpos<0)
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006304 *newpos = len + *newpos;
6305 if (*newpos<0 || *newpos>len) {
Benjamin Peterson29060642009-01-31 22:14:21 +00006306 PyErr_Format(PyExc_IndexError, "position %zd from error handler out of bounds", *newpos);
6307 Py_DECREF(restuple);
6308 return NULL;
Walter Dörwald2e0b18a2003-01-31 17:19:08 +00006309 }
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006310 Py_INCREF(resunicode);
6311 Py_DECREF(restuple);
6312 return resunicode;
6313}
6314
Alexander Belopolsky40018472011-02-26 01:02:56 +00006315static PyObject *
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006316unicode_encode_ucs1(PyObject *unicode,
Ezio Melotti2aa2b3b2011-09-29 00:58:57 +03006317 const char *errors,
Victor Stinnerfcd96532011-11-04 00:28:50 +01006318 unsigned int limit)
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006319{
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006320 /* input state */
6321 Py_ssize_t pos=0, size;
6322 int kind;
6323 void *data;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006324 /* output object */
6325 PyObject *res;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006326 /* pointer into the output */
6327 char *str;
6328 /* current output position */
Martin v. Löwis18e16552006-02-15 17:27:45 +00006329 Py_ssize_t ressize;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006330 const char *encoding = (limit == 256) ? "latin-1" : "ascii";
6331 const char *reason = (limit == 256) ? "ordinal not in range(256)" : "ordinal not in range(128)";
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006332 PyObject *errorHandler = NULL;
6333 PyObject *exc = NULL;
6334 /* the following variable is used for caching string comparisons
6335 * -1=not initialized, 0=unknown, 1=strict, 2=replace, 3=ignore, 4=xmlcharrefreplace */
6336 int known_errorHandler = -1;
6337
Benjamin Petersonbac79492012-01-14 13:34:47 -05006338 if (PyUnicode_READY(unicode) == -1)
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006339 return NULL;
6340 size = PyUnicode_GET_LENGTH(unicode);
6341 kind = PyUnicode_KIND(unicode);
6342 data = PyUnicode_DATA(unicode);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006343 /* allocate enough for a simple encoding without
6344 replacements, if we need more, we'll resize */
Guido van Rossum98297ee2007-11-06 21:34:58 +00006345 if (size == 0)
Christian Heimes72b710a2008-05-26 13:28:38 +00006346 return PyBytes_FromStringAndSize(NULL, 0);
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00006347 res = PyBytes_FromStringAndSize(NULL, size);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006348 if (res == NULL)
Guido van Rossum98297ee2007-11-06 21:34:58 +00006349 return NULL;
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00006350 str = PyBytes_AS_STRING(res);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006351 ressize = size;
6352
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006353 while (pos < size) {
6354 Py_UCS4 c = PyUnicode_READ(kind, data, pos);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006355
Benjamin Peterson29060642009-01-31 22:14:21 +00006356 /* can we encode this? */
6357 if (c<limit) {
6358 /* no overflow check, because we know that the space is enough */
6359 *str++ = (char)c;
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006360 ++pos;
Benjamin Peterson14339b62009-01-31 16:36:08 +00006361 }
Benjamin Peterson29060642009-01-31 22:14:21 +00006362 else {
Benjamin Peterson29060642009-01-31 22:14:21 +00006363 Py_ssize_t requiredsize;
6364 PyObject *repunicode;
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006365 Py_ssize_t repsize, newpos, respos, i;
Benjamin Peterson29060642009-01-31 22:14:21 +00006366 /* startpos for collecting unencodable chars */
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006367 Py_ssize_t collstart = pos;
6368 Py_ssize_t collend = pos;
Benjamin Peterson29060642009-01-31 22:14:21 +00006369 /* find all unecodable characters */
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006370 while ((collend < size) && (PyUnicode_READ(kind, data, collend)>=limit))
Benjamin Peterson29060642009-01-31 22:14:21 +00006371 ++collend;
6372 /* cache callback name lookup (if not done yet, i.e. it's the first error) */
6373 if (known_errorHandler==-1) {
6374 if ((errors==NULL) || (!strcmp(errors, "strict")))
6375 known_errorHandler = 1;
6376 else if (!strcmp(errors, "replace"))
6377 known_errorHandler = 2;
6378 else if (!strcmp(errors, "ignore"))
6379 known_errorHandler = 3;
6380 else if (!strcmp(errors, "xmlcharrefreplace"))
6381 known_errorHandler = 4;
6382 else
6383 known_errorHandler = 0;
6384 }
6385 switch (known_errorHandler) {
6386 case 1: /* strict */
Martin v. Löwis12be46c2011-11-04 19:04:15 +01006387 raise_encode_exception(&exc, encoding, unicode, collstart, collend, reason);
Benjamin Peterson29060642009-01-31 22:14:21 +00006388 goto onError;
6389 case 2: /* replace */
6390 while (collstart++<collend)
6391 *str++ = '?'; /* fall through */
6392 case 3: /* ignore */
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006393 pos = collend;
Benjamin Peterson29060642009-01-31 22:14:21 +00006394 break;
6395 case 4: /* xmlcharrefreplace */
6396 respos = str - PyBytes_AS_STRING(res);
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006397 /* determine replacement size */
6398 for (i = collstart, repsize = 0; i < collend; ++i) {
6399 Py_UCS4 ch = PyUnicode_READ(kind, data, i);
6400 if (ch < 10)
Benjamin Peterson29060642009-01-31 22:14:21 +00006401 repsize += 2+1+1;
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006402 else if (ch < 100)
Benjamin Peterson29060642009-01-31 22:14:21 +00006403 repsize += 2+2+1;
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006404 else if (ch < 1000)
Benjamin Peterson29060642009-01-31 22:14:21 +00006405 repsize += 2+3+1;
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006406 else if (ch < 10000)
Benjamin Peterson29060642009-01-31 22:14:21 +00006407 repsize += 2+4+1;
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006408 else if (ch < 100000)
Benjamin Peterson29060642009-01-31 22:14:21 +00006409 repsize += 2+5+1;
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006410 else if (ch < 1000000)
Benjamin Peterson29060642009-01-31 22:14:21 +00006411 repsize += 2+6+1;
Victor Stinner0d3721d2011-11-22 03:27:53 +01006412 else {
Victor Stinner8faf8212011-12-08 22:14:11 +01006413 assert(ch <= MAX_UNICODE);
Benjamin Peterson29060642009-01-31 22:14:21 +00006414 repsize += 2+7+1;
Victor Stinner0d3721d2011-11-22 03:27:53 +01006415 }
Benjamin Peterson29060642009-01-31 22:14:21 +00006416 }
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006417 requiredsize = respos+repsize+(size-collend);
Benjamin Peterson29060642009-01-31 22:14:21 +00006418 if (requiredsize > ressize) {
6419 if (requiredsize<2*ressize)
6420 requiredsize = 2*ressize;
6421 if (_PyBytes_Resize(&res, requiredsize))
6422 goto onError;
6423 str = PyBytes_AS_STRING(res) + respos;
6424 ressize = requiredsize;
6425 }
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006426 /* generate replacement */
6427 for (i = collstart; i < collend; ++i) {
6428 str += sprintf(str, "&#%d;", PyUnicode_READ(kind, data, i));
Benjamin Peterson29060642009-01-31 22:14:21 +00006429 }
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006430 pos = collend;
Benjamin Peterson29060642009-01-31 22:14:21 +00006431 break;
6432 default:
6433 repunicode = unicode_encode_call_errorhandler(errors, &errorHandler,
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006434 encoding, reason, unicode, &exc,
6435 collstart, collend, &newpos);
6436 if (repunicode == NULL || (PyUnicode_Check(repunicode) &&
Benjamin Petersonbac79492012-01-14 13:34:47 -05006437 PyUnicode_READY(repunicode) == -1))
Benjamin Peterson29060642009-01-31 22:14:21 +00006438 goto onError;
Martin v. Löwis011e8422009-05-05 04:43:17 +00006439 if (PyBytes_Check(repunicode)) {
6440 /* Directly copy bytes result to output. */
6441 repsize = PyBytes_Size(repunicode);
6442 if (repsize > 1) {
6443 /* Make room for all additional bytes. */
Amaury Forgeot d'Arc84ec8d92009-06-29 22:36:49 +00006444 respos = str - PyBytes_AS_STRING(res);
Martin v. Löwis011e8422009-05-05 04:43:17 +00006445 if (_PyBytes_Resize(&res, ressize+repsize-1)) {
6446 Py_DECREF(repunicode);
6447 goto onError;
6448 }
Amaury Forgeot d'Arc84ec8d92009-06-29 22:36:49 +00006449 str = PyBytes_AS_STRING(res) + respos;
Martin v. Löwis011e8422009-05-05 04:43:17 +00006450 ressize += repsize-1;
6451 }
6452 memcpy(str, PyBytes_AsString(repunicode), repsize);
6453 str += repsize;
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006454 pos = newpos;
Martin v. Löwisdb12d452009-05-02 18:52:14 +00006455 Py_DECREF(repunicode);
Martin v. Löwis011e8422009-05-05 04:43:17 +00006456 break;
Martin v. Löwisdb12d452009-05-02 18:52:14 +00006457 }
Benjamin Peterson29060642009-01-31 22:14:21 +00006458 /* need more space? (at least enough for what we
6459 have+the replacement+the rest of the string, so
6460 we won't have to check space for encodable characters) */
6461 respos = str - PyBytes_AS_STRING(res);
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006462 repsize = PyUnicode_GET_LENGTH(repunicode);
6463 requiredsize = respos+repsize+(size-collend);
Benjamin Peterson29060642009-01-31 22:14:21 +00006464 if (requiredsize > ressize) {
6465 if (requiredsize<2*ressize)
6466 requiredsize = 2*ressize;
6467 if (_PyBytes_Resize(&res, requiredsize)) {
6468 Py_DECREF(repunicode);
6469 goto onError;
6470 }
6471 str = PyBytes_AS_STRING(res) + respos;
6472 ressize = requiredsize;
6473 }
6474 /* check if there is anything unencodable in the replacement
6475 and copy it to the output */
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006476 for (i = 0; repsize-->0; ++i, ++str) {
6477 c = PyUnicode_READ_CHAR(repunicode, i);
Benjamin Peterson29060642009-01-31 22:14:21 +00006478 if (c >= limit) {
Martin v. Löwis12be46c2011-11-04 19:04:15 +01006479 raise_encode_exception(&exc, encoding, unicode,
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006480 pos, pos+1, reason);
Benjamin Peterson29060642009-01-31 22:14:21 +00006481 Py_DECREF(repunicode);
6482 goto onError;
6483 }
6484 *str = (char)c;
6485 }
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006486 pos = newpos;
Benjamin Peterson14339b62009-01-31 16:36:08 +00006487 Py_DECREF(repunicode);
Benjamin Peterson14339b62009-01-31 16:36:08 +00006488 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00006489 }
6490 }
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00006491 /* Resize if we allocated to much */
6492 size = str - PyBytes_AS_STRING(res);
6493 if (size < ressize) { /* If this falls res will be NULL */
Alexandre Vassalottibad1b922008-12-27 09:49:09 +00006494 assert(size >= 0);
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00006495 if (_PyBytes_Resize(&res, size) < 0)
6496 goto onError;
6497 }
6498
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006499 Py_XDECREF(errorHandler);
6500 Py_XDECREF(exc);
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00006501 return res;
6502
6503 onError:
6504 Py_XDECREF(res);
6505 Py_XDECREF(errorHandler);
6506 Py_XDECREF(exc);
6507 return NULL;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006508}
6509
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006510/* Deprecated */
Alexander Belopolsky40018472011-02-26 01:02:56 +00006511PyObject *
6512PyUnicode_EncodeLatin1(const Py_UNICODE *p,
Ezio Melotti2aa2b3b2011-09-29 00:58:57 +03006513 Py_ssize_t size,
6514 const char *errors)
Guido van Rossumd57fd912000-03-10 22:53:23 +00006515{
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006516 PyObject *result;
6517 PyObject *unicode = PyUnicode_FromUnicode(p, size);
6518 if (unicode == NULL)
6519 return NULL;
6520 result = unicode_encode_ucs1(unicode, errors, 256);
6521 Py_DECREF(unicode);
6522 return result;
Guido van Rossumd57fd912000-03-10 22:53:23 +00006523}
6524
Alexander Belopolsky40018472011-02-26 01:02:56 +00006525PyObject *
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02006526_PyUnicode_AsLatin1String(PyObject *unicode, const char *errors)
Guido van Rossumd57fd912000-03-10 22:53:23 +00006527{
6528 if (!PyUnicode_Check(unicode)) {
Benjamin Peterson29060642009-01-31 22:14:21 +00006529 PyErr_BadArgument();
6530 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00006531 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02006532 if (PyUnicode_READY(unicode) == -1)
6533 return NULL;
6534 /* Fast path: if it is a one-byte string, construct
6535 bytes object directly. */
6536 if (PyUnicode_KIND(unicode) == PyUnicode_1BYTE_KIND)
6537 return PyBytes_FromStringAndSize(PyUnicode_DATA(unicode),
6538 PyUnicode_GET_LENGTH(unicode));
6539 /* Non-Latin-1 characters present. Defer to above function to
6540 raise the exception. */
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006541 return unicode_encode_ucs1(unicode, errors, 256);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02006542}
6543
6544PyObject*
6545PyUnicode_AsLatin1String(PyObject *unicode)
6546{
6547 return _PyUnicode_AsLatin1String(unicode, NULL);
Guido van Rossumd57fd912000-03-10 22:53:23 +00006548}
6549
6550/* --- 7-bit ASCII Codec -------------------------------------------------- */
6551
Alexander Belopolsky40018472011-02-26 01:02:56 +00006552PyObject *
6553PyUnicode_DecodeASCII(const char *s,
6554 Py_ssize_t size,
6555 const char *errors)
Guido van Rossumd57fd912000-03-10 22:53:23 +00006556{
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006557 const char *starts = s;
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02006558 PyObject *unicode;
Martin v. Löwise9b11c12011-11-08 17:35:34 +01006559 int kind;
6560 void *data;
Martin v. Löwis18e16552006-02-15 17:27:45 +00006561 Py_ssize_t startinpos;
6562 Py_ssize_t endinpos;
6563 Py_ssize_t outpos;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006564 const char *e;
6565 PyObject *errorHandler = NULL;
6566 PyObject *exc = NULL;
Tim Petersced69f82003-09-16 20:30:58 +00006567
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01006568 if (size == 0) {
6569 Py_INCREF(unicode_empty);
6570 return unicode_empty;
6571 }
6572
Guido van Rossumd57fd912000-03-10 22:53:23 +00006573 /* ASCII is equivalent to the first 128 ordinals in Unicode. */
Victor Stinner702c7342011-10-05 13:50:52 +02006574 if (size == 1 && (unsigned char)s[0] < 128)
6575 return get_latin1_char((unsigned char)s[0]);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02006576
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02006577 unicode = PyUnicode_New(size, 127);
6578 if (unicode == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00006579 goto onError;
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02006580
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006581 e = s + size;
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02006582 data = PyUnicode_1BYTE_DATA(unicode);
6583 outpos = ascii_decode(s, e, (Py_UCS1 *)data);
6584 if (outpos == size)
6585 return unicode;
6586
6587 s += outpos;
6588 kind = PyUnicode_1BYTE_KIND;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006589 while (s < e) {
Benjamin Peterson29060642009-01-31 22:14:21 +00006590 register unsigned char c = (unsigned char)*s;
6591 if (c < 128) {
Martin v. Löwise9b11c12011-11-08 17:35:34 +01006592 PyUnicode_WRITE(kind, data, outpos++, c);
Benjamin Peterson29060642009-01-31 22:14:21 +00006593 ++s;
6594 }
6595 else {
6596 startinpos = s-starts;
6597 endinpos = startinpos + 1;
Benjamin Peterson29060642009-01-31 22:14:21 +00006598 if (unicode_decode_call_errorhandler(
6599 errors, &errorHandler,
6600 "ascii", "ordinal not in range(128)",
6601 &starts, &e, &startinpos, &endinpos, &exc, &s,
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02006602 &unicode, &outpos))
Benjamin Peterson29060642009-01-31 22:14:21 +00006603 goto onError;
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02006604 kind = PyUnicode_KIND(unicode);
6605 data = PyUnicode_DATA(unicode);
Benjamin Peterson29060642009-01-31 22:14:21 +00006606 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00006607 }
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02006608 if (unicode_resize(&unicode, outpos) < 0)
Martin v. Löwise9b11c12011-11-08 17:35:34 +01006609 goto onError;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006610 Py_XDECREF(errorHandler);
6611 Py_XDECREF(exc);
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02006612 assert(_PyUnicode_CheckConsistency(unicode, 1));
6613 return unicode;
Tim Petersced69f82003-09-16 20:30:58 +00006614
Benjamin Peterson29060642009-01-31 22:14:21 +00006615 onError:
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02006616 Py_XDECREF(unicode);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006617 Py_XDECREF(errorHandler);
6618 Py_XDECREF(exc);
Guido van Rossumd57fd912000-03-10 22:53:23 +00006619 return NULL;
6620}
6621
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006622/* Deprecated */
Alexander Belopolsky40018472011-02-26 01:02:56 +00006623PyObject *
6624PyUnicode_EncodeASCII(const Py_UNICODE *p,
6625 Py_ssize_t size,
6626 const char *errors)
Guido van Rossumd57fd912000-03-10 22:53:23 +00006627{
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006628 PyObject *result;
6629 PyObject *unicode = PyUnicode_FromUnicode(p, size);
6630 if (unicode == NULL)
6631 return NULL;
6632 result = unicode_encode_ucs1(unicode, errors, 128);
6633 Py_DECREF(unicode);
6634 return result;
Guido van Rossumd57fd912000-03-10 22:53:23 +00006635}
6636
Alexander Belopolsky40018472011-02-26 01:02:56 +00006637PyObject *
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02006638_PyUnicode_AsASCIIString(PyObject *unicode, const char *errors)
Guido van Rossumd57fd912000-03-10 22:53:23 +00006639{
6640 if (!PyUnicode_Check(unicode)) {
Benjamin Peterson29060642009-01-31 22:14:21 +00006641 PyErr_BadArgument();
6642 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00006643 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02006644 if (PyUnicode_READY(unicode) == -1)
6645 return NULL;
6646 /* Fast path: if it is an ASCII-only string, construct bytes object
6647 directly. Else defer to above function to raise the exception. */
6648 if (PyUnicode_MAX_CHAR_VALUE(unicode) < 128)
6649 return PyBytes_FromStringAndSize(PyUnicode_DATA(unicode),
6650 PyUnicode_GET_LENGTH(unicode));
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006651 return unicode_encode_ucs1(unicode, errors, 128);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02006652}
6653
6654PyObject *
6655PyUnicode_AsASCIIString(PyObject *unicode)
6656{
6657 return _PyUnicode_AsASCIIString(unicode, NULL);
Guido van Rossumd57fd912000-03-10 22:53:23 +00006658}
6659
Victor Stinner99b95382011-07-04 14:23:54 +02006660#ifdef HAVE_MBCS
Guido van Rossum2ea3e142000-03-31 17:24:09 +00006661
Guido van Rossumb7a40ba2000-03-28 02:01:52 +00006662/* --- MBCS codecs for Windows -------------------------------------------- */
Guido van Rossum2ea3e142000-03-31 17:24:09 +00006663
Hirokazu Yamamoto35302462009-03-21 13:23:27 +00006664#if SIZEOF_INT < SIZEOF_SIZE_T
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006665#define NEED_RETRY
6666#endif
6667
Victor Stinner3a50e702011-10-18 21:21:00 +02006668#ifndef WC_ERR_INVALID_CHARS
6669# define WC_ERR_INVALID_CHARS 0x0080
6670#endif
6671
6672static char*
6673code_page_name(UINT code_page, PyObject **obj)
6674{
6675 *obj = NULL;
6676 if (code_page == CP_ACP)
6677 return "mbcs";
6678 if (code_page == CP_UTF7)
6679 return "CP_UTF7";
6680 if (code_page == CP_UTF8)
6681 return "CP_UTF8";
6682
6683 *obj = PyBytes_FromFormat("cp%u", code_page);
6684 if (*obj == NULL)
6685 return NULL;
6686 return PyBytes_AS_STRING(*obj);
6687}
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006688
Alexander Belopolsky40018472011-02-26 01:02:56 +00006689static int
Victor Stinner3a50e702011-10-18 21:21:00 +02006690is_dbcs_lead_byte(UINT code_page, const char *s, int offset)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006691{
6692 const char *curr = s + offset;
Victor Stinner3a50e702011-10-18 21:21:00 +02006693 const char *prev;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006694
Victor Stinner3a50e702011-10-18 21:21:00 +02006695 if (!IsDBCSLeadByteEx(code_page, *curr))
6696 return 0;
6697
6698 prev = CharPrevExA(code_page, s, curr, 0);
6699 if (prev == curr)
6700 return 1;
6701 /* FIXME: This code is limited to "true" double-byte encodings,
6702 as it assumes an incomplete character consists of a single
6703 byte. */
6704 if (curr - prev == 2)
6705 return 1;
6706 if (!IsDBCSLeadByteEx(code_page, *prev))
6707 return 1;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006708 return 0;
6709}
6710
Victor Stinner3a50e702011-10-18 21:21:00 +02006711static DWORD
6712decode_code_page_flags(UINT code_page)
6713{
6714 if (code_page == CP_UTF7) {
6715 /* The CP_UTF7 decoder only supports flags=0 */
6716 return 0;
6717 }
6718 else
6719 return MB_ERR_INVALID_CHARS;
6720}
6721
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006722/*
Victor Stinner3a50e702011-10-18 21:21:00 +02006723 * Decode a byte string from a Windows code page into unicode object in strict
6724 * mode.
6725 *
6726 * Returns consumed size if succeed, returns -2 on decode error, or raise a
6727 * WindowsError and returns -1 on other error.
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006728 */
Alexander Belopolsky40018472011-02-26 01:02:56 +00006729static int
Victor Stinner3a50e702011-10-18 21:21:00 +02006730decode_code_page_strict(UINT code_page,
Victor Stinner76a31a62011-11-04 00:05:13 +01006731 PyObject **v,
Victor Stinner3a50e702011-10-18 21:21:00 +02006732 const char *in,
6733 int insize)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006734{
Victor Stinner3a50e702011-10-18 21:21:00 +02006735 const DWORD flags = decode_code_page_flags(code_page);
Victor Stinner24729f32011-11-10 20:31:37 +01006736 wchar_t *out;
Victor Stinner3a50e702011-10-18 21:21:00 +02006737 DWORD outsize;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006738
6739 /* First get the size of the result */
Victor Stinner3a50e702011-10-18 21:21:00 +02006740 assert(insize > 0);
6741 outsize = MultiByteToWideChar(code_page, flags, in, insize, NULL, 0);
6742 if (outsize <= 0)
6743 goto error;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006744
6745 if (*v == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +00006746 /* Create unicode object */
Victor Stinnerab595942011-12-17 04:59:06 +01006747 /* FIXME: don't use _PyUnicode_New(), but allocate a wchar_t* buffer */
Victor Stinner76a31a62011-11-04 00:05:13 +01006748 *v = (PyObject*)_PyUnicode_New(outsize);
Benjamin Peterson29060642009-01-31 22:14:21 +00006749 if (*v == NULL)
6750 return -1;
Victor Stinner3a50e702011-10-18 21:21:00 +02006751 out = PyUnicode_AS_UNICODE(*v);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006752 }
6753 else {
Benjamin Peterson29060642009-01-31 22:14:21 +00006754 /* Extend unicode object */
Victor Stinner3a50e702011-10-18 21:21:00 +02006755 Py_ssize_t n = PyUnicode_GET_SIZE(*v);
Victor Stinner16e6a802011-12-12 13:24:15 +01006756 if (unicode_resize(v, n + outsize) < 0)
Benjamin Peterson29060642009-01-31 22:14:21 +00006757 return -1;
Victor Stinner3a50e702011-10-18 21:21:00 +02006758 out = PyUnicode_AS_UNICODE(*v) + n;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006759 }
6760
6761 /* Do the conversion */
Victor Stinner3a50e702011-10-18 21:21:00 +02006762 outsize = MultiByteToWideChar(code_page, flags, in, insize, out, outsize);
6763 if (outsize <= 0)
6764 goto error;
6765 return insize;
Victor Stinner554f3f02010-06-16 23:33:54 +00006766
Victor Stinner3a50e702011-10-18 21:21:00 +02006767error:
6768 if (GetLastError() == ERROR_NO_UNICODE_TRANSLATION)
6769 return -2;
6770 PyErr_SetFromWindowsErr(0);
Victor Stinner554f3f02010-06-16 23:33:54 +00006771 return -1;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006772}
6773
Victor Stinner3a50e702011-10-18 21:21:00 +02006774/*
6775 * Decode a byte string from a code page into unicode object with an error
6776 * handler.
6777 *
6778 * Returns consumed size if succeed, or raise a WindowsError or
6779 * UnicodeDecodeError exception and returns -1 on error.
6780 */
6781static int
6782decode_code_page_errors(UINT code_page,
Victor Stinner76a31a62011-11-04 00:05:13 +01006783 PyObject **v,
6784 const char *in, const int size,
Victor Stinner3a50e702011-10-18 21:21:00 +02006785 const char *errors)
6786{
6787 const char *startin = in;
6788 const char *endin = in + size;
6789 const DWORD flags = decode_code_page_flags(code_page);
6790 /* Ideally, we should get reason from FormatMessage. This is the Windows
6791 2000 English version of the message. */
6792 const char *reason = "No mapping for the Unicode character exists "
6793 "in the target code page.";
6794 /* each step cannot decode more than 1 character, but a character can be
6795 represented as a surrogate pair */
6796 wchar_t buffer[2], *startout, *out;
6797 int insize, outsize;
6798 PyObject *errorHandler = NULL;
6799 PyObject *exc = NULL;
6800 PyObject *encoding_obj = NULL;
6801 char *encoding;
6802 DWORD err;
6803 int ret = -1;
6804
6805 assert(size > 0);
6806
6807 encoding = code_page_name(code_page, &encoding_obj);
6808 if (encoding == NULL)
6809 return -1;
6810
6811 if (errors == NULL || strcmp(errors, "strict") == 0) {
6812 /* The last error was ERROR_NO_UNICODE_TRANSLATION, then we raise a
6813 UnicodeDecodeError. */
6814 make_decode_exception(&exc, encoding, in, size, 0, 0, reason);
6815 if (exc != NULL) {
6816 PyCodec_StrictErrors(exc);
6817 Py_CLEAR(exc);
6818 }
6819 goto error;
6820 }
6821
6822 if (*v == NULL) {
6823 /* Create unicode object */
6824 if (size > PY_SSIZE_T_MAX / (Py_ssize_t)Py_ARRAY_LENGTH(buffer)) {
6825 PyErr_NoMemory();
6826 goto error;
6827 }
Victor Stinnerab595942011-12-17 04:59:06 +01006828 /* FIXME: don't use _PyUnicode_New(), but allocate a wchar_t* buffer */
Victor Stinner76a31a62011-11-04 00:05:13 +01006829 *v = (PyObject*)_PyUnicode_New(size * Py_ARRAY_LENGTH(buffer));
Victor Stinner3a50e702011-10-18 21:21:00 +02006830 if (*v == NULL)
6831 goto error;
6832 startout = PyUnicode_AS_UNICODE(*v);
6833 }
6834 else {
6835 /* Extend unicode object */
6836 Py_ssize_t n = PyUnicode_GET_SIZE(*v);
6837 if (size > (PY_SSIZE_T_MAX - n) / (Py_ssize_t)Py_ARRAY_LENGTH(buffer)) {
6838 PyErr_NoMemory();
6839 goto error;
6840 }
Victor Stinner16e6a802011-12-12 13:24:15 +01006841 if (unicode_resize(v, n + size * Py_ARRAY_LENGTH(buffer)) < 0)
Victor Stinner3a50e702011-10-18 21:21:00 +02006842 goto error;
6843 startout = PyUnicode_AS_UNICODE(*v) + n;
6844 }
6845
6846 /* Decode the byte string character per character */
6847 out = startout;
6848 while (in < endin)
6849 {
6850 /* Decode a character */
6851 insize = 1;
6852 do
6853 {
6854 outsize = MultiByteToWideChar(code_page, flags,
6855 in, insize,
6856 buffer, Py_ARRAY_LENGTH(buffer));
6857 if (outsize > 0)
6858 break;
6859 err = GetLastError();
6860 if (err != ERROR_NO_UNICODE_TRANSLATION
6861 && err != ERROR_INSUFFICIENT_BUFFER)
6862 {
6863 PyErr_SetFromWindowsErr(0);
6864 goto error;
6865 }
6866 insize++;
6867 }
6868 /* 4=maximum length of a UTF-8 sequence */
6869 while (insize <= 4 && (in + insize) <= endin);
6870
6871 if (outsize <= 0) {
6872 Py_ssize_t startinpos, endinpos, outpos;
6873
6874 startinpos = in - startin;
6875 endinpos = startinpos + 1;
6876 outpos = out - PyUnicode_AS_UNICODE(*v);
6877 if (unicode_decode_call_errorhandler(
6878 errors, &errorHandler,
6879 encoding, reason,
6880 &startin, &endin, &startinpos, &endinpos, &exc, &in,
Victor Stinner596a6c42011-11-09 00:02:18 +01006881 v, &outpos))
Victor Stinner3a50e702011-10-18 21:21:00 +02006882 {
6883 goto error;
6884 }
Victor Stinner596a6c42011-11-09 00:02:18 +01006885 out = PyUnicode_AS_UNICODE(*v) + outpos;
Victor Stinner3a50e702011-10-18 21:21:00 +02006886 }
6887 else {
6888 in += insize;
6889 memcpy(out, buffer, outsize * sizeof(wchar_t));
6890 out += outsize;
6891 }
6892 }
6893
6894 /* write a NUL character at the end */
6895 *out = 0;
6896
6897 /* Extend unicode object */
6898 outsize = out - startout;
6899 assert(outsize <= PyUnicode_WSTR_LENGTH(*v));
Victor Stinner16e6a802011-12-12 13:24:15 +01006900 if (unicode_resize(v, outsize) < 0)
Victor Stinner3a50e702011-10-18 21:21:00 +02006901 goto error;
Victor Stinner76a31a62011-11-04 00:05:13 +01006902 ret = size;
Victor Stinner3a50e702011-10-18 21:21:00 +02006903
6904error:
6905 Py_XDECREF(encoding_obj);
6906 Py_XDECREF(errorHandler);
6907 Py_XDECREF(exc);
6908 return ret;
6909}
6910
Victor Stinner3a50e702011-10-18 21:21:00 +02006911static PyObject *
6912decode_code_page_stateful(int code_page,
Victor Stinner76a31a62011-11-04 00:05:13 +01006913 const char *s, Py_ssize_t size,
6914 const char *errors, Py_ssize_t *consumed)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006915{
Victor Stinner76a31a62011-11-04 00:05:13 +01006916 PyObject *v = NULL;
6917 int chunk_size, final, converted, done;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006918
Victor Stinner3a50e702011-10-18 21:21:00 +02006919 if (code_page < 0) {
6920 PyErr_SetString(PyExc_ValueError, "invalid code page number");
6921 return NULL;
6922 }
6923
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006924 if (consumed)
Benjamin Peterson29060642009-01-31 22:14:21 +00006925 *consumed = 0;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006926
Victor Stinner76a31a62011-11-04 00:05:13 +01006927 do
6928 {
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006929#ifdef NEED_RETRY
Victor Stinner76a31a62011-11-04 00:05:13 +01006930 if (size > INT_MAX) {
6931 chunk_size = INT_MAX;
6932 final = 0;
6933 done = 0;
6934 }
6935 else
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006936#endif
Victor Stinner76a31a62011-11-04 00:05:13 +01006937 {
6938 chunk_size = (int)size;
6939 final = (consumed == NULL);
6940 done = 1;
6941 }
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006942
Victor Stinner76a31a62011-11-04 00:05:13 +01006943 /* Skip trailing lead-byte unless 'final' is set */
6944 if (!final && is_dbcs_lead_byte(code_page, s, chunk_size - 1))
6945 --chunk_size;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006946
Victor Stinner76a31a62011-11-04 00:05:13 +01006947 if (chunk_size == 0 && done) {
6948 if (v != NULL)
6949 break;
6950 Py_INCREF(unicode_empty);
6951 return unicode_empty;
6952 }
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006953
Victor Stinner76a31a62011-11-04 00:05:13 +01006954
6955 converted = decode_code_page_strict(code_page, &v,
6956 s, chunk_size);
6957 if (converted == -2)
6958 converted = decode_code_page_errors(code_page, &v,
6959 s, chunk_size,
6960 errors);
6961 assert(converted != 0);
6962
6963 if (converted < 0) {
6964 Py_XDECREF(v);
6965 return NULL;
6966 }
6967
6968 if (consumed)
6969 *consumed += converted;
6970
6971 s += converted;
6972 size -= converted;
6973 } while (!done);
Victor Stinner3a50e702011-10-18 21:21:00 +02006974
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01006975 return unicode_result(v);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006976}
6977
Alexander Belopolsky40018472011-02-26 01:02:56 +00006978PyObject *
Victor Stinner3a50e702011-10-18 21:21:00 +02006979PyUnicode_DecodeCodePageStateful(int code_page,
6980 const char *s,
6981 Py_ssize_t size,
6982 const char *errors,
6983 Py_ssize_t *consumed)
6984{
6985 return decode_code_page_stateful(code_page, s, size, errors, consumed);
6986}
6987
6988PyObject *
6989PyUnicode_DecodeMBCSStateful(const char *s,
6990 Py_ssize_t size,
6991 const char *errors,
6992 Py_ssize_t *consumed)
6993{
6994 return decode_code_page_stateful(CP_ACP, s, size, errors, consumed);
6995}
6996
6997PyObject *
Alexander Belopolsky40018472011-02-26 01:02:56 +00006998PyUnicode_DecodeMBCS(const char *s,
6999 Py_ssize_t size,
7000 const char *errors)
Guido van Rossumb7a40ba2000-03-28 02:01:52 +00007001{
Thomas Wouters0e3f5912006-08-11 14:57:12 +00007002 return PyUnicode_DecodeMBCSStateful(s, size, errors, NULL);
7003}
7004
Victor Stinner3a50e702011-10-18 21:21:00 +02007005static DWORD
7006encode_code_page_flags(UINT code_page, const char *errors)
7007{
7008 if (code_page == CP_UTF8) {
7009 if (winver.dwMajorVersion >= 6)
7010 /* CP_UTF8 supports WC_ERR_INVALID_CHARS on Windows Vista
7011 and later */
7012 return WC_ERR_INVALID_CHARS;
7013 else
7014 /* CP_UTF8 only supports flags=0 on Windows older than Vista */
7015 return 0;
7016 }
7017 else if (code_page == CP_UTF7) {
7018 /* CP_UTF7 only supports flags=0 */
7019 return 0;
7020 }
7021 else {
7022 if (errors != NULL && strcmp(errors, "replace") == 0)
7023 return 0;
7024 else
7025 return WC_NO_BEST_FIT_CHARS;
7026 }
7027}
7028
Thomas Wouters0e3f5912006-08-11 14:57:12 +00007029/*
Victor Stinner3a50e702011-10-18 21:21:00 +02007030 * Encode a Unicode string to a Windows code page into a byte string in strict
7031 * mode.
7032 *
7033 * Returns consumed characters if succeed, returns -2 on encode error, or raise
7034 * a WindowsError and returns -1 on other error.
Thomas Wouters0e3f5912006-08-11 14:57:12 +00007035 */
Alexander Belopolsky40018472011-02-26 01:02:56 +00007036static int
Victor Stinner3a50e702011-10-18 21:21:00 +02007037encode_code_page_strict(UINT code_page, PyObject **outbytes,
Martin v. Löwis3d325192011-11-04 18:23:06 +01007038 PyObject *unicode, Py_ssize_t offset, int len,
Victor Stinner3a50e702011-10-18 21:21:00 +02007039 const char* errors)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00007040{
Victor Stinner554f3f02010-06-16 23:33:54 +00007041 BOOL usedDefaultChar = FALSE;
Victor Stinner3a50e702011-10-18 21:21:00 +02007042 BOOL *pusedDefaultChar = &usedDefaultChar;
7043 int outsize;
Victor Stinner554f3f02010-06-16 23:33:54 +00007044 PyObject *exc = NULL;
Victor Stinner24729f32011-11-10 20:31:37 +01007045 wchar_t *p;
Victor Stinner2fc507f2011-11-04 20:06:39 +01007046 Py_ssize_t size;
Victor Stinner3a50e702011-10-18 21:21:00 +02007047 const DWORD flags = encode_code_page_flags(code_page, NULL);
7048 char *out;
Victor Stinner2fc507f2011-11-04 20:06:39 +01007049 /* Create a substring so that we can get the UTF-16 representation
7050 of just the slice under consideration. */
7051 PyObject *substring;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00007052
Martin v. Löwis3d325192011-11-04 18:23:06 +01007053 assert(len > 0);
Guido van Rossumb7a40ba2000-03-28 02:01:52 +00007054
Victor Stinner3a50e702011-10-18 21:21:00 +02007055 if (code_page != CP_UTF8 && code_page != CP_UTF7)
Victor Stinner554f3f02010-06-16 23:33:54 +00007056 pusedDefaultChar = &usedDefaultChar;
Victor Stinner3a50e702011-10-18 21:21:00 +02007057 else
Victor Stinner554f3f02010-06-16 23:33:54 +00007058 pusedDefaultChar = NULL;
Victor Stinner554f3f02010-06-16 23:33:54 +00007059
Victor Stinner2fc507f2011-11-04 20:06:39 +01007060 substring = PyUnicode_Substring(unicode, offset, offset+len);
7061 if (substring == NULL)
7062 return -1;
7063 p = PyUnicode_AsUnicodeAndSize(substring, &size);
7064 if (p == NULL) {
7065 Py_DECREF(substring);
7066 return -1;
7067 }
Martin v. Löwis3d325192011-11-04 18:23:06 +01007068
Guido van Rossumb7a40ba2000-03-28 02:01:52 +00007069 /* First get the size of the result */
Victor Stinner3a50e702011-10-18 21:21:00 +02007070 outsize = WideCharToMultiByte(code_page, flags,
7071 p, size,
7072 NULL, 0,
7073 NULL, pusedDefaultChar);
7074 if (outsize <= 0)
7075 goto error;
7076 /* If we used a default char, then we failed! */
Victor Stinner2fc507f2011-11-04 20:06:39 +01007077 if (pusedDefaultChar && *pusedDefaultChar) {
7078 Py_DECREF(substring);
Victor Stinner3a50e702011-10-18 21:21:00 +02007079 return -2;
Victor Stinner2fc507f2011-11-04 20:06:39 +01007080 }
Guido van Rossumb7a40ba2000-03-28 02:01:52 +00007081
Victor Stinner3a50e702011-10-18 21:21:00 +02007082 if (*outbytes == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +00007083 /* Create string object */
Victor Stinner3a50e702011-10-18 21:21:00 +02007084 *outbytes = PyBytes_FromStringAndSize(NULL, outsize);
Victor Stinner2fc507f2011-11-04 20:06:39 +01007085 if (*outbytes == NULL) {
7086 Py_DECREF(substring);
Benjamin Peterson29060642009-01-31 22:14:21 +00007087 return -1;
Victor Stinner2fc507f2011-11-04 20:06:39 +01007088 }
Victor Stinner3a50e702011-10-18 21:21:00 +02007089 out = PyBytes_AS_STRING(*outbytes);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00007090 }
7091 else {
Benjamin Peterson29060642009-01-31 22:14:21 +00007092 /* Extend string object */
Victor Stinner3a50e702011-10-18 21:21:00 +02007093 const Py_ssize_t n = PyBytes_Size(*outbytes);
7094 if (outsize > PY_SSIZE_T_MAX - n) {
7095 PyErr_NoMemory();
Victor Stinner2fc507f2011-11-04 20:06:39 +01007096 Py_DECREF(substring);
Benjamin Peterson29060642009-01-31 22:14:21 +00007097 return -1;
Victor Stinner3a50e702011-10-18 21:21:00 +02007098 }
Victor Stinner2fc507f2011-11-04 20:06:39 +01007099 if (_PyBytes_Resize(outbytes, n + outsize) < 0) {
7100 Py_DECREF(substring);
Victor Stinner3a50e702011-10-18 21:21:00 +02007101 return -1;
Victor Stinner2fc507f2011-11-04 20:06:39 +01007102 }
Victor Stinner3a50e702011-10-18 21:21:00 +02007103 out = PyBytes_AS_STRING(*outbytes) + n;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00007104 }
7105
7106 /* Do the conversion */
Victor Stinner3a50e702011-10-18 21:21:00 +02007107 outsize = WideCharToMultiByte(code_page, flags,
7108 p, size,
7109 out, outsize,
7110 NULL, pusedDefaultChar);
Victor Stinner2fc507f2011-11-04 20:06:39 +01007111 Py_CLEAR(substring);
Victor Stinner3a50e702011-10-18 21:21:00 +02007112 if (outsize <= 0)
7113 goto error;
7114 if (pusedDefaultChar && *pusedDefaultChar)
7115 return -2;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00007116 return 0;
Victor Stinner554f3f02010-06-16 23:33:54 +00007117
Victor Stinner3a50e702011-10-18 21:21:00 +02007118error:
Victor Stinner2fc507f2011-11-04 20:06:39 +01007119 Py_XDECREF(substring);
Victor Stinner3a50e702011-10-18 21:21:00 +02007120 if (GetLastError() == ERROR_NO_UNICODE_TRANSLATION)
7121 return -2;
7122 PyErr_SetFromWindowsErr(0);
Victor Stinner554f3f02010-06-16 23:33:54 +00007123 return -1;
Guido van Rossumb7a40ba2000-03-28 02:01:52 +00007124}
7125
Victor Stinner3a50e702011-10-18 21:21:00 +02007126/*
7127 * Encode a Unicode string to a Windows code page into a byte string using a
7128 * error handler.
7129 *
7130 * Returns consumed characters if succeed, or raise a WindowsError and returns
7131 * -1 on other error.
7132 */
7133static int
7134encode_code_page_errors(UINT code_page, PyObject **outbytes,
Victor Stinner7581cef2011-11-03 22:32:33 +01007135 PyObject *unicode, Py_ssize_t unicode_offset,
Martin v. Löwis3d325192011-11-04 18:23:06 +01007136 Py_ssize_t insize, const char* errors)
Guido van Rossumb7a40ba2000-03-28 02:01:52 +00007137{
Victor Stinner3a50e702011-10-18 21:21:00 +02007138 const DWORD flags = encode_code_page_flags(code_page, errors);
Victor Stinner2fc507f2011-11-04 20:06:39 +01007139 Py_ssize_t pos = unicode_offset;
7140 Py_ssize_t endin = unicode_offset + insize;
Victor Stinner3a50e702011-10-18 21:21:00 +02007141 /* Ideally, we should get reason from FormatMessage. This is the Windows
7142 2000 English version of the message. */
7143 const char *reason = "invalid character";
7144 /* 4=maximum length of a UTF-8 sequence */
7145 char buffer[4];
7146 BOOL usedDefaultChar = FALSE, *pusedDefaultChar;
7147 Py_ssize_t outsize;
7148 char *out;
Victor Stinner3a50e702011-10-18 21:21:00 +02007149 PyObject *errorHandler = NULL;
7150 PyObject *exc = NULL;
7151 PyObject *encoding_obj = NULL;
7152 char *encoding;
Martin v. Löwis3d325192011-11-04 18:23:06 +01007153 Py_ssize_t newpos, newoutsize;
Victor Stinner3a50e702011-10-18 21:21:00 +02007154 PyObject *rep;
7155 int ret = -1;
7156
7157 assert(insize > 0);
7158
7159 encoding = code_page_name(code_page, &encoding_obj);
7160 if (encoding == NULL)
7161 return -1;
7162
7163 if (errors == NULL || strcmp(errors, "strict") == 0) {
7164 /* The last error was ERROR_NO_UNICODE_TRANSLATION,
7165 then we raise a UnicodeEncodeError. */
Martin v. Löwis12be46c2011-11-04 19:04:15 +01007166 make_encode_exception(&exc, encoding, unicode, 0, 0, reason);
Victor Stinner3a50e702011-10-18 21:21:00 +02007167 if (exc != NULL) {
7168 PyCodec_StrictErrors(exc);
7169 Py_DECREF(exc);
7170 }
7171 Py_XDECREF(encoding_obj);
7172 return -1;
7173 }
7174
7175 if (code_page != CP_UTF8 && code_page != CP_UTF7)
7176 pusedDefaultChar = &usedDefaultChar;
7177 else
7178 pusedDefaultChar = NULL;
7179
7180 if (Py_ARRAY_LENGTH(buffer) > PY_SSIZE_T_MAX / insize) {
7181 PyErr_NoMemory();
7182 goto error;
7183 }
7184 outsize = insize * Py_ARRAY_LENGTH(buffer);
7185
7186 if (*outbytes == NULL) {
7187 /* Create string object */
7188 *outbytes = PyBytes_FromStringAndSize(NULL, outsize);
7189 if (*outbytes == NULL)
7190 goto error;
7191 out = PyBytes_AS_STRING(*outbytes);
7192 }
7193 else {
7194 /* Extend string object */
7195 Py_ssize_t n = PyBytes_Size(*outbytes);
7196 if (n > PY_SSIZE_T_MAX - outsize) {
7197 PyErr_NoMemory();
7198 goto error;
7199 }
7200 if (_PyBytes_Resize(outbytes, n + outsize) < 0)
7201 goto error;
7202 out = PyBytes_AS_STRING(*outbytes) + n;
7203 }
7204
7205 /* Encode the string character per character */
Martin v. Löwis3d325192011-11-04 18:23:06 +01007206 while (pos < endin)
Victor Stinner3a50e702011-10-18 21:21:00 +02007207 {
Victor Stinner2fc507f2011-11-04 20:06:39 +01007208 Py_UCS4 ch = PyUnicode_READ_CHAR(unicode, pos);
7209 wchar_t chars[2];
7210 int charsize;
7211 if (ch < 0x10000) {
7212 chars[0] = (wchar_t)ch;
7213 charsize = 1;
7214 }
7215 else {
7216 ch -= 0x10000;
7217 chars[0] = 0xd800 + (ch >> 10);
7218 chars[1] = 0xdc00 + (ch & 0x3ff);
7219 charsize = 2;
7220 }
7221
Victor Stinner3a50e702011-10-18 21:21:00 +02007222 outsize = WideCharToMultiByte(code_page, flags,
Martin v. Löwis3d325192011-11-04 18:23:06 +01007223 chars, charsize,
Victor Stinner3a50e702011-10-18 21:21:00 +02007224 buffer, Py_ARRAY_LENGTH(buffer),
7225 NULL, pusedDefaultChar);
7226 if (outsize > 0) {
7227 if (pusedDefaultChar == NULL || !(*pusedDefaultChar))
7228 {
Martin v. Löwis3d325192011-11-04 18:23:06 +01007229 pos++;
Victor Stinner3a50e702011-10-18 21:21:00 +02007230 memcpy(out, buffer, outsize);
7231 out += outsize;
7232 continue;
7233 }
7234 }
7235 else if (GetLastError() != ERROR_NO_UNICODE_TRANSLATION) {
7236 PyErr_SetFromWindowsErr(0);
7237 goto error;
7238 }
7239
Victor Stinner3a50e702011-10-18 21:21:00 +02007240 rep = unicode_encode_call_errorhandler(
7241 errors, &errorHandler, encoding, reason,
Victor Stinner7581cef2011-11-03 22:32:33 +01007242 unicode, &exc,
Martin v. Löwis3d325192011-11-04 18:23:06 +01007243 pos, pos + 1, &newpos);
Victor Stinner3a50e702011-10-18 21:21:00 +02007244 if (rep == NULL)
7245 goto error;
Martin v. Löwis3d325192011-11-04 18:23:06 +01007246 pos = newpos;
Victor Stinner3a50e702011-10-18 21:21:00 +02007247
7248 if (PyBytes_Check(rep)) {
7249 outsize = PyBytes_GET_SIZE(rep);
7250 if (outsize != 1) {
7251 Py_ssize_t offset = out - PyBytes_AS_STRING(*outbytes);
7252 newoutsize = PyBytes_GET_SIZE(*outbytes) + (outsize - 1);
7253 if (_PyBytes_Resize(outbytes, newoutsize) < 0) {
7254 Py_DECREF(rep);
7255 goto error;
7256 }
7257 out = PyBytes_AS_STRING(*outbytes) + offset;
7258 }
7259 memcpy(out, PyBytes_AS_STRING(rep), outsize);
7260 out += outsize;
7261 }
7262 else {
7263 Py_ssize_t i;
7264 enum PyUnicode_Kind kind;
7265 void *data;
7266
Benjamin Petersonbac79492012-01-14 13:34:47 -05007267 if (PyUnicode_READY(rep) == -1) {
Victor Stinner3a50e702011-10-18 21:21:00 +02007268 Py_DECREF(rep);
7269 goto error;
7270 }
7271
7272 outsize = PyUnicode_GET_LENGTH(rep);
7273 if (outsize != 1) {
7274 Py_ssize_t offset = out - PyBytes_AS_STRING(*outbytes);
7275 newoutsize = PyBytes_GET_SIZE(*outbytes) + (outsize - 1);
7276 if (_PyBytes_Resize(outbytes, newoutsize) < 0) {
7277 Py_DECREF(rep);
7278 goto error;
7279 }
7280 out = PyBytes_AS_STRING(*outbytes) + offset;
7281 }
7282 kind = PyUnicode_KIND(rep);
7283 data = PyUnicode_DATA(rep);
7284 for (i=0; i < outsize; i++) {
7285 Py_UCS4 ch = PyUnicode_READ(kind, data, i);
7286 if (ch > 127) {
Martin v. Löwis12be46c2011-11-04 19:04:15 +01007287 raise_encode_exception(&exc,
Martin v. Löwis3d325192011-11-04 18:23:06 +01007288 encoding, unicode,
7289 pos, pos + 1,
Victor Stinner3a50e702011-10-18 21:21:00 +02007290 "unable to encode error handler result to ASCII");
7291 Py_DECREF(rep);
7292 goto error;
7293 }
7294 *out = (unsigned char)ch;
7295 out++;
7296 }
7297 }
7298 Py_DECREF(rep);
7299 }
7300 /* write a NUL byte */
7301 *out = 0;
7302 outsize = out - PyBytes_AS_STRING(*outbytes);
7303 assert(outsize <= PyBytes_GET_SIZE(*outbytes));
7304 if (_PyBytes_Resize(outbytes, outsize) < 0)
7305 goto error;
7306 ret = 0;
7307
7308error:
7309 Py_XDECREF(encoding_obj);
7310 Py_XDECREF(errorHandler);
7311 Py_XDECREF(exc);
7312 return ret;
7313}
7314
Victor Stinner3a50e702011-10-18 21:21:00 +02007315static PyObject *
7316encode_code_page(int code_page,
Victor Stinner7581cef2011-11-03 22:32:33 +01007317 PyObject *unicode,
Victor Stinner3a50e702011-10-18 21:21:00 +02007318 const char *errors)
7319{
Martin v. Löwis3d325192011-11-04 18:23:06 +01007320 Py_ssize_t len;
Victor Stinner3a50e702011-10-18 21:21:00 +02007321 PyObject *outbytes = NULL;
Victor Stinner7581cef2011-11-03 22:32:33 +01007322 Py_ssize_t offset;
Victor Stinner76a31a62011-11-04 00:05:13 +01007323 int chunk_len, ret, done;
Victor Stinner7581cef2011-11-03 22:32:33 +01007324
Benjamin Petersonbac79492012-01-14 13:34:47 -05007325 if (PyUnicode_READY(unicode) == -1)
Victor Stinner2fc507f2011-11-04 20:06:39 +01007326 return NULL;
7327 len = PyUnicode_GET_LENGTH(unicode);
Guido van Rossum03e29f12000-05-04 15:52:20 +00007328
Victor Stinner3a50e702011-10-18 21:21:00 +02007329 if (code_page < 0) {
7330 PyErr_SetString(PyExc_ValueError, "invalid code page number");
7331 return NULL;
7332 }
7333
Martin v. Löwis3d325192011-11-04 18:23:06 +01007334 if (len == 0)
Victor Stinner76a31a62011-11-04 00:05:13 +01007335 return PyBytes_FromStringAndSize(NULL, 0);
7336
Victor Stinner7581cef2011-11-03 22:32:33 +01007337 offset = 0;
7338 do
7339 {
Thomas Wouters0e3f5912006-08-11 14:57:12 +00007340#ifdef NEED_RETRY
Victor Stinner2fc507f2011-11-04 20:06:39 +01007341 /* UTF-16 encoding may double the size, so use only INT_MAX/2
Martin v. Löwis3d325192011-11-04 18:23:06 +01007342 chunks. */
7343 if (len > INT_MAX/2) {
7344 chunk_len = INT_MAX/2;
Victor Stinner76a31a62011-11-04 00:05:13 +01007345 done = 0;
7346 }
Victor Stinner7581cef2011-11-03 22:32:33 +01007347 else
Thomas Wouters0e3f5912006-08-11 14:57:12 +00007348#endif
Victor Stinner76a31a62011-11-04 00:05:13 +01007349 {
Martin v. Löwis3d325192011-11-04 18:23:06 +01007350 chunk_len = (int)len;
Victor Stinner76a31a62011-11-04 00:05:13 +01007351 done = 1;
7352 }
Victor Stinner2fc507f2011-11-04 20:06:39 +01007353
Victor Stinner76a31a62011-11-04 00:05:13 +01007354 ret = encode_code_page_strict(code_page, &outbytes,
Martin v. Löwis3d325192011-11-04 18:23:06 +01007355 unicode, offset, chunk_len,
Victor Stinner76a31a62011-11-04 00:05:13 +01007356 errors);
7357 if (ret == -2)
7358 ret = encode_code_page_errors(code_page, &outbytes,
7359 unicode, offset,
Martin v. Löwis3d325192011-11-04 18:23:06 +01007360 chunk_len, errors);
Victor Stinner7581cef2011-11-03 22:32:33 +01007361 if (ret < 0) {
7362 Py_XDECREF(outbytes);
7363 return NULL;
7364 }
Thomas Wouters0e3f5912006-08-11 14:57:12 +00007365
Victor Stinner7581cef2011-11-03 22:32:33 +01007366 offset += chunk_len;
Martin v. Löwis3d325192011-11-04 18:23:06 +01007367 len -= chunk_len;
Victor Stinner76a31a62011-11-04 00:05:13 +01007368 } while (!done);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00007369
Victor Stinner3a50e702011-10-18 21:21:00 +02007370 return outbytes;
7371}
7372
7373PyObject *
7374PyUnicode_EncodeMBCS(const Py_UNICODE *p,
7375 Py_ssize_t size,
7376 const char *errors)
7377{
Victor Stinner7581cef2011-11-03 22:32:33 +01007378 PyObject *unicode, *res;
7379 unicode = PyUnicode_FromUnicode(p, size);
7380 if (unicode == NULL)
7381 return NULL;
7382 res = encode_code_page(CP_ACP, unicode, errors);
7383 Py_DECREF(unicode);
7384 return res;
Victor Stinner3a50e702011-10-18 21:21:00 +02007385}
7386
7387PyObject *
7388PyUnicode_EncodeCodePage(int code_page,
7389 PyObject *unicode,
7390 const char *errors)
7391{
Victor Stinner7581cef2011-11-03 22:32:33 +01007392 return encode_code_page(code_page, unicode, errors);
Guido van Rossumb7a40ba2000-03-28 02:01:52 +00007393}
Guido van Rossum2ea3e142000-03-31 17:24:09 +00007394
Alexander Belopolsky40018472011-02-26 01:02:56 +00007395PyObject *
7396PyUnicode_AsMBCSString(PyObject *unicode)
Mark Hammond0ccda1e2003-07-01 00:13:27 +00007397{
7398 if (!PyUnicode_Check(unicode)) {
7399 PyErr_BadArgument();
7400 return NULL;
7401 }
Victor Stinner7581cef2011-11-03 22:32:33 +01007402 return PyUnicode_EncodeCodePage(CP_ACP, unicode, NULL);
Mark Hammond0ccda1e2003-07-01 00:13:27 +00007403}
7404
Thomas Wouters0e3f5912006-08-11 14:57:12 +00007405#undef NEED_RETRY
7406
Victor Stinner99b95382011-07-04 14:23:54 +02007407#endif /* HAVE_MBCS */
Guido van Rossumb7a40ba2000-03-28 02:01:52 +00007408
Guido van Rossumd57fd912000-03-10 22:53:23 +00007409/* --- Character Mapping Codec -------------------------------------------- */
7410
Alexander Belopolsky40018472011-02-26 01:02:56 +00007411PyObject *
7412PyUnicode_DecodeCharmap(const char *s,
7413 Py_ssize_t size,
7414 PyObject *mapping,
7415 const char *errors)
Guido van Rossumd57fd912000-03-10 22:53:23 +00007416{
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007417 const char *starts = s;
Martin v. Löwis18e16552006-02-15 17:27:45 +00007418 Py_ssize_t startinpos;
7419 Py_ssize_t endinpos;
7420 Py_ssize_t outpos;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007421 const char *e;
Victor Stinner7931d9a2011-11-04 00:22:48 +01007422 PyObject *v;
Martin v. Löwis18e16552006-02-15 17:27:45 +00007423 Py_ssize_t extrachars = 0;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007424 PyObject *errorHandler = NULL;
7425 PyObject *exc = NULL;
Tim Petersced69f82003-09-16 20:30:58 +00007426
Guido van Rossumd57fd912000-03-10 22:53:23 +00007427 /* Default to Latin-1 */
7428 if (mapping == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00007429 return PyUnicode_DecodeLatin1(s, size, errors);
Guido van Rossumd57fd912000-03-10 22:53:23 +00007430
Martin v. Löwise9b11c12011-11-08 17:35:34 +01007431 v = PyUnicode_New(size, 127);
Guido van Rossumd57fd912000-03-10 22:53:23 +00007432 if (v == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00007433 goto onError;
Guido van Rossumd57fd912000-03-10 22:53:23 +00007434 if (size == 0)
Victor Stinner7931d9a2011-11-04 00:22:48 +01007435 return v;
Martin v. Löwise9b11c12011-11-08 17:35:34 +01007436 outpos = 0;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007437 e = s + size;
Walter Dörwaldd1c1e102005-10-06 20:29:57 +00007438 if (PyUnicode_CheckExact(mapping)) {
Victor Stinnerebf3ba82011-11-10 20:30:22 +01007439 Py_ssize_t maplen;
Antoine Pitrouaaefac72012-06-16 22:48:21 +02007440 enum PyUnicode_Kind mapkind;
7441 void *mapdata;
Victor Stinnerebf3ba82011-11-10 20:30:22 +01007442 Py_UCS4 x;
7443
Benjamin Petersonbac79492012-01-14 13:34:47 -05007444 if (PyUnicode_READY(mapping) == -1)
Victor Stinnerebf3ba82011-11-10 20:30:22 +01007445 return NULL;
7446
7447 maplen = PyUnicode_GET_LENGTH(mapping);
Antoine Pitrouaaefac72012-06-16 22:48:21 +02007448 mapdata = PyUnicode_DATA(mapping);
7449 mapkind = PyUnicode_KIND(mapping);
Benjamin Peterson29060642009-01-31 22:14:21 +00007450 while (s < e) {
Antoine Pitrouaaefac72012-06-16 22:48:21 +02007451 unsigned char ch;
7452 if (mapkind == PyUnicode_2BYTE_KIND && maplen >= 256) {
7453 enum PyUnicode_Kind outkind = PyUnicode_KIND(v);
7454 if (outkind == PyUnicode_1BYTE_KIND) {
7455 void *outdata = PyUnicode_DATA(v);
7456 Py_UCS4 maxchar = PyUnicode_MAX_CHAR_VALUE(v);
7457 while (s < e) {
7458 unsigned char ch = *s;
7459 x = PyUnicode_READ(PyUnicode_2BYTE_KIND, mapdata, ch);
7460 if (x > maxchar)
7461 goto Error;
7462 PyUnicode_WRITE(PyUnicode_1BYTE_KIND, outdata, outpos++, x);
7463 ++s;
7464 }
7465 break;
7466 }
7467 else if (outkind == PyUnicode_2BYTE_KIND) {
7468 void *outdata = PyUnicode_DATA(v);
7469 while (s < e) {
7470 unsigned char ch = *s;
7471 x = PyUnicode_READ(PyUnicode_2BYTE_KIND, mapdata, ch);
7472 if (x == 0xFFFE)
7473 goto Error;
7474 PyUnicode_WRITE(PyUnicode_2BYTE_KIND, outdata, outpos++, x);
7475 ++s;
7476 }
7477 break;
7478 }
7479 }
7480 ch = *s;
Guido van Rossumd57fd912000-03-10 22:53:23 +00007481
Benjamin Peterson29060642009-01-31 22:14:21 +00007482 if (ch < maplen)
Antoine Pitrouaaefac72012-06-16 22:48:21 +02007483 x = PyUnicode_READ(mapkind, mapdata, ch);
Victor Stinnerebf3ba82011-11-10 20:30:22 +01007484 else
7485 x = 0xfffe; /* invalid value */
Antoine Pitrouaaefac72012-06-16 22:48:21 +02007486Error:
Victor Stinnerebf3ba82011-11-10 20:30:22 +01007487 if (x == 0xfffe)
7488 {
Benjamin Peterson29060642009-01-31 22:14:21 +00007489 /* undefined mapping */
Benjamin Peterson29060642009-01-31 22:14:21 +00007490 startinpos = s-starts;
7491 endinpos = startinpos+1;
7492 if (unicode_decode_call_errorhandler(
7493 errors, &errorHandler,
7494 "charmap", "character maps to <undefined>",
7495 &starts, &e, &startinpos, &endinpos, &exc, &s,
Martin v. Löwise9b11c12011-11-08 17:35:34 +01007496 &v, &outpos)) {
Benjamin Peterson29060642009-01-31 22:14:21 +00007497 goto onError;
7498 }
7499 continue;
7500 }
Victor Stinnerebf3ba82011-11-10 20:30:22 +01007501
Martin v. Löwise9b11c12011-11-08 17:35:34 +01007502 if (unicode_putchar(&v, &outpos, x) < 0)
7503 goto onError;
Benjamin Peterson29060642009-01-31 22:14:21 +00007504 ++s;
Benjamin Peterson14339b62009-01-31 16:36:08 +00007505 }
Walter Dörwaldd1c1e102005-10-06 20:29:57 +00007506 }
7507 else {
Benjamin Peterson29060642009-01-31 22:14:21 +00007508 while (s < e) {
7509 unsigned char ch = *s;
7510 PyObject *w, *x;
Walter Dörwaldd1c1e102005-10-06 20:29:57 +00007511
Benjamin Peterson29060642009-01-31 22:14:21 +00007512 /* Get mapping (char ordinal -> integer, Unicode char or None) */
7513 w = PyLong_FromLong((long)ch);
7514 if (w == NULL)
7515 goto onError;
7516 x = PyObject_GetItem(mapping, w);
7517 Py_DECREF(w);
7518 if (x == NULL) {
7519 if (PyErr_ExceptionMatches(PyExc_LookupError)) {
7520 /* No mapping found means: mapping is undefined. */
7521 PyErr_Clear();
7522 x = Py_None;
7523 Py_INCREF(x);
7524 } else
7525 goto onError;
7526 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00007527
Benjamin Peterson29060642009-01-31 22:14:21 +00007528 /* Apply mapping */
7529 if (PyLong_Check(x)) {
7530 long value = PyLong_AS_LONG(x);
7531 if (value < 0 || value > 65535) {
7532 PyErr_SetString(PyExc_TypeError,
7533 "character mapping must be in range(65536)");
7534 Py_DECREF(x);
7535 goto onError;
7536 }
Martin v. Löwise9b11c12011-11-08 17:35:34 +01007537 if (unicode_putchar(&v, &outpos, value) < 0)
7538 goto onError;
Benjamin Peterson29060642009-01-31 22:14:21 +00007539 }
7540 else if (x == Py_None) {
7541 /* undefined mapping */
Benjamin Peterson29060642009-01-31 22:14:21 +00007542 startinpos = s-starts;
7543 endinpos = startinpos+1;
7544 if (unicode_decode_call_errorhandler(
7545 errors, &errorHandler,
7546 "charmap", "character maps to <undefined>",
7547 &starts, &e, &startinpos, &endinpos, &exc, &s,
Martin v. Löwise9b11c12011-11-08 17:35:34 +01007548 &v, &outpos)) {
Benjamin Peterson29060642009-01-31 22:14:21 +00007549 Py_DECREF(x);
7550 goto onError;
7551 }
7552 Py_DECREF(x);
7553 continue;
7554 }
7555 else if (PyUnicode_Check(x)) {
Martin v. Löwise9b11c12011-11-08 17:35:34 +01007556 Py_ssize_t targetsize;
Benjamin Peterson14339b62009-01-31 16:36:08 +00007557
Benjamin Petersonbac79492012-01-14 13:34:47 -05007558 if (PyUnicode_READY(x) == -1)
Martin v. Löwise9b11c12011-11-08 17:35:34 +01007559 goto onError;
7560 targetsize = PyUnicode_GET_LENGTH(x);
7561
7562 if (targetsize == 1) {
Benjamin Peterson29060642009-01-31 22:14:21 +00007563 /* 1-1 mapping */
Victor Stinner62aa4d02011-11-09 00:03:45 +01007564 if (unicode_putchar(&v, &outpos,
Martin v. Löwise9b11c12011-11-08 17:35:34 +01007565 PyUnicode_READ_CHAR(x, 0)) < 0)
7566 goto onError;
7567 }
Benjamin Peterson29060642009-01-31 22:14:21 +00007568 else if (targetsize > 1) {
7569 /* 1-n mapping */
7570 if (targetsize > extrachars) {
7571 /* resize first */
Benjamin Peterson29060642009-01-31 22:14:21 +00007572 Py_ssize_t needed = (targetsize - extrachars) + \
7573 (targetsize << 2);
7574 extrachars += needed;
7575 /* XXX overflow detection missing */
Victor Stinner16e6a802011-12-12 13:24:15 +01007576 if (unicode_resize(&v,
7577 PyUnicode_GET_LENGTH(v) + needed) < 0)
7578 {
Benjamin Peterson29060642009-01-31 22:14:21 +00007579 Py_DECREF(x);
7580 goto onError;
7581 }
Benjamin Peterson29060642009-01-31 22:14:21 +00007582 }
Victor Stinner1b487b42012-05-03 12:29:04 +02007583 if (unicode_widen(&v, outpos, PyUnicode_MAX_CHAR_VALUE(x)) < 0)
Martin v. Löwise9b11c12011-11-08 17:35:34 +01007584 goto onError;
7585 PyUnicode_CopyCharacters(v, outpos, x, 0, targetsize);
7586 outpos += targetsize;
Benjamin Peterson29060642009-01-31 22:14:21 +00007587 extrachars -= targetsize;
7588 }
7589 /* 1-0 mapping: skip the character */
7590 }
7591 else {
7592 /* wrong return value */
7593 PyErr_SetString(PyExc_TypeError,
7594 "character mapping must return integer, None or str");
Benjamin Peterson14339b62009-01-31 16:36:08 +00007595 Py_DECREF(x);
7596 goto onError;
7597 }
Benjamin Peterson29060642009-01-31 22:14:21 +00007598 Py_DECREF(x);
7599 ++s;
Benjamin Peterson14339b62009-01-31 16:36:08 +00007600 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00007601 }
Victor Stinner16e6a802011-12-12 13:24:15 +01007602 if (unicode_resize(&v, outpos) < 0)
Antoine Pitroua8f63c02011-11-08 18:37:16 +01007603 goto onError;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007604 Py_XDECREF(errorHandler);
7605 Py_XDECREF(exc);
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01007606 return unicode_result(v);
Tim Petersced69f82003-09-16 20:30:58 +00007607
Benjamin Peterson29060642009-01-31 22:14:21 +00007608 onError:
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007609 Py_XDECREF(errorHandler);
7610 Py_XDECREF(exc);
Guido van Rossumd57fd912000-03-10 22:53:23 +00007611 Py_XDECREF(v);
7612 return NULL;
7613}
7614
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007615/* Charmap encoding: the lookup table */
7616
Alexander Belopolsky40018472011-02-26 01:02:56 +00007617struct encoding_map {
Benjamin Peterson29060642009-01-31 22:14:21 +00007618 PyObject_HEAD
7619 unsigned char level1[32];
7620 int count2, count3;
7621 unsigned char level23[1];
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007622};
7623
7624static PyObject*
7625encoding_map_size(PyObject *obj, PyObject* args)
7626{
7627 struct encoding_map *map = (struct encoding_map*)obj;
Benjamin Peterson14339b62009-01-31 16:36:08 +00007628 return PyLong_FromLong(sizeof(*map) - 1 + 16*map->count2 +
Benjamin Peterson29060642009-01-31 22:14:21 +00007629 128*map->count3);
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007630}
7631
7632static PyMethodDef encoding_map_methods[] = {
Benjamin Peterson14339b62009-01-31 16:36:08 +00007633 {"size", encoding_map_size, METH_NOARGS,
Benjamin Peterson29060642009-01-31 22:14:21 +00007634 PyDoc_STR("Return the size (in bytes) of this object") },
7635 { 0 }
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007636};
7637
7638static void
7639encoding_map_dealloc(PyObject* o)
7640{
Benjamin Peterson14339b62009-01-31 16:36:08 +00007641 PyObject_FREE(o);
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007642}
7643
7644static PyTypeObject EncodingMapType = {
Benjamin Peterson14339b62009-01-31 16:36:08 +00007645 PyVarObject_HEAD_INIT(NULL, 0)
Benjamin Peterson29060642009-01-31 22:14:21 +00007646 "EncodingMap", /*tp_name*/
7647 sizeof(struct encoding_map), /*tp_basicsize*/
7648 0, /*tp_itemsize*/
7649 /* methods */
7650 encoding_map_dealloc, /*tp_dealloc*/
7651 0, /*tp_print*/
7652 0, /*tp_getattr*/
7653 0, /*tp_setattr*/
Mark Dickinsone94c6792009-02-02 20:36:42 +00007654 0, /*tp_reserved*/
Benjamin Peterson29060642009-01-31 22:14:21 +00007655 0, /*tp_repr*/
7656 0, /*tp_as_number*/
7657 0, /*tp_as_sequence*/
7658 0, /*tp_as_mapping*/
7659 0, /*tp_hash*/
7660 0, /*tp_call*/
7661 0, /*tp_str*/
7662 0, /*tp_getattro*/
7663 0, /*tp_setattro*/
7664 0, /*tp_as_buffer*/
7665 Py_TPFLAGS_DEFAULT, /*tp_flags*/
7666 0, /*tp_doc*/
7667 0, /*tp_traverse*/
7668 0, /*tp_clear*/
7669 0, /*tp_richcompare*/
7670 0, /*tp_weaklistoffset*/
7671 0, /*tp_iter*/
7672 0, /*tp_iternext*/
7673 encoding_map_methods, /*tp_methods*/
7674 0, /*tp_members*/
7675 0, /*tp_getset*/
7676 0, /*tp_base*/
7677 0, /*tp_dict*/
7678 0, /*tp_descr_get*/
7679 0, /*tp_descr_set*/
7680 0, /*tp_dictoffset*/
7681 0, /*tp_init*/
7682 0, /*tp_alloc*/
7683 0, /*tp_new*/
7684 0, /*tp_free*/
7685 0, /*tp_is_gc*/
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007686};
7687
7688PyObject*
7689PyUnicode_BuildEncodingMap(PyObject* string)
7690{
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007691 PyObject *result;
7692 struct encoding_map *mresult;
7693 int i;
7694 int need_dict = 0;
7695 unsigned char level1[32];
7696 unsigned char level2[512];
7697 unsigned char *mlevel1, *mlevel2, *mlevel3;
7698 int count2 = 0, count3 = 0;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02007699 int kind;
7700 void *data;
Antoine Pitrouaaefac72012-06-16 22:48:21 +02007701 Py_ssize_t length;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02007702 Py_UCS4 ch;
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007703
Antoine Pitrouaaefac72012-06-16 22:48:21 +02007704 if (!PyUnicode_Check(string) || !PyUnicode_GET_LENGTH(string)) {
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007705 PyErr_BadArgument();
7706 return NULL;
7707 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02007708 kind = PyUnicode_KIND(string);
7709 data = PyUnicode_DATA(string);
Antoine Pitrouaaefac72012-06-16 22:48:21 +02007710 length = PyUnicode_GET_LENGTH(string);
7711 length = Py_MIN(length, 256);
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007712 memset(level1, 0xFF, sizeof level1);
7713 memset(level2, 0xFF, sizeof level2);
7714
7715 /* If there isn't a one-to-one mapping of NULL to \0,
7716 or if there are non-BMP characters, we need to use
7717 a mapping dictionary. */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02007718 if (PyUnicode_READ(kind, data, 0) != 0)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007719 need_dict = 1;
Antoine Pitrouaaefac72012-06-16 22:48:21 +02007720 for (i = 1; i < length; i++) {
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007721 int l1, l2;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02007722 ch = PyUnicode_READ(kind, data, i);
7723 if (ch == 0 || ch > 0xFFFF) {
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007724 need_dict = 1;
7725 break;
7726 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02007727 if (ch == 0xFFFE)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007728 /* unmapped character */
7729 continue;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02007730 l1 = ch >> 11;
7731 l2 = ch >> 7;
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007732 if (level1[l1] == 0xFF)
7733 level1[l1] = count2++;
7734 if (level2[l2] == 0xFF)
Benjamin Peterson14339b62009-01-31 16:36:08 +00007735 level2[l2] = count3++;
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007736 }
7737
7738 if (count2 >= 0xFF || count3 >= 0xFF)
7739 need_dict = 1;
7740
7741 if (need_dict) {
7742 PyObject *result = PyDict_New();
7743 PyObject *key, *value;
7744 if (!result)
7745 return NULL;
Antoine Pitrouaaefac72012-06-16 22:48:21 +02007746 for (i = 0; i < length; i++) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02007747 key = PyLong_FromLong(PyUnicode_READ(kind, data, i));
Christian Heimes217cfd12007-12-02 14:31:20 +00007748 value = PyLong_FromLong(i);
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007749 if (!key || !value)
7750 goto failed1;
7751 if (PyDict_SetItem(result, key, value) == -1)
7752 goto failed1;
7753 Py_DECREF(key);
7754 Py_DECREF(value);
7755 }
7756 return result;
7757 failed1:
7758 Py_XDECREF(key);
7759 Py_XDECREF(value);
7760 Py_DECREF(result);
7761 return NULL;
7762 }
7763
7764 /* Create a three-level trie */
7765 result = PyObject_MALLOC(sizeof(struct encoding_map) +
7766 16*count2 + 128*count3 - 1);
7767 if (!result)
7768 return PyErr_NoMemory();
7769 PyObject_Init(result, &EncodingMapType);
7770 mresult = (struct encoding_map*)result;
7771 mresult->count2 = count2;
7772 mresult->count3 = count3;
7773 mlevel1 = mresult->level1;
7774 mlevel2 = mresult->level23;
7775 mlevel3 = mresult->level23 + 16*count2;
7776 memcpy(mlevel1, level1, 32);
7777 memset(mlevel2, 0xFF, 16*count2);
7778 memset(mlevel3, 0, 128*count3);
7779 count3 = 0;
Antoine Pitrouaaefac72012-06-16 22:48:21 +02007780 for (i = 1; i < length; i++) {
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007781 int o1, o2, o3, i2, i3;
Antoine Pitrouaaefac72012-06-16 22:48:21 +02007782 Py_UCS4 ch = PyUnicode_READ(kind, data, i);
7783 if (ch == 0xFFFE)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007784 /* unmapped character */
7785 continue;
Antoine Pitrouaaefac72012-06-16 22:48:21 +02007786 o1 = ch>>11;
7787 o2 = (ch>>7) & 0xF;
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007788 i2 = 16*mlevel1[o1] + o2;
7789 if (mlevel2[i2] == 0xFF)
7790 mlevel2[i2] = count3++;
Antoine Pitrouaaefac72012-06-16 22:48:21 +02007791 o3 = ch & 0x7F;
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007792 i3 = 128*mlevel2[i2] + o3;
7793 mlevel3[i3] = i;
7794 }
7795 return result;
7796}
7797
7798static int
Victor Stinner22168992011-11-20 17:09:18 +01007799encoding_map_lookup(Py_UCS4 c, PyObject *mapping)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007800{
7801 struct encoding_map *map = (struct encoding_map*)mapping;
7802 int l1 = c>>11;
7803 int l2 = (c>>7) & 0xF;
7804 int l3 = c & 0x7F;
7805 int i;
7806
Victor Stinner22168992011-11-20 17:09:18 +01007807 if (c > 0xFFFF)
Benjamin Peterson29060642009-01-31 22:14:21 +00007808 return -1;
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007809 if (c == 0)
7810 return 0;
7811 /* level 1*/
7812 i = map->level1[l1];
7813 if (i == 0xFF) {
7814 return -1;
7815 }
7816 /* level 2*/
7817 i = map->level23[16*i+l2];
7818 if (i == 0xFF) {
7819 return -1;
7820 }
7821 /* level 3 */
7822 i = map->level23[16*map->count2 + 128*i + l3];
7823 if (i == 0) {
7824 return -1;
7825 }
7826 return i;
7827}
7828
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007829/* Lookup the character ch in the mapping. If the character
7830 can't be found, Py_None is returned (or NULL, if another
Fred Drakedb390c12005-10-28 14:39:47 +00007831 error occurred). */
Alexander Belopolsky40018472011-02-26 01:02:56 +00007832static PyObject *
Victor Stinner22168992011-11-20 17:09:18 +01007833charmapencode_lookup(Py_UCS4 c, PyObject *mapping)
Guido van Rossumd57fd912000-03-10 22:53:23 +00007834{
Christian Heimes217cfd12007-12-02 14:31:20 +00007835 PyObject *w = PyLong_FromLong((long)c);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007836 PyObject *x;
7837
7838 if (w == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00007839 return NULL;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007840 x = PyObject_GetItem(mapping, w);
7841 Py_DECREF(w);
7842 if (x == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +00007843 if (PyErr_ExceptionMatches(PyExc_LookupError)) {
7844 /* No mapping found means: mapping is undefined. */
7845 PyErr_Clear();
7846 x = Py_None;
7847 Py_INCREF(x);
7848 return x;
7849 } else
7850 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00007851 }
Walter Dörwaldadc72742003-01-08 22:01:33 +00007852 else if (x == Py_None)
Benjamin Peterson29060642009-01-31 22:14:21 +00007853 return x;
Christian Heimes217cfd12007-12-02 14:31:20 +00007854 else if (PyLong_Check(x)) {
Benjamin Peterson29060642009-01-31 22:14:21 +00007855 long value = PyLong_AS_LONG(x);
7856 if (value < 0 || value > 255) {
7857 PyErr_SetString(PyExc_TypeError,
7858 "character mapping must be in range(256)");
7859 Py_DECREF(x);
7860 return NULL;
7861 }
7862 return x;
Guido van Rossumd57fd912000-03-10 22:53:23 +00007863 }
Christian Heimes72b710a2008-05-26 13:28:38 +00007864 else if (PyBytes_Check(x))
Benjamin Peterson29060642009-01-31 22:14:21 +00007865 return x;
Guido van Rossumd57fd912000-03-10 22:53:23 +00007866 else {
Benjamin Peterson29060642009-01-31 22:14:21 +00007867 /* wrong return value */
7868 PyErr_Format(PyExc_TypeError,
7869 "character mapping must return integer, bytes or None, not %.400s",
7870 x->ob_type->tp_name);
7871 Py_DECREF(x);
7872 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00007873 }
7874}
7875
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007876static int
Guido van Rossum98297ee2007-11-06 21:34:58 +00007877charmapencode_resize(PyObject **outobj, Py_ssize_t *outpos, Py_ssize_t requiredsize)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007878{
Benjamin Peterson14339b62009-01-31 16:36:08 +00007879 Py_ssize_t outsize = PyBytes_GET_SIZE(*outobj);
7880 /* exponentially overallocate to minimize reallocations */
7881 if (requiredsize < 2*outsize)
7882 requiredsize = 2*outsize;
7883 if (_PyBytes_Resize(outobj, requiredsize))
7884 return -1;
7885 return 0;
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007886}
7887
Benjamin Peterson14339b62009-01-31 16:36:08 +00007888typedef enum charmapencode_result {
Benjamin Peterson29060642009-01-31 22:14:21 +00007889 enc_SUCCESS, enc_FAILED, enc_EXCEPTION
Alexander Belopolsky40018472011-02-26 01:02:56 +00007890} charmapencode_result;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007891/* lookup the character, put the result in the output string and adjust
Walter Dörwald827b0552007-05-12 13:23:53 +00007892 various state variables. Resize the output bytes object if not enough
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007893 space is available. Return a new reference to the object that
7894 was put in the output buffer, or Py_None, if the mapping was undefined
7895 (in which case no character was written) or NULL, if a
Andrew M. Kuchling8294de52005-11-02 16:36:12 +00007896 reallocation error occurred. The caller must decref the result */
Alexander Belopolsky40018472011-02-26 01:02:56 +00007897static charmapencode_result
Victor Stinner22168992011-11-20 17:09:18 +01007898charmapencode_output(Py_UCS4 c, PyObject *mapping,
Alexander Belopolsky40018472011-02-26 01:02:56 +00007899 PyObject **outobj, Py_ssize_t *outpos)
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007900{
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007901 PyObject *rep;
7902 char *outstart;
Christian Heimes72b710a2008-05-26 13:28:38 +00007903 Py_ssize_t outsize = PyBytes_GET_SIZE(*outobj);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007904
Christian Heimes90aa7642007-12-19 02:45:37 +00007905 if (Py_TYPE(mapping) == &EncodingMapType) {
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007906 int res = encoding_map_lookup(c, mapping);
Benjamin Peterson29060642009-01-31 22:14:21 +00007907 Py_ssize_t requiredsize = *outpos+1;
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007908 if (res == -1)
7909 return enc_FAILED;
Benjamin Peterson29060642009-01-31 22:14:21 +00007910 if (outsize<requiredsize)
7911 if (charmapencode_resize(outobj, outpos, requiredsize))
7912 return enc_EXCEPTION;
Christian Heimes72b710a2008-05-26 13:28:38 +00007913 outstart = PyBytes_AS_STRING(*outobj);
Benjamin Peterson29060642009-01-31 22:14:21 +00007914 outstart[(*outpos)++] = (char)res;
7915 return enc_SUCCESS;
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007916 }
7917
7918 rep = charmapencode_lookup(c, mapping);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007919 if (rep==NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00007920 return enc_EXCEPTION;
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007921 else if (rep==Py_None) {
Benjamin Peterson29060642009-01-31 22:14:21 +00007922 Py_DECREF(rep);
7923 return enc_FAILED;
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007924 } else {
Benjamin Peterson29060642009-01-31 22:14:21 +00007925 if (PyLong_Check(rep)) {
7926 Py_ssize_t requiredsize = *outpos+1;
7927 if (outsize<requiredsize)
7928 if (charmapencode_resize(outobj, outpos, requiredsize)) {
7929 Py_DECREF(rep);
7930 return enc_EXCEPTION;
7931 }
Christian Heimes72b710a2008-05-26 13:28:38 +00007932 outstart = PyBytes_AS_STRING(*outobj);
Benjamin Peterson29060642009-01-31 22:14:21 +00007933 outstart[(*outpos)++] = (char)PyLong_AS_LONG(rep);
Benjamin Peterson14339b62009-01-31 16:36:08 +00007934 }
Benjamin Peterson29060642009-01-31 22:14:21 +00007935 else {
7936 const char *repchars = PyBytes_AS_STRING(rep);
7937 Py_ssize_t repsize = PyBytes_GET_SIZE(rep);
7938 Py_ssize_t requiredsize = *outpos+repsize;
7939 if (outsize<requiredsize)
7940 if (charmapencode_resize(outobj, outpos, requiredsize)) {
7941 Py_DECREF(rep);
7942 return enc_EXCEPTION;
7943 }
Christian Heimes72b710a2008-05-26 13:28:38 +00007944 outstart = PyBytes_AS_STRING(*outobj);
Benjamin Peterson29060642009-01-31 22:14:21 +00007945 memcpy(outstart + *outpos, repchars, repsize);
7946 *outpos += repsize;
7947 }
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007948 }
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007949 Py_DECREF(rep);
7950 return enc_SUCCESS;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007951}
7952
7953/* handle an error in PyUnicode_EncodeCharmap
7954 Return 0 on success, -1 on error */
Alexander Belopolsky40018472011-02-26 01:02:56 +00007955static int
7956charmap_encoding_error(
Martin v. Löwis23e275b2011-11-02 18:02:51 +01007957 PyObject *unicode, Py_ssize_t *inpos, PyObject *mapping,
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007958 PyObject **exceptionObject,
Walter Dörwalde5402fb2003-08-14 20:25:29 +00007959 int *known_errorHandler, PyObject **errorHandler, const char *errors,
Guido van Rossum98297ee2007-11-06 21:34:58 +00007960 PyObject **res, Py_ssize_t *respos)
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007961{
7962 PyObject *repunicode = NULL; /* initialize to prevent gcc warning */
Martin v. Löwis23e275b2011-11-02 18:02:51 +01007963 Py_ssize_t size, repsize;
Martin v. Löwis18e16552006-02-15 17:27:45 +00007964 Py_ssize_t newpos;
Victor Stinnerae4f7c82011-11-20 18:28:55 +01007965 enum PyUnicode_Kind kind;
7966 void *data;
7967 Py_ssize_t index;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007968 /* startpos for collecting unencodable chars */
Martin v. Löwis18e16552006-02-15 17:27:45 +00007969 Py_ssize_t collstartpos = *inpos;
7970 Py_ssize_t collendpos = *inpos+1;
7971 Py_ssize_t collpos;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007972 char *encoding = "charmap";
7973 char *reason = "character maps to <undefined>";
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007974 charmapencode_result x;
Martin v. Löwis23e275b2011-11-02 18:02:51 +01007975 Py_UCS4 ch;
Brian Curtin2787ea42011-11-02 15:09:37 -05007976 int val;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007977
Benjamin Petersonbac79492012-01-14 13:34:47 -05007978 if (PyUnicode_READY(unicode) == -1)
Martin v. Löwis23e275b2011-11-02 18:02:51 +01007979 return -1;
7980 size = PyUnicode_GET_LENGTH(unicode);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007981 /* find all unencodable characters */
7982 while (collendpos < size) {
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007983 PyObject *rep;
Christian Heimes90aa7642007-12-19 02:45:37 +00007984 if (Py_TYPE(mapping) == &EncodingMapType) {
Martin v. Löwis23e275b2011-11-02 18:02:51 +01007985 ch = PyUnicode_READ_CHAR(unicode, collendpos);
Brian Curtin2787ea42011-11-02 15:09:37 -05007986 val = encoding_map_lookup(ch, mapping);
7987 if (val != -1)
Benjamin Peterson29060642009-01-31 22:14:21 +00007988 break;
7989 ++collendpos;
7990 continue;
7991 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00007992
Martin v. Löwis23e275b2011-11-02 18:02:51 +01007993 ch = PyUnicode_READ_CHAR(unicode, collendpos);
7994 rep = charmapencode_lookup(ch, mapping);
Benjamin Peterson29060642009-01-31 22:14:21 +00007995 if (rep==NULL)
7996 return -1;
7997 else if (rep!=Py_None) {
7998 Py_DECREF(rep);
7999 break;
8000 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00008001 Py_DECREF(rep);
Benjamin Peterson29060642009-01-31 22:14:21 +00008002 ++collendpos;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008003 }
8004 /* cache callback name lookup
8005 * (if not done yet, i.e. it's the first error) */
8006 if (*known_errorHandler==-1) {
Benjamin Peterson29060642009-01-31 22:14:21 +00008007 if ((errors==NULL) || (!strcmp(errors, "strict")))
8008 *known_errorHandler = 1;
8009 else if (!strcmp(errors, "replace"))
8010 *known_errorHandler = 2;
8011 else if (!strcmp(errors, "ignore"))
8012 *known_errorHandler = 3;
8013 else if (!strcmp(errors, "xmlcharrefreplace"))
8014 *known_errorHandler = 4;
8015 else
8016 *known_errorHandler = 0;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008017 }
8018 switch (*known_errorHandler) {
Benjamin Peterson14339b62009-01-31 16:36:08 +00008019 case 1: /* strict */
Martin v. Löwis12be46c2011-11-04 19:04:15 +01008020 raise_encode_exception(exceptionObject, encoding, unicode, collstartpos, collendpos, reason);
Benjamin Peterson14339b62009-01-31 16:36:08 +00008021 return -1;
8022 case 2: /* replace */
8023 for (collpos = collstartpos; collpos<collendpos; ++collpos) {
Benjamin Peterson29060642009-01-31 22:14:21 +00008024 x = charmapencode_output('?', mapping, res, respos);
8025 if (x==enc_EXCEPTION) {
8026 return -1;
8027 }
8028 else if (x==enc_FAILED) {
Martin v. Löwis12be46c2011-11-04 19:04:15 +01008029 raise_encode_exception(exceptionObject, encoding, unicode, collstartpos, collendpos, reason);
Benjamin Peterson29060642009-01-31 22:14:21 +00008030 return -1;
8031 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00008032 }
8033 /* fall through */
8034 case 3: /* ignore */
8035 *inpos = collendpos;
8036 break;
8037 case 4: /* xmlcharrefreplace */
8038 /* generate replacement (temporarily (mis)uses p) */
8039 for (collpos = collstartpos; collpos < collendpos; ++collpos) {
Benjamin Peterson29060642009-01-31 22:14:21 +00008040 char buffer[2+29+1+1];
8041 char *cp;
Martin v. Löwis23e275b2011-11-02 18:02:51 +01008042 sprintf(buffer, "&#%d;", (int)PyUnicode_READ_CHAR(unicode, collpos));
Benjamin Peterson29060642009-01-31 22:14:21 +00008043 for (cp = buffer; *cp; ++cp) {
8044 x = charmapencode_output(*cp, mapping, res, respos);
8045 if (x==enc_EXCEPTION)
8046 return -1;
8047 else if (x==enc_FAILED) {
Martin v. Löwis12be46c2011-11-04 19:04:15 +01008048 raise_encode_exception(exceptionObject, encoding, unicode, collstartpos, collendpos, reason);
Benjamin Peterson29060642009-01-31 22:14:21 +00008049 return -1;
8050 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00008051 }
8052 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00008053 *inpos = collendpos;
8054 break;
8055 default:
8056 repunicode = unicode_encode_call_errorhandler(errors, errorHandler,
Martin v. Löwis23e275b2011-11-02 18:02:51 +01008057 encoding, reason, unicode, exceptionObject,
Benjamin Peterson29060642009-01-31 22:14:21 +00008058 collstartpos, collendpos, &newpos);
Benjamin Peterson14339b62009-01-31 16:36:08 +00008059 if (repunicode == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00008060 return -1;
Martin v. Löwis011e8422009-05-05 04:43:17 +00008061 if (PyBytes_Check(repunicode)) {
8062 /* Directly copy bytes result to output. */
8063 Py_ssize_t outsize = PyBytes_Size(*res);
8064 Py_ssize_t requiredsize;
8065 repsize = PyBytes_Size(repunicode);
8066 requiredsize = *respos + repsize;
8067 if (requiredsize > outsize)
8068 /* Make room for all additional bytes. */
8069 if (charmapencode_resize(res, respos, requiredsize)) {
8070 Py_DECREF(repunicode);
8071 return -1;
8072 }
8073 memcpy(PyBytes_AsString(*res) + *respos,
8074 PyBytes_AsString(repunicode), repsize);
8075 *respos += repsize;
8076 *inpos = newpos;
Martin v. Löwisdb12d452009-05-02 18:52:14 +00008077 Py_DECREF(repunicode);
Martin v. Löwis011e8422009-05-05 04:43:17 +00008078 break;
Martin v. Löwisdb12d452009-05-02 18:52:14 +00008079 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00008080 /* generate replacement */
Benjamin Petersonbac79492012-01-14 13:34:47 -05008081 if (PyUnicode_READY(repunicode) == -1) {
Victor Stinnerae4f7c82011-11-20 18:28:55 +01008082 Py_DECREF(repunicode);
8083 return -1;
8084 }
Victor Stinner9e30aa52011-11-21 02:49:52 +01008085 repsize = PyUnicode_GET_LENGTH(repunicode);
Victor Stinnerae4f7c82011-11-20 18:28:55 +01008086 data = PyUnicode_DATA(repunicode);
8087 kind = PyUnicode_KIND(repunicode);
8088 for (index = 0; index < repsize; index++) {
8089 Py_UCS4 repch = PyUnicode_READ(kind, data, index);
8090 x = charmapencode_output(repch, mapping, res, respos);
Benjamin Peterson29060642009-01-31 22:14:21 +00008091 if (x==enc_EXCEPTION) {
Victor Stinnerae4f7c82011-11-20 18:28:55 +01008092 Py_DECREF(repunicode);
Benjamin Peterson29060642009-01-31 22:14:21 +00008093 return -1;
8094 }
8095 else if (x==enc_FAILED) {
8096 Py_DECREF(repunicode);
Martin v. Löwis12be46c2011-11-04 19:04:15 +01008097 raise_encode_exception(exceptionObject, encoding, unicode, collstartpos, collendpos, reason);
Benjamin Peterson29060642009-01-31 22:14:21 +00008098 return -1;
8099 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00008100 }
8101 *inpos = newpos;
8102 Py_DECREF(repunicode);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008103 }
8104 return 0;
8105}
8106
Alexander Belopolsky40018472011-02-26 01:02:56 +00008107PyObject *
Martin v. Löwis23e275b2011-11-02 18:02:51 +01008108_PyUnicode_EncodeCharmap(PyObject *unicode,
8109 PyObject *mapping,
8110 const char *errors)
Guido van Rossumd57fd912000-03-10 22:53:23 +00008111{
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008112 /* output object */
8113 PyObject *res = NULL;
8114 /* current input position */
Martin v. Löwis18e16552006-02-15 17:27:45 +00008115 Py_ssize_t inpos = 0;
Martin v. Löwis23e275b2011-11-02 18:02:51 +01008116 Py_ssize_t size;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008117 /* current output position */
Martin v. Löwis18e16552006-02-15 17:27:45 +00008118 Py_ssize_t respos = 0;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008119 PyObject *errorHandler = NULL;
8120 PyObject *exc = NULL;
8121 /* the following variable is used for caching string comparisons
8122 * -1=not initialized, 0=unknown, 1=strict, 2=replace,
8123 * 3=ignore, 4=xmlcharrefreplace */
8124 int known_errorHandler = -1;
Guido van Rossumd57fd912000-03-10 22:53:23 +00008125
Benjamin Petersonbac79492012-01-14 13:34:47 -05008126 if (PyUnicode_READY(unicode) == -1)
Martin v. Löwis23e275b2011-11-02 18:02:51 +01008127 return NULL;
8128 size = PyUnicode_GET_LENGTH(unicode);
8129
Guido van Rossumd57fd912000-03-10 22:53:23 +00008130 /* Default to Latin-1 */
8131 if (mapping == NULL)
Martin v. Löwis23e275b2011-11-02 18:02:51 +01008132 return unicode_encode_ucs1(unicode, errors, 256);
Guido van Rossumd57fd912000-03-10 22:53:23 +00008133
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008134 /* allocate enough for a simple encoding without
8135 replacements, if we need more, we'll resize */
Christian Heimes72b710a2008-05-26 13:28:38 +00008136 res = PyBytes_FromStringAndSize(NULL, size);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008137 if (res == NULL)
8138 goto onError;
Marc-André Lemburgb7520772000-08-14 11:29:19 +00008139 if (size == 0)
Benjamin Peterson29060642009-01-31 22:14:21 +00008140 return res;
Guido van Rossumd57fd912000-03-10 22:53:23 +00008141
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008142 while (inpos<size) {
Martin v. Löwis23e275b2011-11-02 18:02:51 +01008143 Py_UCS4 ch = PyUnicode_READ_CHAR(unicode, inpos);
Benjamin Peterson29060642009-01-31 22:14:21 +00008144 /* try to encode it */
Martin v. Löwis23e275b2011-11-02 18:02:51 +01008145 charmapencode_result x = charmapencode_output(ch, mapping, &res, &respos);
Benjamin Peterson29060642009-01-31 22:14:21 +00008146 if (x==enc_EXCEPTION) /* error */
8147 goto onError;
8148 if (x==enc_FAILED) { /* unencodable character */
Martin v. Löwis23e275b2011-11-02 18:02:51 +01008149 if (charmap_encoding_error(unicode, &inpos, mapping,
Benjamin Peterson29060642009-01-31 22:14:21 +00008150 &exc,
8151 &known_errorHandler, &errorHandler, errors,
8152 &res, &respos)) {
8153 goto onError;
8154 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00008155 }
Benjamin Peterson29060642009-01-31 22:14:21 +00008156 else
8157 /* done with this character => adjust input position */
8158 ++inpos;
Guido van Rossumd57fd912000-03-10 22:53:23 +00008159 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00008160
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008161 /* Resize if we allocated to much */
Christian Heimes72b710a2008-05-26 13:28:38 +00008162 if (respos<PyBytes_GET_SIZE(res))
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00008163 if (_PyBytes_Resize(&res, respos) < 0)
8164 goto onError;
Guido van Rossum98297ee2007-11-06 21:34:58 +00008165
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008166 Py_XDECREF(exc);
8167 Py_XDECREF(errorHandler);
8168 return res;
8169
Benjamin Peterson29060642009-01-31 22:14:21 +00008170 onError:
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008171 Py_XDECREF(res);
8172 Py_XDECREF(exc);
8173 Py_XDECREF(errorHandler);
Guido van Rossumd57fd912000-03-10 22:53:23 +00008174 return NULL;
8175}
8176
Martin v. Löwis23e275b2011-11-02 18:02:51 +01008177/* Deprecated */
8178PyObject *
8179PyUnicode_EncodeCharmap(const Py_UNICODE *p,
8180 Py_ssize_t size,
8181 PyObject *mapping,
8182 const char *errors)
8183{
8184 PyObject *result;
8185 PyObject *unicode = PyUnicode_FromUnicode(p, size);
8186 if (unicode == NULL)
8187 return NULL;
8188 result = _PyUnicode_EncodeCharmap(unicode, mapping, errors);
8189 Py_DECREF(unicode);
Victor Stinnerfc026c92011-11-04 00:24:51 +01008190 return result;
Martin v. Löwis23e275b2011-11-02 18:02:51 +01008191}
8192
Alexander Belopolsky40018472011-02-26 01:02:56 +00008193PyObject *
8194PyUnicode_AsCharmapString(PyObject *unicode,
8195 PyObject *mapping)
Guido van Rossumd57fd912000-03-10 22:53:23 +00008196{
8197 if (!PyUnicode_Check(unicode) || mapping == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +00008198 PyErr_BadArgument();
8199 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00008200 }
Martin v. Löwis23e275b2011-11-02 18:02:51 +01008201 return _PyUnicode_EncodeCharmap(unicode, mapping, NULL);
Guido van Rossumd57fd912000-03-10 22:53:23 +00008202}
8203
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008204/* create or adjust a UnicodeTranslateError */
Alexander Belopolsky40018472011-02-26 01:02:56 +00008205static void
8206make_translate_exception(PyObject **exceptionObject,
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008207 PyObject *unicode,
Alexander Belopolsky40018472011-02-26 01:02:56 +00008208 Py_ssize_t startpos, Py_ssize_t endpos,
8209 const char *reason)
Guido van Rossumd57fd912000-03-10 22:53:23 +00008210{
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008211 if (*exceptionObject == NULL) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008212 *exceptionObject = _PyUnicodeTranslateError_Create(
8213 unicode, startpos, endpos, reason);
Guido van Rossumd57fd912000-03-10 22:53:23 +00008214 }
8215 else {
Benjamin Peterson29060642009-01-31 22:14:21 +00008216 if (PyUnicodeTranslateError_SetStart(*exceptionObject, startpos))
8217 goto onError;
8218 if (PyUnicodeTranslateError_SetEnd(*exceptionObject, endpos))
8219 goto onError;
8220 if (PyUnicodeTranslateError_SetReason(*exceptionObject, reason))
8221 goto onError;
8222 return;
8223 onError:
8224 Py_DECREF(*exceptionObject);
8225 *exceptionObject = NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00008226 }
8227}
8228
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008229/* raises a UnicodeTranslateError */
Alexander Belopolsky40018472011-02-26 01:02:56 +00008230static void
8231raise_translate_exception(PyObject **exceptionObject,
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008232 PyObject *unicode,
Alexander Belopolsky40018472011-02-26 01:02:56 +00008233 Py_ssize_t startpos, Py_ssize_t endpos,
8234 const char *reason)
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008235{
8236 make_translate_exception(exceptionObject,
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008237 unicode, startpos, endpos, reason);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008238 if (*exceptionObject != NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00008239 PyCodec_StrictErrors(*exceptionObject);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008240}
8241
8242/* error handling callback helper:
8243 build arguments, call the callback and check the arguments,
8244 put the result into newpos and return the replacement string, which
8245 has to be freed by the caller */
Alexander Belopolsky40018472011-02-26 01:02:56 +00008246static PyObject *
8247unicode_translate_call_errorhandler(const char *errors,
8248 PyObject **errorHandler,
8249 const char *reason,
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008250 PyObject *unicode, PyObject **exceptionObject,
Alexander Belopolsky40018472011-02-26 01:02:56 +00008251 Py_ssize_t startpos, Py_ssize_t endpos,
8252 Py_ssize_t *newpos)
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008253{
Benjamin Peterson142957c2008-07-04 19:55:29 +00008254 static char *argparse = "O!n;translating error handler must return (str, int) tuple";
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008255
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00008256 Py_ssize_t i_newpos;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008257 PyObject *restuple;
8258 PyObject *resunicode;
8259
8260 if (*errorHandler == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +00008261 *errorHandler = PyCodec_LookupError(errors);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008262 if (*errorHandler == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00008263 return NULL;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008264 }
8265
8266 make_translate_exception(exceptionObject,
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008267 unicode, startpos, endpos, reason);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008268 if (*exceptionObject == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00008269 return NULL;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008270
8271 restuple = PyObject_CallFunctionObjArgs(
Benjamin Peterson29060642009-01-31 22:14:21 +00008272 *errorHandler, *exceptionObject, NULL);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008273 if (restuple == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00008274 return NULL;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008275 if (!PyTuple_Check(restuple)) {
Benjamin Petersond75fcb42009-02-19 04:22:03 +00008276 PyErr_SetString(PyExc_TypeError, &argparse[4]);
Benjamin Peterson29060642009-01-31 22:14:21 +00008277 Py_DECREF(restuple);
8278 return NULL;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008279 }
8280 if (!PyArg_ParseTuple(restuple, argparse, &PyUnicode_Type,
Benjamin Peterson29060642009-01-31 22:14:21 +00008281 &resunicode, &i_newpos)) {
8282 Py_DECREF(restuple);
8283 return NULL;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008284 }
Martin v. Löwis18e16552006-02-15 17:27:45 +00008285 if (i_newpos<0)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008286 *newpos = PyUnicode_GET_LENGTH(unicode)+i_newpos;
Martin v. Löwis18e16552006-02-15 17:27:45 +00008287 else
8288 *newpos = i_newpos;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008289 if (*newpos<0 || *newpos>PyUnicode_GET_LENGTH(unicode)) {
Benjamin Peterson29060642009-01-31 22:14:21 +00008290 PyErr_Format(PyExc_IndexError, "position %zd from error handler out of bounds", *newpos);
8291 Py_DECREF(restuple);
8292 return NULL;
Walter Dörwald2e0b18a2003-01-31 17:19:08 +00008293 }
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008294 Py_INCREF(resunicode);
8295 Py_DECREF(restuple);
8296 return resunicode;
8297}
8298
8299/* Lookup the character ch in the mapping and put the result in result,
8300 which must be decrefed by the caller.
8301 Return 0 on success, -1 on error */
Alexander Belopolsky40018472011-02-26 01:02:56 +00008302static int
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008303charmaptranslate_lookup(Py_UCS4 c, PyObject *mapping, PyObject **result)
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008304{
Christian Heimes217cfd12007-12-02 14:31:20 +00008305 PyObject *w = PyLong_FromLong((long)c);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008306 PyObject *x;
8307
8308 if (w == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00008309 return -1;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008310 x = PyObject_GetItem(mapping, w);
8311 Py_DECREF(w);
8312 if (x == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +00008313 if (PyErr_ExceptionMatches(PyExc_LookupError)) {
8314 /* No mapping found means: use 1:1 mapping. */
8315 PyErr_Clear();
8316 *result = NULL;
8317 return 0;
8318 } else
8319 return -1;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008320 }
8321 else if (x == Py_None) {
Benjamin Peterson29060642009-01-31 22:14:21 +00008322 *result = x;
8323 return 0;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008324 }
Christian Heimes217cfd12007-12-02 14:31:20 +00008325 else if (PyLong_Check(x)) {
Benjamin Peterson29060642009-01-31 22:14:21 +00008326 long value = PyLong_AS_LONG(x);
8327 long max = PyUnicode_GetMax();
8328 if (value < 0 || value > max) {
8329 PyErr_Format(PyExc_TypeError,
Guido van Rossum5a2f7e602007-10-24 21:13:09 +00008330 "character mapping must be in range(0x%x)", max+1);
Benjamin Peterson29060642009-01-31 22:14:21 +00008331 Py_DECREF(x);
8332 return -1;
8333 }
8334 *result = x;
8335 return 0;
8336 }
8337 else if (PyUnicode_Check(x)) {
8338 *result = x;
8339 return 0;
8340 }
8341 else {
8342 /* wrong return value */
8343 PyErr_SetString(PyExc_TypeError,
8344 "character mapping must return integer, None or str");
Benjamin Peterson14339b62009-01-31 16:36:08 +00008345 Py_DECREF(x);
8346 return -1;
8347 }
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008348}
8349/* ensure that *outobj is at least requiredsize characters long,
Benjamin Peterson29060642009-01-31 22:14:21 +00008350 if not reallocate and adjust various state variables.
8351 Return 0 on success, -1 on error */
Alexander Belopolsky40018472011-02-26 01:02:56 +00008352static int
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008353charmaptranslate_makespace(Py_UCS4 **outobj, Py_ssize_t *psize,
Benjamin Peterson29060642009-01-31 22:14:21 +00008354 Py_ssize_t requiredsize)
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008355{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008356 Py_ssize_t oldsize = *psize;
Kristjan Valur Jonsson85634d72012-05-31 09:37:31 +00008357 Py_UCS4 *new_outobj;
Walter Dörwald4894c302003-10-24 14:25:28 +00008358 if (requiredsize > oldsize) {
Benjamin Peterson29060642009-01-31 22:14:21 +00008359 /* exponentially overallocate to minimize reallocations */
8360 if (requiredsize < 2 * oldsize)
8361 requiredsize = 2 * oldsize;
Kristjan Valur Jonsson85634d72012-05-31 09:37:31 +00008362 new_outobj = PyMem_Realloc(*outobj, requiredsize * sizeof(Py_UCS4));
8363 if (new_outobj == 0)
Benjamin Peterson29060642009-01-31 22:14:21 +00008364 return -1;
Kristjan Valur Jonsson85634d72012-05-31 09:37:31 +00008365 *outobj = new_outobj;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008366 *psize = requiredsize;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008367 }
8368 return 0;
8369}
8370/* lookup the character, put the result in the output string and adjust
8371 various state variables. Return a new reference to the object that
8372 was put in the output buffer in *result, or Py_None, if the mapping was
8373 undefined (in which case no character was written).
8374 The called must decref result.
8375 Return 0 on success, -1 on error. */
Alexander Belopolsky40018472011-02-26 01:02:56 +00008376static int
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008377charmaptranslate_output(PyObject *input, Py_ssize_t ipos,
8378 PyObject *mapping, Py_UCS4 **output,
8379 Py_ssize_t *osize, Py_ssize_t *opos,
Alexander Belopolsky40018472011-02-26 01:02:56 +00008380 PyObject **res)
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008381{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008382 Py_UCS4 curinp = PyUnicode_READ_CHAR(input, ipos);
8383 if (charmaptranslate_lookup(curinp, mapping, res))
Benjamin Peterson29060642009-01-31 22:14:21 +00008384 return -1;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008385 if (*res==NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +00008386 /* not found => default to 1:1 mapping */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008387 (*output)[(*opos)++] = curinp;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008388 }
8389 else if (*res==Py_None)
Benjamin Peterson29060642009-01-31 22:14:21 +00008390 ;
Christian Heimes217cfd12007-12-02 14:31:20 +00008391 else if (PyLong_Check(*res)) {
Benjamin Peterson29060642009-01-31 22:14:21 +00008392 /* no overflow check, because we know that the space is enough */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008393 (*output)[(*opos)++] = (Py_UCS4)PyLong_AS_LONG(*res);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008394 }
8395 else if (PyUnicode_Check(*res)) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008396 Py_ssize_t repsize;
8397 if (PyUnicode_READY(*res) == -1)
8398 return -1;
8399 repsize = PyUnicode_GET_LENGTH(*res);
Benjamin Peterson29060642009-01-31 22:14:21 +00008400 if (repsize==1) {
8401 /* no overflow check, because we know that the space is enough */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008402 (*output)[(*opos)++] = PyUnicode_READ_CHAR(*res, 0);
Benjamin Peterson29060642009-01-31 22:14:21 +00008403 }
8404 else if (repsize!=0) {
8405 /* more than one character */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008406 Py_ssize_t requiredsize = *opos +
8407 (PyUnicode_GET_LENGTH(input) - ipos) +
Benjamin Peterson29060642009-01-31 22:14:21 +00008408 repsize - 1;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008409 Py_ssize_t i;
8410 if (charmaptranslate_makespace(output, osize, requiredsize))
Benjamin Peterson29060642009-01-31 22:14:21 +00008411 return -1;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008412 for(i = 0; i < repsize; i++)
8413 (*output)[(*opos)++] = PyUnicode_READ_CHAR(*res, i);
Benjamin Peterson29060642009-01-31 22:14:21 +00008414 }
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008415 }
8416 else
Benjamin Peterson29060642009-01-31 22:14:21 +00008417 return -1;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008418 return 0;
8419}
8420
Alexander Belopolsky40018472011-02-26 01:02:56 +00008421PyObject *
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008422_PyUnicode_TranslateCharmap(PyObject *input,
8423 PyObject *mapping,
8424 const char *errors)
Guido van Rossumd57fd912000-03-10 22:53:23 +00008425{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008426 /* input object */
8427 char *idata;
8428 Py_ssize_t size, i;
8429 int kind;
8430 /* output buffer */
8431 Py_UCS4 *output = NULL;
8432 Py_ssize_t osize;
8433 PyObject *res;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008434 /* current output position */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008435 Py_ssize_t opos;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008436 char *reason = "character maps to <undefined>";
8437 PyObject *errorHandler = NULL;
8438 PyObject *exc = NULL;
8439 /* the following variable is used for caching string comparisons
8440 * -1=not initialized, 0=unknown, 1=strict, 2=replace,
8441 * 3=ignore, 4=xmlcharrefreplace */
8442 int known_errorHandler = -1;
8443
Guido van Rossumd57fd912000-03-10 22:53:23 +00008444 if (mapping == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +00008445 PyErr_BadArgument();
8446 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00008447 }
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008448
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008449 if (PyUnicode_READY(input) == -1)
8450 return NULL;
8451 idata = (char*)PyUnicode_DATA(input);
8452 kind = PyUnicode_KIND(input);
8453 size = PyUnicode_GET_LENGTH(input);
8454 i = 0;
8455
8456 if (size == 0) {
8457 Py_INCREF(input);
8458 return input;
8459 }
8460
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008461 /* allocate enough for a simple 1:1 translation without
8462 replacements, if we need more, we'll resize */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008463 osize = size;
8464 output = PyMem_Malloc(osize * sizeof(Py_UCS4));
8465 opos = 0;
8466 if (output == NULL) {
8467 PyErr_NoMemory();
Benjamin Peterson29060642009-01-31 22:14:21 +00008468 goto onError;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008469 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00008470
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008471 while (i<size) {
Benjamin Peterson29060642009-01-31 22:14:21 +00008472 /* try to encode it */
8473 PyObject *x = NULL;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008474 if (charmaptranslate_output(input, i, mapping,
8475 &output, &osize, &opos, &x)) {
Benjamin Peterson29060642009-01-31 22:14:21 +00008476 Py_XDECREF(x);
8477 goto onError;
8478 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00008479 Py_XDECREF(x);
Benjamin Peterson29060642009-01-31 22:14:21 +00008480 if (x!=Py_None) /* it worked => adjust input pointer */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008481 ++i;
Benjamin Peterson29060642009-01-31 22:14:21 +00008482 else { /* untranslatable character */
8483 PyObject *repunicode = NULL; /* initialize to prevent gcc warning */
8484 Py_ssize_t repsize;
8485 Py_ssize_t newpos;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008486 Py_ssize_t uni2;
Benjamin Peterson29060642009-01-31 22:14:21 +00008487 /* startpos for collecting untranslatable chars */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008488 Py_ssize_t collstart = i;
8489 Py_ssize_t collend = i+1;
8490 Py_ssize_t coll;
Guido van Rossumd57fd912000-03-10 22:53:23 +00008491
Benjamin Peterson29060642009-01-31 22:14:21 +00008492 /* find all untranslatable characters */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008493 while (collend < size) {
8494 if (charmaptranslate_lookup(PyUnicode_READ(kind,idata, collend), mapping, &x))
Benjamin Peterson29060642009-01-31 22:14:21 +00008495 goto onError;
8496 Py_XDECREF(x);
8497 if (x!=Py_None)
8498 break;
8499 ++collend;
8500 }
8501 /* cache callback name lookup
8502 * (if not done yet, i.e. it's the first error) */
8503 if (known_errorHandler==-1) {
8504 if ((errors==NULL) || (!strcmp(errors, "strict")))
8505 known_errorHandler = 1;
8506 else if (!strcmp(errors, "replace"))
8507 known_errorHandler = 2;
8508 else if (!strcmp(errors, "ignore"))
8509 known_errorHandler = 3;
8510 else if (!strcmp(errors, "xmlcharrefreplace"))
8511 known_errorHandler = 4;
8512 else
8513 known_errorHandler = 0;
8514 }
8515 switch (known_errorHandler) {
8516 case 1: /* strict */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008517 raise_translate_exception(&exc, input, collstart,
8518 collend, reason);
Benjamin Peterson14339b62009-01-31 16:36:08 +00008519 goto onError;
Benjamin Peterson29060642009-01-31 22:14:21 +00008520 case 2: /* replace */
8521 /* No need to check for space, this is a 1:1 replacement */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008522 for (coll = collstart; coll<collend; coll++)
8523 output[opos++] = '?';
Benjamin Peterson29060642009-01-31 22:14:21 +00008524 /* fall through */
8525 case 3: /* ignore */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008526 i = collend;
Benjamin Peterson29060642009-01-31 22:14:21 +00008527 break;
8528 case 4: /* xmlcharrefreplace */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008529 /* generate replacement (temporarily (mis)uses i) */
8530 for (i = collstart; i < collend; ++i) {
Benjamin Peterson29060642009-01-31 22:14:21 +00008531 char buffer[2+29+1+1];
8532 char *cp;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008533 sprintf(buffer, "&#%d;", PyUnicode_READ(kind, idata, i));
8534 if (charmaptranslate_makespace(&output, &osize,
8535 opos+strlen(buffer)+(size-collend)))
Benjamin Peterson29060642009-01-31 22:14:21 +00008536 goto onError;
8537 for (cp = buffer; *cp; ++cp)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008538 output[opos++] = *cp;
Benjamin Peterson29060642009-01-31 22:14:21 +00008539 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008540 i = collend;
Benjamin Peterson29060642009-01-31 22:14:21 +00008541 break;
8542 default:
8543 repunicode = unicode_translate_call_errorhandler(errors, &errorHandler,
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008544 reason, input, &exc,
8545 collstart, collend, &newpos);
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01008546 if (repunicode == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00008547 goto onError;
Benjamin Peterson9ca3ffa2012-01-01 16:04:29 -06008548 if (PyUnicode_READY(repunicode) == -1) {
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01008549 Py_DECREF(repunicode);
8550 goto onError;
8551 }
Benjamin Peterson29060642009-01-31 22:14:21 +00008552 /* generate replacement */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008553 repsize = PyUnicode_GET_LENGTH(repunicode);
8554 if (charmaptranslate_makespace(&output, &osize,
8555 opos+repsize+(size-collend))) {
Benjamin Peterson29060642009-01-31 22:14:21 +00008556 Py_DECREF(repunicode);
8557 goto onError;
8558 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008559 for (uni2 = 0; repsize-->0; ++uni2)
8560 output[opos++] = PyUnicode_READ_CHAR(repunicode, uni2);
8561 i = newpos;
Benjamin Peterson29060642009-01-31 22:14:21 +00008562 Py_DECREF(repunicode);
Benjamin Peterson14339b62009-01-31 16:36:08 +00008563 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00008564 }
8565 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008566 res = PyUnicode_FromKindAndData(PyUnicode_4BYTE_KIND, output, opos);
8567 if (!res)
8568 goto onError;
8569 PyMem_Free(output);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008570 Py_XDECREF(exc);
8571 Py_XDECREF(errorHandler);
8572 return res;
Guido van Rossumd57fd912000-03-10 22:53:23 +00008573
Benjamin Peterson29060642009-01-31 22:14:21 +00008574 onError:
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008575 PyMem_Free(output);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008576 Py_XDECREF(exc);
8577 Py_XDECREF(errorHandler);
Guido van Rossumd57fd912000-03-10 22:53:23 +00008578 return NULL;
8579}
8580
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008581/* Deprecated. Use PyUnicode_Translate instead. */
8582PyObject *
8583PyUnicode_TranslateCharmap(const Py_UNICODE *p,
8584 Py_ssize_t size,
8585 PyObject *mapping,
8586 const char *errors)
8587{
Christian Heimes5f520f42012-09-11 14:03:25 +02008588 PyObject *result;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008589 PyObject *unicode = PyUnicode_FromUnicode(p, size);
8590 if (!unicode)
8591 return NULL;
Christian Heimes5f520f42012-09-11 14:03:25 +02008592 result = _PyUnicode_TranslateCharmap(unicode, mapping, errors);
8593 Py_DECREF(unicode);
8594 return result;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008595}
8596
Alexander Belopolsky40018472011-02-26 01:02:56 +00008597PyObject *
8598PyUnicode_Translate(PyObject *str,
8599 PyObject *mapping,
8600 const char *errors)
Guido van Rossumd57fd912000-03-10 22:53:23 +00008601{
8602 PyObject *result;
Tim Petersced69f82003-09-16 20:30:58 +00008603
Guido van Rossumd57fd912000-03-10 22:53:23 +00008604 str = PyUnicode_FromObject(str);
8605 if (str == NULL)
Christian Heimes5f520f42012-09-11 14:03:25 +02008606 return NULL;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008607 result = _PyUnicode_TranslateCharmap(str, mapping, errors);
Guido van Rossumd57fd912000-03-10 22:53:23 +00008608 Py_DECREF(str);
8609 return result;
Guido van Rossumd57fd912000-03-10 22:53:23 +00008610}
Tim Petersced69f82003-09-16 20:30:58 +00008611
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008612static Py_UCS4
Victor Stinner9310abb2011-10-05 00:59:23 +02008613fix_decimal_and_space_to_ascii(PyObject *self)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008614{
8615 /* No need to call PyUnicode_READY(self) because this function is only
8616 called as a callback from fixup() which does it already. */
8617 const Py_ssize_t len = PyUnicode_GET_LENGTH(self);
8618 const int kind = PyUnicode_KIND(self);
8619 void *data = PyUnicode_DATA(self);
Victor Stinnere6abb482012-05-02 01:15:40 +02008620 Py_UCS4 maxchar = 127, ch, fixed;
Benjamin Peterson821e4cf2012-01-12 15:40:18 -05008621 int modified = 0;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008622 Py_ssize_t i;
8623
8624 for (i = 0; i < len; ++i) {
8625 ch = PyUnicode_READ(kind, data, i);
8626 fixed = 0;
8627 if (ch > 127) {
8628 if (Py_UNICODE_ISSPACE(ch))
8629 fixed = ' ';
8630 else {
8631 const int decimal = Py_UNICODE_TODECIMAL(ch);
8632 if (decimal >= 0)
8633 fixed = '0' + decimal;
8634 }
8635 if (fixed != 0) {
Benjamin Peterson821e4cf2012-01-12 15:40:18 -05008636 modified = 1;
Victor Stinnere6abb482012-05-02 01:15:40 +02008637 maxchar = MAX_MAXCHAR(maxchar, fixed);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008638 PyUnicode_WRITE(kind, data, i, fixed);
8639 }
Victor Stinnere6abb482012-05-02 01:15:40 +02008640 else
8641 maxchar = MAX_MAXCHAR(maxchar, ch);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008642 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008643 }
8644
Benjamin Peterson821e4cf2012-01-12 15:40:18 -05008645 return (modified) ? maxchar : 0;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008646}
8647
8648PyObject *
8649_PyUnicode_TransformDecimalAndSpaceToASCII(PyObject *unicode)
8650{
8651 if (!PyUnicode_Check(unicode)) {
8652 PyErr_BadInternalCall();
8653 return NULL;
8654 }
8655 if (PyUnicode_READY(unicode) == -1)
8656 return NULL;
8657 if (PyUnicode_MAX_CHAR_VALUE(unicode) <= 127) {
8658 /* If the string is already ASCII, just return the same string */
8659 Py_INCREF(unicode);
8660 return unicode;
8661 }
Victor Stinner9310abb2011-10-05 00:59:23 +02008662 return fixup(unicode, fix_decimal_and_space_to_ascii);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008663}
8664
Alexander Belopolsky942af5a2010-12-04 03:38:46 +00008665PyObject *
8666PyUnicode_TransformDecimalToASCII(Py_UNICODE *s,
8667 Py_ssize_t length)
8668{
Victor Stinnerf0124502011-11-21 23:12:56 +01008669 PyObject *decimal;
Alexander Belopolsky942af5a2010-12-04 03:38:46 +00008670 Py_ssize_t i;
Victor Stinnerf0124502011-11-21 23:12:56 +01008671 Py_UCS4 maxchar;
8672 enum PyUnicode_Kind kind;
8673 void *data;
8674
Victor Stinner99d7ad02012-02-22 13:37:39 +01008675 maxchar = 127;
Alexander Belopolsky942af5a2010-12-04 03:38:46 +00008676 for (i = 0; i < length; i++) {
Victor Stinnerf0124502011-11-21 23:12:56 +01008677 Py_UNICODE ch = s[i];
Alexander Belopolsky942af5a2010-12-04 03:38:46 +00008678 if (ch > 127) {
8679 int decimal = Py_UNICODE_TODECIMAL(ch);
8680 if (decimal >= 0)
Victor Stinnerf0124502011-11-21 23:12:56 +01008681 ch = '0' + decimal;
Victor Stinnere6abb482012-05-02 01:15:40 +02008682 maxchar = MAX_MAXCHAR(maxchar, ch);
Alexander Belopolsky942af5a2010-12-04 03:38:46 +00008683 }
8684 }
Victor Stinnerf0124502011-11-21 23:12:56 +01008685
8686 /* Copy to a new string */
8687 decimal = PyUnicode_New(length, maxchar);
8688 if (decimal == NULL)
8689 return decimal;
8690 kind = PyUnicode_KIND(decimal);
8691 data = PyUnicode_DATA(decimal);
8692 /* Iterate over code points */
8693 for (i = 0; i < length; i++) {
8694 Py_UNICODE ch = s[i];
8695 if (ch > 127) {
8696 int decimal = Py_UNICODE_TODECIMAL(ch);
8697 if (decimal >= 0)
8698 ch = '0' + decimal;
8699 }
8700 PyUnicode_WRITE(kind, data, i, ch);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008701 }
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01008702 return unicode_result(decimal);
Alexander Belopolsky942af5a2010-12-04 03:38:46 +00008703}
Guido van Rossum9e896b32000-04-05 20:11:21 +00008704/* --- Decimal Encoder ---------------------------------------------------- */
8705
Alexander Belopolsky40018472011-02-26 01:02:56 +00008706int
8707PyUnicode_EncodeDecimal(Py_UNICODE *s,
8708 Py_ssize_t length,
8709 char *output,
8710 const char *errors)
Guido van Rossum9e896b32000-04-05 20:11:21 +00008711{
Martin v. Löwis23e275b2011-11-02 18:02:51 +01008712 PyObject *unicode;
Victor Stinner6345be92011-11-25 20:09:01 +01008713 Py_ssize_t i;
Victor Stinner42bf7752011-11-21 22:52:58 +01008714 enum PyUnicode_Kind kind;
8715 void *data;
Guido van Rossum9e896b32000-04-05 20:11:21 +00008716
8717 if (output == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +00008718 PyErr_BadArgument();
8719 return -1;
Guido van Rossum9e896b32000-04-05 20:11:21 +00008720 }
8721
Victor Stinner42bf7752011-11-21 22:52:58 +01008722 unicode = PyUnicode_FromUnicode(s, length);
8723 if (unicode == NULL)
8724 return -1;
8725
Benjamin Petersonbac79492012-01-14 13:34:47 -05008726 if (PyUnicode_READY(unicode) == -1) {
Victor Stinner6345be92011-11-25 20:09:01 +01008727 Py_DECREF(unicode);
8728 return -1;
8729 }
Victor Stinner42bf7752011-11-21 22:52:58 +01008730 kind = PyUnicode_KIND(unicode);
8731 data = PyUnicode_DATA(unicode);
8732
Victor Stinnerb84d7232011-11-22 01:50:07 +01008733 for (i=0; i < length; ) {
Victor Stinner6345be92011-11-25 20:09:01 +01008734 PyObject *exc;
8735 Py_UCS4 ch;
Benjamin Peterson29060642009-01-31 22:14:21 +00008736 int decimal;
Victor Stinner6345be92011-11-25 20:09:01 +01008737 Py_ssize_t startpos;
8738
8739 ch = PyUnicode_READ(kind, data, i);
Tim Petersced69f82003-09-16 20:30:58 +00008740
Benjamin Peterson29060642009-01-31 22:14:21 +00008741 if (Py_UNICODE_ISSPACE(ch)) {
Benjamin Peterson14339b62009-01-31 16:36:08 +00008742 *output++ = ' ';
Victor Stinnerb84d7232011-11-22 01:50:07 +01008743 i++;
Benjamin Peterson29060642009-01-31 22:14:21 +00008744 continue;
Benjamin Peterson14339b62009-01-31 16:36:08 +00008745 }
Benjamin Peterson29060642009-01-31 22:14:21 +00008746 decimal = Py_UNICODE_TODECIMAL(ch);
8747 if (decimal >= 0) {
8748 *output++ = '0' + decimal;
Victor Stinnerb84d7232011-11-22 01:50:07 +01008749 i++;
Benjamin Peterson29060642009-01-31 22:14:21 +00008750 continue;
8751 }
8752 if (0 < ch && ch < 256) {
8753 *output++ = (char)ch;
Victor Stinnerb84d7232011-11-22 01:50:07 +01008754 i++;
Benjamin Peterson29060642009-01-31 22:14:21 +00008755 continue;
8756 }
Victor Stinner6345be92011-11-25 20:09:01 +01008757
Victor Stinner42bf7752011-11-21 22:52:58 +01008758 startpos = i;
Victor Stinner6345be92011-11-25 20:09:01 +01008759 exc = NULL;
8760 raise_encode_exception(&exc, "decimal", unicode,
8761 startpos, startpos+1,
8762 "invalid decimal Unicode string");
8763 Py_XDECREF(exc);
8764 Py_DECREF(unicode);
8765 return -1;
Guido van Rossum9e896b32000-04-05 20:11:21 +00008766 }
8767 /* 0-terminate the output string */
8768 *output++ = '\0';
Victor Stinner42bf7752011-11-21 22:52:58 +01008769 Py_DECREF(unicode);
Guido van Rossum9e896b32000-04-05 20:11:21 +00008770 return 0;
Guido van Rossum9e896b32000-04-05 20:11:21 +00008771}
8772
Guido van Rossumd57fd912000-03-10 22:53:23 +00008773/* --- Helpers ------------------------------------------------------------ */
8774
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008775static Py_ssize_t
Victor Stinner794d5672011-10-10 03:21:36 +02008776any_find_slice(int direction, PyObject* s1, PyObject* s2,
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008777 Py_ssize_t start,
8778 Py_ssize_t end)
8779{
8780 int kind1, kind2, kind;
8781 void *buf1, *buf2;
8782 Py_ssize_t len1, len2, result;
8783
8784 kind1 = PyUnicode_KIND(s1);
8785 kind2 = PyUnicode_KIND(s2);
8786 kind = kind1 > kind2 ? kind1 : kind2;
8787 buf1 = PyUnicode_DATA(s1);
8788 buf2 = PyUnicode_DATA(s2);
8789 if (kind1 != kind)
8790 buf1 = _PyUnicode_AsKind(s1, kind);
8791 if (!buf1)
8792 return -2;
8793 if (kind2 != kind)
8794 buf2 = _PyUnicode_AsKind(s2, kind);
8795 if (!buf2) {
8796 if (kind1 != kind) PyMem_Free(buf1);
8797 return -2;
8798 }
8799 len1 = PyUnicode_GET_LENGTH(s1);
8800 len2 = PyUnicode_GET_LENGTH(s2);
8801
Victor Stinner794d5672011-10-10 03:21:36 +02008802 if (direction > 0) {
Benjamin Petersonead6b532011-12-20 17:23:42 -06008803 switch (kind) {
Victor Stinner794d5672011-10-10 03:21:36 +02008804 case PyUnicode_1BYTE_KIND:
8805 if (PyUnicode_IS_ASCII(s1) && PyUnicode_IS_ASCII(s2))
8806 result = asciilib_find_slice(buf1, len1, buf2, len2, start, end);
8807 else
8808 result = ucs1lib_find_slice(buf1, len1, buf2, len2, start, end);
8809 break;
8810 case PyUnicode_2BYTE_KIND:
8811 result = ucs2lib_find_slice(buf1, len1, buf2, len2, start, end);
8812 break;
8813 case PyUnicode_4BYTE_KIND:
8814 result = ucs4lib_find_slice(buf1, len1, buf2, len2, start, end);
8815 break;
8816 default:
8817 assert(0); result = -2;
8818 }
8819 }
8820 else {
Benjamin Petersonead6b532011-12-20 17:23:42 -06008821 switch (kind) {
Victor Stinner794d5672011-10-10 03:21:36 +02008822 case PyUnicode_1BYTE_KIND:
8823 if (PyUnicode_IS_ASCII(s1) && PyUnicode_IS_ASCII(s2))
8824 result = asciilib_rfind_slice(buf1, len1, buf2, len2, start, end);
8825 else
8826 result = ucs1lib_rfind_slice(buf1, len1, buf2, len2, start, end);
8827 break;
8828 case PyUnicode_2BYTE_KIND:
8829 result = ucs2lib_rfind_slice(buf1, len1, buf2, len2, start, end);
8830 break;
8831 case PyUnicode_4BYTE_KIND:
8832 result = ucs4lib_rfind_slice(buf1, len1, buf2, len2, start, end);
8833 break;
8834 default:
8835 assert(0); result = -2;
8836 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008837 }
8838
8839 if (kind1 != kind)
8840 PyMem_Free(buf1);
8841 if (kind2 != kind)
8842 PyMem_Free(buf2);
8843
8844 return result;
8845}
8846
8847Py_ssize_t
Victor Stinner41a863c2012-02-24 00:37:51 +01008848_PyUnicode_InsertThousandsGrouping(
8849 PyObject *unicode, Py_ssize_t index,
8850 Py_ssize_t n_buffer,
8851 void *digits, Py_ssize_t n_digits,
8852 Py_ssize_t min_width,
8853 const char *grouping, PyObject *thousands_sep,
8854 Py_UCS4 *maxchar)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008855{
Victor Stinner41a863c2012-02-24 00:37:51 +01008856 unsigned int kind, thousands_sep_kind;
Antoine Pitrou842c0f12012-02-24 13:30:46 +01008857 char *data, *thousands_sep_data;
Victor Stinner41a863c2012-02-24 00:37:51 +01008858 Py_ssize_t thousands_sep_len;
8859 Py_ssize_t len;
8860
8861 if (unicode != NULL) {
8862 kind = PyUnicode_KIND(unicode);
Antoine Pitrou842c0f12012-02-24 13:30:46 +01008863 data = (char *) PyUnicode_DATA(unicode) + index * kind;
Victor Stinner41a863c2012-02-24 00:37:51 +01008864 }
8865 else {
8866 kind = PyUnicode_1BYTE_KIND;
8867 data = NULL;
8868 }
8869 thousands_sep_kind = PyUnicode_KIND(thousands_sep);
8870 thousands_sep_data = PyUnicode_DATA(thousands_sep);
8871 thousands_sep_len = PyUnicode_GET_LENGTH(thousands_sep);
8872 if (unicode != NULL && thousands_sep_kind != kind) {
Victor Stinner90f50d42012-02-24 01:44:47 +01008873 if (thousands_sep_kind < kind) {
8874 thousands_sep_data = _PyUnicode_AsKind(thousands_sep, kind);
8875 if (!thousands_sep_data)
8876 return -1;
8877 }
8878 else {
8879 data = _PyUnicode_AsKind(unicode, thousands_sep_kind);
8880 if (!data)
8881 return -1;
8882 }
Victor Stinner41a863c2012-02-24 00:37:51 +01008883 }
8884
Benjamin Petersonead6b532011-12-20 17:23:42 -06008885 switch (kind) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008886 case PyUnicode_1BYTE_KIND:
Victor Stinnerc3cec782011-10-05 21:24:08 +02008887 if (unicode != NULL && PyUnicode_IS_ASCII(unicode))
Victor Stinner41a863c2012-02-24 00:37:51 +01008888 len = asciilib_InsertThousandsGrouping(
Antoine Pitrou842c0f12012-02-24 13:30:46 +01008889 (Py_UCS1 *) data, n_buffer, (Py_UCS1 *) digits, n_digits,
Victor Stinner41a863c2012-02-24 00:37:51 +01008890 min_width, grouping,
Antoine Pitrou842c0f12012-02-24 13:30:46 +01008891 (Py_UCS1 *) thousands_sep_data, thousands_sep_len);
Victor Stinnerc3cec782011-10-05 21:24:08 +02008892 else
Victor Stinner41a863c2012-02-24 00:37:51 +01008893 len = ucs1lib_InsertThousandsGrouping(
Victor Stinnerc3cec782011-10-05 21:24:08 +02008894 (Py_UCS1*)data, n_buffer, (Py_UCS1*)digits, n_digits,
Victor Stinner41a863c2012-02-24 00:37:51 +01008895 min_width, grouping,
Antoine Pitrou842c0f12012-02-24 13:30:46 +01008896 (Py_UCS1 *) 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_2BYTE_KIND:
Victor Stinner41a863c2012-02-24 00:37:51 +01008899 len = ucs2lib_InsertThousandsGrouping(
Antoine Pitrou842c0f12012-02-24 13:30:46 +01008900 (Py_UCS2 *) data, n_buffer, (Py_UCS2 *) digits, n_digits,
Victor Stinner41a863c2012-02-24 00:37:51 +01008901 min_width, grouping,
Antoine Pitrou842c0f12012-02-24 13:30:46 +01008902 (Py_UCS2 *) thousands_sep_data, thousands_sep_len);
Victor Stinner41a863c2012-02-24 00:37:51 +01008903 break;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008904 case PyUnicode_4BYTE_KIND:
Victor Stinner41a863c2012-02-24 00:37:51 +01008905 len = ucs4lib_InsertThousandsGrouping(
Antoine Pitrou842c0f12012-02-24 13:30:46 +01008906 (Py_UCS4 *) data, n_buffer, (Py_UCS4 *) digits, n_digits,
Victor Stinner41a863c2012-02-24 00:37:51 +01008907 min_width, grouping,
Antoine Pitrou842c0f12012-02-24 13:30:46 +01008908 (Py_UCS4 *) thousands_sep_data, thousands_sep_len);
Victor Stinner41a863c2012-02-24 00:37:51 +01008909 break;
8910 default:
8911 assert(0);
8912 return -1;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008913 }
Victor Stinner90f50d42012-02-24 01:44:47 +01008914 if (unicode != NULL && thousands_sep_kind != kind) {
8915 if (thousands_sep_kind < kind)
8916 PyMem_Free(thousands_sep_data);
8917 else
8918 PyMem_Free(data);
8919 }
Victor Stinner41a863c2012-02-24 00:37:51 +01008920 if (unicode == NULL) {
8921 *maxchar = 127;
8922 if (len != n_digits) {
Victor Stinnere6abb482012-05-02 01:15:40 +02008923 *maxchar = MAX_MAXCHAR(*maxchar,
8924 PyUnicode_MAX_CHAR_VALUE(thousands_sep));
Victor Stinner41a863c2012-02-24 00:37:51 +01008925 }
8926 }
8927 return len;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008928}
8929
8930
Thomas Wouters477c8d52006-05-27 19:21:47 +00008931/* helper macro to fixup start/end slice values */
Antoine Pitrouf2c54842010-01-13 08:07:53 +00008932#define ADJUST_INDICES(start, end, len) \
8933 if (end > len) \
8934 end = len; \
8935 else if (end < 0) { \
8936 end += len; \
8937 if (end < 0) \
8938 end = 0; \
8939 } \
8940 if (start < 0) { \
8941 start += len; \
8942 if (start < 0) \
8943 start = 0; \
8944 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00008945
Alexander Belopolsky40018472011-02-26 01:02:56 +00008946Py_ssize_t
8947PyUnicode_Count(PyObject *str,
8948 PyObject *substr,
8949 Py_ssize_t start,
8950 Py_ssize_t end)
Guido van Rossumd57fd912000-03-10 22:53:23 +00008951{
Martin v. Löwis18e16552006-02-15 17:27:45 +00008952 Py_ssize_t result;
Victor Stinner9db1a8b2011-10-23 20:04:37 +02008953 PyObject* str_obj;
8954 PyObject* sub_obj;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008955 int kind1, kind2, kind;
8956 void *buf1 = NULL, *buf2 = NULL;
8957 Py_ssize_t len1, len2;
Tim Petersced69f82003-09-16 20:30:58 +00008958
Victor Stinner9db1a8b2011-10-23 20:04:37 +02008959 str_obj = PyUnicode_FromObject(str);
Benjamin Peterson22a29702012-01-02 09:00:30 -06008960 if (!str_obj)
Benjamin Peterson29060642009-01-31 22:14:21 +00008961 return -1;
Victor Stinner9db1a8b2011-10-23 20:04:37 +02008962 sub_obj = PyUnicode_FromObject(substr);
Benjamin Peterson22a29702012-01-02 09:00:30 -06008963 if (!sub_obj) {
8964 Py_DECREF(str_obj);
8965 return -1;
8966 }
Benjamin Peterson4c13a4a2012-01-02 09:07:38 -06008967 if (PyUnicode_READY(sub_obj) == -1 || PyUnicode_READY(str_obj) == -1) {
Benjamin Peterson5e458f52012-01-02 10:12:13 -06008968 Py_DECREF(sub_obj);
Benjamin Peterson29060642009-01-31 22:14:21 +00008969 Py_DECREF(str_obj);
8970 return -1;
Guido van Rossumd57fd912000-03-10 22:53:23 +00008971 }
Tim Petersced69f82003-09-16 20:30:58 +00008972
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008973 kind1 = PyUnicode_KIND(str_obj);
8974 kind2 = PyUnicode_KIND(sub_obj);
Antoine Pitroue45c0c52012-05-12 15:49:07 +02008975 kind = kind1;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008976 buf1 = PyUnicode_DATA(str_obj);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008977 buf2 = PyUnicode_DATA(sub_obj);
Benjamin Peterson1ff2e352012-05-11 17:41:20 -05008978 if (kind2 != kind) {
Antoine Pitrou758153b2012-05-12 15:51:51 +02008979 if (kind2 > kind) {
8980 Py_DECREF(sub_obj);
8981 Py_DECREF(str_obj);
Antoine Pitroue45c0c52012-05-12 15:49:07 +02008982 return 0;
Antoine Pitrou758153b2012-05-12 15:51:51 +02008983 }
Victor Stinner7931d9a2011-11-04 00:22:48 +01008984 buf2 = _PyUnicode_AsKind(sub_obj, kind);
Benjamin Peterson1ff2e352012-05-11 17:41:20 -05008985 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008986 if (!buf2)
8987 goto onError;
8988 len1 = PyUnicode_GET_LENGTH(str_obj);
8989 len2 = PyUnicode_GET_LENGTH(sub_obj);
8990
8991 ADJUST_INDICES(start, end, len1);
Benjamin Petersonead6b532011-12-20 17:23:42 -06008992 switch (kind) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008993 case PyUnicode_1BYTE_KIND:
Victor Stinnerc3cec782011-10-05 21:24:08 +02008994 if (PyUnicode_IS_ASCII(str_obj) && PyUnicode_IS_ASCII(sub_obj))
8995 result = asciilib_count(
8996 ((Py_UCS1*)buf1) + start, end - start,
8997 buf2, len2, PY_SSIZE_T_MAX
8998 );
8999 else
9000 result = ucs1lib_count(
9001 ((Py_UCS1*)buf1) + start, end - start,
9002 buf2, len2, PY_SSIZE_T_MAX
9003 );
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009004 break;
9005 case PyUnicode_2BYTE_KIND:
9006 result = ucs2lib_count(
9007 ((Py_UCS2*)buf1) + start, end - start,
9008 buf2, len2, PY_SSIZE_T_MAX
9009 );
9010 break;
9011 case PyUnicode_4BYTE_KIND:
9012 result = ucs4lib_count(
9013 ((Py_UCS4*)buf1) + start, end - start,
9014 buf2, len2, PY_SSIZE_T_MAX
9015 );
9016 break;
9017 default:
9018 assert(0); result = 0;
9019 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00009020
9021 Py_DECREF(sub_obj);
9022 Py_DECREF(str_obj);
9023
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009024 if (kind2 != kind)
9025 PyMem_Free(buf2);
9026
Guido van Rossumd57fd912000-03-10 22:53:23 +00009027 return result;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009028 onError:
9029 Py_DECREF(sub_obj);
9030 Py_DECREF(str_obj);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009031 if (kind2 != kind && buf2)
9032 PyMem_Free(buf2);
9033 return -1;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009034}
9035
Alexander Belopolsky40018472011-02-26 01:02:56 +00009036Py_ssize_t
9037PyUnicode_Find(PyObject *str,
9038 PyObject *sub,
9039 Py_ssize_t start,
9040 Py_ssize_t end,
9041 int direction)
Guido van Rossumd57fd912000-03-10 22:53:23 +00009042{
Martin v. Löwis18e16552006-02-15 17:27:45 +00009043 Py_ssize_t result;
Tim Petersced69f82003-09-16 20:30:58 +00009044
Guido van Rossumd57fd912000-03-10 22:53:23 +00009045 str = PyUnicode_FromObject(str);
Benjamin Peterson22a29702012-01-02 09:00:30 -06009046 if (!str)
Benjamin Peterson29060642009-01-31 22:14:21 +00009047 return -2;
Thomas Wouters477c8d52006-05-27 19:21:47 +00009048 sub = PyUnicode_FromObject(sub);
Benjamin Peterson22a29702012-01-02 09:00:30 -06009049 if (!sub) {
9050 Py_DECREF(str);
9051 return -2;
9052 }
9053 if (PyUnicode_READY(sub) == -1 || PyUnicode_READY(str) == -1) {
9054 Py_DECREF(sub);
Benjamin Peterson29060642009-01-31 22:14:21 +00009055 Py_DECREF(str);
9056 return -2;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009057 }
Tim Petersced69f82003-09-16 20:30:58 +00009058
Victor Stinner794d5672011-10-10 03:21:36 +02009059 result = any_find_slice(direction,
9060 str, sub, start, end
9061 );
Thomas Wouters477c8d52006-05-27 19:21:47 +00009062
Guido van Rossumd57fd912000-03-10 22:53:23 +00009063 Py_DECREF(str);
Thomas Wouters477c8d52006-05-27 19:21:47 +00009064 Py_DECREF(sub);
9065
Guido van Rossumd57fd912000-03-10 22:53:23 +00009066 return result;
9067}
9068
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009069Py_ssize_t
9070PyUnicode_FindChar(PyObject *str, Py_UCS4 ch,
9071 Py_ssize_t start, Py_ssize_t end,
9072 int direction)
9073{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009074 int kind;
Antoine Pitrouf0b934b2011-10-13 18:55:09 +02009075 Py_ssize_t result;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009076 if (PyUnicode_READY(str) == -1)
9077 return -2;
Victor Stinner267aa242011-10-02 01:08:37 +02009078 if (start < 0 || end < 0) {
9079 PyErr_SetString(PyExc_IndexError, "string index out of range");
9080 return -2;
9081 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009082 if (end > PyUnicode_GET_LENGTH(str))
9083 end = PyUnicode_GET_LENGTH(str);
9084 kind = PyUnicode_KIND(str);
Antoine Pitrouf0b934b2011-10-13 18:55:09 +02009085 result = findchar(PyUnicode_1BYTE_DATA(str) + kind*start,
9086 kind, end-start, ch, direction);
9087 if (result == -1)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009088 return -1;
Antoine Pitrouf0b934b2011-10-13 18:55:09 +02009089 else
9090 return start + result;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009091}
9092
Alexander Belopolsky40018472011-02-26 01:02:56 +00009093static int
Victor Stinner9db1a8b2011-10-23 20:04:37 +02009094tailmatch(PyObject *self,
9095 PyObject *substring,
Alexander Belopolsky40018472011-02-26 01:02:56 +00009096 Py_ssize_t start,
9097 Py_ssize_t end,
9098 int direction)
Guido van Rossumd57fd912000-03-10 22:53:23 +00009099{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009100 int kind_self;
9101 int kind_sub;
9102 void *data_self;
9103 void *data_sub;
9104 Py_ssize_t offset;
9105 Py_ssize_t i;
9106 Py_ssize_t end_sub;
9107
9108 if (PyUnicode_READY(self) == -1 ||
9109 PyUnicode_READY(substring) == -1)
9110 return 0;
9111
9112 if (PyUnicode_GET_LENGTH(substring) == 0)
Guido van Rossumd57fd912000-03-10 22:53:23 +00009113 return 1;
9114
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009115 ADJUST_INDICES(start, end, PyUnicode_GET_LENGTH(self));
9116 end -= PyUnicode_GET_LENGTH(substring);
Guido van Rossumd57fd912000-03-10 22:53:23 +00009117 if (end < start)
Benjamin Peterson29060642009-01-31 22:14:21 +00009118 return 0;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009119
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009120 kind_self = PyUnicode_KIND(self);
9121 data_self = PyUnicode_DATA(self);
9122 kind_sub = PyUnicode_KIND(substring);
9123 data_sub = PyUnicode_DATA(substring);
9124 end_sub = PyUnicode_GET_LENGTH(substring) - 1;
9125
9126 if (direction > 0)
9127 offset = end;
9128 else
9129 offset = start;
9130
9131 if (PyUnicode_READ(kind_self, data_self, offset) ==
9132 PyUnicode_READ(kind_sub, data_sub, 0) &&
9133 PyUnicode_READ(kind_self, data_self, offset + end_sub) ==
9134 PyUnicode_READ(kind_sub, data_sub, end_sub)) {
9135 /* If both are of the same kind, memcmp is sufficient */
9136 if (kind_self == kind_sub) {
9137 return ! memcmp((char *)data_self +
Martin v. Löwisc47adb02011-10-07 20:55:35 +02009138 (offset * PyUnicode_KIND(substring)),
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009139 data_sub,
9140 PyUnicode_GET_LENGTH(substring) *
Martin v. Löwisc47adb02011-10-07 20:55:35 +02009141 PyUnicode_KIND(substring));
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009142 }
9143 /* otherwise we have to compare each character by first accesing it */
9144 else {
9145 /* We do not need to compare 0 and len(substring)-1 because
9146 the if statement above ensured already that they are equal
9147 when we end up here. */
Antoine Pitrou057119b2012-09-02 17:56:33 +02009148 /* TODO: honor direction and do a forward or backwards search */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009149 for (i = 1; i < end_sub; ++i) {
9150 if (PyUnicode_READ(kind_self, data_self, offset + i) !=
9151 PyUnicode_READ(kind_sub, data_sub, i))
9152 return 0;
9153 }
Benjamin Peterson29060642009-01-31 22:14:21 +00009154 return 1;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009155 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00009156 }
9157
9158 return 0;
9159}
9160
Alexander Belopolsky40018472011-02-26 01:02:56 +00009161Py_ssize_t
9162PyUnicode_Tailmatch(PyObject *str,
9163 PyObject *substr,
9164 Py_ssize_t start,
9165 Py_ssize_t end,
9166 int direction)
Guido van Rossumd57fd912000-03-10 22:53:23 +00009167{
Martin v. Löwis18e16552006-02-15 17:27:45 +00009168 Py_ssize_t result;
Tim Petersced69f82003-09-16 20:30:58 +00009169
Guido van Rossumd57fd912000-03-10 22:53:23 +00009170 str = PyUnicode_FromObject(str);
9171 if (str == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00009172 return -1;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009173 substr = PyUnicode_FromObject(substr);
9174 if (substr == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +00009175 Py_DECREF(str);
9176 return -1;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009177 }
Tim Petersced69f82003-09-16 20:30:58 +00009178
Victor Stinner9db1a8b2011-10-23 20:04:37 +02009179 result = tailmatch(str, substr,
Benjamin Peterson29060642009-01-31 22:14:21 +00009180 start, end, direction);
Guido van Rossumd57fd912000-03-10 22:53:23 +00009181 Py_DECREF(str);
9182 Py_DECREF(substr);
9183 return result;
9184}
9185
Guido van Rossumd57fd912000-03-10 22:53:23 +00009186/* Apply fixfct filter to the Unicode object self and return a
9187 reference to the modified object */
9188
Alexander Belopolsky40018472011-02-26 01:02:56 +00009189static PyObject *
Victor Stinner9310abb2011-10-05 00:59:23 +02009190fixup(PyObject *self,
9191 Py_UCS4 (*fixfct)(PyObject *s))
Guido van Rossumd57fd912000-03-10 22:53:23 +00009192{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009193 PyObject *u;
9194 Py_UCS4 maxchar_old, maxchar_new = 0;
Victor Stinnereaab6042011-12-11 22:22:39 +01009195 PyObject *v;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009196
Victor Stinnerbf6e5602011-12-12 01:53:47 +01009197 u = _PyUnicode_Copy(self);
Guido van Rossumd57fd912000-03-10 22:53:23 +00009198 if (u == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00009199 return NULL;
Victor Stinner87af4f22011-11-21 23:03:47 +01009200 maxchar_old = PyUnicode_MAX_CHAR_VALUE(u);
Marc-André Lemburg8155e0e2001-04-23 14:44:21 +00009201
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009202 /* fix functions return the new maximum character in a string,
9203 if the kind of the resulting unicode object does not change,
9204 everything is fine. Otherwise we need to change the string kind
9205 and re-run the fix function. */
Victor Stinner9310abb2011-10-05 00:59:23 +02009206 maxchar_new = fixfct(u);
Victor Stinnereaab6042011-12-11 22:22:39 +01009207
9208 if (maxchar_new == 0) {
9209 /* no changes */;
9210 if (PyUnicode_CheckExact(self)) {
9211 Py_DECREF(u);
9212 Py_INCREF(self);
9213 return self;
9214 }
9215 else
9216 return u;
9217 }
9218
Victor Stinnere6abb482012-05-02 01:15:40 +02009219 maxchar_new = align_maxchar(maxchar_new);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009220
Victor Stinnereaab6042011-12-11 22:22:39 +01009221 if (maxchar_new == maxchar_old)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009222 return u;
Victor Stinnereaab6042011-12-11 22:22:39 +01009223
9224 /* In case the maximum character changed, we need to
9225 convert the string to the new category. */
9226 v = PyUnicode_New(PyUnicode_GET_LENGTH(self), maxchar_new);
9227 if (v == NULL) {
9228 Py_DECREF(u);
9229 return NULL;
9230 }
9231 if (maxchar_new > maxchar_old) {
9232 /* If the maxchar increased so that the kind changed, not all
9233 characters are representable anymore and we need to fix the
9234 string again. This only happens in very few cases. */
Victor Stinnerd3f08822012-05-29 12:57:52 +02009235 _PyUnicode_FastCopyCharacters(v, 0,
9236 self, 0, PyUnicode_GET_LENGTH(self));
Victor Stinnereaab6042011-12-11 22:22:39 +01009237 maxchar_old = fixfct(v);
9238 assert(maxchar_old > 0 && maxchar_old <= maxchar_new);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009239 }
9240 else {
Victor Stinnerd3f08822012-05-29 12:57:52 +02009241 _PyUnicode_FastCopyCharacters(v, 0,
9242 u, 0, PyUnicode_GET_LENGTH(self));
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009243 }
Victor Stinnereaab6042011-12-11 22:22:39 +01009244 Py_DECREF(u);
9245 assert(_PyUnicode_CheckConsistency(v, 1));
9246 return v;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009247}
9248
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009249static PyObject *
9250ascii_upper_or_lower(PyObject *self, int lower)
Guido van Rossumd57fd912000-03-10 22:53:23 +00009251{
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009252 Py_ssize_t len = PyUnicode_GET_LENGTH(self);
9253 char *resdata, *data = PyUnicode_DATA(self);
9254 PyObject *res;
Tim Petersced69f82003-09-16 20:30:58 +00009255
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009256 res = PyUnicode_New(len, 127);
9257 if (res == NULL)
9258 return NULL;
9259 resdata = PyUnicode_DATA(res);
9260 if (lower)
9261 _Py_bytes_lower(resdata, data, len);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009262 else
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009263 _Py_bytes_upper(resdata, data, len);
9264 return res;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009265}
9266
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009267static Py_UCS4
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009268handle_capital_sigma(int kind, void *data, Py_ssize_t length, Py_ssize_t i)
Guido van Rossumd57fd912000-03-10 22:53:23 +00009269{
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009270 Py_ssize_t j;
9271 int final_sigma;
9272 Py_UCS4 c;
9273 /* U+03A3 is in the Final_Sigma context when, it is found like this:
Tim Petersced69f82003-09-16 20:30:58 +00009274
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009275 \p{cased}\p{case-ignorable}*U+03A3!(\p{case-ignorable}*\p{cased})
9276
9277 where ! is a negation and \p{xxx} is a character with property xxx.
9278 */
9279 for (j = i - 1; j >= 0; j--) {
9280 c = PyUnicode_READ(kind, data, j);
9281 if (!_PyUnicode_IsCaseIgnorable(c))
9282 break;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009283 }
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009284 final_sigma = j >= 0 && _PyUnicode_IsCased(c);
9285 if (final_sigma) {
9286 for (j = i + 1; j < length; j++) {
9287 c = PyUnicode_READ(kind, data, j);
9288 if (!_PyUnicode_IsCaseIgnorable(c))
9289 break;
9290 }
9291 final_sigma = j == length || !_PyUnicode_IsCased(c);
9292 }
9293 return (final_sigma) ? 0x3C2 : 0x3C3;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009294}
9295
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009296static int
9297lower_ucs4(int kind, void *data, Py_ssize_t length, Py_ssize_t i,
9298 Py_UCS4 c, Py_UCS4 *mapped)
Guido van Rossumd57fd912000-03-10 22:53:23 +00009299{
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009300 /* Obscure special case. */
9301 if (c == 0x3A3) {
9302 mapped[0] = handle_capital_sigma(kind, data, length, i);
9303 return 1;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009304 }
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009305 return _PyUnicode_ToLowerFull(c, mapped);
Guido van Rossumd57fd912000-03-10 22:53:23 +00009306}
9307
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009308static Py_ssize_t
9309do_capitalize(int kind, void *data, Py_ssize_t length, Py_UCS4 *res, Py_UCS4 *maxchar)
Guido van Rossumd57fd912000-03-10 22:53:23 +00009310{
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009311 Py_ssize_t i, k = 0;
9312 int n_res, j;
9313 Py_UCS4 c, mapped[3];
Tim Petersced69f82003-09-16 20:30:58 +00009314
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009315 c = PyUnicode_READ(kind, data, 0);
9316 n_res = _PyUnicode_ToUpperFull(c, mapped);
9317 for (j = 0; j < n_res; j++) {
Victor Stinnere6abb482012-05-02 01:15:40 +02009318 *maxchar = MAX_MAXCHAR(*maxchar, mapped[j]);
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009319 res[k++] = mapped[j];
Guido van Rossumd57fd912000-03-10 22:53:23 +00009320 }
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009321 for (i = 1; i < length; i++) {
9322 c = PyUnicode_READ(kind, data, i);
9323 n_res = lower_ucs4(kind, data, length, i, c, mapped);
9324 for (j = 0; j < n_res; j++) {
Victor Stinnere6abb482012-05-02 01:15:40 +02009325 *maxchar = MAX_MAXCHAR(*maxchar, mapped[j]);
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009326 res[k++] = mapped[j];
Marc-André Lemburgfde66e12001-01-29 11:14:16 +00009327 }
Marc-André Lemburgfde66e12001-01-29 11:14:16 +00009328 }
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009329 return k;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009330}
9331
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009332static Py_ssize_t
9333do_swapcase(int kind, void *data, Py_ssize_t length, Py_UCS4 *res, Py_UCS4 *maxchar) {
9334 Py_ssize_t i, k = 0;
9335
9336 for (i = 0; i < length; i++) {
9337 Py_UCS4 c = PyUnicode_READ(kind, data, i), mapped[3];
9338 int n_res, j;
9339 if (Py_UNICODE_ISUPPER(c)) {
9340 n_res = lower_ucs4(kind, data, length, i, c, mapped);
9341 }
9342 else if (Py_UNICODE_ISLOWER(c)) {
9343 n_res = _PyUnicode_ToUpperFull(c, mapped);
9344 }
9345 else {
9346 n_res = 1;
9347 mapped[0] = c;
9348 }
9349 for (j = 0; j < n_res; j++) {
Victor Stinnere6abb482012-05-02 01:15:40 +02009350 *maxchar = MAX_MAXCHAR(*maxchar, mapped[j]);
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009351 res[k++] = mapped[j];
9352 }
9353 }
9354 return k;
9355}
9356
9357static Py_ssize_t
9358do_upper_or_lower(int kind, void *data, Py_ssize_t length, Py_UCS4 *res,
9359 Py_UCS4 *maxchar, int lower)
Guido van Rossumd57fd912000-03-10 22:53:23 +00009360{
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009361 Py_ssize_t i, k = 0;
9362
9363 for (i = 0; i < length; i++) {
9364 Py_UCS4 c = PyUnicode_READ(kind, data, i), mapped[3];
9365 int n_res, j;
9366 if (lower)
9367 n_res = lower_ucs4(kind, data, length, i, c, mapped);
9368 else
9369 n_res = _PyUnicode_ToUpperFull(c, mapped);
9370 for (j = 0; j < n_res; j++) {
Victor Stinnere6abb482012-05-02 01:15:40 +02009371 *maxchar = MAX_MAXCHAR(*maxchar, mapped[j]);
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009372 res[k++] = mapped[j];
9373 }
9374 }
9375 return k;
9376}
9377
9378static Py_ssize_t
9379do_upper(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, 0);
9382}
9383
9384static Py_ssize_t
9385do_lower(int kind, void *data, Py_ssize_t length, Py_UCS4 *res, Py_UCS4 *maxchar)
9386{
9387 return do_upper_or_lower(kind, data, length, res, maxchar, 1);
9388}
9389
Benjamin Petersone51757f2012-01-12 21:10:29 -05009390static Py_ssize_t
Benjamin Petersond5890c82012-01-14 13:23:30 -05009391do_casefold(int kind, void *data, Py_ssize_t length, Py_UCS4 *res, Py_UCS4 *maxchar)
9392{
9393 Py_ssize_t i, k = 0;
9394
9395 for (i = 0; i < length; i++) {
9396 Py_UCS4 c = PyUnicode_READ(kind, data, i);
9397 Py_UCS4 mapped[3];
9398 int j, n_res = _PyUnicode_ToFoldedFull(c, mapped);
9399 for (j = 0; j < n_res; j++) {
Victor Stinnere6abb482012-05-02 01:15:40 +02009400 *maxchar = MAX_MAXCHAR(*maxchar, mapped[j]);
Benjamin Petersond5890c82012-01-14 13:23:30 -05009401 res[k++] = mapped[j];
9402 }
9403 }
9404 return k;
9405}
9406
9407static Py_ssize_t
Benjamin Petersone51757f2012-01-12 21:10:29 -05009408do_title(int kind, void *data, Py_ssize_t length, Py_UCS4 *res, Py_UCS4 *maxchar)
9409{
9410 Py_ssize_t i, k = 0;
9411 int previous_is_cased;
9412
9413 previous_is_cased = 0;
9414 for (i = 0; i < length; i++) {
9415 const Py_UCS4 c = PyUnicode_READ(kind, data, i);
9416 Py_UCS4 mapped[3];
9417 int n_res, j;
9418
9419 if (previous_is_cased)
9420 n_res = lower_ucs4(kind, data, length, i, c, mapped);
9421 else
9422 n_res = _PyUnicode_ToTitleFull(c, mapped);
9423
9424 for (j = 0; j < n_res; j++) {
Victor Stinnere6abb482012-05-02 01:15:40 +02009425 *maxchar = MAX_MAXCHAR(*maxchar, mapped[j]);
Benjamin Petersone51757f2012-01-12 21:10:29 -05009426 res[k++] = mapped[j];
9427 }
9428
9429 previous_is_cased = _PyUnicode_IsCased(c);
9430 }
9431 return k;
9432}
9433
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009434static PyObject *
9435case_operation(PyObject *self,
9436 Py_ssize_t (*perform)(int, void *, Py_ssize_t, Py_UCS4 *, Py_UCS4 *))
9437{
9438 PyObject *res = NULL;
9439 Py_ssize_t length, newlength = 0;
9440 int kind, outkind;
9441 void *data, *outdata;
9442 Py_UCS4 maxchar = 0, *tmp, *tmpend;
9443
Benjamin Petersoneea48462012-01-16 14:28:50 -05009444 assert(PyUnicode_IS_READY(self));
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009445
9446 kind = PyUnicode_KIND(self);
9447 data = PyUnicode_DATA(self);
9448 length = PyUnicode_GET_LENGTH(self);
9449 tmp = PyMem_MALLOC(sizeof(Py_UCS4) * 3 * length);
9450 if (tmp == NULL)
9451 return PyErr_NoMemory();
9452 newlength = perform(kind, data, length, tmp, &maxchar);
9453 res = PyUnicode_New(newlength, maxchar);
9454 if (res == NULL)
9455 goto leave;
9456 tmpend = tmp + newlength;
9457 outdata = PyUnicode_DATA(res);
9458 outkind = PyUnicode_KIND(res);
9459 switch (outkind) {
9460 case PyUnicode_1BYTE_KIND:
9461 _PyUnicode_CONVERT_BYTES(Py_UCS4, Py_UCS1, tmp, tmpend, outdata);
9462 break;
9463 case PyUnicode_2BYTE_KIND:
9464 _PyUnicode_CONVERT_BYTES(Py_UCS4, Py_UCS2, tmp, tmpend, outdata);
9465 break;
9466 case PyUnicode_4BYTE_KIND:
9467 memcpy(outdata, tmp, sizeof(Py_UCS4) * newlength);
9468 break;
9469 default:
9470 assert(0);
9471 break;
9472 }
9473 leave:
9474 PyMem_FREE(tmp);
9475 return res;
9476}
9477
Tim Peters8ce9f162004-08-27 01:49:32 +00009478PyObject *
9479PyUnicode_Join(PyObject *separator, PyObject *seq)
Guido van Rossumd57fd912000-03-10 22:53:23 +00009480{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009481 PyObject *sep = NULL;
Victor Stinnerdd077322011-10-07 17:02:31 +02009482 Py_ssize_t seplen;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009483 PyObject *res = NULL; /* the result */
Tim Peters05eba1f2004-08-27 21:32:02 +00009484 PyObject *fseq; /* PySequence_Fast(seq) */
Antoine Pitrouaf14b792008-08-07 21:50:41 +00009485 Py_ssize_t seqlen; /* len(fseq) -- number of items in sequence */
9486 PyObject **items;
Tim Peters8ce9f162004-08-27 01:49:32 +00009487 PyObject *item;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009488 Py_ssize_t sz, i, res_offset;
Victor Stinnerfb9ea8c2011-10-06 01:45:57 +02009489 Py_UCS4 maxchar;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009490 Py_UCS4 item_maxchar;
Victor Stinnerdd077322011-10-07 17:02:31 +02009491 int use_memcpy;
9492 unsigned char *res_data = NULL, *sep_data = NULL;
9493 PyObject *last_obj;
9494 unsigned int kind = 0;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009495
Tim Peters05eba1f2004-08-27 21:32:02 +00009496 fseq = PySequence_Fast(seq, "");
9497 if (fseq == NULL) {
Benjamin Peterson14339b62009-01-31 16:36:08 +00009498 return NULL;
Tim Peters8ce9f162004-08-27 01:49:32 +00009499 }
9500
Antoine Pitrouaf14b792008-08-07 21:50:41 +00009501 /* NOTE: the following code can't call back into Python code,
9502 * so we are sure that fseq won't be mutated.
Tim Peters91879ab2004-08-27 22:35:44 +00009503 */
Antoine Pitrouaf14b792008-08-07 21:50:41 +00009504
Tim Peters05eba1f2004-08-27 21:32:02 +00009505 seqlen = PySequence_Fast_GET_SIZE(fseq);
9506 /* If empty sequence, return u"". */
9507 if (seqlen == 0) {
Victor Stinnerfb9ea8c2011-10-06 01:45:57 +02009508 Py_DECREF(fseq);
9509 Py_INCREF(unicode_empty);
9510 res = unicode_empty;
9511 return res;
Tim Peters05eba1f2004-08-27 21:32:02 +00009512 }
Victor Stinnerfb9ea8c2011-10-06 01:45:57 +02009513
Tim Peters05eba1f2004-08-27 21:32:02 +00009514 /* If singleton sequence with an exact Unicode, return that. */
Victor Stinnerdd077322011-10-07 17:02:31 +02009515 last_obj = NULL;
Victor Stinnerfb9ea8c2011-10-06 01:45:57 +02009516 items = PySequence_Fast_ITEMS(fseq);
Victor Stinneracf47b82011-10-06 12:32:37 +02009517 if (seqlen == 1) {
9518 if (PyUnicode_CheckExact(items[0])) {
9519 res = items[0];
9520 Py_INCREF(res);
9521 Py_DECREF(fseq);
9522 return res;
9523 }
Victor Stinnerdd077322011-10-07 17:02:31 +02009524 seplen = 0;
Victor Stinnerc6f0df72011-10-06 15:58:54 +02009525 maxchar = 0;
Tim Peters8ce9f162004-08-27 01:49:32 +00009526 }
Antoine Pitrouaf14b792008-08-07 21:50:41 +00009527 else {
Victor Stinneracf47b82011-10-06 12:32:37 +02009528 /* Set up sep and seplen */
9529 if (separator == NULL) {
9530 /* fall back to a blank space separator */
9531 sep = PyUnicode_FromOrdinal(' ');
9532 if (!sep)
9533 goto onError;
Victor Stinnerdd077322011-10-07 17:02:31 +02009534 seplen = 1;
Victor Stinneracf47b82011-10-06 12:32:37 +02009535 maxchar = 32;
Tim Peters05eba1f2004-08-27 21:32:02 +00009536 }
Victor Stinneracf47b82011-10-06 12:32:37 +02009537 else {
9538 if (!PyUnicode_Check(separator)) {
9539 PyErr_Format(PyExc_TypeError,
9540 "separator: expected str instance,"
9541 " %.80s found",
9542 Py_TYPE(separator)->tp_name);
9543 goto onError;
9544 }
9545 if (PyUnicode_READY(separator))
9546 goto onError;
9547 sep = separator;
9548 seplen = PyUnicode_GET_LENGTH(separator);
9549 maxchar = PyUnicode_MAX_CHAR_VALUE(separator);
9550 /* inc refcount to keep this code path symmetric with the
9551 above case of a blank separator */
9552 Py_INCREF(sep);
9553 }
Victor Stinnerdd077322011-10-07 17:02:31 +02009554 last_obj = sep;
Tim Peters05eba1f2004-08-27 21:32:02 +00009555 }
9556
Antoine Pitrouaf14b792008-08-07 21:50:41 +00009557 /* There are at least two things to join, or else we have a subclass
9558 * of str in the sequence.
9559 * Do a pre-pass to figure out the total amount of space we'll
9560 * need (sz), and see whether all argument are strings.
9561 */
9562 sz = 0;
Victor Stinnerdd077322011-10-07 17:02:31 +02009563#ifdef Py_DEBUG
9564 use_memcpy = 0;
9565#else
9566 use_memcpy = 1;
9567#endif
Antoine Pitrouaf14b792008-08-07 21:50:41 +00009568 for (i = 0; i < seqlen; i++) {
9569 const Py_ssize_t old_sz = sz;
9570 item = items[i];
Benjamin Peterson29060642009-01-31 22:14:21 +00009571 if (!PyUnicode_Check(item)) {
9572 PyErr_Format(PyExc_TypeError,
9573 "sequence item %zd: expected str instance,"
9574 " %.80s found",
9575 i, Py_TYPE(item)->tp_name);
9576 goto onError;
9577 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009578 if (PyUnicode_READY(item) == -1)
9579 goto onError;
9580 sz += PyUnicode_GET_LENGTH(item);
9581 item_maxchar = PyUnicode_MAX_CHAR_VALUE(item);
Victor Stinnere6abb482012-05-02 01:15:40 +02009582 maxchar = MAX_MAXCHAR(maxchar, item_maxchar);
Antoine Pitrouaf14b792008-08-07 21:50:41 +00009583 if (i != 0)
9584 sz += seplen;
9585 if (sz < old_sz || sz > PY_SSIZE_T_MAX) {
9586 PyErr_SetString(PyExc_OverflowError,
Benjamin Peterson29060642009-01-31 22:14:21 +00009587 "join() result is too long for a Python string");
Antoine Pitrouaf14b792008-08-07 21:50:41 +00009588 goto onError;
9589 }
Victor Stinnerdd077322011-10-07 17:02:31 +02009590 if (use_memcpy && last_obj != NULL) {
9591 if (PyUnicode_KIND(last_obj) != PyUnicode_KIND(item))
9592 use_memcpy = 0;
9593 }
9594 last_obj = item;
Antoine Pitrouaf14b792008-08-07 21:50:41 +00009595 }
Tim Petersced69f82003-09-16 20:30:58 +00009596
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009597 res = PyUnicode_New(sz, maxchar);
Antoine Pitrouaf14b792008-08-07 21:50:41 +00009598 if (res == NULL)
9599 goto onError;
Tim Peters91879ab2004-08-27 22:35:44 +00009600
Antoine Pitrouaf14b792008-08-07 21:50:41 +00009601 /* Catenate everything. */
Victor Stinnerdd077322011-10-07 17:02:31 +02009602#ifdef Py_DEBUG
9603 use_memcpy = 0;
9604#else
9605 if (use_memcpy) {
9606 res_data = PyUnicode_1BYTE_DATA(res);
9607 kind = PyUnicode_KIND(res);
9608 if (seplen != 0)
9609 sep_data = PyUnicode_1BYTE_DATA(sep);
9610 }
9611#endif
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009612 for (i = 0, res_offset = 0; i < seqlen; ++i) {
Victor Stinnerfb9ea8c2011-10-06 01:45:57 +02009613 Py_ssize_t itemlen;
Antoine Pitrouaf14b792008-08-07 21:50:41 +00009614 item = items[i];
Benjamin Peterson29060642009-01-31 22:14:21 +00009615 /* Copy item, and maybe the separator. */
Victor Stinner9ce5a832011-10-03 23:36:02 +02009616 if (i && seplen != 0) {
Victor Stinnerdd077322011-10-07 17:02:31 +02009617 if (use_memcpy) {
9618 Py_MEMCPY(res_data,
9619 sep_data,
Martin v. Löwisc47adb02011-10-07 20:55:35 +02009620 kind * seplen);
9621 res_data += kind * seplen;
Victor Stinnerdd077322011-10-07 17:02:31 +02009622 }
9623 else {
Victor Stinnerd3f08822012-05-29 12:57:52 +02009624 _PyUnicode_FastCopyCharacters(res, res_offset, sep, 0, seplen);
Victor Stinnerdd077322011-10-07 17:02:31 +02009625 res_offset += seplen;
9626 }
Benjamin Peterson29060642009-01-31 22:14:21 +00009627 }
Victor Stinner9ce5a832011-10-03 23:36:02 +02009628 itemlen = PyUnicode_GET_LENGTH(item);
9629 if (itemlen != 0) {
Victor Stinnerdd077322011-10-07 17:02:31 +02009630 if (use_memcpy) {
9631 Py_MEMCPY(res_data,
9632 PyUnicode_DATA(item),
Martin v. Löwisc47adb02011-10-07 20:55:35 +02009633 kind * itemlen);
9634 res_data += kind * itemlen;
Victor Stinnerdd077322011-10-07 17:02:31 +02009635 }
9636 else {
Victor Stinnerd3f08822012-05-29 12:57:52 +02009637 _PyUnicode_FastCopyCharacters(res, res_offset, item, 0, itemlen);
Victor Stinnerdd077322011-10-07 17:02:31 +02009638 res_offset += itemlen;
9639 }
Victor Stinner9ce5a832011-10-03 23:36:02 +02009640 }
Tim Peters05eba1f2004-08-27 21:32:02 +00009641 }
Victor Stinnerdd077322011-10-07 17:02:31 +02009642 if (use_memcpy)
9643 assert(res_data == PyUnicode_1BYTE_DATA(res)
Martin v. Löwisc47adb02011-10-07 20:55:35 +02009644 + kind * PyUnicode_GET_LENGTH(res));
Victor Stinnerdd077322011-10-07 17:02:31 +02009645 else
9646 assert(res_offset == PyUnicode_GET_LENGTH(res));
Tim Peters8ce9f162004-08-27 01:49:32 +00009647
Tim Peters05eba1f2004-08-27 21:32:02 +00009648 Py_DECREF(fseq);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009649 Py_XDECREF(sep);
Victor Stinnerbb10a1f2011-10-05 01:34:17 +02009650 assert(_PyUnicode_CheckConsistency(res, 1));
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009651 return res;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009652
Benjamin Peterson29060642009-01-31 22:14:21 +00009653 onError:
Tim Peters05eba1f2004-08-27 21:32:02 +00009654 Py_DECREF(fseq);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009655 Py_XDECREF(sep);
Tim Peters8ce9f162004-08-27 01:49:32 +00009656 Py_XDECREF(res);
Guido van Rossumd57fd912000-03-10 22:53:23 +00009657 return NULL;
9658}
9659
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009660#define FILL(kind, data, value, start, length) \
9661 do { \
9662 Py_ssize_t i_ = 0; \
9663 assert(kind != PyUnicode_WCHAR_KIND); \
9664 switch ((kind)) { \
9665 case PyUnicode_1BYTE_KIND: { \
9666 unsigned char * to_ = (unsigned char *)((data)) + (start); \
Victor Stinnerf2c76aa2012-05-03 13:10:40 +02009667 memset(to_, (unsigned char)value, (length)); \
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009668 break; \
9669 } \
9670 case PyUnicode_2BYTE_KIND: { \
9671 Py_UCS2 * to_ = (Py_UCS2 *)((data)) + (start); \
9672 for (; i_ < (length); ++i_, ++to_) *to_ = (value); \
9673 break; \
9674 } \
Benjamin Petersone157cf12012-01-01 15:56:20 -06009675 case PyUnicode_4BYTE_KIND: { \
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009676 Py_UCS4 * to_ = (Py_UCS4 *)((data)) + (start); \
9677 for (; i_ < (length); ++i_, ++to_) *to_ = (value); \
9678 break; \
Benjamin Petersone157cf12012-01-01 15:56:20 -06009679 default: assert(0); \
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009680 } \
9681 } \
9682 } while (0)
9683
Victor Stinnerd3f08822012-05-29 12:57:52 +02009684void
9685_PyUnicode_FastFill(PyObject *unicode, Py_ssize_t start, Py_ssize_t length,
9686 Py_UCS4 fill_char)
9687{
9688 const enum PyUnicode_Kind kind = PyUnicode_KIND(unicode);
9689 const void *data = PyUnicode_DATA(unicode);
9690 assert(PyUnicode_IS_READY(unicode));
9691 assert(unicode_modifiable(unicode));
9692 assert(fill_char <= PyUnicode_MAX_CHAR_VALUE(unicode));
9693 assert(start >= 0);
9694 assert(start + length <= PyUnicode_GET_LENGTH(unicode));
9695 FILL(kind, data, fill_char, start, length);
9696}
9697
Victor Stinner3fe55312012-01-04 00:33:50 +01009698Py_ssize_t
9699PyUnicode_Fill(PyObject *unicode, Py_ssize_t start, Py_ssize_t length,
9700 Py_UCS4 fill_char)
9701{
9702 Py_ssize_t maxlen;
Victor Stinner3fe55312012-01-04 00:33:50 +01009703
9704 if (!PyUnicode_Check(unicode)) {
9705 PyErr_BadInternalCall();
9706 return -1;
9707 }
9708 if (PyUnicode_READY(unicode) == -1)
9709 return -1;
9710 if (unicode_check_modifiable(unicode))
9711 return -1;
9712
Victor Stinnerd3f08822012-05-29 12:57:52 +02009713 if (start < 0) {
9714 PyErr_SetString(PyExc_IndexError, "string index out of range");
9715 return -1;
9716 }
Victor Stinner3fe55312012-01-04 00:33:50 +01009717 if (fill_char > PyUnicode_MAX_CHAR_VALUE(unicode)) {
9718 PyErr_SetString(PyExc_ValueError,
9719 "fill character is bigger than "
9720 "the string maximum character");
9721 return -1;
9722 }
9723
9724 maxlen = PyUnicode_GET_LENGTH(unicode) - start;
9725 length = Py_MIN(maxlen, length);
9726 if (length <= 0)
9727 return 0;
9728
Victor Stinnerd3f08822012-05-29 12:57:52 +02009729 _PyUnicode_FastFill(unicode, start, length, fill_char);
Victor Stinner3fe55312012-01-04 00:33:50 +01009730 return length;
9731}
9732
Victor Stinner9310abb2011-10-05 00:59:23 +02009733static PyObject *
9734pad(PyObject *self,
Alexander Belopolsky40018472011-02-26 01:02:56 +00009735 Py_ssize_t left,
9736 Py_ssize_t right,
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009737 Py_UCS4 fill)
Guido van Rossumd57fd912000-03-10 22:53:23 +00009738{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009739 PyObject *u;
9740 Py_UCS4 maxchar;
Victor Stinner6c7a52a2011-09-28 21:39:17 +02009741 int kind;
9742 void *data;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009743
9744 if (left < 0)
9745 left = 0;
9746 if (right < 0)
9747 right = 0;
9748
Victor Stinnerc4b49542011-12-11 22:44:26 +01009749 if (left == 0 && right == 0)
9750 return unicode_result_unchanged(self);
Guido van Rossumd57fd912000-03-10 22:53:23 +00009751
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009752 if (left > PY_SSIZE_T_MAX - _PyUnicode_LENGTH(self) ||
9753 right > PY_SSIZE_T_MAX - (left + _PyUnicode_LENGTH(self))) {
Neal Norwitz3ce5d922008-08-24 07:08:55 +00009754 PyErr_SetString(PyExc_OverflowError, "padded string is too long");
9755 return NULL;
9756 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009757 maxchar = PyUnicode_MAX_CHAR_VALUE(self);
Victor Stinnere6abb482012-05-02 01:15:40 +02009758 maxchar = MAX_MAXCHAR(maxchar, fill);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009759 u = PyUnicode_New(left + _PyUnicode_LENGTH(self) + right, maxchar);
Victor Stinner6c7a52a2011-09-28 21:39:17 +02009760 if (!u)
9761 return NULL;
9762
9763 kind = PyUnicode_KIND(u);
9764 data = PyUnicode_DATA(u);
9765 if (left)
9766 FILL(kind, data, fill, 0, left);
9767 if (right)
9768 FILL(kind, data, fill, left + _PyUnicode_LENGTH(self), right);
Victor Stinnerd3f08822012-05-29 12:57:52 +02009769 _PyUnicode_FastCopyCharacters(u, left, self, 0, _PyUnicode_LENGTH(self));
Victor Stinnerbb10a1f2011-10-05 01:34:17 +02009770 assert(_PyUnicode_CheckConsistency(u, 1));
9771 return u;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009772}
9773
Alexander Belopolsky40018472011-02-26 01:02:56 +00009774PyObject *
9775PyUnicode_Splitlines(PyObject *string, int keepends)
Guido van Rossumd57fd912000-03-10 22:53:23 +00009776{
Guido van Rossumd57fd912000-03-10 22:53:23 +00009777 PyObject *list;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009778
9779 string = PyUnicode_FromObject(string);
Benjamin Peterson22a29702012-01-02 09:00:30 -06009780 if (string == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00009781 return NULL;
Benjamin Peterson22a29702012-01-02 09:00:30 -06009782 if (PyUnicode_READY(string) == -1) {
9783 Py_DECREF(string);
9784 return NULL;
9785 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00009786
Benjamin Petersonead6b532011-12-20 17:23:42 -06009787 switch (PyUnicode_KIND(string)) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009788 case PyUnicode_1BYTE_KIND:
Victor Stinnerc3cec782011-10-05 21:24:08 +02009789 if (PyUnicode_IS_ASCII(string))
9790 list = asciilib_splitlines(
Victor Stinner7931d9a2011-11-04 00:22:48 +01009791 string, PyUnicode_1BYTE_DATA(string),
Victor Stinnerc3cec782011-10-05 21:24:08 +02009792 PyUnicode_GET_LENGTH(string), keepends);
9793 else
9794 list = ucs1lib_splitlines(
Victor Stinner7931d9a2011-11-04 00:22:48 +01009795 string, PyUnicode_1BYTE_DATA(string),
Victor Stinnerc3cec782011-10-05 21:24:08 +02009796 PyUnicode_GET_LENGTH(string), keepends);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009797 break;
9798 case PyUnicode_2BYTE_KIND:
9799 list = ucs2lib_splitlines(
Victor Stinner7931d9a2011-11-04 00:22:48 +01009800 string, PyUnicode_2BYTE_DATA(string),
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009801 PyUnicode_GET_LENGTH(string), keepends);
9802 break;
9803 case PyUnicode_4BYTE_KIND:
9804 list = ucs4lib_splitlines(
Victor Stinner7931d9a2011-11-04 00:22:48 +01009805 string, PyUnicode_4BYTE_DATA(string),
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009806 PyUnicode_GET_LENGTH(string), keepends);
9807 break;
9808 default:
9809 assert(0);
9810 list = 0;
9811 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00009812 Py_DECREF(string);
9813 return list;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009814}
9815
Alexander Belopolsky40018472011-02-26 01:02:56 +00009816static PyObject *
Victor Stinner9310abb2011-10-05 00:59:23 +02009817split(PyObject *self,
9818 PyObject *substring,
Alexander Belopolsky40018472011-02-26 01:02:56 +00009819 Py_ssize_t maxcount)
Guido van Rossumd57fd912000-03-10 22:53:23 +00009820{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009821 int kind1, kind2, kind;
9822 void *buf1, *buf2;
9823 Py_ssize_t len1, len2;
9824 PyObject* out;
9825
Guido van Rossumd57fd912000-03-10 22:53:23 +00009826 if (maxcount < 0)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00009827 maxcount = PY_SSIZE_T_MAX;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009828
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009829 if (PyUnicode_READY(self) == -1)
9830 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009831
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009832 if (substring == NULL)
Benjamin Petersonead6b532011-12-20 17:23:42 -06009833 switch (PyUnicode_KIND(self)) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009834 case PyUnicode_1BYTE_KIND:
Victor Stinnerc3cec782011-10-05 21:24:08 +02009835 if (PyUnicode_IS_ASCII(self))
9836 return asciilib_split_whitespace(
Victor Stinner7931d9a2011-11-04 00:22:48 +01009837 self, PyUnicode_1BYTE_DATA(self),
Victor Stinnerc3cec782011-10-05 21:24:08 +02009838 PyUnicode_GET_LENGTH(self), maxcount
9839 );
9840 else
9841 return ucs1lib_split_whitespace(
Victor Stinner7931d9a2011-11-04 00:22:48 +01009842 self, PyUnicode_1BYTE_DATA(self),
Victor Stinnerc3cec782011-10-05 21:24:08 +02009843 PyUnicode_GET_LENGTH(self), maxcount
9844 );
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009845 case PyUnicode_2BYTE_KIND:
9846 return ucs2lib_split_whitespace(
Victor Stinner7931d9a2011-11-04 00:22:48 +01009847 self, PyUnicode_2BYTE_DATA(self),
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009848 PyUnicode_GET_LENGTH(self), maxcount
9849 );
9850 case PyUnicode_4BYTE_KIND:
9851 return ucs4lib_split_whitespace(
Victor Stinner7931d9a2011-11-04 00:22:48 +01009852 self, PyUnicode_4BYTE_DATA(self),
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009853 PyUnicode_GET_LENGTH(self), maxcount
9854 );
9855 default:
9856 assert(0);
9857 return NULL;
9858 }
9859
9860 if (PyUnicode_READY(substring) == -1)
9861 return NULL;
9862
9863 kind1 = PyUnicode_KIND(self);
9864 kind2 = PyUnicode_KIND(substring);
9865 kind = kind1 > kind2 ? kind1 : kind2;
9866 buf1 = PyUnicode_DATA(self);
9867 buf2 = PyUnicode_DATA(substring);
9868 if (kind1 != kind)
Victor Stinner7931d9a2011-11-04 00:22:48 +01009869 buf1 = _PyUnicode_AsKind(self, kind);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009870 if (!buf1)
9871 return NULL;
9872 if (kind2 != kind)
Victor Stinner7931d9a2011-11-04 00:22:48 +01009873 buf2 = _PyUnicode_AsKind(substring, kind);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009874 if (!buf2) {
9875 if (kind1 != kind) PyMem_Free(buf1);
9876 return NULL;
9877 }
9878 len1 = PyUnicode_GET_LENGTH(self);
9879 len2 = PyUnicode_GET_LENGTH(substring);
9880
Benjamin Petersonead6b532011-12-20 17:23:42 -06009881 switch (kind) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009882 case PyUnicode_1BYTE_KIND:
Victor Stinnerc3cec782011-10-05 21:24:08 +02009883 if (PyUnicode_IS_ASCII(self) && PyUnicode_IS_ASCII(substring))
9884 out = asciilib_split(
Victor Stinner7931d9a2011-11-04 00:22:48 +01009885 self, buf1, len1, buf2, len2, maxcount);
Victor Stinnerc3cec782011-10-05 21:24:08 +02009886 else
9887 out = ucs1lib_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 case PyUnicode_2BYTE_KIND:
9891 out = ucs2lib_split(
Victor Stinner7931d9a2011-11-04 00:22:48 +01009892 self, buf1, len1, buf2, len2, maxcount);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009893 break;
9894 case PyUnicode_4BYTE_KIND:
9895 out = ucs4lib_split(
Victor Stinner7931d9a2011-11-04 00:22:48 +01009896 self, buf1, len1, buf2, len2, maxcount);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009897 break;
9898 default:
9899 out = NULL;
9900 }
9901 if (kind1 != kind)
9902 PyMem_Free(buf1);
9903 if (kind2 != kind)
9904 PyMem_Free(buf2);
9905 return out;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009906}
9907
Alexander Belopolsky40018472011-02-26 01:02:56 +00009908static PyObject *
Victor Stinner9310abb2011-10-05 00:59:23 +02009909rsplit(PyObject *self,
9910 PyObject *substring,
Alexander Belopolsky40018472011-02-26 01:02:56 +00009911 Py_ssize_t maxcount)
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +00009912{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009913 int kind1, kind2, kind;
9914 void *buf1, *buf2;
9915 Py_ssize_t len1, len2;
9916 PyObject* out;
9917
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +00009918 if (maxcount < 0)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00009919 maxcount = PY_SSIZE_T_MAX;
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +00009920
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009921 if (PyUnicode_READY(self) == -1)
9922 return NULL;
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +00009923
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009924 if (substring == NULL)
Benjamin Petersonead6b532011-12-20 17:23:42 -06009925 switch (PyUnicode_KIND(self)) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009926 case PyUnicode_1BYTE_KIND:
Victor Stinnerc3cec782011-10-05 21:24:08 +02009927 if (PyUnicode_IS_ASCII(self))
9928 return asciilib_rsplit_whitespace(
Victor Stinner7931d9a2011-11-04 00:22:48 +01009929 self, PyUnicode_1BYTE_DATA(self),
Victor Stinnerc3cec782011-10-05 21:24:08 +02009930 PyUnicode_GET_LENGTH(self), maxcount
9931 );
9932 else
9933 return ucs1lib_rsplit_whitespace(
Victor Stinner7931d9a2011-11-04 00:22:48 +01009934 self, PyUnicode_1BYTE_DATA(self),
Victor Stinnerc3cec782011-10-05 21:24:08 +02009935 PyUnicode_GET_LENGTH(self), maxcount
9936 );
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009937 case PyUnicode_2BYTE_KIND:
9938 return ucs2lib_rsplit_whitespace(
Victor Stinner7931d9a2011-11-04 00:22:48 +01009939 self, PyUnicode_2BYTE_DATA(self),
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009940 PyUnicode_GET_LENGTH(self), maxcount
9941 );
9942 case PyUnicode_4BYTE_KIND:
9943 return ucs4lib_rsplit_whitespace(
Victor Stinner7931d9a2011-11-04 00:22:48 +01009944 self, PyUnicode_4BYTE_DATA(self),
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009945 PyUnicode_GET_LENGTH(self), maxcount
9946 );
9947 default:
9948 assert(0);
9949 return NULL;
9950 }
9951
9952 if (PyUnicode_READY(substring) == -1)
9953 return NULL;
9954
9955 kind1 = PyUnicode_KIND(self);
9956 kind2 = PyUnicode_KIND(substring);
9957 kind = kind1 > kind2 ? kind1 : kind2;
9958 buf1 = PyUnicode_DATA(self);
9959 buf2 = PyUnicode_DATA(substring);
9960 if (kind1 != kind)
Victor Stinner7931d9a2011-11-04 00:22:48 +01009961 buf1 = _PyUnicode_AsKind(self, kind);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009962 if (!buf1)
9963 return NULL;
9964 if (kind2 != kind)
Victor Stinner7931d9a2011-11-04 00:22:48 +01009965 buf2 = _PyUnicode_AsKind(substring, kind);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009966 if (!buf2) {
9967 if (kind1 != kind) PyMem_Free(buf1);
9968 return NULL;
9969 }
9970 len1 = PyUnicode_GET_LENGTH(self);
9971 len2 = PyUnicode_GET_LENGTH(substring);
9972
Benjamin Petersonead6b532011-12-20 17:23:42 -06009973 switch (kind) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009974 case PyUnicode_1BYTE_KIND:
Victor Stinnerc3cec782011-10-05 21:24:08 +02009975 if (PyUnicode_IS_ASCII(self) && PyUnicode_IS_ASCII(substring))
9976 out = asciilib_rsplit(
Victor Stinner7931d9a2011-11-04 00:22:48 +01009977 self, buf1, len1, buf2, len2, maxcount);
Victor Stinnerc3cec782011-10-05 21:24:08 +02009978 else
9979 out = ucs1lib_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 case PyUnicode_2BYTE_KIND:
9983 out = ucs2lib_rsplit(
Victor Stinner7931d9a2011-11-04 00:22:48 +01009984 self, buf1, len1, buf2, len2, maxcount);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009985 break;
9986 case PyUnicode_4BYTE_KIND:
9987 out = ucs4lib_rsplit(
Victor Stinner7931d9a2011-11-04 00:22:48 +01009988 self, buf1, len1, buf2, len2, maxcount);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009989 break;
9990 default:
9991 out = NULL;
9992 }
9993 if (kind1 != kind)
9994 PyMem_Free(buf1);
9995 if (kind2 != kind)
9996 PyMem_Free(buf2);
9997 return out;
9998}
9999
10000static Py_ssize_t
Victor Stinnerc3cec782011-10-05 21:24:08 +020010001anylib_find(int kind, PyObject *str1, void *buf1, Py_ssize_t len1,
10002 PyObject *str2, void *buf2, Py_ssize_t len2, Py_ssize_t offset)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010003{
Benjamin Petersonead6b532011-12-20 17:23:42 -060010004 switch (kind) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010005 case PyUnicode_1BYTE_KIND:
Victor Stinnerc3cec782011-10-05 21:24:08 +020010006 if (PyUnicode_IS_ASCII(str1) && PyUnicode_IS_ASCII(str2))
10007 return asciilib_find(buf1, len1, buf2, len2, offset);
10008 else
10009 return ucs1lib_find(buf1, len1, buf2, len2, offset);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010010 case PyUnicode_2BYTE_KIND:
10011 return ucs2lib_find(buf1, len1, buf2, len2, offset);
10012 case PyUnicode_4BYTE_KIND:
10013 return ucs4lib_find(buf1, len1, buf2, len2, offset);
10014 }
10015 assert(0);
10016 return -1;
10017}
10018
10019static Py_ssize_t
Victor Stinnerc3cec782011-10-05 21:24:08 +020010020anylib_count(int kind, PyObject *sstr, void* sbuf, Py_ssize_t slen,
10021 PyObject *str1, void *buf1, Py_ssize_t len1, Py_ssize_t maxcount)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010022{
Benjamin Petersonc0b95d12011-12-20 17:24:05 -060010023 switch (kind) {
10024 case PyUnicode_1BYTE_KIND:
10025 if (PyUnicode_IS_ASCII(sstr) && PyUnicode_IS_ASCII(str1))
10026 return asciilib_count(sbuf, slen, buf1, len1, maxcount);
10027 else
10028 return ucs1lib_count(sbuf, slen, buf1, len1, maxcount);
10029 case PyUnicode_2BYTE_KIND:
10030 return ucs2lib_count(sbuf, slen, buf1, len1, maxcount);
10031 case PyUnicode_4BYTE_KIND:
10032 return ucs4lib_count(sbuf, slen, buf1, len1, maxcount);
10033 }
10034 assert(0);
10035 return 0;
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +000010036}
10037
Alexander Belopolsky40018472011-02-26 01:02:56 +000010038static PyObject *
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010039replace(PyObject *self, PyObject *str1,
10040 PyObject *str2, Py_ssize_t maxcount)
Guido van Rossumd57fd912000-03-10 22:53:23 +000010041{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010042 PyObject *u;
10043 char *sbuf = PyUnicode_DATA(self);
10044 char *buf1 = PyUnicode_DATA(str1);
10045 char *buf2 = PyUnicode_DATA(str2);
10046 int srelease = 0, release1 = 0, release2 = 0;
10047 int skind = PyUnicode_KIND(self);
10048 int kind1 = PyUnicode_KIND(str1);
10049 int kind2 = PyUnicode_KIND(str2);
10050 Py_ssize_t slen = PyUnicode_GET_LENGTH(self);
10051 Py_ssize_t len1 = PyUnicode_GET_LENGTH(str1);
10052 Py_ssize_t len2 = PyUnicode_GET_LENGTH(str2);
Victor Stinner49a0a212011-10-12 23:46:10 +020010053 int mayshrink;
10054 Py_UCS4 maxchar, maxchar_str2;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010055
10056 if (maxcount < 0)
Benjamin Peterson29060642009-01-31 22:14:21 +000010057 maxcount = PY_SSIZE_T_MAX;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010058 else if (maxcount == 0 || slen == 0)
Antoine Pitrouf2c54842010-01-13 08:07:53 +000010059 goto nothing;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010060
Victor Stinner59de0ee2011-10-07 10:01:28 +020010061 if (str1 == str2)
10062 goto nothing;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010063 if (skind < kind1)
10064 /* substring too wide to be present */
10065 goto nothing;
10066
Victor Stinner49a0a212011-10-12 23:46:10 +020010067 maxchar = PyUnicode_MAX_CHAR_VALUE(self);
10068 maxchar_str2 = PyUnicode_MAX_CHAR_VALUE(str2);
10069 /* Replacing str1 with str2 may cause a maxchar reduction in the
10070 result string. */
10071 mayshrink = (maxchar_str2 < maxchar);
Victor Stinnere6abb482012-05-02 01:15:40 +020010072 maxchar = MAX_MAXCHAR(maxchar, maxchar_str2);
Victor Stinner49a0a212011-10-12 23:46:10 +020010073
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010074 if (len1 == len2) {
Thomas Wouters477c8d52006-05-27 19:21:47 +000010075 /* same length */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010076 if (len1 == 0)
Antoine Pitrouf2c54842010-01-13 08:07:53 +000010077 goto nothing;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010078 if (len1 == 1) {
Thomas Wouters477c8d52006-05-27 19:21:47 +000010079 /* replace characters */
Victor Stinner49a0a212011-10-12 23:46:10 +020010080 Py_UCS4 u1, u2;
10081 int rkind;
Victor Stinnerf6441102011-12-18 02:43:08 +010010082 Py_ssize_t index, pos;
10083 char *src;
10084
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010085 u1 = PyUnicode_READ_CHAR(str1, 0);
Victor Stinnerf6441102011-12-18 02:43:08 +010010086 pos = findchar(sbuf, PyUnicode_KIND(self), slen, u1, 1);
10087 if (pos < 0)
Thomas Wouters477c8d52006-05-27 19:21:47 +000010088 goto nothing;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010089 u2 = PyUnicode_READ_CHAR(str2, 0);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010090 u = PyUnicode_New(slen, maxchar);
Thomas Wouters477c8d52006-05-27 19:21:47 +000010091 if (!u)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010092 goto error;
Victor Stinnerd3f08822012-05-29 12:57:52 +020010093 _PyUnicode_FastCopyCharacters(u, 0, self, 0, slen);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010094 rkind = PyUnicode_KIND(u);
Victor Stinnerf6441102011-12-18 02:43:08 +010010095
10096 PyUnicode_WRITE(rkind, PyUnicode_DATA(u), pos, u2);
10097 index = 0;
10098 src = sbuf;
10099 while (--maxcount)
10100 {
10101 pos++;
10102 src += pos * PyUnicode_KIND(self);
10103 slen -= pos;
10104 index += pos;
10105 pos = findchar(src, PyUnicode_KIND(self), slen, u1, 1);
10106 if (pos < 0)
10107 break;
10108 PyUnicode_WRITE(rkind, PyUnicode_DATA(u), index + pos, u2);
10109 }
Victor Stinner49a0a212011-10-12 23:46:10 +020010110 }
10111 else {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010112 int rkind = skind;
10113 char *res;
Victor Stinnerf6441102011-12-18 02:43:08 +010010114 Py_ssize_t i;
Victor Stinner25a4b292011-10-06 12:31:55 +020010115
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010116 if (kind1 < rkind) {
10117 /* widen substring */
10118 buf1 = _PyUnicode_AsKind(str1, rkind);
10119 if (!buf1) goto error;
10120 release1 = 1;
10121 }
Victor Stinnerc3cec782011-10-05 21:24:08 +020010122 i = anylib_find(rkind, self, sbuf, slen, str1, buf1, len1, 0);
Thomas Wouters477c8d52006-05-27 19:21:47 +000010123 if (i < 0)
10124 goto nothing;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010125 if (rkind > kind2) {
10126 /* widen replacement */
10127 buf2 = _PyUnicode_AsKind(str2, rkind);
10128 if (!buf2) goto error;
10129 release2 = 1;
10130 }
10131 else if (rkind < kind2) {
10132 /* widen self and buf1 */
10133 rkind = kind2;
10134 if (release1) PyMem_Free(buf1);
10135 sbuf = _PyUnicode_AsKind(self, rkind);
10136 if (!sbuf) goto error;
10137 srelease = 1;
10138 buf1 = _PyUnicode_AsKind(str1, rkind);
10139 if (!buf1) goto error;
10140 release1 = 1;
10141 }
Victor Stinner49a0a212011-10-12 23:46:10 +020010142 u = PyUnicode_New(slen, maxchar);
10143 if (!u)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010144 goto error;
Victor Stinner49a0a212011-10-12 23:46:10 +020010145 assert(PyUnicode_KIND(u) == rkind);
10146 res = PyUnicode_DATA(u);
Victor Stinner25a4b292011-10-06 12:31:55 +020010147
Martin v. Löwisc47adb02011-10-07 20:55:35 +020010148 memcpy(res, sbuf, rkind * slen);
Antoine Pitrouf2c54842010-01-13 08:07:53 +000010149 /* change everything in-place, starting with this one */
Martin v. Löwisc47adb02011-10-07 20:55:35 +020010150 memcpy(res + rkind * i,
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010151 buf2,
Martin v. Löwisc47adb02011-10-07 20:55:35 +020010152 rkind * len2);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010153 i += len1;
Antoine Pitrouf2c54842010-01-13 08:07:53 +000010154
10155 while ( --maxcount > 0) {
Victor Stinnerc3cec782011-10-05 21:24:08 +020010156 i = anylib_find(rkind, self,
Martin v. Löwisc47adb02011-10-07 20:55:35 +020010157 sbuf+rkind*i, slen-i,
Victor Stinnerc3cec782011-10-05 21:24:08 +020010158 str1, buf1, len1, i);
Antoine Pitrouf2c54842010-01-13 08:07:53 +000010159 if (i == -1)
10160 break;
Martin v. Löwisc47adb02011-10-07 20:55:35 +020010161 memcpy(res + rkind * i,
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010162 buf2,
Martin v. Löwisc47adb02011-10-07 20:55:35 +020010163 rkind * len2);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010164 i += len1;
Antoine Pitrouf2c54842010-01-13 08:07:53 +000010165 }
Guido van Rossumd57fd912000-03-10 22:53:23 +000010166 }
Victor Stinner49a0a212011-10-12 23:46:10 +020010167 }
10168 else {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010169 Py_ssize_t n, i, j, ires;
10170 Py_ssize_t product, new_size;
10171 int rkind = skind;
10172 char *res;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010173
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010174 if (kind1 < rkind) {
Victor Stinner49a0a212011-10-12 23:46:10 +020010175 /* widen substring */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010176 buf1 = _PyUnicode_AsKind(str1, rkind);
10177 if (!buf1) goto error;
10178 release1 = 1;
10179 }
Victor Stinnerc3cec782011-10-05 21:24:08 +020010180 n = anylib_count(rkind, self, sbuf, slen, str1, buf1, len1, maxcount);
Thomas Wouters477c8d52006-05-27 19:21:47 +000010181 if (n == 0)
10182 goto nothing;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010183 if (kind2 < rkind) {
Victor Stinner49a0a212011-10-12 23:46:10 +020010184 /* widen replacement */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010185 buf2 = _PyUnicode_AsKind(str2, rkind);
10186 if (!buf2) goto error;
10187 release2 = 1;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010188 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010189 else if (kind2 > rkind) {
Victor Stinner49a0a212011-10-12 23:46:10 +020010190 /* widen self and buf1 */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010191 rkind = kind2;
10192 sbuf = _PyUnicode_AsKind(self, rkind);
10193 if (!sbuf) goto error;
10194 srelease = 1;
10195 if (release1) PyMem_Free(buf1);
10196 buf1 = _PyUnicode_AsKind(str1, rkind);
10197 if (!buf1) goto error;
10198 release1 = 1;
10199 }
10200 /* new_size = PyUnicode_GET_LENGTH(self) + n * (PyUnicode_GET_LENGTH(str2) -
10201 PyUnicode_GET_LENGTH(str1))); */
10202 product = n * (len2-len1);
10203 if ((product / (len2-len1)) != n) {
10204 PyErr_SetString(PyExc_OverflowError,
10205 "replace string is too long");
10206 goto error;
10207 }
10208 new_size = slen + product;
Victor Stinner49a0a212011-10-12 23:46:10 +020010209 if (new_size == 0) {
10210 Py_INCREF(unicode_empty);
10211 u = unicode_empty;
10212 goto done;
10213 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010214 if (new_size < 0 || new_size > (PY_SSIZE_T_MAX >> (rkind-1))) {
10215 PyErr_SetString(PyExc_OverflowError,
10216 "replace string is too long");
10217 goto error;
10218 }
Victor Stinner49a0a212011-10-12 23:46:10 +020010219 u = PyUnicode_New(new_size, maxchar);
10220 if (!u)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010221 goto error;
Victor Stinner49a0a212011-10-12 23:46:10 +020010222 assert(PyUnicode_KIND(u) == rkind);
10223 res = PyUnicode_DATA(u);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010224 ires = i = 0;
10225 if (len1 > 0) {
Thomas Wouters477c8d52006-05-27 19:21:47 +000010226 while (n-- > 0) {
10227 /* look for next match */
Victor Stinnerc3cec782011-10-05 21:24:08 +020010228 j = anylib_find(rkind, self,
Martin v. Löwisc47adb02011-10-07 20:55:35 +020010229 sbuf + rkind * i, slen-i,
Victor Stinnerc3cec782011-10-05 21:24:08 +020010230 str1, buf1, len1, i);
Antoine Pitrouf2c54842010-01-13 08:07:53 +000010231 if (j == -1)
10232 break;
10233 else if (j > i) {
Thomas Wouters477c8d52006-05-27 19:21:47 +000010234 /* copy unchanged part [i:j] */
Martin v. Löwisc47adb02011-10-07 20:55:35 +020010235 memcpy(res + rkind * ires,
10236 sbuf + rkind * i,
10237 rkind * (j-i));
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010238 ires += j - i;
Thomas Wouters477c8d52006-05-27 19:21:47 +000010239 }
10240 /* copy substitution string */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010241 if (len2 > 0) {
Martin v. Löwisc47adb02011-10-07 20:55:35 +020010242 memcpy(res + rkind * ires,
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010243 buf2,
Martin v. Löwisc47adb02011-10-07 20:55:35 +020010244 rkind * len2);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010245 ires += len2;
Thomas Wouters477c8d52006-05-27 19:21:47 +000010246 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010247 i = j + len1;
Thomas Wouters477c8d52006-05-27 19:21:47 +000010248 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010249 if (i < slen)
Thomas Wouters477c8d52006-05-27 19:21:47 +000010250 /* copy tail [i:] */
Martin v. Löwisc47adb02011-10-07 20:55:35 +020010251 memcpy(res + rkind * ires,
10252 sbuf + rkind * i,
10253 rkind * (slen-i));
Victor Stinner49a0a212011-10-12 23:46:10 +020010254 }
10255 else {
Thomas Wouters477c8d52006-05-27 19:21:47 +000010256 /* interleave */
10257 while (n > 0) {
Martin v. Löwisc47adb02011-10-07 20:55:35 +020010258 memcpy(res + rkind * ires,
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010259 buf2,
Martin v. Löwisc47adb02011-10-07 20:55:35 +020010260 rkind * len2);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010261 ires += len2;
Thomas Wouters477c8d52006-05-27 19:21:47 +000010262 if (--n <= 0)
10263 break;
Martin v. Löwisc47adb02011-10-07 20:55:35 +020010264 memcpy(res + rkind * ires,
10265 sbuf + rkind * i,
10266 rkind);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010267 ires++;
10268 i++;
Thomas Wouters477c8d52006-05-27 19:21:47 +000010269 }
Martin v. Löwisc47adb02011-10-07 20:55:35 +020010270 memcpy(res + rkind * ires,
10271 sbuf + rkind * i,
10272 rkind * (slen-i));
Thomas Wouters477c8d52006-05-27 19:21:47 +000010273 }
Victor Stinner49a0a212011-10-12 23:46:10 +020010274 }
10275
10276 if (mayshrink) {
Victor Stinner25a4b292011-10-06 12:31:55 +020010277 unicode_adjust_maxchar(&u);
10278 if (u == NULL)
10279 goto error;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010280 }
Victor Stinner49a0a212011-10-12 23:46:10 +020010281
10282 done:
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010283 if (srelease)
10284 PyMem_FREE(sbuf);
10285 if (release1)
10286 PyMem_FREE(buf1);
10287 if (release2)
10288 PyMem_FREE(buf2);
Victor Stinnerbb10a1f2011-10-05 01:34:17 +020010289 assert(_PyUnicode_CheckConsistency(u, 1));
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010290 return u;
Thomas Wouters477c8d52006-05-27 19:21:47 +000010291
Benjamin Peterson29060642009-01-31 22:14:21 +000010292 nothing:
Thomas Wouters477c8d52006-05-27 19:21:47 +000010293 /* nothing to replace; return original string (when possible) */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010294 if (srelease)
10295 PyMem_FREE(sbuf);
10296 if (release1)
10297 PyMem_FREE(buf1);
10298 if (release2)
10299 PyMem_FREE(buf2);
Victor Stinnerc4b49542011-12-11 22:44:26 +010010300 return unicode_result_unchanged(self);
10301
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010302 error:
10303 if (srelease && sbuf)
10304 PyMem_FREE(sbuf);
10305 if (release1 && buf1)
10306 PyMem_FREE(buf1);
10307 if (release2 && buf2)
10308 PyMem_FREE(buf2);
10309 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010310}
10311
10312/* --- Unicode Object Methods --------------------------------------------- */
10313
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000010314PyDoc_STRVAR(title__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000010315 "S.title() -> str\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000010316\n\
10317Return a titlecased version of S, i.e. words start with title case\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000010318characters, all remaining cased characters have lower case.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000010319
10320static PyObject*
Victor Stinner9310abb2011-10-05 00:59:23 +020010321unicode_title(PyObject *self)
Guido van Rossumd57fd912000-03-10 22:53:23 +000010322{
Benjamin Petersoneea48462012-01-16 14:28:50 -050010323 if (PyUnicode_READY(self) == -1)
10324 return NULL;
Victor Stinnerb0800dc2012-02-25 00:47:08 +010010325 return case_operation(self, do_title);
Guido van Rossumd57fd912000-03-10 22:53:23 +000010326}
10327
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000010328PyDoc_STRVAR(capitalize__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000010329 "S.capitalize() -> str\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000010330\n\
10331Return a capitalized version of S, i.e. make the first character\n\
Senthil Kumarane51ee8a2010-07-05 12:00:56 +000010332have upper case and the rest lower case.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000010333
10334static PyObject*
Victor Stinner9310abb2011-10-05 00:59:23 +020010335unicode_capitalize(PyObject *self)
Guido van Rossumd57fd912000-03-10 22:53:23 +000010336{
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -050010337 if (PyUnicode_READY(self) == -1)
10338 return NULL;
10339 if (PyUnicode_GET_LENGTH(self) == 0)
10340 return unicode_result_unchanged(self);
Victor Stinnerb0800dc2012-02-25 00:47:08 +010010341 return case_operation(self, do_capitalize);
Guido van Rossumd57fd912000-03-10 22:53:23 +000010342}
10343
Benjamin Petersond5890c82012-01-14 13:23:30 -050010344PyDoc_STRVAR(casefold__doc__,
10345 "S.casefold() -> str\n\
10346\n\
10347Return a version of S suitable for caseless comparisons.");
10348
10349static PyObject *
10350unicode_casefold(PyObject *self)
10351{
10352 if (PyUnicode_READY(self) == -1)
10353 return NULL;
10354 if (PyUnicode_IS_ASCII(self))
10355 return ascii_upper_or_lower(self, 1);
Victor Stinnerb0800dc2012-02-25 00:47:08 +010010356 return case_operation(self, do_casefold);
Benjamin Petersond5890c82012-01-14 13:23:30 -050010357}
10358
10359
Raymond Hettinger4f8f9762003-11-26 08:21:35 +000010360/* Argument converter. Coerces to a single unicode character */
10361
10362static int
10363convert_uc(PyObject *obj, void *addr)
10364{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010365 Py_UCS4 *fillcharloc = (Py_UCS4 *)addr;
Benjamin Peterson14339b62009-01-31 16:36:08 +000010366 PyObject *uniobj;
Raymond Hettinger4f8f9762003-11-26 08:21:35 +000010367
Benjamin Peterson14339b62009-01-31 16:36:08 +000010368 uniobj = PyUnicode_FromObject(obj);
10369 if (uniobj == NULL) {
10370 PyErr_SetString(PyExc_TypeError,
Benjamin Peterson29060642009-01-31 22:14:21 +000010371 "The fill character cannot be converted to Unicode");
Benjamin Peterson14339b62009-01-31 16:36:08 +000010372 return 0;
10373 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010374 if (PyUnicode_GET_LENGTH(uniobj) != 1) {
Benjamin Peterson14339b62009-01-31 16:36:08 +000010375 PyErr_SetString(PyExc_TypeError,
Benjamin Peterson29060642009-01-31 22:14:21 +000010376 "The fill character must be exactly one character long");
Benjamin Peterson14339b62009-01-31 16:36:08 +000010377 Py_DECREF(uniobj);
10378 return 0;
10379 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010380 *fillcharloc = PyUnicode_READ_CHAR(uniobj, 0);
Benjamin Peterson14339b62009-01-31 16:36:08 +000010381 Py_DECREF(uniobj);
10382 return 1;
Raymond Hettinger4f8f9762003-11-26 08:21:35 +000010383}
10384
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000010385PyDoc_STRVAR(center__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000010386 "S.center(width[, fillchar]) -> str\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000010387\n\
Benjamin Peterson142957c2008-07-04 19:55:29 +000010388Return S centered in a string of length width. Padding is\n\
Raymond Hettinger4f8f9762003-11-26 08:21:35 +000010389done using the specified fill character (default is a space)");
Guido van Rossumd57fd912000-03-10 22:53:23 +000010390
10391static PyObject *
Victor Stinner9310abb2011-10-05 00:59:23 +020010392unicode_center(PyObject *self, PyObject *args)
Guido van Rossumd57fd912000-03-10 22:53:23 +000010393{
Martin v. Löwis18e16552006-02-15 17:27:45 +000010394 Py_ssize_t marg, left;
10395 Py_ssize_t width;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010396 Py_UCS4 fillchar = ' ';
10397
Victor Stinnere9a29352011-10-01 02:14:59 +020010398 if (!PyArg_ParseTuple(args, "n|O&:center", &width, convert_uc, &fillchar))
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010399 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010400
Benjamin Petersonbac79492012-01-14 13:34:47 -050010401 if (PyUnicode_READY(self) == -1)
Guido van Rossumd57fd912000-03-10 22:53:23 +000010402 return NULL;
10403
Victor Stinnerc4b49542011-12-11 22:44:26 +010010404 if (PyUnicode_GET_LENGTH(self) >= width)
10405 return unicode_result_unchanged(self);
Guido van Rossumd57fd912000-03-10 22:53:23 +000010406
Victor Stinnerc4b49542011-12-11 22:44:26 +010010407 marg = width - PyUnicode_GET_LENGTH(self);
Guido van Rossumd57fd912000-03-10 22:53:23 +000010408 left = marg / 2 + (marg & width & 1);
10409
Victor Stinner9310abb2011-10-05 00:59:23 +020010410 return pad(self, left, marg - left, fillchar);
Guido van Rossumd57fd912000-03-10 22:53:23 +000010411}
10412
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010413/* This function assumes that str1 and str2 are readied by the caller. */
10414
Marc-André Lemburge5034372000-08-08 08:04:29 +000010415static int
Victor Stinner9db1a8b2011-10-23 20:04:37 +020010416unicode_compare(PyObject *str1, PyObject *str2)
Marc-André Lemburge5034372000-08-08 08:04:29 +000010417{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010418 int kind1, kind2;
10419 void *data1, *data2;
10420 Py_ssize_t len1, len2, i;
Marc-André Lemburge5034372000-08-08 08:04:29 +000010421
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010422 kind1 = PyUnicode_KIND(str1);
10423 kind2 = PyUnicode_KIND(str2);
10424 data1 = PyUnicode_DATA(str1);
10425 data2 = PyUnicode_DATA(str2);
10426 len1 = PyUnicode_GET_LENGTH(str1);
10427 len2 = PyUnicode_GET_LENGTH(str2);
Marc-André Lemburge5034372000-08-08 08:04:29 +000010428
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010429 for (i = 0; i < len1 && i < len2; ++i) {
10430 Py_UCS4 c1, c2;
10431 c1 = PyUnicode_READ(kind1, data1, i);
10432 c2 = PyUnicode_READ(kind2, data2, i);
Fredrik Lundh45714e92001-06-26 16:39:36 +000010433
10434 if (c1 != c2)
10435 return (c1 < c2) ? -1 : 1;
Marc-André Lemburge5034372000-08-08 08:04:29 +000010436 }
10437
10438 return (len1 < len2) ? -1 : (len1 != len2);
10439}
10440
Alexander Belopolsky40018472011-02-26 01:02:56 +000010441int
10442PyUnicode_Compare(PyObject *left, PyObject *right)
Guido van Rossumd57fd912000-03-10 22:53:23 +000010443{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010444 if (PyUnicode_Check(left) && PyUnicode_Check(right)) {
10445 if (PyUnicode_READY(left) == -1 ||
10446 PyUnicode_READY(right) == -1)
10447 return -1;
Victor Stinner9db1a8b2011-10-23 20:04:37 +020010448 return unicode_compare(left, right);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010449 }
Guido van Rossum09dc34f2007-05-04 04:17:33 +000010450 PyErr_Format(PyExc_TypeError,
10451 "Can't compare %.100s and %.100s",
10452 left->ob_type->tp_name,
10453 right->ob_type->tp_name);
Guido van Rossumd57fd912000-03-10 22:53:23 +000010454 return -1;
10455}
10456
Martin v. Löwis5b222132007-06-10 09:51:05 +000010457int
10458PyUnicode_CompareWithASCIIString(PyObject* uni, const char* str)
10459{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010460 Py_ssize_t i;
10461 int kind;
10462 void *data;
10463 Py_UCS4 chr;
10464
Victor Stinner910337b2011-10-03 03:20:16 +020010465 assert(_PyUnicode_CHECK(uni));
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010466 if (PyUnicode_READY(uni) == -1)
10467 return -1;
10468 kind = PyUnicode_KIND(uni);
10469 data = PyUnicode_DATA(uni);
Martin v. Löwis5b222132007-06-10 09:51:05 +000010470 /* Compare Unicode string and source character set string */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010471 for (i = 0; (chr = PyUnicode_READ(kind, data, i)) && str[i]; i++)
10472 if (chr != str[i])
10473 return (chr < (unsigned char)(str[i])) ? -1 : 1;
Benjamin Peterson8667a9b2010-01-09 21:45:28 +000010474 /* This check keeps Python strings that end in '\0' from comparing equal
10475 to C strings identical up to that point. */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010476 if (PyUnicode_GET_LENGTH(uni) != i || chr)
Benjamin Peterson29060642009-01-31 22:14:21 +000010477 return 1; /* uni is longer */
Martin v. Löwis5b222132007-06-10 09:51:05 +000010478 if (str[i])
Benjamin Peterson29060642009-01-31 22:14:21 +000010479 return -1; /* str is longer */
Martin v. Löwis5b222132007-06-10 09:51:05 +000010480 return 0;
10481}
10482
Antoine Pitrou51f3ef92008-12-20 13:14:23 +000010483
Benjamin Peterson29060642009-01-31 22:14:21 +000010484#define TEST_COND(cond) \
Benjamin Peterson14339b62009-01-31 16:36:08 +000010485 ((cond) ? Py_True : Py_False)
Antoine Pitrou51f3ef92008-12-20 13:14:23 +000010486
Alexander Belopolsky40018472011-02-26 01:02:56 +000010487PyObject *
10488PyUnicode_RichCompare(PyObject *left, PyObject *right, int op)
Thomas Wouters00ee7ba2006-08-21 19:07:27 +000010489{
10490 int result;
Benjamin Peterson14339b62009-01-31 16:36:08 +000010491
Antoine Pitrou51f3ef92008-12-20 13:14:23 +000010492 if (PyUnicode_Check(left) && PyUnicode_Check(right)) {
10493 PyObject *v;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010494 if (PyUnicode_READY(left) == -1 ||
10495 PyUnicode_READY(right) == -1)
10496 return NULL;
10497 if (PyUnicode_GET_LENGTH(left) != PyUnicode_GET_LENGTH(right) ||
10498 PyUnicode_KIND(left) != PyUnicode_KIND(right)) {
Antoine Pitrou51f3ef92008-12-20 13:14:23 +000010499 if (op == Py_EQ) {
10500 Py_INCREF(Py_False);
10501 return Py_False;
10502 }
10503 if (op == Py_NE) {
10504 Py_INCREF(Py_True);
10505 return Py_True;
10506 }
10507 }
10508 if (left == right)
10509 result = 0;
10510 else
Victor Stinner9db1a8b2011-10-23 20:04:37 +020010511 result = unicode_compare(left, right);
Benjamin Peterson14339b62009-01-31 16:36:08 +000010512
Antoine Pitrou51f3ef92008-12-20 13:14:23 +000010513 /* Convert the return value to a Boolean */
10514 switch (op) {
10515 case Py_EQ:
10516 v = TEST_COND(result == 0);
10517 break;
10518 case Py_NE:
10519 v = TEST_COND(result != 0);
10520 break;
10521 case Py_LE:
10522 v = TEST_COND(result <= 0);
10523 break;
10524 case Py_GE:
10525 v = TEST_COND(result >= 0);
10526 break;
10527 case Py_LT:
10528 v = TEST_COND(result == -1);
10529 break;
10530 case Py_GT:
10531 v = TEST_COND(result == 1);
10532 break;
10533 default:
10534 PyErr_BadArgument();
10535 return NULL;
10536 }
10537 Py_INCREF(v);
10538 return v;
Thomas Wouters00ee7ba2006-08-21 19:07:27 +000010539 }
Benjamin Peterson14339b62009-01-31 16:36:08 +000010540
Brian Curtindfc80e32011-08-10 20:28:54 -050010541 Py_RETURN_NOTIMPLEMENTED;
Thomas Wouters00ee7ba2006-08-21 19:07:27 +000010542}
10543
Alexander Belopolsky40018472011-02-26 01:02:56 +000010544int
10545PyUnicode_Contains(PyObject *container, PyObject *element)
Guido van Rossum403d68b2000-03-13 15:55:09 +000010546{
Thomas Wouters477c8d52006-05-27 19:21:47 +000010547 PyObject *str, *sub;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010548 int kind1, kind2, kind;
10549 void *buf1, *buf2;
10550 Py_ssize_t len1, len2;
Martin v. Löwis18e16552006-02-15 17:27:45 +000010551 int result;
Guido van Rossum403d68b2000-03-13 15:55:09 +000010552
10553 /* Coerce the two arguments */
Thomas Wouters477c8d52006-05-27 19:21:47 +000010554 sub = PyUnicode_FromObject(element);
10555 if (!sub) {
Benjamin Peterson29060642009-01-31 22:14:21 +000010556 PyErr_Format(PyExc_TypeError,
10557 "'in <string>' requires string as left operand, not %s",
10558 element->ob_type->tp_name);
Thomas Wouters477c8d52006-05-27 19:21:47 +000010559 return -1;
Guido van Rossum403d68b2000-03-13 15:55:09 +000010560 }
10561
Thomas Wouters477c8d52006-05-27 19:21:47 +000010562 str = PyUnicode_FromObject(container);
Benjamin Peterson22a29702012-01-02 09:00:30 -060010563 if (!str) {
Thomas Wouters477c8d52006-05-27 19:21:47 +000010564 Py_DECREF(sub);
10565 return -1;
10566 }
Benjamin Peterson22a29702012-01-02 09:00:30 -060010567 if (PyUnicode_READY(sub) == -1 || PyUnicode_READY(str) == -1) {
10568 Py_DECREF(sub);
10569 Py_DECREF(str);
10570 }
Thomas Wouters477c8d52006-05-27 19:21:47 +000010571
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010572 kind1 = PyUnicode_KIND(str);
10573 kind2 = PyUnicode_KIND(sub);
Benjamin Peterson1ff2e352012-05-11 17:41:20 -050010574 kind = kind1;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010575 buf1 = PyUnicode_DATA(str);
10576 buf2 = PyUnicode_DATA(sub);
Benjamin Peterson1ff2e352012-05-11 17:41:20 -050010577 if (kind2 != kind) {
Antoine Pitrou758153b2012-05-12 15:51:51 +020010578 if (kind2 > kind) {
10579 Py_DECREF(sub);
10580 Py_DECREF(str);
Benjamin Peterson1ff2e352012-05-11 17:41:20 -050010581 return 0;
Antoine Pitrou758153b2012-05-12 15:51:51 +020010582 }
Victor Stinner7931d9a2011-11-04 00:22:48 +010010583 buf2 = _PyUnicode_AsKind(sub, kind);
Benjamin Peterson1ff2e352012-05-11 17:41:20 -050010584 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010585 if (!buf2) {
10586 Py_DECREF(sub);
Benjamin Peterson1ff2e352012-05-11 17:41:20 -050010587 Py_DECREF(str);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010588 return -1;
10589 }
10590 len1 = PyUnicode_GET_LENGTH(str);
10591 len2 = PyUnicode_GET_LENGTH(sub);
10592
Benjamin Petersonead6b532011-12-20 17:23:42 -060010593 switch (kind) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010594 case PyUnicode_1BYTE_KIND:
10595 result = ucs1lib_find(buf1, len1, buf2, len2, 0) != -1;
10596 break;
10597 case PyUnicode_2BYTE_KIND:
10598 result = ucs2lib_find(buf1, len1, buf2, len2, 0) != -1;
10599 break;
10600 case PyUnicode_4BYTE_KIND:
10601 result = ucs4lib_find(buf1, len1, buf2, len2, 0) != -1;
10602 break;
10603 default:
10604 result = -1;
10605 assert(0);
10606 }
Thomas Wouters477c8d52006-05-27 19:21:47 +000010607
10608 Py_DECREF(str);
10609 Py_DECREF(sub);
10610
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010611 if (kind2 != kind)
10612 PyMem_Free(buf2);
10613
Guido van Rossum403d68b2000-03-13 15:55:09 +000010614 return result;
Guido van Rossum403d68b2000-03-13 15:55:09 +000010615}
10616
Guido van Rossumd57fd912000-03-10 22:53:23 +000010617/* Concat to string or Unicode object giving a new Unicode object. */
10618
Alexander Belopolsky40018472011-02-26 01:02:56 +000010619PyObject *
10620PyUnicode_Concat(PyObject *left, PyObject *right)
Guido van Rossumd57fd912000-03-10 22:53:23 +000010621{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010622 PyObject *u = NULL, *v = NULL, *w;
Victor Stinner127226b2011-10-13 01:12:34 +020010623 Py_UCS4 maxchar, maxchar2;
Victor Stinner488fa492011-12-12 00:01:39 +010010624 Py_ssize_t u_len, v_len, new_len;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010625
10626 /* Coerce the two arguments */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010627 u = PyUnicode_FromObject(left);
Guido van Rossumd57fd912000-03-10 22:53:23 +000010628 if (u == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +000010629 goto onError;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010630 v = PyUnicode_FromObject(right);
Guido van Rossumd57fd912000-03-10 22:53:23 +000010631 if (v == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +000010632 goto onError;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010633
10634 /* Shortcuts */
Victor Stinnera464fc12011-10-02 20:39:30 +020010635 if (v == unicode_empty) {
Benjamin Peterson29060642009-01-31 22:14:21 +000010636 Py_DECREF(v);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010637 return u;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010638 }
Victor Stinnera464fc12011-10-02 20:39:30 +020010639 if (u == unicode_empty) {
Benjamin Peterson29060642009-01-31 22:14:21 +000010640 Py_DECREF(u);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010641 return v;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010642 }
10643
Victor Stinner488fa492011-12-12 00:01:39 +010010644 u_len = PyUnicode_GET_LENGTH(u);
10645 v_len = PyUnicode_GET_LENGTH(v);
10646 if (u_len > PY_SSIZE_T_MAX - v_len) {
10647 PyErr_SetString(PyExc_OverflowError,
10648 "strings are too large to concat");
10649 goto onError;
10650 }
10651 new_len = u_len + v_len;
10652
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010653 maxchar = PyUnicode_MAX_CHAR_VALUE(u);
Victor Stinner127226b2011-10-13 01:12:34 +020010654 maxchar2 = PyUnicode_MAX_CHAR_VALUE(v);
Victor Stinnere6abb482012-05-02 01:15:40 +020010655 maxchar = MAX_MAXCHAR(maxchar, maxchar2);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010656
Guido van Rossumd57fd912000-03-10 22:53:23 +000010657 /* Concat the two Unicode strings */
Victor Stinner488fa492011-12-12 00:01:39 +010010658 w = PyUnicode_New(new_len, maxchar);
Guido van Rossumd57fd912000-03-10 22:53:23 +000010659 if (w == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +000010660 goto onError;
Victor Stinnerd3f08822012-05-29 12:57:52 +020010661 _PyUnicode_FastCopyCharacters(w, 0, u, 0, u_len);
10662 _PyUnicode_FastCopyCharacters(w, u_len, v, 0, v_len);
Guido van Rossumd57fd912000-03-10 22:53:23 +000010663 Py_DECREF(u);
10664 Py_DECREF(v);
Victor Stinnerbb10a1f2011-10-05 01:34:17 +020010665 assert(_PyUnicode_CheckConsistency(w, 1));
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010666 return w;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010667
Benjamin Peterson29060642009-01-31 22:14:21 +000010668 onError:
Guido van Rossumd57fd912000-03-10 22:53:23 +000010669 Py_XDECREF(u);
10670 Py_XDECREF(v);
10671 return NULL;
10672}
10673
Walter Dörwald1ab83302007-05-18 17:15:44 +000010674void
Victor Stinner23e56682011-10-03 03:54:37 +020010675PyUnicode_Append(PyObject **p_left, PyObject *right)
Walter Dörwald1ab83302007-05-18 17:15:44 +000010676{
Victor Stinner23e56682011-10-03 03:54:37 +020010677 PyObject *left, *res;
Victor Stinner488fa492011-12-12 00:01:39 +010010678 Py_UCS4 maxchar, maxchar2;
10679 Py_ssize_t left_len, right_len, new_len;
Victor Stinner23e56682011-10-03 03:54:37 +020010680
10681 if (p_left == NULL) {
10682 if (!PyErr_Occurred())
10683 PyErr_BadInternalCall();
Benjamin Peterson14339b62009-01-31 16:36:08 +000010684 return;
10685 }
Victor Stinner23e56682011-10-03 03:54:37 +020010686 left = *p_left;
10687 if (right == NULL || !PyUnicode_Check(left)) {
10688 if (!PyErr_Occurred())
10689 PyErr_BadInternalCall();
10690 goto error;
10691 }
10692
Benjamin Petersonbac79492012-01-14 13:34:47 -050010693 if (PyUnicode_READY(left) == -1)
Victor Stinnere1335c72011-10-04 20:53:03 +020010694 goto error;
Benjamin Petersonbac79492012-01-14 13:34:47 -050010695 if (PyUnicode_READY(right) == -1)
Victor Stinnere1335c72011-10-04 20:53:03 +020010696 goto error;
10697
Victor Stinner488fa492011-12-12 00:01:39 +010010698 /* Shortcuts */
10699 if (left == unicode_empty) {
10700 Py_DECREF(left);
10701 Py_INCREF(right);
10702 *p_left = right;
10703 return;
10704 }
10705 if (right == unicode_empty)
10706 return;
10707
10708 left_len = PyUnicode_GET_LENGTH(left);
10709 right_len = PyUnicode_GET_LENGTH(right);
10710 if (left_len > PY_SSIZE_T_MAX - right_len) {
10711 PyErr_SetString(PyExc_OverflowError,
10712 "strings are too large to concat");
10713 goto error;
10714 }
10715 new_len = left_len + right_len;
10716
10717 if (unicode_modifiable(left)
10718 && PyUnicode_CheckExact(right)
10719 && PyUnicode_KIND(right) <= PyUnicode_KIND(left)
Victor Stinnerb0923652011-10-04 01:17:31 +020010720 /* Don't resize for ascii += latin1. Convert ascii to latin1 requires
10721 to change the structure size, but characters are stored just after
Georg Brandl7597add2011-10-05 16:36:47 +020010722 the structure, and so it requires to move all characters which is
Victor Stinnerb0923652011-10-04 01:17:31 +020010723 not so different than duplicating the string. */
Victor Stinner488fa492011-12-12 00:01:39 +010010724 && !(PyUnicode_IS_ASCII(left) && !PyUnicode_IS_ASCII(right)))
10725 {
10726 /* append inplace */
10727 if (unicode_resize(p_left, new_len) != 0) {
10728 /* XXX if _PyUnicode_Resize() fails, 'left' has been
10729 * deallocated so it cannot be put back into
10730 * 'variable'. The MemoryError is raised when there
10731 * is no value in 'variable', which might (very
10732 * remotely) be a cause of incompatibilities.
10733 */
10734 goto error;
Victor Stinner23e56682011-10-03 03:54:37 +020010735 }
Victor Stinner488fa492011-12-12 00:01:39 +010010736 /* copy 'right' into the newly allocated area of 'left' */
Victor Stinnerd3f08822012-05-29 12:57:52 +020010737 _PyUnicode_FastCopyCharacters(*p_left, left_len, right, 0, right_len);
Victor Stinner23e56682011-10-03 03:54:37 +020010738 }
Victor Stinner488fa492011-12-12 00:01:39 +010010739 else {
10740 maxchar = PyUnicode_MAX_CHAR_VALUE(left);
10741 maxchar2 = PyUnicode_MAX_CHAR_VALUE(right);
Victor Stinnere6abb482012-05-02 01:15:40 +020010742 maxchar = MAX_MAXCHAR(maxchar, maxchar2);
Victor Stinner23e56682011-10-03 03:54:37 +020010743
Victor Stinner488fa492011-12-12 00:01:39 +010010744 /* Concat the two Unicode strings */
10745 res = PyUnicode_New(new_len, maxchar);
10746 if (res == NULL)
10747 goto error;
Victor Stinnerd3f08822012-05-29 12:57:52 +020010748 _PyUnicode_FastCopyCharacters(res, 0, left, 0, left_len);
10749 _PyUnicode_FastCopyCharacters(res, left_len, right, 0, right_len);
Victor Stinner488fa492011-12-12 00:01:39 +010010750 Py_DECREF(left);
10751 *p_left = res;
10752 }
10753 assert(_PyUnicode_CheckConsistency(*p_left, 1));
Victor Stinner23e56682011-10-03 03:54:37 +020010754 return;
10755
10756error:
Victor Stinner488fa492011-12-12 00:01:39 +010010757 Py_CLEAR(*p_left);
Walter Dörwald1ab83302007-05-18 17:15:44 +000010758}
10759
10760void
10761PyUnicode_AppendAndDel(PyObject **pleft, PyObject *right)
10762{
Benjamin Peterson14339b62009-01-31 16:36:08 +000010763 PyUnicode_Append(pleft, right);
10764 Py_XDECREF(right);
Walter Dörwald1ab83302007-05-18 17:15:44 +000010765}
10766
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000010767PyDoc_STRVAR(count__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000010768 "S.count(sub[, start[, end]]) -> int\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000010769\n\
Thomas Wouters477c8d52006-05-27 19:21:47 +000010770Return the number of non-overlapping occurrences of substring sub in\n\
Benjamin Peterson142957c2008-07-04 19:55:29 +000010771string S[start:end]. Optional arguments start and end are\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000010772interpreted as in slice notation.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000010773
10774static PyObject *
Victor Stinner9db1a8b2011-10-23 20:04:37 +020010775unicode_count(PyObject *self, PyObject *args)
Guido van Rossumd57fd912000-03-10 22:53:23 +000010776{
Victor Stinner9db1a8b2011-10-23 20:04:37 +020010777 PyObject *substring;
Martin v. Löwis18e16552006-02-15 17:27:45 +000010778 Py_ssize_t start = 0;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000010779 Py_ssize_t end = PY_SSIZE_T_MAX;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010780 PyObject *result;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010781 int kind1, kind2, kind;
10782 void *buf1, *buf2;
10783 Py_ssize_t len1, len2, iresult;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010784
Jesus Ceaac451502011-04-20 17:09:23 +020010785 if (!stringlib_parse_args_finds_unicode("count", args, &substring,
10786 &start, &end))
Benjamin Peterson29060642009-01-31 22:14:21 +000010787 return NULL;
Tim Petersced69f82003-09-16 20:30:58 +000010788
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010789 kind1 = PyUnicode_KIND(self);
10790 kind2 = PyUnicode_KIND(substring);
Benjamin Petersonb63f49f2012-05-03 18:31:07 -040010791 if (kind2 > kind1)
10792 return PyLong_FromLong(0);
10793 kind = kind1;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010794 buf1 = PyUnicode_DATA(self);
10795 buf2 = PyUnicode_DATA(substring);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010796 if (kind2 != kind)
Victor Stinner7931d9a2011-11-04 00:22:48 +010010797 buf2 = _PyUnicode_AsKind(substring, kind);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010798 if (!buf2) {
10799 Py_DECREF(substring);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010800 return NULL;
10801 }
10802 len1 = PyUnicode_GET_LENGTH(self);
10803 len2 = PyUnicode_GET_LENGTH(substring);
10804
10805 ADJUST_INDICES(start, end, len1);
Benjamin Petersonead6b532011-12-20 17:23:42 -060010806 switch (kind) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010807 case PyUnicode_1BYTE_KIND:
10808 iresult = ucs1lib_count(
10809 ((Py_UCS1*)buf1) + start, end - start,
10810 buf2, len2, PY_SSIZE_T_MAX
10811 );
10812 break;
10813 case PyUnicode_2BYTE_KIND:
10814 iresult = ucs2lib_count(
10815 ((Py_UCS2*)buf1) + start, end - start,
10816 buf2, len2, PY_SSIZE_T_MAX
10817 );
10818 break;
10819 case PyUnicode_4BYTE_KIND:
10820 iresult = ucs4lib_count(
10821 ((Py_UCS4*)buf1) + start, end - start,
10822 buf2, len2, PY_SSIZE_T_MAX
10823 );
10824 break;
10825 default:
10826 assert(0); iresult = 0;
10827 }
10828
10829 result = PyLong_FromSsize_t(iresult);
10830
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010831 if (kind2 != kind)
10832 PyMem_Free(buf2);
Guido van Rossumd57fd912000-03-10 22:53:23 +000010833
10834 Py_DECREF(substring);
Thomas Wouters477c8d52006-05-27 19:21:47 +000010835
Guido van Rossumd57fd912000-03-10 22:53:23 +000010836 return result;
10837}
10838
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000010839PyDoc_STRVAR(encode__doc__,
Victor Stinnerc911bbf2010-11-07 19:04:46 +000010840 "S.encode(encoding='utf-8', errors='strict') -> bytes\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000010841\n\
Victor Stinnere14e2122010-11-07 18:41:46 +000010842Encode S using the codec registered for encoding. Default encoding\n\
10843is 'utf-8'. errors may be given to set a different error\n\
Fred Drakee4315f52000-05-09 19:53:39 +000010844handling scheme. Default is 'strict' meaning that encoding errors raise\n\
Walter Dörwald3aeb6322002-09-02 13:14:32 +000010845a UnicodeEncodeError. Other possible values are 'ignore', 'replace' and\n\
10846'xmlcharrefreplace' as well as any other name registered with\n\
10847codecs.register_error that can handle UnicodeEncodeErrors.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000010848
10849static PyObject *
Victor Stinner9db1a8b2011-10-23 20:04:37 +020010850unicode_encode(PyObject *self, PyObject *args, PyObject *kwargs)
Guido van Rossumd57fd912000-03-10 22:53:23 +000010851{
Benjamin Peterson308d6372009-09-18 21:42:35 +000010852 static char *kwlist[] = {"encoding", "errors", 0};
Guido van Rossumd57fd912000-03-10 22:53:23 +000010853 char *encoding = NULL;
10854 char *errors = NULL;
Guido van Rossum35d94282007-08-27 18:20:11 +000010855
Benjamin Peterson308d6372009-09-18 21:42:35 +000010856 if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|ss:encode",
10857 kwlist, &encoding, &errors))
Guido van Rossumd57fd912000-03-10 22:53:23 +000010858 return NULL;
Victor Stinner9db1a8b2011-10-23 20:04:37 +020010859 return PyUnicode_AsEncodedString(self, encoding, errors);
Marc-André Lemburgd2d45982004-07-08 17:57:32 +000010860}
10861
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000010862PyDoc_STRVAR(expandtabs__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000010863 "S.expandtabs([tabsize]) -> str\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000010864\n\
10865Return a copy of S where all tab characters are expanded using spaces.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000010866If tabsize is not given, a tab size of 8 characters is assumed.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000010867
10868static PyObject*
Victor Stinner9db1a8b2011-10-23 20:04:37 +020010869unicode_expandtabs(PyObject *self, PyObject *args)
Guido van Rossumd57fd912000-03-10 22:53:23 +000010870{
Antoine Pitroue71d5742011-10-04 15:55:09 +020010871 Py_ssize_t i, j, line_pos, src_len, incr;
10872 Py_UCS4 ch;
10873 PyObject *u;
10874 void *src_data, *dest_data;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010875 int tabsize = 8;
Antoine Pitroue71d5742011-10-04 15:55:09 +020010876 int kind;
Antoine Pitroue19aa382011-10-04 16:04:01 +020010877 int found;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010878
10879 if (!PyArg_ParseTuple(args, "|i:expandtabs", &tabsize))
Benjamin Peterson29060642009-01-31 22:14:21 +000010880 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010881
Antoine Pitrou22425222011-10-04 19:10:51 +020010882 if (PyUnicode_READY(self) == -1)
10883 return NULL;
10884
Thomas Wouters7e474022000-07-16 12:04:32 +000010885 /* First pass: determine size of output string */
Antoine Pitroue71d5742011-10-04 15:55:09 +020010886 src_len = PyUnicode_GET_LENGTH(self);
10887 i = j = line_pos = 0;
10888 kind = PyUnicode_KIND(self);
10889 src_data = PyUnicode_DATA(self);
Antoine Pitroue19aa382011-10-04 16:04:01 +020010890 found = 0;
Antoine Pitroue71d5742011-10-04 15:55:09 +020010891 for (; i < src_len; i++) {
10892 ch = PyUnicode_READ(kind, src_data, i);
10893 if (ch == '\t') {
Antoine Pitroue19aa382011-10-04 16:04:01 +020010894 found = 1;
Benjamin Peterson29060642009-01-31 22:14:21 +000010895 if (tabsize > 0) {
Antoine Pitroue71d5742011-10-04 15:55:09 +020010896 incr = tabsize - (line_pos % tabsize); /* cannot overflow */
Benjamin Peterson29060642009-01-31 22:14:21 +000010897 if (j > PY_SSIZE_T_MAX - incr)
Antoine Pitroue71d5742011-10-04 15:55:09 +020010898 goto overflow;
10899 line_pos += incr;
Benjamin Peterson29060642009-01-31 22:14:21 +000010900 j += incr;
Christian Heimesdd15f6c2008-03-16 00:07:10 +000010901 }
Benjamin Peterson29060642009-01-31 22:14:21 +000010902 }
Guido van Rossumd57fd912000-03-10 22:53:23 +000010903 else {
Benjamin Peterson29060642009-01-31 22:14:21 +000010904 if (j > PY_SSIZE_T_MAX - 1)
Antoine Pitroue71d5742011-10-04 15:55:09 +020010905 goto overflow;
10906 line_pos++;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010907 j++;
Antoine Pitroue71d5742011-10-04 15:55:09 +020010908 if (ch == '\n' || ch == '\r')
10909 line_pos = 0;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010910 }
Antoine Pitroue71d5742011-10-04 15:55:09 +020010911 }
Victor Stinnerc4b49542011-12-11 22:44:26 +010010912 if (!found)
10913 return unicode_result_unchanged(self);
Guido van Rossumcd16bf62007-06-13 18:07:49 +000010914
Guido van Rossumd57fd912000-03-10 22:53:23 +000010915 /* Second pass: create output string and fill it */
Antoine Pitroue71d5742011-10-04 15:55:09 +020010916 u = PyUnicode_New(j, PyUnicode_MAX_CHAR_VALUE(self));
Guido van Rossumd57fd912000-03-10 22:53:23 +000010917 if (!u)
10918 return NULL;
Antoine Pitroue71d5742011-10-04 15:55:09 +020010919 dest_data = PyUnicode_DATA(u);
Guido van Rossumd57fd912000-03-10 22:53:23 +000010920
Antoine Pitroue71d5742011-10-04 15:55:09 +020010921 i = j = line_pos = 0;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010922
Antoine Pitroue71d5742011-10-04 15:55:09 +020010923 for (; i < src_len; i++) {
10924 ch = PyUnicode_READ(kind, src_data, i);
10925 if (ch == '\t') {
Benjamin Peterson29060642009-01-31 22:14:21 +000010926 if (tabsize > 0) {
Antoine Pitroue71d5742011-10-04 15:55:09 +020010927 incr = tabsize - (line_pos % tabsize);
10928 line_pos += incr;
Victor Stinnerda79e632012-02-22 13:37:04 +010010929 FILL(kind, dest_data, ' ', j, incr);
10930 j += incr;
Benjamin Peterson29060642009-01-31 22:14:21 +000010931 }
Benjamin Peterson14339b62009-01-31 16:36:08 +000010932 }
Benjamin Peterson29060642009-01-31 22:14:21 +000010933 else {
Antoine Pitroue71d5742011-10-04 15:55:09 +020010934 line_pos++;
10935 PyUnicode_WRITE(kind, dest_data, j, ch);
Christian Heimesdd15f6c2008-03-16 00:07:10 +000010936 j++;
Antoine Pitroue71d5742011-10-04 15:55:09 +020010937 if (ch == '\n' || ch == '\r')
10938 line_pos = 0;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010939 }
Antoine Pitroue71d5742011-10-04 15:55:09 +020010940 }
10941 assert (j == PyUnicode_GET_LENGTH(u));
Victor Stinnerd3df8ab2011-11-22 01:22:34 +010010942 return unicode_result(u);
Christian Heimesdd15f6c2008-03-16 00:07:10 +000010943
Antoine Pitroue71d5742011-10-04 15:55:09 +020010944 overflow:
Christian Heimesdd15f6c2008-03-16 00:07:10 +000010945 PyErr_SetString(PyExc_OverflowError, "new string is too long");
10946 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010947}
10948
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000010949PyDoc_STRVAR(find__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000010950 "S.find(sub[, start[, end]]) -> int\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000010951\n\
10952Return the lowest index in S where substring sub is found,\n\
Senthil Kumaran53516a82011-07-27 23:33:54 +080010953such that sub is contained within S[start:end]. Optional\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000010954arguments start and end are interpreted as in slice notation.\n\
10955\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000010956Return -1 on failure.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000010957
10958static PyObject *
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010959unicode_find(PyObject *self, PyObject *args)
Guido van Rossumd57fd912000-03-10 22:53:23 +000010960{
Victor Stinner9db1a8b2011-10-23 20:04:37 +020010961 PyObject *substring;
Christian Heimes9cd17752007-11-18 19:35:23 +000010962 Py_ssize_t start;
10963 Py_ssize_t end;
Thomas Wouters477c8d52006-05-27 19:21:47 +000010964 Py_ssize_t result;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010965
Jesus Ceaac451502011-04-20 17:09:23 +020010966 if (!stringlib_parse_args_finds_unicode("find", args, &substring,
10967 &start, &end))
Guido van Rossumd57fd912000-03-10 22:53:23 +000010968 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010969
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010970 if (PyUnicode_READY(self) == -1)
10971 return NULL;
10972 if (PyUnicode_READY(substring) == -1)
10973 return NULL;
10974
Victor Stinner7931d9a2011-11-04 00:22:48 +010010975 result = any_find_slice(1, self, substring, start, end);
Guido van Rossumd57fd912000-03-10 22:53:23 +000010976
10977 Py_DECREF(substring);
Thomas Wouters477c8d52006-05-27 19:21:47 +000010978
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010979 if (result == -2)
10980 return NULL;
10981
Christian Heimes217cfd12007-12-02 14:31:20 +000010982 return PyLong_FromSsize_t(result);
Guido van Rossumd57fd912000-03-10 22:53:23 +000010983}
10984
10985static PyObject *
Victor Stinner2fe5ced2011-10-02 00:25:40 +020010986unicode_getitem(PyObject *self, Py_ssize_t index)
Guido van Rossumd57fd912000-03-10 22:53:23 +000010987{
Victor Stinnerb6cd0142012-05-03 02:17:04 +020010988 void *data;
10989 enum PyUnicode_Kind kind;
10990 Py_UCS4 ch;
10991 PyObject *res;
10992
10993 if (!PyUnicode_Check(self) || PyUnicode_READY(self) == -1) {
10994 PyErr_BadArgument();
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010995 return NULL;
Victor Stinnerb6cd0142012-05-03 02:17:04 +020010996 }
10997 if (index < 0 || index >= PyUnicode_GET_LENGTH(self)) {
10998 PyErr_SetString(PyExc_IndexError, "string index out of range");
10999 return NULL;
11000 }
11001 kind = PyUnicode_KIND(self);
11002 data = PyUnicode_DATA(self);
11003 ch = PyUnicode_READ(kind, data, index);
11004 if (ch < 256)
11005 return get_latin1_char(ch);
11006
11007 res = PyUnicode_New(1, ch);
11008 if (res == NULL)
11009 return NULL;
11010 kind = PyUnicode_KIND(res);
11011 data = PyUnicode_DATA(res);
11012 PyUnicode_WRITE(kind, data, 0, ch);
11013 assert(_PyUnicode_CheckConsistency(res, 1));
11014 return res;
Guido van Rossumd57fd912000-03-10 22:53:23 +000011015}
11016
Guido van Rossumc2504932007-09-18 19:42:40 +000011017/* Believe it or not, this produces the same value for ASCII strings
Mark Dickinson57e683e2011-09-24 18:18:40 +010011018 as bytes_hash(). */
Benjamin Peterson8f67d082010-10-17 20:54:53 +000011019static Py_hash_t
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011020unicode_hash(PyObject *self)
Guido van Rossumd57fd912000-03-10 22:53:23 +000011021{
Guido van Rossumc2504932007-09-18 19:42:40 +000011022 Py_ssize_t len;
Mark Dickinson57e683e2011-09-24 18:18:40 +010011023 Py_uhash_t x;
Guido van Rossumc2504932007-09-18 19:42:40 +000011024
Benjamin Petersonf6622c82012-04-09 14:53:07 -040011025#ifdef Py_DEBUG
Benjamin Peterson69e97272012-02-21 11:08:50 -050011026 assert(_Py_HashSecret_Initialized);
Benjamin Petersonf6622c82012-04-09 14:53:07 -040011027#endif
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011028 if (_PyUnicode_HASH(self) != -1)
11029 return _PyUnicode_HASH(self);
11030 if (PyUnicode_READY(self) == -1)
11031 return -1;
11032 len = PyUnicode_GET_LENGTH(self);
Georg Brandl16fa2a12012-02-21 00:50:13 +010011033 /*
11034 We make the hash of the empty string be 0, rather than using
11035 (prefix ^ suffix), since this slightly obfuscates the hash secret
11036 */
11037 if (len == 0) {
11038 _PyUnicode_HASH(self) = 0;
11039 return 0;
11040 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011041
11042 /* The hash function as a macro, gets expanded three times below. */
Georg Brandl2fb477c2012-02-21 00:33:36 +010011043#define HASH(P) \
11044 x ^= (Py_uhash_t) *P << 7; \
11045 while (--len >= 0) \
11046 x = (_PyHASH_MULTIPLIER * x) ^ (Py_uhash_t) *P++; \
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011047
Georg Brandl2fb477c2012-02-21 00:33:36 +010011048 x = (Py_uhash_t) _Py_HashSecret.prefix;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011049 switch (PyUnicode_KIND(self)) {
11050 case PyUnicode_1BYTE_KIND: {
11051 const unsigned char *c = PyUnicode_1BYTE_DATA(self);
11052 HASH(c);
11053 break;
11054 }
11055 case PyUnicode_2BYTE_KIND: {
11056 const Py_UCS2 *s = PyUnicode_2BYTE_DATA(self);
11057 HASH(s);
11058 break;
11059 }
11060 default: {
11061 Py_UCS4 *l;
11062 assert(PyUnicode_KIND(self) == PyUnicode_4BYTE_KIND &&
11063 "Impossible switch case in unicode_hash");
11064 l = PyUnicode_4BYTE_DATA(self);
11065 HASH(l);
11066 break;
11067 }
11068 }
Georg Brandl2fb477c2012-02-21 00:33:36 +010011069 x ^= (Py_uhash_t) PyUnicode_GET_LENGTH(self);
11070 x ^= (Py_uhash_t) _Py_HashSecret.suffix;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011071
Guido van Rossumc2504932007-09-18 19:42:40 +000011072 if (x == -1)
11073 x = -2;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011074 _PyUnicode_HASH(self) = x;
Guido van Rossumc2504932007-09-18 19:42:40 +000011075 return x;
Guido van Rossumd57fd912000-03-10 22:53:23 +000011076}
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011077#undef HASH
Guido van Rossumd57fd912000-03-10 22:53:23 +000011078
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011079PyDoc_STRVAR(index__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000011080 "S.index(sub[, start[, end]]) -> int\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000011081\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011082Like S.find() but raise ValueError when the substring is not found.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000011083
11084static PyObject *
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011085unicode_index(PyObject *self, PyObject *args)
Guido van Rossumd57fd912000-03-10 22:53:23 +000011086{
Martin v. Löwis18e16552006-02-15 17:27:45 +000011087 Py_ssize_t result;
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011088 PyObject *substring;
Christian Heimes9cd17752007-11-18 19:35:23 +000011089 Py_ssize_t start;
11090 Py_ssize_t end;
Guido van Rossumd57fd912000-03-10 22:53:23 +000011091
Jesus Ceaac451502011-04-20 17:09:23 +020011092 if (!stringlib_parse_args_finds_unicode("index", args, &substring,
11093 &start, &end))
Guido van Rossumd57fd912000-03-10 22:53:23 +000011094 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +000011095
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011096 if (PyUnicode_READY(self) == -1)
11097 return NULL;
11098 if (PyUnicode_READY(substring) == -1)
11099 return NULL;
11100
Victor Stinner7931d9a2011-11-04 00:22:48 +010011101 result = any_find_slice(1, self, substring, start, end);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011102
11103 Py_DECREF(substring);
Thomas Wouters477c8d52006-05-27 19:21:47 +000011104
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011105 if (result == -2)
11106 return NULL;
11107
Guido van Rossumd57fd912000-03-10 22:53:23 +000011108 if (result < 0) {
11109 PyErr_SetString(PyExc_ValueError, "substring not found");
11110 return NULL;
11111 }
Thomas Wouters477c8d52006-05-27 19:21:47 +000011112
Christian Heimes217cfd12007-12-02 14:31:20 +000011113 return PyLong_FromSsize_t(result);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011114}
11115
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011116PyDoc_STRVAR(islower__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000011117 "S.islower() -> bool\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000011118\n\
Guido van Rossum77f6a652002-04-03 22:41:51 +000011119Return True if all cased characters in S are lowercase and there is\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011120at least one cased character in S, False otherwise.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000011121
11122static PyObject*
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011123unicode_islower(PyObject *self)
Guido van Rossumd57fd912000-03-10 22:53:23 +000011124{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011125 Py_ssize_t i, length;
11126 int kind;
11127 void *data;
Guido van Rossumd57fd912000-03-10 22:53:23 +000011128 int cased;
11129
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011130 if (PyUnicode_READY(self) == -1)
11131 return NULL;
11132 length = PyUnicode_GET_LENGTH(self);
11133 kind = PyUnicode_KIND(self);
11134 data = PyUnicode_DATA(self);
11135
Guido van Rossumd57fd912000-03-10 22:53:23 +000011136 /* Shortcut for single character strings */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011137 if (length == 1)
11138 return PyBool_FromLong(
11139 Py_UNICODE_ISLOWER(PyUnicode_READ(kind, data, 0)));
Guido van Rossumd57fd912000-03-10 22:53:23 +000011140
Marc-André Lemburg60bc8092000-06-14 09:18:32 +000011141 /* Special case for empty strings */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011142 if (length == 0)
Benjamin Peterson29060642009-01-31 22:14:21 +000011143 return PyBool_FromLong(0);
Marc-André Lemburg60bc8092000-06-14 09:18:32 +000011144
Guido van Rossumd57fd912000-03-10 22:53:23 +000011145 cased = 0;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011146 for (i = 0; i < length; i++) {
11147 const Py_UCS4 ch = PyUnicode_READ(kind, data, i);
Tim Petersced69f82003-09-16 20:30:58 +000011148
Benjamin Peterson29060642009-01-31 22:14:21 +000011149 if (Py_UNICODE_ISUPPER(ch) || Py_UNICODE_ISTITLE(ch))
11150 return PyBool_FromLong(0);
11151 else if (!cased && Py_UNICODE_ISLOWER(ch))
11152 cased = 1;
Guido van Rossumd57fd912000-03-10 22:53:23 +000011153 }
Guido van Rossum77f6a652002-04-03 22:41:51 +000011154 return PyBool_FromLong(cased);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011155}
11156
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011157PyDoc_STRVAR(isupper__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000011158 "S.isupper() -> bool\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000011159\n\
Martin v. Löwis6828e182003-10-18 09:55:08 +000011160Return True if all cased characters in S are uppercase and there is\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011161at least one cased character in S, False otherwise.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000011162
11163static PyObject*
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011164unicode_isupper(PyObject *self)
Guido van Rossumd57fd912000-03-10 22:53:23 +000011165{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011166 Py_ssize_t i, length;
11167 int kind;
11168 void *data;
Guido van Rossumd57fd912000-03-10 22:53:23 +000011169 int cased;
11170
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011171 if (PyUnicode_READY(self) == -1)
11172 return NULL;
11173 length = PyUnicode_GET_LENGTH(self);
11174 kind = PyUnicode_KIND(self);
11175 data = PyUnicode_DATA(self);
11176
Guido van Rossumd57fd912000-03-10 22:53:23 +000011177 /* Shortcut for single character strings */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011178 if (length == 1)
11179 return PyBool_FromLong(
11180 Py_UNICODE_ISUPPER(PyUnicode_READ(kind, data, 0)) != 0);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011181
Marc-André Lemburg60bc8092000-06-14 09:18:32 +000011182 /* Special case for empty strings */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011183 if (length == 0)
Benjamin Peterson29060642009-01-31 22:14:21 +000011184 return PyBool_FromLong(0);
Marc-André Lemburg60bc8092000-06-14 09:18:32 +000011185
Guido van Rossumd57fd912000-03-10 22:53:23 +000011186 cased = 0;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011187 for (i = 0; i < length; i++) {
11188 const Py_UCS4 ch = PyUnicode_READ(kind, data, i);
Tim Petersced69f82003-09-16 20:30:58 +000011189
Benjamin Peterson29060642009-01-31 22:14:21 +000011190 if (Py_UNICODE_ISLOWER(ch) || Py_UNICODE_ISTITLE(ch))
11191 return PyBool_FromLong(0);
11192 else if (!cased && Py_UNICODE_ISUPPER(ch))
11193 cased = 1;
Guido van Rossumd57fd912000-03-10 22:53:23 +000011194 }
Guido van Rossum77f6a652002-04-03 22:41:51 +000011195 return PyBool_FromLong(cased);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011196}
11197
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011198PyDoc_STRVAR(istitle__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000011199 "S.istitle() -> bool\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000011200\n\
Martin v. Löwis6828e182003-10-18 09:55:08 +000011201Return True if S is a titlecased string and there is at least one\n\
11202character in S, i.e. upper- and titlecase characters may only\n\
11203follow uncased characters and lowercase characters only cased ones.\n\
11204Return False otherwise.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000011205
11206static PyObject*
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011207unicode_istitle(PyObject *self)
Guido van Rossumd57fd912000-03-10 22:53:23 +000011208{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011209 Py_ssize_t i, length;
11210 int kind;
11211 void *data;
Guido van Rossumd57fd912000-03-10 22:53:23 +000011212 int cased, previous_is_cased;
11213
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011214 if (PyUnicode_READY(self) == -1)
11215 return NULL;
11216 length = PyUnicode_GET_LENGTH(self);
11217 kind = PyUnicode_KIND(self);
11218 data = PyUnicode_DATA(self);
11219
Guido van Rossumd57fd912000-03-10 22:53:23 +000011220 /* Shortcut for single character strings */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011221 if (length == 1) {
11222 Py_UCS4 ch = PyUnicode_READ(kind, data, 0);
11223 return PyBool_FromLong((Py_UNICODE_ISTITLE(ch) != 0) ||
11224 (Py_UNICODE_ISUPPER(ch) != 0));
11225 }
Guido van Rossumd57fd912000-03-10 22:53:23 +000011226
Marc-André Lemburg60bc8092000-06-14 09:18:32 +000011227 /* Special case for empty strings */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011228 if (length == 0)
Benjamin Peterson29060642009-01-31 22:14:21 +000011229 return PyBool_FromLong(0);
Marc-André Lemburg60bc8092000-06-14 09:18:32 +000011230
Guido van Rossumd57fd912000-03-10 22:53:23 +000011231 cased = 0;
11232 previous_is_cased = 0;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011233 for (i = 0; i < length; i++) {
11234 const Py_UCS4 ch = PyUnicode_READ(kind, data, i);
Tim Petersced69f82003-09-16 20:30:58 +000011235
Benjamin Peterson29060642009-01-31 22:14:21 +000011236 if (Py_UNICODE_ISUPPER(ch) || Py_UNICODE_ISTITLE(ch)) {
11237 if (previous_is_cased)
11238 return PyBool_FromLong(0);
11239 previous_is_cased = 1;
11240 cased = 1;
11241 }
11242 else if (Py_UNICODE_ISLOWER(ch)) {
11243 if (!previous_is_cased)
11244 return PyBool_FromLong(0);
11245 previous_is_cased = 1;
11246 cased = 1;
11247 }
11248 else
11249 previous_is_cased = 0;
Guido van Rossumd57fd912000-03-10 22:53:23 +000011250 }
Guido van Rossum77f6a652002-04-03 22:41:51 +000011251 return PyBool_FromLong(cased);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011252}
11253
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011254PyDoc_STRVAR(isspace__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000011255 "S.isspace() -> bool\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000011256\n\
Martin v. Löwis6828e182003-10-18 09:55:08 +000011257Return True if all characters in S are whitespace\n\
11258and there is at least one character in S, False otherwise.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000011259
11260static PyObject*
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011261unicode_isspace(PyObject *self)
Guido van Rossumd57fd912000-03-10 22:53:23 +000011262{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011263 Py_ssize_t i, length;
11264 int kind;
11265 void *data;
11266
11267 if (PyUnicode_READY(self) == -1)
11268 return NULL;
11269 length = PyUnicode_GET_LENGTH(self);
11270 kind = PyUnicode_KIND(self);
11271 data = PyUnicode_DATA(self);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011272
Guido van Rossumd57fd912000-03-10 22:53:23 +000011273 /* Shortcut for single character strings */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011274 if (length == 1)
11275 return PyBool_FromLong(
11276 Py_UNICODE_ISSPACE(PyUnicode_READ(kind, data, 0)));
Guido van Rossumd57fd912000-03-10 22:53:23 +000011277
Marc-André Lemburg60bc8092000-06-14 09:18:32 +000011278 /* Special case for empty strings */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011279 if (length == 0)
Benjamin Peterson29060642009-01-31 22:14:21 +000011280 return PyBool_FromLong(0);
Marc-André Lemburg60bc8092000-06-14 09:18:32 +000011281
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011282 for (i = 0; i < length; i++) {
11283 const Py_UCS4 ch = PyUnicode_READ(kind, data, i);
Ezio Melotti93e7afc2011-08-22 14:08:38 +030011284 if (!Py_UNICODE_ISSPACE(ch))
Benjamin Peterson29060642009-01-31 22:14:21 +000011285 return PyBool_FromLong(0);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011286 }
Guido van Rossum77f6a652002-04-03 22:41:51 +000011287 return PyBool_FromLong(1);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011288}
11289
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011290PyDoc_STRVAR(isalpha__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000011291 "S.isalpha() -> bool\n\
Marc-André Lemburga7acf422000-07-05 09:49:44 +000011292\n\
Martin v. Löwis6828e182003-10-18 09:55:08 +000011293Return True if all characters in S are alphabetic\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011294and there is at least one character in S, False otherwise.");
Marc-André Lemburga7acf422000-07-05 09:49:44 +000011295
11296static PyObject*
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011297unicode_isalpha(PyObject *self)
Marc-André Lemburga7acf422000-07-05 09:49:44 +000011298{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011299 Py_ssize_t i, length;
11300 int kind;
11301 void *data;
11302
11303 if (PyUnicode_READY(self) == -1)
11304 return NULL;
11305 length = PyUnicode_GET_LENGTH(self);
11306 kind = PyUnicode_KIND(self);
11307 data = PyUnicode_DATA(self);
Marc-André Lemburga7acf422000-07-05 09:49:44 +000011308
Marc-André Lemburga7acf422000-07-05 09:49:44 +000011309 /* Shortcut for single character strings */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011310 if (length == 1)
11311 return PyBool_FromLong(
11312 Py_UNICODE_ISALPHA(PyUnicode_READ(kind, data, 0)));
Marc-André Lemburga7acf422000-07-05 09:49:44 +000011313
11314 /* Special case for empty strings */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011315 if (length == 0)
Benjamin Peterson29060642009-01-31 22:14:21 +000011316 return PyBool_FromLong(0);
Marc-André Lemburga7acf422000-07-05 09:49:44 +000011317
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011318 for (i = 0; i < length; i++) {
11319 if (!Py_UNICODE_ISALPHA(PyUnicode_READ(kind, data, i)))
Benjamin Peterson29060642009-01-31 22:14:21 +000011320 return PyBool_FromLong(0);
Marc-André Lemburga7acf422000-07-05 09:49:44 +000011321 }
Guido van Rossum77f6a652002-04-03 22:41:51 +000011322 return PyBool_FromLong(1);
Marc-André Lemburga7acf422000-07-05 09:49:44 +000011323}
11324
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011325PyDoc_STRVAR(isalnum__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000011326 "S.isalnum() -> bool\n\
Marc-André Lemburga7acf422000-07-05 09:49:44 +000011327\n\
Martin v. Löwis6828e182003-10-18 09:55:08 +000011328Return True if all characters in S are alphanumeric\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011329and there is at least one character in S, False otherwise.");
Marc-André Lemburga7acf422000-07-05 09:49:44 +000011330
11331static PyObject*
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011332unicode_isalnum(PyObject *self)
Marc-André Lemburga7acf422000-07-05 09:49:44 +000011333{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011334 int kind;
11335 void *data;
11336 Py_ssize_t len, i;
11337
11338 if (PyUnicode_READY(self) == -1)
11339 return NULL;
11340
11341 kind = PyUnicode_KIND(self);
11342 data = PyUnicode_DATA(self);
11343 len = PyUnicode_GET_LENGTH(self);
Marc-André Lemburga7acf422000-07-05 09:49:44 +000011344
Marc-André Lemburga7acf422000-07-05 09:49:44 +000011345 /* Shortcut for single character strings */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011346 if (len == 1) {
11347 const Py_UCS4 ch = PyUnicode_READ(kind, data, 0);
11348 return PyBool_FromLong(Py_UNICODE_ISALNUM(ch));
11349 }
Marc-André Lemburga7acf422000-07-05 09:49:44 +000011350
11351 /* Special case for empty strings */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011352 if (len == 0)
Benjamin Peterson29060642009-01-31 22:14:21 +000011353 return PyBool_FromLong(0);
Marc-André Lemburga7acf422000-07-05 09:49:44 +000011354
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011355 for (i = 0; i < len; i++) {
11356 const Py_UCS4 ch = PyUnicode_READ(kind, data, i);
Ezio Melotti93e7afc2011-08-22 14:08:38 +030011357 if (!Py_UNICODE_ISALNUM(ch))
Benjamin Peterson29060642009-01-31 22:14:21 +000011358 return PyBool_FromLong(0);
Marc-André Lemburga7acf422000-07-05 09:49:44 +000011359 }
Guido van Rossum77f6a652002-04-03 22:41:51 +000011360 return PyBool_FromLong(1);
Marc-André Lemburga7acf422000-07-05 09:49:44 +000011361}
11362
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011363PyDoc_STRVAR(isdecimal__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000011364 "S.isdecimal() -> bool\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000011365\n\
Guido van Rossum77f6a652002-04-03 22:41:51 +000011366Return True if there are only decimal characters in S,\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011367False otherwise.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000011368
11369static PyObject*
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011370unicode_isdecimal(PyObject *self)
Guido van Rossumd57fd912000-03-10 22:53:23 +000011371{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011372 Py_ssize_t i, length;
11373 int kind;
11374 void *data;
11375
11376 if (PyUnicode_READY(self) == -1)
11377 return NULL;
11378 length = PyUnicode_GET_LENGTH(self);
11379 kind = PyUnicode_KIND(self);
11380 data = PyUnicode_DATA(self);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011381
Guido van Rossumd57fd912000-03-10 22:53:23 +000011382 /* Shortcut for single character strings */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011383 if (length == 1)
11384 return PyBool_FromLong(
11385 Py_UNICODE_ISDECIMAL(PyUnicode_READ(kind, data, 0)));
Guido van Rossumd57fd912000-03-10 22:53:23 +000011386
Marc-André Lemburg60bc8092000-06-14 09:18:32 +000011387 /* Special case for empty strings */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011388 if (length == 0)
Benjamin Peterson29060642009-01-31 22:14:21 +000011389 return PyBool_FromLong(0);
Marc-André Lemburg60bc8092000-06-14 09:18:32 +000011390
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011391 for (i = 0; i < length; i++) {
11392 if (!Py_UNICODE_ISDECIMAL(PyUnicode_READ(kind, data, i)))
Benjamin Peterson29060642009-01-31 22:14:21 +000011393 return PyBool_FromLong(0);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011394 }
Guido van Rossum77f6a652002-04-03 22:41:51 +000011395 return PyBool_FromLong(1);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011396}
11397
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011398PyDoc_STRVAR(isdigit__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000011399 "S.isdigit() -> bool\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000011400\n\
Martin v. Löwis6828e182003-10-18 09:55:08 +000011401Return True if all characters in S are digits\n\
11402and there is at least one character in S, False otherwise.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000011403
11404static PyObject*
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011405unicode_isdigit(PyObject *self)
Guido van Rossumd57fd912000-03-10 22:53:23 +000011406{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011407 Py_ssize_t i, length;
11408 int kind;
11409 void *data;
11410
11411 if (PyUnicode_READY(self) == -1)
11412 return NULL;
11413 length = PyUnicode_GET_LENGTH(self);
11414 kind = PyUnicode_KIND(self);
11415 data = PyUnicode_DATA(self);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011416
Guido van Rossumd57fd912000-03-10 22:53:23 +000011417 /* Shortcut for single character strings */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011418 if (length == 1) {
11419 const Py_UCS4 ch = PyUnicode_READ(kind, data, 0);
11420 return PyBool_FromLong(Py_UNICODE_ISDIGIT(ch));
11421 }
Guido van Rossumd57fd912000-03-10 22:53:23 +000011422
Marc-André Lemburg60bc8092000-06-14 09:18:32 +000011423 /* Special case for empty strings */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011424 if (length == 0)
Benjamin Peterson29060642009-01-31 22:14:21 +000011425 return PyBool_FromLong(0);
Marc-André Lemburg60bc8092000-06-14 09:18:32 +000011426
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011427 for (i = 0; i < length; i++) {
11428 if (!Py_UNICODE_ISDIGIT(PyUnicode_READ(kind, data, i)))
Benjamin Peterson29060642009-01-31 22:14:21 +000011429 return PyBool_FromLong(0);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011430 }
Guido van Rossum77f6a652002-04-03 22:41:51 +000011431 return PyBool_FromLong(1);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011432}
11433
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011434PyDoc_STRVAR(isnumeric__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000011435 "S.isnumeric() -> bool\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000011436\n\
Guido van Rossum77f6a652002-04-03 22:41:51 +000011437Return True if there are only numeric characters in S,\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011438False otherwise.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000011439
11440static PyObject*
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011441unicode_isnumeric(PyObject *self)
Guido van Rossumd57fd912000-03-10 22:53:23 +000011442{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011443 Py_ssize_t i, length;
11444 int kind;
11445 void *data;
11446
11447 if (PyUnicode_READY(self) == -1)
11448 return NULL;
11449 length = PyUnicode_GET_LENGTH(self);
11450 kind = PyUnicode_KIND(self);
11451 data = PyUnicode_DATA(self);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011452
Guido van Rossumd57fd912000-03-10 22:53:23 +000011453 /* Shortcut for single character strings */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011454 if (length == 1)
11455 return PyBool_FromLong(
11456 Py_UNICODE_ISNUMERIC(PyUnicode_READ(kind, data, 0)));
Guido van Rossumd57fd912000-03-10 22:53:23 +000011457
Marc-André Lemburg60bc8092000-06-14 09:18:32 +000011458 /* Special case for empty strings */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011459 if (length == 0)
Benjamin Peterson29060642009-01-31 22:14:21 +000011460 return PyBool_FromLong(0);
Marc-André Lemburg60bc8092000-06-14 09:18:32 +000011461
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011462 for (i = 0; i < length; i++) {
11463 if (!Py_UNICODE_ISNUMERIC(PyUnicode_READ(kind, data, i)))
Benjamin Peterson29060642009-01-31 22:14:21 +000011464 return PyBool_FromLong(0);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011465 }
Guido van Rossum77f6a652002-04-03 22:41:51 +000011466 return PyBool_FromLong(1);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011467}
11468
Martin v. Löwis47383402007-08-15 07:32:56 +000011469int
11470PyUnicode_IsIdentifier(PyObject *self)
11471{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011472 int kind;
11473 void *data;
11474 Py_ssize_t i;
Ezio Melotti93e7afc2011-08-22 14:08:38 +030011475 Py_UCS4 first;
Martin v. Löwis47383402007-08-15 07:32:56 +000011476
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011477 if (PyUnicode_READY(self) == -1) {
11478 Py_FatalError("identifier not ready");
Benjamin Peterson29060642009-01-31 22:14:21 +000011479 return 0;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011480 }
11481
11482 /* Special case for empty strings */
11483 if (PyUnicode_GET_LENGTH(self) == 0)
11484 return 0;
11485 kind = PyUnicode_KIND(self);
11486 data = PyUnicode_DATA(self);
Martin v. Löwis47383402007-08-15 07:32:56 +000011487
11488 /* PEP 3131 says that the first character must be in
11489 XID_Start and subsequent characters in XID_Continue,
11490 and for the ASCII range, the 2.x rules apply (i.e
Benjamin Peterson14339b62009-01-31 16:36:08 +000011491 start with letters and underscore, continue with
Martin v. Löwis47383402007-08-15 07:32:56 +000011492 letters, digits, underscore). However, given the current
11493 definition of XID_Start and XID_Continue, it is sufficient
11494 to check just for these, except that _ must be allowed
11495 as starting an identifier. */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011496 first = PyUnicode_READ(kind, data, 0);
Benjamin Petersonf413b802011-08-12 22:17:18 -050011497 if (!_PyUnicode_IsXidStart(first) && first != 0x5F /* LOW LINE */)
Martin v. Löwis47383402007-08-15 07:32:56 +000011498 return 0;
11499
Benjamin Peterson9c6e6a02011-09-28 08:09:05 -040011500 for (i = 1; i < PyUnicode_GET_LENGTH(self); i++)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011501 if (!_PyUnicode_IsXidContinue(PyUnicode_READ(kind, data, i)))
Benjamin Peterson29060642009-01-31 22:14:21 +000011502 return 0;
Martin v. Löwis47383402007-08-15 07:32:56 +000011503 return 1;
11504}
11505
11506PyDoc_STRVAR(isidentifier__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000011507 "S.isidentifier() -> bool\n\
Martin v. Löwis47383402007-08-15 07:32:56 +000011508\n\
11509Return True if S is a valid identifier according\n\
11510to the language definition.");
11511
11512static PyObject*
11513unicode_isidentifier(PyObject *self)
11514{
11515 return PyBool_FromLong(PyUnicode_IsIdentifier(self));
11516}
11517
Georg Brandl559e5d72008-06-11 18:37:52 +000011518PyDoc_STRVAR(isprintable__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000011519 "S.isprintable() -> bool\n\
Georg Brandl559e5d72008-06-11 18:37:52 +000011520\n\
11521Return True if all characters in S are considered\n\
11522printable in repr() or S is empty, False otherwise.");
11523
11524static PyObject*
11525unicode_isprintable(PyObject *self)
11526{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011527 Py_ssize_t i, length;
11528 int kind;
11529 void *data;
11530
11531 if (PyUnicode_READY(self) == -1)
11532 return NULL;
11533 length = PyUnicode_GET_LENGTH(self);
11534 kind = PyUnicode_KIND(self);
11535 data = PyUnicode_DATA(self);
Georg Brandl559e5d72008-06-11 18:37:52 +000011536
11537 /* Shortcut for single character strings */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011538 if (length == 1)
11539 return PyBool_FromLong(
11540 Py_UNICODE_ISPRINTABLE(PyUnicode_READ(kind, data, 0)));
Georg Brandl559e5d72008-06-11 18:37:52 +000011541
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011542 for (i = 0; i < length; i++) {
11543 if (!Py_UNICODE_ISPRINTABLE(PyUnicode_READ(kind, data, i))) {
Georg Brandl559e5d72008-06-11 18:37:52 +000011544 Py_RETURN_FALSE;
11545 }
11546 }
11547 Py_RETURN_TRUE;
11548}
11549
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011550PyDoc_STRVAR(join__doc__,
Georg Brandl495f7b52009-10-27 15:28:25 +000011551 "S.join(iterable) -> str\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000011552\n\
11553Return a string which is the concatenation of the strings in the\n\
Georg Brandl495f7b52009-10-27 15:28:25 +000011554iterable. The separator between elements is S.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000011555
11556static PyObject*
Martin v. Löwise3eb1f22001-08-16 13:15:00 +000011557unicode_join(PyObject *self, PyObject *data)
Guido van Rossumd57fd912000-03-10 22:53:23 +000011558{
Martin v. Löwise3eb1f22001-08-16 13:15:00 +000011559 return PyUnicode_Join(self, data);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011560}
11561
Martin v. Löwis18e16552006-02-15 17:27:45 +000011562static Py_ssize_t
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011563unicode_length(PyObject *self)
Guido van Rossumd57fd912000-03-10 22:53:23 +000011564{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011565 if (PyUnicode_READY(self) == -1)
11566 return -1;
11567 return PyUnicode_GET_LENGTH(self);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011568}
11569
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011570PyDoc_STRVAR(ljust__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000011571 "S.ljust(width[, fillchar]) -> str\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000011572\n\
Benjamin Petersonf10a79a2008-10-11 00:49:57 +000011573Return S left-justified in a Unicode string of length width. Padding is\n\
Raymond Hettinger4f8f9762003-11-26 08:21:35 +000011574done using the specified fill character (default is a space).");
Guido van Rossumd57fd912000-03-10 22:53:23 +000011575
11576static PyObject *
Victor Stinner9310abb2011-10-05 00:59:23 +020011577unicode_ljust(PyObject *self, PyObject *args)
Guido van Rossumd57fd912000-03-10 22:53:23 +000011578{
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000011579 Py_ssize_t width;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011580 Py_UCS4 fillchar = ' ';
11581
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000011582 if (!PyArg_ParseTuple(args, "n|O&:ljust", &width, convert_uc, &fillchar))
Guido van Rossumd57fd912000-03-10 22:53:23 +000011583 return NULL;
11584
Benjamin Petersonbac79492012-01-14 13:34:47 -050011585 if (PyUnicode_READY(self) == -1)
Victor Stinnerc4b49542011-12-11 22:44:26 +010011586 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +000011587
Victor Stinnerc4b49542011-12-11 22:44:26 +010011588 if (PyUnicode_GET_LENGTH(self) >= width)
11589 return unicode_result_unchanged(self);
11590
11591 return pad(self, 0, width - PyUnicode_GET_LENGTH(self), fillchar);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011592}
11593
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011594PyDoc_STRVAR(lower__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000011595 "S.lower() -> str\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000011596\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011597Return a copy of the string S converted to lowercase.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000011598
11599static PyObject*
Victor Stinner9310abb2011-10-05 00:59:23 +020011600unicode_lower(PyObject *self)
Guido van Rossumd57fd912000-03-10 22:53:23 +000011601{
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -050011602 if (PyUnicode_READY(self) == -1)
11603 return NULL;
11604 if (PyUnicode_IS_ASCII(self))
11605 return ascii_upper_or_lower(self, 1);
Victor Stinnerb0800dc2012-02-25 00:47:08 +010011606 return case_operation(self, do_lower);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011607}
11608
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011609#define LEFTSTRIP 0
11610#define RIGHTSTRIP 1
11611#define BOTHSTRIP 2
11612
11613/* Arrays indexed by above */
11614static const char *stripformat[] = {"|O:lstrip", "|O:rstrip", "|O:strip"};
11615
11616#define STRIPNAME(i) (stripformat[i]+3)
11617
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011618/* externally visible for str.strip(unicode) */
11619PyObject *
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011620_PyUnicode_XStrip(PyObject *self, int striptype, PyObject *sepobj)
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011621{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011622 void *data;
11623 int kind;
11624 Py_ssize_t i, j, len;
11625 BLOOM_MASK sepmask;
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011626
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011627 if (PyUnicode_READY(self) == -1 || PyUnicode_READY(sepobj) == -1)
11628 return NULL;
11629
11630 kind = PyUnicode_KIND(self);
11631 data = PyUnicode_DATA(self);
11632 len = PyUnicode_GET_LENGTH(self);
11633 sepmask = make_bloom_mask(PyUnicode_KIND(sepobj),
11634 PyUnicode_DATA(sepobj),
11635 PyUnicode_GET_LENGTH(sepobj));
Thomas Wouters477c8d52006-05-27 19:21:47 +000011636
Benjamin Peterson14339b62009-01-31 16:36:08 +000011637 i = 0;
11638 if (striptype != RIGHTSTRIP) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011639 while (i < len &&
11640 BLOOM_MEMBER(sepmask, PyUnicode_READ(kind, data, i), sepobj)) {
Benjamin Peterson29060642009-01-31 22:14:21 +000011641 i++;
11642 }
Benjamin Peterson14339b62009-01-31 16:36:08 +000011643 }
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011644
Benjamin Peterson14339b62009-01-31 16:36:08 +000011645 j = len;
11646 if (striptype != LEFTSTRIP) {
Benjamin Peterson29060642009-01-31 22:14:21 +000011647 do {
11648 j--;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011649 } while (j >= i &&
11650 BLOOM_MEMBER(sepmask, PyUnicode_READ(kind, data, j), sepobj));
Benjamin Peterson29060642009-01-31 22:14:21 +000011651 j++;
Benjamin Peterson14339b62009-01-31 16:36:08 +000011652 }
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011653
Victor Stinner7931d9a2011-11-04 00:22:48 +010011654 return PyUnicode_Substring(self, i, j);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011655}
11656
11657PyObject*
11658PyUnicode_Substring(PyObject *self, Py_ssize_t start, Py_ssize_t end)
11659{
11660 unsigned char *data;
11661 int kind;
Victor Stinner12bab6d2011-10-01 01:53:49 +020011662 Py_ssize_t length;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011663
Victor Stinnerde636f32011-10-01 03:55:54 +020011664 if (PyUnicode_READY(self) == -1)
11665 return NULL;
11666
Victor Stinner684d5fd2012-05-03 02:32:34 +020011667 length = PyUnicode_GET_LENGTH(self);
11668 end = Py_MIN(end, length);
Victor Stinnerde636f32011-10-01 03:55:54 +020011669
Victor Stinner684d5fd2012-05-03 02:32:34 +020011670 if (start == 0 && end == length)
Victor Stinnerc4b49542011-12-11 22:44:26 +010011671 return unicode_result_unchanged(self);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011672
Victor Stinnerde636f32011-10-01 03:55:54 +020011673 if (start < 0 || end < 0) {
Victor Stinner12bab6d2011-10-01 01:53:49 +020011674 PyErr_SetString(PyExc_IndexError, "string index out of range");
11675 return NULL;
11676 }
Victor Stinner684d5fd2012-05-03 02:32:34 +020011677 if (start >= length || end < start) {
Victor Stinner3a7f7972012-05-03 03:36:40 +020011678 Py_INCREF(unicode_empty);
11679 return unicode_empty;
Victor Stinner684d5fd2012-05-03 02:32:34 +020011680 }
Victor Stinner12bab6d2011-10-01 01:53:49 +020011681
Victor Stinner684d5fd2012-05-03 02:32:34 +020011682 length = end - start;
Victor Stinnerb9275c12011-10-05 14:01:42 +020011683 if (PyUnicode_IS_ASCII(self)) {
Victor Stinnerb9275c12011-10-05 14:01:42 +020011684 data = PyUnicode_1BYTE_DATA(self);
Victor Stinnerd3f08822012-05-29 12:57:52 +020011685 return _PyUnicode_FromASCII((char*)(data + start), length);
Victor Stinnerb9275c12011-10-05 14:01:42 +020011686 }
11687 else {
11688 kind = PyUnicode_KIND(self);
11689 data = PyUnicode_1BYTE_DATA(self);
11690 return PyUnicode_FromKindAndData(kind,
Martin v. Löwisc47adb02011-10-07 20:55:35 +020011691 data + kind * start,
Victor Stinnerb9275c12011-10-05 14:01:42 +020011692 length);
11693 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011694}
Guido van Rossumd57fd912000-03-10 22:53:23 +000011695
11696static PyObject *
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011697do_strip(PyObject *self, int striptype)
Guido van Rossumd57fd912000-03-10 22:53:23 +000011698{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011699 int kind;
11700 void *data;
11701 Py_ssize_t len, i, j;
11702
11703 if (PyUnicode_READY(self) == -1)
11704 return NULL;
11705
11706 kind = PyUnicode_KIND(self);
11707 data = PyUnicode_DATA(self);
11708 len = PyUnicode_GET_LENGTH(self);
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011709
Benjamin Peterson14339b62009-01-31 16:36:08 +000011710 i = 0;
11711 if (striptype != RIGHTSTRIP) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011712 while (i < len && Py_UNICODE_ISSPACE(PyUnicode_READ(kind, data, i))) {
Benjamin Peterson14339b62009-01-31 16:36:08 +000011713 i++;
11714 }
11715 }
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011716
Benjamin Peterson14339b62009-01-31 16:36:08 +000011717 j = len;
11718 if (striptype != LEFTSTRIP) {
11719 do {
11720 j--;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011721 } while (j >= i && Py_UNICODE_ISSPACE(PyUnicode_READ(kind, data, j)));
Benjamin Peterson14339b62009-01-31 16:36:08 +000011722 j++;
11723 }
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011724
Victor Stinner7931d9a2011-11-04 00:22:48 +010011725 return PyUnicode_Substring(self, i, j);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011726}
11727
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011728
11729static PyObject *
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011730do_argstrip(PyObject *self, int striptype, PyObject *args)
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011731{
Benjamin Peterson14339b62009-01-31 16:36:08 +000011732 PyObject *sep = NULL;
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011733
Benjamin Peterson14339b62009-01-31 16:36:08 +000011734 if (!PyArg_ParseTuple(args, (char *)stripformat[striptype], &sep))
11735 return NULL;
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011736
Benjamin Peterson14339b62009-01-31 16:36:08 +000011737 if (sep != NULL && sep != Py_None) {
11738 if (PyUnicode_Check(sep))
11739 return _PyUnicode_XStrip(self, striptype, sep);
11740 else {
11741 PyErr_Format(PyExc_TypeError,
Benjamin Peterson29060642009-01-31 22:14:21 +000011742 "%s arg must be None or str",
11743 STRIPNAME(striptype));
Benjamin Peterson14339b62009-01-31 16:36:08 +000011744 return NULL;
11745 }
11746 }
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011747
Benjamin Peterson14339b62009-01-31 16:36:08 +000011748 return do_strip(self, striptype);
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011749}
11750
11751
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011752PyDoc_STRVAR(strip__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000011753 "S.strip([chars]) -> str\n\
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011754\n\
11755Return a copy of the string S with leading and trailing\n\
11756whitespace removed.\n\
Benjamin Peterson142957c2008-07-04 19:55:29 +000011757If chars is given and not None, remove characters in chars instead.");
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011758
11759static PyObject *
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011760unicode_strip(PyObject *self, PyObject *args)
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011761{
Benjamin Peterson14339b62009-01-31 16:36:08 +000011762 if (PyTuple_GET_SIZE(args) == 0)
11763 return do_strip(self, BOTHSTRIP); /* Common case */
11764 else
11765 return do_argstrip(self, BOTHSTRIP, args);
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011766}
11767
11768
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011769PyDoc_STRVAR(lstrip__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000011770 "S.lstrip([chars]) -> str\n\
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011771\n\
11772Return a copy of the string S with leading whitespace removed.\n\
Benjamin Peterson142957c2008-07-04 19:55:29 +000011773If chars is given and not None, remove characters in chars instead.");
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011774
11775static PyObject *
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011776unicode_lstrip(PyObject *self, PyObject *args)
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011777{
Benjamin Peterson14339b62009-01-31 16:36:08 +000011778 if (PyTuple_GET_SIZE(args) == 0)
11779 return do_strip(self, LEFTSTRIP); /* Common case */
11780 else
11781 return do_argstrip(self, LEFTSTRIP, args);
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011782}
11783
11784
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011785PyDoc_STRVAR(rstrip__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000011786 "S.rstrip([chars]) -> str\n\
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011787\n\
11788Return a copy of the string S with trailing whitespace removed.\n\
Benjamin Peterson142957c2008-07-04 19:55:29 +000011789If chars is given and not None, remove characters in chars instead.");
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011790
11791static PyObject *
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011792unicode_rstrip(PyObject *self, PyObject *args)
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011793{
Benjamin Peterson14339b62009-01-31 16:36:08 +000011794 if (PyTuple_GET_SIZE(args) == 0)
11795 return do_strip(self, RIGHTSTRIP); /* Common case */
11796 else
11797 return do_argstrip(self, RIGHTSTRIP, args);
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011798}
11799
11800
Guido van Rossumd57fd912000-03-10 22:53:23 +000011801static PyObject*
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011802unicode_repeat(PyObject *str, Py_ssize_t len)
Guido van Rossumd57fd912000-03-10 22:53:23 +000011803{
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011804 PyObject *u;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011805 Py_ssize_t nchars, n;
Guido van Rossumd57fd912000-03-10 22:53:23 +000011806
Georg Brandl222de0f2009-04-12 12:01:50 +000011807 if (len < 1) {
11808 Py_INCREF(unicode_empty);
Victor Stinnera464fc12011-10-02 20:39:30 +020011809 return unicode_empty;
Georg Brandl222de0f2009-04-12 12:01:50 +000011810 }
Guido van Rossumd57fd912000-03-10 22:53:23 +000011811
Victor Stinnerc4b49542011-12-11 22:44:26 +010011812 /* no repeat, return original string */
11813 if (len == 1)
11814 return unicode_result_unchanged(str);
Tim Peters8f422462000-09-09 06:13:41 +000011815
Benjamin Petersonbac79492012-01-14 13:34:47 -050011816 if (PyUnicode_READY(str) == -1)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011817 return NULL;
11818
Victor Stinnerc759f3e2011-10-01 03:09:58 +020011819 if (PyUnicode_GET_LENGTH(str) > PY_SSIZE_T_MAX / len) {
Victor Stinner67ca64c2011-10-01 02:47:29 +020011820 PyErr_SetString(PyExc_OverflowError,
11821 "repeated string is too long");
11822 return NULL;
11823 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011824 nchars = len * PyUnicode_GET_LENGTH(str);
Victor Stinner67ca64c2011-10-01 02:47:29 +020011825
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011826 u = PyUnicode_New(nchars, PyUnicode_MAX_CHAR_VALUE(str));
Guido van Rossumd57fd912000-03-10 22:53:23 +000011827 if (!u)
11828 return NULL;
Victor Stinner67ca64c2011-10-01 02:47:29 +020011829 assert(PyUnicode_KIND(u) == PyUnicode_KIND(str));
Guido van Rossumd57fd912000-03-10 22:53:23 +000011830
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011831 if (PyUnicode_GET_LENGTH(str) == 1) {
11832 const int kind = PyUnicode_KIND(str);
11833 const Py_UCS4 fill_char = PyUnicode_READ(kind, PyUnicode_DATA(str), 0);
Victor Stinner73f53b52011-12-18 03:26:31 +010011834 if (kind == PyUnicode_1BYTE_KIND) {
11835 void *to = PyUnicode_DATA(u);
Victor Stinner67ca64c2011-10-01 02:47:29 +020011836 memset(to, (unsigned char)fill_char, len);
Victor Stinner73f53b52011-12-18 03:26:31 +010011837 }
11838 else if (kind == PyUnicode_2BYTE_KIND) {
11839 Py_UCS2 *ucs2 = PyUnicode_2BYTE_DATA(u);
Victor Stinner67ca64c2011-10-01 02:47:29 +020011840 for (n = 0; n < len; ++n)
Victor Stinner73f53b52011-12-18 03:26:31 +010011841 ucs2[n] = fill_char;
11842 } else {
11843 Py_UCS4 *ucs4 = PyUnicode_4BYTE_DATA(u);
11844 assert(kind == PyUnicode_4BYTE_KIND);
11845 for (n = 0; n < len; ++n)
11846 ucs4[n] = fill_char;
Victor Stinner67ca64c2011-10-01 02:47:29 +020011847 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011848 }
11849 else {
11850 /* number of characters copied this far */
11851 Py_ssize_t done = PyUnicode_GET_LENGTH(str);
Martin v. Löwisc47adb02011-10-07 20:55:35 +020011852 const Py_ssize_t char_size = PyUnicode_KIND(str);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011853 char *to = (char *) PyUnicode_DATA(u);
11854 Py_MEMCPY(to, PyUnicode_DATA(str),
11855 PyUnicode_GET_LENGTH(str) * char_size);
Benjamin Peterson29060642009-01-31 22:14:21 +000011856 while (done < nchars) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011857 n = (done <= nchars-done) ? done : nchars-done;
11858 Py_MEMCPY(to + (done * char_size), to, n * char_size);
Thomas Wouters477c8d52006-05-27 19:21:47 +000011859 done += n;
Benjamin Peterson29060642009-01-31 22:14:21 +000011860 }
Guido van Rossumd57fd912000-03-10 22:53:23 +000011861 }
11862
Victor Stinnerbb10a1f2011-10-05 01:34:17 +020011863 assert(_PyUnicode_CheckConsistency(u, 1));
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011864 return u;
Guido van Rossumd57fd912000-03-10 22:53:23 +000011865}
11866
Alexander Belopolsky40018472011-02-26 01:02:56 +000011867PyObject *
11868PyUnicode_Replace(PyObject *obj,
11869 PyObject *subobj,
11870 PyObject *replobj,
11871 Py_ssize_t maxcount)
Guido van Rossumd57fd912000-03-10 22:53:23 +000011872{
11873 PyObject *self;
11874 PyObject *str1;
11875 PyObject *str2;
11876 PyObject *result;
11877
11878 self = PyUnicode_FromObject(obj);
Benjamin Peterson22a29702012-01-02 09:00:30 -060011879 if (self == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +000011880 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +000011881 str1 = PyUnicode_FromObject(subobj);
Benjamin Peterson22a29702012-01-02 09:00:30 -060011882 if (str1 == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +000011883 Py_DECREF(self);
11884 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +000011885 }
11886 str2 = PyUnicode_FromObject(replobj);
Benjamin Peterson22a29702012-01-02 09:00:30 -060011887 if (str2 == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +000011888 Py_DECREF(self);
11889 Py_DECREF(str1);
11890 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +000011891 }
Benjamin Peterson22a29702012-01-02 09:00:30 -060011892 if (PyUnicode_READY(self) == -1 ||
11893 PyUnicode_READY(str1) == -1 ||
11894 PyUnicode_READY(str2) == -1)
11895 result = NULL;
11896 else
11897 result = replace(self, str1, str2, maxcount);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011898 Py_DECREF(self);
11899 Py_DECREF(str1);
11900 Py_DECREF(str2);
11901 return result;
11902}
11903
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011904PyDoc_STRVAR(replace__doc__,
Ezio Melottic1897e72010-06-26 18:50:39 +000011905 "S.replace(old, new[, count]) -> str\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000011906\n\
11907Return a copy of S with all occurrences of substring\n\
Georg Brandlf08a9dd2008-06-10 16:57:31 +000011908old replaced by new. If the optional argument count is\n\
11909given, only the first count occurrences are replaced.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000011910
11911static PyObject*
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011912unicode_replace(PyObject *self, PyObject *args)
Guido van Rossumd57fd912000-03-10 22:53:23 +000011913{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011914 PyObject *str1;
11915 PyObject *str2;
Martin v. Löwis18e16552006-02-15 17:27:45 +000011916 Py_ssize_t maxcount = -1;
Guido van Rossumd57fd912000-03-10 22:53:23 +000011917 PyObject *result;
11918
Martin v. Löwis18e16552006-02-15 17:27:45 +000011919 if (!PyArg_ParseTuple(args, "OO|n:replace", &str1, &str2, &maxcount))
Guido van Rossumd57fd912000-03-10 22:53:23 +000011920 return NULL;
Benjamin Peterson22a29702012-01-02 09:00:30 -060011921 if (PyUnicode_READY(self) == -1)
Benjamin Peterson29060642009-01-31 22:14:21 +000011922 return NULL;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011923 str1 = PyUnicode_FromObject(str1);
Benjamin Peterson22a29702012-01-02 09:00:30 -060011924 if (str1 == NULL)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011925 return NULL;
11926 str2 = PyUnicode_FromObject(str2);
Benjamin Peterson22a29702012-01-02 09:00:30 -060011927 if (str2 == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +000011928 Py_DECREF(str1);
11929 return NULL;
Walter Dörwaldf6b56ae2003-02-09 23:42:56 +000011930 }
Benjamin Peterson22a29702012-01-02 09:00:30 -060011931 if (PyUnicode_READY(str1) == -1 || PyUnicode_READY(str2) == -1)
11932 result = NULL;
11933 else
11934 result = replace(self, str1, str2, maxcount);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011935
11936 Py_DECREF(str1);
11937 Py_DECREF(str2);
11938 return result;
11939}
11940
Alexander Belopolsky40018472011-02-26 01:02:56 +000011941static PyObject *
11942unicode_repr(PyObject *unicode)
Guido van Rossumd57fd912000-03-10 22:53:23 +000011943{
Walter Dörwald79e913e2007-05-12 11:08:06 +000011944 PyObject *repr;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011945 Py_ssize_t isize;
11946 Py_ssize_t osize, squote, dquote, i, o;
11947 Py_UCS4 max, quote;
11948 int ikind, okind;
11949 void *idata, *odata;
Walter Dörwald79e913e2007-05-12 11:08:06 +000011950
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011951 if (PyUnicode_READY(unicode) == -1)
Walter Dörwald79e913e2007-05-12 11:08:06 +000011952 return NULL;
11953
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011954 isize = PyUnicode_GET_LENGTH(unicode);
11955 idata = PyUnicode_DATA(unicode);
Walter Dörwald79e913e2007-05-12 11:08:06 +000011956
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011957 /* Compute length of output, quote characters, and
11958 maximum character */
11959 osize = 2; /* quotes */
11960 max = 127;
11961 squote = dquote = 0;
11962 ikind = PyUnicode_KIND(unicode);
11963 for (i = 0; i < isize; i++) {
11964 Py_UCS4 ch = PyUnicode_READ(ikind, idata, i);
11965 switch (ch) {
11966 case '\'': squote++; osize++; break;
11967 case '"': dquote++; osize++; break;
11968 case '\\': case '\t': case '\r': case '\n':
11969 osize += 2; break;
11970 default:
11971 /* Fast-path ASCII */
11972 if (ch < ' ' || ch == 0x7f)
11973 osize += 4; /* \xHH */
11974 else if (ch < 0x7f)
11975 osize++;
11976 else if (Py_UNICODE_ISPRINTABLE(ch)) {
11977 osize++;
11978 max = ch > max ? ch : max;
11979 }
11980 else if (ch < 0x100)
11981 osize += 4; /* \xHH */
11982 else if (ch < 0x10000)
11983 osize += 6; /* \uHHHH */
11984 else
11985 osize += 10; /* \uHHHHHHHH */
11986 }
11987 }
11988
11989 quote = '\'';
11990 if (squote) {
11991 if (dquote)
11992 /* Both squote and dquote present. Use squote,
11993 and escape them */
11994 osize += squote;
11995 else
11996 quote = '"';
11997 }
11998
11999 repr = PyUnicode_New(osize, max);
12000 if (repr == NULL)
12001 return NULL;
12002 okind = PyUnicode_KIND(repr);
12003 odata = PyUnicode_DATA(repr);
12004
12005 PyUnicode_WRITE(okind, odata, 0, quote);
12006 PyUnicode_WRITE(okind, odata, osize-1, quote);
12007
12008 for (i = 0, o = 1; i < isize; i++) {
12009 Py_UCS4 ch = PyUnicode_READ(ikind, idata, i);
Walter Dörwald79e913e2007-05-12 11:08:06 +000012010
12011 /* Escape quotes and backslashes */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012012 if ((ch == quote) || (ch == '\\')) {
12013 PyUnicode_WRITE(okind, odata, o++, '\\');
12014 PyUnicode_WRITE(okind, odata, o++, ch);
Walter Dörwald79e913e2007-05-12 11:08:06 +000012015 continue;
12016 }
12017
Benjamin Peterson29060642009-01-31 22:14:21 +000012018 /* Map special whitespace to '\t', \n', '\r' */
Georg Brandl559e5d72008-06-11 18:37:52 +000012019 if (ch == '\t') {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012020 PyUnicode_WRITE(okind, odata, o++, '\\');
12021 PyUnicode_WRITE(okind, odata, o++, 't');
Walter Dörwald79e913e2007-05-12 11:08:06 +000012022 }
12023 else if (ch == '\n') {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012024 PyUnicode_WRITE(okind, odata, o++, '\\');
12025 PyUnicode_WRITE(okind, odata, o++, 'n');
Walter Dörwald79e913e2007-05-12 11:08:06 +000012026 }
12027 else if (ch == '\r') {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012028 PyUnicode_WRITE(okind, odata, o++, '\\');
12029 PyUnicode_WRITE(okind, odata, o++, 'r');
Walter Dörwald79e913e2007-05-12 11:08:06 +000012030 }
12031
12032 /* Map non-printable US ASCII to '\xhh' */
Georg Brandl559e5d72008-06-11 18:37:52 +000012033 else if (ch < ' ' || ch == 0x7F) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012034 PyUnicode_WRITE(okind, odata, o++, '\\');
12035 PyUnicode_WRITE(okind, odata, o++, 'x');
Victor Stinnerf5cff562011-10-14 02:13:11 +020012036 PyUnicode_WRITE(okind, odata, o++, Py_hexdigits[(ch >> 4) & 0x000F]);
12037 PyUnicode_WRITE(okind, odata, o++, Py_hexdigits[ch & 0x000F]);
Walter Dörwald79e913e2007-05-12 11:08:06 +000012038 }
12039
Georg Brandl559e5d72008-06-11 18:37:52 +000012040 /* Copy ASCII characters as-is */
12041 else if (ch < 0x7F) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012042 PyUnicode_WRITE(okind, odata, o++, ch);
Georg Brandl559e5d72008-06-11 18:37:52 +000012043 }
12044
Benjamin Peterson29060642009-01-31 22:14:21 +000012045 /* Non-ASCII characters */
Georg Brandl559e5d72008-06-11 18:37:52 +000012046 else {
Benjamin Peterson14339b62009-01-31 16:36:08 +000012047 /* Map Unicode whitespace and control characters
Georg Brandl559e5d72008-06-11 18:37:52 +000012048 (categories Z* and C* except ASCII space)
12049 */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012050 if (!Py_UNICODE_ISPRINTABLE(ch)) {
Kristján Valur Jónsson55e5dc82012-06-06 21:58:08 +000012051 PyUnicode_WRITE(okind, odata, o++, '\\');
Georg Brandl559e5d72008-06-11 18:37:52 +000012052 /* Map 8-bit characters to '\xhh' */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012053 if (ch <= 0xff) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012054 PyUnicode_WRITE(okind, odata, o++, 'x');
Victor Stinnerf5cff562011-10-14 02:13:11 +020012055 PyUnicode_WRITE(okind, odata, o++, Py_hexdigits[(ch >> 4) & 0x000F]);
12056 PyUnicode_WRITE(okind, odata, o++, Py_hexdigits[ch & 0x000F]);
Georg Brandl559e5d72008-06-11 18:37:52 +000012057 }
Kristján Valur Jónsson55e5dc82012-06-06 21:58:08 +000012058 /* Map 16-bit characters to '\uxxxx' */
12059 else if (ch <= 0xffff) {
12060 PyUnicode_WRITE(okind, odata, o++, 'u');
Victor Stinnerf5cff562011-10-14 02:13:11 +020012061 PyUnicode_WRITE(okind, odata, o++, Py_hexdigits[(ch >> 12) & 0xF]);
12062 PyUnicode_WRITE(okind, odata, o++, Py_hexdigits[(ch >> 8) & 0xF]);
12063 PyUnicode_WRITE(okind, odata, o++, Py_hexdigits[(ch >> 4) & 0xF]);
12064 PyUnicode_WRITE(okind, odata, o++, Py_hexdigits[ch & 0xF]);
Georg Brandl559e5d72008-06-11 18:37:52 +000012065 }
Kristján Valur Jónsson55e5dc82012-06-06 21:58:08 +000012066 /* Map 21-bit characters to '\U00xxxxxx' */
Georg Brandl559e5d72008-06-11 18:37:52 +000012067 else {
Kristján Valur Jónsson55e5dc82012-06-06 21:58:08 +000012068 PyUnicode_WRITE(okind, odata, o++, 'U');
12069 PyUnicode_WRITE(okind, odata, o++, Py_hexdigits[(ch >> 28) & 0xF]);
12070 PyUnicode_WRITE(okind, odata, o++, Py_hexdigits[(ch >> 24) & 0xF]);
12071 PyUnicode_WRITE(okind, odata, o++, Py_hexdigits[(ch >> 20) & 0xF]);
12072 PyUnicode_WRITE(okind, odata, o++, Py_hexdigits[(ch >> 16) & 0xF]);
Victor Stinnerf5cff562011-10-14 02:13:11 +020012073 PyUnicode_WRITE(okind, odata, o++, Py_hexdigits[(ch >> 12) & 0xF]);
12074 PyUnicode_WRITE(okind, odata, o++, Py_hexdigits[(ch >> 8) & 0xF]);
12075 PyUnicode_WRITE(okind, odata, o++, Py_hexdigits[(ch >> 4) & 0xF]);
12076 PyUnicode_WRITE(okind, odata, o++, Py_hexdigits[ch & 0xF]);
Georg Brandl559e5d72008-06-11 18:37:52 +000012077 }
12078 }
12079 /* Copy characters as-is */
12080 else {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012081 PyUnicode_WRITE(okind, odata, o++, ch);
Georg Brandl559e5d72008-06-11 18:37:52 +000012082 }
12083 }
Walter Dörwald79e913e2007-05-12 11:08:06 +000012084 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012085 /* Closing quote already added at the beginning */
Victor Stinner05d11892011-10-06 01:13:58 +020012086 assert(_PyUnicode_CheckConsistency(repr, 1));
Walter Dörwald79e913e2007-05-12 11:08:06 +000012087 return repr;
Guido van Rossumd57fd912000-03-10 22:53:23 +000012088}
12089
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000012090PyDoc_STRVAR(rfind__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000012091 "S.rfind(sub[, start[, end]]) -> int\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000012092\n\
12093Return the highest index in S where substring sub is found,\n\
Senthil Kumaran53516a82011-07-27 23:33:54 +080012094such that sub is contained within S[start:end]. Optional\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000012095arguments start and end are interpreted as in slice notation.\n\
12096\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000012097Return -1 on failure.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000012098
12099static PyObject *
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012100unicode_rfind(PyObject *self, PyObject *args)
Guido van Rossumd57fd912000-03-10 22:53:23 +000012101{
Victor Stinner9db1a8b2011-10-23 20:04:37 +020012102 PyObject *substring;
Christian Heimes9cd17752007-11-18 19:35:23 +000012103 Py_ssize_t start;
12104 Py_ssize_t end;
Thomas Wouters477c8d52006-05-27 19:21:47 +000012105 Py_ssize_t result;
Guido van Rossumd57fd912000-03-10 22:53:23 +000012106
Jesus Ceaac451502011-04-20 17:09:23 +020012107 if (!stringlib_parse_args_finds_unicode("rfind", args, &substring,
12108 &start, &end))
Benjamin Peterson14339b62009-01-31 16:36:08 +000012109 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +000012110
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012111 if (PyUnicode_READY(self) == -1)
12112 return NULL;
12113 if (PyUnicode_READY(substring) == -1)
12114 return NULL;
12115
Victor Stinner7931d9a2011-11-04 00:22:48 +010012116 result = any_find_slice(-1, self, substring, start, end);
Guido van Rossumd57fd912000-03-10 22:53:23 +000012117
12118 Py_DECREF(substring);
Thomas Wouters477c8d52006-05-27 19:21:47 +000012119
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012120 if (result == -2)
12121 return NULL;
12122
Christian Heimes217cfd12007-12-02 14:31:20 +000012123 return PyLong_FromSsize_t(result);
Guido van Rossumd57fd912000-03-10 22:53:23 +000012124}
12125
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000012126PyDoc_STRVAR(rindex__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000012127 "S.rindex(sub[, start[, end]]) -> int\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000012128\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000012129Like S.rfind() but raise ValueError when the substring is not found.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000012130
12131static PyObject *
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012132unicode_rindex(PyObject *self, PyObject *args)
Guido van Rossumd57fd912000-03-10 22:53:23 +000012133{
Victor Stinner9db1a8b2011-10-23 20:04:37 +020012134 PyObject *substring;
Christian Heimes9cd17752007-11-18 19:35:23 +000012135 Py_ssize_t start;
12136 Py_ssize_t end;
Thomas Wouters477c8d52006-05-27 19:21:47 +000012137 Py_ssize_t result;
Guido van Rossumd57fd912000-03-10 22:53:23 +000012138
Jesus Ceaac451502011-04-20 17:09:23 +020012139 if (!stringlib_parse_args_finds_unicode("rindex", args, &substring,
12140 &start, &end))
Benjamin Peterson14339b62009-01-31 16:36:08 +000012141 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +000012142
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012143 if (PyUnicode_READY(self) == -1)
12144 return NULL;
12145 if (PyUnicode_READY(substring) == -1)
12146 return NULL;
12147
Victor Stinner7931d9a2011-11-04 00:22:48 +010012148 result = any_find_slice(-1, self, substring, start, end);
Guido van Rossumd57fd912000-03-10 22:53:23 +000012149
12150 Py_DECREF(substring);
Thomas Wouters477c8d52006-05-27 19:21:47 +000012151
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012152 if (result == -2)
12153 return NULL;
12154
Guido van Rossumd57fd912000-03-10 22:53:23 +000012155 if (result < 0) {
12156 PyErr_SetString(PyExc_ValueError, "substring not found");
12157 return NULL;
12158 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012159
Christian Heimes217cfd12007-12-02 14:31:20 +000012160 return PyLong_FromSsize_t(result);
Guido van Rossumd57fd912000-03-10 22:53:23 +000012161}
12162
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000012163PyDoc_STRVAR(rjust__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000012164 "S.rjust(width[, fillchar]) -> str\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000012165\n\
Benjamin Petersonf10a79a2008-10-11 00:49:57 +000012166Return S right-justified in a string of length width. Padding is\n\
Raymond Hettinger4f8f9762003-11-26 08:21:35 +000012167done using the specified fill character (default is a space).");
Guido van Rossumd57fd912000-03-10 22:53:23 +000012168
12169static PyObject *
Victor Stinner9310abb2011-10-05 00:59:23 +020012170unicode_rjust(PyObject *self, PyObject *args)
Guido van Rossumd57fd912000-03-10 22:53:23 +000012171{
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000012172 Py_ssize_t width;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012173 Py_UCS4 fillchar = ' ';
12174
Victor Stinnere9a29352011-10-01 02:14:59 +020012175 if (!PyArg_ParseTuple(args, "n|O&:rjust", &width, convert_uc, &fillchar))
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012176 return NULL;
Raymond Hettinger4f8f9762003-11-26 08:21:35 +000012177
Benjamin Petersonbac79492012-01-14 13:34:47 -050012178 if (PyUnicode_READY(self) == -1)
Guido van Rossumd57fd912000-03-10 22:53:23 +000012179 return NULL;
12180
Victor Stinnerc4b49542011-12-11 22:44:26 +010012181 if (PyUnicode_GET_LENGTH(self) >= width)
12182 return unicode_result_unchanged(self);
Guido van Rossumd57fd912000-03-10 22:53:23 +000012183
Victor Stinnerc4b49542011-12-11 22:44:26 +010012184 return pad(self, width - PyUnicode_GET_LENGTH(self), 0, fillchar);
Guido van Rossumd57fd912000-03-10 22:53:23 +000012185}
12186
Alexander Belopolsky40018472011-02-26 01:02:56 +000012187PyObject *
12188PyUnicode_Split(PyObject *s, PyObject *sep, Py_ssize_t maxsplit)
Guido van Rossumd57fd912000-03-10 22:53:23 +000012189{
12190 PyObject *result;
Tim Petersced69f82003-09-16 20:30:58 +000012191
Guido van Rossumd57fd912000-03-10 22:53:23 +000012192 s = PyUnicode_FromObject(s);
12193 if (s == NULL)
Benjamin Peterson14339b62009-01-31 16:36:08 +000012194 return NULL;
Benjamin Peterson29060642009-01-31 22:14:21 +000012195 if (sep != NULL) {
12196 sep = PyUnicode_FromObject(sep);
12197 if (sep == NULL) {
12198 Py_DECREF(s);
12199 return NULL;
12200 }
Guido van Rossumd57fd912000-03-10 22:53:23 +000012201 }
12202
Victor Stinner9310abb2011-10-05 00:59:23 +020012203 result = split(s, sep, maxsplit);
Guido van Rossumd57fd912000-03-10 22:53:23 +000012204
12205 Py_DECREF(s);
12206 Py_XDECREF(sep);
12207 return result;
12208}
12209
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000012210PyDoc_STRVAR(split__doc__,
Ezio Melotticda6b6d2012-02-26 09:39:55 +020012211 "S.split(sep=None, maxsplit=-1) -> list of strings\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000012212\n\
12213Return a list of the words in S, using sep as the\n\
12214delimiter string. If maxsplit is given, at most maxsplit\n\
Alexandre Vassalotti5f8ced22008-05-16 00:03:33 +000012215splits are done. If sep is not specified or is None, any\n\
Alexandre Vassalotti8ae3e052008-05-16 00:41:41 +000012216whitespace string is a separator and empty strings are\n\
12217removed from the result.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000012218
12219static PyObject*
Ezio Melotticda6b6d2012-02-26 09:39:55 +020012220unicode_split(PyObject *self, PyObject *args, PyObject *kwds)
Guido van Rossumd57fd912000-03-10 22:53:23 +000012221{
Ezio Melotticda6b6d2012-02-26 09:39:55 +020012222 static char *kwlist[] = {"sep", "maxsplit", 0};
Guido van Rossumd57fd912000-03-10 22:53:23 +000012223 PyObject *substring = Py_None;
Martin v. Löwis18e16552006-02-15 17:27:45 +000012224 Py_ssize_t maxcount = -1;
Guido van Rossumd57fd912000-03-10 22:53:23 +000012225
Ezio Melotticda6b6d2012-02-26 09:39:55 +020012226 if (!PyArg_ParseTupleAndKeywords(args, kwds, "|On:split",
12227 kwlist, &substring, &maxcount))
Guido van Rossumd57fd912000-03-10 22:53:23 +000012228 return NULL;
12229
12230 if (substring == Py_None)
Benjamin Peterson29060642009-01-31 22:14:21 +000012231 return split(self, NULL, maxcount);
Guido van Rossumd57fd912000-03-10 22:53:23 +000012232 else if (PyUnicode_Check(substring))
Victor Stinner9310abb2011-10-05 00:59:23 +020012233 return split(self, substring, maxcount);
Guido van Rossumd57fd912000-03-10 22:53:23 +000012234 else
Victor Stinner7931d9a2011-11-04 00:22:48 +010012235 return PyUnicode_Split(self, substring, maxcount);
Guido van Rossumd57fd912000-03-10 22:53:23 +000012236}
12237
Thomas Wouters477c8d52006-05-27 19:21:47 +000012238PyObject *
12239PyUnicode_Partition(PyObject *str_in, PyObject *sep_in)
12240{
12241 PyObject* str_obj;
12242 PyObject* sep_obj;
12243 PyObject* out;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012244 int kind1, kind2, kind;
12245 void *buf1 = NULL, *buf2 = NULL;
12246 Py_ssize_t len1, len2;
Thomas Wouters477c8d52006-05-27 19:21:47 +000012247
12248 str_obj = PyUnicode_FromObject(str_in);
Benjamin Peterson22a29702012-01-02 09:00:30 -060012249 if (!str_obj)
Benjamin Peterson29060642009-01-31 22:14:21 +000012250 return NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +000012251 sep_obj = PyUnicode_FromObject(sep_in);
Benjamin Peterson22a29702012-01-02 09:00:30 -060012252 if (!sep_obj) {
12253 Py_DECREF(str_obj);
12254 return NULL;
12255 }
12256 if (PyUnicode_READY(sep_obj) == -1 || PyUnicode_READY(str_obj) == -1) {
12257 Py_DECREF(sep_obj);
Thomas Wouters477c8d52006-05-27 19:21:47 +000012258 Py_DECREF(str_obj);
12259 return NULL;
12260 }
12261
Victor Stinner14f8f022011-10-05 20:58:25 +020012262 kind1 = PyUnicode_KIND(str_obj);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012263 kind2 = PyUnicode_KIND(sep_obj);
Victor Stinner14f8f022011-10-05 20:58:25 +020012264 kind = Py_MAX(kind1, kind2);
12265 buf1 = PyUnicode_DATA(str_obj);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012266 if (kind1 != kind)
Victor Stinner14f8f022011-10-05 20:58:25 +020012267 buf1 = _PyUnicode_AsKind(str_obj, kind);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012268 if (!buf1)
12269 goto onError;
12270 buf2 = PyUnicode_DATA(sep_obj);
12271 if (kind2 != kind)
12272 buf2 = _PyUnicode_AsKind(sep_obj, kind);
12273 if (!buf2)
12274 goto onError;
12275 len1 = PyUnicode_GET_LENGTH(str_obj);
12276 len2 = PyUnicode_GET_LENGTH(sep_obj);
12277
Benjamin Petersonead6b532011-12-20 17:23:42 -060012278 switch (PyUnicode_KIND(str_obj)) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012279 case PyUnicode_1BYTE_KIND:
Victor Stinnerc3cec782011-10-05 21:24:08 +020012280 if (PyUnicode_IS_ASCII(str_obj) && PyUnicode_IS_ASCII(sep_obj))
12281 out = asciilib_partition(str_obj, buf1, len1, sep_obj, buf2, len2);
12282 else
12283 out = ucs1lib_partition(str_obj, buf1, len1, sep_obj, buf2, len2);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012284 break;
12285 case PyUnicode_2BYTE_KIND:
12286 out = ucs2lib_partition(str_obj, buf1, len1, sep_obj, buf2, len2);
12287 break;
12288 case PyUnicode_4BYTE_KIND:
12289 out = ucs4lib_partition(str_obj, buf1, len1, sep_obj, buf2, len2);
12290 break;
12291 default:
12292 assert(0);
12293 out = 0;
12294 }
Thomas Wouters477c8d52006-05-27 19:21:47 +000012295
12296 Py_DECREF(sep_obj);
12297 Py_DECREF(str_obj);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012298 if (kind1 != kind)
12299 PyMem_Free(buf1);
12300 if (kind2 != kind)
12301 PyMem_Free(buf2);
Thomas Wouters477c8d52006-05-27 19:21:47 +000012302
12303 return out;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012304 onError:
12305 Py_DECREF(sep_obj);
12306 Py_DECREF(str_obj);
12307 if (kind1 != kind && buf1)
12308 PyMem_Free(buf1);
12309 if (kind2 != kind && buf2)
12310 PyMem_Free(buf2);
12311 return NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +000012312}
12313
12314
12315PyObject *
12316PyUnicode_RPartition(PyObject *str_in, PyObject *sep_in)
12317{
12318 PyObject* str_obj;
12319 PyObject* sep_obj;
12320 PyObject* out;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012321 int kind1, kind2, kind;
12322 void *buf1 = NULL, *buf2 = NULL;
12323 Py_ssize_t len1, len2;
Thomas Wouters477c8d52006-05-27 19:21:47 +000012324
12325 str_obj = PyUnicode_FromObject(str_in);
12326 if (!str_obj)
Benjamin Peterson29060642009-01-31 22:14:21 +000012327 return NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +000012328 sep_obj = PyUnicode_FromObject(sep_in);
12329 if (!sep_obj) {
12330 Py_DECREF(str_obj);
12331 return NULL;
12332 }
12333
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012334 kind1 = PyUnicode_KIND(str_in);
12335 kind2 = PyUnicode_KIND(sep_obj);
Georg Brandl4cb0de22011-09-28 21:49:49 +020012336 kind = Py_MAX(kind1, kind2);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012337 buf1 = PyUnicode_DATA(str_in);
12338 if (kind1 != kind)
12339 buf1 = _PyUnicode_AsKind(str_in, kind);
12340 if (!buf1)
12341 goto onError;
12342 buf2 = PyUnicode_DATA(sep_obj);
12343 if (kind2 != kind)
12344 buf2 = _PyUnicode_AsKind(sep_obj, kind);
12345 if (!buf2)
12346 goto onError;
12347 len1 = PyUnicode_GET_LENGTH(str_obj);
12348 len2 = PyUnicode_GET_LENGTH(sep_obj);
12349
Benjamin Petersonead6b532011-12-20 17:23:42 -060012350 switch (PyUnicode_KIND(str_in)) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012351 case PyUnicode_1BYTE_KIND:
Victor Stinnerc3cec782011-10-05 21:24:08 +020012352 if (PyUnicode_IS_ASCII(str_obj) && PyUnicode_IS_ASCII(sep_obj))
12353 out = asciilib_rpartition(str_obj, buf1, len1, sep_obj, buf2, len2);
12354 else
12355 out = ucs1lib_rpartition(str_obj, buf1, len1, sep_obj, buf2, len2);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012356 break;
12357 case PyUnicode_2BYTE_KIND:
12358 out = ucs2lib_rpartition(str_obj, buf1, len1, sep_obj, buf2, len2);
12359 break;
12360 case PyUnicode_4BYTE_KIND:
12361 out = ucs4lib_rpartition(str_obj, buf1, len1, sep_obj, buf2, len2);
12362 break;
12363 default:
12364 assert(0);
12365 out = 0;
12366 }
Thomas Wouters477c8d52006-05-27 19:21:47 +000012367
12368 Py_DECREF(sep_obj);
12369 Py_DECREF(str_obj);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012370 if (kind1 != kind)
12371 PyMem_Free(buf1);
12372 if (kind2 != kind)
12373 PyMem_Free(buf2);
Thomas Wouters477c8d52006-05-27 19:21:47 +000012374
12375 return out;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012376 onError:
12377 Py_DECREF(sep_obj);
12378 Py_DECREF(str_obj);
12379 if (kind1 != kind && buf1)
12380 PyMem_Free(buf1);
12381 if (kind2 != kind && buf2)
12382 PyMem_Free(buf2);
12383 return NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +000012384}
12385
12386PyDoc_STRVAR(partition__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000012387 "S.partition(sep) -> (head, sep, tail)\n\
Thomas Wouters477c8d52006-05-27 19:21:47 +000012388\n\
Georg Brandl17cb8a82008-05-30 08:20:09 +000012389Search for the separator sep in S, and return the part before it,\n\
Thomas Wouters477c8d52006-05-27 19:21:47 +000012390the separator itself, and the part after it. If the separator is not\n\
Benjamin Petersonf10a79a2008-10-11 00:49:57 +000012391found, return S and two empty strings.");
Thomas Wouters477c8d52006-05-27 19:21:47 +000012392
12393static PyObject*
Victor Stinner9310abb2011-10-05 00:59:23 +020012394unicode_partition(PyObject *self, PyObject *separator)
Thomas Wouters477c8d52006-05-27 19:21:47 +000012395{
Victor Stinner9310abb2011-10-05 00:59:23 +020012396 return PyUnicode_Partition(self, separator);
Thomas Wouters477c8d52006-05-27 19:21:47 +000012397}
12398
12399PyDoc_STRVAR(rpartition__doc__,
Ezio Melotti5b2b2422010-01-25 11:58:28 +000012400 "S.rpartition(sep) -> (head, sep, tail)\n\
Thomas Wouters477c8d52006-05-27 19:21:47 +000012401\n\
Georg Brandl17cb8a82008-05-30 08:20:09 +000012402Search for the separator sep in S, starting at the end of S, and return\n\
Thomas Wouters477c8d52006-05-27 19:21:47 +000012403the part before it, the separator itself, and the part after it. If the\n\
Benjamin Petersonf10a79a2008-10-11 00:49:57 +000012404separator is not found, return two empty strings and S.");
Thomas Wouters477c8d52006-05-27 19:21:47 +000012405
12406static PyObject*
Victor Stinner9310abb2011-10-05 00:59:23 +020012407unicode_rpartition(PyObject *self, PyObject *separator)
Thomas Wouters477c8d52006-05-27 19:21:47 +000012408{
Victor Stinner9310abb2011-10-05 00:59:23 +020012409 return PyUnicode_RPartition(self, separator);
Thomas Wouters477c8d52006-05-27 19:21:47 +000012410}
12411
Alexander Belopolsky40018472011-02-26 01:02:56 +000012412PyObject *
12413PyUnicode_RSplit(PyObject *s, PyObject *sep, Py_ssize_t maxsplit)
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +000012414{
12415 PyObject *result;
Benjamin Peterson14339b62009-01-31 16:36:08 +000012416
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +000012417 s = PyUnicode_FromObject(s);
12418 if (s == NULL)
Benjamin Peterson14339b62009-01-31 16:36:08 +000012419 return NULL;
Benjamin Peterson29060642009-01-31 22:14:21 +000012420 if (sep != NULL) {
12421 sep = PyUnicode_FromObject(sep);
12422 if (sep == NULL) {
12423 Py_DECREF(s);
12424 return NULL;
12425 }
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +000012426 }
12427
Victor Stinner9310abb2011-10-05 00:59:23 +020012428 result = rsplit(s, sep, maxsplit);
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +000012429
12430 Py_DECREF(s);
12431 Py_XDECREF(sep);
12432 return result;
12433}
12434
12435PyDoc_STRVAR(rsplit__doc__,
Ezio Melotticda6b6d2012-02-26 09:39:55 +020012436 "S.rsplit(sep=None, maxsplit=-1) -> list of strings\n\
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +000012437\n\
12438Return a list of the words in S, using sep as the\n\
12439delimiter string, starting at the end of the string and\n\
12440working to the front. If maxsplit is given, at most maxsplit\n\
12441splits are done. If sep is not specified, any whitespace string\n\
12442is a separator.");
12443
12444static PyObject*
Ezio Melotticda6b6d2012-02-26 09:39:55 +020012445unicode_rsplit(PyObject *self, PyObject *args, PyObject *kwds)
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +000012446{
Ezio Melotticda6b6d2012-02-26 09:39:55 +020012447 static char *kwlist[] = {"sep", "maxsplit", 0};
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +000012448 PyObject *substring = Py_None;
Martin v. Löwis18e16552006-02-15 17:27:45 +000012449 Py_ssize_t maxcount = -1;
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +000012450
Ezio Melotticda6b6d2012-02-26 09:39:55 +020012451 if (!PyArg_ParseTupleAndKeywords(args, kwds, "|On:rsplit",
12452 kwlist, &substring, &maxcount))
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +000012453 return NULL;
12454
12455 if (substring == Py_None)
Benjamin Peterson29060642009-01-31 22:14:21 +000012456 return rsplit(self, NULL, maxcount);
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +000012457 else if (PyUnicode_Check(substring))
Victor Stinner9310abb2011-10-05 00:59:23 +020012458 return rsplit(self, substring, maxcount);
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +000012459 else
Victor Stinner9310abb2011-10-05 00:59:23 +020012460 return PyUnicode_RSplit(self, substring, maxcount);
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +000012461}
12462
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000012463PyDoc_STRVAR(splitlines__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000012464 "S.splitlines([keepends]) -> list of strings\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000012465\n\
12466Return a list of the lines in S, breaking at line boundaries.\n\
Guido van Rossum86662912000-04-11 15:38:46 +000012467Line breaks are not included in the resulting list unless keepends\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000012468is given and true.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000012469
12470static PyObject*
Victor Stinner9db1a8b2011-10-23 20:04:37 +020012471unicode_splitlines(PyObject *self, PyObject *args, PyObject *kwds)
Guido van Rossumd57fd912000-03-10 22:53:23 +000012472{
Mark Dickinson0d5f6ad2011-09-24 09:14:39 +010012473 static char *kwlist[] = {"keepends", 0};
Guido van Rossum86662912000-04-11 15:38:46 +000012474 int keepends = 0;
Guido van Rossumd57fd912000-03-10 22:53:23 +000012475
Mark Dickinson0d5f6ad2011-09-24 09:14:39 +010012476 if (!PyArg_ParseTupleAndKeywords(args, kwds, "|i:splitlines",
12477 kwlist, &keepends))
Guido van Rossumd57fd912000-03-10 22:53:23 +000012478 return NULL;
12479
Victor Stinner9db1a8b2011-10-23 20:04:37 +020012480 return PyUnicode_Splitlines(self, keepends);
Guido van Rossumd57fd912000-03-10 22:53:23 +000012481}
12482
12483static
Guido van Rossumf15a29f2007-05-04 00:41:39 +000012484PyObject *unicode_str(PyObject *self)
Guido van Rossumd57fd912000-03-10 22:53:23 +000012485{
Victor Stinnerc4b49542011-12-11 22:44:26 +010012486 return unicode_result_unchanged(self);
Guido van Rossumd57fd912000-03-10 22:53:23 +000012487}
12488
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000012489PyDoc_STRVAR(swapcase__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000012490 "S.swapcase() -> str\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000012491\n\
12492Return a copy of S with uppercase characters converted to lowercase\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000012493and vice versa.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000012494
12495static PyObject*
Victor Stinner9310abb2011-10-05 00:59:23 +020012496unicode_swapcase(PyObject *self)
Guido van Rossumd57fd912000-03-10 22:53:23 +000012497{
Benjamin Petersoneea48462012-01-16 14:28:50 -050012498 if (PyUnicode_READY(self) == -1)
12499 return NULL;
Victor Stinnerb0800dc2012-02-25 00:47:08 +010012500 return case_operation(self, do_swapcase);
Guido van Rossumd57fd912000-03-10 22:53:23 +000012501}
12502
Georg Brandlceee0772007-11-27 23:48:05 +000012503PyDoc_STRVAR(maketrans__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000012504 "str.maketrans(x[, y[, z]]) -> dict (static method)\n\
Georg Brandlceee0772007-11-27 23:48:05 +000012505\n\
12506Return a translation table usable for str.translate().\n\
12507If there is only one argument, it must be a dictionary mapping Unicode\n\
12508ordinals (integers) or characters to Unicode ordinals, strings or None.\n\
Benjamin Peterson142957c2008-07-04 19:55:29 +000012509Character keys will be then converted to ordinals.\n\
Georg Brandlceee0772007-11-27 23:48:05 +000012510If there are two arguments, they must be strings of equal length, and\n\
12511in the resulting dictionary, each character in x will be mapped to the\n\
12512character at the same position in y. If there is a third argument, it\n\
12513must be a string, whose characters will be mapped to None in the result.");
12514
12515static PyObject*
Victor Stinner9db1a8b2011-10-23 20:04:37 +020012516unicode_maketrans(PyObject *null, PyObject *args)
Georg Brandlceee0772007-11-27 23:48:05 +000012517{
12518 PyObject *x, *y = NULL, *z = NULL;
12519 PyObject *new = NULL, *key, *value;
12520 Py_ssize_t i = 0;
12521 int res;
Benjamin Peterson14339b62009-01-31 16:36:08 +000012522
Georg Brandlceee0772007-11-27 23:48:05 +000012523 if (!PyArg_ParseTuple(args, "O|UU:maketrans", &x, &y, &z))
12524 return NULL;
12525 new = PyDict_New();
12526 if (!new)
12527 return NULL;
12528 if (y != NULL) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012529 int x_kind, y_kind, z_kind;
12530 void *x_data, *y_data, *z_data;
12531
Georg Brandlceee0772007-11-27 23:48:05 +000012532 /* x must be a string too, of equal length */
Georg Brandlceee0772007-11-27 23:48:05 +000012533 if (!PyUnicode_Check(x)) {
12534 PyErr_SetString(PyExc_TypeError, "first maketrans argument must "
12535 "be a string if there is a second argument");
12536 goto err;
12537 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012538 if (PyUnicode_GET_LENGTH(x) != PyUnicode_GET_LENGTH(y)) {
Georg Brandlceee0772007-11-27 23:48:05 +000012539 PyErr_SetString(PyExc_ValueError, "the first two maketrans "
12540 "arguments must have equal length");
12541 goto err;
12542 }
12543 /* create entries for translating chars in x to those in y */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012544 x_kind = PyUnicode_KIND(x);
12545 y_kind = PyUnicode_KIND(y);
12546 x_data = PyUnicode_DATA(x);
12547 y_data = PyUnicode_DATA(y);
12548 for (i = 0; i < PyUnicode_GET_LENGTH(x); i++) {
12549 key = PyLong_FromLong(PyUnicode_READ(x_kind, x_data, i));
Benjamin Peterson53aa1d72011-12-20 13:29:45 -060012550 if (!key)
Georg Brandlceee0772007-11-27 23:48:05 +000012551 goto err;
Benjamin Peterson822c7902011-12-20 13:32:50 -060012552 value = PyLong_FromLong(PyUnicode_READ(y_kind, y_data, i));
Benjamin Peterson53aa1d72011-12-20 13:29:45 -060012553 if (!value) {
12554 Py_DECREF(key);
12555 goto err;
12556 }
Georg Brandlceee0772007-11-27 23:48:05 +000012557 res = PyDict_SetItem(new, key, value);
12558 Py_DECREF(key);
12559 Py_DECREF(value);
12560 if (res < 0)
12561 goto err;
12562 }
12563 /* create entries for deleting chars in z */
12564 if (z != NULL) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012565 z_kind = PyUnicode_KIND(z);
12566 z_data = PyUnicode_DATA(z);
Victor Stinnerc4f281e2011-10-11 22:11:42 +020012567 for (i = 0; i < PyUnicode_GET_LENGTH(z); i++) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012568 key = PyLong_FromLong(PyUnicode_READ(z_kind, z_data, i));
Georg Brandlceee0772007-11-27 23:48:05 +000012569 if (!key)
12570 goto err;
12571 res = PyDict_SetItem(new, key, Py_None);
12572 Py_DECREF(key);
12573 if (res < 0)
12574 goto err;
12575 }
12576 }
12577 } else {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012578 int kind;
12579 void *data;
12580
Georg Brandlceee0772007-11-27 23:48:05 +000012581 /* x must be a dict */
Raymond Hettinger3ad05762009-05-29 22:11:22 +000012582 if (!PyDict_CheckExact(x)) {
Georg Brandlceee0772007-11-27 23:48:05 +000012583 PyErr_SetString(PyExc_TypeError, "if you give only one argument "
12584 "to maketrans it must be a dict");
12585 goto err;
12586 }
12587 /* copy entries into the new dict, converting string keys to int keys */
12588 while (PyDict_Next(x, &i, &key, &value)) {
12589 if (PyUnicode_Check(key)) {
12590 /* convert string keys to integer keys */
12591 PyObject *newkey;
Victor Stinnerc4f281e2011-10-11 22:11:42 +020012592 if (PyUnicode_GET_LENGTH(key) != 1) {
Georg Brandlceee0772007-11-27 23:48:05 +000012593 PyErr_SetString(PyExc_ValueError, "string keys in translate "
12594 "table must be of length 1");
12595 goto err;
12596 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012597 kind = PyUnicode_KIND(key);
12598 data = PyUnicode_DATA(key);
12599 newkey = PyLong_FromLong(PyUnicode_READ(kind, data, 0));
Georg Brandlceee0772007-11-27 23:48:05 +000012600 if (!newkey)
12601 goto err;
12602 res = PyDict_SetItem(new, newkey, value);
12603 Py_DECREF(newkey);
12604 if (res < 0)
12605 goto err;
Christian Heimes217cfd12007-12-02 14:31:20 +000012606 } else if (PyLong_Check(key)) {
Georg Brandlceee0772007-11-27 23:48:05 +000012607 /* just keep integer keys */
12608 if (PyDict_SetItem(new, key, value) < 0)
12609 goto err;
12610 } else {
12611 PyErr_SetString(PyExc_TypeError, "keys in translate table must "
12612 "be strings or integers");
12613 goto err;
12614 }
12615 }
12616 }
12617 return new;
12618 err:
12619 Py_DECREF(new);
12620 return NULL;
12621}
12622
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000012623PyDoc_STRVAR(translate__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000012624 "S.translate(table) -> str\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000012625\n\
12626Return a copy of the string S, where all characters have been mapped\n\
12627through the given translation table, which must be a mapping of\n\
Benjamin Peterson142957c2008-07-04 19:55:29 +000012628Unicode ordinals to Unicode ordinals, strings, or None.\n\
Walter Dörwald5c1ee172002-09-04 20:31:32 +000012629Unmapped characters are left untouched. Characters mapped to None\n\
12630are deleted.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000012631
12632static PyObject*
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012633unicode_translate(PyObject *self, PyObject *table)
Guido van Rossumd57fd912000-03-10 22:53:23 +000012634{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012635 return _PyUnicode_TranslateCharmap(self, table, "ignore");
Guido van Rossumd57fd912000-03-10 22:53:23 +000012636}
12637
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000012638PyDoc_STRVAR(upper__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000012639 "S.upper() -> str\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000012640\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000012641Return a copy of S converted to uppercase.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000012642
12643static PyObject*
Victor Stinner9310abb2011-10-05 00:59:23 +020012644unicode_upper(PyObject *self)
Guido van Rossumd57fd912000-03-10 22:53:23 +000012645{
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -050012646 if (PyUnicode_READY(self) == -1)
12647 return NULL;
12648 if (PyUnicode_IS_ASCII(self))
12649 return ascii_upper_or_lower(self, 0);
Victor Stinnerb0800dc2012-02-25 00:47:08 +010012650 return case_operation(self, do_upper);
Guido van Rossumd57fd912000-03-10 22:53:23 +000012651}
12652
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000012653PyDoc_STRVAR(zfill__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000012654 "S.zfill(width) -> str\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000012655\n\
Benjamin Peterson9aa42992008-09-10 21:57:34 +000012656Pad a numeric string S with zeros on the left, to fill a field\n\
12657of the specified width. The string S is never truncated.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000012658
12659static PyObject *
Victor Stinner9310abb2011-10-05 00:59:23 +020012660unicode_zfill(PyObject *self, PyObject *args)
Guido van Rossumd57fd912000-03-10 22:53:23 +000012661{
Martin v. Löwis18e16552006-02-15 17:27:45 +000012662 Py_ssize_t fill;
Victor Stinner9310abb2011-10-05 00:59:23 +020012663 PyObject *u;
Martin v. Löwis18e16552006-02-15 17:27:45 +000012664 Py_ssize_t width;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012665 int kind;
12666 void *data;
12667 Py_UCS4 chr;
12668
Martin v. Löwis18e16552006-02-15 17:27:45 +000012669 if (!PyArg_ParseTuple(args, "n:zfill", &width))
Guido van Rossumd57fd912000-03-10 22:53:23 +000012670 return NULL;
12671
Benjamin Petersonbac79492012-01-14 13:34:47 -050012672 if (PyUnicode_READY(self) == -1)
Victor Stinnerc4b49542011-12-11 22:44:26 +010012673 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +000012674
Victor Stinnerc4b49542011-12-11 22:44:26 +010012675 if (PyUnicode_GET_LENGTH(self) >= width)
12676 return unicode_result_unchanged(self);
12677
12678 fill = width - PyUnicode_GET_LENGTH(self);
Guido van Rossumd57fd912000-03-10 22:53:23 +000012679
12680 u = pad(self, fill, 0, '0');
12681
Walter Dörwald068325e2002-04-15 13:36:47 +000012682 if (u == NULL)
12683 return NULL;
12684
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012685 kind = PyUnicode_KIND(u);
12686 data = PyUnicode_DATA(u);
12687 chr = PyUnicode_READ(kind, data, fill);
12688
12689 if (chr == '+' || chr == '-') {
Guido van Rossumd57fd912000-03-10 22:53:23 +000012690 /* move sign to beginning of string */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012691 PyUnicode_WRITE(kind, data, 0, chr);
12692 PyUnicode_WRITE(kind, data, fill, '0');
Guido van Rossumd57fd912000-03-10 22:53:23 +000012693 }
12694
Victor Stinnerbb10a1f2011-10-05 01:34:17 +020012695 assert(_PyUnicode_CheckConsistency(u, 1));
Victor Stinner7931d9a2011-11-04 00:22:48 +010012696 return u;
Guido van Rossumd57fd912000-03-10 22:53:23 +000012697}
Guido van Rossumd57fd912000-03-10 22:53:23 +000012698
12699#if 0
Alexander Belopolsky942af5a2010-12-04 03:38:46 +000012700static PyObject *
12701unicode__decimal2ascii(PyObject *self)
12702{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012703 return PyUnicode_TransformDecimalAndSpaceToASCII(self);
Alexander Belopolsky942af5a2010-12-04 03:38:46 +000012704}
Guido van Rossumd57fd912000-03-10 22:53:23 +000012705#endif
12706
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000012707PyDoc_STRVAR(startswith__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000012708 "S.startswith(prefix[, start[, end]]) -> bool\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000012709\n\
Guido van Rossuma7132182003-04-09 19:32:45 +000012710Return True if S starts with the specified prefix, False otherwise.\n\
12711With optional start, test S beginning at that position.\n\
Thomas Wouters0e3f5912006-08-11 14:57:12 +000012712With optional end, stop comparing S at that position.\n\
12713prefix can also be a tuple of strings to try.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000012714
12715static PyObject *
Victor Stinner9db1a8b2011-10-23 20:04:37 +020012716unicode_startswith(PyObject *self,
Benjamin Peterson29060642009-01-31 22:14:21 +000012717 PyObject *args)
Guido van Rossumd57fd912000-03-10 22:53:23 +000012718{
Thomas Wouters0e3f5912006-08-11 14:57:12 +000012719 PyObject *subobj;
Victor Stinner9db1a8b2011-10-23 20:04:37 +020012720 PyObject *substring;
Martin v. Löwis18e16552006-02-15 17:27:45 +000012721 Py_ssize_t start = 0;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000012722 Py_ssize_t end = PY_SSIZE_T_MAX;
Thomas Wouters0e3f5912006-08-11 14:57:12 +000012723 int result;
Guido van Rossumd57fd912000-03-10 22:53:23 +000012724
Jesus Ceaac451502011-04-20 17:09:23 +020012725 if (!stringlib_parse_args_finds("startswith", args, &subobj, &start, &end))
Benjamin Peterson29060642009-01-31 22:14:21 +000012726 return NULL;
Thomas Wouters0e3f5912006-08-11 14:57:12 +000012727 if (PyTuple_Check(subobj)) {
12728 Py_ssize_t i;
12729 for (i = 0; i < PyTuple_GET_SIZE(subobj); i++) {
Victor Stinner9db1a8b2011-10-23 20:04:37 +020012730 substring = PyUnicode_FromObject(PyTuple_GET_ITEM(subobj, i));
Thomas Wouters0e3f5912006-08-11 14:57:12 +000012731 if (substring == NULL)
12732 return NULL;
12733 result = tailmatch(self, substring, start, end, -1);
12734 Py_DECREF(substring);
12735 if (result) {
12736 Py_RETURN_TRUE;
12737 }
12738 }
12739 /* nothing matched */
12740 Py_RETURN_FALSE;
12741 }
Victor Stinner9db1a8b2011-10-23 20:04:37 +020012742 substring = PyUnicode_FromObject(subobj);
Ezio Melottiba42fd52011-04-26 06:09:45 +030012743 if (substring == NULL) {
12744 if (PyErr_ExceptionMatches(PyExc_TypeError))
12745 PyErr_Format(PyExc_TypeError, "startswith first arg must be str or "
12746 "a tuple of str, not %s", Py_TYPE(subobj)->tp_name);
Benjamin Peterson29060642009-01-31 22:14:21 +000012747 return NULL;
Ezio Melottiba42fd52011-04-26 06:09:45 +030012748 }
Thomas Wouters0e3f5912006-08-11 14:57:12 +000012749 result = tailmatch(self, substring, start, end, -1);
Guido van Rossumd57fd912000-03-10 22:53:23 +000012750 Py_DECREF(substring);
Thomas Wouters0e3f5912006-08-11 14:57:12 +000012751 return PyBool_FromLong(result);
Guido van Rossumd57fd912000-03-10 22:53:23 +000012752}
12753
12754
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000012755PyDoc_STRVAR(endswith__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000012756 "S.endswith(suffix[, start[, end]]) -> bool\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000012757\n\
Guido van Rossuma7132182003-04-09 19:32:45 +000012758Return True if S ends with the specified suffix, False otherwise.\n\
12759With optional start, test S beginning at that position.\n\
Thomas Wouters0e3f5912006-08-11 14:57:12 +000012760With optional end, stop comparing S at that position.\n\
12761suffix can also be a tuple of strings to try.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000012762
12763static PyObject *
Victor Stinner9db1a8b2011-10-23 20:04:37 +020012764unicode_endswith(PyObject *self,
Benjamin Peterson29060642009-01-31 22:14:21 +000012765 PyObject *args)
Guido van Rossumd57fd912000-03-10 22:53:23 +000012766{
Thomas Wouters0e3f5912006-08-11 14:57:12 +000012767 PyObject *subobj;
Victor Stinner9db1a8b2011-10-23 20:04:37 +020012768 PyObject *substring;
Martin v. Löwis18e16552006-02-15 17:27:45 +000012769 Py_ssize_t start = 0;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000012770 Py_ssize_t end = PY_SSIZE_T_MAX;
Thomas Wouters0e3f5912006-08-11 14:57:12 +000012771 int result;
Guido van Rossumd57fd912000-03-10 22:53:23 +000012772
Jesus Ceaac451502011-04-20 17:09:23 +020012773 if (!stringlib_parse_args_finds("endswith", args, &subobj, &start, &end))
Benjamin Peterson29060642009-01-31 22:14:21 +000012774 return NULL;
Thomas Wouters0e3f5912006-08-11 14:57:12 +000012775 if (PyTuple_Check(subobj)) {
12776 Py_ssize_t i;
12777 for (i = 0; i < PyTuple_GET_SIZE(subobj); i++) {
Victor Stinner9db1a8b2011-10-23 20:04:37 +020012778 substring = PyUnicode_FromObject(
Benjamin Peterson29060642009-01-31 22:14:21 +000012779 PyTuple_GET_ITEM(subobj, i));
Thomas Wouters0e3f5912006-08-11 14:57:12 +000012780 if (substring == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +000012781 return NULL;
Thomas Wouters0e3f5912006-08-11 14:57:12 +000012782 result = tailmatch(self, substring, start, end, +1);
12783 Py_DECREF(substring);
12784 if (result) {
12785 Py_RETURN_TRUE;
12786 }
12787 }
12788 Py_RETURN_FALSE;
12789 }
Victor Stinner9db1a8b2011-10-23 20:04:37 +020012790 substring = PyUnicode_FromObject(subobj);
Ezio Melottiba42fd52011-04-26 06:09:45 +030012791 if (substring == NULL) {
12792 if (PyErr_ExceptionMatches(PyExc_TypeError))
12793 PyErr_Format(PyExc_TypeError, "endswith first arg must be str or "
12794 "a tuple of str, not %s", Py_TYPE(subobj)->tp_name);
Benjamin Peterson29060642009-01-31 22:14:21 +000012795 return NULL;
Ezio Melottiba42fd52011-04-26 06:09:45 +030012796 }
Thomas Wouters0e3f5912006-08-11 14:57:12 +000012797 result = tailmatch(self, substring, start, end, +1);
Guido van Rossumd57fd912000-03-10 22:53:23 +000012798 Py_DECREF(substring);
Thomas Wouters0e3f5912006-08-11 14:57:12 +000012799 return PyBool_FromLong(result);
Guido van Rossumd57fd912000-03-10 22:53:23 +000012800}
12801
Victor Stinner202fdca2012-05-07 12:47:02 +020012802Py_LOCAL_INLINE(void)
Victor Stinner3b1a74a2012-05-09 22:25:00 +020012803_PyUnicodeWriter_Update(_PyUnicodeWriter *writer)
Victor Stinner202fdca2012-05-07 12:47:02 +020012804{
Victor Stinnerd3f08822012-05-29 12:57:52 +020012805 writer->size = PyUnicode_GET_LENGTH(writer->buffer);
Victor Stinner202fdca2012-05-07 12:47:02 +020012806 writer->maxchar = PyUnicode_MAX_CHAR_VALUE(writer->buffer);
12807 writer->data = PyUnicode_DATA(writer->buffer);
12808 writer->kind = PyUnicode_KIND(writer->buffer);
12809}
12810
Victor Stinnerd3f08822012-05-29 12:57:52 +020012811void
12812_PyUnicodeWriter_Init(_PyUnicodeWriter *writer, Py_ssize_t min_length)
Victor Stinner202fdca2012-05-07 12:47:02 +020012813{
Victor Stinnerd3f08822012-05-29 12:57:52 +020012814 memset(writer, 0, sizeof(*writer));
12815#ifdef Py_DEBUG
12816 writer->kind = 5; /* invalid kind */
12817#endif
12818 writer->min_length = Py_MAX(min_length, 100);
Victor Stinnerd7b7c742012-06-04 22:52:12 +020012819 writer->overallocate = (min_length > 0);
Victor Stinner202fdca2012-05-07 12:47:02 +020012820}
12821
Victor Stinnerd3f08822012-05-29 12:57:52 +020012822int
12823_PyUnicodeWriter_PrepareInternal(_PyUnicodeWriter *writer,
12824 Py_ssize_t length, Py_UCS4 maxchar)
Victor Stinner202fdca2012-05-07 12:47:02 +020012825{
12826 Py_ssize_t newlen;
12827 PyObject *newbuffer;
12828
Victor Stinnerd3f08822012-05-29 12:57:52 +020012829 assert(length > 0);
12830
Victor Stinner202fdca2012-05-07 12:47:02 +020012831 if (length > PY_SSIZE_T_MAX - writer->pos) {
12832 PyErr_NoMemory();
12833 return -1;
12834 }
12835 newlen = writer->pos + length;
12836
Victor Stinnerd3f08822012-05-29 12:57:52 +020012837 if (writer->buffer == NULL) {
Victor Stinnerd7b7c742012-06-04 22:52:12 +020012838 if (writer->overallocate) {
Victor Stinnerd3f08822012-05-29 12:57:52 +020012839 /* overallocate 25% to limit the number of resize */
12840 if (newlen <= (PY_SSIZE_T_MAX - newlen / 4))
12841 newlen += newlen / 4;
12842 if (newlen < writer->min_length)
12843 newlen = writer->min_length;
12844 }
12845 writer->buffer = PyUnicode_New(newlen, maxchar);
12846 if (writer->buffer == NULL)
12847 return -1;
12848 _PyUnicodeWriter_Update(writer);
12849 return 0;
12850 }
Victor Stinner202fdca2012-05-07 12:47:02 +020012851
Victor Stinnerd3f08822012-05-29 12:57:52 +020012852 if (newlen > writer->size) {
Victor Stinnerd7b7c742012-06-04 22:52:12 +020012853 if (writer->overallocate) {
Victor Stinnerd3f08822012-05-29 12:57:52 +020012854 /* overallocate 25% to limit the number of resize */
12855 if (newlen <= (PY_SSIZE_T_MAX - newlen / 4))
12856 newlen += newlen / 4;
12857 if (newlen < writer->min_length)
12858 newlen = writer->min_length;
12859 }
12860
Victor Stinnerd7b7c742012-06-04 22:52:12 +020012861 if (maxchar > writer->maxchar || writer->readonly) {
Victor Stinner202fdca2012-05-07 12:47:02 +020012862 /* resize + widen */
12863 newbuffer = PyUnicode_New(newlen, maxchar);
12864 if (newbuffer == NULL)
12865 return -1;
Victor Stinnerd3f08822012-05-29 12:57:52 +020012866 _PyUnicode_FastCopyCharacters(newbuffer, 0,
12867 writer->buffer, 0, writer->pos);
Victor Stinner202fdca2012-05-07 12:47:02 +020012868 Py_DECREF(writer->buffer);
Victor Stinnerd7b7c742012-06-04 22:52:12 +020012869 writer->readonly = 0;
Victor Stinner202fdca2012-05-07 12:47:02 +020012870 }
12871 else {
12872 newbuffer = resize_compact(writer->buffer, newlen);
12873 if (newbuffer == NULL)
12874 return -1;
12875 }
12876 writer->buffer = newbuffer;
Victor Stinner3b1a74a2012-05-09 22:25:00 +020012877 _PyUnicodeWriter_Update(writer);
Victor Stinner202fdca2012-05-07 12:47:02 +020012878 }
12879 else if (maxchar > writer->maxchar) {
Victor Stinnerd7b7c742012-06-04 22:52:12 +020012880 assert(!writer->readonly);
Victor Stinnerd3f08822012-05-29 12:57:52 +020012881 newbuffer = PyUnicode_New(writer->size, maxchar);
12882 if (newbuffer == NULL)
Victor Stinner202fdca2012-05-07 12:47:02 +020012883 return -1;
Victor Stinnerd3f08822012-05-29 12:57:52 +020012884 _PyUnicode_FastCopyCharacters(newbuffer, 0,
12885 writer->buffer, 0, writer->pos);
12886 Py_DECREF(writer->buffer);
12887 writer->buffer = newbuffer;
Victor Stinner3b1a74a2012-05-09 22:25:00 +020012888 _PyUnicodeWriter_Update(writer);
Victor Stinner202fdca2012-05-07 12:47:02 +020012889 }
12890 return 0;
12891}
12892
Victor Stinnerd3f08822012-05-29 12:57:52 +020012893int
12894_PyUnicodeWriter_WriteStr(_PyUnicodeWriter *writer, PyObject *str)
12895{
12896 Py_UCS4 maxchar;
12897 Py_ssize_t len;
12898
12899 if (PyUnicode_READY(str) == -1)
12900 return -1;
12901 len = PyUnicode_GET_LENGTH(str);
12902 if (len == 0)
12903 return 0;
12904 maxchar = PyUnicode_MAX_CHAR_VALUE(str);
12905 if (maxchar > writer->maxchar || len > writer->size - writer->pos) {
Victor Stinnerd7b7c742012-06-04 22:52:12 +020012906 if (writer->buffer == NULL && !writer->overallocate) {
Victor Stinnerd3f08822012-05-29 12:57:52 +020012907 Py_INCREF(str);
12908 writer->buffer = str;
12909 _PyUnicodeWriter_Update(writer);
Victor Stinnerd7b7c742012-06-04 22:52:12 +020012910 writer->readonly = 1;
Victor Stinnerd3f08822012-05-29 12:57:52 +020012911 writer->size = 0;
12912 writer->pos += len;
12913 return 0;
12914 }
12915 if (_PyUnicodeWriter_PrepareInternal(writer, len, maxchar) == -1)
12916 return -1;
12917 }
12918 _PyUnicode_FastCopyCharacters(writer->buffer, writer->pos,
12919 str, 0, len);
12920 writer->pos += len;
12921 return 0;
12922}
12923
12924PyObject *
Victor Stinner3b1a74a2012-05-09 22:25:00 +020012925_PyUnicodeWriter_Finish(_PyUnicodeWriter *writer)
Victor Stinner202fdca2012-05-07 12:47:02 +020012926{
Victor Stinnerd3f08822012-05-29 12:57:52 +020012927 if (writer->pos == 0) {
12928 Py_XDECREF(writer->buffer);
12929 Py_INCREF(unicode_empty);
12930 return unicode_empty;
12931 }
Victor Stinnerd7b7c742012-06-04 22:52:12 +020012932 if (writer->readonly) {
Victor Stinnerd3f08822012-05-29 12:57:52 +020012933 assert(PyUnicode_GET_LENGTH(writer->buffer) == writer->pos);
12934 return writer->buffer;
12935 }
12936 if (PyUnicode_GET_LENGTH(writer->buffer) != writer->pos) {
12937 PyObject *newbuffer;
12938 newbuffer = resize_compact(writer->buffer, writer->pos);
12939 if (newbuffer == NULL) {
12940 Py_DECREF(writer->buffer);
12941 return NULL;
12942 }
12943 writer->buffer = newbuffer;
Victor Stinner202fdca2012-05-07 12:47:02 +020012944 }
Victor Stinnerf59c28c2012-05-09 03:24:14 +020012945 assert(_PyUnicode_CheckConsistency(writer->buffer, 1));
Victor Stinner202fdca2012-05-07 12:47:02 +020012946 return writer->buffer;
12947}
12948
Victor Stinnerd3f08822012-05-29 12:57:52 +020012949void
Victor Stinner3b1a74a2012-05-09 22:25:00 +020012950_PyUnicodeWriter_Dealloc(_PyUnicodeWriter *writer)
Victor Stinner202fdca2012-05-07 12:47:02 +020012951{
12952 Py_CLEAR(writer->buffer);
12953}
12954
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012955#include "stringlib/unicode_format.h"
Eric Smith8c663262007-08-25 02:26:07 +000012956
12957PyDoc_STRVAR(format__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000012958 "S.format(*args, **kwargs) -> str\n\
Eric Smith8c663262007-08-25 02:26:07 +000012959\n\
Eric Smith51d2fd92010-11-06 19:27:37 +000012960Return a formatted version of S, using substitutions from args and kwargs.\n\
12961The substitutions are identified by braces ('{' and '}').");
Eric Smith8c663262007-08-25 02:26:07 +000012962
Eric Smith27bbca62010-11-04 17:06:58 +000012963PyDoc_STRVAR(format_map__doc__,
12964 "S.format_map(mapping) -> str\n\
12965\n\
Eric Smith51d2fd92010-11-06 19:27:37 +000012966Return a formatted version of S, using substitutions from mapping.\n\
12967The substitutions are identified by braces ('{' and '}').");
Eric Smith27bbca62010-11-04 17:06:58 +000012968
Eric Smith4a7d76d2008-05-30 18:10:19 +000012969static PyObject *
12970unicode__format__(PyObject* self, PyObject* args)
12971{
Victor Stinnerd3f08822012-05-29 12:57:52 +020012972 PyObject *format_spec;
12973 _PyUnicodeWriter writer;
12974 int ret;
Eric Smith4a7d76d2008-05-30 18:10:19 +000012975
12976 if (!PyArg_ParseTuple(args, "U:__format__", &format_spec))
12977 return NULL;
12978
Victor Stinnerd3f08822012-05-29 12:57:52 +020012979 if (PyUnicode_READY(self) == -1)
12980 return NULL;
12981 _PyUnicodeWriter_Init(&writer, 0);
12982 ret = _PyUnicode_FormatAdvancedWriter(&writer,
12983 self, format_spec, 0,
12984 PyUnicode_GET_LENGTH(format_spec));
12985 if (ret == -1) {
12986 _PyUnicodeWriter_Dealloc(&writer);
12987 return NULL;
12988 }
12989 return _PyUnicodeWriter_Finish(&writer);
Eric Smith4a7d76d2008-05-30 18:10:19 +000012990}
12991
Eric Smith8c663262007-08-25 02:26:07 +000012992PyDoc_STRVAR(p_format__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000012993 "S.__format__(format_spec) -> str\n\
Eric Smith8c663262007-08-25 02:26:07 +000012994\n\
Eric Smith51d2fd92010-11-06 19:27:37 +000012995Return a formatted version of S as described by format_spec.");
Eric Smith8c663262007-08-25 02:26:07 +000012996
12997static PyObject *
Victor Stinner9db1a8b2011-10-23 20:04:37 +020012998unicode__sizeof__(PyObject *v)
Georg Brandlc28e1fa2008-06-10 19:20:26 +000012999{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013000 Py_ssize_t size;
13001
13002 /* If it's a compact object, account for base structure +
13003 character data. */
13004 if (PyUnicode_IS_COMPACT_ASCII(v))
13005 size = sizeof(PyASCIIObject) + PyUnicode_GET_LENGTH(v) + 1;
13006 else if (PyUnicode_IS_COMPACT(v))
13007 size = sizeof(PyCompactUnicodeObject) +
Martin v. Löwisc47adb02011-10-07 20:55:35 +020013008 (PyUnicode_GET_LENGTH(v) + 1) * PyUnicode_KIND(v);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013009 else {
13010 /* If it is a two-block object, account for base object, and
13011 for character block if present. */
13012 size = sizeof(PyUnicodeObject);
Victor Stinnerc3c74152011-10-02 20:39:55 +020013013 if (_PyUnicode_DATA_ANY(v))
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013014 size += (PyUnicode_GET_LENGTH(v) + 1) *
Martin v. Löwisc47adb02011-10-07 20:55:35 +020013015 PyUnicode_KIND(v);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013016 }
13017 /* If the wstr pointer is present, account for it unless it is shared
Victor Stinnera3be6132011-10-03 02:16:37 +020013018 with the data pointer. Check if the data is not shared. */
Victor Stinner03490912011-10-03 23:45:12 +020013019 if (_PyUnicode_HAS_WSTR_MEMORY(v))
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013020 size += (PyUnicode_WSTR_LENGTH(v) + 1) * sizeof(wchar_t);
Victor Stinner829c0ad2011-10-03 01:08:02 +020013021 if (_PyUnicode_HAS_UTF8_MEMORY(v))
Victor Stinnere90fe6a2011-10-01 16:48:13 +020013022 size += PyUnicode_UTF8_LENGTH(v) + 1;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013023
13024 return PyLong_FromSsize_t(size);
Georg Brandlc28e1fa2008-06-10 19:20:26 +000013025}
13026
13027PyDoc_STRVAR(sizeof__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000013028 "S.__sizeof__() -> size of S in memory, in bytes");
Georg Brandlc28e1fa2008-06-10 19:20:26 +000013029
13030static PyObject *
Victor Stinner034f6cf2011-09-30 02:26:44 +020013031unicode_getnewargs(PyObject *v)
Guido van Rossum5d9113d2003-01-29 17:58:45 +000013032{
Victor Stinnerbf6e5602011-12-12 01:53:47 +010013033 PyObject *copy = _PyUnicode_Copy(v);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013034 if (!copy)
13035 return NULL;
13036 return Py_BuildValue("(N)", copy);
Guido van Rossum5d9113d2003-01-29 17:58:45 +000013037}
13038
Guido van Rossumd57fd912000-03-10 22:53:23 +000013039static PyMethodDef unicode_methods[] = {
Benjamin Peterson28a4dce2010-12-12 01:33:04 +000013040 {"encode", (PyCFunction) unicode_encode, METH_VARARGS | METH_KEYWORDS, encode__doc__},
Martin v. Löwise3eb1f22001-08-16 13:15:00 +000013041 {"replace", (PyCFunction) unicode_replace, METH_VARARGS, replace__doc__},
Ezio Melotticda6b6d2012-02-26 09:39:55 +020013042 {"split", (PyCFunction) unicode_split, METH_VARARGS | METH_KEYWORDS, split__doc__},
13043 {"rsplit", (PyCFunction) unicode_rsplit, METH_VARARGS | METH_KEYWORDS, rsplit__doc__},
Martin v. Löwise3eb1f22001-08-16 13:15:00 +000013044 {"join", (PyCFunction) unicode_join, METH_O, join__doc__},
13045 {"capitalize", (PyCFunction) unicode_capitalize, METH_NOARGS, capitalize__doc__},
Benjamin Petersond5890c82012-01-14 13:23:30 -050013046 {"casefold", (PyCFunction) unicode_casefold, METH_NOARGS, casefold__doc__},
Martin v. Löwise3eb1f22001-08-16 13:15:00 +000013047 {"title", (PyCFunction) unicode_title, METH_NOARGS, title__doc__},
13048 {"center", (PyCFunction) unicode_center, METH_VARARGS, center__doc__},
13049 {"count", (PyCFunction) unicode_count, METH_VARARGS, count__doc__},
13050 {"expandtabs", (PyCFunction) unicode_expandtabs, METH_VARARGS, expandtabs__doc__},
13051 {"find", (PyCFunction) unicode_find, METH_VARARGS, find__doc__},
Thomas Wouters477c8d52006-05-27 19:21:47 +000013052 {"partition", (PyCFunction) unicode_partition, METH_O, partition__doc__},
Martin v. Löwise3eb1f22001-08-16 13:15:00 +000013053 {"index", (PyCFunction) unicode_index, METH_VARARGS, index__doc__},
13054 {"ljust", (PyCFunction) unicode_ljust, METH_VARARGS, ljust__doc__},
13055 {"lower", (PyCFunction) unicode_lower, METH_NOARGS, lower__doc__},
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000013056 {"lstrip", (PyCFunction) unicode_lstrip, METH_VARARGS, lstrip__doc__},
Martin v. Löwise3eb1f22001-08-16 13:15:00 +000013057 {"rfind", (PyCFunction) unicode_rfind, METH_VARARGS, rfind__doc__},
13058 {"rindex", (PyCFunction) unicode_rindex, METH_VARARGS, rindex__doc__},
13059 {"rjust", (PyCFunction) unicode_rjust, METH_VARARGS, rjust__doc__},
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000013060 {"rstrip", (PyCFunction) unicode_rstrip, METH_VARARGS, rstrip__doc__},
Thomas Wouters477c8d52006-05-27 19:21:47 +000013061 {"rpartition", (PyCFunction) unicode_rpartition, METH_O, rpartition__doc__},
Mark Dickinson0d5f6ad2011-09-24 09:14:39 +010013062 {"splitlines", (PyCFunction) unicode_splitlines, METH_VARARGS | METH_KEYWORDS, splitlines__doc__},
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000013063 {"strip", (PyCFunction) unicode_strip, METH_VARARGS, strip__doc__},
Martin v. Löwise3eb1f22001-08-16 13:15:00 +000013064 {"swapcase", (PyCFunction) unicode_swapcase, METH_NOARGS, swapcase__doc__},
13065 {"translate", (PyCFunction) unicode_translate, METH_O, translate__doc__},
13066 {"upper", (PyCFunction) unicode_upper, METH_NOARGS, upper__doc__},
13067 {"startswith", (PyCFunction) unicode_startswith, METH_VARARGS, startswith__doc__},
13068 {"endswith", (PyCFunction) unicode_endswith, METH_VARARGS, endswith__doc__},
13069 {"islower", (PyCFunction) unicode_islower, METH_NOARGS, islower__doc__},
13070 {"isupper", (PyCFunction) unicode_isupper, METH_NOARGS, isupper__doc__},
13071 {"istitle", (PyCFunction) unicode_istitle, METH_NOARGS, istitle__doc__},
13072 {"isspace", (PyCFunction) unicode_isspace, METH_NOARGS, isspace__doc__},
13073 {"isdecimal", (PyCFunction) unicode_isdecimal, METH_NOARGS, isdecimal__doc__},
13074 {"isdigit", (PyCFunction) unicode_isdigit, METH_NOARGS, isdigit__doc__},
13075 {"isnumeric", (PyCFunction) unicode_isnumeric, METH_NOARGS, isnumeric__doc__},
13076 {"isalpha", (PyCFunction) unicode_isalpha, METH_NOARGS, isalpha__doc__},
13077 {"isalnum", (PyCFunction) unicode_isalnum, METH_NOARGS, isalnum__doc__},
Martin v. Löwis47383402007-08-15 07:32:56 +000013078 {"isidentifier", (PyCFunction) unicode_isidentifier, METH_NOARGS, isidentifier__doc__},
Georg Brandl559e5d72008-06-11 18:37:52 +000013079 {"isprintable", (PyCFunction) unicode_isprintable, METH_NOARGS, isprintable__doc__},
Martin v. Löwise3eb1f22001-08-16 13:15:00 +000013080 {"zfill", (PyCFunction) unicode_zfill, METH_VARARGS, zfill__doc__},
Eric Smith9cd1e092007-08-31 18:39:38 +000013081 {"format", (PyCFunction) do_string_format, METH_VARARGS | METH_KEYWORDS, format__doc__},
Eric Smith27bbca62010-11-04 17:06:58 +000013082 {"format_map", (PyCFunction) do_string_format_map, METH_O, format_map__doc__},
Eric Smith4a7d76d2008-05-30 18:10:19 +000013083 {"__format__", (PyCFunction) unicode__format__, METH_VARARGS, p_format__doc__},
Georg Brandlceee0772007-11-27 23:48:05 +000013084 {"maketrans", (PyCFunction) unicode_maketrans,
13085 METH_VARARGS | METH_STATIC, maketrans__doc__},
Georg Brandlc28e1fa2008-06-10 19:20:26 +000013086 {"__sizeof__", (PyCFunction) unicode__sizeof__, METH_NOARGS, sizeof__doc__},
Walter Dörwald068325e2002-04-15 13:36:47 +000013087#if 0
Alexander Belopolsky942af5a2010-12-04 03:38:46 +000013088 /* These methods are just used for debugging the implementation. */
Alexander Belopolsky942af5a2010-12-04 03:38:46 +000013089 {"_decimal2ascii", (PyCFunction) unicode__decimal2ascii, METH_NOARGS},
Guido van Rossumd57fd912000-03-10 22:53:23 +000013090#endif
13091
Benjamin Peterson14339b62009-01-31 16:36:08 +000013092 {"__getnewargs__", (PyCFunction)unicode_getnewargs, METH_NOARGS},
Guido van Rossumd57fd912000-03-10 22:53:23 +000013093 {NULL, NULL}
13094};
13095
Neil Schemenauerce30bc92002-11-18 16:10:18 +000013096static PyObject *
13097unicode_mod(PyObject *v, PyObject *w)
13098{
Brian Curtindfc80e32011-08-10 20:28:54 -050013099 if (!PyUnicode_Check(v))
13100 Py_RETURN_NOTIMPLEMENTED;
Benjamin Peterson29060642009-01-31 22:14:21 +000013101 return PyUnicode_Format(v, w);
Neil Schemenauerce30bc92002-11-18 16:10:18 +000013102}
13103
13104static PyNumberMethods unicode_as_number = {
Benjamin Peterson14339b62009-01-31 16:36:08 +000013105 0, /*nb_add*/
13106 0, /*nb_subtract*/
13107 0, /*nb_multiply*/
13108 unicode_mod, /*nb_remainder*/
Neil Schemenauerce30bc92002-11-18 16:10:18 +000013109};
13110
Guido van Rossumd57fd912000-03-10 22:53:23 +000013111static PySequenceMethods unicode_as_sequence = {
Benjamin Peterson14339b62009-01-31 16:36:08 +000013112 (lenfunc) unicode_length, /* sq_length */
13113 PyUnicode_Concat, /* sq_concat */
13114 (ssizeargfunc) unicode_repeat, /* sq_repeat */
13115 (ssizeargfunc) unicode_getitem, /* sq_item */
13116 0, /* sq_slice */
13117 0, /* sq_ass_item */
13118 0, /* sq_ass_slice */
13119 PyUnicode_Contains, /* sq_contains */
Guido van Rossumd57fd912000-03-10 22:53:23 +000013120};
13121
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +000013122static PyObject*
Victor Stinner9db1a8b2011-10-23 20:04:37 +020013123unicode_subscript(PyObject* self, PyObject* item)
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +000013124{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013125 if (PyUnicode_READY(self) == -1)
13126 return NULL;
13127
Thomas Wouters00ee7ba2006-08-21 19:07:27 +000013128 if (PyIndex_Check(item)) {
13129 Py_ssize_t i = PyNumber_AsSsize_t(item, PyExc_IndexError);
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +000013130 if (i == -1 && PyErr_Occurred())
13131 return NULL;
13132 if (i < 0)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013133 i += PyUnicode_GET_LENGTH(self);
Victor Stinner9db1a8b2011-10-23 20:04:37 +020013134 return unicode_getitem(self, i);
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +000013135 } else if (PySlice_Check(item)) {
Martin v. Löwis18e16552006-02-15 17:27:45 +000013136 Py_ssize_t start, stop, step, slicelength, cur, i;
Antoine Pitrou7aec4012011-10-04 19:08:01 +020013137 PyObject *result;
13138 void *src_data, *dest_data;
Antoine Pitrou875f29b2011-10-04 20:00:49 +020013139 int src_kind, dest_kind;
Victor Stinnerc80d6d22011-10-05 14:13:28 +020013140 Py_UCS4 ch, max_char, kind_limit;
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +000013141
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013142 if (PySlice_GetIndicesEx(item, PyUnicode_GET_LENGTH(self),
Benjamin Peterson29060642009-01-31 22:14:21 +000013143 &start, &stop, &step, &slicelength) < 0) {
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +000013144 return NULL;
13145 }
13146
13147 if (slicelength <= 0) {
Victor Stinner382955f2011-12-11 21:44:00 +010013148 Py_INCREF(unicode_empty);
13149 return unicode_empty;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013150 } else if (start == 0 && step == 1 &&
Victor Stinnerc4b49542011-12-11 22:44:26 +010013151 slicelength == PyUnicode_GET_LENGTH(self)) {
13152 return unicode_result_unchanged(self);
Thomas Woutersed03b412007-08-28 21:37:11 +000013153 } else if (step == 1) {
Victor Stinner9db1a8b2011-10-23 20:04:37 +020013154 return PyUnicode_Substring(self,
Victor Stinner12bab6d2011-10-01 01:53:49 +020013155 start, start + slicelength);
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +000013156 }
Antoine Pitrou875f29b2011-10-04 20:00:49 +020013157 /* General case */
Antoine Pitrou875f29b2011-10-04 20:00:49 +020013158 src_kind = PyUnicode_KIND(self);
13159 src_data = PyUnicode_DATA(self);
Victor Stinner55c99112011-10-13 01:17:06 +020013160 if (!PyUnicode_IS_ASCII(self)) {
13161 kind_limit = kind_maxchar_limit(src_kind);
13162 max_char = 0;
13163 for (cur = start, i = 0; i < slicelength; cur += step, i++) {
13164 ch = PyUnicode_READ(src_kind, src_data, cur);
13165 if (ch > max_char) {
13166 max_char = ch;
13167 if (max_char >= kind_limit)
13168 break;
13169 }
Victor Stinnerc80d6d22011-10-05 14:13:28 +020013170 }
Antoine Pitrou875f29b2011-10-04 20:00:49 +020013171 }
Victor Stinner55c99112011-10-13 01:17:06 +020013172 else
13173 max_char = 127;
Antoine Pitrou875f29b2011-10-04 20:00:49 +020013174 result = PyUnicode_New(slicelength, max_char);
Antoine Pitrou7aec4012011-10-04 19:08:01 +020013175 if (result == NULL)
13176 return NULL;
Antoine Pitrou875f29b2011-10-04 20:00:49 +020013177 dest_kind = PyUnicode_KIND(result);
Antoine Pitrou7aec4012011-10-04 19:08:01 +020013178 dest_data = PyUnicode_DATA(result);
13179
13180 for (cur = start, i = 0; i < slicelength; cur += step, i++) {
Antoine Pitrou875f29b2011-10-04 20:00:49 +020013181 Py_UCS4 ch = PyUnicode_READ(src_kind, src_data, cur);
13182 PyUnicode_WRITE(dest_kind, dest_data, i, ch);
Antoine Pitrou7aec4012011-10-04 19:08:01 +020013183 }
Victor Stinnerbb10a1f2011-10-05 01:34:17 +020013184 assert(_PyUnicode_CheckConsistency(result, 1));
Antoine Pitrou7aec4012011-10-04 19:08:01 +020013185 return result;
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +000013186 } else {
13187 PyErr_SetString(PyExc_TypeError, "string indices must be integers");
13188 return NULL;
13189 }
13190}
13191
13192static PyMappingMethods unicode_as_mapping = {
Benjamin Peterson14339b62009-01-31 16:36:08 +000013193 (lenfunc)unicode_length, /* mp_length */
13194 (binaryfunc)unicode_subscript, /* mp_subscript */
13195 (objobjargproc)0, /* mp_ass_subscript */
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +000013196};
13197
Guido van Rossumd57fd912000-03-10 22:53:23 +000013198
Guido van Rossumd57fd912000-03-10 22:53:23 +000013199/* Helpers for PyUnicode_Format() */
13200
13201static PyObject *
Martin v. Löwis18e16552006-02-15 17:27:45 +000013202getnextarg(PyObject *args, Py_ssize_t arglen, Py_ssize_t *p_argidx)
Guido van Rossumd57fd912000-03-10 22:53:23 +000013203{
Martin v. Löwis18e16552006-02-15 17:27:45 +000013204 Py_ssize_t argidx = *p_argidx;
Guido van Rossumd57fd912000-03-10 22:53:23 +000013205 if (argidx < arglen) {
Benjamin Peterson29060642009-01-31 22:14:21 +000013206 (*p_argidx)++;
13207 if (arglen < 0)
13208 return args;
13209 else
13210 return PyTuple_GetItem(args, argidx);
Guido van Rossumd57fd912000-03-10 22:53:23 +000013211 }
13212 PyErr_SetString(PyExc_TypeError,
Benjamin Peterson29060642009-01-31 22:14:21 +000013213 "not enough arguments for format string");
Guido van Rossumd57fd912000-03-10 22:53:23 +000013214 return NULL;
13215}
13216
Mark Dickinsonf489caf2009-05-01 11:42:00 +000013217/* Returns a new reference to a PyUnicode object, or NULL on failure. */
Guido van Rossumd57fd912000-03-10 22:53:23 +000013218
Victor Stinnerd3f08822012-05-29 12:57:52 +020013219static int
13220formatfloat(PyObject *v, int flags, int prec, int type,
13221 PyObject **p_output, _PyUnicodeWriter *writer)
Guido van Rossumd57fd912000-03-10 22:53:23 +000013222{
Mark Dickinsonf489caf2009-05-01 11:42:00 +000013223 char *p;
Guido van Rossumd57fd912000-03-10 22:53:23 +000013224 double x;
Victor Stinnerd3f08822012-05-29 12:57:52 +020013225 Py_ssize_t len;
Tim Petersced69f82003-09-16 20:30:58 +000013226
Guido van Rossumd57fd912000-03-10 22:53:23 +000013227 x = PyFloat_AsDouble(v);
13228 if (x == -1.0 && PyErr_Occurred())
Victor Stinnerd3f08822012-05-29 12:57:52 +020013229 return -1;
Mark Dickinsonf489caf2009-05-01 11:42:00 +000013230
Guido van Rossumd57fd912000-03-10 22:53:23 +000013231 if (prec < 0)
Benjamin Peterson29060642009-01-31 22:14:21 +000013232 prec = 6;
Eric Smith0923d1d2009-04-16 20:16:10 +000013233
Eric Smith0923d1d2009-04-16 20:16:10 +000013234 p = PyOS_double_to_string(x, type, prec,
13235 (flags & F_ALT) ? Py_DTSF_ALT : 0, NULL);
Mark Dickinsonf489caf2009-05-01 11:42:00 +000013236 if (p == NULL)
Victor Stinnerd3f08822012-05-29 12:57:52 +020013237 return -1;
13238 len = strlen(p);
13239 if (writer) {
Christian Heimesf4f99392012-09-10 11:48:41 +020013240 if (_PyUnicodeWriter_Prepare(writer, len, 127) == -1) {
13241 PyMem_Free(p);
Victor Stinnerd3f08822012-05-29 12:57:52 +020013242 return -1;
Christian Heimesf4f99392012-09-10 11:48:41 +020013243 }
Victor Stinner184252a2012-06-16 02:57:41 +020013244 unicode_write_cstr(writer->buffer, writer->pos, p, len);
Victor Stinnerd3f08822012-05-29 12:57:52 +020013245 writer->pos += len;
13246 }
13247 else
13248 *p_output = _PyUnicode_FromASCII(p, len);
Eric Smith0923d1d2009-04-16 20:16:10 +000013249 PyMem_Free(p);
Victor Stinnerd3f08822012-05-29 12:57:52 +020013250 return 0;
Guido van Rossumd57fd912000-03-10 22:53:23 +000013251}
13252
Victor Stinnerd0880d52012-04-27 23:40:13 +020013253/* formatlong() emulates the format codes d, u, o, x and X, and
13254 * the F_ALT flag, for Python's long (unbounded) ints. It's not used for
13255 * Python's regular ints.
13256 * Return value: a new PyUnicodeObject*, or NULL if error.
13257 * The output string is of the form
13258 * "-"? ("0x" | "0X")? digit+
13259 * "0x"/"0X" are present only for x and X conversions, with F_ALT
13260 * set in flags. The case of hex digits will be correct,
13261 * There will be at least prec digits, zero-filled on the left if
13262 * necessary to get that many.
13263 * val object to be converted
13264 * flags bitmask of format flags; only F_ALT is looked at
13265 * prec minimum number of digits; 0-fill on left if needed
13266 * type a character in [duoxX]; u acts the same as d
13267 *
13268 * CAUTION: o, x and X conversions on regular ints can never
13269 * produce a '-' sign, but can for Python's unbounded ints.
13270 */
Tim Peters38fd5b62000-09-21 05:43:11 +000013271static PyObject*
13272formatlong(PyObject *val, int flags, int prec, int type)
13273{
Victor Stinnerd0880d52012-04-27 23:40:13 +020013274 PyObject *result = NULL;
Benjamin Peterson14339b62009-01-31 16:36:08 +000013275 char *buf;
Victor Stinnerd0880d52012-04-27 23:40:13 +020013276 Py_ssize_t i;
13277 int sign; /* 1 if '-', else 0 */
13278 int len; /* number of characters */
13279 Py_ssize_t llen;
13280 int numdigits; /* len == numnondigits + numdigits */
13281 int numnondigits = 0;
Tim Peters38fd5b62000-09-21 05:43:11 +000013282
Victor Stinnerd0880d52012-04-27 23:40:13 +020013283 /* Avoid exceeding SSIZE_T_MAX */
13284 if (prec > INT_MAX-3) {
13285 PyErr_SetString(PyExc_OverflowError,
13286 "precision too large");
Benjamin Peterson14339b62009-01-31 16:36:08 +000013287 return NULL;
Victor Stinnerd0880d52012-04-27 23:40:13 +020013288 }
13289
13290 assert(PyLong_Check(val));
13291
13292 switch (type) {
13293 case 'd':
13294 case 'u':
13295 /* Special-case boolean: we want 0/1 */
Victor Stinnerb11d91d2012-04-28 00:25:34 +020013296 if (PyBool_Check(val))
13297 result = PyNumber_ToBase(val, 10);
13298 else
13299 result = Py_TYPE(val)->tp_str(val);
Victor Stinnerd0880d52012-04-27 23:40:13 +020013300 break;
13301 case 'o':
13302 numnondigits = 2;
13303 result = PyNumber_ToBase(val, 8);
13304 break;
13305 case 'x':
13306 case 'X':
13307 numnondigits = 2;
13308 result = PyNumber_ToBase(val, 16);
13309 break;
13310 default:
13311 assert(!"'type' not in [duoxX]");
13312 }
13313 if (!result)
13314 return NULL;
13315
13316 assert(unicode_modifiable(result));
13317 assert(PyUnicode_IS_READY(result));
13318 assert(PyUnicode_IS_ASCII(result));
13319
13320 /* To modify the string in-place, there can only be one reference. */
13321 if (Py_REFCNT(result) != 1) {
13322 PyErr_BadInternalCall();
13323 return NULL;
13324 }
13325 buf = PyUnicode_DATA(result);
13326 llen = PyUnicode_GET_LENGTH(result);
13327 if (llen > INT_MAX) {
13328 PyErr_SetString(PyExc_ValueError,
13329 "string too large in _PyBytes_FormatLong");
13330 return NULL;
13331 }
13332 len = (int)llen;
13333 sign = buf[0] == '-';
13334 numnondigits += sign;
13335 numdigits = len - numnondigits;
13336 assert(numdigits > 0);
13337
13338 /* Get rid of base marker unless F_ALT */
13339 if (((flags & F_ALT) == 0 &&
13340 (type == 'o' || type == 'x' || type == 'X'))) {
13341 assert(buf[sign] == '0');
13342 assert(buf[sign+1] == 'x' || buf[sign+1] == 'X' ||
13343 buf[sign+1] == 'o');
13344 numnondigits -= 2;
13345 buf += 2;
13346 len -= 2;
13347 if (sign)
13348 buf[0] = '-';
13349 assert(len == numnondigits + numdigits);
13350 assert(numdigits > 0);
13351 }
13352
13353 /* Fill with leading zeroes to meet minimum width. */
13354 if (prec > numdigits) {
13355 PyObject *r1 = PyBytes_FromStringAndSize(NULL,
13356 numnondigits + prec);
13357 char *b1;
13358 if (!r1) {
13359 Py_DECREF(result);
13360 return NULL;
13361 }
13362 b1 = PyBytes_AS_STRING(r1);
13363 for (i = 0; i < numnondigits; ++i)
13364 *b1++ = *buf++;
13365 for (i = 0; i < prec - numdigits; i++)
13366 *b1++ = '0';
13367 for (i = 0; i < numdigits; i++)
13368 *b1++ = *buf++;
13369 *b1 = '\0';
13370 Py_DECREF(result);
13371 result = r1;
13372 buf = PyBytes_AS_STRING(result);
13373 len = numnondigits + prec;
13374 }
13375
13376 /* Fix up case for hex conversions. */
13377 if (type == 'X') {
13378 /* Need to convert all lower case letters to upper case.
13379 and need to convert 0x to 0X (and -0x to -0X). */
13380 for (i = 0; i < len; i++)
13381 if (buf[i] >= 'a' && buf[i] <= 'x')
13382 buf[i] -= 'a'-'A';
13383 }
13384 if (!PyUnicode_Check(result) || len != PyUnicode_GET_LENGTH(result)) {
13385 PyObject *unicode;
Victor Stinnerd3f08822012-05-29 12:57:52 +020013386 unicode = _PyUnicode_FromASCII(buf, len);
Victor Stinnerd0880d52012-04-27 23:40:13 +020013387 Py_DECREF(result);
13388 result = unicode;
13389 }
Benjamin Peterson14339b62009-01-31 16:36:08 +000013390 return result;
Tim Peters38fd5b62000-09-21 05:43:11 +000013391}
13392
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013393static Py_UCS4
13394formatchar(PyObject *v)
Guido van Rossumd57fd912000-03-10 22:53:23 +000013395{
Amaury Forgeot d'Arca4db6862008-07-04 21:26:43 +000013396 /* presume that the buffer is at least 3 characters long */
Marc-André Lemburgd4ab4a52000-06-08 17:54:00 +000013397 if (PyUnicode_Check(v)) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013398 if (PyUnicode_GET_LENGTH(v) == 1) {
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013399 return PyUnicode_READ_CHAR(v, 0);
Benjamin Peterson29060642009-01-31 22:14:21 +000013400 }
Benjamin Peterson29060642009-01-31 22:14:21 +000013401 goto onError;
13402 }
13403 else {
13404 /* Integer input truncated to a character */
13405 long x;
13406 x = PyLong_AsLong(v);
13407 if (x == -1 && PyErr_Occurred())
13408 goto onError;
13409
Victor Stinner8faf8212011-12-08 22:14:11 +010013410 if (x < 0 || x > MAX_UNICODE) {
Benjamin Peterson29060642009-01-31 22:14:21 +000013411 PyErr_SetString(PyExc_OverflowError,
13412 "%c arg not in range(0x110000)");
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013413 return (Py_UCS4) -1;
Benjamin Peterson29060642009-01-31 22:14:21 +000013414 }
13415
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013416 return (Py_UCS4) x;
Benjamin Peterson14339b62009-01-31 16:36:08 +000013417 }
Amaury Forgeot d'Arca4db6862008-07-04 21:26:43 +000013418
Benjamin Peterson29060642009-01-31 22:14:21 +000013419 onError:
Marc-André Lemburgd4ab4a52000-06-08 17:54:00 +000013420 PyErr_SetString(PyExc_TypeError,
Benjamin Peterson29060642009-01-31 22:14:21 +000013421 "%c requires int or char");
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013422 return (Py_UCS4) -1;
Guido van Rossumd57fd912000-03-10 22:53:23 +000013423}
13424
Alexander Belopolsky40018472011-02-26 01:02:56 +000013425PyObject *
13426PyUnicode_Format(PyObject *format, PyObject *args)
Guido van Rossumd57fd912000-03-10 22:53:23 +000013427{
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013428 Py_ssize_t fmtcnt, fmtpos, arglen, argidx;
Guido van Rossumd57fd912000-03-10 22:53:23 +000013429 int args_owned = 0;
Guido van Rossumd57fd912000-03-10 22:53:23 +000013430 PyObject *dict = NULL;
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013431 PyObject *temp = NULL;
13432 PyObject *second = NULL;
Victor Stinner9db1a8b2011-10-23 20:04:37 +020013433 PyObject *uformat;
Victor Stinnerf2c76aa2012-05-03 13:10:40 +020013434 void *fmt;
13435 enum PyUnicode_Kind kind, fmtkind;
Victor Stinner3b1a74a2012-05-09 22:25:00 +020013436 _PyUnicodeWriter writer;
Victor Stinneree4544c2012-05-09 22:24:08 +020013437 Py_ssize_t sublen;
13438 Py_UCS4 maxchar;
Tim Petersced69f82003-09-16 20:30:58 +000013439
Guido van Rossumd57fd912000-03-10 22:53:23 +000013440 if (format == NULL || args == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +000013441 PyErr_BadInternalCall();
13442 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +000013443 }
Victor Stinner9db1a8b2011-10-23 20:04:37 +020013444 uformat = PyUnicode_FromObject(format);
Benjamin Peterson22a29702012-01-02 09:00:30 -060013445 if (uformat == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +000013446 return NULL;
Benjamin Peterson22a29702012-01-02 09:00:30 -060013447 if (PyUnicode_READY(uformat) == -1)
13448 Py_DECREF(uformat);
Victor Stinnerf2c76aa2012-05-03 13:10:40 +020013449
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013450 fmt = PyUnicode_DATA(uformat);
13451 fmtkind = PyUnicode_KIND(uformat);
13452 fmtcnt = PyUnicode_GET_LENGTH(uformat);
13453 fmtpos = 0;
Guido van Rossumd57fd912000-03-10 22:53:23 +000013454
Victor Stinnerd3f08822012-05-29 12:57:52 +020013455 _PyUnicodeWriter_Init(&writer, fmtcnt + 100);
Victor Stinnerf2c76aa2012-05-03 13:10:40 +020013456
Guido van Rossumd57fd912000-03-10 22:53:23 +000013457 if (PyTuple_Check(args)) {
Benjamin Peterson29060642009-01-31 22:14:21 +000013458 arglen = PyTuple_Size(args);
13459 argidx = 0;
Guido van Rossumd57fd912000-03-10 22:53:23 +000013460 }
13461 else {
Benjamin Peterson29060642009-01-31 22:14:21 +000013462 arglen = -1;
13463 argidx = -2;
Guido van Rossumd57fd912000-03-10 22:53:23 +000013464 }
Benjamin Peterson28a6cfa2012-08-28 17:55:35 -040013465 if (PyMapping_Check(args) && !PyTuple_Check(args) && !PyUnicode_Check(args))
Benjamin Peterson29060642009-01-31 22:14:21 +000013466 dict = args;
Guido van Rossumd57fd912000-03-10 22:53:23 +000013467
13468 while (--fmtcnt >= 0) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013469 if (PyUnicode_READ(fmtkind, fmt, fmtpos) != '%') {
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013470 Py_ssize_t nonfmtpos;
13471 nonfmtpos = fmtpos++;
13472 while (fmtcnt >= 0 &&
13473 PyUnicode_READ(fmtkind, fmt, fmtpos) != '%') {
13474 fmtpos++;
13475 fmtcnt--;
Benjamin Peterson14339b62009-01-31 16:36:08 +000013476 }
Victor Stinnerf2c76aa2012-05-03 13:10:40 +020013477 if (fmtcnt < 0)
13478 fmtpos--;
Victor Stinneree4544c2012-05-09 22:24:08 +020013479 sublen = fmtpos - nonfmtpos;
13480 maxchar = _PyUnicode_FindMaxChar(uformat,
13481 nonfmtpos, nonfmtpos + sublen);
Victor Stinner3b1a74a2012-05-09 22:25:00 +020013482 if (_PyUnicodeWriter_Prepare(&writer, sublen, maxchar) == -1)
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013483 goto onError;
Victor Stinneree4544c2012-05-09 22:24:08 +020013484
Victor Stinnerd3f08822012-05-29 12:57:52 +020013485 _PyUnicode_FastCopyCharacters(writer.buffer, writer.pos,
13486 uformat, nonfmtpos, sublen);
Victor Stinneree4544c2012-05-09 22:24:08 +020013487 writer.pos += sublen;
Benjamin Peterson14339b62009-01-31 16:36:08 +000013488 }
13489 else {
Benjamin Peterson29060642009-01-31 22:14:21 +000013490 /* Got a format specifier */
13491 int flags = 0;
13492 Py_ssize_t width = -1;
13493 int prec = -1;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013494 Py_UCS4 c = '\0';
Victor Stinnerf2c76aa2012-05-03 13:10:40 +020013495 Py_UCS4 fill;
13496 int sign;
13497 Py_UCS4 signchar;
Benjamin Peterson29060642009-01-31 22:14:21 +000013498 int isnumok;
13499 PyObject *v = NULL;
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013500 void *pbuf = NULL;
13501 Py_ssize_t pindex, len;
Victor Stinneree4544c2012-05-09 22:24:08 +020013502 Py_UCS4 bufmaxchar;
13503 Py_ssize_t buflen;
Guido van Rossumd57fd912000-03-10 22:53:23 +000013504
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013505 fmtpos++;
Victor Stinner438106b2012-05-02 00:41:57 +020013506 c = PyUnicode_READ(fmtkind, fmt, fmtpos);
13507 if (c == '(') {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013508 Py_ssize_t keystart;
Benjamin Peterson29060642009-01-31 22:14:21 +000013509 Py_ssize_t keylen;
13510 PyObject *key;
13511 int pcount = 1;
Christian Heimesa612dc02008-02-24 13:08:18 +000013512
Benjamin Peterson29060642009-01-31 22:14:21 +000013513 if (dict == NULL) {
13514 PyErr_SetString(PyExc_TypeError,
13515 "format requires a mapping");
13516 goto onError;
13517 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013518 ++fmtpos;
Benjamin Peterson29060642009-01-31 22:14:21 +000013519 --fmtcnt;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013520 keystart = fmtpos;
Benjamin Peterson29060642009-01-31 22:14:21 +000013521 /* Skip over balanced parentheses */
13522 while (pcount > 0 && --fmtcnt >= 0) {
Victor Stinnerbff7c962012-05-03 01:44:59 +020013523 c = PyUnicode_READ(fmtkind, fmt, fmtpos);
13524 if (c == ')')
Benjamin Peterson29060642009-01-31 22:14:21 +000013525 --pcount;
Victor Stinnerbff7c962012-05-03 01:44:59 +020013526 else if (c == '(')
Benjamin Peterson29060642009-01-31 22:14:21 +000013527 ++pcount;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013528 fmtpos++;
Benjamin Peterson29060642009-01-31 22:14:21 +000013529 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013530 keylen = fmtpos - keystart - 1;
Benjamin Peterson29060642009-01-31 22:14:21 +000013531 if (fmtcnt < 0 || pcount > 0) {
13532 PyErr_SetString(PyExc_ValueError,
13533 "incomplete format key");
13534 goto onError;
13535 }
Victor Stinner7931d9a2011-11-04 00:22:48 +010013536 key = PyUnicode_Substring(uformat,
Victor Stinner12bab6d2011-10-01 01:53:49 +020013537 keystart, keystart + keylen);
Benjamin Peterson29060642009-01-31 22:14:21 +000013538 if (key == NULL)
13539 goto onError;
13540 if (args_owned) {
13541 Py_DECREF(args);
13542 args_owned = 0;
13543 }
13544 args = PyObject_GetItem(dict, key);
13545 Py_DECREF(key);
13546 if (args == NULL) {
13547 goto onError;
13548 }
13549 args_owned = 1;
13550 arglen = -1;
13551 argidx = -2;
Benjamin Peterson14339b62009-01-31 16:36:08 +000013552 }
Benjamin Peterson29060642009-01-31 22:14:21 +000013553 while (--fmtcnt >= 0) {
Victor Stinner438106b2012-05-02 00:41:57 +020013554 c = PyUnicode_READ(fmtkind, fmt, fmtpos++);
13555 switch (c) {
Benjamin Peterson29060642009-01-31 22:14:21 +000013556 case '-': flags |= F_LJUST; continue;
13557 case '+': flags |= F_SIGN; continue;
13558 case ' ': flags |= F_BLANK; continue;
13559 case '#': flags |= F_ALT; continue;
13560 case '0': flags |= F_ZERO; continue;
13561 }
13562 break;
Benjamin Peterson14339b62009-01-31 16:36:08 +000013563 }
Benjamin Peterson29060642009-01-31 22:14:21 +000013564 if (c == '*') {
13565 v = getnextarg(args, arglen, &argidx);
13566 if (v == NULL)
13567 goto onError;
13568 if (!PyLong_Check(v)) {
13569 PyErr_SetString(PyExc_TypeError,
13570 "* wants int");
13571 goto onError;
13572 }
13573 width = PyLong_AsLong(v);
13574 if (width == -1 && PyErr_Occurred())
13575 goto onError;
13576 if (width < 0) {
13577 flags |= F_LJUST;
13578 width = -width;
13579 }
13580 if (--fmtcnt >= 0)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013581 c = PyUnicode_READ(fmtkind, fmt, fmtpos++);
Benjamin Peterson29060642009-01-31 22:14:21 +000013582 }
13583 else if (c >= '0' && c <= '9') {
13584 width = c - '0';
13585 while (--fmtcnt >= 0) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013586 c = PyUnicode_READ(fmtkind, fmt, fmtpos++);
Benjamin Peterson29060642009-01-31 22:14:21 +000013587 if (c < '0' || c > '9')
13588 break;
Martin v. Löwisb05c0732012-05-15 13:45:49 +020013589 /* Since c is unsigned, the RHS would end up as unsigned,
13590 mixing signed and unsigned comparison. Since c is between
13591 '0' and '9', casting to int is safe. */
13592 if (width > (PY_SSIZE_T_MAX - ((int)c - '0')) / 10) {
Benjamin Peterson29060642009-01-31 22:14:21 +000013593 PyErr_SetString(PyExc_ValueError,
13594 "width too big");
Benjamin Peterson14339b62009-01-31 16:36:08 +000013595 goto onError;
Benjamin Peterson29060642009-01-31 22:14:21 +000013596 }
13597 width = width*10 + (c - '0');
13598 }
13599 }
13600 if (c == '.') {
13601 prec = 0;
13602 if (--fmtcnt >= 0)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013603 c = PyUnicode_READ(fmtkind, fmt, fmtpos++);
Benjamin Peterson29060642009-01-31 22:14:21 +000013604 if (c == '*') {
13605 v = getnextarg(args, arglen, &argidx);
13606 if (v == NULL)
13607 goto onError;
13608 if (!PyLong_Check(v)) {
13609 PyErr_SetString(PyExc_TypeError,
13610 "* wants int");
13611 goto onError;
13612 }
13613 prec = PyLong_AsLong(v);
13614 if (prec == -1 && PyErr_Occurred())
13615 goto onError;
13616 if (prec < 0)
13617 prec = 0;
13618 if (--fmtcnt >= 0)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013619 c = PyUnicode_READ(fmtkind, fmt, fmtpos++);
Benjamin Peterson29060642009-01-31 22:14:21 +000013620 }
13621 else if (c >= '0' && c <= '9') {
13622 prec = c - '0';
13623 while (--fmtcnt >= 0) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013624 c = PyUnicode_READ(fmtkind, fmt, fmtpos++);
Benjamin Peterson29060642009-01-31 22:14:21 +000013625 if (c < '0' || c > '9')
13626 break;
Martin v. Löwisb05c0732012-05-15 13:45:49 +020013627 if (prec > (INT_MAX - ((int)c - '0')) / 10) {
Benjamin Peterson29060642009-01-31 22:14:21 +000013628 PyErr_SetString(PyExc_ValueError,
13629 "prec too big");
13630 goto onError;
13631 }
13632 prec = prec*10 + (c - '0');
13633 }
13634 }
13635 } /* prec */
13636 if (fmtcnt >= 0) {
13637 if (c == 'h' || c == 'l' || c == 'L') {
13638 if (--fmtcnt >= 0)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013639 c = PyUnicode_READ(fmtkind, fmt, fmtpos++);
Benjamin Peterson29060642009-01-31 22:14:21 +000013640 }
13641 }
13642 if (fmtcnt < 0) {
13643 PyErr_SetString(PyExc_ValueError,
13644 "incomplete format");
13645 goto onError;
13646 }
Victor Stinnerd3f08822012-05-29 12:57:52 +020013647 if (fmtcnt == 0)
Victor Stinnerd7b7c742012-06-04 22:52:12 +020013648 writer.overallocate = 0;
Victor Stinneraff3cc62012-04-30 05:19:21 +020013649
13650 if (c == '%') {
Victor Stinner3b1a74a2012-05-09 22:25:00 +020013651 if (_PyUnicodeWriter_Prepare(&writer, 1, '%') == -1)
Victor Stinnerf2c76aa2012-05-03 13:10:40 +020013652 goto onError;
Victor Stinneree4544c2012-05-09 22:24:08 +020013653 PyUnicode_WRITE(writer.kind, writer.data, writer.pos, '%');
13654 writer.pos += 1;
Victor Stinneraff3cc62012-04-30 05:19:21 +020013655 continue;
Benjamin Peterson29060642009-01-31 22:14:21 +000013656 }
Victor Stinneraff3cc62012-04-30 05:19:21 +020013657
Victor Stinneraff3cc62012-04-30 05:19:21 +020013658 v = getnextarg(args, arglen, &argidx);
13659 if (v == NULL)
13660 goto onError;
13661
Benjamin Peterson29060642009-01-31 22:14:21 +000013662 sign = 0;
Victor Stinnerf2c76aa2012-05-03 13:10:40 +020013663 signchar = '\0';
Benjamin Peterson29060642009-01-31 22:14:21 +000013664 fill = ' ';
13665 switch (c) {
13666
Benjamin Peterson29060642009-01-31 22:14:21 +000013667 case 's':
13668 case 'r':
13669 case 'a':
Victor Stinnerd3f08822012-05-29 12:57:52 +020013670 if (PyLong_CheckExact(v) && width == -1 && prec == -1) {
13671 /* Fast path */
13672 if (_PyLong_FormatWriter(&writer, v, 10, flags & F_ALT) == -1)
13673 goto onError;
13674 goto nextarg;
13675 }
13676
Victor Stinner808fc0a2010-03-22 12:50:40 +000013677 if (PyUnicode_CheckExact(v) && c == 's') {
Benjamin Peterson29060642009-01-31 22:14:21 +000013678 temp = v;
13679 Py_INCREF(temp);
Benjamin Peterson14339b62009-01-31 16:36:08 +000013680 }
13681 else {
Benjamin Peterson29060642009-01-31 22:14:21 +000013682 if (c == 's')
13683 temp = PyObject_Str(v);
13684 else if (c == 'r')
13685 temp = PyObject_Repr(v);
13686 else
13687 temp = PyObject_ASCII(v);
Benjamin Peterson29060642009-01-31 22:14:21 +000013688 }
Benjamin Peterson29060642009-01-31 22:14:21 +000013689 break;
13690
13691 case 'i':
13692 case 'd':
13693 case 'u':
13694 case 'o':
13695 case 'x':
13696 case 'X':
Victor Stinnerd3f08822012-05-29 12:57:52 +020013697 if (PyLong_CheckExact(v)
13698 && width == -1 && prec == -1
13699 && !(flags & (F_SIGN | F_BLANK)))
13700 {
13701 /* Fast path */
13702 switch(c)
13703 {
13704 case 'd':
13705 case 'i':
13706 case 'u':
13707 if (_PyLong_FormatWriter(&writer, v, 10, flags & F_ALT) == -1)
13708 goto onError;
13709 goto nextarg;
13710 case 'x':
13711 if (_PyLong_FormatWriter(&writer, v, 16, flags & F_ALT) == -1)
13712 goto onError;
13713 goto nextarg;
13714 case 'o':
13715 if (_PyLong_FormatWriter(&writer, v, 8, flags & F_ALT) == -1)
13716 goto onError;
13717 goto nextarg;
13718 default:
13719 break;
13720 }
13721 }
13722
Benjamin Peterson29060642009-01-31 22:14:21 +000013723 isnumok = 0;
13724 if (PyNumber_Check(v)) {
13725 PyObject *iobj=NULL;
13726
13727 if (PyLong_Check(v)) {
13728 iobj = v;
13729 Py_INCREF(iobj);
13730 }
13731 else {
13732 iobj = PyNumber_Long(v);
13733 }
13734 if (iobj!=NULL) {
13735 if (PyLong_Check(iobj)) {
13736 isnumok = 1;
Victor Stinneraff3cc62012-04-30 05:19:21 +020013737 sign = 1;
Senthil Kumaran9ebe08d2011-07-03 21:03:16 -070013738 temp = formatlong(iobj, flags, prec, (c == 'i'? 'd': c));
Benjamin Peterson29060642009-01-31 22:14:21 +000013739 Py_DECREF(iobj);
Benjamin Peterson29060642009-01-31 22:14:21 +000013740 }
13741 else {
13742 Py_DECREF(iobj);
13743 }
13744 }
13745 }
13746 if (!isnumok) {
13747 PyErr_Format(PyExc_TypeError,
13748 "%%%c format: a number is required, "
13749 "not %.200s", (char)c, Py_TYPE(v)->tp_name);
13750 goto onError;
13751 }
Victor Stinnerf2c76aa2012-05-03 13:10:40 +020013752 if (flags & F_ZERO)
Benjamin Peterson29060642009-01-31 22:14:21 +000013753 fill = '0';
13754 break;
13755
13756 case 'e':
13757 case 'E':
13758 case 'f':
13759 case 'F':
13760 case 'g':
13761 case 'G':
Victor Stinnerd3f08822012-05-29 12:57:52 +020013762 if (width == -1 && prec == -1
13763 && !(flags & (F_SIGN | F_BLANK)))
13764 {
13765 /* Fast path */
13766 if (formatfloat(v, flags, prec, c, NULL, &writer) == -1)
13767 goto onError;
13768 goto nextarg;
13769 }
13770
Benjamin Peterson29060642009-01-31 22:14:21 +000013771 sign = 1;
Victor Stinnerf2c76aa2012-05-03 13:10:40 +020013772 if (flags & F_ZERO)
Benjamin Peterson29060642009-01-31 22:14:21 +000013773 fill = '0';
Victor Stinnerd3f08822012-05-29 12:57:52 +020013774 if (formatfloat(v, flags, prec, c, &temp, NULL) == -1)
13775 temp = NULL;
Benjamin Peterson29060642009-01-31 22:14:21 +000013776 break;
13777
13778 case 'c':
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013779 {
13780 Py_UCS4 ch = formatchar(v);
13781 if (ch == (Py_UCS4) -1)
Benjamin Peterson29060642009-01-31 22:14:21 +000013782 goto onError;
Victor Stinnerd3f08822012-05-29 12:57:52 +020013783 if (width == -1 && prec == -1) {
13784 /* Fast path */
13785 if (_PyUnicodeWriter_Prepare(&writer, 1, ch) == -1)
13786 goto onError;
13787 PyUnicode_WRITE(writer.kind, writer.data, writer.pos, ch);
13788 writer.pos += 1;
13789 goto nextarg;
13790 }
Victor Stinnerb5c3ea32012-05-02 00:29:36 +020013791 temp = PyUnicode_FromOrdinal(ch);
Benjamin Peterson29060642009-01-31 22:14:21 +000013792 break;
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013793 }
Benjamin Peterson29060642009-01-31 22:14:21 +000013794
13795 default:
13796 PyErr_Format(PyExc_ValueError,
13797 "unsupported format character '%c' (0x%x) "
13798 "at index %zd",
13799 (31<=c && c<=126) ? (char)c : '?',
13800 (int)c,
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013801 fmtpos - 1);
Benjamin Peterson29060642009-01-31 22:14:21 +000013802 goto onError;
13803 }
Victor Stinneraff3cc62012-04-30 05:19:21 +020013804 if (temp == NULL)
13805 goto onError;
13806 assert (PyUnicode_Check(temp));
Victor Stinnerd3f08822012-05-29 12:57:52 +020013807
13808 if (width == -1 && prec == -1
13809 && !(flags & (F_SIGN | F_BLANK)))
13810 {
13811 /* Fast path */
13812 if (_PyUnicodeWriter_WriteStr(&writer, temp) == -1)
13813 goto onError;
13814 goto nextarg;
13815 }
13816
Victor Stinneraff3cc62012-04-30 05:19:21 +020013817 if (PyUnicode_READY(temp) == -1) {
13818 Py_CLEAR(temp);
13819 goto onError;
13820 }
13821 kind = PyUnicode_KIND(temp);
13822 pbuf = PyUnicode_DATA(temp);
13823 len = PyUnicode_GET_LENGTH(temp);
13824
13825 if (c == 's' || c == 'r' || c == 'a') {
13826 if (prec >= 0 && len > prec)
13827 len = prec;
13828 }
13829
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013830 /* pbuf is initialized here. */
13831 pindex = 0;
Benjamin Peterson29060642009-01-31 22:14:21 +000013832 if (sign) {
Victor Stinnerf2c76aa2012-05-03 13:10:40 +020013833 Py_UCS4 ch = PyUnicode_READ(kind, pbuf, pindex);
13834 if (ch == '-' || ch == '+') {
13835 signchar = ch;
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013836 len--;
13837 pindex++;
Benjamin Peterson29060642009-01-31 22:14:21 +000013838 }
13839 else if (flags & F_SIGN)
Victor Stinnerf2c76aa2012-05-03 13:10:40 +020013840 signchar = '+';
Benjamin Peterson29060642009-01-31 22:14:21 +000013841 else if (flags & F_BLANK)
Victor Stinnerf2c76aa2012-05-03 13:10:40 +020013842 signchar = ' ';
Benjamin Peterson29060642009-01-31 22:14:21 +000013843 else
13844 sign = 0;
13845 }
13846 if (width < len)
13847 width = len;
Victor Stinneree4544c2012-05-09 22:24:08 +020013848
13849 /* Compute the length and maximum character of the
13850 written characters */
13851 bufmaxchar = 127;
13852 if (!(flags & F_LJUST)) {
13853 if (sign) {
13854 if ((width-1) > len)
Victor Stinnerd3f08822012-05-29 12:57:52 +020013855 bufmaxchar = MAX_MAXCHAR(bufmaxchar, fill);
Victor Stinneree4544c2012-05-09 22:24:08 +020013856 }
13857 else {
13858 if (width > len)
Victor Stinnerd3f08822012-05-29 12:57:52 +020013859 bufmaxchar = MAX_MAXCHAR(bufmaxchar, fill);
Victor Stinneree4544c2012-05-09 22:24:08 +020013860 }
13861 }
13862 maxchar = _PyUnicode_FindMaxChar(temp, 0, pindex+len);
Victor Stinnerd3f08822012-05-29 12:57:52 +020013863 bufmaxchar = MAX_MAXCHAR(bufmaxchar, maxchar);
Victor Stinneree4544c2012-05-09 22:24:08 +020013864
13865 buflen = width;
13866 if (sign && len == width)
13867 buflen++;
13868
Victor Stinner3b1a74a2012-05-09 22:25:00 +020013869 if (_PyUnicodeWriter_Prepare(&writer, buflen, bufmaxchar) == -1)
Victor Stinneree4544c2012-05-09 22:24:08 +020013870 goto onError;
13871
13872 /* Write characters */
Benjamin Peterson29060642009-01-31 22:14:21 +000013873 if (sign) {
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013874 if (fill != ' ') {
Victor Stinneree4544c2012-05-09 22:24:08 +020013875 PyUnicode_WRITE(writer.kind, writer.data, writer.pos, signchar);
13876 writer.pos += 1;
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013877 }
Benjamin Peterson29060642009-01-31 22:14:21 +000013878 if (width > len)
13879 width--;
13880 }
13881 if ((flags & F_ALT) && (c == 'x' || c == 'X' || c == 'o')) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013882 assert(PyUnicode_READ(kind, pbuf, pindex) == '0');
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013883 assert(PyUnicode_READ(kind, pbuf, pindex + 1) == c);
Benjamin Peterson29060642009-01-31 22:14:21 +000013884 if (fill != ' ') {
Victor Stinnerf2c76aa2012-05-03 13:10:40 +020013885 PyUnicode_WRITE(writer.kind, writer.data, writer.pos, '0');
13886 PyUnicode_WRITE(writer.kind, writer.data, writer.pos+1, c);
13887 writer.pos += 2;
13888 pindex += 2;
Benjamin Peterson29060642009-01-31 22:14:21 +000013889 }
Benjamin Peterson29060642009-01-31 22:14:21 +000013890 width -= 2;
13891 if (width < 0)
13892 width = 0;
13893 len -= 2;
13894 }
13895 if (width > len && !(flags & F_LJUST)) {
Victor Stinnerf2c76aa2012-05-03 13:10:40 +020013896 sublen = width - len;
Victor Stinnerf2c76aa2012-05-03 13:10:40 +020013897 FILL(writer.kind, writer.data, fill, writer.pos, sublen);
13898 writer.pos += sublen;
Antoine Pitrou978b9d22011-10-07 12:35:48 +020013899 width = len;
Benjamin Peterson29060642009-01-31 22:14:21 +000013900 }
13901 if (fill == ' ') {
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013902 if (sign) {
Victor Stinneree4544c2012-05-09 22:24:08 +020013903 PyUnicode_WRITE(writer.kind, writer.data, writer.pos, signchar);
13904 writer.pos += 1;
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013905 }
Benjamin Peterson29060642009-01-31 22:14:21 +000013906 if ((flags & F_ALT) && (c == 'x' || c == 'X' || c == 'o')) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013907 assert(PyUnicode_READ(kind, pbuf, pindex) == '0');
13908 assert(PyUnicode_READ(kind, pbuf, pindex+1) == c);
Victor Stinnerf2c76aa2012-05-03 13:10:40 +020013909 PyUnicode_WRITE(writer.kind, writer.data, writer.pos, '0');
13910 PyUnicode_WRITE(writer.kind, writer.data, writer.pos+1, c);
13911 writer.pos += 2;
Victor Stinnerf2c76aa2012-05-03 13:10:40 +020013912 pindex += 2;
Benjamin Peterson14339b62009-01-31 16:36:08 +000013913 }
13914 }
Victor Stinnerf2c76aa2012-05-03 13:10:40 +020013915
Victor Stinnerc9d369f2012-06-16 02:22:37 +020013916 if (len) {
13917 _PyUnicode_FastCopyCharacters(writer.buffer, writer.pos,
13918 temp, pindex, len);
13919 writer.pos += len;
13920 }
Victor Stinnerf2c76aa2012-05-03 13:10:40 +020013921 if (width > len) {
Victor Stinneree4544c2012-05-09 22:24:08 +020013922 sublen = width - len;
Victor Stinnerf2c76aa2012-05-03 13:10:40 +020013923 FILL(writer.kind, writer.data, ' ', writer.pos, sublen);
13924 writer.pos += sublen;
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013925 }
Victor Stinneree4544c2012-05-09 22:24:08 +020013926
Victor Stinnerd3f08822012-05-29 12:57:52 +020013927nextarg:
Benjamin Peterson29060642009-01-31 22:14:21 +000013928 if (dict && (argidx < arglen) && c != '%') {
13929 PyErr_SetString(PyExc_TypeError,
13930 "not all arguments converted during string formatting");
Benjamin Peterson29060642009-01-31 22:14:21 +000013931 goto onError;
13932 }
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013933 Py_CLEAR(temp);
Benjamin Peterson29060642009-01-31 22:14:21 +000013934 } /* '%' */
Guido van Rossumd57fd912000-03-10 22:53:23 +000013935 } /* until end */
13936 if (argidx < arglen && !dict) {
Benjamin Peterson29060642009-01-31 22:14:21 +000013937 PyErr_SetString(PyExc_TypeError,
13938 "not all arguments converted during string formatting");
13939 goto onError;
Guido van Rossumd57fd912000-03-10 22:53:23 +000013940 }
13941
13942 if (args_owned) {
Benjamin Peterson29060642009-01-31 22:14:21 +000013943 Py_DECREF(args);
Guido van Rossumd57fd912000-03-10 22:53:23 +000013944 }
13945 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 return _PyUnicodeWriter_Finish(&writer);
Guido van Rossumd57fd912000-03-10 22:53:23 +000013949
Benjamin Peterson29060642009-01-31 22:14:21 +000013950 onError:
Guido van Rossumd57fd912000-03-10 22:53:23 +000013951 Py_DECREF(uformat);
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013952 Py_XDECREF(temp);
13953 Py_XDECREF(second);
Victor Stinner3b1a74a2012-05-09 22:25:00 +020013954 _PyUnicodeWriter_Dealloc(&writer);
Guido van Rossumd57fd912000-03-10 22:53:23 +000013955 if (args_owned) {
Benjamin Peterson29060642009-01-31 22:14:21 +000013956 Py_DECREF(args);
Guido van Rossumd57fd912000-03-10 22:53:23 +000013957 }
13958 return NULL;
13959}
13960
Jeremy Hylton938ace62002-07-17 16:30:39 +000013961static PyObject *
Guido van Rossume023fe02001-08-30 03:12:59 +000013962unicode_subtype_new(PyTypeObject *type, PyObject *args, PyObject *kwds);
13963
Tim Peters6d6c1a32001-08-02 04:15:00 +000013964static PyObject *
13965unicode_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
13966{
Benjamin Peterson29060642009-01-31 22:14:21 +000013967 PyObject *x = NULL;
Benjamin Peterson14339b62009-01-31 16:36:08 +000013968 static char *kwlist[] = {"object", "encoding", "errors", 0};
13969 char *encoding = NULL;
13970 char *errors = NULL;
Tim Peters6d6c1a32001-08-02 04:15:00 +000013971
Benjamin Peterson14339b62009-01-31 16:36:08 +000013972 if (type != &PyUnicode_Type)
13973 return unicode_subtype_new(type, args, kwds);
13974 if (!PyArg_ParseTupleAndKeywords(args, kwds, "|Oss:str",
Benjamin Peterson29060642009-01-31 22:14:21 +000013975 kwlist, &x, &encoding, &errors))
Benjamin Peterson14339b62009-01-31 16:36:08 +000013976 return NULL;
Victor Stinner382955f2011-12-11 21:44:00 +010013977 if (x == NULL) {
13978 Py_INCREF(unicode_empty);
13979 return unicode_empty;
13980 }
Benjamin Peterson14339b62009-01-31 16:36:08 +000013981 if (encoding == NULL && errors == NULL)
13982 return PyObject_Str(x);
13983 else
Benjamin Peterson29060642009-01-31 22:14:21 +000013984 return PyUnicode_FromEncodedObject(x, encoding, errors);
Tim Peters6d6c1a32001-08-02 04:15:00 +000013985}
13986
Guido van Rossume023fe02001-08-30 03:12:59 +000013987static PyObject *
13988unicode_subtype_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
13989{
Victor Stinner9db1a8b2011-10-23 20:04:37 +020013990 PyObject *unicode, *self;
Victor Stinner07ac3eb2011-10-01 16:16:43 +020013991 Py_ssize_t length, char_size;
13992 int share_wstr, share_utf8;
13993 unsigned int kind;
13994 void *data;
Guido van Rossume023fe02001-08-30 03:12:59 +000013995
Benjamin Peterson14339b62009-01-31 16:36:08 +000013996 assert(PyType_IsSubtype(type, &PyUnicode_Type));
Victor Stinner07ac3eb2011-10-01 16:16:43 +020013997
Victor Stinner9db1a8b2011-10-23 20:04:37 +020013998 unicode = unicode_new(&PyUnicode_Type, args, kwds);
Victor Stinner07ac3eb2011-10-01 16:16:43 +020013999 if (unicode == NULL)
Benjamin Peterson14339b62009-01-31 16:36:08 +000014000 return NULL;
Victor Stinner910337b2011-10-03 03:20:16 +020014001 assert(_PyUnicode_CHECK(unicode));
Benjamin Petersonbac79492012-01-14 13:34:47 -050014002 if (PyUnicode_READY(unicode) == -1) {
Benjamin Peterson22a29702012-01-02 09:00:30 -060014003 Py_DECREF(unicode);
Victor Stinner07ac3eb2011-10-01 16:16:43 +020014004 return NULL;
Benjamin Peterson22a29702012-01-02 09:00:30 -060014005 }
Victor Stinner07ac3eb2011-10-01 16:16:43 +020014006
Victor Stinner9db1a8b2011-10-23 20:04:37 +020014007 self = type->tp_alloc(type, 0);
Victor Stinner07ac3eb2011-10-01 16:16:43 +020014008 if (self == NULL) {
14009 Py_DECREF(unicode);
Benjamin Peterson14339b62009-01-31 16:36:08 +000014010 return NULL;
14011 }
Victor Stinner07ac3eb2011-10-01 16:16:43 +020014012 kind = PyUnicode_KIND(unicode);
14013 length = PyUnicode_GET_LENGTH(unicode);
14014
14015 _PyUnicode_LENGTH(self) = length;
Victor Stinnerfb9ea8c2011-10-06 01:45:57 +020014016#ifdef Py_DEBUG
14017 _PyUnicode_HASH(self) = -1;
14018#else
Victor Stinner07ac3eb2011-10-01 16:16:43 +020014019 _PyUnicode_HASH(self) = _PyUnicode_HASH(unicode);
Victor Stinnerfb9ea8c2011-10-06 01:45:57 +020014020#endif
Victor Stinner07ac3eb2011-10-01 16:16:43 +020014021 _PyUnicode_STATE(self).interned = 0;
14022 _PyUnicode_STATE(self).kind = kind;
14023 _PyUnicode_STATE(self).compact = 0;
Victor Stinner3cf46372011-10-03 14:42:15 +020014024 _PyUnicode_STATE(self).ascii = _PyUnicode_STATE(unicode).ascii;
Victor Stinner07ac3eb2011-10-01 16:16:43 +020014025 _PyUnicode_STATE(self).ready = 1;
14026 _PyUnicode_WSTR(self) = NULL;
14027 _PyUnicode_UTF8_LENGTH(self) = 0;
14028 _PyUnicode_UTF8(self) = NULL;
14029 _PyUnicode_WSTR_LENGTH(self) = 0;
Victor Stinnerc3c74152011-10-02 20:39:55 +020014030 _PyUnicode_DATA_ANY(self) = NULL;
Victor Stinner07ac3eb2011-10-01 16:16:43 +020014031
14032 share_utf8 = 0;
14033 share_wstr = 0;
14034 if (kind == PyUnicode_1BYTE_KIND) {
14035 char_size = 1;
14036 if (PyUnicode_MAX_CHAR_VALUE(unicode) < 128)
14037 share_utf8 = 1;
14038 }
14039 else if (kind == PyUnicode_2BYTE_KIND) {
14040 char_size = 2;
14041 if (sizeof(wchar_t) == 2)
14042 share_wstr = 1;
14043 }
14044 else {
14045 assert(kind == PyUnicode_4BYTE_KIND);
14046 char_size = 4;
14047 if (sizeof(wchar_t) == 4)
14048 share_wstr = 1;
14049 }
14050
14051 /* Ensure we won't overflow the length. */
14052 if (length > (PY_SSIZE_T_MAX / char_size - 1)) {
14053 PyErr_NoMemory();
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020014054 goto onError;
Benjamin Peterson14339b62009-01-31 16:36:08 +000014055 }
Victor Stinner07ac3eb2011-10-01 16:16:43 +020014056 data = PyObject_MALLOC((length + 1) * char_size);
14057 if (data == NULL) {
14058 PyErr_NoMemory();
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020014059 goto onError;
14060 }
14061
Victor Stinnerc3c74152011-10-02 20:39:55 +020014062 _PyUnicode_DATA_ANY(self) = data;
Victor Stinner07ac3eb2011-10-01 16:16:43 +020014063 if (share_utf8) {
14064 _PyUnicode_UTF8_LENGTH(self) = length;
14065 _PyUnicode_UTF8(self) = data;
14066 }
14067 if (share_wstr) {
14068 _PyUnicode_WSTR_LENGTH(self) = length;
14069 _PyUnicode_WSTR(self) = (wchar_t *)data;
14070 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020014071
Victor Stinner07ac3eb2011-10-01 16:16:43 +020014072 Py_MEMCPY(data, PyUnicode_DATA(unicode),
Martin v. Löwisc47adb02011-10-07 20:55:35 +020014073 kind * (length + 1));
Victor Stinnerbb10a1f2011-10-05 01:34:17 +020014074 assert(_PyUnicode_CheckConsistency(self, 1));
Victor Stinnerfb9ea8c2011-10-06 01:45:57 +020014075#ifdef Py_DEBUG
14076 _PyUnicode_HASH(self) = _PyUnicode_HASH(unicode);
14077#endif
Victor Stinnerdd18d3a2011-10-22 11:08:10 +020014078 Py_DECREF(unicode);
Victor Stinner7931d9a2011-11-04 00:22:48 +010014079 return self;
Victor Stinner07ac3eb2011-10-01 16:16:43 +020014080
14081onError:
14082 Py_DECREF(unicode);
14083 Py_DECREF(self);
14084 return NULL;
Guido van Rossume023fe02001-08-30 03:12:59 +000014085}
14086
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000014087PyDoc_STRVAR(unicode_doc,
Nick Coghlan573b1fd2012-08-16 14:13:07 +100014088 "str(object[, encoding[, errors]]) -> str\n\
Tim Peters6d6c1a32001-08-02 04:15:00 +000014089\n\
Nick Coghlan573b1fd2012-08-16 14:13:07 +100014090Create a new string object from the given object. If encoding or\n\
14091errors is specified, then the object must expose a data buffer\n\
14092that will be decoded using the given encoding and error handler.\n\
14093Otherwise, returns the result of object.__str__() (if defined)\n\
14094or repr(object).\n\
14095encoding defaults to sys.getdefaultencoding().\n\
14096errors defaults to 'strict'.");
Tim Peters6d6c1a32001-08-02 04:15:00 +000014097
Guido van Rossum50e9fb92006-08-17 05:42:55 +000014098static PyObject *unicode_iter(PyObject *seq);
14099
Guido van Rossumd57fd912000-03-10 22:53:23 +000014100PyTypeObject PyUnicode_Type = {
Martin v. Löwis9f2e3462007-07-21 17:22:18 +000014101 PyVarObject_HEAD_INIT(&PyType_Type, 0)
Benjamin Peterson14339b62009-01-31 16:36:08 +000014102 "str", /* tp_name */
14103 sizeof(PyUnicodeObject), /* tp_size */
14104 0, /* tp_itemsize */
Guido van Rossumd57fd912000-03-10 22:53:23 +000014105 /* Slots */
Benjamin Peterson14339b62009-01-31 16:36:08 +000014106 (destructor)unicode_dealloc, /* tp_dealloc */
14107 0, /* tp_print */
14108 0, /* tp_getattr */
14109 0, /* tp_setattr */
Mark Dickinsone94c6792009-02-02 20:36:42 +000014110 0, /* tp_reserved */
Benjamin Peterson14339b62009-01-31 16:36:08 +000014111 unicode_repr, /* tp_repr */
14112 &unicode_as_number, /* tp_as_number */
14113 &unicode_as_sequence, /* tp_as_sequence */
14114 &unicode_as_mapping, /* tp_as_mapping */
14115 (hashfunc) unicode_hash, /* tp_hash*/
14116 0, /* tp_call*/
14117 (reprfunc) unicode_str, /* tp_str */
14118 PyObject_GenericGetAttr, /* tp_getattro */
14119 0, /* tp_setattro */
14120 0, /* tp_as_buffer */
14121 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE |
Benjamin Peterson29060642009-01-31 22:14:21 +000014122 Py_TPFLAGS_UNICODE_SUBCLASS, /* tp_flags */
Benjamin Peterson14339b62009-01-31 16:36:08 +000014123 unicode_doc, /* tp_doc */
14124 0, /* tp_traverse */
14125 0, /* tp_clear */
14126 PyUnicode_RichCompare, /* tp_richcompare */
14127 0, /* tp_weaklistoffset */
14128 unicode_iter, /* tp_iter */
14129 0, /* tp_iternext */
14130 unicode_methods, /* tp_methods */
14131 0, /* tp_members */
14132 0, /* tp_getset */
14133 &PyBaseObject_Type, /* tp_base */
14134 0, /* tp_dict */
14135 0, /* tp_descr_get */
14136 0, /* tp_descr_set */
14137 0, /* tp_dictoffset */
14138 0, /* tp_init */
14139 0, /* tp_alloc */
14140 unicode_new, /* tp_new */
14141 PyObject_Del, /* tp_free */
Guido van Rossumd57fd912000-03-10 22:53:23 +000014142};
14143
14144/* Initialize the Unicode implementation */
14145
Victor Stinner3a50e702011-10-18 21:21:00 +020014146int _PyUnicode_Init(void)
Guido van Rossumd57fd912000-03-10 22:53:23 +000014147{
Marc-André Lemburg8155e0e2001-04-23 14:44:21 +000014148 int i;
14149
Thomas Wouters477c8d52006-05-27 19:21:47 +000014150 /* XXX - move this array to unicodectype.c ? */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020014151 Py_UCS2 linebreak[] = {
Thomas Wouters477c8d52006-05-27 19:21:47 +000014152 0x000A, /* LINE FEED */
14153 0x000D, /* CARRIAGE RETURN */
14154 0x001C, /* FILE SEPARATOR */
14155 0x001D, /* GROUP SEPARATOR */
14156 0x001E, /* RECORD SEPARATOR */
14157 0x0085, /* NEXT LINE */
14158 0x2028, /* LINE SEPARATOR */
14159 0x2029, /* PARAGRAPH SEPARATOR */
14160 };
14161
Fred Drakee4315f52000-05-09 19:53:39 +000014162 /* Init the implementation */
Victor Stinnera464fc12011-10-02 20:39:30 +020014163 unicode_empty = PyUnicode_New(0, 0);
Thomas Wouters0e3f5912006-08-11 14:57:12 +000014164 if (!unicode_empty)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020014165 Py_FatalError("Can't create empty string");
Victor Stinnerd3df8ab2011-11-22 01:22:34 +010014166 assert(_PyUnicode_CheckConsistency(unicode_empty, 1));
Thomas Wouters0e3f5912006-08-11 14:57:12 +000014167
Marc-André Lemburg8155e0e2001-04-23 14:44:21 +000014168 for (i = 0; i < 256; i++)
Benjamin Peterson29060642009-01-31 22:14:21 +000014169 unicode_latin1[i] = NULL;
Guido van Rossumcacfc072002-05-24 19:01:59 +000014170 if (PyType_Ready(&PyUnicode_Type) < 0)
Benjamin Peterson29060642009-01-31 22:14:21 +000014171 Py_FatalError("Can't initialize 'unicode'");
Thomas Wouters477c8d52006-05-27 19:21:47 +000014172
14173 /* initialize the linebreak bloom filter */
14174 bloom_linebreak = make_bloom_mask(
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020014175 PyUnicode_2BYTE_KIND, linebreak,
Victor Stinner63941882011-09-29 00:42:28 +020014176 Py_ARRAY_LENGTH(linebreak));
Thomas Wouters0e3f5912006-08-11 14:57:12 +000014177
14178 PyType_Ready(&EncodingMapType);
Victor Stinner3a50e702011-10-18 21:21:00 +020014179
14180#ifdef HAVE_MBCS
14181 winver.dwOSVersionInfoSize = sizeof(winver);
14182 if (!GetVersionEx((OSVERSIONINFO*)&winver)) {
14183 PyErr_SetFromWindowsErr(0);
14184 return -1;
14185 }
14186#endif
14187 return 0;
Guido van Rossumd57fd912000-03-10 22:53:23 +000014188}
14189
14190/* Finalize the Unicode implementation */
14191
Christian Heimesa156e092008-02-16 07:38:31 +000014192int
14193PyUnicode_ClearFreeList(void)
14194{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020014195 return 0;
Christian Heimesa156e092008-02-16 07:38:31 +000014196}
14197
Guido van Rossumd57fd912000-03-10 22:53:23 +000014198void
Thomas Wouters78890102000-07-22 19:25:51 +000014199_PyUnicode_Fini(void)
Guido van Rossumd57fd912000-03-10 22:53:23 +000014200{
Marc-André Lemburg8155e0e2001-04-23 14:44:21 +000014201 int i;
Guido van Rossumd57fd912000-03-10 22:53:23 +000014202
Guido van Rossum4ae8ef82000-10-03 18:09:04 +000014203 Py_XDECREF(unicode_empty);
14204 unicode_empty = NULL;
Barry Warsaw5b4c2282000-10-03 20:45:26 +000014205
Marc-André Lemburg8155e0e2001-04-23 14:44:21 +000014206 for (i = 0; i < 256; i++) {
Benjamin Peterson29060642009-01-31 22:14:21 +000014207 if (unicode_latin1[i]) {
14208 Py_DECREF(unicode_latin1[i]);
14209 unicode_latin1[i] = NULL;
14210 }
Marc-André Lemburg8155e0e2001-04-23 14:44:21 +000014211 }
Martin v. Löwisafe55bb2011-10-09 10:38:36 +020014212 _PyUnicode_ClearStaticStrings();
Christian Heimesa156e092008-02-16 07:38:31 +000014213 (void)PyUnicode_ClearFreeList();
Guido van Rossumd57fd912000-03-10 22:53:23 +000014214}
Martin v. Löwis9a3a9f72003-05-18 12:31:09 +000014215
Walter Dörwald16807132007-05-25 13:52:07 +000014216void
14217PyUnicode_InternInPlace(PyObject **p)
14218{
Victor Stinner9db1a8b2011-10-23 20:04:37 +020014219 register PyObject *s = *p;
Benjamin Peterson14339b62009-01-31 16:36:08 +000014220 PyObject *t;
Victor Stinner4fae54c2011-10-03 02:01:52 +020014221#ifdef Py_DEBUG
14222 assert(s != NULL);
14223 assert(_PyUnicode_CHECK(s));
14224#else
Benjamin Peterson14339b62009-01-31 16:36:08 +000014225 if (s == NULL || !PyUnicode_Check(s))
Victor Stinner4fae54c2011-10-03 02:01:52 +020014226 return;
14227#endif
Benjamin Peterson14339b62009-01-31 16:36:08 +000014228 /* If it's a subclass, we don't really know what putting
14229 it in the interned dict might do. */
14230 if (!PyUnicode_CheckExact(s))
14231 return;
14232 if (PyUnicode_CHECK_INTERNED(s))
14233 return;
14234 if (interned == NULL) {
14235 interned = PyDict_New();
14236 if (interned == NULL) {
14237 PyErr_Clear(); /* Don't leave an exception */
14238 return;
14239 }
14240 }
14241 /* It might be that the GetItem call fails even
14242 though the key is present in the dictionary,
14243 namely when this happens during a stack overflow. */
14244 Py_ALLOW_RECURSION
Victor Stinner7931d9a2011-11-04 00:22:48 +010014245 t = PyDict_GetItem(interned, s);
Benjamin Peterson14339b62009-01-31 16:36:08 +000014246 Py_END_ALLOW_RECURSION
Martin v. Löwis5b222132007-06-10 09:51:05 +000014247
Benjamin Peterson29060642009-01-31 22:14:21 +000014248 if (t) {
14249 Py_INCREF(t);
14250 Py_DECREF(*p);
14251 *p = t;
14252 return;
14253 }
Walter Dörwald16807132007-05-25 13:52:07 +000014254
Benjamin Peterson14339b62009-01-31 16:36:08 +000014255 PyThreadState_GET()->recursion_critical = 1;
Victor Stinner7931d9a2011-11-04 00:22:48 +010014256 if (PyDict_SetItem(interned, s, s) < 0) {
Benjamin Peterson14339b62009-01-31 16:36:08 +000014257 PyErr_Clear();
14258 PyThreadState_GET()->recursion_critical = 0;
14259 return;
14260 }
14261 PyThreadState_GET()->recursion_critical = 0;
14262 /* The two references in interned are not counted by refcnt.
14263 The deallocator will take care of this */
14264 Py_REFCNT(s) -= 2;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020014265 _PyUnicode_STATE(s).interned = SSTATE_INTERNED_MORTAL;
Walter Dörwald16807132007-05-25 13:52:07 +000014266}
14267
14268void
14269PyUnicode_InternImmortal(PyObject **p)
14270{
Benjamin Peterson14339b62009-01-31 16:36:08 +000014271 PyUnicode_InternInPlace(p);
14272 if (PyUnicode_CHECK_INTERNED(*p) != SSTATE_INTERNED_IMMORTAL) {
Victor Stinneraf9e4b82011-10-23 20:07:00 +020014273 _PyUnicode_STATE(*p).interned = SSTATE_INTERNED_IMMORTAL;
Benjamin Peterson14339b62009-01-31 16:36:08 +000014274 Py_INCREF(*p);
14275 }
Walter Dörwald16807132007-05-25 13:52:07 +000014276}
14277
14278PyObject *
14279PyUnicode_InternFromString(const char *cp)
14280{
Benjamin Peterson14339b62009-01-31 16:36:08 +000014281 PyObject *s = PyUnicode_FromString(cp);
14282 if (s == NULL)
14283 return NULL;
14284 PyUnicode_InternInPlace(&s);
14285 return s;
Walter Dörwald16807132007-05-25 13:52:07 +000014286}
14287
Alexander Belopolsky40018472011-02-26 01:02:56 +000014288void
14289_Py_ReleaseInternedUnicodeStrings(void)
Walter Dörwald16807132007-05-25 13:52:07 +000014290{
Benjamin Peterson14339b62009-01-31 16:36:08 +000014291 PyObject *keys;
Victor Stinner9db1a8b2011-10-23 20:04:37 +020014292 PyObject *s;
Benjamin Peterson14339b62009-01-31 16:36:08 +000014293 Py_ssize_t i, n;
14294 Py_ssize_t immortal_size = 0, mortal_size = 0;
Walter Dörwald16807132007-05-25 13:52:07 +000014295
Benjamin Peterson14339b62009-01-31 16:36:08 +000014296 if (interned == NULL || !PyDict_Check(interned))
14297 return;
14298 keys = PyDict_Keys(interned);
14299 if (keys == NULL || !PyList_Check(keys)) {
14300 PyErr_Clear();
14301 return;
14302 }
Walter Dörwald16807132007-05-25 13:52:07 +000014303
Benjamin Peterson14339b62009-01-31 16:36:08 +000014304 /* Since _Py_ReleaseInternedUnicodeStrings() is intended to help a leak
14305 detector, interned unicode strings are not forcibly deallocated;
14306 rather, we give them their stolen references back, and then clear
14307 and DECREF the interned dict. */
Walter Dörwald16807132007-05-25 13:52:07 +000014308
Benjamin Peterson14339b62009-01-31 16:36:08 +000014309 n = PyList_GET_SIZE(keys);
14310 fprintf(stderr, "releasing %" PY_FORMAT_SIZE_T "d interned strings\n",
Benjamin Peterson29060642009-01-31 22:14:21 +000014311 n);
Benjamin Peterson14339b62009-01-31 16:36:08 +000014312 for (i = 0; i < n; i++) {
Victor Stinner9db1a8b2011-10-23 20:04:37 +020014313 s = PyList_GET_ITEM(keys, i);
Victor Stinner6b56a7f2011-10-04 20:04:52 +020014314 if (PyUnicode_READY(s) == -1) {
14315 assert(0 && "could not ready string");
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020014316 fprintf(stderr, "could not ready string\n");
Victor Stinner6b56a7f2011-10-04 20:04:52 +020014317 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020014318 switch (PyUnicode_CHECK_INTERNED(s)) {
Benjamin Peterson14339b62009-01-31 16:36:08 +000014319 case SSTATE_NOT_INTERNED:
14320 /* XXX Shouldn't happen */
14321 break;
14322 case SSTATE_INTERNED_IMMORTAL:
14323 Py_REFCNT(s) += 1;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020014324 immortal_size += PyUnicode_GET_LENGTH(s);
Benjamin Peterson14339b62009-01-31 16:36:08 +000014325 break;
14326 case SSTATE_INTERNED_MORTAL:
14327 Py_REFCNT(s) += 2;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020014328 mortal_size += PyUnicode_GET_LENGTH(s);
Benjamin Peterson14339b62009-01-31 16:36:08 +000014329 break;
14330 default:
14331 Py_FatalError("Inconsistent interned string state.");
14332 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020014333 _PyUnicode_STATE(s).interned = SSTATE_NOT_INTERNED;
Benjamin Peterson14339b62009-01-31 16:36:08 +000014334 }
14335 fprintf(stderr, "total size of all interned strings: "
14336 "%" PY_FORMAT_SIZE_T "d/%" PY_FORMAT_SIZE_T "d "
14337 "mortal/immortal\n", mortal_size, immortal_size);
14338 Py_DECREF(keys);
14339 PyDict_Clear(interned);
14340 Py_DECREF(interned);
14341 interned = NULL;
Walter Dörwald16807132007-05-25 13:52:07 +000014342}
Guido van Rossum50e9fb92006-08-17 05:42:55 +000014343
14344
14345/********************* Unicode Iterator **************************/
14346
14347typedef struct {
Benjamin Peterson14339b62009-01-31 16:36:08 +000014348 PyObject_HEAD
14349 Py_ssize_t it_index;
Victor Stinner9db1a8b2011-10-23 20:04:37 +020014350 PyObject *it_seq; /* Set to NULL when iterator is exhausted */
Guido van Rossum50e9fb92006-08-17 05:42:55 +000014351} unicodeiterobject;
14352
14353static void
14354unicodeiter_dealloc(unicodeiterobject *it)
14355{
Benjamin Peterson14339b62009-01-31 16:36:08 +000014356 _PyObject_GC_UNTRACK(it);
14357 Py_XDECREF(it->it_seq);
14358 PyObject_GC_Del(it);
Guido van Rossum50e9fb92006-08-17 05:42:55 +000014359}
14360
14361static int
14362unicodeiter_traverse(unicodeiterobject *it, visitproc visit, void *arg)
14363{
Benjamin Peterson14339b62009-01-31 16:36:08 +000014364 Py_VISIT(it->it_seq);
14365 return 0;
Guido van Rossum50e9fb92006-08-17 05:42:55 +000014366}
14367
14368static PyObject *
14369unicodeiter_next(unicodeiterobject *it)
14370{
Victor Stinner9db1a8b2011-10-23 20:04:37 +020014371 PyObject *seq, *item;
Guido van Rossum50e9fb92006-08-17 05:42:55 +000014372
Benjamin Peterson14339b62009-01-31 16:36:08 +000014373 assert(it != NULL);
14374 seq = it->it_seq;
14375 if (seq == NULL)
14376 return NULL;
Victor Stinner910337b2011-10-03 03:20:16 +020014377 assert(_PyUnicode_CHECK(seq));
Guido van Rossum50e9fb92006-08-17 05:42:55 +000014378
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020014379 if (it->it_index < PyUnicode_GET_LENGTH(seq)) {
14380 int kind = PyUnicode_KIND(seq);
14381 void *data = PyUnicode_DATA(seq);
14382 Py_UCS4 chr = PyUnicode_READ(kind, data, it->it_index);
14383 item = PyUnicode_FromOrdinal(chr);
Benjamin Peterson14339b62009-01-31 16:36:08 +000014384 if (item != NULL)
14385 ++it->it_index;
14386 return item;
14387 }
Guido van Rossum50e9fb92006-08-17 05:42:55 +000014388
Benjamin Peterson14339b62009-01-31 16:36:08 +000014389 Py_DECREF(seq);
14390 it->it_seq = NULL;
14391 return NULL;
Guido van Rossum50e9fb92006-08-17 05:42:55 +000014392}
14393
14394static PyObject *
14395unicodeiter_len(unicodeiterobject *it)
14396{
Benjamin Peterson14339b62009-01-31 16:36:08 +000014397 Py_ssize_t len = 0;
14398 if (it->it_seq)
Victor Stinnerc4f281e2011-10-11 22:11:42 +020014399 len = PyUnicode_GET_LENGTH(it->it_seq) - it->it_index;
Benjamin Peterson14339b62009-01-31 16:36:08 +000014400 return PyLong_FromSsize_t(len);
Guido van Rossum50e9fb92006-08-17 05:42:55 +000014401}
14402
14403PyDoc_STRVAR(length_hint_doc, "Private method returning an estimate of len(list(it)).");
14404
Kristján Valur Jónsson31668b82012-04-03 10:49:41 +000014405static PyObject *
14406unicodeiter_reduce(unicodeiterobject *it)
14407{
14408 if (it->it_seq != NULL) {
Antoine Pitroua7013882012-04-05 00:04:20 +020014409 return Py_BuildValue("N(O)n", _PyObject_GetBuiltin("iter"),
Kristján Valur Jónsson31668b82012-04-03 10:49:41 +000014410 it->it_seq, it->it_index);
14411 } else {
14412 PyObject *u = PyUnicode_FromUnicode(NULL, 0);
14413 if (u == NULL)
14414 return NULL;
Antoine Pitroua7013882012-04-05 00:04:20 +020014415 return Py_BuildValue("N(N)", _PyObject_GetBuiltin("iter"), u);
Kristján Valur Jónsson31668b82012-04-03 10:49:41 +000014416 }
14417}
14418
14419PyDoc_STRVAR(reduce_doc, "Return state information for pickling.");
14420
14421static PyObject *
14422unicodeiter_setstate(unicodeiterobject *it, PyObject *state)
14423{
14424 Py_ssize_t index = PyLong_AsSsize_t(state);
14425 if (index == -1 && PyErr_Occurred())
14426 return NULL;
14427 if (index < 0)
14428 index = 0;
14429 it->it_index = index;
14430 Py_RETURN_NONE;
14431}
14432
14433PyDoc_STRVAR(setstate_doc, "Set state information for unpickling.");
14434
Guido van Rossum50e9fb92006-08-17 05:42:55 +000014435static PyMethodDef unicodeiter_methods[] = {
Benjamin Peterson14339b62009-01-31 16:36:08 +000014436 {"__length_hint__", (PyCFunction)unicodeiter_len, METH_NOARGS,
Benjamin Peterson29060642009-01-31 22:14:21 +000014437 length_hint_doc},
Kristján Valur Jónsson31668b82012-04-03 10:49:41 +000014438 {"__reduce__", (PyCFunction)unicodeiter_reduce, METH_NOARGS,
14439 reduce_doc},
14440 {"__setstate__", (PyCFunction)unicodeiter_setstate, METH_O,
14441 setstate_doc},
Benjamin Peterson14339b62009-01-31 16:36:08 +000014442 {NULL, NULL} /* sentinel */
Guido van Rossum50e9fb92006-08-17 05:42:55 +000014443};
14444
14445PyTypeObject PyUnicodeIter_Type = {
Benjamin Peterson14339b62009-01-31 16:36:08 +000014446 PyVarObject_HEAD_INIT(&PyType_Type, 0)
14447 "str_iterator", /* tp_name */
14448 sizeof(unicodeiterobject), /* tp_basicsize */
14449 0, /* tp_itemsize */
14450 /* methods */
14451 (destructor)unicodeiter_dealloc, /* tp_dealloc */
14452 0, /* tp_print */
14453 0, /* tp_getattr */
14454 0, /* tp_setattr */
Mark Dickinsone94c6792009-02-02 20:36:42 +000014455 0, /* tp_reserved */
Benjamin Peterson14339b62009-01-31 16:36:08 +000014456 0, /* tp_repr */
14457 0, /* tp_as_number */
14458 0, /* tp_as_sequence */
14459 0, /* tp_as_mapping */
14460 0, /* tp_hash */
14461 0, /* tp_call */
14462 0, /* tp_str */
14463 PyObject_GenericGetAttr, /* tp_getattro */
14464 0, /* tp_setattro */
14465 0, /* tp_as_buffer */
14466 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC,/* tp_flags */
14467 0, /* tp_doc */
14468 (traverseproc)unicodeiter_traverse, /* tp_traverse */
14469 0, /* tp_clear */
14470 0, /* tp_richcompare */
14471 0, /* tp_weaklistoffset */
14472 PyObject_SelfIter, /* tp_iter */
14473 (iternextfunc)unicodeiter_next, /* tp_iternext */
14474 unicodeiter_methods, /* tp_methods */
14475 0,
Guido van Rossum50e9fb92006-08-17 05:42:55 +000014476};
14477
14478static PyObject *
14479unicode_iter(PyObject *seq)
14480{
Benjamin Peterson14339b62009-01-31 16:36:08 +000014481 unicodeiterobject *it;
Guido van Rossum50e9fb92006-08-17 05:42:55 +000014482
Benjamin Peterson14339b62009-01-31 16:36:08 +000014483 if (!PyUnicode_Check(seq)) {
14484 PyErr_BadInternalCall();
14485 return NULL;
14486 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020014487 if (PyUnicode_READY(seq) == -1)
14488 return NULL;
Benjamin Peterson14339b62009-01-31 16:36:08 +000014489 it = PyObject_GC_New(unicodeiterobject, &PyUnicodeIter_Type);
14490 if (it == NULL)
14491 return NULL;
14492 it->it_index = 0;
14493 Py_INCREF(seq);
Victor Stinner9db1a8b2011-10-23 20:04:37 +020014494 it->it_seq = seq;
Benjamin Peterson14339b62009-01-31 16:36:08 +000014495 _PyObject_GC_TRACK(it);
14496 return (PyObject *)it;
Guido van Rossum50e9fb92006-08-17 05:42:55 +000014497}
14498
Martin v. Löwis0d3072e2011-10-31 08:40:56 +010014499
14500size_t
14501Py_UNICODE_strlen(const Py_UNICODE *u)
14502{
14503 int res = 0;
14504 while(*u++)
14505 res++;
14506 return res;
14507}
14508
14509Py_UNICODE*
14510Py_UNICODE_strcpy(Py_UNICODE *s1, const Py_UNICODE *s2)
14511{
14512 Py_UNICODE *u = s1;
14513 while ((*u++ = *s2++));
14514 return s1;
14515}
14516
14517Py_UNICODE*
14518Py_UNICODE_strncpy(Py_UNICODE *s1, const Py_UNICODE *s2, size_t n)
14519{
14520 Py_UNICODE *u = s1;
14521 while ((*u++ = *s2++))
14522 if (n-- == 0)
14523 break;
14524 return s1;
14525}
14526
14527Py_UNICODE*
14528Py_UNICODE_strcat(Py_UNICODE *s1, const Py_UNICODE *s2)
14529{
14530 Py_UNICODE *u1 = s1;
14531 u1 += Py_UNICODE_strlen(u1);
14532 Py_UNICODE_strcpy(u1, s2);
14533 return s1;
14534}
14535
14536int
14537Py_UNICODE_strcmp(const Py_UNICODE *s1, const Py_UNICODE *s2)
14538{
14539 while (*s1 && *s2 && *s1 == *s2)
14540 s1++, s2++;
14541 if (*s1 && *s2)
14542 return (*s1 < *s2) ? -1 : +1;
14543 if (*s1)
14544 return 1;
14545 if (*s2)
14546 return -1;
14547 return 0;
14548}
14549
14550int
14551Py_UNICODE_strncmp(const Py_UNICODE *s1, const Py_UNICODE *s2, size_t n)
14552{
14553 register Py_UNICODE u1, u2;
14554 for (; n != 0; n--) {
14555 u1 = *s1;
14556 u2 = *s2;
14557 if (u1 != u2)
14558 return (u1 < u2) ? -1 : +1;
14559 if (u1 == '\0')
14560 return 0;
14561 s1++;
14562 s2++;
14563 }
14564 return 0;
14565}
14566
14567Py_UNICODE*
14568Py_UNICODE_strchr(const Py_UNICODE *s, Py_UNICODE c)
14569{
14570 const Py_UNICODE *p;
14571 for (p = s; *p; p++)
14572 if (*p == c)
14573 return (Py_UNICODE*)p;
14574 return NULL;
14575}
14576
14577Py_UNICODE*
14578Py_UNICODE_strrchr(const Py_UNICODE *s, Py_UNICODE c)
14579{
14580 const Py_UNICODE *p;
14581 p = s + Py_UNICODE_strlen(s);
14582 while (p != s) {
14583 p--;
14584 if (*p == c)
14585 return (Py_UNICODE*)p;
14586 }
14587 return NULL;
14588}
Victor Stinner331ea922010-08-10 16:37:20 +000014589
Victor Stinner71133ff2010-09-01 23:43:53 +000014590Py_UNICODE*
Victor Stinner9db1a8b2011-10-23 20:04:37 +020014591PyUnicode_AsUnicodeCopy(PyObject *unicode)
Victor Stinner71133ff2010-09-01 23:43:53 +000014592{
Victor Stinner577db2c2011-10-11 22:12:48 +020014593 Py_UNICODE *u, *copy;
Victor Stinner57ffa9d2011-10-23 20:10:08 +020014594 Py_ssize_t len, size;
Victor Stinner71133ff2010-09-01 23:43:53 +000014595
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020014596 if (!PyUnicode_Check(unicode)) {
14597 PyErr_BadArgument();
14598 return NULL;
14599 }
Victor Stinner57ffa9d2011-10-23 20:10:08 +020014600 u = PyUnicode_AsUnicodeAndSize(unicode, &len);
Victor Stinner577db2c2011-10-11 22:12:48 +020014601 if (u == NULL)
14602 return NULL;
Victor Stinner71133ff2010-09-01 23:43:53 +000014603 /* Ensure we won't overflow the size. */
Victor Stinner57ffa9d2011-10-23 20:10:08 +020014604 if (len > ((PY_SSIZE_T_MAX / sizeof(Py_UNICODE)) - 1)) {
Victor Stinner71133ff2010-09-01 23:43:53 +000014605 PyErr_NoMemory();
14606 return NULL;
14607 }
Victor Stinner57ffa9d2011-10-23 20:10:08 +020014608 size = len + 1; /* copy the null character */
Victor Stinner71133ff2010-09-01 23:43:53 +000014609 size *= sizeof(Py_UNICODE);
14610 copy = PyMem_Malloc(size);
14611 if (copy == NULL) {
14612 PyErr_NoMemory();
14613 return NULL;
14614 }
Victor Stinner577db2c2011-10-11 22:12:48 +020014615 memcpy(copy, u, size);
Victor Stinner71133ff2010-09-01 23:43:53 +000014616 return copy;
14617}
Martin v. Löwis5b222132007-06-10 09:51:05 +000014618
Georg Brandl66c221e2010-10-14 07:04:07 +000014619/* A _string module, to export formatter_parser and formatter_field_name_split
14620 to the string.Formatter class implemented in Python. */
14621
14622static PyMethodDef _string_methods[] = {
14623 {"formatter_field_name_split", (PyCFunction) formatter_field_name_split,
14624 METH_O, PyDoc_STR("split the argument as a field name")},
14625 {"formatter_parser", (PyCFunction) formatter_parser,
14626 METH_O, PyDoc_STR("parse the argument as a format string")},
14627 {NULL, NULL}
14628};
14629
14630static struct PyModuleDef _string_module = {
14631 PyModuleDef_HEAD_INIT,
14632 "_string",
14633 PyDoc_STR("string helper module"),
14634 0,
14635 _string_methods,
14636 NULL,
14637 NULL,
14638 NULL,
14639 NULL
14640};
14641
14642PyMODINIT_FUNC
14643PyInit__string(void)
14644{
14645 return PyModule_Create(&_string_module);
14646}
14647
14648
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000014649#ifdef __cplusplus
14650}
14651#endif