3424cd832c56fe892e6f71419c3a87d7eb31ec02
[libt2n] / src / server.cpp
1 /***************************************************************************
2  *   Copyright (C) 2006 by Gerd v. Egidy                                   *
3  *   gve@intra2net.com                                                     *
4  *                                                                         *
5  *   This library is free software; you can redistribute it and/or modify  *
6  *   it under the terms of the GNU Lesser General Public License version   *
7  *   2.1 as published by the Free Software Foundation.                     *
8  *                                                                         *
9  *   This library is distributed in the hope that it will be useful,       *
10  *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
11  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
12  *   GNU Lesser General Public License for more details.                   *
13  *                                                                         *
14  *   You should have received a copy of the GNU Lesser General Public      *
15  *   License along with this program; if not, write to the                 *
16  *   Free Software Foundation, Inc.,                                       *
17  *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
18  ***************************************************************************/
19
20 #include <sstream>
21
22 #include "server.hxx"
23
24 namespace libt2n
25 {
26
27 void connection::check_timeout()
28 {
29     if (timeout != -1 && last_action_time+timeout >= time(NULL))
30         this->close();
31 }
32
33 void connection::reset_timeout()
34 {
35     last_action_time=time(NULL);
36 }
37
38 std::string connection::get_id_string()
39 {
40     std::ostringstream os;
41     os << get_id();
42     return os.str();
43 }
44
45 bool connection::get_packet(std::string& data, unsigned int& conn_id)
46 {
47     // max packet size is unsigned int
48
49     // no size information -> no packet
50     if (buffer.size() < sizeof(unsigned int))
51         return false;
52
53     unsigned int psize=*((unsigned int*)(buffer.data()));
54
55     // enough data for one packet in buffer?
56     if (buffer.size() < sizeof(unsigned int)+psize)
57         return false;
58
59     data.assign(buffer,sizeof(unsigned int),psize);
60     buffer.erase(0,sizeof(unsigned int)+psize);
61
62     return true;
63 }
64
65 server::~server()
66 {
67     std::map<unsigned int, connection*>::iterator ie=connections.end();
68     for(std::map<unsigned int, connection*>::iterator i=connections.begin(); i != ie; i++)
69         delete i->second;
70 }
71
72 int server::add_connection(connection* newconn)
73 {
74     unsigned int cid=next_id++;
75     newconn->set_id(cid);
76     newconn->set_server(this);
77     connections[cid]=newconn;
78     return cid;
79 }
80
81 /**
82     Gets a connection by id
83     
84     \param conn_id Connection ID
85     
86     \retval Pointer to connection object
87 */
88 connection* server::get_connection(unsigned int conn_id)
89 {
90     std::map<unsigned int, connection*>::iterator p=connections.find(conn_id);
91     if (p==connections.end())
92         return NULL;
93     else
94         return p->second;
95 }
96
97 void server::check_timeout()
98 {
99     std::map<unsigned int, connection*>::iterator ie=connections.end();
100     for(std::map<unsigned int, connection*>::iterator i=connections.begin(); i != ie; i++)
101         i->second->check_timeout();
102 }
103
104 bool server::get_packet(std::string& data, unsigned int& conn_id)
105 {
106     // todo: this is somehow unfair: the first connections in the map get checked more
107     // often than the others and can thus block them out
108
109     std::map<unsigned int, connection*>::iterator ie=connections.end();
110     for(std::map<unsigned int, connection*>::iterator i=connections.begin(); i != ie; i++)
111         if (i->second->get_packet(data,conn_id))
112             return true;
113 }
114
115 void server::log(log_level_values level, const char* message)
116 {
117     if (logstream && level >= log_level)
118         (*logstream) << message << std::endl;
119 }
120
121 };