Skip to content

Commit 4c61728

Browse files
anakryikoAlexei Starovoitov
authored andcommitted
bpf: prepare reg_set_min_max for second set of registers
Similarly to is_branch_taken()-related refactorings, start preparing reg_set_min_max() to handle more generic case of two non-const registers. Start with renaming arguments to accommodate later addition of second register as an input argument. Signed-off-by: Andrii Nakryiko <[email protected]> Link: https://lore.kernel.org/r/[email protected] Signed-off-by: Alexei Starovoitov <[email protected]>
1 parent 761a9e5 commit 4c61728

File tree

1 file changed

+40
-40
lines changed

1 file changed

+40
-40
lines changed

kernel/bpf/verifier.c

Lines changed: 40 additions & 40 deletions
Original file line numberDiff line numberDiff line change
@@ -14384,25 +14384,25 @@ static int is_branch_taken(struct bpf_reg_state *reg1, struct bpf_reg_state *reg
1438414384
* simply doing a BPF_K check.
1438514385
* In JEQ/JNE cases we also adjust the var_off values.
1438614386
*/
14387-
static void reg_set_min_max(struct bpf_reg_state *true_reg,
14388-
struct bpf_reg_state *false_reg,
14387+
static void reg_set_min_max(struct bpf_reg_state *true_reg1,
14388+
struct bpf_reg_state *false_reg1,
1438914389
u64 uval, u32 uval32,
1439014390
u8 opcode, bool is_jmp32)
1439114391
{
14392-
struct tnum false_32off = tnum_subreg(false_reg->var_off);
14393-
struct tnum false_64off = false_reg->var_off;
14394-
struct tnum true_32off = tnum_subreg(true_reg->var_off);
14395-
struct tnum true_64off = true_reg->var_off;
14392+
struct tnum false_32off = tnum_subreg(false_reg1->var_off);
14393+
struct tnum false_64off = false_reg1->var_off;
14394+
struct tnum true_32off = tnum_subreg(true_reg1->var_off);
14395+
struct tnum true_64off = true_reg1->var_off;
1439614396
s64 sval = (s64)uval;
1439714397
s32 sval32 = (s32)uval32;
1439814398

1439914399
/* If the dst_reg is a pointer, we can't learn anything about its
1440014400
* variable offset from the compare (unless src_reg were a pointer into
1440114401
* the same object, but we don't bother with that.
14402-
* Since false_reg and true_reg have the same type by construction, we
14402+
* Since false_reg1 and true_reg1 have the same type by construction, we
1440314403
* only need to check one of them for pointerness.
1440414404
*/
14405-
if (__is_pointer_value(false, false_reg))
14405+
if (__is_pointer_value(false, false_reg1))
1440614406
return;
1440714407

1440814408
switch (opcode) {
@@ -14417,20 +14417,20 @@ static void reg_set_min_max(struct bpf_reg_state *true_reg,
1441714417
*/
1441814418
case BPF_JEQ:
1441914419
if (is_jmp32) {
14420-
__mark_reg32_known(true_reg, uval32);
14421-
true_32off = tnum_subreg(true_reg->var_off);
14420+
__mark_reg32_known(true_reg1, uval32);
14421+
true_32off = tnum_subreg(true_reg1->var_off);
1442214422
} else {
14423-
___mark_reg_known(true_reg, uval);
14424-
true_64off = true_reg->var_off;
14423+
___mark_reg_known(true_reg1, uval);
14424+
true_64off = true_reg1->var_off;
1442514425
}
1442614426
break;
1442714427
case BPF_JNE:
1442814428
if (is_jmp32) {
14429-
__mark_reg32_known(false_reg, uval32);
14430-
false_32off = tnum_subreg(false_reg->var_off);
14429+
__mark_reg32_known(false_reg1, uval32);
14430+
false_32off = tnum_subreg(false_reg1->var_off);
1443114431
} else {
14432-
___mark_reg_known(false_reg, uval);
14433-
false_64off = false_reg->var_off;
14432+
___mark_reg_known(false_reg1, uval);
14433+
false_64off = false_reg1->var_off;
1443414434
}
1443514435
break;
1443614436
case BPF_JSET:
@@ -14453,16 +14453,16 @@ static void reg_set_min_max(struct bpf_reg_state *true_reg,
1445314453
u32 false_umax = opcode == BPF_JGT ? uval32 : uval32 - 1;
1445414454
u32 true_umin = opcode == BPF_JGT ? uval32 + 1 : uval32;
1445514455

14456-
false_reg->u32_max_value = min(false_reg->u32_max_value,
14456+
false_reg1->u32_max_value = min(false_reg1->u32_max_value,
1445714457
false_umax);
14458-
true_reg->u32_min_value = max(true_reg->u32_min_value,
14458+
true_reg1->u32_min_value = max(true_reg1->u32_min_value,
1445914459
true_umin);
1446014460
} else {
1446114461
u64 false_umax = opcode == BPF_JGT ? uval : uval - 1;
1446214462
u64 true_umin = opcode == BPF_JGT ? uval + 1 : uval;
1446314463

14464-
false_reg->umax_value = min(false_reg->umax_value, false_umax);
14465-
true_reg->umin_value = max(true_reg->umin_value, true_umin);
14464+
false_reg1->umax_value = min(false_reg1->umax_value, false_umax);
14465+
true_reg1->umin_value = max(true_reg1->umin_value, true_umin);
1446614466
}
1446714467
break;
1446814468
}
@@ -14473,14 +14473,14 @@ static void reg_set_min_max(struct bpf_reg_state *true_reg,
1447314473
s32 false_smax = opcode == BPF_JSGT ? sval32 : sval32 - 1;
1447414474
s32 true_smin = opcode == BPF_JSGT ? sval32 + 1 : sval32;
1447514475

14476-
false_reg->s32_max_value = min(false_reg->s32_max_value, false_smax);
14477-
true_reg->s32_min_value = max(true_reg->s32_min_value, true_smin);
14476+
false_reg1->s32_max_value = min(false_reg1->s32_max_value, false_smax);
14477+
true_reg1->s32_min_value = max(true_reg1->s32_min_value, true_smin);
1447814478
} else {
1447914479
s64 false_smax = opcode == BPF_JSGT ? sval : sval - 1;
1448014480
s64 true_smin = opcode == BPF_JSGT ? sval + 1 : sval;
1448114481

14482-
false_reg->smax_value = min(false_reg->smax_value, false_smax);
14483-
true_reg->smin_value = max(true_reg->smin_value, true_smin);
14482+
false_reg1->smax_value = min(false_reg1->smax_value, false_smax);
14483+
true_reg1->smin_value = max(true_reg1->smin_value, true_smin);
1448414484
}
1448514485
break;
1448614486
}
@@ -14491,16 +14491,16 @@ static void reg_set_min_max(struct bpf_reg_state *true_reg,
1449114491
u32 false_umin = opcode == BPF_JLT ? uval32 : uval32 + 1;
1449214492
u32 true_umax = opcode == BPF_JLT ? uval32 - 1 : uval32;
1449314493

14494-
false_reg->u32_min_value = max(false_reg->u32_min_value,
14494+
false_reg1->u32_min_value = max(false_reg1->u32_min_value,
1449514495
false_umin);
14496-
true_reg->u32_max_value = min(true_reg->u32_max_value,
14496+
true_reg1->u32_max_value = min(true_reg1->u32_max_value,
1449714497
true_umax);
1449814498
} else {
1449914499
u64 false_umin = opcode == BPF_JLT ? uval : uval + 1;
1450014500
u64 true_umax = opcode == BPF_JLT ? uval - 1 : uval;
1450114501

14502-
false_reg->umin_value = max(false_reg->umin_value, false_umin);
14503-
true_reg->umax_value = min(true_reg->umax_value, true_umax);
14502+
false_reg1->umin_value = max(false_reg1->umin_value, false_umin);
14503+
true_reg1->umax_value = min(true_reg1->umax_value, true_umax);
1450414504
}
1450514505
break;
1450614506
}
@@ -14511,14 +14511,14 @@ static void reg_set_min_max(struct bpf_reg_state *true_reg,
1451114511
s32 false_smin = opcode == BPF_JSLT ? sval32 : sval32 + 1;
1451214512
s32 true_smax = opcode == BPF_JSLT ? sval32 - 1 : sval32;
1451314513

14514-
false_reg->s32_min_value = max(false_reg->s32_min_value, false_smin);
14515-
true_reg->s32_max_value = min(true_reg->s32_max_value, true_smax);
14514+
false_reg1->s32_min_value = max(false_reg1->s32_min_value, false_smin);
14515+
true_reg1->s32_max_value = min(true_reg1->s32_max_value, true_smax);
1451614516
} else {
1451714517
s64 false_smin = opcode == BPF_JSLT ? sval : sval + 1;
1451814518
s64 true_smax = opcode == BPF_JSLT ? sval - 1 : sval;
1451914519

14520-
false_reg->smin_value = max(false_reg->smin_value, false_smin);
14521-
true_reg->smax_value = min(true_reg->smax_value, true_smax);
14520+
false_reg1->smin_value = max(false_reg1->smin_value, false_smin);
14521+
true_reg1->smax_value = min(true_reg1->smax_value, true_smax);
1452214522
}
1452314523
break;
1452414524
}
@@ -14527,17 +14527,17 @@ static void reg_set_min_max(struct bpf_reg_state *true_reg,
1452714527
}
1452814528

1452914529
if (is_jmp32) {
14530-
false_reg->var_off = tnum_or(tnum_clear_subreg(false_64off),
14530+
false_reg1->var_off = tnum_or(tnum_clear_subreg(false_64off),
1453114531
tnum_subreg(false_32off));
14532-
true_reg->var_off = tnum_or(tnum_clear_subreg(true_64off),
14532+
true_reg1->var_off = tnum_or(tnum_clear_subreg(true_64off),
1453314533
tnum_subreg(true_32off));
14534-
reg_bounds_sync(false_reg);
14535-
reg_bounds_sync(true_reg);
14534+
reg_bounds_sync(false_reg1);
14535+
reg_bounds_sync(true_reg1);
1453614536
} else {
14537-
false_reg->var_off = false_64off;
14538-
true_reg->var_off = true_64off;
14539-
reg_bounds_sync(false_reg);
14540-
reg_bounds_sync(true_reg);
14537+
false_reg1->var_off = false_64off;
14538+
true_reg1->var_off = true_64off;
14539+
reg_bounds_sync(false_reg1);
14540+
reg_bounds_sync(true_reg1);
1454114541
}
1454214542
}
1454314543

0 commit comments

Comments
 (0)