#define QUERY_CREATE_SHORTCUT_TABLE \
"PRAGMA journal_mode = PERSIST;" \
- "PRAGMA integrity_check;" \
"CREATE TABLE IF NOT EXISTS shortcut_service (\n" \
" id INTEGER PRIMARY KEY AUTOINCREMENT,\n" \
" pkgid TEXT,\n" \
" name TEXT,\n" \
" icon TEXT);"
+static bool is_db_corrupted = false;
+
+static int __check_integrity_cb(void *pid, int argc, char **argv, char **notUsed)
+{
+ if (!strcmp(argv[0], "ok")) {
+ SHORTCUT_ERR("db integrity result : %s" , argv[0]);
+ is_db_corrupted = true;
+ return -1;
+ }
+
+ SHORTCUT_INFO("db integrity result : %s" , argv[0]);
+ return 0;
+}
+
+static int __recover_corrupted_db(sqlite3 *db)
+{
+ int ret = SHORTCUT_ERROR_NONE;
+ int sql_ret;
+ char *errmsg = NULL;
+
+ SHORTCUT_INFO("DB is corrupted, start to recover corrupted db");
+ if (db)
+ sqlite3_close(db);
+ unlink(DB_PATH);
+
+ sql_ret = sqlite3_open_v2(DB_PATH, &db,
+ SQLITE_OPEN_CREATE |SQLITE_OPEN_READWRITE,
+ NULL);
+ if (sql_ret != SQLITE_OK) {
+ SHORTCUT_ERR("Failed to open db[%d]", sql_ret);
+ unlink(DB_PATH);
+ ret = SHORTCUT_ERROR_IO_ERROR;
+ goto out;
+ }
+
+ sql_ret = sqlite3_exec(db, QUERY_CREATE_SHORTCUT_TABLE, NULL, NULL, &errmsg);
+ if (sql_ret != SQLITE_OK) {
+ SHORTCUT_ERR("Failed to exec query[%d][%s]", sql_ret, errmsg);
+ ret = SHORTCUT_ERROR_IO_ERROR;
+ }
+
+out:
+ if (errmsg)
+ sqlite3_free(errmsg);
+
+ return ret;
+}
+
EAPI int shortcut_db_init()
{
int ret = SHORTCUT_ERROR_NONE;
SQLITE_OPEN_CREATE | SQLITE_OPEN_READWRITE, NULL);
if (sql_ret != SQLITE_OK) {
SHORTCUT_ERR("Failed to open db[%d]", sql_ret);
-
- if (sqlite3_errcode(db) == SQLITE_CORRUPT) {
- if (db)
- sqlite3_close(db);
- unlink(DB_PATH);
-
- sql_ret = sqlite3_open_v2(DB_PATH, &db,
- SQLITE_OPEN_CREATE |SQLITE_OPEN_READWRITE,
- NULL);
- if (sql_ret != SQLITE_OK) {
- SHORTCUT_ERR("Failed to open db[%d]", sql_ret);
- unlink(DB_PATH);
- ret = SHORTCUT_ERROR_IO_ERROR;
- goto out;
- }
- } else {
- ret = SHORTCUT_ERROR_IO_ERROR;
- goto out;
- }
+ ret = SHORTCUT_ERROR_IO_ERROR;
+ goto out;
}
- sql_ret = sqlite3_exec(db, QUERY_CREATE_SHORTCUT_TABLE, NULL, NULL,
- &errmsg);
+ sql_ret = sqlite3_exec(db, QUERY_CREATE_SHORTCUT_TABLE,
+ NULL, NULL, &errmsg);
if (sql_ret != SQLITE_OK) {
SHORTCUT_ERR("Failed to exec query [%d][%s]", sql_ret, errmsg);
+ ret = SHORTCUT_ERROR_IO_ERROR;
+ goto out;
+ }
- if (sql_ret == SQLITE_CORRUPT || sql_ret == SQLITE_NOTADB) {
- sqlite3_close(db);
- unlink(DB_PATH);
-
- sql_ret = sqlite3_open_v2(DB_PATH, &db,
- SQLITE_OPEN_CREATE | SQLITE_OPEN_READWRITE,
- NULL);
- if (sql_ret != SQLITE_OK) {
- SHORTCUT_ERR("Failed to open db[%d]", sql_ret);
- unlink(DB_PATH);
- ret = SHORTCUT_ERROR_IO_ERROR;
- goto out;
- }
-
- sqlite3_free(errmsg);
- sql_ret = sqlite3_exec(db, QUERY_CREATE_SHORTCUT_TABLE,
- NULL, NULL, &errmsg);
- if (sql_ret != SQLITE_OK) {
- SHORTCUT_ERR("Failed to exec sqlite, again[%d][%s]",
- sql_ret, errmsg);
- unlink(DB_PATH);
- ret = SHORTCUT_ERROR_IO_ERROR;
- goto out;
- }
- } else {
- ret = SHORTCUT_ERROR_IO_ERROR;
- goto out;
- }
+ sql_ret = sqlite3_exec(db, "PRAGMA integrity_check",
+ __check_integrity_cb, NULL, &errmsg);
+ if (sql_ret != SQLITE_OK || is_db_corrupted) {
+ SHORTCUT_ERR("Failed to exec query[%d][%s]", sql_ret, errmsg);
+ ret = SHORTCUT_ERROR_IO_ERROR;
}
out:
+ if (sql_ret == SQLITE_CORRUPT || sql_ret == SQLITE_NOTADB || is_db_corrupted)
+ ret = __recover_corrupted_db(db);
if (errmsg)
sqlite3_free(errmsg);
if (db)