1

#ifndef NETWORK_SESSION_HH

     2

#define NETWORK_SESSION_HH

     3

     4

#include <map>

     5

#include <stdint.h>

     6

     7

// forward-declare

     8

class NetworkSession;

     9

    10

/*

    11

 * Used to separate packets, ID zero is reserved for NetworkSession use

    12

 */

    13

typedef uint16_t NetworkChannelID;

    14

    15

/*

    16

 * Size of a packet header

    17

 */

    18

const size_t NETWORK_SESSION_HEADER_SIZE = sizeof(uint16_t);

    19

    20

#include "TCP.hh"

    21

#include "UDP.hh"

    22

#include "Node.hh"

    23

    24

class NetworkSession {

    25

    friend class NetworkNode;

    26

    27

    private:

    28

        uint64_t magic;

    29

        NetworkTCPServer *tcp_srv;

    30

        NetworkUDP *udp_srv, *udp_client;

    31

    32

        CL_SlotContainer slots;

    33

    34

        std::map<NetworkAddress, NetworkNode*> nodes;

    35

        std::map<NetworkChannelID, CL_Signal_v2<NetworkPacketInput&, NetworkNode *> > _map_sig_chan_message;

    36

    37

    public:

    38

        NetworkSession (uint64_t magic);

    39

    40

        void listen (const NetworkAddress &addr);

    41

        NetworkNode* connect (const NetworkAddress &addr);

    42

    43

    protected:

    44

        virtual NetworkNode *build_node (NetworkTCPTransport *tcp, NetworkUDP *udp, const NetworkAddress &addr, enum NetworkNodeType type);

    45

    46

        void handle_disconnect (NetworkNode *node);

    47

        void handle_message (NetworkPacketInput &pkt, NetworkNode *node);

    48

    49

    private:

    50

        void on_tcp_client (NetworkTCPTransport *client);

    51

        void on_udp_packet (NetworkPacketInput &pkt, const NetworkAddress &addr);

    52

    53

        CL_Signal_v1<NetworkNode*> _sig_node_connected;

    54

    55

    public:

    56

        void send_all (NetworkChannelID channel_id, const NetworkPacketBuffer &pkt, bool reliable = true);

    57

        void send_all_except (NetworkChannelID channel_id, const NetworkPacketBuffer &pkt, const NetworkNode *node, bool reliable = true);

    58

    59

    60

        CL_Signal_v1<NetworkNode*>& sig_node_connected (void) { return _sig_node_connected; }

    61

        CL_Signal_v2<NetworkPacketInput&, NetworkNode *>& sig_chan_message (NetworkChannelID cid) { return _map_sig_chan_message[cid]; }

    62

};

    63

    64

#endif /* NETWORK_SESSION_HH */