blob: f306cb8e790be427f7d705e7d12008a6f7ea090b [file] [log] [blame]
Benjamin Peterson4116f362008-05-27 00:36:20 +00001/* bytes object implementation */
Guido van Rossum4dfe8a12006-04-22 23:28:04 +00002
Guido van Rossum4dfe8a12006-04-22 23:28:04 +00003#define PY_SSIZE_T_CLEAN
Christian Heimes2c9c7a52008-05-26 13:42:13 +00004
Guido van Rossum4dfe8a12006-04-22 23:28:04 +00005#include "Python.h"
Christian Heimes2c9c7a52008-05-26 13:42:13 +00006
Gregory P. Smith60d241f2007-10-16 06:31:30 +00007#include "bytes_methods.h"
Mark Dickinsonfd24b322008-12-06 15:33:31 +00008#include <stddef.h>
Guido van Rossum4dfe8a12006-04-22 23:28:04 +00009
Neal Norwitz2bad9702007-08-27 06:19:22 +000010static Py_ssize_t
Travis E. Oliphant8ae62b62007-09-23 02:00:13 +000011_getbuffer(PyObject *obj, Py_buffer *view)
Guido van Rossumad7d8d12007-04-13 01:39:34 +000012{
Christian Heimes90aa7642007-12-19 02:45:37 +000013 PyBufferProcs *buffer = Py_TYPE(obj)->tp_as_buffer;
Guido van Rossumad7d8d12007-04-13 01:39:34 +000014
Gregory P. Smith60d241f2007-10-16 06:31:30 +000015 if (buffer == NULL || buffer->bf_getbuffer == NULL)
Guido van Rossuma74184e2007-08-29 04:05:57 +000016 {
17 PyErr_Format(PyExc_TypeError,
18 "Type %.100s doesn't support the buffer API",
Christian Heimes90aa7642007-12-19 02:45:37 +000019 Py_TYPE(obj)->tp_name);
Guido van Rossuma74184e2007-08-29 04:05:57 +000020 return -1;
21 }
Guido van Rossumad7d8d12007-04-13 01:39:34 +000022
Travis E. Oliphantb99f7622007-08-18 11:21:56 +000023 if (buffer->bf_getbuffer(obj, view, PyBUF_SIMPLE) < 0)
24 return -1;
25 return view->len;
Guido van Rossumad7d8d12007-04-13 01:39:34 +000026}
27
Christian Heimes2c9c7a52008-05-26 13:42:13 +000028#ifdef COUNT_ALLOCS
Benjamin Petersona4a37fe2009-01-11 17:13:55 +000029Py_ssize_t null_strings, one_strings;
Christian Heimes2c9c7a52008-05-26 13:42:13 +000030#endif
Guido van Rossum4dfe8a12006-04-22 23:28:04 +000031
Christian Heimes2c9c7a52008-05-26 13:42:13 +000032static PyBytesObject *characters[UCHAR_MAX + 1];
33static PyBytesObject *nullstring;
34
Mark Dickinsonfd24b322008-12-06 15:33:31 +000035/* PyBytesObject_SIZE gives the basic size of a string; any memory allocation
36 for a string of length n should request PyBytesObject_SIZE + n bytes.
37
38 Using PyBytesObject_SIZE instead of sizeof(PyBytesObject) saves
39 3 bytes per string allocation on a typical system.
40*/
41#define PyBytesObject_SIZE (offsetof(PyBytesObject, ob_sval) + 1)
42
Christian Heimes2c9c7a52008-05-26 13:42:13 +000043/*
44 For both PyBytes_FromString() and PyBytes_FromStringAndSize(), the
45 parameter `size' denotes number of characters to allocate, not counting any
46 null terminating character.
47
48 For PyBytes_FromString(), the parameter `str' points to a null-terminated
49 string containing exactly `size' bytes.
50
51 For PyBytes_FromStringAndSize(), the parameter the parameter `str' is
52 either NULL or else points to a string containing at least `size' bytes.
53 For PyBytes_FromStringAndSize(), the string in the `str' parameter does
54 not have to be null-terminated. (Therefore it is safe to construct a
55 substring by calling `PyBytes_FromStringAndSize(origstring, substrlen)'.)
56 If `str' is NULL then PyBytes_FromStringAndSize() will allocate `size+1'
57 bytes (setting the last byte to the null terminating character) and you can
58 fill in the data yourself. If `str' is non-NULL then the resulting
59 PyString object must be treated as immutable and you must not fill in nor
60 alter the data yourself, since the strings may be shared.
61
62 The PyObject member `op->ob_size', which denotes the number of "extra
63 items" in a variable-size object, will contain the number of bytes
64 allocated for string data, not counting the null terminating character. It
65 is therefore equal to the equal to the `size' parameter (for
66 PyBytes_FromStringAndSize()) or the length of the string in the `str'
67 parameter (for PyBytes_FromString()).
68*/
Guido van Rossum4dfe8a12006-04-22 23:28:04 +000069PyObject *
Christian Heimes2c9c7a52008-05-26 13:42:13 +000070PyBytes_FromStringAndSize(const char *str, Py_ssize_t size)
Guido van Rossumd624f182006-04-24 13:47:05 +000071{
Christian Heimes2c9c7a52008-05-26 13:42:13 +000072 register PyBytesObject *op;
73 if (size < 0) {
74 PyErr_SetString(PyExc_SystemError,
75 "Negative size passed to PyBytes_FromStringAndSize");
76 return NULL;
Guido van Rossum98297ee2007-11-06 21:34:58 +000077 }
Christian Heimes2c9c7a52008-05-26 13:42:13 +000078 if (size == 0 && (op = nullstring) != NULL) {
79#ifdef COUNT_ALLOCS
80 null_strings++;
81#endif
82 Py_INCREF(op);
83 return (PyObject *)op;
84 }
85 if (size == 1 && str != NULL &&
86 (op = characters[*str & UCHAR_MAX]) != NULL)
87 {
88#ifdef COUNT_ALLOCS
89 one_strings++;
90#endif
91 Py_INCREF(op);
92 return (PyObject *)op;
93 }
94
Mark Dickinsonfd24b322008-12-06 15:33:31 +000095 if (size > PY_SSIZE_T_MAX - PyBytesObject_SIZE) {
Neal Norwitz3ce5d922008-08-24 07:08:55 +000096 PyErr_SetString(PyExc_OverflowError,
97 "byte string is too large");
98 return NULL;
99 }
100
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000101 /* Inline PyObject_NewVar */
Mark Dickinsonfd24b322008-12-06 15:33:31 +0000102 op = (PyBytesObject *)PyObject_MALLOC(PyBytesObject_SIZE + size);
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000103 if (op == NULL)
104 return PyErr_NoMemory();
105 PyObject_INIT_VAR(op, &PyBytes_Type, size);
106 op->ob_shash = -1;
107 if (str != NULL)
108 Py_MEMCPY(op->ob_sval, str, size);
109 op->ob_sval[size] = '\0';
110 /* share short strings */
111 if (size == 0) {
112 nullstring = op;
113 Py_INCREF(op);
114 } else if (size == 1 && str != NULL) {
115 characters[*str & UCHAR_MAX] = op;
116 Py_INCREF(op);
117 }
118 return (PyObject *) op;
Guido van Rossumd624f182006-04-24 13:47:05 +0000119}
120
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000121PyObject *
122PyBytes_FromString(const char *str)
Guido van Rossum4dfe8a12006-04-22 23:28:04 +0000123{
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000124 register size_t size;
125 register PyBytesObject *op;
Guido van Rossum4dfe8a12006-04-22 23:28:04 +0000126
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000127 assert(str != NULL);
128 size = strlen(str);
Mark Dickinsonfd24b322008-12-06 15:33:31 +0000129 if (size > PY_SSIZE_T_MAX - PyBytesObject_SIZE) {
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000130 PyErr_SetString(PyExc_OverflowError,
Benjamin Peterson4116f362008-05-27 00:36:20 +0000131 "byte string is too long");
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000132 return NULL;
133 }
134 if (size == 0 && (op = nullstring) != NULL) {
135#ifdef COUNT_ALLOCS
136 null_strings++;
137#endif
138 Py_INCREF(op);
139 return (PyObject *)op;
140 }
141 if (size == 1 && (op = characters[*str & UCHAR_MAX]) != NULL) {
142#ifdef COUNT_ALLOCS
143 one_strings++;
144#endif
145 Py_INCREF(op);
146 return (PyObject *)op;
147 }
Guido van Rossum98297ee2007-11-06 21:34:58 +0000148
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000149 /* Inline PyObject_NewVar */
Mark Dickinsonfd24b322008-12-06 15:33:31 +0000150 op = (PyBytesObject *)PyObject_MALLOC(PyBytesObject_SIZE + size);
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000151 if (op == NULL)
152 return PyErr_NoMemory();
153 PyObject_INIT_VAR(op, &PyBytes_Type, size);
154 op->ob_shash = -1;
155 Py_MEMCPY(op->ob_sval, str, size+1);
156 /* share short strings */
157 if (size == 0) {
158 nullstring = op;
159 Py_INCREF(op);
160 } else if (size == 1) {
161 characters[*str & UCHAR_MAX] = op;
162 Py_INCREF(op);
163 }
164 return (PyObject *) op;
165}
Guido van Rossumebea9be2007-04-09 00:49:13 +0000166
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000167PyObject *
168PyBytes_FromFormatV(const char *format, va_list vargs)
169{
170 va_list count;
171 Py_ssize_t n = 0;
172 const char* f;
173 char *s;
174 PyObject* string;
Guido van Rossum343e97f2007-04-09 00:43:24 +0000175
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000176#ifdef VA_LIST_IS_ARRAY
177 Py_MEMCPY(count, vargs, sizeof(va_list));
178#else
179#ifdef __va_copy
180 __va_copy(count, vargs);
181#else
182 count = vargs;
183#endif
184#endif
185 /* step 1: figure out how large a buffer we need */
186 for (f = format; *f; f++) {
187 if (*f == '%') {
188 const char* p = f;
189 while (*++f && *f != '%' && !ISALPHA(*f))
190 ;
Guido van Rossum343e97f2007-04-09 00:43:24 +0000191
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000192 /* skip the 'l' or 'z' in {%ld, %zd, %lu, %zu} since
193 * they don't affect the amount of space we reserve.
194 */
195 if ((*f == 'l' || *f == 'z') &&
196 (f[1] == 'd' || f[1] == 'u'))
197 ++f;
Guido van Rossum4dfe8a12006-04-22 23:28:04 +0000198
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000199 switch (*f) {
200 case 'c':
201 (void)va_arg(count, int);
202 /* fall through... */
203 case '%':
204 n++;
205 break;
206 case 'd': case 'u': case 'i': case 'x':
207 (void) va_arg(count, int);
208 /* 20 bytes is enough to hold a 64-bit
209 integer. Decimal takes the most space.
210 This isn't enough for octal. */
211 n += 20;
212 break;
213 case 's':
214 s = va_arg(count, char*);
215 n += strlen(s);
216 break;
217 case 'p':
218 (void) va_arg(count, int);
219 /* maximum 64-bit pointer representation:
220 * 0xffffffffffffffff
221 * so 19 characters is enough.
222 * XXX I count 18 -- what's the extra for?
223 */
224 n += 19;
225 break;
226 default:
227 /* if we stumble upon an unknown
228 formatting code, copy the rest of
229 the format string to the output
230 string. (we cannot just skip the
231 code, since there's no way to know
232 what's in the argument list) */
233 n += strlen(p);
234 goto expand;
235 }
236 } else
237 n++;
238 }
239 expand:
240 /* step 2: fill the buffer */
241 /* Since we've analyzed how much space we need for the worst case,
242 use sprintf directly instead of the slower PyOS_snprintf. */
243 string = PyBytes_FromStringAndSize(NULL, n);
244 if (!string)
245 return NULL;
Guido van Rossum4dfe8a12006-04-22 23:28:04 +0000246
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000247 s = PyBytes_AsString(string);
Guido van Rossum4dfe8a12006-04-22 23:28:04 +0000248
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000249 for (f = format; *f; f++) {
250 if (*f == '%') {
251 const char* p = f++;
252 Py_ssize_t i;
253 int longflag = 0;
254 int size_tflag = 0;
255 /* parse the width.precision part (we're only
256 interested in the precision value, if any) */
257 n = 0;
258 while (ISDIGIT(*f))
259 n = (n*10) + *f++ - '0';
260 if (*f == '.') {
261 f++;
262 n = 0;
263 while (ISDIGIT(*f))
264 n = (n*10) + *f++ - '0';
265 }
266 while (*f && *f != '%' && !ISALPHA(*f))
267 f++;
268 /* handle the long flag, but only for %ld and %lu.
269 others can be added when necessary. */
270 if (*f == 'l' && (f[1] == 'd' || f[1] == 'u')) {
271 longflag = 1;
272 ++f;
273 }
274 /* handle the size_t flag. */
275 if (*f == 'z' && (f[1] == 'd' || f[1] == 'u')) {
276 size_tflag = 1;
277 ++f;
278 }
Guido van Rossum4dfe8a12006-04-22 23:28:04 +0000279
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000280 switch (*f) {
281 case 'c':
282 *s++ = va_arg(vargs, int);
283 break;
284 case 'd':
285 if (longflag)
286 sprintf(s, "%ld", va_arg(vargs, long));
287 else if (size_tflag)
288 sprintf(s, "%" PY_FORMAT_SIZE_T "d",
289 va_arg(vargs, Py_ssize_t));
290 else
291 sprintf(s, "%d", va_arg(vargs, int));
292 s += strlen(s);
293 break;
294 case 'u':
295 if (longflag)
296 sprintf(s, "%lu",
297 va_arg(vargs, unsigned long));
298 else if (size_tflag)
299 sprintf(s, "%" PY_FORMAT_SIZE_T "u",
300 va_arg(vargs, size_t));
301 else
302 sprintf(s, "%u",
303 va_arg(vargs, unsigned int));
304 s += strlen(s);
305 break;
306 case 'i':
307 sprintf(s, "%i", va_arg(vargs, int));
308 s += strlen(s);
309 break;
310 case 'x':
311 sprintf(s, "%x", va_arg(vargs, int));
312 s += strlen(s);
313 break;
314 case 's':
315 p = va_arg(vargs, char*);
316 i = strlen(p);
317 if (n > 0 && i > n)
318 i = n;
319 Py_MEMCPY(s, p, i);
320 s += i;
321 break;
322 case 'p':
323 sprintf(s, "%p", va_arg(vargs, void*));
324 /* %p is ill-defined: ensure leading 0x. */
325 if (s[1] == 'X')
326 s[1] = 'x';
327 else if (s[1] != 'x') {
328 memmove(s+2, s, strlen(s)+1);
329 s[0] = '0';
330 s[1] = 'x';
331 }
332 s += strlen(s);
333 break;
334 case '%':
335 *s++ = '%';
336 break;
337 default:
338 strcpy(s, p);
339 s += strlen(s);
340 goto end;
341 }
342 } else
343 *s++ = *f;
344 }
345
346 end:
347 _PyBytes_Resize(&string, s - PyBytes_AS_STRING(string));
348 return string;
349}
350
351PyObject *
352PyBytes_FromFormat(const char *format, ...)
353{
354 PyObject* ret;
355 va_list vargs;
356
357#ifdef HAVE_STDARG_PROTOTYPES
358 va_start(vargs, format);
359#else
360 va_start(vargs);
361#endif
362 ret = PyBytes_FromFormatV(format, vargs);
363 va_end(vargs);
364 return ret;
Guido van Rossum4dfe8a12006-04-22 23:28:04 +0000365}
366
367static void
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000368string_dealloc(PyObject *op)
Guido van Rossum4dfe8a12006-04-22 23:28:04 +0000369{
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000370 Py_TYPE(op)->tp_free(op);
Guido van Rossum4dfe8a12006-04-22 23:28:04 +0000371}
372
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000373/* Unescape a backslash-escaped string. If unicode is non-zero,
374 the string is a u-literal. If recode_encoding is non-zero,
375 the string is UTF-8 encoded and should be re-encoded in the
376 specified encoding. */
377
378PyObject *PyBytes_DecodeEscape(const char *s,
379 Py_ssize_t len,
380 const char *errors,
381 Py_ssize_t unicode,
382 const char *recode_encoding)
383{
384 int c;
385 char *p, *buf;
386 const char *end;
387 PyObject *v;
388 Py_ssize_t newlen = recode_encoding ? 4*len:len;
389 v = PyBytes_FromStringAndSize((char *)NULL, newlen);
390 if (v == NULL)
391 return NULL;
392 p = buf = PyBytes_AsString(v);
393 end = s + len;
394 while (s < end) {
395 if (*s != '\\') {
396 non_esc:
397 if (recode_encoding && (*s & 0x80)) {
398 PyObject *u, *w;
399 char *r;
400 const char* t;
401 Py_ssize_t rn;
402 t = s;
403 /* Decode non-ASCII bytes as UTF-8. */
404 while (t < end && (*t & 0x80)) t++;
405 u = PyUnicode_DecodeUTF8(s, t - s, errors);
406 if(!u) goto failed;
407
408 /* Recode them in target encoding. */
409 w = PyUnicode_AsEncodedString(
410 u, recode_encoding, errors);
411 Py_DECREF(u);
412 if (!w) goto failed;
413
414 /* Append bytes to output buffer. */
415 assert(PyBytes_Check(w));
416 r = PyBytes_AS_STRING(w);
417 rn = PyBytes_GET_SIZE(w);
418 Py_MEMCPY(p, r, rn);
419 p += rn;
420 Py_DECREF(w);
421 s = t;
422 } else {
423 *p++ = *s++;
424 }
425 continue;
426 }
427 s++;
428 if (s==end) {
429 PyErr_SetString(PyExc_ValueError,
430 "Trailing \\ in string");
431 goto failed;
432 }
433 switch (*s++) {
434 /* XXX This assumes ASCII! */
435 case '\n': break;
436 case '\\': *p++ = '\\'; break;
437 case '\'': *p++ = '\''; break;
438 case '\"': *p++ = '\"'; break;
439 case 'b': *p++ = '\b'; break;
440 case 'f': *p++ = '\014'; break; /* FF */
441 case 't': *p++ = '\t'; break;
442 case 'n': *p++ = '\n'; break;
443 case 'r': *p++ = '\r'; break;
444 case 'v': *p++ = '\013'; break; /* VT */
445 case 'a': *p++ = '\007'; break; /* BEL, not classic C */
446 case '0': case '1': case '2': case '3':
447 case '4': case '5': case '6': case '7':
448 c = s[-1] - '0';
449 if (s < end && '0' <= *s && *s <= '7') {
450 c = (c<<3) + *s++ - '0';
451 if (s < end && '0' <= *s && *s <= '7')
452 c = (c<<3) + *s++ - '0';
453 }
454 *p++ = c;
455 break;
456 case 'x':
457 if (s+1 < end && ISXDIGIT(s[0]) && ISXDIGIT(s[1])) {
458 unsigned int x = 0;
459 c = Py_CHARMASK(*s);
460 s++;
461 if (ISDIGIT(c))
462 x = c - '0';
463 else if (ISLOWER(c))
464 x = 10 + c - 'a';
465 else
466 x = 10 + c - 'A';
467 x = x << 4;
468 c = Py_CHARMASK(*s);
469 s++;
470 if (ISDIGIT(c))
471 x += c - '0';
472 else if (ISLOWER(c))
473 x += 10 + c - 'a';
474 else
475 x += 10 + c - 'A';
476 *p++ = x;
477 break;
478 }
479 if (!errors || strcmp(errors, "strict") == 0) {
480 PyErr_SetString(PyExc_ValueError,
481 "invalid \\x escape");
482 goto failed;
483 }
484 if (strcmp(errors, "replace") == 0) {
485 *p++ = '?';
486 } else if (strcmp(errors, "ignore") == 0)
487 /* do nothing */;
488 else {
489 PyErr_Format(PyExc_ValueError,
490 "decoding error; unknown "
491 "error handling code: %.400s",
492 errors);
493 goto failed;
494 }
495 default:
496 *p++ = '\\';
497 s--;
498 goto non_esc; /* an arbitry number of unescaped
499 UTF-8 bytes may follow. */
500 }
501 }
502 if (p-buf < newlen)
503 _PyBytes_Resize(&v, p - buf);
504 return v;
505 failed:
506 Py_DECREF(v);
507 return NULL;
508}
509
510/* -------------------------------------------------------------------- */
511/* object api */
512
513Py_ssize_t
514PyBytes_Size(register PyObject *op)
515{
516 if (!PyBytes_Check(op)) {
517 PyErr_Format(PyExc_TypeError,
518 "expected bytes, %.200s found", Py_TYPE(op)->tp_name);
519 return -1;
520 }
521 return Py_SIZE(op);
522}
523
524char *
525PyBytes_AsString(register PyObject *op)
526{
527 if (!PyBytes_Check(op)) {
528 PyErr_Format(PyExc_TypeError,
529 "expected bytes, %.200s found", Py_TYPE(op)->tp_name);
530 return NULL;
531 }
532 return ((PyBytesObject *)op)->ob_sval;
533}
534
535int
536PyBytes_AsStringAndSize(register PyObject *obj,
537 register char **s,
538 register Py_ssize_t *len)
539{
540 if (s == NULL) {
541 PyErr_BadInternalCall();
542 return -1;
543 }
544
545 if (!PyBytes_Check(obj)) {
546 PyErr_Format(PyExc_TypeError,
547 "expected bytes, %.200s found", Py_TYPE(obj)->tp_name);
548 return -1;
549 }
550
551 *s = PyBytes_AS_STRING(obj);
552 if (len != NULL)
553 *len = PyBytes_GET_SIZE(obj);
554 else if (strlen(*s) != (size_t)PyBytes_GET_SIZE(obj)) {
555 PyErr_SetString(PyExc_TypeError,
556 "expected bytes with no null");
557 return -1;
558 }
559 return 0;
560}
Neal Norwitz6968b052007-02-27 19:02:19 +0000561
562/* -------------------------------------------------------------------- */
563/* Methods */
564
Eric Smith0923d1d2009-04-16 20:16:10 +0000565#include "stringlib/stringdefs.h"
Neal Norwitz6968b052007-02-27 19:02:19 +0000566#define STRINGLIB_CHAR char
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000567
Neal Norwitz6968b052007-02-27 19:02:19 +0000568#define STRINGLIB_CMP memcmp
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000569#define STRINGLIB_LEN PyBytes_GET_SIZE
570#define STRINGLIB_NEW PyBytes_FromStringAndSize
571#define STRINGLIB_STR PyBytes_AS_STRING
572/* #define STRINGLIB_WANT_CONTAINS_OBJ 1 */
573
574#define STRINGLIB_EMPTY nullstring
575#define STRINGLIB_CHECK_EXACT PyBytes_CheckExact
576#define STRINGLIB_MUTABLE 0
Neal Norwitz6968b052007-02-27 19:02:19 +0000577
578#include "stringlib/fastsearch.h"
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000579
Neal Norwitz6968b052007-02-27 19:02:19 +0000580#include "stringlib/count.h"
581#include "stringlib/find.h"
582#include "stringlib/partition.h"
Gregory P. Smith60d241f2007-10-16 06:31:30 +0000583#include "stringlib/ctype.h"
584#include "stringlib/transmogrify.h"
Neal Norwitz6968b052007-02-27 19:02:19 +0000585
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000586#define _Py_InsertThousandsGrouping _PyBytes_InsertThousandsGrouping
Eric Smitha3b1ac82009-04-03 14:45:06 +0000587#define _Py_InsertThousandsGroupingLocale _PyBytes_InsertThousandsGroupingLocale
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000588#include "stringlib/localeutil.h"
Neal Norwitz6968b052007-02-27 19:02:19 +0000589
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000590PyObject *
591PyBytes_Repr(PyObject *obj, int smartquotes)
Neal Norwitz6968b052007-02-27 19:02:19 +0000592{
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000593 static const char *hexdigits = "0123456789abcdef";
594 register PyBytesObject* op = (PyBytesObject*) obj;
595 Py_ssize_t length = Py_SIZE(op);
596 size_t newsize = 3 + 4 * length;
597 PyObject *v;
598 if (newsize > PY_SSIZE_T_MAX || (newsize-3) / 4 != length) {
599 PyErr_SetString(PyExc_OverflowError,
600 "bytes object is too large to make repr");
601 return NULL;
602 }
603 v = PyUnicode_FromUnicode(NULL, newsize);
604 if (v == NULL) {
605 return NULL;
606 }
607 else {
608 register Py_ssize_t i;
609 register Py_UNICODE c;
610 register Py_UNICODE *p = PyUnicode_AS_UNICODE(v);
611 int quote;
612
613 /* Figure out which quote to use; single is preferred */
614 quote = '\'';
615 if (smartquotes) {
616 char *test, *start;
617 start = PyBytes_AS_STRING(op);
618 for (test = start; test < start+length; ++test) {
619 if (*test == '"') {
620 quote = '\''; /* back to single */
621 goto decided;
622 }
623 else if (*test == '\'')
624 quote = '"';
625 }
626 decided:
627 ;
628 }
629
630 *p++ = 'b', *p++ = quote;
631 for (i = 0; i < length; i++) {
632 /* There's at least enough room for a hex escape
633 and a closing quote. */
634 assert(newsize - (p - PyUnicode_AS_UNICODE(v)) >= 5);
635 c = op->ob_sval[i];
636 if (c == quote || c == '\\')
637 *p++ = '\\', *p++ = c;
638 else if (c == '\t')
639 *p++ = '\\', *p++ = 't';
640 else if (c == '\n')
641 *p++ = '\\', *p++ = 'n';
642 else if (c == '\r')
643 *p++ = '\\', *p++ = 'r';
644 else if (c < ' ' || c >= 0x7f) {
645 *p++ = '\\';
646 *p++ = 'x';
647 *p++ = hexdigits[(c & 0xf0) >> 4];
648 *p++ = hexdigits[c & 0xf];
649 }
650 else
651 *p++ = c;
652 }
653 assert(newsize - (p - PyUnicode_AS_UNICODE(v)) >= 1);
654 *p++ = quote;
655 *p = '\0';
656 if (PyUnicode_Resize(&v, (p - PyUnicode_AS_UNICODE(v)))) {
657 Py_DECREF(v);
658 return NULL;
659 }
660 return v;
661 }
Neal Norwitz6968b052007-02-27 19:02:19 +0000662}
663
Neal Norwitz6968b052007-02-27 19:02:19 +0000664static PyObject *
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000665string_repr(PyObject *op)
Neal Norwitz6968b052007-02-27 19:02:19 +0000666{
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000667 return PyBytes_Repr(op, 1);
Neal Norwitz6968b052007-02-27 19:02:19 +0000668}
669
Neal Norwitz6968b052007-02-27 19:02:19 +0000670static PyObject *
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000671string_str(PyObject *op)
Neal Norwitz6968b052007-02-27 19:02:19 +0000672{
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000673 if (Py_BytesWarningFlag) {
674 if (PyErr_WarnEx(PyExc_BytesWarning,
675 "str() on a bytes instance", 1))
676 return NULL;
677 }
678 return string_repr(op);
Neal Norwitz6968b052007-02-27 19:02:19 +0000679}
680
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000681static Py_ssize_t
682string_length(PyBytesObject *a)
683{
684 return Py_SIZE(a);
685}
Neal Norwitz6968b052007-02-27 19:02:19 +0000686
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000687/* This is also used by PyBytes_Concat() */
688static PyObject *
689string_concat(PyObject *a, PyObject *b)
690{
691 Py_ssize_t size;
692 Py_buffer va, vb;
693 PyObject *result = NULL;
694
695 va.len = -1;
696 vb.len = -1;
697 if (_getbuffer(a, &va) < 0 ||
698 _getbuffer(b, &vb) < 0) {
699 PyErr_Format(PyExc_TypeError, "can't concat %.100s to %.100s",
700 Py_TYPE(a)->tp_name, Py_TYPE(b)->tp_name);
701 goto done;
702 }
703
704 /* Optimize end cases */
705 if (va.len == 0 && PyBytes_CheckExact(b)) {
706 result = b;
707 Py_INCREF(result);
708 goto done;
709 }
710 if (vb.len == 0 && PyBytes_CheckExact(a)) {
711 result = a;
712 Py_INCREF(result);
713 goto done;
714 }
715
716 size = va.len + vb.len;
717 if (size < 0) {
718 PyErr_NoMemory();
719 goto done;
720 }
721
722 result = PyBytes_FromStringAndSize(NULL, size);
723 if (result != NULL) {
724 memcpy(PyBytes_AS_STRING(result), va.buf, va.len);
725 memcpy(PyBytes_AS_STRING(result) + va.len, vb.buf, vb.len);
726 }
727
728 done:
729 if (va.len != -1)
Martin v. Löwis423be952008-08-13 15:53:07 +0000730 PyBuffer_Release(&va);
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000731 if (vb.len != -1)
Martin v. Löwis423be952008-08-13 15:53:07 +0000732 PyBuffer_Release(&vb);
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000733 return result;
734}
Neal Norwitz6968b052007-02-27 19:02:19 +0000735
736static PyObject *
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000737string_repeat(register PyBytesObject *a, register Py_ssize_t n)
Neal Norwitz6968b052007-02-27 19:02:19 +0000738{
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000739 register Py_ssize_t i;
740 register Py_ssize_t j;
741 register Py_ssize_t size;
742 register PyBytesObject *op;
743 size_t nbytes;
744 if (n < 0)
745 n = 0;
746 /* watch out for overflows: the size can overflow int,
747 * and the # of bytes needed can overflow size_t
748 */
749 size = Py_SIZE(a) * n;
750 if (n && size / n != Py_SIZE(a)) {
751 PyErr_SetString(PyExc_OverflowError,
Benjamin Peterson4116f362008-05-27 00:36:20 +0000752 "repeated bytes are too long");
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000753 return NULL;
754 }
755 if (size == Py_SIZE(a) && PyBytes_CheckExact(a)) {
756 Py_INCREF(a);
757 return (PyObject *)a;
758 }
759 nbytes = (size_t)size;
Mark Dickinsonfd24b322008-12-06 15:33:31 +0000760 if (nbytes + PyBytesObject_SIZE <= nbytes) {
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000761 PyErr_SetString(PyExc_OverflowError,
Benjamin Peterson4116f362008-05-27 00:36:20 +0000762 "repeated bytes are too long");
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000763 return NULL;
764 }
Mark Dickinsonfd24b322008-12-06 15:33:31 +0000765 op = (PyBytesObject *)PyObject_MALLOC(PyBytesObject_SIZE + nbytes);
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000766 if (op == NULL)
767 return PyErr_NoMemory();
768 PyObject_INIT_VAR(op, &PyBytes_Type, size);
769 op->ob_shash = -1;
770 op->ob_sval[size] = '\0';
771 if (Py_SIZE(a) == 1 && n > 0) {
772 memset(op->ob_sval, a->ob_sval[0] , n);
773 return (PyObject *) op;
774 }
775 i = 0;
776 if (i < size) {
777 Py_MEMCPY(op->ob_sval, a->ob_sval, Py_SIZE(a));
778 i = Py_SIZE(a);
779 }
780 while (i < size) {
781 j = (i <= size-i) ? i : size-i;
782 Py_MEMCPY(op->ob_sval+i, op->ob_sval, j);
783 i += j;
784 }
785 return (PyObject *) op;
Neal Norwitz6968b052007-02-27 19:02:19 +0000786}
787
Guido van Rossum98297ee2007-11-06 21:34:58 +0000788static int
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000789string_contains(PyObject *self, PyObject *arg)
Guido van Rossum98297ee2007-11-06 21:34:58 +0000790{
791 Py_ssize_t ival = PyNumber_AsSsize_t(arg, PyExc_ValueError);
792 if (ival == -1 && PyErr_Occurred()) {
793 Py_buffer varg;
794 int pos;
795 PyErr_Clear();
796 if (_getbuffer(arg, &varg) < 0)
797 return -1;
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000798 pos = stringlib_find(PyBytes_AS_STRING(self), Py_SIZE(self),
Guido van Rossum98297ee2007-11-06 21:34:58 +0000799 varg.buf, varg.len, 0);
Martin v. Löwis423be952008-08-13 15:53:07 +0000800 PyBuffer_Release(&varg);
Guido van Rossum98297ee2007-11-06 21:34:58 +0000801 return pos >= 0;
802 }
803 if (ival < 0 || ival >= 256) {
804 PyErr_SetString(PyExc_ValueError, "byte must be in range(0, 256)");
805 return -1;
806 }
807
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000808 return memchr(PyBytes_AS_STRING(self), ival, Py_SIZE(self)) != NULL;
Guido van Rossum98297ee2007-11-06 21:34:58 +0000809}
810
Neal Norwitz6968b052007-02-27 19:02:19 +0000811static PyObject *
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000812string_item(PyBytesObject *a, register Py_ssize_t i)
Neal Norwitz6968b052007-02-27 19:02:19 +0000813{
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000814 if (i < 0 || i >= Py_SIZE(a)) {
Benjamin Peterson4116f362008-05-27 00:36:20 +0000815 PyErr_SetString(PyExc_IndexError, "index out of range");
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000816 return NULL;
817 }
818 return PyLong_FromLong((unsigned char)a->ob_sval[i]);
Neal Norwitz6968b052007-02-27 19:02:19 +0000819}
820
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000821static PyObject*
822string_richcompare(PyBytesObject *a, PyBytesObject *b, int op)
Neal Norwitz6968b052007-02-27 19:02:19 +0000823{
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000824 int c;
825 Py_ssize_t len_a, len_b;
826 Py_ssize_t min_len;
827 PyObject *result;
Neal Norwitz6968b052007-02-27 19:02:19 +0000828
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000829 /* Make sure both arguments are strings. */
830 if (!(PyBytes_Check(a) && PyBytes_Check(b))) {
Barry Warsaw9e9dcd62008-10-17 01:50:37 +0000831 if (Py_BytesWarningFlag && (op == Py_EQ || op == Py_NE) &&
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000832 (PyObject_IsInstance((PyObject*)a,
833 (PyObject*)&PyUnicode_Type) ||
834 PyObject_IsInstance((PyObject*)b,
835 (PyObject*)&PyUnicode_Type))) {
836 if (PyErr_WarnEx(PyExc_BytesWarning,
Georg Brandle5d68ac2008-06-04 11:30:26 +0000837 "Comparison between bytes and string", 1))
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000838 return NULL;
839 }
840 result = Py_NotImplemented;
841 goto out;
842 }
843 if (a == b) {
844 switch (op) {
845 case Py_EQ:case Py_LE:case Py_GE:
846 result = Py_True;
847 goto out;
848 case Py_NE:case Py_LT:case Py_GT:
849 result = Py_False;
850 goto out;
851 }
852 }
853 if (op == Py_EQ) {
854 /* Supporting Py_NE here as well does not save
855 much time, since Py_NE is rarely used. */
856 if (Py_SIZE(a) == Py_SIZE(b)
857 && (a->ob_sval[0] == b->ob_sval[0]
858 && memcmp(a->ob_sval, b->ob_sval, Py_SIZE(a)) == 0)) {
859 result = Py_True;
860 } else {
861 result = Py_False;
862 }
863 goto out;
864 }
865 len_a = Py_SIZE(a); len_b = Py_SIZE(b);
866 min_len = (len_a < len_b) ? len_a : len_b;
867 if (min_len > 0) {
868 c = Py_CHARMASK(*a->ob_sval) - Py_CHARMASK(*b->ob_sval);
869 if (c==0)
870 c = memcmp(a->ob_sval, b->ob_sval, min_len);
871 } else
872 c = 0;
873 if (c == 0)
874 c = (len_a < len_b) ? -1 : (len_a > len_b) ? 1 : 0;
875 switch (op) {
876 case Py_LT: c = c < 0; break;
877 case Py_LE: c = c <= 0; break;
878 case Py_EQ: assert(0); break; /* unreachable */
879 case Py_NE: c = c != 0; break;
880 case Py_GT: c = c > 0; break;
881 case Py_GE: c = c >= 0; break;
882 default:
883 result = Py_NotImplemented;
884 goto out;
885 }
886 result = c ? Py_True : Py_False;
887 out:
888 Py_INCREF(result);
889 return result;
Neal Norwitz6968b052007-02-27 19:02:19 +0000890}
891
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000892static long
893string_hash(PyBytesObject *a)
Neal Norwitz6968b052007-02-27 19:02:19 +0000894{
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000895 register Py_ssize_t len;
896 register unsigned char *p;
897 register long x;
Neal Norwitz6968b052007-02-27 19:02:19 +0000898
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000899 if (a->ob_shash != -1)
900 return a->ob_shash;
901 len = Py_SIZE(a);
902 p = (unsigned char *) a->ob_sval;
903 x = *p << 7;
904 while (--len >= 0)
905 x = (1000003*x) ^ *p++;
906 x ^= Py_SIZE(a);
907 if (x == -1)
908 x = -2;
909 a->ob_shash = x;
910 return x;
Neal Norwitz6968b052007-02-27 19:02:19 +0000911}
912
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000913static PyObject*
914string_subscript(PyBytesObject* self, PyObject* item)
915{
916 if (PyIndex_Check(item)) {
917 Py_ssize_t i = PyNumber_AsSsize_t(item, PyExc_IndexError);
918 if (i == -1 && PyErr_Occurred())
919 return NULL;
920 if (i < 0)
921 i += PyBytes_GET_SIZE(self);
922 if (i < 0 || i >= PyBytes_GET_SIZE(self)) {
923 PyErr_SetString(PyExc_IndexError,
Benjamin Peterson4116f362008-05-27 00:36:20 +0000924 "index out of range");
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000925 return NULL;
926 }
927 return PyLong_FromLong((unsigned char)self->ob_sval[i]);
928 }
929 else if (PySlice_Check(item)) {
930 Py_ssize_t start, stop, step, slicelength, cur, i;
931 char* source_buf;
932 char* result_buf;
933 PyObject* result;
Neal Norwitz6968b052007-02-27 19:02:19 +0000934
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000935 if (PySlice_GetIndicesEx((PySliceObject*)item,
936 PyBytes_GET_SIZE(self),
937 &start, &stop, &step, &slicelength) < 0) {
938 return NULL;
939 }
Neal Norwitz6968b052007-02-27 19:02:19 +0000940
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000941 if (slicelength <= 0) {
942 return PyBytes_FromStringAndSize("", 0);
943 }
944 else if (start == 0 && step == 1 &&
945 slicelength == PyBytes_GET_SIZE(self) &&
946 PyBytes_CheckExact(self)) {
947 Py_INCREF(self);
948 return (PyObject *)self;
949 }
950 else if (step == 1) {
951 return PyBytes_FromStringAndSize(
952 PyBytes_AS_STRING(self) + start,
953 slicelength);
954 }
955 else {
Alexandre Vassalottie2641f42009-04-03 06:38:02 +0000956 source_buf = PyBytes_AS_STRING(self);
957 result = PyBytes_FromStringAndSize(NULL, slicelength);
958 if (result == NULL)
959 return NULL;
Neal Norwitz6968b052007-02-27 19:02:19 +0000960
Alexandre Vassalottie2641f42009-04-03 06:38:02 +0000961 result_buf = PyBytes_AS_STRING(result);
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000962 for (cur = start, i = 0; i < slicelength;
963 cur += step, i++) {
964 result_buf[i] = source_buf[cur];
965 }
966
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000967 return result;
968 }
969 }
970 else {
971 PyErr_Format(PyExc_TypeError,
Benjamin Peterson4116f362008-05-27 00:36:20 +0000972 "byte indices must be integers, not %.200s",
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000973 Py_TYPE(item)->tp_name);
974 return NULL;
975 }
976}
977
978static int
979string_buffer_getbuffer(PyBytesObject *self, Py_buffer *view, int flags)
980{
Martin v. Löwis423be952008-08-13 15:53:07 +0000981 return PyBuffer_FillInfo(view, (PyObject*)self, (void *)self->ob_sval, Py_SIZE(self),
Antoine Pitrou2f89aa62008-08-02 21:02:48 +0000982 1, flags);
Christian Heimes2c9c7a52008-05-26 13:42:13 +0000983}
984
985static PySequenceMethods string_as_sequence = {
986 (lenfunc)string_length, /*sq_length*/
987 (binaryfunc)string_concat, /*sq_concat*/
988 (ssizeargfunc)string_repeat, /*sq_repeat*/
989 (ssizeargfunc)string_item, /*sq_item*/
990 0, /*sq_slice*/
991 0, /*sq_ass_item*/
992 0, /*sq_ass_slice*/
993 (objobjproc)string_contains /*sq_contains*/
994};
995
996static PyMappingMethods string_as_mapping = {
997 (lenfunc)string_length,
998 (binaryfunc)string_subscript,
999 0,
1000};
1001
1002static PyBufferProcs string_as_buffer = {
1003 (getbufferproc)string_buffer_getbuffer,
1004 NULL,
1005};
1006
1007
1008#define LEFTSTRIP 0
1009#define RIGHTSTRIP 1
1010#define BOTHSTRIP 2
1011
1012/* Arrays indexed by above */
1013static const char *stripformat[] = {"|O:lstrip", "|O:rstrip", "|O:strip"};
1014
1015#define STRIPNAME(i) (stripformat[i]+3)
1016
Neal Norwitz6968b052007-02-27 19:02:19 +00001017
1018/* Don't call if length < 2 */
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001019#define Py_STRING_MATCH(target, offset, pattern, length) \
1020 (target[offset] == pattern[0] && \
1021 target[offset+length-1] == pattern[length-1] && \
Neal Norwitz6968b052007-02-27 19:02:19 +00001022 !memcmp(target+offset+1, pattern+1, length-2) )
1023
1024
Neal Norwitz6968b052007-02-27 19:02:19 +00001025/* Overallocate the initial list to reduce the number of reallocs for small
1026 split sizes. Eg, "A A A A A A A A A A".split() (10 elements) has three
1027 resizes, to sizes 4, 8, then 16. Most observed string splits are for human
1028 text (roughly 11 words per line) and field delimited data (usually 1-10
1029 fields). For large strings the split algorithms are bandwidth limited
1030 so increasing the preallocation likely will not improve things.*/
1031
1032#define MAX_PREALLOC 12
1033
1034/* 5 splits gives 6 elements */
1035#define PREALLOC_SIZE(maxsplit) \
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001036 (maxsplit >= MAX_PREALLOC ? MAX_PREALLOC : maxsplit+1)
Neal Norwitz6968b052007-02-27 19:02:19 +00001037
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001038#define SPLIT_ADD(data, left, right) { \
1039 str = PyBytes_FromStringAndSize((data) + (left), \
1040 (right) - (left)); \
1041 if (str == NULL) \
1042 goto onError; \
1043 if (count < MAX_PREALLOC) { \
1044 PyList_SET_ITEM(list, count, str); \
1045 } else { \
1046 if (PyList_Append(list, str)) { \
1047 Py_DECREF(str); \
1048 goto onError; \
1049 } \
1050 else \
1051 Py_DECREF(str); \
1052 } \
1053 count++; }
Neal Norwitz6968b052007-02-27 19:02:19 +00001054
1055/* Always force the list to the expected size. */
Christian Heimes90aa7642007-12-19 02:45:37 +00001056#define FIX_PREALLOC_SIZE(list) Py_SIZE(list) = count
Neal Norwitz6968b052007-02-27 19:02:19 +00001057
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001058#define SKIP_SPACE(s, i, len) { while (i<len && ISSPACE(s[i])) i++; }
1059#define SKIP_NONSPACE(s, i, len) { while (i<len && !ISSPACE(s[i])) i++; }
1060#define RSKIP_SPACE(s, i) { while (i>=0 && ISSPACE(s[i])) i--; }
1061#define RSKIP_NONSPACE(s, i) { while (i>=0 && !ISSPACE(s[i])) i--; }
Neal Norwitz6968b052007-02-27 19:02:19 +00001062
1063Py_LOCAL_INLINE(PyObject *)
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001064split_whitespace(PyBytesObject *self, Py_ssize_t len, Py_ssize_t maxsplit)
Neal Norwitz6968b052007-02-27 19:02:19 +00001065{
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001066 const char *s = PyBytes_AS_STRING(self);
1067 Py_ssize_t i, j, count=0;
1068 PyObject *str;
1069 PyObject *list = PyList_New(PREALLOC_SIZE(maxsplit));
Neal Norwitz6968b052007-02-27 19:02:19 +00001070
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001071 if (list == NULL)
1072 return NULL;
Neal Norwitz6968b052007-02-27 19:02:19 +00001073
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001074 i = j = 0;
Neal Norwitz6968b052007-02-27 19:02:19 +00001075
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001076 while (maxsplit-- > 0) {
1077 SKIP_SPACE(s, i, len);
1078 if (i==len) break;
1079 j = i; i++;
1080 SKIP_NONSPACE(s, i, len);
1081 if (j == 0 && i == len && PyBytes_CheckExact(self)) {
1082 /* No whitespace in self, so just use it as list[0] */
1083 Py_INCREF(self);
1084 PyList_SET_ITEM(list, 0, (PyObject *)self);
1085 count++;
1086 break;
1087 }
1088 SPLIT_ADD(s, j, i);
1089 }
1090
1091 if (i < len) {
1092 /* Only occurs when maxsplit was reached */
1093 /* Skip any remaining whitespace and copy to end of string */
1094 SKIP_SPACE(s, i, len);
1095 if (i != len)
1096 SPLIT_ADD(s, i, len);
1097 }
1098 FIX_PREALLOC_SIZE(list);
1099 return list;
Neal Norwitz6968b052007-02-27 19:02:19 +00001100 onError:
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001101 Py_DECREF(list);
1102 return NULL;
Neal Norwitz6968b052007-02-27 19:02:19 +00001103}
1104
Guido van Rossum8f950672007-09-10 16:53:45 +00001105Py_LOCAL_INLINE(PyObject *)
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001106split_char(PyBytesObject *self, Py_ssize_t len, char ch, Py_ssize_t maxcount)
Guido van Rossum8f950672007-09-10 16:53:45 +00001107{
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001108 const char *s = PyBytes_AS_STRING(self);
1109 register Py_ssize_t i, j, count=0;
1110 PyObject *str;
1111 PyObject *list = PyList_New(PREALLOC_SIZE(maxcount));
Guido van Rossum8f950672007-09-10 16:53:45 +00001112
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001113 if (list == NULL)
1114 return NULL;
Guido van Rossum8f950672007-09-10 16:53:45 +00001115
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001116 i = j = 0;
1117 while ((j < len) && (maxcount-- > 0)) {
1118 for(; j<len; j++) {
1119 /* I found that using memchr makes no difference */
1120 if (s[j] == ch) {
1121 SPLIT_ADD(s, i, j);
1122 i = j = j + 1;
1123 break;
1124 }
1125 }
1126 }
1127 if (i == 0 && count == 0 && PyBytes_CheckExact(self)) {
1128 /* ch not in self, so just use self as list[0] */
1129 Py_INCREF(self);
1130 PyList_SET_ITEM(list, 0, (PyObject *)self);
1131 count++;
1132 }
1133 else if (i <= len) {
1134 SPLIT_ADD(s, i, len);
1135 }
1136 FIX_PREALLOC_SIZE(list);
1137 return list;
Guido van Rossum8f950672007-09-10 16:53:45 +00001138
1139 onError:
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001140 Py_DECREF(list);
1141 return NULL;
Guido van Rossum8f950672007-09-10 16:53:45 +00001142}
1143
Neal Norwitz6968b052007-02-27 19:02:19 +00001144PyDoc_STRVAR(split__doc__,
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001145"B.split([sep[, maxsplit]]) -> list of bytes\n\
Neal Norwitz6968b052007-02-27 19:02:19 +00001146\n\
Guido van Rossum98297ee2007-11-06 21:34:58 +00001147Return a list of the sections in B, using sep as the delimiter.\n\
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001148If sep is not specified or is None, B is split on ASCII whitespace\n\
1149characters (space, tab, return, newline, formfeed, vertical tab).\n\
Guido van Rossum8f950672007-09-10 16:53:45 +00001150If maxsplit is given, at most maxsplit splits are done.");
Neal Norwitz6968b052007-02-27 19:02:19 +00001151
1152static PyObject *
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001153string_split(PyBytesObject *self, PyObject *args)
Neal Norwitz6968b052007-02-27 19:02:19 +00001154{
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001155 Py_ssize_t len = PyBytes_GET_SIZE(self), n, i, j;
1156 Py_ssize_t maxsplit = -1, count=0;
1157 const char *s = PyBytes_AS_STRING(self), *sub;
1158 Py_buffer vsub;
1159 PyObject *list, *str, *subobj = Py_None;
Neal Norwitz6968b052007-02-27 19:02:19 +00001160#ifdef USE_FAST
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001161 Py_ssize_t pos;
Neal Norwitz6968b052007-02-27 19:02:19 +00001162#endif
1163
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001164 if (!PyArg_ParseTuple(args, "|On:split", &subobj, &maxsplit))
1165 return NULL;
1166 if (maxsplit < 0)
1167 maxsplit = PY_SSIZE_T_MAX;
1168 if (subobj == Py_None)
1169 return split_whitespace(self, len, maxsplit);
1170 if (_getbuffer(subobj, &vsub) < 0)
1171 return NULL;
1172 sub = vsub.buf;
1173 n = vsub.len;
Guido van Rossum8f950672007-09-10 16:53:45 +00001174
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001175 if (n == 0) {
1176 PyErr_SetString(PyExc_ValueError, "empty separator");
Martin v. Löwis423be952008-08-13 15:53:07 +00001177 PyBuffer_Release(&vsub);
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001178 return NULL;
1179 }
Amaury Forgeot d'Arc20443f32008-08-22 22:05:20 +00001180 else if (n == 1) {
1181 list = split_char(self, len, sub[0], maxsplit);
1182 PyBuffer_Release(&vsub);
1183 return list;
1184 }
Guido van Rossum8f950672007-09-10 16:53:45 +00001185
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001186 list = PyList_New(PREALLOC_SIZE(maxsplit));
1187 if (list == NULL) {
Martin v. Löwis423be952008-08-13 15:53:07 +00001188 PyBuffer_Release(&vsub);
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001189 return NULL;
1190 }
Neal Norwitz6968b052007-02-27 19:02:19 +00001191
1192#ifdef USE_FAST
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001193 i = j = 0;
1194 while (maxsplit-- > 0) {
1195 pos = fastsearch(s+i, len-i, sub, n, FAST_SEARCH);
1196 if (pos < 0)
1197 break;
1198 j = i+pos;
1199 SPLIT_ADD(s, i, j);
1200 i = j + n;
1201 }
Neal Norwitz6968b052007-02-27 19:02:19 +00001202#else
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001203 i = j = 0;
1204 while ((j+n <= len) && (maxsplit-- > 0)) {
1205 for (; j+n <= len; j++) {
1206 if (Py_STRING_MATCH(s, j, sub, n)) {
1207 SPLIT_ADD(s, i, j);
1208 i = j = j + n;
1209 break;
1210 }
1211 }
1212 }
Neal Norwitz6968b052007-02-27 19:02:19 +00001213#endif
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001214 SPLIT_ADD(s, i, len);
1215 FIX_PREALLOC_SIZE(list);
Martin v. Löwis423be952008-08-13 15:53:07 +00001216 PyBuffer_Release(&vsub);
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001217 return list;
Neal Norwitz6968b052007-02-27 19:02:19 +00001218
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001219 onError:
1220 Py_DECREF(list);
Martin v. Löwis423be952008-08-13 15:53:07 +00001221 PyBuffer_Release(&vsub);
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001222 return NULL;
Guido van Rossum98297ee2007-11-06 21:34:58 +00001223}
1224
Neal Norwitz6968b052007-02-27 19:02:19 +00001225PyDoc_STRVAR(partition__doc__,
1226"B.partition(sep) -> (head, sep, tail)\n\
1227\n\
Georg Brandl17cb8a82008-05-30 08:20:09 +00001228Search for the separator sep in B, and return the part before it,\n\
Neal Norwitz6968b052007-02-27 19:02:19 +00001229the separator itself, and the part after it. If the separator is not\n\
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001230found, returns B and two empty bytes objects.");
Neal Norwitz6968b052007-02-27 19:02:19 +00001231
1232static PyObject *
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001233string_partition(PyBytesObject *self, PyObject *sep_obj)
Neal Norwitz6968b052007-02-27 19:02:19 +00001234{
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001235 const char *sep;
1236 Py_ssize_t sep_len;
Neal Norwitz6968b052007-02-27 19:02:19 +00001237
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001238 if (PyBytes_Check(sep_obj)) {
1239 sep = PyBytes_AS_STRING(sep_obj);
1240 sep_len = PyBytes_GET_SIZE(sep_obj);
1241 }
1242 else if (PyObject_AsCharBuffer(sep_obj, &sep, &sep_len))
1243 return NULL;
Neal Norwitz6968b052007-02-27 19:02:19 +00001244
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001245 return stringlib_partition(
1246 (PyObject*) self,
1247 PyBytes_AS_STRING(self), PyBytes_GET_SIZE(self),
1248 sep_obj, sep, sep_len
1249 );
Neal Norwitz6968b052007-02-27 19:02:19 +00001250}
1251
1252PyDoc_STRVAR(rpartition__doc__,
1253"B.rpartition(sep) -> (tail, sep, head)\n\
1254\n\
Georg Brandl17cb8a82008-05-30 08:20:09 +00001255Search for the separator sep in B, starting at the end of B,\n\
1256and return the part before it, the separator itself, and the\n\
Guido van Rossum98297ee2007-11-06 21:34:58 +00001257part after it. If the separator is not found, returns two empty\n\
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001258bytes objects and B.");
Neal Norwitz6968b052007-02-27 19:02:19 +00001259
1260static PyObject *
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001261string_rpartition(PyBytesObject *self, PyObject *sep_obj)
Neal Norwitz6968b052007-02-27 19:02:19 +00001262{
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001263 const char *sep;
1264 Py_ssize_t sep_len;
Neal Norwitz6968b052007-02-27 19:02:19 +00001265
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001266 if (PyBytes_Check(sep_obj)) {
1267 sep = PyBytes_AS_STRING(sep_obj);
1268 sep_len = PyBytes_GET_SIZE(sep_obj);
1269 }
1270 else if (PyObject_AsCharBuffer(sep_obj, &sep, &sep_len))
1271 return NULL;
Neal Norwitz6968b052007-02-27 19:02:19 +00001272
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001273 return stringlib_rpartition(
1274 (PyObject*) self,
1275 PyBytes_AS_STRING(self), PyBytes_GET_SIZE(self),
1276 sep_obj, sep, sep_len
1277 );
Neal Norwitz6968b052007-02-27 19:02:19 +00001278}
1279
1280Py_LOCAL_INLINE(PyObject *)
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001281rsplit_whitespace(PyBytesObject *self, Py_ssize_t len, Py_ssize_t maxsplit)
Neal Norwitz6968b052007-02-27 19:02:19 +00001282{
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001283 const char *s = PyBytes_AS_STRING(self);
1284 Py_ssize_t i, j, count=0;
1285 PyObject *str;
1286 PyObject *list = PyList_New(PREALLOC_SIZE(maxsplit));
Neal Norwitz6968b052007-02-27 19:02:19 +00001287
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001288 if (list == NULL)
1289 return NULL;
Neal Norwitz6968b052007-02-27 19:02:19 +00001290
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001291 i = j = len-1;
Neal Norwitz6968b052007-02-27 19:02:19 +00001292
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001293 while (maxsplit-- > 0) {
1294 RSKIP_SPACE(s, i);
1295 if (i<0) break;
1296 j = i; i--;
1297 RSKIP_NONSPACE(s, i);
1298 if (j == len-1 && i < 0 && PyBytes_CheckExact(self)) {
1299 /* No whitespace in self, so just use it as list[0] */
1300 Py_INCREF(self);
1301 PyList_SET_ITEM(list, 0, (PyObject *)self);
1302 count++;
1303 break;
1304 }
1305 SPLIT_ADD(s, i + 1, j + 1);
1306 }
1307 if (i >= 0) {
1308 /* Only occurs when maxsplit was reached. Skip any remaining
1309 whitespace and copy to beginning of string. */
1310 RSKIP_SPACE(s, i);
1311 if (i >= 0)
1312 SPLIT_ADD(s, 0, i + 1);
Neal Norwitz6968b052007-02-27 19:02:19 +00001313
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001314 }
1315 FIX_PREALLOC_SIZE(list);
1316 if (PyList_Reverse(list) < 0)
1317 goto onError;
1318 return list;
Neal Norwitz6968b052007-02-27 19:02:19 +00001319 onError:
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001320 Py_DECREF(list);
1321 return NULL;
Neal Norwitz6968b052007-02-27 19:02:19 +00001322}
1323
Guido van Rossum8f950672007-09-10 16:53:45 +00001324Py_LOCAL_INLINE(PyObject *)
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001325rsplit_char(PyBytesObject *self, Py_ssize_t len, char ch, Py_ssize_t maxcount)
Guido van Rossum8f950672007-09-10 16:53:45 +00001326{
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001327 const char *s = PyBytes_AS_STRING(self);
1328 register Py_ssize_t i, j, count=0;
1329 PyObject *str;
1330 PyObject *list = PyList_New(PREALLOC_SIZE(maxcount));
Guido van Rossum8f950672007-09-10 16:53:45 +00001331
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001332 if (list == NULL)
1333 return NULL;
Guido van Rossum8f950672007-09-10 16:53:45 +00001334
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001335 i = j = len - 1;
1336 while ((i >= 0) && (maxcount-- > 0)) {
1337 for (; i >= 0; i--) {
1338 if (s[i] == ch) {
1339 SPLIT_ADD(s, i + 1, j + 1);
1340 j = i = i - 1;
1341 break;
1342 }
1343 }
1344 }
1345 if (i < 0 && count == 0 && PyBytes_CheckExact(self)) {
1346 /* ch not in self, so just use self as list[0] */
1347 Py_INCREF(self);
1348 PyList_SET_ITEM(list, 0, (PyObject *)self);
1349 count++;
1350 }
1351 else if (j >= -1) {
1352 SPLIT_ADD(s, 0, j + 1);
1353 }
1354 FIX_PREALLOC_SIZE(list);
1355 if (PyList_Reverse(list) < 0)
1356 goto onError;
1357 return list;
Guido van Rossum8f950672007-09-10 16:53:45 +00001358
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001359 onError:
1360 Py_DECREF(list);
1361 return NULL;
Guido van Rossum8f950672007-09-10 16:53:45 +00001362}
1363
Neal Norwitz6968b052007-02-27 19:02:19 +00001364PyDoc_STRVAR(rsplit__doc__,
Benjamin Peterson4116f362008-05-27 00:36:20 +00001365"B.rsplit([sep[, maxsplit]]) -> list of bytes\n\
Neal Norwitz6968b052007-02-27 19:02:19 +00001366\n\
Guido van Rossum98297ee2007-11-06 21:34:58 +00001367Return a list of the sections in B, using sep as the delimiter,\n\
1368starting at the end of B and working to the front.\n\
Guido van Rossum8f950672007-09-10 16:53:45 +00001369If sep is not given, B is split on ASCII whitespace characters\n\
1370(space, tab, return, newline, formfeed, vertical tab).\n\
1371If maxsplit is given, at most maxsplit splits are done.");
Neal Norwitz6968b052007-02-27 19:02:19 +00001372
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001373
Neal Norwitz6968b052007-02-27 19:02:19 +00001374static PyObject *
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001375string_rsplit(PyBytesObject *self, PyObject *args)
Neal Norwitz6968b052007-02-27 19:02:19 +00001376{
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001377 Py_ssize_t len = PyBytes_GET_SIZE(self), n, i, j;
1378 Py_ssize_t maxsplit = -1, count=0;
1379 const char *s, *sub;
1380 Py_buffer vsub;
1381 PyObject *list, *str, *subobj = Py_None;
Neal Norwitz6968b052007-02-27 19:02:19 +00001382
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001383 if (!PyArg_ParseTuple(args, "|On:rsplit", &subobj, &maxsplit))
1384 return NULL;
1385 if (maxsplit < 0)
1386 maxsplit = PY_SSIZE_T_MAX;
1387 if (subobj == Py_None)
1388 return rsplit_whitespace(self, len, maxsplit);
1389 if (_getbuffer(subobj, &vsub) < 0)
1390 return NULL;
1391 sub = vsub.buf;
1392 n = vsub.len;
Guido van Rossum8f950672007-09-10 16:53:45 +00001393
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001394 if (n == 0) {
1395 PyErr_SetString(PyExc_ValueError, "empty separator");
Martin v. Löwis423be952008-08-13 15:53:07 +00001396 PyBuffer_Release(&vsub);
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001397 return NULL;
1398 }
Amaury Forgeot d'Arc20443f32008-08-22 22:05:20 +00001399 else if (n == 1) {
1400 list = rsplit_char(self, len, sub[0], maxsplit);
1401 PyBuffer_Release(&vsub);
1402 return list;
1403 }
Guido van Rossum8f950672007-09-10 16:53:45 +00001404
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001405 list = PyList_New(PREALLOC_SIZE(maxsplit));
1406 if (list == NULL) {
Martin v. Löwis423be952008-08-13 15:53:07 +00001407 PyBuffer_Release(&vsub);
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001408 return NULL;
1409 }
Neal Norwitz6968b052007-02-27 19:02:19 +00001410
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001411 j = len;
1412 i = j - n;
Neal Norwitz6968b052007-02-27 19:02:19 +00001413
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001414 s = PyBytes_AS_STRING(self);
1415 while ( (i >= 0) && (maxsplit-- > 0) ) {
1416 for (; i>=0; i--) {
1417 if (Py_STRING_MATCH(s, i, sub, n)) {
1418 SPLIT_ADD(s, i + n, j);
1419 j = i;
1420 i -= n;
1421 break;
1422 }
1423 }
1424 }
1425 SPLIT_ADD(s, 0, j);
1426 FIX_PREALLOC_SIZE(list);
1427 if (PyList_Reverse(list) < 0)
1428 goto onError;
Martin v. Löwis423be952008-08-13 15:53:07 +00001429 PyBuffer_Release(&vsub);
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001430 return list;
Neal Norwitz6968b052007-02-27 19:02:19 +00001431
1432onError:
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001433 Py_DECREF(list);
Martin v. Löwis423be952008-08-13 15:53:07 +00001434 PyBuffer_Release(&vsub);
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001435 return NULL;
Neal Norwitz6968b052007-02-27 19:02:19 +00001436}
1437
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001438#undef SPLIT_ADD
1439#undef MAX_PREALLOC
1440#undef PREALLOC_SIZE
1441
1442
1443PyDoc_STRVAR(join__doc__,
1444"B.join(iterable_of_bytes) -> bytes\n\
Neal Norwitz6968b052007-02-27 19:02:19 +00001445\n\
Georg Brandl17cb8a82008-05-30 08:20:09 +00001446Concatenate any number of bytes objects, with B in between each pair.\n\
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001447Example: b'.'.join([b'ab', b'pq', b'rs']) -> b'ab.pq.rs'.");
1448
Neal Norwitz6968b052007-02-27 19:02:19 +00001449static PyObject *
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001450string_join(PyObject *self, PyObject *orig)
Neal Norwitz6968b052007-02-27 19:02:19 +00001451{
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001452 char *sep = PyBytes_AS_STRING(self);
1453 const Py_ssize_t seplen = PyBytes_GET_SIZE(self);
1454 PyObject *res = NULL;
1455 char *p;
1456 Py_ssize_t seqlen = 0;
1457 size_t sz = 0;
1458 Py_ssize_t i;
1459 PyObject *seq, *item;
Neal Norwitz6968b052007-02-27 19:02:19 +00001460
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001461 seq = PySequence_Fast(orig, "");
1462 if (seq == NULL) {
1463 return NULL;
1464 }
Neal Norwitz6968b052007-02-27 19:02:19 +00001465
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001466 seqlen = PySequence_Size(seq);
1467 if (seqlen == 0) {
1468 Py_DECREF(seq);
1469 return PyBytes_FromString("");
1470 }
1471 if (seqlen == 1) {
1472 item = PySequence_Fast_GET_ITEM(seq, 0);
1473 if (PyBytes_CheckExact(item)) {
1474 Py_INCREF(item);
1475 Py_DECREF(seq);
1476 return item;
1477 }
1478 }
1479
1480 /* There are at least two things to join, or else we have a subclass
1481 * of the builtin types in the sequence.
1482 * Do a pre-pass to figure out the total amount of space we'll
1483 * need (sz), and see whether all argument are bytes.
1484 */
1485 /* XXX Shouldn't we use _getbuffer() on these items instead? */
1486 for (i = 0; i < seqlen; i++) {
1487 const size_t old_sz = sz;
1488 item = PySequence_Fast_GET_ITEM(seq, i);
1489 if (!PyBytes_Check(item) && !PyByteArray_Check(item)) {
1490 PyErr_Format(PyExc_TypeError,
1491 "sequence item %zd: expected bytes,"
1492 " %.80s found",
1493 i, Py_TYPE(item)->tp_name);
1494 Py_DECREF(seq);
1495 return NULL;
1496 }
1497 sz += Py_SIZE(item);
1498 if (i != 0)
1499 sz += seplen;
1500 if (sz < old_sz || sz > PY_SSIZE_T_MAX) {
1501 PyErr_SetString(PyExc_OverflowError,
Benjamin Peterson4116f362008-05-27 00:36:20 +00001502 "join() result is too long for bytes");
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001503 Py_DECREF(seq);
1504 return NULL;
1505 }
1506 }
1507
1508 /* Allocate result space. */
1509 res = PyBytes_FromStringAndSize((char*)NULL, sz);
1510 if (res == NULL) {
1511 Py_DECREF(seq);
1512 return NULL;
1513 }
1514
1515 /* Catenate everything. */
1516 /* I'm not worried about a PyByteArray item growing because there's
1517 nowhere in this function where we release the GIL. */
1518 p = PyBytes_AS_STRING(res);
1519 for (i = 0; i < seqlen; ++i) {
1520 size_t n;
1521 char *q;
1522 if (i) {
1523 Py_MEMCPY(p, sep, seplen);
1524 p += seplen;
1525 }
1526 item = PySequence_Fast_GET_ITEM(seq, i);
1527 n = Py_SIZE(item);
1528 if (PyBytes_Check(item))
1529 q = PyBytes_AS_STRING(item);
1530 else
1531 q = PyByteArray_AS_STRING(item);
1532 Py_MEMCPY(p, q, n);
1533 p += n;
1534 }
1535
1536 Py_DECREF(seq);
1537 return res;
Neal Norwitz6968b052007-02-27 19:02:19 +00001538}
1539
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001540PyObject *
1541_PyBytes_Join(PyObject *sep, PyObject *x)
1542{
1543 assert(sep != NULL && PyBytes_Check(sep));
1544 assert(x != NULL);
1545 return string_join(sep, x);
1546}
1547
1548Py_LOCAL_INLINE(void)
1549string_adjust_indices(Py_ssize_t *start, Py_ssize_t *end, Py_ssize_t len)
1550{
1551 if (*end > len)
1552 *end = len;
1553 else if (*end < 0)
1554 *end += len;
1555 if (*end < 0)
1556 *end = 0;
1557 if (*start < 0)
1558 *start += len;
1559 if (*start < 0)
1560 *start = 0;
1561}
1562
1563Py_LOCAL_INLINE(Py_ssize_t)
1564string_find_internal(PyBytesObject *self, PyObject *args, int dir)
1565{
1566 PyObject *subobj;
1567 const char *sub;
1568 Py_ssize_t sub_len;
1569 Py_ssize_t start=0, end=PY_SSIZE_T_MAX;
1570 PyObject *obj_start=Py_None, *obj_end=Py_None;
1571
1572 if (!PyArg_ParseTuple(args, "O|OO:find/rfind/index/rindex", &subobj,
1573 &obj_start, &obj_end))
1574 return -2;
1575 /* To support None in "start" and "end" arguments, meaning
1576 the same as if they were not passed.
1577 */
1578 if (obj_start != Py_None)
1579 if (!_PyEval_SliceIndex(obj_start, &start))
1580 return -2;
1581 if (obj_end != Py_None)
1582 if (!_PyEval_SliceIndex(obj_end, &end))
1583 return -2;
1584
1585 if (PyBytes_Check(subobj)) {
1586 sub = PyBytes_AS_STRING(subobj);
1587 sub_len = PyBytes_GET_SIZE(subobj);
1588 }
1589 else if (PyObject_AsCharBuffer(subobj, &sub, &sub_len))
1590 /* XXX - the "expected a character buffer object" is pretty
1591 confusing for a non-expert. remap to something else ? */
1592 return -2;
1593
1594 if (dir > 0)
1595 return stringlib_find_slice(
1596 PyBytes_AS_STRING(self), PyBytes_GET_SIZE(self),
1597 sub, sub_len, start, end);
1598 else
1599 return stringlib_rfind_slice(
1600 PyBytes_AS_STRING(self), PyBytes_GET_SIZE(self),
1601 sub, sub_len, start, end);
1602}
1603
1604
1605PyDoc_STRVAR(find__doc__,
Georg Brandl17cb8a82008-05-30 08:20:09 +00001606"B.find(sub[, start[, end]]) -> int\n\
Neal Norwitz6968b052007-02-27 19:02:19 +00001607\n\
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001608Return the lowest index in S where substring sub is found,\n\
1609such that sub is contained within s[start:end]. Optional\n\
1610arguments start and end are interpreted as in slice notation.\n\
Neal Norwitz6968b052007-02-27 19:02:19 +00001611\n\
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001612Return -1 on failure.");
1613
Neal Norwitz6968b052007-02-27 19:02:19 +00001614static PyObject *
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001615string_find(PyBytesObject *self, PyObject *args)
Neal Norwitz6968b052007-02-27 19:02:19 +00001616{
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001617 Py_ssize_t result = string_find_internal(self, args, +1);
1618 if (result == -2)
1619 return NULL;
1620 return PyLong_FromSsize_t(result);
Neal Norwitz6968b052007-02-27 19:02:19 +00001621}
1622
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001623
1624PyDoc_STRVAR(index__doc__,
Georg Brandl17cb8a82008-05-30 08:20:09 +00001625"B.index(sub[, start[, end]]) -> int\n\
Alexandre Vassalotti09121e82007-12-04 05:51:13 +00001626\n\
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001627Like B.find() but raise ValueError when the substring is not found.");
1628
Alexandre Vassalotti09121e82007-12-04 05:51:13 +00001629static PyObject *
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001630string_index(PyBytesObject *self, PyObject *args)
Alexandre Vassalotti09121e82007-12-04 05:51:13 +00001631{
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001632 Py_ssize_t result = string_find_internal(self, args, +1);
1633 if (result == -2)
1634 return NULL;
1635 if (result == -1) {
1636 PyErr_SetString(PyExc_ValueError,
1637 "substring not found");
1638 return NULL;
1639 }
1640 return PyLong_FromSsize_t(result);
Alexandre Vassalotti09121e82007-12-04 05:51:13 +00001641}
1642
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001643
1644PyDoc_STRVAR(rfind__doc__,
Georg Brandl17cb8a82008-05-30 08:20:09 +00001645"B.rfind(sub[, start[, end]]) -> int\n\
Neal Norwitz6968b052007-02-27 19:02:19 +00001646\n\
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001647Return the highest index in B where substring sub is found,\n\
1648such that sub is contained within s[start:end]. Optional\n\
1649arguments start and end are interpreted as in slice notation.\n\
Neal Norwitz6968b052007-02-27 19:02:19 +00001650\n\
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001651Return -1 on failure.");
1652
Neal Norwitz6968b052007-02-27 19:02:19 +00001653static PyObject *
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001654string_rfind(PyBytesObject *self, PyObject *args)
Neal Norwitz6968b052007-02-27 19:02:19 +00001655{
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001656 Py_ssize_t result = string_find_internal(self, args, -1);
1657 if (result == -2)
1658 return NULL;
1659 return PyLong_FromSsize_t(result);
Neal Norwitz6968b052007-02-27 19:02:19 +00001660}
1661
Guido van Rossumad7d8d12007-04-13 01:39:34 +00001662
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001663PyDoc_STRVAR(rindex__doc__,
Georg Brandl17cb8a82008-05-30 08:20:09 +00001664"B.rindex(sub[, start[, end]]) -> int\n\
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001665\n\
1666Like B.rfind() but raise ValueError when the substring is not found.");
1667
1668static PyObject *
1669string_rindex(PyBytesObject *self, PyObject *args)
Guido van Rossumad7d8d12007-04-13 01:39:34 +00001670{
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001671 Py_ssize_t result = string_find_internal(self, args, -1);
1672 if (result == -2)
1673 return NULL;
1674 if (result == -1) {
1675 PyErr_SetString(PyExc_ValueError,
1676 "substring not found");
1677 return NULL;
1678 }
1679 return PyLong_FromSsize_t(result);
Guido van Rossumad7d8d12007-04-13 01:39:34 +00001680}
1681
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001682
1683Py_LOCAL_INLINE(PyObject *)
1684do_xstrip(PyBytesObject *self, int striptype, PyObject *sepobj)
Guido van Rossumad7d8d12007-04-13 01:39:34 +00001685{
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001686 Py_buffer vsep;
1687 char *s = PyBytes_AS_STRING(self);
1688 Py_ssize_t len = PyBytes_GET_SIZE(self);
1689 char *sep;
1690 Py_ssize_t seplen;
1691 Py_ssize_t i, j;
1692
1693 if (_getbuffer(sepobj, &vsep) < 0)
1694 return NULL;
1695 sep = vsep.buf;
1696 seplen = vsep.len;
1697
1698 i = 0;
1699 if (striptype != RIGHTSTRIP) {
1700 while (i < len && memchr(sep, Py_CHARMASK(s[i]), seplen)) {
1701 i++;
1702 }
1703 }
1704
1705 j = len;
1706 if (striptype != LEFTSTRIP) {
1707 do {
1708 j--;
1709 } while (j >= i && memchr(sep, Py_CHARMASK(s[j]), seplen));
1710 j++;
1711 }
1712
Martin v. Löwis423be952008-08-13 15:53:07 +00001713 PyBuffer_Release(&vsep);
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001714
1715 if (i == 0 && j == len && PyBytes_CheckExact(self)) {
1716 Py_INCREF(self);
1717 return (PyObject*)self;
1718 }
1719 else
1720 return PyBytes_FromStringAndSize(s+i, j-i);
Guido van Rossumad7d8d12007-04-13 01:39:34 +00001721}
1722
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001723
1724Py_LOCAL_INLINE(PyObject *)
1725do_strip(PyBytesObject *self, int striptype)
1726{
1727 char *s = PyBytes_AS_STRING(self);
1728 Py_ssize_t len = PyBytes_GET_SIZE(self), i, j;
1729
1730 i = 0;
1731 if (striptype != RIGHTSTRIP) {
1732 while (i < len && ISSPACE(s[i])) {
1733 i++;
1734 }
1735 }
1736
1737 j = len;
1738 if (striptype != LEFTSTRIP) {
1739 do {
1740 j--;
1741 } while (j >= i && ISSPACE(s[j]));
1742 j++;
1743 }
1744
1745 if (i == 0 && j == len && PyBytes_CheckExact(self)) {
1746 Py_INCREF(self);
1747 return (PyObject*)self;
1748 }
1749 else
1750 return PyBytes_FromStringAndSize(s+i, j-i);
1751}
1752
1753
1754Py_LOCAL_INLINE(PyObject *)
1755do_argstrip(PyBytesObject *self, int striptype, PyObject *args)
1756{
1757 PyObject *sep = NULL;
1758
1759 if (!PyArg_ParseTuple(args, (char *)stripformat[striptype], &sep))
1760 return NULL;
1761
1762 if (sep != NULL && sep != Py_None) {
1763 return do_xstrip(self, striptype, sep);
1764 }
1765 return do_strip(self, striptype);
1766}
1767
1768
Guido van Rossumad7d8d12007-04-13 01:39:34 +00001769PyDoc_STRVAR(strip__doc__,
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001770"B.strip([bytes]) -> bytes\n\
Guido van Rossumad7d8d12007-04-13 01:39:34 +00001771\n\
Guido van Rossum8f950672007-09-10 16:53:45 +00001772Strip leading and trailing bytes contained in the argument.\n\
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001773If the argument is omitted, strip trailing ASCII whitespace.");
Guido van Rossumad7d8d12007-04-13 01:39:34 +00001774static PyObject *
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001775string_strip(PyBytesObject *self, PyObject *args)
Guido van Rossumad7d8d12007-04-13 01:39:34 +00001776{
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001777 if (PyTuple_GET_SIZE(args) == 0)
1778 return do_strip(self, BOTHSTRIP); /* Common case */
1779 else
1780 return do_argstrip(self, BOTHSTRIP, args);
Guido van Rossumad7d8d12007-04-13 01:39:34 +00001781}
1782
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001783
Guido van Rossumad7d8d12007-04-13 01:39:34 +00001784PyDoc_STRVAR(lstrip__doc__,
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001785"B.lstrip([bytes]) -> bytes\n\
Guido van Rossumad7d8d12007-04-13 01:39:34 +00001786\n\
Guido van Rossum8f950672007-09-10 16:53:45 +00001787Strip leading bytes contained in the argument.\n\
1788If the argument is omitted, strip leading ASCII whitespace.");
Guido van Rossumad7d8d12007-04-13 01:39:34 +00001789static PyObject *
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001790string_lstrip(PyBytesObject *self, PyObject *args)
Guido van Rossumad7d8d12007-04-13 01:39:34 +00001791{
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001792 if (PyTuple_GET_SIZE(args) == 0)
1793 return do_strip(self, LEFTSTRIP); /* Common case */
1794 else
1795 return do_argstrip(self, LEFTSTRIP, args);
Guido van Rossumad7d8d12007-04-13 01:39:34 +00001796}
1797
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001798
Guido van Rossumad7d8d12007-04-13 01:39:34 +00001799PyDoc_STRVAR(rstrip__doc__,
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001800"B.rstrip([bytes]) -> bytes\n\
Guido van Rossumad7d8d12007-04-13 01:39:34 +00001801\n\
Guido van Rossum8f950672007-09-10 16:53:45 +00001802Strip trailing bytes contained in the argument.\n\
1803If the argument is omitted, strip trailing ASCII whitespace.");
Guido van Rossumad7d8d12007-04-13 01:39:34 +00001804static PyObject *
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001805string_rstrip(PyBytesObject *self, PyObject *args)
Guido van Rossumad7d8d12007-04-13 01:39:34 +00001806{
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001807 if (PyTuple_GET_SIZE(args) == 0)
1808 return do_strip(self, RIGHTSTRIP); /* Common case */
1809 else
1810 return do_argstrip(self, RIGHTSTRIP, args);
Guido van Rossumad7d8d12007-04-13 01:39:34 +00001811}
Neal Norwitz6968b052007-02-27 19:02:19 +00001812
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001813
1814PyDoc_STRVAR(count__doc__,
Georg Brandl17cb8a82008-05-30 08:20:09 +00001815"B.count(sub[, start[, end]]) -> int\n\
Guido van Rossumd624f182006-04-24 13:47:05 +00001816\n\
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001817Return the number of non-overlapping occurrences of substring sub in\n\
1818string S[start:end]. Optional arguments start and end are interpreted\n\
1819as in slice notation.");
1820
1821static PyObject *
1822string_count(PyBytesObject *self, PyObject *args)
1823{
1824 PyObject *sub_obj;
1825 const char *str = PyBytes_AS_STRING(self), *sub;
1826 Py_ssize_t sub_len;
1827 Py_ssize_t start = 0, end = PY_SSIZE_T_MAX;
1828
1829 if (!PyArg_ParseTuple(args, "O|O&O&:count", &sub_obj,
1830 _PyEval_SliceIndex, &start, _PyEval_SliceIndex, &end))
1831 return NULL;
1832
1833 if (PyBytes_Check(sub_obj)) {
1834 sub = PyBytes_AS_STRING(sub_obj);
1835 sub_len = PyBytes_GET_SIZE(sub_obj);
1836 }
1837 else if (PyObject_AsCharBuffer(sub_obj, &sub, &sub_len))
1838 return NULL;
1839
1840 string_adjust_indices(&start, &end, PyBytes_GET_SIZE(self));
1841
1842 return PyLong_FromSsize_t(
1843 stringlib_count(str + start, end - start, sub, sub_len)
1844 );
1845}
1846
1847
1848PyDoc_STRVAR(translate__doc__,
1849"B.translate(table[, deletechars]) -> bytes\n\
1850\n\
1851Return a copy of B, where all characters occurring in the\n\
1852optional argument deletechars are removed, and the remaining\n\
1853characters have been mapped through the given translation\n\
1854table, which must be a bytes object of length 256.");
1855
1856static PyObject *
1857string_translate(PyBytesObject *self, PyObject *args)
1858{
1859 register char *input, *output;
1860 const char *table;
1861 register Py_ssize_t i, c, changed = 0;
1862 PyObject *input_obj = (PyObject*)self;
1863 const char *output_start, *del_table=NULL;
1864 Py_ssize_t inlen, tablen, dellen = 0;
1865 PyObject *result;
1866 int trans_table[256];
1867 PyObject *tableobj, *delobj = NULL;
1868
1869 if (!PyArg_UnpackTuple(args, "translate", 1, 2,
1870 &tableobj, &delobj))
1871 return NULL;
1872
1873 if (PyBytes_Check(tableobj)) {
1874 table = PyBytes_AS_STRING(tableobj);
1875 tablen = PyBytes_GET_SIZE(tableobj);
1876 }
1877 else if (tableobj == Py_None) {
1878 table = NULL;
1879 tablen = 256;
1880 }
1881 else if (PyObject_AsCharBuffer(tableobj, &table, &tablen))
1882 return NULL;
1883
1884 if (tablen != 256) {
1885 PyErr_SetString(PyExc_ValueError,
1886 "translation table must be 256 characters long");
1887 return NULL;
1888 }
1889
1890 if (delobj != NULL) {
1891 if (PyBytes_Check(delobj)) {
1892 del_table = PyBytes_AS_STRING(delobj);
1893 dellen = PyBytes_GET_SIZE(delobj);
1894 }
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001895 else if (PyObject_AsCharBuffer(delobj, &del_table, &dellen))
1896 return NULL;
1897 }
1898 else {
1899 del_table = NULL;
1900 dellen = 0;
1901 }
1902
1903 inlen = PyBytes_GET_SIZE(input_obj);
1904 result = PyBytes_FromStringAndSize((char *)NULL, inlen);
1905 if (result == NULL)
1906 return NULL;
1907 output_start = output = PyBytes_AsString(result);
1908 input = PyBytes_AS_STRING(input_obj);
1909
1910 if (dellen == 0 && table != NULL) {
1911 /* If no deletions are required, use faster code */
1912 for (i = inlen; --i >= 0; ) {
1913 c = Py_CHARMASK(*input++);
1914 if (Py_CHARMASK((*output++ = table[c])) != c)
1915 changed = 1;
1916 }
1917 if (changed || !PyBytes_CheckExact(input_obj))
1918 return result;
1919 Py_DECREF(result);
1920 Py_INCREF(input_obj);
1921 return input_obj;
1922 }
1923
1924 if (table == NULL) {
1925 for (i = 0; i < 256; i++)
1926 trans_table[i] = Py_CHARMASK(i);
1927 } else {
1928 for (i = 0; i < 256; i++)
1929 trans_table[i] = Py_CHARMASK(table[i]);
1930 }
1931
1932 for (i = 0; i < dellen; i++)
1933 trans_table[(int) Py_CHARMASK(del_table[i])] = -1;
1934
1935 for (i = inlen; --i >= 0; ) {
1936 c = Py_CHARMASK(*input++);
1937 if (trans_table[c] != -1)
1938 if (Py_CHARMASK(*output++ = (char)trans_table[c]) == c)
1939 continue;
1940 changed = 1;
1941 }
1942 if (!changed && PyBytes_CheckExact(input_obj)) {
1943 Py_DECREF(result);
1944 Py_INCREF(input_obj);
1945 return input_obj;
1946 }
1947 /* Fix the size of the resulting string */
1948 if (inlen > 0)
1949 _PyBytes_Resize(&result, output - output_start);
1950 return result;
1951}
1952
1953
Georg Brandlabc38772009-04-12 15:51:51 +00001954static PyObject *
1955string_maketrans(PyObject *null, PyObject *args)
1956{
1957 return _Py_bytes_maketrans(args);
1958}
1959
Christian Heimes2c9c7a52008-05-26 13:42:13 +00001960#define FORWARD 1
1961#define REVERSE -1
1962
1963/* find and count characters and substrings */
1964
1965#define findchar(target, target_len, c) \
1966 ((char *)memchr((const void *)(target), c, target_len))
1967
1968/* String ops must return a string. */
1969/* If the object is subclass of string, create a copy */
1970Py_LOCAL(PyBytesObject *)
1971return_self(PyBytesObject *self)
1972{
1973 if (PyBytes_CheckExact(self)) {
1974 Py_INCREF(self);
1975 return self;
1976 }
1977 return (PyBytesObject *)PyBytes_FromStringAndSize(
1978 PyBytes_AS_STRING(self),
1979 PyBytes_GET_SIZE(self));
1980}
1981
1982Py_LOCAL_INLINE(Py_ssize_t)
1983countchar(const char *target, int target_len, char c, Py_ssize_t maxcount)
1984{
1985 Py_ssize_t count=0;
1986 const char *start=target;
1987 const char *end=target+target_len;
1988
1989 while ( (start=findchar(start, end-start, c)) != NULL ) {
1990 count++;
1991 if (count >= maxcount)
1992 break;
1993 start += 1;
1994 }
1995 return count;
1996}
1997
1998Py_LOCAL(Py_ssize_t)
1999findstring(const char *target, Py_ssize_t target_len,
2000 const char *pattern, Py_ssize_t pattern_len,
2001 Py_ssize_t start,
2002 Py_ssize_t end,
2003 int direction)
2004{
2005 if (start < 0) {
2006 start += target_len;
2007 if (start < 0)
2008 start = 0;
2009 }
2010 if (end > target_len) {
2011 end = target_len;
2012 } else if (end < 0) {
2013 end += target_len;
2014 if (end < 0)
2015 end = 0;
2016 }
2017
2018 /* zero-length substrings always match at the first attempt */
2019 if (pattern_len == 0)
2020 return (direction > 0) ? start : end;
2021
2022 end -= pattern_len;
2023
2024 if (direction < 0) {
2025 for (; end >= start; end--)
2026 if (Py_STRING_MATCH(target, end, pattern, pattern_len))
2027 return end;
2028 } else {
2029 for (; start <= end; start++)
2030 if (Py_STRING_MATCH(target, start,pattern,pattern_len))
2031 return start;
2032 }
2033 return -1;
2034}
2035
2036Py_LOCAL_INLINE(Py_ssize_t)
2037countstring(const char *target, Py_ssize_t target_len,
2038 const char *pattern, Py_ssize_t pattern_len,
2039 Py_ssize_t start,
2040 Py_ssize_t end,
2041 int direction, Py_ssize_t maxcount)
2042{
2043 Py_ssize_t count=0;
2044
2045 if (start < 0) {
2046 start += target_len;
2047 if (start < 0)
2048 start = 0;
2049 }
2050 if (end > target_len) {
2051 end = target_len;
2052 } else if (end < 0) {
2053 end += target_len;
2054 if (end < 0)
2055 end = 0;
2056 }
2057
2058 /* zero-length substrings match everywhere */
2059 if (pattern_len == 0 || maxcount == 0) {
2060 if (target_len+1 < maxcount)
2061 return target_len+1;
2062 return maxcount;
2063 }
2064
2065 end -= pattern_len;
2066 if (direction < 0) {
2067 for (; (end >= start); end--)
2068 if (Py_STRING_MATCH(target, end,pattern,pattern_len)) {
2069 count++;
2070 if (--maxcount <= 0) break;
2071 end -= pattern_len-1;
2072 }
2073 } else {
2074 for (; (start <= end); start++)
2075 if (Py_STRING_MATCH(target, start,
2076 pattern, pattern_len)) {
2077 count++;
2078 if (--maxcount <= 0)
2079 break;
2080 start += pattern_len-1;
2081 }
2082 }
2083 return count;
2084}
2085
2086
2087/* Algorithms for different cases of string replacement */
2088
2089/* len(self)>=1, from="", len(to)>=1, maxcount>=1 */
2090Py_LOCAL(PyBytesObject *)
2091replace_interleave(PyBytesObject *self,
2092 const char *to_s, Py_ssize_t to_len,
2093 Py_ssize_t maxcount)
2094{
2095 char *self_s, *result_s;
2096 Py_ssize_t self_len, result_len;
2097 Py_ssize_t count, i, product;
2098 PyBytesObject *result;
2099
2100 self_len = PyBytes_GET_SIZE(self);
2101
2102 /* 1 at the end plus 1 after every character */
2103 count = self_len+1;
2104 if (maxcount < count)
2105 count = maxcount;
2106
2107 /* Check for overflow */
2108 /* result_len = count * to_len + self_len; */
2109 product = count * to_len;
2110 if (product / to_len != count) {
2111 PyErr_SetString(PyExc_OverflowError,
Benjamin Peterson4116f362008-05-27 00:36:20 +00002112 "replacement bytes are too long");
Christian Heimes2c9c7a52008-05-26 13:42:13 +00002113 return NULL;
2114 }
2115 result_len = product + self_len;
2116 if (result_len < 0) {
2117 PyErr_SetString(PyExc_OverflowError,
Benjamin Peterson4116f362008-05-27 00:36:20 +00002118 "replacement bytes are too long");
Christian Heimes2c9c7a52008-05-26 13:42:13 +00002119 return NULL;
2120 }
2121
2122 if (! (result = (PyBytesObject *)
2123 PyBytes_FromStringAndSize(NULL, result_len)) )
2124 return NULL;
2125
2126 self_s = PyBytes_AS_STRING(self);
2127 result_s = PyBytes_AS_STRING(result);
2128
2129 /* TODO: special case single character, which doesn't need memcpy */
2130
2131 /* Lay the first one down (guaranteed this will occur) */
2132 Py_MEMCPY(result_s, to_s, to_len);
2133 result_s += to_len;
2134 count -= 1;
2135
2136 for (i=0; i<count; i++) {
2137 *result_s++ = *self_s++;
2138 Py_MEMCPY(result_s, to_s, to_len);
2139 result_s += to_len;
2140 }
2141
2142 /* Copy the rest of the original string */
2143 Py_MEMCPY(result_s, self_s, self_len-i);
2144
2145 return result;
2146}
2147
2148/* Special case for deleting a single character */
2149/* len(self)>=1, len(from)==1, to="", maxcount>=1 */
2150Py_LOCAL(PyBytesObject *)
2151replace_delete_single_character(PyBytesObject *self,
2152 char from_c, Py_ssize_t maxcount)
2153{
2154 char *self_s, *result_s;
2155 char *start, *next, *end;
2156 Py_ssize_t self_len, result_len;
2157 Py_ssize_t count;
2158 PyBytesObject *result;
2159
2160 self_len = PyBytes_GET_SIZE(self);
2161 self_s = PyBytes_AS_STRING(self);
2162
2163 count = countchar(self_s, self_len, from_c, maxcount);
2164 if (count == 0) {
2165 return return_self(self);
2166 }
2167
2168 result_len = self_len - count; /* from_len == 1 */
2169 assert(result_len>=0);
2170
2171 if ( (result = (PyBytesObject *)
2172 PyBytes_FromStringAndSize(NULL, result_len)) == NULL)
2173 return NULL;
2174 result_s = PyBytes_AS_STRING(result);
2175
2176 start = self_s;
2177 end = self_s + self_len;
2178 while (count-- > 0) {
2179 next = findchar(start, end-start, from_c);
2180 if (next == NULL)
2181 break;
2182 Py_MEMCPY(result_s, start, next-start);
2183 result_s += (next-start);
2184 start = next+1;
2185 }
2186 Py_MEMCPY(result_s, start, end-start);
2187
2188 return result;
2189}
2190
2191/* len(self)>=1, len(from)>=2, to="", maxcount>=1 */
2192
2193Py_LOCAL(PyBytesObject *)
2194replace_delete_substring(PyBytesObject *self,
2195 const char *from_s, Py_ssize_t from_len,
2196 Py_ssize_t maxcount) {
2197 char *self_s, *result_s;
2198 char *start, *next, *end;
2199 Py_ssize_t self_len, result_len;
2200 Py_ssize_t count, offset;
2201 PyBytesObject *result;
2202
2203 self_len = PyBytes_GET_SIZE(self);
2204 self_s = PyBytes_AS_STRING(self);
2205
2206 count = countstring(self_s, self_len,
2207 from_s, from_len,
2208 0, self_len, 1,
2209 maxcount);
2210
2211 if (count == 0) {
2212 /* no matches */
2213 return return_self(self);
2214 }
2215
2216 result_len = self_len - (count * from_len);
2217 assert (result_len>=0);
2218
2219 if ( (result = (PyBytesObject *)
2220 PyBytes_FromStringAndSize(NULL, result_len)) == NULL )
2221 return NULL;
2222
2223 result_s = PyBytes_AS_STRING(result);
2224
2225 start = self_s;
2226 end = self_s + self_len;
2227 while (count-- > 0) {
2228 offset = findstring(start, end-start,
2229 from_s, from_len,
2230 0, end-start, FORWARD);
2231 if (offset == -1)
2232 break;
2233 next = start + offset;
2234
2235 Py_MEMCPY(result_s, start, next-start);
2236
2237 result_s += (next-start);
2238 start = next+from_len;
2239 }
2240 Py_MEMCPY(result_s, start, end-start);
2241 return result;
2242}
2243
2244/* len(self)>=1, len(from)==len(to)==1, maxcount>=1 */
2245Py_LOCAL(PyBytesObject *)
2246replace_single_character_in_place(PyBytesObject *self,
2247 char from_c, char to_c,
2248 Py_ssize_t maxcount)
2249{
2250 char *self_s, *result_s, *start, *end, *next;
2251 Py_ssize_t self_len;
2252 PyBytesObject *result;
2253
2254 /* The result string will be the same size */
2255 self_s = PyBytes_AS_STRING(self);
2256 self_len = PyBytes_GET_SIZE(self);
2257
2258 next = findchar(self_s, self_len, from_c);
2259
2260 if (next == NULL) {
2261 /* No matches; return the original string */
2262 return return_self(self);
2263 }
2264
2265 /* Need to make a new string */
2266 result = (PyBytesObject *) PyBytes_FromStringAndSize(NULL, self_len);
2267 if (result == NULL)
2268 return NULL;
2269 result_s = PyBytes_AS_STRING(result);
2270 Py_MEMCPY(result_s, self_s, self_len);
2271
2272 /* change everything in-place, starting with this one */
2273 start = result_s + (next-self_s);
2274 *start = to_c;
2275 start++;
2276 end = result_s + self_len;
2277
2278 while (--maxcount > 0) {
2279 next = findchar(start, end-start, from_c);
2280 if (next == NULL)
2281 break;
2282 *next = to_c;
2283 start = next+1;
2284 }
2285
2286 return result;
2287}
2288
2289/* len(self)>=1, len(from)==len(to)>=2, maxcount>=1 */
2290Py_LOCAL(PyBytesObject *)
2291replace_substring_in_place(PyBytesObject *self,
2292 const char *from_s, Py_ssize_t from_len,
2293 const char *to_s, Py_ssize_t to_len,
2294 Py_ssize_t maxcount)
2295{
2296 char *result_s, *start, *end;
2297 char *self_s;
2298 Py_ssize_t self_len, offset;
2299 PyBytesObject *result;
2300
2301 /* The result string will be the same size */
2302
2303 self_s = PyBytes_AS_STRING(self);
2304 self_len = PyBytes_GET_SIZE(self);
2305
2306 offset = findstring(self_s, self_len,
2307 from_s, from_len,
2308 0, self_len, FORWARD);
2309 if (offset == -1) {
2310 /* No matches; return the original string */
2311 return return_self(self);
2312 }
2313
2314 /* Need to make a new string */
2315 result = (PyBytesObject *) PyBytes_FromStringAndSize(NULL, self_len);
2316 if (result == NULL)
2317 return NULL;
2318 result_s = PyBytes_AS_STRING(result);
2319 Py_MEMCPY(result_s, self_s, self_len);
2320
2321 /* change everything in-place, starting with this one */
2322 start = result_s + offset;
2323 Py_MEMCPY(start, to_s, from_len);
2324 start += from_len;
2325 end = result_s + self_len;
2326
2327 while ( --maxcount > 0) {
2328 offset = findstring(start, end-start,
2329 from_s, from_len,
2330 0, end-start, FORWARD);
2331 if (offset==-1)
2332 break;
2333 Py_MEMCPY(start+offset, to_s, from_len);
2334 start += offset+from_len;
2335 }
2336
2337 return result;
2338}
2339
2340/* len(self)>=1, len(from)==1, len(to)>=2, maxcount>=1 */
2341Py_LOCAL(PyBytesObject *)
2342replace_single_character(PyBytesObject *self,
2343 char from_c,
2344 const char *to_s, Py_ssize_t to_len,
2345 Py_ssize_t maxcount)
2346{
2347 char *self_s, *result_s;
2348 char *start, *next, *end;
2349 Py_ssize_t self_len, result_len;
2350 Py_ssize_t count, product;
2351 PyBytesObject *result;
2352
2353 self_s = PyBytes_AS_STRING(self);
2354 self_len = PyBytes_GET_SIZE(self);
2355
2356 count = countchar(self_s, self_len, from_c, maxcount);
2357 if (count == 0) {
2358 /* no matches, return unchanged */
2359 return return_self(self);
2360 }
2361
2362 /* use the difference between current and new, hence the "-1" */
2363 /* result_len = self_len + count * (to_len-1) */
2364 product = count * (to_len-1);
2365 if (product / (to_len-1) != count) {
2366 PyErr_SetString(PyExc_OverflowError,
Benjamin Peterson4116f362008-05-27 00:36:20 +00002367 "replacement bytes are too long");
Christian Heimes2c9c7a52008-05-26 13:42:13 +00002368 return NULL;
2369 }
2370 result_len = self_len + product;
2371 if (result_len < 0) {
2372 PyErr_SetString(PyExc_OverflowError,
Benjamin Peterson4116f362008-05-27 00:36:20 +00002373 "replacment bytes are too long");
Christian Heimes2c9c7a52008-05-26 13:42:13 +00002374 return NULL;
2375 }
2376
2377 if ( (result = (PyBytesObject *)
2378 PyBytes_FromStringAndSize(NULL, result_len)) == NULL)
2379 return NULL;
2380 result_s = PyBytes_AS_STRING(result);
2381
2382 start = self_s;
2383 end = self_s + self_len;
2384 while (count-- > 0) {
2385 next = findchar(start, end-start, from_c);
2386 if (next == NULL)
2387 break;
2388
2389 if (next == start) {
2390 /* replace with the 'to' */
2391 Py_MEMCPY(result_s, to_s, to_len);
2392 result_s += to_len;
2393 start += 1;
2394 } else {
2395 /* copy the unchanged old then the 'to' */
2396 Py_MEMCPY(result_s, start, next-start);
2397 result_s += (next-start);
2398 Py_MEMCPY(result_s, to_s, to_len);
2399 result_s += to_len;
2400 start = next+1;
2401 }
2402 }
2403 /* Copy the remainder of the remaining string */
2404 Py_MEMCPY(result_s, start, end-start);
2405
2406 return result;
2407}
2408
2409/* len(self)>=1, len(from)>=2, len(to)>=2, maxcount>=1 */
2410Py_LOCAL(PyBytesObject *)
2411replace_substring(PyBytesObject *self,
2412 const char *from_s, Py_ssize_t from_len,
2413 const char *to_s, Py_ssize_t to_len,
2414 Py_ssize_t maxcount) {
2415 char *self_s, *result_s;
2416 char *start, *next, *end;
2417 Py_ssize_t self_len, result_len;
2418 Py_ssize_t count, offset, product;
2419 PyBytesObject *result;
2420
2421 self_s = PyBytes_AS_STRING(self);
2422 self_len = PyBytes_GET_SIZE(self);
2423
2424 count = countstring(self_s, self_len,
2425 from_s, from_len,
2426 0, self_len, FORWARD, maxcount);
2427 if (count == 0) {
2428 /* no matches, return unchanged */
2429 return return_self(self);
2430 }
2431
2432 /* Check for overflow */
2433 /* result_len = self_len + count * (to_len-from_len) */
2434 product = count * (to_len-from_len);
2435 if (product / (to_len-from_len) != count) {
2436 PyErr_SetString(PyExc_OverflowError,
Benjamin Peterson4116f362008-05-27 00:36:20 +00002437 "replacement bytes are too long");
Christian Heimes2c9c7a52008-05-26 13:42:13 +00002438 return NULL;
2439 }
2440 result_len = self_len + product;
2441 if (result_len < 0) {
2442 PyErr_SetString(PyExc_OverflowError,
Benjamin Peterson4116f362008-05-27 00:36:20 +00002443 "replacement bytes are too long");
Christian Heimes2c9c7a52008-05-26 13:42:13 +00002444 return NULL;
2445 }
2446
2447 if ( (result = (PyBytesObject *)
2448 PyBytes_FromStringAndSize(NULL, result_len)) == NULL)
2449 return NULL;
2450 result_s = PyBytes_AS_STRING(result);
2451
2452 start = self_s;
2453 end = self_s + self_len;
2454 while (count-- > 0) {
2455 offset = findstring(start, end-start,
2456 from_s, from_len,
2457 0, end-start, FORWARD);
2458 if (offset == -1)
2459 break;
2460 next = start+offset;
2461 if (next == start) {
2462 /* replace with the 'to' */
2463 Py_MEMCPY(result_s, to_s, to_len);
2464 result_s += to_len;
2465 start += from_len;
2466 } else {
2467 /* copy the unchanged old then the 'to' */
2468 Py_MEMCPY(result_s, start, next-start);
2469 result_s += (next-start);
2470 Py_MEMCPY(result_s, to_s, to_len);
2471 result_s += to_len;
2472 start = next+from_len;
2473 }
2474 }
2475 /* Copy the remainder of the remaining string */
2476 Py_MEMCPY(result_s, start, end-start);
2477
2478 return result;
2479}
2480
2481
2482Py_LOCAL(PyBytesObject *)
2483replace(PyBytesObject *self,
2484 const char *from_s, Py_ssize_t from_len,
2485 const char *to_s, Py_ssize_t to_len,
2486 Py_ssize_t maxcount)
2487{
2488 if (maxcount < 0) {
2489 maxcount = PY_SSIZE_T_MAX;
2490 } else if (maxcount == 0 || PyBytes_GET_SIZE(self) == 0) {
2491 /* nothing to do; return the original string */
2492 return return_self(self);
2493 }
2494
2495 if (maxcount == 0 ||
2496 (from_len == 0 && to_len == 0)) {
2497 /* nothing to do; return the original string */
2498 return return_self(self);
2499 }
2500
2501 /* Handle zero-length special cases */
2502
2503 if (from_len == 0) {
2504 /* insert the 'to' string everywhere. */
2505 /* >>> "Python".replace("", ".") */
2506 /* '.P.y.t.h.o.n.' */
2507 return replace_interleave(self, to_s, to_len, maxcount);
2508 }
2509
2510 /* Except for "".replace("", "A") == "A" there is no way beyond this */
2511 /* point for an empty self string to generate a non-empty string */
2512 /* Special case so the remaining code always gets a non-empty string */
2513 if (PyBytes_GET_SIZE(self) == 0) {
2514 return return_self(self);
2515 }
2516
2517 if (to_len == 0) {
Georg Brandl17cb8a82008-05-30 08:20:09 +00002518 /* delete all occurrences of 'from' string */
Christian Heimes2c9c7a52008-05-26 13:42:13 +00002519 if (from_len == 1) {
2520 return replace_delete_single_character(
2521 self, from_s[0], maxcount);
2522 } else {
2523 return replace_delete_substring(self, from_s,
2524 from_len, maxcount);
2525 }
2526 }
2527
2528 /* Handle special case where both strings have the same length */
2529
2530 if (from_len == to_len) {
2531 if (from_len == 1) {
2532 return replace_single_character_in_place(
2533 self,
2534 from_s[0],
2535 to_s[0],
2536 maxcount);
2537 } else {
2538 return replace_substring_in_place(
2539 self, from_s, from_len, to_s, to_len,
2540 maxcount);
2541 }
2542 }
2543
2544 /* Otherwise use the more generic algorithms */
2545 if (from_len == 1) {
2546 return replace_single_character(self, from_s[0],
2547 to_s, to_len, maxcount);
2548 } else {
2549 /* len('from')>=2, len('to')>=1 */
2550 return replace_substring(self, from_s, from_len, to_s, to_len,
2551 maxcount);
2552 }
2553}
2554
2555PyDoc_STRVAR(replace__doc__,
2556"B.replace(old, new[, count]) -> bytes\n\
2557\n\
2558Return a copy of B with all occurrences of subsection\n\
2559old replaced by new. If the optional argument count is\n\
2560given, only the first count occurrences are replaced.");
2561
2562static PyObject *
2563string_replace(PyBytesObject *self, PyObject *args)
2564{
2565 Py_ssize_t count = -1;
2566 PyObject *from, *to;
2567 const char *from_s, *to_s;
2568 Py_ssize_t from_len, to_len;
2569
2570 if (!PyArg_ParseTuple(args, "OO|n:replace", &from, &to, &count))
2571 return NULL;
2572
2573 if (PyBytes_Check(from)) {
2574 from_s = PyBytes_AS_STRING(from);
2575 from_len = PyBytes_GET_SIZE(from);
2576 }
2577 else if (PyObject_AsCharBuffer(from, &from_s, &from_len))
2578 return NULL;
2579
2580 if (PyBytes_Check(to)) {
2581 to_s = PyBytes_AS_STRING(to);
2582 to_len = PyBytes_GET_SIZE(to);
2583 }
2584 else if (PyObject_AsCharBuffer(to, &to_s, &to_len))
2585 return NULL;
2586
2587 return (PyObject *)replace((PyBytesObject *) self,
2588 from_s, from_len,
2589 to_s, to_len, count);
2590}
2591
2592/** End DALKE **/
2593
2594/* Matches the end (direction >= 0) or start (direction < 0) of self
2595 * against substr, using the start and end arguments. Returns
2596 * -1 on error, 0 if not found and 1 if found.
2597 */
2598Py_LOCAL(int)
2599_string_tailmatch(PyBytesObject *self, PyObject *substr, Py_ssize_t start,
2600 Py_ssize_t end, int direction)
2601{
2602 Py_ssize_t len = PyBytes_GET_SIZE(self);
2603 Py_ssize_t slen;
2604 const char* sub;
2605 const char* str;
2606
2607 if (PyBytes_Check(substr)) {
2608 sub = PyBytes_AS_STRING(substr);
2609 slen = PyBytes_GET_SIZE(substr);
2610 }
2611 else if (PyObject_AsCharBuffer(substr, &sub, &slen))
2612 return -1;
2613 str = PyBytes_AS_STRING(self);
2614
2615 string_adjust_indices(&start, &end, len);
2616
2617 if (direction < 0) {
2618 /* startswith */
2619 if (start+slen > len)
2620 return 0;
2621 } else {
2622 /* endswith */
2623 if (end-start < slen || start > len)
2624 return 0;
2625
2626 if (end-slen > start)
2627 start = end - slen;
2628 }
2629 if (end-start >= slen)
2630 return ! memcmp(str+start, sub, slen);
2631 return 0;
2632}
2633
2634
2635PyDoc_STRVAR(startswith__doc__,
Georg Brandl17cb8a82008-05-30 08:20:09 +00002636"B.startswith(prefix[, start[, end]]) -> bool\n\
Christian Heimes2c9c7a52008-05-26 13:42:13 +00002637\n\
2638Return True if B starts with the specified prefix, False otherwise.\n\
2639With optional start, test B beginning at that position.\n\
2640With optional end, stop comparing B at that position.\n\
Benjamin Peterson4116f362008-05-27 00:36:20 +00002641prefix can also be a tuple of bytes to try.");
Christian Heimes2c9c7a52008-05-26 13:42:13 +00002642
2643static PyObject *
2644string_startswith(PyBytesObject *self, PyObject *args)
2645{
2646 Py_ssize_t start = 0;
2647 Py_ssize_t end = PY_SSIZE_T_MAX;
2648 PyObject *subobj;
2649 int result;
2650
2651 if (!PyArg_ParseTuple(args, "O|O&O&:startswith", &subobj,
2652 _PyEval_SliceIndex, &start, _PyEval_SliceIndex, &end))
2653 return NULL;
2654 if (PyTuple_Check(subobj)) {
2655 Py_ssize_t i;
2656 for (i = 0; i < PyTuple_GET_SIZE(subobj); i++) {
2657 result = _string_tailmatch(self,
2658 PyTuple_GET_ITEM(subobj, i),
2659 start, end, -1);
2660 if (result == -1)
2661 return NULL;
2662 else if (result) {
2663 Py_RETURN_TRUE;
2664 }
2665 }
2666 Py_RETURN_FALSE;
2667 }
2668 result = _string_tailmatch(self, subobj, start, end, -1);
2669 if (result == -1)
2670 return NULL;
2671 else
2672 return PyBool_FromLong(result);
2673}
2674
2675
2676PyDoc_STRVAR(endswith__doc__,
Georg Brandl17cb8a82008-05-30 08:20:09 +00002677"B.endswith(suffix[, start[, end]]) -> bool\n\
Christian Heimes2c9c7a52008-05-26 13:42:13 +00002678\n\
2679Return True if B ends with the specified suffix, False otherwise.\n\
2680With optional start, test B beginning at that position.\n\
2681With optional end, stop comparing B at that position.\n\
Benjamin Peterson4116f362008-05-27 00:36:20 +00002682suffix can also be a tuple of bytes to try.");
Christian Heimes2c9c7a52008-05-26 13:42:13 +00002683
2684static PyObject *
2685string_endswith(PyBytesObject *self, PyObject *args)
2686{
2687 Py_ssize_t start = 0;
2688 Py_ssize_t end = PY_SSIZE_T_MAX;
2689 PyObject *subobj;
2690 int result;
2691
2692 if (!PyArg_ParseTuple(args, "O|O&O&:endswith", &subobj,
2693 _PyEval_SliceIndex, &start, _PyEval_SliceIndex, &end))
2694 return NULL;
2695 if (PyTuple_Check(subobj)) {
2696 Py_ssize_t i;
2697 for (i = 0; i < PyTuple_GET_SIZE(subobj); i++) {
2698 result = _string_tailmatch(self,
2699 PyTuple_GET_ITEM(subobj, i),
2700 start, end, +1);
2701 if (result == -1)
2702 return NULL;
2703 else if (result) {
2704 Py_RETURN_TRUE;
2705 }
2706 }
2707 Py_RETURN_FALSE;
2708 }
2709 result = _string_tailmatch(self, subobj, start, end, +1);
2710 if (result == -1)
2711 return NULL;
2712 else
2713 return PyBool_FromLong(result);
2714}
2715
2716
2717PyDoc_STRVAR(decode__doc__,
Georg Brandl17cb8a82008-05-30 08:20:09 +00002718"B.decode([encoding[, errors]]) -> str\n\
Christian Heimes2c9c7a52008-05-26 13:42:13 +00002719\n\
Georg Brandl17cb8a82008-05-30 08:20:09 +00002720Decode S using the codec registered for encoding. encoding defaults\n\
Guido van Rossumd624f182006-04-24 13:47:05 +00002721to the default encoding. errors may be given to set a different error\n\
Guido van Rossum98297ee2007-11-06 21:34:58 +00002722handling scheme. Default is 'strict' meaning that encoding errors raise\n\
2723a UnicodeDecodeError. Other possible values are 'ignore' and 'replace'\n\
Christian Heimes2c9c7a52008-05-26 13:42:13 +00002724as well as any other name registerd with codecs.register_error that is\n\
Guido van Rossumd624f182006-04-24 13:47:05 +00002725able to handle UnicodeDecodeErrors.");
2726
2727static PyObject *
Christian Heimes2c9c7a52008-05-26 13:42:13 +00002728string_decode(PyObject *self, PyObject *args)
Guido van Rossumb6f1fdc2007-04-12 22:49:52 +00002729{
Christian Heimes2c9c7a52008-05-26 13:42:13 +00002730 const char *encoding = NULL;
2731 const char *errors = NULL;
Guido van Rossumd624f182006-04-24 13:47:05 +00002732
Christian Heimes2c9c7a52008-05-26 13:42:13 +00002733 if (!PyArg_ParseTuple(args, "|ss:decode", &encoding, &errors))
2734 return NULL;
2735 if (encoding == NULL)
2736 encoding = PyUnicode_GetDefaultEncoding();
Marc-André Lemburgb2750b52008-06-06 12:18:17 +00002737 return PyUnicode_FromEncodedObject(self, encoding, errors);
Guido van Rossumd624f182006-04-24 13:47:05 +00002738}
2739
Guido van Rossum20188312006-05-05 15:15:40 +00002740
Georg Brandl0b9b9e02007-02-27 08:40:54 +00002741PyDoc_STRVAR(fromhex_doc,
Christian Heimes2c9c7a52008-05-26 13:42:13 +00002742"bytes.fromhex(string) -> bytes\n\
Georg Brandl0b9b9e02007-02-27 08:40:54 +00002743\n\
Christian Heimes2c9c7a52008-05-26 13:42:13 +00002744Create a bytes object from a string of hexadecimal numbers.\n\
Guido van Rossum98297ee2007-11-06 21:34:58 +00002745Spaces between two numbers are accepted.\n\
Christian Heimes2c9c7a52008-05-26 13:42:13 +00002746Example: bytes.fromhex('B9 01EF') -> b'\\xb9\\x01\\xef'.");
Georg Brandl0b9b9e02007-02-27 08:40:54 +00002747
2748static int
Guido van Rossumae404e22007-10-26 21:46:44 +00002749hex_digit_to_int(Py_UNICODE c)
Georg Brandl0b9b9e02007-02-27 08:40:54 +00002750{
Christian Heimes2c9c7a52008-05-26 13:42:13 +00002751 if (c >= 128)
2752 return -1;
2753 if (ISDIGIT(c))
2754 return c - '0';
2755 else {
2756 if (ISUPPER(c))
2757 c = TOLOWER(c);
2758 if (c >= 'a' && c <= 'f')
2759 return c - 'a' + 10;
2760 }
2761 return -1;
Georg Brandl0b9b9e02007-02-27 08:40:54 +00002762}
2763
2764static PyObject *
Christian Heimes2c9c7a52008-05-26 13:42:13 +00002765string_fromhex(PyObject *cls, PyObject *args)
Georg Brandl0b9b9e02007-02-27 08:40:54 +00002766{
Christian Heimes2c9c7a52008-05-26 13:42:13 +00002767 PyObject *newstring, *hexobj;
2768 char *buf;
2769 Py_UNICODE *hex;
2770 Py_ssize_t hexlen, byteslen, i, j;
2771 int top, bot;
Georg Brandl0b9b9e02007-02-27 08:40:54 +00002772
Christian Heimes2c9c7a52008-05-26 13:42:13 +00002773 if (!PyArg_ParseTuple(args, "U:fromhex", &hexobj))
2774 return NULL;
2775 assert(PyUnicode_Check(hexobj));
2776 hexlen = PyUnicode_GET_SIZE(hexobj);
2777 hex = PyUnicode_AS_UNICODE(hexobj);
2778 byteslen = hexlen/2; /* This overestimates if there are spaces */
2779 newstring = PyBytes_FromStringAndSize(NULL, byteslen);
2780 if (!newstring)
2781 return NULL;
2782 buf = PyBytes_AS_STRING(newstring);
2783 for (i = j = 0; i < hexlen; i += 2) {
2784 /* skip over spaces in the input */
2785 while (hex[i] == ' ')
2786 i++;
2787 if (i >= hexlen)
2788 break;
2789 top = hex_digit_to_int(hex[i]);
2790 bot = hex_digit_to_int(hex[i+1]);
2791 if (top == -1 || bot == -1) {
2792 PyErr_Format(PyExc_ValueError,
2793 "non-hexadecimal number found in "
2794 "fromhex() arg at position %zd", i);
2795 goto error;
2796 }
2797 buf[j++] = (top << 4) + bot;
2798 }
2799 if (j != byteslen && _PyBytes_Resize(&newstring, j) < 0)
2800 goto error;
2801 return newstring;
Georg Brandl0b9b9e02007-02-27 08:40:54 +00002802
2803 error:
Christian Heimes2c9c7a52008-05-26 13:42:13 +00002804 Py_XDECREF(newstring);
2805 return NULL;
Georg Brandl0b9b9e02007-02-27 08:40:54 +00002806}
2807
Martin v. Löwis00709aa2008-06-04 14:18:43 +00002808PyDoc_STRVAR(sizeof__doc__,
2809"S.__sizeof__() -> size of S in memory, in bytes");
2810
2811static PyObject *
2812string_sizeof(PyBytesObject *v)
2813{
2814 Py_ssize_t res;
Mark Dickinsonfd24b322008-12-06 15:33:31 +00002815 res = PyBytesObject_SIZE + Py_SIZE(v) * Py_TYPE(v)->tp_itemsize;
Martin v. Löwis00709aa2008-06-04 14:18:43 +00002816 return PyLong_FromSsize_t(res);
2817}
2818
Guido van Rossum0dd32e22007-04-11 05:40:58 +00002819
2820static PyObject *
Christian Heimes2c9c7a52008-05-26 13:42:13 +00002821string_getnewargs(PyBytesObject *v)
Guido van Rossum0dd32e22007-04-11 05:40:58 +00002822{
Christian Heimes2c9c7a52008-05-26 13:42:13 +00002823 return Py_BuildValue("(s#)", v->ob_sval, Py_SIZE(v));
Guido van Rossum0dd32e22007-04-11 05:40:58 +00002824}
2825
Guido van Rossum4dfe8a12006-04-22 23:28:04 +00002826
2827static PyMethodDef
Christian Heimes2c9c7a52008-05-26 13:42:13 +00002828string_methods[] = {
2829 {"__getnewargs__", (PyCFunction)string_getnewargs, METH_NOARGS},
2830 {"capitalize", (PyCFunction)stringlib_capitalize, METH_NOARGS,
2831 _Py_capitalize__doc__},
2832 {"center", (PyCFunction)stringlib_center, METH_VARARGS, center__doc__},
2833 {"count", (PyCFunction)string_count, METH_VARARGS, count__doc__},
2834 {"decode", (PyCFunction)string_decode, METH_VARARGS, decode__doc__},
2835 {"endswith", (PyCFunction)string_endswith, METH_VARARGS,
2836 endswith__doc__},
2837 {"expandtabs", (PyCFunction)stringlib_expandtabs, METH_VARARGS,
2838 expandtabs__doc__},
2839 {"find", (PyCFunction)string_find, METH_VARARGS, find__doc__},
2840 {"fromhex", (PyCFunction)string_fromhex, METH_VARARGS|METH_CLASS,
2841 fromhex_doc},
2842 {"index", (PyCFunction)string_index, METH_VARARGS, index__doc__},
2843 {"isalnum", (PyCFunction)stringlib_isalnum, METH_NOARGS,
2844 _Py_isalnum__doc__},
2845 {"isalpha", (PyCFunction)stringlib_isalpha, METH_NOARGS,
2846 _Py_isalpha__doc__},
2847 {"isdigit", (PyCFunction)stringlib_isdigit, METH_NOARGS,
2848 _Py_isdigit__doc__},
2849 {"islower", (PyCFunction)stringlib_islower, METH_NOARGS,
2850 _Py_islower__doc__},
2851 {"isspace", (PyCFunction)stringlib_isspace, METH_NOARGS,
2852 _Py_isspace__doc__},
2853 {"istitle", (PyCFunction)stringlib_istitle, METH_NOARGS,
2854 _Py_istitle__doc__},
2855 {"isupper", (PyCFunction)stringlib_isupper, METH_NOARGS,
2856 _Py_isupper__doc__},
2857 {"join", (PyCFunction)string_join, METH_O, join__doc__},
2858 {"ljust", (PyCFunction)stringlib_ljust, METH_VARARGS, ljust__doc__},
2859 {"lower", (PyCFunction)stringlib_lower, METH_NOARGS, _Py_lower__doc__},
2860 {"lstrip", (PyCFunction)string_lstrip, METH_VARARGS, lstrip__doc__},
Georg Brandlabc38772009-04-12 15:51:51 +00002861 {"maketrans", (PyCFunction)string_maketrans, METH_VARARGS|METH_STATIC,
2862 _Py_maketrans__doc__},
Christian Heimes2c9c7a52008-05-26 13:42:13 +00002863 {"partition", (PyCFunction)string_partition, METH_O, partition__doc__},
2864 {"replace", (PyCFunction)string_replace, METH_VARARGS, replace__doc__},
2865 {"rfind", (PyCFunction)string_rfind, METH_VARARGS, rfind__doc__},
2866 {"rindex", (PyCFunction)string_rindex, METH_VARARGS, rindex__doc__},
2867 {"rjust", (PyCFunction)stringlib_rjust, METH_VARARGS, rjust__doc__},
2868 {"rpartition", (PyCFunction)string_rpartition, METH_O,
2869 rpartition__doc__},
2870 {"rsplit", (PyCFunction)string_rsplit, METH_VARARGS, rsplit__doc__},
2871 {"rstrip", (PyCFunction)string_rstrip, METH_VARARGS, rstrip__doc__},
2872 {"split", (PyCFunction)string_split, METH_VARARGS, split__doc__},
2873 {"splitlines", (PyCFunction)stringlib_splitlines, METH_VARARGS,
2874 splitlines__doc__},
2875 {"startswith", (PyCFunction)string_startswith, METH_VARARGS,
2876 startswith__doc__},
2877 {"strip", (PyCFunction)string_strip, METH_VARARGS, strip__doc__},
2878 {"swapcase", (PyCFunction)stringlib_swapcase, METH_NOARGS,
2879 _Py_swapcase__doc__},
2880 {"title", (PyCFunction)stringlib_title, METH_NOARGS, _Py_title__doc__},
2881 {"translate", (PyCFunction)string_translate, METH_VARARGS,
2882 translate__doc__},
2883 {"upper", (PyCFunction)stringlib_upper, METH_NOARGS, _Py_upper__doc__},
2884 {"zfill", (PyCFunction)stringlib_zfill, METH_VARARGS, zfill__doc__},
Martin v. Löwis00709aa2008-06-04 14:18:43 +00002885 {"__sizeof__", (PyCFunction)string_sizeof, METH_NOARGS,
2886 sizeof__doc__},
Christian Heimes2c9c7a52008-05-26 13:42:13 +00002887 {NULL, NULL} /* sentinel */
Guido van Rossum4dfe8a12006-04-22 23:28:04 +00002888};
2889
Christian Heimes2c9c7a52008-05-26 13:42:13 +00002890static PyObject *
2891str_subtype_new(PyTypeObject *type, PyObject *args, PyObject *kwds);
2892
2893static PyObject *
2894string_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
2895{
Benjamin Petersonc15a0732008-08-26 16:46:47 +00002896 PyObject *x = NULL;
Christian Heimes2c9c7a52008-05-26 13:42:13 +00002897 const char *encoding = NULL;
2898 const char *errors = NULL;
2899 PyObject *new = NULL;
Christian Heimes2c9c7a52008-05-26 13:42:13 +00002900 static char *kwlist[] = {"source", "encoding", "errors", 0};
2901
2902 if (type != &PyBytes_Type)
2903 return str_subtype_new(type, args, kwds);
2904 if (!PyArg_ParseTupleAndKeywords(args, kwds, "|Oss:bytes", kwlist, &x,
2905 &encoding, &errors))
2906 return NULL;
2907 if (x == NULL) {
2908 if (encoding != NULL || errors != NULL) {
2909 PyErr_SetString(PyExc_TypeError,
2910 "encoding or errors without sequence "
2911 "argument");
2912 return NULL;
2913 }
2914 return PyBytes_FromString("");
2915 }
2916
2917 if (PyUnicode_Check(x)) {
2918 /* Encode via the codec registry */
2919 if (encoding == NULL) {
2920 PyErr_SetString(PyExc_TypeError,
2921 "string argument without an encoding");
2922 return NULL;
2923 }
Marc-André Lemburgb2750b52008-06-06 12:18:17 +00002924 new = PyUnicode_AsEncodedString(x, encoding, errors);
Christian Heimes2c9c7a52008-05-26 13:42:13 +00002925 if (new == NULL)
2926 return NULL;
2927 assert(PyBytes_Check(new));
2928 return new;
2929 }
2930
2931 /* If it's not unicode, there can't be encoding or errors */
2932 if (encoding != NULL || errors != NULL) {
2933 PyErr_SetString(PyExc_TypeError,
2934 "encoding or errors without a string argument");
2935 return NULL;
2936 }
Benjamin Petersonc15a0732008-08-26 16:46:47 +00002937 return PyObject_Bytes(x);
2938}
2939
2940PyObject *
2941PyBytes_FromObject(PyObject *x)
2942{
2943 PyObject *new, *it;
2944 Py_ssize_t i, size;
Christian Heimes2c9c7a52008-05-26 13:42:13 +00002945
Benjamin Peterson4b24a422008-08-27 00:28:34 +00002946 if (x == NULL) {
2947 PyErr_BadInternalCall();
2948 return NULL;
2949 }
2950
Christian Heimes2c9c7a52008-05-26 13:42:13 +00002951 /* Is it an int? */
2952 size = PyNumber_AsSsize_t(x, PyExc_ValueError);
2953 if (size == -1 && PyErr_Occurred()) {
2954 PyErr_Clear();
2955 }
2956 else {
2957 if (size < 0) {
2958 PyErr_SetString(PyExc_ValueError, "negative count");
2959 return NULL;
2960 }
2961 new = PyBytes_FromStringAndSize(NULL, size);
2962 if (new == NULL) {
2963 return NULL;
2964 }
2965 if (size > 0) {
2966 memset(((PyBytesObject*)new)->ob_sval, 0, size);
2967 }
2968 return new;
2969 }
2970
2971 /* Use the modern buffer interface */
2972 if (PyObject_CheckBuffer(x)) {
2973 Py_buffer view;
2974 if (PyObject_GetBuffer(x, &view, PyBUF_FULL_RO) < 0)
2975 return NULL;
2976 new = PyBytes_FromStringAndSize(NULL, view.len);
2977 if (!new)
2978 goto fail;
Christian Heimes1a8501c2008-10-02 19:56:01 +00002979 /* XXX(brett.cannon): Better way to get to internal buffer? */
Christian Heimes2c9c7a52008-05-26 13:42:13 +00002980 if (PyBuffer_ToContiguous(((PyBytesObject *)new)->ob_sval,
2981 &view, view.len, 'C') < 0)
2982 goto fail;
Martin v. Löwis423be952008-08-13 15:53:07 +00002983 PyBuffer_Release(&view);
Christian Heimes2c9c7a52008-05-26 13:42:13 +00002984 return new;
2985 fail:
2986 Py_XDECREF(new);
Martin v. Löwis423be952008-08-13 15:53:07 +00002987 PyBuffer_Release(&view);
Christian Heimes2c9c7a52008-05-26 13:42:13 +00002988 return NULL;
2989 }
2990
2991 /* For iterator version, create a string object and resize as needed */
2992 /* XXX(gb): is 64 a good value? also, optimize if length is known */
2993 /* XXX(guido): perhaps use Pysequence_Fast() -- I can't imagine the
2994 input being a truly long iterator. */
2995 size = 64;
2996 new = PyBytes_FromStringAndSize(NULL, size);
2997 if (new == NULL)
2998 return NULL;
2999
3000 /* XXX Optimize this if the arguments is a list, tuple */
3001
3002 /* Get the iterator */
3003 it = PyObject_GetIter(x);
3004 if (it == NULL)
3005 goto error;
3006
3007 /* Run the iterator to exhaustion */
3008 for (i = 0; ; i++) {
3009 PyObject *item;
3010 Py_ssize_t value;
3011
3012 /* Get the next item */
3013 item = PyIter_Next(it);
3014 if (item == NULL) {
3015 if (PyErr_Occurred())
3016 goto error;
3017 break;
3018 }
3019
3020 /* Interpret it as an int (__index__) */
3021 value = PyNumber_AsSsize_t(item, PyExc_ValueError);
3022 Py_DECREF(item);
3023 if (value == -1 && PyErr_Occurred())
3024 goto error;
3025
3026 /* Range check */
3027 if (value < 0 || value >= 256) {
3028 PyErr_SetString(PyExc_ValueError,
3029 "bytes must be in range(0, 256)");
3030 goto error;
3031 }
3032
3033 /* Append the byte */
3034 if (i >= size) {
3035 size *= 2;
3036 if (_PyBytes_Resize(&new, size) < 0)
3037 goto error;
3038 }
3039 ((PyBytesObject *)new)->ob_sval[i] = value;
3040 }
3041 _PyBytes_Resize(&new, i);
3042
3043 /* Clean up and return success */
3044 Py_DECREF(it);
3045 return new;
3046
3047 error:
3048 /* Error handling when new != NULL */
3049 Py_XDECREF(it);
3050 Py_DECREF(new);
3051 return NULL;
3052}
3053
3054static PyObject *
3055str_subtype_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
3056{
3057 PyObject *tmp, *pnew;
3058 Py_ssize_t n;
3059
3060 assert(PyType_IsSubtype(type, &PyBytes_Type));
3061 tmp = string_new(&PyBytes_Type, args, kwds);
3062 if (tmp == NULL)
3063 return NULL;
3064 assert(PyBytes_CheckExact(tmp));
3065 n = PyBytes_GET_SIZE(tmp);
3066 pnew = type->tp_alloc(type, n);
3067 if (pnew != NULL) {
3068 Py_MEMCPY(PyBytes_AS_STRING(pnew),
3069 PyBytes_AS_STRING(tmp), n+1);
3070 ((PyBytesObject *)pnew)->ob_shash =
3071 ((PyBytesObject *)tmp)->ob_shash;
3072 }
3073 Py_DECREF(tmp);
3074 return pnew;
3075}
3076
3077PyDoc_STRVAR(string_doc,
Georg Brandl17cb8a82008-05-30 08:20:09 +00003078"bytes(iterable_of_ints) -> bytes\n\
Christian Heimes2c9c7a52008-05-26 13:42:13 +00003079bytes(string, encoding[, errors]) -> bytes\n\
Georg Brandl17cb8a82008-05-30 08:20:09 +00003080bytes(bytes_or_buffer) -> immutable copy of bytes_or_buffer\n\
3081bytes(memory_view) -> bytes\n\
Guido van Rossum4dfe8a12006-04-22 23:28:04 +00003082\n\
Christian Heimes2c9c7a52008-05-26 13:42:13 +00003083Construct an immutable array of bytes from:\n\
Guido van Rossum98297ee2007-11-06 21:34:58 +00003084 - an iterable yielding integers in range(256)\n\
3085 - a text string encoded using the specified encoding\n\
Christian Heimes2c9c7a52008-05-26 13:42:13 +00003086 - a bytes or a buffer object\n\
3087 - any object implementing the buffer API.");
Guido van Rossum98297ee2007-11-06 21:34:58 +00003088
Christian Heimes2c9c7a52008-05-26 13:42:13 +00003089static PyObject *str_iter(PyObject *seq);
Guido van Rossum4dfe8a12006-04-22 23:28:04 +00003090
Christian Heimes2c9c7a52008-05-26 13:42:13 +00003091PyTypeObject PyBytes_Type = {
3092 PyVarObject_HEAD_INIT(&PyType_Type, 0)
3093 "bytes",
Mark Dickinsonfd24b322008-12-06 15:33:31 +00003094 PyBytesObject_SIZE,
Christian Heimes2c9c7a52008-05-26 13:42:13 +00003095 sizeof(char),
3096 string_dealloc, /* tp_dealloc */
3097 0, /* tp_print */
3098 0, /* tp_getattr */
3099 0, /* tp_setattr */
Mark Dickinsone94c6792009-02-02 20:36:42 +00003100 0, /* tp_reserved */
Christian Heimes2c9c7a52008-05-26 13:42:13 +00003101 (reprfunc)string_repr, /* tp_repr */
3102 0, /* tp_as_number */
3103 &string_as_sequence, /* tp_as_sequence */
3104 &string_as_mapping, /* tp_as_mapping */
3105 (hashfunc)string_hash, /* tp_hash */
3106 0, /* tp_call */
3107 string_str, /* tp_str */
3108 PyObject_GenericGetAttr, /* tp_getattro */
3109 0, /* tp_setattro */
3110 &string_as_buffer, /* tp_as_buffer */
3111 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE |
3112 Py_TPFLAGS_BYTES_SUBCLASS, /* tp_flags */
3113 string_doc, /* tp_doc */
3114 0, /* tp_traverse */
3115 0, /* tp_clear */
3116 (richcmpfunc)string_richcompare, /* tp_richcompare */
3117 0, /* tp_weaklistoffset */
3118 str_iter, /* tp_iter */
3119 0, /* tp_iternext */
3120 string_methods, /* tp_methods */
3121 0, /* tp_members */
3122 0, /* tp_getset */
3123 &PyBaseObject_Type, /* tp_base */
3124 0, /* tp_dict */
3125 0, /* tp_descr_get */
3126 0, /* tp_descr_set */
3127 0, /* tp_dictoffset */
3128 0, /* tp_init */
3129 0, /* tp_alloc */
3130 string_new, /* tp_new */
3131 PyObject_Del, /* tp_free */
Guido van Rossum4dfe8a12006-04-22 23:28:04 +00003132};
Guido van Rossuma5d2d552007-10-26 17:39:48 +00003133
Christian Heimes2c9c7a52008-05-26 13:42:13 +00003134void
3135PyBytes_Concat(register PyObject **pv, register PyObject *w)
3136{
3137 register PyObject *v;
3138 assert(pv != NULL);
3139 if (*pv == NULL)
3140 return;
3141 if (w == NULL) {
3142 Py_DECREF(*pv);
3143 *pv = NULL;
3144 return;
3145 }
3146 v = string_concat(*pv, w);
3147 Py_DECREF(*pv);
3148 *pv = v;
3149}
3150
3151void
3152PyBytes_ConcatAndDel(register PyObject **pv, register PyObject *w)
3153{
3154 PyBytes_Concat(pv, w);
3155 Py_XDECREF(w);
3156}
3157
3158
3159/* The following function breaks the notion that strings are immutable:
3160 it changes the size of a string. We get away with this only if there
3161 is only one module referencing the object. You can also think of it
3162 as creating a new string object and destroying the old one, only
3163 more efficiently. In any case, don't use this if the string may
3164 already be known to some other part of the code...
3165 Note that if there's not enough memory to resize the string, the original
3166 string object at *pv is deallocated, *pv is set to NULL, an "out of
3167 memory" exception is set, and -1 is returned. Else (on success) 0 is
3168 returned, and the value in *pv may or may not be the same as on input.
3169 As always, an extra byte is allocated for a trailing \0 byte (newsize
3170 does *not* include that), and a trailing \0 byte is stored.
3171*/
3172
3173int
3174_PyBytes_Resize(PyObject **pv, Py_ssize_t newsize)
3175{
3176 register PyObject *v;
3177 register PyBytesObject *sv;
3178 v = *pv;
3179 if (!PyBytes_Check(v) || Py_REFCNT(v) != 1 || newsize < 0) {
3180 *pv = 0;
3181 Py_DECREF(v);
3182 PyErr_BadInternalCall();
3183 return -1;
3184 }
3185 /* XXX UNREF/NEWREF interface should be more symmetrical */
3186 _Py_DEC_REFTOTAL;
3187 _Py_ForgetReference(v);
3188 *pv = (PyObject *)
Mark Dickinsonfd24b322008-12-06 15:33:31 +00003189 PyObject_REALLOC((char *)v, PyBytesObject_SIZE + newsize);
Christian Heimes2c9c7a52008-05-26 13:42:13 +00003190 if (*pv == NULL) {
3191 PyObject_Del(v);
3192 PyErr_NoMemory();
3193 return -1;
3194 }
3195 _Py_NewReference(*pv);
3196 sv = (PyBytesObject *) *pv;
3197 Py_SIZE(sv) = newsize;
3198 sv->ob_sval[newsize] = '\0';
3199 sv->ob_shash = -1; /* invalidate cached hash value */
3200 return 0;
3201}
3202
3203/* _PyBytes_FormatLong emulates the format codes d, u, o, x and X, and
3204 * the F_ALT flag, for Python's long (unbounded) ints. It's not used for
3205 * Python's regular ints.
3206 * Return value: a new PyString*, or NULL if error.
3207 * . *pbuf is set to point into it,
3208 * *plen set to the # of chars following that.
3209 * Caller must decref it when done using pbuf.
3210 * The string starting at *pbuf is of the form
3211 * "-"? ("0x" | "0X")? digit+
3212 * "0x"/"0X" are present only for x and X conversions, with F_ALT
3213 * set in flags. The case of hex digits will be correct,
3214 * There will be at least prec digits, zero-filled on the left if
3215 * necessary to get that many.
3216 * val object to be converted
3217 * flags bitmask of format flags; only F_ALT is looked at
3218 * prec minimum number of digits; 0-fill on left if needed
3219 * type a character in [duoxX]; u acts the same as d
3220 *
3221 * CAUTION: o, x and X conversions on regular ints can never
3222 * produce a '-' sign, but can for Python's unbounded ints.
3223 */
3224PyObject*
3225_PyBytes_FormatLong(PyObject *val, int flags, int prec, int type,
3226 char **pbuf, int *plen)
3227{
3228 PyObject *result = NULL;
3229 char *buf;
3230 Py_ssize_t i;
3231 int sign; /* 1 if '-', else 0 */
3232 int len; /* number of characters */
3233 Py_ssize_t llen;
3234 int numdigits; /* len == numnondigits + numdigits */
3235 int numnondigits = 0;
3236
3237 /* Avoid exceeding SSIZE_T_MAX */
Christian Heimesce694b72008-08-24 16:15:19 +00003238 if (prec > INT_MAX-3) {
Christian Heimes2c9c7a52008-05-26 13:42:13 +00003239 PyErr_SetString(PyExc_OverflowError,
3240 "precision too large");
3241 return NULL;
3242 }
3243
3244 switch (type) {
3245 case 'd':
3246 case 'u':
3247 /* Special-case boolean: we want 0/1 */
3248 if (PyBool_Check(val))
3249 result = PyNumber_ToBase(val, 10);
3250 else
3251 result = Py_TYPE(val)->tp_str(val);
3252 break;
3253 case 'o':
3254 numnondigits = 2;
3255 result = PyNumber_ToBase(val, 8);
3256 break;
3257 case 'x':
3258 case 'X':
3259 numnondigits = 2;
3260 result = PyNumber_ToBase(val, 16);
3261 break;
3262 default:
3263 assert(!"'type' not in [duoxX]");
3264 }
3265 if (!result)
3266 return NULL;
3267
Marc-André Lemburg4cc0f242008-08-07 18:54:33 +00003268 buf = _PyUnicode_AsString(result);
Christian Heimes2c9c7a52008-05-26 13:42:13 +00003269 if (!buf) {
3270 Py_DECREF(result);
3271 return NULL;
3272 }
3273
3274 /* To modify the string in-place, there can only be one reference. */
3275 if (Py_REFCNT(result) != 1) {
3276 PyErr_BadInternalCall();
3277 return NULL;
3278 }
3279 llen = PyUnicode_GetSize(result);
3280 if (llen > INT_MAX) {
3281 PyErr_SetString(PyExc_ValueError,
3282 "string too large in _PyBytes_FormatLong");
3283 return NULL;
3284 }
3285 len = (int)llen;
3286 if (buf[len-1] == 'L') {
3287 --len;
3288 buf[len] = '\0';
3289 }
3290 sign = buf[0] == '-';
3291 numnondigits += sign;
3292 numdigits = len - numnondigits;
3293 assert(numdigits > 0);
3294
3295 /* Get rid of base marker unless F_ALT */
3296 if (((flags & F_ALT) == 0 &&
3297 (type == 'o' || type == 'x' || type == 'X'))) {
3298 assert(buf[sign] == '0');
3299 assert(buf[sign+1] == 'x' || buf[sign+1] == 'X' ||
3300 buf[sign+1] == 'o');
3301 numnondigits -= 2;
3302 buf += 2;
3303 len -= 2;
3304 if (sign)
3305 buf[0] = '-';
3306 assert(len == numnondigits + numdigits);
3307 assert(numdigits > 0);
3308 }
3309
3310 /* Fill with leading zeroes to meet minimum width. */
3311 if (prec > numdigits) {
3312 PyObject *r1 = PyBytes_FromStringAndSize(NULL,
3313 numnondigits + prec);
3314 char *b1;
3315 if (!r1) {
3316 Py_DECREF(result);
3317 return NULL;
3318 }
3319 b1 = PyBytes_AS_STRING(r1);
3320 for (i = 0; i < numnondigits; ++i)
3321 *b1++ = *buf++;
3322 for (i = 0; i < prec - numdigits; i++)
3323 *b1++ = '0';
3324 for (i = 0; i < numdigits; i++)
3325 *b1++ = *buf++;
3326 *b1 = '\0';
3327 Py_DECREF(result);
3328 result = r1;
3329 buf = PyBytes_AS_STRING(result);
3330 len = numnondigits + prec;
3331 }
3332
3333 /* Fix up case for hex conversions. */
3334 if (type == 'X') {
3335 /* Need to convert all lower case letters to upper case.
3336 and need to convert 0x to 0X (and -0x to -0X). */
3337 for (i = 0; i < len; i++)
3338 if (buf[i] >= 'a' && buf[i] <= 'x')
3339 buf[i] -= 'a'-'A';
3340 }
3341 *pbuf = buf;
3342 *plen = len;
3343 return result;
3344}
3345
3346void
3347PyBytes_Fini(void)
3348{
3349 int i;
3350 for (i = 0; i < UCHAR_MAX + 1; i++) {
3351 Py_XDECREF(characters[i]);
3352 characters[i] = NULL;
3353 }
3354 Py_XDECREF(nullstring);
3355 nullstring = NULL;
3356}
3357
Benjamin Peterson4116f362008-05-27 00:36:20 +00003358/*********************** Bytes Iterator ****************************/
Guido van Rossuma5d2d552007-10-26 17:39:48 +00003359
3360typedef struct {
Christian Heimes2c9c7a52008-05-26 13:42:13 +00003361 PyObject_HEAD
3362 Py_ssize_t it_index;
3363 PyBytesObject *it_seq; /* Set to NULL when iterator is exhausted */
3364} striterobject;
Guido van Rossuma5d2d552007-10-26 17:39:48 +00003365
3366static void
Christian Heimes2c9c7a52008-05-26 13:42:13 +00003367striter_dealloc(striterobject *it)
Guido van Rossuma5d2d552007-10-26 17:39:48 +00003368{
Christian Heimes2c9c7a52008-05-26 13:42:13 +00003369 _PyObject_GC_UNTRACK(it);
3370 Py_XDECREF(it->it_seq);
3371 PyObject_GC_Del(it);
Guido van Rossuma5d2d552007-10-26 17:39:48 +00003372}
3373
3374static int
Christian Heimes2c9c7a52008-05-26 13:42:13 +00003375striter_traverse(striterobject *it, visitproc visit, void *arg)
Guido van Rossuma5d2d552007-10-26 17:39:48 +00003376{
Christian Heimes2c9c7a52008-05-26 13:42:13 +00003377 Py_VISIT(it->it_seq);
3378 return 0;
Guido van Rossuma5d2d552007-10-26 17:39:48 +00003379}
3380
3381static PyObject *
Christian Heimes2c9c7a52008-05-26 13:42:13 +00003382striter_next(striterobject *it)
Guido van Rossuma5d2d552007-10-26 17:39:48 +00003383{
Christian Heimes2c9c7a52008-05-26 13:42:13 +00003384 PyBytesObject *seq;
3385 PyObject *item;
Guido van Rossuma5d2d552007-10-26 17:39:48 +00003386
Christian Heimes2c9c7a52008-05-26 13:42:13 +00003387 assert(it != NULL);
3388 seq = it->it_seq;
3389 if (seq == NULL)
3390 return NULL;
3391 assert(PyBytes_Check(seq));
Guido van Rossuma5d2d552007-10-26 17:39:48 +00003392
Christian Heimes2c9c7a52008-05-26 13:42:13 +00003393 if (it->it_index < PyBytes_GET_SIZE(seq)) {
3394 item = PyLong_FromLong(
3395 (unsigned char)seq->ob_sval[it->it_index]);
3396 if (item != NULL)
3397 ++it->it_index;
3398 return item;
3399 }
Guido van Rossuma5d2d552007-10-26 17:39:48 +00003400
Christian Heimes2c9c7a52008-05-26 13:42:13 +00003401 Py_DECREF(seq);
3402 it->it_seq = NULL;
3403 return NULL;
Guido van Rossuma5d2d552007-10-26 17:39:48 +00003404}
3405
3406static PyObject *
Christian Heimes2c9c7a52008-05-26 13:42:13 +00003407striter_len(striterobject *it)
Guido van Rossuma5d2d552007-10-26 17:39:48 +00003408{
Christian Heimes2c9c7a52008-05-26 13:42:13 +00003409 Py_ssize_t len = 0;
3410 if (it->it_seq)
3411 len = PyBytes_GET_SIZE(it->it_seq) - it->it_index;
3412 return PyLong_FromSsize_t(len);
Guido van Rossuma5d2d552007-10-26 17:39:48 +00003413}
3414
3415PyDoc_STRVAR(length_hint_doc,
Christian Heimes2c9c7a52008-05-26 13:42:13 +00003416 "Private method returning an estimate of len(list(it)).");
Guido van Rossuma5d2d552007-10-26 17:39:48 +00003417
Christian Heimes2c9c7a52008-05-26 13:42:13 +00003418static PyMethodDef striter_methods[] = {
3419 {"__length_hint__", (PyCFunction)striter_len, METH_NOARGS,
3420 length_hint_doc},
3421 {NULL, NULL} /* sentinel */
Guido van Rossuma5d2d552007-10-26 17:39:48 +00003422};
3423
Christian Heimes2c9c7a52008-05-26 13:42:13 +00003424PyTypeObject PyBytesIter_Type = {
3425 PyVarObject_HEAD_INIT(&PyType_Type, 0)
3426 "bytes_iterator", /* tp_name */
3427 sizeof(striterobject), /* tp_basicsize */
3428 0, /* tp_itemsize */
3429 /* methods */
3430 (destructor)striter_dealloc, /* tp_dealloc */
3431 0, /* tp_print */
3432 0, /* tp_getattr */
3433 0, /* tp_setattr */
Mark Dickinsone94c6792009-02-02 20:36:42 +00003434 0, /* tp_reserved */
Christian Heimes2c9c7a52008-05-26 13:42:13 +00003435 0, /* tp_repr */
3436 0, /* tp_as_number */
3437 0, /* tp_as_sequence */
3438 0, /* tp_as_mapping */
3439 0, /* tp_hash */
3440 0, /* tp_call */
3441 0, /* tp_str */
3442 PyObject_GenericGetAttr, /* tp_getattro */
3443 0, /* tp_setattro */
3444 0, /* tp_as_buffer */
3445 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC,/* tp_flags */
3446 0, /* tp_doc */
3447 (traverseproc)striter_traverse, /* tp_traverse */
3448 0, /* tp_clear */
3449 0, /* tp_richcompare */
3450 0, /* tp_weaklistoffset */
3451 PyObject_SelfIter, /* tp_iter */
3452 (iternextfunc)striter_next, /* tp_iternext */
3453 striter_methods, /* tp_methods */
3454 0,
Guido van Rossuma5d2d552007-10-26 17:39:48 +00003455};
3456
3457static PyObject *
Christian Heimes2c9c7a52008-05-26 13:42:13 +00003458str_iter(PyObject *seq)
Guido van Rossuma5d2d552007-10-26 17:39:48 +00003459{
Christian Heimes2c9c7a52008-05-26 13:42:13 +00003460 striterobject *it;
Guido van Rossuma5d2d552007-10-26 17:39:48 +00003461
Christian Heimes2c9c7a52008-05-26 13:42:13 +00003462 if (!PyBytes_Check(seq)) {
3463 PyErr_BadInternalCall();
3464 return NULL;
3465 }
3466 it = PyObject_GC_New(striterobject, &PyBytesIter_Type);
3467 if (it == NULL)
3468 return NULL;
3469 it->it_index = 0;
3470 Py_INCREF(seq);
3471 it->it_seq = (PyBytesObject *)seq;
3472 _PyObject_GC_TRACK(it);
3473 return (PyObject *)it;
Guido van Rossuma5d2d552007-10-26 17:39:48 +00003474}