aboutsummaryrefslogtreecommitdiffstats
path: root/src/plugins/rdma/input.c
diff options
context:
space:
mode:
Diffstat (limited to 'src/plugins/rdma/input.c')
-rw-r--r--src/plugins/rdma/input.c333
1 files changed, 254 insertions, 79 deletions
diff --git a/src/plugins/rdma/input.c b/src/plugins/rdma/input.c
index 001d1c5d493..2aa98f07e8c 100644
--- a/src/plugins/rdma/input.c
+++ b/src/plugins/rdma/input.c
@@ -45,108 +45,283 @@ rdma_device_input_refill (vlib_main_t * vm, rdma_device_t * rd,
rdma_rxq_t * rxq)
{
u32 n_alloc, n;
- struct ibv_sge sg_entry;
- struct ibv_recv_wr wr, *bad_wr;
- u32 buffers[VLIB_FRAME_SIZE];
+ u32 buffers[VLIB_FRAME_SIZE], *bi = buffers;
+ vlib_buffer_t *bufs[VLIB_FRAME_SIZE], **b = bufs;
+ struct ibv_recv_wr wr[VLIB_FRAME_SIZE], *w = wr;
+ struct ibv_sge sge[VLIB_FRAME_SIZE], *s = sge;
- if (rxq->n_enq >= rxq->size)
+ if (PREDICT_FALSE (rxq->n_enq >= rxq->size))
return;
n_alloc = clib_min (VLIB_FRAME_SIZE, rxq->size - rxq->n_enq);
- n_alloc = vlib_buffer_alloc (vm, buffers, n_alloc);
-
- sg_entry.length = vlib_buffer_get_default_data_size (vm);
- sg_entry.lkey = rd->mr->lkey;
- wr.num_sge = 1;
- wr.sg_list = &sg_entry;
- wr.next = NULL;
- for (n = 0; n < n_alloc; n++)
+ n_alloc = n = vlib_buffer_alloc (vm, buffers, n_alloc);
+ vlib_get_buffers (vm, buffers, bufs, n_alloc);
+
+ while (n >= 4)
{
- vlib_buffer_t *b = vlib_get_buffer (vm, buffers[n]);
- sg_entry.addr = vlib_buffer_get_va (b);
- wr.wr_id = buffers[n];
- if (ibv_post_recv (rxq->qp, &wr, &bad_wr) != 0)
- vlib_buffer_free (vm, buffers + n, 1);
- else
- rxq->n_enq++;
+ if (PREDICT_TRUE (n >= 8))
+ {
+ CLIB_PREFETCH (&s[4 + 0], 4 * sizeof (s[0]), STORE);
+ CLIB_PREFETCH (&w[4 + 0], 4 * sizeof (w[0]), STORE);
+ }
+
+ s[0].addr = vlib_buffer_get_va (b[0]);
+ s[0].length = vlib_buffer_get_default_data_size (vm);
+ s[0].lkey = rd->mr->lkey;
+
+ s[1].addr = vlib_buffer_get_va (b[1]);
+ s[1].length = vlib_buffer_get_default_data_size (vm);
+ s[1].lkey = rd->mr->lkey;
+
+ s[2].addr = vlib_buffer_get_va (b[2]);
+ s[2].length = vlib_buffer_get_default_data_size (vm);
+ s[2].lkey = rd->mr->lkey;
+
+ s[3].addr = vlib_buffer_get_va (b[3]);
+ s[3].length = vlib_buffer_get_default_data_size (vm);
+ s[3].lkey = rd->mr->lkey;
+
+ w[0].wr_id = bi[0];
+ w[0].next = &w[0] + 1;
+ w[0].sg_list = &s[0];
+ w[0].num_sge = 1;
+
+ w[1].wr_id = bi[1];
+ w[1].next = &w[1] + 1;
+ w[1].sg_list = &s[1];
+ w[1].num_sge = 1;
+
+ w[2].wr_id = bi[2];
+ w[2].next = &w[2] + 1;
+ w[2].sg_list = &s[2];
+ w[2].num_sge = 1;
+
+ w[3].wr_id = bi[3];
+ w[3].next = &w[3] + 1;
+ w[3].sg_list = &s[3];
+ w[3].num_sge = 1;
+
+ s += 4;
+ bi += 4;
+ w += 4;
+ b += 4;
+ n -= 4;
}
+
+ while (n >= 1)
+ {
+ s[0].addr = vlib_buffer_get_va (b[0]);
+ s[0].length = vlib_buffer_get_default_data_size (vm);
+ s[0].lkey = rd->mr->lkey;
+
+ w[0].wr_id = bi[0];
+ w[0].next = &w[0] + 1;
+ w[0].sg_list = &s[0];
+ w[0].num_sge = 1;
+
+ s += 1;
+ bi += 1;
+ w += 1;
+ b += 1;
+ n -= 1;
+ }
+
+ w[-1].next = 0; /* fix next pointer in WR linked-list last item */
+
+ w = wr;
+ ibv_post_recv (rxq->qp, wr, &w);
+ n = wr == w ? n_alloc : (uintptr_t) (w - wr);
+
+ if (PREDICT_FALSE (n != n_alloc))
+ vlib_buffer_free (vm, buffers + n, n_alloc - n);
+
+ rxq->n_enq += n;
}
-static_always_inline uword
-rdma_device_input_inline (vlib_main_t * vm, vlib_node_runtime_t * node,
- vlib_frame_t * frame, rdma_device_t * rd, u16 qid)
+static_always_inline void
+rdma_device_input_trace (vlib_main_t * vm, vlib_node_runtime_t * node,
+ const rdma_device_t * rd, u32 n_left, const u32 * bi)
{
- vnet_main_t *vnm = vnet_get_main ();
- rdma_rxq_t *rxq = vec_elt_at_index (rd->rxqs, qid);
- u32 n_trace;
- struct ibv_wc wc[VLIB_FRAME_SIZE];
- u32 next_index = VNET_DEVICE_INPUT_NEXT_ETHERNET_INPUT;
- u32 *bi, *to_next, n_left_to_next;
- int i;
- u32 n_rx_packets = 0, n_rx_bytes = 0;
+ u32 n_trace, i;
- n_rx_packets = ibv_poll_cq (rxq->cq, VLIB_FRAME_SIZE, wc);
+ if (PREDICT_TRUE (0 == (n_trace = vlib_get_trace_count (vm, node))))
+ return;
+
+ i = 0;
+ while (n_trace && n_left)
+ {
+ vlib_buffer_t *b;
+ rdma_input_trace_t *tr;
+ b = vlib_get_buffer (vm, bi[0]);
+ vlib_trace_buffer (vm, node, rd->per_interface_next_index, b,
+ /* follow_chain */ 0);
+ tr = vlib_add_trace (vm, node, b, sizeof (*tr));
+ tr->next_index = rd->per_interface_next_index;
+ tr->hw_if_index = rd->hw_if_index;
+
+ /* next */
+ n_trace--;
+ n_left--;
+ bi++;
+ i++;
+ }
+ vlib_set_trace_count (vm, node, n_trace);
+}
+
+static_always_inline void
+rdma_device_input_ethernet (vlib_main_t * vm, vlib_node_runtime_t * node,
+ const rdma_device_t * rd)
+{
+ vlib_next_frame_t *nf;
+ vlib_frame_t *f;
+ ethernet_input_frame_t *ef;
- if (n_rx_packets <= 0)
- rdma_device_input_refill (vm, rd, rxq);
+ if (PREDICT_FALSE
+ (VNET_DEVICE_INPUT_NEXT_ETHERNET_INPUT != rd->per_interface_next_index))
+ return;
- if (PREDICT_FALSE (rd->per_interface_next_index != ~0))
- next_index = rd->per_interface_next_index;
+ nf =
+ vlib_node_runtime_get_next_frame (vm, node, rd->per_interface_next_index);
+ f = vlib_get_frame (vm, nf->frame_index);
+ f->flags = ETH_INPUT_FRAME_F_SINGLE_SW_IF_IDX;
+ /* FIXME: f->flags |= ETH_INPUT_FRAME_F_IP4_CKSUM_OK; */
- vlib_get_new_next_frame (vm, node, next_index, to_next, n_left_to_next);
+ ef = vlib_frame_scalar_args (f);
+ ef->sw_if_index = rd->sw_if_index;
+ ef->hw_if_index = rd->hw_if_index;
+}
- for (i = 0; i < n_rx_packets; i++)
+static_always_inline u32
+rdma_device_input_load_wc (u32 n_left_from, struct ibv_wc * wc, u32 * to_next,
+ u32 * bufsz)
+{
+ u32 n_rx_bytes[4] = { 0 };
+
+ while (n_left_from >= 4)
{
- u32 bi = wc[i].wr_id;
- vlib_buffer_t *b = vlib_get_buffer (vm, bi);
- b->current_length = wc[i].byte_len;
- vnet_buffer (b)->sw_if_index[VLIB_RX] = rd->sw_if_index;
- vnet_buffer (b)->sw_if_index[VLIB_TX] = ~0;
- to_next[i] = bi;
- n_rx_bytes += wc[i].byte_len;
+ if (PREDICT_TRUE (n_left_from >= 8))
+ {
+ CLIB_PREFETCH (&wc[4 + 0], CLIB_CACHE_LINE_BYTES, LOAD);
+ CLIB_PREFETCH (&wc[4 + 1], CLIB_CACHE_LINE_BYTES, LOAD);
+ CLIB_PREFETCH (&wc[4 + 2], CLIB_CACHE_LINE_BYTES, LOAD);
+ CLIB_PREFETCH (&wc[4 + 3], CLIB_CACHE_LINE_BYTES, LOAD);
+ CLIB_PREFETCH (&bufsz[4 + 0], 4 * sizeof (bufsz[0]), STORE);
+ CLIB_PREFETCH (&to_next[4 + 0], 4 * sizeof (to_next[0]), STORE);
+ }
+
+ to_next[0] = wc[0].wr_id;
+ to_next[1] = wc[1].wr_id;
+ to_next[2] = wc[2].wr_id;
+ to_next[3] = wc[3].wr_id;
+
+ bufsz[0] = wc[0].byte_len;
+ bufsz[1] = wc[1].byte_len;
+ bufsz[2] = wc[2].byte_len;
+ bufsz[3] = wc[3].byte_len;
+
+ n_rx_bytes[0] += wc[0].byte_len;
+ n_rx_bytes[1] += wc[1].byte_len;
+ n_rx_bytes[2] += wc[2].byte_len;
+ n_rx_bytes[3] += wc[3].byte_len;
+
+ wc += 4;
+ to_next += 4;
+ bufsz += 4;
+ n_left_from -= 4;
}
- if (PREDICT_FALSE ((n_trace = vlib_get_trace_count (vm, node))))
+ while (n_left_from >= 1)
{
- u32 n_left = n_rx_packets, i = 0;
- bi = to_next;
+ to_next[0] = wc[0].wr_id;
+ bufsz[0] = wc[0].byte_len;
+ n_rx_bytes[0] += wc[0].byte_len;
- while (n_trace && n_left)
+ wc += 1;
+ to_next += 1;
+ bufsz += 1;
+ n_left_from -= 1;
+ }
+
+ return n_rx_bytes[0] + n_rx_bytes[1] + n_rx_bytes[2] + n_rx_bytes[3];
+}
+
+static_always_inline void
+rdma_device_input_bufs_init (u32 n_left_from, vlib_buffer_t ** bufs,
+ u32 * bufsz, u32 sw_if_index)
+{
+ while (n_left_from >= 4)
+ {
+ if (PREDICT_TRUE (n_left_from >= 8))
{
- vlib_buffer_t *b;
- rdma_input_trace_t *tr;
- b = vlib_get_buffer (vm, bi[0]);
- vlib_trace_buffer (vm, node, next_index, b, /* follow_chain */ 0);
- tr = vlib_add_trace (vm, node, b, sizeof (*tr));
- tr->next_index = next_index;
- tr->hw_if_index = rd->hw_if_index;
-
- /* next */
- n_trace--;
- n_left--;
- bi++;
- i++;
+ vlib_prefetch_buffer_header (bufs[4 + 0], STORE);
+ vlib_prefetch_buffer_header (bufs[4 + 1], STORE);
+ vlib_prefetch_buffer_header (bufs[4 + 2], STORE);
+ vlib_prefetch_buffer_header (bufs[4 + 3], STORE);
+ CLIB_PREFETCH (&bufsz[4 + 0], 4 * sizeof (bufsz[0]), LOAD);
}
- vlib_set_trace_count (vm, node, n_trace);
+
+ bufs[0]->current_length = bufsz[0];
+ bufs[1]->current_length = bufsz[1];
+ bufs[2]->current_length = bufsz[2];
+ bufs[3]->current_length = bufsz[3];
+
+ vnet_buffer (bufs[0])->sw_if_index[VLIB_RX] = sw_if_index;
+ vnet_buffer (bufs[1])->sw_if_index[VLIB_RX] = sw_if_index;
+ vnet_buffer (bufs[2])->sw_if_index[VLIB_RX] = sw_if_index;
+ vnet_buffer (bufs[3])->sw_if_index[VLIB_RX] = sw_if_index;
+
+ vnet_buffer (bufs[0])->sw_if_index[VLIB_TX] = ~0;
+ vnet_buffer (bufs[1])->sw_if_index[VLIB_TX] = ~0;
+ vnet_buffer (bufs[2])->sw_if_index[VLIB_TX] = ~0;
+ vnet_buffer (bufs[3])->sw_if_index[VLIB_TX] = ~0;
+
+ bufs += 4;
+ bufsz += 4;
+ n_left_from -= 4;
}
- if (PREDICT_TRUE (next_index == VNET_DEVICE_INPUT_NEXT_ETHERNET_INPUT))
+ while (n_left_from >= 1)
{
- vlib_next_frame_t *nf;
- vlib_frame_t *f;
- ethernet_input_frame_t *ef;
- nf = vlib_node_runtime_get_next_frame (vm, node, next_index);
- f = vlib_get_frame (vm, nf->frame_index);
- f->flags = ETH_INPUT_FRAME_F_SINGLE_SW_IF_IDX;
-
- ef = vlib_frame_scalar_args (f);
- ef->sw_if_index = rd->sw_if_index;
- ef->hw_if_index = rd->hw_if_index;
- //f->flags |= ETH_INPUT_FRAME_F_IP4_CKSUM_OK;
+ bufs[0]->current_length = bufsz[0];
+ vnet_buffer (bufs[0])->sw_if_index[VLIB_RX] = sw_if_index;
+ vnet_buffer (bufs[0])->sw_if_index[VLIB_TX] = ~0;
+
+ bufs += 1;
+ bufsz += 1;
+ n_left_from -= 1;
}
+}
- n_left_to_next -= n_rx_packets;
- vlib_put_next_frame (vm, node, next_index, n_left_to_next);
+static_always_inline uword
+rdma_device_input_inline (vlib_main_t * vm, vlib_node_runtime_t * node,
+ vlib_frame_t * frame, rdma_device_t * rd, u16 qid)
+{
+ vnet_main_t *vnm = vnet_get_main ();
+ rdma_rxq_t *rxq = vec_elt_at_index (rd->rxqs, qid);
+ struct ibv_wc wc[VLIB_FRAME_SIZE];
+ u32 bufsz[VLIB_FRAME_SIZE];
+ vlib_buffer_t *bufs[VLIB_FRAME_SIZE];
+ u32 *to_next, n_left_to_next;
+ u32 n_rx_packets, n_rx_bytes;
+
+ n_rx_packets = ibv_poll_cq (rxq->cq, VLIB_FRAME_SIZE, wc);
+
+ if (PREDICT_FALSE (n_rx_packets <= 0))
+ {
+ rdma_device_input_refill (vm, rd, rxq);
+ return 0;
+ }
+
+ vlib_get_new_next_frame (vm, node, rd->per_interface_next_index, to_next,
+ n_left_to_next);
+ n_rx_bytes = rdma_device_input_load_wc (n_rx_packets, wc, to_next, bufsz);
+ vlib_get_buffers (vm, to_next, bufs, n_rx_packets);
+ rdma_device_input_bufs_init (n_rx_packets, bufs, bufsz, rd->sw_if_index);
+ rdma_device_input_trace (vm, node, rd, n_rx_packets, to_next);
+ rdma_device_input_ethernet (vm, node, rd);
+
+ vlib_put_next_frame (vm, node, rd->per_interface_next_index,
+ n_left_to_next - n_rx_packets);
vlib_increment_combined_counter
(vnm->interface_main.combined_sw_if_counters +
@@ -154,6 +329,7 @@ rdma_device_input_inline (vlib_main_t * vm, vlib_node_runtime_t * node,
rd->hw_if_index, n_rx_packets, n_rx_bytes);
rxq->n_enq -= n_rx_packets;
+
rdma_device_input_refill (vm, rd, rxq);
return n_rx_packets;
@@ -172,9 +348,8 @@ VLIB_NODE_FN (rdma_input_node) (vlib_main_t * vm,
{
rdma_device_t *rd;
rd = vec_elt_at_index (rm->devices, dq->dev_instance);
- if ((rd->flags & RDMA_DEVICE_F_ADMIN_UP) == 0)
- continue;
- n_rx += rdma_device_input_inline (vm, node, frame, rd, dq->queue_id);
+ if (PREDICT_TRUE (rd->flags & RDMA_DEVICE_F_ADMIN_UP))
+ n_rx += rdma_device_input_inline (vm, node, frame, rd, dq->queue_id);
}
return n_rx;
}