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