592 	struct aa_labelset *ls = labels_set(label);  in __label_remove()  local
594 	AA_BUG(!ls);  in __label_remove()
596 	lockdep_assert_held_write(&ls->lock);  in __label_remove()
605 		rb_erase(&label->node, &ls->root);  in __label_remove()
628 	struct aa_labelset *ls = labels_set(old);  in __label_replace()  local
630 	AA_BUG(!ls);  in __label_replace()
633 	lockdep_assert_held_write(&ls->lock);  in __label_replace()
640 		rb_replace_node(&old->node, &new->node, &ls->root);  in __label_replace()
662 static struct aa_label *__label_insert(struct aa_labelset *ls,  in __label_insert()  argument
667 	AA_BUG(!ls);  in __label_insert()
669 	AA_BUG(labels_set(label) != ls);  in __label_insert()
670 	lockdep_assert_held_write(&ls->lock);  in __label_insert()
674 	new = &ls->root.rb_node;  in __label_insert()
701 	rb_insert_color(&label->node, &ls->root);  in __label_insert()
771 	struct aa_labelset *ls = labels_set(label);  in aa_label_remove()  local
775 	AA_BUG(!ls);  in aa_label_remove()
777 	write_lock_irqsave(&ls->lock, flags);  in aa_label_remove()
779 	write_unlock_irqrestore(&ls->lock, flags);  in aa_label_remove()
807 		struct aa_labelset *ls = labels_set(old);  in aa_label_replace()  local
809 		write_lock_irqsave(&ls->lock, flags);  in aa_label_replace()
812 			write_unlock_irqrestore(&ls->lock, flags);  in aa_label_replace()
813 			ls = labels_set(new);  in aa_label_replace()
814 			write_lock_irqsave(&ls->lock, flags);  in aa_label_replace()
816 		l = __label_insert(ls, new, true);  in aa_label_replace()
818 		write_unlock_irqrestore(&ls->lock, flags);  in aa_label_replace()
835 	struct aa_labelset *ls;  in vec_find()  local
843 	ls = vec_labelset(vec, n);  in vec_find()
844 	read_lock_irqsave(&ls->lock, flags);  in vec_find()
846 	read_unlock_irqrestore(&ls->lock, flags);  in vec_find()
856 	struct aa_labelset *ls;  in vec_create_and_insert_label()  local
866 	ls = labels_set(&vec[len - 1]->label);  in vec_create_and_insert_label()
878 	write_lock_irqsave(&ls->lock, flags);  in vec_create_and_insert_label()
879 	label = __label_insert(ls, new, false);  in vec_create_and_insert_label()
880 	write_unlock_irqrestore(&ls->lock, flags);  in vec_create_and_insert_label()
925 struct aa_label *aa_label_insert(struct aa_labelset *ls, struct aa_label *label)  in aa_label_insert()  argument
930 	AA_BUG(!ls);  in aa_label_insert()
935 		read_lock_irqsave(&ls->lock, flags);  in aa_label_insert()
937 		read_unlock_irqrestore(&ls->lock, flags);  in aa_label_insert()
942 	write_lock_irqsave(&ls->lock, flags);  in aa_label_insert()
943 	l = __label_insert(ls, label, false);  in aa_label_insert()
944 	write_unlock_irqrestore(&ls->lock, flags);  in aa_label_insert()
1051 	struct aa_labelset *ls;  in label_merge_insert()  local
1102 	ls = labels_set(new);  in label_merge_insert()
1103 	write_lock_irqsave(&ls->lock, flags);  in label_merge_insert()
1105 	write_unlock_irqrestore(&ls->lock, flags);  in label_merge_insert()
1139 static struct aa_label *__label_find_merge(struct aa_labelset *ls,  in __label_find_merge()  argument
1145 	AA_BUG(!ls);  in __label_find_merge()
1152 	node  = ls->root.rb_node;  in __label_find_merge()
1182 	struct aa_labelset *ls;  in aa_label_find_merge()  local
1193 	ls = labelset_of_merge(a, b);  in aa_label_find_merge()
1194 	read_lock_irqsave(&ls->lock, flags);  in aa_label_find_merge()
1195 	label = __label_find_merge(ls, a, b);  in aa_label_find_merge()
1196 	read_unlock_irqrestore(&ls->lock, flags);  in aa_label_find_merge()
1445 	struct aa_labelset *ls;  in aa_update_label_name()  local
1459 	ls = labels_set(label);  in aa_update_label_name()
1460 	write_lock_irqsave(&ls->lock, flags);  in aa_update_label_name()
1466 	write_unlock_irqrestore(&ls->lock, flags);  in aa_update_label_name()
1980 void aa_labelset_destroy(struct aa_labelset *ls)  in aa_labelset_destroy()  argument
1985 	AA_BUG(!ls);  in aa_labelset_destroy()
1987 	write_lock_irqsave(&ls->lock, flags);  in aa_labelset_destroy()
1988 	for (node = rb_first(&ls->root); node; node = rb_first(&ls->root)) {  in aa_labelset_destroy()
1997 	write_unlock_irqrestore(&ls->lock, flags);  in aa_labelset_destroy()
2003 void aa_labelset_init(struct aa_labelset *ls)  in aa_labelset_init()  argument
2005 	AA_BUG(!ls);  in aa_labelset_init()
2007 	rwlock_init(&ls->lock);  in aa_labelset_init()
2008 	ls->root = RB_ROOT;  in aa_labelset_init()
2011 static struct aa_label *labelset_next_stale(struct aa_labelset *ls)  in labelset_next_stale()  argument
2017 	AA_BUG(!ls);  in labelset_next_stale()
2019 	read_lock_irqsave(&ls->lock, flags);  in labelset_next_stale()
2021 	__labelset_for_each(ls, node) {  in labelset_next_stale()
2032 	read_unlock_irqrestore(&ls->lock, flags);  in labelset_next_stale()
2052 	struct aa_labelset *ls;  in __label_update()  local
2067 	ls = labels_set(label);  in __label_update()
2068 	write_lock_irqsave(&ls->lock, flags);  in __label_update()
2090 			write_unlock_irqrestore(&ls->lock, flags);  in __label_update()
2092 			write_lock_irqsave(&ls->lock, flags);  in __label_update()
2102 	write_unlock_irqrestore(&ls->lock, flags);  in __label_update()