diff options
author | Nathan Skrzypczak <nathan.skrzypczak@gmail.com> | 2019-08-05 13:43:31 +0200 |
---|---|---|
committer | Florin Coras <florin.coras@gmail.com> | 2019-08-16 14:47:38 +0000 |
commit | ecd1fc7dfa6a36d1774f71093380b3548a22346b (patch) | |
tree | 6ed2568a9950e955c4826941216af153c5b08b2f /src/plugins/hs_apps/sapi/vpp_echo_common.h | |
parent | d25d364d2bc146d3e17ab7c4d558bcb0ce138b89 (diff) |
hsa: Refactor quic_echo to allow other protocols
Type: refactor
Change-Id: Iaef9091e1d057110530255e644fad1c298418966
Signed-off-by: Nathan Skrzypczak <nathan.skrzypczak@gmail.com>
Diffstat (limited to 'src/plugins/hs_apps/sapi/vpp_echo_common.h')
-rw-r--r-- | src/plugins/hs_apps/sapi/vpp_echo_common.h | 321 |
1 files changed, 321 insertions, 0 deletions
diff --git a/src/plugins/hs_apps/sapi/vpp_echo_common.h b/src/plugins/hs_apps/sapi/vpp_echo_common.h new file mode 100644 index 00000000000..2f9d3912bb7 --- /dev/null +++ b/src/plugins/hs_apps/sapi/vpp_echo_common.h @@ -0,0 +1,321 @@ +/* + * Copyright (c) 2019 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. + */ + +#ifndef __included_vpp_echo_common_h__ +#define __included_vpp_echo_common_h__ + +#include <vnet/session/application_interface.h> +#include <vpp/api/vpe_msg_enum.h> + +#define vl_typedefs /* define message structures */ +#include <vpp/api/vpe_all_api_h.h> +#undef vl_typedefs + +/* declare message handlers for each api */ + +#define vl_endianfun /* define message structures */ +#include <vpp/api/vpe_all_api_h.h> +#undef vl_endianfun + +/* instantiate all the print functions we know about */ +#define vl_print(handle, ...) +#define vl_printfun +#include <vpp/api/vpe_all_api_h.h> +#undef vl_printfun + +#define TIMEOUT 10.0 + +#define CHECK(expected, result, _fmt, _args...) \ + if (expected != result) \ + ECHO_FAIL ("expected %d, got %d : " _fmt, expected, result, ##_args); + +#define ECHO_FAIL(_fmt,_args...) \ + { \ + echo_main_t *em = &echo_main; \ + em->has_failed = 1; \ + em->time_to_stop = 1; \ + if (em->log_lvl > 0) \ + clib_warning ("ECHO-ERROR: "_fmt, ##_args); \ + } + +#define ECHO_LOG(lvl, _fmt,_args...) \ + { \ + echo_main_t *em = &echo_main; \ + if (em->log_lvl > lvl) \ + clib_warning (_fmt, ##_args); \ + } + +#define ECHO_REGISTER_PROTO(proto, vft) \ + static void __clib_constructor \ + vpp_echo_init_##proto () \ + { \ + echo_main_t *em = &echo_main; \ + em->available_proto_cb_vft[proto] = &vft; \ + } + +typedef struct +{ + CLIB_CACHE_LINE_ALIGN_MARK (cacheline0); +#define _(type, name) type name; + foreach_app_session_field +#undef _ + u64 vpp_session_handle; + u64 bytes_sent; + u64 bytes_to_send; + volatile u64 bytes_received; + volatile u64 bytes_to_receive; + f64 start; + u32 listener_index; /* listener index in echo session pool */ + u32 idle_cycles; /* consecutive enq/deq with no data */ + volatile u64 accepted_session_count; /* sessions we accepted (as a listener) */ +} echo_session_t; + +typedef enum +{ + ECHO_NO_DATA_SOURCE, + ECHO_TEST_DATA_SOURCE, + ECHO_RX_DATA_SOURCE, + ECHO_INVALID_DATA_SOURCE +} data_source_t; + +enum echo_close_f_t +{ + ECHO_CLOSE_F_INVALID = 0, + ECHO_CLOSE_F_PASSIVE, /* wait for close msg */ + ECHO_CLOSE_F_ACTIVE, /* send close msg */ + ECHO_CLOSE_F_NONE, /* don't bother sending close msg */ +}; + +enum quic_session_type_t +{ + ECHO_SESSION_TYPE_QUIC, + ECHO_SESSION_TYPE_STREAM, + ECHO_SESSION_TYPE_LISTEN, +}; + +enum quic_session_state_t +{ + ECHO_SESSION_STATE_INITIAL, + ECHO_SESSION_STATE_READY, + ECHO_SESSION_STATE_AWAIT_CLOSING, /* Data transfer is done, wait for close evt */ + ECHO_SESSION_STATE_AWAIT_DATA, /* Peer closed, wait for outstanding data */ + ECHO_SESSION_STATE_CLOSING, /* told vpp to close */ + ECHO_SESSION_STATE_CLOSED, /* closed in vpp */ +}; + +typedef enum +{ + STATE_START, + STATE_ATTACHED, + STATE_LISTEN, + STATE_READY, + STATE_DATA_DONE, + STATE_DISCONNECTED, + STATE_DETACHED +} connection_state_t; + +typedef enum echo_test_evt_ +{ + ECHO_EVT_START = 1, /* app starts */ + ECHO_EVT_FIRST_QCONNECT = (1 << 1), /* First connect Quic session sent */ + ECHO_EVT_LAST_QCONNECTED = (1 << 2), /* All Quic session are connected */ + ECHO_EVT_FIRST_SCONNECT = (1 << 3), /* First connect Stream session sent */ + ECHO_EVT_LAST_SCONNECTED = (1 << 4), /* All Stream session are connected */ + ECHO_EVT_LAST_BYTE = (1 << 5), /* Last byte received */ + ECHO_EVT_EXIT = (1 << 6), /* app exits */ +} echo_test_evt_t; + +typedef union session_connected_bundled_msg_ +{ + session_connected_msg_t *mp; + vl_api_connect_uri_reply_t *bmp; +} session_connected_bundled_msg_t; + +typedef struct echo_proto_cb_vft_ +{ + void (*connected_cb) (session_connected_bundled_msg_t * mp, u32 session_index, u8 is_failed); /* Session is connected */ + void (*accepted_cb) (session_accepted_msg_t * mp, echo_session_t * session); /* Session got accepted */ + void (*bound_uri_cb) (session_bound_msg_t * mp, echo_session_t * session); /* Session got bound */ + void (*reset_cb) (session_reset_msg_t * mp, echo_session_t * s); /* Received RESET on session */ + void (*disconnected_cb) (session_disconnected_msg_t * mp, echo_session_t * s); /* Received DISCONNECT on session */ + void (*disconnected_reply_cb) (echo_session_t * s); /* ACK disconnect we sent to vpp */ + void (*cleanup_cb) (echo_session_t * s, u8 parent_died); /* Session should be cleaned up (parent listener may be dead) */ + /* Add CLI options */ + int (*process_opts_cb) (unformat_input_t * a); + void (*set_defaults_before_opts_cb) (void); + void (*set_defaults_after_opts_cb) (void); + void (*print_usage_cb) (void); +} echo_proto_cb_vft_t; + +typedef enum +{ + RETURN_PACKETS_NOTEST, + RETURN_PACKETS_LOG_WRONG, + RETURN_PACKETS_ASSERT, +} test_return_packets_t; + +typedef struct teardown_stat_ +{ + u32 q; /* quic sessions */ + u32 s; /* stream sessions */ +} teardown_stat_t; + +typedef struct +{ + svm_queue_t *vl_input_queue; /* vpe input queue */ + u32 my_client_index; /* API client handle */ + u8 *uri; /* The URI we're playing with */ + echo_session_t *sessions; /* Session pool */ + svm_msg_q_t *our_event_queue; /* Our event queue */ + clib_time_t clib_time; /* For deadman timers */ + u8 *socket_name; + int i_am_master; + u32 listen_session_index; /* Index of vpp listener session */ + + uword *session_index_by_vpp_handles; /* Hash table : quic_echo s_id -> vpp s_handle */ + clib_spinlock_t sid_vpp_handles_lock; /* Hash table lock */ + + uword *shared_segment_handles; /* Hash table : segment_names -> 1 */ + clib_spinlock_t segment_handles_lock; /* Hash table lock */ + echo_proto_cb_vft_t *proto_cb_vft; + svm_msg_q_t *rpc_msq_queue; /* MQ between quic_echo threads */ + fifo_segment_main_t segment_main; + + /* State of the connection, shared between msg RX thread and main thread */ + volatile connection_state_t state; + volatile u8 time_to_stop; /* Signal variables */ + u8 has_failed; /* stores the exit code */ + + /** Flag that decides if socket, instead of svm, api is used to connect to + * vpp. If sock api is used, shm binary api is subsequently bootstrapped + * and all other messages are exchanged using shm IPC. */ + u8 use_sock_api; + + u8 *connect_test_data; + u8 test_return_packets; + u64 bytes_to_send; /* target per stream */ + u64 bytes_to_receive; /* target per stream */ + u32 fifo_size; + u32 rx_buf_size; + u32 tx_buf_size; + data_source_t data_source; /* Use no/dummy/mirrored data */ + u8 send_stream_disconnects; /* actively send disconnect */ + u8 output_json; /* Output stats as JSON */ + u8 log_lvl; /* Verbosity of the logging */ + int max_test_msg; /* Limit the number of incorrect data messages */ + + u8 *appns_id; + u64 appns_flags; + u64 appns_secret; + + pthread_t *data_thread_handles; /* vec of data thread handles */ + pthread_t mq_thread_handle; /* Message queue thread handle */ + u32 *volatile data_thread_args; + + u32 n_connects; /* Target number of connects to send */ + u32 n_sessions; /* Number of sessions to prealloc */ + u32 n_clients; /* Target number of clients doing RX/TX */ + u32 n_rx_threads; /* Number of data threads */ + + volatile u32 n_clients_connected; /* Number of STREAM sessions connected */ + volatile u32 nxt_available_sidx; /* next unused prealloced session_index */ + + /* VNET_API_ERROR_FOO -> "Foo" hash table */ + uword *error_string_by_error_number; + echo_proto_cb_vft_t *available_proto_cb_vft[TRANSPORT_N_PROTO]; + + struct + { + u64 tx_total; + u64 rx_total; + teardown_stat_t reset_count; /* received reset from vpp */ + teardown_stat_t close_count; /* received close from vpp */ + teardown_stat_t active_count; /* sent close to vpp */ + teardown_stat_t clean_count; /* cleaned up stale session */ + } stats; + + struct /* Event based timing : start & end depend on CLI specified events */ + { + f64 start_time; + f64 end_time; + u8 events_sent; + u8 start_event; + u8 end_event; + } timing; + + struct + { + u32 transport_proto; + ip46_address_t ip; + u32 port; + u8 is_ip4; + } uri_elts; +} echo_main_t; + +extern echo_main_t echo_main; + +typedef void (*echo_rpc_t) (void *arg, u32 opaque); + +typedef struct +{ + void *fp; + void *arg; + u32 opaque; +} echo_rpc_msg_t; + +u8 *format_ip4_address (u8 * s, va_list * args); +u8 *format_ip6_address (u8 * s, va_list * args); +u8 *format_ip46_address (u8 * s, va_list * args); +uword unformat_data (unformat_input_t * input, va_list * args); +u8 *format_api_error (u8 * s, va_list * args); +void init_error_string_table (); +u8 *echo_format_app_state (u8 * s, va_list * args); +uword echo_unformat_close (unformat_input_t * input, va_list * args); +uword echo_unformat_timing_event (unformat_input_t * input, va_list * args); +u8 *echo_format_timing_event (u8 * s, va_list * args); +uword unformat_transport_proto (unformat_input_t * input, va_list * args); +u8 *format_transport_proto (u8 * s, va_list * args); +uword unformat_ip4_address (unformat_input_t * input, va_list * args); +uword unformat_ip6_address (unformat_input_t * input, va_list * args); + +void echo_session_handle_add_del (echo_main_t * em, u64 handle, u32 sid); +echo_session_t *echo_session_new (echo_main_t * em); +int echo_send_rpc (echo_main_t * em, void *fp, void *arg, u32 opaque); +echo_session_t *echo_get_session_from_handle (echo_main_t * em, u64 handle); +int wait_for_segment_allocation (u64 segment_handle); +int wait_for_state_change (echo_main_t * em, connection_state_t state, + f64 timeout); +void echo_notify_event (echo_main_t * em, echo_test_evt_t e); +void echo_session_print_stats (echo_main_t * em, echo_session_t * session); + +/* Binary API */ + +void echo_send_attach (echo_main_t * em); +void echo_send_detach (echo_main_t * em); +void echo_send_listen (echo_main_t * em); +void echo_send_unbind (echo_main_t * em); +void echo_send_connect (u8 * uri, u32 opaque); +void echo_send_disconnect_session (u64 handle, u32 opaque); +void echo_api_hookup (echo_main_t * em); + +#endif /* __included_vpp_echo_common_h__ */ + +/* + * fd.io coding-style-patch-verification: ON + * + * Local Variables: + * eval: (c-set-style "gnu") + * End: + */ |