#include "NetworkSession.hh"
#include <cassert>
NetworkSession::NetworkSession (uint64_t magic) :
magic(magic), tcp_srv(NULL), udp_srv(NULL), udp_client(NULL) {
// nothing
}
void NetworkSession::listen (const NetworkAddress &addr) {
assert(tcp_srv == NULL && udp_srv == NULL);
// create TCP/UDP servers
tcp_srv = new NetworkTCPServer(addr);
udp_srv = new NetworkUDP(addr);
// connect signals
slots.connect(tcp_srv->sig_client(), this, &NetworkSession::on_tcp_client);
}
NetworkNode *NetworkSession::build_node (NetworkTCPTransport *tcp, NetworkUDP *udp, const NetworkAddress &addr, enum NetworkNodeType type) {
// XXX: unused
(void) type;
// create node
return new NetworkNode(*this, tcp, udp, addr);
}
NetworkNode* NetworkSession::connect (const NetworkAddress &addr) {
// create new UDP client if needed
if (udp_client == NULL)
udp_client = new NetworkUDP();
// connect
NetworkTCPClient *tcp_client = new NetworkTCPClient(addr);
// build client
NetworkNode *client_node = build_node(tcp_client, udp_client, addr, NETWORK_NODE_CLIENT_SERVER);
// add to nodes
nodes[addr] = client_node;
// return the "server" node
return client_node;
}
void NetworkSession::on_tcp_client (NetworkTCPTransport *tcp_client) {
// get remote address manually, because NetworkTCPServer doesn't pass it in to us
NetworkAddress addr = tcp_client->getRemoteAddress();
// build client
NetworkNode *client_node = build_node(tcp_client, udp_srv, addr, NETWORK_NODE_SERVER_CLIENT);
// add to nodes
nodes[addr] = client_node;
// fire signals
_sig_node_connected(client_node);
}
void NetworkSession::handle_disconnect (NetworkNode *node) {
// remove from nodes
nodes.erase(node->getRemoteAddress());
}
void NetworkSession::send_all (NetworkChannelID channel_id, const NetworkPacket &pkt, bool reliable) {
send_all_except(channel_id, pkt, NULL, reliable);
}
void NetworkSession::send_all_except (NetworkChannelID channel_id, const NetworkPacket &pkt, const NetworkNode *node, bool reliable) {
for (std::map<NetworkAddress, NetworkNode*>::iterator it = nodes.begin(); it != nodes.end(); it++) {
if (it->second == node)
continue;
it->second->send(channel_id, pkt, reliable);
}
}