1

     2

#include "Session.hh"

     3

#include "../Engine.hh"

     4

     5

#include <cassert>

     6

     7

NetworkSession::NetworkSession (uint64_t magic) :

     8

    magic(magic), tcp_srv(NULL), udp_srv(NULL), udp_client(NULL) {

     9

    10

    // nothing

    11

}

    12

    14

    assert(tcp_srv == NULL && udp_srv == NULL);

    15

    16

    // create TCP/UDP servers

    17

    tcp_srv = new NetworkTCPServer(addr);

    18

    udp_srv = new NetworkUDP(addr);

    19

    20

    // connect signals

    21

    slots.connect(tcp_srv->sig_client(), this, &NetworkSession::on_tcp_client);

    22

    slots.connect(udp_srv->sig_packet(), this, &NetworkSession::on_udp_packet);

    23

}

    24

    25

NetworkNode *NetworkSession::build_node (NetworkTCPTransport *tcp, NetworkUDP *udp, const NetworkAddress &addr, enum NetworkNodeType type) {

    26

    // XXX: unused

    27

    (void) type;

    28

    29

    // create node

    30

    return new NetworkNode(*this, tcp, udp, addr);

    31

}

    32

    34

    // XXX: only one connect

    35

    assert(!udp_client);

    36

    37

    // connect TCP

    38

    NetworkTCPClient *tcp_client = new NetworkTCPClient(addr);

    39

    40

    // create UDP socket on same address

    41

    udp_client = new NetworkUDP(tcp_client->get_local_address());

    42

    43

    // remote address

    44

    NetworkAddress remote_address = tcp_client->get_remote_address();

    45

    46

    // build client

    47

    NetworkNode *client_node = build_node(tcp_client, udp_client, remote_address, NETWORK_NODE_CLIENT_SERVER);

    48

    49

    // add to nodes

    50

    nodes[remote_address] = client_node;

    51

    52

    // connect signals

    53

    slots.connect(udp_client->sig_packet(), this, &NetworkSession::on_udp_packet);

    54

    55

    // return the "server" node

    56

    return client_node;

    57

}

    58

    59

void NetworkSession::handle_disconnect (NetworkNode *node) {

    60

    // remove from nodes

    61

    nodes.erase(node->getRemoteAddress());

    62

}

    63

    64

void NetworkSession::handle_message (NetworkPacketInput &pkt, NetworkNode *node) {

    65

    // read the channel id

    66

    NetworkChannelID channel_id = pkt.read_uint16();

    67

    68

    // fire signal

    69

    _map_sig_chan_message[channel_id](pkt, node);

    70

}

    71

    72

void NetworkSession::on_tcp_client (NetworkTCPTransport *tcp_client) {

    73

    // get remote address manually, because NetworkTCPServer doesn't pass it in to us

    74

    NetworkAddress addr = tcp_client->get_remote_address();

    75

    76

    // build client

    77

    NetworkNode *client_node = build_node(tcp_client, udp_srv, addr, NETWORK_NODE_SERVER_CLIENT);

    78

    79

    // add to nodes

    80

    nodes[addr] = client_node;

    81

    82

    // fire signals

    83

    _sig_node_connected(client_node);

    84

}

    85

    86

void NetworkSession::on_udp_packet (NetworkPacketInput &pkt, const NetworkAddress &addr) {

    87

    NetworkNode *node = nodes[addr];

    88

    89

    // drop from unknown sources

    90

    if (!node) {

    91

        Engine::log(WARN, "net_session.on_udp_packet") << "dropping unsolicted UDP packet from " << addr;

    92

        return;

    93

    }

    94

    95

    // handle

    96

    handle_message(pkt, node);

    97

}

    98

    99

void NetworkSession::send_all (NetworkChannelID channel_id, const NetworkPacketBuffer &pkt, bool reliable) {

   100

    send_all_except(channel_id, pkt, NULL, reliable);

   101

}

   102

   103

void NetworkSession::send_all_except (NetworkChannelID channel_id, const NetworkPacketBuffer &pkt, const NetworkNode *node, bool reliable) {

   104

    for (std::map<NetworkAddress, NetworkNode*>::iterator it = nodes.begin(); it != nodes.end(); it++) {

   105

        if (it->second == node)

   106

            continue;

   107

   108

        it->second->send(channel_id, pkt, reliable);

   109

    }

   110

}