| 
									
										
										
										
											2008-05-27 00:36:20 +00:00
										 |  |  | /* bytes object implementation */ | 
					
						
							| 
									
										
										
										
											2007-10-16 06:31:30 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | #define PY_SSIZE_T_CLEAN
 | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | #include "Python.h"
 | 
					
						
							| 
									
										
										
										
											2007-08-18 11:21:56 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | #include "bytes_methods.h"
 | 
					
						
							| 
									
										
										
										
											2008-12-06 15:33:31 +00:00
										 |  |  | #include <stddef.h>
 | 
					
						
							| 
									
										
										
										
											2007-08-18 11:21:56 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2007-08-27 06:19:22 +00:00
										 |  |  | static Py_ssize_t | 
					
						
							| 
									
										
										
										
											2007-09-23 02:00:13 +00:00
										 |  |  | _getbuffer(PyObject *obj, Py_buffer *view) | 
					
						
							| 
									
										
										
										
											2007-04-13 01:39:34 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2007-12-19 02:45:37 +00:00
										 |  |  |     PyBufferProcs *buffer = Py_TYPE(obj)->tp_as_buffer; | 
					
						
							| 
									
										
										
										
											2007-04-13 01:39:34 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2007-10-16 06:31:30 +00:00
										 |  |  |     if (buffer == NULL || buffer->bf_getbuffer == NULL) | 
					
						
							| 
									
										
										
										
											2007-08-29 04:05:57 +00:00
										 |  |  |     { | 
					
						
							| 
									
										
										
										
											2010-06-09 16:38:55 +00:00
										 |  |  |         PyErr_Format(PyExc_TypeError, | 
					
						
							|  |  |  |                      "Type %.100s doesn't support the buffer API", | 
					
						
							|  |  |  |                      Py_TYPE(obj)->tp_name); | 
					
						
							|  |  |  |         return -1; | 
					
						
							| 
									
										
										
										
											2007-08-29 04:05:57 +00:00
										 |  |  |     } | 
					
						
							| 
									
										
										
										
											2007-04-13 01:39:34 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2007-08-18 11:21:56 +00:00
										 |  |  |     if (buffer->bf_getbuffer(obj, view, PyBUF_SIMPLE) < 0) | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |         return -1; | 
					
						
							| 
									
										
										
										
											2007-08-18 11:21:56 +00:00
										 |  |  |     return view->len; | 
					
						
							| 
									
										
										
										
											2007-04-13 01:39:34 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | #ifdef COUNT_ALLOCS
 | 
					
						
							| 
									
										
										
										
											2009-01-11 17:13:55 +00:00
										 |  |  | Py_ssize_t null_strings, one_strings; | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | #endif
 | 
					
						
							| 
									
										
										
										
											2006-04-22 23:28:04 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | static PyBytesObject *characters[UCHAR_MAX + 1]; | 
					
						
							|  |  |  | static PyBytesObject *nullstring; | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-12-06 15:33:31 +00:00
										 |  |  | /* PyBytesObject_SIZE gives the basic size of a string; any memory allocation
 | 
					
						
							|  |  |  |    for a string of length n should request PyBytesObject_SIZE + n bytes. | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |    Using PyBytesObject_SIZE instead of sizeof(PyBytesObject) saves | 
					
						
							|  |  |  |    3 bytes per string allocation on a typical system. | 
					
						
							|  |  |  | */ | 
					
						
							|  |  |  | #define PyBytesObject_SIZE (offsetof(PyBytesObject, ob_sval) + 1)
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | /*
 | 
					
						
							|  |  |  |    For PyBytes_FromString(), the parameter `str' points to a null-terminated | 
					
						
							|  |  |  |    string containing exactly `size' bytes. | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |    For PyBytes_FromStringAndSize(), the parameter the parameter `str' is | 
					
						
							|  |  |  |    either NULL or else points to a string containing at least `size' bytes. | 
					
						
							|  |  |  |    For PyBytes_FromStringAndSize(), the string in the `str' parameter does | 
					
						
							|  |  |  |    not have to be null-terminated.  (Therefore it is safe to construct a | 
					
						
							|  |  |  |    substring by calling `PyBytes_FromStringAndSize(origstring, substrlen)'.) | 
					
						
							|  |  |  |    If `str' is NULL then PyBytes_FromStringAndSize() will allocate `size+1' | 
					
						
							|  |  |  |    bytes (setting the last byte to the null terminating character) and you can | 
					
						
							|  |  |  |    fill in the data yourself.  If `str' is non-NULL then the resulting | 
					
						
							| 
									
										
										
										
											2010-01-13 08:07:53 +00:00
										 |  |  |    PyBytes object must be treated as immutable and you must not fill in nor | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  |    alter the data yourself, since the strings may be shared. | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |    The PyObject member `op->ob_size', which denotes the number of "extra | 
					
						
							|  |  |  |    items" in a variable-size object, will contain the number of bytes | 
					
						
							| 
									
										
										
										
											2011-03-24 22:32:56 +02:00
										 |  |  |    allocated for string data, not counting the null terminating character. | 
					
						
							|  |  |  |    It is therefore equal to the `size' parameter (for | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  |    PyBytes_FromStringAndSize()) or the length of the string in the `str' | 
					
						
							|  |  |  |    parameter (for PyBytes_FromString()). | 
					
						
							|  |  |  | */ | 
					
						
							| 
									
										
										
										
											2006-04-22 23:28:04 +00:00
										 |  |  | PyObject * | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | PyBytes_FromStringAndSize(const char *str, Py_ssize_t size) | 
					
						
							| 
									
										
										
										
											2006-04-24 13:47:05 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     register PyBytesObject *op; | 
					
						
							|  |  |  |     if (size < 0) { | 
					
						
							|  |  |  |         PyErr_SetString(PyExc_SystemError, | 
					
						
							|  |  |  |             "Negative size passed to PyBytes_FromStringAndSize"); | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     if (size == 0 && (op = nullstring) != NULL) { | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | #ifdef COUNT_ALLOCS
 | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |         null_strings++; | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | #endif
 | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |         Py_INCREF(op); | 
					
						
							|  |  |  |         return (PyObject *)op; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     if (size == 1 && str != NULL && | 
					
						
							|  |  |  |         (op = characters[*str & UCHAR_MAX]) != NULL) | 
					
						
							|  |  |  |     { | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | #ifdef COUNT_ALLOCS
 | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |         one_strings++; | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | #endif
 | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |         Py_INCREF(op); | 
					
						
							|  |  |  |         return (PyObject *)op; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     if (size > PY_SSIZE_T_MAX - PyBytesObject_SIZE) { | 
					
						
							|  |  |  |         PyErr_SetString(PyExc_OverflowError, | 
					
						
							|  |  |  |                         "byte string is too large"); | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     /* Inline PyObject_NewVar */ | 
					
						
							|  |  |  |     op = (PyBytesObject *)PyObject_MALLOC(PyBytesObject_SIZE + size); | 
					
						
							|  |  |  |     if (op == NULL) | 
					
						
							|  |  |  |         return PyErr_NoMemory(); | 
					
						
							|  |  |  |     PyObject_INIT_VAR(op, &PyBytes_Type, size); | 
					
						
							|  |  |  |     op->ob_shash = -1; | 
					
						
							|  |  |  |     if (str != NULL) | 
					
						
							|  |  |  |         Py_MEMCPY(op->ob_sval, str, size); | 
					
						
							|  |  |  |     op->ob_sval[size] = '\0'; | 
					
						
							|  |  |  |     /* share short strings */ | 
					
						
							|  |  |  |     if (size == 0) { | 
					
						
							|  |  |  |         nullstring = op; | 
					
						
							|  |  |  |         Py_INCREF(op); | 
					
						
							|  |  |  |     } else if (size == 1 && str != NULL) { | 
					
						
							|  |  |  |         characters[*str & UCHAR_MAX] = op; | 
					
						
							|  |  |  |         Py_INCREF(op); | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     return (PyObject *) op; | 
					
						
							| 
									
										
										
										
											2006-04-24 13:47:05 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | PyObject * | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | PyBytes_FromString(const char *str) | 
					
						
							| 
									
										
										
										
											2006-04-22 23:28:04 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     register size_t size; | 
					
						
							|  |  |  |     register PyBytesObject *op; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     assert(str != NULL); | 
					
						
							|  |  |  |     size = strlen(str); | 
					
						
							|  |  |  |     if (size > PY_SSIZE_T_MAX - PyBytesObject_SIZE) { | 
					
						
							|  |  |  |         PyErr_SetString(PyExc_OverflowError, | 
					
						
							|  |  |  |             "byte string is too long"); | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     if (size == 0 && (op = nullstring) != NULL) { | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | #ifdef COUNT_ALLOCS
 | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |         null_strings++; | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | #endif
 | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |         Py_INCREF(op); | 
					
						
							|  |  |  |         return (PyObject *)op; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     if (size == 1 && (op = characters[*str & UCHAR_MAX]) != NULL) { | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | #ifdef COUNT_ALLOCS
 | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |         one_strings++; | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | #endif
 | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |         Py_INCREF(op); | 
					
						
							|  |  |  |         return (PyObject *)op; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     /* Inline PyObject_NewVar */ | 
					
						
							|  |  |  |     op = (PyBytesObject *)PyObject_MALLOC(PyBytesObject_SIZE + size); | 
					
						
							|  |  |  |     if (op == NULL) | 
					
						
							|  |  |  |         return PyErr_NoMemory(); | 
					
						
							|  |  |  |     PyObject_INIT_VAR(op, &PyBytes_Type, size); | 
					
						
							|  |  |  |     op->ob_shash = -1; | 
					
						
							|  |  |  |     Py_MEMCPY(op->ob_sval, str, size+1); | 
					
						
							|  |  |  |     /* share short strings */ | 
					
						
							|  |  |  |     if (size == 0) { | 
					
						
							|  |  |  |         nullstring = op; | 
					
						
							|  |  |  |         Py_INCREF(op); | 
					
						
							|  |  |  |     } else if (size == 1) { | 
					
						
							|  |  |  |         characters[*str & UCHAR_MAX] = op; | 
					
						
							|  |  |  |         Py_INCREF(op); | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     return (PyObject *) op; | 
					
						
							| 
									
										
										
										
											2006-04-22 23:28:04 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | PyObject * | 
					
						
							|  |  |  | PyBytes_FromFormatV(const char *format, va_list vargs) | 
					
						
							| 
									
										
										
										
											2006-04-22 23:28:04 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     va_list count; | 
					
						
							|  |  |  |     Py_ssize_t n = 0; | 
					
						
							|  |  |  |     const char* f; | 
					
						
							|  |  |  |     char *s; | 
					
						
							|  |  |  |     PyObject* string; | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2010-08-11 17:31:17 +00:00
										 |  |  |     Py_VA_COPY(count, vargs); | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     /* step 1: figure out how large a buffer we need */ | 
					
						
							|  |  |  |     for (f = format; *f; f++) { | 
					
						
							|  |  |  |         if (*f == '%') { | 
					
						
							|  |  |  |             const char* p = f; | 
					
						
							| 
									
										
										
										
											2010-11-05 17:23:41 +00:00
										 |  |  |             while (*++f && *f != '%' && !Py_ISALPHA(*f)) | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |                 ; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |             /* skip the 'l' or 'z' in {%ld, %zd, %lu, %zu} since
 | 
					
						
							|  |  |  |              * they don't affect the amount of space we reserve. | 
					
						
							|  |  |  |              */ | 
					
						
							|  |  |  |             if ((*f == 'l' || *f == 'z') && | 
					
						
							|  |  |  |                             (f[1] == 'd' || f[1] == 'u')) | 
					
						
							|  |  |  |                 ++f; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |             switch (*f) { | 
					
						
							|  |  |  |             case 'c': | 
					
						
							|  |  |  |                 (void)va_arg(count, int); | 
					
						
							|  |  |  |                 /* fall through... */ | 
					
						
							|  |  |  |             case '%': | 
					
						
							|  |  |  |                 n++; | 
					
						
							|  |  |  |                 break; | 
					
						
							|  |  |  |             case 'd': case 'u': case 'i': case 'x': | 
					
						
							|  |  |  |                 (void) va_arg(count, int); | 
					
						
							|  |  |  |                 /* 20 bytes is enough to hold a 64-bit
 | 
					
						
							|  |  |  |                    integer.  Decimal takes the most space. | 
					
						
							|  |  |  |                    This isn't enough for octal. */ | 
					
						
							|  |  |  |                 n += 20; | 
					
						
							|  |  |  |                 break; | 
					
						
							|  |  |  |             case 's': | 
					
						
							|  |  |  |                 s = va_arg(count, char*); | 
					
						
							|  |  |  |                 n += strlen(s); | 
					
						
							|  |  |  |                 break; | 
					
						
							|  |  |  |             case 'p': | 
					
						
							|  |  |  |                 (void) va_arg(count, int); | 
					
						
							|  |  |  |                 /* maximum 64-bit pointer representation:
 | 
					
						
							|  |  |  |                  * 0xffffffffffffffff | 
					
						
							|  |  |  |                  * so 19 characters is enough. | 
					
						
							|  |  |  |                  * XXX I count 18 -- what's the extra for? | 
					
						
							|  |  |  |                  */ | 
					
						
							|  |  |  |                 n += 19; | 
					
						
							|  |  |  |                 break; | 
					
						
							|  |  |  |             default: | 
					
						
							|  |  |  |                 /* if we stumble upon an unknown
 | 
					
						
							|  |  |  |                    formatting code, copy the rest of | 
					
						
							|  |  |  |                    the format string to the output | 
					
						
							|  |  |  |                    string. (we cannot just skip the | 
					
						
							|  |  |  |                    code, since there's no way to know | 
					
						
							|  |  |  |                    what's in the argument list) */ | 
					
						
							|  |  |  |                 n += strlen(p); | 
					
						
							|  |  |  |                 goto expand; | 
					
						
							|  |  |  |             } | 
					
						
							|  |  |  |         } else | 
					
						
							|  |  |  |             n++; | 
					
						
							|  |  |  |     } | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  |  expand: | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     /* step 2: fill the buffer */ | 
					
						
							|  |  |  |     /* Since we've analyzed how much space we need for the worst case,
 | 
					
						
							|  |  |  |        use sprintf directly instead of the slower PyOS_snprintf. */ | 
					
						
							|  |  |  |     string = PyBytes_FromStringAndSize(NULL, n); | 
					
						
							|  |  |  |     if (!string) | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     s = PyBytes_AsString(string); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     for (f = format; *f; f++) { | 
					
						
							|  |  |  |         if (*f == '%') { | 
					
						
							|  |  |  |             const char* p = f++; | 
					
						
							|  |  |  |             Py_ssize_t i; | 
					
						
							|  |  |  |             int longflag = 0; | 
					
						
							|  |  |  |             int size_tflag = 0; | 
					
						
							|  |  |  |             /* parse the width.precision part (we're only
 | 
					
						
							|  |  |  |                interested in the precision value, if any) */ | 
					
						
							|  |  |  |             n = 0; | 
					
						
							| 
									
										
										
										
											2010-11-05 17:23:41 +00:00
										 |  |  |             while (Py_ISDIGIT(*f)) | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |                 n = (n*10) + *f++ - '0'; | 
					
						
							|  |  |  |             if (*f == '.') { | 
					
						
							|  |  |  |                 f++; | 
					
						
							|  |  |  |                 n = 0; | 
					
						
							| 
									
										
										
										
											2010-11-05 17:23:41 +00:00
										 |  |  |                 while (Py_ISDIGIT(*f)) | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |                     n = (n*10) + *f++ - '0'; | 
					
						
							|  |  |  |             } | 
					
						
							| 
									
										
										
										
											2010-11-05 17:23:41 +00:00
										 |  |  |             while (*f && *f != '%' && !Py_ISALPHA(*f)) | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |                 f++; | 
					
						
							|  |  |  |             /* handle the long flag, but only for %ld and %lu.
 | 
					
						
							|  |  |  |                others can be added when necessary. */ | 
					
						
							|  |  |  |             if (*f == 'l' && (f[1] == 'd' || f[1] == 'u')) { | 
					
						
							|  |  |  |                 longflag = 1; | 
					
						
							|  |  |  |                 ++f; | 
					
						
							|  |  |  |             } | 
					
						
							|  |  |  |             /* handle the size_t flag. */ | 
					
						
							|  |  |  |             if (*f == 'z' && (f[1] == 'd' || f[1] == 'u')) { | 
					
						
							|  |  |  |                 size_tflag = 1; | 
					
						
							|  |  |  |                 ++f; | 
					
						
							|  |  |  |             } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |             switch (*f) { | 
					
						
							|  |  |  |             case 'c': | 
					
						
							|  |  |  |                 *s++ = va_arg(vargs, int); | 
					
						
							|  |  |  |                 break; | 
					
						
							|  |  |  |             case 'd': | 
					
						
							|  |  |  |                 if (longflag) | 
					
						
							|  |  |  |                     sprintf(s, "%ld", va_arg(vargs, long)); | 
					
						
							|  |  |  |                 else if (size_tflag) | 
					
						
							|  |  |  |                     sprintf(s, "%" PY_FORMAT_SIZE_T "d", | 
					
						
							|  |  |  |                         va_arg(vargs, Py_ssize_t)); | 
					
						
							|  |  |  |                 else | 
					
						
							|  |  |  |                     sprintf(s, "%d", va_arg(vargs, int)); | 
					
						
							|  |  |  |                 s += strlen(s); | 
					
						
							|  |  |  |                 break; | 
					
						
							|  |  |  |             case 'u': | 
					
						
							|  |  |  |                 if (longflag) | 
					
						
							|  |  |  |                     sprintf(s, "%lu", | 
					
						
							|  |  |  |                         va_arg(vargs, unsigned long)); | 
					
						
							|  |  |  |                 else if (size_tflag) | 
					
						
							|  |  |  |                     sprintf(s, "%" PY_FORMAT_SIZE_T "u", | 
					
						
							|  |  |  |                         va_arg(vargs, size_t)); | 
					
						
							|  |  |  |                 else | 
					
						
							|  |  |  |                     sprintf(s, "%u", | 
					
						
							|  |  |  |                         va_arg(vargs, unsigned int)); | 
					
						
							|  |  |  |                 s += strlen(s); | 
					
						
							|  |  |  |                 break; | 
					
						
							|  |  |  |             case 'i': | 
					
						
							|  |  |  |                 sprintf(s, "%i", va_arg(vargs, int)); | 
					
						
							|  |  |  |                 s += strlen(s); | 
					
						
							|  |  |  |                 break; | 
					
						
							|  |  |  |             case 'x': | 
					
						
							|  |  |  |                 sprintf(s, "%x", va_arg(vargs, int)); | 
					
						
							|  |  |  |                 s += strlen(s); | 
					
						
							|  |  |  |                 break; | 
					
						
							|  |  |  |             case 's': | 
					
						
							|  |  |  |                 p = va_arg(vargs, char*); | 
					
						
							|  |  |  |                 i = strlen(p); | 
					
						
							|  |  |  |                 if (n > 0 && i > n) | 
					
						
							|  |  |  |                     i = n; | 
					
						
							|  |  |  |                 Py_MEMCPY(s, p, i); | 
					
						
							|  |  |  |                 s += i; | 
					
						
							|  |  |  |                 break; | 
					
						
							|  |  |  |             case 'p': | 
					
						
							|  |  |  |                 sprintf(s, "%p", va_arg(vargs, void*)); | 
					
						
							|  |  |  |                 /* %p is ill-defined:  ensure leading 0x. */ | 
					
						
							|  |  |  |                 if (s[1] == 'X') | 
					
						
							|  |  |  |                     s[1] = 'x'; | 
					
						
							|  |  |  |                 else if (s[1] != 'x') { | 
					
						
							|  |  |  |                     memmove(s+2, s, strlen(s)+1); | 
					
						
							|  |  |  |                     s[0] = '0'; | 
					
						
							|  |  |  |                     s[1] = 'x'; | 
					
						
							|  |  |  |                 } | 
					
						
							|  |  |  |                 s += strlen(s); | 
					
						
							|  |  |  |                 break; | 
					
						
							|  |  |  |             case '%': | 
					
						
							|  |  |  |                 *s++ = '%'; | 
					
						
							|  |  |  |                 break; | 
					
						
							|  |  |  |             default: | 
					
						
							|  |  |  |                 strcpy(s, p); | 
					
						
							|  |  |  |                 s += strlen(s); | 
					
						
							|  |  |  |                 goto end; | 
					
						
							|  |  |  |             } | 
					
						
							|  |  |  |         } else | 
					
						
							|  |  |  |             *s++ = *f; | 
					
						
							|  |  |  |     } | 
					
						
							| 
									
										
										
										
											2006-04-22 23:28:04 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  |  end: | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     _PyBytes_Resize(&string, s - PyBytes_AS_STRING(string)); | 
					
						
							|  |  |  |     return string; | 
					
						
							| 
									
										
										
										
											2006-04-22 23:28:04 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | PyObject * | 
					
						
							|  |  |  | PyBytes_FromFormat(const char *format, ...) | 
					
						
							| 
									
										
										
										
											2006-04-22 23:28:04 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     PyObject* ret; | 
					
						
							|  |  |  |     va_list vargs; | 
					
						
							| 
									
										
										
										
											2006-04-22 23:28:04 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | #ifdef HAVE_STDARG_PROTOTYPES
 | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     va_start(vargs, format); | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | #else
 | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     va_start(vargs); | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | #endif
 | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     ret = PyBytes_FromFormatV(format, vargs); | 
					
						
							|  |  |  |     va_end(vargs); | 
					
						
							|  |  |  |     return ret; | 
					
						
							| 
									
										
										
										
											2006-04-22 23:28:04 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | static void | 
					
						
							| 
									
										
										
										
											2009-04-18 15:17:02 +00:00
										 |  |  | bytes_dealloc(PyObject *op) | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     Py_TYPE(op)->tp_free(op); | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | /* Unescape a backslash-escaped string. If unicode is non-zero,
 | 
					
						
							|  |  |  |    the string is a u-literal. If recode_encoding is non-zero, | 
					
						
							|  |  |  |    the string is UTF-8 encoded and should be re-encoded in the | 
					
						
							|  |  |  |    specified encoding.  */ | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | PyObject *PyBytes_DecodeEscape(const char *s, | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |                                 Py_ssize_t len, | 
					
						
							|  |  |  |                                 const char *errors, | 
					
						
							|  |  |  |                                 Py_ssize_t unicode, | 
					
						
							|  |  |  |                                 const char *recode_encoding) | 
					
						
							|  |  |  | { | 
					
						
							|  |  |  |     int c; | 
					
						
							|  |  |  |     char *p, *buf; | 
					
						
							|  |  |  |     const char *end; | 
					
						
							|  |  |  |     PyObject *v; | 
					
						
							|  |  |  |     Py_ssize_t newlen = recode_encoding ? 4*len:len; | 
					
						
							|  |  |  |     v = PyBytes_FromStringAndSize((char *)NULL, newlen); | 
					
						
							|  |  |  |     if (v == NULL) | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  |     p = buf = PyBytes_AsString(v); | 
					
						
							|  |  |  |     end = s + len; | 
					
						
							|  |  |  |     while (s < end) { | 
					
						
							|  |  |  |         if (*s != '\\') { | 
					
						
							|  |  |  |           non_esc: | 
					
						
							|  |  |  |             if (recode_encoding && (*s & 0x80)) { | 
					
						
							|  |  |  |                 PyObject *u, *w; | 
					
						
							|  |  |  |                 char *r; | 
					
						
							|  |  |  |                 const char* t; | 
					
						
							|  |  |  |                 Py_ssize_t rn; | 
					
						
							|  |  |  |                 t = s; | 
					
						
							|  |  |  |                 /* Decode non-ASCII bytes as UTF-8. */ | 
					
						
							|  |  |  |                 while (t < end && (*t & 0x80)) t++; | 
					
						
							|  |  |  |                 u = PyUnicode_DecodeUTF8(s, t - s, errors); | 
					
						
							|  |  |  |                 if(!u) goto failed; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |                 /* Recode them in target encoding. */ | 
					
						
							|  |  |  |                 w = PyUnicode_AsEncodedString( | 
					
						
							|  |  |  |                     u, recode_encoding, errors); | 
					
						
							|  |  |  |                 Py_DECREF(u); | 
					
						
							|  |  |  |                 if (!w)                 goto failed; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |                 /* Append bytes to output buffer. */ | 
					
						
							|  |  |  |                 assert(PyBytes_Check(w)); | 
					
						
							|  |  |  |                 r = PyBytes_AS_STRING(w); | 
					
						
							|  |  |  |                 rn = PyBytes_GET_SIZE(w); | 
					
						
							|  |  |  |                 Py_MEMCPY(p, r, rn); | 
					
						
							|  |  |  |                 p += rn; | 
					
						
							|  |  |  |                 Py_DECREF(w); | 
					
						
							|  |  |  |                 s = t; | 
					
						
							|  |  |  |             } else { | 
					
						
							|  |  |  |                 *p++ = *s++; | 
					
						
							|  |  |  |             } | 
					
						
							|  |  |  |             continue; | 
					
						
							|  |  |  |         } | 
					
						
							|  |  |  |         s++; | 
					
						
							|  |  |  |         if (s==end) { | 
					
						
							|  |  |  |             PyErr_SetString(PyExc_ValueError, | 
					
						
							|  |  |  |                             "Trailing \\ in string"); | 
					
						
							|  |  |  |             goto failed; | 
					
						
							|  |  |  |         } | 
					
						
							|  |  |  |         switch (*s++) { | 
					
						
							|  |  |  |         /* XXX This assumes ASCII! */ | 
					
						
							|  |  |  |         case '\n': break; | 
					
						
							|  |  |  |         case '\\': *p++ = '\\'; break; | 
					
						
							|  |  |  |         case '\'': *p++ = '\''; break; | 
					
						
							|  |  |  |         case '\"': *p++ = '\"'; break; | 
					
						
							|  |  |  |         case 'b': *p++ = '\b'; break; | 
					
						
							|  |  |  |         case 'f': *p++ = '\014'; break; /* FF */ | 
					
						
							|  |  |  |         case 't': *p++ = '\t'; break; | 
					
						
							|  |  |  |         case 'n': *p++ = '\n'; break; | 
					
						
							|  |  |  |         case 'r': *p++ = '\r'; break; | 
					
						
							|  |  |  |         case 'v': *p++ = '\013'; break; /* VT */ | 
					
						
							|  |  |  |         case 'a': *p++ = '\007'; break; /* BEL, not classic C */ | 
					
						
							|  |  |  |         case '0': case '1': case '2': case '3': | 
					
						
							|  |  |  |         case '4': case '5': case '6': case '7': | 
					
						
							|  |  |  |             c = s[-1] - '0'; | 
					
						
							|  |  |  |             if (s < end && '0' <= *s && *s <= '7') { | 
					
						
							|  |  |  |                 c = (c<<3) + *s++ - '0'; | 
					
						
							|  |  |  |                 if (s < end && '0' <= *s && *s <= '7') | 
					
						
							|  |  |  |                     c = (c<<3) + *s++ - '0'; | 
					
						
							|  |  |  |             } | 
					
						
							|  |  |  |             *p++ = c; | 
					
						
							|  |  |  |             break; | 
					
						
							|  |  |  |         case 'x': | 
					
						
							| 
									
										
										
										
											2010-11-05 17:23:41 +00:00
										 |  |  |             if (s+1 < end && Py_ISXDIGIT(s[0]) && Py_ISXDIGIT(s[1])) { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |                 unsigned int x = 0; | 
					
						
							|  |  |  |                 c = Py_CHARMASK(*s); | 
					
						
							|  |  |  |                 s++; | 
					
						
							| 
									
										
										
										
											2010-11-05 17:23:41 +00:00
										 |  |  |                 if (Py_ISDIGIT(c)) | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |                     x = c - '0'; | 
					
						
							| 
									
										
										
										
											2010-11-05 17:23:41 +00:00
										 |  |  |                 else if (Py_ISLOWER(c)) | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |                     x = 10 + c - 'a'; | 
					
						
							|  |  |  |                 else | 
					
						
							|  |  |  |                     x = 10 + c - 'A'; | 
					
						
							|  |  |  |                 x = x << 4; | 
					
						
							|  |  |  |                 c = Py_CHARMASK(*s); | 
					
						
							|  |  |  |                 s++; | 
					
						
							| 
									
										
										
										
											2010-11-05 17:23:41 +00:00
										 |  |  |                 if (Py_ISDIGIT(c)) | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |                     x += c - '0'; | 
					
						
							| 
									
										
										
										
											2010-11-05 17:23:41 +00:00
										 |  |  |                 else if (Py_ISLOWER(c)) | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |                     x += 10 + c - 'a'; | 
					
						
							|  |  |  |                 else | 
					
						
							|  |  |  |                     x += 10 + c - 'A'; | 
					
						
							|  |  |  |                 *p++ = x; | 
					
						
							|  |  |  |                 break; | 
					
						
							|  |  |  |             } | 
					
						
							|  |  |  |             if (!errors || strcmp(errors, "strict") == 0) { | 
					
						
							|  |  |  |                 PyErr_SetString(PyExc_ValueError, | 
					
						
							|  |  |  |                                 "invalid \\x escape"); | 
					
						
							|  |  |  |                 goto failed; | 
					
						
							|  |  |  |             } | 
					
						
							|  |  |  |             if (strcmp(errors, "replace") == 0) { | 
					
						
							|  |  |  |                 *p++ = '?'; | 
					
						
							|  |  |  |             } else if (strcmp(errors, "ignore") == 0) | 
					
						
							|  |  |  |                 /* do nothing */; | 
					
						
							|  |  |  |             else { | 
					
						
							|  |  |  |                 PyErr_Format(PyExc_ValueError, | 
					
						
							|  |  |  |                              "decoding error; unknown " | 
					
						
							|  |  |  |                              "error handling code: %.400s", | 
					
						
							|  |  |  |                              errors); | 
					
						
							|  |  |  |                 goto failed; | 
					
						
							|  |  |  |             } | 
					
						
							|  |  |  |         default: | 
					
						
							|  |  |  |             *p++ = '\\'; | 
					
						
							|  |  |  |             s--; | 
					
						
							| 
									
										
										
										
											2011-03-15 05:18:48 +02:00
										 |  |  |             goto non_esc; /* an arbitrary number of unescaped
 | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |                              UTF-8 bytes may follow. */ | 
					
						
							|  |  |  |         } | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     if (p-buf < newlen) | 
					
						
							|  |  |  |         _PyBytes_Resize(&v, p - buf); | 
					
						
							|  |  |  |     return v; | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  |   failed: | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     Py_DECREF(v); | 
					
						
							|  |  |  |     return NULL; | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | } | 
					
						
							| 
									
										
										
										
											2006-04-22 23:28:04 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | /* -------------------------------------------------------------------- */ | 
					
						
							|  |  |  | /* object api */ | 
					
						
							| 
									
										
										
										
											2006-04-22 23:28:04 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | Py_ssize_t | 
					
						
							|  |  |  | PyBytes_Size(register PyObject *op) | 
					
						
							|  |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     if (!PyBytes_Check(op)) { | 
					
						
							|  |  |  |         PyErr_Format(PyExc_TypeError, | 
					
						
							|  |  |  |              "expected bytes, %.200s found", Py_TYPE(op)->tp_name); | 
					
						
							|  |  |  |         return -1; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     return Py_SIZE(op); | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | } | 
					
						
							| 
									
										
										
										
											2006-05-05 04:34:18 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | char * | 
					
						
							|  |  |  | PyBytes_AsString(register PyObject *op) | 
					
						
							|  |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     if (!PyBytes_Check(op)) { | 
					
						
							|  |  |  |         PyErr_Format(PyExc_TypeError, | 
					
						
							|  |  |  |              "expected bytes, %.200s found", Py_TYPE(op)->tp_name); | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     return ((PyBytesObject *)op)->ob_sval; | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | } | 
					
						
							| 
									
										
										
										
											2007-08-18 11:21:56 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | int | 
					
						
							|  |  |  | PyBytes_AsStringAndSize(register PyObject *obj, | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |                          register char **s, | 
					
						
							|  |  |  |                          register Py_ssize_t *len) | 
					
						
							|  |  |  | { | 
					
						
							|  |  |  |     if (s == NULL) { | 
					
						
							|  |  |  |         PyErr_BadInternalCall(); | 
					
						
							|  |  |  |         return -1; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     if (!PyBytes_Check(obj)) { | 
					
						
							|  |  |  |         PyErr_Format(PyExc_TypeError, | 
					
						
							|  |  |  |              "expected bytes, %.200s found", Py_TYPE(obj)->tp_name); | 
					
						
							|  |  |  |         return -1; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     *s = PyBytes_AS_STRING(obj); | 
					
						
							|  |  |  |     if (len != NULL) | 
					
						
							|  |  |  |         *len = PyBytes_GET_SIZE(obj); | 
					
						
							|  |  |  |     else if (strlen(*s) != (size_t)PyBytes_GET_SIZE(obj)) { | 
					
						
							|  |  |  |         PyErr_SetString(PyExc_TypeError, | 
					
						
							|  |  |  |                         "expected bytes with no null"); | 
					
						
							|  |  |  |         return -1; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     return 0; | 
					
						
							| 
									
										
										
										
											2006-04-22 23:28:04 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | /* -------------------------------------------------------------------- */ | 
					
						
							|  |  |  | /* Methods */ | 
					
						
							| 
									
										
										
										
											2007-04-13 01:39:34 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2009-04-16 20:16:10 +00:00
										 |  |  | #include "stringlib/stringdefs.h"
 | 
					
						
							| 
									
										
										
										
											2007-04-13 01:39:34 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | #include "stringlib/fastsearch.h"
 | 
					
						
							|  |  |  | #include "stringlib/count.h"
 | 
					
						
							|  |  |  | #include "stringlib/find.h"
 | 
					
						
							|  |  |  | #include "stringlib/partition.h"
 | 
					
						
							| 
									
										
										
										
											2010-01-13 08:07:53 +00:00
										 |  |  | #include "stringlib/split.h"
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | #include "stringlib/ctype.h"
 | 
					
						
							| 
									
										
										
										
											2006-04-27 22:54:26 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2009-11-30 01:01:42 +00:00
										 |  |  | #include "stringlib/transmogrify.h"
 | 
					
						
							| 
									
										
										
										
											2006-04-27 22:54:26 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | PyObject * | 
					
						
							|  |  |  | PyBytes_Repr(PyObject *obj, int smartquotes) | 
					
						
							|  |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     static const char *hexdigits = "0123456789abcdef"; | 
					
						
							|  |  |  |     register PyBytesObject* op = (PyBytesObject*) obj; | 
					
						
							| 
									
										
										
										
											2011-09-28 07:41:54 +02:00
										 |  |  |     Py_ssize_t i, length = Py_SIZE(op); | 
					
						
							|  |  |  |     size_t newsize, squotes, dquotes; | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     PyObject *v; | 
					
						
							| 
									
										
										
										
											2011-09-28 07:41:54 +02:00
										 |  |  |     unsigned char quote, *s, *p; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     /* Compute size of output string */ | 
					
						
							|  |  |  |     squotes = dquotes = 0; | 
					
						
							|  |  |  |     newsize = 3; /* b'' */ | 
					
						
							|  |  |  |     s = (unsigned char*)op->ob_sval; | 
					
						
							|  |  |  |     for (i = 0; i < length; i++) { | 
					
						
							|  |  |  |         switch(s[i]) { | 
					
						
							|  |  |  |         case '\'': squotes++; newsize++; break; | 
					
						
							|  |  |  |         case '"':  dquotes++; newsize++; break; | 
					
						
							|  |  |  |         case '\\': case '\t': case '\n': case '\r': | 
					
						
							|  |  |  |             newsize += 2; break; /* \C */ | 
					
						
							|  |  |  |         default: | 
					
						
							|  |  |  |             if (s[i] < ' ' || s[i] >= 0x7f) | 
					
						
							|  |  |  |                 newsize += 4; /* \xHH */ | 
					
						
							|  |  |  |             else | 
					
						
							|  |  |  |                 newsize++; | 
					
						
							|  |  |  |         } | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     quote = '\''; | 
					
						
							|  |  |  |     if (smartquotes && squotes && !dquotes) | 
					
						
							|  |  |  |         quote = '"'; | 
					
						
							|  |  |  |     if (squotes && quote == '\'') | 
					
						
							|  |  |  |         newsize += squotes; | 
					
						
							| 
									
										
										
										
											2011-09-29 04:02:13 +02:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2011-09-28 07:41:54 +02:00
										 |  |  |     if (newsize > (PY_SSIZE_T_MAX - sizeof(PyUnicodeObject) - 1)) { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |         PyErr_SetString(PyExc_OverflowError, | 
					
						
							|  |  |  |             "bytes object is too large to make repr"); | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  |     } | 
					
						
							| 
									
										
										
										
											2011-09-28 07:41:54 +02:00
										 |  |  | 
 | 
					
						
							|  |  |  |     v = PyUnicode_New(newsize, 127); | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     if (v == NULL) { | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  |     } | 
					
						
							| 
									
										
										
										
											2011-09-28 07:41:54 +02:00
										 |  |  |     p = PyUnicode_1BYTE_DATA(v); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     *p++ = 'b', *p++ = quote; | 
					
						
							|  |  |  |     for (i = 0; i < length; i++) { | 
					
						
							|  |  |  |         unsigned char c = op->ob_sval[i]; | 
					
						
							|  |  |  |         if (c == quote || c == '\\') | 
					
						
							|  |  |  |             *p++ = '\\', *p++ = c; | 
					
						
							|  |  |  |         else if (c == '\t') | 
					
						
							|  |  |  |             *p++ = '\\', *p++ = 't'; | 
					
						
							|  |  |  |         else if (c == '\n') | 
					
						
							|  |  |  |             *p++ = '\\', *p++ = 'n'; | 
					
						
							|  |  |  |         else if (c == '\r') | 
					
						
							|  |  |  |             *p++ = '\\', *p++ = 'r'; | 
					
						
							|  |  |  |         else if (c < ' ' || c >= 0x7f) { | 
					
						
							|  |  |  |             *p++ = '\\'; | 
					
						
							|  |  |  |             *p++ = 'x'; | 
					
						
							|  |  |  |             *p++ = hexdigits[(c & 0xf0) >> 4]; | 
					
						
							|  |  |  |             *p++ = hexdigits[c & 0xf]; | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |         } | 
					
						
							| 
									
										
										
										
											2011-09-28 07:41:54 +02:00
										 |  |  |         else | 
					
						
							|  |  |  |             *p++ = c; | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     } | 
					
						
							| 
									
										
										
										
											2011-09-28 07:41:54 +02:00
										 |  |  |     *p++ = quote; | 
					
						
							|  |  |  |     return v; | 
					
						
							| 
									
										
										
										
											2006-04-27 22:54:26 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2006-04-24 13:47:05 +00:00
										 |  |  | static PyObject * | 
					
						
							| 
									
										
										
										
											2009-04-18 15:17:02 +00:00
										 |  |  | bytes_repr(PyObject *op) | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     return PyBytes_Repr(op, 1); | 
					
						
							| 
									
										
										
										
											2006-04-24 13:47:05 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2006-04-27 22:54:26 +00:00
										 |  |  | static PyObject * | 
					
						
							| 
									
										
										
										
											2009-04-18 15:17:02 +00:00
										 |  |  | bytes_str(PyObject *op) | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     if (Py_BytesWarningFlag) { | 
					
						
							|  |  |  |         if (PyErr_WarnEx(PyExc_BytesWarning, | 
					
						
							|  |  |  |                          "str() on a bytes instance", 1)) | 
					
						
							|  |  |  |             return NULL; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     return bytes_repr(op); | 
					
						
							| 
									
										
										
										
											2006-04-27 22:54:26 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | static Py_ssize_t | 
					
						
							| 
									
										
										
										
											2009-04-18 15:17:02 +00:00
										 |  |  | bytes_length(PyBytesObject *a) | 
					
						
							| 
									
										
										
										
											2006-04-22 23:28:04 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     return Py_SIZE(a); | 
					
						
							| 
									
										
										
										
											2006-04-24 13:47:05 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | /* This is also used by PyBytes_Concat() */ | 
					
						
							| 
									
										
										
										
											2006-04-24 13:47:05 +00:00
										 |  |  | static PyObject * | 
					
						
							| 
									
										
										
										
											2009-04-18 15:17:02 +00:00
										 |  |  | bytes_concat(PyObject *a, PyObject *b) | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     Py_ssize_t size; | 
					
						
							|  |  |  |     Py_buffer va, vb; | 
					
						
							|  |  |  |     PyObject *result = NULL; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     va.len = -1; | 
					
						
							|  |  |  |     vb.len = -1; | 
					
						
							|  |  |  |     if (_getbuffer(a, &va) < 0  || | 
					
						
							|  |  |  |         _getbuffer(b, &vb) < 0) { | 
					
						
							|  |  |  |         PyErr_Format(PyExc_TypeError, "can't concat %.100s to %.100s", | 
					
						
							|  |  |  |                      Py_TYPE(a)->tp_name, Py_TYPE(b)->tp_name); | 
					
						
							|  |  |  |         goto done; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     /* Optimize end cases */ | 
					
						
							|  |  |  |     if (va.len == 0 && PyBytes_CheckExact(b)) { | 
					
						
							|  |  |  |         result = b; | 
					
						
							|  |  |  |         Py_INCREF(result); | 
					
						
							|  |  |  |         goto done; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     if (vb.len == 0 && PyBytes_CheckExact(a)) { | 
					
						
							|  |  |  |         result = a; | 
					
						
							|  |  |  |         Py_INCREF(result); | 
					
						
							|  |  |  |         goto done; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     size = va.len + vb.len; | 
					
						
							|  |  |  |     if (size < 0) { | 
					
						
							|  |  |  |         PyErr_NoMemory(); | 
					
						
							|  |  |  |         goto done; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     result = PyBytes_FromStringAndSize(NULL, size); | 
					
						
							|  |  |  |     if (result != NULL) { | 
					
						
							|  |  |  |         memcpy(PyBytes_AS_STRING(result), va.buf, va.len); | 
					
						
							|  |  |  |         memcpy(PyBytes_AS_STRING(result) + va.len, vb.buf, vb.len); | 
					
						
							|  |  |  |     } | 
					
						
							| 
									
										
										
										
											2006-04-24 13:47:05 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  |   done: | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     if (va.len != -1) | 
					
						
							|  |  |  |         PyBuffer_Release(&va); | 
					
						
							|  |  |  |     if (vb.len != -1) | 
					
						
							|  |  |  |         PyBuffer_Release(&vb); | 
					
						
							|  |  |  |     return result; | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | } | 
					
						
							| 
									
										
										
										
											2007-08-24 17:33:11 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | static PyObject * | 
					
						
							| 
									
										
										
										
											2009-04-18 15:17:02 +00:00
										 |  |  | bytes_repeat(register PyBytesObject *a, register Py_ssize_t n) | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     register Py_ssize_t i; | 
					
						
							|  |  |  |     register Py_ssize_t j; | 
					
						
							|  |  |  |     register Py_ssize_t size; | 
					
						
							|  |  |  |     register PyBytesObject *op; | 
					
						
							|  |  |  |     size_t nbytes; | 
					
						
							|  |  |  |     if (n < 0) | 
					
						
							|  |  |  |         n = 0; | 
					
						
							|  |  |  |     /* watch out for overflows:  the size can overflow int,
 | 
					
						
							|  |  |  |      * and the # of bytes needed can overflow size_t | 
					
						
							|  |  |  |      */ | 
					
						
							| 
									
										
										
										
											2010-08-10 18:35:01 +00:00
										 |  |  |     if (n > 0 && Py_SIZE(a) > PY_SSIZE_T_MAX / n) { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |         PyErr_SetString(PyExc_OverflowError, | 
					
						
							|  |  |  |             "repeated bytes are too long"); | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  |     } | 
					
						
							| 
									
										
										
										
											2010-08-10 18:35:01 +00:00
										 |  |  |     size = Py_SIZE(a) * n; | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     if (size == Py_SIZE(a) && PyBytes_CheckExact(a)) { | 
					
						
							|  |  |  |         Py_INCREF(a); | 
					
						
							|  |  |  |         return (PyObject *)a; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     nbytes = (size_t)size; | 
					
						
							|  |  |  |     if (nbytes + PyBytesObject_SIZE <= nbytes) { | 
					
						
							|  |  |  |         PyErr_SetString(PyExc_OverflowError, | 
					
						
							|  |  |  |             "repeated bytes are too long"); | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     op = (PyBytesObject *)PyObject_MALLOC(PyBytesObject_SIZE + nbytes); | 
					
						
							|  |  |  |     if (op == NULL) | 
					
						
							|  |  |  |         return PyErr_NoMemory(); | 
					
						
							|  |  |  |     PyObject_INIT_VAR(op, &PyBytes_Type, size); | 
					
						
							|  |  |  |     op->ob_shash = -1; | 
					
						
							|  |  |  |     op->ob_sval[size] = '\0'; | 
					
						
							|  |  |  |     if (Py_SIZE(a) == 1 && n > 0) { | 
					
						
							|  |  |  |         memset(op->ob_sval, a->ob_sval[0] , n); | 
					
						
							|  |  |  |         return (PyObject *) op; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     i = 0; | 
					
						
							|  |  |  |     if (i < size) { | 
					
						
							|  |  |  |         Py_MEMCPY(op->ob_sval, a->ob_sval, Py_SIZE(a)); | 
					
						
							|  |  |  |         i = Py_SIZE(a); | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     while (i < size) { | 
					
						
							|  |  |  |         j = (i <= size-i)  ?  i  :  size-i; | 
					
						
							|  |  |  |         Py_MEMCPY(op->ob_sval+i, op->ob_sval, j); | 
					
						
							|  |  |  |         i += j; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     return (PyObject *) op; | 
					
						
							| 
									
										
										
										
											2006-04-24 13:47:05 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | static int | 
					
						
							| 
									
										
										
										
											2009-04-18 15:17:02 +00:00
										 |  |  | bytes_contains(PyObject *self, PyObject *arg) | 
					
						
							| 
									
										
										
										
											2006-04-24 13:47:05 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  |     Py_ssize_t ival = PyNumber_AsSsize_t(arg, PyExc_ValueError); | 
					
						
							|  |  |  |     if (ival == -1 && PyErr_Occurred()) { | 
					
						
							| 
									
										
										
										
											2010-06-09 16:38:55 +00:00
										 |  |  |         Py_buffer varg; | 
					
						
							| 
									
										
										
										
											2010-08-15 17:12:55 +00:00
										 |  |  |         Py_ssize_t pos; | 
					
						
							| 
									
										
										
										
											2010-06-09 16:38:55 +00:00
										 |  |  |         PyErr_Clear(); | 
					
						
							|  |  |  |         if (_getbuffer(arg, &varg) < 0) | 
					
						
							|  |  |  |             return -1; | 
					
						
							|  |  |  |         pos = stringlib_find(PyBytes_AS_STRING(self), Py_SIZE(self), | 
					
						
							|  |  |  |                              varg.buf, varg.len, 0); | 
					
						
							|  |  |  |         PyBuffer_Release(&varg); | 
					
						
							|  |  |  |         return pos >= 0; | 
					
						
							| 
									
										
										
										
											2006-04-24 13:47:05 +00:00
										 |  |  |     } | 
					
						
							|  |  |  |     if (ival < 0 || ival >= 256) { | 
					
						
							| 
									
										
										
										
											2010-06-09 16:38:55 +00:00
										 |  |  |         PyErr_SetString(PyExc_ValueError, "byte must be in range(0, 256)"); | 
					
						
							|  |  |  |         return -1; | 
					
						
							| 
									
										
										
										
											2006-04-24 13:47:05 +00:00
										 |  |  |     } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2010-08-15 17:12:55 +00:00
										 |  |  |     return memchr(PyBytes_AS_STRING(self), (int) ival, Py_SIZE(self)) != NULL; | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | static PyObject * | 
					
						
							| 
									
										
										
										
											2009-04-18 15:17:02 +00:00
										 |  |  | bytes_item(PyBytesObject *a, register Py_ssize_t i) | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     if (i < 0 || i >= Py_SIZE(a)) { | 
					
						
							|  |  |  |         PyErr_SetString(PyExc_IndexError, "index out of range"); | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     return PyLong_FromLong((unsigned char)a->ob_sval[i]); | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | static PyObject* | 
					
						
							| 
									
										
										
										
											2009-04-18 15:17:02 +00:00
										 |  |  | bytes_richcompare(PyBytesObject *a, PyBytesObject *b, int op) | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     int c; | 
					
						
							|  |  |  |     Py_ssize_t len_a, len_b; | 
					
						
							|  |  |  |     Py_ssize_t min_len; | 
					
						
							|  |  |  |     PyObject *result; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     /* Make sure both arguments are strings. */ | 
					
						
							|  |  |  |     if (!(PyBytes_Check(a) && PyBytes_Check(b))) { | 
					
						
							|  |  |  |         if (Py_BytesWarningFlag && (op == Py_EQ || op == Py_NE) && | 
					
						
							|  |  |  |             (PyObject_IsInstance((PyObject*)a, | 
					
						
							|  |  |  |                                  (PyObject*)&PyUnicode_Type) || | 
					
						
							|  |  |  |             PyObject_IsInstance((PyObject*)b, | 
					
						
							|  |  |  |                                  (PyObject*)&PyUnicode_Type))) { | 
					
						
							|  |  |  |             if (PyErr_WarnEx(PyExc_BytesWarning, | 
					
						
							|  |  |  |                         "Comparison between bytes and string", 1)) | 
					
						
							|  |  |  |                 return NULL; | 
					
						
							|  |  |  |         } | 
					
						
							|  |  |  |         result = Py_NotImplemented; | 
					
						
							|  |  |  |         goto out; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     if (a == b) { | 
					
						
							|  |  |  |         switch (op) { | 
					
						
							|  |  |  |         case Py_EQ:case Py_LE:case Py_GE: | 
					
						
							|  |  |  |             result = Py_True; | 
					
						
							|  |  |  |             goto out; | 
					
						
							|  |  |  |         case Py_NE:case Py_LT:case Py_GT: | 
					
						
							|  |  |  |             result = Py_False; | 
					
						
							|  |  |  |             goto out; | 
					
						
							|  |  |  |         } | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     if (op == Py_EQ) { | 
					
						
							|  |  |  |         /* Supporting Py_NE here as well does not save
 | 
					
						
							|  |  |  |            much time, since Py_NE is rarely used.  */ | 
					
						
							|  |  |  |         if (Py_SIZE(a) == Py_SIZE(b) | 
					
						
							|  |  |  |             && (a->ob_sval[0] == b->ob_sval[0] | 
					
						
							|  |  |  |             && memcmp(a->ob_sval, b->ob_sval, Py_SIZE(a)) == 0)) { | 
					
						
							|  |  |  |             result = Py_True; | 
					
						
							|  |  |  |         } else { | 
					
						
							|  |  |  |             result = Py_False; | 
					
						
							|  |  |  |         } | 
					
						
							|  |  |  |         goto out; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     len_a = Py_SIZE(a); len_b = Py_SIZE(b); | 
					
						
							|  |  |  |     min_len = (len_a < len_b) ? len_a : len_b; | 
					
						
							|  |  |  |     if (min_len > 0) { | 
					
						
							|  |  |  |         c = Py_CHARMASK(*a->ob_sval) - Py_CHARMASK(*b->ob_sval); | 
					
						
							|  |  |  |         if (c==0) | 
					
						
							|  |  |  |             c = memcmp(a->ob_sval, b->ob_sval, min_len); | 
					
						
							|  |  |  |     } else | 
					
						
							|  |  |  |         c = 0; | 
					
						
							|  |  |  |     if (c == 0) | 
					
						
							|  |  |  |         c = (len_a < len_b) ? -1 : (len_a > len_b) ? 1 : 0; | 
					
						
							|  |  |  |     switch (op) { | 
					
						
							|  |  |  |     case Py_LT: c = c <  0; break; | 
					
						
							|  |  |  |     case Py_LE: c = c <= 0; break; | 
					
						
							|  |  |  |     case Py_EQ: assert(0);  break; /* unreachable */ | 
					
						
							|  |  |  |     case Py_NE: c = c != 0; break; | 
					
						
							|  |  |  |     case Py_GT: c = c >  0; break; | 
					
						
							|  |  |  |     case Py_GE: c = c >= 0; break; | 
					
						
							|  |  |  |     default: | 
					
						
							|  |  |  |         result = Py_NotImplemented; | 
					
						
							|  |  |  |         goto out; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     result = c ? Py_True : Py_False; | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  |   out: | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     Py_INCREF(result); | 
					
						
							|  |  |  |     return result; | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2010-10-17 20:54:53 +00:00
										 |  |  | static Py_hash_t | 
					
						
							| 
									
										
										
										
											2009-04-18 15:17:02 +00:00
										 |  |  | bytes_hash(PyBytesObject *a) | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     register Py_ssize_t len; | 
					
						
							|  |  |  |     register unsigned char *p; | 
					
						
							| 
									
										
										
										
											2011-09-24 18:18:40 +01:00
										 |  |  |     register Py_uhash_t x; | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     if (a->ob_shash != -1) | 
					
						
							|  |  |  |         return a->ob_shash; | 
					
						
							|  |  |  |     len = Py_SIZE(a); | 
					
						
							|  |  |  |     p = (unsigned char *) a->ob_sval; | 
					
						
							| 
									
										
										
										
											2011-09-24 18:18:40 +01:00
										 |  |  |     x = (Py_uhash_t)*p << 7; | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     while (--len >= 0) | 
					
						
							| 
									
										
										
										
											2011-09-24 18:18:40 +01:00
										 |  |  |         x = (1000003U*x) ^ (Py_uhash_t)*p++; | 
					
						
							|  |  |  |     x ^= (Py_uhash_t)Py_SIZE(a); | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     if (x == -1) | 
					
						
							|  |  |  |         x = -2; | 
					
						
							|  |  |  |     a->ob_shash = x; | 
					
						
							|  |  |  |     return x; | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | static PyObject* | 
					
						
							| 
									
										
										
										
											2009-04-18 15:17:02 +00:00
										 |  |  | bytes_subscript(PyBytesObject* self, PyObject* item) | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     if (PyIndex_Check(item)) { | 
					
						
							|  |  |  |         Py_ssize_t i = PyNumber_AsSsize_t(item, PyExc_IndexError); | 
					
						
							|  |  |  |         if (i == -1 && PyErr_Occurred()) | 
					
						
							|  |  |  |             return NULL; | 
					
						
							|  |  |  |         if (i < 0) | 
					
						
							|  |  |  |             i += PyBytes_GET_SIZE(self); | 
					
						
							|  |  |  |         if (i < 0 || i >= PyBytes_GET_SIZE(self)) { | 
					
						
							|  |  |  |             PyErr_SetString(PyExc_IndexError, | 
					
						
							|  |  |  |                             "index out of range"); | 
					
						
							|  |  |  |             return NULL; | 
					
						
							|  |  |  |         } | 
					
						
							|  |  |  |         return PyLong_FromLong((unsigned char)self->ob_sval[i]); | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     else if (PySlice_Check(item)) { | 
					
						
							|  |  |  |         Py_ssize_t start, stop, step, slicelength, cur, i; | 
					
						
							|  |  |  |         char* source_buf; | 
					
						
							|  |  |  |         char* result_buf; | 
					
						
							|  |  |  |         PyObject* result; | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2010-12-03 20:14:31 +00:00
										 |  |  |         if (PySlice_GetIndicesEx(item, | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |                          PyBytes_GET_SIZE(self), | 
					
						
							|  |  |  |                          &start, &stop, &step, &slicelength) < 0) { | 
					
						
							|  |  |  |             return NULL; | 
					
						
							|  |  |  |         } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         if (slicelength <= 0) { | 
					
						
							|  |  |  |             return PyBytes_FromStringAndSize("", 0); | 
					
						
							|  |  |  |         } | 
					
						
							|  |  |  |         else if (start == 0 && step == 1 && | 
					
						
							|  |  |  |                  slicelength == PyBytes_GET_SIZE(self) && | 
					
						
							|  |  |  |                  PyBytes_CheckExact(self)) { | 
					
						
							|  |  |  |             Py_INCREF(self); | 
					
						
							|  |  |  |             return (PyObject *)self; | 
					
						
							|  |  |  |         } | 
					
						
							|  |  |  |         else if (step == 1) { | 
					
						
							|  |  |  |             return PyBytes_FromStringAndSize( | 
					
						
							|  |  |  |                 PyBytes_AS_STRING(self) + start, | 
					
						
							|  |  |  |                 slicelength); | 
					
						
							|  |  |  |         } | 
					
						
							|  |  |  |         else { | 
					
						
							|  |  |  |             source_buf = PyBytes_AS_STRING(self); | 
					
						
							|  |  |  |             result = PyBytes_FromStringAndSize(NULL, slicelength); | 
					
						
							|  |  |  |             if (result == NULL) | 
					
						
							|  |  |  |                 return NULL; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |             result_buf = PyBytes_AS_STRING(result); | 
					
						
							|  |  |  |             for (cur = start, i = 0; i < slicelength; | 
					
						
							|  |  |  |                  cur += step, i++) { | 
					
						
							|  |  |  |                 result_buf[i] = source_buf[cur]; | 
					
						
							|  |  |  |             } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |             return result; | 
					
						
							|  |  |  |         } | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     else { | 
					
						
							|  |  |  |         PyErr_Format(PyExc_TypeError, | 
					
						
							|  |  |  |                      "byte indices must be integers, not %.200s", | 
					
						
							|  |  |  |                      Py_TYPE(item)->tp_name); | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  |     } | 
					
						
							| 
									
										
										
										
											2006-04-22 23:28:04 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2006-12-19 08:30:14 +00:00
										 |  |  | static int | 
					
						
							| 
									
										
										
										
											2009-04-18 15:17:02 +00:00
										 |  |  | bytes_buffer_getbuffer(PyBytesObject *self, Py_buffer *view, int flags) | 
					
						
							| 
									
										
										
										
											2006-12-19 08:30:14 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     return PyBuffer_FillInfo(view, (PyObject*)self, (void *)self->ob_sval, Py_SIZE(self), | 
					
						
							|  |  |  |                              1, flags); | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | } | 
					
						
							| 
									
										
										
										
											2007-04-12 22:49:52 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2009-04-18 15:17:02 +00:00
										 |  |  | static PySequenceMethods bytes_as_sequence = { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     (lenfunc)bytes_length, /*sq_length*/ | 
					
						
							|  |  |  |     (binaryfunc)bytes_concat, /*sq_concat*/ | 
					
						
							|  |  |  |     (ssizeargfunc)bytes_repeat, /*sq_repeat*/ | 
					
						
							|  |  |  |     (ssizeargfunc)bytes_item, /*sq_item*/ | 
					
						
							|  |  |  |     0,                  /*sq_slice*/ | 
					
						
							|  |  |  |     0,                  /*sq_ass_item*/ | 
					
						
							|  |  |  |     0,                  /*sq_ass_slice*/ | 
					
						
							|  |  |  |     (objobjproc)bytes_contains /*sq_contains*/ | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | }; | 
					
						
							| 
									
										
										
										
											2006-12-19 08:30:14 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2009-04-18 15:17:02 +00:00
										 |  |  | static PyMappingMethods bytes_as_mapping = { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     (lenfunc)bytes_length, | 
					
						
							|  |  |  |     (binaryfunc)bytes_subscript, | 
					
						
							|  |  |  |     0, | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | }; | 
					
						
							| 
									
										
										
										
											2006-12-19 08:30:14 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2009-04-18 15:17:02 +00:00
										 |  |  | static PyBufferProcs bytes_as_buffer = { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     (getbufferproc)bytes_buffer_getbuffer, | 
					
						
							|  |  |  |     NULL, | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | }; | 
					
						
							| 
									
										
										
										
											2006-12-19 08:30:14 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | #define LEFTSTRIP 0
 | 
					
						
							|  |  |  | #define RIGHTSTRIP 1
 | 
					
						
							|  |  |  | #define BOTHSTRIP 2
 | 
					
						
							| 
									
										
										
										
											2006-12-19 08:30:14 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | /* Arrays indexed by above */ | 
					
						
							|  |  |  | static const char *stripformat[] = {"|O:lstrip", "|O:rstrip", "|O:strip"}; | 
					
						
							| 
									
										
										
										
											2006-04-24 13:47:05 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | #define STRIPNAME(i) (stripformat[i]+3)
 | 
					
						
							| 
									
										
										
										
											2006-04-24 13:47:05 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | PyDoc_STRVAR(split__doc__, | 
					
						
							|  |  |  | "B.split([sep[, maxsplit]]) -> list of bytes\n\
 | 
					
						
							|  |  |  | \n\ | 
					
						
							|  |  |  | Return a list of the sections in B, using sep as the delimiter.\n\ | 
					
						
							|  |  |  | If sep is not specified or is None, B is split on ASCII whitespace\n\ | 
					
						
							|  |  |  | characters (space, tab, return, newline, formfeed, vertical tab).\n\ | 
					
						
							|  |  |  | If maxsplit is given, at most maxsplit splits are done."); | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | static PyObject * | 
					
						
							| 
									
										
										
										
											2009-04-18 15:17:02 +00:00
										 |  |  | bytes_split(PyBytesObject *self, PyObject *args) | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     Py_ssize_t len = PyBytes_GET_SIZE(self), n; | 
					
						
							|  |  |  |     Py_ssize_t maxsplit = -1; | 
					
						
							|  |  |  |     const char *s = PyBytes_AS_STRING(self), *sub; | 
					
						
							|  |  |  |     Py_buffer vsub; | 
					
						
							|  |  |  |     PyObject *list, *subobj = Py_None; | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     if (!PyArg_ParseTuple(args, "|On:split", &subobj, &maxsplit)) | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  |     if (maxsplit < 0) | 
					
						
							|  |  |  |         maxsplit = PY_SSIZE_T_MAX; | 
					
						
							|  |  |  |     if (subobj == Py_None) | 
					
						
							|  |  |  |         return stringlib_split_whitespace((PyObject*) self, s, len, maxsplit); | 
					
						
							|  |  |  |     if (_getbuffer(subobj, &vsub) < 0) | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  |     sub = vsub.buf; | 
					
						
							|  |  |  |     n = vsub.len; | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     list = stringlib_split((PyObject*) self, s, len, sub, n, maxsplit); | 
					
						
							|  |  |  |     PyBuffer_Release(&vsub); | 
					
						
							|  |  |  |     return list; | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | PyDoc_STRVAR(partition__doc__, | 
					
						
							|  |  |  | "B.partition(sep) -> (head, sep, tail)\n\
 | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | \n\ | 
					
						
							| 
									
										
										
										
											2008-05-30 08:20:09 +00:00
										 |  |  | Search for the separator sep in B, and return the part before it,\n\ | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | the separator itself, and the part after it.  If the separator is not\n\ | 
					
						
							|  |  |  | found, returns B and two empty bytes objects."); | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  | static PyObject * | 
					
						
							| 
									
										
										
										
											2009-04-18 15:17:02 +00:00
										 |  |  | bytes_partition(PyBytesObject *self, PyObject *sep_obj) | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     const char *sep; | 
					
						
							|  |  |  |     Py_ssize_t sep_len; | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     if (PyBytes_Check(sep_obj)) { | 
					
						
							|  |  |  |         sep = PyBytes_AS_STRING(sep_obj); | 
					
						
							|  |  |  |         sep_len = PyBytes_GET_SIZE(sep_obj); | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     else if (PyObject_AsCharBuffer(sep_obj, &sep, &sep_len)) | 
					
						
							|  |  |  |         return NULL; | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     return stringlib_partition( | 
					
						
							|  |  |  |         (PyObject*) self, | 
					
						
							|  |  |  |         PyBytes_AS_STRING(self), PyBytes_GET_SIZE(self), | 
					
						
							|  |  |  |         sep_obj, sep, sep_len | 
					
						
							|  |  |  |         ); | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | PyDoc_STRVAR(rpartition__doc__, | 
					
						
							| 
									
										
										
										
											2010-01-25 11:58:28 +00:00
										 |  |  | "B.rpartition(sep) -> (head, sep, tail)\n\
 | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | \n\ | 
					
						
							| 
									
										
										
										
											2008-05-30 08:20:09 +00:00
										 |  |  | Search for the separator sep in B, starting at the end of B,\n\ | 
					
						
							|  |  |  | and return the part before it, the separator itself, and the\n\ | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | part after it.  If the separator is not found, returns two empty\n\ | 
					
						
							|  |  |  | bytes objects and B."); | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  | static PyObject * | 
					
						
							| 
									
										
										
										
											2009-04-18 15:17:02 +00:00
										 |  |  | bytes_rpartition(PyBytesObject *self, PyObject *sep_obj) | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     const char *sep; | 
					
						
							|  |  |  |     Py_ssize_t sep_len; | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     if (PyBytes_Check(sep_obj)) { | 
					
						
							|  |  |  |         sep = PyBytes_AS_STRING(sep_obj); | 
					
						
							|  |  |  |         sep_len = PyBytes_GET_SIZE(sep_obj); | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     else if (PyObject_AsCharBuffer(sep_obj, &sep, &sep_len)) | 
					
						
							|  |  |  |         return NULL; | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     return stringlib_rpartition( | 
					
						
							|  |  |  |         (PyObject*) self, | 
					
						
							|  |  |  |         PyBytes_AS_STRING(self), PyBytes_GET_SIZE(self), | 
					
						
							|  |  |  |         sep_obj, sep, sep_len | 
					
						
							|  |  |  |         ); | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | PyDoc_STRVAR(rsplit__doc__, | 
					
						
							| 
									
										
										
										
											2008-05-27 00:36:20 +00:00
										 |  |  | "B.rsplit([sep[, maxsplit]]) -> list of bytes\n\
 | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | \n\ | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | Return a list of the sections in B, using sep as the delimiter,\n\ | 
					
						
							|  |  |  | starting at the end of B and working to the front.\n\ | 
					
						
							|  |  |  | If sep is not given, B is split on ASCII whitespace characters\n\ | 
					
						
							|  |  |  | (space, tab, return, newline, formfeed, vertical tab).\n\ | 
					
						
							|  |  |  | If maxsplit is given, at most maxsplit splits are done."); | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  | static PyObject * | 
					
						
							| 
									
										
										
										
											2009-04-18 15:17:02 +00:00
										 |  |  | bytes_rsplit(PyBytesObject *self, PyObject *args) | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     Py_ssize_t len = PyBytes_GET_SIZE(self), n; | 
					
						
							|  |  |  |     Py_ssize_t maxsplit = -1; | 
					
						
							|  |  |  |     const char *s = PyBytes_AS_STRING(self), *sub; | 
					
						
							|  |  |  |     Py_buffer vsub; | 
					
						
							|  |  |  |     PyObject *list, *subobj = Py_None; | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     if (!PyArg_ParseTuple(args, "|On:rsplit", &subobj, &maxsplit)) | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  |     if (maxsplit < 0) | 
					
						
							|  |  |  |         maxsplit = PY_SSIZE_T_MAX; | 
					
						
							|  |  |  |     if (subobj == Py_None) | 
					
						
							|  |  |  |         return stringlib_rsplit_whitespace((PyObject*) self, s, len, maxsplit); | 
					
						
							|  |  |  |     if (_getbuffer(subobj, &vsub) < 0) | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  |     sub = vsub.buf; | 
					
						
							|  |  |  |     n = vsub.len; | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     list = stringlib_rsplit((PyObject*) self, s, len, sub, n, maxsplit); | 
					
						
							|  |  |  |     PyBuffer_Release(&vsub); | 
					
						
							|  |  |  |     return list; | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  | PyDoc_STRVAR(join__doc__, | 
					
						
							|  |  |  | "B.join(iterable_of_bytes) -> bytes\n\
 | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | \n\ | 
					
						
							| 
									
										
										
										
											2008-05-30 08:20:09 +00:00
										 |  |  | Concatenate any number of bytes objects, with B in between each pair.\n\ | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | Example: b'.'.join([b'ab', b'pq', b'rs']) -> b'ab.pq.rs'."); | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  | static PyObject * | 
					
						
							| 
									
										
										
										
											2009-04-18 15:17:02 +00:00
										 |  |  | bytes_join(PyObject *self, PyObject *orig) | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     char *sep = PyBytes_AS_STRING(self); | 
					
						
							|  |  |  |     const Py_ssize_t seplen = PyBytes_GET_SIZE(self); | 
					
						
							|  |  |  |     PyObject *res = NULL; | 
					
						
							|  |  |  |     char *p; | 
					
						
							|  |  |  |     Py_ssize_t seqlen = 0; | 
					
						
							|  |  |  |     size_t sz = 0; | 
					
						
							|  |  |  |     Py_ssize_t i; | 
					
						
							|  |  |  |     PyObject *seq, *item; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     seq = PySequence_Fast(orig, ""); | 
					
						
							|  |  |  |     if (seq == NULL) { | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     seqlen = PySequence_Size(seq); | 
					
						
							|  |  |  |     if (seqlen == 0) { | 
					
						
							|  |  |  |         Py_DECREF(seq); | 
					
						
							|  |  |  |         return PyBytes_FromString(""); | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     if (seqlen == 1) { | 
					
						
							|  |  |  |         item = PySequence_Fast_GET_ITEM(seq, 0); | 
					
						
							|  |  |  |         if (PyBytes_CheckExact(item)) { | 
					
						
							|  |  |  |             Py_INCREF(item); | 
					
						
							|  |  |  |             Py_DECREF(seq); | 
					
						
							|  |  |  |             return item; | 
					
						
							|  |  |  |         } | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     /* There are at least two things to join, or else we have a subclass
 | 
					
						
							|  |  |  |      * of the builtin types in the sequence. | 
					
						
							|  |  |  |      * Do a pre-pass to figure out the total amount of space we'll | 
					
						
							|  |  |  |      * need (sz), and see whether all argument are bytes. | 
					
						
							|  |  |  |      */ | 
					
						
							|  |  |  |     /* XXX Shouldn't we use _getbuffer() on these items instead? */ | 
					
						
							|  |  |  |     for (i = 0; i < seqlen; i++) { | 
					
						
							|  |  |  |         const size_t old_sz = sz; | 
					
						
							|  |  |  |         item = PySequence_Fast_GET_ITEM(seq, i); | 
					
						
							|  |  |  |         if (!PyBytes_Check(item) && !PyByteArray_Check(item)) { | 
					
						
							|  |  |  |             PyErr_Format(PyExc_TypeError, | 
					
						
							|  |  |  |                          "sequence item %zd: expected bytes," | 
					
						
							|  |  |  |                          " %.80s found", | 
					
						
							|  |  |  |                          i, Py_TYPE(item)->tp_name); | 
					
						
							|  |  |  |             Py_DECREF(seq); | 
					
						
							|  |  |  |             return NULL; | 
					
						
							|  |  |  |         } | 
					
						
							|  |  |  |         sz += Py_SIZE(item); | 
					
						
							|  |  |  |         if (i != 0) | 
					
						
							|  |  |  |             sz += seplen; | 
					
						
							|  |  |  |         if (sz < old_sz || sz > PY_SSIZE_T_MAX) { | 
					
						
							|  |  |  |             PyErr_SetString(PyExc_OverflowError, | 
					
						
							|  |  |  |                 "join() result is too long for bytes"); | 
					
						
							|  |  |  |             Py_DECREF(seq); | 
					
						
							|  |  |  |             return NULL; | 
					
						
							|  |  |  |         } | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     /* Allocate result space. */ | 
					
						
							|  |  |  |     res = PyBytes_FromStringAndSize((char*)NULL, sz); | 
					
						
							|  |  |  |     if (res == NULL) { | 
					
						
							|  |  |  |         Py_DECREF(seq); | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     /* Catenate everything. */ | 
					
						
							|  |  |  |     /* I'm not worried about a PyByteArray item growing because there's
 | 
					
						
							|  |  |  |        nowhere in this function where we release the GIL. */ | 
					
						
							|  |  |  |     p = PyBytes_AS_STRING(res); | 
					
						
							|  |  |  |     for (i = 0; i < seqlen; ++i) { | 
					
						
							|  |  |  |         size_t n; | 
					
						
							|  |  |  |         char *q; | 
					
						
							|  |  |  |         if (i) { | 
					
						
							|  |  |  |             Py_MEMCPY(p, sep, seplen); | 
					
						
							|  |  |  |             p += seplen; | 
					
						
							|  |  |  |         } | 
					
						
							|  |  |  |         item = PySequence_Fast_GET_ITEM(seq, i); | 
					
						
							|  |  |  |         n = Py_SIZE(item); | 
					
						
							|  |  |  |         if (PyBytes_Check(item)) | 
					
						
							|  |  |  |             q = PyBytes_AS_STRING(item); | 
					
						
							|  |  |  |         else | 
					
						
							|  |  |  |             q = PyByteArray_AS_STRING(item); | 
					
						
							|  |  |  |         Py_MEMCPY(p, q, n); | 
					
						
							|  |  |  |         p += n; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     Py_DECREF(seq); | 
					
						
							|  |  |  |     return res; | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | PyObject * | 
					
						
							|  |  |  | _PyBytes_Join(PyObject *sep, PyObject *x) | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     assert(sep != NULL && PyBytes_Check(sep)); | 
					
						
							|  |  |  |     assert(x != NULL); | 
					
						
							|  |  |  |     return bytes_join(sep, x); | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2010-01-13 08:07:53 +00:00
										 |  |  | /* helper macro to fixup start/end slice values */ | 
					
						
							|  |  |  | #define ADJUST_INDICES(start, end, len)         \
 | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     if (end > len)                          \ | 
					
						
							|  |  |  |         end = len;                          \ | 
					
						
							|  |  |  |     else if (end < 0) {                     \ | 
					
						
							|  |  |  |         end += len;                         \ | 
					
						
							|  |  |  |         if (end < 0)                        \ | 
					
						
							|  |  |  |         end = 0;                        \ | 
					
						
							|  |  |  |     }                                       \ | 
					
						
							|  |  |  |     if (start < 0) {                        \ | 
					
						
							|  |  |  |         start += len;                       \ | 
					
						
							|  |  |  |         if (start < 0)                      \ | 
					
						
							|  |  |  |         start = 0;                      \ | 
					
						
							|  |  |  |     } | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  | Py_LOCAL_INLINE(Py_ssize_t) | 
					
						
							| 
									
										
										
										
											2009-04-18 15:17:02 +00:00
										 |  |  | bytes_find_internal(PyBytesObject *self, PyObject *args, int dir) | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     PyObject *subobj; | 
					
						
							|  |  |  |     const char *sub; | 
					
						
							|  |  |  |     Py_ssize_t sub_len; | 
					
						
							|  |  |  |     Py_ssize_t start=0, end=PY_SSIZE_T_MAX; | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2011-04-20 17:09:23 +02:00
										 |  |  |     if (!stringlib_parse_args_finds("find/rfind/index/rindex", | 
					
						
							|  |  |  |                                     args, &subobj, &start, &end)) | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |         return -2; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     if (PyBytes_Check(subobj)) { | 
					
						
							|  |  |  |         sub = PyBytes_AS_STRING(subobj); | 
					
						
							|  |  |  |         sub_len = PyBytes_GET_SIZE(subobj); | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     else if (PyObject_AsCharBuffer(subobj, &sub, &sub_len)) | 
					
						
							|  |  |  |         /* XXX - the "expected a character buffer object" is pretty
 | 
					
						
							|  |  |  |            confusing for a non-expert.  remap to something else ? */ | 
					
						
							|  |  |  |         return -2; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     if (dir > 0) | 
					
						
							|  |  |  |         return stringlib_find_slice( | 
					
						
							|  |  |  |             PyBytes_AS_STRING(self), PyBytes_GET_SIZE(self), | 
					
						
							|  |  |  |             sub, sub_len, start, end); | 
					
						
							|  |  |  |     else | 
					
						
							|  |  |  |         return stringlib_rfind_slice( | 
					
						
							|  |  |  |             PyBytes_AS_STRING(self), PyBytes_GET_SIZE(self), | 
					
						
							|  |  |  |             sub, sub_len, start, end); | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | PyDoc_STRVAR(find__doc__, | 
					
						
							| 
									
										
										
										
											2008-05-30 08:20:09 +00:00
										 |  |  | "B.find(sub[, start[, end]]) -> int\n\
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | \n\ | 
					
						
							| 
									
										
										
										
											2010-01-13 08:07:53 +00:00
										 |  |  | Return the lowest index in B where substring sub is found,\n\ | 
					
						
							| 
									
										
										
										
											2011-07-27 23:33:54 +08:00
										 |  |  | such that sub is contained within B[start:end].  Optional\n\ | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | arguments start and end are interpreted as in slice notation.\n\ | 
					
						
							|  |  |  | \n\ | 
					
						
							|  |  |  | Return -1 on failure."); | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | static PyObject * | 
					
						
							| 
									
										
										
										
											2009-04-18 15:17:02 +00:00
										 |  |  | bytes_find(PyBytesObject *self, PyObject *args) | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     Py_ssize_t result = bytes_find_internal(self, args, +1); | 
					
						
							|  |  |  |     if (result == -2) | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  |     return PyLong_FromSsize_t(result); | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | PyDoc_STRVAR(index__doc__, | 
					
						
							| 
									
										
										
										
											2008-05-30 08:20:09 +00:00
										 |  |  | "B.index(sub[, start[, end]]) -> int\n\
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | \n\ | 
					
						
							|  |  |  | Like B.find() but raise ValueError when the substring is not found."); | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | static PyObject * | 
					
						
							| 
									
										
										
										
											2009-04-18 15:17:02 +00:00
										 |  |  | bytes_index(PyBytesObject *self, PyObject *args) | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     Py_ssize_t result = bytes_find_internal(self, args, +1); | 
					
						
							|  |  |  |     if (result == -2) | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  |     if (result == -1) { | 
					
						
							|  |  |  |         PyErr_SetString(PyExc_ValueError, | 
					
						
							|  |  |  |                         "substring not found"); | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     return PyLong_FromSsize_t(result); | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | PyDoc_STRVAR(rfind__doc__, | 
					
						
							| 
									
										
										
										
											2008-05-30 08:20:09 +00:00
										 |  |  | "B.rfind(sub[, start[, end]]) -> int\n\
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | \n\ | 
					
						
							|  |  |  | Return the highest index in B where substring sub is found,\n\ | 
					
						
							| 
									
										
										
										
											2011-07-27 23:33:54 +08:00
										 |  |  | such that sub is contained within B[start:end].  Optional\n\ | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | arguments start and end are interpreted as in slice notation.\n\ | 
					
						
							|  |  |  | \n\ | 
					
						
							|  |  |  | Return -1 on failure."); | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | static PyObject * | 
					
						
							| 
									
										
										
										
											2009-04-18 15:17:02 +00:00
										 |  |  | bytes_rfind(PyBytesObject *self, PyObject *args) | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     Py_ssize_t result = bytes_find_internal(self, args, -1); | 
					
						
							|  |  |  |     if (result == -2) | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  |     return PyLong_FromSsize_t(result); | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | PyDoc_STRVAR(rindex__doc__, | 
					
						
							| 
									
										
										
										
											2008-05-30 08:20:09 +00:00
										 |  |  | "B.rindex(sub[, start[, end]]) -> int\n\
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | \n\ | 
					
						
							|  |  |  | Like B.rfind() but raise ValueError when the substring is not found."); | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | static PyObject * | 
					
						
							| 
									
										
										
										
											2009-04-18 15:17:02 +00:00
										 |  |  | bytes_rindex(PyBytesObject *self, PyObject *args) | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     Py_ssize_t result = bytes_find_internal(self, args, -1); | 
					
						
							|  |  |  |     if (result == -2) | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  |     if (result == -1) { | 
					
						
							|  |  |  |         PyErr_SetString(PyExc_ValueError, | 
					
						
							|  |  |  |                         "substring not found"); | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     return PyLong_FromSsize_t(result); | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | } | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | Py_LOCAL_INLINE(PyObject *) | 
					
						
							|  |  |  | do_xstrip(PyBytesObject *self, int striptype, PyObject *sepobj) | 
					
						
							|  |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     Py_buffer vsep; | 
					
						
							|  |  |  |     char *s = PyBytes_AS_STRING(self); | 
					
						
							|  |  |  |     Py_ssize_t len = PyBytes_GET_SIZE(self); | 
					
						
							|  |  |  |     char *sep; | 
					
						
							|  |  |  |     Py_ssize_t seplen; | 
					
						
							|  |  |  |     Py_ssize_t i, j; | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     if (_getbuffer(sepobj, &vsep) < 0) | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  |     sep = vsep.buf; | 
					
						
							|  |  |  |     seplen = vsep.len; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     i = 0; | 
					
						
							|  |  |  |     if (striptype != RIGHTSTRIP) { | 
					
						
							|  |  |  |         while (i < len && memchr(sep, Py_CHARMASK(s[i]), seplen)) { | 
					
						
							|  |  |  |             i++; | 
					
						
							|  |  |  |         } | 
					
						
							|  |  |  |     } | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     j = len; | 
					
						
							|  |  |  |     if (striptype != LEFTSTRIP) { | 
					
						
							|  |  |  |         do { | 
					
						
							|  |  |  |             j--; | 
					
						
							|  |  |  |         } while (j >= i && memchr(sep, Py_CHARMASK(s[j]), seplen)); | 
					
						
							|  |  |  |         j++; | 
					
						
							|  |  |  |     } | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     PyBuffer_Release(&vsep); | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     if (i == 0 && j == len && PyBytes_CheckExact(self)) { | 
					
						
							|  |  |  |         Py_INCREF(self); | 
					
						
							|  |  |  |         return (PyObject*)self; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     else | 
					
						
							|  |  |  |         return PyBytes_FromStringAndSize(s+i, j-i); | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | Py_LOCAL_INLINE(PyObject *) | 
					
						
							|  |  |  | do_strip(PyBytesObject *self, int striptype) | 
					
						
							|  |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     char *s = PyBytes_AS_STRING(self); | 
					
						
							|  |  |  |     Py_ssize_t len = PyBytes_GET_SIZE(self), i, j; | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     i = 0; | 
					
						
							|  |  |  |     if (striptype != RIGHTSTRIP) { | 
					
						
							| 
									
										
										
										
											2010-11-05 17:23:41 +00:00
										 |  |  |         while (i < len && Py_ISSPACE(s[i])) { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |             i++; | 
					
						
							|  |  |  |         } | 
					
						
							|  |  |  |     } | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     j = len; | 
					
						
							|  |  |  |     if (striptype != LEFTSTRIP) { | 
					
						
							|  |  |  |         do { | 
					
						
							|  |  |  |             j--; | 
					
						
							| 
									
										
										
										
											2010-11-05 17:23:41 +00:00
										 |  |  |         } while (j >= i && Py_ISSPACE(s[j])); | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |         j++; | 
					
						
							|  |  |  |     } | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     if (i == 0 && j == len && PyBytes_CheckExact(self)) { | 
					
						
							|  |  |  |         Py_INCREF(self); | 
					
						
							|  |  |  |         return (PyObject*)self; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     else | 
					
						
							|  |  |  |         return PyBytes_FromStringAndSize(s+i, j-i); | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | Py_LOCAL_INLINE(PyObject *) | 
					
						
							|  |  |  | do_argstrip(PyBytesObject *self, int striptype, PyObject *args) | 
					
						
							|  |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     PyObject *sep = NULL; | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     if (!PyArg_ParseTuple(args, (char *)stripformat[striptype], &sep)) | 
					
						
							|  |  |  |         return NULL; | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     if (sep != NULL && sep != Py_None) { | 
					
						
							|  |  |  |         return do_xstrip(self, striptype, sep); | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     return do_strip(self, striptype); | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | PyDoc_STRVAR(strip__doc__, | 
					
						
							|  |  |  | "B.strip([bytes]) -> bytes\n\
 | 
					
						
							|  |  |  | \n\ | 
					
						
							|  |  |  | Strip leading and trailing bytes contained in the argument.\n\ | 
					
						
							|  |  |  | If the argument is omitted, strip trailing ASCII whitespace."); | 
					
						
							|  |  |  | static PyObject * | 
					
						
							| 
									
										
										
										
											2009-04-18 15:17:02 +00:00
										 |  |  | bytes_strip(PyBytesObject *self, PyObject *args) | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     if (PyTuple_GET_SIZE(args) == 0) | 
					
						
							|  |  |  |         return do_strip(self, BOTHSTRIP); /* Common case */ | 
					
						
							|  |  |  |     else | 
					
						
							|  |  |  |         return do_argstrip(self, BOTHSTRIP, args); | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | } | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | PyDoc_STRVAR(lstrip__doc__, | 
					
						
							|  |  |  | "B.lstrip([bytes]) -> bytes\n\
 | 
					
						
							|  |  |  | \n\ | 
					
						
							|  |  |  | Strip leading bytes contained in the argument.\n\ | 
					
						
							|  |  |  | If the argument is omitted, strip leading ASCII whitespace."); | 
					
						
							|  |  |  | static PyObject * | 
					
						
							| 
									
										
										
										
											2009-04-18 15:17:02 +00:00
										 |  |  | bytes_lstrip(PyBytesObject *self, PyObject *args) | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     if (PyTuple_GET_SIZE(args) == 0) | 
					
						
							|  |  |  |         return do_strip(self, LEFTSTRIP); /* Common case */ | 
					
						
							|  |  |  |     else | 
					
						
							|  |  |  |         return do_argstrip(self, LEFTSTRIP, args); | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | } | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | PyDoc_STRVAR(rstrip__doc__, | 
					
						
							|  |  |  | "B.rstrip([bytes]) -> bytes\n\
 | 
					
						
							|  |  |  | \n\ | 
					
						
							|  |  |  | Strip trailing bytes contained in the argument.\n\ | 
					
						
							|  |  |  | If the argument is omitted, strip trailing ASCII whitespace."); | 
					
						
							|  |  |  | static PyObject * | 
					
						
							| 
									
										
										
										
											2009-04-18 15:17:02 +00:00
										 |  |  | bytes_rstrip(PyBytesObject *self, PyObject *args) | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     if (PyTuple_GET_SIZE(args) == 0) | 
					
						
							|  |  |  |         return do_strip(self, RIGHTSTRIP); /* Common case */ | 
					
						
							|  |  |  |     else | 
					
						
							|  |  |  |         return do_argstrip(self, RIGHTSTRIP, args); | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2007-10-16 06:31:30 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | PyDoc_STRVAR(count__doc__, | 
					
						
							| 
									
										
										
										
											2008-05-30 08:20:09 +00:00
										 |  |  | "B.count(sub[, start[, end]]) -> int\n\
 | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | \n\ | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | Return the number of non-overlapping occurrences of substring sub in\n\ | 
					
						
							| 
									
										
										
										
											2010-01-13 08:07:53 +00:00
										 |  |  | string B[start:end].  Optional arguments start and end are interpreted\n\ | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | as in slice notation."); | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  | static PyObject * | 
					
						
							| 
									
										
										
										
											2009-04-18 15:17:02 +00:00
										 |  |  | bytes_count(PyBytesObject *self, PyObject *args) | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     PyObject *sub_obj; | 
					
						
							|  |  |  |     const char *str = PyBytes_AS_STRING(self), *sub; | 
					
						
							|  |  |  |     Py_ssize_t sub_len; | 
					
						
							|  |  |  |     Py_ssize_t start = 0, end = PY_SSIZE_T_MAX; | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2011-04-20 17:09:23 +02:00
										 |  |  |     if (!stringlib_parse_args_finds("count", args, &sub_obj, &start, &end)) | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |         return NULL; | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     if (PyBytes_Check(sub_obj)) { | 
					
						
							|  |  |  |         sub = PyBytes_AS_STRING(sub_obj); | 
					
						
							|  |  |  |         sub_len = PyBytes_GET_SIZE(sub_obj); | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     else if (PyObject_AsCharBuffer(sub_obj, &sub, &sub_len)) | 
					
						
							|  |  |  |         return NULL; | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     ADJUST_INDICES(start, end, PyBytes_GET_SIZE(self)); | 
					
						
							| 
									
										
										
										
											2007-08-18 11:21:56 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     return PyLong_FromSsize_t( | 
					
						
							|  |  |  |         stringlib_count(str + start, end - start, sub, sub_len, PY_SSIZE_T_MAX) | 
					
						
							|  |  |  |         ); | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | PyDoc_STRVAR(translate__doc__, | 
					
						
							|  |  |  | "B.translate(table[, deletechars]) -> bytes\n\
 | 
					
						
							|  |  |  | \n\ | 
					
						
							|  |  |  | Return a copy of B, where all characters occurring in the\n\ | 
					
						
							|  |  |  | optional argument deletechars are removed, and the remaining\n\ | 
					
						
							|  |  |  | characters have been mapped through the given translation\n\ | 
					
						
							|  |  |  | table, which must be a bytes object of length 256."); | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | static PyObject * | 
					
						
							| 
									
										
										
										
											2009-04-18 15:17:02 +00:00
										 |  |  | bytes_translate(PyBytesObject *self, PyObject *args) | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     register char *input, *output; | 
					
						
							|  |  |  |     const char *table; | 
					
						
							|  |  |  |     register Py_ssize_t i, c, changed = 0; | 
					
						
							|  |  |  |     PyObject *input_obj = (PyObject*)self; | 
					
						
							|  |  |  |     const char *output_start, *del_table=NULL; | 
					
						
							|  |  |  |     Py_ssize_t inlen, tablen, dellen = 0; | 
					
						
							|  |  |  |     PyObject *result; | 
					
						
							|  |  |  |     int trans_table[256]; | 
					
						
							|  |  |  |     PyObject *tableobj, *delobj = NULL; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     if (!PyArg_UnpackTuple(args, "translate", 1, 2, | 
					
						
							|  |  |  |                           &tableobj, &delobj)) | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     if (PyBytes_Check(tableobj)) { | 
					
						
							|  |  |  |         table = PyBytes_AS_STRING(tableobj); | 
					
						
							|  |  |  |         tablen = PyBytes_GET_SIZE(tableobj); | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     else if (tableobj == Py_None) { | 
					
						
							|  |  |  |         table = NULL; | 
					
						
							|  |  |  |         tablen = 256; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     else if (PyObject_AsCharBuffer(tableobj, &table, &tablen)) | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     if (tablen != 256) { | 
					
						
							|  |  |  |         PyErr_SetString(PyExc_ValueError, | 
					
						
							|  |  |  |           "translation table must be 256 characters long"); | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     if (delobj != NULL) { | 
					
						
							|  |  |  |         if (PyBytes_Check(delobj)) { | 
					
						
							|  |  |  |             del_table = PyBytes_AS_STRING(delobj); | 
					
						
							|  |  |  |             dellen = PyBytes_GET_SIZE(delobj); | 
					
						
							|  |  |  |         } | 
					
						
							|  |  |  |         else if (PyObject_AsCharBuffer(delobj, &del_table, &dellen)) | 
					
						
							|  |  |  |             return NULL; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     else { | 
					
						
							|  |  |  |         del_table = NULL; | 
					
						
							|  |  |  |         dellen = 0; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     inlen = PyBytes_GET_SIZE(input_obj); | 
					
						
							|  |  |  |     result = PyBytes_FromStringAndSize((char *)NULL, inlen); | 
					
						
							|  |  |  |     if (result == NULL) | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  |     output_start = output = PyBytes_AsString(result); | 
					
						
							|  |  |  |     input = PyBytes_AS_STRING(input_obj); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     if (dellen == 0 && table != NULL) { | 
					
						
							|  |  |  |         /* If no deletions are required, use faster code */ | 
					
						
							|  |  |  |         for (i = inlen; --i >= 0; ) { | 
					
						
							|  |  |  |             c = Py_CHARMASK(*input++); | 
					
						
							|  |  |  |             if (Py_CHARMASK((*output++ = table[c])) != c) | 
					
						
							|  |  |  |                 changed = 1; | 
					
						
							|  |  |  |         } | 
					
						
							|  |  |  |         if (changed || !PyBytes_CheckExact(input_obj)) | 
					
						
							|  |  |  |             return result; | 
					
						
							|  |  |  |         Py_DECREF(result); | 
					
						
							|  |  |  |         Py_INCREF(input_obj); | 
					
						
							|  |  |  |         return input_obj; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     if (table == NULL) { | 
					
						
							|  |  |  |         for (i = 0; i < 256; i++) | 
					
						
							|  |  |  |             trans_table[i] = Py_CHARMASK(i); | 
					
						
							|  |  |  |     } else { | 
					
						
							|  |  |  |         for (i = 0; i < 256; i++) | 
					
						
							|  |  |  |             trans_table[i] = Py_CHARMASK(table[i]); | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     for (i = 0; i < dellen; i++) | 
					
						
							|  |  |  |         trans_table[(int) Py_CHARMASK(del_table[i])] = -1; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     for (i = inlen; --i >= 0; ) { | 
					
						
							|  |  |  |         c = Py_CHARMASK(*input++); | 
					
						
							|  |  |  |         if (trans_table[c] != -1) | 
					
						
							|  |  |  |             if (Py_CHARMASK(*output++ = (char)trans_table[c]) == c) | 
					
						
							|  |  |  |                 continue; | 
					
						
							|  |  |  |         changed = 1; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     if (!changed && PyBytes_CheckExact(input_obj)) { | 
					
						
							|  |  |  |         Py_DECREF(result); | 
					
						
							|  |  |  |         Py_INCREF(input_obj); | 
					
						
							|  |  |  |         return input_obj; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     /* Fix the size of the resulting string */ | 
					
						
							|  |  |  |     if (inlen > 0) | 
					
						
							|  |  |  |         _PyBytes_Resize(&result, output - output_start); | 
					
						
							|  |  |  |     return result; | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2007-09-10 16:53:45 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2009-04-12 15:51:51 +00:00
										 |  |  | static PyObject * | 
					
						
							| 
									
										
										
										
											2009-04-18 15:17:02 +00:00
										 |  |  | bytes_maketrans(PyObject *null, PyObject *args) | 
					
						
							| 
									
										
										
										
											2009-04-12 15:51:51 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     return _Py_bytes_maketrans(args); | 
					
						
							| 
									
										
										
										
											2009-04-12 15:51:51 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | /* find and count characters and substrings */ | 
					
						
							| 
									
										
										
										
											2007-09-10 16:53:45 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  | #define findchar(target, target_len, c)                         \
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  |   ((char *)memchr((const void *)(target), c, target_len)) | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | /* String ops must return a string.  */ | 
					
						
							|  |  |  | /* If the object is subclass of string, create a copy */ | 
					
						
							|  |  |  | Py_LOCAL(PyBytesObject *) | 
					
						
							|  |  |  | return_self(PyBytesObject *self) | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     if (PyBytes_CheckExact(self)) { | 
					
						
							|  |  |  |         Py_INCREF(self); | 
					
						
							|  |  |  |         return self; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     return (PyBytesObject *)PyBytes_FromStringAndSize( | 
					
						
							|  |  |  |         PyBytes_AS_STRING(self), | 
					
						
							|  |  |  |         PyBytes_GET_SIZE(self)); | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | } | 
					
						
							| 
									
										
										
										
											2007-09-10 16:53:45 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | Py_LOCAL_INLINE(Py_ssize_t) | 
					
						
							| 
									
										
										
										
											2010-08-15 17:12:55 +00:00
										 |  |  | countchar(const char *target, Py_ssize_t target_len, char c, Py_ssize_t maxcount) | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     Py_ssize_t count=0; | 
					
						
							|  |  |  |     const char *start=target; | 
					
						
							|  |  |  |     const char *end=target+target_len; | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     while ( (start=findchar(start, end-start, c)) != NULL ) { | 
					
						
							|  |  |  |         count++; | 
					
						
							|  |  |  |         if (count >= maxcount) | 
					
						
							|  |  |  |             break; | 
					
						
							|  |  |  |         start += 1; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     return count; | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | } | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | /* Algorithms for different cases of string replacement */ | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | /* len(self)>=1, from="", len(to)>=1, maxcount>=1 */ | 
					
						
							|  |  |  | Py_LOCAL(PyBytesObject *) | 
					
						
							|  |  |  | replace_interleave(PyBytesObject *self, | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |                    const char *to_s, Py_ssize_t to_len, | 
					
						
							|  |  |  |                    Py_ssize_t maxcount) | 
					
						
							|  |  |  | { | 
					
						
							|  |  |  |     char *self_s, *result_s; | 
					
						
							|  |  |  |     Py_ssize_t self_len, result_len; | 
					
						
							| 
									
										
										
										
											2010-08-10 18:35:01 +00:00
										 |  |  |     Py_ssize_t count, i; | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     PyBytesObject *result; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     self_len = PyBytes_GET_SIZE(self); | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2010-08-10 18:35:01 +00:00
										 |  |  |     /* 1 at the end plus 1 after every character;
 | 
					
						
							|  |  |  |        count = min(maxcount, self_len + 1) */ | 
					
						
							|  |  |  |     if (maxcount <= self_len) | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |         count = maxcount; | 
					
						
							| 
									
										
										
										
											2010-08-10 18:35:01 +00:00
										 |  |  |     else | 
					
						
							|  |  |  |         /* Can't overflow: self_len + 1 <= maxcount <= PY_SSIZE_T_MAX. */ | 
					
						
							|  |  |  |         count = self_len + 1; | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  |     /* Check for overflow */ | 
					
						
							|  |  |  |     /*   result_len = count * to_len + self_len; */ | 
					
						
							| 
									
										
										
										
											2010-08-10 18:35:01 +00:00
										 |  |  |     assert(count > 0); | 
					
						
							|  |  |  |     if (to_len > (PY_SSIZE_T_MAX - self_len) / count) { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |         PyErr_SetString(PyExc_OverflowError, | 
					
						
							|  |  |  |                         "replacement bytes are too long"); | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  |     } | 
					
						
							| 
									
										
										
										
											2010-08-10 18:35:01 +00:00
										 |  |  |     result_len = count * to_len + self_len; | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  |     if (! (result = (PyBytesObject *) | 
					
						
							|  |  |  |                      PyBytes_FromStringAndSize(NULL, result_len)) ) | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     self_s = PyBytes_AS_STRING(self); | 
					
						
							|  |  |  |     result_s = PyBytes_AS_STRING(result); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     /* TODO: special case single character, which doesn't need memcpy */ | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     /* Lay the first one down (guaranteed this will occur) */ | 
					
						
							|  |  |  |     Py_MEMCPY(result_s, to_s, to_len); | 
					
						
							|  |  |  |     result_s += to_len; | 
					
						
							|  |  |  |     count -= 1; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     for (i=0; i<count; i++) { | 
					
						
							|  |  |  |         *result_s++ = *self_s++; | 
					
						
							|  |  |  |         Py_MEMCPY(result_s, to_s, to_len); | 
					
						
							|  |  |  |         result_s += to_len; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     /* Copy the rest of the original string */ | 
					
						
							|  |  |  |     Py_MEMCPY(result_s, self_s, self_len-i); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     return result; | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | /* Special case for deleting a single character */ | 
					
						
							|  |  |  | /* len(self)>=1, len(from)==1, to="", maxcount>=1 */ | 
					
						
							|  |  |  | Py_LOCAL(PyBytesObject *) | 
					
						
							|  |  |  | replace_delete_single_character(PyBytesObject *self, | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |                                 char from_c, Py_ssize_t maxcount) | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     char *self_s, *result_s; | 
					
						
							|  |  |  |     char *start, *next, *end; | 
					
						
							|  |  |  |     Py_ssize_t self_len, result_len; | 
					
						
							|  |  |  |     Py_ssize_t count; | 
					
						
							|  |  |  |     PyBytesObject *result; | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     self_len = PyBytes_GET_SIZE(self); | 
					
						
							|  |  |  |     self_s = PyBytes_AS_STRING(self); | 
					
						
							| 
									
										
										
										
											2007-09-10 16:53:45 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     count = countchar(self_s, self_len, from_c, maxcount); | 
					
						
							|  |  |  |     if (count == 0) { | 
					
						
							|  |  |  |         return return_self(self); | 
					
						
							|  |  |  |     } | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     result_len = self_len - count;  /* from_len == 1 */ | 
					
						
							|  |  |  |     assert(result_len>=0); | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     if ( (result = (PyBytesObject *) | 
					
						
							|  |  |  |                     PyBytes_FromStringAndSize(NULL, result_len)) == NULL) | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  |     result_s = PyBytes_AS_STRING(result); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     start = self_s; | 
					
						
							|  |  |  |     end = self_s + self_len; | 
					
						
							|  |  |  |     while (count-- > 0) { | 
					
						
							|  |  |  |         next = findchar(start, end-start, from_c); | 
					
						
							|  |  |  |         if (next == NULL) | 
					
						
							|  |  |  |             break; | 
					
						
							|  |  |  |         Py_MEMCPY(result_s, start, next-start); | 
					
						
							|  |  |  |         result_s += (next-start); | 
					
						
							|  |  |  |         start = next+1; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     Py_MEMCPY(result_s, start, end-start); | 
					
						
							| 
									
										
										
										
											2007-09-10 16:53:45 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     return result; | 
					
						
							| 
									
										
										
										
											2007-09-10 16:53:45 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | /* len(self)>=1, len(from)>=2, to="", maxcount>=1 */ | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | Py_LOCAL(PyBytesObject *) | 
					
						
							|  |  |  | replace_delete_substring(PyBytesObject *self, | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |                          const char *from_s, Py_ssize_t from_len, | 
					
						
							|  |  |  |                          Py_ssize_t maxcount) { | 
					
						
							|  |  |  |     char *self_s, *result_s; | 
					
						
							|  |  |  |     char *start, *next, *end; | 
					
						
							|  |  |  |     Py_ssize_t self_len, result_len; | 
					
						
							|  |  |  |     Py_ssize_t count, offset; | 
					
						
							|  |  |  |     PyBytesObject *result; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     self_len = PyBytes_GET_SIZE(self); | 
					
						
							|  |  |  |     self_s = PyBytes_AS_STRING(self); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     count = stringlib_count(self_s, self_len, | 
					
						
							|  |  |  |                             from_s, from_len, | 
					
						
							|  |  |  |                             maxcount); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     if (count == 0) { | 
					
						
							|  |  |  |         /* no matches */ | 
					
						
							|  |  |  |         return return_self(self); | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     result_len = self_len - (count * from_len); | 
					
						
							|  |  |  |     assert (result_len>=0); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     if ( (result = (PyBytesObject *) | 
					
						
							|  |  |  |           PyBytes_FromStringAndSize(NULL, result_len)) == NULL ) | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     result_s = PyBytes_AS_STRING(result); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     start = self_s; | 
					
						
							|  |  |  |     end = self_s + self_len; | 
					
						
							|  |  |  |     while (count-- > 0) { | 
					
						
							|  |  |  |         offset = stringlib_find(start, end-start, | 
					
						
							|  |  |  |                                 from_s, from_len, | 
					
						
							|  |  |  |                                 0); | 
					
						
							|  |  |  |         if (offset == -1) | 
					
						
							|  |  |  |             break; | 
					
						
							|  |  |  |         next = start + offset; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         Py_MEMCPY(result_s, start, next-start); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         result_s += (next-start); | 
					
						
							|  |  |  |         start = next+from_len; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     Py_MEMCPY(result_s, start, end-start); | 
					
						
							|  |  |  |     return result; | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | /* len(self)>=1, len(from)==len(to)==1, maxcount>=1 */ | 
					
						
							|  |  |  | Py_LOCAL(PyBytesObject *) | 
					
						
							|  |  |  | replace_single_character_in_place(PyBytesObject *self, | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |                                   char from_c, char to_c, | 
					
						
							|  |  |  |                                   Py_ssize_t maxcount) | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     char *self_s, *result_s, *start, *end, *next; | 
					
						
							|  |  |  |     Py_ssize_t self_len; | 
					
						
							|  |  |  |     PyBytesObject *result; | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     /* The result string will be the same size */ | 
					
						
							|  |  |  |     self_s = PyBytes_AS_STRING(self); | 
					
						
							|  |  |  |     self_len = PyBytes_GET_SIZE(self); | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     next = findchar(self_s, self_len, from_c); | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     if (next == NULL) { | 
					
						
							|  |  |  |         /* No matches; return the original string */ | 
					
						
							|  |  |  |         return return_self(self); | 
					
						
							|  |  |  |     } | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     /* Need to make a new string */ | 
					
						
							|  |  |  |     result = (PyBytesObject *) PyBytes_FromStringAndSize(NULL, self_len); | 
					
						
							|  |  |  |     if (result == NULL) | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  |     result_s = PyBytes_AS_STRING(result); | 
					
						
							|  |  |  |     Py_MEMCPY(result_s, self_s, self_len); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     /* change everything in-place, starting with this one */ | 
					
						
							|  |  |  |     start =  result_s + (next-self_s); | 
					
						
							|  |  |  |     *start = to_c; | 
					
						
							|  |  |  |     start++; | 
					
						
							|  |  |  |     end = result_s + self_len; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     while (--maxcount > 0) { | 
					
						
							|  |  |  |         next = findchar(start, end-start, from_c); | 
					
						
							|  |  |  |         if (next == NULL) | 
					
						
							|  |  |  |             break; | 
					
						
							|  |  |  |         *next = to_c; | 
					
						
							|  |  |  |         start = next+1; | 
					
						
							|  |  |  |     } | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     return result; | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | } | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | /* len(self)>=1, len(from)==len(to)>=2, maxcount>=1 */ | 
					
						
							|  |  |  | Py_LOCAL(PyBytesObject *) | 
					
						
							|  |  |  | replace_substring_in_place(PyBytesObject *self, | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |                            const char *from_s, Py_ssize_t from_len, | 
					
						
							|  |  |  |                            const char *to_s, Py_ssize_t to_len, | 
					
						
							|  |  |  |                            Py_ssize_t maxcount) | 
					
						
							|  |  |  | { | 
					
						
							|  |  |  |     char *result_s, *start, *end; | 
					
						
							|  |  |  |     char *self_s; | 
					
						
							|  |  |  |     Py_ssize_t self_len, offset; | 
					
						
							|  |  |  |     PyBytesObject *result; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     /* The result string will be the same size */ | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     self_s = PyBytes_AS_STRING(self); | 
					
						
							|  |  |  |     self_len = PyBytes_GET_SIZE(self); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     offset = stringlib_find(self_s, self_len, | 
					
						
							|  |  |  |                             from_s, from_len, | 
					
						
							|  |  |  |                             0); | 
					
						
							|  |  |  |     if (offset == -1) { | 
					
						
							|  |  |  |         /* No matches; return the original string */ | 
					
						
							|  |  |  |         return return_self(self); | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     /* Need to make a new string */ | 
					
						
							|  |  |  |     result = (PyBytesObject *) PyBytes_FromStringAndSize(NULL, self_len); | 
					
						
							|  |  |  |     if (result == NULL) | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  |     result_s = PyBytes_AS_STRING(result); | 
					
						
							|  |  |  |     Py_MEMCPY(result_s, self_s, self_len); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     /* change everything in-place, starting with this one */ | 
					
						
							|  |  |  |     start =  result_s + offset; | 
					
						
							|  |  |  |     Py_MEMCPY(start, to_s, from_len); | 
					
						
							|  |  |  |     start += from_len; | 
					
						
							|  |  |  |     end = result_s + self_len; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     while ( --maxcount > 0) { | 
					
						
							|  |  |  |         offset = stringlib_find(start, end-start, | 
					
						
							|  |  |  |                                 from_s, from_len, | 
					
						
							|  |  |  |                                 0); | 
					
						
							|  |  |  |         if (offset==-1) | 
					
						
							|  |  |  |             break; | 
					
						
							|  |  |  |         Py_MEMCPY(start+offset, to_s, from_len); | 
					
						
							|  |  |  |         start += offset+from_len; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     return result; | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | /* len(self)>=1, len(from)==1, len(to)>=2, maxcount>=1 */ | 
					
						
							|  |  |  | Py_LOCAL(PyBytesObject *) | 
					
						
							|  |  |  | replace_single_character(PyBytesObject *self, | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |                          char from_c, | 
					
						
							|  |  |  |                          const char *to_s, Py_ssize_t to_len, | 
					
						
							|  |  |  |                          Py_ssize_t maxcount) | 
					
						
							|  |  |  | { | 
					
						
							|  |  |  |     char *self_s, *result_s; | 
					
						
							|  |  |  |     char *start, *next, *end; | 
					
						
							|  |  |  |     Py_ssize_t self_len, result_len; | 
					
						
							| 
									
										
										
										
											2010-08-10 18:35:01 +00:00
										 |  |  |     Py_ssize_t count; | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     PyBytesObject *result; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     self_s = PyBytes_AS_STRING(self); | 
					
						
							|  |  |  |     self_len = PyBytes_GET_SIZE(self); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     count = countchar(self_s, self_len, from_c, maxcount); | 
					
						
							|  |  |  |     if (count == 0) { | 
					
						
							|  |  |  |         /* no matches, return unchanged */ | 
					
						
							|  |  |  |         return return_self(self); | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     /* use the difference between current and new, hence the "-1" */ | 
					
						
							|  |  |  |     /*   result_len = self_len + count * (to_len-1)  */ | 
					
						
							| 
									
										
										
										
											2010-08-10 18:35:01 +00:00
										 |  |  |     assert(count > 0); | 
					
						
							|  |  |  |     if (to_len - 1 > (PY_SSIZE_T_MAX - self_len) / count) { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |         PyErr_SetString(PyExc_OverflowError, | 
					
						
							|  |  |  |                         "replacement bytes are too long"); | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  |     } | 
					
						
							| 
									
										
										
										
											2010-08-10 18:35:01 +00:00
										 |  |  |     result_len = self_len + count * (to_len - 1); | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  |     if ( (result = (PyBytesObject *) | 
					
						
							|  |  |  |           PyBytes_FromStringAndSize(NULL, result_len)) == NULL) | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  |     result_s = PyBytes_AS_STRING(result); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     start = self_s; | 
					
						
							|  |  |  |     end = self_s + self_len; | 
					
						
							|  |  |  |     while (count-- > 0) { | 
					
						
							|  |  |  |         next = findchar(start, end-start, from_c); | 
					
						
							|  |  |  |         if (next == NULL) | 
					
						
							|  |  |  |             break; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         if (next == start) { | 
					
						
							|  |  |  |             /* replace with the 'to' */ | 
					
						
							|  |  |  |             Py_MEMCPY(result_s, to_s, to_len); | 
					
						
							|  |  |  |             result_s += to_len; | 
					
						
							|  |  |  |             start += 1; | 
					
						
							|  |  |  |         } else { | 
					
						
							|  |  |  |             /* copy the unchanged old then the 'to' */ | 
					
						
							|  |  |  |             Py_MEMCPY(result_s, start, next-start); | 
					
						
							|  |  |  |             result_s += (next-start); | 
					
						
							|  |  |  |             Py_MEMCPY(result_s, to_s, to_len); | 
					
						
							|  |  |  |             result_s += to_len; | 
					
						
							|  |  |  |             start = next+1; | 
					
						
							|  |  |  |         } | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     /* Copy the remainder of the remaining string */ | 
					
						
							|  |  |  |     Py_MEMCPY(result_s, start, end-start); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     return result; | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | } | 
					
						
							| 
									
										
										
										
											2007-12-04 05:51:13 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | /* len(self)>=1, len(from)>=2, len(to)>=2, maxcount>=1 */ | 
					
						
							|  |  |  | Py_LOCAL(PyBytesObject *) | 
					
						
							|  |  |  | replace_substring(PyBytesObject *self, | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |                   const char *from_s, Py_ssize_t from_len, | 
					
						
							|  |  |  |                   const char *to_s, Py_ssize_t to_len, | 
					
						
							|  |  |  |                   Py_ssize_t maxcount) { | 
					
						
							|  |  |  |     char *self_s, *result_s; | 
					
						
							|  |  |  |     char *start, *next, *end; | 
					
						
							|  |  |  |     Py_ssize_t self_len, result_len; | 
					
						
							| 
									
										
										
										
											2010-08-10 18:35:01 +00:00
										 |  |  |     Py_ssize_t count, offset; | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     PyBytesObject *result; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     self_s = PyBytes_AS_STRING(self); | 
					
						
							|  |  |  |     self_len = PyBytes_GET_SIZE(self); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     count = stringlib_count(self_s, self_len, | 
					
						
							|  |  |  |                             from_s, from_len, | 
					
						
							|  |  |  |                             maxcount); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     if (count == 0) { | 
					
						
							|  |  |  |         /* no matches, return unchanged */ | 
					
						
							|  |  |  |         return return_self(self); | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     /* Check for overflow */ | 
					
						
							|  |  |  |     /*    result_len = self_len + count * (to_len-from_len) */ | 
					
						
							| 
									
										
										
										
											2010-08-10 18:35:01 +00:00
										 |  |  |     assert(count > 0); | 
					
						
							|  |  |  |     if (to_len - from_len > (PY_SSIZE_T_MAX - self_len) / count) { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |         PyErr_SetString(PyExc_OverflowError, | 
					
						
							|  |  |  |                         "replacement bytes are too long"); | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  |     } | 
					
						
							| 
									
										
										
										
											2010-08-10 18:35:01 +00:00
										 |  |  |     result_len = self_len + count * (to_len-from_len); | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  |     if ( (result = (PyBytesObject *) | 
					
						
							|  |  |  |           PyBytes_FromStringAndSize(NULL, result_len)) == NULL) | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  |     result_s = PyBytes_AS_STRING(result); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     start = self_s; | 
					
						
							|  |  |  |     end = self_s + self_len; | 
					
						
							|  |  |  |     while (count-- > 0) { | 
					
						
							|  |  |  |         offset = stringlib_find(start, end-start, | 
					
						
							|  |  |  |                                 from_s, from_len, | 
					
						
							|  |  |  |                                 0); | 
					
						
							|  |  |  |         if (offset == -1) | 
					
						
							|  |  |  |             break; | 
					
						
							|  |  |  |         next = start+offset; | 
					
						
							|  |  |  |         if (next == start) { | 
					
						
							|  |  |  |             /* replace with the 'to' */ | 
					
						
							|  |  |  |             Py_MEMCPY(result_s, to_s, to_len); | 
					
						
							|  |  |  |             result_s += to_len; | 
					
						
							|  |  |  |             start += from_len; | 
					
						
							|  |  |  |         } else { | 
					
						
							|  |  |  |             /* copy the unchanged old then the 'to' */ | 
					
						
							|  |  |  |             Py_MEMCPY(result_s, start, next-start); | 
					
						
							|  |  |  |             result_s += (next-start); | 
					
						
							|  |  |  |             Py_MEMCPY(result_s, to_s, to_len); | 
					
						
							|  |  |  |             result_s += to_len; | 
					
						
							|  |  |  |             start = next+from_len; | 
					
						
							|  |  |  |         } | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     /* Copy the remainder of the remaining string */ | 
					
						
							|  |  |  |     Py_MEMCPY(result_s, start, end-start); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     return result; | 
					
						
							| 
									
										
										
										
											2007-12-04 05:51:13 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  | Py_LOCAL(PyBytesObject *) | 
					
						
							|  |  |  | replace(PyBytesObject *self, | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     const char *from_s, Py_ssize_t from_len, | 
					
						
							|  |  |  |     const char *to_s, Py_ssize_t to_len, | 
					
						
							|  |  |  |     Py_ssize_t maxcount) | 
					
						
							|  |  |  | { | 
					
						
							|  |  |  |     if (maxcount < 0) { | 
					
						
							|  |  |  |         maxcount = PY_SSIZE_T_MAX; | 
					
						
							|  |  |  |     } else if (maxcount == 0 || PyBytes_GET_SIZE(self) == 0) { | 
					
						
							|  |  |  |         /* nothing to do; return the original string */ | 
					
						
							|  |  |  |         return return_self(self); | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     if (maxcount == 0 || | 
					
						
							|  |  |  |         (from_len == 0 && to_len == 0)) { | 
					
						
							|  |  |  |         /* nothing to do; return the original string */ | 
					
						
							|  |  |  |         return return_self(self); | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     /* Handle zero-length special cases */ | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     if (from_len == 0) { | 
					
						
							|  |  |  |         /* insert the 'to' string everywhere.   */ | 
					
						
							|  |  |  |         /*    >>> "Python".replace("", ".")     */ | 
					
						
							|  |  |  |         /*    '.P.y.t.h.o.n.'                   */ | 
					
						
							|  |  |  |         return replace_interleave(self, to_s, to_len, maxcount); | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     /* Except for "".replace("", "A") == "A" there is no way beyond this */ | 
					
						
							|  |  |  |     /* point for an empty self string to generate a non-empty string */ | 
					
						
							|  |  |  |     /* Special case so the remaining code always gets a non-empty string */ | 
					
						
							|  |  |  |     if (PyBytes_GET_SIZE(self) == 0) { | 
					
						
							|  |  |  |         return return_self(self); | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     if (to_len == 0) { | 
					
						
							|  |  |  |         /* delete all occurrences of 'from' string */ | 
					
						
							|  |  |  |         if (from_len == 1) { | 
					
						
							|  |  |  |             return replace_delete_single_character( | 
					
						
							|  |  |  |                 self, from_s[0], maxcount); | 
					
						
							|  |  |  |         } else { | 
					
						
							|  |  |  |             return replace_delete_substring(self, from_s, | 
					
						
							|  |  |  |                                             from_len, maxcount); | 
					
						
							|  |  |  |         } | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     /* Handle special case where both strings have the same length */ | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     if (from_len == to_len) { | 
					
						
							|  |  |  |         if (from_len == 1) { | 
					
						
							|  |  |  |             return replace_single_character_in_place( | 
					
						
							|  |  |  |                 self, | 
					
						
							|  |  |  |                 from_s[0], | 
					
						
							|  |  |  |                 to_s[0], | 
					
						
							|  |  |  |                 maxcount); | 
					
						
							|  |  |  |         } else { | 
					
						
							|  |  |  |             return replace_substring_in_place( | 
					
						
							|  |  |  |                 self, from_s, from_len, to_s, to_len, | 
					
						
							|  |  |  |                 maxcount); | 
					
						
							|  |  |  |         } | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     /* Otherwise use the more generic algorithms */ | 
					
						
							|  |  |  |     if (from_len == 1) { | 
					
						
							|  |  |  |         return replace_single_character(self, from_s[0], | 
					
						
							|  |  |  |                                         to_s, to_len, maxcount); | 
					
						
							|  |  |  |     } else { | 
					
						
							|  |  |  |         /* len('from')>=2, len('to')>=1 */ | 
					
						
							|  |  |  |         return replace_substring(self, from_s, from_len, to_s, to_len, | 
					
						
							|  |  |  |                                  maxcount); | 
					
						
							|  |  |  |     } | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | PyDoc_STRVAR(replace__doc__, | 
					
						
							|  |  |  | "B.replace(old, new[, count]) -> bytes\n\
 | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | \n\ | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | Return a copy of B with all occurrences of subsection\n\ | 
					
						
							|  |  |  | old replaced by new.  If the optional argument count is\n\ | 
					
						
							| 
									
										
										
										
											2010-09-08 12:50:29 +00:00
										 |  |  | given, only first count occurances are replaced."); | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | static PyObject * | 
					
						
							| 
									
										
										
										
											2009-04-18 15:17:02 +00:00
										 |  |  | bytes_replace(PyBytesObject *self, PyObject *args) | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     Py_ssize_t count = -1; | 
					
						
							|  |  |  |     PyObject *from, *to; | 
					
						
							|  |  |  |     const char *from_s, *to_s; | 
					
						
							|  |  |  |     Py_ssize_t from_len, to_len; | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     if (!PyArg_ParseTuple(args, "OO|n:replace", &from, &to, &count)) | 
					
						
							|  |  |  |         return NULL; | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     if (PyBytes_Check(from)) { | 
					
						
							|  |  |  |         from_s = PyBytes_AS_STRING(from); | 
					
						
							|  |  |  |         from_len = PyBytes_GET_SIZE(from); | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     else if (PyObject_AsCharBuffer(from, &from_s, &from_len)) | 
					
						
							|  |  |  |         return NULL; | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     if (PyBytes_Check(to)) { | 
					
						
							|  |  |  |         to_s = PyBytes_AS_STRING(to); | 
					
						
							|  |  |  |         to_len = PyBytes_GET_SIZE(to); | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     else if (PyObject_AsCharBuffer(to, &to_s, &to_len)) | 
					
						
							|  |  |  |         return NULL; | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     return (PyObject *)replace((PyBytesObject *) self, | 
					
						
							|  |  |  |                                from_s, from_len, | 
					
						
							|  |  |  |                                to_s, to_len, count); | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | /** End DALKE **/ | 
					
						
							| 
									
										
										
										
											2007-04-13 01:39:34 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | /* Matches the end (direction >= 0) or start (direction < 0) of self
 | 
					
						
							|  |  |  |  * against substr, using the start and end arguments. Returns | 
					
						
							|  |  |  |  * -1 on error, 0 if not found and 1 if found. | 
					
						
							|  |  |  |  */ | 
					
						
							|  |  |  | Py_LOCAL(int) | 
					
						
							| 
									
										
										
										
											2009-04-18 15:17:02 +00:00
										 |  |  | _bytes_tailmatch(PyBytesObject *self, PyObject *substr, Py_ssize_t start, | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |                   Py_ssize_t end, int direction) | 
					
						
							| 
									
										
										
										
											2007-04-13 01:39:34 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     Py_ssize_t len = PyBytes_GET_SIZE(self); | 
					
						
							|  |  |  |     Py_ssize_t slen; | 
					
						
							|  |  |  |     const char* sub; | 
					
						
							|  |  |  |     const char* str; | 
					
						
							| 
									
										
										
										
											2007-04-13 01:39:34 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     if (PyBytes_Check(substr)) { | 
					
						
							|  |  |  |         sub = PyBytes_AS_STRING(substr); | 
					
						
							|  |  |  |         slen = PyBytes_GET_SIZE(substr); | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     else if (PyObject_AsCharBuffer(substr, &sub, &slen)) | 
					
						
							|  |  |  |         return -1; | 
					
						
							|  |  |  |     str = PyBytes_AS_STRING(self); | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     ADJUST_INDICES(start, end, len); | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     if (direction < 0) { | 
					
						
							|  |  |  |         /* startswith */ | 
					
						
							|  |  |  |         if (start+slen > len) | 
					
						
							|  |  |  |             return 0; | 
					
						
							|  |  |  |     } else { | 
					
						
							|  |  |  |         /* endswith */ | 
					
						
							|  |  |  |         if (end-start < slen || start > len) | 
					
						
							|  |  |  |             return 0; | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |         if (end-slen > start) | 
					
						
							|  |  |  |             start = end - slen; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     if (end-start >= slen) | 
					
						
							|  |  |  |         return ! memcmp(str+start, sub, slen); | 
					
						
							|  |  |  |     return 0; | 
					
						
							| 
									
										
										
										
											2007-04-13 01:39:34 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | PyDoc_STRVAR(startswith__doc__, | 
					
						
							| 
									
										
										
										
											2008-05-30 08:20:09 +00:00
										 |  |  | "B.startswith(prefix[, start[, end]]) -> bool\n\
 | 
					
						
							| 
									
										
										
										
											2007-04-13 01:39:34 +00:00
										 |  |  | \n\ | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | Return True if B starts with the specified prefix, False otherwise.\n\ | 
					
						
							|  |  |  | With optional start, test B beginning at that position.\n\ | 
					
						
							|  |  |  | With optional end, stop comparing B at that position.\n\ | 
					
						
							| 
									
										
										
										
											2008-05-27 00:36:20 +00:00
										 |  |  | prefix can also be a tuple of bytes to try."); | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2007-04-13 01:39:34 +00:00
										 |  |  | static PyObject * | 
					
						
							| 
									
										
										
										
											2009-04-18 15:17:02 +00:00
										 |  |  | bytes_startswith(PyBytesObject *self, PyObject *args) | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     Py_ssize_t start = 0; | 
					
						
							|  |  |  |     Py_ssize_t end = PY_SSIZE_T_MAX; | 
					
						
							|  |  |  |     PyObject *subobj; | 
					
						
							|  |  |  |     int result; | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2011-04-20 17:09:23 +02:00
										 |  |  |     if (!stringlib_parse_args_finds("startswith", args, &subobj, &start, &end)) | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |         return NULL; | 
					
						
							|  |  |  |     if (PyTuple_Check(subobj)) { | 
					
						
							|  |  |  |         Py_ssize_t i; | 
					
						
							|  |  |  |         for (i = 0; i < PyTuple_GET_SIZE(subobj); i++) { | 
					
						
							|  |  |  |             result = _bytes_tailmatch(self, | 
					
						
							|  |  |  |                             PyTuple_GET_ITEM(subobj, i), | 
					
						
							|  |  |  |                             start, end, -1); | 
					
						
							|  |  |  |             if (result == -1) | 
					
						
							|  |  |  |                 return NULL; | 
					
						
							|  |  |  |             else if (result) { | 
					
						
							|  |  |  |                 Py_RETURN_TRUE; | 
					
						
							|  |  |  |             } | 
					
						
							|  |  |  |         } | 
					
						
							|  |  |  |         Py_RETURN_FALSE; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     result = _bytes_tailmatch(self, subobj, start, end, -1); | 
					
						
							| 
									
										
										
										
											2011-04-26 06:09:45 +03:00
										 |  |  |     if (result == -1) { | 
					
						
							|  |  |  |         if (PyErr_ExceptionMatches(PyExc_TypeError)) | 
					
						
							|  |  |  |             PyErr_Format(PyExc_TypeError, "startswith first arg must be bytes " | 
					
						
							|  |  |  |                          "or a tuple of bytes, not %s", Py_TYPE(subobj)->tp_name); | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |         return NULL; | 
					
						
							| 
									
										
										
										
											2011-04-26 06:09:45 +03:00
										 |  |  |     } | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     else | 
					
						
							|  |  |  |         return PyBool_FromLong(result); | 
					
						
							| 
									
										
										
										
											2007-04-13 01:39:34 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  | PyDoc_STRVAR(endswith__doc__, | 
					
						
							| 
									
										
										
										
											2008-05-30 08:20:09 +00:00
										 |  |  | "B.endswith(suffix[, start[, end]]) -> bool\n\
 | 
					
						
							| 
									
										
										
										
											2007-04-13 01:39:34 +00:00
										 |  |  | \n\ | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | Return True if B ends with the specified suffix, False otherwise.\n\ | 
					
						
							|  |  |  | With optional start, test B beginning at that position.\n\ | 
					
						
							|  |  |  | With optional end, stop comparing B at that position.\n\ | 
					
						
							| 
									
										
										
										
											2008-05-27 00:36:20 +00:00
										 |  |  | suffix can also be a tuple of bytes to try."); | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2007-04-13 01:39:34 +00:00
										 |  |  | static PyObject * | 
					
						
							| 
									
										
										
										
											2009-04-18 15:17:02 +00:00
										 |  |  | bytes_endswith(PyBytesObject *self, PyObject *args) | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     Py_ssize_t start = 0; | 
					
						
							|  |  |  |     Py_ssize_t end = PY_SSIZE_T_MAX; | 
					
						
							|  |  |  |     PyObject *subobj; | 
					
						
							|  |  |  |     int result; | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2011-04-20 17:09:23 +02:00
										 |  |  |     if (!stringlib_parse_args_finds("endswith", args, &subobj, &start, &end)) | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |         return NULL; | 
					
						
							|  |  |  |     if (PyTuple_Check(subobj)) { | 
					
						
							|  |  |  |         Py_ssize_t i; | 
					
						
							|  |  |  |         for (i = 0; i < PyTuple_GET_SIZE(subobj); i++) { | 
					
						
							|  |  |  |             result = _bytes_tailmatch(self, | 
					
						
							|  |  |  |                             PyTuple_GET_ITEM(subobj, i), | 
					
						
							|  |  |  |                             start, end, +1); | 
					
						
							|  |  |  |             if (result == -1) | 
					
						
							|  |  |  |                 return NULL; | 
					
						
							|  |  |  |             else if (result) { | 
					
						
							|  |  |  |                 Py_RETURN_TRUE; | 
					
						
							|  |  |  |             } | 
					
						
							|  |  |  |         } | 
					
						
							|  |  |  |         Py_RETURN_FALSE; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     result = _bytes_tailmatch(self, subobj, start, end, +1); | 
					
						
							| 
									
										
										
										
											2011-04-26 06:09:45 +03:00
										 |  |  |     if (result == -1) { | 
					
						
							|  |  |  |         if (PyErr_ExceptionMatches(PyExc_TypeError)) | 
					
						
							|  |  |  |             PyErr_Format(PyExc_TypeError, "endswith first arg must be bytes or " | 
					
						
							|  |  |  |                          "a tuple of bytes, not %s", Py_TYPE(subobj)->tp_name); | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |         return NULL; | 
					
						
							| 
									
										
										
										
											2011-04-26 06:09:45 +03:00
										 |  |  |     } | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     else | 
					
						
							|  |  |  |         return PyBool_FromLong(result); | 
					
						
							| 
									
										
										
										
											2007-04-13 01:39:34 +00:00
										 |  |  | } | 
					
						
							| 
									
										
										
										
											2007-02-27 19:02:19 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  | PyDoc_STRVAR(decode__doc__, | 
					
						
							| 
									
										
										
										
											2010-11-07 19:04:46 +00:00
										 |  |  | "B.decode(encoding='utf-8', errors='strict') -> str\n\
 | 
					
						
							| 
									
										
										
										
											2006-04-24 13:47:05 +00:00
										 |  |  | \n\ | 
					
						
							| 
									
										
										
										
											2010-11-07 18:41:46 +00:00
										 |  |  | Decode B using the codec registered for encoding. Default encoding\n\ | 
					
						
							|  |  |  | is 'utf-8'. errors may be given to set a different error\n\ | 
					
						
							| 
									
										
										
										
											2007-11-06 21:34:58 +00:00
										 |  |  | handling scheme.  Default is 'strict' meaning that encoding errors raise\n\ | 
					
						
							|  |  |  | a UnicodeDecodeError.  Other possible values are 'ignore' and 'replace'\n\ | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | as well as any other name registerd with codecs.register_error that is\n\ | 
					
						
							| 
									
										
										
										
											2006-04-24 13:47:05 +00:00
										 |  |  | able to handle UnicodeDecodeErrors."); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | static PyObject * | 
					
						
							| 
									
										
										
										
											2009-09-18 21:42:35 +00:00
										 |  |  | bytes_decode(PyObject *self, PyObject *args, PyObject *kwargs) | 
					
						
							| 
									
										
										
										
											2007-04-12 22:49:52 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     const char *encoding = NULL; | 
					
						
							|  |  |  |     const char *errors = NULL; | 
					
						
							|  |  |  |     static char *kwlist[] = {"encoding", "errors", 0}; | 
					
						
							| 
									
										
										
										
											2006-05-05 04:34:18 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|ss:decode", kwlist, &encoding, &errors)) | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  |     if (encoding == NULL) | 
					
						
							|  |  |  |         encoding = PyUnicode_GetDefaultEncoding(); | 
					
						
							|  |  |  |     return PyUnicode_FromEncodedObject(self, encoding, errors); | 
					
						
							| 
									
										
										
										
											2006-05-05 04:34:18 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2006-05-05 15:15:40 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2010-01-13 08:07:53 +00:00
										 |  |  | PyDoc_STRVAR(splitlines__doc__, | 
					
						
							|  |  |  | "B.splitlines([keepends]) -> list of lines\n\
 | 
					
						
							|  |  |  | \n\ | 
					
						
							|  |  |  | Return a list of the lines in B, breaking at line boundaries.\n\ | 
					
						
							|  |  |  | Line breaks are not included in the resulting list unless keepends\n\ | 
					
						
							|  |  |  | is given and true."); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | static PyObject* | 
					
						
							| 
									
										
										
										
											2011-09-24 09:14:39 +01:00
										 |  |  | bytes_splitlines(PyObject *self, PyObject *args, PyObject *kwds) | 
					
						
							| 
									
										
										
										
											2010-01-13 08:07:53 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2011-09-24 09:14:39 +01:00
										 |  |  |     static char *kwlist[] = {"keepends", 0}; | 
					
						
							| 
									
										
										
										
											2010-01-13 08:07:53 +00:00
										 |  |  |     int keepends = 0; | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2011-09-24 09:14:39 +01:00
										 |  |  |     if (!PyArg_ParseTupleAndKeywords(args, kwds, "|i:splitlines", | 
					
						
							|  |  |  |                                      kwlist, &keepends)) | 
					
						
							| 
									
										
										
										
											2010-06-09 16:38:55 +00:00
										 |  |  |         return NULL; | 
					
						
							| 
									
										
										
										
											2010-01-13 08:07:53 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  |     return stringlib_splitlines( | 
					
						
							| 
									
										
										
										
											2010-06-09 16:38:55 +00:00
										 |  |  |         (PyObject*) self, PyBytes_AS_STRING(self), | 
					
						
							|  |  |  |         PyBytes_GET_SIZE(self), keepends | 
					
						
							|  |  |  |         ); | 
					
						
							| 
									
										
										
										
											2010-01-13 08:07:53 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2007-02-27 08:40:54 +00:00
										 |  |  | PyDoc_STRVAR(fromhex_doc, | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | "bytes.fromhex(string) -> bytes\n\
 | 
					
						
							| 
									
										
										
										
											2007-02-27 08:40:54 +00:00
										 |  |  | \n\ | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | Create a bytes object from a string of hexadecimal numbers.\n\ | 
					
						
							| 
									
										
										
										
											2007-11-06 21:34:58 +00:00
										 |  |  | Spaces between two numbers are accepted.\n\ | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | Example: bytes.fromhex('B9 01EF') -> b'\\xb9\\x01\\xef'."); | 
					
						
							| 
									
										
										
										
											2007-02-27 08:40:54 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  | static int | 
					
						
							| 
									
										
										
										
											2011-09-29 04:02:13 +02:00
										 |  |  | hex_digit_to_int(Py_UCS4 c) | 
					
						
							| 
									
										
										
										
											2007-02-27 08:40:54 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     if (c >= 128) | 
					
						
							|  |  |  |         return -1; | 
					
						
							| 
									
										
										
										
											2010-11-05 17:23:41 +00:00
										 |  |  |     if (Py_ISDIGIT(c)) | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |         return c - '0'; | 
					
						
							|  |  |  |     else { | 
					
						
							| 
									
										
										
										
											2010-11-05 17:23:41 +00:00
										 |  |  |         if (Py_ISUPPER(c)) | 
					
						
							|  |  |  |             c = Py_TOLOWER(c); | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |         if (c >= 'a' && c <= 'f') | 
					
						
							|  |  |  |             return c - 'a' + 10; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     return -1; | 
					
						
							| 
									
										
										
										
											2007-02-27 08:40:54 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | static PyObject * | 
					
						
							| 
									
										
										
										
											2009-04-18 15:17:02 +00:00
										 |  |  | bytes_fromhex(PyObject *cls, PyObject *args) | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     PyObject *newstring, *hexobj; | 
					
						
							|  |  |  |     char *buf; | 
					
						
							|  |  |  |     Py_ssize_t hexlen, byteslen, i, j; | 
					
						
							|  |  |  |     int top, bot; | 
					
						
							| 
									
										
										
										
											2011-09-28 07:41:54 +02:00
										 |  |  |     void *data; | 
					
						
							|  |  |  |     unsigned int kind; | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  |     if (!PyArg_ParseTuple(args, "U:fromhex", &hexobj)) | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  |     assert(PyUnicode_Check(hexobj)); | 
					
						
							| 
									
										
										
										
											2011-09-28 07:41:54 +02:00
										 |  |  |     if (PyUnicode_READY(hexobj)) | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  |     kind = PyUnicode_KIND(hexobj); | 
					
						
							|  |  |  |     data = PyUnicode_DATA(hexobj); | 
					
						
							|  |  |  |     hexlen = PyUnicode_GET_LENGTH(hexobj); | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     byteslen = hexlen/2; /* This overestimates if there are spaces */ | 
					
						
							|  |  |  |     newstring = PyBytes_FromStringAndSize(NULL, byteslen); | 
					
						
							|  |  |  |     if (!newstring) | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  |     buf = PyBytes_AS_STRING(newstring); | 
					
						
							|  |  |  |     for (i = j = 0; i < hexlen; i += 2) { | 
					
						
							|  |  |  |         /* skip over spaces in the input */ | 
					
						
							| 
									
										
										
										
											2011-09-28 07:41:54 +02:00
										 |  |  |         while (PyUnicode_READ(kind, data, i) == ' ') | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |             i++; | 
					
						
							|  |  |  |         if (i >= hexlen) | 
					
						
							|  |  |  |             break; | 
					
						
							| 
									
										
										
										
											2011-09-28 07:41:54 +02:00
										 |  |  |         top = hex_digit_to_int(PyUnicode_READ(kind, data, i)); | 
					
						
							|  |  |  |         bot = hex_digit_to_int(PyUnicode_READ(kind, data, i+1)); | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |         if (top == -1 || bot == -1) { | 
					
						
							|  |  |  |             PyErr_Format(PyExc_ValueError, | 
					
						
							|  |  |  |                          "non-hexadecimal number found in " | 
					
						
							|  |  |  |                          "fromhex() arg at position %zd", i); | 
					
						
							|  |  |  |             goto error; | 
					
						
							|  |  |  |         } | 
					
						
							|  |  |  |         buf[j++] = (top << 4) + bot; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     if (j != byteslen && _PyBytes_Resize(&newstring, j) < 0) | 
					
						
							|  |  |  |         goto error; | 
					
						
							|  |  |  |     return newstring; | 
					
						
							| 
									
										
										
										
											2007-02-27 08:40:54 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  |   error: | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     Py_XDECREF(newstring); | 
					
						
							|  |  |  |     return NULL; | 
					
						
							| 
									
										
										
										
											2007-02-27 08:40:54 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-06-04 14:18:43 +00:00
										 |  |  | PyDoc_STRVAR(sizeof__doc__, | 
					
						
							| 
									
										
										
										
											2009-04-18 15:17:02 +00:00
										 |  |  | "B.__sizeof__() -> size of B in memory, in bytes"); | 
					
						
							| 
									
										
										
										
											2008-06-04 14:18:43 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  | static PyObject * | 
					
						
							| 
									
										
										
										
											2009-04-18 15:17:02 +00:00
										 |  |  | bytes_sizeof(PyBytesObject *v) | 
					
						
							| 
									
										
										
										
											2008-06-04 14:18:43 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     Py_ssize_t res; | 
					
						
							|  |  |  |     res = PyBytesObject_SIZE + Py_SIZE(v) * Py_TYPE(v)->tp_itemsize; | 
					
						
							|  |  |  |     return PyLong_FromSsize_t(res); | 
					
						
							| 
									
										
										
										
											2008-06-04 14:18:43 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2007-04-11 05:40:58 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  | static PyObject * | 
					
						
							| 
									
										
										
										
											2009-04-18 15:17:02 +00:00
										 |  |  | bytes_getnewargs(PyBytesObject *v) | 
					
						
							| 
									
										
										
										
											2007-04-11 05:40:58 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     return Py_BuildValue("(y#)", v->ob_sval, Py_SIZE(v)); | 
					
						
							| 
									
										
										
										
											2007-04-11 05:40:58 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2006-04-22 23:28:04 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | static PyMethodDef | 
					
						
							| 
									
										
										
										
											2009-04-18 15:17:02 +00:00
										 |  |  | bytes_methods[] = { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     {"__getnewargs__",          (PyCFunction)bytes_getnewargs,  METH_NOARGS}, | 
					
						
							|  |  |  |     {"capitalize", (PyCFunction)stringlib_capitalize, METH_NOARGS, | 
					
						
							|  |  |  |      _Py_capitalize__doc__}, | 
					
						
							|  |  |  |     {"center", (PyCFunction)stringlib_center, METH_VARARGS, center__doc__}, | 
					
						
							|  |  |  |     {"count", (PyCFunction)bytes_count, METH_VARARGS, count__doc__}, | 
					
						
							|  |  |  |     {"decode", (PyCFunction)bytes_decode, METH_VARARGS | METH_KEYWORDS, decode__doc__}, | 
					
						
							|  |  |  |     {"endswith", (PyCFunction)bytes_endswith, METH_VARARGS, | 
					
						
							|  |  |  |      endswith__doc__}, | 
					
						
							|  |  |  |     {"expandtabs", (PyCFunction)stringlib_expandtabs, METH_VARARGS, | 
					
						
							|  |  |  |      expandtabs__doc__}, | 
					
						
							|  |  |  |     {"find", (PyCFunction)bytes_find, METH_VARARGS, find__doc__}, | 
					
						
							|  |  |  |     {"fromhex", (PyCFunction)bytes_fromhex, METH_VARARGS|METH_CLASS, | 
					
						
							|  |  |  |      fromhex_doc}, | 
					
						
							|  |  |  |     {"index", (PyCFunction)bytes_index, METH_VARARGS, index__doc__}, | 
					
						
							|  |  |  |     {"isalnum", (PyCFunction)stringlib_isalnum, METH_NOARGS, | 
					
						
							|  |  |  |      _Py_isalnum__doc__}, | 
					
						
							|  |  |  |     {"isalpha", (PyCFunction)stringlib_isalpha, METH_NOARGS, | 
					
						
							|  |  |  |      _Py_isalpha__doc__}, | 
					
						
							|  |  |  |     {"isdigit", (PyCFunction)stringlib_isdigit, METH_NOARGS, | 
					
						
							|  |  |  |      _Py_isdigit__doc__}, | 
					
						
							|  |  |  |     {"islower", (PyCFunction)stringlib_islower, METH_NOARGS, | 
					
						
							|  |  |  |      _Py_islower__doc__}, | 
					
						
							|  |  |  |     {"isspace", (PyCFunction)stringlib_isspace, METH_NOARGS, | 
					
						
							|  |  |  |      _Py_isspace__doc__}, | 
					
						
							|  |  |  |     {"istitle", (PyCFunction)stringlib_istitle, METH_NOARGS, | 
					
						
							|  |  |  |      _Py_istitle__doc__}, | 
					
						
							|  |  |  |     {"isupper", (PyCFunction)stringlib_isupper, METH_NOARGS, | 
					
						
							|  |  |  |      _Py_isupper__doc__}, | 
					
						
							|  |  |  |     {"join", (PyCFunction)bytes_join, METH_O, join__doc__}, | 
					
						
							|  |  |  |     {"ljust", (PyCFunction)stringlib_ljust, METH_VARARGS, ljust__doc__}, | 
					
						
							|  |  |  |     {"lower", (PyCFunction)stringlib_lower, METH_NOARGS, _Py_lower__doc__}, | 
					
						
							|  |  |  |     {"lstrip", (PyCFunction)bytes_lstrip, METH_VARARGS, lstrip__doc__}, | 
					
						
							|  |  |  |     {"maketrans", (PyCFunction)bytes_maketrans, METH_VARARGS|METH_STATIC, | 
					
						
							|  |  |  |      _Py_maketrans__doc__}, | 
					
						
							|  |  |  |     {"partition", (PyCFunction)bytes_partition, METH_O, partition__doc__}, | 
					
						
							|  |  |  |     {"replace", (PyCFunction)bytes_replace, METH_VARARGS, replace__doc__}, | 
					
						
							|  |  |  |     {"rfind", (PyCFunction)bytes_rfind, METH_VARARGS, rfind__doc__}, | 
					
						
							|  |  |  |     {"rindex", (PyCFunction)bytes_rindex, METH_VARARGS, rindex__doc__}, | 
					
						
							|  |  |  |     {"rjust", (PyCFunction)stringlib_rjust, METH_VARARGS, rjust__doc__}, | 
					
						
							|  |  |  |     {"rpartition", (PyCFunction)bytes_rpartition, METH_O, | 
					
						
							|  |  |  |      rpartition__doc__}, | 
					
						
							|  |  |  |     {"rsplit", (PyCFunction)bytes_rsplit, METH_VARARGS, rsplit__doc__}, | 
					
						
							|  |  |  |     {"rstrip", (PyCFunction)bytes_rstrip, METH_VARARGS, rstrip__doc__}, | 
					
						
							|  |  |  |     {"split", (PyCFunction)bytes_split, METH_VARARGS, split__doc__}, | 
					
						
							| 
									
										
										
										
											2011-09-24 09:14:39 +01:00
										 |  |  |     {"splitlines", (PyCFunction)bytes_splitlines, METH_VARARGS | METH_KEYWORDS, | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |      splitlines__doc__}, | 
					
						
							|  |  |  |     {"startswith", (PyCFunction)bytes_startswith, METH_VARARGS, | 
					
						
							|  |  |  |      startswith__doc__}, | 
					
						
							|  |  |  |     {"strip", (PyCFunction)bytes_strip, METH_VARARGS, strip__doc__}, | 
					
						
							|  |  |  |     {"swapcase", (PyCFunction)stringlib_swapcase, METH_NOARGS, | 
					
						
							|  |  |  |      _Py_swapcase__doc__}, | 
					
						
							|  |  |  |     {"title", (PyCFunction)stringlib_title, METH_NOARGS, _Py_title__doc__}, | 
					
						
							|  |  |  |     {"translate", (PyCFunction)bytes_translate, METH_VARARGS, | 
					
						
							|  |  |  |      translate__doc__}, | 
					
						
							|  |  |  |     {"upper", (PyCFunction)stringlib_upper, METH_NOARGS, _Py_upper__doc__}, | 
					
						
							|  |  |  |     {"zfill", (PyCFunction)stringlib_zfill, METH_VARARGS, zfill__doc__}, | 
					
						
							|  |  |  |     {"__sizeof__", (PyCFunction)bytes_sizeof, METH_NOARGS, | 
					
						
							|  |  |  |      sizeof__doc__}, | 
					
						
							|  |  |  |     {NULL,     NULL}                         /* sentinel */ | 
					
						
							| 
									
										
										
										
											2006-04-22 23:28:04 +00:00
										 |  |  | }; | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | static PyObject * | 
					
						
							|  |  |  | str_subtype_new(PyTypeObject *type, PyObject *args, PyObject *kwds); | 
					
						
							| 
									
										
										
										
											2006-04-22 23:28:04 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | static PyObject * | 
					
						
							| 
									
										
										
										
											2009-04-18 15:17:02 +00:00
										 |  |  | bytes_new(PyTypeObject *type, PyObject *args, PyObject *kwds) | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     PyObject *x = NULL; | 
					
						
							|  |  |  |     const char *encoding = NULL; | 
					
						
							|  |  |  |     const char *errors = NULL; | 
					
						
							|  |  |  |     PyObject *new = NULL; | 
					
						
							|  |  |  |     Py_ssize_t size; | 
					
						
							|  |  |  |     static char *kwlist[] = {"source", "encoding", "errors", 0}; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     if (type != &PyBytes_Type) | 
					
						
							|  |  |  |         return str_subtype_new(type, args, kwds); | 
					
						
							|  |  |  |     if (!PyArg_ParseTupleAndKeywords(args, kwds, "|Oss:bytes", kwlist, &x, | 
					
						
							|  |  |  |                                      &encoding, &errors)) | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  |     if (x == NULL) { | 
					
						
							|  |  |  |         if (encoding != NULL || errors != NULL) { | 
					
						
							|  |  |  |             PyErr_SetString(PyExc_TypeError, | 
					
						
							|  |  |  |                             "encoding or errors without sequence " | 
					
						
							|  |  |  |                             "argument"); | 
					
						
							|  |  |  |             return NULL; | 
					
						
							|  |  |  |         } | 
					
						
							|  |  |  |         return PyBytes_FromString(""); | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     if (PyUnicode_Check(x)) { | 
					
						
							|  |  |  |         /* Encode via the codec registry */ | 
					
						
							|  |  |  |         if (encoding == NULL) { | 
					
						
							|  |  |  |             PyErr_SetString(PyExc_TypeError, | 
					
						
							|  |  |  |                             "string argument without an encoding"); | 
					
						
							|  |  |  |             return NULL; | 
					
						
							|  |  |  |         } | 
					
						
							|  |  |  |         new = PyUnicode_AsEncodedString(x, encoding, errors); | 
					
						
							|  |  |  |         if (new == NULL) | 
					
						
							|  |  |  |             return NULL; | 
					
						
							|  |  |  |         assert(PyBytes_Check(new)); | 
					
						
							|  |  |  |         return new; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     /* Is it an integer? */ | 
					
						
							|  |  |  |     size = PyNumber_AsSsize_t(x, PyExc_OverflowError); | 
					
						
							|  |  |  |     if (size == -1 && PyErr_Occurred()) { | 
					
						
							|  |  |  |         if (PyErr_ExceptionMatches(PyExc_OverflowError)) | 
					
						
							|  |  |  |             return NULL; | 
					
						
							|  |  |  |         PyErr_Clear(); | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     else if (size < 0) { | 
					
						
							|  |  |  |         PyErr_SetString(PyExc_ValueError, "negative count"); | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     else { | 
					
						
							|  |  |  |         new = PyBytes_FromStringAndSize(NULL, size); | 
					
						
							|  |  |  |         if (new == NULL) { | 
					
						
							|  |  |  |             return NULL; | 
					
						
							|  |  |  |         } | 
					
						
							|  |  |  |         if (size > 0) { | 
					
						
							|  |  |  |             memset(((PyBytesObject*)new)->ob_sval, 0, size); | 
					
						
							|  |  |  |         } | 
					
						
							|  |  |  |         return new; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     /* If it's not unicode, there can't be encoding or errors */ | 
					
						
							|  |  |  |     if (encoding != NULL || errors != NULL) { | 
					
						
							|  |  |  |         PyErr_SetString(PyExc_TypeError, | 
					
						
							|  |  |  |             "encoding or errors without a string argument"); | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     return PyObject_Bytes(x); | 
					
						
							| 
									
										
										
										
											2008-08-26 16:46:47 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | PyObject * | 
					
						
							|  |  |  | PyBytes_FromObject(PyObject *x) | 
					
						
							|  |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     PyObject *new, *it; | 
					
						
							|  |  |  |     Py_ssize_t i, size; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     if (x == NULL) { | 
					
						
							|  |  |  |         PyErr_BadInternalCall(); | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     /* Use the modern buffer interface */ | 
					
						
							|  |  |  |     if (PyObject_CheckBuffer(x)) { | 
					
						
							|  |  |  |         Py_buffer view; | 
					
						
							|  |  |  |         if (PyObject_GetBuffer(x, &view, PyBUF_FULL_RO) < 0) | 
					
						
							|  |  |  |             return NULL; | 
					
						
							|  |  |  |         new = PyBytes_FromStringAndSize(NULL, view.len); | 
					
						
							|  |  |  |         if (!new) | 
					
						
							|  |  |  |             goto fail; | 
					
						
							|  |  |  |         /* XXX(brett.cannon): Better way to get to internal buffer? */ | 
					
						
							|  |  |  |         if (PyBuffer_ToContiguous(((PyBytesObject *)new)->ob_sval, | 
					
						
							|  |  |  |                                   &view, view.len, 'C') < 0) | 
					
						
							|  |  |  |             goto fail; | 
					
						
							|  |  |  |         PyBuffer_Release(&view); | 
					
						
							|  |  |  |         return new; | 
					
						
							|  |  |  |       fail: | 
					
						
							|  |  |  |         Py_XDECREF(new); | 
					
						
							|  |  |  |         PyBuffer_Release(&view); | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     if (PyUnicode_Check(x)) { | 
					
						
							|  |  |  |         PyErr_SetString(PyExc_TypeError, | 
					
						
							|  |  |  |                         "cannot convert unicode object to bytes"); | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     if (PyList_CheckExact(x)) { | 
					
						
							|  |  |  |         new = PyBytes_FromStringAndSize(NULL, Py_SIZE(x)); | 
					
						
							|  |  |  |         if (new == NULL) | 
					
						
							|  |  |  |             return NULL; | 
					
						
							|  |  |  |         for (i = 0; i < Py_SIZE(x); i++) { | 
					
						
							|  |  |  |             Py_ssize_t value = PyNumber_AsSsize_t( | 
					
						
							|  |  |  |                 PyList_GET_ITEM(x, i), PyExc_ValueError); | 
					
						
							|  |  |  |             if (value == -1 && PyErr_Occurred()) { | 
					
						
							|  |  |  |                 Py_DECREF(new); | 
					
						
							|  |  |  |                 return NULL; | 
					
						
							|  |  |  |             } | 
					
						
							|  |  |  |             if (value < 0 || value >= 256) { | 
					
						
							|  |  |  |                 PyErr_SetString(PyExc_ValueError, | 
					
						
							|  |  |  |                                 "bytes must be in range(0, 256)"); | 
					
						
							|  |  |  |                 Py_DECREF(new); | 
					
						
							|  |  |  |                 return NULL; | 
					
						
							|  |  |  |             } | 
					
						
							| 
									
										
										
										
											2010-08-15 17:12:55 +00:00
										 |  |  |             ((PyBytesObject *)new)->ob_sval[i] = (char) value; | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |         } | 
					
						
							|  |  |  |         return new; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     if (PyTuple_CheckExact(x)) { | 
					
						
							|  |  |  |         new = PyBytes_FromStringAndSize(NULL, Py_SIZE(x)); | 
					
						
							|  |  |  |         if (new == NULL) | 
					
						
							|  |  |  |             return NULL; | 
					
						
							|  |  |  |         for (i = 0; i < Py_SIZE(x); i++) { | 
					
						
							|  |  |  |             Py_ssize_t value = PyNumber_AsSsize_t( | 
					
						
							|  |  |  |                 PyTuple_GET_ITEM(x, i), PyExc_ValueError); | 
					
						
							|  |  |  |             if (value == -1 && PyErr_Occurred()) { | 
					
						
							|  |  |  |                 Py_DECREF(new); | 
					
						
							|  |  |  |                 return NULL; | 
					
						
							|  |  |  |             } | 
					
						
							|  |  |  |             if (value < 0 || value >= 256) { | 
					
						
							|  |  |  |                 PyErr_SetString(PyExc_ValueError, | 
					
						
							|  |  |  |                                 "bytes must be in range(0, 256)"); | 
					
						
							|  |  |  |                 Py_DECREF(new); | 
					
						
							|  |  |  |                 return NULL; | 
					
						
							|  |  |  |             } | 
					
						
							| 
									
										
										
										
											2010-08-15 17:12:55 +00:00
										 |  |  |             ((PyBytesObject *)new)->ob_sval[i] = (char) value; | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |         } | 
					
						
							|  |  |  |         return new; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     /* For iterator version, create a string object and resize as needed */ | 
					
						
							|  |  |  |     size = _PyObject_LengthHint(x, 64); | 
					
						
							|  |  |  |     if (size == -1 && PyErr_Occurred()) | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  |     /* Allocate an extra byte to prevent PyBytes_FromStringAndSize() from
 | 
					
						
							|  |  |  |        returning a shared empty bytes string. This required because we | 
					
						
							|  |  |  |        want to call _PyBytes_Resize() the returned object, which we can | 
					
						
							|  |  |  |        only do on bytes objects with refcount == 1. */ | 
					
						
							|  |  |  |     size += 1; | 
					
						
							|  |  |  |     new = PyBytes_FromStringAndSize(NULL, size); | 
					
						
							|  |  |  |     if (new == NULL) | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     /* Get the iterator */ | 
					
						
							|  |  |  |     it = PyObject_GetIter(x); | 
					
						
							|  |  |  |     if (it == NULL) | 
					
						
							|  |  |  |         goto error; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     /* Run the iterator to exhaustion */ | 
					
						
							|  |  |  |     for (i = 0; ; i++) { | 
					
						
							|  |  |  |         PyObject *item; | 
					
						
							|  |  |  |         Py_ssize_t value; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         /* Get the next item */ | 
					
						
							|  |  |  |         item = PyIter_Next(it); | 
					
						
							|  |  |  |         if (item == NULL) { | 
					
						
							|  |  |  |             if (PyErr_Occurred()) | 
					
						
							|  |  |  |                 goto error; | 
					
						
							|  |  |  |             break; | 
					
						
							|  |  |  |         } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         /* Interpret it as an int (__index__) */ | 
					
						
							|  |  |  |         value = PyNumber_AsSsize_t(item, PyExc_ValueError); | 
					
						
							|  |  |  |         Py_DECREF(item); | 
					
						
							|  |  |  |         if (value == -1 && PyErr_Occurred()) | 
					
						
							|  |  |  |             goto error; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         /* Range check */ | 
					
						
							|  |  |  |         if (value < 0 || value >= 256) { | 
					
						
							|  |  |  |             PyErr_SetString(PyExc_ValueError, | 
					
						
							|  |  |  |                             "bytes must be in range(0, 256)"); | 
					
						
							|  |  |  |             goto error; | 
					
						
							|  |  |  |         } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         /* Append the byte */ | 
					
						
							|  |  |  |         if (i >= size) { | 
					
						
							|  |  |  |             size = 2 * size + 1; | 
					
						
							|  |  |  |             if (_PyBytes_Resize(&new, size) < 0) | 
					
						
							|  |  |  |                 goto error; | 
					
						
							|  |  |  |         } | 
					
						
							| 
									
										
										
										
											2010-08-15 17:12:55 +00:00
										 |  |  |         ((PyBytesObject *)new)->ob_sval[i] = (char) value; | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     } | 
					
						
							|  |  |  |     _PyBytes_Resize(&new, i); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     /* Clean up and return success */ | 
					
						
							|  |  |  |     Py_DECREF(it); | 
					
						
							|  |  |  |     return new; | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  |   error: | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     /* Error handling when new != NULL */ | 
					
						
							|  |  |  |     Py_XDECREF(it); | 
					
						
							|  |  |  |     Py_DECREF(new); | 
					
						
							|  |  |  |     return NULL; | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | static PyObject * | 
					
						
							|  |  |  | str_subtype_new(PyTypeObject *type, PyObject *args, PyObject *kwds) | 
					
						
							|  |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     PyObject *tmp, *pnew; | 
					
						
							|  |  |  |     Py_ssize_t n; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     assert(PyType_IsSubtype(type, &PyBytes_Type)); | 
					
						
							|  |  |  |     tmp = bytes_new(&PyBytes_Type, args, kwds); | 
					
						
							|  |  |  |     if (tmp == NULL) | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  |     assert(PyBytes_CheckExact(tmp)); | 
					
						
							|  |  |  |     n = PyBytes_GET_SIZE(tmp); | 
					
						
							|  |  |  |     pnew = type->tp_alloc(type, n); | 
					
						
							|  |  |  |     if (pnew != NULL) { | 
					
						
							|  |  |  |         Py_MEMCPY(PyBytes_AS_STRING(pnew), | 
					
						
							|  |  |  |                   PyBytes_AS_STRING(tmp), n+1); | 
					
						
							|  |  |  |         ((PyBytesObject *)pnew)->ob_shash = | 
					
						
							|  |  |  |             ((PyBytesObject *)tmp)->ob_shash; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     Py_DECREF(tmp); | 
					
						
							|  |  |  |     return pnew; | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | } | 
					
						
							| 
									
										
										
										
											2006-04-22 23:28:04 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2009-04-18 15:17:02 +00:00
										 |  |  | PyDoc_STRVAR(bytes_doc, | 
					
						
							| 
									
										
										
										
											2008-05-30 08:20:09 +00:00
										 |  |  | "bytes(iterable_of_ints) -> bytes\n\
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | bytes(string, encoding[, errors]) -> bytes\n\ | 
					
						
							| 
									
										
										
										
											2008-05-30 08:20:09 +00:00
										 |  |  | bytes(bytes_or_buffer) -> immutable copy of bytes_or_buffer\n\ | 
					
						
							|  |  |  | bytes(memory_view) -> bytes\n\ | 
					
						
							| 
									
										
										
										
											2007-11-06 21:34:58 +00:00
										 |  |  | \n\ | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | Construct an immutable array of bytes from:\n\ | 
					
						
							| 
									
										
										
										
											2007-11-06 21:34:58 +00:00
										 |  |  |   - an iterable yielding integers in range(256)\n\ | 
					
						
							|  |  |  |   - a text string encoded using the specified encoding\n\ | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  |   - a bytes or a buffer object\n\ | 
					
						
							|  |  |  |   - any object implementing the buffer API."); | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2009-04-18 15:17:02 +00:00
										 |  |  | static PyObject *bytes_iter(PyObject *seq); | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  | PyTypeObject PyBytes_Type = { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     PyVarObject_HEAD_INIT(&PyType_Type, 0) | 
					
						
							|  |  |  |     "bytes", | 
					
						
							|  |  |  |     PyBytesObject_SIZE, | 
					
						
							|  |  |  |     sizeof(char), | 
					
						
							|  |  |  |     bytes_dealloc,                      /* tp_dealloc */ | 
					
						
							|  |  |  |     0,                                          /* tp_print */ | 
					
						
							|  |  |  |     0,                                          /* tp_getattr */ | 
					
						
							|  |  |  |     0,                                          /* tp_setattr */ | 
					
						
							|  |  |  |     0,                                          /* tp_reserved */ | 
					
						
							|  |  |  |     (reprfunc)bytes_repr,                       /* tp_repr */ | 
					
						
							|  |  |  |     0,                                          /* tp_as_number */ | 
					
						
							|  |  |  |     &bytes_as_sequence,                         /* tp_as_sequence */ | 
					
						
							|  |  |  |     &bytes_as_mapping,                          /* tp_as_mapping */ | 
					
						
							|  |  |  |     (hashfunc)bytes_hash,                       /* tp_hash */ | 
					
						
							|  |  |  |     0,                                          /* tp_call */ | 
					
						
							|  |  |  |     bytes_str,                                  /* tp_str */ | 
					
						
							|  |  |  |     PyObject_GenericGetAttr,                    /* tp_getattro */ | 
					
						
							|  |  |  |     0,                                          /* tp_setattro */ | 
					
						
							|  |  |  |     &bytes_as_buffer,                           /* tp_as_buffer */ | 
					
						
							|  |  |  |     Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE | | 
					
						
							|  |  |  |         Py_TPFLAGS_BYTES_SUBCLASS,              /* tp_flags */ | 
					
						
							|  |  |  |     bytes_doc,                                  /* tp_doc */ | 
					
						
							|  |  |  |     0,                                          /* tp_traverse */ | 
					
						
							|  |  |  |     0,                                          /* tp_clear */ | 
					
						
							|  |  |  |     (richcmpfunc)bytes_richcompare,             /* tp_richcompare */ | 
					
						
							|  |  |  |     0,                                          /* tp_weaklistoffset */ | 
					
						
							|  |  |  |     bytes_iter,                                 /* tp_iter */ | 
					
						
							|  |  |  |     0,                                          /* tp_iternext */ | 
					
						
							|  |  |  |     bytes_methods,                              /* tp_methods */ | 
					
						
							|  |  |  |     0,                                          /* tp_members */ | 
					
						
							|  |  |  |     0,                                          /* tp_getset */ | 
					
						
							|  |  |  |     &PyBaseObject_Type,                         /* tp_base */ | 
					
						
							|  |  |  |     0,                                          /* tp_dict */ | 
					
						
							|  |  |  |     0,                                          /* tp_descr_get */ | 
					
						
							|  |  |  |     0,                                          /* tp_descr_set */ | 
					
						
							|  |  |  |     0,                                          /* tp_dictoffset */ | 
					
						
							|  |  |  |     0,                                          /* tp_init */ | 
					
						
							|  |  |  |     0,                                          /* tp_alloc */ | 
					
						
							|  |  |  |     bytes_new,                                  /* tp_new */ | 
					
						
							|  |  |  |     PyObject_Del,                               /* tp_free */ | 
					
						
							| 
									
										
										
										
											2006-04-22 23:28:04 +00:00
										 |  |  | }; | 
					
						
							| 
									
										
										
										
											2007-10-26 17:39:48 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | void | 
					
						
							|  |  |  | PyBytes_Concat(register PyObject **pv, register PyObject *w) | 
					
						
							|  |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     register PyObject *v; | 
					
						
							|  |  |  |     assert(pv != NULL); | 
					
						
							|  |  |  |     if (*pv == NULL) | 
					
						
							|  |  |  |         return; | 
					
						
							|  |  |  |     if (w == NULL) { | 
					
						
							|  |  |  |         Py_DECREF(*pv); | 
					
						
							|  |  |  |         *pv = NULL; | 
					
						
							|  |  |  |         return; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     v = bytes_concat(*pv, w); | 
					
						
							|  |  |  |     Py_DECREF(*pv); | 
					
						
							|  |  |  |     *pv = v; | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | void | 
					
						
							|  |  |  | PyBytes_ConcatAndDel(register PyObject **pv, register PyObject *w) | 
					
						
							|  |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     PyBytes_Concat(pv, w); | 
					
						
							|  |  |  |     Py_XDECREF(w); | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | /* The following function breaks the notion that strings are immutable:
 | 
					
						
							|  |  |  |    it changes the size of a string.  We get away with this only if there | 
					
						
							|  |  |  |    is only one module referencing the object.  You can also think of it | 
					
						
							|  |  |  |    as creating a new string object and destroying the old one, only | 
					
						
							|  |  |  |    more efficiently.  In any case, don't use this if the string may | 
					
						
							|  |  |  |    already be known to some other part of the code... | 
					
						
							|  |  |  |    Note that if there's not enough memory to resize the string, the original | 
					
						
							|  |  |  |    string object at *pv is deallocated, *pv is set to NULL, an "out of | 
					
						
							|  |  |  |    memory" exception is set, and -1 is returned.  Else (on success) 0 is | 
					
						
							|  |  |  |    returned, and the value in *pv may or may not be the same as on input. | 
					
						
							|  |  |  |    As always, an extra byte is allocated for a trailing \0 byte (newsize | 
					
						
							|  |  |  |    does *not* include that), and a trailing \0 byte is stored. | 
					
						
							|  |  |  | */ | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | int | 
					
						
							|  |  |  | _PyBytes_Resize(PyObject **pv, Py_ssize_t newsize) | 
					
						
							|  |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     register PyObject *v; | 
					
						
							|  |  |  |     register PyBytesObject *sv; | 
					
						
							|  |  |  |     v = *pv; | 
					
						
							|  |  |  |     if (!PyBytes_Check(v) || Py_REFCNT(v) != 1 || newsize < 0) { | 
					
						
							|  |  |  |         *pv = 0; | 
					
						
							|  |  |  |         Py_DECREF(v); | 
					
						
							|  |  |  |         PyErr_BadInternalCall(); | 
					
						
							|  |  |  |         return -1; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     /* XXX UNREF/NEWREF interface should be more symmetrical */ | 
					
						
							|  |  |  |     _Py_DEC_REFTOTAL; | 
					
						
							|  |  |  |     _Py_ForgetReference(v); | 
					
						
							|  |  |  |     *pv = (PyObject *) | 
					
						
							|  |  |  |         PyObject_REALLOC((char *)v, PyBytesObject_SIZE + newsize); | 
					
						
							|  |  |  |     if (*pv == NULL) { | 
					
						
							|  |  |  |         PyObject_Del(v); | 
					
						
							|  |  |  |         PyErr_NoMemory(); | 
					
						
							|  |  |  |         return -1; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     _Py_NewReference(*pv); | 
					
						
							|  |  |  |     sv = (PyBytesObject *) *pv; | 
					
						
							|  |  |  |     Py_SIZE(sv) = newsize; | 
					
						
							|  |  |  |     sv->ob_sval[newsize] = '\0'; | 
					
						
							|  |  |  |     sv->ob_shash = -1;          /* invalidate cached hash value */ | 
					
						
							|  |  |  |     return 0; | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | /* _PyBytes_FormatLong emulates the format codes d, u, o, x and X, and
 | 
					
						
							|  |  |  |  * the F_ALT flag, for Python's long (unbounded) ints.  It's not used for | 
					
						
							|  |  |  |  * Python's regular ints. | 
					
						
							| 
									
										
										
										
											2010-01-13 08:07:53 +00:00
										 |  |  |  * Return value:  a new PyBytes*, or NULL if error. | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  |  *  .  *pbuf is set to point into it, | 
					
						
							|  |  |  |  *     *plen set to the # of chars following that. | 
					
						
							|  |  |  |  *     Caller must decref it when done using pbuf. | 
					
						
							|  |  |  |  *     The string starting at *pbuf is of the form | 
					
						
							|  |  |  |  *         "-"? ("0x" | "0X")? digit+ | 
					
						
							|  |  |  |  *     "0x"/"0X" are present only for x and X conversions, with F_ALT | 
					
						
							|  |  |  |  *         set in flags.  The case of hex digits will be correct, | 
					
						
							|  |  |  |  *     There will be at least prec digits, zero-filled on the left if | 
					
						
							|  |  |  |  *         necessary to get that many. | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |  * val          object to be converted | 
					
						
							|  |  |  |  * flags        bitmask of format flags; only F_ALT is looked at | 
					
						
							|  |  |  |  * prec         minimum number of digits; 0-fill on left if needed | 
					
						
							|  |  |  |  * type         a character in [duoxX]; u acts the same as d | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  |  * | 
					
						
							|  |  |  |  * CAUTION:  o, x and X conversions on regular ints can never | 
					
						
							|  |  |  |  * produce a '-' sign, but can for Python's unbounded ints. | 
					
						
							|  |  |  |  */ | 
					
						
							|  |  |  | PyObject* | 
					
						
							|  |  |  | _PyBytes_FormatLong(PyObject *val, int flags, int prec, int type, | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |                      char **pbuf, int *plen) | 
					
						
							|  |  |  | { | 
					
						
							|  |  |  |     PyObject *result = NULL; | 
					
						
							|  |  |  |     char *buf; | 
					
						
							|  |  |  |     Py_ssize_t i; | 
					
						
							|  |  |  |     int sign;           /* 1 if '-', else 0 */ | 
					
						
							|  |  |  |     int len;            /* number of characters */ | 
					
						
							|  |  |  |     Py_ssize_t llen; | 
					
						
							|  |  |  |     int numdigits;      /* len == numnondigits + numdigits */ | 
					
						
							|  |  |  |     int numnondigits = 0; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     /* Avoid exceeding SSIZE_T_MAX */ | 
					
						
							|  |  |  |     if (prec > INT_MAX-3) { | 
					
						
							|  |  |  |         PyErr_SetString(PyExc_OverflowError, | 
					
						
							|  |  |  |                         "precision too large"); | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     switch (type) { | 
					
						
							|  |  |  |     case 'd': | 
					
						
							|  |  |  |     case 'u': | 
					
						
							|  |  |  |         /* Special-case boolean: we want 0/1 */ | 
					
						
							|  |  |  |         if (PyBool_Check(val)) | 
					
						
							|  |  |  |             result = PyNumber_ToBase(val, 10); | 
					
						
							|  |  |  |         else | 
					
						
							|  |  |  |             result = Py_TYPE(val)->tp_str(val); | 
					
						
							|  |  |  |         break; | 
					
						
							|  |  |  |     case 'o': | 
					
						
							|  |  |  |         numnondigits = 2; | 
					
						
							|  |  |  |         result = PyNumber_ToBase(val, 8); | 
					
						
							|  |  |  |         break; | 
					
						
							|  |  |  |     case 'x': | 
					
						
							|  |  |  |     case 'X': | 
					
						
							|  |  |  |         numnondigits = 2; | 
					
						
							|  |  |  |         result = PyNumber_ToBase(val, 16); | 
					
						
							|  |  |  |         break; | 
					
						
							|  |  |  |     default: | 
					
						
							|  |  |  |         assert(!"'type' not in [duoxX]"); | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     if (!result) | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     buf = _PyUnicode_AsString(result); | 
					
						
							|  |  |  |     if (!buf) { | 
					
						
							|  |  |  |         Py_DECREF(result); | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     /* To modify the string in-place, there can only be one reference. */ | 
					
						
							|  |  |  |     if (Py_REFCNT(result) != 1) { | 
					
						
							|  |  |  |         PyErr_BadInternalCall(); | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     llen = PyUnicode_GetSize(result); | 
					
						
							|  |  |  |     if (llen > INT_MAX) { | 
					
						
							|  |  |  |         PyErr_SetString(PyExc_ValueError, | 
					
						
							|  |  |  |                         "string too large in _PyBytes_FormatLong"); | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     len = (int)llen; | 
					
						
							|  |  |  |     if (buf[len-1] == 'L') { | 
					
						
							|  |  |  |         --len; | 
					
						
							|  |  |  |         buf[len] = '\0'; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     sign = buf[0] == '-'; | 
					
						
							|  |  |  |     numnondigits += sign; | 
					
						
							|  |  |  |     numdigits = len - numnondigits; | 
					
						
							|  |  |  |     assert(numdigits > 0); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     /* Get rid of base marker unless F_ALT */ | 
					
						
							|  |  |  |     if (((flags & F_ALT) == 0 && | 
					
						
							|  |  |  |         (type == 'o' || type == 'x' || type == 'X'))) { | 
					
						
							|  |  |  |         assert(buf[sign] == '0'); | 
					
						
							|  |  |  |         assert(buf[sign+1] == 'x' || buf[sign+1] == 'X' || | 
					
						
							|  |  |  |                buf[sign+1] == 'o'); | 
					
						
							|  |  |  |         numnondigits -= 2; | 
					
						
							|  |  |  |         buf += 2; | 
					
						
							|  |  |  |         len -= 2; | 
					
						
							|  |  |  |         if (sign) | 
					
						
							|  |  |  |             buf[0] = '-'; | 
					
						
							|  |  |  |         assert(len == numnondigits + numdigits); | 
					
						
							|  |  |  |         assert(numdigits > 0); | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     /* Fill with leading zeroes to meet minimum width. */ | 
					
						
							|  |  |  |     if (prec > numdigits) { | 
					
						
							|  |  |  |         PyObject *r1 = PyBytes_FromStringAndSize(NULL, | 
					
						
							|  |  |  |                                 numnondigits + prec); | 
					
						
							|  |  |  |         char *b1; | 
					
						
							|  |  |  |         if (!r1) { | 
					
						
							|  |  |  |             Py_DECREF(result); | 
					
						
							|  |  |  |             return NULL; | 
					
						
							|  |  |  |         } | 
					
						
							|  |  |  |         b1 = PyBytes_AS_STRING(r1); | 
					
						
							|  |  |  |         for (i = 0; i < numnondigits; ++i) | 
					
						
							|  |  |  |             *b1++ = *buf++; | 
					
						
							|  |  |  |         for (i = 0; i < prec - numdigits; i++) | 
					
						
							|  |  |  |             *b1++ = '0'; | 
					
						
							|  |  |  |         for (i = 0; i < numdigits; i++) | 
					
						
							|  |  |  |             *b1++ = *buf++; | 
					
						
							|  |  |  |         *b1 = '\0'; | 
					
						
							|  |  |  |         Py_DECREF(result); | 
					
						
							|  |  |  |         result = r1; | 
					
						
							|  |  |  |         buf = PyBytes_AS_STRING(result); | 
					
						
							|  |  |  |         len = numnondigits + prec; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     /* Fix up case for hex conversions. */ | 
					
						
							|  |  |  |     if (type == 'X') { | 
					
						
							|  |  |  |         /* Need to convert all lower case letters to upper case.
 | 
					
						
							|  |  |  |            and need to convert 0x to 0X (and -0x to -0X). */ | 
					
						
							|  |  |  |         for (i = 0; i < len; i++) | 
					
						
							|  |  |  |             if (buf[i] >= 'a' && buf[i] <= 'x') | 
					
						
							|  |  |  |                 buf[i] -= 'a'-'A'; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     *pbuf = buf; | 
					
						
							|  |  |  |     *plen = len; | 
					
						
							|  |  |  |     return result; | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | void | 
					
						
							|  |  |  | PyBytes_Fini(void) | 
					
						
							|  |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     int i; | 
					
						
							|  |  |  |     for (i = 0; i < UCHAR_MAX + 1; i++) { | 
					
						
							|  |  |  |         Py_XDECREF(characters[i]); | 
					
						
							|  |  |  |         characters[i] = NULL; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     Py_XDECREF(nullstring); | 
					
						
							|  |  |  |     nullstring = NULL; | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-27 00:36:20 +00:00
										 |  |  | /*********************** Bytes Iterator ****************************/ | 
					
						
							| 
									
										
										
										
											2007-10-26 17:39:48 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  | typedef struct { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     PyObject_HEAD | 
					
						
							|  |  |  |     Py_ssize_t it_index; | 
					
						
							|  |  |  |     PyBytesObject *it_seq; /* Set to NULL when iterator is exhausted */ | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | } striterobject; | 
					
						
							| 
									
										
										
										
											2007-10-26 17:39:48 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  | static void | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | striter_dealloc(striterobject *it) | 
					
						
							| 
									
										
										
										
											2007-10-26 17:39:48 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     _PyObject_GC_UNTRACK(it); | 
					
						
							|  |  |  |     Py_XDECREF(it->it_seq); | 
					
						
							|  |  |  |     PyObject_GC_Del(it); | 
					
						
							| 
									
										
										
										
											2007-10-26 17:39:48 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | static int | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | striter_traverse(striterobject *it, visitproc visit, void *arg) | 
					
						
							| 
									
										
										
										
											2007-10-26 17:39:48 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     Py_VISIT(it->it_seq); | 
					
						
							|  |  |  |     return 0; | 
					
						
							| 
									
										
										
										
											2007-10-26 17:39:48 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | static PyObject * | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | striter_next(striterobject *it) | 
					
						
							|  |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     PyBytesObject *seq; | 
					
						
							|  |  |  |     PyObject *item; | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     assert(it != NULL); | 
					
						
							|  |  |  |     seq = it->it_seq; | 
					
						
							|  |  |  |     if (seq == NULL) | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  |     assert(PyBytes_Check(seq)); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     if (it->it_index < PyBytes_GET_SIZE(seq)) { | 
					
						
							|  |  |  |         item = PyLong_FromLong( | 
					
						
							|  |  |  |             (unsigned char)seq->ob_sval[it->it_index]); | 
					
						
							|  |  |  |         if (item != NULL) | 
					
						
							|  |  |  |             ++it->it_index; | 
					
						
							|  |  |  |         return item; | 
					
						
							|  |  |  |     } | 
					
						
							| 
									
										
										
										
											2007-10-26 17:39:48 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     Py_DECREF(seq); | 
					
						
							|  |  |  |     it->it_seq = NULL; | 
					
						
							|  |  |  |     return NULL; | 
					
						
							| 
									
										
										
										
											2007-10-26 17:39:48 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | static PyObject * | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | striter_len(striterobject *it) | 
					
						
							| 
									
										
										
										
											2007-10-26 17:39:48 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     Py_ssize_t len = 0; | 
					
						
							|  |  |  |     if (it->it_seq) | 
					
						
							|  |  |  |         len = PyBytes_GET_SIZE(it->it_seq) - it->it_index; | 
					
						
							|  |  |  |     return PyLong_FromSsize_t(len); | 
					
						
							| 
									
										
										
										
											2007-10-26 17:39:48 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | PyDoc_STRVAR(length_hint_doc, | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |              "Private method returning an estimate of len(list(it))."); | 
					
						
							| 
									
										
										
										
											2007-10-26 17:39:48 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | static PyMethodDef striter_methods[] = { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     {"__length_hint__", (PyCFunction)striter_len, METH_NOARGS, | 
					
						
							|  |  |  |      length_hint_doc}, | 
					
						
							|  |  |  |     {NULL,              NULL}           /* sentinel */ | 
					
						
							| 
									
										
										
										
											2007-10-26 17:39:48 +00:00
										 |  |  | }; | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-05-26 13:42:13 +00:00
										 |  |  | PyTypeObject PyBytesIter_Type = { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     PyVarObject_HEAD_INIT(&PyType_Type, 0) | 
					
						
							|  |  |  |     "bytes_iterator",                           /* tp_name */ | 
					
						
							|  |  |  |     sizeof(striterobject),                      /* tp_basicsize */ | 
					
						
							|  |  |  |     0,                                          /* tp_itemsize */ | 
					
						
							|  |  |  |     /* methods */ | 
					
						
							|  |  |  |     (destructor)striter_dealloc,                /* tp_dealloc */ | 
					
						
							|  |  |  |     0,                                          /* tp_print */ | 
					
						
							|  |  |  |     0,                                          /* tp_getattr */ | 
					
						
							|  |  |  |     0,                                          /* tp_setattr */ | 
					
						
							|  |  |  |     0,                                          /* tp_reserved */ | 
					
						
							|  |  |  |     0,                                          /* tp_repr */ | 
					
						
							|  |  |  |     0,                                          /* tp_as_number */ | 
					
						
							|  |  |  |     0,                                          /* tp_as_sequence */ | 
					
						
							|  |  |  |     0,                                          /* tp_as_mapping */ | 
					
						
							|  |  |  |     0,                                          /* tp_hash */ | 
					
						
							|  |  |  |     0,                                          /* tp_call */ | 
					
						
							|  |  |  |     0,                                          /* tp_str */ | 
					
						
							|  |  |  |     PyObject_GenericGetAttr,                    /* tp_getattro */ | 
					
						
							|  |  |  |     0,                                          /* tp_setattro */ | 
					
						
							|  |  |  |     0,                                          /* tp_as_buffer */ | 
					
						
							|  |  |  |     Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC,/* tp_flags */ | 
					
						
							|  |  |  |     0,                                          /* tp_doc */ | 
					
						
							|  |  |  |     (traverseproc)striter_traverse,     /* tp_traverse */ | 
					
						
							|  |  |  |     0,                                          /* tp_clear */ | 
					
						
							|  |  |  |     0,                                          /* tp_richcompare */ | 
					
						
							|  |  |  |     0,                                          /* tp_weaklistoffset */ | 
					
						
							|  |  |  |     PyObject_SelfIter,                          /* tp_iter */ | 
					
						
							|  |  |  |     (iternextfunc)striter_next,                 /* tp_iternext */ | 
					
						
							|  |  |  |     striter_methods,                            /* tp_methods */ | 
					
						
							|  |  |  |     0, | 
					
						
							| 
									
										
										
										
											2007-10-26 17:39:48 +00:00
										 |  |  | }; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | static PyObject * | 
					
						
							| 
									
										
										
										
											2009-04-18 15:17:02 +00:00
										 |  |  | bytes_iter(PyObject *seq) | 
					
						
							| 
									
										
										
										
											2007-10-26 17:39:48 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2010-05-09 15:52:27 +00:00
										 |  |  |     striterobject *it; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     if (!PyBytes_Check(seq)) { | 
					
						
							|  |  |  |         PyErr_BadInternalCall(); | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  |     } | 
					
						
							|  |  |  |     it = PyObject_GC_New(striterobject, &PyBytesIter_Type); | 
					
						
							|  |  |  |     if (it == NULL) | 
					
						
							|  |  |  |         return NULL; | 
					
						
							|  |  |  |     it->it_index = 0; | 
					
						
							|  |  |  |     Py_INCREF(seq); | 
					
						
							|  |  |  |     it->it_seq = (PyBytesObject *)seq; | 
					
						
							|  |  |  |     _PyObject_GC_TRACK(it); | 
					
						
							|  |  |  |     return (PyObject *)it; | 
					
						
							| 
									
										
										
										
											2007-10-26 17:39:48 +00:00
										 |  |  | } |