自写聊天室_LinuxC实现(0)——初步代码实现
happychat
第一次写聊天室，就给他取了个名字：happychat
转载请注明出处：http://blog.csdn.net/chudongfang2015/article/details/52250340
当前运行测试环境：unbuntu 16.04
用到的工具： mysql数据库 线程库
实现功能：
1.好友管理：
(1)添加好友
(2)删除好友
2.群管理
(1)创建群
(2)加群
(3)退群
(4)解散群
3.文件传送
实现了上传和下载的断点续传
4.聊天界面分屏
利用光标的移动对输入和屏幕聊天记录输出的分屏
5.MD5加密
利用MD5算法，对用户的密码进行加密，对原来的MD5加密进稍微的改动，把字符串类型加密到unsigned int 类型
6.离线传输
对私聊,群聊，传送文件都添加了离线功能，对方上线即时传送
7.即时状态显示
好友列表会及时更新，并用彩色显示当前好友的状态
8.数据库
利用mysql C语言的API函数，应用mysql数据库对聊天记录进行存储，保证聊天记录的有效性
9.多路复用
epoll实现服务端的多路复用，保证服务端连接客户端数据的稳定性
10.容错处理
对于输入内容进行判别，防止在不正规的操作下程序崩溃。
运行所需环境：
1.mysql数据库安装
并且在数据库内有对应的Database 和 Table
安装数据库(unbuntu)：
sudo apt-get install mysql-server mysql-client // 安装数据库
sudo apt-get instal libmysqlclient-dev //安装数据库的包 C语言 API以root用户登陆数据库，并执行以下代码：
CREATE DATABASE happychat;
use happychat
CREATE TABLE message_tbl(send_name varchar(1024),recv_name varchar(1024),mes varchar(2048),time varchar (1024));
这就是大体的功能，先实现了一个Demo ，我会及时对此进行优化，并对每一个模块对进行相应的分析与解释！
先上代码(该代码在ubuntu本地测试通过)：
server.cpp:
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
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 30
#define MAX_CHAR 1024
#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;//
char* IP = "127.0.0.1";//服务器IP
short PORT = 10222;//端口号
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;idata.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 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;idata.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;kdata.mes[k] == -1)
break;
int t1 = 1;
for(int l=0 ;ldata.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)
{
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 ;idata.mes[i] == -1)
break;
int t1 = 1;
for(int l=0;ldata.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 ;idata.mes[i] == -1)
break;
int t1 = 1;
for(int l=0;ldata.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 = inet_addr(IP);
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; icount[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;
}
server对应的Makefile:
CC=gcc
OBJ = happychat_server
all:
$(CC) -o $(OBJ) *.cpp -lpthread -I/usr/include/mysql/ -L/usr/lib/mysql -lmysqlclient `pkg-config --libs --cflags gtk+-2.0` -g
clean:
-rm -rf *.o
/***************************************************************************************************************************/
client.cpp:
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
/*****************same with server**********************/
#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 SIZE_PASS_NAME 30
#define MAX_PACK_CONTIAN 100
#define MAX_CHAR 1024
#define NUM_MAX_DIGIT 10
#define DOWNLINE 0
#define ONLINE 1
#define BUZY 2
/**************************************************/
typedef struct friend_info
{
int statu;
int mes_num;
char name[MAX_CHAR];
}FRIEND_INFO;
typedef struct user_infor{
char username [MAX_CHAR];
FRIEND_INFO friends [MAX_CHAR];
int friends_num;
char group [MAX_CHAR][MAX_CHAR];
int group_num;
}USER_INFOR;
USER_INFOR m_my_infor;
/******************be sure same with server*******************************/
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;
typedef struct pthread_parameter
{
int a;
int b;
}PTHREAD_PAR;
typedef struct prinit_mes
{
char name[MAX_CHAR];
char time[MAX_CHAR];
char mes [MAX_CHAR];
}PRINT_MES;
/************************客户端缓冲区**********************/
/*PACK m_pack_send_friend_see [MAX_CHAR];
PACK m_pack_send_chat_one [MAX_CHAR];
PACK m_pack_send_chat_many [MAX_CHAR];
PACK m_pack_send_send_file [MAX_CHAR];
int m_send_num_friend_see;
int m_send_num_chat_one;
int m_send_num_chat_many;
int n_send_num_send_file;*/
/*****************send*********************/
PACK m_pack_send [MAX_CHAR];
int m_send_num;
/*****************recv*********************/
PACK m_pack_recv_friend_see [MAX_PACK_CONTIAN];
PACK m_pack_recv_chat [MAX_PACK_CONTIAN];
PACK m_pack_recv_send_file [MAX_PACK_CONTIAN];
PACK m_pack_recv_file_mes [MAX_PACK_CONTIAN];
PACK m_pack_recv_file [MAX_PACK_CONTIAN];
int m_recv_num_friend_see;
int m_recv_num_chat;
int m_recv_num_send_file;
int m_recv_num_file_mes;
int m_recv_num_file;
int m_print_mes_flag;
/****************************************************/
int m_flag_group_create;
int m_flag_group_join ;
int m_flag_group_del ;
int m_flag_print_mes;
int m_flag_print_recorde;
// int window_col;
// int window_row;
/****************************************************/
PRINT_MES m_print_mes[7];
int m_print_mes_num;
/***********************function***********************/
void print_file_mes();
void my_err(const char * err_string,int line);
void init();
void sig_close(int i);
void send_pack(int type,char *send_name,char *recv_name,char *mes);
void send_pack_memcpy(int type,char *send_name,char *recv_name,char *mes);
int get_choice(char *choice_t);
int send_login(char username_t[],char password_t[]);
int login();
int login_menu();
int send_registe(char username_t[],char password_t[]);
void registe();
void change_statu(PACK pack_deal_statu_t);
void *deal_statu(void *arg);
void send_file_send(int begin_location,char *file_path);
void *pthread_send_file(void *mes_t);
void *pthread_recv_file(void *par_t);
void init_clien_pthread();
void get_status_mes();
void friends_see();
int judge_same_friend(char add_friend_t[]);
void add_friend();
void del_friend();
int judge_same_group(char *group_name);
void group_see();
void send_mes(char mes_recv_name[],int type);
void print_mes(int id);
void *show_mes(void *username);
void send_mes_to_one();
void group_create();
void group_join();
void group_qiut();
void group_del();
void send_mes_to_group();
int get_file_size(char *file_name);
void send_file();
void file_infor_delete(int id);
void mes_sendfile_fail(int id);
void mes_recv_requir(int id);
void mes_recvfile_fail(int id);
void deal_file_mes(int id);
int file_mes_box();
void print_mes_record(PACK pack_t);
int mes_record();
int main_menu();
int sockfd;
char *IP = "127.0.0.1";
short PORT = 10222;
typedef struct sockaddr SA;
pthread_mutex_t mutex_local_user;
pthread_mutex_t mutex_recv_file;
void print_file_mes()
{
for(int i=1 ;i<=5;i++)
{
printf("hahahahahaha*****\n");
printf("%s\n", m_pack_recv_file_mes[i].data.send_name);
printf("%s",m_pack_recv_file_mes[i].data.recv_name);
for(int j=0;j<=5;j++)
printf("%d\n\n",m_pack_recv_file_mes[i].data.mes[j]);
}
}
void my_err(const char * err_string,int line)
{
fprintf(stderr, "line:%d ", line);
perror(err_string);
exit(1);
}
void init()
{
printf("客户端开始启动\n");
sockfd = socket(AF_INET,SOCK_STREAM,0);//启动socket
struct sockaddr_in addr;
addr.sin_family = AF_INET;
addr.sin_port = htons(PORT);
addr.sin_addr.s_addr = inet_addr(IP);
if(connect(sockfd,(SA*)&addr,sizeof(addr))==-1){
perror("无法连接到服务器");
printf("客户端启动失败\n");
exit(-1);
}
printf("客户端启动成功\n");
}
void sig_close(int i)
{
//关闭客户端的描述符
close(sockfd);
exit(0);
}
/*************************tools******************************/
void send_pack(int type,char *send_name,char *recv_name,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);
strcpy(pack_send_pack.data.mes,mes);
time(&timep);
pack_send_pack.data.time = timep;
if(send(sockfd,&pack_send_pack,sizeof(PACK),0) < 0){
my_err("send",__LINE__);
}
}
void send_pack_memcpy(int type,char *send_name,char *recv_name,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;
if(send(sockfd,&pack_send_pack,sizeof(PACK),0) < 0){
my_err("send",__LINE__);
}
}
int get_choice(char *choice_t)
{
int choice =0;
for(int i=0;i'9')
return -1;
for(int i=0;i 0)
{
sum += length;
printf("length = %d\n", length);
int digit = 0;
while(length != 0)
{
mes[digit++] = length%10;
length /= 10;
}
mes[digit] = -1;
printf("have sended : %d%% %d \n",(int)((double)sum/file_size*100),sum);
printf("%s\n", file_path);
send_pack_memcpy(FILE_SEND,m_my_infor.username,file_path,mes);
if(sum == file_size)
break;
bzero(mes, MAX_CHAR*2);
usleep(100000);
}
// 关闭文件
close(fd);
printf("send finished!!\n");
printf("\n\t\t*********************************\n");
printf("\t\t* 1.show friends *\n");
printf("\t\t* 2.add friends *\n");
printf("\t\t* 3.delete friends *\n");
printf("\t\t* 4.show group *\n");
printf("\t\t* 5.create group *\n");
printf("\t\t* 6.join group *\n");
printf("\t\t* 7.quit group *\n");
printf("\t\t* 8.delete group *\n");
printf("\t\t* 9.chat with one *\n");
printf("\t\t* 10.chat with many *\n");
printf("\t\t* 11.send file *\n");
printf("\t\t* 12.file message box %d *\n",m_recv_num_file_mes);
printf("\t\t* 13.mes recording *\n");
printf("\t\t* 0.exit *\n");
printf("\t\t******************************* *\n");
printf("\t\tchoice：");
}
void *pthread_send_file(void *mes_t)
{
char *mes = (char *)mes_t;
int begin_location = 0;
char file_name[MAX_CHAR];
printf("\033[;31m \nfunction:pthread_send_file\033[0m \n");
for(int i=0 ;ia ;
int begin_location_server = pthread_par->b;
int sum = begin_location_server;
while(1)
{
pthread_mutex_lock(&mutex_recv_file);
int fd;
char file_name[MAX_CHAR];
for(int i=1;i<=m_recv_num_file ;i++)
{
int len = 0;
for(int j=0 ;j= file_size)
{
send_pack(FILE_FINI_RP,m_my_infor.username,"server",file_name);
printf("send finished!!\n");
printf("\n\t\t*********************************\n");
printf("\t\t* 1.show friends *\n");
printf("\t\t* 2.add friends *\n");
printf("\t\t* 3.delete friends *\n");
printf("\t\t* 4.show group *\n");
printf("\t\t* 5.create group *\n");
printf("\t\t* 6.join group *\n");
printf("\t\t* 7.quit group *\n");
printf("\t\t* 8.delete group *\n");
printf("\t\t* 9.chat with one *\n");
printf("\t\t* 10.chat with many *\n");
printf("\t\t* 11.send file *\n");
printf("\t\t* 12.file message box %d *\n",m_recv_num_file_mes);
printf("\t\t* 13.mes recording *\n");
printf("\t\t* 0.exit *\n");
printf("\t\t******************************* *\n");
printf("\t\tchoice：");
return NULL;
}
}
pthread_mutex_unlock(&mutex_recv_file);
usleep(10);
}
}
void *clien_recv_thread(void *arg)
{
int i;
PACK pack_t;
pthread_t pid_send_file;
while(1)
{
if(recv(sockfd,&pack_t,sizeof(PACK),0) < 0){
my_err("recv",__LINE__);
}
//for(int i=1;i <= )
pthread_mutex_lock(&mutex_local_user);
for(int i=1 ;i<= m_my_infor.friends_num;i++)
{
if(strcmp(m_my_infor.friends[i].name,pack_t.data.send_name) == 0)
{
m_my_infor.friends[i].mes_num++;
break;
}
}
switch(pack_t.type)
{
printf("clien_recv_thread:%d\n", pack_t.type);
case FRIEND_SEE:
m_pack_recv_friend_see[++m_recv_num_friend_see] = pack_t;
break;
case GROUP_CREATE:
m_flag_group_create = pack_t.data.mes[0];
break;
case GROUP_JOIN:
m_flag_group_join = pack_t.data.mes[0];
break;
case GROUP_DEL:
m_flag_group_del = pack_t.data.mes[0];
case CHAT_ONE:
m_pack_recv_chat[++m_recv_num_chat] = pack_t;
break;
case CHAT_MANY:
m_pack_recv_chat[++m_recv_num_chat] = pack_t;
break;
//case SEND_FILE:
// m_pack_recv_send_file[++m_recv_num_send_file] = pack_t;
//break;
case FILE_SEND_BEGIN_RP:
pthread_create(&pid_send_file,NULL,pthread_send_file,(void *)pack_t.data.mes);
break;
case FILE_SEND_STOP_RP:
m_pack_recv_file_mes[++m_recv_num_file_mes] = pack_t;
break;
case FILE_RECV_BEGIN:
m_pack_recv_file_mes[++m_recv_num_file_mes] = pack_t;
break;
case FILE_RECV:
pthread_mutex_lock(&mutex_recv_file);
m_pack_recv_file[++m_recv_num_file] = pack_t;
printf("FILE_RECV:%d\n", m_recv_num_file);
pthread_mutex_unlock(&mutex_recv_file);
break;
case FILE_RECV_STOP_RP:
m_pack_recv_file_mes[++m_recv_num_file_mes] = pack_t;
break;
case MES_RECORD:
print_mes_record(pack_t);
break;
}
pthread_mutex_unlock(&mutex_local_user);
usleep(1);
}
}
void init_clien_pthread()
{
pthread_t pid_deal_statu,pid_recv,pid_recv_file;
pthread_create(&pid_deal_statu,NULL,deal_statu,NULL);
pthread_create(&pid_recv,NULL,clien_recv_thread,NULL);
}
/********************friend***************************************/
void get_status_mes()
{
PACK pack_friend_see;
pack_friend_see.type = FRIEND_SEE;
strcpy(pack_friend_see.data.send_name,m_my_infor.username);
strcpy(pack_friend_see.data.recv_name,"server");
memset(pack_friend_see.data.mes,0,sizeof(pack_friend_see.data.mes));
if(send(sockfd,&pack_friend_see,sizeof(PACK),0) < 0){
my_err("send",__LINE__);
}
}
void friends_see()
{
pthread_mutex_lock(&mutex_local_user);
printf("friends num:%d\n", m_my_infor.friends_num);
printf("\n\nthe list of friends:\n");
for(int i=1 ;i<=m_my_infor.friends_num ;i++)
{
printf("m_my_infor.friends[i].statu:%d\n", m_my_infor.friends[i].statu);
switch(m_my_infor.friends[i].statu)
{
case ONLINE:
printf("ID:%d \033[;32m%s\033[0m[ONLINE] ", i,m_my_infor.friends[i].name);
if(m_my_infor.friends[i].mes_num)
printf("\033[;33m%d messages\033[0m\n", m_my_infor.friends[i].mes_num);
else
printf("\n");
break;
case DOWNLINE:
printf("ID:%d \033[;31m%s\033[0m[DOWNLINE] ", i,m_my_infor.friends[i].name);
if(m_my_infor.friends[i].mes_num)
printf("\033[;33m%d messages\033[0m\n", m_my_infor.friends[i].mes_num);
else
printf("\n");
break;
}
}
pthread_mutex_unlock(&mutex_local_user);
}
int judge_same_friend(char add_friend_t[])
{
int i;
for(i=1;i<=m_my_infor.friends_num;i++)
{
if(strcmp(m_my_infor.friends[i].name,add_friend_t) == 0)
return i;
}
return 0;
}
void add_friend()
{
char add_friend_t[MAX_CHAR];
printf("please input the name of friend you want to add:\n");
scanf("%s",add_friend_t);
if(strcmp(add_friend_t,m_my_infor.username) == 0)
{
printf("you can't add youself to be your friend!\n");
return;
}
if(judge_same_friend(add_friend_t))
{
printf("you already have same friends!\n");
return ;
}
printf("m_my_infor.username:%s\n", m_my_infor.username);
send_pack(FRIEND_ADD,m_my_infor.username,"server",add_friend_t);
get_status_mes();
}
void del_friend()
{
char del_friend_t[MAX_CHAR];
printf("please input the name of friend you want to delete:\n");
scanf("%s",del_friend_t);
if(!judge_same_friend(del_friend_t))
{
printf("you don't have this friends on list!\n");
return ;
}
printf("m_my_infor.username:%s\n", m_my_infor.username);
send_pack(FRIEND_DEL,m_my_infor.username,"server",del_friend_t);
get_status_mes();
}
/*************************end_friend************************************/
/***********************group***********************************/
void show_mes_smart(char *name ,char *time ,char *mes)
{
int number = 4;
if(m_print_mes_num == number)
{
for(int i=1;i<=3 ;i++)
m_print_mes[i] = m_print_mes[i+1];
strcpy(m_print_mes[number].name,name);
strcpy(m_print_mes[number].time,time);
strcpy(m_print_mes[number].mes,mes);
}
else{
strcpy(m_print_mes[++m_print_mes_num].name,name);
strcpy(m_print_mes[m_print_mes_num].time,time);
strcpy(m_print_mes[m_print_mes_num].mes,mes);
}
printf("\33[s");
fflush(stdout);
printf("\33[25A\n");
for(int i=1;i<=12;i++)
printf("\33[K\n");
printf("\33[13A\n");
for(int i=1;i<=m_print_mes_num;i++)
{
printf("\033[40;42m%s\033[0m\t%s",m_print_mes[i].name,m_print_mes[i].time);
printf("%s\n", m_print_mes[i].mes);
}
for(int i=1;i<=4- m_print_mes_num ;i++)
{
printf("\n");
printf("\n\n");
}
// printf("\33[11B\n");
printf("\33[u");
fflush(stdout);
}
int judge_same_group(char *group_name)
{
int i;
for(i=1;i<=m_my_infor.group_num;i++)
{
if(strcmp(m_my_infor.group[i],group_name) == 0)
return 1;
}
return 0;
}
void group_see()
{
pthread_mutex_lock(&mutex_local_user);
printf("\n\nthe list of group:\n");
for(int i=1 ;i<=m_my_infor.group_num ;i++)
{
printf("ID:%d %s\n", i,m_my_infor.group[i]);
}
pthread_mutex_unlock(&mutex_local_user);
}
void send_mes(char mes_recv_name[],int type)
{
PACK pack_send_mes;
char mes[MAX_CHAR];
time_t timep;
getchar();
printf("******************please input****************************\n");
while(1)
{
time(&timep);
memset(mes,0,sizeof(mes));
//if(type == CHAT_ONE)
// printf("%s->",m_my_infor.username);
fgets(mes,MAX_CHAR,stdin);
while(mes[0] == 10)
{
printf("\33[1A");
fflush(stdout);
fgets(mes,MAX_CHAR,stdin);
}
if(strcmp(mes,"quit\n") == 0)
break;
printf("\33[1A");
printf("\33[K");
fflush(stdout);
show_mes_smart(m_my_infor.username ,ctime(&timep) ,mes);
//printf("\t%s\n%s\n", m_my_infor.username,ctime(&timep),mes);
send_pack(type,m_my_infor.username,mes_recv_name,mes);
}
m_flag_print_mes = EXIT;
}
void print_mes(int id)
{
char group_print_name[MAX_CHAR];
memset(group_print_name,0,sizeof(group_print_name));
if(m_pack_recv_chat[id].type == CHAT_ONE)
{
show_mes_smart(m_pack_recv_chat[id].data.send_name,ctime(&m_pack_recv_chat[id].data.time),m_pack_recv_chat[id].data.mes);
}
else
{
for(int i=0;ia = file_size;
par_t->b = len;
int digit = 0;
while(len != 0)
{
mes_t[digit++] = len%10;
len /= 10;
}
mes_t[digit] = -1;
//printf("file_name:%s\n", file_name);
send_pack_memcpy(FILE_SEND_BEGIN_RP ,m_my_infor.username ,file_name ,mes_t);
pthread_create(&pid_recv_file,NULL,pthread_recv_file,(void *)par_t);
/***************************************************/
file_infor_delete(id);
/*************************************************/
}
void mes_recvfile_fail(int id)
{
pthread_t pid_recv_file;
char chioce[10];
int begin_location_server;
int file_size;
char file_name[SIZE_PASS_NAME];
char mes_t[MAX_CHAR];
PTHREAD_PAR * par_t = (PTHREAD_PAR *)malloc(sizeof(PTHREAD_PAR));
for(int i=0 ;ia = file_size;
par_t->b = begin_location_server;
printf("the file %s recv failed ,have recved %d%%,do you want recv continue?\n", file_name,(int)((double)begin_location_server/file_size*100));
printf("y/n :");
scanf("%s",chioce);
if(chioce[0] != 'Y' && chioce[0] != 'y')
{
/************************************/
file_infor_delete(id);
/************************************/
return ;
}
printf("&&&&&&&\nbegin_location :%d\n",begin_location_server);
int len = begin_location_server;
int digit = 0;
while(len != 0)
{
mes_t[digit++] = len%10;
len /= 10;
}
mes_t[digit] = -1;
send_pack_memcpy(FILE_SEND_BEGIN_RP ,m_my_infor.username ,file_name ,mes_t);
pthread_create(&pid_recv_file,NULL,pthread_recv_file,(void *)par_t);
/*****************************************************/
file_infor_delete(id);
/*******************************************************/
}
void deal_file_mes(int id)
{
if(m_pack_recv_file_mes[id].type == FILE_SEND_STOP_RP)
{
mes_sendfile_fail(id);
}
else if(m_pack_recv_file_mes[id].type == FILE_RECV_BEGIN)
{
mes_recv_requir(id);
}else if(m_pack_recv_file_mes[id].type == FILE_RECV_STOP_RP)
{
mes_recvfile_fail(id);
}
}
int file_mes_box()
{
char choice_t[100];
int chioce;
do
{
get_status_mes();
printf("pack num_chat:%d\n", m_recv_num_chat);
printf("\n\t\t************file mes box *************\n");
for(int i = 1; i <= m_recv_num_file_mes;i++)
{
if(m_pack_recv_file_mes[i].type == FILE_SEND_STOP_RP)
printf("\t\t* send file %s filed *\n",m_pack_recv_file_mes[i].data.send_name);
if(m_pack_recv_file_mes[i].type == FILE_RECV_BEGIN)
printf("\t\t* %s send file %s to you *\n", m_pack_recv_file_mes[i].data.send_name,m_pack_recv_file_mes[i].data.mes+SIZE_PASS_NAME);
if(m_pack_recv_file_mes[i].type == FILE_RECV_STOP_RP)
printf("\t\t* recv file %s filed *\n", m_pack_recv_file_mes[i].data.mes+NUM_MAX_DIGIT);
}
printf("\t\t* 0.exit *\n");
printf("\t\t******************************* *\n");
printf("\t\tchoice：");
scanf("%s",choice_t);
chioce = get_choice(choice_t);
if(chioce != -1)
deal_file_mes(chioce);
}while(chioce!=0);
return 0;
}
void print_mes_record(PACK pack_t)
{
if(strcmp(pack_t.data.send_name , "server") == 0)
{
printf("\033[40;42m%s\033[0m\t%s",pack_t.data.mes,pack_t.data.mes+1*SIZE_PASS_NAME);
printf("%s\n", pack_t.data.mes+2*SIZE_PASS_NAME);
}
else
m_flag_print_recorde = 1;
}
int mes_record()
{
char username[MAX_CHAR];
printf("please input the username that you want see:\n");
scanf("%s",username);
printf("the recording :\n");
send_pack(MES_RECORD,m_my_infor.username,"server",username);
printf("\33[2J\33[20A********message recordes**************\n");
while(!m_flag_print_recorde) ;
printf("printf recordes finished!!\n");
m_flag_print_recorde = 0;
return 0;
}
int main_menu()
{
char choice_t[100];
int chioce;
do
{
get_status_mes();
//printf("pack num_chat:%d\n", m_recv_num_chat);
printf("\n\t\t*******************************\n");
printf("\t\t* 1.show friends *\n");
printf("\t\t* 2.add friends *\n");
printf("\t\t* 3.delete friends *\n");
printf("\t\t* 4.show group *\n");
printf("\t\t* 5.create group *\n");
printf("\t\t* 6.join group *\n");
printf("\t\t* 7.quit group *\n");
printf("\t\t* 8.delete group *\n");
printf("\t\t* 9.chat with one *\n");
printf("\t\t* 10.chat with many *\n");
printf("\t\t* 11.send file *\n");
printf("\t\t* 12.file message box %d *\n",m_recv_num_file_mes);
printf("\t\t* 13.mes recording *\n");
printf("\t\t* 0.exit *\n");
printf("\t\t******************************* *\n");
printf("\t\tchoice：");
scanf("%s",choice_t);
chioce = get_choice(choice_t);
switch(chioce)
{
case 1:
friends_see();
break;
case 2:
add_friend();
break;
case 3:
del_friend();
break;
case 4:
group_see();
break;
case 5:
group_create();
break;
case 6:
group_join();
break;
case 7:
group_qiut();
break;
case 8:
group_del();
break;
case 9:
send_mes_to_one();
break;
case 10:
send_mes_to_group();
break;
case 11:
send_file();
break;
case 12:
file_mes_box();
break;
case 13:
mes_record();
default:
break;
}
}while(chioce!=0);
return 0;
}
int main(int argc, char const *argv[])
{
int flag =0;
// struct winsize size;
signal(SIGINT,sig_close);//关闭CTRL+C
init();//启动并连接服务器
// ioctl(STDIN_FILENO,TIOCGWINSZ,&size); //get size of window
// window_col = size.ws_col;
// window_row = size.ws_row;
if(login_menu() == 0)
return 0;
init_clien_pthread();
main_menu();
return 0;
}
client对应的Makefile:
CC=g++ -std=c++11
OBJ = happychat_client
all:
$(CC) -o $(OBJ) -cpp ./*.cpp `pkg-config --libs --cflags gtk+-2.0` -pthread -g
clean:
-rm -rf *.o