/*
 * exported_pdu.c
 * exported_pdu helper functions
 * Copyright 2013, Anders Broman <anders-broman@ericsson.com>
 *
 * Wireshark - Network traffic analyzer
 * By Gerald Combs <gerald@wireshark.org>
 * Copyright 1998 Gerald Combs
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License along
 * with this program; if not, write to the Free Software Foundation, Inc.,
 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 */

#include "config.h"
#include <glib.h>

#include <epan/packet.h>
#include <epan/exported_pdu.h>
#include <epan/tap.h>
#include <epan/dissectors/packet-mtp3.h>
#include <epan/dissectors/packet-dvbci.h>

GSList *export_pdu_tap_name_list = NULL;

/**
 * Allocates and fills the exp_pdu_data_t struct according to the wanted_exp_tags
 * bit field of wanted_exp_tags_len bytes length
 * tag_type should be either EXP_PDU_TAG_PROTO_NAME or EXP_PDU_TAG_HEUR_PROTO_NAME
 * proto_name interpretation depends on tag_type value
 *
 * The tags in the tag buffer SHOULD be added in numerical order.
 */
exp_pdu_data_t *
load_export_pdu_tags(packet_info *pinfo, guint tag_type, const char* proto_name,
						guint8 *wanted_exp_tags, guint16 wanted_exp_tags_len)
{
	exp_pdu_data_t *exp_pdu_data;
	int tag_buf_size = 0;
	int str_len = 0;
	int tag_str_len = 0;
	int i = 0;
	gboolean port_type_defined = FALSE;

	exp_pdu_data = (exp_pdu_data_t *)g_malloc(sizeof(exp_pdu_data_t));

	/* If we have a protocol name, calculate the buffer size needed including padding and tag + length */
	if(proto_name){
		str_len = (int)strlen(proto_name);

		/* Ensure that tag length is a multiple of 4 bytes */
		tag_str_len = (str_len + 3) & 0xfffffffc;
		/* Add Tag + length */
		tag_buf_size = tag_str_len + 4;
	}

	/* Check first byte of optional tags bitmap */
	if (wanted_exp_tags_len >= 1) {
		if((wanted_exp_tags[0] & EXP_PDU_TAG_IP_SRC_BIT) == EXP_PDU_TAG_IP_SRC_BIT){
			if(pinfo->net_src.type == AT_IPv4){
				tag_buf_size += 4 + EXP_PDU_TAG_IPV4_SRC_LEN;
			}else if(pinfo->net_src.type == AT_IPv6){
				tag_buf_size += 4 + EXP_PDU_TAG_IPV6_SRC_LEN;
			}
		}

		if((wanted_exp_tags[0] & EXP_PDU_TAG_IP_DST_BIT) == EXP_PDU_TAG_IP_DST_BIT){
			if(pinfo->net_dst.type == AT_IPv4){
				tag_buf_size += 4 + EXP_PDU_TAG_IPV4_DST_LEN;
			}else if(pinfo->net_dst.type == AT_IPv6){
				tag_buf_size += 4 + EXP_PDU_TAG_IPV6_DST_LEN;
			}
		}

		if((wanted_exp_tags[0] & EXP_PDU_TAG_SRC_PORT_BIT) == EXP_PDU_TAG_SRC_PORT_BIT){
			if (!port_type_defined) {
				tag_buf_size= tag_buf_size + EXP_PDU_TAG_PORT_TYPE_LEN + 4;
				port_type_defined = TRUE;
			}
			tag_buf_size= tag_buf_size + EXP_PDU_TAG_SRC_PORT_LEN + 4;
		}

		if((wanted_exp_tags[0] & EXP_PDU_TAG_DST_PORT_BIT) == EXP_PDU_TAG_DST_PORT_BIT){
			if (!port_type_defined) {
				tag_buf_size= tag_buf_size + EXP_PDU_TAG_PORT_TYPE_LEN + 4;
			}
			tag_buf_size= tag_buf_size + EXP_PDU_TAG_DST_PORT_LEN + 4;
		}

		if((wanted_exp_tags[0] & EXP_PDU_TAG_SS7_OPC_BIT) == EXP_PDU_TAG_SS7_OPC_BIT){
			if(pinfo->src.type == AT_SS7PC){
				tag_buf_size += 4 + EXP_PDU_TAG_SS7_OPC_LEN;
			}
		}

		if((wanted_exp_tags[0] & EXP_PDU_TAG_SS7_DPC_BIT) == EXP_PDU_TAG_SS7_DPC_BIT){
			if(pinfo->dst.type == AT_SS7PC){
				tag_buf_size += 4 + EXP_PDU_TAG_SS7_DPC_LEN;
			}
		}

		if((wanted_exp_tags[0] & EXP_PDU_TAG_ORIG_FNO_BIT) == EXP_PDU_TAG_ORIG_FNO_BIT){
			tag_buf_size= tag_buf_size + EXP_PDU_TAG_ORIG_FNO_LEN + 4;
		}
	}

	/* Check second byte of optional tags bitmap */
	if (wanted_exp_tags_len >= 2) {
		if((wanted_exp_tags[1] & EXP_PDU_TAG_DVBCI_EVT_BIT) == EXP_PDU_TAG_DVBCI_EVT_BIT){
			tag_buf_size= tag_buf_size + EXP_PDU_TAG_DVBCI_EVT_LEN + 4;
		}
	}

	/* Add end of options length */
	tag_buf_size+=4;

	exp_pdu_data->tlv_buffer = (guint8 *)g_malloc0(tag_buf_size);
	exp_pdu_data->tlv_buffer_len = tag_buf_size;
	port_type_defined = FALSE;

	if(proto_name){
		exp_pdu_data->tlv_buffer[i] = 0;
		i++;
		exp_pdu_data->tlv_buffer[i] = tag_type;
		i++;
		exp_pdu_data->tlv_buffer[i] = 0;
		i++;
		exp_pdu_data->tlv_buffer[i] = tag_str_len; /* tag length */
		i++;
		memcpy(exp_pdu_data->tlv_buffer+i, proto_name, str_len);
		i = i + tag_str_len;

	}

	/* Check first byte of optional tags bitmap */
	if (wanted_exp_tags_len >= 1) {
		if((wanted_exp_tags[0] & EXP_PDU_TAG_IP_SRC_BIT) == EXP_PDU_TAG_IP_SRC_BIT){
			if(pinfo->net_src.type == AT_IPv4){
				exp_pdu_data->tlv_buffer[i] = 0;
				i++;
				exp_pdu_data->tlv_buffer[i] = EXP_PDU_TAG_IPV4_SRC;
				i++;
				exp_pdu_data->tlv_buffer[i] = 0;
				i++;
				exp_pdu_data->tlv_buffer[i] = EXP_PDU_TAG_IPV4_SRC_LEN; /* tag length */
				i++;
				memcpy(exp_pdu_data->tlv_buffer+i, pinfo->net_src.data, EXP_PDU_TAG_IPV4_SRC_LEN);
				i += EXP_PDU_TAG_IPV4_SRC_LEN;
			}else if(pinfo->net_src.type == AT_IPv6){
				exp_pdu_data->tlv_buffer[i] = 0;
				i++;
				exp_pdu_data->tlv_buffer[i] = EXP_PDU_TAG_IPV6_SRC;
				i++;
				exp_pdu_data->tlv_buffer[i] = 0;
				i++;
				exp_pdu_data->tlv_buffer[i] = EXP_PDU_TAG_IPV6_SRC_LEN; /* tag length */
				i++;
				memcpy(exp_pdu_data->tlv_buffer+i, pinfo->net_src.data, EXP_PDU_TAG_IPV6_SRC_LEN);
				i += EXP_PDU_TAG_IPV6_SRC_LEN;
			}
		}

		if((wanted_exp_tags[0] & EXP_PDU_TAG_IP_DST_BIT) == EXP_PDU_TAG_IP_DST_BIT){
			if(pinfo->net_dst.type == AT_IPv4){
				exp_pdu_data->tlv_buffer[i] = 0;
				i++;
				exp_pdu_data->tlv_buffer[i] = EXP_PDU_TAG_IPV4_DST;
				i++;
				exp_pdu_data->tlv_buffer[i] = 0;
				i++;
				exp_pdu_data->tlv_buffer[i] = EXP_PDU_TAG_IPV4_DST_LEN; /* tag length */
				i++;
				memcpy(exp_pdu_data->tlv_buffer+i, pinfo->net_dst.data, EXP_PDU_TAG_IPV4_DST_LEN);
				i += EXP_PDU_TAG_IPV4_DST_LEN;
			}else if(pinfo->net_dst.type == AT_IPv6){
				exp_pdu_data->tlv_buffer[i] = 0;
				i++;
				exp_pdu_data->tlv_buffer[i] = EXP_PDU_TAG_IPV6_DST;
				i++;
				exp_pdu_data->tlv_buffer[i] = 0;
				i++;
				exp_pdu_data->tlv_buffer[i] = EXP_PDU_TAG_IPV6_DST_LEN; /* tag length */
				i++;
				memcpy(exp_pdu_data->tlv_buffer+i, pinfo->net_dst.data, EXP_PDU_TAG_IPV6_DST_LEN);
				i += EXP_PDU_TAG_IPV6_DST_LEN;
			}
		}

		if((wanted_exp_tags[0] & EXP_PDU_TAG_SRC_PORT_BIT) == EXP_PDU_TAG_SRC_PORT_BIT){
			if (!port_type_defined) {
				exp_pdu_data->tlv_buffer[i] = 0;
				i++;
				exp_pdu_data->tlv_buffer[i] = EXP_PDU_TAG_PORT_TYPE;
				i++;
				exp_pdu_data->tlv_buffer[i] = 0;
				i++;
				exp_pdu_data->tlv_buffer[i] = EXP_PDU_TAG_PORT_TYPE_LEN; /* tag length */
				i++;
				exp_pdu_data->tlv_buffer[i]   = (pinfo->ptype & 0xff000000) >> 24;
				exp_pdu_data->tlv_buffer[i+1] = (pinfo->ptype & 0x00ff0000) >> 16;
				exp_pdu_data->tlv_buffer[i+2] = (pinfo->ptype & 0x0000ff00) >> 8;
				exp_pdu_data->tlv_buffer[i+3] = (pinfo->ptype & 0x000000ff);
				i = i +EXP_PDU_TAG_PORT_TYPE_LEN;
				port_type_defined = TRUE;
			}
			exp_pdu_data->tlv_buffer[i] = 0;
			i++;
			exp_pdu_data->tlv_buffer[i] = EXP_PDU_TAG_SRC_PORT;
			i++;
			exp_pdu_data->tlv_buffer[i] = 0;
			i++;
			exp_pdu_data->tlv_buffer[i] = EXP_PDU_TAG_SRC_PORT_LEN; /* tag length */
			i++;
			exp_pdu_data->tlv_buffer[i]   = (pinfo->srcport & 0xff000000) >> 24;
			exp_pdu_data->tlv_buffer[i+1] = (pinfo->srcport & 0x00ff0000) >> 16;
			exp_pdu_data->tlv_buffer[i+2] = (pinfo->srcport & 0x0000ff00) >> 8;
			exp_pdu_data->tlv_buffer[i+3] = (pinfo->srcport & 0x000000ff);
			i = i +EXP_PDU_TAG_SRC_PORT_LEN;
		}

		if((wanted_exp_tags[0] & EXP_PDU_TAG_DST_PORT_BIT) == EXP_PDU_TAG_DST_PORT_BIT){
			if (!port_type_defined) {
				exp_pdu_data->tlv_buffer[i] = 0;
				i++;
				exp_pdu_data->tlv_buffer[i] = EXP_PDU_TAG_PORT_TYPE;
				i++;
				exp_pdu_data->tlv_buffer[i] = 0;
				i++;
				exp_pdu_data->tlv_buffer[i] = EXP_PDU_TAG_PORT_TYPE_LEN; /* tag length */
				i++;
				exp_pdu_data->tlv_buffer[i]   = (pinfo->ptype & 0xff000000) >> 24;
				exp_pdu_data->tlv_buffer[i+1] = (pinfo->ptype & 0x00ff0000) >> 16;
				exp_pdu_data->tlv_buffer[i+2] = (pinfo->ptype & 0x0000ff00) >> 8;
				exp_pdu_data->tlv_buffer[i+3] = (pinfo->ptype & 0x000000ff);
				i = i +EXP_PDU_TAG_PORT_TYPE_LEN;
			}
			exp_pdu_data->tlv_buffer[i] = 0;
			i++;
			exp_pdu_data->tlv_buffer[i] = EXP_PDU_TAG_DST_PORT;
			i++;
			exp_pdu_data->tlv_buffer[i] = 0;
			i++;
			exp_pdu_data->tlv_buffer[i] = EXP_PDU_TAG_DST_PORT_LEN; /* tag length */
			i++;
			exp_pdu_data->tlv_buffer[i]   = (pinfo->destport & 0xff000000) >> 24;
			exp_pdu_data->tlv_buffer[i+1] = (pinfo->destport & 0x00ff0000) >> 16;
			exp_pdu_data->tlv_buffer[i+2] = (pinfo->destport & 0x0000ff00) >> 8;
			exp_pdu_data->tlv_buffer[i+3] = (pinfo->destport & 0x000000ff);
			i = i +EXP_PDU_TAG_DST_PORT_LEN;
		}

		if((wanted_exp_tags[0] & EXP_PDU_TAG_SS7_OPC_BIT) == EXP_PDU_TAG_SS7_OPC_BIT){
			if(pinfo->src.type == AT_SS7PC){
				const mtp3_addr_pc_t *mtp3_addr = (const mtp3_addr_pc_t *)(pinfo->src.data);
				exp_pdu_data->tlv_buffer[i] = 0;
				i++;
				exp_pdu_data->tlv_buffer[i] = EXP_PDU_TAG_SS7_OPC;
				i++;
				exp_pdu_data->tlv_buffer[i] = 0;
				i++;
				exp_pdu_data->tlv_buffer[i] = EXP_PDU_TAG_SS7_OPC_LEN; /* tag length */
				i++;
				exp_pdu_data->tlv_buffer[i]   = (mtp3_addr->pc & 0xff000000) >> 24;
				exp_pdu_data->tlv_buffer[i+1] = (mtp3_addr->pc & 0x00ff0000) >> 16;
				exp_pdu_data->tlv_buffer[i+2] = (mtp3_addr->pc & 0x0000ff00) >> 8;
				exp_pdu_data->tlv_buffer[i+3] = (mtp3_addr->pc & 0x000000ff);
				exp_pdu_data->tlv_buffer[i+4] = (mtp3_addr->type & 0xff00) >> 8;
				exp_pdu_data->tlv_buffer[i+5] = (mtp3_addr->type & 0x00ff);
				exp_pdu_data->tlv_buffer[i+6] =  mtp3_addr->ni;
				i += EXP_PDU_TAG_SS7_OPC_LEN;
			}
		}

		if((wanted_exp_tags[0] & EXP_PDU_TAG_SS7_DPC_BIT) == EXP_PDU_TAG_SS7_DPC_BIT){
			if(pinfo->dst.type == AT_SS7PC){
				const mtp3_addr_pc_t *mtp3_addr = (const mtp3_addr_pc_t *)(pinfo->dst.data);
				exp_pdu_data->tlv_buffer[i] = 0;
				i++;
				exp_pdu_data->tlv_buffer[i] = EXP_PDU_TAG_SS7_DPC;
				i++;
				exp_pdu_data->tlv_buffer[i] = 0;
				i++;
				exp_pdu_data->tlv_buffer[i] = EXP_PDU_TAG_SS7_DPC_LEN; /* tag length */
				i++;
				exp_pdu_data->tlv_buffer[i]   = (mtp3_addr->pc & 0xff000000) >> 24;
				exp_pdu_data->tlv_buffer[i+1] = (mtp3_addr->pc & 0x00ff0000) >> 16;
				exp_pdu_data->tlv_buffer[i+2] = (mtp3_addr->pc & 0x0000ff00) >> 8;
				exp_pdu_data->tlv_buffer[i+3] = (mtp3_addr->pc & 0x000000ff);
				exp_pdu_data->tlv_buffer[i+4] = (mtp3_addr->type & 0xff00) >> 8;
				exp_pdu_data->tlv_buffer[i+5] = (mtp3_addr->type & 0x00ff);
				exp_pdu_data->tlv_buffer[i+6] =  mtp3_addr->ni;
				i += EXP_PDU_TAG_SS7_DPC_LEN;
			}
		}

		if((wanted_exp_tags[0] & EXP_PDU_TAG_ORIG_FNO_BIT) == EXP_PDU_TAG_ORIG_FNO_BIT){
			exp_pdu_data->tlv_buffer[i] = 0;
			i++;
			exp_pdu_data->tlv_buffer[i] = EXP_PDU_TAG_ORIG_FNO;
			i++;
			exp_pdu_data->tlv_buffer[i] = 0;
			i++;
			exp_pdu_data->tlv_buffer[i] = EXP_PDU_TAG_ORIG_FNO_LEN; /* tag length */
			i++;
			exp_pdu_data->tlv_buffer[i]   = (pinfo->fd->num & 0xff000000) >> 24;
			exp_pdu_data->tlv_buffer[i+1] = (pinfo->fd->num & 0x00ff0000) >> 16;
			exp_pdu_data->tlv_buffer[i+2] = (pinfo->fd->num & 0x0000ff00) >> 8;
			exp_pdu_data->tlv_buffer[i+3] = (pinfo->fd->num & 0x000000ff);
			/*i = i +EXP_PDU_TAG_ORIG_FNO_LEN;*/
		}
	}

	if (wanted_exp_tags_len >= 2) {
		if((wanted_exp_tags[1] & EXP_PDU_TAG_DVBCI_EVT_BIT) == EXP_PDU_TAG_DVBCI_EVT_BIT){
			exp_pdu_data->tlv_buffer[i] = 0;
			i++;
			exp_pdu_data->tlv_buffer[i] = EXP_PDU_TAG_DVBCI_EVT;
			i++;
			exp_pdu_data->tlv_buffer[i] = 0;
			i++;
			exp_pdu_data->tlv_buffer[i] = EXP_PDU_TAG_DVBCI_EVT_LEN;
			i++;
			exp_pdu_data->tlv_buffer[i] = dvbci_get_evt_from_addrs(pinfo);
		}
	}

	return exp_pdu_data;
}

gint
register_export_pdu_tap(const char *name)
{
	gchar *tap_name = g_strdup(name);
	export_pdu_tap_name_list = g_slist_prepend(export_pdu_tap_name_list, tap_name);
	return register_tap(tap_name);
}

static
gint sort_pdu_tap_name_list(gconstpointer a, gconstpointer b)
{
	return g_strcmp0((const char *)a, (const char*)b);
}

GSList *
get_export_pdu_tap_list(void)
{
	export_pdu_tap_name_list = g_slist_sort(export_pdu_tap_name_list, sort_pdu_tap_name_list);
	return export_pdu_tap_name_list;
}

/*
 * Editor modelines  -  http://www.wireshark.org/tools/modelines.html
 *
 * Local variables:
 * c-basic-offset: 8
 * tab-width: 8
 * indent-tabs-mode: t
 * End:
 *
 * vi: set shiftwidth=8 tabstop=8 noexpandtab:
 * :indentSize=8:tabSize=8:noTabs=false:
 */