2 * Copyright (c) 2002 - 2003
3 * NetGroup, Politecnico di Torino (Italy)
6 * Redistribution and use in source and binary forms, with or without
7 * modification, are permitted provided that the following conditions
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.
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.
40 * The goal of this file is to provide a common set of primitives for socket
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.
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
56 #include <errno.h> /* for the errno variable */
57 #include <stdio.h> /* for the stderr file */
58 #include <stdlib.h> /* for malloc() and free() */
62 #define INT_MAX 2147483647
67 #include "sockutils.h"
68 #include "portability.h"
72 * Winsock initialization.
74 * Ask for WinSock 2.2.
76 #define WINSOCK_MAJOR_VERSION 2
77 #define WINSOCK_MINOR_VERSION 2
79 static int sockcount
= 0; /*!< Variable that allows calling the WSAStartup() only one time */
82 /* Some minor differences between UNIX and Win32 */
84 #define SHUT_WR SD_SEND /* The control code for shutdown() is different in Win32 */
87 /* Size of the buffer that has to keep error messages */
88 #define SOCK_ERRBUF_SIZE 1024
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)"
96 * On UN*X, send() and recv() return ssize_t.
98 * On Windows, send() and recv() return an int.
100 * Wth MSVC, there *is* no ssize_t.
102 * With MinGW, there is an ssize_t type; it is either an int (32 bit)
103 * or a long long (64 bit).
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().
109 #if defined(_WIN32) && !defined(_SSIZE_T_DEFINED)
113 /****************************************************
115 * Locally defined functions *
117 ****************************************************/
119 static int sock_ismcastaddr(const struct sockaddr
*saddr
);
121 /****************************************************
125 ****************************************************/
128 * Format an error message given an errno value (UN*X) or a WinSock error
131 void sock_fmterror(const char *caller
, int errcode
, char *errbuf
, int errbuflen
)
135 TCHAR message
[SOCK_ERRBUF_SIZE
]; /* It will be char (if we're using ascii) or wchar_t (if we're using unicode) */
140 retval
= FormatMessage(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
);
147 if ((caller
) && (*caller
))
148 pcap_snprintf(errbuf
, errbuflen
, "%sUnable to get the exact error message", caller
);
150 pcap_snprintf(errbuf
, errbuflen
, "Unable to get the exact error message");
154 if ((caller
) && (*caller
))
155 pcap_snprintf(errbuf
, errbuflen
, "%s%s (code %d)", caller
, message
, errcode
);
157 pcap_snprintf(errbuf
, errbuflen
, "%s (code %d)", message
, errcode
);
165 message
= strerror(errcode
);
167 if ((caller
) && (*caller
))
168 pcap_snprintf(errbuf
, errbuflen
, "%s%s (code %d)", caller
, message
, errcode
);
170 pcap_snprintf(errbuf
, errbuflen
, "%s (code %d)", message
, errcode
);
175 * \brief It retrieves the error message after an error occurred in the socket interface.
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.
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.
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.
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.
192 * \return No return values. The error message is returned in the 'string' parameter.
194 void sock_geterror(const char *caller
, char *errbuf
, int errbuflen
)
199 sock_fmterror(caller
, GetLastError(), errbuf
, errbuflen
);
203 sock_fmterror(caller
, errno
, errbuf
, errbuflen
);
208 * \brief It initializes sockets.
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.
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.
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.
220 * \return '0' if everything is fine, '-1' if some errors occurred. The error message is returned
221 * in the 'errbuf' variable.
224 int sock_init(char *errbuf
, int errbuflen
)
228 WSADATA wsaData
; /* helper variable needed to initialize Winsock */
230 if (WSAStartup(MAKEWORD(WINSOCK_MAJOR_VERSION
,
231 WINSOCK_MINOR_VERSION
), &wsaData
) != 0)
234 pcap_snprintf(errbuf
, errbuflen
, "Failed to initialize Winsock\n");
244 int sock_init(char *errbuf _U_
, int errbuflen _U_
)
251 * \brief It deallocates sockets.
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.
256 * \return No error values.
258 void sock_cleanup(void)
269 * \brief It checks if the sockaddr variable contains a multicast address.
271 * \return '0' if the address is multicast, '-1' if it is not.
273 static int sock_ismcastaddr(const struct sockaddr
*saddr
)
275 if (saddr
->sa_family
== PF_INET
)
277 struct sockaddr_in
*saddr4
= (struct sockaddr_in
*) saddr
;
278 if (IN_MULTICAST(ntohl(saddr4
->sin_addr
.s_addr
))) return 0;
283 struct sockaddr_in6
*saddr6
= (struct sockaddr_in6
*) saddr
;
284 if (IN6_IS_ADDR_MULTICAST(&saddr6
->sin6_addr
)) return 0;
290 * \brief It initializes a network connection both from the client and the server side.
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'.
296 * In case of a server socket, the function calls socket(), bind() and listen().
298 * This function is usually preceeded by the sock_initaddress().
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().
304 * \param server: '1' if this is a server socket, '0' otherwise.
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.
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.
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.
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.
320 SOCKET
sock_open(struct addrinfo
*addrinfo
, int server
, int nconn
, char *errbuf
, int errbuflen
)
323 #if defined(SO_NOSIGPIPE) || defined(IPV6_V6ONLY) || defined(IPV6_BINDV6ONLY)
327 sock
= socket(addrinfo
->ai_family
, addrinfo
->ai_socktype
, addrinfo
->ai_protocol
);
328 if (sock
== INVALID_SOCKET
)
330 sock_geterror("socket(): ", errbuf
, errbuflen
);
331 return INVALID_SOCKET
;
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.
341 if (setsockopt(sock
, SOL_SOCKET
, SO_NOSIGPIPE
, (char *)&on
,
344 sock_geterror("setsockopt(SO_NOSIGPIPE)", errbuf
, errbuflen
);
346 return INVALID_SOCKET
;
350 /* This is a server socket */
353 #if defined(IPV6_V6ONLY) || defined(IPV6_BINDV6ONLY)
355 * Force the use of IPv6-only addresses.
357 * RFC 3493 indicates that you can support IPv4 on an
360 * https://tools.ietf.org/html/rfc3493#section-3.7
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
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.
375 * So we try to disable it, using either the IPV6_V6ONLY
376 * option from RFC 3493:
378 * https://tools.ietf.org/html/rfc3493#section-5.3
380 * or the IPV6_BINDV6ONLY option from older UN*Xes.
383 /* For older systems */
384 #define IPV6_V6ONLY IPV6_BINDV6ONLY
385 #endif /* IPV6_V6ONLY */
386 if (addrinfo
->ai_family
== PF_INET6
)
388 if (setsockopt(sock
, IPPROTO_IPV6
, IPV6_V6ONLY
,
389 (char *)&on
, sizeof (int)) == -1)
392 pcap_snprintf(errbuf
, errbuflen
, "setsockopt(IPV6_V6ONLY)");
394 return INVALID_SOCKET
;
397 #endif /* defined(IPV6_V6ONLY) || defined(IPV6_BINDV6ONLY) */
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)
402 sock_geterror("bind(): ", errbuf
, errbuflen
);
404 return INVALID_SOCKET
;
407 if (addrinfo
->ai_socktype
== SOCK_STREAM
)
408 if (listen(sock
, nconn
) == -1)
410 sock_geterror("listen(): ", errbuf
, errbuflen
);
412 return INVALID_SOCKET
;
415 /* server side ended */
418 else /* we're the client */
420 struct addrinfo
*tempaddrinfo
;
424 tempaddrinfo
= addrinfo
;
426 bufspaceleft
= errbuflen
;
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
437 if (connect(sock
, tempaddrinfo
->ai_addr
, (int) tempaddrinfo
->ai_addrlen
) == -1)
441 char SocketErrorMessage
[SOCK_ERRBUF_SIZE
];
444 * We have to retrieve the error message before any other socket call completes, otherwise
445 * the error message is lost
447 sock_geterror(NULL
, SocketErrorMessage
, sizeof(SocketErrorMessage
));
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
));
452 pcap_snprintf(errbufptr
, bufspaceleft
,
453 "Is the server properly installed on %s? connect() failed: %s", TmpBuffer
, SocketErrorMessage
);
455 /* In case more then one 'connect' fails, we manage to keep all the error messages */
456 msglen
= strlen(errbufptr
);
458 errbufptr
[msglen
] = ' ';
459 errbufptr
[msglen
+ 1] = 0;
461 bufspaceleft
= bufspaceleft
- (msglen
+ 1);
462 errbufptr
+= (msglen
+ 1);
464 tempaddrinfo
= tempaddrinfo
->ai_next
;
471 * Check how we exit from the previous loop
472 * If tempaddrinfo is equal to NULL, it means that all the connect() failed.
474 if (tempaddrinfo
== NULL
)
477 return INVALID_SOCKET
;
485 * \brief Closes the present (TCP and UDP) socket connection.
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.
490 * \param sock: the socket identifier of the connection that has to be closed.
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.
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.
499 * \return '0' if everything is fine, '-1' if some errors occurred. The error message is returned
500 * in the 'errbuf' variable.
502 int sock_close(SOCKET sock
, char *errbuf
, int errbuflen
)
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.
509 if (shutdown(sock
, SHUT_WR
))
511 sock_geterror("shutdown(): ", errbuf
, errbuflen
);
512 /* close the socket anyway */
522 * \brief Checks that the address, port and flags given are valids and it returns an 'addrinfo' structure.
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'.
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).
532 * \param port: a pointer to a user-allocated buffer containing the network port to use.
534 * \param hints: an addrinfo variable (passed by reference) containing the flags needed to create the
535 * addrinfo structure appropriately.
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.
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.
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.
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.
552 * \warning The 'addrinfo' variable has to be deleted by the programmer by calling freeaddrinfo() when
553 * it is no longer needed.
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.
559 int sock_initaddress(const char *host
, const char *port
,
560 struct addrinfo
*hints
, struct addrinfo
**addrinfo
, char *errbuf
, int errbuflen
)
564 retval
= getaddrinfo(host
, port
, hints
, addrinfo
);
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.
574 sock_geterror("getaddrinfo(): ", errbuf
, errbuflen
);
576 pcap_snprintf(errbuf
, errbuflen
, "getaddrinfo() %s", gai_strerror(retval
));
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
587 * This software only supports PF_INET and PF_INET6.
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
594 if (((*addrinfo
)->ai_family
!= PF_INET
) &&
595 ((*addrinfo
)->ai_family
!= PF_INET6
))
598 pcap_snprintf(errbuf
, errbuflen
, "getaddrinfo(): socket type not supported");
599 freeaddrinfo(*addrinfo
);
605 * You can't do multicast (or broadcast) TCP.
607 if (((*addrinfo
)->ai_socktype
== SOCK_STREAM
) &&
608 (sock_ismcastaddr((*addrinfo
)->ai_addr
) == 0))
611 pcap_snprintf(errbuf
, errbuflen
, "getaddrinfo(): multicast addresses are not valid when using TCP streams");
612 freeaddrinfo(*addrinfo
);
621 * \brief It sends the amount of data contained into 'buffer' on the given socket.
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
629 * \param socket: the connected socket currently opened.
631 * \param buffer: a char pointer to a user-allocated buffer in which data is contained.
633 * \param size: number of bytes that have to be sent.
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.
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.
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.
647 int sock_send(SOCKET sock
, const char *buffer
, size_t size
,
648 char *errbuf
, int errbuflen
)
657 pcap_snprintf(errbuf
, errbuflen
,
658 "Can't send more than %u bytes with sock_recv",
663 remaining
= (int)size
;
668 * Another pain... in Linux there's this flag
670 * Requests not to send SIGPIPE on errors on stream-oriented
671 * sockets when the other end breaks the connection.
672 * The EPIPE error is still returned.
674 nsent
= send(sock
, buffer
, remaining
, MSG_NOSIGNAL
);
676 nsent
= send(sock
, buffer
, remaining
, 0);
682 * If the client closed the connection out from
683 * under us, there's no need to log that as an
689 errcode
= GetLastError();
690 if (errcode
== WSAECONNRESET
||
691 errcode
== WSAECONNABORTED
)
694 * WSAECONNABORTED appears to be the error
695 * returned in Winsock when you try to send
696 * on a connection where the peer has closed
701 sock_fmterror("send(): ", errcode
, errbuf
, errbuflen
);
704 if (errcode
== ECONNRESET
|| errcode
== EPIPE
)
707 * EPIPE is what's returned on UN*X when
708 * you try to send on a connection when
709 * the peer has closed the receive side.
713 sock_fmterror("send(): ", errcode
, errbuf
, errbuflen
);
720 } while (remaining
!= 0);
726 * \brief It copies the amount of data contained into 'buffer' into 'tempbuf'.
727 * and it checks for buffer overflows.
729 * This function basically copies 'size' bytes of data contained into 'buffer'
730 * into 'tempbuf', starting at offset 'offset'. Before that, it checks that the
731 * resulting buffer will not be larger than 'totsize'. Finally, it updates
732 * the 'offset' variable in order to point to the first empty location of the buffer.
734 * In case the function is called with 'checkonly' equal to 1, it does not copy
735 * the data into the buffer. It only checks for buffer overflows and it updates the
736 * 'offset' variable. This mode can be useful when the buffer already contains the
737 * data (maybe because the producer writes directly into the target buffer), so
738 * only the buffer overflow check has to be made.
739 * In this case, both 'buffer' and 'tempbuf' can be NULL values.
741 * This function is useful in case the userland application does not know immediately
742 * all the data it has to write into the socket. This function provides a way to create
743 * the "stream" step by step, appending the new data to the old one. Then, when all the
744 * data has been bufferized, the application can call the sock_send() function.
746 * \param buffer: a char pointer to a user-allocated buffer that keeps the data
747 * that has to be copied.
749 * \param size: number of bytes that have to be copied.
751 * \param tempbuf: user-allocated buffer (of size 'totsize') in which data
754 * \param offset: an index into 'tempbuf' which keeps the location of its first
757 * \param totsize: total size of the buffer in which data is being copied.
759 * \param checkonly: '1' if we do not want to copy data into the buffer and we
760 * want just do a buffer ovreflow control, '0' if data has to be copied as well.
762 * \param errbuf: a pointer to an user-allocated buffer that will contain the complete
763 * error message. This buffer has to be at least 'errbuflen' in length.
764 * It can be NULL; in this case the error cannot be printed.
766 * \param errbuflen: length of the buffer that will contains the error. The error message cannot be
767 * larger than 'errbuflen - 1' because the last char is reserved for the string terminator.
769 * \return '0' if everything is fine, '-1' if some errors occurred. The error message
770 * is returned in the 'errbuf' variable. When the function returns, 'tempbuf' will
771 * have the new string appended, and 'offset' will keep the length of that buffer.
772 * In case of 'checkonly == 1', data is not copied, but 'offset' is updated in any case.
774 * \warning This function assumes that the buffer in which data has to be stored is
775 * large 'totbuf' bytes.
777 * \warning In case of 'checkonly', be carefully to call this function *before* copying
778 * the data into the buffer. Otherwise, the control about the buffer overflow is useless.
780 int sock_bufferize(const char *buffer
, int size
, char *tempbuf
, int *offset
, int totsize
, int checkonly
, char *errbuf
, int errbuflen
)
782 if ((*offset
+ size
) > totsize
)
785 pcap_snprintf(errbuf
, errbuflen
, "Not enough space in the temporary send buffer.");
790 memcpy(tempbuf
+ (*offset
), buffer
, size
);
798 * \brief It waits on a connected socket and it manages to receive data.
800 * This function basically calls the recv() socket function and it checks that no
801 * error occurred. If that happens, it writes the error message into 'errbuf'.
803 * This function changes its behavior according to the 'receiveall' flag: if we
804 * want to receive exactly 'size' byte, it loops on the recv() until all the requested
805 * data is arrived. Otherwise, it returns the data currently available.
807 * In case the socket does not have enough data available, it cycles on the recv()
808 * until the requested data (of size 'size') is arrived.
809 * In this case, it blocks until the number of bytes read is equal to 'size'.
811 * \param sock: the connected socket currently opened.
813 * \param buffer: a char pointer to a user-allocated buffer in which data has to be stored
815 * \param size: size of the allocated buffer. WARNING: this indicates the number of bytes
816 * that we are expecting to be read.
820 * SOCK_RECEIVALL_XXX:
822 * if SOCK_RECEIVEALL_NO, return as soon as some data is ready
823 * if SOCK_RECEIVALL_YES, wait until 'size' data has been
824 * received (in case the socket does not have enough data available).
828 * if SOCK_EOF_ISNT_ERROR, if the first read returns 0, just return 0,
829 * and return an error on any subsequent read that returns 0;
830 * if SOCK_EOF_IS_ERROR, if any read returns 0, return an error.
832 * \param errbuf: a pointer to an user-allocated buffer that will contain the complete
833 * error message. This buffer has to be at least 'errbuflen' in length.
834 * It can be NULL; in this case the error cannot be printed.
836 * \param errbuflen: length of the buffer that will contains the error. The error message cannot be
837 * larger than 'errbuflen - 1' because the last char is reserved for the string terminator.
839 * \return the number of bytes read if everything is fine, '-1' if some errors occurred.
840 * The error message is returned in the 'errbuf' variable.
843 int sock_recv(SOCKET sock
, void *buffer
, size_t size
, int flags
,
844 char *errbuf
, int errbuflen
)
852 SOCK_ASSERT("I have been requested to read zero bytes", 1);
859 pcap_snprintf(errbuf
, errbuflen
,
860 "Can't read more than %u bytes with sock_recv",
866 bufp
= (char *) buffer
;
867 remaining
= (int) size
;
870 * We don't use MSG_WAITALL because it's not supported in
874 nread
= recv(sock
, bufp
, remaining
, 0);
882 sock_geterror("recv(): ", errbuf
, errbuflen
);
888 if ((flags
& SOCK_EOF_IS_ERROR
) ||
889 (remaining
!= (int) size
))
892 * Either we've already read some data,
893 * or we're always supposed to return
898 pcap_snprintf(errbuf
, errbuflen
,
899 "The other host terminated the connection.");
908 * Do we want to read the amount requested, or just return
911 if (!(flags
& SOCK_RECEIVEALL_YES
))
914 * Just return what we got.
928 * Receives a datagram from a socket.
930 * Returns the size of the datagram on success or -1 on error.
932 int sock_recv_dgram(SOCKET sock
, void *buffer
, size_t size
,
933 char *errbuf
, int errbuflen
)
937 struct msghdr message
;
943 SOCK_ASSERT("I have been requested to read zero bytes", 1);
950 pcap_snprintf(errbuf
, errbuflen
,
951 "Can't read more than %u bytes with sock_recv_dgram",
958 * This should be a datagram socket, so we should get the
959 * entire datagram in one recv() or recvmsg() call, and
960 * don't need to loop.
963 nread
= recv(sock
, buffer
, size
, 0);
964 if (nread
== SOCKET_ERROR
)
967 * To quote the MSDN documentation for recv(),
968 * "If the datagram or message is larger than
969 * the buffer specified, the buffer is filled
970 * with the first part of the datagram, and recv
971 * generates the error WSAEMSGSIZE. For unreliable
972 * protocols (for example, UDP) the excess data is
975 * So if the message is bigger than the buffer
976 * supplied to us, the excess data is discarded,
977 * and we'll report an error.
979 sock_geterror("recv(): ", errbuf
, errbuflen
);
984 * The Single UNIX Specification says that a recv() on
985 * a socket for a message-oriented protocol will discard
986 * the excess data. It does *not* indicate that the
987 * receive will fail with, for example, EMSGSIZE.
989 * Therefore, we use recvmsg(), which appears to be
990 * the only way to get a "message truncated" indication
991 * when receiving a message for a message-oriented
994 message
.msg_name
= NULL
; /* we don't care who it's from */
995 message
.msg_namelen
= 0;
996 iov
.iov_base
= buffer
;
998 message
.msg_iov
= &iov
;
999 message
.msg_iovlen
= 1;
1000 #ifdef HAVE_STRUCT_MSGHDR_MSG_CONTROL
1001 message
.msg_control
= NULL
; /* we don't care about control information */
1002 message
.msg_controllen
= 0;
1004 #ifdef HAVE_STRUCT_MSGHDR_MSG_FLAGS
1005 message
.msg_flags
= 0;
1007 nread
= recvmsg(sock
, &message
, 0);
1012 sock_geterror("recv(): ", errbuf
, errbuflen
);
1015 #ifdef HAVE_STRUCT_MSGHDR_MSG_FLAGS
1017 * XXX - Solaris supports this, but only if you ask for the
1018 * X/Open version of recvmsg(); should we use that, or will
1019 * that cause other problems?
1021 if (message
.msg_flags
& MSG_TRUNC
)
1024 * Message was bigger than the specified buffer size.
1026 * Report this as an error, as the Microsoft documentation
1027 * implies we'd do in a similar case on Windows.
1029 pcap_snprintf(errbuf
, errbuflen
, "recv(): Message too long");
1032 #endif /* HAVE_STRUCT_MSGHDR_MSG_FLAGS */
1036 * The size we're reading fits in an int, so the return value
1037 * will fit in an int.
1043 * \brief It discards N bytes that are currently waiting to be read on the current socket.
1045 * This function is useful in case we receive a message we cannot understand (e.g.
1046 * wrong version number when receiving a network packet), so that we have to discard all
1047 * data before reading a new message.
1049 * This function will read 'size' bytes from the socket and discard them.
1050 * It defines an internal buffer in which data will be copied; however, in case
1051 * this buffer is not large enough, it will cycle in order to read everything as well.
1053 * \param sock: the connected socket currently opened.
1055 * \param size: number of bytes that have to be discarded.
1057 * \param errbuf: a pointer to an user-allocated buffer that will contain the complete
1058 * error message. This buffer has to be at least 'errbuflen' in length.
1059 * It can be NULL; in this case the error cannot be printed.
1061 * \param errbuflen: length of the buffer that will contains the error. The error message cannot be
1062 * larger than 'errbuflen - 1' because the last char is reserved for the string terminator.
1064 * \return '0' if everything is fine, '-1' if some errors occurred.
1065 * The error message is returned in the 'errbuf' variable.
1067 int sock_discard(SOCKET sock
, int size
, char *errbuf
, int errbuflen
)
1069 #define TEMP_BUF_SIZE 32768
1071 char buffer
[TEMP_BUF_SIZE
]; /* network buffer, to be used when the message is discarded */
1074 * A static allocation avoids the need of a 'malloc()' each time we want to discard a message
1075 * Our feeling is that a buffer if 32KB is enough for most of the application;
1076 * in case this is not enough, the "while" loop discards the message by calling the
1077 * sockrecv() several times.
1078 * We do not want to create a bigger variable because this causes the program to exit on
1079 * some platforms (e.g. BSD)
1081 while (size
> TEMP_BUF_SIZE
)
1083 if (sock_recv(sock
, buffer
, TEMP_BUF_SIZE
, SOCK_RECEIVEALL_YES
, errbuf
, errbuflen
) == -1)
1086 size
-= TEMP_BUF_SIZE
;
1090 * If there is still data to be discarded
1091 * In this case, the data can fit into the temporary buffer
1095 if (sock_recv(sock
, buffer
, size
, SOCK_RECEIVEALL_YES
, errbuf
, errbuflen
) == -1)
1099 SOCK_ASSERT("I'm currently discarding data\n", 1);
1105 * \brief Checks that one host (identified by the sockaddr_storage structure) belongs to an 'allowed list'.
1107 * This function is useful after an accept() call in order to check if the connecting
1108 * host is allowed to connect to me. To do that, we have a buffer that keeps the list of the
1109 * allowed host; this function checks the sockaddr_storage structure of the connecting host
1110 * against this host list, and it returns '0' is the host is included in this list.
1112 * \param hostlist: pointer to a string that contains the list of the allowed host.
1114 * \param sep: a string that keeps the separators used between the hosts (for example the
1115 * space character) in the host list.
1117 * \param from: a sockaddr_storage structure, as it is returned by the accept() call.
1119 * \param errbuf: a pointer to an user-allocated buffer that will contain the complete
1120 * error message. This buffer has to be at least 'errbuflen' in length.
1121 * It can be NULL; in this case the error cannot be printed.
1123 * \param errbuflen: length of the buffer that will contains the error. The error message cannot be
1124 * larger than 'errbuflen - 1' because the last char is reserved for the string terminator.
1126 * \return It returns:
1127 * - '1' if the host list is empty
1128 * - '0' if the host belongs to the host list (and therefore it is allowed to connect)
1129 * - '-1' in case the host does not belong to the host list (and therefore it is not allowed to connect
1130 * - '-2' in case or error. The error message is returned in the 'errbuf' variable.
1132 int sock_check_hostlist(char *hostlist
, const char *sep
, struct sockaddr_storage
*from
, char *errbuf
, int errbuflen
)
1134 /* checks if the connecting host is among the ones allowed */
1135 if ((hostlist
) && (hostlist
[0]))
1137 char *token
; /* temp, needed to separate items into the hostlist */
1138 struct addrinfo
*addrinfo
, *ai_next
;
1143 * The problem is that strtok modifies the original variable by putting '0' at the end of each token
1144 * So, we have to create a new temporary string in which the original content is kept
1146 temphostlist
= strdup(hostlist
);
1147 if (temphostlist
== NULL
)
1149 sock_geterror("sock_check_hostlist(), malloc() failed", errbuf
, errbuflen
);
1153 token
= pcap_strtok_r(temphostlist
, sep
, &lasts
);
1155 /* it avoids a warning in the compilation ('addrinfo used but not initialized') */
1158 while (token
!= NULL
)
1160 struct addrinfo hints
;
1164 memset(&hints
, 0, sizeof(struct addrinfo
));
1165 hints
.ai_family
= PF_UNSPEC
;
1166 hints
.ai_socktype
= SOCK_STREAM
;
1168 retval
= getaddrinfo(token
, "0", &hints
, &addrinfo
);
1172 pcap_snprintf(errbuf
, errbuflen
, "getaddrinfo() %s", gai_strerror(retval
));
1174 SOCK_ASSERT(errbuf
, 1);
1176 /* Get next token */
1177 token
= pcap_strtok_r(NULL
, sep
, &lasts
);
1181 /* ai_next is required to preserve the content of addrinfo, in order to deallocate it properly */
1185 if (sock_cmpaddr(from
, (struct sockaddr_storage
*) ai_next
->ai_addr
) == 0)
1188 freeaddrinfo(addrinfo
);
1193 * If we are here, it means that the current address does not matches
1194 * Let's try with the next one in the header chain
1196 ai_next
= ai_next
->ai_next
;
1199 freeaddrinfo(addrinfo
);
1202 /* Get next token */
1203 token
= pcap_strtok_r(NULL
, sep
, &lasts
);
1208 freeaddrinfo(addrinfo
);
1213 pcap_snprintf(errbuf
, errbuflen
, "The host is not in the allowed host list. Connection refused.");
1219 /* No hostlist, so we have to return 'empty list' */
1224 * \brief Compares two addresses contained into two sockaddr_storage structures.
1226 * This function is useful to compare two addresses, given their internal representation,
1227 * i.e. an sockaddr_storage structure.
1229 * The two structures do not need to be sockaddr_storage; you can have both 'sockaddr_in' and
1230 * sockaddr_in6, properly acsted in order to be compliant to the function interface.
1232 * This function will return '0' if the two addresses matches, '-1' if not.
1234 * \param first: a sockaddr_storage structure, (for example the one that is returned by an
1235 * accept() call), containing the first address to compare.
1237 * \param second: a sockaddr_storage structure containing the second address to compare.
1239 * \return '0' if the addresses are equal, '-1' if they are different.
1241 int sock_cmpaddr(struct sockaddr_storage
*first
, struct sockaddr_storage
*second
)
1243 if (first
->ss_family
== second
->ss_family
)
1245 if (first
->ss_family
== AF_INET
)
1247 if (memcmp(&(((struct sockaddr_in
*) first
)->sin_addr
),
1248 &(((struct sockaddr_in
*) second
)->sin_addr
),
1249 sizeof(struct in_addr
)) == 0)
1252 else /* address family is AF_INET6 */
1254 if (memcmp(&(((struct sockaddr_in6
*) first
)->sin6_addr
),
1255 &(((struct sockaddr_in6
*) second
)->sin6_addr
),
1256 sizeof(struct in6_addr
)) == 0)
1265 * \brief It gets the address/port the system picked for this socket (on connected sockets).
1267 * It is used to return the address and port the server picked for our socket on the local machine.
1269 * - connected sockets
1272 * On unconnected client sockets it does not work because the system dynamically chooses a port
1273 * only when the socket calls a send() call.
1275 * \param sock: the connected socket currently opened.
1277 * \param address: it contains the address that will be returned by the function. This buffer
1278 * must be properly allocated by the user. The address can be either literal or numeric depending
1279 * on the value of 'Flags'.
1281 * \param addrlen: the length of the 'address' buffer.
1283 * \param port: it contains the port that will be returned by the function. This buffer
1284 * must be properly allocated by the user.
1286 * \param portlen: the length of the 'port' buffer.
1288 * \param flags: a set of flags (the ones defined into the getnameinfo() standard socket function)
1289 * that determine if the resulting address must be in numeric / literal form, and so on.
1291 * \param errbuf: a pointer to an user-allocated buffer that will contain the complete
1292 * error message. This buffer has to be at least 'errbuflen' in length.
1293 * It can be NULL; in this case the error cannot be printed.
1295 * \param errbuflen: length of the buffer that will contains the error. The error message cannot be
1296 * larger than 'errbuflen - 1' because the last char is reserved for the string terminator.
1298 * \return It returns '-1' if this function succeeds, '0' otherwise.
1299 * The address and port corresponding are returned back in the buffers 'address' and 'port'.
1300 * In any case, the returned strings are '0' terminated.
1302 * \warning If the socket is using a connectionless protocol, the address may not be available
1303 * until I/O occurs on the socket.
1305 int sock_getmyinfo(SOCKET sock
, char *address
, int addrlen
, char *port
, int portlen
, int flags
, char *errbuf
, int errbuflen
)
1307 struct sockaddr_storage mysockaddr
;
1308 socklen_t sockaddrlen
;
1311 sockaddrlen
= sizeof(struct sockaddr_storage
);
1313 if (getsockname(sock
, (struct sockaddr
*) &mysockaddr
, &sockaddrlen
) == -1)
1315 sock_geterror("getsockname(): ", errbuf
, errbuflen
);
1319 /* Returns the numeric address of the host that triggered the error */
1320 return sock_getascii_addrport(&mysockaddr
, address
, addrlen
, port
, portlen
, flags
, errbuf
, errbuflen
);
1324 * \brief It retrieves two strings containing the address and the port of a given 'sockaddr' variable.
1326 * This function is basically an extended version of the inet_ntop(), which does not exist in
1327 * Winsock because the same result can be obtained by using the getnameinfo().
1328 * However, differently from inet_ntop(), this function is able to return also literal names
1329 * (e.g. 'localhost') dependently from the 'Flags' parameter.
1331 * The function accepts a sockaddr_storage variable (which can be returned by several functions
1332 * like bind(), connect(), accept(), and more) and it transforms its content into a 'human'
1333 * form. So, for instance, it is able to translate an hex address (stored in binary form) into
1334 * a standard IPv6 address like "::1".
1336 * The behavior of this function depends on the parameters we have in the 'Flags' variable, which
1337 * are the ones allowed in the standard getnameinfo() socket function.
1339 * \param sockaddr: a 'sockaddr_in' or 'sockaddr_in6' structure containing the address that
1340 * need to be translated from network form into the presentation form. This structure must be
1341 * zero-ed prior using it, and the address family field must be filled with the proper value.
1342 * The user must cast any 'sockaddr_in' or 'sockaddr_in6' structures to 'sockaddr_storage' before
1343 * calling this function.
1345 * \param address: it contains the address that will be returned by the function. This buffer
1346 * must be properly allocated by the user. The address can be either literal or numeric depending
1347 * on the value of 'Flags'.
1349 * \param addrlen: the length of the 'address' buffer.
1351 * \param port: it contains the port that will be returned by the function. This buffer
1352 * must be properly allocated by the user.
1354 * \param portlen: the length of the 'port' buffer.
1356 * \param flags: a set of flags (the ones defined into the getnameinfo() standard socket function)
1357 * that determine if the resulting address must be in numeric / literal form, and so on.
1359 * \param errbuf: a pointer to an user-allocated buffer that will contain the complete
1360 * error message. This buffer has to be at least 'errbuflen' in length.
1361 * It can be NULL; in this case the error cannot be printed.
1363 * \param errbuflen: length of the buffer that will contains the error. The error message cannot be
1364 * larger than 'errbuflen - 1' because the last char is reserved for the string terminator.
1366 * \return It returns '-1' if this function succeeds, '0' otherwise.
1367 * The address and port corresponding to the given SockAddr are returned back in the buffers 'address'
1369 * In any case, the returned strings are '0' terminated.
1371 int sock_getascii_addrport(const struct sockaddr_storage
*sockaddr
, char *address
, int addrlen
, char *port
, int portlen
, int flags
, char *errbuf
, int errbuflen
)
1373 socklen_t sockaddrlen
;
1374 int retval
; /* Variable that keeps the return value; */
1379 if (sockaddr
->ss_family
== AF_INET
)
1380 sockaddrlen
= sizeof(struct sockaddr_in
);
1382 sockaddrlen
= sizeof(struct sockaddr_in6
);
1384 sockaddrlen
= sizeof(struct sockaddr_storage
);
1387 if ((flags
& NI_NUMERICHOST
) == 0) /* Check that we want literal names */
1389 if ((sockaddr
->ss_family
== AF_INET6
) &&
1390 (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))
1393 strlcpy(address
, SOCKET_NAME_NULL_DAD
, addrlen
);
1398 if (getnameinfo((struct sockaddr
*) sockaddr
, sockaddrlen
, address
, addrlen
, port
, portlen
, flags
) != 0)
1400 /* If the user wants to receive an error message */
1403 sock_geterror("getnameinfo(): ", errbuf
, errbuflen
);
1404 errbuf
[errbuflen
- 1] = 0;
1409 strlcpy(address
, SOCKET_NO_NAME_AVAILABLE
, addrlen
);
1410 address
[addrlen
- 1] = 0;
1415 strlcpy(port
, SOCKET_NO_PORT_AVAILABLE
, portlen
);
1416 port
[portlen
- 1] = 0;
1426 * \brief It translates an address from the 'presentation' form into the 'network' form.
1428 * This function basically replaces inet_pton(), which does not exist in Winsock because
1429 * the same result can be obtained by using the getaddrinfo().
1430 * An additional advantage is that 'Address' can be both a numeric address (e.g. '127.0.0.1',
1431 * like in inet_pton() ) and a literal name (e.g. 'localhost').
1433 * This function does the reverse job of sock_getascii_addrport().
1435 * \param address: a zero-terminated string which contains the name you have to
1436 * translate. The name can be either literal (e.g. 'localhost') or numeric (e.g. '::1').
1438 * \param sockaddr: a user-allocated sockaddr_storage structure which will contains the
1439 * 'network' form of the requested address.
1441 * \param addr_family: a constant which can assume the following values:
1442 * - 'AF_INET' if we want to ping an IPv4 host
1443 * - 'AF_INET6' if we want to ping an IPv6 host
1444 * - 'AF_UNSPEC' if we do not have preferences about the protocol used to ping the host
1446 * \param errbuf: a pointer to an user-allocated buffer that will contain the complete
1447 * error message. This buffer has to be at least 'errbuflen' in length.
1448 * It can be NULL; in this case the error cannot be printed.
1450 * \param errbuflen: length of the buffer that will contains the error. The error message cannot be
1451 * larger than 'errbuflen - 1' because the last char is reserved for the string terminator.
1453 * \return '-1' if the translation succeeded, '-2' if there was some non critical error, '0'
1454 * otherwise. In case it fails, the content of the SockAddr variable remains unchanged.
1455 * A 'non critical error' can occur in case the 'Address' is a literal name, which can be mapped
1456 * to several network addresses (e.g. 'foo.bar.com' => '10.2.2.2' and '10.2.2.3'). In this case
1457 * the content of the SockAddr parameter will be the address corresponding to the first mapping.
1459 * \warning The sockaddr_storage structure MUST be allocated by the user.
1461 int sock_present2network(const char *address
, struct sockaddr_storage
*sockaddr
, int addr_family
, char *errbuf
, int errbuflen
)
1464 struct addrinfo
*addrinfo
;
1465 struct addrinfo hints
;
1467 memset(&hints
, 0, sizeof(hints
));
1469 hints
.ai_family
= addr_family
;
1471 if ((retval
= sock_initaddress(address
, "22222" /* fake port */, &hints
, &addrinfo
, errbuf
, errbuflen
)) == -1)
1474 if (addrinfo
->ai_family
== PF_INET
)
1475 memcpy(sockaddr
, addrinfo
->ai_addr
, sizeof(struct sockaddr_in
));
1477 memcpy(sockaddr
, addrinfo
->ai_addr
, sizeof(struct sockaddr_in6
));
1479 if (addrinfo
->ai_next
!= NULL
)
1481 freeaddrinfo(addrinfo
);
1484 pcap_snprintf(errbuf
, errbuflen
, "More than one socket requested; using the first one returned");
1488 freeaddrinfo(addrinfo
);