Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (c) 2017 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 | |
Florin Coras | 6792ec0 | 2017-03-13 03:49:51 -0700 | [diff] [blame] | 16 | #include <math.h> |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 17 | #include <vlib/vlib.h> |
| 18 | #include <vnet/vnet.h> |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 19 | #include <vppinfra/elog.h> |
Florin Coras | 561af9b | 2017-12-09 10:19:43 -0800 | [diff] [blame] | 20 | #include <vnet/session/transport.h> |
Florin Coras | 6792ec0 | 2017-03-13 03:49:51 -0700 | [diff] [blame] | 21 | #include <vnet/session/application.h> |
Florin Coras | e69f495 | 2017-03-07 10:06:24 -0800 | [diff] [blame] | 22 | #include <vnet/session/session_debug.h> |
Florin Coras | e86a8ed | 2018-01-05 03:20:25 -0800 | [diff] [blame] | 23 | #include <svm/queue.h> |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 24 | |
| 25 | vlib_node_registration_t session_queue_node; |
| 26 | |
| 27 | typedef struct |
| 28 | { |
| 29 | u32 session_index; |
| 30 | u32 server_thread_index; |
| 31 | } session_queue_trace_t; |
| 32 | |
| 33 | /* packet trace format function */ |
| 34 | static u8 * |
| 35 | format_session_queue_trace (u8 * s, va_list * args) |
| 36 | { |
| 37 | CLIB_UNUSED (vlib_main_t * vm) = va_arg (*args, vlib_main_t *); |
| 38 | CLIB_UNUSED (vlib_node_t * node) = va_arg (*args, vlib_node_t *); |
| 39 | session_queue_trace_t *t = va_arg (*args, session_queue_trace_t *); |
| 40 | |
| 41 | s = format (s, "SESSION_QUEUE: session index %d, server thread index %d", |
| 42 | t->session_index, t->server_thread_index); |
| 43 | return s; |
| 44 | } |
| 45 | |
| 46 | vlib_node_registration_t session_queue_node; |
| 47 | |
Florin Coras | 6792ec0 | 2017-03-13 03:49:51 -0700 | [diff] [blame] | 48 | #define foreach_session_queue_error \ |
| 49 | _(TX, "Packets transmitted") \ |
Florin Coras | 93992a9 | 2017-05-24 18:03:56 -0700 | [diff] [blame] | 50 | _(TIMER, "Timer events") \ |
| 51 | _(NO_BUFFER, "Out of buffers") |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 52 | |
| 53 | typedef enum |
| 54 | { |
| 55 | #define _(sym,str) SESSION_QUEUE_ERROR_##sym, |
| 56 | foreach_session_queue_error |
| 57 | #undef _ |
| 58 | SESSION_QUEUE_N_ERROR, |
| 59 | } session_queue_error_t; |
| 60 | |
| 61 | static char *session_queue_error_strings[] = { |
| 62 | #define _(sym,string) string, |
| 63 | foreach_session_queue_error |
| 64 | #undef _ |
| 65 | }; |
| 66 | |
Florin Coras | f6d68ed | 2017-05-07 19:12:02 -0700 | [diff] [blame] | 67 | always_inline void |
| 68 | session_tx_fifo_chain_tail (session_manager_main_t * smm, vlib_main_t * vm, |
| 69 | u8 thread_index, svm_fifo_t * fifo, |
| 70 | vlib_buffer_t * b0, u32 bi0, u8 n_bufs_per_seg, |
Florin Coras | b2215d6 | 2017-08-01 16:56:58 -0700 | [diff] [blame] | 71 | u32 left_from_seg, u32 * left_to_snd0, |
Florin Coras | 1f152cd | 2017-08-18 19:28:03 -0700 | [diff] [blame] | 72 | u16 * n_bufs, u32 * tx_offset, u16 deq_per_buf, |
Florin Coras | 7fb0fe1 | 2018-04-09 09:24:52 -0700 | [diff] [blame] | 73 | u8 peek_data, transport_tx_fn_type_t tx_type) |
Florin Coras | f6d68ed | 2017-05-07 19:12:02 -0700 | [diff] [blame] | 74 | { |
| 75 | vlib_buffer_t *chain_b0, *prev_b0; |
Florin Coras | b2215d6 | 2017-08-01 16:56:58 -0700 | [diff] [blame] | 76 | u32 chain_bi0, to_deq; |
Florin Coras | f6d68ed | 2017-05-07 19:12:02 -0700 | [diff] [blame] | 77 | u16 len_to_deq0, n_bytes_read; |
| 78 | u8 *data0, j; |
| 79 | |
Florin Coras | b2215d6 | 2017-08-01 16:56:58 -0700 | [diff] [blame] | 80 | b0->flags |= VLIB_BUFFER_TOTAL_LENGTH_VALID; |
| 81 | b0->total_length_not_including_first_buffer = 0; |
| 82 | |
Florin Coras | f6d68ed | 2017-05-07 19:12:02 -0700 | [diff] [blame] | 83 | chain_bi0 = bi0; |
| 84 | chain_b0 = b0; |
Florin Coras | b2215d6 | 2017-08-01 16:56:58 -0700 | [diff] [blame] | 85 | to_deq = left_from_seg; |
Florin Coras | f6d68ed | 2017-05-07 19:12:02 -0700 | [diff] [blame] | 86 | for (j = 1; j < n_bufs_per_seg; j++) |
| 87 | { |
| 88 | prev_b0 = chain_b0; |
Florin Coras | b2215d6 | 2017-08-01 16:56:58 -0700 | [diff] [blame] | 89 | len_to_deq0 = clib_min (to_deq, deq_per_buf); |
Florin Coras | f6d68ed | 2017-05-07 19:12:02 -0700 | [diff] [blame] | 90 | |
| 91 | *n_bufs -= 1; |
| 92 | chain_bi0 = smm->tx_buffers[thread_index][*n_bufs]; |
| 93 | _vec_len (smm->tx_buffers[thread_index]) = *n_bufs; |
| 94 | |
| 95 | chain_b0 = vlib_get_buffer (vm, chain_bi0); |
| 96 | chain_b0->current_data = 0; |
| 97 | data0 = vlib_buffer_get_current (chain_b0); |
| 98 | if (peek_data) |
| 99 | { |
Florin Coras | 1f152cd | 2017-08-18 19:28:03 -0700 | [diff] [blame] | 100 | n_bytes_read = svm_fifo_peek (fifo, *tx_offset, len_to_deq0, data0); |
| 101 | *tx_offset += n_bytes_read; |
Florin Coras | f6d68ed | 2017-05-07 19:12:02 -0700 | [diff] [blame] | 102 | } |
| 103 | else |
| 104 | { |
Florin Coras | 7fb0fe1 | 2018-04-09 09:24:52 -0700 | [diff] [blame] | 105 | if (tx_type == TRANSPORT_TX_DGRAM) |
| 106 | { |
| 107 | session_dgram_hdr_t *hdr; |
| 108 | u16 deq_now; |
| 109 | hdr = (session_dgram_hdr_t *) svm_fifo_head (fifo); |
| 110 | deq_now = clib_min (hdr->data_length - hdr->data_offset, |
| 111 | len_to_deq0); |
| 112 | n_bytes_read = svm_fifo_peek (fifo, hdr->data_offset, deq_now, |
| 113 | data0); |
| 114 | ASSERT (n_bytes_read > 0); |
| 115 | |
| 116 | hdr->data_offset += n_bytes_read; |
| 117 | if (hdr->data_offset == hdr->data_length) |
| 118 | svm_fifo_dequeue_drop (fifo, hdr->data_length); |
| 119 | } |
| 120 | else |
| 121 | n_bytes_read = svm_fifo_dequeue_nowait (fifo, len_to_deq0, data0); |
Florin Coras | f6d68ed | 2017-05-07 19:12:02 -0700 | [diff] [blame] | 122 | } |
| 123 | ASSERT (n_bytes_read == len_to_deq0); |
| 124 | chain_b0->current_length = n_bytes_read; |
| 125 | b0->total_length_not_including_first_buffer += chain_b0->current_length; |
| 126 | |
| 127 | /* update previous buffer */ |
| 128 | prev_b0->next_buffer = chain_bi0; |
| 129 | prev_b0->flags |= VLIB_BUFFER_NEXT_PRESENT; |
| 130 | |
| 131 | /* update current buffer */ |
| 132 | chain_b0->next_buffer = 0; |
| 133 | |
Florin Coras | b2215d6 | 2017-08-01 16:56:58 -0700 | [diff] [blame] | 134 | to_deq -= n_bytes_read; |
| 135 | if (to_deq == 0) |
Florin Coras | f6d68ed | 2017-05-07 19:12:02 -0700 | [diff] [blame] | 136 | break; |
| 137 | } |
Florin Coras | 1f152cd | 2017-08-18 19:28:03 -0700 | [diff] [blame] | 138 | ASSERT (to_deq == 0 |
| 139 | && b0->total_length_not_including_first_buffer == left_from_seg); |
Florin Coras | b2215d6 | 2017-08-01 16:56:58 -0700 | [diff] [blame] | 140 | *left_to_snd0 -= left_from_seg; |
Florin Coras | f6d68ed | 2017-05-07 19:12:02 -0700 | [diff] [blame] | 141 | } |
| 142 | |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 143 | always_inline int |
Florin Coras | d79b41e | 2017-03-04 05:37:52 -0800 | [diff] [blame] | 144 | session_tx_fifo_read_and_snd_i (vlib_main_t * vm, vlib_node_runtime_t * node, |
| 145 | session_manager_main_t * smm, |
| 146 | session_fifo_event_t * e0, |
| 147 | stream_session_t * s0, u32 thread_index, |
| 148 | int *n_tx_packets, u8 peek_data) |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 149 | { |
| 150 | u32 n_trace = vlib_get_trace_count (vm, node); |
Florin Coras | f6d68ed | 2017-05-07 19:12:02 -0700 | [diff] [blame] | 151 | u32 left_to_snd0, max_len_to_snd0, len_to_deq0, snd_space0; |
Florin Coras | e87216f | 2017-08-17 16:59:22 -0700 | [diff] [blame] | 152 | u32 n_bufs_per_evt, n_frames_per_evt, n_bufs_per_frame; |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 153 | transport_connection_t *tc0; |
| 154 | transport_proto_vft_t *transport_vft; |
Florin Coras | 561af9b | 2017-12-09 10:19:43 -0800 | [diff] [blame] | 155 | transport_proto_t tp; |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 156 | u32 next_index, next0, *to_next, n_left_to_next, bi0; |
| 157 | vlib_buffer_t *b0; |
Florin Coras | 1f152cd | 2017-08-18 19:28:03 -0700 | [diff] [blame] | 158 | u32 tx_offset = 0, max_dequeue0, n_bytes_per_seg, left_for_seg; |
Florin Coras | f6d68ed | 2017-05-07 19:12:02 -0700 | [diff] [blame] | 159 | u16 snd_mss0, n_bufs_per_seg, n_bufs; |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 160 | u8 *data0; |
Florin Coras | 6792ec0 | 2017-03-13 03:49:51 -0700 | [diff] [blame] | 161 | int i, n_bytes_read; |
Florin Coras | e87216f | 2017-08-17 16:59:22 -0700 | [diff] [blame] | 162 | u32 n_bytes_per_buf, deq_per_buf, deq_per_first_buf; |
Florin Coras | 81a13db | 2018-03-16 08:48:31 -0700 | [diff] [blame] | 163 | u32 bufs_alloc, bufs_now; |
Florin Coras | 7fb0fe1 | 2018-04-09 09:24:52 -0700 | [diff] [blame] | 164 | session_dgram_hdr_t hdr; |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 165 | |
Florin Coras | 561af9b | 2017-12-09 10:19:43 -0800 | [diff] [blame] | 166 | next_index = next0 = smm->session_type_to_next[s0->session_type]; |
Florin Coras | 561af9b | 2017-12-09 10:19:43 -0800 | [diff] [blame] | 167 | tp = session_get_transport_proto (s0); |
| 168 | transport_vft = transport_protocol_get_vft (tp); |
Florin Coras | 7fb0fe1 | 2018-04-09 09:24:52 -0700 | [diff] [blame] | 169 | if (peek_data) |
| 170 | { |
| 171 | if (PREDICT_FALSE (s0->session_state < SESSION_STATE_READY)) |
| 172 | { |
| 173 | /* Can retransmit for closed sessions but can't send new data if |
| 174 | * session is not ready or closed */ |
| 175 | vec_add1 (smm->pending_event_vector[thread_index], *e0); |
| 176 | return 0; |
| 177 | } |
| 178 | tc0 = |
| 179 | transport_vft->get_connection (s0->connection_index, thread_index); |
| 180 | } |
| 181 | else |
| 182 | { |
| 183 | if (s0->session_state == SESSION_STATE_LISTENING) |
| 184 | tc0 = transport_vft->get_listener (s0->connection_index); |
| 185 | else |
| 186 | { |
| 187 | if (PREDICT_FALSE (s0->session_state == SESSION_STATE_CLOSED)) |
| 188 | return 0; |
| 189 | tc0 = transport_vft->get_connection (s0->connection_index, |
| 190 | thread_index); |
| 191 | } |
| 192 | } |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 193 | |
| 194 | /* Make sure we have space to send and there's something to dequeue */ |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 195 | snd_mss0 = transport_vft->send_mss (tc0); |
Florin Coras | c834341 | 2017-05-04 14:25:50 -0700 | [diff] [blame] | 196 | snd_space0 = transport_vft->send_space (tc0); |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 197 | |
Florin Coras | e04c299 | 2017-03-01 08:17:34 -0800 | [diff] [blame] | 198 | /* Can't make any progress */ |
Florin Coras | 6792ec0 | 2017-03-13 03:49:51 -0700 | [diff] [blame] | 199 | if (snd_space0 == 0 || snd_mss0 == 0) |
Florin Coras | e04c299 | 2017-03-01 08:17:34 -0800 | [diff] [blame] | 200 | { |
Dave Barach | acd2a6a | 2017-05-16 17:41:34 -0400 | [diff] [blame] | 201 | vec_add1 (smm->pending_event_vector[thread_index], *e0); |
Florin Coras | e04c299 | 2017-03-01 08:17:34 -0800 | [diff] [blame] | 202 | return 0; |
| 203 | } |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 204 | |
Florin Coras | 6a9b68b | 2017-11-21 04:20:42 -0800 | [diff] [blame] | 205 | /* Allow enqueuing of a new event */ |
| 206 | svm_fifo_unset_event (s0->server_tx_fifo); |
| 207 | |
Florin Coras | 9d06304 | 2017-09-14 03:08:00 -0400 | [diff] [blame] | 208 | /* Check how much we can pull. */ |
| 209 | max_dequeue0 = svm_fifo_max_dequeue (s0->server_tx_fifo); |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 210 | if (peek_data) |
| 211 | { |
Florin Coras | 9d06304 | 2017-09-14 03:08:00 -0400 | [diff] [blame] | 212 | /* Offset in rx fifo from where to peek data */ |
Florin Coras | 1f152cd | 2017-08-18 19:28:03 -0700 | [diff] [blame] | 213 | tx_offset = transport_vft->tx_fifo_offset (tc0); |
Florin Coras | 9d06304 | 2017-09-14 03:08:00 -0400 | [diff] [blame] | 214 | if (PREDICT_FALSE (tx_offset >= max_dequeue0)) |
Florin Coras | 7fb0fe1 | 2018-04-09 09:24:52 -0700 | [diff] [blame] | 215 | return 0; |
| 216 | max_dequeue0 -= tx_offset; |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 217 | } |
Florin Coras | 7fb0fe1 | 2018-04-09 09:24:52 -0700 | [diff] [blame] | 218 | else |
| 219 | { |
| 220 | if (transport_vft->tx_type == TRANSPORT_TX_DGRAM) |
| 221 | { |
| 222 | if (max_dequeue0 < sizeof (hdr)) |
| 223 | return 0; |
| 224 | svm_fifo_peek (s0->server_tx_fifo, 0, sizeof (hdr), (u8 *) & hdr); |
| 225 | ASSERT (hdr.data_length > hdr.data_offset); |
| 226 | max_dequeue0 = hdr.data_length - hdr.data_offset; |
| 227 | } |
| 228 | } |
| 229 | ASSERT (max_dequeue0 > 0); |
Florin Coras | 6792ec0 | 2017-03-13 03:49:51 -0700 | [diff] [blame] | 230 | |
| 231 | /* Ensure we're not writing more than transport window allows */ |
Florin Coras | 3e350af | 2017-03-30 02:54:28 -0700 | [diff] [blame] | 232 | if (max_dequeue0 < snd_space0) |
| 233 | { |
| 234 | /* Constrained by tx queue. Try to send only fully formed segments */ |
| 235 | max_len_to_snd0 = (max_dequeue0 > snd_mss0) ? |
| 236 | max_dequeue0 - max_dequeue0 % snd_mss0 : max_dequeue0; |
| 237 | /* TODO Nagle ? */ |
| 238 | } |
| 239 | else |
| 240 | { |
Florin Coras | 1f152cd | 2017-08-18 19:28:03 -0700 | [diff] [blame] | 241 | /* Expectation is that snd_space0 is already a multiple of snd_mss */ |
Florin Coras | 3e350af | 2017-03-30 02:54:28 -0700 | [diff] [blame] | 242 | max_len_to_snd0 = snd_space0; |
| 243 | } |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 244 | |
Florin Coras | 93992a9 | 2017-05-24 18:03:56 -0700 | [diff] [blame] | 245 | n_bytes_per_buf = vlib_buffer_free_list_buffer_size |
| 246 | (vm, VLIB_BUFFER_DEFAULT_FREE_LIST_INDEX); |
Florin Coras | e87216f | 2017-08-17 16:59:22 -0700 | [diff] [blame] | 247 | ASSERT (n_bytes_per_buf > MAX_HDRS_LEN); |
Florin Coras | f6d68ed | 2017-05-07 19:12:02 -0700 | [diff] [blame] | 248 | n_bytes_per_seg = MAX_HDRS_LEN + snd_mss0; |
| 249 | n_bufs_per_seg = ceil ((double) n_bytes_per_seg / n_bytes_per_buf); |
Florin Coras | e87216f | 2017-08-17 16:59:22 -0700 | [diff] [blame] | 250 | n_bufs_per_evt = ceil ((double) max_len_to_snd0 / n_bytes_per_seg); |
Florin Coras | f6d68ed | 2017-05-07 19:12:02 -0700 | [diff] [blame] | 251 | n_frames_per_evt = ceil ((double) n_bufs_per_evt / VLIB_FRAME_SIZE); |
Florin Coras | e87216f | 2017-08-17 16:59:22 -0700 | [diff] [blame] | 252 | n_bufs_per_frame = n_bufs_per_seg * VLIB_FRAME_SIZE; |
Florin Coras | f6d68ed | 2017-05-07 19:12:02 -0700 | [diff] [blame] | 253 | |
| 254 | deq_per_buf = clib_min (snd_mss0, n_bytes_per_buf); |
Florin Coras | e87216f | 2017-08-17 16:59:22 -0700 | [diff] [blame] | 255 | deq_per_first_buf = clib_min (snd_mss0, n_bytes_per_buf - MAX_HDRS_LEN); |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 256 | |
| 257 | n_bufs = vec_len (smm->tx_buffers[thread_index]); |
| 258 | left_to_snd0 = max_len_to_snd0; |
| 259 | for (i = 0; i < n_frames_per_evt; i++) |
| 260 | { |
| 261 | /* Make sure we have at least one full frame of buffers ready */ |
Florin Coras | e87216f | 2017-08-17 16:59:22 -0700 | [diff] [blame] | 262 | if (PREDICT_FALSE (n_bufs < n_bufs_per_frame)) |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 263 | { |
| 264 | vec_validate (smm->tx_buffers[thread_index], |
Florin Coras | e87216f | 2017-08-17 16:59:22 -0700 | [diff] [blame] | 265 | n_bufs + n_bufs_per_frame - 1); |
Florin Coras | 81a13db | 2018-03-16 08:48:31 -0700 | [diff] [blame] | 266 | bufs_alloc = 0; |
Florin Coras | 93992a9 | 2017-05-24 18:03:56 -0700 | [diff] [blame] | 267 | do |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 268 | { |
Florin Coras | 81a13db | 2018-03-16 08:48:31 -0700 | [diff] [blame] | 269 | bufs_now = |
| 270 | vlib_buffer_alloc (vm, |
| 271 | &smm->tx_buffers[thread_index][n_bufs + |
| 272 | bufs_alloc], |
| 273 | n_bufs_per_frame - bufs_alloc); |
| 274 | bufs_alloc += bufs_now; |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 275 | } |
Florin Coras | 81a13db | 2018-03-16 08:48:31 -0700 | [diff] [blame] | 276 | while (bufs_now > 0 && ((bufs_alloc + n_bufs < n_bufs_per_frame))); |
Florin Coras | 93992a9 | 2017-05-24 18:03:56 -0700 | [diff] [blame] | 277 | |
Florin Coras | 81a13db | 2018-03-16 08:48:31 -0700 | [diff] [blame] | 278 | n_bufs += bufs_alloc; |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 279 | _vec_len (smm->tx_buffers[thread_index]) = n_bufs; |
Florin Coras | 93992a9 | 2017-05-24 18:03:56 -0700 | [diff] [blame] | 280 | |
Florin Coras | e87216f | 2017-08-17 16:59:22 -0700 | [diff] [blame] | 281 | if (PREDICT_FALSE (n_bufs < n_bufs_per_frame)) |
Florin Coras | 93992a9 | 2017-05-24 18:03:56 -0700 | [diff] [blame] | 282 | { |
| 283 | vec_add1 (smm->pending_event_vector[thread_index], *e0); |
| 284 | return -1; |
| 285 | } |
Florin Coras | e87216f | 2017-08-17 16:59:22 -0700 | [diff] [blame] | 286 | ASSERT (n_bufs >= n_bufs_per_frame); |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 287 | } |
| 288 | |
| 289 | vlib_get_next_frame (vm, node, next_index, to_next, n_left_to_next); |
Florin Coras | e87216f | 2017-08-17 16:59:22 -0700 | [diff] [blame] | 290 | while (left_to_snd0 && n_left_to_next) |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 291 | { |
Florin Coras | f6d68ed | 2017-05-07 19:12:02 -0700 | [diff] [blame] | 292 | /* |
| 293 | * Handle first buffer in chain separately |
| 294 | */ |
| 295 | |
Florin Coras | 81a13db | 2018-03-16 08:48:31 -0700 | [diff] [blame] | 296 | len_to_deq0 = clib_min (left_to_snd0, deq_per_first_buf); |
| 297 | if (left_to_snd0 > len_to_deq0 && n_left_to_next > 1) |
| 298 | { |
| 299 | vlib_buffer_t *pb; |
| 300 | u32 pbi = smm->tx_buffers[thread_index][n_bufs - 2]; |
| 301 | pb = vlib_get_buffer (vm, pbi); |
| 302 | vlib_prefetch_buffer_header (pb, LOAD); |
| 303 | } |
| 304 | |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 305 | /* Get free buffer */ |
Florin Coras | 93992a9 | 2017-05-24 18:03:56 -0700 | [diff] [blame] | 306 | ASSERT (n_bufs >= 1); |
Florin Coras | f6d68ed | 2017-05-07 19:12:02 -0700 | [diff] [blame] | 307 | bi0 = smm->tx_buffers[thread_index][--n_bufs]; |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 308 | _vec_len (smm->tx_buffers[thread_index]) = n_bufs; |
| 309 | |
Florin Coras | f6359c8 | 2017-06-19 12:26:09 -0400 | [diff] [blame] | 310 | /* usual speculation, or the enqueue_x1 macro will barf */ |
| 311 | to_next[0] = bi0; |
| 312 | to_next += 1; |
| 313 | n_left_to_next -= 1; |
| 314 | |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 315 | b0 = vlib_get_buffer (vm, bi0); |
| 316 | b0->error = 0; |
Florin Coras | 81a13db | 2018-03-16 08:48:31 -0700 | [diff] [blame] | 317 | b0->flags = VNET_BUFFER_F_LOCALLY_ORIGINATED; |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 318 | b0->current_data = 0; |
Florin Coras | f6d68ed | 2017-05-07 19:12:02 -0700 | [diff] [blame] | 319 | b0->total_length_not_including_first_buffer = 0; |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 320 | |
Florin Coras | f6d68ed | 2017-05-07 19:12:02 -0700 | [diff] [blame] | 321 | data0 = vlib_buffer_make_headroom (b0, MAX_HDRS_LEN); |
| 322 | if (peek_data) |
| 323 | { |
Florin Coras | 1f152cd | 2017-08-18 19:28:03 -0700 | [diff] [blame] | 324 | n_bytes_read = svm_fifo_peek (s0->server_tx_fifo, tx_offset, |
Florin Coras | f6d68ed | 2017-05-07 19:12:02 -0700 | [diff] [blame] | 325 | len_to_deq0, data0); |
Florin Coras | 9d06304 | 2017-09-14 03:08:00 -0400 | [diff] [blame] | 326 | if (n_bytes_read <= 0) |
| 327 | goto dequeue_fail; |
Florin Coras | f6d68ed | 2017-05-07 19:12:02 -0700 | [diff] [blame] | 328 | /* Keep track of progress locally, transport is also supposed to |
| 329 | * increment it independently when pushing the header */ |
Florin Coras | 1f152cd | 2017-08-18 19:28:03 -0700 | [diff] [blame] | 330 | tx_offset += n_bytes_read; |
Florin Coras | f6d68ed | 2017-05-07 19:12:02 -0700 | [diff] [blame] | 331 | } |
| 332 | else |
| 333 | { |
Florin Coras | 7fb0fe1 | 2018-04-09 09:24:52 -0700 | [diff] [blame] | 334 | if (transport_vft->tx_type == TRANSPORT_TX_DGRAM) |
| 335 | { |
| 336 | svm_fifo_t *f = s0->server_tx_fifo; |
| 337 | u16 deq_now; |
| 338 | u32 offset; |
| 339 | |
| 340 | ASSERT (hdr.data_length > hdr.data_offset); |
| 341 | deq_now = clib_min (hdr.data_length - hdr.data_offset, |
| 342 | len_to_deq0); |
| 343 | offset = hdr.data_offset + SESSION_CONN_HDR_LEN; |
| 344 | n_bytes_read = svm_fifo_peek (f, offset, deq_now, data0); |
| 345 | if (PREDICT_FALSE (n_bytes_read <= 0)) |
| 346 | goto dequeue_fail; |
| 347 | |
| 348 | if (s0->session_state == SESSION_STATE_LISTENING) |
| 349 | { |
| 350 | ip_copy (&tc0->rmt_ip, &hdr.rmt_ip, tc0->is_ip4); |
| 351 | tc0->rmt_port = hdr.rmt_port; |
| 352 | } |
| 353 | hdr.data_offset += n_bytes_read; |
| 354 | if (hdr.data_offset == hdr.data_length) |
| 355 | { |
| 356 | offset = hdr.data_length + SESSION_CONN_HDR_LEN; |
| 357 | svm_fifo_dequeue_drop (f, offset); |
| 358 | } |
| 359 | } |
| 360 | else |
| 361 | { |
| 362 | n_bytes_read = svm_fifo_dequeue_nowait (s0->server_tx_fifo, |
| 363 | len_to_deq0, data0); |
| 364 | if (n_bytes_read <= 0) |
| 365 | goto dequeue_fail; |
| 366 | } |
Florin Coras | f6d68ed | 2017-05-07 19:12:02 -0700 | [diff] [blame] | 367 | } |
| 368 | |
Florin Coras | f6d68ed | 2017-05-07 19:12:02 -0700 | [diff] [blame] | 369 | b0->current_length = n_bytes_read; |
Florin Coras | f6d68ed | 2017-05-07 19:12:02 -0700 | [diff] [blame] | 370 | left_to_snd0 -= n_bytes_read; |
| 371 | *n_tx_packets = *n_tx_packets + 1; |
| 372 | |
| 373 | /* |
| 374 | * Fill in the remaining buffers in the chain, if any |
| 375 | */ |
Florin Coras | e87216f | 2017-08-17 16:59:22 -0700 | [diff] [blame] | 376 | if (PREDICT_FALSE (n_bufs_per_seg > 1 && left_to_snd0)) |
Florin Coras | b2215d6 | 2017-08-01 16:56:58 -0700 | [diff] [blame] | 377 | { |
Florin Coras | b2215d6 | 2017-08-01 16:56:58 -0700 | [diff] [blame] | 378 | left_for_seg = clib_min (snd_mss0 - n_bytes_read, left_to_snd0); |
| 379 | session_tx_fifo_chain_tail (smm, vm, thread_index, |
| 380 | s0->server_tx_fifo, b0, bi0, |
| 381 | n_bufs_per_seg, left_for_seg, |
Florin Coras | 1f152cd | 2017-08-18 19:28:03 -0700 | [diff] [blame] | 382 | &left_to_snd0, &n_bufs, &tx_offset, |
Florin Coras | 7fb0fe1 | 2018-04-09 09:24:52 -0700 | [diff] [blame] | 383 | deq_per_buf, peek_data, |
| 384 | transport_vft->tx_type); |
Florin Coras | b2215d6 | 2017-08-01 16:56:58 -0700 | [diff] [blame] | 385 | } |
Florin Coras | f6d68ed | 2017-05-07 19:12:02 -0700 | [diff] [blame] | 386 | |
| 387 | /* Ask transport to push header after current_length and |
| 388 | * total_length_not_including_first_buffer are updated */ |
| 389 | transport_vft->push_header (tc0, b0); |
| 390 | |
| 391 | /* *INDENT-OFF* */ |
| 392 | SESSION_EVT_DBG(SESSION_EVT_DEQ, s0, ({ |
Florin Coras | ec44e34 | 2017-10-16 20:47:56 -0700 | [diff] [blame] | 393 | ed->data[0] = e0->event_type; |
Florin Coras | f6d68ed | 2017-05-07 19:12:02 -0700 | [diff] [blame] | 394 | ed->data[1] = max_dequeue0; |
| 395 | ed->data[2] = len_to_deq0; |
| 396 | ed->data[3] = left_to_snd0; |
| 397 | })); |
| 398 | /* *INDENT-ON* */ |
| 399 | |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 400 | VLIB_BUFFER_TRACE_TRAJECTORY_INIT (b0); |
Florin Coras | 4eeeaaf | 2017-09-05 14:03:37 -0400 | [diff] [blame] | 401 | |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 402 | if (PREDICT_FALSE (n_trace > 0)) |
| 403 | { |
| 404 | session_queue_trace_t *t0; |
| 405 | vlib_trace_buffer (vm, node, next_index, b0, |
| 406 | 1 /* follow_chain */ ); |
| 407 | vlib_set_trace_count (vm, node, --n_trace); |
| 408 | t0 = vlib_add_trace (vm, node, b0, sizeof (*t0)); |
| 409 | t0->session_index = s0->session_index; |
| 410 | t0->server_thread_index = s0->thread_index; |
| 411 | } |
| 412 | |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 413 | vlib_validate_buffer_enqueue_x1 (vm, node, next_index, |
| 414 | to_next, n_left_to_next, |
| 415 | bi0, next0); |
| 416 | } |
| 417 | vlib_put_next_frame (vm, node, next_index, n_left_to_next); |
| 418 | } |
| 419 | |
Florin Coras | 6792ec0 | 2017-03-13 03:49:51 -0700 | [diff] [blame] | 420 | /* If we couldn't dequeue all bytes mark as partially read */ |
| 421 | if (max_len_to_snd0 < max_dequeue0) |
Florin Coras | 7fb0fe1 | 2018-04-09 09:24:52 -0700 | [diff] [blame] | 422 | if (svm_fifo_set_event (s0->server_tx_fifo)) |
| 423 | vec_add1 (smm->pending_event_vector[thread_index], *e0); |
| 424 | |
| 425 | if (!peek_data && transport_vft->tx_type == TRANSPORT_TX_DGRAM) |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 426 | { |
Florin Coras | 7fb0fe1 | 2018-04-09 09:24:52 -0700 | [diff] [blame] | 427 | /* Fix dgram pre header */ |
| 428 | if (max_len_to_snd0 < max_dequeue0) |
| 429 | svm_fifo_overwrite_head (s0->server_tx_fifo, (u8 *) & hdr, |
| 430 | sizeof (session_dgram_pre_hdr_t)); |
| 431 | /* More data needs to be read */ |
| 432 | else if (svm_fifo_max_dequeue (s0->server_tx_fifo) > 0) |
| 433 | vec_add1 (smm->pending_event_vector[thread_index], *e0); |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 434 | } |
| 435 | return 0; |
| 436 | |
| 437 | dequeue_fail: |
Florin Coras | 6792ec0 | 2017-03-13 03:49:51 -0700 | [diff] [blame] | 438 | /* |
| 439 | * Can't read from fifo. If we don't already have an event, save as partially |
| 440 | * read, return buff to free list and return |
| 441 | */ |
| 442 | clib_warning ("dequeue fail"); |
Florin Coras | 6792ec0 | 2017-03-13 03:49:51 -0700 | [diff] [blame] | 443 | if (svm_fifo_set_event (s0->server_tx_fifo)) |
| 444 | { |
Dave Barach | acd2a6a | 2017-05-16 17:41:34 -0400 | [diff] [blame] | 445 | vec_add1 (smm->pending_event_vector[thread_index], *e0); |
Florin Coras | 6792ec0 | 2017-03-13 03:49:51 -0700 | [diff] [blame] | 446 | } |
| 447 | vlib_put_next_frame (vm, node, next_index, n_left_to_next + 1); |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 448 | _vec_len (smm->tx_buffers[thread_index]) += 1; |
| 449 | |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 450 | return 0; |
| 451 | } |
| 452 | |
| 453 | int |
Florin Coras | d79b41e | 2017-03-04 05:37:52 -0800 | [diff] [blame] | 454 | session_tx_fifo_peek_and_snd (vlib_main_t * vm, vlib_node_runtime_t * node, |
| 455 | session_manager_main_t * smm, |
| 456 | session_fifo_event_t * e0, |
| 457 | stream_session_t * s0, u32 thread_index, |
| 458 | int *n_tx_pkts) |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 459 | { |
Florin Coras | d79b41e | 2017-03-04 05:37:52 -0800 | [diff] [blame] | 460 | return session_tx_fifo_read_and_snd_i (vm, node, smm, e0, s0, thread_index, |
| 461 | n_tx_pkts, 1); |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 462 | } |
| 463 | |
| 464 | int |
Florin Coras | d79b41e | 2017-03-04 05:37:52 -0800 | [diff] [blame] | 465 | session_tx_fifo_dequeue_and_snd (vlib_main_t * vm, vlib_node_runtime_t * node, |
| 466 | session_manager_main_t * smm, |
| 467 | session_fifo_event_t * e0, |
| 468 | stream_session_t * s0, u32 thread_index, |
| 469 | int *n_tx_pkts) |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 470 | { |
Florin Coras | d79b41e | 2017-03-04 05:37:52 -0800 | [diff] [blame] | 471 | return session_tx_fifo_read_and_snd_i (vm, node, smm, e0, s0, thread_index, |
| 472 | n_tx_pkts, 0); |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 473 | } |
| 474 | |
Florin Coras | 371ca50 | 2018-02-21 12:07:41 -0800 | [diff] [blame] | 475 | int |
| 476 | session_tx_fifo_dequeue_internal (vlib_main_t * vm, |
| 477 | vlib_node_runtime_t * node, |
| 478 | session_manager_main_t * smm, |
| 479 | session_fifo_event_t * e0, |
| 480 | stream_session_t * s0, u32 thread_index, |
| 481 | int *n_tx_pkts) |
| 482 | { |
| 483 | application_t *app; |
| 484 | app = application_get (s0->opaque); |
| 485 | svm_fifo_unset_event (s0->server_tx_fifo); |
| 486 | return app->cb_fns.builtin_app_tx_callback (s0); |
| 487 | } |
| 488 | |
Dave Barach | 2c25a62 | 2017-06-26 11:35:07 -0400 | [diff] [blame] | 489 | always_inline stream_session_t * |
| 490 | session_event_get_session (session_fifo_event_t * e, u8 thread_index) |
Florin Coras | a546481 | 2017-04-19 13:00:05 -0700 | [diff] [blame] | 491 | { |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 492 | return session_get_if_valid (e->fifo->master_session_index, thread_index); |
Florin Coras | a546481 | 2017-04-19 13:00:05 -0700 | [diff] [blame] | 493 | } |
| 494 | |
Dave Barach | acd2a6a | 2017-05-16 17:41:34 -0400 | [diff] [blame] | 495 | void |
| 496 | dump_thread_0_event_queue (void) |
| 497 | { |
| 498 | session_manager_main_t *smm = vnet_get_session_manager_main (); |
| 499 | vlib_main_t *vm = &vlib_global_main; |
| 500 | u32 my_thread_index = vm->thread_index; |
| 501 | session_fifo_event_t _e, *e = &_e; |
| 502 | stream_session_t *s0; |
| 503 | int i, index; |
| 504 | i8 *headp; |
| 505 | |
Florin Coras | e86a8ed | 2018-01-05 03:20:25 -0800 | [diff] [blame] | 506 | svm_queue_t *q; |
Dave Barach | acd2a6a | 2017-05-16 17:41:34 -0400 | [diff] [blame] | 507 | q = smm->vpp_event_queues[my_thread_index]; |
| 508 | |
| 509 | index = q->head; |
| 510 | |
| 511 | for (i = 0; i < q->cursize; i++) |
| 512 | { |
| 513 | headp = (i8 *) (&q->data[0] + q->elsize * index); |
| 514 | clib_memcpy (e, headp, q->elsize); |
| 515 | |
| 516 | switch (e->event_type) |
| 517 | { |
| 518 | case FIFO_EVENT_APP_TX: |
| 519 | s0 = session_event_get_session (e, my_thread_index); |
| 520 | fformat (stdout, "[%04d] TX session %d\n", i, s0->session_index); |
| 521 | break; |
| 522 | |
| 523 | case FIFO_EVENT_DISCONNECT: |
Florin Coras | cea194d | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 524 | s0 = session_get_from_handle (e->session_handle); |
Dave Barach | acd2a6a | 2017-05-16 17:41:34 -0400 | [diff] [blame] | 525 | fformat (stdout, "[%04d] disconnect session %d\n", i, |
| 526 | s0->session_index); |
| 527 | break; |
| 528 | |
| 529 | case FIFO_EVENT_BUILTIN_RX: |
| 530 | s0 = session_event_get_session (e, my_thread_index); |
| 531 | fformat (stdout, "[%04d] builtin_rx %d\n", i, s0->session_index); |
| 532 | break; |
| 533 | |
| 534 | case FIFO_EVENT_RPC: |
| 535 | fformat (stdout, "[%04d] RPC call %llx with %llx\n", |
| 536 | i, (u64) (e->rpc_args.fp), (u64) (e->rpc_args.arg)); |
| 537 | break; |
| 538 | |
| 539 | default: |
| 540 | fformat (stdout, "[%04d] unhandled event type %d\n", |
| 541 | i, e->event_type); |
| 542 | break; |
| 543 | } |
| 544 | |
| 545 | index++; |
| 546 | |
| 547 | if (index == q->maxsize) |
| 548 | index = 0; |
| 549 | } |
| 550 | } |
| 551 | |
Florin Coras | 6534b7a | 2017-07-18 05:38:03 -0400 | [diff] [blame] | 552 | static u8 |
| 553 | session_node_cmp_event (session_fifo_event_t * e, svm_fifo_t * f) |
| 554 | { |
| 555 | stream_session_t *s; |
| 556 | switch (e->event_type) |
| 557 | { |
| 558 | case FIFO_EVENT_APP_RX: |
| 559 | case FIFO_EVENT_APP_TX: |
| 560 | case FIFO_EVENT_BUILTIN_RX: |
| 561 | if (e->fifo == f) |
| 562 | return 1; |
| 563 | break; |
| 564 | case FIFO_EVENT_DISCONNECT: |
| 565 | break; |
| 566 | case FIFO_EVENT_RPC: |
Florin Coras | cea194d | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 567 | s = session_get_from_handle (e->session_handle); |
Florin Coras | 6534b7a | 2017-07-18 05:38:03 -0400 | [diff] [blame] | 568 | if (!s) |
| 569 | { |
| 570 | clib_warning ("session has event but doesn't exist!"); |
| 571 | break; |
| 572 | } |
| 573 | if (s->server_rx_fifo == f || s->server_tx_fifo == f) |
| 574 | return 1; |
| 575 | break; |
| 576 | default: |
| 577 | break; |
| 578 | } |
| 579 | return 0; |
| 580 | } |
| 581 | |
| 582 | u8 |
| 583 | session_node_lookup_fifo_event (svm_fifo_t * f, session_fifo_event_t * e) |
| 584 | { |
| 585 | session_manager_main_t *smm = vnet_get_session_manager_main (); |
Florin Coras | e86a8ed | 2018-01-05 03:20:25 -0800 | [diff] [blame] | 586 | svm_queue_t *q; |
Florin Coras | 6534b7a | 2017-07-18 05:38:03 -0400 | [diff] [blame] | 587 | session_fifo_event_t *pending_event_vector, *evt; |
| 588 | int i, index, found = 0; |
| 589 | i8 *headp; |
| 590 | u8 thread_index; |
| 591 | |
| 592 | ASSERT (e); |
| 593 | thread_index = f->master_thread_index; |
| 594 | /* |
| 595 | * Search evt queue |
| 596 | */ |
| 597 | q = smm->vpp_event_queues[thread_index]; |
| 598 | index = q->head; |
| 599 | for (i = 0; i < q->cursize; i++) |
| 600 | { |
| 601 | headp = (i8 *) (&q->data[0] + q->elsize * index); |
| 602 | clib_memcpy (e, headp, q->elsize); |
| 603 | found = session_node_cmp_event (e, f); |
| 604 | if (found) |
Florin Coras | 371ca50 | 2018-02-21 12:07:41 -0800 | [diff] [blame] | 605 | return 1; |
Florin Coras | 6534b7a | 2017-07-18 05:38:03 -0400 | [diff] [blame] | 606 | if (++index == q->maxsize) |
| 607 | index = 0; |
| 608 | } |
| 609 | /* |
| 610 | * Search pending events vector |
| 611 | */ |
| 612 | pending_event_vector = smm->pending_event_vector[thread_index]; |
| 613 | vec_foreach (evt, pending_event_vector) |
| 614 | { |
| 615 | found = session_node_cmp_event (evt, f); |
| 616 | if (found) |
| 617 | { |
| 618 | clib_memcpy (e, evt, sizeof (*evt)); |
| 619 | break; |
| 620 | } |
| 621 | } |
| 622 | return found; |
| 623 | } |
| 624 | |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 625 | static uword |
| 626 | session_queue_node_fn (vlib_main_t * vm, vlib_node_runtime_t * node, |
| 627 | vlib_frame_t * frame) |
| 628 | { |
| 629 | session_manager_main_t *smm = vnet_get_session_manager_main (); |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 630 | session_fifo_event_t *my_pending_event_vector, *pending_disconnects, *e; |
Dave Barach | acd2a6a | 2017-05-16 17:41:34 -0400 | [diff] [blame] | 631 | session_fifo_event_t *my_fifo_events; |
Florin Coras | e04c299 | 2017-03-01 08:17:34 -0800 | [diff] [blame] | 632 | u32 n_to_dequeue, n_events; |
Florin Coras | e86a8ed | 2018-01-05 03:20:25 -0800 | [diff] [blame] | 633 | svm_queue_t *q; |
Florin Coras | 6792ec0 | 2017-03-13 03:49:51 -0700 | [diff] [blame] | 634 | application_t *app; |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 635 | int n_tx_packets = 0; |
Damjan Marion | 586afd7 | 2017-04-05 19:18:20 +0200 | [diff] [blame] | 636 | u32 my_thread_index = vm->thread_index; |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 637 | int i, rv; |
Florin Coras | 3e350af | 2017-03-30 02:54:28 -0700 | [diff] [blame] | 638 | f64 now = vlib_time_now (vm); |
Dave Barach | e5f1d27 | 2017-05-10 13:34:04 -0400 | [diff] [blame] | 639 | void (*fp) (void *); |
Florin Coras | 3e350af | 2017-03-30 02:54:28 -0700 | [diff] [blame] | 640 | |
| 641 | SESSION_EVT_DBG (SESSION_EVT_POLL_GAP_TRACK, smm, my_thread_index); |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 642 | |
| 643 | /* |
Florin Coras | 561af9b | 2017-12-09 10:19:43 -0800 | [diff] [blame] | 644 | * Update transport time |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 645 | */ |
Florin Coras | 561af9b | 2017-12-09 10:19:43 -0800 | [diff] [blame] | 646 | transport_update_time (now, my_thread_index); |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 647 | |
| 648 | /* |
| 649 | * Get vpp queue events |
| 650 | */ |
| 651 | q = smm->vpp_event_queues[my_thread_index]; |
| 652 | if (PREDICT_FALSE (q == 0)) |
| 653 | return 0; |
| 654 | |
Dave Barach | acd2a6a | 2017-05-16 17:41:34 -0400 | [diff] [blame] | 655 | my_fifo_events = smm->free_event_vector[my_thread_index]; |
| 656 | |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 657 | /* min number of events we can dequeue without blocking */ |
| 658 | n_to_dequeue = q->cursize; |
Dave Barach | acd2a6a | 2017-05-16 17:41:34 -0400 | [diff] [blame] | 659 | my_pending_event_vector = smm->pending_event_vector[my_thread_index]; |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 660 | pending_disconnects = smm->pending_disconnects[my_thread_index]; |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 661 | |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 662 | if (!n_to_dequeue && !vec_len (my_pending_event_vector) |
| 663 | && !vec_len (pending_disconnects)) |
Florin Coras | e04c299 | 2017-03-01 08:17:34 -0800 | [diff] [blame] | 664 | return 0; |
| 665 | |
Florin Coras | 6792ec0 | 2017-03-13 03:49:51 -0700 | [diff] [blame] | 666 | SESSION_EVT_DBG (SESSION_EVT_DEQ_NODE, 0); |
| 667 | |
Florin Coras | e04c299 | 2017-03-01 08:17:34 -0800 | [diff] [blame] | 668 | /* |
| 669 | * If we didn't manage to process previous events try going |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 670 | * over them again without dequeuing new ones. |
Florin Coras | e04c299 | 2017-03-01 08:17:34 -0800 | [diff] [blame] | 671 | */ |
| 672 | /* XXX: Block senders to sessions that can't keep up */ |
Dave Barach | acd2a6a | 2017-05-16 17:41:34 -0400 | [diff] [blame] | 673 | if (0 && vec_len (my_pending_event_vector) >= 100) |
Florin Coras | 6792ec0 | 2017-03-13 03:49:51 -0700 | [diff] [blame] | 674 | { |
| 675 | clib_warning ("too many fifo events unsolved"); |
| 676 | goto skip_dequeue; |
| 677 | } |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 678 | |
| 679 | /* See you in the next life, don't be late */ |
| 680 | if (pthread_mutex_trylock (&q->mutex)) |
| 681 | return 0; |
| 682 | |
| 683 | for (i = 0; i < n_to_dequeue; i++) |
| 684 | { |
| 685 | vec_add2 (my_fifo_events, e, 1); |
Florin Coras | e86a8ed | 2018-01-05 03:20:25 -0800 | [diff] [blame] | 686 | svm_queue_sub_raw (q, (u8 *) e); |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 687 | } |
| 688 | |
| 689 | /* The other side of the connection is not polling */ |
| 690 | if (q->cursize < (q->maxsize / 8)) |
| 691 | (void) pthread_cond_broadcast (&q->condvar); |
| 692 | pthread_mutex_unlock (&q->mutex); |
| 693 | |
Dave Barach | acd2a6a | 2017-05-16 17:41:34 -0400 | [diff] [blame] | 694 | vec_append (my_fifo_events, my_pending_event_vector); |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 695 | vec_append (my_fifo_events, smm->pending_disconnects[my_thread_index]); |
Dave Barach | acd2a6a | 2017-05-16 17:41:34 -0400 | [diff] [blame] | 696 | |
| 697 | _vec_len (my_pending_event_vector) = 0; |
| 698 | smm->pending_event_vector[my_thread_index] = my_pending_event_vector; |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 699 | _vec_len (smm->pending_disconnects[my_thread_index]) = 0; |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 700 | |
| 701 | skip_dequeue: |
Florin Coras | e04c299 | 2017-03-01 08:17:34 -0800 | [diff] [blame] | 702 | n_events = vec_len (my_fifo_events); |
| 703 | for (i = 0; i < n_events; i++) |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 704 | { |
Florin Coras | a546481 | 2017-04-19 13:00:05 -0700 | [diff] [blame] | 705 | stream_session_t *s0; /* $$$ prefetch 1 ahead maybe */ |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 706 | session_fifo_event_t *e0; |
| 707 | |
| 708 | e0 = &my_fifo_events[i]; |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 709 | |
| 710 | switch (e0->event_type) |
| 711 | { |
Florin Coras | a546481 | 2017-04-19 13:00:05 -0700 | [diff] [blame] | 712 | case FIFO_EVENT_APP_TX: |
| 713 | s0 = session_event_get_session (e0, my_thread_index); |
| 714 | |
Florin Coras | 9d06304 | 2017-09-14 03:08:00 -0400 | [diff] [blame] | 715 | if (PREDICT_FALSE (!s0)) |
Florin Coras | a546481 | 2017-04-19 13:00:05 -0700 | [diff] [blame] | 716 | { |
| 717 | clib_warning ("It's dead, Jim!"); |
| 718 | continue; |
| 719 | } |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 720 | /* Spray packets in per session type frames, since they go to |
| 721 | * different nodes */ |
Florin Coras | e69f495 | 2017-03-07 10:06:24 -0800 | [diff] [blame] | 722 | rv = (smm->session_tx_fns[s0->session_type]) (vm, node, smm, e0, s0, |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 723 | my_thread_index, |
| 724 | &n_tx_packets); |
Florin Coras | 6792ec0 | 2017-03-13 03:49:51 -0700 | [diff] [blame] | 725 | /* Out of buffers */ |
Florin Coras | 93992a9 | 2017-05-24 18:03:56 -0700 | [diff] [blame] | 726 | if (PREDICT_FALSE (rv < 0)) |
Dave Barach | acd2a6a | 2017-05-16 17:41:34 -0400 | [diff] [blame] | 727 | { |
Florin Coras | 93992a9 | 2017-05-24 18:03:56 -0700 | [diff] [blame] | 728 | vlib_node_increment_counter (vm, node->node_index, |
| 729 | SESSION_QUEUE_ERROR_NO_BUFFER, 1); |
Dave Barach | acd2a6a | 2017-05-16 17:41:34 -0400 | [diff] [blame] | 730 | continue; |
| 731 | } |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 732 | break; |
Florin Coras | a546481 | 2017-04-19 13:00:05 -0700 | [diff] [blame] | 733 | case FIFO_EVENT_DISCONNECT: |
Florin Coras | 3cbc04b | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 734 | /* Make sure disconnects run after the pending list is drained */ |
| 735 | if (!e0->postponed) |
| 736 | { |
| 737 | e0->postponed = 1; |
| 738 | vec_add1 (smm->pending_disconnects[my_thread_index], *e0); |
| 739 | continue; |
| 740 | } |
Florin Coras | cea194d | 2017-10-02 00:18:51 -0700 | [diff] [blame] | 741 | s0 = session_get_from_handle (e0->session_handle); |
Florin Coras | 2f8d8fa | 2018-01-26 06:36:04 -0800 | [diff] [blame] | 742 | stream_session_disconnect_transport (s0); |
Florin Coras | 6792ec0 | 2017-03-13 03:49:51 -0700 | [diff] [blame] | 743 | break; |
| 744 | case FIFO_EVENT_BUILTIN_RX: |
Florin Coras | a546481 | 2017-04-19 13:00:05 -0700 | [diff] [blame] | 745 | s0 = session_event_get_session (e0, my_thread_index); |
Florin Coras | c87c91d | 2017-08-16 19:55:49 -0700 | [diff] [blame] | 746 | if (PREDICT_FALSE (!s0)) |
| 747 | continue; |
Florin Coras | 6792ec0 | 2017-03-13 03:49:51 -0700 | [diff] [blame] | 748 | svm_fifo_unset_event (s0->server_rx_fifo); |
Florin Coras | 6792ec0 | 2017-03-13 03:49:51 -0700 | [diff] [blame] | 749 | app = application_get (s0->app_index); |
Florin Coras | 371ca50 | 2018-02-21 12:07:41 -0800 | [diff] [blame] | 750 | app->cb_fns.builtin_app_rx_callback (s0); |
Florin Coras | 6792ec0 | 2017-03-13 03:49:51 -0700 | [diff] [blame] | 751 | break; |
Dave Barach | e5f1d27 | 2017-05-10 13:34:04 -0400 | [diff] [blame] | 752 | case FIFO_EVENT_RPC: |
| 753 | fp = e0->rpc_args.fp; |
| 754 | (*fp) (e0->rpc_args.arg); |
| 755 | break; |
| 756 | |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 757 | default: |
| 758 | clib_warning ("unhandled event type %d", e0->event_type); |
| 759 | } |
| 760 | } |
| 761 | |
Dave Barach | acd2a6a | 2017-05-16 17:41:34 -0400 | [diff] [blame] | 762 | _vec_len (my_fifo_events) = 0; |
| 763 | smm->free_event_vector[my_thread_index] = my_fifo_events; |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 764 | |
| 765 | vlib_node_increment_counter (vm, session_queue_node.index, |
| 766 | SESSION_QUEUE_ERROR_TX, n_tx_packets); |
| 767 | |
Florin Coras | 6792ec0 | 2017-03-13 03:49:51 -0700 | [diff] [blame] | 768 | SESSION_EVT_DBG (SESSION_EVT_DEQ_NODE, 1); |
| 769 | |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 770 | return n_tx_packets; |
| 771 | } |
| 772 | |
| 773 | /* *INDENT-OFF* */ |
| 774 | VLIB_REGISTER_NODE (session_queue_node) = |
| 775 | { |
| 776 | .function = session_queue_node_fn, |
| 777 | .name = "session-queue", |
| 778 | .format_trace = format_session_queue_trace, |
| 779 | .type = VLIB_NODE_TYPE_INPUT, |
| 780 | .n_errors = ARRAY_LEN (session_queue_error_strings), |
| 781 | .error_strings = session_queue_error_strings, |
Florin Coras | e04c299 | 2017-03-01 08:17:34 -0800 | [diff] [blame] | 782 | .state = VLIB_NODE_STATE_DISABLED, |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 783 | }; |
| 784 | /* *INDENT-ON* */ |
| 785 | |
Dave Barach | 2a86391 | 2017-11-28 10:11:42 -0500 | [diff] [blame] | 786 | static clib_error_t * |
| 787 | session_queue_exit (vlib_main_t * vm) |
| 788 | { |
| 789 | if (vec_len (vlib_mains) < 2) |
| 790 | return 0; |
| 791 | |
| 792 | /* |
| 793 | * Shut off (especially) worker-thread session nodes. |
| 794 | * Otherwise, vpp can crash as the main thread unmaps the |
| 795 | * API segment. |
| 796 | */ |
| 797 | vlib_worker_thread_barrier_sync (vm); |
| 798 | session_node_enable_disable (0 /* is_enable */ ); |
| 799 | vlib_worker_thread_barrier_release (vm); |
| 800 | return 0; |
| 801 | } |
| 802 | |
| 803 | VLIB_MAIN_LOOP_EXIT_FUNCTION (session_queue_exit); |
| 804 | |
Dave Barach | 68b0fb0 | 2017-02-28 15:15:56 -0500 | [diff] [blame] | 805 | /* |
| 806 | * fd.io coding-style-patch-verification: ON |
| 807 | * |
| 808 | * Local Variables: |
| 809 | * eval: (c-set-style "gnu") |
| 810 | * End: |
| 811 | */ |