/* * Copyright (c) 2017 Cisco and/or its affiliates. * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at: * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ #include #include #include #include #include #include #include #include vlib_node_registration_t session_queue_node; typedef struct { u32 session_index; u32 server_thread_index; } session_queue_trace_t; /* packet trace format function */ static u8 * format_session_queue_trace (u8 * s, va_list * args) { CLIB_UNUSED (vlib_main_t * vm) = va_arg (*args, vlib_main_t *); CLIB_UNUSED (vlib_node_t * node) = va_arg (*args, vlib_node_t *); session_queue_trace_t *t = va_arg (*args, session_queue_trace_t *); s = format (s, "SESSION_QUEUE: session index %d, server thread index %d", t->session_index, t->server_thread_index); return s; } vlib_node_registration_t session_queue_node; #define foreach_session_queue_error \ _(TX, "Packets transmitted") \ _(TIMER, "Timer events") typedef enum { #define _(sym,str) SESSION_QUEUE_ERROR_##sym, foreach_session_queue_error #undef _ SESSION_QUEUE_N_ERROR, } session_queue_error_t; static char *session_queue_error_strings[] = { #define _(sym,string) string, foreach_session_queue_error #undef _ }; static u32 session_type_to_next[] = { SESSION_QUEUE_NEXT_TCP_IP4_OUTPUT, SESSION_QUEUE_NEXT_IP4_LOOKUP, SESSION_QUEUE_NEXT_TCP_IP6_OUTPUT, SESSION_QUEUE_NEXT_IP6_LOOKUP, }; always_inline int session_tx_fifo_read_and_snd_i (vlib_main_t * vm, vlib_node_runtime_t * node, session_manager_main_t * smm, session_fifo_event_t * e0, stream_session_t * s0, u32 thread_index, int *n_tx_packets, u8 peek_data) { u32 n_trace = vlib_get_trace_count (vm, node); u32 left_to_snd0, max_len_to_snd0, len_to_deq0, n_bufs, snd_space0; u32 n_frame_bytes, n_frames_per_evt; transport_connection_t *tc0; transport_proto_vft_t *transport_vft; u32 next_index, next0, *to_next, n_left_to_next, bi0; vlib_buffer_t *b0; u32 rx_offset = 0, max_dequeue0; u16 snd_mss0; u8 *data0; int i, n_bytes_read; next_index = next0 = session_type_to_next[s0->session_type]; transport_vft = session_get_transport_vft (s0->session_type); tc0 = transport_vft->get_connection (s0->connection_index, thread_index); /* Make sure we have space to send and there's something to dequeue */ snd_space0 = transport_vft->send_space (tc0); snd_mss0 = transport_vft->send_mss (tc0); /* Can't make any progress */ if (snd_space0 == 0 || snd_mss0 == 0) { vec_add1 (smm->evts_partially_read[thread_index], *e0); return 0; } if (peek_data) { /* Offset in rx fifo from where to peek data */ rx_offset = transport_vft->tx_fifo_offset (tc0); } /* Check how much we can pull. If buffering, subtract the offset */ max_dequeue0 = svm_fifo_max_dequeue (s0->server_tx_fifo) - rx_offset; /* Allow enqueuing of a new event */ svm_fifo_unset_event (s0->server_tx_fifo); /* Nothing to read return */ if (max_dequeue0 == 0) { return 0; } /* Ensure we're not writing more than transport window allows */ max_len_to_snd0 = clib_min (max_dequeue0, snd_space0); /* TODO check if transport is willing to send len_to_snd0 * bytes (Nagle) */ n_frame_bytes = snd_mss0 * VLIB_FRAME_SIZE; n_frames_per_evt = ceil ((double) max_len_to_snd0 / n_frame_bytes); n_bufs = vec_len (smm->tx_buffers[thread_index]); left_to_snd0 = max_len_to_snd0; for (i = 0; i < n_frames_per_evt; i++) { /* Make sure we have at least one full frame of buffers ready */ if (PREDICT_FALSE (n_bufs < VLIB_FRAME_SIZE)) { vec_validate (smm->tx_buffers[thread_index], n_bufs + VLIB_FRAME_SIZE - 1); n_bufs += vlib_buffer_alloc (vm, &smm->tx_buffers[thread_index][n_bufs], VLIB_FRAME_SIZE); /* buffer shortage * XXX 0.9 because when debugging we might not get a full frame */ if (PREDICT_FALSE (n_bufs < 0.9 * VLIB_FRAME_SIZE)) { if (svm_fifo_set_event (s0->server_tx_fifo)) { vec_add1 (smm->evts_partially_read[thread_index], *e0); } return -1; } _vec_len (smm->tx_buffers[thread_index]) = n_bufs; } vlib_get_next_frame (vm, node, next_index, to_next, n_left_to_next); while (left_to_snd0 && n_left_to_next) { /* Get free buffer */ n_bufs--; bi0 = smm->tx_buffers[thread_index][n_bufs]; _vec_len (smm->tx_buffers[thread_index]) = n_bufs; b0 = vlib_get_buffer (vm, bi0); b0->error = 0; b0->flags = VLIB_BUFFER_TOTAL_LENGTH_VALID | VNET_BUFFER_LOCALLY_ORIGINATED; b0->current_data = 0; /* RX on the local interface. tx in default fib */ vnet_buffer (b0)->sw_if_index[VLIB_RX] = 0; vnet_buffer (b0)->sw_if_index[VLIB_TX] = (u32) ~ 0; /* usual speculation, or the enqueue_x1 macro will barf */ to_next[0] = bi0; to_next += 1; n_left_to_next -= 1; VLIB_BUFFER_TRACE_TRAJECTORY_INIT (b0); if (PREDICT_FALSE (n_trace > 0)) { session_queue_trace_t *t0; vlib_trace_buffer (vm, node, next_index, b0, 1 /* follow_chain */ ); vlib_set_trace_count (vm, node, --n_trace); t0 = vlib_add_trace (vm, node, b0, sizeof (*t0)); t0->session_index = s0->session_index; t0->server_thread_index = s0->thread_index; } len_to_deq0 = (left_to_snd0 < snd_mss0) ? left_to_snd0 : snd_mss0; /* *INDENT-OFF* */ SESSION_EVT_DBG(SESSION_EVT_DEQ, s0, ({ ed->data[0] = e0->event_id; ed->data[1] = max_dequeue0; ed->data[2] = len_to_deq0; ed->data[3] = left_to_snd0; })); /* *INDENT-ON* */ /* Make room for headers */ data0 = vlib_buffer_make_headroom (b0, MAX_HDRS_LEN); /* Dequeue the data * TODO 1) peek instead of dequeue * 2) buffer chains */ if (peek_data) { n_bytes_read = svm_fifo_peek (s0->server_tx_fifo, s0->pid, rx_offset, len_to_deq0, data0); if (n_bytes_read <= 0) goto dequeue_fail; /* Keep track of progress locally, transport is also supposed to * increment it independently when pushing the header */ rx_offset += n_bytes_read; } else { n_bytes_read = svm_fifo_dequeue_nowait (s0->server_tx_fifo, s0->pid, len_to_deq0, data0); if (n_bytes_read <= 0) goto dequeue_fail; } b0->current_length = n_bytes_read; /* Ask transport to push header */ transport_vft->push_header (tc0, b0); left_to_snd0 -= n_bytes_read; *n_tx_packets = *n_tx_packets + 1; vlib_validate_buffer_enqueue_x1 (vm, node, next_index, to_next, n_left_to_next, bi0, next0); } vlib_put_next_frame (vm, node, next_index, n_left_to_next); } /* If we couldn't dequeue all bytes mark as partially read */ if (max_len_to_snd0 < max_dequeue0) { /* If we don't already have new event */ if (svm_fifo_set_event (s0->server_tx_fifo)) { vec_add1 (smm->evts_partially_read[thread_index], *e0); } } return 0; dequeue_fail: /* * Can't read from fifo. If we don't already have an event, save as partially * read, return buff to free list and return */ clib_warning ("dequeue fail"); if (svm_fifo_set_event (s0->server_tx_fifo)) { vec_add1 (smm->evts_partially_read[thread_index], *e0); } vlib_put_next_frame (vm, node, next_index, n_left_to_next + 1); _vec_len (smm->tx_buffers[thread_index]) += 1; return 0; } int session_tx_fifo_peek_and_snd (vlib_main_t * vm, vlib_node_runtime_t * node, session_manager_main_t * smm, session_fifo_event_t * e0, stream_session_t * s0, u32 thread_index, int *n_tx_pkts) { return session_tx_fifo_read_and_snd_i (vm, node, smm, e0, s0, thread_index, n_tx_pkts, 1); } int session_tx_fifo_dequeue_and_snd (vlib_main_t * vm, vlib_node_runtime_t * node, session_manager_main_t * smm, session_fifo_event_t * e0, stream_session_t * s0, u32 thread_index, int *n_tx_pkts) { return session_tx_fifo_read_and_snd_i (vm, node, smm, e0, s0, thread_index, n_tx_pkts, 0); } static uword session_queue_node_fn (vlib_main_t * vm, vlib_node_runtime_t * node, vlib_frame_t * frame) { session_manager_main_t *smm = vnet_get_session_manager_main (); session_fifo_event_t *my_fifo_events, *e; u32 n_to_dequeue, n_events; unix_shared_memory_queue_t *q; application_t *app; int n_tx_packets = 0; u32 my_thread_index = vm->cpu_index; int i, rv; /* * Update TCP time */ tcp_update_time (vlib_time_now (vm), my_thread_index); /* * Get vpp queue events */ q = smm->vpp_event_queues[my_thread_index]; if (PREDICT_FALSE (q == 0)) return 0; /* min number of events we can dequeue without blocking */ n_to_dequeue = q->cursize; my_fifo_events = smm->fifo_events[my_thread_index]; if (n_to_dequeue == 0 && vec_len (my_fifo_events) == 0) return 0; SESSION_EVT_DBG (SESSION_EVT_DEQ_NODE, 0); /* * If we didn't manage to process previous events try going * over them again without dequeuing new ones. */ /* XXX: Block senders to sessions that can't keep up */ if (vec_len (my_fifo_events) >= 100) { clib_warning ("too many fifo events unsolved"); goto skip_dequeue; } /* See you in the next life, don't be late */ if (pthread_mutex_trylock (&q->mutex)) return 0; for (i = 0; i < n_to_dequeue; i++) { vec_add2 (my_fifo_events, e, 1); unix_shared_memory_queue_sub_raw (q, (u8 *) e); } /* The other side of the connection is not polling */ if (q->cursize < (q->maxsize / 8)) (void) pthread_cond_broadcast (&q->condvar); pthread_mutex_unlock (&q->mutex); smm->fifo_events[my_thread_index] = my_fifo_events; skip_dequeue: n_events = vec_len (my_fifo_events); for (i = 0; i < n_events; i++) { svm_fifo_t *f0; /* $$$ prefetch 1 ahead maybe */ stream_session_t *s0; u32 session_index0; session_fifo_event_t *e0; e0 = &my_fifo_events[i]; f0 = e0->fifo; session_index0 = f0->server_session_index; /* $$$ add multiple event queues, per vpp worker thread */ ASSERT (f0->server_thread_index == my_thread_index); s0 = stream_session_get_if_valid (session_index0, my_thread_index); if (CLIB_DEBUG && !s0) { clib_warning ("It's dead, Jim!"); continue; } if (PREDICT_FALSE (s0->session_state == SESSION_STATE_CLOSED)) continue; ASSERT (s0->thread_index == my_thread_index); switch (e0->event_type) { case FIFO_EVENT_SERVER_TX: /* Spray packets in per session type frames, since they go to * different nodes */ rv = (smm->session_tx_fns[s0->session_type]) (vm, node, smm, e0, s0, my_thread_index, &n_tx_packets); /* Out of buffers */ if (rv < 0) goto done; break; case FIFO_EVENT_SERVER_EXIT: stream_session_disconnect (s0); break; case FIFO_EVENT_BUILTIN_RX: svm_fifo_unset_event (s0->server_rx_fifo); /* Get session's server */ app = application_get (s0->app_index); app->cb_fns.builtin_server_rx_callback (s0); break; default: clib_warning ("unhandled event type %d", e0->event_type); } } done: /* Couldn't process all events. Probably out of buffers */ if (PREDICT_FALSE (i < n_events)) { session_fifo_event_t *partially_read = smm->evts_partially_read[my_thread_index]; vec_add (partially_read, &my_fifo_events[i], n_events - i); vec_free (my_fifo_events); smm->fifo_events[my_thread_index] = partially_read; smm->evts_partially_read[my_thread_index] = 0; } else { vec_free (smm->fifo_events[my_thread_index]); smm->fifo_events[my_thread_index] = smm->evts_partially_read[my_thread_index]; smm->evts_partially_read[my_thread_index] = 0; } vlib_node_increment_counter (vm, session_queue_node.index, SESSION_QUEUE_ERROR_TX, n_tx_packets); SESSION_EVT_DBG (SESSION_EVT_DEQ_NODE, 1); return n_tx_packets; } /* *INDENT-OFF* */ VLIB_REGISTER_NODE (session_queue_node) = { .function = session_queue_node_fn, .name = "session-queue", .format_trace = format_session_queue_trace, .type = VLIB_NODE_TYPE_INPUT, .n_errors = ARRAY_LEN (session_queue_error_strings), .error_strings = session_queue_error_strings, .n_next_nodes = SESSION_QUEUE_N_NEXT, .state = VLIB_NODE_STATE_DISABLED, .next_nodes = { [SESSION_QUEUE_NEXT_DROP] = "error-drop", [SESSION_QUEUE_NEXT_IP4_LOOKUP] = "ip4-lookup", [SESSION_QUEUE_NEXT_IP6_LOOKUP] = "ip6-lookup", [SESSION_QUEUE_NEXT_TCP_IP4_OUTPUT] = "tcp4-output", [SESSION_QUEUE_NEXT_TCP_IP6_OUTPUT] = "tcp6-output", }, }; /* *INDENT-ON* */ /* * fd.io coding-style-patch-verification: ON * * Local Variables: * eval: (c-set-style "gnu") * End: */