1

     2

#include <cassert>

     3

     4

#include "Node.hh"

     5

     6

NetworkNode::NetworkNode (NetworkSession &session, NetworkTCPTransport *tcp, NetworkUDP *udp, const NetworkAddress &address) :

     7

    session(session), tcp(tcp), udp(udp), address(address) {

     8

     9

    // connect signals

    10

    slots.connect(tcp->sig_disconnect(), this, &NetworkNode::on_disconnect);

    11

    slots.connect(tcp->sig_packet(), &session, &NetworkSession::handle_message, this);

    12

    13

}

    14

    15

NetworkNode::~NetworkNode (void) {

    16

    delete tcp;

    17

}

    18

    19

void NetworkNode::on_disconnect (void) {

    20

    // tell session

    21

    session.handle_disconnect(this);

    22

    23

    // fire signal

    24

    _sig_disconnected();

    25

    26

    // delete

    27

//    delete this;

    28

}

    29

    30

void NetworkNode::write_packet_header (NetworkPacketOutput &pkt, NetworkChannelID channel_id) {

    31

    pkt.write_uint16(channel_id);

    32

}

    33

    34

void NetworkNode::send_raw (const NetworkPacketBuffer &pkt, bool reliable) {

    35

    // either tcp or udp

    36

    if (reliable) {

    37

        assert(tcp);

    38

    39

        tcp->write_packet(pkt);

    40

    41

    } else {

    42

        udp->sendto(pkt, address);

    43

    }

    44

    45

}

    46

    47

void NetworkNode::send (NetworkChannelID channel_id, const NetworkPacketBuffer &pkt, bool reliable) {

    48

    assert(channel_id > 0);

    49

    50

    // add our header

    51

    NetworkPacket pkt2;

    52

    53

    write_packet_header(pkt2, channel_id);

    54

    pkt2.write_packet(pkt);

    55

    56

    // send

    57

    send_raw(pkt2, reliable);

    58

}

    59

    60

const NetworkAddress& NetworkNode::getRemoteAddress (void) {

    61

    return address;

    62

}