UNIX网络变成-套接字联网之TCP客户/服务器程序学习

最近开始学习UNIX网络编程，参考书是大师W.Richard Stevens的著作《UNIX网络编程-卷1：套接字联网API》。首先，阅读本书必须有一定的UNIX或者LINUX下C编程的基础知识。

首先，看一个linux下的C/S程序，来了解以下UNIX环境下套接字编程到底是怎么一回事。

/===================================================服务器程序=============================================================/

/* File Name: server.c */

 
#include <stdio.h>  
#include <stdlib.h>  
#include <string.h>  
#include <errno.h>  
#include <sys/types.h>  
#include <sys/socket.h>  
#include <netinet/in.h>
#include <signal.h>
 
/*********************************************宏定义*******************************************************/
#define DEFAULT_PORT 8000  
#define MAXLINE 4096  
/*********************************************函数声明*******************************************************/
void sig_child(int signo);


/*********************************************函数定义*******************************************************/
int main(int argc, char** argv)  
{  
    int     socket_fd, connect_fd;  
    struct  sockaddr_in     servaddr;  
    char    buff[4096];  
    int     n;  
    
    //初始化Socket  
    if( (socket_fd = socket(AF_INET, SOCK_STREAM, 0)) == -1 ){  
      printf("create socket error: %s(errno: %d)\n",strerror(errno),errno);  
      exit(0);  
    }  

    //初始化  
    memset(&servaddr, 0, sizeof(servaddr));  
    servaddr.sin_family = AF_INET;  
    servaddr.sin_addr.s_addr = htonl(INADDR_ANY);//IP地址设置成INADDR_ANY,让系统自动获取本机的IP地址。  
    servaddr.sin_port = htons(DEFAULT_PORT);//设置的端口为DEFAULT_PORT  
 
    //将本地地址绑定到所创建的套接字上  
    if( bind(socket_fd, (struct sockaddr*)&servaddr, sizeof(servaddr)) == -1){  
      printf("bind socket error: %s(errno: %d)\n",strerror(errno),errno);  
      exit(0);  
    }  
    //开始监听是否有客户端连接,并立即处理为客户端建立的子进程的状态
    signal(SIGCHLD, sig_child);// 防止子进程变为僵尸进程

    if( listen(socket_fd, 10) == -1){  
      printf("listen socket error: %s(errno: %d)\n",strerror(errno),errno);  
      exit(0);  
    }  
    printf("======waiting for client's request======\n");
 
    while(1){  
        //阻塞直到有客户端连接，不然多浪费CPU资源。  
        if((connect_fd = accept(socket_fd, (struct sockaddr*)NULL, NULL)) < 0){
      if(errno == EINTR)
        continue;      /*back ro for() */
      else
        printf("accept socket error: %s(errno: %d)",strerror(errno),errno);  
      continue;  
    }
 
    //接受客户端传过来的数据  
    n = recv(connect_fd, buff, MAXLINE, 0);  
    
    //向客户端发送回应数据  
    if(!fork()){ /* 子进程 */  
      if(send(connect_fd, "Hello,you are connected!\n", 26,0) == -1)  
        perror("send error");  
      close(connect_fd);  
      exit(0);  
    }  
    buff[n] = '\0';  
    printf("recv msg from client: %s\n", buff);  
    close(connect_fd);  
    }  
    close(socket_fd);  
}
 
//信号处理函数，用于处理服务器为客户端建立的子进程的状态信息
void sig_child(int signo)
{
  pid_t pid;
  int stat;
 
  //pid = wait(&stat);
  // WNOHANG,如果此时还有未结束的子进程，选择不阻塞
  while ((pid = waitpid(-1,&stat,WNOHANG)) > 0)
    printf("child %d terminated\n",pid);
  return;

}

/****************************************************************************以下是客户端程序*******************************************************************************/

    /* File Name: client.c */  
      
    #include<stdio.h>  
    #include<stdlib.h>  
    #include<string.h>  
    #include<errno.h>  
    #include<sys/types.h>  
    #include<sys/socket.h>  
    #include<netinet/in.h>  
/**************************************************函数定义*****************************************/      
#define MAXLINE 4096
#define max(a,b)    (((a) > (b))? (a):(b))  
#define min(a,b)    (((a) < (b))? (a):(b))
/**************************************************函数声明*****************************************/
void str_cli(FILE *fp, int sockfd);      
/**************************************************函数定义*****************************************/      
    int main(int argc, char** argv)  
    {  
        int    sockfd, n,rec_len;  
        char    recvline[4096], sendline[4096];  
        char    buf[MAXLINE];  
        struct sockaddr_in    servaddr;  
      
    // 提示，如果没有执行参数，则报出使用方法
        if( argc != 2){  
      printf("usage: ./client <ipaddress>\n");  
      exit(0);  
        }  
      
        // 建立SOCKET套接字
        if( (sockfd = socket(AF_INET, SOCK_STREAM, 0)) < 0){  
      printf("create socket error: %s(errno: %d)\n", strerror(errno),errno);  
      exit(0);  
        }  
      
    //
        memset(&servaddr, 0, sizeof(servaddr));  
        servaddr.sin_family = AF_INET;  
        servaddr.sin_port = htons(8000);  
        
    if( inet_pton(AF_INET, argv[1], &servaddr.sin_addr) <= 0){  
      printf("inet_pton error for %s\n",argv[1]);  
      exit(0);  
        }  
      
      
        if( connect(sockfd, (struct sockaddr*)&servaddr, sizeof(servaddr)) < 0){  
      printf("connect error: %s(errno: %d)\n",strerror(errno),errno);  
      exit(0);  
        }  
      
      
        //printf("send msg to server: \n");  
        //fgets(sendline, 4096, stdin);  
        //if( send(sockfd, sendline, strlen(sendline), 0) < 0)  
        //{  
    // printf("send msg error: %s(errno: %d)\n", strerror(errno), errno);  
    //exit(0);  
        //}  
        //if((rec_len = recv(sockfd, buf, MAXLINE,0)) == -1) {  
    // perror("recv error");  
    // exit(1);  
        //}  
        //buf[rec_len]  = '\0';  
        //printf("Received : %s ",buf);  
        //close(sockfd);
    
    str_cli(stdin,sockfd);
    
        exit(0);  
    }  
//
void str_cli(FILE *fp, int sockfd)
{
  int maxfdp1,stdineof;
  fd_set rset;
  //char senline[MAXLINE], recvline[MAXLINE];
  char buf[MAXLINE];
  int n;
 
  stdineof = 0;
  FD_ZERO(&rset);
 
  while(1){
    if(stdineof == 0)
      FD_SET(fileno(fp),&rset);
    //FD_SET(fileno(fp), &rset); // 设置文本描述符读集
    FD_SET(sockfd, &rset);
    maxfdp1 = max(fileno(fp),sockfd) + 1;
    select(maxfdp1, &rset, NULL, NULL,NULL);
    
    // 套接字是可读的
    if (FD_ISSET(sockfd, &rset)){/* socket is readble */
      if ((n = read(sockfd, buf,MAXLINE)) == 0){
    if(stdineof == 1)
      return;
    else{
      printf("str_cli: server terminated prematurely");
      exit(1);
    }
      }
      //fputs(recvline,stdout);
      write(fileno(stdout),buf,n);
    }
    
    // 输入是可读的
    if (FD_ISSET(fileno(fp), &rset)){
      if((n = read(fileno(fp),buf,MAXLINE)) == 0){
    stdineof = 1;
    shutdown(sockfd,SHUT_WR);//发送FIN
    FD_CLR(fileno(fp),&rset);
    continue;
      }
      write(sockfd,buf,n);
    }
  }
}