1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
|
/*
* Copyright (c) 2022 Egor Tensin <Egor.Tensin@gmail.com>
* This file is part of the "cimple" project.
* For details, see https://github.com/egor-tensin/cimple.
* Distributed under the MIT License.
*/
#include "storage_sqlite.h"
#include "log.h"
#include "sql/sqlite_sql.h"
#include "sqlite.h"
#include "storage.h"
#include <sqlite3.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
struct storage_settings_sqlite {
char *path;
};
int storage_settings_create_sqlite(struct storage_settings *settings, const char *path)
{
struct storage_settings_sqlite *sqlite = malloc(sizeof(struct storage_settings_sqlite));
if (!sqlite) {
log_errno("malloc");
return -1;
}
sqlite->path = strdup(path);
if (!sqlite->path) {
log_errno("strdup");
goto free;
}
settings->type = STORAGE_TYPE_SQLITE;
settings->sqlite = sqlite;
return 0;
free:
free(sqlite);
return -1;
}
void storage_settings_destroy_sqlite(const struct storage_settings *settings)
{
free(settings->sqlite->path);
free(settings->sqlite);
}
struct storage_sqlite {
sqlite3 *db;
};
static int storage_upgrade_sqlite_to(struct storage_sqlite *storage, size_t version)
{
static const char *const FMT = "%s PRAGMA user_version = %zu;";
const char *script = sql_sqlite_files[version];
int ret = 0;
ret = snprintf(NULL, 0, FMT, script, version + 1);
size_t nb = (size_t)ret + 1;
ret = 0;
char *full_script = malloc(nb);
if (!full_script) {
log_errno("malloc");
return -1;
}
snprintf(full_script, nb, FMT, script, version + 1);
ret = sqlite_exec_as_transaction(storage->db, full_script);
goto free;
free:
free(full_script);
return ret;
}
static int storage_upgrade_sqlite_from_to(struct storage_sqlite *storage, size_t from, size_t to)
{
int ret = 0;
for (size_t i = from; i < to; ++i) {
log("Upgrading SQLite database from version %zu to version %zu\n", i, i + 1);
ret = storage_upgrade_sqlite_to(storage, i);
if (ret < 0) {
log_err("Failed to upgrade to version %zu\n", i + 1);
return ret;
}
}
return ret;
}
static int storage_upgrade_sqlite(struct storage_sqlite *storage)
{
unsigned int current_version = 0;
int ret = 0;
ret = sqlite_get_user_version(storage->db, ¤t_version);
if (ret < 0)
return ret;
log("SQLite database version: %u\n", current_version);
size_t newest_version = sizeof(sql_sqlite_files) / sizeof(sql_sqlite_files[0]);
log("Newest database version: %zu\n", newest_version);
if (current_version > newest_version) {
log_err("Unknown database version: %u\n", current_version);
return -1;
}
if (current_version == newest_version) {
log("SQLite database already at the newest version\n");
return 0;
}
return storage_upgrade_sqlite_from_to(storage, current_version, newest_version);
}
static int storage_prepare_sqlite(struct storage_sqlite *storage)
{
int ret = 0;
ret = sqlite_set_foreign_keys(storage->db);
if (ret < 0)
return ret;
ret = storage_upgrade_sqlite(storage);
if (ret < 0)
return ret;
return ret;
}
int storage_create_sqlite(struct storage *storage, const struct storage_settings *settings)
{
int ret = 0;
log("Using SQLite database at %s\n", settings->sqlite->path);
struct storage_sqlite *sqlite = malloc(sizeof(struct storage_sqlite));
if (!sqlite) {
log_errno("malloc");
return -1;
}
ret = sqlite_init();
if (ret < 0)
goto free;
ret = sqlite_open_rw(settings->sqlite->path, &sqlite->db);
if (ret < 0)
goto destroy;
ret = storage_prepare_sqlite(sqlite);
if (ret < 0)
goto close;
storage->sqlite = sqlite;
return ret;
close:
sqlite_close(storage->sqlite->db);
destroy:
sqlite_destroy();
free:
free(sqlite);
return ret;
}
void storage_destroy_sqlite(struct storage *storage)
{
sqlite_close(storage->sqlite->db);
sqlite_destroy();
free(storage->sqlite);
}
|
