添加用户认证功能的初步实现

This commit is contained in:
keqingmoe 2024-12-22 22:42:01 +08:00
parent 9cc9b48c18
commit a4aed41ff1
4 changed files with 91 additions and 6 deletions

View File

@ -6,11 +6,11 @@ extern "C"
{
#endif
auto create_token(const char* user_id, const char* secret) noexcept -> char*;
char* create_token(const char* user_id, const char* secret);
auto get_payload(const char* token) noexcept -> char*;
char* get_payload(const char* token);
auto verify_token(const char* token, const char* secret) noexcept -> int;
int verify_token(const char* token, const char* secret);
#ifdef __cplusplus
}

6
include/server/auth.h Normal file
View File

@ -0,0 +1,6 @@
#ifndef SERVER_AUTH_H
#define SERVER_AUTH_H
// Your code here
#endif // SERVER_AUTH_H

79
src/db/auth.cpp Normal file
View File

@ -0,0 +1,79 @@
#include "auth.h"
#include "hash/hash.hpp"
#include <print>
#include <leveldb/db.h>
using leveldb_ptr = leveldb::DB*;
auto user_db = leveldb_ptr{nullptr};
auto iter_round = 10000;
extern "C"
{
auto open_user_db() -> int
{
auto opts = leveldb::Options{};
opts.create_if_missing = true;
auto status = leveldb::DB::Open(opts, "db/user", &user_db);
if (!status.ok()) {
std::println(stderr, "Failed to open user database: {}", status.ToString());
return 0;
}
return 1;
}
auto close_user_db() -> void
{
delete user_db;
}
auto check_user_exists(const char* user_id, int* result) -> int
{
auto value = std::string{};
auto status = user_db->Get(leveldb::ReadOptions{}, user_id, &value);
if (status.ok()) {
*result = 1;
} else if (status.IsNotFound()) {
*result = 0;
} else {
std::println(stderr, "Failed to check user existence: {}", status.ToString());
return 0;
}
return 1;
}
auto set_user_password(const char* user_id, const char* password) -> int
{
auto status = user_db->Put(leveldb::WriteOptions{}, user_id, generate_hash(password, iter_round));
if (!status.ok()) {
std::println(stderr, "Failed to set user password: {}", status.ToString());
return 0;
}
return 1;
}
auto login(const char* user_id, const char* password, int* result) -> int
{
auto value = std::string{};
auto status = user_db->Get(leveldb::ReadOptions{}, user_id, &value);
if (!status.ok()) {
std::println(stderr, "Failed to login: {}", status.ToString());
return 0;
}
*result = validate_password(password, value.data());
return 1;
}
auto logout(const char* user_id) -> int
{
leveldb::WriteOptions write_options;
leveldb::Status status = user_db->Delete(write_options, user_id);
return status.ok();
}
}

View File

@ -8,7 +8,7 @@ using namespace std::literals::chrono_literals;
extern "C"
{
auto create_token(const char* user_id, const char* secret) noexcept -> char*
auto create_token(const char* user_id, const char* secret) -> char*
{
auto token = jwt::create()
.set_type("JWS")
@ -21,14 +21,14 @@ extern "C"
return strdup(token.c_str());
}
auto get_payload(const char* token) noexcept -> char*
auto get_payload(const char* token) -> char*
{
auto decoded_token = jwt::decode(token);
auto payload = decoded_token.get_payload_claim("user_id").as_string();
return strdup(payload.c_str());
}
auto verify_token(const char* token, const char* secret) noexcept -> int
auto verify_token(const char* token, const char* secret) -> int
{
try {
auto decoded_token = jwt::decode(token);