Loading...
1/* SPDX-License-Identifier: GPL-2.0 */
2/*
3 * Copyright (c) 2000-2001,2005 Silicon Graphics, Inc.
4 * All Rights Reserved.
5 */
6#ifndef __XFS_BTREE_H__
7#define __XFS_BTREE_H__
8
9struct xfs_buf;
10struct xfs_inode;
11struct xfs_mount;
12struct xfs_trans;
13struct xfs_ifork;
14struct xfs_perag;
15
16/*
17 * Generic key, ptr and record wrapper structures.
18 *
19 * These are disk format structures, and are converted where necessary
20 * by the btree specific code that needs to interpret them.
21 */
22union xfs_btree_ptr {
23 __be32 s; /* short form ptr */
24 __be64 l; /* long form ptr */
25};
26
27/*
28 * The in-core btree key. Overlapping btrees actually store two keys
29 * per pointer, so we reserve enough memory to hold both. The __*bigkey
30 * items should never be accessed directly.
31 */
32union xfs_btree_key {
33 struct xfs_bmbt_key bmbt;
34 xfs_bmdr_key_t bmbr; /* bmbt root block */
35 xfs_alloc_key_t alloc;
36 struct xfs_inobt_key inobt;
37 struct xfs_rmap_key rmap;
38 struct xfs_rmap_key __rmap_bigkey[2];
39 struct xfs_refcount_key refc;
40};
41
42union xfs_btree_rec {
43 struct xfs_bmbt_rec bmbt;
44 xfs_bmdr_rec_t bmbr; /* bmbt root block */
45 struct xfs_alloc_rec alloc;
46 struct xfs_inobt_rec inobt;
47 struct xfs_rmap_rec rmap;
48 struct xfs_refcount_rec refc;
49};
50
51/*
52 * This nonsense is to make -wlint happy.
53 */
54#define XFS_LOOKUP_EQ ((xfs_lookup_t)XFS_LOOKUP_EQi)
55#define XFS_LOOKUP_LE ((xfs_lookup_t)XFS_LOOKUP_LEi)
56#define XFS_LOOKUP_GE ((xfs_lookup_t)XFS_LOOKUP_GEi)
57
58#define XFS_BTNUM_BNO ((xfs_btnum_t)XFS_BTNUM_BNOi)
59#define XFS_BTNUM_CNT ((xfs_btnum_t)XFS_BTNUM_CNTi)
60#define XFS_BTNUM_BMAP ((xfs_btnum_t)XFS_BTNUM_BMAPi)
61#define XFS_BTNUM_INO ((xfs_btnum_t)XFS_BTNUM_INOi)
62#define XFS_BTNUM_FINO ((xfs_btnum_t)XFS_BTNUM_FINOi)
63#define XFS_BTNUM_RMAP ((xfs_btnum_t)XFS_BTNUM_RMAPi)
64#define XFS_BTNUM_REFC ((xfs_btnum_t)XFS_BTNUM_REFCi)
65
66uint32_t xfs_btree_magic(int crc, xfs_btnum_t btnum);
67
68/*
69 * For logging record fields.
70 */
71#define XFS_BB_MAGIC (1u << 0)
72#define XFS_BB_LEVEL (1u << 1)
73#define XFS_BB_NUMRECS (1u << 2)
74#define XFS_BB_LEFTSIB (1u << 3)
75#define XFS_BB_RIGHTSIB (1u << 4)
76#define XFS_BB_BLKNO (1u << 5)
77#define XFS_BB_LSN (1u << 6)
78#define XFS_BB_UUID (1u << 7)
79#define XFS_BB_OWNER (1u << 8)
80#define XFS_BB_NUM_BITS 5
81#define XFS_BB_ALL_BITS ((1u << XFS_BB_NUM_BITS) - 1)
82#define XFS_BB_NUM_BITS_CRC 9
83#define XFS_BB_ALL_BITS_CRC ((1u << XFS_BB_NUM_BITS_CRC) - 1)
84
85/*
86 * Generic stats interface
87 */
88#define XFS_BTREE_STATS_INC(cur, stat) \
89 XFS_STATS_INC_OFF((cur)->bc_mp, (cur)->bc_statoff + __XBTS_ ## stat)
90#define XFS_BTREE_STATS_ADD(cur, stat, val) \
91 XFS_STATS_ADD_OFF((cur)->bc_mp, (cur)->bc_statoff + __XBTS_ ## stat, val)
92
93enum xbtree_key_contig {
94 XBTREE_KEY_GAP = 0,
95 XBTREE_KEY_CONTIGUOUS,
96 XBTREE_KEY_OVERLAP,
97};
98
99/*
100 * Decide if these two numeric btree key fields are contiguous, overlapping,
101 * or if there's a gap between them. @x should be the field from the high
102 * key and @y should be the field from the low key.
103 */
104static inline enum xbtree_key_contig xbtree_key_contig(uint64_t x, uint64_t y)
105{
106 x++;
107 if (x < y)
108 return XBTREE_KEY_GAP;
109 if (x == y)
110 return XBTREE_KEY_CONTIGUOUS;
111 return XBTREE_KEY_OVERLAP;
112}
113
114struct xfs_btree_ops {
115 /* size of the key and record structures */
116 size_t key_len;
117 size_t rec_len;
118
119 /* cursor operations */
120 struct xfs_btree_cur *(*dup_cursor)(struct xfs_btree_cur *);
121 void (*update_cursor)(struct xfs_btree_cur *src,
122 struct xfs_btree_cur *dst);
123
124 /* update btree root pointer */
125 void (*set_root)(struct xfs_btree_cur *cur,
126 const union xfs_btree_ptr *nptr, int level_change);
127
128 /* block allocation / freeing */
129 int (*alloc_block)(struct xfs_btree_cur *cur,
130 const union xfs_btree_ptr *start_bno,
131 union xfs_btree_ptr *new_bno,
132 int *stat);
133 int (*free_block)(struct xfs_btree_cur *cur, struct xfs_buf *bp);
134
135 /* update last record information */
136 void (*update_lastrec)(struct xfs_btree_cur *cur,
137 const struct xfs_btree_block *block,
138 const union xfs_btree_rec *rec,
139 int ptr, int reason);
140
141 /* records in block/level */
142 int (*get_minrecs)(struct xfs_btree_cur *cur, int level);
143 int (*get_maxrecs)(struct xfs_btree_cur *cur, int level);
144
145 /* records on disk. Matter for the root in inode case. */
146 int (*get_dmaxrecs)(struct xfs_btree_cur *cur, int level);
147
148 /* init values of btree structures */
149 void (*init_key_from_rec)(union xfs_btree_key *key,
150 const union xfs_btree_rec *rec);
151 void (*init_rec_from_cur)(struct xfs_btree_cur *cur,
152 union xfs_btree_rec *rec);
153 void (*init_ptr_from_cur)(struct xfs_btree_cur *cur,
154 union xfs_btree_ptr *ptr);
155 void (*init_high_key_from_rec)(union xfs_btree_key *key,
156 const union xfs_btree_rec *rec);
157
158 /* difference between key value and cursor value */
159 int64_t (*key_diff)(struct xfs_btree_cur *cur,
160 const union xfs_btree_key *key);
161
162 /*
163 * Difference between key2 and key1 -- positive if key1 > key2,
164 * negative if key1 < key2, and zero if equal. If the @mask parameter
165 * is non NULL, each key field to be used in the comparison must
166 * contain a nonzero value.
167 */
168 int64_t (*diff_two_keys)(struct xfs_btree_cur *cur,
169 const union xfs_btree_key *key1,
170 const union xfs_btree_key *key2,
171 const union xfs_btree_key *mask);
172
173 const struct xfs_buf_ops *buf_ops;
174
175 /* check that k1 is lower than k2 */
176 int (*keys_inorder)(struct xfs_btree_cur *cur,
177 const union xfs_btree_key *k1,
178 const union xfs_btree_key *k2);
179
180 /* check that r1 is lower than r2 */
181 int (*recs_inorder)(struct xfs_btree_cur *cur,
182 const union xfs_btree_rec *r1,
183 const union xfs_btree_rec *r2);
184
185 /*
186 * Are these two btree keys immediately adjacent?
187 *
188 * Given two btree keys @key1 and @key2, decide if it is impossible for
189 * there to be a third btree key K satisfying the relationship
190 * @key1 < K < @key2. To determine if two btree records are
191 * immediately adjacent, @key1 should be the high key of the first
192 * record and @key2 should be the low key of the second record.
193 * If the @mask parameter is non NULL, each key field to be used in the
194 * comparison must contain a nonzero value.
195 */
196 enum xbtree_key_contig (*keys_contiguous)(struct xfs_btree_cur *cur,
197 const union xfs_btree_key *key1,
198 const union xfs_btree_key *key2,
199 const union xfs_btree_key *mask);
200};
201
202/*
203 * Reasons for the update_lastrec method to be called.
204 */
205#define LASTREC_UPDATE 0
206#define LASTREC_INSREC 1
207#define LASTREC_DELREC 2
208
209
210union xfs_btree_irec {
211 struct xfs_alloc_rec_incore a;
212 struct xfs_bmbt_irec b;
213 struct xfs_inobt_rec_incore i;
214 struct xfs_rmap_irec r;
215 struct xfs_refcount_irec rc;
216};
217
218/* Per-AG btree information. */
219struct xfs_btree_cur_ag {
220 struct xfs_perag *pag;
221 union {
222 struct xfs_buf *agbp;
223 struct xbtree_afakeroot *afake; /* for staging cursor */
224 };
225 union {
226 struct {
227 unsigned int nr_ops; /* # record updates */
228 unsigned int shape_changes; /* # of extent splits */
229 } refc;
230 struct {
231 bool active; /* allocation cursor state */
232 } abt;
233 };
234};
235
236/* Btree-in-inode cursor information */
237struct xfs_btree_cur_ino {
238 struct xfs_inode *ip;
239 struct xbtree_ifakeroot *ifake; /* for staging cursor */
240 int allocated;
241 short forksize;
242 char whichfork;
243 char flags;
244/* We are converting a delalloc reservation */
245#define XFS_BTCUR_BMBT_WASDEL (1 << 0)
246
247/* For extent swap, ignore owner check in verifier */
248#define XFS_BTCUR_BMBT_INVALID_OWNER (1 << 1)
249};
250
251struct xfs_btree_level {
252 /* buffer pointer */
253 struct xfs_buf *bp;
254
255 /* key/record number */
256 uint16_t ptr;
257
258 /* readahead info */
259#define XFS_BTCUR_LEFTRA (1 << 0) /* left sibling has been read-ahead */
260#define XFS_BTCUR_RIGHTRA (1 << 1) /* right sibling has been read-ahead */
261 uint16_t ra;
262};
263
264/*
265 * Btree cursor structure.
266 * This collects all information needed by the btree code in one place.
267 */
268struct xfs_btree_cur
269{
270 struct xfs_trans *bc_tp; /* transaction we're in, if any */
271 struct xfs_mount *bc_mp; /* file system mount struct */
272 const struct xfs_btree_ops *bc_ops;
273 struct kmem_cache *bc_cache; /* cursor cache */
274 unsigned int bc_flags; /* btree features - below */
275 xfs_btnum_t bc_btnum; /* identifies which btree type */
276 union xfs_btree_irec bc_rec; /* current insert/search record value */
277 uint8_t bc_nlevels; /* number of levels in the tree */
278 uint8_t bc_maxlevels; /* maximum levels for this btree type */
279 int bc_statoff; /* offset of btree stats array */
280
281 /*
282 * Short btree pointers need an agno to be able to turn the pointers
283 * into physical addresses for IO, so the btree cursor switches between
284 * bc_ino and bc_ag based on whether XFS_BTREE_LONG_PTRS is set for the
285 * cursor.
286 */
287 union {
288 struct xfs_btree_cur_ag bc_ag;
289 struct xfs_btree_cur_ino bc_ino;
290 };
291
292 /* Must be at the end of the struct! */
293 struct xfs_btree_level bc_levels[];
294};
295
296/*
297 * Compute the size of a btree cursor that can handle a btree of a given
298 * height. The bc_levels array handles node and leaf blocks, so its size
299 * is exactly nlevels.
300 */
301static inline size_t
302xfs_btree_cur_sizeof(unsigned int nlevels)
303{
304 return struct_size_t(struct xfs_btree_cur, bc_levels, nlevels);
305}
306
307/* cursor flags */
308#define XFS_BTREE_LONG_PTRS (1<<0) /* pointers are 64bits long */
309#define XFS_BTREE_ROOT_IN_INODE (1<<1) /* root may be variable size */
310#define XFS_BTREE_LASTREC_UPDATE (1<<2) /* track last rec externally */
311#define XFS_BTREE_CRC_BLOCKS (1<<3) /* uses extended btree blocks */
312#define XFS_BTREE_OVERLAPPING (1<<4) /* overlapping intervals */
313/*
314 * The root of this btree is a fakeroot structure so that we can stage a btree
315 * rebuild without leaving it accessible via primary metadata. The ops struct
316 * is dynamically allocated and must be freed when the cursor is deleted.
317 */
318#define XFS_BTREE_STAGING (1<<5)
319
320#define XFS_BTREE_NOERROR 0
321#define XFS_BTREE_ERROR 1
322
323/*
324 * Convert from buffer to btree block header.
325 */
326#define XFS_BUF_TO_BLOCK(bp) ((struct xfs_btree_block *)((bp)->b_addr))
327
328/*
329 * Internal long and short btree block checks. They return NULL if the
330 * block is ok or the address of the failed check otherwise.
331 */
332xfs_failaddr_t __xfs_btree_check_lblock(struct xfs_btree_cur *cur,
333 struct xfs_btree_block *block, int level, struct xfs_buf *bp);
334xfs_failaddr_t __xfs_btree_check_sblock(struct xfs_btree_cur *cur,
335 struct xfs_btree_block *block, int level, struct xfs_buf *bp);
336
337/*
338 * Check that block header is ok.
339 */
340int
341xfs_btree_check_block(
342 struct xfs_btree_cur *cur, /* btree cursor */
343 struct xfs_btree_block *block, /* generic btree block pointer */
344 int level, /* level of the btree block */
345 struct xfs_buf *bp); /* buffer containing block, if any */
346
347/*
348 * Check that (long) pointer is ok.
349 */
350bool /* error (0 or EFSCORRUPTED) */
351xfs_btree_check_lptr(
352 struct xfs_btree_cur *cur, /* btree cursor */
353 xfs_fsblock_t fsbno, /* btree block disk address */
354 int level); /* btree block level */
355
356/*
357 * Check that (short) pointer is ok.
358 */
359bool /* error (0 or EFSCORRUPTED) */
360xfs_btree_check_sptr(
361 struct xfs_btree_cur *cur, /* btree cursor */
362 xfs_agblock_t agbno, /* btree block disk address */
363 int level); /* btree block level */
364
365/*
366 * Delete the btree cursor.
367 */
368void
369xfs_btree_del_cursor(
370 struct xfs_btree_cur *cur, /* btree cursor */
371 int error); /* del because of error */
372
373/*
374 * Duplicate the btree cursor.
375 * Allocate a new one, copy the record, re-get the buffers.
376 */
377int /* error */
378xfs_btree_dup_cursor(
379 struct xfs_btree_cur *cur, /* input cursor */
380 struct xfs_btree_cur **ncur);/* output cursor */
381
382/*
383 * Compute first and last byte offsets for the fields given.
384 * Interprets the offsets table, which contains struct field offsets.
385 */
386void
387xfs_btree_offsets(
388 uint32_t fields, /* bitmask of fields */
389 const short *offsets,/* table of field offsets */
390 int nbits, /* number of bits to inspect */
391 int *first, /* output: first byte offset */
392 int *last); /* output: last byte offset */
393
394/*
395 * Get a buffer for the block, return it read in.
396 * Long-form addressing.
397 */
398int /* error */
399xfs_btree_read_bufl(
400 struct xfs_mount *mp, /* file system mount point */
401 struct xfs_trans *tp, /* transaction pointer */
402 xfs_fsblock_t fsbno, /* file system block number */
403 struct xfs_buf **bpp, /* buffer for fsbno */
404 int refval, /* ref count value for buffer */
405 const struct xfs_buf_ops *ops);
406
407/*
408 * Read-ahead the block, don't wait for it, don't return a buffer.
409 * Long-form addressing.
410 */
411void /* error */
412xfs_btree_reada_bufl(
413 struct xfs_mount *mp, /* file system mount point */
414 xfs_fsblock_t fsbno, /* file system block number */
415 xfs_extlen_t count, /* count of filesystem blocks */
416 const struct xfs_buf_ops *ops);
417
418/*
419 * Read-ahead the block, don't wait for it, don't return a buffer.
420 * Short-form addressing.
421 */
422void /* error */
423xfs_btree_reada_bufs(
424 struct xfs_mount *mp, /* file system mount point */
425 xfs_agnumber_t agno, /* allocation group number */
426 xfs_agblock_t agbno, /* allocation group block number */
427 xfs_extlen_t count, /* count of filesystem blocks */
428 const struct xfs_buf_ops *ops);
429
430/*
431 * Initialise a new btree block header
432 */
433void
434xfs_btree_init_block(
435 struct xfs_mount *mp,
436 struct xfs_buf *bp,
437 xfs_btnum_t btnum,
438 __u16 level,
439 __u16 numrecs,
440 __u64 owner);
441
442void
443xfs_btree_init_block_int(
444 struct xfs_mount *mp,
445 struct xfs_btree_block *buf,
446 xfs_daddr_t blkno,
447 xfs_btnum_t btnum,
448 __u16 level,
449 __u16 numrecs,
450 __u64 owner,
451 unsigned int flags);
452
453/*
454 * Common btree core entry points.
455 */
456int xfs_btree_increment(struct xfs_btree_cur *, int, int *);
457int xfs_btree_decrement(struct xfs_btree_cur *, int, int *);
458int xfs_btree_lookup(struct xfs_btree_cur *, xfs_lookup_t, int *);
459int xfs_btree_update(struct xfs_btree_cur *, union xfs_btree_rec *);
460int xfs_btree_new_iroot(struct xfs_btree_cur *, int *, int *);
461int xfs_btree_insert(struct xfs_btree_cur *, int *);
462int xfs_btree_delete(struct xfs_btree_cur *, int *);
463int xfs_btree_get_rec(struct xfs_btree_cur *, union xfs_btree_rec **, int *);
464int xfs_btree_change_owner(struct xfs_btree_cur *cur, uint64_t new_owner,
465 struct list_head *buffer_list);
466
467/*
468 * btree block CRC helpers
469 */
470void xfs_btree_lblock_calc_crc(struct xfs_buf *);
471bool xfs_btree_lblock_verify_crc(struct xfs_buf *);
472void xfs_btree_sblock_calc_crc(struct xfs_buf *);
473bool xfs_btree_sblock_verify_crc(struct xfs_buf *);
474
475/*
476 * Internal btree helpers also used by xfs_bmap.c.
477 */
478void xfs_btree_log_block(struct xfs_btree_cur *, struct xfs_buf *, uint32_t);
479void xfs_btree_log_recs(struct xfs_btree_cur *, struct xfs_buf *, int, int);
480
481/*
482 * Helpers.
483 */
484static inline int xfs_btree_get_numrecs(const struct xfs_btree_block *block)
485{
486 return be16_to_cpu(block->bb_numrecs);
487}
488
489static inline void xfs_btree_set_numrecs(struct xfs_btree_block *block,
490 uint16_t numrecs)
491{
492 block->bb_numrecs = cpu_to_be16(numrecs);
493}
494
495static inline int xfs_btree_get_level(const struct xfs_btree_block *block)
496{
497 return be16_to_cpu(block->bb_level);
498}
499
500
501/*
502 * Min and max functions for extlen, agblock, fileoff, and filblks types.
503 */
504#define XFS_EXTLEN_MIN(a,b) min_t(xfs_extlen_t, (a), (b))
505#define XFS_EXTLEN_MAX(a,b) max_t(xfs_extlen_t, (a), (b))
506#define XFS_AGBLOCK_MIN(a,b) min_t(xfs_agblock_t, (a), (b))
507#define XFS_AGBLOCK_MAX(a,b) max_t(xfs_agblock_t, (a), (b))
508#define XFS_FILEOFF_MIN(a,b) min_t(xfs_fileoff_t, (a), (b))
509#define XFS_FILEOFF_MAX(a,b) max_t(xfs_fileoff_t, (a), (b))
510#define XFS_FILBLKS_MIN(a,b) min_t(xfs_filblks_t, (a), (b))
511#define XFS_FILBLKS_MAX(a,b) max_t(xfs_filblks_t, (a), (b))
512
513xfs_failaddr_t xfs_btree_sblock_v5hdr_verify(struct xfs_buf *bp);
514xfs_failaddr_t xfs_btree_sblock_verify(struct xfs_buf *bp,
515 unsigned int max_recs);
516xfs_failaddr_t xfs_btree_lblock_v5hdr_verify(struct xfs_buf *bp,
517 uint64_t owner);
518xfs_failaddr_t xfs_btree_lblock_verify(struct xfs_buf *bp,
519 unsigned int max_recs);
520
521unsigned int xfs_btree_compute_maxlevels(const unsigned int *limits,
522 unsigned long long records);
523unsigned long long xfs_btree_calc_size(const unsigned int *limits,
524 unsigned long long records);
525unsigned int xfs_btree_space_to_height(const unsigned int *limits,
526 unsigned long long blocks);
527
528/*
529 * Return codes for the query range iterator function are 0 to continue
530 * iterating, and non-zero to stop iterating. Any non-zero value will be
531 * passed up to the _query_range caller. The special value -ECANCELED can be
532 * used to stop iteration, because _query_range never generates that error
533 * code on its own.
534 */
535typedef int (*xfs_btree_query_range_fn)(struct xfs_btree_cur *cur,
536 const union xfs_btree_rec *rec, void *priv);
537
538int xfs_btree_query_range(struct xfs_btree_cur *cur,
539 const union xfs_btree_irec *low_rec,
540 const union xfs_btree_irec *high_rec,
541 xfs_btree_query_range_fn fn, void *priv);
542int xfs_btree_query_all(struct xfs_btree_cur *cur, xfs_btree_query_range_fn fn,
543 void *priv);
544
545typedef int (*xfs_btree_visit_blocks_fn)(struct xfs_btree_cur *cur, int level,
546 void *data);
547/* Visit record blocks. */
548#define XFS_BTREE_VISIT_RECORDS (1 << 0)
549/* Visit leaf blocks. */
550#define XFS_BTREE_VISIT_LEAVES (1 << 1)
551/* Visit all blocks. */
552#define XFS_BTREE_VISIT_ALL (XFS_BTREE_VISIT_RECORDS | \
553 XFS_BTREE_VISIT_LEAVES)
554int xfs_btree_visit_blocks(struct xfs_btree_cur *cur,
555 xfs_btree_visit_blocks_fn fn, unsigned int flags, void *data);
556
557int xfs_btree_count_blocks(struct xfs_btree_cur *cur, xfs_extlen_t *blocks);
558
559union xfs_btree_rec *xfs_btree_rec_addr(struct xfs_btree_cur *cur, int n,
560 struct xfs_btree_block *block);
561union xfs_btree_key *xfs_btree_key_addr(struct xfs_btree_cur *cur, int n,
562 struct xfs_btree_block *block);
563union xfs_btree_key *xfs_btree_high_key_addr(struct xfs_btree_cur *cur, int n,
564 struct xfs_btree_block *block);
565union xfs_btree_ptr *xfs_btree_ptr_addr(struct xfs_btree_cur *cur, int n,
566 struct xfs_btree_block *block);
567int xfs_btree_lookup_get_block(struct xfs_btree_cur *cur, int level,
568 const union xfs_btree_ptr *pp, struct xfs_btree_block **blkp);
569struct xfs_btree_block *xfs_btree_get_block(struct xfs_btree_cur *cur,
570 int level, struct xfs_buf **bpp);
571bool xfs_btree_ptr_is_null(struct xfs_btree_cur *cur,
572 const union xfs_btree_ptr *ptr);
573int64_t xfs_btree_diff_two_ptrs(struct xfs_btree_cur *cur,
574 const union xfs_btree_ptr *a,
575 const union xfs_btree_ptr *b);
576void xfs_btree_get_sibling(struct xfs_btree_cur *cur,
577 struct xfs_btree_block *block,
578 union xfs_btree_ptr *ptr, int lr);
579void xfs_btree_get_keys(struct xfs_btree_cur *cur,
580 struct xfs_btree_block *block, union xfs_btree_key *key);
581union xfs_btree_key *xfs_btree_high_key_from_key(struct xfs_btree_cur *cur,
582 union xfs_btree_key *key);
583typedef bool (*xfs_btree_key_gap_fn)(struct xfs_btree_cur *cur,
584 const union xfs_btree_key *key1,
585 const union xfs_btree_key *key2);
586
587int xfs_btree_has_records(struct xfs_btree_cur *cur,
588 const union xfs_btree_irec *low,
589 const union xfs_btree_irec *high,
590 const union xfs_btree_key *mask,
591 enum xbtree_recpacking *outcome);
592
593bool xfs_btree_has_more_records(struct xfs_btree_cur *cur);
594struct xfs_ifork *xfs_btree_ifork_ptr(struct xfs_btree_cur *cur);
595
596/* Key comparison helpers */
597static inline bool
598xfs_btree_keycmp_lt(
599 struct xfs_btree_cur *cur,
600 const union xfs_btree_key *key1,
601 const union xfs_btree_key *key2)
602{
603 return cur->bc_ops->diff_two_keys(cur, key1, key2, NULL) < 0;
604}
605
606static inline bool
607xfs_btree_keycmp_gt(
608 struct xfs_btree_cur *cur,
609 const union xfs_btree_key *key1,
610 const union xfs_btree_key *key2)
611{
612 return cur->bc_ops->diff_two_keys(cur, key1, key2, NULL) > 0;
613}
614
615static inline bool
616xfs_btree_keycmp_eq(
617 struct xfs_btree_cur *cur,
618 const union xfs_btree_key *key1,
619 const union xfs_btree_key *key2)
620{
621 return cur->bc_ops->diff_two_keys(cur, key1, key2, NULL) == 0;
622}
623
624static inline bool
625xfs_btree_keycmp_le(
626 struct xfs_btree_cur *cur,
627 const union xfs_btree_key *key1,
628 const union xfs_btree_key *key2)
629{
630 return !xfs_btree_keycmp_gt(cur, key1, key2);
631}
632
633static inline bool
634xfs_btree_keycmp_ge(
635 struct xfs_btree_cur *cur,
636 const union xfs_btree_key *key1,
637 const union xfs_btree_key *key2)
638{
639 return !xfs_btree_keycmp_lt(cur, key1, key2);
640}
641
642static inline bool
643xfs_btree_keycmp_ne(
644 struct xfs_btree_cur *cur,
645 const union xfs_btree_key *key1,
646 const union xfs_btree_key *key2)
647{
648 return !xfs_btree_keycmp_eq(cur, key1, key2);
649}
650
651/* Masked key comparison helpers */
652static inline bool
653xfs_btree_masked_keycmp_lt(
654 struct xfs_btree_cur *cur,
655 const union xfs_btree_key *key1,
656 const union xfs_btree_key *key2,
657 const union xfs_btree_key *mask)
658{
659 return cur->bc_ops->diff_two_keys(cur, key1, key2, mask) < 0;
660}
661
662static inline bool
663xfs_btree_masked_keycmp_gt(
664 struct xfs_btree_cur *cur,
665 const union xfs_btree_key *key1,
666 const union xfs_btree_key *key2,
667 const union xfs_btree_key *mask)
668{
669 return cur->bc_ops->diff_two_keys(cur, key1, key2, mask) > 0;
670}
671
672static inline bool
673xfs_btree_masked_keycmp_ge(
674 struct xfs_btree_cur *cur,
675 const union xfs_btree_key *key1,
676 const union xfs_btree_key *key2,
677 const union xfs_btree_key *mask)
678{
679 return !xfs_btree_masked_keycmp_lt(cur, key1, key2, mask);
680}
681
682/* Does this cursor point to the last block in the given level? */
683static inline bool
684xfs_btree_islastblock(
685 struct xfs_btree_cur *cur,
686 int level)
687{
688 struct xfs_btree_block *block;
689 struct xfs_buf *bp;
690
691 block = xfs_btree_get_block(cur, level, &bp);
692
693 if (cur->bc_flags & XFS_BTREE_LONG_PTRS)
694 return block->bb_u.l.bb_rightsib == cpu_to_be64(NULLFSBLOCK);
695 return block->bb_u.s.bb_rightsib == cpu_to_be32(NULLAGBLOCK);
696}
697
698void xfs_btree_set_ptr_null(struct xfs_btree_cur *cur,
699 union xfs_btree_ptr *ptr);
700int xfs_btree_get_buf_block(struct xfs_btree_cur *cur,
701 const union xfs_btree_ptr *ptr, struct xfs_btree_block **block,
702 struct xfs_buf **bpp);
703int xfs_btree_read_buf_block(struct xfs_btree_cur *cur,
704 const union xfs_btree_ptr *ptr, int flags,
705 struct xfs_btree_block **block, struct xfs_buf **bpp);
706void xfs_btree_set_sibling(struct xfs_btree_cur *cur,
707 struct xfs_btree_block *block, const union xfs_btree_ptr *ptr,
708 int lr);
709void xfs_btree_init_block_cur(struct xfs_btree_cur *cur,
710 struct xfs_buf *bp, int level, int numrecs);
711void xfs_btree_copy_ptrs(struct xfs_btree_cur *cur,
712 union xfs_btree_ptr *dst_ptr,
713 const union xfs_btree_ptr *src_ptr, int numptrs);
714void xfs_btree_copy_keys(struct xfs_btree_cur *cur,
715 union xfs_btree_key *dst_key,
716 const union xfs_btree_key *src_key, int numkeys);
717
718static inline struct xfs_btree_cur *
719xfs_btree_alloc_cursor(
720 struct xfs_mount *mp,
721 struct xfs_trans *tp,
722 xfs_btnum_t btnum,
723 uint8_t maxlevels,
724 struct kmem_cache *cache)
725{
726 struct xfs_btree_cur *cur;
727
728 cur = kmem_cache_zalloc(cache, GFP_NOFS | __GFP_NOFAIL);
729 cur->bc_tp = tp;
730 cur->bc_mp = mp;
731 cur->bc_btnum = btnum;
732 cur->bc_maxlevels = maxlevels;
733 cur->bc_cache = cache;
734
735 return cur;
736}
737
738int __init xfs_btree_init_cur_caches(void);
739void xfs_btree_destroy_cur_caches(void);
740
741int xfs_btree_goto_left_edge(struct xfs_btree_cur *cur);
742
743#endif /* __XFS_BTREE_H__ */
1/* SPDX-License-Identifier: GPL-2.0 */
2/*
3 * Copyright (c) 2000-2001,2005 Silicon Graphics, Inc.
4 * All Rights Reserved.
5 */
6#ifndef __XFS_BTREE_H__
7#define __XFS_BTREE_H__
8
9struct xfs_buf;
10struct xfs_inode;
11struct xfs_mount;
12struct xfs_trans;
13struct xfs_ifork;
14struct xfs_perag;
15
16/*
17 * Generic key, ptr and record wrapper structures.
18 *
19 * These are disk format structures, and are converted where necessary
20 * by the btree specific code that needs to interpret them.
21 */
22union xfs_btree_ptr {
23 __be32 s; /* short form ptr */
24 __be64 l; /* long form ptr */
25};
26
27/*
28 * The in-core btree key. Overlapping btrees actually store two keys
29 * per pointer, so we reserve enough memory to hold both. The __*bigkey
30 * items should never be accessed directly.
31 */
32union xfs_btree_key {
33 struct xfs_bmbt_key bmbt;
34 xfs_bmdr_key_t bmbr; /* bmbt root block */
35 xfs_alloc_key_t alloc;
36 struct xfs_inobt_key inobt;
37 struct xfs_rmap_key rmap;
38 struct xfs_rmap_key __rmap_bigkey[2];
39 struct xfs_refcount_key refc;
40};
41
42union xfs_btree_rec {
43 struct xfs_bmbt_rec bmbt;
44 xfs_bmdr_rec_t bmbr; /* bmbt root block */
45 struct xfs_alloc_rec alloc;
46 struct xfs_inobt_rec inobt;
47 struct xfs_rmap_rec rmap;
48 struct xfs_refcount_rec refc;
49};
50
51/*
52 * This nonsense is to make -wlint happy.
53 */
54#define XFS_LOOKUP_EQ ((xfs_lookup_t)XFS_LOOKUP_EQi)
55#define XFS_LOOKUP_LE ((xfs_lookup_t)XFS_LOOKUP_LEi)
56#define XFS_LOOKUP_GE ((xfs_lookup_t)XFS_LOOKUP_GEi)
57
58struct xfs_btree_ops;
59uint32_t xfs_btree_magic(struct xfs_mount *mp, const struct xfs_btree_ops *ops);
60
61/*
62 * For logging record fields.
63 */
64#define XFS_BB_MAGIC (1u << 0)
65#define XFS_BB_LEVEL (1u << 1)
66#define XFS_BB_NUMRECS (1u << 2)
67#define XFS_BB_LEFTSIB (1u << 3)
68#define XFS_BB_RIGHTSIB (1u << 4)
69#define XFS_BB_BLKNO (1u << 5)
70#define XFS_BB_LSN (1u << 6)
71#define XFS_BB_UUID (1u << 7)
72#define XFS_BB_OWNER (1u << 8)
73#define XFS_BB_NUM_BITS 5
74#define XFS_BB_ALL_BITS ((1u << XFS_BB_NUM_BITS) - 1)
75#define XFS_BB_NUM_BITS_CRC 9
76#define XFS_BB_ALL_BITS_CRC ((1u << XFS_BB_NUM_BITS_CRC) - 1)
77
78/*
79 * Generic stats interface
80 */
81#define XFS_BTREE_STATS_INC(cur, stat) \
82 XFS_STATS_INC_OFF((cur)->bc_mp, \
83 (cur)->bc_ops->statoff + __XBTS_ ## stat)
84#define XFS_BTREE_STATS_ADD(cur, stat, val) \
85 XFS_STATS_ADD_OFF((cur)->bc_mp, \
86 (cur)->bc_ops->statoff + __XBTS_ ## stat, val)
87
88enum xbtree_key_contig {
89 XBTREE_KEY_GAP = 0,
90 XBTREE_KEY_CONTIGUOUS,
91 XBTREE_KEY_OVERLAP,
92};
93
94/*
95 * Decide if these two numeric btree key fields are contiguous, overlapping,
96 * or if there's a gap between them. @x should be the field from the high
97 * key and @y should be the field from the low key.
98 */
99static inline enum xbtree_key_contig xbtree_key_contig(uint64_t x, uint64_t y)
100{
101 x++;
102 if (x < y)
103 return XBTREE_KEY_GAP;
104 if (x == y)
105 return XBTREE_KEY_CONTIGUOUS;
106 return XBTREE_KEY_OVERLAP;
107}
108
109#define XFS_BTREE_LONG_PTR_LEN (sizeof(__be64))
110#define XFS_BTREE_SHORT_PTR_LEN (sizeof(__be32))
111
112enum xfs_btree_type {
113 XFS_BTREE_TYPE_AG,
114 XFS_BTREE_TYPE_INODE,
115 XFS_BTREE_TYPE_MEM,
116};
117
118struct xfs_btree_ops {
119 const char *name;
120
121 /* Type of btree - AG-rooted or inode-rooted */
122 enum xfs_btree_type type;
123
124 /* XFS_BTGEO_* flags that determine the geometry of the btree */
125 unsigned int geom_flags;
126
127 /* size of the key, pointer, and record structures */
128 size_t key_len;
129 size_t ptr_len;
130 size_t rec_len;
131
132 /* LRU refcount to set on each btree buffer created */
133 unsigned int lru_refs;
134
135 /* offset of btree stats array */
136 unsigned int statoff;
137
138 /* sick mask for health reporting (only for XFS_BTREE_TYPE_AG) */
139 unsigned int sick_mask;
140
141 /* cursor operations */
142 struct xfs_btree_cur *(*dup_cursor)(struct xfs_btree_cur *);
143 void (*update_cursor)(struct xfs_btree_cur *src,
144 struct xfs_btree_cur *dst);
145
146 /* update btree root pointer */
147 void (*set_root)(struct xfs_btree_cur *cur,
148 const union xfs_btree_ptr *nptr, int level_change);
149
150 /* block allocation / freeing */
151 int (*alloc_block)(struct xfs_btree_cur *cur,
152 const union xfs_btree_ptr *start_bno,
153 union xfs_btree_ptr *new_bno,
154 int *stat);
155 int (*free_block)(struct xfs_btree_cur *cur, struct xfs_buf *bp);
156
157 /* records in block/level */
158 int (*get_minrecs)(struct xfs_btree_cur *cur, int level);
159 int (*get_maxrecs)(struct xfs_btree_cur *cur, int level);
160
161 /* records on disk. Matter for the root in inode case. */
162 int (*get_dmaxrecs)(struct xfs_btree_cur *cur, int level);
163
164 /* init values of btree structures */
165 void (*init_key_from_rec)(union xfs_btree_key *key,
166 const union xfs_btree_rec *rec);
167 void (*init_rec_from_cur)(struct xfs_btree_cur *cur,
168 union xfs_btree_rec *rec);
169 void (*init_ptr_from_cur)(struct xfs_btree_cur *cur,
170 union xfs_btree_ptr *ptr);
171 void (*init_high_key_from_rec)(union xfs_btree_key *key,
172 const union xfs_btree_rec *rec);
173
174 /* difference between key value and cursor value */
175 int64_t (*key_diff)(struct xfs_btree_cur *cur,
176 const union xfs_btree_key *key);
177
178 /*
179 * Difference between key2 and key1 -- positive if key1 > key2,
180 * negative if key1 < key2, and zero if equal. If the @mask parameter
181 * is non NULL, each key field to be used in the comparison must
182 * contain a nonzero value.
183 */
184 int64_t (*diff_two_keys)(struct xfs_btree_cur *cur,
185 const union xfs_btree_key *key1,
186 const union xfs_btree_key *key2,
187 const union xfs_btree_key *mask);
188
189 const struct xfs_buf_ops *buf_ops;
190
191 /* check that k1 is lower than k2 */
192 int (*keys_inorder)(struct xfs_btree_cur *cur,
193 const union xfs_btree_key *k1,
194 const union xfs_btree_key *k2);
195
196 /* check that r1 is lower than r2 */
197 int (*recs_inorder)(struct xfs_btree_cur *cur,
198 const union xfs_btree_rec *r1,
199 const union xfs_btree_rec *r2);
200
201 /*
202 * Are these two btree keys immediately adjacent?
203 *
204 * Given two btree keys @key1 and @key2, decide if it is impossible for
205 * there to be a third btree key K satisfying the relationship
206 * @key1 < K < @key2. To determine if two btree records are
207 * immediately adjacent, @key1 should be the high key of the first
208 * record and @key2 should be the low key of the second record.
209 * If the @mask parameter is non NULL, each key field to be used in the
210 * comparison must contain a nonzero value.
211 */
212 enum xbtree_key_contig (*keys_contiguous)(struct xfs_btree_cur *cur,
213 const union xfs_btree_key *key1,
214 const union xfs_btree_key *key2,
215 const union xfs_btree_key *mask);
216};
217
218/* btree geometry flags */
219#define XFS_BTGEO_OVERLAPPING (1U << 0) /* overlapping intervals */
220
221
222union xfs_btree_irec {
223 struct xfs_alloc_rec_incore a;
224 struct xfs_bmbt_irec b;
225 struct xfs_inobt_rec_incore i;
226 struct xfs_rmap_irec r;
227 struct xfs_refcount_irec rc;
228};
229
230struct xfs_btree_level {
231 /* buffer pointer */
232 struct xfs_buf *bp;
233
234 /* key/record number */
235 uint16_t ptr;
236
237 /* readahead info */
238#define XFS_BTCUR_LEFTRA (1 << 0) /* left sibling has been read-ahead */
239#define XFS_BTCUR_RIGHTRA (1 << 1) /* right sibling has been read-ahead */
240 uint16_t ra;
241};
242
243/*
244 * Btree cursor structure.
245 * This collects all information needed by the btree code in one place.
246 */
247struct xfs_btree_cur
248{
249 struct xfs_trans *bc_tp; /* transaction we're in, if any */
250 struct xfs_mount *bc_mp; /* file system mount struct */
251 const struct xfs_btree_ops *bc_ops;
252 struct kmem_cache *bc_cache; /* cursor cache */
253 unsigned int bc_flags; /* btree features - below */
254 union xfs_btree_irec bc_rec; /* current insert/search record value */
255 uint8_t bc_nlevels; /* number of levels in the tree */
256 uint8_t bc_maxlevels; /* maximum levels for this btree type */
257 struct xfs_group *bc_group;
258
259 /* per-type information */
260 union {
261 struct {
262 struct xfs_inode *ip;
263 short forksize;
264 char whichfork;
265 struct xbtree_ifakeroot *ifake; /* for staging cursor */
266 } bc_ino;
267 struct {
268 struct xfs_buf *agbp;
269 struct xbtree_afakeroot *afake; /* for staging cursor */
270 } bc_ag;
271 struct {
272 struct xfbtree *xfbtree;
273 } bc_mem;
274 };
275
276 /* per-format private data */
277 union {
278 struct {
279 int allocated;
280 } bc_bmap; /* bmapbt */
281 struct {
282 unsigned int nr_ops; /* # record updates */
283 unsigned int shape_changes; /* # of extent splits */
284 } bc_refc; /* refcountbt */
285 };
286
287 /* Must be at the end of the struct! */
288 struct xfs_btree_level bc_levels[];
289};
290
291/*
292 * Compute the size of a btree cursor that can handle a btree of a given
293 * height. The bc_levels array handles node and leaf blocks, so its size
294 * is exactly nlevels.
295 */
296static inline size_t
297xfs_btree_cur_sizeof(unsigned int nlevels)
298{
299 return struct_size_t(struct xfs_btree_cur, bc_levels, nlevels);
300}
301
302/* cursor state flags */
303/*
304 * The root of this btree is a fakeroot structure so that we can stage a btree
305 * rebuild without leaving it accessible via primary metadata. The ops struct
306 * is dynamically allocated and must be freed when the cursor is deleted.
307 */
308#define XFS_BTREE_STAGING (1U << 0)
309
310/* We are converting a delalloc reservation (only for bmbt btrees) */
311#define XFS_BTREE_BMBT_WASDEL (1U << 1)
312
313/* For extent swap, ignore owner check in verifier (only for bmbt btrees) */
314#define XFS_BTREE_BMBT_INVALID_OWNER (1U << 2)
315
316/* Cursor is active (only for allocbt btrees) */
317#define XFS_BTREE_ALLOCBT_ACTIVE (1U << 3)
318
319#define XFS_BTREE_NOERROR 0
320#define XFS_BTREE_ERROR 1
321
322/*
323 * Convert from buffer to btree block header.
324 */
325#define XFS_BUF_TO_BLOCK(bp) ((struct xfs_btree_block *)((bp)->b_addr))
326
327xfs_failaddr_t __xfs_btree_check_block(struct xfs_btree_cur *cur,
328 struct xfs_btree_block *block, int level, struct xfs_buf *bp);
329int __xfs_btree_check_ptr(struct xfs_btree_cur *cur,
330 const union xfs_btree_ptr *ptr, int index, int level);
331
332/*
333 * Check that block header is ok.
334 */
335int
336xfs_btree_check_block(
337 struct xfs_btree_cur *cur, /* btree cursor */
338 struct xfs_btree_block *block, /* generic btree block pointer */
339 int level, /* level of the btree block */
340 struct xfs_buf *bp); /* buffer containing block, if any */
341
342/*
343 * Delete the btree cursor.
344 */
345void
346xfs_btree_del_cursor(
347 struct xfs_btree_cur *cur, /* btree cursor */
348 int error); /* del because of error */
349
350/*
351 * Duplicate the btree cursor.
352 * Allocate a new one, copy the record, re-get the buffers.
353 */
354int /* error */
355xfs_btree_dup_cursor(
356 struct xfs_btree_cur *cur, /* input cursor */
357 struct xfs_btree_cur **ncur);/* output cursor */
358
359/*
360 * Compute first and last byte offsets for the fields given.
361 * Interprets the offsets table, which contains struct field offsets.
362 */
363void
364xfs_btree_offsets(
365 uint32_t fields, /* bitmask of fields */
366 const short *offsets,/* table of field offsets */
367 int nbits, /* number of bits to inspect */
368 int *first, /* output: first byte offset */
369 int *last); /* output: last byte offset */
370
371/*
372 * Initialise a new btree block header
373 */
374void xfs_btree_init_buf(struct xfs_mount *mp, struct xfs_buf *bp,
375 const struct xfs_btree_ops *ops, __u16 level, __u16 numrecs,
376 __u64 owner);
377void xfs_btree_init_block(struct xfs_mount *mp,
378 struct xfs_btree_block *buf, const struct xfs_btree_ops *ops,
379 __u16 level, __u16 numrecs, __u64 owner);
380
381/*
382 * Common btree core entry points.
383 */
384int xfs_btree_increment(struct xfs_btree_cur *, int, int *);
385int xfs_btree_decrement(struct xfs_btree_cur *, int, int *);
386int xfs_btree_lookup(struct xfs_btree_cur *, xfs_lookup_t, int *);
387int xfs_btree_update(struct xfs_btree_cur *, union xfs_btree_rec *);
388int xfs_btree_new_iroot(struct xfs_btree_cur *, int *, int *);
389int xfs_btree_insert(struct xfs_btree_cur *, int *);
390int xfs_btree_delete(struct xfs_btree_cur *, int *);
391int xfs_btree_get_rec(struct xfs_btree_cur *, union xfs_btree_rec **, int *);
392int xfs_btree_change_owner(struct xfs_btree_cur *cur, uint64_t new_owner,
393 struct list_head *buffer_list);
394
395/*
396 * btree block CRC helpers
397 */
398void xfs_btree_fsblock_calc_crc(struct xfs_buf *);
399bool xfs_btree_fsblock_verify_crc(struct xfs_buf *);
400void xfs_btree_agblock_calc_crc(struct xfs_buf *);
401bool xfs_btree_agblock_verify_crc(struct xfs_buf *);
402
403/*
404 * Internal btree helpers also used by xfs_bmap.c.
405 */
406void xfs_btree_log_block(struct xfs_btree_cur *, struct xfs_buf *, uint32_t);
407void xfs_btree_log_recs(struct xfs_btree_cur *, struct xfs_buf *, int, int);
408
409/*
410 * Helpers.
411 */
412static inline int xfs_btree_get_numrecs(const struct xfs_btree_block *block)
413{
414 return be16_to_cpu(block->bb_numrecs);
415}
416
417static inline void xfs_btree_set_numrecs(struct xfs_btree_block *block,
418 uint16_t numrecs)
419{
420 block->bb_numrecs = cpu_to_be16(numrecs);
421}
422
423static inline int xfs_btree_get_level(const struct xfs_btree_block *block)
424{
425 return be16_to_cpu(block->bb_level);
426}
427
428
429/*
430 * Min and max functions for extlen, agblock, fileoff, and filblks types.
431 */
432#define XFS_EXTLEN_MIN(a,b) min_t(xfs_extlen_t, (a), (b))
433#define XFS_EXTLEN_MAX(a,b) max_t(xfs_extlen_t, (a), (b))
434#define XFS_AGBLOCK_MIN(a,b) min_t(xfs_agblock_t, (a), (b))
435#define XFS_AGBLOCK_MAX(a,b) max_t(xfs_agblock_t, (a), (b))
436#define XFS_FILEOFF_MIN(a,b) min_t(xfs_fileoff_t, (a), (b))
437#define XFS_FILEOFF_MAX(a,b) max_t(xfs_fileoff_t, (a), (b))
438#define XFS_FILBLKS_MIN(a,b) min_t(xfs_filblks_t, (a), (b))
439#define XFS_FILBLKS_MAX(a,b) max_t(xfs_filblks_t, (a), (b))
440
441xfs_failaddr_t xfs_btree_agblock_v5hdr_verify(struct xfs_buf *bp);
442xfs_failaddr_t xfs_btree_agblock_verify(struct xfs_buf *bp,
443 unsigned int max_recs);
444xfs_failaddr_t xfs_btree_fsblock_v5hdr_verify(struct xfs_buf *bp,
445 uint64_t owner);
446xfs_failaddr_t xfs_btree_fsblock_verify(struct xfs_buf *bp,
447 unsigned int max_recs);
448xfs_failaddr_t xfs_btree_memblock_verify(struct xfs_buf *bp,
449 unsigned int max_recs);
450
451unsigned int xfs_btree_compute_maxlevels(const unsigned int *limits,
452 unsigned long long records);
453unsigned long long xfs_btree_calc_size(const unsigned int *limits,
454 unsigned long long records);
455unsigned int xfs_btree_space_to_height(const unsigned int *limits,
456 unsigned long long blocks);
457
458/*
459 * Return codes for the query range iterator function are 0 to continue
460 * iterating, and non-zero to stop iterating. Any non-zero value will be
461 * passed up to the _query_range caller. The special value -ECANCELED can be
462 * used to stop iteration, because _query_range never generates that error
463 * code on its own.
464 */
465typedef int (*xfs_btree_query_range_fn)(struct xfs_btree_cur *cur,
466 const union xfs_btree_rec *rec, void *priv);
467
468int xfs_btree_query_range(struct xfs_btree_cur *cur,
469 const union xfs_btree_irec *low_rec,
470 const union xfs_btree_irec *high_rec,
471 xfs_btree_query_range_fn fn, void *priv);
472int xfs_btree_query_all(struct xfs_btree_cur *cur, xfs_btree_query_range_fn fn,
473 void *priv);
474
475typedef int (*xfs_btree_visit_blocks_fn)(struct xfs_btree_cur *cur, int level,
476 void *data);
477/* Visit record blocks. */
478#define XFS_BTREE_VISIT_RECORDS (1 << 0)
479/* Visit leaf blocks. */
480#define XFS_BTREE_VISIT_LEAVES (1 << 1)
481/* Visit all blocks. */
482#define XFS_BTREE_VISIT_ALL (XFS_BTREE_VISIT_RECORDS | \
483 XFS_BTREE_VISIT_LEAVES)
484int xfs_btree_visit_blocks(struct xfs_btree_cur *cur,
485 xfs_btree_visit_blocks_fn fn, unsigned int flags, void *data);
486
487int xfs_btree_count_blocks(struct xfs_btree_cur *cur, xfs_filblks_t *blocks);
488
489union xfs_btree_rec *xfs_btree_rec_addr(struct xfs_btree_cur *cur, int n,
490 struct xfs_btree_block *block);
491union xfs_btree_key *xfs_btree_key_addr(struct xfs_btree_cur *cur, int n,
492 struct xfs_btree_block *block);
493union xfs_btree_key *xfs_btree_high_key_addr(struct xfs_btree_cur *cur, int n,
494 struct xfs_btree_block *block);
495union xfs_btree_ptr *xfs_btree_ptr_addr(struct xfs_btree_cur *cur, int n,
496 struct xfs_btree_block *block);
497int xfs_btree_lookup_get_block(struct xfs_btree_cur *cur, int level,
498 const union xfs_btree_ptr *pp, struct xfs_btree_block **blkp);
499struct xfs_btree_block *xfs_btree_get_block(struct xfs_btree_cur *cur,
500 int level, struct xfs_buf **bpp);
501bool xfs_btree_ptr_is_null(struct xfs_btree_cur *cur,
502 const union xfs_btree_ptr *ptr);
503int64_t xfs_btree_diff_two_ptrs(struct xfs_btree_cur *cur,
504 const union xfs_btree_ptr *a,
505 const union xfs_btree_ptr *b);
506void xfs_btree_get_sibling(struct xfs_btree_cur *cur,
507 struct xfs_btree_block *block,
508 union xfs_btree_ptr *ptr, int lr);
509void xfs_btree_get_keys(struct xfs_btree_cur *cur,
510 struct xfs_btree_block *block, union xfs_btree_key *key);
511union xfs_btree_key *xfs_btree_high_key_from_key(struct xfs_btree_cur *cur,
512 union xfs_btree_key *key);
513typedef bool (*xfs_btree_key_gap_fn)(struct xfs_btree_cur *cur,
514 const union xfs_btree_key *key1,
515 const union xfs_btree_key *key2);
516
517int xfs_btree_has_records(struct xfs_btree_cur *cur,
518 const union xfs_btree_irec *low,
519 const union xfs_btree_irec *high,
520 const union xfs_btree_key *mask,
521 enum xbtree_recpacking *outcome);
522
523bool xfs_btree_has_more_records(struct xfs_btree_cur *cur);
524struct xfs_ifork *xfs_btree_ifork_ptr(struct xfs_btree_cur *cur);
525
526/* Key comparison helpers */
527static inline bool
528xfs_btree_keycmp_lt(
529 struct xfs_btree_cur *cur,
530 const union xfs_btree_key *key1,
531 const union xfs_btree_key *key2)
532{
533 return cur->bc_ops->diff_two_keys(cur, key1, key2, NULL) < 0;
534}
535
536static inline bool
537xfs_btree_keycmp_gt(
538 struct xfs_btree_cur *cur,
539 const union xfs_btree_key *key1,
540 const union xfs_btree_key *key2)
541{
542 return cur->bc_ops->diff_two_keys(cur, key1, key2, NULL) > 0;
543}
544
545static inline bool
546xfs_btree_keycmp_eq(
547 struct xfs_btree_cur *cur,
548 const union xfs_btree_key *key1,
549 const union xfs_btree_key *key2)
550{
551 return cur->bc_ops->diff_two_keys(cur, key1, key2, NULL) == 0;
552}
553
554static inline bool
555xfs_btree_keycmp_le(
556 struct xfs_btree_cur *cur,
557 const union xfs_btree_key *key1,
558 const union xfs_btree_key *key2)
559{
560 return !xfs_btree_keycmp_gt(cur, key1, key2);
561}
562
563static inline bool
564xfs_btree_keycmp_ge(
565 struct xfs_btree_cur *cur,
566 const union xfs_btree_key *key1,
567 const union xfs_btree_key *key2)
568{
569 return !xfs_btree_keycmp_lt(cur, key1, key2);
570}
571
572static inline bool
573xfs_btree_keycmp_ne(
574 struct xfs_btree_cur *cur,
575 const union xfs_btree_key *key1,
576 const union xfs_btree_key *key2)
577{
578 return !xfs_btree_keycmp_eq(cur, key1, key2);
579}
580
581/* Masked key comparison helpers */
582static inline bool
583xfs_btree_masked_keycmp_lt(
584 struct xfs_btree_cur *cur,
585 const union xfs_btree_key *key1,
586 const union xfs_btree_key *key2,
587 const union xfs_btree_key *mask)
588{
589 return cur->bc_ops->diff_two_keys(cur, key1, key2, mask) < 0;
590}
591
592static inline bool
593xfs_btree_masked_keycmp_gt(
594 struct xfs_btree_cur *cur,
595 const union xfs_btree_key *key1,
596 const union xfs_btree_key *key2,
597 const union xfs_btree_key *mask)
598{
599 return cur->bc_ops->diff_two_keys(cur, key1, key2, mask) > 0;
600}
601
602static inline bool
603xfs_btree_masked_keycmp_ge(
604 struct xfs_btree_cur *cur,
605 const union xfs_btree_key *key1,
606 const union xfs_btree_key *key2,
607 const union xfs_btree_key *mask)
608{
609 return !xfs_btree_masked_keycmp_lt(cur, key1, key2, mask);
610}
611
612/* Does this cursor point to the last block in the given level? */
613static inline bool
614xfs_btree_islastblock(
615 struct xfs_btree_cur *cur,
616 int level)
617{
618 struct xfs_btree_block *block;
619 struct xfs_buf *bp;
620
621 block = xfs_btree_get_block(cur, level, &bp);
622
623 if (cur->bc_ops->ptr_len == XFS_BTREE_LONG_PTR_LEN)
624 return block->bb_u.l.bb_rightsib == cpu_to_be64(NULLFSBLOCK);
625 return block->bb_u.s.bb_rightsib == cpu_to_be32(NULLAGBLOCK);
626}
627
628void xfs_btree_set_ptr_null(struct xfs_btree_cur *cur,
629 union xfs_btree_ptr *ptr);
630int xfs_btree_get_buf_block(struct xfs_btree_cur *cur,
631 const union xfs_btree_ptr *ptr, struct xfs_btree_block **block,
632 struct xfs_buf **bpp);
633int xfs_btree_read_buf_block(struct xfs_btree_cur *cur,
634 const union xfs_btree_ptr *ptr, int flags,
635 struct xfs_btree_block **block, struct xfs_buf **bpp);
636void xfs_btree_set_sibling(struct xfs_btree_cur *cur,
637 struct xfs_btree_block *block, const union xfs_btree_ptr *ptr,
638 int lr);
639void xfs_btree_init_block_cur(struct xfs_btree_cur *cur,
640 struct xfs_buf *bp, int level, int numrecs);
641void xfs_btree_copy_ptrs(struct xfs_btree_cur *cur,
642 union xfs_btree_ptr *dst_ptr,
643 const union xfs_btree_ptr *src_ptr, int numptrs);
644void xfs_btree_copy_keys(struct xfs_btree_cur *cur,
645 union xfs_btree_key *dst_key,
646 const union xfs_btree_key *src_key, int numkeys);
647void xfs_btree_init_ptr_from_cur(struct xfs_btree_cur *cur,
648 union xfs_btree_ptr *ptr);
649
650static inline struct xfs_btree_cur *
651xfs_btree_alloc_cursor(
652 struct xfs_mount *mp,
653 struct xfs_trans *tp,
654 const struct xfs_btree_ops *ops,
655 uint8_t maxlevels,
656 struct kmem_cache *cache)
657{
658 struct xfs_btree_cur *cur;
659
660 ASSERT(ops->ptr_len == XFS_BTREE_LONG_PTR_LEN ||
661 ops->ptr_len == XFS_BTREE_SHORT_PTR_LEN);
662
663 /* BMBT allocations can come through from non-transactional context. */
664 cur = kmem_cache_zalloc(cache,
665 GFP_KERNEL | __GFP_NOLOCKDEP | __GFP_NOFAIL);
666 cur->bc_ops = ops;
667 cur->bc_tp = tp;
668 cur->bc_mp = mp;
669 cur->bc_maxlevels = maxlevels;
670 cur->bc_cache = cache;
671
672 return cur;
673}
674
675int __init xfs_btree_init_cur_caches(void);
676void xfs_btree_destroy_cur_caches(void);
677
678int xfs_btree_goto_left_edge(struct xfs_btree_cur *cur);
679
680/* Does this level of the cursor point to the inode root (and not a block)? */
681static inline bool
682xfs_btree_at_iroot(
683 const struct xfs_btree_cur *cur,
684 int level)
685{
686 return cur->bc_ops->type == XFS_BTREE_TYPE_INODE &&
687 level == cur->bc_nlevels - 1;
688}
689
690#endif /* __XFS_BTREE_H__ */