npf_test_subr.c revision 1.1.2.5
1 1.1.2.5 riz /* $NetBSD: npf_test_subr.c,v 1.1.2.5 2012/11/18 21:45:09 riz Exp $ */ 2 1.1.2.2 riz 3 1.1.2.2 riz /* 4 1.1.2.2 riz * NPF initialisation and handler routines. 5 1.1.2.2 riz * 6 1.1.2.2 riz * Public Domain. 7 1.1.2.2 riz */ 8 1.1.2.2 riz 9 1.1.2.2 riz #include <sys/types.h> 10 1.1.2.2 riz #include <net/if.h> 11 1.1.2.2 riz #include <net/if_types.h> 12 1.1.2.2 riz 13 1.1.2.2 riz #include "npf_impl.h" 14 1.1.2.2 riz #include "npf_test.h" 15 1.1.2.2 riz 16 1.1.2.2 riz /* State of the current stream. */ 17 1.1.2.2 riz static npf_state_t cstream_state; 18 1.1.2.2 riz static void * cstream_ptr; 19 1.1.2.2 riz static bool cstream_retval; 20 1.1.2.2 riz 21 1.1.2.5 riz static void npf_state_sample(npf_state_t *, bool); 22 1.1.2.5 riz 23 1.1.2.5 riz void 24 1.1.2.5 riz npf_test_init(void) 25 1.1.2.5 riz { 26 1.1.2.5 riz npf_state_setsampler(npf_state_sample); 27 1.1.2.5 riz } 28 1.1.2.5 riz 29 1.1.2.2 riz int 30 1.1.2.2 riz npf_test_load(const void *xml) 31 1.1.2.2 riz { 32 1.1.2.2 riz prop_dictionary_t npf_dict = prop_dictionary_internalize(xml); 33 1.1.2.2 riz return npfctl_reload(0, npf_dict); 34 1.1.2.2 riz } 35 1.1.2.2 riz 36 1.1.2.4 riz unsigned 37 1.1.2.4 riz npf_test_addif(const char *ifname, unsigned if_idx, bool verbose) 38 1.1.2.4 riz { 39 1.1.2.4 riz ifnet_t *ifp = if_alloc(IFT_OTHER); 40 1.1.2.4 riz 41 1.1.2.4 riz /* 42 1.1.2.4 riz * This is a "fake" interface with explicitly set index. 43 1.1.2.4 riz */ 44 1.1.2.4 riz strlcpy(ifp->if_xname, ifname, sizeof(ifp->if_xname)); 45 1.1.2.4 riz if (verbose) { 46 1.1.2.4 riz printf("+ Interface %s\n", ifp->if_xname); 47 1.1.2.4 riz } 48 1.1.2.4 riz ifp->if_dlt = DLT_NULL; 49 1.1.2.4 riz if_attach(ifp); 50 1.1.2.4 riz ifp->if_index = if_idx; 51 1.1.2.4 riz if_alloc_sadl(ifp); 52 1.1.2.4 riz return if_idx; 53 1.1.2.4 riz } 54 1.1.2.4 riz 55 1.1.2.4 riz unsigned 56 1.1.2.4 riz npf_test_getif(const char *ifname) 57 1.1.2.4 riz { 58 1.1.2.4 riz ifnet_t *ifp = ifunit(ifname); 59 1.1.2.4 riz return ifp ? ifp->if_index : 0; 60 1.1.2.4 riz } 61 1.1.2.4 riz 62 1.1.2.2 riz /* 63 1.1.2.2 riz * State sampler - this routine is called from inside of NPF state engine. 64 1.1.2.2 riz */ 65 1.1.2.5 riz static void 66 1.1.2.2 riz npf_state_sample(npf_state_t *nst, bool retval) 67 1.1.2.2 riz { 68 1.1.2.2 riz /* Pointer will serve as an ID. */ 69 1.1.2.2 riz cstream_ptr = nst; 70 1.1.2.2 riz memcpy(&cstream_state, nst, sizeof(npf_state_t)); 71 1.1.2.2 riz cstream_retval = retval; 72 1.1.2.2 riz } 73 1.1.2.2 riz 74 1.1.2.2 riz int 75 1.1.2.2 riz npf_test_handlepkt(const void *data, size_t len, unsigned idx, 76 1.1.2.2 riz bool forw, int64_t *result) 77 1.1.2.2 riz { 78 1.1.2.2 riz ifnet_t ifp = { .if_index = idx }; 79 1.1.2.2 riz struct mbuf *m; 80 1.1.2.2 riz int i = 0, error; 81 1.1.2.2 riz 82 1.1.2.2 riz m = mbuf_getwithdata(data, len); 83 1.1.2.2 riz error = npf_packet_handler(NULL, &m, &ifp, forw ? PFIL_OUT : PFIL_IN); 84 1.1.2.2 riz if (error) { 85 1.1.2.2 riz assert(m == NULL); 86 1.1.2.2 riz return error; 87 1.1.2.2 riz } 88 1.1.2.2 riz assert(m != NULL); 89 1.1.2.2 riz m_freem(m); 90 1.1.2.2 riz 91 1.1.2.2 riz const int di = forw ? NPF_FLOW_FORW : NPF_FLOW_BACK; 92 1.1.2.2 riz npf_tcpstate_t *fstate = &cstream_state.nst_tcpst[di]; 93 1.1.2.2 riz npf_tcpstate_t *tstate = &cstream_state.nst_tcpst[!di]; 94 1.1.2.2 riz 95 1.1.2.2 riz result[i++] = (intptr_t)cstream_ptr; 96 1.1.2.2 riz result[i++] = cstream_retval; 97 1.1.2.2 riz result[i++] = cstream_state.nst_state; 98 1.1.2.2 riz 99 1.1.2.2 riz result[i++] = fstate->nst_end; 100 1.1.2.2 riz result[i++] = fstate->nst_maxend; 101 1.1.2.2 riz result[i++] = fstate->nst_maxwin; 102 1.1.2.3 jdc result[i++] = fstate->nst_wscale; 103 1.1.2.2 riz 104 1.1.2.2 riz result[i++] = tstate->nst_end; 105 1.1.2.2 riz result[i++] = tstate->nst_maxend; 106 1.1.2.2 riz result[i++] = tstate->nst_maxwin; 107 1.1.2.3 jdc result[i++] = tstate->nst_wscale; 108 1.1.2.2 riz 109 1.1.2.2 riz return 0; 110 1.1.2.2 riz } 111
Indexes created Fri Oct 03 02:09:57 GMT 2025