#include "hyperv_vmbus.h"
 
-#define NUM_PAGES_SPANNED(addr, len) \
-((PAGE_ALIGN(addr + len) >> PAGE_SHIFT) - (addr >> PAGE_SHIFT))
-
 static unsigned long virt_to_hvpfn(void *addr)
 {
        phys_addr_t paddr;
        else
                paddr = __pa(addr);
 
-       return  paddr >> PAGE_SHIFT;
+       return  paddr >> HV_HYP_PAGE_SHIFT;
 }
 
 /*
 
        int pfnsum, pfncount, pfnleft, pfncurr, pfnsize;
 
-       pagecount = size >> PAGE_SHIFT;
+       pagecount = size >> HV_HYP_PAGE_SHIFT;
 
        /* do we need a gpadl body msg */
        pfnsize = MAX_SIZE_CHANNEL_MESSAGE -
                gpadl_header->range[0].byte_count = size;
                for (i = 0; i < pfncount; i++)
                        gpadl_header->range[0].pfn_array[i] = virt_to_hvpfn(
-                               kbuffer + PAGE_SIZE * i);
+                               kbuffer + HV_HYP_PAGE_SIZE * i);
                *msginfo = msgheader;
 
                pfnsum = pfncount;
                         */
                        for (i = 0; i < pfncurr; i++)
                                gpadl_body->pfn[i] = virt_to_hvpfn(
-                                       kbuffer + PAGE_SIZE * (pfnsum + i));
+                                       kbuffer + HV_HYP_PAGE_SIZE * (pfnsum + i));
 
                        /* add to msg header */
                        list_add_tail(&msgbody->msglistentry,
                gpadl_header->range[0].byte_count = size;
                for (i = 0; i < pagecount; i++)
                        gpadl_header->range[0].pfn_array[i] = virt_to_hvpfn(
-                               kbuffer + PAGE_SIZE * i);
+                               kbuffer + HV_HYP_PAGE_SIZE * i);
 
                *msginfo = msgheader;
        }