database-sqlite3.cpp 25 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782
  1. /*
  2. Minetest
  3. Copyright (C) 2013 celeron55, Perttu Ahola <celeron55@gmail.com>
  4. This program is free software; you can redistribute it and/or modify
  5. it under the terms of the GNU Lesser General Public License as published by
  6. the Free Software Foundation; either version 2.1 of the License, or
  7. (at your option) any later version.
  8. This program is distributed in the hope that it will be useful,
  9. but WITHOUT ANY WARRANTY; without even the implied warranty of
  10. MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  11. GNU Lesser General Public License for more details.
  12. You should have received a copy of the GNU Lesser General Public License along
  13. with this program; if not, write to the Free Software Foundation, Inc.,
  14. 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  15. */
  16. /*
  17. SQLite format specification:
  18. blocks:
  19. (PK) INT id
  20. BLOB data
  21. */
  22. #include "database-sqlite3.h"
  23. #include "log.h"
  24. #include "filesys.h"
  25. #include "exceptions.h"
  26. #include "settings.h"
  27. #include "porting.h"
  28. #include "util/string.h"
  29. #include "remoteplayer.h"
  30. #include "server/player_sao.h"
  31. #include <cassert>
  32. // When to print messages when the database is being held locked by another process
  33. // Note: I've seen occasional delays of over 250ms while running minetestmapper.
  34. #define BUSY_INFO_TRESHOLD 100 // Print first informational message after 100ms.
  35. #define BUSY_WARNING_TRESHOLD 250 // Print warning message after 250ms. Lag is increased.
  36. #define BUSY_ERROR_TRESHOLD 1000 // Print error message after 1000ms. Significant lag.
  37. #define BUSY_FATAL_TRESHOLD 3000 // Allow SQLITE_BUSY to be returned, which will cause a minetest crash.
  38. #define BUSY_ERROR_INTERVAL 10000 // Safety net: report again every 10 seconds
  39. #define SQLRES(s, r, m) \
  40. if ((s) != (r)) { \
  41. throw DatabaseException(std::string(m) + ": " +\
  42. sqlite3_errmsg(m_database)); \
  43. }
  44. #define SQLOK(s, m) SQLRES(s, SQLITE_OK, m)
  45. #define PREPARE_STATEMENT(name, query) \
  46. SQLOK(sqlite3_prepare_v2(m_database, query, -1, &m_stmt_##name, NULL),\
  47. "Failed to prepare query '" query "'")
  48. #define SQLOK_ERRSTREAM(s, m) \
  49. if ((s) != SQLITE_OK) { \
  50. errorstream << (m) << ": " \
  51. << sqlite3_errmsg(m_database) << std::endl; \
  52. }
  53. #define FINALIZE_STATEMENT(statement) SQLOK_ERRSTREAM(sqlite3_finalize(statement), \
  54. "Failed to finalize " #statement)
  55. int Database_SQLite3::busyHandler(void *data, int count)
  56. {
  57. s64 &first_time = reinterpret_cast<s64 *>(data)[0];
  58. s64 &prev_time = reinterpret_cast<s64 *>(data)[1];
  59. s64 cur_time = porting::getTimeMs();
  60. if (count == 0) {
  61. first_time = cur_time;
  62. prev_time = first_time;
  63. } else {
  64. while (cur_time < prev_time)
  65. cur_time += s64(1)<<32;
  66. }
  67. if (cur_time - first_time < BUSY_INFO_TRESHOLD) {
  68. ; // do nothing
  69. } else if (cur_time - first_time >= BUSY_INFO_TRESHOLD &&
  70. prev_time - first_time < BUSY_INFO_TRESHOLD) {
  71. infostream << "SQLite3 database has been locked for "
  72. << cur_time - first_time << " ms." << std::endl;
  73. } else if (cur_time - first_time >= BUSY_WARNING_TRESHOLD &&
  74. prev_time - first_time < BUSY_WARNING_TRESHOLD) {
  75. warningstream << "SQLite3 database has been locked for "
  76. << cur_time - first_time << " ms." << std::endl;
  77. } else if (cur_time - first_time >= BUSY_ERROR_TRESHOLD &&
  78. prev_time - first_time < BUSY_ERROR_TRESHOLD) {
  79. errorstream << "SQLite3 database has been locked for "
  80. << cur_time - first_time << " ms; this causes lag." << std::endl;
  81. } else if (cur_time - first_time >= BUSY_FATAL_TRESHOLD &&
  82. prev_time - first_time < BUSY_FATAL_TRESHOLD) {
  83. errorstream << "SQLite3 database has been locked for "
  84. << cur_time - first_time << " ms - giving up!" << std::endl;
  85. } else if ((cur_time - first_time) / BUSY_ERROR_INTERVAL !=
  86. (prev_time - first_time) / BUSY_ERROR_INTERVAL) {
  87. // Safety net: keep reporting every BUSY_ERROR_INTERVAL
  88. errorstream << "SQLite3 database has been locked for "
  89. << (cur_time - first_time) / 1000 << " seconds!" << std::endl;
  90. }
  91. prev_time = cur_time;
  92. // Make sqlite transaction fail if delay exceeds BUSY_FATAL_TRESHOLD
  93. return cur_time - first_time < BUSY_FATAL_TRESHOLD;
  94. }
  95. Database_SQLite3::Database_SQLite3(const std::string &savedir, const std::string &dbname) :
  96. m_savedir(savedir),
  97. m_dbname(dbname)
  98. {
  99. }
  100. void Database_SQLite3::beginSave()
  101. {
  102. verifyDatabase();
  103. SQLRES(sqlite3_step(m_stmt_begin), SQLITE_DONE,
  104. "Failed to start SQLite3 transaction");
  105. sqlite3_reset(m_stmt_begin);
  106. }
  107. void Database_SQLite3::endSave()
  108. {
  109. verifyDatabase();
  110. SQLRES(sqlite3_step(m_stmt_end), SQLITE_DONE,
  111. "Failed to commit SQLite3 transaction");
  112. sqlite3_reset(m_stmt_end);
  113. }
  114. void Database_SQLite3::openDatabase()
  115. {
  116. if (m_database) return;
  117. std::string dbp = m_savedir + DIR_DELIM + m_dbname + ".sqlite";
  118. // Open the database connection
  119. if (!fs::CreateAllDirs(m_savedir)) {
  120. infostream << "Database_SQLite3: Failed to create directory \""
  121. << m_savedir << "\"" << std::endl;
  122. throw FileNotGoodException("Failed to create database "
  123. "save directory");
  124. }
  125. bool needs_create = !fs::PathExists(dbp);
  126. SQLOK(sqlite3_open_v2(dbp.c_str(), &m_database,
  127. SQLITE_OPEN_READWRITE | SQLITE_OPEN_CREATE, NULL),
  128. std::string("Failed to open SQLite3 database file ") + dbp);
  129. SQLOK(sqlite3_busy_handler(m_database, Database_SQLite3::busyHandler,
  130. m_busy_handler_data), "Failed to set SQLite3 busy handler");
  131. if (needs_create) {
  132. createDatabase();
  133. }
  134. std::string query_str = std::string("PRAGMA synchronous = ")
  135. + itos(g_settings->getU16("sqlite_synchronous"));
  136. SQLOK(sqlite3_exec(m_database, query_str.c_str(), NULL, NULL, NULL),
  137. "Failed to modify sqlite3 synchronous mode");
  138. SQLOK(sqlite3_exec(m_database, "PRAGMA foreign_keys = ON", NULL, NULL, NULL),
  139. "Failed to enable sqlite3 foreign key support");
  140. }
  141. void Database_SQLite3::verifyDatabase()
  142. {
  143. if (m_initialized) return;
  144. openDatabase();
  145. PREPARE_STATEMENT(begin, "BEGIN;");
  146. PREPARE_STATEMENT(end, "COMMIT;");
  147. initStatements();
  148. m_initialized = true;
  149. }
  150. Database_SQLite3::~Database_SQLite3()
  151. {
  152. FINALIZE_STATEMENT(m_stmt_begin)
  153. FINALIZE_STATEMENT(m_stmt_end)
  154. SQLOK_ERRSTREAM(sqlite3_close(m_database), "Failed to close database");
  155. }
  156. /*
  157. * Map database
  158. */
  159. MapDatabaseSQLite3::MapDatabaseSQLite3(const std::string &savedir):
  160. Database_SQLite3(savedir, "map"),
  161. MapDatabase()
  162. {
  163. }
  164. MapDatabaseSQLite3::~MapDatabaseSQLite3()
  165. {
  166. FINALIZE_STATEMENT(m_stmt_read)
  167. FINALIZE_STATEMENT(m_stmt_write)
  168. FINALIZE_STATEMENT(m_stmt_list)
  169. FINALIZE_STATEMENT(m_stmt_delete)
  170. }
  171. void MapDatabaseSQLite3::createDatabase()
  172. {
  173. assert(m_database); // Pre-condition
  174. SQLOK(sqlite3_exec(m_database,
  175. "CREATE TABLE IF NOT EXISTS `blocks` (\n"
  176. " `pos` INT PRIMARY KEY,\n"
  177. " `data` BLOB\n"
  178. ");\n",
  179. NULL, NULL, NULL),
  180. "Failed to create database table");
  181. }
  182. void MapDatabaseSQLite3::initStatements()
  183. {
  184. PREPARE_STATEMENT(read, "SELECT `data` FROM `blocks` WHERE `pos` = ? LIMIT 1");
  185. #ifdef __ANDROID__
  186. PREPARE_STATEMENT(write, "INSERT INTO `blocks` (`pos`, `data`) VALUES (?, ?)");
  187. #else
  188. PREPARE_STATEMENT(write, "REPLACE INTO `blocks` (`pos`, `data`) VALUES (?, ?)");
  189. #endif
  190. PREPARE_STATEMENT(delete, "DELETE FROM `blocks` WHERE `pos` = ?");
  191. PREPARE_STATEMENT(list, "SELECT `pos` FROM `blocks`");
  192. verbosestream << "ServerMap: SQLite3 database opened." << std::endl;
  193. }
  194. inline void MapDatabaseSQLite3::bindPos(sqlite3_stmt *stmt, const v3s16 &pos, int index)
  195. {
  196. SQLOK(sqlite3_bind_int64(stmt, index, getBlockAsInteger(pos)),
  197. "Internal error: failed to bind query at " __FILE__ ":" TOSTRING(__LINE__));
  198. }
  199. bool MapDatabaseSQLite3::deleteBlock(const v3s16 &pos)
  200. {
  201. verifyDatabase();
  202. bindPos(m_stmt_delete, pos);
  203. bool good = sqlite3_step(m_stmt_delete) == SQLITE_DONE;
  204. sqlite3_reset(m_stmt_delete);
  205. if (!good) {
  206. warningstream << "deleteBlock: Block failed to delete "
  207. << PP(pos) << ": " << sqlite3_errmsg(m_database) << std::endl;
  208. }
  209. return good;
  210. }
  211. bool MapDatabaseSQLite3::saveBlock(const v3s16 &pos, const std::string &data)
  212. {
  213. verifyDatabase();
  214. #ifdef __ANDROID__
  215. /**
  216. * Note: For some unknown reason SQLite3 fails to REPLACE blocks on Android,
  217. * deleting them and then inserting works.
  218. */
  219. bindPos(m_stmt_read, pos);
  220. if (sqlite3_step(m_stmt_read) == SQLITE_ROW) {
  221. deleteBlock(pos);
  222. }
  223. sqlite3_reset(m_stmt_read);
  224. #endif
  225. bindPos(m_stmt_write, pos);
  226. SQLOK(sqlite3_bind_blob(m_stmt_write, 2, data.data(), data.size(), NULL),
  227. "Internal error: failed to bind query at " __FILE__ ":" TOSTRING(__LINE__));
  228. SQLRES(sqlite3_step(m_stmt_write), SQLITE_DONE, "Failed to save block")
  229. sqlite3_reset(m_stmt_write);
  230. return true;
  231. }
  232. void MapDatabaseSQLite3::loadBlock(const v3s16 &pos, std::string *block)
  233. {
  234. verifyDatabase();
  235. bindPos(m_stmt_read, pos);
  236. if (sqlite3_step(m_stmt_read) != SQLITE_ROW) {
  237. sqlite3_reset(m_stmt_read);
  238. return;
  239. }
  240. const char *data = (const char *) sqlite3_column_blob(m_stmt_read, 0);
  241. size_t len = sqlite3_column_bytes(m_stmt_read, 0);
  242. if (data)
  243. block->assign(data, len);
  244. else
  245. block->clear();
  246. sqlite3_step(m_stmt_read);
  247. // We should never get more than 1 row, so ok to reset
  248. sqlite3_reset(m_stmt_read);
  249. }
  250. void MapDatabaseSQLite3::listAllLoadableBlocks(std::vector<v3s16> &dst)
  251. {
  252. verifyDatabase();
  253. while (sqlite3_step(m_stmt_list) == SQLITE_ROW)
  254. dst.push_back(getIntegerAsBlock(sqlite3_column_int64(m_stmt_list, 0)));
  255. sqlite3_reset(m_stmt_list);
  256. }
  257. /*
  258. * Player Database
  259. */
  260. PlayerDatabaseSQLite3::PlayerDatabaseSQLite3(const std::string &savedir):
  261. Database_SQLite3(savedir, "players"),
  262. PlayerDatabase()
  263. {
  264. }
  265. PlayerDatabaseSQLite3::~PlayerDatabaseSQLite3()
  266. {
  267. FINALIZE_STATEMENT(m_stmt_player_load)
  268. FINALIZE_STATEMENT(m_stmt_player_add)
  269. FINALIZE_STATEMENT(m_stmt_player_update)
  270. FINALIZE_STATEMENT(m_stmt_player_remove)
  271. FINALIZE_STATEMENT(m_stmt_player_list)
  272. FINALIZE_STATEMENT(m_stmt_player_add_inventory)
  273. FINALIZE_STATEMENT(m_stmt_player_add_inventory_items)
  274. FINALIZE_STATEMENT(m_stmt_player_remove_inventory)
  275. FINALIZE_STATEMENT(m_stmt_player_remove_inventory_items)
  276. FINALIZE_STATEMENT(m_stmt_player_load_inventory)
  277. FINALIZE_STATEMENT(m_stmt_player_load_inventory_items)
  278. FINALIZE_STATEMENT(m_stmt_player_metadata_load)
  279. FINALIZE_STATEMENT(m_stmt_player_metadata_add)
  280. FINALIZE_STATEMENT(m_stmt_player_metadata_remove)
  281. };
  282. void PlayerDatabaseSQLite3::createDatabase()
  283. {
  284. assert(m_database); // Pre-condition
  285. SQLOK(sqlite3_exec(m_database,
  286. "CREATE TABLE IF NOT EXISTS `player` ("
  287. "`name` VARCHAR(50) NOT NULL,"
  288. "`pitch` NUMERIC(11, 4) NOT NULL,"
  289. "`yaw` NUMERIC(11, 4) NOT NULL,"
  290. "`posX` NUMERIC(11, 4) NOT NULL,"
  291. "`posY` NUMERIC(11, 4) NOT NULL,"
  292. "`posZ` NUMERIC(11, 4) NOT NULL,"
  293. "`hp` INT NOT NULL,"
  294. "`breath` INT NOT NULL,"
  295. "`creation_date` DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP,"
  296. "`modification_date` DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP,"
  297. "PRIMARY KEY (`name`));",
  298. NULL, NULL, NULL),
  299. "Failed to create player table");
  300. SQLOK(sqlite3_exec(m_database,
  301. "CREATE TABLE IF NOT EXISTS `player_metadata` ("
  302. " `player` VARCHAR(50) NOT NULL,"
  303. " `metadata` VARCHAR(256) NOT NULL,"
  304. " `value` TEXT,"
  305. " PRIMARY KEY(`player`, `metadata`),"
  306. " FOREIGN KEY (`player`) REFERENCES player (`name`) ON DELETE CASCADE );",
  307. NULL, NULL, NULL),
  308. "Failed to create player metadata table");
  309. SQLOK(sqlite3_exec(m_database,
  310. "CREATE TABLE IF NOT EXISTS `player_inventories` ("
  311. " `player` VARCHAR(50) NOT NULL,"
  312. " `inv_id` INT NOT NULL,"
  313. " `inv_width` INT NOT NULL,"
  314. " `inv_name` TEXT NOT NULL DEFAULT '',"
  315. " `inv_size` INT NOT NULL,"
  316. " PRIMARY KEY(player, inv_id),"
  317. " FOREIGN KEY (`player`) REFERENCES player (`name`) ON DELETE CASCADE );",
  318. NULL, NULL, NULL),
  319. "Failed to create player inventory table");
  320. SQLOK(sqlite3_exec(m_database,
  321. "CREATE TABLE `player_inventory_items` ("
  322. " `player` VARCHAR(50) NOT NULL,"
  323. " `inv_id` INT NOT NULL,"
  324. " `slot_id` INT NOT NULL,"
  325. " `item` TEXT NOT NULL DEFAULT '',"
  326. " PRIMARY KEY(player, inv_id, slot_id),"
  327. " FOREIGN KEY (`player`) REFERENCES player (`name`) ON DELETE CASCADE );",
  328. NULL, NULL, NULL),
  329. "Failed to create player inventory items table");
  330. }
  331. void PlayerDatabaseSQLite3::initStatements()
  332. {
  333. PREPARE_STATEMENT(player_load, "SELECT `pitch`, `yaw`, `posX`, `posY`, `posZ`, `hp`, "
  334. "`breath`"
  335. "FROM `player` WHERE `name` = ?")
  336. PREPARE_STATEMENT(player_add, "INSERT INTO `player` (`name`, `pitch`, `yaw`, `posX`, "
  337. "`posY`, `posZ`, `hp`, `breath`) VALUES (?, ?, ?, ?, ?, ?, ?, ?)")
  338. PREPARE_STATEMENT(player_update, "UPDATE `player` SET `pitch` = ?, `yaw` = ?, "
  339. "`posX` = ?, `posY` = ?, `posZ` = ?, `hp` = ?, `breath` = ?, "
  340. "`modification_date` = CURRENT_TIMESTAMP WHERE `name` = ?")
  341. PREPARE_STATEMENT(player_remove, "DELETE FROM `player` WHERE `name` = ?")
  342. PREPARE_STATEMENT(player_list, "SELECT `name` FROM `player`")
  343. PREPARE_STATEMENT(player_add_inventory, "INSERT INTO `player_inventories` "
  344. "(`player`, `inv_id`, `inv_width`, `inv_name`, `inv_size`) VALUES (?, ?, ?, ?, ?)")
  345. PREPARE_STATEMENT(player_add_inventory_items, "INSERT INTO `player_inventory_items` "
  346. "(`player`, `inv_id`, `slot_id`, `item`) VALUES (?, ?, ?, ?)")
  347. PREPARE_STATEMENT(player_remove_inventory, "DELETE FROM `player_inventories` "
  348. "WHERE `player` = ?")
  349. PREPARE_STATEMENT(player_remove_inventory_items, "DELETE FROM `player_inventory_items` "
  350. "WHERE `player` = ?")
  351. PREPARE_STATEMENT(player_load_inventory, "SELECT `inv_id`, `inv_width`, `inv_name`, "
  352. "`inv_size` FROM `player_inventories` WHERE `player` = ? ORDER BY inv_id")
  353. PREPARE_STATEMENT(player_load_inventory_items, "SELECT `slot_id`, `item` "
  354. "FROM `player_inventory_items` WHERE `player` = ? AND `inv_id` = ?")
  355. PREPARE_STATEMENT(player_metadata_load, "SELECT `metadata`, `value` FROM "
  356. "`player_metadata` WHERE `player` = ?")
  357. PREPARE_STATEMENT(player_metadata_add, "INSERT INTO `player_metadata` "
  358. "(`player`, `metadata`, `value`) VALUES (?, ?, ?)")
  359. PREPARE_STATEMENT(player_metadata_remove, "DELETE FROM `player_metadata` "
  360. "WHERE `player` = ?")
  361. verbosestream << "ServerEnvironment: SQLite3 database opened (players)." << std::endl;
  362. }
  363. bool PlayerDatabaseSQLite3::playerDataExists(const std::string &name)
  364. {
  365. verifyDatabase();
  366. str_to_sqlite(m_stmt_player_load, 1, name);
  367. bool res = (sqlite3_step(m_stmt_player_load) == SQLITE_ROW);
  368. sqlite3_reset(m_stmt_player_load);
  369. return res;
  370. }
  371. void PlayerDatabaseSQLite3::savePlayer(RemotePlayer *player)
  372. {
  373. PlayerSAO* sao = player->getPlayerSAO();
  374. sanity_check(sao);
  375. const v3f &pos = sao->getBasePosition();
  376. // Begin save in brace is mandatory
  377. if (!playerDataExists(player->getName())) {
  378. beginSave();
  379. str_to_sqlite(m_stmt_player_add, 1, player->getName());
  380. double_to_sqlite(m_stmt_player_add, 2, sao->getLookPitch());
  381. double_to_sqlite(m_stmt_player_add, 3, sao->getRotation().Y);
  382. double_to_sqlite(m_stmt_player_add, 4, pos.X);
  383. double_to_sqlite(m_stmt_player_add, 5, pos.Y);
  384. double_to_sqlite(m_stmt_player_add, 6, pos.Z);
  385. int64_to_sqlite(m_stmt_player_add, 7, sao->getHP());
  386. int64_to_sqlite(m_stmt_player_add, 8, sao->getBreath());
  387. sqlite3_vrfy(sqlite3_step(m_stmt_player_add), SQLITE_DONE);
  388. sqlite3_reset(m_stmt_player_add);
  389. } else {
  390. beginSave();
  391. double_to_sqlite(m_stmt_player_update, 1, sao->getLookPitch());
  392. double_to_sqlite(m_stmt_player_update, 2, sao->getRotation().Y);
  393. double_to_sqlite(m_stmt_player_update, 3, pos.X);
  394. double_to_sqlite(m_stmt_player_update, 4, pos.Y);
  395. double_to_sqlite(m_stmt_player_update, 5, pos.Z);
  396. int64_to_sqlite(m_stmt_player_update, 6, sao->getHP());
  397. int64_to_sqlite(m_stmt_player_update, 7, sao->getBreath());
  398. str_to_sqlite(m_stmt_player_update, 8, player->getName());
  399. sqlite3_vrfy(sqlite3_step(m_stmt_player_update), SQLITE_DONE);
  400. sqlite3_reset(m_stmt_player_update);
  401. }
  402. // Write player inventories
  403. str_to_sqlite(m_stmt_player_remove_inventory, 1, player->getName());
  404. sqlite3_vrfy(sqlite3_step(m_stmt_player_remove_inventory), SQLITE_DONE);
  405. sqlite3_reset(m_stmt_player_remove_inventory);
  406. str_to_sqlite(m_stmt_player_remove_inventory_items, 1, player->getName());
  407. sqlite3_vrfy(sqlite3_step(m_stmt_player_remove_inventory_items), SQLITE_DONE);
  408. sqlite3_reset(m_stmt_player_remove_inventory_items);
  409. std::vector<const InventoryList*> inventory_lists = sao->getInventory()->getLists();
  410. std::ostringstream oss;
  411. for (u16 i = 0; i < inventory_lists.size(); i++) {
  412. const InventoryList* list = inventory_lists[i];
  413. str_to_sqlite(m_stmt_player_add_inventory, 1, player->getName());
  414. int_to_sqlite(m_stmt_player_add_inventory, 2, i);
  415. int_to_sqlite(m_stmt_player_add_inventory, 3, list->getWidth());
  416. str_to_sqlite(m_stmt_player_add_inventory, 4, list->getName());
  417. int_to_sqlite(m_stmt_player_add_inventory, 5, list->getSize());
  418. sqlite3_vrfy(sqlite3_step(m_stmt_player_add_inventory), SQLITE_DONE);
  419. sqlite3_reset(m_stmt_player_add_inventory);
  420. for (u32 j = 0; j < list->getSize(); j++) {
  421. oss.str("");
  422. oss.clear();
  423. list->getItem(j).serialize(oss);
  424. std::string itemStr = oss.str();
  425. str_to_sqlite(m_stmt_player_add_inventory_items, 1, player->getName());
  426. int_to_sqlite(m_stmt_player_add_inventory_items, 2, i);
  427. int_to_sqlite(m_stmt_player_add_inventory_items, 3, j);
  428. str_to_sqlite(m_stmt_player_add_inventory_items, 4, itemStr);
  429. sqlite3_vrfy(sqlite3_step(m_stmt_player_add_inventory_items), SQLITE_DONE);
  430. sqlite3_reset(m_stmt_player_add_inventory_items);
  431. }
  432. }
  433. str_to_sqlite(m_stmt_player_metadata_remove, 1, player->getName());
  434. sqlite3_vrfy(sqlite3_step(m_stmt_player_metadata_remove), SQLITE_DONE);
  435. sqlite3_reset(m_stmt_player_metadata_remove);
  436. const StringMap &attrs = sao->getMeta().getStrings();
  437. for (const auto &attr : attrs) {
  438. str_to_sqlite(m_stmt_player_metadata_add, 1, player->getName());
  439. str_to_sqlite(m_stmt_player_metadata_add, 2, attr.first);
  440. str_to_sqlite(m_stmt_player_metadata_add, 3, attr.second);
  441. sqlite3_vrfy(sqlite3_step(m_stmt_player_metadata_add), SQLITE_DONE);
  442. sqlite3_reset(m_stmt_player_metadata_add);
  443. }
  444. endSave();
  445. player->onSuccessfulSave();
  446. }
  447. bool PlayerDatabaseSQLite3::loadPlayer(RemotePlayer *player, PlayerSAO *sao)
  448. {
  449. verifyDatabase();
  450. str_to_sqlite(m_stmt_player_load, 1, player->getName());
  451. if (sqlite3_step(m_stmt_player_load) != SQLITE_ROW) {
  452. sqlite3_reset(m_stmt_player_load);
  453. return false;
  454. }
  455. sao->setLookPitch(sqlite_to_float(m_stmt_player_load, 0));
  456. sao->setPlayerYaw(sqlite_to_float(m_stmt_player_load, 1));
  457. sao->setBasePosition(sqlite_to_v3f(m_stmt_player_load, 2));
  458. sao->setHPRaw((u16) MYMIN(sqlite_to_int(m_stmt_player_load, 5), U16_MAX));
  459. sao->setBreath((u16) MYMIN(sqlite_to_int(m_stmt_player_load, 6), U16_MAX), false);
  460. sqlite3_reset(m_stmt_player_load);
  461. // Load inventory
  462. str_to_sqlite(m_stmt_player_load_inventory, 1, player->getName());
  463. while (sqlite3_step(m_stmt_player_load_inventory) == SQLITE_ROW) {
  464. InventoryList *invList = player->inventory.addList(
  465. sqlite_to_string(m_stmt_player_load_inventory, 2),
  466. sqlite_to_uint(m_stmt_player_load_inventory, 3));
  467. invList->setWidth(sqlite_to_uint(m_stmt_player_load_inventory, 1));
  468. u32 invId = sqlite_to_uint(m_stmt_player_load_inventory, 0);
  469. str_to_sqlite(m_stmt_player_load_inventory_items, 1, player->getName());
  470. int_to_sqlite(m_stmt_player_load_inventory_items, 2, invId);
  471. while (sqlite3_step(m_stmt_player_load_inventory_items) == SQLITE_ROW) {
  472. const std::string itemStr = sqlite_to_string(m_stmt_player_load_inventory_items, 1);
  473. if (itemStr.length() > 0) {
  474. ItemStack stack;
  475. stack.deSerialize(itemStr);
  476. invList->changeItem(sqlite_to_uint(m_stmt_player_load_inventory_items, 0), stack);
  477. }
  478. }
  479. sqlite3_reset(m_stmt_player_load_inventory_items);
  480. }
  481. sqlite3_reset(m_stmt_player_load_inventory);
  482. str_to_sqlite(m_stmt_player_metadata_load, 1, sao->getPlayer()->getName());
  483. while (sqlite3_step(m_stmt_player_metadata_load) == SQLITE_ROW) {
  484. std::string attr = sqlite_to_string(m_stmt_player_metadata_load, 0);
  485. std::string value = sqlite_to_string(m_stmt_player_metadata_load, 1);
  486. sao->getMeta().setString(attr, value);
  487. }
  488. sao->getMeta().setModified(false);
  489. sqlite3_reset(m_stmt_player_metadata_load);
  490. return true;
  491. }
  492. bool PlayerDatabaseSQLite3::removePlayer(const std::string &name)
  493. {
  494. if (!playerDataExists(name))
  495. return false;
  496. str_to_sqlite(m_stmt_player_remove, 1, name);
  497. sqlite3_vrfy(sqlite3_step(m_stmt_player_remove), SQLITE_DONE);
  498. sqlite3_reset(m_stmt_player_remove);
  499. return true;
  500. }
  501. void PlayerDatabaseSQLite3::listPlayers(std::vector<std::string> &res)
  502. {
  503. verifyDatabase();
  504. while (sqlite3_step(m_stmt_player_list) == SQLITE_ROW)
  505. res.push_back(sqlite_to_string(m_stmt_player_list, 0));
  506. sqlite3_reset(m_stmt_player_list);
  507. }
  508. /*
  509. * Auth database
  510. */
  511. AuthDatabaseSQLite3::AuthDatabaseSQLite3(const std::string &savedir) :
  512. Database_SQLite3(savedir, "auth"), AuthDatabase()
  513. {
  514. }
  515. AuthDatabaseSQLite3::~AuthDatabaseSQLite3()
  516. {
  517. FINALIZE_STATEMENT(m_stmt_read)
  518. FINALIZE_STATEMENT(m_stmt_write)
  519. FINALIZE_STATEMENT(m_stmt_create)
  520. FINALIZE_STATEMENT(m_stmt_delete)
  521. FINALIZE_STATEMENT(m_stmt_list_names)
  522. FINALIZE_STATEMENT(m_stmt_read_privs)
  523. FINALIZE_STATEMENT(m_stmt_write_privs)
  524. FINALIZE_STATEMENT(m_stmt_delete_privs)
  525. FINALIZE_STATEMENT(m_stmt_last_insert_rowid)
  526. }
  527. void AuthDatabaseSQLite3::createDatabase()
  528. {
  529. assert(m_database); // Pre-condition
  530. SQLOK(sqlite3_exec(m_database,
  531. "CREATE TABLE IF NOT EXISTS `auth` ("
  532. "`id` INTEGER PRIMARY KEY AUTOINCREMENT,"
  533. "`name` VARCHAR(32) UNIQUE,"
  534. "`password` VARCHAR(512),"
  535. "`last_login` INTEGER"
  536. ");",
  537. NULL, NULL, NULL),
  538. "Failed to create auth table");
  539. SQLOK(sqlite3_exec(m_database,
  540. "CREATE TABLE IF NOT EXISTS `user_privileges` ("
  541. "`id` INTEGER,"
  542. "`privilege` VARCHAR(32),"
  543. "PRIMARY KEY (id, privilege)"
  544. "CONSTRAINT fk_id FOREIGN KEY (id) REFERENCES auth (id) ON DELETE CASCADE"
  545. ");",
  546. NULL, NULL, NULL),
  547. "Failed to create auth privileges table");
  548. }
  549. void AuthDatabaseSQLite3::initStatements()
  550. {
  551. PREPARE_STATEMENT(read, "SELECT id, name, password, last_login FROM auth WHERE name = ?");
  552. PREPARE_STATEMENT(write, "UPDATE auth set name = ?, password = ?, last_login = ? WHERE id = ?");
  553. PREPARE_STATEMENT(create, "INSERT INTO auth (name, password, last_login) VALUES (?, ?, ?)");
  554. PREPARE_STATEMENT(delete, "DELETE FROM auth WHERE name = ?");
  555. PREPARE_STATEMENT(list_names, "SELECT name FROM auth ORDER BY name DESC");
  556. PREPARE_STATEMENT(read_privs, "SELECT privilege FROM user_privileges WHERE id = ?");
  557. PREPARE_STATEMENT(write_privs, "INSERT OR IGNORE INTO user_privileges (id, privilege) VALUES (?, ?)");
  558. PREPARE_STATEMENT(delete_privs, "DELETE FROM user_privileges WHERE id = ?");
  559. PREPARE_STATEMENT(last_insert_rowid, "SELECT last_insert_rowid()");
  560. }
  561. bool AuthDatabaseSQLite3::getAuth(const std::string &name, AuthEntry &res)
  562. {
  563. verifyDatabase();
  564. str_to_sqlite(m_stmt_read, 1, name);
  565. if (sqlite3_step(m_stmt_read) != SQLITE_ROW) {
  566. sqlite3_reset(m_stmt_read);
  567. return false;
  568. }
  569. res.id = sqlite_to_uint(m_stmt_read, 0);
  570. res.name = sqlite_to_string(m_stmt_read, 1);
  571. res.password = sqlite_to_string(m_stmt_read, 2);
  572. res.last_login = sqlite_to_int64(m_stmt_read, 3);
  573. sqlite3_reset(m_stmt_read);
  574. int64_to_sqlite(m_stmt_read_privs, 1, res.id);
  575. while (sqlite3_step(m_stmt_read_privs) == SQLITE_ROW) {
  576. res.privileges.emplace_back(sqlite_to_string(m_stmt_read_privs, 0));
  577. }
  578. sqlite3_reset(m_stmt_read_privs);
  579. return true;
  580. }
  581. bool AuthDatabaseSQLite3::saveAuth(const AuthEntry &authEntry)
  582. {
  583. beginSave();
  584. str_to_sqlite(m_stmt_write, 1, authEntry.name);
  585. str_to_sqlite(m_stmt_write, 2, authEntry.password);
  586. int64_to_sqlite(m_stmt_write, 3, authEntry.last_login);
  587. int64_to_sqlite(m_stmt_write, 4, authEntry.id);
  588. sqlite3_vrfy(sqlite3_step(m_stmt_write), SQLITE_DONE);
  589. sqlite3_reset(m_stmt_write);
  590. writePrivileges(authEntry);
  591. endSave();
  592. return true;
  593. }
  594. bool AuthDatabaseSQLite3::createAuth(AuthEntry &authEntry)
  595. {
  596. beginSave();
  597. // id autoincrements
  598. str_to_sqlite(m_stmt_create, 1, authEntry.name);
  599. str_to_sqlite(m_stmt_create, 2, authEntry.password);
  600. int64_to_sqlite(m_stmt_create, 3, authEntry.last_login);
  601. sqlite3_vrfy(sqlite3_step(m_stmt_create), SQLITE_DONE);
  602. sqlite3_reset(m_stmt_create);
  603. // obtain id and write back to original authEntry
  604. sqlite3_step(m_stmt_last_insert_rowid);
  605. authEntry.id = sqlite_to_uint(m_stmt_last_insert_rowid, 0);
  606. sqlite3_reset(m_stmt_last_insert_rowid);
  607. writePrivileges(authEntry);
  608. endSave();
  609. return true;
  610. }
  611. bool AuthDatabaseSQLite3::deleteAuth(const std::string &name)
  612. {
  613. verifyDatabase();
  614. str_to_sqlite(m_stmt_delete, 1, name);
  615. sqlite3_vrfy(sqlite3_step(m_stmt_delete), SQLITE_DONE);
  616. int changes = sqlite3_changes(m_database);
  617. sqlite3_reset(m_stmt_delete);
  618. // privileges deleted by foreign key on delete cascade
  619. return changes > 0;
  620. }
  621. void AuthDatabaseSQLite3::listNames(std::vector<std::string> &res)
  622. {
  623. verifyDatabase();
  624. while (sqlite3_step(m_stmt_list_names) == SQLITE_ROW) {
  625. res.push_back(sqlite_to_string(m_stmt_list_names, 0));
  626. }
  627. sqlite3_reset(m_stmt_list_names);
  628. }
  629. void AuthDatabaseSQLite3::reload()
  630. {
  631. // noop for SQLite
  632. }
  633. void AuthDatabaseSQLite3::writePrivileges(const AuthEntry &authEntry)
  634. {
  635. int64_to_sqlite(m_stmt_delete_privs, 1, authEntry.id);
  636. sqlite3_vrfy(sqlite3_step(m_stmt_delete_privs), SQLITE_DONE);
  637. sqlite3_reset(m_stmt_delete_privs);
  638. for (const std::string &privilege : authEntry.privileges) {
  639. int64_to_sqlite(m_stmt_write_privs, 1, authEntry.id);
  640. str_to_sqlite(m_stmt_write_privs, 2, privilege);
  641. sqlite3_vrfy(sqlite3_step(m_stmt_write_privs), SQLITE_DONE);
  642. sqlite3_reset(m_stmt_write_privs);
  643. }
  644. }