/******************************************************************************
 * $Id$
 *
 * Copyright (c) 2005-2007 Transmission authors and contributors
 *
 * 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.
 *
 * 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 <string.h>
#include <stdio.h>

#include <sys/types.h>

#include "transmission.h"
#include "natpmp.h"
#include "net.h"
#include "peer-mgr.h"
#include "shared.h"
#include "trevent.h"
#include "upnp.h"
#include "utils.h"

struct tr_shared
{
    tr_handle * h;
    tr_timer  * pulseTimer;

    /* Incoming connections */
    int bindPort;
    int bindSocket;

    /* port forwarding */
    int isEnabled;
    int publicPort;
    tr_nat_traversal_status natStatus;
    tr_upnp * upnp;
    tr_natpmp  * natpmp;

    int isShuttingDown;
};

#define NATKEY "Port Mapping: "

/***
****
***/

static const char*
getNatStateStr( int state )
{
    switch( state )
    {
        case TR_NAT_TRAVERSAL_MAPPING:   return "mapping";
        case TR_NAT_TRAVERSAL_MAPPED:    return "mapped";
        case TR_NAT_TRAVERSAL_UNMAPPING: return "unmapping";
        case TR_NAT_TRAVERSAL_UNMAPPED:  return "unmapped";
        case TR_NAT_TRAVERSAL_ERROR:     return "error";
    }

    return "notfound";
}

static void
natPulse( tr_shared * s )
{
    tr_nat_traversal_status status;
    const int port = s->publicPort;
    const int isEnabled = s->isEnabled && !s->isShuttingDown;

    status = tr_natpmpPulse( s->natpmp, port, isEnabled );
    if( status == TR_NAT_TRAVERSAL_ERROR )
        status = tr_upnpPulse( s->upnp, port, isEnabled );
    if( status != s->natStatus ) {
        tr_inf( NATKEY "mapping state changed from '%s' to '%s'", getNatStateStr(s->natStatus), getNatStateStr(status) );
        s->natStatus = status;
    }
}

static void
incomingPeersPulse( tr_shared * s )
{
    if( s->bindSocket >= 0 && ( s->bindPort != s->publicPort ) )
    {
        tr_inf( NATKEY "closing port %d", s->bindPort );
        tr_netClose( s->bindSocket );
        s->bindSocket = -1;
    }

    if( s->bindPort != s->publicPort )
    {
        int socket;
        errno = 0;
        socket = tr_netBindTCP( s->publicPort );
        if( socket >= 0 ) {
            tr_inf( NATKEY "opened port %d to listen for incoming peer connections", s->publicPort );
            s->bindPort = s->publicPort;
            s->bindSocket = socket;
            listen( s->bindSocket, 5 );
        } else {
            tr_err( NATKEY "unable to open port %d to listen for incoming peer connections (errno is %d - %s)",
                    s->publicPort, errno, strerror(errno) );
            s->bindPort = -1;
            s->bindSocket = -1;
        }
    }
    
    for( ;; ) /* check for new incoming peer connections */    
    {
        int socket;
        uint16_t port;
        struct in_addr addr;

        if( s->bindSocket < 0 )
            break;

        socket = tr_netAccept( s->bindSocket, &addr, &port );
        if( socket < 0 )
            break;

        tr_peerMgrAddIncoming( s->h->peerMgr, &addr, port, socket );
    }
}

static int
sharedPulse( void * vshared )
{
    int keepPulsing = 1;
    tr_shared * shared = vshared;

    natPulse( shared );

    if( !shared->isShuttingDown )
    {
        incomingPeersPulse( shared );
    }
    else if( ( shared->natStatus == TR_NAT_TRAVERSAL_ERROR ) || ( shared->natStatus == TR_NAT_TRAVERSAL_UNMAPPED ) )
    {
        tr_dbg( NATKEY "port mapping shut down" );
        shared->h->shared = NULL;
        tr_netClose( shared->bindSocket );
        tr_natpmpClose( shared->natpmp );
        tr_upnpClose( shared->upnp );
        tr_free( shared );
        keepPulsing = 0;
    }

    return keepPulsing;
}

/***
****
***/

tr_shared *
tr_sharedInit( tr_handle * h )
{
    tr_shared * s = tr_new0( tr_shared, 1 );

    s->h            = h;
    s->publicPort   = -1;
    s->bindPort     = -1;
    s->bindSocket   = -1;
    s->natpmp       = tr_natpmpInit();
    s->upnp         = tr_upnpInit();
    s->pulseTimer   = tr_timerNew( h, sharedPulse, s, 500 );
    s->isEnabled    = 0;
    s->natStatus    = TR_NAT_TRAVERSAL_UNMAPPED;

    return s;
}

void
tr_sharedShuttingDown( tr_shared * s )
{
    s->isShuttingDown = 1;
}

void
tr_sharedSetPort( tr_shared * s, int port )
{
    tr_torrent * tor;

    s->publicPort = port;

    for( tor = s->h->torrentList; tor; tor = tor->next )
        tr_torrentChangeMyPort( tor );
}

int
tr_sharedGetPublicPort( const tr_shared * s )
{
    return s->publicPort;
}

void
tr_sharedTraversalEnable( tr_shared * s, int isEnabled )
{
    s->isEnabled = isEnabled;
}

int
tr_sharedTraversalStatus( const tr_shared * s )
{
    return s->natStatus;
}