hsa: Refactor quic_echo to allow other protocols
Type: refactor Change-Id: Iaef9091e1d057110530255e644fad1c298418966 Signed-off-by: Nathan Skrzypczak <nathan.skrzypczak@gmail.com>
This commit is contained in:

committed by
Florin Coras

parent
d25d364d2b
commit
ecd1fc7dfa
@ -34,8 +34,12 @@ if(VPP_BUILD_HS_SAPI_APPS)
|
||||
DEPENDS api_headers
|
||||
NO_INSTALL
|
||||
)
|
||||
add_vpp_executable(quic_echo
|
||||
SOURCES sapi/quic_echo.c
|
||||
add_vpp_executable(vpp_echo
|
||||
SOURCES
|
||||
sapi/vpp_echo.c
|
||||
sapi/vpp_echo_common.c
|
||||
sapi/vpp_echo_bapi.c
|
||||
sapi/vpp_echo_proto_quic.c
|
||||
LINK_LIBRARIES vlibmemoryclient svm vppinfra pthread m rt
|
||||
DEPENDS api_headers
|
||||
NO_INSTALL
|
||||
|
@ -1,158 +0,0 @@
|
||||
/*
|
||||
* 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.
|
||||
*/
|
||||
|
||||
#include <stdio.h>
|
||||
#include <signal.h>
|
||||
|
||||
#include <vnet/session/application_interface.h>
|
||||
|
||||
typedef struct
|
||||
{
|
||||
/* VNET_API_ERROR_FOO -> "Foo" hash table */
|
||||
uword *error_string_by_error_number;
|
||||
} echo_common_main_t;
|
||||
|
||||
echo_common_main_t echo_common_main;
|
||||
|
||||
u8 *
|
||||
format_ip4_address (u8 * s, va_list * args)
|
||||
{
|
||||
u8 *a = va_arg (*args, u8 *);
|
||||
return format (s, "%d.%d.%d.%d", a[0], a[1], a[2], a[3]);
|
||||
}
|
||||
|
||||
u8 *
|
||||
format_ip6_address (u8 * s, va_list * args)
|
||||
{
|
||||
ip6_address_t *a = va_arg (*args, ip6_address_t *);
|
||||
u32 i, i_max_n_zero, max_n_zeros, i_first_zero, n_zeros, last_double_colon;
|
||||
|
||||
i_max_n_zero = ARRAY_LEN (a->as_u16);
|
||||
max_n_zeros = 0;
|
||||
i_first_zero = i_max_n_zero;
|
||||
n_zeros = 0;
|
||||
for (i = 0; i < ARRAY_LEN (a->as_u16); i++)
|
||||
{
|
||||
u32 is_zero = a->as_u16[i] == 0;
|
||||
if (is_zero && i_first_zero >= ARRAY_LEN (a->as_u16))
|
||||
{
|
||||
i_first_zero = i;
|
||||
n_zeros = 0;
|
||||
}
|
||||
n_zeros += is_zero;
|
||||
if ((!is_zero && n_zeros > max_n_zeros)
|
||||
|| (i + 1 >= ARRAY_LEN (a->as_u16) && n_zeros > max_n_zeros))
|
||||
{
|
||||
i_max_n_zero = i_first_zero;
|
||||
max_n_zeros = n_zeros;
|
||||
i_first_zero = ARRAY_LEN (a->as_u16);
|
||||
n_zeros = 0;
|
||||
}
|
||||
}
|
||||
|
||||
last_double_colon = 0;
|
||||
for (i = 0; i < ARRAY_LEN (a->as_u16); i++)
|
||||
{
|
||||
if (i == i_max_n_zero && max_n_zeros > 1)
|
||||
{
|
||||
s = format (s, "::");
|
||||
i += max_n_zeros - 1;
|
||||
last_double_colon = 1;
|
||||
}
|
||||
else
|
||||
{
|
||||
s = format (s, "%s%x",
|
||||
(last_double_colon || i == 0) ? "" : ":",
|
||||
clib_net_to_host_u16 (a->as_u16[i]));
|
||||
last_double_colon = 0;
|
||||
}
|
||||
}
|
||||
|
||||
return s;
|
||||
}
|
||||
|
||||
/* Format an IP46 address. */
|
||||
u8 *
|
||||
format_ip46_address (u8 * s, va_list * args)
|
||||
{
|
||||
ip46_address_t *ip46 = va_arg (*args, ip46_address_t *);
|
||||
ip46_type_t type = va_arg (*args, ip46_type_t);
|
||||
int is_ip4 = 1;
|
||||
|
||||
switch (type)
|
||||
{
|
||||
case IP46_TYPE_ANY:
|
||||
is_ip4 = ip46_address_is_ip4 (ip46);
|
||||
break;
|
||||
case IP46_TYPE_IP4:
|
||||
is_ip4 = 1;
|
||||
break;
|
||||
case IP46_TYPE_IP6:
|
||||
is_ip4 = 0;
|
||||
break;
|
||||
}
|
||||
|
||||
return is_ip4 ?
|
||||
format (s, "%U", format_ip4_address, &ip46->ip4) :
|
||||
format (s, "%U", format_ip6_address, &ip46->ip6);
|
||||
}
|
||||
|
||||
static uword
|
||||
unformat_data (unformat_input_t * input, va_list * args)
|
||||
{
|
||||
u64 _a;
|
||||
u64 *a = va_arg (*args, u64 *);
|
||||
if (unformat (input, "%lluGb", &_a))
|
||||
*a = _a << 30;
|
||||
else if (unformat (input, "%lluMb", &_a))
|
||||
*a = _a << 20;
|
||||
else if (unformat (input, "%lluKb", &_a))
|
||||
*a = _a << 10;
|
||||
else if (unformat (input, "%llu", a))
|
||||
;
|
||||
else
|
||||
return 0;
|
||||
return 1;
|
||||
}
|
||||
|
||||
static u8 *
|
||||
format_api_error (u8 * s, va_list * args)
|
||||
{
|
||||
echo_common_main_t *ecm = &echo_common_main;
|
||||
i32 error = va_arg (*args, u32);
|
||||
uword *p;
|
||||
|
||||
p = hash_get (ecm->error_string_by_error_number, -error);
|
||||
|
||||
if (p)
|
||||
s = format (s, "%s", p[0]);
|
||||
else
|
||||
s = format (s, "%d", error);
|
||||
return s;
|
||||
}
|
||||
|
||||
static void
|
||||
init_error_string_table ()
|
||||
{
|
||||
echo_common_main_t *ecm = &echo_common_main;
|
||||
ecm->error_string_by_error_number = hash_create (0, sizeof (uword));
|
||||
|
||||
#define _(n,v,s) hash_set (ecm->error_string_by_error_number, -v, s);
|
||||
foreach_vnet_api_error;
|
||||
#undef _
|
||||
|
||||
hash_set (ecm->error_string_by_error_number, 99, "Misc");
|
||||
}
|
||||
|
File diff suppressed because it is too large
Load Diff
@ -1,246 +0,0 @@
|
||||
/*
|
||||
* 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.
|
||||
*/
|
||||
|
||||
#include <stdio.h>
|
||||
#include <signal.h>
|
||||
|
||||
#include <vnet/session/application_interface.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 NITER 4000000
|
||||
#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); \
|
||||
}
|
||||
|
||||
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 */
|
||||
} 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
|
||||
{
|
||||
QUIC_SESSION_TYPE_QUIC,
|
||||
QUIC_SESSION_TYPE_STREAM,
|
||||
QUIC_SESSION_TYPE_LISTEN,
|
||||
};
|
||||
|
||||
enum quic_session_state_t
|
||||
{
|
||||
QUIC_SESSION_STATE_INITIAL,
|
||||
QUIC_SESSION_STATE_READY,
|
||||
QUIC_SESSION_STATE_AWAIT_CLOSING, /* Data transfer is done, wait for close evt */
|
||||
QUIC_SESSION_STATE_AWAIT_DATA, /* Peer closed, wait for outstanding data */
|
||||
QUIC_SESSION_STATE_CLOSING, /* told vpp to close */
|
||||
QUIC_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 struct _quic_echo_cb_vft
|
||||
{
|
||||
void (*quic_connected_cb) (session_connected_msg_t * mp, u32 session_index);
|
||||
void (*client_stream_connected_cb) (session_connected_msg_t * mp,
|
||||
u32 session_index);
|
||||
void (*server_stream_connected_cb) (session_connected_msg_t * mp,
|
||||
u32 session_index);
|
||||
void (*quic_accepted_cb) (session_accepted_msg_t * mp, u32 session_index);
|
||||
void (*client_stream_accepted_cb) (session_accepted_msg_t * mp,
|
||||
u32 session_index);
|
||||
void (*server_stream_accepted_cb) (session_accepted_msg_t * mp,
|
||||
u32 session_index);
|
||||
} quic_echo_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 */
|
||||
quic_echo_cb_vft_t cb_vft; /* cb vft for QUIC scenarios */
|
||||
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_quic_disconnects; /* actively send disconnect */
|
||||
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 *data_thread_args;
|
||||
|
||||
u32 n_clients; /* Target number of QUIC sessions */
|
||||
u32 n_stream_clients; /* Target Number of STREAM sessions per QUIC session */
|
||||
volatile u32 n_quic_clients_connected; /* Number of connected QUIC sessions */
|
||||
volatile u32 n_clients_connected; /* Number of STREAM sessions connected */
|
||||
u32 n_rx_threads; /* Number of data threads */
|
||||
volatile u32 nxt_available_sidx; /* next unused prealloced session_index */
|
||||
|
||||
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;
|
||||
} echo_main_t;
|
||||
|
||||
typedef void (*echo_rpc_t) (void *arg, u32 opaque);
|
||||
|
||||
typedef struct
|
||||
{
|
||||
void *fp;
|
||||
void *arg;
|
||||
u32 opaque;
|
||||
} echo_rpc_msg_t;
|
1049
src/plugins/hs_apps/sapi/vpp_echo.c
Normal file
1049
src/plugins/hs_apps/sapi/vpp_echo.c
Normal file
File diff suppressed because it is too large
Load Diff
439
src/plugins/hs_apps/sapi/vpp_echo_bapi.c
Normal file
439
src/plugins/hs_apps/sapi/vpp_echo_bapi.c
Normal file
File diff suppressed because it is too large
Load Diff
562
src/plugins/hs_apps/sapi/vpp_echo_common.c
Normal file
562
src/plugins/hs_apps/sapi/vpp_echo_common.c
Normal file
File diff suppressed because it is too large
Load Diff
321
src/plugins/hs_apps/sapi/vpp_echo_common.h
Normal file
321
src/plugins/hs_apps/sapi/vpp_echo_common.h
Normal file
File diff suppressed because it is too large
Load Diff
507
src/plugins/hs_apps/sapi/vpp_echo_proto_quic.c
Normal file
507
src/plugins/hs_apps/sapi/vpp_echo_proto_quic.c
Normal file
File diff suppressed because it is too large
Load Diff
@ -173,7 +173,7 @@ class QUICEchoExternalTestCase(QUICTestCase):
|
||||
_args = self.server_echo_test_args + list(args)
|
||||
self.worker_server = QUICAppWorker(
|
||||
self.build_dir,
|
||||
"quic_echo",
|
||||
"vpp_echo",
|
||||
_args,
|
||||
self.logger)
|
||||
self.worker_server.start()
|
||||
@ -184,7 +184,7 @@ class QUICEchoExternalTestCase(QUICTestCase):
|
||||
# self.client_echo_test_args += "use-svm-api"
|
||||
self.worker_client = QUICAppWorker(
|
||||
self.build_dir,
|
||||
"quic_echo",
|
||||
"vpp_echo",
|
||||
_args,
|
||||
self.logger)
|
||||
self.worker_client.start()
|
||||
|
Reference in New Issue
Block a user