ip: fix the punt redirect for ip4

Type: fix

Change-Id: I39341f201209931392f315ead5adfddd8b567caf
Signed-off-by: Mohsin Kazmi <sykazmi@cisco.com>
diff --git a/src/vnet/ip/ip4_punt_drop.c b/src/vnet/ip/ip4_punt_drop.c
index aee84b4..65b2401 100644
--- a/src/vnet/ip/ip4_punt_drop.c
+++ b/src/vnet/ip/ip4_punt_drop.c
@@ -334,10 +334,10 @@
 		       vlib_cli_command_t * cmd)
 {
   unformat_input_t _line_input, *line_input = &_line_input;
-  fib_route_path_t *rpaths = NULL, rpath;
-  dpo_proto_t payload_proto;
+  ip46_address_t nh = { 0 };
   clib_error_t *error = 0;
   u32 rx_sw_if_index = ~0;
+  u32 tx_sw_if_index = ~0;
   vnet_main_t *vnm;
   u8 is_add;
 
@@ -358,9 +358,13 @@
       else if (unformat (line_input, "rx %U",
 			 unformat_vnet_sw_interface, vnm, &rx_sw_if_index))
 	;
+      else if (unformat (line_input, "via %U %U",
+			 unformat_ip4_address, unformat_vnet_sw_interface,
+			 &nh.ip4, vnm, &tx_sw_if_index))
+	;
       else if (unformat (line_input, "via %U",
-			 unformat_fib_route_path, &rpath, &payload_proto))
-	vec_add1 (rpaths, rpath);
+			 unformat_vnet_sw_interface, vnm, &tx_sw_if_index))
+	;
       else
 	{
 	  error = unformat_parse_error (line_input);
@@ -376,8 +380,7 @@
 
   if (is_add)
     {
-      if (vec_len (rpaths))
-	ip4_punt_redirect_add_paths (rx_sw_if_index, rpaths);
+      ip4_punt_redirect_add (rx_sw_if_index, tx_sw_if_index, &nh);
     }
   else
     {