1

#ifndef NETWORK_TCP_HH

     2

#define NETWORK_TCP_HH

     3

    10

#include <ClanLib/core.h>

    11

    12

#include "Socket.hh"

    13

#include "Address.hh"

    14

#include "Packet.hh"

    15

#include "Buffer.hh"

    16

    17

// @{

    18

/**

    19

 * Initial input/output buffer size given to NetworkTCPTransport::in/out constructor

    20

 *

    21

 * @see NetworkBufferBase

    22

 */

    23

const size_t NETWORK_TCP_INITIAL_IN_BUF = 4096;

    24

const size_t NETWORK_TCP_INITIAL_OUT_BUF = 0;

    25

// @}

    26

    27

/**

    34

 */

    35

class NetworkTCPTransport {

    36

    protected:

    37

        /**

    38

         * The SOCK_STREAM socket

    39

         */

    40

        NetworkSocket *socket;

    41

    42

        /**

    43

         * Our input buffer associated with socket

    44

         */

    45

        NetworkBufferInput in;

    46

    47

        /**

    48

         * Our output buffer associated with socket

    49

         */

    50

        NetworkBufferOutput out;

    51

    52

        CL_SlotContainer slots; 

    53

    54

    public:

    55

        /**

    56

         * Construct this using the given socket

    57

         */

    58

        NetworkTCPTransport (NetworkSocket *socket);

    59

    60

        /**

    61

         * Deletes the socket

    62

         */

    63

        ~NetworkTCPTransport (void);

    64

    65

    private:

    66

        /**

    67

         * Triggered when the socket is read for recv

    68

         */

    69

        void on_read (void);

    70

    71

        /**

    72

         * Triggered when the socket is read for send

    73

         */

    74

        void on_write (void);

    75

    76

        /**

    77

         * The packet-read signal

    78

         */

    79

        CL_Signal_v1<NetworkPacketInput&> _sig_packet;

    80

    81

        /**

    82

         * The disconnected-signal

    83

         */

    84

        CL_Signal_v0 _sig_disconnect;

    85

    86

    public:

    87

        /**

    88

         * Get this TCP socket's local address

    89

         */

    90

        NetworkAddress get_local_address (void) { return socket->get_local_address(); }

    91

    92

        /**

    93

         * Get this TCP socket's remote address

    94

         */

    95

        NetworkAddress get_remote_address (void) { return socket->get_remote_address(); }

    96

    97

        /**

    98

         * Write the given packet to this socket output, buffering the data if need be

    99

         *

   100

         * @param packet the packet to send

   101

         */

   102

        void write_packet (const NetworkPacketBuffer &packet);

   103

   104

        /**

   105

         * Handle disconnect, disable socket

   106

         */

   107

        void handle_disconnect (void);

   108

   109

        /**

   110

         * A full packet has been received from the remote end

   111

         */

   112

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

   113

   114

        /**

   115

         * The connection has been lost (no specific error code given)

   116

         */

   117

        CL_Signal_v0& sig_disconnect (void) { return _sig_disconnect; }

   118

};

   119

   120

/**

   121

 * A TCP server, listening on the given address, creating NetworkTCPTransports for clients that connect

   122

 */

   123

class NetworkTCPServer {

   124

    private:

   125

        /**

   126

         * The listen() socket

   127

         */

   128

        NetworkSocket socket;

   129

   130

        CL_SlotContainer slots; 

   131

   132

    public:

   133

        /**

   136

         */

   137

        NetworkTCPServer (const NetworkEndpoint &listen_addr);

   138

   139

    private:

   140

        /**

   141

         * The socket is read for read()

   142

         */

   143

        void on_accept (void);

   144

   145

        /**

   146

         * The on_accept signal

   147

         */

   148

        CL_Signal_v1<NetworkTCPTransport *> _sig_client;

   149

   150

    protected:

   151

        /**

   152

         * Called by on_accept to build the NetworkTCPTransport, can be used to override what class gets used.

   153

         *

   154

         * By default, this just constructs a new NetworkTCPTransport and returns it

   155

         *

   156

         * @param socket the socket returned by accept()

   157

         */

   158

        virtual NetworkTCPTransport* buildTransport (NetworkSocket *socket);

   159

   160

    public:

   161

        /**

   162

         * A new client has connected, and a NetworkTCPTransport has been connected for it

   163

         */

   164

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

   165

};

   166

   167

/**

   168

 * A TCP Client that connects to the given address

   169

 */ 

   170

class NetworkTCPClient : public NetworkTCPTransport {

   171

    public:

   172

        /**

   177

         *

   178

         * @param connect_addr the address to connect to

   179

         */

   180

        NetworkTCPClient (const NetworkEndpoint &connect_addr);

   181

};

   182

   183

#endif /* NETWORK_TCP_HH */