这里，对Demo进行了模块化优化，并加上了注释，而且绘制了一个大体的流程图
/*更变：
mysql数据库：
1.数据库:happychat
2.表 :message_tbl
表中变量： recv_name varchar(1024) send_name varchar(1024) mes varchar(2048)
*/
流程图：
模块化划分：
共分为一下几个模块：
mian.cpp :
socket启动及链接，开启线程，开启多路复用
server.cpp
对接收的包按类型进行处理，并把处理结果返回客户端
tools.cpp
对包处理时用到的一些功能函数，对存储信息的数组的一些操作
Data_access.cpp
用户信息文件的读取与保存，数据库的链接，插入，查询，断开链接
md5.cpp
主要是关于md5加密的一些功能函数
debug.cpp
调试用时的输出函数
server.h
结构体声明，函数声明，全局变量声明
下面为各个模块的代码（已加上详细注释）：
mian.cpp :
#include <mysql/mysql.h>
//#include <gtk/gtk.h>
#include <stdio.h>
#include <pthread.h>
#include <sys/socket.h>
#include <fcntl.h>
#include <sys/epoll.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <signal.h>
#include <time.h>
#include <math.h>
using namespace std;
#define LOGIN 1
#define REGISTER 2
#define FRIEND_SEE 3
#define FRIEND_ADD 4
#define FRIEND_DEL 5
#define GROUP_SEE 6
#define GROUP_CREATE 7
#define GROUP_JOIN 8
#define GROUP_QIUT 9
#define GROUP_DEL 10
#define CHAT_ONE 11
#define CHAT_MANY 12
#define FILE_SEND_BEGIN 13
#define FILE_SEND_BEGIN_RP 14
#define FILE_SEND_STOP_RP 15
#define FILE_RECV_RE 16
#define FILE_SEND 17
#define FILE_RECV_BEGIN 18
#define FILE_RECV_BEGIN_RP 19
#define FILE_RECV_STOP_RP 20
#define FILE_RECV 21
#define FILE_FINI_RP 22
#define MES_RECORD 23
#define EXIT -1
#define FILE_STATU_RECV_ING 1
#define FILE_STATU_RECV_STOP 2
#define FILE_STATU_SEND_ING 3
#define FILE_STATU_SEND_FINI 4
#define SIZE_PASS_NAME 10
#define MAX_CHAR 300
#define EPOLL_MAX 200000
#define LISTENMAX 1000
#define USER_MAX 100 //the user on line
#define GROUP_MAX 50
#define FILE_MAX 50
#define NUM_MAX_DIGIT 10
#define DOWNLINE 0
#define ONLINE 1
#define BUZY 2
// #define LOG
#define MYSQL_OPEN
// 储存用户信息结构体
/********************user infor***********************************/
typedef struct infor_user
{
char username[MAX_CHAR]; //用户名
unsigned int password[4]; //经加密后密码
int statu; //用户状态
int socket_id; //用户socketID
char friends[USER_MAX][MAX_CHAR];//begin from 1 好友信息
int friends_num; // 好友数量
char group[GROUP_MAX][MAX_CHAR]; //begin from 1 群组信息
char group_num; //群组数量
}INFO_USER;
//储存群组信息结构体
/*********************group infor***************************/
typedef struct infor_group
{
char group_name[MAX_CHAR]; //组名
int member_num; //成名数量
char member_name[USER_MAX][MAX_CHAR]; //begin from 1 成员名
}INFO_GROUP;
//储存文件信息结构体
/******************file_infor************************************/
typedef struct file_infor
{
char file_name[MAX_CHAR]; //文件名
char file_send_name[MAX_CHAR]; //发送方
char file_recv_name[MAX_CHAR]; //接收方
int file_size; //文件大小
int file_size_now; //文件当前大小
int flag ; //状态
}INFO_FILE;
//包结构体
/*****************************************************************/
typedef struct datas{
char send_name[MAX_CHAR]; //发送方
char recv_name[MAX_CHAR]; //接收方
int send_fd; //发送方fd
int recv_fd; //接收方fd
//time_t time;
char mes[MAX_CHAR*2]; //信息
}DATA;
typedef struct package{
int type;
DATA data;
}PACK;
//线程传参结构体
/*****************************************************************/
typedef struct pthread_parameter
{
int a;
char str1[SIZE_PASS_NAME];
char str2[SIZE_PASS_NAME];
}PTHREAD_PAR;
typedef struct sockaddr SA;//类型转换
/*********************md5*****************************************
/***********************************
* 非线性函数
* (&是与,|是或,~是非,^是异或)
*
* 这些函数是这样设计的：
* 如果X、Y和Z的对应位是独立和均匀的，
* 那么结果的每一位也应是独立和均匀的。
*
* 函数F是按逐位方式操作：如果X，那么Y，否则Z。
* 函数H是逐位奇偶操作符
**********************************/
#define F(x,y,z) ((x & y) | (~x & z))
#define G(x,y,z) ((x & z) | (y & ~z))
#define H(x,y,z) (x^y^z)
#define I(x,y,z) (y ^ (x | ~z))
/**************************************
*向右环移n个单位
* ************************************/
#define ROTATE_LEFT(x,n) ((x << n) | (x >> (32-n)))
/****************************************************
* 每次操作对a，b，c和d中的其中三个作一次非线性函数运算
* F(b,c,d) G(b,c,d) H(b,c,d) I(b,c,d)
*
* 然后将所得结果加上 第四个变量(a)，
* F(b,c,d)+a
*
* 文本的一个子分组(x)
* F(b,c,d)+a+x
*
* 和一个常数(ac)。
* F(b,c,d)+a+x+ac
*
* 再将所得结果向右环移一个不定的数(s)，
* ROTATE_LEFT( F(b,c,d)+a+x+ac , s )
*
* 并加上a，b，c或d中之一(b)。
* ROTATE_LEFT( F(b,c,d)+a+x+ac , s )+b
*
* 最后用该结果取代a，b，c或d中之一(a)。
* a=ROTATE_LEFT( F(b,c,d)+a+x+ac , s )+b
*
* ***************************************************/
#define FF(a,b,c,d,x,s,ac) { a += F(b,c,d) + x + ac; a = ROTATE_LEFT(a,s); a += b; }
#define GG(a,b,c,d,x,s,ac) { a += G(b,c,d) + x + ac; a = ROTATE_LEFT(a,s); a += b; }
#define HH(a,b,c,d,x,s,ac) { a += H(b,c,d) + x + ac; a = ROTATE_LEFT(a,s); a += b; }
#define II(a,b,c,d,x,s,ac) { a += I(b,c,d) + x + ac; a = ROTATE_LEFT(a,s); a += b; }
//储存一个MD5 text信息
typedef struct
{
unsigned int count[2];
//记录当前状态，其数据位数
unsigned int state[4];
//4个数，一共32位 记录用于保存对512bits信息加密的中间结果或者最终结果
unsigned char buffer[64];
//一共64字节，512位
}MD5_CTX;
/**********main***************/
void *deal(void *recv_pack_t);
void *serv_send_thread(void *arg);
void *pthread_check_file(void *arg);
void init_server_pthread();
/************data_access*****************/
int read_infor();
int write_infor();
int conect_mysql_init();
/************debug*****************/
void print_send_pack();
void print_infor_group();
void print_infor_user();
void print_infor_file();
/************tools*****************/
void my_err(const char * err_string,int line);
int find_userinfor(char username_t[]);
void signal_close(int i);
void send_pack(PACK *pack_send_pack_t);
void send_pack_memcpy_server(int type,char *send_name,char *recv_name,int sockfd1,char *mes);
bool md5_judge(int id,char *log_pass);
int judge_usename_same(char username_t[]);
void del_friend_infor(int id,char friend_name[]) ;
void del_group_from_user(char *username,char *groupname);
int find_groupinfor(char groupname_t[]);
int find_fileinfor(char *filename);
/*****************server*********************/
void login(PACK *recv_pack);
void registe(PACK *recv_pack);
void send_statu(PACK *recv_pack);
void friend_add(PACK *recv_pack);
void friend_del(PACK *recv_pack);
void send_mes_to_one(PACK *recv_pack);
void group_create(PACK *recv_pack);
void group_join(PACK *recv_pack);
void group_qiut(PACK *recv_pack);
void group_del_one(int id);
void group_del(PACK *recv_pack);
void send_mes_to_group(PACK *recv_pack);
void file_recv_begin(PACK *recv_pack);
void file_recv(PACK *recv_pack);
void *file_send_send(void *file_send_begin_t);
void file_send_begin(PACK *recv_pack);
void file_send_finish(PACK *recv_pack);
void send_record(PACK *recv_pack);
/**************md5*********************/
void MD5Init(MD5_CTX *context);
void MD5Update(MD5_CTX *context,unsigned char *input,unsigned int inputlen);
void MD5Final(MD5_CTX *context,unsigned char digest[16]);
void MD5Transform(unsigned int state[4],unsigned char block[64]);
void MD5Encode(unsigned char *output,unsigned int *input,unsigned int len);
void MD5Decode(unsigned int *output,unsigned char *input,unsigned int len);
int md5(char encrypt_input[10000],unsigned int decrypt[4]);
INFO_USER m_infor_user [USER_MAX];
int m_user_num;
/**************************************************/
INFO_GROUP m_infor_group [GROUP_MAX]; //begin from 1
int m_group_num;
/******************file_infor************************************/
INFO_FILE m_infor_file [FILE_MAX]; //begin from 1
int m_file_num;
PACK m_pack_send [MAX_CHAR*2];
int m_send_num;
unsigned char PADDING[]={0x80,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,
0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,
0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,
0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0};
int listenfd,epollfd;//
short PORT = 11112;//端口号
int log_file_fd;
int user_infor_fd;
int group_infor_fd;
//线程锁
pthread_mutex_t mutex;
pthread_mutex_t mutex_recv_file;
pthread_mutex_t mutex_check_file;
//处理包的函数，每接受到一个包，开新线程，根据类型进行处理
void *deal(void *recv_pack_t)
{
int i;
PACK *recv_pack = (PACK *)recv_pack_t;
printf("\n\n\ndeal function = %d\n", recv_pack->type);
switch(recv_pack->type)
{
case LOGIN:
login(recv_pack);
break;
case REGISTER:
registe(recv_pack);
break;
case FRIEND_SEE:
send_statu(recv_pack);
break;
case FRIEND_ADD:
friend_add(recv_pack);
break;
case FRIEND_DEL:
friend_del(recv_pack);
break;
case GROUP_CREATE:
group_create(recv_pack);
break;
case GROUP_JOIN:
group_join(recv_pack);
break;
case GROUP_QIUT:
group_qiut(recv_pack);
break;
case GROUP_DEL:
group_del(recv_pack);
break;
case CHAT_ONE:
send_mes_to_one(recv_pack);
break;
case CHAT_MANY:
send_mes_to_group(recv_pack);
break;
case FILE_SEND_BEGIN:
file_recv_begin(recv_pack);
break;
case FILE_SEND:
file_recv(recv_pack);
break;
case FILE_SEND_BEGIN_RP:
file_send_begin(recv_pack);
break;
case FILE_FINI_RP:
file_send_finish(recv_pack);
break;
case MES_RECORD:
send_record(recv_pack);
case EXIT:
break;
}
}
//包发送函数，不断遍历存储包的数组，如果要发送的客户端在线则发送，否则一直等到其上线，
//实现离线功能
/************************send_thread*****************************/
void *serv_send_thread(void *arg)
{
int user_statu = DOWNLINE;
int id_stop;
int i,recv_fd_t,recv_fd_online;
while(1)
{
pthread_mutex_lock(&mutex);//线程锁，保护存包数组
// printf("serv_send_thread:%d\n", m_send_num);
user_statu = DOWNLINE;
for(i = m_send_num-1;i>=0;i--)
{
for(int j =1; j <= m_user_num ;j++)
{
if(strcmp(m_pack_send[i].data.recv_name,m_infor_user[j].username) == 0)
{
user_statu = m_infor_user[j].statu;
if(user_statu == ONLINE)
recv_fd_online = m_infor_user[j].socket_id;
break;
}
}
//上线，则发送包
if(user_statu == ONLINE || m_pack_send[i].type == LOGIN || m_pack_send[i].type == REGISTER)
{
if(user_statu == ONLINE)
recv_fd_t = recv_fd_online;
else
recv_fd_t = m_pack_send[i].data.recv_fd;
if(send(recv_fd_t,&m_pack_send[i],sizeof(PACK),0) < 0){
my_err("send",__LINE__);
}
printf("\n\n\033[;42m*****send pack****\033[0m\n");
printf("\033[;42m*\033[0m type :%d\n",m_pack_send[i].type);
printf("\033[;42m*\033[0m from : %s\n",m_pack_send[i].data.send_name);
printf("\033[;42m*\033[0m to : %s\n",m_pack_send[i].data.recv_name);
printf("\033[;42m*\033[0m mes : %s\n",m_pack_send[i].data.mes);
printf("\033[;42m*\033[0m recv_fd : %d\n",m_pack_send[i].data.recv_fd);
m_send_num-- ;
printf("\033[;42m*\033[0m pack left Now is:%d\n\n", m_send_num);
for(int j = i;j<=m_send_num&&m_send_num;j++)
{
m_pack_send[j] = m_pack_send[j+1];
}
break;
}
}
pthread_mutex_unlock(&mutex);
usleep(1);
}
}
//不断检测文件状态，当文件传输失败，发送提醒给客户端
void *pthread_check_file(void *arg)
{
while(1)
{
pthread_mutex_lock(&mutex_check_file);
for(int i=1 ;i<=m_file_num ;i++)
{
//用户突然下线，发送提醒给客户端
if(m_infor_file[i].file_size <= m_infor_file[i].file_size_now&&(m_infor_file[i].flag == FILE_STATU_RECV_ING ||m_infor_file[i].flag == FILE_STATU_RECV_STOP))
{
char mes_t[MAX_CHAR];
printf("\033[; 35m**********jijmei********?\033[0m \n");
int id = find_userinfor(m_infor_file[i].file_recv_name);
PACK *pthread_check_file_t = (PACK*)malloc(sizeof(PACK));
memset(pthread_check_file_t, 0,sizeof(PACK));
pthread_check_file_t->type = FILE_RECV_BEGIN;
strcpy(pthread_check_file_t->data.send_name ,m_infor_file[i].file_send_name);
strcpy(pthread_check_file_t->data.recv_name,m_infor_file[i].file_recv_name);
int len = m_infor_file[i].file_size;
memset(mes_t,0,sizeof(mes_t));
int digit = 0;
while(len != 0)
{
mes_t[digit++] = len%10;
len /= 10;
}
mes_t[digit] = -1;
for(int j=0 ;j< SIZE_PASS_NAME ;j++)
{
mes_t[NUM_MAX_DIGIT+j] = m_infor_file[i].file_name[j];
}
memcpy(pthread_check_file_t->data.mes,mes_t,sizeof(mes_t));
send_pack(pthread_check_file_t);
free(pthread_check_file_t);
m_infor_file[i].flag = FILE_STATU_SEND_ING;
// m_file_num --;
// for(int j = i;j<=(m_file_num+1)&&m_file_num;j++)
// {
// m_infor_file[j] = m_infor_file[j+1];
// }
break;
}
//文件上传完毕，发送下载提醒给客户端
if(m_infor_file[i].file_size > m_infor_file[i].file_size_now )
{
int id = find_userinfor(m_infor_file[i].file_send_name);
if(m_infor_user[id].statu == DOWNLINE && m_infor_file[i].flag == FILE_STATU_RECV_ING)
{
printf("\033[;35m file_name %s d\033[0m\n", m_infor_file[i].file_name);
char mes[MAX_CHAR];
memset(mes,0,sizeof(mes));
int num = m_infor_file[i].file_size_now;
printf("\033[;35m file_sizejj :%d\033[0m\n",num );
int digit = 0;
while(num != 0)
{
mes[digit++] = num%10;
num /= 10;
}
mes[digit] = -1;
for(int i=0;i<10;i++)
printf("%d\n\n",mes[i]);
printf("\033[;35m file_name %s jjj\033[0m\n", m_infor_file[i].file_name);
PACK pthread_check_file_t;
memset(&pthread_check_file_t, 0,sizeof(PACK));
pthread_check_file_t.type = FILE_SEND_STOP_RP;
strcpy(pthread_check_file_t.data.send_name ,m_infor_file[i].file_name);
strcpy(pthread_check_file_t.data.recv_name,m_infor_file[i].file_send_name);
memcpy(pthread_check_file_t.data.mes,mes,sizeof(mes));
send_pack(&pthread_check_file_t);
printf("jiji$$$$$$$$$$$$$$$$$$$$$$$$$$$$$\n");
m_infor_file[i].flag = FILE_STATU_RECV_STOP;
break;
}
}
//删除服务端缓存文件，并从文件数组中清除该文件信息
if(m_infor_file[i].flag == FILE_STATU_SEND_FINI)
{
unlink(m_infor_file[i].file_name);
m_file_num --;
for(int j = i;j<=(m_file_num+1)&&m_file_num;j++)
{
m_infor_file[j] = m_infor_file[j+1];
}
}
}
pthread_mutex_unlock(&mutex_check_file);
usleep(10);
}
}
//开启线程
void init_server_pthread()
{
printf("\ninit_server_pthread\n");
pthread_t pid_send,pid_file_check;
pthread_create(&pid_send,NULL,serv_send_thread,NULL);
pthread_create(&pid_file_check,NULL,pthread_check_file,NULL);
}
/****************************************************/
int main()
{
int n;
pthread_t pid;
PACK recv_t;
PACK *recv_pack;
int err,socketfd,fd_num;
struct sockaddr_in fromaddr;
socklen_t len = sizeof(fromaddr);
struct epoll_event ev, events[LISTENMAX];
read_infor();
//日志开关，重定向到文件中
#ifdef LOG
if((log_file_fd = open("server_log.txt",O_WRONLY | O_CREAT | O_APPEND, S_IRUSR | S_IWUSR)) < 0)
{
my_err("open",__LINE__);
return 0;
}
dup2(log_file_fd , 1);
#else
#endif
//mysql开关，链接mysql
#ifdef MYSQL_OPEN
conect_mysql_init();
#else
#endif
signal(SIGINT,signal_close);//退出CTRL+C
pthread_mutex_init(&mutex, NULL);
printf("服务器开始启动..\n");
init_server_pthread();
epollfd = epoll_create(EPOLL_MAX);//生成epoll句柄
listenfd = socket(AF_INET,SOCK_STREAM,0);//启动socket
if(listenfd == -1){
perror("创建socket失败");
printf("服务器启动失败\n");
exit(-1);
}
err = 1;
setsockopt(listenfd,SOL_SOCKET,SO_REUSEADDR,&err,sizeof(int));
ev.data.fd = listenfd;//设置与要处理事件相关的文件描述符
ev.events = EPOLLIN;//设置要处理的事件类型
epoll_ctl(epollfd, EPOLL_CTL_ADD, listenfd, &ev);//注册epoll事件
//准备网络通信地址
struct sockaddr_in addr;
addr.sin_family = AF_INET;
addr.sin_port = htons(PORT);
addr.sin_addr.s_addr = htonl(INADDR_ANY);
if(bind(listenfd,(SA*)&addr,sizeof(addr))==-1){//绑定服务器
perror("绑定失败");
printf("服务器启动失败\n");
exit(-1);
}
printf("成功绑定\n");
//设置监听
if(listen(listenfd,10)==-1){
perror("设置监听失败");
printf("服务器启动失败\n");
exit(-1);
}
printf("设置监听成功\n");
printf("初始化服务器成功\n");
printf("服务器开始服务\n");
while(1)
{
fd_num = epoll_wait(epollfd, events, EPOLL_MAX, 1000);
for(int i=0; i<fd_num; i++)
{
if(events[i].data.fd == listenfd)
{
socketfd = accept(listenfd,(SA*)&fromaddr,&len);
printf("%d get conect!\n",socketfd);
ev.data.fd = socketfd;
ev.events = EPOLLIN;//设置监听事件可写
//新增套接字
epoll_ctl(epollfd, EPOLL_CTL_ADD, socketfd, &ev);
}
else if(events[i].events & EPOLLIN)//fd can read
{
n = recv(events[i].data.fd,&recv_t,sizeof(PACK),0);//读取数据
recv_t.data.send_fd = events[i].data.fd;
//输出收到的包信息
printf("\n\n\n\033[;31m ***PACK***\033[0m\n");
printf("\033[;31m*\033[0m type : %d\n", recv_t.type);
printf("\033[;31m*\033[0m send_name: %s\n", recv_t.data.send_name);
printf("\033[;31m*\033[0m recv_name: %s\n",recv_t.data.recv_name);
printf("\033[;31m*\033[0m mes : %s\n", recv_t.data.mes);
printf("\033[;31m*\033[0m send_fd : %d\n", recv_t.data.send_fd);
printf("\033[;31m*\033[0m recv_fd : %d\n", recv_t.data.recv_fd);
printf("\033[;31m*\033[0m send_pack_num:%d\n", m_send_num);
printf("\033[;31m*************\033[0m\n\n\n");
if(n < 0)//recv错误
{
close(events[i].data.fd);
perror("recv");
continue;
}
else if(n == 0)
{
//客户端下线后，把客户端状态设置为下线
for(int j=1;j<=m_user_num;j++)
{
if(events[i].data.fd == m_infor_user[j].socket_id)
{
printf("%s down line!\n",m_infor_user[j].username);
m_infor_user[j].statu = DOWNLINE;
break;
}
}
ev.data.fd = events[i].data.fd;
epoll_ctl(epollfd, EPOLL_CTL_DEL, events[i].data.fd, &ev);//删除套接字
close(events[i].data.fd);
print_infor_user();
continue;
}
//pthread_mutex_lock(&mutex);
//print_send_pack();
//pthread_mutex_unlock(&mutex);
//printf("m_file_num1:%d\n", m_file_num);
//print_infor_file();
// printf("m_file_num2:%d\n", m_file_num);
//printf("hahah ****\n");
int type = 0;
type = recv_t.type;
recv_pack = (PACK*)malloc(sizeof(PACK));
memcpy(recv_pack, &recv_t, sizeof(PACK));
//开启线程处理接受到的包
if(pthread_create(&pid,NULL,deal,(void *)recv_pack) != 0)
my_err("pthread_create",__LINE__);
if(type == FILE_SEND)
{
printf("wati...\n");
usleep(100000);
}
}
}
}
return 0;
}
/^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^/
server.cpp
#include "server.h"
/************************************
* 跟据收到包的内容，找到登录者的信息
* 做密码比对，并返回相应值
* *********************************/
void login(PACK *recv_pack)
{
int id=0;
char login_flag[10];
if((id=find_userinfor(recv_pack->data.send_name)) == 0){//not exit username
login_flag[0] = '2';
}
else if (m_infor_user[id].statu == ONLINE)
{
login_flag[0] = '3';
}
else if( md5_judge(id,recv_pack->data.mes)){//the password is not crrect
login_flag[0] = '1';
//change user infor
m_infor_user[id].socket_id = recv_pack->data.send_fd;
printf("\n\n\033[;45m********login**********\033[0m\n");
printf("\033[;45m*\033[0m %s get online!\n", m_infor_user[id].username);
printf("\033[;45m*\033[0m statu: %d\n", m_infor_user[id].statu);
printf("\033[;45m*\033[0m socket_id:%d\n\n",m_infor_user[id].socket_id);
}
else login_flag[0] = '0';
login_flag[1] = '\0';
/*printf("%s\n", recv_pack->data.send_name);
printf("%s\n", recv_pack->data.mes);
printf("%d\n", m_user_num); */
//包信息赋值
strcpy(recv_pack->data.recv_name,recv_pack->data.send_name);
strcpy(recv_pack->data.send_name,"server");
strcpy(recv_pack->data.mes,login_flag);
recv_pack->data.recv_fd = recv_pack->data.send_fd;
recv_pack->data.send_fd = listenfd;
//发送包
send_pack(recv_pack);
//考虑到中间的数据传输耗时，停顿若干时间，再修改状态
usleep(10);
if(login_flag[0] =='1')
m_infor_user[id].statu = ONLINE;
free(recv_pack);
}
/************************************
* 跟据收到包的内容，判断是否已经注册
* 如果未注册，添加用户信息，并进行密码
* 加密
* *********************************/
void registe(PACK *recv_pack)
{
char registe_flag[10];
if(judge_usename_same(recv_pack->data.send_name)){//the password is not crrect
registe_flag[0] = '1';
//add user infor
m_user_num++;
strcpy(m_infor_user[m_user_num].username,recv_pack->data.send_name);
md5(recv_pack->data.mes,m_infor_user[m_user_num].password);
//服务端提示信息
printf("\n\n\033[;44m&&&&function registe&&&&\033[0m \n");
printf("\033[;44m*\033[0m registe success!\n");
printf("\033[;44m*\033[0m username:%s\n", m_infor_user[m_user_num].username);
//printf("\033[;44m*\033[0m password:%s\n", m_infor_user[m_user_num].password);
printf("\033[;44m*\033[0m user_num:%d\n\n", m_user_num);
m_infor_user[m_user_num].statu = DOWNLINE;
}
else registe_flag[0] = '0';
registe_flag[1] = '\0';
//包信息赋值
strcpy(recv_pack->data.recv_name,recv_pack->data.send_name);
strcpy(recv_pack->data.send_name,"server");
strcpy(recv_pack->data.mes,registe_flag);
recv_pack->data.recv_fd = recv_pack->data.send_fd;
recv_pack->data.send_fd = listenfd;
//发送包
send_pack(recv_pack);
free(recv_pack);
}
/**********************friend******************************/
/************************************
* 跟据收到包的内容，利用字符串处理，
* 把该客户端的朋友信息返回给客户端
* *********************************/
void send_statu(PACK *recv_pack)
{
char str[MAX_CHAR];
char name_t[MAX_CHAR];
char send_statu_mes[MAX_CHAR*2];
int id;
int count = 0;
memset(send_statu_mes,0,sizeof(send_statu_mes));
id = find_userinfor(recv_pack->data.send_name);
send_statu_mes[count++] = m_infor_user[id].friends_num;
//字符串处理
for(int i=1 ;i <= m_infor_user[id].friends_num ;i++)
{
strcpy(name_t,m_infor_user[id].friends[i]);
for(int j=1 ;j <= m_user_num ;j++)
{
if(strcmp(name_t,m_infor_user[j].username) == 0)
{
memset(str,0,sizeof(str));
if(m_infor_user[j].statu == ONLINE)
sprintf(str,"%d%s\0",ONLINE,m_infor_user[j].username);
else
sprintf(str,"%d%s\0",DOWNLINE,m_infor_user[j].username);
printf("str = %s\n",str);
for(int k = 0 ;k< SIZE_PASS_NAME;k++)
{
send_statu_mes[k+count] = str[k];
}
count += SIZE_PASS_NAME;
}
}
}
send_statu_mes[count++] = m_infor_user[id].group_num;
for(int i = 1;i <= m_infor_user[id].group_num;i++)
{
memset(str,0,sizeof(str));
strcpy(name_t,m_infor_user[id].group[i]);
sprintf(str,"%s\0",name_t);
for(int k = 0 ;k< SIZE_PASS_NAME;k++)
{
send_statu_mes[k+count] = str[k];
}
count += SIZE_PASS_NAME;
}
//包信息赋值
strcpy(recv_pack->data.recv_name,recv_pack->data.send_name);
strcpy(recv_pack->data.send_name,"server");
memcpy(recv_pack->data.mes,send_statu_mes,MAX_CHAR*2);
recv_pack->data.recv_fd = recv_pack->data.send_fd;
recv_pack->data.send_fd = listenfd;
send_pack(recv_pack);
free(recv_pack);
}
/************************************
* 跟据收到包的内容，添加朋友信息
* *********************************/
void friend_add(PACK *recv_pack)
{
int id_own,id_friend;
id_own = find_userinfor(recv_pack->data.send_name);
strcpy(m_infor_user[id_own].friends[++(m_infor_user[id_own].friends_num)],recv_pack->data.mes);
id_friend = find_userinfor(recv_pack->data.mes);
strcpy(m_infor_user[id_friend].friends[++(m_infor_user[id_friend].friends_num)],recv_pack->data.send_name);
free(recv_pack);
}
/************************************
* 跟据收到包的内容，删除朋友信息
* *********************************/
void friend_del(PACK *recv_pack)
{
int id_own,id_own_1;
int id_friend,id_friend_1;
id_own = find_userinfor(recv_pack->data.send_name);
del_friend_infor(id_own,recv_pack->data.mes);
id_friend = find_userinfor(recv_pack->data.mes);
del_friend_infor(id_friend,recv_pack->data.send_name);
free(recv_pack);
}
/************************************
* 直接发送信息，并存储到mysql数据库
* *********************************/
void send_mes_to_one(PACK *recv_pack)
{
//存储到数据库
mysql_save_message(recv_pack);
send_pack(recv_pack);
free(recv_pack);
}
/*^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^*/
/****************group*************************/
/************************************
* 跟据收到包的内容，首先判断该群组是否已经创建
* 如果已经创建，则返回出错
* 否则，创建群，并添加客户端为群主
* *********************************/
void group_create(PACK *recv_pack)
{
//判断是否已经建立
for(int i=1;i<= m_group_num;i++)
{
if(strcmp(m_infor_group[i].group_name,recv_pack->data.mes) == 0)
{
strcpy(recv_pack->data.recv_name,recv_pack->data.send_name);
strcpy(recv_pack->data.send_name,"server");
recv_pack->data.mes[0] = 1;
send_pack(recv_pack);
free(recv_pack);
return ;
}
}
//新建群信息
strcpy(m_infor_group[++m_group_num].group_name,recv_pack->data.mes);
strcpy(m_infor_group[m_group_num].member_name[++m_infor_group[m_group_num].member_num],recv_pack->data.send_name);
int id=find_userinfor(recv_pack->data.send_name);
strcpy(m_infor_user[id].group[++m_infor_user[id].group_num],recv_pack->data.mes);
printf("\n\n\033[;32mcreat group : %s successfully!\033[0m \n\n", recv_pack->data.mes);
strcpy(recv_pack->data.recv_name,recv_pack->data.send_name);
strcpy(recv_pack->data.send_name,"server");
recv_pack->data.mes[0] = 2;
send_pack(recv_pack);
free(recv_pack);
}
/************************************
* 跟据收到包的内容，把客户端信息添加到
* 群成员中，若没有该群，返回错误
* *********************************/
void group_join(PACK *recv_pack)
{
//找到该群，并加入
for(int i=1;i<= m_group_num;i++)
{
if(strcmp(m_infor_group[i].group_name,recv_pack->data.mes) == 0)
{
strcpy(m_infor_group[i].member_name[++m_infor_group[i].member_num],recv_pack->data.send_name);
int id=find_userinfor(recv_pack->data.send_name);
strcpy(m_infor_user[id].group[++m_infor_user[id].group_num],recv_pack->data.mes);
strcpy(recv_pack->data.recv_name,recv_pack->data.send_name);
strcpy(recv_pack->data.send_name,"server");
recv_pack->data.mes[0] = 2;
printf("\n\n\033[;32m %s join group : %s successfully!\033[0m \n\n",recv_pack->data.recv_name, recv_pack->data.mes);
print_infor_group();
print_infor_user();
send_pack(recv_pack);
free(recv_pack);
return ;
}
}
//未找到该群
strcpy(recv_pack->data.recv_name,recv_pack->data.send_name);
strcpy(recv_pack->data.send_name,"server");
recv_pack->data.mes[0] = 1;
send_pack(recv_pack);
free(recv_pack);
}
/************************************
* 退出群组，从群组中删除该客户端信息
* 并从客户端信息中删除掉群组
* *********************************/
void group_qiut(PACK *recv_pack)
{
del_group_from_user(recv_pack->data.send_name,recv_pack->data.mes);
for(int i=1 ;i<=m_group_num;i++)
{
if (strcmp(recv_pack->data.mes,m_infor_group[i].group_name) == 0)
{
for(int j=1;j<=m_infor_group[i].member_num;j++)
{
if(strcmp(recv_pack->data.send_name,m_infor_group[i].member_name[j]) == 0)
{
for(int k=j;k <m_infor_group[i].member_num;k++)
{
strcpy(m_infor_group[i].member_name[k],m_infor_group[i].member_name[k+1]);
}
m_infor_group[i].member_num--;
print_infor_group();
}
}
}
}
}
//删除一个指定ID的群组，并从其群成员的信息中删除该组的信息
void group_del_one(int id)
{
for(int i=1;i <= m_infor_group[id].member_num;i++)
{
del_group_from_user(m_infor_group[id].member_name[i],m_infor_group[id].group_name);
}
for(int i = id ;i < m_group_num;i++)
{
m_infor_group[i] = m_infor_group[i+1];
}
m_group_num--;
}
/************************************
* 跟据收到包的内容，判断删除者是否为群主（这里令第一个人为群主）
* 如果是，删除该群
* 否则 返回错误
* *********************************/
void group_del(PACK *recv_pack)
{
for(int i=1;i<=m_group_num;i++)
{
if(strcmp(m_infor_group[i].group_name,recv_pack->data.mes) == 0)
{
//判断是否为群主
if(strcmp(m_infor_group[i].member_name[1],recv_pack->data.send_name) == 0)
{
group_del_one(i);
recv_pack->data.mes[0] = 2;
printf("\n\n\033[;32m delete group : %s successfully!\033[0m \n\n",recv_pack->data.mes);
}
else
recv_pack->data.mes[0] = 1;
}
}
strcpy(recv_pack->data.recv_name,recv_pack->data.send_name);
strcpy(recv_pack->data.send_name,"server");
send_pack(recv_pack);
free(recv_pack);
}
/************************************
* 跟据收到包的内容，利用循环给群成员每一个
* 人都发一遍，并存储到mysql数据库
* *********************************/
void send_mes_to_group(PACK *recv_pack)
{
int id = find_groupinfor(recv_pack->data.recv_name);
int len_mes = strlen(recv_pack->data.mes);
char send_name[SIZE_PASS_NAME];
for(int i=len_mes;i>=0;i--){
recv_pack->data.mes[i+SIZE_PASS_NAME] = recv_pack->data.mes[i];
}
strcpy(send_name ,recv_pack->data.send_name);
for(int i=0;i<SIZE_PASS_NAME;i++){
recv_pack->data.mes[i] = recv_pack->data.send_name[i];
}
strcpy(recv_pack->data.send_name,recv_pack->data.recv_name);
for(int i=1; i<=m_infor_group[id].member_num;i++)
{
strcpy(recv_pack->data.recv_name,m_infor_group[id].member_name[i]);
if(strcmp(send_name , m_infor_group[id].member_name[i]) != 0)
{
mysql_save_message(recv_pack);
//存储到数据库
send_pack(recv_pack);
}
}
free(recv_pack);
}
/*^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^*/
/*****************file*******************/
/************************************
* 跟据收到包的内容，客户端请求发送文件
* 首先判断是否已经有该文件信息
* 如果有，返回该文件的大小
* 否则，建立文件信息，并返回文件大小为0
* *********************************/
void file_recv_begin(PACK *recv_pack)
{
int flag = 0;
int i;
int file_size_now_t;
pthread_mutex_lock(&mutex_check_file);
for(i=1 ;i<= m_file_num ;i++)
{
//文件存在
if(strcmp(m_infor_file[i].file_name,recv_pack->data.mes+NUM_MAX_DIGIT) == 0)
{
file_size_now_t = m_infor_file[i].file_size_now;
flag = 1;
break;
}
}
//不存在，则建立文件信息
if(!flag)
{
file_size_now_t = 0;
strcpy(m_infor_file[++m_file_num].file_name,recv_pack->data.mes+NUM_MAX_DIGIT);
strcpy(m_infor_file[m_file_num].file_send_name,recv_pack->data.send_name);
strcpy(m_infor_file[m_file_num].file_recv_name,recv_pack->data.recv_name);
//解析文件大小
int t=0;
for(int k=0;k<NUM_MAX_DIGIT;k++)
{
if(recv_pack->data.mes[k] == -1)
break;
int t1 = 1;
for(int l=0 ;l<k;l++)
t1*=10;
t += (int)(recv_pack->data.mes[k])*t1;
}
//建立文件信息
m_infor_file[m_file_num].file_size = t;
m_infor_file[m_file_num].file_size_now = 0;
m_infor_file[m_file_num].flag = FILE_STATU_RECV_ING;
}
pthread_mutex_unlock(&mutex_check_file);
recv_pack->type = FILE_SEND_BEGIN_RP;
strcpy(recv_pack->data.recv_name,recv_pack->data.send_name);
int digit = 0;
while(file_size_now_t != 0)
{
recv_pack->data.mes[digit++] = file_size_now_t%10;
file_size_now_t /= 10;
}
recv_pack->data.mes[digit] = -1;
send_pack(recv_pack);
free(recv_pack);
}
/************************************
* 跟据收到包的内容,接收到文件内容
* 就把文件内容写入到文件中
* *********************************/
void file_recv(PACK *recv_pack)
{
//pthread_mutex_lock(&mutex_recv_file);
int fd;
char file_name[MAX_CHAR];
char file_path_t[SIZE_PASS_NAME];
int len = 0;
for(int i=0 ;i<NUM_MAX_DIGIT ;i++)
{
if(recv_pack->data.mes[i] == -1)
break;
int t1 = 1;
for(int l=0;l<i;l++)
t1*=10;
len += (int)recv_pack->data.mes[i]*t1;
}
strcpy(file_name,recv_pack->data.recv_name);
if((fd = open(file_name,O_WRONLY | O_CREAT | O_APPEND, S_IRUSR | S_IWUSR)) < 0)
{
my_err("open",__LINE__);
return ;
}
if(write(fd,recv_pack->data.mes + NUM_MAX_DIGIT,len) < 0)
my_err("write",__LINE__);
// 关闭文件
close(fd);
int id = find_fileinfor(file_name);
m_infor_file[id].file_size_now += len;
free(recv_pack);
//pthread_mutex_unlock(&mutex_recv_file);
//输出文件信息
print_infor_file();
}
/************************************
* 根据文件的起始位置，开始发送，
* 在发送过程中，不断监测接收端的状态
* 若接收端下线，则向接收端发送提醒消息
* *********************************/
void *file_send_send(void *file_send_begin_t)
{
PTHREAD_PAR *file_send_begin = (PTHREAD_PAR *)file_send_begin_t;
int fd;
int length;
int statu;
int id;
int sockfd;
int file_size;
int begin_location = file_send_begin->a;
int sum = begin_location;
char file_name[SIZE_PASS_NAME];
char recv_name[SIZE_PASS_NAME];
char mes[MAX_CHAR*2];
printf("\n\nsending the file.........\n");
strcpy(file_name,file_send_begin->str1);
strcpy(recv_name,file_send_begin->str2);
if((fd = open(file_name,O_RDONLY)) == -1)
{
my_err("open",__LINE__);
return NULL;
}
file_size=lseek(fd, 0, SEEK_END);
//定位到开始位置
lseek(fd ,begin_location ,SEEK_SET);
printf("file_size %d\n", file_size);
printf("begin_location%d\n", begin_location);
bzero(mes, MAX_CHAR*2);
// 每读取一段数据，便将其发送给客户端，循环直到文件读完为止
while((length = read(fd ,mes+NUM_MAX_DIGIT ,MAX_CHAR*2 - NUM_MAX_DIGIT)) > 0)
{
printf("send_::%d\n", sum);
printf("file_size%d\n",file_size);
if(sum >= file_size)
break;
id = find_userinfor(recv_name);
//当文件传输过程中，接收端下线，则向接收端发送提醒消息
if(m_infor_user[id].statu == DOWNLINE)
{
int file_id = find_fileinfor(file_name);
int file_size = m_infor_file[file_id].file_size;
char mes_t[MAX_CHAR];
PACK file_send_stop_t;
memset(&file_send_stop_t, 0,sizeof(PACK));
file_send_stop_t.type = FILE_RECV_STOP_RP;
strcpy(file_send_stop_t.data.send_name ,m_infor_file[file_id].file_send_name);
strcpy(file_send_stop_t.data.recv_name,m_infor_file[file_id].file_recv_name);
memset(mes_t,0,sizeof(mes_t));
//发送服务端已经发送的字节数
int digit = 0;
while(file_size != 0)
{
mes_t[digit++] = file_size%10;
file_size /= 10;
}
mes_t[digit] = -1;
for(int j=0 ;j< SIZE_PASS_NAME ;j++)
{
mes_t[NUM_MAX_DIGIT+j] = m_infor_file[file_id].file_name[j];
}
memcpy(file_send_stop_t.data.mes,mes_t,sizeof(mes_t));
send_pack(&file_send_stop_t);
free(file_send_begin);
return NULL ;
}
sockfd = m_infor_user[id].socket_id;
//在发送数据的前面，为文件已发送的字节数
int t=length;
int digit = 0;
while(t != 0)
{
mes[digit++] = t%10;
t /= 10;
}
mes[digit] = -1;
send_pack_memcpy_server(FILE_RECV,file_name,recv_name,sockfd,mes);
sum += length;
bzero(mes, MAX_CHAR*2);
//为防止发送顺序被打乱，发送时有一定时间间隔
usleep(200000);
}
// 关闭文件
close(fd);
free(file_send_begin);
}
/************************************
* 客户端请求接收文件，解析出客户端已经
* 接收的文件字节数，然后从该字节数开始发送
* *********************************/
void file_send_begin(PACK *recv_pack)
{
PTHREAD_PAR *file_send_begin_t;
file_send_begin_t = (PTHREAD_PAR *)malloc(sizeof(PTHREAD_PAR));
char recv_name[SIZE_PASS_NAME];
char file_name[SIZE_PASS_NAME];
int begin_location=0;
strcpy(recv_name,recv_pack->data.send_name);
strcpy(file_name,recv_pack->data.recv_name);
//解析出客户端已经接收文件字节数
for(int i=0 ;i<NUM_MAX_DIGIT ;i++)
{
if(recv_pack->data.mes[i] == -1)
break;
int t1 = 1;
for(int l=0;l<i;l++)
t1*=10;
begin_location += (int)recv_pack->data.mes[i]*t1;
}
int file_id = find_fileinfor(file_name);
if(begin_location >= m_infor_file[file_id].file_size)
{
m_infor_file[file_id].flag = FILE_STATU_SEND_FINI;
}
file_send_begin_t->a = begin_location;
strcpy(file_send_begin_t->str1,file_name);
strcpy(file_send_begin_t->str2,recv_name);
//断点续传
file_send_send((void *)file_send_begin_t);
free(recv_pack);
}
/************************************
* 客户端接收完，会发送确认信息，
* 然后把文件状态改为:已经发送完毕
* 其他函数对该文件进行相应处理
* *********************************/
void file_send_finish(PACK *recv_pack)
{
int id = find_fileinfor(recv_pack->data.mes);
m_infor_file[id].flag = FILE_STATU_SEND_FINI;
free(recv_pack);
}
/*^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^*/
/^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^/
tools.cpp
#include "server.h"
//输出错误信息
void my_err(const char * err_string,int line)
{
fprintf(stderr, "line:%d ", line);
perror(err_string);
exit(1);
}
//根据姓名找到用户信息
int find_userinfor(char username_t[])
{
int i;
if(m_user_num == 0) return 0;
for(i=1;i<=m_user_num;i++)
{
if(strcmp(m_infor_user[i].username,username_t) == 0)
return i;
}
if(i == m_user_num+1)
return 0;
}
//关闭服务器前 关闭服务器的socket
void signal_close(int i)
{
#ifdef LOG
close(log_file_fd);
#else
#endif
write_infor();
mysql_close();
printf("服务器已经关闭\n");
exit(1);
}
//发送包，把包的内容赋值到发送包的数组
void send_pack(PACK *pack_send_pack_t)
{
pthread_mutex_lock(&mutex);
memcpy(&(m_pack_send[m_send_num++]),pack_send_pack_t,sizeof(PACK));
pthread_mutex_unlock(&mutex);
}
//根据参数信息发送包
void send_pack_memcpy_server(int type,char *send_name,char *recv_name,int sockfd1,char *mes)
{
PACK pack_send_pack;
time_t timep;
pack_send_pack.type = type;
strcpy(pack_send_pack.data.send_name,send_name);
strcpy(pack_send_pack.data.recv_name,recv_name);
memcpy(pack_send_pack.data.mes,mes,MAX_CHAR*2);
time(&timep);
//pack_send_pack.data.time = timep;
printf("sockfd1:%d\n", sockfd1);
if(send(sockfd1,&pack_send_pack,sizeof(PACK),0) < 0){
my_err("send",__LINE__);
}
}
//判断客户端登录密码每是否正确
bool md5_judge(int id,char *log_pass)
{
unsigned int md5_t[4];
md5(log_pass,md5_t);
for(int i=0;i<4;i++)
{
if(m_infor_user[id].password[i] != md5_t[i])
return false;
}
return true;
}
//判断用户信息中是否有当前名字的信息
int judge_usename_same(char username_t[])
{
int i;
if(m_user_num == 0) return 1;
for(i=1;i<=m_user_num;i++)
{
if(strcmp(m_infor_user[i].username,username_t) == 0)
return 0;
}
if(i == m_user_num+1)
return 1;
}
//给定用户ID，从用户信息删除好友
void del_friend_infor(int id,char friend_name[])
{
int id_1;
for(int i = 1 ;i<=m_infor_user[id].friends_num;i++)
{
if(strcmp(m_infor_user[id].friends[i],friend_name) == 0)
{
id_1 = i;
break;
}
}
for(int i = id_1;i<m_infor_user[id].friends_num;i++)
{
strcpy(m_infor_user[id].friends[i],m_infor_user[id].friends[i+1]);
}
m_infor_user[id].friends_num--;
}
//给定用户信息，从用户信息中删除组
void del_group_from_user(char *username,char *groupname)
{
int id = find_userinfor(username);
for(int i = 1;i<=m_infor_user[id].group_num;i++)
{
if(strcmp(m_infor_user[id].group[i],groupname) == 0)
{
for(int j = i ;j < m_infor_user[id].group_num ;j++)
{
strcpy(m_infor_user[id].group[j],m_infor_user[id].group[j+1]);
}
m_infor_user[id].group_num--;
}
}
}
//给定组名，找到信息在数组中位置
int find_groupinfor(char groupname_t[])
{
int i;
if(m_group_num == 0) return 0;
for(i=1;i<=m_group_num;i++)
{
if(strcmp(m_infor_group[i].group_name,groupname_t) == 0)
return i;
}
if(i == m_group_num+1)
return 0;
}
//给定文件名，返回信息在数组中位置
int find_fileinfor(char *filename)
{
for(int i=1 ;i <= m_file_num ;i++)
{
if(strcmp(filename , m_infor_file[i].file_name) == 0)
{
return i;
}
}
return 0;
}
/^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^/
Data_access.cpp
#include "server.h"
//数据库全局变量
/********************mysql*************************/
MYSQL mysql;
MYSQL_RES *res = NULL;
MYSQL_ROW row;
char query_str[MAX_CHAR*4] ;
int rc, fields;
int rows;
//从文件中读取用户信息
int read_infor()
{
INFO_USER read_file_t;
if((user_infor_fd = open("user_infor.txt",O_RDONLY | O_CREAT , S_IRUSR | S_IWUSR)) < 0)
{
my_err("open",__LINE__);
return -1;
}
int length ;
while((length = read(user_infor_fd, &read_file_t, sizeof(INFO_USER))) > 0)
{
read_file_t.statu = DOWNLINE;
m_infor_user[++m_user_num] = read_file_t;
}
close(user_infor_fd);
printf("have read the user_infor.txt.\n");
INFO_GROUP read_file_group;
if((group_infor_fd = open("group_infor.txt",O_RDONLY | O_CREAT , S_IRUSR | S_IWUSR)) < 0)
{
my_err("open",__LINE__);
return -1;
}
while((length = read(group_infor_fd, &read_file_group, sizeof(INFO_GROUP))) > 0)
{
m_infor_group[++m_group_num] = read_file_group;
}
close(group_infor_fd);
printf("have read the group_infor.txt.\n");
}
//服务端关闭时把信息写入到文件中
int write_infor()
{
INFO_USER read_file_t;
if((user_infor_fd = open("user_infor.txt",O_WRONLY | O_CREAT |O_TRUNC , S_IRUSR | S_IWUSR)) < 0)
{
my_err("open",__LINE__);
return -1;
}
int length ;
for(int i= 1 ; i<=m_user_num ;i++)
{
if(write(user_infor_fd, m_infor_user+i, sizeof(INFO_USER)) < 0)
{
my_err("close",__LINE__);
return -1;
}
}
close(user_infor_fd);
printf("have save the user_infor.txt\n");
if((group_infor_fd = open("group_infor.txt",O_WRONLY | O_CREAT |O_TRUNC , S_IRUSR | S_IWUSR)) < 0)
{
my_err("open",__LINE__);
return -1;
}
for(int i= 1 ; i<=m_group_num ;i++)
{
if(write(group_infor_fd, m_infor_group+i, sizeof(INFO_GROUP)) < 0)
{
my_err("close",__LINE__);
return -1;
}
}
close(group_infor_fd);
printf("have save the group_infor.txt.\n");
}
//链接mysql数据库
/***********************mysql**********************************/
int conect_mysql_init()
{
if (NULL == mysql_init(&mysql)) { //初始化mysql变量
printf("mysql_init(): %s\n", mysql_error(&mysql));
return -1;
}
if (NULL == mysql_real_connect(&mysql, //链接mysql数据库
"localhost", //链接的当地名字
"root", //用户名字
"root", //用户密码
"happychat", //所要链接的数据库
0,
NULL,
0)) {
printf("mysql_real_connect(): %s\n", mysql_error(&mysql));
return -1;
}
printf("1. Connected MySQL successful! \n");
}
void mysql_save_message(PACK *recv_pack)
{
#ifdef MYSQL_OPEN
sprintf(query_str, "insert into message_tbl values ('%s','%s' ,'%s')",recv_pack->data.send_name,recv_pack->data.recv_name,recv_pack->data.mes);
rc = mysql_real_query(&mysql, query_str, strlen(query_str)); //对数据库执行 SQL 语句
if (0 != rc) {
printf("mysql_real_query(): %s\n", mysql_error(&mysql));
return ;
}
printf("the message get into the mysql!!!\n");
#else
#endif
}
void mysql_close()
{
mysql_free_result(res);
mysql_close(&mysql);
}
/************************************
* 客户端请求历史记录，通过查询，把mysql数据
* 库中的信息导出，发送给客户端
* *********************************/
void send_record(PACK *recv_pack)
{
char send_name[MAX_CHAR];
char recv_name[MAX_CHAR];
char mes[MAX_CHAR*2];
PACK *pack_send_record_t = (PACK *)malloc(sizeof(PACK));
strcpy(send_name,recv_pack->data.send_name);
strcpy(recv_name,recv_pack->data.mes);
sprintf(query_str,"select send_name ,recv_name,mes from message_tbl where send_name='%s' or send_name = '%s'",send_name,recv_name); //显示 表中数据
rc = mysql_real_query(&mysql, query_str, strlen(query_str));
if (0 != rc) {
printf("mysql_real_query(): %s\n", mysql_error(&mysql));
return ;
}
res = mysql_store_result(&mysql); //获取查询结果
if (NULL == res) {
printf("mysql_restore_result(): %s\n", mysql_error(&mysql));
return ;
}
int rows = mysql_num_rows(res); //获取查询结果的行数
printf("The total rows is: %d\n", rows);
int fields = mysql_num_fields(res); //获取查询结果的列数
printf("The total fields is: %d\n", fields);
//发送查询到的数据
while ((row = mysql_fetch_row(res)))
{
pack_send_record_t->type = MES_RECORD;
strcpy(pack_send_record_t->data.send_name,"server");
strcpy(pack_send_record_t->data.recv_name,send_name);
strcpy(pack_send_record_t->data.mes,row[0]);
//strcpy(pack_send_record_t->data.mes+SIZE_PASS_NAME,row[3]);
strcpy(pack_send_record_t->data.mes+SIZE_PASS_NAME,row[2]);
printf("%s\n",pack_send_record_t->data.mes+SIZE_PASS_NAME);
send_pack(pack_send_record_t);
usleep(100000);
}
usleep(100000);
printf("haha\n");
//发送包，表示已经传输完毕
pack_send_record_t->type = MES_RECORD;
strcpy(pack_send_record_t->data.send_name,"server_end");
strcpy(pack_send_record_t->data.recv_name,send_name);
send_pack(pack_send_record_t);
usleep(10000);
free(pack_send_record_t);
}
/^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^/
md5.cpp
#include "server.h"
/************************
* 函数功能：初始化一个MD5 text
* 函数参数：MD5 text 指针
* ***********************/
//初始化
void MD5Init(MD5_CTX *context)
{
context->count[0] = 0;
context->count[1] = 0;
//分别赋固定值
context->state[0] = 0x67452301;
context->state[1] = 0xEFCDAB89;
context->state[2] = 0x98BADCFE;
context->state[3] = 0x10325476;
}
/************************************************
* 函数功能：对一个MD5 text,把输入的数据进行分组，并进行加密
* 未用到的数据把其储存在MD5 text中。
*
* 参数分析：
* MD5_CTX *context ：一个MD5 text
* unsigned char *input ：新添加的数据
* unsigned int inputlen ：新添加数据的长度(字节)
*
***********************************************/
void MD5Update(MD5_CTX *context,unsigned char *input,unsigned int inputlen)
{
unsigned int i = 0,index = 0,partlen = 0;
//index：当前状态的位数对64取余，其单位是字节
//也可以写作： index=(context->count[0]/8)%64
index = (context->count[0] >> 3) & 0x3F;
//partlen:可以补齐64字节的字节数
partlen = 64 - index;
//下面代码是解决一个unsignde int 无法储存极大数据导致溢出的问题
//当前位数加上新添加的位数，由于inputlen是以字节为单位，所以其转换为位数
//相当于context->count[0] += inputlen*8;
context->count[0] += inputlen << 3;
//当其出现溢出的情况时，通过以下操作把两个16位的数连在一块，生成一个
//32位的二进制数串，从而扩大其储存范围
if(context->count[0] < (inputlen << 3))
context->count[1]++;
//该语句可替换为 context->count[1]+=(inputlen<<3)>>32;
//便于理解
context->count[1] += inputlen >> 29;
//当其输入字节数的大于其可以补足64字节的字节数，进行补足
if(inputlen >= partlen)
{
//向buffer中补足partlen个字节，使其到达64字节
memcpy(&context->buffer[index],input,partlen);
//buffer达到64字节512位，则把其作为一组进行运算
MD5Transform(context->state,context->buffer);
//如果输入的数据还可以组成多个64字节，则把其可以组成
//的作为若干组进行运算
for(i = partlen;i+64 <= inputlen;i+=64)
MD5Transform(context->state,&input[i]);
//恢复0值，照应 下面 把输入 剩余字节(不能组成64字节组) 储存的操作
index = 0;
}
//否则，把输入的数据按顺序放在原来数据后面
else
{
i = 0;
}
//放置剩余数据
memcpy(&context->buffer[index],&input[i],inputlen-i);
}
/*************************************************
* 函数功能：对数据进行补足，并加入数据位数信息，并进一步加密
*
* 参数分析：
* MD5_CTX *context :一个MD5 text
* unsigned char digest[16] :储存加密结果的数组
*************************************************/
void MD5Final(MD5_CTX *context,unsigned int digest[16])
{
unsigned int index = 0,padlen = 0;
//bits： 8个字节，64位
unsigned char bits[8];
//index:对64取余结果
index = (context->count[0] >> 3) & 0x3F;
//因为要填充满足使其位长对512求余的结果等于448（56位）
//所以当其所剩余的数小于56字节，则填充56-index字节，
//否则填充120-index字节
//这里padlen代表其所需填充的字节
padlen = (index < 56)?(56-index):(120-index);
//然后，在这个结果后面附加一个以64位二进制表示的填充前数据长度。
//把填充前数据数据长度转换后放到bit字符数组中
MD5Encode(bits,context->count,8);
//根据已经存储好的数组PADDING，在信息的后面填充一个1和无数个0，
//直到满足上面的条件时才停止用0对信息的填充
//其填充后进行了一系列的加密操作，其定剩余48个字节
MD5Update(context,PADDING,padlen);
//在最后添加进8个字节的数据长度信息，最后凑成一组，进行一次加密处理
MD5Update(context,bits,8);
//把最终得到的加密信息变成字符输出，共16字节
for(int i=0;i<4;i++)
digest[i]= context->state[i];
}
/**********************************************************
* 函数功能：利用位操作，按1->4方式把数字分解成字符
*
* 参数分析：
* unsigned char *output ：输出的字符的数组
* unsigned int *input ：输入数字的数组
* unsigned int len : 输入数字数组的长度（单位：位）
* *********************************************************/
void MD5Encode(unsigned char *output,unsigned int *input,unsigned int len)
{
unsigned int i = 0,j = 0;
while(j < len)
{
//这里& 0xFF为取后8位
//i代表数字数组下标
//j代表字符数组下标
//把数字的8、8-16、16-24、24-32分别赋值给字符
output[j] = input[i] & 0xFF;
output[j+1] = (input[i] >> 8) & 0xFF;
output[j+2] = (input[i] >> 16) & 0xFF;
output[j+3] = (input[i] >> 24) & 0xFF;
i++;
j+=4;
}
}
/**********************************************************
* 函数功能：利用位操作，按4->1方式把字符合成数字
*
* 参数分析：
* unsigned int *output ：输出的数字的数组
* unsigned char *input ：输入字符的数组
* unsigned int len : 输入字符的长度 （单位：位）
* *********************************************************/
void MD5Decode(unsigned int *output,unsigned char *input,unsigned int len)
{
unsigned int i = 0,j = 0;
while(j < len)
{
//利用位操作，把四个单位为1字节的字符，合成一个单位为4字节的数字
//因为FF GG HH II和非线性函数都只能对数字进行处理
//第一个字符占前8位，第二个占8-16位，第三个占16-24位，第四个占
//24-32位。
//i代表数字数组下标
//j代表字符数组下标
output[i] = (input[j]) |
(input[j+1] << 8) |
(input[j+2] << 16) |
(input[j+3] << 24);
i++;
j+=4;
}
}
/**************************************************************
* 函数功能：对512位的block数据进行加密，并把加密结果存入state数组中
* 对512位信息(即block字符数组)进行一次处理，每次处理包括四轮
*state[4]：md5结构中的state[4]，用于保存对512bits信息加密的中间结果或者最终结果
* block[64]：欲加密的512bits信息或其中间数据
***************************************************************/
void MD5Transform(unsigned int state[4],unsigned char block[64])
{
//a b c d继承上一个加密的结果，所以其具有继承性
unsigned int a = state[0];
unsigned int b = state[1];
unsigned int c = state[2];
unsigned int d = state[3];
//这里只需用到16个，我把原来的unsiged int x[64] 改为了 x[16]
unsigned int x[16];
//把字符转化成数字，便于运算
MD5Decode(x,block,64);
//具体函数方式固定，不再赘述
/*************第一轮******************/
FF(a, b, c, d, x[ 0], 7, 0xd76aa478);
FF(d, a, b, c, x[ 1], 12, 0xe8c7b756);
FF(c, d, a, b, x[ 2], 17, 0x242070db);
FF(b, c, d, a, x[ 3], 22, 0xc1bdceee);
FF(a, b, c, d, x[ 4], 7, 0xf57c0faf);
FF(d, a, b, c, x[ 5], 12, 0x4787c62a);
FF(c, d, a, b, x[ 6], 17, 0xa8304613);
FF(b, c, d, a, x[ 7], 22, 0xfd469501);
FF(a, b, c, d, x[ 8], 7, 0x698098d8);
FF(d, a, b, c, x[ 9], 12, 0x8b44f7af);
FF(c, d, a, b, x[10], 17, 0xffff5bb1);
FF(b, c, d, a, x[11], 22, 0x895cd7be);
FF(a, b, c, d, x[12], 7, 0x6b901122);
FF(d, a, b, c, x[13], 12, 0xfd987193);
FF(c, d, a, b, x[14], 17, 0xa679438e);
FF(b, c, d, a, x[15], 22, 0x49b40821);
/*************第二轮*****************/
GG(a, b, c, d, x[ 1], 5, 0xf61e2562);
GG(d, a, b, c, x[ 6], 9, 0xc040b340);
GG(c, d, a, b, x[11], 14, 0x265e5a51);
GG(b, c, d, a, x[ 0], 20, 0xe9b6c7aa);
GG(a, b, c, d, x[ 5], 5, 0xd62f105d);
GG(d, a, b, c, x[10], 9, 0x2441453);
GG(c, d, a, b, x[15], 14, 0xd8a1e681);
GG(b, c, d, a, x[ 4], 20, 0xe7d3fbc8);
GG(a, b, c, d, x[ 9], 5, 0x21e1cde6);
GG(d, a, b, c, x[14], 9, 0xc33707d6);
GG(c, d, a, b, x[ 3], 14, 0xf4d50d87);
GG(b, c, d, a, x[ 8], 20, 0x455a14ed);
GG(a, b, c, d, x[13], 5, 0xa9e3e905);
GG(d, a, b, c, x[ 2], 9, 0xfcefa3f8);
GG(c, d, a, b, x[ 7], 14, 0x676f02d9);
GG(b, c, d, a, x[12], 20, 0x8d2a4c8a);
/*************第三轮*****************/
HH(a, b, c, d, x[ 5], 4, 0xfffa3942);
HH(d, a, b, c, x[ 8], 11, 0x8771f681);
HH(c, d, a, b, x[11], 16, 0x6d9d6122);
HH(b, c, d, a, x[14], 23, 0xfde5380c);
HH(a, b, c, d, x[ 1], 4, 0xa4beea44);
HH(d, a, b, c, x[ 4], 11, 0x4bdecfa9);
HH(c, d, a, b, x[ 7], 16, 0xf6bb4b60);
HH(b, c, d, a, x[10], 23, 0xbebfbc70);
HH(a, b, c, d, x[13], 4, 0x289b7ec6);
HH(d, a, b, c, x[ 0], 11, 0xeaa127fa);
HH(c, d, a, b, x[ 3], 16, 0xd4ef3085);
HH(b, c, d, a, x[ 6], 23, 0x4881d05);
HH(a, b, c, d, x[ 9], 4, 0xd9d4d039);
HH(d, a, b, c, x[12], 11, 0xe6db99e5);
HH(c, d, a, b, x[15], 16, 0x1fa27cf8);
HH(b, c, d, a, x[ 2], 23, 0xc4ac5665);
/*************第四轮******************/
II(a, b, c, d, x[ 0], 6, 0xf4292244);
II(d, a, b, c, x[ 7], 10, 0x432aff97);
II(c, d, a, b, x[14], 15, 0xab9423a7);
II(b, c, d, a, x[ 5], 21, 0xfc93a039);
II(a, b, c, d, x[12], 6, 0x655b59c3);
II(d, a, b, c, x[ 3], 10, 0x8f0ccc92);
II(c, d, a, b, x[10], 15, 0xffeff47d);
II(b, c, d, a, x[ 1], 21, 0x85845dd1);
II(a, b, c, d, x[ 8], 6, 0x6fa87e4f);
II(d, a, b, c, x[15], 10, 0xfe2ce6e0);
II(c, d, a, b, x[ 6], 15, 0xa3014314);
II(b, c, d, a, x[13], 21, 0x4e0811a1);
II(a, b, c, d, x[ 4], 6, 0xf7537e82);
II(d, a, b, c, x[11], 10, 0xbd3af235);
II(c, d, a, b, x[ 2], 15, 0x2ad7d2bb);
II(b, c, d, a, x[ 9], 21, 0xeb86d391);
//更换原来的结果
state[0] += a;
state[1] += b;
state[2] += c;
state[3] += d;
}
int md5(char encrypt_input[10000],unsigned int decrypt[4])
{
MD5_CTX md5; //定义一个MD5 text
MD5Init(&md5);//初始化
int i;
unsigned char *encrypt =(unsigned char *)encrypt_input;//要加密内容
//21232f297a57a5a743894a0e4a801fc3
// ; //加密结果
MD5Update(&md5,encrypt,strlen((char *)encrypt));//进行初步分组加密
MD5Final(&md5,decrypt); //进行后序的补足，并加密
return 0;
}
/^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^/
debug.cpp
#include "server.h"
/***************************debug******************************/
void print_send_pack()
{
for(int i=1;i<=m_send_num;i++)
{
printf("********%d*********\n", i);
printf("type :%d\n", m_pack_send[i].type);
printf("send_name :%s\n", m_pack_send[i].data.send_name);
printf("recv_name :%s\n", m_pack_send[i].data.recv_name);
printf("send_fd :%d\n", m_pack_send[i].data.send_fd);
printf("recv_fd :%s\n", m_pack_send[i].data.recv_fd);
printf("mes :%s\n", m_pack_send[i].data.mes);
printf("***********************\n\n\n");
}
}
void print_infor_group()
{
for(int i=1;i<=m_group_num;i++)
{
printf("\n\n********%d*********\n", i);
printf("group_name :%s\n", m_infor_group[i].group_name);
printf("group_num :%d\n", m_infor_group[i].member_num);
for(int j=1 ;j<=m_infor_group[i].member_num;j++)
printf("*%s\n", m_infor_group[i].member_name[j]);
printf("***********************\n\n\n");
}
}
void print_infor_user()
{
for(int i=1;i<=m_user_num;i++)
{
printf("\n**\033[;33m *****user***%d*********\033[0m \n", i);
printf("user_name :%s\n", m_infor_user[i].username);
printf("int statu :%d\n", m_infor_user[i].statu);
printf("friends_num:%d\n", m_infor_user[i].friends_num);
for(int j=1 ;j<=m_infor_user[i].friends_num;j++)
printf("*%s\n", m_infor_user[i].friends[j]);
printf("user_num :%d\n", m_infor_user[i].group_num);
for(int j=1 ;j<=m_infor_user[i].group_num;j++)
printf("*%s\n", m_infor_user[i].group[j]);
printf("***********************\n\n\n");
}
}
void print_infor_file()
{
pthread_mutex_lock(&mutex_check_file);
for(int i=1;i<=m_file_num;i++)
{
printf("\n\n****file***%d*********\n", i);
printf("file_name :%s\n", m_infor_file[i].file_name);
printf("send_name :%s\n", m_infor_file[i].file_send_name);
printf("recv_name :%s\n", m_infor_file[i].file_recv_name);
printf("file_size :%d\n", m_infor_file[i].file_size);
printf("file_size_now :%d\n", m_infor_file[i].file_size_now);
printf("flag :%d\n", m_infor_file[i].flag);
printf("***********************\n\n\n");
}
pthread_mutex_unlock(&mutex_check_file);
}
/^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^/
server.h
#include <mysql/mysql.h>
//#include <gtk/gtk.h>
#include <stdio.h>
#include <pthread.h>
#include <sys/socket.h>
#include <fcntl.h>
#include <sys/epoll.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <signal.h>
#include <time.h>
#include <math.h>
using namespace std;
#define LOGIN 1
#define REGISTER 2
#define FRIEND_SEE 3
#define FRIEND_ADD 4
#define FRIEND_DEL 5
#define GROUP_SEE 6
#define GROUP_CREATE 7
#define GROUP_JOIN 8
#define GROUP_QIUT 9
#define GROUP_DEL 10
#define CHAT_ONE 11
#define CHAT_MANY 12
#define FILE_SEND_BEGIN 13
#define FILE_SEND_BEGIN_RP 14
#define FILE_SEND_STOP_RP 15
#define FILE_RECV_RE 16
#define FILE_SEND 17
#define FILE_RECV_BEGIN 18
#define FILE_RECV_BEGIN_RP 19
#define FILE_RECV_STOP_RP 20
#define FILE_RECV 21
#define FILE_FINI_RP 22
#define MES_RECORD 23
#define EXIT -1
#define FILE_STATU_RECV_ING 1
#define FILE_STATU_RECV_STOP 2
#define FILE_STATU_SEND_ING 3
#define FILE_STATU_SEND_FINI 4
#define SIZE_PASS_NAME 10
#define MAX_CHAR 300
#define EPOLL_MAX 200000
#define LISTENMAX 1000
#define USER_MAX 100 //the user on line
#define GROUP_MAX 50
#define FILE_MAX 50
#define NUM_MAX_DIGIT 10
#define DOWNLINE 0
#define ONLINE 1
#define BUZY 2
// #define LOG
#define MYSQL_OPEN
/********************user infor***********************************/
typedef struct infor_user
{
char username[MAX_CHAR]; //用户名
unsigned int password[4]; //经加密后密码
int statu; //用户状态
int socket_id; //用户socketID
char friends[USER_MAX][MAX_CHAR];//begin from 1 好友信息
int friends_num; // 好友数量
char group[GROUP_MAX][MAX_CHAR]; //begin from 1 群组信息
char group_num; //群组数量
}INFO_USER;
//储存群组信息结构体
/*********************group infor***************************/
typedef struct infor_group
{
char group_name[MAX_CHAR]; //组名
int member_num; //成名数量
char member_name[USER_MAX][MAX_CHAR]; //begin from 1 成员名
}INFO_GROUP;
//储存文件信息结构体
/******************file_infor************************************/
typedef struct file_infor
{
char file_name[MAX_CHAR]; //文件名
char file_send_name[MAX_CHAR]; //发送方
char file_recv_name[MAX_CHAR]; //接收方
int file_size; //文件大小
int file_size_now; //文件当前大小
int flag ; //状态
}INFO_FILE;
//包结构体
/*****************************************************************/
typedef struct datas{
char send_name[MAX_CHAR]; //发送方
char recv_name[MAX_CHAR]; //接收方
int send_fd; //发送方fd
int recv_fd; //接收方fd
//time_t time;
char mes[MAX_CHAR*2]; //信息
}DATA;
typedef struct package{
int type;
DATA data;
}PACK;
//线程传参结构体
/*****************************************************************/
typedef struct pthread_parameter
{
int a;
char str1[SIZE_PASS_NAME];
char str2[SIZE_PASS_NAME];
}PTHREAD_PAR;
typedef struct sockaddr SA;//类型转换
/*********************md5*****************************************
/***********************************
* 非线性函数
* (&是与,|是或,~是非,^是异或)
*
* 这些函数是这样设计的：
* 如果X、Y和Z的对应位是独立和均匀的，
* 那么结果的每一位也应是独立和均匀的。
*
* 函数F是按逐位方式操作：如果X，那么Y，否则Z。
* 函数H是逐位奇偶操作符
**********************************/
#define F(x,y,z) ((x & y) | (~x & z))
#define G(x,y,z) ((x & z) | (y & ~z))
#define H(x,y,z) (x^y^z)
#define I(x,y,z) (y ^ (x | ~z))
/**************************************
*向右环移n个单位
* ************************************/
#define ROTATE_LEFT(x,n) ((x << n) | (x >> (32-n)))
/****************************************************
* 每次操作对a，b，c和d中的其中三个作一次非线性函数运算
* F(b,c,d) G(b,c,d) H(b,c,d) I(b,c,d)
*
* 然后将所得结果加上 第四个变量(a)，
* F(b,c,d)+a
*
* 文本的一个子分组(x)
* F(b,c,d)+a+x
*
* 和一个常数(ac)。
* F(b,c,d)+a+x+ac
*
* 再将所得结果向右环移一个不定的数(s)，
* ROTATE_LEFT( F(b,c,d)+a+x+ac , s )
*
* 并加上a，b，c或d中之一(b)。
* ROTATE_LEFT( F(b,c,d)+a+x+ac , s )+b
*
* 最后用该结果取代a，b，c或d中之一(a)。
* a=ROTATE_LEFT( F(b,c,d)+a+x+ac , s )+b
*
* ***************************************************/
#define FF(a,b,c,d,x,s,ac) { a += F(b,c,d) + x + ac; a = ROTATE_LEFT(a,s); a += b; }
#define GG(a,b,c,d,x,s,ac) { a += G(b,c,d) + x + ac; a = ROTATE_LEFT(a,s); a += b; }
#define HH(a,b,c,d,x,s,ac) { a += H(b,c,d) + x + ac; a = ROTATE_LEFT(a,s); a += b; }
#define II(a,b,c,d,x,s,ac) { a += I(b,c,d) + x + ac; a = ROTATE_LEFT(a,s); a += b; }
//储存一个MD5 text信息
typedef struct
{
unsigned int count[2];
//记录当前状态，其数据位数
unsigned int state[4];
//4个数，一共32位 记录用于保存对512bits信息加密的中间结果或者最终结果
unsigned char buffer[64];
//一共64字节，512位
}MD5_CTX;
/**********main***************/
void *deal(void *recv_pack_t);
void *serv_send_thread(void *arg);
void *pthread_check_file(void *arg);
void init_server_pthread();
/************data_access*****************/
int read_infor();
int write_infor();
int conect_mysql_init();
void mysql_save_message(PACK *recv_pack);
void mysql_close();
/************debug*****************/
void print_send_pack();
void print_infor_group();
void print_infor_user();
void print_infor_file();
/************tools*****************/
void my_err(const char * err_string,int line);
int find_userinfor(char username_t[]);
void signal_close(int i);
void send_pack(PACK *pack_send_pack_t);
void send_pack_memcpy_server(int type,char *send_name,char *recv_name,int sockfd1,char *mes);
bool md5_judge(int id,char *log_pass);
int judge_usename_same(char username_t[]);
void del_friend_infor(int id,char friend_name[]) ;
void del_group_from_user(char *username,char *groupname);
int find_groupinfor(char groupname_t[]);
int find_fileinfor(char *filename);
/*****************server*********************/
void login(PACK *recv_pack);
void registe(PACK *recv_pack);
void send_statu(PACK *recv_pack);
void friend_add(PACK *recv_pack);
void friend_del(PACK *recv_pack);
void send_mes_to_one(PACK *recv_pack);
void group_create(PACK *recv_pack);
void group_join(PACK *recv_pack);
void group_qiut(PACK *recv_pack);
void group_del_one(int id);
void group_del(PACK *recv_pack);
void send_mes_to_group(PACK *recv_pack);
void file_recv_begin(PACK *recv_pack);
void file_recv(PACK *recv_pack);
void *file_send_send(void *file_send_begin_t);
void file_send_begin(PACK *recv_pack);
void file_send_finish(PACK *recv_pack);
void send_record(PACK *recv_pack);
/**************md5*********************/
void MD5Init(MD5_CTX *context);
void MD5Update(MD5_CTX *context,unsigned char *input,unsigned int inputlen);
void MD5Final(MD5_CTX *context,unsigned char digest[16]);
void MD5Transform(unsigned int state[4],unsigned char block[64]);
void MD5Encode(unsigned char *output,unsigned int *input,unsigned int len);
void MD5Decode(unsigned int *output,unsigned char *input,unsigned int len);
int md5(char encrypt_input[10000],unsigned int decrypt[4]);
extern INFO_USER m_infor_user [USER_MAX] ;
extern int m_user_num;
/**************************************************/
extern INFO_GROUP m_infor_group [GROUP_MAX] ; //begin from 1
extern int m_group_num;
/******************file_infor************************************/
extern INFO_FILE m_infor_file [FILE_MAX]; //begin from 1
extern int m_file_num;
extern PACK m_pack_send [MAX_CHAR*2];
extern int m_send_num;
extern unsigned char PADDING[];
extern int listenfd,epollfd;//
extern short PORT;
extern int log_file_fd;
extern int user_infor_fd;
extern int group_infor_fd;
extern pthread_mutex_t mutex;
extern pthread_mutex_t mutex_recv_file;
extern pthread_mutex_t mutex_check_file;
/^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^/
Makefile
CC=g++
OBJ = happychat_server
all:
$(CC) -o $(OBJ) *.h *.cpp -lpthread -I/usr/include/mysql/ -L/usr/lib/mysql -lmysqlclient -g
# $(CC) -o $(OBJ) ./*.o -I `pkg-config --libs --cflags gtk+-2.0` -pthread -g
# rm -rf *.o
clean:
-rm -rf *.o