KeqingMoe/math
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

重构用户数据库操作，使用 JSON 存储用户信息，简化用户 ID 处理，增强密码和权限管理功能

Browse Source
This commit is contained in:
keqingmoe 2024-12-29 20:58:54 +08:00
parent ba541e0e64
commit 93779768fc
1 changed files with 44 additions and 51 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

95
src/db/auth.cpp
View File

@ -7,26 +7,13 @@
#include <leveldb/db.h> #include <leveldb/db.h>
#include <leveldb/write_batch.h> #include <leveldb/write_batch.h>
#include <nlohmann/json.hpp>
using leveldb_ptr = leveldb::DB*; using leveldb_ptr = leveldb::DB*;
auto user_db = leveldb_ptr{nullptr}; auto user_db = leveldb_ptr{nullptr};
auto iter_round = 10000; auto iter_round = 10000;
auto filter(const std::string_view id) -> bool
{
return id.contains(':');
}
auto mangle_user_id(const std::string_view id) -> std::string
{
return std::format("{}:user_id", id);
}
auto mangle_permission(const std::string_view id) -> std::string
{
return std::format("{}:permission", id);
}
extern "C" extern "C"
{ {
auto open_user_db() -> int auto open_user_db() -> int
@ -35,7 +22,7 @@ extern "C"
opts.create_if_missing = true; opts.create_if_missing = true;
auto status = leveldb::DB::Open(opts, "db/user", &user_db); auto status = leveldb::DB::Open(opts, "db/users", &user_db);
if (!status.ok()) { if (!status.ok()) {
std::println(stderr, "Failed to open user database: {}", status.ToString()); std::println(stderr, "Failed to open user database: {}", status.ToString());
@ -51,10 +38,8 @@ extern "C"
auto check_user_exists(const char* user_id, int* result) -> int auto check_user_exists(const char* user_id, int* result) -> int
{ {
if (filter(user_id)) return 0;
auto value = std::string{}; auto value = std::string{};
auto status = user_db->Get(leveldb::ReadOptions{}, mangle_user_id(user_id), &value); auto status = user_db->Get(leveldb::ReadOptions{}, user_id, &value);
if (status.ok()) { if (status.ok()) {
*result = 1; *result = 1;
} else if (status.IsNotFound()) { } else if (status.IsNotFound()) {
@ -68,10 +53,16 @@ extern "C"
auto set_user_password(const char* user_id, const char* password) -> int auto set_user_password(const char* user_id, const char* password) -> int
{ {
if (filter(user_id)) return 0; auto value = std::string{};
auto status = user_db->Get(leveldb::ReadOptions{}, user_id, &value);
if (!status.ok()) {
std::println(stderr, "Failed to set user password: {}", status.ToString());
return 0;
}
auto json = nlohmann::json::parse(value);
json["hash"] = generate_hash(password, iter_round);
auto hash = generate_hash(password, iter_round); status = user_db->Put(leveldb::WriteOptions{}, user_id, json.dump());
auto status = user_db->Put(leveldb::WriteOptions{}, mangle_user_id(user_id), hash);
if (!status.ok()) { if (!status.ok()) {
std::println(stderr, "Failed to set user password: {}", status.ToString()); std::println(stderr, "Failed to set user password: {}", status.ToString());
return 0; return 0;
@ -81,26 +72,24 @@ extern "C"
auto login(const char* user_id, const char* password, int* result) -> int auto login(const char* user_id, const char* password, int* result) -> int
{ {
if (filter(user_id)) return 0;
auto value = std::string{}; auto value = std::string{};
auto status = user_db->Get(leveldb::ReadOptions{}, mangle_user_id(user_id), &value); auto status = user_db->Get(leveldb::ReadOptions{}, user_id, &value);
if (!status.ok()) { if (!status.ok()) {
if (!status.IsNotFound()) std::println(stderr, "Failed to login: {}", status.ToString()); if (!status.IsNotFound()) std::println(stderr, "Failed to login: {}", status.ToString());
return 0; return 0;
} }
*result = validate_password(password, value.data()); auto json = nlohmann::json::parse(value);
*result = validate_password(password, json["hash"]);
return 1; return 1;
} }
int registe(const char* user_id, const char* password) int registe(const char* user_id, const char* password)
{ {
if (filter(user_id)) return 0; auto json = nlohmann::json{
{"hash", generate_hash(password, iter_round)},
auto batch = leveldb::WriteBatch{}; {"permission", 2}
batch.Put(mangle_user_id(user_id), generate_hash(password, iter_round)); };
batch.Put(mangle_permission(user_id), "2"); auto status = user_db->Put(leveldb::WriteOptions{}, user_id, json.dump());
auto status = user_db->Write(leveldb::WriteOptions{}, &batch);
if (!status.ok()) { if (!status.ok()) {
std::println(stderr, "Failed to register: {}", status.ToString()); std::println(stderr, "Failed to register: {}", status.ToString());
return 0; return 0;
@ -110,12 +99,7 @@ extern "C"
auto delete_user(const char* user_id) -> int auto delete_user(const char* user_id) -> int
{ {
if (filter(user_id)) return 0; auto status = user_db->Delete(leveldb::WriteOptions{}, user_id);
auto batch = leveldb::WriteBatch{};
batch.Delete(mangle_user_id(user_id));
batch.Delete(mangle_permission(user_id));
auto status = user_db->Write(leveldb::WriteOptions{}, &batch);
if (!status.ok()) { if (!status.ok()) {
std::println(stderr, "Failed to delete: {}", status.ToString()); std::println(stderr, "Failed to delete: {}", status.ToString());
return 0; return 0;
@ -125,24 +109,29 @@ extern "C"
int get_user_permission(const char* user_id, int* result) int get_user_permission(const char* user_id, int* result)
{ {
if (filter(user_id)) return 0;
auto value = std::string{}; auto value = std::string{};
auto status = user_db->Get(leveldb::ReadOptions{}, mangle_permission(user_id), &value); auto status = user_db->Get(leveldb::ReadOptions{}, user_id, &value);
if (status.ok()) { if (!status.ok()) {
*result = std::stoi(value);
} else {
if (!status.IsNotFound()) std::println(stderr, "Failed to get user permission: {}", status.ToString()); if (!status.IsNotFound()) std::println(stderr, "Failed to get user permission: {}", status.ToString());
return 0; return 0;
} }
auto json = nlohmann::json::parse(value);
json["permission"].get_to(*result);
return 1; return 1;
} }
int set_user_permission(const char* user_id, int permission) int set_user_permission(const char* user_id, int permission)
{ {
if (filter(user_id)) return 0; auto value = std::string{};
auto status = user_db->Get(leveldb::ReadOptions{}, user_id, &value);
if (!status.ok()) {
std::println(stderr, "Failed to set user password: {}", status.ToString());
return 0;
}
auto json = nlohmann::json::parse(value);
json["permission"] = permission;
auto status = user_db->Put(leveldb::WriteOptions{}, mangle_permission(user_id), std::to_string(permission)); status = user_db->Put(leveldb::WriteOptions{}, user_id, json.dump());
if (!status.ok()) { if (!status.ok()) {
std::println(stderr, "Failed to set user permission: {}", status.ToString()); std::println(stderr, "Failed to set user permission: {}", status.ToString());
return 0; return 0;
@ -152,8 +141,10 @@ extern "C"
int set_admin_password(const char* password) int set_admin_password(const char* password)
{ {
auto hash = generate_hash(password, iter_round); auto json = nlohmann::json{
auto status = user_db->Put(leveldb::WriteOptions{}, "admin_password_hash", hash); {"admin_password_hash", generate_hash(password, iter_round)}
};
auto status = user_db->Put(leveldb::WriteOptions{}, "@", json.dump());
if (!status.ok()) { if (!status.ok()) {
std::println(stderr, "Failed to set admin password: {}", status.ToString()); std::println(stderr, "Failed to set admin password: {}", status.ToString());
return 0; return 0;
@ -164,19 +155,21 @@ extern "C"
int admin_login(const char* password, int* result) int admin_login(const char* password, int* result)
{ {
auto value = std::string{}; auto value = std::string{};
auto status = user_db->Get(leveldb::ReadOptions{}, "admin_password_hash", &value); auto status = user_db->Get(leveldb::ReadOptions{}, "@", &value);
if (!status.ok()) { if (!status.ok()) {
std::println(stderr, "Failed to login admin: {}", status.ToString()); std::println(stderr, "Failed to login admin: {}", status.ToString());
return 0; return 0;
} }
*result = validate_password(password, value.data()); auto json = nlohmann::json::parse(value);
json["admin_password_hash"].get_to(value);
*result = validate_password(password, value);
return 1; return 1;
} }
int has_admin_password(int* result) int has_admin_password(int* result)
{ {
auto value = std::string{}; auto value = std::string{};
auto status = user_db->Get(leveldb::ReadOptions{}, "admin_password_hash", &value); auto status = user_db->Get(leveldb::ReadOptions{}, "@", &value);
if (status.ok()) { if (status.ok()) {
*result = 1; *result = 1;
} else if (status.IsNotFound()) { } else if (status.IsNotFound()) {