1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
|
/*
* The Mana World Server
* Copyright 2004 The Mana World Development Team
*
* This file is part of The Mana World.
*
* The Mana World 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 any later version.
*
* The Mana World 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 The Mana World; if not, write to the Free Software Foundation, Inc.,
* 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
*
* $Id$
*/
#include "connectionhandler.h"
#include "netsession.h"
#include "utils/logger.h"
#ifdef SCRIPT_SUPPORT
#include "script.h"
#endif
#define MAX_CLIENTS 1024
ClientData::ClientData():
inp(0)
{
}
ConnectionHandler::ConnectionHandler()
{
}
void ConnectionHandler::startListen(ListenThreadData *ltd)
{
// Allocate a socket set
SDLNet_SocketSet set = SDLNet_AllocSocketSet(MAX_CLIENTS);
if (!set) {
LOG_FATAL("SDLNet_AllocSocketSet: " << SDLNet_GetError())
exit(1);
}
// Add the server socket to the socket set
if (SDLNet_TCP_AddSocket(set, ltd->socket) < 0) {
LOG_FATAL("SDLNet_AddSocket: " << SDLNet_GetError())
exit(1);
}
// Keep checking for socket activity while running
while (ltd->running) {
int numready = SDLNet_CheckSockets(set, 100);
if (numready == -1) {
LOG_ERROR("SDLNet_CheckSockets: " << SDLNet_GetError())
// When this is a system error, perror may help us
perror("SDLNet_CheckSockets");
}
else if (numready > 0) {
LOG_INFO(numready << " sockets with activity!")
// Check server socket
if (SDLNet_SocketReady(ltd->socket)) {
TCPsocket client = SDLNet_TCP_Accept(ltd->socket);
if (client) {
// Add the client socket to the socket set
if (SDLNet_TCP_AddSocket(set, client) < 0) {
LOG_ERROR("SDLNet_AddSocket: " << SDLNet_GetError())
}
else {
NetComputer *comp = new NetComputer(this);
clients[comp] = client;
computerConnected(comp);
LOG_INFO(clients.size() << " clients connected")
}
}
}
// Check client sockets
std::map<NetComputer*, TCPsocket>::iterator i;
for (i = clients.begin(); i != clients.end(); ) {
NetComputer *comp = (*i).first;
TCPsocket s = (*i).second;
if (SDLNet_SocketReady(s)) {
char buffer[1024];
int result = SDLNet_TCP_Recv(s, buffer, 1024);
if (result <= 0) {
SDLNet_TCP_DelSocket(set, s);
SDLNet_TCP_Close(s);
computerDisconnected(comp);
delete comp;
comp = NULL;
}
else {
// Copy the incoming data to the in buffer of this
// client
buffer[result] = 0;
LOG_INFO("Received " << buffer);
#ifdef SCRIPT_SUPPORT
//script->message(buffer);
#endif
}
}
// Traverse to next client, possibly deleting current
if (comp == NULL) {
std::map<NetComputer*, TCPsocket>::iterator ii = i;
ii++;
clients.erase(i);
i = ii;
}
else {
i++;
}
}
}
}
// - Disconnect all clients (close sockets)
SDLNet_FreeSocketSet(set);
}
void ConnectionHandler::computerConnected(NetComputer *comp)
{
LOG_INFO("A client connected!")
}
void ConnectionHandler::computerDisconnected(NetComputer *comp)
{
LOG_INFO("A client disconnected!")
}
void ConnectionHandler::registerHandler(
unsigned int msgId, MessageHandler *handler)
{
handlers[msgId] = handler;
}
|
