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

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

Email客户端

开发平台：

Visual C++

POPLevel.cpp：源码内容
  1. // POPLevel.cpp : Implementation of CPOPLevel
  2. #include "stdafx.h"
  3. #include "Pophandler.h"
  4. #include "POPLevel.h"
  5. #include "SmartWSA.h"
  7. /////////////////////////////////////////////////////////////////////////////
  8. // CPOPLevel
  10. HRESULT CPOPLevel::FinalConstruct()
  11. {
  12. m_pcSmartWSA = new CSmartWSA;
  13. ATLASSERT(m_pcSmartWSA);
  14. m_pcPOP3 = new CPop3Connection;
  15. ATLASSERT(m_pcPOP3);
  16. return S_OK;
  17. }
  19. void CPOPLevel::FinalRelease()
  20. {
  21. delete m_pcPOP3;
  22. delete m_pcSmartWSA;
  23. }
  26. STDMETHODIMP CPOPLevel::InterfaceSupportsErrorInfo(REFIID riid)
  27. {
  28. static const IID* arr[] = 
  29. {
  30. &IID_IPOPLevel
  31. };
  32. for (int i=0; i < sizeof(arr) / sizeof(arr[0]); i++)
  33. {
  34. if (InlineIsEqualGUID(*arr[i], riid))
  35. return S_OK;
  36. }
  37. return S_FALSE;
  38. }
  41. STDMETHODIMP CPOPLevel::get_Host(BSTR *pVal)
  42. {
  43. AFX_MANAGE_STATE(AfxGetStaticModuleState())
  45. ATLASSERT(pVal);
  46. try
  47. {
  48. *pVal = m_bstHost.copy();
  49. }
  50. catch(...)
  51. {
  52. return E_FAIL;
  53. }
  55. return S_OK;
  56. }
  58. STDMETHODIMP CPOPLevel::put_Host(BSTR newVal)
  59. {
  60. AFX_MANAGE_STATE(AfxGetStaticModuleState())
  62. try
  63. {
  64. m_bstHost = newVal;
  65. }
  66. catch(...)
  67. {
  68. return E_FAIL;
  69. }
  71. return S_OK;
  72. }
  74. STDMETHODIMP CPOPLevel::get_Port(long *pVal)
  75. {
  76. AFX_MANAGE_STATE(AfxGetStaticModuleState())
  78. ATLASSERT(pVal);
  79. *pVal = (LONG)m_dwPort;
  81. return S_OK;
  82. }
  84. STDMETHODIMP CPOPLevel::put_Port(long newVal)
  85. {
  86. AFX_MANAGE_STATE(AfxGetStaticModuleState())
  88. m_dwPort = (DWORD) newVal;
  90. return S_OK;
  91. }
  93. STDMETHODIMP CPOPLevel::get_User(BSTR *pVal)
  94. {
  95. AFX_MANAGE_STATE(AfxGetStaticModuleState())
  97. ATLASSERT(pVal);
  99. try
  100. {
  101. *pVal = m_bstUser.copy();
  102. }
  103. catch(...)
  104. {
  105. return E_FAIL;
  106. }
  108. return S_OK;
  109. }
  111. STDMETHODIMP CPOPLevel::put_User(BSTR newVal)
  112. {
  113. AFX_MANAGE_STATE(AfxGetStaticModuleState())
  115. try
  116. {
  117. m_bstUser = newVal;
  118. }
  119. catch(...)
  120. {
  121. return E_FAIL;
  122. }
  124. return S_OK;
  125. }
  127. STDMETHODIMP CPOPLevel::get_Password(BSTR *pVal)
  128. {
  129. AFX_MANAGE_STATE(AfxGetStaticModuleState())
  131. ATLASSERT(pVal);
  132. try
  133. {
  134. *pVal = m_bstPassword.copy();
  135. }
  136. catch(...)
  137. {
  138. return E_FAIL;
  139. }
  141. return S_OK;
  142. }
  144. STDMETHODIMP CPOPLevel::put_Password(BSTR newVal)
  145. {
  146. AFX_MANAGE_STATE(AfxGetStaticModuleState())
  148. try
  149. {
  150. m_bstPassword = newVal;
  151. }
  152. catch(...)
  153. {
  154. return E_FAIL;
  155. }
  157. return S_OK;
  158. }
  160. STDMETHODIMP CPOPLevel::get_Timeout(long *pVal)
  161. {
  162. AFX_MANAGE_STATE(AfxGetStaticModuleState())
  164. ATLASSERT(pVal);
  165. *pVal = (LONG)m_pcPOP3->GetTimeout();
  167. return S_OK;
  168. }
  170. STDMETHODIMP CPOPLevel::put_Timeout(long newVal)
  171. {
  172. AFX_MANAGE_STATE(AfxGetStaticModuleState())
  174. m_pcPOP3->SetTimeout(newVal);
  176. return S_OK;
  177. }
  179. STDMETHODIMP CPOPLevel::Connect()
  180. {
  181. AFX_MANAGE_STATE(AfxGetStaticModuleState())
  183. BOOL blRet = m_pcPOP3->Connect(m_bstHost, m_bstUser, m_bstPassword, m_dwPort);
  184. if(blRet == false)
  185. return E_FAIL;
  186. //return E_FAIL;
  187. return S_OK;
  188. }
  190. STDMETHODIMP CPOPLevel::Disconnect()
  191. {
  192. AFX_MANAGE_STATE(AfxGetStaticModuleState())
  194. BOOL blRet = m_pcPOP3->Disconnect();
  195. if(blRet == false)
  196. return E_FAIL;
  197. return S_OK;
  198. }
  200. STDMETHODIMP CPOPLevel::GetMailCount(long *a_plCount)
  201. {
  202. AFX_MANAGE_STATE(AfxGetStaticModuleState())
  204. ATLASSERT(a_plCount);
  206. INT iSize;
  207. INT iCount;
  208. BOOL blRet = m_pcPOP3->Statistics(iCount, iSize);
  209. if(blRet == false)
  210. return E_FAIL;
  212. *a_plCount = iCount;
  214. return S_OK;
  215. }
  217. STDMETHODIMP CPOPLevel::GetMail(long a_lMailNum, long a_lDelete)
  218. {
  219. AFX_MANAGE_STATE(AfxGetStaticModuleState())
  221. FILE* pfTemp = NULL;
  222. FILE* pfAttach = NULL;
  223. try
  224. {
  225. // clear attached files
  226. m_bstAttachedFiles = "";
  227. // clear body
  228. m_bstBody = "";
  230. // get mail message
  231. BOOL blRet = m_pcPOP3->Retrieve(a_lMailNum, m_cMessage);
  232. if(blRet == false)
  233. return E_FAIL;
  234. if(a_lDelete > 0)
  235. {
  236. // mark message as del
  237. m_pcPOP3->Delete(a_lMailNum);
  238. }
  239. // write raw body to the file
  240. pfTemp = fopen("temp.mme", "w+b");
  241. ATLASSERT(pfTemp);
  243. CString sRawBody = m_cMessage.GetRawBody();
  244. fputs(sRawBody, pfTemp);
  245. fclose(pfTemp);
  246. pfTemp = NULL;
  248. // get boundary
  249. CString sBoundary = m_cMessage.GetBoundary();
  250. // open file
  251. pfTemp = fopen("temp.mme", "rb");
  252. ATLASSERT(pfTemp);
  253. CHAR text[1024] = "";
  254. bool blIsBodyEmpty = false;
  255. // store position for the empty body
  256. LONG lPos = ftell(pfTemp);
  257. // get opening boundary string
  258. if(sBoundary.GetLength() != 0)
  259. {
  260. while(fgets(text, sizeof(text) - 1, pfTemp))
  261. {
  262. if( !strncmp(text, "--", 2) && !strncmp(text + 2, sBoundary, strlen(sBoundary)))
  263. break;
  264. }
  265. // local headers   
  266. // get header out of the way
  267. while(fgets(text, sizeof(text) - 1, pfTemp))
  268. {
  269. if((*text == 'n') || (*text == 'r') || !*text)
  270. break;
  271. if(strncmp(text, "Content-Disposition: attachment", strlen("Content-Disposition: attachment")) == 0)
  272. {
  273. blIsBodyEmpty = true;
  274. }
  275. }
  276. }
  278. // read body
  279. CString sBody;
  280. if(blIsBodyEmpty == false)
  281. {
  282. memset(text, 0, sizeof(text));
  283. while(fgets(text, sizeof(text) - 1, pfTemp))
  284. {
  285. if(sBoundary.GetLength() != 0)
  286. {
  287. if(!strncmp(text, "--", 2) && !strncmp(text + 2, sBoundary, strlen(sBoundary)))
  288. break;
  289. }
  290. sBody += text;
  291. if(strlen(text) >= 2)
  292. {
  293. if( (text[strlen(text) - 2] == 0x0d) && ((text[strlen(text) - 1] == 0x0a) ) )
  294. {
  295. sBody = sBody.Left(sBody.GetLength() - 2);
  296. sBody += "n";
  297. }
  298. }
  299. }
  300. }
  302. // check is it multipart
  303. CString sContentType = m_cMessage.GetContentType();
  304. if(sContentType.Compare("multipart/mixed") != 0)
  305. {
  306. m_bstBody = sBody;
  307. fclose(pfTemp);
  308. pfTemp = NULL;
  309. remove("temp.mme");
  310. return S_OK;
  311. }
  313. if(sBody.GetLength() > 2)
  314. sBody = sBody.Left(sBody.GetLength() - 2);
  316. m_bstBody = sBody;
  317. if(blIsBodyEmpty == false)
  318. // store new position
  319. lPos = ftell(pfTemp);
  321. // now parse attached files
  322. fseek(pfTemp, lPos, SEEK_SET);
  323. CHAR* pszStr = NULL;
  324. while(true)
  325. {
  326. // local headers   
  327. // get header out of the way
  328. while( (pszStr = fgets(text, sizeof(text) - 1, pfTemp)) )
  329. {
  330. if((*text == 'n') || (*text == 'r') || !*text)
  331. break;
  332. // try to get filename
  333. // Content-Disposition: attachment; filename="New Text Document.txt"
  334. if(strncmp(text, "Content-Disposition: attachment; filename=", strlen("Content-Disposition: attachment; filename=")) == 0)
  335. {
  336. CString sFileName = text + strlen("Content-Disposition: attachment; filename=") + 1;
  337. if(sFileName.GetLength() <= 3)
  338. {
  339. fclose(pfTemp);
  340. pfTemp = NULL;
  341. remove("temp.mme");
  342. return E_FAIL;
  343. }
  344. sFileName = sFileName.Left(sFileName.GetLength() - 3);
  345. // try to open file
  346. pfAttach = fopen(sFileName, "w+b");
  347. ATLASSERT(pfAttach);
  348. m_bstAttachedFiles += (LPCTSTR)sFileName;
  349. m_bstAttachedFiles += "n";
  350. }
  351. }
  353. // read current attachment
  354. memset(text, 0, sizeof(text));
  355. while( (pszStr = fgets(text, sizeof(text) - 1, pfTemp)) )
  356. {
  357. if(!strncmp(text, "--", 2) && !strncmp(text + 2, sBoundary, strlen(sBoundary)))
  358. break;      
  359. LONG x = 0;
  360. LONG triplet = 0; 
  361. LONG qpos = 0;
  362. LONG equals = 0;
  364. while(text[x])
  365. {
  366. if(base64_valid(text[x]))
  367. {
  368. if(text[x] == '=')
  369. {
  370. equals++;
  371. }
  372. else
  373. triplet |= (base64_get_value(text[x]) << ((3 - qpos) * 6));
  374. qpos++;
  375. }
  377. if(qpos == 4)
  378. {
  379. fputc((triplet & 0xFF0000) >> 16, pfAttach);
  380. if(equals < 2)
  381. fputc((triplet & 0xFF00) >> 8, pfAttach);
  382. if(!equals)
  383. fputc((triplet & 0xFF), pfAttach);
  385. triplet = 0;
  386. qpos = 0;
  387. }
  388. x ++;
  389. }
  390. memset(text, 0, sizeof(text));
  391. }
  392. if(pfAttach)
  393. {
  394. fclose(pfAttach);
  395. pfAttach = NULL;
  396. }
  398. if(pszStr == NULL)
  399. {
  400. // eof, stopping
  401. break;
  402. }
  403. }
  405. fclose(pfTemp);
  406. remove("temp.mme");
  407. }
  408. catch (...) 
  409. {
  410. if(pfTemp)
  411. fclose(pfTemp);
  412. if(pfAttach)
  413. fclose(pfAttach);
  414. remove("temp.mme");
  415. return E_FAIL;
  416. }
  417. return S_OK;
  418. }
  420. STDMETHODIMP CPOPLevel::get_From(BSTR *pVal)
  421. {
  422. AFX_MANAGE_STATE(AfxGetStaticModuleState())
  423. ATLASSERT(pVal);
  425. try
  426. {
  427. bstr_t bstFrom;
  428. bstFrom = m_cMessage.GetFrom();
  430. *pVal = bstFrom.copy();
  431. }
  432. catch(...)
  433. {
  434. return E_FAIL;
  435. }
  437. return S_OK;
  438. }
  440. STDMETHODIMP CPOPLevel::get_To(BSTR *pVal)
  441. {
  442. AFX_MANAGE_STATE(AfxGetStaticModuleState())
  443. ATLASSERT(pVal);
  445. try
  446. {
  447. bstr_t bstTo;
  448. bstTo = m_cMessage.GetTo();
  450. *pVal = bstTo.copy();
  451. }
  452. catch(...)
  453. {
  454. return E_FAIL;
  455. }
  457. return S_OK;
  458. }
  460. STDMETHODIMP CPOPLevel::get_CC(BSTR *pVal)
  461. {
  462. AFX_MANAGE_STATE(AfxGetStaticModuleState())
  463. ATLASSERT(pVal);
  465. try
  466. {
  467. bstr_t bstCC;
  468. bstCC = m_cMessage.GetCC();
  470. *pVal = bstCC.copy();
  471. }
  472. catch(...)
  473. {
  474. return E_FAIL;
  475. }
  477. return S_OK;
  478. }
  480. STDMETHODIMP CPOPLevel::get_Date(BSTR *pVal)
  481. {
  482. AFX_MANAGE_STATE(AfxGetStaticModuleState())
  483. ATLASSERT(pVal);
  485. try
  486. {
  487. bstr_t bstDate;
  488. bstDate = m_cMessage.GetDate();
  490. *pVal = bstDate.copy();
  491. }
  492. catch(...)
  493. {
  494. return E_FAIL;
  495. }
  497. return S_OK;
  498. }
  500. STDMETHODIMP CPOPLevel::get_Header(BSTR *pVal)
  501. {
  502. AFX_MANAGE_STATE(AfxGetStaticModuleState())
  503. ATLASSERT(pVal);
  505. try
  506. {
  507. bstr_t bstHeader;
  508. bstHeader = m_cMessage.GetHeader();
  510. *pVal = bstHeader.copy();
  511. }
  512. catch(...)
  513. {
  514. return E_FAIL;
  515. }
  517. return S_OK;
  518. }
  520. STDMETHODIMP CPOPLevel::get_Subject(BSTR *pVal)
  521. {
  522. AFX_MANAGE_STATE(AfxGetStaticModuleState())
  523. ATLASSERT(pVal);
  525. try
  526. {
  527. bstr_t bstSubject;
  528. bstSubject = m_cMessage.GetSubject();
  530. *pVal = bstSubject.copy();
  531. }
  532. catch(...)
  533. {
  534. return E_FAIL;
  535. }
  537. return S_OK;
  538. }
  540. STDMETHODIMP CPOPLevel::get_AttachedFiles(BSTR* pVal)
  541. {
  542. AFX_MANAGE_STATE(AfxGetStaticModuleState())
  543. ATLASSERT(pVal);
  545. try
  546. {
  547. *pVal = m_bstAttachedFiles.copy();
  548. }
  549. catch(...)
  550. {
  551. return E_FAIL;
  552. }
  554. return S_OK;
  555. }
  557. STDMETHODIMP CPOPLevel::get_Body(BSTR *pVal)
  558. {
  559. AFX_MANAGE_STATE(AfxGetStaticModuleState())
  561. ATLASSERT(pVal);
  563. try
  564. {
  565. *pVal = m_bstBody.copy();
  566. }
  567. catch(...)
  568. {
  569. return E_FAIL;
  570. }
  572. return S_OK;
  573. }
  575. /*
  576.  * Eucalyptus
  577.  * by Paul A. Schifferer
  578.  *
  579.  * Module: base64.c
  580.  *     Base64 related functions
  581.  *
  582.  * Copyright