source: code/branches/masterserver/src/modules/masterserver/MasterServerComm.cc @ 7630

/*
 *   ORXONOX - the hottest 3D action shooter ever to exist
 *                    > www.orxonox.net <
 *
 *
 *   License notice:
 *
 *   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.
 *
 *   Author:
 *      Sandro 'smerkli' Merkli
 *   Co-authors:
 *      ...
 *
 */

#include <cstdlib>
#include <cstdio>
#include <cstring>
#include <enet/enet.h>
#include "MasterServerComm.h"

MasterServerComm::MasterServerComm()
{
  /* initialize Enet */
  if (enet_initialize () != 0)
  { fprintf (stderr, "An error occurred while initializing ENet.\n");
    exit(EXIT_FAILURE);
  }
  atexit (enet_deinitialize);


  /* initiate the client */
  this->client = enet_host_create( NULL /* create a client host */,
      1,
      2, /* allow up 2 channels to be used, 0 and 1 */
      0, 
      0 ); 

  /* see if it worked */
  if (this->client == NULL)
  { fprintf (stderr, 
      "An error occurred while trying to create an ENet client host.\n");
    exit (EXIT_FAILURE);
  }
}

MasterServerComm::~MasterServerComm()
{
  enet_host_destroy(this->client);
}

int 
MasterServerComm::connect( char *address, unsigned int port )
{
  /* Connect to address:port. */
  enet_address_set_host( &this->address, address );
  this->address.port = port;

  /* Initiate the connection, allocating the two channels 0 and 1. */
  this->peer = enet_host_connect(this->client, &this->address, 2, 0);    

  if (this->peer == NULL )
  { fprintf( stderr, 
        "No available peers for initiating an ENet connection.\n");
    //exit (EXIT_FAILURE);
    return -1;
  }

  /* Wait up to 2 seconds for the connection attempt to succeed. */
  if (enet_host_service (this->client, &this->event, 2000) > 0 &&
      this->event.type == ENET_EVENT_TYPE_CONNECT )
    fprintf( stdout, "Connection to server succeeded." );
  else
  {
    enet_peer_reset (this->peer);
    fprintf( stdout, "Connection to %s failed.", address );
    //exit(EXIT_FAILURE);
    return -1;
  }

  return 0;
}

int 
MasterServerComm::pollForReply( int (*callback)( char*, ENetEvent* ) )
{ 
  if( enet_host_service( this->client, &this->event, 100 ) >= 0 )
  { 
    char *addrconv = NULL;
    /* check what type of event it is and react accordingly */
    switch (this->event.type)
    { /* new connection, not supposed to happen. */
      case ENET_EVENT_TYPE_CONNECT: break;

      /* disconnect */
      case ENET_EVENT_TYPE_DISCONNECT: 
        /* ?? */ break;

        /* incoming data */
      case ENET_EVENT_TYPE_RECEIVE: 
        addrconv = (char *) calloc( 50, 1 );
        enet_address_get_host_ip( &(this->event.peer->address), addrconv, 49 );

        /* DEBUG */
        printf( "A packet of length %u containing %s was "
          "received from %s on channel %u.\n",
          this->event.packet->dataLength,
          this->event.packet->data,
          addrconv,
          this->event.channelID );
        /* END DEBUG */

        /* call the supplied callback, if any. */
        if( (*callback) != NULL )
          (*callback)( addrconv, &(this->event) );

        enet_packet_destroy( event.packet );
        if( addrconv ) free( addrconv );
        break;
      default: break;
    }
  }
  
  return 0;
}

int 
MasterServerComm::sendRequest( char *data )
{
  /* send the data to the friend */
  /* Create a reliable packet of size 7 containing "packet\0" */
  ENetPacket * packet = enet_packet_create( data, 
      strlen( data ) + 1, 
      ENET_PACKET_FLAG_RELIABLE);

  /* Send the packet to the peer over channel id 0. */
  enet_peer_send (this->peer, 0, packet);

  /* One could just use enet_host_service() instead. */
  enet_host_flush( this->client );
  if( packet ) free( packet );
}

/* sample callback to output debugging info. */
int callb( char *addr, ENetEvent *ev )
{ 
  printf( "A packet of length %u containing %s was "
      "received from %s on channel %u.\n",
      ev->packet->dataLength,
      ev->packet->data,
      addr,
      ev->channelID );
  return 0;
}

/* small testing implementation */
int
main( int argc, char *argv[] )
{
  /* setup object and connect */
  MasterServerComm msc = MasterServerComm();
  if( msc.connect( argv[1], 1234 ) )
    exit(EXIT_FAILURE);
  
  /* send some data and poll for replies */
  char *theinput = (char *)calloc( 100,1 );
  while( true )
  { 
    fgets( theinput, 90, stdin );
    if( !strncmp( theinput, "quit", 4 ) )
      break;

    msc.sendRequest( theinput );
    msc.pollForReply( &callb );
  }

}