linux c++select多人聊天程序

比较简单的多人聊天程序，可直接运行。

主要是实现功能。没有界面也没有多余功能，只是实现群聊天的功能。c/s模式。

server端用select多路复用来做，可以接受多个客户端连接。client端启动2个线程控制发送和接受数据。

==================================================================================================

server.h

#include <unistd.h>

#include <stdio.h>

#include <netinet/in.h>

#include <sys/socket.h>

#include <sys/types.h>

#include <stdlib.h>

#include <arpa/inet.h>

int init(struct sockaddr_in addr)

{

int sock;

int ret;// 记录错误马

sock=socket(AF_INET,SOCK_STREAM,0);

if(sock==-1)return -1;

ret=bind(sock,(struct sockaddr*)&addr,sizeof(addr));

if(ret==-1)return -1;

listen(sock,10);

return sock;

}

int Link(int sock)

{

int ret;//返回值

int Maxfd;//最大描述符号

int acpsock;//accpet函数返回值

struct sockaddr_in acpaddr;//accpet用到的地址

socklen_t len;//地址大小

char buf[20];//接受到的消息内容

int resock;//recv中用到的文件么描述符号、

int client[10];//存储连接的文件描述符

int i,j;//for循环变量

fd_set refd,allfd;

for(i=0;i<10;i++)

client[i]=-1;

FD_ZERO(&refd);

FD_ZERO(&allfd);

Maxfd=sock;

FD_SET(sock,&allfd);

while(1)

{

refd=allfd;

ret=select(Maxfd+1,&refd,0,0,0);

if(ret==-1)return -1;

if(FD_ISSET(sock,&refd))

{

printf("someone connect\n");

len=sizeof(acpaddr);

acpsock=accept(sock,(struct sockaddr*) &acpaddr,&len);

j=0;

while(j<10 && client[j]!=-1)++j;

client[j]=acpsock;

Maxfd=Maxfd>acpsock?Maxfd:acpsock;

FD_SET(acpsock,&allfd);

}

for(i=0;i<10;i++)

{

if((resock=client[i])<0)continue;

if(FD_ISSET(resock,&refd))

{

if((j=recv(resock,buf,20,0))==0)

{

printf("someone exit\n");

close(resock);

FD_CLR(resock,&allfd);

client[i]=-1;

}

else

{

printf("%s",buf);

for(i=0;i<10;i++)

{

if(client[i]!=resock)

{

//printf("%d ,",client[i]);

send(client[i],buf,j,0);

}

}

}

}

}

}

return 0;

}

=====================================================================

server.cpp

#include "server.h"

#define SERVER_PORT 8888

int main()

{

int Sock;

struct sockaddr_in servaddr;

servaddr.sin_family = AF_INET;

servaddr.sin_addr.s_addr = inet_addr("127.0.0.1");//htonl(INADDR_ANY);

servaddr.sin_port = htons(SERVER_PORT);

Sock=init(servaddr);

if(-1==Sock)

{

printf("init error\n");

return -1;

}

printf("初始化成功\n");

Link(Sock);

return 0;

}

============================

client端启动2个线程用于读和写

client.h

#ifndef _CLENT_H_

#define _CLENT_H_

#include <unistd.h>

#include <stdlib.h>

#include <sys/socket.h>

#include <stdio.h>

#include <string.h>

#include <sys/types.h>

#include <netinet/in.h>

#include <arpa/inet.h>

#include <iostream>

using namespace std;

int init( struct sockaddr_in addr)

{

int sock;

int ret;

sock=socket(AF_INET,SOCK_STREAM,0);

ret=connect(sock,(struct sockaddr*)&addr,sizeof(addr));

if(ret==-1)printf("connect erro\n");

return sock;

}

int Chat(int sock)

{

//启动两个线程进行读写

fd_set refd;

int Maxfd;

char talk[20];

int ret;

Maxfd=sock;

while(1)

{

FD_ZERO(&refd);

FD_SET(0,&refd);

FD_SET(sock,&refd);

ret=select(Maxfd+1,&refd,0,0,0);

if (ret==-1)break;

if(FD_ISSET(0,&refd))

{

//scanf("%s",talk);

memset(talk,0,20);

fgets(talk,20,stdin);

send(sock,talk,strlen(talk)+1,0);

}

else if(FD_ISSET(sock,&refd))

{

memset(talk,0,20);

recv(sock,talk,20,0);

printf("%s",talk);

}

}

}

void * ForRead(void *arg)

{

char talk[20];

int* sock;

sock=(int*)arg;

while(1)

{

memset(talk,0,20);

fgets(talk,20,stdin);

send(*sock,talk,strlen(talk)+1,0);

}

}

void * ForWrite(void *arg)

{

char talk[20];

int *sock=(int *)arg;

while(1)

{

memset(talk,0,20);

recv(*sock,talk,20,0);

cout<<talk;

}

}

#endif

====================================================================================================

client.cpp

#include "client.h"

int main()

{

pthread_t rid,wid;

int sock,ret;

struct sockaddr_in client_addr;

client_addr.sin_family = AF_INET;

client_addr.sin_port = htons(8888);

//client_addr1.sin_addr.s_addr = inet_addr("127.0.0,1");

inet_pton(AF_INET, "127.0.0.1", &client_addr.sin_addr);

sock=init(client_addr);

// Chat(sock);

pthread_create(&rid,0,ForRead,&sock);

pthread_create(&wid,0,ForWrite,&sock);

pthread_join(rid,0);

pthread_join(wid,0);

return 0;

}