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

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

Email服务器

开发平台：

Visual C++

CTRLClient.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 "BuffSock.h"
  27. #include "MiscUtils.h"
  28. #include "StrUtils.h"
  29. #include "MD5.h"
  30. #include "Errors.h"
  32. #define STD_CTRL_PORT               6017
  33. #define STD_CTRL_TIMEOUT            90
  34. #define CTRL_LISTFOLLOW_RESULT      100
  35. #define CTRL_WAITDATA_RESULT        101
  36. #define CCLN_ERR_BAD_USAGE          (-10000)
  38. ///////////////////////////////////////////////////////////////////////////////
  39. //  Needed by library functions ( START )
  40. ///////////////////////////////////////////////////////////////////////////////
  41. bool bServerDebug = false;
  42. int iLogRotateHours = 24;
  44. char *SvrGetLogsDir(char *pszLogsDir, int iMaxPath)
  45. {
  47. SysSNPrintf(pszLogsDir, iMaxPath - 1, ".");
  49. return (pszLogsDir);
  51. }
  53. ///////////////////////////////////////////////////////////////////////////////
  54. //  Needed by library functions ( END )
  55. ///////////////////////////////////////////////////////////////////////////////
  57. int CClnGetResponse(BSOCK_HANDLE hBSock, char *pszError, int iMaxError,
  58.     int *piErrorCode, int iTimeout)
  59. {
  61. char szRespBuffer[2048] = "";
  63. if (BSckGetString(hBSock, szRespBuffer, sizeof(szRespBuffer) - 1, iTimeout) == NULL)
  64. return (ErrGetErrorCode());
  66. if ((szRespBuffer[0] != '+') && (szRespBuffer[0] != '-')) {
  67. ErrSetErrorCode(ERR_CCLN_INVALID_RESPONSE, szRespBuffer);
  68. return (ERR_CCLN_INVALID_RESPONSE);
  69. }
  71. char *pszToken = szRespBuffer + 1;
  73. if (piErrorCode != NULL)
  74. *piErrorCode = atoi(pszToken) * ((szRespBuffer[0] == '-') ? -1 : +1);
  76. for (; isdigit(*pszToken); pszToken++);
  78. if (*pszToken != ' ') {
  79. ErrSetErrorCode(ERR_CCLN_INVALID_RESPONSE, szRespBuffer);
  80. return (ERR_CCLN_INVALID_RESPONSE);
  81. }
  83. for (; *pszToken == ' '; pszToken++);
  85. if (pszError != NULL) {
  86. strncpy(pszError, pszToken, iMaxError);
  87. pszError[iMaxError - 1] = '';
  88. }
  90. return (0);
  92. }
  94. int CClnRecvTextFile(const char *pszFileName, BSOCK_HANDLE hBSock, int iTimeout)
  95. {
  97. bool bCloseFile = false;
  98. FILE *pFile = stdout;
  100. if (pszFileName != NULL) {
  101. if ((pFile = fopen(pszFileName, "wt")) == NULL) {
  102. ErrSetErrorCode(ERR_FILE_CREATE, pszFileName);
  103. return (ERR_FILE_CREATE);
  104. }
  106. bCloseFile = true;
  107. }
  109. char szBuffer[2048] = "";
  111. while (BSckGetString(hBSock, szBuffer, sizeof(szBuffer) - 1, iTimeout) != NULL) {
  112. if (strcmp(szBuffer, ".") == 0)
  113. break;
  115. if (szBuffer[0] == '.')
  116. fprintf(pFile, "%sn", szBuffer + 1);
  117. else
  118. fprintf(pFile, "%sn", szBuffer);
  119. }
  121. if (bCloseFile)
  122. fclose(pFile);
  124. return (0);
  126. }
  128. int CClnSendTextFile(const char *pszFileName, BSOCK_HANDLE hBSock, int iTimeout)
  129. {
  131. bool bCloseFile = false;
  132. FILE *pFile = stdin;
  134. if (pszFileName != NULL) {
  135. if ((pFile = fopen(pszFileName, "rt")) == NULL) {
  136. ErrSetErrorCode(ERR_FILE_OPEN, pszFileName);
  137. return (ERR_FILE_OPEN);
  138. }
  140. bCloseFile = true;
  141. }
  143. char szBuffer[2048] = "";
  145. while (MscFGets(szBuffer, sizeof(szBuffer) - 1, pFile) != NULL) {
  146. if (szBuffer[0] == '.')
  147. for (int ii = strlen(szBuffer); ii >= 0; ii--)
  148. szBuffer[ii + 1] = szBuffer[ii];
  150. if (BSckSendString(hBSock, szBuffer, iTimeout) <= 0) {
  151. fclose(pFile);
  152. return (ErrGetErrorCode());
  153. }
  154. }
  156. if (bCloseFile)
  157. fclose(pFile);
  159. return (BSckSendString(hBSock, ".", iTimeout));
  161. }
  163. int CClnSubmitCommand(BSOCK_HANDLE hBSock, char const *pszCommand,
  164.       char *pszError, int iMaxError, char const *pszIOFile, int iTimeout)
  165. {
  167. if (BSckSendString(hBSock, pszCommand, iTimeout) < 0)
  168. return (ErrGetErrorCode());
  170. int iErrorCode = 0;
  172. if (CClnGetResponse(hBSock, pszError, iMaxError, &iErrorCode, iTimeout) < 0)
  173. return (ErrGetErrorCode());
  175. if (iErrorCode < 0) {
  176. ErrSetErrorCode(ERR_CCLN_ERROR_RESPONSE, pszError);
  177. return (ERR_CCLN_ERROR_RESPONSE);
  178. }
  180. if (iErrorCode == CTRL_LISTFOLLOW_RESULT) {
  182. if (CClnRecvTextFile(pszIOFile, hBSock, iTimeout) < 0)
  183. return (ErrGetErrorCode());
  185. } else if (iErrorCode == CTRL_WAITDATA_RESULT) {
  187. if (CClnSendTextFile(pszIOFile, hBSock, iTimeout) < 0)
  188. return (ErrGetErrorCode());
  190. if (CClnGetResponse(hBSock, pszError, iMaxError, &iErrorCode, iTimeout) < 0)
  191. return (ErrGetErrorCode());
  193. }
  195. return (0);
  197. }
  199. BSOCK_HANDLE CClnConnectServer(char const *pszServer, int iPortNo,
  200.        char const *pszUsername, char const *pszPassword,
  201.        bool bUseMD5Auth, int iTimeout)
  202. {
  203. ///////////////////////////////////////////////////////////////////////////////
  204. //  Get server address
  205. ///////////////////////////////////////////////////////////////////////////////
  206. SYS_INET_ADDR SvrAddr;
  208. if (MscGetServerAddress(pszServer, SvrAddr, iPortNo) < 0)
  209. return (INVALID_BSOCK_HANDLE);
  211. ///////////////////////////////////////////////////////////////////////////////
  212. //  Try connect to server
  213. ///////////////////////////////////////////////////////////////////////////////
  214. SYS_SOCKET SockFD = SysCreateSocket(AF_INET, SOCK_STREAM, 0);
  216. if (SockFD == SYS_INVALID_SOCKET)
  217. return (INVALID_BSOCK_HANDLE);
  219. if (SysConnect(SockFD, &SvrAddr, sizeof(SvrAddr), iTimeout) < 0) {
  220. SysCloseSocket(SockFD);
  221. return (INVALID_BSOCK_HANDLE);
  222. }
  224. BSOCK_HANDLE hBSock = BSckAttach(SockFD);
  226. if (hBSock == INVALID_BSOCK_HANDLE) {
  227. SysCloseSocket(SockFD);
  228. return (INVALID_BSOCK_HANDLE);
  229. }
  230. ///////////////////////////////////////////////////////////////////////////////
  231. //  Read welcome message
  232. ///////////////////////////////////////////////////////////////////////////////
  233. int iErrorCode = 0;
  234. char szRTXBuffer[2048] = "";
  236. if (CClnGetResponse(hBSock, szRTXBuffer, sizeof(szRTXBuffer), &iErrorCode, iTimeout) < 0) {
  237. BSckDetach(hBSock, 1);
  238. return (INVALID_BSOCK_HANDLE);
  239. }
  241. if (iErrorCode < 0) {
  242. BSckDetach(hBSock, 1);
  244. ErrSetErrorCode(ERR_CCLN_ERROR_RESPONSE, szRTXBuffer);
  245. return (INVALID_BSOCK_HANDLE);
  246. }
  247. ///////////////////////////////////////////////////////////////////////////////
  248. //  Prepare login
  249. ///////////////////////////////////////////////////////////////////////////////
  250. char szTimeStamp[256] = "";
  252. if (!bUseMD5Auth ||
  253.     (MscExtractServerTimeStamp(szRTXBuffer, szTimeStamp, sizeof(szTimeStamp)) == NULL))
  254. sprintf(szRTXBuffer, ""%s"t"%s"", pszUsername, pszPassword);
  255. else {
  256. ///////////////////////////////////////////////////////////////////////////////
  257. //  Perform MD5 authentication
  258. ///////////////////////////////////////////////////////////////////////////////
  259. char *pszHash = StrSprint("%s%s", szTimeStamp, pszPassword);
  261. if (pszHash == NULL) {
  262. BSckDetach(hBSock, 1);
  263. return (INVALID_BSOCK_HANDLE);
  264. }
  266. char szMD5[128] = "";
  268. do_md5_string(pszHash, strlen(pszHash), szMD5);
  270. SysFree(pszHash);
  272. ///////////////////////////////////////////////////////////////////////////////
  273. //  Add a  #  char in head of password field
  274. ///////////////////////////////////////////////////////////////////////////////
  275. sprintf(szRTXBuffer, ""%s"t"#%s"", pszUsername, szMD5);
  276. }
  278. ///////////////////////////////////////////////////////////////////////////////
  279. //  Send login
  280. ///////////////////////////////////////////////////////////////////////////////
  281. if (BSckSendString(hBSock, szRTXBuffer, iTimeout) < 0) {
  282. BSckDetach(hBSock, 1);
  283. return (INVALID_BSOCK_HANDLE);
  284. }
  286. if (CClnGetResponse(hBSock, szRTXBuffer, sizeof(szRTXBuffer), &iErrorCode, iTimeout) < 0) {
  287. BSckDetach(hBSock, 1);
  288. return (INVALID_BSOCK_HANDLE);
  289. }
  291. if (iErrorCode < 0) {
  292. BSckDetach(hBSock, 1);
  294. ErrSetErrorCode(ERR_CCLN_ERROR_RESPONSE, szRTXBuffer);
  295. return (INVALID_BSOCK_HANDLE);
  296. }
  298. return (hBSock);
  300. }
  302. int CClnQuitConnection(BSOCK_HANDLE hBSock, int iTimeout)
  303. {
  305. CClnSubmitCommand(hBSock, ""quit"", NULL, 0, NULL, iTimeout);
  307. BSckDetach(hBSock, 1);
  309. return (0);
  311. }
  313. int CClnLogError(void)
  314. {
  316. char *pszError = ErrGetErrorStringInfo(ErrGetErrorCode());
  318. if (pszError == NULL)
  319. return (ErrGetErrorCode());
  321. fprintf(stderr, "%sn", pszError);
  323. SysFree(pszError);
  325. return (0);
  327. }
  329. void CClnShowUsage(char const *pszProgName)
  330. {
  332. fprintf(stderr,
  333. "use :  %s  [-snuptfc]  ...n"
  334. "options :n"
  335. "       -s server        = set server addressn"
  336. "       -n port          = set server port [%d]n"
  337. "       -u user          = set usernamen"
  338. "       -p pass          = set passwordn"
  339. "       -t timeout       = set timeout [%d]n"
  340. "       -f filename      = set I/O filename [stdin/stdout]n"
  341. "       -c               = disable MD5 authenticationn",
  342. pszProgName, STD_CTRL_PORT, STD_CTRL_TIMEOUT);
  344. }
  346. int CClnExec(int iArgCount, char *pszArgs[])
  347. {
  349. int ii;
  350. int iPortNo = STD_CTRL_PORT;
  351. int iTimeout = STD_CTRL_TIMEOUT;
  352. bool bUseMD5Auth = true;
  353. char szServer[MAX_HOST_NAME] = "";
  354. char szUsername[256] = "";
  355. char szPassword[256] = "";
  356. char szIOFile[SYS_MAX_PATH] = "";
  358. for (ii = 1; ii < iArgCount; ii++) {
  359. if (pszArgs[ii][0] != '-')
  360. break;
  362. switch (pszArgs[ii][1]) {
  363. case ('s'):
  364. if (++ii < iArgCount)
  365. StrSNCpy(szServer, pszArgs[ii]);
  366. break;
  368. case ('n'):
  369. if (++ii < iArgCount)
  370. iPortNo = atoi(pszArgs[ii]);
  371. break;
  373. case ('u'):
  374. if (++ii < iArgCount)
  375. StrSNCpy(szUsername, pszArgs[ii]);
  376. break;
  378. case ('p'):
  379. if (++ii < iArgCount)
  380. StrSNCpy(szPassword, pszArgs[ii]);
  381. break;
  383. case ('t'):
  384. if (++ii < iArgCount)
  385. iTimeout = atoi(pszArgs[ii]);
  386. break;
  388. case ('f'):
  389. if (++ii < iArgCount)
  390. StrSNCpy(szIOFile, pszArgs[ii]);
  391. break;
  393. case ('c'):
  394. bUseMD5Auth = false;
  395. break;
  397. default:
  398. return (CCLN_ERR_BAD_USAGE);
  399. }
  400. }
  402. if ((strlen(szServer) == 0) || (strlen(szUsername) == 0) ||
  403.     (strlen(szPassword) == 0) || (ii == iArgCount))
  404. return (CCLN_ERR_BAD_USAGE);
  406. int iFirstParam = ii;
  407. int iCmdLength = 0;
  409. for (; ii < iArgCount; ii++)
  410. iCmdLength += strlen(pszArgs[ii]) + 4;
  412. char *pszCommand = (char *) SysAlloc(iCmdLength + 1);
  414. if (pszCommand == NULL)
  415. return (ErrGetErrorCode());
  417. for (ii = iFirstParam; ii < iArgCount; ii++) {
  418. if (ii == iFirstParam)
  419. sprintf(pszCommand, ""%s"", pszArgs[ii]);
  420. else
  421. sprintf(pszCommand + strlen(pszCommand), "t"%s"", pszArgs[ii]);
  422. }
  424. BSOCK_HANDLE hBSock = CClnConnectServer(szServer, iPortNo, szUsername, szPassword,
  425. bUseMD5Auth, iTimeout);
  427. if (hBSock == INVALID_BSOCK_HANDLE) {
  428. ErrorPush();
  429. SysFree(pszCommand);
  430. return (ErrorPop());
  431. }
  433. char szRTXBuffer[2048] = "";
  435. if (CClnSubmitCommand(hBSock, pszCommand, szRTXBuffer, sizeof(szRTXBuffer),
  436.       (strlen(szIOFile) != 0) ? szIOFile : NULL, iTimeout) < 0) {
  437. ErrorPush();
  438. CClnQuitConnection(hBSock, iTimeout);
  439. SysFree(pszCommand);
  440. return (ErrorPop());
  441. }
  443. SysFree(pszCommand);
  445. CClnQuitConnection(hBSock, iTimeout);
  447. return (0);
  449. }
  451. #ifndef __CTRLCLNT_LIBRARY__
  453. int main(int iArgCount, char *pszArgs[])
  454. {
  456. if (SysInitLibrary() < 0) {
  457. CClnLogError();
  458. return (1);
  459. }
  461. int iExecResult = CClnExec(iArgCount, pszArgs);
  463. if (iExecResult == CCLN_ERR_BAD_USAGE) {
  464. CClnShowUsage(pszArgs[0]);
  465. SysCleanupLibrary();
  466. return (2);
  467. } else if (iExecResult < 0) {
  468. CClnLogError();
  469. SysCleanupLibrary();
  470. return (3);
  471. }
  473. SysCleanupLibrary();
  475. return (0);
  477. }
  479. #endif // #ifndef __CTRLCLNT_LIBRARY__