Add LevelDB auth database. (#9476)

* Add leveldb auth database.
This commit is contained in:
luk3yx 2020-04-23 23:07:19 +12:00 committed by GitHub
parent ce5b0932f8
commit 914dbeaa0b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 122 additions and 1 deletions

@ -26,6 +26,7 @@ with this program; if not, write to the Free Software Foundation, Inc.,
#include "log.h"
#include "filesys.h"
#include "exceptions.h"
#include "util/serialize.h"
#include "util/string.h"
#include "leveldb/db.h"
@ -97,5 +98,100 @@ void Database_LevelDB::listAllLoadableBlocks(std::vector<v3s16> &dst)
delete it;
}
#endif // USE_LEVELDB
AuthDatabaseLevelDB::AuthDatabaseLevelDB(const std::string &savedir)
{
leveldb::Options options;
options.create_if_missing = true;
leveldb::Status status = leveldb::DB::Open(options,
savedir + DIR_DELIM + "auth.db", &m_database);
ENSURE_STATUS_OK(status);
}
AuthDatabaseLevelDB::~AuthDatabaseLevelDB()
{
delete m_database;
}
bool AuthDatabaseLevelDB::getAuth(const std::string &name, AuthEntry &res)
{
std::string raw;
leveldb::Status s = m_database->Get(leveldb::ReadOptions(), name, &raw);
if (!s.ok())
return false;
std::istringstream is(raw);
/*
u8 version = 1
std::string password
u16 number of privileges
for each privilege {
std::string privilege
}
s64 last_login
*/
if (readU8(is) > 1)
return false;
res.id = 1;
res.name = name;
res.password = deSerializeString(is);
u16 privilege_count = readU16(is);
res.privileges.clear();
res.privileges.reserve(privilege_count);
for (u16 i = 0; i < privilege_count; i++) {
res.privileges.push_back(deSerializeString(is));
}
res.last_login = readS64(is);
return true;
}
bool AuthDatabaseLevelDB::saveAuth(const AuthEntry &authEntry)
{
std::ostringstream os;
writeU8(os, 1);
os << serializeString(authEntry.password);
size_t privilege_count = authEntry.privileges.size();
FATAL_ERROR_IF(privilege_count > U16_MAX,
"Unsupported number of privileges");
writeU16(os, privilege_count);
for (const std::string &privilege : authEntry.privileges) {
os << serializeString(privilege);
}
writeS64(os, authEntry.last_login);
leveldb::Status s = m_database->Put(leveldb::WriteOptions(),
authEntry.name, os.str());
return s.ok();
}
bool AuthDatabaseLevelDB::createAuth(AuthEntry &authEntry)
{
return saveAuth(authEntry);
}
bool AuthDatabaseLevelDB::deleteAuth(const std::string &name)
{
leveldb::Status s = m_database->Delete(leveldb::WriteOptions(), name);
return s.ok();
}
void AuthDatabaseLevelDB::listNames(std::vector<std::string> &res)
{
leveldb::Iterator* it = m_database->NewIterator(leveldb::ReadOptions());
res.clear();
for (it->SeekToFirst(); it->Valid(); it->Next()) {
res.emplace_back(it->key().ToString());
}
delete it;
}
void AuthDatabaseLevelDB::reload()
{
// No-op for LevelDB.
}
#endif // USE_LEVELDB

@ -45,4 +45,21 @@ private:
leveldb::DB *m_database;
};
class AuthDatabaseLevelDB : public AuthDatabase
{
public:
AuthDatabaseLevelDB(const std::string &savedir);
virtual ~AuthDatabaseLevelDB();
virtual bool getAuth(const std::string &name, AuthEntry &res);
virtual bool saveAuth(const AuthEntry &authEntry);
virtual bool createAuth(AuthEntry &authEntry);
virtual bool deleteAuth(const std::string &name);
virtual void listNames(std::vector<std::string> &res);
virtual void reload();
private:
leveldb::DB *m_database;
};
#endif // USE_LEVELDB

@ -44,6 +44,9 @@ with this program; if not, write to the Free Software Foundation, Inc.,
#if USE_POSTGRESQL
#include "database/database-postgresql.h"
#endif
#if USE_LEVELDB
#include "database/database-leveldb.h"
#endif
#include "server/luaentity_sao.h"
#include "server/player_sao.h"
@ -2187,6 +2190,11 @@ AuthDatabase *ServerEnvironment::openAuthDatabase(
if (name == "files")
return new AuthDatabaseFiles(savedir);
#if USE_LEVELDB
if (name == "leveldb")
return new AuthDatabaseLevelDB(savedir);
#endif
throw BaseException(std::string("Database backend ") + name + " not supported.");
}