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
strEqual(const char * a,const char * b)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.
parseNetworkHandle(const char * arg,net_handle_t * nethandle)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
printUsage(const char * progname)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
~Arguments()69 Arguments::~Arguments() {}
70
parseArguments(int argc,const char * argv[])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
inetSockaddrToString(const sockaddr * sa)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 }
136