1

#ifndef NETWORK_TCP_HH

     2

#define NETWORK_TCP_HH

     3

     4

#include <ClanLib/core.h>

     5

     6

#include "Socket.hh"

     7

#include "Address.hh"

     8

#include "Packet.hh"

     9

#include "Buffer.hh"

    10

    11

// @{

    12

/**

    13

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

    14

 *

    15

 * @see NetworkBufferBase

    16

 */

    17

const size_t NETWORK_TCP_INITIAL_IN_BUF = 4096;

    18

const size_t NETWORK_TCP_INITIAL_OUT_BUF = 0;

    19

// @}

    20

    21

/**

    22

 * A connected TCP socket, so either a client or a server-accept socket

    23

 */

    24

class NetworkTCPTransport {

    25

    protected:

    26

        /**

    27

         * The SOCK_STREAM socket

    28

         */

    29

        NetworkSocket *socket;

    30

    31

        /**

    32

         * Our input buffer associated with socket

    33

         */

    34

        NetworkBufferInput in;

    35

    36

        /**

    37

         * Our output buffer associated with socket

    38

         */

    39

        NetworkBufferOutput out;

    40

    41

        CL_SlotContainer slots; 

    42

    43

    public:

    44

        /**

    45

         * Construct this using the given socket

    46

         */

    47

        NetworkTCPTransport (NetworkSocket *socket);

    48

    49

        /**

    50

         * Deletes the socket

    51

         */

    52

        ~NetworkTCPTransport (void);

    53

    54

    private:

    55

        /**

    56

         * Triggered when the socket is read for recv

    57

         */

    58

        void on_read (void);

    59

    60

        /**

    61

         * Triggered when the socket is read for send

    62

         */

    63

        void on_write (void);

    64

    65

        /**

    66

         * The packet-read signal

    67

         */

    68

        CL_Signal_v1<NetworkPacketInput&> _sig_packet;

    69

    70

        /**

    71

         * The disconnected-signal

    72

         */

    73

        CL_Signal_v0 _sig_disconnect;

    74

    75

    public:

    76

        /**

    77

         * Get this TCP socket's local address

    78

         */

    79

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

    80

    81

        /**

    82

         * Get this TCP socket's remote address

    83

         */

    84

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

    85

    86

        /**

    87

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

    88

         *

    89

         * @param packet the packet to send

    90

         */

    91

        void write_packet (const NetworkPacketBuffer &packet);

    92

    93

        /**

    94

         * Handle disconnect, disable socket

    95

         */

    96

        void handle_disconnect (void);

    97

    98

        /**

    99

         * A full packet has been received from the remote end

   100

         */

   101

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

   102

   103

        /**

   104

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

   105

         */

   106

        CL_Signal_v0& sig_disconnect (void) { return _sig_disconnect; }

   107

};

   108

   109

/**

   110

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

   111

 */

   112

class NetworkTCPServer {

   113

    private:

   114

        /**

   115

         * The listen() socket

   116

         */

   117

        NetworkSocket socket;

   118

   119

        CL_SlotContainer slots; 

   120

   121

    public:

   122

        /**

   123

         * Listen on the specific address.

   124

         */

   126

   127

    private:

   128

        /**

   129

         * The socket is read for read()

   130

         */

   131

        void on_accept (void);

   132

   133

        /**

   134

         * The on_accept signal

   135

         */

   136

        CL_Signal_v1<NetworkTCPTransport *> _sig_client;

   137

   138

    protected:

   139

        /**

   140

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

   141

         *

   142

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

   143

         *

   144

         * @param socket the socket returned by accept()

   145

         */

   146

        virtual NetworkTCPTransport* buildTransport (NetworkSocket *socket);

   147

   148

    public:

   149

        /**

   150

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

   151

         */

   152

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

   153

};

   154

   155

/**

   156

 * A TCP Client that connects to the given address

   157

 */ 

   158

class NetworkTCPClient : public NetworkTCPTransport {

   159

    public:

   160

        /**

   161

         * Create a NetworkTCPTransport, and then connect our socket to the given address.

   162

         *

   163

         * @param connect_addr the address to connect to

   164

         */

   166

};

   167

   168

#endif /* NETWORK_TCP_HH */