多进程实现NTP服务器

多进程实现NTP服务器

服务器端：server.c

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <arpa/inet.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <time.h>
#include <unistd.h>
#include "proto.h"

#define IPSIZE 40

static void func(int fd) //获取时间戳传给客户端
{
　　FILE *fp = fdopen(fd,"r+");
　　fprintf(fp,FMT,(long long)time(NULL));
　　sleep(20); //测试
　　fclose(fp);
}

int main()
{
　　int ret;
　　int lfd,newfd;
　　struct sockaddr_in laddr,raddr;
　　socklen_t lenth;
　　char ip[IPSIZE];
　　pid_t pid;

　　lfd = socket(AF_INET,SOCK_STREAM,0); //创建socket
　　if(lfd <0)
　　{
　　　　perror("socket()");
　　　　exit(1);
　　}

　　int val =1;
　　ret = setsockopt(lfd,SOL_SOCKET,SO_REUSEPORT,&val,sizeof(val)); //设置端口复用
　　if(ret <0)
　　{
　　　　perror("setsockopt()");
　　　　exit(1);
　　}

　　laddr.sin_family = AF_INET; //初始化结构体
　　laddr.sin_port = htons(atoi(PORT));
　　inet_pton(AF_INET,"0.0.0.0",&laddr.sin_addr.s_addr);
　　ret = bind(lfd,(void *)&laddr,sizeof(laddr)); //绑定端口和IP
　　if(ret<0)
　　{
　　　　perror("bind()");
　　　　exit(1);
　　}

　　ret = listen(lfd,200); //监听最大200
　　if(ret <0)
　　{
　　　　perror("listen()");
　　　　exit(1);
　　}

　　lenth = sizeof(raddr);
　　while(1)
　　{
　　　　newfd = accept(lfd,(void *)&raddr,&lenth); //等待客户端连接
　　　　if(newfd<0)
　　　　{
　　　　　　perror("accpt()");
　　　　　　break;
　　　　}

　　　　pid = fork(); //创建子进程
　　　　if(pid<0)
　　　　{
　　　　　　perror("fork()");
　　　　　　close(newfd);
　　　　　　break;
　　　　}
　　　　else if(pid ==0) //子进程
　　　　{
　　　　　　close(lfd); //关闭服务器端socket文件描述符
　　　　　　inet_ntop(AF_INET,&raddr.sin_addr.s_addr,ip,IPSIZE);
　　　　　　printf("-----IP:%s PORT:%d-----\n",ip,ntohs(raddr.sin_port));
　　　　　　func(newfd);
　　　　　　exit(0);
　　　　}
　　　　else //父进程
　　　　　　close(newfd); //关闭客户端socket文件描述符
　　}