您的位置:首页 > 运维架构 > Linux

自写聊天室_LinuxC实现(1)——项目部署遇到问题及解决方法

2016-08-30 17:58 931 查看
我为了保证该聊天室可以在公网上运行,我把服务端部署到了腾讯云上,这之间出了许多问题,正逐一解决:

服务器配置:

操作系统	CentOS 6.5 32位
CPU	1核
内存	1GB
系统盘	20GB(本地磁盘)
公网带宽	1Mbps




本机配置:

操作系统	Ubuntu 16.04 64位
CPU	8核
内存	8GB
系统盘	80GB(本地磁盘)


1.IP地址绑定问题:

服务端只需自动绑定当前IP即可,而客户端则绑定腾讯云的公网IP

这里把代码做了修改:

addr.sin_addr.s_addr = htonl(INADDR_ANY);


2.包传递问题
由于该聊天室全部采用包传递的方式,利用结构体进行传递,而结构体内容如下:

typedef struct datas{
char    send_name[MAX_CHAR];
char    recv_name[MAX_CHAR];
int     send_fd;
int     recv_fd;
time_t  time;
char    mes[MAX_CHAR*2];
}DATA;

typedef struct package{
int type;
DATA  data;
}PACK;


这里recv()和send()都传输sizeof(PACK)个字符 

问题:在传输时发现传输总是乱码

原因:在32位操作系统和64位操作系统下long 的字节数不相同,导致sizeof(PACK)不同,导致乱码(time_t实质就是long类型)

解决:后来我把time_t这一项去掉,得到正确结果

注意:32位机和64位机 字节数的区别

总结如下:

常用数据类型对应字节数

  可用如sizeof(char),sizeof(char*)等得出

 32位编译器:

      char :1个字节

      char*(即指针变量): 4个字节(32位的寻址空间是2^32, 即32个bit,也就是4个字节。同理64位编译器)

      short int : 2个字节

      int:  4个字节

      unsigned int : 4个字节

      float:  4个字节

      double:   8个字节

      long:   4个字节

      long long:  8个字节

      unsigned long:  4个字节

  64位编译器:

      char :1个字节

      char*(即指针变量): 8个字节

      short int : 2个字节

      int:  4个字节

      unsigned int : 4个字节

      float:  4个字节

      double:   8个字节

      long:   8个字节

      long long:  8个字节

      unsigned long:  8个字节

以上占用字节数其实是针对c/c++语言而言的,对于java来说由于其JVM具有跨平台性因此java在32位和64位机下基本数据类型占字节数是一致的(这样才能达到跨平台通信)。

3.包的重复接受问题

客户端发送一个包,客户端会接受到多个包。

问题原因:在使用epoll多路复用时,使用了while循环,在一个包接受完还没来到及清除缓冲区的情况下,进入第二次循环,就导致了一个包被recv多次,这里猜测,epoll判断客户端有消息是根据缓冲区是否为空判断的。

解决方法:这里我想,没有及时清除缓冲区,那我就在循环中加上usleep,等待其清空完,结果成功解决。  但由于这样太耗费时间,我猜想是因为包太大的缘故,导致清空速达减慢,后来我把包空间减小,果然解决了问题。

注意:在使用epoll时一定要注意其是否即使清空缓冲区,避免多次接受。

4.创建线程过多:

问题:在传输文件时,由于每接受一个包就要创建一个线程进行处理,所以这里会出现线程创建过多的情况。

解决方法:[b]为了扩大最大创建线程量,我使用了[b]ulimit -s 命令扩大了线程最多创建数。[/b][/b]



/××××××××××××××××××××××××××××××××××××××××××××××××××××××××××××××××××××××××××××××××××××××××××××/

在网上查了下,原因如下,Linux系统中每个线程都拥有独立的栈空间,而我的系统上调用ulimit -a看到的结果如下:

ulimit -a

core file size          (blocks, -c) 0

data seg size           (kbytes, -d) unlimited

scheduling priority             (-e) 20

file size               (blocks, -f) unlimited

pending signals                 (-i) 16382

max locked memory       (kbytes, -l) 64

max memory size         (kbytes, -m) unlimited

open files                      (-n) 1024

pipe size            (512 bytes, -p) 8

POSIX message queues     (bytes, -q) 819200

real-time priority              (-r) 0
stack size              (kbytes, -s) 8192

cpu time               (seconds, -t) unlimited

max user processes              (-u) unlimited

virtual memory          (kbytes, -v) unlimited

file locks                      (-x) unlimited

可以看到stack size是8M,  400个线程就需要8*400=3200M,虚拟内存不够用。

解决办法有两种:

1.使用ulimit -s 1024*1024命令,将线程栈大小临时设置成1M,经过试验能同时创建2000个线程了。

2.使用pthread_attr_setstacksize在程序中改变线程栈大小。
/××××××××××××××××××××××××××××××××××××××××××××××××××××××××××××××××××××××××××××××××××/




这里给出server.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

/********************user infor***********************************/
typedef struct infor_user
{
char username[MAX_CHAR];
unsigned int password[4];
int statu;//don't foget to change is to 0 when the server begin
int socket_id;
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;

INFO_USER m_infor_user [USER_MAX];
int m_user_num;
/**************************************************/

/*********************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;

INFO_GROUP m_infor_group [GROUP_MAX]; //begin from 1
int m_group_num;

/******************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;

INFO_FILE m_infor_file [FILE_MAX]; //begin from 1
int m_file_num;

/*********************************************************/

typedef struct datas{
char send_name[MAX_CHAR];
char recv_name[MAX_CHAR];
int send_fd;
int recv_fd;
//time_t time;
char mes[MAX_CHAR*2];
}DATA;

typedef struct package{
int type;
DATA data;
}PACK;

/*****************send*********************/
PACK m_pack_send [MAX_CHAR*2];
int m_send_num;

/**************************************/

typedef struct pthread_parameter
{
int a;
char str1[SIZE_PASS_NAME];
char str2[SIZE_PASS_NAME];
}PTHREAD_PAR;

/*********************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;

//第一位1 其后若干个0,用于MD5Final函数时的补足
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};

/***********************************************/

/**********function***************/
void my_err(const char * err_string,int line);
void signal_close(int i);
void send_pack(PACK *pack_send_pack_t);
void print_send_pack();
void print_infor_group();
void print_infor_user();
void print_infor_file();
int find_userinfor(char username_t[]);
void login(PACK *recv_pack);
int judge_usename_same(char username_t[]);
void registe(PACK *recv_pack);
void send_statu(PACK *recv_pack);
void friend_add(PACK *recv_pack);
void del_friend_infor(int id,char friend_name[]);
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 del_group_from_user(char *username,char *groupname);
void group_qiut(PACK *recv_pack);
void group_del_one(int id);
void group_del(PACK *recv_pack);
int find_groupinfor(char groupname_t[]);
void send_mes_to_group(PACK *recv_pack);
int find_fileinfor(char *filename);
void file_recv_begin(PACK *recv_pack);
void file_recv(PACK *recv_pack);
void *pthread_check_file(void *arg);
void send_pack_memcpy_server(int type,char *send_name,char *recv_name,int sockfd1,char *mes);
void *file_send_send(void *file_send_begin_t);
void file_send_begin(PACK *recv_pack);
void file_send_finish(PACK *recv_pack);
void *deal(void *recv_pack_t);
void *serv_send_thread(void *arg);
void init_server_pthread();
int read_infor();
int write_infor();

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]);

int listenfd,epollfd;//
short PORT = 11111;//端口号
typedef struct sockaddr SA;//类型转换
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;

/********************mysql*************************/
MYSQL mysql;
MYSQL_RES *res = NULL;
MYSQL_ROW row;
char query_str[MAX_CHAR*4] ;
int rc, fields;
int rows;

void my_err(const char * err_string,int line)
{
fprintf(stderr, "line:%d ", line);
perror(err_string);
exit(1);
}

void signal_close(int i)
{
//关闭服务器前 关闭服务器的socket
#ifdef LOG
close(log_file_fd);
#else
#endif

write_infor();
mysql_free_result(res);
mysql_close(&mysql);
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);
}

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");
}

/***************************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);
}

/**********************login*******************************************/

/*void set_user_online(char username_t[])
{
int i;
for(i=1;i<=m_send_num;i++)
{
if(strcmp(m_infor_user[i].username,username_t) == 0)
{
m_infor_user[i].statu = ONLINE;
return ;
}
}
}*/

/***********************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");
// query_str = "insert into runoob_tbl values (1,'gaga' ,'justtest', '2015-5-5')"; //插入 SQL 语句
// rc = mysql_real_query(&mysql, query_str, strlen(query_str)); //对数据库执行 SQL 语句
// if (0 != rc) {
// printf("mysql_real_query(): %s\n", mysql_error(&mysql));
// return -1;
// }

/*query_str = "delete from runoob_tbl where runoob_id=2"; //删除 SQL 语句
rc = mysql_real_query(&mysql, query_str, strlen(query_str)); //对数据库 执行 SQL 语句
if (0 != rc) {
printf("mysql_real_query(): %s\n", mysql_error(&mysql));
return -1;
}*/

/*query_str = "select * from runoob_tbl"; //显示 表中数据
rc = mysql_real_query(&mysql, query_str, strlen(query_str));
if (0 != rc) {
printf("mysql_real_query(): %s\n", mysql_error(&mysql));
return -1;
}*/

/* res = mysql_store_result(&mysql); //获取查询结果
if (NULL == res) {
printf("mysql_restore_result(): %s\n", mysql_error(&mysql));
return -1;
}*/

/*rows = mysql_num_rows(res);
printf("The total rows is: %d\n", rows);

fields = mysql_num_fields(res);
printf("The total fields is: %d\n", fields);

while ((row = mysql_fetch_row(res))) {
for (i = 0; i < fields; i++) {
printf("%s\t", row[i]);
}
printf("\n");
}

mysql_free_result(res);
*/
}

/*************************************************************/
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;
}

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;
}

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);
}
/**********************************************************/

/********************registe********************************/
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;
}

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);

}
/**********************************************************/

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);
/*for(int i = 1;i <= m_user_num ;i++)
{
if(strcmp(recv_pack->data.send_name,m_infor_user[i].username) == 0)
{
id = i;
break;
}
}*/

/*for(int i=1;i<=m_user_num;i++)
printf("send_statu:%d,%s\n", i,m_infor_user[i].username);
printf("send_statu:id=%d\n", id);*/

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]);
printf("send_statufriend:%d %s\n",i,name_t);
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);

//printf("friend add m_infor_user[id].friends_num:%d\n", m_infor_user[id].friends_num);
//printf("friend add m_infor_user[id].friends_num:%s\n", m_infor_user[id].friends[m_infor_user[id].friends_num]);
free(recv_pack);
}

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 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);
//printf("friend add m_infor_user[id].friends_num:%d\n", m_infor_user[id].friends_num);
//printf("friend add m_infor_user[id].friends_num:%s\n", m_infor_user[id].friends[m_infor_user[id].friends_num]);
free(recv_pack);

}

void send_mes_to_one(PACK *recv_pack)
{
// #ifdef MYSQL
/* sprintf(query_str, "insert into message_tbl values ('%s','%s' ,'%s','%s')",recv_pack->data.send_name,recv_pack->data.recv_name,recv_pack->data.mes,ctime(&recv_pack->data.time));
rc = mysql_real_query(&mysql, query_str, strlen(query_str));
if (0 != rc) {
printf("mysql_real_query(): %s\n", mysql_error(&mysql));
return ;
}
printf("message get into the mysql!!\n");

// #else
// #endif
*/
send_pack(recv_pack);
free(recv_pack);
}

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 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--;
}
}
}

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();
}
}
}
}
}

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);
//print_infor_group();
free(recv_pack);

}

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;
}

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)
{
// #ifdef MYSQL
/* sprintf(query_str, "insert into message_tbl values ('%s','%s' ,'%s','%s')",recv_pack->data.send_name,recv_pack->data.recv_name,recv_pack->data.mes,ctime(&recv_pack->data.time));
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
*/
send_pack(recv_pack);
}

}

free(recv_pack);
}

/*$************************************************************/

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;
}

/*int get_file_size(char *file_name)
{
FILE *fp = fopen(filename, "r");
fseek(fp, 0, SEEK_END);
file_size=ftell(fp);//得到文件大小

// printf("file_size=%d",file_size);

send(sockfd,&file_size,sizeof(int),0);// 把文件大小数据发送
fseek(fp, 0, SEEK_SET);
close(fp);
}*/

void file_recv_begin(PACK *recv_pack)
{
int flag = 0;
int i;
int file_size_now_t;
printf("jajahaha jiji %d\n", m_file_num);

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;
}
printf("file_recv_begin t :%d\n", t);
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);
//print_infor_file();
free(recv_pack);
}

void file_recv(PACK *recv_pack)
{
printf("jijimei?\n");
pthread_mutex_lock(&mutex_recv_file);
printf("jijile\n");
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;

}

//printf("\033[;33mlen = %d\033[0m \n", len);

strcpy(file_name,recv_pack->data.recv_name);
//you can creat this file when you get the file_send_begin
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);

printf("file_name :%s\n", file_name);

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();
}

/**********************check file statu****************************/

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 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__);
}
}

/*********************deal server to client*****************************/
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);

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(100000);
}
// 关闭文件
close(fd);
printf("guanle mei ?/**/\n");
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);
printf("file namehahha%s\n", file_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;
//printf("begin_loclation :%d\n",begin_location);
strcpy(file_send_begin_t->str1,file_name);
strcpy(file_send_begin_t->str2,recv_name);
//printf("file name%s\n", file_name);
file_send_send((void *)file_send_begin_t);
free(recv_pack);

}

void file_send_finish(PACK *recv_pack)
{
printf("^^^^^^^^^^^^^^^^^^^^^^hahah%s\n",recv_pack->data.mes);
int id = find_fileinfor(recv_pack->data.mes);
m_infor_file[id].flag = FILE_STATU_SEND_FINI;
free(recv_pack);
}

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,time 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 ;
}
printf("jijmei\n");
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*2,row[2]);
send_pack(pack_send_record_t);
usleep(10000);

//for (int i = 0; i < 4; i++) {
// printf("%s\t", row[i]);
//}
//printf("\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);

}

/*****************************************************/

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 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

// #ifdef MYSQL
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");

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__);
}

}
}
return 0;
}

/***************************md5******************************************/

/************************
* 函数功能:初始化一个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;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: