sbin/ping/main.c

   1 /*-
   2  * SPDX-License-Identifier: BSD-2-Clause-FreeBSD
   3  *
   4  * Copyright (C) 2019 Jan Sucan <jansucan@FreeBSD.org>
   5  * All rights reserved.
   6  *
   7  * Redistribution and use in source and binary forms, with or without
   8  * modification, are permitted provided that the following conditions
   9  * are met:
  10  * 1. Redistributions of source code must retain the above copyright
  11  *    notice, this list of conditions and the following disclaimer.
  12  * 2. Redistributions in binary form must reproduce the above copyright
  13  *    notice, this list of conditions and the following disclaimer in the
  14  *    documentation and/or other materials provided with the distribution.
  15  *
  16  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
  17  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  18  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  19  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  20  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  21  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  22  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  23  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  24  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  25  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  26  * SUCH DAMAGE.
  27  */
  28
  29 #include <sys/cdefs.h>
  30 __FBSDID("$FreeBSD$");
  31
  32 #include <sys/types.h>
  33 #include <sys/socket.h>
  34
  35 #include <arpa/inet.h>
  36 #include <netdb.h>
  37 #include <netinet/in.h>
  38
  39 #include <err.h>
  40 #include <stdbool.h>
  41 #include <stdio.h>
  42 #include <stdlib.h>
  43 #include <string.h>
  44 #include <unistd.h>
  45
  46 #include "main.h"
  47 #ifdef INET
  48 #include "ping.h"
  49 #endif
  50 #ifdef INET6
  51 #include "ping6.h"
  52 #endif
  53
  54 #if defined(INET) && defined(INET6)
  55 #define OPTSTR ":46"
  56 #elif defined(INET)
  57 #define OPTSTR ":4"
  58 #elif defined(INET6)
  59 #define OPTSTR ":6"
  60 #else
  61 #define OPTSTR ""
  62 #endif
  63
  64 int
  65 main(int argc, char *argv[])
  66 {
  67 #if defined(INET) && defined(INET6)
  68         struct in_addr a;
  69         struct in6_addr a6;
  70 #endif
  71 #if defined(INET) || defined(INET6)
  72         struct addrinfo hints;
  73 #endif
  74         int ch;
  75 #ifdef INET
  76         bool ipv4 = false;
  77 #endif
  78 #ifdef INET6
  79         bool ipv6 = false;
  80
  81         if (strcmp(getprogname(), "ping6") == 0)
  82                 ipv6 = true;
  83 #endif
  84
  85         while ((ch = getopt(argc, argv, OPTSTR)) != -1) {
  86                 switch(ch) {
  87 #ifdef INET
  88                 case '4':
  89                         ipv4 = true;
  90                         break;
  91 #endif
  92 #ifdef INET6
  93                 case '6':
  94                         ipv6 = true;
  95                         break;
  96 #endif
  97                 default:
  98                         break;
  99                 }
 100         }
 101
 102         if (optind >= argc)
 103                 usage();
 104
 105         optreset = 1;
 106         optind = 1;
 107 #if defined(INET) && defined(INET6)
 108         if (ipv4 && ipv6)
 109                 errx(1, "-4 and -6 cannot be used simultaneously");
 110 #endif
 111
 112 #if defined(INET) && defined(INET6)
 113         if (inet_pton(AF_INET, argv[argc - 1], &a) == 1) {
 114                 if (ipv6)
 115                         errx(1, "IPv6 requested but IPv4 target address "
 116                             "provided");
 117                 hints.ai_family = AF_INET;
 118         }
 119         else if (inet_pton(AF_INET6, argv[argc - 1], &a6) == 1) {
 120                 if (ipv4)
 121                         errx(1, "IPv4 requested but IPv6 target address "
 122                             "provided");
 123                 hints.ai_family = AF_INET6;
 124         } else if (ipv6)
 125                 hints.ai_family = AF_INET6;
 126         else if (ipv4)
 127                 hints.ai_family = AF_INET;
 128         else {
 129                 if (!feature_present("inet6"))
 130                         hints.ai_family = AF_INET;
 131                 else if (!feature_present("inet"))
 132                         hints.ai_family = AF_INET6;
 133                 else {
 134                         struct addrinfo *res;
 135
 136                         memset(&hints, 0, sizeof(hints));
 137                         hints.ai_socktype = SOCK_RAW;
 138                         hints.ai_family = AF_UNSPEC;
 139                         getaddrinfo(argv[argc - 1], NULL, &hints, &res);
 140                         if (res != NULL) {
 141                                 hints.ai_family = res[0].ai_family;
 142                                 freeaddrinfo(res);
 143                         }
 144                 }
 145         }
 146 #elif defined(INET)
 147         hints.ai_family = AF_INET;
 148 #elif defined(INET6)
 149         hints.ai_family = AF_INET6;
 150 #endif
 151
 152 #ifdef INET
 153         if (hints.ai_family == AF_INET)
 154                 return ping(argc, argv);
 155 #endif /* INET */
 156 #ifdef INET6
 157         if (hints.ai_family == AF_INET6)
 158                 return ping6(argc, argv);
 159 #endif /* INET6 */
 160         errx(1, "Unknown host");
 161 }
 162
 163 void
 164 usage(void)
 165 {
 166         (void)fprintf(stderr,
 167             "usage:\n"
 168 #ifdef INET
 169             "\tping [-4AaDdfHnoQqRrv] [-C pcp] [-c count] "
 170             "[-G sweepmaxsize]\n"
 171             "       [-g sweepminsize] [-h sweepincrsize] [-i wait] "
 172             "[-l preload]\n"
 173             "       [-M mask | time] [-m ttl]"
 174 #ifdef IPSEC
 175             "[-P policy] "
 176 #endif
 177             "[-p pattern] [-S src_addr] \n"
 178             "       [-s packetsize] [-t timeout] [-W waittime] [-z tos] "
 179             "IPv4-host\n"
 180             "\tping [-4AaDdfHLnoQqRrv] [-C pcp] [-c count] [-I iface] "
 181             "[-i wait]\n"
 182             "       [-l preload] [-M mask | time] [-m ttl] "
 183 #ifdef IPSEC
 184             "[-P policy] "
 185 #endif
 186             "[-p pattern]\n"
 187             "       [-S src_addr] [-s packetsize] [-T ttl] [-t timeout] [-W waittime]\n"
 188             "            [-z tos] IPv4-mcast-group\n"
 189 #endif /* INET */
 190 #ifdef INET6
 191             "\tping [-6aADd"
 192 #if defined(IPSEC) && !defined(IPSEC_POLICY_IPSEC)
 193             "E"
 194 #endif
 195             "fHnNoOq"
 196 #ifdef IPV6_USE_MIN_MTU
 197             "u"
 198 #endif
 199             "vyY"
 200 #if defined(IPSEC) && !defined(IPSEC_POLICY_IPSEC)
 201             "Z"
 202 #endif
 203             "] "
 204             "[-b bufsiz] [-c count] [-e gateway]\n"
 205             "            [-I interface] [-i wait] [-k addrtype] [-l preload] "
 206             "[-m hoplimit]\n"
 207             "            [-p pattern]"
 208 #if defined(IPSEC) && defined(IPSEC_POLICY_IPSEC)
 209             " [-P policy]"
 210 #endif
 211             " [-S sourceaddr] [-s packetsize] [-t timeout]\n"
 212             "       [-W waittime] [-z tclass] [IPv6-hops ...] IPv6-host\n"
 213 #endif  /* INET6 */
 214             );
 215
 216         exit(1);
 217 }