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

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > Email服务器 > 查看源码
MailDomains.cpp
资源名称：xmail-1.22.rar　[点击查看]
上传用户：woshihumen
上传日期：2013-07-18
资源大小：484k
文件大小：14k
源码类别：

Email服务器

开发平台：

Visual C++

MailDomains.cpp：源码内容
  1. /*
  2.  *  XMail by Davide Libenzi ( Intranet and Internet mail server )
  3.  *  Copyright (C) 1999,..,2004  Davide Libenzi
  4.  *
  5.  *  This program is free software; you can redistribute it and/or modify
  6.  *  it under the terms of the GNU General Public License as published by
  7.  *  the Free Software Foundation; either version 2 of the License, or
  8.  *  (at your option) any later version.
  9.  *
  10.  *  This program is distributed in the hope that it will be useful,
  11.  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  12.  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13.  *  GNU General Public License for more details.
  14.  *
  15.  *  You should have received a copy of the GNU General Public License
  16.  *  along with this program; if not, write to the Free Software
  17.  *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  18.  *
  19.  *  Davide Libenzi <davidel@xmailserver.org>
  20.  *
  21.  */
  23. #include "SysInclude.h"
  24. #include "SysDep.h"
  25. #include "SvrDefines.h"
  26. #include "ShBlocks.h"
  27. #include "ResLocks.h"
  28. #include "StrUtils.h"
  29. #include "SList.h"
  30. #include "BuffSock.h"
  31. #include "MailConfig.h"
  32. #include "MessQueue.h"
  33. #include "MailSvr.h"
  34. #include "MiscUtils.h"
  35. #include "SvrUtils.h"
  36. #include "POP3GwLink.h"
  37. #include "ExtAliases.h"
  38. #include "UsrUtils.h"
  39. #include "UsrAuth.h"
  40. #include "TabIndex.h"
  41. #include "AliasDomain.h"
  42. #include "SMAILUtils.h"
  43. #include "MailDomains.h"
  45. #define MAIL_DOMAINS_DIR            "domains"
  46. #define MAIL_DOMAINS_FILE           "domains.tab"
  47. #define MAIL_DOMAINS_LINE_MAX       512
  49. enum PopDomainFileds {
  50. domDomain = 0,
  52. domMax
  53. };
  55. struct DomainsScanData {
  56. char szTmpDBFile[SYS_MAX_PATH];
  57. FILE *pDBFile;
  58. char szCurrDomain[256];
  59. };
  61. static int MDomRebuildDomainsIndexes(char const *pszDomainsFilePath);
  62. static char *MDomGetDomainsFilePath(char *pszDomainsFilePath, int iMaxPath);
  64. static int iIdxDomains_Domain[] = {
  65. domDomain,
  67. INDEX_SEQUENCE_TERMINATOR
  68. };
  70. int MDomCheckDomainsIndexes(void)
  71. {
  73. char szDomainsFilePath[SYS_MAX_PATH] = "";
  75. MDomGetDomainsFilePath(szDomainsFilePath, sizeof(szDomainsFilePath));
  77. ///////////////////////////////////////////////////////////////////////////////
  78. //  Align RmtDomain-RmtName index
  79. ///////////////////////////////////////////////////////////////////////////////
  80. if (TbixCheckIndex(szDomainsFilePath, iIdxDomains_Domain, false) < 0)
  81. return (ErrGetErrorCode());
  83. return (0);
  85. }
  87. static int MDomRebuildDomainsIndexes(char const *pszDomainsFilePath)
  88. {
  89. ///////////////////////////////////////////////////////////////////////////////
  90. //  Rebuild RmtDomain-RmtName index
  91. ///////////////////////////////////////////////////////////////////////////////
  92. if (TbixCreateIndex(pszDomainsFilePath, iIdxDomains_Domain, false) < 0)
  93. return (ErrGetErrorCode());
  95. return (0);
  97. }
  99. char *MDomGetDomainPath(char const *pszDomain, char *pszDomainPath, int iMaxPath, int iFinalSlash)
  100. {
  101. ///////////////////////////////////////////////////////////////////////////////
  102. //  Make the domain lower-case
  103. ///////////////////////////////////////////////////////////////////////////////
  104. char szLoDomain[SYS_MAX_PATH] = "";
  106. StrSNCpy(szLoDomain, pszDomain);
  107. StrLower(szLoDomain);
  109. CfgGetRootPath(pszDomainPath, iMaxPath);
  111. StrNCat(pszDomainPath, MAIL_DOMAINS_DIR, iMaxPath);
  112. AppendSlash(pszDomainPath);
  113. StrNCat(pszDomainPath, szLoDomain, iMaxPath);
  115. if (iFinalSlash)
  116. AppendSlash(pszDomainPath);
  118. return (pszDomainPath);
  120. }
  122. static char *MDomGetDomainsFilePath(char *pszDomainsFilePath, int iMaxPath)
  123. {
  125. CfgGetRootPath(pszDomainsFilePath, iMaxPath);
  127. StrNCat(pszDomainsFilePath, MAIL_DOMAINS_FILE, iMaxPath);
  129. return (pszDomainsFilePath);
  131. }
  133. int MDomLookupDomain(char const *pszDomain)
  134. {
  136. char szDomainsFilePath[SYS_MAX_PATH] = "";
  138. MDomGetDomainsFilePath(szDomainsFilePath, sizeof(szDomainsFilePath));
  140. char szResLock[SYS_MAX_PATH] = "";
  141. RLCK_HANDLE hResLock = RLckLockSH(CfgGetBasedPath(szDomainsFilePath, szResLock,
  142.   sizeof(szResLock)));
  144. if (hResLock == INVALID_RLCK_HANDLE)
  145. return (ErrGetErrorCode());
  147. ///////////////////////////////////////////////////////////////////////////////
  148. //  Lookup record using the specified index
  149. ///////////////////////////////////////////////////////////////////////////////
  150. char **ppszTabTokens = TbixLookup(szDomainsFilePath, iIdxDomains_Domain, false,
  151.   pszDomain,
  152.   NULL);
  154. if (ppszTabTokens == NULL) {
  155. RLckUnlockSH(hResLock);
  157. ErrSetErrorCode(ERR_DOMAIN_NOT_HANDLED, pszDomain);
  158. return (ERR_DOMAIN_NOT_HANDLED);
  159. }
  161. StrFreeStrings(ppszTabTokens);
  163. RLckUnlockSH(hResLock);
  165. return (0);
  167. }
  169. int MDomAddDomain(char const *pszDomain)
  170. {
  172. char szDomainsFilePath[SYS_MAX_PATH] = "";
  174. MDomGetDomainsFilePath(szDomainsFilePath, sizeof(szDomainsFilePath));
  176. char szResLock[SYS_MAX_PATH] = "";
  177. RLCK_HANDLE hResLock = RLckLockEX(CfgGetBasedPath(szDomainsFilePath, szResLock,
  178.   sizeof(szResLock)));
  180. if (hResLock == INVALID_RLCK_HANDLE)
  181. return (ErrGetErrorCode());
  183. FILE *pDomainsFile = fopen(szDomainsFilePath, "r+t");
  185. if (pDomainsFile == NULL) {
  186. RLckUnlockEX(hResLock);
  188. ErrSetErrorCode(ERR_ALIAS_FILE_NOT_FOUND);
  189. return (ERR_ALIAS_FILE_NOT_FOUND);
  190. }
  192. char szDomainsLine[MAIL_DOMAINS_LINE_MAX] = "";
  194. while (MscFGets(szDomainsLine, sizeof(szDomainsLine) - 1, pDomainsFile) != NULL) {
  195. char **ppszStrings = StrGetTabLineStrings(szDomainsLine);
  197. if (ppszStrings == NULL)
  198. continue;
  200. int iFieldsCount = StrStringsCount(ppszStrings);
  202. if ((iFieldsCount >= domMax) && (stricmp(pszDomain, ppszStrings[domDomain]) == 0)) {
  203. StrFreeStrings(ppszStrings);
  204. fclose(pDomainsFile);
  205. RLckUnlockEX(hResLock);
  207. ErrSetErrorCode(ERR_DOMAIN_ALREADY_HANDLED);
  208. return (ERR_DOMAIN_ALREADY_HANDLED);
  209. }
  211. StrFreeStrings(ppszStrings);
  212. }
  214. fseek(pDomainsFile, 0, SEEK_END);
  216. fprintf(pDomainsFile, ""%s"n", pszDomain);
  218. fclose(pDomainsFile);
  220. ///////////////////////////////////////////////////////////////////////////////
  221. //  Rebuild indexes
  222. ///////////////////////////////////////////////////////////////////////////////
  223. if (MDomRebuildDomainsIndexes(szDomainsFilePath) < 0) {
  224. ErrorPush();
  225. RLckUnlockEX(hResLock);
  226. return (ErrorPop());
  227. }
  228. ///////////////////////////////////////////////////////////////////////////////
  229. //  Create domain directory
  230. ///////////////////////////////////////////////////////////////////////////////
  231. char szDomainPath[SYS_MAX_PATH] = "";
  233. MDomGetDomainPath(pszDomain, szDomainPath, sizeof(szDomainPath), 0);
  235. if (SysMakeDir(szDomainPath) < 0) {
  236. ErrorPush();
  237. RLckUnlockEX(hResLock);
  238. return (ErrorPop());
  239. }
  240. ///////////////////////////////////////////////////////////////////////////////
  241. //  Create cmd alias directory
  242. ///////////////////////////////////////////////////////////////////////////////
  243. if (USmlCreateCmdAliasDomainDir(pszDomain) < 0) {
  244. ErrorPush();
  245. RLckUnlockEX(hResLock);
  246. return (ErrorPop());
  247. }
  249. RLckUnlockEX(hResLock);
  251. return (0);
  253. }
  255. int MDomRemoveDomain(char const *pszDomain)
  256. {
  258. char szDomainsFilePath[SYS_MAX_PATH] = "";
  260. MDomGetDomainsFilePath(szDomainsFilePath, sizeof(szDomainsFilePath));
  262. char szTmpFile[SYS_MAX_PATH] = "";
  264. SysGetTmpFile(szTmpFile);
  266. char szResLock[SYS_MAX_PATH] = "";
  267. RLCK_HANDLE hResLock = RLckLockEX(CfgGetBasedPath(szDomainsFilePath, szResLock,
  268.   sizeof(szResLock)));
  270. if (hResLock == INVALID_RLCK_HANDLE) {
  271. ErrorPush();
  272. CheckRemoveFile(szTmpFile);
  273. return (ErrorPop());
  274. }
  276. FILE *pDomainsFile = fopen(szDomainsFilePath, "rt");
  278. if (pDomainsFile == NULL) {
  279. RLckUnlockEX(hResLock);
  280. CheckRemoveFile(szTmpFile);
  282. ErrSetErrorCode(ERR_DOMAINS_FILE_NOT_FOUND);
  283. return (ERR_DOMAINS_FILE_NOT_FOUND);
  284. }
  286. FILE *pTmpFile = fopen(szTmpFile, "wt");
  288. if (pTmpFile == NULL) {
  289. fclose(pDomainsFile);
  290. RLckUnlockEX(hResLock);
  291. CheckRemoveFile(szTmpFile);
  293. ErrSetErrorCode(ERR_FILE_CREATE);
  294. return (ERR_FILE_CREATE);
  295. }
  297. int iDomainsFound = 0;
  298. char szDomainsLine[MAIL_DOMAINS_LINE_MAX] = "";
  300. while (MscFGets(szDomainsLine, sizeof(szDomainsLine) - 1, pDomainsFile) != NULL) {
  301. char **ppszStrings = StrGetTabLineStrings(szDomainsLine);
  303. if (ppszStrings == NULL)
  304. continue;
  306. int iFieldsCount = StrStringsCount(ppszStrings);
  308. if ((iFieldsCount >= domMax) && (stricmp(pszDomain, ppszStrings[domDomain]) == 0)) {
  310. ++iDomainsFound;
  312. } else
  313. fprintf(pTmpFile, "%sn", szDomainsLine);
  315. StrFreeStrings(ppszStrings);
  316. }
  318. fclose(pDomainsFile);
  319. fclose(pTmpFile);
  321. if (iDomainsFound == 0) {
  322. SysRemove(szTmpFile);
  323. RLckUnlockEX(hResLock);
  325. ErrSetErrorCode(ERR_DOMAIN_NOT_HANDLED);
  326. return (ERR_DOMAIN_NOT_HANDLED);
  327. }
  329. char szTmpAliasFilePath[SYS_MAX_PATH] = "";
  331. sprintf(szTmpAliasFilePath, "%s.tmp", szDomainsFilePath);
  333. if (MscMoveFile(szDomainsFilePath, szTmpAliasFilePath) < 0) {
  334. ErrorPush();
  335. RLckUnlockEX(hResLock);
  336. return (ErrorPop());
  337. }
  339. if (MscMoveFile(szTmpFile, szDomainsFilePath) < 0) {
  340. ErrorPush();
  341. MscMoveFile(szTmpAliasFilePath, szDomainsFilePath);
  342. RLckUnlockEX(hResLock);
  343. return (ErrorPop());
  344. }
  346. SysRemove(szTmpAliasFilePath);
  348. ///////////////////////////////////////////////////////////////////////////////
  349. //  Rebuild indexes
  350. ///////////////////////////////////////////////////////////////////////////////
  351. if (MDomRebuildDomainsIndexes(szDomainsFilePath) < 0) {
  352. ErrorPush();
  353. RLckUnlockEX(hResLock);
  354. return (ErrorPop());
  355. }
  357. RLckUnlockEX(hResLock);
  359. ///////////////////////////////////////////////////////////////////////////////
  360. //  Domain cleanup
  361. ///////////////////////////////////////////////////////////////////////////////
  362. if (UsrRemoveDomainUsers(pszDomain) < 0)
  363. return (ErrGetErrorCode());
  365. if (UsrRemoveDomainAliases(pszDomain) < 0)
  366. return (ErrGetErrorCode());
  368. if (ExAlRemoveDomainAliases(pszDomain) < 0)
  369. return (ErrGetErrorCode());
  371. if (GwLkRemoveDomainLinks(pszDomain) < 0)
  372. return (ErrGetErrorCode());
  374. if (ADomRemoveLinkedDomains(pszDomain) < 0)
  375. return (ErrGetErrorCode());
  377. ///////////////////////////////////////////////////////////////////////////////
  378. //  Try ( if defined ) to drop external auth domain
  379. ///////////////////////////////////////////////////////////////////////////////
  380. UAthDropDomain(AUTH_SERVICE_POP3, pszDomain);
  382. ///////////////////////////////////////////////////////////////////////////////
  383. //  Directory cleanup
  384. ///////////////////////////////////////////////////////////////////////////////
  385. char szDomainPath[SYS_MAX_PATH] = "";
  387. MDomGetDomainPath(pszDomain, szDomainPath, sizeof(szDomainPath), 0);
  389. if (MscClearDirectory(szDomainPath) < 0)
  390. return (ErrGetErrorCode());
  392. if (SysRemoveDir(szDomainPath) < 0)
  393. return (ErrGetErrorCode());
  395. ///////////////////////////////////////////////////////////////////////////////
  396. //  Remove the cmd alias directory
  397. ///////////////////////////////////////////////////////////////////////////////
  398. if (USmlDeleteCmdAliasDomainDir(pszDomain) < 0)
  399. return (ErrGetErrorCode());
  401. return (0);
  403. }
  405. int MDomGetDomainsFileSnapShot(const char *pszFileName)
  406. {
  408. char szDomainsFilePath[SYS_MAX_PATH] = "";
  410. MDomGetDomainsFilePath(szDomainsFilePath, sizeof(szDomainsFilePath));
  412. char szResLock[SYS_MAX_PATH] = "";
  413. RLCK_HANDLE hResLock = RLckLockSH(CfgGetBasedPath(szDomainsFilePath, szResLock,
  414.   sizeof(szResLock)));
  416. if (hResLock == INVALID_RLCK_HANDLE)
  417. return (ErrGetErrorCode());
  419. if (MscCopyFile(pszFileName, szDomainsFilePath) < 0) {
  420. ErrorPush();
  421. RLckUnlockSH(hResLock);
  422. return (ErrorPop());
  423. }
  425. RLckUnlockSH(hResLock);
  427. return (0);
  429. }
  431. DOMLS_HANDLE MDomOpenDB(void)
  432. {
  434. DomainsScanData *pDSD = (DomainsScanData *) SysAlloc(sizeof(DomainsScanData));
  436. if (pDSD == NULL)
  437. return (INVALID_DOMLS_HANDLE);
  439. SysGetTmpFile(pDSD->szTmpDBFile);
  441. if (MDomGetDomainsFileSnapShot(pDSD->szTmpDBFile) < 0) {
  442. CheckRemoveFile(pDSD->szTmpDBFile);
  443. SysFree(pDSD);
  444. return (INVALID_DOMLS_HANDLE);
  445. }
  447. if ((pDSD->pDBFile = fopen(pDSD->szTmpDBFile, "rt")) == NULL) {
  448. SysRemove(pDSD->szTmpDBFile);
  449. SysFree(pDSD);
  450. return (INVALID_DOMLS_HANDLE);
  451. }
  453. return ((DOMLS_HANDLE) pDSD);
  455. }
  457. void MDomCloseDB(DOMLS_HANDLE hDomainsDB)
  458. {
  460. DomainsScanData *pDSD = (DomainsScanData *) hDomainsDB;
  462. fclose(pDSD->pDBFile);
  464. SysRemove(pDSD->szTmpDBFile);
  466. SysFree(pDSD);
  468. }
  470. char const *MDomGetFirstDomain(DOMLS_HANDLE hDomainsDB)
  471. {
  473. DomainsScanData *pDSD = (DomainsScanData *) hDomainsDB;
  475. rewind(pDSD->pDBFile);
  477. const char *pszDomain = NULL;
  478. char szDomainsLine[MAIL_DOMAINS_LINE_MAX] = "";
  480. while ((pszDomain == NULL) &&
  481.        (MscFGets(szDomainsLine, sizeof(szDomainsLine) - 1, pDSD->pDBFile) != NULL)) {
  482. char **ppszStrings = StrGetTabLineStrings(szDomainsLine);
  484. if (ppszStrings == NULL)
  485. continue;
  487. int iFieldsCount = StrStringsCount(ppszStrings);
  489. if (iFieldsCount >= domMax) {
  490. StrSNCpy(pDSD->szCurrDomain, ppszStrings[0]);
  492. pszDomain = pDSD->szCurrDomain;
  493. }
  495. StrFreeStrings(ppszStrings);
  496. }
  498. return (pszDomain);
  500. }
  502. char const *MDomGetNextDomain(DOMLS_HANDLE hDomainsDB)
  503. {
  505. DomainsScanData *pDSD = (DomainsScanData *) hDomainsDB;
  507. const char *pszDomain = NULL;
  508. char szDomainsLine[MAIL_DOMAINS_LINE_MAX] = "";
  510. while ((pszDomain == NULL) &&
  511.        (MscFGets(szDomainsLine, sizeof(szDomainsLine) - 1, pDSD->pDBFile) != NULL)) {
  512. char **ppszStrings = StrGetTabLineStrings(szDomainsLine);
  514. if (ppszStrings == NULL)
  515. continue;
  517. int iFieldsCount = StrStringsCount(ppszStrings);
  519. if (iFieldsCount >= domMax) {
  520. StrSNCpy(pDSD->szCurrDomain, ppszStrings[0]);
  522. pszDomain = pDSD->szCurrDomain;
  523. }
  525. StrFreeStrings(ppszStrings);
  526. }
  528. return (pszDomain);
  530. }
  532. int MDomGetClientDomain(char const *pszFQDN, char *pszClientDomain, int iMaxDomain)
  533. {
  535. for (; pszFQDN != NULL;) {
  536. if (MDomIsHandledDomain(pszFQDN) == 0) {
  537. StrNCpy(pszClientDomain, pszFQDN, iMaxDomain);
  539. return (0);
  540. }
  542. if ((pszFQDN = strchr(pszFQDN, '.')) != NULL)
  543. ++pszFQDN;
  544. }
  546. ErrSetErrorCode(ERR_NO_HANDLED_DOMAIN);
  547. return (ERR_NO_HANDLED_DOMAIN);
  549. }
  551. int MDomIsHandledDomain(char const *pszDomain)
  552. {
  553. ///////////////////////////////////////////////////////////////////////////////
  554. //  Check for alias domain
  555. ///////////////////////////////////////////////////////////////////////////////
  556. char szADomain[MAX_HOST_NAME] = "";
  558. if (ADomLookupDomain(pszDomain, szADomain, true))
  559. pszDomain = szADomain;
  561. return (MDomLookupDomain(pszDomain));
  563. }