從0實現(xiàn)基于Linux socket聊天室-增加公聊、私聊-4

上文中，我們基于多線程的框架，實現(xiàn)了注冊和登錄的功能，這一章，我們在此基礎上來實現(xiàn)公聊、私聊、顯示在線用戶列表功能。

公聊

接著上幾篇的流程圖我們詳細講解公聊流程圖如下：

如上圖所示，我們?nèi)サ袅司W(wǎng)絡連接和客戶端登錄、注冊等功能，直接進入聊天的流程：

1. 客戶端從菜單選擇公聊功能；
2. 輸入要聊天信息；
3. 回車發(fā)送聊天信息；
4. 服務器的子線程收到公聊數(shù)據(jù)之后，進入公聊流程；
5. 查找所有在線用戶，向所有的在線用戶發(fā)送該公聊信息；
6. 客戶端進入聊天后會創(chuàng)建一個子線程，該子線程會循環(huán)接收所有服務器發(fā)送的數(shù)據(jù)信息。

私聊

如上圖所示：

1. 客戶端從菜單選擇私聊功能；
2. 輸入要聊天的對象和聊天信息；
3. 發(fā)送聊天信息給服務器；
4. 服務器的子線程收到公聊數(shù)據(jù)之后，進入公聊流程；
5. 查找所有在線用戶，向所有的在線用戶發(fā)送該公聊信息；
6. 客戶端子線程會循環(huán)接收所有服務器發(fā)送的數(shù)據(jù)信息。

顯示在線用戶

如上圖所示：

1. 客戶端從菜單選擇顯示在線用戶功能；
2. 封裝顯示在線用戶數(shù)據(jù)包，并發(fā)送該數(shù)據(jù)包給服務器；
3. 服務器收到數(shù)據(jù)包后，進入顯示在線用戶功能模塊；
4. 檢查在線用戶數(shù)據(jù)信息數(shù)據(jù)庫，將在線用戶【fd不為-1】名稱封裝到數(shù)據(jù)包中，一次只填充一個，stat字段填充ONLINEUSER_OK；
5. 所有用戶發(fā)送完畢，補充一個數(shù)據(jù)包，stat填充ONLINEUSER_OVER；
6. 收到服務器發(fā)送的在線用戶數(shù)據(jù)包后，客戶端子線程進入顯示在線用戶子模塊；提取數(shù)據(jù)包中在線用戶名字并打印，判斷該數(shù)據(jù)包stat是否為 ONLINEUSER_OVER，如果不是則繼續(xù)接收下一個數(shù)據(jù)包，如果是，則提示用戶顯示完畢。

運行截圖

現(xiàn)在預設：客戶A：yikoulinux客戶B：yikoupeng

公聊

客戶B發(fā)公聊：

客戶A收到信息：

服務器log：

私聊

客戶B私發(fā)信息給A：

客戶A收到的消息：

顯示在線用戶信息

chat.h

#ifndef _TCP_CHAT
#define _TCP_CHAT

#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <stdio.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <string.h>
#include <pthread.h>
#include <stdlib.h>
#include <string.h>

#define SERVER_PORT 8888
//在線用戶
struct ONLINE{ int fd; //-1
int flage; //registed or not
char name[32];
char passwd[32];
};
#define MAX_USER_NUM 64

//C/S通信的結(jié)構(gòu)體
struct protocol{
int cmd;
int state
; char name[32]
; char data[64];
};
/*cmd*/
#define BROADCAST 0X00000001
#define PRIVATE 0X00000002#define REGISTE 0X00000004
#define LOGIN 0X00000008
#define ONLINEUSER 0X00000010
#define LOGOUT 0X00000020

/*return code*/
#define OP_OK 0X80000000
#define ONLINEUSER_OK 0X80000001
#define ONLINEUSER_OVER 0X80000002
#define NAME_EXIST 0X80000003
#define NAME_PWD_NMATCH 0X80000004
#define USER_LOGED 0X80000005
#define USER_NOT_REGIST 0X80000006

#endif

client.c

#include "chat.h"

int sockfd;
int addrlen;
struct sockaddr_in server_addr;

pthread_t pid;

int login_f = -1;

void *func(void *arg)
{
int len;
char buf[128]={0};
struct protocol *msg;

while(1)
{
if(login_f != 1)
{
continue;
}
memset(buf,0,sizeof(buf));
len = read(sockfd,buf,sizeof(buf));
if(len<=0)
{
close(sockfd);
return;
}
msg = (struct protocol *)buf;
// this is show online user,
if((msg->state == ONLINEUSER_OK)&&(msg->cmd == ONLINEUSER))
{
printf("%st",msg->name);
continue;
}
if((msg->state == ONLINEUSER_OVER)&&(msg->cmd == ONLINEUSER))
{
printf("n");
continue;
}

buf[len]='';

printf("%sn",buf);
}
}
void broadcast(int fd)
{
struct protocol msg;

msg.cmd = BROADCAST;
printf("say:n#");

scanf("%s",msg.data);

write(fd,&msg,sizeof(msg));
}
void private(int fd)
{
struct protocol msg;

msg.cmd = PRIVATE;

printf("input name you want to talk:n#");
scanf("%s",msg.name);
printf("say:n#");
scanf("%s",msg.data);

write(fd,&msg,sizeof(msg));
}
void list_online_user(sockfd)
{
struct protocol msg;

msg.cmd = ONLINEUSER;

write(sockfd,&msg,sizeof(msg));
getchar();
getchar();
/*
while(1)
{
read(sockfd ,&msg,sizeof(msg));
if(msg.state == ONLINEUSER_OK)
{
printf("%st",msg.name);
}else{
break;
}
}*/

}int registe(int fd)
{
struct protocol msg,msgback;

msg.cmd = REGISTE;
printf("input your namen");
scanf("%s",msg.name);
printf("input your passwdn");
scanf("%s",msg.data);

write(sockfd,&msg,sizeof(msg));
read(sockfd,&msgback,sizeof(msgback));
if(msgback.state != OP_OK)
{
printf("Name had exist,try again!n");
getchar();
getchar();
return -1;
}else{
printf("Regist success!n");
getchar();
getchar();
return 0 ;
}
}
int login(int fd)
{
struct protocol msg,msgback;

msg.cmd = LOGIN;
printf("input your namen");
scanf("%s",msg.name);
printf("input your passwdn");
scanf("%s",msg.data);

write(sockfd,&msg,sizeof(msg));
read(sockfd,&msgback,sizeof(msgback));
if(msgback.state != OP_OK)
{
printf("Name had exist,try again!n");
getchar();
getchar();
login_f = -1;
return NAME_PWD_NMATCH;
}else{
printf("Login success!n");
getchar();
getchar()
; login_f = 1;
return OP_OK ;
}
}
int logout(int fd)
{
close(fd);
login_f = -1
;}
int main(int argc, char **argv)
{
int sel;
int ret;
int min_sel,max_sel;
int portnumber;
struct protocol msg;

if(argc<3)
{
printf("cmd: %s ip portnumbern",argv[0]);
return;
}
//argv2 存放的是端口號 ，讀取該端口，轉(zhuǎn)換成整型變量
if((portnumber=atoi(argv[2]))<0)
{
fprintf(stderr,"Usage:%s hostname portnumberan",argv[0]);
exit(1);
}
sockfd = socket(PF_INET,SOCK_STREAM,0);
if(sockfd<0)
{
perror("socket() failn");
return;
}

server_addr.sin_family = PF_INET;
server_addr.sin_port = htons(portnumber);
server_addr.sin_addr.s_addr = inet_addr(argv[1]);

addrlen = sizeof(struct sockaddr_in);

connect(sockfd,(struct sockaddr* )&server_addr,addrlen);
pthread_create(&pid, NULL,func, NULL);
while(1)
{
//getchar();
system("clear");
if(login_f == -1){
printf("t 1 娉ㄥ唽 n");
printf("t 2 鐧誨綍 n");
}else if(login_f == 1){
printf("t 3 鍏亰 n");
printf("t 4 縐佽亰 n");
printf("t 5 鍦ㄧ嚎鍒楄〃 n");
}
printf("t 0 閫€鍑?n");

fflush(stdin);
scanf("%d",&sel);
if(sel == 0)
{
break;
}
if(login_f == 1)
{
min_sel = 3
; max_sel = 5;
}else if(login_f == -1)
{
min_sel = 1;
max_sel = 2;
}

if(sel<min_sel || sel > max_sel)
{
printf("Valid choice ,try againn")
; continue;
}
switch(sel)
{
case 1:
registe(sockfd);
break;
case 2:
ret = login(sockfd); break;
case 3:
broadcast(sockfd); break;
case 4:
private(sockfd);
break;
case 5:
list_online_user(sockfd)
; break;
case 0
: logout(sockfd);
break;
default:
break
; }
if(sel == 0)
{
exit(0);
}
}

}

server.c

#include "chat.h"

struct ONLINE online[MAX_USER_NUM];

void del_user_online(int index)
{
int i;
char buf[128]={0};

if(index <0)
{
return;
}
online[index].fd = -1;
sprintf(buf,"%s offlinen",online[index].name);
//通知所有客戶端，某個用戶下線了
for(i=0;i<MAX_USER_NUM;i++)
{
if(online[i].fd == -1)
{
continue;
}
write(online[i].fd,buf,strlen(buf));
}

return;
}
int add_user(int sockfd,struct protocol*msg)
{
int i,index = -1;
char buf[128]={0};

for(i=0;i<64;i++)//添加到在線用戶列表
{
if(online[i].flage == -1)
{
online[i].flage= 1;
strcpy(online[i].name,msg->name);
strcpy(online[i].passwd,msg->data);
printf("regist %s to %d n",msg->name,i);
index = i
; return index
; }
}
return index
;}
void broadcast(int index,struct protocol*msg)
{
int i;
char buf[128]={0};

sprintf(buf,"%s say:%sn",online[index].name ,msg->data);

for(i=0;i<MAX_USER_NUM;i++)
{// jump offline and sender self
if((online[i].fd == -1)||(i == index))
{
continue;
}
write(online[i].fd,buf,strlen(buf));
}
}
int find_dest_user_online(int sockfd,int *index,struct protocol*msg)
{
int i

; for(i=0;i<MAX_USER_NUM;i++)
{
//this pos not use
if(online[i].flage== -1)
{
continue;
}

if((strcmp(msg->name,online[i].name)==0)&&(strcmp(msg->data,online[i].passwd)==0))
{
if(online[i].fd == -1)
{
online[i].fd = sockfd;
*index = i ;
return OP_OK;
}else{
//user had loged
printf("%s had loginn",online[i].name);
return USER_LOGED;
}

}
}
return NAME_PWD_NMATCH;
}
int find_dest_user(char *name)
{
int i;

for(i=0;i<MAX_USER_NUM;i++)
{

if(online[i].flage == -1)
{
continue;
}

if(strcmp(name,online[i].name)==0)
{
return i;
}
}
return -1;
}

void private(int index,struct protocol*msg)
{
int dest_index;
char buf[128]
; //找到那個人
dest_index = find_dest_user(msg->name);

if(dest_index == -1)
{
sprintf(buf,"there is no user :%s n",msg->name);
write(online[index].fd,buf,strlen(buf));
return;
}else{
sprintf(buf,"%s say to %s:%sn",online[index].name ,online[dest_index].name,msg->data);
write(online[dest_index].fd,buf,strlen(buf));
return;
}

}
void list_online_user(int index)
{
int i;
struct protocol msg;

for(i=0;i<MAX_USER_NUM;i++)
{
if(online[i].fd == -1)
{
continue;
}
memset(&msg,0,sizeof(msg));

msg.cmd = ONLINEUSER;
msg.state = ONLINEUSER_OK;
strcpy(msg.name,online[i].name);

printf("list online[i].name =%s n",online[i].name);

write(online[index].fd ,&msg,sizeof(msg));
}
msg.cmd = ONLINEUSER;
msg.state = ONLINEUSER_OVER;

write(online[index].fd ,&msg,sizeof(msg));
}

void registe(int sockfd,int *index,struct protocol*msg)
{
int dest_index;
char buf[128];
struct protocol msg_back;

msg_back.cmd = REGISTE;
//找到那個人
dest_index = find_dest_user(msg->name);

if(dest_index == -1)
{ // this user can registe
*index = add_user(sockfd,msg);

online[*index].flage = 1;
msg_back.state = OP_OK;

printf("user %s regist success!n",msg->name);
write(sockfd,&msg_back,sizeof(msg_back));

return;
}else{
msg_back.state = NAME_EXIST;
printf("user %s exist!n",msg->name);

write(sockfd,&msg_back,sizeof(msg_back));
return;
}
}
void login(int sockfd,int *index,struct protocol*msg)
{
int i; int ret;
char buf[128];
struct protocol msg_back;

msg_back.cmd = LOGIN;

//找到那個人
ret = find_dest_user_online(sockfd,index,msg);

if(ret != OP_OK)
{
msg_back.state = ret;
strcpy(buf,"there is no this usern");
printf("user %s login fail!n",msg->name);

write(sockfd,&msg_back,sizeof(msg_back));
return;
}else{
msg_back.state = OP_OK;
strcpy(msg_back.data,"login successn");
printf("user %s login success!index =%d n",msg->name,*index);
write(online[*index].fd,&msg_back,sizeof(msg_back));
}
//通知所有客戶端，某個用戶上線了
sprintf(buf,"%s onlinen",online[*index].name);

for(i=0;i<MAX_USER_NUM;i++)
{
if(online[i].fd != -1)
{
write(online[i].fd,buf,strlen(buf));
}
}

}
void *func(void *arg)
{
int sockfd = *((int*)arg);
char buf[64];
int len; int index = -1;//該用戶在在線用戶列表的下標
struct protocol msg;

free(arg)
; //客戶端連接后 必須先輸入一個名字,不考慮數(shù)據(jù)類型出錯，以及重名問題,也不考慮人數(shù)滿了的情況

// read(sockfd,&msg,sizeof(msg));

// index = add_user_online(sockfd,&msg);

//進入聊天了
while(1)
{
len = read(sockfd,&msg,sizeof(msg));
if(len<=0)
{//下線
printf("%s offlinen",online[index].name);
//從在線列表中刪除
del_user_online(index);
close(sockfd);
return; }

switch(msg.cmd)
{
case REGISTE:
registe(sockfd,&index,&msg);
break;
case LOGIN:
login(sockfd,&index,&msg);
break;
case BROADCAST:
broadcast(index,&msg);
break;
case PRIVATE:
private(index,&msg);
break;
case ONLINEUSER:
list_online_user(index);
break;
default:
break;
}

}
}

int main(int argc, char **argv)
{
int lsfd,newfd; int addrLen,cliaddrlen;
struct sockaddr_in my_addr;
struct sockaddr_in cli_adr;
char buf[64]="xuezhiqian fuhelen";
pthread_t pid;
int *arg;
int i;
int portnumber;

if(argc<2)
{
printf("cmd: %s portnumbern",argv[0]);
return;
}
/*?????ì??*/
if((portnumber=atoi(argv[1]))<0)
{
fprintf(stderr,"Usage:%s portnumberan",argv[0]);
exit(1);
}
lsfd = socket(PF_INET,SOCK_STREAM,0);
if(lsfd<0)
{
perror("socket() failn");
return;
}
bzero(&my_addr,sizeof(struct sockaddr_in));
my_addr.sin_family = PF_INET;
my_addr.sin_port = htons(portnumber);
my_addr.sin_addr.s_addr = htonl(INADDR_ANY);
addrLen = sizeof(struct sockaddr_in);

if(bind(lsfd,(struct sockaddr* )&my_addr ,addrLen)<0)
{
perror("bind() failn");
return;
}

listen(lsfd,5);
cliaddrlen = sizeof(struct sockaddr_in);

for(i=0;i<64;i++)
{
online[i].fd = -1;
online[i].flage= -1;
}
while(1)
{
newfd = accept(lsfd,(struct sockaddr *)&cli_adr,&cliaddrlen);
printf("client:ip:%s port:%d n",
inet_ntoa(cli_adr.sin_addr),cli_adr.sin_port);

arg = malloc(sizeof(int));
*arg = newfd;//必須搞清楚為什么要申請內(nèi)存
pthread_create(&pid,NULL,func, (void*)arg);
}
close(newfd);
close(lsfd);
}