業(yè)精于勤荒于嬉,行成于思?xì)в陔S!

#include "server.h"
#include <sys/socket.h>
#include <arpa/inet.h>
#include <netinet/in.h>

int sockfd; //服務(wù)器本身的描述符
struct User u[200] = {}; //用來(lái)保存用戶的信息
int size;//數(shù)組的下標(biāo),同時(shí)也是客戶端的個(gè)數(shù)

void init(void) //通信準(zhǔn)備工作
{
    printf("聊天室服務(wù)器馬上啟動(dòng)....\n");
    sockfd = socket(AF_INET,SOCK_STREAM,0);
    if(-1 == sockfd)
    {
        perror("fail to socket");
        printf("服務(wù)器故障!\n");
        exit(-1);
    }
    //準(zhǔn)備通信地址和綁定服務(wù)器
    struct sockaddr_in addr;
    addr.sin_family = AF_INET;
    addr.sin_port = htons(2222); //端口的值,轉(zhuǎn)換成網(wǎng)絡(luò)的格式
    addr.sin_addr.s_addr = inet_addr("172.16.1.21");
    if(-1 == bind(sockfd,(struct sockaddr*)&addr,sizeof(addr)))
    {
        perror("fail to bind");
        exit(-1);
    }
    printf("bind is ok,歡迎訪問(wèn)!\n");
    //設(shè)置監(jiān)聽(tīng)
    if(-1 != listen(sockfd,200))
    {
        printf("監(jiān)聽(tīng)已經(jīng)設(shè)置,一切準(zhǔn)備就緒!\n");
    }
}

void send_msg(char *p_msg) //發(fā)送消息的函數(shù)
{
    int num = 0;
    for(num = 0;num < size;num++)
    {
        if(u[num].fd) //如果是有效的時(shí)候才發(fā)送
        {
            send(u[num].fd,p_msg,sizeof(p_msg),0);
        }
    }
}

void* pthread_deal(void* p) //線程處理函數(shù)
{
    pthread_detach(pthread_self());
    int fd2 = *(int*)p; //取得客戶端的sockfd
    u[size].fd = fd2;  //放入結(jié)構(gòu)體中
    char name[20] = {};
    int res = recv(fd2,name,sizeof(name),0);
    if(res > 0)
    {
        strcpy(u[size].name,name);//存放用戶名
    }
    size++;
    char user[100] = {};
    sprintf(user,"%s悄悄的進(jìn)來(lái)了!(*^__^*) 嘻嘻……\n",name);
    send_msg(user);
    while(1)
    {
        if(recv(fd2,name,sizeof(name),0) == 0) /*返回0表示有客戶端退出*/
        {
            u[size-1].fd = 0; //把退出的客戶端的結(jié)構(gòu)描述符置換成0
        }
    }
}

void start(void)
{
    printf("success to start server,let's go!\n");
    while(1)
    {
        struct sockaddr_in client;//存儲(chǔ)接收到的客戶端的信息
        socklen_t length = sizeof(client);
        //接收客戶端的信息,會(huì)有阻塞效果,sockfd1標(biāo)記客戶端
        int sockfd1 = accept(sockfd,(struct sockaddr*)&client,&length);
        if(sockfd1 == -1)
        {
            perror("fail to accept!");
            continue;  //繼續(xù)連接
        }
        printf("%s連接上來(lái)了\n",inet_ntoa(client.sin_addr));
        /*連接成功之后,就啟動(dòng)線程*/
        pthread_t pid;//線程id
        pthread_create(&pid,0,pthread_deal,&sockfd1);
    }
}

void sig_exit(int signo ) //子定義信號(hào)關(guān)閉服務(wù)器函數(shù)
{
    close(sockfd);//關(guān)閉服務(wù)器端口
    printf("服務(wù)器成功關(guān)閉!\n");
    exit(0);
}
int main(void)
{
    printf("按ctrl+c關(guān)閉聊天室服務(wù)器!\n");
    signal(SIGINT,sig_exit);
    init();//初始化,服務(wù)器通信準(zhǔn)備工作
    start();//啟動(dòng)服務(wù)(開(kāi)始處理聊天信息)
    return 0;
}