1 files changed, 141 insertions, 0 deletions

diff --git a/src/unstable/itoserver.h b/src/unstable/itoserver.h
new file mode 100644
index 0000000..81db4e2
--- /dev/null
+++ b/src/unstable/itoserver.h
@@ -0,0 +1,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/thread.h"
+#include "bu/mutex.h"
+#include "bu/synchroqueue.h"
+#include "bu/hash.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 Thread
+        {
+                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 Thread
+                {
+                friend class Bu::ItoServer::SrvClientLink;
+                public:
+                        ItoClient( ItoServer &rSrv, int nSocket, int nPort,
+                                        int nTimeoutSec, int nTimeoutUSec );
+                        virtual ~ItoClient();
+
+                        typedef SynchroQueue<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;
+                        Mutex 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 SynchroQueue<ItoClient *> ClientQueue;
+                ClientHash hClients;
+                ClientQueue qClientCleanup;
+                Mutex imClients;
+
+                void clientCleanup( int iSocket );
+        };
+}
+
+#endif