Add LevelDB auth database. (#9476)

* Add leveldb auth database.

diff --git a/src/database/database-leveldb.cpp b/src/database/database-leveldb.cpp
index 4a4904c6a0b122f58a5ee217577236b22cf767fb..1aab4c43d06ae7fee949bd33287636b451fa948f 100644 (file)
--- a/src/database/database-leveldb.cpp
+++ b/src/database/database-leveldb.cpp
@@ -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

diff --git a/src/database/database-leveldb.h b/src/database/database-leveldb.h
index d30f9f8f5dd9f733fb98bd879f603c681d0b8884..a9bd0faa486dbf005e5de7447b804a1b300622d0 100644 (file)
--- a/src/database/database-leveldb.h
+++ b/src/database/database-leveldb.h
@@ -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

diff --git a/src/serverenvironment.cpp b/src/serverenvironment.cpp
index 27f0c1e3d0195253d2ee778210418edb620b5163..08d7961189b48e5a25149c7971406427d6cc9a2b 100644 (file)
--- a/src/serverenvironment.cpp
+++ b/src/serverenvironment.cpp
@@ -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.");
 }