114 lines
3.7 KiB
C
114 lines
3.7 KiB
C
#include <cjson/cJSON.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include "includes/sqlite3.h"
|
|
|
|
// cJSON *result_json; // 用于存储查询结果
|
|
|
|
// sqlite3的回调函数
|
|
// sqlite 每查到一条记录，就调用一次这个回调
|
|
int sql_search_callback(void* NotUsed, int argc, char** argv, char** azColName);
|
|
|
|
int main(int argc, char const* argv[]) {
|
|
// result_json = cJSON_CreateObject(); // 创建 JSON 对象
|
|
|
|
// 连接到 SQLite 数据库
|
|
sqlite3* db; // 声明 sqlite3 类型变量，即声明一个数据库对象
|
|
char* zErrMsg = 0; // 用于存储错误信息
|
|
int rc; // 用于存储函数返回值
|
|
|
|
rc = sqlite3_open(
|
|
"../sql_base/green_house.db",
|
|
&db); // 打开指定的数据库文件，如果不存在将创建一个同名的数据库文件
|
|
if (rc) {
|
|
fprintf(stderr, "Can't open database: %s\n", sqlite3_errmsg(db));
|
|
sqlite3_close(db);
|
|
return 1;
|
|
}
|
|
fprintf(stderr, "ok Opened database successfully\n");
|
|
|
|
// 获取前端传递的 JSON 数据
|
|
char request_data[128] = "";
|
|
fgets(request_data, 128, stdin); // 从标准输入中读取一行数据
|
|
|
|
// 解析 JSON 数据
|
|
cJSON* json_buf = cJSON_Parse(request_data); // 解析 JSON 数据
|
|
if (json_buf == NULL) {
|
|
fprintf(stderr, "Error parsing JSON data\n");
|
|
sqlite3_close(db);
|
|
return 1;
|
|
}
|
|
|
|
// 获取用户名和密码
|
|
cJSON* user = cJSON_GetObjectItemCaseSensitive(json_buf, "name");
|
|
cJSON* password = cJSON_GetObjectItemCaseSensitive(json_buf, "password");
|
|
|
|
if (user == NULL || password == NULL || !cJSON_IsString(user) ||
|
|
!cJSON_IsString(password)) { // 是否是非法输入
|
|
fprintf(stderr,
|
|
"Error getting username and/or password from JSON data\n");
|
|
cJSON_Delete(json_buf);
|
|
sqlite3_close(db);
|
|
return 1;
|
|
}
|
|
|
|
// 拿到用户名和密码
|
|
char* username_from_frontend = user->valuestring;
|
|
char* password_from_frontend = password->valuestring;
|
|
|
|
printf("username is %s, pwd is %s\n", username_from_frontend,
|
|
password_from_frontend);
|
|
|
|
// 构建查询语句
|
|
char query_sql[128] = "";
|
|
sprintf(query_sql,
|
|
"insert into login_users (username,password) values('%s','%s');",
|
|
username_from_frontend, password_from_frontend); // 构建插入语句
|
|
|
|
printf("query_sql is %s\n", query_sql);
|
|
|
|
// 执行查询语句
|
|
rc = sqlite3_exec(db, query_sql, NULL, NULL, &zErrMsg); // 执行查询语句
|
|
// rc = sqlite3_exec(db, query_sql, sql_search_callback, 0, &zErrMsg); //
|
|
// 执行查询语句
|
|
if (rc != SQLITE_OK) {
|
|
fprintf(stderr, "SQL error: %s\n", zErrMsg);
|
|
sqlite3_free(zErrMsg);
|
|
cJSON_Delete(json_buf);
|
|
sqlite3_close(db);
|
|
return 1;
|
|
}
|
|
|
|
// 释放资源
|
|
cJSON_Delete(json_buf);
|
|
sqlite3_close(db);
|
|
|
|
return 0;
|
|
}
|
|
|
|
int sql_search_callback(void* NotUsed,
|
|
int argc,
|
|
char** argv,
|
|
char** azColName) {
|
|
// int i;
|
|
// for (i = 0; i < argc; i++)
|
|
// {
|
|
// printf("%s = %s\n", azColName[i], argv[i] ? argv[i] : "NULL");
|
|
// }
|
|
cJSON* result_json = cJSON_CreateObject();
|
|
if (argc == 0) {
|
|
cJSON_AddNumberToObject(result_json, "code", 1);
|
|
cJSON_AddStringToObject(result_json, "msg", "用户名或密码错误");
|
|
} else
|
|
cJSON_AddNumberToObject(result_json, "code", 0);
|
|
|
|
char* result_json_string = cJSON_Print(result_json);
|
|
|
|
// 输出 JSON 结果
|
|
printf("content-type: application/json;charset=utf-8\r\n");
|
|
printf("\r\n");
|
|
printf("%s\n", result_json_string);
|
|
|
|
free(result_json_string);
|
|
return SQLITE_OK;
|
|
} |