lib/test_hmm.c: handle src_pfns and dst_pfns allocation failure
authorDuoming Zhou <duoming@zju.edu.cn>
Tue, 12 Mar 2024 00:59:05 +0000 (08:59 +0800)
committerAndrew Morton <akpm@linux-foundation.org>
Fri, 26 Apr 2024 03:55:48 +0000 (20:55 -0700)
commitc2af060d1c18beaec56351cf9c9bcbbc5af341a3
tree6e56ef09b824bbd3bdeffa411e1b6bf15d9d6bc0
parent4196b48ddd382419e51658bf94a25af195ba9450
lib/test_hmm.c: handle src_pfns and dst_pfns allocation failure

The kcalloc() in dmirror_device_evict_chunk() will return null if the
physical memory has run out.  As a result, if src_pfns or dst_pfns is
dereferenced, the null pointer dereference bug will happen.

Moreover, the device is going away.  If the kcalloc() fails, the pages
mapping a chunk could not be evicted.  So add a __GFP_NOFAIL flag in
kcalloc().

Finally, as there is no need to have physically contiguous memory, Switch
kcalloc() to kvcalloc() in order to avoid failing allocations.

Link: https://lkml.kernel.org/r/20240312005905.9939-1-duoming@zju.edu.cn
Fixes: b2ef9f5a5cb3 ("mm/hmm/test: add selftest driver for HMM")
Signed-off-by: Duoming Zhou <duoming@zju.edu.cn>
Cc: Jérôme Glisse <jglisse@redhat.com>
Signed-off-by: Andrew Morton <akpm@linux-foundation.org>
lib/test_hmm.c