开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > Ftp客户端 > 查看源码
FtpLib.cpp
资源名称：FTP总集.rar　[点击查看]
上传用户：tjfeida
上传日期：2013-03-10
资源大小：1917k
文件大小：28k
源码类别：

Ftp客户端

开发平台：

Visual C++

FtpLib.cpp：源码内容
  1. // Ftp.cpp: implementation of the CFtp class.
  2. //
  3. //////////////////////////////////////////////////////////////////////
  4. /*********************************************
  5. **该文件是属于WolfFTP工程中的。如果有什么问题
  6. **请联系
  7. **         tablejiang@21cn.com
  8. **或者访问
  9. **         http://wolfftp.51.net
  10. **以得到最新的支持。
  11. *********************************************/
  12. #include "stdafx.h"
  13. #include "FtpLib.h"
  15. #ifdef _DEBUG
  16. #undef THIS_FILE
  17. static char THIS_FILE[]=__FILE__;
  18. #define new DEBUG_NEW
  19. #endif
  21. //////////////////////////////////////////////////////////////////////
  22. // Construction/Destruction
  23. //////////////////////////////////////////////////////////////////////
  25. CFtpLib::CFtpLib()
  26. {
  27. m_bRun = true ;
  28. }
  30. CFtpLib::~CFtpLib()
  31. {
  33. }
  34. /***********************************************************
  35. ** @Description:
  36. ** connect the ftp server .
  37. ** @Note:
  38. ** LPCTSTR lpszHostName : the host address .
  39. **
  40. ** @Return: the connected socket handle .
  41. ** @Author: Table.JHM.太子
  42. ** @e-mail: tablejiang@21cn.com
  43. ** Date:     2001 3 26
  44. *************************************************************/
  45. SOCKET CFtpLib::ConnectServer( LPCTSTR lpszHostName )
  46. {
  47. LPHOSTENT lpHostent ;
  48. LPSERVENT lpServent ;
  49. SOCKADDR_IN sockAddr ;
  50. WORD nPort ;
  51. int bConnect ;
  52. int iWinsockErr ;
  53. SOCKET hConnectSocket ;
  55. //get host ip address .
  56. if( !(lpHostent = gethostbyname( lpszHostName ) ) )
  57. {
  58. iWinsockErr = WSAGetLastError( ) ;
  59. sprintf( m_szFtpBuffer , " %d error while get server ip addr!" , iWinsockErr ) ;
  60. //TRACE( NULL , m_szFtpBuffer , MSGBOX_TITLE , MB_OK ) ;
  61. return INVALID_SOCKET ;
  62. }
  64. if( ( hConnectSocket = socket( AF_INET , SOCK_STREAM , 0 ) ) == INVALID_SOCKET )
  65. {
  66. iWinsockErr = WSAGetLastError() ;
  67. sprintf( m_szFtpBuffer , " %d error while create connect socket!" , iWinsockErr )  ;
  68. //TRACE( NULL , m_szFtpBuffer , MSGBOX_TITLE , MB_OK ) ;
  69. return INVALID_SOCKET ;
  70. }
  71. //get default ftp host port .
  72. lpServent = getservbyname( "ftp" , 0 ) ;
  74. if( lpServent == NULL )
  75. nPort = htons( DEFAULT_FTP_PORT ) ;
  76. else
  77. nPort = lpServent->s_port ;
  79. //set server address .
  80. sockAddr.sin_family = AF_INET ;
  81. sockAddr.sin_port = nPort ;
  82. sockAddr.sin_addr = *((LPIN_ADDR)*lpHostent->h_addr_list ) ;
  84. bConnect = connect( hConnectSocket , (LPSOCKADDR)&sockAddr , sizeof( SOCKADDR_IN ) ) ;
  86. if( bConnect == SOCKET_ERROR ) 
  87. {
  88. iWinsockErr = WSAGetLastError( ) ;
  89. sprintf( m_szFtpBuffer , "%d error while connect server !" , iWinsockErr ) ;
  90. //TRACE( NULL , m_szFtpBuffer , MSGBOX_TITLE , MB_OK ) ;
  91. return INVALID_SOCKET ;
  92. }
  94. //check the socket is ready to read .
  95. //IsReadyToRead( hConnectSocket ) ;
  97. //get the server reply code .
  98. UINT nRelpyCode = GetServerReply( hConnectSocket ) ;
  99. if( nRelpyCode >= 400 || nRelpyCode <= 0 )
  100. {
  101. //closesocket( hConnectSocket ) ;
  102. return hConnectSocket ;
  103. }
  104. else
  105. {
  106. return hConnectSocket ;
  107. }
  108. }
  109. /***********************************************************
  110. ** @Description:
  111. ** check the socket connect is ready to read ?
  112. **
  113. ** @Note:
  114. ** SOCKET hSocket : socket handle want be check .
  115. **
  116. ** @Return: if ready to read true ,else false .
  117. ** @Author: Table.JHM.太子
  118. ** @e-mail: tablejiang@21cn.com
  119. ** Date: 2001 3 26
  120. *************************************************************/
  121. BOOL CFtpLib::IsReadyToRead(SOCKET hSocket)
  122. {
  123. FD_SET ReadSet ;
  124. TIMEVAL timeout ;
  125. int nReady ;
  126. int iWinsockErr ;
  128. //set time.
  129. //wait one second .
  130. timeout.tv_sec = 1 ;
  131. timeout.tv_usec = 0 ;
  133. FD_ZERO( &ReadSet ) ;
  134. FD_SET( hSocket , &ReadSet ) ;
  136. //select the socket .
  137. nReady = select( 0 , &ReadSet , 0 , 0 , &timeout ) ;
  139. if( nReady == SOCKET_ERROR )
  140. {
  141. iWinsockErr = WSAGetLastError( ) ;
  142. sprintf( m_szFtpBuffer , " %d error while select function read !" , iWinsockErr ) ;
  143. //TRACE( NULL , m_szFtpBuffer , MSGBOX_TITLE , MB_OK ) ;
  144. return FALSE ;
  145. }
  147. //check socket .
  148. if( FD_ISSET( hSocket , &ReadSet ) )
  149. {
  150. return TRUE ;
  151. }
  152. else
  153. return FALSE ;
  154. }
  155. /***********************************************************
  156. ** @Description:
  157. ** receive the reply string from server .
  158. **
  159. ** @Note:
  160. ** SOCKET hSocket : the connected socket handle.
  161. **
  162. ** @Return: the server reply code.
  163. ** @Author: Table.JHM.太子
  164. ** @e-mail: tablejiang@21cn.com
  165. ** Date:  2001 3 26
  166. ***********************************************************/
  167. int CFtpLib::GetServerReply(SOCKET hSocket)
  168. {
  169. char szBuffer[1024] ;
  170. int iLength ;
  171. int iBuffer = 0 ;
  172. BOOL bNeedReceive = true ;
  173. char szMultiLineHead[MAX_PATH] ;
  174. UINT nReplyCode ;
  175. memset( szMultiLineHead , 0 ,MAX_PATH ) ;
  177. while( bNeedReceive )
  178. {
  179. memset( szBuffer , 0 , 1024 ) ;
  180. iBuffer = 0 ;
  182. //because the reply string possibly is multi line .
  183. //so we must peek the reply info before receive it.
  184. //peek the message .
  185. if( ( iLength = recv( hSocket , szBuffer , 1024 , MSG_PEEK ) ) == SOCKET_ERROR )
  186. {
  187. int iWinsockErr = WSAGetLastError( ) ;
  188. sprintf( m_szFtpBuffer , " %d error while recv server reply!" , iWinsockErr ) ;
  189. //TRACE( NULL , m_szFtpBuffer , MSGBOX_TITLE , MB_OK ) ;
  190. return ( ERROR_FTP_CODE ) ;
  191. }
  193. if( iLength <= 0 )
  194. return (ERROR_FTP_CODE );
  196. //debug
  197. //Log the file .
  198. /*
  199. CFile file ;
  200. file.Open( "e:\receive.txt" , CFile::modeWrite ) ;
  201. file.SeekToEnd( ) ;
  202. file.Write( szBuffer , iLength ) ;
  203. file.Close() ;
  204. */
  207. //find the line end .
  208. while( szBuffer[iBuffer] != 'n' )
  209. {
  210. iBuffer ++ ;
  211. if( iBuffer >= iLength )
  212. break ;
  213. }
  214. //skip the 'n'
  215. iBuffer ++ ;
  216. TRACE( "%drn" , iLength ) ;
  218. //receive server reply string .
  219. if( ( iLength = recv( hSocket , szBuffer , iBuffer , 0 ) ) == SOCKET_ERROR )
  220. {
  221. int iWinsockErr = WSAGetLastError( ) ;
  222. sprintf( m_szFtpBuffer , " %d error while recv server reply!" , iWinsockErr ) ;
  223. //TRACE( NULL , m_szFtpBuffer , MSGBOX_TITLE , MB_OK ) ;
  224. return ( ERROR_FTP_CODE ) ;
  225. }
  227. if( iLength < 1024 )
  228. szBuffer[iLength] = 0 ;
  229. else
  230. szBuffer[1023] = 0 ;
  232. //notify the parent wnd.
  233. SendMsgToParentWnd( szBuffer , false ) ;
  235. if( strlen( szMultiLineHead ) == 0 )
  236. {
  237. nReplyCode = GetReplyCode( szBuffer ) ;
  238. itoa( (int)nReplyCode , szMultiLineHead , 10 ) ;
  239. if( szBuffer[3] == '-' )
  240. bNeedReceive = true ;
  241. else
  242. bNeedReceive = false ;
  243. }
  244. else
  245. {
  246. if( strncmp( szMultiLineHead , szBuffer , 3 ) == 0 )
  247. {
  248. if( szBuffer[3] == '-' )
  249. bNeedReceive = true ;
  250. else
  251. bNeedReceive = false ;
  252. }
  253. else
  254. bNeedReceive = true ;
  255. }
  257. }
  259. nReplyCode = GetReplyCode ( szBuffer ) ;
  261. return nReplyCode ;
  262. }
  263. /***********************************************************
  264. ** @Description:
  265. ** Get server reply command code.
  266. **
  267. ** @Note:
  268. ** LPSTR lpszBuffer : the reply string .
  269. **
  270. ** @Return: server reply code.
  271. ** @Author: Table.JHM.太子
  272. ** @e-mail: tablejiang@21cn.com
  273. ** Date: 2001 3 26
  274. *************************************************************/
  275. UINT CFtpLib::GetReplyCode(LPSTR lpszBuffer)
  276. {
  277. //get the server reply code , 
  278. //the ftp server will return a string after receive command , 
  279. //first three char in this string is the server reply code .
  280. UINT nCode ;
  281. char cBak ;
  282. cBak = *(lpszBuffer + 3 );
  283. *( lpszBuffer + 3 ) = 0 ;
  285. nCode = atoi( lpszBuffer ) ;
  287. *(lpszBuffer + 3 ) = cBak ;
  289. return nCode ;
  291. }
  292. /***********************************************************
  293. ** @Description:
  294. ** send a command to ftp server ,and receive the reply
  295. **
  296. ** @Note:
  297. ** LPSTR szCommandBuffer : the command string 
  298. ** SOCKET hSocket : the connected socket handle.
  299. **
  300. ** @Return: 
  301. ** @Author: Table.JHM.太子
  302. ** @e-mail: tablejiang@21cn.com
  303. ** Date: 2001 3 26
  304. *************************************************************/
  305. UINT CFtpLib::SendFtpCommand(LPSTR szCommandBuffer , SOCKET hSocket )
  306. {
  308. if( !m_bRun )
  309. return ERROR_FTP_CODE ;
  311. //send command . and after send will check server reply .
  312. if( send( hSocket , szCommandBuffer , strlen( szCommandBuffer ) , 0 ) == SOCKET_ERROR ) 
  313. {
  314. int iWinsockErr = WSAGetLastError( ) ;
  315. sprintf( m_szFtpBuffer , " %d error while Send ftp command !" , iWinsockErr ) ;
  316. //TRACE( NULL , m_szFtpBuffer , MSGBOX_TITLE , MB_OK ) ;
  317. return ( ERROR_FTP_CODE ) ;
  318. }
  320. //Send msg to window .
  321. SendMsgToParentWnd( szCommandBuffer , true ) ;
  323. return GetServerReply( hSocket ) ;
  325. }
  326. /***********************************************************
  327. ** @Description:
  328. ** close the ftp control connect socket handle .
  329. *
  330. ** @Note:
  331. ** SOCKET hSocket : the connected socket handle .
  332. **
  333. ** @Return: true .
  334. ** @Author: Table.JHM.太子
  335. ** @e-mail: tablejiang@21cn.com
  336. ** Date: 2001 3 26
  337. *************************************************************/
  338. BOOL CFtpLib::CloseFtpConnect( SOCKET hSocket )
  339. {
  340. closesocket( hSocket ) ;
  341. return TRUE ;
  342. }
  344. /***********************************************************
  345. ** @Description:
  346. ** Create the data link .
  347. **
  348. ** @Note:
  349. ** SOCKET hControlSocket : the control connected socket handle .
  350. ** @Return: if success return the data listen socket handle , 
  351. ** else return INVALID_SOCKET .
  352. **
  353. ** @Author: Table.JHM.太子
  354. ** @e-mail: tablejiang@21cn.com
  355. ** Date: 2001 3 26
  356. *************************************************************/
  357. //this function is used to create date link .
  358. SOCKET CFtpLib::CreateListenCannel(SOCKET hControlSocket)
  359. {
  360. SOCKADDR_IN sockaddr ;
  361. SOCKET hListenSocket ;
  363. if( ( hListenSocket = socket( AF_INET , SOCK_STREAM , 0 ) ) == INVALID_SOCKET )
  364. {
  365. int iWinsockErr = WSAGetLastError( ) ;
  366. sprintf( m_szFtpBuffer , "%d error when create listen socket !" , iWinsockErr ) ;
  367. //TRACE( NULL , m_szFtpBuffer , MSGBOX_TITLE , MB_OK ) ;
  368. return ( INVALID_SOCKET ) ;
  369. }
  371. //make address.
  372. sockaddr.sin_family = AF_INET ;
  373. sockaddr.sin_port = htons( 0 ) ;
  374. sockaddr.sin_addr.s_addr = INADDR_ANY ;
  376. if( bind( hListenSocket , (LPSOCKADDR)&sockaddr , sizeof( sockaddr ) ) )
  377. {
  378. int iWinsockErr = WSAGetLastError( ) ;
  379. sprintf( m_szFtpBuffer , "%d error when bind the listen socket !" , iWinsockErr ) ;
  380. //TRACE( NULL , m_szFtpBuffer , MSGBOX_TITLE , MB_OK ) ;
  381. return ( INVALID_SOCKET ) ;
  382. }
  384. if( listen( hListenSocket , 1 ) )
  385. {
  386. int iWinsockErr = WSAGetLastError( ) ;
  387. sprintf( m_szFtpBuffer , "%d error when listening " , iWinsockErr ) ;
  388. //TRACE( NULL , m_szFtpBuffer , MSGBOX_TITLE , MB_OK ) ;
  389. return ( INVALID_SOCKET ) ;
  390. }
  391. return PrepareDataCannel( hListenSocket , hControlSocket )  ;
  393. }
  394. /***********************************************************
  395. ** @Description:
  396. ** prepare the data cannel , transmit the data.
  397. **
  398. ** @Note:
  399. ** SOCKET hListenSocket : Listen data socket hanle .
  400. ** SOCKET hControlSocket : Control socket handle .
  401. **
  402. ** @Return: the data link connect socket .
  403. ** @Author: Table.JHM.太子
  404. ** @e-mail: tablejiang@21cn.com
  405. ** Date:  2001 3 26
  406. *************************************************************/
  407. SOCKET CFtpLib::PrepareDataCannel(SOCKET hListenSocket, SOCKET hControlSocket)
  408. {
  409. SOCKADDR_IN sockaddr ;
  410. int iLength ;
  411. UINT nPort ;
  412. UINT nReplyCode ;
  414. iLength = sizeof( sockaddr ) ;
  416. // Get the address for the hListenSocket
  417. if( getsockname( hListenSocket , (LPSOCKADDR)&sockaddr , &iLength ) == SOCKET_ERROR )
  418. {
  419. int iWinsockErr = WSAGetLastError( ) ;
  420. sprintf( m_szFtpBuffer , "%d error when get listen socket name!" , iWinsockErr ) ;
  421. //TRACE( NULL , m_szFtpBuffer , MSGBOX_TITLE , MB_OK ) ;
  422. return ( INVALID_SOCKET ) ;
  423. }
  425. nPort = sockaddr.sin_port ;
  426. iLength = sizeof( sockaddr ) ;
  428. if( getsockname( hControlSocket , ( LPSOCKADDR )&sockaddr , &iLength ) == SOCKET_ERROR )
  429. {
  430. int iWinsockErr = WSAGetLastError( ) ;
  431. sprintf( m_szFtpBuffer , "%d error when get control socket name!" , iWinsockErr ) ;
  432. //TRACE( NULL , m_szFtpBuffer , MSGBOX_TITLE , MB_OK ) ;
  433. return ( INVALID_SOCKET ) ;
  434. }
  435. //use PORT command to tell server connect this socket , 
  436. //after server receive this command , server will create a socket to connect 
  437. //this port .
  438. char szCommand[MAX_PATH];
  439. sprintf( szCommand , "PORT %d,%d,%d,%d,%d,%drn" , 
  440. sockaddr.sin_addr.S_un.S_un_b.s_b1,
  441. sockaddr.sin_addr.S_un.S_un_b.s_b2,
  442. sockaddr.sin_addr.S_un.S_un_b.s_b3,
  443. sockaddr.sin_addr.S_un.S_un_b.s_b4,
  444. nPort&0xFF , 
  445. nPort >> 8 );
  447. //send command .
  448. if( ( nReplyCode = SendFtpCommand( szCommand , hControlSocket ) ) != 200 )
  449. {
  450. int iWinsockErr = WSAGetLastError( ) ;
  451. sprintf( m_szFtpBuffer , "%d error when Send ftp command !" , iWinsockErr ) ;
  452. //TRACE( NULL , m_szFtpBuffer , MSGBOX_TITLE , MB_OK ) ;
  453. return ( INVALID_SOCKET ) ;
  454. }
  455. else
  456. return ( hListenSocket ) ;
  457. }
  458. /***********************************************************
  459. ** @Description:
  460. ** Accept data connect socket .
  461. **
  462. ** @Note:
  463. ** SOCKET hListenSocket : the data listen socket handle .
  464. **
  465. ** @Return: 
  466. ** @Author: Table.JHM.太子
  467. ** @e-mail: tablejiang@21cn.com
  468. ** Date: 2001 3 26
  469. *************************************************************/
  470. SOCKET CFtpLib::AcceptDataConnect(SOCKET hListenSocket)
  471. {
  472. SOCKET hDataConnect = 0 ;
  473. SOCKADDR_IN sockaddr ;
  474. int iLength ;
  476. iLength = sizeof( sockaddr ) ;
  478. //accept the data socket .
  479. hDataConnect = accept( hListenSocket , (LPSOCKADDR)&sockaddr , &iLength ) ;
  481. if( hDataConnect == INVALID_SOCKET )
  482. {
  483. int iWinsockErr = WSAGetLastError( ) ;
  484. sprintf( m_szFtpBuffer , "%d error when accept data connect !" , iWinsockErr ) ;
  485. //TRACE( NULL , m_szFtpBuffer , MSGBOX_TITLE , MB_OK ) ;
  486. return ( INVALID_SOCKET ) ;
  487. }
  489. return ( hDataConnect ) ;
  491. }
  492. /***********************************************************
  493. ** @Description:
  494. **
  495. ** @Note:
  496. **
  497. ** @Return: 
  498. ** @Author: Table.JHM.太子
  499. ** @e-mail: tablejiang@21cn.com
  500. ** Date: 2001 3 26
  501. *************************************************************/
  502. UINT CFtpLib::ReadDataCannel(SOCKET hControlSocket, SOCKET hDataSocket, char *Buffer)
  503. {
  504. return 1;
  505. }
  506. /***********************************************************
  507. ** @Description:
  508. ** read the data from data connect and , wirte it to 
  509. ** local file .
  510. **
  511. ** @Note:
  512. ** SOCKET hControlSocket : the control socket handle .
  513. ** SOCKET hDataSocket : the data connect socket handle .
  514. ** HANDLE hFile : the write file handle.
  515. ** __int64 iFileSize : file transmit file size .
  516. **
  517. ** @Return: after transmit complete , the server return code 
  518. ** @Author: Table.JHM.太子
  519. ** @e-mail: tablejiang@21cn.com
  520. ** Date: 2001 3 26
  521. *************************************************************/
  522. UINT CFtpLib::ReadDataCannelToFile( SOCKET hControlSocket, SOCKET hDataSocket, HANDLE hFile , __int64 iFileSize )
  523. {
  524. DWORD nByteRec = 0 ;
  525. __int64 lDataNum = 0 ;
  526. char ReceiveBuffer[2048] ;
  527. m_hDataSocket = hDataSocket ;
  528. int iProgress = 0 ;
  529. int iSpeed = 0 ;
  531. //get start time .
  532. DWORD dwStart = GetTickCount( ) ;
  533. DWORD dwEnd = dwStart ;
  535. //receive data from net , and write it to local file .
  536. do
  537. {
  538. //receive data.
  539. nByteRec = recv( hDataSocket , ReceiveBuffer , 2048 , 0 ) ;
  540. if( nByteRec == SOCKET_ERROR )
  541. break ;
  543. if( nByteRec > 0 )
  544. {
  545. //total receive bytes.
  546. lDataNum += nByteRec ;
  547. if( !WriteFile( hFile , ReceiveBuffer , nByteRec , &nByteRec , NULL ) )
  548. {
  549. break ;
  550. }
  551. //make progress percent.
  552. if( iFileSize != 0 )
  553. iProgress = ( int )( lDataNum * 100 / iFileSize ) ;
  554. else
  555. iProgress = 100 ;
  556. //get end time .
  557. dwEnd = GetTickCount() ;
  558. //get transmit speed .
  559. iSpeed = GetTransmitSpeed( dwStart , dwEnd , lDataNum ) ;
  560. ::PostMessage( m_hParentWnd , FTP_TRANSMIT_DATA_MSG , iProgress , iSpeed ) ;
  562. }
  564. }
  565. while( nByteRec > 0  );
  567. //close data socket handle .
  568. closesocket( hDataSocket ) ;
  569. m_hDataSocket = NULL ;
  571. if( nByteRec == SOCKET_ERROR )
  572. return false ;
  574. return ( GetServerReply( hControlSocket ) ) ;
  576. }
  577. /***********************************************************
  578. ** @Description:
  579. ** global function , use for initialize socket lib .
  580. **
  581. ** @Note:
  582. ** void
  583. **
  584. ** @Return: success return true ,else false.
  585. ** @Author: Table.JHM.太子
  586. ** @e-mail: tablejiang@21cn.com
  587. ** Date: 2001 3 26
  588. *************************************************************/
  589. BOOL InitSocket( )
  590. {
  591. WSADATA wsadata ;
  592. if( WSAStartup( MAKEWORD( 1 , 1 ) , &wsadata ) )
  593. {
  594. return false ;
  595. }
  596. else
  597. {
  598. return true ;
  599. }
  600. }
  601. /***********************************************************
  602. ** @Description:
  603. ** global function , uninit the socket lib.
  604. **
  605. ** @Note:
  606. ** void .
  607. **
  608. ** @Return: success return true ,else return false .
  609. ** @Author: Table.JHM.太子
  610. ** @e-mail: tablejiang@21cn.com
  611. ** Date: 2001 3 26
  612. *************************************************************/
  613. BOOL UnInitSocket( )
  614. {
  615. if( WSACleanup( ) )
  616. {
  617. return false ;
  618. }
  619. else
  620. return true ;
  621. }
  623. /***********************************************************
  624. ** @Description:
  625. ** upload a file to server .
  626. **
  627. ** @Note:
  628. ** SOCKET hControlSocket : connected control socket .
  629. ** SOCKET hDataSocket : connected data socket .
  630. ** HANDLE hFile : the open file handle .
  631. **
  632. ** @Return: 
  633. ** @Author: Table.JHM.太子
  634. ** @e-mail: tablejiang@21cn.com
  635. ** Date: 2001 3 26
  636. *************************************************************/
  637. UINT CFtpLib::WriteDataCannelFromFile(SOCKET hControlSocket, SOCKET hDataSocket, HANDLE hFile )
  638. {
  640. DWORD nByteRec = 0 ;
  641. DWORD nByteSend = 0 ;
  643. char SendBuffer[RECEIVE_SIZE] ;
  645. //backup data socket handle .
  646. m_hDataSocket = hDataSocket ;
  648. __int64 nFileSize = 0 ;
  649. __int64 nSendSize = 0 ;
  650. DWORD dwFileSizeHigh = 0 ;
  651. DWORD dwFileSizeLow = 0 ;
  652. int iPrecent = 0 ;
  653. int iSpeed = 0 ;
  655. //get start time
  656. DWORD dwStart = GetTickCount( ) ;
  657. DWORD dwEnd = dwStart;
  659. //get file size .
  660. dwFileSizeLow = GetFileSize( hFile , &dwFileSizeHigh ) ;
  661. nFileSize = dwFileSizeHigh ;
  662. nFileSize = dwFileSizeLow + (nFileSize << 32 ) ;
  664. //transmit data.
  665. do
  666. {
  667. //read data from file.
  668. if( ReadFile( hFile , SendBuffer , RECEIVE_SIZE , &nByteRec , NULL ) )
  669. {
  670. if( nByteRec > 0 )
  671. {
  672. //send data.
  673. nByteSend = send( hDataSocket , SendBuffer , nByteRec , 0 ) ;
  674. if( nByteSend <= 0 || nByteSend == SOCKET_ERROR )
  675. {
  676. break ;
  677. }
  678. //total transmit bytes
  679. nSendSize += nByteRec ;
  680. //get end time .
  681. dwEnd = GetTickCount() ;
  682. //get transmit precent.
  683. iPrecent = (int) ( nSendSize * 100 / nFileSize );
  684. //get transmit speed .
  685. iSpeed = GetTransmitSpeed( dwStart , dwEnd , nSendSize ) ;
  686. ::PostMessage( m_hParentWnd , FTP_TRANSMIT_DATA_MSG , iPrecent , iSpeed ) ;
  687. //dwStart = GetTickCount( ) ;
  688. }
  689. else
  690. break ;
  691. }
  692. }
  693. while( nByteRec > 0 );
  695. //close handle .
  696. closesocket( hDataSocket ) ;
  697. m_hDataSocket = NULL ;
  699. if( nByteSend == SOCKET_ERROR )
  700. return false ;
  702. return ( GetServerReply( hControlSocket ) ) ;
  704. }
  705. /***********************************************************
  706. ** @Description:
  707. ** this function is parse the server file list info.
  708. ** before you can understand this function ,you must 
  709. ** know the ftp server reply file list format.
  710. **
  711. ** @Note:
  712. ** int * pos[in/out] : the pointer position of file list string.
  713. ** char* word[out] : the buffer use for store current word.
  714. ** int iwordsize[in] : the buffer size .
  715. ** CString* string[in] : the file list string .
  716. **
  717. ** @Return: 
  718. ** @Author: Table.JHM.太子
  719. ** @e-mail: tablejiang@21cn.com
  720. ** Date: 2001 3 26
  721. *************************************************************/
  722. UINT CFtpLib::GetSubItem(int *pos, char *word , int iwordsize , CString* string )
  723. {
  724. int iPos = *pos ;
  725. int iLength = string->GetLength() ;
  727. //zero memory 
  728. memset( word , 0 , iwordsize ) ;
  730. //if pos overtop the string length .
  731. if( iPos >= ( iLength - 1 ) )
  732. return 2 ;
  734. //skip the space char .
  735. while( string->GetAt( iPos ) == ' ' )
  736. {
  737. iPos ++ ;
  739. if( iPos >= ( iLength - 1 ) || string->GetAt( iPos ) == 'r' )
  740. return 0 ;
  741. }
  742. int iWord = 0 ;
  744. //get word string .
  745. while( !( string->GetAt( iPos ) == ' ' || string->GetAt( iPos ) == 'r' &&
  746. string->GetAt( iPos + 1 ) == 'n' ) )
  747. {
  748. word[iWord++] = string->GetAt( iPos++ ) ;
  749. if( iPos >= iLength - 1 )
  750. return 2 ;
  751. }
  753. //if a word complete.
  754. if( string->GetAt( iPos ) == ' ' )
  755. {
  756. *pos = iPos ;
  757. return 1 ;
  758. }
  759. //if a line complete .
  760. if( string->GetAt( iPos ) == 'r' )
  761. {
  762. iPos += 2 ;
  763. *pos = iPos ;
  764. return 2 ;
  765. }
  767. return 0 ;
  769. }
  770. /***********************************************************
  771. ** @Description:
  772. ** send message string to parent window .
  773. **
  774. ** @Note:
  775. ** LPSTR szMsg : the message string .
  776. ** BOOL bCommand : is command or server reply .
  777. **
  778. ** @Return: true .
  779. ** @Author: Table.JHM.太子
  780. ** @e-mail: tablejiang@21cn.com
  781. ** Date: 2001 3 26
  782. *************************************************************/
  783. BOOL CFtpLib::SendMsgToParentWnd(LPSTR szMsg, BOOL bCommand)
  784. {
  785. int iLength = 0 ;
  786. if( bCommand )
  787. {
  788. char* szCommandString ;
  789. memset( m_szFtpBuffer , 0 , sizeof( m_szFtpBuffer ) ) ;
  791. if( strncmp( szMsg , "PASS " , 5 ) == 0 )
  792. {
  793. szMsg[5] = '*' ;
  794. szMsg[6] = '' ;
  795. strcat( szMsg , "rn" ) ;
  796. }
  798. sprintf( m_szFtpBuffer , "COMMAND >   %s" , szMsg ) ;
  799. int iLength = strlen( m_szFtpBuffer ) ;
  800. //new the buffer , this buffer will delete in parent message response function.
  801. szCommandString = new char[iLength + 1] ;
  802. strcpy( szCommandString , m_szFtpBuffer ) ;
  803. szCommandString[iLength] = 0 ;
  804. ::PostMessage( m_hWndInfo.hInfoWnd , FTP_SEND_COMMAND_MSG , 0 , (LPARAM)szCommandString ) ;
  805. }
  806. else
  807. {
  808. char* szReplyString ;
  809. memset( m_szFtpBuffer , 0 , sizeof( m_szFtpBuffer ) ) ;
  810. sprintf( m_szFtpBuffer , "                        %s" , szMsg ) ;
  811. iLength = strlen( m_szFtpBuffer ) ;
  812. //new the buffer , this buffer will delete in parent message response function.
  813. szReplyString = new char[iLength + 1] ;
  814. strcpy( szReplyString , m_szFtpBuffer ) ;
  815. szReplyString[iLength] = 0 ;
  816. ::PostMessage( m_hWndInfo.hInfoWnd , FTP_SERVER_REPLY_MSG , 0 , (LPARAM)szReplyString ) ;
  817. }
  818. return true ;
  820. }
  821. /***********************************************************
  822. ** @Description:
  823. ** user login ftp server.
  824. **
  825. ** @Note:
  826. ** SOCKET hSocket : the connect control socket handle .
  827. ** LPSTR szName : user name .
  828. ** LPSTR szPass : pass word .
  829. **
  830. ** @Return: login success return true ,else return false .
  831. ** @Author: Table.JHM.太子
  832. ** @e-mail: tablejiang@21cn.com
  833. ** Date: 2001 3 26
  834. *************************************************************/
  835. BOOL CFtpLib::UserLogin(SOCKET hSocket, LPSTR szName, LPSTR szPass)
  836. {
  837. char szCommand[MAX_PATH] ;
  838. UINT nReplyCode = 0 ;
  839. //send user name .
  840. sprintf( szCommand , "USER %srn" , szName ) ;
  841. nReplyCode = SendFtpCommand( szCommand , hSocket ) ;
  842. if( nReplyCode >= 400 )
  843. return false ;
  844. //send password 
  845. sprintf( szCommand , "PASS %srn" , szPass ) ;
  846. nReplyCode = SendFtpCommand( szCommand , hSocket ) ;
  847. if( nReplyCode >= 400 )
  848. return false ;
  850. return true ;
  851. }
  852. /***********************************************************
  853. ** @Description:
  854. ** anonymous login function .this function will use 
  855. ** anonymous as user name to login ftp server.
  856. **
  857. ** @Note:
  858. ** SOCKET hSocket : the connect control socket handle .
  859. **
  860. ** @Return: 
  861. ** @Author: Table.JHM.太子
  862. ** @e-mail: tablejiang@21cn.com
  863. ** Date: 2001 3 26
  864. *************************************************************/
  865. BOOL CFtpLib::AnonymousLogin(SOCKET hSocket)
  866. {
  867. //login
  868. char szCommand[MAX_PATH] ;
  869. UINT nReplyCode = 0 ;
  871. sprintf( szCommand , "USER %srn" , "anonymous" ) ;
  872. nReplyCode = SendFtpCommand( szCommand , hSocket ) ;
  873. if( nReplyCode >= 400 )
  874. return false ;
  876. sprintf( szCommand , "PASS %srn" , "anonymous@table.com" ) ;
  877. nReplyCode = SendFtpCommand( szCommand , hSocket ) ;
  878. if( nReplyCode >= 400 )
  879. return false ;
  880. return true ;
  881. }
  882. /***********************************************************
  883. ** @Description:
  884. **
  885. ** @Note:
  886. **
  887. ** @Return: 
  888. ** @Author: Table.JHM.太子
  889. ** @e-mail: tablejiang@21cn.com
  890. ** Date: 2001 3 26
  891. *************************************************************/
  892. /*
  893. BOOL CFtpLib::SetTransmitFilePos(FTPFILEINFO *pFile , HANDLE hFile , SOCKET hSocket )
  894. {
  895. char szBuffer[MAX_PATH] ;
  896. sprintf( szBuffer , "SIZE %srn" , pFile->remotefilename ) ;
  897. //send command . and after send will check server reply .
  898. if( send( hSocket , szBuffer , strlen( szBuffer ) , 0 ) == SOCKET_ERROR ) 
  899. {
  900. int iWinsockErr = WSAGetLastError( ) ;
  901. sprintf( m_szFtpBuffer , " %d error while Send ftp command !" , iWinsockErr ) ;
  902. //TRACE( NULL , m_szFtpBuffer , MSGBOX_TITLE , MB_OK ) ;
  903. return ( ERROR_FTP_CODE ) ;
  904. }
  906. int iLength ;
  908. //peek the message .
  909. if( ( iLength = recv( hSocket , szBuffer , 1024 , MSG_PEEK ) ) == SOCKET_ERROR )
  910. {
  911. int iWinsockErr = WSAGetLastError( ) ;
  912. sprintf( m_szFtpBuffer , " %d error while recv server reply!" , iWinsockErr ) ;
  913. //TRACE( NULL , m_szFtpBuffer , MSGBOX_TITLE , MB_OK ) ;
  914. return ( ERROR_FTP_CODE ) ;
  915. }
  917. if( iLength <= 0 )
  918. return (ERROR_FTP_CODE );
  920. int iBuffer = 0 ;
  922. while( szBuffer[iBuffer] != 'n' )
  923. {
  924. iBuffer ++ ;
  925. if( iBuffer >= iLength )
  926. break ;
  927. }
  928. iBuffer ++ ;
  930. //receive server reply string .
  931. if( ( iLength = recv( hSocket , szBuffer , iBuffer , 0 ) ) == SOCKET_ERROR )
  932. {
  933. int iWinsockErr = WSAGetLastError( ) ;
  934. sprintf( m_szFtpBuffer , " %d error while recv server reply!" , iWinsockErr ) ;
  935. //TRACE( NULL , m_szFtpBuffer , MSGBOX_TITLE , MB_OK ) ;
  936. return ( ERROR_FTP_CODE ) ;
  937. }
  938. if( iLength < 1024 )
  939. szBuffer[iLength] = 0 ;
  940. else
  941. szBuffer[1023] = 0 ;
  943. UINT nReplyCode = GetReplyCode ( szBuffer ) ;
  944. if( nReplyCode >= 400 )
  945. return false ;
  947. char szFileSize[MAX_PATH] ;
  948. __int64 iRemoteFileSize ;
  949. __int64 iLocalFileSize ;
  950. DWORD lSize ;
  951. DWORD hSize ;
  953. strcpy( szFileSize , szBuffer + 4 ) ;
  954. iLength = strlen( szFileSize ) ;
  955. szFileSize[iLength - 2 ] = '' ;
  956. iRemoteFileSize = _atoi64( szFileSize ) ;
  957. lSize = GetFileSize( hFile , &hSize ) ;
  959. iLocalFileSize = hSize ;
  960. iLocalFileSize = iLocalFileSize << 32 ;
  961. iLocalFileSize = lSize ;
  963. if( pFile->bfileput )
  964. {
  965. if( iRemoteFileSize > iLocalFileSize )
  966. {
  967. SendFtpCommand( "RSET 0rn" , hSocket ) ;
  968. return true ;
  969. }
  970. else
  971. {
  973. return true ;
  974. }
  975. }
  976. else
  977. {
  978. if( iRemoteFileSize < iLocalFileSize )
  979. {
  980. SendFtpCommand( "RSET 0rn" , hSocket ) ;
  981. return true ;
  982. }
  983. else
  984. {
  985. return true ;
  986. }
  987. }
  988. return true ;
  989. }
  990. */
  991. /***********************************************************
  992. ** @Description:
  993. ** this function is use for seek file pointer .
  994. **
  995. ** @Note:
  996. ** HANDLE hFile : the open file handle .
  997. ** __int64 distance : the move distance.
  998. ** DWORD MoveMethod : move method , head ,end ??
  999. **
  1000. ** @Return: the file pointer moved distance 
  1001. ** @Author: Table.JHM.太子
  1002. ** @e-mail: tablejiang@21cn.com
  1003. ** Date: 2001 3 26
  1004. *************************************************************/
  1005. __int64 CFtpLib::MyFileSeek(HANDLE hf, __int64 distance, DWORD MoveMethod)
  1006. {
  1007. LARGE_INTEGER li;
  1009. li.QuadPart = distance;
  1011. SetLastError( NO_ERROR ) ;
  1012. //set file pointer.
  1013. li.LowPart = SetFilePointer (hf, li.LowPart, &li.HighPart, MoveMethod);
  1015. if (li.LowPart >= 0 && GetLastError() != NO_ERROR)
  1016. {
  1017. li.QuadPart = -1;
  1018. }
  1019. return li.QuadPart;
  1020. }
  1021. /****************************************************
  1022. ** @Description
  1023. ** this function is get transmit speed .
  1024. **
  1025. ** @Parameter
  1026. ** DWORD dwStart : the start time.
  1027. ** DWORD dwEnd : the end time .
  1028. ** __int64 recData : receive data bytes.
  1029. **
  1030. ** @Return:
  1031. ** @Author: Table.JHM.太子
  1032. ** @e-mail: tablejiang@21cn.com
  1033. ** @Date: 2001 3 26
  1034. ****************************************************/
  1035. int CFtpLib::GetTransmitSpeed( DWORD dwStart , DWORD dwEnd ,  __int64 recData)
  1036. {
  1037. if( dwStart >= dwEnd )
  1038. return 0 ;
  1039. int iSpeed = 0 ;
  1040. iSpeed = (int) ( recData * 1000 / ( dwEnd - dwStart ) ) ;
  1041. return iSpeed ;
  1042. }
  1043. /****************************************************
  1044. ** @Description
  1045. ** stop the command.don't let run.
  1046. **
  1047. ** @Parameter
  1048. ** BOOL bRun : 
  1049. **
  1050. ** @Return:
  1051. ** @Author: Table.JHM.太子
  1052. ** e-mail: tablejiang@21cn.com
  1053. ** @Date: 2001 3 26
  1054. ****************************************************/
  1055. void CFtpLib::SetStopSign(BOOL bRun)
  1056. {
  1057. m_bRun = bRun ;
  1058. }