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

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > TCP/IP协议栈 > 查看源码
popserv.c
资源名称：export.zip　[点击查看]
上传用户：hepax88
上传日期：2007-01-03
资源大小：1101k
文件大小：14k
源码类别：

TCP/IP协议栈

开发平台：

Visual C++

popserv.c：源码内容
  1. /* POP Server state machine - see RFC 937
  2.  *
  3.  *  also see other credits in popcli.c
  4.  *  10/89 Mike Stockett wa7dyx
  5.  *  Modified 5/27/90 by Allen Gwinn, N5CKP, for later NOS releases.
  6.  *  Added to NOS by PA0GRI 2/6/90 (and linted into "standard" C)
  7.  */
  9. #include <stdio.h>
  10. #include <time.h>
  11. #include <sys/stat.h>
  12. #ifdef UNIX
  13. #include <sys/types.h>
  14. #endif
  15. #if defined(__STDC__) || defined(__TURBOC__)
  16. #include <stdarg.h>
  17. #endif
  18. #include <ctype.h>
  19. #include <setjmp.h>
  20. #include "global.h"
  21. #include "mbuf.h"
  22. #include "cmdparse.h"
  23. #include "socket.h"
  24. #include "proc.h"
  25. #include "files.h"
  26. #include "pop.h"
  28. extern char Nospace[];
  30. static struct pop_scb *create_scb(void);
  31. static void delete_scb(struct pop_scb *scb);
  32. static void popserv(int s,void *unused,void *p);
  33. static int poplogin(char *pass,char *username);
  34. static void rrip(register char *s);
  35. int newmail(struct pop_scb *);
  36. int isdeleted(struct pop_scb *,int);
  38. /* I don't know why this isn't static, it isn't called anywhere else {was} */
  39. void pop_sm(struct pop_scb *scb);
  41. static int Spop = -1; /* prototype socket for service */
  43. /* Start up POP receiver service */
  44. int
  45. pop1(argc,argv,p)
  47. int argc;
  48. char *argv[];
  49. void *p;
  51. {
  52. struct sockaddr_in lsocket;
  53. int s;
  55. if (Spop != -1) {
  56. return 0;
  57. }
  59. ksignal(Curproc,0); /* Don't keep the parser waiting */
  60. chname(Curproc,"POP listener");
  62. lsocket.sin_family = AF_INET;
  63. lsocket.sin_addr.s_addr = INADDR_ANY;
  64. if(argc < 2)
  65. lsocket.sin_port = IPPORT_POP;
  66. else
  67. lsocket.sin_port = atoi(argv[1]);
  69. Spop = socket(AF_INET,SOCK_STREAM,0);
  71. bind(Spop,(struct sockaddr *)&lsocket,sizeof(lsocket));
  73. listen(Spop,1);
  75. for (;;) {
  76. if((s = accept(Spop,NULL,(int *)NULL)) == -1)
  77. break; /* Service is shutting down */
  79. /* Spawn a server */
  81. newproc("POP server",2048,popserv,s,NULL,NULL,0);
  82. }
  83. return 0;
  84. }
  86. /* Shutdown POP service (existing connections are allowed to finish) */
  88. int
  89. pop0(argc,argv,p)
  90. int argc;
  91. char *argv[];
  92. void *p;
  94. {
  95. close_s(Spop);
  96. Spop = -1;
  97. return 0;
  98. }
  100. static void
  101. popserv(s,unused,p)
  102. int s;
  103. void *unused;
  104. void *p;
  105. {
  106. struct pop_scb *scb;
  107. FILE *network;
  109. sockowner(s,Curproc); /* We own it now */
  110. logmsg(s,"open POP");
  111. network = fdopen(s,"r+t");
  113. if((scb = create_scb()) == NULL) {
  114. printf(Nospace);
  115. logmsg(s,"close POP - no space");
  116. fclose(network);
  117. return;
  118. }
  120. scb->network = network;
  121. scb->state  = AUTH;
  123. (void) fprintf(network,greeting_msg,Hostname);
  125. loop: if (fgets(scb->buf,BUF_LEN,network) == NULL){
  126. /* He closed on us */
  128. goto quit;
  129. }
  130. scb->count = strlen(scb->buf);
  131. rip(scb->buf);
  132. if (strlen(scb->buf) == 0) /* Ignore blank cmd lines */
  133. goto loop;
  134. pop_sm(scb);
  135. if (scb->state == DONE)
  136. goto quit;
  138. goto loop;
  140. quit:
  141. logmsg(fileno(scb->network),"close POP");
  142. fclose(scb->network);
  143. delete_scb(scb);
  144. }
  147. /* Create control block, initialize */
  149. static struct
  150. pop_scb *create_scb()
  151. {
  152. register struct pop_scb *scb;
  154. if((scb = (struct pop_scb *)callocw(1,sizeof (struct pop_scb))) == NULL)
  155. return NULL;
  157. scb->username[0] = '';
  158. scb->msg_status = NULL;
  159. scb->wf = NULL;
  161. scb->count = scb->folder_file_size = scb->msg_num = 0;
  163. scb->folder_modified = FALSE;
  164. return scb;
  165. }
  168. /* Free resources, delete control block */
  170. static void
  171. delete_scb(scb)
  172. register struct pop_scb *scb;
  173. {
  175. if (scb == NULL)
  176. return;
  177. if (scb->wf != NULL)
  178. fclose(scb->wf);
  179. if (scb->msg_status  != NULL)
  180. free(scb->msg_status);
  182. free(scb);
  183. }
  185. /* replace terminating end of line marker(s) (r and n) with null */
  186. static void
  187. rrip(s)
  188. register char *s;
  189. {
  190. register char *cp;
  192. if((cp = strchr(s,'r')) != NULL)
  193. *cp = '';
  194. if((cp = strchr(s,'n')) != NULL)
  195. *cp = '';
  196. }
  198. /* --------------------- start of POP server code ------------------------ */
  200. #define BITS_PER_WORD 16
  202. #define isSOM(x) ((strncmp(x,"From ",5) == 0))
  204. /* Command string specifications */
  206. static char ackd_cmd[] = "ACKD",
  207. acks_cmd[] = "ACKS",
  208. #ifdef POP_FOLDERS
  209. fold_cmd[] = "FOLD ",
  210. #endif
  211. login_cmd[] = "HELO ",
  212. nack_cmd[] = "NACK",
  213. quit_cmd[] = "QUIT",
  214. read_cmd[] = "READ",
  215. retr_cmd[] = "RETR";
  217. void
  218. pop_sm(scb)
  219. struct pop_scb *scb;
  220. {
  221. char password[40];
  222. void state_error(struct pop_scb *,char *);
  223. void open_folder(struct pop_scb *);
  224. void do_cleanup(struct pop_scb *);
  225. void read_message(struct pop_scb *);
  226. void retrieve_message(struct pop_scb *);
  227. void deletemsg(struct pop_scb *,int);
  228. void get_message(struct pop_scb *,int);
  229. void print_message_length(struct pop_scb *);
  230. void close_folder(struct pop_scb *);
  231. #ifdef POP_FOLDERS
  232. void select_folder(struct pop_scb *);
  233. #endif
  235. if (scb == NULL) /* be certain it is good -- wa6smn */
  236. return;
  238. switch(scb->state) {
  239. case AUTH:
  240. if (strncmp(scb->buf,login_cmd,strlen(login_cmd)) == 0){
  241. sscanf(scb->buf,"HELO %s%s",scb->username,password);
  243. if (!poplogin(scb->username,password)) {
  244. logmsg(fileno(scb->network),"POP access DENIED to %s",
  245.     scb->username);
  246. state_error(scb,"Access DENIED!!");
  247. return;
  248. }
  250. logmsg(fileno(scb->network),"POP access granted to %s",
  251.     scb->username);
  252. open_folder(scb);
  253. } else if (strncmp(scb->buf,quit_cmd,strlen(quit_cmd)) == 0){
  254. do_cleanup(scb);
  255. } else
  256. state_error(scb,"(AUTH) Expected HELO or QUIT command");
  257. break;
  259. case MBOX:
  260. if (strncmp(scb->buf,read_cmd,strlen(read_cmd)) == 0)
  261. read_message(scb);
  263. #ifdef POP_FOLDERS
  264. else if (strncmp(scb->buf,fold_cmd,strlen(fold_cmd)) == 0)
  265. select_folder(scb);
  267. #endif
  269. else if (strncmp(scb->buf,quit_cmd,strlen(quit_cmd)) == 0) {
  270. do_cleanup(scb);
  271. } else
  272. state_error(scb,
  273. #ifdef POP_FOLDERS
  274.     "(MBOX) Expected FOLD, READ, or QUIT command");
  275. #else
  276.     "(MBOX) Expected READ or QUIT command");
  277. #endif
  278. break;
  280. case ITEM:
  281. if (strncmp(scb->buf,read_cmd,strlen(read_cmd)) == 0)
  282. read_message(scb);
  284. #ifdef POP_FOLDERS
  286. else if (strncmp(scb->buf,fold_cmd,strlen(fold_cmd)) == 0)
  287. select_folder(scb);
  288. #endif
  290. else if (strncmp(scb->buf,retr_cmd,strlen(retr_cmd)) == 0)
  291. retrieve_message(scb);
  292. else if (strncmp(scb->buf,quit_cmd,strlen(quit_cmd)) == 0)
  293. do_cleanup(scb);
  294. else
  295. state_error(scb,
  296. #ifdef POP_FOLDERS
  297.    "(ITEM) Expected FOLD, READ, RETR, or QUIT command");
  298. #else
  299.    "(ITEM) Expected READ, RETR, or QUIT command");
  300. #endif
  301. break;
  303. case NEXT:
  304. if (strncmp(scb->buf,ackd_cmd,strlen(ackd_cmd)) == 0){
  305. /* ACKD processing */
  306. deletemsg(scb,scb->msg_num);
  307. scb->msg_num++;
  308. get_message(scb,scb->msg_num);
  309. } else if (strncmp(scb->buf,acks_cmd,strlen(acks_cmd)) == 0){
  310. /* ACKS processing */
  311. scb->msg_num++;
  312. get_message(scb,scb->msg_num);
  313. } else if (strncmp(scb->buf,nack_cmd,strlen(nack_cmd)) == 0){
  314. /* NACK processing */
  315. fseek(scb->wf,scb->curpos,SEEK_SET);
  316. } else {
  317. state_error(scb,"(NEXT) Expected ACKD, ACKS, or NACK command");
  318. return;
  319. }
  321. print_message_length(scb);
  322. scb->state  = ITEM;
  323. break;
  325. case DONE:
  326. do_cleanup(scb);
  327. break;
  329. default:
  330. state_error(scb,"(TOP) State Error!!");
  331. break;
  332. }
  333. }
  335. void
  336. do_cleanup(scb)
  337. struct pop_scb *scb;
  338. {
  339. void close_folder(struct pop_scb *);
  341. close_folder(scb);
  342. (void) fprintf(scb->network,signoff_msg);
  343. scb->state = DONE;
  344. }
  346. void
  347. state_error(scb,msg)
  348. struct pop_scb *scb;
  349. char *msg;
  350. {
  351. (void) fprintf(scb->network,error_rsp,msg);
  352. scb->state = DONE;
  353. }
  355. #ifdef POP_FOLDERS
  357. select_folder(scb)
  358. struct pop_scb *scb;
  359. {
  360. sscanf(scb->buf,"FOLD %s",scb->username);
  362. if (scb->wf != NULL)
  363. close_folder(scb);
  365. open_folder(scb);
  366. }
  368. #endif
  371. void
  372. close_folder(scb)
  373. struct pop_scb *scb;
  374. {
  375. char folder_pathname[64];
  376. char line[BUF_LEN];
  377. FILE *fd;
  378. int deleted = FALSE;
  379. int msg_no = 0;
  380. struct stat folder_stat;
  382. if (scb->wf == NULL)
  383. return;
  385. if (!scb->folder_modified) {
  386. /* no need to re-write the folder if we have not modified it */
  388. fclose(scb->wf);
  389. scb->wf = NULL;
  391. free(scb->msg_status);
  392. scb->msg_status = NULL;
  393. return;
  394. }
  397. sprintf(folder_pathname,"%s/%s.txt",Mailspool,scb->username);
  399. if (newmail(scb)) {
  400. /* copy new mail into the work file and save the
  401.    message count for later */
  403. if ((fd = fopen(folder_pathname,"r")) == NULL) {
  404. state_error(scb,"Unable to add new mail to folder");
  405. return;
  406. }
  408. fseek(scb->wf,0,SEEK_END);
  409. fseek(fd,scb->folder_file_size,SEEK_SET);
  410. while (!feof(fd)) {
  411. fgets(line,BUF_LEN,fd);
  412. fputs(line,scb->wf);
  413. }
  415. fclose(fd);
  416. }
  418. /* now create the updated mail folder */
  420. if ((fd = fopen(folder_pathname,"w")) == NULL){
  421. state_error(scb,"Unable to update mail folder");
  422. return;
  423. }
  425. rewind(scb->wf);
  426. while (!feof(scb->wf)){
  427. fgets(line,BUF_LEN,scb->wf);
  429. if (isSOM(line)){
  430. msg_no++;
  431. if (msg_no <= scb->folder_len)
  432. deleted = isdeleted(scb,msg_no);
  433. else
  434. deleted = FALSE;
  435. }
  437. if (deleted)
  438. continue;
  440. fputs(line,fd);
  441. }
  443. fclose(fd);
  445. /* trash the updated mail folder if it is empty */
  447. if ((stat(folder_pathname,&folder_stat) == 0) && (folder_stat.st_size == 0))
  448. unlink(folder_pathname);
  450. fclose(scb->wf);
  451. scb->wf = NULL;
  453. free(scb->msg_status);
  454. scb->msg_status = NULL;
  455. }
  457. void
  458. open_folder(scb)
  459. struct pop_scb *scb;
  460. {
  461. char folder_pathname[64];
  462. char line[BUF_LEN];
  463. FILE *fd;
  464. FILE *tmpfile();
  465. struct stat folder_stat;
  468. sprintf(folder_pathname,"%s/%s.txt",Mailspool,scb->username);
  469. scb->folder_len       = 0;
  470. scb->folder_file_size = 0;
  471. if (stat(folder_pathname,&folder_stat)){
  472.  (void) fprintf(scb->network,no_mail_rsp);
  473.  return;
  474. }
  476. scb->folder_file_size = folder_stat.st_size;
  477. if ((fd = fopen(folder_pathname,"r")) == NULL){
  478. state_error(scb,"Unable to open mail folder");
  479. return;
  480. }
  482. if ((scb->wf = tmpfile()) == NULL) {
  483. state_error(scb,"Unable to create work folder");
  484. return;
  485. }
  487. while(!feof(fd)) {
  488. fgets(line,BUF_LEN,fd);
  490. /* scan for begining of a message */
  492. if (isSOM(line))
  493. scb->folder_len++;
  495. /* now put  the line in the work file */
  497. fputs(line,scb->wf);
  498. }
  500. fclose(fd);
  502. scb->msg_status_size = (scb->folder_len) / BITS_PER_WORD;
  504. if ((((scb->folder_len) % BITS_PER_WORD) != 0) ||
  505.     (scb->msg_status_size == 0))
  506. scb->msg_status_size++;
  508. if ((scb->msg_status = (unsigned int *) callocw(scb->msg_status_size,
  509. sizeof(unsigned int))) == NULL) {
  510. state_error(scb,"Unable to create message status array");
  511. return;
  512. }
  514. (void) fprintf(scb->network,count_rsp,scb->folder_len);
  516. scb->state  = MBOX;
  517. }
  519. void
  520. read_message(scb)
  521. struct pop_scb *scb;
  522. {
  523. void get_message(struct pop_scb *,int);
  524. void print_message_length(struct pop_scb *);
  526. if (scb == NULL) /* check for null -- wa6smn */
  527. return;
  528. if (scb->buf[sizeof(read_cmd) - 1] == ' ')
  529. scb->msg_num = atoi(&(scb->buf[sizeof(read_cmd) - 1]));
  530. else
  531. scb->msg_num++;
  533. get_message(scb,scb->msg_num);
  534. print_message_length(scb);
  535. scb->state  = ITEM;
  536. }
  538. void
  539. retrieve_message(scb)
  540. struct pop_scb *scb;
  541. {
  542. char line[BUF_LEN];
  543. long cnt;
  545. if (scb == NULL) /* check for null -- wa6smn */
  546. return;
  547. if (scb->msg_len == 0) {
  548. state_error(scb,"Attempt to access a DELETED message!");
  549. return;
  550. }
  552. cnt  = scb->msg_len;
  553. while(!feof(scb->wf) && (cnt > 0)) {
  554. fgets(line,BUF_LEN,scb->wf);
  555. rrip(line);
  557. (void) fprintf(scb->network,msg_line,line);
  558. cnt -= (strlen(line)+2); /* Compensate for CRLF */
  559. }
  561. scb->state = NEXT;
  562. }
  564. void
  565. get_message(scb,msg_no)
  566. struct pop_scb *scb;
  567. int msg_no;
  568. {
  569. char line[BUF_LEN];
  571. if (scb == NULL) /* check for null -- wa6smn */
  572. return;
  573. scb->msg_len = 0;
  574. if (msg_no > scb->folder_len) {
  575. scb->curpos  = 0;
  576. scb->nextpos = 0;
  577. return;
  578. } else {
  579. /* find the message and its length */
  581. rewind(scb->wf);
  582. while (!feof(scb->wf) && (msg_no > -1)) {
  583. if (msg_no > 0)
  584. scb->curpos = ftell(scb->wf);
  586. fgets(line,BUF_LEN,scb->wf);
  587. rrip(line);
  589. if (isSOM(line))
  590. msg_no--;
  592. if (msg_no != 0)
  593. continue;
  595. scb->nextpos  = ftell(scb->wf);
  596. scb->msg_len += (strlen(line)+2); /* Add CRLF */
  597. }
  598. }
  600. if (scb->msg_len > 0)
  601. fseek(scb->wf,scb->curpos,SEEK_SET);
  603. /* we need the pointers even if the message was deleted */
  605. if  (isdeleted(scb,scb->msg_num))
  606. scb->msg_len = 0;
  607. }
  609. static int
  610. poplogin(username,pass)
  611. char *pass;
  612. char *username;
  613. {
  614. char buf[80];
  615. char *cp;
  616. char *cp1;
  617. FILE *fp;
  619. if((fp = fopen(Popusers,"r")) == NULL) {
  620. /* User file doesn't exist */
  621. printf("POP users file %s not foundn",Popusers);
  622. return(FALSE);
  623. }
  625. while(fgets(buf,sizeof(buf),fp),!feof(fp)) {
  626. if(buf[0] == '#')
  627. continue; /* Comment */
  629. if((cp = strchr(buf,':')) == NULL)
  630. /* Bogus entry */
  631. continue;
  633. *cp++ = ''; /* Now points to password */
  634. if(strcmp(username,buf) == 0)
  635. break; /* Found user name */
  636. }
  638. if(feof(fp)) {
  639. /* User name not found in file */
  641. fclose(fp);
  642. return(FALSE);
  643. }
  644. fclose(fp);
  646. if ((cp1 = strchr(cp,':')) == NULL)
  647. return(FALSE);
  649. *cp1 = '';
  650. if(strcmp(cp,pass) != 0) {
  651. /* Password required, but wrong one given */
  653. return(FALSE);
  654. }
  656. /* whew! finally made it!! */
  658. return(TRUE);
  659. }
  661. int
  662. isdeleted(scb,msg_no)
  663. struct pop_scb *scb;
  664. int msg_no;
  665. {
  666. unsigned int mask = 1,offset;
  668. msg_no--;
  669. offset = msg_no / BITS_PER_WORD;
  670. mask <<= msg_no % BITS_PER_WORD;
  671. return (((scb->msg_status[offset]) & mask)? TRUE:FALSE);
  672. }
  674. void
  675. deletemsg(scb,msg_no)
  676. struct pop_scb *scb;
  677. int msg_no;
  678. {
  679. unsigned int mask = 1,offset;
  681. if (scb == NULL) /* check for null -- wa6smn */
  682. return;
  683. msg_no--;
  684. offset = msg_no / BITS_PER_WORD;
  685. mask <<= msg_no % BITS_PER_WORD;
  686. scb->msg_status[offset] |= mask;
  687. scb->folder_modified = TRUE;
  688. }
  690. int
  691. newmail(scb)
  692. struct pop_scb *scb;
  693. {
  694. char folder_pathname[64];
  695. struct stat folder_stat;
  697. sprintf(folder_pathname,"%s/%s.txt",Mailspool,scb->username);
  699. if (stat(folder_pathname,&folder_stat)) {
  700. state_error(scb,"Unable to get old mail folder's status");
  701. return(FALSE);
  702. } else
  703. return ((folder_stat.st_size > scb->folder_file_size)? TRUE:FALSE);
  704. }
  706. void
  707. print_message_length(scb)
  708. struct pop_scb *scb;
  709. {
  710. char *print_control_string;
  712. if (scb == NULL) /* check for null -- wa6smn */
  713. return;
  714. if (scb->msg_len > 0)
  715. print_control_string = length_rsp;
  716. else if (scb->msg_num <= scb->folder_len)
  717. print_control_string = length_rsp;
  718. else
  719. print_control_string = no_more_rsp;
  721. (void)fprintf(scb->network,print_control_string,scb->msg_len,scb->msg_num);
  722. }