mirror of
https://github.com/python/cpython.git
synced 2025-07-30 06:34:15 +00:00
Forward-port of r52136,52138: a review of overflow-detecting code.
* unified the way intobject, longobject and mystrtoul handle values around -sys.maxint-1. * in general, trying to entierely avoid overflows in any computation involving signed ints or longs is extremely involved. Fixed a few simple cases where a compiler might be too clever (but that's all guesswork). * more overflow checks against bad data in marshal.c. * 2.5 specific: fixed a number of places that were still confusing int and Py_ssize_t. Some of them could potentially have caused "real-world" breakage. * list.pop(x): fixing overflow issues on x was messy. I just reverted to PyArg_ParseTuple("n"), which does the right thing. (An obscure test was trying to give a Decimal to list.pop()... doesn't make sense any more IMHO) * trying to write a few tests...
This commit is contained in:
parent
0d2f498a4c
commit
7ccbca93a2
19 changed files with 186 additions and 106 deletions
|
@ -1001,6 +1001,7 @@ getline_via_fgets(FILE *fp)
|
|||
size_t nfree; /* # of free buffer slots; pvend-pvfree */
|
||||
size_t total_v_size; /* total # of slots in buffer */
|
||||
size_t increment; /* amount to increment the buffer */
|
||||
size_t prev_v_size;
|
||||
|
||||
/* Optimize for normal case: avoid _PyString_Resize if at all
|
||||
* possible via first reading into stack buffer "buf".
|
||||
|
@ -1115,8 +1116,11 @@ getline_via_fgets(FILE *fp)
|
|||
/* expand buffer and try again */
|
||||
assert(*(pvend-1) == '\0');
|
||||
increment = total_v_size >> 2; /* mild exponential growth */
|
||||
prev_v_size = total_v_size;
|
||||
total_v_size += increment;
|
||||
if (total_v_size > PY_SSIZE_T_MAX) {
|
||||
/* check for overflow */
|
||||
if (total_v_size <= prev_v_size ||
|
||||
total_v_size > PY_SSIZE_T_MAX) {
|
||||
PyErr_SetString(PyExc_OverflowError,
|
||||
"line is longer than a Python string can hold");
|
||||
Py_DECREF(v);
|
||||
|
@ -1125,7 +1129,7 @@ getline_via_fgets(FILE *fp)
|
|||
if (_PyString_Resize(&v, (int)total_v_size) < 0)
|
||||
return NULL;
|
||||
/* overwrite the trailing null byte */
|
||||
pvfree = BUF(v) + (total_v_size - increment - 1);
|
||||
pvfree = BUF(v) + (prev_v_size - 1);
|
||||
}
|
||||
if (BUF(v) + total_v_size != p)
|
||||
_PyString_Resize(&v, p - BUF(v));
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue