565 	struct aa_labelset *ls = labels_set(label);  in __label_remove()  local
567 	AA_BUG(!ls);  in __label_remove()
569 	lockdep_assert_held_write(&ls->lock);  in __label_remove()
578 		rb_erase(&label->node, &ls->root);  in __label_remove()
601 	struct aa_labelset *ls = labels_set(old);  in __label_replace()  local
603 	AA_BUG(!ls);  in __label_replace()
606 	lockdep_assert_held_write(&ls->lock);  in __label_replace()
613 		rb_replace_node(&old->node, &new->node, &ls->root);  in __label_replace()
635 static struct aa_label *__label_insert(struct aa_labelset *ls,  in __label_insert()  argument
640 	AA_BUG(!ls);  in __label_insert()
642 	AA_BUG(labels_set(label) != ls);  in __label_insert()
643 	lockdep_assert_held_write(&ls->lock);  in __label_insert()
647 	new = &ls->root.rb_node;  in __label_insert()
674 	rb_insert_color(&label->node, &ls->root);  in __label_insert()
744 	struct aa_labelset *ls = labels_set(label);  in aa_label_remove()  local
748 	AA_BUG(!ls);  in aa_label_remove()
750 	write_lock_irqsave(&ls->lock, flags);  in aa_label_remove()
752 	write_unlock_irqrestore(&ls->lock, flags);  in aa_label_remove()
780 		struct aa_labelset *ls = labels_set(old);  in aa_label_replace()  local
782 		write_lock_irqsave(&ls->lock, flags);  in aa_label_replace()
785 			write_unlock_irqrestore(&ls->lock, flags);  in aa_label_replace()
786 			ls = labels_set(new);  in aa_label_replace()
787 			write_lock_irqsave(&ls->lock, flags);  in aa_label_replace()
789 		l = __label_insert(ls, new, true);  in aa_label_replace()
791 		write_unlock_irqrestore(&ls->lock, flags);  in aa_label_replace()
808 	struct aa_labelset *ls;  in vec_find()  local
816 	ls = vec_labelset(vec, n);  in vec_find()
817 	read_lock_irqsave(&ls->lock, flags);  in vec_find()
819 	read_unlock_irqrestore(&ls->lock, flags);  in vec_find()
829 	struct aa_labelset *ls;  in vec_create_and_insert_label()  local
839 	ls = labels_set(&vec[len - 1]->label);  in vec_create_and_insert_label()
851 	write_lock_irqsave(&ls->lock, flags);  in vec_create_and_insert_label()
852 	label = __label_insert(ls, new, false);  in vec_create_and_insert_label()
853 	write_unlock_irqrestore(&ls->lock, flags);  in vec_create_and_insert_label()
898 struct aa_label *aa_label_insert(struct aa_labelset *ls, struct aa_label *label)  in aa_label_insert()  argument
903 	AA_BUG(!ls);  in aa_label_insert()
908 		read_lock_irqsave(&ls->lock, flags);  in aa_label_insert()
910 		read_unlock_irqrestore(&ls->lock, flags);  in aa_label_insert()
915 	write_lock_irqsave(&ls->lock, flags);  in aa_label_insert()
916 	l = __label_insert(ls, label, false);  in aa_label_insert()
917 	write_unlock_irqrestore(&ls->lock, flags);  in aa_label_insert()
1024 	struct aa_labelset *ls;  in label_merge_insert()  local
1075 	ls = labels_set(new);  in label_merge_insert()
1076 	write_lock_irqsave(&ls->lock, flags);  in label_merge_insert()
1078 	write_unlock_irqrestore(&ls->lock, flags);  in label_merge_insert()
1112 static struct aa_label *__label_find_merge(struct aa_labelset *ls,  in __label_find_merge()  argument
1118 	AA_BUG(!ls);  in __label_find_merge()
1125 	node  = ls->root.rb_node;  in __label_find_merge()
1155 	struct aa_labelset *ls;  in aa_label_find_merge()  local
1166 	ls = labelset_of_merge(a, b);  in aa_label_find_merge()
1167 	read_lock_irqsave(&ls->lock, flags);  in aa_label_find_merge()
1168 	label = __label_find_merge(ls, a, b);  in aa_label_find_merge()
1169 	read_unlock_irqrestore(&ls->lock, flags);  in aa_label_find_merge()
1418 	struct aa_labelset *ls;  in aa_update_label_name()  local
1432 	ls = labels_set(label);  in aa_update_label_name()
1433 	write_lock_irqsave(&ls->lock, flags);  in aa_update_label_name()
1439 	write_unlock_irqrestore(&ls->lock, flags);  in aa_update_label_name()
1950 void aa_labelset_destroy(struct aa_labelset *ls)  in aa_labelset_destroy()  argument
1955 	AA_BUG(!ls);  in aa_labelset_destroy()
1957 	write_lock_irqsave(&ls->lock, flags);  in aa_labelset_destroy()
1958 	for (node = rb_first(&ls->root); node; node = rb_first(&ls->root)) {  in aa_labelset_destroy()
1967 	write_unlock_irqrestore(&ls->lock, flags);  in aa_labelset_destroy()
1973 void aa_labelset_init(struct aa_labelset *ls)  in aa_labelset_init()  argument
1975 	AA_BUG(!ls);  in aa_labelset_init()
1977 	rwlock_init(&ls->lock);  in aa_labelset_init()
1978 	ls->root = RB_ROOT;  in aa_labelset_init()
1981 static struct aa_label *labelset_next_stale(struct aa_labelset *ls)  in labelset_next_stale()  argument
1987 	AA_BUG(!ls);  in labelset_next_stale()
1989 	read_lock_irqsave(&ls->lock, flags);  in labelset_next_stale()
1991 	__labelset_for_each(ls, node) {  in labelset_next_stale()
2002 	read_unlock_irqrestore(&ls->lock, flags);  in labelset_next_stale()
2022 	struct aa_labelset *ls;  in __label_update()  local
2037 	ls = labels_set(label);  in __label_update()
2038 	write_lock_irqsave(&ls->lock, flags);  in __label_update()
2060 			write_unlock_irqrestore(&ls->lock, flags);  in __label_update()
2062 			write_lock_irqsave(&ls->lock, flags);  in __label_update()
2072 	write_unlock_irqrestore(&ls->lock, flags);  in __label_update()