Planet
navi homePPSaboutscreenshotsdownloaddevelopmentforum

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

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

more security added by checking also the send packages for leafclassid==0

File size: 28.8 KB
RevLine 
[5566]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:
[9246]12   main-programmer: Christoph Renner rennerc@ee.ethz.ch
13   co-programmer:   Patrick Boenzli  boenzlip@orxonox.ethz.ch
[9347]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 (boenzlip@ee.ethz.ch)
[5566]17*/
18
19
20#define DEBUG_MODULE_NETWORK
21
[5747]22
[5647]23#include "base_object.h"
[5731]24#include "network_protocol.h"
[7954]25#include "udp_socket.h"
26#include "udp_server_socket.h"
[9347]27#include "monitor/connection_monitor.h"
28#include "monitor/network_monitor.h"
[5647]29#include "synchronizeable.h"
[9347]30#include "ip.h"
[6341]31#include "network_game_manager.h"
[6959]32#include "shared_network_data.h"
[7954]33#include "message_manager.h"
34#include "preferences.h"
35#include "zip.h"
[6341]36
[7954]37#include "src/lib/util/loading/resource_manager.h"
38
39#include "network_log.h"
40
[9235]41#include "player_stats.h"
[7954]42
43#include "lib/util/loading/factory.h"
44
[5649]45#include "debug.h"
[6139]46#include "class_list.h"
[6144]47#include <algorithm>
[5647]48
[9347]49
[5566]50#include "network_stream.h"
51
[9347]52
[5594]53using namespace std;
54
[5595]55
[5747]56#define PACKAGE_SIZE  256
[5647]57
[5747]58
[9347]59/**
60 * empty constructor
61 */
[5800]62NetworkStream::NetworkStream()
[5996]63    : DataStream()
[5647]64{
65  this->init();
[5648]66  /* initialize the references */
[9347]67  this->pInfo->nodeType = NET_CLIENT;
[5647]68}
69
[6695]70
[9246]71/**
[9347]72 * start as a client, connect to a server
[9246]73 *  @param host: host name (address)
74 *  @param port: port number
75 */
[7954]76NetworkStream::NetworkStream( std::string host, int port )
[5996]77{
78  this->init();
[9347]79  // init the peers[0] the server to ceonnect to
[7954]80  this->peers[0].socket = new UdpSocket( host, port );
81  this->peers[0].userId = 0;
[9347]82  this->peers[0].nodeType = NET_MASTER_SERVER;
[7954]83  this->peers[0].connectionMonitor = new ConnectionMonitor( 0 );
[9347]84  this->peers[0].ip = this->peers[0].socket->getRemoteAddress();
85  // and set also the localhost
86  this->pInfo->nodeType = NET_CLIENT;
87  // get the local ip address
88  IPaddress ip;
89  SDLNet_ResolveHost( &ip, NULL, port );
90  this->pInfo->ip = ip;
[5996]91}
92
93
[9246]94/**
95 * start as a server
96 *  @param port: at this port
97 */
[7954]98NetworkStream::NetworkStream( int port )
[5647]99{
100  this->init();
[7954]101  this->serverSocket = new UdpServerSocket(port);
[9347]102  this->pInfo->nodeType = NET_MASTER_SERVER;
103  // get the local ip address
104  IPaddress ip;
105  SDLNet_ResolveHost( &ip, NULL, port );
106  this->pInfo->ip = ip;
[5649]107}
108
109
[9246]110/**
111 * generic init functions
112 */
[5647]113void NetworkStream::init()
114{
115  /* set the class id for the base object */
116  this->setClassID(CL_NETWORK_STREAM, "NetworkStream");
[6139]117  this->serverSocket = NULL;
[6341]118  this->networkGameManager = NULL;
[9347]119  this->networkMonitor = NULL;
[9246]120
[9347]121  this->pInfo = new PeerInfo();
122  this->pInfo->userId = 0;
123  this->pInfo->lastAckedState = 0;
124  this->pInfo->lastRecvedState = 0;
125
126
127  this->currentState = 0;
128
[7954]129  remainingBytesToWriteToDict = Preferences::getInstance()->getInt( "compression", "writedict", 0 );
[9246]130
[8623]131  assert( Zip::getInstance()->loadDictionary( "testdict" ) >= 0 );
132  this->dictClient = Zip::getInstance()->loadDictionary( "dict2pl_client" );
133  assert( this->dictClient >= 0 );
134  this->dictServer = Zip::getInstance()->loadDictionary( "dict2p_server" );
135  assert( this->dictServer >= 0 );
[5594]136}
137
[5647]138
[9246]139/**
140 * deconstructor
141 */
[5566]142NetworkStream::~NetworkStream()
[5598]143{
[6139]144  if ( this->serverSocket )
145  {
146    serverSocket->close();
147    delete serverSocket;
[8228]148    serverSocket = NULL;
[6139]149  }
[7954]150  for ( PeerList::iterator i = peers.begin(); i!=peers.end(); i++)
[6139]151  {
[7954]152    if ( i->second.socket )
[6139]153    {
[7954]154      i->second.socket->disconnectServer();
155      delete i->second.socket;
156      i->second.socket = NULL;
[6139]157    }
[9246]158
[7954]159    if ( i->second.handshake )
[6139]160    {
[7954]161      delete i->second.handshake;
162      i->second.handshake = NULL;
[6139]163    }
[9246]164
[8623]165    if ( i->second.connectionMonitor )
166    {
167      delete i->second.connectionMonitor;
168      i->second.connectionMonitor = NULL;
169    }
[6139]170  }
[8228]171  for ( SynchronizeableList::const_iterator it = getSyncBegin(); it != getSyncEnd(); it ++ )
172    (*it)->setNetworkStream( NULL );
[9347]173
174  if( this->pInfo)
175    delete this->pInfo;
176
177  if( this->networkMonitor)
178    delete this->networkMonitor;
[5598]179}
180
[5996]181
[9246]182/**
183 * creates a new instance of the network game manager
184 */
[6695]185void NetworkStream::createNetworkGameManager()
186{
187  this->networkGameManager = NetworkGameManager::getInstance();
188  // setUniqueID( maxCon+2 ) because we need one id for every handshake
189  // and one for handshake to reject client maxCon+1
[7954]190  this->networkGameManager->setUniqueID( SharedNetworkData::getInstance()->getNewUniqueID() );
191  MessageManager::getInstance()->setUniqueID( SharedNetworkData::getInstance()->getNewUniqueID() );
[6695]192}
193
194
[9246]195/**
196 * starts the network handshake
[9347]197 * handsakes are always initialized from the client side first. this starts the handshake and therefore is only
198 * executed as client
[9246]199 */
[6695]200void NetworkStream::startHandshake()
201{
[9347]202  Handshake* hs = new Handshake(this->pInfo->nodeType);
[6695]203  hs->setUniqueID( 0 );
[7954]204  assert( peers[0].handshake == NULL );
205  peers[0].handshake = hs;
[9246]206
[9347]207  // set the preferred nick name
[9235]208  hs->setPreferedNickName( Preferences::getInstance()->getString( "multiplayer", "nickname", "Player" ) );
[9246]209
[7954]210  PRINTF(0)("NetworkStream: Handshake created: %s\n", hs->getName());
[6695]211}
212
213
[9246]214/**
215 * this functions connects a synchronizeable to the networkstream, therefore synchronizeing
216 * it all over the network and creating it on the other platforms (if and only if it is a
217 * server
218 */
[5996]219void NetworkStream::connectSynchronizeable(Synchronizeable& sync)
220{
[6139]221  this->synchronizeables.push_back(&sync);
222  sync.setNetworkStream( this );
223
[9347]224//   this->bActive = true;
[5996]225}
226
[6695]227
[9246]228/**
229 * removes the synchronizeable from the list of synchronized entities
230 */
[6139]231void NetworkStream::disconnectSynchronizeable(Synchronizeable& sync)
232{
[6144]233  // removing the Synchronizeable from the List.
234  std::list<Synchronizeable*>::iterator disconnectSynchro = std::find(this->synchronizeables.begin(), this->synchronizeables.end(), &sync);
235  if (disconnectSynchro != this->synchronizeables.end())
236    this->synchronizeables.erase(disconnectSynchro);
[9246]237
[7954]238  oldSynchronizeables[sync.getUniqueID()] = SDL_GetTicks();
[6139]239}
240
241
[9246]242/**
243 * this is called to process data from the network socket to the synchronizeable and vice versa
244 */
[5604]245void NetworkStream::processData()
246{
[9347]247  // create the network monitor after all the init work and before there is any connection handlings
248  if( this->networkMonitor == NULL)
249    this->networkMonitor = new NetworkMonitor(this);
250
251
[8068]252  int tick = SDL_GetTicks();
[9246]253
[9347]254  this->currentState++;
255  // there was a wrap around
256  if( this->currentState < 0)
257  {
258    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");
259  }
[9246]260
[9347]261  if ( this->pInfo->nodeType == NET_MASTER_SERVER )
[7954]262  {
[9347]263    // execute everytthing the master server shoudl do
[7954]264    if ( serverSocket )
265      serverSocket->update();
[9246]266
[6139]267    this->updateConnectionList();
[7954]268  }
[9347]269  else if( this->pInfo->nodeType == NET_PROXY_SERVER_ACTIVE)
270  {
271    // execute everything the proxy server should do
272  }
[6139]273  else
274  {
[9246]275    // check if the connection is ok else terminate and remove
[7954]276    if ( peers[0].socket && ( !peers[0].socket->isOk() || peers[0].connectionMonitor->hasTimedOut() ) )
[6139]277    {
278      PRINTF(1)("lost connection to server\n");
[5741]279
[7954]280      peers[0].socket->disconnectServer();
281      delete peers[0].socket;
282      peers[0].socket = NULL;
[6498]283
[7954]284      if ( peers[0].handshake )
285        delete peers[0].handshake;
286      peers[0].handshake = NULL;
[9246]287
[8623]288      if ( peers[0].connectionMonitor )
289        delete peers[0].connectionMonitor;
290      peers[0].connectionMonitor = NULL;
[6139]291    }
292  }
293
[7954]294  cleanUpOldSyncList();
295  handleHandshakes();
[9246]296
[9347]297  // update the network monitor
298  this->networkMonitor->process();
299
[7954]300  // order of up/downstream is important!!!!
301  // don't change it
[8068]302  handleDownstream( tick );
303  handleUpstream( tick );
[7954]304}
305
[9246]306
307/**
[9347]308 * if we are a NET_MASTER_SERVER or NET_PROXY_SERVER_ACTIVE update the connection list to accept new
309 * connections (clients) also start the handsake for the new clients
[9246]310 */
[7954]311void NetworkStream::updateConnectionList( )
312{
313  //check for new connections
314
315  NetworkSocket* tempNetworkSocket = serverSocket->getNewSocket();
316
[9246]317  // we got new network node
[7954]318  if ( tempNetworkSocket )
[6139]319  {
[7954]320    int clientId;
[9246]321    // if there is a list of free client id slots, take these
322    if ( freeSocketSlots.size() > 0 )
[6139]323    {
[7954]324      clientId = freeSocketSlots.back();
325      freeSocketSlots.pop_back();
[9246]326    }
327    else
[7954]328    {
329      clientId = 1;
[9246]330
[7954]331      for ( PeerList::iterator it = peers.begin(); it != peers.end(); it++ )
332        if ( it->first >= clientId )
333          clientId = it->first + 1;
[9347]334    }
[9246]335
[9347]336    peers[clientId].socket = tempNetworkSocket;
337    // create new handshake and init its variables
338    peers[clientId].handshake = new Handshake(this->pInfo->nodeType, clientId, this->networkGameManager->getUniqueID(), MessageManager::getInstance()->getUniqueID() );
339    peers[clientId].handshake->setUniqueID(clientId);
[9246]340
[9347]341    peers[clientId].connectionMonitor = new ConnectionMonitor( clientId );
342    peers[clientId].userId = clientId;
[6341]343
[9347]344    PRINTF(0)("num sync: %d\n", synchronizeables.size());
345
346    // get the proxy server informations and write them to the handshake, if any (proxy)
347    assert( this->networkMonitor != NULL);
348    PeerInfo* pi = this->networkMonitor->getFirstChoiceProxy();
349    if( pi != NULL)
[7954]350    {
[9347]351      peers[clientId].handshake->setProxy1Address( pi->ip);
352      PRINTF(0)("proxy1 ip set to: %s\n", pi->ip.ipString().c_str());
[9349]353      PRINTF(0)(" proxy1 ip : %i, port: %i\n", pi->ip.host(), pi->ip.port());
[7954]354    }
[6498]355
[9347]356    pi = this->networkMonitor->getSecondChoiceProxy();
357    if( pi != NULL)
358      peers[clientId].handshake->setProxy2Address( pi->ip);
359
360    // check if the connecting client should reconnect to a proxy server
361    peers[clientId].handshake->setRedirect(this->networkMonitor->isReconnectNextClient());
362
363    // the connecting node of course is a client
364    peers[clientId].nodeType = NET_CLIENT;
365    peers[clientId].ip = peers[clientId].socket->getRemoteAddress();
366
367
368    // check if there are too many clients connected (DEPRECATED: new: the masterserver sends a list of proxy servers)
369//     if ( clientId > SharedNetworkData::getInstance()->getMaxPlayer() )
370//     {
371// //       peers[clientId].handshake->setRedirect(true);
372// //
373// //       peers[clientId].handshake->doReject( "too many connections" );
374//       PRINTF(0)("Will reject client %d because there are to many connections!\n", clientId);
375//     }
376//     else
377//     {
378//       PRINTF(0)("New Client: %d\n", clientId);
379//     }
380    PRINTF(0)("New Client: %d\n", clientId);
381
382
[7954]383  }
[6341]384
[9246]385
386
[7954]387  //check if connections are ok else remove them
[8228]388  for ( PeerList::iterator it = peers.begin(); it != peers.end(); )
[7954]389  {
[9246]390    if (
[7954]391          it->second.socket &&
[9246]392          (
[7954]393            !it->second.socket->isOk()  ||
394            it->second.connectionMonitor->hasTimedOut()
395          )
396       )
397    {
398      std::string reason = "disconnected";
399      if ( it->second.connectionMonitor->hasTimedOut() )
400        reason = "timeout";
401      PRINTF(0)("Client is gone: %d (%s)\n", it->second.userId, reason.c_str());
402
[9246]403
404      // clean up the network data
[7954]405      it->second.socket->disconnectServer();
406      delete it->second.socket;
407      it->second.socket = NULL;
408
[8623]409      if ( it->second.connectionMonitor )
410        delete it->second.connectionMonitor;
411      it->second.connectionMonitor = NULL;
[9246]412
[7954]413      if ( it->second.handshake )
414        delete it->second.handshake;
415      it->second.handshake = NULL;
[9246]416
[7954]417      for ( SynchronizeableList::iterator it2 = synchronizeables.begin(); it2 != synchronizeables.end(); it2++ )
418      {
419        (*it2)->cleanUpUser( it->second.userId );
[6139]420      }
[7954]421
422      NetworkGameManager::getInstance()->signalLeftPlayer(it->second.userId);
423
424      freeSocketSlots.push_back( it->second.userId );
[9246]425
[8228]426      PeerList::iterator delit = it;
427      it++;
[9246]428
[8228]429      peers.erase( delit );
[9246]430
[8228]431      continue;
[6139]432    }
[9246]433
[8228]434    it++;
[6139]435  }
436
437
[7954]438}
[5800]439
[9246]440
[7954]441void NetworkStream::debug()
442{
[9347]443  if( this->isMasterServer())
444    PRINT(0)(" Host ist Server with ID: %i\n", this->pInfo->userId);
[7954]445  else
[9347]446    PRINT(0)(" Host ist Client with ID: %i\n", this->pInfo->userId);
[6695]447
[7954]448  PRINT(0)(" Got %i connected Synchronizeables, showing active Syncs:\n", this->synchronizeables.size());
[6139]449  for (SynchronizeableList::iterator it = synchronizeables.begin(); it!=synchronizeables.end(); it++)
[5996]450  {
[7954]451    if( (*it)->beSynchronized() == true)
452      PRINT(0)("  Synchronizeable of class: %s::%s, with unique ID: %i, Synchronize: %i\n", (*it)->getClassName(), (*it)->getName(),
453               (*it)->getUniqueID(), (*it)->beSynchronized());
454  }
[9347]455  PRINT(0)(" Maximal Connections: %i\n", SharedNetworkData::getInstance()->getMaxPlayer() );
[6959]456
[7954]457}
[6959]458
459
[9246]460/**
461 * @returns the number of synchronizeables registered to this stream
462 */
[7954]463int NetworkStream::getSyncCount()
464{
465  int n = 0;
466  for (SynchronizeableList::iterator it = synchronizeables.begin(); it!=synchronizeables.end(); it++)
467    if( (*it)->beSynchronized() == true)
468      ++n;
[5730]469
[7954]470  //return synchronizeables.size();
471  return n;
472}
[6139]473
[9246]474
[7954]475/**
[9246]476 * check if handshakes completed. if so create the network game manager else remove it again
[7954]477 */
478void NetworkStream::handleHandshakes( )
479{
480  for ( PeerList::iterator it = peers.begin(); it != peers.end(); it++ )
481  {
482    if ( it->second.handshake )
483    {
[9347]484      // handshake finished
[7954]485      if ( it->second.handshake->completed() )
486      {
[9347]487        //handshake is correct
[7954]488        if ( it->second.handshake->ok() )
[6341]489        {
[7954]490          if ( !it->second.handshake->allowDel() )
[6139]491          {
[9347]492            if ( this->pInfo->nodeType == NET_CLIENT )
[6868]493            {
[7954]494              SharedNetworkData::getInstance()->setHostID( it->second.handshake->getHostId() );
[9347]495              this->pInfo->userId = SharedNetworkData::getInstance()->getHostID();
[7954]496
[9347]497              it->second.nodeType = it->second.handshake->getRemoteNodeType();
498              it->second.ip = it->second.socket->getRemoteAddress();
499              // add the new server to the nodes list (it can be a NET_MASTER_SERVER or NET_PROXY_SERVER)
500              this->networkMonitor->addNode(&it->second);
501              // get proxy 1 address and add it
502              this->networkMonitor->addNode(it->second.handshake->getProxy1Address(), NET_PROXY_SERVER_ACTIVE);
503              PRINTF(0)("proxy1 ip read to: %s\n", it->second.handshake->getProxy1Address().ipString().c_str());
[9349]504              PRINTF(0)(" proxy1 ip : %i, port: %i\n", it->second.handshake->getProxy1Address().host(), it->second.handshake->getProxy1Address().port());
[9347]505              // get proxy 2 address and add it
506              this->networkMonitor->addNode(it->second.handshake->getProxy2Address(), NET_PROXY_SERVER_ACTIVE);
507
508              // now check if the server accepted the connection
509              if( it->second.handshake->redirect())
510                this->handleReconnect( it->second.userId);
511
512              // create the new network game manager and init it
[7954]513              this->networkGameManager = NetworkGameManager::getInstance();
514              this->networkGameManager->setUniqueID( it->second.handshake->getNetworkGameManagerId() );
[9347]515              // init the new message manager
[7954]516              MessageManager::getInstance()->setUniqueID( it->second.handshake->getMessageManagerId() );
[6868]517            }
[7954]518
[9246]519
[7954]520            PRINT(0)("handshake finished id=%d\n", it->second.handshake->getNetworkGameManagerId());
521
522            it->second.handshake->del();
[6139]523          }
524          else
525          {
[9347]526            // handshake finished registring new player
[7954]527            if ( it->second.handshake->canDel() )
[6868]528            {
[9347]529              if ( this->pInfo->nodeType == NET_MASTER_SERVER )
[7954]530              {
[9347]531                it->second.nodeType = it->second.handshake->getRemoteNodeType();
532                it->second.ip = it->second.socket->getRemoteAddress();
[9246]533
[9347]534                this->networkMonitor->addNode(&it->second);
535
536                this->handleNewClient( it->second.userId );
537
[9235]538                if ( PlayerStats::getStats( it->second.userId ) && it->second.handshake->getPreferedNickName() != "" )
539                {
540                  PlayerStats::getStats( it->second.userId )->setNickName( it->second.handshake->getPreferedNickName() );
541                }
[7954]542              }
[9246]543
[7954]544              PRINT(0)("handshake finished delete it\n");
545              delete it->second.handshake;
546              it->second.handshake = NULL;
[6868]547            }
[6139]548          }
[7954]549
[6139]550        }
551        else
552        {
[7954]553          PRINT(1)("handshake failed!\n");
554          it->second.socket->disconnectServer();
[6139]555        }
[7954]556      }
[6139]557    }
[5996]558  }
[7954]559}
[5741]560
[9246]561
[7954]562/**
[9347]563 * this functions handles a reconnect event received from the a NET_MASTER_SERVER or NET_PROXY_SERVER
564 */
565void NetworkStream::handleReconnect(int userId)
566{
567  PRINTF(0)("===============================================\n");
568  PRINTF(0)("Client is redirected to the other proxy servers\n");
569  PRINTF(0)("===============================================\n");
570
571  PeerInfo* pInfo = &this->peers[userId];
572
573  // reject the server
574  pInfo->handshake->doReject( "redirected to different server");
575
576  // flush the old synchronization states, since the numbering could be completely different
577  pInfo->lastAckedState = 0;
578  pInfo->lastRecvedState = 0;
579  // not sure if this works as expected
580  if( pInfo->handshake)
581    delete pInfo->handshake;
582
583  // disconnect from the current server and reconnect to proxy server
584  pInfo->socket->reconnectToServer( pInfo->handshake->getProxy1Address().ipString(), pInfo->handshake->getProxy1Address().port());
585
586  // and restart the handshake
587  this->startHandshake();
588}
589
590
591/**
[7954]592 * handle upstream network traffic
593 */
[8068]594void NetworkStream::handleUpstream( int tick )
[7954]595{
596  int offset;
597  int n;
[9246]598
[8068]599  for ( PeerList::reverse_iterator peer = peers.rbegin(); peer != peers.rend(); peer++ )
[5802]600  {
[9246]601    offset = INTSIZE; // reserve enough space for the packet length
602
603    // continue with the next peer if this peer has no socket assigned (therefore no network)
[7954]604    if ( !peer->second.socket )
605      continue;
[9246]606
607    // header informations: current state
[7954]608    n = Converter::intToByteArray( currentState, buf + offset, UDP_PACKET_SIZE - offset );
609    assert( n == INTSIZE );
610    offset += n;
[9246]611
612    // header informations: last acked state
[7954]613    n = Converter::intToByteArray( peer->second.lastAckedState, buf + offset, UDP_PACKET_SIZE - offset );
614    assert( n == INTSIZE );
615    offset += n;
[9246]616
617    // header informations: last recved state
[7954]618    n = Converter::intToByteArray( peer->second.lastRecvedState, buf + offset, UDP_PACKET_SIZE - offset );
619    assert( n == INTSIZE );
620    offset += n;
[9246]621
622    // now write all synchronizeables in the packet
[7954]623    for ( SynchronizeableList::iterator it = synchronizeables.begin(); it != synchronizeables.end(); it++ )
[5810]624    {
[7954]625      int oldOffset = offset;
626      Synchronizeable & sync = **it;
[9246]627
628      // do not include synchronizeables with uninit id and syncs that don't want to be synchronized
[7954]629      if ( !sync.beSynchronized() || sync.getUniqueID() < 0 )
630        continue;
[5730]631
[9246]632      // if handshake not finished only sync handshake
[7954]633      if ( peer->second.handshake && sync.getLeafClassID() != CL_HANDSHAKE )
634        continue;
[9246]635
636      // if we are a server and this is not our handshake
[9347]637      if ( isMasterServer() && sync.getLeafClassID() == CL_HANDSHAKE && sync.getUniqueID() != peer->second.userId )
[7954]638        continue;
[9246]639
640      /* list of synchronizeables that will never be synchronized over the network: */
641      // do not sync null parent
[7954]642      if ( sync.getLeafClassID() == CL_NULL_PARENT )
643        continue;
[6139]644
[9355]645      assert( sync.getLeafClassID() != 0);
646
[7954]647      assert( offset + INTSIZE <= UDP_PACKET_SIZE );
[9246]648
649      // server fakes uniqueid == 0 for handshake
[9347]650      if ( this->isMasterServer() && sync.getUniqueID() < SharedNetworkData::getInstance()->getMaxPlayer() - 1 )
[7954]651        n = Converter::intToByteArray( 0, buf + offset, UDP_PACKET_SIZE - offset );
652      else
653        n = Converter::intToByteArray( sync.getUniqueID(), buf + offset, UDP_PACKET_SIZE - offset );
[9246]654
[7954]655      assert( n == INTSIZE );
656      offset += n;
[9246]657
658      // make space for size
[7954]659      offset += INTSIZE;
[6139]660
[7954]661      n = sync.getStateDiff( peer->second.userId, buf + offset, UDP_PACKET_SIZE-offset, currentState, peer->second.lastAckedState, -1000 );
662      offset += n;
[9246]663
[7954]664      assert( Converter::intToByteArray( n, buf + offset - n - INTSIZE, INTSIZE ) == INTSIZE );
[6341]665
[9246]666      // check if all data bytes == 0 -> remove data and the synchronizeable from the sync process since there is no update
667      // TODO not all synchronizeables like this maybe add Synchronizeable::canRemoveZeroDiff()
668      bool allZero = true;
669      for ( int i = 0; i < n; i++ )
670      {
671         if ( buf[i+oldOffset+2*INTSIZE] != 0 )
672           allZero = false;
673      }
674      // if there is no new data in this synchronizeable reset the data offset to the last state -> dont synchronizes
675      // data that hast not changed
676      if ( allZero )
677      {
678        offset = oldOffset;
679      }
680    } // all synchronizeables written
[6139]681
[9246]682
683
[7954]684    for ( SynchronizeableList::iterator it = synchronizeables.begin(); it != synchronizeables.end(); it++ )
685    {
686      Synchronizeable & sync = **it;
[9246]687
[7954]688      if ( !sync.beSynchronized() || sync.getUniqueID() < 0 )
689        continue;
[9246]690
[7954]691      sync.handleSentState( peer->second.userId, currentState, peer->second.lastAckedState );
692    }
[9246]693
694
[7954]695    assert( Converter::intToByteArray( offset, buf, INTSIZE ) == INTSIZE );
[9246]696
697    // now compress the data with the zip library
[8623]698    int compLength = 0;
[9347]699    if ( this->isMasterServer() )
[8623]700      compLength = Zip::getInstance()->zip( buf, offset, compBuf, UDP_PACKET_SIZE, dictServer );
701    else
702      compLength = Zip::getInstance()->zip( buf, offset, compBuf, UDP_PACKET_SIZE, dictClient );
[9246]703
[8623]704    if ( compLength <= 0 )
[7954]705    {
706      PRINTF(1)("compression failed!\n");
707      continue;
708    }
[9246]709
[7954]710    assert( peer->second.socket->writePacket( compBuf, compLength ) );
[9246]711
[7954]712    if ( this->remainingBytesToWriteToDict > 0 )
[8623]713      writeToNewDict( buf, offset, true );
[9246]714
[8068]715    peer->second.connectionMonitor->processUnzippedOutgoingPacket( tick, buf, offset, currentState );
716    peer->second.connectionMonitor->processZippedOutgoingPacket( tick, compBuf, compLength, currentState );
[9246]717
[5810]718  }
[6139]719}
720
[7954]721/**
722 * handle downstream network traffic
723 */
[8068]724void NetworkStream::handleDownstream( int tick )
[6139]725{
[7954]726  int offset = 0;
[9246]727
[7954]728  int length = 0;
729  int packetLength = 0;
730  int compLength = 0;
731  int uniqueId = 0;
732  int state = 0;
733  int ackedState = 0;
734  int fromState = 0;
735  int syncDataLength = 0;
[9246]736
[7954]737  for ( PeerList::iterator peer = peers.begin(); peer != peers.end(); peer++ )
[5810]738  {
[9246]739
[7954]740    if ( !peer->second.socket )
741      continue;
[5730]742
[7954]743    while ( 0 < (compLength = peer->second.socket->readPacket( compBuf, UDP_PACKET_SIZE )) )
[6139]744    {
[8068]745      peer->second.connectionMonitor->processZippedIncomingPacket( tick, compBuf, compLength );
[9246]746
[7954]747      packetLength = Zip::getInstance()->unZip( compBuf, compLength, buf, UDP_PACKET_SIZE );
[8623]748
[7954]749      if ( packetLength < 4*INTSIZE )
750      {
751        if ( packetLength != 0 )
752          PRINTF(1)("got too small packet: %d\n", packetLength);
753        continue;
754      }
[9246]755
[7954]756      if ( this->remainingBytesToWriteToDict > 0 )
[8623]757        writeToNewDict( buf, packetLength, false );
[9246]758
[7954]759      assert( Converter::byteArrayToInt( buf, &length ) == INTSIZE );
760      assert( Converter::byteArrayToInt( buf + INTSIZE, &state ) == INTSIZE );
761      assert( Converter::byteArrayToInt( buf + 2*INTSIZE, &fromState ) == INTSIZE );
762      assert( Converter::byteArrayToInt( buf + 3*INTSIZE, &ackedState ) == INTSIZE );
763      offset = 4*INTSIZE;
[9246]764
[8623]765      peer->second.connectionMonitor->processUnzippedIncomingPacket( tick, buf, packetLength, state, ackedState );
[6139]766
[9246]767
[9347]768      //if this is an old state drop it
[7954]769      if ( state <= peer->second.lastRecvedState )
770        continue;
[9246]771
[7954]772      if ( packetLength != length )
773      {
774        PRINTF(1)("real packet length (%d) and transmitted packet length (%d) do not match!\n", packetLength, length);
775        peer->second.socket->disconnectServer();
776        continue;
777      }
[9246]778
[7954]779      while ( offset + 2*INTSIZE < length )
780      {
781        assert( offset > 0 );
782        assert( Converter::byteArrayToInt( buf + offset, &uniqueId ) == INTSIZE );
783        offset += INTSIZE;
[9246]784
[7954]785        assert( Converter::byteArrayToInt( buf + offset, &syncDataLength ) == INTSIZE );
786        offset += INTSIZE;
[9246]787
[7954]788        assert( syncDataLength > 0 );
789        assert( syncDataLength < 10000 );
[9246]790
[7954]791        Synchronizeable * sync = NULL;
[9246]792
[7954]793        for ( SynchronizeableList::iterator it = synchronizeables.begin(); it != synchronizeables.end(); it++ )
[9246]794        {
[7954]795        //                                        client thinks his handshake has id 0!!!!!
796          if ( (*it)->getUniqueID() == uniqueId || ( uniqueId == 0 && (*it)->getUniqueID() == peer->second.userId ) )
797          {
798            sync = *it;
799            break;
800          }
801        }
[9246]802
[7954]803        if ( sync == NULL )
804        {
805          PRINTF(0)("could not find sync with id %d. try to create it\n", uniqueId);
806          if ( oldSynchronizeables.find( uniqueId ) != oldSynchronizeables.end() )
807          {
808            offset += syncDataLength;
809            continue;
810          }
[9246]811
[9347]812          if ( !peers[peer->second.userId].isMasterServer() )
[7954]813          {
814            offset += syncDataLength;
815            continue;
816          }
[9246]817
[7954]818          int leafClassId;
819          if ( INTSIZE > length - offset )
820          {
821            offset += syncDataLength;
822            continue;
823          }
[6139]824
[7954]825          Converter::byteArrayToInt( buf + offset, &leafClassId );
[9246]826
[7954]827          assert( leafClassId != 0 );
[9246]828
[7954]829          BaseObject * b = NULL;
830          /* These are some small exeptions in creation: Not all objects can/should be created via Factory */
831          /* Exception 1: NullParent */
832          if( leafClassId == CL_NULL_PARENT || leafClassId == CL_SYNCHRONIZEABLE || leafClassId == CL_NETWORK_GAME_MANAGER )
833          {
834            PRINTF(1)("Can not create Class with ID %x!\n", (int)leafClassId);
835            offset += syncDataLength;
836            continue;
837          }
838          else
839            b = Factory::fabricate( (ClassID)leafClassId );
[5800]840
[7954]841          if ( !b )
842          {
843            PRINTF(1)("Could not fabricate Object with classID %x\n", leafClassId);
844            offset += syncDataLength;
845            continue;
846          }
[5809]847
[7954]848          if ( b->isA(CL_SYNCHRONIZEABLE) )
849          {
850            sync = dynamic_cast<Synchronizeable*>(b);
851            sync->setUniqueID( uniqueId );
852            sync->setSynchronized(true);
[9246]853
[7954]854            PRINTF(0)("Fabricated %s with id %d\n", sync->getClassName(), sync->getUniqueID());
855          }
856          else
857          {
858            PRINTF(1)("Class with ID %x is not a synchronizeable!\n", (int)leafClassId);
859            delete b;
860            offset += syncDataLength;
861            continue;
862          }
863        }
[6139]864
[9246]865
866        int n = sync->setStateDiff( peer->second.userId, buf+offset, syncDataLength, state, fromState );
[7954]867        offset += n;
868        //NETPRINTF(0)("SSSSSEEEEETTTTT: %s %d\n",sync->getClassName(), n);
[6498]869
[7954]870      }
[9246]871
[7954]872      if ( offset != length )
[6139]873      {
[7954]874        PRINTF(0)("offset (%d) != length (%d)\n", offset, length);
875        peer->second.socket->disconnectServer();
[6139]876      }
[9246]877
878
[7954]879      for ( SynchronizeableList::iterator it = synchronizeables.begin(); it != synchronizeables.end(); it++ )
[6139]880      {
[7954]881        Synchronizeable & sync = **it;
[9246]882
[7954]883        if ( !sync.beSynchronized() || sync.getUniqueID() < 0 )
884          continue;
[9246]885
[7954]886        sync.handleRecvState( peer->second.userId, state, fromState );
[6139]887      }
[9246]888
[7954]889      assert( peer->second.lastAckedState <= ackedState );
890      peer->second.lastAckedState = ackedState;
[9246]891
[7954]892      assert( peer->second.lastRecvedState < state );
893      peer->second.lastRecvedState = state;
[8228]894
[6139]895    }
[9246]896
[6139]897  }
[9246]898
[7954]899}
[6139]900
[7954]901/**
902 * is executed when a handshake has finished
903 */
904void NetworkStream::handleNewClient( int userId )
905{
[9347]906  // init and assign the message manager
[7954]907  MessageManager::getInstance()->initUser( userId );
[9347]908  // do all game relevant stuff here
909  networkGameManager->signalNewPlayer( userId );
[9246]910
[9347]911  // register the new client at the network monitor
912//   this->networkMonitor->addClient();
[5604]913}
[6139]914
[9347]915
[7954]916/**
917 * removes old items from oldSynchronizeables
918 */
919void NetworkStream::cleanUpOldSyncList( )
[6139]920{
[7954]921  int now = SDL_GetTicks();
[9246]922
[7954]923  for ( std::map<int,int>::iterator it = oldSynchronizeables.begin(); it != oldSynchronizeables.end();  )
[6139]924  {
[7954]925    if ( it->second < now - 10*1000 )
926    {
927      std::map<int,int>::iterator delIt = it;
928      it++;
929      oldSynchronizeables.erase( delIt );
930      continue;
931    }
932    it++;
[6139]933  }
[7954]934}
935
936/**
937 * writes data to DATA/dicts/newdict
938 * @param data pointer to data
939 * @param length length
940 */
[8623]941void NetworkStream::writeToNewDict( byte * data, int length, bool upstream )
[7954]942{
943  if ( remainingBytesToWriteToDict <= 0 )
944    return;
[9246]945
[7954]946  if ( length > remainingBytesToWriteToDict )
947    length = remainingBytesToWriteToDict;
[9246]948
[7954]949  std::string fileName = ResourceManager::getInstance()->getDataDir();
950  fileName += "/dicts/newdict";
[9246]951
[8623]952  if ( upstream )
953    fileName += "_upstream";
954  else
955    fileName += "_downstream";
[9246]956
[7954]957  FILE * f = fopen( fileName.c_str(), "a" );
[9246]958
[7954]959  if ( !f )
[6139]960  {
[7954]961    PRINTF(2)("could not open %s\n", fileName.c_str());
962    remainingBytesToWriteToDict = 0;
[6139]963    return;
964  }
[9246]965
[7954]966  if ( fwrite( data, 1, length, f ) != length )
[6341]967  {
[7954]968    PRINTF(2)("could not write to file\n");
969    fclose( f );
[6341]970    return;
971  }
[9246]972
[7954]973  fclose( f );
[9246]974
975  remainingBytesToWriteToDict -= length;
[6139]976}
977
978
[6695]979
980
981
982
Note: See TracBrowser for help on using the repository browser.