diff mbox series

stdlib: Fix array bounds protection in insertion sort phase of qsort

Message ID 87edgfhypu.fsf@oldenburg.str.redhat.com
State New
Headers show
Series stdlib: Fix array bounds protection in insertion sort phase of qsort | expand

Commit Message

Florian Weimer Nov. 24, 2023, 5:01 p.m. UTC
The previous check did not do anything because tmp_ptr already
points before run_ptr due to the way it is initialized.

Fixes commit e4d8117b82065dc72e8df80097360e7c05a349b9
("stdlib: Avoid another self-comparison in qsort").

Tested on i686-linux-gnu and x86_64-linux-gnu.

Thanks,
Florian
---
 stdlib/Makefile     |  1 +
 stdlib/qsort.c      |  2 +-
 stdlib/tst-qsort6.c | 60 +++++++++++++++++++++++++++++++++++++++++++++++++++++
 3 files changed, 62 insertions(+), 1 deletion(-)


base-commit: 2e0c0ff95ca0e3122eb5b906ee26a31f284ce5ab

Comments

Adhemerval Zanella Netto Nov. 24, 2023, 6:12 p.m. UTC | #1
On 24/11/23 14:01, Florian Weimer wrote:
> The previous check did not do anything because tmp_ptr already
> points before run_ptr due to the way it is initialized.
> 
> Fixes commit e4d8117b82065dc72e8df80097360e7c05a349b9
> ("stdlib: Avoid another self-comparison in qsort").
> 
> Tested on i686-linux-gnu and x86_64-linux-gnu.
> 
> Thanks,
> Florian
> ---
>  stdlib/Makefile     |  1 +
>  stdlib/qsort.c      |  2 +-
>  stdlib/tst-qsort6.c | 60 +++++++++++++++++++++++++++++++++++++++++++++++++++++
>  3 files changed, 62 insertions(+), 1 deletion(-)
> 
> diff --git a/stdlib/Makefile b/stdlib/Makefile
> index 6194d1cb22..0b154e57c5 100644
> --- a/stdlib/Makefile
> +++ b/stdlib/Makefile
> @@ -216,6 +216,7 @@ tests := \
>    tst-qsort2 \
>    tst-qsort3 \
>    tst-qsort5 \
> +  tst-qsort6 \
>    tst-quick_exit \
>    tst-rand48 \
>    tst-rand48-2 \
> diff --git a/stdlib/qsort.c b/stdlib/qsort.c
> index be01fb5598..62477010b6 100644
> --- a/stdlib/qsort.c
> +++ b/stdlib/qsort.c
> @@ -238,7 +238,7 @@ insertion_sort_qsort_partitions (void *const pbase, size_t total_elems,
>    while ((run_ptr += size) <= end_ptr)
>      {
>        tmp_ptr = run_ptr - size;
> -      while (run_ptr != tmp_ptr && cmp (run_ptr, tmp_ptr, arg) < 0)
> +      while (tmp_ptr != base_ptr && cmp (run_ptr, tmp_ptr, arg) < 0)
>          tmp_ptr -= size;
>  
>        tmp_ptr += size;

The insertion sort is quite convoluted and uses an 'optimization' to avoid
some elements swap that does not leverage the swap_type optimization (it
essentially does a byte-copy).

Also, a threshold of 4 maybe it would be better to use network sort (as
gcc does with gcc_sort).

So I wonder if would be better to just use a simpler algorithm:

static inline void
insertion_sort_qsort_partitions (void *const pbase, size_t total_elems,
                                 size_t size, enum swap_type_t swap_type,
                                 __compar_d_fn_t cmp, void *arg)
{
  int i = 1;
  while (i < total_elems)
    {
      int j = i;
      while (j > 0 && cmp (pbase + ((j - 1) * size), pbase + (j * size), arg)
             > 0)
        {
          do_swap (pbase + ((j - 1) * size), pbase + (j * size), size,
                   swap_type);
          j = j -1;
        }
      i = i + 1;
    }
}

(it shown not regression with your testcase included).

> diff --git a/stdlib/tst-qsort6.c b/stdlib/tst-qsort6.c
> new file mode 100644
> index 0000000000..8ec0a6b633
> --- /dev/null
> +++ b/stdlib/tst-qsort6.c
> @@ -0,0 +1,60 @@
> +/* Test qsort with invalid comparison functions.
> +   Copyright (C) 2023 Free Software Foundation, Inc.
> +   This file is part of the GNU C Library.
> +
> +   The GNU C Library is free software; you can redistribute it and/or
> +   modify it under the terms of the GNU Lesser General Public
> +   License as published by the Free Software Foundation; either
> +   version 2.1 of the License, or (at your option) any later version.
> +
> +   The GNU C Library is distributed in the hope that it will be useful,
> +   but WITHOUT ANY WARRANTY; without even the implied warranty of
> +   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
> +   Lesser General Public License for more details.
> +
> +   You should have received a copy of the GNU Lesser General Public
> +   License along with the GNU C Library; if not, see
> +   <http://www.gnu.org/licenses/>.  */
> +
> +#include <array_length.h>
> +#include <stdlib.h>
> +#include <support/check.h>
> +
> +/* Invalid comparison function that always returns -1.  */
> +static int
> +invalid_compare_1 (const void *a1, const void *b1)
> +{
> +  const int *a = a1;
> +  const int *b = b1;
> +  /* Check that the marker value matches, which means that we are
> +     likely within the array.  */
> +  TEST_COMPARE (*a, 842523635);
> +  TEST_COMPARE (*b, 842523635);
> +  TEST_VERIFY_EXIT (*a == 842523635);
> +  TEST_VERIFY_EXIT (*b == 842523635);
> +  return -1;
> +}
> +
> +/* Invalid comparison function that always returns 1.  */
> +static int
> +invalid_compare_2 (const void *a1, const void *b1)
> +{
> +  const int *a = a1;
> +  const int *b = b1;
> +  TEST_COMPARE (*a, 842523635);
> +  TEST_COMPARE (*b, 842523635);
> +  TEST_VERIFY_EXIT (*a == 842523635);
> +  TEST_VERIFY_EXIT (*b == 842523635);
> +  return 1;
> +}
> +
> +static int
> +do_test (void)
> +{
> +  int array[] = {842523635, 842523635, 842523635, 842523635, 842523635};
> +  qsort (array, array_length (array), sizeof (array[0]), invalid_compare_1);
> +  qsort (array, array_length (array), sizeof (array[0]), invalid_compare_2);
> +  return 0;
> +}
> +
> +#include <support/test-driver.c>
> 
> base-commit: 2e0c0ff95ca0e3122eb5b906ee26a31f284ce5ab
>
Florian Weimer Nov. 24, 2023, 8:01 p.m. UTC | #2
* Adhemerval Zanella Netto:

> So I wonder if would be better to just use a simpler algorithm:
>
> static inline void
> insertion_sort_qsort_partitions (void *const pbase, size_t total_elems,
>                                  size_t size, enum swap_type_t swap_type,
>                                  __compar_d_fn_t cmp, void *arg)
> {
>   int i = 1;
>   while (i < total_elems)
>     {
>       int j = i;
>       while (j > 0 && cmp (pbase + ((j - 1) * size), pbase + (j * size), arg)
>              > 0)
>         {
>           do_swap (pbase + ((j - 1) * size), pbase + (j * size), size,
>                    swap_type);
>           j = j -1;
>         }
>       i = i + 1;
>     }
> }
>
> (it shown not regression with your testcase included).

This approach works for me, too.

Thanks,
Florian
Alexander Monakov Nov. 26, 2023, 4:44 p.m. UTC | #3
On Fri, 24 Nov 2023, Adhemerval Zanella Netto wrote:

> Also, a threshold of 4 maybe it would be better to use network sort (as
> gcc does with gcc_sort).

I've pushed a GCC patch fixing my incorrect use of 'network sort' in favor
of the correct term 'sorting networks':
https://inbox.sourceware.org/gcc-patches/20231126163858.9328-1-amonakov@ispras.ru/T/

Sorry about that.

Alexander
Adhemerval Zanella Netto Nov. 29, 2023, 2:12 p.m. UTC | #4
On 24/11/23 14:01, Florian Weimer wrote:
> The previous check did not do anything because tmp_ptr already
> points before run_ptr due to the way it is initialized.
> 
> Fixes commit e4d8117b82065dc72e8df80097360e7c05a349b9
> ("stdlib: Avoid another self-comparison in qsort").
> 
> Tested on i686-linux-gnu and x86_64-linux-gnu.
> 
> Thanks,
> Florian

LGTM, thanks.

Reviewed-by: Adhemerval Zanella  <adhemerval.zanella@linaro.org>

> ---
>  stdlib/Makefile     |  1 +
>  stdlib/qsort.c      |  2 +-
>  stdlib/tst-qsort6.c | 60 +++++++++++++++++++++++++++++++++++++++++++++++++++++
>  3 files changed, 62 insertions(+), 1 deletion(-)
> 
> diff --git a/stdlib/Makefile b/stdlib/Makefile
> index 6194d1cb22..0b154e57c5 100644
> --- a/stdlib/Makefile
> +++ b/stdlib/Makefile
> @@ -216,6 +216,7 @@ tests := \
>    tst-qsort2 \
>    tst-qsort3 \
>    tst-qsort5 \
> +  tst-qsort6 \
>    tst-quick_exit \
>    tst-rand48 \
>    tst-rand48-2 \
> diff --git a/stdlib/qsort.c b/stdlib/qsort.c
> index be01fb5598..62477010b6 100644
> --- a/stdlib/qsort.c
> +++ b/stdlib/qsort.c
> @@ -238,7 +238,7 @@ insertion_sort_qsort_partitions (void *const pbase, size_t total_elems,
>    while ((run_ptr += size) <= end_ptr)
>      {
>        tmp_ptr = run_ptr - size;
> -      while (run_ptr != tmp_ptr && cmp (run_ptr, tmp_ptr, arg) < 0)
> +      while (tmp_ptr != base_ptr && cmp (run_ptr, tmp_ptr, arg) < 0)
>          tmp_ptr -= size;
>  
>        tmp_ptr += size;
> diff --git a/stdlib/tst-qsort6.c b/stdlib/tst-qsort6.c
> new file mode 100644
> index 0000000000..8ec0a6b633
> --- /dev/null
> +++ b/stdlib/tst-qsort6.c
> @@ -0,0 +1,60 @@
> +/* Test qsort with invalid comparison functions.
> +   Copyright (C) 2023 Free Software Foundation, Inc.
> +   This file is part of the GNU C Library.
> +
> +   The GNU C Library is free software; you can redistribute it and/or
> +   modify it under the terms of the GNU Lesser General Public
> +   License as published by the Free Software Foundation; either
> +   version 2.1 of the License, or (at your option) any later version.
> +
> +   The GNU C Library is distributed in the hope that it will be useful,
> +   but WITHOUT ANY WARRANTY; without even the implied warranty of
> +   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
> +   Lesser General Public License for more details.
> +
> +   You should have received a copy of the GNU Lesser General Public
> +   License along with the GNU C Library; if not, see
> +   <http://www.gnu.org/licenses/>.  */
> +
> +#include <array_length.h>
> +#include <stdlib.h>
> +#include <support/check.h>
> +
> +/* Invalid comparison function that always returns -1.  */
> +static int
> +invalid_compare_1 (const void *a1, const void *b1)
> +{
> +  const int *a = a1;
> +  const int *b = b1;
> +  /* Check that the marker value matches, which means that we are
> +     likely within the array.  */
> +  TEST_COMPARE (*a, 842523635);
> +  TEST_COMPARE (*b, 842523635);

These seems superfluous. 

> +  TEST_VERIFY_EXIT (*a == 842523635);
> +  TEST_VERIFY_EXIT (*b == 842523635);> +  return -1;
> +}
> +
> +/* Invalid comparison function that always returns 1.  */
> +static int
> +invalid_compare_2 (const void *a1, const void *b1)
> +{
> +  const int *a = a1;
> +  const int *b = b1;
> +  TEST_COMPARE (*a, 842523635);
> +  TEST_COMPARE (*b, 842523635);
> +  TEST_VERIFY_EXIT (*a == 842523635);
> +  TEST_VERIFY_EXIT (*b == 842523635);
> +  return 1;
> +}
> +
> +static int
> +do_test (void)
> +{
> +  int array[] = {842523635, 842523635, 842523635, 842523635, 842523635};
> +  qsort (array, array_length (array), sizeof (array[0]), invalid_compare_1);
> +  qsort (array, array_length (array), sizeof (array[0]), invalid_compare_2);
> +  return 0;
> +}
> +
> +#include <support/test-driver.c>
> 
> base-commit: 2e0c0ff95ca0e3122eb5b906ee26a31f284ce5ab
>
diff mbox series

Patch

diff --git a/stdlib/Makefile b/stdlib/Makefile
index 6194d1cb22..0b154e57c5 100644
--- a/stdlib/Makefile
+++ b/stdlib/Makefile
@@ -216,6 +216,7 @@  tests := \
   tst-qsort2 \
   tst-qsort3 \
   tst-qsort5 \
+  tst-qsort6 \
   tst-quick_exit \
   tst-rand48 \
   tst-rand48-2 \
diff --git a/stdlib/qsort.c b/stdlib/qsort.c
index be01fb5598..62477010b6 100644
--- a/stdlib/qsort.c
+++ b/stdlib/qsort.c
@@ -238,7 +238,7 @@  insertion_sort_qsort_partitions (void *const pbase, size_t total_elems,
   while ((run_ptr += size) <= end_ptr)
     {
       tmp_ptr = run_ptr - size;
-      while (run_ptr != tmp_ptr && cmp (run_ptr, tmp_ptr, arg) < 0)
+      while (tmp_ptr != base_ptr && cmp (run_ptr, tmp_ptr, arg) < 0)
         tmp_ptr -= size;
 
       tmp_ptr += size;
diff --git a/stdlib/tst-qsort6.c b/stdlib/tst-qsort6.c
new file mode 100644
index 0000000000..8ec0a6b633
--- /dev/null
+++ b/stdlib/tst-qsort6.c
@@ -0,0 +1,60 @@ 
+/* Test qsort with invalid comparison functions.
+   Copyright (C) 2023 Free Software Foundation, Inc.
+   This file is part of the GNU C Library.
+
+   The GNU C Library is free software; you can redistribute it and/or
+   modify it under the terms of the GNU Lesser General Public
+   License as published by the Free Software Foundation; either
+   version 2.1 of the License, or (at your option) any later version.
+
+   The GNU C Library is distributed in the hope that it will be useful,
+   but WITHOUT ANY WARRANTY; without even the implied warranty of
+   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+   Lesser General Public License for more details.
+
+   You should have received a copy of the GNU Lesser General Public
+   License along with the GNU C Library; if not, see
+   <http://www.gnu.org/licenses/>.  */
+
+#include <array_length.h>
+#include <stdlib.h>
+#include <support/check.h>
+
+/* Invalid comparison function that always returns -1.  */
+static int
+invalid_compare_1 (const void *a1, const void *b1)
+{
+  const int *a = a1;
+  const int *b = b1;
+  /* Check that the marker value matches, which means that we are
+     likely within the array.  */
+  TEST_COMPARE (*a, 842523635);
+  TEST_COMPARE (*b, 842523635);
+  TEST_VERIFY_EXIT (*a == 842523635);
+  TEST_VERIFY_EXIT (*b == 842523635);
+  return -1;
+}
+
+/* Invalid comparison function that always returns 1.  */
+static int
+invalid_compare_2 (const void *a1, const void *b1)
+{
+  const int *a = a1;
+  const int *b = b1;
+  TEST_COMPARE (*a, 842523635);
+  TEST_COMPARE (*b, 842523635);
+  TEST_VERIFY_EXIT (*a == 842523635);
+  TEST_VERIFY_EXIT (*b == 842523635);
+  return 1;
+}
+
+static int
+do_test (void)
+{
+  int array[] = {842523635, 842523635, 842523635, 842523635, 842523635};
+  qsort (array, array_length (array), sizeof (array[0]), invalid_compare_1);
+  qsort (array, array_length (array), sizeof (array[0]), invalid_compare_2);
+  return 0;
+}
+
+#include <support/test-driver.c>