| 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 |  */ | 
|---|
| 63 | NetworkStream::NetworkStream() | 
|---|
| 64 |     : DataStream() | 
|---|
| 65 | { | 
|---|
| 66 |   this->init(); | 
|---|
| 67 |   /* initialize the references */ | 
|---|
| 68 |   this->pInfo->nodeType = NET_UNASSIGNED; | 
|---|
| 69 | } | 
|---|
| 70 |  | 
|---|
| 71 |  | 
|---|
| 72 | NetworkStream::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 |   IPaddress ip; | 
|---|
| 109 |   SDLNet_ResolveHost( &ip, NULL, 0); | 
|---|
| 110 |   this->pInfo->ip = ip; | 
|---|
| 111 | } | 
|---|
| 112 |  | 
|---|
| 113 |  | 
|---|
| 114 |  | 
|---|
| 115 | /** | 
|---|
| 116 |  * generic init functions | 
|---|
| 117 |  */ | 
|---|
| 118 | void NetworkStream::init() | 
|---|
| 119 | { | 
|---|
| 120 |   /* set the class id for the base object */ | 
|---|
| 121 |   this->setClassID(CL_NETWORK_STREAM, "NetworkStream"); | 
|---|
| 122 |   this->clientSocket = NULL; | 
|---|
| 123 |   this->clientSoftSocket = NULL; | 
|---|
| 124 |   this->proxySocket = NULL; | 
|---|
| 125 |   this->networkGameManager = NULL; | 
|---|
| 126 |   this->networkMonitor = NULL; | 
|---|
| 127 |  | 
|---|
| 128 |   this->pInfo = new PeerInfo(); | 
|---|
| 129 |   this->pInfo->userId = NET_UID_UNASSIGNED; | 
|---|
| 130 |   this->pInfo->lastAckedState = 0; | 
|---|
| 131 |   this->pInfo->lastRecvedState = 0; | 
|---|
| 132 |   this->pInfo->bLocal = true; | 
|---|
| 133 |  | 
|---|
| 134 |   this->bRedirect = false; | 
|---|
| 135 |  | 
|---|
| 136 |   this->currentState = 0; | 
|---|
| 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 |  */ | 
|---|
| 151 | NetworkStream::~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 |  */ | 
|---|
| 208 | void 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 |  */ | 
|---|
| 227 | void 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 |  */ | 
|---|
| 246 | void 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 |  */ | 
|---|
| 258 | void 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 |  */ | 
|---|
| 273 | void 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 |  */ | 
|---|
| 295 | void 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 |  */ | 
|---|
| 306 | void 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 |  */ | 
|---|
| 320 | void 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( NET_ID_MASTER_SERVER); | 
|---|
| 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 |  */ | 
|---|
| 401 | void 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 |  | 
|---|
| 490 | //         for ( PeerList::iterator it = peers.begin(); it != peers.end(); it++ ) | 
|---|
| 491 | //           if ( it->first >= userId ) | 
|---|
| 492 | //             userId = it->first + 1; | 
|---|
| 493 |       } | 
|---|
| 494 |  | 
|---|
| 495 |       // this creates a new entry in the peers list | 
|---|
| 496 |       peers[userId].socket = tempNetworkSocket; | 
|---|
| 497 |       peers[userId].nodeType = NET_PROXY_SERVER_ACTIVE; | 
|---|
| 498 |  | 
|---|
| 499 |       // handle the newly connected proxy server | 
|---|
| 500 |       this->handleConnect(userId); | 
|---|
| 501 |  | 
|---|
| 502 |       PRINTF(0)("New proxy connected: %d\n", userId); | 
|---|
| 503 |     } | 
|---|
| 504 |   } | 
|---|
| 505 |  | 
|---|
| 506 |  | 
|---|
| 507 |  | 
|---|
| 508 |   //check if connections are ok else remove them | 
|---|
| 509 |   for ( PeerList::iterator it = peers.begin(); it != peers.end(); ) | 
|---|
| 510 |   { | 
|---|
| 511 |     if ( | 
|---|
| 512 |           it->second.socket && | 
|---|
| 513 |           ( | 
|---|
| 514 |             !it->second.socket->isOk()  || | 
|---|
| 515 |             it->second.connectionMonitor->hasTimedOut() | 
|---|
| 516 |           ) | 
|---|
| 517 |        ) | 
|---|
| 518 |     { | 
|---|
| 519 |       std::string reason = "disconnected"; | 
|---|
| 520 |       if ( it->second.connectionMonitor->hasTimedOut() ) | 
|---|
| 521 |         reason = "timeout"; | 
|---|
| 522 |       PRINTF(0)("Client is gone: %d (%s)\n", it->second.userId, reason.c_str()); | 
|---|
| 523 |  | 
|---|
| 524 |  | 
|---|
| 525 |       this->handleDisconnect( it->second.userId); | 
|---|
| 526 |  | 
|---|
| 527 |       if( SharedNetworkData::getInstance()->isProxyServerActive()) | 
|---|
| 528 |         ProxyControl::getInstance()->signalLeaveClient(it->second.userId); | 
|---|
| 529 |  | 
|---|
| 530 |       it++; | 
|---|
| 531 |       continue; | 
|---|
| 532 |     } | 
|---|
| 533 |  | 
|---|
| 534 |     it++; | 
|---|
| 535 |   } | 
|---|
| 536 | } | 
|---|
| 537 |  | 
|---|
| 538 |  | 
|---|
| 539 | /** | 
|---|
| 540 |  * this handles new connections | 
|---|
| 541 |  * @param userId: the id of the new user node | 
|---|
| 542 |  */ | 
|---|
| 543 | void NetworkStream::handleConnect( int userId) | 
|---|
| 544 | { | 
|---|
| 545 |   // create new handshake and init its variables | 
|---|
| 546 |   this->peers[userId].handshake = new Handshake(this->pInfo->nodeType, userId, this->networkGameManager->getUniqueID(), MessageManager::getInstance()->getUniqueID()); | 
|---|
| 547 |   this->peers[userId].handshake->setUniqueID(userId); | 
|---|
| 548 |  | 
|---|
| 549 |   this->peers[userId].connectionMonitor = new ConnectionMonitor( userId ); | 
|---|
| 550 |   this->peers[userId].userId = userId; | 
|---|
| 551 |   this->peers[userId].bLocal = true; | 
|---|
| 552 |  | 
|---|
| 553 |   PRINTF(0)("num sync: %d\n", synchronizeables.size()); | 
|---|
| 554 |  | 
|---|
| 555 |   // get the proxy server informations and write them to the handshake, if any (proxy) | 
|---|
| 556 |   assert( this->networkMonitor != NULL); | 
|---|
| 557 |   PeerInfo* pi = this->networkMonitor->getFirstChoiceProxy(); | 
|---|
| 558 |   if( pi != NULL) | 
|---|
| 559 |   { | 
|---|
| 560 |     this->peers[userId].handshake->setProxy1Address( pi->ip); | 
|---|
| 561 |   } | 
|---|
| 562 |   pi = this->networkMonitor->getSecondChoiceProxy(); | 
|---|
| 563 |   if( pi != NULL) | 
|---|
| 564 |     this->peers[userId].handshake->setProxy2Address( pi->ip); | 
|---|
| 565 |  | 
|---|
| 566 |   // check if the connecting client should reconnect to a proxy server | 
|---|
| 567 |   if( SharedNetworkData::getInstance()->isMasterServer()) | 
|---|
| 568 |   { | 
|---|
| 569 |     if( this->networkMonitor->isReconnectNextClient()) | 
|---|
| 570 |     { | 
|---|
| 571 |       this->peers[userId].handshake->setRedirect(true); | 
|---|
| 572 |       PRINTF(0)("forwarding client to proxy server because this server is saturated\n"); | 
|---|
| 573 |     } | 
|---|
| 574 |   } | 
|---|
| 575 |  | 
|---|
| 576 |   // the connecting node of course is a client | 
|---|
| 577 |   this->peers[userId].ip = this->peers[userId].socket->getRemoteAddress(); | 
|---|
| 578 | } | 
|---|
| 579 |  | 
|---|
| 580 |  | 
|---|
| 581 | /** | 
|---|
| 582 |  * this handles new soft connections | 
|---|
| 583 |  * @param userId: the id of the new user node | 
|---|
| 584 |  * | 
|---|
| 585 |  * soft connections are connections from clients that are already in the network and don't need a new handshake etc. | 
|---|
| 586 |  * the state of all entitites owned by userId are not deleted and stay | 
|---|
| 587 |  * soft connections can not be redirected therefore they are negotiated between the to parties | 
|---|
| 588 |  */ | 
|---|
| 589 | void NetworkStream::handleSoftConnect( int userId) | 
|---|
| 590 | { | 
|---|
| 591 |   // create new handshake and init its variables | 
|---|
| 592 |   this->peers[userId].handshake = NULL; | 
|---|
| 593 |   this->peers[userId].handshake->setUniqueID(userId); | 
|---|
| 594 |  | 
|---|
| 595 |   this->peers[userId].connectionMonitor = new ConnectionMonitor( userId ); | 
|---|
| 596 |   this->peers[userId].userId = userId; | 
|---|
| 597 |   this->peers[userId].bLocal = true; | 
|---|
| 598 |  | 
|---|
| 599 |   PRINTF(0)("num sync: %d\n", synchronizeables.size()); | 
|---|
| 600 |  | 
|---|
| 601 |   // the connecting node of course is a client | 
|---|
| 602 |   this->peers[userId].ip = this->peers[userId].socket->getRemoteAddress(); | 
|---|
| 603 | } | 
|---|
| 604 |  | 
|---|
| 605 |  | 
|---|
| 606 |  | 
|---|
| 607 |  | 
|---|
| 608 | /** | 
|---|
| 609 |  * some debug output | 
|---|
| 610 |  */ | 
|---|
| 611 | void NetworkStream::debug() | 
|---|
| 612 | { | 
|---|
| 613 |   if( SharedNetworkData::getInstance()->isMasterServer()) { | 
|---|
| 614 |     PRINT(0)(" Host ist Master Server with ID: %i\n", this->pInfo->userId); | 
|---|
| 615 |   } | 
|---|
| 616 |   else if( SharedNetworkData::getInstance()->isProxyServerActive()) { | 
|---|
| 617 |     PRINT(0)(" Host ist Proxy Server with ID: %i\n", this->pInfo->userId); | 
|---|
| 618 |   } | 
|---|
| 619 |   else { | 
|---|
| 620 |     PRINT(0)(" Host ist Client with ID: %i\n", this->pInfo->userId); | 
|---|
| 621 |   } | 
|---|
| 622 |  | 
|---|
| 623 |   PRINT(0)(" Current number of connections is: %i\n", this->peers.size()); | 
|---|
| 624 |   for ( PeerList::iterator it = peers.begin(); it != peers.end(); it++ ) | 
|---|
| 625 |   { | 
|---|
| 626 |     PRINT(0)("  peers[%i] with uniqueId %i and address: %s\n", it->first, it->second.userId, it->second.ip.ipString().c_str()); | 
|---|
| 627 |   } | 
|---|
| 628 |   PRINT(0)("\n\n"); | 
|---|
| 629 |  | 
|---|
| 630 |  | 
|---|
| 631 |   PRINT(0)(" Got %i connected Synchronizeables, showing active Syncs:\n", this->synchronizeables.size()); | 
|---|
| 632 |   for (SynchronizeableList::iterator it = synchronizeables.begin(); it!=synchronizeables.end(); it++) | 
|---|
| 633 |   { | 
|---|
| 634 |     if( (*it)->beSynchronized() == true) | 
|---|
| 635 |       PRINT(0)("  Synchronizeable of class: %s::%s, with unique ID: %i, Synchronize: %i\n", (*it)->getClassCName(), (*it)->getCName(), | 
|---|
| 636 |                (*it)->getUniqueID(), (*it)->beSynchronized()); | 
|---|
| 637 |   } | 
|---|
| 638 |   PRINT(0)(" Maximal Connections: %i\n", SharedNetworkData::getInstance()->getMaxPlayer() ); | 
|---|
| 639 |  | 
|---|
| 640 | } | 
|---|
| 641 |  | 
|---|
| 642 |  | 
|---|
| 643 | /** | 
|---|
| 644 |  * @returns the number of synchronizeables registered to this stream | 
|---|
| 645 |  */ | 
|---|
| 646 | int NetworkStream::getSyncCount() | 
|---|
| 647 | { | 
|---|
| 648 |   int n = 0; | 
|---|
| 649 |   for (SynchronizeableList::iterator it = synchronizeables.begin(); it!=synchronizeables.end(); it++) | 
|---|
| 650 |     if( (*it)->beSynchronized() == true) | 
|---|
| 651 |       ++n; | 
|---|
| 652 |  | 
|---|
| 653 |   //return synchronizeables.size(); | 
|---|
| 654 |   return n; | 
|---|
| 655 | } | 
|---|
| 656 |  | 
|---|
| 657 |  | 
|---|
| 658 | /** | 
|---|
| 659 |  * check if handshakes completed. if so create the network game manager else remove it again | 
|---|
| 660 |  */ | 
|---|
| 661 | void NetworkStream::handleHandshakes( ) | 
|---|
| 662 | { | 
|---|
| 663 |   for ( PeerList::iterator it = peers.begin(); it != peers.end(); it++ ) | 
|---|
| 664 |   { | 
|---|
| 665 |     if ( it->second.handshake ) | 
|---|
| 666 |     { | 
|---|
| 667 |       // handshake finished | 
|---|
| 668 |       if ( it->second.handshake->completed() ) | 
|---|
| 669 |       { | 
|---|
| 670 |         //handshake is correct | 
|---|
| 671 |         if ( it->second.handshake->ok() ) | 
|---|
| 672 |         { | 
|---|
| 673 |           // write the first informations into the node so they can be read from there for case differentiation | 
|---|
| 674 |           it->second.nodeType = it->second.handshake->getRemoteNodeType(); | 
|---|
| 675 |  | 
|---|
| 676 |           // the counter part didn't mark it free for deletion yet | 
|---|
| 677 |           if ( !it->second.handshake->allowDel() ) | 
|---|
| 678 |           { | 
|---|
| 679 |             // make sure this is a connection: | 
|---|
| 680 |             // - client       <==> master server | 
|---|
| 681 |             // - proxy server <==> master server | 
|---|
| 682 |             if(  SharedNetworkData::getInstance()->isClient() || | 
|---|
| 683 |                  SharedNetworkData::getInstance()->isProxyServerActive() && | 
|---|
| 684 |                  SharedNetworkData::getInstance()->isUserMasterServer(it->second.userId)) | 
|---|
| 685 |             { | 
|---|
| 686 |               PRINTF(4)("Handshake: i am in client role\n"); | 
|---|
| 687 |  | 
|---|
| 688 |               SharedNetworkData::getInstance()->setHostID( it->second.handshake->getHostId() ); | 
|---|
| 689 |               this->pInfo->userId = SharedNetworkData::getInstance()->getHostID(); | 
|---|
| 690 |  | 
|---|
| 691 | //               it->second.ip = it->second.socket->getRemoteAddress(); | 
|---|
| 692 |  | 
|---|
| 693 |               it->second.nodeType = it->second.handshake->getRemoteNodeType(); | 
|---|
| 694 |               // it->second.ip = it->second.socket->getRemoteAddress(); | 
|---|
| 695 |               // add the new server to the nodes list (it can be a NET_MASTER_SERVER or NET_PROXY_SERVER) | 
|---|
| 696 |               this->networkMonitor->addNode(&it->second); | 
|---|
| 697 |               // get proxy 1 address and add it | 
|---|
| 698 |               this->networkMonitor->addNode(it->second.handshake->getProxy1Address(), NET_PROXY_SERVER_ACTIVE); | 
|---|
| 699 |               // get proxy 2 address and add it | 
|---|
| 700 |               this->networkMonitor->addNode(it->second.handshake->getProxy2Address(), NET_PROXY_SERVER_ACTIVE); | 
|---|
| 701 |  | 
|---|
| 702 |               // now check if the server accepted the connection | 
|---|
| 703 |               if( SharedNetworkData::getInstance()->isClient() && it->second.handshake->redirect() ) | 
|---|
| 704 |               { | 
|---|
| 705 |                 this->bRedirect = true; | 
|---|
| 706 |               } | 
|---|
| 707 |  | 
|---|
| 708 |               // create the new network game manager and init it | 
|---|
| 709 |               this->networkGameManager = NetworkGameManager::getInstance(); | 
|---|
| 710 |               this->networkGameManager->setUniqueID( it->second.handshake->getNetworkGameManagerId() ); | 
|---|
| 711 |               // init the new message manager | 
|---|
| 712 |               MessageManager::getInstance()->setUniqueID( it->second.handshake->getMessageManagerId() ); | 
|---|
| 713 |             } | 
|---|
| 714 |  | 
|---|
| 715 |             PRINT(0)("handshake finished id=%d\n", it->second.handshake->getNetworkGameManagerId()); | 
|---|
| 716 |             it->second.handshake->del(); | 
|---|
| 717 |  | 
|---|
| 718 |           } | 
|---|
| 719 |           else | 
|---|
| 720 |           { | 
|---|
| 721 |             // handshake finished registring new player | 
|---|
| 722 |             if ( it->second.handshake->canDel() ) | 
|---|
| 723 |             { | 
|---|
| 724 |  | 
|---|
| 725 |               if (  SharedNetworkData::getInstance()->isMasterServer() ) | 
|---|
| 726 |               { | 
|---|
| 727 |                 it->second.ip = it->second.socket->getRemoteAddress(); | 
|---|
| 728 |  | 
|---|
| 729 |                 this->networkMonitor->addNode(&it->second); | 
|---|
| 730 |  | 
|---|
| 731 |                 this->handleNewClient( it->second.userId ); | 
|---|
| 732 |  | 
|---|
| 733 |                 if ( PlayerStats::getStats( it->second.userId ) && it->second.handshake->getPreferedNickName() != "" ) | 
|---|
| 734 |                 { | 
|---|
| 735 |                   PlayerStats::getStats( it->second.userId )->setNickName( it->second.handshake->getPreferedNickName() ); | 
|---|
| 736 |                 } | 
|---|
| 737 |               } | 
|---|
| 738 |               else if ( SharedNetworkData::getInstance()->isProxyServerActive() && it->second.isClient() ) | 
|---|
| 739 |               { | 
|---|
| 740 |                 PRINTF(4)("Handshake: Proxy in server role: connecting %i\n", it->second.userId); | 
|---|
| 741 |  | 
|---|
| 742 |                 it->second.ip = it->second.socket->getRemoteAddress(); | 
|---|
| 743 |  | 
|---|
| 744 |                 this->networkMonitor->addNode(&it->second); | 
|---|
| 745 |  | 
|---|
| 746 |                 // work with the ProxyControl to init the new client | 
|---|
| 747 |                 ProxyControl::getInstance()->signalNewClient( it->second.userId); | 
|---|
| 748 |  | 
|---|
| 749 |                 if ( PlayerStats::getStats( it->second.userId ) && it->second.handshake->getPreferedNickName() != "" ) | 
|---|
| 750 |                 { | 
|---|
| 751 |                   PlayerStats::getStats( it->second.userId )->setNickName( it->second.handshake->getPreferedNickName() ); | 
|---|
| 752 |                 } | 
|---|
| 753 |               } | 
|---|
| 754 |  | 
|---|
| 755 |               PRINT(0)("handshake finished delete it\n"); | 
|---|
| 756 |               delete it->second.handshake; | 
|---|
| 757 |               it->second.handshake = NULL; | 
|---|
| 758 |             } | 
|---|
| 759 |           } | 
|---|
| 760 |  | 
|---|
| 761 |         } | 
|---|
| 762 |         else | 
|---|
| 763 |         { | 
|---|
| 764 |           PRINT(1)("handshake failed!\n"); | 
|---|
| 765 |           it->second.socket->disconnectServer(); | 
|---|
| 766 |         } | 
|---|
| 767 |       } | 
|---|
| 768 |     } | 
|---|
| 769 |   } | 
|---|
| 770 | } | 
|---|
| 771 |  | 
|---|
| 772 |  | 
|---|
| 773 | /** | 
|---|
| 774 |  * this functions handles a reconnect event received from the a NET_MASTER_SERVER or NET_PROXY_SERVER | 
|---|
| 775 |  * @param userId | 
|---|
| 776 |  */ | 
|---|
| 777 | void NetworkStream::handleReconnect(int userId) | 
|---|
| 778 | { | 
|---|
| 779 |   this->bRedirect = false; | 
|---|
| 780 | #warning this peer will be created if it does not yet exist: dangerous | 
|---|
| 781 |   PeerInfo* pInfo = &this->peers[userId]; | 
|---|
| 782 |  | 
|---|
| 783 |   IP proxyIP; | 
|---|
| 784 |   if( this->networkMonitor->isForcedReconnection()) | 
|---|
| 785 |     proxyIP = this->networkMonitor->getForcedReconnectionIP(); | 
|---|
| 786 |   else | 
|---|
| 787 |     proxyIP = pInfo->handshake->getProxy1Address(); | 
|---|
| 788 |  | 
|---|
| 789 |   PRINTF(0)("===============================================\n"); | 
|---|
| 790 |   PRINTF(0)("Client is redirected to the other proxy servers\n"); | 
|---|
| 791 |   PRINTF(0)("  user id: %i\n", userId); | 
|---|
| 792 |   PRINTF(0)("  connecting to: %s\n", proxyIP.ipString().c_str()); | 
|---|
| 793 |   PRINTF(0)("===============================================\n"); | 
|---|
| 794 |  | 
|---|
| 795 |   // flush the old synchronization states, since the numbering could be completely different | 
|---|
| 796 |   pInfo->lastAckedState = 0; | 
|---|
| 797 |   pInfo->lastRecvedState = 0; | 
|---|
| 798 |  | 
|---|
| 799 |   // disconnect from the current server and reconnect to proxy server | 
|---|
| 800 |   this->handleDisconnect( userId); | 
|---|
| 801 |   this->connectToProxyServer(NET_ID_PROXY_SERVER_01, proxyIP.ipString(), 9999); | 
|---|
| 802 | //   this->connectToMasterServer(proxyIP.ipString(), 9999); | 
|---|
| 803 |   #warning the ports are not yet integrated correctly in the ip class | 
|---|
| 804 |  | 
|---|
| 805 |   // and restart the handshake | 
|---|
| 806 |   this->startHandshake( userId); | 
|---|
| 807 | } | 
|---|
| 808 |  | 
|---|
| 809 |  | 
|---|
| 810 |  | 
|---|
| 811 | /** | 
|---|
| 812 |  * reconnects to another server, with full handshake | 
|---|
| 813 |  * @param address of the new server | 
|---|
| 814 |  */ | 
|---|
| 815 | void NetworkStream::reconnectToServer(IP address) | 
|---|
| 816 | { | 
|---|
| 817 |   this->networkMonitor->setForcedReconnection(address); | 
|---|
| 818 |   this->handleReconnect( NET_ID_MASTER_SERVER); | 
|---|
| 819 | } | 
|---|
| 820 |  | 
|---|
| 821 |  | 
|---|
| 822 |  | 
|---|
| 823 | /** | 
|---|
| 824 |  * softly reconnecting to another server | 
|---|
| 825 |  * @param address  of the new server | 
|---|
| 826 |  */ | 
|---|
| 827 | void NetworkStream::softReconnectToServer(IP address) | 
|---|
| 828 | { | 
|---|
| 829 |   this->networkMonitor->setForcedReconnection(address); | 
|---|
| 830 |   this->handleReconnect( NET_ID_MASTER_SERVER); | 
|---|
| 831 | } | 
|---|
| 832 |  | 
|---|
| 833 |  | 
|---|
| 834 | /** | 
|---|
| 835 |  * handles the disconnect event | 
|---|
| 836 |  * @param userId id of the user to remove | 
|---|
| 837 |  */ | 
|---|
| 838 | void NetworkStream::handleDisconnect( int userId ) | 
|---|
| 839 | { | 
|---|
| 840 |   this->peers[userId].socket->disconnectServer(); | 
|---|
| 841 |   delete this->peers[userId].socket; | 
|---|
| 842 |   this->peers[userId].socket = NULL; | 
|---|
| 843 |  | 
|---|
| 844 |   if ( this->peers[userId].handshake ) | 
|---|
| 845 |     delete this->peers[userId].handshake; | 
|---|
| 846 |   this->peers[userId].handshake = NULL; | 
|---|
| 847 |  | 
|---|
| 848 |   if ( this->peers[userId].connectionMonitor ) | 
|---|
| 849 |     delete this->peers[userId].connectionMonitor; | 
|---|
| 850 |   this->peers[userId].connectionMonitor = NULL; | 
|---|
| 851 |  | 
|---|
| 852 |  | 
|---|
| 853 |   for ( SynchronizeableList::iterator it2 = synchronizeables.begin(); it2 != synchronizeables.end(); it2++ )  { | 
|---|
| 854 |     (*it2)->cleanUpUser( userId ); | 
|---|
| 855 |   } | 
|---|
| 856 |  | 
|---|
| 857 |   if( SharedNetworkData::getInstance()->isMasterServer()) | 
|---|
| 858 |     NetworkGameManager::getInstance()->signalLeftPlayer(userId); | 
|---|
| 859 |  | 
|---|
| 860 |   this->freeSocketSlots.push_back( userId ); | 
|---|
| 861 |  | 
|---|
| 862 |   this->networkMonitor->removeNode(&this->peers[userId]); | 
|---|
| 863 |   this->peers.erase( userId); | 
|---|
| 864 | } | 
|---|
| 865 |  | 
|---|
| 866 |  | 
|---|
| 867 |  | 
|---|
| 868 | /** | 
|---|
| 869 |  * handle upstream network traffic | 
|---|
| 870 |  * @param tick: seconds elapsed since last update | 
|---|
| 871 |  */ | 
|---|
| 872 | void NetworkStream::handleUpstream( int tick ) | 
|---|
| 873 | { | 
|---|
| 874 |   int offset; | 
|---|
| 875 |   int n; | 
|---|
| 876 |  | 
|---|
| 877 |   for ( PeerList::reverse_iterator peer = peers.rbegin(); peer != peers.rend(); peer++ ) | 
|---|
| 878 |   { | 
|---|
| 879 |     offset = INTSIZE; // reserve enough space for the packet length | 
|---|
| 880 |  | 
|---|
| 881 |     // continue with the next peer if this peer has no socket assigned (therefore no network) | 
|---|
| 882 |     if ( !peer->second.socket ) | 
|---|
| 883 |       continue; | 
|---|
| 884 |  | 
|---|
| 885 |     // header informations: current state | 
|---|
| 886 |     n = Converter::intToByteArray( currentState, buf + offset, UDP_PACKET_SIZE - offset ); | 
|---|
| 887 |     assert( n == INTSIZE ); | 
|---|
| 888 |     offset += n; | 
|---|
| 889 |  | 
|---|
| 890 |     // header informations: last acked state | 
|---|
| 891 |     n = Converter::intToByteArray( peer->second.lastAckedState, buf + offset, UDP_PACKET_SIZE - offset ); | 
|---|
| 892 |     assert( n == INTSIZE ); | 
|---|
| 893 |     offset += n; | 
|---|
| 894 |  | 
|---|
| 895 |     // header informations: last recved state | 
|---|
| 896 |     n = Converter::intToByteArray( peer->second.lastRecvedState, buf + offset, UDP_PACKET_SIZE - offset ); | 
|---|
| 897 |     assert( n == INTSIZE ); | 
|---|
| 898 |     offset += n; | 
|---|
| 899 |  | 
|---|
| 900 |     // now write all synchronizeables in the packet | 
|---|
| 901 |     for ( SynchronizeableList::iterator it = synchronizeables.begin(); it != synchronizeables.end(); it++ ) | 
|---|
| 902 |     { | 
|---|
| 903 |  | 
|---|
| 904 |       int oldOffset = offset; | 
|---|
| 905 |       Synchronizeable & sync = **it; | 
|---|
| 906 |  | 
|---|
| 907 |  | 
|---|
| 908 |       // do not include synchronizeables with uninit id and syncs that don't want to be synchronized | 
|---|
| 909 |       if ( !sync.beSynchronized() || sync.getUniqueID() <= NET_UID_UNASSIGNED ) | 
|---|
| 910 |         continue; | 
|---|
| 911 |  | 
|---|
| 912 |       // if handshake not finished only sync handshake | 
|---|
| 913 |       if ( peer->second.handshake && sync.getLeafClassID() != CL_HANDSHAKE ) | 
|---|
| 914 |         continue; | 
|---|
| 915 |  | 
|---|
| 916 |       // if we are a server (both master and proxy servers) and this is not our handshake | 
|---|
| 917 |       if ( ( SharedNetworkData::getInstance()->isMasterServer() || | 
|---|
| 918 |              SharedNetworkData::getInstance()->isProxyServerActive() &&  peer->second.isClient()) | 
|---|
| 919 |              && sync.getLeafClassID() == CL_HANDSHAKE && sync.getUniqueID() != peer->second.userId ) | 
|---|
| 920 |         continue; | 
|---|
| 921 |  | 
|---|
| 922 |       /* list of synchronizeables that will never be synchronized over the network: */ | 
|---|
| 923 |       // do not sync null parent | 
|---|
| 924 |       if ( sync.getLeafClassID() == CL_NULL_PARENT ) | 
|---|
| 925 |         continue; | 
|---|
| 926 |  | 
|---|
| 927 |  | 
|---|
| 928 |       assert( sync.getLeafClassID() != 0); | 
|---|
| 929 |  | 
|---|
| 930 |       assert( offset + INTSIZE <= UDP_PACKET_SIZE ); | 
|---|
| 931 |  | 
|---|
| 932 |       // server fakes uniqueid == 0 for handshake synchronizeable | 
|---|
| 933 |       if ( ( SharedNetworkData::getInstance()->isMasterServer() || | 
|---|
| 934 |              SharedNetworkData::getInstance()->isProxyServerActive() &&  peer->second.isClient() ) && | 
|---|
| 935 |              ( sync.getUniqueID() >= 1000 || sync.getUniqueID() <= SharedNetworkData::getInstance()->getMaxPlayer() + 1 + NET_ID_PROXY_MAX)) | 
|---|
| 936 |         n = Converter::intToByteArray( 0, buf + offset, UDP_PACKET_SIZE - offset ); | 
|---|
| 937 |       else | 
|---|
| 938 |         n = Converter::intToByteArray( sync.getUniqueID(), buf + offset, UDP_PACKET_SIZE - offset ); | 
|---|
| 939 |  | 
|---|
| 940 |  | 
|---|
| 941 |       assert( n == INTSIZE ); | 
|---|
| 942 |       offset += n; | 
|---|
| 943 |  | 
|---|
| 944 |       // make space for packet size | 
|---|
| 945 |       offset += INTSIZE; | 
|---|
| 946 |  | 
|---|
| 947 |       n = sync.getStateDiff( peer->second.userId, buf + offset, UDP_PACKET_SIZE-offset, currentState, peer->second.lastAckedState, -1000 ); | 
|---|
| 948 |       offset += n; | 
|---|
| 949 |  | 
|---|
| 950 |       assert( Converter::intToByteArray( n, buf + offset - n - INTSIZE, INTSIZE ) == INTSIZE ); | 
|---|
| 951 |  | 
|---|
| 952 |       // check if all data bytes == 0 -> remove data and the synchronizeable from the sync process since there is no update | 
|---|
| 953 |       // TODO not all synchronizeables like this maybe add Synchronizeable::canRemoveZeroDiff() | 
|---|
| 954 |       bool allZero = true; | 
|---|
| 955 |       for ( int i = 0; i < n; i++ ) | 
|---|
| 956 |       { | 
|---|
| 957 |          if ( buf[i+oldOffset+2*INTSIZE] != 0 ) | 
|---|
| 958 |            allZero = false; | 
|---|
| 959 |       } | 
|---|
| 960 |       // if there is no new data in this synchronizeable reset the data offset to the last state -> dont synchronizes | 
|---|
| 961 |       // data that hast not changed | 
|---|
| 962 |       if ( allZero ) | 
|---|
| 963 |       { | 
|---|
| 964 |         offset = oldOffset; | 
|---|
| 965 |       } | 
|---|
| 966 |     } // all synchronizeables written | 
|---|
| 967 |  | 
|---|
| 968 |  | 
|---|
| 969 |  | 
|---|
| 970 |     for ( SynchronizeableList::iterator it = synchronizeables.begin(); it != synchronizeables.end(); it++ ) | 
|---|
| 971 |     { | 
|---|
| 972 |       Synchronizeable & sync = **it; | 
|---|
| 973 |  | 
|---|
| 974 |       // again exclude all unwanted syncs | 
|---|
| 975 |       if ( !sync.beSynchronized() || sync.getUniqueID() <= NET_UID_UNASSIGNED) | 
|---|
| 976 |         continue; | 
|---|
| 977 |  | 
|---|
| 978 |       sync.handleSentState( peer->second.userId, currentState, peer->second.lastAckedState ); | 
|---|
| 979 |     } | 
|---|
| 980 |  | 
|---|
| 981 |  | 
|---|
| 982 |     assert( Converter::intToByteArray( offset, buf, INTSIZE ) == INTSIZE ); | 
|---|
| 983 |  | 
|---|
| 984 |     // now compress the data with the zip library | 
|---|
| 985 |     int compLength = 0; | 
|---|
| 986 |     if ( SharedNetworkData::getInstance()->isMasterServer() || | 
|---|
| 987 |          SharedNetworkData::getInstance()->isProxyServerActive()) | 
|---|
| 988 |       compLength = Zip::getInstance()->zip( buf, offset, compBuf, UDP_PACKET_SIZE, dictServer ); | 
|---|
| 989 |     else | 
|---|
| 990 |       compLength = Zip::getInstance()->zip( buf, offset, compBuf, UDP_PACKET_SIZE, dictClient ); | 
|---|
| 991 |  | 
|---|
| 992 |     if ( compLength <= 0 ) | 
|---|
| 993 |     { | 
|---|
| 994 |       PRINTF(1)("compression failed!\n"); | 
|---|
| 995 |       continue; | 
|---|
| 996 |     } | 
|---|
| 997 |  | 
|---|
| 998 |     assert( peer->second.socket->writePacket( compBuf, compLength ) ); | 
|---|
| 999 |  | 
|---|
| 1000 |     if ( this->remainingBytesToWriteToDict > 0 ) | 
|---|
| 1001 |       writeToNewDict( buf, offset, true ); | 
|---|
| 1002 |  | 
|---|
| 1003 |     peer->second.connectionMonitor->processUnzippedOutgoingPacket( tick, buf, offset, currentState ); | 
|---|
| 1004 |     peer->second.connectionMonitor->processZippedOutgoingPacket( tick, compBuf, compLength, currentState ); | 
|---|
| 1005 |  | 
|---|
| 1006 |   } | 
|---|
| 1007 | } | 
|---|
| 1008 |  | 
|---|
| 1009 | /** | 
|---|
| 1010 |  * handle downstream network traffic | 
|---|
| 1011 |  */ | 
|---|
| 1012 | void NetworkStream::handleDownstream( int tick ) | 
|---|
| 1013 | { | 
|---|
| 1014 |   int offset = 0; | 
|---|
| 1015 |  | 
|---|
| 1016 |   int length = 0; | 
|---|
| 1017 |   int packetLength = 0; | 
|---|
| 1018 |   int compLength = 0; | 
|---|
| 1019 |   int uniqueId = 0; | 
|---|
| 1020 |   int state = 0; | 
|---|
| 1021 |   int ackedState = 0; | 
|---|
| 1022 |   int fromState = 0; | 
|---|
| 1023 |   int syncDataLength = 0; | 
|---|
| 1024 |  | 
|---|
| 1025 |   for ( PeerList::iterator peer = peers.begin(); peer != peers.end(); peer++ ) | 
|---|
| 1026 |   { | 
|---|
| 1027 |  | 
|---|
| 1028 |     if ( !peer->second.socket ) | 
|---|
| 1029 |       continue; | 
|---|
| 1030 |  | 
|---|
| 1031 |     while ( 0 < (compLength = peer->second.socket->readPacket( compBuf, UDP_PACKET_SIZE )) ) | 
|---|
| 1032 |     { | 
|---|
| 1033 |       peer->second.connectionMonitor->processZippedIncomingPacket( tick, compBuf, compLength ); | 
|---|
| 1034 |  | 
|---|
| 1035 |       packetLength = Zip::getInstance()->unZip( compBuf, compLength, buf, UDP_PACKET_SIZE ); | 
|---|
| 1036 |  | 
|---|
| 1037 |       if ( packetLength < 4*INTSIZE ) | 
|---|
| 1038 |       { | 
|---|
| 1039 |         if ( packetLength != 0 ) | 
|---|
| 1040 |           PRINTF(1)("got too small packet: %d\n", packetLength); | 
|---|
| 1041 |         continue; | 
|---|
| 1042 |       } | 
|---|
| 1043 |  | 
|---|
| 1044 |       if ( this->remainingBytesToWriteToDict > 0 ) | 
|---|
| 1045 |         writeToNewDict( buf, packetLength, false ); | 
|---|
| 1046 |  | 
|---|
| 1047 |       assert( Converter::byteArrayToInt( buf, &length ) == INTSIZE ); | 
|---|
| 1048 |       assert( Converter::byteArrayToInt( buf + INTSIZE, &state ) == INTSIZE ); | 
|---|
| 1049 |       assert( Converter::byteArrayToInt( buf + 2*INTSIZE, &fromState ) == INTSIZE ); | 
|---|
| 1050 |       assert( Converter::byteArrayToInt( buf + 3*INTSIZE, &ackedState ) == INTSIZE ); | 
|---|
| 1051 |       offset = 4*INTSIZE; | 
|---|
| 1052 |  | 
|---|
| 1053 |       peer->second.connectionMonitor->processUnzippedIncomingPacket( tick, buf, packetLength, state, ackedState ); | 
|---|
| 1054 |  | 
|---|
| 1055 |  | 
|---|
| 1056 |       //if this is an old state drop it | 
|---|
| 1057 |       if ( state <= peer->second.lastRecvedState ) | 
|---|
| 1058 |         continue; | 
|---|
| 1059 |  | 
|---|
| 1060 |       if ( packetLength != length ) | 
|---|
| 1061 |       { | 
|---|
| 1062 |         PRINTF(1)("real packet length (%d) and transmitted packet length (%d) do not match!\n", packetLength, length); | 
|---|
| 1063 |         peer->second.socket->disconnectServer(); | 
|---|
| 1064 |         continue; | 
|---|
| 1065 |       } | 
|---|
| 1066 |  | 
|---|
| 1067 |       while ( offset + 2 * INTSIZE < length ) | 
|---|
| 1068 |       { | 
|---|
| 1069 |         // read the unique id of the sync | 
|---|
| 1070 |         assert( offset > 0 ); | 
|---|
| 1071 |         assert( Converter::byteArrayToInt( buf + offset, &uniqueId ) == INTSIZE ); | 
|---|
| 1072 |         offset += INTSIZE; | 
|---|
| 1073 |  | 
|---|
| 1074 |         // read the data length | 
|---|
| 1075 |         assert( Converter::byteArrayToInt( buf + offset, &syncDataLength ) == INTSIZE ); | 
|---|
| 1076 |         offset += INTSIZE; | 
|---|
| 1077 |  | 
|---|
| 1078 |         assert( syncDataLength > 0 ); | 
|---|
| 1079 |         assert( syncDataLength < 10000 ); | 
|---|
| 1080 |  | 
|---|
| 1081 |         Synchronizeable * sync = NULL; | 
|---|
| 1082 |  | 
|---|
| 1083 |         // look for the synchronizeable in question | 
|---|
| 1084 |         for ( SynchronizeableList::iterator it = synchronizeables.begin(); it != synchronizeables.end(); it++ ) | 
|---|
| 1085 |         { | 
|---|
| 1086 |           // client thinks his handshake has a special id: hostId * 1000 (host id of this server) | 
|---|
| 1087 |           if ( (*it)->getUniqueID() == uniqueId ||                                          // so this client exists already go to sync work | 
|---|
| 1088 |                  ( uniqueId == 0  && (*it)->getUniqueID() == peer->second.userId ) )        // so this is a Handshake! | 
|---|
| 1089 |           { | 
|---|
| 1090 |             sync = *it; | 
|---|
| 1091 |             break; | 
|---|
| 1092 |           } | 
|---|
| 1093 |         } | 
|---|
| 1094 |  | 
|---|
| 1095 |         // this synchronizeable does not yet exist! create it | 
|---|
| 1096 |         if ( sync == NULL ) | 
|---|
| 1097 |         { | 
|---|
| 1098 |           PRINTF(0)("could not find sync with id %d. try to create it\n", uniqueId); | 
|---|
| 1099 |  | 
|---|
| 1100 |           // if it is an old synchronizeable already removed, ignore it | 
|---|
| 1101 |           if ( oldSynchronizeables.find( uniqueId ) != oldSynchronizeables.end() ) | 
|---|
| 1102 |           { | 
|---|
| 1103 |             offset += syncDataLength; | 
|---|
| 1104 |             continue; | 
|---|
| 1105 |           } | 
|---|
| 1106 |  | 
|---|
| 1107 |           // if the node we got this unknown sync we ignore it if: | 
|---|
| 1108 |           //  - the remote host is a client | 
|---|
| 1109 |           //  - the remote host is a proxy server and we are master server | 
|---|
| 1110 |           // (since it has no rights to create a new sync) | 
|---|
| 1111 |           if ( peers[peer->second.userId].isClient() || | 
|---|
| 1112 |                (peers[peer->second.userId].isProxyServerActive() && SharedNetworkData::getInstance()->isMasterServer())) | 
|---|
| 1113 |           { | 
|---|
| 1114 |             offset += syncDataLength; | 
|---|
| 1115 |             continue; | 
|---|
| 1116 |           } | 
|---|
| 1117 |  | 
|---|
| 1118 |           int leafClassId; | 
|---|
| 1119 |           if ( INTSIZE > length - offset ) | 
|---|
| 1120 |           { | 
|---|
| 1121 |             offset += syncDataLength; | 
|---|
| 1122 |             continue; | 
|---|
| 1123 |           } | 
|---|
| 1124 |  | 
|---|
| 1125 |           Converter::byteArrayToInt( buf + offset, &leafClassId ); | 
|---|
| 1126 |  | 
|---|
| 1127 |           assert( leafClassId != 0 ); | 
|---|
| 1128 |  | 
|---|
| 1129 |  | 
|---|
| 1130 |           BaseObject * b = NULL; | 
|---|
| 1131 |           /* These are some small exeptions in creation: Not all objects can/should be created via Factory */ | 
|---|
| 1132 |           /* Exception 1: NullParent */ | 
|---|
| 1133 |           if( leafClassId == CL_NULL_PARENT || leafClassId == CL_SYNCHRONIZEABLE || leafClassId == CL_NETWORK_GAME_MANAGER ) | 
|---|
| 1134 |           { | 
|---|
| 1135 |             PRINTF(1)("Don't create Object with ID %x, ignored!\n", (int)leafClassId); | 
|---|
| 1136 |             offset += syncDataLength; | 
|---|
| 1137 |             continue; | 
|---|
| 1138 |           } | 
|---|
| 1139 |           else | 
|---|
| 1140 |             b = Factory::fabricate( (ClassID)leafClassId ); | 
|---|
| 1141 |  | 
|---|
| 1142 |           if ( !b ) | 
|---|
| 1143 |           { | 
|---|
| 1144 |             PRINTF(1)("Could not fabricate Object with classID %x\n", leafClassId); | 
|---|
| 1145 |             offset += syncDataLength; | 
|---|
| 1146 |             continue; | 
|---|
| 1147 |           } | 
|---|
| 1148 |  | 
|---|
| 1149 |           if ( b->isA(CL_SYNCHRONIZEABLE) ) | 
|---|
| 1150 |           { | 
|---|
| 1151 |             sync = dynamic_cast<Synchronizeable*>(b); | 
|---|
| 1152 |             sync->setUniqueID( uniqueId ); | 
|---|
| 1153 |             sync->setSynchronized(true); | 
|---|
| 1154 |  | 
|---|
| 1155 |             PRINTF(0)("Fabricated %s with id %d\n", sync->getClassCName(), sync->getUniqueID()); | 
|---|
| 1156 |           } | 
|---|
| 1157 |           else | 
|---|
| 1158 |           { | 
|---|
| 1159 |             PRINTF(1)("Class with ID %x is not a synchronizeable!\n", (int)leafClassId); | 
|---|
| 1160 |             delete b; | 
|---|
| 1161 |             offset += syncDataLength; | 
|---|
| 1162 |             continue; | 
|---|
| 1163 |           } | 
|---|
| 1164 |         } | 
|---|
| 1165 |  | 
|---|
| 1166 |  | 
|---|
| 1167 |         int n = sync->setStateDiff( peer->second.userId, buf+offset, syncDataLength, state, fromState ); | 
|---|
| 1168 |         offset += n; | 
|---|
| 1169 |  | 
|---|
| 1170 |       } | 
|---|
| 1171 |  | 
|---|
| 1172 |       if ( offset != length ) | 
|---|
| 1173 |       { | 
|---|
| 1174 |         PRINTF(0)("offset (%d) != length (%d)\n", offset, length); | 
|---|
| 1175 |         peer->second.socket->disconnectServer(); | 
|---|
| 1176 |       } | 
|---|
| 1177 |  | 
|---|
| 1178 |  | 
|---|
| 1179 |       for ( SynchronizeableList::iterator it = synchronizeables.begin(); it != synchronizeables.end(); it++ ) | 
|---|
| 1180 |       { | 
|---|
| 1181 |         Synchronizeable & sync = **it; | 
|---|
| 1182 |  | 
|---|
| 1183 |         if ( !sync.beSynchronized() || sync.getUniqueID() <= NET_UID_UNASSIGNED ) | 
|---|
| 1184 |           continue; | 
|---|
| 1185 |  | 
|---|
| 1186 |         sync.handleRecvState( peer->second.userId, state, fromState ); | 
|---|
| 1187 |       } | 
|---|
| 1188 |  | 
|---|
| 1189 |       assert( peer->second.lastAckedState <= ackedState ); | 
|---|
| 1190 |       peer->second.lastAckedState = ackedState; | 
|---|
| 1191 |  | 
|---|
| 1192 |       assert( peer->second.lastRecvedState < state ); | 
|---|
| 1193 |       peer->second.lastRecvedState = state; | 
|---|
| 1194 |  | 
|---|
| 1195 |     } | 
|---|
| 1196 |  | 
|---|
| 1197 |   } | 
|---|
| 1198 |  | 
|---|
| 1199 | } | 
|---|
| 1200 |  | 
|---|
| 1201 | /** | 
|---|
| 1202 |  * is executed when a handshake has finished | 
|---|
| 1203 |  */ | 
|---|
| 1204 | void NetworkStream::handleNewClient( int userId ) | 
|---|
| 1205 | { | 
|---|
| 1206 |   // init and assign the message manager | 
|---|
| 1207 |   MessageManager::getInstance()->initUser( userId ); | 
|---|
| 1208 |   // do all game relevant stuff here | 
|---|
| 1209 |   networkGameManager->signalNewPlayer( userId ); | 
|---|
| 1210 | } | 
|---|
| 1211 |  | 
|---|
| 1212 |  | 
|---|
| 1213 | /** | 
|---|
| 1214 |  * removes old items from oldSynchronizeables | 
|---|
| 1215 |  */ | 
|---|
| 1216 | void NetworkStream::cleanUpOldSyncList( ) | 
|---|
| 1217 | { | 
|---|
| 1218 |   int now = SDL_GetTicks(); | 
|---|
| 1219 |  | 
|---|
| 1220 |   for ( std::map<int,int>::iterator it = oldSynchronizeables.begin(); it != oldSynchronizeables.end();  ) | 
|---|
| 1221 |   { | 
|---|
| 1222 |     if ( it->second < now - 10*1000 ) | 
|---|
| 1223 |     { | 
|---|
| 1224 |       std::map<int,int>::iterator delIt = it; | 
|---|
| 1225 |       it++; | 
|---|
| 1226 |       oldSynchronizeables.erase( delIt ); | 
|---|
| 1227 |       continue; | 
|---|
| 1228 |     } | 
|---|
| 1229 |     it++; | 
|---|
| 1230 |   } | 
|---|
| 1231 | } | 
|---|
| 1232 |  | 
|---|
| 1233 | /** | 
|---|
| 1234 |  * writes data to DATA/dicts/newdict | 
|---|
| 1235 |  * @param data pointer to data | 
|---|
| 1236 |  * @param length length | 
|---|
| 1237 |  */ | 
|---|
| 1238 | void NetworkStream::writeToNewDict( byte * data, int length, bool upstream ) | 
|---|
| 1239 | { | 
|---|
| 1240 |   if ( remainingBytesToWriteToDict <= 0 ) | 
|---|
| 1241 |     return; | 
|---|
| 1242 |  | 
|---|
| 1243 |   if ( length > remainingBytesToWriteToDict ) | 
|---|
| 1244 |     length = remainingBytesToWriteToDict; | 
|---|
| 1245 |  | 
|---|
| 1246 |   std::string fileName = ResourceManager::getInstance()->getDataDir(); | 
|---|
| 1247 |   fileName += "/dicts/newdict"; | 
|---|
| 1248 |  | 
|---|
| 1249 |   if ( upstream ) | 
|---|
| 1250 |     fileName += "_upstream"; | 
|---|
| 1251 |   else | 
|---|
| 1252 |     fileName += "_downstream"; | 
|---|
| 1253 |  | 
|---|
| 1254 |   FILE * f = fopen( fileName.c_str(), "a" ); | 
|---|
| 1255 |  | 
|---|
| 1256 |   if ( !f ) | 
|---|
| 1257 |   { | 
|---|
| 1258 |     PRINTF(2)("could not open %s\n", fileName.c_str()); | 
|---|
| 1259 |     remainingBytesToWriteToDict = 0; | 
|---|
| 1260 |     return; | 
|---|
| 1261 |   } | 
|---|
| 1262 |  | 
|---|
| 1263 |   if ( fwrite( data, 1, length, f ) != length ) | 
|---|
| 1264 |   { | 
|---|
| 1265 |     PRINTF(2)("could not write to file\n"); | 
|---|
| 1266 |     fclose( f ); | 
|---|
| 1267 |     return; | 
|---|
| 1268 |   } | 
|---|
| 1269 |  | 
|---|
| 1270 |   fclose( f ); | 
|---|
| 1271 |  | 
|---|
| 1272 |   remainingBytesToWriteToDict -= length; | 
|---|
| 1273 | } | 
|---|
| 1274 |  | 
|---|
| 1275 |  | 
|---|
| 1276 |  | 
|---|
| 1277 |  | 
|---|
| 1278 |  | 
|---|
| 1279 |  | 
|---|