#include <stdio.h>
#include <string.h>
#include <ctype.h>
#include <stdlib.h>
#include <unistd.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <pthread.h>
#include <sys/stat.h>
#include <time.h>
#define MAXLINE 200
#define MAXNAME 20
#define SERV_PORT 8000
#define MAXCON (100 + 1) // 最大连接数为100，最后一个用于临时连接
#define MAXFILE 10240
#define FINISHFLAG "|_|_|"
struct sockaddr_in servaddr, cliaddr[MAXCON];
socklen_t cliaddr_len;
int listenfd, connfd[MAXCON], n;
char buf[MAXCON][MAXLINE + 50], spemsg[MAXCON][MAXLINE + 50], filebuf[MAXCON][MAXFILE + 50];
// 发送给除特定用户外的用户 发送给特点用户 文件信息缓存
// 略加大数组大小，防止输入字符满导致没有结束符'\0'
// 一个都有对应的缓存，防止消息混叠
char str[INET_ADDRSTRLEN];
char names[MAXCON][MAXNAME];
int used[MAXCON], downloading[MAXCON];
// 记录某个ID是否有用户连接 是否正在下载文件
void *TRD(void *arg); // 每个用户对应一个线程连接
int Process(int ID); // 对接收的信息进行处理
void sendonemsg(int sockfd, char *msg); // 发送信息给单个用户
void sendmsgtoall(int ID); // 发送信息给所有用户
int main(){
// 服务器初始化
listenfd = socket(AF_INET, SOCK_STREAM, 0);
bzero(&servaddr, sizeof(servaddr));
servaddr.sin_family = AF_INET;
servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
servaddr.sin_port = htons(SERV_PORT);
bind(listenfd, (struct sockaddr *)&servaddr, sizeof(servaddr));
listen(listenfd, 20);
memset(names, 0, sizeof(names));
memset(used, 0, sizeof(used));
memset(downloading, 0, sizeof(downloading));
// 预先开启最大用户连接数个线程
pthread_t tids[MAXCON];
int index[MAXCON];
for(int i = 0; i < MAXCON-1; i++) index[i] = i;
for(int i = 0; i < MAXCON-1; i++){
int ret = pthread_create(&tids[i], NULL, TRD, &index[i]);
if(ret != 0){
printf("thread create failed");
return 0;
}
}
printf("Accepting connections ...\n");
while (1) {
// 读取下一个为占用的ID，作为下一个连接的用户ID
int nowID = 0;
for(nowID = 0; nowID < MAXCON-1; nowID++)
if(!used[nowID]) break;
cliaddr_len = sizeof(cliaddr[nowID]);
// 等待连接
connfd[nowID] = accept(listenfd, (struct sockaddr *)&cliaddr[nowID], &cliaddr_len);
if(nowID >= MAXCON-1){
// 再次进行判断是否连接数满，因为第一次判断后可能有用户退出
for(nowID = 0; nowID < MAXCON-1; nowID++)
if(!used[nowID]) break;
if(nowID >= MAXCON-1){
// 服务器连接数满
memset(spemsg[nowID], 0, sizeof(spemsg[nowID]));
strcpy(spemsg[nowID], "Error(1): 聊天室人已满。");
sendonemsg(connfd[nowID], spemsg[nowID]);
close(connfd[nowID]);
continue;
}else{
// 再次判断后未满，设置对应的ID
connfd[nowID] = connfd[MAXCON-1];
cliaddr[nowID].sin_family = cliaddr[MAXCON-1].sin_family;
cliaddr[nowID].sin_port = cliaddr[MAXCON-1].sin_port;
cliaddr[nowID].sin_addr.s_addr = cliaddr[MAXCON-1].sin_addr.s_addr;
}
}
used[nowID] = 1; // 该ID有人占用
}
return 0;
}
inline void sendonemsg(int sockfd, char *msg){
strcat(msg, "\n"); // 发送的消息追加一个回车
write(sockfd, msg, strlen(msg));
}
inline void sendmsgtoall(int ID){
// 对处理信息来源的用户，发送指定消息
if(strlen(spemsg[ID]) != 0) sendonemsg(connfd[ID], spemsg[ID]);
for(int i = 0; i < MAXCON-1; i++){
if(i == ID) continue;
// 发送给除信息来源的用户（对正在下载文件的用户不发送，否则其文件将出错）
else if(used[i] && (!downloading[i])) sendonemsg(connfd[i], buf[ID]);
}
}
inline int Process(int ID){
char op[20];
int p = 0, fun = 0;
if(buf[ID][0] == ':'){
// 分离命令和参数 命令保存在op 参数保存在buf[ID]
p = 0;
memset(op, 0, sizeof(op));
while(buf[ID][0] != ' ' && buf[ID][0] != 0){
op[p] = buf[ID][0]; p++;
char tmp[MAXLINE + 50];//避免未定义行为
strcpy(tmp, buf[ID]+1);
strcpy(buf[ID], tmp);
}
op[p] = 0;
while(buf[ID][0] == ' ' && buf[ID][0] != 0){
char tmp[MAXLINE + 50];
strcpy(tmp, buf[ID]+1);
strcpy(buf[ID], tmp);
}
if(op[1] == 'n'){
// 新用户登陆，判断是否重名
// 重名返回错误信息
for(int i = 0; i < MAXCON-1; i++)
if(used[i]){
if(strcmp(names[i], buf[ID]) == 0){
memset(spemsg[ID], 0, sizeof(spemsg[ID]));
strcpy(spemsg[ID], "Error(2): 姓名重复，请重试。");
sendonemsg(connfd[ID], spemsg[ID]);
return 0;
}
}
// 不重名则发送提示信息
strcpy(names[ID], buf[ID]);
sprintf(buf[ID], "%s(%s:%d)进入了聊天室", names[ID],
inet_ntop(AF_INET, &cliaddr[ID].sin_addr, str, sizeof(str)),
ntohs(cliaddr[ID].sin_port));
memset(spemsg[ID], 0, sizeof(spemsg[ID]));
strcpy(spemsg[ID], "成功进入聊天室");
sendmsgtoall(ID);
return 0;
}
if(op[1] == 'r'){
// 改名，判断是否重名
// 重名返回错误信息
char newname[MAXNAME];
for(int i = 0; i < MAXCON-1; i++){
if(i == ID) continue;
if(used[i]){
if(strcmp(names[i], buf[ID]) == 0){
memset(spemsg[ID], 0, sizeof(spemsg[ID]));
strcpy(spemsg[ID], "Error(3): 姓名重复，请重试。");
sendonemsg(connfd[ID], spemsg[ID]);
return 0;
}
}
}
// 不重名则发送提示信息
strcpy(newname, buf[ID]);
memset(buf[ID], 0, sizeof(buf[ID]));
sprintf(buf[ID], "%s(%s:%d)改名为%s", names[ID],
inet_ntop(AF_INET, &cliaddr[ID].sin_addr, str, sizeof(str)),
ntohs(cliaddr[ID].sin_port), newname);
memset(names[ID], 0, sizeof(names[ID]));
strcpy(names[ID], newname);
memset(spemsg[ID], 0, sizeof(spemsg[ID]));
strcpy(spemsg[ID], "改名成功");
sendmsgtoall(ID);
return 0;
}
if(op[1] == 'q'){
// 用户退出时，给其他所用用户发送提示信息
memset(buf[ID], 0, sizeof(buf[ID]));
sprintf(buf[ID], "%s(%s:%d)离开了聊天室", names[ID],
inet_ntop(AF_INET, &cliaddr[ID].sin_addr, str, sizeof(str)),
ntohs(cliaddr[ID].sin_port));
memset(spemsg[ID], 0, sizeof(spemsg[ID]));
sendmsgtoall(ID);
memset(names[ID], 0, sizeof(names[ID]));
return 1; // 给线程函数返回1，用作后续处理
}
if(op[1] == 's'){
// 给请求的用户发送所有用户信息
memset(spemsg[ID], 0, sizeof(spemsg[ID]));
strcpy(spemsg[ID], "IP Port name");
sendonemsg(connfd[ID], spemsg[ID]);
for(int i = 0; i < MAXCON-1; i++)
if(used[i]){
memset(spemsg[ID], 0, sizeof(spemsg[ID]));
sprintf(spemsg[ID], "%-16s%-7d%s",
inet_ntop(AF_INET, &cliaddr[i].sin_addr, str, sizeof(str)),
ntohs(cliaddr[i].sin_port), names[i]);
sendonemsg(connfd[ID], spemsg[ID]);
}
memset(spemsg[ID], 0, sizeof(spemsg[ID]));
return 0;
}
if(op[1] == 'f'){
// 给请求的用户发送服务器端文件
system("mkdir Files");
system("ls Files > ./files.txt");
FILE *filename = fopen("./files.txt", "r");
memset(spemsg[ID], 0, sizeof(spemsg[ID]));
while(fgets(spemsg[ID], MAXLINE, filename) != NULL){
spemsg[ID][strlen(spemsg[ID])-1] = 0;
sendonemsg(connfd[ID], spemsg[ID]);
}
system("rm -f files.txt");
return 0;
}
if(op[1] == 'u'){
// 服务器端接收文件
char filename[MAXLINE];
char filepath[MAXLINE + 60];
char Command[MAXLINE + 70];
memset(filename, 0, sizeof(filename));
memset(filepath, 0, sizeof(filepath));
memset(Command, 0, sizeof(Command));
strcpy(filename, buf[ID]);
system("mkdir Files");
sprintf(filepath, "./Files/%s", buf[ID]);
sprintf(Command, "rm -f %s", filepath);
// 判断是否已存在同名文件，存在则返回错误信息
FILE *ff = fopen(filepath, "rb");
if(ff != NULL){
memset(spemsg[ID], 0, sizeof(spemsg[ID]));
strcpy(spemsg[ID], "Error(5): 服务器中存在相同名称文件。");
sendonemsg(connfd[ID], sp
