#include <linux/crash_dump.h>
 #include <linux/uaccess.h>
 #include <linux/io.h>
+#include <linux/uio.h>
 
-/**
- * copy_oldmem_page() - copy one page from old kernel memory
- * @pfn: page frame number to be copied
- * @buf: buffer where the copied page is placed
- * @csize: number of bytes to copy
- * @offset: offset in bytes into the page
- * @userbuf: if set, @buf is int he user address space
- *
- * This function copies one page from old kernel memory into buffer pointed by
- * @buf. If @buf is in userspace, set @userbuf to %1. Returns number of bytes
- * copied or negative error in case of failure.
- */
-ssize_t copy_oldmem_page(unsigned long pfn, char *buf,
-                        size_t csize, unsigned long offset,
-                        int userbuf)
+ssize_t copy_oldmem_page(struct iov_iter *iter, unsigned long pfn,
+                        size_t csize, unsigned long offset)
 {
        void *vaddr;
 
        if (!vaddr)
                return -ENOMEM;
 
-       if (userbuf) {
-               if (copy_to_user(buf, vaddr + offset, csize)) {
-                       iounmap(vaddr);
-                       return -EFAULT;
-               }
-       } else {
-               memcpy(buf, vaddr + offset, csize);
-       }
+       csize = copy_to_iter(vaddr + offset, csize, iter);
 
        iounmap(vaddr);
        return csize;
 
 #include <linux/crash_dump.h>
 #include <linux/errno.h>
 #include <linux/io.h>
-#include <linux/memblock.h>
-#include <linux/uaccess.h>
+#include <linux/uio.h>
 #include <asm/memory.h>
 
-/**
- * copy_oldmem_page() - copy one page from old kernel memory
- * @pfn: page frame number to be copied
- * @buf: buffer where the copied page is placed
- * @csize: number of bytes to copy
- * @offset: offset in bytes into the page
- * @userbuf: if set, @buf is in a user address space
- *
- * This function copies one page from old kernel memory into buffer pointed by
- * @buf. If @buf is in userspace, set @userbuf to %1. Returns number of bytes
- * copied or negative error in case of failure.
- */
-ssize_t copy_oldmem_page(unsigned long pfn, char *buf,
-                        size_t csize, unsigned long offset,
-                        int userbuf)
+ssize_t copy_oldmem_page(struct iov_iter *iter, unsigned long pfn,
+                        size_t csize, unsigned long offset)
 {
        void *vaddr;
 
        if (!vaddr)
                return -ENOMEM;
 
-       if (userbuf) {
-               if (copy_to_user((char __user *)buf, vaddr + offset, csize)) {
-                       memunmap(vaddr);
-                       return -EFAULT;
-               }
-       } else {
-               memcpy(buf, vaddr + offset, csize);
-       }
+       csize = copy_to_iter(vaddr + offset, csize, iter);
 
        memunmap(vaddr);
 
 
 #include <linux/errno.h>
 #include <linux/types.h>
 #include <linux/crash_dump.h>
-
+#include <linux/uio.h>
 #include <asm/page.h>
-#include <linux/uaccess.h>
 
-/**
- * copy_oldmem_page - copy one page from "oldmem"
- * @pfn: page frame number to be copied
- * @buf: target memory address for the copy; this can be in kernel address
- *     space or user address space (see @userbuf)
- * @csize: number of bytes to copy
- * @offset: offset in bytes into the page (based on pfn) to begin the copy
- * @userbuf: if set, @buf is in user address space, use copy_to_user(),
- *     otherwise @buf is in kernel address space, use memcpy().
- *
- * Copy a page from "oldmem". For this page, there is no pte mapped
- * in the current kernel. We stitch up a pte, similar to kmap_atomic.
- *
- * Calling copy_to_user() in atomic context is not desirable. Hence first
- * copying the data to a pre-allocated kernel page and then copying to user
- * space in non-atomic context.
- */
-ssize_t
-copy_oldmem_page(unsigned long pfn, char *buf,
-               size_t csize, unsigned long offset, int userbuf)
+ssize_t copy_oldmem_page(struct iov_iter *iter, unsigned long pfn,
+               size_t csize, unsigned long offset)
 {
        void  *vaddr;
 
        if (!csize)
                return 0;
        vaddr = __va(pfn<<PAGE_SHIFT);
-       if (userbuf) {
-               if (copy_to_user(buf, (vaddr + offset), csize)) {
-                       return -EFAULT;
-               }
-       } else
-               memcpy(buf, (vaddr + offset), csize);
+       csize = copy_to_iter(vaddr + offset, csize, iter);
        return csize;
 }
 
 
 // SPDX-License-Identifier: GPL-2.0
 #include <linux/highmem.h>
 #include <linux/crash_dump.h>
+#include <linux/uio.h>
 
-/**
- * copy_oldmem_page - copy one page from "oldmem"
- * @pfn: page frame number to be copied
- * @buf: target memory address for the copy; this can be in kernel address
- *     space or user address space (see @userbuf)
- * @csize: number of bytes to copy
- * @offset: offset in bytes into the page (based on pfn) to begin the copy
- * @userbuf: if set, @buf is in user address space, use copy_to_user(),
- *     otherwise @buf is in kernel address space, use memcpy().
- *
- * Copy a page from "oldmem". For this page, there is no pte mapped
- * in the current kernel.
- */
-ssize_t copy_oldmem_page(unsigned long pfn, char *buf,
-                        size_t csize, unsigned long offset, int userbuf)
+ssize_t copy_oldmem_page(struct iov_iter *iter, unsigned long pfn,
+                        size_t csize, unsigned long offset)
 {
        void  *vaddr;
 
                return 0;
 
        vaddr = kmap_local_pfn(pfn);
-
-       if (!userbuf) {
-               memcpy(buf, vaddr + offset, csize);
-       } else {
-               if (copy_to_user(buf, vaddr + offset, csize))
-                       csize = -EFAULT;
-       }
-
+       csize = copy_to_iter(vaddr + offset, csize, iter);
        kunmap_local(vaddr);
 
        return csize;
 
 #include <asm/kdump.h>
 #include <asm/prom.h>
 #include <asm/firmware.h>
-#include <linux/uaccess.h>
+#include <linux/uio.h>
 #include <asm/rtas.h>
 #include <asm/inst.h>
 
 }
 #endif /* CONFIG_NONSTATIC_KERNEL */
 
-static size_t copy_oldmem_vaddr(void *vaddr, char *buf, size_t csize,
-                               unsigned long offset, int userbuf)
-{
-       if (userbuf) {
-               if (copy_to_user((char __user *)buf, (vaddr + offset), csize))
-                       return -EFAULT;
-       } else
-               memcpy(buf, (vaddr + offset), csize);
-
-       return csize;
-}
-
-/**
- * copy_oldmem_page - copy one page from "oldmem"
- * @pfn: page frame number to be copied
- * @buf: target memory address for the copy; this can be in kernel address
- *      space or user address space (see @userbuf)
- * @csize: number of bytes to copy
- * @offset: offset in bytes into the page (based on pfn) to begin the copy
- * @userbuf: if set, @buf is in user address space, use copy_to_user(),
- *      otherwise @buf is in kernel address space, use memcpy().
- *
- * Copy a page from "oldmem". For this page, there is no pte mapped
- * in the current kernel. We stitch up a pte, similar to kmap_atomic.
- */
-ssize_t copy_oldmem_page(unsigned long pfn, char *buf,
-                       size_t csize, unsigned long offset, int userbuf)
+ssize_t copy_oldmem_page(struct iov_iter *iter, unsigned long pfn,
+                       size_t csize, unsigned long offset)
 {
        void  *vaddr;
        phys_addr_t paddr;
 
        if (memblock_is_region_memory(paddr, csize)) {
                vaddr = __va(paddr);
-               csize = copy_oldmem_vaddr(vaddr, buf, csize, offset, userbuf);
+               csize = copy_to_iter(vaddr + offset, csize, iter);
        } else {
                vaddr = ioremap_cache(paddr, PAGE_SIZE);
-               csize = copy_oldmem_vaddr(vaddr, buf, csize, offset, userbuf);
+               csize = copy_to_iter(vaddr + offset, csize, iter);
                iounmap(vaddr);
        }
 
 
 
 #include <linux/crash_dump.h>
 #include <linux/io.h>
+#include <linux/uio.h>
 
-/**
- * copy_oldmem_page() - copy one page from old kernel memory
- * @pfn: page frame number to be copied
- * @buf: buffer where the copied page is placed
- * @csize: number of bytes to copy
- * @offset: offset in bytes into the page
- * @userbuf: if set, @buf is in a user address space
- *
- * This function copies one page from old kernel memory into buffer pointed by
- * @buf. If @buf is in userspace, set @userbuf to %1. Returns number of bytes
- * copied or negative error in case of failure.
- */
-ssize_t copy_oldmem_page(unsigned long pfn, char *buf,
-                        size_t csize, unsigned long offset,
-                        int userbuf)
+ssize_t copy_oldmem_page(struct iov_iter *iter, unsigned long pfn,
+                        size_t csize, unsigned long offset)
 {
        void *vaddr;
 
        if (!vaddr)
                return -ENOMEM;
 
-       if (userbuf) {
-               if (copy_to_user((char __user *)buf, vaddr + offset, csize)) {
-                       memunmap(vaddr);
-                       return -EFAULT;
-               }
-       } else
-               memcpy(buf, vaddr + offset, csize);
+       csize = copy_to_iter(vaddr + offset, csize, iter);
 
        memunmap(vaddr);
        return csize;
 
 #include <linux/slab.h>
 #include <linux/memblock.h>
 #include <linux/elf.h>
+#include <linux/uio.h>
 #include <asm/asm-offsets.h>
 #include <asm/os_info.h>
 #include <asm/elf.h>
 /*
  * Copy one page from "oldmem"
  */
-ssize_t copy_oldmem_page(unsigned long pfn, char *buf, size_t csize,
-                        unsigned long offset, int userbuf)
+ssize_t copy_oldmem_page(struct iov_iter *iter, unsigned long pfn, size_t csize,
+                        unsigned long offset)
 {
        unsigned long src;
        int rc;
        if (!csize)
                return 0;
        src = pfn_to_phys(pfn) + offset;
-       if (userbuf)
-               rc = copy_oldmem_user((void __force __user *) buf, src, csize);
+
+       /* XXX: pass the iov_iter down to a common function */
+       if (iter_is_iovec(iter))
+               rc = copy_oldmem_user(iter->iov->iov_base, src, csize);
        else
-               rc = copy_oldmem_kernel((void *) buf, src, csize);
+               rc = copy_oldmem_kernel(iter->kvec->iov_base, src, csize);
        return rc;
 }
 
 
 #include <linux/errno.h>
 #include <linux/crash_dump.h>
 #include <linux/io.h>
+#include <linux/uio.h>
 #include <linux/uaccess.h>
 
-/**
- * copy_oldmem_page - copy one page from "oldmem"
- * @pfn: page frame number to be copied
- * @buf: target memory address for the copy; this can be in kernel address
- *     space or user address space (see @userbuf)
- * @csize: number of bytes to copy
- * @offset: offset in bytes into the page (based on pfn) to begin the copy
- * @userbuf: if set, @buf is in user address space, use copy_to_user(),
- *     otherwise @buf is in kernel address space, use memcpy().
- *
- * Copy a page from "oldmem". For this page, there is no pte mapped
- * in the current kernel. We stitch up a pte, similar to kmap_atomic.
- */
-ssize_t copy_oldmem_page(unsigned long pfn, char *buf,
-                               size_t csize, unsigned long offset, int userbuf)
+ssize_t copy_oldmem_page(struct iov_iter *iter, unsigned long pfn,
+                        size_t csize, unsigned long offset)
 {
        void  __iomem *vaddr;
 
                return 0;
 
        vaddr = ioremap(pfn << PAGE_SHIFT, PAGE_SIZE);
-
-       if (userbuf) {
-               if (copy_to_user((void __user *)buf, (vaddr + offset), csize)) {
-                       iounmap(vaddr);
-                       return -EFAULT;
-               }
-       } else
-       memcpy(buf, (vaddr + offset), csize);
-
+       csize = copy_to_iter(vaddr + offset, csize, iter);
        iounmap(vaddr);
+
        return csize;
 }
 
 #include <linux/errno.h>
 #include <linux/highmem.h>
 #include <linux/crash_dump.h>
-
-#include <linux/uaccess.h>
+#include <linux/uio.h>
 
 static inline bool is_crashed_pfn_valid(unsigned long pfn)
 {
 #endif
 }
 
-/**
- * copy_oldmem_page - copy one page from "oldmem"
- * @pfn: page frame number to be copied
- * @buf: target memory address for the copy; this can be in kernel address
- *     space or user address space (see @userbuf)
- * @csize: number of bytes to copy
- * @offset: offset in bytes into the page (based on pfn) to begin the copy
- * @userbuf: if set, @buf is in user address space, use copy_to_user(),
- *     otherwise @buf is in kernel address space, use memcpy().
- *
- * Copy a page from "oldmem". For this page, there might be no pte mapped
- * in the current kernel.
- */
-ssize_t copy_oldmem_page(unsigned long pfn, char *buf, size_t csize,
-                        unsigned long offset, int userbuf)
+ssize_t copy_oldmem_page(struct iov_iter *iter, unsigned long pfn, size_t csize,
+                        unsigned long offset)
 {
        void  *vaddr;
 
                return -EFAULT;
 
        vaddr = kmap_local_pfn(pfn);
-
-       if (!userbuf) {
-               memcpy(buf, vaddr + offset, csize);
-       } else {
-               if (copy_to_user(buf, vaddr + offset, csize))
-                       csize = -EFAULT;
-       }
-
+       csize = copy_to_iter(vaddr + offset, csize, iter);
        kunmap_local(vaddr);
 
        return csize;
 
 
 #include <linux/errno.h>
 #include <linux/crash_dump.h>
-#include <linux/uaccess.h>
+#include <linux/uio.h>
 #include <linux/io.h>
 #include <linux/cc_platform.h>
 
-static ssize_t __copy_oldmem_page(unsigned long pfn, char *buf, size_t csize,
-                                 unsigned long offset, int userbuf,
+static ssize_t __copy_oldmem_page(struct iov_iter *iter, unsigned long pfn,
+                                 size_t csize, unsigned long offset,
                                  bool encrypted)
 {
        void  *vaddr;
        if (!vaddr)
                return -ENOMEM;
 
-       if (userbuf) {
-               if (copy_to_user((void __user *)buf, vaddr + offset, csize)) {
-                       iounmap((void __iomem *)vaddr);
-                       return -EFAULT;
-               }
-       } else
-               memcpy(buf, vaddr + offset, csize);
+       csize = copy_to_iter(vaddr + offset, csize, iter);
 
        iounmap((void __iomem *)vaddr);
        return csize;
 }
 
-/**
- * copy_oldmem_page - copy one page of memory
- * @pfn: page frame number to be copied
- * @buf: target memory address for the copy; this can be in kernel address
- *     space or user address space (see @userbuf)
- * @csize: number of bytes to copy
- * @offset: offset in bytes into the page (based on pfn) to begin the copy
- * @userbuf: if set, @buf is in user address space, use copy_to_user(),
- *     otherwise @buf is in kernel address space, use memcpy().
- *
- * Copy a page from the old kernel's memory. For this page, there is no pte
- * mapped in the current kernel. We stitch up a pte, similar to kmap_atomic.
- */
-ssize_t copy_oldmem_page(unsigned long pfn, char *buf, size_t csize,
-                        unsigned long offset, int userbuf)
+ssize_t copy_oldmem_page(struct iov_iter *iter, unsigned long pfn, size_t csize,
+                        unsigned long offset)
 {
-       return __copy_oldmem_page(pfn, buf, csize, offset, userbuf, false);
+       return __copy_oldmem_page(iter, pfn, csize, offset, false);
 }
 
-/**
+/*
  * copy_oldmem_page_encrypted - same as copy_oldmem_page() above but ioremap the
  * memory with the encryption mask set to accommodate kdump on SME-enabled
  * machines.
  */
-ssize_t copy_oldmem_page_encrypted(unsigned long pfn, char *buf, size_t csize,
-                                  unsigned long offset, int userbuf)
+ssize_t copy_oldmem_page_encrypted(struct iov_iter *iter, unsigned long pfn,
+                                  size_t csize, unsigned long offset)
 {
-       return __copy_oldmem_page(pfn, buf, csize, offset, userbuf, true);
+       return __copy_oldmem_page(iter, pfn, csize, offset, true);
 }
 
 ssize_t elfcorehdr_read(char *buf, size_t count, u64 *ppos)
 
 #include <linux/vmalloc.h>
 #include <linux/pagemap.h>
 #include <linux/uaccess.h>
+#include <linux/uio.h>
 #include <linux/cc_platform.h>
 #include <asm/io.h>
 #include "internal.h"
 }
 
 /* Reads a page from the oldmem device from given offset. */
-ssize_t read_from_oldmem(char *buf, size_t count,
-                        u64 *ppos, int userbuf,
-                        bool encrypted)
+static ssize_t read_from_oldmem_iter(struct iov_iter *iter, size_t count,
+                        u64 *ppos, bool encrypted)
 {
        unsigned long pfn, offset;
        size_t nr_bytes;
 
                /* If pfn is not ram, return zeros for sparse dump files */
                if (!pfn_is_ram(pfn)) {
-                       tmp = 0;
-                       if (!userbuf)
-                               memset(buf, 0, nr_bytes);
-                       else if (clear_user(buf, nr_bytes))
-                               tmp = -EFAULT;
+                       tmp = iov_iter_zero(nr_bytes, iter);
                } else {
                        if (encrypted)
-                               tmp = copy_oldmem_page_encrypted(pfn, buf,
+                               tmp = copy_oldmem_page_encrypted(iter, pfn,
                                                                 nr_bytes,
-                                                                offset,
-                                                                userbuf);
+                                                                offset);
                        else
-                               tmp = copy_oldmem_page(pfn, buf, nr_bytes,
-                                                      offset, userbuf);
+                               tmp = copy_oldmem_page(iter, pfn, nr_bytes,
+                                                      offset);
                }
-               if (tmp < 0) {
+               if (tmp < nr_bytes) {
                        srcu_read_unlock(&vmcore_cb_srcu, idx);
-                       return tmp;
+                       return -EFAULT;
                }
 
                *ppos += nr_bytes;
                count -= nr_bytes;
-               buf += nr_bytes;
                read += nr_bytes;
                ++pfn;
                offset = 0;
        return read;
 }
 
+ssize_t read_from_oldmem(char *buf, size_t count,
+                        u64 *ppos, int userbuf,
+                        bool encrypted)
+{
+       struct iov_iter iter;
+       struct iovec iov;
+       struct kvec kvec;
+
+       if (userbuf) {
+               iov.iov_base = (__force void __user *)buf;
+               iov.iov_len = count;
+               iov_iter_init(&iter, READ, &iov, 1, count);
+       } else {
+               kvec.iov_base = buf;
+               kvec.iov_len = count;
+               iov_iter_kvec(&iter, READ, &kvec, 1, count);
+       }
+
+       return read_from_oldmem_iter(&iter, count, ppos, encrypted);
+}
+
 /*
  * Architectures may override this function to allocate ELF header in 2nd kernel
  */
 /*
  * Architectures which support memory encryption override this.
  */
-ssize_t __weak
-copy_oldmem_page_encrypted(unsigned long pfn, char *buf, size_t csize,
-                          unsigned long offset, int userbuf)
+ssize_t __weak copy_oldmem_page_encrypted(struct iov_iter *iter,
+               unsigned long pfn, size_t csize, unsigned long offset)
 {
-       return copy_oldmem_page(pfn, buf, csize, offset, userbuf);
+       return copy_oldmem_page(iter, pfn, csize, offset);
 }
 
 /*
 
                                  unsigned long from, unsigned long pfn,
                                  unsigned long size, pgprot_t prot);
 
-extern ssize_t copy_oldmem_page(unsigned long, char *, size_t,
-                                               unsigned long, int);
-extern ssize_t copy_oldmem_page_encrypted(unsigned long pfn, char *buf,
-                                         size_t csize, unsigned long offset,
-                                         int userbuf);
+ssize_t copy_oldmem_page(struct iov_iter *i, unsigned long pfn, size_t csize,
+               unsigned long offset);
+ssize_t copy_oldmem_page_encrypted(struct iov_iter *iter, unsigned long pfn,
+                                  size_t csize, unsigned long offset);
 
 void vmcore_cleanup(void);