1 files changed, 270 insertions, 0 deletions

diff --git a/vnet/vnet/rewrite.c b/vnet/vnet/rewrite.c
new file mode 100644
index 00000000000..6c33a2ebcd8
--- /dev/null
+++ b/vnet/vnet/rewrite.c
@@ -0,0 +1,270 @@
+/*
+ * 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.
+ */
+/*
+ * rewrite.c: packet rewrite
+ *
+ * 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/ip/lookup.h>
+
+void vnet_rewrite_copy_slow_path (vnet_rewrite_data_t * p0,
+				  vnet_rewrite_data_t * rw0,
+				  word n_left,
+				  uword most_likely_size)
+{
+  uword n_done = round_pow2 (most_likely_size, sizeof (rw0[0])) / sizeof (rw0[0]);
+
+  p0 -= n_done;
+  rw0 -= n_done;
+
+  /* As we enter the cleanup loop, p0 and rw0 point to the last chunk written
+     by the fast path. Hence, the constant 1, which the 
+     vnet_rewrite_copy_one macro renders as p0[-1] = rw0[-1]. */
+
+  while (n_left > 0)
+    {
+      vnet_rewrite_copy_one (p0, rw0, 1);
+      p0--;
+      rw0--;
+      n_left--;
+    }
+}
+
+u8 * format_vnet_rewrite (u8 * s, va_list * args)
+{
+  vlib_main_t * vm = va_arg (*args, vlib_main_t *);
+  vnet_rewrite_header_t * rw = va_arg (*args, vnet_rewrite_header_t *);
+  u32 max_data_bytes = va_arg (*args, u32);
+  vnet_main_t * vnm = vnet_get_main();
+  vlib_node_t * next;
+  uword indent;
+
+  next = vlib_get_next_node (vm, rw->node_index, rw->next_index);
+
+  indent = format_get_indent (s);
+
+  if (rw->sw_if_index != ~0)
+    {
+      vnet_sw_interface_t * si;
+      si = vnet_get_sw_interface (vnm, rw->sw_if_index);
+      s = format (s, "%U", format_vnet_sw_interface_name, vnm, si);
+    }
+  else
+    s = format (s, "%v", next->name);
+
+  /* Format rewrite string. */
+  if (rw->data_bytes > 0)
+    s = format (s, "\n%U%U",
+		format_white_space, indent,
+		next->format_buffer ? next->format_buffer : format_hex_bytes,
+		rw->data + max_data_bytes - rw->data_bytes,
+		rw->data_bytes);
+
+  return s;
+}
+
+u8 * format_vnet_rewrite_header (u8 * s, va_list * args)
+{
+  vlib_main_t * vm = va_arg (*args, vlib_main_t *);
+  vnet_rewrite_header_t * rw = va_arg (*args, vnet_rewrite_header_t *);
+  u8 * packet_data = va_arg (*args, u8 *);
+  u32 packet_data_bytes = va_arg (*args, u32);
+  vlib_node_t * next;
+
+  next = vlib_get_next_node (vm, rw->node_index, rw->next_index);
+
+  /* Format rewrite string. */
+  s = format (s, "%U",
+	      next->format_buffer ? next->format_buffer : format_hex_bytes,
+	      packet_data, packet_data_bytes);
+
+  return s;
+}
+
+uword unformat_vnet_rewrite (unformat_input_t * input, va_list * args)
+{
+  vlib_main_t * vm = va_arg (*args, vlib_main_t *);
+  vnet_rewrite_header_t * rw = va_arg (*args, vnet_rewrite_header_t *);
+  u32 max_data_bytes = va_arg (*args, u32);
+  vnet_main_t * vnm = vnet_get_main();
+  vlib_node_t * next;
+  u32 next_index, sw_if_index, max_packet_bytes, error;
+  u8 * rw_data;
+
+  rw_data = 0;
+  sw_if_index = ~0;
+  max_packet_bytes = ~0;
+  error = 1;
+
+  /* Parse sw interface. */
+  if (unformat (input, "%U",
+		unformat_vnet_sw_interface, vnm, &sw_if_index))
+    {
+      vnet_hw_interface_t * hi;
+
+      hi = vnet_get_sup_hw_interface (vnm, sw_if_index);
+
+      next_index = hi->output_node_index;
+      max_packet_bytes = hi->max_l3_packet_bytes[VLIB_RX];
+    }
+
+  else if (unformat (input, "%U",
+		     unformat_vlib_node, vm, &next_index))
+    ;
+
+  else
+    goto done;
+
+  next = vlib_get_node (vm, next_index);
+
+  if (next->unformat_buffer
+      && unformat_user (input, next->unformat_buffer, &rw_data))
+    ;
+
+  else if (unformat_user (input, unformat_hex_string, &rw_data)
+	   || unformat (input, "0x%U", unformat_hex_string, &rw_data))
+    ;
+      
+  else
+    goto done;
+
+  /* Re-write does not fit. */
+  if (vec_len (rw_data) >= max_data_bytes)
+    goto done;
+
+  {
+    u32 tmp;
+
+    if (unformat (input, "mtu %d", &tmp)
+	&& tmp < (1 << BITS (rw->max_l3_packet_bytes)))
+      max_packet_bytes = tmp;
+  }
+
+  error = 0;
+  rw->sw_if_index = sw_if_index;
+  rw->max_l3_packet_bytes = max_packet_bytes;
+  rw->next_index = vlib_node_add_next (vm, rw->node_index, next_index);
+  vnet_rewrite_set_data_internal (rw, max_data_bytes, rw_data, vec_len (rw_data));
+
+ done:
+  vec_free (rw_data);
+  return error == 0;
+}
+
+void vnet_rewrite_for_sw_interface (vnet_main_t * vnm,
+				    vnet_l3_packet_type_t packet_type,
+				    u32 sw_if_index,
+				    u32 node_index,
+				    void * dst_address,
+				    vnet_rewrite_header_t * rw,
+				    u32 max_rewrite_bytes)
+{
+  vnet_hw_interface_t * hw = vnet_get_sup_hw_interface (vnm, sw_if_index);
+  vnet_hw_interface_class_t * hc = vnet_get_hw_interface_class (vnm, hw->hw_class_index);
+  static u8 * rw_tmp = 0;
+  uword n_rw_tmp;
+
+  rw->sw_if_index = sw_if_index;
+  rw->node_index = node_index;
+  rw->next_index = vlib_node_add_next (vnm->vlib_main, node_index, hw->output_node_index);
+  rw->max_l3_packet_bytes = hw->max_l3_packet_bytes[VLIB_TX];
+
+  ASSERT (max_rewrite_bytes > 0);
+  vec_reset_length (rw_tmp);
+  vec_validate (rw_tmp, max_rewrite_bytes - 1);
+
+  ASSERT (hc->set_rewrite);
+  n_rw_tmp = hc->set_rewrite (vnm, sw_if_index, packet_type, dst_address, rw_tmp, max_rewrite_bytes);
+
+  ASSERT (n_rw_tmp >= 0 && n_rw_tmp < max_rewrite_bytes);
+  vnet_rewrite_set_data_internal (rw, max_rewrite_bytes, rw_tmp, n_rw_tmp);
+}
+
+void vnet_rewrite_for_tunnel (vnet_main_t * vnm,
+                              u32 tx_sw_if_index,
+                              u32 rewrite_node_index,
+                              u32 post_rewrite_node_index,
+                              vnet_rewrite_header_t * rw,
+                              u8 *rewrite_data,
+                              u32 rewrite_length)
+{
+  ip_adjacency_t * adj = 0;
+  /* 
+   * Installed into vnet_buffer(b)->sw_if_index[VLIB_TX] e.g.
+   * by ip4_rewrite_inline. If the post-rewrite node injects into 
+   * ipX-forward, this will be interpreted as a FIB number. 
+   */
+  rw->sw_if_index = tx_sw_if_index; 
+  rw->node_index = rewrite_node_index;
+  rw->next_index = vlib_node_add_next (vnm->vlib_main, rewrite_node_index, 
+                                       post_rewrite_node_index);
+  rw->max_l3_packet_bytes = (u16) ~0; /* we can't know at this point */
+
+  ASSERT (rewrite_length < sizeof (adj->rewrite_data));
+  /* Leave room for ethernet + VLAN tag */
+  vnet_rewrite_set_data_internal (rw, sizeof(adj->rewrite_data), 
+                                  rewrite_data, rewrite_length);
+}
+
+void serialize_vnet_rewrite (serialize_main_t * m, va_list * va)
+{
+  vnet_rewrite_header_t * rw = va_arg (*va, vnet_rewrite_header_t *);
+  u32 max_data_bytes = va_arg (*va, u32);
+  u8 * p;
+
+  serialize_integer (m, rw->sw_if_index, sizeof (rw->sw_if_index));
+  serialize_integer (m, rw->data_bytes, sizeof (rw->data_bytes));
+  serialize_integer (m, rw->max_l3_packet_bytes, sizeof (rw->max_l3_packet_bytes));
+  p = serialize_get (m, rw->data_bytes);
+  memcpy (p, vnet_rewrite_get_data_internal (rw, max_data_bytes), rw->data_bytes);
+}
+
+void unserialize_vnet_rewrite (serialize_main_t * m, va_list * va)
+{
+  vnet_rewrite_header_t * rw = va_arg (*va, vnet_rewrite_header_t *);
+  u32 max_data_bytes = va_arg (*va, u32);
+  u8 * p;
+
+  /* It is up to user to fill these in. */
+  rw->node_index = ~0;
+  rw->next_index = ~0;
+
+  unserialize_integer (m, &rw->sw_if_index, sizeof (rw->sw_if_index));
+  unserialize_integer (m, &rw->data_bytes, sizeof (rw->data_bytes));
+  unserialize_integer (m, &rw->max_l3_packet_bytes, sizeof (rw->max_l3_packet_bytes));
+  p = unserialize_get (m, rw->data_bytes);
+  memcpy (vnet_rewrite_get_data_internal (rw, max_data_bytes), p, rw->data_bytes);
+}