194 struct rds_ib_mr *rds_ib_reuse_mr(struct rds_ib_mr_pool *pool)  in rds_ib_reuse_mr()  argument
200 	spin_lock_irqsave(&pool->clean_lock, flags);  in rds_ib_reuse_mr()
201 	ret = llist_del_first(&pool->clean_list);  in rds_ib_reuse_mr()
202 	spin_unlock_irqrestore(&pool->clean_lock, flags);  in rds_ib_reuse_mr()
205 		if (pool->pool_type == RDS_IB_MR_8K_POOL)  in rds_ib_reuse_mr()
271 		struct rds_ib_mr_pool *pool = ibmr->pool;  in rds_ib_teardown_mr()  local
273 		atomic_sub(pinned, &pool->free_pinned);  in rds_ib_teardown_mr()
277 static inline unsigned int rds_ib_flush_goal(struct rds_ib_mr_pool *pool, int free_all)  in rds_ib_flush_goal()  argument
281 	item_count = atomic_read(&pool->item_count);  in rds_ib_flush_goal()
333  * Flush our pool of MRs.
338 int rds_ib_flush_mr_pool(struct rds_ib_mr_pool *pool,  in rds_ib_flush_mr_pool()  argument
348 	if (pool->pool_type == RDS_IB_MR_8K_POOL)  in rds_ib_flush_mr_pool()
355 		while (!mutex_trylock(&pool->flush_lock)) {  in rds_ib_flush_mr_pool()
356 			ibmr = rds_ib_reuse_mr(pool);  in rds_ib_flush_mr_pool()
359 				finish_wait(&pool->flush_wait, &wait);  in rds_ib_flush_mr_pool()
363 			prepare_to_wait(&pool->flush_wait, &wait,  in rds_ib_flush_mr_pool()
365 			if (llist_empty(&pool->clean_list))  in rds_ib_flush_mr_pool()
368 			ibmr = rds_ib_reuse_mr(pool);  in rds_ib_flush_mr_pool()
371 				finish_wait(&pool->flush_wait, &wait);  in rds_ib_flush_mr_pool()
375 		finish_wait(&pool->flush_wait, &wait);  in rds_ib_flush_mr_pool()
377 		mutex_lock(&pool->flush_lock);  in rds_ib_flush_mr_pool()
380 		ibmr = rds_ib_reuse_mr(pool);  in rds_ib_flush_mr_pool()
390 	dirty_to_clean = llist_append_to_list(&pool->drop_list, &unmap_list);  in rds_ib_flush_mr_pool()
391 	dirty_to_clean += llist_append_to_list(&pool->free_list, &unmap_list);  in rds_ib_flush_mr_pool()
395 		spin_lock_irqsave(&pool->clean_lock, flags);  in rds_ib_flush_mr_pool()
396 		llist_append_to_list(&pool->clean_list, &unmap_list);  in rds_ib_flush_mr_pool()
397 		spin_unlock_irqrestore(&pool->clean_lock, flags);  in rds_ib_flush_mr_pool()
400 	free_goal = rds_ib_flush_goal(pool, free_all);  in rds_ib_flush_mr_pool()
417 			spin_lock_irqsave(&pool->clean_lock, flags);  in rds_ib_flush_mr_pool()
419 					&pool->clean_list);  in rds_ib_flush_mr_pool()
420 			spin_unlock_irqrestore(&pool->clean_lock, flags);  in rds_ib_flush_mr_pool()
424 	atomic_sub(unpinned, &pool->free_pinned);  in rds_ib_flush_mr_pool()
425 	atomic_sub(dirty_to_clean, &pool->dirty_count);  in rds_ib_flush_mr_pool()
426 	atomic_sub(nfreed, &pool->item_count);  in rds_ib_flush_mr_pool()
429 	mutex_unlock(&pool->flush_lock);  in rds_ib_flush_mr_pool()
430 	if (waitqueue_active(&pool->flush_wait))  in rds_ib_flush_mr_pool()
431 		wake_up(&pool->flush_wait);  in rds_ib_flush_mr_pool()
436 struct rds_ib_mr *rds_ib_try_reuse_ibmr(struct rds_ib_mr_pool *pool)  in rds_ib_try_reuse_ibmr()  argument
442 		ibmr = rds_ib_reuse_mr(pool);  in rds_ib_try_reuse_ibmr()
446 		if (atomic_inc_return(&pool->item_count) <= pool->max_items)  in rds_ib_try_reuse_ibmr()
449 		atomic_dec(&pool->item_count);  in rds_ib_try_reuse_ibmr()
452 			if (pool->pool_type == RDS_IB_MR_8K_POOL)  in rds_ib_try_reuse_ibmr()
460 		if (pool->pool_type == RDS_IB_MR_8K_POOL)  in rds_ib_try_reuse_ibmr()
465 		rds_ib_flush_mr_pool(pool, 0, &ibmr);  in rds_ib_try_reuse_ibmr()
475 	struct rds_ib_mr_pool *pool = container_of(work, struct rds_ib_mr_pool, flush_worker.work);  in rds_ib_mr_pool_flush_worker()  local
477 	rds_ib_flush_mr_pool(pool, 0, NULL);  in rds_ib_mr_pool_flush_worker()
483 	struct rds_ib_mr_pool *pool = ibmr->pool;  in rds_ib_free_mr()  local
498 	/* Return it to the pool's free list */  in rds_ib_free_mr()
501 	atomic_add(ibmr->sg_len, &pool->free_pinned);  in rds_ib_free_mr()
502 	atomic_inc(&pool->dirty_count);  in rds_ib_free_mr()
505 	if (atomic_read(&pool->free_pinned) >= pool->max_free_pinned ||  in rds_ib_free_mr()
506 	    atomic_read(&pool->dirty_count) >= pool->max_items / 5)  in rds_ib_free_mr()
507 		queue_delayed_work(rds_ib_mr_wq, &pool->flush_worker, 10);  in rds_ib_free_mr()
511 			rds_ib_flush_mr_pool(pool, 0, NULL);  in rds_ib_free_mr()
517 					   &pool->flush_worker, 10);  in rds_ib_free_mr()
630 void rds_ib_destroy_mr_pool(struct rds_ib_mr_pool *pool)  in rds_ib_destroy_mr_pool()  argument
632 	cancel_delayed_work_sync(&pool->flush_worker);  in rds_ib_destroy_mr_pool()
633 	rds_ib_flush_mr_pool(pool, 1, NULL);  in rds_ib_destroy_mr_pool()
634 	WARN_ON(atomic_read(&pool->item_count));  in rds_ib_destroy_mr_pool()
635 	WARN_ON(atomic_read(&pool->free_pinned));  in rds_ib_destroy_mr_pool()
636 	kfree(pool);  in rds_ib_destroy_mr_pool()
642 	struct rds_ib_mr_pool *pool;  in rds_ib_create_mr_pool()  local
644 	pool = kzalloc_obj(*pool);  in rds_ib_create_mr_pool()
645 	if (!pool)  in rds_ib_create_mr_pool()
648 	pool->pool_type = pool_type;  in rds_ib_create_mr_pool()
649 	init_llist_head(&pool->free_list);  in rds_ib_create_mr_pool()
650 	init_llist_head(&pool->drop_list);  in rds_ib_create_mr_pool()
651 	init_llist_head(&pool->clean_list);  in rds_ib_create_mr_pool()
652 	spin_lock_init(&pool->clean_lock);  in rds_ib_create_mr_pool()
653 	mutex_init(&pool->flush_lock);  in rds_ib_create_mr_pool()
654 	init_waitqueue_head(&pool->flush_wait);  in rds_ib_create_mr_pool()
655 	INIT_DELAYED_WORK(&pool->flush_worker, rds_ib_mr_pool_flush_worker);  in rds_ib_create_mr_pool()
659 		pool->max_pages = RDS_MR_1M_MSG_SIZE + 1;  in rds_ib_create_mr_pool()
660 		pool->max_items = rds_ibdev->max_1m_mrs;  in rds_ib_create_mr_pool()
663 		pool->max_pages = RDS_MR_8K_MSG_SIZE + 1;  in rds_ib_create_mr_pool()
664 		pool->max_items = rds_ibdev->max_8k_mrs;  in rds_ib_create_mr_pool()
667 	pool->max_free_pinned = pool->max_items * pool->max_pages / 4;  in rds_ib_create_mr_pool()
668 	pool->max_items_soft = rds_ibdev->max_mrs * 3 / 4;  in rds_ib_create_mr_pool()
670 	return pool;  in rds_ib_create_mr_pool()
683  * had their pools freed.  As each pool is freed its work struct is waited on,
684  * so the pool flushing work queue should be idle by the time we get here.