网络编程中简单的点对点聊天程序

利用多进程实现，一个进程负责发送，一个进程负责接收！毕竟文件描述符共享（实际上是两个进程空间中不相关的文件描述符指向同一个文件）

服务器端代码：

/*server03*/#include <sys/types.h>          /* See NOTES */#include <sys/socket.h>#include <netinet/in.h>#include <netinet/ip.h> /* superset of previous */#include <arpa/inet.h>#include <unistd.h>#include <stdlib.h>#include <stdio.h>#include <string.h>#include <signal.h>#define MAX_CLIENT 10#define MAX_READ  1024void handle(int sig)//辅助杀死子进程或者父进程{  printf("recv : %d\n",sig);  exit(0);}int main(){  int serv_fd,con_fd;//服务器端至少要有两个套接字文件描述符--一个用来监听，一个/其余多个用来和客户端通信  struct sockaddr_in serv_addr;//IPV4套接字结构体--服务器  struct sockaddr_in clt_addr;//IPV4套接字结构体--客户端  char addr_dst[INET_ADDRSTRLEN] = {0};//存储客户端IP地址  char buf[MAX_READ]={0};//应用程序自己的缓冲区  int optvar;//地址复用使用的参数  pid_t pid;//子进程PID  socklen_t addr_len;  ssize_t ret ;  signal(SIGUSR1,handle);//注册新号和处理函数  serv_fd = socket(AF_INET,SOCK_STREAM,0);//建立套接字  if(-1 == serv_fd)  {    perror("socket");    exit(-1);  }  if(setsockopt(serv_fd, SOL_SOCKET,SO_REUSEADDR,&optvar,sizeof(optvar))  == -1 )//地址复用  {    perror("setsockopt");    exit(-1);  }  /*设置地址*/  bzero(&serv_addr,sizeof(serv_addr));  serv_addr.sin_family = AF_INET;  serv_addr.sin_port = htons(8001);  serv_addr.sin_addr.s_addr = htons(INADDR_ANY);  if(bind(serv_fd,(struct sockaddr *)&serv_addr,sizeof(serv_addr)) == -1)//绑定端口  {    perror("bind");    exit(-1);  }  /*一旦调用listen函数--套接字就会变成被动套接字--用来监听客户端，让客户端连接他  被动套接字--只能接受连接，不能主动发送连接  做了两个队列：          一个已经完成三次握手，建立连接的队列--客户端发connect请求被响应，已经成功完成连接          一个是未完成成三次握手的队列--正在握手  */  if(listen(serv_fd,MAX_CLIENT)== -1)//开始监听  {    perror("listen");    exit(-1);  }  addr_len = sizeof(clt_addr);  printf("Accepting connections ...\n");  if((con_fd = accept(serv_fd,(struct sockaddr *)&clt_addr,&addr_len)) == -1)//由于是点对点--所以只能处理一个客户端的连接请求  {    perror("accept");    exit(-1);  }  printf("received from %s at PORT %d\n",inet_ntop(AF_INET,&clt_addr.sin_addr,addr_dst,sizeof(addr_dst)),ntohs(clt_addr.sin_port));  pid = fork();//创建子进程  if(pid > 0)//父进程负责读取客户端数据  {    while(1)    {      ret = read(con_fd,buf,sizeof(buf));//读取数据      if(ret == 0)      {        /*如果在读的过程中，对方已关闭，TCP/IP协议会返回一个0数据包*/        printf("client close\n");        break;      }      else if(ret < 0)      {        perror("read");        exit(-1);      }      fputs(buf,stdout);//打印内容      memset(buf,0,sizeof(buf));    }    close(serv_fd);//关闭套接字    close(con_fd);    kill(pid,SIGUSR1);//发送信号给子进程--通知他死期已到  }  else if(pid == 0)//子进程负责向客户端写入数据  {    printf("Please input alphabet:");    while(fgets(buf,sizeof(buf),stdin) != NULL)//从标准输入获取数据    {        write(con_fd,buf,strlen(buf));//给客户端发送数据      memset(buf,0,sizeof(buf));//缓冲区清理      printf("Please input alphabet:");    }  }  else if(pid == -1)//创建子进程错误  {    perror("fork");    close(serv_fd);    close(con_fd);    exit(-1);  }  return 0;}

客户端代码

#include <sys/types.h>          /* See NOTES */#include <sys/socket.h>#include <netinet/in.h>#include <netinet/ip.h> /* superset of previous */#include <arpa/inet.h>#include <unistd.h>#include <stdlib.h>#include <stdio.h>#include <string.h>#include <signal.h>#define MAX_BUF  1024void handle(int sig){  printf("recv : %d\n",sig);  exit(0);}int main(){  int clt_fd;  struct sockaddr_in serv_addr;  char buf[MAX_BUF]={0};  char addr_dst[INET_ADDRSTRLEN] = {0};  ssize_t ret;  pid_t pid;  signal(SIGUSR1,handle);  clt_fd = socket(AF_INET,SOCK_STREAM,0);  if(-1 == clt_fd)  {    perror("socket");    exit(-1);  }   serv_addr.sin_family = AF_INET;  serv_addr.sin_port = htons(8001);  serv_addr.sin_addr.s_addr = inet_addr("192.168.1.110");  if(connect(clt_fd,(struct sockaddr*)&serv_addr,sizeof(serv_addr)) == -1)  {    perror("connect");    exit(-1);  }  printf("Connect successfully\t%s at PORT %d\n",inet_ntop(AF_INET,&serv_addr.sin_addr,addr_dst,sizeof(addr_dst)),ntohs(serv_addr.sin_port));  pid = fork();  if(pid > 0)//父进程负责给服务器端发送数据  {    printf("Please input the alphabet:");    while(fgets(buf,sizeof(buf),stdin) != NULL)//从标准输入获取数据    {      write(clt_fd,buf,strlen(buf));//发送数据      memset(buf,0,sizeof(buf));//缓冲器清理      printf("Please input the alphabet:");    }  }  else if(pid == 0)//子进程负责从服务器端获取数据  {    while(1)    {      ret = read(clt_fd,buf,sizeof(buf));//读数据      if(ret == 0)      {        /*如果在读的过程中，对方已关闭，TCP/IP协议会返回一个0数据包*/        printf("server close\n");        break;      }      else if(ret < 0)      {        perror("read");        exit(-1);      }      fputs(buf,stdout);//打印内容      memset(buf,0,sizeof(buf));    }    close(clt_fd);    kill(getppid(),SIGUSR1);//发送信号给父进程--通知他死期已到  }  else if(pid == -1)//创建子进程错误  {    perror("fork");    close(clt_fd);    exit(-1);  }   return 0;}