blob: 6f7e88dd381dda1cdce5727e18f2a4c4d6798e85 [file] [edit]
/*
* GPAC - Multimedia Framework C SDK
*
* Authors: Jean Le Feuvre
* Copyright (c) Telecom ParisTech 2005-2012
* All rights reserved
*
* This file is part of GPAC / common tools sub-project
*
* GPAC is free software; you can redistribute it and/or modify
* it under the terms of the GNU Lesser General Public License as published by
* the Free Software Foundation; either version 2, or (at your option)
* any later version.
*
* GPAC 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 Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this library; see the file COPYING. If not, write to
* the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA.
*
*/
#if defined(WIN32) || defined(_WIN32_WCE)
#ifdef _WIN32_WCE
#include <winsock.h>
#else
#include <sys/timeb.h>
#include <winsock2.h>
#include <ws2tcpip.h>
#endif
#include <windows.h>
#if !defined(__GNUC__)
#if defined(IPV6_MULTICAST_IF)
#define GPAC_HAS_IPV6 1
#pragma message("Using WinSock IPV6")
#else
#undef GPAC_HAS_IPV6
#pragma message("Using WinSock IPV4")
#endif
#endif
/*common win32 redefs*/
#define EAGAIN WSAEWOULDBLOCK
#define EISCONN WSAEISCONN
#define ENOTCONN WSAENOTCONN
#define ECONNRESET WSAECONNRESET
#define EMSGSIZE WSAEMSGSIZE
#define ECONNABORTED WSAECONNABORTED
#define ENETDOWN WSAENETDOWN
#define LASTSOCKERROR WSAGetLastError()
/*the number of sockets used. This because the WinSock lib needs init*/
static int wsa_init = 0;
#include <gpac/network.h>
/*end-win32*/
#else
/*non-win32*/
#include <unistd.h>
#include <fcntl.h>
#include <netdb.h>
#ifndef __BEOS__
#include <errno.h>
#endif
#ifndef __DARWIN__
#include <sys/time.h>
#endif
#include <netinet/in.h>
#include <netinet/tcp.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <gpac/network.h>
#define INVALID_SOCKET -1
#define SOCKET_ERROR -1
#define LASTSOCKERROR errno
typedef s32 SOCKET;
#define closesocket(v) close(v)
#endif
#ifdef __SYMBIAN32__
#define SSO_CAST
#else
#define SSO_CAST (const char *)
#endif
#define SOCK_MICROSEC_WAIT 500
#ifdef GPAC_HAS_IPV6
static u32 ipv6_check_state = 0;
#endif
/*internal flags*/
enum
{
GF_SOCK_IS_TCP = 1<<9,
GF_SOCK_IS_IPV6 = 1<<10,
GF_SOCK_NON_BLOCKING = 1<<11,
GF_SOCK_IS_MULTICAST = 1<<12,
GF_SOCK_IS_LISTENING = 1<<13,
GF_SOCK_HAS_SOURCE = 1<<14
};
struct __tag_socket
{
u32 flags;
SOCKET socket;
/*destination address for sendto/recvfrom*/
#ifdef GPAC_HAS_IPV6
struct sockaddr_storage dest_addr;
#else
struct sockaddr_in dest_addr;
#endif
u32 dest_addr_len;
};
/*
Some NTP tools
*/
void gf_net_get_ntp(u32 *sec, u32 *frac)
{
struct timeval now;
#ifdef WIN32
s32 gettimeofday(struct timeval *tp, void *tz);
#endif
gettimeofday(&now, NULL);
*sec = (u32) (now.tv_sec) + GF_NTP_SEC_1900_TO_1970;
*frac = (u32) ( (now.tv_usec << 12) + (now.tv_usec << 8) - ((now.tv_usec * 3650) >> 6) );
}
u32 gf_net_has_ipv6()
{
#ifdef GPAC_HAS_IPV6
if (!ipv6_check_state) {
SOCKET s;
#ifdef WIN32
if (!wsa_init) {
WSADATA Data;
if (WSAStartup(0x0202, &Data)!=0) {
ipv6_check_state = 1;
return 0;
}
}
#endif
s = socket(PF_INET6, SOCK_STREAM, 0);
if (!s) ipv6_check_state = 1;
else {
ipv6_check_state = 2;
closesocket(s);
}
#ifdef WIN32
if (!wsa_init) WSACleanup();
#endif
}
return (ipv6_check_state==2);
#else
return 0;
#endif
}
#ifdef GPAC_HAS_IPV6
static struct addrinfo *gf_sk_get_ipv6_addr(char *PeerName, u16 PortNumber, int family, int flags, int sock_type)
{
struct addrinfo *res=NULL;
struct addrinfo hints;
char node[50], portstring[20], *service, *dest;
service = dest = NULL;
memset(&hints, 0, sizeof(hints));
hints.ai_socktype = sock_type;
hints.ai_family = family;
hints.ai_flags = flags;
if (PortNumber) {
sprintf (portstring, "%d", PortNumber);
service = (char *)portstring;
}
if (PeerName) {
gf_strcpy(node, PeerName);
if (node[0]=='[') {
node[strlen(node)-1] = 0;
gf_strcpy(node, &node[1]);
}
dest = (char *) node;
}
if (getaddrinfo((const char *)dest, (const char *)service, &hints, &res) != 0) return NULL;
return res;
}
static Bool gf_sk_ipv6_set_remote_address(GF_Socket *sock, char *address, u16 PortNumber)
{
struct addrinfo *res = gf_sk_get_ipv6_addr(address, PortNumber, AF_UNSPEC, 0, (sock->flags & GF_SOCK_IS_TCP) ? SOCK_STREAM : SOCK_DGRAM);
if (!res) return 0;
memcpy(&sock->dest_addr, res->ai_addr, res->ai_addrlen);
sock->dest_addr_len = res->ai_addrlen;
freeaddrinfo(res);
return 1;
}
#endif
GF_Err gf_sk_get_host_name(char *buffer)
{
s32 ret = gethostname(buffer, GF_MAX_IP_NAME_LEN);
return (ret == SOCKET_ERROR) ? GF_IP_ADDRESS_NOT_FOUND : GF_OK;
}
GF_Err gf_sk_get_local_ip(GF_Socket *sock, char buffer[GF_MAX_IP_NAME_LEN])
{
#ifdef GPAC_HAS_IPV6
char clienthost[NI_MAXHOST];
struct sockaddr_storage clientaddr;
socklen_t addrlen = sizeof(clientaddr);
if (getsockname(sock->socket, (struct sockaddr *)&clientaddr, &addrlen)) return GF_IP_NETWORK_FAILURE;
if (getnameinfo((struct sockaddr *)&clientaddr, addrlen, clienthost, sizeof(clienthost), NULL, 0, NI_NUMERICHOST))
return GF_IP_NETWORK_FAILURE;
gf_strlcpy(buffer, clienthost, GF_MAX_IP_NAME_LEN);
#else
char *ip;
struct sockaddr_in name;
u32 len = sizeof(struct sockaddr_in);
if (getsockname(sock->socket, (struct sockaddr*) &name, &len)) return GF_IP_NETWORK_FAILURE;
ip = inet_ntoa(name.sin_addr);
if (!ip) return GF_IP_NETWORK_FAILURE;
gf_strlcpy(buffer, ip, GF_MAX_IP_NAME_LEN);
#endif
return GF_OK;
}
GF_Socket *gf_sk_new(u32 SocketType)
{
GF_Socket *tmp;
/*init WinSock*/
#ifdef WIN32
WSADATA Data;
if (!wsa_init && (WSAStartup(0x0202, &Data)!=0) ) return NULL;
#endif
if ((SocketType != GF_SOCK_TYPE_UDP) && (SocketType != GF_SOCK_TYPE_TCP)) return NULL;
GF_SAFEALLOC(tmp, GF_Socket);
if (!tmp) return NULL;
if (SocketType == GF_SOCK_TYPE_TCP) tmp->flags |= GF_SOCK_IS_TCP;
#ifdef GPAC_HAS_IPV6
memset(&tmp->dest_addr, 0, sizeof(struct sockaddr_storage));
#else
memset(&tmp->dest_addr, 0, sizeof(struct sockaddr_in));
tmp->dest_addr_len = sizeof(struct sockaddr);
#endif
#ifdef WIN32
wsa_init ++;
#endif
return tmp;
}
GF_Err gf_sk_set_buffer_size(GF_Socket *sock, Bool SendBuffer, u32 NewSize)
{
if (!sock || !sock->socket) return GF_BAD_PARAM;
if (SendBuffer) {
setsockopt(sock->socket, SOL_SOCKET, SO_SNDBUF, (char *) &NewSize, sizeof(u32) );
} else {
setsockopt(sock->socket, SOL_SOCKET, SO_RCVBUF, (char *) &NewSize, sizeof(u32) );
}
return GF_OK;
}
GF_Err gf_sk_set_block_mode(GF_Socket *sock, u32 NonBlockingOn)
{
#ifdef WIN32
s32 res;
u_long val = NonBlockingOn;
res = ioctlsocket(sock->socket, FIONBIO, &val);
if (res) return GF_SERVICE_ERROR;
#else
// s32 flag = fcntl(sock->socket, F_GETFL, 0);
// res = fcntl(sock->socket, F_SETFL, flag | O_NONBLOCK);
// if (res) return GF_SERVICE_ERROR;
#endif
if (NonBlockingOn) {
sock->flags |= GF_SOCK_NON_BLOCKING;
} else {
sock->flags &= ~GF_SOCK_NON_BLOCKING;
}
return GF_OK;
}
static void gf_sk_gf_free(GF_Socket *sock)
{
/*leave multicast*/
if (sock->socket && (sock->flags & GF_SOCK_IS_MULTICAST) ) {
struct ip_mreq mreq;
#ifdef GPAC_HAS_IPV6
struct sockaddr *addr = (struct sockaddr *)&sock->dest_addr;
if (addr->sa_family==AF_INET6) {
struct ipv6_mreq mreq6;
memcpy(&mreq6.ipv6mr_multiaddr, &(((struct sockaddr_in6 *)addr)->sin6_addr), sizeof(struct in6_addr));
mreq6.ipv6mr_interface= 0;
setsockopt(sock->socket, IPPROTO_IPV6, IPV6_ADD_MEMBERSHIP, (char *) &mreq6, sizeof(mreq6));
} else {
mreq.imr_multiaddr.s_addr = ((struct sockaddr_in *)addr)->sin_addr.s_addr;
mreq.imr_interface.s_addr = INADDR_ANY;
setsockopt(sock->socket, IPPROTO_IP, IP_DROP_MEMBERSHIP, (char *) &mreq, sizeof(mreq));
}
#else
mreq.imr_multiaddr.s_addr = sock->dest_addr.sin_addr.s_addr;
mreq.imr_interface.s_addr = INADDR_ANY;
setsockopt(sock->socket, IPPROTO_IP, IP_DROP_MEMBERSHIP, (char *) &mreq, sizeof(mreq));
#endif
}
if (sock->socket) closesocket(sock->socket);
sock->socket = (SOCKET) 0L;
}
void gf_sk_del(GF_Socket *sock)
{
gf_sk_gf_free(sock);
#ifdef WIN32
wsa_init --;
if (!wsa_init) WSACleanup();
#endif
gf_free(sock);
}
void gf_sk_reset(GF_Socket *sock)
{
u32 clear;
if (sock) setsockopt(sock->socket, SOL_SOCKET, SO_ERROR, (char *) &clear, sizeof(u32) );
}
s32 gf_sk_get_handle(GF_Socket *sock)
{
return sock->socket;
}
//connects a socket to a remote peer on a given port
GF_Err gf_sk_connect(GF_Socket *sock, char *PeerName, u16 PortNumber, char *local_ip)
{
s32 ret;
#ifdef GPAC_HAS_IPV6
u32 type = (sock->flags & GF_SOCK_IS_TCP) ? SOCK_STREAM : SOCK_DGRAM;
struct addrinfo *res, *aip;
gf_sk_gf_free(sock);
res = gf_sk_get_ipv6_addr(PeerName, PortNumber, AF_UNSPEC, AI_PASSIVE, type);
if (!res) return GF_IP_CONNECTION_FAILURE;
/*for all interfaces*/
for (aip=res; aip!=NULL; aip=aip->ai_next) {
if (type != (u32) aip->ai_socktype) continue;
sock->socket = socket(aip->ai_family, aip->ai_socktype, aip->ai_protocol);
if (sock->socket == INVALID_SOCKET) {
sock->socket = (SOCKET)NULL;
continue;
}
if (sock->flags & GF_SOCK_NON_BLOCKING) gf_sk_set_block_mode(sock, 1);
if (aip->ai_family==PF_INET6) sock->flags |= GF_SOCK_IS_IPV6;
else sock->flags &= ~GF_SOCK_IS_IPV6;
ret = connect(sock->socket, aip->ai_addr, aip->ai_addrlen);
if (ret == SOCKET_ERROR) {
closesocket(sock->socket);
sock->socket = (SOCKET)NULL;
continue;
}
memcpy(&sock->dest_addr, aip->ai_addr, aip->ai_addrlen);
sock->dest_addr_len = aip->ai_addrlen;
freeaddrinfo(res);
return GF_OK;
}
freeaddrinfo(res);
return GF_IP_CONNECTION_FAILURE;
#else
struct hostent *Host=NULL;
if (!sock->socket)
sock->socket = socket(AF_INET, (sock->flags & GF_SOCK_IS_TCP) ? SOCK_STREAM : SOCK_DGRAM, 0);
/*setup the address*/
sock->dest_addr.sin_family = AF_INET;
sock->dest_addr.sin_port = htons(PortNumber);
/*get the server IP*/
sock->dest_addr.sin_addr.s_addr = inet_addr(PeerName);
if (sock->dest_addr.sin_addr.s_addr==INADDR_NONE) {
Host = gf_gethostbyname(PeerName);
if (Host == NULL) {
switch (LASTSOCKERROR) {
#ifndef __SYMBIAN32__
case ENETDOWN:
return GF_IP_NETWORK_FAILURE;
//case ENOHOST: return GF_IP_ADDRESS_NOT_FOUND;
#endif
default:
return GF_IP_NETWORK_FAILURE;
}
}
memcpy((char *) &sock->dest_addr.sin_addr, Host->h_addr_list[0], sizeof(u32));
}
if (sock->flags & GF_SOCK_IS_TCP) {
ret = connect(sock->socket, (struct sockaddr *) &sock->dest_addr, sizeof(struct sockaddr));
if (ret == SOCKET_ERROR) {
switch (LASTSOCKERROR) {
case EAGAIN:
return GF_IP_SOCK_WOULD_BLOCK;
case EISCONN:
return GF_OK;
default:
return GF_IP_CONNECTION_FAILURE;
}
}
}
#endif
return GF_OK;
}
//binds the given socket to the specified port. If ReUse is true
//this will enable reuse of ports on a single machine
GF_Err gf_sk_bind(GF_Socket *sock, char *local_ip, u16 port, char *peer_name, u16 peer_port, u32 options)
{
#ifdef GPAC_HAS_IPV6
struct addrinfo *res, *aip;
int af;
u32 type;
#else
size_t addrlen;
struct sockaddr_in LocalAdd;
struct hostent *Host;
char buf[GF_MAX_IP_NAME_LEN];
#endif
s32 ret;
s32 optval;
if (!sock || sock->socket) return GF_BAD_PARAM;
#ifdef GPAC_HAS_IPV6
type = (sock->flags & GF_SOCK_IS_TCP) ? SOCK_STREAM : SOCK_DGRAM;
af = (options & GF_SOCK_FORCE_IPV6) ? PF_INET6 : PF_UNSPEC;
if (!gf_net_has_ipv6()) af = PF_INET;
/*probe way to peer: is it V4 or V6? */
if (peer_name && peer_port) {
res = gf_sk_get_ipv6_addr(peer_name, peer_port, af, AI_PASSIVE, type);
if (!res) return GF_IP_CONNECTION_FAILURE;
#ifdef WIN32
/*win32 has troubles redirecting IPV4 datagrams to IPV6 sockets, so override
local family type to avoid IPV4(S)->IPV6(C) UDP*/
af = res->ai_family;
#endif
memcpy(&sock->dest_addr, res->ai_addr, res->ai_addrlen);
sock->dest_addr_len = res->ai_addrlen;
freeaddrinfo(res);
}
res = gf_sk_get_ipv6_addr(NULL, port, af, AI_PASSIVE, type);
if (!res) return GF_IP_CONNECTION_FAILURE;
/*for all interfaces*/
for (aip=res; aip!=NULL; aip=aip->ai_next) {
if (type != (u32) aip->ai_socktype) continue;
#ifdef WIN32
/*recurrent pb with win32: this is not a true dual-stack, listening with a v6 socket for v4 source is
most of the time failing. On win32, only move for connection-less V6 sockets if no other way available*/
if (aip->ai_next && (aip->ai_next->ai_family==PF_INET)) continue;
#endif
sock->socket = socket(aip->ai_family, aip->ai_socktype, aip->ai_protocol);
if (sock->socket == INVALID_SOCKET) {
sock->socket = (SOCKET)NULL;
continue;
}
if (options & GF_SOCK_REUSE_PORT) {
optval = 1;
setsockopt(sock->socket, SOL_SOCKET, SO_REUSEADDR, (const char *) &optval, sizeof(optval));
}
if (sock->flags & GF_SOCK_NON_BLOCKING) gf_sk_set_block_mode(sock, 1);
ret = bind(sock->socket, aip->ai_addr, aip->ai_addrlen);
if (ret == SOCKET_ERROR) {
closesocket(sock->socket);
sock->socket = (SOCKET)NULL;
continue;
}
if (aip->ai_family==PF_INET6) sock->flags |= GF_SOCK_IS_IPV6;
else sock->flags &= ~GF_SOCK_IS_IPV6;
if (peer_name && peer_port)
sock->flags |= GF_SOCK_HAS_SOURCE;
freeaddrinfo(res);
return GF_OK;
}
freeaddrinfo(res);
return GF_IP_CONNECTION_FAILURE;
#else
if (!sock->socket) {
sock->socket = socket(AF_INET, (sock->flags & GF_SOCK_IS_TCP) ? SOCK_STREAM : SOCK_DGRAM, 0);
if (sock->flags & GF_SOCK_NON_BLOCKING) gf_sk_set_block_mode(sock, 1);
sock->flags &= ~GF_SOCK_IS_IPV6;
}
memset((void *) &LocalAdd, 0, sizeof(LocalAdd));
ret = gethostname(buf, GF_MAX_IP_NAME_LEN);
if (ret == SOCKET_ERROR) return GF_IP_ADDRESS_NOT_FOUND;
/*get the IP address*/
Host = gf_gethostbyname(buf);
if (Host == NULL) return GF_IP_ADDRESS_NOT_FOUND;
/*setup the address*/
memcpy((char *) &LocalAdd.sin_addr, Host->h_addr_list[0], sizeof(LocalAdd.sin_addr));
LocalAdd.sin_family = AF_INET;
LocalAdd.sin_addr.s_addr = INADDR_ANY;
LocalAdd.sin_port = htons(port);
addrlen = sizeof(struct sockaddr_in);
if (options & GF_SOCK_REUSE_PORT) {
optval = 1;
setsockopt(sock->socket, SOL_SOCKET, SO_REUSEADDR, SSO_CAST &optval, sizeof(optval));
}
/*bind the socket*/
ret = bind(sock->socket, (struct sockaddr *) &LocalAdd, addrlen);
if (ret == SOCKET_ERROR) return GF_IP_CONNECTION_FAILURE;
if (peer_name && peer_port) {
sock->dest_addr.sin_port = htons(peer_port);
sock->dest_addr.sin_family = AF_INET;
sock->dest_addr.sin_addr.s_addr = inet_addr(peer_name);
if (sock->dest_addr.sin_addr.s_addr == INADDR_NONE) {
Host = gf_gethostbyname(peer_name);
if (Host == NULL) return GF_IP_ADDRESS_NOT_FOUND;
memcpy((char *) &sock->dest_addr.sin_addr, Host->h_addr_list[0], sizeof(u32));
}
sock->flags |= GF_SOCK_HAS_SOURCE;
}
#endif
return GF_OK;
}
//send length bytes of a buffer
GF_Err gf_sk_send(GF_Socket *sock, char *buffer, u32 length)
{
GF_Err e;
u32 Count;
s32 Res;
#ifndef __SYMBIAN32__
u32 ready;
struct timeval timeout;
fd_set Group;
#endif
e = GF_OK;
//the socket must be bound or connected
if (!sock || !sock->socket) return GF_BAD_PARAM;
#ifndef __SYMBIAN32__
//can we write?
FD_ZERO(&Group);
FD_SET(sock->socket, &Group);
timeout.tv_sec = 0;
timeout.tv_usec = SOCK_MICROSEC_WAIT;
ready = select(sock->socket+1, NULL, &Group, NULL, &timeout);
if (ready == SOCKET_ERROR) {
switch (LASTSOCKERROR) {
case EAGAIN:
return GF_IP_SOCK_WOULD_BLOCK;
default:
return GF_IP_NETWORK_FAILURE;
}
}
//should never happen (to check: is writeability is guaranteed for not-connected sockets)
if (!ready || !FD_ISSET(sock->socket, &Group)) {
return GF_IP_NETWORK_EMPTY;
}
#endif
//direct writing
Count = 0;
while (Count < length) {
if (sock->flags & GF_SOCK_IS_TCP) {
Res = send(sock->socket, (char *) &buffer[Count], length - Count, 0);
} else {
Res = sendto(sock->socket, (char *) &buffer[Count], length - Count, 0, (struct sockaddr *) &sock->dest_addr, sock->dest_addr_len);
}
if (Res == SOCKET_ERROR) {
switch (LASTSOCKERROR) {
case EAGAIN:
return GF_IP_SOCK_WOULD_BLOCK;
#ifndef __SYMBIAN32__
case ENOTCONN:
case ECONNRESET:
return GF_IP_CONNECTION_CLOSED;
#endif
default:
return GF_IP_NETWORK_FAILURE;
}
}
Count += Res;
}
return GF_OK;
}
GF_EXPORT
u32 gf_sk_is_multicast_address(char *multi_IPAdd)
{
#ifdef GPAC_HAS_IPV6
u32 val;
struct addrinfo *res;
if (!multi_IPAdd) return 0;
res = gf_sk_get_ipv6_addr(multi_IPAdd, 0, AF_UNSPEC, AI_PASSIVE, SOCK_STREAM);
if (!res) return 0;
val = 0;
if (res->ai_addr->sa_family == AF_INET) {
val = IN_MULTICAST(ntohl(((struct sockaddr_in *)res->ai_addr)->sin_addr.s_addr));
} else if (res->ai_addr->sa_family == AF_INET6) {
val = IN6_IS_ADDR_MULTICAST(& ((struct sockaddr_in6 *)res->ai_addr)->sin6_addr);
}
freeaddrinfo(res);
return val;
#else
if (!multi_IPAdd) return 0;
return ((htonl(inet_addr(multi_IPAdd)) >> 8) & 0x00f00000) == 0x00e00000;
#endif
}
GF_Err gf_sk_setup_multicast(GF_Socket *sock, char *multi_IPAdd, u16 MultiPortNumber, u32 TTL, Bool NoBind, char *local_interface_ip)
{
s32 ret;
u32 flag;
struct ip_mreq M_req;
u32 optval;
#ifdef GPAC_HAS_IPV6
struct sockaddr *addr;
struct addrinfo *res, *aip;
u32 type;
#else
u_long local_add_id;
#endif
if (!sock || sock->socket) return GF_BAD_PARAM;
if (TTL > 255) TTL = 255;
/*check the address*/
if (!gf_sk_is_multicast_address(multi_IPAdd)) return GF_BAD_PARAM;
#ifdef GPAC_HAS_IPV6
type = (sock->flags & GF_SOCK_IS_TCP) ? SOCK_STREAM : SOCK_DGRAM;
res = gf_sk_get_ipv6_addr(local_interface_ip, MultiPortNumber, AF_UNSPEC, AI_PASSIVE, type);
if (!res) {
if (local_interface_ip) {
res = gf_sk_get_ipv6_addr(NULL, MultiPortNumber, AF_UNSPEC, AI_PASSIVE, type);
local_interface_ip = NULL;
}
if (!res) return GF_IP_CONNECTION_FAILURE;
}
/*for all interfaces*/
for (aip=res; aip!=NULL; aip=aip->ai_next) {
if (type != (u32) aip->ai_socktype) continue;
sock->socket = socket(aip->ai_family, aip->ai_socktype, aip->ai_protocol);
if (sock->socket == INVALID_SOCKET) {
sock->socket = (SOCKET)NULL;
continue;
}
/*enable address reuse*/
optval = SO_REUSEADDR;
setsockopt(sock->socket, SOL_SOCKET, SO_REUSEADDR, (const char *) &optval, sizeof(optval));
/*TODO: copy over other properties (recption buffer size & co)*/
if (sock->flags & GF_SOCK_NON_BLOCKING) gf_sk_set_block_mode(sock, 1);
memcpy(&sock->dest_addr, aip->ai_addr, aip->ai_addrlen);
sock->dest_addr_len = aip->ai_addrlen;
if (!NoBind) {
ret = bind(sock->socket, aip->ai_addr, aip->ai_addrlen);
if (ret == SOCKET_ERROR) {
closesocket(sock->socket);
sock->socket = (SOCKET)NULL;
continue;
}
}
if (aip->ai_family==PF_INET6) sock->flags |= GF_SOCK_IS_IPV6;
else sock->flags &= ~GF_SOCK_IS_IPV6;
break;
}
freeaddrinfo(res);
if (!sock->socket) return GF_IP_CONNECTION_FAILURE;
if (!gf_sk_ipv6_set_remote_address(sock, multi_IPAdd, MultiPortNumber))
return GF_IP_CONNECTION_FAILURE;
addr = (struct sockaddr *)&sock->dest_addr;
if (addr->sa_family == AF_INET) {
M_req.imr_multiaddr.s_addr = ((struct sockaddr_in *)addr)->sin_addr.s_addr;
M_req.imr_interface.s_addr = INADDR_ANY;
ret = setsockopt(sock->socket, IPPROTO_IP, IP_ADD_MEMBERSHIP, (char *) &M_req, sizeof(M_req));
if (ret == SOCKET_ERROR) return GF_IP_CONNECTION_FAILURE;
/*set TTL*/
ret = setsockopt(sock->socket, IPPROTO_IP, IP_MULTICAST_TTL, (char *) &TTL, sizeof(TTL));
if (ret == SOCKET_ERROR) return GF_IP_CONNECTION_FAILURE;
/*Disable loopback*/
flag = 1;
ret = setsockopt(sock->socket, IPPROTO_IP, IP_MULTICAST_LOOP, (char *) &flag, sizeof(flag));
if (ret == SOCKET_ERROR) return GF_IP_CONNECTION_FAILURE;
}
if (addr->sa_family == AF_INET6) {
struct ipv6_mreq M_reqV6;
memcpy(&M_reqV6.ipv6mr_multiaddr, &(((struct sockaddr_in6 *)addr)->sin6_addr), sizeof(struct in6_addr));
M_reqV6.ipv6mr_interface = 0;
ret = setsockopt(sock->socket, IPPROTO_IPV6, IPV6_ADD_MEMBERSHIP, (char *) &M_reqV6, sizeof(M_reqV6));
if (ret == SOCKET_ERROR) return GF_IP_CONNECTION_FAILURE;
/*set TTL*/
ret = setsockopt(sock->socket, IPPROTO_IPV6, IPV6_MULTICAST_HOPS, (char *) &TTL, sizeof(TTL));
if (ret == SOCKET_ERROR) return GF_IP_CONNECTION_FAILURE;
/*Disable loopback*/
flag = 1;
ret = setsockopt(sock->socket, IPPROTO_IPV6, IPV6_MULTICAST_LOOP, (char *) &flag, sizeof(flag));
if (ret == SOCKET_ERROR) return GF_IP_CONNECTION_FAILURE;
}
#else
/*enable address reuse*/
optval = SO_REUSEADDR;
setsockopt(sock->socket, SOL_SOCKET, SO_REUSEADDR, SSO_CAST &optval, sizeof(optval));
if (local_interface_ip) local_add_id = inet_addr(local_interface_ip);
else local_add_id = htonl(INADDR_ANY);
if (!NoBind) {
struct sockaddr_in local_address;
local_address.sin_family = AF_INET;
local_address.sin_addr.s_addr = local_add_id;
local_address.sin_port = htons( MultiPortNumber);
ret = bind(sock->socket, (struct sockaddr *) &local_address, sizeof(local_address));
if (ret == SOCKET_ERROR) {
/*retry without specifying the local add*/
local_address.sin_addr.s_addr = local_add_id = htonl(INADDR_ANY);
local_interface_ip = NULL;
ret = bind(sock->socket, (struct sockaddr *) &local_address, sizeof(local_address));
if (ret == SOCKET_ERROR) return GF_IP_CONNECTION_FAILURE;
}
/*setup local interface*/
if (local_interface_ip) {
ret = setsockopt(sock->socket, IPPROTO_IP, IP_MULTICAST_IF, (char *) &local_add_id, sizeof(local_add_id));
if (ret == SOCKET_ERROR) return GF_IP_CONNECTION_FAILURE;
}
}
/*now join the multicast*/
M_req.imr_multiaddr.s_addr = inet_addr(multi_IPAdd);
M_req.imr_interface.s_addr = local_add_id;
ret = setsockopt(sock->socket, IPPROTO_IP, IP_ADD_MEMBERSHIP, (char *) &M_req, sizeof(M_req));
if (ret == SOCKET_ERROR) return GF_IP_CONNECTION_FAILURE;
/*set the Time To Live*/
ret = setsockopt(sock->socket, IPPROTO_IP, IP_MULTICAST_TTL, (char *)&TTL, sizeof(TTL));
if (ret == SOCKET_ERROR) return GF_IP_CONNECTION_FAILURE;
/*Disable loopback*/
flag = 1;
ret = setsockopt(sock->socket, IPPROTO_IP, IP_MULTICAST_LOOP, (char *) &flag, sizeof(flag));
if (ret == SOCKET_ERROR) return GF_IP_CONNECTION_FAILURE;
sock->dest_addr.sin_family = AF_INET;
sock->dest_addr.sin_addr.s_addr = M_req.imr_multiaddr.s_addr;
sock->dest_addr.sin_port = htons( MultiPortNumber);
#endif
sock->flags |= GF_SOCK_IS_MULTICAST;
return GF_OK;
}
//fetch nb bytes on a socket and fill the buffer from startFrom
//length is the allocated size of the receiving buffer
//BytesRead is the number of bytes read from the network
GF_Err gf_sk_receive(GF_Socket *sock, char *buffer, u32 length, u32 startFrom, u32 *BytesRead)
{
GF_Err e;
s32 res;
#ifndef __SYMBIAN32__
u32 ready;
struct timeval timeout;
fd_set Group;
#endif
e = GF_OK;
*BytesRead = 0;
if (!sock->socket) return GF_BAD_PARAM;
if (startFrom >= length) return GF_IO_ERR;
#ifndef __SYMBIAN32__
//can we read?
FD_ZERO(&Group);
FD_SET(sock->socket, &Group);
timeout.tv_sec = 0;
timeout.tv_usec = SOCK_MICROSEC_WAIT;
res = 0;
ready = select(sock->socket+1, &Group, NULL, NULL, &timeout);
if (ready == SOCKET_ERROR) {
switch (LASTSOCKERROR) {
case EAGAIN:
return GF_IP_SOCK_WOULD_BLOCK;
default:
GF_LOG(GF_LOG_ERROR, GF_LOG_CORE, ("[socket] cannot select (error %d)\n", LASTSOCKERROR));
return GF_IP_NETWORK_FAILURE;
}
}
if (!FD_ISSET(sock->socket, &Group)) {
return GF_IP_NETWORK_EMPTY;
}
#endif
if (sock->flags & GF_SOCK_HAS_SOURCE)
res = recvfrom(sock->socket, (char *) buffer + startFrom, length - startFrom, 0, (struct sockaddr *)&sock->dest_addr, &sock->dest_addr_len);
else
res = recv(sock->socket, (char *) buffer + startFrom, length - startFrom, 0);
if (res == SOCKET_ERROR) {
res = LASTSOCKERROR;
switch (res) {
case EAGAIN:
return GF_IP_SOCK_WOULD_BLOCK;
#ifndef __SYMBIAN32__
case EMSGSIZE:
return GF_OUT_OF_MEM;
case ENOTCONN:
case ECONNRESET:
case ECONNABORTED:
return GF_IP_CONNECTION_CLOSED;
#endif
default:
GF_LOG(GF_LOG_ERROR, GF_LOG_CORE, ("[socket] cannot read from network (%d)\n", LASTSOCKERROR));
return GF_IP_NETWORK_FAILURE;
}
}
*BytesRead = res;
return GF_OK;
}
GF_Err gf_sk_listen(GF_Socket *sock, u32 MaxConnection)
{
s32 i;
if (!sock || !sock->socket) return GF_BAD_PARAM;
if (MaxConnection >= SOMAXCONN) MaxConnection = SOMAXCONN;
i = listen(sock->socket, MaxConnection);
if (i == SOCKET_ERROR) return GF_IP_NETWORK_FAILURE;
sock->flags |= GF_SOCK_IS_LISTENING;
return GF_OK;
}
GF_Err gf_sk_accept(GF_Socket *sock, GF_Socket **newConnection)
{
u32 client_address_size;
SOCKET sk;
#ifndef __SYMBIAN32__
u32 ready;
struct timeval timeout;
fd_set Group;
#endif
#ifndef __SYMBIAN32__
u32 res;
#endif
*newConnection = NULL;
if (!sock || !(sock->flags & GF_SOCK_IS_LISTENING) ) return GF_BAD_PARAM;
#ifndef __SYMBIAN32__
//can we read?
FD_ZERO(&Group);
FD_SET(sock->socket, &Group);
timeout.tv_sec = 0;
timeout.tv_usec = SOCK_MICROSEC_WAIT;
res = 0;
ready = select(sock->socket, &Group, NULL, NULL, &timeout);
if (ready == SOCKET_ERROR) {
switch (LASTSOCKERROR) {
case EAGAIN:
return GF_IP_SOCK_WOULD_BLOCK;
default:
return GF_IP_NETWORK_FAILURE;
}
}
if (!ready || !FD_ISSET(sock->socket, &Group)) return GF_IP_NETWORK_EMPTY;
#endif
#ifdef GPAC_HAS_IPV6
client_address_size = sizeof(struct sockaddr_in6);
#else
client_address_size = sizeof(struct sockaddr_in);
#endif
sk = accept(sock->socket, (struct sockaddr *) &sock->dest_addr, &client_address_size);
//we either have an error or we have no connections
if (sk == INVALID_SOCKET) {
// if (sock->flags & GF_SOCK_NON_BLOCKING) return GF_IP_NETWORK_FAILURE;
switch (LASTSOCKERROR) {
case EAGAIN:
return GF_IP_SOCK_WOULD_BLOCK;
default:
return GF_IP_NETWORK_FAILURE;
}
}
(*newConnection) = (GF_Socket *) gf_malloc(sizeof(GF_Socket));
(*newConnection)->socket = sk;
(*newConnection)->flags = sock->flags & ~GF_SOCK_IS_LISTENING;
#ifdef GPAC_HAS_IPV6
memcpy( &(*newConnection)->dest_addr, &sock->dest_addr, client_address_size);
memset(&sock->dest_addr, 0, sizeof(struct sockaddr_in6));
#else
memcpy( &(*newConnection)->dest_addr, &sock->dest_addr, client_address_size);
memset(&sock->dest_addr, 0, sizeof(struct sockaddr_in));
#endif
(*newConnection)->dest_addr_len = client_address_size;
return GF_OK;
}
GF_Err gf_sk_get_local_info(GF_Socket *sock, u16 *Port, u32 *Familly)
{
#ifdef GPAC_HAS_IPV6
struct sockaddr_in6 the_add;
#else
struct sockaddr_in the_add;
#endif
u32 size;
if (!sock || !sock->socket) return GF_BAD_PARAM;
if (Port) {
#ifdef GPAC_HAS_IPV6
size = sizeof(struct sockaddr_in6);
if (getsockname(sock->socket, (struct sockaddr *) &the_add, &size) == SOCKET_ERROR) return GF_IP_NETWORK_FAILURE;
*Port = (u32) ntohs(the_add.sin6_port);
#else
size = sizeof(struct sockaddr_in);
if (getsockname(sock->socket, (struct sockaddr *) &the_add, &size) == SOCKET_ERROR) return GF_IP_NETWORK_FAILURE;
*Port = (u32) ntohs(the_add.sin_port);
#endif
}
if (Familly) {
/* size = 4;
if (getsockopt(sock->socket, SOL_SOCKET, SO_TYPE, (char *) &fam, &size) == SOCKET_ERROR)
return GF_IP_NETWORK_FAILURE;
*Familly = fam;
*/
if (sock->flags & GF_SOCK_IS_TCP) *Familly = GF_SOCK_TYPE_TCP;
else *Familly = GF_SOCK_TYPE_UDP;
}
return GF_OK;
}
//we have to do this for the server sockets as we use only one thread
GF_Err gf_sk_server_mode(GF_Socket *sock, Bool serverOn)
{
u32 one;
if (!sock || !(sock->flags & GF_SOCK_IS_TCP) || !sock->socket)
return GF_BAD_PARAM;
one = serverOn ? 1 : 0;
setsockopt(sock->socket, IPPROTO_TCP, TCP_NODELAY, SSO_CAST &one, sizeof(u32));
#ifndef __SYMBIAN32__
setsockopt(sock->socket, SOL_SOCKET, SO_KEEPALIVE, (char *) &one, sizeof(u32));
#endif
return GF_OK;
}
GF_Err gf_sk_get_remote_address(GF_Socket *sock, char buf[GF_MAX_IP_NAME_LEN])
{
#ifdef GPAC_HAS_IPV6
char clienthost[NI_MAXHOST];
struct sockaddr_in6 * addrptr = (struct sockaddr_in6 *)(&sock->dest_addr_len);
if (!sock || sock->socket) return GF_BAD_PARAM;
if (getnameinfo((struct sockaddr *)addrptr, sock->dest_addr_len, clienthost, sizeof(clienthost), NULL, 0, NI_NUMERICHOST))
return GF_IP_ADDRESS_NOT_FOUND;
gf_strlcpy(buf, clienthost, GF_MAX_IP_NAME_LEN);
#else
if (!sock || !sock->socket) return GF_BAD_PARAM;
gf_strlcpy(buf, inet_ntoa(sock->dest_addr.sin_addr), GF_MAX_IP_NAME_LEN);
#endif
return GF_OK;
}
//send length bytes of a buffer
GF_Err gf_sk_send_to(GF_Socket *sock, char *buffer, u32 length, char *remoteHost, u16 remotePort)
{
u32 Count, remote_add_len;
s32 Res;
#ifdef GPAC_HAS_IPV6
struct sockaddr_storage remote_add;
#else
struct sockaddr_in remote_add;
struct hostent *Host=NULL;
#endif
#ifndef __SYMBIAN32__
u32 ready;
struct timeval timeout;
fd_set Group;
#endif
//the socket must be bound or connected
if (!sock || !sock->socket) return GF_BAD_PARAM;
if (remoteHost && !remotePort) return GF_BAD_PARAM;
#ifndef __SYMBIAN32__
//can we write?
FD_ZERO(&Group);
FD_SET(sock->socket, &Group);
timeout.tv_sec = 0;
timeout.tv_usec = SOCK_MICROSEC_WAIT;
ready = select(sock->socket+1, NULL, &Group, NULL, &timeout);
if (ready == SOCKET_ERROR) {
switch (LASTSOCKERROR) {
case EAGAIN:
return GF_IP_SOCK_WOULD_BLOCK;
default:
return GF_IP_NETWORK_FAILURE;
}
}
if (!ready || !FD_ISSET(sock->socket, &Group)) return GF_IP_NETWORK_EMPTY;
#endif
/*setup the address*/
#ifdef GPAC_HAS_IPV6
remote_add.ss_family = AF_INET6;
//if a remote host is specified, use it. Otherwise use the default host
if (remoteHost) {
//setup the address
struct addrinfo *res = gf_sk_get_ipv6_addr(remoteHost, remotePort, AF_UNSPEC, 0, (sock->flags & GF_SOCK_IS_TCP) ? SOCK_STREAM : SOCK_DGRAM);
if (!res) return GF_IP_ADDRESS_NOT_FOUND;
memcpy(&remote_add, res->ai_addr, res->ai_addrlen);
remote_add_len = res->ai_addrlen;
freeaddrinfo(res);
} else {
struct sockaddr_in6 *remotePtr = (struct sockaddr_in6 *)&remote_add;
struct sockaddr_in6 * addrptr = (struct sockaddr_in6 *)(&sock->dest_addr);
remotePtr->sin6_port = addrptr->sin6_port;
remotePtr->sin6_addr = addrptr->sin6_addr;
remote_add_len = sock->dest_addr_len;
}
#else
remote_add_len = sizeof(struct sockaddr_in);
remote_add.sin_family = AF_INET;
//if a remote host is specified, use it. Otherwise use the default host
if (remoteHost) {
//setup the address
remote_add.sin_port = htons(remotePort);
//get the server IP
Host = gf_gethostbyname(remoteHost);
if (Host == NULL) return GF_IP_ADDRESS_NOT_FOUND;
memcpy((char *) &remote_add.sin_addr, Host->h_addr_list[0], sizeof(u32));
} else {
remote_add.sin_port = sock->dest_addr.sin_port;
remote_add.sin_addr.s_addr = sock->dest_addr.sin_addr.s_addr;
}
#endif
Count = 0;
while (Count < length) {
Res = sendto(sock->socket, (char *) &buffer[Count], length - Count, 0, (struct sockaddr *) &remote_add, remote_add_len);
if (Res == SOCKET_ERROR) {
switch (LASTSOCKERROR) {
case EAGAIN:
return GF_IP_SOCK_WOULD_BLOCK;
default:
return GF_IP_NETWORK_FAILURE;
}
}
Count += Res;
}
return GF_OK;
}
GF_Err gf_sk_receive_wait(GF_Socket *sock, char *buffer, u32 length, u32 startFrom, u32 *BytesRead, u32 Second )
{
GF_Err e;
s32 res;
#ifndef __SYMBIAN32__
u32 ready;
struct timeval timeout;
fd_set Group;
#endif
e = GF_OK;
*BytesRead = 0;
if (startFrom >= length) return GF_OK;
#ifndef __SYMBIAN32__
//can we read?
FD_ZERO(&Group);
FD_SET(sock->socket, &Group);
timeout.tv_sec = Second;
timeout.tv_usec = SOCK_MICROSEC_WAIT;
res = 0;
ready = select(sock->socket+1, &Group, NULL, NULL, &timeout);
if (ready == SOCKET_ERROR) {
switch (LASTSOCKERROR) {
case EAGAIN:
return GF_IP_SOCK_WOULD_BLOCK;
default:
return GF_IP_NETWORK_FAILURE;
}
}
if (!FD_ISSET(sock->socket, &Group)) {
return GF_IP_NETWORK_EMPTY;
}
#endif
res = recv(sock->socket, (char *) buffer + startFrom, length - startFrom, 0);
if (res == SOCKET_ERROR) {
switch (LASTSOCKERROR) {
case EAGAIN:
return GF_IP_SOCK_WOULD_BLOCK;
default:
return GF_IP_NETWORK_FAILURE;
}
}
*BytesRead = res;
return GF_OK;
}
//send length bytes of a buffer
GF_Err gf_sk_send_wait(GF_Socket *sock, char *buffer, u32 length, u32 Second )
{
GF_Err e;
u32 Count;
s32 Res;
#ifndef __SYMBIAN32__
u32 ready;
struct timeval timeout;
fd_set Group;
#endif
e = GF_OK;
//the socket must be bound or connected
if (!sock || !sock->socket) return GF_BAD_PARAM;
#ifndef __SYMBIAN32__
//can we write?
FD_ZERO(&Group);
FD_SET(sock->socket, &Group);
timeout.tv_sec = Second;
timeout.tv_usec = SOCK_MICROSEC_WAIT;
ready = select(sock->socket+1, NULL, &Group, NULL, &timeout);
if (ready == SOCKET_ERROR) {
switch (LASTSOCKERROR) {
case EAGAIN:
return GF_IP_SOCK_WOULD_BLOCK;
default:
return GF_IP_NETWORK_FAILURE;
}
}
//should never happen (to check: is writeability is guaranteed for not-connected sockets)
if (!ready || !FD_ISSET(sock->socket, &Group)) {
return GF_IP_NETWORK_EMPTY;
}
#endif
//direct writing
Count = 0;
while (Count < length) {
Res = send(sock->socket, (char *) &buffer[Count], length - Count, 0);
if (Res == SOCKET_ERROR) {
switch (LASTSOCKERROR) {
case EAGAIN:
return GF_IP_SOCK_WOULD_BLOCK;
#ifndef __SYMBIAN32__
case ECONNRESET:
return GF_IP_CONNECTION_CLOSED;
#endif
default:
return GF_IP_NETWORK_FAILURE;
}
}
Count += Res;
}
return GF_OK;
}
#if 0
//Socket Group for select(). The group is a collection of sockets ready for reading / writing
typedef struct __tag_sock_group
{
//the max time value before a select returns
struct timeval timeout;
fd_set ReadGroup;
fd_set WriteGroup;
} GF_SocketGroup;
#define GF_SOCK_GROUP_READ 0
#define GF_SOCK_GROUP_WRITE 1
GF_SocketGroup *NewSockGroup()
{
GF_SocketGroup *tmp = (GF_SocketGroup*)gf_malloc(sizeof(GF_SocketGroup));
if (!tmp) return NULL;
FD_ZERO(&tmp->ReadGroup);
FD_ZERO(&tmp->WriteGroup);
return tmp;
}
void SKG_Delete(GF_SocketGroup *group)
{
gf_free(group);
}
void SKG_SetWatchTime(GF_SocketGroup *group, u32 DelayInS, u32 DelayInMicroS)
{
group->timeout.tv_sec = DelayInS;
group->timeout.tv_usec = DelayInMicroS;
}
void SKG_AddSocket(GF_SocketGroup *group, GF_Socket *sock, u32 GroupType)
{
switch (GroupType) {
case GF_SOCK_GROUP_READ:
FD_SET(sock->socket, &group->ReadGroup);
return;
case GF_SOCK_GROUP_WRITE:
FD_SET(sock->socket, &group->WriteGroup);
return;
default:
return;
}
}
void SKG_RemoveSocket(GF_SocketGroup *group, GF_Socket *sock, u32 GroupType)
{
switch (GroupType) {
case GF_SOCK_GROUP_READ:
FD_CLR(sock->socket, &group->ReadGroup);
return;
case GF_SOCK_GROUP_WRITE:
FD_CLR(sock->socket, &group->WriteGroup);
return;
default:
return;
}
}
Bool SKG_IsSocketIN(GF_SocketGroup *group, GF_Socket *sock, u32 GroupType)
{
switch (GroupType) {
case GF_SOCK_GROUP_READ:
if (FD_ISSET(sock->socket, &group->ReadGroup)) return 1;
return 0;
case GF_SOCK_GROUP_WRITE:
if (FD_ISSET(sock->socket, &group->WriteGroup)) return 1;
return 0;
default:
return 0;
}
}
u32 SKG_Select(GF_SocketGroup *group)
{
u32 ready, rien = 0;
ready = select(rien, &group->ReadGroup, &group->WriteGroup, NULL, &group->timeout);
if (ready == SOCKET_ERROR) return 0;
return ready;
}
#endif