| OLD | NEW |
| (Empty) |
| 1 /* | |
| 2 PostgreSQL Database Management System | |
| 3 (formerly known as Postgres, then as Postgres95) | |
| 4 | |
| 5 Portions Copyright (c) 1996-2005, The PostgreSQL Global Development Group | |
| 6 | |
| 7 Portions Copyright (c) 1994, The Regents of the University of California | |
| 8 | |
| 9 Permission to use, copy, modify, and distribute this software and its | |
| 10 documentation for any purpose, without fee, and without a written agreement | |
| 11 is hereby granted, provided that the above copyright notice and this paragraph | |
| 12 and the following two paragraphs appear in all copies. | |
| 13 | |
| 14 IN NO EVENT SHALL THE UNIVERSITY OF CALIFORNIA BE LIABLE TO ANY PARTY FOR | |
| 15 DIRECT, INDIRECT, SPECIAL, INCIDENTAL, OR CONSEQUENTIAL DAMAGES, INCLUDING | |
| 16 LOST PROFITS, ARISING OUT OF THE USE OF THIS SOFTWARE AND ITS DOCUMENTATION, | |
| 17 EVEN IF THE UNIVERSITY OF CALIFORNIA HAS BEEN ADVISED OF THE POSSIBILITY OF | |
| 18 SUCH DAMAGE. | |
| 19 | |
| 20 THE UNIVERSITY OF CALIFORNIA SPECIFICALLY DISCLAIMS ANY WARRANTIES, | |
| 21 INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY | |
| 22 AND FITNESS FOR A PARTICULAR PURPOSE. THE SOFTWARE PROVIDED HEREUNDER IS | |
| 23 ON AN "AS IS" BASIS, AND THE UNIVERSITY OF CALIFORNIA HAS NO OBLIGATIONS | |
| 24 TO PROVIDE MAINTENANCE, SUPPORT, UPDATES, ENHANCEMENTS, OR MODIFICATIONS. | |
| 25 | |
| 26 */ | |
| 27 | |
| 28 /*------------------------------------------------------------------------- | |
| 29 * | |
| 30 * getaddrinfo.c | |
| 31 * Support getaddrinfo() on platforms that don't have it. | |
| 32 * | |
| 33 * We also supply getnameinfo() here, assuming that the platform will have | |
| 34 * it if and only if it has getaddrinfo(). If this proves false on some | |
| 35 * platform, we'll need to split this file and provide a separate configure | |
| 36 * test for getnameinfo(). | |
| 37 * | |
| 38 * Copyright (c) 2003-2007, PostgreSQL Global Development Group | |
| 39 * | |
| 40 * Copyright (C) 2007 Jeremy Allison. | |
| 41 * Modified to return multiple IPv4 addresses for Samba. | |
| 42 * | |
| 43 *------------------------------------------------------------------------- | |
| 44 */ | |
| 45 | |
| 46 #include "replace.h" | |
| 47 #include "system/network.h" | |
| 48 | |
| 49 #ifndef SMB_MALLOC | |
| 50 #define SMB_MALLOC(s) malloc(s) | |
| 51 #endif | |
| 52 | |
| 53 #ifndef SMB_STRDUP | |
| 54 #define SMB_STRDUP(s) strdup(s) | |
| 55 #endif | |
| 56 | |
| 57 static int check_hostent_err(struct hostent *hp) | |
| 58 { | |
| 59 if (!hp) { | |
| 60 switch (h_errno) { | |
| 61 case HOST_NOT_FOUND: | |
| 62 case NO_DATA: | |
| 63 return EAI_NONAME; | |
| 64 case TRY_AGAIN: | |
| 65 return EAI_AGAIN; | |
| 66 case NO_RECOVERY: | |
| 67 default: | |
| 68 return EAI_FAIL; | |
| 69 } | |
| 70 } | |
| 71 if (!hp->h_name || hp->h_addrtype != AF_INET) { | |
| 72 return EAI_FAIL; | |
| 73 } | |
| 74 return 0; | |
| 75 } | |
| 76 | |
| 77 static char *canon_name_from_hostent(struct hostent *hp, | |
| 78 int *perr) | |
| 79 { | |
| 80 char *ret = NULL; | |
| 81 | |
| 82 *perr = check_hostent_err(hp); | |
| 83 if (*perr) { | |
| 84 return NULL; | |
| 85 } | |
| 86 ret = SMB_STRDUP(hp->h_name); | |
| 87 if (!ret) { | |
| 88 *perr = EAI_MEMORY; | |
| 89 } | |
| 90 return ret; | |
| 91 } | |
| 92 | |
| 93 static char *get_my_canon_name(int *perr) | |
| 94 { | |
| 95 char name[HOST_NAME_MAX+1]; | |
| 96 | |
| 97 if (gethostname(name, HOST_NAME_MAX) == -1) { | |
| 98 *perr = EAI_FAIL; | |
| 99 return NULL; | |
| 100 } | |
| 101 /* Ensure null termination. */ | |
| 102 name[HOST_NAME_MAX] = '\0'; | |
| 103 return canon_name_from_hostent(gethostbyname(name), perr); | |
| 104 } | |
| 105 | |
| 106 static char *get_canon_name_from_addr(struct in_addr ip, | |
| 107 int *perr) | |
| 108 { | |
| 109 return canon_name_from_hostent( | |
| 110 gethostbyaddr(&ip, sizeof(ip), AF_INET), | |
| 111 perr); | |
| 112 } | |
| 113 | |
| 114 static struct addrinfo *alloc_entry(const struct addrinfo *hints, | |
| 115 struct in_addr ip, | |
| 116 unsigned short port) | |
| 117 { | |
| 118 struct sockaddr_in *psin = NULL; | |
| 119 struct addrinfo *ai = SMB_MALLOC(sizeof(*ai)); | |
| 120 | |
| 121 if (!ai) { | |
| 122 return NULL; | |
| 123 } | |
| 124 memset(ai, '\0', sizeof(*ai)); | |
| 125 | |
| 126 psin = SMB_MALLOC(sizeof(*psin)); | |
| 127 if (!psin) { | |
| 128 free(ai); | |
| 129 return NULL; | |
| 130 } | |
| 131 | |
| 132 memset(psin, '\0', sizeof(*psin)); | |
| 133 | |
| 134 psin->sin_family = AF_INET; | |
| 135 psin->sin_port = htons(port); | |
| 136 psin->sin_addr = ip; | |
| 137 | |
| 138 ai->ai_flags = 0; | |
| 139 ai->ai_family = AF_INET; | |
| 140 ai->ai_socktype = hints->ai_socktype; | |
| 141 ai->ai_protocol = hints->ai_protocol; | |
| 142 ai->ai_addrlen = sizeof(*psin); | |
| 143 ai->ai_addr = (struct sockaddr *) psin; | |
| 144 ai->ai_canonname = NULL; | |
| 145 ai->ai_next = NULL; | |
| 146 | |
| 147 return ai; | |
| 148 } | |
| 149 | |
| 150 /* | |
| 151 * get address info for a single ipv4 address. | |
| 152 * | |
| 153 * Bugs: - servname can only be a number, not text. | |
| 154 */ | |
| 155 | |
| 156 static int getaddr_info_single_addr(const char *service, | |
| 157 uint32_t addr, | |
| 158 const struct addrinfo *hints, | |
| 159 struct addrinfo **res) | |
| 160 { | |
| 161 | |
| 162 struct addrinfo *ai = NULL; | |
| 163 struct in_addr ip; | |
| 164 unsigned short port = 0; | |
| 165 | |
| 166 if (service) { | |
| 167 port = (unsigned short)atoi(service); | |
| 168 } | |
| 169 ip.s_addr = htonl(addr); | |
| 170 | |
| 171 ai = alloc_entry(hints, ip, port); | |
| 172 if (!ai) { | |
| 173 return EAI_MEMORY; | |
| 174 } | |
| 175 | |
| 176 /* If we're asked for the canonical name, | |
| 177 * make sure it returns correctly. */ | |
| 178 if (!(hints->ai_flags & AI_NUMERICSERV) && | |
| 179 hints->ai_flags & AI_CANONNAME) { | |
| 180 int err; | |
| 181 if (addr == INADDR_LOOPBACK || addr == INADDR_ANY) { | |
| 182 ai->ai_canonname = get_my_canon_name(&err); | |
| 183 } else { | |
| 184 ai->ai_canonname = | |
| 185 get_canon_name_from_addr(ip,&err); | |
| 186 } | |
| 187 if (ai->ai_canonname == NULL) { | |
| 188 freeaddrinfo(ai); | |
| 189 return err; | |
| 190 } | |
| 191 } | |
| 192 | |
| 193 *res = ai; | |
| 194 return 0; | |
| 195 } | |
| 196 | |
| 197 /* | |
| 198 * get address info for multiple ipv4 addresses. | |
| 199 * | |
| 200 * Bugs: - servname can only be a number, not text. | |
| 201 */ | |
| 202 | |
| 203 static int getaddr_info_name(const char *node, | |
| 204 const char *service, | |
| 205 const struct addrinfo *hints, | |
| 206 struct addrinfo **res) | |
| 207 { | |
| 208 struct addrinfo *listp = NULL, *prevp = NULL; | |
| 209 char **pptr = NULL; | |
| 210 int err; | |
| 211 struct hostent *hp = NULL; | |
| 212 unsigned short port = 0; | |
| 213 | |
| 214 if (service) { | |
| 215 port = (unsigned short)atoi(service); | |
| 216 } | |
| 217 | |
| 218 hp = gethostbyname(node); | |
| 219 err = check_hostent_err(hp); | |
| 220 if (err) { | |
| 221 return err; | |
| 222 } | |
| 223 | |
| 224 for(pptr = hp->h_addr_list; *pptr; pptr++) { | |
| 225 struct in_addr ip = *(struct in_addr *)*pptr; | |
| 226 struct addrinfo *ai = alloc_entry(hints, ip, port); | |
| 227 | |
| 228 if (!ai) { | |
| 229 freeaddrinfo(listp); | |
| 230 return EAI_MEMORY; | |
| 231 } | |
| 232 | |
| 233 if (!listp) { | |
| 234 listp = ai; | |
| 235 prevp = ai; | |
| 236 ai->ai_canonname = SMB_STRDUP(hp->h_name); | |
| 237 if (!ai->ai_canonname) { | |
| 238 freeaddrinfo(listp); | |
| 239 return EAI_MEMORY; | |
| 240 } | |
| 241 } else { | |
| 242 prevp->ai_next = ai; | |
| 243 prevp = ai; | |
| 244 } | |
| 245 } | |
| 246 *res = listp; | |
| 247 return 0; | |
| 248 } | |
| 249 | |
| 250 /* | |
| 251 * get address info for ipv4 sockets. | |
| 252 * | |
| 253 * Bugs: - servname can only be a number, not text. | |
| 254 */ | |
| 255 | |
| 256 int rep_getaddrinfo(const char *node, | |
| 257 const char *service, | |
| 258 const struct addrinfo * hintp, | |
| 259 struct addrinfo ** res) | |
| 260 { | |
| 261 struct addrinfo hints; | |
| 262 | |
| 263 /* Setup the hints struct. */ | |
| 264 if (hintp == NULL) { | |
| 265 memset(&hints, 0, sizeof(hints)); | |
| 266 hints.ai_family = AF_INET; | |
| 267 hints.ai_socktype = SOCK_STREAM; | |
| 268 } else { | |
| 269 memcpy(&hints, hintp, sizeof(hints)); | |
| 270 } | |
| 271 | |
| 272 if (hints.ai_family != AF_INET && hints.ai_family != AF_UNSPEC) { | |
| 273 return EAI_FAMILY; | |
| 274 } | |
| 275 | |
| 276 if (hints.ai_socktype == 0) { | |
| 277 hints.ai_socktype = SOCK_STREAM; | |
| 278 } | |
| 279 | |
| 280 if (!node && !service) { | |
| 281 return EAI_NONAME; | |
| 282 } | |
| 283 | |
| 284 if (node) { | |
| 285 if (node[0] == '\0') { | |
| 286 return getaddr_info_single_addr(service, | |
| 287 INADDR_ANY, | |
| 288 &hints, | |
| 289 res); | |
| 290 } else if (hints.ai_flags & AI_NUMERICHOST) { | |
| 291 struct in_addr ip; | |
| 292 if (!inet_aton(node, &ip)) { | |
| 293 return EAI_FAIL; | |
| 294 } | |
| 295 return getaddr_info_single_addr(service, | |
| 296 ntohl(ip.s_addr), | |
| 297 &hints, | |
| 298 res); | |
| 299 } else { | |
| 300 return getaddr_info_name(node, | |
| 301 service, | |
| 302 &hints, | |
| 303 res); | |
| 304 } | |
| 305 } else if (hints.ai_flags & AI_PASSIVE) { | |
| 306 return getaddr_info_single_addr(service, | |
| 307 INADDR_ANY, | |
| 308 &hints, | |
| 309 res); | |
| 310 } | |
| 311 return getaddr_info_single_addr(service, | |
| 312 INADDR_LOOPBACK, | |
| 313 &hints, | |
| 314 res); | |
| 315 } | |
| 316 | |
| 317 | |
| 318 void rep_freeaddrinfo(struct addrinfo *res) | |
| 319 { | |
| 320 struct addrinfo *next = NULL; | |
| 321 | |
| 322 for (;res; res = next) { | |
| 323 next = res->ai_next; | |
| 324 if (res->ai_canonname) { | |
| 325 free(res->ai_canonname); | |
| 326 } | |
| 327 if (res->ai_addr) { | |
| 328 free(res->ai_addr); | |
| 329 } | |
| 330 free(res); | |
| 331 } | |
| 332 } | |
| 333 | |
| 334 | |
| 335 const char *rep_gai_strerror(int errcode) | |
| 336 { | |
| 337 #ifdef HAVE_HSTRERROR | |
| 338 int hcode; | |
| 339 | |
| 340 switch (errcode) | |
| 341 { | |
| 342 case EAI_NONAME: | |
| 343 hcode = HOST_NOT_FOUND; | |
| 344 break; | |
| 345 case EAI_AGAIN: | |
| 346 hcode = TRY_AGAIN; | |
| 347 break; | |
| 348 case EAI_FAIL: | |
| 349 default: | |
| 350 hcode = NO_RECOVERY; | |
| 351 break; | |
| 352 } | |
| 353 | |
| 354 return hstrerror(hcode); | |
| 355 #else /* !HAVE_HSTRERROR */ | |
| 356 | |
| 357 switch (errcode) | |
| 358 { | |
| 359 case EAI_NONAME: | |
| 360 return "Unknown host"; | |
| 361 case EAI_AGAIN: | |
| 362 return "Host name lookup failure"; | |
| 363 #ifdef EAI_BADFLAGS | |
| 364 case EAI_BADFLAGS: | |
| 365 return "Invalid argument"; | |
| 366 #endif | |
| 367 #ifdef EAI_FAMILY | |
| 368 case EAI_FAMILY: | |
| 369 return "Address family not supported"; | |
| 370 #endif | |
| 371 #ifdef EAI_MEMORY | |
| 372 case EAI_MEMORY: | |
| 373 return "Not enough memory"; | |
| 374 #endif | |
| 375 #ifdef EAI_NODATA | |
| 376 case EAI_NODATA: | |
| 377 return "No host data of that type was found"; | |
| 378 #endif | |
| 379 #ifdef EAI_SERVICE | |
| 380 case EAI_SERVICE: | |
| 381 return "Class type not found"; | |
| 382 #endif | |
| 383 #ifdef EAI_SOCKTYPE | |
| 384 case EAI_SOCKTYPE: | |
| 385 return "Socket type not supported"; | |
| 386 #endif | |
| 387 default: | |
| 388 return "Unknown server error"; | |
| 389 } | |
| 390 #endif /* HAVE_HSTRERROR */ | |
| 391 } | |
| 392 | |
| 393 static int gethostnameinfo(const struct sockaddr *sa, | |
| 394 char *node, | |
| 395 size_t nodelen, | |
| 396 int flags) | |
| 397 { | |
| 398 int ret = -1; | |
| 399 char *p = NULL; | |
| 400 | |
| 401 if (!(flags & NI_NUMERICHOST)) { | |
| 402 struct hostent *hp = gethostbyaddr( | |
| 403 &((struct sockaddr_in *)sa)->sin_addr, | |
| 404 sizeof(struct in_addr), | |
| 405 sa->sa_family); | |
| 406 ret = check_hostent_err(hp); | |
| 407 if (ret == 0) { | |
| 408 /* Name looked up successfully. */ | |
| 409 ret = snprintf(node, nodelen, "%s", hp->h_name); | |
| 410 if (ret < 0 || (size_t)ret >= nodelen) { | |
| 411 return EAI_MEMORY; | |
| 412 } | |
| 413 if (flags & NI_NOFQDN) { | |
| 414 p = strchr(node,'.'); | |
| 415 if (p) { | |
| 416 *p = '\0'; | |
| 417 } | |
| 418 } | |
| 419 return 0; | |
| 420 } | |
| 421 | |
| 422 if (flags & NI_NAMEREQD) { | |
| 423 /* If we require a name and didn't get one, | |
| 424 * automatically fail. */ | |
| 425 return ret; | |
| 426 } | |
| 427 /* Otherwise just fall into the numeric host code... */ | |
| 428 } | |
| 429 p = inet_ntoa(((struct sockaddr_in *)sa)->sin_addr); | |
| 430 ret = snprintf(node, nodelen, "%s", p); | |
| 431 if (ret < 0 || (size_t)ret >= nodelen) { | |
| 432 return EAI_MEMORY; | |
| 433 } | |
| 434 return 0; | |
| 435 } | |
| 436 | |
| 437 static int getservicenameinfo(const struct sockaddr *sa, | |
| 438 char *service, | |
| 439 size_t servicelen, | |
| 440 int flags) | |
| 441 { | |
| 442 int ret = -1; | |
| 443 int port = ntohs(((struct sockaddr_in *)sa)->sin_port); | |
| 444 | |
| 445 if (!(flags & NI_NUMERICSERV)) { | |
| 446 struct servent *se = getservbyport( | |
| 447 port, | |
| 448 (flags & NI_DGRAM) ? "udp" : "tcp"); | |
| 449 if (se && se->s_name) { | |
| 450 /* Service name looked up successfully. */ | |
| 451 ret = snprintf(service, servicelen, "%s", se->s_name); | |
| 452 if (ret < 0 || (size_t)ret >= servicelen) { | |
| 453 return EAI_MEMORY; | |
| 454 } | |
| 455 return 0; | |
| 456 } | |
| 457 /* Otherwise just fall into the numeric service code... */ | |
| 458 } | |
| 459 ret = snprintf(service, servicelen, "%d", port); | |
| 460 if (ret < 0 || (size_t)ret >= servicelen) { | |
| 461 return EAI_MEMORY; | |
| 462 } | |
| 463 return 0; | |
| 464 } | |
| 465 | |
| 466 /* | |
| 467 * Convert an ipv4 address to a hostname. | |
| 468 * | |
| 469 * Bugs: - No IPv6 support. | |
| 470 */ | |
| 471 int rep_getnameinfo(const struct sockaddr *sa, socklen_t salen, | |
| 472 char *node, size_t nodelen, | |
| 473 char *service, size_t servicelen, int flags) | |
| 474 { | |
| 475 | |
| 476 /* Invalid arguments. */ | |
| 477 if (sa == NULL || (node == NULL && service == NULL)) { | |
| 478 return EAI_FAIL; | |
| 479 } | |
| 480 | |
| 481 if (sa->sa_family != AF_INET) { | |
| 482 return EAI_FAIL; | |
| 483 } | |
| 484 | |
| 485 if (salen < sizeof(struct sockaddr_in)) { | |
| 486 return EAI_FAIL; | |
| 487 } | |
| 488 | |
| 489 if (node) { | |
| 490 return gethostnameinfo(sa, node, nodelen, flags); | |
| 491 } | |
| 492 | |
| 493 if (service) { | |
| 494 return getservicenameinfo(sa, service, servicelen, flags); | |
| 495 } | |
| 496 return 0; | |
| 497 } | |
| OLD | NEW |