2 * Copyright (c) 2015 Cisco and/or its affiliates.
3 * Licensed under the Apache License, Version 2.0 (the "License");
4 * you may not use this file except in compliance with the License.
5 * You may obtain a copy of the License at:
7 * http://www.apache.org/licenses/LICENSE-2.0
9 * Unless required by applicable law or agreed to in writing, software
10 * distributed under the License is distributed on an "AS IS" BASIS,
11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 * See the License for the specific language governing permissions and
13 * limitations under the License.
16 * pg_edit.c: packet generator edits
18 * Copyright (c) 2008 Eliot Dresselhaus
20 * Permission is hereby granted, free of charge, to any person obtaining
21 * a copy of this software and associated documentation files (the
22 * "Software"), to deal in the Software without restriction, including
23 * without limitation the rights to use, copy, modify, merge, publish,
24 * distribute, sublicense, and/or sell copies of the Software, and to
25 * permit persons to whom the Software is furnished to do so, subject to
26 * the following conditions:
28 * The above copyright notice and this permission notice shall be
29 * included in all copies or substantial portions of the Software.
31 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
32 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
33 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
34 * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
35 * LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
36 * OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
37 * WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
40 #include <vlib/vlib.h>
41 #include <vnet/pg/pg.h>
44 pg_edit_set_value_helper (pg_edit_t * e, u64 value, u8 * result)
46 int i, j, n_bits_left;
51 n_bits_left = e->n_bits;
53 j = e->lsb_bit_offset % BITS (v[0]);
55 if (n_bits_left > 0 && j != 0)
57 v[i] = (value & 0xff) << j;
58 value >>= BITS (v[0]) - j;
59 n_bits_left -= BITS (v[0]) - j;
63 while (n_bits_left > 0)
71 /* Convert to network byte order. */
72 for (j = 0; j < i; j++)
73 result[j] = v[i - 1 - j];
77 pg_edit_set_value (pg_edit_t * e, int hi_or_lo, u64 value)
79 pg_edit_alloc_value (e, hi_or_lo);
80 pg_edit_set_value_helper (e, value, e->values[hi_or_lo]);
83 /* Parse an int either %d or 0x%x into network byte order. */
84 uword unformat_pg_number (unformat_input_t * input, va_list * args)
86 u8 * result = va_arg (*args, u8 *);
87 pg_edit_t * e = va_arg (*args, pg_edit_t *);
90 ASSERT (BITS (value) >= e->n_bits);
92 if (! unformat (input, "0x%X", sizeof (value), &value)
93 && ! unformat (input, "%D", sizeof (value), &value))
96 /* Number given does not fit into bit field. */
98 && value >= (u64) 1 << (u64) e->n_bits)
101 pg_edit_set_value_helper (e, value, result);
106 unformat_pg_edit (unformat_input_t * input, va_list * args)
108 unformat_function_t * f = va_arg (*args, unformat_function_t *);
109 pg_edit_t * e = va_arg (*args, pg_edit_t *);
111 pg_edit_alloc_value (e, PG_EDIT_LO);
112 if (! unformat_user (input, f, e->values[PG_EDIT_LO], e))
115 pg_edit_alloc_value (e, PG_EDIT_HI);
116 if (unformat (input, "-%U", f, e->values[PG_EDIT_HI], e))
117 e->type = PG_EDIT_INCREMENT;
118 else if (unformat (input, "+%U", f, e->values[PG_EDIT_HI], e))
119 e->type = PG_EDIT_RANDOM;
121 e->type = PG_EDIT_FIXED;
127 unformat_pg_payload (unformat_input_t * input, va_list * args)
129 pg_stream_t * s = va_arg (*args, pg_stream_t *);
130 pg_main_t * pg = &pg_main;
131 vlib_main_t * vm = pg->vlib_main;
133 u32 i, node_index, len, max_len;
138 if (unformat (input, "incrementing %d", &len))
141 for (i = 0; i < len; i++)
144 else if (unformat (input, "hex 0x%U", unformat_hex_string, &v))
147 else if (unformat (input, "%U", unformat_vlib_node, vm, &node_index))
149 pg_node_t * pn = pg_get_node (node_index);
150 if (! pn->unformat_edit)
152 return unformat (input, "%U", pn->unformat_edit, s);
158 /* Length not including this payload. */
159 max_len = pg_edit_group_n_bytes (s, 0);
160 if (max_len + vec_len (v) >= s->max_packet_bytes)
162 if (s->max_packet_bytes >= max_len)
163 _vec_len (v) = s->max_packet_bytes - max_len;
168 e = pg_create_edit_group (s, sizeof (e[0]), vec_len (v), 0);
170 e->type = PG_EDIT_FIXED;
171 e->n_bits = vec_len (v) * BITS (v[0]);
173 /* Least significant bit is at end of bitstream, since everything is always bigendian. */
174 e->lsb_bit_offset = e->n_bits - BITS (v[0]);
176 e->values[PG_EDIT_LO] = v;