?? mini ftpserver.cpp
字號:
// Mini FtpServer.cpp : Defines the entry point for the console application.
#include "stdafx.h"
#include <string.h>
#include <ws2tcpip.h>
#include <stdlib.h>
#define WSA_RECV 0
#define WSA_SEND 1
#define DATA_BUFSIZE 8192
#define MAX_NAME_LEN 128
#define MAX_PWD_LEN 128
#define MAX_RESP_LEN 1024
#define MAX_REQ_LEN 256
#define MAX_ADDR_LEN 80
#define FTP_PORT 21 // FTP 控制端口
#define DATA_FTP_PORT 20 // FTP 數據端口
#define USER_OK 331
#define LOGGED_IN 230
#define LOGIN_FAILED 530
#define CMD_OK 200
#define OPENING_AMODE 150
#define TRANS_COMPLETE 226
#define CANNOT_FIND 550
#define FTP_QUIT 221
#define CURR_DIR 257
#define DIR_CHANGED 250
#define OS_TYPE 215
#define REPLY_MARKER 504
#define PASSIVE_MODE 227
#define FTP_USER "toldo"
#define FTP_PASS "toldo"
#define DEFAULT_HOME_DIR "C:\\TEMP"
#define MAX_FILE_NUM 1024
#define MODE_PORT 0
#define MODE_PASV 1
#define PORT_BIND 1821
typedef struct {
CHAR buffRecv[DATA_BUFSIZE];
CHAR buffSend[DATA_BUFSIZE];
WSABUF wsaBuf;
SOCKET s;
WSAOVERLAPPED o;
DWORD dwBytesSend;
DWORD dwBytesRecv;
int nStatus;
} SOCKET_INF, *LPSOCKET_INF;
typedef struct {
TCHAR szFileName[MAX_PATH];
DWORD dwFileAttributes;
FILETIME ftCreationTime;
FILETIME ftLastAccessTime;
FILETIME ftLastWriteTime;
DWORD nFileSizeHigh;
DWORD nFileSizeLow;
} FILE_INF, *LPFILE_INF;
DWORD WINAPI ProcessTreadIO( LPVOID lpParam ) ;
BOOL WelcomeInfo( SOCKET s );
int LoginIn( LPSOCKET_INF pSocketInfo );
int SendRes( LPSOCKET_INF pSI );
int RecvReq( LPSOCKET_INF pSI );
int DealCommand( LPSOCKET_INF pSI );
int GetFileList( LPFILE_INF pFI, UINT nArraySize, const char* szPath );
char* GetLocalAddress();
char* HostToNet( char* szPath ) ;
char* NetToHost( char* szPath ) ;
char* RelativeDirectory( char* szDir );
char* AbsoluteDirectory( char* szDir );
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;
// 主函數，控制臺程序開始的地方
void main(void)
{
WSADATA wsaData;
SOCKET sListen, sAccept;
SOCKADDR_IN inetAddr;
DWORD dwFlags;
DWORD dwThreadId;
DWORD dwRecvBytes;
INT nRet;
InitializeCriticalSection(&g_cs);
if (( nRet = WSAStartup(0x0202,&wsaData)) != 0 ) {
printf("錯誤：WSAStartup failed with error %d\n", nRet);
return;
}
// 先取得本地地址
sprintf( g_szLocalAddr,"%s",GetLocalAddress() );
if ((sListen = WSASocket(AF_INET, SOCK_STREAM, 0, NULL, 0,
WSA_FLAG_OVERLAPPED)) == INVALID_SOCKET)
{
printf("錯誤：Failed to get a socket %d\n", WSAGetLastError());
WSACleanup();
return;
}
inetAddr.sin_family = AF_INET;
inetAddr.sin_addr.s_addr = htonl(INADDR_ANY);
inetAddr.sin_port = htons(FTP_PORT);
if (bind(sListen, (PSOCKADDR) &inetAddr, sizeof(inetAddr)) == SOCKET_ERROR)
{
printf("錯誤：bind() failed with error %d\n", WSAGetLastError());
return;
}
if (listen(sListen, SOMAXCONN))
{
printf("錯誤：listen() failed with error %d\n", WSAGetLastError());
return;
}
printf("Mini Ftpserver已經啟動 \n");
printf("Mini Ftpserver開始偵聽 \n");
if ((sAccept = WSASocket(AF_INET, SOCK_STREAM, 0, NULL, 0,
WSA_FLAG_OVERLAPPED)) == INVALID_SOCKET)
{
printf("錯誤：Failed to get a socket %d\n", WSAGetLastError());
return;
}
//創建第一個手動重置對象
if ((g_events[0] = WSACreateEvent()) == WSA_INVALID_EVENT)
{
printf("錯誤：WSACreateEvent failed with error %d\n", WSAGetLastError());
return;
}
// 創建一個線程處理請求
if (CreateThread(NULL, 0, ProcessTreadIO, NULL, 0, &dwThreadId) == NULL)
{
printf("錯誤：CreateThread failed with error %d\n", GetLastError());
return;
}
g_dwEventTotal = 1;
while(TRUE)
{
//處理入站連接
if ((sAccept = accept(sListen, NULL, NULL)) == INVALID_SOCKET)
{
printf("錯誤：accept failed with error %d\n", WSAGetLastError());
return;
}
//回傳歡迎消息
if( !WelcomeInfo( sAccept ) ) break;
//設置ftp根目錄
if( !SetCurrentDirectory( DEFAULT_HOME_DIR ) ) 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;
}
//初始化新的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;
}
//發出接受請求
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;
}
}
g_dwEventTotal++;
//離開臨界區
LeaveCriticalSection(&g_cs);
//使第一個事件有信號。使工作者線程處理其他的事件
if (WSASetEvent(g_events[0]) == FALSE)
{
printf("錯誤：WSASetEvent failed with error %d\n", WSAGetLastError());
return;
}
}
}
//工作者線程處理函數
DWORD WINAPI ProcessTreadIO(LPVOID lpParameter)
{
DWORD dwFlags;
LPSOCKET_INF pSI;
DWORD dwBytesTransferred;
DWORD i;
//處理異步的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( LoginIn(pSI) == LOGGED_IN )
g_bLoggedIn = TRUE;
}
else
{
if(DealCommand( pSI )==FTP_QUIT)
continue;
}
// 緩沖區清除
memset( pSI->buffRecv,0,sizeof(pSI->buffRecv) );
pSI->dwBytesRecv = 0;
}
}
else
{
pSI->dwBytesSend += dwBytesTransferred;
}
// 繼續接收以后到來的數據
if( RecvReq( pSI ) == -1 )
return -1;
}
return 0;
}
// 由于只是簡單的出現一個登錄信息，直接用send就可以了
int SendRes( LPSOCKET_INF pSI )
{
static DWORD dwSendBytes = 0;
pSI->nStatus = WSA_SEND;
memset(&(pSI->o), 0,sizeof(WSAOVERLAPPED));
pSI->o.hEvent = g_events[g_index - WSA_WAIT_EVENT_0];
pSI->wsaBuf.buf = pSI->buffSend + pSI->dwBytesSend;
pSI->wsaBuf.len = strlen( pSI->buffSend ) - pSI->dwBytesSend;
if (WSASend(pSI->s, &(pSI->wsaBuf), 1,&dwSendBytes,
0,&(pSI->o), NULL) == SOCKET_ERROR)
{
if (WSAGetLastError() != ERROR_IO_PENDING)
{
printf("WSASend() failed with error %d\n", WSAGetLastError());
return -1;
}
}
return 0;
}
//接受數據
int RecvReq( LPSOCKET_INF pSI )
{
static DWORD dwRecvBytes = 0;
pSI->nStatus = WSA_RECV;
DWORD dwFlags = 0;
memset(&(pSI->o), 0,sizeof(WSAOVERLAPPED));
pSI->o.hEvent = g_events[g_index - WSA_WAIT_EVENT_0];
pSI->wsaBuf.len = DATA_BUFSIZE;
if (WSARecv(pSI->s, &(pSI->wsaBuf), 1, &dwRecvBytes,
&dwFlags,&(pSI->o), NULL) == SOCKET_ERROR)
{
if (WSAGetLastError() != ERROR_IO_PENDING)
{
printf("WSARecv() failed with error %d\n", WSAGetLastError());
return -1;
}
}
return 0;
}
//顯示歡迎消息
BOOL WelcomeInfo( SOCKET s )
{
char* szWelcomeInfo = "220 歡迎您登錄到Mini FtpServer...\r\n";
if( send( s,szWelcomeInfo,strlen(szWelcomeInfo),0 ) == SOCKET_ERROR )
{
printf("Ftp client error:%d\n", WSAGetLastError() );
return FALSE;
}
// 剛進來，還沒連接，故設置初始狀態為false
g_bLoggedIn = FALSE;
return TRUE;
}
//登錄函數
int LoginIn( LPSOCKET_INF pSocketInfo )
{
const char* szUserOK = "331 User name okay, need password.\r\n";
const char* szLoggedIn = "230 User logged in, proceed.\r\n";
int nRetVal = 0;
static char szUser[MAX_NAME_LEN], szPwd[MAX_PWD_LEN];
LPSOCKET_INF pSI = pSocketInfo;
// 取得登錄用戶名
if( strstr(strupr(pSI->buffRecv),"USER") )
{
sprintf(szUser,"%s",pSI->buffRecv+strlen("USER")+1);
strtok( szUser,"\r\n");
// 響應信息
sprintf(pSI->buffSend,"%s",szUserOK );
if( SendRes(pSI) == -1 ) return -1;
return USER_OK;
}
if( strstr(strupr(pSI->buffRecv),"PASS") || strstr(pSI->buffRecv,"pass") )
{
sprintf(szPwd,"%s",pSI->buffRecv+strlen("PASS")+1 );
strtok( szPwd,"\r\n");
// 判斷用戶名跟口令正確性
if( stricmp( szPwd,FTP_USER) || stricmp(szUser,FTP_PASS) )
{
sprintf(pSI->buffSend,"530 User %s cannot log in.\r\n",szUser );
printf("User %s cannot log in\n",szUser );
nRetVal = LOGIN_FAILED;
}
else
{
sprintf(pSI->buffSend,"%s",szLoggedIn);
printf("User %s logged in\n",szUser );
nRetVal = LOGGED_IN;
}
if( SendRes( pSI ) == -1 )
return -1;
}
return nRetVal;
}
char* ConvertCommaAddress( char* szAddress, WORD wPort )
{
char szPort[10];
sprintf( szPort,"%d,%d",wPort/256,wPort%256 );
char szIpAddr[20];
sprintf( szIpAddr,"%s,",szAddress );
int idx = 0;
while( szIpAddr[idx] )
{
if( szIpAddr[idx] == '.' )
szIpAddr[idx] = ',';
idx ++;
}
sprintf( szAddress,"%s%s",szIpAddr,szPort );
return szAddress;
}
int ConvertDotAddress( char* szAddress, LPDWORD pdwIpAddr, LPWORD pwPort )
{
int idx = 0,i = 0, iCount = 0;
char szIpAddr[MAX_ADDR_LEN]; memset( szIpAddr,0,sizeof(szIpAddr) );
char szPort[MAX_ADDR_LEN]; memset( szPort,0, sizeof(szPort) );
*pdwIpAddr = 0; *pwPort = 0;
while( szAddress[idx] )
{
if( szAddress[idx] == ',' )
{
iCount ++;
szAddress[idx] ='.';
}
if( iCount < 4 )
szIpAddr[idx] = szAddress[idx];
else
szPort[i++] = szAddress[idx];
idx++;
}
if( iCount != 5 ) return -1;
*pdwIpAddr = inet_addr( szIpAddr );
if( *pdwIpAddr == INADDR_NONE ) return -1;
char *pToken = strtok( szPort+1,"." );
if( pToken == NULL ) return -1;
*pwPort = (WORD)(atoi(pToken) * 256);
pToken = strtok(NULL,".");
if( pToken == NULL ) return -1;
*pwPort += (WORD)atoi(pToken);
return 0;
}
UINT FileListToString( char* buff, UINT nBuffSize,BOOL bDetails )
{
FILE_INF fi[MAX_FILE_NUM];
int nFiles = GetFileList( fi, MAX_FILE_NUM, "*.*" );
char szTemp[128];
sprintf( buff,"%s","" );
if( bDetails ) {
for( int i=0; i<nFiles; i++) {
if( strlen(buff)>nBuffSize-128 ) break;
if(!strcmp(fi[i].szFileName,".")) continue;
if(!strcmp(fi[i].szFileName,"..")) continue;
// 時間
SYSTEMTIME st;
FileTimeToSystemTime(&(fi[i].ftLastWriteTime), &st);
char *szNoon = "AM";
if( st.wHour > 12 )
{
st.wHour -= 12;
szNoon = "PM";
}
if( st.wYear >= 2000 )
st.wYear -= 2000;
else st.wYear -= 1900;
sprintf( szTemp,"%02u-%02u-%02u %02u:%02u%s ",
st.wMonth,st.wDay,st.wYear,st.wHour,st.wMonth,szNoon );
strcat( buff,szTemp );
if( fi[i].dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY )
{
strcat(buff,"<DIR>");
strcat(buff," ");
}
else
{
strcat(buff," ");
// 文件大小
sprintf( szTemp,"% 9d ",fi[i].nFileSizeLow );
strcat( buff,szTemp );
}
// 文件名
strcat( buff,fi[i].szFileName );
strcat( buff,"\r\n");
}
}
else
{
for( int i=0; i<nFiles; i++)
{
if( strlen(buff) + strlen( fi[i].szFileName ) + 2 < nBuffSize )
{
strcat( buff, fi[i].szFileName );
strcat( buff, "\r\n");
}
else
break;
}
}
return strlen( buff );
}
DWORD ReadFileToBuffer( const char* szFile, char *buff, DWORD nFileSize )
{
DWORD idx = 0;
DWORD dwBytesLeft = nFileSize;
DWORD dwNumOfBytesRead = 0;
char lpFileName[MAX_PATH];
GetCurrentDirectory( MAX_PATH,lpFileName );
strcat( lpFileName,"\\" );
strcat(lpFileName,szFile );
?? 快捷鍵說明
復制代碼
Ctrl + C
搜索代碼
Ctrl + F
全屏模式
F11
切換主題
Ctrl + Shift + D
顯示快捷鍵
?
增大字號
Ctrl + =
減小字號
Ctrl + -