00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024
00025
00026
00027
00028
00029
00030
00031
00032
00033
00034
00035
00036 #if !defined(AFX_ASYNCSOCKET_H__980B05BD_A250_40D0_AD92_4F31AEC90E74__INCLUDED_)
00037 #define AFX_ASYNCSOCKET_H__980B05BD_A250_40D0_AD92_4F31AEC90E74__INCLUDED_
00038
00039 #if _MSC_VER > 1000
00040 #pragma once
00041 #endif // _MSC_VER > 1000
00042
00043 #pragma warning(disable:4786)
00044
00045 #include "SocketThreadManager.h"
00046
00047 #include "SpoofBase.h"
00048 #include "BlockedBuffer.h"
00049
00050 #include <map>
00051
00052
00053 #define WM_BASE WM_USER
00054 #define WM_SOCKET_GENERAL WM_BASE+1
00055 #define WM_SOCKET_ACCEPT WM_BASE+2
00056 #define WM_SOCKET_CONNECT WM_BASE+3
00057 #define WM_SOCKET_TIMEOUT WM_BASE+4
00058 #define WM_SOCKET_DELETE WM_BASE+5
00059 #define WM_SOCKET_FORCED WM_BASE+6
00060
00061
00062 #define NO_OnSocketTimeout virtual BOOL OnSocketTimeout() {return TRUE;}
00063 #define NO_OnSocketConnect virtual BOOL OnSocketConnect(int iErrorCode) {return TRUE;}
00064 #define NO_OnSocketAccept virtual BOOL OnSocketAccept(int iErrorCode) {return TRUE;}
00065 #define NO_OnSocketClose virtual BOOL OnSocketClose(int iErrorCode) {return TRUE;}
00066 #define NO_OnSocketOOB virtual BOOL OnSocketOOB(int iErrorCode) {return TRUE;}
00067 #define NO_OnSocketWrite virtual BOOL OnSocketWrite(int iErrorCode) {return TRUE;}
00068 #define NO_OnSocketReceive virtual BOOL OnSocketReceive(int iErrorCode) {return TRUE;}
00069
00070
00071 #define CAsyncSocket_Class "CAsyncSocketClass"
00072 #define CAsyncShutdown_Class "CAsyncShutdown"
00073
00074 KOMODIA_NAMESPACE_START
00075
00076 class CGenericCriticalSection;
00077
00078 class CAsyncSocket : virtual protected CSpoofBase
00079 {
00080 friend CBlockedBuffer;
00081 public:
00082
00083 typedef enum _AsyncEvents
00084 {
00085 aeReceive=1,
00086 aeSend=2,
00087 aeOOB=4,
00088 aeClose=8
00089 } AsyncEvents;
00090
00091
00092 typedef BOOL (*MapProc)();
00093 typedef MapProc LPMapProc;
00094 public:
00095
00096 BOOL HasTimeout()const;
00097
00098
00099 void DisableEvents(unsigned char ucEvents);
00100
00101
00102 BOOL IsClosing()const;
00103
00104
00105 void ForceReceiveEvent();
00106
00107
00108 virtual void DeleteSocketFromThread();
00109
00110
00111 virtual void DeleteSocketFromThread(DWORD dwTimeToWait);
00112
00113
00114 static void SimpleMessageMap();
00115
00116
00117 static void SimpleMessageMap(DWORD dwRunTimeMS);
00118
00119
00120
00121
00122
00123
00124 static void SimpleMessageMap(LPMapProc pProc,
00125 BOOL bStopWhenTRUE,
00126 DWORD dwSleep);
00127
00128
00129 static char SimpleMessageMapKey(DWORD dwSleep);
00130
00131
00132
00133 void AllowBlockedBuffer(BOOL bAllow);
00134
00135
00136 virtual BOOL ReAsync();
00137
00138
00139 virtual BOOL ReBlock();
00140
00141
00142
00143 void FreezeThread();
00144
00145
00146 BOOL DisableAsync();
00147
00148
00149
00150 static void Initialize();
00151
00152
00153
00154 static void Shutdown();
00155
00156
00157 BOOL KillTimer();
00158
00159
00160
00161
00162 BOOL SetTimeout(int iMs);
00163
00164
00165
00166 static void SetInstance(HINSTANCE hInst);
00167
00168
00169 CAsyncSocket();
00170 virtual ~CAsyncSocket();
00171 protected:
00172
00173
00174
00175
00176
00177
00178
00179
00180
00181
00182
00183 virtual BOOL OnSocketTimeout()=0;
00184
00185
00186 virtual BOOL OnSocketConnect(int iErrorCode)=0;
00187
00188
00189 virtual BOOL OnSocketAccept(int iErrorCode)=0;
00190
00191
00192 virtual BOOL OnSocketClose(int iErrorCode)=0;
00193
00194
00195
00196 virtual BOOL OnSocketOOB(int iErrorCode)=0;
00197
00198
00199
00200 virtual BOOL OnSocketWrite(int iErrorCode)=0;
00201
00202
00203 virtual BOOL OnSocketReceive(int iErrorCode)=0;
00204
00205
00206 BOOL IsBlocking()const;
00207
00208
00209 int InternalWSAAsyncSelect(unsigned int wMsg,
00210 long lEvent);
00211
00212
00213 void SocketClosing();
00214
00215
00216 int GetSocketID()const;
00217
00218
00219 HWND GetWindowHandle()const;
00220
00221
00222 virtual SOCKET GetAsyncHandle()const=0;
00223
00224
00225 virtual BOOL SetAsync()=0;
00226
00227
00228 void RemoveSocketFromList();
00229
00230
00231 void AddSocketToList();
00232
00233
00234 BOOL IsTimeout()const;
00235
00236
00237 BOOL Block();
00238
00239
00240 virtual BOOL SendBlockedBuffer(const CBlockedBuffer::CBlockedData& rData);
00241
00242
00243 CBlockedBuffer* GetBlockedBuffer()const;
00244
00245
00246 static CSocketThreadManager* GetSocketThreadManager();
00247
00248
00249
00250 virtual BOOL SocketConnected(int iErrorCode);
00251
00252
00253
00254 virtual BOOL SocketClosed(int iErrorCode);
00255
00256
00257
00258 virtual BOOL SocketWrite(int iErrorCode);
00259
00260
00261
00262 virtual void SystemTimeout();
00263
00264
00265 BOOL SetSystemTimeout(int iMS);
00266
00267
00268 BOOL KillSystemTimer();
00269
00270
00271 BOOL HasSystemTimer()const;
00272
00273
00274
00275 virtual void OnSocketDelete();
00276
00277
00278
00279 virtual void SocketCreated();
00280 private:
00281
00282 typedef std::map<int,CAsyncSocket*> SocketMap;
00283 private:
00284
00285 CAsyncSocket(const CAsyncSocket& rSocket);
00286
00287
00288 CAsyncSocket& operator=(const CAsyncSocket& rSocket);
00289
00290
00291 BOOL CheckAsyncSocketValid()const;
00292
00293
00294 void DeAllocateHandle();
00295
00296
00297 void AllocateHandle();
00298
00299
00300 CSocketThreadManager* GetThreadManager()const;
00301
00302
00303 static BOOL RemoveHandlers();
00304
00305
00306 static HINSTANCE GetInstance();
00307
00308
00309 static BOOL SetHandlers();
00310
00311
00312 static BOOL RegisterWindow();
00313
00314
00315 static CAsyncSocket* GetSocketByID(int iSockID);
00316
00317
00318 static void SystemTimerProc(LPVOID lpData);
00319
00320
00321 static void DeleteTimerProc(LPVOID lpData);
00322
00323
00324 static SocketMap m_aSocketMap;
00325
00326
00327 static BOOL m_bWindow;
00328
00329
00330 static HINSTANCE m_hInstance;
00331
00332
00333 static BOOL m_bInitialized;
00334
00335
00336 int m_iSocketID;
00337
00338
00339 BOOL m_bList;
00340
00341
00342 BOOL m_bTimeout;
00343
00344
00345 HWND m_hLocalWindowHandle;
00346
00347
00348 static BOOL m_bShuttingDown;
00349
00350
00351 BOOL m_bFreeze;
00352
00353
00354 BOOL m_bBlocking;
00355
00356
00357 unsigned int m_iMsg;
00358
00359
00360 long m_lEvent;
00361
00362
00363 static CSocketThreadManager* m_pThreadManager;
00364
00365
00366 CSocketThreadManager* m_pLocalThreadManager;
00367
00368
00369 static CGenericCriticalSection* m_pCSection;
00370
00371
00372 CBlockedBuffer* m_pBlockedBuffer;
00373
00374
00375 CSocketThreadManager::TimerID m_aTimerID;
00376
00377
00378 CSocketThreadManager::TimerID m_aDeleteTimerID;
00379
00380
00381 BOOL m_bClosing;
00382
00383
00384 unsigned char m_ucEvents;
00385 private:
00386
00387 class CAsyncShutdown : protected CSpoofBase
00388 {
00389 public:
00390 CAsyncShutdown();
00391 virtual ~CAsyncShutdown();
00392 protected:
00393
00394 virtual void NotifyShutdown();
00395 };
00396 private:
00397
00398 static LRESULT CALLBACK SocketMessageHandler(HWND hwnd,
00399 UINT uMsg,
00400 WPARAM wParam,
00401 LPARAM lParam);
00402 };
00403
00404 KOMODIA_NAMESPACE_END
00405
00406 #endif // !defined(AFX_ASYNCSOCKET_H__980B05BD_A250_40D0_AD92_4F31AEC90E74__INCLUDED_)