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

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > Ftp客户端 > 查看源码
PopGet.cpp
资源名称：Netmanag.zip　[点击查看]
上传用户：geanq888
上传日期：2007-01-03
资源大小：316k
文件大小：12k
源码类别：

Ftp客户端

开发平台：

Visual C++

PopGet.cpp：源码内容
  1. /*----------------------------------------------------------------------------------------------- 
  2. // PopGet.cpp : Defines the class behaviors for the application. 
  3. ---------------------------------------------------------------------------------------------------*/ 
  5. #include "stdafx.h" 
  6. //#include "TesToolbar.h"
  7. #include "Popget.h" 
  9. #ifdef _DEBUG 
  10. #define new DEBUG_NEW 
  11. #undef THIS_FILE 
  12. static char THIS_FILE[] = __FILE__; 
  13. #endif 
  15. #define templist "templist.pop" // <<file used for temp list command
  16. #define temptop  "temptop.pop" // <<file used for temp top command
  18. //////////////////////////////////////////////////////////////////// 
  19. // CPop Class 
  20. ////////////////////////////////////////////////////////////////////// 
  22. ////////////////////////////////////////////////////////////////////// 
  23. // Construction/Destruction 
  24. ////////////////////////////////////////////////////////////////////// 
  25. CPop::CPop() 
  26. { 
  27. if (!m_PopServer.Create()) MessageBox(NULL,"Cant create socket!","Warning",MB_OK);
  28. } 
  29. //{4EEC1C91-6BE1-11d1-8824-00001C302581} 
  31. CPop::~CPop() 
  32. { 
  33.  m_PopServer.Close(); 
  34. } 
  36. BOOL CPop::Connect(CString & Host, int Port, CString & User, CString & Password) 
  37. { 
  38.  char buf [512]; 
  40.  if (!m_PopServer.Connect(Host,Port)) // 110 Pop3 Port 
  41.  { 
  42.   m_ErrorMessage = _T("Server cannot be connected"); 
  43.   return FALSE; 
  44.  } 
  45.  else 
  46.  { 
  47.   if(CheckResponse(CONNECTION_CHECK)==FALSE) 
  48.    return FALSE; 
  50.   wsprintf (buf, "USER %srn", (LPCSTR) User); 
  51.   m_PopServer.Send(buf, strlen (buf)); 
  52.   if(CheckResponse(USER_CHECK)==FALSE) 
  53.    return FALSE; 
  55.   wsprintf (buf, "PASS %srn", (LPCSTR) Password); 
  56.   m_PopServer.Send(buf, strlen (buf)); 
  57.   if (CheckResponse(PASSWORD_CHECK)==FALSE) 
  58.    return FALSE; 
  60.   return TRUE; 
  61.  } 
  63. } 
  65. BOOL CPop::Delete(int MsgNumber) 
  66. { 
  67.  char buf [512]; 
  69.  wsprintf (buf, "DELE %drn",MsgNumber ); 
  70.  m_PopServer.Send(buf, strlen (buf)); 
  71.  if (CheckResponse(DELETE_CHECK)==FALSE) 
  72.   return FALSE; 
  73.  else 
  74.   return TRUE; 
  75. } 
  77. BOOL CPop::Disconnect() 
  78. { 
  79.  char buf [512]; 
  81.  wsprintf (buf, "QUIT rn"); 
  82.  m_PopServer.Send(buf, strlen (buf)); 
  83.  if (CheckResponse(QUIT_CHECK)==FALSE) 
  84.   return FALSE; 
  85.  else 
  86.  {
  87. m_PopServer.Close(); 
  88. return TRUE; 
  89.  }
  90. } 
  92. BOOL CPop::Noop() 
  93. { 
  94.  char buf [512]; 
  96.  wsprintf (buf, "NOOP  rn"); 
  97.  m_PopServer.Send(buf, strlen (buf)); 
  98.  if (CheckResponse(NOOP_CHECK)==FALSE) 
  99.   return FALSE; 
  100.  else 
  101.   return TRUE; 
  102. } 
  104. // Return the Msg Size for given msg number 
  105. int CPop::GetMessageSize(int MsgNumber) 
  106. { 
  107.  if(m_SizeOfMsg.GetSize() < MsgNumber+1) 
  108.   return 0; 
  109.  else 
  110.   return m_SizeOfMsg[MsgNumber+1]; 
  111. } 
  113. BOOL CPop::Reset() 
  114. { 
  115.  char buf [512]; 
  117.  wsprintf (buf, "RSET rn"); 
  118.  m_PopServer.Send(buf, strlen (buf)); 
  119.  if (CheckResponse(RSET_CHECK)==FALSE) 
  120.   return FALSE; 
  121.  else 
  122.   return TRUE; 
  123. } 
  125. // MsgContents will hold the msg body 
  126. BOOL CPop::Retrieve(int  MsgNumber) 
  127. { 
  128.  char buf [512]; 
  130.  m_todisk=FALSE;
  131.  wsprintf (buf, "RETR %drn",MsgNumber );
  132.  m_PopServer.Send(buf, strlen (buf)); 
  133.  if (CheckResponse(RETR_CHECK)==FALSE) 
  134.   return FALSE; 
  135.  else 
  136.   return TRUE; 
  137. } 
  139. BOOL CPop::Retrieve(int  MsgNumber,CString fname) 
  140. { 
  141.  char buf [512]; 
  143.  if (SetOutputFile(fname)) m_todisk=TRUE;
  144.  else
  145.  {
  146. m_ErrorMessage = _T("Error opening output file");
  147.  return FALSE;
  148.  }
  149.  wsprintf (buf, "RETR %drn",MsgNumber );
  150.  m_PopServer.Send(buf, strlen (buf)); 
  151.  if (CheckResponse(RETR_CHECK)==FALSE) 
  152.   return FALSE; 
  153.  else 
  154.   return TRUE; 
  155. } 
  157. BOOL CPop::Statistics() 
  158. { 
  159.  char buf [512]; 
  161.  wsprintf (buf, "STAT rn"); 
  162.  m_PopServer.Send(buf, strlen (buf)); 
  163.  if (CheckResponse(STAT_CHECK)==FALSE) 
  164.   return FALSE; 
  165.  else 
  166.   return TRUE; 
  167. } 
  169. CString CPop::GetMsgContents() 
  170. { 
  171.  return m_MsgContents; 
  172. } 
  174. int CPop::GetNumberOfMails() 
  175. { 
  176.  return m_NumberMail; 
  177. } 
  179. int CPop::GetTotalMailSize() 
  180. { 
  181.  return m_TotalSize; 
  182. } 
  184. BOOL CPop::Connect() 
  185. { 
  186.  return Connect(m_Host, m_port, m_User, m_Password); 
  187. } 
  189. void CPop::SetHost(CString Host) 
  190. { 
  191.  m_Host = Host; 
  192. } 
  194. CString CPop::GetHost() 
  195. { 
  196.  return m_Host; 
  197. } 
  199. void CPop::SetUser(CString User) 
  200. { 
  201.  m_User = User; 
  202. } 
  204. CString CPop::GetUser() 
  205. { 
  206.  return m_User; 
  207. } 
  209. void CPop::SetPassword(CString Password) 
  210. { 
  211.  m_Password = Password; 
  212. } 
  214. CString CPop::GetPassword() 
  215. { 
  216.  return m_Password; 
  217. } 
  219. BOOL CPop::CheckResponse(int ResponseType) 
  220. { 
  221.  char buf[8192]; 
  222.  int totr,skip;
  223.  BOOL ready;
  225.  for (int i=0;i<sizeof(buf);i++) 
  226.   buf[i]=''; 
  228.  Sleep(2000);
  229.  totr=m_PopServer.Receive(buf, sizeof(buf)); 
  230.   
  231.  switch (ResponseType) 
  232.  { 
  233.   case CONNECTION_CHECK: 
  234.    if (strnicmp(buf,"-ERR", 4) == 0) 
  235.    { 
  236.     m_ErrorMessage = _T("Bad Connection"); 
  237.     return FALSE; 
  238.    } 
  239.    break; 
  241.   case USER_CHECK: 
  242.    if (strnicmp(buf,"-ERR", 4) == 0) 
  243.    { 
  244.     m_ErrorMessage = _T("Bad User Name"); 
  245.     return FALSE; 
  246.    } 
  247.    break; 
  248.   case PASSWORD_CHECK: 
  249.    if (strnicmp(buf,"-ERR", 4) == 0) 
  250.    { 
  251.     m_ErrorMessage = _T("Bad Password Name"); 
  252.     return FALSE; 
  253.    } 
  254.    break; 
  255.   case QUIT_CHECK: 
  256.    if (strnicmp(buf,"-ERR", 4) == 0) 
  257.    { 
  258.     m_ErrorMessage = _T("Error occured during QUIT"); 
  259.     return FALSE; 
  260.    } 
  261.    break; 
  262.   case DELETE_CHECK: 
  263.    if (strnicmp(buf,"-ERR", 4) == 0) 
  264.    { 
  265.     m_ErrorMessage = _T("Error occured during DELE"); 
  266.     return FALSE; 
  267.    } 
  268.    break; 
  269.   case RSET_CHECK: 
  270.    if (strnicmp(buf,"-ERR", 4) == 0) 
  271.    { 
  272.     m_ErrorMessage = _T("Error occured during RSET"); 
  273.     return FALSE; 
  274.    } 
  275.    break; 
  276.   case STAT_CHECK: 
  277.    if (strnicmp(buf,"-ERR", 4) == 0) 
  278.    { 
  279.     m_ErrorMessage = _T("Error occured during STAT"); 
  280.     return FALSE; 
  281.    } 
  282.    else 
  283.    { 
  284.     BOOL EmailNumber = TRUE; 
  285.     for (char *p = buf; *p != ''; p++) 
  286.     { 
  287.      if (*p == 't' || *p == ' ') 
  288.      { 
  289.       if(EmailNumber == TRUE) 
  290.       { 
  291.        m_NumberMail = atoi(p); 
  292.        EmailNumber = FALSE; 
  293.       } 
  294.       else 
  295.       { 
  296.        m_TotalSize = atoi(p); 
  297.        return TRUE; 
  298.       } 
  299.      } 
  300.     } 
  301.    } 
  302.    break; 
  303.   case NOOP_CHECK: 
  304.    if (strnicmp(buf,"-ERR", 4) == 0) 
  305.    { 
  306.     m_ErrorMessage = _T("Error occured during NOOP"); 
  307.     return FALSE; 
  308.    } 
  309.    break; 
  310.   case LIST_CHECK: 
  311. if (strnicmp(buf,"-ERR", 4) == 0) 
  312. { 
  313.     m_ErrorMessage = _T("Error occured during LIST"); 
  314. return FALSE; 
  315. } 
  316. else 
  317. { 
  318. m_outputfile.Open(templist,CFile::modeCreate|CFile::modeWrite);
  319. if (m_outputfile==0)
  320. {
  321. //receive all rest data and return false;
  323. if ((totr!=0) && (totr==sizeof(buf))) totr=m_PopServer.Receive(buf, sizeof(buf));
  325. m_ErrorMessage = _T("Error creating temp input file!");
  326. return FALSE;
  327. }
  329.    ready=FALSE;
  330.    if (strnicmp(buf,"+OK rn", 6) == 0) skip=6;
  331.    else  skip=4; //skip +OK(SPACE)
  332.    while (!ready)
  333.    {
  334. if (totr==SOCKET_ERROR)
  335. {
  336. m_outputfile.Close();
  337. ready=TRUE;
  338. m_ErrorMessage = _T("Error reading from socket!");
  339. return FALSE;
  340. }
  341. else
  342. {
  343. m_outputfile.Write(buf+skip,totr-skip);
  344. skip=0;
  345. if (
  346. ((buf[totr-3]=='.')&&(buf[totr-2]=='r')&&(buf[totr-1]=='n')) ||
  347. (totr==0) ||
  348. ((buf[0]='O')&&(buf[1]=='K')&&(buf[totr-3]=='.')&&(buf[totr-2]=='r')&&(buf[totr-1]=='n'))
  349. )
  350. {
  351. ready=TRUE;
  352. m_outputfile.Close();
  353. }
  354. else
  355. {
  356. totr=m_PopServer.Receive(buf, sizeof(buf));
  357. }
  358. }
  359.    }
  360. return MakeMsgList();
  361.    }
  362.    break; 
  363.   case RETR_CHECK: 
  364.    if (strnicmp(buf,"-ERR", 4) == 0) 
  365.    { 
  366.     m_ErrorMessage = _T("Error occured during RETR"); 
  367. if (m_todisk) m_outputfile.Close();
  368.     return FALSE; 
  369.    } 
  370.    else 
  371.    { 
  372.    skip=0;
  373.    if (strnicmp(buf,"+OK", 3) == 0)
  374.    {
  375.    int ii=0;
  376.    while (buf[ii]!='R') ii++;
  377.    skip=ii;
  378.    }
  379. ready=FALSE;
  380.     while (!ready)
  381. {
  382. if (totr==SOCKET_ERROR)
  383. {
  384. if (m_todisk) m_outputfile.Close();
  385. ready=TRUE;
  386. m_ErrorMessage = _T("Error reading from socket!");
  387. return FALSE;
  388. }
  389. else
  390. {
  391. if (m_todisk) m_outputfile.Write(buf+skip,totr-skip);
  392. skip=0;
  393. if (
  394. ((buf[totr-3]=='.')&&(buf[totr-2]=='r')&&(buf[totr-1]=='n')) ||
  395. (totr==0)
  396. )
  397. {
  398. ready=TRUE;
  399. if (m_todisk) m_outputfile.Close();
  400. }
  401. else
  402. {
  403. totr=m_PopServer.Receive(buf, sizeof(buf)); 
  404. }
  405. }
  406. }
  407. if (!m_todisk) m_MsgContents = buf; 
  408.    } 
  409.    break; 
  410.   case TOP_CHECK: 
  411. m_outputfile.Open(temptop,CFile::modeCreate|CFile::modeWrite);
  412. if (m_outputfile==0)
  413. {
  414. //receive all rest data and return false;
  415. if ((totr!=0) && (totr==sizeof(buf))) totr=m_PopServer.Receive(buf, sizeof(buf));
  416. m_ErrorMessage = _T("Error creating temp input file!");
  417. return FALSE;
  418. }
  420.    ready=FALSE;
  421.    if (strnicmp(buf,"+OK rn", 6) == 0) skip=6;
  422.    else  skip=4; //skip +OK(SPACE)
  423.    while (!ready)
  424.    {
  425. if (totr==SOCKET_ERROR)
  426. {
  427. //receive all rest data and return false;
  428. if ((totr!=0) && (totr==sizeof(buf))) totr=m_PopServer.Receive(buf, sizeof(buf));
  429. m_outputfile.Close();
  430. ready=TRUE;
  431. m_ErrorMessage = _T("Error reading from socket!");
  432. return FALSE;
  433. }
  434. else
  435. {
  436. m_outputfile.Write(buf+skip,totr-skip);
  437. skip=0;
  438. if (
  439. ((buf[totr-3]=='.')&&(buf[totr-2]=='r')&&(buf[totr-1]=='n')) ||
  440. (totr==0)
  441. )
  442. {
  443. ready=TRUE;
  444. m_outputfile.Close();
  445. }
  446. else
  447. {
  448. totr=m_PopServer.Receive(buf, sizeof(buf));
  449. }
  450. }
  451.    }
  452.    if (ExtractTop(temptop))
  453.    {
  454. CFile::Remove(temptop);
  455. return TRUE;
  456.    }
  457.    else
  458.    {
  459. CFile::Remove(temptop);
  460. return FALSE;
  461.    }
  464.    break; 
  465.  } 
  466.  return TRUE; 
  467. } 
  469. CString CPop::GetErrorMessage() 
  470. { 
  471.  return m_ErrorMessage; 
  472. } 
  474. BOOL CPop::List() 
  475. { 
  476.  char buf [512]; 
  478.  wsprintf (buf, "LIST  rn"); 
  479.  m_PopServer.Send(buf, strlen (buf)); 
  480.  if (CheckResponse(LIST_CHECK)==FALSE) 
  481.   return FALSE; 
  482.  else 
  483.   return TRUE; 
  484. } 
  486. BOOL CPop::GetTop(int MsgNumber, int Length) 
  487. { 
  488.  char buf [512]; 
  490.  m_todisk=FALSE;
  491.  wsprintf (buf, "TOP %d %drn",MsgNumber,Length );
  492.  m_PopServer.Send(buf, strlen (buf)); 
  493.  if (CheckResponse(TOP_CHECK)==FALSE) 
  494.   return FALSE; 
  495.  else 
  496.   return TRUE; 
  497. } 
  500. void CPop::SetPort(int port)
  501. {
  502. m_port=port;
  503. }
  506. BOOL CPop::SetOutputFile(CString fname)
  507. {
  508. TRY
  509. {
  510. m_outputfile.Open(fname,CFile::modeCreate|CFile::modeWrite);
  511. return TRUE;
  512. }
  513. CATCH( CFileException, e )
  514. {
  515. MessageBox(NULL,"Error opening outputfile!","Error:",MB_OK);
  516. return FALSE;
  517. }
  518. END_CATCH
  519. }
  521. BOOL CPop::ExtractTop(CString fname)
  522. {
  523. CString tcstring;
  524. int pos;
  526. TRY{
  527. CStdioFile f(fname,CFile::modeRead);
  528. while (f.ReadString(tcstring))
  529. {
  530. tcstring.TrimLeft(); tcstring.TrimRight();
  531. if (tcstring.Find('r')!=-1) tcstring=tcstring.Left(tcstring.GetLength()-1);
  532. if (tcstring.Find('n')!=-1) tcstring=tcstring.Left(tcstring.GetLength()-1);
  533. tcstring.TrimLeft(); tcstring.TrimRight();
  535. if (pos=tcstring.Find("To:")==0) t_To=tcstring.Right(tcstring.GetLength()-sizeof("To:"));
  536. if (pos=tcstring.Find("From:")==0) t_From=tcstring.Right(tcstring.GetLength()-sizeof("From:"));
  537. if (pos=tcstring.Find("Subject:")==0) t_Subject=tcstring.Right(tcstring.GetLength()-sizeof("Subject:"));
  538. if (pos=tcstring.Find("Date:")==0) t_Date=tcstring.Right(tcstring.GetLength()-sizeof("Date:"));
  539. }
  540. f.Close();
  541. //if ((t_To=="")||(t_From=="")||(t_Date=="")) return FALSE;
  542. if (t_To == "") return FALSE;
  543. else return TRUE;
  544. }
  545. CATCH( CFileException, e ){
  546. m_ErrorMessage="cant open message file";
  547. return FALSE;
  548. }END_CATCH
  550. }
  552. BOOL CPop::MakeMsgList()
  553. {
  554. m_SizeOfMsg.RemoveAll();
  555. CString tcstring;
  556. int ii;
  558. TRY{
  559. CStdioFile f(templist,CFile::modeRead);
  560. while (f.ReadString(tcstring))
  561. {
  562. tcstring.TrimLeft(); tcstring.TrimRight();
  563. if (tcstring.Find('r')!=-1) tcstring=tcstring.Left(tcstring.GetLength()-1);
  564. if (tcstring.Find('n')!=-1) tcstring=tcstring.Left(tcstring.GetLength()-1);
  565. tcstring.TrimLeft(); tcstring.TrimRight();
  566. ii=tcstring.Find(' ');
  567. if (ii!=-1)
  568. {
  569. tcstring=tcstring.Mid(ii+1,tcstring.GetLength());
  570. m_SizeOfMsg.Add(atoi(tcstring));
  571. }
  572. }
  573. f.Close();
  574. CFile::Remove(templist);
  575. return TRUE;
  576. }
  577. CATCH( CFileException, e ){
  578. m_ErrorMessage="cant open templist file";
  579. CFile::Remove(templist);
  580. return FALSE;
  581. }END_CATCH
  582. }