Commit | Line | Data |
---|---|---|
d3fafed3 MW |
1 | #define _GNU_SOURCE |
2 | ||
3 | #include <ctype.h> | |
4 | #include <errno.h> | |
5 | #include <limits.h> | |
6 | #include <stdio.h> | |
7 | #include <stdlib.h> | |
8 | #include <string.h> | |
9 | ||
10 | #include <getopt.h> | |
11 | ||
12 | #include <netdb.h> | |
13 | ||
14 | struct nameval { const char *name; int val; }; | |
15 | const char *prog = "<unset>"; | |
16 | ||
17 | static int parseint(const char *s, const char *what) | |
18 | { | |
19 | char *q; | |
20 | unsigned long i; | |
21 | int e = errno; | |
22 | ||
23 | if (!isdigit((unsigned char)*s)) goto fail; | |
24 | errno = 0; | |
25 | i = strtoul(s, &q, 0); | |
26 | if (errno || *q || i > INT_MAX) goto fail; | |
27 | errno = e; | |
28 | return ((int)i); | |
29 | ||
30 | fail: | |
31 | fprintf(stderr, "%s: bad numeric %s `%s'\n", prog, what, s); | |
32 | exit(1); | |
33 | } | |
34 | ||
35 | static int lookup(const char *name, const char *what, | |
36 | const struct nameval *nv) | |
37 | { | |
38 | if (isdigit((unsigned char)*name)) | |
39 | return (parseint(name, what)); | |
40 | for (; nv->name; nv++) | |
41 | if (strcmp(nv->name, name) == 0) return nv->val; | |
42 | fprintf(stderr, "%s: unknown %s `%s'\n", prog, what, name); | |
43 | exit(1); | |
44 | } | |
45 | ||
46 | static const char *rlookup(int val, const char *what, | |
47 | const struct nameval *nv) | |
48 | { | |
49 | static char buf[32]; | |
50 | for (; nv->name; nv++) | |
51 | if (nv->val == val) return (nv->name); | |
52 | sprintf(buf, "%s#%d", what, val); | |
53 | return (buf); | |
54 | } | |
55 | ||
56 | static const struct nameval familymap[] = { | |
57 | { "unspec", AF_UNSPEC }, | |
58 | { "inet", AF_INET }, | |
59 | { "inet6", AF_INET6 }, | |
60 | { 0 } | |
61 | }; | |
62 | ||
63 | static const struct nameval typemap[] = { | |
64 | { "any", 0 }, | |
65 | { "stream", SOCK_STREAM }, | |
66 | { "dgram", SOCK_DGRAM }, | |
67 | { "raw", SOCK_RAW }, | |
68 | { 0 } | |
69 | }; | |
70 | ||
71 | static void usage(FILE *fp) | |
72 | { | |
73 | fprintf(fp, | |
74 | "Usage: %s " | |
75 | "[-CNacgimnsuv] [-f FAMILY] [-p PROTOCOL] [-t TYPE] " | |
76 | "NAME [SERVICE]\n", | |
77 | prog); | |
78 | } | |
79 | ||
80 | static void help(FILE *fp) | |
81 | { | |
82 | usage(fp); fputs("\n\ | |
83 | Options:\n\ | |
84 | \n\ | |
85 | -a plain v4 also (with -m) -n numeric host\n\ | |
86 | -c canonify name -N numeric service\n\ | |
87 | -C canonify to IDN (with -c) -s allow unassigned codepoints\n\ | |
88 | -g families with configured addrs -u check result is STD3 hostname\n\ | |
89 | -i convert to IDN -v passive\n\ | |
90 | -m v6-mapped v4 addresses\n\ | |
91 | \n\ | |
92 | -f unspec|inet|inet6 NAME -- hostname or `-' for none\n\ | |
93 | -p PROTOCOL -- /etc/protocols SERVICE -- number or /etc/services\n\ | |
94 | -t any|stream|dgram|raw\n\ | |
95 | ", fp); | |
96 | } | |
97 | ||
98 | int main(int argc, char *argv[]) | |
99 | { | |
100 | struct addrinfo *ai, hint = { .ai_family = AF_UNSPEC }; | |
101 | const char *name; | |
102 | const char *serv = 0; | |
103 | char namebuf[NI_MAXHOST], servbuf[NI_MAXSERV]; | |
104 | int err; | |
105 | ||
106 | prog = strrchr(argv[0], '/'); | |
107 | if (prog) prog++; | |
108 | else prog = argv[0]; | |
109 | ||
110 | for (;;) { | |
111 | int opt = getopt(argc, argv, "hf:t:p:nNcvgamiCus"); | |
112 | if (opt < 0) break; | |
113 | switch (opt) { | |
114 | case 'h': help(stdout); exit(0); | |
115 | case 'f': | |
116 | hint.ai_family = lookup(optarg, "family", familymap); | |
117 | break; | |
118 | case 't': | |
119 | hint.ai_socktype = lookup(optarg, "type", typemap); | |
120 | break; | |
121 | case 'p': | |
122 | if (isdigit((unsigned char)*optarg)) | |
123 | hint.ai_protocol = parseint(optarg, "protocol"); | |
124 | else { | |
125 | struct protoent *p = getprotobyname(optarg); | |
126 | if (!p) { | |
127 | fprintf(stderr, "%s: unknown protocol `%s'\n", prog, optarg); | |
128 | exit(1); | |
129 | } | |
130 | hint.ai_protocol = p->p_proto; | |
131 | } | |
132 | break; | |
133 | case 'C': hint.ai_flags |= AI_CANONIDN; break; | |
134 | case 'N': hint.ai_flags |= AI_NUMERICSERV; break; | |
135 | case 'a': hint.ai_flags |= AI_ALL; break; | |
136 | case 'c': hint.ai_flags |= AI_CANONNAME; break; | |
137 | case 'g': hint.ai_flags |= AI_ADDRCONFIG; break; | |
138 | case 'i': hint.ai_flags |= AI_IDN; break; | |
139 | case 'm': hint.ai_flags |= AI_V4MAPPED; break; | |
140 | case 'n': hint.ai_flags |= AI_NUMERICHOST; break; | |
141 | case 's': hint.ai_flags |= AI_IDN_USE_STD3_ASCII_RULES; break; | |
142 | case 'u': hint.ai_flags |= AI_IDN_ALLOW_UNASSIGNED; break; | |
143 | case 'v': hint.ai_flags |= AI_PASSIVE; break; | |
144 | default: usage(stderr); exit(1); | |
145 | } | |
146 | } | |
147 | ||
148 | argv += optind; | |
149 | if (!*argv) { usage(stderr); exit(1); } name = *argv++; | |
150 | if (*argv) { serv = *argv++; } | |
151 | if (*argv) { usage(stderr); exit(1); } | |
152 | ||
153 | if (strcmp(name, "-") == 0) name = 0; | |
154 | ||
155 | if ((err = getaddrinfo(name, serv, &hint, &ai)) != 0) { | |
156 | fprintf(stderr, "%s: %s\n", prog, gai_strerror(err)); | |
157 | exit(1); | |
158 | } | |
159 | ||
160 | for (; ai; ai = ai->ai_next) { | |
161 | if (ai->ai_canonname) printf("[%s]\n", ai->ai_canonname); | |
162 | fputs(rlookup(ai->ai_family, "family", familymap), stdout); | |
163 | fputc(' ', stdout); | |
164 | fputs(rlookup(ai->ai_socktype, "type", typemap), stdout); | |
165 | fputc(' ', stdout); | |
166 | if ((err = getnameinfo(ai->ai_addr, ai->ai_addrlen, | |
167 | namebuf, sizeof(namebuf), | |
168 | servbuf, sizeof(servbuf), | |
169 | (NI_NUMERICHOST | NI_NUMERICSERV | | |
170 | (ai->ai_socktype == SOCK_DGRAM ? | |
171 | NI_DGRAM : 0)))) != 0) | |
172 | printf("(error: %s)", gai_strerror(err)); | |
173 | else | |
174 | printf("%s %s", namebuf, servbuf); | |
175 | fputc('\n', stdout); | |
176 | } | |
177 | ||
178 | freeaddrinfo(ai); | |
179 | ||
180 | return (0); | |
181 | } |