C语言项目（一）---Linux下基于TCP网络传输的词典在线查询的并发服务器

已于 2022-12-07 21:44:08 修改
阅读量832 收藏 3
点赞数 1
文章标签： 网络 服务器 c语言
于 2022-11-27 11:43:38 首次发布
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/m0_73492050/article/details/128062110
版权

概述

Linux下，采用TCP编程，以多进程的方式实现客户端与服务器的交互，客户端提出注册、登录、退出等要求。服务器在进行具体的操作，应用SQLITE数据库和单词表，完成数据处理。

流程图

客户端流程图：
在这里插入图片描述
服务器流程图：
在这里插入图片描述

代码如下

服务器代码：

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sqlite3.h>
#include <unistd.h>
#include <sys/types.h>         
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <signal.h>
#include <time.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);
void do_register(int acceptfd,MSG *msg,sqlite3 *db);
int do_login(int acceptfd,MSG *msg,sqlite3 *db);
int do_searchword(int acceptfd,MSG *msg,char *word);
int do_query(int acceptfd,MSG *msg,sqlite3 *db);
int history_callback(void *arg,int f_num ,char **f_value,char **f_name);
int do_history(int acceptfd,MSG *msg,sqlite3 *db);
int get_date(char *date);


//./client 192.168.203.130 10000
int main(int argc, 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("socket");
		return -1;
	}

	bzero(&serveraddr,sizeof(serveraddr));
	serveraddr.sin_family = AF_INET;
	serveraddr.sin_port = htons(atoi(argv[2]));
	serveraddr.sin_addr.s_addr = inet_addr(argv[1]);

	if(bind(sockfd,(struct sockaddr *)&serveraddr,sizeof(serveraddr)) < 0){
		perror("bind");
		exit(0);
	}
	if(listen(sockfd,5) < 0){
		perror("listen");
		exit(0);
	}
		
	signal(SIGCHLD,SIG_IGN);

	while(1){
		if((acceptfd = accept(sockfd,NULL,NULL)) < 0){
			perror("accept");
			return -1;
		}
		if((pid = fork())<0){
			perror("fork");
			return -1;
		}
		else if(pid == 0){
			close(sockfd);
			do_client(acceptfd,db);
		}else{
			close(acceptfd);
		}
	}

	return 0;
}



int history_callback(void *arg,int f_num ,char **f_value,char **f_name){
	
	int acceptfd;
	MSG msg;

	acceptfd = *((int *)arg);
	
	sprintf(msg.data,"%s , %s",f_value[1],f_value[2]);
	send(acceptfd,&msg,sizeof(MSG),0);
	return 0;
}
int do_history(int acceptfd,MSG *msg,sqlite3 *db){
	char sql[128] = {0};
	char *errmsg;

	sprintf(sql,"select * from record where name = '%s'",msg->name);

	if(sqlite3_exec(db,sql, history_callback,(void *)&acceptfd,&errmsg) != SQLITE_OK){
		printf("%s\n",errmsg);
	}else{
		printf("Query record done.\n");
	}
	msg->data[0] = '\0';
	send(acceptfd,msg,sizeof(MSG),0);
	return 0;
}

int do_client(int acceptfd,sqlite3 *db){
	MSG msg;
	while(recv(acceptfd,&msg,sizeof(msg),0) > 0){
		printf("type:%d\n",msg.type);
		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;
		dafault:
			printf("Invalid data msg.\n");
		}
	}
	printf("client exit.\n");
	close(acceptfd);
	exit(0);
	return 0;
}


void do_register(int acceptfd,MSG *msg,sqlite3 *db){

	char *errmsg;
	char sql[128] = {0};

	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,"usr name already exist.");
	}else{
		printf("client register ok!\n");
		strcpy(msg->data,"OK!");
	}
	if(send(acceptfd,(void *)msg,sizeof(MSG),0) < 0){
		perror("send");
		exit(0);
	}
	
}


int do_login(int acceptfd,MSG *msg,sqlite3 *db){
	char sql[128] = {0};
	int nrow;
	int ncloumn;
	char **resultp;
	char *errmsg;
	int t=2;
	sprintf(sql,"select * from usr where name = '%s' and pass = '%s'",msg->name,msg->data);
	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");
		t = 1;
	}else{
		strcpy(msg->data,"usr/passwd wrong.");
		t = 0;
	}
	if(send(acceptfd,msg,sizeof(MSG),0) < 0){
		perror("send");
		return -1;
	}
	return t;
}

int do_searchword(int acceptfd,MSG *msg,char *word){
	FILE *fp;
	int len = 0;
	char temp[512] = {0};
	int result;
	char *p;
	if((fp = fopen("dict.txt","r")) == NULL){
		perror("fopen");
		strcpy(msg->data,"Failed to open dict.txt.");
		send(acceptfd,msg,sizeof(MSG),0);
		return -1;
	}
	len = strlen(word);

	while(fgets(temp,512,fp) != NULL){
		result = strncmp(temp,word,len);
		if(result > 0){
			continue;
		}
		if(result == 0){
			if(temp[len] != ' '){
				break;
			}else{	
				p = temp+len;
				while(*p == ' '){
					p++;
				}
				strcpy(msg->data,p);
				fclose(fp);
				return 1;
			}
		}
	}
	fclose(fp);
	return 0;
}
int get_date(char *date){
	time_t t;
	struct tm *tp;
	bzero(tp,sizeof(struct tm));
	time(&t);
	localtime_r(&t,tp);
	sprintf(date,"%d-%d-%d %d:%d:%d",tp->tm_year+1900,tp->tm_mon+1,tp->tm_mday,tp->tm_hour,tp->tm_min,tp->tm_sec);
	return 0;
}

int do_query(int acceptfd,MSG *msg,sqlite3 *db){
	char word[64] = {0};
	int found = 0;
	char date[256] = {0};
	char sql[128] = {0};
	char *errmsg;
	time_t t;
	struct tm *tp;
	t = time(NULL);
	tp = localtime(&t);
	sprintf(date,"%d-%d-%d %d:%d:%d",tp->tm_year+1900,tp->tm_mon+1,tp->tm_mday,tp->tm_hour,tp->tm_min,tp->tm_sec);
	strcpy(word,msg->data);

	found = do_searchword(acceptfd,msg,word);
	if(found == 1)
	{
		//get_date(date);
		puts(date);

		sprintf(sql,"insert into record values('%s','%s','%s')",msg->name,date,word);
		printf("%s\n",sql);
		if(sqlite3_exec(db,sql,NULL,NULL,&errmsg) != SQLITE_OK){
			printf("%s\n",errmsg);
			return -1;
		}else{
			printf("Insert record done.\n");
		}
	}else{  
		strcpy(msg->data,"Not found!");
	}
	if(send(acceptfd,msg,sizeof(MSG),0) < 0){
		perror("send");
		exit(0);
	}

	return 0;
}

客户端代码：
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sqlite3.h>
#include <unistd.h>
#include <sys/types.h>         
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.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_history(int sockfd,MSG *msg);
int do_query(int sockfd,MSG *msg);
int do_login(int sockfd,MSG *msg);
int do_register(int sockfd,MSG *msg);

//./client 192.168.203.130 10000
int main(int argc, 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("socket");
		return -1;
	}

	bzero(&serveraddr,sizeof(serveraddr));
	serveraddr.sin_family = AF_INET;
	serveraddr.sin_port = htons(atoi(argv[2]));
	serveraddr.sin_addr.s_addr = inet_addr(argv[1]);

	if(connect(sockfd,(struct sockaddr *)&serveraddr,sizeof(serveraddr)) < 0){
		perror("fail to connect");
		return -1;
	}
	
	while(1){
		bzero(&msg,sizeof(msg));
		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;
}
int do_register(int sockfd,MSG *msg){
	
	msg->type = R;

	printf("Please input usr name:");
	scanf("%s",msg->name);
	getchar();
	printf("Please input password:\033[8;33m");
	scanf("%s",msg->data);
//	getchar();
	
	if(send(sockfd,(void *)msg,sizeof(MSG),0) == -1){
		printf("fail to send.\n");
		return -1;
	}
	if(recv(sockfd,msg,sizeof(MSG),0) < 0){
		printf("fail to recv.\n");
		return -1;
	}
	printf("\033[0m %s\n",msg->data);
	return 0;
}

int do_login(int sockfd,MSG *msg){
	
	msg->type = L;

	printf("Please input usr name:");
	scanf("%s",msg->name);
	getchar();
	printf("Please input password:\033[8;33m");
	scanf("%s",msg->data);
//	getchar();

	if(send(sockfd,(void *)msg,sizeof(MSG),0) == -1){
		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("\033[0m Login ok!.\n");
		return 1;
	}else{
		printf("\033[0m %s\n",msg->data);
		return 0;
	}
}

int do_query(int sockfd,MSG *msg){
	msg->type = Q;
	puts("-------------------");
	while(1){
		printf("Input word:");
		scanf("%s",msg->data);
		getchar();
		if(strncmp(msg->data,"#",1) == 0){
			break;
		}
		if(send(sockfd,msg,sizeof(MSG),0) < 0){
			perror("send");
			return -1;
		}
		if(recv(sockfd,msg,sizeof(MSG),0) < 0){
			perror("recv");
			return -1;
		}
		printf("%s\n",msg->data);
	}
	return 0;
}

int do_history(int sockfd,MSG *msg){
	msg->type = H;
	
	send(sockfd,msg,sizeof(MSG),0);

	while(1){
		recv(sockfd,msg,sizeof(MSG),0);
		if(msg->data[0]=='\0')
			break;
		printf("%s\n",msg->data);
	}
	return 0;
}

效果展示：

我们依次运行客户端与服务器：
在这里插入图片描述
在这里插入图片描述
然后：注册新用户
在这里插入图片描述
然后，登录我们注册的用户。
在这里插入图片描述
接着：我们查询单词
在这里插入图片描述
然后，我们查询历史记录
在这里插入图片描述
最后，退出系统。
在这里插入图片描述

Unique--ZGY
关注
  • 1
    点赞
  • 踩
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值