Home | History | Annotate | Line # | Download | only in libnpftest
npf_test_subr.c revision 1.1.2.2 
      1  1.1.2.2  riz /*	$NetBSD: npf_test_subr.c,v 1.1.2.2 2012/06/26 00:07:19 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.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.2  riz 
     69  1.1.2.2  riz 	result[i++] = tstate->nst_end;
     70  1.1.2.2  riz 	result[i++] = tstate->nst_maxend;
     71  1.1.2.2  riz 	result[i++] = tstate->nst_maxwin;
     72  1.1.2.2  riz 
     73  1.1.2.2  riz 	return 0;
     74  1.1.2.2  riz }
     75