stringobject.c 23.9 KB
Newer Older
1
/***********************************************************
2 3
Copyright 1991-1995 by Stichting Mathematisch Centrum, Amsterdam,
The Netherlands.
4 5 6

                        All Rights Reserved

7 8
Permission to use, copy, modify, and distribute this software and its
documentation for any purpose and without fee is hereby granted,
9
provided that the above copyright notice appear in all copies and that
10
both that copyright notice and this permission notice appear in
11
supporting documentation, and that the names of Stichting Mathematisch
12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28
Centrum or CWI or Corporation for National Research Initiatives or
CNRI not be used in advertising or publicity pertaining to
distribution of the software without specific, written prior
permission.

While CWI is the initial source for this software, a modified version
is made available by the Corporation for National Research Initiatives
(CNRI) at the Internet address ftp://ftp.python.org.

STICHTING MATHEMATISCH CENTRUM AND CNRI DISCLAIM ALL WARRANTIES WITH
REGARD TO THIS SOFTWARE, INCLUDING ALL IMPLIED WARRANTIES OF
MERCHANTABILITY AND FITNESS, IN NO EVENT SHALL STICHTING MATHEMATISCH
CENTRUM OR CNRI BE LIABLE FOR ANY SPECIAL, INDIRECT OR CONSEQUENTIAL
DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR
PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER
TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
PERFORMANCE OF THIS SOFTWARE.
29 30 31

******************************************************************/

Guido van Rossum's avatar
Guido van Rossum committed
32 33
/* String object implementation */

34
#include "Python.h"
Guido van Rossum's avatar
Guido van Rossum committed
35

36
#include "mymath.h"
37 38
#include <ctype.h>

39 40 41 42
#ifdef COUNT_ALLOCS
int null_strings, one_strings;
#endif

43
#ifdef HAVE_LIMITS_H
44 45 46 47 48 49 50
#include <limits.h>
#else
#ifndef UCHAR_MAX
#define UCHAR_MAX 255
#endif
#endif

51
static PyStringObject *characters[UCHAR_MAX + 1];
52
#ifndef DONT_SHARE_SHORT_STRINGS
53
static PyStringObject *nullstring;
54
#endif
55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70

/*
   Newsizedstringobject() and newstringobject() try in certain cases
   to share string objects.  When the size of the string is zero,
   these routines always return a pointer to the same string object;
   when the size is one, they return a pointer to an already existing
   object if the contents of the string is known.  For
   newstringobject() this is always the case, for
   newsizedstringobject() this is the case when the first argument in
   not NULL.
   A common practice to allocate a string and then fill it in or
   change it must be done carefully.  It is only allowed to change the
   contents of the string if the obect was gotten from
   newsizedstringobject() with a NULL first argument, because in the
   future these routines may try to do even more sharing of objects.
*/
71 72
PyObject *
PyString_FromStringAndSize(str, size)
73
	const char *str;
Guido van Rossum's avatar
Guido van Rossum committed
74 75
	int size;
{
76
	register PyStringObject *op;
77
#ifndef DONT_SHARE_SHORT_STRINGS
78 79 80 81
	if (size == 0 && (op = nullstring) != NULL) {
#ifdef COUNT_ALLOCS
		null_strings++;
#endif
82 83
		Py_INCREF(op);
		return (PyObject *)op;
84
	}
85 86 87
	if (size == 1 && str != NULL &&
	    (op = characters[*str & UCHAR_MAX]) != NULL)
	{
88 89 90
#ifdef COUNT_ALLOCS
		one_strings++;
#endif
91 92
		Py_INCREF(op);
		return (PyObject *)op;
93
	}
94
#endif /* DONT_SHARE_SHORT_STRINGS */
95 96
	op = (PyStringObject *)
		malloc(sizeof(PyStringObject) + size * sizeof(char));
Guido van Rossum's avatar
Guido van Rossum committed
97
	if (op == NULL)
98 99
		return PyErr_NoMemory();
	op->ob_type = &PyString_Type;
Guido van Rossum's avatar
Guido van Rossum committed
100
	op->ob_size = size;
101 102
#ifdef CACHE_HASH
	op->ob_shash = -1;
Guido van Rossum's avatar
Guido van Rossum committed
103 104 105
#endif
#ifdef INTERN_STRINGS
	op->ob_sinterned = NULL;
106
#endif
107
	_Py_NewReference(op);
Guido van Rossum's avatar
Guido van Rossum committed
108 109 110
	if (str != NULL)
		memcpy(op->ob_sval, str, size);
	op->ob_sval[size] = '\0';
111
#ifndef DONT_SHARE_SHORT_STRINGS
112 113
	if (size == 0) {
		nullstring = op;
114
		Py_INCREF(op);
115 116
	} else if (size == 1 && str != NULL) {
		characters[*str & UCHAR_MAX] = op;
117
		Py_INCREF(op);
118
	}
119
#endif
120
	return (PyObject *) op;
Guido van Rossum's avatar
Guido van Rossum committed
121 122
}

123 124
PyObject *
PyString_FromString(str)
125
	const char *str;
Guido van Rossum's avatar
Guido van Rossum committed
126 127
{
	register unsigned int size = strlen(str);
128
	register PyStringObject *op;
129
#ifndef DONT_SHARE_SHORT_STRINGS
130 131 132 133
	if (size == 0 && (op = nullstring) != NULL) {
#ifdef COUNT_ALLOCS
		null_strings++;
#endif
134 135
		Py_INCREF(op);
		return (PyObject *)op;
136 137 138 139 140
	}
	if (size == 1 && (op = characters[*str & UCHAR_MAX]) != NULL) {
#ifdef COUNT_ALLOCS
		one_strings++;
#endif
141 142
		Py_INCREF(op);
		return (PyObject *)op;
143
	}
144
#endif /* DONT_SHARE_SHORT_STRINGS */
145 146
	op = (PyStringObject *)
		malloc(sizeof(PyStringObject) + size * sizeof(char));
Guido van Rossum's avatar
Guido van Rossum committed
147
	if (op == NULL)
148 149
		return PyErr_NoMemory();
	op->ob_type = &PyString_Type;
Guido van Rossum's avatar
Guido van Rossum committed
150
	op->ob_size = size;
151 152
#ifdef CACHE_HASH
	op->ob_shash = -1;
Guido van Rossum's avatar
Guido van Rossum committed
153 154 155
#endif
#ifdef INTERN_STRINGS
	op->ob_sinterned = NULL;
156
#endif
157
	_Py_NewReference(op);
Guido van Rossum's avatar
Guido van Rossum committed
158
	strcpy(op->ob_sval, str);
159
#ifndef DONT_SHARE_SHORT_STRINGS
160 161
	if (size == 0) {
		nullstring = op;
162
		Py_INCREF(op);
163 164
	} else if (size == 1) {
		characters[*str & UCHAR_MAX] = op;
165
		Py_INCREF(op);
166
	}
167
#endif
168
	return (PyObject *) op;
Guido van Rossum's avatar
Guido van Rossum committed
169 170
}

171
static void
172
string_dealloc(op)
173
	PyObject *op;
Guido van Rossum's avatar
Guido van Rossum committed
174
{
175
	PyMem_DEL(op);
Guido van Rossum's avatar
Guido van Rossum committed
176 177
}

178
int
179 180
PyString_Size(op)
	register PyObject *op;
Guido van Rossum's avatar
Guido van Rossum committed
181
{
182 183
	if (!PyString_Check(op)) {
		PyErr_BadInternalCall();
Guido van Rossum's avatar
Guido van Rossum committed
184 185
		return -1;
	}
186
	return ((PyStringObject *)op) -> ob_size;
Guido van Rossum's avatar
Guido van Rossum committed
187 188 189
}

/*const*/ char *
190 191
PyString_AsString(op)
	register PyObject *op;
Guido van Rossum's avatar
Guido van Rossum committed
192
{
193 194
	if (!PyString_Check(op)) {
		PyErr_BadInternalCall();
Guido van Rossum's avatar
Guido van Rossum committed
195 196
		return NULL;
	}
197
	return ((PyStringObject *)op) -> ob_sval;
Guido van Rossum's avatar
Guido van Rossum committed
198 199 200 201
}

/* Methods */

202
static int
203
string_print(op, fp, flags)
204
	PyStringObject *op;
Guido van Rossum's avatar
Guido van Rossum committed
205 206 207 208 209
	FILE *fp;
	int flags;
{
	int i;
	char c;
210
	int quote;
211
	/* XXX Ought to check for interrupts when writing long strings */
212
	if (flags & Py_PRINT_RAW) {
Guido van Rossum's avatar
Guido van Rossum committed
213
		fwrite(op->ob_sval, 1, (int) op->ob_size, fp);
214
		return 0;
Guido van Rossum's avatar
Guido van Rossum committed
215
	}
216 217 218 219 220 221 222

	/* figure out which quote to use; single is prefered */
	quote = '\'';
	if (strchr(op->ob_sval, '\'') && !strchr(op->ob_sval, '"'))
		quote = '"';

	fputc(quote, fp);
Guido van Rossum's avatar
Guido van Rossum committed
223 224
	for (i = 0; i < op->ob_size; i++) {
		c = op->ob_sval[i];
225
		if (c == quote || c == '\\')
Guido van Rossum's avatar
Guido van Rossum committed
226 227
			fprintf(fp, "\\%c", c);
		else if (c < ' ' || c >= 0177)
228
			fprintf(fp, "\\%03o", c & 0377);
Guido van Rossum's avatar
Guido van Rossum committed
229
		else
230
			fputc(c, fp);
Guido van Rossum's avatar
Guido van Rossum committed
231
	}
232
	fputc(quote, fp);
233
	return 0;
Guido van Rossum's avatar
Guido van Rossum committed
234 235
}

236
static PyObject *
237
string_repr(op)
238
	register PyStringObject *op;
Guido van Rossum's avatar
Guido van Rossum committed
239 240 241
{
	/* XXX overflow? */
	int newsize = 2 + 4 * op->ob_size * sizeof(char);
242
	PyObject *v = PyString_FromStringAndSize((char *)NULL, newsize);
Guido van Rossum's avatar
Guido van Rossum committed
243
	if (v == NULL) {
244
		return NULL;
Guido van Rossum's avatar
Guido van Rossum committed
245 246 247 248 249
	}
	else {
		register int i;
		register char c;
		register char *p;
250 251 252 253 254 255 256
		int quote;

		/* figure out which quote to use; single is prefered */
		quote = '\'';
		if (strchr(op->ob_sval, '\'') && !strchr(op->ob_sval, '"'))
			quote = '"';

257
		p = ((PyStringObject *)v)->ob_sval;
258
		*p++ = quote;
Guido van Rossum's avatar
Guido van Rossum committed
259 260
		for (i = 0; i < op->ob_size; i++) {
			c = op->ob_sval[i];
261
			if (c == quote || c == '\\')
Guido van Rossum's avatar
Guido van Rossum committed
262 263
				*p++ = '\\', *p++ = c;
			else if (c < ' ' || c >= 0177) {
264
				sprintf(p, "\\%03o", c & 0377);
Guido van Rossum's avatar
Guido van Rossum committed
265 266 267 268 269 270
				while (*p != '\0')
					p++;
			}
			else
				*p++ = c;
		}
271
		*p++ = quote;
Guido van Rossum's avatar
Guido van Rossum committed
272
		*p = '\0';
273 274
		_PyString_Resize(
			&v, (int) (p - ((PyStringObject *)v)->ob_sval));
Guido van Rossum's avatar
Guido van Rossum committed
275
		return v;
Guido van Rossum's avatar
Guido van Rossum committed
276 277 278 279
	}
}

static int
280
string_length(a)
281
	PyStringObject *a;
Guido van Rossum's avatar
Guido van Rossum committed
282 283 284 285
{
	return a->ob_size;
}

286
static PyObject *
287
string_concat(a, bb)
288 289
	register PyStringObject *a;
	register PyObject *bb;
Guido van Rossum's avatar
Guido van Rossum committed
290 291
{
	register unsigned int size;
292 293 294
	register PyStringObject *op;
	if (!PyString_Check(bb)) {
		PyErr_BadArgument();
Guido van Rossum's avatar
Guido van Rossum committed
295 296
		return NULL;
	}
297
#define b ((PyStringObject *)bb)
Guido van Rossum's avatar
Guido van Rossum committed
298 299
	/* Optimize cases with empty left or right operand */
	if (a->ob_size == 0) {
300
		Py_INCREF(bb);
Guido van Rossum's avatar
Guido van Rossum committed
301 302 303
		return bb;
	}
	if (b->ob_size == 0) {
304 305
		Py_INCREF(a);
		return (PyObject *)a;
Guido van Rossum's avatar
Guido van Rossum committed
306 307
	}
	size = a->ob_size + b->ob_size;
308 309
	op = (PyStringObject *)
		malloc(sizeof(PyStringObject) + size * sizeof(char));
Guido van Rossum's avatar
Guido van Rossum committed
310
	if (op == NULL)
311 312
		return PyErr_NoMemory();
	op->ob_type = &PyString_Type;
Guido van Rossum's avatar
Guido van Rossum committed
313
	op->ob_size = size;
314 315
#ifdef CACHE_HASH
	op->ob_shash = -1;
Guido van Rossum's avatar
Guido van Rossum committed
316 317 318
#endif
#ifdef INTERN_STRINGS
	op->ob_sinterned = NULL;
319
#endif
320
	_Py_NewReference(op);
Guido van Rossum's avatar
Guido van Rossum committed
321 322 323
	memcpy(op->ob_sval, a->ob_sval, (int) a->ob_size);
	memcpy(op->ob_sval + a->ob_size, b->ob_sval, (int) b->ob_size);
	op->ob_sval[size] = '\0';
324
	return (PyObject *) op;
Guido van Rossum's avatar
Guido van Rossum committed
325 326 327
#undef b
}

328
static PyObject *
329
string_repeat(a, n)
330
	register PyStringObject *a;
Guido van Rossum's avatar
Guido van Rossum committed
331 332 333
	register int n;
{
	register int i;
334
	register int size;
335
	register PyStringObject *op;
Guido van Rossum's avatar
Guido van Rossum committed
336 337 338 339
	if (n < 0)
		n = 0;
	size = a->ob_size * n;
	if (size == a->ob_size) {
340 341
		Py_INCREF(a);
		return (PyObject *)a;
Guido van Rossum's avatar
Guido van Rossum committed
342
	}
343 344
	op = (PyStringObject *)
		malloc(sizeof(PyStringObject) + size * sizeof(char));
Guido van Rossum's avatar
Guido van Rossum committed
345
	if (op == NULL)
346 347
		return PyErr_NoMemory();
	op->ob_type = &PyString_Type;
Guido van Rossum's avatar
Guido van Rossum committed
348
	op->ob_size = size;
349 350
#ifdef CACHE_HASH
	op->ob_shash = -1;
Guido van Rossum's avatar
Guido van Rossum committed
351 352 353
#endif
#ifdef INTERN_STRINGS
	op->ob_sinterned = NULL;
354
#endif
355
	_Py_NewReference(op);
Guido van Rossum's avatar
Guido van Rossum committed
356 357 358
	for (i = 0; i < size; i += a->ob_size)
		memcpy(op->ob_sval+i, a->ob_sval, (int) a->ob_size);
	op->ob_sval[size] = '\0';
359
	return (PyObject *) op;
Guido van Rossum's avatar
Guido van Rossum committed
360 361 362 363
}

/* String slice a[i:j] consists of characters a[i] ... a[j-1] */

364
static PyObject *
365
string_slice(a, i, j)
366
	register PyStringObject *a;
Guido van Rossum's avatar
Guido van Rossum committed
367 368 369 370 371 372 373 374 375
	register int i, j; /* May be negative! */
{
	if (i < 0)
		i = 0;
	if (j < 0)
		j = 0; /* Avoid signed/unsigned bug in next line */
	if (j > a->ob_size)
		j = a->ob_size;
	if (i == 0 && j == a->ob_size) { /* It's the same as a */
376 377
		Py_INCREF(a);
		return (PyObject *)a;
Guido van Rossum's avatar
Guido van Rossum committed
378 379 380
	}
	if (j < i)
		j = i;
381
	return PyString_FromStringAndSize(a->ob_sval + i, (int) (j-i));
Guido van Rossum's avatar
Guido van Rossum committed
382 383
}

384
static PyObject *
385
string_item(a, i)
386
	PyStringObject *a;
Guido van Rossum's avatar
Guido van Rossum committed
387 388
	register int i;
{
389
	int c;
390
	PyObject *v;
Guido van Rossum's avatar
Guido van Rossum committed
391
	if (i < 0 || i >= a->ob_size) {
392
		PyErr_SetString(PyExc_IndexError, "string index out of range");
Guido van Rossum's avatar
Guido van Rossum committed
393 394
		return NULL;
	}
395
	c = a->ob_sval[i] & UCHAR_MAX;
396
	v = (PyObject *) characters[c];
397 398 399 400
#ifdef COUNT_ALLOCS
	if (v != NULL)
		one_strings++;
#endif
401
	if (v == NULL) {
402
		v = PyString_FromStringAndSize((char *)NULL, 1);
403 404
		if (v == NULL)
			return NULL;
405 406
		characters[c] = (PyStringObject *) v;
		((PyStringObject *)v)->ob_sval[0] = c;
407
	}
408
	Py_INCREF(v);
409
	return v;
Guido van Rossum's avatar
Guido van Rossum committed
410 411 412
}

static int
413
string_compare(a, b)
414
	PyStringObject *a, *b;
Guido van Rossum's avatar
Guido van Rossum committed
415
{
416 417
	int len_a = a->ob_size, len_b = b->ob_size;
	int min_len = (len_a < len_b) ? len_a : len_b;
418 419
	int cmp;
	if (min_len > 0) {
420
		cmp = Py_CHARMASK(*a->ob_sval) - Py_CHARMASK(*b->ob_sval);
421 422 423 424 425
		if (cmp == 0)
			cmp = memcmp(a->ob_sval, b->ob_sval, min_len);
		if (cmp != 0)
			return cmp;
	}
426
	return (len_a < len_b) ? -1 : (len_a > len_b) ? 1 : 0;
Guido van Rossum's avatar
Guido van Rossum committed
427 428
}

429 430
static long
string_hash(a)
431
	PyStringObject *a;
432
{
433 434 435 436 437 438 439
	register int len;
	register unsigned char *p;
	register long x;

#ifdef CACHE_HASH
	if (a->ob_shash != -1)
		return a->ob_shash;
440 441 442
#ifdef INTERN_STRINGS
	if (a->ob_sinterned != NULL)
		return (a->ob_shash =
443
			((PyStringObject *)(a->ob_sinterned))->ob_shash);
444
#endif
445 446 447 448
#endif
	len = a->ob_size;
	p = (unsigned char *) a->ob_sval;
	x = *p << 7;
449
	while (--len >= 0)
450
		x = (1000003*x) ^ *p++;
451 452 453
	x ^= a->ob_size;
	if (x == -1)
		x = -2;
454 455 456
#ifdef CACHE_HASH
	a->ob_shash = x;
#endif
457 458 459
	return x;
}

460 461 462 463 464 465 466
static int
string_buffer_getreadbuf(self, index, ptr)
	PyStringObject *self;
	int index;
	const void **ptr;
{
	if ( index != 0 ) {
467
		PyErr_SetString(PyExc_SystemError,
468
				"accessing non-existent string segment");
469 470 471 472 473 474 475 476 477 478 479 480
		return -1;
	}
	*ptr = (void *)self->ob_sval;
	return self->ob_size;
}

static int
string_buffer_getwritebuf(self, index, ptr)
	PyStringObject *self;
	int index;
	const void **ptr;
{
481
	PyErr_SetString(PyExc_TypeError,
482
			"Cannot use string as modifiable buffer");
483 484 485 486 487 488 489 490 491 492 493 494 495
	return -1;
}

static int
string_buffer_getsegcount(self, lenp)
	PyStringObject *self;
	int *lenp;
{
	if ( lenp )
		*lenp = self->ob_size;
	return 1;
}

496 497 498 499 500 501 502 503 504 505 506 507 508 509 510
static int
string_buffer_getcharbuf(self, index, ptr)
	PyStringObject *self;
	int index;
	const char **ptr;
{
	if ( index != 0 ) {
		PyErr_SetString(PyExc_SystemError,
				"accessing non-existent string segment");
		return -1;
	}
	*ptr = self->ob_sval;
	return self->ob_size;
}

511
static PySequenceMethods string_as_sequence = {
512 513 514 515 516
	(inquiry)string_length, /*sq_length*/
	(binaryfunc)string_concat, /*sq_concat*/
	(intargfunc)string_repeat, /*sq_repeat*/
	(intargfunc)string_item, /*sq_item*/
	(intintargfunc)string_slice, /*sq_slice*/
517 518
	0,		/*sq_ass_item*/
	0,		/*sq_ass_slice*/
Guido van Rossum's avatar
Guido van Rossum committed
519 520
};

521 522 523 524
static PyBufferProcs string_as_buffer = {
	(getreadbufferproc)string_buffer_getreadbuf,
	(getwritebufferproc)string_buffer_getwritebuf,
	(getsegcountproc)string_buffer_getsegcount,
525
	(getcharbufferproc)string_buffer_getcharbuf,
526 527
};

528 529
PyTypeObject PyString_Type = {
	PyObject_HEAD_INIT(&PyType_Type)
Guido van Rossum's avatar
Guido van Rossum committed
530 531
	0,
	"string",
532
	sizeof(PyStringObject),
Guido van Rossum's avatar
Guido van Rossum committed
533
	sizeof(char),
534 535
	(destructor)string_dealloc, /*tp_dealloc*/
	(printfunc)string_print, /*tp_print*/
Guido van Rossum's avatar
Guido van Rossum committed
536 537
	0,		/*tp_getattr*/
	0,		/*tp_setattr*/
538 539
	(cmpfunc)string_compare, /*tp_compare*/
	(reprfunc)string_repr, /*tp_repr*/
Guido van Rossum's avatar
Guido van Rossum committed
540 541 542
	0,		/*tp_as_number*/
	&string_as_sequence,	/*tp_as_sequence*/
	0,		/*tp_as_mapping*/
543
	(hashfunc)string_hash, /*tp_hash*/
Guido van Rossum's avatar
Guido van Rossum committed
544 545 546 547
	0,		/*tp_call*/
	0,		/*tp_str*/
	0,		/*tp_getattro*/
	0,		/*tp_setattro*/
548
	&string_as_buffer,	/*tp_as_buffer*/
549
	Py_TPFLAGS_DEFAULT,	/*tp_flags*/
Guido van Rossum's avatar
Guido van Rossum committed
550
	0,		/*tp_doc*/
Guido van Rossum's avatar
Guido van Rossum committed
551 552 553
};

void
554 555 556
PyString_Concat(pv, w)
	register PyObject **pv;
	register PyObject *w;
Guido van Rossum's avatar
Guido van Rossum committed
557
{
558
	register PyObject *v;
559 560
	if (*pv == NULL)
		return;
561 562
	if (w == NULL || !PyString_Check(*pv)) {
		Py_DECREF(*pv);
563
		*pv = NULL;
Guido van Rossum's avatar
Guido van Rossum committed
564
		return;
565
	}
566 567
	v = string_concat((PyStringObject *) *pv, w);
	Py_DECREF(*pv);
Guido van Rossum's avatar
Guido van Rossum committed
568 569 570
	*pv = v;
}

571
void
572 573 574
PyString_ConcatAndDel(pv, w)
	register PyObject **pv;
	register PyObject *w;
575
{
576 577
	PyString_Concat(pv, w);
	Py_XDECREF(w);
578 579 580
}


Guido van Rossum's avatar
Guido van Rossum committed
581 582 583 584 585 586 587 588
/* 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... */

int
589 590
_PyString_Resize(pv, newsize)
	PyObject **pv;
Guido van Rossum's avatar
Guido van Rossum committed
591 592
	int newsize;
{
593 594
	register PyObject *v;
	register PyStringObject *sv;
595
	v = *pv;
596
	if (!PyString_Check(v) || v->ob_refcnt != 1) {
Guido van Rossum's avatar
Guido van Rossum committed
597
		*pv = 0;
598 599
		Py_DECREF(v);
		PyErr_BadInternalCall();
Guido van Rossum's avatar
Guido van Rossum committed
600
		return -1;
Guido van Rossum's avatar
Guido van Rossum committed
601
	}
602
	/* XXX UNREF/NEWREF interface should be more symmetrical */
603
#ifdef Py_REF_DEBUG
604
	--_Py_RefTotal;
605
#endif
606 607
	_Py_ForgetReference(v);
	*pv = (PyObject *)
Guido van Rossum's avatar
Guido van Rossum committed
608
		realloc((char *)v,
609
			sizeof(PyStringObject) + newsize * sizeof(char));
Guido van Rossum's avatar
Guido van Rossum committed
610
	if (*pv == NULL) {
611 612
		PyMem_DEL(v);
		PyErr_NoMemory();
Guido van Rossum's avatar
Guido van Rossum committed
613
		return -1;
Guido van Rossum's avatar
Guido van Rossum committed
614
	}
615 616
	_Py_NewReference(*pv);
	sv = (PyStringObject *) *pv;
617 618
	sv->ob_size = newsize;
	sv->ob_sval[newsize] = '\0';
Guido van Rossum's avatar
Guido van Rossum committed
619 620
	return 0;
}
621 622 623

/* Helpers for formatstring */

624
static PyObject *
625
getnextarg(args, arglen, p_argidx)
626
	PyObject *args;
627 628 629 630 631 632 633 634 635
	int arglen;
	int *p_argidx;
{
	int argidx = *p_argidx;
	if (argidx < arglen) {
		(*p_argidx)++;
		if (arglen < 0)
			return args;
		else
636
			return PyTuple_GetItem(args, argidx);
637
	}
638 639
	PyErr_SetString(PyExc_TypeError,
			"not enough arguments for format string");
640 641 642 643 644 645 646 647 648
	return NULL;
}

#define F_LJUST (1<<0)
#define F_SIGN	(1<<1)
#define F_BLANK (1<<2)
#define F_ALT	(1<<3)
#define F_ZERO	(1<<4)

649 650 651
static int
formatfloat(buf, flags, prec, type, v)
	char *buf;
652 653 654
	int flags;
	int prec;
	int type;
655
	PyObject *v;
656 657 658
{
	char fmt[20];
	double x;
659
	if (!PyArg_Parse(v, "d;float argument required", &x))
660
		return -1;
661 662 663 664 665 666 667 668
	if (prec < 0)
		prec = 6;
	if (prec > 50)
		prec = 50; /* Arbitrary limitation */
	if (type == 'f' && fabs(x)/1e25 >= 1e25)
		type = 'g';
	sprintf(fmt, "%%%s.%d%c", (flags&F_ALT) ? "#" : "", prec, type);
	sprintf(buf, fmt, x);
669
	return strlen(buf);
670 671
}

672 673 674
static int
formatint(buf, flags, prec, type, v)
	char *buf;
675 676 677
	int flags;
	int prec;
	int type;
678
	PyObject *v;
679 680 681
{
	char fmt[20];
	long x;
682
	if (!PyArg_Parse(v, "l;int argument required", &x))
683
		return -1;
684 685 686 687
	if (prec < 0)
		prec = 1;
	sprintf(fmt, "%%%s.%dl%c", (flags&F_ALT) ? "#" : "", prec, type);
	sprintf(buf, fmt, x);
688
	return strlen(buf);
689 690
}

691 692 693
static int
formatchar(buf, v)
	char *buf;
694
	PyObject *v;
695
{
696 697
	if (PyString_Check(v)) {
		if (!PyArg_Parse(v, "c;%c requires int or char", &buf[0]))
698
			return -1;
699 700
	}
	else {
701
		if (!PyArg_Parse(v, "b;%c requires int or char", &buf[0]))
702
			return -1;
703 704
	}
	buf[1] = '\0';
705
	return 1;
706 707
}

708

709 710
/* fmt%(v1,v2,...) is roughly equivalent to sprintf(fmt, v1, v2, ...) */

711 712 713 714
PyObject *
PyString_Format(format, args)
	PyObject *format;
	PyObject *args;
715 716 717
{
	char *fmt, *res;
	int fmtcnt, rescnt, reslen, arglen, argidx;
718
	int args_owned = 0;
719 720 721 722
	PyObject *result;
	PyObject *dict = NULL;
	if (format == NULL || !PyString_Check(format) || args == NULL) {
		PyErr_BadInternalCall();
723 724
		return NULL;
	}
725 726
	fmt = PyString_AsString(format);
	fmtcnt = PyString_Size(format);
727
	reslen = rescnt = fmtcnt + 100;
728
	result = PyString_FromStringAndSize((char *)NULL, reslen);
729 730
	if (result == NULL)
		return NULL;
731 732 733
	res = PyString_AsString(result);
	if (PyTuple_Check(args)) {
		arglen = PyTuple_Size(args);
734 735 736 737 738 739
		argidx = 0;
	}
	else {
		arglen = -1;
		argidx = -2;
	}
740 741
	if (args->ob_type->tp_as_mapping)
		dict = args;
742 743 744
	while (--fmtcnt >= 0) {
		if (*fmt != '%') {
			if (--rescnt < 0) {
745 746
				rescnt = fmtcnt + 100;
				reslen += rescnt;
747
				if (_PyString_Resize(&result, reslen) < 0)
748
					return NULL;
749 750
				res = PyString_AsString(result)
					+ reslen - rescnt;
751
				--rescnt;
752 753 754 755 756 757 758 759 760
			}
			*res++ = *fmt++;
		}
		else {
			/* Got a format specifier */
			int flags = 0;
			int width = -1;
			int prec = -1;
			int size = 0;
761
			int c = '\0';
762
			int fill;
763 764
			PyObject *v = NULL;
			PyObject *temp = NULL;
765 766 767
			char *buf;
			int sign;
			int len;
768
			char tmpbuf[120]; /* For format{float,int,char}() */
Guido van Rossum's avatar
Guido van Rossum committed
769
			fmt++;
770 771 772
			if (*fmt == '(') {
				char *keystart;
				int keylen;
773
				PyObject *key;
774
				int pcount = 1;
775 776

				if (dict == NULL) {
777
					PyErr_SetString(PyExc_TypeError,
778 779 780 781 782 783
						 "format requires a mapping"); 
					goto error;
				}
				++fmt;
				--fmtcnt;
				keystart = fmt;
784 785 786 787 788 789
				/* Skip over balanced parentheses */
				while (pcount > 0 && --fmtcnt >= 0) {
					if (*fmt == ')')
						--pcount;
					else if (*fmt == '(')
						++pcount;
790
					fmt++;
791 792 793
				}
				keylen = fmt - keystart - 1;
				if (fmtcnt < 0 || pcount > 0) {
794
					PyErr_SetString(PyExc_ValueError,
795 796 797
						   "incomplete format key");
					goto error;
				}
798 799
				key = PyString_FromStringAndSize(keystart,
								 keylen);
800 801
				if (key == NULL)
					goto error;
802
				if (args_owned) {
803
					Py_DECREF(args);
804 805 806
					args_owned = 0;
				}
				args = PyObject_GetItem(dict, key);
807
				Py_DECREF(key);
808 809 810
				if (args == NULL) {
					goto error;
				}
811
				args_owned = 1;
812 813 814
				arglen = -1;
				argidx = -2;
			}
815 816 817 818 819 820 821 822 823 824 825 826 827 828
			while (--fmtcnt >= 0) {
				switch (c = *fmt++) {
				case '-': flags |= F_LJUST; continue;
				case '+': flags |= F_SIGN; continue;
				case ' ': flags |= F_BLANK; continue;
				case '#': flags |= F_ALT; continue;
				case '0': flags |= F_ZERO; continue;
				}
				break;
			}
			if (c == '*') {
				v = getnextarg(args, arglen, &argidx);
				if (v == NULL)
					goto error;
829 830 831
				if (!PyInt_Check(v)) {
					PyErr_SetString(PyExc_TypeError,
							"* wants int");
832 833
					goto error;
				}
834
				width = PyInt_AsLong(v);
835 836 837 838
				if (width < 0) {
					flags |= F_LJUST;
					width = -width;
				}
839 840 841
				if (--fmtcnt >= 0)
					c = *fmt++;
			}
842
			else if (c >= 0 && isdigit(c)) {
843 844
				width = c - '0';
				while (--fmtcnt >= 0) {
845
					c = Py_CHARMASK(*fmt++);
846 847 848
					if (!isdigit(c))
						break;
					if ((width*10) / 10 != width) {
849 850 851
						PyErr_SetString(
							PyExc_ValueError,
							"width too big");
852 853 854 855 856 857 858 859 860 861 862 863 864
						goto error;
					}
					width = width*10 + (c - '0');
				}
			}
			if (c == '.') {
				prec = 0;
				if (--fmtcnt >= 0)
					c = *fmt++;
				if (c == '*') {
					v = getnextarg(args, arglen, &argidx);
					if (v == NULL)
						goto error;
865 866 867 868
					if (!PyInt_Check(v)) {
						PyErr_SetString(
							PyExc_TypeError,
							"* wants int");
869 870
						goto error;
					}
871
					prec = PyInt_AsLong(v);
872 873 874 875 876
					if (prec < 0)
						prec = 0;
					if (--fmtcnt >= 0)
						c = *fmt++;
				}
877
				else if (c >= 0 && isdigit(c)) {
878 879
					prec = c - '0';
					while (--fmtcnt >= 0) {
880
						c = Py_CHARMASK(*fmt++);
881 882 883
						if (!isdigit(c))
							break;
						if ((prec*10) / 10 != prec) {
884 885
							PyErr_SetString(
							    PyExc_ValueError,
886 887 888 889 890 891 892 893 894 895 896 897 898 899 900
							    "prec too big");
							goto error;
						}
						prec = prec*10 + (c - '0');
					}
				}
			} /* prec */
			if (fmtcnt >= 0) {
				if (c == 'h' || c == 'l' || c == 'L') {
					size = c;
					if (--fmtcnt >= 0)
						c = *fmt++;
				}
			}
			if (fmtcnt < 0) {
901 902
				PyErr_SetString(PyExc_ValueError,
						"incomplete format");
903 904 905 906 907 908 909 910 911 912 913 914 915 916 917
				goto error;
			}
			if (c != '%') {
				v = getnextarg(args, arglen, &argidx);
				if (v == NULL)
					goto error;
			}
			sign = 0;
			fill = ' ';
			switch (c) {
			case '%':
				buf = "%";
				len = 1;
				break;
			case 's':
918
				temp = PyObject_Str(v);
919
				if (temp == NULL)
920
					goto error;
921 922 923 924 925
				if (!PyString_Check(temp)) {
					PyErr_SetString(PyExc_TypeError,
					  "%s argument has non-string str()");
					goto error;
				}
926 927
				buf = PyString_AsString(temp);
				len = PyString_Size(temp);
928 929 930 931 932 933 934 935 936 937 938
				if (prec >= 0 && len > prec)
					len = prec;
				break;
			case 'i':
			case 'd':
			case 'u':
			case 'o':
			case 'x':
			case 'X':
				if (c == 'i')
					c = 'd';
939 940 941
				buf = tmpbuf;
				len = formatint(buf, flags, prec, c, v);
				if (len < 0)
942 943
					goto error;
				sign = (c == 'd');
944
				if (flags&F_ZERO) {
945
					fill = '0';
946 947 948 949 950 951 952 953 954 955 956 957
					if ((flags&F_ALT) &&
					    (c == 'x' || c == 'X') &&
					    buf[0] == '0' && buf[1] == c) {
						*res++ = *buf++;
						*res++ = *buf++;
						rescnt -= 2;
						len -= 2;
						width -= 2;
						if (width < 0)
							width = 0;
					}
				}
958 959 960 961 962 963
				break;
			case 'e':
			case 'E':
			case 'f':
			case 'g':
			case 'G':
964 965 966
				buf = tmpbuf;
				len = formatfloat(buf, flags, prec, c, v);
				if (len < 0)
967 968 969 970 971 972
					goto error;
				sign = 1;
				if (flags&F_ZERO)
					fill = '0';
				break;
			case 'c':
973 974 975
				buf = tmpbuf;
				len = formatchar(buf, v);
				if (len < 0)
976 977 978
					goto error;
				break;
			default:
979 980 981
				PyErr_Format(PyExc_ValueError,
				"unsupported format character '%c' (0x%x)",
					c, c);
982 983 984 985 986 987 988 989 990 991 992 993 994 995 996 997 998
				goto error;
			}
			if (sign) {
				if (*buf == '-' || *buf == '+') {
					sign = *buf++;
					len--;
				}
				else if (flags & F_SIGN)
					sign = '+';
				else if (flags & F_BLANK)
					sign = ' ';
				else
					sign = '\0';
			}
			if (width < len)
				width = len;
			if (rescnt < width + (sign != '\0')) {
999 1000 1001
				reslen -= rescnt;
				rescnt = width + fmtcnt + 100;
				reslen += rescnt;
1002
				if (_PyString_Resize(&result, reslen) < 0)
1003
					return NULL;
1004 1005
				res = PyString_AsString(result)
					+ reslen - rescnt;
1006 1007
			}
			if (sign) {
Guido van Rossum's avatar
Guido van Rossum committed
1008 1009
				if (fill != ' ')
					*res++ = sign;
1010 1011 1012 1013 1014 1015 1016 1017 1018 1019
				rescnt--;
				if (width > len)
					width--;
			}
			if (width > len && !(flags&F_LJUST)) {
				do {
					--rescnt;
					*res++ = fill;
				} while (--width > len);
			}
Guido van Rossum's avatar
Guido van Rossum committed
1020
			if (sign && fill == ' ')
1021
				*res++ = sign;
1022 1023 1024 1025 1026 1027 1028
			memcpy(res, buf, len);
			res += len;
			rescnt -= len;
			while (--width >= len) {
				--rescnt;
				*res++ = ' ';
			}
1029
                        if (dict && (argidx < arglen) && c != '%') {
1030
                                PyErr_SetString(PyExc_TypeError,
1031 1032 1033
                                           "not all arguments converted");
                                goto error;
                        }
1034
			Py_XDECREF(temp);
1035 1036
		} /* '%' */
	} /* until end */
1037
	if (argidx < arglen && !dict) {
1038 1039
		PyErr_SetString(PyExc_TypeError,
				"not all arguments converted");
1040 1041
		goto error;
	}
1042
	if (args_owned) {
1043
		Py_DECREF(args);
1044
	}
1045
	_PyString_Resize(&result, reslen - rescnt);
1046 1047
	return result;
 error:
1048
	Py_DECREF(result);
1049
	if (args_owned) {
1050
		Py_DECREF(args);
1051
	}
1052 1053
	return NULL;
}
Guido van Rossum's avatar
Guido van Rossum committed
1054 1055 1056 1057 1058 1059 1060 1061 1062 1063 1064 1065 1066 1067 1068 1069 1070 1071 1072 1073 1074 1075 1076 1077 1078 1079 1080 1081 1082 1083 1084 1085 1086 1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097 1098 1099 1100 1101 1102 1103 1104 1105 1106 1107


#ifdef INTERN_STRINGS

static PyObject *interned;

void
PyString_InternInPlace(p)
	PyObject **p;
{
	register PyStringObject *s = (PyStringObject *)(*p);
	PyObject *t;
	if (s == NULL || !PyString_Check(s))
		Py_FatalError("PyString_InternInPlace: strings only please!");
	if ((t = s->ob_sinterned) != NULL) {
		if (t == (PyObject *)s)
			return;
		Py_INCREF(t);
		*p = t;
		Py_DECREF(s);
		return;
	}
	if (interned == NULL) {
		interned = PyDict_New();
		if (interned == NULL)
			return;
	}
	if ((t = PyDict_GetItem(interned, (PyObject *)s)) != NULL) {
		Py_INCREF(t);
		*p = s->ob_sinterned = t;
		Py_DECREF(s);
		return;
	}
	t = (PyObject *)s;
	if (PyDict_SetItem(interned, t, t) == 0) {
		s->ob_sinterned = t;
		return;
	}
	PyErr_Clear();
}


PyObject *
PyString_InternFromString(cp)
	const char *cp;
{
	PyObject *s = PyString_FromString(cp);
	if (s == NULL)
		return NULL;
	PyString_InternInPlace(&s);
	return s;
}

#endif
1108 1109 1110 1111 1112 1113 1114 1115 1116 1117 1118 1119 1120

void
PyString_Fini()
{
	int i;
	for (i = 0; i < UCHAR_MAX + 1; i++) {
		Py_XDECREF(characters[i]);
		characters[i] = NULL;
	}
#ifndef DONT_SHARE_SHORT_STRINGS
	Py_XDECREF(nullstring);
	nullstring = NULL;
#endif
1121 1122 1123 1124 1125 1126 1127 1128 1129 1130 1131 1132 1133 1134 1135 1136
#ifdef INTERN_STRINGS
	if (interned) {
		int pos, changed;
		PyObject *key, *value;
		do {
			changed = 0;
			pos = 0;
			while (PyDict_Next(interned, &pos, &key, &value)) {
				if (key->ob_refcnt == 2 && key == value) {
					PyDict_DelItem(interned, key);
					changed = 1;
				}
			}
		} while (changed);
	}
#endif
1137
}