module OPCAP_CLIENT_Tests {

import from OPCAP_Adapter all;
import from OPCAP_Types all;
import from OPCAP_Templates all;

import from IPL4asp_Types all;
import from IPL4asp_PortType all;
import from Osmocom_Types all;
import from Osmocom_VTY_Functions all;
import from TELNETasp_PortType all;
import from Socket_API_Definitions all;

type record IpPort {
	charstring ip,
	integer udp_port
};

modulepar {
	/* local IP address listening for OPCAP connections */
	charstring mp_local_opcap_ip := "127.0.0.1";
	/* local TCP base port for inbound OPCAP connections */
	integer mp_local_opcap_port := 5000;

	/* IP + port for simulating user traffic */
	IpPort mp_traffic_a := { "127.0.0.23", 44423 };
	IpPort mp_traffic_b := { "127.0.0.42", 44442 };
};

type component test_CT extends OPCAP_Adapter_CT {
	timer g_Tguard := 30.0;

	port TELNETasp_PT VTY;

	/* port to generate IP traffic that may or may not be captured */
	port IPL4asp_PT IP;
	var integer g_traffic_conn_id;
};

private altstep as_Tguard() runs on test_CT {
[] g_Tguard.timeout {
	setverdict(fail, "global guard timeout");
	mtc.stop;
	}
}

/* initialize one of the OPCAP servers, wait for client to connect */
private function f_init_one_srv(integer idx, template (present) uint32_t linktype) runs on test_CT {
	/* start guard timer */
	activate(as_Tguard());
	g_Tguard.start;
	log("Waiting for client-", idx, " connection...");
	/* wait for connection */
	f_bind(mp_local_opcap_ip, mp_local_opcap_port+idx, idx);
	f_wait_client_connect(idx);
	/* wait for file header */
	f_opcap_exp(tr_OPCAP_FILE_HDR(linktype), idx);
};


/* global initialization */
private function f_init() runs on test_CT {
	map(self:VTY, system:VTY);
	f_vty_set_prompts(VTY);
	f_vty_transceive(VTY, "enable");

	map(self:IP, system:IP);
	var IPL4asp_Types.Result res

	/* 0 -> 1 */
	res := f_IPL4_connect(IP, mp_traffic_b.ip, mp_traffic_b.udp_port,
			      mp_traffic_a.ip, mp_traffic_a.udp_port, -1, { udp:={} });
	g_traffic_conn_id := res.connId;
}

/* generate user traffic from A -> B */
function f_trafic_pkt_ab(octetstring payload) runs on test_CT {
	IP.send(ASP_Send:{g_traffic_conn_id, omit, payload})
}

/* expect a specified UDP payload on the OPCAP connection 'idx' */
function f_opcap_exp_udp(octetstring udp_payload, integer idx) runs on test_CT {
	var octetstring rx_tail;
	var integer udp_payload_len, rx_pdu_len;
	var OPCAP_PDU rx_pdu;

	udp_payload_len := lengthof(udp_payload);

	/* sadly I couldn't figure out how to create an octetstring template
	 * for 'match an octetstring ending in 'udp_payload' */
	rx_pdu := f_opcap_exp(tr_OPCAP_PKT(?), idx);
	rx_pdu_len := lengthof(rx_pdu.u.packet.payload);
	rx_tail := substr(rx_pdu.u.packet.payload, rx_pdu_len - udp_payload_len, udp_payload_len);
	if (rx_tail != udp_payload) {
		log("captured UDP payload: ", rx_tail, " but expected: ", udp_payload);
		setverdict(fail);
	} else {
		setverdict(pass);
	}
}

/* create an additional pcap-store-connection via the VTY */
function f_vty_create_addl_connection(integer idx) runs on test_CT
{
	f_vty_config3(VTY, { "client", "pcap-store-connectio second-" & int2str(idx) },
			{ "server ip " & mp_local_opcap_ip,
			  "server port " & int2str(mp_local_opcap_port + idx),
			  "connect" }
			);
}



/* wait for inbound client connection and reception of link header */
testcase TC_connect_rx_hdr() runs on test_CT
{
	f_init();
	f_init_one_srv(0, ?);
	setverdict(pass);
}

/* check if client connection is re-started after a close */
testcase TC_reconnect(integer idx := 0) runs on test_CT
{
	f_init();
	f_init_one_srv(idx, ?);
	f_sleep(2.0);

	log("Disconnecting client-", idx);
	f_disconnect(idx);

	f_wait_client_connect(idx);
	f_opcap_exp(tr_OPCAP_FILE_HDR(?), idx);
	setverdict(pass);
}

/* capture a packet that's within the filter */
testcase TC_capture() runs on test_CT
{
	f_init();
	f_init_one_srv(0, ?);

	for (var integer i := 0; i < 10; i := i + 1) {
		var octetstring udp_payload;

		/* we assume 1400 is low enough to avoid IP fragmentation */
		udp_payload := f_rnd_octstring_rnd_len(1400);
		f_trafic_pkt_ab(udp_payload);

		f_opcap_exp_udp(udp_payload, 0);
	}
}

/* wait for inbound client connections and reception of link header */
testcase TC_multi_connect_rx_hdr() runs on test_CT
{
	f_init();
	f_init_one_srv(0, ?);
	f_vty_create_addl_connection(1);
	f_init_one_srv(1, ?);
	setverdict(pass);
}

/* ensure a packet that's within the filter is sent to secondary clients */
testcase TC_multi_capture() runs on test_CT
{
	f_init();
	f_init_one_srv(0, ?);
	f_vty_create_addl_connection(1);
	f_init_one_srv(1, ?);

	for (var integer i := 0; i < 10; i := i + 1) {
		var octetstring udp_payload;

		/* we assume 1400 is low enough to avoid IP fragmentation */
		udp_payload := f_rnd_octstring_rnd_len(1400);
		f_trafic_pkt_ab(udp_payload);

		/* expect packet to arrive on both simulated servers */
		f_opcap_exp_udp(udp_payload, 0);
		f_opcap_exp_udp(udp_payload, 1);
	}
}

/* TODO: ensure a packet outside the filter is dropped */
/* TODO: capture of truncated packet */
/* TODO: stall the receive window */
/* TODO: different link type (ethernet, not SLL) */


control {
	execute( TC_connect_rx_hdr() );
	execute( TC_reconnect() );
	execute( TC_capture() );
	execute( TC_multi_connect_rx_hdr() );
	execute( TC_multi_capture() );
};


};