1  /* Test of reallocarray function.
       2     Copyright (C) 2010-2023 Free Software Foundation, Inc.
       3  
       4     This program is free software: you can redistribute it and/or modify
       5     it under the terms of the GNU General Public License as published by
       6     the Free Software Foundation, either version 3 of the License, or
       7     (at your option) any later version.
       8  
       9     This program is distributed in the hope that it will be useful,
      10     but WITHOUT ANY WARRANTY; without even the implied warranty of
      11     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
      12     GNU General Public License for more details.
      13  
      14     You should have received a copy of the GNU General Public License
      15     along with this program.  If not, see <https://www.gnu.org/licenses/>.  */
      16  
      17  #include <config.h>
      18  
      19  /* Specification.  */
      20  #include <stdlib.h>
      21  
      22  #include <errno.h>
      23  #include <stdint.h>
      24  
      25  #include "signature.h"
      26  SIGNATURE_CHECK (reallocarray, void *, (void *, size_t, size_t));
      27  
      28  #include "macros.h"
      29  
      30  int
      31  main ()
      32  {
      33    /* Check that reallocarray fails when requested to allocate a block
      34       of memory larger than PTRDIFF_MAX or SIZE_MAX bytes.  */
      35    for (size_t n = 2; n != 0; n <<= 1)
      36      {
      37        void *volatile p = NULL;
      38  
      39        if (PTRDIFF_MAX / n + 1 <= SIZE_MAX)
      40          {
      41            p = reallocarray (p, PTRDIFF_MAX / n + 1, n);
      42            ASSERT (p == NULL);
      43            ASSERT (errno == ENOMEM);
      44          }
      45  
      46        p = reallocarray (p, SIZE_MAX / n + 1, n);
      47        ASSERT (p == NULL);
      48        ASSERT (errno == ENOMEM
      49                || errno == EOVERFLOW /* NetBSD */);
      50  
      51        /* Reallocarray should not crash with zero sizes.  */
      52        p = reallocarray (p, 0, n);
      53        p = reallocarray (p, n, 0);
      54        free (p);
      55      }
      56  
      57    return 0;
      58  }