src/vnet/ipfix-export/flow_api.c - fdio/vpp - Gitiles

 /*
  *------------------------------------------------------------------
  * flow_api.c - flow api
  *
  * Copyright (c) 2016 Cisco and/or its affiliates.
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
  * You may obtain a copy of the License at:
  *
  *     http://www.apache.org/licenses/LICENSE-2.0
  *
  * Unless required by applicable law or agreed to in writing, software
  * distributed under the License is distributed on an "AS IS" BASIS,
  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  * See the License for the specific language governing permissions and
  * limitations under the License.
  *------------------------------------------------------------------
  */

 #include <vnet/vnet.h>
 #include <vlibmemory/api.h>
 #include <vnet/ip/ip_types_api.h>
 #include <vnet/udp/udp_local.h>

 #include <vnet/interface.h>
 #include <vnet/api_errno.h>

 #include <vnet/fib/fib_table.h>
 #include <vnet/ipfix-export/flow_report.h>
 #include <vnet/ipfix-export/flow_report_classify.h>

 #include <vnet/format_fns.h>
 #include <vnet/ipfix-export/ipfix_export.api_enum.h>
 #include <vnet/ipfix-export/ipfix_export.api_types.h>

 #define REPLY_MSG_ID_BASE frm->msg_id_base
 #include <vlibapi/api_helper_macros.h>

 ipfix_exporter_t *
 vnet_ipfix_exporter_lookup (ip4_address_t *ipfix_collector)
 {
   flow_report_main_t *frm = &flow_report_main;
   ipfix_exporter_t *exp;

   pool_foreach (exp, frm->exporters)
     {
       if (exp->ipfix_collector.as_u32 == ipfix_collector->as_u32)
 	return exp;
     }

   return NULL;
 }

 /*
  * For backwards compatibility reasons index 0 in the set of exporters
  * is alwyas used for the exporter created via the set_ipfix_exporter
  * API.
  */
 #define USE_INDEX_0   true
 #define USE_ANY_INDEX false

 static int
 vl_api_set_ipfix_exporter_t_internal (
   u32 client_index, vl_api_address_t *mp_collector_address,
   u16 mp_collector_port, vl_api_address_t *mp_src_address, u32 mp_vrf_id,
   u32 mp_path_mtu, u32 mp_template_interval, bool mp_udp_checksum,
   bool use_index_0, bool is_create)
 {
   vlib_main_t *vm = vlib_get_main ();
   flow_report_main_t *frm = &flow_report_main;
   ipfix_exporter_t *exp;
   vl_api_registration_t *reg;
   ip4_address_t collector, src;
   u16 collector_port = UDP_DST_PORT_ipfix;
   u32 path_mtu;
   u32 template_interval;
   u8 udp_checksum;
   u32 fib_id;
   u32 fib_index = ~0;

   reg = vl_api_client_index_to_registration (client_index);
   if (!reg)
     return VNET_API_ERROR_UNIMPLEMENTED;

   /* Collector address is the key for the exporter lookup */
   ip4_address_decode (mp_collector_address->un.ip4, &collector);

   if (use_index_0)
     /*
      * In this case we update the existing exporter. There is no delete
      * for exp[0]
      */
     exp = &frm->exporters[0];
   else
     {
       if (is_create)
 	{
 	  exp = vnet_ipfix_exporter_lookup (&collector);
 	  if (!exp)
 	    {
 	      /* Create a new exporter instead of updating an existing one */
 	      if (pool_elts (frm->exporters) >= IPFIX_EXPORTERS_MAX)
 		return VNET_API_ERROR_INVALID_VALUE;
 	      pool_get (frm->exporters, exp);
 	      if (!exp)
 		return VNET_API_ERROR_INVALID_VALUE;
 	    }
 	}
       else
 	{
 	  /* Delete the exporter */
 	  exp = vnet_ipfix_exporter_lookup (&collector);
 	  if (!exp)
 	    return VNET_API_ERROR_NO_SUCH_ENTRY;

 	  pool_put (frm->exporters, exp);
 	  return 0;
 	}
     }

   if (mp_src_address->af == ADDRESS_IP6 ||
       mp_collector_address->af == ADDRESS_IP6)
     {
       return VNET_API_ERROR_UNIMPLEMENTED;
     }

   collector_port = ntohs (mp_collector_port);
   if (collector_port == (u16) ~ 0)
     collector_port = UDP_DST_PORT_ipfix;
   ip4_address_decode (mp_src_address->un.ip4, &src);
   fib_id = ntohl (mp_vrf_id);

   ip4_main_t *im = &ip4_main;
   if (fib_id == ~0)
     {
       fib_index = ~0;
     }
   else
     {
       uword *p = hash_get (im->fib_index_by_table_id, fib_id);
       if (!p)
 	return VNET_API_ERROR_NO_SUCH_FIB;
       fib_index = p[0];
     }

   path_mtu = ntohl (mp_path_mtu);
   if (path_mtu == ~0)
     path_mtu = 512;		// RFC 7011 section 10.3.3.
   template_interval = ntohl (mp_template_interval);
   if (template_interval == ~0)
     template_interval = 20;
   udp_checksum = mp_udp_checksum;

   if (collector.as_u32 != 0 && src.as_u32 == 0)
     return VNET_API_ERROR_INVALID_VALUE;

   if (path_mtu > 1450 /* vpp does not support fragmentation */ )
     return VNET_API_ERROR_INVALID_VALUE;

   if (path_mtu < 68)
     return VNET_API_ERROR_INVALID_VALUE;

   /* Reset report streams if we are reconfiguring IP addresses */
   if (exp->ipfix_collector.as_u32 != collector.as_u32 ||
       exp->src_address.as_u32 != src.as_u32 ||
       exp->collector_port != collector_port)
     vnet_flow_reports_reset (exp);

   exp->ipfix_collector.as_u32 = collector.as_u32;
   exp->collector_port = collector_port;
   exp->src_address.as_u32 = src.as_u32;
   exp->fib_index = fib_index;
   exp->path_mtu = path_mtu;
   exp->template_interval = template_interval;
   exp->udp_checksum = udp_checksum;

   /* Turn on the flow reporting process */
   vlib_process_signal_event (vm, flow_report_process_node.index, 1, 0);

   return 0;
 }

 static void
 vl_api_set_ipfix_exporter_t_handler (vl_api_set_ipfix_exporter_t *mp)
 {
   vl_api_set_ipfix_exporter_reply_t *rmp;
   flow_report_main_t *frm = &flow_report_main;
   int rv = vl_api_set_ipfix_exporter_t_internal (
     mp->client_index, &mp->collector_address, mp->collector_port,
     &mp->src_address, mp->vrf_id, mp->path_mtu, mp->template_interval,
     mp->udp_checksum, USE_INDEX_0, 0);

   REPLY_MACRO (VL_API_SET_IPFIX_EXPORTER_REPLY);
 }

 static void
 vl_api_ipfix_exporter_create_delete_t_handler (
   vl_api_ipfix_exporter_create_delete_t *mp)
 {
   vl_api_ipfix_exporter_create_delete_reply_t *rmp;
   flow_report_main_t *frm = &flow_report_main;
   int rv = vl_api_set_ipfix_exporter_t_internal (
     mp->client_index, &mp->collector_address, mp->collector_port,
     &mp->src_address, mp->vrf_id, mp->path_mtu, mp->template_interval,
     mp->udp_checksum, USE_ANY_INDEX, mp->is_create);

   REPLY_MACRO (VL_API_IPFIX_EXPORTER_CREATE_DELETE_REPLY);
 }

 static void
 vl_api_ipfix_exporter_dump_t_handler (vl_api_ipfix_exporter_dump_t * mp)
 {
   flow_report_main_t *frm = &flow_report_main;
   ipfix_exporter_t *exp = pool_elt_at_index (flow_report_main.exporters, 0);
   vl_api_registration_t *reg;
   vl_api_ipfix_exporter_details_t *rmp;
   ip4_main_t *im = &ip4_main;
   ip46_address_t collector = {.as_u64[0] = 0,.as_u64[1] = 0 };
   ip46_address_t src = {.as_u64[0] = 0,.as_u64[1] = 0 };
   u32 vrf_id;

   reg = vl_api_client_index_to_registration (mp->client_index);
   if (!reg)
     return;

   rmp = vl_msg_api_alloc (sizeof (*rmp));
   clib_memset (rmp, 0, sizeof (*rmp));
   rmp->_vl_msg_id =
     ntohs ((REPLY_MSG_ID_BASE) + VL_API_IPFIX_EXPORTER_DETAILS);
   rmp->context = mp->context;

   memcpy (&collector.ip4, &exp->ipfix_collector, sizeof (ip4_address_t));
   ip_address_encode (&collector, IP46_TYPE_IP4, &rmp->collector_address);

   rmp->collector_port = htons (exp->collector_port);

   memcpy (&src.ip4, &exp->src_address, sizeof (ip4_address_t));
   ip_address_encode (&src, IP46_TYPE_IP4, &rmp->src_address);

   if (exp->fib_index == ~0)
     vrf_id = ~0;
   else
     vrf_id = im->fibs[exp->fib_index].ft_table_id;
   rmp->vrf_id = htonl (vrf_id);
   rmp->path_mtu = htonl (exp->path_mtu);
   rmp->template_interval = htonl (exp->template_interval);
   rmp->udp_checksum = (exp->udp_checksum != 0);

   vl_api_send_msg (reg, (u8 *) rmp);
 }

 static void
 ipfix_all_fill_details (vl_api_ipfix_all_exporter_details_t *rmp,
 			ipfix_exporter_t *exp)
 {
   ip4_main_t *im = &ip4_main;
   ip46_address_t collector = { .as_u64[0] = 0, .as_u64[1] = 0 };
   ip46_address_t src = { .as_u64[0] = 0, .as_u64[1] = 0 };
   u32 vrf_id;

   memcpy (&collector.ip4, &exp->ipfix_collector, sizeof (ip4_address_t));
   ip_address_encode (&collector, IP46_TYPE_IP4, &rmp->collector_address);

   rmp->collector_port = htons (exp->collector_port);

   memcpy (&src.ip4, &exp->src_address, sizeof (ip4_address_t));
   ip_address_encode (&src, IP46_TYPE_IP4, &rmp->src_address);

   if (exp->fib_index == ~0)
     vrf_id = ~0;
   else
     vrf_id = im->fibs[exp->fib_index].ft_table_id;
   rmp->vrf_id = htonl (vrf_id);
   rmp->path_mtu = htonl (exp->path_mtu);
   rmp->template_interval = htonl (exp->template_interval);
   rmp->udp_checksum = (exp->udp_checksum != 0);
 }

 static void
 ipfix_all_exporter_details (flow_report_main_t *frm, u32 index,
 			    vl_api_registration_t *rp, u32 context)
 {
   ipfix_exporter_t *exp = pool_elt_at_index (frm->exporters, index);

   vl_api_ipfix_all_exporter_details_t *rmp;

   REPLY_MACRO_DETAILS4 (VL_API_IPFIX_ALL_EXPORTER_DETAILS, rp, context,
 			({ ipfix_all_fill_details (rmp, exp); }));
 }

 static void
 vl_api_ipfix_all_exporter_get_t_handler (vl_api_ipfix_all_exporter_get_t *mp)
 {
   flow_report_main_t *frm = &flow_report_main;
   vl_api_ipfix_all_exporter_get_reply_t *rmp;
   int rv = 0;

   REPLY_AND_DETAILS_MACRO (
     VL_API_IPFIX_ALL_EXPORTER_GET_REPLY, frm->exporters,
     ({ ipfix_all_exporter_details (frm, cursor, rp, mp->context); }));
 }

 static void
   vl_api_set_ipfix_classify_stream_t_handler
   (vl_api_set_ipfix_classify_stream_t * mp)
 {
   vl_api_set_ipfix_classify_stream_reply_t *rmp;
   flow_report_classify_main_t *fcm = &flow_report_classify_main;
   flow_report_main_t *frm = &flow_report_main;
   ipfix_exporter_t *exp = &frm->exporters[0];
   u32 domain_id = 0;
   u32 src_port = UDP_DST_PORT_ipfix;
   int rv = 0;

   domain_id = ntohl (mp->domain_id);
   src_port = ntohs (mp->src_port);

   if (fcm->src_port != 0 &&
       (fcm->domain_id != domain_id || fcm->src_port != (u16) src_port))
     {
       int rv = vnet_stream_change (exp, fcm->domain_id, fcm->src_port,
 				   domain_id, (u16) src_port);
       ASSERT (rv == 0);
     }

   fcm->domain_id = domain_id;
   fcm->src_port = (u16) src_port;

   REPLY_MACRO (VL_API_SET_IPFIX_CLASSIFY_STREAM_REPLY);
 }

 static void
   vl_api_ipfix_classify_stream_dump_t_handler
   (vl_api_ipfix_classify_stream_dump_t * mp)
 {
   flow_report_classify_main_t *fcm = &flow_report_classify_main;
   vl_api_registration_t *reg;
   vl_api_ipfix_classify_stream_details_t *rmp;

   reg = vl_api_client_index_to_registration (mp->client_index);
   if (!reg)
     return;

   rmp = vl_msg_api_alloc (sizeof (*rmp));
   clib_memset (rmp, 0, sizeof (*rmp));
   rmp->_vl_msg_id = ntohs (VL_API_IPFIX_CLASSIFY_STREAM_DETAILS);
   rmp->context = mp->context;
   rmp->domain_id = htonl (fcm->domain_id);
   rmp->src_port = htons (fcm->src_port);

   vl_api_send_msg (reg, (u8 *) rmp);
 }

 static void
   vl_api_ipfix_classify_table_add_del_t_handler
   (vl_api_ipfix_classify_table_add_del_t * mp)
 {
   vl_api_ipfix_classify_table_add_del_reply_t *rmp;
   vl_api_registration_t *reg;
   flow_report_classify_main_t *fcm = &flow_report_classify_main;
   flow_report_main_t *frm = &flow_report_main;
   ipfix_exporter_t *exp = &frm->exporters[0];
   vnet_flow_report_add_del_args_t args;
   ipfix_classify_table_t *table;
   int is_add;
   u32 classify_table_index;
   u8 ip_version;
   u8 transport_protocol;
   int rv = 0;

   reg = vl_api_client_index_to_registration (mp->client_index);
   if (!reg)
     return;

   classify_table_index = ntohl (mp->table_id);
   ip_version = (mp->ip_version == ADDRESS_IP4) ? 4 : 6;
   transport_protocol = mp->transport_protocol;
   is_add = mp->is_add;

   if (fcm->src_port == 0)
     {
       /* call set_ipfix_classify_stream first */
       rv = VNET_API_ERROR_UNSPECIFIED;
       goto out;
     }

   clib_memset (&args, 0, sizeof (args));

   table = 0;
   int i;
   for (i = 0; i < vec_len (fcm->tables); i++)
     if (ipfix_classify_table_index_valid (i))
       if (fcm->tables[i].classify_table_index == classify_table_index)
 	{
 	  table = &fcm->tables[i];
 	  break;
 	}

   if (is_add)
     {
       if (table)
 	{
 	  rv = VNET_API_ERROR_VALUE_EXIST;
 	  goto out;
 	}
       table = ipfix_classify_add_table ();
       table->classify_table_index = classify_table_index;
     }
   else
     {
       if (!table)
 	{
 	  rv = VNET_API_ERROR_NO_SUCH_ENTRY;
 	  goto out;
 	}
     }

   table->ip_version = ip_version;
   table->transport_protocol = transport_protocol;

   args.opaque.as_uword = table - fcm->tables;
   args.rewrite_callback = ipfix_classify_template_rewrite;
   args.flow_data_callback = ipfix_classify_send_flows;
   args.is_add = is_add;
   args.domain_id = fcm->domain_id;
   args.src_port = fcm->src_port;

   rv = vnet_flow_report_add_del (exp, &args, NULL);

   /* If deleting, or add failed */
   if (is_add == 0 || (rv && is_add))
     ipfix_classify_delete_table (table - fcm->tables);

 out:
   REPLY_MACRO (VL_API_SET_IPFIX_CLASSIFY_STREAM_REPLY);
 }

 static void
 send_ipfix_classify_table_details (u32 table_index,
 				   vl_api_registration_t * reg, u32 context)
 {
   flow_report_classify_main_t *fcm = &flow_report_classify_main;
   vl_api_ipfix_classify_table_details_t *mp;

   ipfix_classify_table_t *table = &fcm->tables[table_index];

   mp = vl_msg_api_alloc (sizeof (*mp));
   clib_memset (mp, 0, sizeof (*mp));
   mp->_vl_msg_id = ntohs (VL_API_IPFIX_CLASSIFY_TABLE_DETAILS);
   mp->context = context;
   mp->table_id = htonl (table->classify_table_index);
   mp->ip_version = (table->ip_version == 4) ? ADDRESS_IP4 : ADDRESS_IP6;
   mp->transport_protocol = table->transport_protocol;

   vl_api_send_msg (reg, (u8 *) mp);
 }

 static void
   vl_api_ipfix_classify_table_dump_t_handler
   (vl_api_ipfix_classify_table_dump_t * mp)
 {
   flow_report_classify_main_t *fcm = &flow_report_classify_main;
   vl_api_registration_t *reg;
   u32 i;

   reg = vl_api_client_index_to_registration (mp->client_index);
   if (!reg)
     return;

   for (i = 0; i < vec_len (fcm->tables); i++)
     if (ipfix_classify_table_index_valid (i))
       send_ipfix_classify_table_details (i, reg, mp->context);
 }

 static void
 vl_api_ipfix_flush_t_handler (vl_api_ipfix_flush_t * mp)
 {
   flow_report_main_t *frm = &flow_report_main;
   vl_api_ipfix_flush_reply_t *rmp;
   vl_api_registration_t *reg;
   vlib_main_t *vm = vlib_get_main ();
   int rv = 0;

   reg = vl_api_client_index_to_registration (mp->client_index);
   if (!reg)
     return;

   /* poke the flow reporting process */
   vlib_process_signal_event (vm, flow_report_process_node.index,
 			     1 /* type_opaque */ , 0 /* data */ );

   REPLY_MACRO (VL_API_IPFIX_FLUSH_REPLY);
 }

 #include <vnet/ipfix-export/ipfix_export.api.c>
 static clib_error_t *
 flow_api_hookup (vlib_main_t * vm)
 {
   flow_report_main_t *frm = &flow_report_main;
   /*
    * Set up the (msg_name, crc, message-id) table
    */
   REPLY_MSG_ID_BASE = setup_message_id_table ();

   return 0;
 }

 VLIB_API_INIT_FUNCTION (flow_api_hookup);

 /*
  * fd.io coding-style-patch-verification: ON
  *
  * Local Variables:
  * eval: (c-set-style "gnu")
  * End:
  */
	/*
	*------------------------------------------------------------------
	* flow_api.c - flow api
	*
	* Copyright (c) 2016 Cisco and/or its affiliates.
	* Licensed under the Apache License, Version 2.0 (the "License");
	* you may not use this file except in compliance with the License.
	* You may obtain a copy of the License at:
	*
	* http://www.apache.org/licenses/LICENSE-2.0
	*
	* Unless required by applicable law or agreed to in writing, software
	* distributed under the License is distributed on an "AS IS" BASIS,
	* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
	* See the License for the specific language governing permissions and
	* limitations under the License.
	*------------------------------------------------------------------
	*/

	#include <vnet/vnet.h>
	#include <vlibmemory/api.h>
	#include <vnet/ip/ip_types_api.h>
	#include <vnet/udp/udp_local.h>

	#include <vnet/interface.h>
	#include <vnet/api_errno.h>

	#include <vnet/fib/fib_table.h>
	#include <vnet/ipfix-export/flow_report.h>
	#include <vnet/ipfix-export/flow_report_classify.h>

	#include <vnet/format_fns.h>
	#include <vnet/ipfix-export/ipfix_export.api_enum.h>
	#include <vnet/ipfix-export/ipfix_export.api_types.h>

	#define REPLY_MSG_ID_BASE frm->msg_id_base
	#include <vlibapi/api_helper_macros.h>

	ipfix_exporter_t *
	vnet_ipfix_exporter_lookup (ip4_address_t *ipfix_collector)
	{
	flow_report_main_t *frm = &flow_report_main;
	ipfix_exporter_t *exp;

	pool_foreach (exp, frm->exporters)
	{
	if (exp->ipfix_collector.as_u32 == ipfix_collector->as_u32)
	return exp;
	}

	return NULL;
	}

	/*
	* For backwards compatibility reasons index 0 in the set of exporters
	* is alwyas used for the exporter created via the set_ipfix_exporter
	* API.
	*/
	#define USE_INDEX_0 true
	#define USE_ANY_INDEX false

	static int
	vl_api_set_ipfix_exporter_t_internal (
	u32 client_index, vl_api_address_t *mp_collector_address,
	u16 mp_collector_port, vl_api_address_t *mp_src_address, u32 mp_vrf_id,
	u32 mp_path_mtu, u32 mp_template_interval, bool mp_udp_checksum,
	bool use_index_0, bool is_create)
	{
	vlib_main_t *vm = vlib_get_main ();
	flow_report_main_t *frm = &flow_report_main;
	ipfix_exporter_t *exp;
	vl_api_registration_t *reg;
	ip4_address_t collector, src;
	u16 collector_port = UDP_DST_PORT_ipfix;
	u32 path_mtu;
	u32 template_interval;
	u8 udp_checksum;
	u32 fib_id;
	u32 fib_index = ~0;

	reg = vl_api_client_index_to_registration (client_index);
	if (!reg)
	return VNET_API_ERROR_UNIMPLEMENTED;

	/* Collector address is the key for the exporter lookup */
	ip4_address_decode (mp_collector_address->un.ip4, &collector);

	if (use_index_0)
	/*
	* In this case we update the existing exporter. There is no delete
	* for exp[0]
	*/
	exp = &frm->exporters[0];
	else
	{
	if (is_create)
	{
	exp = vnet_ipfix_exporter_lookup (&collector);
	if (!exp)
	{
	/* Create a new exporter instead of updating an existing one */
	if (pool_elts (frm->exporters) >= IPFIX_EXPORTERS_MAX)
	return VNET_API_ERROR_INVALID_VALUE;
	pool_get (frm->exporters, exp);
	if (!exp)
	return VNET_API_ERROR_INVALID_VALUE;
	}
	}
	else
	{
	/* Delete the exporter */
	exp = vnet_ipfix_exporter_lookup (&collector);
	if (!exp)
	return VNET_API_ERROR_NO_SUCH_ENTRY;

	pool_put (frm->exporters, exp);
	return 0;
	}
	}

	if (mp_src_address->af == ADDRESS_IP6 \|\|
	mp_collector_address->af == ADDRESS_IP6)
	{
	return VNET_API_ERROR_UNIMPLEMENTED;
	}

	collector_port = ntohs (mp_collector_port);
	if (collector_port == (u16) ~ 0)
	collector_port = UDP_DST_PORT_ipfix;
	ip4_address_decode (mp_src_address->un.ip4, &src);
	fib_id = ntohl (mp_vrf_id);

	ip4_main_t *im = &ip4_main;
	if (fib_id == ~0)
	{
	fib_index = ~0;
	}
	else
	{
	uword *p = hash_get (im->fib_index_by_table_id, fib_id);
	if (!p)
	return VNET_API_ERROR_NO_SUCH_FIB;
	fib_index = p[0];
	}

	path_mtu = ntohl (mp_path_mtu);
	if (path_mtu == ~0)
	path_mtu = 512; // RFC 7011 section 10.3.3.
	template_interval = ntohl (mp_template_interval);
	if (template_interval == ~0)
	template_interval = 20;
	udp_checksum = mp_udp_checksum;

	if (collector.as_u32 != 0 && src.as_u32 == 0)
	return VNET_API_ERROR_INVALID_VALUE;

	if (path_mtu > 1450 /* vpp does not support fragmentation */ )
	return VNET_API_ERROR_INVALID_VALUE;

	if (path_mtu < 68)
	return VNET_API_ERROR_INVALID_VALUE;

	/* Reset report streams if we are reconfiguring IP addresses */
	if (exp->ipfix_collector.as_u32 != collector.as_u32 \|\|
	exp->src_address.as_u32 != src.as_u32 \|\|
	exp->collector_port != collector_port)
	vnet_flow_reports_reset (exp);

	exp->ipfix_collector.as_u32 = collector.as_u32;
	exp->collector_port = collector_port;
	exp->src_address.as_u32 = src.as_u32;
	exp->fib_index = fib_index;
	exp->path_mtu = path_mtu;
	exp->template_interval = template_interval;
	exp->udp_checksum = udp_checksum;

	/* Turn on the flow reporting process */
	vlib_process_signal_event (vm, flow_report_process_node.index, 1, 0);

	return 0;
	}

	static void
	vl_api_set_ipfix_exporter_t_handler (vl_api_set_ipfix_exporter_t *mp)
	{
	vl_api_set_ipfix_exporter_reply_t *rmp;
	flow_report_main_t *frm = &flow_report_main;
	int rv = vl_api_set_ipfix_exporter_t_internal (
	mp->client_index, &mp->collector_address, mp->collector_port,
	&mp->src_address, mp->vrf_id, mp->path_mtu, mp->template_interval,
	mp->udp_checksum, USE_INDEX_0, 0);

	REPLY_MACRO (VL_API_SET_IPFIX_EXPORTER_REPLY);
	}

	static void
	vl_api_ipfix_exporter_create_delete_t_handler (
	vl_api_ipfix_exporter_create_delete_t *mp)
	{
	vl_api_ipfix_exporter_create_delete_reply_t *rmp;
	flow_report_main_t *frm = &flow_report_main;
	int rv = vl_api_set_ipfix_exporter_t_internal (
	mp->client_index, &mp->collector_address, mp->collector_port,
	&mp->src_address, mp->vrf_id, mp->path_mtu, mp->template_interval,
	mp->udp_checksum, USE_ANY_INDEX, mp->is_create);

	REPLY_MACRO (VL_API_IPFIX_EXPORTER_CREATE_DELETE_REPLY);
	}

	static void
	vl_api_ipfix_exporter_dump_t_handler (vl_api_ipfix_exporter_dump_t * mp)
	{
	flow_report_main_t *frm = &flow_report_main;
	ipfix_exporter_t *exp = pool_elt_at_index (flow_report_main.exporters, 0);
	vl_api_registration_t *reg;
	vl_api_ipfix_exporter_details_t *rmp;
	ip4_main_t *im = &ip4_main;
	ip46_address_t collector = {.as_u64[0] = 0,.as_u64[1] = 0 };
	ip46_address_t src = {.as_u64[0] = 0,.as_u64[1] = 0 };
	u32 vrf_id;

	reg = vl_api_client_index_to_registration (mp->client_index);
	if (!reg)
	return;

	rmp = vl_msg_api_alloc (sizeof (*rmp));
	clib_memset (rmp, 0, sizeof (*rmp));
	rmp->_vl_msg_id =
	ntohs ((REPLY_MSG_ID_BASE) + VL_API_IPFIX_EXPORTER_DETAILS);
	rmp->context = mp->context;

	memcpy (&collector.ip4, &exp->ipfix_collector, sizeof (ip4_address_t));
	ip_address_encode (&collector, IP46_TYPE_IP4, &rmp->collector_address);

	rmp->collector_port = htons (exp->collector_port);

	memcpy (&src.ip4, &exp->src_address, sizeof (ip4_address_t));
	ip_address_encode (&src, IP46_TYPE_IP4, &rmp->src_address);

	if (exp->fib_index == ~0)
	vrf_id = ~0;
	else
	vrf_id = im->fibs[exp->fib_index].ft_table_id;
	rmp->vrf_id = htonl (vrf_id);
	rmp->path_mtu = htonl (exp->path_mtu);
	rmp->template_interval = htonl (exp->template_interval);
	rmp->udp_checksum = (exp->udp_checksum != 0);

	vl_api_send_msg (reg, (u8 *) rmp);
	}

	static void
	ipfix_all_fill_details (vl_api_ipfix_all_exporter_details_t *rmp,
	ipfix_exporter_t *exp)
	{
	ip4_main_t *im = &ip4_main;
	ip46_address_t collector = { .as_u64[0] = 0, .as_u64[1] = 0 };
	ip46_address_t src = { .as_u64[0] = 0, .as_u64[1] = 0 };
	u32 vrf_id;

	memcpy (&collector.ip4, &exp->ipfix_collector, sizeof (ip4_address_t));
	ip_address_encode (&collector, IP46_TYPE_IP4, &rmp->collector_address);

	rmp->collector_port = htons (exp->collector_port);

	memcpy (&src.ip4, &exp->src_address, sizeof (ip4_address_t));
	ip_address_encode (&src, IP46_TYPE_IP4, &rmp->src_address);

	if (exp->fib_index == ~0)
	vrf_id = ~0;
	else
	vrf_id = im->fibs[exp->fib_index].ft_table_id;
	rmp->vrf_id = htonl (vrf_id);
	rmp->path_mtu = htonl (exp->path_mtu);
	rmp->template_interval = htonl (exp->template_interval);
	rmp->udp_checksum = (exp->udp_checksum != 0);
	}

	static void
	ipfix_all_exporter_details (flow_report_main_t *frm, u32 index,
	vl_api_registration_t *rp, u32 context)
	{
	ipfix_exporter_t *exp = pool_elt_at_index (frm->exporters, index);

	vl_api_ipfix_all_exporter_details_t *rmp;

	REPLY_MACRO_DETAILS4 (VL_API_IPFIX_ALL_EXPORTER_DETAILS, rp, context,
	({ ipfix_all_fill_details (rmp, exp); }));
	}

	static void
	vl_api_ipfix_all_exporter_get_t_handler (vl_api_ipfix_all_exporter_get_t *mp)
	{
	flow_report_main_t *frm = &flow_report_main;
	vl_api_ipfix_all_exporter_get_reply_t *rmp;
	int rv = 0;

	REPLY_AND_DETAILS_MACRO (
	VL_API_IPFIX_ALL_EXPORTER_GET_REPLY, frm->exporters,
	({ ipfix_all_exporter_details (frm, cursor, rp, mp->context); }));
	}

	static void
	vl_api_set_ipfix_classify_stream_t_handler
	(vl_api_set_ipfix_classify_stream_t * mp)
	{
	vl_api_set_ipfix_classify_stream_reply_t *rmp;
	flow_report_classify_main_t *fcm = &flow_report_classify_main;
	flow_report_main_t *frm = &flow_report_main;
	ipfix_exporter_t *exp = &frm->exporters[0];
	u32 domain_id = 0;
	u32 src_port = UDP_DST_PORT_ipfix;
	int rv = 0;

	domain_id = ntohl (mp->domain_id);
	src_port = ntohs (mp->src_port);

	if (fcm->src_port != 0 &&
	(fcm->domain_id != domain_id \|\| fcm->src_port != (u16) src_port))
	{
	int rv = vnet_stream_change (exp, fcm->domain_id, fcm->src_port,
	domain_id, (u16) src_port);
	ASSERT (rv == 0);
	}

	fcm->domain_id = domain_id;
	fcm->src_port = (u16) src_port;

	REPLY_MACRO (VL_API_SET_IPFIX_CLASSIFY_STREAM_REPLY);
	}

	static void
	vl_api_ipfix_classify_stream_dump_t_handler
	(vl_api_ipfix_classify_stream_dump_t * mp)
	{
	flow_report_classify_main_t *fcm = &flow_report_classify_main;
	vl_api_registration_t *reg;
	vl_api_ipfix_classify_stream_details_t *rmp;

	reg = vl_api_client_index_to_registration (mp->client_index);
	if (!reg)
	return;

	rmp = vl_msg_api_alloc (sizeof (*rmp));
	clib_memset (rmp, 0, sizeof (*rmp));
	rmp->_vl_msg_id = ntohs (VL_API_IPFIX_CLASSIFY_STREAM_DETAILS);
	rmp->context = mp->context;
	rmp->domain_id = htonl (fcm->domain_id);
	rmp->src_port = htons (fcm->src_port);

	vl_api_send_msg (reg, (u8 *) rmp);
	}

	static void
	vl_api_ipfix_classify_table_add_del_t_handler
	(vl_api_ipfix_classify_table_add_del_t * mp)
	{
	vl_api_ipfix_classify_table_add_del_reply_t *rmp;
	vl_api_registration_t *reg;
	flow_report_classify_main_t *fcm = &flow_report_classify_main;
	flow_report_main_t *frm = &flow_report_main;
	ipfix_exporter_t *exp = &frm->exporters[0];
	vnet_flow_report_add_del_args_t args;
	ipfix_classify_table_t *table;
	int is_add;
	u32 classify_table_index;
	u8 ip_version;
	u8 transport_protocol;
	int rv = 0;

	reg = vl_api_client_index_to_registration (mp->client_index);
	if (!reg)
	return;

	classify_table_index = ntohl (mp->table_id);
	ip_version = (mp->ip_version == ADDRESS_IP4) ? 4 : 6;
	transport_protocol = mp->transport_protocol;
	is_add = mp->is_add;

	if (fcm->src_port == 0)
	{
	/* call set_ipfix_classify_stream first */
	rv = VNET_API_ERROR_UNSPECIFIED;
	goto out;
	}

	clib_memset (&args, 0, sizeof (args));

	table = 0;
	int i;
	for (i = 0; i < vec_len (fcm->tables); i++)
	if (ipfix_classify_table_index_valid (i))
	if (fcm->tables[i].classify_table_index == classify_table_index)
	{
	table = &fcm->tables[i];
	break;
	}

	if (is_add)
	{
	if (table)
	{
	rv = VNET_API_ERROR_VALUE_EXIST;
	goto out;
	}
	table = ipfix_classify_add_table ();
	table->classify_table_index = classify_table_index;
	}
	else
	{
	if (!table)
	{
	rv = VNET_API_ERROR_NO_SUCH_ENTRY;
	goto out;
	}
	}

	table->ip_version = ip_version;
	table->transport_protocol = transport_protocol;

	args.opaque.as_uword = table - fcm->tables;
	args.rewrite_callback = ipfix_classify_template_rewrite;
	args.flow_data_callback = ipfix_classify_send_flows;
	args.is_add = is_add;
	args.domain_id = fcm->domain_id;
	args.src_port = fcm->src_port;

	rv = vnet_flow_report_add_del (exp, &args, NULL);

	/* If deleting, or add failed */
	if (is_add == 0 \|\| (rv && is_add))
	ipfix_classify_delete_table (table - fcm->tables);

	out:
	REPLY_MACRO (VL_API_SET_IPFIX_CLASSIFY_STREAM_REPLY);
	}

	static void
	send_ipfix_classify_table_details (u32 table_index,
	vl_api_registration_t * reg, u32 context)
	{
	flow_report_classify_main_t *fcm = &flow_report_classify_main;
	vl_api_ipfix_classify_table_details_t *mp;

	ipfix_classify_table_t *table = &fcm->tables[table_index];

	mp = vl_msg_api_alloc (sizeof (*mp));
	clib_memset (mp, 0, sizeof (*mp));
	mp->_vl_msg_id = ntohs (VL_API_IPFIX_CLASSIFY_TABLE_DETAILS);
	mp->context = context;
	mp->table_id = htonl (table->classify_table_index);
	mp->ip_version = (table->ip_version == 4) ? ADDRESS_IP4 : ADDRESS_IP6;
	mp->transport_protocol = table->transport_protocol;

	vl_api_send_msg (reg, (u8 *) mp);
	}

	static void
	vl_api_ipfix_classify_table_dump_t_handler
	(vl_api_ipfix_classify_table_dump_t * mp)
	{
	flow_report_classify_main_t *fcm = &flow_report_classify_main;
	vl_api_registration_t *reg;
	u32 i;

	reg = vl_api_client_index_to_registration (mp->client_index);
	if (!reg)
	return;

	for (i = 0; i < vec_len (fcm->tables); i++)
	if (ipfix_classify_table_index_valid (i))
	send_ipfix_classify_table_details (i, reg, mp->context);
	}

	static void
	vl_api_ipfix_flush_t_handler (vl_api_ipfix_flush_t * mp)
	{
	flow_report_main_t *frm = &flow_report_main;
	vl_api_ipfix_flush_reply_t *rmp;
	vl_api_registration_t *reg;
	vlib_main_t *vm = vlib_get_main ();
	int rv = 0;

	reg = vl_api_client_index_to_registration (mp->client_index);
	if (!reg)
	return;

	/* poke the flow reporting process */
	vlib_process_signal_event (vm, flow_report_process_node.index,
	1 /* type_opaque / , 0 / data */ );

	REPLY_MACRO (VL_API_IPFIX_FLUSH_REPLY);
	}

	#include <vnet/ipfix-export/ipfix_export.api.c>
	static clib_error_t *
	flow_api_hookup (vlib_main_t * vm)
	{
	flow_report_main_t *frm = &flow_report_main;
	/*
	* Set up the (msg_name, crc, message-id) table
	*/
	REPLY_MSG_ID_BASE = setup_message_id_table ();

	return 0;
	}

	VLIB_API_INIT_FUNCTION (flow_api_hookup);

	/*
	* fd.io coding-style-patch-verification: ON
	*
	* Local Variables:
	* eval: (c-set-style "gnu")
	* End:
	*/