基于Linux的聊天室(服务器)初成版

  实现注册、登入、下线、私聊、群聊、传文件、在线用户查询,以及管理员(root)的禁言,解禁,踢人等功能,与多客户端进行连接。

/*****************************************************
copyright (C), Nanjing University of ZiJin
File name:server.c
Author: CaoMengDe  Version:0.1    Date: 2017-01-01 14:38
Description:
Funcion List: 
*****************************************************/
#include <sqlite3.h>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <pthread.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>


#define PORTNUMBER 9997


typedef struct node
{
int fd;
char name[1024];
int flagg;
struct node *next;
}Node;


Node* head;
char yourname[1024] = {0};
int init(Node** head);//初始化链表


//void menu();
void registration(int fd);//数据库用来存放用户帐号信息
int signin(int newfd);//登入判断
void* func(void* arg);//线程
int up(int fd);//上线
void down(int fd);//下线
int groupchat(int fd);//群聊
void print(int fd);//在线用户查询
int privatechat(int fd);//私聊
int Prohibition(int fd);//禁言
int Liftban(int fd);//解禁
int Kick(int fd);//踢人
void printlist(int fd);//管理员查询在线人数
int catban(int fd);//查询被禁言用户
int file(int fd);//文件传输
 
int main(int argc, char **argv)
{
int ret;
ret =init(&head);
if(-1 == ret)
{
return -1;
}
   //AF_INET:IPv4
//SOCK_STREAM:流式SOCKET,代表TCP
int socket_fd = socket(AF_INET, SOCK_STREAM, 0);
if (-1 == socket_fd)
{
perror("socket");
exit(1);
}

struct sockaddr_in server_addr;


    //清空addr
memset(&server_addr, 0, sizeof(server_addr));
server_addr.sin_family = AF_INET;
server_addr.sin_port = htons(PORTNUMBER);
//将本机器上所有long数转化为网络字节许。
//UNADDR_ANY 可以接收任意IP地址
server_addr.sin_addr.s_addr = htonl(INADDR_ANY);

if (-1 == bind(socket_fd, (struct sockaddr *)(&server_addr), sizeof(struct sockaddr)))
{
perror("bind");
exit(2);
}

//5 :最大连接数
if(-1 == listen(socket_fd, 5))
{
perror("listen");
exit(3);
}

while (1)
{
struct sockaddr_in client_addr;
   int length = sizeof(struct sockaddr);
   int newfd = accept(socket_fd, (struct sockaddr *)(&client_addr), &length);
if (-1 == newfd)
   {
   perror("accept");
   exit(4);
   }
pthread_t pid;
        pthread_create(&pid, NULL, (void*)func, (void*)newfd);

}

}






void* func(void* arg)
//线程
{
int newfd = (int)arg;
while(1)
{

char order[1024] = {0};
//sleep(1);
int ret = read(newfd, order, 1024);
if(ret == -1)
{
//perror("order");
return NULL;
}
if(strcmp(order, "*registration") == 0)
{
registration(newfd);
}
if(strcmp(order, "*signin") == 0)
{
signin(newfd);
}
if(strcmp(order, "*up") == 0)
{
up(newfd);
}
if(strcmp(order, "*privatechat") == 0)
{
privatechat(newfd);
}
if(strcmp(order, "*groupchat") == 0)
{
groupchat(newfd);
}
if(strcmp(order, "*print") == 0)
{
print(newfd);
}
if(strcmp(order, "*down") == 0)
{
down(newfd);
}
if(strcmp(order, "*file") == 0)
{
//printf("1----------------------------\n");
file(newfd);
}
if(strcmp(order, "*superuser") == 0)
{
superuser(newfd);
}
if(strcmp(order, "*Prohibition") == 0)
{
Prohibition(newfd);
}
if(strcmp(order, "*Liftban") == 0)
{
Liftban(newfd);
}
if(strcmp(order, "*Kick") == 0)
{
Kick(newfd);
}
/*if(strcmp(order, "*printlist") == 0)
{
printlist(newfd);
}*/
if(strcmp(order, "*catban") == 0)
{
catban(newfd);
}
if(strcmp(order, "*exit") == 0)
{
break;
}
memset(order, 0 ,1024);

}

    printf("conect end\n");
close(newfd);
}


int init(Node** head)
//链表的初始化
{
Node* new = (Node*)malloc(sizeof(Node));
    if(NULL == new)
    {
        return -1;
    }
    *head = new;
    (*head)->next = NULL;
return 0;
}




/*void menu()
{
int num;
printf("欢迎来到Coder的聊天室\n");
}*/




void registration(int fd)
//数据库用来存放用户帐号信息
{
int ret;
sqlite3 *db;
char* errmsg;
ret = sqlite3_open("server.db", &db);
if(ret != SQLITE_OK)
    {
        perror("open");
        exit(1);
    }
char sql_create[] = "create table if not exists stu(username text primary key, pwd text);";
    ret = sqlite3_exec(db, sql_create, NULL, NULL, &errmsg);
    if(ret != SQLITE_OK)
    {
        perror("create");
        exit(2);
    }
char buf1[1024] = {0};
char buf2[1024] = {0};
    char flag[1024] = {0};
char buf3[1024] = {0};
ret = read(fd, buf1, 1024);//帐号
if(-1 == ret)
{
perror("zhanghao");
return;
}
ret = read(fd, buf2, 1024);//密码
if(-1 == ret)
{
perror("mima");
return;
}
ret = read(fd, buf3, 1024);//确认密码
if(-1 == ret)
{
perror("2mima");
return;
}
char sql_insert[1024] = {0};
if(strcmp(buf2, buf3) != 0)
{
flag[0] = 'A';
write(fd, flag, 1024);
return;
}
sprintf(sql_insert, "insert into stu values('%s', '%s');", buf1, buf2);
ret = sqlite3_exec(db, sql_insert, NULL, NULL, &errmsg);
if(ret != SQLITE_OK)//判断帐号是否已经存在
    {
        //perror("insert");
//printf("用户名已存在\n");
flag[0] = 'N';
write(fd, flag, 1024);
    }
else
{
//printf("执行成功!\n");
flag[0] = 'Y';
write(fd, flag, 1024);
}
ret = sqlite3_close(db);
if(ret != SQLITE_OK)
    {
        perror("close");
        exit(1);
    }
}






int signin(int newfd)
//登入判断
{
char flag[1024];
int ret;
sqlite3 *db;
char* errmsg;
int row;
int column;
char** result;
ret = sqlite3_open("server.db", &db);
if(ret != SQLITE_OK)
    {
        perror("open");
        exit(1);
    }
char buf1[1024] = {0};
char buf2[1024] = {0};
ret = read(newfd, buf1, 1024);
if(-1 == ret)
{
perror("dengruzhanghao");
return;
}
ret = read(newfd, buf2, 1024);
if(-1 == ret)
{
perror("dengrumima");
return;
}
char sql_query[1024] = {0};
sprintf(sql_query, "select * from stu where username = '%s';", buf1);
ret = sqlite3_get_table(db, sql_query, &result, &row, &column, &errmsg);
if(ret != SQLITE_OK)
{
perror("query1");
return 0;
}
if(row > 0)
{
if(strcmp(buf2, result[1*column + 1]) == 0)
{
if(strcmp("root", buf1) == 0)
{
flag[0] = 'S';
write(newfd, flag, 1024);

}
else
{
flag[0] = 'Y';
write(newfd, flag, 1024);
strcpy(yourname, buf1);
}

}
else
{
flag[0] = 'N';
write(newfd, flag, 1024);
}
}
else
{
flag[0] = 'Z';
write(newfd, flag, 1024);
}

}




int up(int fd)
//上线
{
int ret;
char str[1024] = {0};
char name[1024] = {0};
/*ret = read(fd, str, 1024);
if(-1 == ret)
{
perror("shangxian");
return;
}*/
Node* temp = head;
Node* new = (Node*)malloc(sizeof(Node));
while(temp->next != NULL)
{
temp = temp->next;
}
temp->next = new;
new->next = NULL;
new->fd = fd;
new->flagg = 1;
strcpy(new->name, yourname);
memset(yourname, 0, 1024);
}






void down(int fd)
//下线
{
char t[1024] = "down_success";
Node* temp = head;
while(temp->next->fd != fd)
{
temp = temp->next;
}
Node* temp1 = temp->next;
temp->next = temp->next->next;
free(temp1);
temp1 = NULL;
write(fd, t, 1024);
}






int privatechat(int fd)
//私聊
{
int ret;
Node* temp1 = head;
while(temp1->next->fd != fd)
{
temp1 = temp1->next;
}
if(temp1->next->flagg == 1)
{
char str[1024] = "privatechat_ok";
write(fd, str, 1024);
char t[1024] = {0};
Node* temp = head;
ret = read(fd, t, 1024);
if(-1 == ret)
{
perror("privatechat");
return;
}
while(strcmp(temp->next->name, t)!= 0)
{
temp = temp->next;
if(temp->next == NULL)
{
char m[1024] = "no_find";
write(fd, m, 1024);
return 0;
}
}
int newfd2 = temp->next->fd;
while(1)
{
memset(t, 0, 1024);
ret = read(fd, t, 1024);
if(-1 == ret)
{
perror("2privatechat");
return;
}
if(strcmp(t, "exit") == 0)
{
break;
}
char buffer[1024] = {0};
sprintf(buffer, "%s:%s", temp1->next->name, t);
write(newfd2, buffer, 1024);
}
}
if(temp1->next->flagg == 0)
{
char str[1024] = "privatechat_error";
write(fd, str, 1024);
}

}


 
 
int groupchat(int fd)
//群聊
{
int ret;
Node* temp1 = head;
while(temp1->next->fd != fd)
{
temp1 = temp1->next;
}
if(temp1->next->flagg == 1)
{
char str[1024] = "groupchat_success";
write(fd, str, 1024);
while(1)
{
Node* temp = head;
char buffer[1024] = {0};
ret = read(fd, buffer, 1024);
if(-1 == ret)
{
//perror("qunliao");
return;
}
if(strcmp(buffer, "exit") == 0)
{
break;
}
while(temp->next != NULL)
{
if(temp->next->fd != fd)
{
int newfd = temp->next->fd;
char tt[1024] = {0};
sprintf(tt, "%s:%s", temp1->next->name,buffer);
write(newfd, tt, 1024);
temp = temp->next;

}
else
{
temp = temp->next;
}
}
memset(buffer, 0, 1024);
}
}
if(temp1->next->flagg == 0)
{
char str[1024] = "groupchat_error";
write(fd, str, 1024);


}
}
 
 
 void print(int fd)
 
 //在线用户查询
 {
Node* temp = head;
char buffer[1024] = {0};
while(temp->next != NULL)
{
strcpy(buffer, temp->next->name);
write(fd, buffer, 1024);
   memset(buffer, 0 ,1024);
temp = temp->next;
}

 }
 
 
int superuser(int fd)
//管理员操作
{
Node* new = (Node*)malloc(sizeof(Node));
char str[1024] = "root";
if(NULL == new)
{
perror("superuser");
return -1;
}
Node* temp = head;
while(temp->next != NULL)
{
temp = temp->next;
}
temp->next = new;
new->next = NULL;
strcpy(new->name, str);
new->fd = fd;
new->flagg = 1;
}


int Prohibition(int fd)
//禁言
{
int ret;
char buffer[1024] = {0};
ret = read(fd, buffer, 1024);
if(-1 == ret)
{
perror("jingyan");
return;
}
Node* temp = head;
while(strcmp(temp->next->name, buffer) != 0)
{
temp = temp->next;
if(NULL == temp->next)
{
char t[1024] = "no_find";
write(fd, t, 1024);
return;
}
}
if(temp->next->flagg == 0)
{
char s[1024] = "again_operate";
write(fd, s, 1024);
return;
}
else
{
   temp->next->flagg = 0;
}

}




int Liftban(int fd)
//解禁
{
int ret;
char buffer[1024] = {0};
ret = read(fd, buffer, 1024);
if(-1 == ret)
{
perror("jiejin");
return;
}
Node* temp = head;
while(strcmp(temp->next->name, buffer) != 0)
{
temp = temp->next;
if(NULL == temp->next)
{
char t[1024] = "no_find";
write(fd, t, 1024);
return;
}
}
if(temp->next->flagg == 1)
{
char s[1024] = "again_operate";
write(fd, s, 1024);
return;
}
else
{
   temp->next->flagg = 1;
}
}




int Kick(int fd)
//踢人
{
int ret;
Node* temp = head;
char str[1024] = {0};
ret = read(fd, str, 1024);
if(-1 == ret)
{
perror("tiren");
return;
}
while(strcmp(temp->next->name, str) != 0)
{
temp = temp->next;
if(temp->next == NULL)
{
char m[1024] = "no_find";
write(fd, m, 1024);
return 0;
}
}
int socket_fd = temp->next->fd;
char t[1024] = "kick_success";
write(socket_fd, t, 1024);
Node* temp1 = temp->next;
temp->next = temp->next->next;
free(temp1);
temp1 = NULL;
close(socket_fd);
}


/*void printlist(int fd)
//在线用户查询
{
Node* temp = head;
char buffer[1024] = {0};
while(temp->next != NULL)
{
strcpy(buffer,temp->next->name);
write(fd, buffer, 1024);
memset(buffer, 0 ,1024);
temp = temp->next;
}


}*/


int catban(int fd)
//被禁言用户查询
{
char buffer[1024] = {0};
Node* temp = head;
while(temp->next != NULL)
{
if(temp->next->flagg == 0)
{
strcpy(buffer, temp->next->name);
write(fd, buffer, 1024);
memset(buffer, 0 ,1024);
}
   temp = temp->next;
}



}


int file(int fd)
{
int ret;
Node* temp = head;
char t[1024] = "no_find";
char buffer[1024] = {0};
char file[1024] = {0};
read(fd, buffer, 1024);
while(strcmp(buffer, temp->next->name) != 0)
{
temp = temp->next;
if(NULL == temp->next)
{
write(fd, t, 1024);
return 1;
}

}

memset(buffer, 0, 1024);
int fd2 = temp->next->fd;
read(fd, file, 1024);
/*if(-1 == ret)
{
perror("wenjian");
return;
}*/
write(fd2, "file_accept", 1024);//发送让客户端收的信号
write(fd2, file, 1024);//传递的文件名字
while(ret = read(fd, buffer, 1024))
{
if(strcmp(buffer, "end_file") == 0)
{
break;
}
write(fd2, buffer, 1024);
memset(buffer, 0, 1024);
}
write(fd2, "end_file", 1024);
//printf("22-------------------------------\n");
}


  • 0
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
客户端,采用多线程。一个接收服务器消息,一个发送消息给服务器服务器,采用select()进行IO复用。 编译文件是Makefile。 (1)用户登录: 【1】client端接收用户名和密码->存于结构体中->将结构体发送给server端。 【2】server端接收client发送的结构体->打开存储用户名密码的文件->文件写入链表中->遍历链表验证用户信息。 【3】server端验证正确发送“登陆成功”消息,错误发回“登陆失败”消息。client端接收,“登陆成功”则进入聊天,“登陆失败”则跳出。 【4】若验证成功,server端产生一个新的套接字newfd,将它与用户名封装于同一个结构体中,存储在线用户的信息。 消息、存储在线用户信息结构体: typedef struct message { int type; //服务器用于判断该执行的功能 int fd; int mode; //标志位,表示用户的发言权限,1为正常,0为禁言 char name[NAMELEN]; char mima[NAMELEN]; char from[20]; char to[20]; //聊天时的收信人 char file_name[20]; //发送文件时的文件名 char mtext[100]; //聊天时发送的消息内容 struct message *next; }Mess; (2)一对多聊天: 【1】client端发送欲发送的信息给server端。 【2】server端遍历在线人信息链表,找到每个在线人的套接字描述符,将消息发送给链表中的每个人。 【3】可以通过输入“:)”, “:(”, “bye”来发送笑脸,悲伤脸和退出聊天;检测敏感词汇“fuck”、“shit”,禁止发送。 (3)一对一聊天: 【1】client端发送欲发送的信息和信息的接收者给server端。 【2】server端根据收到的接收者名字在在线人链表中查找该接收者的套接字描述符,找到后就将消息发送给该接收者。 【3】可以通过输入“:)”, “:(”, “bye”来发送笑脸,悲伤脸和退出聊天;检测敏感词汇“fuck”、“shit”,禁止发送。 (4)文件传输 【1】client端发送预发送的文件名和接收者名字到server端。 先打开(不存在则创建)一个文件,将文件内容读到缓冲区buffer,再将buffer的内容复制到结构体Mess中,最后将结构体发送给server端。 【2】server端先将接收到的文件重命名(因为相同文件目录下不能存在同名文件),再将收到的文件和新的文件名一同放入tab1中(并且在tab1开头写“#”)发送给client端。 【3】当client端收到以“#”开头的消息,执行文件接收,先创建一个文件,再写入相应内容。 (5)管理员模式 【1】禁言 【2】解禁

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值