CSDN博客

img swordchueng

How to build a COM to Send and Receive Multi File by ATL

发表于2002/4/25 9:07:00  607人阅读

How to build a COM to Send and Receive Multi File by ATL

 

CodeWorrior April 24,2002.

 

Introduction

 

The COM can send and receive Multi file from Server .

The project code has been built with VC++ 6.0 SP5 / ATL3.0 and tested on Win2K Professional.

 

Getting started

 

1.Create an ATL Project by useing ATL COM AppWizard ,name it as what you want,

    

       at the step 1 :choose Support MFC;

 

2.Insert New ATL object by menu/toolbar,select Control -->lite Control-->

              names:   short name-->TestThread

            attributes:   support ISupportErrorInfo

                       support Connection_Points

                       interface -->Dual

                             thread model -->Apartment

                Misc:  inVisible at runtime

      then click ok;

 

3. At classview ,right click CTestThread select Add Method,

          add three method and the .idl file can found code like following lines

   

 

                interface ITestThread : IDispatch

       {

              [id(1), helpstring("MultiThread Download File From Server")] HRESULT DownFile([in]BSTR bstrString);

              [id(2), helpstring("Show About")] HRESULT About();

              [id(3), helpstring("method SendFile")] HRESULT SendFile([in]BSTR bstrstring,[in]BSTR bstrFileString);

       }

 

4.In TestThread.cpp   add following codes:    

 

// TestThread.cpp : Implementation of CTestThread

 

#include "stdafx.h"

#include "ThreadTest.h"

#include "TestThread.h"

 

#include "AboutDlg.h"

 

/////////////////////////////////////////////////////////////////////////////

// CTestThread

 

////////////////////////////////////////////

//

//     全局

char* msg[10];

 

DWORD WINAPI ReceiveFile(LPVOID);

 

CSocket sockClient;

 

SOCKET_STREAM_FILE_INFO StreamFileInfo;

 

// 全局定义完毕

////////////////////////////////////////////

 

STDMETHODIMP CTestThread::DownFile(BSTR bstrString)

{

       AFX_MANAGE_STATE(AfxGetStaticModuleState())

 

       // TODO: Add your implementation code here

      

       ///////////////////////////////////////////////////////////

      //

      //     负责文件的下载工作,消息字符串在这里需要处理

      //

        ///////////////////////////////////////////////////////////

       USES_CONVERSION;  

     

      //分解消息串,并发送给服务器。

      LPTSTR lpMsg;

      lpMsg = OLE2T(bstrString);

      char* chMsgString;

        chMsgString = lpMsg;

     

      char* msg[10];

      int i = 0;

      char *Split;

        Split=strtok(chMsgString,":");

     

        strcpy(lpMsg,Split);

     

        while(Split!=NULL)

      {    

             Split=strtok(NULL,":");

             if(Split!=NULL)

             {

                    //MessageBox(Split,"TestSplitString",MB_OK);

                    i = i+1;

                    msg[i] = Split;

                    //AfxMessageBox(msg[i]);

             }

      }

 

       AfxSocketInit(NULL);

       sockClient.Create();

       if(!sockClient.Connect(lpMsg,1028))

       {

              AfxMessageBox("Connect to server Fail!");

              sockClient.Close();

              return S_FALSE;

       }

      

       try

       {    

              SOCKET_USERINFO UserInfo;             //// msg[]次序可能需要变动

                                   

              strcpy(UserInfo.UserName,msg[1]);

              strcpy(UserInfo.Database,msg[2]);

              strcpy(UserInfo.DocID,msg[3]);

              strcpy(UserInfo.EditMode,msg[4]);

              strcpy(UserInfo.DotName,msg[5]);

              strcpy(UserInfo.MultiUser,msg[6]);

              strcpy(UserInfo.AttID,msg[7]);

 

              sockClient.Send(&UserInfo,sizeof(SOCKET_USERINFO));

 

              //测试用代码

              //HRESULT hr= sockClient.Receive(&StreamFileInfo,

              //            sizeof(SOCKET_STREAM_FILE_INFO));

              //sockClient.Receive(&UserInfo,sizeof(SOCKET_USERINFO));

             

              while(1)

              {

                     HANDLE hThread;

                     DWORD dwThreadID;

                     hThread = CreateThread(NULL,0,ReceiveFile,0,0,&dwThreadID);

                     Sleep(5000);

                     ::WaitForSingleObject(hThread,200);

                     CloseHandle(hThread);

             

                     try

                     {

                            if(sockClient.Receive((void*)m_szBuffer,4,0))

                            {

                                   //////////////////////////////////////////

                                   //

                                   //       如果接收到信息,则比较,为"end"则结束while

                                   //  循环,否则继续循环创建线程接收文件

                                   //

                                   //////////////////////////////////////////

                                   strcpy(m_szFlag,"end");  

                                   if(m_szBuffer == m_szFlag)

                                   {

                                          break;           //Receive Data =="end" ,Exit While Block

                                   }

                            }

                            else

                                   ///////////////////////////////////////

                                   //

                                   //       如果没有接收到任何信息,退出循环

                                   //    

                                   //////////////////////////////////////

                                   break;  

                     }

                     catch(...)

                     {

                            break;   //while Catch Error ,break;

                     }

 

              }

 

              sockClient.Close();

       }

      

       catch( ... )

       {

              ATLTRACE(_T("Test"));

              sockClient.Close();

              return S_FALSE;

       }

       ::SysFreeString(bstrString);

       return S_OK;

}

 

DWORD WINAPI ReceiveFile(LPVOID p)

{

       CoInitializeEx(0,COINIT_MULTITHREADED);

       ////////////////////////////

       //   以下代码经过测试可以正常接受文件

       //

 

       sockClient.Receive(&StreamFileInfo,sizeof(SOCKET_STREAM_FILE_INFO));

 

       ///////////////////////////////////////////////////////////////

       //

       //     添加代码,判断接收到的文件名是否含有".JPG",有则设置你

       //  保存目录为 "d:/imgtemp/" ,不是则保存为 "c:/info21/"

       //  

       ///////////////////////////////////////////////////////////////

 

       ::CreateDirectory(_T("C://Info21//"),NULL);

       ::SetCurrentDirectory(_T("C://Info21//"));         //设置下载文件的目录

      

       try

       {                         

              //AfxMessageBox(StreamFileInfo.szFileTitle);

              CFile destFile(StreamFileInfo.szFileTitle ,//设置下载后的文件名

                     CFile::modeCreate | CFile::modeWrite | CFile::typeBinary);

              UINT dwRead = 0;                           //接收文件

              while(dwRead<StreamFileInfo.nFileSizeLow)

              {

                     byte* data = new byte[1024];           //1k blocks to the buffer

                     memset(data,0,1024);

                     UINT dw=sockClient.Receive(data, 1024);

                     destFile.Write(data, dw);

                     dwRead+=dw;

              }

             

              SetFileTime((HANDLE)destFile.m_hFile,&StreamFileInfo.ftCreationTime,

                     &StreamFileInfo.ftLastAccessTime,&StreamFileInfo.ftLastWriteTime);

              SetFileAttributes(StreamFileInfo.szFileTitle,StreamFileInfo.dwFileAttributes);

             

              destFile.Close();                            // One File Receive over.

       }

       catch(...)

       {

           MessageBox(NULL,_T("Receive File Error!! And Socket Will DisConnect!!"),_T("Receive File"),       MB_ICONINFORMATION|MB_OK);

              sockClient.Close();

              ////AfxMessageBox("文件下载失败!");

              return S_FALSE;

       }

       MessageBox(NULL,_T("File All Received Complete!!"),_T("Receive File"),MB_OK);

       CoUninitialize();

       return 0;

}

 

STDMETHODIMP CTestThread::About()

{

       AFX_MANAGE_STATE(AfxGetStaticModuleState())

 

       // TODO: Add your implementation code here

      

       ///////////////////////////////////////////////////

       //

       //   Show  About

       //

       ///////////////////////////////////////////////////

 

       CAboutDlg dlg;

       dlg.DoModal();

       return S_OK;

 

}

 

STDMETHODIMP CTestThread::SendFile(BSTR bstrString ,BSTR bstrFileString)

{

       AFX_MANAGE_STATE(AfxGetStaticModuleState())

 

       // TODO: Add your implementation code here

             

       /////////////////////////////////////////////////////

       //     发送文件到服务器

       //  参数为  消息串(bstrString),文件路径(bstrFileString)

       ///////////////////////////////////////////////////////

       USES_CONVERSION;

      

       LPTSTR lpMsg;

      lpMsg = OLE2T(bstrString);

      char* chMsgString;

        chMsgString = lpMsg;

     

      char* msg[10];

      int i = 0;

      char *Split;

        Split=strtok(chMsgString,":");                            ////分解消息串

     

        strcpy(lpMsg,Split);

     

        while(Split!=NULL)

      {    

             Split=strtok(NULL,":");

             if(Split!=NULL)

             {

                    //MessageBox(Split,"TestSplitString",MB_OK);

                    i = i+1;

                    msg[i] = Split;

                    //AfxMessageBox(msg[i]);

             }

      }

 

       AfxSocketInit(NULL);

       sockClient.Create();

       if(!sockClient.Connect(lpMsg,1028))

       {

              AfxMessageBox("Connect to server Fail!");

              sockClient.Close();

              return S_FALSE;

       }

      

       SOCKET_USERINFO UserInfo;             //// msg[]次序可能需要变动

                                               ////与服务器协定

       strcpy(UserInfo.UserName,msg[1]);

       strcpy(UserInfo.Database,msg[2]);

       strcpy(UserInfo.DocID,msg[3]);

       strcpy(UserInfo.EditMode,msg[4]);

       strcpy(UserInfo.DotName,msg[5]);

       strcpy(UserInfo.MultiUser,msg[6]);

       strcpy(UserInfo.AttID,msg[7]);

      

       sockClient.Send(&UserInfo,sizeof(SOCKET_USERINFO));

 

       //::SetCurrentDirectory(bstrFileString);

 

       Recurse(bstrFileString);           //搜索指定目录下的文件并传送到服务器

 

       sockClient.Close();

       ::SysFreeString(bstrString);

       ::SysFreeString(bstrFileString);

       return S_OK;

}

 

void CTestThread::Recurse(BSTR  FilePath)

{

   USES_CONVERSION;

   CFileFind finder;

 

   BSTR pstr = FilePath; 

  

       //查找条件

   CString strWildcard(pstr);

   strWildcard += _T("//*.*");

 

   BOOL bWorking = finder.FindFile(strWildcard);

  

   while (bWorking)

   {

      bWorking = finder.FindNextFile();

 

      //

      // 输出文件名

         CString strFilename = finder.GetFilePath();

        

        

         if(strFilename!= ".")

         {

                if(strFilename!="..")

                {

                    

                     SendFile(T2OLE(strFilename));                   

                }

          }

        

   }

 

   finder.Close();

   ::SysFreeString(FilePath);

}

 

 

HRESULT CTestThread::SendFile(BSTR DocFile)

{

       USES_CONVERSION;

       char* filename = OLE2T(DocFile);

 

       CFile myFile;

       if(!myFile.Open( filename ,/* conversion BSTR to char* */

                                   CFile::modeRead | CFile::typeBinary))

       {

              //文件打开失败直接返回。在 " . " and " .. " 两个目录文件的情况下出现。

              return 0;

       }

 

       SOCKET_STREAM_FILE_INFO              StreamFileInfo;

       WIN32_FIND_DATA             FindFileData;

 

    FindClose(FindFirstFile(myFile.GetFilePath(),&FindFileData));

    memset(&StreamFileInfo,0,sizeof(SOCKET_STREAM_FILE_INFO));

    strcpy(StreamFileInfo.szFileTitle,myFile.GetFileName());

 

    StreamFileInfo.nFileSizeLow         =       FindFileData.nFileSizeLow;     

 

       sockClient.Send(&StreamFileInfo,sizeof(SOCKET_STREAM_FILE_INFO));    

 

       UINT dwRead=0;

       while(dwRead<StreamFileInfo.nFileSizeLow)

       {

              byte* data = new byte[5120];

              UINT dw=myFile.Read(data, 5120);

              sockClient.Send(data, dw);

              dwRead+=dw;

       }

       myFile.Close();

      

       ::SysFreeString(DocFile);

       return S_OK;

}

 

for the About method ,you must add a dialog and named AboutDlg,ok.

 

User Define Struct must be add to TestThread.h file:

 

typedef struct _SOCKET_STREAM_FILE_INFO {

 

    TCHAR       szFileTitle[128];                  

DWORD       dwFileAttributes;                 

FILETIME    ftCreationTime;                    

FILETIME    ftLastAccessTime;                 

FILETIME    ftLastWriteTime;                   

    DWORD       nFileSizeHigh;                     

    DWORD       nFileSizeLow;                      

    DWORD       dwReserved0;                       

    DWORD       dwReserved1;                    

 

} SOCKET_STREAM_FILE_INFO, * PSOCKET_STREAM_FILE_INFO;

 

typedef struct _SOCKET_USERINFO

{

       TCHAR       UserName[128];

       TCHAR   Database[128];

       TCHAR   DocID[256];

       TCHAR   AttID[256];

       TCHAR   DotName[128];

       TCHAR   MultiUser[128];

       TCHAR   EditMode[128];

       TCHAR   UserTime[128];

}SOCKET_USERINFO, * PSOCKET_USERINFO;

 

following two variable must be add to CTestThread class

       char m_szBuffer[4];

       char m_szFlag[4] ;

 

add #include "afxsock.h" to StdAfx.h file

 

Happy programming. :-)

Good Luck!!!  Contract me ,please mail to : CodeWorrior@hotmail.com.

If you need  Source Code ,Give me your e_mail address !!

0 0

相关博文

我的热门文章

img
取 消
img