diff options
Diffstat (limited to 'src/plugins/hs_apps')
-rw-r--r-- | src/plugins/hs_apps/sapi/vpp_echo_common.c | 3 | ||||
-rw-r--r-- | src/plugins/hs_apps/vcl/vcl_test.h | 3 | ||||
-rw-r--r-- | src/plugins/hs_apps/vcl/vcl_test_protos.c | 284 |
3 files changed, 289 insertions, 1 deletions
diff --git a/src/plugins/hs_apps/sapi/vpp_echo_common.c b/src/plugins/hs_apps/sapi/vpp_echo_common.c index 02ce1686ea2..5ce04d1b75b 100644 --- a/src/plugins/hs_apps/sapi/vpp_echo_common.c +++ b/src/plugins/hs_apps/sapi/vpp_echo_common.c @@ -342,6 +342,9 @@ format_transport_proto (u8 * s, va_list * args) case TRANSPORT_PROTO_DTLS: s = format (s, "DTLS"); break; + case TRANSPORT_PROTO_SRTP: + s = format (s, "SRTP"); + break; default: s = format (s, "UNKNOWN"); break; diff --git a/src/plugins/hs_apps/vcl/vcl_test.h b/src/plugins/hs_apps/vcl/vcl_test.h index c22eccc629b..4f67e03f72b 100644 --- a/src/plugins/hs_apps/vcl/vcl_test.h +++ b/src/plugins/hs_apps/vcl/vcl_test.h @@ -141,6 +141,7 @@ typedef struct vcl_test_session vppcom_endpt_t endpt; uint8_t ip[16]; vppcom_data_segment_t ds[2]; + void *opaque; } vcl_test_session_t; static __thread int __wrk_index = 0; @@ -169,7 +170,7 @@ typedef struct typedef struct { - const vcl_test_proto_vft_t *protos[VPPCOM_PROTO_DTLS + 1]; + const vcl_test_proto_vft_t *protos[VPPCOM_PROTO_SRTP + 1]; uint32_t ckpair_index; vcl_test_cfg_t cfg; vcl_test_wrk_t *wrk; diff --git a/src/plugins/hs_apps/vcl/vcl_test_protos.c b/src/plugins/hs_apps/vcl/vcl_test_protos.c index 3c44093156a..60ee09265c9 100644 --- a/src/plugins/hs_apps/vcl/vcl_test_protos.c +++ b/src/plugins/hs_apps/vcl/vcl_test_protos.c @@ -677,6 +677,290 @@ static const vcl_test_proto_vft_t vcl_test_quic = { VCL_TEST_REGISTER_PROTO (VPPCOM_PROTO_QUIC, vcl_test_quic); +static unsigned char test_key[46] = { + 0xe1, 0xf9, 0x7a, 0x0d, 0x3e, 0x01, 0x8b, 0xe0, 0xd6, 0x4f, 0xa3, 0x2c, + 0x06, 0xde, 0x41, 0x39, 0x0e, 0xc6, 0x75, 0xad, 0x49, 0x8a, 0xfe, 0xeb, + 0xb6, 0x96, 0x0b, 0x3a, 0xab, 0xe6, 0xc1, 0x73, 0xc3, 0x17, 0xf2, 0xda, + 0xbe, 0x35, 0x77, 0x93, 0xb6, 0x96, 0x0b, 0x3a, 0xab, 0xe6 +}; + +typedef struct +{ + unsigned char cc : 4; + unsigned char x : 1; + unsigned char p : 1; + unsigned char version : 2; + unsigned char pt : 7; + unsigned char m : 1; + uint16_t seq; + uint32_t ts; + uint32_t ssrc; +} rtp_hdr_t; + +typedef struct +{ + rtp_hdr_t tx_hdr; + rtp_hdr_t rx_hdr; +} rtp_headers_t; + +typedef struct transport_endpt_cfg_srtp_policy +{ + uint32_t ssrc_type; + uint32_t ssrc_value; + uint32_t window_size; + uint8_t allow_repeat_tx; + uint8_t key_len; + uint8_t key[46]; +} transport_endpt_cfg_srtp_policy_t; + +typedef struct transport_endpt_cfg_srtp +{ + transport_endpt_cfg_srtp_policy_t policy[2]; +} transport_endpt_cfg_srtp_t; + +static void +vt_session_add_srtp_policy (vcl_test_session_t *ts, int is_connect) +{ + transport_endpt_cfg_srtp_t *srtp_cfg; + transport_endpt_cfg_srtp_policy_t *test_policy; + uint32_t rx_ssrc, tx_ssrc; + uint32_t cfg_size; + + rx_ssrc = is_connect ? 0xcafebeef : 0xbeefcafe; + tx_ssrc = is_connect ? 0xbeefcafe : 0xcafebeef; + + cfg_size = sizeof (transport_endpt_cfg_srtp_t); + srtp_cfg = malloc (cfg_size); + memset (srtp_cfg, 0, cfg_size); + + test_policy = &srtp_cfg->policy[0]; + test_policy->ssrc_type = 1 /* ssrc_specific */; + test_policy->ssrc_value = rx_ssrc; + memcpy (test_policy->key, test_key, sizeof (test_key)); + test_policy->key_len = sizeof (test_key); + test_policy->window_size = 128; + test_policy->allow_repeat_tx = 1; + + test_policy = &srtp_cfg->policy[1]; + test_policy->ssrc_type = 1 /* ssrc_specific */; + test_policy->ssrc_value = tx_ssrc; + memcpy (test_policy->key, test_key, sizeof (test_key)); + test_policy->key_len = sizeof (test_key); + test_policy->window_size = 128; + test_policy->allow_repeat_tx = 1; + + vppcom_session_attr (ts->fd, VPPCOM_ATTR_SET_ENDPT_EXT_CFG, srtp_cfg, + &cfg_size); + free (srtp_cfg); +} + +static void +vt_srtp_session_init (vcl_test_session_t *ts, int is_connect) +{ + uint32_t rx_ssrc, tx_ssrc; + rtp_headers_t *rtp_hdrs; + rtp_hdr_t *hdr; + + rx_ssrc = is_connect ? 0xcafebeef : 0xbeefcafe; + tx_ssrc = is_connect ? 0xbeefcafe : 0xcafebeef; + + rtp_hdrs = malloc (sizeof (rtp_headers_t)); + memset (rtp_hdrs, 0, sizeof (*rtp_hdrs)); + ts->opaque = rtp_hdrs; + + hdr = &rtp_hdrs->rx_hdr; + hdr->version = 2; + hdr->p = 0; + hdr->x = 0; + hdr->cc = 0; + hdr->m = 0; + hdr->pt = 0x1; + hdr->seq = 0; + hdr->ts = 0; + hdr->ssrc = htonl (rx_ssrc); + + hdr = &rtp_hdrs->tx_hdr; + hdr->version = 2; + hdr->p = 0; + hdr->x = 0; + hdr->cc = 0; + hdr->m = 0; + hdr->pt = 0x1; + hdr->seq = 0; + hdr->ts = 0; + hdr->ssrc = htonl (tx_ssrc); +} + +static int +vt_srtp_write (vcl_test_session_t *ts, void *buf, uint32_t nbytes) +{ + int tx_bytes = 0, nbytes_left = nbytes, rv; + vcl_test_stats_t *stats = &ts->stats; + rtp_hdr_t *hdr; + + hdr = &((rtp_headers_t *) ts->opaque)->tx_hdr; + hdr->seq = htons (ntohs (hdr->seq) + 1); + hdr->ts = htonl (ntohl (hdr->ts) + 1); + + memcpy (buf, hdr, sizeof (*hdr)); + + do + { + stats->tx_xacts++; + rv = vppcom_session_write (ts->fd, buf, nbytes_left); + if (rv < 0) + { + if ((rv == VPPCOM_EAGAIN || rv == VPPCOM_EWOULDBLOCK)) + stats->tx_eagain++; + break; + } + tx_bytes += rv; + nbytes_left = nbytes_left - rv; + buf += rv; + stats->tx_incomp++; + } + while (tx_bytes != nbytes); + + if (tx_bytes < 0) + return 0; + + stats->tx_bytes += tx_bytes; + + return (tx_bytes); +} + +static inline int +vt_srtp_read (vcl_test_session_t *ts, void *buf, uint32_t nbytes) +{ + vcl_test_stats_t *stats = &ts->stats; + rtp_hdr_t *hdr; + int rx_bytes; + + stats->rx_xacts++; + rx_bytes = vppcom_session_read (ts->fd, buf, nbytes); + + if (rx_bytes <= 0) + { + if (rx_bytes == VPPCOM_EAGAIN || rx_bytes == VPPCOM_EWOULDBLOCK) + stats->rx_eagain++; + else + return -1; + } + + if (rx_bytes < nbytes) + stats->rx_incomp++; + + stats->rx_bytes += rx_bytes; + + hdr = &((rtp_headers_t *) ts->opaque)->rx_hdr; + if (((rtp_hdr_t *) buf)->ssrc != hdr->ssrc) + hdr->ssrc = ((rtp_hdr_t *) buf)->ssrc; + return (rx_bytes); +} + +static int +vt_srtp_connect (vcl_test_session_t *ts, vppcom_endpt_t *endpt) +{ + uint32_t flags, flen; + int rv; + + ts->fd = vppcom_session_create (VPPCOM_PROTO_SRTP, 0 /* is_nonblocking */); + if (ts->fd < 0) + { + vterr ("vppcom_session_create()", ts->fd); + return ts->fd; + } + + vt_session_add_srtp_policy (ts, 1 /* is connect */); + + /* Connect is blocking */ + rv = vppcom_session_connect (ts->fd, endpt); + if (rv < 0) + { + vterr ("vppcom_session_connect()", rv); + return rv; + } + + ts->read = vt_srtp_read; + ts->write = vt_srtp_write; + flags = O_NONBLOCK; + flen = sizeof (flags); + vppcom_session_attr (ts->fd, VPPCOM_ATTR_SET_FLAGS, &flags, &flen); + vtinf ("Test session %d (fd %d) connected.", ts->session_index, ts->fd); + + vt_srtp_session_init (ts, 1 /* is connect */); + + return 0; +} + +static int +vt_srtp_listen (vcl_test_session_t *ts, vppcom_endpt_t *endpt) +{ + int rv; + + ts->fd = vppcom_session_create (VPPCOM_PROTO_SRTP, 1 /* is_nonblocking */); + if (ts->fd < 0) + { + vterr ("vppcom_session_create()", ts->fd); + return ts->fd; + } + + vt_session_add_srtp_policy (ts, 0 /* is connect */); + + rv = vppcom_session_bind (ts->fd, endpt); + if (rv < 0) + { + vterr ("vppcom_session_bind()", rv); + return rv; + } + + rv = vppcom_session_listen (ts->fd, 10); + if (rv < 0) + { + vterr ("vppcom_session_listen()", rv); + return rv; + } + + return 0; +} + +static int +vt_srtp_accept (int listen_fd, vcl_test_session_t *ts) +{ + int client_fd; + + client_fd = vppcom_session_accept (listen_fd, &ts->endpt, 0); + if (client_fd < 0) + { + vterr ("vppcom_session_accept()", client_fd); + return client_fd; + } + ts->fd = client_fd; + ts->is_open = 1; + ts->read = vt_srtp_read; + ts->write = vt_srtp_write; + + vt_srtp_session_init (ts, 0 /* is connect */); + + return 0; +} + +static int +vt_srtp_close (vcl_test_session_t *ts) +{ + free (ts->opaque); + return 0; +} + +static const vcl_test_proto_vft_t vcl_test_srtp = { + .open = vt_srtp_connect, + .listen = vt_srtp_listen, + .accept = vt_srtp_accept, + .close = vt_srtp_close, +}; + +VCL_TEST_REGISTER_PROTO (VPPCOM_PROTO_SRTP, vcl_test_srtp); + /* * fd.io coding-style-patch-verification: ON * |