More consistent use of TREE_* macros in AVL comparators

Where is it appropriate and obvious, use TREE_CMP(), TREE_ISIGN() and
TREE_PCMP() instead or direct comparisons. It can make the code a lot
smaller, less error prone, and easier to read.

Sponsored-by: TrueNAS
Reviewed-by: Alexander Motin <alexander.motin@TrueNAS.com>
Reviewed-by: Brian Behlendorf <behlendorf1@llnl.gov>
Signed-off-by: Rob Norris <rob.norris@truenas.com>
Closes #18259
This commit is contained in:
Rob Norris
2026-03-04 04:08:23 +11:00
committed by GitHub
parent 0f90a797dd
commit 1e2c94a043
19 changed files with 88 additions and 199 deletions
+2 -2
View File
@@ -90,8 +90,8 @@ static int
kcf_mech_hash_compar(const void *lhs, const void *rhs)
{
const kcf_mech_entry_t *l = lhs, *r = rhs;
int cmp = strncmp(l->me_name, r->me_name, CRYPTO_MAX_MECH_NAME);
return ((0 < cmp) - (cmp < 0));
return (TREE_ISIGN(strncmp(l->me_name, r->me_name,
CRYPTO_MAX_MECH_NAME)));
}
void
+1 -9
View File
@@ -1213,15 +1213,7 @@ static int
acevals_compare(const void *va, const void *vb)
{
const acevals_t *a = va, *b = vb;
if (a->key == b->key)
return (0);
if (a->key > b->key)
return (1);
else
return (-1);
return (TREE_CMP(a->key, b->key));
}
/*
+5 -19
View File
@@ -251,16 +251,7 @@ snapentry_compare_by_name(const void *a, const void *b)
{
const zfs_snapentry_t *se_a = a;
const zfs_snapentry_t *se_b = b;
int ret;
ret = strcmp(se_a->se_name, se_b->se_name);
if (ret < 0)
return (-1);
else if (ret > 0)
return (1);
else
return (0);
return (TREE_ISIGN(strcmp(se_a->se_name, se_b->se_name)));
}
/*
@@ -272,15 +263,10 @@ snapentry_compare_by_objsetid(const void *a, const void *b)
const zfs_snapentry_t *se_a = a;
const zfs_snapentry_t *se_b = b;
if (se_a->se_spa != se_b->se_spa)
return ((ulong_t)se_a->se_spa < (ulong_t)se_b->se_spa ? -1 : 1);
if (se_a->se_objsetid < se_b->se_objsetid)
return (-1);
else if (se_a->se_objsetid > se_b->se_objsetid)
return (1);
else
return (0);
int cmp = TREE_PCMP(se_a->se_spa, se_b->se_spa);
if (cmp != 0)
return (cmp);
return (TREE_CMP(se_a->se_objsetid, se_b->se_objsetid));
}
/*
+9 -13
View File
@@ -176,11 +176,7 @@ objnode_compare(const void *o1, const void *o2)
{
const struct objnode *obj1 = o1;
const struct objnode *obj2 = o2;
if (obj1->obj < obj2->obj)
return (-1);
if (obj1->obj > obj2->obj)
return (1);
return (0);
return (TREE_CMP(obj1->obj, obj2->obj));
}
@@ -425,11 +421,11 @@ redact_node_compare_start(const void *arg1, const void *arg2)
if (rr2->eos_marker)
return (-1);
int cmp = redact_range_compare(rr1->start_object, rr1->start_blkid,
rr1->datablksz, rr2->start_object, rr2->start_blkid,
rr2->datablksz);
int cmp = redact_range_compare(
rr1->start_object, rr1->start_blkid, rr1->datablksz,
rr2->start_object, rr2->start_blkid, rr2->datablksz);
if (cmp == 0)
cmp = (rn1->thread_num < rn2->thread_num ? -1 : 1);
cmp = TREE_CMP(rn1->thread_num, rn2->thread_num);
return (cmp);
}
@@ -451,11 +447,11 @@ redact_node_compare_end(const void *arg1, const void *arg2)
if (srr2->eos_marker)
return (-1);
int cmp = redact_range_compare(srr1->end_object, srr1->end_blkid,
srr1->datablksz, srr2->end_object, srr2->end_blkid,
srr2->datablksz);
int cmp = redact_range_compare(
srr1->end_object, srr1->end_blkid, srr1->datablksz,
srr2->end_object, srr2->end_blkid, srr2->datablksz);
if (cmp == 0)
cmp = (rn1->thread_num < rn2->thread_num ? -1 : 1);
cmp = TREE_CMP(rn1->thread_num, rn2->thread_num);
return (cmp);
}
+1 -2
View File
@@ -837,8 +837,7 @@ dsl_bookmark_compare(const void *l, const void *r)
(rdbn->dbn_phys.zbm_flags & ZBM_FLAG_HAS_FBN));
if (likely(cmp))
return (cmp);
cmp = strcmp(ldbn->dbn_name, rdbn->dbn_name);
return (TREE_ISIGN(cmp));
return (TREE_ISIGN(strcmp(ldbn->dbn_name, rdbn->dbn_name)));
}
/*
+3 -18
View File
@@ -259,12 +259,7 @@ spa_crypto_key_compare(const void *a, const void *b)
{
const dsl_crypto_key_t *dcka = a;
const dsl_crypto_key_t *dckb = b;
if (dcka->dck_obj < dckb->dck_obj)
return (-1);
if (dcka->dck_obj > dckb->dck_obj)
return (1);
return (0);
return (TREE_CMP(dcka->dck_obj, dckb->dck_obj));
}
/*
@@ -306,12 +301,7 @@ spa_key_mapping_compare(const void *a, const void *b)
{
const dsl_key_mapping_t *kma = a;
const dsl_key_mapping_t *kmb = b;
if (kma->km_dsobj < kmb->km_dsobj)
return (-1);
if (kma->km_dsobj > kmb->km_dsobj)
return (1);
return (0);
return (TREE_CMP(kma->km_dsobj, kmb->km_dsobj));
}
static int
@@ -319,12 +309,7 @@ spa_wkey_compare(const void *a, const void *b)
{
const dsl_wrapping_key_t *wka = a;
const dsl_wrapping_key_t *wkb = b;
if (wka->wk_ddobj < wkb->wk_ddobj)
return (-1);
if (wka->wk_ddobj > wkb->wk_ddobj)
return (1);
return (0);
return (TREE_CMP(wka->wk_ddobj, wkb->wk_ddobj));
}
void
+6 -8
View File
@@ -1001,16 +1001,14 @@ livelist_compare(const void *larg, const void *rarg)
const blkptr_t *r = &((livelist_entry_t *)rarg)->le_bp;
/* Sort them according to dva[0] */
uint64_t l_dva0_vdev = DVA_GET_VDEV(&l->blk_dva[0]);
uint64_t r_dva0_vdev = DVA_GET_VDEV(&r->blk_dva[0]);
if (l_dva0_vdev != r_dva0_vdev)
return (TREE_CMP(l_dva0_vdev, r_dva0_vdev));
int cmp = TREE_CMP(DVA_GET_VDEV(&l->blk_dva[0]),
DVA_GET_VDEV(&r->blk_dva[0]));
if (cmp != 0)
return (cmp);
/* if vdevs are equal, sort by offsets. */
uint64_t l_dva0_offset = DVA_GET_OFFSET(&l->blk_dva[0]);
uint64_t r_dva0_offset = DVA_GET_OFFSET(&r->blk_dva[0]);
return (TREE_CMP(l_dva0_offset, r_dva0_offset));
return (TREE_CMP(DVA_GET_OFFSET(&l->blk_dva[0]),
DVA_GET_OFFSET(&r->blk_dva[0])));
}
struct livelist_iter_arg {
+1 -5
View File
@@ -395,11 +395,7 @@ perm_set_compare(const void *arg1, const void *arg2)
{
const perm_set_t *node1 = (const perm_set_t *)arg1;
const perm_set_t *node2 = (const perm_set_t *)arg2;
int val;
val = strcmp(node1->p_setname, node2->p_setname);
return (TREE_ISIGN(val));
return (TREE_ISIGN(strcmp(node1->p_setname, node2->p_setname)));
}
/*
+1 -6
View File
@@ -1473,12 +1473,7 @@ static int
scan_ds_queue_compare(const void *a, const void *b)
{
const scan_ds_t *sds_a = a, *sds_b = b;
if (sds_a->sds_dsobj < sds_b->sds_dsobj)
return (-1);
if (sds_a->sds_dsobj == sds_b->sds_dsobj)
return (0);
return (1);
return (TREE_CMP(sds_a->sds_dsobj, sds_b->sds_dsobj));
}
static void
+6 -4
View File
@@ -57,10 +57,12 @@ zfs_refcount_compare(const void *x1, const void *x2)
const reference_t *r1 = (const reference_t *)x1;
const reference_t *r2 = (const reference_t *)x2;
int cmp1 = TREE_CMP(r1->ref_holder, r2->ref_holder);
int cmp2 = TREE_CMP(r1->ref_number, r2->ref_number);
int cmp = cmp1 ? cmp1 : cmp2;
return ((cmp || r1->ref_search) ? cmp : TREE_PCMP(r1, r2));
int cmp = TREE_CMP(r1->ref_holder, r2->ref_holder);
if (cmp == 0)
cmp = TREE_CMP(r1->ref_number, r2->ref_number);
if (cmp | r1->ref_search)
return (cmp);
return (TREE_PCMP(r1, r2));
}
void
+1 -4
View File
@@ -2616,11 +2616,8 @@ spa_name_compare(const void *a1, const void *a2)
{
const spa_t *s1 = a1;
const spa_t *s2 = a2;
int s;
s = strcmp(s1->spa_name, s2->spa_name);
return (TREE_ISIGN(s));
return (TREE_ISIGN(strcmp(s1->spa_name, s2->spa_name)));
}
void
+3 -3
View File
@@ -226,10 +226,10 @@ vdev_queue_to_compare(const void *x1, const void *x2)
const zio_t *z1 = (const zio_t *)x1;
const zio_t *z2 = (const zio_t *)x2;
int tcmp = TREE_CMP(z1->io_timestamp >> VDQ_T_SHIFT,
int cmp = TREE_CMP(z1->io_timestamp >> VDQ_T_SHIFT,
z2->io_timestamp >> VDQ_T_SHIFT);
int ocmp = TREE_CMP(z1->io_offset, z2->io_offset);
int cmp = tcmp ? tcmp : ocmp;
if (cmp == 0)
cmp = TREE_CMP(z1->io_offset, z2->io_offset);
if (likely(cmp | (z1->io_queue_state == ZIO_QS_NONE)))
return (cmp);
+2 -5
View File
@@ -84,11 +84,8 @@ domain_compare(const void *arg1, const void *arg2)
{
const fuid_domain_t *node1 = (const fuid_domain_t *)arg1;
const fuid_domain_t *node2 = (const fuid_domain_t *)arg2;
int val;
val = strcmp(node1->f_ksid->kd_name, node2->f_ksid->kd_name);
return (TREE_ISIGN(val));
return (TREE_ISIGN(strcmp(node1->f_ksid->kd_name,
node2->f_ksid->kd_name)));
}
void
+15 -22
View File
@@ -910,33 +910,26 @@ zio_bookmark_compare(const void *x1, const void *x2)
{
const zio_t *z1 = x1;
const zio_t *z2 = x2;
const zbookmark_phys_t *zb1 = &z1->io_bookmark;
const zbookmark_phys_t *zb2 = &z2->io_bookmark;
if (z1->io_bookmark.zb_objset < z2->io_bookmark.zb_objset)
return (-1);
if (z1->io_bookmark.zb_objset > z2->io_bookmark.zb_objset)
return (1);
int cmp = TREE_CMP(zb1->zb_objset, zb2->zb_objset);
if (cmp != 0)
return (cmp);
if (z1->io_bookmark.zb_object < z2->io_bookmark.zb_object)
return (-1);
if (z1->io_bookmark.zb_object > z2->io_bookmark.zb_object)
return (1);
cmp = TREE_CMP(zb1->zb_object, zb2->zb_object);
if (cmp != 0)
return (cmp);
if (z1->io_bookmark.zb_level < z2->io_bookmark.zb_level)
return (-1);
if (z1->io_bookmark.zb_level > z2->io_bookmark.zb_level)
return (1);
cmp = TREE_CMP(zb1->zb_level, zb2->zb_level);
if (cmp != 0)
return (cmp);
if (z1->io_bookmark.zb_blkid < z2->io_bookmark.zb_blkid)
return (-1);
if (z1->io_bookmark.zb_blkid > z2->io_bookmark.zb_blkid)
return (1);
cmp = TREE_CMP(zb1->zb_blkid, zb2->zb_blkid);
if (cmp != 0)
return (cmp);
if (z1 < z2)
return (-1);
if (z1 > z2)
return (1);
return (0);
return (TREE_PCMP(z1, z2));
}
/*