FTP.rar_ftp socket_ftp wininet sock_scoket ftp_socket ftp

// Server.cpp: implementation of the CServer class. // ////////////////////////////////////////////////////////////////////// #include "stdafx.h" #include "MyFtpServer.h" #include "Server.h" #ifdef _DEBUG #undef THIS_FILE static char THIS_FILE[]=__FILE__; #define new DEBUG_NEW #endif ////////////////////////////////////////////////////////////////////// // Construction/Destruction ////////////////////////////////////////////////////////////////////// UINT ProcessTreadIO( LPVOID lpParam ) ; DWORD g_dwEventTotal = 0; DWORD g_index; WSAEVENT g_events[WSA_MAXIMUM_WAIT_EVENTS]; LPSOCKET_INF g_sockets[WSA_MAXIMUM_WAIT_EVENTS]; CRITICAL_SECTION g_cs; //char g_szLocalAddr[MAX_ADDR_LEN]; BOOL g_bLoggedIn; CServer::CServer() { m_Directory[0]='\0'; m_Password[0]='\0'; m_User[0]='\0'; m_WelcomeInfor[0]='\0'; m_bStop=FALSE; } CServer::~CServer() { } UINT ServerThread(LPVOID lpParameter) { SOCKET sListen, sAccept; SOCKADDR_IN inetAddr; DWORD dwFlags; DWORD dwRecvBytes; CServer * server =(CServer*)lpParameter; InitializeCriticalSection(&g_cs); // 先取得本地地址 // sprintf( g_szLocalAddr,"%s",GetLocalAddress() ); if ((sListen = WSASocket(AF_INET, SOCK_STREAM, 0, NULL, 0, WSA_FLAG_OVERLAPPED)) == INVALID_SOCKET) { AfxMessageBox("错误：不能获取socket"); WSACleanup(); return 0; } inetAddr.sin_family = AF_INET; inetAddr.sin_addr.s_addr = htonl(INADDR_ANY); inetAddr.sin_port = htons(server->m_Port);//FTP_PORT); if (bind(sListen, (PSOCKADDR) &inetAddr, sizeof(inetAddr)) == SOCKET_ERROR) { AfxMessageBox("错误：绑定失败!"); return 0; } if (listen(sListen, SOMAXCONN)) { AfxMessageBox("错误：监听失败!"); return 0; } AfxMessageBox("Mini Ftpserver开始侦听 \n"); if ((sAccept = WSASocket(AF_INET, SOCK_STREAM, 0, NULL, 0, WSA_FLAG_OVERLAPPED)) == INVALID_SOCKET) { AfxMessageBox("错误：获取socket失败!"); return 0; } //创建第一个手动重置对象 if ((g_events[0] = WSACreateEvent()) == WSA_INVALID_EVENT) { printf("错误：WSACreateEvent failed with error %d\n", WSAGetLastError()); return 0; } // 创建一个线程处理请求 AfxBeginThread(ProcessTreadIO,(LPVOID)server); /* if (CreateThread(NULL, 0, ProcessTreadIO,(void*)server, 0, &dwThreadId) == NULL) { printf("错误：CreateThread failed with error %d\n", GetLastError()); return 0; } */ g_dwEventTotal = 1; while(!server->m_bStop) { //处理入站连接 if ((sAccept = accept(sListen, NULL, NULL)) == INVALID_SOCKET) { printf("错误：accept failed with error %d\n", WSAGetLastError()); return 0; } //回传欢迎消息 if( !server->WelcomeInfo( sAccept ) ) break; //设置ftp根目录 if( !SetCurrentDirectory( server->m_Directory) ) break; //操作临界区，防止出错 EnterCriticalSection(&g_cs); //创建一个新的SOCKET_INF结构处理接受的数据socket. if ((g_sockets[g_dwEventTotal] = (LPSOCKET_INF) GlobalAlloc(GPTR,sizeof(SOCKET_INF))) == NULL) { printf("错误：GlobalAlloc() failed with error %d\n", GetLastError()); return 0; } //初始化新的SOCKET_INF结构 char buff[DATA_BUFSIZE]; memset( buff,0,DATA_BUFSIZE ); g_sockets[g_dwEventTotal]->wsaBuf.buf = buff; g_sockets[g_dwEventTotal]->wsaBuf.len = DATA_BUFSIZE; g_sockets[g_dwEventTotal]->s = sAccept; memset(&(g_sockets[g_dwEventTotal]->o),0, sizeof(OVERLAPPED)); g_sockets[g_dwEventTotal]->dwBytesSend = 0; g_sockets[g_dwEventTotal]->dwBytesRecv = 0; g_sockets[g_dwEventTotal]->nStatus = WSA_RECV; // 接收 //创建事件 if ((g_sockets[g_dwEventTotal]->o.hEvent = g_events[g_dwEventTotal] = WSACreateEvent()) == WSA_INVALID_EVENT) { printf("WSACreateEvent() failed with error %d\n", WSAGetLastError()); return 0; } //发出接受请求 dwFlags = 0; if (WSARecv(g_sockets[g_dwEventTotal]->s, &(g_sockets[g_dwEventTotal]->wsaBuf), 1, &dwRecvBytes, &dwFlags, &(g_sockets[g_dwEventTotal]->o), NULL) == SOCKET_ERROR) { if (WSAGetLastError() != ERROR_IO_PENDING) { printf("错误：WSARecv() failed with error %d\n", WSAGetLastError()); return 0; } } g_dwEventTotal++; //离开临界区 LeaveCriticalSection(&g_cs); //使第一个事件有信号。使工作者线程处理其他的事件 if (WSASetEvent(g_events[0]) == FALSE) { printf("错误：WSASetEvent failed with error %d\n", WSAGetLastError()); return 0; } } server->m_bStop=FALSE; return 0; } //工作者线程处理函数 UINT ProcessTreadIO(LPVOID lpParameter) { DWORD dwFlags; LPSOCKET_INF pSI;//pSI是指向与客户端通信的socket结构的指针 DWORD dwBytesTransferred; DWORD i; CServer * server=(CServer*)lpParameter; //处理异步的WSASend, WSARecv等请求等 while(TRUE) { if ((g_index = WSAWaitForMultipleEvents(g_dwEventTotal, g_events, FALSE, WSA_INFINITE, FALSE)) == WSA_WAIT_FAILED) { printf("错误：WSAWaitForMultipleEvents failed %d\n", WSAGetLastError()); return 0; } if ((g_index - WSA_WAIT_EVENT_0) == 0) { WSAResetEvent(g_events[0]); continue; } pSI = g_sockets[g_index - WSA_WAIT_EVENT_0]; WSAResetEvent(g_events[g_index - WSA_WAIT_EVENT_0]); if (WSAGetOverlappedResult(pSI->s, &(pSI->o), &dwBytesTransferred, FALSE, &dwFlags) == FALSE || dwBytesTransferred == 0) { printf("Closing socket %d\n", pSI->s); if (closesocket(pSI->s) == SOCKET_ERROR) { printf("错误：closesocket() failed with error %d\n", WSAGetLastError()); } GlobalFree(pSI); WSACloseEvent(g_events[g_index - WSA_WAIT_EVENT_0]); // Cleanup g_sockets and g_events by removing the socket event handle // and socket information structure if they are not at the end of the // arrays. EnterCriticalSection(&g_cs); if ((g_index - WSA_WAIT_EVENT_0) + 1 != g_dwEventTotal) for (i = g_index - WSA_WAIT_EVENT_0; i < g_dwEventTotal; i++) { g_events[i] = g_events[i + 1]; g_sockets[i] = g_sockets[i + 1]; } g_dwEventTotal--; LeaveCriticalSection(&g_cs); continue; } // 已经有数据传递 if( pSI->nStatus == WSA_RECV ) { memcpy( &pSI->buffRecv[pSI->dwBytesRecv],pSI->wsaBuf.buf,dwBytesTransferred); pSI->dwBytesRecv += dwBytesTransferred; printf( "接受:%s\n",pSI->buffRecv); if( pSI->buffRecv[pSI->dwBytesRecv-2] == '\r' // 要保证最后是\r\n && pSI->buffRecv[pSI->dwBytesRecv-1] == '\n' && pSI->dwBytesRecv > 2 ) { if( !g_bLoggedIn ) { if( server->LoginIn(pSI) == LOGGED_IN ) g_bLoggedIn = TRUE; } else { if(server->DealCommand( pSI )==FTP_QUIT) continue; } // 缓冲区清除 memset( pSI->buffRecv,0,sizeof(pSI->buffRecv) ); pSI->dwBytesRecv = 0; } } else { pSI->dwBytesSend += dwBytesTransferred; } // 继续接收以后到来的数据 if(server->RecvReq( pSI ) == -1 ) return -1; } return 0; } void CServer::SetSerConfigInfor(const char *user,const char *password,const char *directory,const char * welcomeinfor,UINT port) { strcpy(m_User,user); strcpy(m_Password,password); strcpy(m_Directory,directory); strcpy(m_WelcomeInfor,welcomeinfor); m_Port=port; } // 由于只是简单的出现一个登录信息，直接用send就可以了，负责发送响应消息�