Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit 10345998 authored by Dean Luick's avatar Dean Luick Committed by Doug Ledford
Browse files

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: default avatarIra Weiny <ira.weiny@intel.com>
Signed-off-by: default avatarDean Luick <dean.luick@intel.com>
Signed-off-by: default avatarDoug Ledford <dledford@redhat.com>
parent 622c202c

drivers/infiniband/hw/hfi1/mmu_rb.c

+34 −0
Original line number Diff line number Diff line
@@ -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)

{

drivers/infiniband/hw/hfi1/mmu_rb.h

+4 −0
Original line number Diff line number Diff line
@@ -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,