/*
  * Copies from a given buffer or iovector to a VMCI Queue.  Uses
- * kmap()/kunmap() to dynamically map/unmap required portions of the queue
+ * kmap_local_page() to dynamically map required portions of the queue
  * by traversing the offset -> page translation structure for the queue.
  * Assumes that offset + size does not wrap around in the queue.
  */
                size_t to_copy;
 
                if (kernel_if->host)
-                       va = kmap(kernel_if->u.h.page[page_index]);
+                       va = kmap_local_page(kernel_if->u.h.page[page_index]);
                else
                        va = kernel_if->u.g.vas[page_index + 1];
                        /* Skip header. */
                if (!copy_from_iter_full((u8 *)va + page_offset, to_copy,
                                         from)) {
                        if (kernel_if->host)
-                               kunmap(kernel_if->u.h.page[page_index]);
+                               kunmap_local(va);
                        return VMCI_ERROR_INVALID_ARGS;
                }
                bytes_copied += to_copy;
                if (kernel_if->host)
-                       kunmap(kernel_if->u.h.page[page_index]);
+                       kunmap_local(va);
        }
 
        return VMCI_SUCCESS;
 
 /*
  * Copies to a given buffer or iovector from a VMCI Queue.  Uses
- * kmap()/kunmap() to dynamically map/unmap required portions of the queue
+ * kmap_local_page() to dynamically map required portions of the queue
  * by traversing the offset -> page translation structure for the queue.
  * Assumes that offset + size does not wrap around in the queue.
  */
                int err;
 
                if (kernel_if->host)
-                       va = kmap(kernel_if->u.h.page[page_index]);
+                       va = kmap_local_page(kernel_if->u.h.page[page_index]);
                else
                        va = kernel_if->u.g.vas[page_index + 1];
                        /* Skip header. */
                err = copy_to_iter((u8 *)va + page_offset, to_copy, to);
                if (err != to_copy) {
                        if (kernel_if->host)
-                               kunmap(kernel_if->u.h.page[page_index]);
+                               kunmap_local(va);
                        return VMCI_ERROR_INVALID_ARGS;
                }
                bytes_copied += to_copy;
                if (kernel_if->host)
-                       kunmap(kernel_if->u.h.page[page_index]);
+                       kunmap_local(va);
        }
 
        return VMCI_SUCCESS;