Mercurial > mplayer.hg
changeset 19541:5177105e3451
remove to allow readding as copy from network.c
author | reimar |
---|---|
date | Sat, 26 Aug 2006 19:20:24 +0000 |
parents | 70a5e89ea4cd |
children | 57980fe3ad96 |
files | stream/tcp.c |
diffstat | 1 files changed, 0 insertions(+), 291 deletions(-) [+] |
line wrap: on
line diff
--- a/stream/tcp.c Sat Aug 26 19:17:04 2006 +0000 +++ /dev/null Thu Jan 01 00:00:00 1970 +0000 @@ -1,291 +0,0 @@ -/* - * 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 <sys/time.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 */ -}