1

#ifndef NETWORK_TCP_HH

     2

#define NETWORK_TCP_HH

     3

     4

#include <ClanLib/core.h>

     5

     6

#include "NetworkSocket.hh"

     7

#include "NetworkAddress.hh"

     8

#include "NetworkPacket.hh"

     9

#include "Error.hh"

    10

    11

const size_t NETWORK_TCP_CHUNK_SIZE = 1024;

    12

const size_t NETWORK_TCP_INITIAL_IN_BUF = 4096;

    13

const size_t NETWORK_TCP_INITIAL_OUT_BUF = 0;

    14

    15

class NetworkBufferError : public Error {

    16

    public:

    17

        NetworkBufferError (const std::string &message) : Error(message) { }

    18

};

    19

    20

class NetworkBuffer {

    21

    private:

    22

        NetworkSocket socket;

    23

    24

        char *buf;

    25

        size_t size, offset;

    26

    27

    public:

    28

        NetworkBuffer (NetworkSocket &socket, size_t size_hint);

    29

        ~NetworkBuffer (void);

    30

    31

    private:

    32

        NetworkBuffer (const NetworkBuffer &copy);

    33

        NetworkBuffer& operator= (const NetworkBuffer &copy);

    34

    35

        void resize (size_t item_size);

    36

        void trim (size_t prefix_size);

    37

    38

    public:    

    39

        void push_write (char *buf_ptr, size_t buf_size);

    40

        void flush_write (void);

    41

        void write_prefix (char *buf, uint16_t prefix);

    42

        void write_prefix (char *buf, uint32_t prefix);

    43

    44

        bool try_read (size_t item_size);

    45

        bool peek_prefix (uint16_t &ref);

    46

        bool peek_prefix (uint32_t &ref);

    47

        template <typename PrefixType> PrefixType read_prefix (char *buf_ptr, size_t buf_max);

    48

};

    49

    50

class NetworkTCPTransport {

    51

    protected:

    52

        NetworkSocket socket;

    53

    54

        NetworkBuffer in, out;

    55

    56

        CL_SlotContainer slots; 

    57

    58

    public:

    59

        NetworkTCPTransport (NetworkSocket socket);

    60

    61

    private:

    62

        void on_read (void);

    63

        void on_write (void);

    64

        void on_disconnected (void);

    65

    66

        CL_Signal_v1<NetworkPacket &> _sig_packet;

    67

        CL_Signal_v0 _sig_disconnect;

    68

    69

    public:

    70

        NetworkAddress getLocalAddress (void) { return socket.get_source_address(); }

    71

        NetworkAddress getRemoteAddress (void) { return socket.get_dest_address(); }

    72

    73

        void write_packet (const NetworkPacket &packet);

    74

    75

        CL_Signal_v1<NetworkPacket&>& sig_packet (void) { return _sig_packet; }

    76

        CL_Signal_v0& sig_disconnect (void) { return _sig_disconnect; }

    77

};

    78

    79

class NetworkTCPServer {

    80

    private:

    81

        NetworkSocket socket;

    82

    83

        CL_SlotContainer slots; 

    84

    85

    public:

    86

        NetworkTCPServer (const NetworkAddress &listen_addr);

    87

    88

    private:

    89

        void on_accept (void);

    90

    91

        CL_Signal_v1<NetworkTCPTransport *> _sig_client;

    92

    93

    protected:

    94

        virtual NetworkTCPTransport* buildTransport (CL_Socket &socket);

    95

    96

    public:

    97

        CL_Signal_v1<NetworkTCPTransport *>& sig_client (void) { return _sig_client; }

    98

};

    99

   100

class NetworkTCPClient : public NetworkTCPTransport {

   101

    public:

   102

        NetworkTCPClient (const NetworkAddress &connect_addr);

   103

};

   104

   105

#endif /* NETWORK_TCP_HH */