blob: 35b424e33ab36276c99595efe776a5b8857023d1 [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,
Fred Drake785d14f2000-05-09 19:54:43 +00004modified by Marc-Andre Lemburg <mal@lemburg.com> according to the
Guido van Rossumd57fd912000-03-10 22:53:23 +00005Unicode Integration Proposal (see file Misc/unicode.txt).
6
Thomas Wouters477c8d52006-05-27 19:21:47 +00007Major speed upgrades to the method implementations at the Reykjavik
8NeedForSpeed sprint, by Fredrik Lundh and Andrew Dalke.
9
Guido van Rossum16b1ad92000-08-03 16:24:25 +000010Copyright (c) Corporation for National Research Initiatives.
Guido van Rossumd57fd912000-03-10 22:53:23 +000011
Fredrik Lundh0fdb90c2001-01-19 09:45:02 +000012--------------------------------------------------------------------
13The original string type implementation is:
Guido van Rossumd57fd912000-03-10 22:53:23 +000014
Benjamin Peterson29060642009-01-31 22:14:21 +000015 Copyright (c) 1999 by Secret Labs AB
16 Copyright (c) 1999 by Fredrik Lundh
Guido van Rossumd57fd912000-03-10 22:53:23 +000017
Fredrik Lundh0fdb90c2001-01-19 09:45:02 +000018By obtaining, using, and/or copying this software and/or its
19associated documentation, you agree that you have read, understood,
20and will comply with the following terms and conditions:
21
22Permission to use, copy, modify, and distribute this software and its
23associated documentation for any purpose and without fee is hereby
24granted, provided that the above copyright notice appears in all
25copies, and that both that copyright notice and this permission notice
26appear in supporting documentation, and that the name of Secret Labs
27AB or the author not be used in advertising or publicity pertaining to
28distribution of the software without specific, written prior
29permission.
30
31SECRET LABS AB AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH REGARD TO
32THIS SOFTWARE, INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY AND
33FITNESS. IN NO EVENT SHALL SECRET LABS AB OR THE AUTHOR BE LIABLE FOR
34ANY SPECIAL, INDIRECT OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
35WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
36ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT
37OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
38--------------------------------------------------------------------
39
40*/
Guido van Rossumd57fd912000-03-10 22:53:23 +000041
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000042#define PY_SSIZE_T_CLEAN
Guido van Rossumd57fd912000-03-10 22:53:23 +000043#include "Python.h"
Marc-André Lemburgd49e5b42000-06-30 14:58:20 +000044#include "ucnhash.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/* Limit for the Unicode object free list */
51
Christian Heimes2202f872008-02-06 14:31:34 +000052#define PyUnicode_MAXFREELIST 1024
Guido van Rossumd57fd912000-03-10 22:53:23 +000053
54/* Limit for the Unicode object free list stay alive optimization.
55
56 The implementation will keep allocated Unicode memory intact for
57 all objects on the free list having a size less than this
Tim Petersced69f82003-09-16 20:30:58 +000058 limit. This reduces malloc() overhead for small Unicode objects.
Guido van Rossumd57fd912000-03-10 22:53:23 +000059
Christian Heimes2202f872008-02-06 14:31:34 +000060 At worst this will result in PyUnicode_MAXFREELIST *
Guido van Rossumfd4b9572000-04-10 13:51:10 +000061 (sizeof(PyUnicodeObject) + KEEPALIVE_SIZE_LIMIT +
Guido van Rossumd57fd912000-03-10 22:53:23 +000062 malloc()-overhead) bytes of unused garbage.
63
64 Setting the limit to 0 effectively turns the feature off.
65
Guido van Rossumfd4b9572000-04-10 13:51:10 +000066 Note: This is an experimental feature ! If you get core dumps when
67 using Unicode objects, turn this feature off.
Guido van Rossumd57fd912000-03-10 22:53:23 +000068
69*/
70
Guido van Rossumfd4b9572000-04-10 13:51:10 +000071#define KEEPALIVE_SIZE_LIMIT 9
Guido van Rossumd57fd912000-03-10 22:53:23 +000072
73/* Endianness switches; defaults to little endian */
74
75#ifdef WORDS_BIGENDIAN
76# define BYTEORDER_IS_BIG_ENDIAN
77#else
78# define BYTEORDER_IS_LITTLE_ENDIAN
79#endif
80
Marc-André Lemburgd4ab4a52000-06-08 17:54:00 +000081/* --- Globals ------------------------------------------------------------
82
83 The globals are initialized by the _PyUnicode_Init() API and should
84 not be used before calling that API.
85
86*/
Guido van Rossumd57fd912000-03-10 22:53:23 +000087
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000088
89#ifdef __cplusplus
90extern "C" {
91#endif
92
Walter Dörwald16807132007-05-25 13:52:07 +000093/* This dictionary holds all interned unicode strings. Note that references
94 to strings in this dictionary are *not* counted in the string's ob_refcnt.
95 When the interned string reaches a refcnt of 0 the string deallocation
96 function will delete the reference from this dictionary.
97
98 Another way to look at this is that to say that the actual reference
Guido van Rossum98297ee2007-11-06 21:34:58 +000099 count of a string is: s->ob_refcnt + (s->state ? 2 : 0)
Walter Dörwald16807132007-05-25 13:52:07 +0000100*/
101static PyObject *interned;
102
Guido van Rossumd57fd912000-03-10 22:53:23 +0000103/* Free list for Unicode objects */
Christian Heimes2202f872008-02-06 14:31:34 +0000104static PyUnicodeObject *free_list;
105static int numfree;
Guido van Rossumd57fd912000-03-10 22:53:23 +0000106
Marc-André Lemburg8155e0e2001-04-23 14:44:21 +0000107/* The empty Unicode object is shared to improve performance. */
108static PyUnicodeObject *unicode_empty;
109
110/* Single character Unicode strings in the Latin-1 range are being
111 shared as well. */
112static PyUnicodeObject *unicode_latin1[256];
113
Christian Heimes190d79e2008-01-30 11:58:22 +0000114/* Fast detection of the most frequent whitespace characters */
115const unsigned char _Py_ascii_whitespace[] = {
Benjamin Peterson14339b62009-01-31 16:36:08 +0000116 0, 0, 0, 0, 0, 0, 0, 0,
Florent Xicluna806d8cf2010-03-30 19:34:18 +0000117/* case 0x0009: * CHARACTER TABULATION */
Christian Heimes1a8501c2008-10-02 19:56:01 +0000118/* case 0x000A: * LINE FEED */
Florent Xicluna806d8cf2010-03-30 19:34:18 +0000119/* case 0x000B: * LINE TABULATION */
Christian Heimes1a8501c2008-10-02 19:56:01 +0000120/* case 0x000C: * FORM FEED */
121/* case 0x000D: * CARRIAGE RETURN */
Benjamin Peterson14339b62009-01-31 16:36:08 +0000122 0, 1, 1, 1, 1, 1, 0, 0,
123 0, 0, 0, 0, 0, 0, 0, 0,
Christian Heimes1a8501c2008-10-02 19:56:01 +0000124/* case 0x001C: * FILE SEPARATOR */
125/* case 0x001D: * GROUP SEPARATOR */
126/* case 0x001E: * RECORD SEPARATOR */
127/* case 0x001F: * UNIT SEPARATOR */
Benjamin Peterson14339b62009-01-31 16:36:08 +0000128 0, 0, 0, 0, 1, 1, 1, 1,
Christian Heimes1a8501c2008-10-02 19:56:01 +0000129/* case 0x0020: * SPACE */
Benjamin Peterson14339b62009-01-31 16:36:08 +0000130 1, 0, 0, 0, 0, 0, 0, 0,
131 0, 0, 0, 0, 0, 0, 0, 0,
132 0, 0, 0, 0, 0, 0, 0, 0,
133 0, 0, 0, 0, 0, 0, 0, 0,
Christian Heimes190d79e2008-01-30 11:58:22 +0000134
Benjamin Peterson14339b62009-01-31 16:36:08 +0000135 0, 0, 0, 0, 0, 0, 0, 0,
136 0, 0, 0, 0, 0, 0, 0, 0,
137 0, 0, 0, 0, 0, 0, 0, 0,
138 0, 0, 0, 0, 0, 0, 0, 0,
139 0, 0, 0, 0, 0, 0, 0, 0,
140 0, 0, 0, 0, 0, 0, 0, 0,
141 0, 0, 0, 0, 0, 0, 0, 0,
142 0, 0, 0, 0, 0, 0, 0, 0
Christian Heimes190d79e2008-01-30 11:58:22 +0000143};
144
Martin v. Löwisdb12d452009-05-02 18:52:14 +0000145static PyObject *unicode_encode_call_errorhandler(const char *errors,
146 PyObject **errorHandler,const char *encoding, const char *reason,
147 const Py_UNICODE *unicode, Py_ssize_t size, PyObject **exceptionObject,
148 Py_ssize_t startpos, Py_ssize_t endpos, Py_ssize_t *newpos);
149
Victor Stinner31be90b2010-04-22 19:38:16 +0000150static void raise_encode_exception(PyObject **exceptionObject,
151 const char *encoding,
152 const Py_UNICODE *unicode, Py_ssize_t size,
153 Py_ssize_t startpos, Py_ssize_t endpos,
154 const char *reason);
155
Christian Heimes190d79e2008-01-30 11:58:22 +0000156/* Same for linebreaks */
157static unsigned char ascii_linebreak[] = {
Benjamin Peterson14339b62009-01-31 16:36:08 +0000158 0, 0, 0, 0, 0, 0, 0, 0,
Christian Heimes1a8501c2008-10-02 19:56:01 +0000159/* 0x000A, * LINE FEED */
Florent Xicluna806d8cf2010-03-30 19:34:18 +0000160/* 0x000B, * LINE TABULATION */
161/* 0x000C, * FORM FEED */
Christian Heimes1a8501c2008-10-02 19:56:01 +0000162/* 0x000D, * CARRIAGE RETURN */
Florent Xicluna806d8cf2010-03-30 19:34:18 +0000163 0, 0, 1, 1, 1, 1, 0, 0,
Benjamin Peterson14339b62009-01-31 16:36:08 +0000164 0, 0, 0, 0, 0, 0, 0, 0,
Christian Heimes1a8501c2008-10-02 19:56:01 +0000165/* 0x001C, * FILE SEPARATOR */
166/* 0x001D, * GROUP SEPARATOR */
167/* 0x001E, * RECORD SEPARATOR */
Benjamin Peterson14339b62009-01-31 16:36:08 +0000168 0, 0, 0, 0, 1, 1, 1, 0,
169 0, 0, 0, 0, 0, 0, 0, 0,
170 0, 0, 0, 0, 0, 0, 0, 0,
171 0, 0, 0, 0, 0, 0, 0, 0,
172 0, 0, 0, 0, 0, 0, 0, 0,
Christian Heimes190d79e2008-01-30 11:58:22 +0000173
Benjamin Peterson14339b62009-01-31 16:36:08 +0000174 0, 0, 0, 0, 0, 0, 0, 0,
175 0, 0, 0, 0, 0, 0, 0, 0,
176 0, 0, 0, 0, 0, 0, 0, 0,
177 0, 0, 0, 0, 0, 0, 0, 0,
178 0, 0, 0, 0, 0, 0, 0, 0,
179 0, 0, 0, 0, 0, 0, 0, 0,
180 0, 0, 0, 0, 0, 0, 0, 0,
181 0, 0, 0, 0, 0, 0, 0, 0
Christian Heimes190d79e2008-01-30 11:58:22 +0000182};
183
184
Martin v. Löwisce9b5a52001-06-27 06:28:56 +0000185Py_UNICODE
Marc-André Lemburg6c6bfb72001-07-20 17:39:11 +0000186PyUnicode_GetMax(void)
Martin v. Löwisce9b5a52001-06-27 06:28:56 +0000187{
Fredrik Lundh8f455852001-06-27 18:59:43 +0000188#ifdef Py_UNICODE_WIDE
Benjamin Peterson14339b62009-01-31 16:36:08 +0000189 return 0x10FFFF;
Martin v. Löwisce9b5a52001-06-27 06:28:56 +0000190#else
Benjamin Peterson14339b62009-01-31 16:36:08 +0000191 /* This is actually an illegal character, so it should
192 not be passed to unichr. */
193 return 0xFFFF;
Martin v. Löwisce9b5a52001-06-27 06:28:56 +0000194#endif
195}
196
Thomas Wouters477c8d52006-05-27 19:21:47 +0000197/* --- Bloom Filters ----------------------------------------------------- */
198
199/* stuff to implement simple "bloom filters" for Unicode characters.
200 to keep things simple, we use a single bitmask, using the least 5
201 bits from each unicode characters as the bit index. */
202
203/* the linebreak mask is set up by Unicode_Init below */
204
Antoine Pitrouf068f942010-01-13 14:19:12 +0000205#if LONG_BIT >= 128
206#define BLOOM_WIDTH 128
207#elif LONG_BIT >= 64
208#define BLOOM_WIDTH 64
209#elif LONG_BIT >= 32
210#define BLOOM_WIDTH 32
211#else
212#error "LONG_BIT is smaller than 32"
213#endif
214
Thomas Wouters477c8d52006-05-27 19:21:47 +0000215#define BLOOM_MASK unsigned long
216
217static BLOOM_MASK bloom_linebreak;
218
Antoine Pitrouf068f942010-01-13 14:19:12 +0000219#define BLOOM_ADD(mask, ch) ((mask |= (1UL << ((ch) & (BLOOM_WIDTH - 1)))))
220#define BLOOM(mask, ch) ((mask & (1UL << ((ch) & (BLOOM_WIDTH - 1)))))
Thomas Wouters477c8d52006-05-27 19:21:47 +0000221
Benjamin Peterson29060642009-01-31 22:14:21 +0000222#define BLOOM_LINEBREAK(ch) \
223 ((ch) < 128U ? ascii_linebreak[(ch)] : \
224 (BLOOM(bloom_linebreak, (ch)) && Py_UNICODE_ISLINEBREAK(ch)))
Thomas Wouters477c8d52006-05-27 19:21:47 +0000225
226Py_LOCAL_INLINE(BLOOM_MASK) make_bloom_mask(Py_UNICODE* ptr, Py_ssize_t len)
227{
228 /* calculate simple bloom-style bitmask for a given unicode string */
229
Antoine Pitrouf068f942010-01-13 14:19:12 +0000230 BLOOM_MASK mask;
Thomas Wouters477c8d52006-05-27 19:21:47 +0000231 Py_ssize_t i;
232
233 mask = 0;
234 for (i = 0; i < len; i++)
Antoine Pitrouf2c54842010-01-13 08:07:53 +0000235 BLOOM_ADD(mask, ptr[i]);
Thomas Wouters477c8d52006-05-27 19:21:47 +0000236
237 return mask;
238}
239
240Py_LOCAL_INLINE(int) unicode_member(Py_UNICODE chr, Py_UNICODE* set, Py_ssize_t setlen)
241{
242 Py_ssize_t i;
243
244 for (i = 0; i < setlen; i++)
245 if (set[i] == chr)
246 return 1;
247
248 return 0;
249}
250
Benjamin Peterson29060642009-01-31 22:14:21 +0000251#define BLOOM_MEMBER(mask, chr, set, setlen) \
Thomas Wouters477c8d52006-05-27 19:21:47 +0000252 BLOOM(mask, chr) && unicode_member(chr, set, setlen)
253
Guido van Rossumd57fd912000-03-10 22:53:23 +0000254/* --- Unicode Object ----------------------------------------------------- */
255
256static
Marc-André Lemburg8155e0e2001-04-23 14:44:21 +0000257int unicode_resize(register PyUnicodeObject *unicode,
Benjamin Peterson29060642009-01-31 22:14:21 +0000258 Py_ssize_t length)
Guido van Rossumd57fd912000-03-10 22:53:23 +0000259{
260 void *oldstr;
Tim Petersced69f82003-09-16 20:30:58 +0000261
Guido van Rossumfd4b9572000-04-10 13:51:10 +0000262 /* Shortcut if there's nothing much to do. */
Guido van Rossumd57fd912000-03-10 22:53:23 +0000263 if (unicode->length == length)
Benjamin Peterson29060642009-01-31 22:14:21 +0000264 goto reset;
Guido van Rossumd57fd912000-03-10 22:53:23 +0000265
Marc-André Lemburg8155e0e2001-04-23 14:44:21 +0000266 /* Resizing shared object (unicode_empty or single character
267 objects) in-place is not allowed. Use PyUnicode_Resize()
268 instead ! */
Thomas Wouters477c8d52006-05-27 19:21:47 +0000269
Benjamin Peterson14339b62009-01-31 16:36:08 +0000270 if (unicode == unicode_empty ||
Benjamin Peterson29060642009-01-31 22:14:21 +0000271 (unicode->length == 1 &&
272 unicode->str[0] < 256U &&
273 unicode_latin1[unicode->str[0]] == unicode)) {
Guido van Rossumd57fd912000-03-10 22:53:23 +0000274 PyErr_SetString(PyExc_SystemError,
Benjamin Peterson142957c2008-07-04 19:55:29 +0000275 "can't resize shared str objects");
Guido van Rossumd57fd912000-03-10 22:53:23 +0000276 return -1;
277 }
278
Thomas Wouters477c8d52006-05-27 19:21:47 +0000279 /* We allocate one more byte to make sure the string is Ux0000 terminated.
280 The overallocation is also used by fastsearch, which assumes that it's
281 safe to look at str[length] (without making any assumptions about what
282 it contains). */
283
Guido van Rossumd57fd912000-03-10 22:53:23 +0000284 oldstr = unicode->str;
Christian Heimesb186d002008-03-18 15:15:01 +0000285 unicode->str = PyObject_REALLOC(unicode->str,
Benjamin Peterson29060642009-01-31 22:14:21 +0000286 sizeof(Py_UNICODE) * (length + 1));
Guido van Rossumd57fd912000-03-10 22:53:23 +0000287 if (!unicode->str) {
Benjamin Peterson29060642009-01-31 22:14:21 +0000288 unicode->str = (Py_UNICODE *)oldstr;
Guido van Rossumd57fd912000-03-10 22:53:23 +0000289 PyErr_NoMemory();
290 return -1;
291 }
292 unicode->str[length] = 0;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000293 unicode->length = length;
Guido van Rossumd57fd912000-03-10 22:53:23 +0000294
Benjamin Peterson29060642009-01-31 22:14:21 +0000295 reset:
Guido van Rossumd57fd912000-03-10 22:53:23 +0000296 /* Reset the object caches */
Marc-André Lemburgbff879c2000-08-03 18:46:08 +0000297 if (unicode->defenc) {
Georg Brandl8ee604b2010-07-29 14:23:06 +0000298 Py_CLEAR(unicode->defenc);
Guido van Rossumd57fd912000-03-10 22:53:23 +0000299 }
300 unicode->hash = -1;
Tim Petersced69f82003-09-16 20:30:58 +0000301
Guido van Rossumd57fd912000-03-10 22:53:23 +0000302 return 0;
303}
304
305/* We allocate one more byte to make sure the string is
Martin v. Löwis47383402007-08-15 07:32:56 +0000306 Ux0000 terminated; some code (e.g. new_identifier)
307 relies on that.
Guido van Rossumd57fd912000-03-10 22:53:23 +0000308
309 XXX This allocator could further be enhanced by assuring that the
Benjamin Peterson29060642009-01-31 22:14:21 +0000310 free list never reduces its size below 1.
Guido van Rossumd57fd912000-03-10 22:53:23 +0000311
312*/
313
314static
Martin v. Löwis18e16552006-02-15 17:27:45 +0000315PyUnicodeObject *_PyUnicode_New(Py_ssize_t length)
Guido van Rossumd57fd912000-03-10 22:53:23 +0000316{
317 register PyUnicodeObject *unicode;
318
Thomas Wouters477c8d52006-05-27 19:21:47 +0000319 /* Optimization for empty strings */
Guido van Rossumd57fd912000-03-10 22:53:23 +0000320 if (length == 0 && unicode_empty != NULL) {
321 Py_INCREF(unicode_empty);
322 return unicode_empty;
323 }
324
Neal Norwitz3ce5d922008-08-24 07:08:55 +0000325 /* Ensure we won't overflow the size. */
326 if (length > ((PY_SSIZE_T_MAX / sizeof(Py_UNICODE)) - 1)) {
327 return (PyUnicodeObject *)PyErr_NoMemory();
328 }
329
Guido van Rossumd57fd912000-03-10 22:53:23 +0000330 /* Unicode freelist & memory allocation */
Christian Heimes2202f872008-02-06 14:31:34 +0000331 if (free_list) {
332 unicode = free_list;
333 free_list = *(PyUnicodeObject **)unicode;
334 numfree--;
Benjamin Peterson29060642009-01-31 22:14:21 +0000335 if (unicode->str) {
336 /* Keep-Alive optimization: we only upsize the buffer,
337 never downsize it. */
338 if ((unicode->length < length) &&
Jeremy Hyltondeb2dc62003-09-16 03:41:45 +0000339 unicode_resize(unicode, length) < 0) {
Benjamin Peterson29060642009-01-31 22:14:21 +0000340 PyObject_DEL(unicode->str);
341 unicode->str = NULL;
342 }
Benjamin Peterson14339b62009-01-31 16:36:08 +0000343 }
Guido van Rossumad98db12001-06-14 17:52:02 +0000344 else {
Benjamin Peterson29060642009-01-31 22:14:21 +0000345 size_t new_size = sizeof(Py_UNICODE) * ((size_t)length + 1);
346 unicode->str = (Py_UNICODE*) PyObject_MALLOC(new_size);
Guido van Rossumad98db12001-06-14 17:52:02 +0000347 }
348 PyObject_INIT(unicode, &PyUnicode_Type);
Guido van Rossumd57fd912000-03-10 22:53:23 +0000349 }
350 else {
Benjamin Peterson29060642009-01-31 22:14:21 +0000351 size_t new_size;
Neil Schemenauer58aa8612002-04-12 03:07:20 +0000352 unicode = PyObject_New(PyUnicodeObject, &PyUnicode_Type);
Guido van Rossumd57fd912000-03-10 22:53:23 +0000353 if (unicode == NULL)
354 return NULL;
Benjamin Peterson29060642009-01-31 22:14:21 +0000355 new_size = sizeof(Py_UNICODE) * ((size_t)length + 1);
356 unicode->str = (Py_UNICODE*) PyObject_MALLOC(new_size);
Guido van Rossumd57fd912000-03-10 22:53:23 +0000357 }
358
Guido van Rossum3c1bb802000-04-27 20:13:50 +0000359 if (!unicode->str) {
Benjamin Peterson29060642009-01-31 22:14:21 +0000360 PyErr_NoMemory();
361 goto onError;
Guido van Rossum3c1bb802000-04-27 20:13:50 +0000362 }
Jeremy Hyltond8082792003-09-16 19:41:39 +0000363 /* Initialize the first element to guard against cases where
Tim Petersced69f82003-09-16 20:30:58 +0000364 * the caller fails before initializing str -- unicode_resize()
365 * reads str[0], and the Keep-Alive optimization can keep memory
366 * allocated for str alive across a call to unicode_dealloc(unicode).
367 * We don't want unicode_resize to read uninitialized memory in
368 * that case.
369 */
Jeremy Hyltond8082792003-09-16 19:41:39 +0000370 unicode->str[0] = 0;
Guido van Rossumd57fd912000-03-10 22:53:23 +0000371 unicode->str[length] = 0;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000372 unicode->length = length;
Guido van Rossumd57fd912000-03-10 22:53:23 +0000373 unicode->hash = -1;
Walter Dörwald16807132007-05-25 13:52:07 +0000374 unicode->state = 0;
Marc-André Lemburgbff879c2000-08-03 18:46:08 +0000375 unicode->defenc = NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +0000376 return unicode;
Barry Warsaw51ac5802000-03-20 16:36:48 +0000377
Benjamin Peterson29060642009-01-31 22:14:21 +0000378 onError:
Amaury Forgeot d'Arc7888d082008-08-01 01:06:32 +0000379 /* XXX UNREF/NEWREF interface should be more symmetrical */
380 _Py_DEC_REFTOTAL;
Barry Warsaw51ac5802000-03-20 16:36:48 +0000381 _Py_ForgetReference((PyObject *)unicode);
Neil Schemenauer58aa8612002-04-12 03:07:20 +0000382 PyObject_Del(unicode);
Barry Warsaw51ac5802000-03-20 16:36:48 +0000383 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +0000384}
385
386static
Guido van Rossum9475a232001-10-05 20:51:39 +0000387void unicode_dealloc(register PyUnicodeObject *unicode)
Guido van Rossumd57fd912000-03-10 22:53:23 +0000388{
Walter Dörwald16807132007-05-25 13:52:07 +0000389 switch (PyUnicode_CHECK_INTERNED(unicode)) {
Benjamin Peterson29060642009-01-31 22:14:21 +0000390 case SSTATE_NOT_INTERNED:
391 break;
Walter Dörwald16807132007-05-25 13:52:07 +0000392
Benjamin Peterson29060642009-01-31 22:14:21 +0000393 case SSTATE_INTERNED_MORTAL:
394 /* revive dead object temporarily for DelItem */
395 Py_REFCNT(unicode) = 3;
396 if (PyDict_DelItem(interned, (PyObject *)unicode) != 0)
397 Py_FatalError(
398 "deletion of interned string failed");
399 break;
Walter Dörwald16807132007-05-25 13:52:07 +0000400
Benjamin Peterson29060642009-01-31 22:14:21 +0000401 case SSTATE_INTERNED_IMMORTAL:
402 Py_FatalError("Immortal interned string died.");
Walter Dörwald16807132007-05-25 13:52:07 +0000403
Benjamin Peterson29060642009-01-31 22:14:21 +0000404 default:
405 Py_FatalError("Inconsistent interned string state.");
Walter Dörwald16807132007-05-25 13:52:07 +0000406 }
407
Guido van Rossum604ddf82001-12-06 20:03:56 +0000408 if (PyUnicode_CheckExact(unicode) &&
Benjamin Peterson29060642009-01-31 22:14:21 +0000409 numfree < PyUnicode_MAXFREELIST) {
Guido van Rossumfd4b9572000-04-10 13:51:10 +0000410 /* Keep-Alive optimization */
Benjamin Peterson29060642009-01-31 22:14:21 +0000411 if (unicode->length >= KEEPALIVE_SIZE_LIMIT) {
412 PyObject_DEL(unicode->str);
413 unicode->str = NULL;
414 unicode->length = 0;
415 }
416 if (unicode->defenc) {
Georg Brandl8ee604b2010-07-29 14:23:06 +0000417 Py_CLEAR(unicode->defenc);
Benjamin Peterson29060642009-01-31 22:14:21 +0000418 }
419 /* Add to free list */
Christian Heimes2202f872008-02-06 14:31:34 +0000420 *(PyUnicodeObject **)unicode = free_list;
421 free_list = unicode;
422 numfree++;
Guido van Rossumd57fd912000-03-10 22:53:23 +0000423 }
424 else {
Benjamin Peterson29060642009-01-31 22:14:21 +0000425 PyObject_DEL(unicode->str);
426 Py_XDECREF(unicode->defenc);
427 Py_TYPE(unicode)->tp_free((PyObject *)unicode);
Guido van Rossumd57fd912000-03-10 22:53:23 +0000428 }
429}
430
Alexandre Vassalottiaa0e5312008-12-27 06:43:58 +0000431static
432int _PyUnicode_Resize(PyUnicodeObject **unicode, Py_ssize_t length)
Marc-André Lemburg8155e0e2001-04-23 14:44:21 +0000433{
434 register PyUnicodeObject *v;
435
436 /* Argument checks */
437 if (unicode == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +0000438 PyErr_BadInternalCall();
439 return -1;
Marc-André Lemburg8155e0e2001-04-23 14:44:21 +0000440 }
Alexandre Vassalottiaa0e5312008-12-27 06:43:58 +0000441 v = *unicode;
Christian Heimes90aa7642007-12-19 02:45:37 +0000442 if (v == NULL || !PyUnicode_Check(v) || Py_REFCNT(v) != 1 || length < 0) {
Benjamin Peterson29060642009-01-31 22:14:21 +0000443 PyErr_BadInternalCall();
444 return -1;
Marc-André Lemburg8155e0e2001-04-23 14:44:21 +0000445 }
446
447 /* Resizing unicode_empty and single character objects is not
448 possible since these are being shared. We simply return a fresh
449 copy with the same Unicode content. */
Tim Petersced69f82003-09-16 20:30:58 +0000450 if (v->length != length &&
Benjamin Peterson29060642009-01-31 22:14:21 +0000451 (v == unicode_empty || v->length == 1)) {
452 PyUnicodeObject *w = _PyUnicode_New(length);
453 if (w == NULL)
454 return -1;
455 Py_UNICODE_COPY(w->str, v->str,
456 length < v->length ? length : v->length);
457 Py_DECREF(*unicode);
458 *unicode = w;
459 return 0;
Marc-André Lemburg8155e0e2001-04-23 14:44:21 +0000460 }
461
462 /* Note that we don't have to modify *unicode for unshared Unicode
463 objects, since we can modify them in-place. */
464 return unicode_resize(v, length);
465}
466
Alexandre Vassalottiaa0e5312008-12-27 06:43:58 +0000467int PyUnicode_Resize(PyObject **unicode, Py_ssize_t length)
468{
469 return _PyUnicode_Resize((PyUnicodeObject **)unicode, length);
470}
Marc-André Lemburg8155e0e2001-04-23 14:44:21 +0000471
Guido van Rossumd57fd912000-03-10 22:53:23 +0000472PyObject *PyUnicode_FromUnicode(const Py_UNICODE *u,
Benjamin Peterson29060642009-01-31 22:14:21 +0000473 Py_ssize_t size)
Guido van Rossumd57fd912000-03-10 22:53:23 +0000474{
475 PyUnicodeObject *unicode;
476
Marc-André Lemburg8155e0e2001-04-23 14:44:21 +0000477 /* If the Unicode data is known at construction time, we can apply
478 some optimizations which share commonly used objects. */
479 if (u != NULL) {
480
Benjamin Peterson29060642009-01-31 22:14:21 +0000481 /* Optimization for empty strings */
482 if (size == 0 && unicode_empty != NULL) {
483 Py_INCREF(unicode_empty);
484 return (PyObject *)unicode_empty;
Benjamin Peterson14339b62009-01-31 16:36:08 +0000485 }
Benjamin Peterson29060642009-01-31 22:14:21 +0000486
487 /* Single character Unicode objects in the Latin-1 range are
488 shared when using this constructor */
489 if (size == 1 && *u < 256) {
490 unicode = unicode_latin1[*u];
491 if (!unicode) {
492 unicode = _PyUnicode_New(1);
493 if (!unicode)
494 return NULL;
495 unicode->str[0] = *u;
496 unicode_latin1[*u] = unicode;
497 }
498 Py_INCREF(unicode);
499 return (PyObject *)unicode;
500 }
Marc-André Lemburg8155e0e2001-04-23 14:44:21 +0000501 }
Tim Petersced69f82003-09-16 20:30:58 +0000502
Guido van Rossumd57fd912000-03-10 22:53:23 +0000503 unicode = _PyUnicode_New(size);
504 if (!unicode)
505 return NULL;
506
507 /* Copy the Unicode data into the new object */
508 if (u != NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +0000509 Py_UNICODE_COPY(unicode->str, u, size);
Guido van Rossumd57fd912000-03-10 22:53:23 +0000510
511 return (PyObject *)unicode;
512}
513
Walter Dörwaldd2034312007-05-18 16:29:38 +0000514PyObject *PyUnicode_FromStringAndSize(const char *u, Py_ssize_t size)
Walter Dörwaldacaa5a12007-05-05 12:00:46 +0000515{
516 PyUnicodeObject *unicode;
Christian Heimes33fe8092008-04-13 13:53:33 +0000517
Benjamin Peterson14339b62009-01-31 16:36:08 +0000518 if (size < 0) {
519 PyErr_SetString(PyExc_SystemError,
Benjamin Peterson29060642009-01-31 22:14:21 +0000520 "Negative size passed to PyUnicode_FromStringAndSize");
Benjamin Peterson14339b62009-01-31 16:36:08 +0000521 return NULL;
522 }
Christian Heimes33fe8092008-04-13 13:53:33 +0000523
Walter Dörwaldacaa5a12007-05-05 12:00:46 +0000524 /* If the Unicode data is known at construction time, we can apply
Martin v. Löwis9c121062007-08-05 20:26:11 +0000525 some optimizations which share commonly used objects.
526 Also, this means the input must be UTF-8, so fall back to the
527 UTF-8 decoder at the end. */
Walter Dörwaldacaa5a12007-05-05 12:00:46 +0000528 if (u != NULL) {
529
Benjamin Peterson29060642009-01-31 22:14:21 +0000530 /* Optimization for empty strings */
531 if (size == 0 && unicode_empty != NULL) {
532 Py_INCREF(unicode_empty);
533 return (PyObject *)unicode_empty;
Benjamin Peterson14339b62009-01-31 16:36:08 +0000534 }
Benjamin Peterson29060642009-01-31 22:14:21 +0000535
536 /* Single characters are shared when using this constructor.
537 Restrict to ASCII, since the input must be UTF-8. */
538 if (size == 1 && Py_CHARMASK(*u) < 128) {
539 unicode = unicode_latin1[Py_CHARMASK(*u)];
540 if (!unicode) {
541 unicode = _PyUnicode_New(1);
542 if (!unicode)
543 return NULL;
544 unicode->str[0] = Py_CHARMASK(*u);
545 unicode_latin1[Py_CHARMASK(*u)] = unicode;
546 }
547 Py_INCREF(unicode);
548 return (PyObject *)unicode;
549 }
Martin v. Löwis9c121062007-08-05 20:26:11 +0000550
551 return PyUnicode_DecodeUTF8(u, size, NULL);
Walter Dörwaldacaa5a12007-05-05 12:00:46 +0000552 }
553
Walter Dörwald55507312007-05-18 13:12:10 +0000554 unicode = _PyUnicode_New(size);
Walter Dörwaldacaa5a12007-05-05 12:00:46 +0000555 if (!unicode)
556 return NULL;
557
Walter Dörwaldacaa5a12007-05-05 12:00:46 +0000558 return (PyObject *)unicode;
559}
560
Walter Dörwaldd2034312007-05-18 16:29:38 +0000561PyObject *PyUnicode_FromString(const char *u)
562{
563 size_t size = strlen(u);
564 if (size > PY_SSIZE_T_MAX) {
565 PyErr_SetString(PyExc_OverflowError, "input too long");
566 return NULL;
567 }
568
569 return PyUnicode_FromStringAndSize(u, size);
570}
571
Guido van Rossumd57fd912000-03-10 22:53:23 +0000572#ifdef HAVE_WCHAR_H
573
Mark Dickinson081dfee2009-03-18 14:47:41 +0000574#if (Py_UNICODE_SIZE == 2) && defined(SIZEOF_WCHAR_T) && (SIZEOF_WCHAR_T == 4)
575# define CONVERT_WCHAR_TO_SURROGATES
576#endif
577
578#ifdef CONVERT_WCHAR_TO_SURROGATES
579
580/* Here sizeof(wchar_t) is 4 but Py_UNICODE_SIZE == 2, so we need
581 to convert from UTF32 to UTF16. */
582
583PyObject *PyUnicode_FromWideChar(register const wchar_t *w,
584 Py_ssize_t size)
585{
586 PyUnicodeObject *unicode;
587 register Py_ssize_t i;
588 Py_ssize_t alloc;
589 const wchar_t *orig_w;
590
591 if (w == NULL) {
592 if (size == 0)
593 return PyUnicode_FromStringAndSize(NULL, 0);
594 PyErr_BadInternalCall();
595 return NULL;
596 }
597
598 if (size == -1) {
599 size = wcslen(w);
600 }
601
602 alloc = size;
603 orig_w = w;
604 for (i = size; i > 0; i--) {
605 if (*w > 0xFFFF)
606 alloc++;
607 w++;
608 }
609 w = orig_w;
610 unicode = _PyUnicode_New(alloc);
611 if (!unicode)
612 return NULL;
613
614 /* Copy the wchar_t data into the new object */
615 {
616 register Py_UNICODE *u;
617 u = PyUnicode_AS_UNICODE(unicode);
618 for (i = size; i > 0; i--) {
619 if (*w > 0xFFFF) {
620 wchar_t ordinal = *w++;
621 ordinal -= 0x10000;
622 *u++ = 0xD800 | (ordinal >> 10);
623 *u++ = 0xDC00 | (ordinal & 0x3FF);
624 }
625 else
626 *u++ = *w++;
627 }
628 }
629 return (PyObject *)unicode;
630}
631
632#else
633
Guido van Rossumd57fd912000-03-10 22:53:23 +0000634PyObject *PyUnicode_FromWideChar(register const wchar_t *w,
Benjamin Peterson29060642009-01-31 22:14:21 +0000635 Py_ssize_t size)
Guido van Rossumd57fd912000-03-10 22:53:23 +0000636{
637 PyUnicodeObject *unicode;
638
639 if (w == NULL) {
Martin v. Löwis790465f2008-04-05 20:41:37 +0000640 if (size == 0)
641 return PyUnicode_FromStringAndSize(NULL, 0);
Benjamin Peterson29060642009-01-31 22:14:21 +0000642 PyErr_BadInternalCall();
643 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +0000644 }
645
Martin v. Löwis790465f2008-04-05 20:41:37 +0000646 if (size == -1) {
647 size = wcslen(w);
648 }
649
Guido van Rossumd57fd912000-03-10 22:53:23 +0000650 unicode = _PyUnicode_New(size);
651 if (!unicode)
652 return NULL;
653
654 /* Copy the wchar_t data into the new object */
Daniel Stutzbach8515eae2010-08-24 21:57:33 +0000655#if Py_UNICODE_SIZE == SIZEOF_WCHAR_T
Guido van Rossumd57fd912000-03-10 22:53:23 +0000656 memcpy(unicode->str, w, size * sizeof(wchar_t));
Tim Petersced69f82003-09-16 20:30:58 +0000657#else
Guido van Rossumd57fd912000-03-10 22:53:23 +0000658 {
Benjamin Peterson29060642009-01-31 22:14:21 +0000659 register Py_UNICODE *u;
660 register Py_ssize_t i;
661 u = PyUnicode_AS_UNICODE(unicode);
662 for (i = size; i > 0; i--)
663 *u++ = *w++;
Guido van Rossumd57fd912000-03-10 22:53:23 +0000664 }
665#endif
666
667 return (PyObject *)unicode;
668}
669
Mark Dickinson081dfee2009-03-18 14:47:41 +0000670#endif /* CONVERT_WCHAR_TO_SURROGATES */
671
672#undef CONVERT_WCHAR_TO_SURROGATES
673
Walter Dörwald346737f2007-05-31 10:44:43 +0000674static void
Mark Dickinson6ce4a9a2009-11-16 17:00:11 +0000675makefmt(char *fmt, int longflag, int longlongflag, int size_tflag,
676 int zeropad, int width, int precision, char c)
Walter Dörwald346737f2007-05-31 10:44:43 +0000677{
Benjamin Peterson14339b62009-01-31 16:36:08 +0000678 *fmt++ = '%';
679 if (width) {
680 if (zeropad)
681 *fmt++ = '0';
682 fmt += sprintf(fmt, "%d", width);
683 }
684 if (precision)
685 fmt += sprintf(fmt, ".%d", precision);
686 if (longflag)
687 *fmt++ = 'l';
Mark Dickinson6ce4a9a2009-11-16 17:00:11 +0000688 else if (longlongflag) {
689 /* longlongflag should only ever be nonzero on machines with
690 HAVE_LONG_LONG defined */
691#ifdef HAVE_LONG_LONG
692 char *f = PY_FORMAT_LONG_LONG;
693 while (*f)
694 *fmt++ = *f++;
695#else
696 /* we shouldn't ever get here */
697 assert(0);
698 *fmt++ = 'l';
699#endif
700 }
Benjamin Peterson14339b62009-01-31 16:36:08 +0000701 else if (size_tflag) {
702 char *f = PY_FORMAT_SIZE_T;
703 while (*f)
704 *fmt++ = *f++;
705 }
706 *fmt++ = c;
707 *fmt = '\0';
Walter Dörwald346737f2007-05-31 10:44:43 +0000708}
709
Walter Dörwaldd2034312007-05-18 16:29:38 +0000710#define appendstring(string) {for (copy = string;*copy;) *s++ = *copy++;}
711
Mark Dickinson6ce4a9a2009-11-16 17:00:11 +0000712/* size of fixed-size buffer for formatting single arguments */
713#define ITEM_BUFFER_LEN 21
714/* maximum number of characters required for output of %ld. 21 characters
715 allows for 64-bit integers (in decimal) and an optional sign. */
716#define MAX_LONG_CHARS 21
717/* maximum number of characters required for output of %lld.
718 We need at most ceil(log10(256)*SIZEOF_LONG_LONG) digits,
719 plus 1 for the sign. 53/22 is an upper bound for log10(256). */
720#define MAX_LONG_LONG_CHARS (2 + (SIZEOF_LONG_LONG*53-1) / 22)
721
Walter Dörwaldd2034312007-05-18 16:29:38 +0000722PyObject *
723PyUnicode_FromFormatV(const char *format, va_list vargs)
724{
Benjamin Peterson14339b62009-01-31 16:36:08 +0000725 va_list count;
726 Py_ssize_t callcount = 0;
727 PyObject **callresults = NULL;
728 PyObject **callresult = NULL;
729 Py_ssize_t n = 0;
730 int width = 0;
731 int precision = 0;
732 int zeropad;
733 const char* f;
734 Py_UNICODE *s;
735 PyObject *string;
736 /* used by sprintf */
Mark Dickinson6ce4a9a2009-11-16 17:00:11 +0000737 char buffer[ITEM_BUFFER_LEN+1];
Benjamin Peterson14339b62009-01-31 16:36:08 +0000738 /* use abuffer instead of buffer, if we need more space
739 * (which can happen if there's a format specifier with width). */
740 char *abuffer = NULL;
741 char *realbuffer;
742 Py_ssize_t abuffersize = 0;
Mark Dickinson6ce4a9a2009-11-16 17:00:11 +0000743 char fmt[61]; /* should be enough for %0width.precisionlld */
Benjamin Peterson14339b62009-01-31 16:36:08 +0000744 const char *copy;
Walter Dörwaldd2034312007-05-18 16:29:38 +0000745
Victor Stinner4a2b7a12010-08-13 14:03:48 +0000746 Py_VA_COPY(count, vargs);
Walter Dörwaldc1651a02009-05-03 22:55:55 +0000747 /* step 1: count the number of %S/%R/%A/%s format specifications
748 * (we call PyObject_Str()/PyObject_Repr()/PyObject_ASCII()/
749 * PyUnicode_DecodeUTF8() for these objects once during step 3 and put the
750 * result in an array) */
Benjamin Peterson14339b62009-01-31 16:36:08 +0000751 for (f = format; *f; f++) {
Walter Dörwaldc1651a02009-05-03 22:55:55 +0000752 if (*f == '%') {
753 if (*(f+1)=='%')
754 continue;
Victor Stinner2b574a22011-03-01 22:48:49 +0000755 if (*(f+1)=='S' || *(f+1)=='R' || *(f+1)=='A' || *(f+1) == 'V')
Walter Dörwaldc1651a02009-05-03 22:55:55 +0000756 ++callcount;
David Malcolm96960882010-11-05 17:23:41 +0000757 while (Py_ISDIGIT((unsigned)*f))
Walter Dörwaldc1651a02009-05-03 22:55:55 +0000758 width = (width*10) + *f++ - '0';
David Malcolm96960882010-11-05 17:23:41 +0000759 while (*++f && *f != '%' && !Py_ISALPHA((unsigned)*f))
Walter Dörwaldc1651a02009-05-03 22:55:55 +0000760 ;
761 if (*f == 's')
762 ++callcount;
763 }
Benjamin Peterson9be0b2e2010-09-12 03:40:54 +0000764 else if (128 <= (unsigned char)*f) {
765 PyErr_Format(PyExc_ValueError,
766 "PyUnicode_FromFormatV() expects an ASCII-encoded format "
Victor Stinner4c7db312010-09-12 07:51:18 +0000767 "string, got a non-ASCII byte: 0x%02x",
Benjamin Peterson9be0b2e2010-09-12 03:40:54 +0000768 (unsigned char)*f);
Benjamin Petersond4ac96a2010-09-12 16:40:53 +0000769 return NULL;
Benjamin Peterson9be0b2e2010-09-12 03:40:54 +0000770 }
Benjamin Peterson14339b62009-01-31 16:36:08 +0000771 }
772 /* step 2: allocate memory for the results of
Walter Dörwaldc1651a02009-05-03 22:55:55 +0000773 * PyObject_Str()/PyObject_Repr()/PyUnicode_DecodeUTF8() calls */
Benjamin Peterson14339b62009-01-31 16:36:08 +0000774 if (callcount) {
775 callresults = PyObject_Malloc(sizeof(PyObject *)*callcount);
776 if (!callresults) {
777 PyErr_NoMemory();
778 return NULL;
779 }
780 callresult = callresults;
781 }
782 /* step 3: figure out how large a buffer we need */
783 for (f = format; *f; f++) {
784 if (*f == '%') {
Mark Dickinson6ce4a9a2009-11-16 17:00:11 +0000785#ifdef HAVE_LONG_LONG
786 int longlongflag = 0;
787#endif
Benjamin Peterson14339b62009-01-31 16:36:08 +0000788 const char* p = f;
789 width = 0;
David Malcolm96960882010-11-05 17:23:41 +0000790 while (Py_ISDIGIT((unsigned)*f))
Benjamin Peterson14339b62009-01-31 16:36:08 +0000791 width = (width*10) + *f++ - '0';
David Malcolm96960882010-11-05 17:23:41 +0000792 while (*++f && *f != '%' && !Py_ISALPHA((unsigned)*f))
Benjamin Peterson14339b62009-01-31 16:36:08 +0000793 ;
Walter Dörwaldd2034312007-05-18 16:29:38 +0000794
Benjamin Peterson14339b62009-01-31 16:36:08 +0000795 /* skip the 'l' or 'z' in {%ld, %zd, %lu, %zu} since
796 * they don't affect the amount of space we reserve.
797 */
Mark Dickinson6ce4a9a2009-11-16 17:00:11 +0000798 if (*f == 'l') {
799 if (f[1] == 'd' || f[1] == 'u') {
800 ++f;
801 }
802#ifdef HAVE_LONG_LONG
803 else if (f[1] == 'l' &&
804 (f[2] == 'd' || f[2] == 'u')) {
805 longlongflag = 1;
806 f += 2;
807 }
808#endif
809 }
810 else if (*f == 'z' && (f[1] == 'd' || f[1] == 'u')) {
Benjamin Peterson29060642009-01-31 22:14:21 +0000811 ++f;
Mark Dickinson6ce4a9a2009-11-16 17:00:11 +0000812 }
Walter Dörwaldd2034312007-05-18 16:29:38 +0000813
Benjamin Peterson14339b62009-01-31 16:36:08 +0000814 switch (*f) {
815 case 'c':
Victor Stinner659eb842011-02-23 12:14:22 +0000816 {
817#ifndef Py_UNICODE_WIDE
818 int ordinal = va_arg(count, int);
819 if (ordinal > 0xffff)
820 n += 2;
821 else
822 n++;
823#else
Benjamin Peterson14339b62009-01-31 16:36:08 +0000824 (void)va_arg(count, int);
Victor Stinner659eb842011-02-23 12:14:22 +0000825 n++;
826#endif
827 break;
828 }
Benjamin Peterson14339b62009-01-31 16:36:08 +0000829 case '%':
830 n++;
831 break;
832 case 'd': case 'u': case 'i': case 'x':
833 (void) va_arg(count, int);
Mark Dickinson6ce4a9a2009-11-16 17:00:11 +0000834#ifdef HAVE_LONG_LONG
835 if (longlongflag) {
836 if (width < MAX_LONG_LONG_CHARS)
837 width = MAX_LONG_LONG_CHARS;
838 }
839 else
840#endif
841 /* MAX_LONG_CHARS is enough to hold a 64-bit integer,
842 including sign. Decimal takes the most space. This
843 isn't enough for octal. If a width is specified we
844 need more (which we allocate later). */
845 if (width < MAX_LONG_CHARS)
846 width = MAX_LONG_CHARS;
Benjamin Peterson14339b62009-01-31 16:36:08 +0000847 n += width;
Mark Dickinson6ce4a9a2009-11-16 17:00:11 +0000848 /* XXX should allow for large precision here too. */
Benjamin Peterson14339b62009-01-31 16:36:08 +0000849 if (abuffersize < width)
850 abuffersize = width;
851 break;
852 case 's':
853 {
854 /* UTF-8 */
Georg Brandl780b2a62009-05-05 09:19:59 +0000855 const char *s = va_arg(count, const char*);
Walter Dörwaldc1651a02009-05-03 22:55:55 +0000856 PyObject *str = PyUnicode_DecodeUTF8(s, strlen(s), "replace");
857 if (!str)
858 goto fail;
859 n += PyUnicode_GET_SIZE(str);
860 /* Remember the str and switch to the next slot */
861 *callresult++ = str;
Benjamin Peterson14339b62009-01-31 16:36:08 +0000862 break;
863 }
864 case 'U':
865 {
866 PyObject *obj = va_arg(count, PyObject *);
867 assert(obj && PyUnicode_Check(obj));
868 n += PyUnicode_GET_SIZE(obj);
869 break;
870 }
871 case 'V':
872 {
873 PyObject *obj = va_arg(count, PyObject *);
874 const char *str = va_arg(count, const char *);
Victor Stinner2b574a22011-03-01 22:48:49 +0000875 PyObject *str_obj;
Benjamin Peterson14339b62009-01-31 16:36:08 +0000876 assert(obj || str);
877 assert(!obj || PyUnicode_Check(obj));
Victor Stinner2b574a22011-03-01 22:48:49 +0000878 if (obj) {
Benjamin Peterson14339b62009-01-31 16:36:08 +0000879 n += PyUnicode_GET_SIZE(obj);
Victor Stinner2b574a22011-03-01 22:48:49 +0000880 *callresult++ = NULL;
881 }
882 else {
883 str_obj = PyUnicode_DecodeUTF8(str, strlen(str), "replace");
884 if (!str_obj)
885 goto fail;
886 n += PyUnicode_GET_SIZE(str_obj);
887 *callresult++ = str_obj;
888 }
Benjamin Peterson14339b62009-01-31 16:36:08 +0000889 break;
890 }
891 case 'S':
892 {
893 PyObject *obj = va_arg(count, PyObject *);
894 PyObject *str;
895 assert(obj);
896 str = PyObject_Str(obj);
897 if (!str)
898 goto fail;
899 n += PyUnicode_GET_SIZE(str);
900 /* Remember the str and switch to the next slot */
901 *callresult++ = str;
902 break;
903 }
904 case 'R':
905 {
906 PyObject *obj = va_arg(count, PyObject *);
907 PyObject *repr;
908 assert(obj);
909 repr = PyObject_Repr(obj);
910 if (!repr)
911 goto fail;
912 n += PyUnicode_GET_SIZE(repr);
913 /* Remember the repr and switch to the next slot */
914 *callresult++ = repr;
915 break;
916 }
917 case 'A':
918 {
919 PyObject *obj = va_arg(count, PyObject *);
920 PyObject *ascii;
921 assert(obj);
922 ascii = PyObject_ASCII(obj);
923 if (!ascii)
924 goto fail;
925 n += PyUnicode_GET_SIZE(ascii);
926 /* Remember the repr and switch to the next slot */
927 *callresult++ = ascii;
928 break;
929 }
930 case 'p':
931 (void) va_arg(count, int);
932 /* maximum 64-bit pointer representation:
933 * 0xffffffffffffffff
934 * so 19 characters is enough.
935 * XXX I count 18 -- what's the extra for?
936 */
937 n += 19;
938 break;
939 default:
940 /* if we stumble upon an unknown
941 formatting code, copy the rest of
942 the format string to the output
943 string. (we cannot just skip the
944 code, since there's no way to know
945 what's in the argument list) */
946 n += strlen(p);
947 goto expand;
948 }
949 } else
950 n++;
951 }
Benjamin Peterson29060642009-01-31 22:14:21 +0000952 expand:
Mark Dickinson6ce4a9a2009-11-16 17:00:11 +0000953 if (abuffersize > ITEM_BUFFER_LEN) {
954 /* add 1 for sprintf's trailing null byte */
955 abuffer = PyObject_Malloc(abuffersize + 1);
Benjamin Peterson14339b62009-01-31 16:36:08 +0000956 if (!abuffer) {
957 PyErr_NoMemory();
958 goto fail;
959 }
960 realbuffer = abuffer;
961 }
962 else
963 realbuffer = buffer;
964 /* step 4: fill the buffer */
965 /* Since we've analyzed how much space we need for the worst case,
966 we don't have to resize the string.
967 There can be no errors beyond this point. */
968 string = PyUnicode_FromUnicode(NULL, n);
969 if (!string)
970 goto fail;
Walter Dörwaldd2034312007-05-18 16:29:38 +0000971
Benjamin Peterson14339b62009-01-31 16:36:08 +0000972 s = PyUnicode_AS_UNICODE(string);
973 callresult = callresults;
Walter Dörwaldd2034312007-05-18 16:29:38 +0000974
Benjamin Peterson14339b62009-01-31 16:36:08 +0000975 for (f = format; *f; f++) {
976 if (*f == '%') {
977 const char* p = f++;
978 int longflag = 0;
Mark Dickinson6ce4a9a2009-11-16 17:00:11 +0000979 int longlongflag = 0;
Benjamin Peterson14339b62009-01-31 16:36:08 +0000980 int size_tflag = 0;
981 zeropad = (*f == '0');
982 /* parse the width.precision part */
983 width = 0;
David Malcolm96960882010-11-05 17:23:41 +0000984 while (Py_ISDIGIT((unsigned)*f))
Benjamin Peterson14339b62009-01-31 16:36:08 +0000985 width = (width*10) + *f++ - '0';
986 precision = 0;
987 if (*f == '.') {
988 f++;
David Malcolm96960882010-11-05 17:23:41 +0000989 while (Py_ISDIGIT((unsigned)*f))
Benjamin Peterson14339b62009-01-31 16:36:08 +0000990 precision = (precision*10) + *f++ - '0';
991 }
Mark Dickinson6ce4a9a2009-11-16 17:00:11 +0000992 /* Handle %ld, %lu, %lld and %llu. */
993 if (*f == 'l') {
994 if (f[1] == 'd' || f[1] == 'u') {
995 longflag = 1;
996 ++f;
997 }
998#ifdef HAVE_LONG_LONG
999 else if (f[1] == 'l' &&
1000 (f[2] == 'd' || f[2] == 'u')) {
1001 longlongflag = 1;
1002 f += 2;
1003 }
1004#endif
Benjamin Peterson14339b62009-01-31 16:36:08 +00001005 }
1006 /* handle the size_t flag. */
1007 if (*f == 'z' && (f[1] == 'd' || f[1] == 'u')) {
1008 size_tflag = 1;
1009 ++f;
1010 }
Walter Dörwaldd2034312007-05-18 16:29:38 +00001011
Benjamin Peterson14339b62009-01-31 16:36:08 +00001012 switch (*f) {
1013 case 'c':
Victor Stinner659eb842011-02-23 12:14:22 +00001014 {
1015 int ordinal = va_arg(vargs, int);
1016#ifndef Py_UNICODE_WIDE
1017 if (ordinal > 0xffff) {
1018 ordinal -= 0x10000;
1019 *s++ = 0xD800 | (ordinal >> 10);
1020 *s++ = 0xDC00 | (ordinal & 0x3FF);
1021 } else
1022#endif
1023 *s++ = ordinal;
Benjamin Peterson14339b62009-01-31 16:36:08 +00001024 break;
Victor Stinner659eb842011-02-23 12:14:22 +00001025 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00001026 case 'd':
Mark Dickinson6ce4a9a2009-11-16 17:00:11 +00001027 makefmt(fmt, longflag, longlongflag, size_tflag, zeropad,
1028 width, precision, 'd');
Benjamin Peterson14339b62009-01-31 16:36:08 +00001029 if (longflag)
1030 sprintf(realbuffer, fmt, va_arg(vargs, long));
Mark Dickinson6ce4a9a2009-11-16 17:00:11 +00001031#ifdef HAVE_LONG_LONG
1032 else if (longlongflag)
1033 sprintf(realbuffer, fmt, va_arg(vargs, PY_LONG_LONG));
1034#endif
Benjamin Peterson14339b62009-01-31 16:36:08 +00001035 else if (size_tflag)
1036 sprintf(realbuffer, fmt, va_arg(vargs, Py_ssize_t));
1037 else
1038 sprintf(realbuffer, fmt, va_arg(vargs, int));
1039 appendstring(realbuffer);
1040 break;
1041 case 'u':
Mark Dickinson6ce4a9a2009-11-16 17:00:11 +00001042 makefmt(fmt, longflag, longlongflag, size_tflag, zeropad,
1043 width, precision, 'u');
Benjamin Peterson14339b62009-01-31 16:36:08 +00001044 if (longflag)
1045 sprintf(realbuffer, fmt, va_arg(vargs, unsigned long));
Mark Dickinson6ce4a9a2009-11-16 17:00:11 +00001046#ifdef HAVE_LONG_LONG
1047 else if (longlongflag)
1048 sprintf(realbuffer, fmt, va_arg(vargs,
1049 unsigned PY_LONG_LONG));
1050#endif
Benjamin Peterson14339b62009-01-31 16:36:08 +00001051 else if (size_tflag)
1052 sprintf(realbuffer, fmt, va_arg(vargs, size_t));
1053 else
1054 sprintf(realbuffer, fmt, va_arg(vargs, unsigned int));
1055 appendstring(realbuffer);
1056 break;
1057 case 'i':
Mark Dickinson6ce4a9a2009-11-16 17:00:11 +00001058 makefmt(fmt, 0, 0, 0, zeropad, width, precision, 'i');
Benjamin Peterson14339b62009-01-31 16:36:08 +00001059 sprintf(realbuffer, fmt, va_arg(vargs, int));
1060 appendstring(realbuffer);
1061 break;
1062 case 'x':
Mark Dickinson6ce4a9a2009-11-16 17:00:11 +00001063 makefmt(fmt, 0, 0, 0, zeropad, width, precision, 'x');
Benjamin Peterson14339b62009-01-31 16:36:08 +00001064 sprintf(realbuffer, fmt, va_arg(vargs, int));
1065 appendstring(realbuffer);
1066 break;
1067 case 's':
1068 {
Walter Dörwaldc1651a02009-05-03 22:55:55 +00001069 /* unused, since we already have the result */
1070 (void) va_arg(vargs, char *);
1071 Py_UNICODE_COPY(s, PyUnicode_AS_UNICODE(*callresult),
1072 PyUnicode_GET_SIZE(*callresult));
1073 s += PyUnicode_GET_SIZE(*callresult);
1074 /* We're done with the unicode()/repr() => forget it */
1075 Py_DECREF(*callresult);
1076 /* switch to next unicode()/repr() result */
1077 ++callresult;
Benjamin Peterson14339b62009-01-31 16:36:08 +00001078 break;
1079 }
1080 case 'U':
1081 {
1082 PyObject *obj = va_arg(vargs, PyObject *);
1083 Py_ssize_t size = PyUnicode_GET_SIZE(obj);
1084 Py_UNICODE_COPY(s, PyUnicode_AS_UNICODE(obj), size);
1085 s += size;
1086 break;
1087 }
1088 case 'V':
1089 {
1090 PyObject *obj = va_arg(vargs, PyObject *);
Victor Stinner2b574a22011-03-01 22:48:49 +00001091 va_arg(vargs, const char *);
Benjamin Peterson14339b62009-01-31 16:36:08 +00001092 if (obj) {
1093 Py_ssize_t size = PyUnicode_GET_SIZE(obj);
1094 Py_UNICODE_COPY(s, PyUnicode_AS_UNICODE(obj), size);
1095 s += size;
1096 } else {
Victor Stinner2b574a22011-03-01 22:48:49 +00001097 Py_UNICODE_COPY(s, PyUnicode_AS_UNICODE(*callresult),
1098 PyUnicode_GET_SIZE(*callresult));
1099 s += PyUnicode_GET_SIZE(*callresult);
1100 Py_DECREF(*callresult);
Benjamin Peterson14339b62009-01-31 16:36:08 +00001101 }
Victor Stinner2b574a22011-03-01 22:48:49 +00001102 ++callresult;
Benjamin Peterson14339b62009-01-31 16:36:08 +00001103 break;
1104 }
1105 case 'S':
1106 case 'R':
Victor Stinner9a909002010-10-18 20:59:24 +00001107 case 'A':
Benjamin Peterson14339b62009-01-31 16:36:08 +00001108 {
1109 Py_UNICODE *ucopy;
1110 Py_ssize_t usize;
1111 Py_ssize_t upos;
1112 /* unused, since we already have the result */
1113 (void) va_arg(vargs, PyObject *);
1114 ucopy = PyUnicode_AS_UNICODE(*callresult);
1115 usize = PyUnicode_GET_SIZE(*callresult);
1116 for (upos = 0; upos<usize;)
1117 *s++ = ucopy[upos++];
1118 /* We're done with the unicode()/repr() => forget it */
1119 Py_DECREF(*callresult);
1120 /* switch to next unicode()/repr() result */
1121 ++callresult;
1122 break;
1123 }
1124 case 'p':
1125 sprintf(buffer, "%p", va_arg(vargs, void*));
1126 /* %p is ill-defined: ensure leading 0x. */
1127 if (buffer[1] == 'X')
1128 buffer[1] = 'x';
1129 else if (buffer[1] != 'x') {
1130 memmove(buffer+2, buffer, strlen(buffer)+1);
1131 buffer[0] = '0';
1132 buffer[1] = 'x';
1133 }
1134 appendstring(buffer);
1135 break;
1136 case '%':
1137 *s++ = '%';
1138 break;
1139 default:
1140 appendstring(p);
1141 goto end;
1142 }
Victor Stinner1205f272010-09-11 00:54:47 +00001143 }
Victor Stinner1205f272010-09-11 00:54:47 +00001144 else
Benjamin Peterson14339b62009-01-31 16:36:08 +00001145 *s++ = *f;
1146 }
Walter Dörwaldd2034312007-05-18 16:29:38 +00001147
Benjamin Peterson29060642009-01-31 22:14:21 +00001148 end:
Benjamin Peterson14339b62009-01-31 16:36:08 +00001149 if (callresults)
1150 PyObject_Free(callresults);
1151 if (abuffer)
1152 PyObject_Free(abuffer);
1153 PyUnicode_Resize(&string, s - PyUnicode_AS_UNICODE(string));
1154 return string;
Benjamin Peterson29060642009-01-31 22:14:21 +00001155 fail:
Benjamin Peterson14339b62009-01-31 16:36:08 +00001156 if (callresults) {
1157 PyObject **callresult2 = callresults;
1158 while (callresult2 < callresult) {
Victor Stinner2b574a22011-03-01 22:48:49 +00001159 Py_XDECREF(*callresult2);
Benjamin Peterson14339b62009-01-31 16:36:08 +00001160 ++callresult2;
1161 }
1162 PyObject_Free(callresults);
1163 }
1164 if (abuffer)
1165 PyObject_Free(abuffer);
1166 return NULL;
Walter Dörwaldd2034312007-05-18 16:29:38 +00001167}
1168
1169#undef appendstring
1170
1171PyObject *
1172PyUnicode_FromFormat(const char *format, ...)
1173{
Benjamin Peterson14339b62009-01-31 16:36:08 +00001174 PyObject* ret;
1175 va_list vargs;
Walter Dörwaldd2034312007-05-18 16:29:38 +00001176
1177#ifdef HAVE_STDARG_PROTOTYPES
Benjamin Peterson14339b62009-01-31 16:36:08 +00001178 va_start(vargs, format);
Walter Dörwaldd2034312007-05-18 16:29:38 +00001179#else
Benjamin Peterson14339b62009-01-31 16:36:08 +00001180 va_start(vargs);
Walter Dörwaldd2034312007-05-18 16:29:38 +00001181#endif
Benjamin Peterson14339b62009-01-31 16:36:08 +00001182 ret = PyUnicode_FromFormatV(format, vargs);
1183 va_end(vargs);
1184 return ret;
Walter Dörwaldd2034312007-05-18 16:29:38 +00001185}
1186
Victor Stinner5593d8a2010-10-02 11:11:27 +00001187/* Helper function for PyUnicode_AsWideChar() and PyUnicode_AsWideCharString():
1188 convert a Unicode object to a wide character string.
1189
Victor Stinnerd88d9832011-09-06 02:00:05 +02001190 - If w is NULL: return the number of wide characters (including the null
Victor Stinner5593d8a2010-10-02 11:11:27 +00001191 character) required to convert the unicode object. Ignore size argument.
1192
Victor Stinnerd88d9832011-09-06 02:00:05 +02001193 - Otherwise: return the number of wide characters (excluding the null
Victor Stinner5593d8a2010-10-02 11:11:27 +00001194 character) written into w. Write at most size wide characters (including
Victor Stinnerd88d9832011-09-06 02:00:05 +02001195 the null character). */
Victor Stinner5593d8a2010-10-02 11:11:27 +00001196static Py_ssize_t
Victor Stinner137c34c2010-09-29 10:25:54 +00001197unicode_aswidechar(PyUnicodeObject *unicode,
1198 wchar_t *w,
1199 Py_ssize_t size)
1200{
1201#if Py_UNICODE_SIZE == SIZEOF_WCHAR_T
Victor Stinner5593d8a2010-10-02 11:11:27 +00001202 Py_ssize_t res;
1203 if (w != NULL) {
1204 res = PyUnicode_GET_SIZE(unicode);
1205 if (size > res)
1206 size = res + 1;
1207 else
1208 res = size;
1209 memcpy(w, unicode->str, size * sizeof(wchar_t));
1210 return res;
1211 }
1212 else
1213 return PyUnicode_GET_SIZE(unicode) + 1;
1214#elif Py_UNICODE_SIZE == 2 && SIZEOF_WCHAR_T == 4
1215 register const Py_UNICODE *u;
1216 const Py_UNICODE *uend;
1217 const wchar_t *worig, *wend;
1218 Py_ssize_t nchar;
1219
Victor Stinner137c34c2010-09-29 10:25:54 +00001220 u = PyUnicode_AS_UNICODE(unicode);
Victor Stinner5593d8a2010-10-02 11:11:27 +00001221 uend = u + PyUnicode_GET_SIZE(unicode);
1222 if (w != NULL) {
1223 worig = w;
1224 wend = w + size;
1225 while (u != uend && w != wend) {
1226 if (0xD800 <= u[0] && u[0] <= 0xDBFF
1227 && 0xDC00 <= u[1] && u[1] <= 0xDFFF)
1228 {
1229 *w = (((u[0] & 0x3FF) << 10) | (u[1] & 0x3FF)) + 0x10000;
1230 u += 2;
1231 }
1232 else {
1233 *w = *u;
1234 u++;
1235 }
1236 w++;
1237 }
1238 if (w != wend)
1239 *w = L'\0';
1240 return w - worig;
1241 }
1242 else {
Victor Stinnerd88d9832011-09-06 02:00:05 +02001243 nchar = 1; /* null character at the end */
Victor Stinner5593d8a2010-10-02 11:11:27 +00001244 while (u != uend) {
1245 if (0xD800 <= u[0] && u[0] <= 0xDBFF
1246 && 0xDC00 <= u[1] && u[1] <= 0xDFFF)
1247 u += 2;
1248 else
1249 u++;
1250 nchar++;
1251 }
1252 }
1253 return nchar;
1254#elif Py_UNICODE_SIZE == 4 && SIZEOF_WCHAR_T == 2
1255 register Py_UNICODE *u, *uend, ordinal;
1256 register Py_ssize_t i;
1257 wchar_t *worig, *wend;
1258 Py_ssize_t nchar;
1259
1260 u = PyUnicode_AS_UNICODE(unicode);
1261 uend = u + PyUnicode_GET_SIZE(u);
1262 if (w != NULL) {
1263 worig = w;
1264 wend = w + size;
1265 while (u != uend && w != wend) {
1266 ordinal = *u;
1267 if (ordinal > 0xffff) {
1268 ordinal -= 0x10000;
1269 *w++ = 0xD800 | (ordinal >> 10);
1270 *w++ = 0xDC00 | (ordinal & 0x3FF);
1271 }
1272 else
1273 *w++ = ordinal;
1274 u++;
1275 }
1276 if (w != wend)
1277 *w = 0;
1278 return w - worig;
1279 }
1280 else {
Victor Stinnerd88d9832011-09-06 02:00:05 +02001281 nchar = 1; /* null character */
Victor Stinner5593d8a2010-10-02 11:11:27 +00001282 while (u != uend) {
1283 if (*u > 0xffff)
1284 nchar += 2;
1285 else
1286 nchar++;
1287 u++;
1288 }
1289 return nchar;
1290 }
1291#else
1292# error "unsupported wchar_t and Py_UNICODE sizes, see issue #8670"
Victor Stinner137c34c2010-09-29 10:25:54 +00001293#endif
1294}
1295
1296Py_ssize_t
Martin v. Löwis4d0d4712010-12-03 20:14:31 +00001297PyUnicode_AsWideChar(PyObject *unicode,
Victor Stinner137c34c2010-09-29 10:25:54 +00001298 wchar_t *w,
1299 Py_ssize_t size)
Guido van Rossumd57fd912000-03-10 22:53:23 +00001300{
1301 if (unicode == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +00001302 PyErr_BadInternalCall();
1303 return -1;
Guido van Rossumd57fd912000-03-10 22:53:23 +00001304 }
Martin v. Löwis4d0d4712010-12-03 20:14:31 +00001305 return unicode_aswidechar((PyUnicodeObject*)unicode, w, size);
Guido van Rossumd57fd912000-03-10 22:53:23 +00001306}
1307
Victor Stinner137c34c2010-09-29 10:25:54 +00001308wchar_t*
Victor Stinnerbeb4135b2010-10-07 01:02:42 +00001309PyUnicode_AsWideCharString(PyObject *unicode,
Victor Stinner137c34c2010-09-29 10:25:54 +00001310 Py_ssize_t *size)
1311{
1312 wchar_t* buffer;
1313 Py_ssize_t buflen;
1314
1315 if (unicode == NULL) {
1316 PyErr_BadInternalCall();
1317 return NULL;
1318 }
1319
Victor Stinnerbeb4135b2010-10-07 01:02:42 +00001320 buflen = unicode_aswidechar((PyUnicodeObject *)unicode, NULL, 0);
Victor Stinner5593d8a2010-10-02 11:11:27 +00001321 if (PY_SSIZE_T_MAX / sizeof(wchar_t) < buflen) {
Victor Stinner137c34c2010-09-29 10:25:54 +00001322 PyErr_NoMemory();
1323 return NULL;
1324 }
1325
Victor Stinner137c34c2010-09-29 10:25:54 +00001326 buffer = PyMem_MALLOC(buflen * sizeof(wchar_t));
1327 if (buffer == NULL) {
1328 PyErr_NoMemory();
1329 return NULL;
1330 }
Victor Stinnerbeb4135b2010-10-07 01:02:42 +00001331 buflen = unicode_aswidechar((PyUnicodeObject *)unicode, buffer, buflen);
Victor Stinner5593d8a2010-10-02 11:11:27 +00001332 if (size != NULL)
1333 *size = buflen;
Victor Stinner137c34c2010-09-29 10:25:54 +00001334 return buffer;
1335}
1336
Guido van Rossumd57fd912000-03-10 22:53:23 +00001337#endif
1338
Marc-André Lemburgcc8764c2002-08-11 12:23:04 +00001339PyObject *PyUnicode_FromOrdinal(int ordinal)
1340{
Guido van Rossum8ac004e2007-07-15 13:00:05 +00001341 Py_UNICODE s[2];
Marc-André Lemburgcc8764c2002-08-11 12:23:04 +00001342
Marc-André Lemburgcc8764c2002-08-11 12:23:04 +00001343 if (ordinal < 0 || ordinal > 0x10ffff) {
Benjamin Peterson29060642009-01-31 22:14:21 +00001344 PyErr_SetString(PyExc_ValueError,
1345 "chr() arg not in range(0x110000)");
1346 return NULL;
Marc-André Lemburgcc8764c2002-08-11 12:23:04 +00001347 }
Guido van Rossum8ac004e2007-07-15 13:00:05 +00001348
1349#ifndef Py_UNICODE_WIDE
1350 if (ordinal > 0xffff) {
1351 ordinal -= 0x10000;
1352 s[0] = 0xD800 | (ordinal >> 10);
1353 s[1] = 0xDC00 | (ordinal & 0x3FF);
1354 return PyUnicode_FromUnicode(s, 2);
Marc-André Lemburgcc8764c2002-08-11 12:23:04 +00001355 }
1356#endif
1357
Hye-Shik Chang40574832004-04-06 07:24:51 +00001358 s[0] = (Py_UNICODE)ordinal;
1359 return PyUnicode_FromUnicode(s, 1);
Marc-André Lemburgcc8764c2002-08-11 12:23:04 +00001360}
1361
Guido van Rossumd57fd912000-03-10 22:53:23 +00001362PyObject *PyUnicode_FromObject(register PyObject *obj)
1363{
Guido van Rossumb8c65bc2001-10-19 02:01:31 +00001364 /* XXX Perhaps we should make this API an alias of
Benjamin Peterson29060642009-01-31 22:14:21 +00001365 PyObject_Str() instead ?! */
Guido van Rossumb8c65bc2001-10-19 02:01:31 +00001366 if (PyUnicode_CheckExact(obj)) {
Benjamin Peterson29060642009-01-31 22:14:21 +00001367 Py_INCREF(obj);
1368 return obj;
Guido van Rossumb8c65bc2001-10-19 02:01:31 +00001369 }
1370 if (PyUnicode_Check(obj)) {
Benjamin Peterson29060642009-01-31 22:14:21 +00001371 /* For a Unicode subtype that's not a Unicode object,
1372 return a true Unicode object with the same data. */
1373 return PyUnicode_FromUnicode(PyUnicode_AS_UNICODE(obj),
1374 PyUnicode_GET_SIZE(obj));
Guido van Rossumb8c65bc2001-10-19 02:01:31 +00001375 }
Guido van Rossum98297ee2007-11-06 21:34:58 +00001376 PyErr_Format(PyExc_TypeError,
1377 "Can't convert '%.100s' object to str implicitly",
Christian Heimes90aa7642007-12-19 02:45:37 +00001378 Py_TYPE(obj)->tp_name);
Guido van Rossum98297ee2007-11-06 21:34:58 +00001379 return NULL;
Marc-André Lemburg5a5c81a2000-07-07 13:46:42 +00001380}
1381
1382PyObject *PyUnicode_FromEncodedObject(register PyObject *obj,
Benjamin Peterson29060642009-01-31 22:14:21 +00001383 const char *encoding,
1384 const char *errors)
Marc-André Lemburg5a5c81a2000-07-07 13:46:42 +00001385{
Antoine Pitroub0fa8312010-09-01 15:10:12 +00001386 Py_buffer buffer;
Marc-André Lemburg5a5c81a2000-07-07 13:46:42 +00001387 PyObject *v;
Tim Petersced69f82003-09-16 20:30:58 +00001388
Guido van Rossumd57fd912000-03-10 22:53:23 +00001389 if (obj == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +00001390 PyErr_BadInternalCall();
1391 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00001392 }
Marc-André Lemburg5a5c81a2000-07-07 13:46:42 +00001393
Antoine Pitroub0fa8312010-09-01 15:10:12 +00001394 /* Decoding bytes objects is the most common case and should be fast */
1395 if (PyBytes_Check(obj)) {
1396 if (PyBytes_GET_SIZE(obj) == 0) {
1397 Py_INCREF(unicode_empty);
1398 v = (PyObject *) unicode_empty;
1399 }
1400 else {
1401 v = PyUnicode_Decode(
1402 PyBytes_AS_STRING(obj), PyBytes_GET_SIZE(obj),
1403 encoding, errors);
1404 }
1405 return v;
1406 }
1407
Guido van Rossumb8c65bc2001-10-19 02:01:31 +00001408 if (PyUnicode_Check(obj)) {
Benjamin Peterson29060642009-01-31 22:14:21 +00001409 PyErr_SetString(PyExc_TypeError,
1410 "decoding str is not supported");
1411 return NULL;
Benjamin Peterson14339b62009-01-31 16:36:08 +00001412 }
Guido van Rossumb8c65bc2001-10-19 02:01:31 +00001413
Antoine Pitroub0fa8312010-09-01 15:10:12 +00001414 /* Retrieve a bytes buffer view through the PEP 3118 buffer interface */
1415 if (PyObject_GetBuffer(obj, &buffer, PyBUF_SIMPLE) < 0) {
1416 PyErr_Format(PyExc_TypeError,
1417 "coercing to str: need bytes, bytearray "
1418 "or buffer-like object, %.80s found",
1419 Py_TYPE(obj)->tp_name);
1420 return NULL;
Marc-André Lemburg6871f6a2001-09-20 12:53:16 +00001421 }
Tim Petersced69f82003-09-16 20:30:58 +00001422
Antoine Pitroub0fa8312010-09-01 15:10:12 +00001423 if (buffer.len == 0) {
Benjamin Peterson29060642009-01-31 22:14:21 +00001424 Py_INCREF(unicode_empty);
Antoine Pitroub0fa8312010-09-01 15:10:12 +00001425 v = (PyObject *) unicode_empty;
Guido van Rossumd57fd912000-03-10 22:53:23 +00001426 }
Tim Petersced69f82003-09-16 20:30:58 +00001427 else
Antoine Pitroub0fa8312010-09-01 15:10:12 +00001428 v = PyUnicode_Decode((char*) buffer.buf, buffer.len, encoding, errors);
Marc-André Lemburgad7c98e2001-01-17 17:09:53 +00001429
Antoine Pitroub0fa8312010-09-01 15:10:12 +00001430 PyBuffer_Release(&buffer);
Marc-André Lemburg5a5c81a2000-07-07 13:46:42 +00001431 return v;
Guido van Rossumd57fd912000-03-10 22:53:23 +00001432}
1433
Victor Stinner600d3be2010-06-10 12:00:55 +00001434/* Convert encoding to lower case and replace '_' with '-' in order to
Victor Stinner37296e82010-06-10 13:36:23 +00001435 catch e.g. UTF_8. Return 0 on error (encoding is longer than lower_len-1),
1436 1 on success. */
1437static int
1438normalize_encoding(const char *encoding,
1439 char *lower,
1440 size_t lower_len)
Guido van Rossumd57fd912000-03-10 22:53:23 +00001441{
Guido van Rossumdaa251c2007-10-25 23:47:33 +00001442 const char *e;
Victor Stinner600d3be2010-06-10 12:00:55 +00001443 char *l;
1444 char *l_end;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00001445
Guido van Rossumdaa251c2007-10-25 23:47:33 +00001446 e = encoding;
1447 l = lower;
Victor Stinner600d3be2010-06-10 12:00:55 +00001448 l_end = &lower[lower_len - 1];
Victor Stinner37296e82010-06-10 13:36:23 +00001449 while (*e) {
1450 if (l == l_end)
1451 return 0;
David Malcolm96960882010-11-05 17:23:41 +00001452 if (Py_ISUPPER(*e)) {
1453 *l++ = Py_TOLOWER(*e++);
Guido van Rossumdaa251c2007-10-25 23:47:33 +00001454 }
1455 else if (*e == '_') {
1456 *l++ = '-';
1457 e++;
1458 }
1459 else {
1460 *l++ = *e++;
1461 }
1462 }
1463 *l = '\0';
Victor Stinner37296e82010-06-10 13:36:23 +00001464 return 1;
Victor Stinner600d3be2010-06-10 12:00:55 +00001465}
1466
1467PyObject *PyUnicode_Decode(const char *s,
1468 Py_ssize_t size,
1469 const char *encoding,
1470 const char *errors)
1471{
1472 PyObject *buffer = NULL, *unicode;
1473 Py_buffer info;
1474 char lower[11]; /* Enough for any encoding shortcut */
1475
1476 if (encoding == NULL)
1477 encoding = PyUnicode_GetDefaultEncoding();
Fred Drakee4315f52000-05-09 19:53:39 +00001478
1479 /* Shortcuts for common default encodings */
Victor Stinner37296e82010-06-10 13:36:23 +00001480 if (normalize_encoding(encoding, lower, sizeof(lower))) {
1481 if (strcmp(lower, "utf-8") == 0)
1482 return PyUnicode_DecodeUTF8(s, size, errors);
1483 else if ((strcmp(lower, "latin-1") == 0) ||
1484 (strcmp(lower, "iso-8859-1") == 0))
1485 return PyUnicode_DecodeLatin1(s, size, errors);
Mark Hammond0ccda1e2003-07-01 00:13:27 +00001486#if defined(MS_WINDOWS) && defined(HAVE_USABLE_WCHAR_T)
Victor Stinner37296e82010-06-10 13:36:23 +00001487 else if (strcmp(lower, "mbcs") == 0)
1488 return PyUnicode_DecodeMBCS(s, size, errors);
Mark Hammond0ccda1e2003-07-01 00:13:27 +00001489#endif
Victor Stinner37296e82010-06-10 13:36:23 +00001490 else if (strcmp(lower, "ascii") == 0)
1491 return PyUnicode_DecodeASCII(s, size, errors);
1492 else if (strcmp(lower, "utf-16") == 0)
1493 return PyUnicode_DecodeUTF16(s, size, errors, 0);
1494 else if (strcmp(lower, "utf-32") == 0)
1495 return PyUnicode_DecodeUTF32(s, size, errors, 0);
1496 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00001497
1498 /* Decode via the codec registry */
Guido van Rossumbe801ac2007-10-08 03:32:34 +00001499 buffer = NULL;
Antoine Pitrouc3b39242009-01-03 16:59:18 +00001500 if (PyBuffer_FillInfo(&info, NULL, (void *)s, size, 1, PyBUF_FULL_RO) < 0)
Guido van Rossumbe801ac2007-10-08 03:32:34 +00001501 goto onError;
Antoine Pitrouee58fa42008-08-19 18:22:14 +00001502 buffer = PyMemoryView_FromBuffer(&info);
Guido van Rossumd57fd912000-03-10 22:53:23 +00001503 if (buffer == NULL)
1504 goto onError;
1505 unicode = PyCodec_Decode(buffer, encoding, errors);
1506 if (unicode == NULL)
1507 goto onError;
1508 if (!PyUnicode_Check(unicode)) {
1509 PyErr_Format(PyExc_TypeError,
Benjamin Peterson142957c2008-07-04 19:55:29 +00001510 "decoder did not return a str object (type=%.400s)",
Christian Heimes90aa7642007-12-19 02:45:37 +00001511 Py_TYPE(unicode)->tp_name);
Guido van Rossumd57fd912000-03-10 22:53:23 +00001512 Py_DECREF(unicode);
1513 goto onError;
1514 }
1515 Py_DECREF(buffer);
1516 return unicode;
Tim Petersced69f82003-09-16 20:30:58 +00001517
Benjamin Peterson29060642009-01-31 22:14:21 +00001518 onError:
Guido van Rossumd57fd912000-03-10 22:53:23 +00001519 Py_XDECREF(buffer);
1520 return NULL;
1521}
1522
Marc-André Lemburgd2d45982004-07-08 17:57:32 +00001523PyObject *PyUnicode_AsDecodedObject(PyObject *unicode,
1524 const char *encoding,
1525 const char *errors)
1526{
1527 PyObject *v;
1528
1529 if (!PyUnicode_Check(unicode)) {
1530 PyErr_BadArgument();
1531 goto onError;
1532 }
1533
1534 if (encoding == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00001535 encoding = PyUnicode_GetDefaultEncoding();
Marc-André Lemburgd2d45982004-07-08 17:57:32 +00001536
1537 /* Decode via the codec registry */
1538 v = PyCodec_Decode(unicode, encoding, errors);
1539 if (v == NULL)
1540 goto onError;
1541 return v;
1542
Benjamin Peterson29060642009-01-31 22:14:21 +00001543 onError:
Marc-André Lemburgd2d45982004-07-08 17:57:32 +00001544 return NULL;
1545}
1546
Marc-André Lemburgb2750b52008-06-06 12:18:17 +00001547PyObject *PyUnicode_AsDecodedUnicode(PyObject *unicode,
1548 const char *encoding,
1549 const char *errors)
1550{
1551 PyObject *v;
1552
1553 if (!PyUnicode_Check(unicode)) {
1554 PyErr_BadArgument();
1555 goto onError;
1556 }
1557
1558 if (encoding == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00001559 encoding = PyUnicode_GetDefaultEncoding();
Marc-André Lemburgb2750b52008-06-06 12:18:17 +00001560
1561 /* Decode via the codec registry */
1562 v = PyCodec_Decode(unicode, encoding, errors);
1563 if (v == NULL)
1564 goto onError;
1565 if (!PyUnicode_Check(v)) {
1566 PyErr_Format(PyExc_TypeError,
Benjamin Peterson142957c2008-07-04 19:55:29 +00001567 "decoder did not return a str object (type=%.400s)",
Marc-André Lemburgb2750b52008-06-06 12:18:17 +00001568 Py_TYPE(v)->tp_name);
1569 Py_DECREF(v);
1570 goto onError;
1571 }
1572 return v;
1573
Benjamin Peterson29060642009-01-31 22:14:21 +00001574 onError:
Marc-André Lemburgb2750b52008-06-06 12:18:17 +00001575 return NULL;
1576}
1577
Guido van Rossumd57fd912000-03-10 22:53:23 +00001578PyObject *PyUnicode_Encode(const Py_UNICODE *s,
Benjamin Peterson29060642009-01-31 22:14:21 +00001579 Py_ssize_t size,
1580 const char *encoding,
1581 const char *errors)
Guido van Rossumd57fd912000-03-10 22:53:23 +00001582{
1583 PyObject *v, *unicode;
Tim Petersced69f82003-09-16 20:30:58 +00001584
Guido van Rossumd57fd912000-03-10 22:53:23 +00001585 unicode = PyUnicode_FromUnicode(s, size);
1586 if (unicode == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00001587 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00001588 v = PyUnicode_AsEncodedString(unicode, encoding, errors);
1589 Py_DECREF(unicode);
1590 return v;
1591}
1592
Marc-André Lemburgd2d45982004-07-08 17:57:32 +00001593PyObject *PyUnicode_AsEncodedObject(PyObject *unicode,
1594 const char *encoding,
1595 const char *errors)
1596{
1597 PyObject *v;
1598
1599 if (!PyUnicode_Check(unicode)) {
1600 PyErr_BadArgument();
1601 goto onError;
1602 }
1603
1604 if (encoding == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00001605 encoding = PyUnicode_GetDefaultEncoding();
Marc-André Lemburgd2d45982004-07-08 17:57:32 +00001606
1607 /* Encode via the codec registry */
1608 v = PyCodec_Encode(unicode, encoding, errors);
1609 if (v == NULL)
1610 goto onError;
1611 return v;
1612
Benjamin Peterson29060642009-01-31 22:14:21 +00001613 onError:
Marc-André Lemburgd2d45982004-07-08 17:57:32 +00001614 return NULL;
1615}
1616
Victor Stinnerad158722010-10-27 00:25:46 +00001617PyObject *
1618PyUnicode_EncodeFSDefault(PyObject *unicode)
Victor Stinnerae6265f2010-05-15 16:27:27 +00001619{
Victor Stinner313a1202010-06-11 23:56:51 +00001620#if defined(MS_WINDOWS) && defined(HAVE_USABLE_WCHAR_T)
Victor Stinnerad158722010-10-27 00:25:46 +00001621 return PyUnicode_EncodeMBCS(PyUnicode_AS_UNICODE(unicode),
1622 PyUnicode_GET_SIZE(unicode),
1623 NULL);
1624#elif defined(__APPLE__)
1625 return PyUnicode_EncodeUTF8(PyUnicode_AS_UNICODE(unicode),
1626 PyUnicode_GET_SIZE(unicode),
1627 "surrogateescape");
1628#else
Victor Stinner3cbf14b2011-04-27 00:24:21 +02001629 PyInterpreterState *interp = PyThreadState_GET()->interp;
1630 /* Bootstrap check: if the filesystem codec is implemented in Python, we
1631 cannot use it to encode and decode filenames before it is loaded. Load
1632 the Python codec requires to encode at least its own filename. Use the C
1633 version of the locale codec until the codec registry is initialized and
1634 the Python codec is loaded.
1635
1636 Py_FileSystemDefaultEncoding is shared between all interpreters, we
1637 cannot only rely on it: check also interp->fscodec_initialized for
1638 subinterpreters. */
1639 if (Py_FileSystemDefaultEncoding && interp->fscodec_initialized) {
Victor Stinnerae6265f2010-05-15 16:27:27 +00001640 return PyUnicode_AsEncodedString(unicode,
1641 Py_FileSystemDefaultEncoding,
1642 "surrogateescape");
Victor Stinnerc39211f2010-09-29 16:35:47 +00001643 }
1644 else {
Victor Stinnerf3170cc2010-10-15 12:04:23 +00001645 /* locale encoding with surrogateescape */
1646 wchar_t *wchar;
1647 char *bytes;
1648 PyObject *bytes_obj;
Victor Stinner2f02a512010-11-08 22:43:46 +00001649 size_t error_pos;
Victor Stinnerf3170cc2010-10-15 12:04:23 +00001650
1651 wchar = PyUnicode_AsWideCharString(unicode, NULL);
1652 if (wchar == NULL)
1653 return NULL;
Victor Stinner2f02a512010-11-08 22:43:46 +00001654 bytes = _Py_wchar2char(wchar, &error_pos);
1655 if (bytes == NULL) {
1656 if (error_pos != (size_t)-1) {
1657 char *errmsg = strerror(errno);
1658 PyObject *exc = NULL;
1659 if (errmsg == NULL)
1660 errmsg = "Py_wchar2char() failed";
1661 raise_encode_exception(&exc,
1662 "filesystemencoding",
1663 PyUnicode_AS_UNICODE(unicode), PyUnicode_GET_SIZE(unicode),
1664 error_pos, error_pos+1,
1665 errmsg);
1666 Py_XDECREF(exc);
1667 }
1668 else
1669 PyErr_NoMemory();
1670 PyMem_Free(wchar);
Victor Stinnerf3170cc2010-10-15 12:04:23 +00001671 return NULL;
Victor Stinner2f02a512010-11-08 22:43:46 +00001672 }
1673 PyMem_Free(wchar);
Victor Stinnerf3170cc2010-10-15 12:04:23 +00001674
1675 bytes_obj = PyBytes_FromString(bytes);
1676 PyMem_Free(bytes);
1677 return bytes_obj;
Victor Stinnerc39211f2010-09-29 16:35:47 +00001678 }
Victor Stinnerad158722010-10-27 00:25:46 +00001679#endif
Victor Stinnerae6265f2010-05-15 16:27:27 +00001680}
1681
Guido van Rossumd57fd912000-03-10 22:53:23 +00001682PyObject *PyUnicode_AsEncodedString(PyObject *unicode,
1683 const char *encoding,
1684 const char *errors)
1685{
1686 PyObject *v;
Victor Stinner600d3be2010-06-10 12:00:55 +00001687 char lower[11]; /* Enough for any encoding shortcut */
Tim Petersced69f82003-09-16 20:30:58 +00001688
Guido van Rossumd57fd912000-03-10 22:53:23 +00001689 if (!PyUnicode_Check(unicode)) {
1690 PyErr_BadArgument();
Amaury Forgeot d'Arcf0481112008-09-05 20:48:47 +00001691 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00001692 }
Fred Drakee4315f52000-05-09 19:53:39 +00001693
Tim Petersced69f82003-09-16 20:30:58 +00001694 if (encoding == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00001695 encoding = PyUnicode_GetDefaultEncoding();
Fred Drakee4315f52000-05-09 19:53:39 +00001696
1697 /* Shortcuts for common default encodings */
Victor Stinner37296e82010-06-10 13:36:23 +00001698 if (normalize_encoding(encoding, lower, sizeof(lower))) {
1699 if (strcmp(lower, "utf-8") == 0)
1700 return PyUnicode_EncodeUTF8(PyUnicode_AS_UNICODE(unicode),
1701 PyUnicode_GET_SIZE(unicode),
1702 errors);
1703 else if ((strcmp(lower, "latin-1") == 0) ||
1704 (strcmp(lower, "iso-8859-1") == 0))
1705 return PyUnicode_EncodeLatin1(PyUnicode_AS_UNICODE(unicode),
1706 PyUnicode_GET_SIZE(unicode),
1707 errors);
Mark Hammond0ccda1e2003-07-01 00:13:27 +00001708#if defined(MS_WINDOWS) && defined(HAVE_USABLE_WCHAR_T)
Victor Stinner37296e82010-06-10 13:36:23 +00001709 else if (strcmp(lower, "mbcs") == 0)
1710 return PyUnicode_EncodeMBCS(PyUnicode_AS_UNICODE(unicode),
1711 PyUnicode_GET_SIZE(unicode),
1712 errors);
Mark Hammond0ccda1e2003-07-01 00:13:27 +00001713#endif
Victor Stinner37296e82010-06-10 13:36:23 +00001714 else if (strcmp(lower, "ascii") == 0)
1715 return PyUnicode_EncodeASCII(PyUnicode_AS_UNICODE(unicode),
1716 PyUnicode_GET_SIZE(unicode),
1717 errors);
1718 }
Victor Stinner59e62db2010-05-15 13:14:32 +00001719 /* During bootstrap, we may need to find the encodings
1720 package, to load the file system encoding, and require the
1721 file system encoding in order to load the encodings
1722 package.
Christian Heimes6a27efa2008-10-30 21:48:26 +00001723
Victor Stinner59e62db2010-05-15 13:14:32 +00001724 Break out of this dependency by assuming that the path to
1725 the encodings module is ASCII-only. XXX could try wcstombs
1726 instead, if the file system encoding is the locale's
1727 encoding. */
Victor Stinner37296e82010-06-10 13:36:23 +00001728 if (Py_FileSystemDefaultEncoding &&
Victor Stinner59e62db2010-05-15 13:14:32 +00001729 strcmp(encoding, Py_FileSystemDefaultEncoding) == 0 &&
1730 !PyThreadState_GET()->interp->codecs_initialized)
1731 return PyUnicode_EncodeASCII(PyUnicode_AS_UNICODE(unicode),
1732 PyUnicode_GET_SIZE(unicode),
1733 errors);
Guido van Rossumd57fd912000-03-10 22:53:23 +00001734
1735 /* Encode via the codec registry */
1736 v = PyCodec_Encode(unicode, encoding, errors);
1737 if (v == NULL)
Amaury Forgeot d'Arcf0481112008-09-05 20:48:47 +00001738 return NULL;
1739
1740 /* The normal path */
1741 if (PyBytes_Check(v))
1742 return v;
1743
1744 /* If the codec returns a buffer, raise a warning and convert to bytes */
Marc-André Lemburgb2750b52008-06-06 12:18:17 +00001745 if (PyByteArray_Check(v)) {
Victor Stinner4a2b7a12010-08-13 14:03:48 +00001746 int error;
Amaury Forgeot d'Arcf0481112008-09-05 20:48:47 +00001747 PyObject *b;
Victor Stinner4a2b7a12010-08-13 14:03:48 +00001748
1749 error = PyErr_WarnFormat(PyExc_RuntimeWarning, 1,
1750 "encoder %s returned bytearray instead of bytes",
1751 encoding);
1752 if (error) {
Amaury Forgeot d'Arcf0481112008-09-05 20:48:47 +00001753 Py_DECREF(v);
1754 return NULL;
Marc-André Lemburgb2750b52008-06-06 12:18:17 +00001755 }
Marc-André Lemburgb2750b52008-06-06 12:18:17 +00001756
Amaury Forgeot d'Arcf0481112008-09-05 20:48:47 +00001757 b = PyBytes_FromStringAndSize(PyByteArray_AS_STRING(v), Py_SIZE(v));
1758 Py_DECREF(v);
1759 return b;
1760 }
1761
1762 PyErr_Format(PyExc_TypeError,
1763 "encoder did not return a bytes object (type=%.400s)",
1764 Py_TYPE(v)->tp_name);
1765 Py_DECREF(v);
Marc-André Lemburgb2750b52008-06-06 12:18:17 +00001766 return NULL;
1767}
1768
1769PyObject *PyUnicode_AsEncodedUnicode(PyObject *unicode,
1770 const char *encoding,
1771 const char *errors)
1772{
1773 PyObject *v;
1774
1775 if (!PyUnicode_Check(unicode)) {
1776 PyErr_BadArgument();
1777 goto onError;
1778 }
1779
1780 if (encoding == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00001781 encoding = PyUnicode_GetDefaultEncoding();
Marc-André Lemburgb2750b52008-06-06 12:18:17 +00001782
1783 /* Encode via the codec registry */
1784 v = PyCodec_Encode(unicode, encoding, errors);
1785 if (v == NULL)
1786 goto onError;
1787 if (!PyUnicode_Check(v)) {
1788 PyErr_Format(PyExc_TypeError,
Benjamin Peterson142957c2008-07-04 19:55:29 +00001789 "encoder did not return an str object (type=%.400s)",
Marc-André Lemburgb2750b52008-06-06 12:18:17 +00001790 Py_TYPE(v)->tp_name);
1791 Py_DECREF(v);
1792 goto onError;
1793 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00001794 return v;
Tim Petersced69f82003-09-16 20:30:58 +00001795
Benjamin Peterson29060642009-01-31 22:14:21 +00001796 onError:
Guido van Rossumd57fd912000-03-10 22:53:23 +00001797 return NULL;
1798}
1799
Marc-André Lemburgbff879c2000-08-03 18:46:08 +00001800PyObject *_PyUnicode_AsDefaultEncodedString(PyObject *unicode,
Benjamin Peterson29060642009-01-31 22:14:21 +00001801 const char *errors)
Marc-André Lemburgbff879c2000-08-03 18:46:08 +00001802{
1803 PyObject *v = ((PyUnicodeObject *)unicode)->defenc;
Marc-André Lemburgbff879c2000-08-03 18:46:08 +00001804 if (v)
1805 return v;
Guido van Rossumf15a29f2007-05-04 00:41:39 +00001806 if (errors != NULL)
1807 Py_FatalError("non-NULL encoding in _PyUnicode_AsDefaultEncodedString");
Guido van Rossum98297ee2007-11-06 21:34:58 +00001808 v = PyUnicode_EncodeUTF8(PyUnicode_AS_UNICODE(unicode),
Guido van Rossum06610092007-08-16 21:02:22 +00001809 PyUnicode_GET_SIZE(unicode),
1810 NULL);
Guido van Rossum98297ee2007-11-06 21:34:58 +00001811 if (!v)
Guido van Rossumf15a29f2007-05-04 00:41:39 +00001812 return NULL;
Guido van Rossume7a0d392007-07-12 07:53:00 +00001813 ((PyUnicodeObject *)unicode)->defenc = v;
Marc-André Lemburgbff879c2000-08-03 18:46:08 +00001814 return v;
1815}
1816
Guido van Rossum00bc0e02007-10-15 02:52:41 +00001817PyObject*
Christian Heimes5894ba72007-11-04 11:43:14 +00001818PyUnicode_DecodeFSDefault(const char *s) {
Guido van Rossum00bc0e02007-10-15 02:52:41 +00001819 Py_ssize_t size = (Py_ssize_t)strlen(s);
Christian Heimes5894ba72007-11-04 11:43:14 +00001820 return PyUnicode_DecodeFSDefaultAndSize(s, size);
1821}
Guido van Rossum00bc0e02007-10-15 02:52:41 +00001822
Christian Heimes5894ba72007-11-04 11:43:14 +00001823PyObject*
1824PyUnicode_DecodeFSDefaultAndSize(const char *s, Py_ssize_t size)
1825{
Victor Stinnerad158722010-10-27 00:25:46 +00001826#if defined(MS_WINDOWS) && defined(HAVE_USABLE_WCHAR_T)
1827 return PyUnicode_DecodeMBCS(s, size, NULL);
1828#elif defined(__APPLE__)
1829 return PyUnicode_DecodeUTF8(s, size, "surrogateescape");
1830#else
Victor Stinner3cbf14b2011-04-27 00:24:21 +02001831 PyInterpreterState *interp = PyThreadState_GET()->interp;
1832 /* Bootstrap check: if the filesystem codec is implemented in Python, we
1833 cannot use it to encode and decode filenames before it is loaded. Load
1834 the Python codec requires to encode at least its own filename. Use the C
1835 version of the locale codec until the codec registry is initialized and
1836 the Python codec is loaded.
1837
1838 Py_FileSystemDefaultEncoding is shared between all interpreters, we
1839 cannot only rely on it: check also interp->fscodec_initialized for
1840 subinterpreters. */
1841 if (Py_FileSystemDefaultEncoding && interp->fscodec_initialized) {
Guido van Rossum00bc0e02007-10-15 02:52:41 +00001842 return PyUnicode_Decode(s, size,
1843 Py_FileSystemDefaultEncoding,
Victor Stinnerb9a20ad2010-04-30 16:37:52 +00001844 "surrogateescape");
Guido van Rossum00bc0e02007-10-15 02:52:41 +00001845 }
1846 else {
Victor Stinnerf3170cc2010-10-15 12:04:23 +00001847 /* locale encoding with surrogateescape */
1848 wchar_t *wchar;
1849 PyObject *unicode;
Victor Stinner168e1172010-10-16 23:16:16 +00001850 size_t len;
Victor Stinnerf3170cc2010-10-15 12:04:23 +00001851
1852 if (s[size] != '\0' || size != strlen(s)) {
1853 PyErr_SetString(PyExc_TypeError, "embedded NUL character");
1854 return NULL;
1855 }
1856
Victor Stinner168e1172010-10-16 23:16:16 +00001857 wchar = _Py_char2wchar(s, &len);
Victor Stinnerf3170cc2010-10-15 12:04:23 +00001858 if (wchar == NULL)
Victor Stinnerd5af0a52010-11-08 23:34:29 +00001859 return PyErr_NoMemory();
Victor Stinnerf3170cc2010-10-15 12:04:23 +00001860
Victor Stinner168e1172010-10-16 23:16:16 +00001861 unicode = PyUnicode_FromWideChar(wchar, len);
Victor Stinnerf3170cc2010-10-15 12:04:23 +00001862 PyMem_Free(wchar);
1863 return unicode;
Guido van Rossum00bc0e02007-10-15 02:52:41 +00001864 }
Victor Stinnerad158722010-10-27 00:25:46 +00001865#endif
Guido van Rossum00bc0e02007-10-15 02:52:41 +00001866}
1867
Martin v. Löwis011e8422009-05-05 04:43:17 +00001868
1869int
Antoine Pitrou13348842012-01-29 18:36:34 +01001870_PyUnicode_HasNULChars(PyObject* s)
1871{
1872 static PyObject *nul = NULL;
1873
1874 if (nul == NULL)
1875 nul = PyUnicode_FromStringAndSize("\0", 1);
1876 if (nul == NULL)
1877 return -1;
1878 return PyUnicode_Contains(s, nul);
1879}
1880
1881
1882int
Martin v. Löwis011e8422009-05-05 04:43:17 +00001883PyUnicode_FSConverter(PyObject* arg, void* addr)
1884{
1885 PyObject *output = NULL;
1886 Py_ssize_t size;
1887 void *data;
Martin v. Löwisc15bdef2009-05-29 14:47:46 +00001888 if (arg == NULL) {
1889 Py_DECREF(*(PyObject**)addr);
1890 return 1;
1891 }
Victor Stinnerdcb24032010-04-22 12:08:36 +00001892 if (PyBytes_Check(arg)) {
Martin v. Löwis011e8422009-05-05 04:43:17 +00001893 output = arg;
1894 Py_INCREF(output);
1895 }
1896 else {
1897 arg = PyUnicode_FromObject(arg);
1898 if (!arg)
1899 return 0;
Victor Stinnerae6265f2010-05-15 16:27:27 +00001900 output = PyUnicode_EncodeFSDefault(arg);
Martin v. Löwis011e8422009-05-05 04:43:17 +00001901 Py_DECREF(arg);
1902 if (!output)
1903 return 0;
1904 if (!PyBytes_Check(output)) {
1905 Py_DECREF(output);
1906 PyErr_SetString(PyExc_TypeError, "encoder failed to return bytes");
1907 return 0;
1908 }
1909 }
Victor Stinner0ea2a462010-04-30 00:22:08 +00001910 size = PyBytes_GET_SIZE(output);
1911 data = PyBytes_AS_STRING(output);
Martin v. Löwis011e8422009-05-05 04:43:17 +00001912 if (size != strlen(data)) {
Benjamin Peterson7a6b44a2011-08-18 13:51:47 -05001913 PyErr_SetString(PyExc_TypeError, "embedded NUL character");
Martin v. Löwis011e8422009-05-05 04:43:17 +00001914 Py_DECREF(output);
1915 return 0;
1916 }
1917 *(PyObject**)addr = output;
Martin v. Löwisc15bdef2009-05-29 14:47:46 +00001918 return Py_CLEANUP_SUPPORTED;
Martin v. Löwis011e8422009-05-05 04:43:17 +00001919}
1920
1921
Victor Stinner47fcb5b2010-08-13 23:59:58 +00001922int
1923PyUnicode_FSDecoder(PyObject* arg, void* addr)
1924{
1925 PyObject *output = NULL;
1926 Py_ssize_t size;
1927 void *data;
1928 if (arg == NULL) {
1929 Py_DECREF(*(PyObject**)addr);
1930 return 1;
1931 }
1932 if (PyUnicode_Check(arg)) {
1933 output = arg;
1934 Py_INCREF(output);
1935 }
1936 else {
1937 arg = PyBytes_FromObject(arg);
1938 if (!arg)
1939 return 0;
1940 output = PyUnicode_DecodeFSDefaultAndSize(PyBytes_AS_STRING(arg),
1941 PyBytes_GET_SIZE(arg));
1942 Py_DECREF(arg);
1943 if (!output)
1944 return 0;
1945 if (!PyUnicode_Check(output)) {
1946 Py_DECREF(output);
1947 PyErr_SetString(PyExc_TypeError, "decoder failed to return unicode");
1948 return 0;
1949 }
1950 }
1951 size = PyUnicode_GET_SIZE(output);
1952 data = PyUnicode_AS_UNICODE(output);
1953 if (size != Py_UNICODE_strlen(data)) {
1954 PyErr_SetString(PyExc_TypeError, "embedded NUL character");
1955 Py_DECREF(output);
1956 return 0;
1957 }
1958 *(PyObject**)addr = output;
1959 return Py_CLEANUP_SUPPORTED;
1960}
1961
1962
Martin v. Löwis5b222132007-06-10 09:51:05 +00001963char*
Marc-André Lemburg4cc0f242008-08-07 18:54:33 +00001964_PyUnicode_AsStringAndSize(PyObject *unicode, Py_ssize_t *psize)
Martin v. Löwis5b222132007-06-10 09:51:05 +00001965{
Christian Heimesf3863112007-11-22 07:46:41 +00001966 PyObject *bytes;
Neal Norwitze0a0a6e2007-08-25 01:04:21 +00001967 if (!PyUnicode_Check(unicode)) {
1968 PyErr_BadArgument();
1969 return NULL;
1970 }
Christian Heimesf3863112007-11-22 07:46:41 +00001971 bytes = _PyUnicode_AsDefaultEncodedString(unicode, NULL);
1972 if (bytes == NULL)
Martin v. Löwis5b222132007-06-10 09:51:05 +00001973 return NULL;
Guido van Rossum7d1df6c2007-08-29 13:53:23 +00001974 if (psize != NULL)
Christian Heimes72b710a2008-05-26 13:28:38 +00001975 *psize = PyBytes_GET_SIZE(bytes);
1976 return PyBytes_AS_STRING(bytes);
Guido van Rossum7d1df6c2007-08-29 13:53:23 +00001977}
1978
1979char*
Marc-André Lemburg4cc0f242008-08-07 18:54:33 +00001980_PyUnicode_AsString(PyObject *unicode)
Guido van Rossum7d1df6c2007-08-29 13:53:23 +00001981{
Marc-André Lemburg4cc0f242008-08-07 18:54:33 +00001982 return _PyUnicode_AsStringAndSize(unicode, NULL);
Martin v. Löwis5b222132007-06-10 09:51:05 +00001983}
1984
Guido van Rossumd57fd912000-03-10 22:53:23 +00001985Py_UNICODE *PyUnicode_AsUnicode(PyObject *unicode)
1986{
1987 if (!PyUnicode_Check(unicode)) {
1988 PyErr_BadArgument();
1989 goto onError;
1990 }
1991 return PyUnicode_AS_UNICODE(unicode);
1992
Benjamin Peterson29060642009-01-31 22:14:21 +00001993 onError:
Guido van Rossumd57fd912000-03-10 22:53:23 +00001994 return NULL;
1995}
1996
Martin v. Löwis18e16552006-02-15 17:27:45 +00001997Py_ssize_t PyUnicode_GetSize(PyObject *unicode)
Guido van Rossumd57fd912000-03-10 22:53:23 +00001998{
1999 if (!PyUnicode_Check(unicode)) {
2000 PyErr_BadArgument();
2001 goto onError;
2002 }
2003 return PyUnicode_GET_SIZE(unicode);
2004
Benjamin Peterson29060642009-01-31 22:14:21 +00002005 onError:
Guido van Rossumd57fd912000-03-10 22:53:23 +00002006 return -1;
2007}
2008
Thomas Wouters78890102000-07-22 19:25:51 +00002009const char *PyUnicode_GetDefaultEncoding(void)
Fred Drakee4315f52000-05-09 19:53:39 +00002010{
Victor Stinner42cb4622010-09-01 19:39:01 +00002011 return "utf-8";
Fred Drakee4315f52000-05-09 19:53:39 +00002012}
2013
Victor Stinner554f3f02010-06-16 23:33:54 +00002014/* create or adjust a UnicodeDecodeError */
2015static void
2016make_decode_exception(PyObject **exceptionObject,
2017 const char *encoding,
2018 const char *input, Py_ssize_t length,
2019 Py_ssize_t startpos, Py_ssize_t endpos,
2020 const char *reason)
2021{
2022 if (*exceptionObject == NULL) {
2023 *exceptionObject = PyUnicodeDecodeError_Create(
2024 encoding, input, length, startpos, endpos, reason);
2025 }
2026 else {
2027 if (PyUnicodeDecodeError_SetStart(*exceptionObject, startpos))
2028 goto onError;
2029 if (PyUnicodeDecodeError_SetEnd(*exceptionObject, endpos))
2030 goto onError;
2031 if (PyUnicodeDecodeError_SetReason(*exceptionObject, reason))
2032 goto onError;
2033 }
2034 return;
2035
2036onError:
2037 Py_DECREF(*exceptionObject);
2038 *exceptionObject = NULL;
2039}
2040
Walter Dörwald3aeb6322002-09-02 13:14:32 +00002041/* error handling callback helper:
2042 build arguments, call the callback and check the arguments,
Fred Drakedb390c12005-10-28 14:39:47 +00002043 if no exception occurred, copy the replacement to the output
Walter Dörwald3aeb6322002-09-02 13:14:32 +00002044 and adjust various state variables.
2045 return 0 on success, -1 on error
2046*/
2047
2048static
2049int unicode_decode_call_errorhandler(const char *errors, PyObject **errorHandler,
Benjamin Peterson29060642009-01-31 22:14:21 +00002050 const char *encoding, const char *reason,
2051 const char **input, const char **inend, Py_ssize_t *startinpos,
2052 Py_ssize_t *endinpos, PyObject **exceptionObject, const char **inptr,
2053 PyUnicodeObject **output, Py_ssize_t *outpos, Py_UNICODE **outptr)
Walter Dörwald3aeb6322002-09-02 13:14:32 +00002054{
Benjamin Peterson142957c2008-07-04 19:55:29 +00002055 static char *argparse = "O!n;decoding error handler must return (str, int) tuple";
Walter Dörwald3aeb6322002-09-02 13:14:32 +00002056
2057 PyObject *restuple = NULL;
2058 PyObject *repunicode = NULL;
Martin v. Löwis18e16552006-02-15 17:27:45 +00002059 Py_ssize_t outsize = PyUnicode_GET_SIZE(*output);
Walter Dörwalde78178e2007-07-30 13:31:40 +00002060 Py_ssize_t insize;
Martin v. Löwis18e16552006-02-15 17:27:45 +00002061 Py_ssize_t requiredsize;
2062 Py_ssize_t newpos;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00002063 Py_UNICODE *repptr;
Walter Dörwalde78178e2007-07-30 13:31:40 +00002064 PyObject *inputobj = NULL;
Martin v. Löwis18e16552006-02-15 17:27:45 +00002065 Py_ssize_t repsize;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00002066 int res = -1;
2067
2068 if (*errorHandler == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +00002069 *errorHandler = PyCodec_LookupError(errors);
2070 if (*errorHandler == NULL)
2071 goto onError;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00002072 }
2073
Victor Stinner554f3f02010-06-16 23:33:54 +00002074 make_decode_exception(exceptionObject,
2075 encoding,
2076 *input, *inend - *input,
2077 *startinpos, *endinpos,
2078 reason);
2079 if (*exceptionObject == NULL)
2080 goto onError;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00002081
2082 restuple = PyObject_CallFunctionObjArgs(*errorHandler, *exceptionObject, NULL);
2083 if (restuple == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00002084 goto onError;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00002085 if (!PyTuple_Check(restuple)) {
Benjamin Petersond75fcb42009-02-19 04:22:03 +00002086 PyErr_SetString(PyExc_TypeError, &argparse[4]);
Benjamin Peterson29060642009-01-31 22:14:21 +00002087 goto onError;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00002088 }
2089 if (!PyArg_ParseTuple(restuple, argparse, &PyUnicode_Type, &repunicode, &newpos))
Benjamin Peterson29060642009-01-31 22:14:21 +00002090 goto onError;
Walter Dörwalde78178e2007-07-30 13:31:40 +00002091
2092 /* Copy back the bytes variables, which might have been modified by the
2093 callback */
2094 inputobj = PyUnicodeDecodeError_GetObject(*exceptionObject);
2095 if (!inputobj)
2096 goto onError;
Christian Heimes72b710a2008-05-26 13:28:38 +00002097 if (!PyBytes_Check(inputobj)) {
Benjamin Peterson29060642009-01-31 22:14:21 +00002098 PyErr_Format(PyExc_TypeError, "exception attribute object must be bytes");
Walter Dörwalde78178e2007-07-30 13:31:40 +00002099 }
Christian Heimes72b710a2008-05-26 13:28:38 +00002100 *input = PyBytes_AS_STRING(inputobj);
2101 insize = PyBytes_GET_SIZE(inputobj);
Walter Dörwalde78178e2007-07-30 13:31:40 +00002102 *inend = *input + insize;
Walter Dörwald36f938f2007-08-10 10:11:43 +00002103 /* we can DECREF safely, as the exception has another reference,
2104 so the object won't go away. */
2105 Py_DECREF(inputobj);
Walter Dörwalde78178e2007-07-30 13:31:40 +00002106
Walter Dörwald3aeb6322002-09-02 13:14:32 +00002107 if (newpos<0)
Benjamin Peterson29060642009-01-31 22:14:21 +00002108 newpos = insize+newpos;
Walter Dörwald2e0b18a2003-01-31 17:19:08 +00002109 if (newpos<0 || newpos>insize) {
Benjamin Peterson29060642009-01-31 22:14:21 +00002110 PyErr_Format(PyExc_IndexError, "position %zd from error handler out of bounds", newpos);
2111 goto onError;
Walter Dörwald2e0b18a2003-01-31 17:19:08 +00002112 }
Walter Dörwald3aeb6322002-09-02 13:14:32 +00002113
2114 /* need more space? (at least enough for what we
2115 have+the replacement+the rest of the string (starting
2116 at the new input position), so we won't have to check space
2117 when there are no errors in the rest of the string) */
2118 repptr = PyUnicode_AS_UNICODE(repunicode);
2119 repsize = PyUnicode_GET_SIZE(repunicode);
2120 requiredsize = *outpos + repsize + insize-newpos;
2121 if (requiredsize > outsize) {
Benjamin Peterson29060642009-01-31 22:14:21 +00002122 if (requiredsize<2*outsize)
2123 requiredsize = 2*outsize;
2124 if (_PyUnicode_Resize(output, requiredsize) < 0)
2125 goto onError;
2126 *outptr = PyUnicode_AS_UNICODE(*output) + *outpos;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00002127 }
2128 *endinpos = newpos;
Walter Dörwalde78178e2007-07-30 13:31:40 +00002129 *inptr = *input + newpos;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00002130 Py_UNICODE_COPY(*outptr, repptr, repsize);
2131 *outptr += repsize;
2132 *outpos += repsize;
Walter Dörwalde78178e2007-07-30 13:31:40 +00002133
Walter Dörwald3aeb6322002-09-02 13:14:32 +00002134 /* we made it! */
2135 res = 0;
2136
Benjamin Peterson29060642009-01-31 22:14:21 +00002137 onError:
Walter Dörwald3aeb6322002-09-02 13:14:32 +00002138 Py_XDECREF(restuple);
2139 return res;
2140}
2141
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00002142/* --- UTF-7 Codec -------------------------------------------------------- */
2143
Antoine Pitrou244651a2009-05-04 18:56:13 +00002144/* See RFC2152 for details. We encode conservatively and decode liberally. */
2145
2146/* Three simple macros defining base-64. */
2147
2148/* Is c a base-64 character? */
2149
2150#define IS_BASE64(c) \
2151 (((c) >= 'A' && (c) <= 'Z') || \
2152 ((c) >= 'a' && (c) <= 'z') || \
2153 ((c) >= '0' && (c) <= '9') || \
2154 (c) == '+' || (c) == '/')
2155
2156/* given that c is a base-64 character, what is its base-64 value? */
2157
2158#define FROM_BASE64(c) \
2159 (((c) >= 'A' && (c) <= 'Z') ? (c) - 'A' : \
2160 ((c) >= 'a' && (c) <= 'z') ? (c) - 'a' + 26 : \
2161 ((c) >= '0' && (c) <= '9') ? (c) - '0' + 52 : \
2162 (c) == '+' ? 62 : 63)
2163
2164/* What is the base-64 character of the bottom 6 bits of n? */
2165
2166#define TO_BASE64(n) \
2167 ("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/"[(n) & 0x3f])
2168
2169/* DECODE_DIRECT: this byte encountered in a UTF-7 string should be
2170 * decoded as itself. We are permissive on decoding; the only ASCII
2171 * byte not decoding to itself is the + which begins a base64
2172 * string. */
2173
2174#define DECODE_DIRECT(c) \
2175 ((c) <= 127 && (c) != '+')
2176
2177/* The UTF-7 encoder treats ASCII characters differently according to
2178 * whether they are Set D, Set O, Whitespace, or special (i.e. none of
2179 * the above). See RFC2152. This array identifies these different
2180 * sets:
2181 * 0 : "Set D"
2182 * alphanumeric and '(),-./:?
2183 * 1 : "Set O"
2184 * !"#$%&*;<=>@[]^_`{|}
2185 * 2 : "whitespace"
2186 * ht nl cr sp
2187 * 3 : special (must be base64 encoded)
2188 * everything else (i.e. +\~ and non-printing codes 0-8 11-12 14-31 127)
2189 */
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00002190
Tim Petersced69f82003-09-16 20:30:58 +00002191static
Antoine Pitrou244651a2009-05-04 18:56:13 +00002192char utf7_category[128] = {
2193/* nul soh stx etx eot enq ack bel bs ht nl vt np cr so si */
2194 3, 3, 3, 3, 3, 3, 3, 3, 3, 2, 2, 3, 3, 2, 3, 3,
2195/* dle dc1 dc2 dc3 dc4 nak syn etb can em sub esc fs gs rs us */
2196 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3,
2197/* sp ! " # $ % & ' ( ) * + , - . / */
2198 2, 1, 1, 1, 1, 1, 1, 0, 0, 0, 1, 3, 0, 0, 0, 0,
2199/* 0 1 2 3 4 5 6 7 8 9 : ; < = > ? */
2200 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 0,
2201/* @ A B C D E F G H I J K L M N O */
2202 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
2203/* P Q R S T U V W X Y Z [ \ ] ^ _ */
2204 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 3, 1, 1, 1,
2205/* ` a b c d e f g h i j k l m n o */
2206 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
2207/* p q r s t u v w x y z { | } ~ del */
2208 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 3, 3,
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00002209};
2210
Antoine Pitrou244651a2009-05-04 18:56:13 +00002211/* ENCODE_DIRECT: this character should be encoded as itself. The
2212 * answer depends on whether we are encoding set O as itself, and also
2213 * on whether we are encoding whitespace as itself. RFC2152 makes it
2214 * clear that the answers to these questions vary between
2215 * applications, so this code needs to be flexible. */
Marc-André Lemburge115ec82005-10-19 22:33:31 +00002216
Antoine Pitrou244651a2009-05-04 18:56:13 +00002217#define ENCODE_DIRECT(c, directO, directWS) \
2218 ((c) < 128 && (c) > 0 && \
2219 ((utf7_category[(c)] == 0) || \
2220 (directWS && (utf7_category[(c)] == 2)) || \
2221 (directO && (utf7_category[(c)] == 1))))
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00002222
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00002223PyObject *PyUnicode_DecodeUTF7(const char *s,
Benjamin Peterson29060642009-01-31 22:14:21 +00002224 Py_ssize_t size,
2225 const char *errors)
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00002226{
Christian Heimes5d14c2b2007-11-20 23:38:09 +00002227 return PyUnicode_DecodeUTF7Stateful(s, size, errors, NULL);
2228}
2229
Antoine Pitrou244651a2009-05-04 18:56:13 +00002230/* The decoder. The only state we preserve is our read position,
2231 * i.e. how many characters we have consumed. So if we end in the
2232 * middle of a shift sequence we have to back off the read position
2233 * and the output to the beginning of the sequence, otherwise we lose
2234 * all the shift state (seen bits, number of bits seen, high
2235 * surrogate). */
2236
Christian Heimes5d14c2b2007-11-20 23:38:09 +00002237PyObject *PyUnicode_DecodeUTF7Stateful(const char *s,
Benjamin Peterson29060642009-01-31 22:14:21 +00002238 Py_ssize_t size,
2239 const char *errors,
2240 Py_ssize_t *consumed)
Christian Heimes5d14c2b2007-11-20 23:38:09 +00002241{
Walter Dörwald3aeb6322002-09-02 13:14:32 +00002242 const char *starts = s;
Martin v. Löwis18e16552006-02-15 17:27:45 +00002243 Py_ssize_t startinpos;
2244 Py_ssize_t endinpos;
2245 Py_ssize_t outpos;
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00002246 const char *e;
2247 PyUnicodeObject *unicode;
2248 Py_UNICODE *p;
2249 const char *errmsg = "";
2250 int inShift = 0;
Antoine Pitrou244651a2009-05-04 18:56:13 +00002251 Py_UNICODE *shiftOutStart;
2252 unsigned int base64bits = 0;
2253 unsigned long base64buffer = 0;
2254 Py_UNICODE surrogate = 0;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00002255 PyObject *errorHandler = NULL;
2256 PyObject *exc = NULL;
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00002257
2258 unicode = _PyUnicode_New(size);
2259 if (!unicode)
2260 return NULL;
Christian Heimes5d14c2b2007-11-20 23:38:09 +00002261 if (size == 0) {
2262 if (consumed)
2263 *consumed = 0;
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00002264 return (PyObject *)unicode;
Christian Heimes5d14c2b2007-11-20 23:38:09 +00002265 }
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00002266
2267 p = unicode->str;
Antoine Pitrou244651a2009-05-04 18:56:13 +00002268 shiftOutStart = p;
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00002269 e = s + size;
2270
2271 while (s < e) {
Walter Dörwald3aeb6322002-09-02 13:14:32 +00002272 Py_UNICODE ch;
Benjamin Peterson29060642009-01-31 22:14:21 +00002273 restart:
Antoine Pitrou5ffd9e92008-07-25 18:05:24 +00002274 ch = (unsigned char) *s;
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00002275
Antoine Pitrou244651a2009-05-04 18:56:13 +00002276 if (inShift) { /* in a base-64 section */
2277 if (IS_BASE64(ch)) { /* consume a base-64 character */
2278 base64buffer = (base64buffer << 6) | FROM_BASE64(ch);
2279 base64bits += 6;
2280 s++;
2281 if (base64bits >= 16) {
2282 /* we have enough bits for a UTF-16 value */
2283 Py_UNICODE outCh = (Py_UNICODE)
2284 (base64buffer >> (base64bits-16));
2285 base64bits -= 16;
2286 base64buffer &= (1 << base64bits) - 1; /* clear high bits */
2287 if (surrogate) {
2288 /* expecting a second surrogate */
2289 if (outCh >= 0xDC00 && outCh <= 0xDFFF) {
2290#ifdef Py_UNICODE_WIDE
2291 *p++ = (((surrogate & 0x3FF)<<10)
2292 | (outCh & 0x3FF)) + 0x10000;
2293#else
2294 *p++ = surrogate;
2295 *p++ = outCh;
2296#endif
2297 surrogate = 0;
Antoine Pitrou5418ee02011-11-15 01:42:21 +01002298 continue;
Antoine Pitrou244651a2009-05-04 18:56:13 +00002299 }
2300 else {
Antoine Pitrou5418ee02011-11-15 01:42:21 +01002301 *p++ = surrogate;
Antoine Pitrou244651a2009-05-04 18:56:13 +00002302 surrogate = 0;
Antoine Pitrou244651a2009-05-04 18:56:13 +00002303 }
2304 }
Antoine Pitrou5418ee02011-11-15 01:42:21 +01002305 if (outCh >= 0xD800 && outCh <= 0xDBFF) {
Antoine Pitrou244651a2009-05-04 18:56:13 +00002306 /* first surrogate */
2307 surrogate = outCh;
2308 }
Antoine Pitrou244651a2009-05-04 18:56:13 +00002309 else {
2310 *p++ = outCh;
2311 }
2312 }
2313 }
2314 else { /* now leaving a base-64 section */
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00002315 inShift = 0;
2316 s++;
Antoine Pitrou244651a2009-05-04 18:56:13 +00002317 if (surrogate) {
Antoine Pitrou5418ee02011-11-15 01:42:21 +01002318 *p++ = surrogate;
2319 surrogate = 0;
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00002320 }
Antoine Pitrou244651a2009-05-04 18:56:13 +00002321 if (base64bits > 0) { /* left-over bits */
2322 if (base64bits >= 6) {
2323 /* We've seen at least one base-64 character */
2324 errmsg = "partial character in shift sequence";
2325 goto utf7Error;
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00002326 }
Antoine Pitrou244651a2009-05-04 18:56:13 +00002327 else {
2328 /* Some bits remain; they should be zero */
2329 if (base64buffer != 0) {
2330 errmsg = "non-zero padding bits in shift sequence";
2331 goto utf7Error;
2332 }
2333 }
2334 }
2335 if (ch != '-') {
2336 /* '-' is absorbed; other terminating
2337 characters are preserved */
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00002338 *p++ = ch;
2339 }
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00002340 }
2341 }
2342 else if ( ch == '+' ) {
Walter Dörwald3aeb6322002-09-02 13:14:32 +00002343 startinpos = s-starts;
Antoine Pitrou244651a2009-05-04 18:56:13 +00002344 s++; /* consume '+' */
2345 if (s < e && *s == '-') { /* '+-' encodes '+' */
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00002346 s++;
2347 *p++ = '+';
Antoine Pitrou244651a2009-05-04 18:56:13 +00002348 }
2349 else { /* begin base64-encoded section */
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00002350 inShift = 1;
Antoine Pitrou244651a2009-05-04 18:56:13 +00002351 shiftOutStart = p;
2352 base64bits = 0;
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00002353 }
2354 }
Antoine Pitrou244651a2009-05-04 18:56:13 +00002355 else if (DECODE_DIRECT(ch)) { /* character decodes as itself */
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00002356 *p++ = ch;
2357 s++;
2358 }
Antoine Pitrou244651a2009-05-04 18:56:13 +00002359 else {
2360 startinpos = s-starts;
2361 s++;
2362 errmsg = "unexpected special character";
2363 goto utf7Error;
2364 }
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00002365 continue;
Antoine Pitrou244651a2009-05-04 18:56:13 +00002366utf7Error:
Walter Dörwald3aeb6322002-09-02 13:14:32 +00002367 outpos = p-PyUnicode_AS_UNICODE(unicode);
2368 endinpos = s-starts;
2369 if (unicode_decode_call_errorhandler(
Benjamin Peterson29060642009-01-31 22:14:21 +00002370 errors, &errorHandler,
2371 "utf7", errmsg,
2372 &starts, &e, &startinpos, &endinpos, &exc, &s,
2373 &unicode, &outpos, &p))
2374 goto onError;
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00002375 }
2376
Antoine Pitrou244651a2009-05-04 18:56:13 +00002377 /* end of string */
2378
2379 if (inShift && !consumed) { /* in shift sequence, no more to follow */
2380 /* if we're in an inconsistent state, that's an error */
2381 if (surrogate ||
2382 (base64bits >= 6) ||
2383 (base64bits > 0 && base64buffer != 0)) {
2384 outpos = p-PyUnicode_AS_UNICODE(unicode);
2385 endinpos = size;
2386 if (unicode_decode_call_errorhandler(
2387 errors, &errorHandler,
2388 "utf7", "unterminated shift sequence",
2389 &starts, &e, &startinpos, &endinpos, &exc, &s,
2390 &unicode, &outpos, &p))
2391 goto onError;
2392 if (s < e)
2393 goto restart;
2394 }
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00002395 }
Antoine Pitrou244651a2009-05-04 18:56:13 +00002396
2397 /* return state */
Christian Heimes5d14c2b2007-11-20 23:38:09 +00002398 if (consumed) {
Antoine Pitrou244651a2009-05-04 18:56:13 +00002399 if (inShift) {
2400 p = shiftOutStart; /* back off output */
Christian Heimes5d14c2b2007-11-20 23:38:09 +00002401 *consumed = startinpos;
Antoine Pitrou244651a2009-05-04 18:56:13 +00002402 }
2403 else {
Christian Heimes5d14c2b2007-11-20 23:38:09 +00002404 *consumed = s-starts;
Antoine Pitrou244651a2009-05-04 18:56:13 +00002405 }
Christian Heimes5d14c2b2007-11-20 23:38:09 +00002406 }
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00002407
Jeremy Hyltondeb2dc62003-09-16 03:41:45 +00002408 if (_PyUnicode_Resize(&unicode, p - PyUnicode_AS_UNICODE(unicode)) < 0)
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00002409 goto onError;
2410
Walter Dörwald3aeb6322002-09-02 13:14:32 +00002411 Py_XDECREF(errorHandler);
2412 Py_XDECREF(exc);
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00002413 return (PyObject *)unicode;
2414
Benjamin Peterson29060642009-01-31 22:14:21 +00002415 onError:
Walter Dörwald3aeb6322002-09-02 13:14:32 +00002416 Py_XDECREF(errorHandler);
2417 Py_XDECREF(exc);
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00002418 Py_DECREF(unicode);
2419 return NULL;
2420}
2421
2422
2423PyObject *PyUnicode_EncodeUTF7(const Py_UNICODE *s,
Benjamin Peterson29060642009-01-31 22:14:21 +00002424 Py_ssize_t size,
Antoine Pitrou244651a2009-05-04 18:56:13 +00002425 int base64SetO,
2426 int base64WhiteSpace,
Benjamin Peterson29060642009-01-31 22:14:21 +00002427 const char *errors)
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00002428{
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00002429 PyObject *v;
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00002430 /* It might be possible to tighten this worst case */
Alexandre Vassalottie85bd982009-07-21 00:39:03 +00002431 Py_ssize_t allocated = 8 * size;
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00002432 int inShift = 0;
Martin v. Löwis18e16552006-02-15 17:27:45 +00002433 Py_ssize_t i = 0;
Antoine Pitrou244651a2009-05-04 18:56:13 +00002434 unsigned int base64bits = 0;
2435 unsigned long base64buffer = 0;
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00002436 char * out;
2437 char * start;
2438
2439 if (size == 0)
Benjamin Peterson29060642009-01-31 22:14:21 +00002440 return PyBytes_FromStringAndSize(NULL, 0);
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00002441
Alexandre Vassalottie85bd982009-07-21 00:39:03 +00002442 if (allocated / 8 != size)
Neal Norwitz3ce5d922008-08-24 07:08:55 +00002443 return PyErr_NoMemory();
2444
Antoine Pitrou244651a2009-05-04 18:56:13 +00002445 v = PyBytes_FromStringAndSize(NULL, allocated);
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00002446 if (v == NULL)
2447 return NULL;
2448
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00002449 start = out = PyBytes_AS_STRING(v);
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00002450 for (;i < size; ++i) {
2451 Py_UNICODE ch = s[i];
2452
Antoine Pitrou244651a2009-05-04 18:56:13 +00002453 if (inShift) {
2454 if (ENCODE_DIRECT(ch, !base64SetO, !base64WhiteSpace)) {
2455 /* shifting out */
2456 if (base64bits) { /* output remaining bits */
2457 *out++ = TO_BASE64(base64buffer << (6-base64bits));
2458 base64buffer = 0;
2459 base64bits = 0;
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00002460 }
2461 inShift = 0;
Antoine Pitrou244651a2009-05-04 18:56:13 +00002462 /* Characters not in the BASE64 set implicitly unshift the sequence
2463 so no '-' is required, except if the character is itself a '-' */
2464 if (IS_BASE64(ch) || ch == '-') {
2465 *out++ = '-';
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00002466 }
Antoine Pitrou244651a2009-05-04 18:56:13 +00002467 *out++ = (char) ch;
2468 }
2469 else {
2470 goto encode_char;
Tim Petersced69f82003-09-16 20:30:58 +00002471 }
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00002472 }
Antoine Pitrou244651a2009-05-04 18:56:13 +00002473 else { /* not in a shift sequence */
2474 if (ch == '+') {
2475 *out++ = '+';
2476 *out++ = '-';
2477 }
2478 else if (ENCODE_DIRECT(ch, !base64SetO, !base64WhiteSpace)) {
2479 *out++ = (char) ch;
2480 }
2481 else {
2482 *out++ = '+';
2483 inShift = 1;
2484 goto encode_char;
2485 }
2486 }
2487 continue;
2488encode_char:
2489#ifdef Py_UNICODE_WIDE
2490 if (ch >= 0x10000) {
2491 /* code first surrogate */
2492 base64bits += 16;
2493 base64buffer = (base64buffer << 16) | 0xd800 | ((ch-0x10000) >> 10);
2494 while (base64bits >= 6) {
2495 *out++ = TO_BASE64(base64buffer >> (base64bits-6));
2496 base64bits -= 6;
2497 }
2498 /* prepare second surrogate */
2499 ch = 0xDC00 | ((ch-0x10000) & 0x3FF);
2500 }
2501#endif
2502 base64bits += 16;
2503 base64buffer = (base64buffer << 16) | ch;
2504 while (base64bits >= 6) {
2505 *out++ = TO_BASE64(base64buffer >> (base64bits-6));
2506 base64bits -= 6;
2507 }
Hye-Shik Chang1bc09b72004-01-03 19:35:43 +00002508 }
Antoine Pitrou244651a2009-05-04 18:56:13 +00002509 if (base64bits)
2510 *out++= TO_BASE64(base64buffer << (6-base64bits) );
2511 if (inShift)
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00002512 *out++ = '-';
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00002513 if (_PyBytes_Resize(&v, out - start) < 0)
2514 return NULL;
2515 return v;
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00002516}
2517
Antoine Pitrou244651a2009-05-04 18:56:13 +00002518#undef IS_BASE64
2519#undef FROM_BASE64
2520#undef TO_BASE64
2521#undef DECODE_DIRECT
2522#undef ENCODE_DIRECT
Marc-André Lemburgc60e6f72001-09-20 10:35:46 +00002523
Guido van Rossumd57fd912000-03-10 22:53:23 +00002524/* --- UTF-8 Codec -------------------------------------------------------- */
2525
Tim Petersced69f82003-09-16 20:30:58 +00002526static
Guido van Rossumd57fd912000-03-10 22:53:23 +00002527char utf8_code_length[256] = {
Ezio Melotti57221d02010-07-01 07:32:02 +00002528 /* Map UTF-8 encoded prefix byte to sequence length. Zero means
2529 illegal prefix. See RFC 3629 for details */
2530 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, /* 00-0F */
2531 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
Victor Stinner4a2b7a12010-08-13 14:03:48 +00002532 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
Guido van Rossumd57fd912000-03-10 22:53:23 +00002533 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
2534 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
2535 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
2536 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
Ezio Melotti57221d02010-07-01 07:32:02 +00002537 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, /* 70-7F */
2538 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, /* 80-8F */
Guido van Rossumd57fd912000-03-10 22:53:23 +00002539 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
2540 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
Ezio Melotti57221d02010-07-01 07:32:02 +00002541 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, /* B0-BF */
2542 0, 0, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, /* C0-C1 + C2-CF */
2543 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, /* D0-DF */
2544 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, /* E0-EF */
2545 4, 4, 4, 4, 4, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 /* F0-F4 + F5-FF */
Guido van Rossumd57fd912000-03-10 22:53:23 +00002546};
2547
Guido van Rossumd57fd912000-03-10 22:53:23 +00002548PyObject *PyUnicode_DecodeUTF8(const char *s,
Benjamin Peterson29060642009-01-31 22:14:21 +00002549 Py_ssize_t size,
2550 const char *errors)
Guido van Rossumd57fd912000-03-10 22:53:23 +00002551{
Walter Dörwald69652032004-09-07 20:24:22 +00002552 return PyUnicode_DecodeUTF8Stateful(s, size, errors, NULL);
2553}
2554
Antoine Pitrouab868312009-01-10 15:40:25 +00002555/* Mask to check or force alignment of a pointer to C 'long' boundaries */
2556#define LONG_PTR_MASK (size_t) (SIZEOF_LONG - 1)
2557
2558/* Mask to quickly check whether a C 'long' contains a
2559 non-ASCII, UTF8-encoded char. */
2560#if (SIZEOF_LONG == 8)
2561# define ASCII_CHAR_MASK 0x8080808080808080L
2562#elif (SIZEOF_LONG == 4)
2563# define ASCII_CHAR_MASK 0x80808080L
2564#else
2565# error C 'long' size should be either 4 or 8!
2566#endif
2567
Walter Dörwald69652032004-09-07 20:24:22 +00002568PyObject *PyUnicode_DecodeUTF8Stateful(const char *s,
Benjamin Peterson29060642009-01-31 22:14:21 +00002569 Py_ssize_t size,
2570 const char *errors,
2571 Py_ssize_t *consumed)
Walter Dörwald69652032004-09-07 20:24:22 +00002572{
Walter Dörwald3aeb6322002-09-02 13:14:32 +00002573 const char *starts = s;
Guido van Rossumd57fd912000-03-10 22:53:23 +00002574 int n;
Ezio Melotti57221d02010-07-01 07:32:02 +00002575 int k;
Martin v. Löwis18e16552006-02-15 17:27:45 +00002576 Py_ssize_t startinpos;
2577 Py_ssize_t endinpos;
2578 Py_ssize_t outpos;
Antoine Pitrouab868312009-01-10 15:40:25 +00002579 const char *e, *aligned_end;
Guido van Rossumd57fd912000-03-10 22:53:23 +00002580 PyUnicodeObject *unicode;
2581 Py_UNICODE *p;
Marc-André Lemburg9542f482000-07-17 18:23:13 +00002582 const char *errmsg = "";
Walter Dörwald3aeb6322002-09-02 13:14:32 +00002583 PyObject *errorHandler = NULL;
2584 PyObject *exc = NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00002585
2586 /* Note: size will always be longer than the resulting Unicode
2587 character count */
2588 unicode = _PyUnicode_New(size);
2589 if (!unicode)
2590 return NULL;
Walter Dörwald69652032004-09-07 20:24:22 +00002591 if (size == 0) {
2592 if (consumed)
2593 *consumed = 0;
Guido van Rossumd57fd912000-03-10 22:53:23 +00002594 return (PyObject *)unicode;
Walter Dörwald69652032004-09-07 20:24:22 +00002595 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00002596
2597 /* Unpack UTF-8 encoded data */
2598 p = unicode->str;
2599 e = s + size;
Antoine Pitrouab868312009-01-10 15:40:25 +00002600 aligned_end = (const char *) ((size_t) e & ~LONG_PTR_MASK);
Guido van Rossumd57fd912000-03-10 22:53:23 +00002601
2602 while (s < e) {
Marc-André Lemburge12896e2000-07-07 17:51:08 +00002603 Py_UCS4 ch = (unsigned char)*s;
Guido van Rossumd57fd912000-03-10 22:53:23 +00002604
2605 if (ch < 0x80) {
Antoine Pitrouab868312009-01-10 15:40:25 +00002606 /* Fast path for runs of ASCII characters. Given that common UTF-8
2607 input will consist of an overwhelming majority of ASCII
2608 characters, we try to optimize for this case by checking
2609 as many characters as a C 'long' can contain.
2610 First, check if we can do an aligned read, as most CPUs have
2611 a penalty for unaligned reads.
2612 */
2613 if (!((size_t) s & LONG_PTR_MASK)) {
2614 /* Help register allocation */
2615 register const char *_s = s;
2616 register Py_UNICODE *_p = p;
2617 while (_s < aligned_end) {
2618 /* Read a whole long at a time (either 4 or 8 bytes),
2619 and do a fast unrolled copy if it only contains ASCII
2620 characters. */
2621 unsigned long data = *(unsigned long *) _s;
2622 if (data & ASCII_CHAR_MASK)
2623 break;
2624 _p[0] = (unsigned char) _s[0];
2625 _p[1] = (unsigned char) _s[1];
2626 _p[2] = (unsigned char) _s[2];
2627 _p[3] = (unsigned char) _s[3];
2628#if (SIZEOF_LONG == 8)
2629 _p[4] = (unsigned char) _s[4];
2630 _p[5] = (unsigned char) _s[5];
2631 _p[6] = (unsigned char) _s[6];
2632 _p[7] = (unsigned char) _s[7];
2633#endif
2634 _s += SIZEOF_LONG;
2635 _p += SIZEOF_LONG;
2636 }
2637 s = _s;
2638 p = _p;
2639 if (s == e)
2640 break;
2641 ch = (unsigned char)*s;
2642 }
2643 }
2644
2645 if (ch < 0x80) {
Marc-André Lemburge12896e2000-07-07 17:51:08 +00002646 *p++ = (Py_UNICODE)ch;
Guido van Rossumd57fd912000-03-10 22:53:23 +00002647 s++;
2648 continue;
2649 }
2650
2651 n = utf8_code_length[ch];
2652
Marc-André Lemburg9542f482000-07-17 18:23:13 +00002653 if (s + n > e) {
Benjamin Peterson29060642009-01-31 22:14:21 +00002654 if (consumed)
2655 break;
2656 else {
2657 errmsg = "unexpected end of data";
2658 startinpos = s-starts;
Ezio Melotti57221d02010-07-01 07:32:02 +00002659 endinpos = startinpos+1;
2660 for (k=1; (k < size-startinpos) && ((s[k]&0xC0) == 0x80); k++)
2661 endinpos++;
Benjamin Peterson29060642009-01-31 22:14:21 +00002662 goto utf8Error;
2663 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00002664 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00002665
2666 switch (n) {
2667
2668 case 0:
Ezio Melotti57221d02010-07-01 07:32:02 +00002669 errmsg = "invalid start byte";
Benjamin Peterson29060642009-01-31 22:14:21 +00002670 startinpos = s-starts;
2671 endinpos = startinpos+1;
2672 goto utf8Error;
Guido van Rossumd57fd912000-03-10 22:53:23 +00002673
2674 case 1:
Marc-André Lemburg9542f482000-07-17 18:23:13 +00002675 errmsg = "internal error";
Benjamin Peterson29060642009-01-31 22:14:21 +00002676 startinpos = s-starts;
2677 endinpos = startinpos+1;
2678 goto utf8Error;
Guido van Rossumd57fd912000-03-10 22:53:23 +00002679
2680 case 2:
Marc-André Lemburg9542f482000-07-17 18:23:13 +00002681 if ((s[1] & 0xc0) != 0x80) {
Ezio Melotti57221d02010-07-01 07:32:02 +00002682 errmsg = "invalid continuation byte";
Benjamin Peterson29060642009-01-31 22:14:21 +00002683 startinpos = s-starts;
Ezio Melotti57221d02010-07-01 07:32:02 +00002684 endinpos = startinpos + 1;
Benjamin Peterson29060642009-01-31 22:14:21 +00002685 goto utf8Error;
2686 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00002687 ch = ((s[0] & 0x1f) << 6) + (s[1] & 0x3f);
Ezio Melotti57221d02010-07-01 07:32:02 +00002688 assert ((ch > 0x007F) && (ch <= 0x07FF));
2689 *p++ = (Py_UNICODE)ch;
Guido van Rossumd57fd912000-03-10 22:53:23 +00002690 break;
2691
2692 case 3:
Ezio Melotti9bf2b3a2010-07-03 04:52:19 +00002693 /* Decoding UTF-8 sequences in range \xed\xa0\x80-\xed\xbf\xbf
2694 will result in surrogates in range d800-dfff. Surrogates are
2695 not valid UTF-8 so they are rejected.
2696 See http://www.unicode.org/versions/Unicode5.2.0/ch03.pdf
2697 (table 3-7) and http://www.rfc-editor.org/rfc/rfc3629.txt */
Tim Petersced69f82003-09-16 20:30:58 +00002698 if ((s[1] & 0xc0) != 0x80 ||
Ezio Melotti57221d02010-07-01 07:32:02 +00002699 (s[2] & 0xc0) != 0x80 ||
2700 ((unsigned char)s[0] == 0xE0 &&
2701 (unsigned char)s[1] < 0xA0) ||
2702 ((unsigned char)s[0] == 0xED &&
2703 (unsigned char)s[1] > 0x9F)) {
2704 errmsg = "invalid continuation byte";
Benjamin Peterson29060642009-01-31 22:14:21 +00002705 startinpos = s-starts;
Ezio Melotti57221d02010-07-01 07:32:02 +00002706 endinpos = startinpos + 1;
2707
2708 /* if s[1] first two bits are 1 and 0, then the invalid
2709 continuation byte is s[2], so increment endinpos by 1,
2710 if not, s[1] is invalid and endinpos doesn't need to
2711 be incremented. */
2712 if ((s[1] & 0xC0) == 0x80)
2713 endinpos++;
Benjamin Peterson29060642009-01-31 22:14:21 +00002714 goto utf8Error;
2715 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00002716 ch = ((s[0] & 0x0f) << 12) + ((s[1] & 0x3f) << 6) + (s[2] & 0x3f);
Ezio Melotti57221d02010-07-01 07:32:02 +00002717 assert ((ch > 0x07FF) && (ch <= 0xFFFF));
2718 *p++ = (Py_UNICODE)ch;
Marc-André Lemburge12896e2000-07-07 17:51:08 +00002719 break;
2720
2721 case 4:
2722 if ((s[1] & 0xc0) != 0x80 ||
2723 (s[2] & 0xc0) != 0x80 ||
Ezio Melotti57221d02010-07-01 07:32:02 +00002724 (s[3] & 0xc0) != 0x80 ||
2725 ((unsigned char)s[0] == 0xF0 &&
2726 (unsigned char)s[1] < 0x90) ||
2727 ((unsigned char)s[0] == 0xF4 &&
2728 (unsigned char)s[1] > 0x8F)) {
2729 errmsg = "invalid continuation byte";
Benjamin Peterson29060642009-01-31 22:14:21 +00002730 startinpos = s-starts;
Ezio Melotti57221d02010-07-01 07:32:02 +00002731 endinpos = startinpos + 1;
2732 if ((s[1] & 0xC0) == 0x80) {
2733 endinpos++;
2734 if ((s[2] & 0xC0) == 0x80)
2735 endinpos++;
2736 }
Benjamin Peterson29060642009-01-31 22:14:21 +00002737 goto utf8Error;
2738 }
Marc-André Lemburge12896e2000-07-07 17:51:08 +00002739 ch = ((s[0] & 0x7) << 18) + ((s[1] & 0x3f) << 12) +
Ezio Melotti57221d02010-07-01 07:32:02 +00002740 ((s[2] & 0x3f) << 6) + (s[3] & 0x3f);
2741 assert ((ch > 0xFFFF) && (ch <= 0x10ffff));
2742
Fredrik Lundh8f455852001-06-27 18:59:43 +00002743#ifdef Py_UNICODE_WIDE
Benjamin Peterson29060642009-01-31 22:14:21 +00002744 *p++ = (Py_UNICODE)ch;
Martin v. Löwis0ba70cc2001-06-26 22:22:37 +00002745#else
Marc-André Lemburge12896e2000-07-07 17:51:08 +00002746 /* compute and append the two surrogates: */
Tim Petersced69f82003-09-16 20:30:58 +00002747
Marc-André Lemburge12896e2000-07-07 17:51:08 +00002748 /* translate from 10000..10FFFF to 0..FFFF */
2749 ch -= 0x10000;
Tim Petersced69f82003-09-16 20:30:58 +00002750
Marc-André Lemburge12896e2000-07-07 17:51:08 +00002751 /* high surrogate = top 10 bits added to D800 */
2752 *p++ = (Py_UNICODE)(0xD800 + (ch >> 10));
Tim Petersced69f82003-09-16 20:30:58 +00002753
Marc-André Lemburge12896e2000-07-07 17:51:08 +00002754 /* low surrogate = bottom 10 bits added to DC00 */
Fredrik Lundh45714e92001-06-26 16:39:36 +00002755 *p++ = (Py_UNICODE)(0xDC00 + (ch & 0x03FF));
Martin v. Löwis0ba70cc2001-06-26 22:22:37 +00002756#endif
Guido van Rossumd57fd912000-03-10 22:53:23 +00002757 break;
Guido van Rossumd57fd912000-03-10 22:53:23 +00002758 }
2759 s += n;
Benjamin Peterson29060642009-01-31 22:14:21 +00002760 continue;
Tim Petersced69f82003-09-16 20:30:58 +00002761
Benjamin Peterson29060642009-01-31 22:14:21 +00002762 utf8Error:
2763 outpos = p-PyUnicode_AS_UNICODE(unicode);
2764 if (unicode_decode_call_errorhandler(
2765 errors, &errorHandler,
Victor Stinnercbe01342012-02-14 01:17:45 +01002766 "utf-8", errmsg,
Benjamin Peterson29060642009-01-31 22:14:21 +00002767 &starts, &e, &startinpos, &endinpos, &exc, &s,
2768 &unicode, &outpos, &p))
2769 goto onError;
2770 aligned_end = (const char *) ((size_t) e & ~LONG_PTR_MASK);
Guido van Rossumd57fd912000-03-10 22:53:23 +00002771 }
Walter Dörwald69652032004-09-07 20:24:22 +00002772 if (consumed)
Benjamin Peterson29060642009-01-31 22:14:21 +00002773 *consumed = s-starts;
Guido van Rossumd57fd912000-03-10 22:53:23 +00002774
2775 /* Adjust length */
Jeremy Hyltondeb2dc62003-09-16 03:41:45 +00002776 if (_PyUnicode_Resize(&unicode, p - unicode->str) < 0)
Guido van Rossumd57fd912000-03-10 22:53:23 +00002777 goto onError;
2778
Walter Dörwald3aeb6322002-09-02 13:14:32 +00002779 Py_XDECREF(errorHandler);
2780 Py_XDECREF(exc);
Guido van Rossumd57fd912000-03-10 22:53:23 +00002781 return (PyObject *)unicode;
2782
Benjamin Peterson29060642009-01-31 22:14:21 +00002783 onError:
Walter Dörwald3aeb6322002-09-02 13:14:32 +00002784 Py_XDECREF(errorHandler);
2785 Py_XDECREF(exc);
Guido van Rossumd57fd912000-03-10 22:53:23 +00002786 Py_DECREF(unicode);
2787 return NULL;
2788}
2789
Antoine Pitrouab868312009-01-10 15:40:25 +00002790#undef ASCII_CHAR_MASK
2791
Victor Stinnerf933e1a2010-10-20 22:58:25 +00002792#ifdef __APPLE__
2793
2794/* Simplified UTF-8 decoder using surrogateescape error handler,
2795 used to decode the command line arguments on Mac OS X. */
2796
2797wchar_t*
2798_Py_DecodeUTF8_surrogateescape(const char *s, Py_ssize_t size)
2799{
2800 int n;
2801 const char *e;
2802 wchar_t *unicode, *p;
2803
2804 /* Note: size will always be longer than the resulting Unicode
2805 character count */
2806 if (PY_SSIZE_T_MAX / sizeof(wchar_t) < (size + 1)) {
2807 PyErr_NoMemory();
2808 return NULL;
2809 }
2810 unicode = PyMem_Malloc((size + 1) * sizeof(wchar_t));
2811 if (!unicode)
2812 return NULL;
2813
2814 /* Unpack UTF-8 encoded data */
2815 p = unicode;
2816 e = s + size;
2817 while (s < e) {
2818 Py_UCS4 ch = (unsigned char)*s;
2819
2820 if (ch < 0x80) {
2821 *p++ = (wchar_t)ch;
2822 s++;
2823 continue;
2824 }
2825
2826 n = utf8_code_length[ch];
2827 if (s + n > e) {
2828 goto surrogateescape;
2829 }
2830
2831 switch (n) {
2832 case 0:
2833 case 1:
2834 goto surrogateescape;
2835
2836 case 2:
2837 if ((s[1] & 0xc0) != 0x80)
2838 goto surrogateescape;
2839 ch = ((s[0] & 0x1f) << 6) + (s[1] & 0x3f);
2840 assert ((ch > 0x007F) && (ch <= 0x07FF));
2841 *p++ = (wchar_t)ch;
2842 break;
2843
2844 case 3:
2845 /* Decoding UTF-8 sequences in range \xed\xa0\x80-\xed\xbf\xbf
2846 will result in surrogates in range d800-dfff. Surrogates are
2847 not valid UTF-8 so they are rejected.
2848 See http://www.unicode.org/versions/Unicode5.2.0/ch03.pdf
2849 (table 3-7) and http://www.rfc-editor.org/rfc/rfc3629.txt */
2850 if ((s[1] & 0xc0) != 0x80 ||
2851 (s[2] & 0xc0) != 0x80 ||
2852 ((unsigned char)s[0] == 0xE0 &&
2853 (unsigned char)s[1] < 0xA0) ||
2854 ((unsigned char)s[0] == 0xED &&
2855 (unsigned char)s[1] > 0x9F)) {
2856
2857 goto surrogateescape;
2858 }
2859 ch = ((s[0] & 0x0f) << 12) + ((s[1] & 0x3f) << 6) + (s[2] & 0x3f);
2860 assert ((ch > 0x07FF) && (ch <= 0xFFFF));
2861 *p++ = (Py_UNICODE)ch;
2862 break;
2863
2864 case 4:
2865 if ((s[1] & 0xc0) != 0x80 ||
2866 (s[2] & 0xc0) != 0x80 ||
2867 (s[3] & 0xc0) != 0x80 ||
2868 ((unsigned char)s[0] == 0xF0 &&
2869 (unsigned char)s[1] < 0x90) ||
2870 ((unsigned char)s[0] == 0xF4 &&
2871 (unsigned char)s[1] > 0x8F)) {
2872 goto surrogateescape;
2873 }
2874 ch = ((s[0] & 0x7) << 18) + ((s[1] & 0x3f) << 12) +
2875 ((s[2] & 0x3f) << 6) + (s[3] & 0x3f);
2876 assert ((ch > 0xFFFF) && (ch <= 0x10ffff));
2877
2878#if SIZEOF_WCHAR_T == 4
2879 *p++ = (wchar_t)ch;
2880#else
2881 /* compute and append the two surrogates: */
2882
2883 /* translate from 10000..10FFFF to 0..FFFF */
2884 ch -= 0x10000;
2885
2886 /* high surrogate = top 10 bits added to D800 */
2887 *p++ = (wchar_t)(0xD800 + (ch >> 10));
2888
2889 /* low surrogate = bottom 10 bits added to DC00 */
2890 *p++ = (wchar_t)(0xDC00 + (ch & 0x03FF));
2891#endif
2892 break;
2893 }
2894 s += n;
2895 continue;
2896
2897 surrogateescape:
2898 *p++ = 0xDC00 + ch;
2899 s++;
2900 }
2901 *p = L'\0';
2902 return unicode;
2903}
2904
2905#endif /* __APPLE__ */
Antoine Pitrouab868312009-01-10 15:40:25 +00002906
Tim Peters602f7402002-04-27 18:03:26 +00002907/* Allocation strategy: if the string is short, convert into a stack buffer
2908 and allocate exactly as much space needed at the end. Else allocate the
2909 maximum possible needed (4 result bytes per Unicode character), and return
2910 the excess memory at the end.
Martin v. Löwis2a7ff352002-04-21 09:59:45 +00002911*/
Tim Peters7e3d9612002-04-21 03:26:37 +00002912PyObject *
2913PyUnicode_EncodeUTF8(const Py_UNICODE *s,
Benjamin Peterson29060642009-01-31 22:14:21 +00002914 Py_ssize_t size,
2915 const char *errors)
Guido van Rossumd57fd912000-03-10 22:53:23 +00002916{
Tim Peters602f7402002-04-27 18:03:26 +00002917#define MAX_SHORT_UNICHARS 300 /* largest size we'll do on the stack */
Tim Peters0eca65c2002-04-21 17:28:06 +00002918
Guido van Rossum98297ee2007-11-06 21:34:58 +00002919 Py_ssize_t i; /* index into s of next input byte */
2920 PyObject *result; /* result string object */
2921 char *p; /* next free byte in output buffer */
2922 Py_ssize_t nallocated; /* number of result bytes allocated */
2923 Py_ssize_t nneeded; /* number of result bytes needed */
Tim Peters602f7402002-04-27 18:03:26 +00002924 char stackbuf[MAX_SHORT_UNICHARS * 4];
Martin v. Löwisdb12d452009-05-02 18:52:14 +00002925 PyObject *errorHandler = NULL;
2926 PyObject *exc = NULL;
Marc-André Lemburgbd3be8f2002-02-07 11:33:49 +00002927
Tim Peters602f7402002-04-27 18:03:26 +00002928 assert(s != NULL);
2929 assert(size >= 0);
Guido van Rossumd57fd912000-03-10 22:53:23 +00002930
Tim Peters602f7402002-04-27 18:03:26 +00002931 if (size <= MAX_SHORT_UNICHARS) {
2932 /* Write into the stack buffer; nallocated can't overflow.
2933 * At the end, we'll allocate exactly as much heap space as it
2934 * turns out we need.
2935 */
2936 nallocated = Py_SAFE_DOWNCAST(sizeof(stackbuf), size_t, int);
Guido van Rossum98297ee2007-11-06 21:34:58 +00002937 result = NULL; /* will allocate after we're done */
Tim Peters602f7402002-04-27 18:03:26 +00002938 p = stackbuf;
2939 }
2940 else {
2941 /* Overallocate on the heap, and give the excess back at the end. */
2942 nallocated = size * 4;
2943 if (nallocated / 4 != size) /* overflow! */
2944 return PyErr_NoMemory();
Christian Heimes72b710a2008-05-26 13:28:38 +00002945 result = PyBytes_FromStringAndSize(NULL, nallocated);
Guido van Rossum98297ee2007-11-06 21:34:58 +00002946 if (result == NULL)
Tim Peters602f7402002-04-27 18:03:26 +00002947 return NULL;
Christian Heimes72b710a2008-05-26 13:28:38 +00002948 p = PyBytes_AS_STRING(result);
Tim Peters602f7402002-04-27 18:03:26 +00002949 }
Martin v. Löwis2a7ff352002-04-21 09:59:45 +00002950
Tim Peters602f7402002-04-27 18:03:26 +00002951 for (i = 0; i < size;) {
Marc-André Lemburge12896e2000-07-07 17:51:08 +00002952 Py_UCS4 ch = s[i++];
Marc-André Lemburg3688a882002-02-06 18:09:02 +00002953
Martin v. Löwis2a7ff352002-04-21 09:59:45 +00002954 if (ch < 0x80)
Tim Peters602f7402002-04-27 18:03:26 +00002955 /* Encode ASCII */
Guido van Rossumd57fd912000-03-10 22:53:23 +00002956 *p++ = (char) ch;
Marc-André Lemburg3688a882002-02-06 18:09:02 +00002957
Guido van Rossumd57fd912000-03-10 22:53:23 +00002958 else if (ch < 0x0800) {
Tim Peters602f7402002-04-27 18:03:26 +00002959 /* Encode Latin-1 */
Marc-André Lemburgdc724d62002-02-06 18:20:19 +00002960 *p++ = (char)(0xc0 | (ch >> 6));
2961 *p++ = (char)(0x80 | (ch & 0x3f));
Victor Stinner31be90b2010-04-22 19:38:16 +00002962 } else if (0xD800 <= ch && ch <= 0xDFFF) {
Martin v. Löwisdb12d452009-05-02 18:52:14 +00002963#ifndef Py_UNICODE_WIDE
Victor Stinner31be90b2010-04-22 19:38:16 +00002964 /* Special case: check for high and low surrogate */
2965 if (ch <= 0xDBFF && i != size && 0xDC00 <= s[i] && s[i] <= 0xDFFF) {
2966 Py_UCS4 ch2 = s[i];
2967 /* Combine the two surrogates to form a UCS4 value */
2968 ch = ((ch - 0xD800) << 10 | (ch2 - 0xDC00)) + 0x10000;
2969 i++;
2970
2971 /* Encode UCS4 Unicode ordinals */
2972 *p++ = (char)(0xf0 | (ch >> 18));
2973 *p++ = (char)(0x80 | ((ch >> 12) & 0x3f));
Tim Peters602f7402002-04-27 18:03:26 +00002974 *p++ = (char)(0x80 | ((ch >> 6) & 0x3f));
2975 *p++ = (char)(0x80 | (ch & 0x3f));
Victor Stinner31be90b2010-04-22 19:38:16 +00002976 } else {
Victor Stinner445a6232010-04-22 20:01:57 +00002977#endif
Victor Stinner31be90b2010-04-22 19:38:16 +00002978 Py_ssize_t newpos;
2979 PyObject *rep;
2980 Py_ssize_t repsize, k;
2981 rep = unicode_encode_call_errorhandler
2982 (errors, &errorHandler, "utf-8", "surrogates not allowed",
2983 s, size, &exc, i-1, i, &newpos);
2984 if (!rep)
2985 goto error;
2986
2987 if (PyBytes_Check(rep))
2988 repsize = PyBytes_GET_SIZE(rep);
2989 else
2990 repsize = PyUnicode_GET_SIZE(rep);
2991
2992 if (repsize > 4) {
2993 Py_ssize_t offset;
2994
2995 if (result == NULL)
2996 offset = p - stackbuf;
2997 else
2998 offset = p - PyBytes_AS_STRING(result);
2999
3000 if (nallocated > PY_SSIZE_T_MAX - repsize + 4) {
3001 /* integer overflow */
3002 PyErr_NoMemory();
3003 goto error;
3004 }
3005 nallocated += repsize - 4;
3006 if (result != NULL) {
3007 if (_PyBytes_Resize(&result, nallocated) < 0)
3008 goto error;
3009 } else {
3010 result = PyBytes_FromStringAndSize(NULL, nallocated);
3011 if (result == NULL)
3012 goto error;
3013 Py_MEMCPY(PyBytes_AS_STRING(result), stackbuf, offset);
3014 }
3015 p = PyBytes_AS_STRING(result) + offset;
3016 }
3017
3018 if (PyBytes_Check(rep)) {
3019 char *prep = PyBytes_AS_STRING(rep);
3020 for(k = repsize; k > 0; k--)
3021 *p++ = *prep++;
3022 } else /* rep is unicode */ {
3023 Py_UNICODE *prep = PyUnicode_AS_UNICODE(rep);
3024 Py_UNICODE c;
3025
3026 for(k=0; k<repsize; k++) {
3027 c = prep[k];
3028 if (0x80 <= c) {
3029 raise_encode_exception(&exc, "utf-8", s, size,
3030 i-1, i, "surrogates not allowed");
3031 goto error;
3032 }
3033 *p++ = (char)prep[k];
3034 }
3035 }
3036 Py_DECREF(rep);
Victor Stinner445a6232010-04-22 20:01:57 +00003037#ifndef Py_UNICODE_WIDE
Victor Stinner31be90b2010-04-22 19:38:16 +00003038 }
Victor Stinner445a6232010-04-22 20:01:57 +00003039#endif
Victor Stinner31be90b2010-04-22 19:38:16 +00003040 } else if (ch < 0x10000) {
3041 *p++ = (char)(0xe0 | (ch >> 12));
3042 *p++ = (char)(0x80 | ((ch >> 6) & 0x3f));
3043 *p++ = (char)(0x80 | (ch & 0x3f));
3044 } else /* ch >= 0x10000 */ {
Tim Peters602f7402002-04-27 18:03:26 +00003045 /* Encode UCS4 Unicode ordinals */
3046 *p++ = (char)(0xf0 | (ch >> 18));
3047 *p++ = (char)(0x80 | ((ch >> 12) & 0x3f));
3048 *p++ = (char)(0x80 | ((ch >> 6) & 0x3f));
3049 *p++ = (char)(0x80 | (ch & 0x3f));
3050 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00003051 }
Tim Peters0eca65c2002-04-21 17:28:06 +00003052
Guido van Rossum98297ee2007-11-06 21:34:58 +00003053 if (result == NULL) {
Tim Peters602f7402002-04-27 18:03:26 +00003054 /* This was stack allocated. */
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00003055 nneeded = p - stackbuf;
Tim Peters602f7402002-04-27 18:03:26 +00003056 assert(nneeded <= nallocated);
Christian Heimes72b710a2008-05-26 13:28:38 +00003057 result = PyBytes_FromStringAndSize(stackbuf, nneeded);
Tim Peters602f7402002-04-27 18:03:26 +00003058 }
3059 else {
Christian Heimesf3863112007-11-22 07:46:41 +00003060 /* Cut back to size actually needed. */
Christian Heimes72b710a2008-05-26 13:28:38 +00003061 nneeded = p - PyBytes_AS_STRING(result);
Tim Peters602f7402002-04-27 18:03:26 +00003062 assert(nneeded <= nallocated);
Christian Heimes72b710a2008-05-26 13:28:38 +00003063 _PyBytes_Resize(&result, nneeded);
Tim Peters602f7402002-04-27 18:03:26 +00003064 }
Martin v. Löwisdb12d452009-05-02 18:52:14 +00003065 Py_XDECREF(errorHandler);
3066 Py_XDECREF(exc);
Guido van Rossum98297ee2007-11-06 21:34:58 +00003067 return result;
Martin v. Löwisdb12d452009-05-02 18:52:14 +00003068 error:
3069 Py_XDECREF(errorHandler);
3070 Py_XDECREF(exc);
3071 Py_XDECREF(result);
3072 return NULL;
Martin v. Löwis2a7ff352002-04-21 09:59:45 +00003073
Tim Peters602f7402002-04-27 18:03:26 +00003074#undef MAX_SHORT_UNICHARS
Guido van Rossumd57fd912000-03-10 22:53:23 +00003075}
3076
Guido van Rossumd57fd912000-03-10 22:53:23 +00003077PyObject *PyUnicode_AsUTF8String(PyObject *unicode)
3078{
Guido van Rossumd57fd912000-03-10 22:53:23 +00003079 if (!PyUnicode_Check(unicode)) {
3080 PyErr_BadArgument();
3081 return NULL;
3082 }
Barry Warsaw2dd4abf2000-08-18 06:58:15 +00003083 return PyUnicode_EncodeUTF8(PyUnicode_AS_UNICODE(unicode),
Benjamin Peterson29060642009-01-31 22:14:21 +00003084 PyUnicode_GET_SIZE(unicode),
3085 NULL);
Guido van Rossumd57fd912000-03-10 22:53:23 +00003086}
3087
Walter Dörwald41980ca2007-08-16 21:55:45 +00003088/* --- UTF-32 Codec ------------------------------------------------------- */
3089
3090PyObject *
3091PyUnicode_DecodeUTF32(const char *s,
Benjamin Peterson29060642009-01-31 22:14:21 +00003092 Py_ssize_t size,
3093 const char *errors,
3094 int *byteorder)
Walter Dörwald41980ca2007-08-16 21:55:45 +00003095{
3096 return PyUnicode_DecodeUTF32Stateful(s, size, errors, byteorder, NULL);
3097}
3098
3099PyObject *
3100PyUnicode_DecodeUTF32Stateful(const char *s,
Benjamin Peterson29060642009-01-31 22:14:21 +00003101 Py_ssize_t size,
3102 const char *errors,
3103 int *byteorder,
3104 Py_ssize_t *consumed)
Walter Dörwald41980ca2007-08-16 21:55:45 +00003105{
3106 const char *starts = s;
3107 Py_ssize_t startinpos;
3108 Py_ssize_t endinpos;
3109 Py_ssize_t outpos;
3110 PyUnicodeObject *unicode;
3111 Py_UNICODE *p;
3112#ifndef Py_UNICODE_WIDE
Antoine Pitroucc0cfd32010-06-11 21:46:32 +00003113 int pairs = 0;
Mark Dickinson7db923c2010-06-12 09:10:14 +00003114 const unsigned char *qq;
Walter Dörwald41980ca2007-08-16 21:55:45 +00003115#else
3116 const int pairs = 0;
3117#endif
Mark Dickinson7db923c2010-06-12 09:10:14 +00003118 const unsigned char *q, *e;
Walter Dörwald41980ca2007-08-16 21:55:45 +00003119 int bo = 0; /* assume native ordering by default */
3120 const char *errmsg = "";
Walter Dörwald41980ca2007-08-16 21:55:45 +00003121 /* Offsets from q for retrieving bytes in the right order. */
3122#ifdef BYTEORDER_IS_LITTLE_ENDIAN
3123 int iorder[] = {0, 1, 2, 3};
3124#else
3125 int iorder[] = {3, 2, 1, 0};
3126#endif
3127 PyObject *errorHandler = NULL;
3128 PyObject *exc = NULL;
Victor Stinner313a1202010-06-11 23:56:51 +00003129
Walter Dörwald41980ca2007-08-16 21:55:45 +00003130 q = (unsigned char *)s;
3131 e = q + size;
3132
3133 if (byteorder)
3134 bo = *byteorder;
3135
3136 /* Check for BOM marks (U+FEFF) in the input and adjust current
3137 byte order setting accordingly. In native mode, the leading BOM
3138 mark is skipped, in all other modes, it is copied to the output
3139 stream as-is (giving a ZWNBSP character). */
3140 if (bo == 0) {
3141 if (size >= 4) {
3142 const Py_UCS4 bom = (q[iorder[3]] << 24) | (q[iorder[2]] << 16) |
Benjamin Peterson29060642009-01-31 22:14:21 +00003143 (q[iorder[1]] << 8) | q[iorder[0]];
Walter Dörwald41980ca2007-08-16 21:55:45 +00003144#ifdef BYTEORDER_IS_LITTLE_ENDIAN
Benjamin Peterson29060642009-01-31 22:14:21 +00003145 if (bom == 0x0000FEFF) {
3146 q += 4;
3147 bo = -1;
3148 }
3149 else if (bom == 0xFFFE0000) {
3150 q += 4;
3151 bo = 1;
3152 }
Walter Dörwald41980ca2007-08-16 21:55:45 +00003153#else
Benjamin Peterson29060642009-01-31 22:14:21 +00003154 if (bom == 0x0000FEFF) {
3155 q += 4;
3156 bo = 1;
3157 }
3158 else if (bom == 0xFFFE0000) {
3159 q += 4;
3160 bo = -1;
3161 }
Walter Dörwald41980ca2007-08-16 21:55:45 +00003162#endif
Benjamin Peterson29060642009-01-31 22:14:21 +00003163 }
Walter Dörwald41980ca2007-08-16 21:55:45 +00003164 }
3165
3166 if (bo == -1) {
3167 /* force LE */
3168 iorder[0] = 0;
3169 iorder[1] = 1;
3170 iorder[2] = 2;
3171 iorder[3] = 3;
3172 }
3173 else if (bo == 1) {
3174 /* force BE */
3175 iorder[0] = 3;
3176 iorder[1] = 2;
3177 iorder[2] = 1;
3178 iorder[3] = 0;
3179 }
3180
Antoine Pitroucc0cfd32010-06-11 21:46:32 +00003181 /* On narrow builds we split characters outside the BMP into two
3182 codepoints => count how much extra space we need. */
3183#ifndef Py_UNICODE_WIDE
3184 for (qq = q; qq < e; qq += 4)
3185 if (qq[iorder[2]] != 0 || qq[iorder[3]] != 0)
3186 pairs++;
3187#endif
3188
3189 /* This might be one to much, because of a BOM */
3190 unicode = _PyUnicode_New((size+3)/4+pairs);
3191 if (!unicode)
3192 return NULL;
3193 if (size == 0)
3194 return (PyObject *)unicode;
3195
3196 /* Unpack UTF-32 encoded data */
3197 p = unicode->str;
3198
Walter Dörwald41980ca2007-08-16 21:55:45 +00003199 while (q < e) {
Benjamin Peterson29060642009-01-31 22:14:21 +00003200 Py_UCS4 ch;
3201 /* remaining bytes at the end? (size should be divisible by 4) */
3202 if (e-q<4) {
3203 if (consumed)
3204 break;
3205 errmsg = "truncated data";
3206 startinpos = ((const char *)q)-starts;
3207 endinpos = ((const char *)e)-starts;
3208 goto utf32Error;
3209 /* The remaining input chars are ignored if the callback
3210 chooses to skip the input */
3211 }
3212 ch = (q[iorder[3]] << 24) | (q[iorder[2]] << 16) |
3213 (q[iorder[1]] << 8) | q[iorder[0]];
Walter Dörwald41980ca2007-08-16 21:55:45 +00003214
Benjamin Peterson29060642009-01-31 22:14:21 +00003215 if (ch >= 0x110000)
3216 {
3217 errmsg = "codepoint not in range(0x110000)";
3218 startinpos = ((const char *)q)-starts;
3219 endinpos = startinpos+4;
3220 goto utf32Error;
3221 }
Walter Dörwald41980ca2007-08-16 21:55:45 +00003222#ifndef Py_UNICODE_WIDE
Benjamin Peterson29060642009-01-31 22:14:21 +00003223 if (ch >= 0x10000)
3224 {
3225 *p++ = 0xD800 | ((ch-0x10000) >> 10);
3226 *p++ = 0xDC00 | ((ch-0x10000) & 0x3FF);
3227 }
3228 else
Walter Dörwald41980ca2007-08-16 21:55:45 +00003229#endif
Benjamin Peterson29060642009-01-31 22:14:21 +00003230 *p++ = ch;
3231 q += 4;
3232 continue;
3233 utf32Error:
3234 outpos = p-PyUnicode_AS_UNICODE(unicode);
3235 if (unicode_decode_call_errorhandler(
3236 errors, &errorHandler,
3237 "utf32", errmsg,
3238 &starts, (const char **)&e, &startinpos, &endinpos, &exc, (const char **)&q,
3239 &unicode, &outpos, &p))
3240 goto onError;
Walter Dörwald41980ca2007-08-16 21:55:45 +00003241 }
3242
3243 if (byteorder)
3244 *byteorder = bo;
3245
3246 if (consumed)
Benjamin Peterson29060642009-01-31 22:14:21 +00003247 *consumed = (const char *)q-starts;
Walter Dörwald41980ca2007-08-16 21:55:45 +00003248
3249 /* Adjust length */
3250 if (_PyUnicode_Resize(&unicode, p - unicode->str) < 0)
3251 goto onError;
3252
3253 Py_XDECREF(errorHandler);
3254 Py_XDECREF(exc);
3255 return (PyObject *)unicode;
3256
Benjamin Peterson29060642009-01-31 22:14:21 +00003257 onError:
Walter Dörwald41980ca2007-08-16 21:55:45 +00003258 Py_DECREF(unicode);
3259 Py_XDECREF(errorHandler);
3260 Py_XDECREF(exc);
3261 return NULL;
3262}
3263
3264PyObject *
3265PyUnicode_EncodeUTF32(const Py_UNICODE *s,
Benjamin Peterson29060642009-01-31 22:14:21 +00003266 Py_ssize_t size,
3267 const char *errors,
3268 int byteorder)
Walter Dörwald41980ca2007-08-16 21:55:45 +00003269{
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00003270 PyObject *v;
Walter Dörwald41980ca2007-08-16 21:55:45 +00003271 unsigned char *p;
Neal Norwitz3ce5d922008-08-24 07:08:55 +00003272 Py_ssize_t nsize, bytesize;
Walter Dörwald41980ca2007-08-16 21:55:45 +00003273#ifndef Py_UNICODE_WIDE
Neal Norwitz3ce5d922008-08-24 07:08:55 +00003274 Py_ssize_t i, pairs;
Walter Dörwald41980ca2007-08-16 21:55:45 +00003275#else
3276 const int pairs = 0;
3277#endif
3278 /* Offsets from p for storing byte pairs in the right order. */
3279#ifdef BYTEORDER_IS_LITTLE_ENDIAN
3280 int iorder[] = {0, 1, 2, 3};
3281#else
3282 int iorder[] = {3, 2, 1, 0};
3283#endif
3284
Benjamin Peterson29060642009-01-31 22:14:21 +00003285#define STORECHAR(CH) \
3286 do { \
3287 p[iorder[3]] = ((CH) >> 24) & 0xff; \
3288 p[iorder[2]] = ((CH) >> 16) & 0xff; \
3289 p[iorder[1]] = ((CH) >> 8) & 0xff; \
3290 p[iorder[0]] = (CH) & 0xff; \
3291 p += 4; \
Walter Dörwald41980ca2007-08-16 21:55:45 +00003292 } while(0)
3293
3294 /* In narrow builds we can output surrogate pairs as one codepoint,
3295 so we need less space. */
3296#ifndef Py_UNICODE_WIDE
3297 for (i = pairs = 0; i < size-1; i++)
Benjamin Peterson29060642009-01-31 22:14:21 +00003298 if (0xD800 <= s[i] && s[i] <= 0xDBFF &&
3299 0xDC00 <= s[i+1] && s[i+1] <= 0xDFFF)
3300 pairs++;
Walter Dörwald41980ca2007-08-16 21:55:45 +00003301#endif
Neal Norwitz3ce5d922008-08-24 07:08:55 +00003302 nsize = (size - pairs + (byteorder == 0));
3303 bytesize = nsize * 4;
3304 if (bytesize / 4 != nsize)
Benjamin Peterson29060642009-01-31 22:14:21 +00003305 return PyErr_NoMemory();
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00003306 v = PyBytes_FromStringAndSize(NULL, bytesize);
Walter Dörwald41980ca2007-08-16 21:55:45 +00003307 if (v == NULL)
3308 return NULL;
3309
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00003310 p = (unsigned char *)PyBytes_AS_STRING(v);
Walter Dörwald41980ca2007-08-16 21:55:45 +00003311 if (byteorder == 0)
Benjamin Peterson29060642009-01-31 22:14:21 +00003312 STORECHAR(0xFEFF);
Walter Dörwald41980ca2007-08-16 21:55:45 +00003313 if (size == 0)
Guido van Rossum98297ee2007-11-06 21:34:58 +00003314 goto done;
Walter Dörwald41980ca2007-08-16 21:55:45 +00003315
3316 if (byteorder == -1) {
3317 /* force LE */
3318 iorder[0] = 0;
3319 iorder[1] = 1;
3320 iorder[2] = 2;
3321 iorder[3] = 3;
3322 }
3323 else if (byteorder == 1) {
3324 /* force BE */
3325 iorder[0] = 3;
3326 iorder[1] = 2;
3327 iorder[2] = 1;
3328 iorder[3] = 0;
3329 }
3330
3331 while (size-- > 0) {
Benjamin Peterson29060642009-01-31 22:14:21 +00003332 Py_UCS4 ch = *s++;
Walter Dörwald41980ca2007-08-16 21:55:45 +00003333#ifndef Py_UNICODE_WIDE
Benjamin Peterson29060642009-01-31 22:14:21 +00003334 if (0xD800 <= ch && ch <= 0xDBFF && size > 0) {
3335 Py_UCS4 ch2 = *s;
3336 if (0xDC00 <= ch2 && ch2 <= 0xDFFF) {
3337 ch = (((ch & 0x3FF)<<10) | (ch2 & 0x3FF)) + 0x10000;
3338 s++;
3339 size--;
3340 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00003341 }
Walter Dörwald41980ca2007-08-16 21:55:45 +00003342#endif
3343 STORECHAR(ch);
3344 }
Guido van Rossum98297ee2007-11-06 21:34:58 +00003345
3346 done:
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00003347 return v;
Walter Dörwald41980ca2007-08-16 21:55:45 +00003348#undef STORECHAR
3349}
3350
3351PyObject *PyUnicode_AsUTF32String(PyObject *unicode)
3352{
3353 if (!PyUnicode_Check(unicode)) {
3354 PyErr_BadArgument();
3355 return NULL;
3356 }
3357 return PyUnicode_EncodeUTF32(PyUnicode_AS_UNICODE(unicode),
Benjamin Peterson29060642009-01-31 22:14:21 +00003358 PyUnicode_GET_SIZE(unicode),
3359 NULL,
3360 0);
Walter Dörwald41980ca2007-08-16 21:55:45 +00003361}
3362
Guido van Rossumd57fd912000-03-10 22:53:23 +00003363/* --- UTF-16 Codec ------------------------------------------------------- */
3364
Tim Peters772747b2001-08-09 22:21:55 +00003365PyObject *
3366PyUnicode_DecodeUTF16(const char *s,
Benjamin Peterson29060642009-01-31 22:14:21 +00003367 Py_ssize_t size,
3368 const char *errors,
3369 int *byteorder)
Guido van Rossumd57fd912000-03-10 22:53:23 +00003370{
Walter Dörwald69652032004-09-07 20:24:22 +00003371 return PyUnicode_DecodeUTF16Stateful(s, size, errors, byteorder, NULL);
3372}
3373
Antoine Pitrouab868312009-01-10 15:40:25 +00003374/* Two masks for fast checking of whether a C 'long' may contain
3375 UTF16-encoded surrogate characters. This is an efficient heuristic,
3376 assuming that non-surrogate characters with a code point >= 0x8000 are
3377 rare in most input.
3378 FAST_CHAR_MASK is used when the input is in native byte ordering,
3379 SWAPPED_FAST_CHAR_MASK when the input is in byteswapped ordering.
Benjamin Peterson29060642009-01-31 22:14:21 +00003380*/
Antoine Pitrouab868312009-01-10 15:40:25 +00003381#if (SIZEOF_LONG == 8)
3382# define FAST_CHAR_MASK 0x8000800080008000L
3383# define SWAPPED_FAST_CHAR_MASK 0x0080008000800080L
3384#elif (SIZEOF_LONG == 4)
3385# define FAST_CHAR_MASK 0x80008000L
3386# define SWAPPED_FAST_CHAR_MASK 0x00800080L
3387#else
3388# error C 'long' size should be either 4 or 8!
3389#endif
3390
Walter Dörwald69652032004-09-07 20:24:22 +00003391PyObject *
3392PyUnicode_DecodeUTF16Stateful(const char *s,
Benjamin Peterson29060642009-01-31 22:14:21 +00003393 Py_ssize_t size,
3394 const char *errors,
3395 int *byteorder,
3396 Py_ssize_t *consumed)
Walter Dörwald69652032004-09-07 20:24:22 +00003397{
Walter Dörwald3aeb6322002-09-02 13:14:32 +00003398 const char *starts = s;
Martin v. Löwis18e16552006-02-15 17:27:45 +00003399 Py_ssize_t startinpos;
3400 Py_ssize_t endinpos;
3401 Py_ssize_t outpos;
Guido van Rossumd57fd912000-03-10 22:53:23 +00003402 PyUnicodeObject *unicode;
3403 Py_UNICODE *p;
Antoine Pitrouab868312009-01-10 15:40:25 +00003404 const unsigned char *q, *e, *aligned_end;
Tim Peters772747b2001-08-09 22:21:55 +00003405 int bo = 0; /* assume native ordering by default */
Antoine Pitrouab868312009-01-10 15:40:25 +00003406 int native_ordering = 0;
Marc-André Lemburg9542f482000-07-17 18:23:13 +00003407 const char *errmsg = "";
Tim Peters772747b2001-08-09 22:21:55 +00003408 /* Offsets from q for retrieving byte pairs in the right order. */
3409#ifdef BYTEORDER_IS_LITTLE_ENDIAN
3410 int ihi = 1, ilo = 0;
3411#else
3412 int ihi = 0, ilo = 1;
3413#endif
Walter Dörwald3aeb6322002-09-02 13:14:32 +00003414 PyObject *errorHandler = NULL;
3415 PyObject *exc = NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00003416
3417 /* Note: size will always be longer than the resulting Unicode
3418 character count */
3419 unicode = _PyUnicode_New(size);
3420 if (!unicode)
3421 return NULL;
3422 if (size == 0)
3423 return (PyObject *)unicode;
3424
3425 /* Unpack UTF-16 encoded data */
3426 p = unicode->str;
Tim Peters772747b2001-08-09 22:21:55 +00003427 q = (unsigned char *)s;
Antoine Pitroub4bbee22012-07-21 00:45:14 +02003428 e = q + size;
Guido van Rossumd57fd912000-03-10 22:53:23 +00003429
3430 if (byteorder)
Tim Peters772747b2001-08-09 22:21:55 +00003431 bo = *byteorder;
Guido van Rossumd57fd912000-03-10 22:53:23 +00003432
Marc-André Lemburg489b56e2001-05-21 20:30:15 +00003433 /* Check for BOM marks (U+FEFF) in the input and adjust current
3434 byte order setting accordingly. In native mode, the leading BOM
3435 mark is skipped, in all other modes, it is copied to the output
3436 stream as-is (giving a ZWNBSP character). */
3437 if (bo == 0) {
Walter Dörwald69652032004-09-07 20:24:22 +00003438 if (size >= 2) {
3439 const Py_UNICODE bom = (q[ihi] << 8) | q[ilo];
Marc-André Lemburg489b56e2001-05-21 20:30:15 +00003440#ifdef BYTEORDER_IS_LITTLE_ENDIAN
Benjamin Peterson29060642009-01-31 22:14:21 +00003441 if (bom == 0xFEFF) {
3442 q += 2;
3443 bo = -1;
3444 }
3445 else if (bom == 0xFFFE) {
3446 q += 2;
3447 bo = 1;
3448 }
Tim Petersced69f82003-09-16 20:30:58 +00003449#else
Benjamin Peterson29060642009-01-31 22:14:21 +00003450 if (bom == 0xFEFF) {
3451 q += 2;
3452 bo = 1;
3453 }
3454 else if (bom == 0xFFFE) {
3455 q += 2;
3456 bo = -1;
3457 }
Marc-André Lemburg489b56e2001-05-21 20:30:15 +00003458#endif
Benjamin Peterson29060642009-01-31 22:14:21 +00003459 }
Marc-André Lemburg489b56e2001-05-21 20:30:15 +00003460 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00003461
Tim Peters772747b2001-08-09 22:21:55 +00003462 if (bo == -1) {
3463 /* force LE */
3464 ihi = 1;
3465 ilo = 0;
3466 }
3467 else if (bo == 1) {
3468 /* force BE */
3469 ihi = 0;
3470 ilo = 1;
3471 }
Antoine Pitrouab868312009-01-10 15:40:25 +00003472#ifdef BYTEORDER_IS_LITTLE_ENDIAN
3473 native_ordering = ilo < ihi;
3474#else
3475 native_ordering = ilo > ihi;
3476#endif
Tim Peters772747b2001-08-09 22:21:55 +00003477
Antoine Pitrouab868312009-01-10 15:40:25 +00003478 aligned_end = (const unsigned char *) ((size_t) e & ~LONG_PTR_MASK);
Antoine Pitroub4bbee22012-07-21 00:45:14 +02003479 while (1) {
Benjamin Peterson29060642009-01-31 22:14:21 +00003480 Py_UNICODE ch;
Antoine Pitroub4bbee22012-07-21 00:45:14 +02003481 if (e - q < 2) {
3482 /* remaining byte at the end? (size should be even) */
3483 if (q == e || consumed)
3484 break;
3485 errmsg = "truncated data";
3486 startinpos = ((const char *)q) - starts;
3487 endinpos = ((const char *)e) - starts;
3488 outpos = p - PyUnicode_AS_UNICODE(unicode);
3489 goto utf16Error;
3490 /* The remaining input chars are ignored if the callback
3491 chooses to skip the input */
3492 }
Antoine Pitrouab868312009-01-10 15:40:25 +00003493 /* First check for possible aligned read of a C 'long'. Unaligned
3494 reads are more expensive, better to defer to another iteration. */
3495 if (!((size_t) q & LONG_PTR_MASK)) {
3496 /* Fast path for runs of non-surrogate chars. */
3497 register const unsigned char *_q = q;
3498 Py_UNICODE *_p = p;
3499 if (native_ordering) {
3500 /* Native ordering is simple: as long as the input cannot
3501 possibly contain a surrogate char, do an unrolled copy
3502 of several 16-bit code points to the target object.
3503 The non-surrogate check is done on several input bytes
3504 at a time (as many as a C 'long' can contain). */
3505 while (_q < aligned_end) {
3506 unsigned long data = * (unsigned long *) _q;
3507 if (data & FAST_CHAR_MASK)
3508 break;
3509 _p[0] = ((unsigned short *) _q)[0];
3510 _p[1] = ((unsigned short *) _q)[1];
3511#if (SIZEOF_LONG == 8)
3512 _p[2] = ((unsigned short *) _q)[2];
3513 _p[3] = ((unsigned short *) _q)[3];
3514#endif
3515 _q += SIZEOF_LONG;
3516 _p += SIZEOF_LONG / 2;
3517 }
3518 }
3519 else {
3520 /* Byteswapped ordering is similar, but we must decompose
3521 the copy bytewise, and take care of zero'ing out the
3522 upper bytes if the target object is in 32-bit units
3523 (that is, in UCS-4 builds). */
3524 while (_q < aligned_end) {
3525 unsigned long data = * (unsigned long *) _q;
3526 if (data & SWAPPED_FAST_CHAR_MASK)
3527 break;
3528 /* Zero upper bytes in UCS-4 builds */
3529#if (Py_UNICODE_SIZE > 2)
3530 _p[0] = 0;
3531 _p[1] = 0;
3532#if (SIZEOF_LONG == 8)
3533 _p[2] = 0;
3534 _p[3] = 0;
3535#endif
3536#endif
Antoine Pitroud6e8de12009-01-11 23:56:55 +00003537 /* Issue #4916; UCS-4 builds on big endian machines must
3538 fill the two last bytes of each 4-byte unit. */
3539#if (!defined(BYTEORDER_IS_LITTLE_ENDIAN) && Py_UNICODE_SIZE > 2)
3540# define OFF 2
3541#else
3542# define OFF 0
Antoine Pitrouab868312009-01-10 15:40:25 +00003543#endif
Antoine Pitroud6e8de12009-01-11 23:56:55 +00003544 ((unsigned char *) _p)[OFF + 1] = _q[0];
3545 ((unsigned char *) _p)[OFF + 0] = _q[1];
3546 ((unsigned char *) _p)[OFF + 1 + Py_UNICODE_SIZE] = _q[2];
3547 ((unsigned char *) _p)[OFF + 0 + Py_UNICODE_SIZE] = _q[3];
3548#if (SIZEOF_LONG == 8)
3549 ((unsigned char *) _p)[OFF + 1 + 2 * Py_UNICODE_SIZE] = _q[4];
3550 ((unsigned char *) _p)[OFF + 0 + 2 * Py_UNICODE_SIZE] = _q[5];
3551 ((unsigned char *) _p)[OFF + 1 + 3 * Py_UNICODE_SIZE] = _q[6];
3552 ((unsigned char *) _p)[OFF + 0 + 3 * Py_UNICODE_SIZE] = _q[7];
3553#endif
3554#undef OFF
Antoine Pitrouab868312009-01-10 15:40:25 +00003555 _q += SIZEOF_LONG;
3556 _p += SIZEOF_LONG / 2;
3557 }
3558 }
3559 p = _p;
3560 q = _q;
Antoine Pitroub4bbee22012-07-21 00:45:14 +02003561 if (e - q < 2)
3562 continue;
Antoine Pitrouab868312009-01-10 15:40:25 +00003563 }
Benjamin Peterson29060642009-01-31 22:14:21 +00003564 ch = (q[ihi] << 8) | q[ilo];
Walter Dörwald3aeb6322002-09-02 13:14:32 +00003565
Benjamin Peterson14339b62009-01-31 16:36:08 +00003566 q += 2;
Benjamin Peterson29060642009-01-31 22:14:21 +00003567
3568 if (ch < 0xD800 || ch > 0xDFFF) {
3569 *p++ = ch;
3570 continue;
3571 }
3572
3573 /* UTF-16 code pair: */
Antoine Pitroub4bbee22012-07-21 00:45:14 +02003574 if (e - q < 2) {
Benjamin Peterson29060642009-01-31 22:14:21 +00003575 errmsg = "unexpected end of data";
3576 startinpos = (((const char *)q) - 2) - starts;
Antoine Pitroub4bbee22012-07-21 00:45:14 +02003577 endinpos = ((const char *)e) - starts;
Benjamin Peterson29060642009-01-31 22:14:21 +00003578 goto utf16Error;
3579 }
3580 if (0xD800 <= ch && ch <= 0xDBFF) {
3581 Py_UNICODE ch2 = (q[ihi] << 8) | q[ilo];
3582 q += 2;
3583 if (0xDC00 <= ch2 && ch2 <= 0xDFFF) {
Fredrik Lundh8f455852001-06-27 18:59:43 +00003584#ifndef Py_UNICODE_WIDE
Benjamin Peterson29060642009-01-31 22:14:21 +00003585 *p++ = ch;
3586 *p++ = ch2;
Martin v. Löwis0ba70cc2001-06-26 22:22:37 +00003587#else
Benjamin Peterson29060642009-01-31 22:14:21 +00003588 *p++ = (((ch & 0x3FF)<<10) | (ch2 & 0x3FF)) + 0x10000;
Martin v. Löwis0ba70cc2001-06-26 22:22:37 +00003589#endif
Benjamin Peterson29060642009-01-31 22:14:21 +00003590 continue;
3591 }
3592 else {
Martin v. Löwis0ba70cc2001-06-26 22:22:37 +00003593 errmsg = "illegal UTF-16 surrogate";
Benjamin Peterson29060642009-01-31 22:14:21 +00003594 startinpos = (((const char *)q)-4)-starts;
3595 endinpos = startinpos+2;
3596 goto utf16Error;
3597 }
3598
Benjamin Peterson14339b62009-01-31 16:36:08 +00003599 }
Benjamin Peterson29060642009-01-31 22:14:21 +00003600 errmsg = "illegal encoding";
3601 startinpos = (((const char *)q)-2)-starts;
3602 endinpos = startinpos+2;
3603 /* Fall through to report the error */
Martin v. Löwis0ba70cc2001-06-26 22:22:37 +00003604
Benjamin Peterson29060642009-01-31 22:14:21 +00003605 utf16Error:
3606 outpos = p - PyUnicode_AS_UNICODE(unicode);
3607 if (unicode_decode_call_errorhandler(
Antoine Pitrouab868312009-01-10 15:40:25 +00003608 errors,
3609 &errorHandler,
3610 "utf16", errmsg,
3611 &starts,
3612 (const char **)&e,
3613 &startinpos,
3614 &endinpos,
3615 &exc,
3616 (const char **)&q,
3617 &unicode,
3618 &outpos,
3619 &p))
Benjamin Peterson29060642009-01-31 22:14:21 +00003620 goto onError;
Antoine Pitroub4bbee22012-07-21 00:45:14 +02003621 /* Update data because unicode_decode_call_errorhandler might have
3622 changed the input object. */
3623 aligned_end = (const unsigned char *) ((size_t) e & ~LONG_PTR_MASK);
Antoine Pitrouab868312009-01-10 15:40:25 +00003624 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00003625
3626 if (byteorder)
3627 *byteorder = bo;
3628
Walter Dörwald69652032004-09-07 20:24:22 +00003629 if (consumed)
Benjamin Peterson29060642009-01-31 22:14:21 +00003630 *consumed = (const char *)q-starts;
Walter Dörwald69652032004-09-07 20:24:22 +00003631
Guido van Rossumd57fd912000-03-10 22:53:23 +00003632 /* Adjust length */
Jeremy Hyltondeb2dc62003-09-16 03:41:45 +00003633 if (_PyUnicode_Resize(&unicode, p - unicode->str) < 0)
Guido van Rossumd57fd912000-03-10 22:53:23 +00003634 goto onError;
3635
Walter Dörwald3aeb6322002-09-02 13:14:32 +00003636 Py_XDECREF(errorHandler);
3637 Py_XDECREF(exc);
Guido van Rossumd57fd912000-03-10 22:53:23 +00003638 return (PyObject *)unicode;
3639
Benjamin Peterson29060642009-01-31 22:14:21 +00003640 onError:
Guido van Rossumd57fd912000-03-10 22:53:23 +00003641 Py_DECREF(unicode);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00003642 Py_XDECREF(errorHandler);
3643 Py_XDECREF(exc);
Guido van Rossumd57fd912000-03-10 22:53:23 +00003644 return NULL;
3645}
3646
Antoine Pitrouab868312009-01-10 15:40:25 +00003647#undef FAST_CHAR_MASK
3648#undef SWAPPED_FAST_CHAR_MASK
3649
Tim Peters772747b2001-08-09 22:21:55 +00003650PyObject *
3651PyUnicode_EncodeUTF16(const Py_UNICODE *s,
Benjamin Peterson29060642009-01-31 22:14:21 +00003652 Py_ssize_t size,
3653 const char *errors,
3654 int byteorder)
Guido van Rossumd57fd912000-03-10 22:53:23 +00003655{
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00003656 PyObject *v;
Tim Peters772747b2001-08-09 22:21:55 +00003657 unsigned char *p;
Neal Norwitz3ce5d922008-08-24 07:08:55 +00003658 Py_ssize_t nsize, bytesize;
Hye-Shik Chang4a264fb2003-12-19 01:59:56 +00003659#ifdef Py_UNICODE_WIDE
Neal Norwitz3ce5d922008-08-24 07:08:55 +00003660 Py_ssize_t i, pairs;
Hye-Shik Chang4a264fb2003-12-19 01:59:56 +00003661#else
3662 const int pairs = 0;
3663#endif
Tim Peters772747b2001-08-09 22:21:55 +00003664 /* Offsets from p for storing byte pairs in the right order. */
3665#ifdef BYTEORDER_IS_LITTLE_ENDIAN
3666 int ihi = 1, ilo = 0;
3667#else
3668 int ihi = 0, ilo = 1;
3669#endif
3670
Benjamin Peterson29060642009-01-31 22:14:21 +00003671#define STORECHAR(CH) \
3672 do { \
3673 p[ihi] = ((CH) >> 8) & 0xff; \
3674 p[ilo] = (CH) & 0xff; \
3675 p += 2; \
Tim Peters772747b2001-08-09 22:21:55 +00003676 } while(0)
Guido van Rossumd57fd912000-03-10 22:53:23 +00003677
Hye-Shik Chang4a264fb2003-12-19 01:59:56 +00003678#ifdef Py_UNICODE_WIDE
Martin v. Löwis0ba70cc2001-06-26 22:22:37 +00003679 for (i = pairs = 0; i < size; i++)
Benjamin Peterson29060642009-01-31 22:14:21 +00003680 if (s[i] >= 0x10000)
3681 pairs++;
Hye-Shik Chang4a264fb2003-12-19 01:59:56 +00003682#endif
Neal Norwitz3ce5d922008-08-24 07:08:55 +00003683 /* 2 * (size + pairs + (byteorder == 0)) */
3684 if (size > PY_SSIZE_T_MAX ||
3685 size > PY_SSIZE_T_MAX - pairs - (byteorder == 0))
Benjamin Peterson29060642009-01-31 22:14:21 +00003686 return PyErr_NoMemory();
Neal Norwitz3ce5d922008-08-24 07:08:55 +00003687 nsize = size + pairs + (byteorder == 0);
3688 bytesize = nsize * 2;
3689 if (bytesize / 2 != nsize)
Benjamin Peterson29060642009-01-31 22:14:21 +00003690 return PyErr_NoMemory();
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00003691 v = PyBytes_FromStringAndSize(NULL, bytesize);
Guido van Rossumd57fd912000-03-10 22:53:23 +00003692 if (v == NULL)
3693 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00003694
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00003695 p = (unsigned char *)PyBytes_AS_STRING(v);
Guido van Rossumd57fd912000-03-10 22:53:23 +00003696 if (byteorder == 0)
Benjamin Peterson29060642009-01-31 22:14:21 +00003697 STORECHAR(0xFEFF);
Marc-André Lemburg063e0cb2000-07-07 11:27:45 +00003698 if (size == 0)
Guido van Rossum98297ee2007-11-06 21:34:58 +00003699 goto done;
Tim Peters772747b2001-08-09 22:21:55 +00003700
3701 if (byteorder == -1) {
3702 /* force LE */
3703 ihi = 1;
3704 ilo = 0;
3705 }
3706 else if (byteorder == 1) {
3707 /* force BE */
3708 ihi = 0;
3709 ilo = 1;
3710 }
3711
Martin v. Löwis0ba70cc2001-06-26 22:22:37 +00003712 while (size-- > 0) {
Benjamin Peterson29060642009-01-31 22:14:21 +00003713 Py_UNICODE ch = *s++;
3714 Py_UNICODE ch2 = 0;
Hye-Shik Chang4a264fb2003-12-19 01:59:56 +00003715#ifdef Py_UNICODE_WIDE
Benjamin Peterson29060642009-01-31 22:14:21 +00003716 if (ch >= 0x10000) {
3717 ch2 = 0xDC00 | ((ch-0x10000) & 0x3FF);
3718 ch = 0xD800 | ((ch-0x10000) >> 10);
3719 }
Hye-Shik Chang4a264fb2003-12-19 01:59:56 +00003720#endif
Tim Peters772747b2001-08-09 22:21:55 +00003721 STORECHAR(ch);
3722 if (ch2)
3723 STORECHAR(ch2);
Martin v. Löwis0ba70cc2001-06-26 22:22:37 +00003724 }
Guido van Rossum98297ee2007-11-06 21:34:58 +00003725
3726 done:
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00003727 return v;
Tim Peters772747b2001-08-09 22:21:55 +00003728#undef STORECHAR
Guido van Rossumd57fd912000-03-10 22:53:23 +00003729}
3730
3731PyObject *PyUnicode_AsUTF16String(PyObject *unicode)
3732{
3733 if (!PyUnicode_Check(unicode)) {
3734 PyErr_BadArgument();
3735 return NULL;
3736 }
3737 return PyUnicode_EncodeUTF16(PyUnicode_AS_UNICODE(unicode),
Benjamin Peterson29060642009-01-31 22:14:21 +00003738 PyUnicode_GET_SIZE(unicode),
3739 NULL,
3740 0);
Guido van Rossumd57fd912000-03-10 22:53:23 +00003741}
3742
3743/* --- Unicode Escape Codec ----------------------------------------------- */
3744
Fredrik Lundh06d12682001-01-24 07:59:11 +00003745static _PyUnicode_Name_CAPI *ucnhash_CAPI = NULL;
Marc-André Lemburg0f774e32000-06-28 16:43:35 +00003746
Guido van Rossumd57fd912000-03-10 22:53:23 +00003747PyObject *PyUnicode_DecodeUnicodeEscape(const char *s,
Benjamin Peterson29060642009-01-31 22:14:21 +00003748 Py_ssize_t size,
3749 const char *errors)
Guido van Rossumd57fd912000-03-10 22:53:23 +00003750{
Walter Dörwald3aeb6322002-09-02 13:14:32 +00003751 const char *starts = s;
Martin v. Löwis18e16552006-02-15 17:27:45 +00003752 Py_ssize_t startinpos;
3753 Py_ssize_t endinpos;
3754 Py_ssize_t outpos;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00003755 int i;
Guido van Rossumd57fd912000-03-10 22:53:23 +00003756 PyUnicodeObject *v;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00003757 Py_UNICODE *p;
Guido van Rossumd57fd912000-03-10 22:53:23 +00003758 const char *end;
Fredrik Lundhccc74732001-02-18 22:13:49 +00003759 char* message;
3760 Py_UCS4 chr = 0xffffffff; /* in case 'getcode' messes up */
Walter Dörwald3aeb6322002-09-02 13:14:32 +00003761 PyObject *errorHandler = NULL;
3762 PyObject *exc = NULL;
Fredrik Lundhccc74732001-02-18 22:13:49 +00003763
Guido van Rossumd57fd912000-03-10 22:53:23 +00003764 /* Escaped strings will always be longer than the resulting
3765 Unicode string, so we start with size here and then reduce the
Walter Dörwald3aeb6322002-09-02 13:14:32 +00003766 length after conversion to the true value.
3767 (but if the error callback returns a long replacement string
3768 we'll have to allocate more space) */
Guido van Rossumd57fd912000-03-10 22:53:23 +00003769 v = _PyUnicode_New(size);
3770 if (v == NULL)
3771 goto onError;
3772 if (size == 0)
3773 return (PyObject *)v;
Fredrik Lundhccc74732001-02-18 22:13:49 +00003774
Walter Dörwald3aeb6322002-09-02 13:14:32 +00003775 p = PyUnicode_AS_UNICODE(v);
Guido van Rossumd57fd912000-03-10 22:53:23 +00003776 end = s + size;
Fredrik Lundhccc74732001-02-18 22:13:49 +00003777
Guido van Rossumd57fd912000-03-10 22:53:23 +00003778 while (s < end) {
3779 unsigned char c;
Marc-André Lemburg063e0cb2000-07-07 11:27:45 +00003780 Py_UNICODE x;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00003781 int digits;
Guido van Rossumd57fd912000-03-10 22:53:23 +00003782
3783 /* Non-escape characters are interpreted as Unicode ordinals */
3784 if (*s != '\\') {
Fredrik Lundhccc74732001-02-18 22:13:49 +00003785 *p++ = (unsigned char) *s++;
Guido van Rossumd57fd912000-03-10 22:53:23 +00003786 continue;
3787 }
3788
Walter Dörwald3aeb6322002-09-02 13:14:32 +00003789 startinpos = s-starts;
Guido van Rossumd57fd912000-03-10 22:53:23 +00003790 /* \ - Escapes */
3791 s++;
Guido van Rossum8ce8a782007-11-01 19:42:39 +00003792 c = *s++;
3793 if (s > end)
3794 c = '\0'; /* Invalid after \ */
3795 switch (c) {
Guido van Rossumd57fd912000-03-10 22:53:23 +00003796
Benjamin Peterson29060642009-01-31 22:14:21 +00003797 /* \x escapes */
Guido van Rossumd57fd912000-03-10 22:53:23 +00003798 case '\n': break;
3799 case '\\': *p++ = '\\'; break;
3800 case '\'': *p++ = '\''; break;
3801 case '\"': *p++ = '\"'; break;
3802 case 'b': *p++ = '\b'; break;
3803 case 'f': *p++ = '\014'; break; /* FF */
3804 case 't': *p++ = '\t'; break;
3805 case 'n': *p++ = '\n'; break;
3806 case 'r': *p++ = '\r'; break;
3807 case 'v': *p++ = '\013'; break; /* VT */
3808 case 'a': *p++ = '\007'; break; /* BEL, not classic C */
3809
Benjamin Peterson29060642009-01-31 22:14:21 +00003810 /* \OOO (octal) escapes */
Guido van Rossumd57fd912000-03-10 22:53:23 +00003811 case '0': case '1': case '2': case '3':
3812 case '4': case '5': case '6': case '7':
Guido van Rossum0e4f6572000-05-01 21:27:20 +00003813 x = s[-1] - '0';
Guido van Rossum8ce8a782007-11-01 19:42:39 +00003814 if (s < end && '0' <= *s && *s <= '7') {
Guido van Rossum0e4f6572000-05-01 21:27:20 +00003815 x = (x<<3) + *s++ - '0';
Guido van Rossum8ce8a782007-11-01 19:42:39 +00003816 if (s < end && '0' <= *s && *s <= '7')
Guido van Rossum0e4f6572000-05-01 21:27:20 +00003817 x = (x<<3) + *s++ - '0';
Guido van Rossumd57fd912000-03-10 22:53:23 +00003818 }
Guido van Rossum0e4f6572000-05-01 21:27:20 +00003819 *p++ = x;
Guido van Rossumd57fd912000-03-10 22:53:23 +00003820 break;
3821
Benjamin Peterson29060642009-01-31 22:14:21 +00003822 /* hex escapes */
3823 /* \xXX */
Guido van Rossumd57fd912000-03-10 22:53:23 +00003824 case 'x':
Fredrik Lundhccc74732001-02-18 22:13:49 +00003825 digits = 2;
3826 message = "truncated \\xXX escape";
3827 goto hexescape;
Guido van Rossumd57fd912000-03-10 22:53:23 +00003828
Benjamin Peterson29060642009-01-31 22:14:21 +00003829 /* \uXXXX */
Guido van Rossumd57fd912000-03-10 22:53:23 +00003830 case 'u':
Fredrik Lundhccc74732001-02-18 22:13:49 +00003831 digits = 4;
3832 message = "truncated \\uXXXX escape";
3833 goto hexescape;
Guido van Rossumd57fd912000-03-10 22:53:23 +00003834
Benjamin Peterson29060642009-01-31 22:14:21 +00003835 /* \UXXXXXXXX */
Fredrik Lundhdf846752000-09-03 11:29:49 +00003836 case 'U':
Fredrik Lundhccc74732001-02-18 22:13:49 +00003837 digits = 8;
3838 message = "truncated \\UXXXXXXXX escape";
3839 hexescape:
3840 chr = 0;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00003841 outpos = p-PyUnicode_AS_UNICODE(v);
3842 if (s+digits>end) {
3843 endinpos = size;
3844 if (unicode_decode_call_errorhandler(
Benjamin Peterson29060642009-01-31 22:14:21 +00003845 errors, &errorHandler,
3846 "unicodeescape", "end of string in escape sequence",
3847 &starts, &end, &startinpos, &endinpos, &exc, &s,
3848 &v, &outpos, &p))
Walter Dörwald3aeb6322002-09-02 13:14:32 +00003849 goto onError;
3850 goto nextByte;
3851 }
3852 for (i = 0; i < digits; ++i) {
Fredrik Lundhccc74732001-02-18 22:13:49 +00003853 c = (unsigned char) s[i];
David Malcolm96960882010-11-05 17:23:41 +00003854 if (!Py_ISXDIGIT(c)) {
Walter Dörwald3aeb6322002-09-02 13:14:32 +00003855 endinpos = (s+i+1)-starts;
3856 if (unicode_decode_call_errorhandler(
Benjamin Peterson29060642009-01-31 22:14:21 +00003857 errors, &errorHandler,
3858 "unicodeescape", message,
3859 &starts, &end, &startinpos, &endinpos, &exc, &s,
3860 &v, &outpos, &p))
Fredrik Lundhdf846752000-09-03 11:29:49 +00003861 goto onError;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00003862 goto nextByte;
Fredrik Lundhdf846752000-09-03 11:29:49 +00003863 }
3864 chr = (chr<<4) & ~0xF;
3865 if (c >= '0' && c <= '9')
3866 chr += c - '0';
3867 else if (c >= 'a' && c <= 'f')
3868 chr += 10 + c - 'a';
3869 else
3870 chr += 10 + c - 'A';
3871 }
3872 s += i;
Jeremy Hylton504de6b2003-10-06 05:08:26 +00003873 if (chr == 0xffffffff && PyErr_Occurred())
Walter Dörwald3aeb6322002-09-02 13:14:32 +00003874 /* _decoding_error will have already written into the
3875 target buffer. */
3876 break;
Fredrik Lundhccc74732001-02-18 22:13:49 +00003877 store:
Fredrik Lundhdf846752000-09-03 11:29:49 +00003878 /* when we get here, chr is a 32-bit unicode character */
3879 if (chr <= 0xffff)
3880 /* UCS-2 character */
3881 *p++ = (Py_UNICODE) chr;
3882 else if (chr <= 0x10ffff) {
Marc-André Lemburg6c6bfb72001-07-20 17:39:11 +00003883 /* UCS-4 character. Either store directly, or as
Walter Dörwald8c077222002-03-25 11:16:18 +00003884 surrogate pair. */
Fredrik Lundh8f455852001-06-27 18:59:43 +00003885#ifdef Py_UNICODE_WIDE
Martin v. Löwis0ba70cc2001-06-26 22:22:37 +00003886 *p++ = chr;
3887#else
Fredrik Lundhdf846752000-09-03 11:29:49 +00003888 chr -= 0x10000L;
3889 *p++ = 0xD800 + (Py_UNICODE) (chr >> 10);
Fredrik Lundh45714e92001-06-26 16:39:36 +00003890 *p++ = 0xDC00 + (Py_UNICODE) (chr & 0x03FF);
Martin v. Löwis0ba70cc2001-06-26 22:22:37 +00003891#endif
Fredrik Lundhdf846752000-09-03 11:29:49 +00003892 } else {
Walter Dörwald3aeb6322002-09-02 13:14:32 +00003893 endinpos = s-starts;
3894 outpos = p-PyUnicode_AS_UNICODE(v);
3895 if (unicode_decode_call_errorhandler(
Benjamin Peterson29060642009-01-31 22:14:21 +00003896 errors, &errorHandler,
3897 "unicodeescape", "illegal Unicode character",
3898 &starts, &end, &startinpos, &endinpos, &exc, &s,
3899 &v, &outpos, &p))
Fredrik Lundhdf846752000-09-03 11:29:49 +00003900 goto onError;
3901 }
Fredrik Lundhccc74732001-02-18 22:13:49 +00003902 break;
3903
Benjamin Peterson29060642009-01-31 22:14:21 +00003904 /* \N{name} */
Fredrik Lundhccc74732001-02-18 22:13:49 +00003905 case 'N':
3906 message = "malformed \\N character escape";
3907 if (ucnhash_CAPI == NULL) {
3908 /* load the unicode data module */
Benjamin Petersonb173f782009-05-05 22:31:58 +00003909 ucnhash_CAPI = (_PyUnicode_Name_CAPI *)PyCapsule_Import(PyUnicodeData_CAPSULE_NAME, 1);
Fredrik Lundhccc74732001-02-18 22:13:49 +00003910 if (ucnhash_CAPI == NULL)
3911 goto ucnhashError;
3912 }
3913 if (*s == '{') {
3914 const char *start = s+1;
3915 /* look for the closing brace */
3916 while (*s != '}' && s < end)
3917 s++;
3918 if (s > start && s < end && *s == '}') {
3919 /* found a name. look it up in the unicode database */
3920 message = "unknown Unicode character name";
3921 s++;
Martin v. Löwis480f1bb2006-03-09 23:38:20 +00003922 if (ucnhash_CAPI->getcode(NULL, start, (int)(s-start-1), &chr))
Fredrik Lundhccc74732001-02-18 22:13:49 +00003923 goto store;
3924 }
3925 }
Walter Dörwald3aeb6322002-09-02 13:14:32 +00003926 endinpos = s-starts;
3927 outpos = p-PyUnicode_AS_UNICODE(v);
3928 if (unicode_decode_call_errorhandler(
Benjamin Peterson29060642009-01-31 22:14:21 +00003929 errors, &errorHandler,
3930 "unicodeescape", message,
3931 &starts, &end, &startinpos, &endinpos, &exc, &s,
3932 &v, &outpos, &p))
Fredrik Lundhccc74732001-02-18 22:13:49 +00003933 goto onError;
Fredrik Lundhccc74732001-02-18 22:13:49 +00003934 break;
3935
3936 default:
Walter Dörwald8c077222002-03-25 11:16:18 +00003937 if (s > end) {
Walter Dörwald3aeb6322002-09-02 13:14:32 +00003938 message = "\\ at end of string";
3939 s--;
3940 endinpos = s-starts;
3941 outpos = p-PyUnicode_AS_UNICODE(v);
3942 if (unicode_decode_call_errorhandler(
Benjamin Peterson29060642009-01-31 22:14:21 +00003943 errors, &errorHandler,
3944 "unicodeescape", message,
3945 &starts, &end, &startinpos, &endinpos, &exc, &s,
3946 &v, &outpos, &p))
Walter Dörwald8c077222002-03-25 11:16:18 +00003947 goto onError;
3948 }
3949 else {
3950 *p++ = '\\';
3951 *p++ = (unsigned char)s[-1];
3952 }
Fredrik Lundhccc74732001-02-18 22:13:49 +00003953 break;
Guido van Rossumd57fd912000-03-10 22:53:23 +00003954 }
Benjamin Peterson29060642009-01-31 22:14:21 +00003955 nextByte:
Walter Dörwald3aeb6322002-09-02 13:14:32 +00003956 ;
Guido van Rossumd57fd912000-03-10 22:53:23 +00003957 }
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00003958 if (_PyUnicode_Resize(&v, p - PyUnicode_AS_UNICODE(v)) < 0)
Walter Dörwald3aeb6322002-09-02 13:14:32 +00003959 goto onError;
Walter Dörwaldd4ade082003-08-15 15:00:26 +00003960 Py_XDECREF(errorHandler);
3961 Py_XDECREF(exc);
Guido van Rossumd57fd912000-03-10 22:53:23 +00003962 return (PyObject *)v;
Walter Dörwald8c077222002-03-25 11:16:18 +00003963
Benjamin Peterson29060642009-01-31 22:14:21 +00003964 ucnhashError:
Fredrik Lundh06d12682001-01-24 07:59:11 +00003965 PyErr_SetString(
3966 PyExc_UnicodeError,
3967 "\\N escapes not supported (can't load unicodedata module)"
3968 );
Hye-Shik Chang4af5c8c2006-03-07 15:39:21 +00003969 Py_XDECREF(v);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00003970 Py_XDECREF(errorHandler);
3971 Py_XDECREF(exc);
Fredrik Lundhf6056062001-01-20 11:15:25 +00003972 return NULL;
3973
Benjamin Peterson29060642009-01-31 22:14:21 +00003974 onError:
Guido van Rossumd57fd912000-03-10 22:53:23 +00003975 Py_XDECREF(v);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00003976 Py_XDECREF(errorHandler);
3977 Py_XDECREF(exc);
Guido van Rossumd57fd912000-03-10 22:53:23 +00003978 return NULL;
3979}
3980
3981/* Return a Unicode-Escape string version of the Unicode object.
3982
3983 If quotes is true, the string is enclosed in u"" or u'' quotes as
3984 appropriate.
3985
3986*/
3987
Thomas Wouters477c8d52006-05-27 19:21:47 +00003988Py_LOCAL_INLINE(const Py_UNICODE *) findchar(const Py_UNICODE *s,
Benjamin Peterson29060642009-01-31 22:14:21 +00003989 Py_ssize_t size,
3990 Py_UNICODE ch)
Thomas Wouters477c8d52006-05-27 19:21:47 +00003991{
3992 /* like wcschr, but doesn't stop at NULL characters */
3993
3994 while (size-- > 0) {
3995 if (*s == ch)
3996 return s;
3997 s++;
3998 }
3999
4000 return NULL;
4001}
Barry Warsaw51ac5802000-03-20 16:36:48 +00004002
Walter Dörwald79e913e2007-05-12 11:08:06 +00004003static const char *hexdigits = "0123456789abcdef";
4004
4005PyObject *PyUnicode_EncodeUnicodeEscape(const Py_UNICODE *s,
Benjamin Peterson29060642009-01-31 22:14:21 +00004006 Py_ssize_t size)
Guido van Rossumd57fd912000-03-10 22:53:23 +00004007{
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00004008 PyObject *repr;
Guido van Rossumd57fd912000-03-10 22:53:23 +00004009 char *p;
Guido van Rossumd57fd912000-03-10 22:53:23 +00004010
Neal Norwitz3ce5d922008-08-24 07:08:55 +00004011#ifdef Py_UNICODE_WIDE
4012 const Py_ssize_t expandsize = 10;
4013#else
4014 const Py_ssize_t expandsize = 6;
4015#endif
4016
Thomas Wouters89f507f2006-12-13 04:49:30 +00004017 /* XXX(nnorwitz): rather than over-allocating, it would be
4018 better to choose a different scheme. Perhaps scan the
4019 first N-chars of the string and allocate based on that size.
4020 */
4021 /* Initial allocation is based on the longest-possible unichr
4022 escape.
4023
4024 In wide (UTF-32) builds '\U00xxxxxx' is 10 chars per source
4025 unichr, so in this case it's the longest unichr escape. In
4026 narrow (UTF-16) builds this is five chars per source unichr
4027 since there are two unichrs in the surrogate pair, so in narrow
4028 (UTF-16) builds it's not the longest unichr escape.
4029
4030 In wide or narrow builds '\uxxxx' is 6 chars per source unichr,
4031 so in the narrow (UTF-16) build case it's the longest unichr
4032 escape.
4033 */
4034
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00004035 if (size == 0)
4036 return PyBytes_FromStringAndSize(NULL, 0);
4037
Neal Norwitz3ce5d922008-08-24 07:08:55 +00004038 if (size > (PY_SSIZE_T_MAX - 2 - 1) / expandsize)
Benjamin Peterson29060642009-01-31 22:14:21 +00004039 return PyErr_NoMemory();
Neal Norwitz3ce5d922008-08-24 07:08:55 +00004040
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00004041 repr = PyBytes_FromStringAndSize(NULL,
Benjamin Peterson29060642009-01-31 22:14:21 +00004042 2
4043 + expandsize*size
4044 + 1);
Guido van Rossumd57fd912000-03-10 22:53:23 +00004045 if (repr == NULL)
4046 return NULL;
4047
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00004048 p = PyBytes_AS_STRING(repr);
Guido van Rossumd57fd912000-03-10 22:53:23 +00004049
Guido van Rossumd57fd912000-03-10 22:53:23 +00004050 while (size-- > 0) {
4051 Py_UNICODE ch = *s++;
Marc-André Lemburg80d1dd52001-07-25 16:05:59 +00004052
Walter Dörwald79e913e2007-05-12 11:08:06 +00004053 /* Escape backslashes */
4054 if (ch == '\\') {
Guido van Rossumd57fd912000-03-10 22:53:23 +00004055 *p++ = '\\';
4056 *p++ = (char) ch;
Walter Dörwald79e913e2007-05-12 11:08:06 +00004057 continue;
Tim Petersced69f82003-09-16 20:30:58 +00004058 }
Marc-André Lemburg80d1dd52001-07-25 16:05:59 +00004059
Guido van Rossum0d42e0c2001-07-20 16:36:21 +00004060#ifdef Py_UNICODE_WIDE
Martin v. Löwis0ba70cc2001-06-26 22:22:37 +00004061 /* Map 21-bit characters to '\U00xxxxxx' */
4062 else if (ch >= 0x10000) {
4063 *p++ = '\\';
4064 *p++ = 'U';
Walter Dörwald79e913e2007-05-12 11:08:06 +00004065 *p++ = hexdigits[(ch >> 28) & 0x0000000F];
4066 *p++ = hexdigits[(ch >> 24) & 0x0000000F];
4067 *p++ = hexdigits[(ch >> 20) & 0x0000000F];
4068 *p++ = hexdigits[(ch >> 16) & 0x0000000F];
4069 *p++ = hexdigits[(ch >> 12) & 0x0000000F];
4070 *p++ = hexdigits[(ch >> 8) & 0x0000000F];
4071 *p++ = hexdigits[(ch >> 4) & 0x0000000F];
4072 *p++ = hexdigits[ch & 0x0000000F];
Benjamin Peterson29060642009-01-31 22:14:21 +00004073 continue;
Martin v. Löwis0ba70cc2001-06-26 22:22:37 +00004074 }
Thomas Wouters89f507f2006-12-13 04:49:30 +00004075#else
Benjamin Peterson29060642009-01-31 22:14:21 +00004076 /* Map UTF-16 surrogate pairs to '\U00xxxxxx' */
4077 else if (ch >= 0xD800 && ch < 0xDC00) {
4078 Py_UNICODE ch2;
4079 Py_UCS4 ucs;
Tim Petersced69f82003-09-16 20:30:58 +00004080
Benjamin Peterson29060642009-01-31 22:14:21 +00004081 ch2 = *s++;
4082 size--;
Georg Brandl78eef3de2010-08-01 20:51:02 +00004083 if (ch2 >= 0xDC00 && ch2 <= 0xDFFF) {
Benjamin Peterson29060642009-01-31 22:14:21 +00004084 ucs = (((ch & 0x03FF) << 10) | (ch2 & 0x03FF)) + 0x00010000;
4085 *p++ = '\\';
4086 *p++ = 'U';
4087 *p++ = hexdigits[(ucs >> 28) & 0x0000000F];
4088 *p++ = hexdigits[(ucs >> 24) & 0x0000000F];
4089 *p++ = hexdigits[(ucs >> 20) & 0x0000000F];
4090 *p++ = hexdigits[(ucs >> 16) & 0x0000000F];
4091 *p++ = hexdigits[(ucs >> 12) & 0x0000000F];
4092 *p++ = hexdigits[(ucs >> 8) & 0x0000000F];
4093 *p++ = hexdigits[(ucs >> 4) & 0x0000000F];
4094 *p++ = hexdigits[ucs & 0x0000000F];
4095 continue;
4096 }
4097 /* Fall through: isolated surrogates are copied as-is */
4098 s--;
4099 size++;
Benjamin Peterson14339b62009-01-31 16:36:08 +00004100 }
Thomas Wouters89f507f2006-12-13 04:49:30 +00004101#endif
Marc-André Lemburg6c6bfb72001-07-20 17:39:11 +00004102
Guido van Rossumd57fd912000-03-10 22:53:23 +00004103 /* Map 16-bit characters to '\uxxxx' */
Marc-André Lemburg6c6bfb72001-07-20 17:39:11 +00004104 if (ch >= 256) {
Guido van Rossumd57fd912000-03-10 22:53:23 +00004105 *p++ = '\\';
4106 *p++ = 'u';
Walter Dörwald79e913e2007-05-12 11:08:06 +00004107 *p++ = hexdigits[(ch >> 12) & 0x000F];
4108 *p++ = hexdigits[(ch >> 8) & 0x000F];
4109 *p++ = hexdigits[(ch >> 4) & 0x000F];
4110 *p++ = hexdigits[ch & 0x000F];
Guido van Rossumd57fd912000-03-10 22:53:23 +00004111 }
Marc-André Lemburg80d1dd52001-07-25 16:05:59 +00004112
Ka-Ping Yeefa004ad2001-01-24 17:19:08 +00004113 /* Map special whitespace to '\t', \n', '\r' */
4114 else if (ch == '\t') {
4115 *p++ = '\\';
4116 *p++ = 't';
4117 }
4118 else if (ch == '\n') {
4119 *p++ = '\\';
4120 *p++ = 'n';
4121 }
4122 else if (ch == '\r') {
4123 *p++ = '\\';
4124 *p++ = 'r';
4125 }
Marc-André Lemburg80d1dd52001-07-25 16:05:59 +00004126
Ka-Ping Yeefa004ad2001-01-24 17:19:08 +00004127 /* Map non-printable US ASCII to '\xhh' */
Marc-André Lemburg11326de2001-11-28 12:56:20 +00004128 else if (ch < ' ' || ch >= 0x7F) {
Guido van Rossumd57fd912000-03-10 22:53:23 +00004129 *p++ = '\\';
Ka-Ping Yeefa004ad2001-01-24 17:19:08 +00004130 *p++ = 'x';
Walter Dörwald79e913e2007-05-12 11:08:06 +00004131 *p++ = hexdigits[(ch >> 4) & 0x000F];
4132 *p++ = hexdigits[ch & 0x000F];
Tim Petersced69f82003-09-16 20:30:58 +00004133 }
Marc-André Lemburg80d1dd52001-07-25 16:05:59 +00004134
Guido van Rossumd57fd912000-03-10 22:53:23 +00004135 /* Copy everything else as-is */
4136 else
4137 *p++ = (char) ch;
4138 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00004139
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00004140 assert(p - PyBytes_AS_STRING(repr) > 0);
4141 if (_PyBytes_Resize(&repr, p - PyBytes_AS_STRING(repr)) < 0)
4142 return NULL;
4143 return repr;
Guido van Rossumd57fd912000-03-10 22:53:23 +00004144}
4145
Alexandre Vassalotti2056bed2008-12-27 19:46:35 +00004146PyObject *PyUnicode_AsUnicodeEscapeString(PyObject *unicode)
Guido van Rossumd57fd912000-03-10 22:53:23 +00004147{
Alexandre Vassalotti9cb6f7f2008-12-27 09:09:15 +00004148 PyObject *s;
Guido van Rossumd57fd912000-03-10 22:53:23 +00004149 if (!PyUnicode_Check(unicode)) {
4150 PyErr_BadArgument();
4151 return NULL;
4152 }
Walter Dörwald79e913e2007-05-12 11:08:06 +00004153 s = PyUnicode_EncodeUnicodeEscape(PyUnicode_AS_UNICODE(unicode),
4154 PyUnicode_GET_SIZE(unicode));
Alexandre Vassalotti9cb6f7f2008-12-27 09:09:15 +00004155 return s;
Guido van Rossumd57fd912000-03-10 22:53:23 +00004156}
4157
4158/* --- Raw Unicode Escape Codec ------------------------------------------- */
4159
4160PyObject *PyUnicode_DecodeRawUnicodeEscape(const char *s,
Benjamin Peterson29060642009-01-31 22:14:21 +00004161 Py_ssize_t size,
4162 const char *errors)
Guido van Rossumd57fd912000-03-10 22:53:23 +00004163{
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004164 const char *starts = s;
Martin v. Löwis18e16552006-02-15 17:27:45 +00004165 Py_ssize_t startinpos;
4166 Py_ssize_t endinpos;
4167 Py_ssize_t outpos;
Guido van Rossumd57fd912000-03-10 22:53:23 +00004168 PyUnicodeObject *v;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004169 Py_UNICODE *p;
Guido van Rossumd57fd912000-03-10 22:53:23 +00004170 const char *end;
4171 const char *bs;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004172 PyObject *errorHandler = NULL;
4173 PyObject *exc = NULL;
Tim Petersced69f82003-09-16 20:30:58 +00004174
Guido van Rossumd57fd912000-03-10 22:53:23 +00004175 /* Escaped strings will always be longer than the resulting
4176 Unicode string, so we start with size here and then reduce the
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004177 length after conversion to the true value. (But decoding error
4178 handler might have to resize the string) */
Guido van Rossumd57fd912000-03-10 22:53:23 +00004179 v = _PyUnicode_New(size);
4180 if (v == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00004181 goto onError;
Guido van Rossumd57fd912000-03-10 22:53:23 +00004182 if (size == 0)
Benjamin Peterson29060642009-01-31 22:14:21 +00004183 return (PyObject *)v;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004184 p = PyUnicode_AS_UNICODE(v);
Guido van Rossumd57fd912000-03-10 22:53:23 +00004185 end = s + size;
4186 while (s < end) {
Benjamin Peterson29060642009-01-31 22:14:21 +00004187 unsigned char c;
4188 Py_UCS4 x;
4189 int i;
Martin v. Löwis9a3a9f72003-05-18 12:31:09 +00004190 int count;
Guido van Rossumd57fd912000-03-10 22:53:23 +00004191
Benjamin Peterson29060642009-01-31 22:14:21 +00004192 /* Non-escape characters are interpreted as Unicode ordinals */
4193 if (*s != '\\') {
4194 *p++ = (unsigned char)*s++;
4195 continue;
Benjamin Peterson14339b62009-01-31 16:36:08 +00004196 }
Benjamin Peterson29060642009-01-31 22:14:21 +00004197 startinpos = s-starts;
4198
4199 /* \u-escapes are only interpreted iff the number of leading
4200 backslashes if odd */
4201 bs = s;
4202 for (;s < end;) {
4203 if (*s != '\\')
4204 break;
4205 *p++ = (unsigned char)*s++;
4206 }
4207 if (((s - bs) & 1) == 0 ||
4208 s >= end ||
4209 (*s != 'u' && *s != 'U')) {
4210 continue;
4211 }
4212 p--;
4213 count = *s=='u' ? 4 : 8;
4214 s++;
4215
4216 /* \uXXXX with 4 hex digits, \Uxxxxxxxx with 8 */
4217 outpos = p-PyUnicode_AS_UNICODE(v);
4218 for (x = 0, i = 0; i < count; ++i, ++s) {
4219 c = (unsigned char)*s;
David Malcolm96960882010-11-05 17:23:41 +00004220 if (!Py_ISXDIGIT(c)) {
Benjamin Peterson29060642009-01-31 22:14:21 +00004221 endinpos = s-starts;
4222 if (unicode_decode_call_errorhandler(
4223 errors, &errorHandler,
4224 "rawunicodeescape", "truncated \\uXXXX",
4225 &starts, &end, &startinpos, &endinpos, &exc, &s,
4226 &v, &outpos, &p))
4227 goto onError;
4228 goto nextByte;
4229 }
4230 x = (x<<4) & ~0xF;
4231 if (c >= '0' && c <= '9')
4232 x += c - '0';
4233 else if (c >= 'a' && c <= 'f')
4234 x += 10 + c - 'a';
4235 else
4236 x += 10 + c - 'A';
4237 }
Christian Heimesfe337bf2008-03-23 21:54:12 +00004238 if (x <= 0xffff)
Benjamin Peterson29060642009-01-31 22:14:21 +00004239 /* UCS-2 character */
4240 *p++ = (Py_UNICODE) x;
Christian Heimesfe337bf2008-03-23 21:54:12 +00004241 else if (x <= 0x10ffff) {
Benjamin Peterson29060642009-01-31 22:14:21 +00004242 /* UCS-4 character. Either store directly, or as
4243 surrogate pair. */
Christian Heimesfe337bf2008-03-23 21:54:12 +00004244#ifdef Py_UNICODE_WIDE
Benjamin Peterson29060642009-01-31 22:14:21 +00004245 *p++ = (Py_UNICODE) x;
Christian Heimesfe337bf2008-03-23 21:54:12 +00004246#else
Benjamin Peterson29060642009-01-31 22:14:21 +00004247 x -= 0x10000L;
4248 *p++ = 0xD800 + (Py_UNICODE) (x >> 10);
4249 *p++ = 0xDC00 + (Py_UNICODE) (x & 0x03FF);
Christian Heimesfe337bf2008-03-23 21:54:12 +00004250#endif
4251 } else {
4252 endinpos = s-starts;
4253 outpos = p-PyUnicode_AS_UNICODE(v);
Martin v. Löwis9a3a9f72003-05-18 12:31:09 +00004254 if (unicode_decode_call_errorhandler(
4255 errors, &errorHandler,
4256 "rawunicodeescape", "\\Uxxxxxxxx out of range",
Benjamin Peterson29060642009-01-31 22:14:21 +00004257 &starts, &end, &startinpos, &endinpos, &exc, &s,
4258 &v, &outpos, &p))
4259 goto onError;
Martin v. Löwis9a3a9f72003-05-18 12:31:09 +00004260 }
Benjamin Peterson29060642009-01-31 22:14:21 +00004261 nextByte:
4262 ;
Guido van Rossumd57fd912000-03-10 22:53:23 +00004263 }
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004264 if (_PyUnicode_Resize(&v, p - PyUnicode_AS_UNICODE(v)) < 0)
Benjamin Peterson29060642009-01-31 22:14:21 +00004265 goto onError;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004266 Py_XDECREF(errorHandler);
4267 Py_XDECREF(exc);
Guido van Rossumd57fd912000-03-10 22:53:23 +00004268 return (PyObject *)v;
Tim Petersced69f82003-09-16 20:30:58 +00004269
Benjamin Peterson29060642009-01-31 22:14:21 +00004270 onError:
Guido van Rossumd57fd912000-03-10 22:53:23 +00004271 Py_XDECREF(v);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004272 Py_XDECREF(errorHandler);
4273 Py_XDECREF(exc);
Guido van Rossumd57fd912000-03-10 22:53:23 +00004274 return NULL;
4275}
4276
4277PyObject *PyUnicode_EncodeRawUnicodeEscape(const Py_UNICODE *s,
Benjamin Peterson29060642009-01-31 22:14:21 +00004278 Py_ssize_t size)
Guido van Rossumd57fd912000-03-10 22:53:23 +00004279{
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00004280 PyObject *repr;
Guido van Rossumd57fd912000-03-10 22:53:23 +00004281 char *p;
4282 char *q;
4283
Martin v. Löwis9a3a9f72003-05-18 12:31:09 +00004284#ifdef Py_UNICODE_WIDE
Neal Norwitz3ce5d922008-08-24 07:08:55 +00004285 const Py_ssize_t expandsize = 10;
Martin v. Löwis9a3a9f72003-05-18 12:31:09 +00004286#else
Neal Norwitz3ce5d922008-08-24 07:08:55 +00004287 const Py_ssize_t expandsize = 6;
Martin v. Löwis9a3a9f72003-05-18 12:31:09 +00004288#endif
Benjamin Peterson14339b62009-01-31 16:36:08 +00004289
Neal Norwitz3ce5d922008-08-24 07:08:55 +00004290 if (size > PY_SSIZE_T_MAX / expandsize)
Benjamin Peterson29060642009-01-31 22:14:21 +00004291 return PyErr_NoMemory();
Benjamin Peterson14339b62009-01-31 16:36:08 +00004292
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00004293 repr = PyBytes_FromStringAndSize(NULL, expandsize * size);
Guido van Rossumd57fd912000-03-10 22:53:23 +00004294 if (repr == NULL)
4295 return NULL;
Marc-André Lemburgb7520772000-08-14 11:29:19 +00004296 if (size == 0)
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00004297 return repr;
Guido van Rossumd57fd912000-03-10 22:53:23 +00004298
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00004299 p = q = PyBytes_AS_STRING(repr);
Guido van Rossumd57fd912000-03-10 22:53:23 +00004300 while (size-- > 0) {
4301 Py_UNICODE ch = *s++;
Martin v. Löwis9a3a9f72003-05-18 12:31:09 +00004302#ifdef Py_UNICODE_WIDE
Benjamin Peterson29060642009-01-31 22:14:21 +00004303 /* Map 32-bit characters to '\Uxxxxxxxx' */
4304 if (ch >= 0x10000) {
Martin v. Löwis9a3a9f72003-05-18 12:31:09 +00004305 *p++ = '\\';
4306 *p++ = 'U';
Walter Dörwalddb5d33e2007-05-12 11:13:47 +00004307 *p++ = hexdigits[(ch >> 28) & 0xf];
4308 *p++ = hexdigits[(ch >> 24) & 0xf];
4309 *p++ = hexdigits[(ch >> 20) & 0xf];
4310 *p++ = hexdigits[(ch >> 16) & 0xf];
4311 *p++ = hexdigits[(ch >> 12) & 0xf];
4312 *p++ = hexdigits[(ch >> 8) & 0xf];
4313 *p++ = hexdigits[(ch >> 4) & 0xf];
4314 *p++ = hexdigits[ch & 15];
Tim Petersced69f82003-09-16 20:30:58 +00004315 }
Martin v. Löwis9a3a9f72003-05-18 12:31:09 +00004316 else
Christian Heimesfe337bf2008-03-23 21:54:12 +00004317#else
Benjamin Peterson29060642009-01-31 22:14:21 +00004318 /* Map UTF-16 surrogate pairs to '\U00xxxxxx' */
4319 if (ch >= 0xD800 && ch < 0xDC00) {
4320 Py_UNICODE ch2;
4321 Py_UCS4 ucs;
Christian Heimesfe337bf2008-03-23 21:54:12 +00004322
Benjamin Peterson29060642009-01-31 22:14:21 +00004323 ch2 = *s++;
4324 size--;
Georg Brandl78eef3de2010-08-01 20:51:02 +00004325 if (ch2 >= 0xDC00 && ch2 <= 0xDFFF) {
Benjamin Peterson29060642009-01-31 22:14:21 +00004326 ucs = (((ch & 0x03FF) << 10) | (ch2 & 0x03FF)) + 0x00010000;
4327 *p++ = '\\';
4328 *p++ = 'U';
4329 *p++ = hexdigits[(ucs >> 28) & 0xf];
4330 *p++ = hexdigits[(ucs >> 24) & 0xf];
4331 *p++ = hexdigits[(ucs >> 20) & 0xf];
4332 *p++ = hexdigits[(ucs >> 16) & 0xf];
4333 *p++ = hexdigits[(ucs >> 12) & 0xf];
4334 *p++ = hexdigits[(ucs >> 8) & 0xf];
4335 *p++ = hexdigits[(ucs >> 4) & 0xf];
4336 *p++ = hexdigits[ucs & 0xf];
4337 continue;
4338 }
4339 /* Fall through: isolated surrogates are copied as-is */
4340 s--;
4341 size++;
4342 }
Martin v. Löwis9a3a9f72003-05-18 12:31:09 +00004343#endif
Benjamin Peterson29060642009-01-31 22:14:21 +00004344 /* Map 16-bit characters to '\uxxxx' */
4345 if (ch >= 256) {
Guido van Rossumd57fd912000-03-10 22:53:23 +00004346 *p++ = '\\';
4347 *p++ = 'u';
Walter Dörwalddb5d33e2007-05-12 11:13:47 +00004348 *p++ = hexdigits[(ch >> 12) & 0xf];
4349 *p++ = hexdigits[(ch >> 8) & 0xf];
4350 *p++ = hexdigits[(ch >> 4) & 0xf];
4351 *p++ = hexdigits[ch & 15];
Guido van Rossumd57fd912000-03-10 22:53:23 +00004352 }
Benjamin Peterson29060642009-01-31 22:14:21 +00004353 /* Copy everything else as-is */
4354 else
Guido van Rossumd57fd912000-03-10 22:53:23 +00004355 *p++ = (char) ch;
4356 }
Guido van Rossum98297ee2007-11-06 21:34:58 +00004357 size = p - q;
4358
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00004359 assert(size > 0);
4360 if (_PyBytes_Resize(&repr, size) < 0)
4361 return NULL;
4362 return repr;
Guido van Rossumd57fd912000-03-10 22:53:23 +00004363}
4364
4365PyObject *PyUnicode_AsRawUnicodeEscapeString(PyObject *unicode)
4366{
Alexandre Vassalotti9cb6f7f2008-12-27 09:09:15 +00004367 PyObject *s;
Guido van Rossumd57fd912000-03-10 22:53:23 +00004368 if (!PyUnicode_Check(unicode)) {
Walter Dörwald711005d2007-05-12 12:03:26 +00004369 PyErr_BadArgument();
4370 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00004371 }
Walter Dörwald711005d2007-05-12 12:03:26 +00004372 s = PyUnicode_EncodeRawUnicodeEscape(PyUnicode_AS_UNICODE(unicode),
4373 PyUnicode_GET_SIZE(unicode));
4374
Alexandre Vassalotti9cb6f7f2008-12-27 09:09:15 +00004375 return s;
Guido van Rossumd57fd912000-03-10 22:53:23 +00004376}
4377
Walter Dörwalda47d1c02005-08-30 10:23:14 +00004378/* --- Unicode Internal Codec ------------------------------------------- */
4379
4380PyObject *_PyUnicode_DecodeUnicodeInternal(const char *s,
Benjamin Peterson29060642009-01-31 22:14:21 +00004381 Py_ssize_t size,
4382 const char *errors)
Walter Dörwalda47d1c02005-08-30 10:23:14 +00004383{
4384 const char *starts = s;
Martin v. Löwis18e16552006-02-15 17:27:45 +00004385 Py_ssize_t startinpos;
4386 Py_ssize_t endinpos;
4387 Py_ssize_t outpos;
Walter Dörwalda47d1c02005-08-30 10:23:14 +00004388 PyUnicodeObject *v;
4389 Py_UNICODE *p;
4390 const char *end;
4391 const char *reason;
4392 PyObject *errorHandler = NULL;
4393 PyObject *exc = NULL;
4394
Neal Norwitzd43069c2006-01-08 01:12:10 +00004395#ifdef Py_UNICODE_WIDE
4396 Py_UNICODE unimax = PyUnicode_GetMax();
4397#endif
4398
Thomas Wouters89f507f2006-12-13 04:49:30 +00004399 /* XXX overflow detection missing */
Walter Dörwalda47d1c02005-08-30 10:23:14 +00004400 v = _PyUnicode_New((size+Py_UNICODE_SIZE-1)/ Py_UNICODE_SIZE);
4401 if (v == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00004402 goto onError;
Walter Dörwalda47d1c02005-08-30 10:23:14 +00004403 if (PyUnicode_GetSize((PyObject *)v) == 0)
Benjamin Peterson29060642009-01-31 22:14:21 +00004404 return (PyObject *)v;
Walter Dörwalda47d1c02005-08-30 10:23:14 +00004405 p = PyUnicode_AS_UNICODE(v);
4406 end = s + size;
4407
4408 while (s < end) {
Thomas Wouters477c8d52006-05-27 19:21:47 +00004409 memcpy(p, s, sizeof(Py_UNICODE));
Walter Dörwalda47d1c02005-08-30 10:23:14 +00004410 /* We have to sanity check the raw data, otherwise doom looms for
4411 some malformed UCS-4 data. */
4412 if (
Benjamin Peterson29060642009-01-31 22:14:21 +00004413#ifdef Py_UNICODE_WIDE
Walter Dörwalda47d1c02005-08-30 10:23:14 +00004414 *p > unimax || *p < 0 ||
Benjamin Peterson29060642009-01-31 22:14:21 +00004415#endif
Walter Dörwalda47d1c02005-08-30 10:23:14 +00004416 end-s < Py_UNICODE_SIZE
4417 )
Benjamin Peterson29060642009-01-31 22:14:21 +00004418 {
Walter Dörwalda47d1c02005-08-30 10:23:14 +00004419 startinpos = s - starts;
4420 if (end-s < Py_UNICODE_SIZE) {
4421 endinpos = end-starts;
4422 reason = "truncated input";
4423 }
4424 else {
4425 endinpos = s - starts + Py_UNICODE_SIZE;
4426 reason = "illegal code point (> 0x10FFFF)";
4427 }
4428 outpos = p - PyUnicode_AS_UNICODE(v);
4429 if (unicode_decode_call_errorhandler(
4430 errors, &errorHandler,
4431 "unicode_internal", reason,
Walter Dörwalde78178e2007-07-30 13:31:40 +00004432 &starts, &end, &startinpos, &endinpos, &exc, &s,
Alexandre Vassalottiaa0e5312008-12-27 06:43:58 +00004433 &v, &outpos, &p)) {
Walter Dörwalda47d1c02005-08-30 10:23:14 +00004434 goto onError;
4435 }
4436 }
4437 else {
4438 p++;
4439 s += Py_UNICODE_SIZE;
4440 }
4441 }
4442
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004443 if (_PyUnicode_Resize(&v, p - PyUnicode_AS_UNICODE(v)) < 0)
Walter Dörwalda47d1c02005-08-30 10:23:14 +00004444 goto onError;
4445 Py_XDECREF(errorHandler);
4446 Py_XDECREF(exc);
4447 return (PyObject *)v;
4448
Benjamin Peterson29060642009-01-31 22:14:21 +00004449 onError:
Walter Dörwalda47d1c02005-08-30 10:23:14 +00004450 Py_XDECREF(v);
4451 Py_XDECREF(errorHandler);
4452 Py_XDECREF(exc);
4453 return NULL;
4454}
4455
Guido van Rossumd57fd912000-03-10 22:53:23 +00004456/* --- Latin-1 Codec ------------------------------------------------------ */
4457
4458PyObject *PyUnicode_DecodeLatin1(const char *s,
Benjamin Peterson29060642009-01-31 22:14:21 +00004459 Py_ssize_t size,
4460 const char *errors)
Guido van Rossumd57fd912000-03-10 22:53:23 +00004461{
4462 PyUnicodeObject *v;
4463 Py_UNICODE *p;
Antoine Pitrouab868312009-01-10 15:40:25 +00004464 const char *e, *unrolled_end;
Tim Petersced69f82003-09-16 20:30:58 +00004465
Guido van Rossumd57fd912000-03-10 22:53:23 +00004466 /* Latin-1 is equivalent to the first 256 ordinals in Unicode. */
Hye-Shik Chang4a264fb2003-12-19 01:59:56 +00004467 if (size == 1) {
Benjamin Peterson29060642009-01-31 22:14:21 +00004468 Py_UNICODE r = *(unsigned char*)s;
4469 return PyUnicode_FromUnicode(&r, 1);
Marc-André Lemburg8155e0e2001-04-23 14:44:21 +00004470 }
4471
Guido van Rossumd57fd912000-03-10 22:53:23 +00004472 v = _PyUnicode_New(size);
4473 if (v == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00004474 goto onError;
Guido van Rossumd57fd912000-03-10 22:53:23 +00004475 if (size == 0)
Benjamin Peterson29060642009-01-31 22:14:21 +00004476 return (PyObject *)v;
Guido van Rossumd57fd912000-03-10 22:53:23 +00004477 p = PyUnicode_AS_UNICODE(v);
Antoine Pitrouab868312009-01-10 15:40:25 +00004478 e = s + size;
4479 /* Unrolling the copy makes it much faster by reducing the looping
4480 overhead. This is similar to what many memcpy() implementations do. */
4481 unrolled_end = e - 4;
4482 while (s < unrolled_end) {
4483 p[0] = (unsigned char) s[0];
4484 p[1] = (unsigned char) s[1];
4485 p[2] = (unsigned char) s[2];
4486 p[3] = (unsigned char) s[3];
4487 s += 4;
4488 p += 4;
4489 }
4490 while (s < e)
4491 *p++ = (unsigned char) *s++;
Guido van Rossumd57fd912000-03-10 22:53:23 +00004492 return (PyObject *)v;
Tim Petersced69f82003-09-16 20:30:58 +00004493
Benjamin Peterson29060642009-01-31 22:14:21 +00004494 onError:
Guido van Rossumd57fd912000-03-10 22:53:23 +00004495 Py_XDECREF(v);
4496 return NULL;
4497}
4498
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004499/* create or adjust a UnicodeEncodeError */
4500static void make_encode_exception(PyObject **exceptionObject,
Benjamin Peterson29060642009-01-31 22:14:21 +00004501 const char *encoding,
4502 const Py_UNICODE *unicode, Py_ssize_t size,
4503 Py_ssize_t startpos, Py_ssize_t endpos,
4504 const char *reason)
Guido van Rossumd57fd912000-03-10 22:53:23 +00004505{
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004506 if (*exceptionObject == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +00004507 *exceptionObject = PyUnicodeEncodeError_Create(
4508 encoding, unicode, size, startpos, endpos, reason);
Guido van Rossumd57fd912000-03-10 22:53:23 +00004509 }
4510 else {
Benjamin Peterson29060642009-01-31 22:14:21 +00004511 if (PyUnicodeEncodeError_SetStart(*exceptionObject, startpos))
4512 goto onError;
4513 if (PyUnicodeEncodeError_SetEnd(*exceptionObject, endpos))
4514 goto onError;
4515 if (PyUnicodeEncodeError_SetReason(*exceptionObject, reason))
4516 goto onError;
4517 return;
4518 onError:
4519 Py_DECREF(*exceptionObject);
4520 *exceptionObject = NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00004521 }
4522}
4523
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004524/* raises a UnicodeEncodeError */
4525static void raise_encode_exception(PyObject **exceptionObject,
Benjamin Peterson29060642009-01-31 22:14:21 +00004526 const char *encoding,
4527 const Py_UNICODE *unicode, Py_ssize_t size,
4528 Py_ssize_t startpos, Py_ssize_t endpos,
4529 const char *reason)
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004530{
4531 make_encode_exception(exceptionObject,
Benjamin Peterson29060642009-01-31 22:14:21 +00004532 encoding, unicode, size, startpos, endpos, reason);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004533 if (*exceptionObject != NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00004534 PyCodec_StrictErrors(*exceptionObject);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004535}
4536
4537/* error handling callback helper:
4538 build arguments, call the callback and check the arguments,
4539 put the result into newpos and return the replacement string, which
4540 has to be freed by the caller */
4541static PyObject *unicode_encode_call_errorhandler(const char *errors,
Benjamin Peterson29060642009-01-31 22:14:21 +00004542 PyObject **errorHandler,
4543 const char *encoding, const char *reason,
4544 const Py_UNICODE *unicode, Py_ssize_t size, PyObject **exceptionObject,
4545 Py_ssize_t startpos, Py_ssize_t endpos,
4546 Py_ssize_t *newpos)
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004547{
Martin v. Löwisdb12d452009-05-02 18:52:14 +00004548 static char *argparse = "On;encoding error handler must return (str/bytes, int) tuple";
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004549
4550 PyObject *restuple;
4551 PyObject *resunicode;
4552
4553 if (*errorHandler == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +00004554 *errorHandler = PyCodec_LookupError(errors);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004555 if (*errorHandler == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00004556 return NULL;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004557 }
4558
4559 make_encode_exception(exceptionObject,
Benjamin Peterson29060642009-01-31 22:14:21 +00004560 encoding, unicode, size, startpos, endpos, reason);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004561 if (*exceptionObject == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00004562 return NULL;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004563
4564 restuple = PyObject_CallFunctionObjArgs(
Benjamin Peterson29060642009-01-31 22:14:21 +00004565 *errorHandler, *exceptionObject, NULL);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004566 if (restuple == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00004567 return NULL;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004568 if (!PyTuple_Check(restuple)) {
Martin v. Löwisdb12d452009-05-02 18:52:14 +00004569 PyErr_SetString(PyExc_TypeError, &argparse[3]);
Benjamin Peterson29060642009-01-31 22:14:21 +00004570 Py_DECREF(restuple);
4571 return NULL;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004572 }
Martin v. Löwisdb12d452009-05-02 18:52:14 +00004573 if (!PyArg_ParseTuple(restuple, argparse,
Benjamin Peterson29060642009-01-31 22:14:21 +00004574 &resunicode, newpos)) {
4575 Py_DECREF(restuple);
4576 return NULL;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004577 }
Martin v. Löwisdb12d452009-05-02 18:52:14 +00004578 if (!PyUnicode_Check(resunicode) && !PyBytes_Check(resunicode)) {
4579 PyErr_SetString(PyExc_TypeError, &argparse[3]);
4580 Py_DECREF(restuple);
4581 return NULL;
4582 }
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004583 if (*newpos<0)
Benjamin Peterson29060642009-01-31 22:14:21 +00004584 *newpos = size+*newpos;
Walter Dörwald2e0b18a2003-01-31 17:19:08 +00004585 if (*newpos<0 || *newpos>size) {
Benjamin Peterson29060642009-01-31 22:14:21 +00004586 PyErr_Format(PyExc_IndexError, "position %zd from error handler out of bounds", *newpos);
4587 Py_DECREF(restuple);
4588 return NULL;
Walter Dörwald2e0b18a2003-01-31 17:19:08 +00004589 }
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004590 Py_INCREF(resunicode);
4591 Py_DECREF(restuple);
4592 return resunicode;
4593}
4594
4595static PyObject *unicode_encode_ucs1(const Py_UNICODE *p,
Benjamin Peterson29060642009-01-31 22:14:21 +00004596 Py_ssize_t size,
4597 const char *errors,
4598 int limit)
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004599{
4600 /* output object */
4601 PyObject *res;
4602 /* pointers to the beginning and end+1 of input */
4603 const Py_UNICODE *startp = p;
4604 const Py_UNICODE *endp = p + size;
4605 /* pointer to the beginning of the unencodable characters */
4606 /* const Py_UNICODE *badp = NULL; */
4607 /* pointer into the output */
4608 char *str;
4609 /* current output position */
Martin v. Löwis18e16552006-02-15 17:27:45 +00004610 Py_ssize_t ressize;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00004611 const char *encoding = (limit == 256) ? "latin-1" : "ascii";
4612 const char *reason = (limit == 256) ? "ordinal not in range(256)" : "ordinal not in range(128)";
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004613 PyObject *errorHandler = NULL;
4614 PyObject *exc = NULL;
4615 /* the following variable is used for caching string comparisons
4616 * -1=not initialized, 0=unknown, 1=strict, 2=replace, 3=ignore, 4=xmlcharrefreplace */
4617 int known_errorHandler = -1;
4618
4619 /* allocate enough for a simple encoding without
4620 replacements, if we need more, we'll resize */
Guido van Rossum98297ee2007-11-06 21:34:58 +00004621 if (size == 0)
Christian Heimes72b710a2008-05-26 13:28:38 +00004622 return PyBytes_FromStringAndSize(NULL, 0);
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00004623 res = PyBytes_FromStringAndSize(NULL, size);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004624 if (res == NULL)
Guido van Rossum98297ee2007-11-06 21:34:58 +00004625 return NULL;
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00004626 str = PyBytes_AS_STRING(res);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004627 ressize = size;
4628
4629 while (p<endp) {
Benjamin Peterson29060642009-01-31 22:14:21 +00004630 Py_UNICODE c = *p;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004631
Benjamin Peterson29060642009-01-31 22:14:21 +00004632 /* can we encode this? */
4633 if (c<limit) {
4634 /* no overflow check, because we know that the space is enough */
4635 *str++ = (char)c;
4636 ++p;
Benjamin Peterson14339b62009-01-31 16:36:08 +00004637 }
Benjamin Peterson29060642009-01-31 22:14:21 +00004638 else {
4639 Py_ssize_t unicodepos = p-startp;
4640 Py_ssize_t requiredsize;
4641 PyObject *repunicode;
4642 Py_ssize_t repsize;
4643 Py_ssize_t newpos;
4644 Py_ssize_t respos;
4645 Py_UNICODE *uni2;
4646 /* startpos for collecting unencodable chars */
4647 const Py_UNICODE *collstart = p;
4648 const Py_UNICODE *collend = p;
4649 /* find all unecodable characters */
4650 while ((collend < endp) && ((*collend)>=limit))
4651 ++collend;
4652 /* cache callback name lookup (if not done yet, i.e. it's the first error) */
4653 if (known_errorHandler==-1) {
4654 if ((errors==NULL) || (!strcmp(errors, "strict")))
4655 known_errorHandler = 1;
4656 else if (!strcmp(errors, "replace"))
4657 known_errorHandler = 2;
4658 else if (!strcmp(errors, "ignore"))
4659 known_errorHandler = 3;
4660 else if (!strcmp(errors, "xmlcharrefreplace"))
4661 known_errorHandler = 4;
4662 else
4663 known_errorHandler = 0;
4664 }
4665 switch (known_errorHandler) {
4666 case 1: /* strict */
4667 raise_encode_exception(&exc, encoding, startp, size, collstart-startp, collend-startp, reason);
4668 goto onError;
4669 case 2: /* replace */
4670 while (collstart++<collend)
4671 *str++ = '?'; /* fall through */
4672 case 3: /* ignore */
4673 p = collend;
4674 break;
4675 case 4: /* xmlcharrefreplace */
4676 respos = str - PyBytes_AS_STRING(res);
4677 /* determine replacement size (temporarily (mis)uses p) */
4678 for (p = collstart, repsize = 0; p < collend; ++p) {
4679 if (*p<10)
4680 repsize += 2+1+1;
4681 else if (*p<100)
4682 repsize += 2+2+1;
4683 else if (*p<1000)
4684 repsize += 2+3+1;
4685 else if (*p<10000)
4686 repsize += 2+4+1;
Hye-Shik Chang40e95092003-12-22 01:31:13 +00004687#ifndef Py_UNICODE_WIDE
Benjamin Peterson29060642009-01-31 22:14:21 +00004688 else
4689 repsize += 2+5+1;
Hye-Shik Chang40e95092003-12-22 01:31:13 +00004690#else
Benjamin Peterson29060642009-01-31 22:14:21 +00004691 else if (*p<100000)
4692 repsize += 2+5+1;
4693 else if (*p<1000000)
4694 repsize += 2+6+1;
4695 else
4696 repsize += 2+7+1;
Hye-Shik Chang4a264fb2003-12-19 01:59:56 +00004697#endif
Benjamin Peterson29060642009-01-31 22:14:21 +00004698 }
4699 requiredsize = respos+repsize+(endp-collend);
4700 if (requiredsize > ressize) {
4701 if (requiredsize<2*ressize)
4702 requiredsize = 2*ressize;
4703 if (_PyBytes_Resize(&res, requiredsize))
4704 goto onError;
4705 str = PyBytes_AS_STRING(res) + respos;
4706 ressize = requiredsize;
4707 }
4708 /* generate replacement (temporarily (mis)uses p) */
4709 for (p = collstart; p < collend; ++p) {
4710 str += sprintf(str, "&#%d;", (int)*p);
4711 }
4712 p = collend;
4713 break;
4714 default:
4715 repunicode = unicode_encode_call_errorhandler(errors, &errorHandler,
4716 encoding, reason, startp, size, &exc,
4717 collstart-startp, collend-startp, &newpos);
4718 if (repunicode == NULL)
4719 goto onError;
Martin v. Löwis011e8422009-05-05 04:43:17 +00004720 if (PyBytes_Check(repunicode)) {
4721 /* Directly copy bytes result to output. */
4722 repsize = PyBytes_Size(repunicode);
4723 if (repsize > 1) {
4724 /* Make room for all additional bytes. */
Amaury Forgeot d'Arc84ec8d92009-06-29 22:36:49 +00004725 respos = str - PyBytes_AS_STRING(res);
Martin v. Löwis011e8422009-05-05 04:43:17 +00004726 if (_PyBytes_Resize(&res, ressize+repsize-1)) {
4727 Py_DECREF(repunicode);
4728 goto onError;
4729 }
Amaury Forgeot d'Arc84ec8d92009-06-29 22:36:49 +00004730 str = PyBytes_AS_STRING(res) + respos;
Martin v. Löwis011e8422009-05-05 04:43:17 +00004731 ressize += repsize-1;
4732 }
4733 memcpy(str, PyBytes_AsString(repunicode), repsize);
4734 str += repsize;
4735 p = startp + newpos;
Martin v. Löwisdb12d452009-05-02 18:52:14 +00004736 Py_DECREF(repunicode);
Martin v. Löwis011e8422009-05-05 04:43:17 +00004737 break;
Martin v. Löwisdb12d452009-05-02 18:52:14 +00004738 }
Benjamin Peterson29060642009-01-31 22:14:21 +00004739 /* need more space? (at least enough for what we
4740 have+the replacement+the rest of the string, so
4741 we won't have to check space for encodable characters) */
4742 respos = str - PyBytes_AS_STRING(res);
4743 repsize = PyUnicode_GET_SIZE(repunicode);
4744 requiredsize = respos+repsize+(endp-collend);
4745 if (requiredsize > ressize) {
4746 if (requiredsize<2*ressize)
4747 requiredsize = 2*ressize;
4748 if (_PyBytes_Resize(&res, requiredsize)) {
4749 Py_DECREF(repunicode);
4750 goto onError;
4751 }
4752 str = PyBytes_AS_STRING(res) + respos;
4753 ressize = requiredsize;
4754 }
4755 /* check if there is anything unencodable in the replacement
4756 and copy it to the output */
4757 for (uni2 = PyUnicode_AS_UNICODE(repunicode);repsize-->0; ++uni2, ++str) {
4758 c = *uni2;
4759 if (c >= limit) {
4760 raise_encode_exception(&exc, encoding, startp, size,
4761 unicodepos, unicodepos+1, reason);
4762 Py_DECREF(repunicode);
4763 goto onError;
4764 }
4765 *str = (char)c;
4766 }
4767 p = startp + newpos;
Benjamin Peterson14339b62009-01-31 16:36:08 +00004768 Py_DECREF(repunicode);
Benjamin Peterson14339b62009-01-31 16:36:08 +00004769 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00004770 }
4771 }
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00004772 /* Resize if we allocated to much */
4773 size = str - PyBytes_AS_STRING(res);
4774 if (size < ressize) { /* If this falls res will be NULL */
Alexandre Vassalottibad1b922008-12-27 09:49:09 +00004775 assert(size >= 0);
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00004776 if (_PyBytes_Resize(&res, size) < 0)
4777 goto onError;
4778 }
4779
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004780 Py_XDECREF(errorHandler);
4781 Py_XDECREF(exc);
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00004782 return res;
4783
4784 onError:
4785 Py_XDECREF(res);
4786 Py_XDECREF(errorHandler);
4787 Py_XDECREF(exc);
4788 return NULL;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004789}
4790
Guido van Rossumd57fd912000-03-10 22:53:23 +00004791PyObject *PyUnicode_EncodeLatin1(const Py_UNICODE *p,
Benjamin Peterson29060642009-01-31 22:14:21 +00004792 Py_ssize_t size,
4793 const char *errors)
Guido van Rossumd57fd912000-03-10 22:53:23 +00004794{
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004795 return unicode_encode_ucs1(p, size, errors, 256);
Guido van Rossumd57fd912000-03-10 22:53:23 +00004796}
4797
4798PyObject *PyUnicode_AsLatin1String(PyObject *unicode)
4799{
4800 if (!PyUnicode_Check(unicode)) {
Benjamin Peterson29060642009-01-31 22:14:21 +00004801 PyErr_BadArgument();
4802 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00004803 }
4804 return PyUnicode_EncodeLatin1(PyUnicode_AS_UNICODE(unicode),
Benjamin Peterson29060642009-01-31 22:14:21 +00004805 PyUnicode_GET_SIZE(unicode),
4806 NULL);
Guido van Rossumd57fd912000-03-10 22:53:23 +00004807}
4808
4809/* --- 7-bit ASCII Codec -------------------------------------------------- */
4810
Guido van Rossumd57fd912000-03-10 22:53:23 +00004811PyObject *PyUnicode_DecodeASCII(const char *s,
Benjamin Peterson29060642009-01-31 22:14:21 +00004812 Py_ssize_t size,
4813 const char *errors)
Guido van Rossumd57fd912000-03-10 22:53:23 +00004814{
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004815 const char *starts = s;
Guido van Rossumd57fd912000-03-10 22:53:23 +00004816 PyUnicodeObject *v;
4817 Py_UNICODE *p;
Martin v. Löwis18e16552006-02-15 17:27:45 +00004818 Py_ssize_t startinpos;
4819 Py_ssize_t endinpos;
4820 Py_ssize_t outpos;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004821 const char *e;
4822 PyObject *errorHandler = NULL;
4823 PyObject *exc = NULL;
Tim Petersced69f82003-09-16 20:30:58 +00004824
Guido van Rossumd57fd912000-03-10 22:53:23 +00004825 /* ASCII is equivalent to the first 128 ordinals in Unicode. */
Marc-André Lemburg8155e0e2001-04-23 14:44:21 +00004826 if (size == 1 && *(unsigned char*)s < 128) {
Benjamin Peterson29060642009-01-31 22:14:21 +00004827 Py_UNICODE r = *(unsigned char*)s;
4828 return PyUnicode_FromUnicode(&r, 1);
Marc-André Lemburg8155e0e2001-04-23 14:44:21 +00004829 }
Tim Petersced69f82003-09-16 20:30:58 +00004830
Guido van Rossumd57fd912000-03-10 22:53:23 +00004831 v = _PyUnicode_New(size);
4832 if (v == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00004833 goto onError;
Guido van Rossumd57fd912000-03-10 22:53:23 +00004834 if (size == 0)
Benjamin Peterson29060642009-01-31 22:14:21 +00004835 return (PyObject *)v;
Guido van Rossumd57fd912000-03-10 22:53:23 +00004836 p = PyUnicode_AS_UNICODE(v);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004837 e = s + size;
4838 while (s < e) {
Benjamin Peterson29060642009-01-31 22:14:21 +00004839 register unsigned char c = (unsigned char)*s;
4840 if (c < 128) {
4841 *p++ = c;
4842 ++s;
4843 }
4844 else {
4845 startinpos = s-starts;
4846 endinpos = startinpos + 1;
4847 outpos = p - (Py_UNICODE *)PyUnicode_AS_UNICODE(v);
4848 if (unicode_decode_call_errorhandler(
4849 errors, &errorHandler,
4850 "ascii", "ordinal not in range(128)",
4851 &starts, &e, &startinpos, &endinpos, &exc, &s,
4852 &v, &outpos, &p))
4853 goto onError;
4854 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00004855 }
Martin v. Löwis5b222132007-06-10 09:51:05 +00004856 if (p - PyUnicode_AS_UNICODE(v) < PyUnicode_GET_SIZE(v))
Benjamin Peterson29060642009-01-31 22:14:21 +00004857 if (_PyUnicode_Resize(&v, p - PyUnicode_AS_UNICODE(v)) < 0)
4858 goto onError;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004859 Py_XDECREF(errorHandler);
4860 Py_XDECREF(exc);
Guido van Rossumd57fd912000-03-10 22:53:23 +00004861 return (PyObject *)v;
Tim Petersced69f82003-09-16 20:30:58 +00004862
Benjamin Peterson29060642009-01-31 22:14:21 +00004863 onError:
Guido van Rossumd57fd912000-03-10 22:53:23 +00004864 Py_XDECREF(v);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004865 Py_XDECREF(errorHandler);
4866 Py_XDECREF(exc);
Guido van Rossumd57fd912000-03-10 22:53:23 +00004867 return NULL;
4868}
4869
Guido van Rossumd57fd912000-03-10 22:53:23 +00004870PyObject *PyUnicode_EncodeASCII(const Py_UNICODE *p,
Benjamin Peterson29060642009-01-31 22:14:21 +00004871 Py_ssize_t size,
4872 const char *errors)
Guido van Rossumd57fd912000-03-10 22:53:23 +00004873{
Walter Dörwald3aeb6322002-09-02 13:14:32 +00004874 return unicode_encode_ucs1(p, size, errors, 128);
Guido van Rossumd57fd912000-03-10 22:53:23 +00004875}
4876
4877PyObject *PyUnicode_AsASCIIString(PyObject *unicode)
4878{
4879 if (!PyUnicode_Check(unicode)) {
Benjamin Peterson29060642009-01-31 22:14:21 +00004880 PyErr_BadArgument();
4881 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00004882 }
4883 return PyUnicode_EncodeASCII(PyUnicode_AS_UNICODE(unicode),
Benjamin Peterson29060642009-01-31 22:14:21 +00004884 PyUnicode_GET_SIZE(unicode),
4885 NULL);
Guido van Rossumd57fd912000-03-10 22:53:23 +00004886}
4887
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00004888#if defined(MS_WINDOWS) && defined(HAVE_USABLE_WCHAR_T)
Guido van Rossum2ea3e142000-03-31 17:24:09 +00004889
Guido van Rossumb7a40ba2000-03-28 02:01:52 +00004890/* --- MBCS codecs for Windows -------------------------------------------- */
Guido van Rossum2ea3e142000-03-31 17:24:09 +00004891
Hirokazu Yamamoto35302462009-03-21 13:23:27 +00004892#if SIZEOF_INT < SIZEOF_SIZE_T
Thomas Wouters0e3f5912006-08-11 14:57:12 +00004893#define NEED_RETRY
4894#endif
4895
4896/* XXX This code is limited to "true" double-byte encodings, as
4897 a) it assumes an incomplete character consists of a single byte, and
4898 b) IsDBCSLeadByte (probably) does not work for non-DBCS multi-byte
Benjamin Peterson29060642009-01-31 22:14:21 +00004899 encodings, see IsDBCSLeadByteEx documentation. */
Thomas Wouters0e3f5912006-08-11 14:57:12 +00004900
4901static int is_dbcs_lead_byte(const char *s, int offset)
4902{
4903 const char *curr = s + offset;
4904
4905 if (IsDBCSLeadByte(*curr)) {
Benjamin Peterson29060642009-01-31 22:14:21 +00004906 const char *prev = CharPrev(s, curr);
4907 return (prev == curr) || !IsDBCSLeadByte(*prev) || (curr - prev == 2);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00004908 }
4909 return 0;
4910}
4911
4912/*
4913 * Decode MBCS string into unicode object. If 'final' is set, converts
4914 * trailing lead-byte too. Returns consumed size if succeed, -1 otherwise.
4915 */
4916static int decode_mbcs(PyUnicodeObject **v,
Benjamin Peterson29060642009-01-31 22:14:21 +00004917 const char *s, /* MBCS string */
4918 int size, /* sizeof MBCS string */
Victor Stinner554f3f02010-06-16 23:33:54 +00004919 int final,
4920 const char *errors)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00004921{
4922 Py_UNICODE *p;
Victor Stinner554f3f02010-06-16 23:33:54 +00004923 Py_ssize_t n;
4924 DWORD usize;
4925 DWORD flags;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00004926
4927 assert(size >= 0);
4928
Victor Stinner554f3f02010-06-16 23:33:54 +00004929 /* check and handle 'errors' arg */
4930 if (errors==NULL || strcmp(errors, "strict")==0)
4931 flags = MB_ERR_INVALID_CHARS;
4932 else if (strcmp(errors, "ignore")==0)
4933 flags = 0;
4934 else {
4935 PyErr_Format(PyExc_ValueError,
4936 "mbcs encoding does not support errors='%s'",
4937 errors);
4938 return -1;
4939 }
4940
Thomas Wouters0e3f5912006-08-11 14:57:12 +00004941 /* Skip trailing lead-byte unless 'final' is set */
4942 if (!final && size >= 1 && is_dbcs_lead_byte(s, size - 1))
Benjamin Peterson29060642009-01-31 22:14:21 +00004943 --size;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00004944
4945 /* First get the size of the result */
4946 if (size > 0) {
Victor Stinner554f3f02010-06-16 23:33:54 +00004947 usize = MultiByteToWideChar(CP_ACP, flags, s, size, NULL, 0);
4948 if (usize==0)
4949 goto mbcs_decode_error;
4950 } else
4951 usize = 0;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00004952
4953 if (*v == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +00004954 /* Create unicode object */
4955 *v = _PyUnicode_New(usize);
4956 if (*v == NULL)
4957 return -1;
Victor Stinner554f3f02010-06-16 23:33:54 +00004958 n = 0;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00004959 }
4960 else {
Benjamin Peterson29060642009-01-31 22:14:21 +00004961 /* Extend unicode object */
4962 n = PyUnicode_GET_SIZE(*v);
4963 if (_PyUnicode_Resize(v, n + usize) < 0)
4964 return -1;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00004965 }
4966
4967 /* Do the conversion */
Victor Stinner554f3f02010-06-16 23:33:54 +00004968 if (usize > 0) {
Benjamin Peterson29060642009-01-31 22:14:21 +00004969 p = PyUnicode_AS_UNICODE(*v) + n;
Victor Stinner554f3f02010-06-16 23:33:54 +00004970 if (0 == MultiByteToWideChar(CP_ACP, flags, s, size, p, usize)) {
4971 goto mbcs_decode_error;
Benjamin Peterson29060642009-01-31 22:14:21 +00004972 }
Thomas Wouters0e3f5912006-08-11 14:57:12 +00004973 }
Thomas Wouters0e3f5912006-08-11 14:57:12 +00004974 return size;
Victor Stinner554f3f02010-06-16 23:33:54 +00004975
4976mbcs_decode_error:
4977 /* If the last error was ERROR_NO_UNICODE_TRANSLATION, then
4978 we raise a UnicodeDecodeError - else it is a 'generic'
4979 windows error
4980 */
4981 if (GetLastError()==ERROR_NO_UNICODE_TRANSLATION) {
4982 /* Ideally, we should get reason from FormatMessage - this
4983 is the Windows 2000 English version of the message
4984 */
4985 PyObject *exc = NULL;
4986 const char *reason = "No mapping for the Unicode character exists "
4987 "in the target multi-byte code page.";
4988 make_decode_exception(&exc, "mbcs", s, size, 0, 0, reason);
4989 if (exc != NULL) {
4990 PyCodec_StrictErrors(exc);
4991 Py_DECREF(exc);
4992 }
4993 } else {
4994 PyErr_SetFromWindowsErrWithFilename(0, NULL);
4995 }
4996 return -1;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00004997}
4998
4999PyObject *PyUnicode_DecodeMBCSStateful(const char *s,
Benjamin Peterson29060642009-01-31 22:14:21 +00005000 Py_ssize_t size,
5001 const char *errors,
5002 Py_ssize_t *consumed)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00005003{
5004 PyUnicodeObject *v = NULL;
5005 int done;
5006
5007 if (consumed)
Benjamin Peterson29060642009-01-31 22:14:21 +00005008 *consumed = 0;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00005009
5010#ifdef NEED_RETRY
5011 retry:
5012 if (size > INT_MAX)
Victor Stinner554f3f02010-06-16 23:33:54 +00005013 done = decode_mbcs(&v, s, INT_MAX, 0, errors);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00005014 else
5015#endif
Victor Stinner554f3f02010-06-16 23:33:54 +00005016 done = decode_mbcs(&v, s, (int)size, !consumed, errors);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00005017
5018 if (done < 0) {
5019 Py_XDECREF(v);
Benjamin Peterson29060642009-01-31 22:14:21 +00005020 return NULL;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00005021 }
5022
5023 if (consumed)
Benjamin Peterson29060642009-01-31 22:14:21 +00005024 *consumed += done;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00005025
5026#ifdef NEED_RETRY
5027 if (size > INT_MAX) {
Benjamin Peterson29060642009-01-31 22:14:21 +00005028 s += done;
5029 size -= done;
5030 goto retry;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00005031 }
5032#endif
5033
5034 return (PyObject *)v;
5035}
5036
Guido van Rossumb7a40ba2000-03-28 02:01:52 +00005037PyObject *PyUnicode_DecodeMBCS(const char *s,
Benjamin Peterson29060642009-01-31 22:14:21 +00005038 Py_ssize_t size,
5039 const char *errors)
Guido van Rossumb7a40ba2000-03-28 02:01:52 +00005040{
Thomas Wouters0e3f5912006-08-11 14:57:12 +00005041 return PyUnicode_DecodeMBCSStateful(s, size, errors, NULL);
5042}
5043
5044/*
5045 * Convert unicode into string object (MBCS).
5046 * Returns 0 if succeed, -1 otherwise.
5047 */
5048static int encode_mbcs(PyObject **repr,
Benjamin Peterson29060642009-01-31 22:14:21 +00005049 const Py_UNICODE *p, /* unicode */
Victor Stinner554f3f02010-06-16 23:33:54 +00005050 int size, /* size of unicode */
5051 const char* errors)
Thomas Wouters0e3f5912006-08-11 14:57:12 +00005052{
Victor Stinner554f3f02010-06-16 23:33:54 +00005053 BOOL usedDefaultChar = FALSE;
5054 BOOL *pusedDefaultChar;
5055 int mbcssize;
5056 Py_ssize_t n;
5057 PyObject *exc = NULL;
5058 DWORD flags;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00005059
5060 assert(size >= 0);
Guido van Rossumb7a40ba2000-03-28 02:01:52 +00005061
Victor Stinner554f3f02010-06-16 23:33:54 +00005062 /* check and handle 'errors' arg */
5063 if (errors==NULL || strcmp(errors, "strict")==0) {
5064 flags = WC_NO_BEST_FIT_CHARS;
5065 pusedDefaultChar = &usedDefaultChar;
5066 } else if (strcmp(errors, "replace")==0) {
5067 flags = 0;
5068 pusedDefaultChar = NULL;
5069 } else {
5070 PyErr_Format(PyExc_ValueError,
5071 "mbcs encoding does not support errors='%s'",
5072 errors);
5073 return -1;
5074 }
5075
Guido van Rossumb7a40ba2000-03-28 02:01:52 +00005076 /* First get the size of the result */
Thomas Wouters0e3f5912006-08-11 14:57:12 +00005077 if (size > 0) {
Victor Stinner554f3f02010-06-16 23:33:54 +00005078 mbcssize = WideCharToMultiByte(CP_ACP, flags, p, size, NULL, 0,
5079 NULL, pusedDefaultChar);
Benjamin Peterson29060642009-01-31 22:14:21 +00005080 if (mbcssize == 0) {
5081 PyErr_SetFromWindowsErrWithFilename(0, NULL);
5082 return -1;
5083 }
Victor Stinner554f3f02010-06-16 23:33:54 +00005084 /* If we used a default char, then we failed! */
5085 if (pusedDefaultChar && *pusedDefaultChar)
5086 goto mbcs_encode_error;
5087 } else {
5088 mbcssize = 0;
Guido van Rossumb7a40ba2000-03-28 02:01:52 +00005089 }
5090
Thomas Wouters0e3f5912006-08-11 14:57:12 +00005091 if (*repr == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +00005092 /* Create string object */
5093 *repr = PyBytes_FromStringAndSize(NULL, mbcssize);
5094 if (*repr == NULL)
5095 return -1;
Victor Stinner554f3f02010-06-16 23:33:54 +00005096 n = 0;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00005097 }
5098 else {
Benjamin Peterson29060642009-01-31 22:14:21 +00005099 /* Extend string object */
5100 n = PyBytes_Size(*repr);
5101 if (_PyBytes_Resize(repr, n + mbcssize) < 0)
5102 return -1;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00005103 }
5104
5105 /* Do the conversion */
5106 if (size > 0) {
Benjamin Peterson29060642009-01-31 22:14:21 +00005107 char *s = PyBytes_AS_STRING(*repr) + n;
Victor Stinner554f3f02010-06-16 23:33:54 +00005108 if (0 == WideCharToMultiByte(CP_ACP, flags, p, size, s, mbcssize,
5109 NULL, pusedDefaultChar)) {
Benjamin Peterson29060642009-01-31 22:14:21 +00005110 PyErr_SetFromWindowsErrWithFilename(0, NULL);
5111 return -1;
5112 }
Victor Stinner554f3f02010-06-16 23:33:54 +00005113 if (pusedDefaultChar && *pusedDefaultChar)
5114 goto mbcs_encode_error;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00005115 }
Thomas Wouters0e3f5912006-08-11 14:57:12 +00005116 return 0;
Victor Stinner554f3f02010-06-16 23:33:54 +00005117
5118mbcs_encode_error:
5119 raise_encode_exception(&exc, "mbcs", p, size, 0, 0, "invalid character");
5120 Py_XDECREF(exc);
5121 return -1;
Guido van Rossumb7a40ba2000-03-28 02:01:52 +00005122}
5123
5124PyObject *PyUnicode_EncodeMBCS(const Py_UNICODE *p,
Benjamin Peterson29060642009-01-31 22:14:21 +00005125 Py_ssize_t size,
5126 const char *errors)
Guido van Rossumb7a40ba2000-03-28 02:01:52 +00005127{
Thomas Wouters0e3f5912006-08-11 14:57:12 +00005128 PyObject *repr = NULL;
5129 int ret;
Guido van Rossum03e29f12000-05-04 15:52:20 +00005130
Thomas Wouters0e3f5912006-08-11 14:57:12 +00005131#ifdef NEED_RETRY
Benjamin Peterson29060642009-01-31 22:14:21 +00005132 retry:
Thomas Wouters0e3f5912006-08-11 14:57:12 +00005133 if (size > INT_MAX)
Victor Stinner554f3f02010-06-16 23:33:54 +00005134 ret = encode_mbcs(&repr, p, INT_MAX, errors);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00005135 else
5136#endif
Victor Stinner554f3f02010-06-16 23:33:54 +00005137 ret = encode_mbcs(&repr, p, (int)size, errors);
Guido van Rossumb7a40ba2000-03-28 02:01:52 +00005138
Thomas Wouters0e3f5912006-08-11 14:57:12 +00005139 if (ret < 0) {
Benjamin Peterson29060642009-01-31 22:14:21 +00005140 Py_XDECREF(repr);
5141 return NULL;
Guido van Rossumb7a40ba2000-03-28 02:01:52 +00005142 }
Thomas Wouters0e3f5912006-08-11 14:57:12 +00005143
5144#ifdef NEED_RETRY
5145 if (size > INT_MAX) {
Benjamin Peterson29060642009-01-31 22:14:21 +00005146 p += INT_MAX;
5147 size -= INT_MAX;
5148 goto retry;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00005149 }
5150#endif
5151
Guido van Rossumb7a40ba2000-03-28 02:01:52 +00005152 return repr;
5153}
Guido van Rossum2ea3e142000-03-31 17:24:09 +00005154
Mark Hammond0ccda1e2003-07-01 00:13:27 +00005155PyObject *PyUnicode_AsMBCSString(PyObject *unicode)
5156{
5157 if (!PyUnicode_Check(unicode)) {
5158 PyErr_BadArgument();
5159 return NULL;
5160 }
5161 return PyUnicode_EncodeMBCS(PyUnicode_AS_UNICODE(unicode),
Benjamin Peterson29060642009-01-31 22:14:21 +00005162 PyUnicode_GET_SIZE(unicode),
5163 NULL);
Mark Hammond0ccda1e2003-07-01 00:13:27 +00005164}
5165
Thomas Wouters0e3f5912006-08-11 14:57:12 +00005166#undef NEED_RETRY
5167
Martin v. Löwis6238d2b2002-06-30 15:26:10 +00005168#endif /* MS_WINDOWS */
Guido van Rossumb7a40ba2000-03-28 02:01:52 +00005169
Guido van Rossumd57fd912000-03-10 22:53:23 +00005170/* --- Character Mapping Codec -------------------------------------------- */
5171
Guido van Rossumd57fd912000-03-10 22:53:23 +00005172PyObject *PyUnicode_DecodeCharmap(const char *s,
Benjamin Peterson29060642009-01-31 22:14:21 +00005173 Py_ssize_t size,
5174 PyObject *mapping,
5175 const char *errors)
Guido van Rossumd57fd912000-03-10 22:53:23 +00005176{
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005177 const char *starts = s;
Martin v. Löwis18e16552006-02-15 17:27:45 +00005178 Py_ssize_t startinpos;
5179 Py_ssize_t endinpos;
5180 Py_ssize_t outpos;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005181 const char *e;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005182 PyUnicodeObject *v;
5183 Py_UNICODE *p;
Martin v. Löwis18e16552006-02-15 17:27:45 +00005184 Py_ssize_t extrachars = 0;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005185 PyObject *errorHandler = NULL;
5186 PyObject *exc = NULL;
Walter Dörwaldd1c1e102005-10-06 20:29:57 +00005187 Py_UNICODE *mapstring = NULL;
Martin v. Löwis18e16552006-02-15 17:27:45 +00005188 Py_ssize_t maplen = 0;
Tim Petersced69f82003-09-16 20:30:58 +00005189
Guido van Rossumd57fd912000-03-10 22:53:23 +00005190 /* Default to Latin-1 */
5191 if (mapping == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00005192 return PyUnicode_DecodeLatin1(s, size, errors);
Guido van Rossumd57fd912000-03-10 22:53:23 +00005193
5194 v = _PyUnicode_New(size);
5195 if (v == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00005196 goto onError;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005197 if (size == 0)
Benjamin Peterson29060642009-01-31 22:14:21 +00005198 return (PyObject *)v;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005199 p = PyUnicode_AS_UNICODE(v);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005200 e = s + size;
Walter Dörwaldd1c1e102005-10-06 20:29:57 +00005201 if (PyUnicode_CheckExact(mapping)) {
Benjamin Peterson29060642009-01-31 22:14:21 +00005202 mapstring = PyUnicode_AS_UNICODE(mapping);
5203 maplen = PyUnicode_GET_SIZE(mapping);
5204 while (s < e) {
5205 unsigned char ch = *s;
5206 Py_UNICODE x = 0xfffe; /* illegal value */
Guido van Rossumd57fd912000-03-10 22:53:23 +00005207
Benjamin Peterson29060642009-01-31 22:14:21 +00005208 if (ch < maplen)
5209 x = mapstring[ch];
Guido van Rossumd57fd912000-03-10 22:53:23 +00005210
Benjamin Peterson29060642009-01-31 22:14:21 +00005211 if (x == 0xfffe) {
5212 /* undefined mapping */
5213 outpos = p-PyUnicode_AS_UNICODE(v);
5214 startinpos = s-starts;
5215 endinpos = startinpos+1;
5216 if (unicode_decode_call_errorhandler(
5217 errors, &errorHandler,
5218 "charmap", "character maps to <undefined>",
5219 &starts, &e, &startinpos, &endinpos, &exc, &s,
5220 &v, &outpos, &p)) {
5221 goto onError;
5222 }
5223 continue;
5224 }
5225 *p++ = x;
5226 ++s;
Benjamin Peterson14339b62009-01-31 16:36:08 +00005227 }
Walter Dörwaldd1c1e102005-10-06 20:29:57 +00005228 }
5229 else {
Benjamin Peterson29060642009-01-31 22:14:21 +00005230 while (s < e) {
5231 unsigned char ch = *s;
5232 PyObject *w, *x;
Walter Dörwaldd1c1e102005-10-06 20:29:57 +00005233
Benjamin Peterson29060642009-01-31 22:14:21 +00005234 /* Get mapping (char ordinal -> integer, Unicode char or None) */
5235 w = PyLong_FromLong((long)ch);
5236 if (w == NULL)
5237 goto onError;
5238 x = PyObject_GetItem(mapping, w);
5239 Py_DECREF(w);
5240 if (x == NULL) {
5241 if (PyErr_ExceptionMatches(PyExc_LookupError)) {
5242 /* No mapping found means: mapping is undefined. */
5243 PyErr_Clear();
5244 x = Py_None;
5245 Py_INCREF(x);
5246 } else
5247 goto onError;
5248 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00005249
Benjamin Peterson29060642009-01-31 22:14:21 +00005250 /* Apply mapping */
5251 if (PyLong_Check(x)) {
5252 long value = PyLong_AS_LONG(x);
Antoine Pitrou6f80f5d2012-09-23 19:55:21 +02005253 if (value < 0 || value > 0x10FFFF) {
Benjamin Peterson29060642009-01-31 22:14:21 +00005254 PyErr_SetString(PyExc_TypeError,
Antoine Pitrou6f80f5d2012-09-23 19:55:21 +02005255 "character mapping must be in range(0x110000)");
Benjamin Peterson29060642009-01-31 22:14:21 +00005256 Py_DECREF(x);
5257 goto onError;
5258 }
Antoine Pitrou6f80f5d2012-09-23 19:55:21 +02005259
5260#ifndef Py_UNICODE_WIDE
5261 if (value > 0xFFFF) {
5262 /* see the code for 1-n mapping below */
5263 if (extrachars < 2) {
5264 /* resize first */
5265 Py_ssize_t oldpos = p - PyUnicode_AS_UNICODE(v);
5266 Py_ssize_t needed = 10 - extrachars;
5267 extrachars += needed;
5268 /* XXX overflow detection missing */
5269 if (_PyUnicode_Resize(&v,
5270 PyUnicode_GET_SIZE(v) + needed) < 0) {
5271 Py_DECREF(x);
5272 goto onError;
5273 }
5274 p = PyUnicode_AS_UNICODE(v) + oldpos;
5275 }
5276 value -= 0x10000;
5277 *p++ = 0xD800 | (value >> 10);
5278 *p++ = 0xDC00 | (value & 0x3FF);
5279 extrachars -= 2;
5280 }
5281 else
5282#endif
Benjamin Peterson29060642009-01-31 22:14:21 +00005283 *p++ = (Py_UNICODE)value;
5284 }
5285 else if (x == Py_None) {
5286 /* undefined mapping */
5287 outpos = p-PyUnicode_AS_UNICODE(v);
5288 startinpos = s-starts;
5289 endinpos = startinpos+1;
5290 if (unicode_decode_call_errorhandler(
5291 errors, &errorHandler,
5292 "charmap", "character maps to <undefined>",
5293 &starts, &e, &startinpos, &endinpos, &exc, &s,
5294 &v, &outpos, &p)) {
5295 Py_DECREF(x);
5296 goto onError;
5297 }
5298 Py_DECREF(x);
5299 continue;
5300 }
5301 else if (PyUnicode_Check(x)) {
5302 Py_ssize_t targetsize = PyUnicode_GET_SIZE(x);
Benjamin Peterson14339b62009-01-31 16:36:08 +00005303
Benjamin Peterson29060642009-01-31 22:14:21 +00005304 if (targetsize == 1)
5305 /* 1-1 mapping */
5306 *p++ = *PyUnicode_AS_UNICODE(x);
Benjamin Peterson14339b62009-01-31 16:36:08 +00005307
Benjamin Peterson29060642009-01-31 22:14:21 +00005308 else if (targetsize > 1) {
5309 /* 1-n mapping */
5310 if (targetsize > extrachars) {
5311 /* resize first */
5312 Py_ssize_t oldpos = p - PyUnicode_AS_UNICODE(v);
5313 Py_ssize_t needed = (targetsize - extrachars) + \
5314 (targetsize << 2);
5315 extrachars += needed;
5316 /* XXX overflow detection missing */
5317 if (_PyUnicode_Resize(&v,
5318 PyUnicode_GET_SIZE(v) + needed) < 0) {
5319 Py_DECREF(x);
5320 goto onError;
5321 }
5322 p = PyUnicode_AS_UNICODE(v) + oldpos;
5323 }
5324 Py_UNICODE_COPY(p,
5325 PyUnicode_AS_UNICODE(x),
5326 targetsize);
5327 p += targetsize;
5328 extrachars -= targetsize;
5329 }
5330 /* 1-0 mapping: skip the character */
5331 }
5332 else {
5333 /* wrong return value */
5334 PyErr_SetString(PyExc_TypeError,
5335 "character mapping must return integer, None or str");
Benjamin Peterson14339b62009-01-31 16:36:08 +00005336 Py_DECREF(x);
5337 goto onError;
5338 }
Benjamin Peterson29060642009-01-31 22:14:21 +00005339 Py_DECREF(x);
5340 ++s;
Benjamin Peterson14339b62009-01-31 16:36:08 +00005341 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00005342 }
5343 if (p - PyUnicode_AS_UNICODE(v) < PyUnicode_GET_SIZE(v))
Benjamin Peterson29060642009-01-31 22:14:21 +00005344 if (_PyUnicode_Resize(&v, p - PyUnicode_AS_UNICODE(v)) < 0)
5345 goto onError;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005346 Py_XDECREF(errorHandler);
5347 Py_XDECREF(exc);
Guido van Rossumd57fd912000-03-10 22:53:23 +00005348 return (PyObject *)v;
Tim Petersced69f82003-09-16 20:30:58 +00005349
Benjamin Peterson29060642009-01-31 22:14:21 +00005350 onError:
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005351 Py_XDECREF(errorHandler);
5352 Py_XDECREF(exc);
Guido van Rossumd57fd912000-03-10 22:53:23 +00005353 Py_XDECREF(v);
5354 return NULL;
5355}
5356
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00005357/* Charmap encoding: the lookup table */
5358
5359struct encoding_map{
Benjamin Peterson29060642009-01-31 22:14:21 +00005360 PyObject_HEAD
5361 unsigned char level1[32];
5362 int count2, count3;
5363 unsigned char level23[1];
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00005364};
5365
5366static PyObject*
5367encoding_map_size(PyObject *obj, PyObject* args)
5368{
5369 struct encoding_map *map = (struct encoding_map*)obj;
Benjamin Peterson14339b62009-01-31 16:36:08 +00005370 return PyLong_FromLong(sizeof(*map) - 1 + 16*map->count2 +
Benjamin Peterson29060642009-01-31 22:14:21 +00005371 128*map->count3);
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00005372}
5373
5374static PyMethodDef encoding_map_methods[] = {
Benjamin Peterson14339b62009-01-31 16:36:08 +00005375 {"size", encoding_map_size, METH_NOARGS,
Benjamin Peterson29060642009-01-31 22:14:21 +00005376 PyDoc_STR("Return the size (in bytes) of this object") },
5377 { 0 }
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00005378};
5379
5380static void
5381encoding_map_dealloc(PyObject* o)
5382{
Benjamin Peterson14339b62009-01-31 16:36:08 +00005383 PyObject_FREE(o);
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00005384}
5385
5386static PyTypeObject EncodingMapType = {
Benjamin Peterson14339b62009-01-31 16:36:08 +00005387 PyVarObject_HEAD_INIT(NULL, 0)
Benjamin Peterson29060642009-01-31 22:14:21 +00005388 "EncodingMap", /*tp_name*/
5389 sizeof(struct encoding_map), /*tp_basicsize*/
5390 0, /*tp_itemsize*/
5391 /* methods */
5392 encoding_map_dealloc, /*tp_dealloc*/
5393 0, /*tp_print*/
5394 0, /*tp_getattr*/
5395 0, /*tp_setattr*/
Mark Dickinsone94c6792009-02-02 20:36:42 +00005396 0, /*tp_reserved*/
Benjamin Peterson29060642009-01-31 22:14:21 +00005397 0, /*tp_repr*/
5398 0, /*tp_as_number*/
5399 0, /*tp_as_sequence*/
5400 0, /*tp_as_mapping*/
5401 0, /*tp_hash*/
5402 0, /*tp_call*/
5403 0, /*tp_str*/
5404 0, /*tp_getattro*/
5405 0, /*tp_setattro*/
5406 0, /*tp_as_buffer*/
5407 Py_TPFLAGS_DEFAULT, /*tp_flags*/
5408 0, /*tp_doc*/
5409 0, /*tp_traverse*/
5410 0, /*tp_clear*/
5411 0, /*tp_richcompare*/
5412 0, /*tp_weaklistoffset*/
5413 0, /*tp_iter*/
5414 0, /*tp_iternext*/
5415 encoding_map_methods, /*tp_methods*/
5416 0, /*tp_members*/
5417 0, /*tp_getset*/
5418 0, /*tp_base*/
5419 0, /*tp_dict*/
5420 0, /*tp_descr_get*/
5421 0, /*tp_descr_set*/
5422 0, /*tp_dictoffset*/
5423 0, /*tp_init*/
5424 0, /*tp_alloc*/
5425 0, /*tp_new*/
5426 0, /*tp_free*/
5427 0, /*tp_is_gc*/
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00005428};
5429
5430PyObject*
5431PyUnicode_BuildEncodingMap(PyObject* string)
5432{
5433 Py_UNICODE *decode;
5434 PyObject *result;
5435 struct encoding_map *mresult;
5436 int i;
5437 int need_dict = 0;
5438 unsigned char level1[32];
5439 unsigned char level2[512];
5440 unsigned char *mlevel1, *mlevel2, *mlevel3;
5441 int count2 = 0, count3 = 0;
5442
5443 if (!PyUnicode_Check(string) || PyUnicode_GetSize(string) != 256) {
5444 PyErr_BadArgument();
5445 return NULL;
5446 }
5447 decode = PyUnicode_AS_UNICODE(string);
5448 memset(level1, 0xFF, sizeof level1);
5449 memset(level2, 0xFF, sizeof level2);
5450
5451 /* If there isn't a one-to-one mapping of NULL to \0,
5452 or if there are non-BMP characters, we need to use
5453 a mapping dictionary. */
5454 if (decode[0] != 0)
5455 need_dict = 1;
5456 for (i = 1; i < 256; i++) {
5457 int l1, l2;
5458 if (decode[i] == 0
Benjamin Peterson29060642009-01-31 22:14:21 +00005459#ifdef Py_UNICODE_WIDE
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00005460 || decode[i] > 0xFFFF
Benjamin Peterson29060642009-01-31 22:14:21 +00005461#endif
5462 ) {
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00005463 need_dict = 1;
5464 break;
5465 }
5466 if (decode[i] == 0xFFFE)
5467 /* unmapped character */
5468 continue;
5469 l1 = decode[i] >> 11;
5470 l2 = decode[i] >> 7;
5471 if (level1[l1] == 0xFF)
5472 level1[l1] = count2++;
5473 if (level2[l2] == 0xFF)
Benjamin Peterson14339b62009-01-31 16:36:08 +00005474 level2[l2] = count3++;
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00005475 }
5476
5477 if (count2 >= 0xFF || count3 >= 0xFF)
5478 need_dict = 1;
5479
5480 if (need_dict) {
5481 PyObject *result = PyDict_New();
5482 PyObject *key, *value;
5483 if (!result)
5484 return NULL;
5485 for (i = 0; i < 256; i++) {
5486 key = value = NULL;
Christian Heimes217cfd12007-12-02 14:31:20 +00005487 key = PyLong_FromLong(decode[i]);
5488 value = PyLong_FromLong(i);
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00005489 if (!key || !value)
5490 goto failed1;
5491 if (PyDict_SetItem(result, key, value) == -1)
5492 goto failed1;
5493 Py_DECREF(key);
5494 Py_DECREF(value);
5495 }
5496 return result;
5497 failed1:
5498 Py_XDECREF(key);
5499 Py_XDECREF(value);
5500 Py_DECREF(result);
5501 return NULL;
5502 }
5503
5504 /* Create a three-level trie */
5505 result = PyObject_MALLOC(sizeof(struct encoding_map) +
5506 16*count2 + 128*count3 - 1);
5507 if (!result)
5508 return PyErr_NoMemory();
5509 PyObject_Init(result, &EncodingMapType);
5510 mresult = (struct encoding_map*)result;
5511 mresult->count2 = count2;
5512 mresult->count3 = count3;
5513 mlevel1 = mresult->level1;
5514 mlevel2 = mresult->level23;
5515 mlevel3 = mresult->level23 + 16*count2;
5516 memcpy(mlevel1, level1, 32);
5517 memset(mlevel2, 0xFF, 16*count2);
5518 memset(mlevel3, 0, 128*count3);
5519 count3 = 0;
5520 for (i = 1; i < 256; i++) {
5521 int o1, o2, o3, i2, i3;
5522 if (decode[i] == 0xFFFE)
5523 /* unmapped character */
5524 continue;
5525 o1 = decode[i]>>11;
5526 o2 = (decode[i]>>7) & 0xF;
5527 i2 = 16*mlevel1[o1] + o2;
5528 if (mlevel2[i2] == 0xFF)
5529 mlevel2[i2] = count3++;
5530 o3 = decode[i] & 0x7F;
5531 i3 = 128*mlevel2[i2] + o3;
5532 mlevel3[i3] = i;
5533 }
5534 return result;
5535}
5536
5537static int
5538encoding_map_lookup(Py_UNICODE c, PyObject *mapping)
5539{
5540 struct encoding_map *map = (struct encoding_map*)mapping;
5541 int l1 = c>>11;
5542 int l2 = (c>>7) & 0xF;
5543 int l3 = c & 0x7F;
5544 int i;
5545
5546#ifdef Py_UNICODE_WIDE
5547 if (c > 0xFFFF) {
Benjamin Peterson29060642009-01-31 22:14:21 +00005548 return -1;
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00005549 }
5550#endif
5551 if (c == 0)
5552 return 0;
5553 /* level 1*/
5554 i = map->level1[l1];
5555 if (i == 0xFF) {
5556 return -1;
5557 }
5558 /* level 2*/
5559 i = map->level23[16*i+l2];
5560 if (i == 0xFF) {
5561 return -1;
5562 }
5563 /* level 3 */
5564 i = map->level23[16*map->count2 + 128*i + l3];
5565 if (i == 0) {
5566 return -1;
5567 }
5568 return i;
5569}
5570
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005571/* Lookup the character ch in the mapping. If the character
5572 can't be found, Py_None is returned (or NULL, if another
Fred Drakedb390c12005-10-28 14:39:47 +00005573 error occurred). */
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005574static PyObject *charmapencode_lookup(Py_UNICODE c, PyObject *mapping)
Guido van Rossumd57fd912000-03-10 22:53:23 +00005575{
Christian Heimes217cfd12007-12-02 14:31:20 +00005576 PyObject *w = PyLong_FromLong((long)c);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005577 PyObject *x;
5578
5579 if (w == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00005580 return NULL;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005581 x = PyObject_GetItem(mapping, w);
5582 Py_DECREF(w);
5583 if (x == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +00005584 if (PyErr_ExceptionMatches(PyExc_LookupError)) {
5585 /* No mapping found means: mapping is undefined. */
5586 PyErr_Clear();
5587 x = Py_None;
5588 Py_INCREF(x);
5589 return x;
5590 } else
5591 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005592 }
Walter Dörwaldadc72742003-01-08 22:01:33 +00005593 else if (x == Py_None)
Benjamin Peterson29060642009-01-31 22:14:21 +00005594 return x;
Christian Heimes217cfd12007-12-02 14:31:20 +00005595 else if (PyLong_Check(x)) {
Benjamin Peterson29060642009-01-31 22:14:21 +00005596 long value = PyLong_AS_LONG(x);
5597 if (value < 0 || value > 255) {
5598 PyErr_SetString(PyExc_TypeError,
5599 "character mapping must be in range(256)");
5600 Py_DECREF(x);
5601 return NULL;
5602 }
5603 return x;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005604 }
Christian Heimes72b710a2008-05-26 13:28:38 +00005605 else if (PyBytes_Check(x))
Benjamin Peterson29060642009-01-31 22:14:21 +00005606 return x;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005607 else {
Benjamin Peterson29060642009-01-31 22:14:21 +00005608 /* wrong return value */
5609 PyErr_Format(PyExc_TypeError,
5610 "character mapping must return integer, bytes or None, not %.400s",
5611 x->ob_type->tp_name);
5612 Py_DECREF(x);
5613 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005614 }
5615}
5616
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00005617static int
Guido van Rossum98297ee2007-11-06 21:34:58 +00005618charmapencode_resize(PyObject **outobj, Py_ssize_t *outpos, Py_ssize_t requiredsize)
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00005619{
Benjamin Peterson14339b62009-01-31 16:36:08 +00005620 Py_ssize_t outsize = PyBytes_GET_SIZE(*outobj);
5621 /* exponentially overallocate to minimize reallocations */
5622 if (requiredsize < 2*outsize)
5623 requiredsize = 2*outsize;
5624 if (_PyBytes_Resize(outobj, requiredsize))
5625 return -1;
5626 return 0;
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00005627}
5628
Benjamin Peterson14339b62009-01-31 16:36:08 +00005629typedef enum charmapencode_result {
Benjamin Peterson29060642009-01-31 22:14:21 +00005630 enc_SUCCESS, enc_FAILED, enc_EXCEPTION
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00005631}charmapencode_result;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005632/* lookup the character, put the result in the output string and adjust
Walter Dörwald827b0552007-05-12 13:23:53 +00005633 various state variables. Resize the output bytes object if not enough
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005634 space is available. Return a new reference to the object that
5635 was put in the output buffer, or Py_None, if the mapping was undefined
5636 (in which case no character was written) or NULL, if a
Andrew M. Kuchling8294de52005-11-02 16:36:12 +00005637 reallocation error occurred. The caller must decref the result */
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005638static
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00005639charmapencode_result charmapencode_output(Py_UNICODE c, PyObject *mapping,
Benjamin Peterson29060642009-01-31 22:14:21 +00005640 PyObject **outobj, Py_ssize_t *outpos)
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005641{
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00005642 PyObject *rep;
5643 char *outstart;
Christian Heimes72b710a2008-05-26 13:28:38 +00005644 Py_ssize_t outsize = PyBytes_GET_SIZE(*outobj);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005645
Christian Heimes90aa7642007-12-19 02:45:37 +00005646 if (Py_TYPE(mapping) == &EncodingMapType) {
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00005647 int res = encoding_map_lookup(c, mapping);
Benjamin Peterson29060642009-01-31 22:14:21 +00005648 Py_ssize_t requiredsize = *outpos+1;
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00005649 if (res == -1)
5650 return enc_FAILED;
Benjamin Peterson29060642009-01-31 22:14:21 +00005651 if (outsize<requiredsize)
5652 if (charmapencode_resize(outobj, outpos, requiredsize))
5653 return enc_EXCEPTION;
Christian Heimes72b710a2008-05-26 13:28:38 +00005654 outstart = PyBytes_AS_STRING(*outobj);
Benjamin Peterson29060642009-01-31 22:14:21 +00005655 outstart[(*outpos)++] = (char)res;
5656 return enc_SUCCESS;
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00005657 }
5658
5659 rep = charmapencode_lookup(c, mapping);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005660 if (rep==NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00005661 return enc_EXCEPTION;
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00005662 else if (rep==Py_None) {
Benjamin Peterson29060642009-01-31 22:14:21 +00005663 Py_DECREF(rep);
5664 return enc_FAILED;
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00005665 } else {
Benjamin Peterson29060642009-01-31 22:14:21 +00005666 if (PyLong_Check(rep)) {
5667 Py_ssize_t requiredsize = *outpos+1;
5668 if (outsize<requiredsize)
5669 if (charmapencode_resize(outobj, outpos, requiredsize)) {
5670 Py_DECREF(rep);
5671 return enc_EXCEPTION;
5672 }
Christian Heimes72b710a2008-05-26 13:28:38 +00005673 outstart = PyBytes_AS_STRING(*outobj);
Benjamin Peterson29060642009-01-31 22:14:21 +00005674 outstart[(*outpos)++] = (char)PyLong_AS_LONG(rep);
Benjamin Peterson14339b62009-01-31 16:36:08 +00005675 }
Benjamin Peterson29060642009-01-31 22:14:21 +00005676 else {
5677 const char *repchars = PyBytes_AS_STRING(rep);
5678 Py_ssize_t repsize = PyBytes_GET_SIZE(rep);
5679 Py_ssize_t requiredsize = *outpos+repsize;
5680 if (outsize<requiredsize)
5681 if (charmapencode_resize(outobj, outpos, requiredsize)) {
5682 Py_DECREF(rep);
5683 return enc_EXCEPTION;
5684 }
Christian Heimes72b710a2008-05-26 13:28:38 +00005685 outstart = PyBytes_AS_STRING(*outobj);
Benjamin Peterson29060642009-01-31 22:14:21 +00005686 memcpy(outstart + *outpos, repchars, repsize);
5687 *outpos += repsize;
5688 }
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005689 }
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00005690 Py_DECREF(rep);
5691 return enc_SUCCESS;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005692}
5693
5694/* handle an error in PyUnicode_EncodeCharmap
5695 Return 0 on success, -1 on error */
5696static
5697int charmap_encoding_error(
Martin v. Löwis18e16552006-02-15 17:27:45 +00005698 const Py_UNICODE *p, Py_ssize_t size, Py_ssize_t *inpos, PyObject *mapping,
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005699 PyObject **exceptionObject,
Walter Dörwalde5402fb2003-08-14 20:25:29 +00005700 int *known_errorHandler, PyObject **errorHandler, const char *errors,
Guido van Rossum98297ee2007-11-06 21:34:58 +00005701 PyObject **res, Py_ssize_t *respos)
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005702{
5703 PyObject *repunicode = NULL; /* initialize to prevent gcc warning */
Martin v. Löwis18e16552006-02-15 17:27:45 +00005704 Py_ssize_t repsize;
5705 Py_ssize_t newpos;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005706 Py_UNICODE *uni2;
5707 /* startpos for collecting unencodable chars */
Martin v. Löwis18e16552006-02-15 17:27:45 +00005708 Py_ssize_t collstartpos = *inpos;
5709 Py_ssize_t collendpos = *inpos+1;
5710 Py_ssize_t collpos;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005711 char *encoding = "charmap";
5712 char *reason = "character maps to <undefined>";
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00005713 charmapencode_result x;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005714
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005715 /* find all unencodable characters */
5716 while (collendpos < size) {
Thomas Wouters73e5a5b2006-06-08 15:35:45 +00005717 PyObject *rep;
Christian Heimes90aa7642007-12-19 02:45:37 +00005718 if (Py_TYPE(mapping) == &EncodingMapType) {
Benjamin Peterson29060642009-01-31 22:14:21 +00005719 int res = encoding_map_lookup(p[collendpos], mapping);
5720 if (res != -1)
5721 break;
5722 ++collendpos;
5723 continue;
5724 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00005725
Benjamin Peterson29060642009-01-31 22:14:21 +00005726 rep = charmapencode_lookup(p[collendpos], mapping);
5727 if (rep==NULL)
5728 return -1;
5729 else if (rep!=Py_None) {
5730 Py_DECREF(rep);
5731 break;
5732 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00005733 Py_DECREF(rep);
Benjamin Peterson29060642009-01-31 22:14:21 +00005734 ++collendpos;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005735 }
5736 /* cache callback name lookup
5737 * (if not done yet, i.e. it's the first error) */
5738 if (*known_errorHandler==-1) {
Benjamin Peterson29060642009-01-31 22:14:21 +00005739 if ((errors==NULL) || (!strcmp(errors, "strict")))
5740 *known_errorHandler = 1;
5741 else if (!strcmp(errors, "replace"))
5742 *known_errorHandler = 2;
5743 else if (!strcmp(errors, "ignore"))
5744 *known_errorHandler = 3;
5745 else if (!strcmp(errors, "xmlcharrefreplace"))
5746 *known_errorHandler = 4;
5747 else
5748 *known_errorHandler = 0;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005749 }
5750 switch (*known_errorHandler) {
Benjamin Peterson14339b62009-01-31 16:36:08 +00005751 case 1: /* strict */
5752 raise_encode_exception(exceptionObject, encoding, p, size, collstartpos, collendpos, reason);
5753 return -1;
5754 case 2: /* replace */
5755 for (collpos = collstartpos; collpos<collendpos; ++collpos) {
Benjamin Peterson29060642009-01-31 22:14:21 +00005756 x = charmapencode_output('?', mapping, res, respos);
5757 if (x==enc_EXCEPTION) {
5758 return -1;
5759 }
5760 else if (x==enc_FAILED) {
5761 raise_encode_exception(exceptionObject, encoding, p, size, collstartpos, collendpos, reason);
5762 return -1;
5763 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00005764 }
5765 /* fall through */
5766 case 3: /* ignore */
5767 *inpos = collendpos;
5768 break;
5769 case 4: /* xmlcharrefreplace */
5770 /* generate replacement (temporarily (mis)uses p) */
5771 for (collpos = collstartpos; collpos < collendpos; ++collpos) {
Benjamin Peterson29060642009-01-31 22:14:21 +00005772 char buffer[2+29+1+1];
5773 char *cp;
5774 sprintf(buffer, "&#%d;", (int)p[collpos]);
5775 for (cp = buffer; *cp; ++cp) {
5776 x = charmapencode_output(*cp, mapping, res, respos);
5777 if (x==enc_EXCEPTION)
5778 return -1;
5779 else if (x==enc_FAILED) {
5780 raise_encode_exception(exceptionObject, encoding, p, size, collstartpos, collendpos, reason);
5781 return -1;
5782 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00005783 }
5784 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00005785 *inpos = collendpos;
5786 break;
5787 default:
5788 repunicode = unicode_encode_call_errorhandler(errors, errorHandler,
Benjamin Peterson29060642009-01-31 22:14:21 +00005789 encoding, reason, p, size, exceptionObject,
5790 collstartpos, collendpos, &newpos);
Benjamin Peterson14339b62009-01-31 16:36:08 +00005791 if (repunicode == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00005792 return -1;
Martin v. Löwis011e8422009-05-05 04:43:17 +00005793 if (PyBytes_Check(repunicode)) {
5794 /* Directly copy bytes result to output. */
5795 Py_ssize_t outsize = PyBytes_Size(*res);
5796 Py_ssize_t requiredsize;
5797 repsize = PyBytes_Size(repunicode);
5798 requiredsize = *respos + repsize;
5799 if (requiredsize > outsize)
5800 /* Make room for all additional bytes. */
5801 if (charmapencode_resize(res, respos, requiredsize)) {
5802 Py_DECREF(repunicode);
5803 return -1;
5804 }
5805 memcpy(PyBytes_AsString(*res) + *respos,
5806 PyBytes_AsString(repunicode), repsize);
5807 *respos += repsize;
5808 *inpos = newpos;
Martin v. Löwisdb12d452009-05-02 18:52:14 +00005809 Py_DECREF(repunicode);
Martin v. Löwis011e8422009-05-05 04:43:17 +00005810 break;
Martin v. Löwisdb12d452009-05-02 18:52:14 +00005811 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00005812 /* generate replacement */
5813 repsize = PyUnicode_GET_SIZE(repunicode);
5814 for (uni2 = PyUnicode_AS_UNICODE(repunicode); repsize-->0; ++uni2) {
Benjamin Peterson29060642009-01-31 22:14:21 +00005815 x = charmapencode_output(*uni2, mapping, res, respos);
5816 if (x==enc_EXCEPTION) {
5817 return -1;
5818 }
5819 else if (x==enc_FAILED) {
5820 Py_DECREF(repunicode);
5821 raise_encode_exception(exceptionObject, encoding, p, size, collstartpos, collendpos, reason);
5822 return -1;
5823 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00005824 }
5825 *inpos = newpos;
5826 Py_DECREF(repunicode);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005827 }
5828 return 0;
5829}
5830
Guido van Rossumd57fd912000-03-10 22:53:23 +00005831PyObject *PyUnicode_EncodeCharmap(const Py_UNICODE *p,
Benjamin Peterson29060642009-01-31 22:14:21 +00005832 Py_ssize_t size,
5833 PyObject *mapping,
5834 const char *errors)
Guido van Rossumd57fd912000-03-10 22:53:23 +00005835{
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005836 /* output object */
5837 PyObject *res = NULL;
5838 /* current input position */
Martin v. Löwis18e16552006-02-15 17:27:45 +00005839 Py_ssize_t inpos = 0;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005840 /* current output position */
Martin v. Löwis18e16552006-02-15 17:27:45 +00005841 Py_ssize_t respos = 0;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005842 PyObject *errorHandler = NULL;
5843 PyObject *exc = NULL;
5844 /* the following variable is used for caching string comparisons
5845 * -1=not initialized, 0=unknown, 1=strict, 2=replace,
5846 * 3=ignore, 4=xmlcharrefreplace */
5847 int known_errorHandler = -1;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005848
5849 /* Default to Latin-1 */
5850 if (mapping == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00005851 return PyUnicode_EncodeLatin1(p, size, errors);
Guido van Rossumd57fd912000-03-10 22:53:23 +00005852
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005853 /* allocate enough for a simple encoding without
5854 replacements, if we need more, we'll resize */
Christian Heimes72b710a2008-05-26 13:28:38 +00005855 res = PyBytes_FromStringAndSize(NULL, size);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005856 if (res == NULL)
5857 goto onError;
Marc-André Lemburgb7520772000-08-14 11:29:19 +00005858 if (size == 0)
Benjamin Peterson29060642009-01-31 22:14:21 +00005859 return res;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005860
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005861 while (inpos<size) {
Benjamin Peterson29060642009-01-31 22:14:21 +00005862 /* try to encode it */
5863 charmapencode_result x = charmapencode_output(p[inpos], mapping, &res, &respos);
5864 if (x==enc_EXCEPTION) /* error */
5865 goto onError;
5866 if (x==enc_FAILED) { /* unencodable character */
5867 if (charmap_encoding_error(p, size, &inpos, mapping,
5868 &exc,
5869 &known_errorHandler, &errorHandler, errors,
5870 &res, &respos)) {
5871 goto onError;
5872 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00005873 }
Benjamin Peterson29060642009-01-31 22:14:21 +00005874 else
5875 /* done with this character => adjust input position */
5876 ++inpos;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005877 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00005878
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005879 /* Resize if we allocated to much */
Christian Heimes72b710a2008-05-26 13:28:38 +00005880 if (respos<PyBytes_GET_SIZE(res))
Alexandre Vassalotti44531cb2008-12-27 09:16:49 +00005881 if (_PyBytes_Resize(&res, respos) < 0)
5882 goto onError;
Guido van Rossum98297ee2007-11-06 21:34:58 +00005883
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005884 Py_XDECREF(exc);
5885 Py_XDECREF(errorHandler);
5886 return res;
5887
Benjamin Peterson29060642009-01-31 22:14:21 +00005888 onError:
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005889 Py_XDECREF(res);
5890 Py_XDECREF(exc);
5891 Py_XDECREF(errorHandler);
Guido van Rossumd57fd912000-03-10 22:53:23 +00005892 return NULL;
5893}
5894
5895PyObject *PyUnicode_AsCharmapString(PyObject *unicode,
Benjamin Peterson29060642009-01-31 22:14:21 +00005896 PyObject *mapping)
Guido van Rossumd57fd912000-03-10 22:53:23 +00005897{
5898 if (!PyUnicode_Check(unicode) || mapping == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +00005899 PyErr_BadArgument();
5900 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005901 }
5902 return PyUnicode_EncodeCharmap(PyUnicode_AS_UNICODE(unicode),
Benjamin Peterson29060642009-01-31 22:14:21 +00005903 PyUnicode_GET_SIZE(unicode),
5904 mapping,
5905 NULL);
Guido van Rossumd57fd912000-03-10 22:53:23 +00005906}
5907
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005908/* create or adjust a UnicodeTranslateError */
5909static void make_translate_exception(PyObject **exceptionObject,
Benjamin Peterson29060642009-01-31 22:14:21 +00005910 const Py_UNICODE *unicode, Py_ssize_t size,
5911 Py_ssize_t startpos, Py_ssize_t endpos,
5912 const char *reason)
Guido van Rossumd57fd912000-03-10 22:53:23 +00005913{
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005914 if (*exceptionObject == NULL) {
Benjamin Peterson14339b62009-01-31 16:36:08 +00005915 *exceptionObject = PyUnicodeTranslateError_Create(
Benjamin Peterson29060642009-01-31 22:14:21 +00005916 unicode, size, startpos, endpos, reason);
Guido van Rossumd57fd912000-03-10 22:53:23 +00005917 }
5918 else {
Benjamin Peterson29060642009-01-31 22:14:21 +00005919 if (PyUnicodeTranslateError_SetStart(*exceptionObject, startpos))
5920 goto onError;
5921 if (PyUnicodeTranslateError_SetEnd(*exceptionObject, endpos))
5922 goto onError;
5923 if (PyUnicodeTranslateError_SetReason(*exceptionObject, reason))
5924 goto onError;
5925 return;
5926 onError:
5927 Py_DECREF(*exceptionObject);
5928 *exceptionObject = NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00005929 }
5930}
5931
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005932/* raises a UnicodeTranslateError */
5933static void raise_translate_exception(PyObject **exceptionObject,
Benjamin Peterson29060642009-01-31 22:14:21 +00005934 const Py_UNICODE *unicode, Py_ssize_t size,
5935 Py_ssize_t startpos, Py_ssize_t endpos,
5936 const char *reason)
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005937{
5938 make_translate_exception(exceptionObject,
Benjamin Peterson29060642009-01-31 22:14:21 +00005939 unicode, size, startpos, endpos, reason);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005940 if (*exceptionObject != NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00005941 PyCodec_StrictErrors(*exceptionObject);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005942}
5943
5944/* error handling callback helper:
5945 build arguments, call the callback and check the arguments,
5946 put the result into newpos and return the replacement string, which
5947 has to be freed by the caller */
5948static PyObject *unicode_translate_call_errorhandler(const char *errors,
Benjamin Peterson29060642009-01-31 22:14:21 +00005949 PyObject **errorHandler,
5950 const char *reason,
5951 const Py_UNICODE *unicode, Py_ssize_t size, PyObject **exceptionObject,
5952 Py_ssize_t startpos, Py_ssize_t endpos,
5953 Py_ssize_t *newpos)
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005954{
Benjamin Peterson142957c2008-07-04 19:55:29 +00005955 static char *argparse = "O!n;translating error handler must return (str, int) tuple";
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005956
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00005957 Py_ssize_t i_newpos;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005958 PyObject *restuple;
5959 PyObject *resunicode;
5960
5961 if (*errorHandler == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +00005962 *errorHandler = PyCodec_LookupError(errors);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005963 if (*errorHandler == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00005964 return NULL;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005965 }
5966
5967 make_translate_exception(exceptionObject,
Benjamin Peterson29060642009-01-31 22:14:21 +00005968 unicode, size, startpos, endpos, reason);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005969 if (*exceptionObject == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00005970 return NULL;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005971
5972 restuple = PyObject_CallFunctionObjArgs(
Benjamin Peterson29060642009-01-31 22:14:21 +00005973 *errorHandler, *exceptionObject, NULL);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005974 if (restuple == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00005975 return NULL;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005976 if (!PyTuple_Check(restuple)) {
Benjamin Petersond75fcb42009-02-19 04:22:03 +00005977 PyErr_SetString(PyExc_TypeError, &argparse[4]);
Benjamin Peterson29060642009-01-31 22:14:21 +00005978 Py_DECREF(restuple);
5979 return NULL;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005980 }
5981 if (!PyArg_ParseTuple(restuple, argparse, &PyUnicode_Type,
Benjamin Peterson29060642009-01-31 22:14:21 +00005982 &resunicode, &i_newpos)) {
5983 Py_DECREF(restuple);
5984 return NULL;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005985 }
Martin v. Löwis18e16552006-02-15 17:27:45 +00005986 if (i_newpos<0)
Benjamin Peterson29060642009-01-31 22:14:21 +00005987 *newpos = size+i_newpos;
Martin v. Löwis18e16552006-02-15 17:27:45 +00005988 else
5989 *newpos = i_newpos;
Walter Dörwald2e0b18a2003-01-31 17:19:08 +00005990 if (*newpos<0 || *newpos>size) {
Benjamin Peterson29060642009-01-31 22:14:21 +00005991 PyErr_Format(PyExc_IndexError, "position %zd from error handler out of bounds", *newpos);
5992 Py_DECREF(restuple);
5993 return NULL;
Walter Dörwald2e0b18a2003-01-31 17:19:08 +00005994 }
Walter Dörwald3aeb6322002-09-02 13:14:32 +00005995 Py_INCREF(resunicode);
5996 Py_DECREF(restuple);
5997 return resunicode;
5998}
5999
6000/* Lookup the character ch in the mapping and put the result in result,
6001 which must be decrefed by the caller.
6002 Return 0 on success, -1 on error */
6003static
6004int charmaptranslate_lookup(Py_UNICODE c, PyObject *mapping, PyObject **result)
6005{
Christian Heimes217cfd12007-12-02 14:31:20 +00006006 PyObject *w = PyLong_FromLong((long)c);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006007 PyObject *x;
6008
6009 if (w == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00006010 return -1;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006011 x = PyObject_GetItem(mapping, w);
6012 Py_DECREF(w);
6013 if (x == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +00006014 if (PyErr_ExceptionMatches(PyExc_LookupError)) {
6015 /* No mapping found means: use 1:1 mapping. */
6016 PyErr_Clear();
6017 *result = NULL;
6018 return 0;
6019 } else
6020 return -1;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006021 }
6022 else if (x == Py_None) {
Benjamin Peterson29060642009-01-31 22:14:21 +00006023 *result = x;
6024 return 0;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006025 }
Christian Heimes217cfd12007-12-02 14:31:20 +00006026 else if (PyLong_Check(x)) {
Benjamin Peterson29060642009-01-31 22:14:21 +00006027 long value = PyLong_AS_LONG(x);
6028 long max = PyUnicode_GetMax();
6029 if (value < 0 || value > max) {
6030 PyErr_Format(PyExc_TypeError,
Guido van Rossum5a2f7e602007-10-24 21:13:09 +00006031 "character mapping must be in range(0x%x)", max+1);
Benjamin Peterson29060642009-01-31 22:14:21 +00006032 Py_DECREF(x);
6033 return -1;
6034 }
6035 *result = x;
6036 return 0;
6037 }
6038 else if (PyUnicode_Check(x)) {
6039 *result = x;
6040 return 0;
6041 }
6042 else {
6043 /* wrong return value */
6044 PyErr_SetString(PyExc_TypeError,
6045 "character mapping must return integer, None or str");
Benjamin Peterson14339b62009-01-31 16:36:08 +00006046 Py_DECREF(x);
6047 return -1;
6048 }
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006049}
6050/* ensure that *outobj is at least requiredsize characters long,
Benjamin Peterson29060642009-01-31 22:14:21 +00006051 if not reallocate and adjust various state variables.
6052 Return 0 on success, -1 on error */
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006053static
Walter Dörwald4894c302003-10-24 14:25:28 +00006054int charmaptranslate_makespace(PyObject **outobj, Py_UNICODE **outp,
Benjamin Peterson29060642009-01-31 22:14:21 +00006055 Py_ssize_t requiredsize)
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006056{
Martin v. Löwis18e16552006-02-15 17:27:45 +00006057 Py_ssize_t oldsize = PyUnicode_GET_SIZE(*outobj);
Walter Dörwald4894c302003-10-24 14:25:28 +00006058 if (requiredsize > oldsize) {
Benjamin Peterson29060642009-01-31 22:14:21 +00006059 /* remember old output position */
6060 Py_ssize_t outpos = *outp-PyUnicode_AS_UNICODE(*outobj);
6061 /* exponentially overallocate to minimize reallocations */
6062 if (requiredsize < 2 * oldsize)
6063 requiredsize = 2 * oldsize;
6064 if (PyUnicode_Resize(outobj, requiredsize) < 0)
6065 return -1;
6066 *outp = PyUnicode_AS_UNICODE(*outobj) + outpos;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006067 }
6068 return 0;
6069}
6070/* lookup the character, put the result in the output string and adjust
6071 various state variables. Return a new reference to the object that
6072 was put in the output buffer in *result, or Py_None, if the mapping was
6073 undefined (in which case no character was written).
6074 The called must decref result.
6075 Return 0 on success, -1 on error. */
6076static
Walter Dörwald4894c302003-10-24 14:25:28 +00006077int charmaptranslate_output(const Py_UNICODE *startinp, const Py_UNICODE *curinp,
Benjamin Peterson29060642009-01-31 22:14:21 +00006078 Py_ssize_t insize, PyObject *mapping, PyObject **outobj, Py_UNICODE **outp,
6079 PyObject **res)
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006080{
Walter Dörwald4894c302003-10-24 14:25:28 +00006081 if (charmaptranslate_lookup(*curinp, mapping, res))
Benjamin Peterson29060642009-01-31 22:14:21 +00006082 return -1;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006083 if (*res==NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +00006084 /* not found => default to 1:1 mapping */
6085 *(*outp)++ = *curinp;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006086 }
6087 else if (*res==Py_None)
Benjamin Peterson29060642009-01-31 22:14:21 +00006088 ;
Christian Heimes217cfd12007-12-02 14:31:20 +00006089 else if (PyLong_Check(*res)) {
Benjamin Peterson29060642009-01-31 22:14:21 +00006090 /* no overflow check, because we know that the space is enough */
6091 *(*outp)++ = (Py_UNICODE)PyLong_AS_LONG(*res);
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006092 }
6093 else if (PyUnicode_Check(*res)) {
Benjamin Peterson29060642009-01-31 22:14:21 +00006094 Py_ssize_t repsize = PyUnicode_GET_SIZE(*res);
6095 if (repsize==1) {
6096 /* no overflow check, because we know that the space is enough */
6097 *(*outp)++ = *PyUnicode_AS_UNICODE(*res);
6098 }
6099 else if (repsize!=0) {
6100 /* more than one character */
6101 Py_ssize_t requiredsize = (*outp-PyUnicode_AS_UNICODE(*outobj)) +
6102 (insize - (curinp-startinp)) +
6103 repsize - 1;
6104 if (charmaptranslate_makespace(outobj, outp, requiredsize))
6105 return -1;
6106 memcpy(*outp, PyUnicode_AS_UNICODE(*res), sizeof(Py_UNICODE)*repsize);
6107 *outp += repsize;
6108 }
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006109 }
6110 else
Benjamin Peterson29060642009-01-31 22:14:21 +00006111 return -1;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006112 return 0;
6113}
6114
6115PyObject *PyUnicode_TranslateCharmap(const Py_UNICODE *p,
Benjamin Peterson29060642009-01-31 22:14:21 +00006116 Py_ssize_t size,
6117 PyObject *mapping,
6118 const char *errors)
Guido van Rossumd57fd912000-03-10 22:53:23 +00006119{
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006120 /* output object */
6121 PyObject *res = NULL;
6122 /* pointers to the beginning and end+1 of input */
6123 const Py_UNICODE *startp = p;
6124 const Py_UNICODE *endp = p + size;
6125 /* pointer into the output */
6126 Py_UNICODE *str;
6127 /* current output position */
Martin v. Löwis18e16552006-02-15 17:27:45 +00006128 Py_ssize_t respos = 0;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006129 char *reason = "character maps to <undefined>";
6130 PyObject *errorHandler = NULL;
6131 PyObject *exc = NULL;
6132 /* the following variable is used for caching string comparisons
6133 * -1=not initialized, 0=unknown, 1=strict, 2=replace,
6134 * 3=ignore, 4=xmlcharrefreplace */
6135 int known_errorHandler = -1;
6136
Guido van Rossumd57fd912000-03-10 22:53:23 +00006137 if (mapping == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +00006138 PyErr_BadArgument();
6139 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00006140 }
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006141
6142 /* allocate enough for a simple 1:1 translation without
6143 replacements, if we need more, we'll resize */
6144 res = PyUnicode_FromUnicode(NULL, size);
6145 if (res == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00006146 goto onError;
Guido van Rossumd57fd912000-03-10 22:53:23 +00006147 if (size == 0)
Benjamin Peterson29060642009-01-31 22:14:21 +00006148 return res;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006149 str = PyUnicode_AS_UNICODE(res);
Guido van Rossumd57fd912000-03-10 22:53:23 +00006150
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006151 while (p<endp) {
Benjamin Peterson29060642009-01-31 22:14:21 +00006152 /* try to encode it */
6153 PyObject *x = NULL;
6154 if (charmaptranslate_output(startp, p, size, mapping, &res, &str, &x)) {
6155 Py_XDECREF(x);
6156 goto onError;
6157 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00006158 Py_XDECREF(x);
Benjamin Peterson29060642009-01-31 22:14:21 +00006159 if (x!=Py_None) /* it worked => adjust input pointer */
6160 ++p;
6161 else { /* untranslatable character */
6162 PyObject *repunicode = NULL; /* initialize to prevent gcc warning */
6163 Py_ssize_t repsize;
6164 Py_ssize_t newpos;
6165 Py_UNICODE *uni2;
6166 /* startpos for collecting untranslatable chars */
6167 const Py_UNICODE *collstart = p;
6168 const Py_UNICODE *collend = p+1;
6169 const Py_UNICODE *coll;
Guido van Rossumd57fd912000-03-10 22:53:23 +00006170
Benjamin Peterson29060642009-01-31 22:14:21 +00006171 /* find all untranslatable characters */
6172 while (collend < endp) {
6173 if (charmaptranslate_lookup(*collend, mapping, &x))
6174 goto onError;
6175 Py_XDECREF(x);
6176 if (x!=Py_None)
6177 break;
6178 ++collend;
6179 }
6180 /* cache callback name lookup
6181 * (if not done yet, i.e. it's the first error) */
6182 if (known_errorHandler==-1) {
6183 if ((errors==NULL) || (!strcmp(errors, "strict")))
6184 known_errorHandler = 1;
6185 else if (!strcmp(errors, "replace"))
6186 known_errorHandler = 2;
6187 else if (!strcmp(errors, "ignore"))
6188 known_errorHandler = 3;
6189 else if (!strcmp(errors, "xmlcharrefreplace"))
6190 known_errorHandler = 4;
6191 else
6192 known_errorHandler = 0;
6193 }
6194 switch (known_errorHandler) {
6195 case 1: /* strict */
6196 raise_translate_exception(&exc, startp, size, collstart-startp, collend-startp, reason);
Benjamin Peterson14339b62009-01-31 16:36:08 +00006197 goto onError;
Benjamin Peterson29060642009-01-31 22:14:21 +00006198 case 2: /* replace */
6199 /* No need to check for space, this is a 1:1 replacement */
6200 for (coll = collstart; coll<collend; ++coll)
6201 *str++ = '?';
6202 /* fall through */
6203 case 3: /* ignore */
6204 p = collend;
6205 break;
6206 case 4: /* xmlcharrefreplace */
6207 /* generate replacement (temporarily (mis)uses p) */
6208 for (p = collstart; p < collend; ++p) {
6209 char buffer[2+29+1+1];
6210 char *cp;
6211 sprintf(buffer, "&#%d;", (int)*p);
6212 if (charmaptranslate_makespace(&res, &str,
6213 (str-PyUnicode_AS_UNICODE(res))+strlen(buffer)+(endp-collend)))
6214 goto onError;
6215 for (cp = buffer; *cp; ++cp)
6216 *str++ = *cp;
6217 }
6218 p = collend;
6219 break;
6220 default:
6221 repunicode = unicode_translate_call_errorhandler(errors, &errorHandler,
6222 reason, startp, size, &exc,
6223 collstart-startp, collend-startp, &newpos);
6224 if (repunicode == NULL)
6225 goto onError;
6226 /* generate replacement */
6227 repsize = PyUnicode_GET_SIZE(repunicode);
6228 if (charmaptranslate_makespace(&res, &str,
6229 (str-PyUnicode_AS_UNICODE(res))+repsize+(endp-collend))) {
6230 Py_DECREF(repunicode);
6231 goto onError;
6232 }
6233 for (uni2 = PyUnicode_AS_UNICODE(repunicode); repsize-->0; ++uni2)
6234 *str++ = *uni2;
6235 p = startp + newpos;
6236 Py_DECREF(repunicode);
Benjamin Peterson14339b62009-01-31 16:36:08 +00006237 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00006238 }
6239 }
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006240 /* Resize if we allocated to much */
6241 respos = str-PyUnicode_AS_UNICODE(res);
Walter Dörwald4894c302003-10-24 14:25:28 +00006242 if (respos<PyUnicode_GET_SIZE(res)) {
Benjamin Peterson29060642009-01-31 22:14:21 +00006243 if (PyUnicode_Resize(&res, respos) < 0)
6244 goto onError;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006245 }
6246 Py_XDECREF(exc);
6247 Py_XDECREF(errorHandler);
6248 return res;
Guido van Rossumd57fd912000-03-10 22:53:23 +00006249
Benjamin Peterson29060642009-01-31 22:14:21 +00006250 onError:
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006251 Py_XDECREF(res);
6252 Py_XDECREF(exc);
6253 Py_XDECREF(errorHandler);
Guido van Rossumd57fd912000-03-10 22:53:23 +00006254 return NULL;
6255}
6256
6257PyObject *PyUnicode_Translate(PyObject *str,
Benjamin Peterson29060642009-01-31 22:14:21 +00006258 PyObject *mapping,
6259 const char *errors)
Guido van Rossumd57fd912000-03-10 22:53:23 +00006260{
6261 PyObject *result;
Tim Petersced69f82003-09-16 20:30:58 +00006262
Guido van Rossumd57fd912000-03-10 22:53:23 +00006263 str = PyUnicode_FromObject(str);
6264 if (str == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00006265 goto onError;
Guido van Rossumd57fd912000-03-10 22:53:23 +00006266 result = PyUnicode_TranslateCharmap(PyUnicode_AS_UNICODE(str),
Benjamin Peterson29060642009-01-31 22:14:21 +00006267 PyUnicode_GET_SIZE(str),
6268 mapping,
6269 errors);
Guido van Rossumd57fd912000-03-10 22:53:23 +00006270 Py_DECREF(str);
6271 return result;
Tim Petersced69f82003-09-16 20:30:58 +00006272
Benjamin Peterson29060642009-01-31 22:14:21 +00006273 onError:
Guido van Rossumd57fd912000-03-10 22:53:23 +00006274 Py_XDECREF(str);
6275 return NULL;
6276}
Tim Petersced69f82003-09-16 20:30:58 +00006277
Alexander Belopolsky942af5a2010-12-04 03:38:46 +00006278PyObject *
6279PyUnicode_TransformDecimalToASCII(Py_UNICODE *s,
6280 Py_ssize_t length)
6281{
6282 PyObject *result;
6283 Py_UNICODE *p; /* write pointer into result */
6284 Py_ssize_t i;
6285 /* Copy to a new string */
6286 result = (PyObject *)_PyUnicode_New(length);
6287 Py_UNICODE_COPY(PyUnicode_AS_UNICODE(result), s, length);
6288 if (result == NULL)
6289 return result;
6290 p = PyUnicode_AS_UNICODE(result);
6291 /* Iterate over code points */
6292 for (i = 0; i < length; i++) {
6293 Py_UNICODE ch =s[i];
6294 if (ch > 127) {
6295 int decimal = Py_UNICODE_TODECIMAL(ch);
6296 if (decimal >= 0)
6297 p[i] = '0' + decimal;
6298 }
6299 }
6300 return result;
6301}
Guido van Rossum9e896b32000-04-05 20:11:21 +00006302/* --- Decimal Encoder ---------------------------------------------------- */
6303
6304int PyUnicode_EncodeDecimal(Py_UNICODE *s,
Benjamin Peterson29060642009-01-31 22:14:21 +00006305 Py_ssize_t length,
6306 char *output,
6307 const char *errors)
Guido van Rossum9e896b32000-04-05 20:11:21 +00006308{
6309 Py_UNICODE *p, *end;
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006310 PyObject *errorHandler = NULL;
6311 PyObject *exc = NULL;
6312 const char *encoding = "decimal";
6313 const char *reason = "invalid decimal Unicode string";
6314 /* the following variable is used for caching string comparisons
6315 * -1=not initialized, 0=unknown, 1=strict, 2=replace, 3=ignore, 4=xmlcharrefreplace */
6316 int known_errorHandler = -1;
Guido van Rossum9e896b32000-04-05 20:11:21 +00006317
6318 if (output == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +00006319 PyErr_BadArgument();
6320 return -1;
Guido van Rossum9e896b32000-04-05 20:11:21 +00006321 }
6322
6323 p = s;
6324 end = s + length;
6325 while (p < end) {
Benjamin Peterson29060642009-01-31 22:14:21 +00006326 register Py_UNICODE ch = *p;
6327 int decimal;
6328 PyObject *repunicode;
6329 Py_ssize_t repsize;
6330 Py_ssize_t newpos;
6331 Py_UNICODE *uni2;
6332 Py_UNICODE *collstart;
6333 Py_UNICODE *collend;
Tim Petersced69f82003-09-16 20:30:58 +00006334
Benjamin Peterson29060642009-01-31 22:14:21 +00006335 if (Py_UNICODE_ISSPACE(ch)) {
Benjamin Peterson14339b62009-01-31 16:36:08 +00006336 *output++ = ' ';
Benjamin Peterson29060642009-01-31 22:14:21 +00006337 ++p;
6338 continue;
Benjamin Peterson14339b62009-01-31 16:36:08 +00006339 }
Benjamin Peterson29060642009-01-31 22:14:21 +00006340 decimal = Py_UNICODE_TODECIMAL(ch);
6341 if (decimal >= 0) {
6342 *output++ = '0' + decimal;
6343 ++p;
6344 continue;
6345 }
6346 if (0 < ch && ch < 256) {
6347 *output++ = (char)ch;
6348 ++p;
6349 continue;
6350 }
6351 /* All other characters are considered unencodable */
6352 collstart = p;
Victor Stinnerab1d16b2011-11-22 01:45:37 +01006353 for (collend = p+1; collend < end; collend++) {
Benjamin Peterson29060642009-01-31 22:14:21 +00006354 if ((0 < *collend && *collend < 256) ||
Victor Stinnerab1d16b2011-11-22 01:45:37 +01006355 Py_UNICODE_ISSPACE(*collend) ||
6356 0 <= Py_UNICODE_TODECIMAL(*collend))
Benjamin Peterson29060642009-01-31 22:14:21 +00006357 break;
6358 }
6359 /* cache callback name lookup
6360 * (if not done yet, i.e. it's the first error) */
6361 if (known_errorHandler==-1) {
6362 if ((errors==NULL) || (!strcmp(errors, "strict")))
6363 known_errorHandler = 1;
6364 else if (!strcmp(errors, "replace"))
6365 known_errorHandler = 2;
6366 else if (!strcmp(errors, "ignore"))
6367 known_errorHandler = 3;
6368 else if (!strcmp(errors, "xmlcharrefreplace"))
6369 known_errorHandler = 4;
6370 else
6371 known_errorHandler = 0;
6372 }
6373 switch (known_errorHandler) {
6374 case 1: /* strict */
6375 raise_encode_exception(&exc, encoding, s, length, collstart-s, collend-s, reason);
6376 goto onError;
6377 case 2: /* replace */
6378 for (p = collstart; p < collend; ++p)
6379 *output++ = '?';
6380 /* fall through */
6381 case 3: /* ignore */
6382 p = collend;
6383 break;
6384 case 4: /* xmlcharrefreplace */
6385 /* generate replacement (temporarily (mis)uses p) */
6386 for (p = collstart; p < collend; ++p)
6387 output += sprintf(output, "&#%d;", (int)*p);
6388 p = collend;
6389 break;
6390 default:
6391 repunicode = unicode_encode_call_errorhandler(errors, &errorHandler,
6392 encoding, reason, s, length, &exc,
6393 collstart-s, collend-s, &newpos);
6394 if (repunicode == NULL)
6395 goto onError;
Martin v. Löwisdb12d452009-05-02 18:52:14 +00006396 if (!PyUnicode_Check(repunicode)) {
Martin v. Löwis011e8422009-05-05 04:43:17 +00006397 /* Byte results not supported, since they have no decimal property. */
Martin v. Löwisdb12d452009-05-02 18:52:14 +00006398 PyErr_SetString(PyExc_TypeError, "error handler should return unicode");
6399 Py_DECREF(repunicode);
6400 goto onError;
6401 }
Benjamin Peterson29060642009-01-31 22:14:21 +00006402 /* generate replacement */
6403 repsize = PyUnicode_GET_SIZE(repunicode);
6404 for (uni2 = PyUnicode_AS_UNICODE(repunicode); repsize-->0; ++uni2) {
6405 Py_UNICODE ch = *uni2;
6406 if (Py_UNICODE_ISSPACE(ch))
6407 *output++ = ' ';
6408 else {
6409 decimal = Py_UNICODE_TODECIMAL(ch);
6410 if (decimal >= 0)
6411 *output++ = '0' + decimal;
6412 else if (0 < ch && ch < 256)
6413 *output++ = (char)ch;
6414 else {
6415 Py_DECREF(repunicode);
6416 raise_encode_exception(&exc, encoding,
6417 s, length, collstart-s, collend-s, reason);
6418 goto onError;
6419 }
6420 }
6421 }
6422 p = s + newpos;
6423 Py_DECREF(repunicode);
6424 }
Guido van Rossum9e896b32000-04-05 20:11:21 +00006425 }
6426 /* 0-terminate the output string */
6427 *output++ = '\0';
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006428 Py_XDECREF(exc);
6429 Py_XDECREF(errorHandler);
Guido van Rossum9e896b32000-04-05 20:11:21 +00006430 return 0;
6431
Benjamin Peterson29060642009-01-31 22:14:21 +00006432 onError:
Walter Dörwald3aeb6322002-09-02 13:14:32 +00006433 Py_XDECREF(exc);
6434 Py_XDECREF(errorHandler);
Guido van Rossum9e896b32000-04-05 20:11:21 +00006435 return -1;
6436}
6437
Guido van Rossumd57fd912000-03-10 22:53:23 +00006438/* --- Helpers ------------------------------------------------------------ */
6439
Eric Smith8c663262007-08-25 02:26:07 +00006440#include "stringlib/unicodedefs.h"
Thomas Wouters477c8d52006-05-27 19:21:47 +00006441#include "stringlib/fastsearch.h"
Antoine Pitrouf2c54842010-01-13 08:07:53 +00006442
Thomas Wouters477c8d52006-05-27 19:21:47 +00006443#include "stringlib/count.h"
6444#include "stringlib/find.h"
6445#include "stringlib/partition.h"
Antoine Pitrouf2c54842010-01-13 08:07:53 +00006446#include "stringlib/split.h"
Thomas Wouters477c8d52006-05-27 19:21:47 +00006447
Eric Smith5807c412008-05-11 21:00:57 +00006448#define _Py_InsertThousandsGrouping _PyUnicode_InsertThousandsGrouping
Eric Smitha3b1ac82009-04-03 14:45:06 +00006449#define _Py_InsertThousandsGroupingLocale _PyUnicode_InsertThousandsGroupingLocale
Eric Smith5807c412008-05-11 21:00:57 +00006450#include "stringlib/localeutil.h"
6451
Thomas Wouters477c8d52006-05-27 19:21:47 +00006452/* helper macro to fixup start/end slice values */
Antoine Pitrouf2c54842010-01-13 08:07:53 +00006453#define ADJUST_INDICES(start, end, len) \
6454 if (end > len) \
6455 end = len; \
6456 else if (end < 0) { \
6457 end += len; \
6458 if (end < 0) \
6459 end = 0; \
6460 } \
6461 if (start < 0) { \
6462 start += len; \
6463 if (start < 0) \
6464 start = 0; \
6465 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00006466
Ezio Melotti93e7afc2011-08-22 14:08:38 +03006467/* _Py_UNICODE_NEXT is a private macro used to retrieve the character pointed
6468 * by 'ptr', possibly combining surrogate pairs on narrow builds.
6469 * 'ptr' and 'end' must be Py_UNICODE*, with 'ptr' pointing at the character
6470 * that should be returned and 'end' pointing to the end of the buffer.
6471 * ('end' is used on narrow builds to detect a lone surrogate at the
6472 * end of the buffer that should be returned unchanged.)
6473 * The ptr and end arguments should be side-effect free and ptr must an lvalue.
6474 * The type of the returned char is always Py_UCS4.
6475 *
6476 * Note: the macro advances ptr to next char, so it might have side-effects
6477 * (especially if used with other macros).
6478 */
6479
6480/* helper macros used by _Py_UNICODE_NEXT */
6481#define _Py_UNICODE_IS_HIGH_SURROGATE(ch) (0xD800 <= ch && ch <= 0xDBFF)
6482#define _Py_UNICODE_IS_LOW_SURROGATE(ch) (0xDC00 <= ch && ch <= 0xDFFF)
6483/* Join two surrogate characters and return a single Py_UCS4 value. */
6484#define _Py_UNICODE_JOIN_SURROGATES(high, low) \
6485 (((((Py_UCS4)(high) & 0x03FF) << 10) | \
6486 ((Py_UCS4)(low) & 0x03FF)) + 0x10000)
6487
6488#ifdef Py_UNICODE_WIDE
6489#define _Py_UNICODE_NEXT(ptr, end) *(ptr)++
6490#else
6491#define _Py_UNICODE_NEXT(ptr, end) \
6492 (((_Py_UNICODE_IS_HIGH_SURROGATE(*(ptr)) && (ptr) < (end)) && \
6493 _Py_UNICODE_IS_LOW_SURROGATE((ptr)[1])) ? \
6494 ((ptr) += 2,_Py_UNICODE_JOIN_SURROGATES((ptr)[-2], (ptr)[-1])) : \
6495 (Py_UCS4)*(ptr)++)
6496#endif
6497
Martin v. Löwis18e16552006-02-15 17:27:45 +00006498Py_ssize_t PyUnicode_Count(PyObject *str,
Thomas Wouters477c8d52006-05-27 19:21:47 +00006499 PyObject *substr,
6500 Py_ssize_t start,
6501 Py_ssize_t end)
Guido van Rossumd57fd912000-03-10 22:53:23 +00006502{
Martin v. Löwis18e16552006-02-15 17:27:45 +00006503 Py_ssize_t result;
Thomas Wouters477c8d52006-05-27 19:21:47 +00006504 PyUnicodeObject* str_obj;
6505 PyUnicodeObject* sub_obj;
Tim Petersced69f82003-09-16 20:30:58 +00006506
Thomas Wouters477c8d52006-05-27 19:21:47 +00006507 str_obj = (PyUnicodeObject*) PyUnicode_FromObject(str);
6508 if (!str_obj)
Benjamin Peterson29060642009-01-31 22:14:21 +00006509 return -1;
Thomas Wouters477c8d52006-05-27 19:21:47 +00006510 sub_obj = (PyUnicodeObject*) PyUnicode_FromObject(substr);
6511 if (!sub_obj) {
Benjamin Peterson29060642009-01-31 22:14:21 +00006512 Py_DECREF(str_obj);
6513 return -1;
Guido van Rossumd57fd912000-03-10 22:53:23 +00006514 }
Tim Petersced69f82003-09-16 20:30:58 +00006515
Antoine Pitrouf2c54842010-01-13 08:07:53 +00006516 ADJUST_INDICES(start, end, str_obj->length);
Thomas Wouters477c8d52006-05-27 19:21:47 +00006517 result = stringlib_count(
Antoine Pitrouf2c54842010-01-13 08:07:53 +00006518 str_obj->str + start, end - start, sub_obj->str, sub_obj->length,
6519 PY_SSIZE_T_MAX
Thomas Wouters477c8d52006-05-27 19:21:47 +00006520 );
6521
6522 Py_DECREF(sub_obj);
6523 Py_DECREF(str_obj);
6524
Guido van Rossumd57fd912000-03-10 22:53:23 +00006525 return result;
6526}
6527
Martin v. Löwis18e16552006-02-15 17:27:45 +00006528Py_ssize_t PyUnicode_Find(PyObject *str,
Thomas Wouters477c8d52006-05-27 19:21:47 +00006529 PyObject *sub,
6530 Py_ssize_t start,
6531 Py_ssize_t end,
6532 int direction)
Guido van Rossumd57fd912000-03-10 22:53:23 +00006533{
Martin v. Löwis18e16552006-02-15 17:27:45 +00006534 Py_ssize_t result;
Tim Petersced69f82003-09-16 20:30:58 +00006535
Guido van Rossumd57fd912000-03-10 22:53:23 +00006536 str = PyUnicode_FromObject(str);
Thomas Wouters477c8d52006-05-27 19:21:47 +00006537 if (!str)
Benjamin Peterson29060642009-01-31 22:14:21 +00006538 return -2;
Thomas Wouters477c8d52006-05-27 19:21:47 +00006539 sub = PyUnicode_FromObject(sub);
6540 if (!sub) {
Benjamin Peterson29060642009-01-31 22:14:21 +00006541 Py_DECREF(str);
6542 return -2;
Guido van Rossumd57fd912000-03-10 22:53:23 +00006543 }
Tim Petersced69f82003-09-16 20:30:58 +00006544
Thomas Wouters477c8d52006-05-27 19:21:47 +00006545 if (direction > 0)
6546 result = stringlib_find_slice(
6547 PyUnicode_AS_UNICODE(str), PyUnicode_GET_SIZE(str),
6548 PyUnicode_AS_UNICODE(sub), PyUnicode_GET_SIZE(sub),
6549 start, end
6550 );
6551 else
6552 result = stringlib_rfind_slice(
6553 PyUnicode_AS_UNICODE(str), PyUnicode_GET_SIZE(str),
6554 PyUnicode_AS_UNICODE(sub), PyUnicode_GET_SIZE(sub),
6555 start, end
6556 );
6557
Guido van Rossumd57fd912000-03-10 22:53:23 +00006558 Py_DECREF(str);
Thomas Wouters477c8d52006-05-27 19:21:47 +00006559 Py_DECREF(sub);
6560
Guido van Rossumd57fd912000-03-10 22:53:23 +00006561 return result;
6562}
6563
Tim Petersced69f82003-09-16 20:30:58 +00006564static
Guido van Rossumd57fd912000-03-10 22:53:23 +00006565int tailmatch(PyUnicodeObject *self,
Benjamin Peterson29060642009-01-31 22:14:21 +00006566 PyUnicodeObject *substring,
6567 Py_ssize_t start,
6568 Py_ssize_t end,
6569 int direction)
Guido van Rossumd57fd912000-03-10 22:53:23 +00006570{
Guido van Rossumd57fd912000-03-10 22:53:23 +00006571 if (substring->length == 0)
6572 return 1;
6573
Antoine Pitrouf2c54842010-01-13 08:07:53 +00006574 ADJUST_INDICES(start, end, self->length);
Guido van Rossumd57fd912000-03-10 22:53:23 +00006575 end -= substring->length;
6576 if (end < start)
Benjamin Peterson29060642009-01-31 22:14:21 +00006577 return 0;
Guido van Rossumd57fd912000-03-10 22:53:23 +00006578
6579 if (direction > 0) {
Benjamin Peterson29060642009-01-31 22:14:21 +00006580 if (Py_UNICODE_MATCH(self, end, substring))
6581 return 1;
Guido van Rossumd57fd912000-03-10 22:53:23 +00006582 } else {
6583 if (Py_UNICODE_MATCH(self, start, substring))
Benjamin Peterson29060642009-01-31 22:14:21 +00006584 return 1;
Guido van Rossumd57fd912000-03-10 22:53:23 +00006585 }
6586
6587 return 0;
6588}
6589
Martin v. Löwis18e16552006-02-15 17:27:45 +00006590Py_ssize_t PyUnicode_Tailmatch(PyObject *str,
Benjamin Peterson29060642009-01-31 22:14:21 +00006591 PyObject *substr,
6592 Py_ssize_t start,
6593 Py_ssize_t end,
6594 int direction)
Guido van Rossumd57fd912000-03-10 22:53:23 +00006595{
Martin v. Löwis18e16552006-02-15 17:27:45 +00006596 Py_ssize_t result;
Tim Petersced69f82003-09-16 20:30:58 +00006597
Guido van Rossumd57fd912000-03-10 22:53:23 +00006598 str = PyUnicode_FromObject(str);
6599 if (str == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00006600 return -1;
Guido van Rossumd57fd912000-03-10 22:53:23 +00006601 substr = PyUnicode_FromObject(substr);
6602 if (substr == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +00006603 Py_DECREF(str);
6604 return -1;
Guido van Rossumd57fd912000-03-10 22:53:23 +00006605 }
Tim Petersced69f82003-09-16 20:30:58 +00006606
Guido van Rossumd57fd912000-03-10 22:53:23 +00006607 result = tailmatch((PyUnicodeObject *)str,
Benjamin Peterson29060642009-01-31 22:14:21 +00006608 (PyUnicodeObject *)substr,
6609 start, end, direction);
Guido van Rossumd57fd912000-03-10 22:53:23 +00006610 Py_DECREF(str);
6611 Py_DECREF(substr);
6612 return result;
6613}
6614
Guido van Rossumd57fd912000-03-10 22:53:23 +00006615/* Apply fixfct filter to the Unicode object self and return a
6616 reference to the modified object */
6617
Tim Petersced69f82003-09-16 20:30:58 +00006618static
Guido van Rossumd57fd912000-03-10 22:53:23 +00006619PyObject *fixup(PyUnicodeObject *self,
Benjamin Peterson29060642009-01-31 22:14:21 +00006620 int (*fixfct)(PyUnicodeObject *s))
Guido van Rossumd57fd912000-03-10 22:53:23 +00006621{
6622
6623 PyUnicodeObject *u;
6624
Marc-André Lemburg8155e0e2001-04-23 14:44:21 +00006625 u = (PyUnicodeObject*) PyUnicode_FromUnicode(NULL, self->length);
Guido van Rossumd57fd912000-03-10 22:53:23 +00006626 if (u == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00006627 return NULL;
Marc-André Lemburg8155e0e2001-04-23 14:44:21 +00006628
6629 Py_UNICODE_COPY(u->str, self->str, self->length);
6630
Tim Peters7a29bd52001-09-12 03:03:31 +00006631 if (!fixfct(u) && PyUnicode_CheckExact(self)) {
Benjamin Peterson29060642009-01-31 22:14:21 +00006632 /* fixfct should return TRUE if it modified the buffer. If
6633 FALSE, return a reference to the original buffer instead
6634 (to save space, not time) */
6635 Py_INCREF(self);
6636 Py_DECREF(u);
6637 return (PyObject*) self;
Guido van Rossumd57fd912000-03-10 22:53:23 +00006638 }
6639 return (PyObject*) u;
6640}
6641
Tim Petersced69f82003-09-16 20:30:58 +00006642static
Guido van Rossumd57fd912000-03-10 22:53:23 +00006643int fixupper(PyUnicodeObject *self)
6644{
Martin v. Löwis18e16552006-02-15 17:27:45 +00006645 Py_ssize_t len = self->length;
Guido van Rossumd57fd912000-03-10 22:53:23 +00006646 Py_UNICODE *s = self->str;
6647 int status = 0;
Tim Petersced69f82003-09-16 20:30:58 +00006648
Guido van Rossumd57fd912000-03-10 22:53:23 +00006649 while (len-- > 0) {
Benjamin Peterson29060642009-01-31 22:14:21 +00006650 register Py_UNICODE ch;
Tim Petersced69f82003-09-16 20:30:58 +00006651
Benjamin Peterson29060642009-01-31 22:14:21 +00006652 ch = Py_UNICODE_TOUPPER(*s);
6653 if (ch != *s) {
Guido van Rossumd57fd912000-03-10 22:53:23 +00006654 status = 1;
Benjamin Peterson29060642009-01-31 22:14:21 +00006655 *s = ch;
6656 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00006657 s++;
6658 }
6659
6660 return status;
6661}
6662
Tim Petersced69f82003-09-16 20:30:58 +00006663static
Guido van Rossumd57fd912000-03-10 22:53:23 +00006664int fixlower(PyUnicodeObject *self)
6665{
Martin v. Löwis18e16552006-02-15 17:27:45 +00006666 Py_ssize_t len = self->length;
Guido van Rossumd57fd912000-03-10 22:53:23 +00006667 Py_UNICODE *s = self->str;
6668 int status = 0;
Tim Petersced69f82003-09-16 20:30:58 +00006669
Guido van Rossumd57fd912000-03-10 22:53:23 +00006670 while (len-- > 0) {
Benjamin Peterson29060642009-01-31 22:14:21 +00006671 register Py_UNICODE ch;
Tim Petersced69f82003-09-16 20:30:58 +00006672
Benjamin Peterson29060642009-01-31 22:14:21 +00006673 ch = Py_UNICODE_TOLOWER(*s);
6674 if (ch != *s) {
Guido van Rossumd57fd912000-03-10 22:53:23 +00006675 status = 1;
Benjamin Peterson29060642009-01-31 22:14:21 +00006676 *s = ch;
6677 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00006678 s++;
6679 }
6680
6681 return status;
6682}
6683
Tim Petersced69f82003-09-16 20:30:58 +00006684static
Guido van Rossumd57fd912000-03-10 22:53:23 +00006685int fixswapcase(PyUnicodeObject *self)
6686{
Martin v. Löwis18e16552006-02-15 17:27:45 +00006687 Py_ssize_t len = self->length;
Guido van Rossumd57fd912000-03-10 22:53:23 +00006688 Py_UNICODE *s = self->str;
6689 int status = 0;
Tim Petersced69f82003-09-16 20:30:58 +00006690
Guido van Rossumd57fd912000-03-10 22:53:23 +00006691 while (len-- > 0) {
6692 if (Py_UNICODE_ISUPPER(*s)) {
6693 *s = Py_UNICODE_TOLOWER(*s);
6694 status = 1;
6695 } else if (Py_UNICODE_ISLOWER(*s)) {
6696 *s = Py_UNICODE_TOUPPER(*s);
6697 status = 1;
6698 }
6699 s++;
6700 }
6701
6702 return status;
6703}
6704
Tim Petersced69f82003-09-16 20:30:58 +00006705static
Guido van Rossumd57fd912000-03-10 22:53:23 +00006706int fixcapitalize(PyUnicodeObject *self)
6707{
Martin v. Löwis18e16552006-02-15 17:27:45 +00006708 Py_ssize_t len = self->length;
Marc-André Lemburgfde66e12001-01-29 11:14:16 +00006709 Py_UNICODE *s = self->str;
6710 int status = 0;
Tim Petersced69f82003-09-16 20:30:58 +00006711
Marc-André Lemburgfde66e12001-01-29 11:14:16 +00006712 if (len == 0)
Benjamin Peterson29060642009-01-31 22:14:21 +00006713 return 0;
Ezio Melottiee8d9982011-08-15 09:09:57 +03006714 if (!Py_UNICODE_ISUPPER(*s)) {
Benjamin Peterson29060642009-01-31 22:14:21 +00006715 *s = Py_UNICODE_TOUPPER(*s);
6716 status = 1;
Guido van Rossumd57fd912000-03-10 22:53:23 +00006717 }
Marc-André Lemburgfde66e12001-01-29 11:14:16 +00006718 s++;
6719 while (--len > 0) {
Ezio Melottiee8d9982011-08-15 09:09:57 +03006720 if (!Py_UNICODE_ISLOWER(*s)) {
Marc-André Lemburgfde66e12001-01-29 11:14:16 +00006721 *s = Py_UNICODE_TOLOWER(*s);
6722 status = 1;
6723 }
6724 s++;
6725 }
6726 return status;
Guido van Rossumd57fd912000-03-10 22:53:23 +00006727}
6728
6729static
6730int fixtitle(PyUnicodeObject *self)
6731{
6732 register Py_UNICODE *p = PyUnicode_AS_UNICODE(self);
6733 register Py_UNICODE *e;
6734 int previous_is_cased;
6735
6736 /* Shortcut for single character strings */
6737 if (PyUnicode_GET_SIZE(self) == 1) {
Benjamin Peterson29060642009-01-31 22:14:21 +00006738 Py_UNICODE ch = Py_UNICODE_TOTITLE(*p);
6739 if (*p != ch) {
6740 *p = ch;
6741 return 1;
6742 }
6743 else
6744 return 0;
Guido van Rossumd57fd912000-03-10 22:53:23 +00006745 }
Tim Petersced69f82003-09-16 20:30:58 +00006746
Guido van Rossumd57fd912000-03-10 22:53:23 +00006747 e = p + PyUnicode_GET_SIZE(self);
6748 previous_is_cased = 0;
6749 for (; p < e; p++) {
Benjamin Peterson29060642009-01-31 22:14:21 +00006750 register const Py_UNICODE ch = *p;
Tim Petersced69f82003-09-16 20:30:58 +00006751
Benjamin Peterson29060642009-01-31 22:14:21 +00006752 if (previous_is_cased)
6753 *p = Py_UNICODE_TOLOWER(ch);
6754 else
6755 *p = Py_UNICODE_TOTITLE(ch);
Tim Petersced69f82003-09-16 20:30:58 +00006756
Benjamin Peterson29060642009-01-31 22:14:21 +00006757 if (Py_UNICODE_ISLOWER(ch) ||
6758 Py_UNICODE_ISUPPER(ch) ||
6759 Py_UNICODE_ISTITLE(ch))
6760 previous_is_cased = 1;
6761 else
6762 previous_is_cased = 0;
Guido van Rossumd57fd912000-03-10 22:53:23 +00006763 }
6764 return 1;
6765}
6766
Tim Peters8ce9f162004-08-27 01:49:32 +00006767PyObject *
6768PyUnicode_Join(PyObject *separator, PyObject *seq)
Guido van Rossumd57fd912000-03-10 22:53:23 +00006769{
Skip Montanaro6543b452004-09-16 03:28:13 +00006770 const Py_UNICODE blank = ' ';
6771 const Py_UNICODE *sep = &blank;
Thomas Wouters477c8d52006-05-27 19:21:47 +00006772 Py_ssize_t seplen = 1;
Tim Peters05eba1f2004-08-27 21:32:02 +00006773 PyUnicodeObject *res = NULL; /* the result */
Tim Peters05eba1f2004-08-27 21:32:02 +00006774 Py_UNICODE *res_p; /* pointer to free byte in res's string area */
6775 PyObject *fseq; /* PySequence_Fast(seq) */
Antoine Pitrouaf14b792008-08-07 21:50:41 +00006776 Py_ssize_t seqlen; /* len(fseq) -- number of items in sequence */
6777 PyObject **items;
Tim Peters8ce9f162004-08-27 01:49:32 +00006778 PyObject *item;
Antoine Pitrouaf14b792008-08-07 21:50:41 +00006779 Py_ssize_t sz, i;
Guido van Rossumd57fd912000-03-10 22:53:23 +00006780
Tim Peters05eba1f2004-08-27 21:32:02 +00006781 fseq = PySequence_Fast(seq, "");
6782 if (fseq == NULL) {
Benjamin Peterson14339b62009-01-31 16:36:08 +00006783 return NULL;
Tim Peters8ce9f162004-08-27 01:49:32 +00006784 }
6785
Antoine Pitrouaf14b792008-08-07 21:50:41 +00006786 /* NOTE: the following code can't call back into Python code,
6787 * so we are sure that fseq won't be mutated.
Tim Peters91879ab2004-08-27 22:35:44 +00006788 */
Antoine Pitrouaf14b792008-08-07 21:50:41 +00006789
Tim Peters05eba1f2004-08-27 21:32:02 +00006790 seqlen = PySequence_Fast_GET_SIZE(fseq);
6791 /* If empty sequence, return u"". */
6792 if (seqlen == 0) {
Benjamin Peterson14339b62009-01-31 16:36:08 +00006793 res = _PyUnicode_New(0); /* empty sequence; return u"" */
6794 goto Done;
Tim Peters05eba1f2004-08-27 21:32:02 +00006795 }
Antoine Pitrouaf14b792008-08-07 21:50:41 +00006796 items = PySequence_Fast_ITEMS(fseq);
Tim Peters05eba1f2004-08-27 21:32:02 +00006797 /* If singleton sequence with an exact Unicode, return that. */
6798 if (seqlen == 1) {
Benjamin Peterson29060642009-01-31 22:14:21 +00006799 item = items[0];
6800 if (PyUnicode_CheckExact(item)) {
6801 Py_INCREF(item);
6802 res = (PyUnicodeObject *)item;
6803 goto Done;
6804 }
Tim Peters8ce9f162004-08-27 01:49:32 +00006805 }
Antoine Pitrouaf14b792008-08-07 21:50:41 +00006806 else {
6807 /* Set up sep and seplen */
6808 if (separator == NULL) {
6809 sep = &blank;
6810 seplen = 1;
Tim Peters05eba1f2004-08-27 21:32:02 +00006811 }
Antoine Pitrouaf14b792008-08-07 21:50:41 +00006812 else {
6813 if (!PyUnicode_Check(separator)) {
6814 PyErr_Format(PyExc_TypeError,
6815 "separator: expected str instance,"
6816 " %.80s found",
6817 Py_TYPE(separator)->tp_name);
6818 goto onError;
6819 }
6820 sep = PyUnicode_AS_UNICODE(separator);
6821 seplen = PyUnicode_GET_SIZE(separator);
Tim Peters05eba1f2004-08-27 21:32:02 +00006822 }
6823 }
6824
Antoine Pitrouaf14b792008-08-07 21:50:41 +00006825 /* There are at least two things to join, or else we have a subclass
6826 * of str in the sequence.
6827 * Do a pre-pass to figure out the total amount of space we'll
6828 * need (sz), and see whether all argument are strings.
6829 */
6830 sz = 0;
6831 for (i = 0; i < seqlen; i++) {
6832 const Py_ssize_t old_sz = sz;
6833 item = items[i];
Benjamin Peterson29060642009-01-31 22:14:21 +00006834 if (!PyUnicode_Check(item)) {
6835 PyErr_Format(PyExc_TypeError,
6836 "sequence item %zd: expected str instance,"
6837 " %.80s found",
6838 i, Py_TYPE(item)->tp_name);
6839 goto onError;
6840 }
Antoine Pitrouaf14b792008-08-07 21:50:41 +00006841 sz += PyUnicode_GET_SIZE(item);
6842 if (i != 0)
6843 sz += seplen;
6844 if (sz < old_sz || sz > PY_SSIZE_T_MAX) {
6845 PyErr_SetString(PyExc_OverflowError,
Benjamin Peterson29060642009-01-31 22:14:21 +00006846 "join() result is too long for a Python string");
Antoine Pitrouaf14b792008-08-07 21:50:41 +00006847 goto onError;
6848 }
6849 }
Tim Petersced69f82003-09-16 20:30:58 +00006850
Antoine Pitrouaf14b792008-08-07 21:50:41 +00006851 res = _PyUnicode_New(sz);
6852 if (res == NULL)
6853 goto onError;
Tim Peters91879ab2004-08-27 22:35:44 +00006854
Antoine Pitrouaf14b792008-08-07 21:50:41 +00006855 /* Catenate everything. */
6856 res_p = PyUnicode_AS_UNICODE(res);
6857 for (i = 0; i < seqlen; ++i) {
6858 Py_ssize_t itemlen;
6859 item = items[i];
6860 itemlen = PyUnicode_GET_SIZE(item);
Benjamin Peterson29060642009-01-31 22:14:21 +00006861 /* Copy item, and maybe the separator. */
6862 if (i) {
6863 Py_UNICODE_COPY(res_p, sep, seplen);
6864 res_p += seplen;
6865 }
6866 Py_UNICODE_COPY(res_p, PyUnicode_AS_UNICODE(item), itemlen);
6867 res_p += itemlen;
Tim Peters05eba1f2004-08-27 21:32:02 +00006868 }
Tim Peters8ce9f162004-08-27 01:49:32 +00006869
Benjamin Peterson29060642009-01-31 22:14:21 +00006870 Done:
Tim Peters05eba1f2004-08-27 21:32:02 +00006871 Py_DECREF(fseq);
Guido van Rossumd57fd912000-03-10 22:53:23 +00006872 return (PyObject *)res;
6873
Benjamin Peterson29060642009-01-31 22:14:21 +00006874 onError:
Tim Peters05eba1f2004-08-27 21:32:02 +00006875 Py_DECREF(fseq);
Tim Peters8ce9f162004-08-27 01:49:32 +00006876 Py_XDECREF(res);
Guido van Rossumd57fd912000-03-10 22:53:23 +00006877 return NULL;
6878}
6879
Tim Petersced69f82003-09-16 20:30:58 +00006880static
6881PyUnicodeObject *pad(PyUnicodeObject *self,
Benjamin Peterson29060642009-01-31 22:14:21 +00006882 Py_ssize_t left,
6883 Py_ssize_t right,
6884 Py_UNICODE fill)
Guido van Rossumd57fd912000-03-10 22:53:23 +00006885{
6886 PyUnicodeObject *u;
6887
6888 if (left < 0)
6889 left = 0;
6890 if (right < 0)
6891 right = 0;
6892
Tim Peters7a29bd52001-09-12 03:03:31 +00006893 if (left == 0 && right == 0 && PyUnicode_CheckExact(self)) {
Guido van Rossumd57fd912000-03-10 22:53:23 +00006894 Py_INCREF(self);
6895 return self;
6896 }
6897
Neal Norwitz3ce5d922008-08-24 07:08:55 +00006898 if (left > PY_SSIZE_T_MAX - self->length ||
6899 right > PY_SSIZE_T_MAX - (left + self->length)) {
6900 PyErr_SetString(PyExc_OverflowError, "padded string is too long");
6901 return NULL;
6902 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00006903 u = _PyUnicode_New(left + self->length + right);
6904 if (u) {
6905 if (left)
6906 Py_UNICODE_FILL(u->str, fill, left);
6907 Py_UNICODE_COPY(u->str + left, self->str, self->length);
6908 if (right)
6909 Py_UNICODE_FILL(u->str + left + self->length, fill, right);
6910 }
6911
6912 return u;
6913}
6914
Antoine Pitrouf2c54842010-01-13 08:07:53 +00006915PyObject *PyUnicode_Splitlines(PyObject *string, int keepends)
Guido van Rossumd57fd912000-03-10 22:53:23 +00006916{
Guido van Rossumd57fd912000-03-10 22:53:23 +00006917 PyObject *list;
Guido van Rossumd57fd912000-03-10 22:53:23 +00006918
6919 string = PyUnicode_FromObject(string);
6920 if (string == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00006921 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00006922
Antoine Pitrouf2c54842010-01-13 08:07:53 +00006923 list = stringlib_splitlines(
6924 (PyObject*) string, PyUnicode_AS_UNICODE(string),
6925 PyUnicode_GET_SIZE(string), keepends);
Guido van Rossumd57fd912000-03-10 22:53:23 +00006926
6927 Py_DECREF(string);
6928 return list;
Guido van Rossumd57fd912000-03-10 22:53:23 +00006929}
6930
Tim Petersced69f82003-09-16 20:30:58 +00006931static
Guido van Rossumd57fd912000-03-10 22:53:23 +00006932PyObject *split(PyUnicodeObject *self,
Benjamin Peterson29060642009-01-31 22:14:21 +00006933 PyUnicodeObject *substring,
6934 Py_ssize_t maxcount)
Guido van Rossumd57fd912000-03-10 22:53:23 +00006935{
Guido van Rossumd57fd912000-03-10 22:53:23 +00006936 if (maxcount < 0)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006937 maxcount = PY_SSIZE_T_MAX;
Guido van Rossumd57fd912000-03-10 22:53:23 +00006938
Guido van Rossumd57fd912000-03-10 22:53:23 +00006939 if (substring == NULL)
Antoine Pitrouf2c54842010-01-13 08:07:53 +00006940 return stringlib_split_whitespace(
6941 (PyObject*) self, self->str, self->length, maxcount
6942 );
Guido van Rossumd57fd912000-03-10 22:53:23 +00006943
Antoine Pitrouf2c54842010-01-13 08:07:53 +00006944 return stringlib_split(
6945 (PyObject*) self, self->str, self->length,
6946 substring->str, substring->length,
6947 maxcount
6948 );
Guido van Rossumd57fd912000-03-10 22:53:23 +00006949}
6950
Tim Petersced69f82003-09-16 20:30:58 +00006951static
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +00006952PyObject *rsplit(PyUnicodeObject *self,
Benjamin Peterson29060642009-01-31 22:14:21 +00006953 PyUnicodeObject *substring,
6954 Py_ssize_t maxcount)
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +00006955{
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +00006956 if (maxcount < 0)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006957 maxcount = PY_SSIZE_T_MAX;
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +00006958
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +00006959 if (substring == NULL)
Antoine Pitrouf2c54842010-01-13 08:07:53 +00006960 return stringlib_rsplit_whitespace(
6961 (PyObject*) self, self->str, self->length, maxcount
6962 );
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +00006963
Antoine Pitrouf2c54842010-01-13 08:07:53 +00006964 return stringlib_rsplit(
6965 (PyObject*) self, self->str, self->length,
6966 substring->str, substring->length,
6967 maxcount
6968 );
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +00006969}
6970
6971static
Guido van Rossumd57fd912000-03-10 22:53:23 +00006972PyObject *replace(PyUnicodeObject *self,
Benjamin Peterson29060642009-01-31 22:14:21 +00006973 PyUnicodeObject *str1,
6974 PyUnicodeObject *str2,
6975 Py_ssize_t maxcount)
Guido van Rossumd57fd912000-03-10 22:53:23 +00006976{
6977 PyUnicodeObject *u;
6978
6979 if (maxcount < 0)
Benjamin Peterson29060642009-01-31 22:14:21 +00006980 maxcount = PY_SSIZE_T_MAX;
Antoine Pitrouf2c54842010-01-13 08:07:53 +00006981 else if (maxcount == 0 || self->length == 0)
6982 goto nothing;
Guido van Rossumd57fd912000-03-10 22:53:23 +00006983
Thomas Wouters477c8d52006-05-27 19:21:47 +00006984 if (str1->length == str2->length) {
Antoine Pitroucbfdee32010-01-13 08:58:08 +00006985 Py_ssize_t i;
Thomas Wouters477c8d52006-05-27 19:21:47 +00006986 /* same length */
Antoine Pitrouf2c54842010-01-13 08:07:53 +00006987 if (str1->length == 0)
6988 goto nothing;
Thomas Wouters477c8d52006-05-27 19:21:47 +00006989 if (str1->length == 1) {
6990 /* replace characters */
6991 Py_UNICODE u1, u2;
6992 if (!findchar(self->str, self->length, str1->str[0]))
6993 goto nothing;
6994 u = (PyUnicodeObject*) PyUnicode_FromUnicode(NULL, self->length);
6995 if (!u)
6996 return NULL;
6997 Py_UNICODE_COPY(u->str, self->str, self->length);
6998 u1 = str1->str[0];
6999 u2 = str2->str[0];
7000 for (i = 0; i < u->length; i++)
7001 if (u->str[i] == u1) {
7002 if (--maxcount < 0)
7003 break;
7004 u->str[i] = u2;
7005 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00007006 } else {
Antoine Pitrouf2c54842010-01-13 08:07:53 +00007007 i = stringlib_find(
7008 self->str, self->length, str1->str, str1->length, 0
Guido van Rossumd57fd912000-03-10 22:53:23 +00007009 );
Thomas Wouters477c8d52006-05-27 19:21:47 +00007010 if (i < 0)
7011 goto nothing;
7012 u = (PyUnicodeObject*) PyUnicode_FromUnicode(NULL, self->length);
7013 if (!u)
7014 return NULL;
7015 Py_UNICODE_COPY(u->str, self->str, self->length);
Antoine Pitrouf2c54842010-01-13 08:07:53 +00007016
7017 /* change everything in-place, starting with this one */
7018 Py_UNICODE_COPY(u->str+i, str2->str, str2->length);
7019 i += str1->length;
7020
7021 while ( --maxcount > 0) {
7022 i = stringlib_find(self->str+i, self->length-i,
7023 str1->str, str1->length,
7024 i);
7025 if (i == -1)
7026 break;
7027 Py_UNICODE_COPY(u->str+i, str2->str, str2->length);
7028 i += str1->length;
7029 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00007030 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00007031 } else {
Thomas Wouters477c8d52006-05-27 19:21:47 +00007032
Victor Stinnerab1d16b2011-11-22 01:45:37 +01007033 Py_ssize_t n, i, j;
Thomas Wouters477c8d52006-05-27 19:21:47 +00007034 Py_ssize_t product, new_size, delta;
Guido van Rossumd57fd912000-03-10 22:53:23 +00007035 Py_UNICODE *p;
7036
7037 /* replace strings */
Antoine Pitrouf2c54842010-01-13 08:07:53 +00007038 n = stringlib_count(self->str, self->length, str1->str, str1->length,
7039 maxcount);
Thomas Wouters477c8d52006-05-27 19:21:47 +00007040 if (n == 0)
7041 goto nothing;
7042 /* new_size = self->length + n * (str2->length - str1->length)); */
7043 delta = (str2->length - str1->length);
7044 if (delta == 0) {
7045 new_size = self->length;
Guido van Rossumd57fd912000-03-10 22:53:23 +00007046 } else {
Thomas Wouters477c8d52006-05-27 19:21:47 +00007047 product = n * (str2->length - str1->length);
7048 if ((product / (str2->length - str1->length)) != n) {
7049 PyErr_SetString(PyExc_OverflowError,
7050 "replace string is too long");
7051 return NULL;
7052 }
7053 new_size = self->length + product;
7054 if (new_size < 0) {
7055 PyErr_SetString(PyExc_OverflowError,
7056 "replace string is too long");
7057 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00007058 }
7059 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00007060 u = _PyUnicode_New(new_size);
7061 if (!u)
7062 return NULL;
7063 i = 0;
7064 p = u->str;
Thomas Wouters477c8d52006-05-27 19:21:47 +00007065 if (str1->length > 0) {
7066 while (n-- > 0) {
7067 /* look for next match */
Antoine Pitrouf2c54842010-01-13 08:07:53 +00007068 j = stringlib_find(self->str+i, self->length-i,
7069 str1->str, str1->length,
7070 i);
7071 if (j == -1)
7072 break;
7073 else if (j > i) {
Thomas Wouters477c8d52006-05-27 19:21:47 +00007074 /* copy unchanged part [i:j] */
7075 Py_UNICODE_COPY(p, self->str+i, j-i);
7076 p += j - i;
7077 }
7078 /* copy substitution string */
7079 if (str2->length > 0) {
7080 Py_UNICODE_COPY(p, str2->str, str2->length);
7081 p += str2->length;
7082 }
7083 i = j + str1->length;
7084 }
7085 if (i < self->length)
7086 /* copy tail [i:] */
7087 Py_UNICODE_COPY(p, self->str+i, self->length-i);
7088 } else {
7089 /* interleave */
7090 while (n > 0) {
7091 Py_UNICODE_COPY(p, str2->str, str2->length);
7092 p += str2->length;
7093 if (--n <= 0)
7094 break;
7095 *p++ = self->str[i++];
7096 }
7097 Py_UNICODE_COPY(p, self->str+i, self->length-i);
7098 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00007099 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00007100 return (PyObject *) u;
Thomas Wouters477c8d52006-05-27 19:21:47 +00007101
Benjamin Peterson29060642009-01-31 22:14:21 +00007102 nothing:
Thomas Wouters477c8d52006-05-27 19:21:47 +00007103 /* nothing to replace; return original string (when possible) */
7104 if (PyUnicode_CheckExact(self)) {
7105 Py_INCREF(self);
7106 return (PyObject *) self;
7107 }
7108 return PyUnicode_FromUnicode(self->str, self->length);
Guido van Rossumd57fd912000-03-10 22:53:23 +00007109}
7110
7111/* --- Unicode Object Methods --------------------------------------------- */
7112
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007113PyDoc_STRVAR(title__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +00007114 "S.title() -> str\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +00007115\n\
7116Return a titlecased version of S, i.e. words start with title case\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007117characters, all remaining cased characters have lower case.");
Guido van Rossumd57fd912000-03-10 22:53:23 +00007118
7119static PyObject*
Martin v. Löwise3eb1f22001-08-16 13:15:00 +00007120unicode_title(PyUnicodeObject *self)
Guido van Rossumd57fd912000-03-10 22:53:23 +00007121{
Guido van Rossumd57fd912000-03-10 22:53:23 +00007122 return fixup(self, fixtitle);
7123}
7124
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007125PyDoc_STRVAR(capitalize__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +00007126 "S.capitalize() -> str\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +00007127\n\
7128Return a capitalized version of S, i.e. make the first character\n\
Senthil Kumarane51ee8a2010-07-05 12:00:56 +00007129have upper case and the rest lower case.");
Guido van Rossumd57fd912000-03-10 22:53:23 +00007130
7131static PyObject*
Martin v. Löwise3eb1f22001-08-16 13:15:00 +00007132unicode_capitalize(PyUnicodeObject *self)
Guido van Rossumd57fd912000-03-10 22:53:23 +00007133{
Guido van Rossumd57fd912000-03-10 22:53:23 +00007134 return fixup(self, fixcapitalize);
7135}
7136
7137#if 0
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007138PyDoc_STRVAR(capwords__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +00007139 "S.capwords() -> str\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +00007140\n\
7141Apply .capitalize() to all words in S and return the result with\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007142normalized whitespace (all whitespace strings are replaced by ' ').");
Guido van Rossumd57fd912000-03-10 22:53:23 +00007143
7144static PyObject*
Martin v. Löwise3eb1f22001-08-16 13:15:00 +00007145unicode_capwords(PyUnicodeObject *self)
Guido van Rossumd57fd912000-03-10 22:53:23 +00007146{
7147 PyObject *list;
7148 PyObject *item;
Martin v. Löwis18e16552006-02-15 17:27:45 +00007149 Py_ssize_t i;
Guido van Rossumd57fd912000-03-10 22:53:23 +00007150
Guido van Rossumd57fd912000-03-10 22:53:23 +00007151 /* Split into words */
7152 list = split(self, NULL, -1);
7153 if (!list)
7154 return NULL;
7155
7156 /* Capitalize each word */
7157 for (i = 0; i < PyList_GET_SIZE(list); i++) {
7158 item = fixup((PyUnicodeObject *)PyList_GET_ITEM(list, i),
Benjamin Peterson29060642009-01-31 22:14:21 +00007159 fixcapitalize);
Guido van Rossumd57fd912000-03-10 22:53:23 +00007160 if (item == NULL)
7161 goto onError;
7162 Py_DECREF(PyList_GET_ITEM(list, i));
7163 PyList_SET_ITEM(list, i, item);
7164 }
7165
7166 /* Join the words to form a new string */
7167 item = PyUnicode_Join(NULL, list);
7168
Benjamin Peterson29060642009-01-31 22:14:21 +00007169 onError:
Guido van Rossumd57fd912000-03-10 22:53:23 +00007170 Py_DECREF(list);
7171 return (PyObject *)item;
7172}
7173#endif
7174
Raymond Hettinger4f8f9762003-11-26 08:21:35 +00007175/* Argument converter. Coerces to a single unicode character */
7176
7177static int
7178convert_uc(PyObject *obj, void *addr)
7179{
Benjamin Peterson14339b62009-01-31 16:36:08 +00007180 Py_UNICODE *fillcharloc = (Py_UNICODE *)addr;
7181 PyObject *uniobj;
7182 Py_UNICODE *unistr;
Raymond Hettinger4f8f9762003-11-26 08:21:35 +00007183
Benjamin Peterson14339b62009-01-31 16:36:08 +00007184 uniobj = PyUnicode_FromObject(obj);
7185 if (uniobj == NULL) {
7186 PyErr_SetString(PyExc_TypeError,
Benjamin Peterson29060642009-01-31 22:14:21 +00007187 "The fill character cannot be converted to Unicode");
Benjamin Peterson14339b62009-01-31 16:36:08 +00007188 return 0;
7189 }
7190 if (PyUnicode_GET_SIZE(uniobj) != 1) {
7191 PyErr_SetString(PyExc_TypeError,
Benjamin Peterson29060642009-01-31 22:14:21 +00007192 "The fill character must be exactly one character long");
Benjamin Peterson14339b62009-01-31 16:36:08 +00007193 Py_DECREF(uniobj);
7194 return 0;
7195 }
7196 unistr = PyUnicode_AS_UNICODE(uniobj);
7197 *fillcharloc = unistr[0];
7198 Py_DECREF(uniobj);
7199 return 1;
Raymond Hettinger4f8f9762003-11-26 08:21:35 +00007200}
7201
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007202PyDoc_STRVAR(center__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +00007203 "S.center(width[, fillchar]) -> str\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +00007204\n\
Benjamin Peterson142957c2008-07-04 19:55:29 +00007205Return S centered in a string of length width. Padding is\n\
Raymond Hettinger4f8f9762003-11-26 08:21:35 +00007206done using the specified fill character (default is a space)");
Guido van Rossumd57fd912000-03-10 22:53:23 +00007207
7208static PyObject *
7209unicode_center(PyUnicodeObject *self, PyObject *args)
7210{
Martin v. Löwis18e16552006-02-15 17:27:45 +00007211 Py_ssize_t marg, left;
7212 Py_ssize_t width;
Raymond Hettinger4f8f9762003-11-26 08:21:35 +00007213 Py_UNICODE fillchar = ' ';
Guido van Rossumd57fd912000-03-10 22:53:23 +00007214
Thomas Woutersde017742006-02-16 19:34:37 +00007215 if (!PyArg_ParseTuple(args, "n|O&:center", &width, convert_uc, &fillchar))
Guido van Rossumd57fd912000-03-10 22:53:23 +00007216 return NULL;
7217
Tim Peters7a29bd52001-09-12 03:03:31 +00007218 if (self->length >= width && PyUnicode_CheckExact(self)) {
Guido van Rossumd57fd912000-03-10 22:53:23 +00007219 Py_INCREF(self);
7220 return (PyObject*) self;
7221 }
7222
7223 marg = width - self->length;
7224 left = marg / 2 + (marg & width & 1);
7225
Raymond Hettinger4f8f9762003-11-26 08:21:35 +00007226 return (PyObject*) pad(self, left, marg - left, fillchar);
Guido van Rossumd57fd912000-03-10 22:53:23 +00007227}
7228
Marc-André Lemburge5034372000-08-08 08:04:29 +00007229#if 0
7230
7231/* This code should go into some future Unicode collation support
7232 module. The basic comparison should compare ordinals on a naive
Georg Brandlc6c31782009-06-08 13:41:29 +00007233 basis (this is what Java does and thus Jython too). */
Marc-André Lemburge5034372000-08-08 08:04:29 +00007234
Marc-André Lemburg1e7205a2000-07-04 09:51:07 +00007235/* speedy UTF-16 code point order comparison */
7236/* gleaned from: */
7237/* http://www-4.ibm.com/software/developer/library/utf16.html?dwzone=unicode */
7238
Marc-André Lemburge12896e2000-07-07 17:51:08 +00007239static short utf16Fixup[32] =
Marc-André Lemburg1e7205a2000-07-04 09:51:07 +00007240{
Marc-André Lemburg1e7205a2000-07-04 09:51:07 +00007241 0, 0, 0, 0, 0, 0, 0, 0,
Tim Petersced69f82003-09-16 20:30:58 +00007242 0, 0, 0, 0, 0, 0, 0, 0,
7243 0, 0, 0, 0, 0, 0, 0, 0,
Marc-André Lemburge12896e2000-07-07 17:51:08 +00007244 0, 0, 0, 0x2000, -0x800, -0x800, -0x800, -0x800
Marc-André Lemburg1e7205a2000-07-04 09:51:07 +00007245};
7246
Guido van Rossumd57fd912000-03-10 22:53:23 +00007247static int
7248unicode_compare(PyUnicodeObject *str1, PyUnicodeObject *str2)
7249{
Martin v. Löwis18e16552006-02-15 17:27:45 +00007250 Py_ssize_t len1, len2;
Marc-André Lemburg1e7205a2000-07-04 09:51:07 +00007251
Guido van Rossumd57fd912000-03-10 22:53:23 +00007252 Py_UNICODE *s1 = str1->str;
7253 Py_UNICODE *s2 = str2->str;
7254
7255 len1 = str1->length;
7256 len2 = str2->length;
Tim Petersced69f82003-09-16 20:30:58 +00007257
Guido van Rossumd57fd912000-03-10 22:53:23 +00007258 while (len1 > 0 && len2 > 0) {
Tim Petersced69f82003-09-16 20:30:58 +00007259 Py_UNICODE c1, c2;
Marc-André Lemburg1e7205a2000-07-04 09:51:07 +00007260
7261 c1 = *s1++;
7262 c2 = *s2++;
Fredrik Lundh45714e92001-06-26 16:39:36 +00007263
Benjamin Peterson29060642009-01-31 22:14:21 +00007264 if (c1 > (1<<11) * 26)
7265 c1 += utf16Fixup[c1>>11];
7266 if (c2 > (1<<11) * 26)
Marc-André Lemburg1e7205a2000-07-04 09:51:07 +00007267 c2 += utf16Fixup[c2>>11];
Marc-André Lemburg1e7205a2000-07-04 09:51:07 +00007268 /* now c1 and c2 are in UTF-32-compatible order */
Fredrik Lundh45714e92001-06-26 16:39:36 +00007269
7270 if (c1 != c2)
7271 return (c1 < c2) ? -1 : 1;
Tim Petersced69f82003-09-16 20:30:58 +00007272
Marc-André Lemburg1e7205a2000-07-04 09:51:07 +00007273 len1--; len2--;
Guido van Rossumd57fd912000-03-10 22:53:23 +00007274 }
7275
7276 return (len1 < len2) ? -1 : (len1 != len2);
7277}
7278
Marc-André Lemburge5034372000-08-08 08:04:29 +00007279#else
7280
7281static int
7282unicode_compare(PyUnicodeObject *str1, PyUnicodeObject *str2)
7283{
Martin v. Löwis18e16552006-02-15 17:27:45 +00007284 register Py_ssize_t len1, len2;
Marc-André Lemburge5034372000-08-08 08:04:29 +00007285
7286 Py_UNICODE *s1 = str1->str;
7287 Py_UNICODE *s2 = str2->str;
7288
7289 len1 = str1->length;
7290 len2 = str2->length;
Tim Petersced69f82003-09-16 20:30:58 +00007291
Marc-André Lemburge5034372000-08-08 08:04:29 +00007292 while (len1 > 0 && len2 > 0) {
Tim Petersced69f82003-09-16 20:30:58 +00007293 Py_UNICODE c1, c2;
Marc-André Lemburge5034372000-08-08 08:04:29 +00007294
Fredrik Lundh45714e92001-06-26 16:39:36 +00007295 c1 = *s1++;
7296 c2 = *s2++;
7297
7298 if (c1 != c2)
7299 return (c1 < c2) ? -1 : 1;
7300
Marc-André Lemburge5034372000-08-08 08:04:29 +00007301 len1--; len2--;
7302 }
7303
7304 return (len1 < len2) ? -1 : (len1 != len2);
7305}
7306
7307#endif
7308
Guido van Rossumd57fd912000-03-10 22:53:23 +00007309int PyUnicode_Compare(PyObject *left,
Benjamin Peterson29060642009-01-31 22:14:21 +00007310 PyObject *right)
Guido van Rossumd57fd912000-03-10 22:53:23 +00007311{
Guido van Rossum09dc34f2007-05-04 04:17:33 +00007312 if (PyUnicode_Check(left) && PyUnicode_Check(right))
7313 return unicode_compare((PyUnicodeObject *)left,
7314 (PyUnicodeObject *)right);
Guido van Rossum09dc34f2007-05-04 04:17:33 +00007315 PyErr_Format(PyExc_TypeError,
7316 "Can't compare %.100s and %.100s",
7317 left->ob_type->tp_name,
7318 right->ob_type->tp_name);
Guido van Rossumd57fd912000-03-10 22:53:23 +00007319 return -1;
7320}
7321
Martin v. Löwis5b222132007-06-10 09:51:05 +00007322int
7323PyUnicode_CompareWithASCIIString(PyObject* uni, const char* str)
7324{
7325 int i;
7326 Py_UNICODE *id;
7327 assert(PyUnicode_Check(uni));
7328 id = PyUnicode_AS_UNICODE(uni);
7329 /* Compare Unicode string and source character set string */
7330 for (i = 0; id[i] && str[i]; i++)
Benjamin Peterson29060642009-01-31 22:14:21 +00007331 if (id[i] != str[i])
7332 return ((int)id[i] < (int)str[i]) ? -1 : 1;
Benjamin Peterson8667a9b2010-01-09 21:45:28 +00007333 /* This check keeps Python strings that end in '\0' from comparing equal
7334 to C strings identical up to that point. */
Benjamin Petersona23831f2010-04-25 21:54:00 +00007335 if (PyUnicode_GET_SIZE(uni) != i || id[i])
Benjamin Peterson29060642009-01-31 22:14:21 +00007336 return 1; /* uni is longer */
Martin v. Löwis5b222132007-06-10 09:51:05 +00007337 if (str[i])
Benjamin Peterson29060642009-01-31 22:14:21 +00007338 return -1; /* str is longer */
Martin v. Löwis5b222132007-06-10 09:51:05 +00007339 return 0;
7340}
7341
Antoine Pitrou51f3ef92008-12-20 13:14:23 +00007342
Benjamin Peterson29060642009-01-31 22:14:21 +00007343#define TEST_COND(cond) \
Benjamin Peterson14339b62009-01-31 16:36:08 +00007344 ((cond) ? Py_True : Py_False)
Antoine Pitrou51f3ef92008-12-20 13:14:23 +00007345
Thomas Wouters00ee7ba2006-08-21 19:07:27 +00007346PyObject *PyUnicode_RichCompare(PyObject *left,
7347 PyObject *right,
7348 int op)
7349{
7350 int result;
Benjamin Peterson14339b62009-01-31 16:36:08 +00007351
Antoine Pitrou51f3ef92008-12-20 13:14:23 +00007352 if (PyUnicode_Check(left) && PyUnicode_Check(right)) {
7353 PyObject *v;
7354 if (((PyUnicodeObject *) left)->length !=
7355 ((PyUnicodeObject *) right)->length) {
7356 if (op == Py_EQ) {
7357 Py_INCREF(Py_False);
7358 return Py_False;
7359 }
7360 if (op == Py_NE) {
7361 Py_INCREF(Py_True);
7362 return Py_True;
7363 }
7364 }
7365 if (left == right)
7366 result = 0;
7367 else
7368 result = unicode_compare((PyUnicodeObject *)left,
7369 (PyUnicodeObject *)right);
Benjamin Peterson14339b62009-01-31 16:36:08 +00007370
Antoine Pitrou51f3ef92008-12-20 13:14:23 +00007371 /* Convert the return value to a Boolean */
7372 switch (op) {
7373 case Py_EQ:
7374 v = TEST_COND(result == 0);
7375 break;
7376 case Py_NE:
7377 v = TEST_COND(result != 0);
7378 break;
7379 case Py_LE:
7380 v = TEST_COND(result <= 0);
7381 break;
7382 case Py_GE:
7383 v = TEST_COND(result >= 0);
7384 break;
7385 case Py_LT:
7386 v = TEST_COND(result == -1);
7387 break;
7388 case Py_GT:
7389 v = TEST_COND(result == 1);
7390 break;
7391 default:
7392 PyErr_BadArgument();
7393 return NULL;
7394 }
7395 Py_INCREF(v);
7396 return v;
Thomas Wouters00ee7ba2006-08-21 19:07:27 +00007397 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00007398
Antoine Pitrou51f3ef92008-12-20 13:14:23 +00007399 Py_INCREF(Py_NotImplemented);
7400 return Py_NotImplemented;
Thomas Wouters00ee7ba2006-08-21 19:07:27 +00007401}
7402
Guido van Rossum403d68b2000-03-13 15:55:09 +00007403int PyUnicode_Contains(PyObject *container,
Benjamin Peterson29060642009-01-31 22:14:21 +00007404 PyObject *element)
Guido van Rossum403d68b2000-03-13 15:55:09 +00007405{
Thomas Wouters477c8d52006-05-27 19:21:47 +00007406 PyObject *str, *sub;
Martin v. Löwis18e16552006-02-15 17:27:45 +00007407 int result;
Guido van Rossum403d68b2000-03-13 15:55:09 +00007408
7409 /* Coerce the two arguments */
Thomas Wouters477c8d52006-05-27 19:21:47 +00007410 sub = PyUnicode_FromObject(element);
7411 if (!sub) {
Benjamin Peterson29060642009-01-31 22:14:21 +00007412 PyErr_Format(PyExc_TypeError,
7413 "'in <string>' requires string as left operand, not %s",
7414 element->ob_type->tp_name);
Thomas Wouters477c8d52006-05-27 19:21:47 +00007415 return -1;
Guido van Rossum403d68b2000-03-13 15:55:09 +00007416 }
7417
Thomas Wouters477c8d52006-05-27 19:21:47 +00007418 str = PyUnicode_FromObject(container);
7419 if (!str) {
7420 Py_DECREF(sub);
7421 return -1;
7422 }
7423
7424 result = stringlib_contains_obj(str, sub);
7425
7426 Py_DECREF(str);
7427 Py_DECREF(sub);
7428
Guido van Rossum403d68b2000-03-13 15:55:09 +00007429 return result;
Guido van Rossum403d68b2000-03-13 15:55:09 +00007430}
7431
Guido van Rossumd57fd912000-03-10 22:53:23 +00007432/* Concat to string or Unicode object giving a new Unicode object. */
7433
7434PyObject *PyUnicode_Concat(PyObject *left,
Benjamin Peterson29060642009-01-31 22:14:21 +00007435 PyObject *right)
Guido van Rossumd57fd912000-03-10 22:53:23 +00007436{
7437 PyUnicodeObject *u = NULL, *v = NULL, *w;
7438
7439 /* Coerce the two arguments */
7440 u = (PyUnicodeObject *)PyUnicode_FromObject(left);
7441 if (u == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00007442 goto onError;
Guido van Rossumd57fd912000-03-10 22:53:23 +00007443 v = (PyUnicodeObject *)PyUnicode_FromObject(right);
7444 if (v == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00007445 goto onError;
Guido van Rossumd57fd912000-03-10 22:53:23 +00007446
7447 /* Shortcuts */
7448 if (v == unicode_empty) {
Benjamin Peterson29060642009-01-31 22:14:21 +00007449 Py_DECREF(v);
7450 return (PyObject *)u;
Guido van Rossumd57fd912000-03-10 22:53:23 +00007451 }
7452 if (u == unicode_empty) {
Benjamin Peterson29060642009-01-31 22:14:21 +00007453 Py_DECREF(u);
7454 return (PyObject *)v;
Guido van Rossumd57fd912000-03-10 22:53:23 +00007455 }
7456
7457 /* Concat the two Unicode strings */
7458 w = _PyUnicode_New(u->length + v->length);
7459 if (w == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00007460 goto onError;
Guido van Rossumd57fd912000-03-10 22:53:23 +00007461 Py_UNICODE_COPY(w->str, u->str, u->length);
7462 Py_UNICODE_COPY(w->str + u->length, v->str, v->length);
7463
7464 Py_DECREF(u);
7465 Py_DECREF(v);
7466 return (PyObject *)w;
7467
Benjamin Peterson29060642009-01-31 22:14:21 +00007468 onError:
Guido van Rossumd57fd912000-03-10 22:53:23 +00007469 Py_XDECREF(u);
7470 Py_XDECREF(v);
7471 return NULL;
7472}
7473
Walter Dörwald1ab83302007-05-18 17:15:44 +00007474void
7475PyUnicode_Append(PyObject **pleft, PyObject *right)
7476{
Benjamin Peterson14339b62009-01-31 16:36:08 +00007477 PyObject *new;
7478 if (*pleft == NULL)
7479 return;
7480 if (right == NULL || !PyUnicode_Check(*pleft)) {
7481 Py_DECREF(*pleft);
7482 *pleft = NULL;
7483 return;
7484 }
7485 new = PyUnicode_Concat(*pleft, right);
7486 Py_DECREF(*pleft);
7487 *pleft = new;
Walter Dörwald1ab83302007-05-18 17:15:44 +00007488}
7489
7490void
7491PyUnicode_AppendAndDel(PyObject **pleft, PyObject *right)
7492{
Benjamin Peterson14339b62009-01-31 16:36:08 +00007493 PyUnicode_Append(pleft, right);
7494 Py_XDECREF(right);
Walter Dörwald1ab83302007-05-18 17:15:44 +00007495}
7496
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007497PyDoc_STRVAR(count__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +00007498 "S.count(sub[, start[, end]]) -> int\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +00007499\n\
Thomas Wouters477c8d52006-05-27 19:21:47 +00007500Return the number of non-overlapping occurrences of substring sub in\n\
Benjamin Peterson142957c2008-07-04 19:55:29 +00007501string S[start:end]. Optional arguments start and end are\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007502interpreted as in slice notation.");
Guido van Rossumd57fd912000-03-10 22:53:23 +00007503
7504static PyObject *
7505unicode_count(PyUnicodeObject *self, PyObject *args)
7506{
7507 PyUnicodeObject *substring;
Martin v. Löwis18e16552006-02-15 17:27:45 +00007508 Py_ssize_t start = 0;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007509 Py_ssize_t end = PY_SSIZE_T_MAX;
Guido van Rossumd57fd912000-03-10 22:53:23 +00007510 PyObject *result;
7511
Jesus Ceaac451502011-04-20 17:09:23 +02007512 if (!stringlib_parse_args_finds_unicode("count", args, &substring,
7513 &start, &end))
Benjamin Peterson29060642009-01-31 22:14:21 +00007514 return NULL;
Tim Petersced69f82003-09-16 20:30:58 +00007515
Antoine Pitrouf2c54842010-01-13 08:07:53 +00007516 ADJUST_INDICES(start, end, self->length);
Christian Heimes217cfd12007-12-02 14:31:20 +00007517 result = PyLong_FromSsize_t(
Thomas Wouters477c8d52006-05-27 19:21:47 +00007518 stringlib_count(self->str + start, end - start,
Antoine Pitrouf2c54842010-01-13 08:07:53 +00007519 substring->str, substring->length,
7520 PY_SSIZE_T_MAX)
Thomas Wouters477c8d52006-05-27 19:21:47 +00007521 );
Guido van Rossumd57fd912000-03-10 22:53:23 +00007522
7523 Py_DECREF(substring);
Thomas Wouters477c8d52006-05-27 19:21:47 +00007524
Guido van Rossumd57fd912000-03-10 22:53:23 +00007525 return result;
7526}
7527
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007528PyDoc_STRVAR(encode__doc__,
Victor Stinnerc911bbf2010-11-07 19:04:46 +00007529 "S.encode(encoding='utf-8', errors='strict') -> bytes\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +00007530\n\
Victor Stinnere14e2122010-11-07 18:41:46 +00007531Encode S using the codec registered for encoding. Default encoding\n\
7532is 'utf-8'. errors may be given to set a different error\n\
Fred Drakee4315f52000-05-09 19:53:39 +00007533handling scheme. Default is 'strict' meaning that encoding errors raise\n\
Walter Dörwald3aeb6322002-09-02 13:14:32 +00007534a UnicodeEncodeError. Other possible values are 'ignore', 'replace' and\n\
7535'xmlcharrefreplace' as well as any other name registered with\n\
7536codecs.register_error that can handle UnicodeEncodeErrors.");
Guido van Rossumd57fd912000-03-10 22:53:23 +00007537
7538static PyObject *
Benjamin Peterson308d6372009-09-18 21:42:35 +00007539unicode_encode(PyUnicodeObject *self, PyObject *args, PyObject *kwargs)
Guido van Rossumd57fd912000-03-10 22:53:23 +00007540{
Benjamin Peterson308d6372009-09-18 21:42:35 +00007541 static char *kwlist[] = {"encoding", "errors", 0};
Guido van Rossumd57fd912000-03-10 22:53:23 +00007542 char *encoding = NULL;
7543 char *errors = NULL;
Guido van Rossum35d94282007-08-27 18:20:11 +00007544
Benjamin Peterson308d6372009-09-18 21:42:35 +00007545 if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|ss:encode",
7546 kwlist, &encoding, &errors))
Guido van Rossumd57fd912000-03-10 22:53:23 +00007547 return NULL;
Georg Brandl3b9406b2010-12-03 07:54:09 +00007548 return PyUnicode_AsEncodedString((PyObject *)self, encoding, errors);
Marc-André Lemburgd2d45982004-07-08 17:57:32 +00007549}
7550
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007551PyDoc_STRVAR(expandtabs__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +00007552 "S.expandtabs([tabsize]) -> str\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +00007553\n\
7554Return a copy of S where all tab characters are expanded using spaces.\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007555If tabsize is not given, a tab size of 8 characters is assumed.");
Guido van Rossumd57fd912000-03-10 22:53:23 +00007556
7557static PyObject*
7558unicode_expandtabs(PyUnicodeObject *self, PyObject *args)
7559{
7560 Py_UNICODE *e;
7561 Py_UNICODE *p;
7562 Py_UNICODE *q;
Christian Heimesdd15f6c2008-03-16 00:07:10 +00007563 Py_UNICODE *qe;
7564 Py_ssize_t i, j, incr;
Guido van Rossumd57fd912000-03-10 22:53:23 +00007565 PyUnicodeObject *u;
7566 int tabsize = 8;
7567
7568 if (!PyArg_ParseTuple(args, "|i:expandtabs", &tabsize))
Benjamin Peterson29060642009-01-31 22:14:21 +00007569 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00007570
Thomas Wouters7e474022000-07-16 12:04:32 +00007571 /* First pass: determine size of output string */
Christian Heimesdd15f6c2008-03-16 00:07:10 +00007572 i = 0; /* chars up to and including most recent \n or \r */
7573 j = 0; /* chars since most recent \n or \r (use in tab calculations) */
7574 e = self->str + self->length; /* end of input */
Guido van Rossumd57fd912000-03-10 22:53:23 +00007575 for (p = self->str; p < e; p++)
7576 if (*p == '\t') {
Benjamin Peterson29060642009-01-31 22:14:21 +00007577 if (tabsize > 0) {
7578 incr = tabsize - (j % tabsize); /* cannot overflow */
7579 if (j > PY_SSIZE_T_MAX - incr)
7580 goto overflow1;
7581 j += incr;
Christian Heimesdd15f6c2008-03-16 00:07:10 +00007582 }
Benjamin Peterson29060642009-01-31 22:14:21 +00007583 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00007584 else {
Benjamin Peterson29060642009-01-31 22:14:21 +00007585 if (j > PY_SSIZE_T_MAX - 1)
7586 goto overflow1;
Guido van Rossumd57fd912000-03-10 22:53:23 +00007587 j++;
7588 if (*p == '\n' || *p == '\r') {
Benjamin Peterson29060642009-01-31 22:14:21 +00007589 if (i > PY_SSIZE_T_MAX - j)
7590 goto overflow1;
Guido van Rossumd57fd912000-03-10 22:53:23 +00007591 i += j;
Christian Heimesdd15f6c2008-03-16 00:07:10 +00007592 j = 0;
Guido van Rossumd57fd912000-03-10 22:53:23 +00007593 }
7594 }
7595
Christian Heimesdd15f6c2008-03-16 00:07:10 +00007596 if (i > PY_SSIZE_T_MAX - j)
Benjamin Peterson29060642009-01-31 22:14:21 +00007597 goto overflow1;
Guido van Rossumcd16bf62007-06-13 18:07:49 +00007598
Guido van Rossumd57fd912000-03-10 22:53:23 +00007599 /* Second pass: create output string and fill it */
7600 u = _PyUnicode_New(i + j);
7601 if (!u)
7602 return NULL;
7603
Christian Heimesdd15f6c2008-03-16 00:07:10 +00007604 j = 0; /* same as in first pass */
7605 q = u->str; /* next output char */
7606 qe = u->str + u->length; /* end of output */
Guido van Rossumd57fd912000-03-10 22:53:23 +00007607
7608 for (p = self->str; p < e; p++)
7609 if (*p == '\t') {
Benjamin Peterson29060642009-01-31 22:14:21 +00007610 if (tabsize > 0) {
7611 i = tabsize - (j % tabsize);
7612 j += i;
7613 while (i--) {
7614 if (q >= qe)
7615 goto overflow2;
7616 *q++ = ' ';
Christian Heimesdd15f6c2008-03-16 00:07:10 +00007617 }
Benjamin Peterson29060642009-01-31 22:14:21 +00007618 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00007619 }
Benjamin Peterson29060642009-01-31 22:14:21 +00007620 else {
7621 if (q >= qe)
7622 goto overflow2;
7623 *q++ = *p;
Christian Heimesdd15f6c2008-03-16 00:07:10 +00007624 j++;
Guido van Rossumd57fd912000-03-10 22:53:23 +00007625 if (*p == '\n' || *p == '\r')
7626 j = 0;
7627 }
7628
7629 return (PyObject*) u;
Christian Heimesdd15f6c2008-03-16 00:07:10 +00007630
7631 overflow2:
7632 Py_DECREF(u);
7633 overflow1:
7634 PyErr_SetString(PyExc_OverflowError, "new string is too long");
7635 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00007636}
7637
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007638PyDoc_STRVAR(find__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +00007639 "S.find(sub[, start[, end]]) -> int\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +00007640\n\
7641Return the lowest index in S where substring sub is found,\n\
Senthil Kumaran53516a82011-07-27 23:33:54 +08007642such that sub is contained within S[start:end]. Optional\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +00007643arguments start and end are interpreted as in slice notation.\n\
7644\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007645Return -1 on failure.");
Guido van Rossumd57fd912000-03-10 22:53:23 +00007646
7647static PyObject *
7648unicode_find(PyUnicodeObject *self, PyObject *args)
7649{
Jesus Ceaac451502011-04-20 17:09:23 +02007650 PyUnicodeObject *substring;
Christian Heimes9cd17752007-11-18 19:35:23 +00007651 Py_ssize_t start;
7652 Py_ssize_t end;
Thomas Wouters477c8d52006-05-27 19:21:47 +00007653 Py_ssize_t result;
Guido van Rossumd57fd912000-03-10 22:53:23 +00007654
Jesus Ceaac451502011-04-20 17:09:23 +02007655 if (!stringlib_parse_args_finds_unicode("find", args, &substring,
7656 &start, &end))
Guido van Rossumd57fd912000-03-10 22:53:23 +00007657 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00007658
Thomas Wouters477c8d52006-05-27 19:21:47 +00007659 result = stringlib_find_slice(
7660 PyUnicode_AS_UNICODE(self), PyUnicode_GET_SIZE(self),
7661 PyUnicode_AS_UNICODE(substring), PyUnicode_GET_SIZE(substring),
7662 start, end
7663 );
Guido van Rossumd57fd912000-03-10 22:53:23 +00007664
7665 Py_DECREF(substring);
Thomas Wouters477c8d52006-05-27 19:21:47 +00007666
Christian Heimes217cfd12007-12-02 14:31:20 +00007667 return PyLong_FromSsize_t(result);
Guido van Rossumd57fd912000-03-10 22:53:23 +00007668}
7669
7670static PyObject *
Martin v. Löwis18e16552006-02-15 17:27:45 +00007671unicode_getitem(PyUnicodeObject *self, Py_ssize_t index)
Guido van Rossumd57fd912000-03-10 22:53:23 +00007672{
7673 if (index < 0 || index >= self->length) {
7674 PyErr_SetString(PyExc_IndexError, "string index out of range");
7675 return NULL;
7676 }
7677
7678 return (PyObject*) PyUnicode_FromUnicode(&self->str[index], 1);
7679}
7680
Guido van Rossumc2504932007-09-18 19:42:40 +00007681/* Believe it or not, this produces the same value for ASCII strings
7682 as string_hash(). */
Benjamin Peterson8f67d082010-10-17 20:54:53 +00007683static Py_hash_t
Neil Schemenauerf8c37d12007-09-07 20:49:04 +00007684unicode_hash(PyUnicodeObject *self)
Guido van Rossumd57fd912000-03-10 22:53:23 +00007685{
Guido van Rossumc2504932007-09-18 19:42:40 +00007686 Py_ssize_t len;
7687 Py_UNICODE *p;
Benjamin Peterson8f67d082010-10-17 20:54:53 +00007688 Py_hash_t x;
Guido van Rossumc2504932007-09-18 19:42:40 +00007689
Benjamin Petersonf6622c82012-04-09 14:53:07 -04007690#ifdef Py_DEBUG
Benjamin Peterson69e97272012-02-21 11:08:50 -05007691 assert(_Py_HashSecret_Initialized);
Benjamin Petersonf6622c82012-04-09 14:53:07 -04007692#endif
Guido van Rossumc2504932007-09-18 19:42:40 +00007693 if (self->hash != -1)
7694 return self->hash;
Christian Heimes90aa7642007-12-19 02:45:37 +00007695 len = Py_SIZE(self);
Georg Brandl2daf6ae2012-02-20 19:54:16 +01007696 /*
7697 We make the hash of the empty string be 0, rather than using
7698 (prefix ^ suffix), since this slightly obfuscates the hash secret
7699 */
7700 if (len == 0) {
7701 self->hash = 0;
7702 return 0;
7703 }
Guido van Rossumc2504932007-09-18 19:42:40 +00007704 p = self->str;
Georg Brandl2daf6ae2012-02-20 19:54:16 +01007705 x = _Py_HashSecret.prefix;
7706 x ^= *p << 7;
Guido van Rossumc2504932007-09-18 19:42:40 +00007707 while (--len >= 0)
Gregory P. Smith63e6c322012-01-14 15:31:34 -08007708 x = (_PyHASH_MULTIPLIER*x) ^ *p++;
Christian Heimes90aa7642007-12-19 02:45:37 +00007709 x ^= Py_SIZE(self);
Georg Brandl2daf6ae2012-02-20 19:54:16 +01007710 x ^= _Py_HashSecret.suffix;
Guido van Rossumc2504932007-09-18 19:42:40 +00007711 if (x == -1)
7712 x = -2;
7713 self->hash = x;
7714 return x;
Guido van Rossumd57fd912000-03-10 22:53:23 +00007715}
7716
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007717PyDoc_STRVAR(index__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +00007718 "S.index(sub[, start[, end]]) -> int\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +00007719\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007720Like S.find() but raise ValueError when the substring is not found.");
Guido van Rossumd57fd912000-03-10 22:53:23 +00007721
7722static PyObject *
7723unicode_index(PyUnicodeObject *self, PyObject *args)
7724{
Martin v. Löwis18e16552006-02-15 17:27:45 +00007725 Py_ssize_t result;
Jesus Ceaac451502011-04-20 17:09:23 +02007726 PyUnicodeObject *substring;
Christian Heimes9cd17752007-11-18 19:35:23 +00007727 Py_ssize_t start;
7728 Py_ssize_t end;
Guido van Rossumd57fd912000-03-10 22:53:23 +00007729
Jesus Ceaac451502011-04-20 17:09:23 +02007730 if (!stringlib_parse_args_finds_unicode("index", args, &substring,
7731 &start, &end))
Guido van Rossumd57fd912000-03-10 22:53:23 +00007732 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00007733
Thomas Wouters477c8d52006-05-27 19:21:47 +00007734 result = stringlib_find_slice(
7735 PyUnicode_AS_UNICODE(self), PyUnicode_GET_SIZE(self),
7736 PyUnicode_AS_UNICODE(substring), PyUnicode_GET_SIZE(substring),
7737 start, end
7738 );
Guido van Rossumd57fd912000-03-10 22:53:23 +00007739
7740 Py_DECREF(substring);
Thomas Wouters477c8d52006-05-27 19:21:47 +00007741
Guido van Rossumd57fd912000-03-10 22:53:23 +00007742 if (result < 0) {
7743 PyErr_SetString(PyExc_ValueError, "substring not found");
7744 return NULL;
7745 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00007746
Christian Heimes217cfd12007-12-02 14:31:20 +00007747 return PyLong_FromSsize_t(result);
Guido van Rossumd57fd912000-03-10 22:53:23 +00007748}
7749
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007750PyDoc_STRVAR(islower__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +00007751 "S.islower() -> bool\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +00007752\n\
Guido van Rossum77f6a652002-04-03 22:41:51 +00007753Return True if all cased characters in S are lowercase and there is\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007754at least one cased character in S, False otherwise.");
Guido van Rossumd57fd912000-03-10 22:53:23 +00007755
7756static PyObject*
Martin v. Löwise3eb1f22001-08-16 13:15:00 +00007757unicode_islower(PyUnicodeObject *self)
Guido van Rossumd57fd912000-03-10 22:53:23 +00007758{
7759 register const Py_UNICODE *p = PyUnicode_AS_UNICODE(self);
7760 register const Py_UNICODE *e;
7761 int cased;
7762
Guido van Rossumd57fd912000-03-10 22:53:23 +00007763 /* Shortcut for single character strings */
7764 if (PyUnicode_GET_SIZE(self) == 1)
Benjamin Peterson29060642009-01-31 22:14:21 +00007765 return PyBool_FromLong(Py_UNICODE_ISLOWER(*p));
Guido van Rossumd57fd912000-03-10 22:53:23 +00007766
Marc-André Lemburg60bc8092000-06-14 09:18:32 +00007767 /* Special case for empty strings */
Martin v. Löwisdea59e52006-01-05 10:00:36 +00007768 if (PyUnicode_GET_SIZE(self) == 0)
Benjamin Peterson29060642009-01-31 22:14:21 +00007769 return PyBool_FromLong(0);
Marc-André Lemburg60bc8092000-06-14 09:18:32 +00007770
Guido van Rossumd57fd912000-03-10 22:53:23 +00007771 e = p + PyUnicode_GET_SIZE(self);
7772 cased = 0;
Ezio Melotti93e7afc2011-08-22 14:08:38 +03007773 while (p < e) {
7774 const Py_UCS4 ch = _Py_UNICODE_NEXT(p, e);
Tim Petersced69f82003-09-16 20:30:58 +00007775
Benjamin Peterson29060642009-01-31 22:14:21 +00007776 if (Py_UNICODE_ISUPPER(ch) || Py_UNICODE_ISTITLE(ch))
7777 return PyBool_FromLong(0);
7778 else if (!cased && Py_UNICODE_ISLOWER(ch))
7779 cased = 1;
Guido van Rossumd57fd912000-03-10 22:53:23 +00007780 }
Guido van Rossum77f6a652002-04-03 22:41:51 +00007781 return PyBool_FromLong(cased);
Guido van Rossumd57fd912000-03-10 22:53:23 +00007782}
7783
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007784PyDoc_STRVAR(isupper__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +00007785 "S.isupper() -> bool\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +00007786\n\
Martin v. Löwis6828e182003-10-18 09:55:08 +00007787Return True if all cased characters in S are uppercase and there is\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007788at least one cased character in S, False otherwise.");
Guido van Rossumd57fd912000-03-10 22:53:23 +00007789
7790static PyObject*
Martin v. Löwise3eb1f22001-08-16 13:15:00 +00007791unicode_isupper(PyUnicodeObject *self)
Guido van Rossumd57fd912000-03-10 22:53:23 +00007792{
7793 register const Py_UNICODE *p = PyUnicode_AS_UNICODE(self);
7794 register const Py_UNICODE *e;
7795 int cased;
7796
Guido van Rossumd57fd912000-03-10 22:53:23 +00007797 /* Shortcut for single character strings */
7798 if (PyUnicode_GET_SIZE(self) == 1)
Benjamin Peterson29060642009-01-31 22:14:21 +00007799 return PyBool_FromLong(Py_UNICODE_ISUPPER(*p) != 0);
Guido van Rossumd57fd912000-03-10 22:53:23 +00007800
Marc-André Lemburg60bc8092000-06-14 09:18:32 +00007801 /* Special case for empty strings */
Martin v. Löwisdea59e52006-01-05 10:00:36 +00007802 if (PyUnicode_GET_SIZE(self) == 0)
Benjamin Peterson29060642009-01-31 22:14:21 +00007803 return PyBool_FromLong(0);
Marc-André Lemburg60bc8092000-06-14 09:18:32 +00007804
Guido van Rossumd57fd912000-03-10 22:53:23 +00007805 e = p + PyUnicode_GET_SIZE(self);
7806 cased = 0;
Ezio Melotti93e7afc2011-08-22 14:08:38 +03007807 while (p < e) {
7808 const Py_UCS4 ch = _Py_UNICODE_NEXT(p, e);
Tim Petersced69f82003-09-16 20:30:58 +00007809
Benjamin Peterson29060642009-01-31 22:14:21 +00007810 if (Py_UNICODE_ISLOWER(ch) || Py_UNICODE_ISTITLE(ch))
7811 return PyBool_FromLong(0);
7812 else if (!cased && Py_UNICODE_ISUPPER(ch))
7813 cased = 1;
Guido van Rossumd57fd912000-03-10 22:53:23 +00007814 }
Guido van Rossum77f6a652002-04-03 22:41:51 +00007815 return PyBool_FromLong(cased);
Guido van Rossumd57fd912000-03-10 22:53:23 +00007816}
7817
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007818PyDoc_STRVAR(istitle__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +00007819 "S.istitle() -> bool\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +00007820\n\
Martin v. Löwis6828e182003-10-18 09:55:08 +00007821Return True if S is a titlecased string and there is at least one\n\
7822character in S, i.e. upper- and titlecase characters may only\n\
7823follow uncased characters and lowercase characters only cased ones.\n\
7824Return False otherwise.");
Guido van Rossumd57fd912000-03-10 22:53:23 +00007825
7826static PyObject*
Martin v. Löwise3eb1f22001-08-16 13:15:00 +00007827unicode_istitle(PyUnicodeObject *self)
Guido van Rossumd57fd912000-03-10 22:53:23 +00007828{
7829 register const Py_UNICODE *p = PyUnicode_AS_UNICODE(self);
7830 register const Py_UNICODE *e;
7831 int cased, previous_is_cased;
7832
Guido van Rossumd57fd912000-03-10 22:53:23 +00007833 /* Shortcut for single character strings */
7834 if (PyUnicode_GET_SIZE(self) == 1)
Benjamin Peterson29060642009-01-31 22:14:21 +00007835 return PyBool_FromLong((Py_UNICODE_ISTITLE(*p) != 0) ||
7836 (Py_UNICODE_ISUPPER(*p) != 0));
Guido van Rossumd57fd912000-03-10 22:53:23 +00007837
Marc-André Lemburg60bc8092000-06-14 09:18:32 +00007838 /* Special case for empty strings */
Martin v. Löwisdea59e52006-01-05 10:00:36 +00007839 if (PyUnicode_GET_SIZE(self) == 0)
Benjamin Peterson29060642009-01-31 22:14:21 +00007840 return PyBool_FromLong(0);
Marc-André Lemburg60bc8092000-06-14 09:18:32 +00007841
Guido van Rossumd57fd912000-03-10 22:53:23 +00007842 e = p + PyUnicode_GET_SIZE(self);
7843 cased = 0;
7844 previous_is_cased = 0;
Ezio Melotti93e7afc2011-08-22 14:08:38 +03007845 while (p < e) {
7846 const Py_UCS4 ch = _Py_UNICODE_NEXT(p, e);
Tim Petersced69f82003-09-16 20:30:58 +00007847
Benjamin Peterson29060642009-01-31 22:14:21 +00007848 if (Py_UNICODE_ISUPPER(ch) || Py_UNICODE_ISTITLE(ch)) {
7849 if (previous_is_cased)
7850 return PyBool_FromLong(0);
7851 previous_is_cased = 1;
7852 cased = 1;
7853 }
7854 else if (Py_UNICODE_ISLOWER(ch)) {
7855 if (!previous_is_cased)
7856 return PyBool_FromLong(0);
7857 previous_is_cased = 1;
7858 cased = 1;
7859 }
7860 else
7861 previous_is_cased = 0;
Guido van Rossumd57fd912000-03-10 22:53:23 +00007862 }
Guido van Rossum77f6a652002-04-03 22:41:51 +00007863 return PyBool_FromLong(cased);
Guido van Rossumd57fd912000-03-10 22:53:23 +00007864}
7865
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007866PyDoc_STRVAR(isspace__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +00007867 "S.isspace() -> bool\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +00007868\n\
Martin v. Löwis6828e182003-10-18 09:55:08 +00007869Return True if all characters in S are whitespace\n\
7870and there is at least one character in S, False otherwise.");
Guido van Rossumd57fd912000-03-10 22:53:23 +00007871
7872static PyObject*
Martin v. Löwise3eb1f22001-08-16 13:15:00 +00007873unicode_isspace(PyUnicodeObject *self)
Guido van Rossumd57fd912000-03-10 22:53:23 +00007874{
7875 register const Py_UNICODE *p = PyUnicode_AS_UNICODE(self);
7876 register const Py_UNICODE *e;
7877
Guido van Rossumd57fd912000-03-10 22:53:23 +00007878 /* Shortcut for single character strings */
7879 if (PyUnicode_GET_SIZE(self) == 1 &&
Benjamin Peterson29060642009-01-31 22:14:21 +00007880 Py_UNICODE_ISSPACE(*p))
7881 return PyBool_FromLong(1);
Guido van Rossumd57fd912000-03-10 22:53:23 +00007882
Marc-André Lemburg60bc8092000-06-14 09:18:32 +00007883 /* Special case for empty strings */
Martin v. Löwisdea59e52006-01-05 10:00:36 +00007884 if (PyUnicode_GET_SIZE(self) == 0)
Benjamin Peterson29060642009-01-31 22:14:21 +00007885 return PyBool_FromLong(0);
Marc-André Lemburg60bc8092000-06-14 09:18:32 +00007886
Guido van Rossumd57fd912000-03-10 22:53:23 +00007887 e = p + PyUnicode_GET_SIZE(self);
Ezio Melotti93e7afc2011-08-22 14:08:38 +03007888 while (p < e) {
7889 const Py_UCS4 ch = _Py_UNICODE_NEXT(p, e);
7890 if (!Py_UNICODE_ISSPACE(ch))
Benjamin Peterson29060642009-01-31 22:14:21 +00007891 return PyBool_FromLong(0);
Guido van Rossumd57fd912000-03-10 22:53:23 +00007892 }
Guido van Rossum77f6a652002-04-03 22:41:51 +00007893 return PyBool_FromLong(1);
Guido van Rossumd57fd912000-03-10 22:53:23 +00007894}
7895
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007896PyDoc_STRVAR(isalpha__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +00007897 "S.isalpha() -> bool\n\
Marc-André Lemburga7acf422000-07-05 09:49:44 +00007898\n\
Martin v. Löwis6828e182003-10-18 09:55:08 +00007899Return True if all characters in S are alphabetic\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007900and there is at least one character in S, False otherwise.");
Marc-André Lemburga7acf422000-07-05 09:49:44 +00007901
7902static PyObject*
Martin v. Löwise3eb1f22001-08-16 13:15:00 +00007903unicode_isalpha(PyUnicodeObject *self)
Marc-André Lemburga7acf422000-07-05 09:49:44 +00007904{
7905 register const Py_UNICODE *p = PyUnicode_AS_UNICODE(self);
7906 register const Py_UNICODE *e;
7907
Marc-André Lemburga7acf422000-07-05 09:49:44 +00007908 /* Shortcut for single character strings */
7909 if (PyUnicode_GET_SIZE(self) == 1 &&
Benjamin Peterson29060642009-01-31 22:14:21 +00007910 Py_UNICODE_ISALPHA(*p))
7911 return PyBool_FromLong(1);
Marc-André Lemburga7acf422000-07-05 09:49:44 +00007912
7913 /* Special case for empty strings */
Martin v. Löwisdea59e52006-01-05 10:00:36 +00007914 if (PyUnicode_GET_SIZE(self) == 0)
Benjamin Peterson29060642009-01-31 22:14:21 +00007915 return PyBool_FromLong(0);
Marc-André Lemburga7acf422000-07-05 09:49:44 +00007916
7917 e = p + PyUnicode_GET_SIZE(self);
Ezio Melotti93e7afc2011-08-22 14:08:38 +03007918 while (p < e) {
7919 if (!Py_UNICODE_ISALPHA(_Py_UNICODE_NEXT(p, e)))
Benjamin Peterson29060642009-01-31 22:14:21 +00007920 return PyBool_FromLong(0);
Marc-André Lemburga7acf422000-07-05 09:49:44 +00007921 }
Guido van Rossum77f6a652002-04-03 22:41:51 +00007922 return PyBool_FromLong(1);
Marc-André Lemburga7acf422000-07-05 09:49:44 +00007923}
7924
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007925PyDoc_STRVAR(isalnum__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +00007926 "S.isalnum() -> bool\n\
Marc-André Lemburga7acf422000-07-05 09:49:44 +00007927\n\
Martin v. Löwis6828e182003-10-18 09:55:08 +00007928Return True if all characters in S are alphanumeric\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007929and there is at least one character in S, False otherwise.");
Marc-André Lemburga7acf422000-07-05 09:49:44 +00007930
7931static PyObject*
Martin v. Löwise3eb1f22001-08-16 13:15:00 +00007932unicode_isalnum(PyUnicodeObject *self)
Marc-André Lemburga7acf422000-07-05 09:49:44 +00007933{
7934 register const Py_UNICODE *p = PyUnicode_AS_UNICODE(self);
7935 register const Py_UNICODE *e;
7936
Marc-André Lemburga7acf422000-07-05 09:49:44 +00007937 /* Shortcut for single character strings */
7938 if (PyUnicode_GET_SIZE(self) == 1 &&
Benjamin Peterson29060642009-01-31 22:14:21 +00007939 Py_UNICODE_ISALNUM(*p))
7940 return PyBool_FromLong(1);
Marc-André Lemburga7acf422000-07-05 09:49:44 +00007941
7942 /* Special case for empty strings */
Martin v. Löwisdea59e52006-01-05 10:00:36 +00007943 if (PyUnicode_GET_SIZE(self) == 0)
Benjamin Peterson29060642009-01-31 22:14:21 +00007944 return PyBool_FromLong(0);
Marc-André Lemburga7acf422000-07-05 09:49:44 +00007945
7946 e = p + PyUnicode_GET_SIZE(self);
Ezio Melotti93e7afc2011-08-22 14:08:38 +03007947 while (p < e) {
7948 const Py_UCS4 ch = _Py_UNICODE_NEXT(p, e);
7949 if (!Py_UNICODE_ISALNUM(ch))
Benjamin Peterson29060642009-01-31 22:14:21 +00007950 return PyBool_FromLong(0);
Marc-André Lemburga7acf422000-07-05 09:49:44 +00007951 }
Guido van Rossum77f6a652002-04-03 22:41:51 +00007952 return PyBool_FromLong(1);
Marc-André Lemburga7acf422000-07-05 09:49:44 +00007953}
7954
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007955PyDoc_STRVAR(isdecimal__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +00007956 "S.isdecimal() -> bool\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +00007957\n\
Guido van Rossum77f6a652002-04-03 22:41:51 +00007958Return True if there are only decimal characters in S,\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007959False otherwise.");
Guido van Rossumd57fd912000-03-10 22:53:23 +00007960
7961static PyObject*
Martin v. Löwise3eb1f22001-08-16 13:15:00 +00007962unicode_isdecimal(PyUnicodeObject *self)
Guido van Rossumd57fd912000-03-10 22:53:23 +00007963{
7964 register const Py_UNICODE *p = PyUnicode_AS_UNICODE(self);
7965 register const Py_UNICODE *e;
7966
Guido van Rossumd57fd912000-03-10 22:53:23 +00007967 /* Shortcut for single character strings */
7968 if (PyUnicode_GET_SIZE(self) == 1 &&
Benjamin Peterson29060642009-01-31 22:14:21 +00007969 Py_UNICODE_ISDECIMAL(*p))
7970 return PyBool_FromLong(1);
Guido van Rossumd57fd912000-03-10 22:53:23 +00007971
Marc-André Lemburg60bc8092000-06-14 09:18:32 +00007972 /* Special case for empty strings */
Martin v. Löwisdea59e52006-01-05 10:00:36 +00007973 if (PyUnicode_GET_SIZE(self) == 0)
Benjamin Peterson29060642009-01-31 22:14:21 +00007974 return PyBool_FromLong(0);
Marc-André Lemburg60bc8092000-06-14 09:18:32 +00007975
Guido van Rossumd57fd912000-03-10 22:53:23 +00007976 e = p + PyUnicode_GET_SIZE(self);
Ezio Melotti93e7afc2011-08-22 14:08:38 +03007977 while (p < e) {
7978 if (!Py_UNICODE_ISDECIMAL(_Py_UNICODE_NEXT(p, e)))
Benjamin Peterson29060642009-01-31 22:14:21 +00007979 return PyBool_FromLong(0);
Guido van Rossumd57fd912000-03-10 22:53:23 +00007980 }
Guido van Rossum77f6a652002-04-03 22:41:51 +00007981 return PyBool_FromLong(1);
Guido van Rossumd57fd912000-03-10 22:53:23 +00007982}
7983
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00007984PyDoc_STRVAR(isdigit__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +00007985 "S.isdigit() -> bool\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +00007986\n\
Martin v. Löwis6828e182003-10-18 09:55:08 +00007987Return True if all characters in S are digits\n\
7988and there is at least one character in S, False otherwise.");
Guido van Rossumd57fd912000-03-10 22:53:23 +00007989
7990static PyObject*
Martin v. Löwise3eb1f22001-08-16 13:15:00 +00007991unicode_isdigit(PyUnicodeObject *self)
Guido van Rossumd57fd912000-03-10 22:53:23 +00007992{
7993 register const Py_UNICODE *p = PyUnicode_AS_UNICODE(self);
7994 register const Py_UNICODE *e;
7995
Guido van Rossumd57fd912000-03-10 22:53:23 +00007996 /* Shortcut for single character strings */
7997 if (PyUnicode_GET_SIZE(self) == 1 &&
Benjamin Peterson29060642009-01-31 22:14:21 +00007998 Py_UNICODE_ISDIGIT(*p))
7999 return PyBool_FromLong(1);
Guido van Rossumd57fd912000-03-10 22:53:23 +00008000
Marc-André Lemburg60bc8092000-06-14 09:18:32 +00008001 /* Special case for empty strings */
Martin v. Löwisdea59e52006-01-05 10:00:36 +00008002 if (PyUnicode_GET_SIZE(self) == 0)
Benjamin Peterson29060642009-01-31 22:14:21 +00008003 return PyBool_FromLong(0);
Marc-André Lemburg60bc8092000-06-14 09:18:32 +00008004
Guido van Rossumd57fd912000-03-10 22:53:23 +00008005 e = p + PyUnicode_GET_SIZE(self);
Ezio Melotti93e7afc2011-08-22 14:08:38 +03008006 while (p < e) {
8007 if (!Py_UNICODE_ISDIGIT(_Py_UNICODE_NEXT(p, e)))
Benjamin Peterson29060642009-01-31 22:14:21 +00008008 return PyBool_FromLong(0);
Guido van Rossumd57fd912000-03-10 22:53:23 +00008009 }
Guido van Rossum77f6a652002-04-03 22:41:51 +00008010 return PyBool_FromLong(1);
Guido van Rossumd57fd912000-03-10 22:53:23 +00008011}
8012
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008013PyDoc_STRVAR(isnumeric__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +00008014 "S.isnumeric() -> bool\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +00008015\n\
Guido van Rossum77f6a652002-04-03 22:41:51 +00008016Return True if there are only numeric characters in S,\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008017False otherwise.");
Guido van Rossumd57fd912000-03-10 22:53:23 +00008018
8019static PyObject*
Martin v. Löwise3eb1f22001-08-16 13:15:00 +00008020unicode_isnumeric(PyUnicodeObject *self)
Guido van Rossumd57fd912000-03-10 22:53:23 +00008021{
8022 register const Py_UNICODE *p = PyUnicode_AS_UNICODE(self);
8023 register const Py_UNICODE *e;
8024
Guido van Rossumd57fd912000-03-10 22:53:23 +00008025 /* Shortcut for single character strings */
8026 if (PyUnicode_GET_SIZE(self) == 1 &&
Benjamin Peterson29060642009-01-31 22:14:21 +00008027 Py_UNICODE_ISNUMERIC(*p))
8028 return PyBool_FromLong(1);
Guido van Rossumd57fd912000-03-10 22:53:23 +00008029
Marc-André Lemburg60bc8092000-06-14 09:18:32 +00008030 /* Special case for empty strings */
Martin v. Löwisdea59e52006-01-05 10:00:36 +00008031 if (PyUnicode_GET_SIZE(self) == 0)
Benjamin Peterson29060642009-01-31 22:14:21 +00008032 return PyBool_FromLong(0);
Marc-André Lemburg60bc8092000-06-14 09:18:32 +00008033
Guido van Rossumd57fd912000-03-10 22:53:23 +00008034 e = p + PyUnicode_GET_SIZE(self);
Ezio Melotti93e7afc2011-08-22 14:08:38 +03008035 while (p < e) {
8036 if (!Py_UNICODE_ISNUMERIC(_Py_UNICODE_NEXT(p, e)))
Benjamin Peterson29060642009-01-31 22:14:21 +00008037 return PyBool_FromLong(0);
Guido van Rossumd57fd912000-03-10 22:53:23 +00008038 }
Guido van Rossum77f6a652002-04-03 22:41:51 +00008039 return PyBool_FromLong(1);
Guido van Rossumd57fd912000-03-10 22:53:23 +00008040}
8041
Martin v. Löwis47383402007-08-15 07:32:56 +00008042int
8043PyUnicode_IsIdentifier(PyObject *self)
8044{
Benjamin Petersonf413b802011-08-12 22:17:18 -05008045 const Py_UNICODE *p = PyUnicode_AS_UNICODE((PyUnicodeObject*)self);
Ezio Melotti93e7afc2011-08-22 14:08:38 +03008046 const Py_UNICODE *e;
8047 Py_UCS4 first;
Martin v. Löwis47383402007-08-15 07:32:56 +00008048
8049 /* Special case for empty strings */
Ezio Melotti93e7afc2011-08-22 14:08:38 +03008050 if (PyUnicode_GET_SIZE(self) == 0)
Benjamin Peterson29060642009-01-31 22:14:21 +00008051 return 0;
Martin v. Löwis47383402007-08-15 07:32:56 +00008052
8053 /* PEP 3131 says that the first character must be in
8054 XID_Start and subsequent characters in XID_Continue,
8055 and for the ASCII range, the 2.x rules apply (i.e
Benjamin Peterson14339b62009-01-31 16:36:08 +00008056 start with letters and underscore, continue with
Martin v. Löwis47383402007-08-15 07:32:56 +00008057 letters, digits, underscore). However, given the current
8058 definition of XID_Start and XID_Continue, it is sufficient
8059 to check just for these, except that _ must be allowed
8060 as starting an identifier. */
Ezio Melotti93e7afc2011-08-22 14:08:38 +03008061 e = p + PyUnicode_GET_SIZE(self);
8062 first = _Py_UNICODE_NEXT(p, e);
Benjamin Petersonf413b802011-08-12 22:17:18 -05008063 if (!_PyUnicode_IsXidStart(first) && first != 0x5F /* LOW LINE */)
Martin v. Löwis47383402007-08-15 07:32:56 +00008064 return 0;
8065
Ezio Melotti93e7afc2011-08-22 14:08:38 +03008066 while (p < e)
8067 if (!_PyUnicode_IsXidContinue(_Py_UNICODE_NEXT(p, e)))
Benjamin Peterson29060642009-01-31 22:14:21 +00008068 return 0;
Martin v. Löwis47383402007-08-15 07:32:56 +00008069 return 1;
8070}
8071
8072PyDoc_STRVAR(isidentifier__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +00008073 "S.isidentifier() -> bool\n\
Martin v. Löwis47383402007-08-15 07:32:56 +00008074\n\
8075Return True if S is a valid identifier according\n\
8076to the language definition.");
8077
8078static PyObject*
8079unicode_isidentifier(PyObject *self)
8080{
8081 return PyBool_FromLong(PyUnicode_IsIdentifier(self));
8082}
8083
Georg Brandl559e5d72008-06-11 18:37:52 +00008084PyDoc_STRVAR(isprintable__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +00008085 "S.isprintable() -> bool\n\
Georg Brandl559e5d72008-06-11 18:37:52 +00008086\n\
8087Return True if all characters in S are considered\n\
8088printable in repr() or S is empty, False otherwise.");
8089
8090static PyObject*
8091unicode_isprintable(PyObject *self)
8092{
8093 register const Py_UNICODE *p = PyUnicode_AS_UNICODE(self);
8094 register const Py_UNICODE *e;
8095
8096 /* Shortcut for single character strings */
8097 if (PyUnicode_GET_SIZE(self) == 1 && Py_UNICODE_ISPRINTABLE(*p)) {
8098 Py_RETURN_TRUE;
8099 }
8100
8101 e = p + PyUnicode_GET_SIZE(self);
Ezio Melotti93e7afc2011-08-22 14:08:38 +03008102 while (p < e) {
8103 if (!Py_UNICODE_ISPRINTABLE(_Py_UNICODE_NEXT(p, e))) {
Georg Brandl559e5d72008-06-11 18:37:52 +00008104 Py_RETURN_FALSE;
8105 }
8106 }
8107 Py_RETURN_TRUE;
8108}
8109
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008110PyDoc_STRVAR(join__doc__,
Georg Brandl495f7b52009-10-27 15:28:25 +00008111 "S.join(iterable) -> str\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +00008112\n\
8113Return a string which is the concatenation of the strings in the\n\
Georg Brandl495f7b52009-10-27 15:28:25 +00008114iterable. The separator between elements is S.");
Guido van Rossumd57fd912000-03-10 22:53:23 +00008115
8116static PyObject*
Martin v. Löwise3eb1f22001-08-16 13:15:00 +00008117unicode_join(PyObject *self, PyObject *data)
Guido van Rossumd57fd912000-03-10 22:53:23 +00008118{
Martin v. Löwise3eb1f22001-08-16 13:15:00 +00008119 return PyUnicode_Join(self, data);
Guido van Rossumd57fd912000-03-10 22:53:23 +00008120}
8121
Martin v. Löwis18e16552006-02-15 17:27:45 +00008122static Py_ssize_t
Guido van Rossumd57fd912000-03-10 22:53:23 +00008123unicode_length(PyUnicodeObject *self)
8124{
8125 return self->length;
8126}
8127
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008128PyDoc_STRVAR(ljust__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +00008129 "S.ljust(width[, fillchar]) -> str\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +00008130\n\
Benjamin Petersonf10a79a2008-10-11 00:49:57 +00008131Return S left-justified in a Unicode string of length width. Padding is\n\
Raymond Hettinger4f8f9762003-11-26 08:21:35 +00008132done using the specified fill character (default is a space).");
Guido van Rossumd57fd912000-03-10 22:53:23 +00008133
8134static PyObject *
8135unicode_ljust(PyUnicodeObject *self, PyObject *args)
8136{
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00008137 Py_ssize_t width;
Raymond Hettinger4f8f9762003-11-26 08:21:35 +00008138 Py_UNICODE fillchar = ' ';
8139
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00008140 if (!PyArg_ParseTuple(args, "n|O&:ljust", &width, convert_uc, &fillchar))
Guido van Rossumd57fd912000-03-10 22:53:23 +00008141 return NULL;
8142
Tim Peters7a29bd52001-09-12 03:03:31 +00008143 if (self->length >= width && PyUnicode_CheckExact(self)) {
Guido van Rossumd57fd912000-03-10 22:53:23 +00008144 Py_INCREF(self);
8145 return (PyObject*) self;
8146 }
8147
Raymond Hettinger4f8f9762003-11-26 08:21:35 +00008148 return (PyObject*) pad(self, 0, width - self->length, fillchar);
Guido van Rossumd57fd912000-03-10 22:53:23 +00008149}
8150
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008151PyDoc_STRVAR(lower__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +00008152 "S.lower() -> str\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +00008153\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008154Return a copy of the string S converted to lowercase.");
Guido van Rossumd57fd912000-03-10 22:53:23 +00008155
8156static PyObject*
Martin v. Löwise3eb1f22001-08-16 13:15:00 +00008157unicode_lower(PyUnicodeObject *self)
Guido van Rossumd57fd912000-03-10 22:53:23 +00008158{
Guido van Rossumd57fd912000-03-10 22:53:23 +00008159 return fixup(self, fixlower);
8160}
8161
Walter Dörwaldde02bcb2002-04-22 17:42:37 +00008162#define LEFTSTRIP 0
8163#define RIGHTSTRIP 1
8164#define BOTHSTRIP 2
8165
8166/* Arrays indexed by above */
8167static const char *stripformat[] = {"|O:lstrip", "|O:rstrip", "|O:strip"};
8168
8169#define STRIPNAME(i) (stripformat[i]+3)
8170
Walter Dörwaldde02bcb2002-04-22 17:42:37 +00008171/* externally visible for str.strip(unicode) */
8172PyObject *
8173_PyUnicode_XStrip(PyUnicodeObject *self, int striptype, PyObject *sepobj)
8174{
Benjamin Peterson14339b62009-01-31 16:36:08 +00008175 Py_UNICODE *s = PyUnicode_AS_UNICODE(self);
8176 Py_ssize_t len = PyUnicode_GET_SIZE(self);
8177 Py_UNICODE *sep = PyUnicode_AS_UNICODE(sepobj);
8178 Py_ssize_t seplen = PyUnicode_GET_SIZE(sepobj);
8179 Py_ssize_t i, j;
Walter Dörwaldde02bcb2002-04-22 17:42:37 +00008180
Benjamin Peterson29060642009-01-31 22:14:21 +00008181 BLOOM_MASK sepmask = make_bloom_mask(sep, seplen);
Thomas Wouters477c8d52006-05-27 19:21:47 +00008182
Benjamin Peterson14339b62009-01-31 16:36:08 +00008183 i = 0;
8184 if (striptype != RIGHTSTRIP) {
Benjamin Peterson29060642009-01-31 22:14:21 +00008185 while (i < len && BLOOM_MEMBER(sepmask, s[i], sep, seplen)) {
8186 i++;
8187 }
Benjamin Peterson14339b62009-01-31 16:36:08 +00008188 }
Walter Dörwaldde02bcb2002-04-22 17:42:37 +00008189
Benjamin Peterson14339b62009-01-31 16:36:08 +00008190 j = len;
8191 if (striptype != LEFTSTRIP) {
Benjamin Peterson29060642009-01-31 22:14:21 +00008192 do {
8193 j--;
8194 } while (j >= i && BLOOM_MEMBER(sepmask, s[j], sep, seplen));
8195 j++;
Benjamin Peterson14339b62009-01-31 16:36:08 +00008196 }
Walter Dörwaldde02bcb2002-04-22 17:42:37 +00008197
Benjamin Peterson14339b62009-01-31 16:36:08 +00008198 if (i == 0 && j == len && PyUnicode_CheckExact(self)) {
Benjamin Peterson29060642009-01-31 22:14:21 +00008199 Py_INCREF(self);
8200 return (PyObject*)self;
Benjamin Peterson14339b62009-01-31 16:36:08 +00008201 }
8202 else
Benjamin Peterson29060642009-01-31 22:14:21 +00008203 return PyUnicode_FromUnicode(s+i, j-i);
Walter Dörwaldde02bcb2002-04-22 17:42:37 +00008204}
8205
Guido van Rossumd57fd912000-03-10 22:53:23 +00008206
8207static PyObject *
Walter Dörwaldde02bcb2002-04-22 17:42:37 +00008208do_strip(PyUnicodeObject *self, int striptype)
Guido van Rossumd57fd912000-03-10 22:53:23 +00008209{
Benjamin Peterson14339b62009-01-31 16:36:08 +00008210 Py_UNICODE *s = PyUnicode_AS_UNICODE(self);
8211 Py_ssize_t len = PyUnicode_GET_SIZE(self), i, j;
Walter Dörwaldde02bcb2002-04-22 17:42:37 +00008212
Benjamin Peterson14339b62009-01-31 16:36:08 +00008213 i = 0;
8214 if (striptype != RIGHTSTRIP) {
8215 while (i < len && Py_UNICODE_ISSPACE(s[i])) {
8216 i++;
8217 }
8218 }
Walter Dörwaldde02bcb2002-04-22 17:42:37 +00008219
Benjamin Peterson14339b62009-01-31 16:36:08 +00008220 j = len;
8221 if (striptype != LEFTSTRIP) {
8222 do {
8223 j--;
8224 } while (j >= i && Py_UNICODE_ISSPACE(s[j]));
8225 j++;
8226 }
Walter Dörwaldde02bcb2002-04-22 17:42:37 +00008227
Benjamin Peterson14339b62009-01-31 16:36:08 +00008228 if (i == 0 && j == len && PyUnicode_CheckExact(self)) {
8229 Py_INCREF(self);
8230 return (PyObject*)self;
8231 }
8232 else
8233 return PyUnicode_FromUnicode(s+i, j-i);
Guido van Rossumd57fd912000-03-10 22:53:23 +00008234}
8235
Walter Dörwaldde02bcb2002-04-22 17:42:37 +00008236
8237static PyObject *
8238do_argstrip(PyUnicodeObject *self, int striptype, PyObject *args)
8239{
Benjamin Peterson14339b62009-01-31 16:36:08 +00008240 PyObject *sep = NULL;
Walter Dörwaldde02bcb2002-04-22 17:42:37 +00008241
Benjamin Peterson14339b62009-01-31 16:36:08 +00008242 if (!PyArg_ParseTuple(args, (char *)stripformat[striptype], &sep))
8243 return NULL;
Walter Dörwaldde02bcb2002-04-22 17:42:37 +00008244
Benjamin Peterson14339b62009-01-31 16:36:08 +00008245 if (sep != NULL && sep != Py_None) {
8246 if (PyUnicode_Check(sep))
8247 return _PyUnicode_XStrip(self, striptype, sep);
8248 else {
8249 PyErr_Format(PyExc_TypeError,
Benjamin Peterson29060642009-01-31 22:14:21 +00008250 "%s arg must be None or str",
8251 STRIPNAME(striptype));
Benjamin Peterson14339b62009-01-31 16:36:08 +00008252 return NULL;
8253 }
8254 }
Walter Dörwaldde02bcb2002-04-22 17:42:37 +00008255
Benjamin Peterson14339b62009-01-31 16:36:08 +00008256 return do_strip(self, striptype);
Walter Dörwaldde02bcb2002-04-22 17:42:37 +00008257}
8258
8259
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008260PyDoc_STRVAR(strip__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +00008261 "S.strip([chars]) -> str\n\
Walter Dörwaldde02bcb2002-04-22 17:42:37 +00008262\n\
8263Return a copy of the string S with leading and trailing\n\
8264whitespace removed.\n\
Benjamin Peterson142957c2008-07-04 19:55:29 +00008265If chars is given and not None, remove characters in chars instead.");
Walter Dörwaldde02bcb2002-04-22 17:42:37 +00008266
8267static PyObject *
8268unicode_strip(PyUnicodeObject *self, PyObject *args)
8269{
Benjamin Peterson14339b62009-01-31 16:36:08 +00008270 if (PyTuple_GET_SIZE(args) == 0)
8271 return do_strip(self, BOTHSTRIP); /* Common case */
8272 else
8273 return do_argstrip(self, BOTHSTRIP, args);
Walter Dörwaldde02bcb2002-04-22 17:42:37 +00008274}
8275
8276
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008277PyDoc_STRVAR(lstrip__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +00008278 "S.lstrip([chars]) -> str\n\
Walter Dörwaldde02bcb2002-04-22 17:42:37 +00008279\n\
8280Return a copy of the string S with leading whitespace removed.\n\
Benjamin Peterson142957c2008-07-04 19:55:29 +00008281If chars is given and not None, remove characters in chars instead.");
Walter Dörwaldde02bcb2002-04-22 17:42:37 +00008282
8283static PyObject *
8284unicode_lstrip(PyUnicodeObject *self, PyObject *args)
8285{
Benjamin Peterson14339b62009-01-31 16:36:08 +00008286 if (PyTuple_GET_SIZE(args) == 0)
8287 return do_strip(self, LEFTSTRIP); /* Common case */
8288 else
8289 return do_argstrip(self, LEFTSTRIP, args);
Walter Dörwaldde02bcb2002-04-22 17:42:37 +00008290}
8291
8292
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008293PyDoc_STRVAR(rstrip__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +00008294 "S.rstrip([chars]) -> str\n\
Walter Dörwaldde02bcb2002-04-22 17:42:37 +00008295\n\
8296Return a copy of the string S with trailing whitespace removed.\n\
Benjamin Peterson142957c2008-07-04 19:55:29 +00008297If chars is given and not None, remove characters in chars instead.");
Walter Dörwaldde02bcb2002-04-22 17:42:37 +00008298
8299static PyObject *
8300unicode_rstrip(PyUnicodeObject *self, PyObject *args)
8301{
Benjamin Peterson14339b62009-01-31 16:36:08 +00008302 if (PyTuple_GET_SIZE(args) == 0)
8303 return do_strip(self, RIGHTSTRIP); /* Common case */
8304 else
8305 return do_argstrip(self, RIGHTSTRIP, args);
Walter Dörwaldde02bcb2002-04-22 17:42:37 +00008306}
8307
8308
Guido van Rossumd57fd912000-03-10 22:53:23 +00008309static PyObject*
Martin v. Löwis18e16552006-02-15 17:27:45 +00008310unicode_repeat(PyUnicodeObject *str, Py_ssize_t len)
Guido van Rossumd57fd912000-03-10 22:53:23 +00008311{
8312 PyUnicodeObject *u;
8313 Py_UNICODE *p;
Martin v. Löwis18e16552006-02-15 17:27:45 +00008314 Py_ssize_t nchars;
Tim Peters8f422462000-09-09 06:13:41 +00008315 size_t nbytes;
Guido van Rossumd57fd912000-03-10 22:53:23 +00008316
Georg Brandl222de0f2009-04-12 12:01:50 +00008317 if (len < 1) {
8318 Py_INCREF(unicode_empty);
8319 return (PyObject *)unicode_empty;
8320 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00008321
Tim Peters7a29bd52001-09-12 03:03:31 +00008322 if (len == 1 && PyUnicode_CheckExact(str)) {
Guido van Rossumd57fd912000-03-10 22:53:23 +00008323 /* no repeat, return original string */
8324 Py_INCREF(str);
8325 return (PyObject*) str;
8326 }
Tim Peters8f422462000-09-09 06:13:41 +00008327
8328 /* ensure # of chars needed doesn't overflow int and # of bytes
8329 * needed doesn't overflow size_t
8330 */
8331 nchars = len * str->length;
Georg Brandl222de0f2009-04-12 12:01:50 +00008332 if (nchars / len != str->length) {
Tim Peters8f422462000-09-09 06:13:41 +00008333 PyErr_SetString(PyExc_OverflowError,
8334 "repeated string is too long");
8335 return NULL;
8336 }
8337 nbytes = (nchars + 1) * sizeof(Py_UNICODE);
8338 if (nbytes / sizeof(Py_UNICODE) != (size_t)(nchars + 1)) {
8339 PyErr_SetString(PyExc_OverflowError,
8340 "repeated string is too long");
8341 return NULL;
8342 }
8343 u = _PyUnicode_New(nchars);
Guido van Rossumd57fd912000-03-10 22:53:23 +00008344 if (!u)
8345 return NULL;
8346
8347 p = u->str;
8348
Georg Brandl222de0f2009-04-12 12:01:50 +00008349 if (str->length == 1) {
Thomas Wouters477c8d52006-05-27 19:21:47 +00008350 Py_UNICODE_FILL(p, str->str[0], len);
8351 } else {
Georg Brandl222de0f2009-04-12 12:01:50 +00008352 Py_ssize_t done = str->length; /* number of characters copied this far */
8353 Py_UNICODE_COPY(p, str->str, str->length);
Benjamin Peterson29060642009-01-31 22:14:21 +00008354 while (done < nchars) {
Christian Heimescc47b052008-03-25 14:56:36 +00008355 Py_ssize_t n = (done <= nchars-done) ? done : nchars-done;
Thomas Wouters477c8d52006-05-27 19:21:47 +00008356 Py_UNICODE_COPY(p+done, p, n);
8357 done += n;
Benjamin Peterson29060642009-01-31 22:14:21 +00008358 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00008359 }
8360
8361 return (PyObject*) u;
8362}
8363
8364PyObject *PyUnicode_Replace(PyObject *obj,
Benjamin Peterson29060642009-01-31 22:14:21 +00008365 PyObject *subobj,
8366 PyObject *replobj,
8367 Py_ssize_t maxcount)
Guido van Rossumd57fd912000-03-10 22:53:23 +00008368{
8369 PyObject *self;
8370 PyObject *str1;
8371 PyObject *str2;
8372 PyObject *result;
8373
8374 self = PyUnicode_FromObject(obj);
8375 if (self == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00008376 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00008377 str1 = PyUnicode_FromObject(subobj);
8378 if (str1 == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +00008379 Py_DECREF(self);
8380 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00008381 }
8382 str2 = PyUnicode_FromObject(replobj);
8383 if (str2 == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +00008384 Py_DECREF(self);
8385 Py_DECREF(str1);
8386 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00008387 }
Tim Petersced69f82003-09-16 20:30:58 +00008388 result = replace((PyUnicodeObject *)self,
Benjamin Peterson29060642009-01-31 22:14:21 +00008389 (PyUnicodeObject *)str1,
8390 (PyUnicodeObject *)str2,
8391 maxcount);
Guido van Rossumd57fd912000-03-10 22:53:23 +00008392 Py_DECREF(self);
8393 Py_DECREF(str1);
8394 Py_DECREF(str2);
8395 return result;
8396}
8397
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008398PyDoc_STRVAR(replace__doc__,
Ezio Melottic1897e72010-06-26 18:50:39 +00008399 "S.replace(old, new[, count]) -> str\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +00008400\n\
8401Return a copy of S with all occurrences of substring\n\
Georg Brandlf08a9dd2008-06-10 16:57:31 +00008402old replaced by new. If the optional argument count is\n\
8403given, only the first count occurrences are replaced.");
Guido van Rossumd57fd912000-03-10 22:53:23 +00008404
8405static PyObject*
8406unicode_replace(PyUnicodeObject *self, PyObject *args)
8407{
8408 PyUnicodeObject *str1;
8409 PyUnicodeObject *str2;
Martin v. Löwis18e16552006-02-15 17:27:45 +00008410 Py_ssize_t maxcount = -1;
Guido van Rossumd57fd912000-03-10 22:53:23 +00008411 PyObject *result;
8412
Martin v. Löwis18e16552006-02-15 17:27:45 +00008413 if (!PyArg_ParseTuple(args, "OO|n:replace", &str1, &str2, &maxcount))
Guido van Rossumd57fd912000-03-10 22:53:23 +00008414 return NULL;
8415 str1 = (PyUnicodeObject *)PyUnicode_FromObject((PyObject *)str1);
8416 if (str1 == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00008417 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00008418 str2 = (PyUnicodeObject *)PyUnicode_FromObject((PyObject *)str2);
Walter Dörwaldf6b56ae2003-02-09 23:42:56 +00008419 if (str2 == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +00008420 Py_DECREF(str1);
8421 return NULL;
Walter Dörwaldf6b56ae2003-02-09 23:42:56 +00008422 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00008423
8424 result = replace(self, str1, str2, maxcount);
8425
8426 Py_DECREF(str1);
8427 Py_DECREF(str2);
8428 return result;
8429}
8430
8431static
8432PyObject *unicode_repr(PyObject *unicode)
8433{
Walter Dörwald79e913e2007-05-12 11:08:06 +00008434 PyObject *repr;
Walter Dörwald1ab83302007-05-18 17:15:44 +00008435 Py_UNICODE *p;
Walter Dörwald79e913e2007-05-12 11:08:06 +00008436 Py_UNICODE *s = PyUnicode_AS_UNICODE(unicode);
8437 Py_ssize_t size = PyUnicode_GET_SIZE(unicode);
8438
8439 /* XXX(nnorwitz): rather than over-allocating, it would be
8440 better to choose a different scheme. Perhaps scan the
8441 first N-chars of the string and allocate based on that size.
8442 */
8443 /* Initial allocation is based on the longest-possible unichr
8444 escape.
8445
8446 In wide (UTF-32) builds '\U00xxxxxx' is 10 chars per source
8447 unichr, so in this case it's the longest unichr escape. In
8448 narrow (UTF-16) builds this is five chars per source unichr
8449 since there are two unichrs in the surrogate pair, so in narrow
8450 (UTF-16) builds it's not the longest unichr escape.
8451
8452 In wide or narrow builds '\uxxxx' is 6 chars per source unichr,
8453 so in the narrow (UTF-16) build case it's the longest unichr
8454 escape.
8455 */
8456
Walter Dörwald1ab83302007-05-18 17:15:44 +00008457 repr = PyUnicode_FromUnicode(NULL,
Benjamin Peterson29060642009-01-31 22:14:21 +00008458 2 /* quotes */
Walter Dörwald79e913e2007-05-12 11:08:06 +00008459#ifdef Py_UNICODE_WIDE
Benjamin Peterson29060642009-01-31 22:14:21 +00008460 + 10*size
Walter Dörwald79e913e2007-05-12 11:08:06 +00008461#else
Benjamin Peterson29060642009-01-31 22:14:21 +00008462 + 6*size
Walter Dörwald79e913e2007-05-12 11:08:06 +00008463#endif
Benjamin Peterson29060642009-01-31 22:14:21 +00008464 + 1);
Walter Dörwald79e913e2007-05-12 11:08:06 +00008465 if (repr == NULL)
8466 return NULL;
8467
Walter Dörwald1ab83302007-05-18 17:15:44 +00008468 p = PyUnicode_AS_UNICODE(repr);
Walter Dörwald79e913e2007-05-12 11:08:06 +00008469
8470 /* Add quote */
8471 *p++ = (findchar(s, size, '\'') &&
8472 !findchar(s, size, '"')) ? '"' : '\'';
8473 while (size-- > 0) {
8474 Py_UNICODE ch = *s++;
8475
8476 /* Escape quotes and backslashes */
Walter Dörwald1ab83302007-05-18 17:15:44 +00008477 if ((ch == PyUnicode_AS_UNICODE(repr)[0]) || (ch == '\\')) {
Walter Dörwald79e913e2007-05-12 11:08:06 +00008478 *p++ = '\\';
Walter Dörwald1ab83302007-05-18 17:15:44 +00008479 *p++ = ch;
Walter Dörwald79e913e2007-05-12 11:08:06 +00008480 continue;
8481 }
8482
Benjamin Peterson29060642009-01-31 22:14:21 +00008483 /* Map special whitespace to '\t', \n', '\r' */
Georg Brandl559e5d72008-06-11 18:37:52 +00008484 if (ch == '\t') {
Walter Dörwald79e913e2007-05-12 11:08:06 +00008485 *p++ = '\\';
8486 *p++ = 't';
8487 }
8488 else if (ch == '\n') {
8489 *p++ = '\\';
8490 *p++ = 'n';
8491 }
8492 else if (ch == '\r') {
8493 *p++ = '\\';
8494 *p++ = 'r';
8495 }
8496
8497 /* Map non-printable US ASCII to '\xhh' */
Georg Brandl559e5d72008-06-11 18:37:52 +00008498 else if (ch < ' ' || ch == 0x7F) {
Walter Dörwald79e913e2007-05-12 11:08:06 +00008499 *p++ = '\\';
8500 *p++ = 'x';
8501 *p++ = hexdigits[(ch >> 4) & 0x000F];
8502 *p++ = hexdigits[ch & 0x000F];
8503 }
8504
Georg Brandl559e5d72008-06-11 18:37:52 +00008505 /* Copy ASCII characters as-is */
8506 else if (ch < 0x7F) {
8507 *p++ = ch;
8508 }
8509
Benjamin Peterson29060642009-01-31 22:14:21 +00008510 /* Non-ASCII characters */
Georg Brandl559e5d72008-06-11 18:37:52 +00008511 else {
8512 Py_UCS4 ucs = ch;
8513
8514#ifndef Py_UNICODE_WIDE
8515 Py_UNICODE ch2 = 0;
8516 /* Get code point from surrogate pair */
8517 if (size > 0) {
8518 ch2 = *s;
8519 if (ch >= 0xD800 && ch < 0xDC00 && ch2 >= 0xDC00
Benjamin Peterson29060642009-01-31 22:14:21 +00008520 && ch2 <= 0xDFFF) {
Benjamin Peterson14339b62009-01-31 16:36:08 +00008521 ucs = (((ch & 0x03FF) << 10) | (ch2 & 0x03FF))
Benjamin Peterson29060642009-01-31 22:14:21 +00008522 + 0x00010000;
Benjamin Peterson14339b62009-01-31 16:36:08 +00008523 s++;
Georg Brandl559e5d72008-06-11 18:37:52 +00008524 size--;
8525 }
8526 }
8527#endif
Benjamin Peterson14339b62009-01-31 16:36:08 +00008528 /* Map Unicode whitespace and control characters
Georg Brandl559e5d72008-06-11 18:37:52 +00008529 (categories Z* and C* except ASCII space)
8530 */
8531 if (!Py_UNICODE_ISPRINTABLE(ucs)) {
8532 /* Map 8-bit characters to '\xhh' */
8533 if (ucs <= 0xff) {
8534 *p++ = '\\';
8535 *p++ = 'x';
8536 *p++ = hexdigits[(ch >> 4) & 0x000F];
8537 *p++ = hexdigits[ch & 0x000F];
8538 }
8539 /* Map 21-bit characters to '\U00xxxxxx' */
8540 else if (ucs >= 0x10000) {
8541 *p++ = '\\';
8542 *p++ = 'U';
8543 *p++ = hexdigits[(ucs >> 28) & 0x0000000F];
8544 *p++ = hexdigits[(ucs >> 24) & 0x0000000F];
8545 *p++ = hexdigits[(ucs >> 20) & 0x0000000F];
8546 *p++ = hexdigits[(ucs >> 16) & 0x0000000F];
8547 *p++ = hexdigits[(ucs >> 12) & 0x0000000F];
8548 *p++ = hexdigits[(ucs >> 8) & 0x0000000F];
8549 *p++ = hexdigits[(ucs >> 4) & 0x0000000F];
8550 *p++ = hexdigits[ucs & 0x0000000F];
8551 }
8552 /* Map 16-bit characters to '\uxxxx' */
8553 else {
8554 *p++ = '\\';
8555 *p++ = 'u';
8556 *p++ = hexdigits[(ucs >> 12) & 0x000F];
8557 *p++ = hexdigits[(ucs >> 8) & 0x000F];
8558 *p++ = hexdigits[(ucs >> 4) & 0x000F];
8559 *p++ = hexdigits[ucs & 0x000F];
8560 }
8561 }
8562 /* Copy characters as-is */
8563 else {
8564 *p++ = ch;
8565#ifndef Py_UNICODE_WIDE
8566 if (ucs >= 0x10000)
8567 *p++ = ch2;
8568#endif
8569 }
8570 }
Walter Dörwald79e913e2007-05-12 11:08:06 +00008571 }
8572 /* Add quote */
Walter Dörwald1ab83302007-05-18 17:15:44 +00008573 *p++ = PyUnicode_AS_UNICODE(repr)[0];
Walter Dörwald79e913e2007-05-12 11:08:06 +00008574
8575 *p = '\0';
Alexandre Vassalottiaa0e5312008-12-27 06:43:58 +00008576 PyUnicode_Resize(&repr, p - PyUnicode_AS_UNICODE(repr));
Walter Dörwald79e913e2007-05-12 11:08:06 +00008577 return repr;
Guido van Rossumd57fd912000-03-10 22:53:23 +00008578}
8579
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008580PyDoc_STRVAR(rfind__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +00008581 "S.rfind(sub[, start[, end]]) -> int\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +00008582\n\
8583Return the highest index in S where substring sub is found,\n\
Senthil Kumaran53516a82011-07-27 23:33:54 +08008584such that sub is contained within S[start:end]. Optional\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +00008585arguments start and end are interpreted as in slice notation.\n\
8586\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008587Return -1 on failure.");
Guido van Rossumd57fd912000-03-10 22:53:23 +00008588
8589static PyObject *
8590unicode_rfind(PyUnicodeObject *self, PyObject *args)
8591{
Jesus Ceaac451502011-04-20 17:09:23 +02008592 PyUnicodeObject *substring;
Christian Heimes9cd17752007-11-18 19:35:23 +00008593 Py_ssize_t start;
8594 Py_ssize_t end;
Thomas Wouters477c8d52006-05-27 19:21:47 +00008595 Py_ssize_t result;
Guido van Rossumd57fd912000-03-10 22:53:23 +00008596
Jesus Ceaac451502011-04-20 17:09:23 +02008597 if (!stringlib_parse_args_finds_unicode("rfind", args, &substring,
8598 &start, &end))
Benjamin Peterson14339b62009-01-31 16:36:08 +00008599 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00008600
Thomas Wouters477c8d52006-05-27 19:21:47 +00008601 result = stringlib_rfind_slice(
8602 PyUnicode_AS_UNICODE(self), PyUnicode_GET_SIZE(self),
8603 PyUnicode_AS_UNICODE(substring), PyUnicode_GET_SIZE(substring),
8604 start, end
8605 );
Guido van Rossumd57fd912000-03-10 22:53:23 +00008606
8607 Py_DECREF(substring);
Thomas Wouters477c8d52006-05-27 19:21:47 +00008608
Christian Heimes217cfd12007-12-02 14:31:20 +00008609 return PyLong_FromSsize_t(result);
Guido van Rossumd57fd912000-03-10 22:53:23 +00008610}
8611
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008612PyDoc_STRVAR(rindex__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +00008613 "S.rindex(sub[, start[, end]]) -> int\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +00008614\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008615Like S.rfind() but raise ValueError when the substring is not found.");
Guido van Rossumd57fd912000-03-10 22:53:23 +00008616
8617static PyObject *
8618unicode_rindex(PyUnicodeObject *self, PyObject *args)
8619{
Jesus Ceaac451502011-04-20 17:09:23 +02008620 PyUnicodeObject *substring;
Christian Heimes9cd17752007-11-18 19:35:23 +00008621 Py_ssize_t start;
8622 Py_ssize_t end;
Thomas Wouters477c8d52006-05-27 19:21:47 +00008623 Py_ssize_t result;
Guido van Rossumd57fd912000-03-10 22:53:23 +00008624
Jesus Ceaac451502011-04-20 17:09:23 +02008625 if (!stringlib_parse_args_finds_unicode("rindex", args, &substring,
8626 &start, &end))
Benjamin Peterson14339b62009-01-31 16:36:08 +00008627 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00008628
Thomas Wouters477c8d52006-05-27 19:21:47 +00008629 result = stringlib_rfind_slice(
8630 PyUnicode_AS_UNICODE(self), PyUnicode_GET_SIZE(self),
8631 PyUnicode_AS_UNICODE(substring), PyUnicode_GET_SIZE(substring),
8632 start, end
8633 );
Guido van Rossumd57fd912000-03-10 22:53:23 +00008634
8635 Py_DECREF(substring);
Thomas Wouters477c8d52006-05-27 19:21:47 +00008636
Guido van Rossumd57fd912000-03-10 22:53:23 +00008637 if (result < 0) {
8638 PyErr_SetString(PyExc_ValueError, "substring not found");
8639 return NULL;
8640 }
Christian Heimes217cfd12007-12-02 14:31:20 +00008641 return PyLong_FromSsize_t(result);
Guido van Rossumd57fd912000-03-10 22:53:23 +00008642}
8643
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008644PyDoc_STRVAR(rjust__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +00008645 "S.rjust(width[, fillchar]) -> str\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +00008646\n\
Benjamin Petersonf10a79a2008-10-11 00:49:57 +00008647Return S right-justified in a string of length width. Padding is\n\
Raymond Hettinger4f8f9762003-11-26 08:21:35 +00008648done using the specified fill character (default is a space).");
Guido van Rossumd57fd912000-03-10 22:53:23 +00008649
8650static PyObject *
8651unicode_rjust(PyUnicodeObject *self, PyObject *args)
8652{
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00008653 Py_ssize_t width;
Raymond Hettinger4f8f9762003-11-26 08:21:35 +00008654 Py_UNICODE fillchar = ' ';
8655
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00008656 if (!PyArg_ParseTuple(args, "n|O&:rjust", &width, convert_uc, &fillchar))
Guido van Rossumd57fd912000-03-10 22:53:23 +00008657 return NULL;
8658
Tim Peters7a29bd52001-09-12 03:03:31 +00008659 if (self->length >= width && PyUnicode_CheckExact(self)) {
Guido van Rossumd57fd912000-03-10 22:53:23 +00008660 Py_INCREF(self);
8661 return (PyObject*) self;
8662 }
8663
Raymond Hettinger4f8f9762003-11-26 08:21:35 +00008664 return (PyObject*) pad(self, width - self->length, 0, fillchar);
Guido van Rossumd57fd912000-03-10 22:53:23 +00008665}
8666
Guido van Rossumd57fd912000-03-10 22:53:23 +00008667PyObject *PyUnicode_Split(PyObject *s,
Benjamin Peterson29060642009-01-31 22:14:21 +00008668 PyObject *sep,
8669 Py_ssize_t maxsplit)
Guido van Rossumd57fd912000-03-10 22:53:23 +00008670{
8671 PyObject *result;
Tim Petersced69f82003-09-16 20:30:58 +00008672
Guido van Rossumd57fd912000-03-10 22:53:23 +00008673 s = PyUnicode_FromObject(s);
8674 if (s == NULL)
Benjamin Peterson14339b62009-01-31 16:36:08 +00008675 return NULL;
Benjamin Peterson29060642009-01-31 22:14:21 +00008676 if (sep != NULL) {
8677 sep = PyUnicode_FromObject(sep);
8678 if (sep == NULL) {
8679 Py_DECREF(s);
8680 return NULL;
8681 }
Guido van Rossumd57fd912000-03-10 22:53:23 +00008682 }
8683
8684 result = split((PyUnicodeObject *)s, (PyUnicodeObject *)sep, maxsplit);
8685
8686 Py_DECREF(s);
8687 Py_XDECREF(sep);
8688 return result;
8689}
8690
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008691PyDoc_STRVAR(split__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +00008692 "S.split([sep[, maxsplit]]) -> list of strings\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +00008693\n\
8694Return a list of the words in S, using sep as the\n\
8695delimiter string. If maxsplit is given, at most maxsplit\n\
Alexandre Vassalotti5f8ced22008-05-16 00:03:33 +00008696splits are done. If sep is not specified or is None, any\n\
Alexandre Vassalotti8ae3e052008-05-16 00:41:41 +00008697whitespace string is a separator and empty strings are\n\
8698removed from the result.");
Guido van Rossumd57fd912000-03-10 22:53:23 +00008699
8700static PyObject*
8701unicode_split(PyUnicodeObject *self, PyObject *args)
8702{
8703 PyObject *substring = Py_None;
Martin v. Löwis18e16552006-02-15 17:27:45 +00008704 Py_ssize_t maxcount = -1;
Guido van Rossumd57fd912000-03-10 22:53:23 +00008705
Martin v. Löwis18e16552006-02-15 17:27:45 +00008706 if (!PyArg_ParseTuple(args, "|On:split", &substring, &maxcount))
Guido van Rossumd57fd912000-03-10 22:53:23 +00008707 return NULL;
8708
8709 if (substring == Py_None)
Benjamin Peterson29060642009-01-31 22:14:21 +00008710 return split(self, NULL, maxcount);
Guido van Rossumd57fd912000-03-10 22:53:23 +00008711 else if (PyUnicode_Check(substring))
Benjamin Peterson29060642009-01-31 22:14:21 +00008712 return split(self, (PyUnicodeObject *)substring, maxcount);
Guido van Rossumd57fd912000-03-10 22:53:23 +00008713 else
Benjamin Peterson29060642009-01-31 22:14:21 +00008714 return PyUnicode_Split((PyObject *)self, substring, maxcount);
Guido van Rossumd57fd912000-03-10 22:53:23 +00008715}
8716
Thomas Wouters477c8d52006-05-27 19:21:47 +00008717PyObject *
8718PyUnicode_Partition(PyObject *str_in, PyObject *sep_in)
8719{
8720 PyObject* str_obj;
8721 PyObject* sep_obj;
8722 PyObject* out;
8723
8724 str_obj = PyUnicode_FromObject(str_in);
8725 if (!str_obj)
Benjamin Peterson29060642009-01-31 22:14:21 +00008726 return NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +00008727 sep_obj = PyUnicode_FromObject(sep_in);
8728 if (!sep_obj) {
8729 Py_DECREF(str_obj);
8730 return NULL;
8731 }
8732
8733 out = stringlib_partition(
8734 str_obj, PyUnicode_AS_UNICODE(str_obj), PyUnicode_GET_SIZE(str_obj),
8735 sep_obj, PyUnicode_AS_UNICODE(sep_obj), PyUnicode_GET_SIZE(sep_obj)
8736 );
8737
8738 Py_DECREF(sep_obj);
8739 Py_DECREF(str_obj);
8740
8741 return out;
8742}
8743
8744
8745PyObject *
8746PyUnicode_RPartition(PyObject *str_in, PyObject *sep_in)
8747{
8748 PyObject* str_obj;
8749 PyObject* sep_obj;
8750 PyObject* out;
8751
8752 str_obj = PyUnicode_FromObject(str_in);
8753 if (!str_obj)
Benjamin Peterson29060642009-01-31 22:14:21 +00008754 return NULL;
Thomas Wouters477c8d52006-05-27 19:21:47 +00008755 sep_obj = PyUnicode_FromObject(sep_in);
8756 if (!sep_obj) {
8757 Py_DECREF(str_obj);
8758 return NULL;
8759 }
8760
8761 out = stringlib_rpartition(
8762 str_obj, PyUnicode_AS_UNICODE(str_obj), PyUnicode_GET_SIZE(str_obj),
8763 sep_obj, PyUnicode_AS_UNICODE(sep_obj), PyUnicode_GET_SIZE(sep_obj)
8764 );
8765
8766 Py_DECREF(sep_obj);
8767 Py_DECREF(str_obj);
8768
8769 return out;
8770}
8771
8772PyDoc_STRVAR(partition__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +00008773 "S.partition(sep) -> (head, sep, tail)\n\
Thomas Wouters477c8d52006-05-27 19:21:47 +00008774\n\
Georg Brandl17cb8a82008-05-30 08:20:09 +00008775Search for the separator sep in S, and return the part before it,\n\
Thomas Wouters477c8d52006-05-27 19:21:47 +00008776the separator itself, and the part after it. If the separator is not\n\
Benjamin Petersonf10a79a2008-10-11 00:49:57 +00008777found, return S and two empty strings.");
Thomas Wouters477c8d52006-05-27 19:21:47 +00008778
8779static PyObject*
8780unicode_partition(PyUnicodeObject *self, PyObject *separator)
8781{
8782 return PyUnicode_Partition((PyObject *)self, separator);
8783}
8784
8785PyDoc_STRVAR(rpartition__doc__,
Ezio Melotti5b2b2422010-01-25 11:58:28 +00008786 "S.rpartition(sep) -> (head, sep, tail)\n\
Thomas Wouters477c8d52006-05-27 19:21:47 +00008787\n\
Georg Brandl17cb8a82008-05-30 08:20:09 +00008788Search for the separator sep in S, starting at the end of S, and return\n\
Thomas Wouters477c8d52006-05-27 19:21:47 +00008789the part before it, the separator itself, and the part after it. If the\n\
Benjamin Petersonf10a79a2008-10-11 00:49:57 +00008790separator is not found, return two empty strings and S.");
Thomas Wouters477c8d52006-05-27 19:21:47 +00008791
8792static PyObject*
8793unicode_rpartition(PyUnicodeObject *self, PyObject *separator)
8794{
8795 return PyUnicode_RPartition((PyObject *)self, separator);
8796}
8797
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +00008798PyObject *PyUnicode_RSplit(PyObject *s,
Benjamin Peterson29060642009-01-31 22:14:21 +00008799 PyObject *sep,
8800 Py_ssize_t maxsplit)
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +00008801{
8802 PyObject *result;
Benjamin Peterson14339b62009-01-31 16:36:08 +00008803
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +00008804 s = PyUnicode_FromObject(s);
8805 if (s == NULL)
Benjamin Peterson14339b62009-01-31 16:36:08 +00008806 return NULL;
Benjamin Peterson29060642009-01-31 22:14:21 +00008807 if (sep != NULL) {
8808 sep = PyUnicode_FromObject(sep);
8809 if (sep == NULL) {
8810 Py_DECREF(s);
8811 return NULL;
8812 }
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +00008813 }
8814
8815 result = rsplit((PyUnicodeObject *)s, (PyUnicodeObject *)sep, maxsplit);
8816
8817 Py_DECREF(s);
8818 Py_XDECREF(sep);
8819 return result;
8820}
8821
8822PyDoc_STRVAR(rsplit__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +00008823 "S.rsplit([sep[, maxsplit]]) -> list of strings\n\
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +00008824\n\
8825Return a list of the words in S, using sep as the\n\
8826delimiter string, starting at the end of the string and\n\
8827working to the front. If maxsplit is given, at most maxsplit\n\
8828splits are done. If sep is not specified, any whitespace string\n\
8829is a separator.");
8830
8831static PyObject*
8832unicode_rsplit(PyUnicodeObject *self, PyObject *args)
8833{
8834 PyObject *substring = Py_None;
Martin v. Löwis18e16552006-02-15 17:27:45 +00008835 Py_ssize_t maxcount = -1;
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +00008836
Martin v. Löwis18e16552006-02-15 17:27:45 +00008837 if (!PyArg_ParseTuple(args, "|On:rsplit", &substring, &maxcount))
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +00008838 return NULL;
8839
8840 if (substring == Py_None)
Benjamin Peterson29060642009-01-31 22:14:21 +00008841 return rsplit(self, NULL, maxcount);
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +00008842 else if (PyUnicode_Check(substring))
Benjamin Peterson29060642009-01-31 22:14:21 +00008843 return rsplit(self, (PyUnicodeObject *)substring, maxcount);
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +00008844 else
Benjamin Peterson29060642009-01-31 22:14:21 +00008845 return PyUnicode_RSplit((PyObject *)self, substring, maxcount);
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +00008846}
8847
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008848PyDoc_STRVAR(splitlines__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +00008849 "S.splitlines([keepends]) -> list of strings\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +00008850\n\
8851Return a list of the lines in S, breaking at line boundaries.\n\
Guido van Rossum86662912000-04-11 15:38:46 +00008852Line breaks are not included in the resulting list unless keepends\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008853is given and true.");
Guido van Rossumd57fd912000-03-10 22:53:23 +00008854
8855static PyObject*
8856unicode_splitlines(PyUnicodeObject *self, PyObject *args)
8857{
Guido van Rossum86662912000-04-11 15:38:46 +00008858 int keepends = 0;
Guido van Rossumd57fd912000-03-10 22:53:23 +00008859
Guido van Rossum86662912000-04-11 15:38:46 +00008860 if (!PyArg_ParseTuple(args, "|i:splitlines", &keepends))
Guido van Rossumd57fd912000-03-10 22:53:23 +00008861 return NULL;
8862
Guido van Rossum86662912000-04-11 15:38:46 +00008863 return PyUnicode_Splitlines((PyObject *)self, keepends);
Guido van Rossumd57fd912000-03-10 22:53:23 +00008864}
8865
8866static
Guido van Rossumf15a29f2007-05-04 00:41:39 +00008867PyObject *unicode_str(PyObject *self)
Guido van Rossumd57fd912000-03-10 22:53:23 +00008868{
Walter Dörwald346737f2007-05-31 10:44:43 +00008869 if (PyUnicode_CheckExact(self)) {
8870 Py_INCREF(self);
8871 return self;
8872 } else
8873 /* Subtype -- return genuine unicode string with the same value. */
8874 return PyUnicode_FromUnicode(PyUnicode_AS_UNICODE(self),
8875 PyUnicode_GET_SIZE(self));
Guido van Rossumd57fd912000-03-10 22:53:23 +00008876}
8877
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008878PyDoc_STRVAR(swapcase__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +00008879 "S.swapcase() -> str\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +00008880\n\
8881Return a copy of S with uppercase characters converted to lowercase\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008882and vice versa.");
Guido van Rossumd57fd912000-03-10 22:53:23 +00008883
8884static PyObject*
Martin v. Löwise3eb1f22001-08-16 13:15:00 +00008885unicode_swapcase(PyUnicodeObject *self)
Guido van Rossumd57fd912000-03-10 22:53:23 +00008886{
Guido van Rossumd57fd912000-03-10 22:53:23 +00008887 return fixup(self, fixswapcase);
8888}
8889
Georg Brandlceee0772007-11-27 23:48:05 +00008890PyDoc_STRVAR(maketrans__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +00008891 "str.maketrans(x[, y[, z]]) -> dict (static method)\n\
Georg Brandlceee0772007-11-27 23:48:05 +00008892\n\
8893Return a translation table usable for str.translate().\n\
8894If there is only one argument, it must be a dictionary mapping Unicode\n\
8895ordinals (integers) or characters to Unicode ordinals, strings or None.\n\
Benjamin Peterson142957c2008-07-04 19:55:29 +00008896Character keys will be then converted to ordinals.\n\
Georg Brandlceee0772007-11-27 23:48:05 +00008897If there are two arguments, they must be strings of equal length, and\n\
8898in the resulting dictionary, each character in x will be mapped to the\n\
8899character at the same position in y. If there is a third argument, it\n\
8900must be a string, whose characters will be mapped to None in the result.");
8901
8902static PyObject*
8903unicode_maketrans(PyUnicodeObject *null, PyObject *args)
8904{
8905 PyObject *x, *y = NULL, *z = NULL;
8906 PyObject *new = NULL, *key, *value;
8907 Py_ssize_t i = 0;
8908 int res;
Benjamin Peterson14339b62009-01-31 16:36:08 +00008909
Georg Brandlceee0772007-11-27 23:48:05 +00008910 if (!PyArg_ParseTuple(args, "O|UU:maketrans", &x, &y, &z))
8911 return NULL;
8912 new = PyDict_New();
8913 if (!new)
8914 return NULL;
8915 if (y != NULL) {
8916 /* x must be a string too, of equal length */
8917 Py_ssize_t ylen = PyUnicode_GET_SIZE(y);
8918 if (!PyUnicode_Check(x)) {
8919 PyErr_SetString(PyExc_TypeError, "first maketrans argument must "
8920 "be a string if there is a second argument");
8921 goto err;
8922 }
8923 if (PyUnicode_GET_SIZE(x) != ylen) {
8924 PyErr_SetString(PyExc_ValueError, "the first two maketrans "
8925 "arguments must have equal length");
8926 goto err;
8927 }
8928 /* create entries for translating chars in x to those in y */
8929 for (i = 0; i < PyUnicode_GET_SIZE(x); i++) {
Christian Heimes217cfd12007-12-02 14:31:20 +00008930 key = PyLong_FromLong(PyUnicode_AS_UNICODE(x)[i]);
Benjamin Peterson53aa1d72011-12-20 13:29:45 -06008931 if (!key)
Georg Brandlceee0772007-11-27 23:48:05 +00008932 goto err;
Benjamin Peterson53aa1d72011-12-20 13:29:45 -06008933 value = PyLong_FromLong(PyUnicode_AS_UNICODE(y)[i]);
8934 if (!value) {
8935 Py_DECREF(key);
8936 goto err;
8937 }
Georg Brandlceee0772007-11-27 23:48:05 +00008938 res = PyDict_SetItem(new, key, value);
8939 Py_DECREF(key);
8940 Py_DECREF(value);
8941 if (res < 0)
8942 goto err;
8943 }
8944 /* create entries for deleting chars in z */
8945 if (z != NULL) {
8946 for (i = 0; i < PyUnicode_GET_SIZE(z); i++) {
Christian Heimes217cfd12007-12-02 14:31:20 +00008947 key = PyLong_FromLong(PyUnicode_AS_UNICODE(z)[i]);
Georg Brandlceee0772007-11-27 23:48:05 +00008948 if (!key)
8949 goto err;
8950 res = PyDict_SetItem(new, key, Py_None);
8951 Py_DECREF(key);
8952 if (res < 0)
8953 goto err;
8954 }
8955 }
8956 } else {
8957 /* x must be a dict */
Raymond Hettinger3ad05762009-05-29 22:11:22 +00008958 if (!PyDict_CheckExact(x)) {
Georg Brandlceee0772007-11-27 23:48:05 +00008959 PyErr_SetString(PyExc_TypeError, "if you give only one argument "
8960 "to maketrans it must be a dict");
8961 goto err;
8962 }
8963 /* copy entries into the new dict, converting string keys to int keys */
8964 while (PyDict_Next(x, &i, &key, &value)) {
8965 if (PyUnicode_Check(key)) {
8966 /* convert string keys to integer keys */
8967 PyObject *newkey;
8968 if (PyUnicode_GET_SIZE(key) != 1) {
8969 PyErr_SetString(PyExc_ValueError, "string keys in translate "
8970 "table must be of length 1");
8971 goto err;
8972 }
Christian Heimes217cfd12007-12-02 14:31:20 +00008973 newkey = PyLong_FromLong(PyUnicode_AS_UNICODE(key)[0]);
Georg Brandlceee0772007-11-27 23:48:05 +00008974 if (!newkey)
8975 goto err;
8976 res = PyDict_SetItem(new, newkey, value);
8977 Py_DECREF(newkey);
8978 if (res < 0)
8979 goto err;
Christian Heimes217cfd12007-12-02 14:31:20 +00008980 } else if (PyLong_Check(key)) {
Georg Brandlceee0772007-11-27 23:48:05 +00008981 /* just keep integer keys */
8982 if (PyDict_SetItem(new, key, value) < 0)
8983 goto err;
8984 } else {
8985 PyErr_SetString(PyExc_TypeError, "keys in translate table must "
8986 "be strings or integers");
8987 goto err;
8988 }
8989 }
8990 }
8991 return new;
8992 err:
8993 Py_DECREF(new);
8994 return NULL;
8995}
8996
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00008997PyDoc_STRVAR(translate__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +00008998 "S.translate(table) -> str\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +00008999\n\
9000Return a copy of the string S, where all characters have been mapped\n\
9001through the given translation table, which must be a mapping of\n\
Benjamin Peterson142957c2008-07-04 19:55:29 +00009002Unicode ordinals to Unicode ordinals, strings, or None.\n\
Walter Dörwald5c1ee172002-09-04 20:31:32 +00009003Unmapped characters are left untouched. Characters mapped to None\n\
9004are deleted.");
Guido van Rossumd57fd912000-03-10 22:53:23 +00009005
9006static PyObject*
Martin v. Löwise3eb1f22001-08-16 13:15:00 +00009007unicode_translate(PyUnicodeObject *self, PyObject *table)
Guido van Rossumd57fd912000-03-10 22:53:23 +00009008{
Georg Brandlceee0772007-11-27 23:48:05 +00009009 return PyUnicode_TranslateCharmap(self->str, self->length, table, "ignore");
Guido van Rossumd57fd912000-03-10 22:53:23 +00009010}
9011
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00009012PyDoc_STRVAR(upper__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +00009013 "S.upper() -> str\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +00009014\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00009015Return a copy of S converted to uppercase.");
Guido van Rossumd57fd912000-03-10 22:53:23 +00009016
9017static PyObject*
Martin v. Löwise3eb1f22001-08-16 13:15:00 +00009018unicode_upper(PyUnicodeObject *self)
Guido van Rossumd57fd912000-03-10 22:53:23 +00009019{
Guido van Rossumd57fd912000-03-10 22:53:23 +00009020 return fixup(self, fixupper);
9021}
9022
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00009023PyDoc_STRVAR(zfill__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +00009024 "S.zfill(width) -> str\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +00009025\n\
Benjamin Peterson9aa42992008-09-10 21:57:34 +00009026Pad a numeric string S with zeros on the left, to fill a field\n\
9027of the specified width. The string S is never truncated.");
Guido van Rossumd57fd912000-03-10 22:53:23 +00009028
9029static PyObject *
9030unicode_zfill(PyUnicodeObject *self, PyObject *args)
9031{
Martin v. Löwis18e16552006-02-15 17:27:45 +00009032 Py_ssize_t fill;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009033 PyUnicodeObject *u;
9034
Martin v. Löwis18e16552006-02-15 17:27:45 +00009035 Py_ssize_t width;
9036 if (!PyArg_ParseTuple(args, "n:zfill", &width))
Guido van Rossumd57fd912000-03-10 22:53:23 +00009037 return NULL;
9038
9039 if (self->length >= width) {
Walter Dörwald0fe940c2002-04-15 18:42:15 +00009040 if (PyUnicode_CheckExact(self)) {
9041 Py_INCREF(self);
9042 return (PyObject*) self;
9043 }
9044 else
9045 return PyUnicode_FromUnicode(
9046 PyUnicode_AS_UNICODE(self),
9047 PyUnicode_GET_SIZE(self)
Benjamin Peterson29060642009-01-31 22:14:21 +00009048 );
Guido van Rossumd57fd912000-03-10 22:53:23 +00009049 }
9050
9051 fill = width - self->length;
9052
9053 u = pad(self, fill, 0, '0');
9054
Walter Dörwald068325e2002-04-15 13:36:47 +00009055 if (u == NULL)
9056 return NULL;
9057
Guido van Rossumd57fd912000-03-10 22:53:23 +00009058 if (u->str[fill] == '+' || u->str[fill] == '-') {
9059 /* move sign to beginning of string */
9060 u->str[0] = u->str[fill];
9061 u->str[fill] = '0';
9062 }
9063
9064 return (PyObject*) u;
9065}
Guido van Rossumd57fd912000-03-10 22:53:23 +00009066
9067#if 0
9068static PyObject*
Martin v. Löwise3eb1f22001-08-16 13:15:00 +00009069unicode_freelistsize(PyUnicodeObject *self)
Guido van Rossumd57fd912000-03-10 22:53:23 +00009070{
Christian Heimes2202f872008-02-06 14:31:34 +00009071 return PyLong_FromLong(numfree);
Guido van Rossumd57fd912000-03-10 22:53:23 +00009072}
Alexander Belopolsky942af5a2010-12-04 03:38:46 +00009073
9074static PyObject *
9075unicode__decimal2ascii(PyObject *self)
9076{
9077 return PyUnicode_TransformDecimalToASCII(PyUnicode_AS_UNICODE(self),
9078 PyUnicode_GET_SIZE(self));
9079}
Guido van Rossumd57fd912000-03-10 22:53:23 +00009080#endif
9081
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00009082PyDoc_STRVAR(startswith__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +00009083 "S.startswith(prefix[, start[, end]]) -> bool\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +00009084\n\
Guido van Rossuma7132182003-04-09 19:32:45 +00009085Return True if S starts with the specified prefix, False otherwise.\n\
9086With optional start, test S beginning at that position.\n\
Thomas Wouters0e3f5912006-08-11 14:57:12 +00009087With optional end, stop comparing S at that position.\n\
9088prefix can also be a tuple of strings to try.");
Guido van Rossumd57fd912000-03-10 22:53:23 +00009089
9090static PyObject *
9091unicode_startswith(PyUnicodeObject *self,
Benjamin Peterson29060642009-01-31 22:14:21 +00009092 PyObject *args)
Guido van Rossumd57fd912000-03-10 22:53:23 +00009093{
Thomas Wouters0e3f5912006-08-11 14:57:12 +00009094 PyObject *subobj;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009095 PyUnicodeObject *substring;
Martin v. Löwis18e16552006-02-15 17:27:45 +00009096 Py_ssize_t start = 0;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00009097 Py_ssize_t end = PY_SSIZE_T_MAX;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00009098 int result;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009099
Jesus Ceaac451502011-04-20 17:09:23 +02009100 if (!stringlib_parse_args_finds("startswith", args, &subobj, &start, &end))
Benjamin Peterson29060642009-01-31 22:14:21 +00009101 return NULL;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00009102 if (PyTuple_Check(subobj)) {
9103 Py_ssize_t i;
9104 for (i = 0; i < PyTuple_GET_SIZE(subobj); i++) {
9105 substring = (PyUnicodeObject *)PyUnicode_FromObject(
Benjamin Peterson29060642009-01-31 22:14:21 +00009106 PyTuple_GET_ITEM(subobj, i));
Thomas Wouters0e3f5912006-08-11 14:57:12 +00009107 if (substring == NULL)
9108 return NULL;
9109 result = tailmatch(self, substring, start, end, -1);
9110 Py_DECREF(substring);
9111 if (result) {
9112 Py_RETURN_TRUE;
9113 }
9114 }
9115 /* nothing matched */
9116 Py_RETURN_FALSE;
9117 }
9118 substring = (PyUnicodeObject *)PyUnicode_FromObject(subobj);
Ezio Melottiba42fd52011-04-26 06:09:45 +03009119 if (substring == NULL) {
9120 if (PyErr_ExceptionMatches(PyExc_TypeError))
9121 PyErr_Format(PyExc_TypeError, "startswith first arg must be str or "
9122 "a tuple of str, not %s", Py_TYPE(subobj)->tp_name);
Benjamin Peterson29060642009-01-31 22:14:21 +00009123 return NULL;
Ezio Melottiba42fd52011-04-26 06:09:45 +03009124 }
Thomas Wouters0e3f5912006-08-11 14:57:12 +00009125 result = tailmatch(self, substring, start, end, -1);
Guido van Rossumd57fd912000-03-10 22:53:23 +00009126 Py_DECREF(substring);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00009127 return PyBool_FromLong(result);
Guido van Rossumd57fd912000-03-10 22:53:23 +00009128}
9129
9130
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00009131PyDoc_STRVAR(endswith__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +00009132 "S.endswith(suffix[, start[, end]]) -> bool\n\
Guido van Rossumd57fd912000-03-10 22:53:23 +00009133\n\
Guido van Rossuma7132182003-04-09 19:32:45 +00009134Return True if S ends with the specified suffix, False otherwise.\n\
9135With optional start, test S beginning at that position.\n\
Thomas Wouters0e3f5912006-08-11 14:57:12 +00009136With optional end, stop comparing S at that position.\n\
9137suffix can also be a tuple of strings to try.");
Guido van Rossumd57fd912000-03-10 22:53:23 +00009138
9139static PyObject *
9140unicode_endswith(PyUnicodeObject *self,
Benjamin Peterson29060642009-01-31 22:14:21 +00009141 PyObject *args)
Guido van Rossumd57fd912000-03-10 22:53:23 +00009142{
Thomas Wouters0e3f5912006-08-11 14:57:12 +00009143 PyObject *subobj;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009144 PyUnicodeObject *substring;
Martin v. Löwis18e16552006-02-15 17:27:45 +00009145 Py_ssize_t start = 0;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00009146 Py_ssize_t end = PY_SSIZE_T_MAX;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00009147 int result;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009148
Jesus Ceaac451502011-04-20 17:09:23 +02009149 if (!stringlib_parse_args_finds("endswith", args, &subobj, &start, &end))
Benjamin Peterson29060642009-01-31 22:14:21 +00009150 return NULL;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00009151 if (PyTuple_Check(subobj)) {
9152 Py_ssize_t i;
9153 for (i = 0; i < PyTuple_GET_SIZE(subobj); i++) {
9154 substring = (PyUnicodeObject *)PyUnicode_FromObject(
Benjamin Peterson29060642009-01-31 22:14:21 +00009155 PyTuple_GET_ITEM(subobj, i));
Thomas Wouters0e3f5912006-08-11 14:57:12 +00009156 if (substring == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00009157 return NULL;
Thomas Wouters0e3f5912006-08-11 14:57:12 +00009158 result = tailmatch(self, substring, start, end, +1);
9159 Py_DECREF(substring);
9160 if (result) {
9161 Py_RETURN_TRUE;
9162 }
9163 }
9164 Py_RETURN_FALSE;
9165 }
9166 substring = (PyUnicodeObject *)PyUnicode_FromObject(subobj);
Ezio Melottiba42fd52011-04-26 06:09:45 +03009167 if (substring == NULL) {
9168 if (PyErr_ExceptionMatches(PyExc_TypeError))
9169 PyErr_Format(PyExc_TypeError, "endswith first arg must be str or "
9170 "a tuple of str, not %s", Py_TYPE(subobj)->tp_name);
Benjamin Peterson29060642009-01-31 22:14:21 +00009171 return NULL;
Ezio Melottiba42fd52011-04-26 06:09:45 +03009172 }
Thomas Wouters0e3f5912006-08-11 14:57:12 +00009173 result = tailmatch(self, substring, start, end, +1);
Guido van Rossumd57fd912000-03-10 22:53:23 +00009174 Py_DECREF(substring);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00009175 return PyBool_FromLong(result);
Guido van Rossumd57fd912000-03-10 22:53:23 +00009176}
9177
Eric Smith8c663262007-08-25 02:26:07 +00009178#include "stringlib/string_format.h"
9179
9180PyDoc_STRVAR(format__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +00009181 "S.format(*args, **kwargs) -> str\n\
Eric Smith8c663262007-08-25 02:26:07 +00009182\n\
Eric Smith51d2fd92010-11-06 19:27:37 +00009183Return a formatted version of S, using substitutions from args and kwargs.\n\
9184The substitutions are identified by braces ('{' and '}').");
Eric Smith8c663262007-08-25 02:26:07 +00009185
Eric Smith27bbca62010-11-04 17:06:58 +00009186PyDoc_STRVAR(format_map__doc__,
9187 "S.format_map(mapping) -> str\n\
9188\n\
Eric Smith51d2fd92010-11-06 19:27:37 +00009189Return a formatted version of S, using substitutions from mapping.\n\
9190The substitutions are identified by braces ('{' and '}').");
Eric Smith27bbca62010-11-04 17:06:58 +00009191
Eric Smith4a7d76d2008-05-30 18:10:19 +00009192static PyObject *
9193unicode__format__(PyObject* self, PyObject* args)
9194{
9195 PyObject *format_spec;
9196
9197 if (!PyArg_ParseTuple(args, "U:__format__", &format_spec))
9198 return NULL;
9199
9200 return _PyUnicode_FormatAdvanced(self,
9201 PyUnicode_AS_UNICODE(format_spec),
9202 PyUnicode_GET_SIZE(format_spec));
9203}
9204
Eric Smith8c663262007-08-25 02:26:07 +00009205PyDoc_STRVAR(p_format__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +00009206 "S.__format__(format_spec) -> str\n\
Eric Smith8c663262007-08-25 02:26:07 +00009207\n\
Eric Smith51d2fd92010-11-06 19:27:37 +00009208Return a formatted version of S as described by format_spec.");
Eric Smith8c663262007-08-25 02:26:07 +00009209
9210static PyObject *
Georg Brandlc28e1fa2008-06-10 19:20:26 +00009211unicode__sizeof__(PyUnicodeObject *v)
9212{
Robert Schuppeniesfbe94c52008-07-14 10:13:31 +00009213 return PyLong_FromSsize_t(sizeof(PyUnicodeObject) +
9214 sizeof(Py_UNICODE) * (v->length + 1));
Georg Brandlc28e1fa2008-06-10 19:20:26 +00009215}
9216
9217PyDoc_STRVAR(sizeof__doc__,
Benjamin Peterson29060642009-01-31 22:14:21 +00009218 "S.__sizeof__() -> size of S in memory, in bytes");
Georg Brandlc28e1fa2008-06-10 19:20:26 +00009219
9220static PyObject *
Guido van Rossum5d9113d2003-01-29 17:58:45 +00009221unicode_getnewargs(PyUnicodeObject *v)
9222{
Benjamin Peterson14339b62009-01-31 16:36:08 +00009223 return Py_BuildValue("(u#)", v->str, v->length);
Guido van Rossum5d9113d2003-01-29 17:58:45 +00009224}
9225
Guido van Rossumd57fd912000-03-10 22:53:23 +00009226static PyMethodDef unicode_methods[] = {
Benjamin Peterson28a4dce2010-12-12 01:33:04 +00009227 {"encode", (PyCFunction) unicode_encode, METH_VARARGS | METH_KEYWORDS, encode__doc__},
Martin v. Löwise3eb1f22001-08-16 13:15:00 +00009228 {"replace", (PyCFunction) unicode_replace, METH_VARARGS, replace__doc__},
9229 {"split", (PyCFunction) unicode_split, METH_VARARGS, split__doc__},
Hye-Shik Chang3ae811b2003-12-15 18:49:53 +00009230 {"rsplit", (PyCFunction) unicode_rsplit, METH_VARARGS, rsplit__doc__},
Martin v. Löwise3eb1f22001-08-16 13:15:00 +00009231 {"join", (PyCFunction) unicode_join, METH_O, join__doc__},
9232 {"capitalize", (PyCFunction) unicode_capitalize, METH_NOARGS, capitalize__doc__},
9233 {"title", (PyCFunction) unicode_title, METH_NOARGS, title__doc__},
9234 {"center", (PyCFunction) unicode_center, METH_VARARGS, center__doc__},
9235 {"count", (PyCFunction) unicode_count, METH_VARARGS, count__doc__},
9236 {"expandtabs", (PyCFunction) unicode_expandtabs, METH_VARARGS, expandtabs__doc__},
9237 {"find", (PyCFunction) unicode_find, METH_VARARGS, find__doc__},
Thomas Wouters477c8d52006-05-27 19:21:47 +00009238 {"partition", (PyCFunction) unicode_partition, METH_O, partition__doc__},
Martin v. Löwise3eb1f22001-08-16 13:15:00 +00009239 {"index", (PyCFunction) unicode_index, METH_VARARGS, index__doc__},
9240 {"ljust", (PyCFunction) unicode_ljust, METH_VARARGS, ljust__doc__},
9241 {"lower", (PyCFunction) unicode_lower, METH_NOARGS, lower__doc__},
Walter Dörwaldde02bcb2002-04-22 17:42:37 +00009242 {"lstrip", (PyCFunction) unicode_lstrip, METH_VARARGS, lstrip__doc__},
Martin v. Löwise3eb1f22001-08-16 13:15:00 +00009243 {"rfind", (PyCFunction) unicode_rfind, METH_VARARGS, rfind__doc__},
9244 {"rindex", (PyCFunction) unicode_rindex, METH_VARARGS, rindex__doc__},
9245 {"rjust", (PyCFunction) unicode_rjust, METH_VARARGS, rjust__doc__},
Walter Dörwaldde02bcb2002-04-22 17:42:37 +00009246 {"rstrip", (PyCFunction) unicode_rstrip, METH_VARARGS, rstrip__doc__},
Thomas Wouters477c8d52006-05-27 19:21:47 +00009247 {"rpartition", (PyCFunction) unicode_rpartition, METH_O, rpartition__doc__},
Martin v. Löwise3eb1f22001-08-16 13:15:00 +00009248 {"splitlines", (PyCFunction) unicode_splitlines, METH_VARARGS, splitlines__doc__},
Walter Dörwaldde02bcb2002-04-22 17:42:37 +00009249 {"strip", (PyCFunction) unicode_strip, METH_VARARGS, strip__doc__},
Martin v. Löwise3eb1f22001-08-16 13:15:00 +00009250 {"swapcase", (PyCFunction) unicode_swapcase, METH_NOARGS, swapcase__doc__},
9251 {"translate", (PyCFunction) unicode_translate, METH_O, translate__doc__},
9252 {"upper", (PyCFunction) unicode_upper, METH_NOARGS, upper__doc__},
9253 {"startswith", (PyCFunction) unicode_startswith, METH_VARARGS, startswith__doc__},
9254 {"endswith", (PyCFunction) unicode_endswith, METH_VARARGS, endswith__doc__},
9255 {"islower", (PyCFunction) unicode_islower, METH_NOARGS, islower__doc__},
9256 {"isupper", (PyCFunction) unicode_isupper, METH_NOARGS, isupper__doc__},
9257 {"istitle", (PyCFunction) unicode_istitle, METH_NOARGS, istitle__doc__},
9258 {"isspace", (PyCFunction) unicode_isspace, METH_NOARGS, isspace__doc__},
9259 {"isdecimal", (PyCFunction) unicode_isdecimal, METH_NOARGS, isdecimal__doc__},
9260 {"isdigit", (PyCFunction) unicode_isdigit, METH_NOARGS, isdigit__doc__},
9261 {"isnumeric", (PyCFunction) unicode_isnumeric, METH_NOARGS, isnumeric__doc__},
9262 {"isalpha", (PyCFunction) unicode_isalpha, METH_NOARGS, isalpha__doc__},
9263 {"isalnum", (PyCFunction) unicode_isalnum, METH_NOARGS, isalnum__doc__},
Martin v. Löwis47383402007-08-15 07:32:56 +00009264 {"isidentifier", (PyCFunction) unicode_isidentifier, METH_NOARGS, isidentifier__doc__},
Georg Brandl559e5d72008-06-11 18:37:52 +00009265 {"isprintable", (PyCFunction) unicode_isprintable, METH_NOARGS, isprintable__doc__},
Martin v. Löwise3eb1f22001-08-16 13:15:00 +00009266 {"zfill", (PyCFunction) unicode_zfill, METH_VARARGS, zfill__doc__},
Eric Smith9cd1e092007-08-31 18:39:38 +00009267 {"format", (PyCFunction) do_string_format, METH_VARARGS | METH_KEYWORDS, format__doc__},
Eric Smith27bbca62010-11-04 17:06:58 +00009268 {"format_map", (PyCFunction) do_string_format_map, METH_O, format_map__doc__},
Eric Smith4a7d76d2008-05-30 18:10:19 +00009269 {"__format__", (PyCFunction) unicode__format__, METH_VARARGS, p_format__doc__},
Georg Brandlceee0772007-11-27 23:48:05 +00009270 {"maketrans", (PyCFunction) unicode_maketrans,
9271 METH_VARARGS | METH_STATIC, maketrans__doc__},
Georg Brandlc28e1fa2008-06-10 19:20:26 +00009272 {"__sizeof__", (PyCFunction) unicode__sizeof__, METH_NOARGS, sizeof__doc__},
Walter Dörwald068325e2002-04-15 13:36:47 +00009273#if 0
Martin v. Löwise3eb1f22001-08-16 13:15:00 +00009274 {"capwords", (PyCFunction) unicode_capwords, METH_NOARGS, capwords__doc__},
Guido van Rossumd57fd912000-03-10 22:53:23 +00009275#endif
9276
9277#if 0
Alexander Belopolsky942af5a2010-12-04 03:38:46 +00009278 /* These methods are just used for debugging the implementation. */
Martin v. Löwise3eb1f22001-08-16 13:15:00 +00009279 {"freelistsize", (PyCFunction) unicode_freelistsize, METH_NOARGS},
Alexander Belopolsky942af5a2010-12-04 03:38:46 +00009280 {"_decimal2ascii", (PyCFunction) unicode__decimal2ascii, METH_NOARGS},
Guido van Rossumd57fd912000-03-10 22:53:23 +00009281#endif
9282
Benjamin Peterson14339b62009-01-31 16:36:08 +00009283 {"__getnewargs__", (PyCFunction)unicode_getnewargs, METH_NOARGS},
Guido van Rossumd57fd912000-03-10 22:53:23 +00009284 {NULL, NULL}
9285};
9286
Neil Schemenauerce30bc92002-11-18 16:10:18 +00009287static PyObject *
9288unicode_mod(PyObject *v, PyObject *w)
9289{
Benjamin Peterson29060642009-01-31 22:14:21 +00009290 if (!PyUnicode_Check(v)) {
9291 Py_INCREF(Py_NotImplemented);
9292 return Py_NotImplemented;
9293 }
9294 return PyUnicode_Format(v, w);
Neil Schemenauerce30bc92002-11-18 16:10:18 +00009295}
9296
9297static PyNumberMethods unicode_as_number = {
Benjamin Peterson14339b62009-01-31 16:36:08 +00009298 0, /*nb_add*/
9299 0, /*nb_subtract*/
9300 0, /*nb_multiply*/
9301 unicode_mod, /*nb_remainder*/
Neil Schemenauerce30bc92002-11-18 16:10:18 +00009302};
9303
Guido van Rossumd57fd912000-03-10 22:53:23 +00009304static PySequenceMethods unicode_as_sequence = {
Benjamin Peterson14339b62009-01-31 16:36:08 +00009305 (lenfunc) unicode_length, /* sq_length */
9306 PyUnicode_Concat, /* sq_concat */
9307 (ssizeargfunc) unicode_repeat, /* sq_repeat */
9308 (ssizeargfunc) unicode_getitem, /* sq_item */
9309 0, /* sq_slice */
9310 0, /* sq_ass_item */
9311 0, /* sq_ass_slice */
9312 PyUnicode_Contains, /* sq_contains */
Guido van Rossumd57fd912000-03-10 22:53:23 +00009313};
9314
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00009315static PyObject*
9316unicode_subscript(PyUnicodeObject* self, PyObject* item)
9317{
Thomas Wouters00ee7ba2006-08-21 19:07:27 +00009318 if (PyIndex_Check(item)) {
9319 Py_ssize_t i = PyNumber_AsSsize_t(item, PyExc_IndexError);
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00009320 if (i == -1 && PyErr_Occurred())
9321 return NULL;
9322 if (i < 0)
Martin v. Löwisdea59e52006-01-05 10:00:36 +00009323 i += PyUnicode_GET_SIZE(self);
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00009324 return unicode_getitem(self, i);
9325 } else if (PySlice_Check(item)) {
Martin v. Löwis18e16552006-02-15 17:27:45 +00009326 Py_ssize_t start, stop, step, slicelength, cur, i;
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00009327 Py_UNICODE* source_buf;
9328 Py_UNICODE* result_buf;
9329 PyObject* result;
9330
Martin v. Löwis4d0d4712010-12-03 20:14:31 +00009331 if (PySlice_GetIndicesEx(item, PyUnicode_GET_SIZE(self),
Benjamin Peterson29060642009-01-31 22:14:21 +00009332 &start, &stop, &step, &slicelength) < 0) {
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00009333 return NULL;
9334 }
9335
9336 if (slicelength <= 0) {
9337 return PyUnicode_FromUnicode(NULL, 0);
Thomas Woutersed03b412007-08-28 21:37:11 +00009338 } else if (start == 0 && step == 1 && slicelength == self->length &&
9339 PyUnicode_CheckExact(self)) {
9340 Py_INCREF(self);
9341 return (PyObject *)self;
9342 } else if (step == 1) {
9343 return PyUnicode_FromUnicode(self->str + start, slicelength);
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00009344 } else {
9345 source_buf = PyUnicode_AS_UNICODE((PyObject*)self);
Christian Heimesb186d002008-03-18 15:15:01 +00009346 result_buf = (Py_UNICODE *)PyObject_MALLOC(slicelength*
9347 sizeof(Py_UNICODE));
Benjamin Peterson14339b62009-01-31 16:36:08 +00009348
Benjamin Peterson29060642009-01-31 22:14:21 +00009349 if (result_buf == NULL)
9350 return PyErr_NoMemory();
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00009351
9352 for (cur = start, i = 0; i < slicelength; cur += step, i++) {
9353 result_buf[i] = source_buf[cur];
9354 }
Tim Petersced69f82003-09-16 20:30:58 +00009355
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00009356 result = PyUnicode_FromUnicode(result_buf, slicelength);
Christian Heimesb186d002008-03-18 15:15:01 +00009357 PyObject_FREE(result_buf);
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00009358 return result;
9359 }
9360 } else {
9361 PyErr_SetString(PyExc_TypeError, "string indices must be integers");
9362 return NULL;
9363 }
9364}
9365
9366static PyMappingMethods unicode_as_mapping = {
Benjamin Peterson14339b62009-01-31 16:36:08 +00009367 (lenfunc)unicode_length, /* mp_length */
9368 (binaryfunc)unicode_subscript, /* mp_subscript */
9369 (objobjargproc)0, /* mp_ass_subscript */
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +00009370};
9371
Guido van Rossumd57fd912000-03-10 22:53:23 +00009372
Guido van Rossumd57fd912000-03-10 22:53:23 +00009373/* Helpers for PyUnicode_Format() */
9374
9375static PyObject *
Martin v. Löwis18e16552006-02-15 17:27:45 +00009376getnextarg(PyObject *args, Py_ssize_t arglen, Py_ssize_t *p_argidx)
Guido van Rossumd57fd912000-03-10 22:53:23 +00009377{
Martin v. Löwis18e16552006-02-15 17:27:45 +00009378 Py_ssize_t argidx = *p_argidx;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009379 if (argidx < arglen) {
Benjamin Peterson29060642009-01-31 22:14:21 +00009380 (*p_argidx)++;
9381 if (arglen < 0)
9382 return args;
9383 else
9384 return PyTuple_GetItem(args, argidx);
Guido van Rossumd57fd912000-03-10 22:53:23 +00009385 }
9386 PyErr_SetString(PyExc_TypeError,
Benjamin Peterson29060642009-01-31 22:14:21 +00009387 "not enough arguments for format string");
Guido van Rossumd57fd912000-03-10 22:53:23 +00009388 return NULL;
9389}
9390
Mark Dickinsonf489caf2009-05-01 11:42:00 +00009391/* Returns a new reference to a PyUnicode object, or NULL on failure. */
Guido van Rossumd57fd912000-03-10 22:53:23 +00009392
Mark Dickinsonf489caf2009-05-01 11:42:00 +00009393static PyObject *
9394formatfloat(PyObject *v, int flags, int prec, int type)
Guido van Rossumd57fd912000-03-10 22:53:23 +00009395{
Mark Dickinsonf489caf2009-05-01 11:42:00 +00009396 char *p;
9397 PyObject *result;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009398 double x;
Tim Petersced69f82003-09-16 20:30:58 +00009399
Guido van Rossumd57fd912000-03-10 22:53:23 +00009400 x = PyFloat_AsDouble(v);
9401 if (x == -1.0 && PyErr_Occurred())
Mark Dickinsonf489caf2009-05-01 11:42:00 +00009402 return NULL;
9403
Guido van Rossumd57fd912000-03-10 22:53:23 +00009404 if (prec < 0)
Benjamin Peterson29060642009-01-31 22:14:21 +00009405 prec = 6;
Eric Smith0923d1d2009-04-16 20:16:10 +00009406
Eric Smith0923d1d2009-04-16 20:16:10 +00009407 p = PyOS_double_to_string(x, type, prec,
9408 (flags & F_ALT) ? Py_DTSF_ALT : 0, NULL);
Mark Dickinsonf489caf2009-05-01 11:42:00 +00009409 if (p == NULL)
9410 return NULL;
9411 result = PyUnicode_FromStringAndSize(p, strlen(p));
Eric Smith0923d1d2009-04-16 20:16:10 +00009412 PyMem_Free(p);
9413 return result;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009414}
9415
Tim Peters38fd5b62000-09-21 05:43:11 +00009416static PyObject*
9417formatlong(PyObject *val, int flags, int prec, int type)
9418{
Benjamin Peterson14339b62009-01-31 16:36:08 +00009419 char *buf;
9420 int len;
9421 PyObject *str; /* temporary string object. */
9422 PyObject *result;
Tim Peters38fd5b62000-09-21 05:43:11 +00009423
Benjamin Peterson14339b62009-01-31 16:36:08 +00009424 str = _PyBytes_FormatLong(val, flags, prec, type, &buf, &len);
9425 if (!str)
9426 return NULL;
9427 result = PyUnicode_FromStringAndSize(buf, len);
9428 Py_DECREF(str);
9429 return result;
Tim Peters38fd5b62000-09-21 05:43:11 +00009430}
9431
Guido van Rossumd57fd912000-03-10 22:53:23 +00009432static int
9433formatchar(Py_UNICODE *buf,
Marc-André Lemburgf28dd832000-06-30 10:29:57 +00009434 size_t buflen,
9435 PyObject *v)
Guido van Rossumd57fd912000-03-10 22:53:23 +00009436{
Amaury Forgeot d'Arca4db6862008-07-04 21:26:43 +00009437 /* presume that the buffer is at least 3 characters long */
Marc-André Lemburgd4ab4a52000-06-08 17:54:00 +00009438 if (PyUnicode_Check(v)) {
Benjamin Peterson29060642009-01-31 22:14:21 +00009439 if (PyUnicode_GET_SIZE(v) == 1) {
9440 buf[0] = PyUnicode_AS_UNICODE(v)[0];
9441 buf[1] = '\0';
9442 return 1;
9443 }
9444#ifndef Py_UNICODE_WIDE
9445 if (PyUnicode_GET_SIZE(v) == 2) {
9446 /* Decode a valid surrogate pair */
9447 int c0 = PyUnicode_AS_UNICODE(v)[0];
9448 int c1 = PyUnicode_AS_UNICODE(v)[1];
9449 if (0xD800 <= c0 && c0 <= 0xDBFF &&
9450 0xDC00 <= c1 && c1 <= 0xDFFF) {
9451 buf[0] = c0;
9452 buf[1] = c1;
9453 buf[2] = '\0';
9454 return 2;
9455 }
9456 }
9457#endif
9458 goto onError;
9459 }
9460 else {
9461 /* Integer input truncated to a character */
9462 long x;
9463 x = PyLong_AsLong(v);
9464 if (x == -1 && PyErr_Occurred())
9465 goto onError;
9466
9467 if (x < 0 || x > 0x10ffff) {
9468 PyErr_SetString(PyExc_OverflowError,
9469 "%c arg not in range(0x110000)");
9470 return -1;
9471 }
9472
9473#ifndef Py_UNICODE_WIDE
9474 if (x > 0xffff) {
9475 x -= 0x10000;
9476 buf[0] = (Py_UNICODE)(0xD800 | (x >> 10));
9477 buf[1] = (Py_UNICODE)(0xDC00 | (x & 0x3FF));
9478 return 2;
9479 }
9480#endif
9481 buf[0] = (Py_UNICODE) x;
Benjamin Peterson14339b62009-01-31 16:36:08 +00009482 buf[1] = '\0';
9483 return 1;
9484 }
Amaury Forgeot d'Arca4db6862008-07-04 21:26:43 +00009485
Benjamin Peterson29060642009-01-31 22:14:21 +00009486 onError:
Marc-André Lemburgd4ab4a52000-06-08 17:54:00 +00009487 PyErr_SetString(PyExc_TypeError,
Benjamin Peterson29060642009-01-31 22:14:21 +00009488 "%c requires int or char");
Marc-André Lemburgd4ab4a52000-06-08 17:54:00 +00009489 return -1;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009490}
9491
Marc-André Lemburgf28dd832000-06-30 10:29:57 +00009492/* fmt%(v1,v2,...) is roughly equivalent to sprintf(fmt, v1, v2, ...)
Mark Dickinsonf489caf2009-05-01 11:42:00 +00009493 FORMATBUFLEN is the length of the buffer in which chars are formatted.
Marc-André Lemburgf28dd832000-06-30 10:29:57 +00009494*/
Mark Dickinsonf489caf2009-05-01 11:42:00 +00009495#define FORMATBUFLEN (size_t)10
Marc-André Lemburgf28dd832000-06-30 10:29:57 +00009496
Guido van Rossumd57fd912000-03-10 22:53:23 +00009497PyObject *PyUnicode_Format(PyObject *format,
Benjamin Peterson29060642009-01-31 22:14:21 +00009498 PyObject *args)
Guido van Rossumd57fd912000-03-10 22:53:23 +00009499{
9500 Py_UNICODE *fmt, *res;
Martin v. Löwis18e16552006-02-15 17:27:45 +00009501 Py_ssize_t fmtcnt, rescnt, reslen, arglen, argidx;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009502 int args_owned = 0;
9503 PyUnicodeObject *result = NULL;
9504 PyObject *dict = NULL;
9505 PyObject *uformat;
Tim Petersced69f82003-09-16 20:30:58 +00009506
Guido van Rossumd57fd912000-03-10 22:53:23 +00009507 if (format == NULL || args == NULL) {
Benjamin Peterson29060642009-01-31 22:14:21 +00009508 PyErr_BadInternalCall();
9509 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009510 }
9511 uformat = PyUnicode_FromObject(format);
Fred Drakee4315f52000-05-09 19:53:39 +00009512 if (uformat == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00009513 return NULL;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009514 fmt = PyUnicode_AS_UNICODE(uformat);
9515 fmtcnt = PyUnicode_GET_SIZE(uformat);
9516
9517 reslen = rescnt = fmtcnt + 100;
9518 result = _PyUnicode_New(reslen);
9519 if (result == NULL)
Benjamin Peterson29060642009-01-31 22:14:21 +00009520 goto onError;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009521 res = PyUnicode_AS_UNICODE(result);
9522
9523 if (PyTuple_Check(args)) {
Benjamin Peterson29060642009-01-31 22:14:21 +00009524 arglen = PyTuple_Size(args);
9525 argidx = 0;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009526 }
9527 else {
Benjamin Peterson29060642009-01-31 22:14:21 +00009528 arglen = -1;
9529 argidx = -2;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009530 }
Benjamin Peterson28a6cfa2012-08-28 17:55:35 -04009531 if (PyMapping_Check(args) && !PyTuple_Check(args) && !PyUnicode_Check(args))
Benjamin Peterson29060642009-01-31 22:14:21 +00009532 dict = args;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009533
9534 while (--fmtcnt >= 0) {
Benjamin Peterson29060642009-01-31 22:14:21 +00009535 if (*fmt != '%') {
9536 if (--rescnt < 0) {
9537 rescnt = fmtcnt + 100;
9538 reslen += rescnt;
9539 if (_PyUnicode_Resize(&result, reslen) < 0)
9540 goto onError;
9541 res = PyUnicode_AS_UNICODE(result) + reslen - rescnt;
9542 --rescnt;
Benjamin Peterson14339b62009-01-31 16:36:08 +00009543 }
Benjamin Peterson29060642009-01-31 22:14:21 +00009544 *res++ = *fmt++;
Benjamin Peterson14339b62009-01-31 16:36:08 +00009545 }
9546 else {
Benjamin Peterson29060642009-01-31 22:14:21 +00009547 /* Got a format specifier */
9548 int flags = 0;
9549 Py_ssize_t width = -1;
9550 int prec = -1;
9551 Py_UNICODE c = '\0';
9552 Py_UNICODE fill;
9553 int isnumok;
9554 PyObject *v = NULL;
9555 PyObject *temp = NULL;
9556 Py_UNICODE *pbuf;
9557 Py_UNICODE sign;
9558 Py_ssize_t len;
Mark Dickinsonf489caf2009-05-01 11:42:00 +00009559 Py_UNICODE formatbuf[FORMATBUFLEN]; /* For formatchar() */
Guido van Rossumd57fd912000-03-10 22:53:23 +00009560
Benjamin Peterson29060642009-01-31 22:14:21 +00009561 fmt++;
9562 if (*fmt == '(') {
9563 Py_UNICODE *keystart;
9564 Py_ssize_t keylen;
9565 PyObject *key;
9566 int pcount = 1;
Christian Heimesa612dc02008-02-24 13:08:18 +00009567
Benjamin Peterson29060642009-01-31 22:14:21 +00009568 if (dict == NULL) {
9569 PyErr_SetString(PyExc_TypeError,
9570 "format requires a mapping");
9571 goto onError;
9572 }
9573 ++fmt;
9574 --fmtcnt;
9575 keystart = fmt;
9576 /* Skip over balanced parentheses */
9577 while (pcount > 0 && --fmtcnt >= 0) {
9578 if (*fmt == ')')
9579 --pcount;
9580 else if (*fmt == '(')
9581 ++pcount;
9582 fmt++;
9583 }
9584 keylen = fmt - keystart - 1;
9585 if (fmtcnt < 0 || pcount > 0) {
9586 PyErr_SetString(PyExc_ValueError,
9587 "incomplete format key");
9588 goto onError;
9589 }
9590#if 0
9591 /* keys are converted to strings using UTF-8 and
9592 then looked up since Python uses strings to hold
9593 variables names etc. in its namespaces and we
9594 wouldn't want to break common idioms. */
9595 key = PyUnicode_EncodeUTF8(keystart,
9596 keylen,
9597 NULL);
9598#else
9599 key = PyUnicode_FromUnicode(keystart, keylen);
9600#endif
9601 if (key == NULL)
9602 goto onError;
9603 if (args_owned) {
9604 Py_DECREF(args);
9605 args_owned = 0;
9606 }
9607 args = PyObject_GetItem(dict, key);
9608 Py_DECREF(key);
9609 if (args == NULL) {
9610 goto onError;
9611 }
9612 args_owned = 1;
9613 arglen = -1;
9614 argidx = -2;
Benjamin Peterson14339b62009-01-31 16:36:08 +00009615 }
Benjamin Peterson29060642009-01-31 22:14:21 +00009616 while (--fmtcnt >= 0) {
9617 switch (c = *fmt++) {
9618 case '-': flags |= F_LJUST; continue;
9619 case '+': flags |= F_SIGN; continue;
9620 case ' ': flags |= F_BLANK; continue;
9621 case '#': flags |= F_ALT; continue;
9622 case '0': flags |= F_ZERO; continue;
9623 }
9624 break;
Benjamin Peterson14339b62009-01-31 16:36:08 +00009625 }
Benjamin Peterson29060642009-01-31 22:14:21 +00009626 if (c == '*') {
9627 v = getnextarg(args, arglen, &argidx);
9628 if (v == NULL)
9629 goto onError;
9630 if (!PyLong_Check(v)) {
9631 PyErr_SetString(PyExc_TypeError,
9632 "* wants int");
9633 goto onError;
9634 }
9635 width = PyLong_AsLong(v);
9636 if (width == -1 && PyErr_Occurred())
9637 goto onError;
9638 if (width < 0) {
9639 flags |= F_LJUST;
9640 width = -width;
9641 }
9642 if (--fmtcnt >= 0)
9643 c = *fmt++;
9644 }
9645 else if (c >= '0' && c <= '9') {
9646 width = c - '0';
9647 while (--fmtcnt >= 0) {
9648 c = *fmt++;
9649 if (c < '0' || c > '9')
9650 break;
Mark Dickinsonfb90c092012-10-28 10:18:03 +00009651 if (width > (PY_SSIZE_T_MAX - ((int)c - '0')) / 10) {
Benjamin Peterson29060642009-01-31 22:14:21 +00009652 PyErr_SetString(PyExc_ValueError,
9653 "width too big");
Benjamin Peterson14339b62009-01-31 16:36:08 +00009654 goto onError;
Benjamin Peterson29060642009-01-31 22:14:21 +00009655 }
9656 width = width*10 + (c - '0');
9657 }
9658 }
9659 if (c == '.') {
9660 prec = 0;
9661 if (--fmtcnt >= 0)
9662 c = *fmt++;
9663 if (c == '*') {
9664 v = getnextarg(args, arglen, &argidx);
9665 if (v == NULL)
9666 goto onError;
9667 if (!PyLong_Check(v)) {
9668 PyErr_SetString(PyExc_TypeError,
9669 "* wants int");
9670 goto onError;
9671 }
9672 prec = PyLong_AsLong(v);
9673 if (prec == -1 && PyErr_Occurred())
9674 goto onError;
9675 if (prec < 0)
9676 prec = 0;
9677 if (--fmtcnt >= 0)
9678 c = *fmt++;
9679 }
9680 else if (c >= '0' && c <= '9') {
9681 prec = c - '0';
9682 while (--fmtcnt >= 0) {
Stefan Krah99212f62010-07-19 17:58:26 +00009683 c = *fmt++;
Benjamin Peterson29060642009-01-31 22:14:21 +00009684 if (c < '0' || c > '9')
9685 break;
Mark Dickinsonfb90c092012-10-28 10:18:03 +00009686 if (prec > (INT_MAX - ((int)c - '0')) / 10) {
Benjamin Peterson29060642009-01-31 22:14:21 +00009687 PyErr_SetString(PyExc_ValueError,
9688 "prec too big");
9689 goto onError;
9690 }
9691 prec = prec*10 + (c - '0');
9692 }
9693 }
9694 } /* prec */
9695 if (fmtcnt >= 0) {
9696 if (c == 'h' || c == 'l' || c == 'L') {
9697 if (--fmtcnt >= 0)
9698 c = *fmt++;
9699 }
9700 }
9701 if (fmtcnt < 0) {
9702 PyErr_SetString(PyExc_ValueError,
9703 "incomplete format");
9704 goto onError;
9705 }
9706 if (c != '%') {
9707 v = getnextarg(args, arglen, &argidx);
9708 if (v == NULL)
9709 goto onError;
9710 }
9711 sign = 0;
9712 fill = ' ';
9713 switch (c) {
9714
9715 case '%':
9716 pbuf = formatbuf;
9717 /* presume that buffer length is at least 1 */
9718 pbuf[0] = '%';
9719 len = 1;
9720 break;
9721
9722 case 's':
9723 case 'r':
9724 case 'a':
Victor Stinner808fc0a2010-03-22 12:50:40 +00009725 if (PyUnicode_CheckExact(v) && c == 's') {
Benjamin Peterson29060642009-01-31 22:14:21 +00009726 temp = v;
9727 Py_INCREF(temp);
Benjamin Peterson14339b62009-01-31 16:36:08 +00009728 }
9729 else {
Benjamin Peterson29060642009-01-31 22:14:21 +00009730 if (c == 's')
9731 temp = PyObject_Str(v);
9732 else if (c == 'r')
9733 temp = PyObject_Repr(v);
9734 else
9735 temp = PyObject_ASCII(v);
9736 if (temp == NULL)
9737 goto onError;
9738 if (PyUnicode_Check(temp))
9739 /* nothing to do */;
9740 else {
9741 Py_DECREF(temp);
9742 PyErr_SetString(PyExc_TypeError,
9743 "%s argument has non-string str()");
9744 goto onError;
9745 }
9746 }
9747 pbuf = PyUnicode_AS_UNICODE(temp);
9748 len = PyUnicode_GET_SIZE(temp);
9749 if (prec >= 0 && len > prec)
9750 len = prec;
9751 break;
9752
9753 case 'i':
9754 case 'd':
9755 case 'u':
9756 case 'o':
9757 case 'x':
9758 case 'X':
Benjamin Peterson29060642009-01-31 22:14:21 +00009759 isnumok = 0;
9760 if (PyNumber_Check(v)) {
9761 PyObject *iobj=NULL;
9762
9763 if (PyLong_Check(v)) {
9764 iobj = v;
9765 Py_INCREF(iobj);
9766 }
9767 else {
9768 iobj = PyNumber_Long(v);
9769 }
9770 if (iobj!=NULL) {
9771 if (PyLong_Check(iobj)) {
9772 isnumok = 1;
Senthil Kumaran9ebe08d2011-07-03 21:03:16 -07009773 temp = formatlong(iobj, flags, prec, (c == 'i'? 'd': c));
Benjamin Peterson29060642009-01-31 22:14:21 +00009774 Py_DECREF(iobj);
9775 if (!temp)
9776 goto onError;
9777 pbuf = PyUnicode_AS_UNICODE(temp);
9778 len = PyUnicode_GET_SIZE(temp);
9779 sign = 1;
9780 }
9781 else {
9782 Py_DECREF(iobj);
9783 }
9784 }
9785 }
9786 if (!isnumok) {
9787 PyErr_Format(PyExc_TypeError,
9788 "%%%c format: a number is required, "
9789 "not %.200s", (char)c, Py_TYPE(v)->tp_name);
9790 goto onError;
9791 }
9792 if (flags & F_ZERO)
9793 fill = '0';
9794 break;
9795
9796 case 'e':
9797 case 'E':
9798 case 'f':
9799 case 'F':
9800 case 'g':
9801 case 'G':
Mark Dickinsonf489caf2009-05-01 11:42:00 +00009802 temp = formatfloat(v, flags, prec, c);
9803 if (!temp)
Benjamin Peterson29060642009-01-31 22:14:21 +00009804 goto onError;
Mark Dickinsonf489caf2009-05-01 11:42:00 +00009805 pbuf = PyUnicode_AS_UNICODE(temp);
9806 len = PyUnicode_GET_SIZE(temp);
Benjamin Peterson29060642009-01-31 22:14:21 +00009807 sign = 1;
9808 if (flags & F_ZERO)
9809 fill = '0';
9810 break;
9811
9812 case 'c':
9813 pbuf = formatbuf;
9814 len = formatchar(pbuf, sizeof(formatbuf)/sizeof(Py_UNICODE), v);
9815 if (len < 0)
9816 goto onError;
9817 break;
9818
9819 default:
9820 PyErr_Format(PyExc_ValueError,
9821 "unsupported format character '%c' (0x%x) "
9822 "at index %zd",
9823 (31<=c && c<=126) ? (char)c : '?',
9824 (int)c,
9825 (Py_ssize_t)(fmt - 1 -
9826 PyUnicode_AS_UNICODE(uformat)));
9827 goto onError;
9828 }
9829 if (sign) {
9830 if (*pbuf == '-' || *pbuf == '+') {
9831 sign = *pbuf++;
9832 len--;
9833 }
9834 else if (flags & F_SIGN)
9835 sign = '+';
9836 else if (flags & F_BLANK)
9837 sign = ' ';
9838 else
9839 sign = 0;
9840 }
9841 if (width < len)
9842 width = len;
9843 if (rescnt - (sign != 0) < width) {
9844 reslen -= rescnt;
9845 rescnt = width + fmtcnt + 100;
9846 reslen += rescnt;
9847 if (reslen < 0) {
9848 Py_XDECREF(temp);
9849 PyErr_NoMemory();
9850 goto onError;
9851 }
9852 if (_PyUnicode_Resize(&result, reslen) < 0) {
9853 Py_XDECREF(temp);
9854 goto onError;
9855 }
9856 res = PyUnicode_AS_UNICODE(result)
9857 + reslen - rescnt;
9858 }
9859 if (sign) {
9860 if (fill != ' ')
9861 *res++ = sign;
9862 rescnt--;
9863 if (width > len)
9864 width--;
9865 }
9866 if ((flags & F_ALT) && (c == 'x' || c == 'X' || c == 'o')) {
9867 assert(pbuf[0] == '0');
9868 assert(pbuf[1] == c);
9869 if (fill != ' ') {
9870 *res++ = *pbuf++;
9871 *res++ = *pbuf++;
9872 }
9873 rescnt -= 2;
9874 width -= 2;
9875 if (width < 0)
9876 width = 0;
9877 len -= 2;
9878 }
9879 if (width > len && !(flags & F_LJUST)) {
9880 do {
9881 --rescnt;
9882 *res++ = fill;
9883 } while (--width > len);
9884 }
9885 if (fill == ' ') {
9886 if (sign)
9887 *res++ = sign;
9888 if ((flags & F_ALT) && (c == 'x' || c == 'X' || c == 'o')) {
9889 assert(pbuf[0] == '0');
9890 assert(pbuf[1] == c);
9891 *res++ = *pbuf++;
9892 *res++ = *pbuf++;
Benjamin Peterson14339b62009-01-31 16:36:08 +00009893 }
9894 }
Benjamin Peterson29060642009-01-31 22:14:21 +00009895 Py_UNICODE_COPY(res, pbuf, len);
9896 res += len;
9897 rescnt -= len;
9898 while (--width >= len) {
9899 --rescnt;
9900 *res++ = ' ';
9901 }
9902 if (dict && (argidx < arglen) && c != '%') {
9903 PyErr_SetString(PyExc_TypeError,
9904 "not all arguments converted during string formatting");
Thomas Woutersa96affe2006-03-12 00:29:36 +00009905 Py_XDECREF(temp);
Benjamin Peterson29060642009-01-31 22:14:21 +00009906 goto onError;
9907 }
9908 Py_XDECREF(temp);
9909 } /* '%' */
Guido van Rossumd57fd912000-03-10 22:53:23 +00009910 } /* until end */
9911 if (argidx < arglen && !dict) {
Benjamin Peterson29060642009-01-31 22:14:21 +00009912 PyErr_SetString(PyExc_TypeError,
9913 "not all arguments converted during string formatting");
9914 goto onError;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009915 }
9916
Thomas Woutersa96affe2006-03-12 00:29:36 +00009917 if (_PyUnicode_Resize(&result, reslen - rescnt) < 0)
Benjamin Peterson29060642009-01-31 22:14:21 +00009918 goto onError;
Guido van Rossumd57fd912000-03-10 22:53:23 +00009919 if (args_owned) {
Benjamin Peterson29060642009-01-31 22:14:21 +00009920 Py_DECREF(args);
Guido van Rossumd57fd912000-03-10 22:53:23 +00009921 }
9922 Py_DECREF(uformat);
Guido van Rossumd57fd912000-03-10 22:53:23 +00009923 return (PyObject *)result;
9924
Benjamin Peterson29060642009-01-31 22:14:21 +00009925 onError:
Guido van Rossumd57fd912000-03-10 22:53:23 +00009926 Py_XDECREF(result);
9927 Py_DECREF(uformat);
9928 if (args_owned) {
Benjamin Peterson29060642009-01-31 22:14:21 +00009929 Py_DECREF(args);
Guido van Rossumd57fd912000-03-10 22:53:23 +00009930 }
9931 return NULL;
9932}
9933
Jeremy Hylton938ace62002-07-17 16:30:39 +00009934static PyObject *
Guido van Rossume023fe02001-08-30 03:12:59 +00009935unicode_subtype_new(PyTypeObject *type, PyObject *args, PyObject *kwds);
9936
Tim Peters6d6c1a32001-08-02 04:15:00 +00009937static PyObject *
9938unicode_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
9939{
Benjamin Peterson29060642009-01-31 22:14:21 +00009940 PyObject *x = NULL;
Benjamin Peterson14339b62009-01-31 16:36:08 +00009941 static char *kwlist[] = {"object", "encoding", "errors", 0};
9942 char *encoding = NULL;
9943 char *errors = NULL;
Tim Peters6d6c1a32001-08-02 04:15:00 +00009944
Benjamin Peterson14339b62009-01-31 16:36:08 +00009945 if (type != &PyUnicode_Type)
9946 return unicode_subtype_new(type, args, kwds);
9947 if (!PyArg_ParseTupleAndKeywords(args, kwds, "|Oss:str",
Benjamin Peterson29060642009-01-31 22:14:21 +00009948 kwlist, &x, &encoding, &errors))
Benjamin Peterson14339b62009-01-31 16:36:08 +00009949 return NULL;
9950 if (x == NULL)
9951 return (PyObject *)_PyUnicode_New(0);
9952 if (encoding == NULL && errors == NULL)
9953 return PyObject_Str(x);
9954 else
Benjamin Peterson29060642009-01-31 22:14:21 +00009955 return PyUnicode_FromEncodedObject(x, encoding, errors);
Tim Peters6d6c1a32001-08-02 04:15:00 +00009956}
9957
Guido van Rossume023fe02001-08-30 03:12:59 +00009958static PyObject *
9959unicode_subtype_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
9960{
Benjamin Peterson14339b62009-01-31 16:36:08 +00009961 PyUnicodeObject *tmp, *pnew;
9962 Py_ssize_t n;
Guido van Rossume023fe02001-08-30 03:12:59 +00009963
Benjamin Peterson14339b62009-01-31 16:36:08 +00009964 assert(PyType_IsSubtype(type, &PyUnicode_Type));
9965 tmp = (PyUnicodeObject *)unicode_new(&PyUnicode_Type, args, kwds);
9966 if (tmp == NULL)
9967 return NULL;
9968 assert(PyUnicode_Check(tmp));
9969 pnew = (PyUnicodeObject *) type->tp_alloc(type, n = tmp->length);
9970 if (pnew == NULL) {
9971 Py_DECREF(tmp);
9972 return NULL;
9973 }
9974 pnew->str = (Py_UNICODE*) PyObject_MALLOC(sizeof(Py_UNICODE) * (n+1));
9975 if (pnew->str == NULL) {
9976 _Py_ForgetReference((PyObject *)pnew);
9977 PyObject_Del(pnew);
9978 Py_DECREF(tmp);
9979 return PyErr_NoMemory();
9980 }
9981 Py_UNICODE_COPY(pnew->str, tmp->str, n+1);
9982 pnew->length = n;
9983 pnew->hash = tmp->hash;
9984 Py_DECREF(tmp);
9985 return (PyObject *)pnew;
Guido van Rossume023fe02001-08-30 03:12:59 +00009986}
9987
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00009988PyDoc_STRVAR(unicode_doc,
Chris Jerdonek83fe2e12012-10-07 14:48:36 -07009989"str(object='') -> str\n\
9990str(bytes_or_buffer[, encoding[, errors]]) -> str\n\
Tim Peters6d6c1a32001-08-02 04:15:00 +00009991\n\
Nick Coghlan573b1fd2012-08-16 14:13:07 +10009992Create a new string object from the given object. If encoding or\n\
9993errors is specified, then the object must expose a data buffer\n\
9994that will be decoded using the given encoding and error handler.\n\
9995Otherwise, returns the result of object.__str__() (if defined)\n\
9996or repr(object).\n\
9997encoding defaults to sys.getdefaultencoding().\n\
9998errors defaults to 'strict'.");
Tim Peters6d6c1a32001-08-02 04:15:00 +00009999
Guido van Rossum50e9fb92006-08-17 05:42:55 +000010000static PyObject *unicode_iter(PyObject *seq);
10001
Guido van Rossumd57fd912000-03-10 22:53:23 +000010002PyTypeObject PyUnicode_Type = {
Martin v. Löwis9f2e3462007-07-21 17:22:18 +000010003 PyVarObject_HEAD_INIT(&PyType_Type, 0)
Benjamin Peterson14339b62009-01-31 16:36:08 +000010004 "str", /* tp_name */
10005 sizeof(PyUnicodeObject), /* tp_size */
10006 0, /* tp_itemsize */
Guido van Rossumd57fd912000-03-10 22:53:23 +000010007 /* Slots */
Benjamin Peterson14339b62009-01-31 16:36:08 +000010008 (destructor)unicode_dealloc, /* tp_dealloc */
10009 0, /* tp_print */
10010 0, /* tp_getattr */
10011 0, /* tp_setattr */
Mark Dickinsone94c6792009-02-02 20:36:42 +000010012 0, /* tp_reserved */
Benjamin Peterson14339b62009-01-31 16:36:08 +000010013 unicode_repr, /* tp_repr */
10014 &unicode_as_number, /* tp_as_number */
10015 &unicode_as_sequence, /* tp_as_sequence */
10016 &unicode_as_mapping, /* tp_as_mapping */
10017 (hashfunc) unicode_hash, /* tp_hash*/
10018 0, /* tp_call*/
10019 (reprfunc) unicode_str, /* tp_str */
10020 PyObject_GenericGetAttr, /* tp_getattro */
10021 0, /* tp_setattro */
10022 0, /* tp_as_buffer */
10023 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE |
Benjamin Peterson29060642009-01-31 22:14:21 +000010024 Py_TPFLAGS_UNICODE_SUBCLASS, /* tp_flags */
Benjamin Peterson14339b62009-01-31 16:36:08 +000010025 unicode_doc, /* tp_doc */
10026 0, /* tp_traverse */
10027 0, /* tp_clear */
10028 PyUnicode_RichCompare, /* tp_richcompare */
10029 0, /* tp_weaklistoffset */
10030 unicode_iter, /* tp_iter */
10031 0, /* tp_iternext */
10032 unicode_methods, /* tp_methods */
10033 0, /* tp_members */
10034 0, /* tp_getset */
10035 &PyBaseObject_Type, /* tp_base */
10036 0, /* tp_dict */
10037 0, /* tp_descr_get */
10038 0, /* tp_descr_set */
10039 0, /* tp_dictoffset */
10040 0, /* tp_init */
10041 0, /* tp_alloc */
10042 unicode_new, /* tp_new */
10043 PyObject_Del, /* tp_free */
Guido van Rossumd57fd912000-03-10 22:53:23 +000010044};
10045
10046/* Initialize the Unicode implementation */
10047
Thomas Wouters78890102000-07-22 19:25:51 +000010048void _PyUnicode_Init(void)
Guido van Rossumd57fd912000-03-10 22:53:23 +000010049{
Marc-André Lemburg8155e0e2001-04-23 14:44:21 +000010050 int i;
10051
Thomas Wouters477c8d52006-05-27 19:21:47 +000010052 /* XXX - move this array to unicodectype.c ? */
10053 Py_UNICODE linebreak[] = {
10054 0x000A, /* LINE FEED */
10055 0x000D, /* CARRIAGE RETURN */
10056 0x001C, /* FILE SEPARATOR */
10057 0x001D, /* GROUP SEPARATOR */
10058 0x001E, /* RECORD SEPARATOR */
10059 0x0085, /* NEXT LINE */
10060 0x2028, /* LINE SEPARATOR */
10061 0x2029, /* PARAGRAPH SEPARATOR */
10062 };
10063
Fred Drakee4315f52000-05-09 19:53:39 +000010064 /* Init the implementation */
Christian Heimes2202f872008-02-06 14:31:34 +000010065 free_list = NULL;
10066 numfree = 0;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010067 unicode_empty = _PyUnicode_New(0);
Thomas Wouters0e3f5912006-08-11 14:57:12 +000010068 if (!unicode_empty)
Benjamin Peterson29060642009-01-31 22:14:21 +000010069 return;
Thomas Wouters0e3f5912006-08-11 14:57:12 +000010070
Marc-André Lemburg8155e0e2001-04-23 14:44:21 +000010071 for (i = 0; i < 256; i++)
Benjamin Peterson29060642009-01-31 22:14:21 +000010072 unicode_latin1[i] = NULL;
Guido van Rossumcacfc072002-05-24 19:01:59 +000010073 if (PyType_Ready(&PyUnicode_Type) < 0)
Benjamin Peterson29060642009-01-31 22:14:21 +000010074 Py_FatalError("Can't initialize 'unicode'");
Thomas Wouters477c8d52006-05-27 19:21:47 +000010075
10076 /* initialize the linebreak bloom filter */
10077 bloom_linebreak = make_bloom_mask(
10078 linebreak, sizeof(linebreak) / sizeof(linebreak[0])
10079 );
Thomas Wouters0e3f5912006-08-11 14:57:12 +000010080
10081 PyType_Ready(&EncodingMapType);
Benjamin Petersonc4311282012-10-30 23:21:10 -040010082
10083 if (PyType_Ready(&PyFieldNameIter_Type) < 0)
10084 Py_FatalError("Can't initialize field name iterator type");
10085
10086 if (PyType_Ready(&PyFormatterIter_Type) < 0)
10087 Py_FatalError("Can't initialize formatter iter type");
Guido van Rossumd57fd912000-03-10 22:53:23 +000010088}
10089
10090/* Finalize the Unicode implementation */
10091
Christian Heimesa156e092008-02-16 07:38:31 +000010092int
10093PyUnicode_ClearFreeList(void)
10094{
10095 int freelist_size = numfree;
10096 PyUnicodeObject *u;
10097
10098 for (u = free_list; u != NULL;) {
Benjamin Peterson29060642009-01-31 22:14:21 +000010099 PyUnicodeObject *v = u;
10100 u = *(PyUnicodeObject **)u;
10101 if (v->str)
10102 PyObject_DEL(v->str);
10103 Py_XDECREF(v->defenc);
10104 PyObject_Del(v);
10105 numfree--;
Christian Heimesa156e092008-02-16 07:38:31 +000010106 }
10107 free_list = NULL;
10108 assert(numfree == 0);
10109 return freelist_size;
10110}
10111
Guido van Rossumd57fd912000-03-10 22:53:23 +000010112void
Thomas Wouters78890102000-07-22 19:25:51 +000010113_PyUnicode_Fini(void)
Guido van Rossumd57fd912000-03-10 22:53:23 +000010114{
Marc-André Lemburg8155e0e2001-04-23 14:44:21 +000010115 int i;
Guido van Rossumd57fd912000-03-10 22:53:23 +000010116
Guido van Rossum4ae8ef82000-10-03 18:09:04 +000010117 Py_XDECREF(unicode_empty);
10118 unicode_empty = NULL;
Barry Warsaw5b4c2282000-10-03 20:45:26 +000010119
Marc-André Lemburg8155e0e2001-04-23 14:44:21 +000010120 for (i = 0; i < 256; i++) {
Benjamin Peterson29060642009-01-31 22:14:21 +000010121 if (unicode_latin1[i]) {
10122 Py_DECREF(unicode_latin1[i]);
10123 unicode_latin1[i] = NULL;
10124 }
Marc-André Lemburg8155e0e2001-04-23 14:44:21 +000010125 }
Christian Heimesa156e092008-02-16 07:38:31 +000010126 (void)PyUnicode_ClearFreeList();
Guido van Rossumd57fd912000-03-10 22:53:23 +000010127}
Martin v. Löwis9a3a9f72003-05-18 12:31:09 +000010128
Walter Dörwald16807132007-05-25 13:52:07 +000010129void
10130PyUnicode_InternInPlace(PyObject **p)
10131{
Benjamin Peterson14339b62009-01-31 16:36:08 +000010132 register PyUnicodeObject *s = (PyUnicodeObject *)(*p);
10133 PyObject *t;
10134 if (s == NULL || !PyUnicode_Check(s))
10135 Py_FatalError(
10136 "PyUnicode_InternInPlace: unicode strings only please!");
10137 /* If it's a subclass, we don't really know what putting
10138 it in the interned dict might do. */
10139 if (!PyUnicode_CheckExact(s))
10140 return;
10141 if (PyUnicode_CHECK_INTERNED(s))
10142 return;
10143 if (interned == NULL) {
10144 interned = PyDict_New();
10145 if (interned == NULL) {
10146 PyErr_Clear(); /* Don't leave an exception */
10147 return;
10148 }
10149 }
10150 /* It might be that the GetItem call fails even
10151 though the key is present in the dictionary,
10152 namely when this happens during a stack overflow. */
10153 Py_ALLOW_RECURSION
Benjamin Peterson29060642009-01-31 22:14:21 +000010154 t = PyDict_GetItem(interned, (PyObject *)s);
Benjamin Peterson14339b62009-01-31 16:36:08 +000010155 Py_END_ALLOW_RECURSION
Martin v. Löwis5b222132007-06-10 09:51:05 +000010156
Benjamin Peterson29060642009-01-31 22:14:21 +000010157 if (t) {
10158 Py_INCREF(t);
10159 Py_DECREF(*p);
10160 *p = t;
10161 return;
10162 }
Walter Dörwald16807132007-05-25 13:52:07 +000010163
Benjamin Peterson14339b62009-01-31 16:36:08 +000010164 PyThreadState_GET()->recursion_critical = 1;
10165 if (PyDict_SetItem(interned, (PyObject *)s, (PyObject *)s) < 0) {
10166 PyErr_Clear();
10167 PyThreadState_GET()->recursion_critical = 0;
10168 return;
10169 }
10170 PyThreadState_GET()->recursion_critical = 0;
10171 /* The two references in interned are not counted by refcnt.
10172 The deallocator will take care of this */
10173 Py_REFCNT(s) -= 2;
10174 PyUnicode_CHECK_INTERNED(s) = SSTATE_INTERNED_MORTAL;
Walter Dörwald16807132007-05-25 13:52:07 +000010175}
10176
10177void
10178PyUnicode_InternImmortal(PyObject **p)
10179{
Benjamin Peterson14339b62009-01-31 16:36:08 +000010180 PyUnicode_InternInPlace(p);
10181 if (PyUnicode_CHECK_INTERNED(*p) != SSTATE_INTERNED_IMMORTAL) {
10182 PyUnicode_CHECK_INTERNED(*p) = SSTATE_INTERNED_IMMORTAL;
10183 Py_INCREF(*p);
10184 }
Walter Dörwald16807132007-05-25 13:52:07 +000010185}
10186
10187PyObject *
10188PyUnicode_InternFromString(const char *cp)
10189{
Benjamin Peterson14339b62009-01-31 16:36:08 +000010190 PyObject *s = PyUnicode_FromString(cp);
10191 if (s == NULL)
10192 return NULL;
10193 PyUnicode_InternInPlace(&s);
10194 return s;
Walter Dörwald16807132007-05-25 13:52:07 +000010195}
10196
10197void _Py_ReleaseInternedUnicodeStrings(void)
10198{
Benjamin Peterson14339b62009-01-31 16:36:08 +000010199 PyObject *keys;
10200 PyUnicodeObject *s;
10201 Py_ssize_t i, n;
10202 Py_ssize_t immortal_size = 0, mortal_size = 0;
Walter Dörwald16807132007-05-25 13:52:07 +000010203
Benjamin Peterson14339b62009-01-31 16:36:08 +000010204 if (interned == NULL || !PyDict_Check(interned))
10205 return;
10206 keys = PyDict_Keys(interned);
10207 if (keys == NULL || !PyList_Check(keys)) {
10208 PyErr_Clear();
10209 return;
10210 }
Walter Dörwald16807132007-05-25 13:52:07 +000010211
Benjamin Peterson14339b62009-01-31 16:36:08 +000010212 /* Since _Py_ReleaseInternedUnicodeStrings() is intended to help a leak
10213 detector, interned unicode strings are not forcibly deallocated;
10214 rather, we give them their stolen references back, and then clear
10215 and DECREF the interned dict. */
Walter Dörwald16807132007-05-25 13:52:07 +000010216
Benjamin Peterson14339b62009-01-31 16:36:08 +000010217 n = PyList_GET_SIZE(keys);
10218 fprintf(stderr, "releasing %" PY_FORMAT_SIZE_T "d interned strings\n",
Benjamin Peterson29060642009-01-31 22:14:21 +000010219 n);
Benjamin Peterson14339b62009-01-31 16:36:08 +000010220 for (i = 0; i < n; i++) {
10221 s = (PyUnicodeObject *) PyList_GET_ITEM(keys, i);
10222 switch (s->state) {
10223 case SSTATE_NOT_INTERNED:
10224 /* XXX Shouldn't happen */
10225 break;
10226 case SSTATE_INTERNED_IMMORTAL:
10227 Py_REFCNT(s) += 1;
10228 immortal_size += s->length;
10229 break;
10230 case SSTATE_INTERNED_MORTAL:
10231 Py_REFCNT(s) += 2;
10232 mortal_size += s->length;
10233 break;
10234 default:
10235 Py_FatalError("Inconsistent interned string state.");
10236 }
10237 s->state = SSTATE_NOT_INTERNED;
10238 }
10239 fprintf(stderr, "total size of all interned strings: "
10240 "%" PY_FORMAT_SIZE_T "d/%" PY_FORMAT_SIZE_T "d "
10241 "mortal/immortal\n", mortal_size, immortal_size);
10242 Py_DECREF(keys);
10243 PyDict_Clear(interned);
10244 Py_DECREF(interned);
10245 interned = NULL;
Walter Dörwald16807132007-05-25 13:52:07 +000010246}
Guido van Rossum50e9fb92006-08-17 05:42:55 +000010247
10248
10249/********************* Unicode Iterator **************************/
10250
10251typedef struct {
Benjamin Peterson14339b62009-01-31 16:36:08 +000010252 PyObject_HEAD
10253 Py_ssize_t it_index;
10254 PyUnicodeObject *it_seq; /* Set to NULL when iterator is exhausted */
Guido van Rossum50e9fb92006-08-17 05:42:55 +000010255} unicodeiterobject;
10256
10257static void
10258unicodeiter_dealloc(unicodeiterobject *it)
10259{
Benjamin Peterson14339b62009-01-31 16:36:08 +000010260 _PyObject_GC_UNTRACK(it);
10261 Py_XDECREF(it->it_seq);
10262 PyObject_GC_Del(it);
Guido van Rossum50e9fb92006-08-17 05:42:55 +000010263}
10264
10265static int
10266unicodeiter_traverse(unicodeiterobject *it, visitproc visit, void *arg)
10267{
Benjamin Peterson14339b62009-01-31 16:36:08 +000010268 Py_VISIT(it->it_seq);
10269 return 0;
Guido van Rossum50e9fb92006-08-17 05:42:55 +000010270}
10271
10272static PyObject *
10273unicodeiter_next(unicodeiterobject *it)
10274{
Benjamin Peterson14339b62009-01-31 16:36:08 +000010275 PyUnicodeObject *seq;
10276 PyObject *item;
Guido van Rossum50e9fb92006-08-17 05:42:55 +000010277
Benjamin Peterson14339b62009-01-31 16:36:08 +000010278 assert(it != NULL);
10279 seq = it->it_seq;
10280 if (seq == NULL)
10281 return NULL;
10282 assert(PyUnicode_Check(seq));
Guido van Rossum50e9fb92006-08-17 05:42:55 +000010283
Benjamin Peterson14339b62009-01-31 16:36:08 +000010284 if (it->it_index < PyUnicode_GET_SIZE(seq)) {
10285 item = PyUnicode_FromUnicode(
Benjamin Peterson29060642009-01-31 22:14:21 +000010286 PyUnicode_AS_UNICODE(seq)+it->it_index, 1);
Benjamin Peterson14339b62009-01-31 16:36:08 +000010287 if (item != NULL)
10288 ++it->it_index;
10289 return item;
10290 }
Guido van Rossum50e9fb92006-08-17 05:42:55 +000010291
Benjamin Peterson14339b62009-01-31 16:36:08 +000010292 Py_DECREF(seq);
10293 it->it_seq = NULL;
10294 return NULL;
Guido van Rossum50e9fb92006-08-17 05:42:55 +000010295}
10296
10297static PyObject *
10298unicodeiter_len(unicodeiterobject *it)
10299{
Benjamin Peterson14339b62009-01-31 16:36:08 +000010300 Py_ssize_t len = 0;
10301 if (it->it_seq)
10302 len = PyUnicode_GET_SIZE(it->it_seq) - it->it_index;
10303 return PyLong_FromSsize_t(len);
Guido van Rossum50e9fb92006-08-17 05:42:55 +000010304}
10305
10306PyDoc_STRVAR(length_hint_doc, "Private method returning an estimate of len(list(it)).");
10307
10308static PyMethodDef unicodeiter_methods[] = {
Benjamin Peterson14339b62009-01-31 16:36:08 +000010309 {"__length_hint__", (PyCFunction)unicodeiter_len, METH_NOARGS,
Benjamin Peterson29060642009-01-31 22:14:21 +000010310 length_hint_doc},
Benjamin Peterson14339b62009-01-31 16:36:08 +000010311 {NULL, NULL} /* sentinel */
Guido van Rossum50e9fb92006-08-17 05:42:55 +000010312};
10313
10314PyTypeObject PyUnicodeIter_Type = {
Benjamin Peterson14339b62009-01-31 16:36:08 +000010315 PyVarObject_HEAD_INIT(&PyType_Type, 0)
10316 "str_iterator", /* tp_name */
10317 sizeof(unicodeiterobject), /* tp_basicsize */
10318 0, /* tp_itemsize */
10319 /* methods */
10320 (destructor)unicodeiter_dealloc, /* tp_dealloc */
10321 0, /* tp_print */
10322 0, /* tp_getattr */
10323 0, /* tp_setattr */
Mark Dickinsone94c6792009-02-02 20:36:42 +000010324 0, /* tp_reserved */
Benjamin Peterson14339b62009-01-31 16:36:08 +000010325 0, /* tp_repr */
10326 0, /* tp_as_number */
10327 0, /* tp_as_sequence */
10328 0, /* tp_as_mapping */
10329 0, /* tp_hash */
10330 0, /* tp_call */
10331 0, /* tp_str */
10332 PyObject_GenericGetAttr, /* tp_getattro */
10333 0, /* tp_setattro */
10334 0, /* tp_as_buffer */
10335 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC,/* tp_flags */
10336 0, /* tp_doc */
10337 (traverseproc)unicodeiter_traverse, /* tp_traverse */
10338 0, /* tp_clear */
10339 0, /* tp_richcompare */
10340 0, /* tp_weaklistoffset */
10341 PyObject_SelfIter, /* tp_iter */
10342 (iternextfunc)unicodeiter_next, /* tp_iternext */
10343 unicodeiter_methods, /* tp_methods */
10344 0,
Guido van Rossum50e9fb92006-08-17 05:42:55 +000010345};
10346
10347static PyObject *
10348unicode_iter(PyObject *seq)
10349{
Benjamin Peterson14339b62009-01-31 16:36:08 +000010350 unicodeiterobject *it;
Guido van Rossum50e9fb92006-08-17 05:42:55 +000010351
Benjamin Peterson14339b62009-01-31 16:36:08 +000010352 if (!PyUnicode_Check(seq)) {
10353 PyErr_BadInternalCall();
10354 return NULL;
10355 }
10356 it = PyObject_GC_New(unicodeiterobject, &PyUnicodeIter_Type);
10357 if (it == NULL)
10358 return NULL;
10359 it->it_index = 0;
10360 Py_INCREF(seq);
10361 it->it_seq = (PyUnicodeObject *)seq;
10362 _PyObject_GC_TRACK(it);
10363 return (PyObject *)it;
Guido van Rossum50e9fb92006-08-17 05:42:55 +000010364}
10365
Martin v. Löwis5b222132007-06-10 09:51:05 +000010366size_t
10367Py_UNICODE_strlen(const Py_UNICODE *u)
10368{
10369 int res = 0;
10370 while(*u++)
10371 res++;
10372 return res;
10373}
10374
10375Py_UNICODE*
10376Py_UNICODE_strcpy(Py_UNICODE *s1, const Py_UNICODE *s2)
10377{
10378 Py_UNICODE *u = s1;
10379 while ((*u++ = *s2++));
10380 return s1;
10381}
10382
10383Py_UNICODE*
10384Py_UNICODE_strncpy(Py_UNICODE *s1, const Py_UNICODE *s2, size_t n)
10385{
10386 Py_UNICODE *u = s1;
10387 while ((*u++ = *s2++))
10388 if (n-- == 0)
10389 break;
10390 return s1;
10391}
10392
Victor Stinnerc4eb7652010-09-01 23:43:50 +000010393Py_UNICODE*
10394Py_UNICODE_strcat(Py_UNICODE *s1, const Py_UNICODE *s2)
10395{
10396 Py_UNICODE *u1 = s1;
10397 u1 += Py_UNICODE_strlen(u1);
10398 Py_UNICODE_strcpy(u1, s2);
10399 return s1;
10400}
10401
Martin v. Löwis5b222132007-06-10 09:51:05 +000010402int
10403Py_UNICODE_strcmp(const Py_UNICODE *s1, const Py_UNICODE *s2)
10404{
10405 while (*s1 && *s2 && *s1 == *s2)
10406 s1++, s2++;
10407 if (*s1 && *s2)
10408 return (*s1 < *s2) ? -1 : +1;
10409 if (*s1)
10410 return 1;
10411 if (*s2)
10412 return -1;
10413 return 0;
10414}
10415
Victor Stinneref8d95c2010-08-16 22:03:11 +000010416int
10417Py_UNICODE_strncmp(const Py_UNICODE *s1, const Py_UNICODE *s2, size_t n)
10418{
10419 register Py_UNICODE u1, u2;
10420 for (; n != 0; n--) {
10421 u1 = *s1;
10422 u2 = *s2;
10423 if (u1 != u2)
10424 return (u1 < u2) ? -1 : +1;
10425 if (u1 == '\0')
10426 return 0;
10427 s1++;
10428 s2++;
10429 }
10430 return 0;
10431}
10432
Martin v. Löwis5b222132007-06-10 09:51:05 +000010433Py_UNICODE*
10434Py_UNICODE_strchr(const Py_UNICODE *s, Py_UNICODE c)
10435{
10436 const Py_UNICODE *p;
10437 for (p = s; *p; p++)
10438 if (*p == c)
10439 return (Py_UNICODE*)p;
10440 return NULL;
10441}
10442
Victor Stinner331ea922010-08-10 16:37:20 +000010443Py_UNICODE*
10444Py_UNICODE_strrchr(const Py_UNICODE *s, Py_UNICODE c)
10445{
10446 const Py_UNICODE *p;
10447 p = s + Py_UNICODE_strlen(s);
10448 while (p != s) {
10449 p--;
10450 if (*p == c)
10451 return (Py_UNICODE*)p;
10452 }
10453 return NULL;
10454}
10455
Victor Stinner71133ff2010-09-01 23:43:53 +000010456Py_UNICODE*
Victor Stinner46408602010-09-03 16:18:00 +000010457PyUnicode_AsUnicodeCopy(PyObject *object)
Victor Stinner71133ff2010-09-01 23:43:53 +000010458{
10459 PyUnicodeObject *unicode = (PyUnicodeObject *)object;
10460 Py_UNICODE *copy;
10461 Py_ssize_t size;
10462
10463 /* Ensure we won't overflow the size. */
10464 if (PyUnicode_GET_SIZE(unicode) > ((PY_SSIZE_T_MAX / sizeof(Py_UNICODE)) - 1)) {
10465 PyErr_NoMemory();
10466 return NULL;
10467 }
10468 size = PyUnicode_GET_SIZE(unicode) + 1; /* copy the nul character */
10469 size *= sizeof(Py_UNICODE);
10470 copy = PyMem_Malloc(size);
10471 if (copy == NULL) {
10472 PyErr_NoMemory();
10473 return NULL;
10474 }
10475 memcpy(copy, PyUnicode_AS_UNICODE(unicode), size);
10476 return copy;
10477}
Martin v. Löwis5b222132007-06-10 09:51:05 +000010478
Georg Brandl66c221e2010-10-14 07:04:07 +000010479/* A _string module, to export formatter_parser and formatter_field_name_split
10480 to the string.Formatter class implemented in Python. */
10481
10482static PyMethodDef _string_methods[] = {
10483 {"formatter_field_name_split", (PyCFunction) formatter_field_name_split,
10484 METH_O, PyDoc_STR("split the argument as a field name")},
10485 {"formatter_parser", (PyCFunction) formatter_parser,
10486 METH_O, PyDoc_STR("parse the argument as a format string")},
10487 {NULL, NULL}
10488};
10489
10490static struct PyModuleDef _string_module = {
10491 PyModuleDef_HEAD_INIT,
10492 "_string",
10493 PyDoc_STR("string helper module"),
10494 0,
10495 _string_methods,
10496 NULL,
10497 NULL,
10498 NULL,
10499 NULL
10500};
10501
10502PyMODINIT_FUNC
10503PyInit__string(void)
10504{
10505 return PyModule_Create(&_string_module);
10506}
10507
10508
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000010509#ifdef __cplusplus
10510}
10511#endif