Planet
navi homePPSaboutscreenshotsdownloaddevelopmentforum

source: code/branches/ipv6/src/libraries/network/ClientConnection.cc @ 7395

Last change on this file since 7395 was 7294, checked in by adrfried, 14 years ago

made enet 1.3 compatible

  • Property svn:eol-style set to native
File size: 4.4 KB
Line 
1/*
2 *   ORXONOX - the hottest 3D action shooter ever to exist
3 *                    > www.orxonox.net <
4 *
5 *
6 *   License notice:
7 *
8 *   This program is free software; you can redistribute it and/or
9 *   modify it under the terms of the GNU General Public License
10 *   as published by the Free Software Foundation; either version 2
11 *   of the License, or (at your option) any later version.
12 *
13 *   This program is distributed in the hope that it will be useful,
14 *   but WITHOUT ANY WARRANTY; without even the implied warranty of
15 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16 *   GNU General Public License for more details.
17 *
18 *   You should have received a copy of the GNU General Public License
19 *   along with this program; if not, write to the Free Software
20 *   Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
21 *
22 *   Author:
23 *      Oliver Scheuss
24 *   Co-authors:
25 *      ...
26 *
27 */
28
29#include "ClientConnection.h"
30
31#include <cassert>
32#define WIN32_LEAN_AND_MEAN
33#include <enet/enet.h>
34#include "util/Debug.h"
35
36namespace orxonox
37{
38  const unsigned int NETWORK_CLIENT_WAIT_TIME = 1;
39  const unsigned int NETWORK_CLIENT_CONNECTION_TIMEOUT = 3000; //millisecs
40  const unsigned int NETWORK_CLIENT_MAX_CONNECTIONS = 1;
41  const unsigned int NETWORK_CLIENT_CHANNELS = 1;
42
43
44  ClientConnection::ClientConnection():
45    established_(false),
46    server_(NULL)
47  {
48    this->serverAddress_ = new ENetAddress();
49    //set standard address and port
50    enet_address_set_host(this->serverAddress_, "127.0.0.1");
51    serverAddress_->port = NETWORK_PORT;
52  }
53
54  ClientConnection::~ClientConnection(){
55    if(this->established_)
56      closeConnection();
57    delete this->serverAddress_; // surely was created
58  }
59
60  void ClientConnection::setServerAddress( const std::string& serverAddress ) {
61    enet_address_set_host (this->serverAddress_, serverAddress.c_str());
62  }
63
64  void ClientConnection::setPort( unsigned int port ) {
65    this->serverAddress_->port = port;
66  }
67
68  bool ClientConnection::establishConnection()
69  {
70    ENetEvent event;
71
72    this->host_ = enet_host_create(NULL, NETWORK_CLIENT_MAX_CONNECTIONS, 0, 0, 0);
73    if ( this->host_ == NULL )
74    {
75      COUT(2) << "ClientConnection: host_ == NULL" << std::endl;
76      // error handling
77      return false;
78    }
79    this->server_ = enet_host_connect(this->host_, serverAddress_, NETWORK_CLIENT_CHANNELS, 0);
80    if ( this->server_==NULL )
81    {
82      COUT(2) << "ClientConnection: server == NULL" << std::endl;
83      // error handling
84      return false;
85    }
86    // handshake
87    for( unsigned int i=0; i<NETWORK_CLIENT_CONNECTION_TIMEOUT/NETWORK_CLIENT_WAIT_TIME; i++ )
88    {
89      if( enet_host_service(this->host_, &event, NETWORK_CLIENT_WAIT_TIME)>=0 && event.type == ENET_EVENT_TYPE_CONNECT )
90      {
91        this->established_=true;
92        return true;
93      }
94    }
95    COUT(1) << "Could not connect to server" << endl;
96    return false;
97  }
98
99  bool ClientConnection::closeConnection() {
100    ENetEvent event;
101
102    if ( !this->established_ )
103      return true;
104    this->established_ = false;
105    enet_peer_disconnect(this->server_, 0);
106    for( unsigned int i=0; i<NETWORK_CLIENT_CONNECTION_TIMEOUT/NETWORK_CLIENT_WAIT_TIME; i++)
107    {
108      if ( enet_host_service(this->host_, &event, NETWORK_CLIENT_WAIT_TIME) >= 0 )
109      {
110        switch (event.type)
111        {
112          case ENET_EVENT_TYPE_NONE:
113          case ENET_EVENT_TYPE_CONNECT:
114            break;
115          case ENET_EVENT_TYPE_RECEIVE:
116            enet_packet_destroy(event.packet);
117            break;
118          case ENET_EVENT_TYPE_DISCONNECT:
119            COUT(4) << "received disconnect confirmation from server" << endl;
120            this->connectionClosed();
121            return true;
122        }
123      }
124    }
125    enet_peer_reset( this->server_ );
126    this->connectionClosed();
127    return false;
128  }
129
130
131  bool ClientConnection::addPacket(ENetPacket *packet) {
132    assert( this->server_ );
133    assert( packet );
134    return Connection::addPacket( packet, this->server_ );
135  }
136
137  void ClientConnection::addPeer(ENetEvent* event)
138  {
139    assert(0);
140  }
141  void ClientConnection::removePeer(ENetEvent* event)
142  {
143    this->established_=false;
144    COUT(1) << "Received disconnect Packet from Server!" << endl;
145        // server closed the connection
146    this->connectionClosed();
147  }
148
149  uint32_t ClientConnection::getRTT()
150  {
151    if (server_)
152        return server_->roundTripTime;
153    else
154        return 0;
155  }
156
157}
Note: See TracBrowser for help on using the repository browser.