#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;

}