multinetwork/common.cpp

   1 /*
   2  * Copyright (C) 2016 The Android Open Source Project
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  *      http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17
  18 #include "common.h"
  19
  20 #include <android/api-level.h>
  21 #include <arpa/inet.h>
  22 #include <assert.h>
  23 #include <ctype.h>
  24 #include <errno.h>
  25 #include <stdlib.h>
  26 #include <string.h>
  27 #include <iostream>
  28
  29
  30 namespace {
  31
  32 bool strEqual(const char *a, const char *b) {
  33     return strcmp(a, b) == 0;
  34 }
  35
  36 // Allow specifying network handles in decimal and hexadecimal.
  37 bool parseNetworkHandle(const char *arg, net_handle_t *nethandle) {
  38     if (arg == nullptr || !isdigit(arg[0]) || nethandle == nullptr) {
  39         return false;
  40     }
  41
  42     net_handle_t nh;
  43     char *end = nullptr;
  44
  45     nh = strtoull(arg, &end, 0);
  46     if (end != nullptr && *end == '\0') {
  47         *nethandle = nh;
  48         return true;
  49     }
  50     return false;
  51 }
  52
  53 }  // namespace
  54
  55
  56 void printUsage(const char *progname) {
  57     std::cerr << "Usage: " << progname
  58               << " [--nethandle <nethandle>]"
  59               << " [--mode explicit|process]"
  60               << " [--family unspec|ipv4|ipv6]"
  61               << " <argument>"
  62               << std::endl;
  63     std::cerr << std::endl;
  64     std::cerr << "Learn nethandle values from 'dumpsys connectivity --short' "
  65               << "or 'dumpsys connectivity --diag'"
  66               << std::endl;
  67 }
  68
  69 Arguments::~Arguments() {}
  70
  71 bool Arguments::parseArguments(int argc, const char* argv[]) {
  72     if (argc < 1 || argv == nullptr) { return false; }
  73
  74     for (int i = 1; i < argc; i++) {
  75         if (strEqual(argv[i], "--nethandle")) {
  76             i++;
  77             if (argc == i) break;
  78             if (!parseNetworkHandle(argv[i], &nethandle)) {
  79                 std::cerr << "Failed to parse nethandle: '" << argv[i] << "'"
  80                           << std::endl;
  81                 break;
  82             }
  83         } else if (strEqual(argv[i], "--family")) {
  84             i++;
  85             if (argc == i) break;
  86             if (strEqual(argv[i], "unspec")) {
  87                 family = AF_UNSPEC;
  88             } else if (strEqual(argv[i], "ipv4")) {
  89                 family = AF_INET;
  90             } else if (strEqual(argv[i], "ipv6")) {
  91                 family = AF_INET6;
  92             } else {
  93                 break;
  94             }
  95         } else if (strEqual(argv[i], "--mode")) {
  96             i++;
  97             if (argc == i) break;
  98             if (strEqual(argv[i], "explicit")) {
  99                 api_mode = ApiMode::EXPLICIT;
 100             } else if (strEqual(argv[i], "process")) {
 101                 api_mode = ApiMode::PROCESS;
 102             } else {
 103                 break;
 104             }
 105         } else if (arg1 == nullptr) {
 106             arg1 = argv[i];
 107         } else {
 108             arg1 = nullptr;
 109             break;
 110         }
 111     }
 112
 113     if (arg1 != nullptr) {
 114         return true;
 115     }
 116
 117     printUsage(argv[0]);
 118     return false;
 119 }
 120
 121
 122 std::string inetSockaddrToString(const sockaddr* sa) {
 123     const bool is_ipv6 = (sa->sa_family == AF_INET6);
 124     char host[INET6_ADDRSTRLEN];
 125     char port[sizeof("65535")];
 126     getnameinfo(sa, is_ipv6 ? sizeof(sockaddr_in6) : sizeof(sockaddr_in),
 127                 host, sizeof(host),
 128                 port, sizeof(port),
 129                 NI_NUMERICHOST | NI_NUMERICSERV);
 130
 131     if (port[0] == '0' || port[0] == '\0') {
 132         return std::string(host);
 133     }
 134     return (is_ipv6 ? "[" : "") + std::string(host) + (is_ipv6 ? "]:" : ":") + std::string(port);
 135 }