106 	struct rb_root *sl_id_map = &to_mdev(ibdev)->sriov.sl_id_map;  in id_map_find_by_sl_id()
133 	struct mlx4_ib_sriov *sriov = &dev->sriov;  in id_map_ent_timeout()  local
134 	struct rb_root *sl_id_map = &sriov->sl_id_map;  in id_map_ent_timeout()
137 	spin_lock(&sriov->id_map_lock);  in id_map_ent_timeout()
138 	db_ent = (struct id_map_entry *)idr_find(&sriov->pv_id_table, pv_id);  in id_map_ent_timeout()
144 	idr_remove(&sriov->pv_id_table, pv_id);  in id_map_ent_timeout()
148 	spin_unlock(&sriov->id_map_lock);  in id_map_ent_timeout()
154 	struct mlx4_ib_sriov *sriov = &to_mdev(ibdev)->sriov;  in id_map_find_del()  local
155 	struct rb_root *sl_id_map = &sriov->sl_id_map;  in id_map_find_del()
158 	spin_lock(&sriov->id_map_lock);  in id_map_find_del()
159 	ent = (struct id_map_entry *)idr_find(&sriov->pv_id_table, pv_cm_id);  in id_map_find_del()
165 	idr_remove(&sriov->pv_id_table, pv_cm_id);  in id_map_find_del()
167 	spin_unlock(&sriov->id_map_lock);  in id_map_find_del()
172 	struct rb_root *sl_id_map = &to_mdev(ibdev)->sriov.sl_id_map;  in sl_id_map_add()
208 	struct mlx4_ib_sriov *sriov = &to_mdev(ibdev)->sriov;  in id_map_alloc()  local
223 		spin_lock(&to_mdev(ibdev)->sriov.id_map_lock);  in id_map_alloc()
224 		ret = idr_get_new_above(&sriov->pv_id_table, ent,  in id_map_alloc()
232 		spin_unlock(&sriov->id_map_lock);  in id_map_alloc()
233 	} while (ret == -EAGAIN && idr_pre_get(&sriov->pv_id_table, GFP_KERNEL));  in id_map_alloc()
236 		spin_lock(&sriov->id_map_lock);  in id_map_alloc()
237 		list_add_tail(&ent->list, &sriov->cm_list);  in id_map_alloc()
238 		spin_unlock(&sriov->id_map_lock);  in id_map_alloc()
251 	struct mlx4_ib_sriov *sriov = &to_mdev(ibdev)->sriov;  in id_map_get()  local
253 	spin_lock(&sriov->id_map_lock);  in id_map_get()
259 		ent = (struct id_map_entry *)idr_find(&sriov->pv_id_table, *pv_cm_id);  in id_map_get()
260 	spin_unlock(&sriov->id_map_lock);  in id_map_get()
267 	struct mlx4_ib_sriov *sriov = &to_mdev(ibdev)->sriov;  in schedule_delayed()  local
270 	spin_lock(&sriov->id_map_lock);  in schedule_delayed()
271 	spin_lock_irqsave(&sriov->going_down_lock, flags);  in schedule_delayed()
273 	if (!sriov->is_going_down) {  in schedule_delayed()
277 	spin_unlock_irqrestore(&sriov->going_down_lock, flags);  in schedule_delayed()
278 	spin_unlock(&sriov->id_map_lock);  in schedule_delayed()
366 	spin_lock_init(&dev->sriov.id_map_lock);  in mlx4_ib_cm_paravirt_init()
367 	INIT_LIST_HEAD(&dev->sriov.cm_list);  in mlx4_ib_cm_paravirt_init()
368 	dev->sriov.sl_id_map = RB_ROOT;  in mlx4_ib_cm_paravirt_init()
369 	idr_init(&dev->sriov.pv_id_table);  in mlx4_ib_cm_paravirt_init()
370 	idr_pre_get(&dev->sriov.pv_id_table, GFP_KERNEL);  in mlx4_ib_cm_paravirt_init()
377 	struct mlx4_ib_sriov *sriov = &dev->sriov;  in mlx4_ib_cm_paravirt_clean()  local
378 	struct rb_root *sl_id_map = &sriov->sl_id_map;  in mlx4_ib_cm_paravirt_clean()
385 	spin_lock(&sriov->id_map_lock);  in mlx4_ib_cm_paravirt_clean()
386 	list_for_each_entry_safe(map, tmp_map, &dev->sriov.cm_list, list) {  in mlx4_ib_cm_paravirt_clean()
393 	spin_unlock(&sriov->id_map_lock);  in mlx4_ib_cm_paravirt_clean()
399 	spin_lock(&sriov->id_map_lock);  in mlx4_ib_cm_paravirt_clean()
407 			idr_remove(&sriov->pv_id_table, (int) ent->pv_cm_id);  in mlx4_ib_cm_paravirt_clean()
409 		list_splice_init(&dev->sriov.cm_list, &lh);  in mlx4_ib_cm_paravirt_clean()
423 			idr_remove(&sriov->pv_id_table, (int) map->pv_cm_id);  in mlx4_ib_cm_paravirt_clean()
427 		list_for_each_entry_safe(map, tmp_map, &dev->sriov.cm_list, list) {  in mlx4_ib_cm_paravirt_clean()
433 	spin_unlock(&sriov->id_map_lock);  in mlx4_ib_cm_paravirt_clean()