32 #include <sys/types.h>
39 #include "db1-ast/include/db.h"
41 #define MAX_DB_FIELD 256
43 ({ typeof (a) _a = (a); \
44 typeof (b) _b = (b); \
47 static sqlite3 *astdb;
49 #define DEFINE_SQL_STATEMENT(stmt,sql) static sqlite3_stmt *stmt; \
50 const char stmt##_sql[] = sql;
52 DEFINE_SQL_STATEMENT(put_stmt,
"INSERT OR REPLACE INTO astdb (key, value) VALUES (?, ?)")
53 DEFINE_SQL_STATEMENT(create_astdb_stmt, "CREATE TABLE IF NOT EXISTS astdb(key VARCHAR(256), value VARCHAR(256), PRIMARY KEY(key))")
55 static
int db_execute_transaction_sql(const
char *sql)
60 sqlite3_exec(astdb, sql, NULL, NULL, &errmsg);
62 fprintf(stderr,
"Error executing SQL: %s\n", errmsg);
70 static int ast_db_begin_transaction(
void)
72 return db_execute_transaction_sql(
"BEGIN TRANSACTION");
75 static int ast_db_commit_transaction(
void)
77 return db_execute_transaction_sql(
"COMMIT");
80 static int ast_db_rollback_transaction(
void)
82 return db_execute_transaction_sql(
"ROLLBACK");
85 static int db_put_raw(
const char *key,
size_t keylen,
const char *value,
size_t valuelen)
89 if (sqlite3_bind_text(put_stmt, 1, key, keylen, SQLITE_STATIC) != SQLITE_OK) {
90 fprintf(stderr,
"Couldn't bind key to stmt: %s\n", sqlite3_errmsg(astdb));
92 }
else if (sqlite3_bind_text(put_stmt, 2, value, valuelen, SQLITE_STATIC) != SQLITE_OK) {
93 fprintf(stderr,
"Couldn't bind value to stmt: %s\n", sqlite3_errmsg(astdb));
95 }
else if (sqlite3_step(put_stmt) != SQLITE_DONE) {
96 fprintf(stderr,
"Couldn't execute statement: %s\n", sqlite3_errmsg(astdb));
99 sqlite3_reset(put_stmt);
104 static int convert_bdb_to_sqlite3(
const char *bdb_dbname)
107 DBT key = { 0, }, value = { 0, }, last_key = { 0, };
109 char last_key_s[MAX_DB_FIELD];
111 if (!(bdb = dbopen(bdb_dbname, O_RDONLY, AST_FILE_MODE, DB_BTREE, NULL))) {
112 fprintf(stderr,
"Unable to open Asterisk database '%s'\n", bdb_dbname);
116 if (bdb->seq(bdb, &last_key, &value, R_LAST)) {
121 memcpy(last_key_s, last_key.data, MIN(last_key.size - 1,
sizeof(last_key_s)));
122 last_key_s[last_key.size - 1] =
'\0';
123 for (res = bdb->seq(bdb, &key, &value, R_FIRST);
124 !res; res = bdb->seq(bdb, &key, &value, R_NEXT)) {
125 last = !strcmp(key.data, last_key_s);
126 db_put_raw((
const char *) key.data, key.size - 1, (
const char *) value.data, value.size - 1);
137 static int init_stmt(sqlite3_stmt **stmt,
const char *sql,
size_t len)
139 if (sqlite3_prepare(astdb, sql, len, stmt, NULL) != SQLITE_OK) {
140 fprintf(stderr,
"Couldn't prepare statement '%s': %s\n", sql, sqlite3_errmsg(astdb));
147 static int db_create_astdb(
void)
149 if (init_stmt(&create_astdb_stmt, create_astdb_stmt_sql,
sizeof(create_astdb_stmt_sql))) {
153 ast_db_begin_transaction();
154 if (sqlite3_step(create_astdb_stmt) != SQLITE_DONE) {
155 fprintf(stderr,
"Couldn't create astdb table: %s\n", sqlite3_errmsg(astdb));
156 ast_db_rollback_transaction();
157 sqlite3_reset(create_astdb_stmt);
161 ast_db_commit_transaction();
162 sqlite3_reset(create_astdb_stmt);
167 static int init_statements(
void)
171 return init_stmt(&put_stmt, put_stmt_sql,
sizeof(put_stmt_sql));
174 static int db_open(
const char *dbname)
176 if (sqlite3_open(dbname, &astdb) != SQLITE_OK) {
177 fprintf(stderr,
"Unable to open Asterisk database '%s': %s\n", dbname, sqlite3_errmsg(astdb));
178 sqlite3_close(astdb);
185 static int sql_db_init(
const char *dbname)
187 if (
db_open(dbname) || db_create_astdb() || init_statements()) {
194 int main(
int argc,
char *argv[])
197 struct stat dont_care;
200 fprintf(stderr,
"%s takes the path of astdb as its only argument\n", basename(argv[0]));
204 if (stat(argv[1], &dont_care)) {
205 fprintf(stderr,
"Unable to open %s: %s\n", argv[1], strerror(errno));
209 if (!(dbname = alloca(strlen(argv[1]) +
sizeof(
".sqlite3")))) {
213 strcpy(dbname, argv[1]);
214 strcat(dbname,
".sqlite3");
216 if (!stat(dbname, &dont_care)) {
217 fprintf(stderr,
"%s already exists!\n", dbname);
221 if (sql_db_init(dbname)) {
225 if (convert_bdb_to_sqlite3(argv[1])) {
226 fprintf(stderr,
"Database conversion failed!\n");
228 sqlite3_close(astdb);
231 sqlite3_close(astdb);
static int db_open(struct realtime_sqlite3_db *db)
Open a database and appropriately set debugging on the db handle.
Asterisk main include file. File version handling, generic pbx functions.
int main(int argc, char *argv[])