aboutsummaryrefslogtreecommitdiff
path: root/src/itoserver.h
blob: 902c68468f6f84598091305f3abc83b634aba011 (plain)
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
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
/*
 * Copyright (C) 2007-2011 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_ITO_SERVER_H
#define BU_ITO_SERVER_H

#include <stdint.h>

#ifndef WIN32
	#include <sys/select.h>
#endif

#include "bu/string.h"
#include "bu/list.h"
#include "bu/ito.h"
#include "bu/itomutex.h"
#include "bu/itoqueue.h"
#include "bu/set.h"

#include "bu/clientlink.h"
#include "bu/clientlinkfactory.h"

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 Threading Serving
	 */
	class ItoServer : public Ito
	{
		friend class ItoClient;
		friend class SrvClientLinkFactory;
	public:
		ItoServer();
		virtual ~ItoServer();

		void addPort( int nPort, int nPoolSize=40 );
		void addPort( const String &sAddr, int nPort, int nPoolSize=40 );

		//void scan();
		void setTimeout( int nTimeoutSec, int nTimeoutUSec=0 );

		void addClient( int nSocket, int nPort );

		virtual void onNewConnection( Client *pClient, int nPort )=0;
		virtual void onClosedConnection( Client *pClient )=0;

	protected:
		virtual void run();

	private:
		class SrvClientLink;
		class ItoClient : public Ito
		{
		friend class Bu::ItoServer::SrvClientLink;
		public:
			ItoClient( ItoServer &rSrv, int nSocket, int nPort,
					int nTimeoutSec, int nTimeoutUSec );
			virtual ~ItoClient();

			typedef ItoQueue<Bu::String *> StringQueue;
			StringQueue qMsg;

		protected:
			virtual void run();

		private:
			ItoServer &rSrv;
			Client *pClient;
			fd_set fdActive;
			int iSocket;
			int iPort;
			int nTimeoutSec;
			int nTimeoutUSec;
			ItoMutex imProto;
		};

		class SrvClientLink : public Bu::ClientLink
		{
		public:
			SrvClientLink( ItoClient *pClient );
			virtual ~SrvClientLink();

			virtual void sendMessage( const Bu::String &sMsg );

		private:
			ItoClient *pClient;
		};

		class SrvClientLinkFactory : public Bu::ClientLinkFactory
		{
		public:
			SrvClientLinkFactory( ItoServer &rSrv );
			virtual ~SrvClientLinkFactory();

			virtual Bu::ClientLink *createLink( Bu::Client *pClient );

		private:
			ItoServer &rSrv;
		};

		int nTimeoutSec;
		int nTimeoutUSec;
		fd_set fdActive;
		typedef Hash<int,TcpServerSocket *> ServerHash;
		ServerHash hServers;
		typedef Hash<int,ItoClient *> ClientHash;
		typedef ItoQueue<ItoClient *> ClientQueue;
		ClientHash hClients;
		ClientQueue qClientCleanup;
		ItoMutex imClients;

		void clientCleanup( int iSocket );
	};
}

#endif