网络编程之多人聊天室服务端

#include
#include
#include
#include
#include
#include
#define MAX_CLNT 100
#define BUF_SIZE 1024

pthread_mutex_t mutx;          //定义互斥锁
int clnt_count = 0;            //统计客户端数量
int clnt_socks[MAX_CLNT];        //存储客户端对应的套接字

void send_msg(char* msg,int len);            
void* handle_clnt(void* arg);

int main(int argc,char **argv)
{
    printf("Hello World!\n");
    //保证输入格式正确
    if(argc < 3)
    {
        printf("Usage: %s \n",argv[0]);
        return 0;
    }
    //定义套接字
    int serv_sock,clnt_sock;
    int clnt_addr_size;
    pthread_t t_id;
    //初始化互斥锁
    pthread_mutex_init(&mutx,NULL);
    //创建字节流套接字
    serv_sock = socket(AF_INET,SOCK_STREAM,0);
    
    //准备通信地址
    struct sockaddr_in sAddr,cAddr;
    memset(&sAddr,0,sizeof(sAddr));
    sAddr.sin_family = AF_INET;
    sAddr.sin_port = htons(atoi(argv[2]));
    sAddr.sin_addr.s_addr = inet_addr(argv[1]);
    
    //绑定通信地址
    bind(serv_sock,(struct sockaddr*)&sAddr,sizeof(sAddr));
    //监听
    listen(serv_sock,5);

    while(1)
    {
        //等待连接
        clnt_addr_size = sizeof(cAddr);
        clnt_sock = accept(serv_sock,(struct sockaddr*)&cAddr,&clnt_addr_size);

        //对共享资源进行加锁
        pthread_mutex_lock(&mutx);
        //用一个数组专门存放套接字【有多人连接进来，连接一个存放一个】
        clnt_socks[clnt_count++] = clnt_sock;    
       //对共享资源进行解锁
       pthread_mutex_unlock(&mutx);

        //创建线程，用来为客户端服务
        pthread_create(&t_id,NULL,handle_clnt,(void*)&clnt_sock);
        pthread_detach(t_id);
        printf("Connected client IP: %s \n",inet_ntoa(cAddr.sin_addr));
    }

    close(serv_sock);
    return 0;
}

void* handle_clnt(void* arg)
{    
    //先将套接字强转回来
    int clnt_sock = *((int*)arg);
    int str_len = 0;
    int i;
    char msg[BUF_SIZE];

    //从套接字读数据到缓存区
    while((str_len = read(clnt_sock,msg,sizeof(msg))) != 0)
    {    //读
        send_msg(msg,str_len);
    }//读到0个字节，表示客户下线，退出循环

    //客户下线，移除其socket
    //上锁，封闭式的移除
    pthread_mutex_lock(&mutx);
    for(i = 0; i < clnt_count; i++)    //遍历所有客户
    {
        if(clnt_sock == clnt_socks[i])        //若某客户下线
        {
            while(i < clnt_count-1)        //若第i个后面还有人？？？
            {
                clnt_socks[i++] = clnt_socks[i];    //后面的依次往前移动一个位置
            }
            break;
        }
    }
    clnt_count--;
    pthread_mutex_unlock(&mutx);        //解锁
    close(clnt_sock);                //关闭该套接字
    printf("%d off line~~\n",clnt_sock);
    return NULL;
}

void send_msg(char* msg,int len)//发信息给所有人
{
    int i;
    //上锁，关闭共享资源
    pthread_mutex_lock(&mutx);
    for(i = 0; i < clnt_count; i++)
    {    //写自己读到的数据，发送给每个客户的
        write(clnt_socks[i],msg,len);
    }
    //解锁
    pthread_mutex_unlock(&mutx);
}

/*******************************************
服务器：
准备工作：
    用一个数组统计连接进来的客户端的数量和存放他们的套接字
    即：每连接进来一个客户端，数组的成员加一，对应一个套接字
    
    
1.循环等待连接，每有一个客户端连接进来，上锁，让客户端
            数量加一，多保存一个套接字，解锁
2.创建一个线程，让他去为这个客户端服务
3.线程函数中，从这个客户端读取数据
4.若读到数据，就调用数据处理函数
5.数据处理函数中，上锁，用for循环封闭式给每个客户端发送
            自己读到的信息，发完解锁        
6.若读不到信息，上锁
a.在数组中遍历所有客户端        
b.若遇到下线的客户端    clnt_sock == clnt_socks[i]    【即若套接字对的上】
c.若他后面还有人，则把他的下一个人调到他的位置，所有人都循环前移一位