win32FTP程序设计

掌握socket基于事件机制的网络程序设计，掌握多线程技术的FTP Server端设计方法，掌握FTP标准基本协议及其程序的实现，掌握文件内容的网络传输设计方法。

利用CFtpServer类接收和解析客户端命令，编写FTP客户端程序，服务器端使用多线程，实现多用户同时登录管理；

利用CFtpConnection和CInternetSession类，编写FTP客户端，实现简单文件操作功能。

1. FTP服务器实现

   1.服务器窗口界面设计

   

   服务器主界面

   2.功能实现：

   （1）初始化FTP并启动监听

   BOOL CFTPServer::Start()

   {

       if (m_bRunning)

           return FALSE;

    

       // create dummy window for message routing

   if (!CWnd::CreateEx(0, AfxRegisterWndClass(0), "FTP Server Notification Sink", WS_POPUP, 0,0,0,0, NULL, 0))

       {

           AddTraceLine(0, "Failed to create notification window.");

           return FALSE;

       }

       // created the listen socket

       if (m_ListenSocket.Create(m_nPort))

       {

           // start listening

           if (m_ListenSocket.Listen())

           {

               m_ListenSocket.m_pWndServer = this;

               m_bRunning = TRUE;    

               

               SetTimer(1, m_nStatisticsInterval, NULL);

    

               AddTraceLine(0, "FTP Server started on port %d.", m_nPort);

               return TRUE;

           }

       }

       AddTraceLine(0, "FTP Server failed to listen on port %d.", m_nPort);

    

       // destroy notification window

       if (IsWindow(m_hWnd))

           DestroyWindow();

       m_hWnd = NULL;

    

       return FALSE;

   }

    

   （2）停止FTP

   通知各线程停止监听，关闭进程，释放连接，实现代码如下

   void CFTPServer::Stop()

   {

       if (!m_bRunning)

           return;

    

       // stop statistics timer

       KillTimer(1);

    

       m_bRunning = FALSE;    

       m_ListenSocket.Close();

    

       CConnectThread* pThread = NULL;

    

       // close all running threads

       do

       {

           m_CriticalSection.Lock();

    

           POSITION pos = m_ThreadList.GetHeadPosition();

           if (pos != NULL)

           {

               pThread = (CConnectThread *)m_ThreadList.GetAt(pos);

           

               m_CriticalSection.Unlock();

    

               // save thread members

               int nThreadID = pThread->m_nThreadID;

               HANDLE hThread = pThread->m_hThread;

    

               AddTraceLine(0, "[%u] Shutting down thread...", nThreadID);

    

               // tell thread to stop

               pThread->SetThreadPriority(THREAD_PRIORITY_HIGHEST);

               pThread->PostThreadMessage(WM_QUIT,0,0);

    

               // wait for thread to end, while keeping the messages pumping (max 5 seconds)

               if (WaitWithMessageLoop(hThread, 5000) == FALSE)

               {

                   // thread doesn't want to stopped

                   AddTraceLine(0, "[%u] Problem while killing thread.", nThreadID);

                   // don't try again, so remove

                   m_CriticalSection.Lock();

                   POSITION rmPos = m_ThreadList.Find(pThread);

                   if (rmPos != NULL)

                       m_ThreadList.RemoveAt(rmPos);

                   m_CriticalSection.Unlock();

               }

               else

               {

                   AddTraceLine(0, "[%u] Thread successfully stopped.", nThreadID);

               }

           }

           else

           {

               m_CriticalSection.Unlock();    

               pThread = NULL;

           }

       }

       while (pThread != NULL);

    

       AddTraceLine(0, "FTP Server stopped.");

    

       if (IsWindow(m_hWnd))

           DestroyWindow();

    

       m_hWnd = NULL;

   }

   （3）账号操作

   用户账号操作包括：添加，编辑、删除和保存，具体实现代码如下：

   //添加账号

   void CUserAccountPage::OnAddUser()

   {

       CAddUserDlg dlg;

       if (dlg.DoModal() == IDOK)

       {

           for (int i=0; i<m_UsersList.GetItemCount(); i++)

           {

               CString strName;

               strName = m_UsersList.GetItemText(i, 0);

               if (strName.CompareNoCase(dlg.m_strName) == 0)

               {

                   AfxMessageBox("Sorry, this user already exists!");

                   return;

               }

           }

    

           CUser user;

           user.m_strName = dlg.m_strName;

           user.m_strPassword = "";

    

           int nItem = m_UsersList.InsertItem(0, user.m_strName, 0);

           if (nItem <= m_nPreviousIndex)

               m_nPreviousIndex++;

    

           // add home directory item

           user.m_bAllowCreateDirectory = FALSE;

           user.m_bAllowDelete = FALSE;

           user.m_bAllowDownload = TRUE;

           user.m_bAllowRename = FALSE;

           user.m_bAllowUpload = FALSE;

           user.m_strHomeDirectory = "";

    

           int index = m_UserArray.Add(user);

           

           m_UsersList.SetItemData(nItem, index);

           m_UsersList.SetItemState(nItem, LVIS_SELECTED | LVIS_FOCUSED , LVIS_SELECTED | LVIS_FOCUSED);

    

           OnSelchangeUserlist();

    

           // update user manager

           theServer.m_UserManager.UpdateUserList(m_UserArray);

           SetModified();

    

           // launch directory browser

           PostMessage(WM_COMMAND, IDC_BROWSE);

       }

   }

    

   //更改用户名

   void CUserAccountPage::OnEditUser()

   {

       // get selected user

       int nSelIndex = m_UsersList.GetNextItem(-1, LVNI_ALL | LVNI_SELECTED);

   if(nSelIndex == -1)

   return;

    

       int nUserIndex = m_UsersList.GetItemData(nSelIndex);

    

       CAddUserDlg dlg;

       dlg.m_strTitle = "修改";

       dlg.m_strName = m_UserArray[nUserIndex].m_strName;

    

       if (dlg.DoModal() == IDOK)

       {

           // check if user already exists

           for (int i=0; i<m_UsersList.GetItemCount(); i++)

           {

               if (i != nSelIndex)

               {

                   CString strName;

                   strName = m_UsersList.GetItemText(i, 0);

                   if (strName.CompareNoCase(dlg.m_strName) == 0)

                   {

                       AfxMessageBox("Sorry, this user already exists!");

                       return;

                   }

               }

           }

    

           m_UserArray[nUserIndex].m_strName = dlg.m_strName;

    

           m_UsersList.DeleteItem(nSelIndex);

           nSelIndex = m_UsersList.InsertItem(0, dlg.m_strName, 0);

    

           m_UsersList.SetItemData(nSelIndex, nUserIndex);

           m_UsersList.SetItemState(nSelIndex, LVIS_SELECTED | LVIS_FOCUSED , LVIS_SELECTED | LVIS_FOCUSED);

           m_nPreviousIndex = nSelIndex;

           

           OnSelchangeUserlist();

    

           // update user manager

           theServer.m_UserManager.UpdateUserList(m_UserArray);

           SetModified(FALSE);

       }

   }

   //更改用户目录

   void CUserAccountPage::OnBrowse()

   {

       CString strDir = BrowseForFolder(m_hWnd, "Select a home directory:", BIF_RETURNONLYFSDIRS);

       if (!strDir.IsEmpty())

       {

           m_strHomeDirectory = strDir;

           UpdateData(FALSE);

       }    

   }

    

   //删除账号

   void CUserAccountPage::OnDelUser()

   {

       int nSelIndex = m_UsersList.GetNextItem(-1, LVNI_ALL | LVNI_SELECTED);

   if(nSelIndex == -1)

   return;

    

       CString strText;

    

       strText.Format("Are you sure you want to delete '%s'?", m_UsersList.GetItemText(nSelIndex, 0));

       if (MessageBox(strText, "FTP Server", MB_YESNO | MB_ICONQUESTION) != IDYES)

       {

           return;

       }

    

       int nUserIndex = m_UsersList.GetItemData(nSelIndex);

    

       // remove user from array

       m_UserArray.RemoveAt(nUserIndex);

    

       m_nPreviousIndex = -1;

       // update item data values

       

       m_UsersList.SetRedraw(FALSE);

       m_UsersList.DeleteAllItems();

       

       // update user list

       for (int i=0; i < m_UserArray.GetSize(); i++)

       {

           int nIndex = m_UsersList.InsertItem(0, m_UserArray[i].m_strName, 0);

           m_UsersList.SetItemData(nIndex, i);

       }    

       m_UsersList.SetRedraw(TRUE);

    

       // update user manager

       theServer.m_UserManager.UpdateUserList(m_UserArray);

       SetModified(FALSE);

    

       m_UsersList.SetItemState(nSelIndex-1, LVIS_SELECTED | LVIS_FOCUSED , LVIS_SELECTED | LVIS_FOCUSED);

       OnSelchangeUserlist();

   }

    

   //保存当前修改

   BOOL CUserAccountPage::UpdateAccount(int nSelIndex)

   {

       if (nSelIndex == -1)

       {

           // get selected user

           nSelIndex = m_UsersList.GetNextItem(-1, LVNI_ALL | LVNI_SELECTED);

       }

    

       if (nSelIndex != -1)

       {

           UpdateData();

    

           int nUserIndex = m_UsersList.GetItemData(nSelIndex);

    

           m_UserArray[nUserIndex].m_strPassword = m_strPassword;

           m_UserArray[nUserIndex].m_bAccountDisabled = m_bDisableAccount;

    

           // check if it's a valid directory

           if (GetFileAttributes(m_strHomeDirectory) == 0xFFFFFFFF)

           {

               MessageBox("Please enter a valid home directory", "FTP Server", MB_OK | MB_ICONEXCLAMATION);

               GetDlgItem(IDC_HOME_DIRECTORY)->SetFocus();

               m_UsersList.SetItemState(m_nPreviousIndex, LVIS_SELECTED | LVIS_FOCUSED , LVIS_SELECTED | LVIS_FOCUSED);

               return FALSE;

           }

    

           m_UserArray[nUserIndex].m_strHomeDirectory = m_strHomeDirectory;

           m_UserArray[nUserIndex].m_bAllowCreateDirectory = m_bAllowCreateDirectory;

           m_UserArray[nUserIndex].m_bAllowDelete = m_bAllowDelete;

           m_UserArray[nUserIndex].m_bAllowDownload = m_bAllowDownload;

           m_UserArray[nUserIndex].m_bAllowRename = m_bAllowRename;

           m_UserArray[nUserIndex].m_bAllowUpload = m_bAllowUpload;

           

           // update user manager

           theServer.m_UserManager.UpdateUserList(m_UserArray);

           return TRUE;

       }

       return FALSE;

   }

    

   （4）查看在线用户及其使用的线程号

   在线用户显示页面主要代码如下：

   BOOL COnlineUsersPage::OnInitDialog()

   {

       CDialog::OnInitDialog();

    

       m_OnlineUsers.InsertColumn(0, "ThreadID");        

       m_OnlineUsers.InsertColumn(1, "Username");    

       m_OnlineUsers.InsertColumn(2, "IP Adress");    

       m_OnlineUsers.InsertColumn(3, "Login Time");

       

       DWORD dwStyle = m_OnlineUsers.GetExtendedStyle();

       dwStyle |= LVS_EX_FULLROWSELECT;

   m_OnlineUsers.SetExtendedStyle(dwStyle);

       return TRUE;

   }

   void COnlineUsersPage::OnSize(UINT nType, int cx, int cy)

   {

       CDialog::OnSize(nType, cx, cy);

       

       if (IsWindow(::GetDlgItem(m_hWnd, IDC_ONLINE_USERS)))

       {

           CRect rect;

           GetClientRect(rect);

           m_OnlineUsers.MoveWindow(rect);

           m_OnlineUsers.SetColumnWidth(0, 0);

           m_OnlineUsers.SetColumnWidth(1, rect.Width()/3-2);

           m_OnlineUsers.SetColumnWidth(2, rect.Width()/3-2);

           m_OnlineUsers.SetColumnWidth(3, rect.Width()/3-2);

       }    

   }

    

   void COnlineUsersPage::AddUser(DWORD nThreadID, LPCTSTR lpszName, LPCTSTR lpszAddress)

   {

       CString strThreadID;

       strThreadID.Format("%d", nThreadID);

    

       LVFINDINFO info;

       

       info.flags = LVFI_PARTIAL|LVFI_STRING;

       info.psz = (LPCTSTR)strThreadID;

    

       int nIndex = m_OnlineUsers.FindItem(&info);

       if (nIndex == -1)

       {

           nIndex = m_OnlineUsers.InsertItem(0, strThreadID);

       }

    

       m_OnlineUsers.SetItemText(nIndex, 1, lpszName);

       m_OnlineUsers.SetItemText(nIndex, 2, lpszAddress);

       m_OnlineUsers.SetItemText(nIndex, 3, CTime::GetCurrentTime().Format("%H:%M:%S"));

       

   }

   void COnlineUsersPage::RemoveUser(DWORD nThreadID)

   {

       LVFINDINFO info;

       

       CString strThreadID;

       strThreadID.Format("%d", nThreadID);

    

       info.flags = LVFI_PARTIAL|LVFI_STRING;

       info.psz = (LPCTSTR)strThreadID;

    

       int nIndex = m_OnlineUsers.FindItem(&info);

       if (nIndex != -1)

       {

           m_OnlineUsers.DeleteItem(nIndex);

       }

   }

   void COnlineUsersPage::OnContextMenu(CWnd* pWnd, CPoint point)

   {

       // get selected user

       int nIndex = m_OnlineUsers.GetNextItem(-1, LVNI_ALL | LVNI_SELECTED);

   if(nIndex == -1)

   return;

    

       CMenu menu;

       menu.LoadMenu(MAKEINTRESOURCE(IDR_ONLINE_MENU));

       menu.GetSubMenu(0)->TrackPopupMenu(0, point.x, point.y, this, NULL);            

    

   }

   void COnlineUsersPage::OnKickUser()

   {

       int nIndex = m_OnlineUsers.GetNextItem(-1, LVNI_ALL | LVNI_SELECTED);

       while (nIndex != -1)

       {

           CString strThreadID = m_OnlineUsers.GetItemText(nIndex, 0);

           PostThreadMessage(atoi(strThreadID), WM_QUIT, 0 ,0);

           nIndex = m_OnlineUsers.GetNextItem(nIndex, LVNI_ALL | LVNI_SELECTED);

       }

   }

    

    

   void COnlineUsersPage::OnCancel()

   {

   //    CDialog::OnCancel();

   }

    

   void COnlineUsersPage::OnOK()

   {

   //    CDialog::OnOK();

   }

    

2. FTP客户端实现

   为检验FTP客户端的可用性和进一步掌握FTP标准基本协议及文件内容的网络传输设计方法，我实现了一个简单的FTP客户端程序。

   1.客户端窗口界面设计

FTP客户端程序主界面

 

2.功能实现：

（1）服务器的连接与断开

void CFTPClientDlg::OnButtonConnect()

{

    // TODO: Add your control notification handler code here

    UpdateData(TRUE);

    if(!m_ftpConnection){

        if(m_str_address!=""){

            m_ftpConnection=m_internetSession.GetFtpConnection(m_str_address,m_str_username,m_str_password);

            if(m_ftpConnection){

                m_ftpConnection->GetCurrentDirectory(m_str_server_dir);

                UpdateData(FALSE);

                LoadFileList();

                m_btn_connect.SetWindowText("断开");

            }

        }

    }

    else{

        m_ftpConnection->Close();

        delete m_ftpConnection;

        m_ftpConnection=NULL;

 

        m_btn_connect.SetWindowText("连接");

        m_str_server_dir="";

        m_listbox_files.ResetContent();

        

        UpdateData(FALSE);

    }

}

（2）加载文件、目录信息

void CFTPClientDlg::LoadFileList()

{

    m_listbox_files.ResetContent();

    CFtpFileFind fileFind(m_ftpConnection);

    CString fileName;

    BOOL bMoreFiles=fileFind.FindFile();

    while(bMoreFiles)

    {

        bMoreFiles=fileFind.FindNextFile();

        fileName=fileFind.GetFileName();

        if(fileFind.IsDirectory()){

            fileName+=" <dir>";

        }

        m_listbox_files.AddString(fileName);    

    }

    fileFind.Close();

}

（3）下载文件

void CFTPClientDlg::OnButtonDownload()

{

    // TODO: Add your control notification handler code here

    UpdateData(TRUE);

    if(m_str_filename!=""){

        if(m_str_filename.Right(5)=="<dir>"){

            MessageBox("不能下载目录");

        }

        else{

            CFileDialog saveDialog(FALSE,NULL,m_str_filename);

            if(saveDialog.DoModal()==IDOK)

            {

                if(!m_ftpConnection->GetFile(m_str_filename,saveDialog.GetFileName()),FALSE)

                    MessageBox("文件下载失败！");

                else

                    MessageBox("成功下载 "+m_str_filename);

            }

        }

    }

}

 

五. 实验结果

服务器运行效果如下：

主界面，运行状态显示

 

账号列表，编辑用户信息

 

在线用户列表

客户端运行效果如下：

 

不足的是，本次设计的FTP客户端程序较简单，只是为了简单验证服务器的可用性，所以只实现了文件的下载功能，后期将会尝试加上文件上传、删除以及目录创建等功能。