Sorry for wasting your time.
I send the wrong version of this patchset, this is the older version.
I will send the right one later.
On 2023/3/20 10:47, Wupeng Ma wrote:
> From: Ma Wupeng <mawupeng1@xxxxxxxxxx>
>
> While testing mlock, we have a problem if the len of mlock is ULONG_MAX.
> The return value of mlock is zero. But nothing will be locked since the
> len in do_mlock overflows to zero due to the following code in mlock:
>
> len = PAGE_ALIGN(len + (offset_in_page(start)));
>
> The same problem happens in munlock.
>
> Add new check and return -EINVAL to fix this overflowing scenarios since
> they are absolutely wrong.
>
> Similar logic is used to fix problems with multiple syscalls.
>
> Here is the testcases:
>
> #include <stdio.h>
> #define _GNU_SOURCE
> #include <sys/mman.h>
> #include <fcntl.h>
> #include <errno.h>
> #include <unistd.h>
> #include <string.h>
> #include <limits.h>
> #include <numaif.h>
>
> extern int errno;
> int main(void)
> {
> int fd;
> int ret;
> void *addr;
> int size = getpagesize();
>
> fd = open("/tmp/testfile", O_RDWR | O_CREAT);
> if (fd < 0) {
> printf("open file error! errno: %d\n", errno);
> return -1;
> }
> printf("open success\n");
> addr = mmap(0, size, PROT_READ | PROT_WRITE, MAP_PRIVATE | MAP_ANONYMOUS, -1, 0);
> if (addr == MAP_FAILED) {
> printf("open file error! errno: %d\n", errno);
> close(fd);
> return -1;
> }
> printf("mmap success\n");
> memset(addr, 0, size);
>
> printf("==== mlock ====\n");
> ret = mlock(addr, ULONG_MAX);
> if (ret == -1 && errno == EINVAL)
> printf("mlock test passed\n");
> else
> printf("mlock test failed, ret: %d, errno: %d\n", ret, errno);
>
> printf("==== set_mempolicy_home_node ====\n");
> ret = syscall(450, addr, ULONG_MAX, 0, 0);
> if (ret == -1 && errno == EINVAL)
> printf("set_mempolicy_home_node test passed\n");
> else
> printf("set_mempolicy_home_node test failed, ret: %d, errno: %d\n", ret, errno);
>
> printf("==== mbind ====\n");
> unsigned long nodemask = 1Ul << 0;
> long max_node = 8 * sizeof(nodemask);
> ret = mbind(addr, ULONG_MAX, MPOL_BIND, &nodemask, max_node, MPOL_MF_MOVE_ALL);
> if (ret == -1 && errno == EINVAL)
> printf("mbind test passed\n");
> else
> printf("mbind test failed, ret: %d, errno: %d\n", ret, errno);
>
> printf("==== msync ====\n");
> ret = msync(addr, ULONG_MAX, MS_ASYNC);
> if (ret == -1 && errno == ENOMEM)
> printf("mbind test passed\n");
> else
> printf("mbind test failed, ret: %d, errno: %d\n", ret, errno);
>
> munmap(mmap, size);
>
> return 0;
> }
>
> Changelog since v3[3]:
> - rebase to the latest master
> - add simple testcases
>
> Changelog since v2[2]:
> - modified the way of checking overflows based on Andrew's comments
>
> Changelog since v1[1]:
> - only check overflow rather than access_ok to keep backward-compatibility
>
> [1]: https://lore.kernel.org/lkml/20221228141701.c64add46c4b09aa17f605baf@xxxxxxxxxxxxxxxxxxxx/T/
> [2]: https://lore.kernel.org/linux-mm/20230116115813.2956935-5-mawupeng1@xxxxxxxxxx/T/
> [3]: https://lore.kernel.org/linux-mm/de4149c7-0e6e-2035-3fb8-2f9da9633704@xxxxxxxxxx/T/
>
> Ma Wupeng (4):
> mm/mlock: return EINVAL if len overflows for mlock/munlock
> mm/mempolicy: return EINVAL for if len overflows for
> set_mempolicy_home_node
> mm/mempolicy: return EINVAL if len overflows for mbind
> mm/msync: return ENOMEM if len overflows for msync
>
> mm/mempolicy.c | 6 ++++--
> mm/mlock.c | 8 ++++++++
> mm/msync.c | 3 ++-
> 3 files changed, 14 insertions(+), 3 deletions(-)
>
