WebSVN – HomeAutomation – Diff – /trunk/PcSoftware/Atom/src/Socket/server.cpp


/***************************************************************************
 *   Copyright (C) January 4, 2009 by Mattias Runge                             *
 *   mattias@runge.se                                                      *
 *   server.cpp                                            *
 *                                                                         *
 *   This program is free software; you can redistribute it and/or modify  *
 *   it under the terms of the GNU General Public License as published by  *
 *   the Free Software Foundation; either version 2 of the License, or     *
 *   (at your option) any later version.                                   *
 *                                                                         *
 *   This program is distributed in the hope that it will be useful,       *
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
 *   GNU General Public License for more details.                          *
 *                                                                         *
 *   You should have received a copy of the GNU General Public License     *
 *   along with this program; if not, write to the                         *
 *   Free Software Foundation, Inc.,                                       *
 *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
 ***************************************************************************/
 
#include "server.h"
 
Server::Server(int port, string name)
{
    setSettings(port, name);
}
 
Server::~Server()
{
    stop();
 
    for (map<int, AsyncSocket*>::iterator iter = myClientSockets.begin(); iter != myClientSockets.end(); iter++)
    {
        if (iter->second->isConnected())
        {
            iter->second->stop();
        }
 
        delete iter->second;
    }
}
 
void Server::setSettings(int port, string name)
{
    myName = name;
    mySocket.setPort(port);
}
 
void Server::run()
{
    Logger &log = Logger::getInstance();
 
    try
    {
        mySocket.startListen();
 
        log.add(myName + " is listening for connections on port " + itos(mySocket.getPort()) + ".\n");
 
        while (true)
        {
            AsyncSocket *newSocket = new AsyncSocket();
 
            if (mySocket.accept(newSocket))
            {
                newSocket->sendData("Welcome to Atom " + ftos(VERSION) + " - " + myName + "\r\n");
 
                log.add(myName + " accepted new client with id " + itos(newSocket->getId()) + ".\n");
                myClientSockets[newSocket->getId()] = newSocket;
                myClientSockets[newSocket->getId()]->eventSetCallback(this);
                myClientSockets[newSocket->getId()]->start();
            }
        }
    }
    catch (SocketException* e)
    {
        log.add(myName + " caught an exception:" + e->getDescription() + ".\n");
    }
}
 
void Server::sendToAll(string data)
{
    Logger &log = Logger::getInstance();
 
    for (map<int, AsyncSocket*>::iterator iter = myClientSockets.begin(); iter != myClientSockets.end(); iter++)
    {
        if (iter->second->isConnected())
        {
            //log.add("Server: sent data to client " + itos(iter->second->getSocket()) + ".\n";
 
            try
            {
                iter->second->sendData(data);
            }
            catch (SocketException* e)
            {
                log.add(myName + " caught an exception:" + e->getDescription() + ".\n");
            }
        }
        else
        {
            log.add(myName + " had a client disconnect.\n");
            delete iter->second;
            myClientSockets.erase(iter);
        }
    }
}
 
void Server::sendTo(int id, string data)
{
    Logger &log = Logger::getInstance();
 
    if (myClientSockets.find(id) != myClientSockets.end())
    {
        if (myClientSockets[id]->isConnected())
        {
            //log.add("Server: sent data to client " + itos(iter->second->getSocket()) + ".\n";
 
            try
            {
                myClientSockets[id]->sendData(data);
            }
            catch (SocketException* e)
            {
                log.add(myName + " caught an exception:" + e->getDescription() + ".\n");
            }
        }
        else
        {
            log.add(myName + " had a client disconnect.\n");
           
            delete myClientSockets[id];
            myClientSockets[id]=0;
            myClientSockets.erase(id);
        }
    }
}
 
void Server::disconnectClient(int id)
{
    Logger &log = Logger::getInstance();
 
    if (myClientSockets.find(id) != myClientSockets.end())
    {
//      log.add(myName + " had a client disconnect!.\n");
       
//      delete myClientSockets[id];
 
//      myClientSockets.erase(id);
//delete myClientSockets[id];
//          myClientSockets[id]=0;
//          myClientSockets.erase(id);
    }
}
 
void Server::handleEvent(int id, SocketEvent socketEvent)
{
    switch (socketEvent.getType())
    {
        case SocketEvent::TYPE_CONNECTION_CLOSED:
        case SocketEvent::TYPE_CONNECTION_DIED:
        case SocketEvent::TYPE_CONNECTION_RESET:
        disconnectClient(id);
        break;
 
        case SocketEvent::TYPE_DATA:
        handleClientData(id, socketEvent.getData());
        break;
    }
}
 
void Server::handleClientData(int id, string data)
{
    Logger &log = Logger::getInstance();
    log.add(myName + " received data from client " + itos(id) + ": \"" + data + "\"\n");
}
 

Subversion Repositories HomeAutomation

(root)/trunk/PcSoftware/Atom/src/Socket/server.cpp – Rev 1026 → 1122

Rev 1026		Rev 1122
1	/***************************************************************************	1	/***************************************************************************
2	* Copyright (C) January 4, 2009 by Mattias Runge *	2	* Copyright (C) January 4, 2009 by Mattias Runge *
3	* mattias@runge.se *	3	* mattias@runge.se *
4	* server.cpp *	4	* server.cpp *
5	* *	5	* *
6	* This program is free software; you can redistribute it and/or modify *	6	* This program is free software; you can redistribute it and/or modify *
7	* it under the terms of the GNU General Public License as published by *	7	* it under the terms of the GNU General Public License as published by *
8	* the Free Software Foundation; either version 2 of the License, or *	8	* the Free Software Foundation; either version 2 of the License, or *
9	* (at your option) any later version. *	9	* (at your option) any later version. *
10	* *	10	* *
11	* This program is distributed in the hope that it will be useful, *	11	* This program is distributed in the hope that it will be useful, *
12	* but WITHOUT ANY WARRANTY; without even the implied warranty of *	12	* but WITHOUT ANY WARRANTY; without even the implied warranty of *
13	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *	13	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
14	* GNU General Public License for more details. *	14	* GNU General Public License for more details. *
15	* *	15	* *
16	* You should have received a copy of the GNU General Public License *	16	* You should have received a copy of the GNU General Public License *
17	* along with this program; if not, write to the *	17	* along with this program; if not, write to the *
18	* Free Software Foundation, Inc., *	18	* Free Software Foundation, Inc., *
19	* 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. *	19	* 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. *
20	***************************************************************************/	20	***************************************************************************/
21		21
22	#include "server.h"	22	#include "server.h"
23		23
24	Server::Server(int port, string name)	24	Server::Server(int port, string name)
25	{	25	{
26	setSettings(port, name);	26	setSettings(port, name);
27	}	27	}
28		28
29	Server::~Server()	29	Server::~Server()
30	{	30	{
31	stop();	31	stop();
32		32
33	for (map<int, AsyncSocket*>::iterator iter = myClientSockets.begin(); iter != myClientSockets.end(); iter++)	33	for (map<int, AsyncSocket*>::iterator iter = myClientSockets.begin(); iter != myClientSockets.end(); iter++)
34	{	34	{
35	if (iter->second->isConnected())	35	if (iter->second->isConnected())
36	{	36	{
37	iter->second->stop();	37	iter->second->stop();
38	}	38	}
39		39
40	delete iter->second;	40	delete iter->second;
41	}	41	}
42	}	42	}
43		43
44	void Server::setSettings(int port, string name)	44	void Server::setSettings(int port, string name)
45	{	45	{
46	myName = name;	46	myName = name;
47	mySocket.setPort(port);	47	mySocket.setPort(port);
48	}	48	}
49		49
50	void Server::run()	50	void Server::run()
51	{	51	{
52	Logger &log = Logger::getInstance();	52	Logger &log = Logger::getInstance();
53		53
54	try	54	try
55	{	55	{
56	mySocket.startListen();	56	mySocket.startListen();
57		57
58	log.add(myName + " is listening for connections on port " + itos(mySocket.getPort()) + ".\n");	58	log.add(myName + " is listening for connections on port " + itos(mySocket.getPort()) + ".\n");
59		59
60	while (true)	60	while (true)
61	{	61	{
62	AsyncSocket *newSocket = new AsyncSocket();	62	AsyncSocket *newSocket = new AsyncSocket();
63		63
64	if (mySocket.accept(newSocket))	64	if (mySocket.accept(newSocket))
65	{	65	{
66	newSocket->sendData("Welcome to Atom " + ftos(VERSION) + " - " + myName + "\n");	66	newSocket->sendData("Welcome to Atom " + ftos(VERSION) + " - " + myName + "\r\n");
67		67
68	log.add(myName + " accepted new client with id " + itos(newSocket->getId()) + ".\n");	68	log.add(myName + " accepted new client with id " + itos(newSocket->getId()) + ".\n");
69	myClientSockets[newSocket->getId()] = newSocket;	69	myClientSockets[newSocket->getId()] = newSocket;
70	myClientSockets[newSocket->getId()]->eventSetCallback(this);	70	myClientSockets[newSocket->getId()]->eventSetCallback(this);
71	myClientSockets[newSocket->getId()]->start();	71	myClientSockets[newSocket->getId()]->start();
72	}	72	}
73	}	73	}
74	}	74	}
75	catch (SocketException* e)	75	catch (SocketException* e)
76	{	76	{
77	log.add(myName + " caught an exception:" + e->getDescription() + ".\n");	77	log.add(myName + " caught an exception:" + e->getDescription() + ".\n");
78	}	78	}
79	}	79	}
80		80
81	void Server::sendToAll(string data)	81	void Server::sendToAll(string data)
82	{	82	{
83	Logger &log = Logger::getInstance();	83	Logger &log = Logger::getInstance();
84		84
85	for (map<int, AsyncSocket*>::iterator iter = myClientSockets.begin(); iter != myClientSockets.end(); iter++)	85	for (map<int, AsyncSocket*>::iterator iter = myClientSockets.begin(); iter != myClientSockets.end(); iter++)
86	{	86	{
87	if (iter->second->isConnected())	87	if (iter->second->isConnected())
88	{	88	{
89	//log.add("Server: sent data to client " + itos(iter->second->getSocket()) + ".\n";	89	//log.add("Server: sent data to client " + itos(iter->second->getSocket()) + ".\n";
90		90
91	try	91	try
92	{	92	{
93	iter->second->sendData(data);	93	iter->second->sendData(data);
94	}	94	}
95	catch (SocketException* e)	95	catch (SocketException* e)
96	{	96	{
97	log.add(myName + " caught an exception:" + e->getDescription() + ".\n");	97	log.add(myName + " caught an exception:" + e->getDescription() + ".\n");
98	}	98	}
99	}	99	}
100	else	100	else
101	{	101	{
102	log.add(myName + " had a client disconnect.\n");	102	log.add(myName + " had a client disconnect.\n");
103	delete iter->second;	103	delete iter->second;
104	myClientSockets.erase(iter);	104	myClientSockets.erase(iter);
105	}	105	}
106	}	106	}
107	}	107	}
108		108
109	void Server::sendTo(int id, string data)	109	void Server::sendTo(int id, string data)
110	{	110	{
111	Logger &log = Logger::getInstance();	111	Logger &log = Logger::getInstance();
112		112
113	if (myClientSockets.find(id) != myClientSockets.end())	113	if (myClientSockets.find(id) != myClientSockets.end())
114	{	114	{
115	if (myClientSockets[id]->isConnected())	115	if (myClientSockets[id]->isConnected())
116	{	116	{
117	//log.add("Server: sent data to client " + itos(iter->second->getSocket()) + ".\n";	117	//log.add("Server: sent data to client " + itos(iter->second->getSocket()) + ".\n";
118		118
119	try	119	try
120	{	120	{
121	myClientSockets[id]->sendData(data);	121	myClientSockets[id]->sendData(data);
122	}	122	}
123	catch (SocketException* e)	123	catch (SocketException* e)
124	{	124	{
125	log.add(myName + " caught an exception:" + e->getDescription() + ".\n");	125	log.add(myName + " caught an exception:" + e->getDescription() + ".\n");
126	}	126	}
127	}	127	}
128	else	128	else
129	{	129	{
130	log.add(myName + " had a client disconnect.\n");	130	log.add(myName + " had a client disconnect.\n");
-		131
131	delete myClientSockets[id];	132	delete myClientSockets[id];
-		133	myClientSockets[id]=0;
132	myClientSockets.erase(id);	134	myClientSockets.erase(id);
133	}	135	}
134	}	136	}
135	}	137	}
136		138
137	void Server::disconnectClient(int id)	139	void Server::disconnectClient(int id)
138	{	140	{
139	Logger &log = Logger::getInstance();	141	Logger &log = Logger::getInstance();
140		142
141	if (myClientSockets.find(id) != myClientSockets.end())	143	if (myClientSockets.find(id) != myClientSockets.end())
142	{	144	{
143	log.add(myName + " had a client ~~disconnect~~.\n");	145	// log.add(myName + " had a client disconnect!.\n");
-		146
144	delete myClientSockets[id];	147	// delete myClientSockets[id];
-		148
145	myClientSockets.erase(id);	149	// myClientSockets.erase(id);
-		150	//delete myClientSockets[id];
-		151	// myClientSockets[id]=0;
-		152	// myClientSockets.erase(id);
146	}	153	}
147	}	154	}
148		155
149	void Server::handleEvent(int id, SocketEvent socketEvent)	156	void Server::handleEvent(int id, SocketEvent socketEvent)
150	{	157	{
151	switch (socketEvent.getType())	158	switch (socketEvent.getType())
152	{	159	{
153	case SocketEvent::TYPE_CONNECTION_CLOSED:	160	case SocketEvent::TYPE_CONNECTION_CLOSED:
154	case SocketEvent::TYPE_CONNECTION_DIED:	161	case SocketEvent::TYPE_CONNECTION_DIED:
155	case SocketEvent::TYPE_CONNECTION_RESET:	162	case SocketEvent::TYPE_CONNECTION_RESET:
156	disconnectClient(id);	163	disconnectClient(id);
157	break;	164	break;
158		165
159	case SocketEvent::TYPE_DATA:	166	case SocketEvent::TYPE_DATA:
160	handleClientData(id, socketEvent.getData());	167	handleClientData(id, socketEvent.getData());
161	break;	168	break;
162	}	169	}
163	}	170	}
164		171
165	void Server::handleClientData(int id, string data)	172	void Server::handleClientData(int id, string data)
166	{	173	{
167	Logger &log = Logger::getInstance();	174	Logger &log = Logger::getInstance();
168	log.add(myName + " received data from client " + itos(id) + ": \"" + data + "\"\n");	175	log.add(myName + " received data from client " + itos(id) + ": \"" + data + "\"\n");
169	}	176	}
170		177