diff mbox series

[RFC] bpf: s390: add JIT support for multi-function programs

Message ID 20190826182036.17456-1-yauheni.kaliuta@redhat.com
State RFC
Delegated to: BPF Maintainers
Headers show
Series [RFC] bpf: s390: add JIT support for multi-function programs | expand

Commit Message

Yauheni Kaliuta Aug. 26, 2019, 6:20 p.m. UTC
This adds support for bpf-to-bpf function calls in the s390 JIT
compiler. The JIT compiler converts the bpf call instructions to
native branch instructions. After a round of the usual passes, the
start addresses of the JITed images for the callee functions are
known. Finally, to fixup the branch target addresses, we need to
perform an extra pass.

Because of the address range in which JITed images are allocated on
s390, the offsets of the start addresses of these images from
__bpf_call_base are as large as 64 bits. So, for a function call,
the imm field of the instruction cannot be used to determine the
callee's address. Use bpf_jit_get_func_addr() helper instead.

The patch borrows a lot from:

8c11ea5ce13d bpf, arm64: fix getting subprog addr from aux for calls
e2c95a61656d bpf, ppc64: generalize fetching subprog into bpf_jit_get_func_addr
8484ce8306f9 bpf: powerpc64: add JIT support for multi-function programs

(including the commit message).

test_verifier (5.3-rc6):

without patch:
Summary: 1501 PASSED, 0 SKIPPED, 47 FAILED

with patch:
Summary: 1540 PASSED, 0 SKIPPED, 8 FAILED

Signed-off-by: Yauheni Kaliuta <yauheni.kaliuta@redhat.com>
---
 arch/s390/net/bpf_jit_comp.c | 63 +++++++++++++++++++++++++++++-------
 1 file changed, 52 insertions(+), 11 deletions(-)

Comments

Ilya Leoshkevich Aug. 27, 2019, 1:21 p.m. UTC | #1
> Am 26.08.2019 um 20:20 schrieb Yauheni Kaliuta <yauheni.kaliuta@redhat.com>:
> 
> test_verifier (5.3-rc6):
> 
> without patch:
> Summary: 1501 PASSED, 0 SKIPPED, 47 FAILED
> 
> with patch:
> Summary: 1540 PASSED, 0 SKIPPED, 8 FAILED

Are you per chance running with a testsuite patch like this one?

--- a/tools/testing/selftests/bpf/test_verifier.c
+++ b/tools/testing/selftests/bpf/test_verifier.c
@@ -846,7 +846,7 @@ static int do_prog_test_run(int fd_prog, bool unpriv, uint32_t expected_val,
 				tmp, &size_tmp, &retval, NULL);
 	if (unpriv)
 		set_admin(false);
-	if (err && errno != 524/*ENOTSUPP*/ && errno != EPERM) {
+	if (err && errno != EPERM) {
 		printf("Unexpected bpf_prog_test_run error ");
 		return err;
 	}

Without it, all the failures appear to be masked for me.
Ilya Leoshkevich Aug. 27, 2019, 1:46 p.m. UTC | #2
> Am 27.08.2019 um 15:21 schrieb Ilya Leoshkevich <iii@linux.ibm.com>:
> 
>> Am 26.08.2019 um 20:20 schrieb Yauheni Kaliuta <yauheni.kaliuta@redhat.com>:
>> 
>> test_verifier (5.3-rc6):
>> 
>> without patch:
>> Summary: 1501 PASSED, 0 SKIPPED, 47 FAILED
>> 
>> with patch:
>> Summary: 1540 PASSED, 0 SKIPPED, 8 FAILED
> 
> Are you per chance running with a testsuite patch like this one?
> 
> --- a/tools/testing/selftests/bpf/test_verifier.c
> +++ b/tools/testing/selftests/bpf/test_verifier.c
> @@ -846,7 +846,7 @@ static int do_prog_test_run(int fd_prog, bool unpriv, uint32_t expected_val,
> 				tmp, &size_tmp, &retval, NULL);
> 	if (unpriv)
> 		set_admin(false);
> -	if (err && errno != 524/*ENOTSUPP*/ && errno != EPERM) {
> +	if (err && errno != EPERM) {
> 		printf("Unexpected bpf_prog_test_run error ");
> 		return err;
> 	}
> 
> Without it, all the failures appear to be masked for me.

Hmm, I'm sorry, I thought about it a bit more, and the patch I posted
above doesn't make any sense, because the failures you fixed are during
load, and not run time.

Now I think you are using CONFIG_BPF_JIT_ALWAYS_ON for your testing,
is that right? If yes, it would be nice to mention this in the commit
message.
Yauheni Kaliuta Aug. 27, 2019, 2:21 p.m. UTC | #3
Hi, Ilya!

>>>>> On Tue, 27 Aug 2019 15:46:43 +0200, Ilya Leoshkevich  wrote:

 >> Am 27.08.2019 um 15:21 schrieb Ilya Leoshkevich <iii@linux.ibm.com>:
 >> 
 >>> Am 26.08.2019 um 20:20 schrieb Yauheni Kaliuta <yauheni.kaliuta@redhat.com>:
 >>> 
 >>> test_verifier (5.3-rc6):
 >>> 
 >>> without patch:
 >>> Summary: 1501 PASSED, 0 SKIPPED, 47 FAILED
 >>> 
 >>> with patch:
 >>> Summary: 1540 PASSED, 0 SKIPPED, 8 FAILED
 >> 
 >> Are you per chance running with a testsuite patch like this one?
 >> 
 >> --- a/tools/testing/selftests/bpf/test_verifier.c
 >> +++ b/tools/testing/selftests/bpf/test_verifier.c
 >> @@ -846,7 +846,7 @@ static int do_prog_test_run(int fd_prog, bool unpriv, uint32_t expected_val,
 >> tmp, &size_tmp, &retval, NULL);
 >> if (unpriv)
 >> set_admin(false);
 >> -	if (err && errno != 524/*ENOTSUPP*/ && errno != EPERM) {
 >> +	if (err && errno != EPERM) {
 >> printf("Unexpected bpf_prog_test_run error ");
 >> return err;
 >> }
 >> 
 >> Without it, all the failures appear to be masked for me.

 > Hmm, I'm sorry, I thought about it a bit more, and the patch I
 > posted above doesn't make any sense, because the failures you
 > fixed are during load, and not run time.

 > Now I think you are using CONFIG_BPF_JIT_ALWAYS_ON for your
 > testing, is that right? If yes, it would be nice to mention

Right.

 > this in the commit message.

Sure. Should I post non-RFC v2 or wait for some more comments?
Yauheni Kaliuta Aug. 27, 2019, 2:25 p.m. UTC | #4
Hi, Ilya!

>>>>> On Tue, 27 Aug 2019 15:21:30 +0200, Ilya Leoshkevich  wrote:

 >> Am 26.08.2019 um 20:20 schrieb Yauheni Kaliuta <yauheni.kaliuta@redhat.com>:
 >> 
 >> test_verifier (5.3-rc6):
 >> 
 >> without patch:
 >> Summary: 1501 PASSED, 0 SKIPPED, 47 FAILED
 >> 
 >> with patch:
 >> Summary: 1540 PASSED, 0 SKIPPED, 8 FAILED

 > Are you per chance running with a testsuite patch like this one?

 > --- a/tools/testing/selftests/bpf/test_verifier.c
 > +++ b/tools/testing/selftests/bpf/test_verifier.c
 > @@ -846,7 +846,7 @@ static int do_prog_test_run(int fd_prog, bool unpriv, uint32_t expected_val,
 >  				tmp, &size_tmp, &retval, NULL);
 >  	if (unpriv)
 >  		set_admin(false);
 > -	if (err && errno != 524/*ENOTSUPP*/ && errno != EPERM) {
 > +	if (err && errno != EPERM) {
 >  		printf("Unexpected bpf_prog_test_run error ");
 >  		return err;
 >  	}

 > Without it, all the failures appear to be masked for me.

BTW, I have several failures because of low BPF_SIZE_MAX. If I
increase it, some tests pass (#585/p ld_abs: vlan + abs, test 1),
but some crash (#587/p ld_abs: jump around ld_abs, haven't
found the reason yet).

Have you observed anything like that?
Ilya Leoshkevich Aug. 27, 2019, 2:37 p.m. UTC | #5
> Am 27.08.2019 um 16:21 schrieb Yauheni Kaliuta <yauheni.kaliuta@redhat.com>:
> 
> Hi, Ilya!
> 
>>>>>> On Tue, 27 Aug 2019 15:46:43 +0200, Ilya Leoshkevich  wrote:
> 
>>> Am 27.08.2019 um 15:21 schrieb Ilya Leoshkevich <iii@linux.ibm.com>:
>>> 
>>>> Am 26.08.2019 um 20:20 schrieb Yauheni Kaliuta <yauheni.kaliuta@redhat.com>:
>>>> 
>>>> test_verifier (5.3-rc6):
>>>> 
>>>> without patch:
>>>> Summary: 1501 PASSED, 0 SKIPPED, 47 FAILED
>>>> 
>>>> with patch:
>>>> Summary: 1540 PASSED, 0 SKIPPED, 8 FAILED
>>> 
>>> Are you per chance running with a testsuite patch like this one?
>>> 
>>> --- a/tools/testing/selftests/bpf/test_verifier.c
>>> +++ b/tools/testing/selftests/bpf/test_verifier.c
>>> @@ -846,7 +846,7 @@ static int do_prog_test_run(int fd_prog, bool unpriv, uint32_t expected_val,
>>> tmp, &size_tmp, &retval, NULL);
>>> if (unpriv)
>>> set_admin(false);
>>> -	if (err && errno != 524/*ENOTSUPP*/ && errno != EPERM) {
>>> +	if (err && errno != EPERM) {
>>> printf("Unexpected bpf_prog_test_run error ");
>>> return err;
>>> }
>>> 
>>> Without it, all the failures appear to be masked for me.
> 
>> Hmm, I'm sorry, I thought about it a bit more, and the patch I
>> posted above doesn't make any sense, because the failures you
>> fixed are during load, and not run time.
> 
>> Now I think you are using CONFIG_BPF_JIT_ALWAYS_ON for your
>> testing, is that right? If yes, it would be nice to mention
> 
> Right.
> 
>> this in the commit message.
> 
> Sure. Should I post non-RFC v2 or wait for some more comments?

So far I only spotted a minor issue:

+		if (ret < 0)
+			return ret;

Right now bpf_jit_insn returns 0 or -1, but bpf_jit_get_func_addr
returns 0 or -errno. This does not affect anything in the end, but just
to be uniform, maybe return -1 here or -EINVAL in the default: branch?


I don't see any other obvious problems with the patch, but I'd like to
take some time to understand how exactly some parts of it work before
acking it. So I think it's fine to post a non-RFC version.
Ilya Leoshkevich Aug. 27, 2019, 2:46 p.m. UTC | #6
> Am 27.08.2019 um 16:25 schrieb Yauheni Kaliuta <yauheni.kaliuta@redhat.com>:
> 
> Hi, Ilya!
> 
>>>>>> On Tue, 27 Aug 2019 15:21:30 +0200, Ilya Leoshkevich  wrote:
> 
>>> Am 26.08.2019 um 20:20 schrieb Yauheni Kaliuta <yauheni.kaliuta@redhat.com>:
>>> 
>>> test_verifier (5.3-rc6):
>>> 
>>> without patch:
>>> Summary: 1501 PASSED, 0 SKIPPED, 47 FAILED
>>> 
>>> with patch:
>>> Summary: 1540 PASSED, 0 SKIPPED, 8 FAILED
> 
>> Are you per chance running with a testsuite patch like this one?
> 
>> --- a/tools/testing/selftests/bpf/test_verifier.c
>> +++ b/tools/testing/selftests/bpf/test_verifier.c
>> @@ -846,7 +846,7 @@ static int do_prog_test_run(int fd_prog, bool unpriv, uint32_t expected_val,
>> 				tmp, &size_tmp, &retval, NULL);
>> 	if (unpriv)
>> 		set_admin(false);
>> -	if (err && errno != 524/*ENOTSUPP*/ && errno != EPERM) {
>> +	if (err && errno != EPERM) {
>> 		printf("Unexpected bpf_prog_test_run error ");
>> 		return err;
>> 	}
> 
>> Without it, all the failures appear to be masked for me.
> 
> BTW, I have several failures because of low BPF_SIZE_MAX. If I
> increase it, some tests pass (#585/p ld_abs: vlan + abs, test 1),
> but some crash (#587/p ld_abs: jump around ld_abs, haven't
> found the reason yet).
> 
> Have you observed anything like that?

Yes, this is because right now JIT generates clrj and friends,
which can jump only by +-32k. Improving this is actually my next task
(after fixing more or less "obvious" test suite problems).
Yauheni Kaliuta Aug. 27, 2019, 2:48 p.m. UTC | #7
Hi, Ilya!

>>>>> On Tue, 27 Aug 2019 16:37:04 +0200, Ilya Leoshkevich  wrote:

 >> Am 27.08.2019 um 16:21 schrieb Yauheni Kaliuta <yauheni.kaliuta@redhat.com>:
 >> 
 >> Hi, Ilya!
 >> 
 >>>>>>> On Tue, 27 Aug 2019 15:46:43 +0200, Ilya Leoshkevich  wrote:
 >> 
 >>>> Am 27.08.2019 um 15:21 schrieb Ilya Leoshkevich <iii@linux.ibm.com>:
 >>>> 
 >>>>> Am 26.08.2019 um 20:20 schrieb Yauheni Kaliuta <yauheni.kaliuta@redhat.com>:
 >>>>> 
 >>>>> test_verifier (5.3-rc6):
 >>>>> 
 >>>>> without patch:
 >>>>> Summary: 1501 PASSED, 0 SKIPPED, 47 FAILED
 >>>>> 
 >>>>> with patch:
 >>>>> Summary: 1540 PASSED, 0 SKIPPED, 8 FAILED
 >>>> 
 >>>> Are you per chance running with a testsuite patch like this one?
 >>>> 
 >>>> --- a/tools/testing/selftests/bpf/test_verifier.c
 >>>> +++ b/tools/testing/selftests/bpf/test_verifier.c
 >>>> @@ -846,7 +846,7 @@ static int do_prog_test_run(int fd_prog, bool
 >>>> unpriv, uint32_t expected_val,
 >>>> tmp, &size_tmp, &retval, NULL);
 >>>> if (unpriv)
 >>>> set_admin(false);
 >>>> -	if (err && errno != 524/*ENOTSUPP*/ && errno != EPERM) {
 >>>> +	if (err && errno != EPERM) {
 >>>> printf("Unexpected bpf_prog_test_run error ");
 >>>> return err;
 >>>> }
 >>>> 
 >>>> Without it, all the failures appear to be masked for me.
 >> 
 >>> Hmm, I'm sorry, I thought about it a bit more, and the patch I
 >>> posted above doesn't make any sense, because the failures you
 >>> fixed are during load, and not run time.
 >> 
 >>> Now I think you are using CONFIG_BPF_JIT_ALWAYS_ON for your
 >>> testing, is that right? If yes, it would be nice to mention
 >> 
 >> Right.
 >> 
 >>> this in the commit message.
 >> 
 >> Sure. Should I post non-RFC v2 or wait for some more comments?

 > So far I only spotted a minor issue:

 > +		if (ret < 0)
 > +			return ret;

 > Right now bpf_jit_insn returns 0 or -1, but
 > bpf_jit_get_func_addr returns 0 or -errno. This does not
 > affect anything in the end, but just to be uniform, maybe
 > return -1 here or -EINVAL in the default: branch?

ok. I choose "return -1" since changing default to -EINVAL sounds
as unrelated change to the patch.

 > I don't see any other obvious problems with the patch, but I'd
 > like to take some time to understand how exactly some parts of
 > it work before acking it. So I think it's fine to post a
 > non-RFC version.

Good, thanks!
Yauheni Kaliuta Aug. 27, 2019, 3:05 p.m. UTC | #8
Hi, Ilya!

>>>>> On Tue, 27 Aug 2019 16:46:46 +0200, Ilya Leoshkevich  wrote:

 >> Am 27.08.2019 um 16:25 schrieb Yauheni Kaliuta <yauheni.kaliuta@redhat.com>:
 >> 
 >> Hi, Ilya!
 >> 
 >>>>>>> On Tue, 27 Aug 2019 15:21:30 +0200, Ilya Leoshkevich  wrote:
 >> 
 >>>> Am 26.08.2019 um 20:20 schrieb Yauheni Kaliuta <yauheni.kaliuta@redhat.com>:
 >>>> 
 >>>> test_verifier (5.3-rc6):
 >>>> 
 >>>> without patch:
 >>>> Summary: 1501 PASSED, 0 SKIPPED, 47 FAILED
 >>>> 
 >>>> with patch:
 >>>> Summary: 1540 PASSED, 0 SKIPPED, 8 FAILED
 >> 
 >>> Are you per chance running with a testsuite patch like this one?
 >> 
 >>> --- a/tools/testing/selftests/bpf/test_verifier.c
 >>> +++ b/tools/testing/selftests/bpf/test_verifier.c
 >>> @@ -846,7 +846,7 @@ static int do_prog_test_run(int fd_prog, bool unpriv, uint32_t expected_val,
 >>> tmp, &size_tmp, &retval, NULL);
 >>> if (unpriv)
 >>> set_admin(false);
 >>> -	if (err && errno != 524/*ENOTSUPP*/ && errno != EPERM) {
 >>> +	if (err && errno != EPERM) {
 >>> printf("Unexpected bpf_prog_test_run error ");
 >>> return err;
 >>> }
 >> 
 >>> Without it, all the failures appear to be masked for me.
 >> 
 >> BTW, I have several failures because of low BPF_SIZE_MAX. If I
 >> increase it, some tests pass (#585/p ld_abs: vlan + abs, test 1),
 >> but some crash (#587/p ld_abs: jump around ld_abs, haven't
 >> found the reason yet).
 >> 
 >> Have you observed anything like that?

 > Yes, this is because right now JIT generates clrj and friends,
 > which can jump only by +-32k. Improving this is actually my
 > next task (after fixing more or less "obvious" test suite
 > problems).

ah, great. Sorry for the noise.
diff mbox series

Patch

diff --git a/arch/s390/net/bpf_jit_comp.c b/arch/s390/net/bpf_jit_comp.c
index e636728ab452..39329c20dcbb 100644
--- a/arch/s390/net/bpf_jit_comp.c
+++ b/arch/s390/net/bpf_jit_comp.c
@@ -502,7 +502,8 @@  static void bpf_jit_epilogue(struct bpf_jit *jit, u32 stack_depth)
  * NOTE: Use noinline because for gcov (-fprofile-arcs) gcc allocates a lot of
  * stack space for the large switch statement.
  */
-static noinline int bpf_jit_insn(struct bpf_jit *jit, struct bpf_prog *fp, int i)
+static noinline int bpf_jit_insn(struct bpf_jit *jit, struct bpf_prog *fp,
+				 int i, bool extra_pass)
 {
 	struct bpf_insn *insn = &fp->insnsi[i];
 	int jmp_off, last, insn_count = 1;
@@ -1011,10 +1012,14 @@  static noinline int bpf_jit_insn(struct bpf_jit *jit, struct bpf_prog *fp, int i
 	 */
 	case BPF_JMP | BPF_CALL:
 	{
-		/*
-		 * b0 = (__bpf_call_base + imm)(b1, b2, b3, b4, b5)
-		 */
-		const u64 func = (u64)__bpf_call_base + imm;
+		u64 func;
+		bool func_addr_fixed;
+		int ret;
+
+		ret = bpf_jit_get_func_addr(fp, insn, extra_pass,
+					    &func, &func_addr_fixed);
+		if (ret < 0)
+			return ret;
 
 		REG_SET_SEEN(BPF_REG_5);
 		jit->seen |= SEEN_FUNC;
@@ -1281,7 +1286,7 @@  static noinline int bpf_jit_insn(struct bpf_jit *jit, struct bpf_prog *fp, int i
 /*
  * Compile eBPF program into s390x code
  */
-static int bpf_jit_prog(struct bpf_jit *jit, struct bpf_prog *fp)
+static int bpf_jit_prog(struct bpf_jit *jit, struct bpf_prog *fp, bool extra_pass)
 {
 	int i, insn_count;
 
@@ -1290,7 +1295,7 @@  static int bpf_jit_prog(struct bpf_jit *jit, struct bpf_prog *fp)
 
 	bpf_jit_prologue(jit, fp->aux->stack_depth);
 	for (i = 0; i < fp->len; i += insn_count) {
-		insn_count = bpf_jit_insn(jit, fp, i);
+		insn_count = bpf_jit_insn(jit, fp, i, extra_pass);
 		if (insn_count < 0)
 			return -1;
 		/* Next instruction address */
@@ -1309,6 +1314,12 @@  bool bpf_jit_needs_zext(void)
 	return true;
 }
 
+
+struct s390_jit_data {
+	struct bpf_binary_header *header;
+	struct bpf_jit ctx;
+};
+
 /*
  * Compile eBPF program "fp"
  */
@@ -1316,7 +1327,9 @@  struct bpf_prog *bpf_int_jit_compile(struct bpf_prog *fp)
 {
 	struct bpf_prog *tmp, *orig_fp = fp;
 	struct bpf_binary_header *header;
+	struct s390_jit_data *jit_data;
 	bool tmp_blinded = false;
+	bool extra_pass = false;
 	struct bpf_jit jit;
 	int pass;
 
@@ -1335,6 +1348,22 @@  struct bpf_prog *bpf_int_jit_compile(struct bpf_prog *fp)
 		fp = tmp;
 	}
 
+	jit_data = fp->aux->jit_data;
+	if (!jit_data) {
+		jit_data = kzalloc(sizeof(*jit_data), GFP_KERNEL);
+		if (!jit_data) {
+			fp = orig_fp;
+			goto out;
+		}
+		fp->aux->jit_data = jit_data;
+	}
+	if (jit_data->ctx.addrs) {
+		jit = jit_data->ctx;
+		header = jit_data->header;
+		extra_pass = true;
+		goto skip_init_ctx;
+	}
+
 	memset(&jit, 0, sizeof(jit));
 	jit.addrs = kcalloc(fp->len + 1, sizeof(*jit.addrs), GFP_KERNEL);
 	if (jit.addrs == NULL) {
@@ -1347,7 +1376,7 @@  struct bpf_prog *bpf_int_jit_compile(struct bpf_prog *fp)
 	 *   - 3:   Calculate program size and addrs arrray
 	 */
 	for (pass = 1; pass <= 3; pass++) {
-		if (bpf_jit_prog(&jit, fp)) {
+		if (bpf_jit_prog(&jit, fp, extra_pass)) {
 			fp = orig_fp;
 			goto free_addrs;
 		}
@@ -1359,12 +1388,14 @@  struct bpf_prog *bpf_int_jit_compile(struct bpf_prog *fp)
 		fp = orig_fp;
 		goto free_addrs;
 	}
+
 	header = bpf_jit_binary_alloc(jit.size, &jit.prg_buf, 2, jit_fill_hole);
 	if (!header) {
 		fp = orig_fp;
 		goto free_addrs;
 	}
-	if (bpf_jit_prog(&jit, fp)) {
+skip_init_ctx:
+	if (bpf_jit_prog(&jit, fp, extra_pass)) {
 		bpf_jit_binary_free(header);
 		fp = orig_fp;
 		goto free_addrs;
@@ -1373,12 +1404,22 @@  struct bpf_prog *bpf_int_jit_compile(struct bpf_prog *fp)
 		bpf_jit_dump(fp->len, jit.size, pass, jit.prg_buf);
 		print_fn_code(jit.prg_buf, jit.size_prg);
 	}
-	bpf_jit_binary_lock_ro(header);
+	if (!fp->is_func || extra_pass) {
+		bpf_jit_binary_lock_ro(header);
+	} else {
+		jit_data->header = header;
+		jit_data->ctx = jit;
+	}
 	fp->bpf_func = (void *) jit.prg_buf;
 	fp->jited = 1;
 	fp->jited_len = jit.size;
+
+	if (!fp->is_func || extra_pass) {
 free_addrs:
-	kfree(jit.addrs);
+		kfree(jit.addrs);
+		kfree(jit_data);
+		fp->aux->jit_data = NULL;
+	}
 out:
 	if (tmp_blinded)
 		bpf_jit_prog_release_other(fp, fp == orig_fp ?