初步修复

Server/ServiceManager/SimpleServerFrame.cpp

@@ -0,0 +1,110 @@
+
+#include "stdafx.h"
+#include <crtdbg.h>
+#include "SimpleServerFrame.h"
+
+CSimpleServerFrame::CSimpleServerFrame()
+{
+	m_hThread = INVALID_HANDLE_VALUE;
+	m_hSocket = INVALID_SOCKET;
+	m_bAlive = true;
+	m_nPort = 0;
+}
+
+CSimpleServerFrame::~CSimpleServerFrame()
+{
+	Close();
+}
+
+bool CSimpleServerFrame::Open(USHORT nAcceptPort)
+{
+	if(m_hThread != INVALID_HANDLE_VALUE || m_hSocket != INVALID_SOCKET)
+		return false;
+
+	SOCKADDR_IN addr;
+	memset (&addr, 0, sizeof(addr));
+
+	m_hSocket = ::socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
+	if (m_hSocket == INVALID_SOCKET)
+		return false;
+	
+	addr.sin_family = AF_INET;
+	addr.sin_port = htons(nAcceptPort);
+	addr.sin_addr.S_un.S_addr = htonl(INADDR_ANY);
+
+	if (::bind(m_hSocket, (struct sockaddr *)&addr, sizeof(addr)) == SOCKET_ERROR)
+	{
+		closesocket(m_hSocket);
+		return false;
+	}
+
+	if (listen(m_hSocket, 0) == SOCKET_ERROR){
+		closesocket(m_hSocket);
+		return false;
+	}
+
+	m_hThread = (HANDLE) _beginthreadex(NULL, 0, _threadmain, (void*)this, 0, NULL);
+	return true;
+}
+
+void CSimpleServerFrame::Close()
+{
+	while(m_hThread != INVALID_HANDLE_VALUE)
+	{
+		m_bAlive = false;
+		if (WaitForSingleObject(m_hThread, 2000) != WAIT_TIMEOUT)
+		{
+			CloseHandle(m_hThread);
+			m_hThread = INVALID_HANDLE_VALUE;
+		}
+	}
+
+	if (m_hSocket != INVALID_SOCKET)
+	{
+		closesocket(m_hSocket);
+		m_hSocket = INVALID_SOCKET;
+	}
+}
+
+void CSimpleServerFrame::ThreadMain()
+{
+	char buffer[1024];
+	int readbytes, addrlen = sizeof(sockaddr_in);
+	unsigned long totalbytes;
+	SOCKADDR_IN addr;
+	fd_set fdset;
+	timeval tm;
+
+	while (m_bAlive && m_hSocket != INVALID_SOCKET)
+	{
+		tm.tv_sec = 0;
+		tm.tv_usec = INFINITE;
+
+		FD_ZERO(&fdset);
+		FD_SET(m_hSocket, &fdset);
+
+		if (select(FD_SETSIZE, (fd_set*)&fdset, (fd_set*)0, (fd_set*)0, &tm) != SOCKET_ERROR)
+		{
+			if( FD_ISSET( m_hSocket, &fdset ) )
+			{
+				SOCKET socket = accept(m_hSocket, (sockaddr*)&addr, &addrlen);
+				if (socket != INVALID_SOCKET)
+				{
+					ioctlsocket(m_hSocket, FIONREAD, &totalbytes);
+					readbytes = recv(socket, buffer, sizeof(buffer), 0);
+					Recv(&socket, buffer, readbytes);
+
+					closesocket(socket);
+					socket = INVALID_SOCKET;
+				}
+			}
+		}
+		Sleep(500);
+	}
+}
+
+unsigned int __stdcall CSimpleServerFrame::_threadmain(void * pParam)
+{
+	((CSimpleServerFrame*)pParam)->ThreadMain();
+	return 0;
+}