Linux聊天室

pub.h

#ifndef PUB_H
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <fcntl.h>
#include <errno.h>
#include <string.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <sys/epoll.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <dirent.h>

#define BUFSIZE 262144 //256k

#define MAX_EVENT 200

#endif

pub.c

#include "pub.h"

//聊天室最大人数限制
#define MAXCHAT_P 100

//存放socket描述符
int sts[MAXCHAT_P]={0};//目前最多允许MAXCHAT_P个人加入在线聊天

//聊天室当前人数
int chatsum=0;


/*
*功能：添加一个聊天的人
*参数: st socket描述符
*/
int addchatpeople(int st)
{
	int i;
	for(i=0;i<MAXCHAT_P;i++)
	{
		
		if(sts[i]==0)
		{
			sts[i]=st;
			chatsum++;
			return 1;
		}
	}
	return 0;//满了
	
}

/*
*功能：一个聊天的退出了
*参数: st socket描述符
*/
void deletechatpeople(int st)
{
	int i;
	for(i=0;i<MAXCHAT_P;i++)
	{
		if(sts[i]==st)
		{
			sts[i]=0;	
			chatsum--;
		}
	}
}


/*
*功能：将聊天记录发给聊天室其他成员
*参数: buf 发送的内容
       currst socket描述符
*/
void sendtocharroom(char * buf,int currst)
{
	int i;
	for(i=0;i<MAXCHAT_P;i++)
	{
		if(sts[i]!=0&&sts[i]!=currst)
		{
			send(sts[i], buf, strlen(buf), 0);
		}
	}
	
}


/*
*功能：创建参数port指定端口号的server端socket
*参数: port 端口
*/
int socket_create(int port)
{
	//创建TCP Socket
	int st = socket(AF_INET, SOCK_STREAM, 0);
	
	int on = 1;
	//设置地址和端口重用
	if (setsockopt(st, SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on)) == -1)
	{
		printf("setsockopt failed %s\n", strerror(errno));
		return 0;
	}
	
	struct sockaddr_in addr;
	memset(&addr, 0, sizeof(addr));
	addr.sin_family = AF_INET;
	addr.sin_port = htons(p