1 #include "Python.h"
2 #include "asdl.h"
3 
4 asdl_seq *
asdl_seq_new(int size,PyArena * arena)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 *
asdl_int_seq_new(int size,PyArena * arena)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 }
65