23  * @mark is an xa_mark_t; a small number indicating one of the mark bits.
66 static inline void xa_mark_set(struct xarray *xa, xa_mark_t mark)  in xa_mark_set()  argument
68 	if (!(xa->xa_flags & XA_FLAGS_MARK(mark)))  in xa_mark_set()
69 		xa->xa_flags |= XA_FLAGS_MARK(mark);  in xa_mark_set()
72 static inline void xa_mark_clear(struct xarray *xa, xa_mark_t mark)  in xa_mark_clear()  argument
74 	if (xa->xa_flags & XA_FLAGS_MARK(mark))  in xa_mark_clear()
75 		xa->xa_flags &= ~(XA_FLAGS_MARK(mark));  in xa_mark_clear()
78 static inline unsigned long *node_marks(struct xa_node *node, xa_mark_t mark)  in node_marks()  argument
80 	return node->marks[(__force unsigned)mark];  in node_marks()
84 		unsigned int offset, xa_mark_t mark)  in node_get_mark()  argument
86 	return test_bit(offset, node_marks(node, mark));  in node_get_mark()
91 				xa_mark_t mark)  in node_set_mark()  argument
93 	return __test_and_set_bit(offset, node_marks(node, mark));  in node_set_mark()
98 				xa_mark_t mark)  in node_clear_mark()  argument
100 	return __test_and_clear_bit(offset, node_marks(node, mark));  in node_clear_mark()
103 static inline bool node_any_mark(struct xa_node *node, xa_mark_t mark)  in node_any_mark()  argument
105 	return !bitmap_empty(node_marks(node, mark), XA_CHUNK_SIZE);  in node_any_mark()
108 static inline void node_mark_all(struct xa_node *node, xa_mark_t mark)  in node_mark_all()  argument
110 	bitmap_fill(node_marks(node, mark), XA_CHUNK_SIZE);  in node_mark_all()
113 #define mark_inc(mark) do { \  argument
114 	mark = (__force xa_mark_t)((__force unsigned)(mark) + 1); \
121  * Set a mark on the first entry if any entry has it set.  Clear marks on
126 	unsigned int mark = 0;  in xas_squash_marks()  local
133 		unsigned long *marks = xas->xa_node->marks[mark];  in xas_squash_marks()
138 	} while (mark++ != (__force unsigned)XA_MARK_MAX);  in xas_squash_marks()
578 		xa_mark_t mark = 0;  in xas_expand()  local
590 		/* Propagate the aggregated mark info to the new child */  in xas_expand()
592 			if (xa_track_free(xa) && mark == XA_FREE_MARK) {  in xas_expand()
598 			} else if (xa_marked(xa, mark)) {  in xas_expand()
599 				node_set_mark(node, 0, mark);  in xas_expand()
601 			if (mark == XA_MARK_MAX)  in xas_expand()
603 			mark_inc(mark);  in xas_expand()
813 		 * so the mark clearing will appear to happen before the  in xas_store()
847  * xas_get_mark() - Returns the state of this mark.
849  * @mark: Mark number.
851  * Return: true if the mark is set, false if the mark is clear or @xas
854 bool xas_get_mark(const struct xa_state *xas, xa_mark_t mark)  in xas_get_mark()  argument
859 		return xa_marked(xas->xa, mark);  in xas_get_mark()
860 	return node_get_mark(xas->xa_node, xas->xa_offset, mark);  in xas_get_mark()
865  * xas_set_mark() - Sets the mark on this entry and its parents.
867  * @mark: Mark number.
869  * Sets the specified mark on this entry, and walks up the tree setting it
873 void xas_set_mark(const struct xa_state *xas, xa_mark_t mark)  in xas_set_mark()  argument
882 		if (node_set_mark(node, offset, mark))  in xas_set_mark()
888 	if (!xa_marked(xas->xa, mark))  in xas_set_mark()
889 		xa_mark_set(xas->xa, mark);  in xas_set_mark()
894  * xas_clear_mark() - Clears the mark on this entry and its parents.
896  * @mark: Mark number.
898  * Clears the specified mark on this entry, and walks back to the head
902 void xas_clear_mark(const struct xa_state *xas, xa_mark_t mark)  in xas_clear_mark()  argument
911 		if (!node_clear_mark(node, offset, mark))  in xas_clear_mark()
913 		if (node_any_mark(node, mark))  in xas_clear_mark()
920 	if (xa_marked(xas->xa, mark))  in xas_clear_mark()
921 		xa_mark_clear(xas->xa, mark);  in xas_clear_mark()
930  * free entries with a mark, we need to set it on all entries.  All other
938 	xa_mark_t mark = 0;  in xas_init_marks()  local
941 		if (xa_track_free(xas->xa) && mark == XA_FREE_MARK)  in xas_init_marks()
942 			xas_set_mark(xas, mark);  in xas_init_marks()
944 			xas_clear_mark(xas, mark);  in xas_init_marks()
945 		if (mark == XA_MARK_MAX)  in xas_init_marks()
947 		mark_inc(mark);  in xas_init_marks()
956 	xa_mark_t mark = XA_MARK_0;  in node_get_marks()  local
959 		if (node_get_mark(node, offset, mark))  in node_get_marks()
960 			marks |= 1 << (__force unsigned int)mark;  in node_get_marks()
961 		if (mark == XA_MARK_MAX)  in node_get_marks()
963 		mark_inc(mark);  in node_get_marks()
972 	xa_mark_t mark = XA_MARK_0;  in node_set_marks()  local
975 		if (marks & (1 << (__force unsigned int)mark)) {  in node_set_marks()
976 			node_set_mark(node, offset, mark);  in node_set_marks()
978 				node_mark_all(child, mark);  in node_set_marks()
980 		if (mark == XA_MARK_MAX)  in node_set_marks()
982 		mark_inc(mark);  in node_set_marks()
1283  * @mark: Mark number to search for.
1300 void *xas_find_marked(struct xa_state *xas, unsigned long max, xa_mark_t mark)  in xas_find_marked()  argument
1321 			if (xa_marked(xas->xa, mark))  in xas_find_marked()
1348 		offset = xas_find_chunk(xas, advance, mark);  in xas_find_marked()
1361 		if (!entry && !(xa_track_free(xas->xa) && mark == XA_FREE_MARK))  in xas_find_marked()
1883  * __xa_set_mark() - Set this mark on this entry while locked.
1886  * @mark: Mark number.
1888  * Attempting to set a mark on a %NULL entry does not succeed.
1892 void __xa_set_mark(struct xarray *xa, unsigned long index, xa_mark_t mark)  in __xa_set_mark()  argument
1898 		xas_set_mark(&xas, mark);  in __xa_set_mark()
1903  * __xa_clear_mark() - Clear this mark on this entry while locked.
1906  * @mark: Mark number.
1910 void __xa_clear_mark(struct xarray *xa, unsigned long index, xa_mark_t mark)  in __xa_clear_mark()  argument
1916 		xas_clear_mark(&xas, mark);  in __xa_clear_mark()
1921  * xa_get_mark() - Inquire whether this mark is set on this entry.
1924  * @mark: Mark number.
1930  * Return: True if the entry at @index has this mark set, false if it doesn't.
1932 bool xa_get_mark(struct xarray *xa, unsigned long index, xa_mark_t mark)  in xa_get_mark()  argument
1939 	while (xas_get_mark(&xas, mark)) {  in xa_get_mark()
1953  * xa_set_mark() - Set this mark on this entry.
1956  * @mark: Mark number.
1958  * Attempting to set a mark on a %NULL entry does not succeed.
1962 void xa_set_mark(struct xarray *xa, unsigned long index, xa_mark_t mark)  in xa_set_mark()  argument
1965 	__xa_set_mark(xa, index, mark);  in xa_set_mark()
1971  * xa_clear_mark() - Clear this mark on this entry.
1974  * @mark: Mark number.
1976  * Clearing a mark always succeeds.
1980 void xa_clear_mark(struct xarray *xa, unsigned long index, xa_mark_t mark)  in xa_clear_mark()  argument
1983 	__xa_clear_mark(xa, index, mark);  in xa_clear_mark()
2106 			unsigned long max, unsigned int n, xa_mark_t mark)  in xas_extract_marked()  argument
2112 	xas_for_each_marked(xas, entry, max, mark) {  in xas_extract_marked()
2136  * The @filter may be an XArray mark value, in which case entries which are
2137  * marked with that mark will be copied.  It may also be %XA_PRESENT, in