/*

	oscpack -- Open Sound Control packet manipulation library

	http://www.audiomulch.com/~rossb/oscpack

	Copyright (c) 2004-2005 Ross Bencina <rossb@audiomulch.com>

	Permission is hereby granted, free of charge, to any person obtaining

	a copy of this software and associated documentation files

	(the "Software"), to deal in the Software without restriction,

	including without limitation the rights to use, copy, modify, merge,

	publish, distribute, sublicense, and/or sell copies of the Software,

	and to permit persons to whom the Software is furnished to do so,

	subject to the following conditions:

	The above copyright notice and this permission notice shall be

	included in all copies or substantial portions of the Software.

	Any person wishing to distribute modifications to the Software is

	requested to send the modifications to the original developer so that

	they can be incorporated into the canonical version.

	THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,

	EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF

	MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.

	IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR

	ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF

	CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION

	WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

*/

#include "ip/UdpSocket.h"

#include <vector>

#include <algorithm>

#include <stdexcept>

#include <assert.h>

#include <signal.h>

#include <math.h>

#include <errno.h>

#include <string.h> // for memset

#include <pthread.h>

#include <unistd.h>

#include <stdlib.h>

#include <stdio.h>

#include <netdb.h>

#include <sys/types.h>

#include <sys/socket.h>

#include <sys/time.h>

#include <netinet/in.h> // for sockaddr_in

#include "ip/PacketListener.h"

#include "ip/TimerListener.h"

#if defined(__APPLE__) && !defined(_SOCKLEN_T)

// pre system 10.3 didn have socklen_t

typedef ssize_t socklen_t;

#endif

static void SockaddrFromIpEndpointName( struct sockaddr_in& sockAddr, const IpEndpointName& endpoint )

{

    memset( (char *)&sockAddr, 0, sizeof(sockAddr ) );

    sockAddr.sin_family = AF_INET;

	sockAddr.sin_addr.s_addr = 

		(endpoint.address == IpEndpointName::ANY_ADDRESS)

		? INADDR_ANY

		: htonl( endpoint.address );

	sockAddr.sin_port =

		(endpoint.port == IpEndpointName::ANY_PORT)

		? 0

		: htons( endpoint.port );

}

static IpEndpointName IpEndpointNameFromSockaddr( const struct sockaddr_in& sockAddr )

{

	return IpEndpointName( 

		(sockAddr.sin_addr.s_addr == INADDR_ANY) 

			? IpEndpointName::ANY_ADDRESS 

			: ntohl( sockAddr.sin_addr.s_addr ),

		(sockAddr.sin_port == 0)

			? IpEndpointName::ANY_PORT

			: ntohs( sockAddr.sin_port )

		);

}

class UdpSocket::Implementation{

	bool isBound_;

	bool isConnected_;

	int socket_;

	struct sockaddr_in connectedAddr_;

	struct sockaddr_in sendToAddr_;

public:

	Implementation()

		: isBound_( false )

		, isConnected_( false )

		, socket_( -1 )

	{

		if( (socket_ = socket( AF_INET, SOCK_DGRAM, 0 )) == -1 ){

            throw std::runtime_error("unable to create udp socket\n");

        }

		memset( &sendToAddr_, 0, sizeof(sendToAddr_) );

        sendToAddr_.sin_family = AF_INET;

	}

	~Implementation()

	{

		if (socket_ != -1) close(socket_);

	}

	IpEndpointName LocalEndpointFor( const IpEndpointName& remoteEndpoint ) const

	{

		assert( isBound_ );

		// first connect the socket to the remote server

        struct sockaddr_in connectSockAddr;

		SockaddrFromIpEndpointName( connectSockAddr, remoteEndpoint );

        if (connect(socket_, (struct sockaddr *)&connectSockAddr, sizeof(connectSockAddr)) < 0) {

            throw std::runtime_error("unable to connect udp socket\n");

        }

        // get the address

        struct sockaddr_in sockAddr;

        memset( (char *)&sockAddr, 0, sizeof(sockAddr ) );

        socklen_t length = sizeof(sockAddr);

        if (getsockname(socket_, (struct sockaddr *)&sockAddr, &length) < 0) {

            throw std::runtime_error("unable to getsockname\n");

        }

		if( isConnected_ ){

			// reconnect to the connected address

			if (connect(socket_, (struct sockaddr *)&connectedAddr_, sizeof(connectedAddr_)) < 0) {

				throw std::runtime_error("unable to connect udp socket\n");

			}

		}else{

			// unconnect from the remote address

			struct sockaddr_in unconnectSockAddr;

			memset( (char *)&unconnectSockAddr, 0, sizeof(unconnectSockAddr ) );

			unconnectSockAddr.sin_family = AF_UNSPEC;

			// address fields are zero

			int connectResult = connect(socket_, (struct sockaddr *)&unconnectSockAddr, sizeof(unconnectSockAddr));

			if ( connectResult < 0 && errno != EAFNOSUPPORT ) {

				throw std::runtime_error("unable to un-connect udp socket\n");

			}

		}

		return IpEndpointNameFromSockaddr( sockAddr );

	}

	void Connect( const IpEndpointName& remoteEndpoint )

	{

		SockaddrFromIpEndpointName( connectedAddr_, remoteEndpoint );

        if (connect(socket_, (struct sockaddr *)&connectedAddr_, sizeof(connectedAddr_)) < 0) {

            throw std::runtime_error("unable to connect udp socket\n");

        }

		isConnected_ = true;

	}

	void Send( const char *data, int size )

	{

		assert( isConnected_ );

        send( socket_, data, size, 0 );

	}

    void SendTo( const IpEndpointName& remoteEndpoint, const char *data, int size )

	{

		sendToAddr_.sin_addr.s_addr = htonl( remoteEndpoint.address );

        sendToAddr_.sin_port = htons( remoteEndpoint.port );

        sendto( socket_, data, size, 0, (sockaddr*)&sendToAddr_, sizeof(sendToAddr_) );

	}

	void Bind( const IpEndpointName& localEndpoint )

	{

		struct sockaddr_in bindSockAddr;

		SockaddrFromIpEndpointName( bindSockAddr, localEndpoint );

        if (bind(socket_, (struct sockaddr *)&bindSockAddr, sizeof(bindSockAddr)) < 0) {

            throw std::runtime_error("unable to bind udp socket\n");

        }

		isBound_ = true;

	}

	bool IsBound() const { return isBound_; }

    int ReceiveFrom( IpEndpointName& remoteEndpoint, char *data, int size )

	{

		assert( isBound_ );

		struct sockaddr_in fromAddr;

        socklen_t fromAddrLen = sizeof(fromAddr);

        int result = recvfrom(socket_, data, size, 0,

                    (struct sockaddr *) &fromAddr, (socklen_t*)&fromAddrLen);

		if( result < 0 )

			return 0;

		remoteEndpoint.address = ntohl(fromAddr.sin_addr.s_addr);

		remoteEndpoint.port = ntohs(fromAddr.sin_port);

		return result;

	}

	int Socket() { return socket_; }

};

UdpSocket::UdpSocket()

{

	impl_ = new Implementation();

}

UdpSocket::~UdpSocket()

{

	delete impl_;

}

IpEndpointName UdpSocket::LocalEndpointFor( const IpEndpointName& remoteEndpoint ) const

{

	return impl_->LocalEndpointFor( remoteEndpoint );

}

void UdpSocket::Connect( const IpEndpointName& remoteEndpoint )

{

	impl_->Connect( remoteEndpoint );

}

void UdpSocket::Send( const char *data, int size )

{

	impl_->Send( data, size );

}

void UdpSocket::SendTo( const IpEndpointName& remoteEndpoint, const char *data, int size )

{

	impl_->SendTo( remoteEndpoint, data, size );

}

void UdpSocket::Bind( const IpEndpointName& localEndpoint )

{

	impl_->Bind( localEndpoint );

}

bool UdpSocket::IsBound() const

{

	return impl_->IsBound();

}

int UdpSocket::ReceiveFrom( IpEndpointName& remoteEndpoint, char *data, int size )

{

	return impl_->ReceiveFrom( remoteEndpoint, data, size );

}

struct AttachedTimerListener{

	AttachedTimerListener( int id, int p, TimerListener *tl )

		: initialDelayMs( id )

		, periodMs( p )

		, listener( tl ) {}

	int initialDelayMs;

	int periodMs;

	TimerListener *listener;

};

static bool CompareScheduledTimerCalls( 

		const std::pair< double, AttachedTimerListener > & lhs, const std::pair< double, AttachedTimerListener > & rhs )

{

	return lhs.first < rhs.first;

}

SocketReceiveMultiplexer *multiplexerInstanceToAbortWithSigInt_ = 0;

extern "C" /*static*/ void InterruptSignalHandler( int );

/*static*/ void InterruptSignalHandler( int )

{

	multiplexerInstanceToAbortWithSigInt_->AsynchronousBreak();

	signal( SIGINT, SIG_DFL );

}

class SocketReceiveMultiplexer::Implementation{

	std::vector< std::pair< PacketListener*, UdpSocket* > > socketListeners_;

	std::vector< AttachedTimerListener > timerListeners_;

	volatile bool break_;

	int breakPipe_[2]; // [0] is the reader descriptor and [1] the writer

	double GetCurrentTimeMs() const

	{

		struct timeval t;

		gettimeofday( &t, 0 );

		return ((double)t.tv_sec*1000.) + ((double)t.tv_usec / 1000.);

	}

public:

    Implementation()

	{

		if( pipe(breakPipe_) != 0 )

			throw std::runtime_error( "creation of asynchronous break pipes failed\n" );

	}

    ~Implementation()

	{

		close( breakPipe_[0] );

		close( breakPipe_[1] );

	}

    void AttachSocketListener( UdpSocket *socket, PacketListener *listener )

	{

		assert( std::find( socketListeners_.begin(), socketListeners_.end(), std::make_pair(listener, socket) ) == socketListeners_.end() );

		// we don't check that the same socket has been added multiple times, even though this is an error

		socketListeners_.push_back( std::make_pair( listener, socket ) );

	}

    void DetachSocketListener( UdpSocket *socket, PacketListener *listener )

	{

		std::vector< std::pair< PacketListener*, UdpSocket* > >::iterator i = 

				std::find( socketListeners_.begin(), socketListeners_.end(), std::make_pair(listener, socket) );

		assert( i != socketListeners_.end() );

		socketListeners_.erase( i );

	}

    void AttachPeriodicTimerListener( int periodMilliseconds, TimerListener *listener )

	{

		timerListeners_.push_back( AttachedTimerListener( periodMilliseconds, periodMilliseconds, listener ) );

	}

	void AttachPeriodicTimerListener( int initialDelayMilliseconds, int periodMilliseconds, TimerListener *listener )

	{

		timerListeners_.push_back( AttachedTimerListener( initialDelayMilliseconds, periodMilliseconds, listener ) );

	}

    void DetachPeriodicTimerListener( TimerListener *listener )

	{

		std::vector< AttachedTimerListener >::iterator i = timerListeners_.begin();

		while( i != timerListeners_.end() ){

			if( i->listener == listener )

				break;

			++i;

		}

		assert( i != timerListeners_.end() );

		timerListeners_.erase( i );

	}

    void Run()

	{

		break_ = false;

		// configure the master fd_set for select()

		fd_set masterfds, tempfds;

		FD_ZERO( &masterfds );

		FD_ZERO( &tempfds );