source: code/branches/mergeFS18/src/libraries/network/LANDiscoverable.cc @ 12034

/*
 *   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:
 *      Oliver Scheuss
 *   Co-authors:
 *      ...
 *
 */

#include "LANDiscoverable.h"

#include <enet/enet.h>
#include <cassert>
#include <cstring>

#include "util/Output.h"
#include "packet/ServerInformation.h"
#include "core/config/ConfigValueIncludes.h"
#include "core/CoreIncludes.h"


namespace orxonox
{
  const char* LAN_DISCOVERY_MESSAGE = "Orxonox Client";
  const char* LAN_DISCOVERY_ACK     = "Orxonox Server Ack";

  RegisterClassNoArgs(LANDiscoverable);

  LANDiscoverable::LANDiscoverable() : bActive_(false), host_(nullptr)
  {
    /* register object in orxonox */
    RegisterObject(LANDiscoverable);

    this->setConfigValues();
  }

  void LANDiscoverable::setConfigValues()
  {
    /* update ownName string from orxonox.ini config file, if it
     * has changed.
     */
    SetConfigValueExternal(ownName, "Discovery", "ownName", "OrxServer");
  }

  LANDiscoverable::~LANDiscoverable()
  {
    if( this->host_ )
    {
      enet_host_destroy( this->host_ );
      assert( this->bActive_ );
    }
  }

  void LANDiscoverable::setActivity(bool bActive)
  {
    if( bActive == this->bActive_ )        // no change
      return;

    if( bActive )
    {
      ENetAddress bindAddress;
      memset(& bindAddress, 0, sizeof(ENetAddress));
      bindAddress.host = ENET_HOST_ANY;
      bindAddress.port = LAN_DISCOVERY_PORT;
      assert( this->host_ == nullptr );
      this->host_ = enet_host_create( &bindAddress, 10, 0, 0, 0 );
      if ( this->host_ == nullptr )
          orxout(internal_error, context::network) << "LANDiscoverable: host_ == nullptr" << endl;
    }
    else
    {
      enet_host_destroy( this->host_ );
      this->host_ = nullptr;
    }
      this->bActive_ = bActive;
  }

  void LANDiscoverable::update()
  {
    ENetEvent event;

    if( this->bActive_==false )
      return;
    assert(this->host_);

    while( enet_host_service( this->host_, &event, 0 ) > 0 )
    {
      switch(event.type)
      {
        case ENET_EVENT_TYPE_CONNECT:
            orxout(verbose, context::network) << "Received LAN discovery connect from client " << event.peer->host->receivedAddress << endl;
            break;
        case ENET_EVENT_TYPE_DISCONNECT:
        case ENET_EVENT_TYPE_NONE:
          break;
        case ENET_EVENT_TYPE_RECEIVE:
          if( strcmp( LAN_DISCOVERY_MESSAGE, (char*)event.packet->data ) == 0 )      // check for a suitable orxonox client
          {
            orxout(internal_info, context::network) << "Received LAN discovery message from client " << event.peer->host->receivedAddress << endl;
            packet::ServerInformation info;
            info.setServerName(this->ownName);
            info.setClientNumber(this->clientNumber);
            info.send(event.peer);
            enet_host_flush(this->host_);
          }
          break;
      }
    }
  }

}