aboutsummaryrefslogtreecommitdiff
path: root/src/itoserver.h
blob: f11960c8d6d83859b5da71f5b10dba351fb5b224 (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
/*
 * Copyright (C) 2007 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>

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

namespace Bu
{
	class ServerSocket;
	class Socket;
	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.
	 *@author Mike Buland
	 *@ingroup Threading Serving
	 */
	class ItoServer : public Ito
	{
		friend class ItoClient;
	public:
		ItoServer();
		virtual ~ItoServer();

		void addPort( int nPort, int nPoolSize=40 );
		void addPort( const FString &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;

		virtual void *run();

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

			virtual void *run();

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

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

		void clientCleanup( int iSocket );
	};
}

#endif