engine/src/network/NetworkHost.cpp

   1 /*
   2 Copyright (C) 2008  Mathias Gottschlag
   3
   4 Permission is hereby granted, free of charge, to any person obtaining a copy of
   5 this software and associated documentation files (the "Software"), to deal in the
   6 Software without restriction, including without limitation the rights to use,
   7 copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the
   8 Software, and to permit persons to whom the Software is furnished to do so,
   9 subject to the following conditions:
  10
  11 The above copyright notice and this permission notice shall be included in all
  12 copies or substantial portions of the Software.
  13
  14 THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED,
  15 INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A
  16 PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
  17 HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
  18 OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
  19 SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  20 */
  21
  22 #include "network/NetworkHost.h"
  23 #include "network/Connection.h"
  24 #include "core/Logger.h"
  25
  26 #include <stdio.h>
  27
  28 namespace peak
  29 {
  30         NetworkHost::NetworkHost()
  31         {
  32                 host = 0;
  33         }
  34         NetworkHost::~NetworkHost()
  35         {
  36         }
  37
  38         bool NetworkHost::init(int port)
  39         {
  40                 // Create network host
  41                 ENetAddress address;
  42                 address.host = ENET_HOST_ANY;
  43                 address.port = port;
  44                 host = enet_host_create(&address, 32, 200000, 20000);
  45                 if (!host)
  46                 {
  47                         LERROR("Could not create network host.\n");
  48                         return false;
  49                 }
  50                 LINFO("Created network host.\n");
  51                 return true;
  52         }
  53         bool NetworkHost::shutdown(void)
  54         {
  55                 for (unsigned int i = 0; i < connections.size(); i++)
  56                 {
  57                         Connection *conn = connections[i];
  58                         conn->destroy();
  59                         delete connections[i];
  60                 }
  61                 connections.clear();
  62                 if (host)
  63                         enet_host_destroy(host);
  64                 host = 0;
  65                 return true;
  66         }
  67
  68         Connection *NetworkHost::getNewConnection(void)
  69         {
  70                 if (newconnections.empty())
  71                 {
  72                         return 0;
  73                 }
  74
  75                 Connection *conn = newconnections.front();
  76                 newconnections.pop();
  77                 return conn;
  78         }
  79         void NetworkHost::closeConnection(Connection *conn)
  80         {
  81                 for (unsigned int i = 0; i < connections.size(); i++)
  82                 {
  83                         if (connections[i] == conn)
  84                         {
  85                                 connections[i]->destroy();
  86                                 delete connections[i];
  87                                 connections.erase(connections.begin() + i);
  88                                 return;
  89                         }
  90                 }
  91         }
  92
  93         bool NetworkHost::doWork(void)
  94         {
  95                 if (!host)
  96                         return false;
  97
  98
  99                 // Receive data
 100                 ENetEvent event;
 101                 while (enet_host_service(host, &event, 0) > 0)
 102                 {
 103                         switch (event.type)
 104                         {
 105                         case ENET_EVENT_TYPE_CONNECT:
 106                         {
 107                                 LINFO("Host: Someone connected.\n");
 108                                 Connection *newconn = new Connection(event.peer);
 109                                 connections.push_back(newconn);
 110                                 newconnections.push(newconn);
 111                                 break;
 112                         }
 113                         case ENET_EVENT_TYPE_RECEIVE:
 114                                 //printf("Host: Received data.\n");
 115                                 // Push received data into connection
 116                                 for (unsigned int i = 0; i < connections.size(); i++)
 117                                 {
 118                                         if (event.peer == connections[i]->getPeer())
 119                                         {
 120                                                 Buffer *buffer = new Buffer;
 121                                                 buffer->writeData(event.packet->data,
 122                                                                 event.packet->dataLength, true);
 123                                                 connections[i]->injectData(buffer);
 124                                         }
 125                                 }
 126                                 break;
 127                         case ENET_EVENT_TYPE_DISCONNECT:
 128                                 LINFO("Host: Someone disconnected.\n");
 129                                 for (unsigned int i = 0; i < connections.size(); i++)
 130                                 {
 131                                         if (event.peer == connections[i]->getPeer())
 132                                         {
 133                                                 connections[i]->destroy();
 134                                         }
 135                                 }
 136                                 break;
 137                         default:
 138                                 break;
 139                         }
 140                 }
 141                 // Send data
 142
 143                 return true;
 144         }
 145 }
 146