1 #include "Python.h"

     2 #include "asdl.h"

     3 

     4 asdl_seq *

     5 asdl_seq_new(int size, PyArena *arena)

     6 {

     7 	asdl_seq *seq = NULL;

     8 	size_t n = (size ? (sizeof(void *) * (size - 1)) : 0);

     9 

    10 	/* check size is sane */

    11 	if (size < 0 || size == INT_MIN || 

    12 		(size && ((size - 1) > (PY_SIZE_MAX / sizeof(void *))))) {

    13 		PyErr_NoMemory();

    14 		return NULL;

    15 	}

    16 

    17 	/* check if size can be added safely */

    18 	if (n > PY_SIZE_MAX - sizeof(asdl_seq)) {

    19 		PyErr_NoMemory();

    20 		return NULL;

    21 	}

    22 

    23 	n += sizeof(asdl_seq);

    24 

    25 	seq = (asdl_seq *)PyArena_Malloc(arena, n);

    26 	if (!seq) {

    27 		PyErr_NoMemory();

    28 		return NULL;

    29 	}

    30 	memset(seq, 0, n);

    31 	seq->size = size;

    32 	return seq;

    33 }

    34 

    35 asdl_int_seq *

    36 asdl_int_seq_new(int size, PyArena *arena)

    37 {

    38 	asdl_int_seq *seq = NULL;

    39 	size_t n = (size ? (sizeof(void *) * (size - 1)) : 0);

    40 

    41 	/* check size is sane */

    42 	if (size < 0 || size == INT_MIN || 

    43 		(size && ((size - 1) > (PY_SIZE_MAX / sizeof(void *))))) {

    44 			PyErr_NoMemory();

    45 			return NULL;

    46 	}

    47 

    48 	/* check if size can be added safely */

    49 	if (n > PY_SIZE_MAX - sizeof(asdl_seq)) {

    50 		PyErr_NoMemory();

    51 		return NULL;

    52 	}

    53 

    54 	n += sizeof(asdl_seq);

    55 

    56 	seq = (asdl_int_seq *)PyArena_Malloc(arena, n);

    57 	if (!seq) {

    58 		PyErr_NoMemory();

    59 		return NULL;

    60 	}

    61 	memset(seq, 0, n);

    62 	seq->size = size;

    63 	return seq;

    64 }