diff options
Diffstat (limited to 'src/vnet/sctp')
-rw-r--r-- | src/vnet/sctp/builtin_client.c | 829 | ||||
-rw-r--r-- | src/vnet/sctp/builtin_client.h | 121 | ||||
-rw-r--r-- | src/vnet/sctp/builtin_server.c | 469 |
3 files changed, 0 insertions, 1419 deletions
diff --git a/src/vnet/sctp/builtin_client.c b/src/vnet/sctp/builtin_client.c deleted file mode 100644 index 12f053297c2..00000000000 --- a/src/vnet/sctp/builtin_client.c +++ /dev/null @@ -1,829 +0,0 @@ -/* - * Copyright (c) 2018 SUSE LLC. - * 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 <vnet/vnet.h> -#include <vnet/plugin/plugin.h> -#include <vnet/sctp/builtin_client.h> - -#include <vlibapi/api.h> -#include <vlibmemory/api.h> -#include <vpp/app/version.h> - -tclient_main_t tclient_main; - -#define SCTP_BUILTIN_CLIENT_DBG (0) - -static void -signal_evt_to_cli_i (int *code) -{ - tclient_main_t *tm = &tclient_main; - ASSERT (vlib_get_thread_index () == 0); - vlib_process_signal_event (tm->vlib_main, tm->cli_node_index, *code, 0); -} - -static void -signal_evt_to_cli (int code) -{ - if (vlib_get_thread_index () != 0) - vl_api_rpc_call_main_thread (signal_evt_to_cli_i, (u8 *) & code, - sizeof (code)); - else - signal_evt_to_cli_i (&code); -} - -static void -send_test_chunk (tclient_main_t * tm, session_t * s) -{ - u8 *test_data = tm->connect_test_data; - int test_buf_offset; - u32 bytes_this_chunk; - session_fifo_event_t evt; - svm_fifo_t *txf; - int rv; - - ASSERT (vec_len (test_data) > 0); - - test_buf_offset = s->bytes_sent % vec_len (test_data); - bytes_this_chunk = vec_len (test_data) - test_buf_offset; - bytes_this_chunk = bytes_this_chunk < s->bytes_to_send - ? bytes_this_chunk : s->bytes_to_send; - - txf = s->server_tx_fifo; - rv = svm_fifo_enqueue_nowait (txf, bytes_this_chunk, - test_data + test_buf_offset); - - /* If we managed to enqueue data... */ - if (rv > 0) - { - /* Account for it... */ - s->bytes_to_send -= rv; - s->bytes_sent += rv; - - if (SCTP_BUILTIN_CLIENT_DBG) - { - /* *INDENT-OFF* */ - ELOG_TYPE_DECLARE (e) = - { - .format = "tx-enq: xfer %d bytes, sent %u remain %u", - .format_args = "i4i4i4", - }; - /* *INDENT-ON* */ - struct - { - u32 data[3]; - } *ed; - ed = ELOG_DATA (&vlib_global_main.elog_main, e); - ed->data[0] = rv; - ed->data[1] = s->bytes_sent; - ed->data[2] = s->bytes_to_send; - } - - /* Poke the session layer */ - if (svm_fifo_set_event (txf)) - { - /* Fabricate TX event, send to vpp */ - evt.fifo = txf; - evt.event_type = FIFO_EVENT_APP_TX; - - if (svm_queue_add - (tm->vpp_event_queue[txf->master_thread_index], (u8 *) & evt, - 0 /* do wait for mutex */ )) - clib_warning ("could not enqueue event"); - } - } -} - -static void -receive_test_chunk (tclient_main_t * tm, session_t * s) -{ - svm_fifo_t *rx_fifo = s->server_rx_fifo; - u32 my_thread_index = vlib_get_thread_index (); - int n_read, i; - - /* Allow enqueuing of new event */ - // svm_fifo_unset_event (rx_fifo); - - if (tm->test_bytes) - { - n_read = svm_fifo_dequeue_nowait (rx_fifo, - vec_len (tm->rx_buf[my_thread_index]), - tm->rx_buf[my_thread_index]); - } - else - { - n_read = svm_fifo_max_dequeue (rx_fifo); - svm_fifo_dequeue_drop (rx_fifo, n_read); - } - - if (SCTP_BUILTIN_CLIENT_DBG) - clib_warning ("Receiving test chunk; n_read = %d", n_read); - - if (n_read > 0) - { - if (SCTP_BUILTIN_CLIENT_DBG) - { - /* *INDENT-OFF* */ - ELOG_TYPE_DECLARE (e) = - { - .format = "rx-deq: %d bytes", - .format_args = "i4", - }; - /* *INDENT-ON* */ - struct - { - u32 data[1]; - } *ed; - ed = ELOG_DATA (&vlib_global_main.elog_main, e); - ed->data[0] = n_read; - } - - if (tm->test_bytes) - { - for (i = 0; i < n_read; i++) - { - if (tm->rx_buf[my_thread_index][i] - != ((s->bytes_received + i) & 0xff)) - { - clib_warning ("read %d error at byte %lld, 0x%x not 0x%x", - n_read, s->bytes_received + i, - tm->rx_buf[my_thread_index][i], - ((s->bytes_received + i) & 0xff)); - tm->test_failed = 1; - } - } - } - - if (s->bytes_to_receive < n_read) - { - s->bytes_to_receive = 0; - s->bytes_received += s->bytes_received; - } - else - { - s->bytes_to_receive -= n_read; - s->bytes_received += n_read; - } - } -} - -static uword -builtin_client_node_fn (vlib_main_t * vm, vlib_node_runtime_t * node, - vlib_frame_t * frame) -{ - tclient_main_t *tm = &tclient_main; - int my_thread_index = vlib_get_thread_index (); - session_t *sp; - int i; - int delete_session; - u32 *connection_indices; - u32 *connections_this_batch; - u32 nconnections_this_batch; - - connection_indices = tm->connection_index_by_thread[my_thread_index]; - connections_this_batch = - tm->connections_this_batch_by_thread[my_thread_index]; - - if ((tm->run_test == 0) || - ((vec_len (connection_indices) == 0) - && vec_len (connections_this_batch) == 0)) - return 0; - - /* Grab another pile of connections */ - if (PREDICT_FALSE (vec_len (connections_this_batch) == 0)) - { - nconnections_this_batch = - clib_min (tm->connections_per_batch, vec_len (connection_indices)); - - ASSERT (nconnections_this_batch > 0); - vec_validate (connections_this_batch, nconnections_this_batch - 1); - clib_memcpy (connections_this_batch, - connection_indices + vec_len (connection_indices) - - nconnections_this_batch, - nconnections_this_batch * sizeof (u32)); - _vec_len (connection_indices) -= nconnections_this_batch; - } - - if (PREDICT_FALSE (tm->prev_conns != tm->connections_per_batch - && tm->prev_conns == vec_len (connections_this_batch))) - { - tm->repeats++; - tm->prev_conns = vec_len (connections_this_batch); - if (tm->repeats == 500000) - { - clib_warning ("stuck clients"); - } - } - else - { - tm->prev_conns = vec_len (connections_this_batch); - tm->repeats = 0; - } - - for (i = 0; i < vec_len (connections_this_batch); i++) - { - delete_session = 1; - - sp = pool_elt_at_index (tm->sessions, connections_this_batch[i]); - - if (sp->bytes_to_send > 0) - { - send_test_chunk (tm, sp); - delete_session = 0; - } - - if (sp->bytes_to_receive > 0) - { - receive_test_chunk (tm, sp); - delete_session = 0; - } - if (PREDICT_FALSE (delete_session == 1)) - { - u32 index, thread_index; - stream_session_t *s; - - __sync_fetch_and_add (&tm->tx_total, sp->bytes_sent); - __sync_fetch_and_add (&tm->rx_total, sp->bytes_received); - - session_parse_handle (sp->vpp_session_handle, - &index, &thread_index); - s = session_get_if_valid (index, thread_index); - - if (s) - { - vnet_disconnect_args_t _a, *a = &_a; - a->handle = session_handle (s); - a->app_index = tm->app_index; - vnet_disconnect_session (a); - - vec_delete (connections_this_batch, 1, i); - i--; - __sync_fetch_and_add (&tm->ready_connections, -1); - } - else - clib_warning ("session AWOL?"); - - /* Kick the debug CLI process */ - if (tm->ready_connections == 0) - { - signal_evt_to_cli (2); - } - } - } - - tm->connection_index_by_thread[my_thread_index] = connection_indices; - tm->connections_this_batch_by_thread[my_thread_index] = - connections_this_batch; - return 0; -} - -/* *INDENT-OFF* */ -VLIB_REGISTER_NODE (builtin_sctp_client_node) = -{ - .function = builtin_client_node_fn, - .name = "builtin-sctp-client", - .type = VLIB_NODE_TYPE_INPUT, - .state = VLIB_NODE_STATE_DISABLED, -}; -/* *INDENT-ON* */ - -static int -create_api_loopback (tclient_main_t * tm) -{ - api_main_t *am = &api_main; - vl_shmem_hdr_t *shmem_hdr; - - shmem_hdr = am->shmem_hdr; - tm->vl_input_queue = shmem_hdr->vl_input_queue; - tm->my_client_index = - vl_api_memclnt_create_internal ("sctp_test_client", tm->vl_input_queue); - return 0; -} - -static int -sctp_test_clients_init (vlib_main_t * vm) -{ - tclient_main_t *tm = &tclient_main; - vlib_thread_main_t *vtm = vlib_get_thread_main (); - u32 num_threads; - int i; - - if (create_api_loopback (tm)) - return -1; - - num_threads = 1 /* main thread */ + vtm->n_threads; - - /* Init test data. Big buffer */ - vec_validate (tm->connect_test_data, 1024 * 1024 - 1); - for (i = 0; i < vec_len (tm->connect_test_data); i++) - tm->connect_test_data[i] = i & 0xff; - - vec_validate (tm->rx_buf, num_threads - 1); - for (i = 0; i < num_threads; i++) - vec_validate (tm->rx_buf[i], vec_len (tm->connect_test_data) - 1); - - tm->is_init = 1; - - vec_validate (tm->connection_index_by_thread, vtm->n_vlib_mains); - vec_validate (tm->connections_this_batch_by_thread, vtm->n_vlib_mains); - vec_validate (tm->vpp_event_queue, vtm->n_vlib_mains); - - return 0; -} - -static int -builtin_session_connected_callback (u32 app_index, u32 api_context, - stream_session_t * s, u8 is_fail) -{ - tclient_main_t *tm = &tclient_main; - session_t *session; - u32 session_index; - u8 thread_index = vlib_get_thread_index (); - - if (is_fail) - { - clib_warning ("connection %d failed!", api_context); - signal_evt_to_cli (-1); - return 0; - } - - ASSERT (s->thread_index == thread_index); - - if (!tm->vpp_event_queue[thread_index]) - tm->vpp_event_queue[thread_index] = - session_manager_get_vpp_event_queue (thread_index); - - /* - * Setup session - */ - clib_spinlock_lock_if_init (&tm->sessions_lock); - pool_get (tm->sessions, session); - clib_spinlock_unlock_if_init (&tm->sessions_lock); - - memset (session, 0, sizeof (*session)); - session_index = session - tm->sessions; - session->bytes_to_send = tm->bytes_to_send; - session->bytes_to_receive = tm->no_return ? 0ULL : tm->bytes_to_send; - session->server_rx_fifo = s->server_rx_fifo; - session->server_rx_fifo->client_session_index = session_index; - session->server_tx_fifo = s->server_tx_fifo; - session->server_tx_fifo->client_session_index = session_index; - session->vpp_session_handle = session_handle (s); - - vec_add1 (tm->connection_index_by_thread[thread_index], session_index); - __sync_fetch_and_add (&tm->ready_connections, 1); - if (tm->ready_connections == tm->expected_connections) - { - tm->run_test = 1; - /* Signal the CLI process that the action is starting... */ - signal_evt_to_cli (1); - } - - return 0; -} - -static void -builtin_session_reset_callback (stream_session_t * s) -{ - if (s->session_state == SESSION_STATE_READY) - clib_warning ("Reset active connection %U", format_stream_session, s, 2); - stream_session_cleanup (s); - return; -} - -static int -builtin_session_create_callback (stream_session_t * s) -{ - return 0; -} - -static void -builtin_session_disconnect_callback (stream_session_t * s) -{ - tclient_main_t *tm = &tclient_main; - vnet_disconnect_args_t _a, *a = &_a; - a->handle = session_handle (s); - a->app_index = tm->app_index; - vnet_disconnect_session (a); - return; -} - -static int -builtin_server_rx_callback (stream_session_t * s) -{ - return 0; -} - -/* *INDENT-OFF* */ -static session_cb_vft_t builtin_clients = { - .session_reset_callback = builtin_session_reset_callback, - .session_connected_callback = builtin_session_connected_callback, - .session_accept_callback = builtin_session_create_callback, - .session_disconnect_callback = builtin_session_disconnect_callback, - .builtin_server_rx_callback = builtin_server_rx_callback -}; -/* *INDENT-ON* */ - -static clib_error_t * -attach_builtin_test_clients_app (u8 * appns_id, u64 appns_flags, - u64 appns_secret) -{ - u32 prealloc_fifos, segment_size = 2 << 20; - tclient_main_t *tm = &tclient_main; - vnet_app_attach_args_t _a, *a = &_a; - u64 options[16]; - clib_error_t *error = 0; - - memset (a, 0, sizeof (*a)); - memset (options, 0, sizeof (options)); - - a->api_client_index = tm->my_client_index; - a->session_cb_vft = &builtin_clients; - - prealloc_fifos = tm->prealloc_fifos ? tm->expected_connections : 1; - - if (tm->private_segment_size) - segment_size = tm->private_segment_size; - - options[APP_OPTIONS_ACCEPT_COOKIE] = 0x12345678; - options[APP_OPTIONS_SEGMENT_SIZE] = segment_size; - options[APP_OPTIONS_RX_FIFO_SIZE] = tm->fifo_size; - options[APP_OPTIONS_TX_FIFO_SIZE] = tm->fifo_size; - options[APP_OPTIONS_PRIVATE_SEGMENT_COUNT] = tm->private_segment_count; - options[APP_OPTIONS_PREALLOC_FIFO_PAIRS] = prealloc_fifos; - - options[APP_OPTIONS_FLAGS] = APP_OPTIONS_FLAGS_IS_BUILTIN; - if (appns_id) - { - options[APP_OPTIONS_FLAGS] |= appns_flags; - options[APP_OPTIONS_NAMESPACE_SECRET] = appns_secret; - } - a->options = options; - a->namespace_id = appns_id; - - if ((error = vnet_application_attach (a))) - return error; - - tm->app_index = a->app_index; - return 0; -} - -static void * -tclient_thread_fn (void *arg) -{ - return 0; -} - -/** Start a transmit thread */ -int -start_tx_pthread_sctp (tclient_main_t * tm) -{ - if (tm->client_thread_handle == 0) - { - int rv = pthread_create (&tm->client_thread_handle, - NULL /*attr */ , - tclient_thread_fn, 0); - if (rv) - { - tm->client_thread_handle = 0; - return -1; - } - } - return 0; -} - -clib_error_t * -clients_connect_sctp (vlib_main_t * vm, u8 * uri, u32 n_clients) -{ - tclient_main_t *tm = &tclient_main; - vnet_connect_args_t _a, *a = &_a; - clib_error_t *error = 0; - int i; - for (i = 0; i < n_clients; i++) - { - memset (a, 0, sizeof (*a)); - - a->uri = (char *) uri; - a->api_context = i; - a->app_index = tm->app_index; - a->mp = 0; - - if ((error = vnet_connect_uri (a))) - return error; - - - /* Crude pacing for call setups */ - if ((i % 4) == 0) - vlib_process_suspend (vm, 10e-6); - ASSERT (i + 1 >= tm->ready_connections); - while (i + 1 - tm->ready_connections > 1000) - { - vlib_process_suspend (vm, 100e-6); - } - } - return 0; -} - -#define CLI_OUTPUT(_fmt, _args...) \ - if (!tm->no_output) \ - vlib_cli_output(vm, _fmt, ##_args) - -static clib_error_t * -test_sctp_clients_command_fn (vlib_main_t * vm, - unformat_input_t * input, - vlib_cli_command_t * cmd) -{ - tclient_main_t *tm = &tclient_main; - vlib_thread_main_t *thread_main = vlib_get_thread_main (); - uword *event_data = 0, event_type; - u8 *default_connect_uri = (u8 *) "sctp://6.0.1.1/1234", *uri, *appns_id = 0; - u64 tmp, total_bytes, appns_flags = 0, appns_secret = 0; - f64 test_timeout = 20.0, syn_timeout = 20.0, delta; - f64 time_before_connects; - u32 n_clients = 1; - int preallocate_sessions = 0; - char *transfer_type; - clib_error_t *error = 0; - int i; - - tm->bytes_to_send = 8192; - tm->no_return = 0; - tm->fifo_size = 64 << 10; - tm->connections_per_batch = 1000; - tm->private_segment_count = 0; - tm->private_segment_size = 0; - tm->no_output = 0; - tm->test_bytes = 0; - tm->test_failed = 0; - tm->vlib_main = vm; - if (thread_main->n_vlib_mains > 1) - clib_spinlock_init (&tm->sessions_lock); - vec_free (tm->connect_uri); - - while (unformat_check_input (input) != UNFORMAT_END_OF_INPUT) - { - if (unformat (input, "nclients %d", &n_clients)) - ; - else if (unformat (input, "mbytes %lld", &tmp)) - tm->bytes_to_send = tmp << 20; - else if (unformat (input, "gbytes %lld", &tmp)) - tm->bytes_to_send = tmp << 30; - else if (unformat (input, "bytes %lld", &tm->bytes_to_send)) - ; - else if (unformat (input, "uri %s", &tm->connect_uri)) - ; - else if (unformat (input, "test-timeout %f", &test_timeout)) - ; - else if (unformat (input, "syn-timeout %f", &syn_timeout)) - ; - else if (unformat (input, "no-return")) - tm->no_return = 1; - else if (unformat (input, "fifo-size %d", &tm->fifo_size)) - tm->fifo_size <<= 10; - else if (unformat (input, "private-segment-count %d", - &tm->private_segment_count)) - ; - else if (unformat (input, "private-segment-size %U", - unformat_memory_size, &tmp)) - { - if (tmp >= 0x100000000ULL) - return clib_error_return - (0, "private segment size %lld (%llu) too large", tmp, tmp); - tm->private_segment_size = tmp; - } - else if (unformat (input, "preallocate-fifos")) - tm->prealloc_fifos = 1; - else if (unformat (input, "preallocate-sessions")) - preallocate_sessions = 1; - else - if (unformat (input, "client-batch %d", &tm->connections_per_batch)) - ; - else if (unformat (input, "appns %_%v%_", &appns_id)) - ; - else if (unformat (input, "all-scope")) - appns_flags |= (APP_OPTIONS_FLAGS_USE_GLOBAL_SCOPE - | APP_OPTIONS_FLAGS_USE_LOCAL_SCOPE); - else if (unformat (input, "local-scope")) - appns_flags = APP_OPTIONS_FLAGS_USE_LOCAL_SCOPE; - else if (unformat (input, "global-scope")) - appns_flags = APP_OPTIONS_FLAGS_USE_GLOBAL_SCOPE; - else if (unformat (input, "secret %lu", &appns_secret)) - ; - else if (unformat (input, "no-output")) - tm->no_output = 1; - else if (unformat (input, "test-bytes")) - tm->test_bytes = 1; - else - return clib_error_return (0, "unknown input `%U'", - format_unformat_error, input); - } - - /* Store cli process node index for signalling */ - tm->cli_node_index = vlib_get_current_process (vm)->node_runtime.node_index; - - if (tm->is_init == 0) - { - if (sctp_test_clients_init (vm)) - return clib_error_return (0, "failed init"); - } - - tm->ready_connections = 0; - tm->expected_connections = n_clients; - tm->rx_total = 0; - tm->tx_total = 0; - - uri = default_connect_uri; - if (tm->connect_uri) - uri = tm->connect_uri; - -#if SCTP_BUILTIN_CLIENT_PTHREAD - start_tx_pthread (); -#endif - - vlib_worker_thread_barrier_sync (vm); - vnet_session_enable_disable (vm, 1 /* turn on SCTP, etc. */ ); - vlib_worker_thread_barrier_release (vm); - - if (tm->test_client_attached == 0) - { - if ((error = attach_builtin_test_clients_app (appns_id, appns_flags, - appns_secret))) - { - vec_free (appns_id); - clib_error_report (error); - return error; - } - vec_free (appns_id); - } - tm->test_client_attached = 1; - - /* Turn on the builtin client input nodes */ - for (i = 0; i < thread_main->n_vlib_mains; i++) - vlib_node_set_state (vlib_mains[i], builtin_sctp_client_node.index, - VLIB_NODE_STATE_POLLING); - - if (preallocate_sessions) - { - session_t *sp __attribute__ ((unused)); - for (i = 0; i < n_clients; i++) - pool_get (tm->sessions, sp); - for (i = 0; i < n_clients; i++) - pool_put_index (tm->sessions, i); - } - - /* Fire off connect requests */ - time_before_connects = vlib_time_now (vm); - if ((error = clients_connect_sctp (vm, uri, n_clients))) - return error; - - /* Park until the sessions come up, or ten seconds elapse... */ - vlib_process_wait_for_event_or_clock (vm, syn_timeout); - - event_type = vlib_process_get_events (vm, &event_data); - switch (event_type) - { - case ~0: - CLI_OUTPUT ("Timeout with only %d sessions active...", - tm->ready_connections); - error = - clib_error_return (0, "failed: syn timeout (%f) with %d sessions", - syn_timeout, tm->ready_connections); - goto cleanup; - - case 1: - delta = vlib_time_now (vm) - time_before_connects; - if (delta != 0.0) - CLI_OUTPUT ("%d three-way handshakes in %.2f seconds %.2f/s", - n_clients, delta, ((f64) n_clients) / delta); - - tm->test_start_time = vlib_time_now (tm->vlib_main); - CLI_OUTPUT ("Test started at %.6f", tm->test_start_time); - break; - - default: - CLI_OUTPUT ("unexpected event(1): %d", event_type); - error = clib_error_return (0, "failed: unexpected event(1): %d", - event_type); - goto cleanup; - } - - /* Now wait for the sessions to finish... */ - vlib_process_wait_for_event_or_clock (vm, test_timeout); - event_type = vlib_process_get_events (vm, &event_data); - switch (event_type) - { - case ~0: - CLI_OUTPUT ("Timeout with %d sessions still active...", - tm->ready_connections); - error = clib_error_return (0, "failed: timeout with %d sessions", - tm->ready_connections); - goto cleanup; - - case 2: - tm->test_end_time = vlib_time_now (vm); - CLI_OUTPUT ("Test finished at %.6f", tm->test_end_time); - break; - - default: - CLI_OUTPUT ("unexpected event(2): %d", event_type); - error = clib_error_return (0, "failed: unexpected event(2): %d", - event_type); - goto cleanup; - } - - delta = tm->test_end_time - tm->test_start_time; - - if (delta != 0.0) - { - total_bytes = (tm->no_return ? tm->tx_total : tm->rx_total); - transfer_type = tm->no_return ? "half-duplex" : "full-duplex"; - CLI_OUTPUT ("%lld bytes (%lld mbytes, %lld gbytes) in %.2f seconds", - total_bytes, total_bytes / (1ULL << 20), - total_bytes / (1ULL << 30), delta); - CLI_OUTPUT ("%.2f bytes/second %s", ((f64) total_bytes) / (delta), - transfer_type); - CLI_OUTPUT ("%.4f gbit/second %s", - (((f64) total_bytes * 8.0) / delta / 1e9), transfer_type); - } - else - { - CLI_OUTPUT ("zero delta-t?"); - error = clib_error_return (0, "failed: zero delta-t"); - goto cleanup; - } - - if (tm->test_bytes && tm->test_failed) - error = clib_error_return (0, "failed: test bytes"); - -cleanup: - tm->run_test = 0; - for (i = 0; i < vec_len (tm->connection_index_by_thread); i++) - { - vec_reset_length (tm->connection_index_by_thread[i]); - vec_reset_length (tm->connections_this_batch_by_thread[i]); - } - - pool_free (tm->sessions); - - /* Detach the application, so we can use different fifo sizes next time */ - if (tm->test_client_attached) - { - vnet_app_detach_args_t _da, *da = &_da; - int rv; - - da->app_index = tm->app_index; - rv = vnet_application_detach (da); - if (rv) - { - error = clib_error_return (0, "failed: app detach"); - CLI_OUTPUT ("WARNING: app detach failed..."); - } - tm->test_client_attached = 0; - tm->app_index = ~0; - } - if (error) - CLI_OUTPUT ("test failed"); - return error; -} - -/* *INDENT-OFF* */ -VLIB_CLI_COMMAND (test_clients_command, static) = -{ - .path = "test sctp clients", - .short_help = "test sctp clients [nclients %d] [[m|g]bytes <bytes>] " - "[test-timeout <time>][syn-timeout <time>][no-return][fifo-size <size>]" - "[private-segment-count <count>][private-segment-size <bytes>[m|g]]" - "[preallocate-fifos][preallocate-sessions][client-batch <batch-size>]" - "[uri <sctp://ip/port>][test-bytes][no-output]", - .function = test_sctp_clients_command_fn, - .is_mp_safe = 1, -}; -/* *INDENT-ON* */ - -clib_error_t * -sctp_test_clients_main_init (vlib_main_t * vm) -{ - tclient_main_t *tm = &tclient_main; - tm->is_init = 0; - return 0; -} - -VLIB_INIT_FUNCTION (sctp_test_clients_main_init); - -/* - * fd.io coding-style-patch-verification: ON - * - * Local Variables: - * eval: (c-set-style "gnu") - * End: - */ diff --git a/src/vnet/sctp/builtin_client.h b/src/vnet/sctp/builtin_client.h deleted file mode 100644 index ecf22d8e420..00000000000 --- a/src/vnet/sctp/builtin_client.h +++ /dev/null @@ -1,121 +0,0 @@ -/* - * Copyright (c) 2018 SUSE LLC. - * 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. - */ -#ifndef __included_tclient_h__ -#define __included_tclient_h__ - -#include <vnet/vnet.h> -#include <vnet/ip/ip.h> -#include <vnet/ethernet/ethernet.h> - -#include <vppinfra/hash.h> -#include <vppinfra/error.h> -#include <svm/queue.h> -#include <svm/svm_fifo_segment.h> -#include <vnet/session/session.h> -#include <vnet/session/application_interface.h> - -typedef struct -{ - u64 bytes_to_send; - u64 bytes_sent; - u64 bytes_to_receive; - u64 bytes_received; - - svm_fifo_t *server_rx_fifo; - svm_fifo_t *server_tx_fifo; - - u64 vpp_session_handle; -} session_t; - -typedef struct -{ - /* - * Application setup parameters - */ - svm_queue_t *vl_input_queue; /**< vpe input queue */ - svm_queue_t **vpp_event_queue; - - u32 cli_node_index; /**< cli process node index */ - u32 my_client_index; /**< loopback API client handle */ - u32 app_index; /**< app index after attach */ - - /* - * Configuration params - */ - u8 *connect_uri; /**< URI for slave's connect */ - u64 bytes_to_send; /**< Bytes to send */ - u32 configured_segment_size; - u32 fifo_size; - u32 expected_connections; /**< Number of clients/connections */ - u32 connections_per_batch; /**< Connections to rx/tx at once */ - u32 private_segment_count; /**< Number of private fifo segs */ - u32 private_segment_size; /**< size of private fifo segs */ - - /* - * Test state variables - */ - session_t *sessions; /**< Session pool, shared */ - clib_spinlock_t sessions_lock; - u8 **rx_buf; /**< intermediate rx buffers */ - u8 *connect_test_data; /**< Pre-computed test data */ - u32 **connection_index_by_thread; - u32 **connections_this_batch_by_thread; /**< active connection batch */ - pthread_t client_thread_handle; - - volatile u32 ready_connections; - volatile u32 finished_connections; - volatile u64 rx_total; - volatile u64 tx_total; - volatile int run_test; /**< Signal start of test */ - - f64 test_start_time; - f64 test_end_time; - u32 prev_conns; - u32 repeats; - /* - * Flags - */ - u8 is_init; - u8 test_client_attached; - u8 no_return; - u8 test_return_packets; - int i_am_master; - int drop_packets; /**< drop all packets */ - u8 prealloc_fifos; /**< Request fifo preallocation */ - u8 no_output; - u8 test_bytes; - u8 test_failed; - - /* - * Convenience - */ - vlib_main_t *vlib_main; - vnet_main_t *vnet_main; - ethernet_main_t *ethernet_main; -} tclient_main_t; - -extern tclient_main_t tclient_main; - -vlib_node_registration_t tclient_node; - -#endif /* __included_tclient_h__ */ - -/* - * fd.io coding-style-patch-verification: ON - * - * Local Variables: - * eval: (c-set-style "gnu") - * End: - */ diff --git a/src/vnet/sctp/builtin_server.c b/src/vnet/sctp/builtin_server.c deleted file mode 100644 index e67c4abad9b..00000000000 --- a/src/vnet/sctp/builtin_server.c +++ /dev/null @@ -1,469 +0,0 @@ -/* - * Copyright (c) 2018 SUSE LLC. - * 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 <vnet/vnet.h> -#include <vlibmemory/api.h> -#include <vnet/session/application.h> -#include <vnet/session/application_interface.h> - -typedef struct -{ - /* - * Server app parameters - */ - svm_queue_t **vpp_queue; - svm_queue_t *vl_input_queue; /**< Sever's event queue */ - - u32 app_index; /**< Server app index */ - u32 my_client_index; /**< API client handle */ - u32 node_index; /**< process node index for evnt scheduling */ - - /* - * Config params - */ - u8 no_echo; /**< Don't echo traffic */ - u32 fifo_size; /**< Fifo size */ - u32 rcv_buffer_size; /**< Rcv buffer size */ - u32 prealloc_fifos; /**< Preallocate fifos */ - u32 private_segment_count; /**< Number of private segments */ - u32 private_segment_size; /**< Size of private segments */ - char *server_uri; /**< Server URI */ - - /* - * Test state - */ - u8 **rx_buf; /**< Per-thread RX buffer */ - u64 byte_index; - u32 **rx_retries; - - vlib_main_t *vlib_main; -} builtin_server_main_t; - -builtin_server_main_t builtin_server_main; - -int -builtin_sctp_session_accept_callback (stream_session_t * s) -{ - builtin_server_main_t *bsm = &builtin_server_main; - - bsm->vpp_queue[s->thread_index] = - session_manager_get_vpp_event_queue (s->thread_index); - s->session_state = SESSION_STATE_READY; - bsm->byte_index = 0; - vec_validate (bsm->rx_retries[s->thread_index], s->session_index); - bsm->rx_retries[s->thread_index][s->session_index] = 0; - return 0; -} - -void -builtin_sctp_session_disconnect_callback (stream_session_t * s) -{ - builtin_server_main_t *bsm = &builtin_server_main; - vnet_disconnect_args_t _a, *a = &_a; - - a->handle = session_handle (s); - a->app_index = bsm->app_index; - vnet_disconnect_session (a); -} - -void -builtin_sctp_session_reset_callback (stream_session_t * s) -{ - clib_warning ("Reset session %U", format_stream_session, s, 2); - stream_session_cleanup (s); -} - - -int -builtin_sctp_session_connected_callback (u32 app_index, u32 api_context, - stream_session_t * s, u8 is_fail) -{ - clib_warning ("called..."); - return -1; -} - -int -builtin_sctp_add_segment_callback (u32 client_index, - const ssvm_private_t * fs) -{ - clib_warning ("called..."); - return -1; -} - -int -builtin_sctp_redirect_connect_callback (u32 client_index, void *mp) -{ - clib_warning ("called..."); - return -1; -} - -void -test_bytes_sctp (builtin_server_main_t * bsm, int actual_transfer) -{ - int i; - u32 my_thread_id = vlib_get_thread_index (); - - for (i = 0; i < actual_transfer; i++) - { - if (bsm->rx_buf[my_thread_id][i] != ((bsm->byte_index + i) & 0xff)) - { - clib_warning ("at %lld expected %d got %d", bsm->byte_index + i, - (bsm->byte_index + i) & 0xff, - bsm->rx_buf[my_thread_id][i]); - } - } - bsm->byte_index += actual_transfer; -} - -/* - * If no-echo, just read the data and be done with it - */ -int -builtin_sctp_server_rx_callback_no_echo (stream_session_t * s) -{ - builtin_server_main_t *bsm = &builtin_server_main; - u32 my_thread_id = vlib_get_thread_index (); - int actual_transfer; - svm_fifo_t *rx_fifo; - - rx_fifo = s->server_rx_fifo; - - do - { - actual_transfer = - svm_fifo_dequeue_nowait (rx_fifo, bsm->rcv_buffer_size, - bsm->rx_buf[my_thread_id]); - } - while (actual_transfer > 0); - return 0; -} - -int -builtin_sctp_server_rx_callback (stream_session_t * s) -{ - u32 n_written, max_dequeue, max_enqueue, max_transfer; - int actual_transfer; - svm_fifo_t *tx_fifo, *rx_fifo; - builtin_server_main_t *bsm = &builtin_server_main; - session_fifo_event_t evt; - u32 thread_index = vlib_get_thread_index (); - - ASSERT (s->thread_index == thread_index); - - rx_fifo = s->server_rx_fifo; - tx_fifo = s->server_tx_fifo; - - ASSERT (rx_fifo->master_thread_index == thread_index); - ASSERT (tx_fifo->master_thread_index == thread_index); - - max_dequeue = svm_fifo_max_dequeue (s->server_rx_fifo); - max_enqueue = svm_fifo_max_enqueue (s->server_tx_fifo); - - if (PREDICT_FALSE (max_dequeue == 0)) - return 0; - - /* Number of bytes we're going to copy */ - max_transfer = (max_dequeue < max_enqueue) ? max_dequeue : max_enqueue; - - /* No space in tx fifo */ - if (PREDICT_FALSE (max_transfer == 0)) - { - /* XXX timeout for session that are stuck */ - - rx_event: - /* Program self-tap to retry */ - if (svm_fifo_set_event (rx_fifo)) - { - svm_queue_t *q; - evt.fifo = rx_fifo; - evt.event_type = FIFO_EVENT_BUILTIN_RX; - - q = bsm->vpp_queue[thread_index]; - if (PREDICT_FALSE (q->cursize == q->maxsize)) - clib_warning ("out of event queue space"); - else if (svm_queue_add (q, (u8 *) & evt, 0)) - clib_warning ("failed to enqueue self-tap"); - - if (bsm->rx_retries[thread_index][s->session_index] == 500000) - { - clib_warning ("session stuck: %U", format_stream_session, s, 2); - } - if (bsm->rx_retries[thread_index][s->session_index] < 500001) - bsm->rx_retries[thread_index][s->session_index]++; - } - - return 0; - } - - _vec_len (bsm->rx_buf[thread_index]) = max_transfer; - - actual_transfer = svm_fifo_dequeue_nowait (rx_fifo, max_transfer, - bsm->rx_buf[thread_index]); - ASSERT (actual_transfer == max_transfer); - -// test_bytes (bsm, actual_transfer); - - /* - * Echo back - */ - - n_written = svm_fifo_enqueue_nowait (tx_fifo, actual_transfer, - bsm->rx_buf[thread_index]); - - if (n_written != max_transfer) - clib_warning ("short trout!"); - - if (svm_fifo_set_event (tx_fifo)) - { - /* Fabricate TX event, send to vpp */ - evt.fifo = tx_fifo; - evt.event_type = FIFO_EVENT_APP_TX; - - if (svm_queue_add (bsm->vpp_queue[s->thread_index], - (u8 *) & evt, 0 /* do wait for mutex */ )) - clib_warning ("failed to enqueue tx evt"); - } - - if (PREDICT_FALSE (n_written < max_dequeue)) - goto rx_event; - - return 0; -} - -static session_cb_vft_t builtin_session_cb_vft = { - .session_accept_callback = builtin_sctp_session_accept_callback, - .session_disconnect_callback = builtin_sctp_session_disconnect_callback, - .session_connected_callback = builtin_sctp_session_connected_callback, - .add_segment_callback = builtin_sctp_add_segment_callback, - .redirect_connect_callback = builtin_sctp_redirect_connect_callback, - .builtin_server_rx_callback = builtin_sctp_server_rx_callback, - .session_reset_callback = builtin_sctp_session_reset_callback -}; - -/* Abuse VPP's input queue */ -static int -create_api_loopback (vlib_main_t * vm) -{ - builtin_server_main_t *bsm = &builtin_server_main; - api_main_t *am = &api_main; - vl_shmem_hdr_t *shmem_hdr; - - shmem_hdr = am->shmem_hdr; - bsm->vl_input_queue = shmem_hdr->vl_input_queue; - bsm->my_client_index = - vl_api_memclnt_create_internal ("sctp_test_server", bsm->vl_input_queue); - return 0; -} - -static int -server_attach (u8 * appns_id, u64 appns_flags, u64 appns_secret) -{ - builtin_server_main_t *bsm = &builtin_server_main; - u64 options[APP_OPTIONS_N_OPTIONS]; - vnet_app_attach_args_t _a, *a = &_a; - u32 segment_size = 512 << 20; - - memset (a, 0, sizeof (*a)); - memset (options, 0, sizeof (options)); - - if (bsm->no_echo) - builtin_session_cb_vft.builtin_server_rx_callback = - builtin_sctp_server_rx_callback_no_echo; - else - builtin_session_cb_vft.builtin_server_rx_callback = - builtin_sctp_server_rx_callback; - - if (bsm->private_segment_size) - segment_size = bsm->private_segment_size; - - a->api_client_index = bsm->my_client_index; - a->session_cb_vft = &builtin_session_cb_vft; - a->options = options; - a->options[APP_OPTIONS_SEGMENT_SIZE] = segment_size; - a->options[APP_OPTIONS_RX_FIFO_SIZE] = bsm->fifo_size; - a->options[APP_OPTIONS_TX_FIFO_SIZE] = bsm->fifo_size; - a->options[APP_OPTIONS_PRIVATE_SEGMENT_COUNT] = bsm->private_segment_count; - a->options[APP_OPTIONS_PREALLOC_FIFO_PAIRS] = - bsm->prealloc_fifos ? bsm->prealloc_fifos : 1; - - a->options[APP_OPTIONS_FLAGS] = APP_OPTIONS_FLAGS_IS_BUILTIN; - if (appns_id) - { - a->namespace_id = appns_id; - a->options[APP_OPTIONS_FLAGS] |= appns_flags; - a->options[APP_OPTIONS_NAMESPACE_SECRET] = appns_secret; - } - - if (vnet_application_attach (a)) - { - clib_warning ("failed to attach server"); - return -1; - } - bsm->app_index = a->app_index; - return 0; -} - -static int -server_listen () -{ - builtin_server_main_t *bsm = &builtin_server_main; - vnet_bind_args_t _a, *a = &_a; - memset (a, 0, sizeof (*a)); - a->app_index = bsm->app_index; - a->uri = bsm->server_uri; - return vnet_bind_uri (a); -} - -static int -server_create (vlib_main_t * vm, u8 * appns_id, u64 appns_flags, - u64 appns_secret) -{ - builtin_server_main_t *bsm = &builtin_server_main; - vlib_thread_main_t *vtm = vlib_get_thread_main (); - u32 num_threads; - int i; - - if (bsm->my_client_index == (u32) ~ 0) - { - if (create_api_loopback (vm)) - { - clib_warning ("failed to create api loopback"); - return -1; - } - } - - num_threads = 1 /* main thread */ + vtm->n_threads; - vec_validate (builtin_server_main.vpp_queue, num_threads - 1); - vec_validate (bsm->rx_buf, num_threads - 1); - vec_validate (bsm->rx_retries, num_threads - 1); - - for (i = 0; i < num_threads; i++) - vec_validate (bsm->rx_buf[i], bsm->rcv_buffer_size); - - if (server_attach (appns_id, appns_flags, appns_secret)) - { - clib_warning ("failed to attach server"); - return -1; - } - if (server_listen ()) - { - clib_warning ("failed to start listening"); - return -1; - } - return 0; -} - -static clib_error_t * -server_create_command_fn (vlib_main_t * vm, unformat_input_t * input, - vlib_cli_command_t * cmd) -{ - builtin_server_main_t *bsm = &builtin_server_main; - u8 server_uri_set = 0, *appns_id = 0; - u64 tmp, appns_flags = 0, appns_secret = 0; - int rv; - - bsm->no_echo = 0; - bsm->fifo_size = 64 << 10; - bsm->rcv_buffer_size = 128 << 10; - bsm->prealloc_fifos = 0; - bsm->private_segment_count = 0; - bsm->private_segment_size = 0; - vec_free (bsm->server_uri); - - while (unformat_check_input (input) != UNFORMAT_END_OF_INPUT) - { - if (unformat (input, "no-echo")) - bsm->no_echo = 1; - else if (unformat (input, "fifo-size %d", &bsm->fifo_size)) - bsm->fifo_size <<= 10; - else if (unformat (input, "rcv-buf-size %d", &bsm->rcv_buffer_size)) - ; - else if (unformat (input, "prealloc-fifos %d", &bsm->prealloc_fifos)) - ; - else if (unformat (input, "private-segment-count %d", - &bsm->private_segment_count)) - ; - else if (unformat (input, "private-segment-size %U", - unformat_memory_size, &tmp)) - { - if (tmp >= 0x100000000ULL) - return clib_error_return - (0, "private segment size %lld (%llu) too large", tmp, tmp); - bsm->private_segment_size = tmp; - } - else if (unformat (input, "uri %s", &bsm->server_uri)) - server_uri_set = 1; - else if (unformat (input, "appns %_%v%_", &appns_id)) - ; - else if (unformat (input, "all-scope")) - appns_flags |= (APP_OPTIONS_FLAGS_USE_GLOBAL_SCOPE - | APP_OPTIONS_FLAGS_USE_LOCAL_SCOPE); - else if (unformat (input, "local-scope")) - appns_flags |= APP_OPTIONS_FLAGS_USE_LOCAL_SCOPE; - else if (unformat (input, "global-scope")) - appns_flags |= APP_OPTIONS_FLAGS_USE_GLOBAL_SCOPE; - else if (unformat (input, "secret %lu", &appns_secret)) - ; - else - return clib_error_return (0, "unknown input `%U'", - format_unformat_error, input); - } - - vnet_session_enable_disable (vm, 1 /* turn on SCTP, etc. */ ); - - if (!server_uri_set) - bsm->server_uri = (char *) format (0, "sctp://0.0.0.0/1234%c", 0); - - rv = server_create (vm, appns_id, appns_flags, appns_secret); - vec_free (appns_id); - switch (rv) - { - case 0: - break; - default: - return clib_error_return (0, "server_create returned %d", rv); - } - - return 0; -} - -/* *INDENT-OFF* */ -VLIB_CLI_COMMAND (server_create_command, static) = -{ - .path = "test sctp server", - .short_help = "test sctp server [no echo][fifo-size <mbytes>] " - "[rcv-buf-size <bytes>][prealloc-fifos <count>]" - "[private-segment-count <count>][private-segment-size <bytes[m|g]>]" - "[uri <sctp://ip/port>]", - .function = server_create_command_fn, -}; -/* *INDENT-ON* */ - -clib_error_t * -builtin_sctp_server_main_init (vlib_main_t * vm) -{ - builtin_server_main_t *bsm = &builtin_server_main; - bsm->my_client_index = ~0; - return 0; -} - -VLIB_INIT_FUNCTION (builtin_sctp_server_main_init); - -/* -* fd.io coding-style-patch-verification: ON -* -* Local Variables: -* eval: (c-set-style "gnu") -* End: -*/ |