Source-Changes-HG archive

[Date Prev][Date Next][Thread Prev][Thread Next][Date Index][Thread Index][Old Index]

[src/trunk]: src/tests/lib/libbpfjit Add BPF_COP and BPF_COPX tests.



details:   https://anonhg.NetBSD.org/src/rev/d40c38f842bc
branches:  trunk
changeset: 796927:d40c38f842bc
user:      alnsn <alnsn%NetBSD.org@localhost>
date:      Wed Jun 25 18:04:05 2014 +0000

description:
Add BPF_COP and BPF_COPX tests.

diffstat:

 tests/lib/libbpfjit/t_cop.c |  572 ++++++++++++++++++++++++++++++++++++++++++++
 1 files changed, 572 insertions(+), 0 deletions(-)

diffs (truncated from 576 to 300 lines):

diff -r 341380495119 -r d40c38f842bc tests/lib/libbpfjit/t_cop.c
--- /dev/null   Thu Jan 01 00:00:00 1970 +0000
+++ b/tests/lib/libbpfjit/t_cop.c       Wed Jun 25 18:04:05 2014 +0000
@@ -0,0 +1,572 @@
+/*     $NetBSD: t_cop.c,v 1.1 2014/06/25 18:04:05 alnsn Exp $ */
+
+/*-
+ * Copyright (c) 2014 Alexander Nasonov.
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ *
+ * 1. Redistributions of source code must retain the above copyright
+ *    notice, this list of conditions and the following disclaimer.
+ * 2. Redistributions in binary form must reproduce the above copyright
+ *    notice, this list of conditions and the following disclaimer in
+ *    the documentation and/or other materials provided with the
+ *    distribution.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+ * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+ * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
+ * FOR A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE
+ * COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
+ * INCIDENTAL, SPECIAL, EXEMPLARY OR CONSEQUENTIAL DAMAGES (INCLUDING,
+ * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
+ * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
+ * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
+ * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
+ * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
+ * SUCH DAMAGE.
+ */
+
+#include <sys/cdefs.h>
+__RCSID("$NetBSD: t_cop.c,v 1.1 2014/06/25 18:04:05 alnsn Exp $");
+
+#include <atf-c.h>
+#include <stdint.h>
+#include <string.h>
+
+#define __BPF_PRIVATE
+#include <net/bpf.h>
+#include <net/bpfjit.h>
+
+static uint32_t retA(const bpf_ctx_t *bc, bpf_args_t *args, uint32_t A);
+static uint32_t retBL(const bpf_ctx_t *bc, bpf_args_t *args, uint32_t A);
+static uint32_t retWL(const bpf_ctx_t *bc, bpf_args_t *args, uint32_t A);
+static uint32_t retNF(const bpf_ctx_t *bc, bpf_args_t *args, uint32_t A);
+static uint32_t setARG(const bpf_ctx_t *bc, bpf_args_t *args, uint32_t A);
+
+static const bpf_copfunc_t copfuncs[] = {
+       &retA,
+       &retBL,
+       &retWL,
+       &retNF,
+       &setARG
+};
+
+static const bpf_ctx_t ctx = {
+       .copfuncs = copfuncs,
+       .nfuncs = sizeof(copfuncs) / sizeof(copfuncs[0]),
+       .extwords = 0
+};
+
+static uint32_t
+retA(const bpf_ctx_t *bc, bpf_args_t *args, uint32_t A)
+{
+
+       return A;
+}
+
+static uint32_t
+retBL(const bpf_ctx_t *bc, bpf_args_t *args, uint32_t A)
+{
+
+       return args->buflen;
+}
+
+static uint32_t
+retWL(const bpf_ctx_t *bc, bpf_args_t *args, uint32_t A)
+{
+
+       return args->wirelen;
+}
+
+static uint32_t
+retNF(const bpf_ctx_t *bc, bpf_args_t *args, uint32_t A)
+{
+
+       return bc->nfuncs;
+}
+
+/*
+ * COP function with a side effect.
+ */
+static uint32_t
+setARG(const bpf_ctx_t *bc, bpf_args_t *args, uint32_t A)
+{
+       bool *arg = (bool *)args->arg;
+       bool old = *arg;
+
+       *arg = true;
+       return old;
+}
+
+ATF_TC(bpfjit_cop_no_ctx);
+ATF_TC_HEAD(bpfjit_cop_no_ctx, tc)
+{
+       atf_tc_set_md_var(tc, "descr", "Test that bpf program with BPF_COP "
+           "instruction isn't valid without a context");
+}
+
+ATF_TC_BODY(bpfjit_cop_no_ctx, tc)
+{
+       static struct bpf_insn insns[] = {
+               BPF_STMT(BPF_MISC+BPF_COP, 0),
+               BPF_STMT(BPF_RET+BPF_K, 7)
+       };
+
+       bpfjit_func_t code;
+
+       size_t insn_count = sizeof(insns) / sizeof(insns[0]);
+
+       ATF_CHECK(!bpf_validate(insns, insn_count));
+
+       code = bpfjit_generate_code(NULL, insns, insn_count);
+       ATF_CHECK(code == NULL);
+}
+
+ATF_TC(bpfjit_cop_ret_A);
+ATF_TC_HEAD(bpfjit_cop_ret_A, tc)
+{
+       atf_tc_set_md_var(tc, "descr", "Test coprocessor function "
+           "that returns a content of the A register");
+}
+
+ATF_TC_BODY(bpfjit_cop_ret_A, tc)
+{
+       static struct bpf_insn insns[] = {
+               BPF_STMT(BPF_LD+BPF_IMM, 13),
+               BPF_STMT(BPF_MISC+BPF_COP, 0), // retA
+               BPF_STMT(BPF_RET+BPF_A, 0)
+       };
+
+       bpfjit_func_t code;
+       uint8_t pkt[1] = { 0 };
+       bpf_args_t args = {
+               .pkt = pkt,
+               .buflen = sizeof(pkt),
+               .wirelen = sizeof(pkt),
+       };
+
+       size_t insn_count = sizeof(insns) / sizeof(insns[0]);
+
+       code = bpfjit_generate_code(&ctx, insns, insn_count);
+       ATF_REQUIRE(code != NULL);
+
+       ATF_CHECK(code(&ctx, &args) == 13);
+
+       bpfjit_free_code(code);
+}
+
+ATF_TC(bpfjit_cop_ret_buflen);
+ATF_TC_HEAD(bpfjit_cop_ret_buflen, tc)
+{
+       atf_tc_set_md_var(tc, "descr", "Test coprocessor function "
+           "that returns the buflen argument");
+}
+
+ATF_TC_BODY(bpfjit_cop_ret_buflen, tc)
+{
+       static struct bpf_insn insns[] = {
+               BPF_STMT(BPF_LD+BPF_IMM, 13),
+               BPF_STMT(BPF_MISC+BPF_COP, 1), // retBL
+               BPF_STMT(BPF_RET+BPF_A, 0)
+       };
+
+       bpfjit_func_t code;
+       uint8_t pkt[1] = { 0 };
+       bpf_args_t args = {
+               .pkt = pkt,
+               .buflen = sizeof(pkt),
+               .wirelen = sizeof(pkt)
+       };
+
+       size_t insn_count = sizeof(insns) / sizeof(insns[0]);
+
+       code = bpfjit_generate_code(&ctx, insns, insn_count);
+       ATF_REQUIRE(code != NULL);
+
+       ATF_CHECK(code(&ctx, &args) == sizeof(pkt));
+
+       bpfjit_free_code(code);
+}
+
+ATF_TC(bpfjit_cop_ret_wirelen);
+ATF_TC_HEAD(bpfjit_cop_ret_wirelen, tc)
+{
+       atf_tc_set_md_var(tc, "descr", "Test coprocessor function "
+           "that returns the wirelen argument");
+}
+
+ATF_TC_BODY(bpfjit_cop_ret_wirelen, tc)
+{
+       static struct bpf_insn insns[] = {
+               BPF_STMT(BPF_LD+BPF_IMM, 13),
+               BPF_STMT(BPF_MISC+BPF_COP, 2), // retWL
+               BPF_STMT(BPF_RET+BPF_A, 0)
+       };
+
+       bpfjit_func_t code;
+       uint8_t pkt[1] = { 0 };
+       bpf_args_t args = {
+               .pkt = pkt,
+               .buflen = sizeof(pkt),
+               .wirelen = sizeof(pkt)
+       };
+
+       size_t insn_count = sizeof(insns) / sizeof(insns[0]);
+
+       code = bpfjit_generate_code(&ctx, insns, insn_count);
+       ATF_REQUIRE(code != NULL);
+
+       ATF_CHECK(code(&ctx, &args) == sizeof(pkt));
+
+       bpfjit_free_code(code);
+}
+
+ATF_TC(bpfjit_cop_ret_nfuncs);
+ATF_TC_HEAD(bpfjit_cop_ret_nfuncs, tc)
+{
+       atf_tc_set_md_var(tc, "descr", "Test coprocessor function "
+           "that returns nfuncs member of the context argument");
+}
+
+ATF_TC_BODY(bpfjit_cop_ret_nfuncs, tc)
+{
+       static struct bpf_insn insns[] = {
+               BPF_STMT(BPF_LD+BPF_IMM, 13),
+               BPF_STMT(BPF_MISC+BPF_COP, 3), // retNF
+               BPF_STMT(BPF_RET+BPF_A, 0)
+       };
+
+       bpfjit_func_t code;
+       uint8_t pkt[1] = { 0 };
+       bpf_args_t args = {
+               .pkt = pkt,
+               .buflen = sizeof(pkt),
+               .wirelen = sizeof(pkt)
+       };
+
+       size_t insn_count = sizeof(insns) / sizeof(insns[0]);
+
+       code = bpfjit_generate_code(&ctx, insns, insn_count);
+       ATF_REQUIRE(code != NULL);
+
+       ATF_CHECK(code(&ctx, &args) == ctx.nfuncs);
+
+       bpfjit_free_code(code);
+}
+
+ATF_TC(bpfjit_cop_side_effect);
+ATF_TC_HEAD(bpfjit_cop_side_effect, tc)
+{
+       atf_tc_set_md_var(tc, "descr",
+           "Test that ABC optimization doesn't skip BPF_COP call");
+}
+
+ATF_TC_BODY(bpfjit_cop_side_effect, tc)
+{
+       static struct bpf_insn insns[] = {
+               BPF_STMT(BPF_LD+BPF_IMM, 13),
+               BPF_STMT(BPF_LD+BPF_B+BPF_ABS, 0),
+               BPF_STMT(BPF_MISC+BPF_COP, 4), // setARG
+               BPF_STMT(BPF_LD+BPF_B+BPF_ABS, 99999),
+               BPF_STMT(BPF_RET+BPF_A, 0)
+       };
+
+       bpfjit_func_t code;
+       bool arg = false;
+       uint8_t pkt[1] = { 0 };
+       bpf_args_t args = {
+               .pkt = pkt,
+               .buflen = sizeof(pkt),
+               .wirelen = sizeof(pkt),
+               .mem = NULL,
+               .arg = &arg
+       };
+
+       size_t insn_count = sizeof(insns) / sizeof(insns[0]);
+
+       code = bpfjit_generate_code(&ctx, insns, insn_count);
+       ATF_REQUIRE(code != NULL);
+
+       ATF_CHECK(code(&ctx, &args) == 0);
+       ATF_CHECK(arg == true);
+
+       bpfjit_free_code(code);



Home | Main Index | Thread Index | Old Index