mirror of
https://github.com/python/cpython.git
synced 2025-10-30 21:21:22 +00:00
9d53457e59
Added checks for integer overflows, contributed by Google. Some are only available if asserts are left in the code, in cases where they can't be triggered from Python code.
64 lines
1.2 KiB
C
64 lines
1.2 KiB
C
#include "Python.h"
|
|
#include "asdl.h"
|
|
|
|
asdl_seq *
|
|
asdl_seq_new(int size, PyArena *arena)
|
|
{
|
|
asdl_seq *seq = NULL;
|
|
size_t n = (size ? (sizeof(void *) * (size - 1)) : 0);
|
|
|
|
/* check size is sane */
|
|
if (size < 0 || size == INT_MIN ||
|
|
(size && ((size - 1) > (PY_SIZE_MAX / sizeof(void *))))) {
|
|
PyErr_NoMemory();
|
|
return NULL;
|
|
}
|
|
|
|
/* check if size can be added safely */
|
|
if (n > PY_SIZE_MAX - sizeof(asdl_seq)) {
|
|
PyErr_NoMemory();
|
|
return NULL;
|
|
}
|
|
|
|
n += sizeof(asdl_seq);
|
|
|
|
seq = (asdl_seq *)PyArena_Malloc(arena, n);
|
|
if (!seq) {
|
|
PyErr_NoMemory();
|
|
return NULL;
|
|
}
|
|
memset(seq, 0, n);
|
|
seq->size = size;
|
|
return seq;
|
|
}
|
|
|
|
asdl_int_seq *
|
|
asdl_int_seq_new(int size, PyArena *arena)
|
|
{
|
|
asdl_int_seq *seq = NULL;
|
|
size_t n = (size ? (sizeof(void *) * (size - 1)) : 0);
|
|
|
|
/* check size is sane */
|
|
if (size < 0 || size == INT_MIN ||
|
|
(size && ((size - 1) > (PY_SIZE_MAX / sizeof(void *))))) {
|
|
PyErr_NoMemory();
|
|
return NULL;
|
|
}
|
|
|
|
/* check if size can be added safely */
|
|
if (n > PY_SIZE_MAX - sizeof(asdl_seq)) {
|
|
PyErr_NoMemory();
|
|
return NULL;
|
|
}
|
|
|
|
n += sizeof(asdl_seq);
|
|
|
|
seq = (asdl_int_seq *)PyArena_Malloc(arena, n);
|
|
if (!seq) {
|
|
PyErr_NoMemory();
|
|
return NULL;
|
|
}
|
|
memset(seq, 0, n);
|
|
seq->size = size;
|
|
return seq;
|
|
}
|