day2 在线词典项目注册登录的实现

server.c

#include <stdio.h>
#include <stdlib.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <string.h>
#include <unistd.h>
#include <sqlite3.h>
#include <signal.h>


#define N  32


#define R 1  // user - register
#define L 2  // user - login
#define Q 3  // user - query
#define H 4  // user - history


#define DATABASE  "my.db"


//定义通信双方的信息结构体
typedef struct {
	int type;
	char name[N];
	char data[256];
}MSG;


int do_client(int acceptfd, sqlite3 *db);
int do_register(int acceptfd, MSG *msg, sqlite3 *db);
int do_login(int acceptfd, MSG *msg, sqlite3 *db);
int do_query(int acceptfd, MSG *msg, sqlite3 *db);
int do_history(int acceptfd, MSG *msg, sqlite3 *db);


int main(int argc, const char *argv[])
{
	int sockfd;
	struct sockaddr_in serveraddr;
	int n;
	MSG msg;
	sqlite3 *db;
	int acceptfd;
	pid_t pid;


	if (argc != 3)
	{
		printf("Usage:%s serverip port\n", argv[0]);
		return -1;
	}

	//打开数据库
	if(sqlite3_open(DATABASE, &db) != SQLITE_OK)
	{
		printf("%s\n", sqlite3_errmsg(db));

		return -1;
	}
	else
	{
		printf("open DATABASE success\n");
	}
	/*创建套接字*/
	if((sockfd = socket(AF_INET, SOCK_STREAM, 0)) < 0)
	{
		perror("fail to socket\n");
		return -1;
	}

	/*设置通信地址组结构体*/
	bzero(&serveraddr, sizeof(serveraddr));
	serveraddr.sin_family = AF_INET;
	serveraddr.sin_addr.s_addr = inet_addr(argv[1]);
	serveraddr.sin_port = htons(atoi(argv[2]));

	/*绑定通信结构体*/
	if (bind(sockfd, (struct sockaddr *)&serveraddr, sizeof(serveraddr)) < 0)
	{
		perror("fail to bind\n");

		return -1;
	}

	//设置为监听,模式
	if (listen(sockfd, 5) < 0)
	{
		printf("fail to listen\n");

		return -1;
	}

	//处理僵尸进程
	signal(SIGCHLD, SIG_IGN);


	//实现并发
	while(1)
	{
		if((acceptfd = accept(sockfd, NULL, NULL)) < 0)
		{
			perror("fail to accept");
			return -1;
		}

		if((pid = fork()) < 0)
		{
			perror("fail to fork");
			return -1;
		}
		else if(pid == 0) //子进程
		{
			//处理客户端具体消息
			close(sockfd);
			do_client(acceptfd, db);
		}
		else  //父进程,接受客户端的请求
		{
			close(acceptfd);
		}
	}

	return 0;
}



int do_client(int acceptfd, sqlite3 *db)
{
	MSG msg;
	while(recv(acceptfd, &msg, sizeof(msg), 0) > 0)
	{
		switch(msg.type)
		{
		case R:
			do_register(acceptfd, &msg, db);
			break;
		case L:
			do_login(acceptfd, &msg, db);
			break;
		case Q:
			do_query(acceptfd, &msg, db);
			break;
		case H:
			do_history(acceptfd, &msg, db);
			break;
		default:
			printf("invalid data msg\n");

		}
	}

	printf("client exit\n");
	close(acceptfd);
	exit(0);

	return 0;
}


int do_register(int acceptfd, MSG *msg, sqlite3 *db)
{
	char *errmsg;
	char sql[128];

	sprintf(sql, "insert into usr values('%s', %s);", msg->name, msg->data);
	printf("%s\n", sql);

	if( sqlite3_exec(db, sql, NULL, NULL, &errmsg) != SQLITE_OK)
	{
		printf("%s\n", errmsg);
		strcpy(msg->data, "user name already exist");
	}else{
		printf("client register ok!\n");
		strcpy(msg->data, "OK!");
	}

	if(send(acceptfd, msg, sizeof(MSG), 0) < 0)
	{
		perror("fail to send");
		return -1;
	}
	return 1;
	
}

int do_login(int acceptfd, MSG *msg, sqlite3 *db)
{
	char sql[128] = {};
	char *errmsg;
	char **resultp;             //sql语句查询的结果;
	int nrow;                   //查询的行数;
	int ncloumn;                //查询的列数;

	sprintf(sql, "select * from usr where name = '%s' and pass = '%s';", msg->name, msg->data);
	printf("%s\n", sql);

	if(sqlite3_get_table(db, sql, &resultp, &nrow, &ncloumn, &errmsg) != SQLITE_OK)
	{
		printf("%s\n", errmsg);
		return -1;
	}else{
		printf("get_table ok\n");
	}

	//查询成功,数据库存在此用户;
	if(nrow == 1)                
	{
		strcpy(msg->data, "OK");
		send(acceptfd, msg, sizeof(MSG), 0);
		return 1;
	}

	//密码或用户名错误;
	if(nrow == 0)
	{
		strcpy(msg->data, "usr/passwd wring");
		send(acceptfd, msg, sizeof(MSG), 0);
	}else{
		printf("%s\n", msg->data);
	}

	return 0;
}

int do_query(int acceptfd, MSG *msg, sqlite3 *db )
{
	printf("query ...\n");

	return 1;
}

int do_history(int acceptfd, MSG *msg, sqlite3 *db)
{
	printf("history ...\n");

	return 0;
}

client.c

#include <stdio.h>
#include <stdlib.h>
#include <strings.h>
#include <sys/types.h>          
#include <sys/socket.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <unistd.h>


#define N 32

#define R 1                   // user - register   ,用户注册;
#define L 2                   // user - login      ,用户登录;
#define Q 3                   // user - query      ,用户查询;
#define H 4                   // user - history    ,历史记录;

/*通信双方的网络信息结构体*/
typedef struct {
	int type;                 //类型;
	char name[N];             //用户名;
	char data[256];           //用户数据;
}MSG;

/*注册模块*/
int do_register(int sockfd, MSG *msg)
{
	msg->type = R;

	printf("Input name:");
	scanf("%s", msg->name);
	getchar();

	printf("Input passwd:");
	scanf("%s", msg->data);

	//将信息发送给服务器
	if(send(sockfd, msg, sizeof(MSG), 0) < 0)
	{
		printf("fail to send\n");
		return -1;
	}

	//等待读取服务器回馈
	if( recv(sockfd, msg, sizeof(MSG), 0) < 0)
	{
		printf("fail to recv\n");
		return -1;
	}

	//ok ! or usr alread exist
	printf("%s\n", msg->data);
	return;

}

/*登录模块*/
int do_login(int sockfd, MSG *msg)
{
	msg->type = L;

	printf("Input name:");
	scanf("%s", msg->name);
	getchar();

	printf("Input passwd:");
	scanf("%s", msg->data);

	if(send(sockfd, msg, sizeof(MSG), 0) < 0)
	{
		printf("fail to send\n");
		return -1;
	}

	//等待读取服务器回馈
	if( recv(sockfd, msg, sizeof(MSG), 0) < 0)
	{
		printf("fail to recv\n");
		return -1;
	}

	if(strncmp(msg->data, "OK", 3) == 0)
	{
		printf("login ok!\n");
		return 1;
	}

	return 0;
}

int do_query(int sockfd, MSG *msg)
{
	printf("query...\n");

	return 1;
}

int do_history(int sockfd, MSG *msg)
{
	printf("history...\n");

	return 0;
}



int main(int argc, const char *argv[])
{
	int sockfd;
	struct sockaddr_in serveraddr;
	int n;
	MSG msg;


	/*参数判断*/
	if(argc != 3)
	{
		printf("Usage:%s serverip port\n", argv[0]);
		return -1;
	}

	/*创建套接字*/
	if( (sockfd = socket(AF_INET, SOCK_STREAM, 0)) < 0)
	{
		perror("fail to socket\n");
		return -1;
	}

	/*设置通信地址族结构体*/
	bzero(&serveraddr, sizeof(serveraddr));
	serveraddr.sin_family = AF_INET;

	//inet_addr将点分十进制表示的IP地址转换为整数表示的形式
	serveraddr.sin_addr.s_addr = inet_addr(argv[1]);

	//htons->是主机字节序到网络字节序;atoi()函数是讲端口还转换为整数形式;
	serveraddr.sin_port = htons(atoi(argv[2]));

	/*向服务器发起连接申请*/
	if( (connect(sockfd, (struct sockaddr *)&serveraddr, sizeof(serveraddr))) < 0)
	{
		perror("fail to connect");
		return -1;
	}

	/*一级菜单*/
	while(1) {
		printf("***********************************************************\n");
		printf("* 1.register                 2.login               3.quit *\n");
		printf("***********************************************************\n");
		printf("Please choose:");

		scanf("%d", &n);
		getchar();

		switch(n)
		{
		case 1:
			do_register(sockfd, &msg);
			break;
		case 2:
			if( do_login(sockfd, &msg) == 1)
			{
				goto next;
			}
			break;
		case 3:
			close(sockfd);
			exit(0);
			break;
		default:
			printf("Invalid data cmd\n");
		}
	}

	/*二级菜单*/
next:
	while(1){
		printf("***********************************************************\n");
		printf("* 1.query_word           2.history_record          3.quit *\n");
		printf("***********************************************************\n");
		printf("Please choose:");

		scanf("%d", &n);
		getchar();

		switch(n)
		{
		case 1:
			do_query(sockfd, &msg);
			break;
		case 2:
			do_history(sockfd, &msg);
			break;
		case 3:
			close(sockfd);
			exit(0);
			break;
		default:
			printf("Invalid data cmd\n");
		}

	}
	return 0;
}