2023-02-16 01:36:22

by Mike Kravetz

[permalink] [raw]
Subject: [PATCH] hugetlb: check for undefined shift on 32 bit architectures

Users can specify the hugetlb page size in the mmap, shmget and
memfd_create system calls. This is done by using 6 bits within the
flags argument to encode the base-2 logarithm of the desired page size.
The routine hstate_sizelog() uses the log2 value to find the
corresponding hugetlb hstate structure. Converting the log2 value
(page_size_log) to potential hugetlb page size is the simple statement:

1UL << page_size_log

Because only 6 bits are used for page_size_log, the left shift can not
be greater than 63. This is fine on 64 bit architectures where a long
is 64 bits. However, if a value greater than 31 is passed on a 32 bit
architecture (where long is 32 bits) the shift will result in undefined
behavior. This was generally not an issue as the result of the
undefined shift had to exactly match hugetlb page size to proceed.

Recent improvements in runtime checking have resulted in this undefined
behavior throwing errors such as reported below.

Fix by comparing page_size_log to BITS_PER_LONG before doing shift.

Reported-by: Naresh Kamboju <[email protected]>
Link: https://lore.kernel.org/lkml/CA+G9fYuei_Tr-vN9GS7SfFyU1y9hNysnf=PB7kT0=yv4MiPgVg@mail.gmail.com/
Fixes: 42d7395feb56 ("mm: support more pagesizes for MAP_HUGETLB/SHM_HUGETLB")
Cc: <[email protected]>
Signed-off-by: Mike Kravetz <[email protected]>
---
include/linux/hugetlb.h | 5 ++++-
1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/include/linux/hugetlb.h b/include/linux/hugetlb.h
index df6dd624ccfe..8b45720f9475 100644
--- a/include/linux/hugetlb.h
+++ b/include/linux/hugetlb.h
@@ -781,7 +781,10 @@ static inline struct hstate *hstate_sizelog(int page_size_log)
if (!page_size_log)
return &default_hstate;

- return size_to_hstate(1UL << page_size_log);
+ if (page_size_log < BITS_PER_LONG)
+ return size_to_hstate(1UL << page_size_log);
+
+ return NULL;
}

static inline struct hstate *hstate_vma(struct vm_area_struct *vma)
--
2.39.1



2023-02-16 02:08:59

by Jesper Juhl

[permalink] [raw]
Subject: Re: [PATCH] hugetlb: check for undefined shift on 32 bit architectures

On Wed, 15 Feb 2023, Mike Kravetz wrote:

> Users can specify the hugetlb page size in the mmap, shmget and
> memfd_create system calls. This is done by using 6 bits within the
> flags argument to encode the base-2 logarithm of the desired page size.
> The routine hstate_sizelog() uses the log2 value to find the
> corresponding hugetlb hstate structure. Converting the log2 value
> (page_size_log) to potential hugetlb page size is the simple statement:
>
> 1UL << page_size_log
>
> Because only 6 bits are used for page_size_log, the left shift can not
> be greater than 63. This is fine on 64 bit architectures where a long
> is 64 bits. However, if a value greater than 31 is passed on a 32 bit
> architecture (where long is 32 bits) the shift will result in undefined
> behavior. This was generally not an issue as the result of the
> undefined shift had to exactly match hugetlb page size to proceed.
>
> Recent improvements in runtime checking have resulted in this undefined
> behavior throwing errors such as reported below.
>
> Fix by comparing page_size_log to BITS_PER_LONG before doing shift.
>
> Reported-by: Naresh Kamboju <[email protected]>
> Link: https://lore.kernel.org/lkml/CA+G9fYuei_Tr-vN9GS7SfFyU1y9hNysnf=PB7kT0=yv4MiPgVg@mail.gmail.com/
> Fixes: 42d7395feb56 ("mm: support more pagesizes for MAP_HUGETLB/SHM_HUGETLB")
> Cc: <[email protected]>
> Signed-off-by: Mike Kravetz <[email protected]>

Looks good to me.

Reviewed-by: Jesper Juhl <[email protected]>


2023-02-16 03:02:06

by Muchun Song

[permalink] [raw]
Subject: Re: [PATCH] hugetlb: check for undefined shift on 32 bit architectures



> On Feb 16, 2023, at 09:35, Mike Kravetz <[email protected]> wrote:
>
> Users can specify the hugetlb page size in the mmap, shmget and
> memfd_create system calls. This is done by using 6 bits within the
> flags argument to encode the base-2 logarithm of the desired page size.
> The routine hstate_sizelog() uses the log2 value to find the
> corresponding hugetlb hstate structure. Converting the log2 value
> (page_size_log) to potential hugetlb page size is the simple statement:
>
> 1UL << page_size_log
>
> Because only 6 bits are used for page_size_log, the left shift can not
> be greater than 63. This is fine on 64 bit architectures where a long
> is 64 bits. However, if a value greater than 31 is passed on a 32 bit
> architecture (where long is 32 bits) the shift will result in undefined
> behavior. This was generally not an issue as the result of the
> undefined shift had to exactly match hugetlb page size to proceed.
>
> Recent improvements in runtime checking have resulted in this undefined
> behavior throwing errors such as reported below.
>
> Fix by comparing page_size_log to BITS_PER_LONG before doing shift.
>
> Reported-by: Naresh Kamboju <[email protected]>
> Link: https://lore.kernel.org/lkml/CA+G9fYuei_Tr-vN9GS7SfFyU1y9hNysnf=PB7kT0=yv4MiPgVg@mail.gmail.com/
> Fixes: 42d7395feb56 ("mm: support more pagesizes for MAP_HUGETLB/SHM_HUGETLB")
> Cc: <[email protected]>
> Signed-off-by: Mike Kravetz <[email protected]>

Acked-by: Muchun Song <[email protected]>

Thanks.


2023-02-16 15:35:26

by Naresh Kamboju

[permalink] [raw]
Subject: Re: [PATCH] hugetlb: check for undefined shift on 32 bit architectures

On Thu, 16 Feb 2023 at 07:06, Mike Kravetz <[email protected]> wrote:
>
> Users can specify the hugetlb page size in the mmap, shmget and
> memfd_create system calls. This is done by using 6 bits within the
> flags argument to encode the base-2 logarithm of the desired page size.
> The routine hstate_sizelog() uses the log2 value to find the
> corresponding hugetlb hstate structure. Converting the log2 value
> (page_size_log) to potential hugetlb page size is the simple statement:
>
> 1UL << page_size_log
>
> Because only 6 bits are used for page_size_log, the left shift can not
> be greater than 63. This is fine on 64 bit architectures where a long
> is 64 bits. However, if a value greater than 31 is passed on a 32 bit
> architecture (where long is 32 bits) the shift will result in undefined
> behavior. This was generally not an issue as the result of the
> undefined shift had to exactly match hugetlb page size to proceed.
>
> Recent improvements in runtime checking have resulted in this undefined
> behavior throwing errors such as reported below.
>
> Fix by comparing page_size_log to BITS_PER_LONG before doing shift.

This proposed fix tested and confirmed that reported issues were fixed.

>
> Reported-by: Naresh Kamboju <[email protected]>
> Link: https://lore.kernel.org/lkml/CA+G9fYuei_Tr-vN9GS7SfFyU1y9hNysnf=PB7kT0=yv4MiPgVg@mail.gmail.com/
> Fixes: 42d7395feb56 ("mm: support more pagesizes for MAP_HUGETLB/SHM_HUGETLB")
> Cc: <[email protected]>
> Signed-off-by: Mike Kravetz <[email protected]>

Tested-by: Linux Kernel Functional Testing <[email protected]>
Tested-by: Naresh Kamboju <[email protected]>

> ---
> include/linux/hugetlb.h | 5 ++++-
> 1 file changed, 4 insertions(+), 1 deletion(-)
>
> diff --git a/include/linux/hugetlb.h b/include/linux/hugetlb.h
> index df6dd624ccfe..8b45720f9475 100644
> --- a/include/linux/hugetlb.h
> +++ b/include/linux/hugetlb.h
> @@ -781,7 +781,10 @@ static inline struct hstate *hstate_sizelog(int page_size_log)
> if (!page_size_log)
> return &default_hstate;
>
> - return size_to_hstate(1UL << page_size_log);
> + if (page_size_log < BITS_PER_LONG)
> + return size_to_hstate(1UL << page_size_log);
> +
> + return NULL;
> }
>
> static inline struct hstate *hstate_vma(struct vm_area_struct *vma)
> --
> 2.39.1


--
Linaro LKFT
https://lkft.linaro.org