2 files changed, 54 insertions, 4 deletions

diff --git a/arch/frv/mm/cache-page.c b/arch/frv/mm/cache-page.c
index 0261cbe153b5..b24ade27a0f0 100644
--- a/arch/frv/mm/cache-page.c
+++ b/arch/frv/mm/cache-page.c
@@ -26,11 +26,11 @@ void flush_dcache_page(struct page *page)
 
         dampr2 = __get_DAMPR(2);
 
-        vaddr = kmap_atomic(page, __KM_CACHE);
+        vaddr = kmap_atomic_primary(page, __KM_CACHE);
 
         frv_dcache_writeback((unsigned long) vaddr, (unsigned long) vaddr + PAGE_SIZE);
 
-        kunmap_atomic(vaddr, __KM_CACHE);
+        kunmap_atomic_primary(vaddr, __KM_CACHE);
 
         if (dampr2) {
                 __set_DAMPR(2, dampr2);
@@ -54,12 +54,12 @@ void flush_icache_user_range(struct vm_area_struct *vma, struct page *page,
 
         dampr2 = __get_DAMPR(2);
 
-        vaddr = kmap_atomic(page, __KM_CACHE);
+        vaddr = kmap_atomic_primary(page, __KM_CACHE);
 
         start = (start & ~PAGE_MASK) | (unsigned long) vaddr;
         frv_cache_wback_inv(start, start + len);
 
-        kunmap_atomic(vaddr, __KM_CACHE);
+        kunmap_atomic_primary(vaddr, __KM_CACHE);
 
         if (dampr2) {
                 __set_DAMPR(2, dampr2);
diff --git a/arch/frv/mm/highmem.c b/arch/frv/mm/highmem.c
index eadd07658075..61088dcc1594 100644
--- a/arch/frv/mm/highmem.c
+++ b/arch/frv/mm/highmem.c
@@ -36,3 +36,53 @@ struct page *kmap_atomic_to_page(void *ptr)
 {
         return virt_to_page(ptr);
 }
+
+void *__kmap_atomic(struct page *page)
+{
+        unsigned long paddr;
+        int type;
+
+        pagefault_disable();
+        type = kmap_atomic_idx_push();
+        paddr = page_to_phys(page);
+
+        switch (type) {
+        /*
+         * The first 4 primary maps are reserved for architecture code
+         */
+        case 0:         return __kmap_atomic_primary(4, paddr, 6);
+        case 1:         return __kmap_atomic_primary(5, paddr, 7);
+        case 2:         return __kmap_atomic_primary(6, paddr, 8);
+        case 3:         return __kmap_atomic_primary(7, paddr, 9);
+        case 4:         return __kmap_atomic_primary(8, paddr, 10);
+
+        case 5 ... 5 + NR_TLB_LINES - 1:
+                return __kmap_atomic_secondary(type - 5, paddr);
+
+        default:
+                BUG();
+                return NULL;
+        }
+}
+EXPORT_SYMBOL(__kmap_atomic);
+
+void __kunmap_atomic(void *kvaddr)
+{
+        int type = kmap_atomic_idx_pop();
+        switch (type) {
+        case 0:         __kunmap_atomic_primary(4, 6);  break;
+        case 1:         __kunmap_atomic_primary(5, 7);  break;
+        case 2:         __kunmap_atomic_primary(6, 8);  break;
+        case 3:         __kunmap_atomic_primary(7, 9);  break;
+        case 4:         __kunmap_atomic_primary(8, 10); break;
+
+        case 5 ... 5 + NR_TLB_LINES - 1:
+                __kunmap_atomic_secondary(type - 5, kvaddr);
+                break;
+
+        default:
+                BUG();
+        }
+        pagefault_enable();
+}
+EXPORT_SYMBOL(__kunmap_atomic);