Mini-FTPserver/ListenSocket.cpp at master · codecopy/Mini-FTPserver · GitHub

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
// ListenSocket.cpp : implementation file
//

#include "stdafx.h"
#include "FTPserver.h"
#include "FTPserverDlg.h"
#include "ListenSocket.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

/////////////////////////////////////////////////////////////////////////////
// CListenSocket

CListenSocket::CListenSocket()
:m_pWndServer(NULL)
{
}

CListenSocket::~CListenSocket()
{
}


// Do not edit the following lines, which are needed by ClassWizard.
#if 0
BEGIN_MESSAGE_MAP(CListenSocket, CAsyncSocket)
	//{{AFX_MSG_MAP(CListenSocket)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()
#endif	// 0

/////////////////////////////////////////////////////////////////////////////
// CListenSocket member functions

/********************************************************************/
/*																	*/
/* 函数名称 : CListenSocket::OnAccept								*/
/* 作用     : 接收到新连接时触发									*/
/*																	*/
/********************************************************************/
void CListenSocket::OnAccept(int nErrorCode)
{
	// 创建临时的SOCKET
	CSocket tmpSocket;

	// 接受连接
	Accept(tmpSocket);

	// 使用SUSPENDED模式创建新的线程，这样可以在执行线程前设置线程信息
	CConnectThread* pThread = (CConnectThread*)AfxBeginThread(RUNTIME_CLASS(CConnectThread), THREAD_PRIORITY_NORMAL, 0, CREATE_SUSPENDED);
	if (!pThread)
	{
		tmpSocket.Close();
		TRACE("Could not create thread\n");
		return;
	}

	// 添加线程指针到线程链表
	CFTPserverDlg *pWnd = (CFTPserverDlg *)m_pWndServer;
	pWnd->m_CriticalSection.Lock();
    pWnd->m_ThreadList.AddTail(pThread);

	pWnd->m_CriticalSection.Unlock();

	// 传递窗口指针
	pThread->m_pWndServer = m_pWndServer;

	// 传递SOCKET
	pThread->m_hSocket = tmpSocket.Detach();

	// 重新启动SUSPENDED模式下的线程
	pThread->ResumeThread();

	CAsyncSocket::OnAccept(nErrorCode);
}