学习 UNIX网络编程卷1：套接字 笔记1-实现一个简单的回射客户服务器程序

一：客户端程序

1、头文件：unp.h

#ifndef _unp_h
#define _unp_h
#include <sys/types.h> /* basic system data types */
#include <sys/socket.h> /* basic socket definitions */
#include <sys/time.h> /* timeval{} for select() */
#include <time.h>  /* timespec{} for pselect() */
#include <netinet/in.h> /* sockaddr_in{} and other Internet defns */
#include <arpa/inet.h> /* inet(3) functions */
#include <errno.h>
#include <fcntl.h>  /* for nonblocking */
#include <netdb.h>
#include <signal.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/stat.h> /* for S_xxx file mode constants */
#include <sys/uio.h>  /* for iovec{} and readv/writev */
#include <unistd.h>
#include <sys/wait.h>
#include <sys/un.h>  /* for Unix domain sockets */
#define MAXLINE 4096
#define LISTENQ 5
#define SERV_PORT   9877
int Socket(int sockfd, int type, int protocol);
void Bind(int sockfd, const struct sockaddr *myaddr, socklen_t addrlen);
void Listen(int sockfd,int backlog);
void Connect(int sockfd, const struct sockaddr *servaddr, socklen_t addrlen);
int Accept(int sockfd, struct sockaddr *cliaddr, socklen_t *addrlen);
void Write(int fd,void *buf,size_t count);
void Close(int sockfd);
void err_sys(char *err_str);
int Fork();
void str_echo(int sockfd);
ssize_t writen(int fd, char *buf, size_t n);
ssize_t Writen(int fd, char *buf, size_t n);
void str_cli(FILE *fp, int sockfd);
ssize_t readline(int fd, void *vptr, size_t maxlen);
ssize_t Readline(int fd, void *vptr, size_t maxlen);
#endif

2、客户端主程序：clien.c

#include "unp.h"

int main(int argc,char **argv)
{
    char ip_addr[20];
    int sockfd;
    struct sockaddr_in servaddr;
    /**add for ipaddress **/
    /*
    if (argc != 2)
    {
        err_sys("usage:clien <IPaddress>");
        return -1;
    }
    */

    bzero(&servaddr, sizeof(servaddr));
    bzero(ip_addr,sizeof(ip_addr));
    /**add for ipaddress**/
    //strcpy(ip_addr,argv[1]);
    sockfd = Socket(AF_INET, SOCK_STREAM, 0);

    servaddr.sin_family = AF_INET;
    servaddr.sin_port = htons(SERV_PORT);
    servaddr.sin_addr.s_addr = inet_addr("127.0.0.1");
    //servaddr.sin_addr.s_addr = inet_addr(ip_addr);
    Connect(sockfd, (struct sockaddr *)&servaddr, sizeof(servaddr));

    str_cli(stdin,sockfd);

    return 0;

}

3、客户端包裹函数：pack.c

#include "unp.h"

void err_sys(char *err_str)
{
  perror(err_str);
}

int Socket(int sockfd, int type, int protocol)
{
  int n=0;
  if ((n=socket(sockfd,type,protocol))<0)
  {
    err_sys("socket error");
  }
  return n;
}

void Bind(int sockfd, const struct sockaddr *myaddr, socklen_t addrlen)
{
  int n=0;
  if ((n = bind(sockfd,myaddr,addrlen))<0)
  {
    err_sys("bind error");
  }
}

void Listen(int sockfd,int backlog)
{
  char *ptr=NULL;
  if ((ptr=getenv("LISTENQ"))!=NULL)
  {
    backlog=atoi(ptr);
  }
  if (listen(sockfd,backlog)<0)
  {
    err_sys("listen error");
  }
}

void Connect(int sockfd,const struct sockaddr *servaddr, socklen_t addrlen)
{
  if(connect(sockfd,servaddr,addrlen)<0)
  {
    err_sys("connect error");
  }
  else
  {
    printf("connect success!
");
  }
}

int Accept(int sockfd, struct sockaddr *cliaddr, socklen_t *addrlen)
{
  int n;
  again:
    if ((n=accept(sockfd,cliaddr,addrlen))<0)
    {
#ifndef EPROTO
      if(errno == EPROTO || errno == ECONNABORTED)
#else
      if(errno == ECONNABORTED)
#endif
      {
          goto again;
      }
      else
      {
        err_sys("accept error");
      }
    }
  return n;
}

ssize_t Writen(int fd,char *buf, size_t n)
{
  ssize_t ret;
  ret = writen(fd,buf,n);
  if (ret < 0)
  {
    err_sys("writen error");
    return -1;
  }
  return ret;
}

ssize_t Readline(int fd, void *vptr, size_t maxlen)
{
  ssize_t n;
  n=readline(fd,vptr,maxlen);
  if (n<0)
  {
    err_sys("readline error");
    return -1;
  }
  return n;
}

4、客户端功能行函数：func.c

#include "unp.h"
ssize_t writen(int fd, char *buf, size_t n)
{
  size_t nleft;
  ssize_t nwritten;
  char *ptr;
  ptr = buf;
  nleft = n ;
  while(nleft > 0)
  {
    nwritten = write(fd, ptr, nleft);
    if (nwritten < 0 && errno == EINTR)
    {
      nwritten = 0;      
    }
    else if (nwritten < 0)
    {
      err_sys("write error");
      return -1;
    }

    nleft -= nwritten;
    ptr += nwritten;
  }
  return n;
}

void str_echo(int sockfd)
{
  ssize_t n;
  char buf[MAXLINE];
  again:
    while ((n = read(sockfd, buf, MAXLINE))>0)
      Writen(sockfd, buf, n);
  if (n < 0 && errno == EINTR)
  {
    goto again;
  }
  else if (n < 0)
  {
    err_sys("str_echo error");
  }
}
ssize_t readline(int fd, void *vptr, size_t maxlen)
{
    ssize_t n,rc;
    char c,*ptr;
    ptr = vptr;
    for (n = 1; n < maxlen; n++)
    {
        again:
            if ((rc = read(fd, &c, 1)) == 1)
            {
                *ptr++ = c;
                if (c == '
')
                {
                    break;
                }
            }
            else if (rc == 0)
            {
                *ptr = 0;
                return (n-1);
            }
            else
            {
                if (errno == EINTR)        
                {
                    goto again;
                }
                return -1;
            }
    }
    *ptr = 0;
    return n;
}

void str_cli(FILE *fp, int sockfd)
{
    char sendline[MAXLINE],recvline[MAXLINE];
    bzero(sendline,sizeof(sendline));
    bzero(recvline,sizeof(recvline));

    while(fgets(sendline,MAXLINE,fp) != NULL)
    {
        Writen(sockfd,sendline,strlen(sendline));
        if (Readline(sockfd, recvline, MAXLINE) == 0)
        {
            err_sys("str_cli Readline error");
        }
        fputs(recvline, stdout);
    }
}

5、客户端Makefile

clien:clien.o pack.o func.o
    gcc clien.o pack.o func.o -o clien
clien.o:clien.c unp.h 
    gcc -c clien.c -o clien.o
pack.o:pack.c unp.h 
    gcc -c pack.c -o pack.o
func.o:func.c unp.h 
    gcc -c func.c -o func.o    
.PHONY:clean
clean:
    rm -f *.o

二：服务器程序

1、头文件：unp.h

#ifndef _unp_h
#define _unp_h
#include <sys/types.h> /* basic system data types */
#include <sys/socket.h> /* basic socket definitions */
#include <sys/time.h> /* timeval{} for select() */
#include <time.h>  /* timespec{} for pselect() */
#include <netinet/in.h> /* sockaddr_in{} and other Internet defns */
#include <arpa/inet.h> /* inet(3) functions */
#include <errno.h>
#include <fcntl.h>  /* for nonblocking */
#include <netdb.h>
#include <signal.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/stat.h> /* for S_xxx file mode constants */
#include <sys/uio.h>  /* for iovec{} and readv/writev */
#include <unistd.h>
#include <sys/wait.h>
#include <sys/un.h>  /* for Unix domain sockets */
#define MAXLINE 4096
#define LISTENQ 5
#define SERV_PORT   9877
typedef void (*sighandler_t)(int);
sighandler_t Signal(int signum, sighandler_t handler);
int Socket(int sockfd, int type, int protocol);
void Bind(int sockfd, const struct sockaddr *myaddr, socklen_t addrlen);
void Listen(int sockfd,int backlog);
void Connect(int sockfd, const struct sockaddr *servaddr, socklen_t addrlen);
int Accept(int sockfd, struct sockaddr *cliaddr, socklen_t *addrlen);
void Write(int fd,void *buf,size_t count);
void Close(int sockfd);
void err_sys(char *err_str);
int Fork();
void str_echo(int sockfd);
ssize_t writen(int fd, char *buf, size_t n);
ssize_t Writen(int fd, char *buf, size_t n);
void sig_chlid(int signo);
#endif

2、服务器主程序：serv.c

#include "unp.h"
void sig_chld(int );

int main(int argc, char const *argv[])
{
    int listenfd,connfd;
    pid_t childpid;
    socklen_t clilen;
    struct sockaddr_in cliaddr,servaddr;
    listenfd = Socket(AF_INET, SOCK_STREAM, 0);//监听套接字创建
    bzero(&servaddr, sizeof(servaddr));
    servaddr.sin_family = AF_INET;//服务器地址族
    servaddr.sin_addr.s_addr = htonl(INADDR_ANY);//服务器地址赋值，一般服务器地址不指定
    servaddr.sin_port = htons(SERV_PORT);//服务器端口赋值
    Bind(listenfd,(struct sockaddr*)&servaddr, sizeof(servaddr));//绑定监听套接字与服务器的端口
    Listen(listenfd, LISTENQ);  //监听
    Signal(SIGCHLD, sig_chld);  //回收僵死子进程
    while(1)
    {
        clilen = sizeof(cliaddr);
        connfd = Accept(listenfd, (struct sockaddr *)&cliaddr, &clilen);//accept创建已连接套接字connfd
        if (connfd<0) {
          if(errno == EINTR)//处理系统中断
          {
            continue;
          }
          else
          {
            err_sys("accept error");
          }
        }
        if ((childpid = Fork()) == 0)//创建子进程
        {
            Close(listenfd);//子进程关闭监听套接字
            str_echo(connfd);//调用回射函数
            exit(0);
        }
        else//父进程
        {
            Close(connfd);//父进程关闭已连接套接字
        }

    }
    return 0;
}

3、包裹函数：pack.c

#include "unp.h"

void err_sys(char *err_str)
{
  perror(err_str);
}

int Socket(int sockfd, int type, int protocol)
{
  int n=0;
  if ((n=socket(sockfd,type,protocol))<0)
  {
    err_sys("socket error");
  }
  printf("socket success!sockfd=%d
",n);
  return n;
}

void Bind(int sockfd, const struct sockaddr *myaddr, socklen_t addrlen)
{
  int n=0;
  if ((n = bind(sockfd,myaddr,addrlen))<0)
  {
    err_sys("bind error");
  }
  printf("bind success
");
}

void Listen(int sockfd,int backlog)
{
  char *ptr=NULL;
  if ((ptr=getenv("LISTENQ"))!=NULL)
  {
    backlog=atoi(ptr);
  }
  if (listen(sockfd,backlog)<0)
  {
    err_sys("listen error");
  }
  printf("listen success
");
}

void Connect(int sockfd, const struct sockaddr *servaddr, socklen_t addrlen)
{
  if(connect(sockfd,servaddr,addrlen)<0)
  {
    err_sys("connect error");
  }
  printf("connect success!
");
}

int Accept(int sockfd, struct sockaddr *cliaddr, socklen_t *addrlen)
{
  int n;
  again:
    if ((n=accept(sockfd,cliaddr,addrlen))<0)
    {
#ifndef EPROTO
      if(errno == EPROTO || errno == ECONNABORTED)
#else
      if(errno == ECONNABORTED)
#endif
      {
          goto again;
      }
      else
      {
        err_sys("accept error");
      }
    }
  return n;
}

/*
char* Inet_ntop(int family, const void *addrptr,char *strptr,size_t len)
{
  char buff[1024+1];
  bzero(buff,sizeof(buff));
  if(inet_ntop(family,addrptr,strptr,len)==NULL)
  {
    err_sys("inet_error");
  }
  strcpy(buff,strptr);
  return buff;
}
*/
void Write(int fd,void *buf,size_t count)
{
    ssize_t n;
    n=write(fd,buf,count);
    if(n!=count)
    {
        err_sys("write error");
    }

}
void Close(int sockfd)
{
    if(close(sockfd)==-1)
    {
        err_sys("close error");
    }
}

int Fork()
{
  pid_t pid;
  if ((pid = fork()) < 0)
  {
    err_sys("fork error");
  }
  else
    return pid;
}

ssize_t Writen(int fd,char *buf, size_t n)
{
  ssize_t ret;
  ret = writen(fd,buf,n);
  if (ret < 0)
  {
    err_sys("writen error");
    return -1;
  }
  return ret;
}

sighandler_t Signal(int signum, sighandler_t handler)
{
  if (signal(signum, handler)==SIG_ERR)
  {
    err_sys("signal error");
    exit(-1);
  }
  
}

4、功能性函数：func.c

#include "unp.h"

ssize_t writen(int fd, char *buf, size_t n)
{
  size_t nleft;
  ssize_t nwritten;
  char *ptr;
  ptr = buf;
  nleft = n ;
  while(nleft > 0)
  {
    nwritten = write(fd, ptr, nleft);
    if (nwritten < 0 && errno == EINTR)
    {
      nwritten = 0;
    }
    else if (nwritten < 0)
    {
      err_sys("write error");
      return -1;
    }

    nleft -= nwritten;
    ptr += nwritten;
  }
  return n;
}

void str_echo(int sockfd)
{
  ssize_t n;
  char buf[MAXLINE];
  again:
    while ((n = read(sockfd, buf, MAXLINE))>0)
      Writen(sockfd, buf, n);
  if (n < 0 && errno == EINTR)
  {
    goto again;
  }
  else if (n < 0)
  {
    err_sys("str_echo error");
  }
}

void sig_chld(int signum)
{
    pid_t pid;
    int stat;

    while(pid = waitpid(-1, &stat, WNOHANG) >0 )
        printf("chlid %d terminated!
",pid);
    return ;
}

5、服务器Makefile

serv:serv.o pack.o func.o
    gcc serv.o pack.o func.o -o serv
serv.o:serv.c unp.h
    gcc -c serv.c -o serv.o
pack.o:pack.c unp.h
    gcc -c pack.c -o pack.o
func.o:func.c unp.h
    gcc -c func.c -o func.o
.PHONY:clean
clean:
    rm -f *.o