• Gregory P. Smith's avatar
    Merge in release25-maint r60793: · 9d53457e
    Gregory P. Smith yazdı
     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.
    9d53457e
asdl.c 1.21 KB
#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;
}