jizhao的qmail-smtpd.c修改版(增加本地域之间的验证)

artxing

jizhao的qmail-smtpd.c,很多人说没有本地域之间验证的问题,我之前叫大家从gadfly的补丁里找出本地验证的部分加上就行了.可大家好象还是有问题.那我自己来把它加上吧.大家试一下,效果如何.

1. #include "sig.h"
   
2. #include "readwrite.h"
   
3. #include "stralloc.h"
   
4. #include "substdio.h"
   
5. #include "alloc.h"
   
6. #include "auto_qmail.h"
   
7. #include "control.h"
   
8. #include "received.h"
   
9. #include "constmap.h"
   
10. #include "error.h"
    
11. #include "ipme.h"
    
12. #include "ip.h"
    
13. #include "qmail.h"
    
14. #include "str.h"
    
15. #include "fmt.h"
    
16. #include "scan.h"
    
17. #include "byte.h"
    
18. #include "case.h"
    
19. #include "env.h"
    
20. #include "now.h"
    
21. #include "exit.h"
    
22. #include "rcpthosts.h"
    
23. #include "timeoutread.h"
    
24. #include "timeoutwrite.h"
    
25. #include "commands.h"
    
26. #include "wait.h"
    
27. #include "fd.h"
    
28. 
    
29. /*******Added by Jizhao******/
    
30. #include <stdio.h>;
    
31. #include <regex.h>;
    
32. #include <unistd.h>;
    
33. #include <sys/types.h>;
    
34. #include <limits.h>;
    
35. #include <sys/stat.h>;
    
36. #include <fcntl.h>;
    
37. #include <time.h>;
    
38. #include <string.h>;
    
39. /*******Added by Jizhao******/
    
40. 
    
41. #define AUTHCRAM
    
42. #define MAXHOPS 100
    
43. unsigned int databytes = 0;
    
44. int timeout = 1200;
    
45. 
    
46. /************************* Added by Jizhao*********************************/
    
47. /*
    
48. 全局变量用于记录认证的email地址
    
49. 如果是虚拟域的用户则是邮件地址
    
50. 如果是主域用户则在函数authenticate
    
51. 里面赋值后，需加上defaultdomain里面的内容
    
52. */
    
53. stralloc authusername={0};
    
54. stralloc completeusername={0};
    
55. /*
    
56. 如果是远程邮件服务器连接本地的
    
57. smtp,则远程主机使用的是不带认证
    
58. 的连接方式，即在连接建立以后，
    
59. 远程主机使用的是helo命令，这样
    
60. 远程主机就只能向本机发送本地域
    
61. 的用户，因为如果用ehlo的话，远程
    
62. 主机不可能知道每个帐户的密码
    
63. ehlo 记录，smtp连接使用的是
    
64. helo还是ehlo
    
65. 0:helo 1:ehlo
    
66. */
    
67. int ehlo = 0;
    
68. /*
    
69. 检查邮件地址只能由字母、数字、下划线、减号
    
70. 和点组成，其余均视为为非法地址
    
71. */
    
72. #define SUBSLEN 20
    
73. #define EBUFLEN 128 /* error buffer length */
    
74. 
    
75. int CheckMailAddress(char * mail, char * msg)
    
76. {
    
77. size_t len; /* store error message length */
    
78. regex_t re; /* store compilned regular expression */
    
79. regmatch_t subs[SUBSLEN]; /* store matched string position */
    
80. char errbuf[EBUFLEN]; /* store error message */
    
81. int err, i;
    
82. char pattern[] = "([a-zA-Z\\.0-9_-])+@([a-zA-Z\\.0-9_-])+";
    
83. 
    
84. /* compile regular expression */
    
85. err = regcomp (&re, pattern, REG_EXTENDED);
    
86. if (err)
    
87. {
    
88. len = regerror (err, &re, errbuf, sizeof (errbuf));
    
89. sprintf(msg, "error: regcomp: %s\r\n", errbuf);
    
90. return 0;
    
91. }
    
92. 
    
93. /* execute pattern match */
    
94. err = regexec (&re, mail, (size_t)SUBSLEN, subs, 0);
    
95. if (err == REG_NOMATCH)
    
96. {
    
97. sprintf (msg, "Invalid mail address\r\n");
    
98. regfree (&re);
    
99. return 0;
    
100. }
     
101. else if (err)
     
102. {
     
103. len = regerror (err, &re, errbuf, sizeof (errbuf));
     
104. sprintf(msg, "error: regexec: %s\r\n", errbuf);
     
105. return 0;
     
106. }
     
107. regfree(&re);
     
108. if (subs[0].rm_so == 0 && subs[0].rm_eo == strlen(mail))
     
109. {
     
110. return 1;
     
111. }
     
112. sprintf (msg, "Invalid mail address\r\n");
     
113. return 0;
     
114. }
     
115. /*
     
116. 初始设置：每天最多可发送的邮件数
     
117. 具体每个那个用户每天可发信的数量
     
118. 限制可在该用户目录下的mailsperday
     
119. 文件中的第二个域修改
     
120. */
     
121. #define MAX_MAILS_PERDAY 150
     
122. /*
     
123. user: 用户完整mail地址
     
124. dir : 返回用户的目录，如果用户存在
     
125. 返回值：如果发生错误返回 -1，
     
126. 无此用户返回 -2,
     
127. 否则返回0
     
128. */
     
129. int getuserdir(const char *user, char * dir)
     
130. {
     
131. FILE * fp = NULL;
     
132. char command[150];
     
133. int len;
     
134. 
     
135. if (strlen(user) >; 97) return 0;
     
136. sprintf(command, "/home/vpopmail/bin/vuserinfo %s -d", user);
     
137. fp = popen(command, "r");
     
138. if (fp ==NULL) return -1;
     
139. len = read(fileno(fp),dir,PATH_MAX);
     
140. if (len == -1) return -1;
     
141. if (strstr(dir, "no such user") != 0) return -2; //无此帐户
     
142. if (fp != NULL) pclose(fp);
     
143. return 0;
     
144. }
     
145. /*
     
146. 在每个用户的目录下有一个文件mailsperday记录
     
147. 当天该用户利用smtp和web mail发信的情况,该文
     
148. 件只有一行，分为4个域，每个域宽度为10个字符，
     
149. 域之间用一个空格分开，所以改行总长度为43
     
150. 
     
151. 记录日期(10个字符)
     
152. 每天最多可发送邮件的数目
     
153. 记录当天通过smtp已发信的数目
     
154. 记录当天通过webmail已发送邮件的数目
     
155. 
     
156. 函数返回值：
     
157. 0: 通过发信次数检查
     
158. -1: 内存分配失败
     
159. -2: 文件操作失败
     
160. -3: 发信次数已达限制，不能发信
     
161. */
     
162. 
     
163. int checkmaxmailsperday(const char* dir/*要检查的用户的目录*/)
     
164. {
     
165. int fd, len;
     
166. char info[50];
     
167. char temp[50];
     
168. short year, month, day;
     
169. short limit, smtp, webmail;
     
170. char *path;
     
171. time_t cur_time;
     
172. struct tm *cur_tm;
     
173. char filename[]="/mailsperday";
     
174. int i=0;
     
175. 
     
176. /*给path分配空间，并将path置为有完整路径的文件名*/
     
177. len = strlen(dir);
     
178. path = (char*)calloc(strlen(dir)+13, 1);
     
179. if (path == 0) return -1;
     
180. strcpy(path, dir);
     
181. /*在这里不知道为什么strcat不好用*/
     
182. for(i=len-1;i<len+11;i++)
     
183. path[i] = filename[i-len+1];
     
184. path[i] = '\0';
     
185. 
     
186. if ((fd = open(path,O_CREAT|O_RDWR, S_IRUSR|S_IWUSR)) == -1) return -2;
     
187. if ((len = read(fd, info, 43)) == -1) return -2;
     
188. 
     
189. 
     
190. cur_time = time(NULL);
     
191. cur_tm = localtime(&cur_time);
     
192. year = cur_tm->;tm_year + 1900;
     
193. month = cur_tm->;tm_mon + 1;
     
194. day = cur_tm->;tm_mday;
     
195. limit = MAX_MAILS_PERDAY;
     
196. smtp = 0;
     
197. webmail = 0;
     
198. 
     
199. /*如果读到了43个字符，说明文件已存在，
     
200. 不是刚刚创建出来的，那么提取文件中的
     
201. 数据
     
202. */
     
203. if (len >;= 43)
     
204. {
     
205. strncpy(temp, info, 4);
     
206. year = atoi(temp);
     
207. 
     
208. memset(temp, 0, 50);
     
209. strncpy(temp, info+5, 2);
     
210. month = atoi(temp);
     
211. 
     
212. memset(temp, 0, 50);
     
213. strncpy(temp, info+8, 2);
     
214. day = atoi(temp);
     
215. 
     
216. memset(temp, 0, 50);
     
217. strncpy(temp, info+11, 10);
     
218. limit = atoi(temp);
     
219. 
     
220. memset(temp, 0, 50);
     
221. strncpy(temp, info+22, 10);
     
222. smtp = atoi(temp);
     
223. 
     
224. memset(temp, 0, 50);
     
225. strncpy(temp, info+33, 10);
     
226. webmail = atoi(temp);
     
227. 
     
228. /*
     
229. 如果文件中记录的时间不是今天，
     
230. 则置smtp和webmail的值为0，
     
231. 时间置为今天
     
232. */
     
233. if (!(year == cur_tm->;tm_year+1900 && month == cur_tm->;tm_mon+1
     
234. && day == cur_tm->;tm_mday))
     
235. {
     
236. year = cur_tm->;tm_year + 1900;
     
237. month = cur_tm->;tm_mon + 1;
     
238. day = cur_tm->;tm_mday;
     
239. smtp = 0;
     
240. webmail = 0;
     
241. }
     
242. }
     
243. 
     
244. /*判断发信是否超出次数限制*/
     
245. if (smtp + webmail + 1 >; limit)
     
246. {
     
247. return -3; /*发信次数超过限制*/
     
248. }
     
249. else
     
250. {
     
251. smtp += 1;
     
252. memset(temp, 0, 50);
     
253. sprintf(temp, "%4d/%2d/%2d %10d %10d %10d\n",
     
254. year, month, day, limit, smtp, webmail);
     
255. lseek(fd, 0, SEEK_SET);
     
256. len = write(fd, temp, 44);
     
257. if (len != 44) return -2;
     
258. }
     
259. close(fd);
     
260. if (path != 0 ) free(path);
     
261. return 0;
     
262. }
     
263. 
     
264. /************************* Added by Jizhao*********************************/
     
265. 
     
266. 
     
267. int safewrite(fd,buf,len) int fd; char *buf; int len;
     
268. {
     
269.   int r;
     
270.   r = timeoutwrite(timeout,fd,buf,len);
     
271.   if (r <= 0) _exit(1);
     
272.   return r;
     
273. }
     
274. 
     
275. char ssoutbuf[512];
     
276. substdio ssout = SUBSTDIO_FDBUF(safewrite,1,ssoutbuf,sizeof ssoutbuf);
     
277. 
     
278. void flush() { substdio_flush(&ssout); }
     
279. void out(s) char *s; { substdio_puts(&ssout,s); }
     
280. 
     
281. void die_read() { _exit(1); }
     
282. void die_alarm() { out("451 timeout (#4.4.2)\r\n"); flush(); _exit(1); }
     
283. void die_nomem() { out("421 out of memory (#4.3.0)\r\n"); flush(); _exit(1); }
     
284. void die_control() { out("421 unable to read controls (#4.3.0)\r\n"); flush(); _exit(1); }
     
285. void die_ipme() { out("421 unable to figure out my IP addresses (#4.3.0)\r\n"); flush(); _exit(1); }
     
286. void straynewline() { out("451 See http://pobox.com/~djb/docs/smtplf.html.\r\n"); flush(); _exit(1); }
     
287. 
     
288. void err_bmf() { out("553 sorry, your envelope sender is in my badmailfrom list (#5.7.1)\r\n"); }
     
289. void err_nogateway() { out("553 sorry, that domain isn't in my list of allowed rcpthosts (#5.7.1)\r\n"); }
     
290. void err_unimpl() { out("502 unimplemented (#5.5.1)\r\n"); }
     
291. void err_syntax() { out("555 syntax error (#5.5.4)\r\n"); }
     
292. void err_wantmail() { out("503 MAIL first (#5.5.1)\r\n"); }
     
293. void err_wantrcpt() { out("503 RCPT first (#5.5.1)\r\n"); }
     
294. void err_noop() { out("250 ok\r\n"); }
     
295. void err_vrfy() { out("252 send some mail, i'll try my best\r\n"); }
     
296. void err_qqt() { out("451 qqt failure (#4.3.0)\r\n"); }
     
297. 
     
298. int err_child() { out("454 oops, problem with child and I can't auth (#4.3.0)\r\n"); return -1; }
     
299. int err_fork() { out("454 oops, child won't start and I can't auth (#4.3.0)\r\n"); return -1; }
     
300. int err_pipe() { out("454 oops, unable to open pipe and I can't auth (#4.3.0)\r\n"); return -1; }
     
301. int err_write() { out("454 oops, unable to write pipe and I can't auth (#4.3.0)\r\n"); return -1; }
     
302. void err_authd() { out("503 you're already authenticated (#5.5.0)\r\n"); }
     
303. void err_authmail() { out("503 no auth during mail transaction (#5.5.0)\r\n"); }
     
304. int err_noauth() { out("504 auth type unimplemented (#5.5.1)\r\n"); return -1; }
     
305. int err_authabrt() { out("501 auth exchange cancelled (#5.0.0)\r\n"); return -1; }
     
306. int err_input() { out("501 malformed auth input (#5.5.4)\r\n"); return -1; }
     
307. 
     
308. /************************* Added by Jizhao*********************************/
     
309. void outsafe(stralloc *sa)
     
310. {
     
311. int i;
     
312. 
     
313. char ch[200];
     
314. for (i = 0;i < sa->;len;++i)
     
315. {
     
316. ch[i] = sa->;s[i]; if (ch[i] < 33) ch[i] = '?'; if (ch[i] >; 126) ch[i] = '?';
     
317. }
     
318. ch[i-1]='\0';
     
319. out(ch);
     
320. }
     
321. 
     
322. /*
     
323. 处理用户认证的用户名为完整形式，即
     
324. username@domain
     
325. */
     
326. void complete_authusername()
     
327. {
     
328. stralloc temp = {0}, mydefaultdomain={0};
     
329. int isaddress = 0; /*如果为0，说明需要加上defaultdomain,否则不需要加*/
     
330. int len = 0, i = 0;
     
331. 
     
332. len = authusername.len;
     
333. for (i = 0; i < len; i++)
     
334. {
     
335. if (authusername.s[i] == '@')
     
336. {
     
337. isaddress = 1;
     
338. break;
     
339. }
     
340. }
     
341. 
     
342. if (isaddress == 0)
     
343. {
     
344. if (control_readfile(&mydefaultdomain,"/var/qmail/control/defaultdomain",1) != 1)
     
345. {
     
346. die_nomem();
     
347. }
     
348. /*
     
349. 由于从defaultdomain中读出的字符串末尾有
     
350. 一个特殊字符，所以要先去掉它，再放到放到
     
351. mydefaultdomain里面去
     
352. */
     
353. if (!stralloc_copyb(&completeusername, authusername.s, authusername.len-1))
     
354. {
     
355. die_nomem();
     
356. }
     
357. if (!stralloc_cats(&completeusername, "@") )
     
358. {
     
359. die_nomem();
     
360. }
     
361. if (!stralloc_cat(&completeusername, &mydefaultdomain))
     
362. {
     
363. die_nomem();
     
364. }
     
365. }
     
366. else
     
367. {
     
368. if (!stralloc_cat(&completeusername, &authusername))
     
369. {
     
370. die_nomem();
     
371. }
     
372. }
     
373. }
     
374. 
     
375. /*********************************Added by jizhao**************************/
     
376. 
     
377. 
     
378. stralloc greeting = {0};
     
379. 
     
380. void smtp_greet(code) char *code;
     
381. {
     
382.   substdio_puts(&ssout,code);
     
383.   substdio_put(&ssout,greeting.s,greeting.len);
     
384. }
     
385. void smtp_help()
     
386. {
     
387.   out("214 qmail home page: http://pobox.com/~djb/qmail.html\r\n");
     
388. }
     
389. void smtp_quit()
     
390. {
     
391.   smtp_greet("221 "); out("\r\n"); flush(); _exit(0);
     
392. }
     
393. 
     
394. char *remoteip;
     
395. char *remotehost;
     
396. char *remoteinfo;
     
397. char *local;
     
398. char *relayclient;
     
399. 
     
400. stralloc helohost = {0};
     
401. char *fakehelo; /* pointer into helohost, or 0 */
     
402. 
     
403. void dohelo(arg) char *arg; {
     
404.   if (!stralloc_copys(&helohost,arg)) die_nomem();
     
405.   if (!stralloc_0(&helohost)) die_nomem();
     
406.   fakehelo = case_diffs(remotehost,helohost.s) ? helohost.s : 0;
     
407. }
     
408. 
     
409. int liphostok = 0;
     
410. stralloc liphost = {0};
     
411. int bmfok = 0;
     
412. stralloc bmf = {0};
     
413. struct constmap mapbmf;
     
414. 
     
415. void setup()
     
416. {
     
417.   char *x;
     
418.   unsigned long u;
     
419. 
     
420.   if (control_init() == -1) die_control();
     
421.   if (control_rldef(&greeting,"control/smtpgreeting",1,(char *) 0) != 1)
     
422.     die_control();
     
423.   liphostok = control_rldef(&liphost,"control/localiphost",1,(char *) 0);
     
424.   if (liphostok == -1) die_control();
     
425.   if (control_readint(&timeout,"control/timeoutsmtpd") == -1) die_control();
     
426.   if (timeout <= 0) timeout = 1;
     
427. 
     
428.   if (rcpthosts_init() == -1) die_control();
     
429. 
     
430.   bmfok = control_readfile(&bmf,"control/badmailfrom",0);
     
431.   if (bmfok == -1) die_control();
     
432.   if (bmfok)
     
433.     if (!constmap_init(&mapbmf,bmf.s,bmf.len,0)) die_nomem();
     
434. 
     
435.   if (control_readint(&databytes,"control/databytes") == -1) die_control();
     
436.   x = env_get("DATABYTES");
     
437.   if (x) { scan_ulong(x,&u); databytes = u; }
     
438.   if (!(databytes + 1)) --databytes;
     
439. 
     
440.   remoteip = env_get("TCPREMOTEIP");
     
441.   if (!remoteip) remoteip = "unknown";
     
442.   local = env_get("TCPLOCALHOST");
     
443.   if (!local) local = env_get("TCPLOCALIP");
     
444.   if (!local) local = "unknown";
     
445.   remotehost = env_get("TCPREMOTEHOST");
     
446.   if (!remotehost) remotehost = "unknown";
     
447.   remoteinfo = env_get("TCPREMOTEINFO");
     
448.   relayclient = env_get("RELAYCLIENT");
     
449.   dohelo(remotehost);
     
450. }
     
451. 
     
452. 
     
453. stralloc addr = {0}; /* will be 0-terminated, if addrparse returns 1 */
     
454. 
     
455. int addrparse(arg)
     
456. char *arg;
     
457. {
     
458.   int i;
     
459.   char ch;
     
460.   char terminator;
     
461.   struct ip_address ip;
     
462.   int flagesc;
     
463.   int flagquoted;
     
464.   
     
465. /********Added by Jizhao****/
     
466. char error[150];
     
467. /********Added by Jizhao****/
     
468.   
     
469. 
     
470.   terminator = '>;';
     
471.   i = str_chr(arg,'<');
     
472.   if (arg[i])
     
473.     arg += i + 1;
     
474.   else { /* partner should go read rfc 821 */
     
475.     terminator = ' ';
     
476.     arg += str_chr(arg,':');
     
477.     if (*arg == ':') ++arg;
     
478.     while (*arg == ' ') ++arg;
     
479.   }
     
480. 
     
481.   /* strip source route */
     
482.   if (*arg == '@') while (*arg) if (*arg++ == ':') break;
     
483. 
     
484.   if (!stralloc_copys(&addr,"")) die_nomem();
     
485.   flagesc = 0;
     
486.   flagquoted = 0;
     
487.   for (i = 0;ch = arg[i];++i) { /* copy arg to addr, stripping quotes */
     
488.     if (flagesc) {
     
489.       if (!stralloc_append(&addr,&ch)) die_nomem();
     
490.       flagesc = 0;
     
491.     }
     
492.     else {
     
493.       if (!flagquoted && (ch == terminator)) break;
     
494.       switch(ch) {
     
495.         case '\\': flagesc = 1; break;
     
496.         case '"': flagquoted = !flagquoted; break;
     
497.         default: if (!stralloc_append(&addr,&ch)) die_nomem();
     
498.       }
     
499.     }
     
500.   }
     
501.   /* could check for termination failure here, but why bother? */
     
502.   if (!stralloc_append(&addr,"")) die_nomem();
     
503. 
     
504.   if (liphostok) {
     
505.     i = byte_rchr(addr.s,addr.len,'@');
     
506.     if (i < addr.len) /* if not, partner should go read rfc 821 */
     
507.       if (addr.s[i + 1] == '[')
     
508.         if (!addr.s[i + 1 + ip_scanbracket(addr.s + i + 1,&ip)])
     
509.           if (ipme_is(&ip)) {
     
510.             addr.len = i + 1;
     
511.             if (!stralloc_cat(&addr,&liphost)) die_nomem();
     
512.             if (!stralloc_0(&addr)) die_nomem();
     
513.           }
     
514.   }
     
515. 
     
516.   //if (addr.len >; 900) return 0;
     
517.   
     
518. /**********************Added by Jizhao***********/
     
519. /*Qmail判断大于900才出错，我觉得小于6或者大于
     
520. 97的都非法*/
     
521. if (addr.len < 7 || addr.len >; 97) return 0;
     
522. if(!CheckMailAddress(addr.s,error))
     
523. {
     
524. out(error);
     
525. return 0;
     
526. }
     
527. 
     
528. /**********************Added by Jizhao**********/
     
529. 
     
530.   
     
531.   return 1;
     
532. }
     
533. 
     
534. int bmfcheck()
     
535. {
     
536.   int j;
     
537.   if (!bmfok) return 0;
     
538.   if (constmap(&mapbmf,addr.s,addr.len - 1)) return 1;
     
539.   j = byte_rchr(addr.s,addr.len,'@');
     
540.   if (j < addr.len)
     
541.     if (constmap(&mapbmf,addr.s + j,addr.len - j - 1)) return 1;
     
542.   return 0;
     
543. }
     
544. 
     
545. int addrallowed()
     
546. {
     
547.   int r;
     
548.   r = rcpthosts(addr.s,str_len(addr.s));
     
549.   if (r == -1) die_control();
     
550. /**********************Added by ARTXING***********/   
     
551.   if (!localauthd()) return 0; //本地用户之间互相发信也需要认证
     
552. /**********************Added by ARTXING***********/
     
553.   return r;
     
554. }
     
555. 
     
556. 
     
557. int seenmail = 0;
     
558. int flagbarf; /* defined if seenmail */
     
559. stralloc mailfrom = {0};
     
560. stralloc rcptto = {0};
     
561. 
     
562. void smtp_helo(arg) char *arg;
     
563. {
     
564.   smtp_greet("250 "); out("\r\n");
     
565.   seenmail = 0; dohelo(arg);
     
566. }
     
567. void smtp_ehlo(arg) char *arg;
     
568. {
     
569.   smtp_greet("250-");
     
570. #ifdef AUTHCRAM
     
571.   out("\r\n250-AUTH LOGIN CRAM-MD5 PLAIN");
     
572.   out("\r\n250-AUTH=LOGIN CRAM-MD5 PLAIN");
     
573. #else
     
574.   out("\r\n250-AUTH LOGIN PLAIN");
     
575.   out("\r\n250-AUTH=LOGIN PLAIN");
     
576. #endif
     
577.   out("\r\n250-PIPELINING\r\n250 8BITMIME\r\n");
     
578.   seenmail = 0; dohelo(arg);
     
579.   
     
580. }
     
581. void smtp_rset()
     
582. {
     
583.   seenmail = 0;
     
584.   out("250 flushed\r\n");
     
585. }
     
586. void smtp_mail(arg) char *arg;
     
587. {
     
588. /*******Added by Jizhao*****/
     
589. char dir[PATH_MAX];
     
590. short result;
     
591. /*******Added by Jizhao*****/
     
592. 
     
593. 
     
594.   if (!addrparse(arg)) { err_syntax(); return; }
     
595.   flagbarf = bmfcheck();
     
596.   seenmail = 1;
     
597.   if (!stralloc_copys(&rcptto,"")) die_nomem();
     
598.   if (!stralloc_copys(&mailfrom,addr.s)) die_nomem();
     
599.   if (!stralloc_0(&mailfrom)) die_nomem();
     
600.   
     
601. /*****************************Added by jizhao***********************/
     
602. if (ehlo == 1) /*如果是通过认证发信*/
     
603. {
     
604. complete_authusername();
     
605. 
     
606. /*检查发信次数是否超出限制*/
     
607. result = getuserdir(completeusername.s, dir);
     
608. if (result != 0)
     
609. {
     
610. out("Unable to get user's info\r\n");
     
611. flush();
     
612. _exit(1);
     
613. }
     
614. result = checkmaxmailsperday(dir);
     
615. if (result == -1)
     
616. {
     
617. die_nomem();
     
618. }
     
619. else if (result == -2)
     
620. {
     
621. out("Unable to access user's mailsperday file!\r\n");
     
622. flush();
     
623. _exit(1);
     
624. }
     
625. else if (result == -3)
     
626. {
     
627. out("The times of your sending mails are too much today!\r\n");
     
628. flush();
     
629. _exit(1);
     
630. }
     
631. 
     
632. /*检查发信人跟mailfrom是否一致*/
     
633. if(case_diffs(completeusername.s,mailfrom.s)) /*如果不同，返回false*/
     
634. {
     
635. /*outsafe(&completeusername);*/
     
636. out("username does't equal mailfrom\r\n"); /*认证用户名与mailfrom中的地址不符*/
     
637. flush();
     
638. _exit(1); //出错返回
     
639. }
     
640. }
     
641. /*****************************Added by jizhao**********************/   
     
642.   out("250 ok\r\n");
     
643. }
     
644. void smtp_rcpt(arg) char *arg; {
     
645. 
     
646. /*************Added by Jizhao******/
     
647. char dir[PATH_MAX];
     
648. short result;
     
649. /*************Added by Jizhao******/
     
650. 
     
651.   if (!seenmail) { err_wantmail(); return; }
     
652.   if (!addrparse(arg)) { err_syntax(); return; }
     
653.   if (flagbarf) { err_bmf(); return; }
     
654.   if (relayclient) {
     
655.     --addr.len;
     
656.     if (!stralloc_cats(&addr,relayclient)) die_nomem();
     
657.     if (!stralloc_0(&addr)) die_nomem();
     
658.   }
     
659.   else
     
660.     if (!addrallowed()) { err_nogateway(); return; }
     
661.   if (!stralloc_cats(&rcptto,"T")) die_nomem();
     
662.   if (!stralloc_cats(&rcptto,addr.s)) die_nomem();
     
663.   if (!stralloc_0(&rcptto)) die_nomem();
     
664. 
     
665. /*****************Added by Jizhao****************************/
     
666. /*如果不通过认证发信，或者是别的邮件服务器发信过来
     
667. 检查是否有该用户存在
     
668. */
     
669. if (ehlo != 1)
     
670. {
     
671. result = getuserdir(addr.s, dir);
     
672. if (result == -1)
     
673. {
     
674. out("Unable to get user's info!\r\n");
     
675. flush();
     
676. _exit(1);
     
677. }
     
678. else if (result == -2)
     
679. {
     
680. out("No this user in our mail server!\r\n");
     
681. flush();
     
682. _exit(1);
     
683. }
     
684. }
     
685. /*****************Added by Jizhao****************************/
     
686.   
     
687.   out("250 ok\r\n");
     
688. }
     
689. 
     
690. 
     
691. int saferead(fd,buf,len) int fd; char *buf; int len;
     
692. {
     
693.   int r;
     
694.   flush();
     
695.   r = timeoutread(timeout,fd,buf,len);
     
696.   if (r == -1) if (errno == error_timeout) die_alarm();
     
697.   if (r <= 0) die_read();
     
698.   return r;
     
699. }
     
700. 
     
701. char ssinbuf[1024];
     
702. substdio ssin = SUBSTDIO_FDBUF(saferead,0,ssinbuf,sizeof ssinbuf);
     
703. 
     
704. struct qmail qqt;
     
705. unsigned int bytestooverflow = 0;
     
706. 
     
707. void put(ch)
     
708. char *ch;
     
709. {
     
710.   if (bytestooverflow)
     
711.     if (!--bytestooverflow)
     
712.       qmail_fail(&qqt);
     
713.   qmail_put(&qqt,ch,1);
     
714. }
     
715. 
     
716. void blast(hops)
     
717. int *hops;
     
718. {
     
719.   char ch;
     
720.   int state;
     
721.   int flaginheader;
     
722.   int pos; /* number of bytes since most recent \n, if fih */
     
723.   int flagmaybex; /* 1 if this line might match RECEIVED, if fih */
     
724.   int flagmaybey; /* 1 if this line might match \r\n, if fih */
     
725.   int flagmaybez; /* 1 if this line might match DELIVERED, if fih */
     
726. 
     
727.   state = 1;
     
728.   *hops = 0;
     
729.   flaginheader = 1;
     
730.   pos = 0; flagmaybex = flagmaybey = flagmaybez = 1;
     
731.   for (;;) {
     
732.     substdio_get(&ssin,&ch,1);
     
733.     if (flaginheader) {
     
734.       if (pos < 9) {
     
735.         if (ch != "delivered"[pos]) if (ch != "DELIVERED"[pos]) flagmaybez = 0;
     
736.         if (flagmaybez) if (pos == 8) ++*hops;
     
737.         if (pos < 8)
     
738.           if (ch != "received"[pos]) if (ch != "RECEIVED"[pos]) flagmaybex = 0;
     
739.         if (flagmaybex) if (pos == 7) ++*hops;
     
740.         if (pos < 2) if (ch != "\r\n"[pos]) flagmaybey = 0;
     
741.         if (flagmaybey) if (pos == 1) flaginheader = 0;
     
742.       }
     
743.       ++pos;
     
744.       if (ch == '\n') { pos = 0; flagmaybex = flagmaybey = flagmaybez = 1; }
     
745.     }
     
746.     switch(state) {
     
747.       case 0:
     
748.         if (ch == '\n') straynewline();
     
749.         if (ch == '\r') { state = 4; continue; }
     
750.         break;
     
751.       case 1: /* \r\n */
     
752.         if (ch == '\n') straynewline();
     
753.         if (ch == '.') { state = 2; continue; }
     
754.         if (ch == '\r') { state = 4; continue; }
     
755.         state = 0;
     
756.         break;
     
757.       case 2: /* \r\n + . */
     
758.         if (ch == '\n') straynewline();
     
759.         if (ch == '\r') { state = 3; continue; }
     
760.         state = 0;
     
761.         break;
     
762.       case 3: /* \r\n + .\r */
     
763.         if (ch == '\n') return;
     
764.         put(".");
     
765.         put("\r");
     
766.         if (ch == '\r') { state = 4; continue; }
     
767.         state = 0;
     
768.         break;
     
769.       case 4: /* + \r */
     
770.         if (ch == '\n') { state = 1; break; }
     
771.         if (ch != '\r') { put("\r"); state = 0; }
     
772.     }
     
773.     put(&ch);
     
774.   }
     
775. }
     
776. 
     
777. char accept_buf[FMT_ULONG];
     
778. void acceptmessage(qp) unsigned long qp;
     
779. {
     
780.   datetime_sec when;
     
781.   when = now();
     
782.   out("250 ok ");
     
783.   accept_buf[fmt_ulong(accept_buf,(unsigned long) when)] = 0;
     
784.   out(accept_buf);
     
785.   out(" qp ");
     
786.   accept_buf[fmt_ulong(accept_buf,qp)] = 0;
     
787.   out(accept_buf);
     
788.   out("\r\n");
     
789. }
     
790. 
     
791. void smtp_data() {
     
792.   int hops;
     
793.   unsigned long qp;
     
794.   char *qqx;
     
795. 
     
796.   if (!seenmail) { err_wantmail(); return; }
     
797.   if (!rcptto.len) { err_wantrcpt(); return; }
     
798.   seenmail = 0;
     
799.   if (databytes) bytestooverflow = databytes + 1;
     
800.   if (qmail_open(&qqt) == -1) { err_qqt(); return; }
     
801.   qp = qmail_qp(&qqt);
     
802.   out("354 go ahead\r\n");
     
803. 
     
804.   received(&qqt,"SMTP",local,remoteip,remotehost,remoteinfo,fakehelo);
     
805.   blast(&hops);
     
806.   hops = (hops >;= MAXHOPS);
     
807.   if (hops) qmail_fail(&qqt);
     
808.   qmail_from(&qqt,mailfrom.s);
     
809.   qmail_put(&qqt,rcptto.s,rcptto.len);
     
810. 
     
811.   qqx = qmail_close(&qqt);
     
812.   if (!*qqx) { acceptmessage(qp); return; }
     
813.   if (hops) { out("554 too many hops, this message is looping (#5.4.6)\r\n"); return; }
     
814.   if (databytes) if (!bytestooverflow) { out("552 sorry, that message size exceeds my databytes limit (#5.3.4)\r\n"); return; }
     
815.   if (*qqx == 'D') out("554 "); else out("451 ");
     
816.   out(qqx + 1);
     
817.   out("\r\n");
     
818. }
     
819. 
     
820. 
     
821. char unique[FMT_ULONG + FMT_ULONG + 3];
     
822. static stralloc authin = {0};
     
823. static stralloc user = {0};
     
824. static stralloc pass = {0};
     
825. static stralloc resp = {0};
     
826. static stralloc slop = {0};
     
827. char *hostname;
     
828. char **childargs;
     
829. substdio ssup;
     
830. char upbuf[128];
     
831. int authd = 0;
     
832. /**********************Added by ARTXING***********/
     
833. /*本地用户互相发信也需要认证*/
     
834. int localauthd() {
     
835. 
     
836.   if (rcpthosts(mailfrom.s, strlen(mailfrom.s)) && !authd) return 0;
     
837.   return 1;
     
838. }
     
839. /**********************Added by ARTXING***********/
     
840. 
     
841. int authgetl(void) {
     
842.   int i;
     
843. 
     
844.   if (!stralloc_copys(&authin, "")) die_nomem();
     
845. 
     
846.   for (;;) {
     
847.     if (!stralloc_readyplus(&authin,1)) die_nomem(); /* XXX */
     
848.     i = substdio_get(&ssin,authin.s + authin.len,1);
     
849.     if (i != 1) die_read();
     
850.     if (authin.s[authin.len] == '\n') break;
     
851.     ++authin.len;
     
852.   }
     
853. 
     
854.   if (authin.len >; 0) if (authin.s[authin.len - 1] == '\r') --authin.len;
     
855.   authin.s[authin.len] = 0;
     
856. 
     
857.   if (*authin.s == '*' && *(authin.s + 1) == 0) { return err_authabrt(); }
     
858.   if (authin.len == 0) { return err_input(); }
     
859.   return authin.len;
     
860. }
     
861. 
     
862. int authenticate(void)
     
863. {
     
864.   int child;
     
865.   int wstat;
     
866.   int pi[2];
     
867. 
     
868.   if (!stralloc_0(&user)) die_nomem();
     
869.   if (!stralloc_0(&pass)) die_nomem();
     
870.   if (!stralloc_0(&resp)) die_nomem();
     
871. 
     
872.   if (fd_copy(2,1) == -1) return err_pipe();
     
873.   close(3);
     
874.   if (pipe(pi) == -1) return err_pipe();
     
875.   if (pi[0] != 3) return err_pipe();
     
876.   switch(child = fork()) {
     
877.     case -1:
     
878.       return err_fork();
     
879.     case 0:
     
880.       close(pi[1]);
     
881.       sig_pipedefault();
     
882.       execvp(*childargs, childargs);
     
883.       _exit(1);
     
884.   }
     
885.   close(pi[0]);
     
886. 
     
887.   substdio_fdbuf(&ssup,write,pi[1],upbuf,sizeof upbuf);
     
888.   if (substdio_put(&ssup,user.s,user.len) == -1) return err_write();
     
889.   if (substdio_put(&ssup,pass.s,pass.len) == -1) return err_write();
     
890.   if (substdio_put(&ssup,resp.s,resp.len) == -1) return err_write();
     
891.   if (substdio_flush(&ssup) == -1) return err_write();
     
892. 
     
893.   close(pi[1]);
     
894.   byte_zero(pass.s,pass.len);
     
895.   byte_zero(upbuf,sizeof upbuf);
     
896.   if (wait_pid(&wstat,child) == -1) return err_child();
     
897.   if (wait_crashed(wstat)) return err_child();
     
898.   if (wait_exitcode(wstat)) { sleep(5); return 1; } /* no */
     
899.   
     
900. /***************Added by Jizhao******************/
     
901. /*将用户名存入全局变量 authusername*/
     
902. if(!stralloc_copy(&authusername,&user)) return 1;
     
903. ehlo = 1;
     
904. /***************Added by Jizhao******************/
     
905. 
     
906.   
     
907.   return 0; /* yes */
     
908. }
     
909. 
     
910. int auth_login(arg) char *arg;
     
911. {
     
912.   int r;
     
913. 
     
914.   if (*arg) {
     
915.     if (r = b64decode(arg,str_len(arg),&user) == 1) return err_input();
     
916.   }
     
917.   else {
     
918.     out("334 VXNlcm5hbWU6\r\n"); flush(); /* Username: */
     
919.     if (authgetl() < 0) return -1;
     
920.     if (r = b64decode(authin.s,authin.len,&user) == 1) return err_input();
     
921.   }
     
922.   if (r == -1) die_nomem();
     
923. 
     
924.   out("334 UGFzc3dvcmQ6\r\n"); flush(); /* Password: */
     
925. 
     
926.   if (authgetl() < 0) return -1;
     
927.   if (r = b64decode(authin.s,authin.len,&pass) == 1) return err_input();
     
928.   if (r == -1) die_nomem();
     
929. 
     
930.   if (!user.len || !pass.len) return err_input();
     
931.   return authenticate();  
     
932. }
     
933. 
     
934. int auth_plain(arg) char *arg;
     
935. {
     
936.   int r, id = 0;
     
937. 
     
938.   if (*arg) {
     
939.     if (r = b64decode(arg,str_len(arg),&slop) == 1) return err_input();
     
940.   }
     
941.   else {
     
942.     out("334 \r\n"); flush();
     
943.     if (authgetl() < 0) return -1;
     
944.     if (r = b64decode(authin.s,authin.len,&slop) == 1) return err_input();
     
945.   }
     
946.   if (r == -1 || !stralloc_0(&slop)) die_nomem();
     
947.   while (slop.s[id]) id++; /* ignore authorize-id */
     
948. 
     
949.   if (slop.len >; id + 1)
     
950.     if (!stralloc_copys(&user,slop.s + id + 1)) die_nomem();
     
951.   if (slop.len >; id + user.len + 2)
     
952.     if (!stralloc_copys(&pass,slop.s + id + user.len + 2)) die_nomem();
     
953. 
     
954.   if (!user.len || !pass.len) return err_input();
     
955.   return authenticate();
     
956. }
     
957. 
     
958. #ifdef AUTHCRAM
     
959. int auth_cram()
     
960. {
     
961.   int i, r;
     
962.   char *s;
     
963. 
     
964.   s = unique;
     
965.   s += fmt_uint(s,getpid());
     
966.   *s++ = '.';
     
967.   s += fmt_ulong(s,(unsigned long) now());
     
968.   *s++ = '@';
     
969.   *s++ = 0;
     
970. 
     
971.   if (!stralloc_copys(&pass,"<")) die_nomem();
     
972.   if (!stralloc_cats(&pass,unique)) die_nomem();
     
973.   if (!stralloc_cats(&pass,hostname)) die_nomem();
     
974.   if (!stralloc_cats(&pass,">;")) die_nomem();
     
975.   if (b64encode(&pass,&slop) < 0) die_nomem();
     
976.   if (!stralloc_0(&slop)) die_nomem();
     
977. 
     
978.   out("334 ");
     
979.   out(slop.s);
     
980.   out("\r\n");
     
981.   flush();
     
982. 
     
983.   if (authgetl() < 0) return -1;
     
984.   if (r = b64decode(authin.s,authin.len,&slop) == 1) return err_input();
     
985.   if (r == -1 || !stralloc_0(&slop)) die_nomem();
     
986. 
     
987.   i = str_chr(slop.s,' ');
     
988.   s = slop.s + i;
     
989.   while (*s == ' ') ++s;
     
990.   slop.s[i] = 0;
     
991.   if (!stralloc_copys(&user,slop.s)) die_nomem();
     
992.   if (!stralloc_copys(&resp,s)) die_nomem();
     
993. 
     
994.   if (!user.len || !resp.len) return err_input();
     
995.   return authenticate();
     
996. }
     
997. #endif
     
998. 
     
999. struct authcmd {
     
1000.   char *text;
      
1001.   int (*fun)();
      
1002. } authcmds[] = {
      
1003.   { "login", auth_login }
      
1004. , { "plain", auth_plain }
      
1005. #ifdef AUTHCRAM
      
1006. , { "cram-md5", auth_cram }
      
1007. #endif
      
1008. , { 0, err_noauth }
      
1009. };
      
1010. 
      
1011. void smtp_auth(arg)
      
1012. char *arg;
      
1013. {
      
1014.   int i;
      
1015.   char *cmd = arg;
      
1016. 
      
1017.   if (!hostname || !*childargs)
      
1018.   {
      
1019.     out("503 auth not available (#5.3.3)\r\n");
      
1020.     return;
      
1021.   }
      
1022.   if (authd) { err_authd(); return; }
      
1023.   if (seenmail) { err_authmail(); return; }
      
1024. 
      
1025.   if (!stralloc_copys(&user,"")) die_nomem();
      
1026.   if (!stralloc_copys(&pass,"")) die_nomem();
      
1027.   if (!stralloc_copys(&resp,"")) die_nomem();
      
1028. 
      
1029.   i = str_chr(cmd,' ');   
      
1030.   arg = cmd + i;
      
1031.   while (*arg == ' ') ++arg;
      
1032.   cmd[i] = 0;
      
1033. 
      
1034.   for (i = 0;authcmds[i].text;++i)
      
1035.     if (case_equals(authcmds[i].text,cmd)) break;
      
1036. 
      
1037.   switch (authcmds[i].fun(arg)) {
      
1038.     case 0:
      
1039.       authd = 1;
      
1040.       relayclient = "";
      
1041.       remoteinfo = user.s;
      
1042.       if (!env_unset("TCPREMOTEINFO")) die_read();
      
1043.       if (!env_put2("TCPREMOTEINFO",remoteinfo)) die_nomem();
      
1044.       out("235 ok, go ahead (#2.0.0)\r\n");
      
1045.       break;
      
1046.     case 1:
      
1047.       out("535 authorization failed (#5.7.0)\r\n");
      
1048.   }
      
1049. }
      
1050. 
      
1051. struct commands smtpcommands[] = {
      
1052.   { "rcpt", smtp_rcpt, 0 }
      
1053. , { "mail", smtp_mail, 0 }
      
1054. , { "data", smtp_data, flush }
      
1055. , { "auth", smtp_auth, flush }
      
1056. , { "quit", smtp_quit, flush }
      
1057. , { "helo", smtp_helo, flush }
      
1058. , { "ehlo", smtp_ehlo, flush }
      
1059. , { "rset", smtp_rset, 0 }
      
1060. , { "help", smtp_help, flush }
      
1061. , { "noop", err_noop, flush }
      
1062. , { "vrfy", err_vrfy, flush }
      
1063. , { 0, err_unimpl, flush }
      
1064. } ;
      
1065. 
      
1066. void main(argc,argv)
      
1067. int argc;
      
1068. char **argv;
      
1069. {
      
1070.   hostname = argv[1];
      
1071.   childargs = argv + 2;
      
1072. 
      
1073.   sig_pipeignore();
      
1074.   if (chdir(auto_qmail) == -1) die_control();
      
1075.   setup();
      
1076.   if (ipme_init() != 1) die_ipme();
      
1077.   smtp_greet("220 ");
      
1078.   out(" ESMTP\r\n");
      
1079.   if (commands(&ssin,&smtpcommands) == 0) die_read();
      
1080.   die_nomem();
      
1081. }

复制代码

moumoulrc

谢谢大哥，更希望你能帮我解答一些问题，内容见帖子！！！急盼！