Linux套接字+Sqlite实例：客户端-服务器应用程序代码

服务器

#include
#include
#include
#include
#include
#include
#include
#include 
#include
#include
#include

#define N 32
#define R 1 // 用户注册
#define L 2 // 用户登录
#define Q 3 // 单词查询
#define H 4 // 查询历史记录
#define DATABASE "my.db" // 数据库名称

// 消息结构体，用于客户端与服务器之间通信
typedef struct {
	int type;          // 消息类型（注册、登录、查询、历史记录）
	char name[32];     // 用户名
	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_query(int acceptfd, MSG *msg, sqlite3 *db);
int do_history(int acceptfd, MSG *msg, sqlite3 *db);
int do_searchword(int acceptfd, MSG *msg, char word[]);
int get_date(char *date);

// 主函数，启动服务器并监听客户端连接
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)
	{
		perror("fail to listen");
	}

	// 避免僵尸进程
	signal(SIGCHLD, SIG_IGN);

	// 主循环，接受客户端连接并 fork 子进程处理
	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");
		}
	}

	close(acceptfd);
	exit(0);
	return 0;
}

// 处理用户注册请求
void do_register(int acceptfd, MSG *msg, sqlite3 *db)
{
	printf("do_register\n");
	char *errmsg;
	char sql[512];

	// 构建插入 SQL 语句
	snprintf(sql, sizeof(sql), "insert into usr values('%s', '%s');", msg->name, msg->data);
	printf("%s\n", sql);

	// 执行 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, msg, sizeof(MSG), 0) < 0)
	{
		perror("fail to send");
		return;
	}
}

// 处理用户登录请求
int do_login(int acceptfd, MSG *msg , sqlite3 *db)
{
	char sql[512] = {};
	char *errmsg;
	int nrow, ncloumn;
	char **resultp;

	// 构造查询语句
	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");
	}

	// 如果查到1行，说明用户名密码正确
	if(nrow == 1)
	{
		strcpy(msg->data, "OK");
		send(acceptfd, msg, sizeof(MSG), 0);
		return 1;
	}
	else
	{
		strcpy(msg->data, "usr/passwd wrong.");
		send(acceptfd, msg, sizeof(MSG), 0);
	}
	return 0;
}

// 获取当前系统时间，格式化为字符串
int get_date(char *date)
{
	time_t t;
	struct tm *tp;
	time(&t);
	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);
	return 0;
}

// 处理单词查询请求
int do_query(int acceptfd, MSG *msg , sqlite3 *db)
{
	char word[64];
	int found = 0;
	char date[128] = {};
	char sql[512] = {};
	char *errmsg;

	strcpy(word, msg->data); // 提取要查询的单词

	// 查询单词
	found = do_searchword(acceptfd, msg, word);
	printf("查询一个单词完毕.\n");

	// 如果查到了该单词，则记录历史记录
	if(found == 1)
	{
		get_date(date); // 获取当前时间
		sprintf(sql, "insert into record values('%s', '%s', '%s')", msg->name, date, word);

		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!");
	}

	send(acceptfd, msg, sizeof(MSG), 0); // 把查询结果发送给客户端
	return 0;
}

// 在本地词典文件中搜索单词
int do_searchword(int acceptfd, MSG* msg, char word[])
{
	FILE *fp;
	int len = 0;
	char temp[512] = {};
	int result; 
	char *p;

	if((fp = fopen("dict.txt", "r")) == NULL)
	{
		perror("Fail to fopen");
		strcpy(msg->data, "Failed to open dict.txt");
		send(acceptfd, msg, sizeof(MSG), 0);
		return -1;
	}

	len = strlen(word);
	printf("%s, len = %d\n", word, len);

	// 按行读取并查找匹配单词
	while(fgets(temp, 512, fp) != NULL)
	{
		result = strncmp(temp, word, len);

		if(result < 0)
			continue;

		if(result > 0 || ((result == 0) && (temp[len] != ' ')))
			break;

		// 找到单词
		p = temp + len;
		while(*p == ' ') p++;
		strcpy(msg->data, p);
		printf("found word: %s\n", msg->data);
		fclose(fp);
		return 1;
	}

	fclose(fp);
	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[512] = {};
	char *errmsg;

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

// 执行查询，使用回调函数逐条发送
sqlite3_exec(db, sql语句, 回调函数, 回调函数参数, 错误信息指针);
	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;
}

客户端

#include 
#include 
#include           // 数据类型定义
#include          // 套接字 API
#include          // sockaddr_in 结构体
#include           // inet_addr 函数
#include              // 字符串函数
#include              // close 函数等

#define N 32  // 用户名最大长度

// 定义操作类型常量
#define R 1   // 注册 register
#define L 2   // 登录 login
#define Q 3   // 查询 query
#define H 4   // 查询历史记录 history

// 定义客户端与服务器之间通信的数据结构
typedef struct {
	int type;           // 消息类型：R, L, Q, H
	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;
	}

	// 显示服务器返回的注册结果
	printf("%s\n", msg->data);

	return 0;
}

// 登录功能
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;
	}
	else 
	{
		printf("%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)
		{
			printf("Fail to send.\n");
			return -1;
		}

		// 接收服务器返回的查询结果
		if(recv(sockfd, msg, sizeof(MSG), 0) < 0)
		{
			printf("Fail to recv.\n");
			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;
}

// 主函数，客户端程序入口
// 用法： ./client  
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;
	serveraddr.sin_addr.s_addr = inet_addr(argv[1]);  // 服务器 IP
	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;
}

来自网上开源项目