#include<string.h>
#include<signal.h>
#include<stdio.h>
#include<sys/socket.h>
#include<stdlib.h>
#include<netdb.h>
#include<pthread.h>
#include<memory.h>
#include<semaphore.h>
int Thread_num=0,count=0; //定义客户端计数器，写线程计数器
int sockfd;
sem_t sem,sem2;   
pthread_mutex_t tmutex,cmutex;
pthread_attr_t pattr; //定义要创建的线程属性
void ct_thread(char* arg,int acpfd);
char chatct[1024]; //定义聊天缓存
void get_sys_time(char* tbuf)  //函数功能：得到当前系统时间并进行裁剪
{
    long t=time(0);
    char *stime=ctime(&t);
    int i=11;
    tbuf[0]=‘(‘;
    for(;i<20;i++)
        tbuf[i-10]=stime[i];
    tbuf[i-11]=‘)‘;
    tbuf[i-10]=‘\0‘;
}
void* do_read(void* arg)    //函数功能：读线程，用于接收客户端发来的信息，保存到缓冲区
{
    int acfd=(int)arg;
    char timeb[13];
//    char name[7];
//    memset(name,0,sizeof(name));
//    read(acfd,name,sizeof(name));
//    name[sizeof(name)-1]=‘\0‘;
//    puts(name);
    while(1)
    {    
        int ret;
        char buf[1024];
        memset(buf,0,sizeof(buf));
        ret=read(acfd,buf,sizeof(buf));
        sem_wait(&sem2);
        if(ret<0)
        {
            perror("do_read errro");
            continue;
        }
        else if(ret==0)
        {
            close(acfd);
            sem_post(&sem2);
            printf("a person exit\n");
            break;
        }
        else
        {
            memset(timeb,0,sizeof(timeb));
//            strcpy(chatct,name);
//            puts(chatct);
//            strcat(chatct,":");
//            puts(chatct);
            get_sys_time(timeb);
            strcpy(chatct,buf);
            strcat(chatct,timeb);
//            puts(chatct);
    //        printf("do_read %lu\n",pthread_self());
            sem_post(&sem);
        }
    //    puts(chatct);
    }
    return (void*)0;
}
void* do_write(void* arg)  //函数功能：写线程，用于将缓冲区的数据发送到连接上来的每个客户端，当所有客户端接收到消息后，清空缓冲区。
{
    int fd=(int)arg;
    while(1)
    {
        sem_wait(&sem);
        if(write(fd,chatct,sizeof(chatct))<0)
        {
            Thread_num--;
            if(count==Thread_num)
            {    
                memset(chatct,0,sizeof(chatct));
                pthread_mutex_lock(&cmutex);
                count=0;
                pthread_mutex_unlock(&cmutex);
                sem_post(&sem2);
                
            }
            else if(count<Thread_num)
            {
                sem_post(&sem);
            }
            close(fd);
            break;
        }
        else
        {
            pthread_mutex_lock(&cmutex);
            count++;
            pthread_mutex_unlock(&cmutex);
        }
        if(count<Thread_num)
        {
            sem_post(&sem);
            usleep(1);
        }
        else if(count==Thread_num)
        {    
            memset(chatct,0,sizeof(chatct));
            pthread_mutex_lock(&cmutex);
            count=0;
            pthread_mutex_unlock(&cmutex);
            sem_post(&sem2);
        }
    }
    return (void*)0;
}
void do_thread(int acpfd)  //函数功能：对于每个连接上来的客户端 创建一个读线程，一个写线程
{
//    char* start="------------------welcome my chatroom--------------------\n";
//    write(acpfd,start,strlen(start));
    pthread_mutex_lock(&cmutex);
    Thread_num++;     //每创建一个 线程个数计数加1
    pthread_mutex_unlock(&cmutex);
    ct_thread("read",acpfd);
    ct_thread("write",acpfd);
}
void ct_thread(char* arg,int acpfd) //函数功能，根据传参不同，创建不同类型的函数。
{
    pthread_t pt;
/*    if(!strcmp(arg,"create"))
    {
        if(pthread_create(&pt,&pattr,thread_ct,(void*)acpfd)<0)
        {
            perror("thread creat error");
            exit(1);
        }
    }
*/  //以分离方式创建线程可避免资源无法回收  
    if(!strcmp(arg,"read"))
    {
        if(pthread_create(&pt,&pattr,do_read,(void*)acpfd)<0)
        {
            perror("thread creat error");
            exit(1);
        }
    }
    else if(!strcmp(arg,"write"))
    {
        if(pthread_create(&pt,&pattr,do_write,(void*)acpfd)<0)
        {
            perror("thread creat error");
            exit(1);
        }
    }
}
void mysighand(int signo) //函数功能：当程序结束，处理返回的信号，释放资源
{
    if(signo==SIGINT)
    {
        printf("server close!\n");
        close(sockfd);
        sem_destroy(&sem);
        sem_destroy(&sem2);
        pthread_mutex_destroy(&cmutex);
        pthread_mutex_destroy(&tmutex);
        pthread_attr_destroy(&pattr);
        exit(1);
    }
}
int main(int argc,char* argv[])
{
    if(argc<2)
    {
        perror("argc error");
        exit(1);
    }
    if(signal(SIGINT,mysighand)==SIG_ERR) //登记信号处理函数
    {    
        perror("signal error");
        exit(1);
    }
/*  1 初始化
    sem_init(&sem,0,0);
    sem_init(&sem2,0,1);
    pthread_mutex_init(&tmutex,NULL);
    pthread_mutex_init(&cmutex,NULL);
    pthread_attr_init(&pattr);
*/
　  if(pthread_attr_setdetachstate(&pattr,PTHREAD_CREATE_DETACHED)<0) //设置分离属性
    {
        perror("setdetached error");
        exit(1);
    }
    memset(chatct,0,sizeof(chatct)); //初始化缓存区
     sockfd=socket(AF_INET,SOCK_STREAM,0);//以TCP方式创建socket
    if(sockfd<0)
    {
        perror("sockfd error");
        exit(1);
    }
    struct sockaddr_in ser;
    ser.sin_family=AF_INET;　　　　　　　　//IP类型:IPV4
    ser.sin_port=htons(atoi(argv[1]));  //主机字节序转换成网络字节序
    ser.sin_addr.s_addr=INADDR_ANY;    　//主机所有可访问IP　
    if(bind(sockfd,(struct sockaddr*)&ser,sizeof(ser))<0) //socket和IP绑定
    {
        perror("bind error");
        exit(1);
    }
    if(listen(sockfd,10)<0)         //进行监听等待客户端连接
    {
        perror("listen error");
        exit(1);
    }    
    while(1)
    {
        int acpfd;
        if((acpfd=accept(sockfd,NULL,NULL))<0)  //处理每个连接上来的客户端，如果无客户端连接，则阻塞
            break;
        else
        {
             do_thread(acpfd); //执行处理连接函数
        }
    }
    return 0;
}