Planet
navi homePPSaboutscreenshotsdownloaddevelopmentforum

source: orxonox.OLD/branches/proxy/src/lib/network/network_stream.cc @ 9652

Last change on this file since 9652 was 9652, checked in by patrick, 18 years ago

proxy server now temporarily does not open server ports (so i can test on one client). proxy server id bug fixed

File size: 39.1 KB
Line 
1/*
2   orxonox - the future of 3D-vertical-scrollers
3
4   Copyright (C) 2004 orx
5
6   This program is free software; you can redistribute it and/or modify
7   it under the terms of the GNU General Public License as published by
8   the Free Software Foundation; either version 2, or (at your option)
9   any later version.
10
11### File Specific:
12   main-programmer: Christoph Renner rennerc@ee.ethz.ch
13   co-programmer:   Patrick Boenzli  patrick@orxonox.ethz.ch
14
15     June 2006: finishing work on the network stream for pps presentation (rennerc@ee.ethz.ch)
16     July 2006: some code rearangement and integration of the proxy server mechanism (patrick@orxonox.ethz.ch)
17*/
18
19
20#define DEBUG_MODULE_NETWORK
21
22#include "proxy/proxy_control.h"
23
24#include "base_object.h"
25#include "network_protocol.h"
26#include "udp_socket.h"
27#include "udp_server_socket.h"
28#include "monitor/connection_monitor.h"
29#include "monitor/network_monitor.h"
30#include "synchronizeable.h"
31#include "ip.h"
32#include "network_game_manager.h"
33#include "shared_network_data.h"
34#include "message_manager.h"
35#include "preferences.h"
36#include "zip.h"
37
38#include "src/lib/util/loading/resource_manager.h"
39
40#include "network_log.h"
41
42#include "player_stats.h"
43
44#include "lib/util/loading/factory.h"
45
46#include "debug.h"
47#include "class_list.h"
48#include <algorithm>
49
50
51#include "network_stream.h"
52
53
54#include "converter.h"
55
56
57#define PACKAGE_SIZE  256
58
59
60/**
61 * empty constructor
62 */
63NetworkStream::NetworkStream()
64    : DataStream()
65{
66  this->init();
67  /* initialize the references */
68  this->pInfo->nodeType = NET_UNASSIGNED;
69}
70
71
72NetworkStream::NetworkStream( int nodeType)
73{
74  this->init();
75
76  this->pInfo->nodeType = nodeType;
77
78  switch( nodeType)
79  {
80    case NET_MASTER_SERVER:
81      // init the shared network data
82      SharedNetworkData::getInstance()->setHostID(NET_ID_MASTER_SERVER);
83      this->pInfo->userId = NET_ID_MASTER_SERVER;
84      this->pInfo->nodeType = NET_MASTER_SERVER;
85
86      break;
87    case NET_PROXY_SERVER_ACTIVE:
88      // init the shared network data
89      SharedNetworkData::getInstance()->setHostID(NET_ID_PROXY_SERVER_01);
90      this->pInfo->nodeType = NET_PROXY_SERVER_ACTIVE;
91
92      break;
93    case NET_PROXY_SERVER_PASSIVE:
94      // init the shared network data
95      SharedNetworkData::getInstance()->setHostID(NET_ID_PROXY_SERVER_01);
96      this->pInfo->nodeType = NET_PROXY_SERVER_PASSIVE;
97
98      break;
99    case NET_CLIENT:
100      SharedNetworkData::getInstance()->setHostID(NET_ID_UNASSIGNED);
101      this->pInfo->nodeType = NET_CLIENT;
102      break;
103  }
104
105  SharedNetworkData::getInstance()->setDefaultSyncStream(this);
106
107  // get the local ip address
108  IP ip("localhost", 0);
109  this->pInfo->ip = ip;
110}
111
112
113
114/**
115 * generic init functions
116 */
117void NetworkStream::init()
118{
119  /* set the class id for the base object */
120  this->setClassID(CL_NETWORK_STREAM, "NetworkStream");
121  this->clientSocket = NULL;
122  this->clientSoftSocket = NULL;
123  this->proxySocket = NULL;
124  this->networkGameManager = NULL;
125  this->networkMonitor = NULL;
126
127  this->pInfo = new PeerInfo();
128  this->pInfo->userId = NET_UID_UNASSIGNED;
129  this->pInfo->lastAckedState = 0;
130  this->pInfo->lastRecvedState = 0;
131  this->pInfo->bLocal = true;
132
133  this->bRedirect = false;
134
135  this->currentState = 0;
136  this->redirectionUID = NET_ID_MASTER_SERVER;
137
138  remainingBytesToWriteToDict = Preferences::getInstance()->getInt( "compression", "writedict", 0 );
139
140  assert( Zip::getInstance()->loadDictionary( "testdict" ) >= 0 );
141  this->dictClient = Zip::getInstance()->loadDictionary( "dict2pl_client" );
142  assert( this->dictClient >= 0 );
143  this->dictServer = Zip::getInstance()->loadDictionary( "dict2p_server" );
144  assert( this->dictServer >= 0 );
145}
146
147
148/**
149 * deconstructor
150 */
151NetworkStream::~NetworkStream()
152{
153  if ( this->clientSocket )
154  {
155    this->clientSocket->close();
156    delete this->clientSocket;
157    this->clientSocket = NULL;
158  }
159  if ( this->clientSoftSocket )
160  {
161    this->clientSoftSocket->close();
162    delete this->clientSoftSocket;
163    this->clientSoftSocket = NULL;
164  }
165  if ( this->proxySocket)
166  {
167    proxySocket->close();
168    delete proxySocket;
169    proxySocket = NULL;
170  }
171  for ( PeerList::iterator i = peers.begin(); i!=peers.end(); i++)
172  {
173    if ( i->second.socket )
174    {
175      i->second.socket->disconnectServer();
176      delete i->second.socket;
177      i->second.socket = NULL;
178    }
179
180    if ( i->second.handshake )
181    {
182      delete i->second.handshake;
183      i->second.handshake = NULL;
184    }
185
186    if ( i->second.connectionMonitor )
187    {
188      delete i->second.connectionMonitor;
189      i->second.connectionMonitor = NULL;
190    }
191  }
192  for ( SynchronizeableList::const_iterator it = getSyncBegin(); it != getSyncEnd(); it ++ )
193    (*it)->setNetworkStream( NULL );
194
195  if( this->pInfo)
196    delete this->pInfo;
197
198  if( this->networkMonitor)
199    delete this->networkMonitor;
200}
201
202
203/**
204 * establish a connection to a remote master server
205 * @param host: host name
206 * @param port: the port number
207 */
208void NetworkStream::connectToMasterServer(std::string host, int port)
209{
210  int node = NET_ID_MASTER_SERVER;
211  // this create the new node in the peers map
212  this->peers[node].socket = new UdpSocket( host, port );
213  this->peers[node].userId = NET_ID_MASTER_SERVER;
214
215  this->peers[node].nodeType = NET_MASTER_SERVER;
216  this->peers[node].connectionMonitor = new ConnectionMonitor( NET_ID_MASTER_SERVER );
217  this->peers[node].ip = this->peers[node].socket->getRemoteAddress();
218  this->peers[node].bLocal = true;
219}
220
221
222/**
223 * establish a connection to a remote proxy server
224 * @param host: host name
225 * @param port: the port number
226 */
227void NetworkStream::connectToProxyServer(int proxyId, std::string host, int port)
228{
229  PRINTF(0)("connect to proxy %s, this is proxyId %i\n", host.c_str(), proxyId);
230
231  // this creates the new proxyId in the peers map
232  this->peers[proxyId].socket = new UdpSocket( host, port );
233  this->peers[proxyId].userId = proxyId;
234
235  this->peers[proxyId].nodeType = NET_PROXY_SERVER_ACTIVE;
236  this->peers[proxyId].connectionMonitor = new ConnectionMonitor( proxyId );
237  this->peers[proxyId].ip = this->peers[proxyId].socket->getRemoteAddress();
238  this->peers[proxyId].bLocal = true;
239}
240
241
242/**
243 * create a server
244 * @param port: interface port for all clients
245 */
246void NetworkStream::createServer(int clientPort, int proxyPort, int clientSoftPort)
247{
248  PRINTF(0)(" Creating new Server: listening for clients on port %i and for proxies on port %i", clientPort, proxyPort);
249  this->clientSocket= new UdpServerSocket(clientPort);
250  this->clientSoftSocket= new UdpServerSocket(clientSoftPort);
251  this->proxySocket = new UdpServerSocket(proxyPort);
252}
253
254
255/**
256 * creates a new instance of the network game manager
257 */
258void NetworkStream::createNetworkGameManager()
259{
260  this->networkGameManager = NetworkGameManager::getInstance();
261
262  this->networkGameManager->setUniqueID( SharedNetworkData::getInstance()->getNewUniqueID() );
263  MessageManager::getInstance()->setUniqueID( SharedNetworkData::getInstance()->getNewUniqueID() );
264}
265
266
267/**
268 * starts the network handshake
269 * handsakes are always initialized from the client side first. this starts the handshake and therefore is only
270 * executed as client
271 * @param userId: start handshake for this user id (optional, default == 0)
272 */
273void NetworkStream::startHandshake(int userId)
274{
275  Handshake* hs = new Handshake(this->pInfo->nodeType);
276  // fake the unique id
277  hs->setUniqueID( userId);
278  assert( this->peers[userId].handshake == NULL );
279  this->peers[userId].handshake = hs;
280  this->peers[userId].bLocal = true;
281
282  // set the preferred nick name
283  hs->setPreferedNickName( Preferences::getInstance()->getString( "multiplayer", "nickname", "Player" ) );
284
285  PRINTF(0)("NetworkStream: Handshake created: %s\n", hs->getCName());
286}
287
288
289/**
290 * this functions connects a synchronizeable to the networkstream, therefore synchronizeing
291 * it all over the network and creating it on the other platforms (if and only if it is a
292 * server
293 * @param sync: the synchronizeable to add
294 */
295void NetworkStream::connectSynchronizeable(Synchronizeable& sync)
296{
297  this->synchronizeables.push_back(&sync);
298  sync.setNetworkStream( this );
299}
300
301
302/**
303 * removes the synchronizeable from the list of synchronized entities
304 * @param sync: the syncronizeable to remove
305 */
306void NetworkStream::disconnectSynchronizeable(Synchronizeable& sync)
307{
308  // removing the Synchronizeable from the List.
309  std::list<Synchronizeable*>::iterator disconnectSynchro = std::find(this->synchronizeables.begin(), this->synchronizeables.end(), &sync);
310  if (disconnectSynchro != this->synchronizeables.end())
311    this->synchronizeables.erase(disconnectSynchro);
312
313  oldSynchronizeables[sync.getUniqueID()] = SDL_GetTicks();
314}
315
316
317/**
318 * this is called to process data from the network socket to the synchronizeable and vice versa
319 */
320void NetworkStream::processData()
321{
322  // create the network monitor after all the init work and before there is any connection handlings
323  if( this->networkMonitor == NULL)
324  {
325    this->networkMonitor = new NetworkMonitor(this);
326    SharedNetworkData::getInstance()->setNetworkMonitor( this->networkMonitor);
327  }
328
329
330  int tick = SDL_GetTicks();
331
332  this->currentState++;
333  // there was a wrap around
334  if( this->currentState < 0)
335  {
336    PRINTF(1)("A wrap around in the state variable as occured. The server was running so long? Pls restart server or write a mail to the supporters!\n");
337  }
338
339  if ( SharedNetworkData::getInstance()->isMasterServer())
340  {
341    // execute everytthing the master server shoudl do
342    if ( this->clientSocket )
343      this->clientSocket->update();
344    if ( this->clientSoftSocket)
345      this->clientSoftSocket->update();
346    if( this->proxySocket)
347      this->proxySocket->update();
348
349    this->updateConnectionList();
350  }
351  else if( SharedNetworkData::getInstance()->isProxyServerActive())
352  {
353    //execute everything the proxy server should do
354    if ( this->clientSocket )
355      this->clientSocket->update();
356    if ( this->clientSoftSocket)
357      this->clientSoftSocket->update();
358    if( this->proxySocket)
359      this->proxySocket->update();
360
361    this->updateConnectionList();
362  }
363
364#warning make this more modular: every proxy/master server connection should be watched for termination
365  if( !SharedNetworkData::getInstance()->isMasterServer())
366  {
367    // check if the connection is ok else terminate and remove
368    if ( !peers.empty() && peers[NET_ID_MASTER_SERVER].socket &&
369          ( !peers[NET_ID_MASTER_SERVER].socket->isOk() ||
370          peers[NET_ID_MASTER_SERVER].connectionMonitor->hasTimedOut() ) )
371    {
372      this->handleDisconnect( NET_ID_MASTER_SERVER);
373      PRINTF(1)("lost connection to server\n");
374    }
375    // check if there is a redirection command
376    if( this->bRedirect)
377    {
378      this->handleReconnect( this->redirectionUID);
379    }
380  }
381
382  this->cleanUpOldSyncList();
383  this->handleHandshakes();
384
385  // update the network monitor
386  this->networkMonitor->process();
387
388  // order of up/downstream is important!!!!
389  // don't change it
390  this->handleDownstream( tick );
391  this->handleUpstream( tick );
392}
393
394
395/**
396 * @brief handles incoming connections
397 *
398 * if we are a NET_MASTER_SERVER or NET_PROXY_SERVER_ACTIVE update the connection list to accept new connections (clients)
399 * start and initialize the handsake for the new clients
400 */
401void NetworkStream::updateConnectionList( )
402{
403  //check for new connections
404
405  NetworkSocket* tempNetworkSocket = NULL;
406  int userId;
407
408  if( this->clientSocket != NULL)
409  {
410    tempNetworkSocket = this->clientSocket->getNewSocket();
411
412    // we got new NET_CLIENT connecting
413    if ( tempNetworkSocket )
414    {
415      // get a userId
416//       if ( freeSocketSlots.size() > 0 )
417//       {
418//         // this should never be called
419//         userId = freeSocketSlots.back();
420//         freeSocketSlots.pop_back();
421//       }
422//       else
423      {
424        // each server (proxy and master) have an address space for new network nodes of 1000 nodes
425        // the first NET_ID_PROXY_MAX are always reserved for proxy servers
426        userId = SharedNetworkData::getInstance()->getHostID() * 1000 + NET_ID_PROXY_MAX + 1;
427
428        for ( PeerList::iterator it = peers.begin(); it != peers.end(); it++ )
429          if ( it->first >= userId )
430            userId = it->first + 1;
431
432        // make sure that this server only uses an address space of 1000
433        assert( userId < (SharedNetworkData::getInstance()->getHostID() + 1) * 1000);
434      }
435      // this creates a new entry in the peers list
436      peers[userId].socket = tempNetworkSocket;
437      peers[userId].nodeType = NET_CLIENT;
438
439      // handle the newly connected client
440      this->handleConnect(userId);
441
442      PRINTF(0)("New Client: %d\n", userId);
443    }
444  }
445
446  if( this->clientSoftSocket != NULL)
447  {
448    tempNetworkSocket = this->clientSoftSocket->getNewSocket();
449
450    // we got new NET_CLIENT connecting
451    if ( tempNetworkSocket )
452    {
453
454      // this creates a new entry in the peers list
455      peers[userId].socket = tempNetworkSocket;
456      peers[userId].nodeType = NET_CLIENT;
457
458      // handle the newly connected client
459      this->handleSoftConnect(userId);
460
461      PRINTF(0)("New Client softly connected: %d :D\n", userId);
462    }
463  }
464
465
466  if( this->proxySocket != NULL)
467  {
468    tempNetworkSocket = this->proxySocket->getNewSocket();
469
470    // we got new NET_PROXY_SERVER_ACTIVE connecting
471    if ( tempNetworkSocket )
472    {
473      // determine the network node id
474//       if ( freeSocketSlots.size() > 0 )
475//       {
476//         userId = freeSocketSlots.back();
477//         freeSocketSlots.pop_back();
478//       }
479//       else
480      {
481        userId = 1;
482
483        for( int i = 0; i < NET_ID_PROXY_MAX; i++)
484        {
485          if( this->peers.find( i) != this->peers.end())
486            userId = i;
487          break;
488        }
489        userId++;
490
491//         for ( PeerList::iterator it = peers.begin(); it != peers.end(); it++ )
492//           if ( it->first >= userId )
493//             userId = it->first + 1;
494      }
495
496      // this creates a new entry in the peers list
497      peers[userId].socket = tempNetworkSocket;
498      peers[userId].nodeType = NET_PROXY_SERVER_ACTIVE;
499
500      // handle the newly connected proxy server
501      this->handleConnect(userId);
502
503      PRINTF(0)("New proxy connected: %d\n", userId);
504    }
505  }
506
507
508
509  //check if connections are ok else remove them
510  for ( PeerList::iterator it = peers.begin(); it != peers.end(); )
511  {
512    if (
513          it->second.socket &&
514          (
515            !it->second.socket->isOk()  ||
516            it->second.connectionMonitor->hasTimedOut()
517          )
518       )
519    {
520      std::string reason = "disconnected";
521      if ( it->second.connectionMonitor->hasTimedOut() )
522        reason = "timeout";
523      PRINTF(0)("Client is gone: %d (%s)\n", it->second.userId, reason.c_str());
524
525
526      this->handleDisconnect( it->second.userId);
527
528      if( SharedNetworkData::getInstance()->isProxyServerActive())
529        ProxyControl::getInstance()->signalLeaveClient(it->second.userId);
530
531      it++;
532      continue;
533    }
534
535    it++;
536  }
537}
538
539
540/**
541 * this handles new connections
542 * @param userId: the id of the new user node
543 */
544void NetworkStream::handleConnect( int userId)
545{
546  // create new handshake and init its variables
547  this->peers[userId].handshake = new Handshake(this->pInfo->nodeType, userId, this->networkGameManager->getUniqueID(), MessageManager::getInstance()->getUniqueID());
548  this->peers[userId].handshake->setUniqueID(userId);
549
550  this->peers[userId].connectionMonitor = new ConnectionMonitor( userId );
551  this->peers[userId].userId = userId;
552  this->peers[userId].bLocal = true;
553
554  PRINTF(0)("num sync: %d\n", synchronizeables.size());
555
556  // get the proxy server informations and write them to the handshake, if any (proxy)
557  assert( this->networkMonitor != NULL);
558  PeerInfo* pi = this->networkMonitor->getFirstChoiceProxy();
559  if( pi != NULL)
560  {
561    this->peers[userId].handshake->setProxy1Address( pi->ip);
562  }
563  pi = this->networkMonitor->getSecondChoiceProxy();
564  if( pi != NULL)
565    this->peers[userId].handshake->setProxy2Address( pi->ip);
566
567  // check if the connecting client should reconnect to a proxy server
568  if( SharedNetworkData::getInstance()->isMasterServer())
569  {
570    if( this->networkMonitor->isReconnectNextClient())
571    {
572      this->peers[userId].handshake->setRedirect(true);
573      PRINTF(0)("forwarding client to proxy server because this server is saturated\n");
574    }
575  }
576
577  // the connecting node of course is a client
578  this->peers[userId].ip = this->peers[userId].socket->getRemoteAddress();
579}
580
581
582/**
583 * this handles new soft connections
584 * @param userId: the id of the new user node
585 *
586 * soft connections are connections from clients that are already in the network and don't need a new handshake etc.
587 * the state of all entitites owned by userId are not deleted and stay
588 * soft connections can not be redirected therefore they are negotiated between the to parties
589 */
590void NetworkStream::handleSoftConnect( int userId)
591{
592  // create new handshake and init its variables
593  this->peers[userId].handshake = NULL;
594  this->peers[userId].handshake->setUniqueID(userId);
595
596  this->peers[userId].connectionMonitor = new ConnectionMonitor( userId );
597  this->peers[userId].userId = userId;
598  this->peers[userId].bLocal = true;
599
600  PRINTF(0)("num sync: %d\n", synchronizeables.size());
601
602  // the connecting node of course is a client
603  this->peers[userId].ip = this->peers[userId].socket->getRemoteAddress();
604}
605
606
607
608
609/**
610 * some debug output
611 */
612void NetworkStream::debug()
613{
614  if( SharedNetworkData::getInstance()->isMasterServer()) {
615    PRINT(0)(" Host ist Master Server with ID: %i\n", this->pInfo->userId);
616  }
617  else if( SharedNetworkData::getInstance()->isProxyServerActive()) {
618    PRINT(0)(" Host ist Proxy Server with ID: %i\n", this->pInfo->userId);
619  }
620  else {
621    PRINT(0)(" Host ist Client with ID: %i\n", this->pInfo->userId);
622  }
623
624  PRINT(0)(" Current number of connections is: %i\n", this->peers.size());
625  for ( PeerList::iterator it = peers.begin(); it != peers.end(); it++ )
626  {
627    PRINT(0)("  peers[%i] with uniqueId %i and address: %s\n", it->first, it->second.userId, it->second.ip.ipString().c_str());
628  }
629  PRINT(0)("\n\n");
630
631
632  PRINT(0)(" Got %i connected Synchronizeables, showing active Syncs:\n", this->synchronizeables.size());
633  for (SynchronizeableList::iterator it = synchronizeables.begin(); it!=synchronizeables.end(); it++)
634  {
635    if( (*it)->beSynchronized() == true)
636      PRINT(0)("  Synchronizeable of class: %s::%s, with unique ID: %i, Synchronize: %i\n", (*it)->getClassCName(), (*it)->getCName(),
637               (*it)->getUniqueID(), (*it)->beSynchronized());
638  }
639  PRINT(0)(" Maximal Connections: %i\n", SharedNetworkData::getInstance()->getMaxPlayer() );
640
641}
642
643
644/**
645 * @returns the number of synchronizeables registered to this stream
646 */
647int NetworkStream::getSyncCount()
648{
649  int n = 0;
650  for (SynchronizeableList::iterator it = synchronizeables.begin(); it!=synchronizeables.end(); it++)
651    if( (*it)->beSynchronized() == true)
652      ++n;
653
654  //return synchronizeables.size();
655  return n;
656}
657
658
659/**
660 * check if handshakes completed. if so create the network game manager else remove it again
661 */
662void NetworkStream::handleHandshakes( )
663{
664  for ( PeerList::iterator it = peers.begin(); it != peers.end(); it++ )
665  {
666    if ( it->second.handshake )
667    {
668      // handshake finished
669      if ( it->second.handshake->completed() )
670      {
671        //handshake is correct
672        if ( it->second.handshake->ok() )
673        {
674          // write the first informations into the node so they can be read from there for case differentiation
675          it->second.nodeType = it->second.handshake->getRemoteNodeType();
676
677          // the counter part didn't mark it free for deletion yet
678          if ( !it->second.handshake->allowDel() )
679          {
680            // make sure this is a connection:
681            // - client       <==> master server
682            // - proxy server <==> master server
683            if(  SharedNetworkData::getInstance()->isClient() ||
684                 SharedNetworkData::getInstance()->isProxyServerActive() &&
685                 SharedNetworkData::getInstance()->isUserMasterServer(it->second.userId))
686            {
687              PRINTF(4)("Handshake: i am in client role\n");
688
689              SharedNetworkData::getInstance()->setHostID( it->second.handshake->getHostId() );
690              this->pInfo->userId = SharedNetworkData::getInstance()->getHostID();
691
692//               it->second.ip = it->second.socket->getRemoteAddress();
693
694              it->second.nodeType = it->second.handshake->getRemoteNodeType();
695              // it->second.ip = it->second.socket->getRemoteAddress();
696              // add the new server to the nodes list (it can be a NET_MASTER_SERVER or NET_PROXY_SERVER)
697              this->networkMonitor->addNode(&it->second);
698              // get proxy 1 address and add it
699              this->networkMonitor->addNode(it->second.handshake->getProxy1Address(), NET_PROXY_SERVER_ACTIVE);
700              // get proxy 2 address and add it
701              this->networkMonitor->addNode(it->second.handshake->getProxy2Address(), NET_PROXY_SERVER_ACTIVE);
702
703              // now check if the server accepted the connection
704              if( SharedNetworkData::getInstance()->isClient() && it->second.handshake->redirect() )
705              {
706                this->bRedirect = true;
707              }
708
709              // create the new network game manager and init it
710              this->networkGameManager = NetworkGameManager::getInstance();
711              this->networkGameManager->setUniqueID( it->second.handshake->getNetworkGameManagerId() );
712              // init the new message manager
713              MessageManager::getInstance()->setUniqueID( it->second.handshake->getMessageManagerId() );
714            }
715
716            PRINT(0)("handshake finished id=%d\n", it->second.handshake->getNetworkGameManagerId());
717            it->second.handshake->del();
718
719          }
720          else
721          {
722            // handshake finished registring new player
723            if ( it->second.handshake->canDel() )
724            {
725
726              if (  SharedNetworkData::getInstance()->isMasterServer() )
727              {
728                it->second.ip = it->second.socket->getRemoteAddress();
729
730                this->networkMonitor->addNode(&it->second);
731
732                this->handleNewClient( it->second.userId );
733
734                if ( PlayerStats::getStats( it->second.userId ) && it->second.handshake->getPreferedNickName() != "" )
735                {
736                  PlayerStats::getStats( it->second.userId )->setNickName( it->second.handshake->getPreferedNickName() );
737                }
738              }
739              else if ( SharedNetworkData::getInstance()->isProxyServerActive() && it->second.isClient() )
740              {
741                PRINTF(4)("Handshake: Proxy in server role: connecting %i\n", it->second.userId);
742
743                it->second.ip = it->second.socket->getRemoteAddress();
744
745                this->networkMonitor->addNode(&it->second);
746
747                // work with the ProxyControl to init the new client
748                ProxyControl::getInstance()->signalNewClient( it->second.userId);
749
750                if ( PlayerStats::getStats( it->second.userId ) && it->second.handshake->getPreferedNickName() != "" )
751                {
752                  PlayerStats::getStats( it->second.userId )->setNickName( it->second.handshake->getPreferedNickName() );
753                }
754              }
755
756              PRINT(0)("handshake finished delete it\n");
757              delete it->second.handshake;
758              it->second.handshake = NULL;
759            }
760          }
761
762        }
763        else
764        {
765          PRINT(1)("handshake failed!\n");
766          it->second.socket->disconnectServer();
767        }
768      }
769    }
770  }
771}
772
773
774/**
775 * this functions handles a reconnect event received from the a NET_MASTER_SERVER or NET_PROXY_SERVER
776 * @param userId
777 */
778void NetworkStream::handleReconnect(int userId)
779{
780  this->bRedirect = false;
781#warning this peer will be created if it does not yet exist: dangerous
782  PeerInfo* pInfo = &this->peers[userId];
783
784  IP proxyIP;
785  if( this->networkMonitor->isForcedReconnection())
786    proxyIP = this->networkMonitor->getForcedReconnectionIP();
787  else
788    proxyIP = pInfo->handshake->getProxy1Address();
789
790  PRINTF(0)("===============================================\n");
791  PRINTF(0)("Client is redirected to the other proxy servers\n");
792  PRINTF(0)("  user id: %i\n", userId);
793  PRINTF(0)("  connecting to: %s\n", proxyIP.ipString().c_str());
794  PRINTF(0)("===============================================\n");
795
796  // flush the old synchronization states, since the numbering could be completely different
797  pInfo->lastAckedState = 0;
798  pInfo->lastRecvedState = 0;
799
800  // disconnect from the current server and reconnect to proxy server
801  this->handleDisconnect( userId);
802  this->connectToProxyServer(NET_ID_PROXY_SERVER_01, proxyIP.ipString(), 9999);
803//   this->connectToMasterServer(proxyIP.ipString(), 9999);
804  #warning the ports are not yet integrated correctly in the ip class
805
806  // and restart the handshake
807  this->startHandshake( userId);
808}
809
810
811
812/**
813 * reconnects to another server, with full handshake
814 * @param address of the new server
815 */
816void NetworkStream::reconnectToServer(IP address)
817{
818  ///TODO make a redirection struct and push it to the network monitor
819  this->networkMonitor->setForcedReconnection(address);
820
821  // reconnect (depending on how we are connected at the moment)
822  if ( peers.find( NET_ID_MASTER_SERVER) != peers.end() )
823    this->redirectionUID = NET_ID_MASTER_SERVER;
824  else if( peers.find( NET_ID_PROXY_SERVER_01) != peers.end() )
825    this->redirectionUID = NET_ID_PROXY_SERVER_01;
826
827  this->bRedirect = true;
828}
829
830
831
832/**
833 * softly reconnecting to another server
834 * @param address  of the new server
835 */
836void NetworkStream::softReconnectToServer(IP address)
837{
838  this->networkMonitor->setForcedReconnection(address);
839  this->handleReconnect( NET_ID_MASTER_SERVER);
840}
841
842
843/**
844 * handles the disconnect event
845 * @param userId id of the user to remove
846 */
847void NetworkStream::handleDisconnect( int userId )
848{
849  this->peers[userId].socket->disconnectServer();
850  delete this->peers[userId].socket;
851  this->peers[userId].socket = NULL;
852
853  if ( this->peers[userId].handshake )
854    delete this->peers[userId].handshake;
855  this->peers[userId].handshake = NULL;
856
857  if ( this->peers[userId].connectionMonitor )
858    delete this->peers[userId].connectionMonitor;
859  this->peers[userId].connectionMonitor = NULL;
860
861
862  for ( SynchronizeableList::iterator it2 = synchronizeables.begin(); it2 != synchronizeables.end(); it2++ )  {
863    (*it2)->cleanUpUser( userId );
864  }
865
866  if( SharedNetworkData::getInstance()->isMasterServer())
867    NetworkGameManager::getInstance()->signalLeftPlayer(userId);
868
869  this->freeSocketSlots.push_back( userId );
870
871  this->networkMonitor->removeNode(&this->peers[userId]);
872  this->peers.erase( userId);
873}
874
875
876
877/**
878 * handle upstream network traffic
879 * @param tick: seconds elapsed since last update
880 */
881void NetworkStream::handleUpstream( int tick )
882{
883  int offset;
884  int n;
885
886  for ( PeerList::reverse_iterator peer = peers.rbegin(); peer != peers.rend(); peer++ )
887  {
888    offset = INTSIZE; // reserve enough space for the packet length
889
890    // continue with the next peer if this peer has no socket assigned (therefore no network)
891    if ( !peer->second.socket )
892      continue;
893
894    // header informations: current state
895    n = Converter::intToByteArray( currentState, buf + offset, UDP_PACKET_SIZE - offset );
896    assert( n == INTSIZE );
897    offset += n;
898
899    // header informations: last acked state
900    n = Converter::intToByteArray( peer->second.lastAckedState, buf + offset, UDP_PACKET_SIZE - offset );
901    assert( n == INTSIZE );
902    offset += n;
903
904    // header informations: last recved state
905    n = Converter::intToByteArray( peer->second.lastRecvedState, buf + offset, UDP_PACKET_SIZE - offset );
906    assert( n == INTSIZE );
907    offset += n;
908
909    // now write all synchronizeables in the packet
910    for ( SynchronizeableList::iterator it = synchronizeables.begin(); it != synchronizeables.end(); it++ )
911    {
912
913      int oldOffset = offset;
914      Synchronizeable & sync = **it;
915
916
917      // do not include synchronizeables with uninit id and syncs that don't want to be synchronized
918      if ( !sync.beSynchronized() || sync.getUniqueID() <= NET_UID_UNASSIGNED )
919        continue;
920
921      // if handshake not finished only sync handshake
922      if ( peer->second.handshake && sync.getLeafClassID() != CL_HANDSHAKE )
923        continue;
924
925      // if we are a server (both master and proxy servers) and this is not our handshake
926      if ( ( SharedNetworkData::getInstance()->isMasterServer() ||
927             SharedNetworkData::getInstance()->isProxyServerActive() &&  peer->second.isClient())
928             && sync.getLeafClassID() == CL_HANDSHAKE && sync.getUniqueID() != peer->second.userId )
929        continue;
930
931      /* list of synchronizeables that will never be synchronized over the network: */
932      // do not sync null parent
933      if ( sync.getLeafClassID() == CL_NULL_PARENT )
934        continue;
935
936
937      assert( sync.getLeafClassID() != 0);
938
939      assert( offset + INTSIZE <= UDP_PACKET_SIZE );
940
941      // server fakes uniqueid == 0 for handshake synchronizeable
942      if ( ( SharedNetworkData::getInstance()->isMasterServer() ||
943             SharedNetworkData::getInstance()->isProxyServerActive() &&  peer->second.isClient() ) &&
944             ( sync.getUniqueID() >= 1000 || sync.getUniqueID() <= SharedNetworkData::getInstance()->getMaxPlayer() + 1 + NET_ID_PROXY_MAX))
945        n = Converter::intToByteArray( 0, buf + offset, UDP_PACKET_SIZE - offset );
946      else
947        n = Converter::intToByteArray( sync.getUniqueID(), buf + offset, UDP_PACKET_SIZE - offset );
948
949
950      assert( n == INTSIZE );
951      offset += n;
952
953      // make space for packet size
954      offset += INTSIZE;
955
956      n = sync.getStateDiff( peer->second.userId, buf + offset, UDP_PACKET_SIZE-offset, currentState, peer->second.lastAckedState, -1000 );
957      offset += n;
958
959      assert( Converter::intToByteArray( n, buf + offset - n - INTSIZE, INTSIZE ) == INTSIZE );
960
961      // check if all data bytes == 0 -> remove data and the synchronizeable from the sync process since there is no update
962      // TODO not all synchronizeables like this maybe add Synchronizeable::canRemoveZeroDiff()
963      bool allZero = true;
964      for ( int i = 0; i < n; i++ )
965      {
966         if ( buf[i+oldOffset+2*INTSIZE] != 0 )
967           allZero = false;
968      }
969      // if there is no new data in this synchronizeable reset the data offset to the last state -> dont synchronizes
970      // data that hast not changed
971      if ( allZero )
972      {
973        offset = oldOffset;
974      }
975    } // all synchronizeables written
976
977
978
979    for ( SynchronizeableList::iterator it = synchronizeables.begin(); it != synchronizeables.end(); it++ )
980    {
981      Synchronizeable & sync = **it;
982
983      // again exclude all unwanted syncs
984      if ( !sync.beSynchronized() || sync.getUniqueID() <= NET_UID_UNASSIGNED)
985        continue;
986
987      sync.handleSentState( peer->second.userId, currentState, peer->second.lastAckedState );
988    }
989
990
991    assert( Converter::intToByteArray( offset, buf, INTSIZE ) == INTSIZE );
992
993    // now compress the data with the zip library
994    int compLength = 0;
995    if ( SharedNetworkData::getInstance()->isMasterServer() ||
996         SharedNetworkData::getInstance()->isProxyServerActive())
997      compLength = Zip::getInstance()->zip( buf, offset, compBuf, UDP_PACKET_SIZE, dictServer );
998    else
999      compLength = Zip::getInstance()->zip( buf, offset, compBuf, UDP_PACKET_SIZE, dictClient );
1000
1001    if ( compLength <= 0 )
1002    {
1003      PRINTF(1)("compression failed!\n");
1004      continue;
1005    }
1006
1007    assert( peer->second.socket->writePacket( compBuf, compLength ) );
1008
1009    if ( this->remainingBytesToWriteToDict > 0 )
1010      writeToNewDict( buf, offset, true );
1011
1012    peer->second.connectionMonitor->processUnzippedOutgoingPacket( tick, buf, offset, currentState );
1013    peer->second.connectionMonitor->processZippedOutgoingPacket( tick, compBuf, compLength, currentState );
1014
1015  }
1016}
1017
1018/**
1019 * handle downstream network traffic
1020 */
1021void NetworkStream::handleDownstream( int tick )
1022{
1023  int offset = 0;
1024
1025  int length = 0;
1026  int packetLength = 0;
1027  int compLength = 0;
1028  int uniqueId = 0;
1029  int state = 0;
1030  int ackedState = 0;
1031  int fromState = 0;
1032  int syncDataLength = 0;
1033
1034  for ( PeerList::iterator peer = peers.begin(); peer != peers.end(); peer++ )
1035  {
1036
1037    if ( !peer->second.socket )
1038      continue;
1039
1040    while ( 0 < (compLength = peer->second.socket->readPacket( compBuf, UDP_PACKET_SIZE )) )
1041    {
1042      peer->second.connectionMonitor->processZippedIncomingPacket( tick, compBuf, compLength );
1043
1044      packetLength = Zip::getInstance()->unZip( compBuf, compLength, buf, UDP_PACKET_SIZE );
1045
1046      if ( packetLength < 4*INTSIZE )
1047      {
1048        if ( packetLength != 0 )
1049          PRINTF(1)("got too small packet: %d\n", packetLength);
1050        continue;
1051      }
1052
1053      if ( this->remainingBytesToWriteToDict > 0 )
1054        writeToNewDict( buf, packetLength, false );
1055
1056      assert( Converter::byteArrayToInt( buf, &length ) == INTSIZE );
1057      assert( Converter::byteArrayToInt( buf + INTSIZE, &state ) == INTSIZE );
1058      assert( Converter::byteArrayToInt( buf + 2*INTSIZE, &fromState ) == INTSIZE );
1059      assert( Converter::byteArrayToInt( buf + 3*INTSIZE, &ackedState ) == INTSIZE );
1060      offset = 4*INTSIZE;
1061
1062      peer->second.connectionMonitor->processUnzippedIncomingPacket( tick, buf, packetLength, state, ackedState );
1063
1064
1065      //if this is an old state drop it
1066      if ( state <= peer->second.lastRecvedState )
1067        continue;
1068
1069      if ( packetLength != length )
1070      {
1071        PRINTF(1)("real packet length (%d) and transmitted packet length (%d) do not match!\n", packetLength, length);
1072        peer->second.socket->disconnectServer();
1073        continue;
1074      }
1075
1076      while ( offset + 2 * INTSIZE < length )
1077      {
1078        // read the unique id of the sync
1079        assert( offset > 0 );
1080        assert( Converter::byteArrayToInt( buf + offset, &uniqueId ) == INTSIZE );
1081        offset += INTSIZE;
1082
1083        // read the data length
1084        assert( Converter::byteArrayToInt( buf + offset, &syncDataLength ) == INTSIZE );
1085        offset += INTSIZE;
1086
1087        assert( syncDataLength > 0 );
1088        assert( syncDataLength < 10000 );
1089
1090        Synchronizeable * sync = NULL;
1091
1092        // look for the synchronizeable in question
1093        for ( SynchronizeableList::iterator it = synchronizeables.begin(); it != synchronizeables.end(); it++ )
1094        {
1095          // client thinks his handshake has a special id: hostId * 1000 (host id of this server)
1096          if ( (*it)->getUniqueID() == uniqueId ||                                          // so this client exists already go to sync work
1097                 ( uniqueId == 0  && (*it)->getUniqueID() == peer->second.userId ) )        // so this is a Handshake!
1098          {
1099            sync = *it;
1100            break;
1101          }
1102        }
1103
1104        // this synchronizeable does not yet exist! create it
1105        if ( sync == NULL )
1106        {
1107          PRINTF(0)("could not find sync with id %d. try to create it\n", uniqueId);
1108
1109          // if it is an old synchronizeable already removed, ignore it
1110          if ( oldSynchronizeables.find( uniqueId ) != oldSynchronizeables.end() )
1111          {
1112            offset += syncDataLength;
1113            continue;
1114          }
1115
1116          // if the node we got this unknown sync we ignore it if:
1117          //  - the remote host is a client
1118          //  - the remote host is a proxy server and we are master server
1119          // (since it has no rights to create a new sync)
1120          if ( peers[peer->second.userId].isClient() ||
1121               (peers[peer->second.userId].isProxyServerActive() && SharedNetworkData::getInstance()->isMasterServer()))
1122          {
1123            offset += syncDataLength;
1124            continue;
1125          }
1126
1127          int leafClassId;
1128          if ( INTSIZE > length - offset )
1129          {
1130            offset += syncDataLength;
1131            continue;
1132          }
1133
1134          Converter::byteArrayToInt( buf + offset, &leafClassId );
1135
1136          assert( leafClassId != 0 );
1137
1138
1139          BaseObject * b = NULL;
1140          /* These are some small exeptions in creation: Not all objects can/should be created via Factory */
1141          /* Exception 1: NullParent */
1142          if( leafClassId == CL_NULL_PARENT || leafClassId == CL_SYNCHRONIZEABLE || leafClassId == CL_NETWORK_GAME_MANAGER )
1143          {
1144            PRINTF(1)("Don't create Object with ID %x, ignored!\n", (int)leafClassId);
1145            offset += syncDataLength;
1146            continue;
1147          }
1148          else
1149            b = Factory::fabricate( (ClassID)leafClassId );
1150
1151          if ( !b )
1152          {
1153            PRINTF(1)("Could not fabricate Object with classID %x\n", leafClassId);
1154            offset += syncDataLength;
1155            continue;
1156          }
1157
1158          if ( b->isA(CL_SYNCHRONIZEABLE) )
1159          {
1160            sync = dynamic_cast<Synchronizeable*>(b);
1161            sync->setUniqueID( uniqueId );
1162            sync->setSynchronized(true);
1163
1164            PRINTF(0)("Fabricated %s with id %d\n", sync->getClassCName(), sync->getUniqueID());
1165          }
1166          else
1167          {
1168            PRINTF(1)("Class with ID %x is not a synchronizeable!\n", (int)leafClassId);
1169            delete b;
1170            offset += syncDataLength;
1171            continue;
1172          }
1173        }
1174
1175
1176        int n = sync->setStateDiff( peer->second.userId, buf+offset, syncDataLength, state, fromState );
1177        offset += n;
1178
1179      }
1180
1181      if ( offset != length )
1182      {
1183        PRINTF(0)("offset (%d) != length (%d)\n", offset, length);
1184        peer->second.socket->disconnectServer();
1185      }
1186
1187
1188      for ( SynchronizeableList::iterator it = synchronizeables.begin(); it != synchronizeables.end(); it++ )
1189      {
1190        Synchronizeable & sync = **it;
1191
1192        if ( !sync.beSynchronized() || sync.getUniqueID() <= NET_UID_UNASSIGNED )
1193          continue;
1194
1195        sync.handleRecvState( peer->second.userId, state, fromState );
1196      }
1197
1198      assert( peer->second.lastAckedState <= ackedState );
1199      peer->second.lastAckedState = ackedState;
1200
1201      assert( peer->second.lastRecvedState < state );
1202      peer->second.lastRecvedState = state;
1203
1204    }
1205
1206  }
1207
1208}
1209
1210/**
1211 * is executed when a handshake has finished
1212 */
1213void NetworkStream::handleNewClient( int userId )
1214{
1215  // init and assign the message manager
1216  MessageManager::getInstance()->initUser( userId );
1217  // do all game relevant stuff here
1218  networkGameManager->signalNewPlayer( userId );
1219}
1220
1221
1222/**
1223 * removes old items from oldSynchronizeables
1224 */
1225void NetworkStream::cleanUpOldSyncList( )
1226{
1227  int now = SDL_GetTicks();
1228
1229  for ( std::map<int,int>::iterator it = oldSynchronizeables.begin(); it != oldSynchronizeables.end();  )
1230  {
1231    if ( it->second < now - 10*1000 )
1232    {
1233      std::map<int,int>::iterator delIt = it;
1234      it++;
1235      oldSynchronizeables.erase( delIt );
1236      continue;
1237    }
1238    it++;
1239  }
1240}
1241
1242/**
1243 * writes data to DATA/dicts/newdict
1244 * @param data pointer to data
1245 * @param length length
1246 */
1247void NetworkStream::writeToNewDict( byte * data, int length, bool upstream )
1248{
1249  if ( remainingBytesToWriteToDict <= 0 )
1250    return;
1251
1252  if ( length > remainingBytesToWriteToDict )
1253    length = remainingBytesToWriteToDict;
1254
1255  std::string fileName = ResourceManager::getInstance()->getDataDir();
1256  fileName += "/dicts/newdict";
1257
1258  if ( upstream )
1259    fileName += "_upstream";
1260  else
1261    fileName += "_downstream";
1262
1263  FILE * f = fopen( fileName.c_str(), "a" );
1264
1265  if ( !f )
1266  {
1267    PRINTF(2)("could not open %s\n", fileName.c_str());
1268    remainingBytesToWriteToDict = 0;
1269    return;
1270  }
1271
1272  if ( fwrite( data, 1, length, f ) != length )
1273  {
1274    PRINTF(2)("could not write to file\n");
1275    fclose( f );
1276    return;
1277  }
1278
1279  fclose( f );
1280
1281  remainingBytesToWriteToDict -= length;
1282}
1283
1284
1285
1286
1287
1288
Note: See TracBrowser for help on using the repository browser.