Linux网络编程之实现多方消息收发demo

// 服务端

#include <stdio.h>
#include <sys/types.h>         
#include <sys/socket.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <stdlib.h>
#include <string.h>

int main(int argc, char **argv)
{
   int s_fd;
   int c_fd;
   char readBuf[128];
   int n_read;
   char msg[128] = {0};

   int mark = 0;

   struct sockaddr_in s_addr;
   struct sockaddr_in c_addr;

   if(argc != 3){
           printf("param is no good\n");
           exit(-1);
   }

   memset(&s_addr, 0, sizeof(struct sockaddr_in));
   memset(&c_addr, 0, sizeof(struct sockaddr_in));

   s_fd = socket(AF_INET, SOCK_STREAM, 0);
   if(s_fd == -1){
          perror("socket");
          exit(-1);
   }

   s_addr.sin_family = AF_INET;
   s_addr.sin_port = htons(atoi(argv[2]));
   inet_aton(argv[1], &s_addr.sin_addr);

   bind(s_fd, (struct sockaddr *)&s_addr, sizeof(struct sockaddr_in));

   listen(s_fd, 10);
   int clen = sizeof(struct sockaddr_in);
   
   while(1){

             c_fd = accept(s_fd, (struct sockaddr *)&c_addr, &clen);   
             if(c_fd == -1){
                      perror("accept");
             }
   
             mark++;
             printf("get connect: %s\n",inet_ntoa(c_addr.sin_addr));

             if(fork() == 0){

                       if(fork() == 0){
                               
                               while(1){

                                        sprintf(msg, "welcome No.%d clinet", mark);
                                        
                                        write(c_fd, msg, strlen(msg));
                                        sleep(3);
                               }
                       }
 
                         
                       while(1){
  
                                memset(readBuf,0,sizeof(readBuf));
                                n_read = read(c_fd, readBuf, 128);   
                                if(n_read == -1){
                                          perror("read");            
  
                                }else{
      
                                          printf("get massage: %d, %s\n",n_read,readBuf); 
                                }

                      }
                      break; 
                       
             }

   }

    return 0;

}

// 客户端

#include <stdio.h>
#include <sys/types.h>         
#include <sys/socket.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <stdlib.h>
#include <string.h>

int main(int argc, char **argv)
{
   int c_fd;
   char readBuf[128];
   int n_read;
   char msg[128] = {0};

   struct sockaddr_in c_addr;
   memset(&c_addr, 0, sizeof(struct sockaddr_in));

   c_fd = socket(AF_INET, SOCK_STREAM, 0);
   if(c_fd == -1){
          perror("socket");
          exit(-1);
   }

   c_addr.sin_family = AF_INET;
   c_addr.sin_port = htons(atoi(argv[2]));
   inet_aton(argv[1], &c_addr.sin_addr);

   if(connect(c_fd, (struct sockaddr *)&c_addr, sizeof(struct sockaddr_in)) == -1){

                     perror("connect");
                     exit(-1);

   }
   
    while(1){
             
             if(fork() == 0){
                       
                       while(1){

                                memset(msg, 0, sizeof(msg));   
                                
                                printf("input: ");
                                gets(msg);
                                write(c_fd, msg, strlen(msg));
                       }
           } 
           
            while(1){
                     
                      memset(readBuf,0,sizeof(readBuf));
                      n_read = read(c_fd, readBuf, 128);
                      if(n_read == -1){
                                 perror("read");

                      }else{

                                 printf("get massage from server: %d, %s\n",n_read,readBuf);
                      }
            }

   }

  

   return 0;

}