src/plugins/af_packet/node.c

   1 /*
   2  *------------------------------------------------------------------
   3  * af_packet.c - linux kernel packet interface
   4  *
   5  * Copyright (c) 2016 Cisco and/or its affiliates.
   6  * Licensed under the Apache License, Version 2.0 (the "License");
   7  * you may not use this file except in compliance with the License.
   8  * You may obtain a copy of the License at:
   9  *
  10  *     http://www.apache.org/licenses/LICENSE-2.0
  11  *
  12  * Unless required by applicable law or agreed to in writing, software
  13  * distributed under the License is distributed on an "AS IS" BASIS,
  14  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15  * See the License for the specific language governing permissions and
  16  * limitations under the License.
  17  *------------------------------------------------------------------
  18  */
  19
  20 #include <linux/if_packet.h>
  21
  22 #include <vlib/vlib.h>
  23 #include <vlib/unix/unix.h>
  24 #include <vnet/ip/ip.h>
  25 #include <vnet/ethernet/ethernet.h>
  26 #include <vnet/interface/rx_queue_funcs.h>
  27 #include <vnet/feature/feature.h>
  28 #include <vnet/ethernet/packet.h>
  29
  30 #include <af_packet/af_packet.h>
  31 #include <vnet/devices/virtio/virtio_std.h>
  32
  33 #define foreach_af_packet_input_error                                         \
  34   _ (PARTIAL_PKT, "partial packet")                                           \
  35   _ (TIMEDOUT_BLK, "timed out block")                                         \
  36   _ (TOTAL_RECV_BLK, "total received block")
  37 typedef enum
  38 {
  39 #define _(f,s) AF_PACKET_INPUT_ERROR_##f,
  40   foreach_af_packet_input_error
  41 #undef _
  42     AF_PACKET_INPUT_N_ERROR,
  43 } af_packet_input_error_t;
  44
  45 static char *af_packet_input_error_strings[] = {
  46 #define _(n,s) s,
  47   foreach_af_packet_input_error
  48 #undef _
  49 };
  50
  51 typedef struct
  52 {
  53   u32 next_index;
  54   u32 hw_if_index;
  55   u16 queue_id;
  56   int block;
  57   u32 pkt_num;
  58   void *block_start;
  59   block_desc_t bd;
  60   union
  61   {
  62     tpacket3_hdr_t tph3;
  63     tpacket2_hdr_t tph2;
  64   };
  65   vnet_virtio_net_hdr_t vnet_hdr;
  66   u8 is_v3;
  67 } af_packet_input_trace_t;
  68
  69 static u8 *
  70 format_af_packet_input_trace (u8 * s, va_list * args)
  71 {
  72   CLIB_UNUSED (vlib_main_t * vm) = va_arg (*args, vlib_main_t *);
  73   CLIB_UNUSED (vlib_node_t * node) = va_arg (*args, vlib_node_t *);
  74   af_packet_input_trace_t *t = va_arg (*args, af_packet_input_trace_t *);
  75   u32 indent = format_get_indent (s);
  76
  77   s = format (s, "af_packet: hw_if_index %d rx-queue %u next-index %d",
  78               t->hw_if_index, t->queue_id, t->next_index);
  79
  80   if (t->is_v3)
  81     {
  82       s = format (
  83         s, "\n%Ublock %u:\n%Uaddress %p version %u seq_num %lu pkt_num %u",
  84         format_white_space, indent + 2, t->block, format_white_space,
  85         indent + 4, t->block_start, t->bd.version, t->bd.hdr.bh1.seq_num,
  86         t->pkt_num);
  87       s = format (
  88         s,
  89         "\n%Utpacket3_hdr:\n%Ustatus 0x%x len %u snaplen %u mac %u net %u"
  90         "\n%Usec 0x%x nsec 0x%x vlan %U"
  91 #ifdef TP_STATUS_VLAN_TPID_VALID
  92         " vlan_tpid %u"
  93 #endif
  94         ,
  95         format_white_space, indent + 2, format_white_space, indent + 4,
  96         t->tph3.tp_status, t->tph3.tp_len, t->tph3.tp_snaplen, t->tph3.tp_mac,
  97         t->tph3.tp_net, format_white_space, indent + 4, t->tph3.tp_sec,
  98         t->tph3.tp_nsec, format_ethernet_vlan_tci, t->tph3.hv1.tp_vlan_tci
  99 #ifdef TP_STATUS_VLAN_TPID_VALID
 100         ,
 101         t->tph3.hv1.tp_vlan_tpid
 102 #endif
 103       );
 104     }
 105   else
 106     {
 107       s = format (
 108         s,
 109         "\n%Utpacket2_hdr:\n%Ustatus 0x%x len %u snaplen %u mac %u net %u"
 110         "\n%Usec 0x%x nsec 0x%x vlan %U"
 111 #ifdef TP_STATUS_VLAN_TPID_VALID
 112         " vlan_tpid %u"
 113 #endif
 114         ,
 115         format_white_space, indent + 2, format_white_space, indent + 4,
 116         t->tph2.tp_status, t->tph2.tp_len, t->tph2.tp_snaplen, t->tph2.tp_mac,
 117         t->tph2.tp_net, format_white_space, indent + 4, t->tph2.tp_sec,
 118         t->tph2.tp_nsec, format_ethernet_vlan_tci, t->tph2.tp_vlan_tci
 119 #ifdef TP_STATUS_VLAN_TPID_VALID
 120         ,
 121         t->tph2.tp_vlan_tpid
 122 #endif
 123       );
 124     }
 125
 126   s = format (s,
 127               "\n%Uvnet-hdr:\n%Uflags 0x%02x gso_type 0x%02x hdr_len %u"
 128               "\n%Ugso_size %u csum_start %u csum_offset %u",
 129               format_white_space, indent + 2, format_white_space, indent + 4,
 130               t->vnet_hdr.flags, t->vnet_hdr.gso_type, t->vnet_hdr.hdr_len,
 131               format_white_space, indent + 4, t->vnet_hdr.gso_size,
 132               t->vnet_hdr.csum_start, t->vnet_hdr.csum_offset);
 133   return s;
 134 }
 135
 136 always_inline void
 137 buffer_add_to_chain (vlib_buffer_t *b, vlib_buffer_t *first_b,
 138                      vlib_buffer_t *prev_b, u32 bi)
 139 {
 140   /* update first buffer */
 141   first_b->total_length_not_including_first_buffer += b->current_length;
 142
 143   /* update previous buffer */
 144   prev_b->next_buffer = bi;
 145   prev_b->flags |= VLIB_BUFFER_NEXT_PRESENT;
 146
 147   /* update current buffer */
 148   b->next_buffer = ~0;
 149 }
 150
 151 static_always_inline void
 152 fill_gso_offload (vlib_buffer_t *b, u32 gso_size, u8 l4_hdr_sz)
 153 {
 154   b->flags |= VNET_BUFFER_F_GSO;
 155   vnet_buffer2 (b)->gso_size = gso_size;
 156   vnet_buffer2 (b)->gso_l4_hdr_sz = l4_hdr_sz;
 157 }
 158
 159 static_always_inline void
 160 fill_cksum_offload (vlib_buffer_t *b, u8 *l4_hdr_sz, u8 is_ip)
 161 {
 162   vnet_buffer_oflags_t oflags = 0;
 163   u16 l2hdr_sz = 0;
 164   u16 ethertype = 0;
 165   u8 l4_proto = 0;
 166
 167   if (is_ip)
 168     {
 169       switch (b->data[0] & 0xf0)
 170         {
 171         case 0x40:
 172           ethertype = ETHERNET_TYPE_IP4;
 173           break;
 174         case 0x60:
 175           ethertype = ETHERNET_TYPE_IP6;
 176           break;
 177         }
 178     }
 179   else
 180     {
 181       ethernet_header_t *eth = (ethernet_header_t *) b->data;
 182       ethertype = clib_net_to_host_u16 (eth->type);
 183       l2hdr_sz = sizeof (ethernet_header_t);
 184       if (ethernet_frame_is_tagged (ethertype))
 185         {
 186           ethernet_vlan_header_t *vlan = (ethernet_vlan_header_t *) (eth + 1);
 187
 188           ethertype = clib_net_to_host_u16 (vlan->type);
 189           l2hdr_sz += sizeof (*vlan);
 190           if (ethertype == ETHERNET_TYPE_VLAN)
 191             {
 192               vlan++;
 193               ethertype = clib_net_to_host_u16 (vlan->type);
 194               l2hdr_sz += sizeof (*vlan);
 195             }
 196         }
 197     }
 198
 199   vnet_buffer (b)->l2_hdr_offset = 0;
 200   vnet_buffer (b)->l3_hdr_offset = l2hdr_sz;
 201
 202   if (ethertype == ETHERNET_TYPE_IP4)
 203     {
 204       ip4_header_t *ip4 = (ip4_header_t *) (b->data + l2hdr_sz);
 205       vnet_buffer (b)->l4_hdr_offset = l2hdr_sz + ip4_header_bytes (ip4);
 206       b->flags |= (VNET_BUFFER_F_IS_IP4 | VNET_BUFFER_F_L2_HDR_OFFSET_VALID |
 207                    VNET_BUFFER_F_L3_HDR_OFFSET_VALID |
 208                    VNET_BUFFER_F_L4_HDR_OFFSET_VALID);
 209
 210       l4_proto = ip4->protocol;
 211     }
 212   else if (ethertype == ETHERNET_TYPE_IP6)
 213     {
 214       ip6_header_t *ip6 = (ip6_header_t *) (b->data + l2hdr_sz);
 215       b->flags |= (VNET_BUFFER_F_IS_IP6 | VNET_BUFFER_F_L2_HDR_OFFSET_VALID |
 216                    VNET_BUFFER_F_L3_HDR_OFFSET_VALID |
 217                    VNET_BUFFER_F_L4_HDR_OFFSET_VALID);
 218       u16 ip6_hdr_len = sizeof (ip6_header_t);
 219
 220       if (ip6_ext_hdr (ip6->protocol))
 221         {
 222           ip6_ext_header_t *p = (void *) (ip6 + 1);
 223           ip6_hdr_len += ip6_ext_header_len (p);
 224           while (ip6_ext_hdr (p->next_hdr))
 225             {
 226               ip6_hdr_len += ip6_ext_header_len (p);
 227               p = ip6_ext_next_header (p);
 228             }
 229           l4_proto = p->next_hdr;
 230         }
 231       else
 232         l4_proto = ip6->protocol;
 233       vnet_buffer (b)->l4_hdr_offset = l2hdr_sz + ip6_hdr_len;
 234     }
 235
 236   if (l4_proto == IP_PROTOCOL_TCP)
 237     {
 238       oflags |= VNET_BUFFER_OFFLOAD_F_TCP_CKSUM;
 239       tcp_header_t *tcp =
 240         (tcp_header_t *) (b->data + vnet_buffer (b)->l4_hdr_offset);
 241       *l4_hdr_sz = tcp_header_bytes (tcp);
 242     }
 243   else if (l4_proto == IP_PROTOCOL_UDP)
 244     {
 245       oflags |= VNET_BUFFER_OFFLOAD_F_UDP_CKSUM;
 246       *l4_hdr_sz = sizeof (udp_header_t);
 247     }
 248
 249   if (oflags)
 250     vnet_buffer_offload_flags_set (b, oflags);
 251 }
 252
 253 always_inline uword
 254 af_packet_v3_device_input_fn (vlib_main_t *vm, vlib_node_runtime_t *node,
 255                               vlib_frame_t *frame, af_packet_if_t *apif,
 256                               u16 queue_id, u8 is_cksum_gso_enabled)
 257 {
 258   af_packet_main_t *apm = &af_packet_main;
 259   af_packet_queue_t *rx_queue = vec_elt_at_index (apif->rx_queues, queue_id);
 260   tpacket3_hdr_t *tph;
 261   u32 next_index;
 262   u32 n_free_bufs;
 263   u32 n_rx_packets = 0;
 264   u32 n_rx_bytes = 0;
 265   u32 timedout_blk = 0;
 266   u32 total = 0;
 267   u32 *to_next = 0;
 268   u32 block = rx_queue->next_rx_block;
 269   u32 block_nr = rx_queue->rx_req->req3.tp_block_nr;
 270   u8 *block_start = 0;
 271   uword n_trace = vlib_get_trace_count (vm, node);
 272   u32 thread_index = vm->thread_index;
 273   u32 n_buffer_bytes = vlib_buffer_get_default_data_size (vm);
 274   u32 min_bufs = rx_queue->rx_req->req3.tp_frame_size / n_buffer_bytes;
 275   u32 num_pkts = 0;
 276   u32 rx_frame_offset = 0;
 277   block_desc_t *bd = 0;
 278   vlib_buffer_t bt = {};
 279   u8 is_ip = (apif->mode == AF_PACKET_IF_MODE_IP);
 280
 281   if (is_ip)
 282     next_index = VNET_DEVICE_INPUT_NEXT_IP4_INPUT;
 283   else
 284     {
 285       next_index = VNET_DEVICE_INPUT_NEXT_ETHERNET_INPUT;
 286       if (PREDICT_FALSE (apif->per_interface_next_index != ~0))
 287         next_index = apif->per_interface_next_index;
 288
 289       /* redirect if feature path enabled */
 290       vnet_feature_start_device_input (apif->sw_if_index, &next_index, &bt);
 291     }
 292
 293   if ((((block_desc_t *) (block_start = rx_queue->rx_ring[block]))
 294          ->hdr.bh1.block_status &
 295        TP_STATUS_USER) != 0)
 296     {
 297       u32 n_required = 0;
 298       bd = (block_desc_t *) block_start;
 299
 300       if (PREDICT_FALSE (rx_queue->is_rx_pending))
 301         {
 302           num_pkts = rx_queue->num_rx_pkts;
 303           rx_frame_offset = rx_queue->rx_frame_offset;
 304           rx_queue->is_rx_pending = 0;
 305         }
 306       else
 307         {
 308           num_pkts = bd->hdr.bh1.num_pkts;
 309           rx_frame_offset = bd->hdr.bh1.offset_to_first_pkt;
 310           total++;
 311
 312           if (TP_STATUS_BLK_TMO & bd->hdr.bh1.block_status)
 313             timedout_blk++;
 314         }
 315
 316       n_required = clib_max (num_pkts, VLIB_FRAME_SIZE);
 317       n_free_bufs = vec_len (apm->rx_buffers[thread_index]);
 318       if (PREDICT_FALSE (n_free_bufs < n_required))
 319         {
 320           vec_validate (apm->rx_buffers[thread_index],
 321                         n_required + n_free_bufs - 1);
 322           n_free_bufs += vlib_buffer_alloc (
 323             vm, &apm->rx_buffers[thread_index][n_free_bufs], n_required);
 324           vec_set_len (apm->rx_buffers[thread_index], n_free_bufs);
 325         }
 326
 327       while (num_pkts && (n_free_bufs >= min_bufs))
 328         {
 329           u32 next0 = next_index;
 330           u32 n_left_to_next;
 331
 332           vlib_get_next_frame (vm, node, next_index, to_next, n_left_to_next);
 333
 334           while (num_pkts && n_left_to_next && (n_free_bufs >= min_bufs))
 335             {
 336               tph = (tpacket3_hdr_t *) (block_start + rx_frame_offset);
 337
 338               if (num_pkts > 1)
 339                 CLIB_PREFETCH (block_start + rx_frame_offset +
 340                                  tph->tp_next_offset,
 341                                2 * CLIB_CACHE_LINE_BYTES, LOAD);
 342
 343               vlib_buffer_t *b0 = 0, *first_b0 = 0, *prev_b0 = 0;
 344               vnet_virtio_net_hdr_t *vnet_hdr = 0;
 345               u32 data_len = tph->tp_snaplen;
 346               u32 offset = 0;
 347               u32 bi0 = ~0, first_bi0 = ~0;
 348               u8 l4_hdr_sz = 0;
 349
 350               if (is_cksum_gso_enabled)
 351                 vnet_hdr =
 352                   (vnet_virtio_net_hdr_t *) ((u8 *) tph + tph->tp_mac -
 353                                              sizeof (vnet_virtio_net_hdr_t));
 354
 355               // save current state and return
 356               if (PREDICT_FALSE (((data_len / n_buffer_bytes) + 1) >
 357                                  vec_len (apm->rx_buffers[thread_index])))
 358                 {
 359                   rx_queue->rx_frame_offset = rx_frame_offset;
 360                   rx_queue->num_rx_pkts = num_pkts;
 361                   rx_queue->is_rx_pending = 1;
 362                   vlib_put_next_frame (vm, node, next_index, n_left_to_next);
 363                   goto done;
 364                 }
 365
 366               while (data_len)
 367                 {
 368                   /* grab free buffer */
 369                   u32 last_empty_buffer =
 370                     vec_len (apm->rx_buffers[thread_index]) - 1;
 371                   bi0 = apm->rx_buffers[thread_index][last_empty_buffer];
 372                   vec_set_len (apm->rx_buffers[thread_index],
 373                                last_empty_buffer);
 374                   n_free_bufs--;
 375
 376                   /* copy data */
 377                   u32 bytes_to_copy =
 378                     data_len > n_buffer_bytes ? n_buffer_bytes : data_len;
 379                   u32 vlan_len = 0;
 380                   u32 bytes_copied = 0;
 381
 382                   b0 = vlib_get_buffer (vm, bi0);
 383                   b0->current_data = 0;
 384
 385                   /* Kernel removes VLAN headers, so reconstruct VLAN */
 386                   if (PREDICT_FALSE (tph->tp_status & TP_STATUS_VLAN_VALID))
 387                     {
 388                       if (PREDICT_TRUE (offset == 0))
 389                         {
 390                           clib_memcpy_fast (vlib_buffer_get_current (b0),
 391                                             (u8 *) tph + tph->tp_mac,
 392                                             sizeof (ethernet_header_t));
 393                           ethernet_header_t *eth =
 394                             vlib_buffer_get_current (b0);
 395                           ethernet_vlan_header_t *vlan =
 396                             (ethernet_vlan_header_t *) (eth + 1);
 397                           vlan->priority_cfi_and_id =
 398                             clib_host_to_net_u16 (tph->hv1.tp_vlan_tci);
 399                           vlan->type = eth->type;
 400                           eth->type =
 401                             clib_host_to_net_u16 (ETHERNET_TYPE_VLAN);
 402                           vlan_len = sizeof (ethernet_vlan_header_t);
 403                           bytes_copied = sizeof (ethernet_header_t);
 404                         }
 405                     }
 406                   clib_memcpy_fast (((u8 *) vlib_buffer_get_current (b0)) +
 407                                       bytes_copied + vlan_len,
 408                                     (u8 *) tph + tph->tp_mac + offset +
 409                                       bytes_copied,
 410                                     (bytes_to_copy - bytes_copied));
 411
 412                   /* fill buffer header */
 413                   b0->current_length = bytes_to_copy + vlan_len;
 414
 415                   if (offset == 0)
 416                     {
 417                       b0->total_length_not_including_first_buffer = 0;
 418                       b0->flags = VLIB_BUFFER_TOTAL_LENGTH_VALID;
 419                       vnet_buffer (b0)->sw_if_index[VLIB_RX] =
 420                         apif->sw_if_index;
 421                       vnet_buffer (b0)->sw_if_index[VLIB_TX] = (u32) ~0;
 422                       first_b0 = b0;
 423                       first_bi0 = bi0;
 424                       if (is_cksum_gso_enabled)
 425                         {
 426                           if (vnet_hdr->flags & VIRTIO_NET_HDR_F_NEEDS_CSUM)
 427                             fill_cksum_offload (first_b0, &l4_hdr_sz, is_ip);
 428                           if (vnet_hdr->gso_type & (VIRTIO_NET_HDR_GSO_TCPV4 |
 429                                                     VIRTIO_NET_HDR_GSO_TCPV6))
 430                             fill_gso_offload (first_b0, vnet_hdr->gso_size,
 431                                               l4_hdr_sz);
 432                         }
 433                     }
 434                   else
 435                     buffer_add_to_chain (b0, first_b0, prev_b0, bi0);
 436
 437                   prev_b0 = b0;
 438                   offset += bytes_to_copy;
 439                   data_len -= bytes_to_copy;
 440                 }
 441               n_rx_packets++;
 442               n_rx_bytes += tph->tp_snaplen;
 443               to_next[0] = first_bi0;
 444               to_next += 1;
 445               n_left_to_next--;
 446
 447               /* drop partial packets */
 448               if (PREDICT_FALSE (tph->tp_len != tph->tp_snaplen))
 449                 {
 450                   next0 = VNET_DEVICE_INPUT_NEXT_DROP;
 451                   first_b0->error =
 452                     node->errors[AF_PACKET_INPUT_ERROR_PARTIAL_PKT];
 453                 }
 454               else
 455                 {
 456                   if (PREDICT_FALSE (apif->mode == AF_PACKET_IF_MODE_IP))
 457                     {
 458                       switch (first_b0->data[0] & 0xf0)
 459                         {
 460                         case 0x40:
 461                           next0 = VNET_DEVICE_INPUT_NEXT_IP4_INPUT;
 462                           break;
 463                         case 0x60:
 464                           next0 = VNET_DEVICE_INPUT_NEXT_IP6_INPUT;
 465                           break;
 466                         default:
 467                           next0 = VNET_DEVICE_INPUT_NEXT_DROP;
 468                           break;
 469                         }
 470                       if (PREDICT_FALSE (apif->per_interface_next_index != ~0))
 471                         next0 = apif->per_interface_next_index;
 472                     }
 473                   else
 474                     {
 475                       next0 = VNET_DEVICE_INPUT_NEXT_ETHERNET_INPUT;
 476                       if (PREDICT_FALSE (apif->per_interface_next_index != ~0))
 477                         next0 = apif->per_interface_next_index;
 478                       /* copy feature arc data from template */
 479                       first_b0->current_config_index = bt.current_config_index;
 480                       vnet_buffer (first_b0)->feature_arc_index =
 481                         vnet_buffer (&bt)->feature_arc_index;
 482                     }
 483                 }
 484
 485               /* trace */
 486               if (PREDICT_FALSE (n_trace > 0 &&
 487                                  vlib_trace_buffer (vm, node, next0, first_b0,
 488                                                     /* follow_chain */ 0)))
 489                 {
 490                   af_packet_input_trace_t *tr;
 491                   vlib_set_trace_count (vm, node, --n_trace);
 492                   tr = vlib_add_trace (vm, node, first_b0, sizeof (*tr));
 493                   tr->is_v3 = 1;
 494                   tr->next_index = next0;
 495                   tr->hw_if_index = apif->hw_if_index;
 496                   tr->queue_id = queue_id;
 497                   tr->block = block;
 498                   tr->block_start = bd;
 499                   tr->pkt_num = bd->hdr.bh1.num_pkts - num_pkts;
 500                   clib_memcpy_fast (&tr->bd, bd, sizeof (block_desc_t));
 501                   clib_memcpy_fast (&tr->tph3, tph, sizeof (tpacket3_hdr_t));
 502                   if (is_cksum_gso_enabled)
 503                     clib_memcpy_fast (&tr->vnet_hdr, vnet_hdr,
 504                                       sizeof (vnet_virtio_net_hdr_t));
 505                   else
 506                     clib_memset_u8 (&tr->vnet_hdr, 0,
 507                                     sizeof (vnet_virtio_net_hdr_t));
 508                 }
 509
 510               /* enque and take next packet */
 511               vlib_validate_buffer_enqueue_x1 (vm, node, next_index, to_next,
 512                                                n_left_to_next, first_bi0,
 513                                                next0);
 514
 515               /* next packet */
 516               num_pkts--;
 517               rx_frame_offset += tph->tp_next_offset;
 518             }
 519
 520           vlib_put_next_frame (vm, node, next_index, n_left_to_next);
 521         }
 522
 523       if (PREDICT_TRUE (num_pkts == 0))
 524         {
 525           bd->hdr.bh1.block_status = TP_STATUS_KERNEL;
 526           block = (block + 1) % block_nr;
 527         }
 528       else
 529         {
 530           rx_queue->rx_frame_offset = rx_frame_offset;
 531           rx_queue->num_rx_pkts = num_pkts;
 532           rx_queue->is_rx_pending = 1;
 533         }
 534     }
 535
 536   rx_queue->next_rx_block = block;
 537
 538 done:
 539
 540   if (apm->polling_count == 0)
 541     {
 542       if ((((block_desc_t *) (block_start = rx_queue->rx_ring[block]))
 543              ->hdr.bh1.block_status &
 544            TP_STATUS_USER) != 0)
 545         vlib_node_set_state (vm, node->node_index, VLIB_NODE_STATE_POLLING);
 546       else
 547         vlib_node_set_state (vm, node->node_index, VLIB_NODE_STATE_INTERRUPT);
 548     }
 549
 550   vlib_error_count (vm, node->node_index, AF_PACKET_INPUT_ERROR_TOTAL_RECV_BLK,
 551                     total);
 552   vlib_error_count (vm, node->node_index, AF_PACKET_INPUT_ERROR_TIMEDOUT_BLK,
 553                     timedout_blk);
 554
 555   vlib_increment_combined_counter
 556     (vnet_get_main ()->interface_main.combined_sw_if_counters
 557      + VNET_INTERFACE_COUNTER_RX,
 558      vlib_get_thread_index (), apif->hw_if_index, n_rx_packets, n_rx_bytes);
 559
 560   vnet_device_increment_rx_packets (thread_index, n_rx_packets);
 561   return n_rx_packets;
 562 }
 563
 564 always_inline uword
 565 af_packet_v2_device_input_fn (vlib_main_t *vm, vlib_node_runtime_t *node,
 566                               vlib_frame_t *frame, af_packet_if_t *apif,
 567                               u16 queue_id, u8 is_cksum_gso_enabled)
 568 {
 569   af_packet_main_t *apm = &af_packet_main;
 570   af_packet_queue_t *rx_queue = vec_elt_at_index (apif->rx_queues, queue_id);
 571   tpacket2_hdr_t *tph;
 572   u32 next_index;
 573   u32 block = 0;
 574   u32 rx_frame;
 575   u32 n_free_bufs;
 576   u32 n_rx_packets = 0;
 577   u32 n_rx_bytes = 0;
 578   u32 *to_next = 0;
 579   u32 frame_size = rx_queue->rx_req->req.tp_frame_size;
 580   u32 frame_num = rx_queue->rx_req->req.tp_frame_nr;
 581   u8 *block_start = rx_queue->rx_ring[block];
 582   uword n_trace = vlib_get_trace_count (vm, node);
 583   u32 thread_index = vm->thread_index;
 584   u32 n_buffer_bytes = vlib_buffer_get_default_data_size (vm);
 585   u32 min_bufs = rx_queue->rx_req->req.tp_frame_size / n_buffer_bytes;
 586   u8 is_ip = (apif->mode == AF_PACKET_IF_MODE_IP);
 587   vlib_buffer_t bt = {};
 588
 589   if (is_ip)
 590     {
 591       next_index = VNET_DEVICE_INPUT_NEXT_IP4_INPUT;
 592     }
 593   else
 594     {
 595       next_index = VNET_DEVICE_INPUT_NEXT_ETHERNET_INPUT;
 596       if (PREDICT_FALSE (apif->per_interface_next_index != ~0))
 597         next_index = apif->per_interface_next_index;
 598
 599       /* redirect if feature path enabled */
 600       vnet_feature_start_device_input (apif->sw_if_index, &next_index, &bt);
 601     }
 602
 603   n_free_bufs = vec_len (apm->rx_buffers[thread_index]);
 604   if (PREDICT_FALSE (n_free_bufs < VLIB_FRAME_SIZE))
 605     {
 606       vec_validate (apm->rx_buffers[thread_index],
 607                     VLIB_FRAME_SIZE + n_free_bufs - 1);
 608       n_free_bufs += vlib_buffer_alloc (
 609         vm, &apm->rx_buffers[thread_index][n_free_bufs], VLIB_FRAME_SIZE);
 610       vec_set_len (apm->rx_buffers[thread_index], n_free_bufs);
 611     }
 612
 613   rx_frame = rx_queue->next_rx_frame;
 614   tph = (tpacket2_hdr_t *) (block_start + rx_frame * frame_size);
 615   while ((tph->tp_status & TP_STATUS_USER) && (n_free_bufs > min_bufs))
 616     {
 617       vlib_buffer_t *b0 = 0, *first_b0 = 0, *prev_b0 = 0;
 618       u32 next0 = next_index;
 619
 620       u32 n_left_to_next;
 621       vlib_get_next_frame (vm, node, next_index, to_next, n_left_to_next);
 622       while ((tph->tp_status & TP_STATUS_USER) && (n_free_bufs > min_bufs) &&
 623              n_left_to_next)
 624         {
 625           vnet_virtio_net_hdr_t *vnet_hdr = 0;
 626           u32 data_len = tph->tp_snaplen;
 627           u32 offset = 0;
 628           u32 bi0 = 0, first_bi0 = 0;
 629           u8 l4_hdr_sz = 0;
 630
 631           if (is_cksum_gso_enabled)
 632             vnet_hdr =
 633               (vnet_virtio_net_hdr_t *) ((u8 *) tph + tph->tp_mac -
 634                                          sizeof (vnet_virtio_net_hdr_t));
 635           while (data_len)
 636             {
 637               /* grab free buffer */
 638               u32 last_empty_buffer =
 639                 vec_len (apm->rx_buffers[thread_index]) - 1;
 640               bi0 = apm->rx_buffers[thread_index][last_empty_buffer];
 641               b0 = vlib_get_buffer (vm, bi0);
 642               vec_set_len (apm->rx_buffers[thread_index], last_empty_buffer);
 643               n_free_bufs--;
 644
 645               /* copy data */
 646               u32 bytes_to_copy =
 647                 data_len > n_buffer_bytes ? n_buffer_bytes : data_len;
 648               u32 vlan_len = 0;
 649               u32 bytes_copied = 0;
 650               b0->current_data = 0;
 651               /* Kernel removes VLAN headers, so reconstruct VLAN */
 652               if (PREDICT_FALSE (tph->tp_status & TP_STATUS_VLAN_VALID))
 653                 {
 654                   if (PREDICT_TRUE (offset == 0))
 655                     {
 656                       clib_memcpy_fast (vlib_buffer_get_current (b0),
 657                                         (u8 *) tph + tph->tp_mac,
 658                                         sizeof (ethernet_header_t));
 659                       ethernet_header_t *eth = vlib_buffer_get_current (b0);
 660                       ethernet_vlan_header_t *vlan =
 661                         (ethernet_vlan_header_t *) (eth + 1);
 662                       vlan->priority_cfi_and_id =
 663                         clib_host_to_net_u16 (tph->tp_vlan_tci);
 664                       vlan->type = eth->type;
 665                       eth->type = clib_host_to_net_u16 (ETHERNET_TYPE_VLAN);
 666                       vlan_len = sizeof (ethernet_vlan_header_t);
 667                       bytes_copied = sizeof (ethernet_header_t);
 668                     }
 669                 }
 670               clib_memcpy_fast (((u8 *) vlib_buffer_get_current (b0)) +
 671                                   bytes_copied + vlan_len,
 672                                 (u8 *) tph + tph->tp_mac + offset +
 673                                   bytes_copied,
 674                                 (bytes_to_copy - bytes_copied));
 675
 676               /* fill buffer header */
 677               b0->current_length = bytes_to_copy + vlan_len;
 678
 679               if (offset == 0)
 680                 {
 681                   b0->total_length_not_including_first_buffer = 0;
 682                   b0->flags = VLIB_BUFFER_TOTAL_LENGTH_VALID;
 683                   vnet_buffer (b0)->sw_if_index[VLIB_RX] = apif->sw_if_index;
 684                   vnet_buffer (b0)->sw_if_index[VLIB_TX] = (u32) ~0;
 685                   first_bi0 = bi0;
 686                   first_b0 = vlib_get_buffer (vm, first_bi0);
 687
 688                   if (is_cksum_gso_enabled)
 689                     {
 690                       if (vnet_hdr->flags & VIRTIO_NET_HDR_F_NEEDS_CSUM)
 691                         fill_cksum_offload (first_b0, &l4_hdr_sz, is_ip);
 692                       if (vnet_hdr->gso_type & (VIRTIO_NET_HDR_GSO_TCPV4 |
 693                                                 VIRTIO_NET_HDR_GSO_TCPV6))
 694                         fill_gso_offload (first_b0, vnet_hdr->gso_size,
 695                                           l4_hdr_sz);
 696                     }
 697                 }
 698               else
 699                 buffer_add_to_chain (b0, first_b0, prev_b0, bi0);
 700
 701               prev_b0 = b0;
 702               offset += bytes_to_copy;
 703               data_len -= bytes_to_copy;
 704             }
 705           n_rx_packets++;
 706           n_rx_bytes += tph->tp_snaplen;
 707           to_next[0] = first_bi0;
 708           to_next += 1;
 709           n_left_to_next--;
 710
 711           /* drop partial packets */
 712           if (PREDICT_FALSE (tph->tp_len != tph->tp_snaplen))
 713             {
 714               next0 = VNET_DEVICE_INPUT_NEXT_DROP;
 715               first_b0->error =
 716                 node->errors[AF_PACKET_INPUT_ERROR_PARTIAL_PKT];
 717             }
 718           else
 719             {
 720               if (PREDICT_FALSE (is_ip))
 721                 {
 722                   switch (first_b0->data[0] & 0xf0)
 723                     {
 724                     case 0x40:
 725                       next0 = VNET_DEVICE_INPUT_NEXT_IP4_INPUT;
 726                       break;
 727                     case 0x60:
 728                       next0 = VNET_DEVICE_INPUT_NEXT_IP6_INPUT;
 729                       break;
 730                     default:
 731                       next0 = VNET_DEVICE_INPUT_NEXT_DROP;
 732                       break;
 733                     }
 734                   if (PREDICT_FALSE (apif->per_interface_next_index != ~0))
 735                     next0 = apif->per_interface_next_index;
 736                 }
 737               else
 738                 {
 739                   next0 = VNET_DEVICE_INPUT_NEXT_ETHERNET_INPUT;
 740                   if (PREDICT_FALSE (apif->per_interface_next_index != ~0))
 741                     next0 = apif->per_interface_next_index;
 742                   /* copy feature arc data from template */
 743                   first_b0->current_config_index = bt.current_config_index;
 744                   vnet_buffer (first_b0)->feature_arc_index =
 745                     vnet_buffer (&bt)->feature_arc_index;
 746                 }
 747             }
 748
 749           /* trace */
 750           if (PREDICT_FALSE (n_trace > 0 &&
 751                              vlib_trace_buffer (vm, node, next0, first_b0,
 752                                                 /* follow_chain */ 0)))
 753             {
 754               af_packet_input_trace_t *tr;
 755               vlib_set_trace_count (vm, node, --n_trace);
 756               tr = vlib_add_trace (vm, node, first_b0, sizeof (*tr));
 757               tr->is_v3 = 0;
 758               tr->next_index = next0;
 759               tr->hw_if_index = apif->hw_if_index;
 760               tr->queue_id = queue_id;
 761               clib_memcpy_fast (&tr->tph2, tph, sizeof (struct tpacket2_hdr));
 762               if (is_cksum_gso_enabled)
 763                 clib_memcpy_fast (&tr->vnet_hdr, vnet_hdr,
 764                                   sizeof (vnet_virtio_net_hdr_t));
 765               else
 766                 clib_memset_u8 (&tr->vnet_hdr, 0,
 767                                 sizeof (vnet_virtio_net_hdr_t));
 768             }
 769
 770           /* enque and take next packet */
 771           vlib_validate_buffer_enqueue_x1 (vm, node, next_index, to_next,
 772                                            n_left_to_next, first_bi0, next0);
 773
 774           /* next packet */
 775           tph->tp_status = TP_STATUS_KERNEL;
 776           rx_frame = (rx_frame + 1) % frame_num;
 777           tph = (struct tpacket2_hdr *) (block_start + rx_frame * frame_size);
 778         }
 779
 780       vlib_put_next_frame (vm, node, next_index, n_left_to_next);
 781     }
 782
 783   rx_queue->next_rx_frame = rx_frame;
 784
 785   vlib_increment_combined_counter (
 786     vnet_get_main ()->interface_main.combined_sw_if_counters +
 787       VNET_INTERFACE_COUNTER_RX,
 788     vlib_get_thread_index (), apif->hw_if_index, n_rx_packets, n_rx_bytes);
 789
 790   vnet_device_increment_rx_packets (thread_index, n_rx_packets);
 791   return n_rx_packets;
 792 }
 793
 794 always_inline uword
 795 af_packet_device_input_fn (vlib_main_t *vm, vlib_node_runtime_t *node,
 796                            vlib_frame_t *frame, af_packet_if_t *apif,
 797                            u16 queue_id, u8 is_cksum_gso_enabled)
 798
 799 {
 800   if (apif->version == TPACKET_V3)
 801     return af_packet_v3_device_input_fn (vm, node, frame, apif, queue_id,
 802                                          is_cksum_gso_enabled);
 803   else
 804     return af_packet_v2_device_input_fn (vm, node, frame, apif, queue_id,
 805                                          is_cksum_gso_enabled);
 806 }
 807
 808 VLIB_NODE_FN (af_packet_input_node) (vlib_main_t * vm,
 809                                      vlib_node_runtime_t * node,
 810                                      vlib_frame_t * frame)
 811 {
 812   u32 n_rx_packets = 0;
 813   af_packet_main_t *apm = &af_packet_main;
 814   vnet_hw_if_rxq_poll_vector_t *pv;
 815   pv = vnet_hw_if_get_rxq_poll_vector (vm, node);
 816   for (int i = 0; i < vec_len (pv); i++)
 817     {
 818       af_packet_if_t *apif;
 819       apif = vec_elt_at_index (apm->interfaces, pv[i].dev_instance);
 820       if (apif->is_admin_up)
 821         {
 822           if (apif->is_cksum_gso_enabled)
 823             n_rx_packets += af_packet_device_input_fn (vm, node, frame, apif,
 824                                                        pv[i].queue_id, 1);
 825           else
 826             n_rx_packets += af_packet_device_input_fn (vm, node, frame, apif,
 827                                                        pv[i].queue_id, 0);
 828         }
 829     }
 830   return n_rx_packets;
 831 }
 832
 833 VLIB_REGISTER_NODE (af_packet_input_node) = {
 834   .name = "af-packet-input",
 835   .flags = VLIB_NODE_FLAG_TRACE_SUPPORTED,
 836   .sibling_of = "device-input",
 837   .format_trace = format_af_packet_input_trace,
 838   .type = VLIB_NODE_TYPE_INPUT,
 839   .state = VLIB_NODE_STATE_INTERRUPT,
 840   .n_errors = AF_PACKET_INPUT_N_ERROR,
 841   .error_strings = af_packet_input_error_strings,
 842 };
 843
 844
 845 /*
 846  * fd.io coding-style-patch-verification: ON
 847  *
 848  * Local Variables:
 849  * eval: (c-set-style "gnu")
 850  * End:
 851  */