Сообщение было отмечено slowCheetah как решение
Решение
сразу после публикации поста реализовал то, что хотел, но есть небольшая проблема - бывает так, сокет серера получив сообщение FD_ACCEPT и вызвав OnAccept после этого все сообщения принимает с кодом 0 - что бы ни приходило, всегда 0 - а почему поянть никак не могу
кто может помогите, вот пример реализации этих классов:
Класс сокетов:
| C++ | 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
| /**
* @brief Класс исключений для сокетов.
*/
class CSocketException : public Exceptions::CException
{
public:
CSocketException(const string& strError, const string& aFile, const size_t uLine) throw()
: Exceptions::CException("CSocketException: " + strError, aFile, uLine) {
}
};
/**
* @brief Обертка над сокетами Winsock2.
*/
class CSocket
{
public:
/** @brief Типы соединений сокета */
enum { TYPE_TCP, TYPE_UDP };
CSocket() throw() : m_bIsConnected(false), m_handle(NULL), m_uPort(0), m_bStop(false), m_iType(TYPE_TCP) { }
CSocket(const string& aIpAdress, const unsigned short uPort, int iType = TYPE_TCP) throw(CSocketException);
CSocket(SOCKET handle, const string& aIpAdress, int iType = TYPE_TCP) throw();
virtual ~CSocket() throw();
void Create(int iConnectionType = TYPE_TCP) throw(CSocketException);
void Close() throw(CSocketException);
void Connect(const string& aAddress, const unsigned short uPort) throw(CSocketException);
void Bind() throw(CSocketException);
void Listen(int iQueueLength = SOMAXCONN) throw(CSocketException);
bool Accept(CSocket ** pNewSocket) throw();
bool Accept(SOCKET& sock, string& aIpAddress) throw();
bool Shutdown(int nFlag) throw();
int Read(PBYTE pBuffer, int nLen, sockaddr_in * clientAddress = NULL, int * addrSize = NULL) throw();
int Recv(PBYTE pBuffer, int nLen) throw();
int Recvfrom(PBYTE pBuffer, int nLen, sockaddr_in * clientAddress, int * addrSize) throw();
int Write(PBYTE pBuffer, int nLen, sockaddr_in * destAddress = NULL) throw();
int Send(PBYTE pBuffer, int nLen) throw();
int Sendto(PBYTE pBuffer, int nLen, sockaddr_in * destAddress) throw();
/** @brief Переходим к блокируемым и неблокируемым сокетам */
void BecomeBlocking() throw(CSocketException);
void BecomeNonBlocking() throw(CSocketException);
/** @brief Возвращаем статус соединения */
bool IsConnected() const throw() {
return m_bIsConnected;
}
/** @brief Возвращаем тип соединения */
int GetType() const throw() {
return m_iType;
}
/** @brief Возвращаем описатель */
SOCKET GetSocket() const throw() {
return m_handle;
}
/** @brief Возвращаем номер порта */
unsigned short GetPort() const throw() {
return m_uPort;
}
/** @brief Возвращаем IP адрес */
const string& GetAdress() const throw() {
return m_strAdress;
}
// Возвращаем описатель сокета
SOCKET GetHandle() const throw() {
return m_handle;
}
// Запуск
bool Start() throw();
// на выполнение
private:
// нити событий класса сокетов
static unsigned __stdcall Proc(LPVOID lpParam) throw(CSocketException);
protected:
/** @brief Сообщения класса */
virtual void OnConnect();
virtual void OnAccept();
virtual void OnSend();
virtual void OnReceive();
virtual void OnClose();
private:
SOCKET m_handle;
int m_iType;
unsigned short m_uPort;
string m_strAdress;
bool m_bIsConnected;
bool m_bStop;
Wrappers::CHandle m_hThread;
private:
CSocket(const CSocket&);
CSocket& operator=(const CSocket&);
}; |
|
Класс сокетов - реализация:
| C++ | 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
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
| CSocket::CSocket(const string &aIpAdress, const unsigned short uPort, int iType/* = TYPE_TCP*/)
throw(CSocketException)
: m_handle(NULL), m_strAdress(aIpAdress), m_uPort(uPort), m_iType(iType), m_bIsConnected(false), m_bStop(false)
{
Create(m_iType);
}
CSocket::CSocket(SOCKET handle, const string& aIpAdress, int iType/* = TYPE_TCP */) throw()
: m_handle(handle), m_strAdress(aIpAdress), m_iType(iType), m_bIsConnected(true), m_bStop(false)
{
}
CSocket::~CSocket() throw() {
m_bStop = true;
if (m_hThread.IsValid())
WaitForSingleObject(m_hThread.Get(), INFINITE);
Close();
m_hThread.Attach(NULL);
}
/**
* @brief Инициализируем библиотеку, инициализируем сокет.
* @param iConnectionType Тип соединения.
*/
void CSocket::Create(int iConnectionType/* = TYPE_TCP*/) throw(CSocketException)
{
WSAData wsaData;
WSAStartup(0x202, &wsaData);
m_iType = iConnectionType;
m_handle = socket(AF_INET, m_iType ? SOCK_DGRAM : SOCK_STREAM, 0);
if (m_handle == INVALID_SOCKET) {
Close();
throw CSocketException(IDS_SOCKET_FAIL, __FILE__, __LINE__);
}
}
/**
* @brief Закрываем соединение.
* @return void.
*/
void CSocket::Close() throw() {
if (m_handle) {
closesocket(m_handle);
m_handle = NULL;
m_bIsConnected = false;
}
}
/**
* @brief Соединение с удаленным сокетом.
* @param aAddress Адрес удаленного сервера.
* @param uPort Порт, к которому подсоединяемся.
*/
void CSocket::Connect(const string& aAddress, const unsigned short uPort) throw(CSocketException) {
sockaddr_in clientService;
clientService.sin_family = AF_INET;
clientService.sin_port = htons(uPort);
clientService.sin_addr.s_addr = inet_addr(aAddress.c_str());
if (connect(m_handle, reinterpret_cast<SOCKADDR*>(&clientService), sizeof clientService) == SOCKET_ERROR) {
Close();
throw CSocketException(IDS_SOCKET_CONNECT_ERROR, __FILE__, __LINE__);
}
m_bIsConnected = true;
}
/**
* @brief Привязка сокета к локальному адресу.
* @return void.
*/
void CSocket::Bind() throw(CSocketException) {
sockaddr_in localAdress;
localAdress.sin_family = AF_INET;
localAdress.sin_port = htons(m_uPort);
localAdress.sin_addr.s_addr = inet_addr(m_strAdress.c_str());
if (bind(m_handle, reinterpret_cast<SOCKADDR*>(&localAdress), sizeof localAdress)) {
Close();
throw CSocketException(IDS_SOCKET_BIND_ERROR, __FILE__, __LINE__);
}
}
/**
* @brief Получение входящих запросов.
* @return Результат операции.
*/
bool CSocket::Accept(CSocket ** pNewSocket) throw() {
sockaddr_in address;
int iAddrLen = sizeof address;
SOCKET requestedSocketHandle = INVALID_SOCKET;
requestedSocketHandle = accept(m_handle, reinterpret_cast<SOCKADDR*>(&address), &iAddrLen);
if (requestedSocketHandle == SOCKET_ERROR) {
Close();
return false;
}
*pNewSocket = new CSocket(requestedSocketHandle,
static_cast<const string>(::inet_ntoa(address.sin_addr)));
return true;
}
/**
* @brief Получение входящих запросов.
* @param sock Входящий клиентский сокет.
* @param sock Адрес клиента.
* @return Результат операции.
*/
bool CSocket::Accept(SOCKET& sock, string& aIpAddress) throw() {
sockaddr_in address;
int iAddrLen = sizeof address;
sock = accept(m_handle, reinterpret_cast<SOCKADDR*>(&address), &iAddrLen);
if (sock == SOCKET_ERROR) {
Close();
return false;
}
aIpAddress = static_cast<const string>(::inet_ntoa(address.sin_addr));
return true;
}
/**
* @brief Ожидание входящих подключений.
* @brief iQueueLength Максимальный размер очереди подключений.
*/
void CSocket::Listen(int iQueueLength /*= SOMAXCONN*/) throw(CSocketException) {
if (listen(m_handle, iQueueLength) == SOCKET_ERROR) {
Close();
throw CSocketException(IDS_SOCKET_LISTEN_ERROR, __FILE__, __LINE__);
}
}
/**
* @brief Переходим к блокируемым сокетам.
* @return
*/
void CSocket::BecomeBlocking() throw(CSocketException)
{
unsigned long Blocking = 0;
if (ioctlsocket(m_handle, FIONBIO, &Blocking) == SOCKET_ERROR)
{
Close();
throw CSocketException(IDS_SOCKET_LISTEN_ERROR, __FILE__, __LINE__);
}
}
/**
* @brief Переходим к неблокируемым сокетам.
* @return
*/
void CSocket::BecomeNonBlocking() throw(CSocketException)
{
unsigned long NonBlock = 1;
if (ioctlsocket(m_handle, FIONBIO, &NonBlock) == SOCKET_ERROR)
{
Close();
throw CSocketException(IDS_SOCKET_LISTEN_ERROR, __FILE__, __LINE__);
}
}
/**
* @brief Отмена приема или передачи данных.
* @returns Результат операции.
*/
bool CSocket::Shutdown(int nFlag) throw() {
return shutdown(m_handle, nFlag) == 0;
}
/**
* @brief Чтение данных из потока.
* @param pData Буффер данных.
* @param nLen Размер буффера в байтах.
* @param clientAddress Адрес отправителя.
* @param addrSize Размер адреса в байтах.
* @return Количество принятых байт.
*/
int CSocket::Read(PBYTE pBuffer, int nLen, sockaddr_in * clientAddress/* = NULL*/, int * addrSize/* = NULL*/)
{
if (m_iType == TYPE_TCP) {
return Recv(pBuffer, nLen);
}
else if (m_iType == TYPE_UDP) {
return Recvfrom(pBuffer, nLen, clientAddress, addrSize);
}
return SOCKET_ERROR;
}
/**
* @brief Передаем данные по протоколу TCP.
* @param pData Буффер данных.
* @param nLen Размер буффера в байтах.
* @return Количество принятых байт.
*/
int CSocket::Recv(PBYTE pBuffer, int nLen) throw() {
return recv(m_handle, reinterpret_cast<char *>(pBuffer), nLen, 0);
}
/**
* @brief Передаем данные по протоколу UDP.
* @param pData Буффер данных.
* @param nLen Размер буффера в байтах.
* @param clientAddress Адрес отправителя.
* @param addrSize Размер адреса в байтах.
* @return Количество принятых байт.
*/
int CSocket::Recvfrom(PBYTE pBuffer, int nLen, sockaddr_in * clientAddress, int * addrSize) throw() {
return recvfrom(m_handle, reinterpret_cast<char *>(pBuffer), nLen, 0,
reinterpret_cast<sockaddr*>(clientAddress), addrSize);
}
/**
* @brief Запись данных в поток.
* @param pBuffer Буффер с данными.
* @param nLen Размер буффера в байтах.
* @param destAddress Адрес получателя.
* @return Количество переданных байт.
*/
int CSocket::Write(PBYTE pBuffer, int nLen, sockaddr_in * destAddress/* = NULL*/)
{
if (m_iType == TYPE_TCP) {
return Send(pBuffer, nLen);
}
else if (m_iType == TYPE_UDP) {
return Sendto(pBuffer, nLen, destAddress);
}
return SOCKET_ERROR;
}
/**
* @brief Принимаем данные по протоколу UDP.
* @param pBuffer Буффер с данными.
* @param nLen Размер буффера в байтах.
* @param destAddress Адрес получателя.
* @return Количество переданных байт.
*/
int CSocket::Sendto(PBYTE pBuffer, int nLen, sockaddr_in * destAddress) throw() {
return sendto(m_handle, reinterpret_cast<char *>(pBuffer), nLen, 0,
reinterpret_cast<sockaddr*>(destAddress), sizeof sockaddr_in);
}
/**
* @brief Принимаем данные по протоколу TCP.
* @param pBuffer Буффер с данными.
* @param nLen Размер буффера в байтах.
* @return Количество переданных байт.
*/
int CSocket::Send(PBYTE pBuffer, int nLen) throw() {
return send(m_handle, reinterpret_cast<char *>(pBuffer), nLen, 0);
}
/**
* @brief Основная нить сообщений класса.
* @param lpParam Параметры процедуры.
*/
unsigned __stdcall CSocket::Proc(LPVOID lpParam) throw(CSocketException)
{
if (lpParam == NULL)
throw CSocketException(IDS_SOCKET_EVENT_LOOP_FAILED, __FILE__, __LINE__);
CSocket * pThisObj = reinterpret_cast<CSocket *>(lpParam);
WSANETWORKEVENTS NetworkEvents;
ZeroMemory(&NetworkEvents, sizeof WSANETWORKEVENTS);
Wrappers::CHandle hNetworkEvent( WSACreateEvent() );
if (!hNetworkEvent.IsValid())
throw CSocketException(IDS_SOCKET_EVENT_LOOP_FAILED, __FILE__, __LINE__);
// Подписка на сообщения
if (WSAEventSelect(pThisObj->m_handle, hNetworkEvent.Get(),
FD_CONNECT | FD_READ | FD_WRITE | FD_ACCEPT | FD_CONNECT | FD_CLOSE) == SOCKET_ERROR)
throw CSocketException(IDS_SOCKET_EVENT_LOOP_FAILED, __FILE__, __LINE__);
while (!pThisObj->m_bStop)
{
HANDLE hNetevent = hNetworkEvent.Get();
if (WSAWaitForMultipleEvents(1, &hNetevent, FALSE, WSA_INFINITE, FALSE) == WSA_WAIT_FAILED)
throw CSocketException(IDS_SOCKET_EVENT_LOOP_FAILED, __FILE__, __LINE__);
// Определяем какое событие произошло для заданного сокета
if (WSAEnumNetworkEvents(pThisObj->m_handle, hNetworkEvent.Get(), &NetworkEvents) == SOCKET_ERROR)
throw CSocketException(IDS_SOCKET_EVENT_LOOP_FAILED, __FILE__, __LINE__);
//
// Выбираем события
//
if (NetworkEvents.lNetworkEvents & FD_READ)
{
pThisObj->OnReceive();
}
else if (NetworkEvents.lNetworkEvents & FD_CONNECT)
{
pThisObj->OnConnect();
}
else if (NetworkEvents.lNetworkEvents & FD_ACCEPT)
{
pThisObj->OnAccept();
}
else if (NetworkEvents.lNetworkEvents & FD_WRITE)
{
pThisObj->OnSend();
}
else if (NetworkEvents.lNetworkEvents & FD_CLOSE)
{
pThisObj->OnClose();
}
}
return 0;
}
/**
* @brief Запуск нити сообщений класса.
* @return Результат операции.
*/
bool CSocket::Start() throw() {
unsigned uThreadID = 0;
m_hThread.Attach( (HANDLE)_beginthreadex(0, 0, Proc, this, 0, &uThreadID) );
return m_hThread.IsValid() == TRUE;
}
/**
* @brief Сообщение подключения.
* @return void
*/
void CSocket::OnConnect()
{
}
/**
* @brief Сообщение подключения клиента к сокету.
* @return void
*/
void CSocket::OnAccept()
{
}
/**
* @brief Сообщения отправки данных в поток.
* @return void
*/
void CSocket::OnSend()
{
}
/**
* @brief Сообщение получения данных из потока.
* @return void
*/
void CSocket::OnReceive()
{
}
/**
* @brief Сообщения закрытия сокета.
* @return void
*/
void CSocket::OnClose()
{
}
} |
|
Класс сокетов сервера:
| C++ | 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
| /**
* @brief Интерфейс объектов получения данных.
*/
interface INetSender
{
virtual void OnAccept() = 0;
virtual void OnReceive() = 0;
virtual void OnSend() = 0;
virtual void OnClose() = 0;
};
/**
* @brief Базовый класс сокетов сервера.
*/
class CServerSocket : public CSocket
{
INetSender * m_pSender;
CClientSocket * m_lastConnected;
typedef std::vector<CClientSocket *> SocketsVec;
SocketsVec m_vecClientSockets;
public:
CServerSocket() throw() : CSocket(), m_pSender(NULL), m_lastConnected(NULL) {
// но только не забыть потом запустить нить событий вручную
}
CServerSocket(const string& aIpAdress, const unsigned short uPort, int iType = TYPE_TCP) throw(CSocketException)
: CSocket(aIpAdress, uPort, iType)
{
Bind();
Listen();
Start();
m_lastConnected = NULL;
}
~CServerSocket() throw() {
for (int i = 0; i != m_vecClientSockets.size(); ++i)
delete m_vecClientSockets[i];
m_vecClientSockets.clear();
}
// Событие получения входящих запросов
virtual void OnAccept()
{
string aClientAddress;
SOCKET clientSock = INVALID_SOCKET;
if (CSocket::Accept(clientSock, aClientAddress)) {
CClientSocket * pClient = new CClientSocket(clientSock, aClientAddress, GetType());
m_vecClientSockets.push_back(pClient);
m_lastConnected = pClient;
}
m_pSender->OnAccept();
}
// Событие приема данных
virtual void OnReceive()
{
m_pSender->OnReceive();
}
// Событие отправки данных
virtual void OnSend()
{
m_pSender->OnSend();
}
// Событие закрытия соединения
virtual void OnClose()
{
m_pSender->OnClose();
}
// Установить объект-получатель данных
void SetReceiverObj(INetSender * pSender) throw(CSocketException)
{
if (pSender == NULL)
throw CSocketException(IDS_SOCKET_INVALID_BUFFER, __FILE__, __LINE__);
m_pSender = pSender;
}
// Возвращаем текущего клиента из списка
CClientSocket * GetCurrentClient() const throw() {
for (int i = 0; i != m_vecClientSockets.size(); ++i) {
FD_SET readerSet;
FD_ZERO(&readerSet);
timeval timeValue = { 0, 0 };
FD_SET(m_vecClientSockets[i]->GetHandle(), &readerSet);
if (::select(0, &readerSet, NULL, NULL, &timeValue) == SOCKET_ERROR) {
return m_lastConnected;
}
return m_vecClientSockets[i];
}
return m_lastConnected;
}
}; |
|
Класс сокетов клиента:
| C++ | 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
| /**
* @brief Интерфейс объектов получения данных.
*/
interface INetReceiver
{
virtual void OnReceive() = 0;
virtual void OnSend() = 0;
virtual void OnClose() = 0;
};
/**
* @brief Базовый класс клиентских сокетов.
*/
class CClientSocket : public CSocket
{
INetReceiver * m_pReceiver;
public:
CClientSocket() throw() : m_pReceiver(NULL) {
// но только не забыть потом вызвать SetReceiverObj, ConnectServer и StartEventLoop
}
CClientSocket(SOCKET handle, const string& aIpAdress, int iType = TYPE_TCP) throw(CSocketException)
: CSocket(handle, aIpAdress, iType) {
Create(iType);
// не забыть потом вызвать SetReceiverObj и StartEventLoop()
}
CClientSocket(int iType, INetReceiver * pRecvObj, const string& aServAddress, const unsigned short uServPort)
throw(CSocketException) : CSocket()
{
SetReceiverObj(pRecvObj);
ConnectServer(aServAddress, uServPort, iType);
}
// Событие приема данных
virtual void OnReceive()
{
m_pReceiver->OnReceive();
}
// Событие отправки данных
virtual void OnSend()
{
m_pReceiver->OnSend();
}
// Событие закрытия соединения
virtual void OnClose()
{
m_pReceiver->OnClose();
}
// Установить объект-получатель данных
void SetReceiverObj(INetReceiver * pReceiver) throw(CSocketException)
{
if (pReceiver == NULL)
throw CSocketException(IDS_SOCKET_INVALID_BUFFER, __FILE__, __LINE__);
m_pReceiver = pReceiver;
}
// Подключение к удаленному серверу
void ConnectServer(const string& aServAddress, const unsigned short uServPort, int iType) throw(CSocketException)
{
CSocket::Create(iType);
CSocket::Connect(aServAddress, uServPort);
Start();
}
}; |
|
Пример использования сокетов сервера:
| C++ | 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
| #include "ServerSocket.h"
#include "ClientSocket.h"
static CONST INT PORT = 0x700;
static CHAR ADDRESS[] = "127.0.0.1";
class CChatSocket : public INetSender, public INetReceiver
{
CServerSocket * m_pServSocket;
CClientSocket * m_pConnected;
public:
CChatSocket() throw() {
m_pServSocket = new CServerSocket(ADDRESS, PORT);
m_pServSocket->SetReceiverObj(this);
}
~CChatSocket() throw() {
delete m_pServSocket;
}
virtual void INetSender::OnAccept()
{
m_pConnected = m_pServSocket->GetCurrentClient();
m_pConnected->SetReceiverObj(this);
m_pConnected->Start();
}
virtual void INetSender::OnReceive()
{
}
virtual void INetSender::OnClose()
{
}
virtual void INetSender::OnSend()
{
}
virtual void INetReceiver::OnReceive()
{
const int nBuffSize = 0xFFFF;
BYTE Buffer[nBuffSize];
ZeroMemory(Buffer, nBuffSize);
while( m_pServSocket->GetCurrentClient()->Read(Buffer, nBuffSize) )
{
}
}
virtual void INetReceiver::OnSend()
{
}
virtual void INetReceiver::OnClose()
{
}
};
int main()
{
CChatSocket chatSocket;
while (INFINITE) ;
return 0;
} |
|
И главная проблема, как я уже писал - сокет серера получив сообщение FD_ACCEPT и вызвав OnAccept после этого все сообщения принимает с кодом 0
Еще раз метод обработки сообщений:
| C++ | 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
| unsigned __stdcall CSocket::Proc(LPVOID lpParam) throw(CSocketException)
{
if (lpParam == NULL)
throw CSocketException(IDS_SOCKET_EVENT_LOOP_FAILED, __FILE__, __LINE__);
CSocket * pThisObj = reinterpret_cast<CSocket *>(lpParam);
WSANETWORKEVENTS NetworkEvents;
ZeroMemory(&NetworkEvents, sizeof WSANETWORKEVENTS);
Wrappers::CHandle hNetworkEvent( WSACreateEvent() );
if (!hNetworkEvent.IsValid())
throw CSocketException(IDS_SOCKET_EVENT_LOOP_FAILED, __FILE__, __LINE__);
// Подписка на сообщения
if (WSAEventSelect(pThisObj->m_handle, hNetworkEvent.Get(),
FD_CONNECT | FD_READ | FD_WRITE | FD_ACCEPT | FD_CONNECT | FD_CLOSE) == SOCKET_ERROR)
throw CSocketException(IDS_SOCKET_EVENT_LOOP_FAILED, __FILE__, __LINE__);
while (!pThisObj->m_bStop)
{
HANDLE hNetevent = hNetworkEvent.Get();
if (WSAWaitForMultipleEvents(1, &hNetevent, FALSE, WSA_INFINITE, FALSE) == WSA_WAIT_FAILED)
throw CSocketException(IDS_SOCKET_EVENT_LOOP_FAILED, __FILE__, __LINE__);
// Определяем какое событие произошло для заданного сокета
if (WSAEnumNetworkEvents(pThisObj->m_handle, hNetworkEvent.Get(), &NetworkEvents) == SOCKET_ERROR)
throw CSocketException(IDS_SOCKET_EVENT_LOOP_FAILED, __FILE__, __LINE__);
//
// Выбираем события
//
if (NetworkEvents.lNetworkEvents & FD_READ)
{
pThisObj->OnReceive();
}
else if (NetworkEvents.lNetworkEvents & FD_CONNECT)
{
pThisObj->OnConnect();
}
else if (NetworkEvents.lNetworkEvents & FD_ACCEPT)
{
pThisObj->OnAccept();
}
else if (NetworkEvents.lNetworkEvents & FD_WRITE)
{
pThisObj->OnSend();
}
else if (NetworkEvents.lNetworkEvents & FD_CLOSE)
{
pThisObj->OnClose();
}
}
return 0;
} |
|
0
|