#include <asm/pgtable.h>
  #include <asm/tlbflush.h>
  
+ struct page *dmw_virt_to_page(unsigned long kaddr)
+ {
+       return pfn_to_page(virt_to_pfn(kaddr));
+ }
+ EXPORT_SYMBOL_GPL(dmw_virt_to_page);
+ 
+ struct page *tlb_virt_to_page(unsigned long kaddr)
+ {
+       return pfn_to_page(pte_pfn(*virt_to_kpte(kaddr)));
+ }
+ EXPORT_SYMBOL_GPL(tlb_virt_to_page);
+ 
  pgd_t *pgd_alloc(struct mm_struct *mm)
  {
 -      pgd_t *ret, *init;
 +      pgd_t *init, *ret = NULL;
 +      struct ptdesc *ptdesc = pagetable_alloc(GFP_KERNEL & ~__GFP_HIGHMEM, 0);
  
 -      ret = (pgd_t *) __get_free_page(GFP_KERNEL);
 -      if (ret) {
 +      if (ptdesc) {
 +              ret = (pgd_t *)ptdesc_address(ptdesc);
                init = pgd_offset(&init_mm, 0UL);
                pgd_init(ret);
                memcpy(ret + USER_PTRS_PER_PGD, init + USER_PTRS_PER_PGD,