/src/Python-3.8.3/Python/asdl.c
Line | Count | Source |
1 | | #include "Python.h" |
2 | | #include "asdl.h" |
3 | | |
4 | | asdl_seq * |
5 | | _Py_asdl_seq_new(Py_ssize_t size, PyArena *arena) |
6 | 302 | { |
7 | 302 | asdl_seq *seq = NULL; |
8 | 302 | size_t n; |
9 | | |
10 | | /* check size is sane */ |
11 | 302 | if (size < 0 || |
12 | 302 | (size && (((size_t)size - 1) > (SIZE_MAX / sizeof(void *))))) { |
13 | 0 | PyErr_NoMemory(); |
14 | 0 | return NULL; |
15 | 0 | } |
16 | 302 | n = (size ? (sizeof(void *) * (size - 1)) : 0); |
17 | | |
18 | | /* check if size can be added safely */ |
19 | 302 | if (n > SIZE_MAX - sizeof(asdl_seq)) { |
20 | 0 | PyErr_NoMemory(); |
21 | 0 | return NULL; |
22 | 0 | } |
23 | 302 | n += sizeof(asdl_seq); |
24 | | |
25 | 302 | seq = (asdl_seq *)PyArena_Malloc(arena, n); |
26 | 302 | if (!seq) { |
27 | 0 | PyErr_NoMemory(); |
28 | 0 | return NULL; |
29 | 0 | } |
30 | 302 | memset(seq, 0, n); |
31 | 302 | seq->size = size; |
32 | 302 | return seq; |
33 | 302 | } |
34 | | |
35 | | asdl_int_seq * |
36 | | _Py_asdl_int_seq_new(Py_ssize_t size, PyArena *arena) |
37 | 22 | { |
38 | 22 | asdl_int_seq *seq = NULL; |
39 | 22 | size_t n; |
40 | | |
41 | | /* check size is sane */ |
42 | 22 | if (size < 0 || |
43 | 22 | (size && (((size_t)size - 1) > (SIZE_MAX / sizeof(void *))))) { |
44 | 0 | PyErr_NoMemory(); |
45 | 0 | return NULL; |
46 | 0 | } |
47 | 22 | n = (size ? (sizeof(void *) * (size - 1)) : 0); |
48 | | |
49 | | /* check if size can be added safely */ |
50 | 22 | if (n > SIZE_MAX - sizeof(asdl_seq)) { |
51 | 0 | PyErr_NoMemory(); |
52 | 0 | return NULL; |
53 | 0 | } |
54 | 22 | n += sizeof(asdl_seq); |
55 | | |
56 | 22 | seq = (asdl_int_seq *)PyArena_Malloc(arena, n); |
57 | 22 | if (!seq) { |
58 | 0 | PyErr_NoMemory(); |
59 | 0 | return NULL; |
60 | 0 | } |
61 | 22 | memset(seq, 0, n); |
62 | 22 | seq->size = size; |
63 | 22 | return seq; |
64 | 22 | } |