vnet/vnet/pg/edit.c

   1 /*
   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:
   6  *
   7  *     http://www.apache.org/licenses/LICENSE-2.0
   8  *
   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.
  14  */
  15 /*
  16  * pg_edit.c: packet generator edits
  17  *
  18  * Copyright (c) 2008 Eliot Dresselhaus
  19  *
  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:
  27  *
  28  * The above copyright notice and this permission notice shall be
  29  * included in all copies or substantial portions of the Software.
  30  *
  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.
  38  */
  39
  40 #include <vlib/vlib.h>
  41 #include <vnet/pg/pg.h>
  42
  43 static void
  44 pg_edit_set_value_helper (pg_edit_t * e, u64 value, u8 * result)
  45 {
  46   int i, j, n_bits_left;
  47   u8 * v, tmp[8];
  48
  49   v = tmp;
  50
  51   n_bits_left = e->n_bits;
  52   i = 0;
  53   j = e->lsb_bit_offset % BITS (v[0]);
  54
  55   if (n_bits_left > 0 && j != 0)
  56     {
  57       v[i] = (value & 0xff) << j;
  58       value >>= BITS (v[0]) - j;
  59       n_bits_left -= BITS (v[0]) - j;
  60       i += 1;
  61     }
  62
  63   while (n_bits_left > 0)
  64     {
  65       v[i] = value & 0xff;
  66       value >>= 8;
  67       n_bits_left -= 8;
  68       i += 1;
  69     }
  70
  71   /* Convert to network byte order. */
  72   for (j = 0; j < i; j++)
  73     result[j] = v[i - 1 - j];
  74 }
  75
  76 void
  77 pg_edit_set_value (pg_edit_t * e, int hi_or_lo, u64 value)
  78 {
  79   pg_edit_alloc_value (e, hi_or_lo);
  80   pg_edit_set_value_helper (e, value, e->values[hi_or_lo]);
  81 }
  82
  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)
  85 {
  86   u8 * result = va_arg (*args, u8 *);
  87   pg_edit_t * e = va_arg (*args, pg_edit_t *);
  88   u64 value;
  89
  90   ASSERT (BITS (value) >= e->n_bits);
  91
  92   if (! unformat (input, "0x%X", sizeof (value), &value)
  93       && ! unformat (input, "%D", sizeof (value), &value))
  94     return 0;
  95
  96   /* Number given does not fit into bit field. */
  97   if (e->n_bits < 64
  98       && value >= (u64) 1 << (u64) e->n_bits)
  99     return 0;
 100
 101   pg_edit_set_value_helper (e, value, result);
 102   return 1;
 103 }
 104
 105 uword
 106 unformat_pg_edit (unformat_input_t * input, va_list * args)
 107 {
 108   unformat_function_t * f = va_arg (*args, unformat_function_t *);
 109   pg_edit_t * e = va_arg (*args, pg_edit_t *);
 110
 111   pg_edit_alloc_value (e, PG_EDIT_LO);
 112   if (! unformat_user (input, f, e->values[PG_EDIT_LO], e))
 113     return 0;
 114
 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;
 120   else
 121     e->type = PG_EDIT_FIXED;
 122
 123   return 1;
 124 }
 125
 126 uword
 127 unformat_pg_payload (unformat_input_t * input, va_list * args)
 128 {
 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;
 132   pg_edit_t * e;
 133   u32 i, node_index, len, max_len;
 134   u8 * v;
 135
 136   v = 0;
 137
 138   if (unformat (input, "incrementing %d", &len))
 139     {
 140       vec_resize (v, len);
 141       for (i = 0; i < len; i++)
 142         v[i] = i;
 143     }
 144   else if (unformat (input, "hex 0x%U", unformat_hex_string, &v))
 145     ;
 146
 147   else if (unformat (input, "%U", unformat_vlib_node, vm, &node_index))
 148     {
 149       pg_node_t * pn = pg_get_node (node_index);
 150       if (! pn->unformat_edit)
 151         return 0;
 152       return unformat (input, "%U", pn->unformat_edit, s);
 153     }
 154
 155   else
 156     return 0;
 157
 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)
 161     {
 162       if (s->max_packet_bytes >= max_len)
 163         _vec_len (v) = s->max_packet_bytes - max_len;
 164       else
 165         _vec_len (v) = 0;
 166     }
 167
 168   e = pg_create_edit_group (s, sizeof (e[0]), vec_len (v), 0);
 169
 170   e->type = PG_EDIT_FIXED;
 171   e->n_bits = vec_len (v) * BITS (v[0]);
 172
 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]);
 175
 176   e->values[PG_EDIT_LO] = v;
 177
 178   return 1;
 179 }