#include <stdio.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <pthread.h>
#include <string.h>
#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 <IP> <Port> \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.若他後面還有人,則把他的下一個人調到他的位置,所有人都循環前移一位