1 files changed, 14 insertions, 9 deletions

diff --git a/arch/um/kernel/skas/tlb.c b/arch/um/kernel/skas/tlb.c
index 6230999c672c..4b5fd2049547 100644
--- a/arch/um/kernel/skas/tlb.c
+++ b/arch/um/kernel/skas/tlb.c
@@ -18,7 +18,8 @@
 #include "os.h"
 #include "tlb.h"
 
-static void do_ops(union mm_context *mmu, struct host_vm_op *ops, int last)
+static void *do_ops(union mm_context *mmu, struct host_vm_op *ops, int last,
+                    int finished, void *flush)
 {
         struct host_vm_op *op;
         int i;
@@ -27,24 +28,28 @@ static void do_ops(union mm_context *mmu, struct host_vm_op *ops, int last)
                 op = &ops[i];
                 switch(op->type){
                 case MMAP:
-                        map(&mmu->skas.id, op->u.mmap.addr, op->u.mmap.len,
-                            op->u.mmap.r, op->u.mmap.w, op->u.mmap.x,
-                            op->u.mmap.fd, op->u.mmap.offset);
+                        flush = map(&mmu->skas.id, op->u.mmap.addr,
+                                    op->u.mmap.len, op->u.mmap.r, op->u.mmap.w,
+                                    op->u.mmap.x, op->u.mmap.fd,
+                                    op->u.mmap.offset, finished, flush);
                         break;
                 case MUNMAP:
-                        unmap(&mmu->skas.id, (void *) op->u.munmap.addr,
-                              op->u.munmap.len);
+                        flush = unmap(&mmu->skas.id, (void *) op->u.munmap.addr,
+                                      op->u.munmap.len, finished, flush);
                         break;
                 case MPROTECT:
-                        protect(&mmu->skas.id, op->u.mprotect.addr,
-                                op->u.mprotect.len, op->u.mprotect.r,
-                                op->u.mprotect.w, op->u.mprotect.x);
+                        flush = protect(&mmu->skas.id, op->u.mprotect.addr,
+                                        op->u.mprotect.len, op->u.mprotect.r,
+                                        op->u.mprotect.w, op->u.mprotect.x,
+                                        finished, flush);
                         break;
                 default:
                         printk("Unknown op type %d in do_ops\n", op->type);
                         break;
                 }
         }
+
+        return flush;
 }
 
 extern int proc_mm;