Linux TCP实现简单聊天程序

Linux TCP实现简单聊天程序

/*tcp_chat_server.c*/
#include
#include
#include
#include
#include
#include
#include
#include
#include
#define MAXBUF 1024
int main(int argc,char *argv[])
{
  int pid;
  int sockfd,client_fd;
  socklen_t len;
  struct sockaddr_in server_addr,client_addr;
  unsigned int myport,lisnum;
  char buf[MAXBUF+1];
  if(argv[2])//对端口号进行赋值
    myport=atoi(argv[2]);
  else
    myport=7575;
  if(argv[3])//对监听队列大小进行赋值
    lisnum=atoi(argv[3]);
  else
     lisnum=5;
  if((sockfd=socket(AF_INET,SOCK_STREAM,0))==-1)
  {
   perror("socket");
   exit(EXIT_FAILURE);
  }
  bzero(&server_addr,sizeof(server_addr));
  server_addr.sin_family=AF_INET;//地址协议
  server_addr.sin_port=htons(myport);//端口号
  if(argv[1])//将点分十进制转换为网络顺序IP地址
     server_addr.sin_addr.s_addr=inet_addr(argv[1]);
  else
     server_addr.sin_addr.s_addr=INADDR_ANY;//设置本机地址
  if(bind(sockfd,(struct sockaddr*)&server_addr,sizeof(struct sockaddr))==-1)
  {
    perror("bind");
    exit(EXIT_FAILURE);
  }
  if(listen(sockfd,lisnum)==-1)
  {
    perror("listen");
    exit(EXIT_FAILURE);
  }
  printf("wait for connection\n");
  len=sizeof(struct sockaddr);
  if((client_fd=accept(sockfd,(struct sockaddr*)&client_addr,&len))==-1)
  {//堵塞等待连接
    perror("accept");
    exit(EXIT_FAILURE);
  }
  else
    printf("server:got connection from %s,port %d,socket %d\n",inet_ntoa(client_addr.sin_addr),ntohs(client_addr.sin_port),client_fd);
  if(-1==(pid=fork()))//创建新进程
  {
   perror("fork");
   exit(EXIT_FAILURE);
  }
  else if(pid==0)
  {
    while(1)
    {
      bzero(buf,MAXBUF+1);
      printf("input the message to send:\n");
      fgets(buf,MAXBUF,stdin);
      /*if(!strncasecmp(buf,"quit",4));
      {
         printf("I will close the connection!");
         break;
      }*/
      len=send(client_fd,buf,strlen(buf)-1,0);
      if(len<0)
      {
        printf("message'%s' send failure! error code is %d,errno  message is '%s'\n",buf,errno,strerror(errno));
        break;
      }
   }
 }
 else
 {
  while(1)
  {
   bzero(buf,MAXBUF+1);
   len=recv(client_fd,buf,MAXBUF,0);
   if(len>0)
     printf("message recv successful:'%s',%dByte recv\n",buf,len);
   else if (len<0)
   {
     printf("recv failure!errno code is %d,errno message is '%s'\n",errno,strerror(errno));
     break;
   }
  }
}
close(client_fd);
close(sockfd);
return 0;
}
      
         
   
   
  
  
  

/*tcp_chat_client.c*/
#include
#include
#include
#include
#include
#include
#include
#include
#include
#define MAXBUF 1024
int main(int argc,char *argv[])
{
  int sockfd,len;
  struct sockaddr_in ser_addr;
  char buf[MAXBUF+1];
  if(argc!=3)
  {
    printf("error format,it must be:\n%s IP port\n",argv[0]);
    exit(EXIT_FAILURE);
  }
  if((sockfd=socket(AF_INET,SOCK_STREAM,0))==-1)
  {
   perror("socket");
   exit(errno);
  }
  printf("socket created\n");
  bzero(&ser_addr,sizeof(ser_addr));
  ser_addr.sin_family=AF_INET;
  ser_addr.sin_port=htons(atoi(argv[2]));/*atoi（）把命令行字符串转化为整数*/
  /*inet_aton()将点分十进制IP转化为网络流IP，并用新的IP更新第二个参数的结构体*/
  if(inet_aton(argv[1],(struct in_addr*)&ser_addr.sin_addr.s_addr)==0)
  {
   perror(argv[1]);
   exit(errno);
  }
  if(connect(sockfd,(struct sockaddr*)&ser_addr,sizeof(ser_addr))==-1)//发起连接请求
  {
   perror("connect");
   exit(errno);
  }
  printf("server connected\n");
  pid_t pid;
  if(-1==(pid=fork()))//创建新进程
  {
   perror("fork");
   exit(EXIT_FAILURE);
  }
  else if(pid==0)//子进程用于数据接收
  {
    while(1)
    {
      bzero(buf,MAXBUF+1);
      len=recv(sockfd,buf,MAXBUF,0);
      if(len>0)
          printf(" recv successful:'%s',%dByte recv\n",buf,len);
      else if (len<0)
      {
         perror("recv");
         break;
      }
      else
      {
        printf(" the other one close,quit\n");
        break;
      }
    }
 }
else//父进程用于数据发送
{
 while(1)
 {
   bzero(buf,MAXBUF+1);
   printf("pls send message to send:\n");
   fgets(buf,MAXBUF,stdin);
   /*if(!strncasecmp(buf,"quit",4));
    {
      printf("I will quit!\n");
      break;
    }*/
   len=send(sockfd,buf,strlen(buf)-1,0);
   if(len<0)
   {
     perror("send");
     break;
   }
  }
}
close(sockfd);
return 0;
}