Use kmap instead of page_address as it's not always in low memory.
Acked-by: Björn Töpel <bjorn.topel@intel.com>
Signed-off-by: Ivan Khoronzhuk <ivan.khoronzhuk@linaro.org>
Signed-off-by: Daniel Borkmann <daniel@iogearbox.net>
 #include <linux/netdevice.h>
 #include <linux/rtnetlink.h>
 #include <linux/idr.h>
+#include <linux/highmem.h>
 
 #include "xdp_umem.h"
 #include "xsk_queue.h"
        umem->zc = false;
 }
 
+static void xdp_umem_unmap_pages(struct xdp_umem *umem)
+{
+       unsigned int i;
+
+       for (i = 0; i < umem->npgs; i++)
+               kunmap(umem->pgs[i]);
+}
+
 static void xdp_umem_unpin_pages(struct xdp_umem *umem)
 {
        unsigned int i;
 
        xsk_reuseq_destroy(umem);
 
+       xdp_umem_unmap_pages(umem);
        xdp_umem_unpin_pages(umem);
 
        kfree(umem->pages);
        }
 
        for (i = 0; i < umem->npgs; i++)
-               umem->pages[i].addr = page_address(umem->pgs[i]);
+               umem->pages[i].addr = kmap(umem->pgs[i]);
 
        return 0;