This makes the rtvec_alloc argument size_t catching overflow and
truncated arguments (from "invalid" testcases), verifying the
argument against INT_MAX which is the limit set by the int
typed rtvec_def.num_elem member.

Bootstrapped and tested on x86_64-unknown-linux-gnu, pushed.

2021-05-12  Richard Biener  <rguent...@suse.de>

        PR middle-end/100547
        * rtl.h (rtvec_alloc): Make argument size_t.
        * rtl.c (rtvec_alloc): Verify the count is less than INT_MAX.
---
 gcc/rtl.c | 5 ++++-
 gcc/rtl.h | 2 +-
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/gcc/rtl.c b/gcc/rtl.c
index 035eadc3514..b0ba1ff684c 100644
--- a/gcc/rtl.c
+++ b/gcc/rtl.c
@@ -158,10 +158,13 @@ static size_t rtvec_alloc_sizes;
    Store the length, and initialize all elements to zero.  */
 
 rtvec
-rtvec_alloc (int n)
+rtvec_alloc (size_t n)
 {
   rtvec rt;
 
+  /* rtvec_def.num_elem is an int.  */
+  gcc_assert (n < INT_MAX);
+
   rt = ggc_alloc_rtvec_sized (n);
   /* Clear out the vector.  */
   memset (&rt->elem[0], 0, n * sizeof (rtx));
diff --git a/gcc/rtl.h b/gcc/rtl.h
index c5f3d20fae4..35178b5bfac 100644
--- a/gcc/rtl.h
+++ b/gcc/rtl.h
@@ -2976,7 +2976,7 @@ extern rtx rtx_alloc_stat_v (RTX_CODE MEM_STAT_DECL, int);
               (sizeof (struct hwivec_def)                      \
                + ((NWORDS)-1) * sizeof (HOST_WIDE_INT)))       \
 
-extern rtvec rtvec_alloc (int);
+extern rtvec rtvec_alloc (size_t);
 extern rtvec shallow_copy_rtvec (rtvec);
 extern bool shared_const_p (const_rtx);
 extern rtx copy_rtx (rtx);
-- 
2.26.2

Reply via email to