libt2n: (gerd) add ip communication

[libt2n] / src / server.cpp

diff --git a/src/server.cpp b/src/server.cpp
index c8255ee..20cce1e 100644 (file)
--- a/src/server.cpp
+++ b/src/server.cpp
@@ -17,84 +17,76 @@
  *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
  ***************************************************************************/
 
+#include <sstream>
+
 #include "server.hxx"
 
 namespace libt2n
 {
 
-void connection::check_timeout()
+void server_connection::check_timeout()
 {
     if (timeout != -1 && last_action_time+timeout >= time(NULL))
         this->close();
 }
 
-void connection::reset_timeout()
+void server_connection::reset_timeout()
 {
     last_action_time=time(NULL);
 }
 
-bool connection::get_packet(std::string& data, unsigned int& conn_id)
-{
-    // max packet size is unsigned int
-
-    // no size information -> no packet
-    if (buffer.size() < sizeof(unsigned int))
-        return false;
-
-    unsigned int psize=*((unsigned int*)(buffer.data()));
-
-    // enough data for one packet in buffer?
-    if (buffer.size() < sizeof(unsigned int)+psize)
-        return false;
-
-    data.assign(buffer,sizeof(unsigned int),psize);
-    buffer.erase(0,sizeof(unsigned int)+psize);
-
-    return true;
-}
-
 server::~server()
 {
-    std::map<unsigned int, connection*>::iterator ie=connections.end();
-    for(std::map<unsigned int, connection*>::iterator i=connections.begin(); i != ie; i++)
+    std::map<unsigned int, server_connection*>::iterator ie=connections.end();
+    for(std::map<unsigned int, server_connection*>::iterator i=connections.begin(); i != ie; i++)
         delete i->second;
 }
 
-int server::add_connection(connection* newconn)
+int server::add_connection(server_connection* newconn)
 {
-    connections[next_id]=newconn;
-    return next_id++;
+    unsigned int cid=next_id++;
+    newconn->set_id(cid);
+    newconn->set_server(this);
+    connections[cid]=newconn;
+    return cid;
 }
 
 /**
-    Gets a connection by id
+    @brief Gets a connection by id
     
-    \param conn_id Connection ID
+    @param conn_id Connection ID
     
-    \retval Pointer to connection object
+    @retval Pointer to connection object
 */
-connection* server::get_connection(unsigned int conn_id)
+server_connection* server::get_connection(unsigned int conn_id)
 {
-    std::map<unsigned int, connection*>::iterator p=connections.find(conn_id);
+    std::map<unsigned int, server_connection*>::iterator p=connections.find(conn_id);
     if (p==connections.end())
         return NULL;
     else
         return p->second;
 }
 
-void server::check_timeout()
+/// check for timeouts, remove closed connections. don't forget to call this from time to time.
+void server::cleanup()
 {
-    std::map<unsigned int, connection*>::iterator ie=connections.end();
-    for(std::map<unsigned int, connection*>::iterator i=connections.begin(); i != ie; i++)
+    std::map<unsigned int, server_connection*>::iterator ie=connections.end();
+    for(std::map<unsigned int, server_connection*>::iterator i=connections.begin(); i != ie; i++)
         i->second->check_timeout();
-}
 
-void server::fill_connection_buffers(void)
-{
-    std::map<unsigned int, connection*>::iterator ie=connections.end();
-    for(std::map<unsigned int, connection*>::iterator i=connections.begin(); i != ie; i++)
-        if (!i->second->is_closed())
-            i->second->fill_buffer();
+    for(std::map<unsigned int, server_connection*>::iterator i=connections.begin(); i != ie;)
+    {
+        if (i->second->is_closed() && !i->second->packet_available())
+        {
+            // closed and no usable data in buffer -> remove
+            delete i->second;
+            connections.erase(i);
+            i=connections.begin();
+            ie=connections.end();
+        }
+        else
+            i++;
+    }
 }
 
 bool server::get_packet(std::string& data, unsigned int& conn_id)
@@ -102,10 +94,15 @@ bool server::get_packet(std::string& data, unsigned int& conn_id)
     // todo: this is somehow unfair: the first connections in the map get checked more
     // often than the others and can thus block them out
 
-    std::map<unsigned int, connection*>::iterator ie=connections.end();
-    for(std::map<unsigned int, connection*>::iterator i=connections.begin(); i != ie; i++)
-        if (i->second->get_packet(data,conn_id))
+    std::map<unsigned int, server_connection*>::iterator ie=connections.end();
+    for(std::map<unsigned int, server_connection*>::iterator i=connections.begin(); i != ie; i++)
+        if (i->second->get_packet(data))
+        {
+            conn_id=i->first;
             return true;
+        }
+
+    return false;
 }
 
 void server::log(log_level_values level, const char* message)