码迷,mamicode.com
首页 > 系统相关 > 详细

基于LINUX的多功能聊天室

时间:2015-02-02 09:28:06      阅读:143      评论:0      收藏:0      [点我收藏+]

标签:

原文:基于LINUX的多功能聊天室

基于LINUX的多功能聊天室

其实这个项目在我电脑已经躺了多时,最初写完项目规划后,我就认认真真地去实现了它,后来拿着这个项目区参加了面试,同样面试官也拿这个项目来问我,当然我是做过一遍了,而且为了面试,我将什么strcpy,strlen等最常用的函数都自己实现了一遍,说着,我感觉自己有点挺用功的样子呢!

后来,工作也定下来了,等三方,然后继续帮助我的导师做项目,经过老师的威逼利诱下,我屈服了,又把智能家居系统作为项目,同时也是我的毕业设计,而且功能还要十分完善的说,好吧,别人的简单,但我也无所谓,因为看到技术二字,我必然是两眼放光的那种,特别喜欢那种突然我想到咋么解决了,然后就帮功能实现了那种感觉,像是上瘾了一样,(果然是当程序员的料)

项目回想与思考

嗯,我的项目基本上的功能和项目规划里面基本上是一致的,而且功能也是非常完善了,基本上能够实现了我所需要的功能,但是我也要思考我在整个项目过程中的得失,我在项目的思考中,其实把他定义错了,因为是一个非常小的聊天室的功能,其实我无需使用数据库来作为登陆与否的查询方法,其实完全可以使用链表的方法来做客户端的登录,这样才是不会造成资源的浪费

项目代码

我将分两篇来讲述我的项目,第一篇(也就是这篇为服务器篇);第二篇为客户端篇,当然里面的代码基本上全是我个人所写,也参考过一部分别的代码!

Makefile

基本上我也忘得差不多了,对这个项目,我感觉淡淡的陌生感,但是重拾还是非常容易的嘛!看一个项目咋么形成,Makefile不容错过!

main = insert.o chatroom.o sql.o passwd.o string.o cmd.o socket.o
chatroom:$(main)
	@gcc $(main) -o chatroom -lsqlite3
passwd.o: passwd.c
	@gcc passwd.c -c
insert.o: insert.c	
	@gcc insert.c -c
sql.o:sql.c
	@gcc sql.c -c
string.o:string.c
	@gcc string.c -c
cmd.o:cmd.c
	@gcc cmd.c -c
socket.o:socket.c
	@gcc socket.c -c
.PHONY: clean cleanall
clean:
	@rm *.o
cleanall:
	@rm *.o  chatroom

通过这个我们不难发现这个项目调动了什么,和哪些文件!

passwd.c文件分析

先不说我们把代码贴出来先!

#include"data.h"
void reg_db(sqlite3 *db,char **errmsg,char *name,char *passwd)//注册插入
{
	char sql[1024];
	int rc;
	sprintf(sql,"insert into user(name,passwd)values('%s','%s')",name,passwd);
	rc = sqlite3_exec(db,sql,NULL,NULL,errmsg);
	is_sqlite(rc);
}
void log_db(sqlite3 *db,char **errmsg,char *name,int sockfd)//登陆插入
{
	char sql[1024];
	int rc;
	sprintf(sql,"insert into online(name,socket,flag)values('%s',%d,1)",name,sockfd);
	rc = sqlite3_exec(db,sql,NULL,NULL,errmsg);
	is_sqlite(rc);
}
void delete_online_db(sqlite3 *db,char **errmsg,int sockfd)//删除登陆
{
	char sql[1024];
	int rc;
	sprintf(sql,"delete from online where socket = %d",sockfd);
	rc = sqlite3_exec(db,sql,NULL,NULL,errmsg);
	is_sqlite(rc);
}
int read_user(sqlite3 *db,char **errmsg,char *user)//询问有无此用户
{
	int rc;
	int i;
	sqlite3_stmt *stmt = NULL;
	rc = sqlite3_prepare(db,"select * from user",-1,&stmt,0); 
    is_sqlite_ok(rc);
	rc = sqlite3_step(stmt);
	int userflag = 1;
    while(rc == SQLITE_ROW)
	{
       userflag = my_strcmp(user,sqlite3_column_text(stmt,1));
	   if(userflag == 0 )
	   {
			return USERIN;
	   }	
	   rc = sqlite3_step(stmt);
	}
	return USEROUT;
}
int read_id(sqlite3 *db,char **errmsg,char *user)//询问有无此用户
{
	int rc;
	int i;
	sqlite3_stmt *stmt = NULL;
	rc = sqlite3_prepare(db,"select * from user",-1,&stmt,0); 
    is_sqlite_ok(rc);
	rc = sqlite3_step(stmt);
	int userflag = 1;
    while(rc == SQLITE_ROW)
	{
       userflag = my_strcmp(user,sqlite3_column_text(stmt,1));
	   if(userflag == 0 )
	   {
        	return atoi(sqlite3_column_text(stmt,0));
	   }	
	   rc = sqlite3_step(stmt);
	}
	return USEROUT;
}
void read_id_name(sqlite3 *db,char **errmsg,vpChat temp)//询问有无此用户
{
	int rc;
	int i;
	sqlite3_stmt *stmt = NULL;
	rc = sqlite3_prepare(db,"select * from user",-1,&stmt,0); 
    is_sqlite_ok(rc);
	rc = sqlite3_step(stmt);
	int userflag = 1;
    while(rc == SQLITE_ROW)
	{
       userflag = my_atoi(sqlite3_column_text(stmt,0));
	   if(userflag == temp->flag )
	   {
		   my_strcpy(temp->name,sqlite3_column_text(stmt,1));
	   }	
	   rc = sqlite3_step(stmt);
	}
}
int read_pass(sqlite3 *db,char **errmsg,char *user,char *passwd)//用于用户验证
{
	int rc;
	int i;
	sqlite3_stmt *stmt = NULL;
	rc = sqlite3_prepare(db,"select * from user",-1,&stmt,0); 
    is_sqlite_ok(rc);
	rc = sqlite3_step(stmt);
	int userflag = 1;
	int passwdflag = 1;
    while(rc == SQLITE_ROW)
	{
       userflag = my_strcmp(user,sqlite3_column_text(stmt,1));
       passwdflag = my_strcmp(passwd,sqlite3_column_text(stmt,2));
	   if(userflag == 0 && passwdflag == 0)
	   {
			return PASSWDOK;
	   }	
	   rc = sqlite3_step(stmt);
	}
	return PASSWDNO;
}
int read_online_ok(sqlite3 *db,char **errmsg,char *user)//用于验证在线用户
{
	int rc;
	int i;
	sqlite3_stmt *stmt = NULL;
	rc = sqlite3_prepare(db,"select * from online",-1,&stmt,0); 
    is_sqlite_ok(rc);
	rc = sqlite3_step(stmt);
	int userflag = 1;
    while(rc == SQLITE_ROW)
	{
       userflag = my_strcmp(user,sqlite3_column_text(stmt,1));
	   if(userflag == 0)
	   {
			return ONLINEIN;
	   }	
	   rc = sqlite3_step(stmt);
	}
	return ONLINEOUT;
}

哦,这个原来是用来登录注册用的呀!
那么我在来看看passwd.h呢!

void reg_db(sqlite3 *db,char **errmsg,char *name,char *passwd);
int read_user(sqlite3 *db,char **errmsg,char *user);
int read_pass(sqlite3 *db,char **errmsg,char *user,char *passwd);//用于用户验证
int read_online_ok(sqlite3 *db,char **errmsg,char *user);
void delete_online_db(sqlite3 *db,char **errmsg,int sockfd);
int read_id(sqlite3 *db,char **errmsg,char *user);//询问有无此用户
void read_id_name(sqlite3 *db,char **errmsg,vpChat temp);//询问有无此用户

原来是对函数进行申明啊!

但是#include"data.h" 这个文件又是干什么的呢?
那么data.h按照猜测肯定是各类头文件的引用喽!

#ifndef __DATA__
#define __DATA__
#define MAXSIZE 100
#include<stdio.h>
#include<stdlib.h>
#include<errno.h>
#include<string.h>
#include<netdb.h>
#include<netinet/in.h>
#include<sys/socket.h>
#include<arpa/inet.h>
#include<ctype.h>
#include<sys/types.h>
#include<sys/stat.h>
#include<fcntl.h>
#include<sqlite3.h>
#include<unistd.h>
#include<signal.h>
#include<time.h>
#include<termios.h>
#include<assert.h>
#define portnumber 8000 //宏定义端口号
struct chat
{
	char name[20];//用户名
	char passwd[20];//密码
	int cmd;//命令
	int revert;//返回值
	char toname[20];//目标客户名
    char msg[800];//发送信息
	int flag;
	int sockfd;
	char time[40];
	char filename[40];
};
typedef struct chat stChat;
typedef struct chat* vpChat;
enum cmd
{
	LOGID = 0,
	REG = 1,
	LOG = 2,
	USERHELP = 3,
	USEREXIT = 4,
	CHAT = 3,
	ALL = 4,
	SMILE = 5,
	WELCOME = 6,
	EXIT = 7,
	HELP = 8,
	PASSWD = 9,
	BOOT = 10,
	STEP = 11,
	BAN = 12,
	SET = 13,
	SEE = 14,
	DATA = 15,
	SEND = 16,
	CHANGE = 17
    
};
enum revert
{
	USERIN = 1,
	USEROUT = -1,
	PASSWDOK = 2,
	PASSWDNO = -2,
	ONLINEIN = 3,
	ONLINEOUT = -3,
	REGNO = -4,
	REGOK = 4,
	MYFLAGOK = -5,
	MYFLAGNO = -6,
	TOFLAGOK = -7,
	TOFLAGNO = -8,
	CHATOK = 1,
	ALLOK = 6,
	SMILEOK = 6,
	WELCOMEOK  = 6,
	SEEOK = 7,
    BOOTOK = 1,
    DATAOK = 8,
	SENDOK = 2
};
#endif

果然与我们的猜想差不多啊!

insert.c文件分析

#include"data.h"
void insert_server()//服务器运行提示
{
	system("reset");
	printf("\t\t\t\t\t**********************************\n");
	printf("\t\t\t\t\t*      欢迎使用                  *\n");
	printf("\t\t\t\t\t*           聊天室服务器         *\n");
	printf("\t\t\t\t\t**********************************\n");
}

sogo,这文件原来是给我们看服务器启动时的界面的呀!

sql.c文件分析

看这名字,肯定跟数据库有关!

#include"data.h"
void is_malloc_ok(vpChat *list)
{
   *list = (vpChat)malloc(sizeof(stChat));
   if(*list == NULL)
   {
       exit(1); 
   }
}
void is_sqlite(int rc)  //测试数据库
{
    if(rc == SQLITE_OK)
	{
        printf("sqlite %d succse \n",__LINE__);
	}
	else
	{
		printf("数据库发生错误,请使用SQLITE3 看数据库!\n");
     	printf("sqlite %d error\n",__LINE__);
		exit(1);
	}
}

void is_sqlite_ok(int rc)
{
    if(rc == SQLITE_OK)
	{
	   printf("sqlite %d succse \n",__LINE__);
	}
	else
	{
		printf("sqlite %d error\n",__LINE__);
	}
}

void open_db(sqlite3 **db)//打开数据库
{
	int rc;
	rc = sqlite3_open("server.db",db);
	is_sqlite(rc);
}
void creat_user_db(sqlite3 *db,char **errmsg)//建立user数据表
{
	int rc;
    rc = sqlite3_exec(db,"create table user(id integer primary key autoincrement,name text,passwd text)",NULL,NULL,errmsg);
	is_sqlite_ok(rc);
}
void creat_data_db(sqlite3 *db,char **errmsg)//建立data数据表
{
	int rc;
    rc = sqlite3_exec(db,"create table data(id integer primary key autoincrement,time text,name text,toname text,msg text)",NULL,NULL,errmsg);
	is_sqlite_ok(rc);
}
void creat_online_db(sqlite3 *db,char **errmsg)//建立online数据表
{
	int rc;
    rc = sqlite3_exec(db,"create table online(id integer primary key autoincrement,name text,socket integer,flag integer)",NULL,NULL,errmsg);
	is_sqlite_ok(rc);
}
void creat_server_db(sqlite3 *db,char **errmsg)//建立server数据表
{
	int rc;
    rc = sqlite3_exec(db,"create table server(id integer primary key autoincrement,time text)",NULL,NULL,errmsg);
	is_sqlite_ok(rc);
}
void insert_server_db(sqlite3 *db,char *time,char **errmsg)//向server数据库插入数据
{
         int rc;
        char sql[1024];
	    sprintf(sql,"insert into server(time) values('%s')",time);
	    rc = sqlite3_exec(db,sql,NULL,NULL,errmsg);
	    is_sqlite(rc);
        printf("数据库已经录入完毕\n");
}
void insert_data_db(sqlite3 *db,char **errmsg,char *time,vpChat temp)//插入聊天记录
{
         int rc;
        char sql[1024];
	    sprintf(sql,"insert into data(time,name,toname,msg) values('%s','%s','%s','%s')",time,temp->name,temp->toname,temp->msg);
	    rc = sqlite3_exec(db,sql,NULL,NULL,errmsg);
	    is_sqlite(rc);
        printf("聊天记录已经录入完毕\n");
}
void delete_clean_db(sqlite3 *db,char *tablename,char **errmsg)//清空数据库
{
	char sql[1024];
	int rc;
    sprintf(sql,"delete from %s where name !='root'",tablename);
	rc = sqlite3_exec(db,"delete from student",NULL,NULL,errmsg);
	is_sqlite(rc);
}
void read_db_ok(sqlite3 *db,char *errmsg,char*tablename)//用于检测数据库
{
	int rc;
	int i;
	char sql[1024];
	sqlite3_stmt *stmt = NULL;
    sprintf(sql,"select * from %s ",tablename);
	rc = sqlite3_prepare(db,sql,-1,&stmt,0); 
    is_sqlite_ok(rc);
	rc = sqlite3_step(stmt);
	int ncolumn;
	ncolumn = sqlite3_column_count(stmt);
                 while(rc == SQLITE_ROW)
				 {
                     for(i = 0; i < ncolumn; i++)
					 {
                         printf("%s|",sqlite3_column_text(stmt,i));
					 }
					 printf("\n");
					 rc = sqlite3_step(stmt);
				 }
}
int read_online_fd(sqlite3 *db,char **errmsg,char *user)//获取FD
{
	int rc;
	int i;
	char fd[100];
	sqlite3_stmt *stmt = NULL;
	rc = sqlite3_prepare(db,"select * from online",-1,&stmt,0); 
    is_sqlite_ok(rc);
	rc = sqlite3_step(stmt);
	int userflag = 1;
    while(rc == SQLITE_ROW)
	{
       userflag = my_strcmp(user,sqlite3_column_text(stmt,1));
	   if(userflag == 0)
	   {
            my_strcpy(fd,sqlite3_column_text(stmt,2));

			return my_atoi(fd);
	   }	
	   rc = sqlite3_step(stmt);
	}
	return ONLINEOUT;
}
int read_online_flag(sqlite3 *db,char **errmsg,char *user)//用于验证是否禁言
{
	int rc;
	int i;
	char flag[10];
	sqlite3_stmt *stmt = NULL;
	rc = sqlite3_prepare(db,"select * from online",-1,&stmt,0); 
    is_sqlite_ok(rc);
	rc = sqlite3_step(stmt);
	int userflag = 1;
    while(rc == SQLITE_ROW)
	{
       userflag = my_strcmp(user,sqlite3_column_text(stmt,1));
	   if(userflag == 0)
	   {
            my_strcpy(flag,sqlite3_column_text(stmt,3));

			return my_atoi(flag);
	   }	
	   rc = sqlite3_step(stmt);
	}
	return -10;
}
void write_online_all(sqlite3 *db,char **errmsg,vpChat temp)//向在线用户发送信息
{
	int rc;
  	int flag;
	char fd[100];
	sqlite3_stmt *stmt = NULL;
	rc = sqlite3_prepare(db,"select * from online",-1,&stmt,0); 
    is_sqlite_ok(rc);
	rc = sqlite3_step(stmt);
	int userflag = 1;
    while(rc == SQLITE_ROW)
	{
	        flag = my_strcmp(temp->name,sqlite3_column_text(stmt,1));
            my_strcpy(fd,sqlite3_column_text(stmt,2));
			if(flag != 0)
			{
			   write( my_atoi(fd),temp,sizeof(stChat));
			}
		    sleep(1);	
	        rc = sqlite3_step(stmt);
	}
}
int update_passwd(sqlite3 *db,char **errmsg,char *name,char *passwd)//修改密码
{
         int rc;
         char sql[1024];
	     sprintf(sql,"update user set passwd = '%s'where name = '%s'",passwd,name);
	    rc = sqlite3_exec(db,sql,NULL,NULL,errmsg);
        if(rc == SQLITE_OK)
	     {
			 return 1;
	     }
	     else
	     {
			 return 0;
   	     }

}
int update_user(sqlite3 *db,char **errmsg,char *name,char *toname)//修改密码
{
         int rc;
         char sql[1024];
	     sprintf(sql,"update user set name = '%s'where name = '%s'",toname,name);
	    rc = sqlite3_exec(db,sql,NULL,NULL,errmsg);
        if(rc == SQLITE_OK)
	     {
			 return 1;
	     }
	     else
	     {
			 return 0;
   	     }

}
int update_db_data(sqlite3 *db,char **errmsg,char *name,char *toname)//修改密码
{
         int rc1;
		 int rc2;
         char sql[1024];
	     sprintf(sql,"update data set name = '%s'where name = '%s'",toname,name);
	    rc1 = sqlite3_exec(db,sql,NULL,NULL,errmsg);
		memset(sql,0,1024);
	     sprintf(sql,"update data set toname = '%s'where toname = '%s'",toname,name);
	    rc2 = sqlite3_exec(db,sql,NULL,NULL,errmsg);
        if(rc1 == SQLITE_OK && rc2 == SQLITE_OK)
	     {
			 return 1;
	     }
	     else
	     {
			 return 0;
   	     }

}
int update_flag(sqlite3 *db,char **errmsg,char *name,int flag)//禁言解禁操作
{
         int rc;
         char sql[1024];
	     sprintf(sql,"update online set flag = %d where name = '%s'",flag,name);
	    rc = sqlite3_exec(db,sql,NULL,NULL,errmsg);
        if(rc == SQLITE_OK)
	     {
			 return 1;
	     }
	     else
	     {
			 return 0;
   	     }
}
int delete_user(sqlite3 *db,char **errmsg,char *name)//注销用户
{
         int rc;
         char sql[1024];
	     sprintf(sql,"delete from user where name = '%s'",name);
	     rc = sqlite3_exec(db,sql,NULL,NULL,errmsg);
         if(rc == SQLITE_OK)
	     {
			 return 1;
	     }
	     else
	     {
			 return 0;
   	     }

}
void read_online_all(sqlite3 *db,char **errmsg,vpChat temp)//向在线用户发送信息
{
	int rc;
	sqlite3_stmt *stmt = NULL;
	rc = sqlite3_prepare(db,"select * from online",-1,&stmt,0); 
    is_sqlite_ok(rc);
	rc = sqlite3_step(stmt);
	int userflag = 1;
    while(rc == SQLITE_ROW)
	{
	        my_strcpy(temp->msg,sqlite3_column_text(stmt,1));
			mywrite(temp);
		    sleep(1);	
	        rc = sqlite3_step(stmt);
    }
}
void read_data(sqlite3 *db,char **errmsg,vpChat temp)//向在线用户发送信息
{
	int rc;
	char name[80];
	char toname[80];
	sqlite3_stmt *stmt = NULL;
	rc = sqlite3_prepare(db,"select * from data",-1,&stmt,0); 
    is_sqlite_ok(rc);
	rc = sqlite3_step(stmt);
	int userflag = 1;
    while(rc == SQLITE_ROW)
	{
	        my_strcpy(name,sqlite3_column_text(stmt,2));
	        my_strcpy(toname,sqlite3_column_text(stmt,3));
		if(	my_strcmp(temp->name,name) == 0)
		{
            strcat(temp->msg,"你");      
            strcat(temp->msg,"给");      
            strcat(temp->msg,toname);      
            strcat(temp->msg,"发了");      
            strcat(temp->msg,sqlite3_column_text(stmt,4));
            strcat(temp->msg,"\n");      
	        my_strcpy(temp->time,sqlite3_column_text(stmt,1));
	        mywrite(temp);
			memset(temp->msg,0,sizeof(temp->msg));
	        sleep(1);		
		}
		if(my_strcmp(temp->name,toname) == 0)
		{
            strcat(temp->msg,toname);      
            strcat(temp->msg,"给你发了");      
            strcat(temp->msg,sqlite3_column_text(stmt,4));
            strcat(temp->msg,"\n");      
	        my_strcpy(temp->time,sqlite3_column_text(stmt,1));
	        mywrite(temp);
			memset(temp->msg,0,sizeof(temp->msg));
	        sleep(1);		
		}
	        rc = sqlite3_step(stmt);
    }
}

这个是数据库的各类操作呢!
那么它的头文件估计也是函数申明

void is_sqlite(int rc);  //测试数据库
void is_malloc_ok(vpChat *list);
void is_sqlite_ok(int rc);
void open_db(sqlite3 **db);//打开数据库
void creat_user_db(sqlite3 *db,char **errmsg);//建立user数据表
void creat_data_db(sqlite3 *db,char **errmsg);//建立data数据表
void creat_online_db(sqlite3 *db,char **errmsg);//建立online数据表
void creat_server_db(sqlite3 *db,char **errmsg);//建立server数据表
void insert_server_db(sqlite3 *db,char *time,char **errmsg);
void read_db_ok(sqlite3 *db,char *errmsg,char*tablename);
void delete_clean_db(sqlite3 *db,char *tablename,char **errmsg);
int read_online_fd(sqlite3 *db,char **errmsg,char *user);
int read_online_flag(sqlite3 *db,char **errmsg,char *user);
void write_online_all(sqlite3 *db,char **errmsg,vpChat temp);
int update_passwd(sqlite3 *db,char **errmsg,char *name,char *passwd);
void insert_data_db(sqlite3 *db,char **errmsg,char *time,vpChat temp);
int update_flag(sqlite3 *db,char **errmsg,char *name,int flag);
int delete_user(sqlite3 *db,char **errmsg,char *name);
void read_online_all(sqlite3 *db,char **errmsg,vpChat temp);
void read_data(sqlite3 *db,char **errmsg,vpChat temp);
int update_user(sqlite3 *db,char **errmsg,char *name,char *toname);
int update_db_data(sqlite3 *db,char **errmsg,char *name,char *toname);

干货来了,string.c

很明显是我为了面试准备所写的函数!经测是有效的!

#include"data.h"
int my_strlen(const char *str)
{
	int len;
	while(*str++ != '\0')
	{
		len++;
	}
	return len;
}
int my_strcmp(const char *str1,const char *str2)
{
	if(str1 != NULL && str2 != NULL)
	{
	}
	else
	{
		return -2;
	}

	int i;
	while((*str1)!='\0' && (*str2) != '\0')
	{
		if((*str1) == (*str2))
		{
			str1++;
			str2++;
		}
		else
		{
			break;
		}
	}
	if((*str1) == '\0' && (*str2) == '\0')
	{
		return 0;
	}
	else
	{
		return (*str1) > (*str2)? 1 : -1;
	}
}
char  *my_strcpy( char *dest,const char *src)
{
	if(dest != NULL && src != NULL)
	{
	}
	else
	{
		return NULL;
	}
	char *adress = dest;
	while(((*dest++) = (*src++)) != '\0');
	return adress;
}
int my_atoi(const char *str)
{
	if(str != NULL)
	{
	}
	else
	{
		return -1;
	}
	const char* p = str;
	int minus = 0;
	long result = 0;
	if(*p == '-')
	{
		minus = 1;
		p++;
	}
	else if(*p == '+')
	{
		p++;
	}
	while(*p != '\0')
	{
		if(*p < '0' || *p >'9')
		{
			return -1;
		}
		result = result*10 + ((*p) - '0');
		p++;
	}
	minus = 1 ? -result:result;
	return result;
}

char * my_strcat(char *dest,const char *src)
{
	if(dest != NULL && src != NULL)
	{
	}
	else
	{
		return NULL;
	}
	char *address = dest;
	while((*dest) != '\0')
	{
		dest++;
	}
	while(((*dest++) = (*src++)) != '\0');
	return address;
}

CMD.c 命令解析

#include"data.h"
int cmd_user(sqlite3 *db,char **errmsg,vpChat temp,int sockfd)//用语判别用户信息
{
	int flag;
	int sayflag;
	printf("cmd = %d\n",temp->cmd);
   switch(temp->cmd )
   {
    case REG:
		  {
	         flag = read_user(db,errmsg,temp->name);
	         if(flag == USERIN)
	         {
		         return REGNO;//注册重名
	         }
	          else
	         {
                reg_db(db,errmsg,temp->name,temp->passwd);
                temp->flag = read_id(db,errmsg,temp->name);
                return REGOK;
	         }
		     break;
		  }
     case LOG:
		  {
			  int flagpasswd;
			  flag = read_online_ok(db,errmsg,temp->name);
			  if(flag == ONLINEIN)
			  {
				  return ONLINEIN;
			  }
			  else
			  {
				  flagpasswd = read_pass(db,errmsg,temp->name,temp->passwd);
				  if(flagpasswd == PASSWDOK)
				  {
                      log_db(db,errmsg,temp->name,sockfd);
					  return PASSWDOK;
				  }
				  else
				  {
					  return PASSWDNO;
				  }
			  }
			  break;
		  }  
     case LOGID:
		  {
			  int flagpasswd;
              read_id_name(db,errmsg,temp);//询问有无此用户
			  flag = read_online_ok(db,errmsg,temp->name);
			  if(flag == ONLINEIN)
			  {
				  return ONLINEIN;
			  }
			  else
			  {
				  flagpasswd = read_pass(db,errmsg,temp->name,temp->passwd);
				  if(flagpasswd == PASSWDOK)
				  {
                      log_db(db,errmsg,temp->name,sockfd);
					  return PASSWDOK;
				  }
				  else
				  {
					  return PASSWDNO;
				  }
			  }
			  break;
		  }  
	  case CHAT:
		   {
			   int tempfd;
              tempfd = read_online_fd(db,errmsg,temp->toname);
			  if(tempfd == ONLINEOUT)
			  {
                   return ONLINEOUT;
			  }
			  else
			  {
                  sayflag = read_online_flag(db,errmsg,temp->name); 
				  if(sayflag == 0)
				  {
					  return MYFLAGNO;
				  }
				  else
				  { 
                     temp->flag = flag;
					 temp->sockfd = tempfd;
					 insert_data_db(db,errmsg,temp->time,temp);
				     return CHATOK;
				  }
			  }
		      break;
		   }
	  case ALL:
		   {
               sayflag = read_online_flag(db,errmsg,temp->name); 
			   if(sayflag == 0)
			   {

				   return MYFLAGNO;
			   }
			   else
			   {
				   insert_data_db(db,errmsg,temp->time,temp);
			       return ALLOK;
			   }
			   break;
		   }
	  case SMILE:
		   {
               sayflag = read_online_flag(db,errmsg,temp->name); 
			   if(sayflag == 0)
			   {
				   return MYFLAGNO;
			   }
			   else
			   {
					 insert_data_db(db,errmsg,temp->time,temp);
			       return SMILEOK;
			   }
			   break;
		   }
       case WELCOME:
		   {
               sayflag = read_online_flag(db,errmsg,temp->name); 
			   if(sayflag == 0)
			   {
				   return MYFLAGNO;
			   }
			   else
			   {
					 insert_data_db(db,errmsg,temp->time,temp);
			       return WELCOMEOK;
			   }
			   break;
		   }
	   case PASSWD:
		   {
             flag =  update_passwd(db,errmsg,temp->name,temp->passwd);
             return flag;
			 break;
		   }
	   case BOOT:
		   {  
			  int tempfd;
              tempfd = read_online_fd(db,errmsg,temp->toname);
			  if(tempfd == ONLINEOUT)
			  {
                   return ONLINEOUT;
			  }
			  else
			  {
					 temp->sockfd = tempfd;
					 return BOOTOK;
			  }
			  break;
		   }
		case STEP:
		   {
			  int tempfd;
              tempfd = read_online_fd(db,errmsg,temp->toname);
			  if(tempfd == ONLINEOUT)
			  {
                   return ONLINEOUT;
			  }
			  else
			  {
                     flag =  update_flag(db,errmsg,temp->toname,0);
					 if(flag == 1)
					 {
					    temp->sockfd = tempfd;
					 }
					 return flag;
			  }
			  break;
		   }
	    case BAN:
		   {

			  int tempfd;
              tempfd = read_online_fd(db,errmsg,temp->toname);
			  if(tempfd == ONLINEOUT)
			  {
                   return ONLINEOUT;
			  }
			  else
			  {
                     sayflag = read_online_flag(db,errmsg,temp->toname); 
					 if(sayflag == 1)
					 {
						 return TOFLAGOK;
					 }
					 else
					 {
                         flag =  update_flag(db,errmsg,temp->toname,1);
					     if(flag == 1)
					     {
					            temp->sockfd = tempfd;
					     }
					     return flag;
					 }
			  }
			  break;
		   }
		  case SET:
		   {

			  int tempfd;
              tempfd = read_online_fd(db,errmsg,temp->toname);
			  if(tempfd == ONLINEOUT)
			  {
                  flag = delete_user(db,errmsg,temp->toname);
    			  return flag;
			  }
			  else
			  {
				  return ONLINEIN;
			  }
			   break;
		   }
		  case SEE:
		   {
			   return SEEOK;
			   break;
		   }
		  case DATA:
		   {
               return DATAOK;
			   break;
		   }
		  case SEND:
		   {
			  int tempfd;
              tempfd = read_online_fd(db,errmsg,temp->toname);
			  if(tempfd == ONLINEOUT)
			  {
    			  return ONLINEOUT;
			  }
			  else
			  {
				   temp->sockfd = tempfd;
				   return SENDOK;
			  }
			   break;
		   }
		  case CHANGE:
		   {
	         flag = read_user(db,errmsg,temp->toname);
	         if(flag == USERIN)
	         {
		         return REGNO;//注册重名
	         }
             else
			 {
                 flag = update_user(db,errmsg,temp->name,temp->toname);//????
                 if(flag == 1)
				 {
                      sayflag = update_db_data(db,errmsg,temp->name,temp->toname);//修改密码
                     return sayflag;
				 }
				 else
				{
					return 0;
				}
			 }
			   break;
		   }

   }
}

那么它的头文件是估计也是用于申明函数的!

int cmd_user(sqlite3 *db,char **errmsg,vpChat temp,int sockfd);

socket.c文件分析

#include"data.h"
void mybzero(struct sockaddr_in *sin)//对server_addr_in  结构进行赋值
{
	bzero(sin,sizeof(struct sockaddr_in));   /* 先清零 */
	sin->sin_family=AF_INET;                 //
	sin->sin_addr.s_addr=htonl(INADDR_ANY);  //表示接受任何ip地址   将ip地址转换成网络字节序
	sin->sin_port=htons(portnumber);         //将端口号转换成网络字节序
}
int mysocket()//调用socket函数创建一个TCP协议套接口
{
	int lfd;
	if((lfd = socket(AF_INET,SOCK_STREAM,0)) == -1) // AF_INET:IPV4;SOCK_STREAM:TCP
	{
		fprintf(stderr,"Socket error:%s\n\a",strerror(errno));
		exit(1);
	}
	return lfd;
}
void mybind(int lfd,struct sockaddr_in *sin)	// 调用bind函数 将serer_addr结构绑定到sockfd上 
{
	if(bind(lfd,(struct sockaddr *)(sin),sizeof(struct sockaddr))==-1)
	{
		fprintf(stderr,"Bind error:%s\n\a",strerror(errno));
		exit(1);
	}
}
void mylisten(int lfd)// 开始监听端口   等待客户的请求
{
	if(listen(lfd,20)==-1)
	{
		fprintf(stderr,"Listen error:%s\n\a",strerror(errno));
		exit(1);
	}
}
int myaccept(int lfd,struct sockaddr_in *cin,socklen_t * addr_len)// 接受客户端的请求 
{
	int cfd;
	if((cfd=accept(lfd,(struct sockaddr *)cin,addr_len))==-1)
	{
		fprintf(stderr,"Accept error:%s\n\a",strerror(errno));
		exit(1);
	}
	return cfd;
}
void mywrite(vpChat temp)
{
	int num;
    if((num = write(temp->sockfd,temp,sizeof(stChat))) == -1)
    {
        printf("send error!\n");
    }
}

服务器主程序

朱程序里面主要是打开数据库,初始化数据库,及用于显示数据库的命令,下面为代码部分

#include"data.h"
#include"package.h"
void display(stChat temp)//用于显示发送命令
{
	printf("name = %s\n",temp.name);
	printf("passwd = %s\n",temp.passwd);
	printf("cmd = %d\n",temp.cmd);
	printf("revert = %d\n",temp.revert);
    printf("toname = %s\n",temp.toname);
	printf("msg = %s\n",temp.msg);
	printf("flag = %d\n",temp.flag);
	printf("sockfd = %d\n",temp.sockfd);
	printf("time = %s\n",temp.time);
	printf("filename = %s\n",temp.filename);
}
int main()
{
	time_t timep;
	time(&timep);
	char *timedata = ctime(&timep);
	sqlite3 *db = NULL;
	char *errmsg;
	open_db(&db);//打开数据库
    creat_user_db(db,&errmsg);
    creat_data_db(db,&errmsg);
    creat_online_db(db,&errmsg);
    creat_server_db(db,&errmsg);
    insert_server_db(db,timedata,&errmsg);//向server数据库插入数据
	insert_server();
    int  lfd;
	int cfd;
	int sfd;
	int rdy;
    stChat temp;
	struct sockaddr_in sin;
	struct sockaddr_in cin;
	int client[FD_SETSIZE];  /* 客户端连接的套接字描述符数组 */
	int maxi;
	int maxfd;                        /* 最大连接数 */
	fd_set rset;
	fd_set allset;
	socklen_t addr_len;         /* 地址结构长度 */
	int i;
	int n;
	int len;
	int opt = 1;   /* 套接字选项 */
	char addr_p[20];
    mybzero(&sin);
    lfd = mysocket();
	/*设置套接字选项 使用默认选项*/
	setsockopt(lfd, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt));
    mybind(lfd,&sin);	// 调用bind函数 将serer_addr结构绑定到sockfd上 
    mylisten(lfd);// 开始监听端口   等待客户的请求
	printf("\t\t\t服务器开始等待客户端链接\n");
	maxfd = lfd;                                /*对最大文件描述符进行初始化*/
	maxi = -1;
	/*初始化客户端连接描述符集合*/
	for(i = 0;i < FD_SETSIZE;i++)
	{
        client[i] = -1;
	}

	FD_ZERO(&allset);                     /* 清空文件描述符集合 */
	FD_SET(lfd,&allset);                    /* 将监听字设置在集合内 */
	/* 开始服务程序的死循环 */
	while(1)
	{
		rset = allset;
        
		/*得到当前可以读的文件描述符数*/
		rdy = select(maxfd + 1, &rset, NULL, NULL, NULL);
		
		if(FD_ISSET(lfd, &rset))
		{
		
			addr_len = sizeof(sin);
            cfd = myaccept(lfd,&cin,&addr_len);// 接受客户端的请求 
            /*查找一个空闲位置*/
			for(i = 0; i<FD_SETSIZE; i++)
			{       //printf("%d\t",client[i]);
				if(client[i] <= 0)
				{
					client[i] = cfd;   /* 将处理该客户端的连接套接字设置到该位置 */
					break;
				}
			}

		/* 太多的客户端连接   服务器拒绝俄请求  跳出循环 */
			if(i == FD_SETSIZE)
			{
				printf("too many clients");
				exit(1);
			}

			FD_SET(cfd, &allset);     /* 设置连接集合 */

			if(cfd > maxfd)                  /* 新的连接描述符 */
			{
				maxfd = cfd;
			}

			if(i > maxi)
			{
				maxi = i;
			}

			if(--rdy <= 0)                /* 减少一个连接描述符 */
			{
				continue;
			}

		}
        
		/* 对每一个连接描述符做处理 */
		for(i = 0;i< FD_SETSIZE;i++)
		{   
			if((sfd = client[i]) < 0)
			{
				continue;
			}

			if(FD_ISSET(sfd, &rset))
			{
			    printf("客户端sfd = %d已经成功链接\n",sfd);
				n = read(sfd,&temp,sizeof(stChat));
				if(n == 0)
				{
					printf("客户端sfd = %d已经离开本服务器. \n",sfd);
				   	delete_online_db(db,&errmsg,sfd);
					fflush(stdout);                                    /* 刷新 输出终端 */
					close(sfd);
					FD_CLR(sfd, &allset);                        /*清空连接描述符数组*/
					client[i] = -1;
				}
				else
				{
					temp.sockfd = sfd;
					/* 将客户端地址转换成字符串 */
					inet_ntop(AF_INET, &cin.sin_addr, addr_p, sizeof(addr_p));
					addr_p[strlen(addr_p)] = '\0';
					/*打印客户端地址 和 端口号*/
					printf("客户端的Ip是%s, 端口是 %d\n",addr_p,ntohs(cin.sin_port));
                    int revert ;
					revert = cmd_user(db,&errmsg,&temp,sfd);
					temp.revert = revert;
					printf("开始向客户端发送命令!\n");
				//	printf("以下为命令结构体!\n");
                 //   display(temp);
				    if(revert < 5)
					{  
                         mywrite(&temp);
					}
					else if(revert == DATAOK)
					{
                        read_data(db,&errmsg,&temp);//向在线用户发送信息
					}
					else if(revert == SEEOK)
					{
                         read_online_all(db,&errmsg,&temp);//向在线用户发送信息
					}
					else if(revert == ALLOK || revert == SMILEOK || revert == WELCOMEOK)
					{
                         write_online_all(db,&errmsg,&temp);
					}
					printf("发送完毕!\n");
					memset(&temp,0,sizeof(stChat));//清空发送数据结构体
					/* 谐函数出错 */
					if(n == 1)
					{
						exit(1);
					}
				}

				/*如果没有可以读的套接字   退出循环*/
				if(--rdy <= 0)
				{
					break;
				}


			}
		}

	}

	close(lfd);       /* 关闭链接套接字 */
	return 0;
}

项目地址

其实它就在我的github里面。你可怜点击链接进入,当然希望您在看的过程中能够followe me 一下,感谢!

基于LINUX的多功能聊天室

标签:

原文地址:http://www.cnblogs.com/lonelyxmas/p/4266729.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!