2008-05-26 21:36:20 -03:00
|
|
|
/* bytes object implementation */
|
2007-10-16 03:31:30 -03:00
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
#define PY_SSIZE_T_CLEAN
|
2007-02-27 15:02:19 -04:00
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
#include "Python.h"
|
2007-08-18 08:21:56 -03:00
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
#include "bytes_methods.h"
|
2008-12-06 11:33:31 -04:00
|
|
|
#include <stddef.h>
|
2007-08-18 08:21:56 -03:00
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
#ifdef COUNT_ALLOCS
|
2009-01-11 13:13:55 -04:00
|
|
|
Py_ssize_t null_strings, one_strings;
|
2008-05-26 10:42:13 -03:00
|
|
|
#endif
|
2006-04-22 20:28:04 -03:00
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
static PyBytesObject *characters[UCHAR_MAX + 1];
|
|
|
|
static PyBytesObject *nullstring;
|
|
|
|
|
2008-12-06 11:33:31 -04: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 10:42:13 -03: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 04:07:53 -04:00
|
|
|
PyBytes object must be treated as immutable and you must not fill in nor
|
2008-05-26 10:42:13 -03: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 17:32:56 -03:00
|
|
|
allocated for string data, not counting the null terminating character.
|
|
|
|
It is therefore equal to the `size' parameter (for
|
2008-05-26 10:42:13 -03:00
|
|
|
PyBytes_FromStringAndSize()) or the length of the string in the `str'
|
|
|
|
parameter (for PyBytes_FromString()).
|
|
|
|
*/
|
2006-04-22 20:28:04 -03:00
|
|
|
PyObject *
|
2008-05-26 10:42:13 -03:00
|
|
|
PyBytes_FromStringAndSize(const char *str, Py_ssize_t size)
|
2006-04-24 10:47:05 -03:00
|
|
|
{
|
2013-08-13 15:18:52 -03:00
|
|
|
PyBytesObject *op;
|
2010-05-09 12:52:27 -03:00
|
|
|
if (size < 0) {
|
|
|
|
PyErr_SetString(PyExc_SystemError,
|
|
|
|
"Negative size passed to PyBytes_FromStringAndSize");
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
if (size == 0 && (op = nullstring) != NULL) {
|
2008-05-26 10:42:13 -03:00
|
|
|
#ifdef COUNT_ALLOCS
|
2010-05-09 12:52:27 -03:00
|
|
|
null_strings++;
|
2008-05-26 10:42:13 -03:00
|
|
|
#endif
|
2010-05-09 12:52:27 -03:00
|
|
|
Py_INCREF(op);
|
|
|
|
return (PyObject *)op;
|
|
|
|
}
|
|
|
|
if (size == 1 && str != NULL &&
|
|
|
|
(op = characters[*str & UCHAR_MAX]) != NULL)
|
|
|
|
{
|
2008-05-26 10:42:13 -03:00
|
|
|
#ifdef COUNT_ALLOCS
|
2010-05-09 12:52:27 -03:00
|
|
|
one_strings++;
|
2008-05-26 10:42:13 -03:00
|
|
|
#endif
|
2010-05-09 12:52:27 -03: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();
|
2013-12-04 04:27:47 -04:00
|
|
|
(void)PyObject_INIT_VAR(op, &PyBytes_Type, size);
|
2010-05-09 12:52:27 -03:00
|
|
|
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 10:47:05 -03:00
|
|
|
}
|
|
|
|
|
|
|
|
PyObject *
|
2008-05-26 10:42:13 -03:00
|
|
|
PyBytes_FromString(const char *str)
|
2006-04-22 20:28:04 -03:00
|
|
|
{
|
2013-08-13 15:18:52 -03:00
|
|
|
size_t size;
|
|
|
|
PyBytesObject *op;
|
2010-05-09 12:52:27 -03:00
|
|
|
|
|
|
|
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 10:42:13 -03:00
|
|
|
#ifdef COUNT_ALLOCS
|
2010-05-09 12:52:27 -03:00
|
|
|
null_strings++;
|
2008-05-26 10:42:13 -03:00
|
|
|
#endif
|
2010-05-09 12:52:27 -03:00
|
|
|
Py_INCREF(op);
|
|
|
|
return (PyObject *)op;
|
|
|
|
}
|
|
|
|
if (size == 1 && (op = characters[*str & UCHAR_MAX]) != NULL) {
|
2008-05-26 10:42:13 -03:00
|
|
|
#ifdef COUNT_ALLOCS
|
2010-05-09 12:52:27 -03:00
|
|
|
one_strings++;
|
2008-05-26 10:42:13 -03:00
|
|
|
#endif
|
2010-05-09 12:52:27 -03:00
|
|
|
Py_INCREF(op);
|
|
|
|
return (PyObject *)op;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Inline PyObject_NewVar */
|
|
|
|
op = (PyBytesObject *)PyObject_MALLOC(PyBytesObject_SIZE + size);
|
|
|
|
if (op == NULL)
|
|
|
|
return PyErr_NoMemory();
|
2013-12-04 04:27:47 -04:00
|
|
|
(void)PyObject_INIT_VAR(op, &PyBytes_Type, size);
|
2010-05-09 12:52:27 -03:00
|
|
|
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 20:28:04 -03:00
|
|
|
}
|
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
PyObject *
|
|
|
|
PyBytes_FromFormatV(const char *format, va_list vargs)
|
2006-04-22 20:28:04 -03:00
|
|
|
{
|
2010-05-09 12:52:27 -03:00
|
|
|
va_list count;
|
|
|
|
Py_ssize_t n = 0;
|
|
|
|
const char* f;
|
|
|
|
char *s;
|
|
|
|
PyObject* string;
|
2008-05-26 10:42:13 -03:00
|
|
|
|
2010-08-11 14:31:17 -03:00
|
|
|
Py_VA_COPY(count, vargs);
|
2010-05-09 12:52:27 -03: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 14:23:41 -03:00
|
|
|
while (*++f && *f != '%' && !Py_ISALPHA(*f))
|
2010-05-09 12:52:27 -03: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':
|
2013-12-13 07:14:44 -04:00
|
|
|
{
|
|
|
|
int c = va_arg(count, int);
|
|
|
|
if (c < 0 || c > 255) {
|
|
|
|
PyErr_SetString(PyExc_OverflowError,
|
|
|
|
"PyBytes_FromFormatV(): %c format "
|
|
|
|
"expects an integer in range [0; 255]");
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
n++;
|
|
|
|
break;
|
|
|
|
}
|
2010-05-09 12:52:27 -03:00
|
|
|
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 10:42:13 -03:00
|
|
|
expand:
|
2010-05-09 12:52:27 -03: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 14:23:41 -03:00
|
|
|
while (Py_ISDIGIT(*f))
|
2010-05-09 12:52:27 -03:00
|
|
|
n = (n*10) + *f++ - '0';
|
|
|
|
if (*f == '.') {
|
|
|
|
f++;
|
|
|
|
n = 0;
|
2010-11-05 14:23:41 -03:00
|
|
|
while (Py_ISDIGIT(*f))
|
2010-05-09 12:52:27 -03:00
|
|
|
n = (n*10) + *f++ - '0';
|
|
|
|
}
|
2010-11-05 14:23:41 -03:00
|
|
|
while (*f && *f != '%' && !Py_ISALPHA(*f))
|
2010-05-09 12:52:27 -03: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':
|
2013-12-13 07:14:44 -04:00
|
|
|
{
|
|
|
|
int c = va_arg(vargs, int);
|
|
|
|
/* c has been checked for overflow in the first step */
|
|
|
|
*s++ = (unsigned char)c;
|
2010-05-09 12:52:27 -03:00
|
|
|
break;
|
2013-12-13 07:14:44 -04:00
|
|
|
}
|
2010-05-09 12:52:27 -03:00
|
|
|
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 20:28:04 -03:00
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
end:
|
2010-05-09 12:52:27 -03:00
|
|
|
_PyBytes_Resize(&string, s - PyBytes_AS_STRING(string));
|
|
|
|
return string;
|
2006-04-22 20:28:04 -03:00
|
|
|
}
|
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
PyObject *
|
|
|
|
PyBytes_FromFormat(const char *format, ...)
|
2006-04-22 20:28:04 -03:00
|
|
|
{
|
2010-05-09 12:52:27 -03:00
|
|
|
PyObject* ret;
|
|
|
|
va_list vargs;
|
2006-04-22 20:28:04 -03:00
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
#ifdef HAVE_STDARG_PROTOTYPES
|
2010-05-09 12:52:27 -03:00
|
|
|
va_start(vargs, format);
|
2008-05-26 10:42:13 -03:00
|
|
|
#else
|
2010-05-09 12:52:27 -03:00
|
|
|
va_start(vargs);
|
2008-05-26 10:42:13 -03:00
|
|
|
#endif
|
2010-05-09 12:52:27 -03:00
|
|
|
ret = PyBytes_FromFormatV(format, vargs);
|
|
|
|
va_end(vargs);
|
|
|
|
return ret;
|
2006-04-22 20:28:04 -03:00
|
|
|
}
|
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
static void
|
2009-04-18 12:17:02 -03:00
|
|
|
bytes_dealloc(PyObject *op)
|
2008-05-26 10:42:13 -03:00
|
|
|
{
|
2010-05-09 12:52:27 -03:00
|
|
|
Py_TYPE(op)->tp_free(op);
|
2008-05-26 10:42:13 -03: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 12:52:27 -03: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 14:23:41 -03:00
|
|
|
if (s+1 < end && Py_ISXDIGIT(s[0]) && Py_ISXDIGIT(s[1])) {
|
2010-05-09 12:52:27 -03:00
|
|
|
unsigned int x = 0;
|
|
|
|
c = Py_CHARMASK(*s);
|
|
|
|
s++;
|
2010-11-05 14:23:41 -03:00
|
|
|
if (Py_ISDIGIT(c))
|
2010-05-09 12:52:27 -03:00
|
|
|
x = c - '0';
|
2010-11-05 14:23:41 -03:00
|
|
|
else if (Py_ISLOWER(c))
|
2010-05-09 12:52:27 -03:00
|
|
|
x = 10 + c - 'a';
|
|
|
|
else
|
|
|
|
x = 10 + c - 'A';
|
|
|
|
x = x << 4;
|
|
|
|
c = Py_CHARMASK(*s);
|
|
|
|
s++;
|
2010-11-05 14:23:41 -03:00
|
|
|
if (Py_ISDIGIT(c))
|
2010-05-09 12:52:27 -03:00
|
|
|
x += c - '0';
|
2010-11-05 14:23:41 -03:00
|
|
|
else if (Py_ISLOWER(c))
|
2010-05-09 12:52:27 -03:00
|
|
|
x += 10 + c - 'a';
|
|
|
|
else
|
|
|
|
x += 10 + c - 'A';
|
|
|
|
*p++ = x;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
if (!errors || strcmp(errors, "strict") == 0) {
|
2013-02-10 11:36:00 -04:00
|
|
|
PyErr_Format(PyExc_ValueError,
|
|
|
|
"invalid \\x escape at position %d",
|
2013-02-10 11:42:01 -04:00
|
|
|
s - 2 - (end - len));
|
2010-05-09 12:52:27 -03:00
|
|
|
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;
|
|
|
|
}
|
2013-01-25 17:31:43 -04:00
|
|
|
/* skip \x */
|
|
|
|
if (s < end && Py_ISXDIGIT(s[0]))
|
|
|
|
s++; /* and a hexdigit */
|
|
|
|
break;
|
2010-05-09 12:52:27 -03:00
|
|
|
default:
|
|
|
|
*p++ = '\\';
|
|
|
|
s--;
|
2011-03-15 00:18:48 -03:00
|
|
|
goto non_esc; /* an arbitrary number of unescaped
|
2010-05-09 12:52:27 -03:00
|
|
|
UTF-8 bytes may follow. */
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if (p-buf < newlen)
|
|
|
|
_PyBytes_Resize(&v, p - buf);
|
|
|
|
return v;
|
2008-05-26 10:42:13 -03:00
|
|
|
failed:
|
2010-05-09 12:52:27 -03:00
|
|
|
Py_DECREF(v);
|
|
|
|
return NULL;
|
2008-05-26 10:42:13 -03:00
|
|
|
}
|
2006-04-22 20:28:04 -03:00
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
/* -------------------------------------------------------------------- */
|
|
|
|
/* object api */
|
2006-04-22 20:28:04 -03:00
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
Py_ssize_t
|
2013-08-13 15:18:52 -03:00
|
|
|
PyBytes_Size(PyObject *op)
|
2008-05-26 10:42:13 -03:00
|
|
|
{
|
2010-05-09 12:52:27 -03: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 10:42:13 -03:00
|
|
|
}
|
2006-05-05 01:34:18 -03:00
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
char *
|
2013-08-13 15:18:52 -03:00
|
|
|
PyBytes_AsString(PyObject *op)
|
2008-05-26 10:42:13 -03:00
|
|
|
{
|
2010-05-09 12:52:27 -03: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 10:42:13 -03:00
|
|
|
}
|
2007-08-18 08:21:56 -03:00
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
int
|
2013-08-13 15:18:52 -03:00
|
|
|
PyBytes_AsStringAndSize(PyObject *obj,
|
|
|
|
char **s,
|
|
|
|
Py_ssize_t *len)
|
2010-05-09 12:52:27 -03:00
|
|
|
{
|
|
|
|
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 20:28:04 -03:00
|
|
|
}
|
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
/* -------------------------------------------------------------------- */
|
|
|
|
/* Methods */
|
2007-04-12 22:39:34 -03:00
|
|
|
|
2009-04-16 17:16:10 -03:00
|
|
|
#include "stringlib/stringdefs.h"
|
2007-04-12 22:39:34 -03:00
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
#include "stringlib/fastsearch.h"
|
|
|
|
#include "stringlib/count.h"
|
|
|
|
#include "stringlib/find.h"
|
2012-10-16 16:07:23 -03:00
|
|
|
#include "stringlib/join.h"
|
2008-05-26 10:42:13 -03:00
|
|
|
#include "stringlib/partition.h"
|
2010-01-13 04:07:53 -04:00
|
|
|
#include "stringlib/split.h"
|
2008-05-26 10:42:13 -03:00
|
|
|
#include "stringlib/ctype.h"
|
2006-04-27 19:54:26 -03:00
|
|
|
|
2009-11-29 21:01:42 -04:00
|
|
|
#include "stringlib/transmogrify.h"
|
2006-04-27 19:54:26 -03:00
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
PyObject *
|
|
|
|
PyBytes_Repr(PyObject *obj, int smartquotes)
|
|
|
|
{
|
2013-08-13 15:18:52 -03:00
|
|
|
PyBytesObject* op = (PyBytesObject*) obj;
|
2011-09-28 02:41:54 -03:00
|
|
|
Py_ssize_t i, length = Py_SIZE(op);
|
2014-09-29 20:12:26 -03:00
|
|
|
Py_ssize_t newsize, squotes, dquotes;
|
2010-05-09 12:52:27 -03:00
|
|
|
PyObject *v;
|
2011-09-28 02:41:54 -03: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++) {
|
2014-09-29 20:01:18 -03:00
|
|
|
Py_ssize_t incr = 1;
|
2011-09-28 02:41:54 -03:00
|
|
|
switch(s[i]) {
|
2014-09-29 20:01:18 -03:00
|
|
|
case '\'': squotes++; break;
|
|
|
|
case '"': dquotes++; break;
|
2011-09-28 02:41:54 -03:00
|
|
|
case '\\': case '\t': case '\n': case '\r':
|
2014-09-29 20:01:18 -03:00
|
|
|
incr = 2; break; /* \C */
|
2011-09-28 02:41:54 -03:00
|
|
|
default:
|
|
|
|
if (s[i] < ' ' || s[i] >= 0x7f)
|
2014-09-29 20:01:18 -03:00
|
|
|
incr = 4; /* \xHH */
|
2011-09-28 02:41:54 -03:00
|
|
|
}
|
2014-09-29 20:01:18 -03:00
|
|
|
if (newsize > PY_SSIZE_T_MAX - incr)
|
|
|
|
goto overflow;
|
|
|
|
newsize += incr;
|
2011-09-28 02:41:54 -03:00
|
|
|
}
|
|
|
|
quote = '\'';
|
|
|
|
if (smartquotes && squotes && !dquotes)
|
|
|
|
quote = '"';
|
2014-09-29 20:01:18 -03:00
|
|
|
if (squotes && quote == '\'') {
|
|
|
|
if (newsize > PY_SSIZE_T_MAX - squotes)
|
|
|
|
goto overflow;
|
2011-09-28 02:41:54 -03:00
|
|
|
newsize += squotes;
|
2010-05-09 12:52:27 -03:00
|
|
|
}
|
2011-09-28 02:41:54 -03:00
|
|
|
|
|
|
|
v = PyUnicode_New(newsize, 127);
|
2010-05-09 12:52:27 -03:00
|
|
|
if (v == NULL) {
|
|
|
|
return NULL;
|
|
|
|
}
|
2011-09-28 02:41:54 -03: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';
|
2011-10-13 21:13:11 -03:00
|
|
|
*p++ = Py_hexdigits[(c & 0xf0) >> 4];
|
|
|
|
*p++ = Py_hexdigits[c & 0xf];
|
2010-05-09 12:52:27 -03:00
|
|
|
}
|
2011-09-28 02:41:54 -03:00
|
|
|
else
|
|
|
|
*p++ = c;
|
2010-05-09 12:52:27 -03:00
|
|
|
}
|
2011-09-28 02:41:54 -03:00
|
|
|
*p++ = quote;
|
2012-04-27 08:55:39 -03:00
|
|
|
assert(_PyUnicode_CheckConsistency(v, 1));
|
2011-09-28 02:41:54 -03:00
|
|
|
return v;
|
2014-09-29 20:01:18 -03:00
|
|
|
|
|
|
|
overflow:
|
|
|
|
PyErr_SetString(PyExc_OverflowError,
|
|
|
|
"bytes object is too large to make repr");
|
|
|
|
return NULL;
|
2006-04-27 19:54:26 -03:00
|
|
|
}
|
|
|
|
|
2006-04-24 10:47:05 -03:00
|
|
|
static PyObject *
|
2009-04-18 12:17:02 -03:00
|
|
|
bytes_repr(PyObject *op)
|
2008-05-26 10:42:13 -03:00
|
|
|
{
|
2010-05-09 12:52:27 -03:00
|
|
|
return PyBytes_Repr(op, 1);
|
2006-04-24 10:47:05 -03:00
|
|
|
}
|
|
|
|
|
2006-04-27 19:54:26 -03:00
|
|
|
static PyObject *
|
2009-04-18 12:17:02 -03:00
|
|
|
bytes_str(PyObject *op)
|
2008-05-26 10:42:13 -03:00
|
|
|
{
|
2010-05-09 12:52:27 -03:00
|
|
|
if (Py_BytesWarningFlag) {
|
|
|
|
if (PyErr_WarnEx(PyExc_BytesWarning,
|
|
|
|
"str() on a bytes instance", 1))
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
return bytes_repr(op);
|
2006-04-27 19:54:26 -03:00
|
|
|
}
|
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
static Py_ssize_t
|
2009-04-18 12:17:02 -03:00
|
|
|
bytes_length(PyBytesObject *a)
|
2006-04-22 20:28:04 -03:00
|
|
|
{
|
2010-05-09 12:52:27 -03:00
|
|
|
return Py_SIZE(a);
|
2006-04-24 10:47:05 -03:00
|
|
|
}
|
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
/* This is also used by PyBytes_Concat() */
|
2006-04-24 10:47:05 -03:00
|
|
|
static PyObject *
|
2009-04-18 12:17:02 -03:00
|
|
|
bytes_concat(PyObject *a, PyObject *b)
|
2008-05-26 10:42:13 -03:00
|
|
|
{
|
2010-05-09 12:52:27 -03:00
|
|
|
Py_ssize_t size;
|
|
|
|
Py_buffer va, vb;
|
|
|
|
PyObject *result = NULL;
|
|
|
|
|
|
|
|
va.len = -1;
|
|
|
|
vb.len = -1;
|
2015-02-02 19:21:08 -04:00
|
|
|
if (PyObject_GetBuffer(a, &va, PyBUF_SIMPLE) != 0 ||
|
|
|
|
PyObject_GetBuffer(b, &vb, PyBUF_SIMPLE) != 0) {
|
2010-05-09 12:52:27 -03:00
|
|
|
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 10:47:05 -03:00
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
done:
|
2010-05-09 12:52:27 -03:00
|
|
|
if (va.len != -1)
|
|
|
|
PyBuffer_Release(&va);
|
|
|
|
if (vb.len != -1)
|
|
|
|
PyBuffer_Release(&vb);
|
|
|
|
return result;
|
2008-05-26 10:42:13 -03:00
|
|
|
}
|
2007-08-24 14:33:11 -03:00
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
static PyObject *
|
2013-08-13 15:18:52 -03:00
|
|
|
bytes_repeat(PyBytesObject *a, Py_ssize_t n)
|
2008-05-26 10:42:13 -03:00
|
|
|
{
|
2013-08-13 15:18:52 -03:00
|
|
|
Py_ssize_t i;
|
|
|
|
Py_ssize_t j;
|
|
|
|
Py_ssize_t size;
|
|
|
|
PyBytesObject *op;
|
2010-05-09 12:52:27 -03:00
|
|
|
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 15:35:01 -03:00
|
|
|
if (n > 0 && Py_SIZE(a) > PY_SSIZE_T_MAX / n) {
|
2010-05-09 12:52:27 -03:00
|
|
|
PyErr_SetString(PyExc_OverflowError,
|
|
|
|
"repeated bytes are too long");
|
|
|
|
return NULL;
|
|
|
|
}
|
2010-08-10 15:35:01 -03:00
|
|
|
size = Py_SIZE(a) * n;
|
2010-05-09 12:52:27 -03: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();
|
2013-12-04 04:27:47 -04:00
|
|
|
(void)PyObject_INIT_VAR(op, &PyBytes_Type, size);
|
2010-05-09 12:52:27 -03:00
|
|
|
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 10:47:05 -03:00
|
|
|
}
|
|
|
|
|
|
|
|
static int
|
2009-04-18 12:17:02 -03:00
|
|
|
bytes_contains(PyObject *self, PyObject *arg)
|
2006-04-24 10:47:05 -03:00
|
|
|
{
|
2008-05-26 10:42:13 -03:00
|
|
|
Py_ssize_t ival = PyNumber_AsSsize_t(arg, PyExc_ValueError);
|
|
|
|
if (ival == -1 && PyErr_Occurred()) {
|
2010-06-09 13:38:55 -03:00
|
|
|
Py_buffer varg;
|
2010-08-15 14:12:55 -03:00
|
|
|
Py_ssize_t pos;
|
2010-06-09 13:38:55 -03:00
|
|
|
PyErr_Clear();
|
2015-02-02 19:21:08 -04:00
|
|
|
if (PyObject_GetBuffer(arg, &varg, PyBUF_SIMPLE) != 0)
|
2010-06-09 13:38:55 -03:00
|
|
|
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 10:47:05 -03:00
|
|
|
}
|
|
|
|
if (ival < 0 || ival >= 256) {
|
2010-06-09 13:38:55 -03:00
|
|
|
PyErr_SetString(PyExc_ValueError, "byte must be in range(0, 256)");
|
|
|
|
return -1;
|
2006-04-24 10:47:05 -03:00
|
|
|
}
|
|
|
|
|
2010-08-15 14:12:55 -03:00
|
|
|
return memchr(PyBytes_AS_STRING(self), (int) ival, Py_SIZE(self)) != NULL;
|
2008-05-26 10:42:13 -03:00
|
|
|
}
|
|
|
|
|
|
|
|
static PyObject *
|
2013-08-13 15:18:52 -03:00
|
|
|
bytes_item(PyBytesObject *a, Py_ssize_t i)
|
2008-05-26 10:42:13 -03:00
|
|
|
{
|
2010-05-09 12:52:27 -03: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 10:42:13 -03:00
|
|
|
}
|
|
|
|
|
2013-11-04 06:08:10 -04:00
|
|
|
Py_LOCAL(int)
|
|
|
|
bytes_compare_eq(PyBytesObject *a, PyBytesObject *b)
|
|
|
|
{
|
|
|
|
int cmp;
|
|
|
|
Py_ssize_t len;
|
|
|
|
|
|
|
|
len = Py_SIZE(a);
|
|
|
|
if (Py_SIZE(b) != len)
|
|
|
|
return 0;
|
|
|
|
|
|
|
|
if (a->ob_sval[0] != b->ob_sval[0])
|
|
|
|
return 0;
|
|
|
|
|
|
|
|
cmp = memcmp(a->ob_sval, b->ob_sval, len);
|
|
|
|
return (cmp == 0);
|
|
|
|
}
|
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
static PyObject*
|
2009-04-18 12:17:02 -03:00
|
|
|
bytes_richcompare(PyBytesObject *a, PyBytesObject *b, int op)
|
2008-05-26 10:42:13 -03:00
|
|
|
{
|
2010-05-09 12:52:27 -03: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;
|
|
|
|
}
|
2013-11-04 06:08:10 -04:00
|
|
|
else if (a == b) {
|
2010-05-09 12:52:27 -03:00
|
|
|
switch (op) {
|
2013-11-04 06:23:05 -04:00
|
|
|
case Py_EQ:
|
|
|
|
case Py_LE:
|
|
|
|
case Py_GE:
|
|
|
|
/* a string is equal to itself */
|
2010-05-09 12:52:27 -03:00
|
|
|
result = Py_True;
|
2013-11-04 06:08:10 -04:00
|
|
|
break;
|
2013-11-04 06:23:05 -04:00
|
|
|
case Py_NE:
|
|
|
|
case Py_LT:
|
|
|
|
case Py_GT:
|
2010-05-09 12:52:27 -03:00
|
|
|
result = Py_False;
|
2013-11-04 06:08:10 -04:00
|
|
|
break;
|
2013-11-04 06:23:05 -04:00
|
|
|
default:
|
|
|
|
PyErr_BadArgument();
|
|
|
|
return NULL;
|
2010-05-09 12:52:27 -03:00
|
|
|
}
|
|
|
|
}
|
2013-11-04 06:08:10 -04:00
|
|
|
else if (op == Py_EQ || op == Py_NE) {
|
|
|
|
int eq = bytes_compare_eq(a, b);
|
|
|
|
eq ^= (op == Py_NE);
|
|
|
|
result = eq ? Py_True : Py_False;
|
|
|
|
}
|
|
|
|
else {
|
2013-11-04 06:23:05 -04:00
|
|
|
len_a = Py_SIZE(a);
|
|
|
|
len_b = Py_SIZE(b);
|
|
|
|
min_len = Py_MIN(len_a, len_b);
|
2013-11-04 06:08:10 -04:00
|
|
|
if (min_len > 0) {
|
|
|
|
c = Py_CHARMASK(*a->ob_sval) - Py_CHARMASK(*b->ob_sval);
|
2013-11-04 06:23:05 -04:00
|
|
|
if (c == 0)
|
2013-11-04 06:08:10 -04:00
|
|
|
c = memcmp(a->ob_sval, b->ob_sval, min_len);
|
2010-05-09 12:52:27 -03:00
|
|
|
}
|
2013-11-04 06:08:10 -04:00
|
|
|
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_GT: c = c > 0; break;
|
|
|
|
case Py_GE: c = c >= 0; break;
|
|
|
|
default:
|
2013-11-04 06:23:05 -04:00
|
|
|
PyErr_BadArgument();
|
|
|
|
return NULL;
|
2013-11-04 06:08:10 -04:00
|
|
|
}
|
|
|
|
result = c ? Py_True : Py_False;
|
2010-05-09 12:52:27 -03:00
|
|
|
}
|
2013-11-04 06:08:10 -04:00
|
|
|
|
2010-05-09 12:52:27 -03:00
|
|
|
Py_INCREF(result);
|
|
|
|
return result;
|
2008-05-26 10:42:13 -03:00
|
|
|
}
|
|
|
|
|
2010-10-17 17:54:53 -03:00
|
|
|
static Py_hash_t
|
2009-04-18 12:17:02 -03:00
|
|
|
bytes_hash(PyBytesObject *a)
|
2008-05-26 10:42:13 -03:00
|
|
|
{
|
2011-11-21 15:46:33 -04:00
|
|
|
if (a->ob_shash == -1) {
|
|
|
|
/* Can't fail */
|
2013-11-20 06:46:18 -04:00
|
|
|
a->ob_shash = _Py_HashBytes(a->ob_sval, Py_SIZE(a));
|
2011-11-21 15:46:33 -04:00
|
|
|
}
|
|
|
|
return a->ob_shash;
|
2008-05-26 10:42:13 -03:00
|
|
|
}
|
|
|
|
|
|
|
|
static PyObject*
|
2009-04-18 12:17:02 -03:00
|
|
|
bytes_subscript(PyBytesObject* self, PyObject* item)
|
2008-05-26 10:42:13 -03:00
|
|
|
{
|
2010-05-09 12:52:27 -03: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 16:14:31 -04:00
|
|
|
if (PySlice_GetIndicesEx(item,
|
2010-05-09 12:52:27 -03: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 20:28:04 -03:00
|
|
|
}
|
|
|
|
|
2006-12-19 04:30:14 -04:00
|
|
|
static int
|
2009-04-18 12:17:02 -03:00
|
|
|
bytes_buffer_getbuffer(PyBytesObject *self, Py_buffer *view, int flags)
|
2006-12-19 04:30:14 -04:00
|
|
|
{
|
2010-05-09 12:52:27 -03:00
|
|
|
return PyBuffer_FillInfo(view, (PyObject*)self, (void *)self->ob_sval, Py_SIZE(self),
|
|
|
|
1, flags);
|
2008-05-26 10:42:13 -03:00
|
|
|
}
|
2007-04-12 19:49:52 -03:00
|
|
|
|
2009-04-18 12:17:02 -03:00
|
|
|
static PySequenceMethods bytes_as_sequence = {
|
2010-05-09 12:52:27 -03: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 10:42:13 -03:00
|
|
|
};
|
2006-12-19 04:30:14 -04:00
|
|
|
|
2009-04-18 12:17:02 -03:00
|
|
|
static PyMappingMethods bytes_as_mapping = {
|
2010-05-09 12:52:27 -03:00
|
|
|
(lenfunc)bytes_length,
|
|
|
|
(binaryfunc)bytes_subscript,
|
|
|
|
0,
|
2008-05-26 10:42:13 -03:00
|
|
|
};
|
2006-12-19 04:30:14 -04:00
|
|
|
|
2009-04-18 12:17:02 -03:00
|
|
|
static PyBufferProcs bytes_as_buffer = {
|
2010-05-09 12:52:27 -03:00
|
|
|
(getbufferproc)bytes_buffer_getbuffer,
|
|
|
|
NULL,
|
2008-05-26 10:42:13 -03:00
|
|
|
};
|
2006-12-19 04:30:14 -04:00
|
|
|
|
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
#define LEFTSTRIP 0
|
|
|
|
#define RIGHTSTRIP 1
|
|
|
|
#define BOTHSTRIP 2
|
2006-12-19 04:30:14 -04:00
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
/* Arrays indexed by above */
|
|
|
|
static const char *stripformat[] = {"|O:lstrip", "|O:rstrip", "|O:strip"};
|
2006-04-24 10:47:05 -03:00
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
#define STRIPNAME(i) (stripformat[i]+3)
|
2006-04-24 10:47:05 -03:00
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
PyDoc_STRVAR(split__doc__,
|
2012-02-26 03:39:55 -04:00
|
|
|
"B.split(sep=None, maxsplit=-1) -> list of bytes\n\
|
2008-05-26 10:42:13 -03:00
|
|
|
\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 15:02:19 -04:00
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
static PyObject *
|
2012-02-26 03:39:55 -04:00
|
|
|
bytes_split(PyBytesObject *self, PyObject *args, PyObject *kwds)
|
2007-02-27 15:02:19 -04:00
|
|
|
{
|
2012-02-26 03:39:55 -04:00
|
|
|
static char *kwlist[] = {"sep", "maxsplit", 0};
|
2010-05-09 12:52:27 -03: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 15:02:19 -04:00
|
|
|
|
2012-02-26 03:39:55 -04:00
|
|
|
if (!PyArg_ParseTupleAndKeywords(args, kwds, "|On:split",
|
|
|
|
kwlist, &subobj, &maxsplit))
|
2010-05-09 12:52:27 -03:00
|
|
|
return NULL;
|
|
|
|
if (maxsplit < 0)
|
|
|
|
maxsplit = PY_SSIZE_T_MAX;
|
|
|
|
if (subobj == Py_None)
|
|
|
|
return stringlib_split_whitespace((PyObject*) self, s, len, maxsplit);
|
2015-02-02 19:21:08 -04:00
|
|
|
if (PyObject_GetBuffer(subobj, &vsub, PyBUF_SIMPLE) != 0)
|
2010-05-09 12:52:27 -03:00
|
|
|
return NULL;
|
|
|
|
sub = vsub.buf;
|
|
|
|
n = vsub.len;
|
2008-05-26 10:42:13 -03:00
|
|
|
|
2010-05-09 12:52:27 -03:00
|
|
|
list = stringlib_split((PyObject*) self, s, len, sub, n, maxsplit);
|
|
|
|
PyBuffer_Release(&vsub);
|
|
|
|
return list;
|
2007-02-27 15:02:19 -04:00
|
|
|
}
|
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
PyDoc_STRVAR(partition__doc__,
|
|
|
|
"B.partition(sep) -> (head, sep, tail)\n\
|
2007-02-27 15:02:19 -04:00
|
|
|
\n\
|
2008-05-30 05:20:09 -03:00
|
|
|
Search for the separator sep in B, and return the part before it,\n\
|
2008-05-26 10:42:13 -03: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 15:02:19 -04:00
|
|
|
|
|
|
|
static PyObject *
|
2009-04-18 12:17:02 -03:00
|
|
|
bytes_partition(PyBytesObject *self, PyObject *sep_obj)
|
2007-02-27 15:02:19 -04:00
|
|
|
{
|
2015-02-02 19:21:08 -04:00
|
|
|
Py_buffer sep = {NULL, NULL};
|
|
|
|
PyObject *res;
|
2007-02-27 15:02:19 -04:00
|
|
|
|
2015-02-02 19:21:08 -04:00
|
|
|
if (PyObject_GetBuffer(sep_obj, &sep, PyBUF_SIMPLE) != 0)
|
2010-05-09 12:52:27 -03:00
|
|
|
return NULL;
|
2007-02-27 15:02:19 -04:00
|
|
|
|
2015-02-02 19:21:08 -04:00
|
|
|
res = stringlib_partition(
|
2010-05-09 12:52:27 -03:00
|
|
|
(PyObject*) self,
|
|
|
|
PyBytes_AS_STRING(self), PyBytes_GET_SIZE(self),
|
2015-02-02 19:21:08 -04:00
|
|
|
sep_obj, sep.buf, sep.len
|
2010-05-09 12:52:27 -03:00
|
|
|
);
|
2015-02-02 19:21:08 -04:00
|
|
|
PyBuffer_Release(&sep);
|
|
|
|
return res;
|
2007-02-27 15:02:19 -04:00
|
|
|
}
|
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
PyDoc_STRVAR(rpartition__doc__,
|
2010-01-25 07:58:28 -04:00
|
|
|
"B.rpartition(sep) -> (head, sep, tail)\n\
|
2007-02-27 15:02:19 -04:00
|
|
|
\n\
|
2008-05-30 05:20:09 -03: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 10:42:13 -03:00
|
|
|
part after it. If the separator is not found, returns two empty\n\
|
|
|
|
bytes objects and B.");
|
2007-02-27 15:02:19 -04:00
|
|
|
|
|
|
|
static PyObject *
|
2009-04-18 12:17:02 -03:00
|
|
|
bytes_rpartition(PyBytesObject *self, PyObject *sep_obj)
|
2007-02-27 15:02:19 -04:00
|
|
|
{
|
2015-02-02 19:21:08 -04:00
|
|
|
Py_buffer sep = {NULL, NULL};
|
|
|
|
PyObject *res;
|
2007-02-27 15:02:19 -04:00
|
|
|
|
2015-02-02 19:21:08 -04:00
|
|
|
if (PyObject_GetBuffer(sep_obj, &sep, PyBUF_SIMPLE) != 0)
|
2010-05-09 12:52:27 -03:00
|
|
|
return NULL;
|
2007-02-27 15:02:19 -04:00
|
|
|
|
2015-02-02 19:21:08 -04:00
|
|
|
res = stringlib_rpartition(
|
2010-05-09 12:52:27 -03:00
|
|
|
(PyObject*) self,
|
|
|
|
PyBytes_AS_STRING(self), PyBytes_GET_SIZE(self),
|
2015-02-02 19:21:08 -04:00
|
|
|
sep_obj, sep.buf, sep.len
|
2010-05-09 12:52:27 -03:00
|
|
|
);
|
2015-02-02 19:21:08 -04:00
|
|
|
PyBuffer_Release(&sep);
|
|
|
|
return res;
|
2007-02-27 15:02:19 -04:00
|
|
|
}
|
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
PyDoc_STRVAR(rsplit__doc__,
|
2012-02-26 03:39:55 -04:00
|
|
|
"B.rsplit(sep=None, maxsplit=-1) -> list of bytes\n\
|
2007-02-27 15:02:19 -04:00
|
|
|
\n\
|
2008-05-26 10:42:13 -03: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 15:02:19 -04:00
|
|
|
|
|
|
|
static PyObject *
|
2012-02-26 03:39:55 -04:00
|
|
|
bytes_rsplit(PyBytesObject *self, PyObject *args, PyObject *kwds)
|
2008-05-26 10:42:13 -03:00
|
|
|
{
|
2012-02-26 03:39:55 -04:00
|
|
|
static char *kwlist[] = {"sep", "maxsplit", 0};
|
2010-05-09 12:52:27 -03: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 10:42:13 -03:00
|
|
|
|
2012-02-26 03:39:55 -04:00
|
|
|
if (!PyArg_ParseTupleAndKeywords(args, kwds, "|On:rsplit",
|
|
|
|
kwlist, &subobj, &maxsplit))
|
2010-05-09 12:52:27 -03:00
|
|
|
return NULL;
|
|
|
|
if (maxsplit < 0)
|
|
|
|
maxsplit = PY_SSIZE_T_MAX;
|
|
|
|
if (subobj == Py_None)
|
|
|
|
return stringlib_rsplit_whitespace((PyObject*) self, s, len, maxsplit);
|
2015-02-02 19:21:08 -04:00
|
|
|
if (PyObject_GetBuffer(subobj, &vsub, PyBUF_SIMPLE) != 0)
|
2010-05-09 12:52:27 -03:00
|
|
|
return NULL;
|
|
|
|
sub = vsub.buf;
|
|
|
|
n = vsub.len;
|
2008-05-26 10:42:13 -03:00
|
|
|
|
2010-05-09 12:52:27 -03:00
|
|
|
list = stringlib_rsplit((PyObject*) self, s, len, sub, n, maxsplit);
|
|
|
|
PyBuffer_Release(&vsub);
|
|
|
|
return list;
|
2007-02-27 15:02:19 -04:00
|
|
|
}
|
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
|
|
|
|
PyDoc_STRVAR(join__doc__,
|
|
|
|
"B.join(iterable_of_bytes) -> bytes\n\
|
2007-02-27 15:02:19 -04:00
|
|
|
\n\
|
2008-05-30 05:20:09 -03:00
|
|
|
Concatenate any number of bytes objects, with B in between each pair.\n\
|
2008-05-26 10:42:13 -03:00
|
|
|
Example: b'.'.join([b'ab', b'pq', b'rs']) -> b'ab.pq.rs'.");
|
2007-02-27 15:02:19 -04:00
|
|
|
|
|
|
|
static PyObject *
|
2012-10-16 16:07:23 -03:00
|
|
|
bytes_join(PyObject *self, PyObject *iterable)
|
2008-05-26 10:42:13 -03:00
|
|
|
{
|
2012-10-16 16:07:23 -03:00
|
|
|
return stringlib_bytes_join(self, iterable);
|
2007-02-27 15:02:19 -04:00
|
|
|
}
|
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
PyObject *
|
|
|
|
_PyBytes_Join(PyObject *sep, PyObject *x)
|
2007-02-27 15:02:19 -04:00
|
|
|
{
|
2010-05-09 12:52:27 -03:00
|
|
|
assert(sep != NULL && PyBytes_Check(sep));
|
|
|
|
assert(x != NULL);
|
|
|
|
return bytes_join(sep, x);
|
2007-02-27 15:02:19 -04:00
|
|
|
}
|
|
|
|
|
2010-01-13 04:07:53 -04:00
|
|
|
/* helper macro to fixup start/end slice values */
|
|
|
|
#define ADJUST_INDICES(start, end, len) \
|
2010-05-09 12:52:27 -03: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 15:02:19 -04:00
|
|
|
|
|
|
|
Py_LOCAL_INLINE(Py_ssize_t)
|
2009-04-18 12:17:02 -03:00
|
|
|
bytes_find_internal(PyBytesObject *self, PyObject *args, int dir)
|
2008-05-26 10:42:13 -03:00
|
|
|
{
|
2010-05-09 12:52:27 -03:00
|
|
|
PyObject *subobj;
|
2011-10-20 18:54:17 -03:00
|
|
|
char byte;
|
|
|
|
Py_buffer subbuf;
|
2010-05-09 12:52:27 -03:00
|
|
|
const char *sub;
|
|
|
|
Py_ssize_t sub_len;
|
|
|
|
Py_ssize_t start=0, end=PY_SSIZE_T_MAX;
|
2011-10-20 18:54:17 -03:00
|
|
|
Py_ssize_t res;
|
2010-05-09 12:52:27 -03:00
|
|
|
|
2011-10-20 18:54:17 -03:00
|
|
|
if (!stringlib_parse_args_finds_byte("find/rfind/index/rindex",
|
|
|
|
args, &subobj, &byte, &start, &end))
|
2010-05-09 12:52:27 -03:00
|
|
|
return -2;
|
|
|
|
|
2011-10-20 18:54:17 -03:00
|
|
|
if (subobj) {
|
2015-02-02 19:21:08 -04:00
|
|
|
if (PyObject_GetBuffer(subobj, &subbuf, PyBUF_SIMPLE) != 0)
|
2011-10-20 18:54:17 -03:00
|
|
|
return -2;
|
|
|
|
|
|
|
|
sub = subbuf.buf;
|
|
|
|
sub_len = subbuf.len;
|
|
|
|
}
|
|
|
|
else {
|
|
|
|
sub = &byte;
|
|
|
|
sub_len = 1;
|
2010-05-09 12:52:27 -03:00
|
|
|
}
|
|
|
|
|
|
|
|
if (dir > 0)
|
2011-10-20 18:54:17 -03:00
|
|
|
res = stringlib_find_slice(
|
2010-05-09 12:52:27 -03:00
|
|
|
PyBytes_AS_STRING(self), PyBytes_GET_SIZE(self),
|
|
|
|
sub, sub_len, start, end);
|
|
|
|
else
|
2011-10-20 18:54:17 -03:00
|
|
|
res = stringlib_rfind_slice(
|
2010-05-09 12:52:27 -03:00
|
|
|
PyBytes_AS_STRING(self), PyBytes_GET_SIZE(self),
|
|
|
|
sub, sub_len, start, end);
|
2011-10-20 18:54:17 -03:00
|
|
|
|
|
|
|
if (subobj)
|
|
|
|
PyBuffer_Release(&subbuf);
|
|
|
|
|
|
|
|
return res;
|
2007-02-27 15:02:19 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
PyDoc_STRVAR(find__doc__,
|
2008-05-30 05:20:09 -03:00
|
|
|
"B.find(sub[, start[, end]]) -> int\n\
|
2008-05-26 10:42:13 -03:00
|
|
|
\n\
|
2010-01-13 04:07:53 -04:00
|
|
|
Return the lowest index in B where substring sub is found,\n\
|
2011-07-27 12:33:54 -03:00
|
|
|
such that sub is contained within B[start:end]. Optional\n\
|
2008-05-26 10:42:13 -03:00
|
|
|
arguments start and end are interpreted as in slice notation.\n\
|
|
|
|
\n\
|
|
|
|
Return -1 on failure.");
|
2007-02-27 15:02:19 -04:00
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
static PyObject *
|
2009-04-18 12:17:02 -03:00
|
|
|
bytes_find(PyBytesObject *self, PyObject *args)
|
2008-05-26 10:42:13 -03:00
|
|
|
{
|
2010-05-09 12:52:27 -03:00
|
|
|
Py_ssize_t result = bytes_find_internal(self, args, +1);
|
|
|
|
if (result == -2)
|
|
|
|
return NULL;
|
|
|
|
return PyLong_FromSsize_t(result);
|
2007-02-27 15:02:19 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
PyDoc_STRVAR(index__doc__,
|
2008-05-30 05:20:09 -03:00
|
|
|
"B.index(sub[, start[, end]]) -> int\n\
|
2008-05-26 10:42:13 -03:00
|
|
|
\n\
|
|
|
|
Like B.find() but raise ValueError when the substring is not found.");
|
2007-02-27 15:02:19 -04:00
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
static PyObject *
|
2009-04-18 12:17:02 -03:00
|
|
|
bytes_index(PyBytesObject *self, PyObject *args)
|
2008-05-26 10:42:13 -03:00
|
|
|
{
|
2010-05-09 12:52:27 -03: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 15:02:19 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
PyDoc_STRVAR(rfind__doc__,
|
2008-05-30 05:20:09 -03:00
|
|
|
"B.rfind(sub[, start[, end]]) -> int\n\
|
2008-05-26 10:42:13 -03:00
|
|
|
\n\
|
|
|
|
Return the highest index in B where substring sub is found,\n\
|
2011-07-27 12:33:54 -03:00
|
|
|
such that sub is contained within B[start:end]. Optional\n\
|
2008-05-26 10:42:13 -03:00
|
|
|
arguments start and end are interpreted as in slice notation.\n\
|
|
|
|
\n\
|
|
|
|
Return -1 on failure.");
|
2007-02-27 15:02:19 -04:00
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
static PyObject *
|
2009-04-18 12:17:02 -03:00
|
|
|
bytes_rfind(PyBytesObject *self, PyObject *args)
|
2008-05-26 10:42:13 -03:00
|
|
|
{
|
2010-05-09 12:52:27 -03:00
|
|
|
Py_ssize_t result = bytes_find_internal(self, args, -1);
|
|
|
|
if (result == -2)
|
|
|
|
return NULL;
|
|
|
|
return PyLong_FromSsize_t(result);
|
2007-02-27 15:02:19 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
PyDoc_STRVAR(rindex__doc__,
|
2008-05-30 05:20:09 -03:00
|
|
|
"B.rindex(sub[, start[, end]]) -> int\n\
|
2008-05-26 10:42:13 -03:00
|
|
|
\n\
|
|
|
|
Like B.rfind() but raise ValueError when the substring is not found.");
|
2007-02-27 15:02:19 -04:00
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
static PyObject *
|
2009-04-18 12:17:02 -03:00
|
|
|
bytes_rindex(PyBytesObject *self, PyObject *args)
|
2008-05-26 10:42:13 -03:00
|
|
|
{
|
2010-05-09 12:52:27 -03: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 10:42:13 -03:00
|
|
|
}
|
2007-02-27 15:02:19 -04:00
|
|
|
|
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
Py_LOCAL_INLINE(PyObject *)
|
|
|
|
do_xstrip(PyBytesObject *self, int striptype, PyObject *sepobj)
|
|
|
|
{
|
2010-05-09 12:52:27 -03: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 10:42:13 -03:00
|
|
|
|
2015-02-02 19:21:08 -04:00
|
|
|
if (PyObject_GetBuffer(sepobj, &vsep, PyBUF_SIMPLE) != 0)
|
2010-05-09 12:52:27 -03:00
|
|
|
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 15:02:19 -04:00
|
|
|
|
2010-05-09 12:52:27 -03:00
|
|
|
j = len;
|
|
|
|
if (striptype != LEFTSTRIP) {
|
|
|
|
do {
|
|
|
|
j--;
|
|
|
|
} while (j >= i && memchr(sep, Py_CHARMASK(s[j]), seplen));
|
|
|
|
j++;
|
|
|
|
}
|
2007-02-27 15:02:19 -04:00
|
|
|
|
2010-05-09 12:52:27 -03:00
|
|
|
PyBuffer_Release(&vsep);
|
2007-02-27 15:02:19 -04:00
|
|
|
|
2010-05-09 12:52:27 -03: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 15:02:19 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
Py_LOCAL_INLINE(PyObject *)
|
|
|
|
do_strip(PyBytesObject *self, int striptype)
|
|
|
|
{
|
2010-05-09 12:52:27 -03:00
|
|
|
char *s = PyBytes_AS_STRING(self);
|
|
|
|
Py_ssize_t len = PyBytes_GET_SIZE(self), i, j;
|
2007-02-27 15:02:19 -04:00
|
|
|
|
2010-05-09 12:52:27 -03:00
|
|
|
i = 0;
|
|
|
|
if (striptype != RIGHTSTRIP) {
|
2010-11-05 14:23:41 -03:00
|
|
|
while (i < len && Py_ISSPACE(s[i])) {
|
2010-05-09 12:52:27 -03:00
|
|
|
i++;
|
|
|
|
}
|
|
|
|
}
|
2007-02-27 15:02:19 -04:00
|
|
|
|
2010-05-09 12:52:27 -03:00
|
|
|
j = len;
|
|
|
|
if (striptype != LEFTSTRIP) {
|
|
|
|
do {
|
|
|
|
j--;
|
2010-11-05 14:23:41 -03:00
|
|
|
} while (j >= i && Py_ISSPACE(s[j]));
|
2010-05-09 12:52:27 -03:00
|
|
|
j++;
|
|
|
|
}
|
2007-02-27 15:02:19 -04:00
|
|
|
|
2010-05-09 12:52:27 -03: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 15:02:19 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
Py_LOCAL_INLINE(PyObject *)
|
|
|
|
do_argstrip(PyBytesObject *self, int striptype, PyObject *args)
|
|
|
|
{
|
2010-05-09 12:52:27 -03:00
|
|
|
PyObject *sep = NULL;
|
2007-02-27 15:02:19 -04:00
|
|
|
|
2013-10-19 15:03:34 -03:00
|
|
|
if (!PyArg_ParseTuple(args, stripformat[striptype], &sep))
|
2010-05-09 12:52:27 -03:00
|
|
|
return NULL;
|
2007-02-27 15:02:19 -04:00
|
|
|
|
2010-05-09 12:52:27 -03:00
|
|
|
if (sep != NULL && sep != Py_None) {
|
|
|
|
return do_xstrip(self, striptype, sep);
|
|
|
|
}
|
|
|
|
return do_strip(self, striptype);
|
2007-02-27 15:02:19 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
PyDoc_STRVAR(strip__doc__,
|
|
|
|
"B.strip([bytes]) -> bytes\n\
|
|
|
|
\n\
|
|
|
|
Strip leading and trailing bytes contained in the argument.\n\
|
2012-01-22 16:31:21 -04:00
|
|
|
If the argument is omitted, strip leading and trailing ASCII whitespace.");
|
2008-05-26 10:42:13 -03:00
|
|
|
static PyObject *
|
2009-04-18 12:17:02 -03:00
|
|
|
bytes_strip(PyBytesObject *self, PyObject *args)
|
2007-02-27 15:02:19 -04:00
|
|
|
{
|
2010-05-09 12:52:27 -03:00
|
|
|
if (PyTuple_GET_SIZE(args) == 0)
|
|
|
|
return do_strip(self, BOTHSTRIP); /* Common case */
|
|
|
|
else
|
|
|
|
return do_argstrip(self, BOTHSTRIP, args);
|
2008-05-26 10:42:13 -03:00
|
|
|
}
|
2007-02-27 15:02:19 -04:00
|
|
|
|
|
|
|
|
2008-05-26 10:42:13 -03: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 12:17:02 -03:00
|
|
|
bytes_lstrip(PyBytesObject *self, PyObject *args)
|
2008-05-26 10:42:13 -03:00
|
|
|
{
|
2010-05-09 12:52:27 -03:00
|
|
|
if (PyTuple_GET_SIZE(args) == 0)
|
|
|
|
return do_strip(self, LEFTSTRIP); /* Common case */
|
|
|
|
else
|
|
|
|
return do_argstrip(self, LEFTSTRIP, args);
|
2008-05-26 10:42:13 -03:00
|
|
|
}
|
2007-02-27 15:02:19 -04:00
|
|
|
|
|
|
|
|
2008-05-26 10:42:13 -03: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 12:17:02 -03:00
|
|
|
bytes_rstrip(PyBytesObject *self, PyObject *args)
|
2008-05-26 10:42:13 -03:00
|
|
|
{
|
2010-05-09 12:52:27 -03:00
|
|
|
if (PyTuple_GET_SIZE(args) == 0)
|
|
|
|
return do_strip(self, RIGHTSTRIP); /* Common case */
|
|
|
|
else
|
|
|
|
return do_argstrip(self, RIGHTSTRIP, args);
|
2007-02-27 15:02:19 -04:00
|
|
|
}
|
|
|
|
|
2007-10-16 03:31:30 -03:00
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
PyDoc_STRVAR(count__doc__,
|
2008-05-30 05:20:09 -03:00
|
|
|
"B.count(sub[, start[, end]]) -> int\n\
|
2007-02-27 15:02:19 -04:00
|
|
|
\n\
|
2008-05-26 10:42:13 -03:00
|
|
|
Return the number of non-overlapping occurrences of substring sub in\n\
|
2010-01-13 04:07:53 -04:00
|
|
|
string B[start:end]. Optional arguments start and end are interpreted\n\
|
2008-05-26 10:42:13 -03:00
|
|
|
as in slice notation.");
|
2007-02-27 15:02:19 -04:00
|
|
|
|
|
|
|
static PyObject *
|
2009-04-18 12:17:02 -03:00
|
|
|
bytes_count(PyBytesObject *self, PyObject *args)
|
2007-02-27 15:02:19 -04:00
|
|
|
{
|
2010-05-09 12:52:27 -03:00
|
|
|
PyObject *sub_obj;
|
|
|
|
const char *str = PyBytes_AS_STRING(self), *sub;
|
|
|
|
Py_ssize_t sub_len;
|
2011-10-20 18:54:17 -03:00
|
|
|
char byte;
|
2010-05-09 12:52:27 -03:00
|
|
|
Py_ssize_t start = 0, end = PY_SSIZE_T_MAX;
|
2007-02-27 15:02:19 -04:00
|
|
|
|
2011-10-20 18:54:17 -03:00
|
|
|
Py_buffer vsub;
|
|
|
|
PyObject *count_obj;
|
|
|
|
|
|
|
|
if (!stringlib_parse_args_finds_byte("count", args, &sub_obj, &byte,
|
|
|
|
&start, &end))
|
2010-05-09 12:52:27 -03:00
|
|
|
return NULL;
|
2007-02-27 15:02:19 -04:00
|
|
|
|
2011-10-20 18:54:17 -03:00
|
|
|
if (sub_obj) {
|
2015-02-02 19:21:08 -04:00
|
|
|
if (PyObject_GetBuffer(sub_obj, &vsub, PyBUF_SIMPLE) != 0)
|
2011-10-20 18:54:17 -03:00
|
|
|
return NULL;
|
|
|
|
|
|
|
|
sub = vsub.buf;
|
|
|
|
sub_len = vsub.len;
|
|
|
|
}
|
|
|
|
else {
|
|
|
|
sub = &byte;
|
|
|
|
sub_len = 1;
|
2010-05-09 12:52:27 -03:00
|
|
|
}
|
2007-02-27 15:02:19 -04:00
|
|
|
|
2010-05-09 12:52:27 -03:00
|
|
|
ADJUST_INDICES(start, end, PyBytes_GET_SIZE(self));
|
2007-08-18 08:21:56 -03:00
|
|
|
|
2011-10-20 18:54:17 -03:00
|
|
|
count_obj = PyLong_FromSsize_t(
|
2010-05-09 12:52:27 -03:00
|
|
|
stringlib_count(str + start, end - start, sub, sub_len, PY_SSIZE_T_MAX)
|
|
|
|
);
|
2011-10-20 18:54:17 -03:00
|
|
|
|
|
|
|
if (sub_obj)
|
|
|
|
PyBuffer_Release(&vsub);
|
|
|
|
|
|
|
|
return count_obj;
|
2007-02-27 15:02:19 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
|
2008-05-26 10:42:13 -03: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 15:02:19 -04:00
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
static PyObject *
|
2009-04-18 12:17:02 -03:00
|
|
|
bytes_translate(PyBytesObject *self, PyObject *args)
|
2008-05-26 10:42:13 -03:00
|
|
|
{
|
2013-08-13 15:18:52 -03:00
|
|
|
char *input, *output;
|
2015-02-02 19:21:08 -04:00
|
|
|
Py_buffer table_view = {NULL, NULL};
|
|
|
|
Py_buffer del_table_view = {NULL, NULL};
|
2010-05-09 12:52:27 -03:00
|
|
|
const char *table;
|
2013-08-13 15:18:52 -03:00
|
|
|
Py_ssize_t i, c, changed = 0;
|
2010-05-09 12:52:27 -03:00
|
|
|
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;
|
|
|
|
}
|
2015-02-02 19:21:08 -04:00
|
|
|
else {
|
|
|
|
if (PyObject_GetBuffer(tableobj, &table_view, PyBUF_SIMPLE) != 0)
|
|
|
|
return NULL;
|
|
|
|
table = table_view.buf;
|
|
|
|
tablen = table_view.len;
|
|
|
|
}
|
2010-05-09 12:52:27 -03:00
|
|
|
|
|
|
|
if (tablen != 256) {
|
|
|
|
PyErr_SetString(PyExc_ValueError,
|
|
|
|
"translation table must be 256 characters long");
|
2015-02-02 19:21:08 -04:00
|
|
|
PyBuffer_Release(&table_view);
|
2010-05-09 12:52:27 -03:00
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (delobj != NULL) {
|
|
|
|
if (PyBytes_Check(delobj)) {
|
|
|
|
del_table = PyBytes_AS_STRING(delobj);
|
|
|
|
dellen = PyBytes_GET_SIZE(delobj);
|
|
|
|
}
|
2015-02-02 19:21:08 -04:00
|
|
|
else {
|
|
|
|
if (PyObject_GetBuffer(delobj, &del_table_view, PyBUF_SIMPLE) != 0) {
|
|
|
|
PyBuffer_Release(&table_view);
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
del_table = del_table_view.buf;
|
|
|
|
dellen = del_table_view.len;
|
|
|
|
}
|
2010-05-09 12:52:27 -03:00
|
|
|
}
|
|
|
|
else {
|
|
|
|
del_table = NULL;
|
|
|
|
dellen = 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
inlen = PyBytes_GET_SIZE(input_obj);
|
|
|
|
result = PyBytes_FromStringAndSize((char *)NULL, inlen);
|
2015-02-02 19:21:08 -04:00
|
|
|
if (result == NULL) {
|
|
|
|
PyBuffer_Release(&del_table_view);
|
|
|
|
PyBuffer_Release(&table_view);
|
2010-05-09 12:52:27 -03:00
|
|
|
return NULL;
|
2015-02-02 19:21:08 -04:00
|
|
|
}
|
2010-05-09 12:52:27 -03:00
|
|
|
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;
|
|
|
|
}
|
2015-02-02 19:21:08 -04:00
|
|
|
if (!changed && PyBytes_CheckExact(input_obj)) {
|
|
|
|
Py_INCREF(input_obj);
|
|
|
|
Py_DECREF(result);
|
|
|
|
result = input_obj;
|
|
|
|
}
|
|
|
|
PyBuffer_Release(&del_table_view);
|
|
|
|
PyBuffer_Release(&table_view);
|
|
|
|
return result;
|
2010-05-09 12:52:27 -03:00
|
|
|
}
|
|
|
|
|
|
|
|
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]);
|
|
|
|
}
|
2015-02-02 19:21:08 -04:00
|
|
|
PyBuffer_Release(&table_view);
|
2010-05-09 12:52:27 -03:00
|
|
|
|
|
|
|
for (i = 0; i < dellen; i++)
|
|
|
|
trans_table[(int) Py_CHARMASK(del_table[i])] = -1;
|
2015-02-02 19:21:08 -04:00
|
|
|
PyBuffer_Release(&del_table_view);
|
2010-05-09 12:52:27 -03:00
|
|
|
|
|
|
|
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 15:02:19 -04:00
|
|
|
}
|
|
|
|
|
2007-09-10 13:53:45 -03:00
|
|
|
|
2009-04-12 12:51:51 -03:00
|
|
|
static PyObject *
|
2009-04-18 12:17:02 -03:00
|
|
|
bytes_maketrans(PyObject *null, PyObject *args)
|
2009-04-12 12:51:51 -03:00
|
|
|
{
|
2010-05-09 12:52:27 -03:00
|
|
|
return _Py_bytes_maketrans(args);
|
2009-04-12 12:51:51 -03:00
|
|
|
}
|
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
/* find and count characters and substrings */
|
2007-09-10 13:53:45 -03:00
|
|
|
|
2010-05-09 12:52:27 -03:00
|
|
|
#define findchar(target, target_len, c) \
|
2008-05-26 10:42:13 -03:00
|
|
|
((char *)memchr((const void *)(target), c, target_len))
|
2007-02-27 15:02:19 -04:00
|
|
|
|
2008-05-26 10:42:13 -03: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 15:02:19 -04:00
|
|
|
{
|
2010-05-09 12:52:27 -03: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 10:42:13 -03:00
|
|
|
}
|
2007-09-10 13:53:45 -03:00
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
Py_LOCAL_INLINE(Py_ssize_t)
|
2010-08-15 14:12:55 -03:00
|
|
|
countchar(const char *target, Py_ssize_t target_len, char c, Py_ssize_t maxcount)
|
2008-05-26 10:42:13 -03:00
|
|
|
{
|
2010-05-09 12:52:27 -03:00
|
|
|
Py_ssize_t count=0;
|
|
|
|
const char *start=target;
|
|
|
|
const char *end=target+target_len;
|
2007-02-27 15:02:19 -04:00
|
|
|
|
2010-05-09 12:52:27 -03:00
|
|
|
while ( (start=findchar(start, end-start, c)) != NULL ) {
|
|
|
|
count++;
|
|
|
|
if (count >= maxcount)
|
|
|
|
break;
|
|
|
|
start += 1;
|
|
|
|
}
|
|
|
|
return count;
|
2008-05-26 10:42:13 -03:00
|
|
|
}
|
2007-02-27 15:02:19 -04:00
|
|
|
|
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
/* Algorithms for different cases of string replacement */
|
2007-02-27 15:02:19 -04:00
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
/* len(self)>=1, from="", len(to)>=1, maxcount>=1 */
|
|
|
|
Py_LOCAL(PyBytesObject *)
|
|
|
|
replace_interleave(PyBytesObject *self,
|
2010-05-09 12:52:27 -03: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 15:35:01 -03:00
|
|
|
Py_ssize_t count, i;
|
2010-05-09 12:52:27 -03:00
|
|
|
PyBytesObject *result;
|
|
|
|
|
|
|
|
self_len = PyBytes_GET_SIZE(self);
|
|
|
|
|
2010-08-10 15:35:01 -03:00
|
|
|
/* 1 at the end plus 1 after every character;
|
|
|
|
count = min(maxcount, self_len + 1) */
|
|
|
|
if (maxcount <= self_len)
|
2010-05-09 12:52:27 -03:00
|
|
|
count = maxcount;
|
2010-08-10 15:35:01 -03:00
|
|
|
else
|
|
|
|
/* Can't overflow: self_len + 1 <= maxcount <= PY_SSIZE_T_MAX. */
|
|
|
|
count = self_len + 1;
|
2010-05-09 12:52:27 -03:00
|
|
|
|
|
|
|
/* Check for overflow */
|
|
|
|
/* result_len = count * to_len + self_len; */
|
2010-08-10 15:35:01 -03:00
|
|
|
assert(count > 0);
|
|
|
|
if (to_len > (PY_SSIZE_T_MAX - self_len) / count) {
|
2010-05-09 12:52:27 -03:00
|
|
|
PyErr_SetString(PyExc_OverflowError,
|
|
|
|
"replacement bytes are too long");
|
|
|
|
return NULL;
|
|
|
|
}
|
2010-08-10 15:35:01 -03:00
|
|
|
result_len = count * to_len + self_len;
|
2010-05-09 12:52:27 -03: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 15:02:19 -04:00
|
|
|
}
|
|
|
|
|
2008-05-26 10:42:13 -03: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 12:52:27 -03:00
|
|
|
char from_c, Py_ssize_t maxcount)
|
2008-05-26 10:42:13 -03:00
|
|
|
{
|
2010-05-09 12:52:27 -03: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 10:42:13 -03:00
|
|
|
|
2010-05-09 12:52:27 -03:00
|
|
|
self_len = PyBytes_GET_SIZE(self);
|
|
|
|
self_s = PyBytes_AS_STRING(self);
|
2007-09-10 13:53:45 -03:00
|
|
|
|
2010-05-09 12:52:27 -03:00
|
|
|
count = countchar(self_s, self_len, from_c, maxcount);
|
|
|
|
if (count == 0) {
|
|
|
|
return return_self(self);
|
|
|
|
}
|
2008-05-26 10:42:13 -03:00
|
|
|
|
2010-05-09 12:52:27 -03:00
|
|
|
result_len = self_len - count; /* from_len == 1 */
|
|
|
|
assert(result_len>=0);
|
2008-05-26 10:42:13 -03:00
|
|
|
|
2010-05-09 12:52:27 -03: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 13:53:45 -03:00
|
|
|
|
2010-05-09 12:52:27 -03:00
|
|
|
return result;
|
2007-09-10 13:53:45 -03:00
|
|
|
}
|
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
/* len(self)>=1, len(from)>=2, to="", maxcount>=1 */
|
2007-02-27 15:02:19 -04:00
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
Py_LOCAL(PyBytesObject *)
|
|
|
|
replace_delete_substring(PyBytesObject *self,
|
2010-05-09 12:52:27 -03: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 15:02:19 -04:00
|
|
|
}
|
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
/* len(self)>=1, len(from)==len(to)==1, maxcount>=1 */
|
|
|
|
Py_LOCAL(PyBytesObject *)
|
|
|
|
replace_single_character_in_place(PyBytesObject *self,
|
2010-05-09 12:52:27 -03:00
|
|
|
char from_c, char to_c,
|
|
|
|
Py_ssize_t maxcount)
|
2007-02-27 15:02:19 -04:00
|
|
|
{
|
2010-05-09 12:52:27 -03:00
|
|
|
char *self_s, *result_s, *start, *end, *next;
|
|
|
|
Py_ssize_t self_len;
|
|
|
|
PyBytesObject *result;
|
2007-02-27 15:02:19 -04:00
|
|
|
|
2010-05-09 12:52:27 -03:00
|
|
|
/* The result string will be the same size */
|
|
|
|
self_s = PyBytes_AS_STRING(self);
|
|
|
|
self_len = PyBytes_GET_SIZE(self);
|
2007-02-27 15:02:19 -04:00
|
|
|
|
2010-05-09 12:52:27 -03:00
|
|
|
next = findchar(self_s, self_len, from_c);
|
2007-02-27 15:02:19 -04:00
|
|
|
|
2010-05-09 12:52:27 -03:00
|
|
|
if (next == NULL) {
|
|
|
|
/* No matches; return the original string */
|
|
|
|
return return_self(self);
|
|
|
|
}
|
2008-05-26 10:42:13 -03:00
|
|
|
|
2010-05-09 12:52:27 -03: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 15:02:19 -04:00
|
|
|
|
2010-05-09 12:52:27 -03:00
|
|
|
return result;
|
2008-05-26 10:42:13 -03:00
|
|
|
}
|
2007-02-27 15:02:19 -04:00
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
/* len(self)>=1, len(from)==len(to)>=2, maxcount>=1 */
|
|
|
|
Py_LOCAL(PyBytesObject *)
|
|
|
|
replace_substring_in_place(PyBytesObject *self,
|
2010-05-09 12:52:27 -03: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 15:02:19 -04:00
|
|
|
}
|
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
/* len(self)>=1, len(from)==1, len(to)>=2, maxcount>=1 */
|
|
|
|
Py_LOCAL(PyBytesObject *)
|
|
|
|
replace_single_character(PyBytesObject *self,
|
2010-05-09 12:52:27 -03: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 15:35:01 -03:00
|
|
|
Py_ssize_t count;
|
2010-05-09 12:52:27 -03: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 15:35:01 -03:00
|
|
|
assert(count > 0);
|
|
|
|
if (to_len - 1 > (PY_SSIZE_T_MAX - self_len) / count) {
|
2010-05-09 12:52:27 -03:00
|
|
|
PyErr_SetString(PyExc_OverflowError,
|
|
|
|
"replacement bytes are too long");
|
|
|
|
return NULL;
|
|
|
|
}
|
2010-08-10 15:35:01 -03:00
|
|
|
result_len = self_len + count * (to_len - 1);
|
2010-05-09 12:52:27 -03: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 10:42:13 -03:00
|
|
|
}
|
2007-12-04 01:51:13 -04:00
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
/* len(self)>=1, len(from)>=2, len(to)>=2, maxcount>=1 */
|
|
|
|
Py_LOCAL(PyBytesObject *)
|
|
|
|
replace_substring(PyBytesObject *self,
|
2010-05-09 12:52:27 -03: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 15:35:01 -03:00
|
|
|
Py_ssize_t count, offset;
|
2010-05-09 12:52:27 -03: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 15:35:01 -03:00
|
|
|
assert(count > 0);
|
|
|
|
if (to_len - from_len > (PY_SSIZE_T_MAX - self_len) / count) {
|
2010-05-09 12:52:27 -03:00
|
|
|
PyErr_SetString(PyExc_OverflowError,
|
|
|
|
"replacement bytes are too long");
|
|
|
|
return NULL;
|
|
|
|
}
|
2010-08-10 15:35:01 -03:00
|
|
|
result_len = self_len + count * (to_len-from_len);
|
2010-05-09 12:52:27 -03: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 01:51:13 -04:00
|
|
|
}
|
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
|
|
|
|
Py_LOCAL(PyBytesObject *)
|
|
|
|
replace(PyBytesObject *self,
|
2010-05-09 12:52:27 -03: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 15:02:19 -04:00
|
|
|
}
|
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
PyDoc_STRVAR(replace__doc__,
|
|
|
|
"B.replace(old, new[, count]) -> bytes\n\
|
2007-02-27 15:02:19 -04:00
|
|
|
\n\
|
2008-05-26 10:42:13 -03: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 09:50:29 -03:00
|
|
|
given, only first count occurances are replaced.");
|
2008-05-26 10:42:13 -03:00
|
|
|
|
2007-02-27 15:02:19 -04:00
|
|
|
static PyObject *
|
2009-04-18 12:17:02 -03:00
|
|
|
bytes_replace(PyBytesObject *self, PyObject *args)
|
2007-02-27 15:02:19 -04:00
|
|
|
{
|
2015-02-02 19:21:08 -04:00
|
|
|
PyObject *res;
|
|
|
|
Py_buffer old = {NULL, NULL};
|
|
|
|
Py_buffer new = {NULL, NULL};
|
2010-05-09 12:52:27 -03:00
|
|
|
Py_ssize_t count = -1;
|
2007-02-27 15:02:19 -04:00
|
|
|
|
2015-02-02 19:21:08 -04:00
|
|
|
if (!PyArg_ParseTuple(args, "y*y*|n:replace", &old, &new, &count))
|
2010-05-09 12:52:27 -03:00
|
|
|
return NULL;
|
2007-02-27 15:02:19 -04:00
|
|
|
|
2015-02-02 19:21:08 -04:00
|
|
|
res = (PyObject *)replace((PyBytesObject *) self,
|
|
|
|
(const char *)old.buf, old.len,
|
|
|
|
(const char *)new.buf, new.len, count);
|
2007-02-27 15:02:19 -04:00
|
|
|
|
2015-02-02 19:21:08 -04:00
|
|
|
PyBuffer_Release(&old);
|
|
|
|
PyBuffer_Release(&new);
|
|
|
|
return res;
|
2007-02-27 15:02:19 -04:00
|
|
|
}
|
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
/** End DALKE **/
|
2007-04-12 22:39:34 -03:00
|
|
|
|
2008-05-26 10:42:13 -03: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 12:17:02 -03:00
|
|
|
_bytes_tailmatch(PyBytesObject *self, PyObject *substr, Py_ssize_t start,
|
2010-05-09 12:52:27 -03:00
|
|
|
Py_ssize_t end, int direction)
|
2007-04-12 22:39:34 -03:00
|
|
|
{
|
2010-05-09 12:52:27 -03:00
|
|
|
Py_ssize_t len = PyBytes_GET_SIZE(self);
|
|
|
|
Py_ssize_t slen;
|
2015-02-02 19:21:08 -04:00
|
|
|
Py_buffer sub_view = {NULL, NULL};
|
2010-05-09 12:52:27 -03:00
|
|
|
const char* sub;
|
|
|
|
const char* str;
|
2007-04-12 22:39:34 -03:00
|
|
|
|
2010-05-09 12:52:27 -03:00
|
|
|
if (PyBytes_Check(substr)) {
|
|
|
|
sub = PyBytes_AS_STRING(substr);
|
|
|
|
slen = PyBytes_GET_SIZE(substr);
|
|
|
|
}
|
2015-02-02 19:21:08 -04:00
|
|
|
else {
|
|
|
|
if (PyObject_GetBuffer(substr, &sub_view, PyBUF_SIMPLE) != 0)
|
|
|
|
return -1;
|
|
|
|
sub = sub_view.buf;
|
|
|
|
slen = sub_view.len;
|
|
|
|
}
|
2010-05-09 12:52:27 -03:00
|
|
|
str = PyBytes_AS_STRING(self);
|
2008-05-26 10:42:13 -03:00
|
|
|
|
2010-05-09 12:52:27 -03:00
|
|
|
ADJUST_INDICES(start, end, len);
|
2008-05-26 10:42:13 -03:00
|
|
|
|
2010-05-09 12:52:27 -03:00
|
|
|
if (direction < 0) {
|
|
|
|
/* startswith */
|
|
|
|
if (start+slen > len)
|
2015-02-02 19:21:08 -04:00
|
|
|
goto notfound;
|
2010-05-09 12:52:27 -03:00
|
|
|
} else {
|
|
|
|
/* endswith */
|
|
|
|
if (end-start < slen || start > len)
|
2015-02-02 19:21:08 -04:00
|
|
|
goto notfound;
|
2008-05-26 10:42:13 -03:00
|
|
|
|
2010-05-09 12:52:27 -03:00
|
|
|
if (end-slen > start)
|
|
|
|
start = end - slen;
|
|
|
|
}
|
2015-02-02 19:21:08 -04:00
|
|
|
if (end-start < slen)
|
|
|
|
goto notfound;
|
|
|
|
if (memcmp(str+start, sub, slen) != 0)
|
|
|
|
goto notfound;
|
|
|
|
|
|
|
|
PyBuffer_Release(&sub_view);
|
|
|
|
return 1;
|
|
|
|
|
|
|
|
notfound:
|
|
|
|
PyBuffer_Release(&sub_view);
|
2010-05-09 12:52:27 -03:00
|
|
|
return 0;
|
2007-04-12 22:39:34 -03:00
|
|
|
}
|
|
|
|
|
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
PyDoc_STRVAR(startswith__doc__,
|
2008-05-30 05:20:09 -03:00
|
|
|
"B.startswith(prefix[, start[, end]]) -> bool\n\
|
2007-04-12 22:39:34 -03:00
|
|
|
\n\
|
2008-05-26 10:42:13 -03: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-26 21:36:20 -03:00
|
|
|
prefix can also be a tuple of bytes to try.");
|
2008-05-26 10:42:13 -03:00
|
|
|
|
2007-04-12 22:39:34 -03:00
|
|
|
static PyObject *
|
2009-04-18 12:17:02 -03:00
|
|
|
bytes_startswith(PyBytesObject *self, PyObject *args)
|
2008-05-26 10:42:13 -03:00
|
|
|
{
|
2010-05-09 12:52:27 -03:00
|
|
|
Py_ssize_t start = 0;
|
|
|
|
Py_ssize_t end = PY_SSIZE_T_MAX;
|
|
|
|
PyObject *subobj;
|
|
|
|
int result;
|
|
|
|
|
2011-04-20 12:09:23 -03:00
|
|
|
if (!stringlib_parse_args_finds("startswith", args, &subobj, &start, &end))
|
2010-05-09 12:52:27 -03: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 00: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 12:52:27 -03:00
|
|
|
return NULL;
|
2011-04-26 00:09:45 -03:00
|
|
|
}
|
2010-05-09 12:52:27 -03:00
|
|
|
else
|
|
|
|
return PyBool_FromLong(result);
|
2007-04-12 22:39:34 -03:00
|
|
|
}
|
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
|
|
|
|
PyDoc_STRVAR(endswith__doc__,
|
2008-05-30 05:20:09 -03:00
|
|
|
"B.endswith(suffix[, start[, end]]) -> bool\n\
|
2007-04-12 22:39:34 -03:00
|
|
|
\n\
|
2008-05-26 10:42:13 -03: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-26 21:36:20 -03:00
|
|
|
suffix can also be a tuple of bytes to try.");
|
2008-05-26 10:42:13 -03:00
|
|
|
|
2007-04-12 22:39:34 -03:00
|
|
|
static PyObject *
|
2009-04-18 12:17:02 -03:00
|
|
|
bytes_endswith(PyBytesObject *self, PyObject *args)
|
2008-05-26 10:42:13 -03:00
|
|
|
{
|
2010-05-09 12:52:27 -03:00
|
|
|
Py_ssize_t start = 0;
|
|
|
|
Py_ssize_t end = PY_SSIZE_T_MAX;
|
|
|
|
PyObject *subobj;
|
|
|
|
int result;
|
|
|
|
|
2011-04-20 12:09:23 -03:00
|
|
|
if (!stringlib_parse_args_finds("endswith", args, &subobj, &start, &end))
|
2010-05-09 12:52:27 -03: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 00: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 12:52:27 -03:00
|
|
|
return NULL;
|
2011-04-26 00:09:45 -03:00
|
|
|
}
|
2010-05-09 12:52:27 -03:00
|
|
|
else
|
|
|
|
return PyBool_FromLong(result);
|
2007-04-12 22:39:34 -03:00
|
|
|
}
|
2007-02-27 15:02:19 -04:00
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
|
|
|
|
PyDoc_STRVAR(decode__doc__,
|
2010-11-07 15:04:46 -04:00
|
|
|
"B.decode(encoding='utf-8', errors='strict') -> str\n\
|
2006-04-24 10:47:05 -03:00
|
|
|
\n\
|
2010-11-07 14:41:46 -04: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 17:34:58 -04: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 10:42:13 -03:00
|
|
|
as well as any other name registerd with codecs.register_error that is\n\
|
2006-04-24 10:47:05 -03:00
|
|
|
able to handle UnicodeDecodeErrors.");
|
|
|
|
|
|
|
|
static PyObject *
|
2009-09-18 18:42:35 -03:00
|
|
|
bytes_decode(PyObject *self, PyObject *args, PyObject *kwargs)
|
2007-04-12 19:49:52 -03:00
|
|
|
{
|
2010-05-09 12:52:27 -03:00
|
|
|
const char *encoding = NULL;
|
|
|
|
const char *errors = NULL;
|
|
|
|
static char *kwlist[] = {"encoding", "errors", 0};
|
2006-05-05 01:34:18 -03:00
|
|
|
|
2010-05-09 12:52:27 -03:00
|
|
|
if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|ss:decode", kwlist, &encoding, &errors))
|
|
|
|
return NULL;
|
|
|
|
return PyUnicode_FromEncodedObject(self, encoding, errors);
|
2006-05-05 01:34:18 -03:00
|
|
|
}
|
|
|
|
|
2006-05-05 12:15:40 -03:00
|
|
|
|
2010-01-13 04:07:53 -04: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 05:14:39 -03:00
|
|
|
bytes_splitlines(PyObject *self, PyObject *args, PyObject *kwds)
|
2010-01-13 04:07:53 -04:00
|
|
|
{
|
2011-09-24 05:14:39 -03:00
|
|
|
static char *kwlist[] = {"keepends", 0};
|
2010-01-13 04:07:53 -04:00
|
|
|
int keepends = 0;
|
|
|
|
|
2011-09-24 05:14:39 -03:00
|
|
|
if (!PyArg_ParseTupleAndKeywords(args, kwds, "|i:splitlines",
|
|
|
|
kwlist, &keepends))
|
2010-06-09 13:38:55 -03:00
|
|
|
return NULL;
|
2010-01-13 04:07:53 -04:00
|
|
|
|
|
|
|
return stringlib_splitlines(
|
2010-06-09 13:38:55 -03:00
|
|
|
(PyObject*) self, PyBytes_AS_STRING(self),
|
|
|
|
PyBytes_GET_SIZE(self), keepends
|
|
|
|
);
|
2010-01-13 04:07:53 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
|
2007-02-27 04:40:54 -04:00
|
|
|
PyDoc_STRVAR(fromhex_doc,
|
2008-05-26 10:42:13 -03:00
|
|
|
"bytes.fromhex(string) -> bytes\n\
|
2007-02-27 04:40:54 -04:00
|
|
|
\n\
|
2008-05-26 10:42:13 -03:00
|
|
|
Create a bytes object from a string of hexadecimal numbers.\n\
|
2007-11-06 17:34:58 -04:00
|
|
|
Spaces between two numbers are accepted.\n\
|
2008-05-26 10:42:13 -03:00
|
|
|
Example: bytes.fromhex('B9 01EF') -> b'\\xb9\\x01\\xef'.");
|
2007-02-27 04:40:54 -04:00
|
|
|
|
|
|
|
static int
|
2011-09-28 23:02:13 -03:00
|
|
|
hex_digit_to_int(Py_UCS4 c)
|
2007-02-27 04:40:54 -04:00
|
|
|
{
|
2010-05-09 12:52:27 -03:00
|
|
|
if (c >= 128)
|
|
|
|
return -1;
|
2010-11-05 14:23:41 -03:00
|
|
|
if (Py_ISDIGIT(c))
|
2010-05-09 12:52:27 -03:00
|
|
|
return c - '0';
|
|
|
|
else {
|
2010-11-05 14:23:41 -03:00
|
|
|
if (Py_ISUPPER(c))
|
|
|
|
c = Py_TOLOWER(c);
|
2010-05-09 12:52:27 -03:00
|
|
|
if (c >= 'a' && c <= 'f')
|
|
|
|
return c - 'a' + 10;
|
|
|
|
}
|
|
|
|
return -1;
|
2007-02-27 04:40:54 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
static PyObject *
|
2009-04-18 12:17:02 -03:00
|
|
|
bytes_fromhex(PyObject *cls, PyObject *args)
|
2008-05-26 10:42:13 -03:00
|
|
|
{
|
2010-05-09 12:52:27 -03:00
|
|
|
PyObject *newstring, *hexobj;
|
|
|
|
char *buf;
|
|
|
|
Py_ssize_t hexlen, byteslen, i, j;
|
|
|
|
int top, bot;
|
2011-09-28 02:41:54 -03:00
|
|
|
void *data;
|
|
|
|
unsigned int kind;
|
2010-05-09 12:52:27 -03:00
|
|
|
|
|
|
|
if (!PyArg_ParseTuple(args, "U:fromhex", &hexobj))
|
|
|
|
return NULL;
|
|
|
|
assert(PyUnicode_Check(hexobj));
|
2011-09-28 02:41:54 -03:00
|
|
|
if (PyUnicode_READY(hexobj))
|
|
|
|
return NULL;
|
|
|
|
kind = PyUnicode_KIND(hexobj);
|
|
|
|
data = PyUnicode_DATA(hexobj);
|
|
|
|
hexlen = PyUnicode_GET_LENGTH(hexobj);
|
|
|
|
|
2010-05-09 12:52:27 -03: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 02:41:54 -03:00
|
|
|
while (PyUnicode_READ(kind, data, i) == ' ')
|
2010-05-09 12:52:27 -03:00
|
|
|
i++;
|
|
|
|
if (i >= hexlen)
|
|
|
|
break;
|
2011-09-28 02:41:54 -03: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 12:52:27 -03: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 04:40:54 -04:00
|
|
|
|
|
|
|
error:
|
2010-05-09 12:52:27 -03:00
|
|
|
Py_XDECREF(newstring);
|
|
|
|
return NULL;
|
2007-02-27 04:40:54 -04:00
|
|
|
}
|
|
|
|
|
2007-04-11 02:40:58 -03:00
|
|
|
static PyObject *
|
2009-04-18 12:17:02 -03:00
|
|
|
bytes_getnewargs(PyBytesObject *v)
|
2007-04-11 02:40:58 -03:00
|
|
|
{
|
2010-05-09 12:52:27 -03:00
|
|
|
return Py_BuildValue("(y#)", v->ob_sval, Py_SIZE(v));
|
2007-04-11 02:40:58 -03:00
|
|
|
}
|
|
|
|
|
2006-04-22 20:28:04 -03:00
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
static PyMethodDef
|
2009-04-18 12:17:02 -03:00
|
|
|
bytes_methods[] = {
|
2010-05-09 12:52:27 -03: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__},
|
2013-11-16 13:10:57 -04:00
|
|
|
{"expandtabs", (PyCFunction)stringlib_expandtabs, METH_VARARGS | METH_KEYWORDS,
|
2010-05-09 12:52:27 -03:00
|
|
|
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__},
|
2012-02-26 03:39:55 -04:00
|
|
|
{"rsplit", (PyCFunction)bytes_rsplit, METH_VARARGS | METH_KEYWORDS, rsplit__doc__},
|
2010-05-09 12:52:27 -03:00
|
|
|
{"rstrip", (PyCFunction)bytes_rstrip, METH_VARARGS, rstrip__doc__},
|
2012-02-26 03:39:55 -04:00
|
|
|
{"split", (PyCFunction)bytes_split, METH_VARARGS | METH_KEYWORDS, split__doc__},
|
2011-09-24 05:14:39 -03:00
|
|
|
{"splitlines", (PyCFunction)bytes_splitlines, METH_VARARGS | METH_KEYWORDS,
|
2010-05-09 12:52:27 -03: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__},
|
|
|
|
{NULL, NULL} /* sentinel */
|
2006-04-22 20:28:04 -03:00
|
|
|
};
|
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
static PyObject *
|
|
|
|
str_subtype_new(PyTypeObject *type, PyObject *args, PyObject *kwds);
|
2006-04-22 20:28:04 -03:00
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
static PyObject *
|
2009-04-18 12:17:02 -03:00
|
|
|
bytes_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
|
2008-05-26 10:42:13 -03:00
|
|
|
{
|
2010-05-09 12:52:27 -03:00
|
|
|
PyObject *x = NULL;
|
|
|
|
const char *encoding = NULL;
|
|
|
|
const char *errors = NULL;
|
|
|
|
PyObject *new = NULL;
|
2012-12-19 17:27:41 -04:00
|
|
|
PyObject *func;
|
2010-05-09 12:52:27 -03:00
|
|
|
Py_ssize_t size;
|
|
|
|
static char *kwlist[] = {"source", "encoding", "errors", 0};
|
2012-12-19 17:27:41 -04:00
|
|
|
_Py_IDENTIFIER(__bytes__);
|
2010-05-09 12:52:27 -03:00
|
|
|
|
|
|
|
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;
|
|
|
|
}
|
2012-12-19 17:27:41 -04:00
|
|
|
|
2014-12-02 03:24:06 -04:00
|
|
|
/* 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;
|
|
|
|
}
|
|
|
|
|
2012-12-19 17:27:41 -04:00
|
|
|
/* We'd like to call PyObject_Bytes here, but we need to check for an
|
|
|
|
integer argument before deferring to PyBytes_FromObject, something
|
|
|
|
PyObject_Bytes doesn't do. */
|
|
|
|
func = _PyObject_LookupSpecial(x, &PyId___bytes__);
|
|
|
|
if (func != NULL) {
|
|
|
|
new = PyObject_CallFunctionObjArgs(func, NULL);
|
|
|
|
Py_DECREF(func);
|
|
|
|
if (new == NULL)
|
|
|
|
return NULL;
|
|
|
|
if (!PyBytes_Check(new)) {
|
|
|
|
PyErr_Format(PyExc_TypeError,
|
|
|
|
"__bytes__ returned non-bytes (type %.200s)",
|
|
|
|
Py_TYPE(new)->tp_name);
|
|
|
|
Py_DECREF(new);
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
return new;
|
|
|
|
}
|
|
|
|
else if (PyErr_Occurred())
|
|
|
|
return NULL;
|
|
|
|
|
2010-05-09 12:52:27 -03:00
|
|
|
/* 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);
|
2012-12-19 17:27:41 -04:00
|
|
|
if (new == NULL)
|
2010-05-09 12:52:27 -03:00
|
|
|
return NULL;
|
2012-12-19 17:27:41 -04:00
|
|
|
if (size > 0)
|
2010-05-09 12:52:27 -03:00
|
|
|
memset(((PyBytesObject*)new)->ob_sval, 0, size);
|
|
|
|
return new;
|
|
|
|
}
|
|
|
|
|
2012-12-19 17:27:41 -04:00
|
|
|
return PyBytes_FromObject(x);
|
2008-08-26 13:46:47 -03:00
|
|
|
}
|
|
|
|
|
|
|
|
PyObject *
|
|
|
|
PyBytes_FromObject(PyObject *x)
|
|
|
|
{
|
2010-05-09 12:52:27 -03:00
|
|
|
PyObject *new, *it;
|
|
|
|
Py_ssize_t i, size;
|
|
|
|
|
|
|
|
if (x == NULL) {
|
|
|
|
PyErr_BadInternalCall();
|
|
|
|
return NULL;
|
|
|
|
}
|
2012-05-25 02:58:30 -03:00
|
|
|
|
|
|
|
if (PyBytes_CheckExact(x)) {
|
|
|
|
Py_INCREF(x);
|
|
|
|
return x;
|
|
|
|
}
|
|
|
|
|
2010-05-09 12:52:27 -03:00
|
|
|
/* 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;
|
|
|
|
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 14:12:55 -03:00
|
|
|
((PyBytesObject *)new)->ob_sval[i] = (char) value;
|
2010-05-09 12:52:27 -03: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 14:12:55 -03:00
|
|
|
((PyBytesObject *)new)->ob_sval[i] = (char) value;
|
2010-05-09 12:52:27 -03:00
|
|
|
}
|
|
|
|
return new;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* For iterator version, create a string object and resize as needed */
|
2012-10-06 09:03:24 -03:00
|
|
|
size = PyObject_LengthHint(x, 64);
|
2010-05-09 12:52:27 -03:00
|
|
|
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 14:12:55 -03:00
|
|
|
((PyBytesObject *)new)->ob_sval[i] = (char) value;
|
2010-05-09 12:52:27 -03:00
|
|
|
}
|
|
|
|
_PyBytes_Resize(&new, i);
|
|
|
|
|
|
|
|
/* Clean up and return success */
|
|
|
|
Py_DECREF(it);
|
|
|
|
return new;
|
2008-05-26 10:42:13 -03:00
|
|
|
|
|
|
|
error:
|
2010-05-09 12:52:27 -03:00
|
|
|
Py_XDECREF(it);
|
2013-10-28 23:14:22 -03:00
|
|
|
Py_XDECREF(new);
|
2010-05-09 12:52:27 -03:00
|
|
|
return NULL;
|
2008-05-26 10:42:13 -03:00
|
|
|
}
|
|
|
|
|
|
|
|
static PyObject *
|
|
|
|
str_subtype_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
|
|
|
|
{
|
2010-05-09 12:52:27 -03: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 10:42:13 -03:00
|
|
|
}
|
2006-04-22 20:28:04 -03:00
|
|
|
|
2009-04-18 12:17:02 -03:00
|
|
|
PyDoc_STRVAR(bytes_doc,
|
2008-05-30 05:20:09 -03:00
|
|
|
"bytes(iterable_of_ints) -> bytes\n\
|
2008-05-26 10:42:13 -03:00
|
|
|
bytes(string, encoding[, errors]) -> bytes\n\
|
2008-05-30 05:20:09 -03:00
|
|
|
bytes(bytes_or_buffer) -> immutable copy of bytes_or_buffer\n\
|
2011-12-17 18:18:07 -04:00
|
|
|
bytes(int) -> bytes object of size given by the parameter initialized with null bytes\n\
|
|
|
|
bytes() -> empty bytes object\n\
|
2007-11-06 17:34:58 -04:00
|
|
|
\n\
|
2008-05-26 10:42:13 -03:00
|
|
|
Construct an immutable array of bytes from:\n\
|
2007-11-06 17:34:58 -04:00
|
|
|
- an iterable yielding integers in range(256)\n\
|
|
|
|
- a text string encoded using the specified encoding\n\
|
2011-12-17 18:18:07 -04:00
|
|
|
- any object implementing the buffer API.\n\
|
|
|
|
- an integer");
|
2008-05-26 10:42:13 -03:00
|
|
|
|
2009-04-18 12:17:02 -03:00
|
|
|
static PyObject *bytes_iter(PyObject *seq);
|
2008-05-26 10:42:13 -03:00
|
|
|
|
|
|
|
PyTypeObject PyBytes_Type = {
|
2010-05-09 12:52:27 -03: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 20:28:04 -03:00
|
|
|
};
|
2007-10-26 14:39:48 -03:00
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
void
|
2013-08-13 15:18:52 -03:00
|
|
|
PyBytes_Concat(PyObject **pv, PyObject *w)
|
2008-05-26 10:42:13 -03:00
|
|
|
{
|
2013-08-13 15:18:52 -03:00
|
|
|
PyObject *v;
|
2010-05-09 12:52:27 -03:00
|
|
|
assert(pv != NULL);
|
|
|
|
if (*pv == NULL)
|
|
|
|
return;
|
|
|
|
if (w == NULL) {
|
2013-02-02 12:45:22 -04:00
|
|
|
Py_CLEAR(*pv);
|
2010-05-09 12:52:27 -03:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
v = bytes_concat(*pv, w);
|
|
|
|
Py_DECREF(*pv);
|
|
|
|
*pv = v;
|
2008-05-26 10:42:13 -03:00
|
|
|
}
|
|
|
|
|
|
|
|
void
|
2013-08-13 15:18:52 -03:00
|
|
|
PyBytes_ConcatAndDel(PyObject **pv, PyObject *w)
|
2008-05-26 10:42:13 -03:00
|
|
|
{
|
2010-05-09 12:52:27 -03:00
|
|
|
PyBytes_Concat(pv, w);
|
|
|
|
Py_XDECREF(w);
|
2008-05-26 10:42:13 -03: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)
|
|
|
|
{
|
2013-08-13 15:18:52 -03:00
|
|
|
PyObject *v;
|
|
|
|
PyBytesObject *sv;
|
2010-05-09 12:52:27 -03:00
|
|
|
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 10:42:13 -03:00
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
PyBytes_Fini(void)
|
|
|
|
{
|
2010-05-09 12:52:27 -03:00
|
|
|
int i;
|
2013-02-02 12:45:22 -04:00
|
|
|
for (i = 0; i < UCHAR_MAX + 1; i++)
|
|
|
|
Py_CLEAR(characters[i]);
|
|
|
|
Py_CLEAR(nullstring);
|
2008-05-26 10:42:13 -03:00
|
|
|
}
|
|
|
|
|
2008-05-26 21:36:20 -03:00
|
|
|
/*********************** Bytes Iterator ****************************/
|
2007-10-26 14:39:48 -03:00
|
|
|
|
|
|
|
typedef struct {
|
2010-05-09 12:52:27 -03:00
|
|
|
PyObject_HEAD
|
|
|
|
Py_ssize_t it_index;
|
|
|
|
PyBytesObject *it_seq; /* Set to NULL when iterator is exhausted */
|
2008-05-26 10:42:13 -03:00
|
|
|
} striterobject;
|
2007-10-26 14:39:48 -03:00
|
|
|
|
|
|
|
static void
|
2008-05-26 10:42:13 -03:00
|
|
|
striter_dealloc(striterobject *it)
|
2007-10-26 14:39:48 -03:00
|
|
|
{
|
2010-05-09 12:52:27 -03:00
|
|
|
_PyObject_GC_UNTRACK(it);
|
|
|
|
Py_XDECREF(it->it_seq);
|
|
|
|
PyObject_GC_Del(it);
|
2007-10-26 14:39:48 -03:00
|
|
|
}
|
|
|
|
|
|
|
|
static int
|
2008-05-26 10:42:13 -03:00
|
|
|
striter_traverse(striterobject *it, visitproc visit, void *arg)
|
2007-10-26 14:39:48 -03:00
|
|
|
{
|
2010-05-09 12:52:27 -03:00
|
|
|
Py_VISIT(it->it_seq);
|
|
|
|
return 0;
|
2007-10-26 14:39:48 -03:00
|
|
|
}
|
|
|
|
|
|
|
|
static PyObject *
|
2008-05-26 10:42:13 -03:00
|
|
|
striter_next(striterobject *it)
|
|
|
|
{
|
2010-05-09 12:52:27 -03:00
|
|
|
PyBytesObject *seq;
|
|
|
|
PyObject *item;
|
2008-05-26 10:42:13 -03:00
|
|
|
|
2010-05-09 12:52:27 -03: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 14:39:48 -03:00
|
|
|
|
2010-05-09 12:52:27 -03:00
|
|
|
Py_DECREF(seq);
|
|
|
|
it->it_seq = NULL;
|
|
|
|
return NULL;
|
2007-10-26 14:39:48 -03:00
|
|
|
}
|
|
|
|
|
|
|
|
static PyObject *
|
2008-05-26 10:42:13 -03:00
|
|
|
striter_len(striterobject *it)
|
2007-10-26 14:39:48 -03:00
|
|
|
{
|
2010-05-09 12:52:27 -03: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 14:39:48 -03:00
|
|
|
}
|
|
|
|
|
|
|
|
PyDoc_STRVAR(length_hint_doc,
|
2010-05-09 12:52:27 -03:00
|
|
|
"Private method returning an estimate of len(list(it)).");
|
2007-10-26 14:39:48 -03:00
|
|
|
|
2012-04-03 07:49:41 -03:00
|
|
|
static PyObject *
|
|
|
|
striter_reduce(striterobject *it)
|
|
|
|
{
|
|
|
|
if (it->it_seq != NULL) {
|
2012-04-04 19:04:20 -03:00
|
|
|
return Py_BuildValue("N(O)n", _PyObject_GetBuiltin("iter"),
|
2012-04-03 07:49:41 -03:00
|
|
|
it->it_seq, it->it_index);
|
|
|
|
} else {
|
|
|
|
PyObject *u = PyUnicode_FromUnicode(NULL, 0);
|
|
|
|
if (u == NULL)
|
|
|
|
return NULL;
|
2012-04-04 19:04:20 -03:00
|
|
|
return Py_BuildValue("N(N)", _PyObject_GetBuiltin("iter"), u);
|
2012-04-03 07:49:41 -03:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
PyDoc_STRVAR(reduce_doc, "Return state information for pickling.");
|
|
|
|
|
|
|
|
static PyObject *
|
|
|
|
striter_setstate(striterobject *it, PyObject *state)
|
|
|
|
{
|
|
|
|
Py_ssize_t index = PyLong_AsSsize_t(state);
|
|
|
|
if (index == -1 && PyErr_Occurred())
|
|
|
|
return NULL;
|
2014-03-05 09:47:57 -04:00
|
|
|
if (it->it_seq != NULL) {
|
|
|
|
if (index < 0)
|
|
|
|
index = 0;
|
|
|
|
else if (index > PyBytes_GET_SIZE(it->it_seq))
|
|
|
|
index = PyBytes_GET_SIZE(it->it_seq); /* iterator exhausted */
|
|
|
|
it->it_index = index;
|
|
|
|
}
|
2012-04-03 07:49:41 -03:00
|
|
|
Py_RETURN_NONE;
|
|
|
|
}
|
|
|
|
|
|
|
|
PyDoc_STRVAR(setstate_doc, "Set state information for unpickling.");
|
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
static PyMethodDef striter_methods[] = {
|
2010-05-09 12:52:27 -03:00
|
|
|
{"__length_hint__", (PyCFunction)striter_len, METH_NOARGS,
|
|
|
|
length_hint_doc},
|
2012-04-03 07:49:41 -03:00
|
|
|
{"__reduce__", (PyCFunction)striter_reduce, METH_NOARGS,
|
|
|
|
reduce_doc},
|
|
|
|
{"__setstate__", (PyCFunction)striter_setstate, METH_O,
|
|
|
|
setstate_doc},
|
2010-05-09 12:52:27 -03:00
|
|
|
{NULL, NULL} /* sentinel */
|
2007-10-26 14:39:48 -03:00
|
|
|
};
|
|
|
|
|
2008-05-26 10:42:13 -03:00
|
|
|
PyTypeObject PyBytesIter_Type = {
|
2010-05-09 12:52:27 -03: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 14:39:48 -03:00
|
|
|
};
|
|
|
|
|
|
|
|
static PyObject *
|
2009-04-18 12:17:02 -03:00
|
|
|
bytes_iter(PyObject *seq)
|
2007-10-26 14:39:48 -03:00
|
|
|
{
|
2010-05-09 12:52:27 -03: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 14:39:48 -03:00
|
|
|
}
|