基于TCP协议的服务器编程

#include<netdb.h>
#include<sys/socket.h>
#include<unistd.h>
#include<string.h>
#include<stdio.h>
#include<stdlib.h>
#include<memory.h>
#include<signal.h>
#include<time.h>

int sockfd;

void sig_handler(int signo)
{
   if(signo == SIGINT)
   {
       printf("server close ");
    /*
   *    步骤6：关闭socket
   */
       close(sockfd);
       exit(1);
   }
}

/*输出连接上来的客户端相关信息*/
void out_addr(struct sockaddr_in *clientaddr)
{
    //将端口从网络字节序转换成主机字节序
   int port = ntohs(clientaddr->sin_port);
   char ip[16];
   memset(ip ,0,sizeof(ip));
   //将ip地址从网络字节序转换成点分十进制
   inet_ntop(AF_INET,&clientaddr->sin_addr.s_addr,ip,sizeof(ip));
   printf("client:%s(%d) connected ",ip,port);
}

void do_service(int fd)
{
   //获得系统时间
   long t = time(0);
   char *s =ctime(&t);//转成字符串形式
   size_t size = strlen(s) * sizeof(char);
   //将服务器端获得的系统时间写回到客户端
   if(write(fd,s,size) != size )
   {
       perror("write error");
   }
}

int main(int argc,char *argv[])
{
   if(argc < 2)
   {
       printf("usage: %s #port ", argv[0]);
       exit(1);
   }
   if(signal(SIGINT,sig_handler) == SIG_ERR)
   {
       perror("signal sigint error");
       exit(1);
   }

   /*步骤1：创建socket（套接字）
   *       注：socket创建在内核中，是一个结构体
   *       AF_INET：IPV4
   *       SOCK_STREAM:tcp协议
   */
   sockfd = socket(AF_INET,SOCK_STREAM,0);
    if (sockfd < 0)
   {
       perror("socket error");
       exit(1);
   }

   /*
   * 步骤2：调用bind函数将socket和地址（包括ip、port）进行绑定
   */
   struct sockaddr_in serveraddr;
   memset(&serveraddr,0,sizeof(serveraddr));
    //往地址中填入ip、port、internet地址族类型
    serveraddr.sin_family = AF_INET;//IPV4
   serveraddr.sin_port = htons(atoi(argv[1]));//端口
   serveraddr.sin_addr.s_addr = INADDR_ANY;
   if(bind(sockfd,(struct sockaddr*)&serveraddr,sizeof(serveraddr) < 0 )
   {
       perror("bind error");
       exit(1);
   }

   /*
   * 步骤3：调用listen函数启动监听(指定port监听)
   *         通知系统去接受来自客户端的连接请求
   *         将接受到的客户端连接请求放置到对应的队列中
   *         第二个参数：指定队列的长度
   */
   if(listen(socket,10) < 0 )
   {
       perror("listen error");
       exit(1);
   }

   /*
   * 步骤4：调用accept函数从队列当中获得一个客户端的请求连接
   *           并返回新的socket描述符
   * 注意：若没有客户端连接，调用此函数后会阻塞，
   */
   struct sockaddr_in clientaddr;
   socklen_t clientaddr_len = sizeof(clientaddr);
   while(1)
   {
       int fd = accept(sockfd,(struct sockaddr*)&clientaddr,&clientaddr_len);
       if(fd < 0)
       {
           perror("accept error");
           continue;
       }
   /*
   *   步骤5：调用I/O函数（read/write）和连接的客户端进行双向的通信
   */
   out_addr(&clientaddr);
   do_service(fd);
   /*
   *    步骤6：关闭socket
   */
   close(fd);

   }

   return 0;
}