npf_test_subr.c revision 1.1.2.3 1 1.1.2.3 jdc /* $NetBSD: npf_test_subr.c,v 1.1.2.3 2012/07/25 20:45:24 jdc 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.2 riz /*
29 1.1.2.2 riz * State sampler - this routine is called from inside of NPF state engine.
30 1.1.2.2 riz */
31 1.1.2.2 riz void
32 1.1.2.2 riz npf_state_sample(npf_state_t *nst, bool retval)
33 1.1.2.2 riz {
34 1.1.2.2 riz /* Pointer will serve as an ID. */
35 1.1.2.2 riz cstream_ptr = nst;
36 1.1.2.2 riz memcpy(&cstream_state, nst, sizeof(npf_state_t));
37 1.1.2.2 riz cstream_retval = retval;
38 1.1.2.2 riz }
39 1.1.2.2 riz
40 1.1.2.2 riz int
41 1.1.2.2 riz npf_test_handlepkt(const void *data, size_t len, unsigned idx,
42 1.1.2.2 riz bool forw, int64_t *result)
43 1.1.2.2 riz {
44 1.1.2.2 riz ifnet_t ifp = { .if_index = idx };
45 1.1.2.2 riz struct mbuf *m;
46 1.1.2.2 riz int i = 0, error;
47 1.1.2.2 riz
48 1.1.2.2 riz m = mbuf_getwithdata(data, len);
49 1.1.2.2 riz error = npf_packet_handler(NULL, &m, &ifp, forw ? PFIL_OUT : PFIL_IN);
50 1.1.2.2 riz if (error) {
51 1.1.2.2 riz assert(m == NULL);
52 1.1.2.2 riz return error;
53 1.1.2.2 riz }
54 1.1.2.2 riz assert(m != NULL);
55 1.1.2.2 riz m_freem(m);
56 1.1.2.2 riz
57 1.1.2.2 riz const int di = forw ? NPF_FLOW_FORW : NPF_FLOW_BACK;
58 1.1.2.2 riz npf_tcpstate_t *fstate = &cstream_state.nst_tcpst[di];
59 1.1.2.2 riz npf_tcpstate_t *tstate = &cstream_state.nst_tcpst[!di];
60 1.1.2.2 riz
61 1.1.2.2 riz result[i++] = (intptr_t)cstream_ptr;
62 1.1.2.2 riz result[i++] = cstream_retval;
63 1.1.2.2 riz result[i++] = cstream_state.nst_state;
64 1.1.2.2 riz
65 1.1.2.2 riz result[i++] = fstate->nst_end;
66 1.1.2.2 riz result[i++] = fstate->nst_maxend;
67 1.1.2.2 riz result[i++] = fstate->nst_maxwin;
68 1.1.2.3 jdc result[i++] = fstate->nst_wscale;
69 1.1.2.2 riz
70 1.1.2.2 riz result[i++] = tstate->nst_end;
71 1.1.2.2 riz result[i++] = tstate->nst_maxend;
72 1.1.2.2 riz result[i++] = tstate->nst_maxwin;
73 1.1.2.3 jdc result[i++] = tstate->nst_wscale;
74 1.1.2.2 riz
75 1.1.2.2 riz return 0;
76 1.1.2.2 riz }
77