Age Owner Branch data TLA Line data Source code
1 : : /*-------------------------------------------------------------------------
2 : : *
3 : : * ip.c
4 : : * IPv6-aware network access.
5 : : *
6 : : * Portions Copyright (c) 1996-2024, PostgreSQL Global Development Group
7 : : * Portions Copyright (c) 1994, Regents of the University of California
8 : : *
9 : : *
10 : : * IDENTIFICATION
11 : : * src/common/ip.c
12 : : *
13 : : * This file and the IPV6 implementation were initially provided by
14 : : * Nigel Kukard <nkukard@lbsd.net>, Linux Based Systems Design
15 : : * http://www.lbsd.net.
16 : : *
17 : : *-------------------------------------------------------------------------
18 : : */
19 : :
20 : : #ifndef FRONTEND
21 : : #include "postgres.h"
22 : : #else
23 : : #include "postgres_fe.h"
24 : : #endif
25 : :
26 : : #include <unistd.h>
27 : : #include <sys/stat.h>
28 : : #include <sys/socket.h>
29 : : #include <netdb.h>
30 : : #include <netinet/in.h>
31 : : #include <netinet/tcp.h>
32 : : #include <arpa/inet.h>
33 : : #include <sys/file.h>
34 : :
35 : : #include "common/ip.h"
36 : :
37 : :
38 : :
39 : : static int getaddrinfo_unix(const char *path,
40 : : const struct addrinfo *hintsp,
41 : : struct addrinfo **result);
42 : :
43 : : static int getnameinfo_unix(const struct sockaddr_un *sa, int salen,
44 : : char *node, int nodelen,
45 : : char *service, int servicelen,
46 : : int flags);
47 : :
48 : :
49 : : /*
50 : : * pg_getaddrinfo_all - get address info for Unix, IPv4 and IPv6 sockets
51 : : */
52 : : int
2781 heikki.linnakangas@i 53 :CBC 16148 : pg_getaddrinfo_all(const char *hostname, const char *servname,
54 : : const struct addrinfo *hintp, struct addrinfo **result)
55 : : {
56 : : int rc;
57 : :
58 : : /* not all versions of getaddrinfo() zero *result on failure */
59 : 16148 : *result = NULL;
60 : :
61 [ + + ]: 16148 : if (hintp->ai_family == AF_UNIX)
62 : 12236 : return getaddrinfo_unix(servname, hintp, result);
63 : :
64 : : /* NULL has special meaning to getaddrinfo(). */
65 [ + - + - ]: 3912 : rc = getaddrinfo((!hostname || hostname[0] == '\0') ? NULL : hostname,
66 : : servname, hintp, result);
67 : :
68 : 3912 : return rc;
69 : : }
70 : :
71 : :
72 : : /*
73 : : * pg_freeaddrinfo_all - free addrinfo structures for IPv4, IPv6, or Unix
74 : : *
75 : : * Note: the ai_family field of the original hint structure must be passed
76 : : * so that we can tell whether the addrinfo struct was built by the system's
77 : : * getaddrinfo() routine or our own getaddrinfo_unix() routine. Some versions
78 : : * of getaddrinfo() might be willing to return AF_UNIX addresses, so it's
79 : : * not safe to look at ai_family in the addrinfo itself.
80 : : */
81 : : void
2489 tgl@sss.pgh.pa.us 82 : 16148 : pg_freeaddrinfo_all(int hint_ai_family, struct addrinfo *ai)
83 : : {
2781 heikki.linnakangas@i 84 [ + + ]: 16148 : if (hint_ai_family == AF_UNIX)
85 : : {
86 : : /* struct was built by getaddrinfo_unix (see pg_getaddrinfo_all) */
87 [ + + ]: 24472 : while (ai != NULL)
88 : : {
89 : 12236 : struct addrinfo *p = ai;
90 : :
91 : 12236 : ai = ai->ai_next;
92 : 12236 : free(p->ai_addr);
93 : 12236 : free(p);
94 : : }
95 : : }
96 : : else
97 : : {
98 : : /* struct was built by getaddrinfo() */
99 [ + - ]: 3912 : if (ai != NULL)
100 : 3912 : freeaddrinfo(ai);
101 : : }
102 : 16148 : }
103 : :
104 : :
105 : : /*
106 : : * pg_getnameinfo_all - get name info for Unix, IPv4 and IPv6 sockets
107 : : *
108 : : * The API of this routine differs from the standard getnameinfo() definition
109 : : * in two ways: first, the addr parameter is declared as sockaddr_storage
110 : : * rather than struct sockaddr, and second, the node and service fields are
111 : : * guaranteed to be filled with something even on failure return.
112 : : */
113 : : int
2489 tgl@sss.pgh.pa.us 114 : 23838 : pg_getnameinfo_all(const struct sockaddr_storage *addr, int salen,
115 : : char *node, int nodelen,
116 : : char *service, int servicelen,
117 : : int flags)
118 : : {
119 : : int rc;
120 : :
2781 heikki.linnakangas@i 121 [ + - + + ]: 23838 : if (addr && addr->ss_family == AF_UNIX)
122 : 22442 : rc = getnameinfo_unix((const struct sockaddr_un *) addr, salen,
123 : : node, nodelen,
124 : : service, servicelen,
125 : : flags);
126 : : else
127 : 1396 : rc = getnameinfo((const struct sockaddr *) addr, salen,
128 : : node, nodelen,
129 : : service, servicelen,
130 : : flags);
131 : :
132 [ - + ]: 23838 : if (rc != 0)
133 : : {
2781 heikki.linnakangas@i 134 [ # # ]:UBC 0 : if (node)
135 : 0 : strlcpy(node, "???", nodelen);
136 [ # # ]: 0 : if (service)
137 : 0 : strlcpy(service, "???", servicelen);
138 : : }
139 : :
2781 heikki.linnakangas@i 140 :CBC 23838 : return rc;
141 : : }
142 : :
143 : :
144 : : /* -------
145 : : * getaddrinfo_unix - get unix socket info using IPv6-compatible API
146 : : *
147 : : * Bugs: only one addrinfo is set even though hintsp is NULL or
148 : : * ai_socktype is 0
149 : : * AI_CANONNAME is not supported.
150 : : * -------
151 : : */
152 : : static int
2489 tgl@sss.pgh.pa.us 153 : 12236 : getaddrinfo_unix(const char *path, const struct addrinfo *hintsp,
154 : : struct addrinfo **result)
155 : : {
638 peter@eisentraut.org 156 : 12236 : struct addrinfo hints = {0};
157 : : struct addrinfo *aip;
158 : : struct sockaddr_un *unp;
159 : :
2781 heikki.linnakangas@i 160 : 12236 : *result = NULL;
161 : :
162 [ - + ]: 12236 : if (strlen(path) >= sizeof(unp->sun_path))
2781 heikki.linnakangas@i 163 :UBC 0 : return EAI_FAIL;
164 : :
2781 heikki.linnakangas@i 165 [ - + ]:CBC 12236 : if (hintsp == NULL)
166 : : {
2781 heikki.linnakangas@i 167 :UBC 0 : hints.ai_family = AF_UNIX;
168 : 0 : hints.ai_socktype = SOCK_STREAM;
169 : : }
170 : : else
2781 heikki.linnakangas@i 171 :CBC 12236 : memcpy(&hints, hintsp, sizeof(hints));
172 : :
173 [ - + ]: 12236 : if (hints.ai_socktype == 0)
2781 heikki.linnakangas@i 174 :UBC 0 : hints.ai_socktype = SOCK_STREAM;
175 : :
2781 heikki.linnakangas@i 176 [ - + ]:CBC 12236 : if (hints.ai_family != AF_UNIX)
177 : : {
178 : : /* shouldn't have been called */
2781 heikki.linnakangas@i 179 :UBC 0 : return EAI_FAIL;
180 : : }
181 : :
2781 heikki.linnakangas@i 182 :CBC 12236 : aip = calloc(1, sizeof(struct addrinfo));
183 [ - + ]: 12236 : if (aip == NULL)
2781 heikki.linnakangas@i 184 :UBC 0 : return EAI_MEMORY;
185 : :
2781 heikki.linnakangas@i 186 :CBC 12236 : unp = calloc(1, sizeof(struct sockaddr_un));
187 [ - + ]: 12236 : if (unp == NULL)
188 : : {
2781 heikki.linnakangas@i 189 :UBC 0 : free(aip);
190 : 0 : return EAI_MEMORY;
191 : : }
192 : :
2781 heikki.linnakangas@i 193 :CBC 12236 : aip->ai_family = AF_UNIX;
194 : 12236 : aip->ai_socktype = hints.ai_socktype;
195 : 12236 : aip->ai_protocol = hints.ai_protocol;
196 : 12236 : aip->ai_next = NULL;
197 : 12236 : aip->ai_canonname = NULL;
198 : 12236 : *result = aip;
199 : :
200 : 12236 : unp->sun_family = AF_UNIX;
201 : 12236 : aip->ai_addr = (struct sockaddr *) unp;
202 : 12236 : aip->ai_addrlen = sizeof(struct sockaddr_un);
203 : :
204 : 12236 : strcpy(unp->sun_path, path);
205 : :
206 : : /*
207 : : * If the supplied path starts with @, replace that with a zero byte for
208 : : * the internal representation. In that mode, the entire sun_path is the
209 : : * address, including trailing zero bytes. But we set the address length
210 : : * to only include the length of the original string. That way the
211 : : * trailing zero bytes won't show up in any network or socket lists of the
212 : : * operating system. This is just a convention, also followed by other
213 : : * packages.
214 : : */
1236 peter@eisentraut.org 215 [ - + ]: 12236 : if (path[0] == '@')
216 : : {
1236 peter@eisentraut.org 217 :UBC 0 : unp->sun_path[0] = '\0';
218 : 0 : aip->ai_addrlen = offsetof(struct sockaddr_un, sun_path) + strlen(path);
219 : : }
220 : :
2781 heikki.linnakangas@i 221 :CBC 12236 : return 0;
222 : : }
223 : :
224 : : /*
225 : : * Convert an address to a hostname.
226 : : */
227 : : static int
2489 tgl@sss.pgh.pa.us 228 : 22442 : getnameinfo_unix(const struct sockaddr_un *sa, int salen,
229 : : char *node, int nodelen,
230 : : char *service, int servicelen,
231 : : int flags)
232 : : {
233 : : int ret;
234 : :
235 : : /* Invalid arguments. */
2781 heikki.linnakangas@i 236 [ + - + - : 22442 : if (sa == NULL || sa->sun_family != AF_UNIX ||
+ + ]
237 [ - + ]: 11516 : (node == NULL && service == NULL))
2781 heikki.linnakangas@i 238 :UBC 0 : return EAI_FAIL;
239 : :
2781 heikki.linnakangas@i 240 [ + + ]:CBC 22442 : if (node)
241 : : {
242 : 10926 : ret = snprintf(node, nodelen, "%s", "[local]");
2069 tgl@sss.pgh.pa.us 243 [ + - - + ]: 10926 : if (ret < 0 || ret >= nodelen)
2781 heikki.linnakangas@i 244 :UBC 0 : return EAI_MEMORY;
245 : : }
246 : :
2781 heikki.linnakangas@i 247 [ + + ]:CBC 22442 : if (service)
248 : : {
249 : : /*
250 : : * Check whether it looks like an abstract socket, but it could also
251 : : * just be an empty string.
252 : : */
1236 peter@eisentraut.org 253 [ + + - + ]: 22435 : if (sa->sun_path[0] == '\0' && sa->sun_path[1] != '\0')
1236 peter@eisentraut.org 254 :UBC 0 : ret = snprintf(service, servicelen, "@%s", sa->sun_path + 1);
255 : : else
1236 peter@eisentraut.org 256 :CBC 22435 : ret = snprintf(service, servicelen, "%s", sa->sun_path);
2069 tgl@sss.pgh.pa.us 257 [ + - - + ]: 22435 : if (ret < 0 || ret >= servicelen)
2781 heikki.linnakangas@i 258 :UBC 0 : return EAI_MEMORY;
259 : : }
260 : :
2781 heikki.linnakangas@i 261 :CBC 22442 : return 0;
262 : : }
|
