聊天程序的简单实现

功能:

实现两个主机之间的局域网聊天


实现关键:

如何同时处理套接字的输入和标准输入 --- select函数


客户端的程序

  1. #include "unp.h"
  3. void cli_echo(FILE *fp, int sockfd)
  4. {
  5. 	int maxfdp1, stdineof;
  6. 	fd_set rset;
  7. 	char buf[1000];
  8. 	int n;
  10. 	stdineof=0;
  11. 	FD_ZERO(&rset);
  12. 	for(;;){
  13. 		if(stdineof==0)
  14. 			FD_SET(fileno(fp),&rset);
  15. 		FD_SET(sockfd,&rset);
  16. 		maxfdp1=max(fileno(fp), sockfd)+1;
  17. 		Select(maxfdp1, &rset, NULL, NULL, NULL);
  19. 		if(FD_ISSET(sockfd, &rset)) {//socket is readable
  20. 			if( (n=Read(sockfd, buf, 1000))==0){
  21. 				if(stdineof==1)
  22. 					return;//normal termination
  23. 				else
  24. 					err_quit("cli_echo: server terminated prematurely");
  25. 			}
  26. 			Write(fileno(stdout),buf,n);			
  27. 		}
  29. 		if(FD_ISSET(fileno(fp), &rset)){//input is readable
  30. 			if( (n=Read(fileno(fp),buf,1000))==0){
  31. 				stdineof=1;
  32. 				Shutdown(sockfd, SHUT_WR);
  33. 				FD_CLR(fileno(fp), &rset);
  34. 				continue;
  35. 			}
  36. 			Writen(sockfd, buf, n);
  37. 		}
  38. 	}
  39. }
  41. int
  42. main(int argc, char **argv)
  43. {
  44. 	int sockfd;
  45. 	struct sockaddr_in servaddr;
  47. 	if(argc!=2)
  48. 		err_quit("usage: tcpcli <IPaddress>");
  49. 	
  50. 	sockfd=Socket(AF_INET, SOCK_STREAM, 0);
  51. 	
  52. 	bzero(&servaddr, sizeof(servaddr));
  53. 	servaddr.sin_family=AF_INET;
  54. 	servaddr.sin_port=htons(1234);
  55. 	Inet_pton(AF_INET,argv[1], &servaddr.sin_addr);
  57. 	Connect(sockfd, (SA*) &servaddr, sizeof(servaddr));
  59. 	cli_echo(stdin,sockfd);
  61. 	exit(0);
  63. }

服务器端的程序

  1. #include "unp.h"
  2. #include <time.h>
  4. void srv_echo(FILE *fp, int sockfd)
  5. {
  6. 	int maxfdp1, stdineof;
  7. 	fd_set rset;
  8. 	char buf[1000];
  9. 	int n;
  11. 	stdineof=0;
  12. 	FD_ZERO(&rset);
  13. 	for(;;){
  14. 		if(stdineof==0)
  15. 			FD_SET(fileno(fp),&rset);
  16. 		FD_SET(sockfd,&rset);
  17. 		maxfdp1=max(fileno(fp), sockfd)+1;
  18. 		Select(maxfdp1, &rset, NULL, NULL, NULL);
  20. 		if(FD_ISSET(sockfd, &rset)) {//socket is readable
  21. 			if( (n=Read(sockfd, buf, 1000))==0){
  22. 				if(stdineof==1)
  23. 					return;//normal termination
  24. 				else
  25. 					err_quit("srv_echo: client terminated prematurely");
  26. 			}
  27. 			Write(fileno(stdout),buf,n);			
  28. 		}
  30. 		if(FD_ISSET(fileno(fp), &rset)){//input is readable
  31. 			if( (n=Read(fileno(fp),buf,1000))==0){
  32. 				stdineof=1;
  33. 				Shutdown(sockfd, SHUT_WR);
  34. 				FD_CLR(fileno(fp), &rset);
  35. 				continue;
  36. 			}
  37. 			Writen(sockfd, buf, n);
  38. 		}
  39. 	}
  40. }
  42. int 
  43. main(int argc, char **argv)
  44. {
  45. 	int i, maxi, maxfd, listenfd, connfd, sockfd;
  46. 	socklen_t clilen;
  47. 	struct sockaddr_in cliaddr, servaddr;
  49. 	listenfd = Socket(AF_INET, SOCK_STREAM,0);
  51. 	bzero(&servaddr, sizeof(servaddr));
  52. 	servaddr.sin_family=AF_INET;
  53. 	servaddr.sin_addr.s_addr=htonl(INADDR_ANY);
  54. 	servaddr.sin_port=htons(1234);
  56. 	Bind(listenfd, (SA*)&servaddr, sizeof(servaddr));
  58. 	Listen(listenfd, LISTENQ);
  60. 	clilen=sizeof(cliaddr);
  61. 	connfd=Accept(listenfd, (SA*) &cliaddr, &clilen);
  62. 	
  63. 	srv_echo(stdin,connfd);
  64. 	
  65. 	Close(connfd);
  66. 	return 0;
  67. }


实现结果:

1. 先开启服务端 ./srv_echo

2. 再开启客户端 ./cli_echo 127.0.0.1

3. 任意在服务端或者客户端输入内容, 对端都将收到该内容







附件列表

    【推广】 免费学中医，健康全家人
    原文地址：https://www.cnblogs.com/gremount/p/5803107.html