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;
}
文章来源地址https://www.toymoban.com/news/detail-467858.html
文章来源:https://www.toymoban.com/news/detail-467858.html
到了这里,关于day2 在线词典项目注册登录的实现的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!