AppPkg/Applications/Python/Python-2.7.10/Python/asdl.c

   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 }