| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728 |
- /*
- * Copyright (C) 2011-2013 Michael Tuexen
- * Copyright (C) 2018-2020 Felix Weinrank
- *
- * All rights reserved.
- *
- * Redistribution and use in source and binary forms, with or without
- * modification, are permitted provided that the following conditions
- * are met:
- * 1. Redistributions of source code must retain the above copyright
- * notice, this list of conditions and the following disclaimer.
- * 2. Redistributions in binary form must reproduce the above copyright
- * notice, this list of conditions and the following disclaimer in the
- * documentation and/or other materials provided with the distribution.
- * 3. Neither the name of the project nor the names of its contributors
- * may be used to endorse or promote products derived from this software
- * without specific prior written permission.
- *
- * THIS SOFTWARE IS PROVIDED BY THE PROJECT AND CONTRIBUTORS ``AS IS'' AND
- * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
- * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
- * ARE DISCLAIMED. IN NO EVENT SHALL THE PROJECT OR CONTRIBUTORS BE LIABLE
- * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
- * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
- * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
- * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
- * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
- * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
- * SUCH DAMAGE.
- */
- /*
- * Usage: ekr_loop [client_port] [server_port] [crc32c offloading <0/1>]
- */
- #ifdef _WIN32
- #define _CRT_SECURE_NO_WARNINGS
- #endif
- #include <stdio.h>
- #include <stdlib.h>
- #include <string.h>
- #include <stdarg.h>
- #include <sys/types.h>
- #ifndef _WIN32
- #include <sys/socket.h>
- #include <netinet/in.h>
- #include <arpa/inet.h>
- #include <errno.h>
- #include <pthread.h>
- #include <unistd.h>
- #else
- #include <winsock2.h>
- #include <ws2tcpip.h>
- #endif
- #include <usrsctp.h>
- #include "programs_helper.h"
- #define MAX_PACKET_SIZE (1<<16)
- #define LINE_LENGTH (1<<20)
- #define DISCARD_PPID 39
- #define NUMBER_OF_STEPS 10
- static uint8_t crc32c_offloading = 0;
- struct upcall_meta {
- uint8_t sender;
- void* addr;
- };
- #ifdef _WIN32
- static DWORD WINAPI
- #else
- static void *
- #endif
- handle_packets(void *arg)
- {
- #ifdef _WIN32
- SOCKET *fdp;
- #else
- int *fdp;
- #endif
- char *dump_buf;
- struct sctp_common_header *hdr;
- ssize_t length;
- char buf[MAX_PACKET_SIZE];
- uint32_t received_crc32c, computed_crc32c;
- #ifdef _WIN32
- fdp = (SOCKET *)arg;
- #else
- fdp = (int *)arg;
- #endif
- for (;;) {
- #if defined(__NetBSD__)
- pthread_testcancel();
- #endif
- length = recv(*fdp, buf, MAX_PACKET_SIZE, 0);
- if (length > 0) {
- if ((dump_buf = usrsctp_dumppacket(buf, (size_t)length, SCTP_DUMP_INBOUND)) != NULL) {
- //debug_printf_clean("%s\n", dump_buf);
- usrsctp_freedumpbuffer(dump_buf);
- }
- if (crc32c_offloading) {
- if ((size_t)length >= sizeof(struct sctp_common_header)) {
- hdr = (struct sctp_common_header *)buf;
- received_crc32c = hdr->crc32c;
- hdr->crc32c = htonl(0);
- computed_crc32c = usrsctp_crc32c(buf, (size_t)length);
- hdr->crc32c = received_crc32c;
- if (received_crc32c == computed_crc32c) {
- usrsctp_conninput(fdp, buf, (size_t)length, 0);
- } else {
- debug_printf("Wrong CRC32c: expected %08x received %08x\n",
- ntohl(computed_crc32c), ntohl(received_crc32c));
- }
- } else {
- debug_printf("Packet too short: length %zd", length);
- }
- } else {
- usrsctp_conninput(fdp, buf, (size_t)length, 0);
- }
- }
- }
- #ifdef _WIN32
- return 0;
- #else
- return (NULL);
- #endif
- }
- static int
- conn_output(void *addr, void *buf, size_t length, uint8_t tos, uint8_t set_df)
- {
- char *dump_buf;
- struct sctp_common_header *hdr;
- #ifdef _WIN32
- SOCKET *fdp;
- #else
- int *fdp;
- #endif
- #ifdef _WIN32
- fdp = (SOCKET *)addr;
- #else
- fdp = (int *)addr;
- #endif
- if (crc32c_offloading && length >= sizeof(struct sctp_common_header)) {
- hdr = (struct sctp_common_header *)buf;
- hdr->crc32c = usrsctp_crc32c(buf, (size_t)length);
- }
- if ((dump_buf = usrsctp_dumppacket(buf, length, SCTP_DUMP_OUTBOUND)) != NULL) {
- //debug_printf_clean("%s\n", dump_buf);
- usrsctp_freedumpbuffer(dump_buf);
- }
- #ifdef _WIN32
- if (send(*fdp, buf, (int)length, 0) == SOCKET_ERROR) {
- return (WSAGetLastError());
- #else
- if (send(*fdp, buf, length, 0) < 0) {
- return (errno);
- #endif
- } else {
- return (0);
- }
- }
- static void
- handle_upcall(struct socket *sock, void *data, int flgs)
- {
- char *buf;
- int events;
- struct upcall_meta *meta;
- meta = data;
- buf = malloc(MAX_PACKET_SIZE);
- while ((events = usrsctp_get_events(sock)) && (events & SCTP_EVENT_READ)) {
- struct sctp_recvv_rn rn;
- ssize_t n;
- union sctp_sockstore addr;
- int flags = 0;
- socklen_t len = (socklen_t)sizeof(addr);
- unsigned int infotype = 0;
- socklen_t infolen = sizeof(struct sctp_recvv_rn);
- memset(&rn, 0, sizeof(struct sctp_recvv_rn));
- n = usrsctp_recvv(sock, buf, MAX_PACKET_SIZE, (struct sockaddr *) &addr, &len, (void *)&rn, &infolen, &infotype, &flags);
- if (n < 0) {
- perror("usrsctp_recvv");
- break;
- } else if (n > 0) {
- if (flags & MSG_NOTIFICATION) {
- debug_printf("MSG RCV: Notification of length %d received.\n", (int)n);
- } else {
- debug_printf("MSG RCV: Data length %d, addr %p:%u, stream %u, SSN %u, TSN %u, PPID %u, context %u, %s%s.\n",
- (int)n,
- addr.sconn.sconn_addr,
- ntohs(addr.sconn.sconn_port),
- rn.recvv_rcvinfo.rcv_sid,
- rn.recvv_rcvinfo.rcv_ssn,
- rn.recvv_rcvinfo.rcv_tsn,
- ntohl(rn.recvv_rcvinfo.rcv_ppid),
- rn.recvv_rcvinfo.rcv_context,
- (rn.recvv_rcvinfo.rcv_flags & SCTP_UNORDERED) ? "unordered" : "ordered",
- (flags & MSG_EOR) ? ", EOR" : "");
- }
- } else {
- debug_printf("Connection closed!\n");
- usrsctp_deregister_address(meta->addr);
- usrsctp_close(sock);
- break;
- }
- }
- free(buf);
- return;
- }
- static void
- print_addresses(struct socket *sock)
- {
- int i, n;
- struct sockaddr *addrs, *addr;
- #if !defined(HAVE_SA_LEN)
- int sa_len;
- #endif
- debug_printf("Addresses: ");
- n = usrsctp_getladdrs(sock, 0, &addrs);
- addr = addrs;
- for (i = 0; i < n; i++) {
- switch (addr->sa_family) {
- case AF_INET:
- {
- struct sockaddr_in *sin;
- char buf[INET_ADDRSTRLEN];
- const char *name;
- sin = (struct sockaddr_in *)addr;
- name = inet_ntop(AF_INET, &sin->sin_addr, buf, INET_ADDRSTRLEN);
- debug_printf_clean("%s:%d", name, ntohs(sin->sin_port));
- #if !defined(HAVE_SA_LEN)
- sa_len = (int)sizeof(struct sockaddr_in);
- #endif
- break;
- }
- case AF_INET6:
- {
- struct sockaddr_in6 *sin6;
- char buf[INET6_ADDRSTRLEN];
- const char *name;
- sin6 = (struct sockaddr_in6 *)addr;
- name = inet_ntop(AF_INET6, &sin6->sin6_addr, buf, INET6_ADDRSTRLEN);
- debug_printf_clean("%s:%d", name, ntohs(sin6->sin6_port));
- #if !defined(HAVE_SA_LEN)
- sa_len = (int)sizeof(struct sockaddr_in6);
- #endif
- break;
- }
- case AF_CONN:
- {
- struct sockaddr_conn *sconn;
- sconn = (struct sockaddr_conn *)addr;
- debug_printf_clean("%p:%d", sconn->sconn_addr, ntohs(sconn->sconn_port));
- #if !defined(HAVE_SA_LEN)
- sa_len = (int)sizeof(struct sockaddr_conn);
- #endif
- break;
- }
- default:
- debug_printf_clean("Unknown family: %d", addr->sa_family);
- #if !defined(HAVE_SA_LEN)
- sa_len = (int)sizeof(struct sockaddr);
- #endif
- break;
- }
- #if !defined(HAVE_SA_LEN)
- addr = (struct sockaddr *)((char *)addr + sa_len);
- #else
- addr = (struct sockaddr *)((caddr_t)addr + addr->sa_len);
- #endif
- if (i != n - 1) {
- debug_printf_clean(",");
- }
- }
- if (n > 0) {
- usrsctp_freeladdrs(addrs);
- }
- debug_printf_clean("<->");
- n = usrsctp_getpaddrs(sock, 0, &addrs);
- addr = addrs;
- for (i = 0; i < n; i++) {
- switch (addr->sa_family) {
- case AF_INET:
- {
- struct sockaddr_in *sin;
- char buf[INET_ADDRSTRLEN];
- const char *name;
- sin = (struct sockaddr_in *)addr;
- name = inet_ntop(AF_INET, &sin->sin_addr, buf, INET_ADDRSTRLEN);
- debug_printf_clean("%s:%d", name, ntohs(sin->sin_port));
- #if !defined(HAVE_SA_LEN)
- sa_len = (int)sizeof(struct sockaddr_in);
- #endif
- break;
- }
- case AF_INET6:
- {
- struct sockaddr_in6 *sin6;
- char buf[INET6_ADDRSTRLEN];
- const char *name;
- sin6 = (struct sockaddr_in6 *)addr;
- name = inet_ntop(AF_INET6, &sin6->sin6_addr, buf, INET6_ADDRSTRLEN);
- debug_printf_clean("%s:%d", name, ntohs(sin6->sin6_port));
- #if !defined(HAVE_SA_LEN)
- sa_len = (int)sizeof(struct sockaddr_in6);
- #endif
- break;
- }
- case AF_CONN:
- {
- struct sockaddr_conn *sconn;
- sconn = (struct sockaddr_conn *)addr;
- debug_printf_clean("%p:%d", sconn->sconn_addr, ntohs(sconn->sconn_port));
- #if !defined(HAVE_SA_LEN)
- sa_len = (int)sizeof(struct sockaddr_conn);
- #endif
- break;
- }
- default:
- debug_printf_clean("Unknown family: %d", addr->sa_family);
- #if !defined(HAVE_SA_LEN)
- sa_len = (int)sizeof(struct sockaddr);
- #endif
- break;
- }
- #if !defined(HAVE_SA_LEN)
- addr = (struct sockaddr *)((char *)addr + sa_len);
- #else
- addr = (struct sockaddr *)((caddr_t)addr + addr->sa_len);
- #endif
- if (i != n - 1) {
- debug_printf_clean(",");
- }
- }
- if (n > 0) {
- usrsctp_freepaddrs(addrs);
- }
- debug_printf_clean("\n");
- }
- int
- main(int argc, char *argv[])
- {
- struct sockaddr_in sin_s, sin_c;
- struct sockaddr_conn sconn;
- struct sctp_paddrparams paddrparams;
- #ifdef _WIN32
- SOCKET fd_c, fd_s;
- #else
- int fd_c, fd_s, rc;
- #endif
- struct socket *s_c, *s_s, *s_l;
- #ifdef _WIN32
- HANDLE tid_c, tid_s;
- #else
- pthread_t tid_c, tid_s;
- #endif
- int i, j, cur_buf_size, snd_buf_size, rcv_buf_size, sendv_retries_left, on;
- socklen_t opt_len;
- struct sctp_sndinfo sndinfo;
- char *line;
- #ifdef _WIN32
- WSADATA wsaData;
- #endif
- uint16_t client_port = 9900;
- uint16_t server_port = 9901;
- struct upcall_meta upcall_meta_client;
- struct upcall_meta upcall_meta_server;
- #ifdef OUTPUT_TO_LOGFILE
- FILE *logfile;
- logfile = fopen("ekr_loop_upcall.log", "a+");
- if (logfile == NULL) {
- debug_printf("Failed creating logfile\n");
- exit(EXIT_FAILURE);
- }
- debug_set_target(logfile);
- #endif
- if (argc > 1) {
- client_port = atoi(argv[1]);
- }
- if (argc > 2) {
- server_port = atoi(argv[2]);
- }
- if (argc > 3) {
- crc32c_offloading = atoi(argv[3]);
- }
- debug_printf("Starting program\n");
- debug_printf("Config:\n\tClient Port:\t%d\n\tServer Port:\t%d\n\tCRC32C Calc:\t%s\n", client_port, server_port, crc32c_offloading ? "offloaded" : "NOT offloaded");
- #ifdef _WIN32
- if (WSAStartup(MAKEWORD(2,2), &wsaData) != 0) {
- debug_printf("WSAStartup failed\n");
- exit (EXIT_FAILURE);
- }
- #endif
- usrsctp_init(0, conn_output, debug_printf_stack);
- #ifdef SCTP_DEBUG
- usrsctp_sysctl_set_sctp_debug_on(SCTP_DEBUG_NONE);
- #endif // SCTP_DEBUG
- if (crc32c_offloading) {
- usrsctp_enable_crc32c_offload();
- }
- /* set up a connected UDP socket */
- #ifdef _WIN32
- if ((fd_c = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP)) == INVALID_SOCKET) {
- debug_printf("socket() failed with error: %d\n", WSAGetLastError());
- exit(EXIT_FAILURE);
- }
- if ((fd_s = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP)) == INVALID_SOCKET) {
- debug_printf("socket() failed with error: %d\n", WSAGetLastError());
- exit(EXIT_FAILURE);
- }
- #else
- if ((fd_c = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP)) < 0) {
- perror("socket");
- exit(EXIT_FAILURE);
- }
- if ((fd_s = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP)) < 0) {
- perror("socket");
- exit(EXIT_FAILURE);
- }
- #endif
- memset(&sin_c, 0, sizeof(struct sockaddr_in));
- sin_c.sin_family = AF_INET;
- #ifdef HAVE_SIN_LEN
- sin_c.sin_len = sizeof(struct sockaddr_in);
- #endif
- sin_c.sin_port = htons(client_port);
- sin_c.sin_addr.s_addr = htonl(INADDR_LOOPBACK);
- memset(&sin_s, 0, sizeof(struct sockaddr_in));
- sin_s.sin_family = AF_INET;
- #ifdef HAVE_SIN_LEN
- sin_s.sin_len = sizeof(struct sockaddr_in);
- #endif
- sin_s.sin_port = htons(server_port);
- sin_s.sin_addr.s_addr = htonl(INADDR_LOOPBACK);
- #ifdef _WIN32
- if (bind(fd_c, (struct sockaddr *)&sin_c, sizeof(struct sockaddr_in)) == SOCKET_ERROR) {
- debug_printf("bind() failed with error: %d\n", WSAGetLastError());
- exit(EXIT_FAILURE);
- }
- if (bind(fd_s, (struct sockaddr *)&sin_s, sizeof(struct sockaddr_in)) == SOCKET_ERROR) {
- debug_printf("bind() failed with error: %d\n", WSAGetLastError());
- exit(EXIT_FAILURE);
- }
- #else
- if (bind(fd_c, (struct sockaddr *)&sin_c, sizeof(struct sockaddr_in)) < 0) {
- perror("bind");
- exit(EXIT_FAILURE);
- }
- if (bind(fd_s, (struct sockaddr *)&sin_s, sizeof(struct sockaddr_in)) < 0) {
- perror("bind");
- exit(EXIT_FAILURE);
- }
- #endif
- #ifdef _WIN32
- if (connect(fd_c, (struct sockaddr *)&sin_s, sizeof(struct sockaddr_in)) == SOCKET_ERROR) {
- debug_printf("connect() failed with error: %d\n", WSAGetLastError());
- exit(EXIT_FAILURE);
- }
- if (connect(fd_s, (struct sockaddr *)&sin_c, sizeof(struct sockaddr_in)) == SOCKET_ERROR) {
- debug_printf("connect() failed with error: %d\n", WSAGetLastError());
- exit(EXIT_FAILURE);
- }
- #else
- if (connect(fd_c, (struct sockaddr *)&sin_s, sizeof(struct sockaddr_in)) < 0) {
- perror("connect");
- exit(EXIT_FAILURE);
- }
- if (connect(fd_s, (struct sockaddr *)&sin_c, sizeof(struct sockaddr_in)) < 0) {
- perror("connect");
- exit(EXIT_FAILURE);
- }
- #endif
- #ifdef _WIN32
- if ((tid_c = CreateThread(NULL, 0, &handle_packets, (void *)&fd_c, 0, NULL)) == NULL) {
- debug_printf("CreateThread() failed with error: %d\n", GetLastError());
- exit(EXIT_FAILURE);
- }
- if ((tid_s = CreateThread(NULL, 0, &handle_packets, (void *)&fd_s, 0, NULL)) == NULL) {
- debug_printf("CreateThread() failed with error: %d\n", GetLastError());
- exit(EXIT_FAILURE);
- }
- #else
- if ((rc = pthread_create(&tid_c, NULL, &handle_packets, (void *)&fd_c)) != 0) {
- debug_printf("pthread_create tid_c: %s\n", strerror(rc));
- exit(EXIT_FAILURE);
- }
- if ((rc = pthread_create(&tid_s, NULL, &handle_packets, (void *)&fd_s)) != 0) {
- debug_printf("pthread_create tid_s: %s\n", strerror(rc));
- exit(EXIT_FAILURE);
- };
- #endif
- usrsctp_sysctl_set_sctp_ecn_enable(0);
- usrsctp_register_address((void *)&fd_c);
- usrsctp_register_address((void *)&fd_s);
- if ((s_c = usrsctp_socket(AF_CONN, SOCK_STREAM, IPPROTO_SCTP, NULL, NULL, 0, NULL)) == NULL) {
- perror("usrsctp_socket");
- exit(EXIT_FAILURE);
- }
- upcall_meta_client.addr = &fd_c;
- upcall_meta_client.sender = 1;
- usrsctp_set_upcall(s_c, handle_upcall, &upcall_meta_client);
- opt_len = (socklen_t)sizeof(int);
- cur_buf_size = 0;
- if (usrsctp_getsockopt(s_c, SOL_SOCKET, SO_SNDBUF, &cur_buf_size, &opt_len) < 0) {
- perror("usrsctp_getsockopt");
- exit(EXIT_FAILURE);
- }
- debug_printf("Change send socket buffer size from %d ", cur_buf_size);
- snd_buf_size = 1<<22; /* 4 MB */
- if (usrsctp_setsockopt(s_c, SOL_SOCKET, SO_SNDBUF, &snd_buf_size, sizeof(int)) < 0) {
- perror("usrsctp_setsockopt");
- exit(EXIT_FAILURE);
- }
- opt_len = (socklen_t)sizeof(int);
- cur_buf_size = 0;
- if (usrsctp_getsockopt(s_c, SOL_SOCKET, SO_SNDBUF, &cur_buf_size, &opt_len) < 0) {
- perror("usrsctp_getsockopt");
- exit(EXIT_FAILURE);
- }
- debug_printf_clean("to %d.\n", cur_buf_size);
- memset(&paddrparams, 0, sizeof(struct sctp_paddrparams));
- paddrparams.spp_address.ss_family = AF_CONN;
- #ifdef HAVE_SCONN_LEN
- paddrparams.spp_address.ss_len = sizeof(struct sockaddr_conn);
- #endif
- paddrparams.spp_flags = SPP_PMTUD_DISABLE;
- paddrparams.spp_pathmtu = 9000;
- if (usrsctp_setsockopt(s_c, IPPROTO_SCTP, SCTP_PEER_ADDR_PARAMS, &paddrparams, sizeof(struct sctp_paddrparams)) < 0) {
- perror("usrsctp_setsockopt");
- exit(EXIT_FAILURE);
- }
- if ((s_l = usrsctp_socket(AF_CONN, SOCK_STREAM, IPPROTO_SCTP, NULL, NULL, 0, NULL)) == NULL) {
- perror("usrsctp_socket");
- exit(EXIT_FAILURE);
- }
- opt_len = (socklen_t)sizeof(int);
- cur_buf_size = 0;
- if (usrsctp_getsockopt(s_l, SOL_SOCKET, SO_RCVBUF, &cur_buf_size, &opt_len) < 0) {
- perror("usrsctp_getsockopt");
- exit(EXIT_FAILURE);
- }
- debug_printf("Change receive socket buffer size from %d ", cur_buf_size);
- rcv_buf_size = 1<<16; /* 64 KB */
- if (usrsctp_setsockopt(s_l, SOL_SOCKET, SO_RCVBUF, &rcv_buf_size, sizeof(int)) < 0) {
- perror("usrsctp_setsockopt");
- exit(EXIT_FAILURE);
- }
- opt_len = (socklen_t)sizeof(int);
- cur_buf_size = 0;
- if (usrsctp_getsockopt(s_l, SOL_SOCKET, SO_RCVBUF, &cur_buf_size, &opt_len) < 0) {
- perror("usrsctp_getsockopt");
- exit(EXIT_FAILURE);
- }
- debug_printf_clean("to %d.\n", cur_buf_size);
- on = 1;
- if (usrsctp_setsockopt(s_l, IPPROTO_SCTP, SCTP_RECVRCVINFO, &on, sizeof(int)) < 0) {
- perror("usrsctp_setsockopt");
- exit(EXIT_FAILURE);
- }
- /* Bind the client side. */
- memset(&sconn, 0, sizeof(struct sockaddr_conn));
- sconn.sconn_family = AF_CONN;
- #ifdef HAVE_SCONN_LEN
- sconn.sconn_len = sizeof(struct sockaddr_conn);
- #endif
- sconn.sconn_port = htons(5001);
- sconn.sconn_addr = &fd_c;
- if (usrsctp_bind(s_c, (struct sockaddr *)&sconn, sizeof(struct sockaddr_conn)) < 0) {
- perror("usrsctp_bind");
- exit(EXIT_FAILURE);
- }
- /* Bind the server side. */
- memset(&sconn, 0, sizeof(struct sockaddr_conn));
- sconn.sconn_family = AF_CONN;
- #ifdef HAVE_SCONN_LEN
- sconn.sconn_len = sizeof(struct sockaddr_conn);
- #endif
- sconn.sconn_port = htons(5001);
- sconn.sconn_addr = &fd_s;
- if (usrsctp_bind(s_l, (struct sockaddr *)&sconn, sizeof(struct sockaddr_conn)) < 0) {
- perror("usrsctp_bind");
- exit(EXIT_FAILURE);
- }
- /* Make server side passive... */
- if (usrsctp_listen(s_l, 1) < 0) {
- perror("usrsctp_listen");
- exit(EXIT_FAILURE);
- }
- /* Initiate the handshake */
- memset(&sconn, 0, sizeof(struct sockaddr_conn));
- sconn.sconn_family = AF_CONN;
- #ifdef HAVE_SCONN_LEN
- sconn.sconn_len = sizeof(struct sockaddr_conn);
- #endif
- sconn.sconn_port = htons(5001);
- sconn.sconn_addr = &fd_c;
- if (usrsctp_connect(s_c, (struct sockaddr *)&sconn, sizeof(struct sockaddr_conn)) < 0) {
- perror("usrsctp_connect");
- exit(EXIT_FAILURE);
- }
- if ((s_s = usrsctp_accept(s_l, NULL, NULL)) == NULL) {
- perror("usrsctp_accept");
- exit(EXIT_FAILURE);
- }
- upcall_meta_server.addr = &fd_s;
- upcall_meta_server.sender = 0;
- usrsctp_set_upcall(s_s, handle_upcall, &upcall_meta_server);
- usrsctp_close(s_l);
- print_addresses(s_s);
- if ((line = malloc(LINE_LENGTH)) == NULL) {
- exit(EXIT_FAILURE);
- }
- memset(line, 'A', LINE_LENGTH);
- sndinfo.snd_sid = 1;
- sndinfo.snd_ppid = htonl(DISCARD_PPID);
- sndinfo.snd_context = 0;
- sndinfo.snd_assoc_id = 0;
- for (i = 0; i < NUMBER_OF_STEPS; i++) {
- if (i % 2) {
- sndinfo.snd_flags = SCTP_UNORDERED;
- } else {
- sndinfo.snd_flags = 0;
- }
- for (j = 0; j < 2; j++) {
- /* Send a 1 MB message */
- sendv_retries_left = 120;
- debug_printf("usrscp_sendv - step %d - call %d flags %x\n", i, j + 1, sndinfo.snd_flags);
- while (usrsctp_sendv(s_c, line, LINE_LENGTH, NULL, 0, (void *)&sndinfo,
- (socklen_t)sizeof(struct sctp_sndinfo), SCTP_SENDV_SNDINFO, 0) < 0) {
- debug_printf("usrsctp_sendv - errno: %d - %s\n", errno, strerror(errno));
- if (errno != EWOULDBLOCK || !sendv_retries_left) {
- exit(EXIT_FAILURE);
- }
- sendv_retries_left--;
- #ifdef _WIN32
- Sleep(1000);
- #else
- sleep(1);
- #endif
- }
- }
- debug_printf("Sending done, sleeping\n");
- #ifdef _WIN32
- Sleep(1000);
- #else
- sleep(1);
- #endif
- }
- free(line);
- usrsctp_shutdown(s_c, SHUT_WR);
- while (usrsctp_finish() != 0) {
- debug_printf("Waiting for usrsctp_finish()\n");
- #ifdef _WIN32
- Sleep(1000);
- #else
- sleep(1);
- #endif
- }
- #ifdef _WIN32
- TerminateThread(tid_c, 0);
- WaitForSingleObject(tid_c, INFINITE);
- TerminateThread(tid_s, 0);
- WaitForSingleObject(tid_s, INFINITE);
- if (closesocket(fd_c) == SOCKET_ERROR) {
- debug_printf("closesocket() failed with error: %d\n", WSAGetLastError());
- exit(EXIT_FAILURE);
- }
- if (closesocket(fd_s) == SOCKET_ERROR) {
- debug_printf("closesocket() failed with error: %d\n", WSAGetLastError());
- exit(EXIT_FAILURE);
- }
- WSACleanup();
- #else
- pthread_cancel(tid_c);
- pthread_join(tid_c, NULL);
- pthread_cancel(tid_s);
- pthread_join(tid_s, NULL);
- if (close(fd_c) < 0) {
- perror("close");
- exit(EXIT_FAILURE);
- }
- if (close(fd_s) < 0) {
- perror("close");
- exit(EXIT_FAILURE);
- }
- #endif
- return (0);
- }
|