forked from Minki/linux
IB/hfi1: Add evict operation to the mmu rb handler
Allow users to clear nodes from the rb tree based on their evict callback. Reviewed-by: Ira Weiny <ira.weiny@intel.com> Signed-off-by: Dean Luick <dean.luick@intel.com> Signed-off-by: Doug Ledford <dledford@redhat.com>
This commit is contained in:
parent
622c202c4a
commit
1034599805
@ -206,6 +206,40 @@ struct mmu_rb_node *hfi1_mmu_rb_extract(struct mmu_rb_handler *handler,
|
||||
return node;
|
||||
}
|
||||
|
||||
void hfi1_mmu_rb_evict(struct mmu_rb_handler *handler, void *evict_arg)
|
||||
{
|
||||
struct mmu_rb_node *rbnode;
|
||||
struct rb_node *node, *next;
|
||||
struct list_head del_list;
|
||||
unsigned long flags;
|
||||
bool stop = false;
|
||||
|
||||
INIT_LIST_HEAD(&del_list);
|
||||
|
||||
spin_lock_irqsave(&handler->lock, flags);
|
||||
for (node = rb_first(&handler->root); node; node = next) {
|
||||
next = rb_next(node);
|
||||
rbnode = rb_entry(node, struct mmu_rb_node, node);
|
||||
if (handler->ops->evict(handler->ops_arg, rbnode, evict_arg,
|
||||
&stop)) {
|
||||
__mmu_int_rb_remove(rbnode, &handler->root);
|
||||
list_add(&rbnode->list, &del_list);
|
||||
}
|
||||
if (stop)
|
||||
break;
|
||||
}
|
||||
spin_unlock_irqrestore(&handler->lock, flags);
|
||||
|
||||
down_write(&handler->mm->mmap_sem);
|
||||
while (!list_empty(&del_list)) {
|
||||
rbnode = list_first_entry(&del_list, struct mmu_rb_node, list);
|
||||
list_del(&rbnode->list);
|
||||
handler->ops->remove(handler->ops_arg, rbnode,
|
||||
handler->mm);
|
||||
}
|
||||
up_write(&handler->mm->mmap_sem);
|
||||
}
|
||||
|
||||
void hfi1_mmu_rb_remove(struct mmu_rb_handler *handler,
|
||||
struct mmu_rb_node *node)
|
||||
{
|
||||
|
@ -54,6 +54,7 @@ struct mmu_rb_node {
|
||||
unsigned long len;
|
||||
unsigned long __last;
|
||||
struct rb_node node;
|
||||
struct list_head list;
|
||||
};
|
||||
|
||||
struct mmu_rb_ops {
|
||||
@ -63,6 +64,8 @@ struct mmu_rb_ops {
|
||||
void (*remove)(void *ops_arg, struct mmu_rb_node *mnode,
|
||||
struct mm_struct *mm);
|
||||
int (*invalidate)(void *ops_arg, struct mmu_rb_node *node);
|
||||
int (*evict)(void *ops_arg, struct mmu_rb_node *mnode,
|
||||
void *evict_arg, bool *stop);
|
||||
};
|
||||
|
||||
int hfi1_mmu_rb_register(void *ops_arg, struct mm_struct *mm,
|
||||
@ -71,6 +74,7 @@ int hfi1_mmu_rb_register(void *ops_arg, struct mm_struct *mm,
|
||||
void hfi1_mmu_rb_unregister(struct mmu_rb_handler *handler);
|
||||
int hfi1_mmu_rb_insert(struct mmu_rb_handler *handler,
|
||||
struct mmu_rb_node *mnode);
|
||||
void hfi1_mmu_rb_evict(struct mmu_rb_handler *handler, void *evict_arg);
|
||||
void hfi1_mmu_rb_remove(struct mmu_rb_handler *handler,
|
||||
struct mmu_rb_node *mnode);
|
||||
struct mmu_rb_node *hfi1_mmu_rb_extract(struct mmu_rb_handler *handler,
|
||||
|
Loading…
Reference in New Issue
Block a user