Mercurial > mplayer.hg
view stream/tcp.c @ 19405:0797e1b4a4be
Replace stdint.h with inttypes.h.
author | eugeni |
---|---|
date | Tue, 15 Aug 2006 22:46:56 +0000 |
parents | 2a9d669e5ff6 |
children | 3f19764369ac |
line wrap: on
line source
/* * Copyright (C) 2001 Bertrand BAUDET, 2006 Benjamin Zores * Network helpers for TCP connections * (originally borrowed from network.c, * by Bertrand BAUDET <bertrand_baudet@yahoo.com>). * * This program is free software; you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation; either version 2 of the License, or * (at your option) any later version. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * * You should have received a copy of the GNU General Public License * along with this program; if not, write to the Free Software Foundation, * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA */ #include "config.h" #include <stdlib.h> #include <string.h> #include <errno.h> #include <unistd.h> #include <stdlib.h> #include <sys/types.h> #include <ctype.h> #include <netdb.h> #include <unistd.h> #include <fcntl.h> #ifndef HAVE_WINSOCK2 #include <netinet/in.h> #include <sys/socket.h> #include <arpa/inet.h> #define closesocket close #else #include <winsock2.h> #include <ws2tcpip.h> #endif #include "mp_msg.h" #include "help_mp.h" #include "tcp.h" /* IPv6 options */ int network_prefer_ipv4 = 0; /* Converts an address family constant to a string */ static const char * af2String (int af) { switch (af) { case AF_INET: return "AF_INET"; #ifdef HAVE_AF_INET6 case AF_INET6: return "AF_INET6"; #endif default: return "Unknown address family!"; } } static int connect2Server_with_af (char *host, int port, int af ,int verb) { int socket_server_fd; int err; socklen_t err_len; int ret, count = 0; fd_set set; struct timeval tv; union { struct sockaddr_in four; #ifdef HAVE_AF_INET6 struct sockaddr_in6 six; #endif } server_address; size_t server_address_size; void *our_s_addr; /* Pointer to sin_addr or sin6_addr */ struct hostent *hp = NULL; char buf[255]; #ifdef HAVE_WINSOCK2 u_long val; #endif socket_server_fd = socket (af, SOCK_STREAM, 0); if (socket_server_fd == -1) return TCP_ERROR_FATAL; switch (af) { case AF_INET: our_s_addr = (void *) &server_address.four.sin_addr; break; #ifdef HAVE_AF_INET6 case AF_INET6: our_s_addr = (void *) &server_address.six.sin6_addr; break; #endif default: mp_msg (MSGT_NETWORK, MSGL_ERR, MSGTR_MPDEMUX_NW_UnknownAF, af); return TCP_ERROR_FATAL; } memset (&server_address, 0, sizeof (server_address)); #ifndef HAVE_WINSOCK2 #ifdef USE_ATON if (inet_aton (host, our_s_addr) !=1) #else if (inet_pton (af, host, our_s_addr) !=1) #endif /* USE_ATON */ #else if (inet_addr (host) == INADDR_NONE) #endif /* HAVE_WINSOCK2 */ { if (verb) mp_msg (MSGT_NETWORK, MSGL_STATUS, MSGTR_MPDEMUX_NW_ResolvingHostForAF, host, af2String (af)); #ifdef HAVE_GETHOSTBYNAME2 hp = (struct hostent *) gethostbyname2 (host, af); #else hp = (struct hostent *) gethostbyname (host); #endif /* HAVE_GETHOSTBYNAME2 */ if (!hp) { if (verb) mp_msg (MSGT_NETWORK, MSGL_ERR, MSGTR_MPDEMUX_NW_CantResolv, af2String (af), host); return TCP_ERROR_FATAL; } memcpy (our_s_addr, (void *) hp->h_addr_list[0], hp->h_length); } #ifdef HAVE_WINSOCK2 else { unsigned long addr = inet_addr (host); memcpy (our_s_addr, (void *) &addr, sizeof (addr)); } #endif /* HAVE_WINSOCK2 */ switch (af) { case AF_INET: server_address.four.sin_family = af; server_address.four.sin_port = htons (port); server_address_size = sizeof (server_address.four); break; #ifdef HAVE_AF_INET6 case AF_INET6: server_address.six.sin6_family = af; server_address.six.sin6_port = htons (port); server_address_size = sizeof (server_address.six); break; #endif /* HAVE_AF_INET6 */ default: mp_msg (MSGT_NETWORK, MSGL_ERR, MSGTR_MPDEMUX_NW_UnknownAF, af); return TCP_ERROR_FATAL; } #if defined(USE_ATON) || defined(HAVE_WINSOCK2) strncpy (buf, inet_ntoa (*((struct in_addr *) our_s_addr)), 255); #else inet_ntop (af, our_s_addr, buf, 255); #endif /* USE_ATON || HAVE_WINSOCK2 */ if (verb) mp_msg (MSGT_NETWORK, MSGL_STATUS, MSGTR_MPDEMUX_NW_ConnectingToServer, host, buf, port); /* Turn the socket as non blocking so we can timeout on the connection */ #ifndef HAVE_WINSOCK2 fcntl (socket_server_fd, F_SETFL, fcntl (socket_server_fd, F_GETFL) | O_NONBLOCK); #else val = 1; ioctlsocket (socket_server_fd, FIONBIO, &val); #endif /* HAVE_WINSOCK2 */ if (connect (socket_server_fd, (struct sockaddr *) &server_address, server_address_size) == -1) { #ifndef HAVE_WINSOCK2 if (errno != EINPROGRESS) #else if ((WSAGetLastError () != WSAEINPROGRESS) && (WSAGetLastError () != WSAEWOULDBLOCK)) #endif /* HAVE_WINSOCK2 */ { if (verb) mp_msg (MSGT_NETWORK, MSGL_ERR, MSGTR_MPDEMUX_NW_CantConnect2Server, af2String (af)); closesocket (socket_server_fd); return TCP_ERROR_PORT; } } tv.tv_sec = 0; tv.tv_usec = 500000; FD_ZERO (&set); FD_SET (socket_server_fd, &set); /* When the connection will be made, we will have a writable fd */ while ((ret = select (socket_server_fd + 1, NULL, &set, NULL, &tv)) == 0) { if (ret < 0) mp_msg (MSGT_NETWORK, MSGL_ERR, MSGTR_MPDEMUX_NW_SelectFailed); else if (ret > 0) break; else if (count > 30 || mp_input_check_interrupt (500)) { if (count > 30) mp_msg (MSGT_NETWORK, MSGL_ERR, MSGTR_MPDEMUX_NW_ConnTimeout); else mp_msg (MSGT_NETWORK, MSGL_V, "Connection interuppted by user\n"); return TCP_ERROR_TIMEOUT; } count++; FD_ZERO (&set); FD_SET (socket_server_fd, &set); tv.tv_sec = 0; tv.tv_usec = 500000; } /* Turn back the socket as blocking */ #ifndef HAVE_WINSOCK2 fcntl (socket_server_fd, F_SETFL, fcntl (socket_server_fd, F_GETFL) & ~O_NONBLOCK); #else val = 0; ioctlsocket (socket_server_fd, FIONBIO, &val); #endif /* HAVE_WINSOCK2 */ /* Check if there were any error */ err_len = sizeof (int); ret = getsockopt (socket_server_fd, SOL_SOCKET, SO_ERROR, &err, &err_len); if (ret < 0) { mp_msg (MSGT_NETWORK, MSGL_ERR, MSGTR_MPDEMUX_NW_GetSockOptFailed, strerror (errno)); return TCP_ERROR_FATAL; } if (err > 0) { mp_msg (MSGT_NETWORK, MSGL_ERR, MSGTR_MPDEMUX_NW_ConnectError, strerror (err)); return TCP_ERROR_PORT; } return socket_server_fd; } int connect2Server (char *host, int port, int verb) { #ifdef HAVE_AF_INET6 int r; int s = TCP_ERROR_FATAL; r = connect2Server_with_af (host, port, network_prefer_ipv4 ? AF_INET:AF_INET6, verb); if (r > TCP_ERROR_PORT) return r; s = connect2Server_with_af (host, port, network_prefer_ipv4 ? AF_INET6:AF_INET, verb); if (s == TCP_ERROR_FATAL) return r; return s; #else return connect2Server_with_af (host, port, AF_INET,verb); #endif /* HAVE_AF_INET6 */ }