1 /* 2 * Copyright (c) 2002 - 2003 3 * NetGroup, Politecnico di Torino (Italy) 4 * All rights reserved. 5 * 6 * Redistribution and use in source and binary forms, with or without 7 * modification, are permitted provided that the following conditions 8 * are met: 9 * 10 * 1. Redistributions of source code must retain the above copyright 11 * notice, this list of conditions and the following disclaimer. 12 * 2. Redistributions in binary form must reproduce the above copyright 13 * notice, this list of conditions and the following disclaimer in the 14 * documentation and/or other materials provided with the distribution. 15 * 3. Neither the name of the Politecnico di Torino nor the names of its 16 * contributors may be used to endorse or promote products derived from 17 * this software without specific prior written permission. 18 * 19 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 20 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 21 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 22 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 23 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 24 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 25 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 26 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 27 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 28 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 29 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 30 * 31 */ 32 33 #ifdef HAVE_CONFIG_H 34 #include <config.h> 35 #endif 36 37 /* 38 * \file sockutils.c 39 * 40 * The goal of this file is to provide a common set of primitives for socket 41 * manipulation. 42 * 43 * Although the socket interface defined in the RFC 2553 (and its updates) 44 * is excellent, there are still differences between the behavior of those 45 * routines on UN*X and Windows, and between UN*Xes. 46 * 47 * These calls provide an interface similar to the socket interface, but 48 * that hides the differences between operating systems. It does not 49 * attempt to significantly improve on the socket interface in other 50 * ways. 51 */ 52 53 #include "ftmacros.h" 54 55 #include <string.h> 56 #include <errno.h> /* for the errno variable */ 57 #include <stdio.h> /* for the stderr file */ 58 #include <stdlib.h> /* for malloc() and free() */ 59 #ifdef HAVE_LIMITS_H 60 #include <limits.h> 61 #else 62 #define INT_MAX 2147483647 63 #endif 64 65 #include "pcap-int.h" 66 67 #include "sockutils.h" 68 #include "portability.h" 69 70 #ifdef _WIN32 71 /* 72 * Winsock initialization. 73 * 74 * Ask for WinSock 2.2. 75 */ 76 #define WINSOCK_MAJOR_VERSION 2 77 #define WINSOCK_MINOR_VERSION 2 78 79 static int sockcount = 0; /*!< Variable that allows calling the WSAStartup() only one time */ 80 #endif 81 82 /* Some minor differences between UNIX and Win32 */ 83 #ifdef _WIN32 84 #define SHUT_WR SD_SEND /* The control code for shutdown() is different in Win32 */ 85 #endif 86 87 /* Size of the buffer that has to keep error messages */ 88 #define SOCK_ERRBUF_SIZE 1024 89 90 /* Constants; used in order to keep strings here */ 91 #define SOCKET_NO_NAME_AVAILABLE "No name available" 92 #define SOCKET_NO_PORT_AVAILABLE "No port available" 93 #define SOCKET_NAME_NULL_DAD "Null address (possibly DAD Phase)" 94 95 /* 96 * On UN*X, send() and recv() return ssize_t. 97 * 98 * On Windows, send() and recv() return an int. 99 * 100 * Wth MSVC, there *is* no ssize_t. 101 * 102 * With MinGW, there is an ssize_t type; it is either an int (32 bit) 103 * or a long long (64 bit). 104 * 105 * So, on Windows, if we don't have ssize_t defined, define it as an 106 * int, so we can use it, on all platforms, as the type of variables 107 * that hold the return values from send() and recv(). 108 */ 109 #if defined(_WIN32) && !defined(_SSIZE_T_DEFINED) 110 typedef int ssize_t; 111 #endif 112 113 /**************************************************** 114 * * 115 * Locally defined functions * 116 * * 117 ****************************************************/ 118 119 static int sock_ismcastaddr(const struct sockaddr *saddr); 120 121 /**************************************************** 122 * * 123 * Function bodies * 124 * * 125 ****************************************************/ 126 127 /* 128 * Format an error message given an errno value (UN*X) or a WinSock error 129 * (Windows). 130 */ 131 void sock_fmterror(const char *caller, int errcode, char *errbuf, int errbuflen) 132 { 133 #ifdef _WIN32 134 int retval; 135 char message[SOCK_ERRBUF_SIZE]; /* We're forcing "ANSI" */ 136 137 if (errbuf == NULL) 138 return; 139 140 retval = FormatMessageA(FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_IGNORE_INSERTS | 141 FORMAT_MESSAGE_MAX_WIDTH_MASK, 142 NULL, errcode, MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), 143 message, sizeof(message) / sizeof(TCHAR), NULL); 144 145 if (retval == 0) 146 { 147 if ((caller) && (*caller)) 148 pcap_snprintf(errbuf, errbuflen, "%sUnable to get the exact error message", caller); 149 else 150 pcap_snprintf(errbuf, errbuflen, "Unable to get the exact error message"); 151 } 152 else 153 { 154 if ((caller) && (*caller)) 155 pcap_snprintf(errbuf, errbuflen, "%s%s (code %d)", caller, message, errcode); 156 else 157 pcap_snprintf(errbuf, errbuflen, "%s (code %d)", message, errcode); 158 } 159 #else 160 char *message; 161 162 if (errbuf == NULL) 163 return; 164 165 message = strerror(errcode); 166 167 if ((caller) && (*caller)) 168 pcap_snprintf(errbuf, errbuflen, "%s%s (code %d)", caller, message, errcode); 169 else 170 pcap_snprintf(errbuf, errbuflen, "%s (code %d)", message, errcode); 171 #endif 172 } 173 174 /* 175 * \brief It retrieves the error message after an error occurred in the socket interface. 176 * 177 * This function is defined because of the different way errors are returned in UNIX 178 * and Win32. This function provides a consistent way to retrieve the error message 179 * (after a socket error occurred) on all the platforms. 180 * 181 * \param caller: a pointer to a user-allocated string which contains a message that has 182 * to be printed *before* the true error message. It could be, for example, 'this error 183 * comes from the recv() call at line 31'. It may be NULL. 184 * 185 * \param errbuf: a pointer to an user-allocated buffer that will contain the complete 186 * error message. This buffer has to be at least 'errbuflen' in length. 187 * It can be NULL; in this case the error cannot be printed. 188 * 189 * \param errbuflen: length of the buffer that will contains the error. The error message cannot be 190 * larger than 'errbuflen - 1' because the last char is reserved for the string terminator. 191 * 192 * \return No return values. The error message is returned in the 'string' parameter. 193 */ 194 void sock_geterror(const char *caller, char *errbuf, int errbuflen) 195 { 196 #ifdef _WIN32 197 if (errbuf == NULL) 198 return; 199 sock_fmterror(caller, GetLastError(), errbuf, errbuflen); 200 #else 201 if (errbuf == NULL) 202 return; 203 sock_fmterror(caller, errno, errbuf, errbuflen); 204 #endif 205 } 206 207 /* 208 * \brief It initializes sockets. 209 * 210 * This function is pretty useless on UNIX, since socket initialization is not required. 211 * However it is required on Win32. In UNIX, this function appears to be completely empty. 212 * 213 * \param errbuf: a pointer to an user-allocated buffer that will contain the complete 214 * error message. This buffer has to be at least 'errbuflen' in length. 215 * It can be NULL; in this case the error cannot be printed. 216 * 217 * \param errbuflen: length of the buffer that will contains the error. The error message cannot be 218 * larger than 'errbuflen - 1' because the last char is reserved for the string terminator. 219 * 220 * \return '0' if everything is fine, '-1' if some errors occurred. The error message is returned 221 * in the 'errbuf' variable. 222 */ 223 #ifdef _WIN32 224 int sock_init(char *errbuf, int errbuflen) 225 { 226 if (sockcount == 0) 227 { 228 WSADATA wsaData; /* helper variable needed to initialize Winsock */ 229 230 if (WSAStartup(MAKEWORD(WINSOCK_MAJOR_VERSION, 231 WINSOCK_MINOR_VERSION), &wsaData) != 0) 232 { 233 if (errbuf) 234 pcap_snprintf(errbuf, errbuflen, "Failed to initialize Winsock\n"); 235 236 WSACleanup(); 237 238 return -1; 239 } 240 } 241 242 sockcount++; 243 #else 244 int sock_init(char *errbuf _U_, int errbuflen _U_) 245 { 246 #endif 247 return 0; 248 } 249 250 /* 251 * \brief It deallocates sockets. 252 * 253 * This function is pretty useless on UNIX, since socket deallocation is not required. 254 * However it is required on Win32. In UNIX, this function appears to be completely empty. 255 * 256 * \return No error values. 257 */ 258 void sock_cleanup(void) 259 { 260 #ifdef _WIN32 261 sockcount--; 262 263 if (sockcount == 0) 264 WSACleanup(); 265 #endif 266 } 267 268 /* 269 * \brief It checks if the sockaddr variable contains a multicast address. 270 * 271 * \return '0' if the address is multicast, '-1' if it is not. 272 */ 273 static int sock_ismcastaddr(const struct sockaddr *saddr) 274 { 275 if (saddr->sa_family == PF_INET) 276 { 277 struct sockaddr_in *saddr4 = (struct sockaddr_in *) saddr; 278 if (IN_MULTICAST(ntohl(saddr4->sin_addr.s_addr))) return 0; 279 else return -1; 280 } 281 else 282 { 283 struct sockaddr_in6 *saddr6 = (struct sockaddr_in6 *) saddr; 284 if (IN6_IS_ADDR_MULTICAST(&saddr6->sin6_addr)) return 0; 285 else return -1; 286 } 287 } 288 289 /* 290 * \brief It initializes a network connection both from the client and the server side. 291 * 292 * In case of a client socket, this function calls socket() and connect(). 293 * In the meanwhile, it checks for any socket error. 294 * If an error occurs, it writes the error message into 'errbuf'. 295 * 296 * In case of a server socket, the function calls socket(), bind() and listen(). 297 * 298 * This function is usually preceeded by the sock_initaddress(). 299 * 300 * \param addrinfo: pointer to an addrinfo variable which will be used to 301 * open the socket and such. This variable is the one returned by the previous call to 302 * sock_initaddress(). 303 * 304 * \param server: '1' if this is a server socket, '0' otherwise. 305 * 306 * \param nconn: number of the connections that are allowed to wait into the listen() call. 307 * This value has no meanings in case of a client socket. 308 * 309 * \param errbuf: a pointer to an user-allocated buffer that will contain the complete 310 * error message. This buffer has to be at least 'errbuflen' in length. 311 * It can be NULL; in this case the error cannot be printed. 312 * 313 * \param errbuflen: length of the buffer that will contains the error. The error message cannot be 314 * larger than 'errbuflen - 1' because the last char is reserved for the string terminator. 315 * 316 * \return the socket that has been opened (that has to be used in the following sockets calls) 317 * if everything is fine, INVALID_SOCKET if some errors occurred. The error message is returned 318 * in the 'errbuf' variable. 319 */ 320 SOCKET sock_open(struct addrinfo *addrinfo, int server, int nconn, char *errbuf, int errbuflen) 321 { 322 SOCKET sock; 323 #if defined(SO_NOSIGPIPE) || defined(IPV6_V6ONLY) || defined(IPV6_BINDV6ONLY) 324 int on = 1; 325 #endif 326 327 sock = socket(addrinfo->ai_family, addrinfo->ai_socktype, addrinfo->ai_protocol); 328 if (sock == INVALID_SOCKET) 329 { 330 sock_geterror("socket(): ", errbuf, errbuflen); 331 return INVALID_SOCKET; 332 } 333 334 /* 335 * Disable SIGPIPE, if we have SO_NOSIGPIPE. We don't want to 336 * have to deal with signals if the peer closes the connection, 337 * especially in client programs, which may not even be aware that 338 * they're sending to sockets. 339 */ 340 #ifdef SO_NOSIGPIPE 341 if (setsockopt(sock, SOL_SOCKET, SO_NOSIGPIPE, (char *)&on, 342 sizeof (int)) == -1) 343 { 344 sock_geterror("setsockopt(SO_NOSIGPIPE)", errbuf, errbuflen); 345 closesocket(sock); 346 return INVALID_SOCKET; 347 } 348 #endif 349 350 /* This is a server socket */ 351 if (server) 352 { 353 #if defined(IPV6_V6ONLY) || defined(IPV6_BINDV6ONLY) 354 /* 355 * Force the use of IPv6-only addresses. 356 * 357 * RFC 3493 indicates that you can support IPv4 on an 358 * IPv6 socket: 359 * 360 * https://tools.ietf.org/html/rfc3493#section-3.7 361 * 362 * and that this is the default behavior. This means 363 * that if we first create an IPv6 socket bound to the 364 * "any" address, it is, in effect, also bound to the 365 * IPv4 "any" address, so when we create an IPv4 socket 366 * and try to bind it to the IPv4 "any" address, it gets 367 * EADDRINUSE. 368 * 369 * Not all network stacks support IPv4 on IPv6 sockets; 370 * pre-NT 6 Windows stacks don't support it, and the 371 * OpenBSD stack doesn't support it for security reasons 372 * (see the OpenBSD inet6(4) man page). Therefore, we 373 * don't want to rely on this behavior. 374 * 375 * So we try to disable it, using either the IPV6_V6ONLY 376 * option from RFC 3493: 377 * 378 * https://tools.ietf.org/html/rfc3493#section-5.3 379 * 380 * or the IPV6_BINDV6ONLY option from older UN*Xes. 381 */ 382 #ifndef IPV6_V6ONLY 383 /* For older systems */ 384 #define IPV6_V6ONLY IPV6_BINDV6ONLY 385 #endif /* IPV6_V6ONLY */ 386 if (addrinfo->ai_family == PF_INET6) 387 { 388 if (setsockopt(sock, IPPROTO_IPV6, IPV6_V6ONLY, 389 (char *)&on, sizeof (int)) == -1) 390 { 391 if (errbuf) 392 pcap_snprintf(errbuf, errbuflen, "setsockopt(IPV6_V6ONLY)"); 393 closesocket(sock); 394 return INVALID_SOCKET; 395 } 396 } 397 #endif /* defined(IPV6_V6ONLY) || defined(IPV6_BINDV6ONLY) */ 398 399 /* WARNING: if the address is a mcast one, I should place the proper Win32 code here */ 400 if (bind(sock, addrinfo->ai_addr, (int) addrinfo->ai_addrlen) != 0) 401 { 402 sock_geterror("bind(): ", errbuf, errbuflen); 403 closesocket(sock); 404 return INVALID_SOCKET; 405 } 406 407 if (addrinfo->ai_socktype == SOCK_STREAM) 408 if (listen(sock, nconn) == -1) 409 { 410 sock_geterror("listen(): ", errbuf, errbuflen); 411 closesocket(sock); 412 return INVALID_SOCKET; 413 } 414 415 /* server side ended */ 416 return sock; 417 } 418 else /* we're the client */ 419 { 420 struct addrinfo *tempaddrinfo; 421 char *errbufptr; 422 size_t bufspaceleft; 423 424 tempaddrinfo = addrinfo; 425 errbufptr = errbuf; 426 bufspaceleft = errbuflen; 427 *errbufptr = 0; 428 429 /* 430 * We have to loop though all the addinfo returned. 431 * For instance, we can have both IPv6 and IPv4 addresses, but the service we're trying 432 * to connect to is unavailable in IPv6, so we have to try in IPv4 as well 433 */ 434 while (tempaddrinfo) 435 { 436 437 if (connect(sock, tempaddrinfo->ai_addr, (int) tempaddrinfo->ai_addrlen) == -1) 438 { 439 size_t msglen; 440 char TmpBuffer[100]; 441 char SocketErrorMessage[SOCK_ERRBUF_SIZE]; 442 443 /* 444 * We have to retrieve the error message before any other socket call completes, otherwise 445 * the error message is lost 446 */ 447 sock_geterror(NULL, SocketErrorMessage, sizeof(SocketErrorMessage)); 448 449 /* Returns the numeric address of the host that triggered the error */ 450 sock_getascii_addrport((struct sockaddr_storage *) tempaddrinfo->ai_addr, TmpBuffer, sizeof(TmpBuffer), NULL, 0, NI_NUMERICHOST, TmpBuffer, sizeof(TmpBuffer)); 451 452 pcap_snprintf(errbufptr, bufspaceleft, 453 "Is the server properly installed on %s? connect() failed: %s", TmpBuffer, SocketErrorMessage); 454 455 /* In case more then one 'connect' fails, we manage to keep all the error messages */ 456 msglen = strlen(errbufptr); 457 458 errbufptr[msglen] = ' '; 459 errbufptr[msglen + 1] = 0; 460 461 bufspaceleft = bufspaceleft - (msglen + 1); 462 errbufptr += (msglen + 1); 463 464 tempaddrinfo = tempaddrinfo->ai_next; 465 } 466 else 467 break; 468 } 469 470 /* 471 * Check how we exit from the previous loop 472 * If tempaddrinfo is equal to NULL, it means that all the connect() failed. 473 */ 474 if (tempaddrinfo == NULL) 475 { 476 closesocket(sock); 477 return INVALID_SOCKET; 478 } 479 else 480 return sock; 481 } 482 } 483 484 /* 485 * \brief Closes the present (TCP and UDP) socket connection. 486 * 487 * This function sends a shutdown() on the socket in order to disable send() calls 488 * (while recv() ones are still allowed). Then, it closes the socket. 489 * 490 * \param sock: the socket identifier of the connection that has to be closed. 491 * 492 * \param errbuf: a pointer to an user-allocated buffer that will contain the complete 493 * error message. This buffer has to be at least 'errbuflen' in length. 494 * It can be NULL; in this case the error cannot be printed. 495 * 496 * \param errbuflen: length of the buffer that will contains the error. The error message cannot be 497 * larger than 'errbuflen - 1' because the last char is reserved for the string terminator. 498 * 499 * \return '0' if everything is fine, '-1' if some errors occurred. The error message is returned 500 * in the 'errbuf' variable. 501 */ 502 int sock_close(SOCKET sock, char *errbuf, int errbuflen) 503 { 504 /* 505 * SHUT_WR: subsequent calls to the send function are disallowed. 506 * For TCP sockets, a FIN will be sent after all data is sent and 507 * acknowledged by the Server. 508 */ 509 if (shutdown(sock, SHUT_WR)) 510 { 511 sock_geterror("shutdown(): ", errbuf, errbuflen); 512 /* close the socket anyway */ 513 closesocket(sock); 514 return -1; 515 } 516 517 closesocket(sock); 518 return 0; 519 } 520 521 /* 522 * \brief Checks that the address, port and flags given are valids and it returns an 'addrinfo' structure. 523 * 524 * This function basically calls the getaddrinfo() calls, and it performs a set of sanity checks 525 * to control that everything is fine (e.g. a TCP socket cannot have a mcast address, and such). 526 * If an error occurs, it writes the error message into 'errbuf'. 527 * 528 * \param host: a pointer to a string identifying the host. It can be 529 * a host name, a numeric literal address, or NULL or "" (useful 530 * in case of a server socket which has to bind to all addresses). 531 * 532 * \param port: a pointer to a user-allocated buffer containing the network port to use. 533 * 534 * \param hints: an addrinfo variable (passed by reference) containing the flags needed to create the 535 * addrinfo structure appropriately. 536 * 537 * \param addrinfo: it represents the true returning value. This is a pointer to an addrinfo variable 538 * (passed by reference), which will be allocated by this function and returned back to the caller. 539 * This variable will be used in the next sockets calls. 540 * 541 * \param errbuf: a pointer to an user-allocated buffer that will contain the complete 542 * error message. This buffer has to be at least 'errbuflen' in length. 543 * It can be NULL; in this case the error cannot be printed. 544 * 545 * \param errbuflen: length of the buffer that will contains the error. The error message cannot be 546 * larger than 'errbuflen - 1' because the last char is reserved for the string terminator. 547 * 548 * \return '0' if everything is fine, '-1' if some errors occurred. The error message is returned 549 * in the 'errbuf' variable. The addrinfo variable that has to be used in the following sockets calls is 550 * returned into the addrinfo parameter. 551 * 552 * \warning The 'addrinfo' variable has to be deleted by the programmer by calling freeaddrinfo() when 553 * it is no longer needed. 554 * 555 * \warning This function requires the 'hints' variable as parameter. The semantic of this variable is the same 556 * of the one of the corresponding variable used into the standard getaddrinfo() socket function. We suggest 557 * the programmer to look at that function in order to set the 'hints' variable appropriately. 558 */ 559 int sock_initaddress(const char *host, const char *port, 560 struct addrinfo *hints, struct addrinfo **addrinfo, char *errbuf, int errbuflen) 561 { 562 int retval; 563 564 retval = getaddrinfo(host, port, hints, addrinfo); 565 if (retval != 0) 566 { 567 /* 568 * if the getaddrinfo() fails, you have to use gai_strerror(), instead of using the standard 569 * error routines (errno) in UNIX; Winsock suggests using the GetLastError() instead. 570 */ 571 if (errbuf) 572 { 573 #ifdef _WIN32 574 sock_geterror("getaddrinfo(): ", errbuf, errbuflen); 575 #else 576 pcap_snprintf(errbuf, errbuflen, "getaddrinfo() %s", gai_strerror(retval)); 577 #endif 578 } 579 return -1; 580 } 581 /* 582 * \warning SOCKET: I should check all the accept() in order to bind to all addresses in case 583 * addrinfo has more han one pointers 584 */ 585 586 /* 587 * This software only supports PF_INET and PF_INET6. 588 * 589 * XXX - should we just check that at least *one* address is 590 * either PF_INET or PF_INET6, and, when using the list, 591 * ignore all addresses that are neither? (What, no IPX 592 * support? :-)) 593 */ 594 if (((*addrinfo)->ai_family != PF_INET) && 595 ((*addrinfo)->ai_family != PF_INET6)) 596 { 597 if (errbuf) 598 pcap_snprintf(errbuf, errbuflen, "getaddrinfo(): socket type not supported"); 599 freeaddrinfo(*addrinfo); 600 *addrinfo = NULL; 601 return -1; 602 } 603 604 /* 605 * You can't do multicast (or broadcast) TCP. 606 */ 607 if (((*addrinfo)->ai_socktype == SOCK_STREAM) && 608 (sock_ismcastaddr((*addrinfo)->ai_addr) == 0)) 609 { 610 if (errbuf) 611 pcap_snprintf(errbuf, errbuflen, "getaddrinfo(): multicast addresses are not valid when using TCP streams"); 612 freeaddrinfo(*addrinfo); 613 *addrinfo = NULL; 614 return -1; 615 } 616 617 return 0; 618 } 619 620 /* 621 * \brief It sends the amount of data contained into 'buffer' on the given socket. 622 * 623 * This function basically calls the send() socket function and it checks that all 624 * the data specified in 'buffer' (of size 'size') will be sent. If an error occurs, 625 * it writes the error message into 'errbuf'. 626 * In case the socket buffer does not have enough space, it loops until all data 627 * has been sent. 628 * 629 * \param socket: the connected socket currently opened. 630 * 631 * \param buffer: a char pointer to a user-allocated buffer in which data is contained. 632 * 633 * \param size: number of bytes that have to be sent. 634 * 635 * \param errbuf: a pointer to an user-allocated buffer that will contain the complete 636 * error message. This buffer has to be at least 'errbuflen' in length. 637 * It can be NULL; in this case the error cannot be printed. 638 * 639 * \param errbuflen: length of the buffer that will contains the error. The error message cannot be 640 * larger than 'errbuflen - 1' because the last char is reserved for the string terminator. 641 * 642 * \return '0' if everything is fine, '-1' if an error other than 643 * "connection reset" or "peer has closed the receive side" occurred, 644 * '-2' if we got one of those errors. 645 * For errors, an error message is returned in the 'errbuf' variable. 646 */ 647 int sock_send(SOCKET sock, const char *buffer, size_t size, 648 char *errbuf, int errbuflen) 649 { 650 int remaining; 651 ssize_t nsent; 652 653 if (size > INT_MAX) 654 { 655 if (errbuf) 656 { 657 pcap_snprintf(errbuf, errbuflen, 658 "Can't send more than %u bytes with sock_recv", 659 INT_MAX); 660 } 661 return -1; 662 } 663 remaining = (int)size; 664 665 do { 666 #ifdef MSG_NOSIGNAL 667 /* 668 * Send with MSG_NOSIGNAL, so that we don't get SIGPIPE 669 * on errors on stream-oriented sockets when the other 670 * end breaks the connection. 671 * The EPIPE error is still returned. 672 */ 673 nsent = send(sock, buffer, remaining, MSG_NOSIGNAL); 674 #else 675 nsent = send(sock, buffer, remaining, 0); 676 #endif 677 678 if (nsent == -1) 679 { 680 /* 681 * If the client closed the connection out from 682 * under us, there's no need to log that as an 683 * error. 684 */ 685 int errcode; 686 687 #ifdef _WIN32 688 errcode = GetLastError(); 689 if (errcode == WSAECONNRESET || 690 errcode == WSAECONNABORTED) 691 { 692 /* 693 * WSAECONNABORTED appears to be the error 694 * returned in Winsock when you try to send 695 * on a connection where the peer has closed 696 * the receive side. 697 */ 698 return -2; 699 } 700 sock_fmterror("send(): ", errcode, errbuf, errbuflen); 701 #else 702 errcode = errno; 703 if (errcode == ECONNRESET || errcode == EPIPE) 704 { 705 /* 706 * EPIPE is what's returned on UN*X when 707 * you try to send on a connection when 708 * the peer has closed the receive side. 709 */ 710 return -2; 711 } 712 sock_fmterror("send(): ", errcode, errbuf, errbuflen); 713 #endif 714 return -1; 715 } 716 717 remaining -= nsent; 718 buffer += nsent; 719 } while (remaining != 0); 720 721 return 0; 722 } 723 724 /* 725 * \brief It copies the amount of data contained into 'buffer' into 'tempbuf'. 726 * and it checks for buffer overflows. 727 * 728 * This function basically copies 'size' bytes of data contained into 'buffer' 729 * into 'tempbuf', starting at offset 'offset'. Before that, it checks that the 730 * resulting buffer will not be larger than 'totsize'. Finally, it updates 731 * the 'offset' variable in order to point to the first empty location of the buffer. 732 * 733 * In case the function is called with 'checkonly' equal to 1, it does not copy 734 * the data into the buffer. It only checks for buffer overflows and it updates the 735 * 'offset' variable. This mode can be useful when the buffer already contains the 736 * data (maybe because the producer writes directly into the target buffer), so 737 * only the buffer overflow check has to be made. 738 * In this case, both 'buffer' and 'tempbuf' can be NULL values. 739 * 740 * This function is useful in case the userland application does not know immediately 741 * all the data it has to write into the socket. This function provides a way to create 742 * the "stream" step by step, appending the new data to the old one. Then, when all the 743 * data has been bufferized, the application can call the sock_send() function. 744 * 745 * \param buffer: a char pointer to a user-allocated buffer that keeps the data 746 * that has to be copied. 747 * 748 * \param size: number of bytes that have to be copied. 749 * 750 * \param tempbuf: user-allocated buffer (of size 'totsize') in which data 751 * has to be copied. 752 * 753 * \param offset: an index into 'tempbuf' which keeps the location of its first 754 * empty location. 755 * 756 * \param totsize: total size of the buffer in which data is being copied. 757 * 758 * \param checkonly: '1' if we do not want to copy data into the buffer and we 759 * want just do a buffer ovreflow control, '0' if data has to be copied as well. 760 * 761 * \param errbuf: a pointer to an user-allocated buffer that will contain the complete 762 * error message. This buffer has to be at least 'errbuflen' in length. 763 * It can be NULL; in this case the error cannot be printed. 764 * 765 * \param errbuflen: length of the buffer that will contains the error. The error message cannot be 766 * larger than 'errbuflen - 1' because the last char is reserved for the string terminator. 767 * 768 * \return '0' if everything is fine, '-1' if some errors occurred. The error message 769 * is returned in the 'errbuf' variable. When the function returns, 'tempbuf' will 770 * have the new string appended, and 'offset' will keep the length of that buffer. 771 * In case of 'checkonly == 1', data is not copied, but 'offset' is updated in any case. 772 * 773 * \warning This function assumes that the buffer in which data has to be stored is 774 * large 'totbuf' bytes. 775 * 776 * \warning In case of 'checkonly', be carefully to call this function *before* copying 777 * the data into the buffer. Otherwise, the control about the buffer overflow is useless. 778 */ 779 int sock_bufferize(const char *buffer, int size, char *tempbuf, int *offset, int totsize, int checkonly, char *errbuf, int errbuflen) 780 { 781 if ((*offset + size) > totsize) 782 { 783 if (errbuf) 784 pcap_snprintf(errbuf, errbuflen, "Not enough space in the temporary send buffer."); 785 return -1; 786 } 787 788 if (!checkonly) 789 memcpy(tempbuf + (*offset), buffer, size); 790 791 (*offset) += size; 792 793 return 0; 794 } 795 796 /* 797 * \brief It waits on a connected socket and it manages to receive data. 798 * 799 * This function basically calls the recv() socket function and it checks that no 800 * error occurred. If that happens, it writes the error message into 'errbuf'. 801 * 802 * This function changes its behavior according to the 'receiveall' flag: if we 803 * want to receive exactly 'size' byte, it loops on the recv() until all the requested 804 * data is arrived. Otherwise, it returns the data currently available. 805 * 806 * In case the socket does not have enough data available, it cycles on the recv() 807 * until the requested data (of size 'size') is arrived. 808 * In this case, it blocks until the number of bytes read is equal to 'size'. 809 * 810 * \param sock: the connected socket currently opened. 811 * 812 * \param buffer: a char pointer to a user-allocated buffer in which data has to be stored 813 * 814 * \param size: size of the allocated buffer. WARNING: this indicates the number of bytes 815 * that we are expecting to be read. 816 * 817 * \param flags: 818 * 819 * SOCK_RECEIVALL_XXX: 820 * 821 * if SOCK_RECEIVEALL_NO, return as soon as some data is ready 822 * if SOCK_RECEIVALL_YES, wait until 'size' data has been 823 * received (in case the socket does not have enough data available). 824 * 825 * SOCK_EOF_XXX: 826 * 827 * if SOCK_EOF_ISNT_ERROR, if the first read returns 0, just return 0, 828 * and return an error on any subsequent read that returns 0; 829 * if SOCK_EOF_IS_ERROR, if any read returns 0, return an error. 830 * 831 * \param errbuf: a pointer to an user-allocated buffer that will contain the complete 832 * error message. This buffer has to be at least 'errbuflen' in length. 833 * It can be NULL; in this case the error cannot be printed. 834 * 835 * \param errbuflen: length of the buffer that will contains the error. The error message cannot be 836 * larger than 'errbuflen - 1' because the last char is reserved for the string terminator. 837 * 838 * \return the number of bytes read if everything is fine, '-1' if some errors occurred. 839 * The error message is returned in the 'errbuf' variable. 840 */ 841 842 int sock_recv(SOCKET sock, void *buffer, size_t size, int flags, 843 char *errbuf, int errbuflen) 844 { 845 char *bufp = buffer; 846 int remaining; 847 ssize_t nread; 848 849 if (size == 0) 850 { 851 SOCK_DEBUG_MESSAGE("I have been requested to read zero bytes"); 852 return 0; 853 } 854 if (size > INT_MAX) 855 { 856 if (errbuf) 857 { 858 pcap_snprintf(errbuf, errbuflen, 859 "Can't read more than %u bytes with sock_recv", 860 INT_MAX); 861 } 862 return -1; 863 } 864 865 bufp = (char *) buffer; 866 remaining = (int) size; 867 868 /* 869 * We don't use MSG_WAITALL because it's not supported in 870 * Win32. 871 */ 872 for (;;) { 873 nread = recv(sock, bufp, remaining, 0); 874 875 if (nread == -1) 876 { 877 #ifndef _WIN32 878 if (errno == EINTR) 879 return -3; 880 #endif 881 sock_geterror("recv(): ", errbuf, errbuflen); 882 return -1; 883 } 884 885 if (nread == 0) 886 { 887 if ((flags & SOCK_EOF_IS_ERROR) || 888 (remaining != (int) size)) 889 { 890 /* 891 * Either we've already read some data, 892 * or we're always supposed to return 893 * an error on EOF. 894 */ 895 if (errbuf) 896 { 897 pcap_snprintf(errbuf, errbuflen, 898 "The other host terminated the connection."); 899 } 900 return -1; 901 } 902 else 903 return 0; 904 } 905 906 /* 907 * Do we want to read the amount requested, or just return 908 * what we got? 909 */ 910 if (!(flags & SOCK_RECEIVEALL_YES)) 911 { 912 /* 913 * Just return what we got. 914 */ 915 return (int) nread; 916 } 917 918 bufp += nread; 919 remaining -= nread; 920 921 if (remaining == 0) 922 return (int) size; 923 } 924 } 925 926 /* 927 * Receives a datagram from a socket. 928 * 929 * Returns the size of the datagram on success or -1 on error. 930 */ 931 int sock_recv_dgram(SOCKET sock, void *buffer, size_t size, 932 char *errbuf, int errbuflen) 933 { 934 ssize_t nread; 935 #ifndef _WIN32 936 struct msghdr message; 937 struct iovec iov; 938 #endif 939 940 if (size == 0) 941 { 942 SOCK_DEBUG_MESSAGE("I have been requested to read zero bytes"); 943 return 0; 944 } 945 if (size > INT_MAX) 946 { 947 if (errbuf) 948 { 949 pcap_snprintf(errbuf, errbuflen, 950 "Can't read more than %u bytes with sock_recv_dgram", 951 INT_MAX); 952 } 953 return -1; 954 } 955 956 /* 957 * This should be a datagram socket, so we should get the 958 * entire datagram in one recv() or recvmsg() call, and 959 * don't need to loop. 960 */ 961 #ifdef _WIN32 962 nread = recv(sock, buffer, size, 0); 963 if (nread == SOCKET_ERROR) 964 { 965 /* 966 * To quote the MSDN documentation for recv(), 967 * "If the datagram or message is larger than 968 * the buffer specified, the buffer is filled 969 * with the first part of the datagram, and recv 970 * generates the error WSAEMSGSIZE. For unreliable 971 * protocols (for example, UDP) the excess data is 972 * lost..." 973 * 974 * So if the message is bigger than the buffer 975 * supplied to us, the excess data is discarded, 976 * and we'll report an error. 977 */ 978 sock_geterror("recv(): ", errbuf, errbuflen); 979 return -1; 980 } 981 #else /* _WIN32 */ 982 /* 983 * The Single UNIX Specification says that a recv() on 984 * a socket for a message-oriented protocol will discard 985 * the excess data. It does *not* indicate that the 986 * receive will fail with, for example, EMSGSIZE. 987 * 988 * Therefore, we use recvmsg(), which appears to be 989 * the only way to get a "message truncated" indication 990 * when receiving a message for a message-oriented 991 * protocol. 992 */ 993 message.msg_name = NULL; /* we don't care who it's from */ 994 message.msg_namelen = 0; 995 iov.iov_base = buffer; 996 iov.iov_len = size; 997 message.msg_iov = &iov; 998 message.msg_iovlen = 1; 999 #ifdef HAVE_STRUCT_MSGHDR_MSG_CONTROL 1000 message.msg_control = NULL; /* we don't care about control information */ 1001 message.msg_controllen = 0; 1002 #endif 1003 #ifdef HAVE_STRUCT_MSGHDR_MSG_FLAGS 1004 message.msg_flags = 0; 1005 #endif 1006 nread = recvmsg(sock, &message, 0); 1007 if (nread == -1) 1008 { 1009 if (errno == EINTR) 1010 return -3; 1011 sock_geterror("recv(): ", errbuf, errbuflen); 1012 return -1; 1013 } 1014 #ifdef HAVE_STRUCT_MSGHDR_MSG_FLAGS 1015 /* 1016 * XXX - Solaris supports this, but only if you ask for the 1017 * X/Open version of recvmsg(); should we use that, or will 1018 * that cause other problems? 1019 */ 1020 if (message.msg_flags & MSG_TRUNC) 1021 { 1022 /* 1023 * Message was bigger than the specified buffer size. 1024 * 1025 * Report this as an error, as the Microsoft documentation 1026 * implies we'd do in a similar case on Windows. 1027 */ 1028 pcap_snprintf(errbuf, errbuflen, "recv(): Message too long"); 1029 return -1; 1030 } 1031 #endif /* HAVE_STRUCT_MSGHDR_MSG_FLAGS */ 1032 #endif /* _WIN32 */ 1033 1034 /* 1035 * The size we're reading fits in an int, so the return value 1036 * will fit in an int. 1037 */ 1038 return (int)nread; 1039 } 1040 1041 /* 1042 * \brief It discards N bytes that are currently waiting to be read on the current socket. 1043 * 1044 * This function is useful in case we receive a message we cannot understand (e.g. 1045 * wrong version number when receiving a network packet), so that we have to discard all 1046 * data before reading a new message. 1047 * 1048 * This function will read 'size' bytes from the socket and discard them. 1049 * It defines an internal buffer in which data will be copied; however, in case 1050 * this buffer is not large enough, it will cycle in order to read everything as well. 1051 * 1052 * \param sock: the connected socket currently opened. 1053 * 1054 * \param size: number of bytes that have to be discarded. 1055 * 1056 * \param errbuf: a pointer to an user-allocated buffer that will contain the complete 1057 * error message. This buffer has to be at least 'errbuflen' in length. 1058 * It can be NULL; in this case the error cannot be printed. 1059 * 1060 * \param errbuflen: length of the buffer that will contains the error. The error message cannot be 1061 * larger than 'errbuflen - 1' because the last char is reserved for the string terminator. 1062 * 1063 * \return '0' if everything is fine, '-1' if some errors occurred. 1064 * The error message is returned in the 'errbuf' variable. 1065 */ 1066 int sock_discard(SOCKET sock, int size, char *errbuf, int errbuflen) 1067 { 1068 #define TEMP_BUF_SIZE 32768 1069 1070 char buffer[TEMP_BUF_SIZE]; /* network buffer, to be used when the message is discarded */ 1071 1072 /* 1073 * A static allocation avoids the need of a 'malloc()' each time we want to discard a message 1074 * Our feeling is that a buffer if 32KB is enough for most of the application; 1075 * in case this is not enough, the "while" loop discards the message by calling the 1076 * sockrecv() several times. 1077 * We do not want to create a bigger variable because this causes the program to exit on 1078 * some platforms (e.g. BSD) 1079 */ 1080 while (size > TEMP_BUF_SIZE) 1081 { 1082 if (sock_recv(sock, buffer, TEMP_BUF_SIZE, SOCK_RECEIVEALL_YES, errbuf, errbuflen) == -1) 1083 return -1; 1084 1085 size -= TEMP_BUF_SIZE; 1086 } 1087 1088 /* 1089 * If there is still data to be discarded 1090 * In this case, the data can fit into the temporary buffer 1091 */ 1092 if (size) 1093 { 1094 if (sock_recv(sock, buffer, size, SOCK_RECEIVEALL_YES, errbuf, errbuflen) == -1) 1095 return -1; 1096 } 1097 1098 SOCK_DEBUG_MESSAGE("I'm currently discarding data\n"); 1099 1100 return 0; 1101 } 1102 1103 /* 1104 * \brief Checks that one host (identified by the sockaddr_storage structure) belongs to an 'allowed list'. 1105 * 1106 * This function is useful after an accept() call in order to check if the connecting 1107 * host is allowed to connect to me. To do that, we have a buffer that keeps the list of the 1108 * allowed host; this function checks the sockaddr_storage structure of the connecting host 1109 * against this host list, and it returns '0' is the host is included in this list. 1110 * 1111 * \param hostlist: pointer to a string that contains the list of the allowed host. 1112 * 1113 * \param sep: a string that keeps the separators used between the hosts (for example the 1114 * space character) in the host list. 1115 * 1116 * \param from: a sockaddr_storage structure, as it is returned by the accept() call. 1117 * 1118 * \param errbuf: a pointer to an user-allocated buffer that will contain the complete 1119 * error message. This buffer has to be at least 'errbuflen' in length. 1120 * It can be NULL; in this case the error cannot be printed. 1121 * 1122 * \param errbuflen: length of the buffer that will contains the error. The error message cannot be 1123 * larger than 'errbuflen - 1' because the last char is reserved for the string terminator. 1124 * 1125 * \return It returns: 1126 * - '1' if the host list is empty 1127 * - '0' if the host belongs to the host list (and therefore it is allowed to connect) 1128 * - '-1' in case the host does not belong to the host list (and therefore it is not allowed to connect 1129 * - '-2' in case or error. The error message is returned in the 'errbuf' variable. 1130 */ 1131 int sock_check_hostlist(char *hostlist, const char *sep, struct sockaddr_storage *from, char *errbuf, int errbuflen) 1132 { 1133 /* checks if the connecting host is among the ones allowed */ 1134 if ((hostlist) && (hostlist[0])) 1135 { 1136 char *token; /* temp, needed to separate items into the hostlist */ 1137 struct addrinfo *addrinfo, *ai_next; 1138 char *temphostlist; 1139 char *lasts; 1140 1141 /* 1142 * The problem is that strtok modifies the original variable by putting '0' at the end of each token 1143 * So, we have to create a new temporary string in which the original content is kept 1144 */ 1145 temphostlist = strdup(hostlist); 1146 if (temphostlist == NULL) 1147 { 1148 sock_geterror("sock_check_hostlist(), malloc() failed", errbuf, errbuflen); 1149 return -2; 1150 } 1151 1152 token = pcap_strtok_r(temphostlist, sep, &lasts); 1153 1154 /* it avoids a warning in the compilation ('addrinfo used but not initialized') */ 1155 addrinfo = NULL; 1156 1157 while (token != NULL) 1158 { 1159 struct addrinfo hints; 1160 int retval; 1161 1162 addrinfo = NULL; 1163 memset(&hints, 0, sizeof(struct addrinfo)); 1164 hints.ai_family = PF_UNSPEC; 1165 hints.ai_socktype = SOCK_STREAM; 1166 1167 retval = getaddrinfo(token, "0", &hints, &addrinfo); 1168 if (retval != 0) 1169 { 1170 if (errbuf) 1171 pcap_snprintf(errbuf, errbuflen, "getaddrinfo() %s", gai_strerror(retval)); 1172 1173 SOCK_DEBUG_MESSAGE(errbuf); 1174 1175 /* Get next token */ 1176 token = pcap_strtok_r(NULL, sep, &lasts); 1177 continue; 1178 } 1179 1180 /* ai_next is required to preserve the content of addrinfo, in order to deallocate it properly */ 1181 ai_next = addrinfo; 1182 while (ai_next) 1183 { 1184 if (sock_cmpaddr(from, (struct sockaddr_storage *) ai_next->ai_addr) == 0) 1185 { 1186 free(temphostlist); 1187 freeaddrinfo(addrinfo); 1188 return 0; 1189 } 1190 1191 /* 1192 * If we are here, it means that the current address does not matches 1193 * Let's try with the next one in the header chain 1194 */ 1195 ai_next = ai_next->ai_next; 1196 } 1197 1198 freeaddrinfo(addrinfo); 1199 addrinfo = NULL; 1200 1201 /* Get next token */ 1202 token = pcap_strtok_r(NULL, sep, &lasts); 1203 } 1204 1205 if (addrinfo) 1206 { 1207 freeaddrinfo(addrinfo); 1208 addrinfo = NULL; 1209 } 1210 1211 if (errbuf) 1212 pcap_snprintf(errbuf, errbuflen, "The host is not in the allowed host list. Connection refused."); 1213 1214 free(temphostlist); 1215 return -1; 1216 } 1217 1218 /* No hostlist, so we have to return 'empty list' */ 1219 return 1; 1220 } 1221 1222 /* 1223 * \brief Compares two addresses contained into two sockaddr_storage structures. 1224 * 1225 * This function is useful to compare two addresses, given their internal representation, 1226 * i.e. an sockaddr_storage structure. 1227 * 1228 * The two structures do not need to be sockaddr_storage; you can have both 'sockaddr_in' and 1229 * sockaddr_in6, properly acsted in order to be compliant to the function interface. 1230 * 1231 * This function will return '0' if the two addresses matches, '-1' if not. 1232 * 1233 * \param first: a sockaddr_storage structure, (for example the one that is returned by an 1234 * accept() call), containing the first address to compare. 1235 * 1236 * \param second: a sockaddr_storage structure containing the second address to compare. 1237 * 1238 * \return '0' if the addresses are equal, '-1' if they are different. 1239 */ 1240 int sock_cmpaddr(struct sockaddr_storage *first, struct sockaddr_storage *second) 1241 { 1242 if (first->ss_family == second->ss_family) 1243 { 1244 if (first->ss_family == AF_INET) 1245 { 1246 if (memcmp(&(((struct sockaddr_in *) first)->sin_addr), 1247 &(((struct sockaddr_in *) second)->sin_addr), 1248 sizeof(struct in_addr)) == 0) 1249 return 0; 1250 } 1251 else /* address family is AF_INET6 */ 1252 { 1253 if (memcmp(&(((struct sockaddr_in6 *) first)->sin6_addr), 1254 &(((struct sockaddr_in6 *) second)->sin6_addr), 1255 sizeof(struct in6_addr)) == 0) 1256 return 0; 1257 } 1258 } 1259 1260 return -1; 1261 } 1262 1263 /* 1264 * \brief It gets the address/port the system picked for this socket (on connected sockets). 1265 * 1266 * It is used to return the address and port the server picked for our socket on the local machine. 1267 * It works only on: 1268 * - connected sockets 1269 * - server sockets 1270 * 1271 * On unconnected client sockets it does not work because the system dynamically chooses a port 1272 * only when the socket calls a send() call. 1273 * 1274 * \param sock: the connected socket currently opened. 1275 * 1276 * \param address: it contains the address that will be returned by the function. This buffer 1277 * must be properly allocated by the user. The address can be either literal or numeric depending 1278 * on the value of 'Flags'. 1279 * 1280 * \param addrlen: the length of the 'address' buffer. 1281 * 1282 * \param port: it contains the port that will be returned by the function. This buffer 1283 * must be properly allocated by the user. 1284 * 1285 * \param portlen: the length of the 'port' buffer. 1286 * 1287 * \param flags: a set of flags (the ones defined into the getnameinfo() standard socket function) 1288 * that determine if the resulting address must be in numeric / literal form, and so on. 1289 * 1290 * \param errbuf: a pointer to an user-allocated buffer that will contain the complete 1291 * error message. This buffer has to be at least 'errbuflen' in length. 1292 * It can be NULL; in this case the error cannot be printed. 1293 * 1294 * \param errbuflen: length of the buffer that will contains the error. The error message cannot be 1295 * larger than 'errbuflen - 1' because the last char is reserved for the string terminator. 1296 * 1297 * \return It returns '-1' if this function succeeds, '0' otherwise. 1298 * The address and port corresponding are returned back in the buffers 'address' and 'port'. 1299 * In any case, the returned strings are '0' terminated. 1300 * 1301 * \warning If the socket is using a connectionless protocol, the address may not be available 1302 * until I/O occurs on the socket. 1303 */ 1304 int sock_getmyinfo(SOCKET sock, char *address, int addrlen, char *port, int portlen, int flags, char *errbuf, int errbuflen) 1305 { 1306 struct sockaddr_storage mysockaddr; 1307 socklen_t sockaddrlen; 1308 1309 1310 sockaddrlen = sizeof(struct sockaddr_storage); 1311 1312 if (getsockname(sock, (struct sockaddr *) &mysockaddr, &sockaddrlen) == -1) 1313 { 1314 sock_geterror("getsockname(): ", errbuf, errbuflen); 1315 return 0; 1316 } 1317 1318 /* Returns the numeric address of the host that triggered the error */ 1319 return sock_getascii_addrport(&mysockaddr, address, addrlen, port, portlen, flags, errbuf, errbuflen); 1320 } 1321 1322 /* 1323 * \brief It retrieves two strings containing the address and the port of a given 'sockaddr' variable. 1324 * 1325 * This function is basically an extended version of the inet_ntop(), which does not exist in 1326 * Winsock because the same result can be obtained by using the getnameinfo(). 1327 * However, differently from inet_ntop(), this function is able to return also literal names 1328 * (e.g. 'localhost') dependently from the 'Flags' parameter. 1329 * 1330 * The function accepts a sockaddr_storage variable (which can be returned by several functions 1331 * like bind(), connect(), accept(), and more) and it transforms its content into a 'human' 1332 * form. So, for instance, it is able to translate an hex address (stored in binary form) into 1333 * a standard IPv6 address like "::1". 1334 * 1335 * The behavior of this function depends on the parameters we have in the 'Flags' variable, which 1336 * are the ones allowed in the standard getnameinfo() socket function. 1337 * 1338 * \param sockaddr: a 'sockaddr_in' or 'sockaddr_in6' structure containing the address that 1339 * need to be translated from network form into the presentation form. This structure must be 1340 * zero-ed prior using it, and the address family field must be filled with the proper value. 1341 * The user must cast any 'sockaddr_in' or 'sockaddr_in6' structures to 'sockaddr_storage' before 1342 * calling this function. 1343 * 1344 * \param address: it contains the address that will be returned by the function. This buffer 1345 * must be properly allocated by the user. The address can be either literal or numeric depending 1346 * on the value of 'Flags'. 1347 * 1348 * \param addrlen: the length of the 'address' buffer. 1349 * 1350 * \param port: it contains the port that will be returned by the function. This buffer 1351 * must be properly allocated by the user. 1352 * 1353 * \param portlen: the length of the 'port' buffer. 1354 * 1355 * \param flags: a set of flags (the ones defined into the getnameinfo() standard socket function) 1356 * that determine if the resulting address must be in numeric / literal form, and so on. 1357 * 1358 * \param errbuf: a pointer to an user-allocated buffer that will contain the complete 1359 * error message. This buffer has to be at least 'errbuflen' in length. 1360 * It can be NULL; in this case the error cannot be printed. 1361 * 1362 * \param errbuflen: length of the buffer that will contains the error. The error message cannot be 1363 * larger than 'errbuflen - 1' because the last char is reserved for the string terminator. 1364 * 1365 * \return It returns '-1' if this function succeeds, '0' otherwise. 1366 * The address and port corresponding to the given SockAddr are returned back in the buffers 'address' 1367 * and 'port'. 1368 * In any case, the returned strings are '0' terminated. 1369 */ 1370 int sock_getascii_addrport(const struct sockaddr_storage *sockaddr, char *address, int addrlen, char *port, int portlen, int flags, char *errbuf, int errbuflen) 1371 { 1372 socklen_t sockaddrlen; 1373 int retval; /* Variable that keeps the return value; */ 1374 1375 retval = -1; 1376 1377 #ifdef _WIN32 1378 if (sockaddr->ss_family == AF_INET) 1379 sockaddrlen = sizeof(struct sockaddr_in); 1380 else 1381 sockaddrlen = sizeof(struct sockaddr_in6); 1382 #else 1383 sockaddrlen = sizeof(struct sockaddr_storage); 1384 #endif 1385 1386 if ((flags & NI_NUMERICHOST) == 0) /* Check that we want literal names */ 1387 { 1388 if ((sockaddr->ss_family == AF_INET6) && 1389 (memcmp(&((struct sockaddr_in6 *) sockaddr)->sin6_addr, "\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0", sizeof(struct in6_addr)) == 0)) 1390 { 1391 if (address) 1392 strlcpy(address, SOCKET_NAME_NULL_DAD, addrlen); 1393 return retval; 1394 } 1395 } 1396 1397 if (getnameinfo((struct sockaddr *) sockaddr, sockaddrlen, address, addrlen, port, portlen, flags) != 0) 1398 { 1399 /* If the user wants to receive an error message */ 1400 if (errbuf) 1401 { 1402 sock_geterror("getnameinfo(): ", errbuf, errbuflen); 1403 errbuf[errbuflen - 1] = 0; 1404 } 1405 1406 if (address) 1407 { 1408 strlcpy(address, SOCKET_NO_NAME_AVAILABLE, addrlen); 1409 address[addrlen - 1] = 0; 1410 } 1411 1412 if (port) 1413 { 1414 strlcpy(port, SOCKET_NO_PORT_AVAILABLE, portlen); 1415 port[portlen - 1] = 0; 1416 } 1417 1418 retval = 0; 1419 } 1420 1421 return retval; 1422 } 1423 1424 /* 1425 * \brief It translates an address from the 'presentation' form into the 'network' form. 1426 * 1427 * This function basically replaces inet_pton(), which does not exist in Winsock because 1428 * the same result can be obtained by using the getaddrinfo(). 1429 * An additional advantage is that 'Address' can be both a numeric address (e.g. '127.0.0.1', 1430 * like in inet_pton() ) and a literal name (e.g. 'localhost'). 1431 * 1432 * This function does the reverse job of sock_getascii_addrport(). 1433 * 1434 * \param address: a zero-terminated string which contains the name you have to 1435 * translate. The name can be either literal (e.g. 'localhost') or numeric (e.g. '::1'). 1436 * 1437 * \param sockaddr: a user-allocated sockaddr_storage structure which will contains the 1438 * 'network' form of the requested address. 1439 * 1440 * \param addr_family: a constant which can assume the following values: 1441 * - 'AF_INET' if we want to ping an IPv4 host 1442 * - 'AF_INET6' if we want to ping an IPv6 host 1443 * - 'AF_UNSPEC' if we do not have preferences about the protocol used to ping the host 1444 * 1445 * \param errbuf: a pointer to an user-allocated buffer that will contain the complete 1446 * error message. This buffer has to be at least 'errbuflen' in length. 1447 * It can be NULL; in this case the error cannot be printed. 1448 * 1449 * \param errbuflen: length of the buffer that will contains the error. The error message cannot be 1450 * larger than 'errbuflen - 1' because the last char is reserved for the string terminator. 1451 * 1452 * \return '-1' if the translation succeeded, '-2' if there was some non critical error, '0' 1453 * otherwise. In case it fails, the content of the SockAddr variable remains unchanged. 1454 * A 'non critical error' can occur in case the 'Address' is a literal name, which can be mapped 1455 * to several network addresses (e.g. 'foo.bar.com' => '10.2.2.2' and '10.2.2.3'). In this case 1456 * the content of the SockAddr parameter will be the address corresponding to the first mapping. 1457 * 1458 * \warning The sockaddr_storage structure MUST be allocated by the user. 1459 */ 1460 int sock_present2network(const char *address, struct sockaddr_storage *sockaddr, int addr_family, char *errbuf, int errbuflen) 1461 { 1462 int retval; 1463 struct addrinfo *addrinfo; 1464 struct addrinfo hints; 1465 1466 memset(&hints, 0, sizeof(hints)); 1467 1468 hints.ai_family = addr_family; 1469 1470 if ((retval = sock_initaddress(address, "22222" /* fake port */, &hints, &addrinfo, errbuf, errbuflen)) == -1) 1471 return 0; 1472 1473 if (addrinfo->ai_family == PF_INET) 1474 memcpy(sockaddr, addrinfo->ai_addr, sizeof(struct sockaddr_in)); 1475 else 1476 memcpy(sockaddr, addrinfo->ai_addr, sizeof(struct sockaddr_in6)); 1477 1478 if (addrinfo->ai_next != NULL) 1479 { 1480 freeaddrinfo(addrinfo); 1481 1482 if (errbuf) 1483 pcap_snprintf(errbuf, errbuflen, "More than one socket requested; using the first one returned"); 1484 return -2; 1485 } 1486 1487 freeaddrinfo(addrinfo); 1488 return -1; 1489 } 1490