1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
|
/*
* Copyright (C) 2007-2019 Xagasoft, All rights reserved.
*
* This file is part of the libbu++ library and is released under the
* terms of the license contained in the file LICENSE.
*/
#ifndef BU_SERVER_H
#define BU_SERVER_H
#include <stdint.h>
#ifndef WIN32
#include <sys/select.h>
#endif
#include "bu/string.h"
#include "bu/list.h"
#include "bu/clientlink.h"
#include "bu/clientlinkfactory.h"
#include "bu/hash.h"
#include "bu/config.h"
#ifndef PROFILE_BU_SERVER
// #define PROFILE_BU_SERVER 1
#endif
#ifdef PROFILE_BU_SERVER
#include "bu/profiler.h"
#endif
namespace Bu
{
class TcpServerSocket;
class TcpSocket;
class Client;
/**
* Core of a network server. This class is distinct from a ServerSocket in
* that a ServerSocket is one listening socket, nothing more. Socket will
* manage a pool of both ServerSockets and connected Sockets along with
* their protocols and buffers.
*
* To start serving on a new port, use the addPort functions. Each call to
* addPort creates a new ServerSocket, starts it listening, and adds it to
* the server pool.
*
* All of the real work is done by scan, which will wait for up
* to the timeout set by setTimeout before returning if there is no data
* pending. scan should probably be called in some sort of tight
* loop, possibly in it's own thread, or in the main control loop.
*
* In order to use a Server you must subclass it and implement the pure
* virtual functions. These allow you to receive notification of events
* happening within the server itself, and actually makes it useful.
*@ingroup Serving
*/
class Server
{
public:
Server();
virtual ~Server();
#ifdef WIN32
typedef unsigned int socket_t;
#else
typedef int socket_t;
#endif
void addPort( int nPort, int nPoolSize=40 );
void addPort( const String &sAddr, int nPort, int nPoolSize=40 );
virtual void scan();
void setTimeout( int nTimeoutSec, int nTimeoutUSec=0 );
void addClient( socket_t nSocket, int nPort );
void setAutoTick( bool bEnable=true );
void tick();
virtual void onNewConnection( Client *pClient, int nPort )=0;
virtual void onClosedConnection( Client *pClient )=0;
void shutdown();
private:
void closeClient( socket_t iSocket );
class SrvClientLink : public Bu::ClientLink
{
public:
SrvClientLink( Bu::Client *pClient );
virtual ~SrvClientLink();
virtual void sendMessage( const Bu::String &sMsg );
private:
Bu::Client *pClient;
};
class SrvClientLinkFactory : public Bu::ClientLinkFactory
{
public:
SrvClientLinkFactory();
virtual ~SrvClientLinkFactory();
virtual Bu::ClientLink *createLink( Bu::Client *pClient );
};
int nTimeoutSec;
int nTimeoutUSec;
fd_set fdActive;
typedef Hash<socket_t,TcpServerSocket *> SrvHash;
SrvHash hServers;
typedef Hash<socket_t,Client *> ClientHash;
ClientHash hClients;
bool bAutoTick;
};
}
#endif
|
