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