|
33
|
1 |
|
|
|
2 |
#include "NetworkSession.hh"
|
|
|
3 |
|
|
|
4 |
NetworkSession::NetworkSession (NetworkSessionMagic magic) :
|
|
|
5 |
magic(magic), tcp_srv(NULL), udp_srv(NULL), udp_client(NULL) {
|
|
|
6 |
|
|
|
7 |
// nothing
|
|
|
8 |
}
|
|
|
9 |
|
|
|
10 |
void NetworkSession::listen (const NetworkAddress &addr) {
|
|
|
11 |
assert(tcp_srv == NULL && udp_srv == NULL);
|
|
|
12 |
|
|
|
13 |
// create TCP/UDP servers
|
|
|
14 |
tcp_srv = new NetworkTCPServer(addr);
|
|
|
15 |
udp_srv = new NetworkUDP(addr);
|
|
|
16 |
|
|
|
17 |
// connect signals
|
|
34
|
18 |
slots.connect(tcp->sig_client(), this, &NetworkSession::on_tcp_client);
|
|
33
|
19 |
}
|
|
|
20 |
|
|
36
|
21 |
NetworkNode *NetworkSession::build_node (NetworkTCPTransport *tcp_client, NetworkUDP *udp, const NetworkAddress &addr, , enum NetworkNodeType type) {
|
|
|
22 |
// XXX: unused
|
|
|
23 |
(void) type;
|
|
|
24 |
|
|
|
25 |
// create node
|
|
|
26 |
return new NetworkNode(*this, tcp_client, addr);
|
|
|
27 |
}
|
|
|
28 |
|
|
33
|
29 |
NetworkNode* NetworkSession::connect (const NetworkAddress &addr) {
|
|
|
30 |
// create new UDP client if needed
|
|
|
31 |
if (udp_client == NULL)
|
|
|
32 |
udp_client = new NetworkUDP();
|
|
|
33 |
|
|
|
34 |
// connect
|
|
|
35 |
NetworkTCPClient *tcp_client = new NetworkTCPClient(addr);
|
|
36
|
36 |
|
|
|
37 |
// build client
|
|
|
38 |
NetworkNode *client_node = build_node(tcp_client, udp_client, addr, NETWORK_NODE_CLIENT_SERVER);
|
|
33
|
39 |
|
|
|
40 |
// add to nodes
|
|
|
41 |
nodes[addr] = client_node;
|
|
|
42 |
|
|
|
43 |
// bind signals
|
|
34
|
44 |
slots.connect(tcp_client->sig_disconnect(), this, &NetworkSession::on_disconnect, client_node);
|
|
33
|
45 |
}
|
|
|
46 |
|
|
36
|
47 |
void NetworkSession::on_tcp_client (NetworkTCPTransport *tcp_client) {
|
|
33
|
48 |
// get remote address manually, because NetworkTCPServer doesn't pass it in to us
|
|
|
49 |
NetworkAddress addr = tcp_client->getRemoteAddress();
|
|
|
50 |
|
|
36
|
51 |
// build client
|
|
|
52 |
NetworkNode *client_node = build_node(tcp_client, udp_server, addr, NETWORK_NODE_SERVER_CLIENT);
|
|
33
|
53 |
|
|
|
54 |
// add to nodes
|
|
|
55 |
nodes[addr] = client_node;
|
|
|
56 |
|
|
|
57 |
// fire signals
|
|
34
|
58 |
_sig_node_connected(node);
|
|
33
|
59 |
}
|
|
|
60 |
|
|
36
|
61 |
void NetworkSession::handle_disconnect (NetworkNode *node) {
|
|
33
|
62 |
// remove from nodes
|
|
|
63 |
nodes.erase(node->getRemoteAddress());
|
|
|
64 |
}
|
|
|
65 |
|
|
|
66 |
void NetworkSession::send_all (NetworkChannelID channel_id, const NetworkPacket &pkt, bool reliable) {
|
|
|
67 |
send_all_except(NULL, reliable);
|
|
|
68 |
}
|
|
|
69 |
|
|
|
70 |
void NetworkSession::send_all_except (NetworkChannelID channel_id, const NetworkPacket &pkt, const NetworkNode *node, bool reliable) {
|
|
|
71 |
for (std::map<NetworkAddress, NetworkNode*>::iterator it = nodes.begin(); it != nodes.end(); it++) {
|
|
|
72 |
if (it->second == node)
|
|
|
73 |
continue;
|
|
|
74 |
|
|
|
75 |
it->second->send(channel_id, pkt, reliable);
|
|
|
76 |
}
|
|
|
77 |
}
|