Planet
navi homePPSaboutscreenshotsdownloaddevelopmentforum

source: code/trunk/src/libraries/network/MasterServerComm.cc @ 7801

Last change on this file since 7801 was 7801, checked in by dafrick, 13 years ago

Merging presentation2 branch back to trunk.

File size: 7.2 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 *      Sandro 'smerkli' Merkli
24 *   Co-authors:
25 *      ...
26 *
27 */
28
29#include "MasterServerComm.h"
30
31namespace orxonox
32{
33
34  MasterServerComm::MasterServerComm()
35  { /* nothing anymore, everything's been outsourced to
36     * the initialize method to facilitate debugging
37     */
38    /* register object in orxonox */
39  } 
40
41  int MasterServerComm::initialize()
42  {
43    /* initialize Enet */
44    if( enet_initialize () != 0 )
45    { COUT(1) << "An error occurred while initializing ENet.\n";
46      return 1;
47    }
48
49    /* initialize the event holder */
50//     this->event = (ENetEvent *)calloc( sizeof(ENetEvent), 1 );
51   
52
53    /* initiate the client */
54    this->client = enet_host_create( NULL /* create a client host */,
55        1,
56        2, /* allow up 2 channels to be used, 0 and 1 */
57        0, 
58        0 ); 
59
60    /* see if it worked */
61    if (this->client == NULL)
62    { COUT(1) << "An error occurred while trying to create an " 
63        << "ENet client host.\n";
64      return 1;
65    }
66
67    return 0;
68  }
69
70  MasterServerComm::~MasterServerComm()
71  {
72    /* destroy the enet facilities */
73    enet_host_destroy(this->client);
74  }
75
76  int MasterServerComm::connect( const char *address, unsigned int port )
77  {
78    /* Connect to address:port. */
79    enet_address_set_host( &this->address, address );
80    this->address.port = port;
81
82    /* Initiate the connection, allocating the two channels 0 and 1. */
83    this->peer = enet_host_connect(this->client, &this->address, 2, 0);   
84
85    if( this->peer == NULL )
86    { COUT(2) << "ERROR: No available peers for initiating an ENet"
87        << " connection.\n";
88      return -1;
89    }
90
91    /* Wait up to 2 seconds for the connection attempt to succeed. */
92    if (enet_host_service (this->client, &this->event, 500) > 0 &&
93        this->event.type == ENET_EVENT_TYPE_CONNECT )
94      COUT(3) << "Connection to master server succeeded.\n";
95    else
96    {
97      enet_peer_reset (this->peer);
98      COUT(2) << "ERROR: connection to " << address << " failed.\n";
99      return -1;
100    }
101
102    /* all fine */
103    return 0;
104  }
105 
106void MasterServerComm::update()
107{
108  while( enet_host_service( this->client, &this->event, 1 ) );
109}
110
111
112  int MasterServerComm::disconnect( void )
113  {
114    while( enet_host_service( this->client, &this->event, 1 ) );
115    enet_peer_disconnect( this->peer, 0 );
116
117    /* Allow up to 1 second for the disconnect to succeed
118     * and drop any packets received packets.
119     */
120    while (enet_host_service (this->client, &this->event, 1000) > 0)
121    {
122      switch (this->event.type)
123      {
124        case ENET_EVENT_TYPE_RECEIVE:
125          enet_packet_destroy (event.packet);
126          break;
127
128        case ENET_EVENT_TYPE_DISCONNECT:
129          COUT(4) << "Disconnect from master server successful.\n"; 
130          return 0;
131        default: break;
132      }
133    }
134
135    /* We've arrived here, so the disconnect attempt didn't
136     * succeed yet, hence: force the connection down.           
137     */
138    enet_peer_reset( this->peer );
139
140    /* done */
141    return 0;
142  }
143
144  /* NOTE this is to be reimplemented soon to return
145   * a structure containing
146   * - addrconv
147   * - the event
148   * so we can also make callbacks from objects
149   */
150  int MasterServerComm::pollForReply( int (*callback)( char*, ENetEvent* ),
151    int delayms )
152  { 
153    /* see whether anything happened */
154    /* WORK MARK REMOVE THIS OUTPUT */
155    COUT(2) << "polling masterserver...\n";
156
157    /* address buffer */
158    char *addrconv = NULL;
159    int retval = 0;
160
161    /* enet_host_service returns 0 if no event occured */
162    /* just newly set below test to >0 from >= 0, to be tested */
163    if( enet_host_service( this->client, &this->event, delayms ) > 0 )
164    { 
165      /* check what type of event it is and react accordingly */
166      switch (this->event.type)
167      { /* new connection, not supposed to happen. */
168        case ENET_EVENT_TYPE_CONNECT: break;
169
170        /* disconnect */
171        case ENET_EVENT_TYPE_DISCONNECT: /* ?? */ break;
172
173        /* incoming data */
174        case ENET_EVENT_TYPE_RECEIVE: 
175          addrconv = (char *) calloc( 50, 1 );
176          if( !addrconv ) 
177          { COUT(2) << "MasterServerComm.cc: Could not allocate memory!\n";
178            break;
179          }
180
181          /* resolve IP */
182          enet_address_get_host_ip( &(this->event.peer->address), 
183            addrconv, 49 );
184
185          /* DEBUG */
186          COUT(3) << "MasterServer Debug: A packet of length " 
187            << this->event.packet->dataLength
188            << " containing " << this->event.packet->data
189            << " was received from " << addrconv
190            << " on channel " << this->event.channelID;
191          /* END DEBUG */
192
193          /* call the supplied callback, if any. */
194          if( (*callback) != NULL )
195            retval = (*callback)( addrconv, &(this->event) );
196
197          /* clean up */
198          enet_packet_destroy( event.packet );
199          if( addrconv ) 
200            free( addrconv );
201
202          break;
203        default: break;
204      }
205
206      /* event handled, return 0 */
207      return retval;
208    }
209
210    /* show that no event occured */
211    return 0;
212  }
213
214  int MasterServerComm::sendRequest( const char *data )
215  {
216    /* send the data to the friend */
217    /* Create a reliable packet of size 7 containing "packet\0" */
218    ENetPacket * packet = enet_packet_create( data, 
219        strlen( data ) + 1, 
220        ENET_PACKET_FLAG_RELIABLE);
221
222    /* Send the packet to the peer over channel id 0. */
223    enet_peer_send (this->peer, 0, packet);
224
225    /* One could just use enet_host_service() instead. */
226    enet_host_flush( this->client );
227   
228    /* free the packet */
229    // PLEASE: never do this, because enet will free the packet once it's delivered. this will cause double frees
230//     enet_packet_destroy( packet );
231
232    /* all done. */
233    return 0;
234  }
235
236  int MasterServerComm::sendRequest( std::string data )
237  {
238    /* send the data to the friend */
239    /* Create a reliable packet of size 7 containing "packet\0" */
240    ENetPacket * packet = enet_packet_create( data.c_str(), 
241        data.length() + 1, 
242        ENET_PACKET_FLAG_RELIABLE);
243
244    /* Send the packet to the peer over channel id 0. */
245    enet_peer_send (this->peer, 0, packet);
246
247    /* One could just use enet_host_service() instead. */
248    enet_host_flush( this->client );
249    // PLEASE: never do this, because enet will free the packet once it's delivered. this will cause double frees
250//     enet_packet_destroy( packet );
251
252    /* all done. */
253    return 0;
254  }
255
256}
Note: See TracBrowser for help on using the repository browser.