blob: 0da565a612b33c9741d5d104e5b73e922f3fe6f7 [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 = \
Antoine Pitrouca8aa4a2012-09-20 20:56:47 +0200162 _iter + _Py_SIZE_ROUND_DOWN(n, 4); \
Antoine Pitroue459a082011-10-11 20:58:41 +0200163 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 quickly check whether a C 'long' contains a
4639 non-ASCII, UTF8-encoded char. */
4640#if (SIZEOF_LONG == 8)
Mark Dickinson01ac8b62012-07-07 14:08:48 +02004641# define ASCII_CHAR_MASK 0x8080808080808080UL
Antoine Pitrouab868312009-01-10 15:40:25 +00004642#elif (SIZEOF_LONG == 4)
Mark Dickinson01ac8b62012-07-07 14:08:48 +02004643# define ASCII_CHAR_MASK 0x80808080UL
Antoine Pitrouab868312009-01-10 15:40:25 +00004644#else
4645# error C 'long' size should be either 4 or 8!
4646#endif
4647
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004648static Py_ssize_t
4649ascii_decode(const char *start, const char *end, Py_UCS1 *dest)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02004650{
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004651 const char *p = start;
Antoine Pitrouca8aa4a2012-09-20 20:56:47 +02004652 const char *aligned_end = (const char *) _Py_ALIGN_DOWN(end, SIZEOF_LONG);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02004653
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004654#if SIZEOF_LONG <= SIZEOF_VOID_P
Antoine Pitrouca8aa4a2012-09-20 20:56:47 +02004655 assert(_Py_IS_ALIGNED(dest, SIZEOF_LONG));
4656 if (_Py_IS_ALIGNED(p, SIZEOF_LONG)) {
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004657 /* Fast path, see in STRINGLIB(utf8_decode) for
4658 an explanation. */
4659 /* Help register allocation */
4660 register const char *_p = p;
4661 register Py_UCS1 * q = dest;
4662 while (_p < aligned_end) {
4663 unsigned long value = *(const unsigned long *) _p;
4664 if (value & ASCII_CHAR_MASK)
Benjamin Peterson29060642009-01-31 22:14:21 +00004665 break;
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004666 *((unsigned long *)q) = value;
4667 _p += SIZEOF_LONG;
4668 q += SIZEOF_LONG;
Benjamin Peterson14339b62009-01-31 16:36:08 +00004669 }
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004670 p = _p;
4671 while (p < end) {
4672 if ((unsigned char)*p & 0x80)
4673 break;
4674 *q++ = *p++;
Guido van Rossumd57fd912000-03-10 22:53:23 +00004675 }
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004676 return p - start;
Guido van Rossumd57fd912000-03-10 22:53:23 +00004677 }
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004678#endif
4679 while (p < end) {
4680 /* Fast path, see in STRINGLIB(utf8_decode) in stringlib/codecs.h
4681 for an explanation. */
Antoine Pitrouca8aa4a2012-09-20 20:56:47 +02004682 if (_Py_IS_ALIGNED(p, SIZEOF_LONG)) {
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004683 /* Help register allocation */
4684 register const char *_p = p;
4685 while (_p < aligned_end) {
4686 unsigned long value = *(unsigned long *) _p;
4687 if (value & ASCII_CHAR_MASK)
4688 break;
4689 _p += SIZEOF_LONG;
4690 }
4691 p = _p;
4692 if (_p == end)
4693 break;
4694 }
4695 if ((unsigned char)*p & 0x80)
4696 break;
4697 ++p;
4698 }
4699 memcpy(dest, start, p - start);
4700 return p - start;
Guido van Rossumd57fd912000-03-10 22:53:23 +00004701}
Antoine Pitrouab868312009-01-10 15:40:25 +00004702
Victor Stinner785938e2011-12-11 20:09:03 +01004703PyObject *
4704PyUnicode_DecodeUTF8Stateful(const char *s,
4705 Py_ssize_t size,
4706 const char *errors,
4707 Py_ssize_t *consumed)
4708{
Victor Stinner785938e2011-12-11 20:09:03 +01004709 PyObject *unicode;
Victor Stinner785938e2011-12-11 20:09:03 +01004710 const char *starts = s;
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004711 const char *end = s + size;
4712 Py_ssize_t outpos;
4713
4714 Py_ssize_t startinpos;
4715 Py_ssize_t endinpos;
4716 const char *errmsg = "";
4717 PyObject *errorHandler = NULL;
4718 PyObject *exc = NULL;
Victor Stinner785938e2011-12-11 20:09:03 +01004719
4720 if (size == 0) {
4721 if (consumed)
4722 *consumed = 0;
Victor Stinner382955f2011-12-11 21:44:00 +01004723 Py_INCREF(unicode_empty);
4724 return unicode_empty;
Victor Stinner785938e2011-12-11 20:09:03 +01004725 }
4726
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004727 /* ASCII is equivalent to the first 128 ordinals in Unicode. */
4728 if (size == 1 && (unsigned char)s[0] < 128) {
Victor Stinner785938e2011-12-11 20:09:03 +01004729 if (consumed)
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004730 *consumed = 1;
4731 return get_latin1_char((unsigned char)s[0]);
Victor Stinner785938e2011-12-11 20:09:03 +01004732 }
4733
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004734 unicode = PyUnicode_New(size, 127);
Victor Stinner785938e2011-12-11 20:09:03 +01004735 if (!unicode)
4736 return NULL;
Victor Stinner785938e2011-12-11 20:09:03 +01004737
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004738 outpos = ascii_decode(s, end, PyUnicode_1BYTE_DATA(unicode));
4739 s += outpos;
4740 while (s < end) {
4741 Py_UCS4 ch;
4742 int kind = PyUnicode_KIND(unicode);
4743 if (kind == PyUnicode_1BYTE_KIND) {
4744 if (PyUnicode_IS_ASCII(unicode))
4745 ch = asciilib_utf8_decode(&s, end,
4746 PyUnicode_1BYTE_DATA(unicode), &outpos);
4747 else
4748 ch = ucs1lib_utf8_decode(&s, end,
4749 PyUnicode_1BYTE_DATA(unicode), &outpos);
4750 } else if (kind == PyUnicode_2BYTE_KIND) {
4751 ch = ucs2lib_utf8_decode(&s, end,
4752 PyUnicode_2BYTE_DATA(unicode), &outpos);
4753 } else {
4754 assert(kind == PyUnicode_4BYTE_KIND);
4755 ch = ucs4lib_utf8_decode(&s, end,
4756 PyUnicode_4BYTE_DATA(unicode), &outpos);
4757 }
4758
4759 switch (ch) {
4760 case 0:
4761 if (s == end || consumed)
4762 goto End;
4763 errmsg = "unexpected end of data";
4764 startinpos = s - starts;
4765 endinpos = startinpos + 1;
4766 while (endinpos < size && (starts[endinpos] & 0xC0) == 0x80)
4767 endinpos++;
4768 break;
4769 case 1:
4770 errmsg = "invalid start byte";
4771 startinpos = s - starts;
4772 endinpos = startinpos + 1;
4773 break;
4774 case 2:
4775 errmsg = "invalid continuation byte";
4776 startinpos = s - starts;
4777 endinpos = startinpos + 1;
4778 while (endinpos < size && (starts[endinpos] & 0xC0) == 0x80)
4779 endinpos++;
4780 break;
4781 default:
4782 if (unicode_putchar(&unicode, &outpos, ch) < 0)
4783 goto onError;
4784 continue;
4785 }
4786
4787 if (unicode_decode_call_errorhandler(
4788 errors, &errorHandler,
4789 "utf-8", errmsg,
4790 &starts, &end, &startinpos, &endinpos, &exc, &s,
4791 &unicode, &outpos))
4792 goto onError;
Victor Stinner785938e2011-12-11 20:09:03 +01004793 }
4794
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004795End:
4796 if (unicode_resize(&unicode, outpos) < 0)
4797 goto onError;
4798
4799 if (consumed)
4800 *consumed = s - starts;
4801
4802 Py_XDECREF(errorHandler);
4803 Py_XDECREF(exc);
4804 assert(_PyUnicode_CheckConsistency(unicode, 1));
4805 return unicode;
4806
4807onError:
4808 Py_XDECREF(errorHandler);
4809 Py_XDECREF(exc);
4810 Py_XDECREF(unicode);
4811 return NULL;
Victor Stinner785938e2011-12-11 20:09:03 +01004812}
4813
Victor Stinnerf933e1a2010-10-20 22:58:25 +00004814#ifdef __APPLE__
4815
4816/* Simplified UTF-8 decoder using surrogateescape error handler,
4817 used to decode the command line arguments on Mac OS X. */
4818
4819wchar_t*
4820_Py_DecodeUTF8_surrogateescape(const char *s, Py_ssize_t size)
4821{
Victor Stinnerf933e1a2010-10-20 22:58:25 +00004822 const char *e;
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004823 wchar_t *unicode;
4824 Py_ssize_t outpos;
Victor Stinnerf933e1a2010-10-20 22:58:25 +00004825
4826 /* Note: size will always be longer than the resulting Unicode
4827 character count */
4828 if (PY_SSIZE_T_MAX / sizeof(wchar_t) < (size + 1)) {
4829 PyErr_NoMemory();
4830 return NULL;
4831 }
4832 unicode = PyMem_Malloc((size + 1) * sizeof(wchar_t));
4833 if (!unicode)
4834 return NULL;
4835
4836 /* Unpack UTF-8 encoded data */
Victor Stinnerf933e1a2010-10-20 22:58:25 +00004837 e = s + size;
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004838 outpos = 0;
Victor Stinnerf933e1a2010-10-20 22:58:25 +00004839 while (s < e) {
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004840 Py_UCS4 ch;
Victor Stinnerf933e1a2010-10-20 22:58:25 +00004841#if SIZEOF_WCHAR_T == 4
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004842 ch = ucs4lib_utf8_decode(&s, e, (Py_UCS4 *)unicode, &outpos);
Victor Stinnerf933e1a2010-10-20 22:58:25 +00004843#else
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004844 ch = ucs2lib_utf8_decode(&s, e, (Py_UCS2 *)unicode, &outpos);
Victor Stinnerf933e1a2010-10-20 22:58:25 +00004845#endif
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004846 if (ch > 0xFF) {
4847#if SIZEOF_WCHAR_T == 4
4848 assert(0);
4849#else
4850 assert(Py_UNICODE_IS_SURROGATE(ch));
4851 /* compute and append the two surrogates: */
4852 unicode[outpos++] = (wchar_t)Py_UNICODE_HIGH_SURROGATE(ch);
4853 unicode[outpos++] = (wchar_t)Py_UNICODE_LOW_SURROGATE(ch);
4854#endif
Victor Stinnerf933e1a2010-10-20 22:58:25 +00004855 }
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004856 else {
4857 if (!ch && s == e)
4858 break;
4859 /* surrogateescape */
4860 unicode[outpos++] = 0xDC00 + (unsigned char)*s++;
4861 }
Victor Stinnerf933e1a2010-10-20 22:58:25 +00004862 }
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02004863 unicode[outpos] = L'\0';
Victor Stinnerf933e1a2010-10-20 22:58:25 +00004864 return unicode;
4865}
4866
4867#endif /* __APPLE__ */
Antoine Pitrouab868312009-01-10 15:40:25 +00004868
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02004869/* Primary internal function which creates utf8 encoded bytes objects.
4870
4871 Allocation strategy: if the string is short, convert into a stack buffer
Tim Peters602f7402002-04-27 18:03:26 +00004872 and allocate exactly as much space needed at the end. Else allocate the
4873 maximum possible needed (4 result bytes per Unicode character), and return
4874 the excess memory at the end.
Martin v. Löwis2a7ff352002-04-21 09:59:45 +00004875*/
Tim Peters7e3d9612002-04-21 03:26:37 +00004876PyObject *
Victor Stinner7931d9a2011-11-04 00:22:48 +01004877_PyUnicode_AsUTF8String(PyObject *unicode, const char *errors)
Guido van Rossumd57fd912000-03-10 22:53:23 +00004878{
Victor Stinner6099a032011-12-18 14:22:26 +01004879 enum PyUnicode_Kind kind;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02004880 void *data;
4881 Py_ssize_t size;
Marc-André Lemburgbd3be8f2002-02-07 11:33:49 +00004882
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02004883 if (!PyUnicode_Check(unicode)) {
4884 PyErr_BadArgument();
4885 return NULL;
4886 }
4887
4888 if (PyUnicode_READY(unicode) == -1)
4889 return NULL;
4890
Victor Stinnere90fe6a2011-10-01 16:48:13 +02004891 if (PyUnicode_UTF8(unicode))
4892 return PyBytes_FromStringAndSize(PyUnicode_UTF8(unicode),
4893 PyUnicode_UTF8_LENGTH(unicode));
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02004894
4895 kind = PyUnicode_KIND(unicode);
4896 data = PyUnicode_DATA(unicode);
4897 size = PyUnicode_GET_LENGTH(unicode);
4898
Benjamin Petersonead6b532011-12-20 17:23:42 -06004899 switch (kind) {
Victor Stinner6099a032011-12-18 14:22:26 +01004900 default:
4901 assert(0);
4902 case PyUnicode_1BYTE_KIND:
4903 /* the string cannot be ASCII, or PyUnicode_UTF8() would be set */
4904 assert(!PyUnicode_IS_ASCII(unicode));
4905 return ucs1lib_utf8_encoder(unicode, data, size, errors);
4906 case PyUnicode_2BYTE_KIND:
4907 return ucs2lib_utf8_encoder(unicode, data, size, errors);
4908 case PyUnicode_4BYTE_KIND:
4909 return ucs4lib_utf8_encoder(unicode, data, size, errors);
Tim Peters602f7402002-04-27 18:03:26 +00004910 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00004911}
4912
Alexander Belopolsky40018472011-02-26 01:02:56 +00004913PyObject *
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02004914PyUnicode_EncodeUTF8(const Py_UNICODE *s,
4915 Py_ssize_t size,
4916 const char *errors)
4917{
4918 PyObject *v, *unicode;
4919
4920 unicode = PyUnicode_FromUnicode(s, size);
4921 if (unicode == NULL)
4922 return NULL;
4923 v = _PyUnicode_AsUTF8String(unicode, errors);
4924 Py_DECREF(unicode);
4925 return v;
4926}
4927
4928PyObject *
Alexander Belopolsky40018472011-02-26 01:02:56 +00004929PyUnicode_AsUTF8String(PyObject *unicode)
Guido van Rossumd57fd912000-03-10 22:53:23 +00004930{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02004931 return _PyUnicode_AsUTF8String(unicode, NULL);
Guido van Rossumd57fd912000-03-10 22:53:23 +00004932}
4933
Walter Dörwald41980ca2007-08-16 21:55:45 +00004934/* --- UTF-32 Codec ------------------------------------------------------- */
4935
4936PyObject *
4937PyUnicode_DecodeUTF32(const char *s,
Benjamin Peterson29060642009-01-31 22:14:21 +00004938 Py_ssize_t size,
4939 const char *errors,
4940 int *byteorder)
Walter Dörwald41980ca2007-08-16 21:55:45 +00004941{
4942 return PyUnicode_DecodeUTF32Stateful(s, size, errors, byteorder, NULL);
4943}
4944
4945PyObject *
4946PyUnicode_DecodeUTF32Stateful(const char *s,
Benjamin Peterson29060642009-01-31 22:14:21 +00004947 Py_ssize_t size,
4948 const char *errors,
4949 int *byteorder,
4950 Py_ssize_t *consumed)
Walter Dörwald41980ca2007-08-16 21:55:45 +00004951{
4952 const char *starts = s;
4953 Py_ssize_t startinpos;
4954 Py_ssize_t endinpos;
4955 Py_ssize_t outpos;
Victor Stinner7931d9a2011-11-04 00:22:48 +01004956 PyObject *unicode;
Mark Dickinson7db923c2010-06-12 09:10:14 +00004957 const unsigned char *q, *e;
Walter Dörwald41980ca2007-08-16 21:55:45 +00004958 int bo = 0; /* assume native ordering by default */
4959 const char *errmsg = "";
Walter Dörwald41980ca2007-08-16 21:55:45 +00004960 /* Offsets from q for retrieving bytes in the right order. */
4961#ifdef BYTEORDER_IS_LITTLE_ENDIAN
4962 int iorder[] = {0, 1, 2, 3};
4963#else
4964 int iorder[] = {3, 2, 1, 0};
4965#endif
4966 PyObject *errorHandler = NULL;
4967 PyObject *exc = NULL;
Victor Stinner313a1202010-06-11 23:56:51 +00004968
Walter Dörwald41980ca2007-08-16 21:55:45 +00004969 q = (unsigned char *)s;
4970 e = q + size;
4971
4972 if (byteorder)
4973 bo = *byteorder;
4974
4975 /* Check for BOM marks (U+FEFF) in the input and adjust current
4976 byte order setting accordingly. In native mode, the leading BOM
4977 mark is skipped, in all other modes, it is copied to the output
4978 stream as-is (giving a ZWNBSP character). */
4979 if (bo == 0) {
4980 if (size >= 4) {
4981 const Py_UCS4 bom = (q[iorder[3]] << 24) | (q[iorder[2]] << 16) |
Benjamin Peterson29060642009-01-31 22:14:21 +00004982 (q[iorder[1]] << 8) | q[iorder[0]];
Walter Dörwald41980ca2007-08-16 21:55:45 +00004983#ifdef BYTEORDER_IS_LITTLE_ENDIAN
Benjamin Peterson29060642009-01-31 22:14:21 +00004984 if (bom == 0x0000FEFF) {
4985 q += 4;
4986 bo = -1;
4987 }
4988 else if (bom == 0xFFFE0000) {
4989 q += 4;
4990 bo = 1;
4991 }
Walter Dörwald41980ca2007-08-16 21:55:45 +00004992#else
Benjamin Peterson29060642009-01-31 22:14:21 +00004993 if (bom == 0x0000FEFF) {
4994 q += 4;
4995 bo = 1;
4996 }
4997 else if (bom == 0xFFFE0000) {
4998 q += 4;
4999 bo = -1;
5000 }
Walter Dörwald41980ca2007-08-16 21:55:45 +00005001#endif
Benjamin Peterson29060642009-01-31 22:14:21 +00005002 }
Walter Dörwald41980ca2007-08-16 21:55:45 +00005003 }
5004
5005 if (bo == -1) {
5006 /* force LE */
5007 iorder[0] = 0;
5008 iorder[1] = 1;
5009 iorder[2] = 2;
5010 iorder[3] = 3;
5011 }
5012 else if (bo == 1) {
5013 /* force BE */
5014 iorder[0] = 3;
5015 iorder[1] = 2;
5016 iorder[2] = 1;
5017 iorder[3] = 0;
5018 }
5019
Antoine Pitroucc0cfd32010-06-11 21:46:32 +00005020 /* This might be one to much, because of a BOM */
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005021 unicode = PyUnicode_New((size+3)/4, 127);
Antoine Pitroucc0cfd32010-06-11 21:46:32 +00005022 if (!unicode)
5023 return NULL;
5024 if (size == 0)
Victor Stinner7931d9a2011-11-04 00:22:48 +01005025 return unicode;
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005026 outpos = 0;
Antoine Pitroucc0cfd32010-06-11 21:46:32 +00005027
Walter Dörwald41980ca2007-08-16 21:55:45 +00005028 while (q < e) {
Benjamin Peterson29060642009-01-31 22:14:21 +00005029 Py_UCS4 ch;
5030 /* remaining bytes at the end? (size should be divisible by 4) */
5031 if (e-q<4) {
5032 if (consumed)
5033 break;
5034 errmsg = "truncated data";
5035 startinpos = ((const char *)q)-starts;
5036 endinpos = ((const char *)e)-starts;
5037 goto utf32Error;
5038 /* The remaining input chars are ignored if the callback
5039 chooses to skip the input */
5040 }
5041 ch = (q[iorder[3]] << 24) | (q[iorder[2]] << 16) |
5042 (q[iorder[1]] << 8) | q[iorder[0]];
Walter Dörwald41980ca2007-08-16 21:55:45 +00005043
Benjamin Peterson29060642009-01-31 22:14:21 +00005044 if (ch >= 0x110000)
5045 {
5046 errmsg = "codepoint not in range(0x110000)";
5047 startinpos = ((const char *)q)-starts;
5048 endinpos = startinpos+4;
5049 goto utf32Error;
5050 }
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005051 if (unicode_putchar(&unicode, &outpos, ch) < 0)
5052 goto onError;
Benjamin Peterson29060642009-01-31 22:14:21 +00005053 q += 4;
5054 continue;
5055 utf32Error:
Benjamin Peterson29060642009-01-31 22:14:21 +00005056 if (unicode_decode_call_errorhandler(
5057 errors, &errorHandler,
5058 "utf32", errmsg,
5059 &starts, (const char **)&e, &startinpos, &endinpos, &exc, (const char **)&q,
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005060 &unicode, &outpos))
Benjamin Peterson29060642009-01-31 22:14:21 +00005061 goto onError;
Walter Dörwald41980ca2007-08-16 21:55:45 +00005062 }
5063
5064 if (byteorder)
5065 *byteorder = bo;
5066
5067 if (consumed)
Benjamin Peterson29060642009-01-31 22:14:21 +00005068 *consumed = (const char *)q-starts;
Walter Dörwald41980ca2007-08-16 21:55:45 +00005069
5070 /* Adjust length */
Victor Stinner16e6a802011-12-12 13:24:15 +01005071 if (unicode_resize(&unicode, outpos) < 0)
Walter Dörwald41980ca2007-08-16 21:55:45 +00005072 goto onError;
5073
5074 Py_XDECREF(errorHandler);
5075 Py_XDECREF(exc);
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01005076 return unicode_result(unicode);
Walter Dörwald41980ca2007-08-16 21:55:45 +00005077
Benjamin Peterson29060642009-01-31 22:14:21 +00005078 onError:
Walter Dörwald41980ca2007-08-16 21:55:45 +00005079 Py_DECREF(unicode);
5080 Py_XDECREF(errorHandler);
5081 Py_XDECREF(exc);
5082 return NULL;
5083}
5084
5085PyObject *
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005086_PyUnicode_EncodeUTF32(PyObject *str,
5087 const char *errors,
5088 int byteorder)
Walter Dörwald41980ca2007-08-16 21:55:45 +00005089{
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005090 int kind;
5091 void *data;
5092 Py_ssize_t len;
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00005093 PyObject *v;
Walter Dörwald41980ca2007-08-16 21:55:45 +00005094 unsigned char *p;
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005095 Py_ssize_t nsize, bytesize, i;
Walter Dörwald41980ca2007-08-16 21:55:45 +00005096 /* Offsets from p for storing byte pairs in the right order. */
5097#ifdef BYTEORDER_IS_LITTLE_ENDIAN
5098 int iorder[] = {0, 1, 2, 3};
5099#else
5100 int iorder[] = {3, 2, 1, 0};
5101#endif
5102
Benjamin Peterson29060642009-01-31 22:14:21 +00005103#define STORECHAR(CH) \
5104 do { \
5105 p[iorder[3]] = ((CH) >> 24) & 0xff; \
5106 p[iorder[2]] = ((CH) >> 16) & 0xff; \
5107 p[iorder[1]] = ((CH) >> 8) & 0xff; \
5108 p[iorder[0]] = (CH) & 0xff; \
5109 p += 4; \
Walter Dörwald41980ca2007-08-16 21:55:45 +00005110 } while(0)
5111
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005112 if (!PyUnicode_Check(str)) {
5113 PyErr_BadArgument();
5114 return NULL;
5115 }
Benjamin Petersonbac79492012-01-14 13:34:47 -05005116 if (PyUnicode_READY(str) == -1)
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005117 return NULL;
5118 kind = PyUnicode_KIND(str);
5119 data = PyUnicode_DATA(str);
5120 len = PyUnicode_GET_LENGTH(str);
5121
5122 nsize = len + (byteorder == 0);
Neal Norwitz3ce5d922008-08-24 07:08:55 +00005123 bytesize = nsize * 4;
5124 if (bytesize / 4 != nsize)
Benjamin Peterson29060642009-01-31 22:14:21 +00005125 return PyErr_NoMemory();
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00005126 v = PyBytes_FromStringAndSize(NULL, bytesize);
Walter Dörwald41980ca2007-08-16 21:55:45 +00005127 if (v == NULL)
5128 return NULL;
5129
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00005130 p = (unsigned char *)PyBytes_AS_STRING(v);
Walter Dörwald41980ca2007-08-16 21:55:45 +00005131 if (byteorder == 0)
Benjamin Peterson29060642009-01-31 22:14:21 +00005132 STORECHAR(0xFEFF);
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005133 if (len == 0)
Guido van Rossum98297ee2007-11-06 21:34:58 +00005134 goto done;
Walter Dörwald41980ca2007-08-16 21:55:45 +00005135
5136 if (byteorder == -1) {
5137 /* force LE */
5138 iorder[0] = 0;
5139 iorder[1] = 1;
5140 iorder[2] = 2;
5141 iorder[3] = 3;
5142 }
5143 else if (byteorder == 1) {
5144 /* force BE */
5145 iorder[0] = 3;
5146 iorder[1] = 2;
5147 iorder[2] = 1;
5148 iorder[3] = 0;
5149 }
5150
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005151 for (i = 0; i < len; i++)
5152 STORECHAR(PyUnicode_READ(kind, data, i));
Guido van Rossum98297ee2007-11-06 21:34:58 +00005153
5154 done:
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00005155 return v;
Walter Dörwald41980ca2007-08-16 21:55:45 +00005156#undef STORECHAR
5157}
5158
Alexander Belopolsky40018472011-02-26 01:02:56 +00005159PyObject *
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005160PyUnicode_EncodeUTF32(const Py_UNICODE *s,
5161 Py_ssize_t size,
5162 const char *errors,
5163 int byteorder)
5164{
5165 PyObject *result;
5166 PyObject *tmp = PyUnicode_FromUnicode(s, size);
5167 if (tmp == NULL)
5168 return NULL;
5169 result = _PyUnicode_EncodeUTF32(tmp, errors, byteorder);
5170 Py_DECREF(tmp);
5171 return result;
5172}
5173
5174PyObject *
Alexander Belopolsky40018472011-02-26 01:02:56 +00005175PyUnicode_AsUTF32String(PyObject *unicode)
Walter Dörwald41980ca2007-08-16 21:55:45 +00005176{
Victor Stinnerb960b342011-11-20 19:12:52 +01005177 return _PyUnicode_EncodeUTF32(unicode, NULL, 0);
Walter Dörwald41980ca2007-08-16 21:55:45 +00005178}
5179
Guido van Rossumd57fd912000-03-10 22:53:23 +00005180/* --- UTF-16 Codec ------------------------------------------------------- */
5181
Tim Peters772747b2001-08-09 22:21:55 +00005182PyObject *
5183PyUnicode_DecodeUTF16(const char *s,
Benjamin Peterson29060642009-01-31 22:14:21 +00005184 Py_ssize_t size,
5185 const char *errors,
5186 int *byteorder)
Guido van Rossumd57fd912000-03-10 22:53:23 +00005187{
Walter Dörwald69652032004-09-07 20:24:22 +00005188 return PyUnicode_DecodeUTF16Stateful(s, size, errors, byteorder, NULL);
5189}
5190
5191PyObject *
5192PyUnicode_DecodeUTF16Stateful(const char *s,
Benjamin Peterson29060642009-01-31 22:14:21 +00005193 Py_ssize_t size,
5194 const char *errors,
5195 int *byteorder,
5196 Py_ssize_t *consumed)
Walter Dörwald69652032004-09-07 20:24:22 +00005197{
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005198 const char *starts = s;
Martin v. Löwis18e16552006-02-15 17:27:45 +00005199 Py_ssize_t startinpos;
5200 Py_ssize_t endinpos;
5201 Py_ssize_t outpos;
Victor Stinner7931d9a2011-11-04 00:22:48 +01005202 PyObject *unicode;
Antoine Pitrou63065d72012-05-15 23:48:04 +02005203 const unsigned char *q, *e;
Tim Peters772747b2001-08-09 22:21:55 +00005204 int bo = 0; /* assume native ordering by default */
Antoine Pitrou63065d72012-05-15 23:48:04 +02005205 int native_ordering;
Marc-André Lemburg9542f482000-07-17 18:23:13 +00005206 const char *errmsg = "";
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005207 PyObject *errorHandler = NULL;
5208 PyObject *exc = NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005209
Tim Peters772747b2001-08-09 22:21:55 +00005210 q = (unsigned char *)s;
Antoine Pitrou63065d72012-05-15 23:48:04 +02005211 e = q + size;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005212
5213 if (byteorder)
Tim Peters772747b2001-08-09 22:21:55 +00005214 bo = *byteorder;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005215
Marc-André Lemburg489b56e2001-05-21 20:30:15 +00005216 /* Check for BOM marks (U+FEFF) in the input and adjust current
5217 byte order setting accordingly. In native mode, the leading BOM
5218 mark is skipped, in all other modes, it is copied to the output
5219 stream as-is (giving a ZWNBSP character). */
Antoine Pitrou63065d72012-05-15 23:48:04 +02005220 if (bo == 0 && size >= 2) {
5221 const Py_UCS4 bom = (q[1] << 8) | q[0];
5222 if (bom == 0xFEFF) {
5223 q += 2;
5224 bo = -1;
Benjamin Peterson29060642009-01-31 22:14:21 +00005225 }
Antoine Pitrou63065d72012-05-15 23:48:04 +02005226 else if (bom == 0xFFFE) {
5227 q += 2;
5228 bo = 1;
5229 }
5230 if (byteorder)
5231 *byteorder = bo;
Marc-André Lemburg489b56e2001-05-21 20:30:15 +00005232 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00005233
Antoine Pitrou63065d72012-05-15 23:48:04 +02005234 if (q == e) {
5235 if (consumed)
5236 *consumed = size;
5237 Py_INCREF(unicode_empty);
5238 return unicode_empty;
Tim Peters772747b2001-08-09 22:21:55 +00005239 }
Antoine Pitrou63065d72012-05-15 23:48:04 +02005240
Antoine Pitrouab868312009-01-10 15:40:25 +00005241#ifdef BYTEORDER_IS_LITTLE_ENDIAN
Antoine Pitrou63065d72012-05-15 23:48:04 +02005242 native_ordering = bo <= 0;
Antoine Pitrouab868312009-01-10 15:40:25 +00005243#else
Antoine Pitrou63065d72012-05-15 23:48:04 +02005244 native_ordering = bo >= 0;
Antoine Pitrouab868312009-01-10 15:40:25 +00005245#endif
Tim Peters772747b2001-08-09 22:21:55 +00005246
Antoine Pitrou63065d72012-05-15 23:48:04 +02005247 /* Note: size will always be longer than the resulting Unicode
5248 character count */
5249 unicode = PyUnicode_New((e - q + 1) / 2, 127);
5250 if (!unicode)
5251 return NULL;
5252
5253 outpos = 0;
5254 while (1) {
5255 Py_UCS4 ch = 0;
5256 if (e - q >= 2) {
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005257 int kind = PyUnicode_KIND(unicode);
Antoine Pitrou63065d72012-05-15 23:48:04 +02005258 if (kind == PyUnicode_1BYTE_KIND) {
5259 if (PyUnicode_IS_ASCII(unicode))
5260 ch = asciilib_utf16_decode(&q, e,
5261 PyUnicode_1BYTE_DATA(unicode), &outpos,
5262 native_ordering);
5263 else
5264 ch = ucs1lib_utf16_decode(&q, e,
5265 PyUnicode_1BYTE_DATA(unicode), &outpos,
5266 native_ordering);
5267 } else if (kind == PyUnicode_2BYTE_KIND) {
5268 ch = ucs2lib_utf16_decode(&q, e,
5269 PyUnicode_2BYTE_DATA(unicode), &outpos,
5270 native_ordering);
5271 } else {
5272 assert(kind == PyUnicode_4BYTE_KIND);
5273 ch = ucs4lib_utf16_decode(&q, e,
5274 PyUnicode_4BYTE_DATA(unicode), &outpos,
5275 native_ordering);
Antoine Pitrouab868312009-01-10 15:40:25 +00005276 }
Antoine Pitrouab868312009-01-10 15:40:25 +00005277 }
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005278
Antoine Pitrou63065d72012-05-15 23:48:04 +02005279 switch (ch)
5280 {
5281 case 0:
5282 /* remaining byte at the end? (size should be even) */
5283 if (q == e || consumed)
5284 goto End;
5285 errmsg = "truncated data";
5286 startinpos = ((const char *)q) - starts;
5287 endinpos = ((const char *)e) - starts;
5288 break;
5289 /* The remaining input chars are ignored if the callback
5290 chooses to skip the input */
5291 case 1:
5292 errmsg = "unexpected end of data";
5293 startinpos = ((const char *)q) - 2 - starts;
5294 endinpos = ((const char *)e) - starts;
5295 break;
5296 case 2:
5297 errmsg = "illegal encoding";
5298 startinpos = ((const char *)q) - 2 - starts;
5299 endinpos = startinpos + 2;
5300 break;
5301 case 3:
5302 errmsg = "illegal UTF-16 surrogate";
5303 startinpos = ((const char *)q) - 4 - starts;
5304 endinpos = startinpos + 2;
5305 break;
5306 default:
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005307 if (unicode_putchar(&unicode, &outpos, ch) < 0)
5308 goto onError;
Benjamin Peterson29060642009-01-31 22:14:21 +00005309 continue;
5310 }
5311
Benjamin Peterson29060642009-01-31 22:14:21 +00005312 if (unicode_decode_call_errorhandler(
Antoine Pitrouab868312009-01-10 15:40:25 +00005313 errors,
5314 &errorHandler,
5315 "utf16", errmsg,
5316 &starts,
5317 (const char **)&e,
5318 &startinpos,
5319 &endinpos,
5320 &exc,
5321 (const char **)&q,
5322 &unicode,
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005323 &outpos))
Benjamin Peterson29060642009-01-31 22:14:21 +00005324 goto onError;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005325 }
5326
Antoine Pitrou63065d72012-05-15 23:48:04 +02005327End:
Walter Dörwald69652032004-09-07 20:24:22 +00005328 if (consumed)
Benjamin Peterson29060642009-01-31 22:14:21 +00005329 *consumed = (const char *)q-starts;
Walter Dörwald69652032004-09-07 20:24:22 +00005330
Guido van Rossumd57fd912000-03-10 22:53:23 +00005331 /* Adjust length */
Victor Stinner16e6a802011-12-12 13:24:15 +01005332 if (unicode_resize(&unicode, outpos) < 0)
Guido van Rossumd57fd912000-03-10 22:53:23 +00005333 goto onError;
5334
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005335 Py_XDECREF(errorHandler);
5336 Py_XDECREF(exc);
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01005337 return unicode_result(unicode);
Guido van Rossumd57fd912000-03-10 22:53:23 +00005338
Benjamin Peterson29060642009-01-31 22:14:21 +00005339 onError:
Guido van Rossumd57fd912000-03-10 22:53:23 +00005340 Py_DECREF(unicode);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005341 Py_XDECREF(errorHandler);
5342 Py_XDECREF(exc);
Guido van Rossumd57fd912000-03-10 22:53:23 +00005343 return NULL;
5344}
5345
Tim Peters772747b2001-08-09 22:21:55 +00005346PyObject *
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005347_PyUnicode_EncodeUTF16(PyObject *str,
5348 const char *errors,
5349 int byteorder)
Guido van Rossumd57fd912000-03-10 22:53:23 +00005350{
Antoine Pitrou27f6a3b2012-06-15 22:15:23 +02005351 enum PyUnicode_Kind kind;
5352 const void *data;
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005353 Py_ssize_t len;
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00005354 PyObject *v;
Antoine Pitrou27f6a3b2012-06-15 22:15:23 +02005355 unsigned short *out;
5356 Py_ssize_t bytesize;
5357 Py_ssize_t pairs;
5358#ifdef WORDS_BIGENDIAN
5359 int native_ordering = byteorder >= 0;
Tim Peters772747b2001-08-09 22:21:55 +00005360#else
Antoine Pitrou27f6a3b2012-06-15 22:15:23 +02005361 int native_ordering = byteorder <= 0;
Tim Peters772747b2001-08-09 22:21:55 +00005362#endif
5363
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005364 if (!PyUnicode_Check(str)) {
5365 PyErr_BadArgument();
5366 return NULL;
5367 }
Benjamin Petersonbac79492012-01-14 13:34:47 -05005368 if (PyUnicode_READY(str) == -1)
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005369 return NULL;
5370 kind = PyUnicode_KIND(str);
5371 data = PyUnicode_DATA(str);
5372 len = PyUnicode_GET_LENGTH(str);
Victor Stinner0e368262011-11-10 20:12:49 +01005373
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005374 pairs = 0;
Antoine Pitrou27f6a3b2012-06-15 22:15:23 +02005375 if (kind == PyUnicode_4BYTE_KIND) {
5376 const Py_UCS4 *in = (const Py_UCS4 *)data;
5377 const Py_UCS4 *end = in + len;
5378 while (in < end)
5379 if (*in++ >= 0x10000)
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005380 pairs++;
Antoine Pitrou27f6a3b2012-06-15 22:15:23 +02005381 }
5382 if (len > PY_SSIZE_T_MAX / 2 - pairs - (byteorder == 0))
Benjamin Peterson29060642009-01-31 22:14:21 +00005383 return PyErr_NoMemory();
Antoine Pitrou27f6a3b2012-06-15 22:15:23 +02005384 bytesize = (len + pairs + (byteorder == 0)) * 2;
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00005385 v = PyBytes_FromStringAndSize(NULL, bytesize);
Guido van Rossumd57fd912000-03-10 22:53:23 +00005386 if (v == NULL)
5387 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005388
Antoine Pitrou27f6a3b2012-06-15 22:15:23 +02005389 /* output buffer is 2-bytes aligned */
Antoine Pitrouca8aa4a2012-09-20 20:56:47 +02005390 assert(_Py_IS_ALIGNED(PyBytes_AS_STRING(v), 2));
Antoine Pitrou27f6a3b2012-06-15 22:15:23 +02005391 out = (unsigned short *)PyBytes_AS_STRING(v);
Guido van Rossumd57fd912000-03-10 22:53:23 +00005392 if (byteorder == 0)
Antoine Pitrou27f6a3b2012-06-15 22:15:23 +02005393 *out++ = 0xFEFF;
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005394 if (len == 0)
Guido van Rossum98297ee2007-11-06 21:34:58 +00005395 goto done;
Tim Peters772747b2001-08-09 22:21:55 +00005396
Antoine Pitrou27f6a3b2012-06-15 22:15:23 +02005397 switch (kind) {
5398 case PyUnicode_1BYTE_KIND: {
5399 ucs1lib_utf16_encode(out, (const Py_UCS1 *)data, len, native_ordering);
5400 break;
Tim Peters772747b2001-08-09 22:21:55 +00005401 }
Antoine Pitrou27f6a3b2012-06-15 22:15:23 +02005402 case PyUnicode_2BYTE_KIND: {
5403 ucs2lib_utf16_encode(out, (const Py_UCS2 *)data, len, native_ordering);
5404 break;
Tim Peters772747b2001-08-09 22:21:55 +00005405 }
Antoine Pitrou27f6a3b2012-06-15 22:15:23 +02005406 case PyUnicode_4BYTE_KIND: {
5407 ucs4lib_utf16_encode(out, (const Py_UCS4 *)data, len, native_ordering);
5408 break;
5409 }
5410 default:
5411 assert(0);
Martin v. Löwis0ba70cc2001-06-26 22:22:37 +00005412 }
Guido van Rossum98297ee2007-11-06 21:34:58 +00005413
5414 done:
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00005415 return v;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005416}
5417
Alexander Belopolsky40018472011-02-26 01:02:56 +00005418PyObject *
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005419PyUnicode_EncodeUTF16(const Py_UNICODE *s,
5420 Py_ssize_t size,
5421 const char *errors,
5422 int byteorder)
5423{
5424 PyObject *result;
5425 PyObject *tmp = PyUnicode_FromUnicode(s, size);
5426 if (tmp == NULL)
5427 return NULL;
5428 result = _PyUnicode_EncodeUTF16(tmp, errors, byteorder);
5429 Py_DECREF(tmp);
5430 return result;
5431}
5432
5433PyObject *
Alexander Belopolsky40018472011-02-26 01:02:56 +00005434PyUnicode_AsUTF16String(PyObject *unicode)
Guido van Rossumd57fd912000-03-10 22:53:23 +00005435{
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005436 return _PyUnicode_EncodeUTF16(unicode, NULL, 0);
Guido van Rossumd57fd912000-03-10 22:53:23 +00005437}
5438
5439/* --- Unicode Escape Codec ----------------------------------------------- */
5440
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02005441/* Helper function for PyUnicode_DecodeUnicodeEscape, determines
5442 if all the escapes in the string make it still a valid ASCII string.
5443 Returns -1 if any escapes were found which cause the string to
5444 pop out of ASCII range. Otherwise returns the length of the
5445 required buffer to hold the string.
5446 */
Antoine Pitrou53bb5482011-10-10 23:49:24 +02005447static Py_ssize_t
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02005448length_of_escaped_ascii_string(const char *s, Py_ssize_t size)
5449{
5450 const unsigned char *p = (const unsigned char *)s;
5451 const unsigned char *end = p + size;
5452 Py_ssize_t length = 0;
5453
5454 if (size < 0)
5455 return -1;
5456
5457 for (; p < end; ++p) {
5458 if (*p > 127) {
5459 /* Non-ASCII */
5460 return -1;
5461 }
5462 else if (*p != '\\') {
5463 /* Normal character */
5464 ++length;
5465 }
5466 else {
5467 /* Backslash-escape, check next char */
5468 ++p;
5469 /* Escape sequence reaches till end of string or
5470 non-ASCII follow-up. */
5471 if (p >= end || *p > 127)
5472 return -1;
5473 switch (*p) {
5474 case '\n':
5475 /* backslash + \n result in zero characters */
5476 break;
5477 case '\\': case '\'': case '\"':
5478 case 'b': case 'f': case 't':
5479 case 'n': case 'r': case 'v': case 'a':
5480 ++length;
5481 break;
5482 case '0': case '1': case '2': case '3':
5483 case '4': case '5': case '6': case '7':
5484 case 'x': case 'u': case 'U': case 'N':
5485 /* these do not guarantee ASCII characters */
5486 return -1;
5487 default:
5488 /* count the backslash + the other character */
5489 length += 2;
5490 }
5491 }
5492 }
5493 return length;
5494}
5495
Fredrik Lundh06d12682001-01-24 07:59:11 +00005496static _PyUnicode_Name_CAPI *ucnhash_CAPI = NULL;
Marc-André Lemburg0f774e32000-06-28 16:43:35 +00005497
Alexander Belopolsky40018472011-02-26 01:02:56 +00005498PyObject *
5499PyUnicode_DecodeUnicodeEscape(const char *s,
Ezio Melotti2aa2b3b2011-09-29 00:58:57 +03005500 Py_ssize_t size,
Victor Stinnerc17f5402011-09-29 00:16:58 +02005501 const char *errors)
Guido van Rossumd57fd912000-03-10 22:53:23 +00005502{
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005503 const char *starts = s;
Martin v. Löwis18e16552006-02-15 17:27:45 +00005504 Py_ssize_t startinpos;
5505 Py_ssize_t endinpos;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02005506 int j;
Victor Stinner7931d9a2011-11-04 00:22:48 +01005507 PyObject *v;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005508 const char *end;
Fredrik Lundhccc74732001-02-18 22:13:49 +00005509 char* message;
5510 Py_UCS4 chr = 0xffffffff; /* in case 'getcode' messes up */
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005511 PyObject *errorHandler = NULL;
5512 PyObject *exc = NULL;
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005513 Py_ssize_t len;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02005514 Py_ssize_t i;
Fredrik Lundhccc74732001-02-18 22:13:49 +00005515
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005516 len = length_of_escaped_ascii_string(s, size);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02005517
5518 /* After length_of_escaped_ascii_string() there are two alternatives,
5519 either the string is pure ASCII with named escapes like \n, etc.
5520 and we determined it's exact size (common case)
5521 or it contains \x, \u, ... escape sequences. then we create a
5522 legacy wchar string and resize it at the end of this function. */
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005523 if (len >= 0) {
5524 v = PyUnicode_New(len, 127);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02005525 if (!v)
5526 goto onError;
5527 assert(PyUnicode_KIND(v) == PyUnicode_1BYTE_KIND);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02005528 }
5529 else {
5530 /* Escaped strings will always be longer than the resulting
5531 Unicode string, so we start with size here and then reduce the
5532 length after conversion to the true value.
5533 (but if the error callback returns a long replacement string
5534 we'll have to allocate more space) */
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005535 v = PyUnicode_New(size, 127);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02005536 if (!v)
5537 goto onError;
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005538 len = size;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02005539 }
5540
Guido van Rossumd57fd912000-03-10 22:53:23 +00005541 if (size == 0)
Victor Stinner7931d9a2011-11-04 00:22:48 +01005542 return v;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02005543 i = 0;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005544 end = s + size;
Fredrik Lundhccc74732001-02-18 22:13:49 +00005545
Guido van Rossumd57fd912000-03-10 22:53:23 +00005546 while (s < end) {
5547 unsigned char c;
Victor Stinner24729f32011-11-10 20:31:37 +01005548 Py_UCS4 x;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005549 int digits;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005550
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005551 /* The only case in which i == ascii_length is a backslash
5552 followed by a newline. */
5553 assert(i <= len);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02005554
Guido van Rossumd57fd912000-03-10 22:53:23 +00005555 /* Non-escape characters are interpreted as Unicode ordinals */
5556 if (*s != '\\') {
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005557 if (unicode_putchar(&v, &i, (unsigned char) *s++) < 0)
5558 goto onError;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005559 continue;
5560 }
5561
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005562 startinpos = s-starts;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005563 /* \ - Escapes */
5564 s++;
Guido van Rossum8ce8a782007-11-01 19:42:39 +00005565 c = *s++;
5566 if (s > end)
5567 c = '\0'; /* Invalid after \ */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02005568
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005569 /* The only case in which i == ascii_length is a backslash
5570 followed by a newline. */
5571 assert(i < len || (i == len && c == '\n'));
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02005572
Guido van Rossum8ce8a782007-11-01 19:42:39 +00005573 switch (c) {
Guido van Rossumd57fd912000-03-10 22:53:23 +00005574
Benjamin Peterson29060642009-01-31 22:14:21 +00005575 /* \x escapes */
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005576#define WRITECHAR(ch) \
5577 do { \
5578 if (unicode_putchar(&v, &i, ch) < 0) \
5579 goto onError; \
5580 }while(0)
5581
Guido van Rossumd57fd912000-03-10 22:53:23 +00005582 case '\n': break;
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005583 case '\\': WRITECHAR('\\'); break;
5584 case '\'': WRITECHAR('\''); break;
5585 case '\"': WRITECHAR('\"'); break;
5586 case 'b': WRITECHAR('\b'); break;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02005587 /* FF */
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005588 case 'f': WRITECHAR('\014'); break;
5589 case 't': WRITECHAR('\t'); break;
5590 case 'n': WRITECHAR('\n'); break;
5591 case 'r': WRITECHAR('\r'); break;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02005592 /* VT */
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005593 case 'v': WRITECHAR('\013'); break;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02005594 /* BEL, not classic C */
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005595 case 'a': WRITECHAR('\007'); break;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005596
Benjamin Peterson29060642009-01-31 22:14:21 +00005597 /* \OOO (octal) escapes */
Guido van Rossumd57fd912000-03-10 22:53:23 +00005598 case '0': case '1': case '2': case '3':
5599 case '4': case '5': case '6': case '7':
Guido van Rossum0e4f6572000-05-01 21:27:20 +00005600 x = s[-1] - '0';
Guido van Rossum8ce8a782007-11-01 19:42:39 +00005601 if (s < end && '0' <= *s && *s <= '7') {
Guido van Rossum0e4f6572000-05-01 21:27:20 +00005602 x = (x<<3) + *s++ - '0';
Guido van Rossum8ce8a782007-11-01 19:42:39 +00005603 if (s < end && '0' <= *s && *s <= '7')
Guido van Rossum0e4f6572000-05-01 21:27:20 +00005604 x = (x<<3) + *s++ - '0';
Guido van Rossumd57fd912000-03-10 22:53:23 +00005605 }
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005606 WRITECHAR(x);
Guido van Rossumd57fd912000-03-10 22:53:23 +00005607 break;
5608
Benjamin Peterson29060642009-01-31 22:14:21 +00005609 /* hex escapes */
5610 /* \xXX */
Guido van Rossumd57fd912000-03-10 22:53:23 +00005611 case 'x':
Fredrik Lundhccc74732001-02-18 22:13:49 +00005612 digits = 2;
5613 message = "truncated \\xXX escape";
5614 goto hexescape;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005615
Benjamin Peterson29060642009-01-31 22:14:21 +00005616 /* \uXXXX */
Guido van Rossumd57fd912000-03-10 22:53:23 +00005617 case 'u':
Fredrik Lundhccc74732001-02-18 22:13:49 +00005618 digits = 4;
5619 message = "truncated \\uXXXX escape";
5620 goto hexescape;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005621
Benjamin Peterson29060642009-01-31 22:14:21 +00005622 /* \UXXXXXXXX */
Fredrik Lundhdf846752000-09-03 11:29:49 +00005623 case 'U':
Fredrik Lundhccc74732001-02-18 22:13:49 +00005624 digits = 8;
5625 message = "truncated \\UXXXXXXXX escape";
5626 hexescape:
5627 chr = 0;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005628 if (s+digits>end) {
5629 endinpos = size;
5630 if (unicode_decode_call_errorhandler(
Benjamin Peterson29060642009-01-31 22:14:21 +00005631 errors, &errorHandler,
5632 "unicodeescape", "end of string in escape sequence",
5633 &starts, &end, &startinpos, &endinpos, &exc, &s,
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005634 &v, &i))
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005635 goto onError;
5636 goto nextByte;
5637 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02005638 for (j = 0; j < digits; ++j) {
5639 c = (unsigned char) s[j];
David Malcolm96960882010-11-05 17:23:41 +00005640 if (!Py_ISXDIGIT(c)) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02005641 endinpos = (s+j+1)-starts;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005642 if (unicode_decode_call_errorhandler(
Benjamin Peterson29060642009-01-31 22:14:21 +00005643 errors, &errorHandler,
5644 "unicodeescape", message,
5645 &starts, &end, &startinpos, &endinpos, &exc, &s,
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005646 &v, &i))
Fredrik Lundhdf846752000-09-03 11:29:49 +00005647 goto onError;
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005648 len = PyUnicode_GET_LENGTH(v);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005649 goto nextByte;
Fredrik Lundhdf846752000-09-03 11:29:49 +00005650 }
5651 chr = (chr<<4) & ~0xF;
5652 if (c >= '0' && c <= '9')
5653 chr += c - '0';
5654 else if (c >= 'a' && c <= 'f')
5655 chr += 10 + c - 'a';
5656 else
5657 chr += 10 + c - 'A';
5658 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02005659 s += j;
Jeremy Hylton504de6b2003-10-06 05:08:26 +00005660 if (chr == 0xffffffff && PyErr_Occurred())
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005661 /* _decoding_error will have already written into the
5662 target buffer. */
5663 break;
Fredrik Lundhccc74732001-02-18 22:13:49 +00005664 store:
Fredrik Lundhdf846752000-09-03 11:29:49 +00005665 /* when we get here, chr is a 32-bit unicode character */
Victor Stinner8faf8212011-12-08 22:14:11 +01005666 if (chr <= MAX_UNICODE) {
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005667 WRITECHAR(chr);
Fredrik Lundhdf846752000-09-03 11:29:49 +00005668 } else {
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005669 endinpos = s-starts;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005670 if (unicode_decode_call_errorhandler(
Benjamin Peterson29060642009-01-31 22:14:21 +00005671 errors, &errorHandler,
5672 "unicodeescape", "illegal Unicode character",
5673 &starts, &end, &startinpos, &endinpos, &exc, &s,
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005674 &v, &i))
Fredrik Lundhdf846752000-09-03 11:29:49 +00005675 goto onError;
5676 }
Fredrik Lundhccc74732001-02-18 22:13:49 +00005677 break;
5678
Benjamin Peterson29060642009-01-31 22:14:21 +00005679 /* \N{name} */
Fredrik Lundhccc74732001-02-18 22:13:49 +00005680 case 'N':
5681 message = "malformed \\N character escape";
5682 if (ucnhash_CAPI == NULL) {
5683 /* load the unicode data module */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02005684 ucnhash_CAPI = (_PyUnicode_Name_CAPI *)PyCapsule_Import(
5685 PyUnicodeData_CAPSULE_NAME, 1);
Fredrik Lundhccc74732001-02-18 22:13:49 +00005686 if (ucnhash_CAPI == NULL)
5687 goto ucnhashError;
5688 }
5689 if (*s == '{') {
5690 const char *start = s+1;
5691 /* look for the closing brace */
5692 while (*s != '}' && s < end)
5693 s++;
5694 if (s > start && s < end && *s == '}') {
5695 /* found a name. look it up in the unicode database */
5696 message = "unknown Unicode character name";
5697 s++;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02005698 if (ucnhash_CAPI->getcode(NULL, start, (int)(s-start-1),
Ezio Melotti931b8aa2011-10-21 21:57:36 +03005699 &chr, 0))
Fredrik Lundhccc74732001-02-18 22:13:49 +00005700 goto store;
5701 }
5702 }
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005703 endinpos = s-starts;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005704 if (unicode_decode_call_errorhandler(
Benjamin Peterson29060642009-01-31 22:14:21 +00005705 errors, &errorHandler,
5706 "unicodeescape", message,
5707 &starts, &end, &startinpos, &endinpos, &exc, &s,
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005708 &v, &i))
Fredrik Lundhccc74732001-02-18 22:13:49 +00005709 goto onError;
Fredrik Lundhccc74732001-02-18 22:13:49 +00005710 break;
5711
5712 default:
Walter Dörwald8c077222002-03-25 11:16:18 +00005713 if (s > end) {
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005714 message = "\\ at end of string";
5715 s--;
5716 endinpos = s-starts;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005717 if (unicode_decode_call_errorhandler(
Benjamin Peterson29060642009-01-31 22:14:21 +00005718 errors, &errorHandler,
5719 "unicodeescape", message,
5720 &starts, &end, &startinpos, &endinpos, &exc, &s,
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005721 &v, &i))
Walter Dörwald8c077222002-03-25 11:16:18 +00005722 goto onError;
5723 }
5724 else {
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005725 WRITECHAR('\\');
5726 WRITECHAR(s[-1]);
Walter Dörwald8c077222002-03-25 11:16:18 +00005727 }
Fredrik Lundhccc74732001-02-18 22:13:49 +00005728 break;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005729 }
Benjamin Peterson29060642009-01-31 22:14:21 +00005730 nextByte:
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005731 ;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005732 }
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005733#undef WRITECHAR
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02005734
Victor Stinner16e6a802011-12-12 13:24:15 +01005735 if (unicode_resize(&v, i) < 0)
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005736 goto onError;
Walter Dörwaldd4ade082003-08-15 15:00:26 +00005737 Py_XDECREF(errorHandler);
5738 Py_XDECREF(exc);
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01005739 return unicode_result(v);
Walter Dörwald8c077222002-03-25 11:16:18 +00005740
Benjamin Peterson29060642009-01-31 22:14:21 +00005741 ucnhashError:
Fredrik Lundh06d12682001-01-24 07:59:11 +00005742 PyErr_SetString(
5743 PyExc_UnicodeError,
5744 "\\N escapes not supported (can't load unicodedata module)"
5745 );
Hye-Shik Chang4af5c8c2006-03-07 15:39:21 +00005746 Py_XDECREF(v);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005747 Py_XDECREF(errorHandler);
5748 Py_XDECREF(exc);
Fredrik Lundhf6056062001-01-20 11:15:25 +00005749 return NULL;
5750
Benjamin Peterson29060642009-01-31 22:14:21 +00005751 onError:
Guido van Rossumd57fd912000-03-10 22:53:23 +00005752 Py_XDECREF(v);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005753 Py_XDECREF(errorHandler);
5754 Py_XDECREF(exc);
Guido van Rossumd57fd912000-03-10 22:53:23 +00005755 return NULL;
5756}
5757
5758/* Return a Unicode-Escape string version of the Unicode object.
5759
5760 If quotes is true, the string is enclosed in u"" or u'' quotes as
5761 appropriate.
5762
5763*/
5764
Alexander Belopolsky40018472011-02-26 01:02:56 +00005765PyObject *
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005766PyUnicode_AsUnicodeEscapeString(PyObject *unicode)
Guido van Rossumd57fd912000-03-10 22:53:23 +00005767{
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005768 Py_ssize_t i, len;
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00005769 PyObject *repr;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005770 char *p;
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005771 int kind;
5772 void *data;
5773 Py_ssize_t expandsize = 0;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005774
Thomas Wouters89f507f2006-12-13 04:49:30 +00005775 /* Initial allocation is based on the longest-possible unichr
5776 escape.
5777
5778 In wide (UTF-32) builds '\U00xxxxxx' is 10 chars per source
5779 unichr, so in this case it's the longest unichr escape. In
5780 narrow (UTF-16) builds this is five chars per source unichr
5781 since there are two unichrs in the surrogate pair, so in narrow
5782 (UTF-16) builds it's not the longest unichr escape.
5783
5784 In wide or narrow builds '\uxxxx' is 6 chars per source unichr,
5785 so in the narrow (UTF-16) build case it's the longest unichr
5786 escape.
5787 */
5788
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005789 if (!PyUnicode_Check(unicode)) {
5790 PyErr_BadArgument();
5791 return NULL;
5792 }
Benjamin Petersonbac79492012-01-14 13:34:47 -05005793 if (PyUnicode_READY(unicode) == -1)
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005794 return NULL;
5795 len = PyUnicode_GET_LENGTH(unicode);
5796 kind = PyUnicode_KIND(unicode);
5797 data = PyUnicode_DATA(unicode);
Benjamin Petersonead6b532011-12-20 17:23:42 -06005798 switch (kind) {
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005799 case PyUnicode_1BYTE_KIND: expandsize = 4; break;
5800 case PyUnicode_2BYTE_KIND: expandsize = 6; break;
5801 case PyUnicode_4BYTE_KIND: expandsize = 10; break;
5802 }
5803
5804 if (len == 0)
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00005805 return PyBytes_FromStringAndSize(NULL, 0);
5806
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005807 if (len > (PY_SSIZE_T_MAX - 2 - 1) / expandsize)
Benjamin Peterson29060642009-01-31 22:14:21 +00005808 return PyErr_NoMemory();
Neal Norwitz3ce5d922008-08-24 07:08:55 +00005809
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00005810 repr = PyBytes_FromStringAndSize(NULL,
Benjamin Peterson29060642009-01-31 22:14:21 +00005811 2
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005812 + expandsize*len
Benjamin Peterson29060642009-01-31 22:14:21 +00005813 + 1);
Guido van Rossumd57fd912000-03-10 22:53:23 +00005814 if (repr == NULL)
5815 return NULL;
5816
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00005817 p = PyBytes_AS_STRING(repr);
Guido van Rossumd57fd912000-03-10 22:53:23 +00005818
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005819 for (i = 0; i < len; i++) {
Victor Stinner3326cb62011-11-10 20:15:25 +01005820 Py_UCS4 ch = PyUnicode_READ(kind, data, i);
Marc-André Lemburg80d1dd52001-07-25 16:05:59 +00005821
Walter Dörwald79e913e2007-05-12 11:08:06 +00005822 /* Escape backslashes */
5823 if (ch == '\\') {
Guido van Rossumd57fd912000-03-10 22:53:23 +00005824 *p++ = '\\';
5825 *p++ = (char) ch;
Walter Dörwald79e913e2007-05-12 11:08:06 +00005826 continue;
Tim Petersced69f82003-09-16 20:30:58 +00005827 }
Marc-André Lemburg80d1dd52001-07-25 16:05:59 +00005828
Martin v. Löwis0ba70cc2001-06-26 22:22:37 +00005829 /* Map 21-bit characters to '\U00xxxxxx' */
5830 else if (ch >= 0x10000) {
Victor Stinner8faf8212011-12-08 22:14:11 +01005831 assert(ch <= MAX_UNICODE);
Martin v. Löwis0ba70cc2001-06-26 22:22:37 +00005832 *p++ = '\\';
5833 *p++ = 'U';
Victor Stinnerf5cff562011-10-14 02:13:11 +02005834 *p++ = Py_hexdigits[(ch >> 28) & 0x0000000F];
5835 *p++ = Py_hexdigits[(ch >> 24) & 0x0000000F];
5836 *p++ = Py_hexdigits[(ch >> 20) & 0x0000000F];
5837 *p++ = Py_hexdigits[(ch >> 16) & 0x0000000F];
5838 *p++ = Py_hexdigits[(ch >> 12) & 0x0000000F];
5839 *p++ = Py_hexdigits[(ch >> 8) & 0x0000000F];
5840 *p++ = Py_hexdigits[(ch >> 4) & 0x0000000F];
5841 *p++ = Py_hexdigits[ch & 0x0000000F];
Benjamin Peterson29060642009-01-31 22:14:21 +00005842 continue;
Martin v. Löwis0ba70cc2001-06-26 22:22:37 +00005843 }
Marc-André Lemburg6c6bfb72001-07-20 17:39:11 +00005844
Guido van Rossumd57fd912000-03-10 22:53:23 +00005845 /* Map 16-bit characters to '\uxxxx' */
Marc-André Lemburg6c6bfb72001-07-20 17:39:11 +00005846 if (ch >= 256) {
Guido van Rossumd57fd912000-03-10 22:53:23 +00005847 *p++ = '\\';
5848 *p++ = 'u';
Victor Stinnerf5cff562011-10-14 02:13:11 +02005849 *p++ = Py_hexdigits[(ch >> 12) & 0x000F];
5850 *p++ = Py_hexdigits[(ch >> 8) & 0x000F];
5851 *p++ = Py_hexdigits[(ch >> 4) & 0x000F];
5852 *p++ = Py_hexdigits[ch & 0x000F];
Guido van Rossumd57fd912000-03-10 22:53:23 +00005853 }
Marc-André Lemburg80d1dd52001-07-25 16:05:59 +00005854
Ka-Ping Yeefa004ad2001-01-24 17:19:08 +00005855 /* Map special whitespace to '\t', \n', '\r' */
5856 else if (ch == '\t') {
5857 *p++ = '\\';
5858 *p++ = 't';
5859 }
5860 else if (ch == '\n') {
5861 *p++ = '\\';
5862 *p++ = 'n';
5863 }
5864 else if (ch == '\r') {
5865 *p++ = '\\';
5866 *p++ = 'r';
5867 }
Marc-André Lemburg80d1dd52001-07-25 16:05:59 +00005868
Ka-Ping Yeefa004ad2001-01-24 17:19:08 +00005869 /* Map non-printable US ASCII to '\xhh' */
Marc-André Lemburg11326de2001-11-28 12:56:20 +00005870 else if (ch < ' ' || ch >= 0x7F) {
Guido van Rossumd57fd912000-03-10 22:53:23 +00005871 *p++ = '\\';
Ka-Ping Yeefa004ad2001-01-24 17:19:08 +00005872 *p++ = 'x';
Victor Stinnerf5cff562011-10-14 02:13:11 +02005873 *p++ = Py_hexdigits[(ch >> 4) & 0x000F];
5874 *p++ = Py_hexdigits[ch & 0x000F];
Tim Petersced69f82003-09-16 20:30:58 +00005875 }
Marc-André Lemburg80d1dd52001-07-25 16:05:59 +00005876
Guido van Rossumd57fd912000-03-10 22:53:23 +00005877 /* Copy everything else as-is */
5878 else
5879 *p++ = (char) ch;
5880 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00005881
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00005882 assert(p - PyBytes_AS_STRING(repr) > 0);
5883 if (_PyBytes_Resize(&repr, p - PyBytes_AS_STRING(repr)) < 0)
5884 return NULL;
5885 return repr;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005886}
5887
Alexander Belopolsky40018472011-02-26 01:02:56 +00005888PyObject *
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005889PyUnicode_EncodeUnicodeEscape(const Py_UNICODE *s,
5890 Py_ssize_t size)
Guido van Rossumd57fd912000-03-10 22:53:23 +00005891{
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005892 PyObject *result;
5893 PyObject *tmp = PyUnicode_FromUnicode(s, size);
5894 if (tmp == NULL)
Guido van Rossumd57fd912000-03-10 22:53:23 +00005895 return NULL;
Martin v. Löwis1db7c132011-11-10 18:24:32 +01005896 result = PyUnicode_AsUnicodeEscapeString(tmp);
5897 Py_DECREF(tmp);
5898 return result;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005899}
5900
5901/* --- Raw Unicode Escape Codec ------------------------------------------- */
5902
Alexander Belopolsky40018472011-02-26 01:02:56 +00005903PyObject *
5904PyUnicode_DecodeRawUnicodeEscape(const char *s,
Ezio Melotti2aa2b3b2011-09-29 00:58:57 +03005905 Py_ssize_t size,
5906 const char *errors)
Guido van Rossumd57fd912000-03-10 22:53:23 +00005907{
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005908 const char *starts = s;
Martin v. Löwis18e16552006-02-15 17:27:45 +00005909 Py_ssize_t startinpos;
5910 Py_ssize_t endinpos;
5911 Py_ssize_t outpos;
Victor Stinner7931d9a2011-11-04 00:22:48 +01005912 PyObject *v;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005913 const char *end;
5914 const char *bs;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005915 PyObject *errorHandler = NULL;
5916 PyObject *exc = NULL;
Tim Petersced69f82003-09-16 20:30:58 +00005917
Guido van Rossumd57fd912000-03-10 22:53:23 +00005918 /* Escaped strings will always be longer than the resulting
5919 Unicode string, so we start with size here and then reduce the
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005920 length after conversion to the true value. (But decoding error
5921 handler might have to resize the string) */
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005922 v = PyUnicode_New(size, 127);
Guido van Rossumd57fd912000-03-10 22:53:23 +00005923 if (v == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00005924 goto onError;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005925 if (size == 0)
Victor Stinner7931d9a2011-11-04 00:22:48 +01005926 return v;
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005927 outpos = 0;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005928 end = s + size;
5929 while (s < end) {
Benjamin Peterson29060642009-01-31 22:14:21 +00005930 unsigned char c;
5931 Py_UCS4 x;
5932 int i;
Martin v. Löwis9a3a9f72003-05-18 12:31:09 +00005933 int count;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005934
Benjamin Peterson29060642009-01-31 22:14:21 +00005935 /* Non-escape characters are interpreted as Unicode ordinals */
5936 if (*s != '\\') {
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005937 if (unicode_putchar(&v, &outpos, (unsigned char)*s++) < 0)
5938 goto onError;
Benjamin Peterson29060642009-01-31 22:14:21 +00005939 continue;
Benjamin Peterson14339b62009-01-31 16:36:08 +00005940 }
Benjamin Peterson29060642009-01-31 22:14:21 +00005941 startinpos = s-starts;
5942
5943 /* \u-escapes are only interpreted iff the number of leading
5944 backslashes if odd */
5945 bs = s;
5946 for (;s < end;) {
5947 if (*s != '\\')
5948 break;
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005949 if (unicode_putchar(&v, &outpos, (unsigned char)*s++) < 0)
5950 goto onError;
Benjamin Peterson29060642009-01-31 22:14:21 +00005951 }
5952 if (((s - bs) & 1) == 0 ||
5953 s >= end ||
5954 (*s != 'u' && *s != 'U')) {
5955 continue;
5956 }
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005957 outpos--;
Benjamin Peterson29060642009-01-31 22:14:21 +00005958 count = *s=='u' ? 4 : 8;
5959 s++;
5960
5961 /* \uXXXX with 4 hex digits, \Uxxxxxxxx with 8 */
Benjamin Peterson29060642009-01-31 22:14:21 +00005962 for (x = 0, i = 0; i < count; ++i, ++s) {
5963 c = (unsigned char)*s;
David Malcolm96960882010-11-05 17:23:41 +00005964 if (!Py_ISXDIGIT(c)) {
Benjamin Peterson29060642009-01-31 22:14:21 +00005965 endinpos = s-starts;
5966 if (unicode_decode_call_errorhandler(
5967 errors, &errorHandler,
5968 "rawunicodeescape", "truncated \\uXXXX",
5969 &starts, &end, &startinpos, &endinpos, &exc, &s,
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005970 &v, &outpos))
Benjamin Peterson29060642009-01-31 22:14:21 +00005971 goto onError;
5972 goto nextByte;
5973 }
5974 x = (x<<4) & ~0xF;
5975 if (c >= '0' && c <= '9')
5976 x += c - '0';
5977 else if (c >= 'a' && c <= 'f')
5978 x += 10 + c - 'a';
5979 else
5980 x += 10 + c - 'A';
5981 }
Victor Stinner8faf8212011-12-08 22:14:11 +01005982 if (x <= MAX_UNICODE) {
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005983 if (unicode_putchar(&v, &outpos, x) < 0)
5984 goto onError;
Christian Heimesfe337bf2008-03-23 21:54:12 +00005985 } else {
5986 endinpos = s-starts;
Martin v. Löwis9a3a9f72003-05-18 12:31:09 +00005987 if (unicode_decode_call_errorhandler(
5988 errors, &errorHandler,
5989 "rawunicodeescape", "\\Uxxxxxxxx out of range",
Benjamin Peterson29060642009-01-31 22:14:21 +00005990 &starts, &end, &startinpos, &endinpos, &exc, &s,
Martin v. Löwise9b11c12011-11-08 17:35:34 +01005991 &v, &outpos))
Benjamin Peterson29060642009-01-31 22:14:21 +00005992 goto onError;
Martin v. Löwis9a3a9f72003-05-18 12:31:09 +00005993 }
Benjamin Peterson29060642009-01-31 22:14:21 +00005994 nextByte:
5995 ;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005996 }
Victor Stinner16e6a802011-12-12 13:24:15 +01005997 if (unicode_resize(&v, outpos) < 0)
Benjamin Peterson29060642009-01-31 22:14:21 +00005998 goto onError;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005999 Py_XDECREF(errorHandler);
6000 Py_XDECREF(exc);
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01006001 return unicode_result(v);
Tim Petersced69f82003-09-16 20:30:58 +00006002
Benjamin Peterson29060642009-01-31 22:14:21 +00006003 onError:
Guido van Rossumd57fd912000-03-10 22:53:23 +00006004 Py_XDECREF(v);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006005 Py_XDECREF(errorHandler);
6006 Py_XDECREF(exc);
Guido van Rossumd57fd912000-03-10 22:53:23 +00006007 return NULL;
6008}
6009
Martin v. Löwis1db7c132011-11-10 18:24:32 +01006010
Alexander Belopolsky40018472011-02-26 01:02:56 +00006011PyObject *
Martin v. Löwis1db7c132011-11-10 18:24:32 +01006012PyUnicode_AsRawUnicodeEscapeString(PyObject *unicode)
Guido van Rossumd57fd912000-03-10 22:53:23 +00006013{
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00006014 PyObject *repr;
Guido van Rossumd57fd912000-03-10 22:53:23 +00006015 char *p;
6016 char *q;
Martin v. Löwis1db7c132011-11-10 18:24:32 +01006017 Py_ssize_t expandsize, pos;
6018 int kind;
6019 void *data;
6020 Py_ssize_t len;
Guido van Rossumd57fd912000-03-10 22:53:23 +00006021
Martin v. Löwis1db7c132011-11-10 18:24:32 +01006022 if (!PyUnicode_Check(unicode)) {
6023 PyErr_BadArgument();
6024 return NULL;
6025 }
Benjamin Petersonbac79492012-01-14 13:34:47 -05006026 if (PyUnicode_READY(unicode) == -1)
Martin v. Löwis1db7c132011-11-10 18:24:32 +01006027 return NULL;
6028 kind = PyUnicode_KIND(unicode);
6029 data = PyUnicode_DATA(unicode);
6030 len = PyUnicode_GET_LENGTH(unicode);
Benjamin Peterson1518e872011-11-23 10:44:52 -06006031 /* 4 byte characters can take up 10 bytes, 2 byte characters can take up 6
6032 bytes, and 1 byte characters 4. */
6033 expandsize = kind * 2 + 2;
Victor Stinner0e368262011-11-10 20:12:49 +01006034
Martin v. Löwis1db7c132011-11-10 18:24:32 +01006035 if (len > PY_SSIZE_T_MAX / expandsize)
Benjamin Peterson29060642009-01-31 22:14:21 +00006036 return PyErr_NoMemory();
Benjamin Peterson14339b62009-01-31 16:36:08 +00006037
Martin v. Löwis1db7c132011-11-10 18:24:32 +01006038 repr = PyBytes_FromStringAndSize(NULL, expandsize * len);
Guido van Rossumd57fd912000-03-10 22:53:23 +00006039 if (repr == NULL)
6040 return NULL;
Martin v. Löwis1db7c132011-11-10 18:24:32 +01006041 if (len == 0)
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00006042 return repr;
Guido van Rossumd57fd912000-03-10 22:53:23 +00006043
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00006044 p = q = PyBytes_AS_STRING(repr);
Martin v. Löwis1db7c132011-11-10 18:24:32 +01006045 for (pos = 0; pos < len; pos++) {
6046 Py_UCS4 ch = PyUnicode_READ(kind, data, pos);
Benjamin Peterson29060642009-01-31 22:14:21 +00006047 /* Map 32-bit characters to '\Uxxxxxxxx' */
6048 if (ch >= 0x10000) {
Victor Stinner8faf8212011-12-08 22:14:11 +01006049 assert(ch <= MAX_UNICODE);
Martin v. Löwis9a3a9f72003-05-18 12:31:09 +00006050 *p++ = '\\';
6051 *p++ = 'U';
Victor Stinnerf5cff562011-10-14 02:13:11 +02006052 *p++ = Py_hexdigits[(ch >> 28) & 0xf];
6053 *p++ = Py_hexdigits[(ch >> 24) & 0xf];
6054 *p++ = Py_hexdigits[(ch >> 20) & 0xf];
6055 *p++ = Py_hexdigits[(ch >> 16) & 0xf];
6056 *p++ = Py_hexdigits[(ch >> 12) & 0xf];
6057 *p++ = Py_hexdigits[(ch >> 8) & 0xf];
6058 *p++ = Py_hexdigits[(ch >> 4) & 0xf];
6059 *p++ = Py_hexdigits[ch & 15];
Tim Petersced69f82003-09-16 20:30:58 +00006060 }
Benjamin Peterson29060642009-01-31 22:14:21 +00006061 /* Map 16-bit characters to '\uxxxx' */
Martin v. Löwis1db7c132011-11-10 18:24:32 +01006062 else if (ch >= 256) {
Guido van Rossumd57fd912000-03-10 22:53:23 +00006063 *p++ = '\\';
6064 *p++ = 'u';
Victor Stinnerf5cff562011-10-14 02:13:11 +02006065 *p++ = Py_hexdigits[(ch >> 12) & 0xf];
6066 *p++ = Py_hexdigits[(ch >> 8) & 0xf];
6067 *p++ = Py_hexdigits[(ch >> 4) & 0xf];
6068 *p++ = Py_hexdigits[ch & 15];
Guido van Rossumd57fd912000-03-10 22:53:23 +00006069 }
Benjamin Peterson29060642009-01-31 22:14:21 +00006070 /* Copy everything else as-is */
6071 else
Guido van Rossumd57fd912000-03-10 22:53:23 +00006072 *p++ = (char) ch;
6073 }
Guido van Rossum98297ee2007-11-06 21:34:58 +00006074
Martin v. Löwis1db7c132011-11-10 18:24:32 +01006075 assert(p > q);
6076 if (_PyBytes_Resize(&repr, p - q) < 0)
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00006077 return NULL;
6078 return repr;
Guido van Rossumd57fd912000-03-10 22:53:23 +00006079}
6080
Alexander Belopolsky40018472011-02-26 01:02:56 +00006081PyObject *
Martin v. Löwis1db7c132011-11-10 18:24:32 +01006082PyUnicode_EncodeRawUnicodeEscape(const Py_UNICODE *s,
6083 Py_ssize_t size)
Guido van Rossumd57fd912000-03-10 22:53:23 +00006084{
Martin v. Löwis1db7c132011-11-10 18:24:32 +01006085 PyObject *result;
6086 PyObject *tmp = PyUnicode_FromUnicode(s, size);
6087 if (tmp == NULL)
Walter Dörwald711005d2007-05-12 12:03:26 +00006088 return NULL;
Martin v. Löwis1db7c132011-11-10 18:24:32 +01006089 result = PyUnicode_AsRawUnicodeEscapeString(tmp);
6090 Py_DECREF(tmp);
6091 return result;
Guido van Rossumd57fd912000-03-10 22:53:23 +00006092}
6093
Walter Dörwalda47d1c02005-08-30 10:23:14 +00006094/* --- Unicode Internal Codec ------------------------------------------- */
6095
Alexander Belopolsky40018472011-02-26 01:02:56 +00006096PyObject *
6097_PyUnicode_DecodeUnicodeInternal(const char *s,
Ezio Melotti2aa2b3b2011-09-29 00:58:57 +03006098 Py_ssize_t size,
6099 const char *errors)
Walter Dörwalda47d1c02005-08-30 10:23:14 +00006100{
6101 const char *starts = s;
Martin v. Löwis18e16552006-02-15 17:27:45 +00006102 Py_ssize_t startinpos;
6103 Py_ssize_t endinpos;
6104 Py_ssize_t outpos;
Victor Stinner7931d9a2011-11-04 00:22:48 +01006105 PyObject *v;
Walter Dörwalda47d1c02005-08-30 10:23:14 +00006106 const char *end;
6107 const char *reason;
6108 PyObject *errorHandler = NULL;
6109 PyObject *exc = NULL;
6110
Victor Stinner9f4b1e92011-11-10 20:56:30 +01006111 if (PyErr_WarnEx(PyExc_DeprecationWarning,
Ezio Melotti11060a42011-11-16 09:39:10 +02006112 "unicode_internal codec has been deprecated",
Victor Stinner9f4b1e92011-11-10 20:56:30 +01006113 1))
6114 return NULL;
6115
Thomas Wouters89f507f2006-12-13 04:49:30 +00006116 /* XXX overflow detection missing */
Martin v. Löwise9b11c12011-11-08 17:35:34 +01006117 v = PyUnicode_New((size+Py_UNICODE_SIZE-1)/ Py_UNICODE_SIZE, 127);
Walter Dörwalda47d1c02005-08-30 10:23:14 +00006118 if (v == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00006119 goto onError;
Martin v. Löwise9b11c12011-11-08 17:35:34 +01006120 if (PyUnicode_GET_LENGTH(v) == 0)
Victor Stinner7931d9a2011-11-04 00:22:48 +01006121 return v;
Martin v. Löwise9b11c12011-11-08 17:35:34 +01006122 outpos = 0;
Walter Dörwalda47d1c02005-08-30 10:23:14 +00006123 end = s + size;
6124
6125 while (s < end) {
Antoine Pitrou0290c7a2011-11-11 13:29:12 +01006126 Py_UNICODE uch;
Antoine Pitrou44c6aff2011-11-11 02:59:42 +01006127 Py_UCS4 ch;
6128 /* We copy the raw representation one byte at a time because the
6129 pointer may be unaligned (see test_codeccallbacks). */
Antoine Pitrou0290c7a2011-11-11 13:29:12 +01006130 ((char *) &uch)[0] = s[0];
6131 ((char *) &uch)[1] = s[1];
Antoine Pitrou44c6aff2011-11-11 02:59:42 +01006132#ifdef Py_UNICODE_WIDE
Antoine Pitrou0290c7a2011-11-11 13:29:12 +01006133 ((char *) &uch)[2] = s[2];
6134 ((char *) &uch)[3] = s[3];
Antoine Pitrou44c6aff2011-11-11 02:59:42 +01006135#endif
Antoine Pitrou0290c7a2011-11-11 13:29:12 +01006136 ch = uch;
6137
Walter Dörwalda47d1c02005-08-30 10:23:14 +00006138 /* We have to sanity check the raw data, otherwise doom looms for
6139 some malformed UCS-4 data. */
6140 if (
Benjamin Peterson29060642009-01-31 22:14:21 +00006141#ifdef Py_UNICODE_WIDE
Martin v. Löwise9b11c12011-11-08 17:35:34 +01006142 ch > 0x10ffff ||
Benjamin Peterson29060642009-01-31 22:14:21 +00006143#endif
Walter Dörwalda47d1c02005-08-30 10:23:14 +00006144 end-s < Py_UNICODE_SIZE
6145 )
Benjamin Peterson29060642009-01-31 22:14:21 +00006146 {
Walter Dörwalda47d1c02005-08-30 10:23:14 +00006147 startinpos = s - starts;
6148 if (end-s < Py_UNICODE_SIZE) {
6149 endinpos = end-starts;
6150 reason = "truncated input";
6151 }
6152 else {
6153 endinpos = s - starts + Py_UNICODE_SIZE;
6154 reason = "illegal code point (> 0x10FFFF)";
6155 }
Walter Dörwalda47d1c02005-08-30 10:23:14 +00006156 if (unicode_decode_call_errorhandler(
6157 errors, &errorHandler,
6158 "unicode_internal", reason,
Walter Dörwalde78178e2007-07-30 13:31:40 +00006159 &starts, &end, &startinpos, &endinpos, &exc, &s,
Victor Stinner9f4b1e92011-11-10 20:56:30 +01006160 &v, &outpos))
Walter Dörwalda47d1c02005-08-30 10:23:14 +00006161 goto onError;
Victor Stinner9f4b1e92011-11-10 20:56:30 +01006162 continue;
6163 }
6164
6165 s += Py_UNICODE_SIZE;
6166#ifndef Py_UNICODE_WIDE
Victor Stinner551ac952011-11-29 22:58:13 +01006167 if (Py_UNICODE_IS_HIGH_SURROGATE(ch) && s < end)
Victor Stinner9f4b1e92011-11-10 20:56:30 +01006168 {
Antoine Pitrou0290c7a2011-11-11 13:29:12 +01006169 Py_UNICODE uch2;
6170 ((char *) &uch2)[0] = s[0];
6171 ((char *) &uch2)[1] = s[1];
Victor Stinner551ac952011-11-29 22:58:13 +01006172 if (Py_UNICODE_IS_LOW_SURROGATE(uch2))
Victor Stinner9f4b1e92011-11-10 20:56:30 +01006173 {
Victor Stinner551ac952011-11-29 22:58:13 +01006174 ch = Py_UNICODE_JOIN_SURROGATES(uch, uch2);
Victor Stinner9f4b1e92011-11-10 20:56:30 +01006175 s += Py_UNICODE_SIZE;
Walter Dörwalda47d1c02005-08-30 10:23:14 +00006176 }
6177 }
Victor Stinner9f4b1e92011-11-10 20:56:30 +01006178#endif
6179
6180 if (unicode_putchar(&v, &outpos, ch) < 0)
6181 goto onError;
Walter Dörwalda47d1c02005-08-30 10:23:14 +00006182 }
6183
Victor Stinner16e6a802011-12-12 13:24:15 +01006184 if (unicode_resize(&v, outpos) < 0)
Walter Dörwalda47d1c02005-08-30 10:23:14 +00006185 goto onError;
6186 Py_XDECREF(errorHandler);
6187 Py_XDECREF(exc);
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01006188 return unicode_result(v);
Walter Dörwalda47d1c02005-08-30 10:23:14 +00006189
Benjamin Peterson29060642009-01-31 22:14:21 +00006190 onError:
Walter Dörwalda47d1c02005-08-30 10:23:14 +00006191 Py_XDECREF(v);
6192 Py_XDECREF(errorHandler);
6193 Py_XDECREF(exc);
6194 return NULL;
6195}
6196
Guido van Rossumd57fd912000-03-10 22:53:23 +00006197/* --- Latin-1 Codec ------------------------------------------------------ */
6198
Alexander Belopolsky40018472011-02-26 01:02:56 +00006199PyObject *
6200PyUnicode_DecodeLatin1(const char *s,
Ezio Melotti2aa2b3b2011-09-29 00:58:57 +03006201 Py_ssize_t size,
6202 const char *errors)
Guido van Rossumd57fd912000-03-10 22:53:23 +00006203{
Guido van Rossumd57fd912000-03-10 22:53:23 +00006204 /* Latin-1 is equivalent to the first 256 ordinals in Unicode. */
Victor Stinnere57b1c02011-09-28 22:20:48 +02006205 return _PyUnicode_FromUCS1((unsigned char*)s, size);
Guido van Rossumd57fd912000-03-10 22:53:23 +00006206}
6207
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006208/* create or adjust a UnicodeEncodeError */
Alexander Belopolsky40018472011-02-26 01:02:56 +00006209static void
6210make_encode_exception(PyObject **exceptionObject,
Ezio Melotti2aa2b3b2011-09-29 00:58:57 +03006211 const char *encoding,
Martin v. Löwis9e816682011-11-02 12:45:42 +01006212 PyObject *unicode,
6213 Py_ssize_t startpos, Py_ssize_t endpos,
6214 const char *reason)
6215{
6216 if (*exceptionObject == NULL) {
6217 *exceptionObject = PyObject_CallFunction(
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006218 PyExc_UnicodeEncodeError, "sOnns",
Martin v. Löwis9e816682011-11-02 12:45:42 +01006219 encoding, unicode, startpos, endpos, reason);
6220 }
6221 else {
6222 if (PyUnicodeEncodeError_SetStart(*exceptionObject, startpos))
6223 goto onError;
6224 if (PyUnicodeEncodeError_SetEnd(*exceptionObject, endpos))
6225 goto onError;
6226 if (PyUnicodeEncodeError_SetReason(*exceptionObject, reason))
6227 goto onError;
6228 return;
6229 onError:
6230 Py_DECREF(*exceptionObject);
6231 *exceptionObject = NULL;
6232 }
6233}
6234
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006235/* raises a UnicodeEncodeError */
Alexander Belopolsky40018472011-02-26 01:02:56 +00006236static void
6237raise_encode_exception(PyObject **exceptionObject,
Ezio Melotti2aa2b3b2011-09-29 00:58:57 +03006238 const char *encoding,
Martin v. Löwis9e816682011-11-02 12:45:42 +01006239 PyObject *unicode,
6240 Py_ssize_t startpos, Py_ssize_t endpos,
6241 const char *reason)
6242{
Martin v. Löwis12be46c2011-11-04 19:04:15 +01006243 make_encode_exception(exceptionObject,
Martin v. Löwis9e816682011-11-02 12:45:42 +01006244 encoding, unicode, startpos, endpos, reason);
6245 if (*exceptionObject != NULL)
6246 PyCodec_StrictErrors(*exceptionObject);
6247}
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006248
6249/* error handling callback helper:
6250 build arguments, call the callback and check the arguments,
6251 put the result into newpos and return the replacement string, which
6252 has to be freed by the caller */
Alexander Belopolsky40018472011-02-26 01:02:56 +00006253static PyObject *
6254unicode_encode_call_errorhandler(const char *errors,
Ezio Melotti2aa2b3b2011-09-29 00:58:57 +03006255 PyObject **errorHandler,
6256 const char *encoding, const char *reason,
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006257 PyObject *unicode, PyObject **exceptionObject,
Ezio Melotti2aa2b3b2011-09-29 00:58:57 +03006258 Py_ssize_t startpos, Py_ssize_t endpos,
6259 Py_ssize_t *newpos)
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006260{
Martin v. Löwisdb12d452009-05-02 18:52:14 +00006261 static char *argparse = "On;encoding error handler must return (str/bytes, int) tuple";
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006262 Py_ssize_t len;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006263 PyObject *restuple;
6264 PyObject *resunicode;
6265
6266 if (*errorHandler == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +00006267 *errorHandler = PyCodec_LookupError(errors);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006268 if (*errorHandler == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00006269 return NULL;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006270 }
6271
Benjamin Petersonbac79492012-01-14 13:34:47 -05006272 if (PyUnicode_READY(unicode) == -1)
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006273 return NULL;
6274 len = PyUnicode_GET_LENGTH(unicode);
6275
Martin v. Löwis12be46c2011-11-04 19:04:15 +01006276 make_encode_exception(exceptionObject,
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006277 encoding, unicode, startpos, endpos, reason);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006278 if (*exceptionObject == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00006279 return NULL;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006280
6281 restuple = PyObject_CallFunctionObjArgs(
Benjamin Peterson29060642009-01-31 22:14:21 +00006282 *errorHandler, *exceptionObject, NULL);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006283 if (restuple == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00006284 return NULL;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006285 if (!PyTuple_Check(restuple)) {
Martin v. Löwisdb12d452009-05-02 18:52:14 +00006286 PyErr_SetString(PyExc_TypeError, &argparse[3]);
Benjamin Peterson29060642009-01-31 22:14:21 +00006287 Py_DECREF(restuple);
6288 return NULL;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006289 }
Martin v. Löwisdb12d452009-05-02 18:52:14 +00006290 if (!PyArg_ParseTuple(restuple, argparse,
Benjamin Peterson29060642009-01-31 22:14:21 +00006291 &resunicode, newpos)) {
6292 Py_DECREF(restuple);
6293 return NULL;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006294 }
Martin v. Löwisdb12d452009-05-02 18:52:14 +00006295 if (!PyUnicode_Check(resunicode) && !PyBytes_Check(resunicode)) {
6296 PyErr_SetString(PyExc_TypeError, &argparse[3]);
6297 Py_DECREF(restuple);
6298 return NULL;
6299 }
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006300 if (*newpos<0)
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006301 *newpos = len + *newpos;
6302 if (*newpos<0 || *newpos>len) {
Benjamin Peterson29060642009-01-31 22:14:21 +00006303 PyErr_Format(PyExc_IndexError, "position %zd from error handler out of bounds", *newpos);
6304 Py_DECREF(restuple);
6305 return NULL;
Walter Dörwald2e0b18a2003-01-31 17:19:08 +00006306 }
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006307 Py_INCREF(resunicode);
6308 Py_DECREF(restuple);
6309 return resunicode;
6310}
6311
Alexander Belopolsky40018472011-02-26 01:02:56 +00006312static PyObject *
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006313unicode_encode_ucs1(PyObject *unicode,
Ezio Melotti2aa2b3b2011-09-29 00:58:57 +03006314 const char *errors,
Victor Stinnerfcd96532011-11-04 00:28:50 +01006315 unsigned int limit)
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006316{
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006317 /* input state */
6318 Py_ssize_t pos=0, size;
6319 int kind;
6320 void *data;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006321 /* output object */
6322 PyObject *res;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006323 /* pointer into the output */
6324 char *str;
6325 /* current output position */
Martin v. Löwis18e16552006-02-15 17:27:45 +00006326 Py_ssize_t ressize;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006327 const char *encoding = (limit == 256) ? "latin-1" : "ascii";
6328 const char *reason = (limit == 256) ? "ordinal not in range(256)" : "ordinal not in range(128)";
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006329 PyObject *errorHandler = NULL;
6330 PyObject *exc = NULL;
6331 /* the following variable is used for caching string comparisons
6332 * -1=not initialized, 0=unknown, 1=strict, 2=replace, 3=ignore, 4=xmlcharrefreplace */
6333 int known_errorHandler = -1;
6334
Benjamin Petersonbac79492012-01-14 13:34:47 -05006335 if (PyUnicode_READY(unicode) == -1)
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006336 return NULL;
6337 size = PyUnicode_GET_LENGTH(unicode);
6338 kind = PyUnicode_KIND(unicode);
6339 data = PyUnicode_DATA(unicode);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006340 /* allocate enough for a simple encoding without
6341 replacements, if we need more, we'll resize */
Guido van Rossum98297ee2007-11-06 21:34:58 +00006342 if (size == 0)
Christian Heimes72b710a2008-05-26 13:28:38 +00006343 return PyBytes_FromStringAndSize(NULL, 0);
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00006344 res = PyBytes_FromStringAndSize(NULL, size);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006345 if (res == NULL)
Guido van Rossum98297ee2007-11-06 21:34:58 +00006346 return NULL;
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00006347 str = PyBytes_AS_STRING(res);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006348 ressize = size;
6349
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006350 while (pos < size) {
6351 Py_UCS4 c = PyUnicode_READ(kind, data, pos);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006352
Benjamin Peterson29060642009-01-31 22:14:21 +00006353 /* can we encode this? */
6354 if (c<limit) {
6355 /* no overflow check, because we know that the space is enough */
6356 *str++ = (char)c;
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006357 ++pos;
Benjamin Peterson14339b62009-01-31 16:36:08 +00006358 }
Benjamin Peterson29060642009-01-31 22:14:21 +00006359 else {
Benjamin Peterson29060642009-01-31 22:14:21 +00006360 Py_ssize_t requiredsize;
6361 PyObject *repunicode;
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006362 Py_ssize_t repsize, newpos, respos, i;
Benjamin Peterson29060642009-01-31 22:14:21 +00006363 /* startpos for collecting unencodable chars */
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006364 Py_ssize_t collstart = pos;
6365 Py_ssize_t collend = pos;
Benjamin Peterson29060642009-01-31 22:14:21 +00006366 /* find all unecodable characters */
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006367 while ((collend < size) && (PyUnicode_READ(kind, data, collend)>=limit))
Benjamin Peterson29060642009-01-31 22:14:21 +00006368 ++collend;
6369 /* cache callback name lookup (if not done yet, i.e. it's the first error) */
6370 if (known_errorHandler==-1) {
6371 if ((errors==NULL) || (!strcmp(errors, "strict")))
6372 known_errorHandler = 1;
6373 else if (!strcmp(errors, "replace"))
6374 known_errorHandler = 2;
6375 else if (!strcmp(errors, "ignore"))
6376 known_errorHandler = 3;
6377 else if (!strcmp(errors, "xmlcharrefreplace"))
6378 known_errorHandler = 4;
6379 else
6380 known_errorHandler = 0;
6381 }
6382 switch (known_errorHandler) {
6383 case 1: /* strict */
Martin v. Löwis12be46c2011-11-04 19:04:15 +01006384 raise_encode_exception(&exc, encoding, unicode, collstart, collend, reason);
Benjamin Peterson29060642009-01-31 22:14:21 +00006385 goto onError;
6386 case 2: /* replace */
6387 while (collstart++<collend)
6388 *str++ = '?'; /* fall through */
6389 case 3: /* ignore */
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006390 pos = collend;
Benjamin Peterson29060642009-01-31 22:14:21 +00006391 break;
6392 case 4: /* xmlcharrefreplace */
6393 respos = str - PyBytes_AS_STRING(res);
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006394 /* determine replacement size */
6395 for (i = collstart, repsize = 0; i < collend; ++i) {
6396 Py_UCS4 ch = PyUnicode_READ(kind, data, i);
6397 if (ch < 10)
Benjamin Peterson29060642009-01-31 22:14:21 +00006398 repsize += 2+1+1;
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006399 else if (ch < 100)
Benjamin Peterson29060642009-01-31 22:14:21 +00006400 repsize += 2+2+1;
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006401 else if (ch < 1000)
Benjamin Peterson29060642009-01-31 22:14:21 +00006402 repsize += 2+3+1;
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006403 else if (ch < 10000)
Benjamin Peterson29060642009-01-31 22:14:21 +00006404 repsize += 2+4+1;
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006405 else if (ch < 100000)
Benjamin Peterson29060642009-01-31 22:14:21 +00006406 repsize += 2+5+1;
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006407 else if (ch < 1000000)
Benjamin Peterson29060642009-01-31 22:14:21 +00006408 repsize += 2+6+1;
Victor Stinner0d3721d2011-11-22 03:27:53 +01006409 else {
Victor Stinner8faf8212011-12-08 22:14:11 +01006410 assert(ch <= MAX_UNICODE);
Benjamin Peterson29060642009-01-31 22:14:21 +00006411 repsize += 2+7+1;
Victor Stinner0d3721d2011-11-22 03:27:53 +01006412 }
Benjamin Peterson29060642009-01-31 22:14:21 +00006413 }
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006414 requiredsize = respos+repsize+(size-collend);
Benjamin Peterson29060642009-01-31 22:14:21 +00006415 if (requiredsize > ressize) {
6416 if (requiredsize<2*ressize)
6417 requiredsize = 2*ressize;
6418 if (_PyBytes_Resize(&res, requiredsize))
6419 goto onError;
6420 str = PyBytes_AS_STRING(res) + respos;
6421 ressize = requiredsize;
6422 }
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006423 /* generate replacement */
6424 for (i = collstart; i < collend; ++i) {
6425 str += sprintf(str, "&#%d;", PyUnicode_READ(kind, data, i));
Benjamin Peterson29060642009-01-31 22:14:21 +00006426 }
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006427 pos = collend;
Benjamin Peterson29060642009-01-31 22:14:21 +00006428 break;
6429 default:
6430 repunicode = unicode_encode_call_errorhandler(errors, &errorHandler,
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006431 encoding, reason, unicode, &exc,
6432 collstart, collend, &newpos);
6433 if (repunicode == NULL || (PyUnicode_Check(repunicode) &&
Benjamin Petersonbac79492012-01-14 13:34:47 -05006434 PyUnicode_READY(repunicode) == -1))
Benjamin Peterson29060642009-01-31 22:14:21 +00006435 goto onError;
Martin v. Löwis011e8422009-05-05 04:43:17 +00006436 if (PyBytes_Check(repunicode)) {
6437 /* Directly copy bytes result to output. */
6438 repsize = PyBytes_Size(repunicode);
6439 if (repsize > 1) {
6440 /* Make room for all additional bytes. */
Amaury Forgeot d'Arc84ec8d92009-06-29 22:36:49 +00006441 respos = str - PyBytes_AS_STRING(res);
Martin v. Löwis011e8422009-05-05 04:43:17 +00006442 if (_PyBytes_Resize(&res, ressize+repsize-1)) {
6443 Py_DECREF(repunicode);
6444 goto onError;
6445 }
Amaury Forgeot d'Arc84ec8d92009-06-29 22:36:49 +00006446 str = PyBytes_AS_STRING(res) + respos;
Martin v. Löwis011e8422009-05-05 04:43:17 +00006447 ressize += repsize-1;
6448 }
6449 memcpy(str, PyBytes_AsString(repunicode), repsize);
6450 str += repsize;
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006451 pos = newpos;
Martin v. Löwisdb12d452009-05-02 18:52:14 +00006452 Py_DECREF(repunicode);
Martin v. Löwis011e8422009-05-05 04:43:17 +00006453 break;
Martin v. Löwisdb12d452009-05-02 18:52:14 +00006454 }
Benjamin Peterson29060642009-01-31 22:14:21 +00006455 /* need more space? (at least enough for what we
6456 have+the replacement+the rest of the string, so
6457 we won't have to check space for encodable characters) */
6458 respos = str - PyBytes_AS_STRING(res);
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006459 repsize = PyUnicode_GET_LENGTH(repunicode);
6460 requiredsize = respos+repsize+(size-collend);
Benjamin Peterson29060642009-01-31 22:14:21 +00006461 if (requiredsize > ressize) {
6462 if (requiredsize<2*ressize)
6463 requiredsize = 2*ressize;
6464 if (_PyBytes_Resize(&res, requiredsize)) {
6465 Py_DECREF(repunicode);
6466 goto onError;
6467 }
6468 str = PyBytes_AS_STRING(res) + respos;
6469 ressize = requiredsize;
6470 }
6471 /* check if there is anything unencodable in the replacement
6472 and copy it to the output */
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006473 for (i = 0; repsize-->0; ++i, ++str) {
6474 c = PyUnicode_READ_CHAR(repunicode, i);
Benjamin Peterson29060642009-01-31 22:14:21 +00006475 if (c >= limit) {
Martin v. Löwis12be46c2011-11-04 19:04:15 +01006476 raise_encode_exception(&exc, encoding, unicode,
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006477 pos, pos+1, reason);
Benjamin Peterson29060642009-01-31 22:14:21 +00006478 Py_DECREF(repunicode);
6479 goto onError;
6480 }
6481 *str = (char)c;
6482 }
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006483 pos = newpos;
Benjamin Peterson14339b62009-01-31 16:36:08 +00006484 Py_DECREF(repunicode);
Benjamin Peterson14339b62009-01-31 16:36:08 +00006485 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00006486 }
6487 }
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00006488 /* Resize if we allocated to much */
6489 size = str - PyBytes_AS_STRING(res);
6490 if (size < ressize) { /* If this falls res will be NULL */
Alexandre Vassalottibad1b922008-12-27 09:49:09 +00006491 assert(size >= 0);
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00006492 if (_PyBytes_Resize(&res, size) < 0)
6493 goto onError;
6494 }
6495
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006496 Py_XDECREF(errorHandler);
6497 Py_XDECREF(exc);
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00006498 return res;
6499
6500 onError:
6501 Py_XDECREF(res);
6502 Py_XDECREF(errorHandler);
6503 Py_XDECREF(exc);
6504 return NULL;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006505}
6506
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006507/* Deprecated */
Alexander Belopolsky40018472011-02-26 01:02:56 +00006508PyObject *
6509PyUnicode_EncodeLatin1(const Py_UNICODE *p,
Ezio Melotti2aa2b3b2011-09-29 00:58:57 +03006510 Py_ssize_t size,
6511 const char *errors)
Guido van Rossumd57fd912000-03-10 22:53:23 +00006512{
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006513 PyObject *result;
6514 PyObject *unicode = PyUnicode_FromUnicode(p, size);
6515 if (unicode == NULL)
6516 return NULL;
6517 result = unicode_encode_ucs1(unicode, errors, 256);
6518 Py_DECREF(unicode);
6519 return result;
Guido van Rossumd57fd912000-03-10 22:53:23 +00006520}
6521
Alexander Belopolsky40018472011-02-26 01:02:56 +00006522PyObject *
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02006523_PyUnicode_AsLatin1String(PyObject *unicode, const char *errors)
Guido van Rossumd57fd912000-03-10 22:53:23 +00006524{
6525 if (!PyUnicode_Check(unicode)) {
Benjamin Peterson29060642009-01-31 22:14:21 +00006526 PyErr_BadArgument();
6527 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00006528 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02006529 if (PyUnicode_READY(unicode) == -1)
6530 return NULL;
6531 /* Fast path: if it is a one-byte string, construct
6532 bytes object directly. */
6533 if (PyUnicode_KIND(unicode) == PyUnicode_1BYTE_KIND)
6534 return PyBytes_FromStringAndSize(PyUnicode_DATA(unicode),
6535 PyUnicode_GET_LENGTH(unicode));
6536 /* Non-Latin-1 characters present. Defer to above function to
6537 raise the exception. */
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006538 return unicode_encode_ucs1(unicode, errors, 256);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02006539}
6540
6541PyObject*
6542PyUnicode_AsLatin1String(PyObject *unicode)
6543{
6544 return _PyUnicode_AsLatin1String(unicode, NULL);
Guido van Rossumd57fd912000-03-10 22:53:23 +00006545}
6546
6547/* --- 7-bit ASCII Codec -------------------------------------------------- */
6548
Alexander Belopolsky40018472011-02-26 01:02:56 +00006549PyObject *
6550PyUnicode_DecodeASCII(const char *s,
6551 Py_ssize_t size,
6552 const char *errors)
Guido van Rossumd57fd912000-03-10 22:53:23 +00006553{
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006554 const char *starts = s;
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02006555 PyObject *unicode;
Martin v. Löwise9b11c12011-11-08 17:35:34 +01006556 int kind;
6557 void *data;
Martin v. Löwis18e16552006-02-15 17:27:45 +00006558 Py_ssize_t startinpos;
6559 Py_ssize_t endinpos;
6560 Py_ssize_t outpos;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006561 const char *e;
6562 PyObject *errorHandler = NULL;
6563 PyObject *exc = NULL;
Tim Petersced69f82003-09-16 20:30:58 +00006564
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01006565 if (size == 0) {
6566 Py_INCREF(unicode_empty);
6567 return unicode_empty;
6568 }
6569
Guido van Rossumd57fd912000-03-10 22:53:23 +00006570 /* ASCII is equivalent to the first 128 ordinals in Unicode. */
Victor Stinner702c7342011-10-05 13:50:52 +02006571 if (size == 1 && (unsigned char)s[0] < 128)
6572 return get_latin1_char((unsigned char)s[0]);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02006573
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02006574 unicode = PyUnicode_New(size, 127);
6575 if (unicode == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00006576 goto onError;
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02006577
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006578 e = s + size;
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02006579 data = PyUnicode_1BYTE_DATA(unicode);
6580 outpos = ascii_decode(s, e, (Py_UCS1 *)data);
6581 if (outpos == size)
6582 return unicode;
6583
6584 s += outpos;
6585 kind = PyUnicode_1BYTE_KIND;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006586 while (s < e) {
Benjamin Peterson29060642009-01-31 22:14:21 +00006587 register unsigned char c = (unsigned char)*s;
6588 if (c < 128) {
Martin v. Löwise9b11c12011-11-08 17:35:34 +01006589 PyUnicode_WRITE(kind, data, outpos++, c);
Benjamin Peterson29060642009-01-31 22:14:21 +00006590 ++s;
6591 }
6592 else {
6593 startinpos = s-starts;
6594 endinpos = startinpos + 1;
Benjamin Peterson29060642009-01-31 22:14:21 +00006595 if (unicode_decode_call_errorhandler(
6596 errors, &errorHandler,
6597 "ascii", "ordinal not in range(128)",
6598 &starts, &e, &startinpos, &endinpos, &exc, &s,
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02006599 &unicode, &outpos))
Benjamin Peterson29060642009-01-31 22:14:21 +00006600 goto onError;
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02006601 kind = PyUnicode_KIND(unicode);
6602 data = PyUnicode_DATA(unicode);
Benjamin Peterson29060642009-01-31 22:14:21 +00006603 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00006604 }
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02006605 if (unicode_resize(&unicode, outpos) < 0)
Martin v. Löwise9b11c12011-11-08 17:35:34 +01006606 goto onError;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006607 Py_XDECREF(errorHandler);
6608 Py_XDECREF(exc);
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02006609 assert(_PyUnicode_CheckConsistency(unicode, 1));
6610 return unicode;
Tim Petersced69f82003-09-16 20:30:58 +00006611
Benjamin Peterson29060642009-01-31 22:14:21 +00006612 onError:
Antoine Pitrouca5f91b2012-05-10 16:36:02 +02006613 Py_XDECREF(unicode);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006614 Py_XDECREF(errorHandler);
6615 Py_XDECREF(exc);
Guido van Rossumd57fd912000-03-10 22:53:23 +00006616 return NULL;
6617}
6618
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006619/* Deprecated */
Alexander Belopolsky40018472011-02-26 01:02:56 +00006620PyObject *
6621PyUnicode_EncodeASCII(const Py_UNICODE *p,
6622 Py_ssize_t size,
6623 const char *errors)
Guido van Rossumd57fd912000-03-10 22:53:23 +00006624{
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006625 PyObject *result;
6626 PyObject *unicode = PyUnicode_FromUnicode(p, size);
6627 if (unicode == NULL)
6628 return NULL;
6629 result = unicode_encode_ucs1(unicode, errors, 128);
6630 Py_DECREF(unicode);
6631 return result;
Guido van Rossumd57fd912000-03-10 22:53:23 +00006632}
6633
Alexander Belopolsky40018472011-02-26 01:02:56 +00006634PyObject *
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02006635_PyUnicode_AsASCIIString(PyObject *unicode, const char *errors)
Guido van Rossumd57fd912000-03-10 22:53:23 +00006636{
6637 if (!PyUnicode_Check(unicode)) {
Benjamin Peterson29060642009-01-31 22:14:21 +00006638 PyErr_BadArgument();
6639 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00006640 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02006641 if (PyUnicode_READY(unicode) == -1)
6642 return NULL;
6643 /* Fast path: if it is an ASCII-only string, construct bytes object
6644 directly. Else defer to above function to raise the exception. */
6645 if (PyUnicode_MAX_CHAR_VALUE(unicode) < 128)
6646 return PyBytes_FromStringAndSize(PyUnicode_DATA(unicode),
6647 PyUnicode_GET_LENGTH(unicode));
Martin v. Löwis23e275b2011-11-02 18:02:51 +01006648 return unicode_encode_ucs1(unicode, errors, 128);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02006649}
6650
6651PyObject *
6652PyUnicode_AsASCIIString(PyObject *unicode)
6653{
6654 return _PyUnicode_AsASCIIString(unicode, NULL);
Guido van Rossumd57fd912000-03-10 22:53:23 +00006655}
6656
Victor Stinner99b95382011-07-04 14:23:54 +02006657#ifdef HAVE_MBCS
Guido van Rossum2ea3e142000-03-31 17:24:09 +00006658
Guido van Rossumb7a40ba2000-03-28 02:01:52 +00006659/* --- MBCS codecs for Windows -------------------------------------------- */
Guido van Rossum2ea3e142000-03-31 17:24:09 +00006660
Hirokazu Yamamoto35302462009-03-21 13:23:27 +00006661#if SIZEOF_INT < SIZEOF_SIZE_T
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006662#define NEED_RETRY
6663#endif
6664
Victor Stinner3a50e702011-10-18 21:21:00 +02006665#ifndef WC_ERR_INVALID_CHARS
6666# define WC_ERR_INVALID_CHARS 0x0080
6667#endif
6668
6669static char*
6670code_page_name(UINT code_page, PyObject **obj)
6671{
6672 *obj = NULL;
6673 if (code_page == CP_ACP)
6674 return "mbcs";
6675 if (code_page == CP_UTF7)
6676 return "CP_UTF7";
6677 if (code_page == CP_UTF8)
6678 return "CP_UTF8";
6679
6680 *obj = PyBytes_FromFormat("cp%u", code_page);
6681 if (*obj == NULL)
6682 return NULL;
6683 return PyBytes_AS_STRING(*obj);
6684}
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006685
Alexander Belopolsky40018472011-02-26 01:02:56 +00006686static int
Victor Stinner3a50e702011-10-18 21:21:00 +02006687is_dbcs_lead_byte(UINT code_page, const char *s, int offset)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006688{
6689 const char *curr = s + offset;
Victor Stinner3a50e702011-10-18 21:21:00 +02006690 const char *prev;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006691
Victor Stinner3a50e702011-10-18 21:21:00 +02006692 if (!IsDBCSLeadByteEx(code_page, *curr))
6693 return 0;
6694
6695 prev = CharPrevExA(code_page, s, curr, 0);
6696 if (prev == curr)
6697 return 1;
6698 /* FIXME: This code is limited to "true" double-byte encodings,
6699 as it assumes an incomplete character consists of a single
6700 byte. */
6701 if (curr - prev == 2)
6702 return 1;
6703 if (!IsDBCSLeadByteEx(code_page, *prev))
6704 return 1;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006705 return 0;
6706}
6707
Victor Stinner3a50e702011-10-18 21:21:00 +02006708static DWORD
6709decode_code_page_flags(UINT code_page)
6710{
6711 if (code_page == CP_UTF7) {
6712 /* The CP_UTF7 decoder only supports flags=0 */
6713 return 0;
6714 }
6715 else
6716 return MB_ERR_INVALID_CHARS;
6717}
6718
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006719/*
Victor Stinner3a50e702011-10-18 21:21:00 +02006720 * Decode a byte string from a Windows code page into unicode object in strict
6721 * mode.
6722 *
6723 * Returns consumed size if succeed, returns -2 on decode error, or raise a
6724 * WindowsError and returns -1 on other error.
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006725 */
Alexander Belopolsky40018472011-02-26 01:02:56 +00006726static int
Victor Stinner3a50e702011-10-18 21:21:00 +02006727decode_code_page_strict(UINT code_page,
Victor Stinner76a31a62011-11-04 00:05:13 +01006728 PyObject **v,
Victor Stinner3a50e702011-10-18 21:21:00 +02006729 const char *in,
6730 int insize)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006731{
Victor Stinner3a50e702011-10-18 21:21:00 +02006732 const DWORD flags = decode_code_page_flags(code_page);
Victor Stinner24729f32011-11-10 20:31:37 +01006733 wchar_t *out;
Victor Stinner3a50e702011-10-18 21:21:00 +02006734 DWORD outsize;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006735
6736 /* First get the size of the result */
Victor Stinner3a50e702011-10-18 21:21:00 +02006737 assert(insize > 0);
6738 outsize = MultiByteToWideChar(code_page, flags, in, insize, NULL, 0);
6739 if (outsize <= 0)
6740 goto error;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006741
6742 if (*v == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +00006743 /* Create unicode object */
Victor Stinnerab595942011-12-17 04:59:06 +01006744 /* FIXME: don't use _PyUnicode_New(), but allocate a wchar_t* buffer */
Victor Stinner76a31a62011-11-04 00:05:13 +01006745 *v = (PyObject*)_PyUnicode_New(outsize);
Benjamin Peterson29060642009-01-31 22:14:21 +00006746 if (*v == NULL)
6747 return -1;
Victor Stinner3a50e702011-10-18 21:21:00 +02006748 out = PyUnicode_AS_UNICODE(*v);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006749 }
6750 else {
Benjamin Peterson29060642009-01-31 22:14:21 +00006751 /* Extend unicode object */
Victor Stinner3a50e702011-10-18 21:21:00 +02006752 Py_ssize_t n = PyUnicode_GET_SIZE(*v);
Victor Stinner16e6a802011-12-12 13:24:15 +01006753 if (unicode_resize(v, n + outsize) < 0)
Benjamin Peterson29060642009-01-31 22:14:21 +00006754 return -1;
Victor Stinner3a50e702011-10-18 21:21:00 +02006755 out = PyUnicode_AS_UNICODE(*v) + n;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006756 }
6757
6758 /* Do the conversion */
Victor Stinner3a50e702011-10-18 21:21:00 +02006759 outsize = MultiByteToWideChar(code_page, flags, in, insize, out, outsize);
6760 if (outsize <= 0)
6761 goto error;
6762 return insize;
Victor Stinner554f3f02010-06-16 23:33:54 +00006763
Victor Stinner3a50e702011-10-18 21:21:00 +02006764error:
6765 if (GetLastError() == ERROR_NO_UNICODE_TRANSLATION)
6766 return -2;
6767 PyErr_SetFromWindowsErr(0);
Victor Stinner554f3f02010-06-16 23:33:54 +00006768 return -1;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006769}
6770
Victor Stinner3a50e702011-10-18 21:21:00 +02006771/*
6772 * Decode a byte string from a code page into unicode object with an error
6773 * handler.
6774 *
6775 * Returns consumed size if succeed, or raise a WindowsError or
6776 * UnicodeDecodeError exception and returns -1 on error.
6777 */
6778static int
6779decode_code_page_errors(UINT code_page,
Victor Stinner76a31a62011-11-04 00:05:13 +01006780 PyObject **v,
6781 const char *in, const int size,
Victor Stinner3a50e702011-10-18 21:21:00 +02006782 const char *errors)
6783{
6784 const char *startin = in;
6785 const char *endin = in + size;
6786 const DWORD flags = decode_code_page_flags(code_page);
6787 /* Ideally, we should get reason from FormatMessage. This is the Windows
6788 2000 English version of the message. */
6789 const char *reason = "No mapping for the Unicode character exists "
6790 "in the target code page.";
6791 /* each step cannot decode more than 1 character, but a character can be
6792 represented as a surrogate pair */
6793 wchar_t buffer[2], *startout, *out;
6794 int insize, outsize;
6795 PyObject *errorHandler = NULL;
6796 PyObject *exc = NULL;
6797 PyObject *encoding_obj = NULL;
6798 char *encoding;
6799 DWORD err;
6800 int ret = -1;
6801
6802 assert(size > 0);
6803
6804 encoding = code_page_name(code_page, &encoding_obj);
6805 if (encoding == NULL)
6806 return -1;
6807
6808 if (errors == NULL || strcmp(errors, "strict") == 0) {
6809 /* The last error was ERROR_NO_UNICODE_TRANSLATION, then we raise a
6810 UnicodeDecodeError. */
6811 make_decode_exception(&exc, encoding, in, size, 0, 0, reason);
6812 if (exc != NULL) {
6813 PyCodec_StrictErrors(exc);
6814 Py_CLEAR(exc);
6815 }
6816 goto error;
6817 }
6818
6819 if (*v == NULL) {
6820 /* Create unicode object */
6821 if (size > PY_SSIZE_T_MAX / (Py_ssize_t)Py_ARRAY_LENGTH(buffer)) {
6822 PyErr_NoMemory();
6823 goto error;
6824 }
Victor Stinnerab595942011-12-17 04:59:06 +01006825 /* FIXME: don't use _PyUnicode_New(), but allocate a wchar_t* buffer */
Victor Stinner76a31a62011-11-04 00:05:13 +01006826 *v = (PyObject*)_PyUnicode_New(size * Py_ARRAY_LENGTH(buffer));
Victor Stinner3a50e702011-10-18 21:21:00 +02006827 if (*v == NULL)
6828 goto error;
6829 startout = PyUnicode_AS_UNICODE(*v);
6830 }
6831 else {
6832 /* Extend unicode object */
6833 Py_ssize_t n = PyUnicode_GET_SIZE(*v);
6834 if (size > (PY_SSIZE_T_MAX - n) / (Py_ssize_t)Py_ARRAY_LENGTH(buffer)) {
6835 PyErr_NoMemory();
6836 goto error;
6837 }
Victor Stinner16e6a802011-12-12 13:24:15 +01006838 if (unicode_resize(v, n + size * Py_ARRAY_LENGTH(buffer)) < 0)
Victor Stinner3a50e702011-10-18 21:21:00 +02006839 goto error;
6840 startout = PyUnicode_AS_UNICODE(*v) + n;
6841 }
6842
6843 /* Decode the byte string character per character */
6844 out = startout;
6845 while (in < endin)
6846 {
6847 /* Decode a character */
6848 insize = 1;
6849 do
6850 {
6851 outsize = MultiByteToWideChar(code_page, flags,
6852 in, insize,
6853 buffer, Py_ARRAY_LENGTH(buffer));
6854 if (outsize > 0)
6855 break;
6856 err = GetLastError();
6857 if (err != ERROR_NO_UNICODE_TRANSLATION
6858 && err != ERROR_INSUFFICIENT_BUFFER)
6859 {
6860 PyErr_SetFromWindowsErr(0);
6861 goto error;
6862 }
6863 insize++;
6864 }
6865 /* 4=maximum length of a UTF-8 sequence */
6866 while (insize <= 4 && (in + insize) <= endin);
6867
6868 if (outsize <= 0) {
6869 Py_ssize_t startinpos, endinpos, outpos;
6870
6871 startinpos = in - startin;
6872 endinpos = startinpos + 1;
6873 outpos = out - PyUnicode_AS_UNICODE(*v);
6874 if (unicode_decode_call_errorhandler(
6875 errors, &errorHandler,
6876 encoding, reason,
6877 &startin, &endin, &startinpos, &endinpos, &exc, &in,
Victor Stinner596a6c42011-11-09 00:02:18 +01006878 v, &outpos))
Victor Stinner3a50e702011-10-18 21:21:00 +02006879 {
6880 goto error;
6881 }
Victor Stinner596a6c42011-11-09 00:02:18 +01006882 out = PyUnicode_AS_UNICODE(*v) + outpos;
Victor Stinner3a50e702011-10-18 21:21:00 +02006883 }
6884 else {
6885 in += insize;
6886 memcpy(out, buffer, outsize * sizeof(wchar_t));
6887 out += outsize;
6888 }
6889 }
6890
6891 /* write a NUL character at the end */
6892 *out = 0;
6893
6894 /* Extend unicode object */
6895 outsize = out - startout;
6896 assert(outsize <= PyUnicode_WSTR_LENGTH(*v));
Victor Stinner16e6a802011-12-12 13:24:15 +01006897 if (unicode_resize(v, outsize) < 0)
Victor Stinner3a50e702011-10-18 21:21:00 +02006898 goto error;
Victor Stinner76a31a62011-11-04 00:05:13 +01006899 ret = size;
Victor Stinner3a50e702011-10-18 21:21:00 +02006900
6901error:
6902 Py_XDECREF(encoding_obj);
6903 Py_XDECREF(errorHandler);
6904 Py_XDECREF(exc);
6905 return ret;
6906}
6907
Victor Stinner3a50e702011-10-18 21:21:00 +02006908static PyObject *
6909decode_code_page_stateful(int code_page,
Victor Stinner76a31a62011-11-04 00:05:13 +01006910 const char *s, Py_ssize_t size,
6911 const char *errors, Py_ssize_t *consumed)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006912{
Victor Stinner76a31a62011-11-04 00:05:13 +01006913 PyObject *v = NULL;
6914 int chunk_size, final, converted, done;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006915
Victor Stinner3a50e702011-10-18 21:21:00 +02006916 if (code_page < 0) {
6917 PyErr_SetString(PyExc_ValueError, "invalid code page number");
6918 return NULL;
6919 }
6920
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006921 if (consumed)
Benjamin Peterson29060642009-01-31 22:14:21 +00006922 *consumed = 0;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006923
Victor Stinner76a31a62011-11-04 00:05:13 +01006924 do
6925 {
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006926#ifdef NEED_RETRY
Victor Stinner76a31a62011-11-04 00:05:13 +01006927 if (size > INT_MAX) {
6928 chunk_size = INT_MAX;
6929 final = 0;
6930 done = 0;
6931 }
6932 else
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006933#endif
Victor Stinner76a31a62011-11-04 00:05:13 +01006934 {
6935 chunk_size = (int)size;
6936 final = (consumed == NULL);
6937 done = 1;
6938 }
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006939
Victor Stinner76a31a62011-11-04 00:05:13 +01006940 /* Skip trailing lead-byte unless 'final' is set */
6941 if (!final && is_dbcs_lead_byte(code_page, s, chunk_size - 1))
6942 --chunk_size;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006943
Victor Stinner76a31a62011-11-04 00:05:13 +01006944 if (chunk_size == 0 && done) {
6945 if (v != NULL)
6946 break;
6947 Py_INCREF(unicode_empty);
6948 return unicode_empty;
6949 }
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006950
Victor Stinner76a31a62011-11-04 00:05:13 +01006951
6952 converted = decode_code_page_strict(code_page, &v,
6953 s, chunk_size);
6954 if (converted == -2)
6955 converted = decode_code_page_errors(code_page, &v,
6956 s, chunk_size,
6957 errors);
6958 assert(converted != 0);
6959
6960 if (converted < 0) {
6961 Py_XDECREF(v);
6962 return NULL;
6963 }
6964
6965 if (consumed)
6966 *consumed += converted;
6967
6968 s += converted;
6969 size -= converted;
6970 } while (!done);
Victor Stinner3a50e702011-10-18 21:21:00 +02006971
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01006972 return unicode_result(v);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006973}
6974
Alexander Belopolsky40018472011-02-26 01:02:56 +00006975PyObject *
Victor Stinner3a50e702011-10-18 21:21:00 +02006976PyUnicode_DecodeCodePageStateful(int code_page,
6977 const char *s,
6978 Py_ssize_t size,
6979 const char *errors,
6980 Py_ssize_t *consumed)
6981{
6982 return decode_code_page_stateful(code_page, s, size, errors, consumed);
6983}
6984
6985PyObject *
6986PyUnicode_DecodeMBCSStateful(const char *s,
6987 Py_ssize_t size,
6988 const char *errors,
6989 Py_ssize_t *consumed)
6990{
6991 return decode_code_page_stateful(CP_ACP, s, size, errors, consumed);
6992}
6993
6994PyObject *
Alexander Belopolsky40018472011-02-26 01:02:56 +00006995PyUnicode_DecodeMBCS(const char *s,
6996 Py_ssize_t size,
6997 const char *errors)
Guido van Rossumb7a40ba2000-03-28 02:01:52 +00006998{
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006999 return PyUnicode_DecodeMBCSStateful(s, size, errors, NULL);
7000}
7001
Victor Stinner3a50e702011-10-18 21:21:00 +02007002static DWORD
7003encode_code_page_flags(UINT code_page, const char *errors)
7004{
7005 if (code_page == CP_UTF8) {
7006 if (winver.dwMajorVersion >= 6)
7007 /* CP_UTF8 supports WC_ERR_INVALID_CHARS on Windows Vista
7008 and later */
7009 return WC_ERR_INVALID_CHARS;
7010 else
7011 /* CP_UTF8 only supports flags=0 on Windows older than Vista */
7012 return 0;
7013 }
7014 else if (code_page == CP_UTF7) {
7015 /* CP_UTF7 only supports flags=0 */
7016 return 0;
7017 }
7018 else {
7019 if (errors != NULL && strcmp(errors, "replace") == 0)
7020 return 0;
7021 else
7022 return WC_NO_BEST_FIT_CHARS;
7023 }
7024}
7025
Thomas Wouters0e3f5912006-08-11 14:57:12 +00007026/*
Victor Stinner3a50e702011-10-18 21:21:00 +02007027 * Encode a Unicode string to a Windows code page into a byte string in strict
7028 * mode.
7029 *
7030 * Returns consumed characters if succeed, returns -2 on encode error, or raise
7031 * a WindowsError and returns -1 on other error.
Thomas Wouters0e3f5912006-08-11 14:57:12 +00007032 */
Alexander Belopolsky40018472011-02-26 01:02:56 +00007033static int
Victor Stinner3a50e702011-10-18 21:21:00 +02007034encode_code_page_strict(UINT code_page, PyObject **outbytes,
Martin v. Löwis3d325192011-11-04 18:23:06 +01007035 PyObject *unicode, Py_ssize_t offset, int len,
Victor Stinner3a50e702011-10-18 21:21:00 +02007036 const char* errors)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00007037{
Victor Stinner554f3f02010-06-16 23:33:54 +00007038 BOOL usedDefaultChar = FALSE;
Victor Stinner3a50e702011-10-18 21:21:00 +02007039 BOOL *pusedDefaultChar = &usedDefaultChar;
7040 int outsize;
Victor Stinner554f3f02010-06-16 23:33:54 +00007041 PyObject *exc = NULL;
Victor Stinner24729f32011-11-10 20:31:37 +01007042 wchar_t *p;
Victor Stinner2fc507f2011-11-04 20:06:39 +01007043 Py_ssize_t size;
Victor Stinner3a50e702011-10-18 21:21:00 +02007044 const DWORD flags = encode_code_page_flags(code_page, NULL);
7045 char *out;
Victor Stinner2fc507f2011-11-04 20:06:39 +01007046 /* Create a substring so that we can get the UTF-16 representation
7047 of just the slice under consideration. */
7048 PyObject *substring;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00007049
Martin v. Löwis3d325192011-11-04 18:23:06 +01007050 assert(len > 0);
Guido van Rossumb7a40ba2000-03-28 02:01:52 +00007051
Victor Stinner3a50e702011-10-18 21:21:00 +02007052 if (code_page != CP_UTF8 && code_page != CP_UTF7)
Victor Stinner554f3f02010-06-16 23:33:54 +00007053 pusedDefaultChar = &usedDefaultChar;
Victor Stinner3a50e702011-10-18 21:21:00 +02007054 else
Victor Stinner554f3f02010-06-16 23:33:54 +00007055 pusedDefaultChar = NULL;
Victor Stinner554f3f02010-06-16 23:33:54 +00007056
Victor Stinner2fc507f2011-11-04 20:06:39 +01007057 substring = PyUnicode_Substring(unicode, offset, offset+len);
7058 if (substring == NULL)
7059 return -1;
7060 p = PyUnicode_AsUnicodeAndSize(substring, &size);
7061 if (p == NULL) {
7062 Py_DECREF(substring);
7063 return -1;
7064 }
Martin v. Löwis3d325192011-11-04 18:23:06 +01007065
Guido van Rossumb7a40ba2000-03-28 02:01:52 +00007066 /* First get the size of the result */
Victor Stinner3a50e702011-10-18 21:21:00 +02007067 outsize = WideCharToMultiByte(code_page, flags,
7068 p, size,
7069 NULL, 0,
7070 NULL, pusedDefaultChar);
7071 if (outsize <= 0)
7072 goto error;
7073 /* If we used a default char, then we failed! */
Victor Stinner2fc507f2011-11-04 20:06:39 +01007074 if (pusedDefaultChar && *pusedDefaultChar) {
7075 Py_DECREF(substring);
Victor Stinner3a50e702011-10-18 21:21:00 +02007076 return -2;
Victor Stinner2fc507f2011-11-04 20:06:39 +01007077 }
Guido van Rossumb7a40ba2000-03-28 02:01:52 +00007078
Victor Stinner3a50e702011-10-18 21:21:00 +02007079 if (*outbytes == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +00007080 /* Create string object */
Victor Stinner3a50e702011-10-18 21:21:00 +02007081 *outbytes = PyBytes_FromStringAndSize(NULL, outsize);
Victor Stinner2fc507f2011-11-04 20:06:39 +01007082 if (*outbytes == NULL) {
7083 Py_DECREF(substring);
Benjamin Peterson29060642009-01-31 22:14:21 +00007084 return -1;
Victor Stinner2fc507f2011-11-04 20:06:39 +01007085 }
Victor Stinner3a50e702011-10-18 21:21:00 +02007086 out = PyBytes_AS_STRING(*outbytes);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00007087 }
7088 else {
Benjamin Peterson29060642009-01-31 22:14:21 +00007089 /* Extend string object */
Victor Stinner3a50e702011-10-18 21:21:00 +02007090 const Py_ssize_t n = PyBytes_Size(*outbytes);
7091 if (outsize > PY_SSIZE_T_MAX - n) {
7092 PyErr_NoMemory();
Victor Stinner2fc507f2011-11-04 20:06:39 +01007093 Py_DECREF(substring);
Benjamin Peterson29060642009-01-31 22:14:21 +00007094 return -1;
Victor Stinner3a50e702011-10-18 21:21:00 +02007095 }
Victor Stinner2fc507f2011-11-04 20:06:39 +01007096 if (_PyBytes_Resize(outbytes, n + outsize) < 0) {
7097 Py_DECREF(substring);
Victor Stinner3a50e702011-10-18 21:21:00 +02007098 return -1;
Victor Stinner2fc507f2011-11-04 20:06:39 +01007099 }
Victor Stinner3a50e702011-10-18 21:21:00 +02007100 out = PyBytes_AS_STRING(*outbytes) + n;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00007101 }
7102
7103 /* Do the conversion */
Victor Stinner3a50e702011-10-18 21:21:00 +02007104 outsize = WideCharToMultiByte(code_page, flags,
7105 p, size,
7106 out, outsize,
7107 NULL, pusedDefaultChar);
Victor Stinner2fc507f2011-11-04 20:06:39 +01007108 Py_CLEAR(substring);
Victor Stinner3a50e702011-10-18 21:21:00 +02007109 if (outsize <= 0)
7110 goto error;
7111 if (pusedDefaultChar && *pusedDefaultChar)
7112 return -2;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00007113 return 0;
Victor Stinner554f3f02010-06-16 23:33:54 +00007114
Victor Stinner3a50e702011-10-18 21:21:00 +02007115error:
Victor Stinner2fc507f2011-11-04 20:06:39 +01007116 Py_XDECREF(substring);
Victor Stinner3a50e702011-10-18 21:21:00 +02007117 if (GetLastError() == ERROR_NO_UNICODE_TRANSLATION)
7118 return -2;
7119 PyErr_SetFromWindowsErr(0);
Victor Stinner554f3f02010-06-16 23:33:54 +00007120 return -1;
Guido van Rossumb7a40ba2000-03-28 02:01:52 +00007121}
7122
Victor Stinner3a50e702011-10-18 21:21:00 +02007123/*
7124 * Encode a Unicode string to a Windows code page into a byte string using a
7125 * error handler.
7126 *
7127 * Returns consumed characters if succeed, or raise a WindowsError and returns
7128 * -1 on other error.
7129 */
7130static int
7131encode_code_page_errors(UINT code_page, PyObject **outbytes,
Victor Stinner7581cef2011-11-03 22:32:33 +01007132 PyObject *unicode, Py_ssize_t unicode_offset,
Martin v. Löwis3d325192011-11-04 18:23:06 +01007133 Py_ssize_t insize, const char* errors)
Guido van Rossumb7a40ba2000-03-28 02:01:52 +00007134{
Victor Stinner3a50e702011-10-18 21:21:00 +02007135 const DWORD flags = encode_code_page_flags(code_page, errors);
Victor Stinner2fc507f2011-11-04 20:06:39 +01007136 Py_ssize_t pos = unicode_offset;
7137 Py_ssize_t endin = unicode_offset + insize;
Victor Stinner3a50e702011-10-18 21:21:00 +02007138 /* Ideally, we should get reason from FormatMessage. This is the Windows
7139 2000 English version of the message. */
7140 const char *reason = "invalid character";
7141 /* 4=maximum length of a UTF-8 sequence */
7142 char buffer[4];
7143 BOOL usedDefaultChar = FALSE, *pusedDefaultChar;
7144 Py_ssize_t outsize;
7145 char *out;
Victor Stinner3a50e702011-10-18 21:21:00 +02007146 PyObject *errorHandler = NULL;
7147 PyObject *exc = NULL;
7148 PyObject *encoding_obj = NULL;
7149 char *encoding;
Martin v. Löwis3d325192011-11-04 18:23:06 +01007150 Py_ssize_t newpos, newoutsize;
Victor Stinner3a50e702011-10-18 21:21:00 +02007151 PyObject *rep;
7152 int ret = -1;
7153
7154 assert(insize > 0);
7155
7156 encoding = code_page_name(code_page, &encoding_obj);
7157 if (encoding == NULL)
7158 return -1;
7159
7160 if (errors == NULL || strcmp(errors, "strict") == 0) {
7161 /* The last error was ERROR_NO_UNICODE_TRANSLATION,
7162 then we raise a UnicodeEncodeError. */
Martin v. Löwis12be46c2011-11-04 19:04:15 +01007163 make_encode_exception(&exc, encoding, unicode, 0, 0, reason);
Victor Stinner3a50e702011-10-18 21:21:00 +02007164 if (exc != NULL) {
7165 PyCodec_StrictErrors(exc);
7166 Py_DECREF(exc);
7167 }
7168 Py_XDECREF(encoding_obj);
7169 return -1;
7170 }
7171
7172 if (code_page != CP_UTF8 && code_page != CP_UTF7)
7173 pusedDefaultChar = &usedDefaultChar;
7174 else
7175 pusedDefaultChar = NULL;
7176
7177 if (Py_ARRAY_LENGTH(buffer) > PY_SSIZE_T_MAX / insize) {
7178 PyErr_NoMemory();
7179 goto error;
7180 }
7181 outsize = insize * Py_ARRAY_LENGTH(buffer);
7182
7183 if (*outbytes == NULL) {
7184 /* Create string object */
7185 *outbytes = PyBytes_FromStringAndSize(NULL, outsize);
7186 if (*outbytes == NULL)
7187 goto error;
7188 out = PyBytes_AS_STRING(*outbytes);
7189 }
7190 else {
7191 /* Extend string object */
7192 Py_ssize_t n = PyBytes_Size(*outbytes);
7193 if (n > PY_SSIZE_T_MAX - outsize) {
7194 PyErr_NoMemory();
7195 goto error;
7196 }
7197 if (_PyBytes_Resize(outbytes, n + outsize) < 0)
7198 goto error;
7199 out = PyBytes_AS_STRING(*outbytes) + n;
7200 }
7201
7202 /* Encode the string character per character */
Martin v. Löwis3d325192011-11-04 18:23:06 +01007203 while (pos < endin)
Victor Stinner3a50e702011-10-18 21:21:00 +02007204 {
Victor Stinner2fc507f2011-11-04 20:06:39 +01007205 Py_UCS4 ch = PyUnicode_READ_CHAR(unicode, pos);
7206 wchar_t chars[2];
7207 int charsize;
7208 if (ch < 0x10000) {
7209 chars[0] = (wchar_t)ch;
7210 charsize = 1;
7211 }
7212 else {
7213 ch -= 0x10000;
7214 chars[0] = 0xd800 + (ch >> 10);
7215 chars[1] = 0xdc00 + (ch & 0x3ff);
7216 charsize = 2;
7217 }
7218
Victor Stinner3a50e702011-10-18 21:21:00 +02007219 outsize = WideCharToMultiByte(code_page, flags,
Martin v. Löwis3d325192011-11-04 18:23:06 +01007220 chars, charsize,
Victor Stinner3a50e702011-10-18 21:21:00 +02007221 buffer, Py_ARRAY_LENGTH(buffer),
7222 NULL, pusedDefaultChar);
7223 if (outsize > 0) {
7224 if (pusedDefaultChar == NULL || !(*pusedDefaultChar))
7225 {
Martin v. Löwis3d325192011-11-04 18:23:06 +01007226 pos++;
Victor Stinner3a50e702011-10-18 21:21:00 +02007227 memcpy(out, buffer, outsize);
7228 out += outsize;
7229 continue;
7230 }
7231 }
7232 else if (GetLastError() != ERROR_NO_UNICODE_TRANSLATION) {
7233 PyErr_SetFromWindowsErr(0);
7234 goto error;
7235 }
7236
Victor Stinner3a50e702011-10-18 21:21:00 +02007237 rep = unicode_encode_call_errorhandler(
7238 errors, &errorHandler, encoding, reason,
Victor Stinner7581cef2011-11-03 22:32:33 +01007239 unicode, &exc,
Martin v. Löwis3d325192011-11-04 18:23:06 +01007240 pos, pos + 1, &newpos);
Victor Stinner3a50e702011-10-18 21:21:00 +02007241 if (rep == NULL)
7242 goto error;
Martin v. Löwis3d325192011-11-04 18:23:06 +01007243 pos = newpos;
Victor Stinner3a50e702011-10-18 21:21:00 +02007244
7245 if (PyBytes_Check(rep)) {
7246 outsize = PyBytes_GET_SIZE(rep);
7247 if (outsize != 1) {
7248 Py_ssize_t offset = out - PyBytes_AS_STRING(*outbytes);
7249 newoutsize = PyBytes_GET_SIZE(*outbytes) + (outsize - 1);
7250 if (_PyBytes_Resize(outbytes, newoutsize) < 0) {
7251 Py_DECREF(rep);
7252 goto error;
7253 }
7254 out = PyBytes_AS_STRING(*outbytes) + offset;
7255 }
7256 memcpy(out, PyBytes_AS_STRING(rep), outsize);
7257 out += outsize;
7258 }
7259 else {
7260 Py_ssize_t i;
7261 enum PyUnicode_Kind kind;
7262 void *data;
7263
Benjamin Petersonbac79492012-01-14 13:34:47 -05007264 if (PyUnicode_READY(rep) == -1) {
Victor Stinner3a50e702011-10-18 21:21:00 +02007265 Py_DECREF(rep);
7266 goto error;
7267 }
7268
7269 outsize = PyUnicode_GET_LENGTH(rep);
7270 if (outsize != 1) {
7271 Py_ssize_t offset = out - PyBytes_AS_STRING(*outbytes);
7272 newoutsize = PyBytes_GET_SIZE(*outbytes) + (outsize - 1);
7273 if (_PyBytes_Resize(outbytes, newoutsize) < 0) {
7274 Py_DECREF(rep);
7275 goto error;
7276 }
7277 out = PyBytes_AS_STRING(*outbytes) + offset;
7278 }
7279 kind = PyUnicode_KIND(rep);
7280 data = PyUnicode_DATA(rep);
7281 for (i=0; i < outsize; i++) {
7282 Py_UCS4 ch = PyUnicode_READ(kind, data, i);
7283 if (ch > 127) {
Martin v. Löwis12be46c2011-11-04 19:04:15 +01007284 raise_encode_exception(&exc,
Martin v. Löwis3d325192011-11-04 18:23:06 +01007285 encoding, unicode,
7286 pos, pos + 1,
Victor Stinner3a50e702011-10-18 21:21:00 +02007287 "unable to encode error handler result to ASCII");
7288 Py_DECREF(rep);
7289 goto error;
7290 }
7291 *out = (unsigned char)ch;
7292 out++;
7293 }
7294 }
7295 Py_DECREF(rep);
7296 }
7297 /* write a NUL byte */
7298 *out = 0;
7299 outsize = out - PyBytes_AS_STRING(*outbytes);
7300 assert(outsize <= PyBytes_GET_SIZE(*outbytes));
7301 if (_PyBytes_Resize(outbytes, outsize) < 0)
7302 goto error;
7303 ret = 0;
7304
7305error:
7306 Py_XDECREF(encoding_obj);
7307 Py_XDECREF(errorHandler);
7308 Py_XDECREF(exc);
7309 return ret;
7310}
7311
Victor Stinner3a50e702011-10-18 21:21:00 +02007312static PyObject *
7313encode_code_page(int code_page,
Victor Stinner7581cef2011-11-03 22:32:33 +01007314 PyObject *unicode,
Victor Stinner3a50e702011-10-18 21:21:00 +02007315 const char *errors)
7316{
Martin v. Löwis3d325192011-11-04 18:23:06 +01007317 Py_ssize_t len;
Victor Stinner3a50e702011-10-18 21:21:00 +02007318 PyObject *outbytes = NULL;
Victor Stinner7581cef2011-11-03 22:32:33 +01007319 Py_ssize_t offset;
Victor Stinner76a31a62011-11-04 00:05:13 +01007320 int chunk_len, ret, done;
Victor Stinner7581cef2011-11-03 22:32:33 +01007321
Benjamin Petersonbac79492012-01-14 13:34:47 -05007322 if (PyUnicode_READY(unicode) == -1)
Victor Stinner2fc507f2011-11-04 20:06:39 +01007323 return NULL;
7324 len = PyUnicode_GET_LENGTH(unicode);
Guido van Rossum03e29f12000-05-04 15:52:20 +00007325
Victor Stinner3a50e702011-10-18 21:21:00 +02007326 if (code_page < 0) {
7327 PyErr_SetString(PyExc_ValueError, "invalid code page number");
7328 return NULL;
7329 }
7330
Martin v. Löwis3d325192011-11-04 18:23:06 +01007331 if (len == 0)
Victor Stinner76a31a62011-11-04 00:05:13 +01007332 return PyBytes_FromStringAndSize(NULL, 0);
7333
Victor Stinner7581cef2011-11-03 22:32:33 +01007334 offset = 0;
7335 do
7336 {
Thomas Wouters0e3f5912006-08-11 14:57:12 +00007337#ifdef NEED_RETRY
Victor Stinner2fc507f2011-11-04 20:06:39 +01007338 /* UTF-16 encoding may double the size, so use only INT_MAX/2
Martin v. Löwis3d325192011-11-04 18:23:06 +01007339 chunks. */
7340 if (len > INT_MAX/2) {
7341 chunk_len = INT_MAX/2;
Victor Stinner76a31a62011-11-04 00:05:13 +01007342 done = 0;
7343 }
Victor Stinner7581cef2011-11-03 22:32:33 +01007344 else
Thomas Wouters0e3f5912006-08-11 14:57:12 +00007345#endif
Victor Stinner76a31a62011-11-04 00:05:13 +01007346 {
Martin v. Löwis3d325192011-11-04 18:23:06 +01007347 chunk_len = (int)len;
Victor Stinner76a31a62011-11-04 00:05:13 +01007348 done = 1;
7349 }
Victor Stinner2fc507f2011-11-04 20:06:39 +01007350
Victor Stinner76a31a62011-11-04 00:05:13 +01007351 ret = encode_code_page_strict(code_page, &outbytes,
Martin v. Löwis3d325192011-11-04 18:23:06 +01007352 unicode, offset, chunk_len,
Victor Stinner76a31a62011-11-04 00:05:13 +01007353 errors);
7354 if (ret == -2)
7355 ret = encode_code_page_errors(code_page, &outbytes,
7356 unicode, offset,
Martin v. Löwis3d325192011-11-04 18:23:06 +01007357 chunk_len, errors);
Victor Stinner7581cef2011-11-03 22:32:33 +01007358 if (ret < 0) {
7359 Py_XDECREF(outbytes);
7360 return NULL;
7361 }
Thomas Wouters0e3f5912006-08-11 14:57:12 +00007362
Victor Stinner7581cef2011-11-03 22:32:33 +01007363 offset += chunk_len;
Martin v. Löwis3d325192011-11-04 18:23:06 +01007364 len -= chunk_len;
Victor Stinner76a31a62011-11-04 00:05:13 +01007365 } while (!done);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00007366
Victor Stinner3a50e702011-10-18 21:21:00 +02007367 return outbytes;
7368}
7369
7370PyObject *
7371PyUnicode_EncodeMBCS(const Py_UNICODE *p,
7372 Py_ssize_t size,
7373 const char *errors)
7374{
Victor Stinner7581cef2011-11-03 22:32:33 +01007375 PyObject *unicode, *res;
7376 unicode = PyUnicode_FromUnicode(p, size);
7377 if (unicode == NULL)
7378 return NULL;
7379 res = encode_code_page(CP_ACP, unicode, errors);
7380 Py_DECREF(unicode);
7381 return res;
Victor Stinner3a50e702011-10-18 21:21:00 +02007382}
7383
7384PyObject *
7385PyUnicode_EncodeCodePage(int code_page,
7386 PyObject *unicode,
7387 const char *errors)
7388{
Victor Stinner7581cef2011-11-03 22:32:33 +01007389 return encode_code_page(code_page, unicode, errors);
Guido van Rossumb7a40ba2000-03-28 02:01:52 +00007390}
Guido van Rossum2ea3e142000-03-31 17:24:09 +00007391
Alexander Belopolsky40018472011-02-26 01:02:56 +00007392PyObject *
7393PyUnicode_AsMBCSString(PyObject *unicode)
Mark Hammond0ccda1e2003-07-01 00:13:27 +00007394{
7395 if (!PyUnicode_Check(unicode)) {
7396 PyErr_BadArgument();
7397 return NULL;
7398 }
Victor Stinner7581cef2011-11-03 22:32:33 +01007399 return PyUnicode_EncodeCodePage(CP_ACP, unicode, NULL);
Mark Hammond0ccda1e2003-07-01 00:13:27 +00007400}
7401
Thomas Wouters0e3f5912006-08-11 14:57:12 +00007402#undef NEED_RETRY
7403
Victor Stinner99b95382011-07-04 14:23:54 +02007404#endif /* HAVE_MBCS */
Guido van Rossumb7a40ba2000-03-28 02:01:52 +00007405
Guido van Rossumd57fd912000-03-10 22:53:23 +00007406/* --- Character Mapping Codec -------------------------------------------- */
7407
Alexander Belopolsky40018472011-02-26 01:02:56 +00007408PyObject *
7409PyUnicode_DecodeCharmap(const char *s,
7410 Py_ssize_t size,
7411 PyObject *mapping,
7412 const char *errors)
Guido van Rossumd57fd912000-03-10 22:53:23 +00007413{
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007414 const char *starts = s;
Martin v. Löwis18e16552006-02-15 17:27:45 +00007415 Py_ssize_t startinpos;
7416 Py_ssize_t endinpos;
7417 Py_ssize_t outpos;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007418 const char *e;
Victor Stinner7931d9a2011-11-04 00:22:48 +01007419 PyObject *v;
Martin v. Löwis18e16552006-02-15 17:27:45 +00007420 Py_ssize_t extrachars = 0;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007421 PyObject *errorHandler = NULL;
7422 PyObject *exc = NULL;
Tim Petersced69f82003-09-16 20:30:58 +00007423
Guido van Rossumd57fd912000-03-10 22:53:23 +00007424 /* Default to Latin-1 */
7425 if (mapping == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00007426 return PyUnicode_DecodeLatin1(s, size, errors);
Guido van Rossumd57fd912000-03-10 22:53:23 +00007427
Martin v. Löwise9b11c12011-11-08 17:35:34 +01007428 v = PyUnicode_New(size, 127);
Guido van Rossumd57fd912000-03-10 22:53:23 +00007429 if (v == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00007430 goto onError;
Guido van Rossumd57fd912000-03-10 22:53:23 +00007431 if (size == 0)
Victor Stinner7931d9a2011-11-04 00:22:48 +01007432 return v;
Martin v. Löwise9b11c12011-11-08 17:35:34 +01007433 outpos = 0;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007434 e = s + size;
Walter Dörwaldd1c1e102005-10-06 20:29:57 +00007435 if (PyUnicode_CheckExact(mapping)) {
Victor Stinnerebf3ba82011-11-10 20:30:22 +01007436 Py_ssize_t maplen;
Antoine Pitrouaaefac72012-06-16 22:48:21 +02007437 enum PyUnicode_Kind mapkind;
7438 void *mapdata;
Victor Stinnerebf3ba82011-11-10 20:30:22 +01007439 Py_UCS4 x;
7440
Benjamin Petersonbac79492012-01-14 13:34:47 -05007441 if (PyUnicode_READY(mapping) == -1)
Victor Stinnerebf3ba82011-11-10 20:30:22 +01007442 return NULL;
7443
7444 maplen = PyUnicode_GET_LENGTH(mapping);
Antoine Pitrouaaefac72012-06-16 22:48:21 +02007445 mapdata = PyUnicode_DATA(mapping);
7446 mapkind = PyUnicode_KIND(mapping);
Benjamin Peterson29060642009-01-31 22:14:21 +00007447 while (s < e) {
Antoine Pitrouaaefac72012-06-16 22:48:21 +02007448 unsigned char ch;
7449 if (mapkind == PyUnicode_2BYTE_KIND && maplen >= 256) {
7450 enum PyUnicode_Kind outkind = PyUnicode_KIND(v);
7451 if (outkind == PyUnicode_1BYTE_KIND) {
7452 void *outdata = PyUnicode_DATA(v);
7453 Py_UCS4 maxchar = PyUnicode_MAX_CHAR_VALUE(v);
7454 while (s < e) {
7455 unsigned char ch = *s;
7456 x = PyUnicode_READ(PyUnicode_2BYTE_KIND, mapdata, ch);
7457 if (x > maxchar)
7458 goto Error;
7459 PyUnicode_WRITE(PyUnicode_1BYTE_KIND, outdata, outpos++, x);
7460 ++s;
7461 }
7462 break;
7463 }
7464 else if (outkind == PyUnicode_2BYTE_KIND) {
7465 void *outdata = PyUnicode_DATA(v);
7466 while (s < e) {
7467 unsigned char ch = *s;
7468 x = PyUnicode_READ(PyUnicode_2BYTE_KIND, mapdata, ch);
7469 if (x == 0xFFFE)
7470 goto Error;
7471 PyUnicode_WRITE(PyUnicode_2BYTE_KIND, outdata, outpos++, x);
7472 ++s;
7473 }
7474 break;
7475 }
7476 }
7477 ch = *s;
Guido van Rossumd57fd912000-03-10 22:53:23 +00007478
Benjamin Peterson29060642009-01-31 22:14:21 +00007479 if (ch < maplen)
Antoine Pitrouaaefac72012-06-16 22:48:21 +02007480 x = PyUnicode_READ(mapkind, mapdata, ch);
Victor Stinnerebf3ba82011-11-10 20:30:22 +01007481 else
7482 x = 0xfffe; /* invalid value */
Antoine Pitrouaaefac72012-06-16 22:48:21 +02007483Error:
Victor Stinnerebf3ba82011-11-10 20:30:22 +01007484 if (x == 0xfffe)
7485 {
Benjamin Peterson29060642009-01-31 22:14:21 +00007486 /* undefined mapping */
Benjamin Peterson29060642009-01-31 22:14:21 +00007487 startinpos = s-starts;
7488 endinpos = startinpos+1;
7489 if (unicode_decode_call_errorhandler(
7490 errors, &errorHandler,
7491 "charmap", "character maps to <undefined>",
7492 &starts, &e, &startinpos, &endinpos, &exc, &s,
Martin v. Löwise9b11c12011-11-08 17:35:34 +01007493 &v, &outpos)) {
Benjamin Peterson29060642009-01-31 22:14:21 +00007494 goto onError;
7495 }
7496 continue;
7497 }
Victor Stinnerebf3ba82011-11-10 20:30:22 +01007498
Martin v. Löwise9b11c12011-11-08 17:35:34 +01007499 if (unicode_putchar(&v, &outpos, x) < 0)
7500 goto onError;
Benjamin Peterson29060642009-01-31 22:14:21 +00007501 ++s;
Benjamin Peterson14339b62009-01-31 16:36:08 +00007502 }
Walter Dörwaldd1c1e102005-10-06 20:29:57 +00007503 }
7504 else {
Benjamin Peterson29060642009-01-31 22:14:21 +00007505 while (s < e) {
7506 unsigned char ch = *s;
7507 PyObject *w, *x;
Walter Dörwaldd1c1e102005-10-06 20:29:57 +00007508
Benjamin Peterson29060642009-01-31 22:14:21 +00007509 /* Get mapping (char ordinal -> integer, Unicode char or None) */
7510 w = PyLong_FromLong((long)ch);
7511 if (w == NULL)
7512 goto onError;
7513 x = PyObject_GetItem(mapping, w);
7514 Py_DECREF(w);
7515 if (x == NULL) {
7516 if (PyErr_ExceptionMatches(PyExc_LookupError)) {
7517 /* No mapping found means: mapping is undefined. */
7518 PyErr_Clear();
7519 x = Py_None;
7520 Py_INCREF(x);
7521 } else
7522 goto onError;
7523 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00007524
Benjamin Peterson29060642009-01-31 22:14:21 +00007525 /* Apply mapping */
7526 if (PyLong_Check(x)) {
7527 long value = PyLong_AS_LONG(x);
Antoine Pitroua1f76552012-09-23 20:00:04 +02007528 if (value < 0 || value > MAX_UNICODE) {
7529 PyErr_Format(PyExc_TypeError,
7530 "character mapping must be in range(0x%lx)",
7531 (unsigned long)MAX_UNICODE + 1);
Benjamin Peterson29060642009-01-31 22:14:21 +00007532 Py_DECREF(x);
7533 goto onError;
7534 }
Martin v. Löwise9b11c12011-11-08 17:35:34 +01007535 if (unicode_putchar(&v, &outpos, value) < 0)
7536 goto onError;
Benjamin Peterson29060642009-01-31 22:14:21 +00007537 }
7538 else if (x == Py_None) {
7539 /* undefined mapping */
Benjamin Peterson29060642009-01-31 22:14:21 +00007540 startinpos = s-starts;
7541 endinpos = startinpos+1;
7542 if (unicode_decode_call_errorhandler(
7543 errors, &errorHandler,
7544 "charmap", "character maps to <undefined>",
7545 &starts, &e, &startinpos, &endinpos, &exc, &s,
Martin v. Löwise9b11c12011-11-08 17:35:34 +01007546 &v, &outpos)) {
Benjamin Peterson29060642009-01-31 22:14:21 +00007547 Py_DECREF(x);
7548 goto onError;
7549 }
7550 Py_DECREF(x);
7551 continue;
7552 }
7553 else if (PyUnicode_Check(x)) {
Martin v. Löwise9b11c12011-11-08 17:35:34 +01007554 Py_ssize_t targetsize;
Benjamin Peterson14339b62009-01-31 16:36:08 +00007555
Benjamin Petersonbac79492012-01-14 13:34:47 -05007556 if (PyUnicode_READY(x) == -1)
Martin v. Löwise9b11c12011-11-08 17:35:34 +01007557 goto onError;
7558 targetsize = PyUnicode_GET_LENGTH(x);
7559
7560 if (targetsize == 1) {
Benjamin Peterson29060642009-01-31 22:14:21 +00007561 /* 1-1 mapping */
Victor Stinner62aa4d02011-11-09 00:03:45 +01007562 if (unicode_putchar(&v, &outpos,
Martin v. Löwise9b11c12011-11-08 17:35:34 +01007563 PyUnicode_READ_CHAR(x, 0)) < 0)
7564 goto onError;
7565 }
Benjamin Peterson29060642009-01-31 22:14:21 +00007566 else if (targetsize > 1) {
7567 /* 1-n mapping */
7568 if (targetsize > extrachars) {
7569 /* resize first */
Benjamin Peterson29060642009-01-31 22:14:21 +00007570 Py_ssize_t needed = (targetsize - extrachars) + \
7571 (targetsize << 2);
7572 extrachars += needed;
7573 /* XXX overflow detection missing */
Victor Stinner16e6a802011-12-12 13:24:15 +01007574 if (unicode_resize(&v,
7575 PyUnicode_GET_LENGTH(v) + needed) < 0)
7576 {
Benjamin Peterson29060642009-01-31 22:14:21 +00007577 Py_DECREF(x);
7578 goto onError;
7579 }
Benjamin Peterson29060642009-01-31 22:14:21 +00007580 }
Victor Stinner1b487b42012-05-03 12:29:04 +02007581 if (unicode_widen(&v, outpos, PyUnicode_MAX_CHAR_VALUE(x)) < 0)
Martin v. Löwise9b11c12011-11-08 17:35:34 +01007582 goto onError;
7583 PyUnicode_CopyCharacters(v, outpos, x, 0, targetsize);
7584 outpos += targetsize;
Benjamin Peterson29060642009-01-31 22:14:21 +00007585 extrachars -= targetsize;
7586 }
7587 /* 1-0 mapping: skip the character */
7588 }
7589 else {
7590 /* wrong return value */
7591 PyErr_SetString(PyExc_TypeError,
7592 "character mapping must return integer, None or str");
Benjamin Peterson14339b62009-01-31 16:36:08 +00007593 Py_DECREF(x);
7594 goto onError;
7595 }
Benjamin Peterson29060642009-01-31 22:14:21 +00007596 Py_DECREF(x);
7597 ++s;
Benjamin Peterson14339b62009-01-31 16:36:08 +00007598 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00007599 }
Victor Stinner16e6a802011-12-12 13:24:15 +01007600 if (unicode_resize(&v, outpos) < 0)
Antoine Pitroua8f63c02011-11-08 18:37:16 +01007601 goto onError;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007602 Py_XDECREF(errorHandler);
7603 Py_XDECREF(exc);
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01007604 return unicode_result(v);
Tim Petersced69f82003-09-16 20:30:58 +00007605
Benjamin Peterson29060642009-01-31 22:14:21 +00007606 onError:
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007607 Py_XDECREF(errorHandler);
7608 Py_XDECREF(exc);
Guido van Rossumd57fd912000-03-10 22:53:23 +00007609 Py_XDECREF(v);
7610 return NULL;
7611}
7612
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007613/* Charmap encoding: the lookup table */
7614
Alexander Belopolsky40018472011-02-26 01:02:56 +00007615struct encoding_map {
Benjamin Peterson29060642009-01-31 22:14:21 +00007616 PyObject_HEAD
7617 unsigned char level1[32];
7618 int count2, count3;
7619 unsigned char level23[1];
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007620};
7621
7622static PyObject*
7623encoding_map_size(PyObject *obj, PyObject* args)
7624{
7625 struct encoding_map *map = (struct encoding_map*)obj;
Benjamin Peterson14339b62009-01-31 16:36:08 +00007626 return PyLong_FromLong(sizeof(*map) - 1 + 16*map->count2 +
Benjamin Peterson29060642009-01-31 22:14:21 +00007627 128*map->count3);
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007628}
7629
7630static PyMethodDef encoding_map_methods[] = {
Benjamin Peterson14339b62009-01-31 16:36:08 +00007631 {"size", encoding_map_size, METH_NOARGS,
Benjamin Peterson29060642009-01-31 22:14:21 +00007632 PyDoc_STR("Return the size (in bytes) of this object") },
7633 { 0 }
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007634};
7635
7636static void
7637encoding_map_dealloc(PyObject* o)
7638{
Benjamin Peterson14339b62009-01-31 16:36:08 +00007639 PyObject_FREE(o);
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007640}
7641
7642static PyTypeObject EncodingMapType = {
Benjamin Peterson14339b62009-01-31 16:36:08 +00007643 PyVarObject_HEAD_INIT(NULL, 0)
Benjamin Peterson29060642009-01-31 22:14:21 +00007644 "EncodingMap", /*tp_name*/
7645 sizeof(struct encoding_map), /*tp_basicsize*/
7646 0, /*tp_itemsize*/
7647 /* methods */
7648 encoding_map_dealloc, /*tp_dealloc*/
7649 0, /*tp_print*/
7650 0, /*tp_getattr*/
7651 0, /*tp_setattr*/
Mark Dickinsone94c6792009-02-02 20:36:42 +00007652 0, /*tp_reserved*/
Benjamin Peterson29060642009-01-31 22:14:21 +00007653 0, /*tp_repr*/
7654 0, /*tp_as_number*/
7655 0, /*tp_as_sequence*/
7656 0, /*tp_as_mapping*/
7657 0, /*tp_hash*/
7658 0, /*tp_call*/
7659 0, /*tp_str*/
7660 0, /*tp_getattro*/
7661 0, /*tp_setattro*/
7662 0, /*tp_as_buffer*/
7663 Py_TPFLAGS_DEFAULT, /*tp_flags*/
7664 0, /*tp_doc*/
7665 0, /*tp_traverse*/
7666 0, /*tp_clear*/
7667 0, /*tp_richcompare*/
7668 0, /*tp_weaklistoffset*/
7669 0, /*tp_iter*/
7670 0, /*tp_iternext*/
7671 encoding_map_methods, /*tp_methods*/
7672 0, /*tp_members*/
7673 0, /*tp_getset*/
7674 0, /*tp_base*/
7675 0, /*tp_dict*/
7676 0, /*tp_descr_get*/
7677 0, /*tp_descr_set*/
7678 0, /*tp_dictoffset*/
7679 0, /*tp_init*/
7680 0, /*tp_alloc*/
7681 0, /*tp_new*/
7682 0, /*tp_free*/
7683 0, /*tp_is_gc*/
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007684};
7685
7686PyObject*
7687PyUnicode_BuildEncodingMap(PyObject* string)
7688{
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007689 PyObject *result;
7690 struct encoding_map *mresult;
7691 int i;
7692 int need_dict = 0;
7693 unsigned char level1[32];
7694 unsigned char level2[512];
7695 unsigned char *mlevel1, *mlevel2, *mlevel3;
7696 int count2 = 0, count3 = 0;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02007697 int kind;
7698 void *data;
Antoine Pitrouaaefac72012-06-16 22:48:21 +02007699 Py_ssize_t length;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02007700 Py_UCS4 ch;
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007701
Antoine Pitrouaaefac72012-06-16 22:48:21 +02007702 if (!PyUnicode_Check(string) || !PyUnicode_GET_LENGTH(string)) {
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007703 PyErr_BadArgument();
7704 return NULL;
7705 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02007706 kind = PyUnicode_KIND(string);
7707 data = PyUnicode_DATA(string);
Antoine Pitrouaaefac72012-06-16 22:48:21 +02007708 length = PyUnicode_GET_LENGTH(string);
7709 length = Py_MIN(length, 256);
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007710 memset(level1, 0xFF, sizeof level1);
7711 memset(level2, 0xFF, sizeof level2);
7712
7713 /* If there isn't a one-to-one mapping of NULL to \0,
7714 or if there are non-BMP characters, we need to use
7715 a mapping dictionary. */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02007716 if (PyUnicode_READ(kind, data, 0) != 0)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007717 need_dict = 1;
Antoine Pitrouaaefac72012-06-16 22:48:21 +02007718 for (i = 1; i < length; i++) {
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007719 int l1, l2;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02007720 ch = PyUnicode_READ(kind, data, i);
7721 if (ch == 0 || ch > 0xFFFF) {
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007722 need_dict = 1;
7723 break;
7724 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02007725 if (ch == 0xFFFE)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007726 /* unmapped character */
7727 continue;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02007728 l1 = ch >> 11;
7729 l2 = ch >> 7;
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007730 if (level1[l1] == 0xFF)
7731 level1[l1] = count2++;
7732 if (level2[l2] == 0xFF)
Benjamin Peterson14339b62009-01-31 16:36:08 +00007733 level2[l2] = count3++;
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007734 }
7735
7736 if (count2 >= 0xFF || count3 >= 0xFF)
7737 need_dict = 1;
7738
7739 if (need_dict) {
7740 PyObject *result = PyDict_New();
7741 PyObject *key, *value;
7742 if (!result)
7743 return NULL;
Antoine Pitrouaaefac72012-06-16 22:48:21 +02007744 for (i = 0; i < length; i++) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02007745 key = PyLong_FromLong(PyUnicode_READ(kind, data, i));
Christian Heimes217cfd12007-12-02 14:31:20 +00007746 value = PyLong_FromLong(i);
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007747 if (!key || !value)
7748 goto failed1;
7749 if (PyDict_SetItem(result, key, value) == -1)
7750 goto failed1;
7751 Py_DECREF(key);
7752 Py_DECREF(value);
7753 }
7754 return result;
7755 failed1:
7756 Py_XDECREF(key);
7757 Py_XDECREF(value);
7758 Py_DECREF(result);
7759 return NULL;
7760 }
7761
7762 /* Create a three-level trie */
7763 result = PyObject_MALLOC(sizeof(struct encoding_map) +
7764 16*count2 + 128*count3 - 1);
7765 if (!result)
7766 return PyErr_NoMemory();
7767 PyObject_Init(result, &EncodingMapType);
7768 mresult = (struct encoding_map*)result;
7769 mresult->count2 = count2;
7770 mresult->count3 = count3;
7771 mlevel1 = mresult->level1;
7772 mlevel2 = mresult->level23;
7773 mlevel3 = mresult->level23 + 16*count2;
7774 memcpy(mlevel1, level1, 32);
7775 memset(mlevel2, 0xFF, 16*count2);
7776 memset(mlevel3, 0, 128*count3);
7777 count3 = 0;
Antoine Pitrouaaefac72012-06-16 22:48:21 +02007778 for (i = 1; i < length; i++) {
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007779 int o1, o2, o3, i2, i3;
Antoine Pitrouaaefac72012-06-16 22:48:21 +02007780 Py_UCS4 ch = PyUnicode_READ(kind, data, i);
7781 if (ch == 0xFFFE)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007782 /* unmapped character */
7783 continue;
Antoine Pitrouaaefac72012-06-16 22:48:21 +02007784 o1 = ch>>11;
7785 o2 = (ch>>7) & 0xF;
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007786 i2 = 16*mlevel1[o1] + o2;
7787 if (mlevel2[i2] == 0xFF)
7788 mlevel2[i2] = count3++;
Antoine Pitrouaaefac72012-06-16 22:48:21 +02007789 o3 = ch & 0x7F;
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007790 i3 = 128*mlevel2[i2] + o3;
7791 mlevel3[i3] = i;
7792 }
7793 return result;
7794}
7795
7796static int
Victor Stinner22168992011-11-20 17:09:18 +01007797encoding_map_lookup(Py_UCS4 c, PyObject *mapping)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007798{
7799 struct encoding_map *map = (struct encoding_map*)mapping;
7800 int l1 = c>>11;
7801 int l2 = (c>>7) & 0xF;
7802 int l3 = c & 0x7F;
7803 int i;
7804
Victor Stinner22168992011-11-20 17:09:18 +01007805 if (c > 0xFFFF)
Benjamin Peterson29060642009-01-31 22:14:21 +00007806 return -1;
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007807 if (c == 0)
7808 return 0;
7809 /* level 1*/
7810 i = map->level1[l1];
7811 if (i == 0xFF) {
7812 return -1;
7813 }
7814 /* level 2*/
7815 i = map->level23[16*i+l2];
7816 if (i == 0xFF) {
7817 return -1;
7818 }
7819 /* level 3 */
7820 i = map->level23[16*map->count2 + 128*i + l3];
7821 if (i == 0) {
7822 return -1;
7823 }
7824 return i;
7825}
7826
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007827/* Lookup the character ch in the mapping. If the character
7828 can't be found, Py_None is returned (or NULL, if another
Fred Drakedb390c12005-10-28 14:39:47 +00007829 error occurred). */
Alexander Belopolsky40018472011-02-26 01:02:56 +00007830static PyObject *
Victor Stinner22168992011-11-20 17:09:18 +01007831charmapencode_lookup(Py_UCS4 c, PyObject *mapping)
Guido van Rossumd57fd912000-03-10 22:53:23 +00007832{
Christian Heimes217cfd12007-12-02 14:31:20 +00007833 PyObject *w = PyLong_FromLong((long)c);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007834 PyObject *x;
7835
7836 if (w == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00007837 return NULL;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007838 x = PyObject_GetItem(mapping, w);
7839 Py_DECREF(w);
7840 if (x == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +00007841 if (PyErr_ExceptionMatches(PyExc_LookupError)) {
7842 /* No mapping found means: mapping is undefined. */
7843 PyErr_Clear();
7844 x = Py_None;
7845 Py_INCREF(x);
7846 return x;
7847 } else
7848 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00007849 }
Walter Dörwaldadc72742003-01-08 22:01:33 +00007850 else if (x == Py_None)
Benjamin Peterson29060642009-01-31 22:14:21 +00007851 return x;
Christian Heimes217cfd12007-12-02 14:31:20 +00007852 else if (PyLong_Check(x)) {
Benjamin Peterson29060642009-01-31 22:14:21 +00007853 long value = PyLong_AS_LONG(x);
7854 if (value < 0 || value > 255) {
7855 PyErr_SetString(PyExc_TypeError,
7856 "character mapping must be in range(256)");
7857 Py_DECREF(x);
7858 return NULL;
7859 }
7860 return x;
Guido van Rossumd57fd912000-03-10 22:53:23 +00007861 }
Christian Heimes72b710a2008-05-26 13:28:38 +00007862 else if (PyBytes_Check(x))
Benjamin Peterson29060642009-01-31 22:14:21 +00007863 return x;
Guido van Rossumd57fd912000-03-10 22:53:23 +00007864 else {
Benjamin Peterson29060642009-01-31 22:14:21 +00007865 /* wrong return value */
7866 PyErr_Format(PyExc_TypeError,
7867 "character mapping must return integer, bytes or None, not %.400s",
7868 x->ob_type->tp_name);
7869 Py_DECREF(x);
7870 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00007871 }
7872}
7873
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007874static int
Guido van Rossum98297ee2007-11-06 21:34:58 +00007875charmapencode_resize(PyObject **outobj, Py_ssize_t *outpos, Py_ssize_t requiredsize)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007876{
Benjamin Peterson14339b62009-01-31 16:36:08 +00007877 Py_ssize_t outsize = PyBytes_GET_SIZE(*outobj);
7878 /* exponentially overallocate to minimize reallocations */
7879 if (requiredsize < 2*outsize)
7880 requiredsize = 2*outsize;
7881 if (_PyBytes_Resize(outobj, requiredsize))
7882 return -1;
7883 return 0;
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007884}
7885
Benjamin Peterson14339b62009-01-31 16:36:08 +00007886typedef enum charmapencode_result {
Benjamin Peterson29060642009-01-31 22:14:21 +00007887 enc_SUCCESS, enc_FAILED, enc_EXCEPTION
Alexander Belopolsky40018472011-02-26 01:02:56 +00007888} charmapencode_result;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007889/* lookup the character, put the result in the output string and adjust
Walter Dörwald827b0552007-05-12 13:23:53 +00007890 various state variables. Resize the output bytes object if not enough
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007891 space is available. Return a new reference to the object that
7892 was put in the output buffer, or Py_None, if the mapping was undefined
7893 (in which case no character was written) or NULL, if a
Andrew M. Kuchling8294de52005-11-02 16:36:12 +00007894 reallocation error occurred. The caller must decref the result */
Alexander Belopolsky40018472011-02-26 01:02:56 +00007895static charmapencode_result
Victor Stinner22168992011-11-20 17:09:18 +01007896charmapencode_output(Py_UCS4 c, PyObject *mapping,
Alexander Belopolsky40018472011-02-26 01:02:56 +00007897 PyObject **outobj, Py_ssize_t *outpos)
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007898{
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007899 PyObject *rep;
7900 char *outstart;
Christian Heimes72b710a2008-05-26 13:28:38 +00007901 Py_ssize_t outsize = PyBytes_GET_SIZE(*outobj);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007902
Christian Heimes90aa7642007-12-19 02:45:37 +00007903 if (Py_TYPE(mapping) == &EncodingMapType) {
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007904 int res = encoding_map_lookup(c, mapping);
Benjamin Peterson29060642009-01-31 22:14:21 +00007905 Py_ssize_t requiredsize = *outpos+1;
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007906 if (res == -1)
7907 return enc_FAILED;
Benjamin Peterson29060642009-01-31 22:14:21 +00007908 if (outsize<requiredsize)
7909 if (charmapencode_resize(outobj, outpos, requiredsize))
7910 return enc_EXCEPTION;
Christian Heimes72b710a2008-05-26 13:28:38 +00007911 outstart = PyBytes_AS_STRING(*outobj);
Benjamin Peterson29060642009-01-31 22:14:21 +00007912 outstart[(*outpos)++] = (char)res;
7913 return enc_SUCCESS;
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007914 }
7915
7916 rep = charmapencode_lookup(c, mapping);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007917 if (rep==NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00007918 return enc_EXCEPTION;
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007919 else if (rep==Py_None) {
Benjamin Peterson29060642009-01-31 22:14:21 +00007920 Py_DECREF(rep);
7921 return enc_FAILED;
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007922 } else {
Benjamin Peterson29060642009-01-31 22:14:21 +00007923 if (PyLong_Check(rep)) {
7924 Py_ssize_t requiredsize = *outpos+1;
7925 if (outsize<requiredsize)
7926 if (charmapencode_resize(outobj, outpos, requiredsize)) {
7927 Py_DECREF(rep);
7928 return enc_EXCEPTION;
7929 }
Christian Heimes72b710a2008-05-26 13:28:38 +00007930 outstart = PyBytes_AS_STRING(*outobj);
Benjamin Peterson29060642009-01-31 22:14:21 +00007931 outstart[(*outpos)++] = (char)PyLong_AS_LONG(rep);
Benjamin Peterson14339b62009-01-31 16:36:08 +00007932 }
Benjamin Peterson29060642009-01-31 22:14:21 +00007933 else {
7934 const char *repchars = PyBytes_AS_STRING(rep);
7935 Py_ssize_t repsize = PyBytes_GET_SIZE(rep);
7936 Py_ssize_t requiredsize = *outpos+repsize;
7937 if (outsize<requiredsize)
7938 if (charmapencode_resize(outobj, outpos, requiredsize)) {
7939 Py_DECREF(rep);
7940 return enc_EXCEPTION;
7941 }
Christian Heimes72b710a2008-05-26 13:28:38 +00007942 outstart = PyBytes_AS_STRING(*outobj);
Benjamin Peterson29060642009-01-31 22:14:21 +00007943 memcpy(outstart + *outpos, repchars, repsize);
7944 *outpos += repsize;
7945 }
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007946 }
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007947 Py_DECREF(rep);
7948 return enc_SUCCESS;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007949}
7950
7951/* handle an error in PyUnicode_EncodeCharmap
7952 Return 0 on success, -1 on error */
Alexander Belopolsky40018472011-02-26 01:02:56 +00007953static int
7954charmap_encoding_error(
Martin v. Löwis23e275b2011-11-02 18:02:51 +01007955 PyObject *unicode, Py_ssize_t *inpos, PyObject *mapping,
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007956 PyObject **exceptionObject,
Walter Dörwalde5402fb2003-08-14 20:25:29 +00007957 int *known_errorHandler, PyObject **errorHandler, const char *errors,
Guido van Rossum98297ee2007-11-06 21:34:58 +00007958 PyObject **res, Py_ssize_t *respos)
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007959{
7960 PyObject *repunicode = NULL; /* initialize to prevent gcc warning */
Martin v. Löwis23e275b2011-11-02 18:02:51 +01007961 Py_ssize_t size, repsize;
Martin v. Löwis18e16552006-02-15 17:27:45 +00007962 Py_ssize_t newpos;
Victor Stinnerae4f7c82011-11-20 18:28:55 +01007963 enum PyUnicode_Kind kind;
7964 void *data;
7965 Py_ssize_t index;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007966 /* startpos for collecting unencodable chars */
Martin v. Löwis18e16552006-02-15 17:27:45 +00007967 Py_ssize_t collstartpos = *inpos;
7968 Py_ssize_t collendpos = *inpos+1;
7969 Py_ssize_t collpos;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007970 char *encoding = "charmap";
7971 char *reason = "character maps to <undefined>";
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007972 charmapencode_result x;
Martin v. Löwis23e275b2011-11-02 18:02:51 +01007973 Py_UCS4 ch;
Brian Curtin2787ea42011-11-02 15:09:37 -05007974 int val;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007975
Benjamin Petersonbac79492012-01-14 13:34:47 -05007976 if (PyUnicode_READY(unicode) == -1)
Martin v. Löwis23e275b2011-11-02 18:02:51 +01007977 return -1;
7978 size = PyUnicode_GET_LENGTH(unicode);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007979 /* find all unencodable characters */
7980 while (collendpos < size) {
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00007981 PyObject *rep;
Christian Heimes90aa7642007-12-19 02:45:37 +00007982 if (Py_TYPE(mapping) == &EncodingMapType) {
Martin v. Löwis23e275b2011-11-02 18:02:51 +01007983 ch = PyUnicode_READ_CHAR(unicode, collendpos);
Brian Curtin2787ea42011-11-02 15:09:37 -05007984 val = encoding_map_lookup(ch, mapping);
7985 if (val != -1)
Benjamin Peterson29060642009-01-31 22:14:21 +00007986 break;
7987 ++collendpos;
7988 continue;
7989 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00007990
Martin v. Löwis23e275b2011-11-02 18:02:51 +01007991 ch = PyUnicode_READ_CHAR(unicode, collendpos);
7992 rep = charmapencode_lookup(ch, mapping);
Benjamin Peterson29060642009-01-31 22:14:21 +00007993 if (rep==NULL)
7994 return -1;
7995 else if (rep!=Py_None) {
7996 Py_DECREF(rep);
7997 break;
7998 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00007999 Py_DECREF(rep);
Benjamin Peterson29060642009-01-31 22:14:21 +00008000 ++collendpos;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008001 }
8002 /* cache callback name lookup
8003 * (if not done yet, i.e. it's the first error) */
8004 if (*known_errorHandler==-1) {
Benjamin Peterson29060642009-01-31 22:14:21 +00008005 if ((errors==NULL) || (!strcmp(errors, "strict")))
8006 *known_errorHandler = 1;
8007 else if (!strcmp(errors, "replace"))
8008 *known_errorHandler = 2;
8009 else if (!strcmp(errors, "ignore"))
8010 *known_errorHandler = 3;
8011 else if (!strcmp(errors, "xmlcharrefreplace"))
8012 *known_errorHandler = 4;
8013 else
8014 *known_errorHandler = 0;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008015 }
8016 switch (*known_errorHandler) {
Benjamin Peterson14339b62009-01-31 16:36:08 +00008017 case 1: /* strict */
Martin v. Löwis12be46c2011-11-04 19:04:15 +01008018 raise_encode_exception(exceptionObject, encoding, unicode, collstartpos, collendpos, reason);
Benjamin Peterson14339b62009-01-31 16:36:08 +00008019 return -1;
8020 case 2: /* replace */
8021 for (collpos = collstartpos; collpos<collendpos; ++collpos) {
Benjamin Peterson29060642009-01-31 22:14:21 +00008022 x = charmapencode_output('?', mapping, res, respos);
8023 if (x==enc_EXCEPTION) {
8024 return -1;
8025 }
8026 else if (x==enc_FAILED) {
Martin v. Löwis12be46c2011-11-04 19:04:15 +01008027 raise_encode_exception(exceptionObject, encoding, unicode, collstartpos, collendpos, reason);
Benjamin Peterson29060642009-01-31 22:14:21 +00008028 return -1;
8029 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00008030 }
8031 /* fall through */
8032 case 3: /* ignore */
8033 *inpos = collendpos;
8034 break;
8035 case 4: /* xmlcharrefreplace */
8036 /* generate replacement (temporarily (mis)uses p) */
8037 for (collpos = collstartpos; collpos < collendpos; ++collpos) {
Benjamin Peterson29060642009-01-31 22:14:21 +00008038 char buffer[2+29+1+1];
8039 char *cp;
Martin v. Löwis23e275b2011-11-02 18:02:51 +01008040 sprintf(buffer, "&#%d;", (int)PyUnicode_READ_CHAR(unicode, collpos));
Benjamin Peterson29060642009-01-31 22:14:21 +00008041 for (cp = buffer; *cp; ++cp) {
8042 x = charmapencode_output(*cp, mapping, res, respos);
8043 if (x==enc_EXCEPTION)
8044 return -1;
8045 else if (x==enc_FAILED) {
Martin v. Löwis12be46c2011-11-04 19:04:15 +01008046 raise_encode_exception(exceptionObject, encoding, unicode, collstartpos, collendpos, reason);
Benjamin Peterson29060642009-01-31 22:14:21 +00008047 return -1;
8048 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00008049 }
8050 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00008051 *inpos = collendpos;
8052 break;
8053 default:
8054 repunicode = unicode_encode_call_errorhandler(errors, errorHandler,
Martin v. Löwis23e275b2011-11-02 18:02:51 +01008055 encoding, reason, unicode, exceptionObject,
Benjamin Peterson29060642009-01-31 22:14:21 +00008056 collstartpos, collendpos, &newpos);
Benjamin Peterson14339b62009-01-31 16:36:08 +00008057 if (repunicode == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00008058 return -1;
Martin v. Löwis011e8422009-05-05 04:43:17 +00008059 if (PyBytes_Check(repunicode)) {
8060 /* Directly copy bytes result to output. */
8061 Py_ssize_t outsize = PyBytes_Size(*res);
8062 Py_ssize_t requiredsize;
8063 repsize = PyBytes_Size(repunicode);
8064 requiredsize = *respos + repsize;
8065 if (requiredsize > outsize)
8066 /* Make room for all additional bytes. */
8067 if (charmapencode_resize(res, respos, requiredsize)) {
8068 Py_DECREF(repunicode);
8069 return -1;
8070 }
8071 memcpy(PyBytes_AsString(*res) + *respos,
8072 PyBytes_AsString(repunicode), repsize);
8073 *respos += repsize;
8074 *inpos = newpos;
Martin v. Löwisdb12d452009-05-02 18:52:14 +00008075 Py_DECREF(repunicode);
Martin v. Löwis011e8422009-05-05 04:43:17 +00008076 break;
Martin v. Löwisdb12d452009-05-02 18:52:14 +00008077 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00008078 /* generate replacement */
Benjamin Petersonbac79492012-01-14 13:34:47 -05008079 if (PyUnicode_READY(repunicode) == -1) {
Victor Stinnerae4f7c82011-11-20 18:28:55 +01008080 Py_DECREF(repunicode);
8081 return -1;
8082 }
Victor Stinner9e30aa52011-11-21 02:49:52 +01008083 repsize = PyUnicode_GET_LENGTH(repunicode);
Victor Stinnerae4f7c82011-11-20 18:28:55 +01008084 data = PyUnicode_DATA(repunicode);
8085 kind = PyUnicode_KIND(repunicode);
8086 for (index = 0; index < repsize; index++) {
8087 Py_UCS4 repch = PyUnicode_READ(kind, data, index);
8088 x = charmapencode_output(repch, mapping, res, respos);
Benjamin Peterson29060642009-01-31 22:14:21 +00008089 if (x==enc_EXCEPTION) {
Victor Stinnerae4f7c82011-11-20 18:28:55 +01008090 Py_DECREF(repunicode);
Benjamin Peterson29060642009-01-31 22:14:21 +00008091 return -1;
8092 }
8093 else if (x==enc_FAILED) {
8094 Py_DECREF(repunicode);
Martin v. Löwis12be46c2011-11-04 19:04:15 +01008095 raise_encode_exception(exceptionObject, encoding, unicode, collstartpos, collendpos, reason);
Benjamin Peterson29060642009-01-31 22:14:21 +00008096 return -1;
8097 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00008098 }
8099 *inpos = newpos;
8100 Py_DECREF(repunicode);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008101 }
8102 return 0;
8103}
8104
Alexander Belopolsky40018472011-02-26 01:02:56 +00008105PyObject *
Martin v. Löwis23e275b2011-11-02 18:02:51 +01008106_PyUnicode_EncodeCharmap(PyObject *unicode,
8107 PyObject *mapping,
8108 const char *errors)
Guido van Rossumd57fd912000-03-10 22:53:23 +00008109{
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008110 /* output object */
8111 PyObject *res = NULL;
8112 /* current input position */
Martin v. Löwis18e16552006-02-15 17:27:45 +00008113 Py_ssize_t inpos = 0;
Martin v. Löwis23e275b2011-11-02 18:02:51 +01008114 Py_ssize_t size;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008115 /* current output position */
Martin v. Löwis18e16552006-02-15 17:27:45 +00008116 Py_ssize_t respos = 0;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008117 PyObject *errorHandler = NULL;
8118 PyObject *exc = NULL;
8119 /* the following variable is used for caching string comparisons
8120 * -1=not initialized, 0=unknown, 1=strict, 2=replace,
8121 * 3=ignore, 4=xmlcharrefreplace */
8122 int known_errorHandler = -1;
Guido van Rossumd57fd912000-03-10 22:53:23 +00008123
Benjamin Petersonbac79492012-01-14 13:34:47 -05008124 if (PyUnicode_READY(unicode) == -1)
Martin v. Löwis23e275b2011-11-02 18:02:51 +01008125 return NULL;
8126 size = PyUnicode_GET_LENGTH(unicode);
8127
Guido van Rossumd57fd912000-03-10 22:53:23 +00008128 /* Default to Latin-1 */
8129 if (mapping == NULL)
Martin v. Löwis23e275b2011-11-02 18:02:51 +01008130 return unicode_encode_ucs1(unicode, errors, 256);
Guido van Rossumd57fd912000-03-10 22:53:23 +00008131
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008132 /* allocate enough for a simple encoding without
8133 replacements, if we need more, we'll resize */
Christian Heimes72b710a2008-05-26 13:28:38 +00008134 res = PyBytes_FromStringAndSize(NULL, size);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008135 if (res == NULL)
8136 goto onError;
Marc-André Lemburgb7520772000-08-14 11:29:19 +00008137 if (size == 0)
Benjamin Peterson29060642009-01-31 22:14:21 +00008138 return res;
Guido van Rossumd57fd912000-03-10 22:53:23 +00008139
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008140 while (inpos<size) {
Martin v. Löwis23e275b2011-11-02 18:02:51 +01008141 Py_UCS4 ch = PyUnicode_READ_CHAR(unicode, inpos);
Benjamin Peterson29060642009-01-31 22:14:21 +00008142 /* try to encode it */
Martin v. Löwis23e275b2011-11-02 18:02:51 +01008143 charmapencode_result x = charmapencode_output(ch, mapping, &res, &respos);
Benjamin Peterson29060642009-01-31 22:14:21 +00008144 if (x==enc_EXCEPTION) /* error */
8145 goto onError;
8146 if (x==enc_FAILED) { /* unencodable character */
Martin v. Löwis23e275b2011-11-02 18:02:51 +01008147 if (charmap_encoding_error(unicode, &inpos, mapping,
Benjamin Peterson29060642009-01-31 22:14:21 +00008148 &exc,
8149 &known_errorHandler, &errorHandler, errors,
8150 &res, &respos)) {
8151 goto onError;
8152 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00008153 }
Benjamin Peterson29060642009-01-31 22:14:21 +00008154 else
8155 /* done with this character => adjust input position */
8156 ++inpos;
Guido van Rossumd57fd912000-03-10 22:53:23 +00008157 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00008158
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008159 /* Resize if we allocated to much */
Christian Heimes72b710a2008-05-26 13:28:38 +00008160 if (respos<PyBytes_GET_SIZE(res))
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00008161 if (_PyBytes_Resize(&res, respos) < 0)
8162 goto onError;
Guido van Rossum98297ee2007-11-06 21:34:58 +00008163
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008164 Py_XDECREF(exc);
8165 Py_XDECREF(errorHandler);
8166 return res;
8167
Benjamin Peterson29060642009-01-31 22:14:21 +00008168 onError:
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008169 Py_XDECREF(res);
8170 Py_XDECREF(exc);
8171 Py_XDECREF(errorHandler);
Guido van Rossumd57fd912000-03-10 22:53:23 +00008172 return NULL;
8173}
8174
Martin v. Löwis23e275b2011-11-02 18:02:51 +01008175/* Deprecated */
8176PyObject *
8177PyUnicode_EncodeCharmap(const Py_UNICODE *p,
8178 Py_ssize_t size,
8179 PyObject *mapping,
8180 const char *errors)
8181{
8182 PyObject *result;
8183 PyObject *unicode = PyUnicode_FromUnicode(p, size);
8184 if (unicode == NULL)
8185 return NULL;
8186 result = _PyUnicode_EncodeCharmap(unicode, mapping, errors);
8187 Py_DECREF(unicode);
Victor Stinnerfc026c92011-11-04 00:24:51 +01008188 return result;
Martin v. Löwis23e275b2011-11-02 18:02:51 +01008189}
8190
Alexander Belopolsky40018472011-02-26 01:02:56 +00008191PyObject *
8192PyUnicode_AsCharmapString(PyObject *unicode,
8193 PyObject *mapping)
Guido van Rossumd57fd912000-03-10 22:53:23 +00008194{
8195 if (!PyUnicode_Check(unicode) || mapping == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +00008196 PyErr_BadArgument();
8197 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00008198 }
Martin v. Löwis23e275b2011-11-02 18:02:51 +01008199 return _PyUnicode_EncodeCharmap(unicode, mapping, NULL);
Guido van Rossumd57fd912000-03-10 22:53:23 +00008200}
8201
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008202/* create or adjust a UnicodeTranslateError */
Alexander Belopolsky40018472011-02-26 01:02:56 +00008203static void
8204make_translate_exception(PyObject **exceptionObject,
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008205 PyObject *unicode,
Alexander Belopolsky40018472011-02-26 01:02:56 +00008206 Py_ssize_t startpos, Py_ssize_t endpos,
8207 const char *reason)
Guido van Rossumd57fd912000-03-10 22:53:23 +00008208{
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008209 if (*exceptionObject == NULL) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008210 *exceptionObject = _PyUnicodeTranslateError_Create(
8211 unicode, startpos, endpos, reason);
Guido van Rossumd57fd912000-03-10 22:53:23 +00008212 }
8213 else {
Benjamin Peterson29060642009-01-31 22:14:21 +00008214 if (PyUnicodeTranslateError_SetStart(*exceptionObject, startpos))
8215 goto onError;
8216 if (PyUnicodeTranslateError_SetEnd(*exceptionObject, endpos))
8217 goto onError;
8218 if (PyUnicodeTranslateError_SetReason(*exceptionObject, reason))
8219 goto onError;
8220 return;
8221 onError:
8222 Py_DECREF(*exceptionObject);
8223 *exceptionObject = NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00008224 }
8225}
8226
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008227/* raises a UnicodeTranslateError */
Alexander Belopolsky40018472011-02-26 01:02:56 +00008228static void
8229raise_translate_exception(PyObject **exceptionObject,
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008230 PyObject *unicode,
Alexander Belopolsky40018472011-02-26 01:02:56 +00008231 Py_ssize_t startpos, Py_ssize_t endpos,
8232 const char *reason)
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008233{
8234 make_translate_exception(exceptionObject,
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008235 unicode, startpos, endpos, reason);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008236 if (*exceptionObject != NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00008237 PyCodec_StrictErrors(*exceptionObject);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008238}
8239
8240/* error handling callback helper:
8241 build arguments, call the callback and check the arguments,
8242 put the result into newpos and return the replacement string, which
8243 has to be freed by the caller */
Alexander Belopolsky40018472011-02-26 01:02:56 +00008244static PyObject *
8245unicode_translate_call_errorhandler(const char *errors,
8246 PyObject **errorHandler,
8247 const char *reason,
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008248 PyObject *unicode, PyObject **exceptionObject,
Alexander Belopolsky40018472011-02-26 01:02:56 +00008249 Py_ssize_t startpos, Py_ssize_t endpos,
8250 Py_ssize_t *newpos)
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008251{
Benjamin Peterson142957c2008-07-04 19:55:29 +00008252 static char *argparse = "O!n;translating error handler must return (str, int) tuple";
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008253
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00008254 Py_ssize_t i_newpos;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008255 PyObject *restuple;
8256 PyObject *resunicode;
8257
8258 if (*errorHandler == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +00008259 *errorHandler = PyCodec_LookupError(errors);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008260 if (*errorHandler == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00008261 return NULL;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008262 }
8263
8264 make_translate_exception(exceptionObject,
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008265 unicode, startpos, endpos, reason);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008266 if (*exceptionObject == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00008267 return NULL;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008268
8269 restuple = PyObject_CallFunctionObjArgs(
Benjamin Peterson29060642009-01-31 22:14:21 +00008270 *errorHandler, *exceptionObject, NULL);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008271 if (restuple == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00008272 return NULL;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008273 if (!PyTuple_Check(restuple)) {
Benjamin Petersond75fcb42009-02-19 04:22:03 +00008274 PyErr_SetString(PyExc_TypeError, &argparse[4]);
Benjamin Peterson29060642009-01-31 22:14:21 +00008275 Py_DECREF(restuple);
8276 return NULL;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008277 }
8278 if (!PyArg_ParseTuple(restuple, argparse, &PyUnicode_Type,
Benjamin Peterson29060642009-01-31 22:14:21 +00008279 &resunicode, &i_newpos)) {
8280 Py_DECREF(restuple);
8281 return NULL;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008282 }
Martin v. Löwis18e16552006-02-15 17:27:45 +00008283 if (i_newpos<0)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008284 *newpos = PyUnicode_GET_LENGTH(unicode)+i_newpos;
Martin v. Löwis18e16552006-02-15 17:27:45 +00008285 else
8286 *newpos = i_newpos;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008287 if (*newpos<0 || *newpos>PyUnicode_GET_LENGTH(unicode)) {
Benjamin Peterson29060642009-01-31 22:14:21 +00008288 PyErr_Format(PyExc_IndexError, "position %zd from error handler out of bounds", *newpos);
8289 Py_DECREF(restuple);
8290 return NULL;
Walter Dörwald2e0b18a2003-01-31 17:19:08 +00008291 }
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008292 Py_INCREF(resunicode);
8293 Py_DECREF(restuple);
8294 return resunicode;
8295}
8296
8297/* Lookup the character ch in the mapping and put the result in result,
8298 which must be decrefed by the caller.
8299 Return 0 on success, -1 on error */
Alexander Belopolsky40018472011-02-26 01:02:56 +00008300static int
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008301charmaptranslate_lookup(Py_UCS4 c, PyObject *mapping, PyObject **result)
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008302{
Christian Heimes217cfd12007-12-02 14:31:20 +00008303 PyObject *w = PyLong_FromLong((long)c);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008304 PyObject *x;
8305
8306 if (w == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00008307 return -1;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008308 x = PyObject_GetItem(mapping, w);
8309 Py_DECREF(w);
8310 if (x == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +00008311 if (PyErr_ExceptionMatches(PyExc_LookupError)) {
8312 /* No mapping found means: use 1:1 mapping. */
8313 PyErr_Clear();
8314 *result = NULL;
8315 return 0;
8316 } else
8317 return -1;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008318 }
8319 else if (x == Py_None) {
Benjamin Peterson29060642009-01-31 22:14:21 +00008320 *result = x;
8321 return 0;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008322 }
Christian Heimes217cfd12007-12-02 14:31:20 +00008323 else if (PyLong_Check(x)) {
Benjamin Peterson29060642009-01-31 22:14:21 +00008324 long value = PyLong_AS_LONG(x);
8325 long max = PyUnicode_GetMax();
8326 if (value < 0 || value > max) {
8327 PyErr_Format(PyExc_TypeError,
Guido van Rossum5a2f7e602007-10-24 21:13:09 +00008328 "character mapping must be in range(0x%x)", max+1);
Benjamin Peterson29060642009-01-31 22:14:21 +00008329 Py_DECREF(x);
8330 return -1;
8331 }
8332 *result = x;
8333 return 0;
8334 }
8335 else if (PyUnicode_Check(x)) {
8336 *result = x;
8337 return 0;
8338 }
8339 else {
8340 /* wrong return value */
8341 PyErr_SetString(PyExc_TypeError,
8342 "character mapping must return integer, None or str");
Benjamin Peterson14339b62009-01-31 16:36:08 +00008343 Py_DECREF(x);
8344 return -1;
8345 }
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008346}
8347/* ensure that *outobj is at least requiredsize characters long,
Benjamin Peterson29060642009-01-31 22:14:21 +00008348 if not reallocate and adjust various state variables.
8349 Return 0 on success, -1 on error */
Alexander Belopolsky40018472011-02-26 01:02:56 +00008350static int
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008351charmaptranslate_makespace(Py_UCS4 **outobj, Py_ssize_t *psize,
Benjamin Peterson29060642009-01-31 22:14:21 +00008352 Py_ssize_t requiredsize)
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008353{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008354 Py_ssize_t oldsize = *psize;
Kristjan Valur Jonsson85634d72012-05-31 09:37:31 +00008355 Py_UCS4 *new_outobj;
Walter Dörwald4894c302003-10-24 14:25:28 +00008356 if (requiredsize > oldsize) {
Benjamin Peterson29060642009-01-31 22:14:21 +00008357 /* exponentially overallocate to minimize reallocations */
8358 if (requiredsize < 2 * oldsize)
8359 requiredsize = 2 * oldsize;
Kristjan Valur Jonsson85634d72012-05-31 09:37:31 +00008360 new_outobj = PyMem_Realloc(*outobj, requiredsize * sizeof(Py_UCS4));
8361 if (new_outobj == 0)
Benjamin Peterson29060642009-01-31 22:14:21 +00008362 return -1;
Kristjan Valur Jonsson85634d72012-05-31 09:37:31 +00008363 *outobj = new_outobj;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008364 *psize = requiredsize;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008365 }
8366 return 0;
8367}
8368/* lookup the character, put the result in the output string and adjust
8369 various state variables. Return a new reference to the object that
8370 was put in the output buffer in *result, or Py_None, if the mapping was
8371 undefined (in which case no character was written).
8372 The called must decref result.
8373 Return 0 on success, -1 on error. */
Alexander Belopolsky40018472011-02-26 01:02:56 +00008374static int
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008375charmaptranslate_output(PyObject *input, Py_ssize_t ipos,
8376 PyObject *mapping, Py_UCS4 **output,
8377 Py_ssize_t *osize, Py_ssize_t *opos,
Alexander Belopolsky40018472011-02-26 01:02:56 +00008378 PyObject **res)
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008379{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008380 Py_UCS4 curinp = PyUnicode_READ_CHAR(input, ipos);
8381 if (charmaptranslate_lookup(curinp, mapping, res))
Benjamin Peterson29060642009-01-31 22:14:21 +00008382 return -1;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008383 if (*res==NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +00008384 /* not found => default to 1:1 mapping */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008385 (*output)[(*opos)++] = curinp;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008386 }
8387 else if (*res==Py_None)
Benjamin Peterson29060642009-01-31 22:14:21 +00008388 ;
Christian Heimes217cfd12007-12-02 14:31:20 +00008389 else if (PyLong_Check(*res)) {
Benjamin Peterson29060642009-01-31 22:14:21 +00008390 /* no overflow check, because we know that the space is enough */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008391 (*output)[(*opos)++] = (Py_UCS4)PyLong_AS_LONG(*res);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008392 }
8393 else if (PyUnicode_Check(*res)) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008394 Py_ssize_t repsize;
8395 if (PyUnicode_READY(*res) == -1)
8396 return -1;
8397 repsize = PyUnicode_GET_LENGTH(*res);
Benjamin Peterson29060642009-01-31 22:14:21 +00008398 if (repsize==1) {
8399 /* no overflow check, because we know that the space is enough */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008400 (*output)[(*opos)++] = PyUnicode_READ_CHAR(*res, 0);
Benjamin Peterson29060642009-01-31 22:14:21 +00008401 }
8402 else if (repsize!=0) {
8403 /* more than one character */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008404 Py_ssize_t requiredsize = *opos +
8405 (PyUnicode_GET_LENGTH(input) - ipos) +
Benjamin Peterson29060642009-01-31 22:14:21 +00008406 repsize - 1;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008407 Py_ssize_t i;
8408 if (charmaptranslate_makespace(output, osize, requiredsize))
Benjamin Peterson29060642009-01-31 22:14:21 +00008409 return -1;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008410 for(i = 0; i < repsize; i++)
8411 (*output)[(*opos)++] = PyUnicode_READ_CHAR(*res, i);
Benjamin Peterson29060642009-01-31 22:14:21 +00008412 }
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008413 }
8414 else
Benjamin Peterson29060642009-01-31 22:14:21 +00008415 return -1;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008416 return 0;
8417}
8418
Alexander Belopolsky40018472011-02-26 01:02:56 +00008419PyObject *
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008420_PyUnicode_TranslateCharmap(PyObject *input,
8421 PyObject *mapping,
8422 const char *errors)
Guido van Rossumd57fd912000-03-10 22:53:23 +00008423{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008424 /* input object */
8425 char *idata;
8426 Py_ssize_t size, i;
8427 int kind;
8428 /* output buffer */
8429 Py_UCS4 *output = NULL;
8430 Py_ssize_t osize;
8431 PyObject *res;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008432 /* current output position */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008433 Py_ssize_t opos;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008434 char *reason = "character maps to <undefined>";
8435 PyObject *errorHandler = NULL;
8436 PyObject *exc = NULL;
8437 /* the following variable is used for caching string comparisons
8438 * -1=not initialized, 0=unknown, 1=strict, 2=replace,
8439 * 3=ignore, 4=xmlcharrefreplace */
8440 int known_errorHandler = -1;
8441
Guido van Rossumd57fd912000-03-10 22:53:23 +00008442 if (mapping == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +00008443 PyErr_BadArgument();
8444 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00008445 }
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008446
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008447 if (PyUnicode_READY(input) == -1)
8448 return NULL;
8449 idata = (char*)PyUnicode_DATA(input);
8450 kind = PyUnicode_KIND(input);
8451 size = PyUnicode_GET_LENGTH(input);
8452 i = 0;
8453
8454 if (size == 0) {
8455 Py_INCREF(input);
8456 return input;
8457 }
8458
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008459 /* allocate enough for a simple 1:1 translation without
8460 replacements, if we need more, we'll resize */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008461 osize = size;
8462 output = PyMem_Malloc(osize * sizeof(Py_UCS4));
8463 opos = 0;
8464 if (output == NULL) {
8465 PyErr_NoMemory();
Benjamin Peterson29060642009-01-31 22:14:21 +00008466 goto onError;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008467 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00008468
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008469 while (i<size) {
Benjamin Peterson29060642009-01-31 22:14:21 +00008470 /* try to encode it */
8471 PyObject *x = NULL;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008472 if (charmaptranslate_output(input, i, mapping,
8473 &output, &osize, &opos, &x)) {
Benjamin Peterson29060642009-01-31 22:14:21 +00008474 Py_XDECREF(x);
8475 goto onError;
8476 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00008477 Py_XDECREF(x);
Benjamin Peterson29060642009-01-31 22:14:21 +00008478 if (x!=Py_None) /* it worked => adjust input pointer */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008479 ++i;
Benjamin Peterson29060642009-01-31 22:14:21 +00008480 else { /* untranslatable character */
8481 PyObject *repunicode = NULL; /* initialize to prevent gcc warning */
8482 Py_ssize_t repsize;
8483 Py_ssize_t newpos;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008484 Py_ssize_t uni2;
Benjamin Peterson29060642009-01-31 22:14:21 +00008485 /* startpos for collecting untranslatable chars */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008486 Py_ssize_t collstart = i;
8487 Py_ssize_t collend = i+1;
8488 Py_ssize_t coll;
Guido van Rossumd57fd912000-03-10 22:53:23 +00008489
Benjamin Peterson29060642009-01-31 22:14:21 +00008490 /* find all untranslatable characters */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008491 while (collend < size) {
8492 if (charmaptranslate_lookup(PyUnicode_READ(kind,idata, collend), mapping, &x))
Benjamin Peterson29060642009-01-31 22:14:21 +00008493 goto onError;
8494 Py_XDECREF(x);
8495 if (x!=Py_None)
8496 break;
8497 ++collend;
8498 }
8499 /* cache callback name lookup
8500 * (if not done yet, i.e. it's the first error) */
8501 if (known_errorHandler==-1) {
8502 if ((errors==NULL) || (!strcmp(errors, "strict")))
8503 known_errorHandler = 1;
8504 else if (!strcmp(errors, "replace"))
8505 known_errorHandler = 2;
8506 else if (!strcmp(errors, "ignore"))
8507 known_errorHandler = 3;
8508 else if (!strcmp(errors, "xmlcharrefreplace"))
8509 known_errorHandler = 4;
8510 else
8511 known_errorHandler = 0;
8512 }
8513 switch (known_errorHandler) {
8514 case 1: /* strict */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008515 raise_translate_exception(&exc, input, collstart,
8516 collend, reason);
Benjamin Peterson14339b62009-01-31 16:36:08 +00008517 goto onError;
Benjamin Peterson29060642009-01-31 22:14:21 +00008518 case 2: /* replace */
8519 /* No need to check for space, this is a 1:1 replacement */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008520 for (coll = collstart; coll<collend; coll++)
8521 output[opos++] = '?';
Benjamin Peterson29060642009-01-31 22:14:21 +00008522 /* fall through */
8523 case 3: /* ignore */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008524 i = collend;
Benjamin Peterson29060642009-01-31 22:14:21 +00008525 break;
8526 case 4: /* xmlcharrefreplace */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008527 /* generate replacement (temporarily (mis)uses i) */
8528 for (i = collstart; i < collend; ++i) {
Benjamin Peterson29060642009-01-31 22:14:21 +00008529 char buffer[2+29+1+1];
8530 char *cp;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008531 sprintf(buffer, "&#%d;", PyUnicode_READ(kind, idata, i));
8532 if (charmaptranslate_makespace(&output, &osize,
8533 opos+strlen(buffer)+(size-collend)))
Benjamin Peterson29060642009-01-31 22:14:21 +00008534 goto onError;
8535 for (cp = buffer; *cp; ++cp)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008536 output[opos++] = *cp;
Benjamin Peterson29060642009-01-31 22:14:21 +00008537 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008538 i = collend;
Benjamin Peterson29060642009-01-31 22:14:21 +00008539 break;
8540 default:
8541 repunicode = unicode_translate_call_errorhandler(errors, &errorHandler,
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008542 reason, input, &exc,
8543 collstart, collend, &newpos);
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01008544 if (repunicode == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00008545 goto onError;
Benjamin Peterson9ca3ffa2012-01-01 16:04:29 -06008546 if (PyUnicode_READY(repunicode) == -1) {
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01008547 Py_DECREF(repunicode);
8548 goto onError;
8549 }
Benjamin Peterson29060642009-01-31 22:14:21 +00008550 /* generate replacement */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008551 repsize = PyUnicode_GET_LENGTH(repunicode);
8552 if (charmaptranslate_makespace(&output, &osize,
8553 opos+repsize+(size-collend))) {
Benjamin Peterson29060642009-01-31 22:14:21 +00008554 Py_DECREF(repunicode);
8555 goto onError;
8556 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008557 for (uni2 = 0; repsize-->0; ++uni2)
8558 output[opos++] = PyUnicode_READ_CHAR(repunicode, uni2);
8559 i = newpos;
Benjamin Peterson29060642009-01-31 22:14:21 +00008560 Py_DECREF(repunicode);
Benjamin Peterson14339b62009-01-31 16:36:08 +00008561 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00008562 }
8563 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008564 res = PyUnicode_FromKindAndData(PyUnicode_4BYTE_KIND, output, opos);
8565 if (!res)
8566 goto onError;
8567 PyMem_Free(output);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008568 Py_XDECREF(exc);
8569 Py_XDECREF(errorHandler);
8570 return res;
Guido van Rossumd57fd912000-03-10 22:53:23 +00008571
Benjamin Peterson29060642009-01-31 22:14:21 +00008572 onError:
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008573 PyMem_Free(output);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00008574 Py_XDECREF(exc);
8575 Py_XDECREF(errorHandler);
Guido van Rossumd57fd912000-03-10 22:53:23 +00008576 return NULL;
8577}
8578
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008579/* Deprecated. Use PyUnicode_Translate instead. */
8580PyObject *
8581PyUnicode_TranslateCharmap(const Py_UNICODE *p,
8582 Py_ssize_t size,
8583 PyObject *mapping,
8584 const char *errors)
8585{
Christian Heimes5f520f42012-09-11 14:03:25 +02008586 PyObject *result;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008587 PyObject *unicode = PyUnicode_FromUnicode(p, size);
8588 if (!unicode)
8589 return NULL;
Christian Heimes5f520f42012-09-11 14:03:25 +02008590 result = _PyUnicode_TranslateCharmap(unicode, mapping, errors);
8591 Py_DECREF(unicode);
8592 return result;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008593}
8594
Alexander Belopolsky40018472011-02-26 01:02:56 +00008595PyObject *
8596PyUnicode_Translate(PyObject *str,
8597 PyObject *mapping,
8598 const char *errors)
Guido van Rossumd57fd912000-03-10 22:53:23 +00008599{
8600 PyObject *result;
Tim Petersced69f82003-09-16 20:30:58 +00008601
Guido van Rossumd57fd912000-03-10 22:53:23 +00008602 str = PyUnicode_FromObject(str);
8603 if (str == NULL)
Christian Heimes5f520f42012-09-11 14:03:25 +02008604 return NULL;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008605 result = _PyUnicode_TranslateCharmap(str, mapping, errors);
Guido van Rossumd57fd912000-03-10 22:53:23 +00008606 Py_DECREF(str);
8607 return result;
Guido van Rossumd57fd912000-03-10 22:53:23 +00008608}
Tim Petersced69f82003-09-16 20:30:58 +00008609
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008610static Py_UCS4
Victor Stinner9310abb2011-10-05 00:59:23 +02008611fix_decimal_and_space_to_ascii(PyObject *self)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008612{
8613 /* No need to call PyUnicode_READY(self) because this function is only
8614 called as a callback from fixup() which does it already. */
8615 const Py_ssize_t len = PyUnicode_GET_LENGTH(self);
8616 const int kind = PyUnicode_KIND(self);
8617 void *data = PyUnicode_DATA(self);
Victor Stinnere6abb482012-05-02 01:15:40 +02008618 Py_UCS4 maxchar = 127, ch, fixed;
Benjamin Peterson821e4cf2012-01-12 15:40:18 -05008619 int modified = 0;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008620 Py_ssize_t i;
8621
8622 for (i = 0; i < len; ++i) {
8623 ch = PyUnicode_READ(kind, data, i);
8624 fixed = 0;
8625 if (ch > 127) {
8626 if (Py_UNICODE_ISSPACE(ch))
8627 fixed = ' ';
8628 else {
8629 const int decimal = Py_UNICODE_TODECIMAL(ch);
8630 if (decimal >= 0)
8631 fixed = '0' + decimal;
8632 }
8633 if (fixed != 0) {
Benjamin Peterson821e4cf2012-01-12 15:40:18 -05008634 modified = 1;
Victor Stinnere6abb482012-05-02 01:15:40 +02008635 maxchar = MAX_MAXCHAR(maxchar, fixed);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008636 PyUnicode_WRITE(kind, data, i, fixed);
8637 }
Victor Stinnere6abb482012-05-02 01:15:40 +02008638 else
8639 maxchar = MAX_MAXCHAR(maxchar, ch);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008640 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008641 }
8642
Benjamin Peterson821e4cf2012-01-12 15:40:18 -05008643 return (modified) ? maxchar : 0;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008644}
8645
8646PyObject *
8647_PyUnicode_TransformDecimalAndSpaceToASCII(PyObject *unicode)
8648{
8649 if (!PyUnicode_Check(unicode)) {
8650 PyErr_BadInternalCall();
8651 return NULL;
8652 }
8653 if (PyUnicode_READY(unicode) == -1)
8654 return NULL;
8655 if (PyUnicode_MAX_CHAR_VALUE(unicode) <= 127) {
8656 /* If the string is already ASCII, just return the same string */
8657 Py_INCREF(unicode);
8658 return unicode;
8659 }
Victor Stinner9310abb2011-10-05 00:59:23 +02008660 return fixup(unicode, fix_decimal_and_space_to_ascii);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008661}
8662
Alexander Belopolsky942af5a2010-12-04 03:38:46 +00008663PyObject *
8664PyUnicode_TransformDecimalToASCII(Py_UNICODE *s,
8665 Py_ssize_t length)
8666{
Victor Stinnerf0124502011-11-21 23:12:56 +01008667 PyObject *decimal;
Alexander Belopolsky942af5a2010-12-04 03:38:46 +00008668 Py_ssize_t i;
Victor Stinnerf0124502011-11-21 23:12:56 +01008669 Py_UCS4 maxchar;
8670 enum PyUnicode_Kind kind;
8671 void *data;
8672
Victor Stinner99d7ad02012-02-22 13:37:39 +01008673 maxchar = 127;
Alexander Belopolsky942af5a2010-12-04 03:38:46 +00008674 for (i = 0; i < length; i++) {
Victor Stinnerf0124502011-11-21 23:12:56 +01008675 Py_UNICODE ch = s[i];
Alexander Belopolsky942af5a2010-12-04 03:38:46 +00008676 if (ch > 127) {
8677 int decimal = Py_UNICODE_TODECIMAL(ch);
8678 if (decimal >= 0)
Victor Stinnerf0124502011-11-21 23:12:56 +01008679 ch = '0' + decimal;
Victor Stinnere6abb482012-05-02 01:15:40 +02008680 maxchar = MAX_MAXCHAR(maxchar, ch);
Alexander Belopolsky942af5a2010-12-04 03:38:46 +00008681 }
8682 }
Victor Stinnerf0124502011-11-21 23:12:56 +01008683
8684 /* Copy to a new string */
8685 decimal = PyUnicode_New(length, maxchar);
8686 if (decimal == NULL)
8687 return decimal;
8688 kind = PyUnicode_KIND(decimal);
8689 data = PyUnicode_DATA(decimal);
8690 /* Iterate over code points */
8691 for (i = 0; i < length; i++) {
8692 Py_UNICODE ch = s[i];
8693 if (ch > 127) {
8694 int decimal = Py_UNICODE_TODECIMAL(ch);
8695 if (decimal >= 0)
8696 ch = '0' + decimal;
8697 }
8698 PyUnicode_WRITE(kind, data, i, ch);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008699 }
Victor Stinnerd3df8ab2011-11-22 01:22:34 +01008700 return unicode_result(decimal);
Alexander Belopolsky942af5a2010-12-04 03:38:46 +00008701}
Guido van Rossum9e896b32000-04-05 20:11:21 +00008702/* --- Decimal Encoder ---------------------------------------------------- */
8703
Alexander Belopolsky40018472011-02-26 01:02:56 +00008704int
8705PyUnicode_EncodeDecimal(Py_UNICODE *s,
8706 Py_ssize_t length,
8707 char *output,
8708 const char *errors)
Guido van Rossum9e896b32000-04-05 20:11:21 +00008709{
Martin v. Löwis23e275b2011-11-02 18:02:51 +01008710 PyObject *unicode;
Victor Stinner6345be92011-11-25 20:09:01 +01008711 Py_ssize_t i;
Victor Stinner42bf7752011-11-21 22:52:58 +01008712 enum PyUnicode_Kind kind;
8713 void *data;
Guido van Rossum9e896b32000-04-05 20:11:21 +00008714
8715 if (output == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +00008716 PyErr_BadArgument();
8717 return -1;
Guido van Rossum9e896b32000-04-05 20:11:21 +00008718 }
8719
Victor Stinner42bf7752011-11-21 22:52:58 +01008720 unicode = PyUnicode_FromUnicode(s, length);
8721 if (unicode == NULL)
8722 return -1;
8723
Benjamin Petersonbac79492012-01-14 13:34:47 -05008724 if (PyUnicode_READY(unicode) == -1) {
Victor Stinner6345be92011-11-25 20:09:01 +01008725 Py_DECREF(unicode);
8726 return -1;
8727 }
Victor Stinner42bf7752011-11-21 22:52:58 +01008728 kind = PyUnicode_KIND(unicode);
8729 data = PyUnicode_DATA(unicode);
8730
Victor Stinnerb84d7232011-11-22 01:50:07 +01008731 for (i=0; i < length; ) {
Victor Stinner6345be92011-11-25 20:09:01 +01008732 PyObject *exc;
8733 Py_UCS4 ch;
Benjamin Peterson29060642009-01-31 22:14:21 +00008734 int decimal;
Victor Stinner6345be92011-11-25 20:09:01 +01008735 Py_ssize_t startpos;
8736
8737 ch = PyUnicode_READ(kind, data, i);
Tim Petersced69f82003-09-16 20:30:58 +00008738
Benjamin Peterson29060642009-01-31 22:14:21 +00008739 if (Py_UNICODE_ISSPACE(ch)) {
Benjamin Peterson14339b62009-01-31 16:36:08 +00008740 *output++ = ' ';
Victor Stinnerb84d7232011-11-22 01:50:07 +01008741 i++;
Benjamin Peterson29060642009-01-31 22:14:21 +00008742 continue;
Benjamin Peterson14339b62009-01-31 16:36:08 +00008743 }
Benjamin Peterson29060642009-01-31 22:14:21 +00008744 decimal = Py_UNICODE_TODECIMAL(ch);
8745 if (decimal >= 0) {
8746 *output++ = '0' + decimal;
Victor Stinnerb84d7232011-11-22 01:50:07 +01008747 i++;
Benjamin Peterson29060642009-01-31 22:14:21 +00008748 continue;
8749 }
8750 if (0 < ch && ch < 256) {
8751 *output++ = (char)ch;
Victor Stinnerb84d7232011-11-22 01:50:07 +01008752 i++;
Benjamin Peterson29060642009-01-31 22:14:21 +00008753 continue;
8754 }
Victor Stinner6345be92011-11-25 20:09:01 +01008755
Victor Stinner42bf7752011-11-21 22:52:58 +01008756 startpos = i;
Victor Stinner6345be92011-11-25 20:09:01 +01008757 exc = NULL;
8758 raise_encode_exception(&exc, "decimal", unicode,
8759 startpos, startpos+1,
8760 "invalid decimal Unicode string");
8761 Py_XDECREF(exc);
8762 Py_DECREF(unicode);
8763 return -1;
Guido van Rossum9e896b32000-04-05 20:11:21 +00008764 }
8765 /* 0-terminate the output string */
8766 *output++ = '\0';
Victor Stinner42bf7752011-11-21 22:52:58 +01008767 Py_DECREF(unicode);
Guido van Rossum9e896b32000-04-05 20:11:21 +00008768 return 0;
Guido van Rossum9e896b32000-04-05 20:11:21 +00008769}
8770
Guido van Rossumd57fd912000-03-10 22:53:23 +00008771/* --- Helpers ------------------------------------------------------------ */
8772
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008773static Py_ssize_t
Victor Stinner794d5672011-10-10 03:21:36 +02008774any_find_slice(int direction, PyObject* s1, PyObject* s2,
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008775 Py_ssize_t start,
8776 Py_ssize_t end)
8777{
8778 int kind1, kind2, kind;
8779 void *buf1, *buf2;
8780 Py_ssize_t len1, len2, result;
8781
8782 kind1 = PyUnicode_KIND(s1);
8783 kind2 = PyUnicode_KIND(s2);
8784 kind = kind1 > kind2 ? kind1 : kind2;
8785 buf1 = PyUnicode_DATA(s1);
8786 buf2 = PyUnicode_DATA(s2);
8787 if (kind1 != kind)
8788 buf1 = _PyUnicode_AsKind(s1, kind);
8789 if (!buf1)
8790 return -2;
8791 if (kind2 != kind)
8792 buf2 = _PyUnicode_AsKind(s2, kind);
8793 if (!buf2) {
8794 if (kind1 != kind) PyMem_Free(buf1);
8795 return -2;
8796 }
8797 len1 = PyUnicode_GET_LENGTH(s1);
8798 len2 = PyUnicode_GET_LENGTH(s2);
8799
Victor Stinner794d5672011-10-10 03:21:36 +02008800 if (direction > 0) {
Benjamin Petersonead6b532011-12-20 17:23:42 -06008801 switch (kind) {
Victor Stinner794d5672011-10-10 03:21:36 +02008802 case PyUnicode_1BYTE_KIND:
8803 if (PyUnicode_IS_ASCII(s1) && PyUnicode_IS_ASCII(s2))
8804 result = asciilib_find_slice(buf1, len1, buf2, len2, start, end);
8805 else
8806 result = ucs1lib_find_slice(buf1, len1, buf2, len2, start, end);
8807 break;
8808 case PyUnicode_2BYTE_KIND:
8809 result = ucs2lib_find_slice(buf1, len1, buf2, len2, start, end);
8810 break;
8811 case PyUnicode_4BYTE_KIND:
8812 result = ucs4lib_find_slice(buf1, len1, buf2, len2, start, end);
8813 break;
8814 default:
8815 assert(0); result = -2;
8816 }
8817 }
8818 else {
Benjamin Petersonead6b532011-12-20 17:23:42 -06008819 switch (kind) {
Victor Stinner794d5672011-10-10 03:21:36 +02008820 case PyUnicode_1BYTE_KIND:
8821 if (PyUnicode_IS_ASCII(s1) && PyUnicode_IS_ASCII(s2))
8822 result = asciilib_rfind_slice(buf1, len1, buf2, len2, start, end);
8823 else
8824 result = ucs1lib_rfind_slice(buf1, len1, buf2, len2, start, end);
8825 break;
8826 case PyUnicode_2BYTE_KIND:
8827 result = ucs2lib_rfind_slice(buf1, len1, buf2, len2, start, end);
8828 break;
8829 case PyUnicode_4BYTE_KIND:
8830 result = ucs4lib_rfind_slice(buf1, len1, buf2, len2, start, end);
8831 break;
8832 default:
8833 assert(0); result = -2;
8834 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008835 }
8836
8837 if (kind1 != kind)
8838 PyMem_Free(buf1);
8839 if (kind2 != kind)
8840 PyMem_Free(buf2);
8841
8842 return result;
8843}
8844
8845Py_ssize_t
Victor Stinner41a863c2012-02-24 00:37:51 +01008846_PyUnicode_InsertThousandsGrouping(
8847 PyObject *unicode, Py_ssize_t index,
8848 Py_ssize_t n_buffer,
8849 void *digits, Py_ssize_t n_digits,
8850 Py_ssize_t min_width,
8851 const char *grouping, PyObject *thousands_sep,
8852 Py_UCS4 *maxchar)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008853{
Victor Stinner41a863c2012-02-24 00:37:51 +01008854 unsigned int kind, thousands_sep_kind;
Antoine Pitrou842c0f12012-02-24 13:30:46 +01008855 char *data, *thousands_sep_data;
Victor Stinner41a863c2012-02-24 00:37:51 +01008856 Py_ssize_t thousands_sep_len;
8857 Py_ssize_t len;
8858
8859 if (unicode != NULL) {
8860 kind = PyUnicode_KIND(unicode);
Antoine Pitrou842c0f12012-02-24 13:30:46 +01008861 data = (char *) PyUnicode_DATA(unicode) + index * kind;
Victor Stinner41a863c2012-02-24 00:37:51 +01008862 }
8863 else {
8864 kind = PyUnicode_1BYTE_KIND;
8865 data = NULL;
8866 }
8867 thousands_sep_kind = PyUnicode_KIND(thousands_sep);
8868 thousands_sep_data = PyUnicode_DATA(thousands_sep);
8869 thousands_sep_len = PyUnicode_GET_LENGTH(thousands_sep);
8870 if (unicode != NULL && thousands_sep_kind != kind) {
Victor Stinner90f50d42012-02-24 01:44:47 +01008871 if (thousands_sep_kind < kind) {
8872 thousands_sep_data = _PyUnicode_AsKind(thousands_sep, kind);
8873 if (!thousands_sep_data)
8874 return -1;
8875 }
8876 else {
8877 data = _PyUnicode_AsKind(unicode, thousands_sep_kind);
8878 if (!data)
8879 return -1;
8880 }
Victor Stinner41a863c2012-02-24 00:37:51 +01008881 }
8882
Benjamin Petersonead6b532011-12-20 17:23:42 -06008883 switch (kind) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008884 case PyUnicode_1BYTE_KIND:
Victor Stinnerc3cec782011-10-05 21:24:08 +02008885 if (unicode != NULL && PyUnicode_IS_ASCII(unicode))
Victor Stinner41a863c2012-02-24 00:37:51 +01008886 len = asciilib_InsertThousandsGrouping(
Antoine Pitrou842c0f12012-02-24 13:30:46 +01008887 (Py_UCS1 *) data, n_buffer, (Py_UCS1 *) digits, n_digits,
Victor Stinner41a863c2012-02-24 00:37:51 +01008888 min_width, grouping,
Antoine Pitrou842c0f12012-02-24 13:30:46 +01008889 (Py_UCS1 *) thousands_sep_data, thousands_sep_len);
Victor Stinnerc3cec782011-10-05 21:24:08 +02008890 else
Victor Stinner41a863c2012-02-24 00:37:51 +01008891 len = ucs1lib_InsertThousandsGrouping(
Victor Stinnerc3cec782011-10-05 21:24:08 +02008892 (Py_UCS1*)data, n_buffer, (Py_UCS1*)digits, n_digits,
Victor Stinner41a863c2012-02-24 00:37:51 +01008893 min_width, grouping,
Antoine Pitrou842c0f12012-02-24 13:30:46 +01008894 (Py_UCS1 *) thousands_sep_data, thousands_sep_len);
Victor Stinner41a863c2012-02-24 00:37:51 +01008895 break;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008896 case PyUnicode_2BYTE_KIND:
Victor Stinner41a863c2012-02-24 00:37:51 +01008897 len = ucs2lib_InsertThousandsGrouping(
Antoine Pitrou842c0f12012-02-24 13:30:46 +01008898 (Py_UCS2 *) data, n_buffer, (Py_UCS2 *) digits, n_digits,
Victor Stinner41a863c2012-02-24 00:37:51 +01008899 min_width, grouping,
Antoine Pitrou842c0f12012-02-24 13:30:46 +01008900 (Py_UCS2 *) thousands_sep_data, thousands_sep_len);
Victor Stinner41a863c2012-02-24 00:37:51 +01008901 break;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008902 case PyUnicode_4BYTE_KIND:
Victor Stinner41a863c2012-02-24 00:37:51 +01008903 len = ucs4lib_InsertThousandsGrouping(
Antoine Pitrou842c0f12012-02-24 13:30:46 +01008904 (Py_UCS4 *) data, n_buffer, (Py_UCS4 *) digits, n_digits,
Victor Stinner41a863c2012-02-24 00:37:51 +01008905 min_width, grouping,
Antoine Pitrou842c0f12012-02-24 13:30:46 +01008906 (Py_UCS4 *) thousands_sep_data, thousands_sep_len);
Victor Stinner41a863c2012-02-24 00:37:51 +01008907 break;
8908 default:
8909 assert(0);
8910 return -1;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008911 }
Victor Stinner90f50d42012-02-24 01:44:47 +01008912 if (unicode != NULL && thousands_sep_kind != kind) {
8913 if (thousands_sep_kind < kind)
8914 PyMem_Free(thousands_sep_data);
8915 else
8916 PyMem_Free(data);
8917 }
Victor Stinner41a863c2012-02-24 00:37:51 +01008918 if (unicode == NULL) {
8919 *maxchar = 127;
8920 if (len != n_digits) {
Victor Stinnere6abb482012-05-02 01:15:40 +02008921 *maxchar = MAX_MAXCHAR(*maxchar,
8922 PyUnicode_MAX_CHAR_VALUE(thousands_sep));
Victor Stinner41a863c2012-02-24 00:37:51 +01008923 }
8924 }
8925 return len;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008926}
8927
8928
Thomas Wouters477c8d52006-05-27 19:21:47 +00008929/* helper macro to fixup start/end slice values */
Antoine Pitrouf2c54842010-01-13 08:07:53 +00008930#define ADJUST_INDICES(start, end, len) \
8931 if (end > len) \
8932 end = len; \
8933 else if (end < 0) { \
8934 end += len; \
8935 if (end < 0) \
8936 end = 0; \
8937 } \
8938 if (start < 0) { \
8939 start += len; \
8940 if (start < 0) \
8941 start = 0; \
8942 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00008943
Alexander Belopolsky40018472011-02-26 01:02:56 +00008944Py_ssize_t
8945PyUnicode_Count(PyObject *str,
8946 PyObject *substr,
8947 Py_ssize_t start,
8948 Py_ssize_t end)
Guido van Rossumd57fd912000-03-10 22:53:23 +00008949{
Martin v. Löwis18e16552006-02-15 17:27:45 +00008950 Py_ssize_t result;
Victor Stinner9db1a8b2011-10-23 20:04:37 +02008951 PyObject* str_obj;
8952 PyObject* sub_obj;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008953 int kind1, kind2, kind;
8954 void *buf1 = NULL, *buf2 = NULL;
8955 Py_ssize_t len1, len2;
Tim Petersced69f82003-09-16 20:30:58 +00008956
Victor Stinner9db1a8b2011-10-23 20:04:37 +02008957 str_obj = PyUnicode_FromObject(str);
Benjamin Peterson22a29702012-01-02 09:00:30 -06008958 if (!str_obj)
Benjamin Peterson29060642009-01-31 22:14:21 +00008959 return -1;
Victor Stinner9db1a8b2011-10-23 20:04:37 +02008960 sub_obj = PyUnicode_FromObject(substr);
Benjamin Peterson22a29702012-01-02 09:00:30 -06008961 if (!sub_obj) {
8962 Py_DECREF(str_obj);
8963 return -1;
8964 }
Benjamin Peterson4c13a4a2012-01-02 09:07:38 -06008965 if (PyUnicode_READY(sub_obj) == -1 || PyUnicode_READY(str_obj) == -1) {
Benjamin Peterson5e458f52012-01-02 10:12:13 -06008966 Py_DECREF(sub_obj);
Benjamin Peterson29060642009-01-31 22:14:21 +00008967 Py_DECREF(str_obj);
8968 return -1;
Guido van Rossumd57fd912000-03-10 22:53:23 +00008969 }
Tim Petersced69f82003-09-16 20:30:58 +00008970
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008971 kind1 = PyUnicode_KIND(str_obj);
8972 kind2 = PyUnicode_KIND(sub_obj);
Antoine Pitroue45c0c52012-05-12 15:49:07 +02008973 kind = kind1;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008974 buf1 = PyUnicode_DATA(str_obj);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008975 buf2 = PyUnicode_DATA(sub_obj);
Benjamin Peterson1ff2e352012-05-11 17:41:20 -05008976 if (kind2 != kind) {
Antoine Pitrou758153b2012-05-12 15:51:51 +02008977 if (kind2 > kind) {
8978 Py_DECREF(sub_obj);
8979 Py_DECREF(str_obj);
Antoine Pitroue45c0c52012-05-12 15:49:07 +02008980 return 0;
Antoine Pitrou758153b2012-05-12 15:51:51 +02008981 }
Victor Stinner7931d9a2011-11-04 00:22:48 +01008982 buf2 = _PyUnicode_AsKind(sub_obj, kind);
Benjamin Peterson1ff2e352012-05-11 17:41:20 -05008983 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008984 if (!buf2)
8985 goto onError;
8986 len1 = PyUnicode_GET_LENGTH(str_obj);
8987 len2 = PyUnicode_GET_LENGTH(sub_obj);
8988
8989 ADJUST_INDICES(start, end, len1);
Benjamin Petersonead6b532011-12-20 17:23:42 -06008990 switch (kind) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02008991 case PyUnicode_1BYTE_KIND:
Victor Stinnerc3cec782011-10-05 21:24:08 +02008992 if (PyUnicode_IS_ASCII(str_obj) && PyUnicode_IS_ASCII(sub_obj))
8993 result = asciilib_count(
8994 ((Py_UCS1*)buf1) + start, end - start,
8995 buf2, len2, PY_SSIZE_T_MAX
8996 );
8997 else
8998 result = ucs1lib_count(
8999 ((Py_UCS1*)buf1) + start, end - start,
9000 buf2, len2, PY_SSIZE_T_MAX
9001 );
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009002 break;
9003 case PyUnicode_2BYTE_KIND:
9004 result = ucs2lib_count(
9005 ((Py_UCS2*)buf1) + start, end - start,
9006 buf2, len2, PY_SSIZE_T_MAX
9007 );
9008 break;
9009 case PyUnicode_4BYTE_KIND:
9010 result = ucs4lib_count(
9011 ((Py_UCS4*)buf1) + start, end - start,
9012 buf2, len2, PY_SSIZE_T_MAX
9013 );
9014 break;
9015 default:
9016 assert(0); result = 0;
9017 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00009018
9019 Py_DECREF(sub_obj);
9020 Py_DECREF(str_obj);
9021
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009022 if (kind2 != kind)
9023 PyMem_Free(buf2);
9024
Guido van Rossumd57fd912000-03-10 22:53:23 +00009025 return result;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009026 onError:
9027 Py_DECREF(sub_obj);
9028 Py_DECREF(str_obj);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009029 if (kind2 != kind && buf2)
9030 PyMem_Free(buf2);
9031 return -1;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009032}
9033
Alexander Belopolsky40018472011-02-26 01:02:56 +00009034Py_ssize_t
9035PyUnicode_Find(PyObject *str,
9036 PyObject *sub,
9037 Py_ssize_t start,
9038 Py_ssize_t end,
9039 int direction)
Guido van Rossumd57fd912000-03-10 22:53:23 +00009040{
Martin v. Löwis18e16552006-02-15 17:27:45 +00009041 Py_ssize_t result;
Tim Petersced69f82003-09-16 20:30:58 +00009042
Guido van Rossumd57fd912000-03-10 22:53:23 +00009043 str = PyUnicode_FromObject(str);
Benjamin Peterson22a29702012-01-02 09:00:30 -06009044 if (!str)
Benjamin Peterson29060642009-01-31 22:14:21 +00009045 return -2;
Thomas Wouters477c8d52006-05-27 19:21:47 +00009046 sub = PyUnicode_FromObject(sub);
Benjamin Peterson22a29702012-01-02 09:00:30 -06009047 if (!sub) {
9048 Py_DECREF(str);
9049 return -2;
9050 }
9051 if (PyUnicode_READY(sub) == -1 || PyUnicode_READY(str) == -1) {
9052 Py_DECREF(sub);
Benjamin Peterson29060642009-01-31 22:14:21 +00009053 Py_DECREF(str);
9054 return -2;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009055 }
Tim Petersced69f82003-09-16 20:30:58 +00009056
Victor Stinner794d5672011-10-10 03:21:36 +02009057 result = any_find_slice(direction,
9058 str, sub, start, end
9059 );
Thomas Wouters477c8d52006-05-27 19:21:47 +00009060
Guido van Rossumd57fd912000-03-10 22:53:23 +00009061 Py_DECREF(str);
Thomas Wouters477c8d52006-05-27 19:21:47 +00009062 Py_DECREF(sub);
9063
Guido van Rossumd57fd912000-03-10 22:53:23 +00009064 return result;
9065}
9066
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009067Py_ssize_t
9068PyUnicode_FindChar(PyObject *str, Py_UCS4 ch,
9069 Py_ssize_t start, Py_ssize_t end,
9070 int direction)
9071{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009072 int kind;
Antoine Pitrouf0b934b2011-10-13 18:55:09 +02009073 Py_ssize_t result;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009074 if (PyUnicode_READY(str) == -1)
9075 return -2;
Victor Stinner267aa242011-10-02 01:08:37 +02009076 if (start < 0 || end < 0) {
9077 PyErr_SetString(PyExc_IndexError, "string index out of range");
9078 return -2;
9079 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009080 if (end > PyUnicode_GET_LENGTH(str))
9081 end = PyUnicode_GET_LENGTH(str);
9082 kind = PyUnicode_KIND(str);
Antoine Pitrouf0b934b2011-10-13 18:55:09 +02009083 result = findchar(PyUnicode_1BYTE_DATA(str) + kind*start,
9084 kind, end-start, ch, direction);
9085 if (result == -1)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009086 return -1;
Antoine Pitrouf0b934b2011-10-13 18:55:09 +02009087 else
9088 return start + result;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009089}
9090
Alexander Belopolsky40018472011-02-26 01:02:56 +00009091static int
Victor Stinner9db1a8b2011-10-23 20:04:37 +02009092tailmatch(PyObject *self,
9093 PyObject *substring,
Alexander Belopolsky40018472011-02-26 01:02:56 +00009094 Py_ssize_t start,
9095 Py_ssize_t end,
9096 int direction)
Guido van Rossumd57fd912000-03-10 22:53:23 +00009097{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009098 int kind_self;
9099 int kind_sub;
9100 void *data_self;
9101 void *data_sub;
9102 Py_ssize_t offset;
9103 Py_ssize_t i;
9104 Py_ssize_t end_sub;
9105
9106 if (PyUnicode_READY(self) == -1 ||
9107 PyUnicode_READY(substring) == -1)
9108 return 0;
9109
9110 if (PyUnicode_GET_LENGTH(substring) == 0)
Guido van Rossumd57fd912000-03-10 22:53:23 +00009111 return 1;
9112
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009113 ADJUST_INDICES(start, end, PyUnicode_GET_LENGTH(self));
9114 end -= PyUnicode_GET_LENGTH(substring);
Guido van Rossumd57fd912000-03-10 22:53:23 +00009115 if (end < start)
Benjamin Peterson29060642009-01-31 22:14:21 +00009116 return 0;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009117
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009118 kind_self = PyUnicode_KIND(self);
9119 data_self = PyUnicode_DATA(self);
9120 kind_sub = PyUnicode_KIND(substring);
9121 data_sub = PyUnicode_DATA(substring);
9122 end_sub = PyUnicode_GET_LENGTH(substring) - 1;
9123
9124 if (direction > 0)
9125 offset = end;
9126 else
9127 offset = start;
9128
9129 if (PyUnicode_READ(kind_self, data_self, offset) ==
9130 PyUnicode_READ(kind_sub, data_sub, 0) &&
9131 PyUnicode_READ(kind_self, data_self, offset + end_sub) ==
9132 PyUnicode_READ(kind_sub, data_sub, end_sub)) {
9133 /* If both are of the same kind, memcmp is sufficient */
9134 if (kind_self == kind_sub) {
9135 return ! memcmp((char *)data_self +
Martin v. Löwisc47adb02011-10-07 20:55:35 +02009136 (offset * PyUnicode_KIND(substring)),
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009137 data_sub,
9138 PyUnicode_GET_LENGTH(substring) *
Martin v. Löwisc47adb02011-10-07 20:55:35 +02009139 PyUnicode_KIND(substring));
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009140 }
9141 /* otherwise we have to compare each character by first accesing it */
9142 else {
9143 /* We do not need to compare 0 and len(substring)-1 because
9144 the if statement above ensured already that they are equal
9145 when we end up here. */
Antoine Pitrou057119b2012-09-02 17:56:33 +02009146 /* TODO: honor direction and do a forward or backwards search */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009147 for (i = 1; i < end_sub; ++i) {
9148 if (PyUnicode_READ(kind_self, data_self, offset + i) !=
9149 PyUnicode_READ(kind_sub, data_sub, i))
9150 return 0;
9151 }
Benjamin Peterson29060642009-01-31 22:14:21 +00009152 return 1;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009153 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00009154 }
9155
9156 return 0;
9157}
9158
Alexander Belopolsky40018472011-02-26 01:02:56 +00009159Py_ssize_t
9160PyUnicode_Tailmatch(PyObject *str,
9161 PyObject *substr,
9162 Py_ssize_t start,
9163 Py_ssize_t end,
9164 int direction)
Guido van Rossumd57fd912000-03-10 22:53:23 +00009165{
Martin v. Löwis18e16552006-02-15 17:27:45 +00009166 Py_ssize_t result;
Tim Petersced69f82003-09-16 20:30:58 +00009167
Guido van Rossumd57fd912000-03-10 22:53:23 +00009168 str = PyUnicode_FromObject(str);
9169 if (str == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00009170 return -1;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009171 substr = PyUnicode_FromObject(substr);
9172 if (substr == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +00009173 Py_DECREF(str);
9174 return -1;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009175 }
Tim Petersced69f82003-09-16 20:30:58 +00009176
Victor Stinner9db1a8b2011-10-23 20:04:37 +02009177 result = tailmatch(str, substr,
Benjamin Peterson29060642009-01-31 22:14:21 +00009178 start, end, direction);
Guido van Rossumd57fd912000-03-10 22:53:23 +00009179 Py_DECREF(str);
9180 Py_DECREF(substr);
9181 return result;
9182}
9183
Guido van Rossumd57fd912000-03-10 22:53:23 +00009184/* Apply fixfct filter to the Unicode object self and return a
9185 reference to the modified object */
9186
Alexander Belopolsky40018472011-02-26 01:02:56 +00009187static PyObject *
Victor Stinner9310abb2011-10-05 00:59:23 +02009188fixup(PyObject *self,
9189 Py_UCS4 (*fixfct)(PyObject *s))
Guido van Rossumd57fd912000-03-10 22:53:23 +00009190{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009191 PyObject *u;
9192 Py_UCS4 maxchar_old, maxchar_new = 0;
Victor Stinnereaab6042011-12-11 22:22:39 +01009193 PyObject *v;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009194
Victor Stinnerbf6e5602011-12-12 01:53:47 +01009195 u = _PyUnicode_Copy(self);
Guido van Rossumd57fd912000-03-10 22:53:23 +00009196 if (u == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00009197 return NULL;
Victor Stinner87af4f22011-11-21 23:03:47 +01009198 maxchar_old = PyUnicode_MAX_CHAR_VALUE(u);
Marc-André Lemburg8155e0e2001-04-23 14:44:21 +00009199
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009200 /* fix functions return the new maximum character in a string,
9201 if the kind of the resulting unicode object does not change,
9202 everything is fine. Otherwise we need to change the string kind
9203 and re-run the fix function. */
Victor Stinner9310abb2011-10-05 00:59:23 +02009204 maxchar_new = fixfct(u);
Victor Stinnereaab6042011-12-11 22:22:39 +01009205
9206 if (maxchar_new == 0) {
9207 /* no changes */;
9208 if (PyUnicode_CheckExact(self)) {
9209 Py_DECREF(u);
9210 Py_INCREF(self);
9211 return self;
9212 }
9213 else
9214 return u;
9215 }
9216
Victor Stinnere6abb482012-05-02 01:15:40 +02009217 maxchar_new = align_maxchar(maxchar_new);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009218
Victor Stinnereaab6042011-12-11 22:22:39 +01009219 if (maxchar_new == maxchar_old)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009220 return u;
Victor Stinnereaab6042011-12-11 22:22:39 +01009221
9222 /* In case the maximum character changed, we need to
9223 convert the string to the new category. */
9224 v = PyUnicode_New(PyUnicode_GET_LENGTH(self), maxchar_new);
9225 if (v == NULL) {
9226 Py_DECREF(u);
9227 return NULL;
9228 }
9229 if (maxchar_new > maxchar_old) {
9230 /* If the maxchar increased so that the kind changed, not all
9231 characters are representable anymore and we need to fix the
9232 string again. This only happens in very few cases. */
Victor Stinnerd3f08822012-05-29 12:57:52 +02009233 _PyUnicode_FastCopyCharacters(v, 0,
9234 self, 0, PyUnicode_GET_LENGTH(self));
Victor Stinnereaab6042011-12-11 22:22:39 +01009235 maxchar_old = fixfct(v);
9236 assert(maxchar_old > 0 && maxchar_old <= maxchar_new);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009237 }
9238 else {
Victor Stinnerd3f08822012-05-29 12:57:52 +02009239 _PyUnicode_FastCopyCharacters(v, 0,
9240 u, 0, PyUnicode_GET_LENGTH(self));
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009241 }
Victor Stinnereaab6042011-12-11 22:22:39 +01009242 Py_DECREF(u);
9243 assert(_PyUnicode_CheckConsistency(v, 1));
9244 return v;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009245}
9246
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009247static PyObject *
9248ascii_upper_or_lower(PyObject *self, int lower)
Guido van Rossumd57fd912000-03-10 22:53:23 +00009249{
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009250 Py_ssize_t len = PyUnicode_GET_LENGTH(self);
9251 char *resdata, *data = PyUnicode_DATA(self);
9252 PyObject *res;
Tim Petersced69f82003-09-16 20:30:58 +00009253
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009254 res = PyUnicode_New(len, 127);
9255 if (res == NULL)
9256 return NULL;
9257 resdata = PyUnicode_DATA(res);
9258 if (lower)
9259 _Py_bytes_lower(resdata, data, len);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009260 else
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009261 _Py_bytes_upper(resdata, data, len);
9262 return res;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009263}
9264
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009265static Py_UCS4
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009266handle_capital_sigma(int kind, void *data, Py_ssize_t length, Py_ssize_t i)
Guido van Rossumd57fd912000-03-10 22:53:23 +00009267{
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009268 Py_ssize_t j;
9269 int final_sigma;
9270 Py_UCS4 c;
9271 /* U+03A3 is in the Final_Sigma context when, it is found like this:
Tim Petersced69f82003-09-16 20:30:58 +00009272
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009273 \p{cased}\p{case-ignorable}*U+03A3!(\p{case-ignorable}*\p{cased})
9274
9275 where ! is a negation and \p{xxx} is a character with property xxx.
9276 */
9277 for (j = i - 1; j >= 0; j--) {
9278 c = PyUnicode_READ(kind, data, j);
9279 if (!_PyUnicode_IsCaseIgnorable(c))
9280 break;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009281 }
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009282 final_sigma = j >= 0 && _PyUnicode_IsCased(c);
9283 if (final_sigma) {
9284 for (j = i + 1; j < length; j++) {
9285 c = PyUnicode_READ(kind, data, j);
9286 if (!_PyUnicode_IsCaseIgnorable(c))
9287 break;
9288 }
9289 final_sigma = j == length || !_PyUnicode_IsCased(c);
9290 }
9291 return (final_sigma) ? 0x3C2 : 0x3C3;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009292}
9293
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009294static int
9295lower_ucs4(int kind, void *data, Py_ssize_t length, Py_ssize_t i,
9296 Py_UCS4 c, Py_UCS4 *mapped)
Guido van Rossumd57fd912000-03-10 22:53:23 +00009297{
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009298 /* Obscure special case. */
9299 if (c == 0x3A3) {
9300 mapped[0] = handle_capital_sigma(kind, data, length, i);
9301 return 1;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009302 }
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009303 return _PyUnicode_ToLowerFull(c, mapped);
Guido van Rossumd57fd912000-03-10 22:53:23 +00009304}
9305
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009306static Py_ssize_t
9307do_capitalize(int kind, void *data, Py_ssize_t length, Py_UCS4 *res, Py_UCS4 *maxchar)
Guido van Rossumd57fd912000-03-10 22:53:23 +00009308{
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009309 Py_ssize_t i, k = 0;
9310 int n_res, j;
9311 Py_UCS4 c, mapped[3];
Tim Petersced69f82003-09-16 20:30:58 +00009312
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009313 c = PyUnicode_READ(kind, data, 0);
9314 n_res = _PyUnicode_ToUpperFull(c, mapped);
9315 for (j = 0; j < n_res; j++) {
Victor Stinnere6abb482012-05-02 01:15:40 +02009316 *maxchar = MAX_MAXCHAR(*maxchar, mapped[j]);
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009317 res[k++] = mapped[j];
Guido van Rossumd57fd912000-03-10 22:53:23 +00009318 }
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009319 for (i = 1; i < length; i++) {
9320 c = PyUnicode_READ(kind, data, i);
9321 n_res = lower_ucs4(kind, data, length, i, c, mapped);
9322 for (j = 0; j < n_res; j++) {
Victor Stinnere6abb482012-05-02 01:15:40 +02009323 *maxchar = MAX_MAXCHAR(*maxchar, mapped[j]);
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009324 res[k++] = mapped[j];
Marc-André Lemburgfde66e12001-01-29 11:14:16 +00009325 }
Marc-André Lemburgfde66e12001-01-29 11:14:16 +00009326 }
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009327 return k;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009328}
9329
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009330static Py_ssize_t
9331do_swapcase(int kind, void *data, Py_ssize_t length, Py_UCS4 *res, Py_UCS4 *maxchar) {
9332 Py_ssize_t i, k = 0;
9333
9334 for (i = 0; i < length; i++) {
9335 Py_UCS4 c = PyUnicode_READ(kind, data, i), mapped[3];
9336 int n_res, j;
9337 if (Py_UNICODE_ISUPPER(c)) {
9338 n_res = lower_ucs4(kind, data, length, i, c, mapped);
9339 }
9340 else if (Py_UNICODE_ISLOWER(c)) {
9341 n_res = _PyUnicode_ToUpperFull(c, mapped);
9342 }
9343 else {
9344 n_res = 1;
9345 mapped[0] = c;
9346 }
9347 for (j = 0; j < n_res; j++) {
Victor Stinnere6abb482012-05-02 01:15:40 +02009348 *maxchar = MAX_MAXCHAR(*maxchar, mapped[j]);
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009349 res[k++] = mapped[j];
9350 }
9351 }
9352 return k;
9353}
9354
9355static Py_ssize_t
9356do_upper_or_lower(int kind, void *data, Py_ssize_t length, Py_UCS4 *res,
9357 Py_UCS4 *maxchar, int lower)
Guido van Rossumd57fd912000-03-10 22:53:23 +00009358{
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009359 Py_ssize_t i, k = 0;
9360
9361 for (i = 0; i < length; i++) {
9362 Py_UCS4 c = PyUnicode_READ(kind, data, i), mapped[3];
9363 int n_res, j;
9364 if (lower)
9365 n_res = lower_ucs4(kind, data, length, i, c, mapped);
9366 else
9367 n_res = _PyUnicode_ToUpperFull(c, mapped);
9368 for (j = 0; j < n_res; j++) {
Victor Stinnere6abb482012-05-02 01:15:40 +02009369 *maxchar = MAX_MAXCHAR(*maxchar, mapped[j]);
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009370 res[k++] = mapped[j];
9371 }
9372 }
9373 return k;
9374}
9375
9376static Py_ssize_t
9377do_upper(int kind, void *data, Py_ssize_t length, Py_UCS4 *res, Py_UCS4 *maxchar)
9378{
9379 return do_upper_or_lower(kind, data, length, res, maxchar, 0);
9380}
9381
9382static Py_ssize_t
9383do_lower(int kind, void *data, Py_ssize_t length, Py_UCS4 *res, Py_UCS4 *maxchar)
9384{
9385 return do_upper_or_lower(kind, data, length, res, maxchar, 1);
9386}
9387
Benjamin Petersone51757f2012-01-12 21:10:29 -05009388static Py_ssize_t
Benjamin Petersond5890c82012-01-14 13:23:30 -05009389do_casefold(int kind, void *data, Py_ssize_t length, Py_UCS4 *res, Py_UCS4 *maxchar)
9390{
9391 Py_ssize_t i, k = 0;
9392
9393 for (i = 0; i < length; i++) {
9394 Py_UCS4 c = PyUnicode_READ(kind, data, i);
9395 Py_UCS4 mapped[3];
9396 int j, n_res = _PyUnicode_ToFoldedFull(c, mapped);
9397 for (j = 0; j < n_res; j++) {
Victor Stinnere6abb482012-05-02 01:15:40 +02009398 *maxchar = MAX_MAXCHAR(*maxchar, mapped[j]);
Benjamin Petersond5890c82012-01-14 13:23:30 -05009399 res[k++] = mapped[j];
9400 }
9401 }
9402 return k;
9403}
9404
9405static Py_ssize_t
Benjamin Petersone51757f2012-01-12 21:10:29 -05009406do_title(int kind, void *data, Py_ssize_t length, Py_UCS4 *res, Py_UCS4 *maxchar)
9407{
9408 Py_ssize_t i, k = 0;
9409 int previous_is_cased;
9410
9411 previous_is_cased = 0;
9412 for (i = 0; i < length; i++) {
9413 const Py_UCS4 c = PyUnicode_READ(kind, data, i);
9414 Py_UCS4 mapped[3];
9415 int n_res, j;
9416
9417 if (previous_is_cased)
9418 n_res = lower_ucs4(kind, data, length, i, c, mapped);
9419 else
9420 n_res = _PyUnicode_ToTitleFull(c, mapped);
9421
9422 for (j = 0; j < n_res; j++) {
Victor Stinnere6abb482012-05-02 01:15:40 +02009423 *maxchar = MAX_MAXCHAR(*maxchar, mapped[j]);
Benjamin Petersone51757f2012-01-12 21:10:29 -05009424 res[k++] = mapped[j];
9425 }
9426
9427 previous_is_cased = _PyUnicode_IsCased(c);
9428 }
9429 return k;
9430}
9431
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009432static PyObject *
9433case_operation(PyObject *self,
9434 Py_ssize_t (*perform)(int, void *, Py_ssize_t, Py_UCS4 *, Py_UCS4 *))
9435{
9436 PyObject *res = NULL;
9437 Py_ssize_t length, newlength = 0;
9438 int kind, outkind;
9439 void *data, *outdata;
9440 Py_UCS4 maxchar = 0, *tmp, *tmpend;
9441
Benjamin Petersoneea48462012-01-16 14:28:50 -05009442 assert(PyUnicode_IS_READY(self));
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -05009443
9444 kind = PyUnicode_KIND(self);
9445 data = PyUnicode_DATA(self);
9446 length = PyUnicode_GET_LENGTH(self);
9447 tmp = PyMem_MALLOC(sizeof(Py_UCS4) * 3 * length);
9448 if (tmp == NULL)
9449 return PyErr_NoMemory();
9450 newlength = perform(kind, data, length, tmp, &maxchar);
9451 res = PyUnicode_New(newlength, maxchar);
9452 if (res == NULL)
9453 goto leave;
9454 tmpend = tmp + newlength;
9455 outdata = PyUnicode_DATA(res);
9456 outkind = PyUnicode_KIND(res);
9457 switch (outkind) {
9458 case PyUnicode_1BYTE_KIND:
9459 _PyUnicode_CONVERT_BYTES(Py_UCS4, Py_UCS1, tmp, tmpend, outdata);
9460 break;
9461 case PyUnicode_2BYTE_KIND:
9462 _PyUnicode_CONVERT_BYTES(Py_UCS4, Py_UCS2, tmp, tmpend, outdata);
9463 break;
9464 case PyUnicode_4BYTE_KIND:
9465 memcpy(outdata, tmp, sizeof(Py_UCS4) * newlength);
9466 break;
9467 default:
9468 assert(0);
9469 break;
9470 }
9471 leave:
9472 PyMem_FREE(tmp);
9473 return res;
9474}
9475
Tim Peters8ce9f162004-08-27 01:49:32 +00009476PyObject *
9477PyUnicode_Join(PyObject *separator, PyObject *seq)
Guido van Rossumd57fd912000-03-10 22:53:23 +00009478{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009479 PyObject *sep = NULL;
Victor Stinnerdd077322011-10-07 17:02:31 +02009480 Py_ssize_t seplen;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009481 PyObject *res = NULL; /* the result */
Tim Peters05eba1f2004-08-27 21:32:02 +00009482 PyObject *fseq; /* PySequence_Fast(seq) */
Antoine Pitrouaf14b792008-08-07 21:50:41 +00009483 Py_ssize_t seqlen; /* len(fseq) -- number of items in sequence */
9484 PyObject **items;
Tim Peters8ce9f162004-08-27 01:49:32 +00009485 PyObject *item;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009486 Py_ssize_t sz, i, res_offset;
Victor Stinnerfb9ea8c2011-10-06 01:45:57 +02009487 Py_UCS4 maxchar;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009488 Py_UCS4 item_maxchar;
Victor Stinnerdd077322011-10-07 17:02:31 +02009489 int use_memcpy;
9490 unsigned char *res_data = NULL, *sep_data = NULL;
9491 PyObject *last_obj;
9492 unsigned int kind = 0;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009493
Tim Peters05eba1f2004-08-27 21:32:02 +00009494 fseq = PySequence_Fast(seq, "");
9495 if (fseq == NULL) {
Benjamin Peterson14339b62009-01-31 16:36:08 +00009496 return NULL;
Tim Peters8ce9f162004-08-27 01:49:32 +00009497 }
9498
Antoine Pitrouaf14b792008-08-07 21:50:41 +00009499 /* NOTE: the following code can't call back into Python code,
9500 * so we are sure that fseq won't be mutated.
Tim Peters91879ab2004-08-27 22:35:44 +00009501 */
Antoine Pitrouaf14b792008-08-07 21:50:41 +00009502
Tim Peters05eba1f2004-08-27 21:32:02 +00009503 seqlen = PySequence_Fast_GET_SIZE(fseq);
9504 /* If empty sequence, return u"". */
9505 if (seqlen == 0) {
Victor Stinnerfb9ea8c2011-10-06 01:45:57 +02009506 Py_DECREF(fseq);
9507 Py_INCREF(unicode_empty);
9508 res = unicode_empty;
9509 return res;
Tim Peters05eba1f2004-08-27 21:32:02 +00009510 }
Victor Stinnerfb9ea8c2011-10-06 01:45:57 +02009511
Tim Peters05eba1f2004-08-27 21:32:02 +00009512 /* If singleton sequence with an exact Unicode, return that. */
Victor Stinnerdd077322011-10-07 17:02:31 +02009513 last_obj = NULL;
Victor Stinnerfb9ea8c2011-10-06 01:45:57 +02009514 items = PySequence_Fast_ITEMS(fseq);
Victor Stinneracf47b82011-10-06 12:32:37 +02009515 if (seqlen == 1) {
9516 if (PyUnicode_CheckExact(items[0])) {
9517 res = items[0];
9518 Py_INCREF(res);
9519 Py_DECREF(fseq);
9520 return res;
9521 }
Victor Stinnerdd077322011-10-07 17:02:31 +02009522 seplen = 0;
Victor Stinnerc6f0df72011-10-06 15:58:54 +02009523 maxchar = 0;
Tim Peters8ce9f162004-08-27 01:49:32 +00009524 }
Antoine Pitrouaf14b792008-08-07 21:50:41 +00009525 else {
Victor Stinneracf47b82011-10-06 12:32:37 +02009526 /* Set up sep and seplen */
9527 if (separator == NULL) {
9528 /* fall back to a blank space separator */
9529 sep = PyUnicode_FromOrdinal(' ');
9530 if (!sep)
9531 goto onError;
Victor Stinnerdd077322011-10-07 17:02:31 +02009532 seplen = 1;
Victor Stinneracf47b82011-10-06 12:32:37 +02009533 maxchar = 32;
Tim Peters05eba1f2004-08-27 21:32:02 +00009534 }
Victor Stinneracf47b82011-10-06 12:32:37 +02009535 else {
9536 if (!PyUnicode_Check(separator)) {
9537 PyErr_Format(PyExc_TypeError,
9538 "separator: expected str instance,"
9539 " %.80s found",
9540 Py_TYPE(separator)->tp_name);
9541 goto onError;
9542 }
9543 if (PyUnicode_READY(separator))
9544 goto onError;
9545 sep = separator;
9546 seplen = PyUnicode_GET_LENGTH(separator);
9547 maxchar = PyUnicode_MAX_CHAR_VALUE(separator);
9548 /* inc refcount to keep this code path symmetric with the
9549 above case of a blank separator */
9550 Py_INCREF(sep);
9551 }
Victor Stinnerdd077322011-10-07 17:02:31 +02009552 last_obj = sep;
Tim Peters05eba1f2004-08-27 21:32:02 +00009553 }
9554
Antoine Pitrouaf14b792008-08-07 21:50:41 +00009555 /* There are at least two things to join, or else we have a subclass
9556 * of str in the sequence.
9557 * Do a pre-pass to figure out the total amount of space we'll
9558 * need (sz), and see whether all argument are strings.
9559 */
9560 sz = 0;
Victor Stinnerdd077322011-10-07 17:02:31 +02009561#ifdef Py_DEBUG
9562 use_memcpy = 0;
9563#else
9564 use_memcpy = 1;
9565#endif
Antoine Pitrouaf14b792008-08-07 21:50:41 +00009566 for (i = 0; i < seqlen; i++) {
9567 const Py_ssize_t old_sz = sz;
9568 item = items[i];
Benjamin Peterson29060642009-01-31 22:14:21 +00009569 if (!PyUnicode_Check(item)) {
9570 PyErr_Format(PyExc_TypeError,
9571 "sequence item %zd: expected str instance,"
9572 " %.80s found",
9573 i, Py_TYPE(item)->tp_name);
9574 goto onError;
9575 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009576 if (PyUnicode_READY(item) == -1)
9577 goto onError;
9578 sz += PyUnicode_GET_LENGTH(item);
9579 item_maxchar = PyUnicode_MAX_CHAR_VALUE(item);
Victor Stinnere6abb482012-05-02 01:15:40 +02009580 maxchar = MAX_MAXCHAR(maxchar, item_maxchar);
Antoine Pitrouaf14b792008-08-07 21:50:41 +00009581 if (i != 0)
9582 sz += seplen;
9583 if (sz < old_sz || sz > PY_SSIZE_T_MAX) {
9584 PyErr_SetString(PyExc_OverflowError,
Benjamin Peterson29060642009-01-31 22:14:21 +00009585 "join() result is too long for a Python string");
Antoine Pitrouaf14b792008-08-07 21:50:41 +00009586 goto onError;
9587 }
Victor Stinnerdd077322011-10-07 17:02:31 +02009588 if (use_memcpy && last_obj != NULL) {
9589 if (PyUnicode_KIND(last_obj) != PyUnicode_KIND(item))
9590 use_memcpy = 0;
9591 }
9592 last_obj = item;
Antoine Pitrouaf14b792008-08-07 21:50:41 +00009593 }
Tim Petersced69f82003-09-16 20:30:58 +00009594
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009595 res = PyUnicode_New(sz, maxchar);
Antoine Pitrouaf14b792008-08-07 21:50:41 +00009596 if (res == NULL)
9597 goto onError;
Tim Peters91879ab2004-08-27 22:35:44 +00009598
Antoine Pitrouaf14b792008-08-07 21:50:41 +00009599 /* Catenate everything. */
Victor Stinnerdd077322011-10-07 17:02:31 +02009600#ifdef Py_DEBUG
9601 use_memcpy = 0;
9602#else
9603 if (use_memcpy) {
9604 res_data = PyUnicode_1BYTE_DATA(res);
9605 kind = PyUnicode_KIND(res);
9606 if (seplen != 0)
9607 sep_data = PyUnicode_1BYTE_DATA(sep);
9608 }
9609#endif
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009610 for (i = 0, res_offset = 0; i < seqlen; ++i) {
Victor Stinnerfb9ea8c2011-10-06 01:45:57 +02009611 Py_ssize_t itemlen;
Antoine Pitrouaf14b792008-08-07 21:50:41 +00009612 item = items[i];
Benjamin Peterson29060642009-01-31 22:14:21 +00009613 /* Copy item, and maybe the separator. */
Victor Stinner9ce5a832011-10-03 23:36:02 +02009614 if (i && seplen != 0) {
Victor Stinnerdd077322011-10-07 17:02:31 +02009615 if (use_memcpy) {
9616 Py_MEMCPY(res_data,
9617 sep_data,
Martin v. Löwisc47adb02011-10-07 20:55:35 +02009618 kind * seplen);
9619 res_data += kind * seplen;
Victor Stinnerdd077322011-10-07 17:02:31 +02009620 }
9621 else {
Victor Stinnerd3f08822012-05-29 12:57:52 +02009622 _PyUnicode_FastCopyCharacters(res, res_offset, sep, 0, seplen);
Victor Stinnerdd077322011-10-07 17:02:31 +02009623 res_offset += seplen;
9624 }
Benjamin Peterson29060642009-01-31 22:14:21 +00009625 }
Victor Stinner9ce5a832011-10-03 23:36:02 +02009626 itemlen = PyUnicode_GET_LENGTH(item);
9627 if (itemlen != 0) {
Victor Stinnerdd077322011-10-07 17:02:31 +02009628 if (use_memcpy) {
9629 Py_MEMCPY(res_data,
9630 PyUnicode_DATA(item),
Martin v. Löwisc47adb02011-10-07 20:55:35 +02009631 kind * itemlen);
9632 res_data += kind * itemlen;
Victor Stinnerdd077322011-10-07 17:02:31 +02009633 }
9634 else {
Victor Stinnerd3f08822012-05-29 12:57:52 +02009635 _PyUnicode_FastCopyCharacters(res, res_offset, item, 0, itemlen);
Victor Stinnerdd077322011-10-07 17:02:31 +02009636 res_offset += itemlen;
9637 }
Victor Stinner9ce5a832011-10-03 23:36:02 +02009638 }
Tim Peters05eba1f2004-08-27 21:32:02 +00009639 }
Victor Stinnerdd077322011-10-07 17:02:31 +02009640 if (use_memcpy)
9641 assert(res_data == PyUnicode_1BYTE_DATA(res)
Martin v. Löwisc47adb02011-10-07 20:55:35 +02009642 + kind * PyUnicode_GET_LENGTH(res));
Victor Stinnerdd077322011-10-07 17:02:31 +02009643 else
9644 assert(res_offset == PyUnicode_GET_LENGTH(res));
Tim Peters8ce9f162004-08-27 01:49:32 +00009645
Tim Peters05eba1f2004-08-27 21:32:02 +00009646 Py_DECREF(fseq);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009647 Py_XDECREF(sep);
Victor Stinnerbb10a1f2011-10-05 01:34:17 +02009648 assert(_PyUnicode_CheckConsistency(res, 1));
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009649 return res;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009650
Benjamin Peterson29060642009-01-31 22:14:21 +00009651 onError:
Tim Peters05eba1f2004-08-27 21:32:02 +00009652 Py_DECREF(fseq);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009653 Py_XDECREF(sep);
Tim Peters8ce9f162004-08-27 01:49:32 +00009654 Py_XDECREF(res);
Guido van Rossumd57fd912000-03-10 22:53:23 +00009655 return NULL;
9656}
9657
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009658#define FILL(kind, data, value, start, length) \
9659 do { \
9660 Py_ssize_t i_ = 0; \
9661 assert(kind != PyUnicode_WCHAR_KIND); \
9662 switch ((kind)) { \
9663 case PyUnicode_1BYTE_KIND: { \
9664 unsigned char * to_ = (unsigned char *)((data)) + (start); \
Victor Stinnerf2c76aa2012-05-03 13:10:40 +02009665 memset(to_, (unsigned char)value, (length)); \
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009666 break; \
9667 } \
9668 case PyUnicode_2BYTE_KIND: { \
9669 Py_UCS2 * to_ = (Py_UCS2 *)((data)) + (start); \
9670 for (; i_ < (length); ++i_, ++to_) *to_ = (value); \
9671 break; \
9672 } \
Benjamin Petersone157cf12012-01-01 15:56:20 -06009673 case PyUnicode_4BYTE_KIND: { \
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009674 Py_UCS4 * to_ = (Py_UCS4 *)((data)) + (start); \
9675 for (; i_ < (length); ++i_, ++to_) *to_ = (value); \
9676 break; \
Benjamin Petersone157cf12012-01-01 15:56:20 -06009677 default: assert(0); \
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009678 } \
9679 } \
9680 } while (0)
9681
Victor Stinnerd3f08822012-05-29 12:57:52 +02009682void
9683_PyUnicode_FastFill(PyObject *unicode, Py_ssize_t start, Py_ssize_t length,
9684 Py_UCS4 fill_char)
9685{
9686 const enum PyUnicode_Kind kind = PyUnicode_KIND(unicode);
9687 const void *data = PyUnicode_DATA(unicode);
9688 assert(PyUnicode_IS_READY(unicode));
9689 assert(unicode_modifiable(unicode));
9690 assert(fill_char <= PyUnicode_MAX_CHAR_VALUE(unicode));
9691 assert(start >= 0);
9692 assert(start + length <= PyUnicode_GET_LENGTH(unicode));
9693 FILL(kind, data, fill_char, start, length);
9694}
9695
Victor Stinner3fe55312012-01-04 00:33:50 +01009696Py_ssize_t
9697PyUnicode_Fill(PyObject *unicode, Py_ssize_t start, Py_ssize_t length,
9698 Py_UCS4 fill_char)
9699{
9700 Py_ssize_t maxlen;
Victor Stinner3fe55312012-01-04 00:33:50 +01009701
9702 if (!PyUnicode_Check(unicode)) {
9703 PyErr_BadInternalCall();
9704 return -1;
9705 }
9706 if (PyUnicode_READY(unicode) == -1)
9707 return -1;
9708 if (unicode_check_modifiable(unicode))
9709 return -1;
9710
Victor Stinnerd3f08822012-05-29 12:57:52 +02009711 if (start < 0) {
9712 PyErr_SetString(PyExc_IndexError, "string index out of range");
9713 return -1;
9714 }
Victor Stinner3fe55312012-01-04 00:33:50 +01009715 if (fill_char > PyUnicode_MAX_CHAR_VALUE(unicode)) {
9716 PyErr_SetString(PyExc_ValueError,
9717 "fill character is bigger than "
9718 "the string maximum character");
9719 return -1;
9720 }
9721
9722 maxlen = PyUnicode_GET_LENGTH(unicode) - start;
9723 length = Py_MIN(maxlen, length);
9724 if (length <= 0)
9725 return 0;
9726
Victor Stinnerd3f08822012-05-29 12:57:52 +02009727 _PyUnicode_FastFill(unicode, start, length, fill_char);
Victor Stinner3fe55312012-01-04 00:33:50 +01009728 return length;
9729}
9730
Victor Stinner9310abb2011-10-05 00:59:23 +02009731static PyObject *
9732pad(PyObject *self,
Alexander Belopolsky40018472011-02-26 01:02:56 +00009733 Py_ssize_t left,
9734 Py_ssize_t right,
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009735 Py_UCS4 fill)
Guido van Rossumd57fd912000-03-10 22:53:23 +00009736{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009737 PyObject *u;
9738 Py_UCS4 maxchar;
Victor Stinner6c7a52a2011-09-28 21:39:17 +02009739 int kind;
9740 void *data;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009741
9742 if (left < 0)
9743 left = 0;
9744 if (right < 0)
9745 right = 0;
9746
Victor Stinnerc4b49542011-12-11 22:44:26 +01009747 if (left == 0 && right == 0)
9748 return unicode_result_unchanged(self);
Guido van Rossumd57fd912000-03-10 22:53:23 +00009749
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009750 if (left > PY_SSIZE_T_MAX - _PyUnicode_LENGTH(self) ||
9751 right > PY_SSIZE_T_MAX - (left + _PyUnicode_LENGTH(self))) {
Neal Norwitz3ce5d922008-08-24 07:08:55 +00009752 PyErr_SetString(PyExc_OverflowError, "padded string is too long");
9753 return NULL;
9754 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009755 maxchar = PyUnicode_MAX_CHAR_VALUE(self);
Victor Stinnere6abb482012-05-02 01:15:40 +02009756 maxchar = MAX_MAXCHAR(maxchar, fill);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009757 u = PyUnicode_New(left + _PyUnicode_LENGTH(self) + right, maxchar);
Victor Stinner6c7a52a2011-09-28 21:39:17 +02009758 if (!u)
9759 return NULL;
9760
9761 kind = PyUnicode_KIND(u);
9762 data = PyUnicode_DATA(u);
9763 if (left)
9764 FILL(kind, data, fill, 0, left);
9765 if (right)
9766 FILL(kind, data, fill, left + _PyUnicode_LENGTH(self), right);
Victor Stinnerd3f08822012-05-29 12:57:52 +02009767 _PyUnicode_FastCopyCharacters(u, left, self, 0, _PyUnicode_LENGTH(self));
Victor Stinnerbb10a1f2011-10-05 01:34:17 +02009768 assert(_PyUnicode_CheckConsistency(u, 1));
9769 return u;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009770}
9771
Alexander Belopolsky40018472011-02-26 01:02:56 +00009772PyObject *
9773PyUnicode_Splitlines(PyObject *string, int keepends)
Guido van Rossumd57fd912000-03-10 22:53:23 +00009774{
Guido van Rossumd57fd912000-03-10 22:53:23 +00009775 PyObject *list;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009776
9777 string = PyUnicode_FromObject(string);
Benjamin Peterson22a29702012-01-02 09:00:30 -06009778 if (string == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00009779 return NULL;
Benjamin Peterson22a29702012-01-02 09:00:30 -06009780 if (PyUnicode_READY(string) == -1) {
9781 Py_DECREF(string);
9782 return NULL;
9783 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00009784
Benjamin Petersonead6b532011-12-20 17:23:42 -06009785 switch (PyUnicode_KIND(string)) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009786 case PyUnicode_1BYTE_KIND:
Victor Stinnerc3cec782011-10-05 21:24:08 +02009787 if (PyUnicode_IS_ASCII(string))
9788 list = asciilib_splitlines(
Victor Stinner7931d9a2011-11-04 00:22:48 +01009789 string, PyUnicode_1BYTE_DATA(string),
Victor Stinnerc3cec782011-10-05 21:24:08 +02009790 PyUnicode_GET_LENGTH(string), keepends);
9791 else
9792 list = ucs1lib_splitlines(
Victor Stinner7931d9a2011-11-04 00:22:48 +01009793 string, PyUnicode_1BYTE_DATA(string),
Victor Stinnerc3cec782011-10-05 21:24:08 +02009794 PyUnicode_GET_LENGTH(string), keepends);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009795 break;
9796 case PyUnicode_2BYTE_KIND:
9797 list = ucs2lib_splitlines(
Victor Stinner7931d9a2011-11-04 00:22:48 +01009798 string, PyUnicode_2BYTE_DATA(string),
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009799 PyUnicode_GET_LENGTH(string), keepends);
9800 break;
9801 case PyUnicode_4BYTE_KIND:
9802 list = ucs4lib_splitlines(
Victor Stinner7931d9a2011-11-04 00:22:48 +01009803 string, PyUnicode_4BYTE_DATA(string),
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009804 PyUnicode_GET_LENGTH(string), keepends);
9805 break;
9806 default:
9807 assert(0);
9808 list = 0;
9809 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00009810 Py_DECREF(string);
9811 return list;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009812}
9813
Alexander Belopolsky40018472011-02-26 01:02:56 +00009814static PyObject *
Victor Stinner9310abb2011-10-05 00:59:23 +02009815split(PyObject *self,
9816 PyObject *substring,
Alexander Belopolsky40018472011-02-26 01:02:56 +00009817 Py_ssize_t maxcount)
Guido van Rossumd57fd912000-03-10 22:53:23 +00009818{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009819 int kind1, kind2, kind;
9820 void *buf1, *buf2;
9821 Py_ssize_t len1, len2;
9822 PyObject* out;
9823
Guido van Rossumd57fd912000-03-10 22:53:23 +00009824 if (maxcount < 0)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00009825 maxcount = PY_SSIZE_T_MAX;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009826
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009827 if (PyUnicode_READY(self) == -1)
9828 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009829
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009830 if (substring == NULL)
Benjamin Petersonead6b532011-12-20 17:23:42 -06009831 switch (PyUnicode_KIND(self)) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009832 case PyUnicode_1BYTE_KIND:
Victor Stinnerc3cec782011-10-05 21:24:08 +02009833 if (PyUnicode_IS_ASCII(self))
9834 return asciilib_split_whitespace(
Victor Stinner7931d9a2011-11-04 00:22:48 +01009835 self, PyUnicode_1BYTE_DATA(self),
Victor Stinnerc3cec782011-10-05 21:24:08 +02009836 PyUnicode_GET_LENGTH(self), maxcount
9837 );
9838 else
9839 return ucs1lib_split_whitespace(
Victor Stinner7931d9a2011-11-04 00:22:48 +01009840 self, PyUnicode_1BYTE_DATA(self),
Victor Stinnerc3cec782011-10-05 21:24:08 +02009841 PyUnicode_GET_LENGTH(self), maxcount
9842 );
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009843 case PyUnicode_2BYTE_KIND:
9844 return ucs2lib_split_whitespace(
Victor Stinner7931d9a2011-11-04 00:22:48 +01009845 self, PyUnicode_2BYTE_DATA(self),
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009846 PyUnicode_GET_LENGTH(self), maxcount
9847 );
9848 case PyUnicode_4BYTE_KIND:
9849 return ucs4lib_split_whitespace(
Victor Stinner7931d9a2011-11-04 00:22:48 +01009850 self, PyUnicode_4BYTE_DATA(self),
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009851 PyUnicode_GET_LENGTH(self), maxcount
9852 );
9853 default:
9854 assert(0);
9855 return NULL;
9856 }
9857
9858 if (PyUnicode_READY(substring) == -1)
9859 return NULL;
9860
9861 kind1 = PyUnicode_KIND(self);
9862 kind2 = PyUnicode_KIND(substring);
9863 kind = kind1 > kind2 ? kind1 : kind2;
9864 buf1 = PyUnicode_DATA(self);
9865 buf2 = PyUnicode_DATA(substring);
9866 if (kind1 != kind)
Victor Stinner7931d9a2011-11-04 00:22:48 +01009867 buf1 = _PyUnicode_AsKind(self, kind);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009868 if (!buf1)
9869 return NULL;
9870 if (kind2 != kind)
Victor Stinner7931d9a2011-11-04 00:22:48 +01009871 buf2 = _PyUnicode_AsKind(substring, kind);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009872 if (!buf2) {
9873 if (kind1 != kind) PyMem_Free(buf1);
9874 return NULL;
9875 }
9876 len1 = PyUnicode_GET_LENGTH(self);
9877 len2 = PyUnicode_GET_LENGTH(substring);
9878
Benjamin Petersonead6b532011-12-20 17:23:42 -06009879 switch (kind) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009880 case PyUnicode_1BYTE_KIND:
Victor Stinnerc3cec782011-10-05 21:24:08 +02009881 if (PyUnicode_IS_ASCII(self) && PyUnicode_IS_ASCII(substring))
9882 out = asciilib_split(
Victor Stinner7931d9a2011-11-04 00:22:48 +01009883 self, buf1, len1, buf2, len2, maxcount);
Victor Stinnerc3cec782011-10-05 21:24:08 +02009884 else
9885 out = ucs1lib_split(
Victor Stinner7931d9a2011-11-04 00:22:48 +01009886 self, buf1, len1, buf2, len2, maxcount);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009887 break;
9888 case PyUnicode_2BYTE_KIND:
9889 out = ucs2lib_split(
Victor Stinner7931d9a2011-11-04 00:22:48 +01009890 self, buf1, len1, buf2, len2, maxcount);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009891 break;
9892 case PyUnicode_4BYTE_KIND:
9893 out = ucs4lib_split(
Victor Stinner7931d9a2011-11-04 00:22:48 +01009894 self, buf1, len1, buf2, len2, maxcount);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009895 break;
9896 default:
9897 out = NULL;
9898 }
9899 if (kind1 != kind)
9900 PyMem_Free(buf1);
9901 if (kind2 != kind)
9902 PyMem_Free(buf2);
9903 return out;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009904}
9905
Alexander Belopolsky40018472011-02-26 01:02:56 +00009906static PyObject *
Victor Stinner9310abb2011-10-05 00:59:23 +02009907rsplit(PyObject *self,
9908 PyObject *substring,
Alexander Belopolsky40018472011-02-26 01:02:56 +00009909 Py_ssize_t maxcount)
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +00009910{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009911 int kind1, kind2, kind;
9912 void *buf1, *buf2;
9913 Py_ssize_t len1, len2;
9914 PyObject* out;
9915
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +00009916 if (maxcount < 0)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00009917 maxcount = PY_SSIZE_T_MAX;
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +00009918
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009919 if (PyUnicode_READY(self) == -1)
9920 return NULL;
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +00009921
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009922 if (substring == NULL)
Benjamin Petersonead6b532011-12-20 17:23:42 -06009923 switch (PyUnicode_KIND(self)) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009924 case PyUnicode_1BYTE_KIND:
Victor Stinnerc3cec782011-10-05 21:24:08 +02009925 if (PyUnicode_IS_ASCII(self))
9926 return asciilib_rsplit_whitespace(
Victor Stinner7931d9a2011-11-04 00:22:48 +01009927 self, PyUnicode_1BYTE_DATA(self),
Victor Stinnerc3cec782011-10-05 21:24:08 +02009928 PyUnicode_GET_LENGTH(self), maxcount
9929 );
9930 else
9931 return ucs1lib_rsplit_whitespace(
Victor Stinner7931d9a2011-11-04 00:22:48 +01009932 self, PyUnicode_1BYTE_DATA(self),
Victor Stinnerc3cec782011-10-05 21:24:08 +02009933 PyUnicode_GET_LENGTH(self), maxcount
9934 );
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009935 case PyUnicode_2BYTE_KIND:
9936 return ucs2lib_rsplit_whitespace(
Victor Stinner7931d9a2011-11-04 00:22:48 +01009937 self, PyUnicode_2BYTE_DATA(self),
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009938 PyUnicode_GET_LENGTH(self), maxcount
9939 );
9940 case PyUnicode_4BYTE_KIND:
9941 return ucs4lib_rsplit_whitespace(
Victor Stinner7931d9a2011-11-04 00:22:48 +01009942 self, PyUnicode_4BYTE_DATA(self),
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009943 PyUnicode_GET_LENGTH(self), maxcount
9944 );
9945 default:
9946 assert(0);
9947 return NULL;
9948 }
9949
9950 if (PyUnicode_READY(substring) == -1)
9951 return NULL;
9952
9953 kind1 = PyUnicode_KIND(self);
9954 kind2 = PyUnicode_KIND(substring);
9955 kind = kind1 > kind2 ? kind1 : kind2;
9956 buf1 = PyUnicode_DATA(self);
9957 buf2 = PyUnicode_DATA(substring);
9958 if (kind1 != kind)
Victor Stinner7931d9a2011-11-04 00:22:48 +01009959 buf1 = _PyUnicode_AsKind(self, kind);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009960 if (!buf1)
9961 return NULL;
9962 if (kind2 != kind)
Victor Stinner7931d9a2011-11-04 00:22:48 +01009963 buf2 = _PyUnicode_AsKind(substring, kind);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009964 if (!buf2) {
9965 if (kind1 != kind) PyMem_Free(buf1);
9966 return NULL;
9967 }
9968 len1 = PyUnicode_GET_LENGTH(self);
9969 len2 = PyUnicode_GET_LENGTH(substring);
9970
Benjamin Petersonead6b532011-12-20 17:23:42 -06009971 switch (kind) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009972 case PyUnicode_1BYTE_KIND:
Victor Stinnerc3cec782011-10-05 21:24:08 +02009973 if (PyUnicode_IS_ASCII(self) && PyUnicode_IS_ASCII(substring))
9974 out = asciilib_rsplit(
Victor Stinner7931d9a2011-11-04 00:22:48 +01009975 self, buf1, len1, buf2, len2, maxcount);
Victor Stinnerc3cec782011-10-05 21:24:08 +02009976 else
9977 out = ucs1lib_rsplit(
Victor Stinner7931d9a2011-11-04 00:22:48 +01009978 self, buf1, len1, buf2, len2, maxcount);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009979 break;
9980 case PyUnicode_2BYTE_KIND:
9981 out = ucs2lib_rsplit(
Victor Stinner7931d9a2011-11-04 00:22:48 +01009982 self, buf1, len1, buf2, len2, maxcount);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009983 break;
9984 case PyUnicode_4BYTE_KIND:
9985 out = ucs4lib_rsplit(
Victor Stinner7931d9a2011-11-04 00:22:48 +01009986 self, buf1, len1, buf2, len2, maxcount);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02009987 break;
9988 default:
9989 out = NULL;
9990 }
9991 if (kind1 != kind)
9992 PyMem_Free(buf1);
9993 if (kind2 != kind)
9994 PyMem_Free(buf2);
9995 return out;
9996}
9997
9998static Py_ssize_t
Victor Stinnerc3cec782011-10-05 21:24:08 +02009999anylib_find(int kind, PyObject *str1, void *buf1, Py_ssize_t len1,
10000 PyObject *str2, void *buf2, Py_ssize_t len2, Py_ssize_t offset)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010001{
Benjamin Petersonead6b532011-12-20 17:23:42 -060010002 switch (kind) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010003 case PyUnicode_1BYTE_KIND:
Victor Stinnerc3cec782011-10-05 21:24:08 +020010004 if (PyUnicode_IS_ASCII(str1) && PyUnicode_IS_ASCII(str2))
10005 return asciilib_find(buf1, len1, buf2, len2, offset);
10006 else
10007 return ucs1lib_find(buf1, len1, buf2, len2, offset);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010008 case PyUnicode_2BYTE_KIND:
10009 return ucs2lib_find(buf1, len1, buf2, len2, offset);
10010 case PyUnicode_4BYTE_KIND:
10011 return ucs4lib_find(buf1, len1, buf2, len2, offset);
10012 }
10013 assert(0);
10014 return -1;
10015}
10016
10017static Py_ssize_t
Victor Stinnerc3cec782011-10-05 21:24:08 +020010018anylib_count(int kind, PyObject *sstr, void* sbuf, Py_ssize_t slen,
10019 PyObject *str1, void *buf1, Py_ssize_t len1, Py_ssize_t maxcount)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010020{
Benjamin Petersonc0b95d12011-12-20 17:24:05 -060010021 switch (kind) {
10022 case PyUnicode_1BYTE_KIND:
10023 if (PyUnicode_IS_ASCII(sstr) && PyUnicode_IS_ASCII(str1))
10024 return asciilib_count(sbuf, slen, buf1, len1, maxcount);
10025 else
10026 return ucs1lib_count(sbuf, slen, buf1, len1, maxcount);
10027 case PyUnicode_2BYTE_KIND:
10028 return ucs2lib_count(sbuf, slen, buf1, len1, maxcount);
10029 case PyUnicode_4BYTE_KIND:
10030 return ucs4lib_count(sbuf, slen, buf1, len1, maxcount);
10031 }
10032 assert(0);
10033 return 0;
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +000010034}
10035
Alexander Belopolsky40018472011-02-26 01:02:56 +000010036static PyObject *
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010037replace(PyObject *self, PyObject *str1,
10038 PyObject *str2, Py_ssize_t maxcount)
Guido van Rossumd57fd912000-03-10 22:53:23 +000010039{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010040 PyObject *u;
10041 char *sbuf = PyUnicode_DATA(self);
10042 char *buf1 = PyUnicode_DATA(str1);
10043 char *buf2 = PyUnicode_DATA(str2);
10044 int srelease = 0, release1 = 0, release2 = 0;
10045 int skind = PyUnicode_KIND(self);
10046 int kind1 = PyUnicode_KIND(str1);
10047 int kind2 = PyUnicode_KIND(str2);
10048 Py_ssize_t slen = PyUnicode_GET_LENGTH(self);
10049 Py_ssize_t len1 = PyUnicode_GET_LENGTH(str1);
10050 Py_ssize_t len2 = PyUnicode_GET_LENGTH(str2);
Victor Stinner49a0a212011-10-12 23:46:10 +020010051 int mayshrink;
10052 Py_UCS4 maxchar, maxchar_str2;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010053
10054 if (maxcount < 0)
Benjamin Peterson29060642009-01-31 22:14:21 +000010055 maxcount = PY_SSIZE_T_MAX;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010056 else if (maxcount == 0 || slen == 0)
Antoine Pitrouf2c54842010-01-13 08:07:53 +000010057 goto nothing;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010058
Victor Stinner59de0ee2011-10-07 10:01:28 +020010059 if (str1 == str2)
10060 goto nothing;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010061 if (skind < kind1)
10062 /* substring too wide to be present */
10063 goto nothing;
10064
Victor Stinner49a0a212011-10-12 23:46:10 +020010065 maxchar = PyUnicode_MAX_CHAR_VALUE(self);
10066 maxchar_str2 = PyUnicode_MAX_CHAR_VALUE(str2);
10067 /* Replacing str1 with str2 may cause a maxchar reduction in the
10068 result string. */
10069 mayshrink = (maxchar_str2 < maxchar);
Victor Stinnere6abb482012-05-02 01:15:40 +020010070 maxchar = MAX_MAXCHAR(maxchar, maxchar_str2);
Victor Stinner49a0a212011-10-12 23:46:10 +020010071
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010072 if (len1 == len2) {
Thomas Wouters477c8d52006-05-27 19:21:47 +000010073 /* same length */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010074 if (len1 == 0)
Antoine Pitrouf2c54842010-01-13 08:07:53 +000010075 goto nothing;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010076 if (len1 == 1) {
Thomas Wouters477c8d52006-05-27 19:21:47 +000010077 /* replace characters */
Victor Stinner49a0a212011-10-12 23:46:10 +020010078 Py_UCS4 u1, u2;
10079 int rkind;
Victor Stinnerf6441102011-12-18 02:43:08 +010010080 Py_ssize_t index, pos;
10081 char *src;
10082
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010083 u1 = PyUnicode_READ_CHAR(str1, 0);
Victor Stinnerf6441102011-12-18 02:43:08 +010010084 pos = findchar(sbuf, PyUnicode_KIND(self), slen, u1, 1);
10085 if (pos < 0)
Thomas Wouters477c8d52006-05-27 19:21:47 +000010086 goto nothing;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010087 u2 = PyUnicode_READ_CHAR(str2, 0);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010088 u = PyUnicode_New(slen, maxchar);
Thomas Wouters477c8d52006-05-27 19:21:47 +000010089 if (!u)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010090 goto error;
Victor Stinnerd3f08822012-05-29 12:57:52 +020010091 _PyUnicode_FastCopyCharacters(u, 0, self, 0, slen);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010092 rkind = PyUnicode_KIND(u);
Victor Stinnerf6441102011-12-18 02:43:08 +010010093
10094 PyUnicode_WRITE(rkind, PyUnicode_DATA(u), pos, u2);
10095 index = 0;
10096 src = sbuf;
10097 while (--maxcount)
10098 {
10099 pos++;
10100 src += pos * PyUnicode_KIND(self);
10101 slen -= pos;
10102 index += pos;
10103 pos = findchar(src, PyUnicode_KIND(self), slen, u1, 1);
10104 if (pos < 0)
10105 break;
10106 PyUnicode_WRITE(rkind, PyUnicode_DATA(u), index + pos, u2);
10107 }
Victor Stinner49a0a212011-10-12 23:46:10 +020010108 }
10109 else {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010110 int rkind = skind;
10111 char *res;
Victor Stinnerf6441102011-12-18 02:43:08 +010010112 Py_ssize_t i;
Victor Stinner25a4b292011-10-06 12:31:55 +020010113
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010114 if (kind1 < rkind) {
10115 /* widen substring */
10116 buf1 = _PyUnicode_AsKind(str1, rkind);
10117 if (!buf1) goto error;
10118 release1 = 1;
10119 }
Victor Stinnerc3cec782011-10-05 21:24:08 +020010120 i = anylib_find(rkind, self, sbuf, slen, str1, buf1, len1, 0);
Thomas Wouters477c8d52006-05-27 19:21:47 +000010121 if (i < 0)
10122 goto nothing;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010123 if (rkind > kind2) {
10124 /* widen replacement */
10125 buf2 = _PyUnicode_AsKind(str2, rkind);
10126 if (!buf2) goto error;
10127 release2 = 1;
10128 }
10129 else if (rkind < kind2) {
10130 /* widen self and buf1 */
10131 rkind = kind2;
10132 if (release1) PyMem_Free(buf1);
10133 sbuf = _PyUnicode_AsKind(self, rkind);
10134 if (!sbuf) goto error;
10135 srelease = 1;
10136 buf1 = _PyUnicode_AsKind(str1, rkind);
10137 if (!buf1) goto error;
10138 release1 = 1;
10139 }
Victor Stinner49a0a212011-10-12 23:46:10 +020010140 u = PyUnicode_New(slen, maxchar);
10141 if (!u)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010142 goto error;
Victor Stinner49a0a212011-10-12 23:46:10 +020010143 assert(PyUnicode_KIND(u) == rkind);
10144 res = PyUnicode_DATA(u);
Victor Stinner25a4b292011-10-06 12:31:55 +020010145
Martin v. Löwisc47adb02011-10-07 20:55:35 +020010146 memcpy(res, sbuf, rkind * slen);
Antoine Pitrouf2c54842010-01-13 08:07:53 +000010147 /* change everything in-place, starting with this one */
Martin v. Löwisc47adb02011-10-07 20:55:35 +020010148 memcpy(res + rkind * i,
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010149 buf2,
Martin v. Löwisc47adb02011-10-07 20:55:35 +020010150 rkind * len2);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010151 i += len1;
Antoine Pitrouf2c54842010-01-13 08:07:53 +000010152
10153 while ( --maxcount > 0) {
Victor Stinnerc3cec782011-10-05 21:24:08 +020010154 i = anylib_find(rkind, self,
Martin v. Löwisc47adb02011-10-07 20:55:35 +020010155 sbuf+rkind*i, slen-i,
Victor Stinnerc3cec782011-10-05 21:24:08 +020010156 str1, buf1, len1, i);
Antoine Pitrouf2c54842010-01-13 08:07:53 +000010157 if (i == -1)
10158 break;
Martin v. Löwisc47adb02011-10-07 20:55:35 +020010159 memcpy(res + rkind * i,
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010160 buf2,
Martin v. Löwisc47adb02011-10-07 20:55:35 +020010161 rkind * len2);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010162 i += len1;
Antoine Pitrouf2c54842010-01-13 08:07:53 +000010163 }
Guido van Rossumd57fd912000-03-10 22:53:23 +000010164 }
Victor Stinner49a0a212011-10-12 23:46:10 +020010165 }
10166 else {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010167 Py_ssize_t n, i, j, ires;
10168 Py_ssize_t product, new_size;
10169 int rkind = skind;
10170 char *res;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010171
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010172 if (kind1 < rkind) {
Victor Stinner49a0a212011-10-12 23:46:10 +020010173 /* widen substring */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010174 buf1 = _PyUnicode_AsKind(str1, rkind);
10175 if (!buf1) goto error;
10176 release1 = 1;
10177 }
Victor Stinnerc3cec782011-10-05 21:24:08 +020010178 n = anylib_count(rkind, self, sbuf, slen, str1, buf1, len1, maxcount);
Thomas Wouters477c8d52006-05-27 19:21:47 +000010179 if (n == 0)
10180 goto nothing;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010181 if (kind2 < rkind) {
Victor Stinner49a0a212011-10-12 23:46:10 +020010182 /* widen replacement */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010183 buf2 = _PyUnicode_AsKind(str2, rkind);
10184 if (!buf2) goto error;
10185 release2 = 1;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010186 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010187 else if (kind2 > rkind) {
Victor Stinner49a0a212011-10-12 23:46:10 +020010188 /* widen self and buf1 */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010189 rkind = kind2;
10190 sbuf = _PyUnicode_AsKind(self, rkind);
10191 if (!sbuf) goto error;
10192 srelease = 1;
10193 if (release1) PyMem_Free(buf1);
10194 buf1 = _PyUnicode_AsKind(str1, rkind);
10195 if (!buf1) goto error;
10196 release1 = 1;
10197 }
10198 /* new_size = PyUnicode_GET_LENGTH(self) + n * (PyUnicode_GET_LENGTH(str2) -
10199 PyUnicode_GET_LENGTH(str1))); */
10200 product = n * (len2-len1);
10201 if ((product / (len2-len1)) != n) {
10202 PyErr_SetString(PyExc_OverflowError,
10203 "replace string is too long");
10204 goto error;
10205 }
10206 new_size = slen + product;
Victor Stinner49a0a212011-10-12 23:46:10 +020010207 if (new_size == 0) {
10208 Py_INCREF(unicode_empty);
10209 u = unicode_empty;
10210 goto done;
10211 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010212 if (new_size < 0 || new_size > (PY_SSIZE_T_MAX >> (rkind-1))) {
10213 PyErr_SetString(PyExc_OverflowError,
10214 "replace string is too long");
10215 goto error;
10216 }
Victor Stinner49a0a212011-10-12 23:46:10 +020010217 u = PyUnicode_New(new_size, maxchar);
10218 if (!u)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010219 goto error;
Victor Stinner49a0a212011-10-12 23:46:10 +020010220 assert(PyUnicode_KIND(u) == rkind);
10221 res = PyUnicode_DATA(u);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010222 ires = i = 0;
10223 if (len1 > 0) {
Thomas Wouters477c8d52006-05-27 19:21:47 +000010224 while (n-- > 0) {
10225 /* look for next match */
Victor Stinnerc3cec782011-10-05 21:24:08 +020010226 j = anylib_find(rkind, self,
Martin v. Löwisc47adb02011-10-07 20:55:35 +020010227 sbuf + rkind * i, slen-i,
Victor Stinnerc3cec782011-10-05 21:24:08 +020010228 str1, buf1, len1, i);
Antoine Pitrouf2c54842010-01-13 08:07:53 +000010229 if (j == -1)
10230 break;
10231 else if (j > i) {
Thomas Wouters477c8d52006-05-27 19:21:47 +000010232 /* copy unchanged part [i:j] */
Martin v. Löwisc47adb02011-10-07 20:55:35 +020010233 memcpy(res + rkind * ires,
10234 sbuf + rkind * i,
10235 rkind * (j-i));
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010236 ires += j - i;
Thomas Wouters477c8d52006-05-27 19:21:47 +000010237 }
10238 /* copy substitution string */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010239 if (len2 > 0) {
Martin v. Löwisc47adb02011-10-07 20:55:35 +020010240 memcpy(res + rkind * ires,
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010241 buf2,
Martin v. Löwisc47adb02011-10-07 20:55:35 +020010242 rkind * len2);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010243 ires += len2;
Thomas Wouters477c8d52006-05-27 19:21:47 +000010244 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010245 i = j + len1;
Thomas Wouters477c8d52006-05-27 19:21:47 +000010246 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010247 if (i < slen)
Thomas Wouters477c8d52006-05-27 19:21:47 +000010248 /* copy tail [i:] */
Martin v. Löwisc47adb02011-10-07 20:55:35 +020010249 memcpy(res + rkind * ires,
10250 sbuf + rkind * i,
10251 rkind * (slen-i));
Victor Stinner49a0a212011-10-12 23:46:10 +020010252 }
10253 else {
Thomas Wouters477c8d52006-05-27 19:21:47 +000010254 /* interleave */
10255 while (n > 0) {
Martin v. Löwisc47adb02011-10-07 20:55:35 +020010256 memcpy(res + rkind * ires,
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010257 buf2,
Martin v. Löwisc47adb02011-10-07 20:55:35 +020010258 rkind * len2);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010259 ires += len2;
Thomas Wouters477c8d52006-05-27 19:21:47 +000010260 if (--n <= 0)
10261 break;
Martin v. Löwisc47adb02011-10-07 20:55:35 +020010262 memcpy(res + rkind * ires,
10263 sbuf + rkind * i,
10264 rkind);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010265 ires++;
10266 i++;
Thomas Wouters477c8d52006-05-27 19:21:47 +000010267 }
Martin v. Löwisc47adb02011-10-07 20:55:35 +020010268 memcpy(res + rkind * ires,
10269 sbuf + rkind * i,
10270 rkind * (slen-i));
Thomas Wouters477c8d52006-05-27 19:21:47 +000010271 }
Victor Stinner49a0a212011-10-12 23:46:10 +020010272 }
10273
10274 if (mayshrink) {
Victor Stinner25a4b292011-10-06 12:31:55 +020010275 unicode_adjust_maxchar(&u);
10276 if (u == NULL)
10277 goto error;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010278 }
Victor Stinner49a0a212011-10-12 23:46:10 +020010279
10280 done:
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010281 if (srelease)
10282 PyMem_FREE(sbuf);
10283 if (release1)
10284 PyMem_FREE(buf1);
10285 if (release2)
10286 PyMem_FREE(buf2);
Victor Stinnerbb10a1f2011-10-05 01:34:17 +020010287 assert(_PyUnicode_CheckConsistency(u, 1));
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010288 return u;
Thomas Wouters477c8d52006-05-27 19:21:47 +000010289
Benjamin Peterson29060642009-01-31 22:14:21 +000010290 nothing:
Thomas Wouters477c8d52006-05-27 19:21:47 +000010291 /* nothing to replace; return original string (when possible) */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010292 if (srelease)
10293 PyMem_FREE(sbuf);
10294 if (release1)
10295 PyMem_FREE(buf1);
10296 if (release2)
10297 PyMem_FREE(buf2);
Victor Stinnerc4b49542011-12-11 22:44:26 +010010298 return unicode_result_unchanged(self);
10299
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010300 error:
10301 if (srelease && sbuf)
10302 PyMem_FREE(sbuf);
10303 if (release1 && buf1)
10304 PyMem_FREE(buf1);
10305 if (release2 && buf2)
10306 PyMem_FREE(buf2);
10307 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010308}
10309
10310/* --- Unicode Object Methods --------------------------------------------- */
10311
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000010312PyDoc_STRVAR(title__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000010313 "S.title() -> str\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000010314\n\
10315Return a titlecased version of S, i.e. words start with title case\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000010316characters, all remaining cased characters have lower case.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000010317
10318static PyObject*
Victor Stinner9310abb2011-10-05 00:59:23 +020010319unicode_title(PyObject *self)
Guido van Rossumd57fd912000-03-10 22:53:23 +000010320{
Benjamin Petersoneea48462012-01-16 14:28:50 -050010321 if (PyUnicode_READY(self) == -1)
10322 return NULL;
Victor Stinnerb0800dc2012-02-25 00:47:08 +010010323 return case_operation(self, do_title);
Guido van Rossumd57fd912000-03-10 22:53:23 +000010324}
10325
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000010326PyDoc_STRVAR(capitalize__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000010327 "S.capitalize() -> str\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000010328\n\
10329Return a capitalized version of S, i.e. make the first character\n\
Senthil Kumarane51ee8a2010-07-05 12:00:56 +000010330have upper case and the rest lower case.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000010331
10332static PyObject*
Victor Stinner9310abb2011-10-05 00:59:23 +020010333unicode_capitalize(PyObject *self)
Guido van Rossumd57fd912000-03-10 22:53:23 +000010334{
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -050010335 if (PyUnicode_READY(self) == -1)
10336 return NULL;
10337 if (PyUnicode_GET_LENGTH(self) == 0)
10338 return unicode_result_unchanged(self);
Victor Stinnerb0800dc2012-02-25 00:47:08 +010010339 return case_operation(self, do_capitalize);
Guido van Rossumd57fd912000-03-10 22:53:23 +000010340}
10341
Benjamin Petersond5890c82012-01-14 13:23:30 -050010342PyDoc_STRVAR(casefold__doc__,
10343 "S.casefold() -> str\n\
10344\n\
10345Return a version of S suitable for caseless comparisons.");
10346
10347static PyObject *
10348unicode_casefold(PyObject *self)
10349{
10350 if (PyUnicode_READY(self) == -1)
10351 return NULL;
10352 if (PyUnicode_IS_ASCII(self))
10353 return ascii_upper_or_lower(self, 1);
Victor Stinnerb0800dc2012-02-25 00:47:08 +010010354 return case_operation(self, do_casefold);
Benjamin Petersond5890c82012-01-14 13:23:30 -050010355}
10356
10357
Raymond Hettinger4f8f9762003-11-26 08:21:35 +000010358/* Argument converter. Coerces to a single unicode character */
10359
10360static int
10361convert_uc(PyObject *obj, void *addr)
10362{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010363 Py_UCS4 *fillcharloc = (Py_UCS4 *)addr;
Benjamin Peterson14339b62009-01-31 16:36:08 +000010364 PyObject *uniobj;
Raymond Hettinger4f8f9762003-11-26 08:21:35 +000010365
Benjamin Peterson14339b62009-01-31 16:36:08 +000010366 uniobj = PyUnicode_FromObject(obj);
10367 if (uniobj == NULL) {
10368 PyErr_SetString(PyExc_TypeError,
Benjamin Peterson29060642009-01-31 22:14:21 +000010369 "The fill character cannot be converted to Unicode");
Benjamin Peterson14339b62009-01-31 16:36:08 +000010370 return 0;
10371 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010372 if (PyUnicode_GET_LENGTH(uniobj) != 1) {
Benjamin Peterson14339b62009-01-31 16:36:08 +000010373 PyErr_SetString(PyExc_TypeError,
Benjamin Peterson29060642009-01-31 22:14:21 +000010374 "The fill character must be exactly one character long");
Benjamin Peterson14339b62009-01-31 16:36:08 +000010375 Py_DECREF(uniobj);
10376 return 0;
10377 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010378 *fillcharloc = PyUnicode_READ_CHAR(uniobj, 0);
Benjamin Peterson14339b62009-01-31 16:36:08 +000010379 Py_DECREF(uniobj);
10380 return 1;
Raymond Hettinger4f8f9762003-11-26 08:21:35 +000010381}
10382
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000010383PyDoc_STRVAR(center__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000010384 "S.center(width[, fillchar]) -> str\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000010385\n\
Benjamin Peterson142957c2008-07-04 19:55:29 +000010386Return S centered in a string of length width. Padding is\n\
Raymond Hettinger4f8f9762003-11-26 08:21:35 +000010387done using the specified fill character (default is a space)");
Guido van Rossumd57fd912000-03-10 22:53:23 +000010388
10389static PyObject *
Victor Stinner9310abb2011-10-05 00:59:23 +020010390unicode_center(PyObject *self, PyObject *args)
Guido van Rossumd57fd912000-03-10 22:53:23 +000010391{
Martin v. Löwis18e16552006-02-15 17:27:45 +000010392 Py_ssize_t marg, left;
10393 Py_ssize_t width;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010394 Py_UCS4 fillchar = ' ';
10395
Victor Stinnere9a29352011-10-01 02:14:59 +020010396 if (!PyArg_ParseTuple(args, "n|O&:center", &width, convert_uc, &fillchar))
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010397 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010398
Benjamin Petersonbac79492012-01-14 13:34:47 -050010399 if (PyUnicode_READY(self) == -1)
Guido van Rossumd57fd912000-03-10 22:53:23 +000010400 return NULL;
10401
Victor Stinnerc4b49542011-12-11 22:44:26 +010010402 if (PyUnicode_GET_LENGTH(self) >= width)
10403 return unicode_result_unchanged(self);
Guido van Rossumd57fd912000-03-10 22:53:23 +000010404
Victor Stinnerc4b49542011-12-11 22:44:26 +010010405 marg = width - PyUnicode_GET_LENGTH(self);
Guido van Rossumd57fd912000-03-10 22:53:23 +000010406 left = marg / 2 + (marg & width & 1);
10407
Victor Stinner9310abb2011-10-05 00:59:23 +020010408 return pad(self, left, marg - left, fillchar);
Guido van Rossumd57fd912000-03-10 22:53:23 +000010409}
10410
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010411/* This function assumes that str1 and str2 are readied by the caller. */
10412
Marc-André Lemburge5034372000-08-08 08:04:29 +000010413static int
Victor Stinner9db1a8b2011-10-23 20:04:37 +020010414unicode_compare(PyObject *str1, PyObject *str2)
Marc-André Lemburge5034372000-08-08 08:04:29 +000010415{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010416 int kind1, kind2;
10417 void *data1, *data2;
10418 Py_ssize_t len1, len2, i;
Marc-André Lemburge5034372000-08-08 08:04:29 +000010419
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010420 kind1 = PyUnicode_KIND(str1);
10421 kind2 = PyUnicode_KIND(str2);
10422 data1 = PyUnicode_DATA(str1);
10423 data2 = PyUnicode_DATA(str2);
10424 len1 = PyUnicode_GET_LENGTH(str1);
10425 len2 = PyUnicode_GET_LENGTH(str2);
Marc-André Lemburge5034372000-08-08 08:04:29 +000010426
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010427 for (i = 0; i < len1 && i < len2; ++i) {
10428 Py_UCS4 c1, c2;
10429 c1 = PyUnicode_READ(kind1, data1, i);
10430 c2 = PyUnicode_READ(kind2, data2, i);
Fredrik Lundh45714e92001-06-26 16:39:36 +000010431
10432 if (c1 != c2)
10433 return (c1 < c2) ? -1 : 1;
Marc-André Lemburge5034372000-08-08 08:04:29 +000010434 }
10435
10436 return (len1 < len2) ? -1 : (len1 != len2);
10437}
10438
Alexander Belopolsky40018472011-02-26 01:02:56 +000010439int
10440PyUnicode_Compare(PyObject *left, PyObject *right)
Guido van Rossumd57fd912000-03-10 22:53:23 +000010441{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010442 if (PyUnicode_Check(left) && PyUnicode_Check(right)) {
10443 if (PyUnicode_READY(left) == -1 ||
10444 PyUnicode_READY(right) == -1)
10445 return -1;
Victor Stinner9db1a8b2011-10-23 20:04:37 +020010446 return unicode_compare(left, right);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010447 }
Guido van Rossum09dc34f2007-05-04 04:17:33 +000010448 PyErr_Format(PyExc_TypeError,
10449 "Can't compare %.100s and %.100s",
10450 left->ob_type->tp_name,
10451 right->ob_type->tp_name);
Guido van Rossumd57fd912000-03-10 22:53:23 +000010452 return -1;
10453}
10454
Martin v. Löwis5b222132007-06-10 09:51:05 +000010455int
10456PyUnicode_CompareWithASCIIString(PyObject* uni, const char* str)
10457{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010458 Py_ssize_t i;
10459 int kind;
10460 void *data;
10461 Py_UCS4 chr;
10462
Victor Stinner910337b2011-10-03 03:20:16 +020010463 assert(_PyUnicode_CHECK(uni));
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010464 if (PyUnicode_READY(uni) == -1)
10465 return -1;
10466 kind = PyUnicode_KIND(uni);
10467 data = PyUnicode_DATA(uni);
Martin v. Löwis5b222132007-06-10 09:51:05 +000010468 /* Compare Unicode string and source character set string */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010469 for (i = 0; (chr = PyUnicode_READ(kind, data, i)) && str[i]; i++)
10470 if (chr != str[i])
10471 return (chr < (unsigned char)(str[i])) ? -1 : 1;
Benjamin Peterson8667a9b2010-01-09 21:45:28 +000010472 /* This check keeps Python strings that end in '\0' from comparing equal
10473 to C strings identical up to that point. */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010474 if (PyUnicode_GET_LENGTH(uni) != i || chr)
Benjamin Peterson29060642009-01-31 22:14:21 +000010475 return 1; /* uni is longer */
Martin v. Löwis5b222132007-06-10 09:51:05 +000010476 if (str[i])
Benjamin Peterson29060642009-01-31 22:14:21 +000010477 return -1; /* str is longer */
Martin v. Löwis5b222132007-06-10 09:51:05 +000010478 return 0;
10479}
10480
Antoine Pitrou51f3ef92008-12-20 13:14:23 +000010481
Benjamin Peterson29060642009-01-31 22:14:21 +000010482#define TEST_COND(cond) \
Benjamin Peterson14339b62009-01-31 16:36:08 +000010483 ((cond) ? Py_True : Py_False)
Antoine Pitrou51f3ef92008-12-20 13:14:23 +000010484
Alexander Belopolsky40018472011-02-26 01:02:56 +000010485PyObject *
10486PyUnicode_RichCompare(PyObject *left, PyObject *right, int op)
Thomas Wouters00ee7ba2006-08-21 19:07:27 +000010487{
10488 int result;
Benjamin Peterson14339b62009-01-31 16:36:08 +000010489
Antoine Pitrou51f3ef92008-12-20 13:14:23 +000010490 if (PyUnicode_Check(left) && PyUnicode_Check(right)) {
10491 PyObject *v;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010492 if (PyUnicode_READY(left) == -1 ||
10493 PyUnicode_READY(right) == -1)
10494 return NULL;
10495 if (PyUnicode_GET_LENGTH(left) != PyUnicode_GET_LENGTH(right) ||
10496 PyUnicode_KIND(left) != PyUnicode_KIND(right)) {
Antoine Pitrou51f3ef92008-12-20 13:14:23 +000010497 if (op == Py_EQ) {
10498 Py_INCREF(Py_False);
10499 return Py_False;
10500 }
10501 if (op == Py_NE) {
10502 Py_INCREF(Py_True);
10503 return Py_True;
10504 }
10505 }
10506 if (left == right)
10507 result = 0;
10508 else
Victor Stinner9db1a8b2011-10-23 20:04:37 +020010509 result = unicode_compare(left, right);
Benjamin Peterson14339b62009-01-31 16:36:08 +000010510
Antoine Pitrou51f3ef92008-12-20 13:14:23 +000010511 /* Convert the return value to a Boolean */
10512 switch (op) {
10513 case Py_EQ:
10514 v = TEST_COND(result == 0);
10515 break;
10516 case Py_NE:
10517 v = TEST_COND(result != 0);
10518 break;
10519 case Py_LE:
10520 v = TEST_COND(result <= 0);
10521 break;
10522 case Py_GE:
10523 v = TEST_COND(result >= 0);
10524 break;
10525 case Py_LT:
10526 v = TEST_COND(result == -1);
10527 break;
10528 case Py_GT:
10529 v = TEST_COND(result == 1);
10530 break;
10531 default:
10532 PyErr_BadArgument();
10533 return NULL;
10534 }
10535 Py_INCREF(v);
10536 return v;
Thomas Wouters00ee7ba2006-08-21 19:07:27 +000010537 }
Benjamin Peterson14339b62009-01-31 16:36:08 +000010538
Brian Curtindfc80e32011-08-10 20:28:54 -050010539 Py_RETURN_NOTIMPLEMENTED;
Thomas Wouters00ee7ba2006-08-21 19:07:27 +000010540}
10541
Alexander Belopolsky40018472011-02-26 01:02:56 +000010542int
10543PyUnicode_Contains(PyObject *container, PyObject *element)
Guido van Rossum403d68b2000-03-13 15:55:09 +000010544{
Thomas Wouters477c8d52006-05-27 19:21:47 +000010545 PyObject *str, *sub;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010546 int kind1, kind2, kind;
10547 void *buf1, *buf2;
10548 Py_ssize_t len1, len2;
Martin v. Löwis18e16552006-02-15 17:27:45 +000010549 int result;
Guido van Rossum403d68b2000-03-13 15:55:09 +000010550
10551 /* Coerce the two arguments */
Thomas Wouters477c8d52006-05-27 19:21:47 +000010552 sub = PyUnicode_FromObject(element);
10553 if (!sub) {
Benjamin Peterson29060642009-01-31 22:14:21 +000010554 PyErr_Format(PyExc_TypeError,
10555 "'in <string>' requires string as left operand, not %s",
10556 element->ob_type->tp_name);
Thomas Wouters477c8d52006-05-27 19:21:47 +000010557 return -1;
Guido van Rossum403d68b2000-03-13 15:55:09 +000010558 }
10559
Thomas Wouters477c8d52006-05-27 19:21:47 +000010560 str = PyUnicode_FromObject(container);
Benjamin Peterson22a29702012-01-02 09:00:30 -060010561 if (!str) {
Thomas Wouters477c8d52006-05-27 19:21:47 +000010562 Py_DECREF(sub);
10563 return -1;
10564 }
Benjamin Peterson22a29702012-01-02 09:00:30 -060010565 if (PyUnicode_READY(sub) == -1 || PyUnicode_READY(str) == -1) {
10566 Py_DECREF(sub);
10567 Py_DECREF(str);
10568 }
Thomas Wouters477c8d52006-05-27 19:21:47 +000010569
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010570 kind1 = PyUnicode_KIND(str);
10571 kind2 = PyUnicode_KIND(sub);
Benjamin Peterson1ff2e352012-05-11 17:41:20 -050010572 kind = kind1;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010573 buf1 = PyUnicode_DATA(str);
10574 buf2 = PyUnicode_DATA(sub);
Benjamin Peterson1ff2e352012-05-11 17:41:20 -050010575 if (kind2 != kind) {
Antoine Pitrou758153b2012-05-12 15:51:51 +020010576 if (kind2 > kind) {
10577 Py_DECREF(sub);
10578 Py_DECREF(str);
Benjamin Peterson1ff2e352012-05-11 17:41:20 -050010579 return 0;
Antoine Pitrou758153b2012-05-12 15:51:51 +020010580 }
Victor Stinner7931d9a2011-11-04 00:22:48 +010010581 buf2 = _PyUnicode_AsKind(sub, kind);
Benjamin Peterson1ff2e352012-05-11 17:41:20 -050010582 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010583 if (!buf2) {
10584 Py_DECREF(sub);
Benjamin Peterson1ff2e352012-05-11 17:41:20 -050010585 Py_DECREF(str);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010586 return -1;
10587 }
10588 len1 = PyUnicode_GET_LENGTH(str);
10589 len2 = PyUnicode_GET_LENGTH(sub);
10590
Benjamin Petersonead6b532011-12-20 17:23:42 -060010591 switch (kind) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010592 case PyUnicode_1BYTE_KIND:
10593 result = ucs1lib_find(buf1, len1, buf2, len2, 0) != -1;
10594 break;
10595 case PyUnicode_2BYTE_KIND:
10596 result = ucs2lib_find(buf1, len1, buf2, len2, 0) != -1;
10597 break;
10598 case PyUnicode_4BYTE_KIND:
10599 result = ucs4lib_find(buf1, len1, buf2, len2, 0) != -1;
10600 break;
10601 default:
10602 result = -1;
10603 assert(0);
10604 }
Thomas Wouters477c8d52006-05-27 19:21:47 +000010605
10606 Py_DECREF(str);
10607 Py_DECREF(sub);
10608
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010609 if (kind2 != kind)
10610 PyMem_Free(buf2);
10611
Guido van Rossum403d68b2000-03-13 15:55:09 +000010612 return result;
Guido van Rossum403d68b2000-03-13 15:55:09 +000010613}
10614
Guido van Rossumd57fd912000-03-10 22:53:23 +000010615/* Concat to string or Unicode object giving a new Unicode object. */
10616
Alexander Belopolsky40018472011-02-26 01:02:56 +000010617PyObject *
10618PyUnicode_Concat(PyObject *left, PyObject *right)
Guido van Rossumd57fd912000-03-10 22:53:23 +000010619{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010620 PyObject *u = NULL, *v = NULL, *w;
Victor Stinner127226b2011-10-13 01:12:34 +020010621 Py_UCS4 maxchar, maxchar2;
Victor Stinner488fa492011-12-12 00:01:39 +010010622 Py_ssize_t u_len, v_len, new_len;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010623
10624 /* Coerce the two arguments */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010625 u = PyUnicode_FromObject(left);
Guido van Rossumd57fd912000-03-10 22:53:23 +000010626 if (u == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +000010627 goto onError;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010628 v = PyUnicode_FromObject(right);
Guido van Rossumd57fd912000-03-10 22:53:23 +000010629 if (v == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +000010630 goto onError;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010631
10632 /* Shortcuts */
Victor Stinnera464fc12011-10-02 20:39:30 +020010633 if (v == unicode_empty) {
Benjamin Peterson29060642009-01-31 22:14:21 +000010634 Py_DECREF(v);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010635 return u;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010636 }
Victor Stinnera464fc12011-10-02 20:39:30 +020010637 if (u == unicode_empty) {
Benjamin Peterson29060642009-01-31 22:14:21 +000010638 Py_DECREF(u);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010639 return v;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010640 }
10641
Victor Stinner488fa492011-12-12 00:01:39 +010010642 u_len = PyUnicode_GET_LENGTH(u);
10643 v_len = PyUnicode_GET_LENGTH(v);
10644 if (u_len > PY_SSIZE_T_MAX - v_len) {
10645 PyErr_SetString(PyExc_OverflowError,
10646 "strings are too large to concat");
10647 goto onError;
10648 }
10649 new_len = u_len + v_len;
10650
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010651 maxchar = PyUnicode_MAX_CHAR_VALUE(u);
Victor Stinner127226b2011-10-13 01:12:34 +020010652 maxchar2 = PyUnicode_MAX_CHAR_VALUE(v);
Victor Stinnere6abb482012-05-02 01:15:40 +020010653 maxchar = MAX_MAXCHAR(maxchar, maxchar2);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010654
Guido van Rossumd57fd912000-03-10 22:53:23 +000010655 /* Concat the two Unicode strings */
Victor Stinner488fa492011-12-12 00:01:39 +010010656 w = PyUnicode_New(new_len, maxchar);
Guido van Rossumd57fd912000-03-10 22:53:23 +000010657 if (w == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +000010658 goto onError;
Victor Stinnerd3f08822012-05-29 12:57:52 +020010659 _PyUnicode_FastCopyCharacters(w, 0, u, 0, u_len);
10660 _PyUnicode_FastCopyCharacters(w, u_len, v, 0, v_len);
Guido van Rossumd57fd912000-03-10 22:53:23 +000010661 Py_DECREF(u);
10662 Py_DECREF(v);
Victor Stinnerbb10a1f2011-10-05 01:34:17 +020010663 assert(_PyUnicode_CheckConsistency(w, 1));
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010664 return w;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010665
Benjamin Peterson29060642009-01-31 22:14:21 +000010666 onError:
Guido van Rossumd57fd912000-03-10 22:53:23 +000010667 Py_XDECREF(u);
10668 Py_XDECREF(v);
10669 return NULL;
10670}
10671
Walter Dörwald1ab83302007-05-18 17:15:44 +000010672void
Victor Stinner23e56682011-10-03 03:54:37 +020010673PyUnicode_Append(PyObject **p_left, PyObject *right)
Walter Dörwald1ab83302007-05-18 17:15:44 +000010674{
Victor Stinner23e56682011-10-03 03:54:37 +020010675 PyObject *left, *res;
Victor Stinner488fa492011-12-12 00:01:39 +010010676 Py_UCS4 maxchar, maxchar2;
10677 Py_ssize_t left_len, right_len, new_len;
Victor Stinner23e56682011-10-03 03:54:37 +020010678
10679 if (p_left == NULL) {
10680 if (!PyErr_Occurred())
10681 PyErr_BadInternalCall();
Benjamin Peterson14339b62009-01-31 16:36:08 +000010682 return;
10683 }
Victor Stinner23e56682011-10-03 03:54:37 +020010684 left = *p_left;
10685 if (right == NULL || !PyUnicode_Check(left)) {
10686 if (!PyErr_Occurred())
10687 PyErr_BadInternalCall();
10688 goto error;
10689 }
10690
Benjamin Petersonbac79492012-01-14 13:34:47 -050010691 if (PyUnicode_READY(left) == -1)
Victor Stinnere1335c72011-10-04 20:53:03 +020010692 goto error;
Benjamin Petersonbac79492012-01-14 13:34:47 -050010693 if (PyUnicode_READY(right) == -1)
Victor Stinnere1335c72011-10-04 20:53:03 +020010694 goto error;
10695
Victor Stinner488fa492011-12-12 00:01:39 +010010696 /* Shortcuts */
10697 if (left == unicode_empty) {
10698 Py_DECREF(left);
10699 Py_INCREF(right);
10700 *p_left = right;
10701 return;
10702 }
10703 if (right == unicode_empty)
10704 return;
10705
10706 left_len = PyUnicode_GET_LENGTH(left);
10707 right_len = PyUnicode_GET_LENGTH(right);
10708 if (left_len > PY_SSIZE_T_MAX - right_len) {
10709 PyErr_SetString(PyExc_OverflowError,
10710 "strings are too large to concat");
10711 goto error;
10712 }
10713 new_len = left_len + right_len;
10714
10715 if (unicode_modifiable(left)
10716 && PyUnicode_CheckExact(right)
10717 && PyUnicode_KIND(right) <= PyUnicode_KIND(left)
Victor Stinnerb0923652011-10-04 01:17:31 +020010718 /* Don't resize for ascii += latin1. Convert ascii to latin1 requires
10719 to change the structure size, but characters are stored just after
Georg Brandl7597add2011-10-05 16:36:47 +020010720 the structure, and so it requires to move all characters which is
Victor Stinnerb0923652011-10-04 01:17:31 +020010721 not so different than duplicating the string. */
Victor Stinner488fa492011-12-12 00:01:39 +010010722 && !(PyUnicode_IS_ASCII(left) && !PyUnicode_IS_ASCII(right)))
10723 {
10724 /* append inplace */
10725 if (unicode_resize(p_left, new_len) != 0) {
10726 /* XXX if _PyUnicode_Resize() fails, 'left' has been
10727 * deallocated so it cannot be put back into
10728 * 'variable'. The MemoryError is raised when there
10729 * is no value in 'variable', which might (very
10730 * remotely) be a cause of incompatibilities.
10731 */
10732 goto error;
Victor Stinner23e56682011-10-03 03:54:37 +020010733 }
Victor Stinner488fa492011-12-12 00:01:39 +010010734 /* copy 'right' into the newly allocated area of 'left' */
Victor Stinnerd3f08822012-05-29 12:57:52 +020010735 _PyUnicode_FastCopyCharacters(*p_left, left_len, right, 0, right_len);
Victor Stinner23e56682011-10-03 03:54:37 +020010736 }
Victor Stinner488fa492011-12-12 00:01:39 +010010737 else {
10738 maxchar = PyUnicode_MAX_CHAR_VALUE(left);
10739 maxchar2 = PyUnicode_MAX_CHAR_VALUE(right);
Victor Stinnere6abb482012-05-02 01:15:40 +020010740 maxchar = MAX_MAXCHAR(maxchar, maxchar2);
Victor Stinner23e56682011-10-03 03:54:37 +020010741
Victor Stinner488fa492011-12-12 00:01:39 +010010742 /* Concat the two Unicode strings */
10743 res = PyUnicode_New(new_len, maxchar);
10744 if (res == NULL)
10745 goto error;
Victor Stinnerd3f08822012-05-29 12:57:52 +020010746 _PyUnicode_FastCopyCharacters(res, 0, left, 0, left_len);
10747 _PyUnicode_FastCopyCharacters(res, left_len, right, 0, right_len);
Victor Stinner488fa492011-12-12 00:01:39 +010010748 Py_DECREF(left);
10749 *p_left = res;
10750 }
10751 assert(_PyUnicode_CheckConsistency(*p_left, 1));
Victor Stinner23e56682011-10-03 03:54:37 +020010752 return;
10753
10754error:
Victor Stinner488fa492011-12-12 00:01:39 +010010755 Py_CLEAR(*p_left);
Walter Dörwald1ab83302007-05-18 17:15:44 +000010756}
10757
10758void
10759PyUnicode_AppendAndDel(PyObject **pleft, PyObject *right)
10760{
Benjamin Peterson14339b62009-01-31 16:36:08 +000010761 PyUnicode_Append(pleft, right);
10762 Py_XDECREF(right);
Walter Dörwald1ab83302007-05-18 17:15:44 +000010763}
10764
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000010765PyDoc_STRVAR(count__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000010766 "S.count(sub[, start[, end]]) -> int\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000010767\n\
Thomas Wouters477c8d52006-05-27 19:21:47 +000010768Return the number of non-overlapping occurrences of substring sub in\n\
Benjamin Peterson142957c2008-07-04 19:55:29 +000010769string S[start:end]. Optional arguments start and end are\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000010770interpreted as in slice notation.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000010771
10772static PyObject *
Victor Stinner9db1a8b2011-10-23 20:04:37 +020010773unicode_count(PyObject *self, PyObject *args)
Guido van Rossumd57fd912000-03-10 22:53:23 +000010774{
Victor Stinner9db1a8b2011-10-23 20:04:37 +020010775 PyObject *substring;
Martin v. Löwis18e16552006-02-15 17:27:45 +000010776 Py_ssize_t start = 0;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000010777 Py_ssize_t end = PY_SSIZE_T_MAX;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010778 PyObject *result;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010779 int kind1, kind2, kind;
10780 void *buf1, *buf2;
10781 Py_ssize_t len1, len2, iresult;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010782
Jesus Ceaac451502011-04-20 17:09:23 +020010783 if (!stringlib_parse_args_finds_unicode("count", args, &substring,
10784 &start, &end))
Benjamin Peterson29060642009-01-31 22:14:21 +000010785 return NULL;
Tim Petersced69f82003-09-16 20:30:58 +000010786
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010787 kind1 = PyUnicode_KIND(self);
10788 kind2 = PyUnicode_KIND(substring);
Benjamin Petersonb63f49f2012-05-03 18:31:07 -040010789 if (kind2 > kind1)
10790 return PyLong_FromLong(0);
10791 kind = kind1;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010792 buf1 = PyUnicode_DATA(self);
10793 buf2 = PyUnicode_DATA(substring);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010794 if (kind2 != kind)
Victor Stinner7931d9a2011-11-04 00:22:48 +010010795 buf2 = _PyUnicode_AsKind(substring, kind);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010796 if (!buf2) {
10797 Py_DECREF(substring);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010798 return NULL;
10799 }
10800 len1 = PyUnicode_GET_LENGTH(self);
10801 len2 = PyUnicode_GET_LENGTH(substring);
10802
10803 ADJUST_INDICES(start, end, len1);
Benjamin Petersonead6b532011-12-20 17:23:42 -060010804 switch (kind) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010805 case PyUnicode_1BYTE_KIND:
10806 iresult = ucs1lib_count(
10807 ((Py_UCS1*)buf1) + start, end - start,
10808 buf2, len2, PY_SSIZE_T_MAX
10809 );
10810 break;
10811 case PyUnicode_2BYTE_KIND:
10812 iresult = ucs2lib_count(
10813 ((Py_UCS2*)buf1) + start, end - start,
10814 buf2, len2, PY_SSIZE_T_MAX
10815 );
10816 break;
10817 case PyUnicode_4BYTE_KIND:
10818 iresult = ucs4lib_count(
10819 ((Py_UCS4*)buf1) + start, end - start,
10820 buf2, len2, PY_SSIZE_T_MAX
10821 );
10822 break;
10823 default:
10824 assert(0); iresult = 0;
10825 }
10826
10827 result = PyLong_FromSsize_t(iresult);
10828
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010829 if (kind2 != kind)
10830 PyMem_Free(buf2);
Guido van Rossumd57fd912000-03-10 22:53:23 +000010831
10832 Py_DECREF(substring);
Thomas Wouters477c8d52006-05-27 19:21:47 +000010833
Guido van Rossumd57fd912000-03-10 22:53:23 +000010834 return result;
10835}
10836
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000010837PyDoc_STRVAR(encode__doc__,
Victor Stinnerc911bbf2010-11-07 19:04:46 +000010838 "S.encode(encoding='utf-8', errors='strict') -> bytes\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000010839\n\
Victor Stinnere14e2122010-11-07 18:41:46 +000010840Encode S using the codec registered for encoding. Default encoding\n\
10841is 'utf-8'. errors may be given to set a different error\n\
Fred Drakee4315f52000-05-09 19:53:39 +000010842handling scheme. Default is 'strict' meaning that encoding errors raise\n\
Walter Dörwald3aeb6322002-09-02 13:14:32 +000010843a UnicodeEncodeError. Other possible values are 'ignore', 'replace' and\n\
10844'xmlcharrefreplace' as well as any other name registered with\n\
10845codecs.register_error that can handle UnicodeEncodeErrors.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000010846
10847static PyObject *
Victor Stinner9db1a8b2011-10-23 20:04:37 +020010848unicode_encode(PyObject *self, PyObject *args, PyObject *kwargs)
Guido van Rossumd57fd912000-03-10 22:53:23 +000010849{
Benjamin Peterson308d6372009-09-18 21:42:35 +000010850 static char *kwlist[] = {"encoding", "errors", 0};
Guido van Rossumd57fd912000-03-10 22:53:23 +000010851 char *encoding = NULL;
10852 char *errors = NULL;
Guido van Rossum35d94282007-08-27 18:20:11 +000010853
Benjamin Peterson308d6372009-09-18 21:42:35 +000010854 if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|ss:encode",
10855 kwlist, &encoding, &errors))
Guido van Rossumd57fd912000-03-10 22:53:23 +000010856 return NULL;
Victor Stinner9db1a8b2011-10-23 20:04:37 +020010857 return PyUnicode_AsEncodedString(self, encoding, errors);
Marc-André Lemburgd2d45982004-07-08 17:57:32 +000010858}
10859
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000010860PyDoc_STRVAR(expandtabs__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000010861 "S.expandtabs([tabsize]) -> str\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000010862\n\
10863Return a copy of S where all tab characters are expanded using spaces.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000010864If tabsize is not given, a tab size of 8 characters is assumed.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000010865
10866static PyObject*
Victor Stinner9db1a8b2011-10-23 20:04:37 +020010867unicode_expandtabs(PyObject *self, PyObject *args)
Guido van Rossumd57fd912000-03-10 22:53:23 +000010868{
Antoine Pitroue71d5742011-10-04 15:55:09 +020010869 Py_ssize_t i, j, line_pos, src_len, incr;
10870 Py_UCS4 ch;
10871 PyObject *u;
10872 void *src_data, *dest_data;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010873 int tabsize = 8;
Antoine Pitroue71d5742011-10-04 15:55:09 +020010874 int kind;
Antoine Pitroue19aa382011-10-04 16:04:01 +020010875 int found;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010876
10877 if (!PyArg_ParseTuple(args, "|i:expandtabs", &tabsize))
Benjamin Peterson29060642009-01-31 22:14:21 +000010878 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010879
Antoine Pitrou22425222011-10-04 19:10:51 +020010880 if (PyUnicode_READY(self) == -1)
10881 return NULL;
10882
Thomas Wouters7e474022000-07-16 12:04:32 +000010883 /* First pass: determine size of output string */
Antoine Pitroue71d5742011-10-04 15:55:09 +020010884 src_len = PyUnicode_GET_LENGTH(self);
10885 i = j = line_pos = 0;
10886 kind = PyUnicode_KIND(self);
10887 src_data = PyUnicode_DATA(self);
Antoine Pitroue19aa382011-10-04 16:04:01 +020010888 found = 0;
Antoine Pitroue71d5742011-10-04 15:55:09 +020010889 for (; i < src_len; i++) {
10890 ch = PyUnicode_READ(kind, src_data, i);
10891 if (ch == '\t') {
Antoine Pitroue19aa382011-10-04 16:04:01 +020010892 found = 1;
Benjamin Peterson29060642009-01-31 22:14:21 +000010893 if (tabsize > 0) {
Antoine Pitroue71d5742011-10-04 15:55:09 +020010894 incr = tabsize - (line_pos % tabsize); /* cannot overflow */
Benjamin Peterson29060642009-01-31 22:14:21 +000010895 if (j > PY_SSIZE_T_MAX - incr)
Antoine Pitroue71d5742011-10-04 15:55:09 +020010896 goto overflow;
10897 line_pos += incr;
Benjamin Peterson29060642009-01-31 22:14:21 +000010898 j += incr;
Christian Heimesdd15f6c2008-03-16 00:07:10 +000010899 }
Benjamin Peterson29060642009-01-31 22:14:21 +000010900 }
Guido van Rossumd57fd912000-03-10 22:53:23 +000010901 else {
Benjamin Peterson29060642009-01-31 22:14:21 +000010902 if (j > PY_SSIZE_T_MAX - 1)
Antoine Pitroue71d5742011-10-04 15:55:09 +020010903 goto overflow;
10904 line_pos++;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010905 j++;
Antoine Pitroue71d5742011-10-04 15:55:09 +020010906 if (ch == '\n' || ch == '\r')
10907 line_pos = 0;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010908 }
Antoine Pitroue71d5742011-10-04 15:55:09 +020010909 }
Victor Stinnerc4b49542011-12-11 22:44:26 +010010910 if (!found)
10911 return unicode_result_unchanged(self);
Guido van Rossumcd16bf62007-06-13 18:07:49 +000010912
Guido van Rossumd57fd912000-03-10 22:53:23 +000010913 /* Second pass: create output string and fill it */
Antoine Pitroue71d5742011-10-04 15:55:09 +020010914 u = PyUnicode_New(j, PyUnicode_MAX_CHAR_VALUE(self));
Guido van Rossumd57fd912000-03-10 22:53:23 +000010915 if (!u)
10916 return NULL;
Antoine Pitroue71d5742011-10-04 15:55:09 +020010917 dest_data = PyUnicode_DATA(u);
Guido van Rossumd57fd912000-03-10 22:53:23 +000010918
Antoine Pitroue71d5742011-10-04 15:55:09 +020010919 i = j = line_pos = 0;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010920
Antoine Pitroue71d5742011-10-04 15:55:09 +020010921 for (; i < src_len; i++) {
10922 ch = PyUnicode_READ(kind, src_data, i);
10923 if (ch == '\t') {
Benjamin Peterson29060642009-01-31 22:14:21 +000010924 if (tabsize > 0) {
Antoine Pitroue71d5742011-10-04 15:55:09 +020010925 incr = tabsize - (line_pos % tabsize);
10926 line_pos += incr;
Victor Stinnerda79e632012-02-22 13:37:04 +010010927 FILL(kind, dest_data, ' ', j, incr);
10928 j += incr;
Benjamin Peterson29060642009-01-31 22:14:21 +000010929 }
Benjamin Peterson14339b62009-01-31 16:36:08 +000010930 }
Benjamin Peterson29060642009-01-31 22:14:21 +000010931 else {
Antoine Pitroue71d5742011-10-04 15:55:09 +020010932 line_pos++;
10933 PyUnicode_WRITE(kind, dest_data, j, ch);
Christian Heimesdd15f6c2008-03-16 00:07:10 +000010934 j++;
Antoine Pitroue71d5742011-10-04 15:55:09 +020010935 if (ch == '\n' || ch == '\r')
10936 line_pos = 0;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010937 }
Antoine Pitroue71d5742011-10-04 15:55:09 +020010938 }
10939 assert (j == PyUnicode_GET_LENGTH(u));
Victor Stinnerd3df8ab2011-11-22 01:22:34 +010010940 return unicode_result(u);
Christian Heimesdd15f6c2008-03-16 00:07:10 +000010941
Antoine Pitroue71d5742011-10-04 15:55:09 +020010942 overflow:
Christian Heimesdd15f6c2008-03-16 00:07:10 +000010943 PyErr_SetString(PyExc_OverflowError, "new string is too long");
10944 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010945}
10946
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000010947PyDoc_STRVAR(find__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000010948 "S.find(sub[, start[, end]]) -> int\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000010949\n\
10950Return the lowest index in S where substring sub is found,\n\
Senthil Kumaran53516a82011-07-27 23:33:54 +080010951such that sub is contained within S[start:end]. Optional\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000010952arguments start and end are interpreted as in slice notation.\n\
10953\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000010954Return -1 on failure.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000010955
10956static PyObject *
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010957unicode_find(PyObject *self, PyObject *args)
Guido van Rossumd57fd912000-03-10 22:53:23 +000010958{
Victor Stinner9db1a8b2011-10-23 20:04:37 +020010959 PyObject *substring;
Christian Heimes9cd17752007-11-18 19:35:23 +000010960 Py_ssize_t start;
10961 Py_ssize_t end;
Thomas Wouters477c8d52006-05-27 19:21:47 +000010962 Py_ssize_t result;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010963
Jesus Ceaac451502011-04-20 17:09:23 +020010964 if (!stringlib_parse_args_finds_unicode("find", args, &substring,
10965 &start, &end))
Guido van Rossumd57fd912000-03-10 22:53:23 +000010966 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010967
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010968 if (PyUnicode_READY(self) == -1)
10969 return NULL;
10970 if (PyUnicode_READY(substring) == -1)
10971 return NULL;
10972
Victor Stinner7931d9a2011-11-04 00:22:48 +010010973 result = any_find_slice(1, self, substring, start, end);
Guido van Rossumd57fd912000-03-10 22:53:23 +000010974
10975 Py_DECREF(substring);
Thomas Wouters477c8d52006-05-27 19:21:47 +000010976
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010977 if (result == -2)
10978 return NULL;
10979
Christian Heimes217cfd12007-12-02 14:31:20 +000010980 return PyLong_FromSsize_t(result);
Guido van Rossumd57fd912000-03-10 22:53:23 +000010981}
10982
10983static PyObject *
Victor Stinner2fe5ced2011-10-02 00:25:40 +020010984unicode_getitem(PyObject *self, Py_ssize_t index)
Guido van Rossumd57fd912000-03-10 22:53:23 +000010985{
Victor Stinnerb6cd0142012-05-03 02:17:04 +020010986 void *data;
10987 enum PyUnicode_Kind kind;
10988 Py_UCS4 ch;
10989 PyObject *res;
10990
10991 if (!PyUnicode_Check(self) || PyUnicode_READY(self) == -1) {
10992 PyErr_BadArgument();
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020010993 return NULL;
Victor Stinnerb6cd0142012-05-03 02:17:04 +020010994 }
10995 if (index < 0 || index >= PyUnicode_GET_LENGTH(self)) {
10996 PyErr_SetString(PyExc_IndexError, "string index out of range");
10997 return NULL;
10998 }
10999 kind = PyUnicode_KIND(self);
11000 data = PyUnicode_DATA(self);
11001 ch = PyUnicode_READ(kind, data, index);
11002 if (ch < 256)
11003 return get_latin1_char(ch);
11004
11005 res = PyUnicode_New(1, ch);
11006 if (res == NULL)
11007 return NULL;
11008 kind = PyUnicode_KIND(res);
11009 data = PyUnicode_DATA(res);
11010 PyUnicode_WRITE(kind, data, 0, ch);
11011 assert(_PyUnicode_CheckConsistency(res, 1));
11012 return res;
Guido van Rossumd57fd912000-03-10 22:53:23 +000011013}
11014
Guido van Rossumc2504932007-09-18 19:42:40 +000011015/* Believe it or not, this produces the same value for ASCII strings
Mark Dickinson57e683e2011-09-24 18:18:40 +010011016 as bytes_hash(). */
Benjamin Peterson8f67d082010-10-17 20:54:53 +000011017static Py_hash_t
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011018unicode_hash(PyObject *self)
Guido van Rossumd57fd912000-03-10 22:53:23 +000011019{
Guido van Rossumc2504932007-09-18 19:42:40 +000011020 Py_ssize_t len;
Mark Dickinson57e683e2011-09-24 18:18:40 +010011021 Py_uhash_t x;
Guido van Rossumc2504932007-09-18 19:42:40 +000011022
Benjamin Petersonf6622c82012-04-09 14:53:07 -040011023#ifdef Py_DEBUG
Benjamin Peterson69e97272012-02-21 11:08:50 -050011024 assert(_Py_HashSecret_Initialized);
Benjamin Petersonf6622c82012-04-09 14:53:07 -040011025#endif
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011026 if (_PyUnicode_HASH(self) != -1)
11027 return _PyUnicode_HASH(self);
11028 if (PyUnicode_READY(self) == -1)
11029 return -1;
11030 len = PyUnicode_GET_LENGTH(self);
Georg Brandl16fa2a12012-02-21 00:50:13 +010011031 /*
11032 We make the hash of the empty string be 0, rather than using
11033 (prefix ^ suffix), since this slightly obfuscates the hash secret
11034 */
11035 if (len == 0) {
11036 _PyUnicode_HASH(self) = 0;
11037 return 0;
11038 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011039
11040 /* The hash function as a macro, gets expanded three times below. */
Georg Brandl2fb477c2012-02-21 00:33:36 +010011041#define HASH(P) \
11042 x ^= (Py_uhash_t) *P << 7; \
11043 while (--len >= 0) \
11044 x = (_PyHASH_MULTIPLIER * x) ^ (Py_uhash_t) *P++; \
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011045
Georg Brandl2fb477c2012-02-21 00:33:36 +010011046 x = (Py_uhash_t) _Py_HashSecret.prefix;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011047 switch (PyUnicode_KIND(self)) {
11048 case PyUnicode_1BYTE_KIND: {
11049 const unsigned char *c = PyUnicode_1BYTE_DATA(self);
11050 HASH(c);
11051 break;
11052 }
11053 case PyUnicode_2BYTE_KIND: {
11054 const Py_UCS2 *s = PyUnicode_2BYTE_DATA(self);
11055 HASH(s);
11056 break;
11057 }
11058 default: {
11059 Py_UCS4 *l;
11060 assert(PyUnicode_KIND(self) == PyUnicode_4BYTE_KIND &&
11061 "Impossible switch case in unicode_hash");
11062 l = PyUnicode_4BYTE_DATA(self);
11063 HASH(l);
11064 break;
11065 }
11066 }
Georg Brandl2fb477c2012-02-21 00:33:36 +010011067 x ^= (Py_uhash_t) PyUnicode_GET_LENGTH(self);
11068 x ^= (Py_uhash_t) _Py_HashSecret.suffix;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011069
Guido van Rossumc2504932007-09-18 19:42:40 +000011070 if (x == -1)
11071 x = -2;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011072 _PyUnicode_HASH(self) = x;
Guido van Rossumc2504932007-09-18 19:42:40 +000011073 return x;
Guido van Rossumd57fd912000-03-10 22:53:23 +000011074}
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011075#undef HASH
Guido van Rossumd57fd912000-03-10 22:53:23 +000011076
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011077PyDoc_STRVAR(index__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000011078 "S.index(sub[, start[, end]]) -> int\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000011079\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011080Like S.find() but raise ValueError when the substring is not found.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000011081
11082static PyObject *
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011083unicode_index(PyObject *self, PyObject *args)
Guido van Rossumd57fd912000-03-10 22:53:23 +000011084{
Martin v. Löwis18e16552006-02-15 17:27:45 +000011085 Py_ssize_t result;
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011086 PyObject *substring;
Christian Heimes9cd17752007-11-18 19:35:23 +000011087 Py_ssize_t start;
11088 Py_ssize_t end;
Guido van Rossumd57fd912000-03-10 22:53:23 +000011089
Jesus Ceaac451502011-04-20 17:09:23 +020011090 if (!stringlib_parse_args_finds_unicode("index", args, &substring,
11091 &start, &end))
Guido van Rossumd57fd912000-03-10 22:53:23 +000011092 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +000011093
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011094 if (PyUnicode_READY(self) == -1)
11095 return NULL;
11096 if (PyUnicode_READY(substring) == -1)
11097 return NULL;
11098
Victor Stinner7931d9a2011-11-04 00:22:48 +010011099 result = any_find_slice(1, self, substring, start, end);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011100
11101 Py_DECREF(substring);
Thomas Wouters477c8d52006-05-27 19:21:47 +000011102
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011103 if (result == -2)
11104 return NULL;
11105
Guido van Rossumd57fd912000-03-10 22:53:23 +000011106 if (result < 0) {
11107 PyErr_SetString(PyExc_ValueError, "substring not found");
11108 return NULL;
11109 }
Thomas Wouters477c8d52006-05-27 19:21:47 +000011110
Christian Heimes217cfd12007-12-02 14:31:20 +000011111 return PyLong_FromSsize_t(result);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011112}
11113
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011114PyDoc_STRVAR(islower__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000011115 "S.islower() -> bool\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000011116\n\
Guido van Rossum77f6a652002-04-03 22:41:51 +000011117Return True if all cased characters in S are lowercase and there is\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011118at least one cased character in S, False otherwise.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000011119
11120static PyObject*
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011121unicode_islower(PyObject *self)
Guido van Rossumd57fd912000-03-10 22:53:23 +000011122{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011123 Py_ssize_t i, length;
11124 int kind;
11125 void *data;
Guido van Rossumd57fd912000-03-10 22:53:23 +000011126 int cased;
11127
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011128 if (PyUnicode_READY(self) == -1)
11129 return NULL;
11130 length = PyUnicode_GET_LENGTH(self);
11131 kind = PyUnicode_KIND(self);
11132 data = PyUnicode_DATA(self);
11133
Guido van Rossumd57fd912000-03-10 22:53:23 +000011134 /* Shortcut for single character strings */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011135 if (length == 1)
11136 return PyBool_FromLong(
11137 Py_UNICODE_ISLOWER(PyUnicode_READ(kind, data, 0)));
Guido van Rossumd57fd912000-03-10 22:53:23 +000011138
Marc-André Lemburg60bc8092000-06-14 09:18:32 +000011139 /* Special case for empty strings */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011140 if (length == 0)
Benjamin Peterson29060642009-01-31 22:14:21 +000011141 return PyBool_FromLong(0);
Marc-André Lemburg60bc8092000-06-14 09:18:32 +000011142
Guido van Rossumd57fd912000-03-10 22:53:23 +000011143 cased = 0;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011144 for (i = 0; i < length; i++) {
11145 const Py_UCS4 ch = PyUnicode_READ(kind, data, i);
Tim Petersced69f82003-09-16 20:30:58 +000011146
Benjamin Peterson29060642009-01-31 22:14:21 +000011147 if (Py_UNICODE_ISUPPER(ch) || Py_UNICODE_ISTITLE(ch))
11148 return PyBool_FromLong(0);
11149 else if (!cased && Py_UNICODE_ISLOWER(ch))
11150 cased = 1;
Guido van Rossumd57fd912000-03-10 22:53:23 +000011151 }
Guido van Rossum77f6a652002-04-03 22:41:51 +000011152 return PyBool_FromLong(cased);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011153}
11154
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011155PyDoc_STRVAR(isupper__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000011156 "S.isupper() -> bool\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000011157\n\
Martin v. Löwis6828e182003-10-18 09:55:08 +000011158Return True if all cased characters in S are uppercase and there is\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011159at least one cased character in S, False otherwise.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000011160
11161static PyObject*
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011162unicode_isupper(PyObject *self)
Guido van Rossumd57fd912000-03-10 22:53:23 +000011163{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011164 Py_ssize_t i, length;
11165 int kind;
11166 void *data;
Guido van Rossumd57fd912000-03-10 22:53:23 +000011167 int cased;
11168
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011169 if (PyUnicode_READY(self) == -1)
11170 return NULL;
11171 length = PyUnicode_GET_LENGTH(self);
11172 kind = PyUnicode_KIND(self);
11173 data = PyUnicode_DATA(self);
11174
Guido van Rossumd57fd912000-03-10 22:53:23 +000011175 /* Shortcut for single character strings */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011176 if (length == 1)
11177 return PyBool_FromLong(
11178 Py_UNICODE_ISUPPER(PyUnicode_READ(kind, data, 0)) != 0);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011179
Marc-André Lemburg60bc8092000-06-14 09:18:32 +000011180 /* Special case for empty strings */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011181 if (length == 0)
Benjamin Peterson29060642009-01-31 22:14:21 +000011182 return PyBool_FromLong(0);
Marc-André Lemburg60bc8092000-06-14 09:18:32 +000011183
Guido van Rossumd57fd912000-03-10 22:53:23 +000011184 cased = 0;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011185 for (i = 0; i < length; i++) {
11186 const Py_UCS4 ch = PyUnicode_READ(kind, data, i);
Tim Petersced69f82003-09-16 20:30:58 +000011187
Benjamin Peterson29060642009-01-31 22:14:21 +000011188 if (Py_UNICODE_ISLOWER(ch) || Py_UNICODE_ISTITLE(ch))
11189 return PyBool_FromLong(0);
11190 else if (!cased && Py_UNICODE_ISUPPER(ch))
11191 cased = 1;
Guido van Rossumd57fd912000-03-10 22:53:23 +000011192 }
Guido van Rossum77f6a652002-04-03 22:41:51 +000011193 return PyBool_FromLong(cased);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011194}
11195
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011196PyDoc_STRVAR(istitle__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000011197 "S.istitle() -> bool\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000011198\n\
Martin v. Löwis6828e182003-10-18 09:55:08 +000011199Return True if S is a titlecased string and there is at least one\n\
11200character in S, i.e. upper- and titlecase characters may only\n\
11201follow uncased characters and lowercase characters only cased ones.\n\
11202Return False otherwise.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000011203
11204static PyObject*
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011205unicode_istitle(PyObject *self)
Guido van Rossumd57fd912000-03-10 22:53:23 +000011206{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011207 Py_ssize_t i, length;
11208 int kind;
11209 void *data;
Guido van Rossumd57fd912000-03-10 22:53:23 +000011210 int cased, previous_is_cased;
11211
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011212 if (PyUnicode_READY(self) == -1)
11213 return NULL;
11214 length = PyUnicode_GET_LENGTH(self);
11215 kind = PyUnicode_KIND(self);
11216 data = PyUnicode_DATA(self);
11217
Guido van Rossumd57fd912000-03-10 22:53:23 +000011218 /* Shortcut for single character strings */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011219 if (length == 1) {
11220 Py_UCS4 ch = PyUnicode_READ(kind, data, 0);
11221 return PyBool_FromLong((Py_UNICODE_ISTITLE(ch) != 0) ||
11222 (Py_UNICODE_ISUPPER(ch) != 0));
11223 }
Guido van Rossumd57fd912000-03-10 22:53:23 +000011224
Marc-André Lemburg60bc8092000-06-14 09:18:32 +000011225 /* Special case for empty strings */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011226 if (length == 0)
Benjamin Peterson29060642009-01-31 22:14:21 +000011227 return PyBool_FromLong(0);
Marc-André Lemburg60bc8092000-06-14 09:18:32 +000011228
Guido van Rossumd57fd912000-03-10 22:53:23 +000011229 cased = 0;
11230 previous_is_cased = 0;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011231 for (i = 0; i < length; i++) {
11232 const Py_UCS4 ch = PyUnicode_READ(kind, data, i);
Tim Petersced69f82003-09-16 20:30:58 +000011233
Benjamin Peterson29060642009-01-31 22:14:21 +000011234 if (Py_UNICODE_ISUPPER(ch) || Py_UNICODE_ISTITLE(ch)) {
11235 if (previous_is_cased)
11236 return PyBool_FromLong(0);
11237 previous_is_cased = 1;
11238 cased = 1;
11239 }
11240 else if (Py_UNICODE_ISLOWER(ch)) {
11241 if (!previous_is_cased)
11242 return PyBool_FromLong(0);
11243 previous_is_cased = 1;
11244 cased = 1;
11245 }
11246 else
11247 previous_is_cased = 0;
Guido van Rossumd57fd912000-03-10 22:53:23 +000011248 }
Guido van Rossum77f6a652002-04-03 22:41:51 +000011249 return PyBool_FromLong(cased);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011250}
11251
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011252PyDoc_STRVAR(isspace__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000011253 "S.isspace() -> bool\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000011254\n\
Martin v. Löwis6828e182003-10-18 09:55:08 +000011255Return True if all characters in S are whitespace\n\
11256and there is at least one character in S, False otherwise.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000011257
11258static PyObject*
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011259unicode_isspace(PyObject *self)
Guido van Rossumd57fd912000-03-10 22:53:23 +000011260{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011261 Py_ssize_t i, length;
11262 int kind;
11263 void *data;
11264
11265 if (PyUnicode_READY(self) == -1)
11266 return NULL;
11267 length = PyUnicode_GET_LENGTH(self);
11268 kind = PyUnicode_KIND(self);
11269 data = PyUnicode_DATA(self);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011270
Guido van Rossumd57fd912000-03-10 22:53:23 +000011271 /* Shortcut for single character strings */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011272 if (length == 1)
11273 return PyBool_FromLong(
11274 Py_UNICODE_ISSPACE(PyUnicode_READ(kind, data, 0)));
Guido van Rossumd57fd912000-03-10 22:53:23 +000011275
Marc-André Lemburg60bc8092000-06-14 09:18:32 +000011276 /* Special case for empty strings */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011277 if (length == 0)
Benjamin Peterson29060642009-01-31 22:14:21 +000011278 return PyBool_FromLong(0);
Marc-André Lemburg60bc8092000-06-14 09:18:32 +000011279
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011280 for (i = 0; i < length; i++) {
11281 const Py_UCS4 ch = PyUnicode_READ(kind, data, i);
Ezio Melotti93e7afc2011-08-22 14:08:38 +030011282 if (!Py_UNICODE_ISSPACE(ch))
Benjamin Peterson29060642009-01-31 22:14:21 +000011283 return PyBool_FromLong(0);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011284 }
Guido van Rossum77f6a652002-04-03 22:41:51 +000011285 return PyBool_FromLong(1);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011286}
11287
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011288PyDoc_STRVAR(isalpha__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000011289 "S.isalpha() -> bool\n\
Marc-André Lemburga7acf422000-07-05 09:49:44 +000011290\n\
Martin v. Löwis6828e182003-10-18 09:55:08 +000011291Return True if all characters in S are alphabetic\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011292and there is at least one character in S, False otherwise.");
Marc-André Lemburga7acf422000-07-05 09:49:44 +000011293
11294static PyObject*
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011295unicode_isalpha(PyObject *self)
Marc-André Lemburga7acf422000-07-05 09:49:44 +000011296{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011297 Py_ssize_t i, length;
11298 int kind;
11299 void *data;
11300
11301 if (PyUnicode_READY(self) == -1)
11302 return NULL;
11303 length = PyUnicode_GET_LENGTH(self);
11304 kind = PyUnicode_KIND(self);
11305 data = PyUnicode_DATA(self);
Marc-André Lemburga7acf422000-07-05 09:49:44 +000011306
Marc-André Lemburga7acf422000-07-05 09:49:44 +000011307 /* Shortcut for single character strings */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011308 if (length == 1)
11309 return PyBool_FromLong(
11310 Py_UNICODE_ISALPHA(PyUnicode_READ(kind, data, 0)));
Marc-André Lemburga7acf422000-07-05 09:49:44 +000011311
11312 /* Special case for empty strings */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011313 if (length == 0)
Benjamin Peterson29060642009-01-31 22:14:21 +000011314 return PyBool_FromLong(0);
Marc-André Lemburga7acf422000-07-05 09:49:44 +000011315
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011316 for (i = 0; i < length; i++) {
11317 if (!Py_UNICODE_ISALPHA(PyUnicode_READ(kind, data, i)))
Benjamin Peterson29060642009-01-31 22:14:21 +000011318 return PyBool_FromLong(0);
Marc-André Lemburga7acf422000-07-05 09:49:44 +000011319 }
Guido van Rossum77f6a652002-04-03 22:41:51 +000011320 return PyBool_FromLong(1);
Marc-André Lemburga7acf422000-07-05 09:49:44 +000011321}
11322
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011323PyDoc_STRVAR(isalnum__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000011324 "S.isalnum() -> bool\n\
Marc-André Lemburga7acf422000-07-05 09:49:44 +000011325\n\
Martin v. Löwis6828e182003-10-18 09:55:08 +000011326Return True if all characters in S are alphanumeric\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011327and there is at least one character in S, False otherwise.");
Marc-André Lemburga7acf422000-07-05 09:49:44 +000011328
11329static PyObject*
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011330unicode_isalnum(PyObject *self)
Marc-André Lemburga7acf422000-07-05 09:49:44 +000011331{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011332 int kind;
11333 void *data;
11334 Py_ssize_t len, i;
11335
11336 if (PyUnicode_READY(self) == -1)
11337 return NULL;
11338
11339 kind = PyUnicode_KIND(self);
11340 data = PyUnicode_DATA(self);
11341 len = PyUnicode_GET_LENGTH(self);
Marc-André Lemburga7acf422000-07-05 09:49:44 +000011342
Marc-André Lemburga7acf422000-07-05 09:49:44 +000011343 /* Shortcut for single character strings */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011344 if (len == 1) {
11345 const Py_UCS4 ch = PyUnicode_READ(kind, data, 0);
11346 return PyBool_FromLong(Py_UNICODE_ISALNUM(ch));
11347 }
Marc-André Lemburga7acf422000-07-05 09:49:44 +000011348
11349 /* Special case for empty strings */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011350 if (len == 0)
Benjamin Peterson29060642009-01-31 22:14:21 +000011351 return PyBool_FromLong(0);
Marc-André Lemburga7acf422000-07-05 09:49:44 +000011352
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011353 for (i = 0; i < len; i++) {
11354 const Py_UCS4 ch = PyUnicode_READ(kind, data, i);
Ezio Melotti93e7afc2011-08-22 14:08:38 +030011355 if (!Py_UNICODE_ISALNUM(ch))
Benjamin Peterson29060642009-01-31 22:14:21 +000011356 return PyBool_FromLong(0);
Marc-André Lemburga7acf422000-07-05 09:49:44 +000011357 }
Guido van Rossum77f6a652002-04-03 22:41:51 +000011358 return PyBool_FromLong(1);
Marc-André Lemburga7acf422000-07-05 09:49:44 +000011359}
11360
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011361PyDoc_STRVAR(isdecimal__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000011362 "S.isdecimal() -> bool\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000011363\n\
Guido van Rossum77f6a652002-04-03 22:41:51 +000011364Return True if there are only decimal characters in S,\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011365False otherwise.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000011366
11367static PyObject*
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011368unicode_isdecimal(PyObject *self)
Guido van Rossumd57fd912000-03-10 22:53:23 +000011369{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011370 Py_ssize_t i, length;
11371 int kind;
11372 void *data;
11373
11374 if (PyUnicode_READY(self) == -1)
11375 return NULL;
11376 length = PyUnicode_GET_LENGTH(self);
11377 kind = PyUnicode_KIND(self);
11378 data = PyUnicode_DATA(self);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011379
Guido van Rossumd57fd912000-03-10 22:53:23 +000011380 /* Shortcut for single character strings */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011381 if (length == 1)
11382 return PyBool_FromLong(
11383 Py_UNICODE_ISDECIMAL(PyUnicode_READ(kind, data, 0)));
Guido van Rossumd57fd912000-03-10 22:53:23 +000011384
Marc-André Lemburg60bc8092000-06-14 09:18:32 +000011385 /* Special case for empty strings */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011386 if (length == 0)
Benjamin Peterson29060642009-01-31 22:14:21 +000011387 return PyBool_FromLong(0);
Marc-André Lemburg60bc8092000-06-14 09:18:32 +000011388
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011389 for (i = 0; i < length; i++) {
11390 if (!Py_UNICODE_ISDECIMAL(PyUnicode_READ(kind, data, i)))
Benjamin Peterson29060642009-01-31 22:14:21 +000011391 return PyBool_FromLong(0);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011392 }
Guido van Rossum77f6a652002-04-03 22:41:51 +000011393 return PyBool_FromLong(1);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011394}
11395
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011396PyDoc_STRVAR(isdigit__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000011397 "S.isdigit() -> bool\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000011398\n\
Martin v. Löwis6828e182003-10-18 09:55:08 +000011399Return True if all characters in S are digits\n\
11400and there is at least one character in S, False otherwise.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000011401
11402static PyObject*
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011403unicode_isdigit(PyObject *self)
Guido van Rossumd57fd912000-03-10 22:53:23 +000011404{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011405 Py_ssize_t i, length;
11406 int kind;
11407 void *data;
11408
11409 if (PyUnicode_READY(self) == -1)
11410 return NULL;
11411 length = PyUnicode_GET_LENGTH(self);
11412 kind = PyUnicode_KIND(self);
11413 data = PyUnicode_DATA(self);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011414
Guido van Rossumd57fd912000-03-10 22:53:23 +000011415 /* Shortcut for single character strings */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011416 if (length == 1) {
11417 const Py_UCS4 ch = PyUnicode_READ(kind, data, 0);
11418 return PyBool_FromLong(Py_UNICODE_ISDIGIT(ch));
11419 }
Guido van Rossumd57fd912000-03-10 22:53:23 +000011420
Marc-André Lemburg60bc8092000-06-14 09:18:32 +000011421 /* Special case for empty strings */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011422 if (length == 0)
Benjamin Peterson29060642009-01-31 22:14:21 +000011423 return PyBool_FromLong(0);
Marc-André Lemburg60bc8092000-06-14 09:18:32 +000011424
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011425 for (i = 0; i < length; i++) {
11426 if (!Py_UNICODE_ISDIGIT(PyUnicode_READ(kind, data, i)))
Benjamin Peterson29060642009-01-31 22:14:21 +000011427 return PyBool_FromLong(0);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011428 }
Guido van Rossum77f6a652002-04-03 22:41:51 +000011429 return PyBool_FromLong(1);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011430}
11431
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011432PyDoc_STRVAR(isnumeric__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000011433 "S.isnumeric() -> bool\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000011434\n\
Guido van Rossum77f6a652002-04-03 22:41:51 +000011435Return True if there are only numeric characters in S,\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011436False otherwise.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000011437
11438static PyObject*
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011439unicode_isnumeric(PyObject *self)
Guido van Rossumd57fd912000-03-10 22:53:23 +000011440{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011441 Py_ssize_t i, length;
11442 int kind;
11443 void *data;
11444
11445 if (PyUnicode_READY(self) == -1)
11446 return NULL;
11447 length = PyUnicode_GET_LENGTH(self);
11448 kind = PyUnicode_KIND(self);
11449 data = PyUnicode_DATA(self);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011450
Guido van Rossumd57fd912000-03-10 22:53:23 +000011451 /* Shortcut for single character strings */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011452 if (length == 1)
11453 return PyBool_FromLong(
11454 Py_UNICODE_ISNUMERIC(PyUnicode_READ(kind, data, 0)));
Guido van Rossumd57fd912000-03-10 22:53:23 +000011455
Marc-André Lemburg60bc8092000-06-14 09:18:32 +000011456 /* Special case for empty strings */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011457 if (length == 0)
Benjamin Peterson29060642009-01-31 22:14:21 +000011458 return PyBool_FromLong(0);
Marc-André Lemburg60bc8092000-06-14 09:18:32 +000011459
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011460 for (i = 0; i < length; i++) {
11461 if (!Py_UNICODE_ISNUMERIC(PyUnicode_READ(kind, data, i)))
Benjamin Peterson29060642009-01-31 22:14:21 +000011462 return PyBool_FromLong(0);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011463 }
Guido van Rossum77f6a652002-04-03 22:41:51 +000011464 return PyBool_FromLong(1);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011465}
11466
Martin v. Löwis47383402007-08-15 07:32:56 +000011467int
11468PyUnicode_IsIdentifier(PyObject *self)
11469{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011470 int kind;
11471 void *data;
11472 Py_ssize_t i;
Ezio Melotti93e7afc2011-08-22 14:08:38 +030011473 Py_UCS4 first;
Martin v. Löwis47383402007-08-15 07:32:56 +000011474
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011475 if (PyUnicode_READY(self) == -1) {
11476 Py_FatalError("identifier not ready");
Benjamin Peterson29060642009-01-31 22:14:21 +000011477 return 0;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011478 }
11479
11480 /* Special case for empty strings */
11481 if (PyUnicode_GET_LENGTH(self) == 0)
11482 return 0;
11483 kind = PyUnicode_KIND(self);
11484 data = PyUnicode_DATA(self);
Martin v. Löwis47383402007-08-15 07:32:56 +000011485
11486 /* PEP 3131 says that the first character must be in
11487 XID_Start and subsequent characters in XID_Continue,
11488 and for the ASCII range, the 2.x rules apply (i.e
Benjamin Peterson14339b62009-01-31 16:36:08 +000011489 start with letters and underscore, continue with
Martin v. Löwis47383402007-08-15 07:32:56 +000011490 letters, digits, underscore). However, given the current
11491 definition of XID_Start and XID_Continue, it is sufficient
11492 to check just for these, except that _ must be allowed
11493 as starting an identifier. */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011494 first = PyUnicode_READ(kind, data, 0);
Benjamin Petersonf413b802011-08-12 22:17:18 -050011495 if (!_PyUnicode_IsXidStart(first) && first != 0x5F /* LOW LINE */)
Martin v. Löwis47383402007-08-15 07:32:56 +000011496 return 0;
11497
Benjamin Peterson9c6e6a02011-09-28 08:09:05 -040011498 for (i = 1; i < PyUnicode_GET_LENGTH(self); i++)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011499 if (!_PyUnicode_IsXidContinue(PyUnicode_READ(kind, data, i)))
Benjamin Peterson29060642009-01-31 22:14:21 +000011500 return 0;
Martin v. Löwis47383402007-08-15 07:32:56 +000011501 return 1;
11502}
11503
11504PyDoc_STRVAR(isidentifier__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000011505 "S.isidentifier() -> bool\n\
Martin v. Löwis47383402007-08-15 07:32:56 +000011506\n\
11507Return True if S is a valid identifier according\n\
11508to the language definition.");
11509
11510static PyObject*
11511unicode_isidentifier(PyObject *self)
11512{
11513 return PyBool_FromLong(PyUnicode_IsIdentifier(self));
11514}
11515
Georg Brandl559e5d72008-06-11 18:37:52 +000011516PyDoc_STRVAR(isprintable__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000011517 "S.isprintable() -> bool\n\
Georg Brandl559e5d72008-06-11 18:37:52 +000011518\n\
11519Return True if all characters in S are considered\n\
11520printable in repr() or S is empty, False otherwise.");
11521
11522static PyObject*
11523unicode_isprintable(PyObject *self)
11524{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011525 Py_ssize_t i, length;
11526 int kind;
11527 void *data;
11528
11529 if (PyUnicode_READY(self) == -1)
11530 return NULL;
11531 length = PyUnicode_GET_LENGTH(self);
11532 kind = PyUnicode_KIND(self);
11533 data = PyUnicode_DATA(self);
Georg Brandl559e5d72008-06-11 18:37:52 +000011534
11535 /* Shortcut for single character strings */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011536 if (length == 1)
11537 return PyBool_FromLong(
11538 Py_UNICODE_ISPRINTABLE(PyUnicode_READ(kind, data, 0)));
Georg Brandl559e5d72008-06-11 18:37:52 +000011539
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011540 for (i = 0; i < length; i++) {
11541 if (!Py_UNICODE_ISPRINTABLE(PyUnicode_READ(kind, data, i))) {
Georg Brandl559e5d72008-06-11 18:37:52 +000011542 Py_RETURN_FALSE;
11543 }
11544 }
11545 Py_RETURN_TRUE;
11546}
11547
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011548PyDoc_STRVAR(join__doc__,
Georg Brandl495f7b52009-10-27 15:28:25 +000011549 "S.join(iterable) -> str\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000011550\n\
11551Return a string which is the concatenation of the strings in the\n\
Georg Brandl495f7b52009-10-27 15:28:25 +000011552iterable. The separator between elements is S.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000011553
11554static PyObject*
Martin v. Löwise3eb1f22001-08-16 13:15:00 +000011555unicode_join(PyObject *self, PyObject *data)
Guido van Rossumd57fd912000-03-10 22:53:23 +000011556{
Martin v. Löwise3eb1f22001-08-16 13:15:00 +000011557 return PyUnicode_Join(self, data);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011558}
11559
Martin v. Löwis18e16552006-02-15 17:27:45 +000011560static Py_ssize_t
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011561unicode_length(PyObject *self)
Guido van Rossumd57fd912000-03-10 22:53:23 +000011562{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011563 if (PyUnicode_READY(self) == -1)
11564 return -1;
11565 return PyUnicode_GET_LENGTH(self);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011566}
11567
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011568PyDoc_STRVAR(ljust__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000011569 "S.ljust(width[, fillchar]) -> str\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000011570\n\
Benjamin Petersonf10a79a2008-10-11 00:49:57 +000011571Return S left-justified in a Unicode string of length width. Padding is\n\
Raymond Hettinger4f8f9762003-11-26 08:21:35 +000011572done using the specified fill character (default is a space).");
Guido van Rossumd57fd912000-03-10 22:53:23 +000011573
11574static PyObject *
Victor Stinner9310abb2011-10-05 00:59:23 +020011575unicode_ljust(PyObject *self, PyObject *args)
Guido van Rossumd57fd912000-03-10 22:53:23 +000011576{
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000011577 Py_ssize_t width;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011578 Py_UCS4 fillchar = ' ';
11579
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000011580 if (!PyArg_ParseTuple(args, "n|O&:ljust", &width, convert_uc, &fillchar))
Guido van Rossumd57fd912000-03-10 22:53:23 +000011581 return NULL;
11582
Benjamin Petersonbac79492012-01-14 13:34:47 -050011583 if (PyUnicode_READY(self) == -1)
Victor Stinnerc4b49542011-12-11 22:44:26 +010011584 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +000011585
Victor Stinnerc4b49542011-12-11 22:44:26 +010011586 if (PyUnicode_GET_LENGTH(self) >= width)
11587 return unicode_result_unchanged(self);
11588
11589 return pad(self, 0, width - PyUnicode_GET_LENGTH(self), fillchar);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011590}
11591
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011592PyDoc_STRVAR(lower__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000011593 "S.lower() -> str\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000011594\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011595Return a copy of the string S converted to lowercase.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000011596
11597static PyObject*
Victor Stinner9310abb2011-10-05 00:59:23 +020011598unicode_lower(PyObject *self)
Guido van Rossumd57fd912000-03-10 22:53:23 +000011599{
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -050011600 if (PyUnicode_READY(self) == -1)
11601 return NULL;
11602 if (PyUnicode_IS_ASCII(self))
11603 return ascii_upper_or_lower(self, 1);
Victor Stinnerb0800dc2012-02-25 00:47:08 +010011604 return case_operation(self, do_lower);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011605}
11606
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011607#define LEFTSTRIP 0
11608#define RIGHTSTRIP 1
11609#define BOTHSTRIP 2
11610
11611/* Arrays indexed by above */
11612static const char *stripformat[] = {"|O:lstrip", "|O:rstrip", "|O:strip"};
11613
11614#define STRIPNAME(i) (stripformat[i]+3)
11615
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011616/* externally visible for str.strip(unicode) */
11617PyObject *
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011618_PyUnicode_XStrip(PyObject *self, int striptype, PyObject *sepobj)
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011619{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011620 void *data;
11621 int kind;
11622 Py_ssize_t i, j, len;
11623 BLOOM_MASK sepmask;
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011624
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011625 if (PyUnicode_READY(self) == -1 || PyUnicode_READY(sepobj) == -1)
11626 return NULL;
11627
11628 kind = PyUnicode_KIND(self);
11629 data = PyUnicode_DATA(self);
11630 len = PyUnicode_GET_LENGTH(self);
11631 sepmask = make_bloom_mask(PyUnicode_KIND(sepobj),
11632 PyUnicode_DATA(sepobj),
11633 PyUnicode_GET_LENGTH(sepobj));
Thomas Wouters477c8d52006-05-27 19:21:47 +000011634
Benjamin Peterson14339b62009-01-31 16:36:08 +000011635 i = 0;
11636 if (striptype != RIGHTSTRIP) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011637 while (i < len &&
11638 BLOOM_MEMBER(sepmask, PyUnicode_READ(kind, data, i), sepobj)) {
Benjamin Peterson29060642009-01-31 22:14:21 +000011639 i++;
11640 }
Benjamin Peterson14339b62009-01-31 16:36:08 +000011641 }
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011642
Benjamin Peterson14339b62009-01-31 16:36:08 +000011643 j = len;
11644 if (striptype != LEFTSTRIP) {
Benjamin Peterson29060642009-01-31 22:14:21 +000011645 do {
11646 j--;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011647 } while (j >= i &&
11648 BLOOM_MEMBER(sepmask, PyUnicode_READ(kind, data, j), sepobj));
Benjamin Peterson29060642009-01-31 22:14:21 +000011649 j++;
Benjamin Peterson14339b62009-01-31 16:36:08 +000011650 }
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011651
Victor Stinner7931d9a2011-11-04 00:22:48 +010011652 return PyUnicode_Substring(self, i, j);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011653}
11654
11655PyObject*
11656PyUnicode_Substring(PyObject *self, Py_ssize_t start, Py_ssize_t end)
11657{
11658 unsigned char *data;
11659 int kind;
Victor Stinner12bab6d2011-10-01 01:53:49 +020011660 Py_ssize_t length;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011661
Victor Stinnerde636f32011-10-01 03:55:54 +020011662 if (PyUnicode_READY(self) == -1)
11663 return NULL;
11664
Victor Stinner684d5fd2012-05-03 02:32:34 +020011665 length = PyUnicode_GET_LENGTH(self);
11666 end = Py_MIN(end, length);
Victor Stinnerde636f32011-10-01 03:55:54 +020011667
Victor Stinner684d5fd2012-05-03 02:32:34 +020011668 if (start == 0 && end == length)
Victor Stinnerc4b49542011-12-11 22:44:26 +010011669 return unicode_result_unchanged(self);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011670
Victor Stinnerde636f32011-10-01 03:55:54 +020011671 if (start < 0 || end < 0) {
Victor Stinner12bab6d2011-10-01 01:53:49 +020011672 PyErr_SetString(PyExc_IndexError, "string index out of range");
11673 return NULL;
11674 }
Victor Stinner684d5fd2012-05-03 02:32:34 +020011675 if (start >= length || end < start) {
Victor Stinner3a7f79772012-05-03 03:36:40 +020011676 Py_INCREF(unicode_empty);
11677 return unicode_empty;
Victor Stinner684d5fd2012-05-03 02:32:34 +020011678 }
Victor Stinner12bab6d2011-10-01 01:53:49 +020011679
Victor Stinner684d5fd2012-05-03 02:32:34 +020011680 length = end - start;
Victor Stinnerb9275c12011-10-05 14:01:42 +020011681 if (PyUnicode_IS_ASCII(self)) {
Victor Stinnerb9275c12011-10-05 14:01:42 +020011682 data = PyUnicode_1BYTE_DATA(self);
Victor Stinnerd3f08822012-05-29 12:57:52 +020011683 return _PyUnicode_FromASCII((char*)(data + start), length);
Victor Stinnerb9275c12011-10-05 14:01:42 +020011684 }
11685 else {
11686 kind = PyUnicode_KIND(self);
11687 data = PyUnicode_1BYTE_DATA(self);
11688 return PyUnicode_FromKindAndData(kind,
Martin v. Löwisc47adb02011-10-07 20:55:35 +020011689 data + kind * start,
Victor Stinnerb9275c12011-10-05 14:01:42 +020011690 length);
11691 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011692}
Guido van Rossumd57fd912000-03-10 22:53:23 +000011693
11694static PyObject *
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011695do_strip(PyObject *self, int striptype)
Guido van Rossumd57fd912000-03-10 22:53:23 +000011696{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011697 int kind;
11698 void *data;
11699 Py_ssize_t len, i, j;
11700
11701 if (PyUnicode_READY(self) == -1)
11702 return NULL;
11703
11704 kind = PyUnicode_KIND(self);
11705 data = PyUnicode_DATA(self);
11706 len = PyUnicode_GET_LENGTH(self);
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011707
Benjamin Peterson14339b62009-01-31 16:36:08 +000011708 i = 0;
11709 if (striptype != RIGHTSTRIP) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011710 while (i < len && Py_UNICODE_ISSPACE(PyUnicode_READ(kind, data, i))) {
Benjamin Peterson14339b62009-01-31 16:36:08 +000011711 i++;
11712 }
11713 }
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011714
Benjamin Peterson14339b62009-01-31 16:36:08 +000011715 j = len;
11716 if (striptype != LEFTSTRIP) {
11717 do {
11718 j--;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011719 } while (j >= i && Py_UNICODE_ISSPACE(PyUnicode_READ(kind, data, j)));
Benjamin Peterson14339b62009-01-31 16:36:08 +000011720 j++;
11721 }
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011722
Victor Stinner7931d9a2011-11-04 00:22:48 +010011723 return PyUnicode_Substring(self, i, j);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011724}
11725
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011726
11727static PyObject *
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011728do_argstrip(PyObject *self, int striptype, PyObject *args)
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011729{
Benjamin Peterson14339b62009-01-31 16:36:08 +000011730 PyObject *sep = NULL;
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011731
Benjamin Peterson14339b62009-01-31 16:36:08 +000011732 if (!PyArg_ParseTuple(args, (char *)stripformat[striptype], &sep))
11733 return NULL;
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011734
Benjamin Peterson14339b62009-01-31 16:36:08 +000011735 if (sep != NULL && sep != Py_None) {
11736 if (PyUnicode_Check(sep))
11737 return _PyUnicode_XStrip(self, striptype, sep);
11738 else {
11739 PyErr_Format(PyExc_TypeError,
Benjamin Peterson29060642009-01-31 22:14:21 +000011740 "%s arg must be None or str",
11741 STRIPNAME(striptype));
Benjamin Peterson14339b62009-01-31 16:36:08 +000011742 return NULL;
11743 }
11744 }
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011745
Benjamin Peterson14339b62009-01-31 16:36:08 +000011746 return do_strip(self, striptype);
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011747}
11748
11749
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011750PyDoc_STRVAR(strip__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000011751 "S.strip([chars]) -> str\n\
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011752\n\
11753Return a copy of the string S with leading and trailing\n\
11754whitespace removed.\n\
Benjamin Peterson142957c2008-07-04 19:55:29 +000011755If chars is given and not None, remove characters in chars instead.");
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011756
11757static PyObject *
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011758unicode_strip(PyObject *self, PyObject *args)
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011759{
Benjamin Peterson14339b62009-01-31 16:36:08 +000011760 if (PyTuple_GET_SIZE(args) == 0)
11761 return do_strip(self, BOTHSTRIP); /* Common case */
11762 else
11763 return do_argstrip(self, BOTHSTRIP, args);
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011764}
11765
11766
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011767PyDoc_STRVAR(lstrip__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000011768 "S.lstrip([chars]) -> str\n\
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011769\n\
11770Return a copy of the string S with leading whitespace removed.\n\
Benjamin Peterson142957c2008-07-04 19:55:29 +000011771If chars is given and not None, remove characters in chars instead.");
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011772
11773static PyObject *
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011774unicode_lstrip(PyObject *self, PyObject *args)
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011775{
Benjamin Peterson14339b62009-01-31 16:36:08 +000011776 if (PyTuple_GET_SIZE(args) == 0)
11777 return do_strip(self, LEFTSTRIP); /* Common case */
11778 else
11779 return do_argstrip(self, LEFTSTRIP, args);
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011780}
11781
11782
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011783PyDoc_STRVAR(rstrip__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000011784 "S.rstrip([chars]) -> str\n\
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011785\n\
11786Return a copy of the string S with trailing whitespace removed.\n\
Benjamin Peterson142957c2008-07-04 19:55:29 +000011787If chars is given and not None, remove characters in chars instead.");
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011788
11789static PyObject *
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011790unicode_rstrip(PyObject *self, PyObject *args)
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011791{
Benjamin Peterson14339b62009-01-31 16:36:08 +000011792 if (PyTuple_GET_SIZE(args) == 0)
11793 return do_strip(self, RIGHTSTRIP); /* Common case */
11794 else
11795 return do_argstrip(self, RIGHTSTRIP, args);
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000011796}
11797
11798
Guido van Rossumd57fd912000-03-10 22:53:23 +000011799static PyObject*
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011800unicode_repeat(PyObject *str, Py_ssize_t len)
Guido van Rossumd57fd912000-03-10 22:53:23 +000011801{
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011802 PyObject *u;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011803 Py_ssize_t nchars, n;
Guido van Rossumd57fd912000-03-10 22:53:23 +000011804
Georg Brandl222de0f2009-04-12 12:01:50 +000011805 if (len < 1) {
11806 Py_INCREF(unicode_empty);
Victor Stinnera464fc12011-10-02 20:39:30 +020011807 return unicode_empty;
Georg Brandl222de0f2009-04-12 12:01:50 +000011808 }
Guido van Rossumd57fd912000-03-10 22:53:23 +000011809
Victor Stinnerc4b49542011-12-11 22:44:26 +010011810 /* no repeat, return original string */
11811 if (len == 1)
11812 return unicode_result_unchanged(str);
Tim Peters8f422462000-09-09 06:13:41 +000011813
Benjamin Petersonbac79492012-01-14 13:34:47 -050011814 if (PyUnicode_READY(str) == -1)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011815 return NULL;
11816
Victor Stinnerc759f3e2011-10-01 03:09:58 +020011817 if (PyUnicode_GET_LENGTH(str) > PY_SSIZE_T_MAX / len) {
Victor Stinner67ca64c2011-10-01 02:47:29 +020011818 PyErr_SetString(PyExc_OverflowError,
11819 "repeated string is too long");
11820 return NULL;
11821 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011822 nchars = len * PyUnicode_GET_LENGTH(str);
Victor Stinner67ca64c2011-10-01 02:47:29 +020011823
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011824 u = PyUnicode_New(nchars, PyUnicode_MAX_CHAR_VALUE(str));
Guido van Rossumd57fd912000-03-10 22:53:23 +000011825 if (!u)
11826 return NULL;
Victor Stinner67ca64c2011-10-01 02:47:29 +020011827 assert(PyUnicode_KIND(u) == PyUnicode_KIND(str));
Guido van Rossumd57fd912000-03-10 22:53:23 +000011828
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011829 if (PyUnicode_GET_LENGTH(str) == 1) {
11830 const int kind = PyUnicode_KIND(str);
11831 const Py_UCS4 fill_char = PyUnicode_READ(kind, PyUnicode_DATA(str), 0);
Victor Stinner73f53b52011-12-18 03:26:31 +010011832 if (kind == PyUnicode_1BYTE_KIND) {
11833 void *to = PyUnicode_DATA(u);
Victor Stinner67ca64c2011-10-01 02:47:29 +020011834 memset(to, (unsigned char)fill_char, len);
Victor Stinner73f53b52011-12-18 03:26:31 +010011835 }
11836 else if (kind == PyUnicode_2BYTE_KIND) {
11837 Py_UCS2 *ucs2 = PyUnicode_2BYTE_DATA(u);
Victor Stinner67ca64c2011-10-01 02:47:29 +020011838 for (n = 0; n < len; ++n)
Victor Stinner73f53b52011-12-18 03:26:31 +010011839 ucs2[n] = fill_char;
11840 } else {
11841 Py_UCS4 *ucs4 = PyUnicode_4BYTE_DATA(u);
11842 assert(kind == PyUnicode_4BYTE_KIND);
11843 for (n = 0; n < len; ++n)
11844 ucs4[n] = fill_char;
Victor Stinner67ca64c2011-10-01 02:47:29 +020011845 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011846 }
11847 else {
11848 /* number of characters copied this far */
11849 Py_ssize_t done = PyUnicode_GET_LENGTH(str);
Martin v. Löwisc47adb02011-10-07 20:55:35 +020011850 const Py_ssize_t char_size = PyUnicode_KIND(str);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011851 char *to = (char *) PyUnicode_DATA(u);
11852 Py_MEMCPY(to, PyUnicode_DATA(str),
11853 PyUnicode_GET_LENGTH(str) * char_size);
Benjamin Peterson29060642009-01-31 22:14:21 +000011854 while (done < nchars) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011855 n = (done <= nchars-done) ? done : nchars-done;
11856 Py_MEMCPY(to + (done * char_size), to, n * char_size);
Thomas Wouters477c8d52006-05-27 19:21:47 +000011857 done += n;
Benjamin Peterson29060642009-01-31 22:14:21 +000011858 }
Guido van Rossumd57fd912000-03-10 22:53:23 +000011859 }
11860
Victor Stinnerbb10a1f2011-10-05 01:34:17 +020011861 assert(_PyUnicode_CheckConsistency(u, 1));
Victor Stinner9db1a8b2011-10-23 20:04:37 +020011862 return u;
Guido van Rossumd57fd912000-03-10 22:53:23 +000011863}
11864
Alexander Belopolsky40018472011-02-26 01:02:56 +000011865PyObject *
11866PyUnicode_Replace(PyObject *obj,
11867 PyObject *subobj,
11868 PyObject *replobj,
11869 Py_ssize_t maxcount)
Guido van Rossumd57fd912000-03-10 22:53:23 +000011870{
11871 PyObject *self;
11872 PyObject *str1;
11873 PyObject *str2;
11874 PyObject *result;
11875
11876 self = PyUnicode_FromObject(obj);
Benjamin Peterson22a29702012-01-02 09:00:30 -060011877 if (self == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +000011878 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +000011879 str1 = PyUnicode_FromObject(subobj);
Benjamin Peterson22a29702012-01-02 09:00:30 -060011880 if (str1 == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +000011881 Py_DECREF(self);
11882 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +000011883 }
11884 str2 = PyUnicode_FromObject(replobj);
Benjamin Peterson22a29702012-01-02 09:00:30 -060011885 if (str2 == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +000011886 Py_DECREF(self);
11887 Py_DECREF(str1);
11888 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +000011889 }
Benjamin Peterson22a29702012-01-02 09:00:30 -060011890 if (PyUnicode_READY(self) == -1 ||
11891 PyUnicode_READY(str1) == -1 ||
11892 PyUnicode_READY(str2) == -1)
11893 result = NULL;
11894 else
11895 result = replace(self, str1, str2, maxcount);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011896 Py_DECREF(self);
11897 Py_DECREF(str1);
11898 Py_DECREF(str2);
11899 return result;
11900}
11901
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000011902PyDoc_STRVAR(replace__doc__,
Ezio Melottic1897e72010-06-26 18:50:39 +000011903 "S.replace(old, new[, count]) -> str\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000011904\n\
11905Return a copy of S with all occurrences of substring\n\
Georg Brandlf08a9dd2008-06-10 16:57:31 +000011906old replaced by new. If the optional argument count is\n\
11907given, only the first count occurrences are replaced.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000011908
11909static PyObject*
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011910unicode_replace(PyObject *self, PyObject *args)
Guido van Rossumd57fd912000-03-10 22:53:23 +000011911{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011912 PyObject *str1;
11913 PyObject *str2;
Martin v. Löwis18e16552006-02-15 17:27:45 +000011914 Py_ssize_t maxcount = -1;
Guido van Rossumd57fd912000-03-10 22:53:23 +000011915 PyObject *result;
11916
Martin v. Löwis18e16552006-02-15 17:27:45 +000011917 if (!PyArg_ParseTuple(args, "OO|n:replace", &str1, &str2, &maxcount))
Guido van Rossumd57fd912000-03-10 22:53:23 +000011918 return NULL;
Benjamin Peterson22a29702012-01-02 09:00:30 -060011919 if (PyUnicode_READY(self) == -1)
Benjamin Peterson29060642009-01-31 22:14:21 +000011920 return NULL;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011921 str1 = PyUnicode_FromObject(str1);
Benjamin Peterson22a29702012-01-02 09:00:30 -060011922 if (str1 == NULL)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011923 return NULL;
11924 str2 = PyUnicode_FromObject(str2);
Benjamin Peterson22a29702012-01-02 09:00:30 -060011925 if (str2 == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +000011926 Py_DECREF(str1);
11927 return NULL;
Walter Dörwaldf6b56ae2003-02-09 23:42:56 +000011928 }
Benjamin Peterson22a29702012-01-02 09:00:30 -060011929 if (PyUnicode_READY(str1) == -1 || PyUnicode_READY(str2) == -1)
11930 result = NULL;
11931 else
11932 result = replace(self, str1, str2, maxcount);
Guido van Rossumd57fd912000-03-10 22:53:23 +000011933
11934 Py_DECREF(str1);
11935 Py_DECREF(str2);
11936 return result;
11937}
11938
Alexander Belopolsky40018472011-02-26 01:02:56 +000011939static PyObject *
11940unicode_repr(PyObject *unicode)
Guido van Rossumd57fd912000-03-10 22:53:23 +000011941{
Walter Dörwald79e913e2007-05-12 11:08:06 +000011942 PyObject *repr;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011943 Py_ssize_t isize;
11944 Py_ssize_t osize, squote, dquote, i, o;
11945 Py_UCS4 max, quote;
11946 int ikind, okind;
11947 void *idata, *odata;
Walter Dörwald79e913e2007-05-12 11:08:06 +000011948
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011949 if (PyUnicode_READY(unicode) == -1)
Walter Dörwald79e913e2007-05-12 11:08:06 +000011950 return NULL;
11951
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011952 isize = PyUnicode_GET_LENGTH(unicode);
11953 idata = PyUnicode_DATA(unicode);
Walter Dörwald79e913e2007-05-12 11:08:06 +000011954
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020011955 /* Compute length of output, quote characters, and
11956 maximum character */
11957 osize = 2; /* quotes */
11958 max = 127;
11959 squote = dquote = 0;
11960 ikind = PyUnicode_KIND(unicode);
11961 for (i = 0; i < isize; i++) {
11962 Py_UCS4 ch = PyUnicode_READ(ikind, idata, i);
11963 switch (ch) {
11964 case '\'': squote++; osize++; break;
11965 case '"': dquote++; osize++; break;
11966 case '\\': case '\t': case '\r': case '\n':
11967 osize += 2; break;
11968 default:
11969 /* Fast-path ASCII */
11970 if (ch < ' ' || ch == 0x7f)
11971 osize += 4; /* \xHH */
11972 else if (ch < 0x7f)
11973 osize++;
11974 else if (Py_UNICODE_ISPRINTABLE(ch)) {
11975 osize++;
11976 max = ch > max ? ch : max;
11977 }
11978 else if (ch < 0x100)
11979 osize += 4; /* \xHH */
11980 else if (ch < 0x10000)
11981 osize += 6; /* \uHHHH */
11982 else
11983 osize += 10; /* \uHHHHHHHH */
11984 }
11985 }
11986
11987 quote = '\'';
11988 if (squote) {
11989 if (dquote)
11990 /* Both squote and dquote present. Use squote,
11991 and escape them */
11992 osize += squote;
11993 else
11994 quote = '"';
11995 }
11996
11997 repr = PyUnicode_New(osize, max);
11998 if (repr == NULL)
11999 return NULL;
12000 okind = PyUnicode_KIND(repr);
12001 odata = PyUnicode_DATA(repr);
12002
12003 PyUnicode_WRITE(okind, odata, 0, quote);
12004 PyUnicode_WRITE(okind, odata, osize-1, quote);
12005
12006 for (i = 0, o = 1; i < isize; i++) {
12007 Py_UCS4 ch = PyUnicode_READ(ikind, idata, i);
Walter Dörwald79e913e2007-05-12 11:08:06 +000012008
12009 /* Escape quotes and backslashes */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012010 if ((ch == quote) || (ch == '\\')) {
12011 PyUnicode_WRITE(okind, odata, o++, '\\');
12012 PyUnicode_WRITE(okind, odata, o++, ch);
Walter Dörwald79e913e2007-05-12 11:08:06 +000012013 continue;
12014 }
12015
Benjamin Peterson29060642009-01-31 22:14:21 +000012016 /* Map special whitespace to '\t', \n', '\r' */
Georg Brandl559e5d72008-06-11 18:37:52 +000012017 if (ch == '\t') {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012018 PyUnicode_WRITE(okind, odata, o++, '\\');
12019 PyUnicode_WRITE(okind, odata, o++, 't');
Walter Dörwald79e913e2007-05-12 11:08:06 +000012020 }
12021 else if (ch == '\n') {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012022 PyUnicode_WRITE(okind, odata, o++, '\\');
12023 PyUnicode_WRITE(okind, odata, o++, 'n');
Walter Dörwald79e913e2007-05-12 11:08:06 +000012024 }
12025 else if (ch == '\r') {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012026 PyUnicode_WRITE(okind, odata, o++, '\\');
12027 PyUnicode_WRITE(okind, odata, o++, 'r');
Walter Dörwald79e913e2007-05-12 11:08:06 +000012028 }
12029
12030 /* Map non-printable US ASCII to '\xhh' */
Georg Brandl559e5d72008-06-11 18:37:52 +000012031 else if (ch < ' ' || ch == 0x7F) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012032 PyUnicode_WRITE(okind, odata, o++, '\\');
12033 PyUnicode_WRITE(okind, odata, o++, 'x');
Victor Stinnerf5cff562011-10-14 02:13:11 +020012034 PyUnicode_WRITE(okind, odata, o++, Py_hexdigits[(ch >> 4) & 0x000F]);
12035 PyUnicode_WRITE(okind, odata, o++, Py_hexdigits[ch & 0x000F]);
Walter Dörwald79e913e2007-05-12 11:08:06 +000012036 }
12037
Georg Brandl559e5d72008-06-11 18:37:52 +000012038 /* Copy ASCII characters as-is */
12039 else if (ch < 0x7F) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012040 PyUnicode_WRITE(okind, odata, o++, ch);
Georg Brandl559e5d72008-06-11 18:37:52 +000012041 }
12042
Benjamin Peterson29060642009-01-31 22:14:21 +000012043 /* Non-ASCII characters */
Georg Brandl559e5d72008-06-11 18:37:52 +000012044 else {
Benjamin Peterson14339b62009-01-31 16:36:08 +000012045 /* Map Unicode whitespace and control characters
Georg Brandl559e5d72008-06-11 18:37:52 +000012046 (categories Z* and C* except ASCII space)
12047 */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012048 if (!Py_UNICODE_ISPRINTABLE(ch)) {
Kristján Valur Jónsson55e5dc82012-06-06 21:58:08 +000012049 PyUnicode_WRITE(okind, odata, o++, '\\');
Georg Brandl559e5d72008-06-11 18:37:52 +000012050 /* Map 8-bit characters to '\xhh' */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012051 if (ch <= 0xff) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012052 PyUnicode_WRITE(okind, odata, o++, 'x');
Victor Stinnerf5cff562011-10-14 02:13:11 +020012053 PyUnicode_WRITE(okind, odata, o++, Py_hexdigits[(ch >> 4) & 0x000F]);
12054 PyUnicode_WRITE(okind, odata, o++, Py_hexdigits[ch & 0x000F]);
Georg Brandl559e5d72008-06-11 18:37:52 +000012055 }
Kristján Valur Jónsson55e5dc82012-06-06 21:58:08 +000012056 /* Map 16-bit characters to '\uxxxx' */
12057 else if (ch <= 0xffff) {
12058 PyUnicode_WRITE(okind, odata, o++, 'u');
Victor Stinnerf5cff562011-10-14 02:13:11 +020012059 PyUnicode_WRITE(okind, odata, o++, Py_hexdigits[(ch >> 12) & 0xF]);
12060 PyUnicode_WRITE(okind, odata, o++, Py_hexdigits[(ch >> 8) & 0xF]);
12061 PyUnicode_WRITE(okind, odata, o++, Py_hexdigits[(ch >> 4) & 0xF]);
12062 PyUnicode_WRITE(okind, odata, o++, Py_hexdigits[ch & 0xF]);
Georg Brandl559e5d72008-06-11 18:37:52 +000012063 }
Kristján Valur Jónsson55e5dc82012-06-06 21:58:08 +000012064 /* Map 21-bit characters to '\U00xxxxxx' */
Georg Brandl559e5d72008-06-11 18:37:52 +000012065 else {
Kristján Valur Jónsson55e5dc82012-06-06 21:58:08 +000012066 PyUnicode_WRITE(okind, odata, o++, 'U');
12067 PyUnicode_WRITE(okind, odata, o++, Py_hexdigits[(ch >> 28) & 0xF]);
12068 PyUnicode_WRITE(okind, odata, o++, Py_hexdigits[(ch >> 24) & 0xF]);
12069 PyUnicode_WRITE(okind, odata, o++, Py_hexdigits[(ch >> 20) & 0xF]);
12070 PyUnicode_WRITE(okind, odata, o++, Py_hexdigits[(ch >> 16) & 0xF]);
Victor Stinnerf5cff562011-10-14 02:13:11 +020012071 PyUnicode_WRITE(okind, odata, o++, Py_hexdigits[(ch >> 12) & 0xF]);
12072 PyUnicode_WRITE(okind, odata, o++, Py_hexdigits[(ch >> 8) & 0xF]);
12073 PyUnicode_WRITE(okind, odata, o++, Py_hexdigits[(ch >> 4) & 0xF]);
12074 PyUnicode_WRITE(okind, odata, o++, Py_hexdigits[ch & 0xF]);
Georg Brandl559e5d72008-06-11 18:37:52 +000012075 }
12076 }
12077 /* Copy characters as-is */
12078 else {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012079 PyUnicode_WRITE(okind, odata, o++, ch);
Georg Brandl559e5d72008-06-11 18:37:52 +000012080 }
12081 }
Walter Dörwald79e913e2007-05-12 11:08:06 +000012082 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012083 /* Closing quote already added at the beginning */
Victor Stinner05d11892011-10-06 01:13:58 +020012084 assert(_PyUnicode_CheckConsistency(repr, 1));
Walter Dörwald79e913e2007-05-12 11:08:06 +000012085 return repr;
Guido van Rossumd57fd912000-03-10 22:53:23 +000012086}
12087
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000012088PyDoc_STRVAR(rfind__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000012089 "S.rfind(sub[, start[, end]]) -> int\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000012090\n\
12091Return the highest index in S where substring sub is found,\n\
Senthil Kumaran53516a82011-07-27 23:33:54 +080012092such that sub is contained within S[start:end]. Optional\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000012093arguments start and end are interpreted as in slice notation.\n\
12094\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000012095Return -1 on failure.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000012096
12097static PyObject *
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012098unicode_rfind(PyObject *self, PyObject *args)
Guido van Rossumd57fd912000-03-10 22:53:23 +000012099{
Victor Stinner9db1a8b2011-10-23 20:04:37 +020012100 PyObject *substring;
Christian Heimes9cd17752007-11-18 19:35:23 +000012101 Py_ssize_t start;
12102 Py_ssize_t end;
Thomas Wouters477c8d52006-05-27 19:21:47 +000012103 Py_ssize_t result;
Guido van Rossumd57fd912000-03-10 22:53:23 +000012104
Jesus Ceaac451502011-04-20 17:09:23 +020012105 if (!stringlib_parse_args_finds_unicode("rfind", args, &substring,
12106 &start, &end))
Benjamin Peterson14339b62009-01-31 16:36:08 +000012107 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +000012108
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012109 if (PyUnicode_READY(self) == -1)
12110 return NULL;
12111 if (PyUnicode_READY(substring) == -1)
12112 return NULL;
12113
Victor Stinner7931d9a2011-11-04 00:22:48 +010012114 result = any_find_slice(-1, self, substring, start, end);
Guido van Rossumd57fd912000-03-10 22:53:23 +000012115
12116 Py_DECREF(substring);
Thomas Wouters477c8d52006-05-27 19:21:47 +000012117
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012118 if (result == -2)
12119 return NULL;
12120
Christian Heimes217cfd12007-12-02 14:31:20 +000012121 return PyLong_FromSsize_t(result);
Guido van Rossumd57fd912000-03-10 22:53:23 +000012122}
12123
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000012124PyDoc_STRVAR(rindex__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000012125 "S.rindex(sub[, start[, end]]) -> int\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000012126\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000012127Like S.rfind() but raise ValueError when the substring is not found.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000012128
12129static PyObject *
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012130unicode_rindex(PyObject *self, PyObject *args)
Guido van Rossumd57fd912000-03-10 22:53:23 +000012131{
Victor Stinner9db1a8b2011-10-23 20:04:37 +020012132 PyObject *substring;
Christian Heimes9cd17752007-11-18 19:35:23 +000012133 Py_ssize_t start;
12134 Py_ssize_t end;
Thomas Wouters477c8d52006-05-27 19:21:47 +000012135 Py_ssize_t result;
Guido van Rossumd57fd912000-03-10 22:53:23 +000012136
Jesus Ceaac451502011-04-20 17:09:23 +020012137 if (!stringlib_parse_args_finds_unicode("rindex", args, &substring,
12138 &start, &end))
Benjamin Peterson14339b62009-01-31 16:36:08 +000012139 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +000012140
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012141 if (PyUnicode_READY(self) == -1)
12142 return NULL;
12143 if (PyUnicode_READY(substring) == -1)
12144 return NULL;
12145
Victor Stinner7931d9a2011-11-04 00:22:48 +010012146 result = any_find_slice(-1, self, substring, start, end);
Guido van Rossumd57fd912000-03-10 22:53:23 +000012147
12148 Py_DECREF(substring);
Thomas Wouters477c8d52006-05-27 19:21:47 +000012149
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012150 if (result == -2)
12151 return NULL;
12152
Guido van Rossumd57fd912000-03-10 22:53:23 +000012153 if (result < 0) {
12154 PyErr_SetString(PyExc_ValueError, "substring not found");
12155 return NULL;
12156 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012157
Christian Heimes217cfd12007-12-02 14:31:20 +000012158 return PyLong_FromSsize_t(result);
Guido van Rossumd57fd912000-03-10 22:53:23 +000012159}
12160
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000012161PyDoc_STRVAR(rjust__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000012162 "S.rjust(width[, fillchar]) -> str\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000012163\n\
Benjamin Petersonf10a79a2008-10-11 00:49:57 +000012164Return S right-justified in a string of length width. Padding is\n\
Raymond Hettinger4f8f9762003-11-26 08:21:35 +000012165done using the specified fill character (default is a space).");
Guido van Rossumd57fd912000-03-10 22:53:23 +000012166
12167static PyObject *
Victor Stinner9310abb2011-10-05 00:59:23 +020012168unicode_rjust(PyObject *self, PyObject *args)
Guido van Rossumd57fd912000-03-10 22:53:23 +000012169{
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000012170 Py_ssize_t width;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012171 Py_UCS4 fillchar = ' ';
12172
Victor Stinnere9a29352011-10-01 02:14:59 +020012173 if (!PyArg_ParseTuple(args, "n|O&:rjust", &width, convert_uc, &fillchar))
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012174 return NULL;
Raymond Hettinger4f8f9762003-11-26 08:21:35 +000012175
Benjamin Petersonbac79492012-01-14 13:34:47 -050012176 if (PyUnicode_READY(self) == -1)
Guido van Rossumd57fd912000-03-10 22:53:23 +000012177 return NULL;
12178
Victor Stinnerc4b49542011-12-11 22:44:26 +010012179 if (PyUnicode_GET_LENGTH(self) >= width)
12180 return unicode_result_unchanged(self);
Guido van Rossumd57fd912000-03-10 22:53:23 +000012181
Victor Stinnerc4b49542011-12-11 22:44:26 +010012182 return pad(self, width - PyUnicode_GET_LENGTH(self), 0, fillchar);
Guido van Rossumd57fd912000-03-10 22:53:23 +000012183}
12184
Alexander Belopolsky40018472011-02-26 01:02:56 +000012185PyObject *
12186PyUnicode_Split(PyObject *s, PyObject *sep, Py_ssize_t maxsplit)
Guido van Rossumd57fd912000-03-10 22:53:23 +000012187{
12188 PyObject *result;
Tim Petersced69f82003-09-16 20:30:58 +000012189
Guido van Rossumd57fd912000-03-10 22:53:23 +000012190 s = PyUnicode_FromObject(s);
12191 if (s == NULL)
Benjamin Peterson14339b62009-01-31 16:36:08 +000012192 return NULL;
Benjamin Peterson29060642009-01-31 22:14:21 +000012193 if (sep != NULL) {
12194 sep = PyUnicode_FromObject(sep);
12195 if (sep == NULL) {
12196 Py_DECREF(s);
12197 return NULL;
12198 }
Guido van Rossumd57fd912000-03-10 22:53:23 +000012199 }
12200
Victor Stinner9310abb2011-10-05 00:59:23 +020012201 result = split(s, sep, maxsplit);
Guido van Rossumd57fd912000-03-10 22:53:23 +000012202
12203 Py_DECREF(s);
12204 Py_XDECREF(sep);
12205 return result;
12206}
12207
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000012208PyDoc_STRVAR(split__doc__,
Ezio Melotticda6b6d2012-02-26 09:39:55 +020012209 "S.split(sep=None, maxsplit=-1) -> list of strings\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000012210\n\
12211Return a list of the words in S, using sep as the\n\
12212delimiter string. If maxsplit is given, at most maxsplit\n\
Alexandre Vassalotti5f8ced22008-05-16 00:03:33 +000012213splits are done. If sep is not specified or is None, any\n\
Alexandre Vassalotti8ae3e052008-05-16 00:41:41 +000012214whitespace string is a separator and empty strings are\n\
12215removed from the result.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000012216
12217static PyObject*
Ezio Melotticda6b6d2012-02-26 09:39:55 +020012218unicode_split(PyObject *self, PyObject *args, PyObject *kwds)
Guido van Rossumd57fd912000-03-10 22:53:23 +000012219{
Ezio Melotticda6b6d2012-02-26 09:39:55 +020012220 static char *kwlist[] = {"sep", "maxsplit", 0};
Guido van Rossumd57fd912000-03-10 22:53:23 +000012221 PyObject *substring = Py_None;
Martin v. Löwis18e16552006-02-15 17:27:45 +000012222 Py_ssize_t maxcount = -1;
Guido van Rossumd57fd912000-03-10 22:53:23 +000012223
Ezio Melotticda6b6d2012-02-26 09:39:55 +020012224 if (!PyArg_ParseTupleAndKeywords(args, kwds, "|On:split",
12225 kwlist, &substring, &maxcount))
Guido van Rossumd57fd912000-03-10 22:53:23 +000012226 return NULL;
12227
12228 if (substring == Py_None)
Benjamin Peterson29060642009-01-31 22:14:21 +000012229 return split(self, NULL, maxcount);
Guido van Rossumd57fd912000-03-10 22:53:23 +000012230 else if (PyUnicode_Check(substring))
Victor Stinner9310abb2011-10-05 00:59:23 +020012231 return split(self, substring, maxcount);
Guido van Rossumd57fd912000-03-10 22:53:23 +000012232 else
Victor Stinner7931d9a2011-11-04 00:22:48 +010012233 return PyUnicode_Split(self, substring, maxcount);
Guido van Rossumd57fd912000-03-10 22:53:23 +000012234}
12235
Thomas Wouters477c8d52006-05-27 19:21:47 +000012236PyObject *
12237PyUnicode_Partition(PyObject *str_in, PyObject *sep_in)
12238{
12239 PyObject* str_obj;
12240 PyObject* sep_obj;
12241 PyObject* out;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012242 int kind1, kind2, kind;
12243 void *buf1 = NULL, *buf2 = NULL;
12244 Py_ssize_t len1, len2;
Thomas Wouters477c8d52006-05-27 19:21:47 +000012245
12246 str_obj = PyUnicode_FromObject(str_in);
Benjamin Peterson22a29702012-01-02 09:00:30 -060012247 if (!str_obj)
Benjamin Peterson29060642009-01-31 22:14:21 +000012248 return NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +000012249 sep_obj = PyUnicode_FromObject(sep_in);
Benjamin Peterson22a29702012-01-02 09:00:30 -060012250 if (!sep_obj) {
12251 Py_DECREF(str_obj);
12252 return NULL;
12253 }
12254 if (PyUnicode_READY(sep_obj) == -1 || PyUnicode_READY(str_obj) == -1) {
12255 Py_DECREF(sep_obj);
Thomas Wouters477c8d52006-05-27 19:21:47 +000012256 Py_DECREF(str_obj);
12257 return NULL;
12258 }
12259
Victor Stinner14f8f022011-10-05 20:58:25 +020012260 kind1 = PyUnicode_KIND(str_obj);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012261 kind2 = PyUnicode_KIND(sep_obj);
Victor Stinner14f8f022011-10-05 20:58:25 +020012262 kind = Py_MAX(kind1, kind2);
12263 buf1 = PyUnicode_DATA(str_obj);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012264 if (kind1 != kind)
Victor Stinner14f8f022011-10-05 20:58:25 +020012265 buf1 = _PyUnicode_AsKind(str_obj, kind);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012266 if (!buf1)
12267 goto onError;
12268 buf2 = PyUnicode_DATA(sep_obj);
12269 if (kind2 != kind)
12270 buf2 = _PyUnicode_AsKind(sep_obj, kind);
12271 if (!buf2)
12272 goto onError;
12273 len1 = PyUnicode_GET_LENGTH(str_obj);
12274 len2 = PyUnicode_GET_LENGTH(sep_obj);
12275
Benjamin Petersonead6b532011-12-20 17:23:42 -060012276 switch (PyUnicode_KIND(str_obj)) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012277 case PyUnicode_1BYTE_KIND:
Victor Stinnerc3cec782011-10-05 21:24:08 +020012278 if (PyUnicode_IS_ASCII(str_obj) && PyUnicode_IS_ASCII(sep_obj))
12279 out = asciilib_partition(str_obj, buf1, len1, sep_obj, buf2, len2);
12280 else
12281 out = ucs1lib_partition(str_obj, buf1, len1, sep_obj, buf2, len2);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012282 break;
12283 case PyUnicode_2BYTE_KIND:
12284 out = ucs2lib_partition(str_obj, buf1, len1, sep_obj, buf2, len2);
12285 break;
12286 case PyUnicode_4BYTE_KIND:
12287 out = ucs4lib_partition(str_obj, buf1, len1, sep_obj, buf2, len2);
12288 break;
12289 default:
12290 assert(0);
12291 out = 0;
12292 }
Thomas Wouters477c8d52006-05-27 19:21:47 +000012293
12294 Py_DECREF(sep_obj);
12295 Py_DECREF(str_obj);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012296 if (kind1 != kind)
12297 PyMem_Free(buf1);
12298 if (kind2 != kind)
12299 PyMem_Free(buf2);
Thomas Wouters477c8d52006-05-27 19:21:47 +000012300
12301 return out;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012302 onError:
12303 Py_DECREF(sep_obj);
12304 Py_DECREF(str_obj);
12305 if (kind1 != kind && buf1)
12306 PyMem_Free(buf1);
12307 if (kind2 != kind && buf2)
12308 PyMem_Free(buf2);
12309 return NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +000012310}
12311
12312
12313PyObject *
12314PyUnicode_RPartition(PyObject *str_in, PyObject *sep_in)
12315{
12316 PyObject* str_obj;
12317 PyObject* sep_obj;
12318 PyObject* out;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012319 int kind1, kind2, kind;
12320 void *buf1 = NULL, *buf2 = NULL;
12321 Py_ssize_t len1, len2;
Thomas Wouters477c8d52006-05-27 19:21:47 +000012322
12323 str_obj = PyUnicode_FromObject(str_in);
12324 if (!str_obj)
Benjamin Peterson29060642009-01-31 22:14:21 +000012325 return NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +000012326 sep_obj = PyUnicode_FromObject(sep_in);
12327 if (!sep_obj) {
12328 Py_DECREF(str_obj);
12329 return NULL;
12330 }
12331
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012332 kind1 = PyUnicode_KIND(str_in);
12333 kind2 = PyUnicode_KIND(sep_obj);
Georg Brandl4cb0de22011-09-28 21:49:49 +020012334 kind = Py_MAX(kind1, kind2);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012335 buf1 = PyUnicode_DATA(str_in);
12336 if (kind1 != kind)
12337 buf1 = _PyUnicode_AsKind(str_in, kind);
12338 if (!buf1)
12339 goto onError;
12340 buf2 = PyUnicode_DATA(sep_obj);
12341 if (kind2 != kind)
12342 buf2 = _PyUnicode_AsKind(sep_obj, kind);
12343 if (!buf2)
12344 goto onError;
12345 len1 = PyUnicode_GET_LENGTH(str_obj);
12346 len2 = PyUnicode_GET_LENGTH(sep_obj);
12347
Benjamin Petersonead6b532011-12-20 17:23:42 -060012348 switch (PyUnicode_KIND(str_in)) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012349 case PyUnicode_1BYTE_KIND:
Victor Stinnerc3cec782011-10-05 21:24:08 +020012350 if (PyUnicode_IS_ASCII(str_obj) && PyUnicode_IS_ASCII(sep_obj))
12351 out = asciilib_rpartition(str_obj, buf1, len1, sep_obj, buf2, len2);
12352 else
12353 out = ucs1lib_rpartition(str_obj, buf1, len1, sep_obj, buf2, len2);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012354 break;
12355 case PyUnicode_2BYTE_KIND:
12356 out = ucs2lib_rpartition(str_obj, buf1, len1, sep_obj, buf2, len2);
12357 break;
12358 case PyUnicode_4BYTE_KIND:
12359 out = ucs4lib_rpartition(str_obj, buf1, len1, sep_obj, buf2, len2);
12360 break;
12361 default:
12362 assert(0);
12363 out = 0;
12364 }
Thomas Wouters477c8d52006-05-27 19:21:47 +000012365
12366 Py_DECREF(sep_obj);
12367 Py_DECREF(str_obj);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012368 if (kind1 != kind)
12369 PyMem_Free(buf1);
12370 if (kind2 != kind)
12371 PyMem_Free(buf2);
Thomas Wouters477c8d52006-05-27 19:21:47 +000012372
12373 return out;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012374 onError:
12375 Py_DECREF(sep_obj);
12376 Py_DECREF(str_obj);
12377 if (kind1 != kind && buf1)
12378 PyMem_Free(buf1);
12379 if (kind2 != kind && buf2)
12380 PyMem_Free(buf2);
12381 return NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +000012382}
12383
12384PyDoc_STRVAR(partition__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000012385 "S.partition(sep) -> (head, sep, tail)\n\
Thomas Wouters477c8d52006-05-27 19:21:47 +000012386\n\
Georg Brandl17cb8a82008-05-30 08:20:09 +000012387Search for the separator sep in S, and return the part before it,\n\
Thomas Wouters477c8d52006-05-27 19:21:47 +000012388the separator itself, and the part after it. If the separator is not\n\
Benjamin Petersonf10a79a2008-10-11 00:49:57 +000012389found, return S and two empty strings.");
Thomas Wouters477c8d52006-05-27 19:21:47 +000012390
12391static PyObject*
Victor Stinner9310abb2011-10-05 00:59:23 +020012392unicode_partition(PyObject *self, PyObject *separator)
Thomas Wouters477c8d52006-05-27 19:21:47 +000012393{
Victor Stinner9310abb2011-10-05 00:59:23 +020012394 return PyUnicode_Partition(self, separator);
Thomas Wouters477c8d52006-05-27 19:21:47 +000012395}
12396
12397PyDoc_STRVAR(rpartition__doc__,
Ezio Melotti5b2b2422010-01-25 11:58:28 +000012398 "S.rpartition(sep) -> (head, sep, tail)\n\
Thomas Wouters477c8d52006-05-27 19:21:47 +000012399\n\
Georg Brandl17cb8a82008-05-30 08:20:09 +000012400Search for the separator sep in S, starting at the end of S, and return\n\
Thomas Wouters477c8d52006-05-27 19:21:47 +000012401the part before it, the separator itself, and the part after it. If the\n\
Benjamin Petersonf10a79a2008-10-11 00:49:57 +000012402separator is not found, return two empty strings and S.");
Thomas Wouters477c8d52006-05-27 19:21:47 +000012403
12404static PyObject*
Victor Stinner9310abb2011-10-05 00:59:23 +020012405unicode_rpartition(PyObject *self, PyObject *separator)
Thomas Wouters477c8d52006-05-27 19:21:47 +000012406{
Victor Stinner9310abb2011-10-05 00:59:23 +020012407 return PyUnicode_RPartition(self, separator);
Thomas Wouters477c8d52006-05-27 19:21:47 +000012408}
12409
Alexander Belopolsky40018472011-02-26 01:02:56 +000012410PyObject *
12411PyUnicode_RSplit(PyObject *s, PyObject *sep, Py_ssize_t maxsplit)
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +000012412{
12413 PyObject *result;
Benjamin Peterson14339b62009-01-31 16:36:08 +000012414
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +000012415 s = PyUnicode_FromObject(s);
12416 if (s == NULL)
Benjamin Peterson14339b62009-01-31 16:36:08 +000012417 return NULL;
Benjamin Peterson29060642009-01-31 22:14:21 +000012418 if (sep != NULL) {
12419 sep = PyUnicode_FromObject(sep);
12420 if (sep == NULL) {
12421 Py_DECREF(s);
12422 return NULL;
12423 }
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +000012424 }
12425
Victor Stinner9310abb2011-10-05 00:59:23 +020012426 result = rsplit(s, sep, maxsplit);
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +000012427
12428 Py_DECREF(s);
12429 Py_XDECREF(sep);
12430 return result;
12431}
12432
12433PyDoc_STRVAR(rsplit__doc__,
Ezio Melotticda6b6d2012-02-26 09:39:55 +020012434 "S.rsplit(sep=None, maxsplit=-1) -> list of strings\n\
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +000012435\n\
12436Return a list of the words in S, using sep as the\n\
12437delimiter string, starting at the end of the string and\n\
12438working to the front. If maxsplit is given, at most maxsplit\n\
12439splits are done. If sep is not specified, any whitespace string\n\
12440is a separator.");
12441
12442static PyObject*
Ezio Melotticda6b6d2012-02-26 09:39:55 +020012443unicode_rsplit(PyObject *self, PyObject *args, PyObject *kwds)
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +000012444{
Ezio Melotticda6b6d2012-02-26 09:39:55 +020012445 static char *kwlist[] = {"sep", "maxsplit", 0};
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +000012446 PyObject *substring = Py_None;
Martin v. Löwis18e16552006-02-15 17:27:45 +000012447 Py_ssize_t maxcount = -1;
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +000012448
Ezio Melotticda6b6d2012-02-26 09:39:55 +020012449 if (!PyArg_ParseTupleAndKeywords(args, kwds, "|On:rsplit",
12450 kwlist, &substring, &maxcount))
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +000012451 return NULL;
12452
12453 if (substring == Py_None)
Benjamin Peterson29060642009-01-31 22:14:21 +000012454 return rsplit(self, NULL, maxcount);
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +000012455 else if (PyUnicode_Check(substring))
Victor Stinner9310abb2011-10-05 00:59:23 +020012456 return rsplit(self, substring, maxcount);
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +000012457 else
Victor Stinner9310abb2011-10-05 00:59:23 +020012458 return PyUnicode_RSplit(self, substring, maxcount);
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +000012459}
12460
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000012461PyDoc_STRVAR(splitlines__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000012462 "S.splitlines([keepends]) -> list of strings\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000012463\n\
12464Return a list of the lines in S, breaking at line boundaries.\n\
Guido van Rossum86662912000-04-11 15:38:46 +000012465Line breaks are not included in the resulting list unless keepends\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000012466is given and true.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000012467
12468static PyObject*
Victor Stinner9db1a8b2011-10-23 20:04:37 +020012469unicode_splitlines(PyObject *self, PyObject *args, PyObject *kwds)
Guido van Rossumd57fd912000-03-10 22:53:23 +000012470{
Mark Dickinson0d5f6ad2011-09-24 09:14:39 +010012471 static char *kwlist[] = {"keepends", 0};
Guido van Rossum86662912000-04-11 15:38:46 +000012472 int keepends = 0;
Guido van Rossumd57fd912000-03-10 22:53:23 +000012473
Mark Dickinson0d5f6ad2011-09-24 09:14:39 +010012474 if (!PyArg_ParseTupleAndKeywords(args, kwds, "|i:splitlines",
12475 kwlist, &keepends))
Guido van Rossumd57fd912000-03-10 22:53:23 +000012476 return NULL;
12477
Victor Stinner9db1a8b2011-10-23 20:04:37 +020012478 return PyUnicode_Splitlines(self, keepends);
Guido van Rossumd57fd912000-03-10 22:53:23 +000012479}
12480
12481static
Guido van Rossumf15a29f2007-05-04 00:41:39 +000012482PyObject *unicode_str(PyObject *self)
Guido van Rossumd57fd912000-03-10 22:53:23 +000012483{
Victor Stinnerc4b49542011-12-11 22:44:26 +010012484 return unicode_result_unchanged(self);
Guido van Rossumd57fd912000-03-10 22:53:23 +000012485}
12486
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000012487PyDoc_STRVAR(swapcase__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000012488 "S.swapcase() -> str\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000012489\n\
12490Return a copy of S with uppercase characters converted to lowercase\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000012491and vice versa.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000012492
12493static PyObject*
Victor Stinner9310abb2011-10-05 00:59:23 +020012494unicode_swapcase(PyObject *self)
Guido van Rossumd57fd912000-03-10 22:53:23 +000012495{
Benjamin Petersoneea48462012-01-16 14:28:50 -050012496 if (PyUnicode_READY(self) == -1)
12497 return NULL;
Victor Stinnerb0800dc2012-02-25 00:47:08 +010012498 return case_operation(self, do_swapcase);
Guido van Rossumd57fd912000-03-10 22:53:23 +000012499}
12500
Georg Brandlceee0772007-11-27 23:48:05 +000012501PyDoc_STRVAR(maketrans__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000012502 "str.maketrans(x[, y[, z]]) -> dict (static method)\n\
Georg Brandlceee0772007-11-27 23:48:05 +000012503\n\
12504Return a translation table usable for str.translate().\n\
12505If there is only one argument, it must be a dictionary mapping Unicode\n\
12506ordinals (integers) or characters to Unicode ordinals, strings or None.\n\
Benjamin Peterson142957c2008-07-04 19:55:29 +000012507Character keys will be then converted to ordinals.\n\
Georg Brandlceee0772007-11-27 23:48:05 +000012508If there are two arguments, they must be strings of equal length, and\n\
12509in the resulting dictionary, each character in x will be mapped to the\n\
12510character at the same position in y. If there is a third argument, it\n\
12511must be a string, whose characters will be mapped to None in the result.");
12512
12513static PyObject*
Victor Stinner9db1a8b2011-10-23 20:04:37 +020012514unicode_maketrans(PyObject *null, PyObject *args)
Georg Brandlceee0772007-11-27 23:48:05 +000012515{
12516 PyObject *x, *y = NULL, *z = NULL;
12517 PyObject *new = NULL, *key, *value;
12518 Py_ssize_t i = 0;
12519 int res;
Benjamin Peterson14339b62009-01-31 16:36:08 +000012520
Georg Brandlceee0772007-11-27 23:48:05 +000012521 if (!PyArg_ParseTuple(args, "O|UU:maketrans", &x, &y, &z))
12522 return NULL;
12523 new = PyDict_New();
12524 if (!new)
12525 return NULL;
12526 if (y != NULL) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012527 int x_kind, y_kind, z_kind;
12528 void *x_data, *y_data, *z_data;
12529
Georg Brandlceee0772007-11-27 23:48:05 +000012530 /* x must be a string too, of equal length */
Georg Brandlceee0772007-11-27 23:48:05 +000012531 if (!PyUnicode_Check(x)) {
12532 PyErr_SetString(PyExc_TypeError, "first maketrans argument must "
12533 "be a string if there is a second argument");
12534 goto err;
12535 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012536 if (PyUnicode_GET_LENGTH(x) != PyUnicode_GET_LENGTH(y)) {
Georg Brandlceee0772007-11-27 23:48:05 +000012537 PyErr_SetString(PyExc_ValueError, "the first two maketrans "
12538 "arguments must have equal length");
12539 goto err;
12540 }
12541 /* create entries for translating chars in x to those in y */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012542 x_kind = PyUnicode_KIND(x);
12543 y_kind = PyUnicode_KIND(y);
12544 x_data = PyUnicode_DATA(x);
12545 y_data = PyUnicode_DATA(y);
12546 for (i = 0; i < PyUnicode_GET_LENGTH(x); i++) {
12547 key = PyLong_FromLong(PyUnicode_READ(x_kind, x_data, i));
Benjamin Peterson53aa1d72011-12-20 13:29:45 -060012548 if (!key)
Georg Brandlceee0772007-11-27 23:48:05 +000012549 goto err;
Benjamin Peterson822c7902011-12-20 13:32:50 -060012550 value = PyLong_FromLong(PyUnicode_READ(y_kind, y_data, i));
Benjamin Peterson53aa1d72011-12-20 13:29:45 -060012551 if (!value) {
12552 Py_DECREF(key);
12553 goto err;
12554 }
Georg Brandlceee0772007-11-27 23:48:05 +000012555 res = PyDict_SetItem(new, key, value);
12556 Py_DECREF(key);
12557 Py_DECREF(value);
12558 if (res < 0)
12559 goto err;
12560 }
12561 /* create entries for deleting chars in z */
12562 if (z != NULL) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012563 z_kind = PyUnicode_KIND(z);
12564 z_data = PyUnicode_DATA(z);
Victor Stinnerc4f281e2011-10-11 22:11:42 +020012565 for (i = 0; i < PyUnicode_GET_LENGTH(z); i++) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012566 key = PyLong_FromLong(PyUnicode_READ(z_kind, z_data, i));
Georg Brandlceee0772007-11-27 23:48:05 +000012567 if (!key)
12568 goto err;
12569 res = PyDict_SetItem(new, key, Py_None);
12570 Py_DECREF(key);
12571 if (res < 0)
12572 goto err;
12573 }
12574 }
12575 } else {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012576 int kind;
12577 void *data;
12578
Georg Brandlceee0772007-11-27 23:48:05 +000012579 /* x must be a dict */
Raymond Hettinger3ad05762009-05-29 22:11:22 +000012580 if (!PyDict_CheckExact(x)) {
Georg Brandlceee0772007-11-27 23:48:05 +000012581 PyErr_SetString(PyExc_TypeError, "if you give only one argument "
12582 "to maketrans it must be a dict");
12583 goto err;
12584 }
12585 /* copy entries into the new dict, converting string keys to int keys */
12586 while (PyDict_Next(x, &i, &key, &value)) {
12587 if (PyUnicode_Check(key)) {
12588 /* convert string keys to integer keys */
12589 PyObject *newkey;
Victor Stinnerc4f281e2011-10-11 22:11:42 +020012590 if (PyUnicode_GET_LENGTH(key) != 1) {
Georg Brandlceee0772007-11-27 23:48:05 +000012591 PyErr_SetString(PyExc_ValueError, "string keys in translate "
12592 "table must be of length 1");
12593 goto err;
12594 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012595 kind = PyUnicode_KIND(key);
12596 data = PyUnicode_DATA(key);
12597 newkey = PyLong_FromLong(PyUnicode_READ(kind, data, 0));
Georg Brandlceee0772007-11-27 23:48:05 +000012598 if (!newkey)
12599 goto err;
12600 res = PyDict_SetItem(new, newkey, value);
12601 Py_DECREF(newkey);
12602 if (res < 0)
12603 goto err;
Christian Heimes217cfd12007-12-02 14:31:20 +000012604 } else if (PyLong_Check(key)) {
Georg Brandlceee0772007-11-27 23:48:05 +000012605 /* just keep integer keys */
12606 if (PyDict_SetItem(new, key, value) < 0)
12607 goto err;
12608 } else {
12609 PyErr_SetString(PyExc_TypeError, "keys in translate table must "
12610 "be strings or integers");
12611 goto err;
12612 }
12613 }
12614 }
12615 return new;
12616 err:
12617 Py_DECREF(new);
12618 return NULL;
12619}
12620
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000012621PyDoc_STRVAR(translate__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000012622 "S.translate(table) -> str\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000012623\n\
12624Return a copy of the string S, where all characters have been mapped\n\
12625through the given translation table, which must be a mapping of\n\
Benjamin Peterson142957c2008-07-04 19:55:29 +000012626Unicode ordinals to Unicode ordinals, strings, or None.\n\
Walter Dörwald5c1ee172002-09-04 20:31:32 +000012627Unmapped characters are left untouched. Characters mapped to None\n\
12628are deleted.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000012629
12630static PyObject*
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012631unicode_translate(PyObject *self, PyObject *table)
Guido van Rossumd57fd912000-03-10 22:53:23 +000012632{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012633 return _PyUnicode_TranslateCharmap(self, table, "ignore");
Guido van Rossumd57fd912000-03-10 22:53:23 +000012634}
12635
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000012636PyDoc_STRVAR(upper__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000012637 "S.upper() -> str\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000012638\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000012639Return a copy of S converted to uppercase.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000012640
12641static PyObject*
Victor Stinner9310abb2011-10-05 00:59:23 +020012642unicode_upper(PyObject *self)
Guido van Rossumd57fd912000-03-10 22:53:23 +000012643{
Benjamin Petersonb2bf01d2012-01-11 18:17:06 -050012644 if (PyUnicode_READY(self) == -1)
12645 return NULL;
12646 if (PyUnicode_IS_ASCII(self))
12647 return ascii_upper_or_lower(self, 0);
Victor Stinnerb0800dc2012-02-25 00:47:08 +010012648 return case_operation(self, do_upper);
Guido van Rossumd57fd912000-03-10 22:53:23 +000012649}
12650
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000012651PyDoc_STRVAR(zfill__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000012652 "S.zfill(width) -> str\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000012653\n\
Benjamin Peterson9aa42992008-09-10 21:57:34 +000012654Pad a numeric string S with zeros on the left, to fill a field\n\
12655of the specified width. The string S is never truncated.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000012656
12657static PyObject *
Victor Stinner9310abb2011-10-05 00:59:23 +020012658unicode_zfill(PyObject *self, PyObject *args)
Guido van Rossumd57fd912000-03-10 22:53:23 +000012659{
Martin v. Löwis18e16552006-02-15 17:27:45 +000012660 Py_ssize_t fill;
Victor Stinner9310abb2011-10-05 00:59:23 +020012661 PyObject *u;
Martin v. Löwis18e16552006-02-15 17:27:45 +000012662 Py_ssize_t width;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012663 int kind;
12664 void *data;
12665 Py_UCS4 chr;
12666
Martin v. Löwis18e16552006-02-15 17:27:45 +000012667 if (!PyArg_ParseTuple(args, "n:zfill", &width))
Guido van Rossumd57fd912000-03-10 22:53:23 +000012668 return NULL;
12669
Benjamin Petersonbac79492012-01-14 13:34:47 -050012670 if (PyUnicode_READY(self) == -1)
Victor Stinnerc4b49542011-12-11 22:44:26 +010012671 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +000012672
Victor Stinnerc4b49542011-12-11 22:44:26 +010012673 if (PyUnicode_GET_LENGTH(self) >= width)
12674 return unicode_result_unchanged(self);
12675
12676 fill = width - PyUnicode_GET_LENGTH(self);
Guido van Rossumd57fd912000-03-10 22:53:23 +000012677
12678 u = pad(self, fill, 0, '0');
12679
Walter Dörwald068325e2002-04-15 13:36:47 +000012680 if (u == NULL)
12681 return NULL;
12682
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012683 kind = PyUnicode_KIND(u);
12684 data = PyUnicode_DATA(u);
12685 chr = PyUnicode_READ(kind, data, fill);
12686
12687 if (chr == '+' || chr == '-') {
Guido van Rossumd57fd912000-03-10 22:53:23 +000012688 /* move sign to beginning of string */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012689 PyUnicode_WRITE(kind, data, 0, chr);
12690 PyUnicode_WRITE(kind, data, fill, '0');
Guido van Rossumd57fd912000-03-10 22:53:23 +000012691 }
12692
Victor Stinnerbb10a1f2011-10-05 01:34:17 +020012693 assert(_PyUnicode_CheckConsistency(u, 1));
Victor Stinner7931d9a2011-11-04 00:22:48 +010012694 return u;
Guido van Rossumd57fd912000-03-10 22:53:23 +000012695}
Guido van Rossumd57fd912000-03-10 22:53:23 +000012696
12697#if 0
Alexander Belopolsky942af5a2010-12-04 03:38:46 +000012698static PyObject *
12699unicode__decimal2ascii(PyObject *self)
12700{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012701 return PyUnicode_TransformDecimalAndSpaceToASCII(self);
Alexander Belopolsky942af5a2010-12-04 03:38:46 +000012702}
Guido van Rossumd57fd912000-03-10 22:53:23 +000012703#endif
12704
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000012705PyDoc_STRVAR(startswith__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000012706 "S.startswith(prefix[, start[, end]]) -> bool\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000012707\n\
Guido van Rossuma7132182003-04-09 19:32:45 +000012708Return True if S starts with the specified prefix, False otherwise.\n\
12709With optional start, test S beginning at that position.\n\
Thomas Wouters0e3f5912006-08-11 14:57:12 +000012710With optional end, stop comparing S at that position.\n\
12711prefix can also be a tuple of strings to try.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000012712
12713static PyObject *
Victor Stinner9db1a8b2011-10-23 20:04:37 +020012714unicode_startswith(PyObject *self,
Benjamin Peterson29060642009-01-31 22:14:21 +000012715 PyObject *args)
Guido van Rossumd57fd912000-03-10 22:53:23 +000012716{
Thomas Wouters0e3f5912006-08-11 14:57:12 +000012717 PyObject *subobj;
Victor Stinner9db1a8b2011-10-23 20:04:37 +020012718 PyObject *substring;
Martin v. Löwis18e16552006-02-15 17:27:45 +000012719 Py_ssize_t start = 0;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000012720 Py_ssize_t end = PY_SSIZE_T_MAX;
Thomas Wouters0e3f5912006-08-11 14:57:12 +000012721 int result;
Guido van Rossumd57fd912000-03-10 22:53:23 +000012722
Jesus Ceaac451502011-04-20 17:09:23 +020012723 if (!stringlib_parse_args_finds("startswith", args, &subobj, &start, &end))
Benjamin Peterson29060642009-01-31 22:14:21 +000012724 return NULL;
Thomas Wouters0e3f5912006-08-11 14:57:12 +000012725 if (PyTuple_Check(subobj)) {
12726 Py_ssize_t i;
12727 for (i = 0; i < PyTuple_GET_SIZE(subobj); i++) {
Victor Stinner9db1a8b2011-10-23 20:04:37 +020012728 substring = PyUnicode_FromObject(PyTuple_GET_ITEM(subobj, i));
Thomas Wouters0e3f5912006-08-11 14:57:12 +000012729 if (substring == NULL)
12730 return NULL;
12731 result = tailmatch(self, substring, start, end, -1);
12732 Py_DECREF(substring);
12733 if (result) {
12734 Py_RETURN_TRUE;
12735 }
12736 }
12737 /* nothing matched */
12738 Py_RETURN_FALSE;
12739 }
Victor Stinner9db1a8b2011-10-23 20:04:37 +020012740 substring = PyUnicode_FromObject(subobj);
Ezio Melottiba42fd52011-04-26 06:09:45 +030012741 if (substring == NULL) {
12742 if (PyErr_ExceptionMatches(PyExc_TypeError))
12743 PyErr_Format(PyExc_TypeError, "startswith first arg must be str or "
12744 "a tuple of str, not %s", Py_TYPE(subobj)->tp_name);
Benjamin Peterson29060642009-01-31 22:14:21 +000012745 return NULL;
Ezio Melottiba42fd52011-04-26 06:09:45 +030012746 }
Thomas Wouters0e3f5912006-08-11 14:57:12 +000012747 result = tailmatch(self, substring, start, end, -1);
Guido van Rossumd57fd912000-03-10 22:53:23 +000012748 Py_DECREF(substring);
Thomas Wouters0e3f5912006-08-11 14:57:12 +000012749 return PyBool_FromLong(result);
Guido van Rossumd57fd912000-03-10 22:53:23 +000012750}
12751
12752
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000012753PyDoc_STRVAR(endswith__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000012754 "S.endswith(suffix[, start[, end]]) -> bool\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +000012755\n\
Guido van Rossuma7132182003-04-09 19:32:45 +000012756Return True if S ends with the specified suffix, False otherwise.\n\
12757With optional start, test S beginning at that position.\n\
Thomas Wouters0e3f5912006-08-11 14:57:12 +000012758With optional end, stop comparing S at that position.\n\
12759suffix can also be a tuple of strings to try.");
Guido van Rossumd57fd912000-03-10 22:53:23 +000012760
12761static PyObject *
Victor Stinner9db1a8b2011-10-23 20:04:37 +020012762unicode_endswith(PyObject *self,
Benjamin Peterson29060642009-01-31 22:14:21 +000012763 PyObject *args)
Guido van Rossumd57fd912000-03-10 22:53:23 +000012764{
Thomas Wouters0e3f5912006-08-11 14:57:12 +000012765 PyObject *subobj;
Victor Stinner9db1a8b2011-10-23 20:04:37 +020012766 PyObject *substring;
Martin v. Löwis18e16552006-02-15 17:27:45 +000012767 Py_ssize_t start = 0;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000012768 Py_ssize_t end = PY_SSIZE_T_MAX;
Thomas Wouters0e3f5912006-08-11 14:57:12 +000012769 int result;
Guido van Rossumd57fd912000-03-10 22:53:23 +000012770
Jesus Ceaac451502011-04-20 17:09:23 +020012771 if (!stringlib_parse_args_finds("endswith", args, &subobj, &start, &end))
Benjamin Peterson29060642009-01-31 22:14:21 +000012772 return NULL;
Thomas Wouters0e3f5912006-08-11 14:57:12 +000012773 if (PyTuple_Check(subobj)) {
12774 Py_ssize_t i;
12775 for (i = 0; i < PyTuple_GET_SIZE(subobj); i++) {
Victor Stinner9db1a8b2011-10-23 20:04:37 +020012776 substring = PyUnicode_FromObject(
Benjamin Peterson29060642009-01-31 22:14:21 +000012777 PyTuple_GET_ITEM(subobj, i));
Thomas Wouters0e3f5912006-08-11 14:57:12 +000012778 if (substring == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +000012779 return NULL;
Thomas Wouters0e3f5912006-08-11 14:57:12 +000012780 result = tailmatch(self, substring, start, end, +1);
12781 Py_DECREF(substring);
12782 if (result) {
12783 Py_RETURN_TRUE;
12784 }
12785 }
12786 Py_RETURN_FALSE;
12787 }
Victor Stinner9db1a8b2011-10-23 20:04:37 +020012788 substring = PyUnicode_FromObject(subobj);
Ezio Melottiba42fd52011-04-26 06:09:45 +030012789 if (substring == NULL) {
12790 if (PyErr_ExceptionMatches(PyExc_TypeError))
12791 PyErr_Format(PyExc_TypeError, "endswith first arg must be str or "
12792 "a tuple of str, not %s", Py_TYPE(subobj)->tp_name);
Benjamin Peterson29060642009-01-31 22:14:21 +000012793 return NULL;
Ezio Melottiba42fd52011-04-26 06:09:45 +030012794 }
Thomas Wouters0e3f5912006-08-11 14:57:12 +000012795 result = tailmatch(self, substring, start, end, +1);
Guido van Rossumd57fd912000-03-10 22:53:23 +000012796 Py_DECREF(substring);
Thomas Wouters0e3f5912006-08-11 14:57:12 +000012797 return PyBool_FromLong(result);
Guido van Rossumd57fd912000-03-10 22:53:23 +000012798}
12799
Victor Stinner202fdca2012-05-07 12:47:02 +020012800Py_LOCAL_INLINE(void)
Victor Stinner3b1a74a2012-05-09 22:25:00 +020012801_PyUnicodeWriter_Update(_PyUnicodeWriter *writer)
Victor Stinner202fdca2012-05-07 12:47:02 +020012802{
Victor Stinnerd3f08822012-05-29 12:57:52 +020012803 writer->size = PyUnicode_GET_LENGTH(writer->buffer);
Victor Stinner202fdca2012-05-07 12:47:02 +020012804 writer->maxchar = PyUnicode_MAX_CHAR_VALUE(writer->buffer);
12805 writer->data = PyUnicode_DATA(writer->buffer);
12806 writer->kind = PyUnicode_KIND(writer->buffer);
12807}
12808
Victor Stinnerd3f08822012-05-29 12:57:52 +020012809void
12810_PyUnicodeWriter_Init(_PyUnicodeWriter *writer, Py_ssize_t min_length)
Victor Stinner202fdca2012-05-07 12:47:02 +020012811{
Victor Stinnerd3f08822012-05-29 12:57:52 +020012812 memset(writer, 0, sizeof(*writer));
12813#ifdef Py_DEBUG
12814 writer->kind = 5; /* invalid kind */
12815#endif
12816 writer->min_length = Py_MAX(min_length, 100);
Victor Stinnerd7b7c742012-06-04 22:52:12 +020012817 writer->overallocate = (min_length > 0);
Victor Stinner202fdca2012-05-07 12:47:02 +020012818}
12819
Victor Stinnerd3f08822012-05-29 12:57:52 +020012820int
12821_PyUnicodeWriter_PrepareInternal(_PyUnicodeWriter *writer,
12822 Py_ssize_t length, Py_UCS4 maxchar)
Victor Stinner202fdca2012-05-07 12:47:02 +020012823{
12824 Py_ssize_t newlen;
12825 PyObject *newbuffer;
12826
Victor Stinnerd3f08822012-05-29 12:57:52 +020012827 assert(length > 0);
12828
Victor Stinner202fdca2012-05-07 12:47:02 +020012829 if (length > PY_SSIZE_T_MAX - writer->pos) {
12830 PyErr_NoMemory();
12831 return -1;
12832 }
12833 newlen = writer->pos + length;
12834
Victor Stinnerd3f08822012-05-29 12:57:52 +020012835 if (writer->buffer == NULL) {
Victor Stinnerd7b7c742012-06-04 22:52:12 +020012836 if (writer->overallocate) {
Victor Stinnerd3f08822012-05-29 12:57:52 +020012837 /* overallocate 25% to limit the number of resize */
12838 if (newlen <= (PY_SSIZE_T_MAX - newlen / 4))
12839 newlen += newlen / 4;
12840 if (newlen < writer->min_length)
12841 newlen = writer->min_length;
12842 }
12843 writer->buffer = PyUnicode_New(newlen, maxchar);
12844 if (writer->buffer == NULL)
12845 return -1;
12846 _PyUnicodeWriter_Update(writer);
12847 return 0;
12848 }
Victor Stinner202fdca2012-05-07 12:47:02 +020012849
Victor Stinnerd3f08822012-05-29 12:57:52 +020012850 if (newlen > writer->size) {
Victor Stinnerd7b7c742012-06-04 22:52:12 +020012851 if (writer->overallocate) {
Victor Stinnerd3f08822012-05-29 12:57:52 +020012852 /* overallocate 25% to limit the number of resize */
12853 if (newlen <= (PY_SSIZE_T_MAX - newlen / 4))
12854 newlen += newlen / 4;
12855 if (newlen < writer->min_length)
12856 newlen = writer->min_length;
12857 }
12858
Victor Stinnerd7b7c742012-06-04 22:52:12 +020012859 if (maxchar > writer->maxchar || writer->readonly) {
Victor Stinner202fdca2012-05-07 12:47:02 +020012860 /* resize + widen */
12861 newbuffer = PyUnicode_New(newlen, maxchar);
12862 if (newbuffer == NULL)
12863 return -1;
Victor Stinnerd3f08822012-05-29 12:57:52 +020012864 _PyUnicode_FastCopyCharacters(newbuffer, 0,
12865 writer->buffer, 0, writer->pos);
Victor Stinner202fdca2012-05-07 12:47:02 +020012866 Py_DECREF(writer->buffer);
Victor Stinnerd7b7c742012-06-04 22:52:12 +020012867 writer->readonly = 0;
Victor Stinner202fdca2012-05-07 12:47:02 +020012868 }
12869 else {
12870 newbuffer = resize_compact(writer->buffer, newlen);
12871 if (newbuffer == NULL)
12872 return -1;
12873 }
12874 writer->buffer = newbuffer;
Victor Stinner3b1a74a2012-05-09 22:25:00 +020012875 _PyUnicodeWriter_Update(writer);
Victor Stinner202fdca2012-05-07 12:47:02 +020012876 }
12877 else if (maxchar > writer->maxchar) {
Victor Stinnerd7b7c742012-06-04 22:52:12 +020012878 assert(!writer->readonly);
Victor Stinnerd3f08822012-05-29 12:57:52 +020012879 newbuffer = PyUnicode_New(writer->size, maxchar);
12880 if (newbuffer == NULL)
Victor Stinner202fdca2012-05-07 12:47:02 +020012881 return -1;
Victor Stinnerd3f08822012-05-29 12:57:52 +020012882 _PyUnicode_FastCopyCharacters(newbuffer, 0,
12883 writer->buffer, 0, writer->pos);
12884 Py_DECREF(writer->buffer);
12885 writer->buffer = newbuffer;
Victor Stinner3b1a74a2012-05-09 22:25:00 +020012886 _PyUnicodeWriter_Update(writer);
Victor Stinner202fdca2012-05-07 12:47:02 +020012887 }
12888 return 0;
12889}
12890
Victor Stinnerd3f08822012-05-29 12:57:52 +020012891int
12892_PyUnicodeWriter_WriteStr(_PyUnicodeWriter *writer, PyObject *str)
12893{
12894 Py_UCS4 maxchar;
12895 Py_ssize_t len;
12896
12897 if (PyUnicode_READY(str) == -1)
12898 return -1;
12899 len = PyUnicode_GET_LENGTH(str);
12900 if (len == 0)
12901 return 0;
12902 maxchar = PyUnicode_MAX_CHAR_VALUE(str);
12903 if (maxchar > writer->maxchar || len > writer->size - writer->pos) {
Victor Stinnerd7b7c742012-06-04 22:52:12 +020012904 if (writer->buffer == NULL && !writer->overallocate) {
Victor Stinnerd3f08822012-05-29 12:57:52 +020012905 Py_INCREF(str);
12906 writer->buffer = str;
12907 _PyUnicodeWriter_Update(writer);
Victor Stinnerd7b7c742012-06-04 22:52:12 +020012908 writer->readonly = 1;
Victor Stinnerd3f08822012-05-29 12:57:52 +020012909 writer->size = 0;
12910 writer->pos += len;
12911 return 0;
12912 }
12913 if (_PyUnicodeWriter_PrepareInternal(writer, len, maxchar) == -1)
12914 return -1;
12915 }
12916 _PyUnicode_FastCopyCharacters(writer->buffer, writer->pos,
12917 str, 0, len);
12918 writer->pos += len;
12919 return 0;
12920}
12921
12922PyObject *
Victor Stinner3b1a74a2012-05-09 22:25:00 +020012923_PyUnicodeWriter_Finish(_PyUnicodeWriter *writer)
Victor Stinner202fdca2012-05-07 12:47:02 +020012924{
Victor Stinnerd3f08822012-05-29 12:57:52 +020012925 if (writer->pos == 0) {
12926 Py_XDECREF(writer->buffer);
12927 Py_INCREF(unicode_empty);
12928 return unicode_empty;
12929 }
Victor Stinnerd7b7c742012-06-04 22:52:12 +020012930 if (writer->readonly) {
Victor Stinnerd3f08822012-05-29 12:57:52 +020012931 assert(PyUnicode_GET_LENGTH(writer->buffer) == writer->pos);
12932 return writer->buffer;
12933 }
12934 if (PyUnicode_GET_LENGTH(writer->buffer) != writer->pos) {
12935 PyObject *newbuffer;
12936 newbuffer = resize_compact(writer->buffer, writer->pos);
12937 if (newbuffer == NULL) {
12938 Py_DECREF(writer->buffer);
12939 return NULL;
12940 }
12941 writer->buffer = newbuffer;
Victor Stinner202fdca2012-05-07 12:47:02 +020012942 }
Victor Stinnerf59c28c2012-05-09 03:24:14 +020012943 assert(_PyUnicode_CheckConsistency(writer->buffer, 1));
Victor Stinner202fdca2012-05-07 12:47:02 +020012944 return writer->buffer;
12945}
12946
Victor Stinnerd3f08822012-05-29 12:57:52 +020012947void
Victor Stinner3b1a74a2012-05-09 22:25:00 +020012948_PyUnicodeWriter_Dealloc(_PyUnicodeWriter *writer)
Victor Stinner202fdca2012-05-07 12:47:02 +020012949{
12950 Py_CLEAR(writer->buffer);
12951}
12952
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012953#include "stringlib/unicode_format.h"
Eric Smith8c663262007-08-25 02:26:07 +000012954
12955PyDoc_STRVAR(format__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000012956 "S.format(*args, **kwargs) -> str\n\
Eric Smith8c663262007-08-25 02:26:07 +000012957\n\
Eric Smith51d2fd92010-11-06 19:27:37 +000012958Return a formatted version of S, using substitutions from args and kwargs.\n\
12959The substitutions are identified by braces ('{' and '}').");
Eric Smith8c663262007-08-25 02:26:07 +000012960
Eric Smith27bbca62010-11-04 17:06:58 +000012961PyDoc_STRVAR(format_map__doc__,
12962 "S.format_map(mapping) -> str\n\
12963\n\
Eric Smith51d2fd92010-11-06 19:27:37 +000012964Return a formatted version of S, using substitutions from mapping.\n\
12965The substitutions are identified by braces ('{' and '}').");
Eric Smith27bbca62010-11-04 17:06:58 +000012966
Eric Smith4a7d76d2008-05-30 18:10:19 +000012967static PyObject *
12968unicode__format__(PyObject* self, PyObject* args)
12969{
Victor Stinnerd3f08822012-05-29 12:57:52 +020012970 PyObject *format_spec;
12971 _PyUnicodeWriter writer;
12972 int ret;
Eric Smith4a7d76d2008-05-30 18:10:19 +000012973
12974 if (!PyArg_ParseTuple(args, "U:__format__", &format_spec))
12975 return NULL;
12976
Victor Stinnerd3f08822012-05-29 12:57:52 +020012977 if (PyUnicode_READY(self) == -1)
12978 return NULL;
12979 _PyUnicodeWriter_Init(&writer, 0);
12980 ret = _PyUnicode_FormatAdvancedWriter(&writer,
12981 self, format_spec, 0,
12982 PyUnicode_GET_LENGTH(format_spec));
12983 if (ret == -1) {
12984 _PyUnicodeWriter_Dealloc(&writer);
12985 return NULL;
12986 }
12987 return _PyUnicodeWriter_Finish(&writer);
Eric Smith4a7d76d2008-05-30 18:10:19 +000012988}
12989
Eric Smith8c663262007-08-25 02:26:07 +000012990PyDoc_STRVAR(p_format__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000012991 "S.__format__(format_spec) -> str\n\
Eric Smith8c663262007-08-25 02:26:07 +000012992\n\
Eric Smith51d2fd92010-11-06 19:27:37 +000012993Return a formatted version of S as described by format_spec.");
Eric Smith8c663262007-08-25 02:26:07 +000012994
12995static PyObject *
Victor Stinner9db1a8b2011-10-23 20:04:37 +020012996unicode__sizeof__(PyObject *v)
Georg Brandlc28e1fa2008-06-10 19:20:26 +000012997{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020012998 Py_ssize_t size;
12999
13000 /* If it's a compact object, account for base structure +
13001 character data. */
13002 if (PyUnicode_IS_COMPACT_ASCII(v))
13003 size = sizeof(PyASCIIObject) + PyUnicode_GET_LENGTH(v) + 1;
13004 else if (PyUnicode_IS_COMPACT(v))
13005 size = sizeof(PyCompactUnicodeObject) +
Martin v. Löwisc47adb02011-10-07 20:55:35 +020013006 (PyUnicode_GET_LENGTH(v) + 1) * PyUnicode_KIND(v);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013007 else {
13008 /* If it is a two-block object, account for base object, and
13009 for character block if present. */
13010 size = sizeof(PyUnicodeObject);
Victor Stinnerc3c74152011-10-02 20:39:55 +020013011 if (_PyUnicode_DATA_ANY(v))
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013012 size += (PyUnicode_GET_LENGTH(v) + 1) *
Martin v. Löwisc47adb02011-10-07 20:55:35 +020013013 PyUnicode_KIND(v);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013014 }
13015 /* If the wstr pointer is present, account for it unless it is shared
Victor Stinnera3be6132011-10-03 02:16:37 +020013016 with the data pointer. Check if the data is not shared. */
Victor Stinner03490912011-10-03 23:45:12 +020013017 if (_PyUnicode_HAS_WSTR_MEMORY(v))
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013018 size += (PyUnicode_WSTR_LENGTH(v) + 1) * sizeof(wchar_t);
Victor Stinner829c0ad2011-10-03 01:08:02 +020013019 if (_PyUnicode_HAS_UTF8_MEMORY(v))
Victor Stinnere90fe6a2011-10-01 16:48:13 +020013020 size += PyUnicode_UTF8_LENGTH(v) + 1;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013021
13022 return PyLong_FromSsize_t(size);
Georg Brandlc28e1fa2008-06-10 19:20:26 +000013023}
13024
13025PyDoc_STRVAR(sizeof__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +000013026 "S.__sizeof__() -> size of S in memory, in bytes");
Georg Brandlc28e1fa2008-06-10 19:20:26 +000013027
13028static PyObject *
Victor Stinner034f6cf2011-09-30 02:26:44 +020013029unicode_getnewargs(PyObject *v)
Guido van Rossum5d9113d2003-01-29 17:58:45 +000013030{
Victor Stinnerbf6e5602011-12-12 01:53:47 +010013031 PyObject *copy = _PyUnicode_Copy(v);
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013032 if (!copy)
13033 return NULL;
13034 return Py_BuildValue("(N)", copy);
Guido van Rossum5d9113d2003-01-29 17:58:45 +000013035}
13036
Guido van Rossumd57fd912000-03-10 22:53:23 +000013037static PyMethodDef unicode_methods[] = {
Benjamin Peterson28a4dce2010-12-12 01:33:04 +000013038 {"encode", (PyCFunction) unicode_encode, METH_VARARGS | METH_KEYWORDS, encode__doc__},
Martin v. Löwise3eb1f22001-08-16 13:15:00 +000013039 {"replace", (PyCFunction) unicode_replace, METH_VARARGS, replace__doc__},
Ezio Melotticda6b6d2012-02-26 09:39:55 +020013040 {"split", (PyCFunction) unicode_split, METH_VARARGS | METH_KEYWORDS, split__doc__},
13041 {"rsplit", (PyCFunction) unicode_rsplit, METH_VARARGS | METH_KEYWORDS, rsplit__doc__},
Martin v. Löwise3eb1f22001-08-16 13:15:00 +000013042 {"join", (PyCFunction) unicode_join, METH_O, join__doc__},
13043 {"capitalize", (PyCFunction) unicode_capitalize, METH_NOARGS, capitalize__doc__},
Benjamin Petersond5890c82012-01-14 13:23:30 -050013044 {"casefold", (PyCFunction) unicode_casefold, METH_NOARGS, casefold__doc__},
Martin v. Löwise3eb1f22001-08-16 13:15:00 +000013045 {"title", (PyCFunction) unicode_title, METH_NOARGS, title__doc__},
13046 {"center", (PyCFunction) unicode_center, METH_VARARGS, center__doc__},
13047 {"count", (PyCFunction) unicode_count, METH_VARARGS, count__doc__},
13048 {"expandtabs", (PyCFunction) unicode_expandtabs, METH_VARARGS, expandtabs__doc__},
13049 {"find", (PyCFunction) unicode_find, METH_VARARGS, find__doc__},
Thomas Wouters477c8d52006-05-27 19:21:47 +000013050 {"partition", (PyCFunction) unicode_partition, METH_O, partition__doc__},
Martin v. Löwise3eb1f22001-08-16 13:15:00 +000013051 {"index", (PyCFunction) unicode_index, METH_VARARGS, index__doc__},
13052 {"ljust", (PyCFunction) unicode_ljust, METH_VARARGS, ljust__doc__},
13053 {"lower", (PyCFunction) unicode_lower, METH_NOARGS, lower__doc__},
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000013054 {"lstrip", (PyCFunction) unicode_lstrip, METH_VARARGS, lstrip__doc__},
Martin v. Löwise3eb1f22001-08-16 13:15:00 +000013055 {"rfind", (PyCFunction) unicode_rfind, METH_VARARGS, rfind__doc__},
13056 {"rindex", (PyCFunction) unicode_rindex, METH_VARARGS, rindex__doc__},
13057 {"rjust", (PyCFunction) unicode_rjust, METH_VARARGS, rjust__doc__},
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000013058 {"rstrip", (PyCFunction) unicode_rstrip, METH_VARARGS, rstrip__doc__},
Thomas Wouters477c8d52006-05-27 19:21:47 +000013059 {"rpartition", (PyCFunction) unicode_rpartition, METH_O, rpartition__doc__},
Mark Dickinson0d5f6ad2011-09-24 09:14:39 +010013060 {"splitlines", (PyCFunction) unicode_splitlines, METH_VARARGS | METH_KEYWORDS, splitlines__doc__},
Walter Dörwaldde02bcb2002-04-22 17:42:37 +000013061 {"strip", (PyCFunction) unicode_strip, METH_VARARGS, strip__doc__},
Martin v. Löwise3eb1f22001-08-16 13:15:00 +000013062 {"swapcase", (PyCFunction) unicode_swapcase, METH_NOARGS, swapcase__doc__},
13063 {"translate", (PyCFunction) unicode_translate, METH_O, translate__doc__},
13064 {"upper", (PyCFunction) unicode_upper, METH_NOARGS, upper__doc__},
13065 {"startswith", (PyCFunction) unicode_startswith, METH_VARARGS, startswith__doc__},
13066 {"endswith", (PyCFunction) unicode_endswith, METH_VARARGS, endswith__doc__},
13067 {"islower", (PyCFunction) unicode_islower, METH_NOARGS, islower__doc__},
13068 {"isupper", (PyCFunction) unicode_isupper, METH_NOARGS, isupper__doc__},
13069 {"istitle", (PyCFunction) unicode_istitle, METH_NOARGS, istitle__doc__},
13070 {"isspace", (PyCFunction) unicode_isspace, METH_NOARGS, isspace__doc__},
13071 {"isdecimal", (PyCFunction) unicode_isdecimal, METH_NOARGS, isdecimal__doc__},
13072 {"isdigit", (PyCFunction) unicode_isdigit, METH_NOARGS, isdigit__doc__},
13073 {"isnumeric", (PyCFunction) unicode_isnumeric, METH_NOARGS, isnumeric__doc__},
13074 {"isalpha", (PyCFunction) unicode_isalpha, METH_NOARGS, isalpha__doc__},
13075 {"isalnum", (PyCFunction) unicode_isalnum, METH_NOARGS, isalnum__doc__},
Martin v. Löwis47383402007-08-15 07:32:56 +000013076 {"isidentifier", (PyCFunction) unicode_isidentifier, METH_NOARGS, isidentifier__doc__},
Georg Brandl559e5d72008-06-11 18:37:52 +000013077 {"isprintable", (PyCFunction) unicode_isprintable, METH_NOARGS, isprintable__doc__},
Martin v. Löwise3eb1f22001-08-16 13:15:00 +000013078 {"zfill", (PyCFunction) unicode_zfill, METH_VARARGS, zfill__doc__},
Eric Smith9cd1e092007-08-31 18:39:38 +000013079 {"format", (PyCFunction) do_string_format, METH_VARARGS | METH_KEYWORDS, format__doc__},
Eric Smith27bbca62010-11-04 17:06:58 +000013080 {"format_map", (PyCFunction) do_string_format_map, METH_O, format_map__doc__},
Eric Smith4a7d76d2008-05-30 18:10:19 +000013081 {"__format__", (PyCFunction) unicode__format__, METH_VARARGS, p_format__doc__},
Georg Brandlceee0772007-11-27 23:48:05 +000013082 {"maketrans", (PyCFunction) unicode_maketrans,
13083 METH_VARARGS | METH_STATIC, maketrans__doc__},
Georg Brandlc28e1fa2008-06-10 19:20:26 +000013084 {"__sizeof__", (PyCFunction) unicode__sizeof__, METH_NOARGS, sizeof__doc__},
Walter Dörwald068325e2002-04-15 13:36:47 +000013085#if 0
Alexander Belopolsky942af5a2010-12-04 03:38:46 +000013086 /* These methods are just used for debugging the implementation. */
Alexander Belopolsky942af5a2010-12-04 03:38:46 +000013087 {"_decimal2ascii", (PyCFunction) unicode__decimal2ascii, METH_NOARGS},
Guido van Rossumd57fd912000-03-10 22:53:23 +000013088#endif
13089
Benjamin Peterson14339b62009-01-31 16:36:08 +000013090 {"__getnewargs__", (PyCFunction)unicode_getnewargs, METH_NOARGS},
Guido van Rossumd57fd912000-03-10 22:53:23 +000013091 {NULL, NULL}
13092};
13093
Neil Schemenauerce30bc92002-11-18 16:10:18 +000013094static PyObject *
13095unicode_mod(PyObject *v, PyObject *w)
13096{
Brian Curtindfc80e32011-08-10 20:28:54 -050013097 if (!PyUnicode_Check(v))
13098 Py_RETURN_NOTIMPLEMENTED;
Benjamin Peterson29060642009-01-31 22:14:21 +000013099 return PyUnicode_Format(v, w);
Neil Schemenauerce30bc92002-11-18 16:10:18 +000013100}
13101
13102static PyNumberMethods unicode_as_number = {
Benjamin Peterson14339b62009-01-31 16:36:08 +000013103 0, /*nb_add*/
13104 0, /*nb_subtract*/
13105 0, /*nb_multiply*/
13106 unicode_mod, /*nb_remainder*/
Neil Schemenauerce30bc92002-11-18 16:10:18 +000013107};
13108
Guido van Rossumd57fd912000-03-10 22:53:23 +000013109static PySequenceMethods unicode_as_sequence = {
Benjamin Peterson14339b62009-01-31 16:36:08 +000013110 (lenfunc) unicode_length, /* sq_length */
13111 PyUnicode_Concat, /* sq_concat */
13112 (ssizeargfunc) unicode_repeat, /* sq_repeat */
13113 (ssizeargfunc) unicode_getitem, /* sq_item */
13114 0, /* sq_slice */
13115 0, /* sq_ass_item */
13116 0, /* sq_ass_slice */
13117 PyUnicode_Contains, /* sq_contains */
Guido van Rossumd57fd912000-03-10 22:53:23 +000013118};
13119
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +000013120static PyObject*
Victor Stinner9db1a8b2011-10-23 20:04:37 +020013121unicode_subscript(PyObject* self, PyObject* item)
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +000013122{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013123 if (PyUnicode_READY(self) == -1)
13124 return NULL;
13125
Thomas Wouters00ee7ba2006-08-21 19:07:27 +000013126 if (PyIndex_Check(item)) {
13127 Py_ssize_t i = PyNumber_AsSsize_t(item, PyExc_IndexError);
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +000013128 if (i == -1 && PyErr_Occurred())
13129 return NULL;
13130 if (i < 0)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013131 i += PyUnicode_GET_LENGTH(self);
Victor Stinner9db1a8b2011-10-23 20:04:37 +020013132 return unicode_getitem(self, i);
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +000013133 } else if (PySlice_Check(item)) {
Martin v. Löwis18e16552006-02-15 17:27:45 +000013134 Py_ssize_t start, stop, step, slicelength, cur, i;
Antoine Pitrou7aec4012011-10-04 19:08:01 +020013135 PyObject *result;
13136 void *src_data, *dest_data;
Antoine Pitrou875f29b2011-10-04 20:00:49 +020013137 int src_kind, dest_kind;
Victor Stinnerc80d6d22011-10-05 14:13:28 +020013138 Py_UCS4 ch, max_char, kind_limit;
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +000013139
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013140 if (PySlice_GetIndicesEx(item, PyUnicode_GET_LENGTH(self),
Benjamin Peterson29060642009-01-31 22:14:21 +000013141 &start, &stop, &step, &slicelength) < 0) {
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +000013142 return NULL;
13143 }
13144
13145 if (slicelength <= 0) {
Victor Stinner382955f2011-12-11 21:44:00 +010013146 Py_INCREF(unicode_empty);
13147 return unicode_empty;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013148 } else if (start == 0 && step == 1 &&
Victor Stinnerc4b49542011-12-11 22:44:26 +010013149 slicelength == PyUnicode_GET_LENGTH(self)) {
13150 return unicode_result_unchanged(self);
Thomas Woutersed03b412007-08-28 21:37:11 +000013151 } else if (step == 1) {
Victor Stinner9db1a8b2011-10-23 20:04:37 +020013152 return PyUnicode_Substring(self,
Victor Stinner12bab6d2011-10-01 01:53:49 +020013153 start, start + slicelength);
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +000013154 }
Antoine Pitrou875f29b2011-10-04 20:00:49 +020013155 /* General case */
Antoine Pitrou875f29b2011-10-04 20:00:49 +020013156 src_kind = PyUnicode_KIND(self);
13157 src_data = PyUnicode_DATA(self);
Victor Stinner55c99112011-10-13 01:17:06 +020013158 if (!PyUnicode_IS_ASCII(self)) {
13159 kind_limit = kind_maxchar_limit(src_kind);
13160 max_char = 0;
13161 for (cur = start, i = 0; i < slicelength; cur += step, i++) {
13162 ch = PyUnicode_READ(src_kind, src_data, cur);
13163 if (ch > max_char) {
13164 max_char = ch;
13165 if (max_char >= kind_limit)
13166 break;
13167 }
Victor Stinnerc80d6d22011-10-05 14:13:28 +020013168 }
Antoine Pitrou875f29b2011-10-04 20:00:49 +020013169 }
Victor Stinner55c99112011-10-13 01:17:06 +020013170 else
13171 max_char = 127;
Antoine Pitrou875f29b2011-10-04 20:00:49 +020013172 result = PyUnicode_New(slicelength, max_char);
Antoine Pitrou7aec4012011-10-04 19:08:01 +020013173 if (result == NULL)
13174 return NULL;
Antoine Pitrou875f29b2011-10-04 20:00:49 +020013175 dest_kind = PyUnicode_KIND(result);
Antoine Pitrou7aec4012011-10-04 19:08:01 +020013176 dest_data = PyUnicode_DATA(result);
13177
13178 for (cur = start, i = 0; i < slicelength; cur += step, i++) {
Antoine Pitrou875f29b2011-10-04 20:00:49 +020013179 Py_UCS4 ch = PyUnicode_READ(src_kind, src_data, cur);
13180 PyUnicode_WRITE(dest_kind, dest_data, i, ch);
Antoine Pitrou7aec4012011-10-04 19:08:01 +020013181 }
Victor Stinnerbb10a1f2011-10-05 01:34:17 +020013182 assert(_PyUnicode_CheckConsistency(result, 1));
Antoine Pitrou7aec4012011-10-04 19:08:01 +020013183 return result;
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +000013184 } else {
13185 PyErr_SetString(PyExc_TypeError, "string indices must be integers");
13186 return NULL;
13187 }
13188}
13189
13190static PyMappingMethods unicode_as_mapping = {
Benjamin Peterson14339b62009-01-31 16:36:08 +000013191 (lenfunc)unicode_length, /* mp_length */
13192 (binaryfunc)unicode_subscript, /* mp_subscript */
13193 (objobjargproc)0, /* mp_ass_subscript */
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +000013194};
13195
Guido van Rossumd57fd912000-03-10 22:53:23 +000013196
Guido van Rossumd57fd912000-03-10 22:53:23 +000013197/* Helpers for PyUnicode_Format() */
13198
13199static PyObject *
Martin v. Löwis18e16552006-02-15 17:27:45 +000013200getnextarg(PyObject *args, Py_ssize_t arglen, Py_ssize_t *p_argidx)
Guido van Rossumd57fd912000-03-10 22:53:23 +000013201{
Martin v. Löwis18e16552006-02-15 17:27:45 +000013202 Py_ssize_t argidx = *p_argidx;
Guido van Rossumd57fd912000-03-10 22:53:23 +000013203 if (argidx < arglen) {
Benjamin Peterson29060642009-01-31 22:14:21 +000013204 (*p_argidx)++;
13205 if (arglen < 0)
13206 return args;
13207 else
13208 return PyTuple_GetItem(args, argidx);
Guido van Rossumd57fd912000-03-10 22:53:23 +000013209 }
13210 PyErr_SetString(PyExc_TypeError,
Benjamin Peterson29060642009-01-31 22:14:21 +000013211 "not enough arguments for format string");
Guido van Rossumd57fd912000-03-10 22:53:23 +000013212 return NULL;
13213}
13214
Mark Dickinsonf489caf2009-05-01 11:42:00 +000013215/* Returns a new reference to a PyUnicode object, or NULL on failure. */
Guido van Rossumd57fd912000-03-10 22:53:23 +000013216
Victor Stinnerd3f08822012-05-29 12:57:52 +020013217static int
13218formatfloat(PyObject *v, int flags, int prec, int type,
13219 PyObject **p_output, _PyUnicodeWriter *writer)
Guido van Rossumd57fd912000-03-10 22:53:23 +000013220{
Mark Dickinsonf489caf2009-05-01 11:42:00 +000013221 char *p;
Guido van Rossumd57fd912000-03-10 22:53:23 +000013222 double x;
Victor Stinnerd3f08822012-05-29 12:57:52 +020013223 Py_ssize_t len;
Tim Petersced69f82003-09-16 20:30:58 +000013224
Guido van Rossumd57fd912000-03-10 22:53:23 +000013225 x = PyFloat_AsDouble(v);
13226 if (x == -1.0 && PyErr_Occurred())
Victor Stinnerd3f08822012-05-29 12:57:52 +020013227 return -1;
Mark Dickinsonf489caf2009-05-01 11:42:00 +000013228
Guido van Rossumd57fd912000-03-10 22:53:23 +000013229 if (prec < 0)
Benjamin Peterson29060642009-01-31 22:14:21 +000013230 prec = 6;
Eric Smith0923d1d2009-04-16 20:16:10 +000013231
Eric Smith0923d1d2009-04-16 20:16:10 +000013232 p = PyOS_double_to_string(x, type, prec,
13233 (flags & F_ALT) ? Py_DTSF_ALT : 0, NULL);
Mark Dickinsonf489caf2009-05-01 11:42:00 +000013234 if (p == NULL)
Victor Stinnerd3f08822012-05-29 12:57:52 +020013235 return -1;
13236 len = strlen(p);
13237 if (writer) {
Christian Heimesf4f99392012-09-10 11:48:41 +020013238 if (_PyUnicodeWriter_Prepare(writer, len, 127) == -1) {
13239 PyMem_Free(p);
Victor Stinnerd3f08822012-05-29 12:57:52 +020013240 return -1;
Christian Heimesf4f99392012-09-10 11:48:41 +020013241 }
Victor Stinner184252a2012-06-16 02:57:41 +020013242 unicode_write_cstr(writer->buffer, writer->pos, p, len);
Victor Stinnerd3f08822012-05-29 12:57:52 +020013243 writer->pos += len;
13244 }
13245 else
13246 *p_output = _PyUnicode_FromASCII(p, len);
Eric Smith0923d1d2009-04-16 20:16:10 +000013247 PyMem_Free(p);
Victor Stinnerd3f08822012-05-29 12:57:52 +020013248 return 0;
Guido van Rossumd57fd912000-03-10 22:53:23 +000013249}
13250
Victor Stinnerd0880d52012-04-27 23:40:13 +020013251/* formatlong() emulates the format codes d, u, o, x and X, and
13252 * the F_ALT flag, for Python's long (unbounded) ints. It's not used for
13253 * Python's regular ints.
13254 * Return value: a new PyUnicodeObject*, or NULL if error.
13255 * The output string is of the form
13256 * "-"? ("0x" | "0X")? digit+
13257 * "0x"/"0X" are present only for x and X conversions, with F_ALT
13258 * set in flags. The case of hex digits will be correct,
13259 * There will be at least prec digits, zero-filled on the left if
13260 * necessary to get that many.
13261 * val object to be converted
13262 * flags bitmask of format flags; only F_ALT is looked at
13263 * prec minimum number of digits; 0-fill on left if needed
13264 * type a character in [duoxX]; u acts the same as d
13265 *
13266 * CAUTION: o, x and X conversions on regular ints can never
13267 * produce a '-' sign, but can for Python's unbounded ints.
13268 */
Tim Peters38fd5b62000-09-21 05:43:11 +000013269static PyObject*
13270formatlong(PyObject *val, int flags, int prec, int type)
13271{
Victor Stinnerd0880d52012-04-27 23:40:13 +020013272 PyObject *result = NULL;
Benjamin Peterson14339b62009-01-31 16:36:08 +000013273 char *buf;
Victor Stinnerd0880d52012-04-27 23:40:13 +020013274 Py_ssize_t i;
13275 int sign; /* 1 if '-', else 0 */
13276 int len; /* number of characters */
13277 Py_ssize_t llen;
13278 int numdigits; /* len == numnondigits + numdigits */
13279 int numnondigits = 0;
Tim Peters38fd5b62000-09-21 05:43:11 +000013280
Victor Stinnerd0880d52012-04-27 23:40:13 +020013281 /* Avoid exceeding SSIZE_T_MAX */
13282 if (prec > INT_MAX-3) {
13283 PyErr_SetString(PyExc_OverflowError,
13284 "precision too large");
Benjamin Peterson14339b62009-01-31 16:36:08 +000013285 return NULL;
Victor Stinnerd0880d52012-04-27 23:40:13 +020013286 }
13287
13288 assert(PyLong_Check(val));
13289
13290 switch (type) {
13291 case 'd':
13292 case 'u':
13293 /* Special-case boolean: we want 0/1 */
Victor Stinnerb11d91d2012-04-28 00:25:34 +020013294 if (PyBool_Check(val))
13295 result = PyNumber_ToBase(val, 10);
13296 else
13297 result = Py_TYPE(val)->tp_str(val);
Victor Stinnerd0880d52012-04-27 23:40:13 +020013298 break;
13299 case 'o':
13300 numnondigits = 2;
13301 result = PyNumber_ToBase(val, 8);
13302 break;
13303 case 'x':
13304 case 'X':
13305 numnondigits = 2;
13306 result = PyNumber_ToBase(val, 16);
13307 break;
13308 default:
13309 assert(!"'type' not in [duoxX]");
13310 }
13311 if (!result)
13312 return NULL;
13313
13314 assert(unicode_modifiable(result));
13315 assert(PyUnicode_IS_READY(result));
13316 assert(PyUnicode_IS_ASCII(result));
13317
13318 /* To modify the string in-place, there can only be one reference. */
13319 if (Py_REFCNT(result) != 1) {
13320 PyErr_BadInternalCall();
13321 return NULL;
13322 }
13323 buf = PyUnicode_DATA(result);
13324 llen = PyUnicode_GET_LENGTH(result);
13325 if (llen > INT_MAX) {
13326 PyErr_SetString(PyExc_ValueError,
13327 "string too large in _PyBytes_FormatLong");
13328 return NULL;
13329 }
13330 len = (int)llen;
13331 sign = buf[0] == '-';
13332 numnondigits += sign;
13333 numdigits = len - numnondigits;
13334 assert(numdigits > 0);
13335
13336 /* Get rid of base marker unless F_ALT */
13337 if (((flags & F_ALT) == 0 &&
13338 (type == 'o' || type == 'x' || type == 'X'))) {
13339 assert(buf[sign] == '0');
13340 assert(buf[sign+1] == 'x' || buf[sign+1] == 'X' ||
13341 buf[sign+1] == 'o');
13342 numnondigits -= 2;
13343 buf += 2;
13344 len -= 2;
13345 if (sign)
13346 buf[0] = '-';
13347 assert(len == numnondigits + numdigits);
13348 assert(numdigits > 0);
13349 }
13350
13351 /* Fill with leading zeroes to meet minimum width. */
13352 if (prec > numdigits) {
13353 PyObject *r1 = PyBytes_FromStringAndSize(NULL,
13354 numnondigits + prec);
13355 char *b1;
13356 if (!r1) {
13357 Py_DECREF(result);
13358 return NULL;
13359 }
13360 b1 = PyBytes_AS_STRING(r1);
13361 for (i = 0; i < numnondigits; ++i)
13362 *b1++ = *buf++;
13363 for (i = 0; i < prec - numdigits; i++)
13364 *b1++ = '0';
13365 for (i = 0; i < numdigits; i++)
13366 *b1++ = *buf++;
13367 *b1 = '\0';
13368 Py_DECREF(result);
13369 result = r1;
13370 buf = PyBytes_AS_STRING(result);
13371 len = numnondigits + prec;
13372 }
13373
13374 /* Fix up case for hex conversions. */
13375 if (type == 'X') {
13376 /* Need to convert all lower case letters to upper case.
13377 and need to convert 0x to 0X (and -0x to -0X). */
13378 for (i = 0; i < len; i++)
13379 if (buf[i] >= 'a' && buf[i] <= 'x')
13380 buf[i] -= 'a'-'A';
13381 }
13382 if (!PyUnicode_Check(result) || len != PyUnicode_GET_LENGTH(result)) {
13383 PyObject *unicode;
Victor Stinnerd3f08822012-05-29 12:57:52 +020013384 unicode = _PyUnicode_FromASCII(buf, len);
Victor Stinnerd0880d52012-04-27 23:40:13 +020013385 Py_DECREF(result);
13386 result = unicode;
13387 }
Benjamin Peterson14339b62009-01-31 16:36:08 +000013388 return result;
Tim Peters38fd5b62000-09-21 05:43:11 +000013389}
13390
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013391static Py_UCS4
13392formatchar(PyObject *v)
Guido van Rossumd57fd912000-03-10 22:53:23 +000013393{
Amaury Forgeot d'Arca4db6862008-07-04 21:26:43 +000013394 /* presume that the buffer is at least 3 characters long */
Marc-André Lemburgd4ab4a52000-06-08 17:54:00 +000013395 if (PyUnicode_Check(v)) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013396 if (PyUnicode_GET_LENGTH(v) == 1) {
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013397 return PyUnicode_READ_CHAR(v, 0);
Benjamin Peterson29060642009-01-31 22:14:21 +000013398 }
Benjamin Peterson29060642009-01-31 22:14:21 +000013399 goto onError;
13400 }
13401 else {
13402 /* Integer input truncated to a character */
13403 long x;
13404 x = PyLong_AsLong(v);
13405 if (x == -1 && PyErr_Occurred())
13406 goto onError;
13407
Victor Stinner8faf8212011-12-08 22:14:11 +010013408 if (x < 0 || x > MAX_UNICODE) {
Benjamin Peterson29060642009-01-31 22:14:21 +000013409 PyErr_SetString(PyExc_OverflowError,
13410 "%c arg not in range(0x110000)");
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013411 return (Py_UCS4) -1;
Benjamin Peterson29060642009-01-31 22:14:21 +000013412 }
13413
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013414 return (Py_UCS4) x;
Benjamin Peterson14339b62009-01-31 16:36:08 +000013415 }
Amaury Forgeot d'Arca4db6862008-07-04 21:26:43 +000013416
Benjamin Peterson29060642009-01-31 22:14:21 +000013417 onError:
Marc-André Lemburgd4ab4a52000-06-08 17:54:00 +000013418 PyErr_SetString(PyExc_TypeError,
Benjamin Peterson29060642009-01-31 22:14:21 +000013419 "%c requires int or char");
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013420 return (Py_UCS4) -1;
Guido van Rossumd57fd912000-03-10 22:53:23 +000013421}
13422
Alexander Belopolsky40018472011-02-26 01:02:56 +000013423PyObject *
13424PyUnicode_Format(PyObject *format, PyObject *args)
Guido van Rossumd57fd912000-03-10 22:53:23 +000013425{
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013426 Py_ssize_t fmtcnt, fmtpos, arglen, argidx;
Guido van Rossumd57fd912000-03-10 22:53:23 +000013427 int args_owned = 0;
Guido van Rossumd57fd912000-03-10 22:53:23 +000013428 PyObject *dict = NULL;
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013429 PyObject *temp = NULL;
13430 PyObject *second = NULL;
Victor Stinner9db1a8b2011-10-23 20:04:37 +020013431 PyObject *uformat;
Victor Stinnerf2c76aa2012-05-03 13:10:40 +020013432 void *fmt;
13433 enum PyUnicode_Kind kind, fmtkind;
Victor Stinner3b1a74a2012-05-09 22:25:00 +020013434 _PyUnicodeWriter writer;
Victor Stinneree4544c2012-05-09 22:24:08 +020013435 Py_ssize_t sublen;
13436 Py_UCS4 maxchar;
Tim Petersced69f82003-09-16 20:30:58 +000013437
Guido van Rossumd57fd912000-03-10 22:53:23 +000013438 if (format == NULL || args == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +000013439 PyErr_BadInternalCall();
13440 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +000013441 }
Victor Stinner9db1a8b2011-10-23 20:04:37 +020013442 uformat = PyUnicode_FromObject(format);
Benjamin Peterson22a29702012-01-02 09:00:30 -060013443 if (uformat == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +000013444 return NULL;
Benjamin Peterson22a29702012-01-02 09:00:30 -060013445 if (PyUnicode_READY(uformat) == -1)
13446 Py_DECREF(uformat);
Victor Stinnerf2c76aa2012-05-03 13:10:40 +020013447
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013448 fmt = PyUnicode_DATA(uformat);
13449 fmtkind = PyUnicode_KIND(uformat);
13450 fmtcnt = PyUnicode_GET_LENGTH(uformat);
13451 fmtpos = 0;
Guido van Rossumd57fd912000-03-10 22:53:23 +000013452
Victor Stinnerd3f08822012-05-29 12:57:52 +020013453 _PyUnicodeWriter_Init(&writer, fmtcnt + 100);
Victor Stinnerf2c76aa2012-05-03 13:10:40 +020013454
Guido van Rossumd57fd912000-03-10 22:53:23 +000013455 if (PyTuple_Check(args)) {
Benjamin Peterson29060642009-01-31 22:14:21 +000013456 arglen = PyTuple_Size(args);
13457 argidx = 0;
Guido van Rossumd57fd912000-03-10 22:53:23 +000013458 }
13459 else {
Benjamin Peterson29060642009-01-31 22:14:21 +000013460 arglen = -1;
13461 argidx = -2;
Guido van Rossumd57fd912000-03-10 22:53:23 +000013462 }
Benjamin Peterson28a6cfa2012-08-28 17:55:35 -040013463 if (PyMapping_Check(args) && !PyTuple_Check(args) && !PyUnicode_Check(args))
Benjamin Peterson29060642009-01-31 22:14:21 +000013464 dict = args;
Guido van Rossumd57fd912000-03-10 22:53:23 +000013465
13466 while (--fmtcnt >= 0) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013467 if (PyUnicode_READ(fmtkind, fmt, fmtpos) != '%') {
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013468 Py_ssize_t nonfmtpos;
13469 nonfmtpos = fmtpos++;
13470 while (fmtcnt >= 0 &&
13471 PyUnicode_READ(fmtkind, fmt, fmtpos) != '%') {
13472 fmtpos++;
13473 fmtcnt--;
Benjamin Peterson14339b62009-01-31 16:36:08 +000013474 }
Victor Stinnerf2c76aa2012-05-03 13:10:40 +020013475 if (fmtcnt < 0)
13476 fmtpos--;
Victor Stinneree4544c2012-05-09 22:24:08 +020013477 sublen = fmtpos - nonfmtpos;
13478 maxchar = _PyUnicode_FindMaxChar(uformat,
13479 nonfmtpos, nonfmtpos + sublen);
Victor Stinner3b1a74a2012-05-09 22:25:00 +020013480 if (_PyUnicodeWriter_Prepare(&writer, sublen, maxchar) == -1)
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013481 goto onError;
Victor Stinneree4544c2012-05-09 22:24:08 +020013482
Victor Stinnerd3f08822012-05-29 12:57:52 +020013483 _PyUnicode_FastCopyCharacters(writer.buffer, writer.pos,
13484 uformat, nonfmtpos, sublen);
Victor Stinneree4544c2012-05-09 22:24:08 +020013485 writer.pos += sublen;
Benjamin Peterson14339b62009-01-31 16:36:08 +000013486 }
13487 else {
Benjamin Peterson29060642009-01-31 22:14:21 +000013488 /* Got a format specifier */
13489 int flags = 0;
13490 Py_ssize_t width = -1;
13491 int prec = -1;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013492 Py_UCS4 c = '\0';
Victor Stinnerf2c76aa2012-05-03 13:10:40 +020013493 Py_UCS4 fill;
13494 int sign;
13495 Py_UCS4 signchar;
Benjamin Peterson29060642009-01-31 22:14:21 +000013496 int isnumok;
13497 PyObject *v = NULL;
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013498 void *pbuf = NULL;
13499 Py_ssize_t pindex, len;
Victor Stinneree4544c2012-05-09 22:24:08 +020013500 Py_UCS4 bufmaxchar;
13501 Py_ssize_t buflen;
Guido van Rossumd57fd912000-03-10 22:53:23 +000013502
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013503 fmtpos++;
Victor Stinner438106b2012-05-02 00:41:57 +020013504 c = PyUnicode_READ(fmtkind, fmt, fmtpos);
13505 if (c == '(') {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013506 Py_ssize_t keystart;
Benjamin Peterson29060642009-01-31 22:14:21 +000013507 Py_ssize_t keylen;
13508 PyObject *key;
13509 int pcount = 1;
Christian Heimesa612dc02008-02-24 13:08:18 +000013510
Benjamin Peterson29060642009-01-31 22:14:21 +000013511 if (dict == NULL) {
13512 PyErr_SetString(PyExc_TypeError,
13513 "format requires a mapping");
13514 goto onError;
13515 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013516 ++fmtpos;
Benjamin Peterson29060642009-01-31 22:14:21 +000013517 --fmtcnt;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013518 keystart = fmtpos;
Benjamin Peterson29060642009-01-31 22:14:21 +000013519 /* Skip over balanced parentheses */
13520 while (pcount > 0 && --fmtcnt >= 0) {
Victor Stinnerbff7c962012-05-03 01:44:59 +020013521 c = PyUnicode_READ(fmtkind, fmt, fmtpos);
13522 if (c == ')')
Benjamin Peterson29060642009-01-31 22:14:21 +000013523 --pcount;
Victor Stinnerbff7c962012-05-03 01:44:59 +020013524 else if (c == '(')
Benjamin Peterson29060642009-01-31 22:14:21 +000013525 ++pcount;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013526 fmtpos++;
Benjamin Peterson29060642009-01-31 22:14:21 +000013527 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013528 keylen = fmtpos - keystart - 1;
Benjamin Peterson29060642009-01-31 22:14:21 +000013529 if (fmtcnt < 0 || pcount > 0) {
13530 PyErr_SetString(PyExc_ValueError,
13531 "incomplete format key");
13532 goto onError;
13533 }
Victor Stinner7931d9a2011-11-04 00:22:48 +010013534 key = PyUnicode_Substring(uformat,
Victor Stinner12bab6d2011-10-01 01:53:49 +020013535 keystart, keystart + keylen);
Benjamin Peterson29060642009-01-31 22:14:21 +000013536 if (key == NULL)
13537 goto onError;
13538 if (args_owned) {
13539 Py_DECREF(args);
13540 args_owned = 0;
13541 }
13542 args = PyObject_GetItem(dict, key);
13543 Py_DECREF(key);
13544 if (args == NULL) {
13545 goto onError;
13546 }
13547 args_owned = 1;
13548 arglen = -1;
13549 argidx = -2;
Benjamin Peterson14339b62009-01-31 16:36:08 +000013550 }
Benjamin Peterson29060642009-01-31 22:14:21 +000013551 while (--fmtcnt >= 0) {
Victor Stinner438106b2012-05-02 00:41:57 +020013552 c = PyUnicode_READ(fmtkind, fmt, fmtpos++);
13553 switch (c) {
Benjamin Peterson29060642009-01-31 22:14:21 +000013554 case '-': flags |= F_LJUST; continue;
13555 case '+': flags |= F_SIGN; continue;
13556 case ' ': flags |= F_BLANK; continue;
13557 case '#': flags |= F_ALT; continue;
13558 case '0': flags |= F_ZERO; continue;
13559 }
13560 break;
Benjamin Peterson14339b62009-01-31 16:36:08 +000013561 }
Benjamin Peterson29060642009-01-31 22:14:21 +000013562 if (c == '*') {
13563 v = getnextarg(args, arglen, &argidx);
13564 if (v == NULL)
13565 goto onError;
13566 if (!PyLong_Check(v)) {
13567 PyErr_SetString(PyExc_TypeError,
13568 "* wants int");
13569 goto onError;
13570 }
13571 width = PyLong_AsLong(v);
13572 if (width == -1 && PyErr_Occurred())
13573 goto onError;
13574 if (width < 0) {
13575 flags |= F_LJUST;
13576 width = -width;
13577 }
13578 if (--fmtcnt >= 0)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013579 c = PyUnicode_READ(fmtkind, fmt, fmtpos++);
Benjamin Peterson29060642009-01-31 22:14:21 +000013580 }
13581 else if (c >= '0' && c <= '9') {
13582 width = c - '0';
13583 while (--fmtcnt >= 0) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013584 c = PyUnicode_READ(fmtkind, fmt, fmtpos++);
Benjamin Peterson29060642009-01-31 22:14:21 +000013585 if (c < '0' || c > '9')
13586 break;
Martin v. Löwisb05c0732012-05-15 13:45:49 +020013587 /* Since c is unsigned, the RHS would end up as unsigned,
13588 mixing signed and unsigned comparison. Since c is between
13589 '0' and '9', casting to int is safe. */
13590 if (width > (PY_SSIZE_T_MAX - ((int)c - '0')) / 10) {
Benjamin Peterson29060642009-01-31 22:14:21 +000013591 PyErr_SetString(PyExc_ValueError,
13592 "width too big");
Benjamin Peterson14339b62009-01-31 16:36:08 +000013593 goto onError;
Benjamin Peterson29060642009-01-31 22:14:21 +000013594 }
13595 width = width*10 + (c - '0');
13596 }
13597 }
13598 if (c == '.') {
13599 prec = 0;
13600 if (--fmtcnt >= 0)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013601 c = PyUnicode_READ(fmtkind, fmt, fmtpos++);
Benjamin Peterson29060642009-01-31 22:14:21 +000013602 if (c == '*') {
13603 v = getnextarg(args, arglen, &argidx);
13604 if (v == NULL)
13605 goto onError;
13606 if (!PyLong_Check(v)) {
13607 PyErr_SetString(PyExc_TypeError,
13608 "* wants int");
13609 goto onError;
13610 }
13611 prec = PyLong_AsLong(v);
13612 if (prec == -1 && PyErr_Occurred())
13613 goto onError;
13614 if (prec < 0)
13615 prec = 0;
13616 if (--fmtcnt >= 0)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013617 c = PyUnicode_READ(fmtkind, fmt, fmtpos++);
Benjamin Peterson29060642009-01-31 22:14:21 +000013618 }
13619 else if (c >= '0' && c <= '9') {
13620 prec = c - '0';
13621 while (--fmtcnt >= 0) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013622 c = PyUnicode_READ(fmtkind, fmt, fmtpos++);
Benjamin Peterson29060642009-01-31 22:14:21 +000013623 if (c < '0' || c > '9')
13624 break;
Martin v. Löwisb05c0732012-05-15 13:45:49 +020013625 if (prec > (INT_MAX - ((int)c - '0')) / 10) {
Benjamin Peterson29060642009-01-31 22:14:21 +000013626 PyErr_SetString(PyExc_ValueError,
13627 "prec too big");
13628 goto onError;
13629 }
13630 prec = prec*10 + (c - '0');
13631 }
13632 }
13633 } /* prec */
13634 if (fmtcnt >= 0) {
13635 if (c == 'h' || c == 'l' || c == 'L') {
13636 if (--fmtcnt >= 0)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013637 c = PyUnicode_READ(fmtkind, fmt, fmtpos++);
Benjamin Peterson29060642009-01-31 22:14:21 +000013638 }
13639 }
13640 if (fmtcnt < 0) {
13641 PyErr_SetString(PyExc_ValueError,
13642 "incomplete format");
13643 goto onError;
13644 }
Victor Stinnerd3f08822012-05-29 12:57:52 +020013645 if (fmtcnt == 0)
Victor Stinnerd7b7c742012-06-04 22:52:12 +020013646 writer.overallocate = 0;
Victor Stinneraff3cc62012-04-30 05:19:21 +020013647
13648 if (c == '%') {
Victor Stinner3b1a74a2012-05-09 22:25:00 +020013649 if (_PyUnicodeWriter_Prepare(&writer, 1, '%') == -1)
Victor Stinnerf2c76aa2012-05-03 13:10:40 +020013650 goto onError;
Victor Stinneree4544c2012-05-09 22:24:08 +020013651 PyUnicode_WRITE(writer.kind, writer.data, writer.pos, '%');
13652 writer.pos += 1;
Victor Stinneraff3cc62012-04-30 05:19:21 +020013653 continue;
Benjamin Peterson29060642009-01-31 22:14:21 +000013654 }
Victor Stinneraff3cc62012-04-30 05:19:21 +020013655
Victor Stinneraff3cc62012-04-30 05:19:21 +020013656 v = getnextarg(args, arglen, &argidx);
13657 if (v == NULL)
13658 goto onError;
13659
Benjamin Peterson29060642009-01-31 22:14:21 +000013660 sign = 0;
Victor Stinnerf2c76aa2012-05-03 13:10:40 +020013661 signchar = '\0';
Benjamin Peterson29060642009-01-31 22:14:21 +000013662 fill = ' ';
13663 switch (c) {
13664
Benjamin Peterson29060642009-01-31 22:14:21 +000013665 case 's':
13666 case 'r':
13667 case 'a':
Victor Stinnerd3f08822012-05-29 12:57:52 +020013668 if (PyLong_CheckExact(v) && width == -1 && prec == -1) {
13669 /* Fast path */
13670 if (_PyLong_FormatWriter(&writer, v, 10, flags & F_ALT) == -1)
13671 goto onError;
13672 goto nextarg;
13673 }
13674
Victor Stinner808fc0a2010-03-22 12:50:40 +000013675 if (PyUnicode_CheckExact(v) && c == 's') {
Benjamin Peterson29060642009-01-31 22:14:21 +000013676 temp = v;
13677 Py_INCREF(temp);
Benjamin Peterson14339b62009-01-31 16:36:08 +000013678 }
13679 else {
Benjamin Peterson29060642009-01-31 22:14:21 +000013680 if (c == 's')
13681 temp = PyObject_Str(v);
13682 else if (c == 'r')
13683 temp = PyObject_Repr(v);
13684 else
13685 temp = PyObject_ASCII(v);
Benjamin Peterson29060642009-01-31 22:14:21 +000013686 }
Benjamin Peterson29060642009-01-31 22:14:21 +000013687 break;
13688
13689 case 'i':
13690 case 'd':
13691 case 'u':
13692 case 'o':
13693 case 'x':
13694 case 'X':
Victor Stinnerd3f08822012-05-29 12:57:52 +020013695 if (PyLong_CheckExact(v)
13696 && width == -1 && prec == -1
13697 && !(flags & (F_SIGN | F_BLANK)))
13698 {
13699 /* Fast path */
13700 switch(c)
13701 {
13702 case 'd':
13703 case 'i':
13704 case 'u':
13705 if (_PyLong_FormatWriter(&writer, v, 10, flags & F_ALT) == -1)
13706 goto onError;
13707 goto nextarg;
13708 case 'x':
13709 if (_PyLong_FormatWriter(&writer, v, 16, flags & F_ALT) == -1)
13710 goto onError;
13711 goto nextarg;
13712 case 'o':
13713 if (_PyLong_FormatWriter(&writer, v, 8, flags & F_ALT) == -1)
13714 goto onError;
13715 goto nextarg;
13716 default:
13717 break;
13718 }
13719 }
13720
Benjamin Peterson29060642009-01-31 22:14:21 +000013721 isnumok = 0;
13722 if (PyNumber_Check(v)) {
13723 PyObject *iobj=NULL;
13724
13725 if (PyLong_Check(v)) {
13726 iobj = v;
13727 Py_INCREF(iobj);
13728 }
13729 else {
13730 iobj = PyNumber_Long(v);
13731 }
13732 if (iobj!=NULL) {
13733 if (PyLong_Check(iobj)) {
13734 isnumok = 1;
Victor Stinneraff3cc62012-04-30 05:19:21 +020013735 sign = 1;
Senthil Kumaran9ebe08d2011-07-03 21:03:16 -070013736 temp = formatlong(iobj, flags, prec, (c == 'i'? 'd': c));
Benjamin Peterson29060642009-01-31 22:14:21 +000013737 Py_DECREF(iobj);
Benjamin Peterson29060642009-01-31 22:14:21 +000013738 }
13739 else {
13740 Py_DECREF(iobj);
13741 }
13742 }
13743 }
13744 if (!isnumok) {
13745 PyErr_Format(PyExc_TypeError,
13746 "%%%c format: a number is required, "
13747 "not %.200s", (char)c, Py_TYPE(v)->tp_name);
13748 goto onError;
13749 }
Victor Stinnerf2c76aa2012-05-03 13:10:40 +020013750 if (flags & F_ZERO)
Benjamin Peterson29060642009-01-31 22:14:21 +000013751 fill = '0';
13752 break;
13753
13754 case 'e':
13755 case 'E':
13756 case 'f':
13757 case 'F':
13758 case 'g':
13759 case 'G':
Victor Stinnerd3f08822012-05-29 12:57:52 +020013760 if (width == -1 && prec == -1
13761 && !(flags & (F_SIGN | F_BLANK)))
13762 {
13763 /* Fast path */
13764 if (formatfloat(v, flags, prec, c, NULL, &writer) == -1)
13765 goto onError;
13766 goto nextarg;
13767 }
13768
Benjamin Peterson29060642009-01-31 22:14:21 +000013769 sign = 1;
Victor Stinnerf2c76aa2012-05-03 13:10:40 +020013770 if (flags & F_ZERO)
Benjamin Peterson29060642009-01-31 22:14:21 +000013771 fill = '0';
Victor Stinnerd3f08822012-05-29 12:57:52 +020013772 if (formatfloat(v, flags, prec, c, &temp, NULL) == -1)
13773 temp = NULL;
Benjamin Peterson29060642009-01-31 22:14:21 +000013774 break;
13775
13776 case 'c':
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013777 {
13778 Py_UCS4 ch = formatchar(v);
13779 if (ch == (Py_UCS4) -1)
Benjamin Peterson29060642009-01-31 22:14:21 +000013780 goto onError;
Victor Stinnerd3f08822012-05-29 12:57:52 +020013781 if (width == -1 && prec == -1) {
13782 /* Fast path */
13783 if (_PyUnicodeWriter_Prepare(&writer, 1, ch) == -1)
13784 goto onError;
13785 PyUnicode_WRITE(writer.kind, writer.data, writer.pos, ch);
13786 writer.pos += 1;
13787 goto nextarg;
13788 }
Victor Stinnerb5c3ea32012-05-02 00:29:36 +020013789 temp = PyUnicode_FromOrdinal(ch);
Benjamin Peterson29060642009-01-31 22:14:21 +000013790 break;
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013791 }
Benjamin Peterson29060642009-01-31 22:14:21 +000013792
13793 default:
13794 PyErr_Format(PyExc_ValueError,
13795 "unsupported format character '%c' (0x%x) "
13796 "at index %zd",
13797 (31<=c && c<=126) ? (char)c : '?',
13798 (int)c,
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013799 fmtpos - 1);
Benjamin Peterson29060642009-01-31 22:14:21 +000013800 goto onError;
13801 }
Victor Stinneraff3cc62012-04-30 05:19:21 +020013802 if (temp == NULL)
13803 goto onError;
13804 assert (PyUnicode_Check(temp));
Victor Stinnerd3f08822012-05-29 12:57:52 +020013805
13806 if (width == -1 && prec == -1
13807 && !(flags & (F_SIGN | F_BLANK)))
13808 {
13809 /* Fast path */
13810 if (_PyUnicodeWriter_WriteStr(&writer, temp) == -1)
13811 goto onError;
13812 goto nextarg;
13813 }
13814
Victor Stinneraff3cc62012-04-30 05:19:21 +020013815 if (PyUnicode_READY(temp) == -1) {
13816 Py_CLEAR(temp);
13817 goto onError;
13818 }
13819 kind = PyUnicode_KIND(temp);
13820 pbuf = PyUnicode_DATA(temp);
13821 len = PyUnicode_GET_LENGTH(temp);
13822
13823 if (c == 's' || c == 'r' || c == 'a') {
13824 if (prec >= 0 && len > prec)
13825 len = prec;
13826 }
13827
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013828 /* pbuf is initialized here. */
13829 pindex = 0;
Benjamin Peterson29060642009-01-31 22:14:21 +000013830 if (sign) {
Victor Stinnerf2c76aa2012-05-03 13:10:40 +020013831 Py_UCS4 ch = PyUnicode_READ(kind, pbuf, pindex);
13832 if (ch == '-' || ch == '+') {
13833 signchar = ch;
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013834 len--;
13835 pindex++;
Benjamin Peterson29060642009-01-31 22:14:21 +000013836 }
13837 else if (flags & F_SIGN)
Victor Stinnerf2c76aa2012-05-03 13:10:40 +020013838 signchar = '+';
Benjamin Peterson29060642009-01-31 22:14:21 +000013839 else if (flags & F_BLANK)
Victor Stinnerf2c76aa2012-05-03 13:10:40 +020013840 signchar = ' ';
Benjamin Peterson29060642009-01-31 22:14:21 +000013841 else
13842 sign = 0;
13843 }
13844 if (width < len)
13845 width = len;
Victor Stinneree4544c2012-05-09 22:24:08 +020013846
13847 /* Compute the length and maximum character of the
13848 written characters */
13849 bufmaxchar = 127;
13850 if (!(flags & F_LJUST)) {
13851 if (sign) {
13852 if ((width-1) > len)
Victor Stinnerd3f08822012-05-29 12:57:52 +020013853 bufmaxchar = MAX_MAXCHAR(bufmaxchar, fill);
Victor Stinneree4544c2012-05-09 22:24:08 +020013854 }
13855 else {
13856 if (width > len)
Victor Stinnerd3f08822012-05-29 12:57:52 +020013857 bufmaxchar = MAX_MAXCHAR(bufmaxchar, fill);
Victor Stinneree4544c2012-05-09 22:24:08 +020013858 }
13859 }
13860 maxchar = _PyUnicode_FindMaxChar(temp, 0, pindex+len);
Victor Stinnerd3f08822012-05-29 12:57:52 +020013861 bufmaxchar = MAX_MAXCHAR(bufmaxchar, maxchar);
Victor Stinneree4544c2012-05-09 22:24:08 +020013862
13863 buflen = width;
13864 if (sign && len == width)
13865 buflen++;
13866
Victor Stinner3b1a74a2012-05-09 22:25:00 +020013867 if (_PyUnicodeWriter_Prepare(&writer, buflen, bufmaxchar) == -1)
Victor Stinneree4544c2012-05-09 22:24:08 +020013868 goto onError;
13869
13870 /* Write characters */
Benjamin Peterson29060642009-01-31 22:14:21 +000013871 if (sign) {
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013872 if (fill != ' ') {
Victor Stinneree4544c2012-05-09 22:24:08 +020013873 PyUnicode_WRITE(writer.kind, writer.data, writer.pos, signchar);
13874 writer.pos += 1;
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013875 }
Benjamin Peterson29060642009-01-31 22:14:21 +000013876 if (width > len)
13877 width--;
13878 }
13879 if ((flags & F_ALT) && (c == 'x' || c == 'X' || c == 'o')) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013880 assert(PyUnicode_READ(kind, pbuf, pindex) == '0');
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013881 assert(PyUnicode_READ(kind, pbuf, pindex + 1) == c);
Benjamin Peterson29060642009-01-31 22:14:21 +000013882 if (fill != ' ') {
Victor Stinnerf2c76aa2012-05-03 13:10:40 +020013883 PyUnicode_WRITE(writer.kind, writer.data, writer.pos, '0');
13884 PyUnicode_WRITE(writer.kind, writer.data, writer.pos+1, c);
13885 writer.pos += 2;
13886 pindex += 2;
Benjamin Peterson29060642009-01-31 22:14:21 +000013887 }
Benjamin Peterson29060642009-01-31 22:14:21 +000013888 width -= 2;
13889 if (width < 0)
13890 width = 0;
13891 len -= 2;
13892 }
13893 if (width > len && !(flags & F_LJUST)) {
Victor Stinnerf2c76aa2012-05-03 13:10:40 +020013894 sublen = width - len;
Victor Stinnerf2c76aa2012-05-03 13:10:40 +020013895 FILL(writer.kind, writer.data, fill, writer.pos, sublen);
13896 writer.pos += sublen;
Antoine Pitrou978b9d22011-10-07 12:35:48 +020013897 width = len;
Benjamin Peterson29060642009-01-31 22:14:21 +000013898 }
13899 if (fill == ' ') {
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013900 if (sign) {
Victor Stinneree4544c2012-05-09 22:24:08 +020013901 PyUnicode_WRITE(writer.kind, writer.data, writer.pos, signchar);
13902 writer.pos += 1;
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013903 }
Benjamin Peterson29060642009-01-31 22:14:21 +000013904 if ((flags & F_ALT) && (c == 'x' || c == 'X' || c == 'o')) {
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020013905 assert(PyUnicode_READ(kind, pbuf, pindex) == '0');
13906 assert(PyUnicode_READ(kind, pbuf, pindex+1) == c);
Victor Stinnerf2c76aa2012-05-03 13:10:40 +020013907 PyUnicode_WRITE(writer.kind, writer.data, writer.pos, '0');
13908 PyUnicode_WRITE(writer.kind, writer.data, writer.pos+1, c);
13909 writer.pos += 2;
Victor Stinnerf2c76aa2012-05-03 13:10:40 +020013910 pindex += 2;
Benjamin Peterson14339b62009-01-31 16:36:08 +000013911 }
13912 }
Victor Stinnerf2c76aa2012-05-03 13:10:40 +020013913
Victor Stinnerc9d369f2012-06-16 02:22:37 +020013914 if (len) {
13915 _PyUnicode_FastCopyCharacters(writer.buffer, writer.pos,
13916 temp, pindex, len);
13917 writer.pos += len;
13918 }
Victor Stinnerf2c76aa2012-05-03 13:10:40 +020013919 if (width > len) {
Victor Stinneree4544c2012-05-09 22:24:08 +020013920 sublen = width - len;
Victor Stinnerf2c76aa2012-05-03 13:10:40 +020013921 FILL(writer.kind, writer.data, ' ', writer.pos, sublen);
13922 writer.pos += sublen;
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013923 }
Victor Stinneree4544c2012-05-09 22:24:08 +020013924
Victor Stinnerd3f08822012-05-29 12:57:52 +020013925nextarg:
Benjamin Peterson29060642009-01-31 22:14:21 +000013926 if (dict && (argidx < arglen) && c != '%') {
13927 PyErr_SetString(PyExc_TypeError,
13928 "not all arguments converted during string formatting");
Benjamin Peterson29060642009-01-31 22:14:21 +000013929 goto onError;
13930 }
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013931 Py_CLEAR(temp);
Benjamin Peterson29060642009-01-31 22:14:21 +000013932 } /* '%' */
Guido van Rossumd57fd912000-03-10 22:53:23 +000013933 } /* until end */
13934 if (argidx < arglen && !dict) {
Benjamin Peterson29060642009-01-31 22:14:21 +000013935 PyErr_SetString(PyExc_TypeError,
13936 "not all arguments converted during string formatting");
13937 goto onError;
Guido van Rossumd57fd912000-03-10 22:53:23 +000013938 }
13939
13940 if (args_owned) {
Benjamin Peterson29060642009-01-31 22:14:21 +000013941 Py_DECREF(args);
Guido van Rossumd57fd912000-03-10 22:53:23 +000013942 }
13943 Py_DECREF(uformat);
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013944 Py_XDECREF(temp);
13945 Py_XDECREF(second);
Victor Stinner3b1a74a2012-05-09 22:25:00 +020013946 return _PyUnicodeWriter_Finish(&writer);
Guido van Rossumd57fd912000-03-10 22:53:23 +000013947
Benjamin Peterson29060642009-01-31 22:14:21 +000013948 onError:
Guido van Rossumd57fd912000-03-10 22:53:23 +000013949 Py_DECREF(uformat);
Antoine Pitrou5c0ba362011-10-07 01:54:09 +020013950 Py_XDECREF(temp);
13951 Py_XDECREF(second);
Victor Stinner3b1a74a2012-05-09 22:25:00 +020013952 _PyUnicodeWriter_Dealloc(&writer);
Guido van Rossumd57fd912000-03-10 22:53:23 +000013953 if (args_owned) {
Benjamin Peterson29060642009-01-31 22:14:21 +000013954 Py_DECREF(args);
Guido van Rossumd57fd912000-03-10 22:53:23 +000013955 }
13956 return NULL;
13957}
13958
Jeremy Hylton938ace62002-07-17 16:30:39 +000013959static PyObject *
Guido van Rossume023fe02001-08-30 03:12:59 +000013960unicode_subtype_new(PyTypeObject *type, PyObject *args, PyObject *kwds);
13961
Tim Peters6d6c1a32001-08-02 04:15:00 +000013962static PyObject *
13963unicode_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
13964{
Benjamin Peterson29060642009-01-31 22:14:21 +000013965 PyObject *x = NULL;
Benjamin Peterson14339b62009-01-31 16:36:08 +000013966 static char *kwlist[] = {"object", "encoding", "errors", 0};
13967 char *encoding = NULL;
13968 char *errors = NULL;
Tim Peters6d6c1a32001-08-02 04:15:00 +000013969
Benjamin Peterson14339b62009-01-31 16:36:08 +000013970 if (type != &PyUnicode_Type)
13971 return unicode_subtype_new(type, args, kwds);
13972 if (!PyArg_ParseTupleAndKeywords(args, kwds, "|Oss:str",
Benjamin Peterson29060642009-01-31 22:14:21 +000013973 kwlist, &x, &encoding, &errors))
Benjamin Peterson14339b62009-01-31 16:36:08 +000013974 return NULL;
Victor Stinner382955f2011-12-11 21:44:00 +010013975 if (x == NULL) {
13976 Py_INCREF(unicode_empty);
13977 return unicode_empty;
13978 }
Benjamin Peterson14339b62009-01-31 16:36:08 +000013979 if (encoding == NULL && errors == NULL)
13980 return PyObject_Str(x);
13981 else
Benjamin Peterson29060642009-01-31 22:14:21 +000013982 return PyUnicode_FromEncodedObject(x, encoding, errors);
Tim Peters6d6c1a32001-08-02 04:15:00 +000013983}
13984
Guido van Rossume023fe02001-08-30 03:12:59 +000013985static PyObject *
13986unicode_subtype_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
13987{
Victor Stinner9db1a8b2011-10-23 20:04:37 +020013988 PyObject *unicode, *self;
Victor Stinner07ac3eb2011-10-01 16:16:43 +020013989 Py_ssize_t length, char_size;
13990 int share_wstr, share_utf8;
13991 unsigned int kind;
13992 void *data;
Guido van Rossume023fe02001-08-30 03:12:59 +000013993
Benjamin Peterson14339b62009-01-31 16:36:08 +000013994 assert(PyType_IsSubtype(type, &PyUnicode_Type));
Victor Stinner07ac3eb2011-10-01 16:16:43 +020013995
Victor Stinner9db1a8b2011-10-23 20:04:37 +020013996 unicode = unicode_new(&PyUnicode_Type, args, kwds);
Victor Stinner07ac3eb2011-10-01 16:16:43 +020013997 if (unicode == NULL)
Benjamin Peterson14339b62009-01-31 16:36:08 +000013998 return NULL;
Victor Stinner910337b2011-10-03 03:20:16 +020013999 assert(_PyUnicode_CHECK(unicode));
Benjamin Petersonbac79492012-01-14 13:34:47 -050014000 if (PyUnicode_READY(unicode) == -1) {
Benjamin Peterson22a29702012-01-02 09:00:30 -060014001 Py_DECREF(unicode);
Victor Stinner07ac3eb2011-10-01 16:16:43 +020014002 return NULL;
Benjamin Peterson22a29702012-01-02 09:00:30 -060014003 }
Victor Stinner07ac3eb2011-10-01 16:16:43 +020014004
Victor Stinner9db1a8b2011-10-23 20:04:37 +020014005 self = type->tp_alloc(type, 0);
Victor Stinner07ac3eb2011-10-01 16:16:43 +020014006 if (self == NULL) {
14007 Py_DECREF(unicode);
Benjamin Peterson14339b62009-01-31 16:36:08 +000014008 return NULL;
14009 }
Victor Stinner07ac3eb2011-10-01 16:16:43 +020014010 kind = PyUnicode_KIND(unicode);
14011 length = PyUnicode_GET_LENGTH(unicode);
14012
14013 _PyUnicode_LENGTH(self) = length;
Victor Stinnerfb9ea8c2011-10-06 01:45:57 +020014014#ifdef Py_DEBUG
14015 _PyUnicode_HASH(self) = -1;
14016#else
Victor Stinner07ac3eb2011-10-01 16:16:43 +020014017 _PyUnicode_HASH(self) = _PyUnicode_HASH(unicode);
Victor Stinnerfb9ea8c2011-10-06 01:45:57 +020014018#endif
Victor Stinner07ac3eb2011-10-01 16:16:43 +020014019 _PyUnicode_STATE(self).interned = 0;
14020 _PyUnicode_STATE(self).kind = kind;
14021 _PyUnicode_STATE(self).compact = 0;
Victor Stinner3cf46372011-10-03 14:42:15 +020014022 _PyUnicode_STATE(self).ascii = _PyUnicode_STATE(unicode).ascii;
Victor Stinner07ac3eb2011-10-01 16:16:43 +020014023 _PyUnicode_STATE(self).ready = 1;
14024 _PyUnicode_WSTR(self) = NULL;
14025 _PyUnicode_UTF8_LENGTH(self) = 0;
14026 _PyUnicode_UTF8(self) = NULL;
14027 _PyUnicode_WSTR_LENGTH(self) = 0;
Victor Stinnerc3c74152011-10-02 20:39:55 +020014028 _PyUnicode_DATA_ANY(self) = NULL;
Victor Stinner07ac3eb2011-10-01 16:16:43 +020014029
14030 share_utf8 = 0;
14031 share_wstr = 0;
14032 if (kind == PyUnicode_1BYTE_KIND) {
14033 char_size = 1;
14034 if (PyUnicode_MAX_CHAR_VALUE(unicode) < 128)
14035 share_utf8 = 1;
14036 }
14037 else if (kind == PyUnicode_2BYTE_KIND) {
14038 char_size = 2;
14039 if (sizeof(wchar_t) == 2)
14040 share_wstr = 1;
14041 }
14042 else {
14043 assert(kind == PyUnicode_4BYTE_KIND);
14044 char_size = 4;
14045 if (sizeof(wchar_t) == 4)
14046 share_wstr = 1;
14047 }
14048
14049 /* Ensure we won't overflow the length. */
14050 if (length > (PY_SSIZE_T_MAX / char_size - 1)) {
14051 PyErr_NoMemory();
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020014052 goto onError;
Benjamin Peterson14339b62009-01-31 16:36:08 +000014053 }
Victor Stinner07ac3eb2011-10-01 16:16:43 +020014054 data = PyObject_MALLOC((length + 1) * char_size);
14055 if (data == NULL) {
14056 PyErr_NoMemory();
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020014057 goto onError;
14058 }
14059
Victor Stinnerc3c74152011-10-02 20:39:55 +020014060 _PyUnicode_DATA_ANY(self) = data;
Victor Stinner07ac3eb2011-10-01 16:16:43 +020014061 if (share_utf8) {
14062 _PyUnicode_UTF8_LENGTH(self) = length;
14063 _PyUnicode_UTF8(self) = data;
14064 }
14065 if (share_wstr) {
14066 _PyUnicode_WSTR_LENGTH(self) = length;
14067 _PyUnicode_WSTR(self) = (wchar_t *)data;
14068 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020014069
Victor Stinner07ac3eb2011-10-01 16:16:43 +020014070 Py_MEMCPY(data, PyUnicode_DATA(unicode),
Martin v. Löwisc47adb02011-10-07 20:55:35 +020014071 kind * (length + 1));
Victor Stinnerbb10a1f2011-10-05 01:34:17 +020014072 assert(_PyUnicode_CheckConsistency(self, 1));
Victor Stinnerfb9ea8c2011-10-06 01:45:57 +020014073#ifdef Py_DEBUG
14074 _PyUnicode_HASH(self) = _PyUnicode_HASH(unicode);
14075#endif
Victor Stinnerdd18d3a2011-10-22 11:08:10 +020014076 Py_DECREF(unicode);
Victor Stinner7931d9a2011-11-04 00:22:48 +010014077 return self;
Victor Stinner07ac3eb2011-10-01 16:16:43 +020014078
14079onError:
14080 Py_DECREF(unicode);
14081 Py_DECREF(self);
14082 return NULL;
Guido van Rossume023fe02001-08-30 03:12:59 +000014083}
14084
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +000014085PyDoc_STRVAR(unicode_doc,
Nick Coghlan573b1fd2012-08-16 14:13:07 +100014086 "str(object[, encoding[, errors]]) -> str\n\
Tim Peters6d6c1a32001-08-02 04:15:00 +000014087\n\
Nick Coghlan573b1fd2012-08-16 14:13:07 +100014088Create a new string object from the given object. If encoding or\n\
14089errors is specified, then the object must expose a data buffer\n\
14090that will be decoded using the given encoding and error handler.\n\
14091Otherwise, returns the result of object.__str__() (if defined)\n\
14092or repr(object).\n\
14093encoding defaults to sys.getdefaultencoding().\n\
14094errors defaults to 'strict'.");
Tim Peters6d6c1a32001-08-02 04:15:00 +000014095
Guido van Rossum50e9fb92006-08-17 05:42:55 +000014096static PyObject *unicode_iter(PyObject *seq);
14097
Guido van Rossumd57fd912000-03-10 22:53:23 +000014098PyTypeObject PyUnicode_Type = {
Martin v. Löwis9f2e3462007-07-21 17:22:18 +000014099 PyVarObject_HEAD_INIT(&PyType_Type, 0)
Benjamin Peterson14339b62009-01-31 16:36:08 +000014100 "str", /* tp_name */
14101 sizeof(PyUnicodeObject), /* tp_size */
14102 0, /* tp_itemsize */
Guido van Rossumd57fd912000-03-10 22:53:23 +000014103 /* Slots */
Benjamin Peterson14339b62009-01-31 16:36:08 +000014104 (destructor)unicode_dealloc, /* tp_dealloc */
14105 0, /* tp_print */
14106 0, /* tp_getattr */
14107 0, /* tp_setattr */
Mark Dickinsone94c6792009-02-02 20:36:42 +000014108 0, /* tp_reserved */
Benjamin Peterson14339b62009-01-31 16:36:08 +000014109 unicode_repr, /* tp_repr */
14110 &unicode_as_number, /* tp_as_number */
14111 &unicode_as_sequence, /* tp_as_sequence */
14112 &unicode_as_mapping, /* tp_as_mapping */
14113 (hashfunc) unicode_hash, /* tp_hash*/
14114 0, /* tp_call*/
14115 (reprfunc) unicode_str, /* tp_str */
14116 PyObject_GenericGetAttr, /* tp_getattro */
14117 0, /* tp_setattro */
14118 0, /* tp_as_buffer */
14119 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE |
Benjamin Peterson29060642009-01-31 22:14:21 +000014120 Py_TPFLAGS_UNICODE_SUBCLASS, /* tp_flags */
Benjamin Peterson14339b62009-01-31 16:36:08 +000014121 unicode_doc, /* tp_doc */
14122 0, /* tp_traverse */
14123 0, /* tp_clear */
14124 PyUnicode_RichCompare, /* tp_richcompare */
14125 0, /* tp_weaklistoffset */
14126 unicode_iter, /* tp_iter */
14127 0, /* tp_iternext */
14128 unicode_methods, /* tp_methods */
14129 0, /* tp_members */
14130 0, /* tp_getset */
14131 &PyBaseObject_Type, /* tp_base */
14132 0, /* tp_dict */
14133 0, /* tp_descr_get */
14134 0, /* tp_descr_set */
14135 0, /* tp_dictoffset */
14136 0, /* tp_init */
14137 0, /* tp_alloc */
14138 unicode_new, /* tp_new */
14139 PyObject_Del, /* tp_free */
Guido van Rossumd57fd912000-03-10 22:53:23 +000014140};
14141
14142/* Initialize the Unicode implementation */
14143
Victor Stinner3a50e702011-10-18 21:21:00 +020014144int _PyUnicode_Init(void)
Guido van Rossumd57fd912000-03-10 22:53:23 +000014145{
Marc-André Lemburg8155e0e2001-04-23 14:44:21 +000014146 int i;
14147
Thomas Wouters477c8d52006-05-27 19:21:47 +000014148 /* XXX - move this array to unicodectype.c ? */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020014149 Py_UCS2 linebreak[] = {
Thomas Wouters477c8d52006-05-27 19:21:47 +000014150 0x000A, /* LINE FEED */
14151 0x000D, /* CARRIAGE RETURN */
14152 0x001C, /* FILE SEPARATOR */
14153 0x001D, /* GROUP SEPARATOR */
14154 0x001E, /* RECORD SEPARATOR */
14155 0x0085, /* NEXT LINE */
14156 0x2028, /* LINE SEPARATOR */
14157 0x2029, /* PARAGRAPH SEPARATOR */
14158 };
14159
Fred Drakee4315f52000-05-09 19:53:39 +000014160 /* Init the implementation */
Victor Stinnera464fc12011-10-02 20:39:30 +020014161 unicode_empty = PyUnicode_New(0, 0);
Thomas Wouters0e3f5912006-08-11 14:57:12 +000014162 if (!unicode_empty)
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020014163 Py_FatalError("Can't create empty string");
Victor Stinnerd3df8ab2011-11-22 01:22:34 +010014164 assert(_PyUnicode_CheckConsistency(unicode_empty, 1));
Thomas Wouters0e3f5912006-08-11 14:57:12 +000014165
Marc-André Lemburg8155e0e2001-04-23 14:44:21 +000014166 for (i = 0; i < 256; i++)
Benjamin Peterson29060642009-01-31 22:14:21 +000014167 unicode_latin1[i] = NULL;
Guido van Rossumcacfc072002-05-24 19:01:59 +000014168 if (PyType_Ready(&PyUnicode_Type) < 0)
Benjamin Peterson29060642009-01-31 22:14:21 +000014169 Py_FatalError("Can't initialize 'unicode'");
Thomas Wouters477c8d52006-05-27 19:21:47 +000014170
14171 /* initialize the linebreak bloom filter */
14172 bloom_linebreak = make_bloom_mask(
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020014173 PyUnicode_2BYTE_KIND, linebreak,
Victor Stinner63941882011-09-29 00:42:28 +020014174 Py_ARRAY_LENGTH(linebreak));
Thomas Wouters0e3f5912006-08-11 14:57:12 +000014175
14176 PyType_Ready(&EncodingMapType);
Victor Stinner3a50e702011-10-18 21:21:00 +020014177
14178#ifdef HAVE_MBCS
14179 winver.dwOSVersionInfoSize = sizeof(winver);
14180 if (!GetVersionEx((OSVERSIONINFO*)&winver)) {
14181 PyErr_SetFromWindowsErr(0);
14182 return -1;
14183 }
14184#endif
14185 return 0;
Guido van Rossumd57fd912000-03-10 22:53:23 +000014186}
14187
14188/* Finalize the Unicode implementation */
14189
Christian Heimesa156e092008-02-16 07:38:31 +000014190int
14191PyUnicode_ClearFreeList(void)
14192{
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020014193 return 0;
Christian Heimesa156e092008-02-16 07:38:31 +000014194}
14195
Guido van Rossumd57fd912000-03-10 22:53:23 +000014196void
Thomas Wouters78890102000-07-22 19:25:51 +000014197_PyUnicode_Fini(void)
Guido van Rossumd57fd912000-03-10 22:53:23 +000014198{
Marc-André Lemburg8155e0e2001-04-23 14:44:21 +000014199 int i;
Guido van Rossumd57fd912000-03-10 22:53:23 +000014200
Guido van Rossum4ae8ef82000-10-03 18:09:04 +000014201 Py_XDECREF(unicode_empty);
14202 unicode_empty = NULL;
Barry Warsaw5b4c2282000-10-03 20:45:26 +000014203
Marc-André Lemburg8155e0e2001-04-23 14:44:21 +000014204 for (i = 0; i < 256; i++) {
Benjamin Peterson29060642009-01-31 22:14:21 +000014205 if (unicode_latin1[i]) {
14206 Py_DECREF(unicode_latin1[i]);
14207 unicode_latin1[i] = NULL;
14208 }
Marc-André Lemburg8155e0e2001-04-23 14:44:21 +000014209 }
Martin v. Löwisafe55bb2011-10-09 10:38:36 +020014210 _PyUnicode_ClearStaticStrings();
Christian Heimesa156e092008-02-16 07:38:31 +000014211 (void)PyUnicode_ClearFreeList();
Guido van Rossumd57fd912000-03-10 22:53:23 +000014212}
Martin v. Löwis9a3a9f72003-05-18 12:31:09 +000014213
Walter Dörwald16807132007-05-25 13:52:07 +000014214void
14215PyUnicode_InternInPlace(PyObject **p)
14216{
Victor Stinner9db1a8b2011-10-23 20:04:37 +020014217 register PyObject *s = *p;
Benjamin Peterson14339b62009-01-31 16:36:08 +000014218 PyObject *t;
Victor Stinner4fae54c2011-10-03 02:01:52 +020014219#ifdef Py_DEBUG
14220 assert(s != NULL);
14221 assert(_PyUnicode_CHECK(s));
14222#else
Benjamin Peterson14339b62009-01-31 16:36:08 +000014223 if (s == NULL || !PyUnicode_Check(s))
Victor Stinner4fae54c2011-10-03 02:01:52 +020014224 return;
14225#endif
Benjamin Peterson14339b62009-01-31 16:36:08 +000014226 /* If it's a subclass, we don't really know what putting
14227 it in the interned dict might do. */
14228 if (!PyUnicode_CheckExact(s))
14229 return;
14230 if (PyUnicode_CHECK_INTERNED(s))
14231 return;
14232 if (interned == NULL) {
14233 interned = PyDict_New();
14234 if (interned == NULL) {
14235 PyErr_Clear(); /* Don't leave an exception */
14236 return;
14237 }
14238 }
14239 /* It might be that the GetItem call fails even
14240 though the key is present in the dictionary,
14241 namely when this happens during a stack overflow. */
14242 Py_ALLOW_RECURSION
Victor Stinner7931d9a2011-11-04 00:22:48 +010014243 t = PyDict_GetItem(interned, s);
Benjamin Peterson14339b62009-01-31 16:36:08 +000014244 Py_END_ALLOW_RECURSION
Martin v. Löwis5b222132007-06-10 09:51:05 +000014245
Benjamin Peterson29060642009-01-31 22:14:21 +000014246 if (t) {
14247 Py_INCREF(t);
14248 Py_DECREF(*p);
14249 *p = t;
14250 return;
14251 }
Walter Dörwald16807132007-05-25 13:52:07 +000014252
Benjamin Peterson14339b62009-01-31 16:36:08 +000014253 PyThreadState_GET()->recursion_critical = 1;
Victor Stinner7931d9a2011-11-04 00:22:48 +010014254 if (PyDict_SetItem(interned, s, s) < 0) {
Benjamin Peterson14339b62009-01-31 16:36:08 +000014255 PyErr_Clear();
14256 PyThreadState_GET()->recursion_critical = 0;
14257 return;
14258 }
14259 PyThreadState_GET()->recursion_critical = 0;
14260 /* The two references in interned are not counted by refcnt.
14261 The deallocator will take care of this */
14262 Py_REFCNT(s) -= 2;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020014263 _PyUnicode_STATE(s).interned = SSTATE_INTERNED_MORTAL;
Walter Dörwald16807132007-05-25 13:52:07 +000014264}
14265
14266void
14267PyUnicode_InternImmortal(PyObject **p)
14268{
Benjamin Peterson14339b62009-01-31 16:36:08 +000014269 PyUnicode_InternInPlace(p);
14270 if (PyUnicode_CHECK_INTERNED(*p) != SSTATE_INTERNED_IMMORTAL) {
Victor Stinneraf9e4b82011-10-23 20:07:00 +020014271 _PyUnicode_STATE(*p).interned = SSTATE_INTERNED_IMMORTAL;
Benjamin Peterson14339b62009-01-31 16:36:08 +000014272 Py_INCREF(*p);
14273 }
Walter Dörwald16807132007-05-25 13:52:07 +000014274}
14275
14276PyObject *
14277PyUnicode_InternFromString(const char *cp)
14278{
Benjamin Peterson14339b62009-01-31 16:36:08 +000014279 PyObject *s = PyUnicode_FromString(cp);
14280 if (s == NULL)
14281 return NULL;
14282 PyUnicode_InternInPlace(&s);
14283 return s;
Walter Dörwald16807132007-05-25 13:52:07 +000014284}
14285
Alexander Belopolsky40018472011-02-26 01:02:56 +000014286void
14287_Py_ReleaseInternedUnicodeStrings(void)
Walter Dörwald16807132007-05-25 13:52:07 +000014288{
Benjamin Peterson14339b62009-01-31 16:36:08 +000014289 PyObject *keys;
Victor Stinner9db1a8b2011-10-23 20:04:37 +020014290 PyObject *s;
Benjamin Peterson14339b62009-01-31 16:36:08 +000014291 Py_ssize_t i, n;
14292 Py_ssize_t immortal_size = 0, mortal_size = 0;
Walter Dörwald16807132007-05-25 13:52:07 +000014293
Benjamin Peterson14339b62009-01-31 16:36:08 +000014294 if (interned == NULL || !PyDict_Check(interned))
14295 return;
14296 keys = PyDict_Keys(interned);
14297 if (keys == NULL || !PyList_Check(keys)) {
14298 PyErr_Clear();
14299 return;
14300 }
Walter Dörwald16807132007-05-25 13:52:07 +000014301
Benjamin Peterson14339b62009-01-31 16:36:08 +000014302 /* Since _Py_ReleaseInternedUnicodeStrings() is intended to help a leak
14303 detector, interned unicode strings are not forcibly deallocated;
14304 rather, we give them their stolen references back, and then clear
14305 and DECREF the interned dict. */
Walter Dörwald16807132007-05-25 13:52:07 +000014306
Benjamin Peterson14339b62009-01-31 16:36:08 +000014307 n = PyList_GET_SIZE(keys);
14308 fprintf(stderr, "releasing %" PY_FORMAT_SIZE_T "d interned strings\n",
Benjamin Peterson29060642009-01-31 22:14:21 +000014309 n);
Benjamin Peterson14339b62009-01-31 16:36:08 +000014310 for (i = 0; i < n; i++) {
Victor Stinner9db1a8b2011-10-23 20:04:37 +020014311 s = PyList_GET_ITEM(keys, i);
Victor Stinner6b56a7f2011-10-04 20:04:52 +020014312 if (PyUnicode_READY(s) == -1) {
14313 assert(0 && "could not ready string");
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020014314 fprintf(stderr, "could not ready string\n");
Victor Stinner6b56a7f2011-10-04 20:04:52 +020014315 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020014316 switch (PyUnicode_CHECK_INTERNED(s)) {
Benjamin Peterson14339b62009-01-31 16:36:08 +000014317 case SSTATE_NOT_INTERNED:
14318 /* XXX Shouldn't happen */
14319 break;
14320 case SSTATE_INTERNED_IMMORTAL:
14321 Py_REFCNT(s) += 1;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020014322 immortal_size += PyUnicode_GET_LENGTH(s);
Benjamin Peterson14339b62009-01-31 16:36:08 +000014323 break;
14324 case SSTATE_INTERNED_MORTAL:
14325 Py_REFCNT(s) += 2;
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020014326 mortal_size += PyUnicode_GET_LENGTH(s);
Benjamin Peterson14339b62009-01-31 16:36:08 +000014327 break;
14328 default:
14329 Py_FatalError("Inconsistent interned string state.");
14330 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020014331 _PyUnicode_STATE(s).interned = SSTATE_NOT_INTERNED;
Benjamin Peterson14339b62009-01-31 16:36:08 +000014332 }
14333 fprintf(stderr, "total size of all interned strings: "
14334 "%" PY_FORMAT_SIZE_T "d/%" PY_FORMAT_SIZE_T "d "
14335 "mortal/immortal\n", mortal_size, immortal_size);
14336 Py_DECREF(keys);
14337 PyDict_Clear(interned);
14338 Py_DECREF(interned);
14339 interned = NULL;
Walter Dörwald16807132007-05-25 13:52:07 +000014340}
Guido van Rossum50e9fb92006-08-17 05:42:55 +000014341
14342
14343/********************* Unicode Iterator **************************/
14344
14345typedef struct {
Benjamin Peterson14339b62009-01-31 16:36:08 +000014346 PyObject_HEAD
14347 Py_ssize_t it_index;
Victor Stinner9db1a8b2011-10-23 20:04:37 +020014348 PyObject *it_seq; /* Set to NULL when iterator is exhausted */
Guido van Rossum50e9fb92006-08-17 05:42:55 +000014349} unicodeiterobject;
14350
14351static void
14352unicodeiter_dealloc(unicodeiterobject *it)
14353{
Benjamin Peterson14339b62009-01-31 16:36:08 +000014354 _PyObject_GC_UNTRACK(it);
14355 Py_XDECREF(it->it_seq);
14356 PyObject_GC_Del(it);
Guido van Rossum50e9fb92006-08-17 05:42:55 +000014357}
14358
14359static int
14360unicodeiter_traverse(unicodeiterobject *it, visitproc visit, void *arg)
14361{
Benjamin Peterson14339b62009-01-31 16:36:08 +000014362 Py_VISIT(it->it_seq);
14363 return 0;
Guido van Rossum50e9fb92006-08-17 05:42:55 +000014364}
14365
14366static PyObject *
14367unicodeiter_next(unicodeiterobject *it)
14368{
Victor Stinner9db1a8b2011-10-23 20:04:37 +020014369 PyObject *seq, *item;
Guido van Rossum50e9fb92006-08-17 05:42:55 +000014370
Benjamin Peterson14339b62009-01-31 16:36:08 +000014371 assert(it != NULL);
14372 seq = it->it_seq;
14373 if (seq == NULL)
14374 return NULL;
Victor Stinner910337b2011-10-03 03:20:16 +020014375 assert(_PyUnicode_CHECK(seq));
Guido van Rossum50e9fb92006-08-17 05:42:55 +000014376
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020014377 if (it->it_index < PyUnicode_GET_LENGTH(seq)) {
14378 int kind = PyUnicode_KIND(seq);
14379 void *data = PyUnicode_DATA(seq);
14380 Py_UCS4 chr = PyUnicode_READ(kind, data, it->it_index);
14381 item = PyUnicode_FromOrdinal(chr);
Benjamin Peterson14339b62009-01-31 16:36:08 +000014382 if (item != NULL)
14383 ++it->it_index;
14384 return item;
14385 }
Guido van Rossum50e9fb92006-08-17 05:42:55 +000014386
Benjamin Peterson14339b62009-01-31 16:36:08 +000014387 Py_DECREF(seq);
14388 it->it_seq = NULL;
14389 return NULL;
Guido van Rossum50e9fb92006-08-17 05:42:55 +000014390}
14391
14392static PyObject *
14393unicodeiter_len(unicodeiterobject *it)
14394{
Benjamin Peterson14339b62009-01-31 16:36:08 +000014395 Py_ssize_t len = 0;
14396 if (it->it_seq)
Victor Stinnerc4f281e2011-10-11 22:11:42 +020014397 len = PyUnicode_GET_LENGTH(it->it_seq) - it->it_index;
Benjamin Peterson14339b62009-01-31 16:36:08 +000014398 return PyLong_FromSsize_t(len);
Guido van Rossum50e9fb92006-08-17 05:42:55 +000014399}
14400
14401PyDoc_STRVAR(length_hint_doc, "Private method returning an estimate of len(list(it)).");
14402
Kristján Valur Jónsson31668b82012-04-03 10:49:41 +000014403static PyObject *
14404unicodeiter_reduce(unicodeiterobject *it)
14405{
14406 if (it->it_seq != NULL) {
Antoine Pitroua7013882012-04-05 00:04:20 +020014407 return Py_BuildValue("N(O)n", _PyObject_GetBuiltin("iter"),
Kristján Valur Jónsson31668b82012-04-03 10:49:41 +000014408 it->it_seq, it->it_index);
14409 } else {
14410 PyObject *u = PyUnicode_FromUnicode(NULL, 0);
14411 if (u == NULL)
14412 return NULL;
Antoine Pitroua7013882012-04-05 00:04:20 +020014413 return Py_BuildValue("N(N)", _PyObject_GetBuiltin("iter"), u);
Kristján Valur Jónsson31668b82012-04-03 10:49:41 +000014414 }
14415}
14416
14417PyDoc_STRVAR(reduce_doc, "Return state information for pickling.");
14418
14419static PyObject *
14420unicodeiter_setstate(unicodeiterobject *it, PyObject *state)
14421{
14422 Py_ssize_t index = PyLong_AsSsize_t(state);
14423 if (index == -1 && PyErr_Occurred())
14424 return NULL;
14425 if (index < 0)
14426 index = 0;
14427 it->it_index = index;
14428 Py_RETURN_NONE;
14429}
14430
14431PyDoc_STRVAR(setstate_doc, "Set state information for unpickling.");
14432
Guido van Rossum50e9fb92006-08-17 05:42:55 +000014433static PyMethodDef unicodeiter_methods[] = {
Benjamin Peterson14339b62009-01-31 16:36:08 +000014434 {"__length_hint__", (PyCFunction)unicodeiter_len, METH_NOARGS,
Benjamin Peterson29060642009-01-31 22:14:21 +000014435 length_hint_doc},
Kristján Valur Jónsson31668b82012-04-03 10:49:41 +000014436 {"__reduce__", (PyCFunction)unicodeiter_reduce, METH_NOARGS,
14437 reduce_doc},
14438 {"__setstate__", (PyCFunction)unicodeiter_setstate, METH_O,
14439 setstate_doc},
Benjamin Peterson14339b62009-01-31 16:36:08 +000014440 {NULL, NULL} /* sentinel */
Guido van Rossum50e9fb92006-08-17 05:42:55 +000014441};
14442
14443PyTypeObject PyUnicodeIter_Type = {
Benjamin Peterson14339b62009-01-31 16:36:08 +000014444 PyVarObject_HEAD_INIT(&PyType_Type, 0)
14445 "str_iterator", /* tp_name */
14446 sizeof(unicodeiterobject), /* tp_basicsize */
14447 0, /* tp_itemsize */
14448 /* methods */
14449 (destructor)unicodeiter_dealloc, /* tp_dealloc */
14450 0, /* tp_print */
14451 0, /* tp_getattr */
14452 0, /* tp_setattr */
Mark Dickinsone94c6792009-02-02 20:36:42 +000014453 0, /* tp_reserved */
Benjamin Peterson14339b62009-01-31 16:36:08 +000014454 0, /* tp_repr */
14455 0, /* tp_as_number */
14456 0, /* tp_as_sequence */
14457 0, /* tp_as_mapping */
14458 0, /* tp_hash */
14459 0, /* tp_call */
14460 0, /* tp_str */
14461 PyObject_GenericGetAttr, /* tp_getattro */
14462 0, /* tp_setattro */
14463 0, /* tp_as_buffer */
14464 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC,/* tp_flags */
14465 0, /* tp_doc */
14466 (traverseproc)unicodeiter_traverse, /* tp_traverse */
14467 0, /* tp_clear */
14468 0, /* tp_richcompare */
14469 0, /* tp_weaklistoffset */
14470 PyObject_SelfIter, /* tp_iter */
14471 (iternextfunc)unicodeiter_next, /* tp_iternext */
14472 unicodeiter_methods, /* tp_methods */
14473 0,
Guido van Rossum50e9fb92006-08-17 05:42:55 +000014474};
14475
14476static PyObject *
14477unicode_iter(PyObject *seq)
14478{
Benjamin Peterson14339b62009-01-31 16:36:08 +000014479 unicodeiterobject *it;
Guido van Rossum50e9fb92006-08-17 05:42:55 +000014480
Benjamin Peterson14339b62009-01-31 16:36:08 +000014481 if (!PyUnicode_Check(seq)) {
14482 PyErr_BadInternalCall();
14483 return NULL;
14484 }
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020014485 if (PyUnicode_READY(seq) == -1)
14486 return NULL;
Benjamin Peterson14339b62009-01-31 16:36:08 +000014487 it = PyObject_GC_New(unicodeiterobject, &PyUnicodeIter_Type);
14488 if (it == NULL)
14489 return NULL;
14490 it->it_index = 0;
14491 Py_INCREF(seq);
Victor Stinner9db1a8b2011-10-23 20:04:37 +020014492 it->it_seq = seq;
Benjamin Peterson14339b62009-01-31 16:36:08 +000014493 _PyObject_GC_TRACK(it);
14494 return (PyObject *)it;
Guido van Rossum50e9fb92006-08-17 05:42:55 +000014495}
14496
Martin v. Löwis0d3072e2011-10-31 08:40:56 +010014497
14498size_t
14499Py_UNICODE_strlen(const Py_UNICODE *u)
14500{
14501 int res = 0;
14502 while(*u++)
14503 res++;
14504 return res;
14505}
14506
14507Py_UNICODE*
14508Py_UNICODE_strcpy(Py_UNICODE *s1, const Py_UNICODE *s2)
14509{
14510 Py_UNICODE *u = s1;
14511 while ((*u++ = *s2++));
14512 return s1;
14513}
14514
14515Py_UNICODE*
14516Py_UNICODE_strncpy(Py_UNICODE *s1, const Py_UNICODE *s2, size_t n)
14517{
14518 Py_UNICODE *u = s1;
14519 while ((*u++ = *s2++))
14520 if (n-- == 0)
14521 break;
14522 return s1;
14523}
14524
14525Py_UNICODE*
14526Py_UNICODE_strcat(Py_UNICODE *s1, const Py_UNICODE *s2)
14527{
14528 Py_UNICODE *u1 = s1;
14529 u1 += Py_UNICODE_strlen(u1);
14530 Py_UNICODE_strcpy(u1, s2);
14531 return s1;
14532}
14533
14534int
14535Py_UNICODE_strcmp(const Py_UNICODE *s1, const Py_UNICODE *s2)
14536{
14537 while (*s1 && *s2 && *s1 == *s2)
14538 s1++, s2++;
14539 if (*s1 && *s2)
14540 return (*s1 < *s2) ? -1 : +1;
14541 if (*s1)
14542 return 1;
14543 if (*s2)
14544 return -1;
14545 return 0;
14546}
14547
14548int
14549Py_UNICODE_strncmp(const Py_UNICODE *s1, const Py_UNICODE *s2, size_t n)
14550{
14551 register Py_UNICODE u1, u2;
14552 for (; n != 0; n--) {
14553 u1 = *s1;
14554 u2 = *s2;
14555 if (u1 != u2)
14556 return (u1 < u2) ? -1 : +1;
14557 if (u1 == '\0')
14558 return 0;
14559 s1++;
14560 s2++;
14561 }
14562 return 0;
14563}
14564
14565Py_UNICODE*
14566Py_UNICODE_strchr(const Py_UNICODE *s, Py_UNICODE c)
14567{
14568 const Py_UNICODE *p;
14569 for (p = s; *p; p++)
14570 if (*p == c)
14571 return (Py_UNICODE*)p;
14572 return NULL;
14573}
14574
14575Py_UNICODE*
14576Py_UNICODE_strrchr(const Py_UNICODE *s, Py_UNICODE c)
14577{
14578 const Py_UNICODE *p;
14579 p = s + Py_UNICODE_strlen(s);
14580 while (p != s) {
14581 p--;
14582 if (*p == c)
14583 return (Py_UNICODE*)p;
14584 }
14585 return NULL;
14586}
Victor Stinner331ea922010-08-10 16:37:20 +000014587
Victor Stinner71133ff2010-09-01 23:43:53 +000014588Py_UNICODE*
Victor Stinner9db1a8b2011-10-23 20:04:37 +020014589PyUnicode_AsUnicodeCopy(PyObject *unicode)
Victor Stinner71133ff2010-09-01 23:43:53 +000014590{
Victor Stinner577db2c2011-10-11 22:12:48 +020014591 Py_UNICODE *u, *copy;
Victor Stinner57ffa9d2011-10-23 20:10:08 +020014592 Py_ssize_t len, size;
Victor Stinner71133ff2010-09-01 23:43:53 +000014593
Martin v. Löwisd63a3b82011-09-28 07:41:54 +020014594 if (!PyUnicode_Check(unicode)) {
14595 PyErr_BadArgument();
14596 return NULL;
14597 }
Victor Stinner57ffa9d2011-10-23 20:10:08 +020014598 u = PyUnicode_AsUnicodeAndSize(unicode, &len);
Victor Stinner577db2c2011-10-11 22:12:48 +020014599 if (u == NULL)
14600 return NULL;
Victor Stinner71133ff2010-09-01 23:43:53 +000014601 /* Ensure we won't overflow the size. */
Victor Stinner57ffa9d2011-10-23 20:10:08 +020014602 if (len > ((PY_SSIZE_T_MAX / sizeof(Py_UNICODE)) - 1)) {
Victor Stinner71133ff2010-09-01 23:43:53 +000014603 PyErr_NoMemory();
14604 return NULL;
14605 }
Victor Stinner57ffa9d2011-10-23 20:10:08 +020014606 size = len + 1; /* copy the null character */
Victor Stinner71133ff2010-09-01 23:43:53 +000014607 size *= sizeof(Py_UNICODE);
14608 copy = PyMem_Malloc(size);
14609 if (copy == NULL) {
14610 PyErr_NoMemory();
14611 return NULL;
14612 }
Victor Stinner577db2c2011-10-11 22:12:48 +020014613 memcpy(copy, u, size);
Victor Stinner71133ff2010-09-01 23:43:53 +000014614 return copy;
14615}
Martin v. Löwis5b222132007-06-10 09:51:05 +000014616
Georg Brandl66c221e2010-10-14 07:04:07 +000014617/* A _string module, to export formatter_parser and formatter_field_name_split
14618 to the string.Formatter class implemented in Python. */
14619
14620static PyMethodDef _string_methods[] = {
14621 {"formatter_field_name_split", (PyCFunction) formatter_field_name_split,
14622 METH_O, PyDoc_STR("split the argument as a field name")},
14623 {"formatter_parser", (PyCFunction) formatter_parser,
14624 METH_O, PyDoc_STR("parse the argument as a format string")},
14625 {NULL, NULL}
14626};
14627
14628static struct PyModuleDef _string_module = {
14629 PyModuleDef_HEAD_INIT,
14630 "_string",
14631 PyDoc_STR("string helper module"),
14632 0,
14633 _string_methods,
14634 NULL,
14635 NULL,
14636 NULL,
14637 NULL
14638};
14639
14640PyMODINIT_FUNC
14641PyInit__string(void)
14642{
14643 return PyModule_Create(&_string_module);
14644}
14645
14646
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000014647#ifdef __cplusplus
14648}
14649#endif