25  * @mark is an xa_mark_t; a small number indicating one of the mark bits.
68 static inline void xa_mark_set(struct xarray *xa, xa_mark_t mark)  in xa_mark_set()  argument
70 	if (!(xa->xa_flags & XA_FLAGS_MARK(mark)))  in xa_mark_set()
71 		xa->xa_flags |= XA_FLAGS_MARK(mark);  in xa_mark_set()
74 static inline void xa_mark_clear(struct xarray *xa, xa_mark_t mark)  in xa_mark_clear()  argument
76 	if (xa->xa_flags & XA_FLAGS_MARK(mark))  in xa_mark_clear()
77 		xa->xa_flags &= ~(XA_FLAGS_MARK(mark));  in xa_mark_clear()
80 static inline unsigned long *node_marks(struct xa_node *node, xa_mark_t mark)  in node_marks()  argument
82 	return node->marks[(__force unsigned)mark];  in node_marks()
86 		unsigned int offset, xa_mark_t mark)  in node_get_mark()  argument
88 	return test_bit(offset, node_marks(node, mark));  in node_get_mark()
93 				xa_mark_t mark)  in node_set_mark()  argument
95 	return __test_and_set_bit(offset, node_marks(node, mark));  in node_set_mark()
100 				xa_mark_t mark)  in node_clear_mark()  argument
102 	return __test_and_clear_bit(offset, node_marks(node, mark));  in node_clear_mark()
105 static inline bool node_any_mark(struct xa_node *node, xa_mark_t mark)  in node_any_mark()  argument
107 	return !bitmap_empty(node_marks(node, mark), XA_CHUNK_SIZE);  in node_any_mark()
110 static inline void node_mark_all(struct xa_node *node, xa_mark_t mark)  in node_mark_all()  argument
112 	bitmap_fill(node_marks(node, mark), XA_CHUNK_SIZE);  in node_mark_all()
115 #define mark_inc(mark) do { \  argument
116 	mark = (__force xa_mark_t)((__force unsigned)(mark) + 1); \
123  * Set a mark on the first entry if any entry has it set.  Clear marks on
128 	unsigned int mark = 0;  in xas_squash_marks()  local
135 		unsigned long *marks = xas->xa_node->marks[mark];  in xas_squash_marks()
140 	} while (mark++ != (__force unsigned)XA_MARK_MAX);  in xas_squash_marks()
579 		xa_mark_t mark = 0;  in xas_expand()  local
591 		/* Propagate the aggregated mark info to the new child */  in xas_expand()
593 			if (xa_track_free(xa) && mark == XA_FREE_MARK) {  in xas_expand()
599 			} else if (xa_marked(xa, mark)) {  in xas_expand()
600 				node_set_mark(node, 0, mark);  in xas_expand()
602 			if (mark == XA_MARK_MAX)  in xas_expand()
604 			mark_inc(mark);  in xas_expand()
816 		 * so the mark clearing will appear to happen before the  in xas_store()
850  * xas_get_mark() - Returns the state of this mark.
852  * @mark: Mark number.
854  * Return: true if the mark is set, false if the mark is clear or @xas
857 bool xas_get_mark(const struct xa_state *xas, xa_mark_t mark)  in xas_get_mark()  argument
862 		return xa_marked(xas->xa, mark);  in xas_get_mark()
863 	return node_get_mark(xas->xa_node, xas->xa_offset, mark);  in xas_get_mark()
868  * xas_set_mark() - Sets the mark on this entry and its parents.
870  * @mark: Mark number.
872  * Sets the specified mark on this entry, and walks up the tree setting it
876 void xas_set_mark(const struct xa_state *xas, xa_mark_t mark)  in xas_set_mark()  argument
885 		if (node_set_mark(node, offset, mark))  in xas_set_mark()
891 	if (!xa_marked(xas->xa, mark))  in xas_set_mark()
892 		xa_mark_set(xas->xa, mark);  in xas_set_mark()
897  * xas_clear_mark() - Clears the mark on this entry and its parents.
899  * @mark: Mark number.
901  * Clears the specified mark on this entry, and walks back to the head
905 void xas_clear_mark(const struct xa_state *xas, xa_mark_t mark)  in xas_clear_mark()  argument
914 		if (!node_clear_mark(node, offset, mark))  in xas_clear_mark()
916 		if (node_any_mark(node, mark))  in xas_clear_mark()
923 	if (xa_marked(xas->xa, mark))  in xas_clear_mark()
924 		xa_mark_clear(xas->xa, mark);  in xas_clear_mark()
933  * free entries with a mark, we need to set it on all entries.  All other
941 	xa_mark_t mark = 0;  in xas_init_marks()  local
944 		if (xa_track_free(xas->xa) && mark == XA_FREE_MARK)  in xas_init_marks()
945 			xas_set_mark(xas, mark);  in xas_init_marks()
947 			xas_clear_mark(xas, mark);  in xas_init_marks()
948 		if (mark == XA_MARK_MAX)  in xas_init_marks()
950 		mark_inc(mark);  in xas_init_marks()
959 	xa_mark_t mark = XA_MARK_0;  in node_get_marks()  local
962 		if (node_get_mark(node, offset, mark))  in node_get_marks()
963 			marks |= 1 << (__force unsigned int)mark;  in node_get_marks()
964 		if (mark == XA_MARK_MAX)  in node_get_marks()
966 		mark_inc(mark);  in node_get_marks()
975 	xa_mark_t mark = XA_MARK_0;  in node_set_marks()  local
978 		if (marks & (1 << (__force unsigned int)mark)) {  in node_set_marks()
979 			node_set_mark(node, offset, mark);  in node_set_marks()
981 				node_mark_all(child, mark);  in node_set_marks()
983 		if (mark == XA_MARK_MAX)  in node_set_marks()
985 		mark_inc(mark);  in node_set_marks()
1289  * @mark: Mark number to search for.
1306 void *xas_find_marked(struct xa_state *xas, unsigned long max, xa_mark_t mark)  in xas_find_marked()  argument
1327 			if (xa_marked(xas->xa, mark))  in xas_find_marked()
1354 		offset = xas_find_chunk(xas, advance, mark);  in xas_find_marked()
1367 		if (!entry && !(xa_track_free(xas->xa) && mark == XA_FREE_MARK))  in xas_find_marked()
1908  * __xa_set_mark() - Set this mark on this entry while locked.
1911  * @mark: Mark number.
1913  * Attempting to set a mark on a %NULL entry does not succeed.
1917 void __xa_set_mark(struct xarray *xa, unsigned long index, xa_mark_t mark)  in __xa_set_mark()  argument
1923 		xas_set_mark(&xas, mark);  in __xa_set_mark()
1928  * __xa_clear_mark() - Clear this mark on this entry while locked.
1931  * @mark: Mark number.
1935 void __xa_clear_mark(struct xarray *xa, unsigned long index, xa_mark_t mark)  in __xa_clear_mark()  argument
1941 		xas_clear_mark(&xas, mark);  in __xa_clear_mark()
1946  * xa_get_mark() - Inquire whether this mark is set on this entry.
1949  * @mark: Mark number.
1955  * Return: True if the entry at @index has this mark set, false if it doesn't.
1957 bool xa_get_mark(struct xarray *xa, unsigned long index, xa_mark_t mark)  in xa_get_mark()  argument
1964 	while (xas_get_mark(&xas, mark)) {  in xa_get_mark()
1978  * xa_set_mark() - Set this mark on this entry.
1981  * @mark: Mark number.
1983  * Attempting to set a mark on a %NULL entry does not succeed.
1987 void xa_set_mark(struct xarray *xa, unsigned long index, xa_mark_t mark)  in xa_set_mark()  argument
1990 	__xa_set_mark(xa, index, mark);  in xa_set_mark()
1996  * xa_clear_mark() - Clear this mark on this entry.
1999  * @mark: Mark number.
2001  * Clearing a mark always succeeds.
2005 void xa_clear_mark(struct xarray *xa, unsigned long index, xa_mark_t mark)  in xa_clear_mark()  argument
2008 	__xa_clear_mark(xa, index, mark);  in xa_clear_mark()
2131 			unsigned long max, unsigned int n, xa_mark_t mark)  in xas_extract_marked()  argument
2137 	xas_for_each_marked(xas, entry, max, mark) {  in xas_extract_marked()
2161  * The @filter may be an XArray mark value, in which case entries which are
2162  * marked with that mark will be copied.  It may also be %XA_PRESENT, in