| /* |
| * linux/mm/vmalloc.c |
| * |
| * Copyright (C) 1993 Linus Torvalds |
| * Support of BIGMEM added by Gerhard Wichert, Siemens AG, July 1999 |
| * SMP-safe vmalloc/vfree/ioremap, Tigran Aivazian <tigran@veritas.com>, May 2000 |
| */ |
| |
| #include <linux/config.h> |
| #include <linux/slab.h> |
| #include <linux/vmalloc.h> |
| #include <linux/spinlock.h> |
| #include <linux/highmem.h> |
| #include <linux/smp_lock.h> |
| |
| #include <asm/uaccess.h> |
| #include <asm/pgalloc.h> |
| #include <asm/tlbflush.h> |
| |
| rwlock_t vmlist_lock = RW_LOCK_UNLOCKED; |
| struct vm_struct * vmlist; |
| |
| static inline void free_area_pte(pmd_t * pmd, unsigned long address, unsigned long size) |
| { |
| pte_t * pte; |
| unsigned long end; |
| |
| if (pmd_none(*pmd)) |
| return; |
| if (pmd_bad(*pmd)) { |
| pmd_ERROR(*pmd); |
| pmd_clear(pmd); |
| return; |
| } |
| pte = pte_offset_kernel(pmd, address); |
| address &= ~PMD_MASK; |
| end = address + size; |
| if (end > PMD_SIZE) |
| end = PMD_SIZE; |
| do { |
| pte_t page; |
| page = ptep_get_and_clear(pte); |
| address += PAGE_SIZE; |
| pte++; |
| if (pte_none(page)) |
| continue; |
| if (pte_present(page)) { |
| struct page *ptpage; |
| unsigned long pfn = pte_pfn(page); |
| if (!pfn_valid(pfn)) |
| continue; |
| ptpage = pfn_to_page(pfn); |
| if (!PageReserved(ptpage)) |
| __free_page(ptpage); |
| continue; |
| } |
| printk(KERN_CRIT "Whee.. Swapped out page in kernel page table\n"); |
| } while (address < end); |
| } |
| |
| static inline void free_area_pmd(pgd_t * dir, unsigned long address, unsigned long size) |
| { |
| pmd_t * pmd; |
| unsigned long end; |
| |
| if (pgd_none(*dir)) |
| return; |
| if (pgd_bad(*dir)) { |
| pgd_ERROR(*dir); |
| pgd_clear(dir); |
| return; |
| } |
| pmd = pmd_offset(dir, address); |
| address &= ~PGDIR_MASK; |
| end = address + size; |
| if (end > PGDIR_SIZE) |
| end = PGDIR_SIZE; |
| do { |
| free_area_pte(pmd, address, end - address); |
| address = (address + PMD_SIZE) & PMD_MASK; |
| pmd++; |
| } while (address < end); |
| } |
| |
| void vmfree_area_pages(unsigned long start, unsigned long size) |
| { |
| pgd_t * dir; |
| unsigned long address = start; |
| unsigned long end = start + size; |
| |
| dir = pgd_offset_k(address); |
| flush_cache_all(); |
| do { |
| free_area_pmd(dir, address, end - address); |
| address = (address + PGDIR_SIZE) & PGDIR_MASK; |
| dir++; |
| } while (address && (address < end)); |
| flush_tlb_kernel_range(start, end); |
| } |
| |
| static inline int alloc_area_pte (pte_t * pte, unsigned long address, |
| unsigned long size, int gfp_mask, pgprot_t prot) |
| { |
| unsigned long end; |
| |
| address &= ~PMD_MASK; |
| end = address + size; |
| if (end > PMD_SIZE) |
| end = PMD_SIZE; |
| do { |
| struct page * page; |
| spin_unlock(&init_mm.page_table_lock); |
| page = alloc_page(gfp_mask); |
| spin_lock(&init_mm.page_table_lock); |
| if (!pte_none(*pte)) |
| printk(KERN_ERR "alloc_area_pte: page already exists\n"); |
| if (!page) |
| return -ENOMEM; |
| set_pte(pte, mk_pte(page, prot)); |
| address += PAGE_SIZE; |
| pte++; |
| } while (address < end); |
| return 0; |
| } |
| |
| static inline int alloc_area_pmd(pmd_t * pmd, unsigned long address, unsigned long size, int gfp_mask, pgprot_t prot) |
| { |
| unsigned long end; |
| |
| address &= ~PGDIR_MASK; |
| end = address + size; |
| if (end > PGDIR_SIZE) |
| end = PGDIR_SIZE; |
| do { |
| pte_t * pte = pte_alloc_kernel(&init_mm, pmd, address); |
| if (!pte) |
| return -ENOMEM; |
| if (alloc_area_pte(pte, address, end - address, gfp_mask, prot)) |
| return -ENOMEM; |
| address = (address + PMD_SIZE) & PMD_MASK; |
| pmd++; |
| } while (address < end); |
| return 0; |
| } |
| |
| inline int vmalloc_area_pages (unsigned long address, unsigned long size, |
| int gfp_mask, pgprot_t prot) |
| { |
| pgd_t * dir; |
| unsigned long end = address + size; |
| int ret; |
| |
| dir = pgd_offset_k(address); |
| spin_lock(&init_mm.page_table_lock); |
| do { |
| pmd_t *pmd; |
| |
| pmd = pmd_alloc(&init_mm, dir, address); |
| ret = -ENOMEM; |
| if (!pmd) |
| break; |
| |
| ret = -ENOMEM; |
| if (alloc_area_pmd(pmd, address, end - address, gfp_mask, prot)) |
| break; |
| |
| address = (address + PGDIR_SIZE) & PGDIR_MASK; |
| dir++; |
| |
| ret = 0; |
| } while (address && (address < end)); |
| spin_unlock(&init_mm.page_table_lock); |
| flush_cache_all(); |
| return ret; |
| } |
| |
| struct vm_struct * get_vm_area(unsigned long size, unsigned long flags) |
| { |
| unsigned long addr; |
| struct vm_struct **p, *tmp, *area; |
| |
| area = (struct vm_struct *) kmalloc(sizeof(*area), GFP_KERNEL); |
| if (!area) |
| return NULL; |
| size += PAGE_SIZE; |
| addr = VMALLOC_START; |
| write_lock(&vmlist_lock); |
| for (p = &vmlist; (tmp = *p) ; p = &tmp->next) { |
| if ((size + addr) < addr) |
| goto out; |
| if (size + addr <= (unsigned long) tmp->addr) |
| break; |
| addr = tmp->size + (unsigned long) tmp->addr; |
| if (addr > VMALLOC_END-size) |
| goto out; |
| } |
| area->flags = flags; |
| area->addr = (void *)addr; |
| area->size = size; |
| area->next = *p; |
| *p = area; |
| write_unlock(&vmlist_lock); |
| return area; |
| |
| out: |
| write_unlock(&vmlist_lock); |
| kfree(area); |
| return NULL; |
| } |
| |
| void vfree(void * addr) |
| { |
| struct vm_struct **p, *tmp; |
| |
| if (!addr) |
| return; |
| if ((PAGE_SIZE-1) & (unsigned long) addr) { |
| printk(KERN_ERR "Trying to vfree() bad address (%p)\n", addr); |
| return; |
| } |
| write_lock(&vmlist_lock); |
| for (p = &vmlist ; (tmp = *p) ; p = &tmp->next) { |
| if (tmp->addr == addr) { |
| *p = tmp->next; |
| vmfree_area_pages(VMALLOC_VMADDR(tmp->addr), tmp->size); |
| write_unlock(&vmlist_lock); |
| kfree(tmp); |
| return; |
| } |
| } |
| write_unlock(&vmlist_lock); |
| printk(KERN_ERR "Trying to vfree() nonexistent vm area (%p)\n", addr); |
| } |
| |
| /* |
| * Allocate any pages |
| */ |
| |
| void * vmalloc (unsigned long size) |
| { |
| return __vmalloc(size, GFP_KERNEL | __GFP_HIGHMEM, PAGE_KERNEL); |
| } |
| |
| /* |
| * Allocate ISA addressable pages for broke crap |
| */ |
| |
| void * vmalloc_dma (unsigned long size) |
| { |
| return __vmalloc(size, GFP_KERNEL|GFP_DMA, PAGE_KERNEL); |
| } |
| |
| /* |
| * vmalloc 32bit PA addressable pages - eg for PCI 32bit devices |
| */ |
| |
| void * vmalloc_32(unsigned long size) |
| { |
| return __vmalloc(size, GFP_KERNEL, PAGE_KERNEL); |
| } |
| |
| void * __vmalloc (unsigned long size, int gfp_mask, pgprot_t prot) |
| { |
| void * addr; |
| struct vm_struct *area; |
| |
| size = PAGE_ALIGN(size); |
| if (!size || (size >> PAGE_SHIFT) > num_physpages) { |
| BUG(); |
| return NULL; |
| } |
| area = get_vm_area(size, VM_ALLOC); |
| if (!area) |
| return NULL; |
| addr = area->addr; |
| if (vmalloc_area_pages(VMALLOC_VMADDR(addr), size, gfp_mask, prot)) { |
| vfree(addr); |
| return NULL; |
| } |
| return addr; |
| } |
| |
| long vread(char *buf, char *addr, unsigned long count) |
| { |
| struct vm_struct *tmp; |
| char *vaddr, *buf_start = buf; |
| unsigned long n; |
| |
| /* Don't allow overflow */ |
| if ((unsigned long) addr + count < count) |
| count = -(unsigned long) addr; |
| |
| read_lock(&vmlist_lock); |
| for (tmp = vmlist; tmp; tmp = tmp->next) { |
| vaddr = (char *) tmp->addr; |
| if (addr >= vaddr + tmp->size - PAGE_SIZE) |
| continue; |
| while (addr < vaddr) { |
| if (count == 0) |
| goto finished; |
| *buf = '\0'; |
| buf++; |
| addr++; |
| count--; |
| } |
| n = vaddr + tmp->size - PAGE_SIZE - addr; |
| do { |
| if (count == 0) |
| goto finished; |
| *buf = *addr; |
| buf++; |
| addr++; |
| count--; |
| } while (--n > 0); |
| } |
| finished: |
| read_unlock(&vmlist_lock); |
| return buf - buf_start; |
| } |
| |
| long vwrite(char *buf, char *addr, unsigned long count) |
| { |
| struct vm_struct *tmp; |
| char *vaddr, *buf_start = buf; |
| unsigned long n; |
| |
| /* Don't allow overflow */ |
| if ((unsigned long) addr + count < count) |
| count = -(unsigned long) addr; |
| |
| read_lock(&vmlist_lock); |
| for (tmp = vmlist; tmp; tmp = tmp->next) { |
| vaddr = (char *) tmp->addr; |
| if (addr >= vaddr + tmp->size - PAGE_SIZE) |
| continue; |
| while (addr < vaddr) { |
| if (count == 0) |
| goto finished; |
| buf++; |
| addr++; |
| count--; |
| } |
| n = vaddr + tmp->size - PAGE_SIZE - addr; |
| do { |
| if (count == 0) |
| goto finished; |
| *addr = *buf; |
| buf++; |
| addr++; |
| count--; |
| } while (--n > 0); |
| } |
| finished: |
| read_unlock(&vmlist_lock); |
| return buf - buf_start; |
| } |