1272343Sngie/* $NetBSD: h_bpf.h,v 1.2 2014/07/08 21:44:26 alnsn Exp $ */ 2272343Sngie 3272343Sngie/*- 4272343Sngie * Copyright (c) 2014 Alexander Nasonov. 5272343Sngie * All rights reserved. 6272343Sngie * 7272343Sngie * Redistribution and use in source and binary forms, with or without 8272343Sngie * modification, are permitted provided that the following conditions 9272343Sngie * are met: 10272343Sngie * 1. Redistributions of source code must retain the above copyright 11272343Sngie * notice, this list of conditions and the following disclaimer. 12272343Sngie * 2. Redistributions in binary form must reproduce the above copyright 13272343Sngie * notice, this list of conditions and the following disclaimer in the 14272343Sngie * documentation and/or other materials provided with the distribution. 15272343Sngie * 16272343Sngie * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS 17272343Sngie * OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED 18272343Sngie * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE 19272343Sngie * DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 20272343Sngie * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 21272343Sngie * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR 22272343Sngie * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 23272343Sngie * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 24272343Sngie * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 25272343Sngie * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 26272343Sngie * SUCH DAMAGE. 27272343Sngie */ 28272343Sngie 29272343Sngie#ifndef _TESTS_NET_BPF_H_BPF_H_ 30272343Sngie#define _TESTS_NET_BPF_H_BPF_H_ 31272343Sngie 32272343Sngie#include <sys/param.h> 33272343Sngie#include <sys/mbuf.h> 34272343Sngie 35272343Sngie#include <net/bpf.h> 36272343Sngie#include <net/bpfjit.h> 37272343Sngie 38272343Sngie#include <rump/rump.h> 39272343Sngie#include <rump/rump_syscalls.h> 40272343Sngie 41272343Sngie#include <stdint.h> 42272343Sngie#include <string.h> 43272343Sngie 44272343Sngie/* XXX These declarations don't look kosher. */ 45272343Sngieint rumpns_bpf_validate(const struct bpf_insn *, int); 46272343Sngieunsigned int rumpns_bpf_filter_ext(const bpf_ctx_t *, 47272343Sngie const struct bpf_insn *, bpf_args_t *); 48272343Sngiebpfjit_func_t rumpns_bpfjit_generate_code(const bpf_ctx_t *, 49272343Sngie const struct bpf_insn *, size_t); 50272343Sngievoid rumpns_bpfjit_free_code(bpfjit_func_t); 51272343Sngie 52272343Sngie/* 53272343Sngie * Init mbuf chain with one or two chunks. The first chunk holds 54272343Sngie * [pkt, pkt + split] bytes, the second chunk (if it's not empty) 55272343Sngie * holds (pkt + split, pkt + pktsize) bytes. 56272343Sngie * The function returns (const uint8_t *)mb1. 57272343Sngie */ 58272343Sngiestatic inline const uint8_t * 59272343Sngieinit_mchain2(struct mbuf *mb1, struct mbuf *mb2, 60272343Sngie unsigned char pkt[], size_t pktsize, size_t split) 61272343Sngie{ 62272343Sngie 63272343Sngie (void)memset(mb1, 0, sizeof(*mb1)); 64272343Sngie mb1->m_data = (char *)pkt; 65272343Sngie mb1->m_next = (split < pktsize) ? mb2 : NULL; 66272343Sngie mb1->m_len = (split < pktsize) ? split : pktsize; 67272343Sngie 68272343Sngie if (split < pktsize) { 69272343Sngie (void)memset(mb2, 0, sizeof(*mb2)); 70272343Sngie mb2->m_next = NULL; 71272343Sngie mb2->m_data = (char *)&pkt[split]; 72272343Sngie mb2->m_len = pktsize - split; 73272343Sngie } 74272343Sngie 75272343Sngie return (const uint8_t*)mb1; 76272343Sngie} 77272343Sngie 78272343Sngie/* 79272343Sngie * Compile and run a filter program. 80272343Sngie */ 81272343Sngiestatic inline unsigned int 82272343Sngieexec_prog(struct bpf_insn *insns, size_t insn_count, 83272343Sngie unsigned char pkt[], size_t pktsize) 84272343Sngie{ 85272343Sngie bpfjit_func_t fn; 86272343Sngie bpf_args_t args; 87272343Sngie unsigned int res; 88272343Sngie 89272343Sngie args.pkt = (const uint8_t *)pkt; 90272343Sngie args.buflen = pktsize; 91272343Sngie args.wirelen = pktsize; 92272343Sngie 93272343Sngie rump_schedule(); 94272343Sngie fn = rumpns_bpfjit_generate_code(NULL, insns, insn_count); 95272343Sngie rump_unschedule(); 96272343Sngie 97272343Sngie res = fn(NULL, &args); 98272343Sngie 99272343Sngie rump_schedule(); 100272343Sngie rumpns_bpfjit_free_code(fn); 101272343Sngie rump_unschedule(); 102272343Sngie 103272343Sngie return res; 104272343Sngie} 105272343Sngie 106272343Sngie/* 107272343Sngie * Interpret a filter program with mbuf chain passed to bpf_filter_ext(). 108272343Sngie */ 109272343Sngiestatic inline unsigned int 110272343Sngieinterp_prog_mchain2(struct bpf_insn *insns, 111272343Sngie unsigned char pkt[], size_t pktsize, size_t split) 112272343Sngie{ 113272343Sngie uint32_t mem[BPF_MEMWORDS]; 114272343Sngie struct mbuf mb1, mb2; 115272343Sngie bpf_args_t args; 116272343Sngie unsigned int res; 117272343Sngie 118272343Sngie args.pkt = init_mchain2(&mb1, &mb2, pkt, pktsize, split); 119272343Sngie args.buflen = 0; 120272343Sngie args.wirelen = pktsize; 121272343Sngie args.mem = mem; 122272343Sngie 123272343Sngie rump_schedule(); 124272343Sngie res = rumpns_bpf_filter_ext(NULL, insns, &args); 125272343Sngie rump_unschedule(); 126272343Sngie 127272343Sngie return res; 128272343Sngie} 129272343Sngie 130272343Sngie/* 131272343Sngie * Compile and run a filter program with mbuf chain passed to compiled function. 132272343Sngie */ 133272343Sngiestatic inline unsigned int 134272343Sngieexec_prog_mchain2(struct bpf_insn *insns, size_t insn_count, 135272343Sngie unsigned char pkt[], size_t pktsize, size_t split) 136272343Sngie{ 137272343Sngie bpfjit_func_t fn; 138272343Sngie struct mbuf mb1, mb2; 139272343Sngie bpf_args_t args; 140272343Sngie unsigned int res; 141272343Sngie 142272343Sngie args.pkt = init_mchain2(&mb1, &mb2, pkt, pktsize, split); 143272343Sngie args.buflen = 0; 144272343Sngie args.wirelen = pktsize; 145272343Sngie 146272343Sngie rump_schedule(); 147272343Sngie fn = rumpns_bpfjit_generate_code(NULL, insns, insn_count); 148272343Sngie rump_unschedule(); 149272343Sngie 150272343Sngie res = fn(NULL, &args); 151272343Sngie 152272343Sngie rump_schedule(); 153272343Sngie rumpns_bpfjit_free_code(fn); 154272343Sngie rump_unschedule(); 155272343Sngie 156272343Sngie return res; 157272343Sngie} 158272343Sngie 159272343Sngiestatic inline bool 160272343Sngieprog_validate(struct bpf_insn *insns, size_t insn_count) 161272343Sngie{ 162272343Sngie bool res; 163272343Sngie 164272343Sngie rump_schedule(); 165272343Sngie res = rumpns_bpf_validate(insns, insn_count); 166272343Sngie rump_unschedule(); 167272343Sngie 168272343Sngie return res; 169272343Sngie} 170272343Sngie 171272343Sngie#endif /* _TESTS_NET_BPF_H_BPF_H_ */ 172