Changeset 380 for code/branches/FICN/src/network/ConnectionManager.cc

Timestamp:

Dec 4, 2007, 8:22:24 PM (16 years ago)

Author:

scheusso

Message:

problem in connectionmanager

File:

• code/branches/FICN/src/network/ConnectionManager.cc (modified) (8 diffs)

code/branches/FICN/src/network/ConnectionManager.cc

@@ -16 +16 @@
   boost::thread_group network_threads;
 
-  void test(){
-    return;
-  }
-
   ConnectionManager::ConnectionManager(){
     quit=false;
-    client=NULL;
     bindAddress.host = ENET_HOST_ANY;
     bindAddress.port = NETWORK_PORT;
@@ -29 +24 @@
   ConnectionManager::ConnectionManager(int port, std::string address){
     quit=false;
-    client=NULL;
     enet_address_set_host (& bindAddress, address.c_str());
     bindAddress.port = NETWORK_PORT;
@@ -36 +30 @@
   ConnectionManager::ConnectionManager(int port, const char *address){
     quit=false;
-    client=NULL;
     enet_address_set_host (& bindAddress, address);
     bindAddress.port = NETWORK_PORT;
@@ -47 +40 @@
         return NULL;
   }
+  
+  ENetPacket *ConnectionManager::getPacket(int &clientID){
+    ENetAddress address;
+    ENetPacket *packet=getPacket(address);
+    clientID=clientMap.find(address)->second;
+    return packet;
+  }
 
   bool ConnectionManager::queueEmpty(){
@@ -65 +65 @@
 
   bool ConnectionManager::addPacket(ENetPacket *packet, ENetPeer *peer){
-    if(client==NULL)
-      return false;
-    ClientList *temp=client;
-    while(peer->host != temp->event->peer->host){
-      temp=temp->next;
-      if(temp==NULL)
-        return false;
-    }
-    if(enet_peer_send(temp->event->peer, temp->ID, packet)!=0)
-      return false;
-    return true;
-  }
-
-  bool ConnectionManager::addPacket(ENetPacket *packet, int ID){
-    if(client==NULL)
-      return false;
-    ClientList *temp=client;
-    while(ID != temp->ID){
-      temp=temp->next;
-      if(temp==NULL)
-        return false;
-    }
-    if(enet_peer_send(temp->event->peer, temp->ID, packet)!=0)
-      return false;
-    else
-      return true;
-  }
-
+    if(clientVector.size()==0)
+      return false;
+    if(enet_peer_send(peer, clientMap.find(peer->address)->second , packet)!=0)
+      return false;
+    return true;
+  }
+  
+  bool ConnectionManager::addPacket(ENetPacket *packet, int clientID){
+    if(clientVector.size()==0)
+      return false;
+    if(enet_peer_send(&(peerMap.find(clientVector[clientID])->second), clientID, packet)!=0)
+      return false;
+    return true;
+  }
+  
   bool ConnectionManager::addPacketAll(ENetPacket *packet){
-    ClientList *temp=client;
-    while(temp!=NULL){
-      if(enet_peer_send(temp->event->peer, temp->ID, packet)!=0)
+    for(int i=0; i<clientVector.size(); i++){
+      if(enet_peer_send(&(peerMap.find(clientVector[i])->second), i, packet)!=0)
          return false;
     }
@@ -163 +149 @@
     bool disconnected=false;
     ENetEvent event;
-    ClientList *temp=client;
-    while(temp!=NULL){
-      enet_peer_disconnect(temp->event->peer, 0);
+    for(int i=0; i<clientVector.size(); i++){
+      enet_peer_disconnect(&(peerMap.find(clientVector[i])->second), 0);
       while( !disconnected && enet_host_service(server, &event, NETWORK_WAIT_TIMEOUT) > 0){
         switch (event.type)
@@ -179 +164 @@
         }
       }
-      temp = temp->next;
       disconnected=false;
     }
@@ -192 +176 @@
 
   bool ConnectionManager::clientDisconnect(ENetPeer *peer){
-    ClientList *temp=client;
-    // do we have to remove the first item ?
-    if(temp->event->peer->host==peer->host){
-      if(temp->next==NULL){
-        client=NULL;
-      } else{
-        client=temp->next;
-      }
-      delete temp;
-      return true;
-    } else {
-      while(temp->next!=NULL){
-        if(temp->next->event->peer->host==peer->host){
-          // do a correct relink and delete the disconnected client
-          ClientList *temp2=temp->next;
-          temp->next=temp2->next;
-          delete temp2;
-          return true;
-        } else
-          //otherwise keep on searching ;)
-          temp=temp->next;
-      }
-    }
-    return false;
+    return clientDisconnect(*peer);
+  }
+  
+  bool ConnectionManager::clientDisconnect(ENetPeer peer){
+    std::map<ENetAddress, int>::iterator it;
+    it=clientMap.find(peer.address);
+    clientVector.erase(clientVector.begin()+it->second);
+    clientMap.erase(it);
+    peerMap.erase(peerMap.find(peer.address));
+    return true;
   }
 
   bool ConnectionManager::addClient(ENetEvent *event){
-
-    // first client?
-    if(client==NULL){
-      client =new ClientList;
-      client->ID=1;
-      client->next=NULL;
-      client->event = event;
-    } else {
-      // otherwise add a new element to clientlist
-      ClientList *temp = client;
-      int i=1;
-      while(temp->next != NULL){
-        temp=temp->next;
-        i++;
-      }
-      temp->next=new ClientList;
-      temp=temp->next;
-      temp->ID=i+1;
-      temp->event=event;
-      temp->next=NULL;
-    }
-    return true;
-  }
-
-
+    clientVector.push_back((event->peer->address));
+    clientMap[event->peer->address]=clientVector.size()-1;
+    peerMap[event->peer->address]=*(event->peer);
+    return true;
+  }
+  
+  int ConnectionManager::getClientID(ENetPeer peer){
+    return clientMap.find(peer.address)->second;
+  }
+  
+  int ConnectionManager::getClientID(ENetAddress address){
+    return clientMap.find(address)->second;
+  }
+  
+  ENetPeer ConnectionManager::getClientPeer(int clientID){
+    return peerMap.find(clientVector[clientID])->second;
+  }
+  
 }