eaf5d29d48bd80cd12e03c030c82e3cde355b363
[vpp.git] / src / plugins / ping / ping.c
1 /*
2  * Copyright (c) 2016 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 #include <stddef.h>
17
18 #include <vlib/vlib.h>
19 #include <vnet/fib/ip6_fib.h>
20 #include <vnet/fib/ip4_fib.h>
21 #include <vnet/fib/fib_entry.h>
22 #include <vnet/plugin/plugin.h>
23 #include <vpp/app/version.h>
24
25 #include <vnet/ip/icmp4.h>
26 #include <ping/ping.h>
27
28 ping_main_t ping_main;
29
30 /**
31  * @file
32  * @brief IPv4 and IPv6 ICMP Ping.
33  *
34  * This file contains code to support IPv4 or IPv6 ICMP ECHO_REQUEST to
35  * network hosts.
36  *
37  */
38
39 typedef struct
40 {
41   u16 id;
42   u16 seq;
43   u32 cli_process_node;
44   u8 is_ip6;
45 } icmp_echo_trace_t;
46
47
48 u8 *
49 format_icmp_echo_trace (u8 * s, va_list * va)
50 {
51   CLIB_UNUSED (vlib_main_t * vm) = va_arg (*va, vlib_main_t *);
52   CLIB_UNUSED (vlib_node_t * node) = va_arg (*va, vlib_node_t *);
53   icmp_echo_trace_t *t = va_arg (*va, icmp_echo_trace_t *);
54
55   s =
56     format (s, "ICMP%s echo id %d seq %d", t->is_ip6 ? "6" : "4", t->id,
57             t->seq);
58   if (t->cli_process_node == PING_CLI_UNKNOWN_NODE)
59     {
60       s = format (s, " (unknown)");
61     }
62   else
63     {
64       s = format (s, " send to cli node %d", t->cli_process_node);
65     }
66
67   return s;
68 }
69
70
71 static u8 *
72 format_ip46_ping_result (u8 * s, va_list * args)
73 {
74   send_ip46_ping_result_t res = va_arg (*args, send_ip46_ping_result_t);
75
76   switch (res)
77     {
78 #define _(v, n) case SEND_PING_##v: s = format(s, "%s", n);break;
79       foreach_ip46_ping_result
80 #undef _
81     }
82
83   return (s);
84 }
85
86
87 /*
88  * Poor man's get-set-clear functions
89  * for manipulation of icmp_id -> cli_process_id
90  * mappings.
91  *
92  * There should normally be very few (0..1..2) of these
93  * mappings, so the linear search is a good strategy.
94  *
95  * Make them thread-safe via a simple spinlock.
96  *
97  */
98
99
100 static_always_inline uword
101 get_cli_process_id_by_icmp_id_mt (vlib_main_t * vm, u16 icmp_id)
102 {
103   ping_main_t *pm = &ping_main;
104   uword cli_process_id = PING_CLI_UNKNOWN_NODE;
105   ping_run_t *pr;
106
107   clib_spinlock_lock_if_init (&pm->ping_run_check_lock);
108   vec_foreach (pr, pm->active_ping_runs)
109   {
110     if (pr->icmp_id == icmp_id)
111       {
112         cli_process_id = pr->cli_process_id;
113         break;
114       }
115   }
116   clib_spinlock_unlock_if_init (&pm->ping_run_check_lock);
117   return cli_process_id;
118 }
119
120
121 static_always_inline void
122 set_cli_process_id_by_icmp_id_mt (vlib_main_t * vm, u16 icmp_id,
123                                   uword cli_process_id)
124 {
125   ping_main_t *pm = &ping_main;
126   ping_run_t *pr;
127
128   clib_spinlock_lock_if_init (&pm->ping_run_check_lock);
129   vec_foreach (pr, pm->active_ping_runs)
130   {
131     if (pr->icmp_id == icmp_id)
132       {
133         pr->cli_process_id = cli_process_id;
134         goto have_found_and_set;
135       }
136   }
137   /* no such key yet - add a new one */
138   ping_run_t new_pr = {.icmp_id = icmp_id,.cli_process_id = cli_process_id };
139   vec_add1 (pm->active_ping_runs, new_pr);
140 have_found_and_set:
141   clib_spinlock_unlock_if_init (&pm->ping_run_check_lock);
142 }
143
144
145 static_always_inline void
146 clear_cli_process_id_by_icmp_id_mt (vlib_main_t * vm, u16 icmp_id)
147 {
148   ping_main_t *pm = &ping_main;
149   ping_run_t *pr;
150
151   clib_spinlock_lock_if_init (&pm->ping_run_check_lock);
152   vec_foreach (pr, pm->active_ping_runs)
153   {
154     if (pr->icmp_id == icmp_id)
155       {
156         vec_del1 (pm->active_ping_runs, pm->active_ping_runs - pr);
157         break;
158       }
159   }
160   clib_spinlock_unlock_if_init (&pm->ping_run_check_lock);
161 }
162
163 static_always_inline int
164 ip46_get_icmp_id_and_seq (vlib_main_t * vm, vlib_buffer_t * b0,
165                           u16 * out_icmp_id, u16 * out_icmp_seq, int is_ip6)
166 {
167   int l4_offset;
168   if (is_ip6)
169     {
170       ip6_header_t *ip6 = vlib_buffer_get_current (b0);
171       if (ip6->protocol != IP_PROTOCOL_ICMP6)
172         {
173           return 0;
174         }
175       l4_offset = sizeof (*ip6);        // IPv6 EH
176     }
177   else
178     {
179       ip4_header_t *ip4 = vlib_buffer_get_current (b0);
180       l4_offset = ip4_header_bytes (ip4);
181
182     }
183   icmp46_header_t *icmp46 = vlib_buffer_get_current (b0) + l4_offset;
184   icmp46_echo_request_t *icmp46_echo = (icmp46_echo_request_t *) (icmp46 + 1);
185
186   *out_icmp_id = clib_net_to_host_u16 (icmp46_echo->id);
187   *out_icmp_seq = clib_net_to_host_u16 (icmp46_echo->seq);
188   return 1;
189 }
190
191 /*
192  * post the buffer to a given cli process node - the caller should forget bi0 after return.
193  */
194
195 static_always_inline void
196 ip46_post_icmp_reply_event (vlib_main_t * vm, uword cli_process_id, u32 bi0,
197                             int is_ip6)
198 {
199   vlib_buffer_t *b0 = vlib_get_buffer (vm, bi0);
200   u64 nowts = clib_cpu_time_now ();
201
202   /* Pass the timestamp to the cli_process thanks to the vnet_buffer unused metadata field */
203
204   /* Camping on unused data... just ensure statically that there is enough space */
205   STATIC_ASSERT (ARRAY_LEN (vnet_buffer (b0)->unused) *
206                  sizeof (vnet_buffer (b0)->unused[0]) > sizeof (nowts),
207                  "ping reply timestamp fits within remaining space of vnet_buffer unused data");
208   u64 *pnowts = (void *) &vnet_buffer (b0)->unused[0];
209   *pnowts = nowts;
210
211   u32 event_id = is_ip6 ? PING_RESPONSE_IP6 : PING_RESPONSE_IP4;
212   vlib_process_signal_event_mt (vm, cli_process_id, event_id, bi0);
213 }
214
215
216 static_always_inline void
217 ip46_echo_reply_maybe_trace_buffer (vlib_main_t * vm,
218                                     vlib_node_runtime_t * node,
219                                     uword cli_process_id, u16 id, u16 seq,
220                                     vlib_buffer_t * b0, int is_ip6)
221 {
222   if (PREDICT_FALSE (b0->flags & VLIB_BUFFER_IS_TRACED))
223     {
224       icmp_echo_trace_t *tr = vlib_add_trace (vm, node, b0, sizeof (*tr));
225       tr->id = id;
226       tr->seq = seq;
227       tr->cli_process_node = cli_process_id;
228       tr->is_ip6 = is_ip6;
229     }
230 }
231
232
233 static_always_inline uword
234 ip46_icmp_echo_reply_inner_node_fn (vlib_main_t * vm,
235                                     vlib_node_runtime_t * node,
236                                     vlib_frame_t * frame, int do_trace,
237                                     int is_ip6)
238 {
239   u32 n_left_from, *from, *to_next;
240   icmp46_echo_reply_next_t next_index;
241
242   from = vlib_frame_vector_args (frame);
243   n_left_from = frame->n_vectors;
244
245   next_index = node->cached_next_index;
246
247   while (n_left_from > 0)
248     {
249       u32 n_left_to_next;
250       vlib_get_next_frame (vm, node, next_index, to_next, n_left_to_next);
251
252       while (n_left_from > 0 && n_left_to_next > 0)
253         {
254           u32 bi0;
255           vlib_buffer_t *b0;
256           /*
257            * The buffers (replies) are either posted to the CLI thread
258            * awaiting for them for subsequent analysis and disposal,
259            * or are sent to the punt node.
260            *
261            * So the only "next" node is a punt, normally.
262            */
263           u32 next0 = ICMP46_ECHO_REPLY_NEXT_PUNT;
264
265           bi0 = from[0];
266           b0 = vlib_get_buffer (vm, bi0);
267           from += 1;
268           n_left_from -= 1;
269
270           u16 icmp_id = ~0;
271           u16 icmp_seq = ~0;
272           uword cli_process_id = PING_CLI_UNKNOWN_NODE;
273
274           if (ip46_get_icmp_id_and_seq (vm, b0, &icmp_id, &icmp_seq, is_ip6))
275             {
276               cli_process_id = get_cli_process_id_by_icmp_id_mt (vm, icmp_id);
277             }
278
279           if (do_trace)
280             ip46_echo_reply_maybe_trace_buffer (vm, node, cli_process_id,
281                                                 icmp_id, icmp_seq, b0,
282                                                 is_ip6);
283
284           if (~0 == cli_process_id)
285             {
286               /* no outstanding requests for this reply, punt */
287               /* speculatively enqueue b0 to the current next frame */
288               to_next[0] = bi0;
289               to_next += 1;
290               n_left_to_next -= 1;
291               /* verify speculative enqueue, maybe switch current next frame */
292               vlib_validate_buffer_enqueue_x1 (vm, node, next_index,
293                                                to_next, n_left_to_next,
294                                                bi0, next0);
295             }
296           else
297             {
298               /* Post the buffer to CLI thread. It will take care of freeing it. */
299               ip46_post_icmp_reply_event (vm, cli_process_id, bi0, is_ip6);
300             }
301         }
302       vlib_put_next_frame (vm, node, next_index, n_left_to_next);
303     }
304   return frame->n_vectors;
305 }
306
307 /*
308  * select "with-trace" or "without-trace" codepaths upfront.
309  */
310 static_always_inline uword
311 ip46_icmp_echo_reply_outer_node_fn (vlib_main_t * vm,
312                                     vlib_node_runtime_t * node,
313                                     vlib_frame_t * frame, int is_ip6)
314 {
315   if (node->flags & VLIB_NODE_FLAG_TRACE)
316     return ip46_icmp_echo_reply_inner_node_fn (vm, node, frame,
317                                                1 /* do_trace */ , is_ip6);
318   else
319     return ip46_icmp_echo_reply_inner_node_fn (vm, node, frame,
320                                                0 /* do_trace */ , is_ip6);
321 }
322
323 static uword
324 ip4_icmp_echo_reply_node_fn (vlib_main_t * vm,
325                              vlib_node_runtime_t * node, vlib_frame_t * frame)
326 {
327   return ip46_icmp_echo_reply_outer_node_fn (vm, node, frame,
328                                              0 /* is_ip6 */ );
329 }
330
331 static uword
332 ip6_icmp_echo_reply_node_fn (vlib_main_t * vm,
333                              vlib_node_runtime_t * node, vlib_frame_t * frame)
334 {
335   return ip46_icmp_echo_reply_outer_node_fn (vm, node, frame,
336                                              1 /* is_ip6 */ );
337 }
338
339 /* *INDENT-OFF* */
340 VLIB_REGISTER_NODE (ip6_icmp_echo_reply_node, static) =
341 {
342   .function = ip6_icmp_echo_reply_node_fn,
343   .name = "ip6-icmp-echo-reply",
344   .vector_size = sizeof (u32),
345   .format_trace = format_icmp_echo_trace,
346   .n_next_nodes = ICMP46_ECHO_REPLY_N_NEXT,
347   .next_nodes = {
348     [ICMP46_ECHO_REPLY_NEXT_DROP] = "ip6-drop",
349     [ICMP46_ECHO_REPLY_NEXT_PUNT] = "ip6-punt",
350   },
351 };
352
353 VLIB_REGISTER_NODE (ip4_icmp_echo_reply_node, static) =
354 {
355   .function = ip4_icmp_echo_reply_node_fn,
356   .name = "ip4-icmp-echo-reply",
357   .vector_size = sizeof (u32),
358   .format_trace = format_icmp_echo_trace,
359   .n_next_nodes = ICMP46_ECHO_REPLY_N_NEXT,
360   .next_nodes = {
361     [ICMP46_ECHO_REPLY_NEXT_DROP] = "ip4-drop",
362     [ICMP46_ECHO_REPLY_NEXT_PUNT] = "ip4-punt",
363   },
364 };
365 /* *INDENT-ON* */
366
367 static uword
368 ip4_icmp_echo_request (vlib_main_t * vm,
369                        vlib_node_runtime_t * node, vlib_frame_t * frame)
370 {
371   uword n_packets = frame->n_vectors;
372   u32 *from, *to_next;
373   u32 n_left_from, n_left_to_next, next;
374   ip4_main_t *i4m = &ip4_main;
375   u16 *fragment_ids, *fid;
376   u8 host_config_ttl = i4m->host_config.ttl;
377
378   from = vlib_frame_vector_args (frame);
379   n_left_from = n_packets;
380   next = node->cached_next_index;
381
382   if (node->flags & VLIB_NODE_FLAG_TRACE)
383     vlib_trace_frame_buffers_only (vm, node, from, frame->n_vectors,
384                                    /* stride */ 1,
385                                    sizeof (icmp_input_trace_t));
386
387   /* Get random fragment IDs for replies. */
388   fid = fragment_ids = clib_random_buffer_get_data (&vm->random_buffer,
389                                                     n_packets *
390                                                     sizeof (fragment_ids[0]));
391
392   while (n_left_from > 0)
393     {
394       vlib_get_next_frame (vm, node, next, to_next, n_left_to_next);
395
396       while (n_left_from > 2 && n_left_to_next > 2)
397         {
398           vlib_buffer_t *p0, *p1;
399           ip4_header_t *ip0, *ip1;
400           icmp46_header_t *icmp0, *icmp1;
401           u32 bi0, src0, dst0;
402           u32 bi1, src1, dst1;
403           ip_csum_t sum0, sum1;
404
405           bi0 = to_next[0] = from[0];
406           bi1 = to_next[1] = from[1];
407
408           from += 2;
409           n_left_from -= 2;
410           to_next += 2;
411           n_left_to_next -= 2;
412
413           p0 = vlib_get_buffer (vm, bi0);
414           p1 = vlib_get_buffer (vm, bi1);
415           ip0 = vlib_buffer_get_current (p0);
416           ip1 = vlib_buffer_get_current (p1);
417           icmp0 = ip4_next_header (ip0);
418           icmp1 = ip4_next_header (ip1);
419
420           vnet_buffer (p0)->sw_if_index[VLIB_RX] =
421             vnet_main.local_interface_sw_if_index;
422           vnet_buffer (p1)->sw_if_index[VLIB_RX] =
423             vnet_main.local_interface_sw_if_index;
424
425           /* Update ICMP checksum. */
426           sum0 = icmp0->checksum;
427           sum1 = icmp1->checksum;
428
429           ASSERT (icmp0->type == ICMP4_echo_request);
430           ASSERT (icmp1->type == ICMP4_echo_request);
431           sum0 = ip_csum_update (sum0, ICMP4_echo_request, ICMP4_echo_reply,
432                                  icmp46_header_t, type);
433           sum1 = ip_csum_update (sum1, ICMP4_echo_request, ICMP4_echo_reply,
434                                  icmp46_header_t, type);
435           icmp0->type = ICMP4_echo_reply;
436           icmp1->type = ICMP4_echo_reply;
437
438           icmp0->checksum = ip_csum_fold (sum0);
439           icmp1->checksum = ip_csum_fold (sum1);
440
441           src0 = ip0->src_address.data_u32;
442           src1 = ip1->src_address.data_u32;
443           dst0 = ip0->dst_address.data_u32;
444           dst1 = ip1->dst_address.data_u32;
445
446           /* Swap source and destination address.
447              Does not change checksum. */
448           ip0->src_address.data_u32 = dst0;
449           ip1->src_address.data_u32 = dst1;
450           ip0->dst_address.data_u32 = src0;
451           ip1->dst_address.data_u32 = src1;
452
453           /* Update IP checksum. */
454           sum0 = ip0->checksum;
455           sum1 = ip1->checksum;
456
457           sum0 = ip_csum_update (sum0, ip0->ttl, host_config_ttl,
458                                  ip4_header_t, ttl);
459           sum1 = ip_csum_update (sum1, ip1->ttl, host_config_ttl,
460                                  ip4_header_t, ttl);
461           ip0->ttl = host_config_ttl;
462           ip1->ttl = host_config_ttl;
463
464           /* New fragment id. */
465           sum0 = ip_csum_update (sum0, ip0->fragment_id, fid[0],
466                                  ip4_header_t, fragment_id);
467           sum1 = ip_csum_update (sum1, ip1->fragment_id, fid[1],
468                                  ip4_header_t, fragment_id);
469           ip0->fragment_id = fid[0];
470           ip1->fragment_id = fid[1];
471           fid += 2;
472
473           ip0->checksum = ip_csum_fold (sum0);
474           ip1->checksum = ip_csum_fold (sum1);
475
476           ASSERT (ip0->checksum == ip4_header_checksum (ip0));
477           ASSERT (ip1->checksum == ip4_header_checksum (ip1));
478
479           p0->flags |= VNET_BUFFER_F_LOCALLY_ORIGINATED;
480           p1->flags |= VNET_BUFFER_F_LOCALLY_ORIGINATED;
481         }
482
483       while (n_left_from > 0 && n_left_to_next > 0)
484         {
485           vlib_buffer_t *p0;
486           ip4_header_t *ip0;
487           icmp46_header_t *icmp0;
488           u32 bi0, src0, dst0;
489           ip_csum_t sum0;
490
491           bi0 = to_next[0] = from[0];
492
493           from += 1;
494           n_left_from -= 1;
495           to_next += 1;
496           n_left_to_next -= 1;
497
498           p0 = vlib_get_buffer (vm, bi0);
499           ip0 = vlib_buffer_get_current (p0);
500           icmp0 = ip4_next_header (ip0);
501
502           vnet_buffer (p0)->sw_if_index[VLIB_RX] =
503             vnet_main.local_interface_sw_if_index;
504
505           /* Update ICMP checksum. */
506           sum0 = icmp0->checksum;
507
508           ASSERT (icmp0->type == ICMP4_echo_request);
509           sum0 = ip_csum_update (sum0, ICMP4_echo_request, ICMP4_echo_reply,
510                                  icmp46_header_t, type);
511           icmp0->type = ICMP4_echo_reply;
512           icmp0->checksum = ip_csum_fold (sum0);
513
514           src0 = ip0->src_address.data_u32;
515           dst0 = ip0->dst_address.data_u32;
516           ip0->src_address.data_u32 = dst0;
517           ip0->dst_address.data_u32 = src0;
518
519           /* Update IP checksum. */
520           sum0 = ip0->checksum;
521
522           sum0 = ip_csum_update (sum0, ip0->ttl, host_config_ttl,
523                                  ip4_header_t, ttl);
524           ip0->ttl = host_config_ttl;
525
526           sum0 = ip_csum_update (sum0, ip0->fragment_id, fid[0],
527                                  ip4_header_t, fragment_id);
528           ip0->fragment_id = fid[0];
529           fid += 1;
530
531           ip0->checksum = ip_csum_fold (sum0);
532
533           ASSERT (ip0->checksum == ip4_header_checksum (ip0));
534
535           p0->flags |= VNET_BUFFER_F_LOCALLY_ORIGINATED;
536         }
537
538       vlib_put_next_frame (vm, node, next, n_left_to_next);
539     }
540
541   vlib_error_count (vm, ip4_icmp_input_node.index,
542                     ICMP4_ERROR_ECHO_REPLIES_SENT, frame->n_vectors);
543
544   return frame->n_vectors;
545 }
546
547 static u8 *
548 format_icmp_input_trace (u8 * s, va_list * va)
549 {
550   CLIB_UNUSED (vlib_main_t * vm) = va_arg (*va, vlib_main_t *);
551   CLIB_UNUSED (vlib_node_t * node) = va_arg (*va, vlib_node_t *);
552   icmp_input_trace_t *t = va_arg (*va, icmp_input_trace_t *);
553
554   s = format (s, "%U",
555               format_ip4_header, t->packet_data, sizeof (t->packet_data));
556
557   return s;
558 }
559
560 /* *INDENT-OFF* */
561 VLIB_REGISTER_NODE (ip4_icmp_echo_request_node,static) = {
562   .function = ip4_icmp_echo_request,
563   .name = "ip4-icmp-echo-request",
564
565   .vector_size = sizeof (u32),
566
567   .format_trace = format_icmp_input_trace,
568
569   .n_next_nodes = 1,
570   .next_nodes = {
571     [0] = "ip4-load-balance",
572   },
573 };
574 /* *INDENT-ON* */
575
576 /*
577  * A swarm of address-family agnostic helper functions
578  * for building and sending the ICMP echo request.
579  *
580  * Deliberately mostly "static" rather than "static inline"
581  * so one can trace them sanely if needed in debugger, if needed.
582  *
583  */
584
585 static_always_inline u8
586 get_icmp_echo_payload_byte (int offset)
587 {
588   return (offset % 256);
589 }
590
591 /* Fill in the ICMP ECHO structure, return the safety-checked and possibly shrunk data_len */
592 static u16
593 init_icmp46_echo_request (vlib_main_t * vm, vlib_buffer_t * b0,
594                           int l4_header_offset,
595                           icmp46_echo_request_t * icmp46_echo, u16 seq_host,
596                           u16 id_host, u64 now, u16 data_len)
597 {
598   int i;
599
600
601   int l34_len =
602     l4_header_offset + sizeof (icmp46_header_t) +
603     offsetof (icmp46_echo_request_t, data);
604   int max_data_len = vlib_buffer_get_default_data_size (vm) - l34_len;
605
606   int first_buf_data_len = data_len < max_data_len ? data_len : max_data_len;
607
608   int payload_offset = 0;
609   for (i = 0; i < first_buf_data_len; i++)
610     icmp46_echo->data[i] = get_icmp_echo_payload_byte (payload_offset++);
611
612   /* inspired by vlib_buffer_add_data */
613   vlib_buffer_t *hb = b0;
614   int remaining_data_len = data_len - first_buf_data_len;
615   while (remaining_data_len)
616     {
617       int this_buf_data_len =
618         remaining_data_len <
619         vlib_buffer_get_default_data_size (vm) ? remaining_data_len :
620         vlib_buffer_get_default_data_size (vm);
621       int n_alloc = vlib_buffer_alloc (vm, &b0->next_buffer, 1);
622       if (n_alloc < 1)
623         {
624           /* That is how much we have so far - return it... */
625           return (data_len - remaining_data_len);
626         }
627       b0->flags |= VLIB_BUFFER_NEXT_PRESENT;
628       /* move on to the newly acquired buffer */
629       b0 = vlib_get_buffer (vm, b0->next_buffer);
630       /* initialize the data */
631       for (i = 0; i < this_buf_data_len; i++)
632         {
633           b0->data[i] = get_icmp_echo_payload_byte (payload_offset++);
634         }
635       b0->current_length = this_buf_data_len;
636       b0->current_data = 0;
637       remaining_data_len -= this_buf_data_len;
638     }
639   hb->flags |= VLIB_BUFFER_TOTAL_LENGTH_VALID;
640   hb->current_length = l34_len + first_buf_data_len;
641   hb->total_length_not_including_first_buffer = data_len - first_buf_data_len;
642
643   icmp46_echo->time_sent = now;
644   icmp46_echo->seq = clib_host_to_net_u16 (seq_host);
645   icmp46_echo->id = clib_host_to_net_u16 (id_host);
646   return data_len;
647 }
648
649
650 static u32
651 ip46_fib_index_from_table_id (u32 table_id, int is_ip6)
652 {
653   u32 fib_index = is_ip6 ?
654     ip6_fib_index_from_table_id (table_id) :
655     ip4_fib_index_from_table_id (table_id);
656   return fib_index;
657 }
658
659 static fib_node_index_t
660 ip46_fib_table_lookup_host (u32 fib_index, ip46_address_t * pa46, int is_ip6)
661 {
662   fib_node_index_t fib_entry_index = is_ip6 ?
663     ip6_fib_table_lookup (fib_index, &pa46->ip6, 128) :
664     ip4_fib_table_lookup (ip4_fib_get (fib_index), &pa46->ip4, 32);
665   return fib_entry_index;
666 }
667
668 static u32
669 ip46_get_resolving_interface (u32 fib_index, ip46_address_t * pa46,
670                               int is_ip6)
671 {
672   u32 sw_if_index = ~0;
673   if (~0 != fib_index)
674     {
675       fib_node_index_t fib_entry_index;
676       fib_entry_index = ip46_fib_table_lookup_host (fib_index, pa46, is_ip6);
677       sw_if_index = fib_entry_get_resolving_interface (fib_entry_index);
678     }
679   return sw_if_index;
680 }
681
682 static u32
683 ip46_fib_table_get_index_for_sw_if_index (u32 sw_if_index, int is_ip6)
684 {
685   u32 fib_table_index = is_ip6 ?
686     ip6_fib_table_get_index_for_sw_if_index (sw_if_index) :
687     ip4_fib_table_get_index_for_sw_if_index (sw_if_index);
688   return fib_table_index;
689
690 }
691
692
693 static int
694 ip46_fill_l3_header (ip46_address_t * pa46, vlib_buffer_t * b0, int is_ip6)
695 {
696   if (is_ip6)
697     {
698       ip6_header_t *ip6 = vlib_buffer_get_current (b0);
699       /* Fill in ip6 header fields */
700       ip6->ip_version_traffic_class_and_flow_label =
701         clib_host_to_net_u32 (0x6 << 28);
702       ip6->payload_length = 0;  /* will be set later */
703       ip6->protocol = IP_PROTOCOL_ICMP6;
704       ip6->hop_limit = 255;
705       ip6->dst_address = pa46->ip6;
706       ip6->src_address = pa46->ip6;
707       return (sizeof (ip6_header_t));
708     }
709   else
710     {
711       ip4_header_t *ip4 = vlib_buffer_get_current (b0);
712       /* Fill in ip4 header fields */
713       ip4->checksum = 0;
714       ip4->ip_version_and_header_length = 0x45;
715       ip4->tos = 0;
716       ip4->length = 0;          /* will be set later */
717       ip4->fragment_id = 0;
718       ip4->flags_and_fragment_offset = 0;
719       ip4->ttl = 0xff;
720       ip4->protocol = IP_PROTOCOL_ICMP;
721       ip4->src_address = pa46->ip4;
722       ip4->dst_address = pa46->ip4;
723       return (sizeof (ip4_header_t));
724     }
725 }
726
727 static int
728 ip46_set_src_address (u32 sw_if_index, vlib_buffer_t * b0, int is_ip6)
729 {
730   int res;
731   if (is_ip6)
732     {
733       ip6_main_t *im = &ip6_main;
734       ip6_header_t *ip6 = vlib_buffer_get_current (b0);
735       res =
736         ip6_src_address_for_packet (&im->lookup_main, sw_if_index,
737                                     &ip6->dst_address, &ip6->src_address);
738     }
739   else
740     {
741       ip4_main_t *im = &ip4_main;
742       ip4_header_t *ip4 = vlib_buffer_get_current (b0);
743       res =
744         ip4_src_address_for_packet (&im->lookup_main, sw_if_index,
745                                     &ip4->src_address);
746       /* IP4 and IP6 paths have the inverse logic. Harmonize. */
747       res = !res;
748     }
749   return res;
750 }
751
752 static void
753 ip46_print_buffer_src_address (vlib_main_t * vm, vlib_buffer_t * b0,
754                                int is_ip6)
755 {
756   void *format_addr_func;
757   void *paddr;
758   if (is_ip6)
759     {
760       ip6_header_t *ip6 = vlib_buffer_get_current (b0);
761       format_addr_func = format_ip6_address;
762       paddr = &ip6->src_address;
763     }
764   else
765     {
766       ip4_header_t *ip4 = vlib_buffer_get_current (b0);
767       format_addr_func = format_ip4_address;
768       paddr = &ip4->src_address;
769     }
770   vlib_cli_output (vm, "Source address: %U ", format_addr_func, paddr);
771 }
772
773 static u16
774 ip46_fill_icmp_request_at (vlib_main_t * vm, int l4_offset, u16 seq_host,
775                            u16 id_host, u16 data_len, vlib_buffer_t * b0,
776                            int is_ip6)
777 {
778   icmp46_header_t *icmp46 = vlib_buffer_get_current (b0) + l4_offset;
779
780   icmp46->type = is_ip6 ? ICMP6_echo_request : ICMP4_echo_request;
781   icmp46->code = 0;
782   icmp46->checksum = 0;
783
784   icmp46_echo_request_t *icmp46_echo = (icmp46_echo_request_t *) (icmp46 + 1);
785
786   data_len =
787     init_icmp46_echo_request (vm, b0, l4_offset, icmp46_echo, seq_host,
788                               id_host, clib_cpu_time_now (), data_len);
789   return data_len;
790 }
791
792
793 /* Compute ICMP4 checksum with multibuffer support. */
794 u16
795 ip4_icmp_compute_checksum (vlib_main_t * vm, vlib_buffer_t * p0,
796                            ip4_header_t * ip0)
797 {
798   ip_csum_t sum0;
799   u32 ip_header_length, payload_length_host_byte_order;
800   u32 n_this_buffer, n_bytes_left, n_ip_bytes_this_buffer;
801   u16 sum16;
802   void *data_this_buffer;
803
804   ip_header_length = ip4_header_bytes (ip0);
805   payload_length_host_byte_order =
806     clib_net_to_host_u16 (ip0->length) - ip_header_length;
807
808   /* ICMP4 checksum does not include the IP header */
809   sum0 = 0;
810
811   n_bytes_left = n_this_buffer = payload_length_host_byte_order;
812   data_this_buffer = (void *) ip0 + ip_header_length;
813   n_ip_bytes_this_buffer =
814     p0->current_length - (((u8 *) ip0 - p0->data) - p0->current_data);
815   if (n_this_buffer + ip_header_length > n_ip_bytes_this_buffer)
816     {
817       n_this_buffer = n_ip_bytes_this_buffer > ip_header_length ?
818         n_ip_bytes_this_buffer - ip_header_length : 0;
819     }
820   while (1)
821     {
822       sum0 = ip_incremental_checksum (sum0, data_this_buffer, n_this_buffer);
823       n_bytes_left -= n_this_buffer;
824       if (n_bytes_left == 0)
825         break;
826
827       ASSERT (p0->flags & VLIB_BUFFER_NEXT_PRESENT);
828       p0 = vlib_get_buffer (vm, p0->next_buffer);
829       data_this_buffer = vlib_buffer_get_current (p0);
830       n_this_buffer = p0->current_length;
831     }
832
833   sum16 = ~ip_csum_fold (sum0);
834
835   return sum16;
836 }
837
838
839 static void
840 ip46_fix_len_and_csum (vlib_main_t * vm, int l4_offset, u16 data_len,
841                        vlib_buffer_t * b0, int is_ip6)
842 {
843   u16 payload_length =
844     data_len + sizeof (icmp46_header_t) + offsetof (icmp46_echo_request_t,
845                                                     data);
846   u16 total_length = payload_length + l4_offset;
847   icmp46_header_t *icmp46 = vlib_buffer_get_current (b0) + l4_offset;
848   icmp46->checksum = 0;
849
850   if (is_ip6)
851     {
852       ip6_header_t *ip6 = vlib_buffer_get_current (b0);
853       ip6->payload_length = clib_host_to_net_u16 (payload_length);
854
855       int bogus_length = 0;
856       icmp46->checksum =
857         ip6_tcp_udp_icmp_compute_checksum (vm, b0, ip6, &bogus_length);
858     }
859   else
860     {
861       ip4_header_t *ip4 = vlib_buffer_get_current (b0);
862       ip4->length = clib_host_to_net_u16 (total_length);
863
864       ip4->checksum = ip4_header_checksum (ip4);
865       icmp46->checksum = ip4_icmp_compute_checksum (vm, b0, ip4);
866     }
867 }
868
869 static u16
870 at_most_a_frame (u32 count)
871 {
872   return count > VLIB_FRAME_SIZE ? VLIB_FRAME_SIZE : count;
873 }
874
875 static int
876 ip46_enqueue_packet (vlib_main_t * vm, vlib_buffer_t * b0, u32 burst,
877                      int is_ip6)
878 {
879   vlib_frame_t *f = 0;
880   u32 lookup_node_index =
881     is_ip6 ? ip6_lookup_node.index : ip4_lookup_node.index;
882   int n_sent = 0;
883
884   u16 n_to_send;
885
886   /*
887    * Enqueue the packet, possibly as one or more frames of copies to make
888    * bursts. We enqueue b0 as the very last buffer, when there is no possibility
889    * for error in vlib_buffer_copy, so as to allow the caller to free it
890    * in case we encounter the error in the middle of the loop.
891    */
892   for (n_to_send = at_most_a_frame (burst), burst -= n_to_send; n_to_send > 0;
893        n_to_send = at_most_a_frame (burst), burst -= n_to_send)
894     {
895       f = vlib_get_frame_to_node (vm, lookup_node_index);
896       /* f can not be NULL here - frame allocation failure causes panic */
897
898       u32 *to_next = vlib_frame_vector_args (f);
899       f->n_vectors = n_to_send;
900
901       while (n_to_send > 1)
902         {
903           vlib_buffer_t *b0copy = vlib_buffer_copy (vm, b0);
904           if (PREDICT_FALSE (b0copy == NULL))
905             goto ship_and_ret;
906           *to_next++ = vlib_get_buffer_index (vm, b0copy);
907           n_to_send--;
908           n_sent++;
909         }
910
911       /* n_to_send is guaranteed to equal 1 here */
912       if (burst > 0)
913         {
914           /* not the last burst, so still make a copy for the last buffer */
915           vlib_buffer_t *b0copy = vlib_buffer_copy (vm, b0);
916           if (PREDICT_FALSE (b0copy == NULL))
917             goto ship_and_ret;
918           n_to_send--;
919           *to_next++ = vlib_get_buffer_index (vm, b0copy);
920         }
921       else
922         {
923           /* put the original buffer as the last one of an error-free run */
924           *to_next++ = vlib_get_buffer_index (vm, b0);
925         }
926       vlib_put_frame_to_node (vm, lookup_node_index, f);
927       n_sent += f->n_vectors;
928     }
929   return n_sent;
930   /*
931    * We reach here in case we already enqueued one or more buffers
932    * and maybe one or more frames but could not make more copies.
933    * There is an outstanding frame - so ship it and return.
934    * Caller will have to free the b0 in this case, since
935    * we did not enqueue it here yet.
936    */
937 ship_and_ret:
938   n_sent += f->n_vectors;
939   vlib_put_frame_to_node (vm, lookup_node_index, f);
940   return n_sent;
941 }
942
943
944 /*
945  * An address-family agnostic ping send function.
946  */
947
948 #define ERROR_OUT(e) do { err = e; goto done; } while (0)
949
950 static send_ip46_ping_result_t
951 send_ip46_ping (vlib_main_t * vm,
952                 u32 table_id,
953                 ip46_address_t * pa46,
954                 u32 sw_if_index,
955                 u16 seq_host, u16 id_host, u16 data_len, u32 burst,
956                 u8 verbose, int is_ip6)
957 {
958   int err = SEND_PING_OK;
959   u32 bi0 = 0;
960   int n_buf0 = 0;
961   vlib_buffer_t *b0;
962
963   n_buf0 = vlib_buffer_alloc (vm, &bi0, 1);
964   if (n_buf0 < 1)
965     ERROR_OUT (SEND_PING_ALLOC_FAIL);
966
967   b0 = vlib_get_buffer (vm, bi0);
968   VLIB_BUFFER_TRACE_TRAJECTORY_INIT (b0);
969
970   /*
971    * if the user did not provide a source interface,
972    * perform a resolution and use an interface
973    * via which it succeeds.
974    */
975   u32 fib_index;
976   if (~0 == sw_if_index)
977     {
978       fib_index = ip46_fib_index_from_table_id (table_id, is_ip6);
979       sw_if_index = ip46_get_resolving_interface (fib_index, pa46, is_ip6);
980     }
981   else
982     fib_index =
983       ip46_fib_table_get_index_for_sw_if_index (sw_if_index, is_ip6);
984
985   if (~0 == fib_index)
986     ERROR_OUT (SEND_PING_NO_TABLE);
987   if (~0 == sw_if_index)
988     ERROR_OUT (SEND_PING_NO_INTERFACE);
989
990   vnet_buffer (b0)->sw_if_index[VLIB_RX] = sw_if_index;
991   vnet_buffer (b0)->sw_if_index[VLIB_TX] = fib_index;
992
993   int l4_header_offset = ip46_fill_l3_header (pa46, b0, is_ip6);
994
995   /* set the src address in the buffer */
996   if (!ip46_set_src_address (sw_if_index, b0, is_ip6))
997     ERROR_OUT (SEND_PING_NO_SRC_ADDRESS);
998   if (verbose)
999     ip46_print_buffer_src_address (vm, b0, is_ip6);
1000
1001   data_len =
1002     ip46_fill_icmp_request_at (vm, l4_header_offset, seq_host, id_host,
1003                                data_len, b0, is_ip6);
1004
1005   ip46_fix_len_and_csum (vm, l4_header_offset, data_len, b0, is_ip6);
1006
1007   int n_sent = ip46_enqueue_packet (vm, b0, burst, is_ip6);
1008   if (n_sent < burst)
1009     err = SEND_PING_NO_BUFFERS;
1010
1011 done:
1012   if (err != SEND_PING_OK)
1013     {
1014       if (n_buf0 > 0)
1015         vlib_buffer_free (vm, &bi0, 1);
1016     }
1017   return err;
1018 }
1019
1020 static send_ip46_ping_result_t
1021 send_ip6_ping (vlib_main_t * vm,
1022                u32 table_id, ip6_address_t * pa6,
1023                u32 sw_if_index, u16 seq_host, u16 id_host, u16 data_len,
1024                u32 burst, u8 verbose)
1025 {
1026   ip46_address_t target;
1027   target.ip6 = *pa6;
1028   return send_ip46_ping (vm, table_id, &target, sw_if_index, seq_host,
1029                          id_host, data_len, burst, verbose, 1 /* is_ip6 */ );
1030 }
1031
1032 static send_ip46_ping_result_t
1033 send_ip4_ping (vlib_main_t * vm,
1034                u32 table_id, ip4_address_t * pa4,
1035                u32 sw_if_index, u16 seq_host, u16 id_host, u16 data_len,
1036                u32 burst, u8 verbose)
1037 {
1038   ip46_address_t target;
1039   ip46_address_set_ip4 (&target, pa4);
1040   return send_ip46_ping (vm, table_id, &target, sw_if_index, seq_host,
1041                          id_host, data_len, burst, verbose, 0 /* is_ip6 */ );
1042 }
1043
1044 static void
1045 print_ip46_icmp_reply (vlib_main_t * vm, u32 bi0, int is_ip6)
1046 {
1047   vlib_buffer_t *b0 = vlib_get_buffer (vm, bi0);
1048   int l4_offset;
1049   void *paddr;
1050   void *format_addr_func;
1051   u16 payload_length;
1052   u8 ttl;
1053   if (is_ip6)
1054     {
1055       ip6_header_t *ip6 = vlib_buffer_get_current (b0);
1056       paddr = (void *) &ip6->src_address;
1057       format_addr_func = (void *) format_ip6_address;
1058       ttl = ip6->hop_limit;
1059       l4_offset = sizeof (ip6_header_t);        // FIXME - EH processing ?
1060       payload_length = clib_net_to_host_u16 (ip6->payload_length);
1061     }
1062   else
1063     {
1064       ip4_header_t *ip4 = vlib_buffer_get_current (b0);
1065       paddr = (void *) &ip4->src_address;
1066       format_addr_func = (void *) format_ip4_address;
1067       ttl = ip4->ttl;
1068       l4_offset = ip4_header_bytes (ip4);
1069       payload_length =
1070         clib_net_to_host_u16 (ip4->length) + ip4_header_bytes (ip4);
1071     }
1072   icmp46_header_t *icmp = vlib_buffer_get_current (b0) + l4_offset;
1073   icmp46_echo_request_t *icmp_echo = (icmp46_echo_request_t *) (icmp + 1);
1074   u64 *dataplane_ts = (u64 *) & vnet_buffer (b0)->unused[0];
1075
1076   f64 clocks_per_second = ((f64) vm->clib_time.clocks_per_second);
1077   f64 rtt =
1078     ((f64) (*dataplane_ts - icmp_echo->time_sent)) / clocks_per_second;
1079
1080   vlib_cli_output (vm,
1081                    "%d bytes from %U: icmp_seq=%d ttl=%d time=%.4f ms",
1082                    payload_length,
1083                    format_addr_func,
1084                    paddr,
1085                    clib_host_to_net_u16 (icmp_echo->seq), ttl, rtt * 1000.0);
1086 }
1087
1088 /*
1089  * Perform the ping run with the given parameters in the current CLI process.
1090  * Depending on whether pa4 or pa6 is set, runs IPv4 or IPv6 ping.
1091  * The amusing side effect is of course if both are set, then both pings are sent.
1092  * This behavior can be used to ping a dualstack host over IPv4 and IPv6 at once.
1093  */
1094
1095 static void
1096 run_ping_ip46_address (vlib_main_t * vm, u32 table_id, ip4_address_t * pa4,
1097                        ip6_address_t * pa6, u32 sw_if_index,
1098                        f64 ping_interval, u32 ping_repeat, u32 data_len,
1099                        u32 ping_burst, u32 verbose)
1100 {
1101   int i;
1102   uword curr_proc = vlib_current_process (vm);
1103   u32 n_replies = 0;
1104   u32 n_requests = 0;
1105   u16 icmp_id;
1106
1107   static u32 rand_seed = 0;
1108
1109   if (PREDICT_FALSE (!rand_seed))
1110     rand_seed = random_default_seed ();
1111
1112   icmp_id = random_u32 (&rand_seed) & 0xffff;
1113
1114   while (~0 != get_cli_process_id_by_icmp_id_mt (vm, icmp_id))
1115     {
1116       vlib_cli_output (vm, "ICMP ID collision at %d, incrementing", icmp_id);
1117       icmp_id++;
1118     }
1119
1120   set_cli_process_id_by_icmp_id_mt (vm, icmp_id, curr_proc);
1121
1122   for (i = 1; i <= ping_repeat; i++)
1123     {
1124       send_ip46_ping_result_t res = SEND_PING_OK;
1125       f64 sleep_interval;
1126       f64 time_ping_sent = vlib_time_now (vm);
1127       if (pa6)
1128         {
1129           res = send_ip6_ping (vm, table_id,
1130                                pa6, sw_if_index, i, icmp_id,
1131                                data_len, ping_burst, verbose);
1132           if (SEND_PING_OK == res)
1133             n_requests += ping_burst;
1134           else
1135             vlib_cli_output (vm, "Failed: %U", format_ip46_ping_result, res);
1136         }
1137       if (pa4)
1138         {
1139           res = send_ip4_ping (vm, table_id, pa4,
1140                                sw_if_index, i, icmp_id, data_len,
1141                                ping_burst, verbose);
1142           if (SEND_PING_OK == res)
1143             n_requests += ping_burst;
1144           else
1145             vlib_cli_output (vm, "Failed: %U", format_ip46_ping_result, res);
1146         }
1147
1148       /* Collect and print the responses until it is time to send a next ping */
1149
1150       while ((i <= ping_repeat)
1151              &&
1152              ((sleep_interval =
1153                time_ping_sent + ping_interval - vlib_time_now (vm)) > 0.0))
1154         {
1155           uword event_type, *event_data = 0;
1156           vlib_process_wait_for_event_or_clock (vm, sleep_interval);
1157           event_type = vlib_process_get_events (vm, &event_data);
1158           switch (event_type)
1159             {
1160             case ~0:            /* no events => timeout */
1161               break;
1162             case PING_RESPONSE_IP6:
1163               /* fall-through */
1164             case PING_RESPONSE_IP4:
1165               {
1166                 int ii;
1167                 int is_ip6 = (event_type == PING_RESPONSE_IP6);
1168                 for (ii = 0; ii < vec_len (event_data); ii++)
1169                   {
1170                     u32 bi0 = event_data[ii];
1171                     print_ip46_icmp_reply (vm, bi0, is_ip6);
1172                     n_replies++;
1173                     if (0 != bi0)
1174                       vlib_buffer_free (vm, &bi0, 1);
1175                   }
1176               }
1177               break;
1178             default:
1179               /* someone pressed a key, abort */
1180               vlib_cli_output (vm, "Aborted due to a keypress.");
1181               goto double_break;
1182               break;
1183             }
1184           vec_free (event_data);
1185         }
1186     }
1187 double_break:
1188   vlib_cli_output (vm, "\n");
1189   {
1190     float loss =
1191       (0 ==
1192        n_requests) ? 0 : 100.0 * ((float) n_requests -
1193                                   (float) n_replies) / (float) n_requests;
1194     vlib_cli_output (vm,
1195                      "Statistics: %u sent, %u received, %f%% packet loss\n",
1196                      n_requests, n_replies, loss);
1197     clear_cli_process_id_by_icmp_id_mt (vm, icmp_id);
1198   }
1199 }
1200
1201
1202
1203 static clib_error_t *
1204 ping_ip_address (vlib_main_t * vm,
1205                  unformat_input_t * input, vlib_cli_command_t * cmd)
1206 {
1207   ip4_address_t a4;
1208   ip6_address_t a6;
1209   clib_error_t *error = 0;
1210   u32 ping_repeat = 5;
1211   u32 ping_burst = 1;
1212   u8 ping_ip4, ping_ip6;
1213   vnet_main_t *vnm = vnet_get_main ();
1214   u32 data_len = PING_DEFAULT_DATA_LEN;
1215   u32 verbose = 0;
1216   f64 ping_interval = PING_DEFAULT_INTERVAL;
1217   u32 sw_if_index, table_id;
1218
1219   table_id = 0;
1220   ping_ip4 = ping_ip6 = 0;
1221   sw_if_index = ~0;
1222
1223   if (unformat (input, "%U", unformat_ip4_address, &a4))
1224     {
1225       ping_ip4 = 1;
1226     }
1227   else if (unformat (input, "%U", unformat_ip6_address, &a6))
1228     {
1229       ping_ip6 = 1;
1230     }
1231   else if (unformat (input, "ipv4"))
1232     {
1233       if (unformat (input, "%U", unformat_ip4_address, &a4))
1234         {
1235           ping_ip4 = 1;
1236         }
1237       else
1238         {
1239           error =
1240             clib_error_return (0,
1241                                "expecting IPv4 address but got `%U'",
1242                                format_unformat_error, input);
1243         }
1244     }
1245   else if (unformat (input, "ipv6"))
1246     {
1247       if (unformat (input, "%U", unformat_ip6_address, &a6))
1248         {
1249           ping_ip6 = 1;
1250         }
1251       else
1252         {
1253           error =
1254             clib_error_return (0,
1255                                "expecting IPv6 address but got `%U'",
1256                                format_unformat_error, input);
1257         }
1258     }
1259   else
1260     {
1261       error =
1262         clib_error_return (0,
1263                            "expecting IP4/IP6 address `%U'. Usage: ping <addr> [source <intf>] [size <datasz>] [repeat <count>] [verbose]",
1264                            format_unformat_error, input);
1265       goto done;
1266     }
1267
1268   /* allow for the second AF in the same ping */
1269   if (!ping_ip4 && (unformat (input, "ipv4")))
1270     {
1271       if (unformat (input, "%U", unformat_ip4_address, &a4))
1272         {
1273           ping_ip4 = 1;
1274         }
1275     }
1276   else if (!ping_ip6 && (unformat (input, "ipv6")))
1277     {
1278       if (unformat (input, "%U", unformat_ip6_address, &a6))
1279         {
1280           ping_ip6 = 1;
1281         }
1282     }
1283
1284   /* parse the rest of the parameters  in a cycle */
1285   while (!unformat_eof (input, NULL))
1286     {
1287       if (unformat (input, "source"))
1288         {
1289           if (!unformat_user
1290               (input, unformat_vnet_sw_interface, vnm, &sw_if_index))
1291             {
1292               error =
1293                 clib_error_return (0,
1294                                    "unknown interface `%U'",
1295                                    format_unformat_error, input);
1296               goto done;
1297             }
1298         }
1299       else if (unformat (input, "size"))
1300         {
1301           if (!unformat (input, "%u", &data_len))
1302             {
1303               error =
1304                 clib_error_return (0,
1305                                    "expecting size but got `%U'",
1306                                    format_unformat_error, input);
1307               goto done;
1308             }
1309           if (data_len > PING_MAXIMUM_DATA_SIZE)
1310             {
1311               error =
1312                 clib_error_return (0,
1313                                    "%d is bigger than maximum allowed payload size %d",
1314                                    data_len, PING_MAXIMUM_DATA_SIZE);
1315               goto done;
1316             }
1317         }
1318       else if (unformat (input, "table-id"))
1319         {
1320           if (!unformat (input, "%u", &table_id))
1321             {
1322               error =
1323                 clib_error_return (0,
1324                                    "expecting table-id but got `%U'",
1325                                    format_unformat_error, input);
1326               goto done;
1327             }
1328         }
1329       else if (unformat (input, "interval"))
1330         {
1331           if (!unformat (input, "%f", &ping_interval))
1332             {
1333               error =
1334                 clib_error_return (0,
1335                                    "expecting interval (floating point number) got `%U'",
1336                                    format_unformat_error, input);
1337               goto done;
1338             }
1339         }
1340       else if (unformat (input, "repeat"))
1341         {
1342           if (!unformat (input, "%u", &ping_repeat))
1343             {
1344               error =
1345                 clib_error_return (0,
1346                                    "expecting repeat count but got `%U'",
1347                                    format_unformat_error, input);
1348               goto done;
1349             }
1350         }
1351       else if (unformat (input, "burst"))
1352         {
1353           if (!unformat (input, "%u", &ping_burst))
1354             {
1355               error =
1356                 clib_error_return (0,
1357                                    "expecting burst count but got `%U'",
1358                                    format_unformat_error, input);
1359               goto done;
1360             }
1361         }
1362       else if (unformat (input, "verbose"))
1363         {
1364           verbose = 1;
1365         }
1366       else
1367         {
1368           error = clib_error_return (0, "unknown input `%U'",
1369                                      format_unformat_error, input);
1370           goto done;
1371         }
1372     }
1373
1374 /*
1375  * Operationally, one won't (and shouldn't) need to send more than a frame worth of pings.
1376  * But it may be handy during the debugging.
1377  */
1378
1379 #ifdef CLIB_DEBUG
1380 #define MAX_PING_BURST (10*VLIB_FRAME_SIZE)
1381 #else
1382 #define MAX_PING_BURST (VLIB_FRAME_SIZE)
1383 #endif
1384
1385   if (ping_burst < 1 || ping_burst > MAX_PING_BURST)
1386     return clib_error_return (0, "burst size must be between 1 and %u",
1387                               MAX_PING_BURST);
1388
1389   run_ping_ip46_address (vm, table_id, ping_ip4 ? &a4 : NULL,
1390                          ping_ip6 ? &a6 : NULL, sw_if_index, ping_interval,
1391                          ping_repeat, data_len, ping_burst, verbose);
1392 done:
1393   return error;
1394 }
1395
1396 /*?
1397  * This command sends an ICMP ECHO_REQUEST to network hosts. The address
1398  * can be an IPv4 or IPv6 address (or both at the same time).
1399  *
1400  * @cliexpar
1401  * @parblock
1402  * Example of how ping an IPv4 address:
1403  * @cliexstart{ping 172.16.1.2 source GigabitEthernet2/0/0 repeat 2}
1404  * 64 bytes from 172.16.1.2: icmp_seq=1 ttl=64 time=.1090 ms
1405  * 64 bytes from 172.16.1.2: icmp_seq=2 ttl=64 time=.0914 ms
1406  *
1407  * Statistics: 2 sent, 2 received, 0% packet loss
1408  * @cliexend
1409  *
1410  * Example of how ping both an IPv4 address and IPv6 address at the same time:
1411  * @cliexstart{ping 172.16.1.2 ipv6 fe80::24a5:f6ff:fe9c:3a36 source GigabitEthernet2/0/0 repeat 2 verbose}
1412  * Adjacency index: 10, sw_if_index: 1
1413  * Adj: ip6-discover-neighbor
1414  * Adj Interface: 0
1415  * Forced set interface: 1
1416  * Adjacency index: 0, sw_if_index: 4294967295
1417  * Adj: ip4-miss
1418  * Adj Interface: 0
1419  * Forced set interface: 1
1420  * Source address: 172.16.1.1
1421  * 64 bytes from 172.16.1.2: icmp_seq=1 ttl=64 time=.1899 ms
1422  * Adjacency index: 10, sw_if_index: 1
1423  * Adj: ip6-discover-neighbor
1424  * Adj Interface: 0
1425  * Forced set interface: 1
1426  * Adjacency index: 0, sw_if_index: 4294967295
1427  * Adj: ip4-miss
1428  * Adj Interface: 0
1429  * Forced set interface: 1
1430  * Source address: 172.16.1.1
1431  * 64 bytes from 172.16.1.2: icmp_seq=2 ttl=64 time=.0910 ms
1432  *
1433  * Statistics: 4 sent, 2 received, 50% packet loss
1434  * @cliexend
1435  * @endparblock
1436 ?*/
1437 /* *INDENT-OFF* */
1438 VLIB_CLI_COMMAND (ping_command, static) =
1439 {
1440   .path = "ping",
1441   .function = ping_ip_address,
1442   .short_help = "ping {<ip-addr> | ipv4 <ip4-addr> | ipv6 <ip6-addr>}"
1443   " [ipv4 <ip4-addr> | ipv6 <ip6-addr>] [source <interface>]"
1444   " [size <pktsize:60>] [interval <sec:1>] [repeat <cnt:5>] [table-id <id:0>]"
1445   " [burst <count:1>] [verbose]",
1446   .is_mp_safe = 1,
1447 };
1448 /* *INDENT-ON* */
1449
1450 static clib_error_t *
1451 ping_cli_init (vlib_main_t * vm)
1452 {
1453   vlib_thread_main_t *tm = vlib_get_thread_main ();
1454   ping_main_t *pm = &ping_main;
1455
1456   pm->ip6_main = &ip6_main;
1457   pm->ip4_main = &ip4_main;
1458   icmp6_register_type (vm, ICMP6_echo_reply, ip6_icmp_echo_reply_node.index);
1459   ip4_icmp_register_type (vm, ICMP4_echo_reply,
1460                           ip4_icmp_echo_reply_node.index);
1461   if (tm->n_vlib_mains > 1)
1462     clib_spinlock_init (&pm->ping_run_check_lock);
1463
1464   ip4_icmp_register_type (vm, ICMP4_echo_request,
1465                           ip4_icmp_echo_request_node.index);
1466
1467   return 0;
1468 }
1469
1470 VLIB_INIT_FUNCTION (ping_cli_init);
1471
1472 /* *INDENT-OFF* */
1473 VLIB_PLUGIN_REGISTER () = {
1474     .version = VPP_BUILD_VER,
1475     .description = "Ping (ping)",
1476 };
1477 /* *INDENT-ON* */
1478
1479 /*
1480  * fd.io coding-style-patch-verification: ON
1481  *
1482  * Local Variables:
1483  * eval: (c-set-style "gnu")
1484  * End:
1485  */