[5566] | 1 | /*! |
---|
| 2 | * @file network_stream.h |
---|
| 3 | * implementation of a network pipe |
---|
| 4 | */ |
---|
[5587] | 5 | |
---|
[5566] | 6 | #ifndef _NETWORK_STREAM |
---|
| 7 | #define _NETWORK_STREAM |
---|
| 8 | |
---|
[6139] | 9 | #include <vector> |
---|
| 10 | #include <list> |
---|
[7954] | 11 | #include <map> |
---|
[6139] | 12 | |
---|
[5582] | 13 | #include "data_stream.h" |
---|
[6139] | 14 | #include "server_socket.h" |
---|
| 15 | #include "handshake.h" |
---|
[9406] | 16 | #include "monitor/connection_monitor.h" |
---|
[7954] | 17 | #include "udp_server_socket.h" |
---|
[9406] | 18 | #include "peer_info.h" |
---|
[5566] | 19 | |
---|
[9406] | 20 | #include "shared_network_data.h" |
---|
| 21 | |
---|
[5647] | 22 | class Synchronizeable; |
---|
| 23 | class NetworkSocket; |
---|
[6139] | 24 | class ServerSocket; |
---|
[6341] | 25 | class NetworkGameManager; |
---|
[9406] | 26 | class NetworkMonitor; |
---|
[5589] | 27 | |
---|
[9246] | 28 | |
---|
[6139] | 29 | typedef std::list<Synchronizeable*> SynchronizeableList; |
---|
[7954] | 30 | typedef std::map<int,PeerInfo> PeerList; |
---|
[5809] | 31 | |
---|
| 32 | |
---|
[5649] | 33 | class NetworkStream : public DataStream |
---|
[5566] | 34 | { |
---|
| 35 | |
---|
[5996] | 36 | public: |
---|
| 37 | NetworkStream(); |
---|
[9406] | 38 | NetworkStream(int nodeType); |
---|
| 39 | virtual ~NetworkStream(); |
---|
[5804] | 40 | |
---|
[5996] | 41 | void init(); |
---|
[5804] | 42 | |
---|
[9450] | 43 | void connectToMasterServer(std::string host, int port); |
---|
| 44 | void connectToProxyServer(int proxyId, std::string host, int port); |
---|
[9406] | 45 | void createServer(int port); |
---|
| 46 | |
---|
[6695] | 47 | void createNetworkGameManager(); |
---|
[9450] | 48 | void startHandshake(int userId = NET_ID_MASTER_SERVER); |
---|
[6695] | 49 | |
---|
[9406] | 50 | /* synchronizeable interface */ |
---|
[5996] | 51 | void connectSynchronizeable(Synchronizeable& sync); |
---|
[6139] | 52 | void disconnectSynchronizeable(Synchronizeable& sync); |
---|
[5566] | 53 | |
---|
[9406] | 54 | inline int getMaxConnections(){ return SharedNetworkData::getInstance()->getMaxPlayer(); } |
---|
[5996] | 55 | |
---|
[9406] | 56 | /* functions for the peerInfo information retreival */ |
---|
| 57 | inline bool isUserIdActive( int userID ) { return (peers.find(userID) != peers.end()); } |
---|
| 58 | inline bool isUserMasterServer( int userID ){ if ( !isUserIdActive(userID) ) return false; return peers[userID].isMasterServer(); } |
---|
| 59 | inline bool isUserProxyServer( int userID ){ if ( !isUserIdActive(userID) ) return false; return peers[userID].isProxyServer(); } |
---|
| 60 | inline bool isUserClient( int userID ){ if ( !isUserIdActive(userID) ) return false; return peers[userID].isClient(); } |
---|
[6139] | 61 | |
---|
[9406] | 62 | /* peering interface */ |
---|
| 63 | inline PeerInfo & getPeerInfo( int userId ) { return peers[userId]; } |
---|
| 64 | inline PeerInfo* getPeerInfo() { return this->pInfo; } |
---|
| 65 | inline PeerList getPeerInfoList() { return this->peers; } |
---|
| 66 | |
---|
| 67 | /* data processing*/ |
---|
[5996] | 68 | virtual void processData(); |
---|
| 69 | |
---|
[9406] | 70 | /* debugging */ |
---|
| 71 | void debug(); |
---|
| 72 | |
---|
| 73 | |
---|
| 74 | private: |
---|
| 75 | |
---|
[6341] | 76 | inline SynchronizeableList::const_iterator getSyncBegin(){ return synchronizeables.begin(); } |
---|
| 77 | inline SynchronizeableList::const_iterator getSyncEnd(){ return synchronizeables.end(); } |
---|
[9406] | 78 | void cleanUpOldSyncList(); |
---|
[6695] | 79 | int getSyncCount(); |
---|
[6341] | 80 | |
---|
[6695] | 81 | void updateConnectionList(); |
---|
[9419] | 82 | |
---|
[9406] | 83 | /* handle processes */ |
---|
[7954] | 84 | void handleHandshakes(); |
---|
[8068] | 85 | void handleUpstream( int tick ); |
---|
[9246] | 86 | void handleDownstream(int tick ); |
---|
[9406] | 87 | |
---|
| 88 | /* handle events*/ |
---|
[7954] | 89 | void handleNewClient( int userId ); |
---|
[9406] | 90 | void handleReconnect( int userId ); |
---|
[9419] | 91 | void handleDisconnect( int userId ); |
---|
[9246] | 92 | |
---|
[8623] | 93 | void writeToNewDict( byte * data, int length, bool upstream ); |
---|
[6139] | 94 | |
---|
| 95 | |
---|
[6695] | 96 | private: |
---|
[9406] | 97 | PeerList peers; //!< list of the network node informations |
---|
[6341] | 98 | |
---|
[9406] | 99 | PeerInfo* pInfo; //!< the info about the local peer node (not in the PeerList) |
---|
[9246] | 100 | |
---|
[9406] | 101 | std::list<int> freeSocketSlots; //!< list of free sockets (to ensure not to resycle sockets) |
---|
[7954] | 102 | int currentState; //!< current state id |
---|
[6695] | 103 | |
---|
[9406] | 104 | NetworkMonitor* networkMonitor; //!< the network monitor |
---|
| 105 | NetworkGameManager* networkGameManager; //!< reference to the network game manager |
---|
| 106 | ServerSocket* serverSocket; //!< the listening socket of the server |
---|
[7954] | 107 | |
---|
| 108 | std::map<int,int> oldSynchronizeables; //!< used to save recently deleted sync ids to not recreate them |
---|
[9406] | 109 | SynchronizeableList synchronizeables; //!< list of the synchronizeables |
---|
[9246] | 110 | |
---|
[7954] | 111 | byte buf[UDP_PACKET_SIZE]; //!< used by handleUp/Downstream |
---|
| 112 | byte compBuf[UDP_PACKET_SIZE]; //!< used by handleUp/Downstream |
---|
| 113 | int remainingBytesToWriteToDict; //!< if > 0 NetworkStream will write packets to DATA/dicts/newdict |
---|
[8623] | 114 | |
---|
[9406] | 115 | int dictServer; //!< the zip dict for the server |
---|
| 116 | int dictClient; //!< the zip dict for the client |
---|
[9433] | 117 | |
---|
| 118 | bool bRedirect; //!< true if the master server sent a redirect command |
---|
[5587] | 119 | }; |
---|
[5566] | 120 | #endif /* _NETWORK_STREAM */ |
---|