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

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

Ftp客户端

开发平台：

Visual C++

FileQueue.cpp：源码内容
  1. // FileQueue.cpp: implementation of the CFileQueue 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 "QuickFTP.h"
  14. #include "FileQueue.h"
  16. #ifdef _DEBUG
  17. #undef THIS_FILE
  18. static char THIS_FILE[]=__FILE__;
  19. #define new DEBUG_NEW
  20. #endif
  22. //////////////////////////////////////////////////////////////////////
  23. // Construction/Destruction
  24. //////////////////////////////////////////////////////////////////////
  26. CFileQueue::CFileQueue()
  27. {
  28. InitializeCriticalSection( &m_cs );
  29. m_pFileTransmitQueue = NULL ;
  30. m_bLetRun = true ;
  31. //m_SaveFileName[0] = 0 ;
  32. //strcpy( m_SaveFileName , "FileQueue.QFQ" ) ;
  33. m_hParentWnd = NULL ;
  34. }
  36. CFileQueue::~CFileQueue()
  37. {
  38. DeleteAllFileQueue( ) ;
  39. DeleteCriticalSection( &m_cs );
  40. }
  41. /****************************************************
  42. ** @Description
  43. **
  44. **
  45. ** @Parameter
  46. **
  47. **
  48. ** @Return:
  49. ** @Author: JHM
  50. ** e-mail: tablejiang@21cn.com
  51. ** @Date: 2001 3 26
  52. ****************************************************/
  53. BOOL CFileQueue::AddItem(FTPFILEINFO *pFileInfo)
  54. {
  55. EnterCriticalSection( &m_cs ) ;
  57. if( pFileInfo == NULL )
  58. return true ;
  60. FTPFILEINFO* pFileItem ;
  61. FTPFILEINFO*    pCurItem ;
  62. FTPFILEINFO* pPreItem ;
  64. BOOL bFind = false ;
  66. pFileItem = new FTPFILEINFO ;
  67. memcpy( pFileItem , pFileInfo , sizeof( FTPFILEINFO ) ) ;
  68. if( pFileItem->state == FILE_STATE_RUNNING )
  69. pFileItem->state = FILE_STATE_READY ;
  70. pFileItem->pNext = NULL ;
  72. if( m_pFileTransmitQueue == NULL )
  73. {
  74. m_pFileTransmitQueue = pFileItem ;
  75. SaveQueue( ) ;
  76. LeaveCriticalSection( &m_cs ) ;
  77. ::PostMessage( m_hParentWnd , REFRESH_WND_MSG , 0 , 0 ) ;
  78. return true ;
  79. }
  81. pCurItem = m_pFileTransmitQueue ;
  82. pPreItem = m_pFileTransmitQueue ;
  84. while( pCurItem != NULL )
  85. {
  86. if( strcmp( pFileInfo->site.host , pCurItem->site.host ) == 0 )
  87. {
  88. if( strcmp( pFileInfo->remotefilename , pCurItem->remotefilename ) == 0 )
  89. {
  90. if( strcmp( pFileInfo->remotepath , pCurItem->remotepath ) == 0 )
  91. {
  92. if( pFileInfo->bfileput == pCurItem->bfileput )
  93. {
  94. if( pFileInfo->bIsDirectory == pCurItem->bIsDirectory )
  95. {
  96. if(strcmp(  pFileInfo->site.user , pFileInfo->site.user ) == 0 )
  97. {
  98. bFind = true ;
  99. break ;
  100. }
  101. }
  102. }
  103. }
  104. }
  105. }
  106. pPreItem = pCurItem ;
  107. pCurItem = pCurItem->pNext ;
  108. }
  110. if( bFind )
  111. {
  112. delete pFileItem ;
  113. return true ;
  114. }
  115. else
  116. pPreItem->pNext = pFileItem ;
  118. SaveQueue( ) ;
  120. LeaveCriticalSection( &m_cs ) ;
  122. if( m_hParentWnd )
  123. ::PostMessage( m_hParentWnd , REFRESH_WND_MSG , 0 , 0 ) ;
  124. return true ;
  125. }
  126. /****************************************************
  127. ** @Description
  128. **
  129. **
  130. ** @Parameter
  131. **
  132. **
  133. ** @Return:
  134. ** @Author: JHM
  135. ** e-mail: tablejiang@21cn.com
  136. ** @Date: 2001 3 26
  137. ****************************************************/
  138. BOOL CFileQueue::DeleteItem(FTPFILEINFO *pFileInfo)
  139. {
  140. FTPFILEINFO* pFront ;
  141. FTPFILEINFO* pCurrent ;
  142. BOOL  bFind = false ;
  144. EnterCriticalSection( &m_cs ) ;
  146. if( m_pFileTransmitQueue == NULL )
  147. {
  148. LeaveCriticalSection( &m_cs ) ;
  149. return true ;
  150. }
  152. pCurrent = m_pFileTransmitQueue ;
  153. pFront = m_pFileTransmitQueue ;
  155. while( pCurrent != NULL )
  156. {
  157. if( FtpFileCmp( pFileInfo , pCurrent ) )
  158. {
  159. bFind = true ;
  160. break ;
  161. }
  163. pFront = pCurrent ;
  164. pCurrent = pCurrent->pNext ;
  165. }
  167. if( bFind == false )
  168. {
  169. LeaveCriticalSection( &m_cs ) ;
  170. return false ;
  171. }
  173. if( pCurrent == m_pFileTransmitQueue )
  174. {
  175. m_pFileTransmitQueue = pCurrent->pNext ;
  176. }
  177. else
  178. {
  179. pFront->pNext = pCurrent->pNext ;
  180. }
  182. delete pCurrent ;
  184. SaveQueue( ) ;
  186. LeaveCriticalSection( &m_cs ) ;
  188. if( m_hParentWnd )
  189. ::PostMessage( m_hParentWnd , REFRESH_WND_MSG , 0 , 0 ) ;
  190. return true ;
  191. }
  192. /****************************************************
  193. ** @Description
  194. **
  195. **
  196. ** @Parameter
  197. **
  198. **
  199. ** @Return:
  200. ** @Author: JHM
  201. ** e-mail: tablejiang@21cn.com
  202. ** @Date: 2001 3 26
  203. ****************************************************/
  204. BOOL CFileQueue::IsInFileQueue(FTPFILEINFO *pFileInfo)
  205. {
  206. FTPFILEINFO* pFront ;
  207. FTPFILEINFO* pCurrent ;
  208. BOOL  bFind = false ;
  210. EnterCriticalSection( &m_cs ) ;
  212. if( m_pFileTransmitQueue == NULL )
  213. {
  214. LeaveCriticalSection( &m_cs ) ;
  215. return false ;
  216. }
  218. pCurrent = m_pFileTransmitQueue ;
  219. pFront = m_pFileTransmitQueue ;
  221. while( pCurrent != NULL )
  222. {
  223. if( FtpFileCmp( pFileInfo , pCurrent ) )
  224. {
  225. bFind = true ;
  226. break ;
  227. }
  228. pFront = pCurrent ;
  229. pCurrent = pCurrent->pNext ;
  230. }
  232. LeaveCriticalSection( &m_cs ) ;
  233. return bFind ;
  234. }
  235. /****************************************************
  236. ** @Description
  237. **
  238. **
  239. ** @Parameter
  240. **
  241. **
  242. ** @Return:
  243. ** @Author: JHM
  244. ** e-mail: tablejiang@21cn.com
  245. ** @Date: 2001 3 26
  246. ****************************************************/
  247. BOOL CFileQueue::DeleteAllFileQueue()
  248. {
  249. FTPFILEINFO* pCurrent ;
  251. EnterCriticalSection( &m_cs ) ;
  253. if( m_pFileTransmitQueue== NULL)
  254. {
  255. LeaveCriticalSection( &m_cs ) ;
  256. return true ;
  257. }
  259. pCurrent = m_pFileTransmitQueue ;
  260. while( m_pFileTransmitQueue != NULL)
  261. {
  262. m_pFileTransmitQueue = pCurrent->pNext ;
  263. delete pCurrent ;
  264. pCurrent = m_pFileTransmitQueue ;
  265. }
  267. m_pFileTransmitQueue = NULL ;
  269. LeaveCriticalSection( &m_cs ) ;
  271. if( m_hParentWnd )
  272. ::PostMessage( m_hParentWnd , REFRESH_WND_MSG , 0 , 0 ) ;
  273. return true ;
  274. }
  275. /****************************************************
  276. ** @Description
  277. **
  278. **
  279. ** @Parameter
  280. **
  281. **
  282. ** @Return:
  283. ** @Author: JHM
  284. ** e-mail: tablejiang@21cn.com
  285. ** @Date: 2001 3 26
  286. ****************************************************/
  287. BOOL CFileQueue::IsEmpty()
  288. {
  289. EnterCriticalSection( &m_cs ) ;
  291. if( m_pFileTransmitQueue == NULL )
  292. {
  293. LeaveCriticalSection( &m_cs ) ;
  294. return true ;
  295. }
  296. else
  297. {
  298. LeaveCriticalSection( &m_cs ) ;
  299. return false ;
  300. }
  301. }
  302. /****************************************************
  303. ** @Description
  304. **
  305. **
  306. ** @Parameter
  307. **
  308. **
  309. ** @Return:
  310. ** @Author: JHM
  311. ** e-mail: tablejiang@21cn.com
  312. ** @Date: 2001 3 26
  313. ****************************************************/
  314. FTPFILEINFO* CFileQueue::GetNextItem()
  315. {
  316. if( IsEmpty( ) )
  317. return NULL ;
  319. EnterCriticalSection( &m_cs ) ;
  321. FTPFILEINFO* pCur ;
  322. pCur = m_pFileTransmitQueue ;
  323. while( pCur != NULL )
  324. {
  325. if( pCur->state == FILE_STATE_READY )
  326. break ;
  327. pCur = pCur->pNext ;
  328. }
  330. LeaveCriticalSection( &m_cs ) ;
  332. return pCur ;
  333. }
  334. /****************************************************
  335. ** @Description
  336. **
  337. **
  338. ** @Parameter
  339. **
  340. **
  341. ** @Return:
  342. ** @Author: JHM
  343. ** e-mail: tablejiang@21cn.com
  344. ** @Date: 2001 3 26
  345. ****************************************************/
  346. BOOL CFileQueue::SaveQueue()
  347. {
  348. CFile file ;
  349. if( !file.Open( m_SaveFileName , CFile::modeCreate|CFile::modeWrite ) )
  350. return false ;
  352. EnterCriticalSection( &m_cs ) ;
  354. FTPFILEINFO* pCur ;
  355. pCur = m_pFileTransmitQueue ;
  357. try
  358. {
  359. while( pCur != NULL )
  360. {
  361. file.Write( pCur , sizeof( FTPFILEINFO ) ) ;
  362. pCur = pCur->pNext ;
  363. }
  364. }
  365. catch(...)
  366. {
  367. TRACE( "file error!" ) ;
  368. }
  370. LeaveCriticalSection( &m_cs ) ;
  372. file.Close() ;
  374. return true ;
  375. }
  376. /****************************************************
  377. ** @Description
  378. **
  379. **
  380. ** @Parameter
  381. **
  382. **
  383. ** @Return:
  384. ** @Author: JHM
  385. ** e-mail: tablejiang@21cn.com
  386. ** @Date: 2001 3 26
  387. ****************************************************/
  388. BOOL CFileQueue::SetItemState(int iState , FTPFILEINFO* pFileInfo )
  389. {
  390. FTPFILEINFO* pFront ;
  391. FTPFILEINFO* pCurrent ;
  392. BOOL  bFind = false ;
  394. EnterCriticalSection( &m_cs ) ;
  396. if( m_pFileTransmitQueue == NULL )
  397. {
  398. LeaveCriticalSection( &m_cs ) ;
  399. return false ;
  400. }
  402. pCurrent = m_pFileTransmitQueue ;
  403. pFront = m_pFileTransmitQueue ;
  405. while( pCurrent != NULL )
  406. {
  407. if( FtpFileCmp( pFileInfo , pCurrent ) )
  408. {
  409. pCurrent->state = iState ;
  410. bFind = true ;
  411. break ;
  412. }
  414. pFront = pCurrent ;
  415. pCurrent = pCurrent->pNext ;
  416. }
  418. if( bFind == false )
  419. {
  420. LeaveCriticalSection( &m_cs ) ;
  421. return false ;
  422. }
  424. SaveQueue( ) ;
  426. LeaveCriticalSection( &m_cs ) ;
  428. if( m_hParentWnd )
  429. ::PostMessage( m_hParentWnd , REFRESH_WND_MSG , 0 , 0 ) ;
  430. return true ;
  431. }
  433. /****************************************************
  434. ** @Description
  435. **
  436. **
  437. ** @Parameter
  438. **
  439. **
  440. ** @Return:
  441. ** @Author: JHM
  442. ** e-mail: tablejiang@21cn.com
  443. ** @Date: 2001 3 26
  444. ****************************************************/
  445. void CFileQueue::LockTheFileList()
  446. {
  447. EnterCriticalSection( &m_cs ) ;
  448. }
  449. /****************************************************
  450. ** @Description
  451. **
  452. **
  453. ** @Parameter
  454. **
  455. **
  456. ** @Return:
  457. ** @Author: JHM
  458. ** e-mail: tablejiang@21cn.com
  459. ** @Date: 2001 3 26
  460. ****************************************************/
  461. void CFileQueue::UnLockTheFileList()
  462. {
  463. LeaveCriticalSection( &m_cs ) ;
  464. }
  465. /****************************************************
  466. ** @Description
  467. **
  468. **
  469. ** @Parameter
  470. **
  471. **
  472. ** @Return:
  473. ** @Author: JHM
  474. ** e-mail: tablejiang@21cn.com
  475. ** @Date: 2001 3 26
  476. ****************************************************/
  477. FTPFILEINFO* CFileQueue::GetSameSiteFile(FTPFILEINFO *pFile)
  478. {
  479. EnterCriticalSection( &m_cs ) ;
  481. FTPFILEINFO* pCur ;
  483. if( m_pFileTransmitQueue == NULL )
  484. {
  485. LeaveCriticalSection( &m_cs ) ;
  486. return NULL ;
  487. }
  489. pCur = m_pFileTransmitQueue ;
  490. while( pCur != NULL )
  491. {
  492. if( pCur->state == FILE_STATE_READY )
  493. {
  494. if( strcmp( pCur->site.host , pFile->site.host ) == 0 )
  495. {
  496. if( strcmp( pCur->site.user , pFile->site.user ) == 0 )
  497. {
  498. if( strcmp( pCur->site.pass , pFile->site.pass ) == 0 )
  499. {
  500. break ;
  501. }
  502. }
  503. }
  504. }
  505. pCur = pCur->pNext ;
  506. }
  507. LeaveCriticalSection( &m_cs ) ;
  508. return pCur ;
  509. }
  510. /****************************************************
  511. ** @Description
  512. **
  513. **
  514. ** @Parameter
  515. **
  516. **
  517. ** @Return:
  518. ** @Author: JHM
  519. ** e-mail: tablejiang@21cn.com
  520. ** @Date: 2001 3 26
  521. ****************************************************/
  522. BOOL CFileQueue::LoadQueue()
  523. {
  524. CFile file ;
  525. if( !file.Open( m_SaveFileName , CFile::modeRead ) )
  526. return false ;
  528. EnterCriticalSection( &m_cs ) ;
  530. FTPFILEINFO FtpFile ;
  531. UINT iRead ;
  533. try
  534. {
  535. while( iRead )
  536. {
  537. iRead = file.Read( &FtpFile , sizeof( FTPFILEINFO ) ) ;
  538. if( iRead == sizeof( FTPFILEINFO ) )
  539. {
  540. AddItem( &FtpFile ) ;
  541. }
  542. }
  543. }
  544. catch(...)
  545. {
  546. TRACE( "file error!" ) ;
  547. }
  549. LeaveCriticalSection( &m_cs ) ;
  551. file.Close() ;
  553. return true ;
  554. }
  555. /****************************************************
  556. ** @Description
  557. **
  558. **
  559. ** @Parameter
  560. **
  561. **
  562. ** @Return:
  563. ** @Author: JHM
  564. ** e-mail: tablejiang@21cn.com
  565. ** @Date: 2001 3 26
  566. ****************************************************/
  567. BOOL CFileQueue::SetSaveFileName(LPCTSTR filename)
  568. {
  569. strcpy( m_SaveFileName , filename ) ;
  570. DeleteAllFileQueue( ) ;
  571. LoadQueue( ) ;
  572. return true ;
  573. }
  574. /****************************************************
  575. ** @Description
  576. **
  577. **
  578. ** @Parameter
  579. **
  580. **
  581. ** @Return:
  582. ** @Author: JHM
  583. ** e-mail: tablejiang@21cn.com
  584. ** @Date: 2001 3 26
  585. ****************************************************/
  586. BOOL CFileQueue::FtpFileCmp(FTPFILEINFO *pItem1, FTPFILEINFO *pItem2)
  587. {
  588. //if( pItem1->bfileput == pItem2->bfileput )
  589. //{
  590. if( pItem1->filesize == pItem2->filesize )
  591. {
  592. if( strcmp( pItem1->site.host , pItem2->site.host ) == 0 )
  593. {
  594. if( strcmp( pItem1->remotefilename , pItem2->remotefilename ) == 0 )
  595. {
  596. if( strcmp( pItem1->remotepath , pItem2->remotepath ) == 0 )
  597. {
  598. if( strcmp( pItem1->site.user , pItem2->site.user ) == 0 )
  599. {
  600. return true ;
  601. }
  602. }
  603. }
  604. }
  605. }
  606. //}
  607. return false ;
  608. }