src/vnet/pg/stream.c - fdio/vpp - Gitiles

 /*
  * Copyright (c) 2015 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.
  */
 /*
  * pg_stream.c: packet generator streams
  *
  * Copyright (c) 2008 Eliot Dresselhaus
  *
  * Permission is hereby granted, free of charge, to any person obtaining
  * a copy of this software and associated documentation files (the
  * "Software"), to deal in the Software without restriction, including
  * without limitation the rights to use, copy, modify, merge, publish,
  * distribute, sublicense, and/or sell copies of the Software, and to
  * permit persons to whom the Software is furnished to do so, subject to
  * the following conditions:
  *
  * The above copyright notice and this permission notice shall be
  * included in all copies or substantial portions of the Software.
  *
  *  THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
  *  EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  *  MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  *  NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
  *  LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
  *  OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
  *  WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  */

 #include <vnet/vnet.h>
 #include <vnet/pg/pg.h>
 #include <vnet/ethernet/ethernet.h>
 #include <vnet/ip/ip.h>
 #include <vnet/mpls/mpls.h>
 #include <vnet/devices/devices.h>

 /* Mark stream active or inactive. */
 void
 pg_stream_enable_disable (pg_main_t * pg, pg_stream_t * s, int want_enabled)
 {
   vlib_main_t *vm;
   vnet_main_t *vnm = vnet_get_main ();
   pg_interface_t *pi = pool_elt_at_index (pg->interfaces, s->pg_if_index);

   want_enabled = want_enabled != 0;

   if (pg_stream_is_enabled (s) == want_enabled)
     /* No change necessary. */
     return;

   if (want_enabled)
     s->n_packets_generated = 0;

   /* Toggle enabled flag. */
   s->flags ^= PG_STREAM_FLAGS_IS_ENABLED;

   ASSERT (!pool_is_free (pg->streams, s));

   vec_validate (pg->enabled_streams, s->worker_index);
   pg->enabled_streams[s->worker_index] =
     clib_bitmap_set (pg->enabled_streams[s->worker_index], s - pg->streams,
 		     want_enabled);

   if (want_enabled)
     {
       vnet_hw_interface_set_flags (vnm, pi->hw_if_index,
 				   VNET_HW_INTERFACE_FLAG_LINK_UP);

       vnet_sw_interface_set_flags (vnm, pi->sw_if_index,
 				   VNET_SW_INTERFACE_FLAG_ADMIN_UP);
     }

   if (vlib_num_workers ())
     vm = vlib_get_worker_vlib_main (s->worker_index);
   else
     vm = vlib_get_main ();

   vlib_node_set_state (vm, pg_input_node.index,
 		       (clib_bitmap_is_zero
 			(pg->enabled_streams[s->worker_index]) ?
 			VLIB_NODE_STATE_DISABLED : VLIB_NODE_STATE_POLLING));

   s->packet_accumulator = 0;
   s->time_last_generate = 0;
 }

 static u8 *
 format_pg_output_trace (u8 * s, va_list * va)
 {
   CLIB_UNUSED (vlib_main_t * vm) = va_arg (*va, vlib_main_t *);
   CLIB_UNUSED (vlib_node_t * node) = va_arg (*va, vlib_node_t *);
   pg_output_trace_t *t = va_arg (*va, pg_output_trace_t *);
   u32 indent = format_get_indent (s);

   s = format (s, "%Ubuffer 0x%x: %U",
 	      format_white_space, indent,
 	      t->buffer_index, format_vnet_buffer, &t->buffer);

   s = format (s, "\n%U%U", format_white_space, indent,
 	      format_ethernet_header_with_length, t->buffer.pre_data,
 	      sizeof (t->buffer.pre_data));

   return s;
 }

 static u8 *
 format_pg_interface_name (u8 * s, va_list * args)
 {
   pg_main_t *pg = &pg_main;
   u32 if_index = va_arg (*args, u32);
   pg_interface_t *pi;

   pi = pool_elt_at_index (pg->interfaces, if_index);
   s = format (s, "pg%d", pi->id);

   return s;
 }

 static clib_error_t *
 pg_interface_admin_up_down (vnet_main_t * vnm, u32 hw_if_index, u32 flags)
 {
   u32 hw_flags = 0;

   if (flags & VNET_SW_INTERFACE_FLAG_ADMIN_UP)
     hw_flags = VNET_HW_INTERFACE_FLAG_LINK_UP;

   vnet_hw_interface_set_flags (vnm, hw_if_index, hw_flags);

   return 0;
 }

 /* *INDENT-OFF* */
 VNET_DEVICE_CLASS (pg_dev_class) = {
   .name = "pg",
   .tx_function = pg_output,
   .format_device_name = format_pg_interface_name,
   .format_tx_trace = format_pg_output_trace,
   .admin_up_down_function = pg_interface_admin_up_down,
 };
 /* *INDENT-ON* */

 static u8 *
 pg_build_rewrite (vnet_main_t * vnm,
 		  u32 sw_if_index,
 		  vnet_link_t link_type, const void *dst_address)
 {
   u8 *rewrite = NULL;
   u16 *h;

   vec_validate (rewrite, sizeof (*h) - 1);
   h = (u16 *) rewrite;
   h[0] = clib_host_to_net_u16 (vnet_link_to_l3_proto (link_type));

   return (rewrite);
 }

 /* *INDENT-OFF* */
 VNET_HW_INTERFACE_CLASS (pg_interface_class,static) = {
   .name = "Packet generator",
   .build_rewrite = pg_build_rewrite,
 };
 /* *INDENT-ON* */

 static u32
 pg_eth_flag_change (vnet_main_t * vnm, vnet_hw_interface_t * hi, u32 flags)
 {
   /* nothing for now */
   return 0;
 }

 u32
 pg_interface_add_or_get (pg_main_t * pg, uword if_id)
 {
   vnet_main_t *vnm = vnet_get_main ();
   vlib_main_t *vm = vlib_get_main ();
   pg_interface_t *pi;
   vnet_hw_interface_t *hi;
   uword *p;
   u32 i;

   p = hash_get (pg->if_index_by_if_id, if_id);

   if (p)
     {
       return p[0];
     }
   else
     {
       u8 hw_addr[6];
       f64 now = vlib_time_now (vm);
       u32 rnd;

       pool_get (pg->interfaces, pi);
       i = pi - pg->interfaces;

       rnd = (u32) (now * 1e6);
       rnd = random_u32 (&rnd);
       clib_memcpy_fast (hw_addr + 2, &rnd, sizeof (rnd));
       hw_addr[0] = 2;
       hw_addr[1] = 0xfe;

       pi->id = if_id;
       ethernet_register_interface (vnm, pg_dev_class.index, i, hw_addr,
 				   &pi->hw_if_index, pg_eth_flag_change);
       hi = vnet_get_hw_interface (vnm, pi->hw_if_index);
       pi->sw_if_index = hi->sw_if_index;

       hash_set (pg->if_index_by_if_id, if_id, i);

       if (vlib_num_workers ())
 	{
 	  pi->lockp = clib_mem_alloc_aligned (CLIB_CACHE_LINE_BYTES,
 					      CLIB_CACHE_LINE_BYTES);
 	  *pi->lockp = 0;
 	}
     }

   return i;
 }

 static void
 do_edit (pg_stream_t * stream,
 	 pg_edit_group_t * g, pg_edit_t * e, uword want_commit)
 {
   u32 i, i0, i1, mask, n_bits_left;
   u8 *v, *s, *m;

   i0 = e->lsb_bit_offset / BITS (u8);

   /* Make space for edit in value and mask. */
   vec_validate (g->fixed_packet_data, i0);
   vec_validate (g->fixed_packet_data_mask, i0);

   if (e->type != PG_EDIT_FIXED)
     {
       switch (e->type)
 	{
 	case PG_EDIT_RANDOM:
 	case PG_EDIT_INCREMENT:
 	  e->last_increment_value = pg_edit_get_value (e, PG_EDIT_LO);
 	  break;

 	default:
 	  break;
 	}

       if (want_commit)
 	{
 	  ASSERT (e->type != PG_EDIT_INVALID_TYPE);
 	  vec_add1 (g->non_fixed_edits, e[0]);
 	}
       return;
     }

   s = g->fixed_packet_data;
   m = g->fixed_packet_data_mask;

   n_bits_left = e->n_bits;
   i0 = e->lsb_bit_offset / BITS (u8);
   i1 = e->lsb_bit_offset % BITS (u8);

   v = e->values[PG_EDIT_LO];
   i = pg_edit_n_alloc_bytes (e) - 1;

   /* Odd low order bits?. */
   if (i1 != 0 && n_bits_left > 0)
     {
       u32 n = clib_min (n_bits_left, BITS (u8) - i1);

       mask = pow2_mask (n) << i1;

       ASSERT (i0 < vec_len (s));
       ASSERT (i < vec_len (v));
       ASSERT ((v[i] & ~mask) == 0);

       s[i0] |= v[i] & mask;
       m[i0] |= mask;

       i0--;
       i--;
       n_bits_left -= n;
     }

   /* Even bytes. */
   while (n_bits_left >= 8)
     {
       ASSERT (i0 < vec_len (s));
       ASSERT (i < vec_len (v));

       s[i0] = v[i];
       m[i0] = ~0;

       i0--;
       i--;
       n_bits_left -= 8;
     }

   /* Odd high order bits. */
   if (n_bits_left > 0)
     {
       mask = pow2_mask (n_bits_left);

       ASSERT (i0 < vec_len (s));
       ASSERT (i < vec_len (v));
       ASSERT ((v[i] & ~mask) == 0);

       s[i0] |= v[i] & mask;
       m[i0] |= mask;
     }

   if (want_commit)
     pg_edit_free (e);
 }

 void
 pg_edit_group_get_fixed_packet_data (pg_stream_t * s,
 				     u32 group_index,
 				     void *packet_data,
 				     void *packet_data_mask)
 {
   pg_edit_group_t *g = pg_stream_get_group (s, group_index);
   pg_edit_t *e;

   vec_foreach (e, g->edits) do_edit (s, g, e, /* want_commit */ 0);

   clib_memcpy_fast (packet_data, g->fixed_packet_data,
 		    vec_len (g->fixed_packet_data));
   clib_memcpy_fast (packet_data_mask, g->fixed_packet_data_mask,
 		    vec_len (g->fixed_packet_data_mask));
 }

 static void
 perform_fixed_edits (pg_stream_t * s)
 {
   pg_edit_group_t *g;
   pg_edit_t *e;
   word i;

   for (i = vec_len (s->edit_groups) - 1; i >= 0; i--)
     {
       g = vec_elt_at_index (s->edit_groups, i);
       vec_foreach (e, g->edits) do_edit (s, g, e, /* want_commit */ 1);

       /* All edits have either been performed or added to
          g->non_fixed_edits.  So, we can delete the vector. */
       vec_free (g->edits);
     }

   vec_free (s->fixed_packet_data_mask);
   vec_free (s->fixed_packet_data);
   vec_foreach (g, s->edit_groups)
   {
     int i;
     g->start_byte_offset = vec_len (s->fixed_packet_data);

     /* Relocate and copy non-fixed edits from group to stream. */
     vec_foreach (e, g->non_fixed_edits)
       e->lsb_bit_offset += g->start_byte_offset * BITS (u8);

     for (i = 0; i < vec_len (g->non_fixed_edits); i++)
       ASSERT (g->non_fixed_edits[i].type != PG_EDIT_INVALID_TYPE);

     vec_add (s->non_fixed_edits,
 	     g->non_fixed_edits, vec_len (g->non_fixed_edits));
     vec_free (g->non_fixed_edits);

     vec_add (s->fixed_packet_data,
 	     g->fixed_packet_data, vec_len (g->fixed_packet_data));
     vec_add (s->fixed_packet_data_mask,
 	     g->fixed_packet_data_mask, vec_len (g->fixed_packet_data_mask));
   }
 }

 void
 pg_stream_add (pg_main_t * pg, pg_stream_t * s_init)
 {
   vlib_main_t *vm = vlib_get_main ();
   pg_stream_t *s;
   uword *p;

   if (!pg->stream_index_by_name)
     pg->stream_index_by_name
       = hash_create_vec (0, sizeof (s->name[0]), sizeof (uword));

   /* Delete any old stream with the same name. */
   if (s_init->name
       && (p = hash_get_mem (pg->stream_index_by_name, s_init->name)))
     {
       pg_stream_del (pg, p[0]);
     }

   pool_get (pg->streams, s);
   s[0] = s_init[0];

   /* Give it a name. */
   if (!s->name)
     s->name = format (0, "stream%d", s - pg->streams);
   else
     s->name = vec_dup (s->name);

   hash_set_mem (pg->stream_index_by_name, s->name, s - pg->streams);

   /* Get fixed part of buffer data. */
   if (s->edit_groups)
     perform_fixed_edits (s);

   /* Determine packet size. */
   switch (s->packet_size_edit_type)
     {
     case PG_EDIT_INCREMENT:
     case PG_EDIT_RANDOM:
       if (s->min_packet_bytes == s->max_packet_bytes)
 	s->packet_size_edit_type = PG_EDIT_FIXED;
       break;

     default:
       /* Get packet size from fixed edits. */
       s->packet_size_edit_type = PG_EDIT_FIXED;
       if (!s->replay_packet_templates)
 	s->min_packet_bytes = s->max_packet_bytes =
 	  vec_len (s->fixed_packet_data);
       break;
     }

   s->last_increment_packet_size = s->min_packet_bytes;

   {
     int n;

     s->buffer_bytes = vlib_buffer_get_default_data_size (vm);
     n = s->max_packet_bytes / s->buffer_bytes;
     n += (s->max_packet_bytes % s->buffer_bytes) != 0;

     vec_resize (s->buffer_indices, n);
   }

   /* Find an interface to use. */
   s->pg_if_index = pg_interface_add_or_get (pg, s->if_id);

   if (s->sw_if_index[VLIB_RX] == ~0)
     {
       pg_interface_t *pi = pool_elt_at_index (pg->interfaces, s->pg_if_index);
       /*
        * Default the RX interface if unset. It's a bad mistake to
        * set [VLIB_TX] prior to ip lookup, since the ip lookup code
        * interprets [VLIB_TX] as a fib index...
        */
       s->sw_if_index[VLIB_RX] = pi->sw_if_index;
     }

   /* Connect the graph. */
   s->next_index = vlib_node_add_next (vm, device_input_node.index,
 				      s->node_index);
 }

 void
 pg_stream_del (pg_main_t * pg, uword index)
 {
   pg_stream_t *s;
   pg_buffer_index_t *bi;

   s = pool_elt_at_index (pg->streams, index);

   pg_stream_enable_disable (pg, s, /* want_enabled */ 0);
   hash_unset_mem (pg->stream_index_by_name, s->name);

   vec_foreach (bi, s->buffer_indices)
   {
     clib_fifo_free (bi->buffer_fifo);
   }

   pg_stream_free (s);
   pool_put (pg->streams, s);
 }

 void
 pg_stream_change (pg_main_t * pg, pg_stream_t * s)
 {
   /* Determine packet size. */
   switch (s->packet_size_edit_type)
     {
     case PG_EDIT_INCREMENT:
     case PG_EDIT_RANDOM:
       if (s->min_packet_bytes == s->max_packet_bytes)
 	s->packet_size_edit_type = PG_EDIT_FIXED;
     case PG_EDIT_FIXED:
       break;

     default:
       /* Get packet size from fixed edits. */
       s->packet_size_edit_type = PG_EDIT_FIXED;
       if (!s->replay_packet_templates)
 	s->min_packet_bytes = s->max_packet_bytes =
 	  vec_len (s->fixed_packet_data);
       break;
     }

   s->last_increment_packet_size = s->min_packet_bytes;
 }


 /*
  * fd.io coding-style-patch-verification: ON
  *
  * Local Variables:
  * eval: (c-set-style "gnu")
  * End:
  */
	/*
	* Copyright (c) 2015 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.
	*/
	/*
	* pg_stream.c: packet generator streams
	*
	* Copyright (c) 2008 Eliot Dresselhaus
	*
	* Permission is hereby granted, free of charge, to any person obtaining
	* a copy of this software and associated documentation files (the
	* "Software"), to deal in the Software without restriction, including
	* without limitation the rights to use, copy, modify, merge, publish,
	* distribute, sublicense, and/or sell copies of the Software, and to
	* permit persons to whom the Software is furnished to do so, subject to
	* the following conditions:
	*
	* The above copyright notice and this permission notice shall be
	* included in all copies or substantial portions of the Software.
	*
	* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
	* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
	* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
	* NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
	* LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
	* OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
	* WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
	*/

	#include <vnet/vnet.h>
	#include <vnet/pg/pg.h>
	#include <vnet/ethernet/ethernet.h>
	#include <vnet/ip/ip.h>
	#include <vnet/mpls/mpls.h>
	#include <vnet/devices/devices.h>

	/* Mark stream active or inactive. */
	void
	pg_stream_enable_disable (pg_main_t * pg, pg_stream_t * s, int want_enabled)
	{
	vlib_main_t *vm;
	vnet_main_t *vnm = vnet_get_main ();
	pg_interface_t *pi = pool_elt_at_index (pg->interfaces, s->pg_if_index);

	want_enabled = want_enabled != 0;

	if (pg_stream_is_enabled (s) == want_enabled)
	/* No change necessary. */
	return;

	if (want_enabled)
	s->n_packets_generated = 0;

	/* Toggle enabled flag. */
	s->flags ^= PG_STREAM_FLAGS_IS_ENABLED;

	ASSERT (!pool_is_free (pg->streams, s));

	vec_validate (pg->enabled_streams, s->worker_index);
	pg->enabled_streams[s->worker_index] =
	clib_bitmap_set (pg->enabled_streams[s->worker_index], s - pg->streams,
	want_enabled);

	if (want_enabled)
	{
	vnet_hw_interface_set_flags (vnm, pi->hw_if_index,
	VNET_HW_INTERFACE_FLAG_LINK_UP);

	vnet_sw_interface_set_flags (vnm, pi->sw_if_index,
	VNET_SW_INTERFACE_FLAG_ADMIN_UP);
	}

	if (vlib_num_workers ())
	vm = vlib_get_worker_vlib_main (s->worker_index);
	else
	vm = vlib_get_main ();

	vlib_node_set_state (vm, pg_input_node.index,
	(clib_bitmap_is_zero
	(pg->enabled_streams[s->worker_index]) ?
	VLIB_NODE_STATE_DISABLED : VLIB_NODE_STATE_POLLING));

	s->packet_accumulator = 0;
	s->time_last_generate = 0;
	}

	static u8 *
	format_pg_output_trace (u8 * s, va_list * va)
	{
	CLIB_UNUSED (vlib_main_t * vm) = va_arg (va, vlib_main_t );
	CLIB_UNUSED (vlib_node_t * node) = va_arg (va, vlib_node_t );
	pg_output_trace_t t = va_arg (va, pg_output_trace_t *);
	u32 indent = format_get_indent (s);

	s = format (s, "%Ubuffer 0x%x: %U",
	format_white_space, indent,
	t->buffer_index, format_vnet_buffer, &t->buffer);

	s = format (s, "\n%U%U", format_white_space, indent,
	format_ethernet_header_with_length, t->buffer.pre_data,
	sizeof (t->buffer.pre_data));

	return s;
	}

	static u8 *
	format_pg_interface_name (u8 * s, va_list * args)
	{
	pg_main_t *pg = &pg_main;
	u32 if_index = va_arg (*args, u32);
	pg_interface_t *pi;

	pi = pool_elt_at_index (pg->interfaces, if_index);
	s = format (s, "pg%d", pi->id);

	return s;
	}

	static clib_error_t *
	pg_interface_admin_up_down (vnet_main_t * vnm, u32 hw_if_index, u32 flags)
	{
	u32 hw_flags = 0;

	if (flags & VNET_SW_INTERFACE_FLAG_ADMIN_UP)
	hw_flags = VNET_HW_INTERFACE_FLAG_LINK_UP;

	vnet_hw_interface_set_flags (vnm, hw_if_index, hw_flags);

	return 0;
	}

	/* INDENT-OFF */
	VNET_DEVICE_CLASS (pg_dev_class) = {
	.name = "pg",
	.tx_function = pg_output,
	.format_device_name = format_pg_interface_name,
	.format_tx_trace = format_pg_output_trace,
	.admin_up_down_function = pg_interface_admin_up_down,
	};
	/* INDENT-ON */

	static u8 *
	pg_build_rewrite (vnet_main_t * vnm,
	u32 sw_if_index,
	vnet_link_t link_type, const void *dst_address)
	{
	u8 *rewrite = NULL;
	u16 *h;

	vec_validate (rewrite, sizeof (*h) - 1);
	h = (u16 *) rewrite;
	h[0] = clib_host_to_net_u16 (vnet_link_to_l3_proto (link_type));

	return (rewrite);
	}

	/* INDENT-OFF */
	VNET_HW_INTERFACE_CLASS (pg_interface_class,static) = {
	.name = "Packet generator",
	.build_rewrite = pg_build_rewrite,
	};
	/* INDENT-ON */

	static u32
	pg_eth_flag_change (vnet_main_t * vnm, vnet_hw_interface_t * hi, u32 flags)
	{
	/* nothing for now */
	return 0;
	}

	u32
	pg_interface_add_or_get (pg_main_t * pg, uword if_id)
	{
	vnet_main_t *vnm = vnet_get_main ();
	vlib_main_t *vm = vlib_get_main ();
	pg_interface_t *pi;
	vnet_hw_interface_t *hi;
	uword *p;
	u32 i;

	p = hash_get (pg->if_index_by_if_id, if_id);

	if (p)
	{
	return p[0];
	}
	else
	{
	u8 hw_addr[6];
	f64 now = vlib_time_now (vm);
	u32 rnd;

	pool_get (pg->interfaces, pi);
	i = pi - pg->interfaces;

	rnd = (u32) (now * 1e6);
	rnd = random_u32 (&rnd);
	clib_memcpy_fast (hw_addr + 2, &rnd, sizeof (rnd));
	hw_addr[0] = 2;
	hw_addr[1] = 0xfe;

	pi->id = if_id;
	ethernet_register_interface (vnm, pg_dev_class.index, i, hw_addr,
	&pi->hw_if_index, pg_eth_flag_change);
	hi = vnet_get_hw_interface (vnm, pi->hw_if_index);
	pi->sw_if_index = hi->sw_if_index;

	hash_set (pg->if_index_by_if_id, if_id, i);

	if (vlib_num_workers ())
	{
	pi->lockp = clib_mem_alloc_aligned (CLIB_CACHE_LINE_BYTES,
	CLIB_CACHE_LINE_BYTES);
	*pi->lockp = 0;
	}
	}

	return i;
	}

	static void
	do_edit (pg_stream_t * stream,
	pg_edit_group_t * g, pg_edit_t * e, uword want_commit)
	{
	u32 i, i0, i1, mask, n_bits_left;
	u8 v, s, *m;

	i0 = e->lsb_bit_offset / BITS (u8);

	/* Make space for edit in value and mask. */
	vec_validate (g->fixed_packet_data, i0);
	vec_validate (g->fixed_packet_data_mask, i0);

	if (e->type != PG_EDIT_FIXED)
	{
	switch (e->type)
	{
	case PG_EDIT_RANDOM:
	case PG_EDIT_INCREMENT:
	e->last_increment_value = pg_edit_get_value (e, PG_EDIT_LO);
	break;

	default:
	break;
	}

	if (want_commit)
	{
	ASSERT (e->type != PG_EDIT_INVALID_TYPE);
	vec_add1 (g->non_fixed_edits, e[0]);
	}
	return;
	}

	s = g->fixed_packet_data;
	m = g->fixed_packet_data_mask;

	n_bits_left = e->n_bits;
	i0 = e->lsb_bit_offset / BITS (u8);
	i1 = e->lsb_bit_offset % BITS (u8);

	v = e->values[PG_EDIT_LO];
	i = pg_edit_n_alloc_bytes (e) - 1;

	/* Odd low order bits?. */
	if (i1 != 0 && n_bits_left > 0)
	{
	u32 n = clib_min (n_bits_left, BITS (u8) - i1);

	mask = pow2_mask (n) << i1;

	ASSERT (i0 < vec_len (s));
	ASSERT (i < vec_len (v));
	ASSERT ((v[i] & ~mask) == 0);

	s[i0] \|= v[i] & mask;
	m[i0] \|= mask;

	i0--;
	i--;
	n_bits_left -= n;
	}

	/* Even bytes. */
	while (n_bits_left >= 8)
	{
	ASSERT (i0 < vec_len (s));
	ASSERT (i < vec_len (v));

	s[i0] = v[i];
	m[i0] = ~0;

	i0--;
	i--;
	n_bits_left -= 8;
	}

	/* Odd high order bits. */
	if (n_bits_left > 0)
	{
	mask = pow2_mask (n_bits_left);

	ASSERT (i0 < vec_len (s));
	ASSERT (i < vec_len (v));
	ASSERT ((v[i] & ~mask) == 0);

	s[i0] \|= v[i] & mask;
	m[i0] \|= mask;
	}

	if (want_commit)
	pg_edit_free (e);
	}

	void
	pg_edit_group_get_fixed_packet_data (pg_stream_t * s,
	u32 group_index,
	void *packet_data,
	void *packet_data_mask)
	{
	pg_edit_group_t *g = pg_stream_get_group (s, group_index);
	pg_edit_t *e;

	vec_foreach (e, g->edits) do_edit (s, g, e, /* want_commit */ 0);

	clib_memcpy_fast (packet_data, g->fixed_packet_data,
	vec_len (g->fixed_packet_data));
	clib_memcpy_fast (packet_data_mask, g->fixed_packet_data_mask,
	vec_len (g->fixed_packet_data_mask));
	}

	static void
	perform_fixed_edits (pg_stream_t * s)
	{
	pg_edit_group_t *g;
	pg_edit_t *e;
	word i;

	for (i = vec_len (s->edit_groups) - 1; i >= 0; i--)
	{
	g = vec_elt_at_index (s->edit_groups, i);
	vec_foreach (e, g->edits) do_edit (s, g, e, /* want_commit */ 1);

	/* All edits have either been performed or added to
	g->non_fixed_edits. So, we can delete the vector. */
	vec_free (g->edits);
	}

	vec_free (s->fixed_packet_data_mask);
	vec_free (s->fixed_packet_data);
	vec_foreach (g, s->edit_groups)
	{
	int i;
	g->start_byte_offset = vec_len (s->fixed_packet_data);

	/* Relocate and copy non-fixed edits from group to stream. */
	vec_foreach (e, g->non_fixed_edits)
	e->lsb_bit_offset += g->start_byte_offset * BITS (u8);

	for (i = 0; i < vec_len (g->non_fixed_edits); i++)
	ASSERT (g->non_fixed_edits[i].type != PG_EDIT_INVALID_TYPE);

	vec_add (s->non_fixed_edits,
	g->non_fixed_edits, vec_len (g->non_fixed_edits));
	vec_free (g->non_fixed_edits);

	vec_add (s->fixed_packet_data,
	g->fixed_packet_data, vec_len (g->fixed_packet_data));
	vec_add (s->fixed_packet_data_mask,
	g->fixed_packet_data_mask, vec_len (g->fixed_packet_data_mask));
	}
	}

	void
	pg_stream_add (pg_main_t * pg, pg_stream_t * s_init)
	{
	vlib_main_t *vm = vlib_get_main ();
	pg_stream_t *s;
	uword *p;

	if (!pg->stream_index_by_name)
	pg->stream_index_by_name
	= hash_create_vec (0, sizeof (s->name[0]), sizeof (uword));

	/* Delete any old stream with the same name. */
	if (s_init->name
	&& (p = hash_get_mem (pg->stream_index_by_name, s_init->name)))
	{
	pg_stream_del (pg, p[0]);
	}

	pool_get (pg->streams, s);
	s[0] = s_init[0];

	/* Give it a name. */
	if (!s->name)
	s->name = format (0, "stream%d", s - pg->streams);
	else
	s->name = vec_dup (s->name);

	hash_set_mem (pg->stream_index_by_name, s->name, s - pg->streams);

	/* Get fixed part of buffer data. */
	if (s->edit_groups)
	perform_fixed_edits (s);

	/* Determine packet size. */
	switch (s->packet_size_edit_type)
	{
	case PG_EDIT_INCREMENT:
	case PG_EDIT_RANDOM:
	if (s->min_packet_bytes == s->max_packet_bytes)
	s->packet_size_edit_type = PG_EDIT_FIXED;
	break;

	default:
	/* Get packet size from fixed edits. */
	s->packet_size_edit_type = PG_EDIT_FIXED;
	if (!s->replay_packet_templates)
	s->min_packet_bytes = s->max_packet_bytes =
	vec_len (s->fixed_packet_data);
	break;
	}

	s->last_increment_packet_size = s->min_packet_bytes;

	{
	int n;

	s->buffer_bytes = vlib_buffer_get_default_data_size (vm);
	n = s->max_packet_bytes / s->buffer_bytes;
	n += (s->max_packet_bytes % s->buffer_bytes) != 0;

	vec_resize (s->buffer_indices, n);
	}

	/* Find an interface to use. */
	s->pg_if_index = pg_interface_add_or_get (pg, s->if_id);

	if (s->sw_if_index[VLIB_RX] == ~0)
	{
	pg_interface_t *pi = pool_elt_at_index (pg->interfaces, s->pg_if_index);
	/*
	* Default the RX interface if unset. It's a bad mistake to
	* set [VLIB_TX] prior to ip lookup, since the ip lookup code
	* interprets [VLIB_TX] as a fib index...
	*/
	s->sw_if_index[VLIB_RX] = pi->sw_if_index;
	}

	/* Connect the graph. */
	s->next_index = vlib_node_add_next (vm, device_input_node.index,
	s->node_index);
	}

	void
	pg_stream_del (pg_main_t * pg, uword index)
	{
	pg_stream_t *s;
	pg_buffer_index_t *bi;

	s = pool_elt_at_index (pg->streams, index);

	pg_stream_enable_disable (pg, s, /* want_enabled */ 0);
	hash_unset_mem (pg->stream_index_by_name, s->name);

	vec_foreach (bi, s->buffer_indices)
	{
	clib_fifo_free (bi->buffer_fifo);
	}

	pg_stream_free (s);
	pool_put (pg->streams, s);
	}

	void
	pg_stream_change (pg_main_t * pg, pg_stream_t * s)
	{
	/* Determine packet size. */
	switch (s->packet_size_edit_type)
	{
	case PG_EDIT_INCREMENT:
	case PG_EDIT_RANDOM:
	if (s->min_packet_bytes == s->max_packet_bytes)
	s->packet_size_edit_type = PG_EDIT_FIXED;
	case PG_EDIT_FIXED:
	break;

	default:
	/* Get packet size from fixed edits. */
	s->packet_size_edit_type = PG_EDIT_FIXED;
	if (!s->replay_packet_templates)
	s->min_packet_bytes = s->max_packet_bytes =
	vec_len (s->fixed_packet_data);
	break;
	}

	s->last_increment_packet_size = s->min_packet_bytes;
	}


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