ftpservernew.cpp

ftp协议实现程序

// FtpServerNew.cpp : 定义控制台应用程序的入口点。
//

#include "stdafx.h"
#include "DataStructAndConstant.h"
#include "TreadFunction.h"
CRITICAL_SECTION  PerSocketVectorProtection;//作为操作vector的全局互斥量
CRITICAL_SECTION  AddIocpProtection;//为了添加完成端口处理socket的全局变量
HANDLE CompetionPort;//作为全局的完成端口的句柄
extern bool DeleteVectorItem(LPPER_HANDLE_DATA pPerHandleData);


PerSocketDataVector cPerSocketDataVector;
int _tmain(int argc, char* argv[])
{
	SOCKADDR_IN InternetAddr;
	SOCKET Listen,Accept;
	SYSTEM_INFO SystenInfo;
	DWORD ThreadID;
	WSADATA wsadata;
	DWORD Ret;
	if (Ret = WSAStartup(0x2020,&wsadata) != 0)
	{
		printf("WSAStartup failed with error %d\n",Ret);
		return 0;
	}
   //打开一个空的完成端口


	if ((CompetionPort = CreateIoCompletionPort(INVALID_HANDLE_VALUE,NULL,0,0)) == NULL)
	{
		printf("CreateIoCompletionPort failed with error %d\n",GetLastError());
		return 0;
	}
	else
	{
		printf("CreateIoCompletionPort Success……\r\n");
	}
	GetSystemInfo(&SystenInfo);
	// 开启cpu个数的2倍个的线程 
	for (ULONG i=0; i < 6+SystenInfo.dwNumberOfProcessors*2; i++)
	{
		HANDLE ThreadHandle;
		//创建服务器工作线程，并且向线程传送完成端口
		if ((ThreadHandle = CreateThread(NULL,0,ServerWorkerThread,CompetionPort,0,&ThreadID)) == NULL)
		{
			printf("CreateThread failed with error %d\n" ,GetLastError());
			return 0;
		}
		CloseHandle(ThreadHandle);
	}
	printf("Start %d Threats……\r\n",6+SystenInfo.dwNumberOfProcessors*2);
	//打开一个服务器socket
	if ((Listen = WSASocket(AF_INET, SOCK_STREAM, 0, NULL, 0, WSA_FLAG_OVERLAPPED)) == INVALID_SOCKET)
	{
		printf("WSASocket() failed with error %d\n", WSAGetLastError());
		return 0;
	}
	InternetAddr.sin_family = AF_INET;
	InternetAddr.sin_addr.S_un.S_addr = htonl(INADDR_ANY);
	InternetAddr.sin_port = htons(LISTENPORT);
	if (bind(Listen,(LPSOCKADDR)&InternetAddr,sizeof(InternetAddr)) == SOCKET_ERROR)
	{
		printf("bind failed with error %d\n",WSAGetLastError());
		return 0;
	}
    if (listen(Listen,5) == SOCKET_ERROR)
	{
		printf("listen failed with error %d\n",WSAGetLastError());
		return 0;
	}
	printf("Ftp Server Start……… \r\n");
	InitializeCriticalSection(&PerSocketVectorProtection);
	InitializeCriticalSection(&AddIocpProtection);
     //接收连接并且分发给完成端口
	while (TRUE)
	{
		if ((Accept = WSAAccept(Listen,NULL,NULL,NULL,0)) == SOCKET_ERROR)
		{
			printf("WSAAccept failed with error %d\n",WSAGetLastError());
			continue;
		}
         //创建与套接字相关的套接字信息结构
		LPPER_SOCKET_DATA pAcceptSocketData=new PER_SOCKET_DATA;
		EnterCriticalSection(&PerSocketVectorProtection);
			cPerSocketDataVector.push_back(pAcceptSocketData);
		LeaveCriticalSection(&PerSocketVectorProtection);
		memset((void*)pAcceptSocketData,0,sizeof(PER_SOCKET_DATA));
		pAcceptSocketData->cPerHandleData.Socket=Accept;
		pAcceptSocketData->cPerHandleData.uHandleInfo.cCtrlInfo.EventDataValid=CreateEvent(NULL,TRUE,FALSE,NULL);
		pAcceptSocketData->cPerHandleData.enCurrentHandleType=HandleControl;
		pAcceptSocketData->cPerHandleData.uHandleInfo.cCtrlInfo.pDataLinkSocketDataBuffer=(void *)new PER_SOCKET_DATA;
		strcpy(pAcceptSocketData->cPerHandleData.uHandleInfo .cCtrlInfo.strCurrentWorkPath,DEFAULTWORKPATH);
		//与我们的创建的那个完成端口关联起来,将关键项也与指定的一个完成端口关联
		EnterCriticalSection(&AddIocpProtection);
		if ((CreateIoCompletionPort((HANDLE)Accept,CompetionPort,reinterpret_cast<DWORD>(&(pAcceptSocketData->cPerHandleData)),0)) == NULL)
		{
			printf("CreateIoCompletionPort failed with error%d\n",GetLastError());
			LeaveCriticalSection(&AddIocpProtection);
			closesocket(Accept);
			continue;
		}
		LeaveCriticalSection(&AddIocpProtection);
		int zero = 0; 
		if ( setsockopt(pAcceptSocketData->cPerHandleData.Socket, SOL_SOCKET, SO_SNDBUF, (char *)&zero, sizeof(zero) ) == SOCKET_ERROR )
		{
			printf("Set socket option Error %d",WSAGetLastError());
		}
		zero = 0; 
		if ( setsockopt(pAcceptSocketData->cPerHandleData.Socket, SOL_SOCKET, SO_RCVBUF, (char *)&zero, sizeof(zero) ) == SOCKET_ERROR )
		{
			printf("Set socket option Error %d",WSAGetLastError());
		}
 // 创建同下面的WSARecv调用相关的IO套接字信息结构体
		pAcceptSocketData->cPerIoData.cBuffer.buf=pAcceptSocketData->cPerIoData.stBuffer;
		strcpy(pAcceptSocketData->cPerIoData.cBuffer.buf,"220 Welcome to Access BlackBeast FTP Server...\r\n");
		pAcceptSocketData->cPerIoData.cBuffer.len=(DWORD)strlen(pAcceptSocketData->cPerIoData.cBuffer.buf);
		pAcceptSocketData->cPerIoData.IoType=ControlIoSend;
		pAcceptSocketData->cPerIoData.uIoInfo.cControlIoSendInfo.enNextToDo=ToReceive;
	
									
		DWORD SendBytes=0;
		if (WSASend(pAcceptSocketData->cPerHandleData.Socket,&(pAcceptSocketData->cPerIoData.cBuffer),1,&SendBytes,0,&(pAcceptSocketData->cPerIoData.Overlapped),NULL) ==SOCKET_ERROR ) 
		{ 
			if (WSAGetLastError() != ERROR_IO_PENDING) 
			{ 
				printf("WSASend() fialed with error %d\n",WSAGetLastError()); 
				DeleteVectorItem(&(pAcceptSocketData->cPerHandleData));
				closesocket(Accept);
				continue; 
			} 
		} 
		
	
	}
	return 0;
}