网络编程三、实现双方聊天

大概思路：

服务器：
  服务器在listen 之后会阻塞在accept 处，等待客户端的连接，若想多方聊天（2人以上）就必须把accept（）放在while（1）里面且在其中fork一个子进程如第二张代码图所示。那么我们就可以在while循环里不断地检测客户端的连接。连接之后，创建子进程来发送数据；父进程接收数据，这样就可以实现接收和发送两不误。
要实现多人聊天就必须：在while中创建一个子进程来负责收发数据，在这个子进程中再fork一个子进程负责收或者发数据。
  客户端也是一样的，在connect()之后，在while（1）循环里创建一个子进程，子进程来接收数据；父进程发数据。
为啥多人聊天客户端和服务端的收发混乱由于两个客户端在争抢资源且gets的光标不知道被发到哪个客户端上去。导致消息错乱。

具体代码演示：

客户端：

#include <sys/types.h>          /* See NOTES */
#include <sys/socket.h>
#include <arpa/inet.h>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
int main(int argc,char**argv){
        if(argc != 3){
                printf("input error\n");
                exit(-1);
        }       
//socket
		 int soc_fd;
        soc_fd=socket(AF_INET,SOCK_STREAM,0);
//bind
        struct sockaddr_in bind_addr;
        bind_addr.sin_family=AF_INET;
        bind_addr.sin_port=htons(atoi(argv[2]));
        inet_aton(argv[1],&bind_addr.sin_addr);
        bind(soc_fd,(struct sockaddr*)&bind_addr,sizeof(struct sockaddr_in));
//listen
        listen(soc_fd,10);
//accept
        struct sockaddr_in c_addr;
        int c_len=sizeof(struct sockaddr_in);

        char readbuf[128];
        int read_ret;
        char getmsg[128];
        while(1){
                int cli_fd=accept(soc_fd,(struct sockaddr *)&c_addr.sin_addr,&c_len);
                if(cli_fd != -1)
                        printf("connected\n");

                printf("get connect:%s\n",inet_ntoa(c_addr.sin_addr));
                //write 
                if(fork()==0){
                        while(1){
                                memset(getmsg,0,sizeof(getmsg));
                                printf("input: ");
                                gets(getmsg);
                                write(cli_fd,getmsg,strlen(getmsg));
                       }
                }
                //read
                while(1){
                                memset(readbuf,0,128);
                                read_ret=read(cli_fd,readbuf,128);
                                if(read_ret==-1){
                                        perror("read");
                                }
                                else{
                                        printf("read content:%d  %s\n",read_ret,readbuf);
                                }
                }
        }
        return 0;

}                          

若要实现多人聊天则需要如下代码

while(1){
                        int cli_fd=accept(soc_fd,(struct sockaddr *)&c_addr.sin_addr,&c_len);
                        if(cli_fd != -1)
                                printf("connected\n");

                        printf("get connect:%s\n",inet_ntoa(c_addr.sin_addr));
                //write 
                if(fork()==0){
                        if(fork()==0){
                                while(1){
                                        memset(getmsg,0,sizeof(getmsg));
                                        printf("input: ");
                                        gets(getmsg);
                                        write(cli_fd,getmsg,strlen(getmsg));
                                }
                        }
                //read

                        while(1){
                                memset(readbuf,0,128);
                                read_ret=read(cli_fd,readbuf,128);
                                if(read_ret==-1){
                                        perror("read");
                                }
                                else{
                                        printf("read content:%d  %s\n",read_ret,readbuf);
                                }
                        }
                }
        }