/* * Ouroboros - Copyright (C) 2016 - 2021 * * Ouroboros ping application * * Dimitri Staessens <dimitri@ouroboros.rocks> * Sander Vrijders <sander@ouroboros.rocks> * * 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 copyright holder 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 COPYRIGHT HOLDERS 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 * COPYRIGHT HOLDER 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. */ #define _POSIX_C_SOURCE 199506L #define __XSI_VISIBLE 500 #include <ouroboros/dev.h> #include <ouroboros/fccntl.h> #include <ouroboros/fqueue.h> #include <ouroboros/qos.h> #include "time_utils.h" #include <limits.h> #include <stdio.h> #include <string.h> #include <pthread.h> #include <stdint.h> #include <stdbool.h> #include <signal.h> #include <stdlib.h> #include <sys/time.h> #include <arpa/inet.h> #include <math.h> #include <errno.h> #include <float.h> #define OPING_BUF_SIZE 1500 #define ECHO_REQUEST 0 #define ECHO_REPLY 1 #define OPING_MAX_FLOWS 256 struct c { char * s_apn; int interval; uint32_t count; int size; bool timestamp; qosspec_t qs; /* stats */ uint32_t sent; uint32_t rcvd; size_t ooo; bool quiet; double rtt_min; double rtt_max; double rtt_avg; double rtt_m2; pthread_t reader_pt; pthread_t writer_pt; } client; struct s { struct timespec times[OPING_MAX_FLOWS]; fset_t * flows; fqueue_t * fq; pthread_mutex_t lock; pthread_t cleaner_pt; pthread_t accept_pt; pthread_t server_pt; } server; struct oping_msg { uint32_t type; uint32_t id; uint64_t tv_sec; uint64_t tv_nsec; } __attribute__((packed)); #include "oping_client.c" #include "oping_server.c" static void usage(void) { printf("Usage: oping [OPTION]...\n" "Checks liveness between a client and a server\n" "and reports the Round Trip Time (RTT)\n\n" " -l, --listen Run in server mode\n" "\n" " -c, --count Number of packets\n" " -d, --duration Duration of the test (default 1s)\n" " -i, --interval Interval (default 1000ms)\n" " -n, --server-name Name of the oping server\n" " -q, --qos QoS (raw, best, video, voice, data)" "\n" " -s, --size Payload size (B, default 64)\n" " -Q, --quiet Only print final statistics\n" " -D, --timeofday Print time of day before each line" "\n" " --help Display this help text and exit\n"); } /* Times are in ms. */ static int time_mul(const char * rem) { if (strcmp(rem, "ms") == 0 || strcmp(rem, "") == 0) return 1; else if(strcmp(rem, "s") == 0) return 1000; else if (strcmp(rem, "m") == 0) return 60 * 1000; else if (strcmp(rem, "h") == 0) return 60 * 60 * 1000; else if (strcmp(rem, "d") == 0) return 60 * 60 * 24 * 1000; printf("Unknown time unit: %s.\n", rem); exit(EXIT_FAILURE); } int main(int argc, char ** argv) { int ret = -1; char * rem = NULL; bool serv = false; long duration = 0; char * qos = NULL; argc--; argv++; client.s_apn = NULL; client.interval = 1000; client.size = 64; client.count = INT_MAX; client.timestamp = false; client.qs = qos_raw; client.quiet = false; while (argc > 0) { if (strcmp(*argv, "-i") == 0 || strcmp(*argv, "--interval") == 0) { client.interval = strtol(*(++argv), &rem, 10); client.interval *= time_mul(rem); --argc; } else if (strcmp(*argv, "-n") == 0 || strcmp(*argv, "--server-name") == 0) { client.s_apn = *(++argv); --argc; } else if (strcmp(*argv, "-c") == 0 || strcmp(*argv, "--count") == 0) { client.count = strtol(*(++argv), &rem, 10); --argc; } else if (strcmp(*argv, "-d") == 0 || strcmp(*argv, "--duration") == 0) { duration = strtol(*(++argv), &rem, 10); duration *= time_mul(rem); --argc; } else if (strcmp(*argv, "-s") == 0 || strcmp(*argv, "--size") == 0) { client.size = strtol(*(++argv), &rem, 10); --argc; } else if (strcmp(*argv, "-q") == 0 || strcmp(*argv, "--qos") == 0) { qos = *(++argv); --argc; } else if (strcmp(*argv, "-l") == 0 || strcmp(*argv, "--listen") == 0) { serv = true; } else if (strcmp(*argv, "-D") == 0 || strcmp(*argv, "--timeofday") == 0) { client.timestamp = true; } else if (strcmp(*argv, "-Q") == 0 || strcmp(*argv, "--quiet") == 0) { client.quiet = true; } else { goto fail; } argc--; argv++; } if (duration > 0) { if (client.interval == 0) client.count = duration * 10; else client.count = duration / client.interval; } if (qos != NULL) { if (strcmp(qos, "best") == 0) client.qs = qos_best_effort; else if (strcmp(qos, "raw") == 0) client.qs = qos_raw; else if (strcmp(qos, "video") == 0) client.qs = qos_video; else if (strcmp(qos, "voice") == 0) client.qs = qos_voice; else if (strcmp(qos, "data") == 0) client.qs = qos_data; else if (strcmp(qos, "raw_crypt") == 0) client.qs = qos_raw_crypt; else printf("Unknown QoS cube, defaulting to raw.\n"); } if (serv) { ret = server_main(); } else { if (client.s_apn == NULL) { printf("No server specified.\n"); usage(); exit(EXIT_FAILURE); } if (client.interval > 10000) { printf("Ping interval truncated to 10s.\n"); client.interval = 10000; } if (client.size > OPING_BUF_SIZE) { printf("Packet size truncated to %d bytes.\n", OPING_BUF_SIZE); client.size = OPING_BUF_SIZE; } if (client.size < 64) { printf("Packet size set to 64 bytes.\n"); client.size = 64; } ret = client_main(); } if (ret < 0) exit(EXIT_FAILURE); exit(EXIT_SUCCESS); fail: usage(); exit(EXIT_FAILURE); }