2 * copyright (c) 2010 Sveriges Television AB <info@casparcg.com>
\r
4 * This file is part of CasparCG.
\r
6 * CasparCG is free software: you can redistribute it and/or modify
\r
7 * it under the terms of the GNU General Public License as published by
\r
8 * the Free Software Foundation, either version 3 of the License, or
\r
9 * (at your option) any later version.
\r
11 * CasparCG is distributed in the hope that it will be useful,
\r
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
\r
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
\r
14 * GNU General Public License for more details.
\r
16 * You should have received a copy of the GNU General Public License
\r
17 * along with CasparCG. If not, see <http://www.gnu.org/licenses/>.
\r
21 // AsyncEventServer.h: interface for the AsyncServer class.
\r
22 //////////////////////////////////////////////////////////////////////
\r
24 #if !defined(AFX_ASYNCEVENTSERVER_H__0BFA29CB_BE4C_46A0_9CAE_E233ED27A8EC__INCLUDED_)
\r
25 #define AFX_ASYNCEVENTSERVER_H__0BFA29CB_BE4C_46A0_9CAE_E233ED27A8EC__INCLUDED_
\r
31 #include <common/memory/safe_ptr.h>
\r
36 #include <functional>
\r
38 #include "..\util\thread.h"
\r
40 #include "ProtocolStrategy.h"
\r
42 #include <tbb\mutex.h>
\r
48 typedef std::shared_ptr<SocketInfo> SocketInfoPtr;
\r
50 typedef std::function<void(caspar::IO::SocketInfoPtr)> ClientDisconnectEvent;
\r
52 class AsyncEventServer : public IRunnable
\r
54 static long instanceCount_;
\r
57 AsyncEventServer(const AsyncEventServer&);
\r
58 AsyncEventServer& operator=(const AsyncEventServer&);
\r
61 explicit AsyncEventServer(const safe_ptr<IProtocolStrategy>& pProtocol, int port);
\r
62 ~AsyncEventServer();
\r
65 void SetProtocolStrategy(safe_ptr<IProtocolStrategy> pPS) {
\r
66 pProtocolStrategy_ = pPS;
\r
71 void SetClientDisconnectHandler(ClientDisconnectEvent handler);
\r
74 Thread listenThread_;
\r
75 void Run(HANDLE stopEvent);
\r
76 bool OnUnhandledException(const std::exception&) throw();
\r
78 bool OnAccept(SocketInfoPtr&);
\r
79 bool OnRead(SocketInfoPtr&);
\r
80 void OnWrite(SocketInfoPtr&);
\r
81 void OnClose(SocketInfoPtr&);
\r
82 void OnError(HANDLE waitEvent, int errorCode);
\r
84 SocketInfoPtr pListenSocketInfo_;
\r
85 safe_ptr<IProtocolStrategy> pProtocolStrategy_;
\r
88 friend class SocketInfo;
\r
89 void DoSend(SocketInfo&);
\r
90 void DisconnectClient(SocketInfo&);
\r
92 void LogSocketError(const TCHAR* pStr, int socketError = 0);
\r
94 class SocketInfoCollection
\r
96 SocketInfoCollection(const SocketInfoCollection&);
\r
97 SocketInfoCollection& operator=(const SocketInfoCollection&);
\r
99 typedef std::map<HANDLE, SocketInfoPtr> SocketInfoMap;
\r
100 typedef std::vector<HANDLE> HandleVector;
\r
103 SocketInfoCollection();
\r
104 ~SocketInfoCollection();
\r
106 bool AddSocketInfo(SocketInfoPtr& pSocketInfo);
\r
107 void RemoveSocketInfo(SocketInfoPtr& pSocketInfo);
\r
108 void RemoveSocketInfo(HANDLE);
\r
109 void CopyCollectionToArray(HANDLE*, int maxCount);
\r
111 bool FindSocketInfo(HANDLE, SocketInfoPtr& pResult);
\r
116 void ClearDirty() {
\r
120 std::size_t Size() {
\r
121 return waitEvents_.size();
\r
125 ClientDisconnectEvent onSocketInfoRemoved;
\r
128 SocketInfoMap socketInfoMap_;
\r
129 HandleVector waitEvents_;
\r
133 SocketInfoCollection socketInfoCollection_;
\r
136 typedef std::tr1::shared_ptr<AsyncEventServer> AsyncEventServerPtr;
\r
139 } //namespace caspar
\r
141 #endif // !defined(AFX_ASYNCEVENTSERVER_H__0BFA29CB_BE4C_46A0_9CAE_E233ED27A8EC__INCLUDED_)
\r