aboutsummaryrefslogtreecommitdiffstats
path: root/src/dblayer.c
diff options
context:
space:
mode:
Diffstat (limited to 'src/dblayer.c')
-rw-r--r--src/dblayer.c227
1 files changed, 187 insertions, 40 deletions
diff --git a/src/dblayer.c b/src/dblayer.c
index 98a4d64..b55a760 100644
--- a/src/dblayer.c
+++ b/src/dblayer.c
@@ -12,18 +12,15 @@
* Give access to the database.
*
* -------------------------------------------------------------------------- */
-#include <stdio.h>
-#include <stdlib.h>
-#include <string.h>
-
#include "dblayer.h"
+#include "dbhelper.h"
#include "ckutil.h"
const char * const DB_NAME = "/ckdb";
/* figure out the database name */
void make_db_name(char *ret, const char *confPath) {
- char db_path[200];
+ char db_path[STR_L];
strcpy(db_path, confPath);
strcat(db_path, DB_NAME);
@@ -32,14 +29,15 @@ void make_db_name(char *ret, const char *confPath) {
/* Check if the db file exists*/
int db_exists(const UserOpt *opt) {
- char db_path[200];
+ char db_path[STR_L];
make_db_name(db_path, opt->confDir);
return util_is_file_rw(db_path);
}
/* check if db has the correct tables */
int check_initialized_DB(sqlite3 *db) {
- char *sql = "SELECT * FROM SQLITE_MASTER WHERE type='table';";
+ char sql[STR_M];
+ dbh_form_query_select_all_tables(sql);
sqlite3_stmt *stmt;
int rc = sqlite3_prepare_v2(db, sql, strlen(sql), &stmt, NULL);
@@ -47,13 +45,13 @@ int check_initialized_DB(sqlite3 *db) {
int program_table_ok, config_table_ok, rel_table_ok = 0;
while (sqlite3_step(stmt) != SQLITE_DONE) {
const unsigned char *tmpbuf = sqlite3_column_text(stmt, 1);
- if (strcmp((char *)tmpbuf, "PROGRAM") == 0) {
+ if (strcmp((char *)tmpbuf, TBL_PROGRAM) == 0) {
program_table_ok = 1;
}
- if (strcmp((char *)tmpbuf, "CONFIG") == 0) {
+ if (strcmp((char *)tmpbuf, TBL_CONFIG) == 0) {
config_table_ok = 1;
}
- if (strcmp((char *)tmpbuf, "REL") == 0) {
+ if (strcmp((char *)tmpbuf, TBL_REL) == 0) {
rel_table_ok = 1;
}
}
@@ -82,7 +80,7 @@ void close_DB(DB *db) {
DB init_make_DB(const UserOpt *opt) {
sqlite3 *db;
- char db_path[200];
+ char db_path[STR_L];
int rc;
make_db_name(db_path, opt->confDir);
@@ -97,7 +95,7 @@ DB init_make_DB(const UserOpt *opt) {
DB open_DB(const UserOpt *opt) {
sqlite3 *db;
int rc;
- char db_path[200];
+ char db_path[STR_L];
make_db_name(db_path, opt->confDir);
rc = sqlite3_open(db_path, &db);
@@ -114,18 +112,8 @@ DB open_DB(const UserOpt *opt) {
}
void init_make_tables(DB *db) {
- char * sql =
- "CREATE TABLE REL("
- "pId INT NOT NULL,"
- "cId INT NOT NULL);"
- "CREATE TABLE PROGRAM("
- "id INT NOT NULL PRIMARY KEY,"
- "name TEXT NOT NULL);"
- "CREATE TABLE CONFIG("
- "id INT NOT NULL PRIMARY KEY,"
- "path TEXT NOT NULL,"
- "secret INT NOT NULL,"
- "prime INT NOT NULL);";
+ char sql[STR_L];
+ dbh_form_query_make_tables(sql);
char *err_msg = NULL;
int rc = sqlite3_exec(db->ptr, sql, 0, 0, &err_msg);
@@ -141,10 +129,9 @@ int get_next_valid_id_from_table(DB *db, const char* tableName) {
sqlite3_stmt *stmt;
int rc;
- char sql[100] = "SELECT id FROM ";
- strcat(sql, tableName);
- strcat(sql, " ORDER BY id;");
-
+ char sql[STR_M];
+ dbh_form_query_select_id_from(sql, tableName);
+
rc = sqlite3_prepare_v2(db->ptr, sql, -1, &stmt, 0);
if (rc != SQLITE_OK) {
return -1;
@@ -163,33 +150,193 @@ int get_next_valid_id_from_table(DB *db, const char* tableName) {
return id;
}
-void insert_to_program_table(DB *db, const char *name) {
+int insert_to_program_table(DB *db, const char *name) {
sqlite3_stmt *stmt;
int rc;
- char * sql =
- "INSERT INTO PROGRAM VALUES(?, ?);";
+ char sql[STR_L];
+ dbh_form_query_insert_program(sql);
rc = sqlite3_prepare_v2(db->ptr, sql, -1, &stmt, 0);
if (rc != SQLITE_OK) {
- printf("Error\n");
- return;
+ printf("Error while preparing insert to program sql.\n");
+ db->error = SQL_ERR_SQLITE;
+ return -1;
}
- int id = get_next_valid_id_from_table(db, "PROGRAM");
+ int id = get_next_valid_id_from_table(db, TBL_PROGRAM);
if (id == -1) {
db->error = SQL_ERR_SQLITE;
- return;
+ return -1;
}
sqlite3_bind_int(stmt, 1, id);
sqlite3_bind_text(stmt, 2, name, strlen(name), 0);
if (sqlite3_step(stmt) != SQLITE_DONE) {
- printf("Error\n");
- return;
+ printf("Error while excecuting insert to program sql.\n");
+ db->error = SQL_ERR_SQLITE;
+ return -1;
}
sqlite3_finalize(stmt);
+ return id;
}
-int add_insert_program_to_db(DB *db, const char *name) {
- insert_to_program_table(db,name);
- return 0;
+int insert_to_config_table(DB *db, const char *path, const int secret, const int prime) {
+ sqlite3_stmt *stmt;
+ int rc;
+
+ char sql[STR_L];
+ dbh_form_query_insert_config(sql);
+ rc = sqlite3_prepare_v2(db->ptr, sql, -1, &stmt, 0);
+ if (rc != SQLITE_OK) {
+ printf("Error while preparing insert to config sql.\n");
+ db->error = SQL_ERR_SQLITE;
+ return -1;
+ }
+ int id = get_next_valid_id_from_table(db, TBL_CONFIG);
+ if (id == -1) {
+ db->error = SQL_ERR_SQLITE;
+ return -1;
+ }
+ sqlite3_bind_int(stmt, 1, id);
+ sqlite3_bind_text(stmt, 2, path, strlen(path), 0);
+ sqlite3_bind_int(stmt, 3, secret);
+ sqlite3_bind_int(stmt, 4, prime);
+ if (sqlite3_step(stmt) != SQLITE_DONE) {
+ printf("Error while excecuting insert to config sql.\n");
+ db->error = SQL_ERR_SQLITE;
+ return-1;
+ }
+ sqlite3_finalize(stmt);
+ return id;
+}
+
+int insert_to_rel_table(DB *db, const int pid, const int cid) {
+ sqlite3_stmt *stmt;
+ int rc;
+
+ char sql[STR_M];
+ dhb_form_query_insert_relationship(sql);
+ rc = sqlite3_prepare_v2(db->ptr, sql, -1, &stmt, 0);
+ if (rc != SQLITE_OK) {
+ db->error = SQL_ERR_SQLITE;
+ printf("Error while preparing insert to rel sql.\n");
+ return -1;
+ }
+ sqlite3_bind_int(stmt, 1, pid);
+ sqlite3_bind_int(stmt, 2, cid);
+ if (sqlite3_step(stmt) != SQLITE_DONE) {
+ db->error = SQL_ERR_SQLITE;
+ printf("Error while excecuting insert to rel sql.\n");
+ return-1;
+ }
+ sqlite3_finalize(stmt);
+ return 1;
+}
+
+int program_exists(DB *db, const char* name) {
+ sqlite3_stmt *stmt;
+ int rc;
+
+ char sql[STR_M];
+ dhb_form_query_find_program(sql);
+
+ rc = sqlite3_prepare_v2(db->ptr, sql, -1, &stmt, 0);
+ if (rc != SQLITE_OK) {
+ printf("Error while preparing program_exists sql.\n");
+ return -2;
+ }
+ sqlite3_bind_text(stmt, 1, name, strlen(name), 0);
+ int id = -1;
+ while (sqlite3_step(stmt) == SQLITE_ROW) {
+ id = sqlite3_column_int(stmt, 0);
+ break;
+ }
+ sqlite3_finalize(stmt);
+ return id;
+}
+
+int config_exists(DB *db, const int pid, const char* path) {
+ sqlite3_stmt *stmt;
+ int rc;
+
+ char sql[STR_M];
+ dhb_form_query_find_config(sql);
+
+ rc = sqlite3_prepare_v2(db->ptr, sql, -1, &stmt, 0);
+ if (rc != SQLITE_OK) {
+ printf("Error while preparing config_exists sql.\n");
+ return -2;
+ }
+ sqlite3_bind_text(stmt, 1, path, strlen(path), 0);
+ int id = -1;
+ while (sqlite3_step(stmt) == SQLITE_ROW) {
+ id = sqlite3_column_int(stmt, 0);
+ break;
+ }
+ sqlite3_finalize(stmt);
+ return id;
+}
+
+int add_insert_relationship(DB *db, const int pid, const int cid) {
+ // do checks
+ return insert_to_rel_table(db, pid, cid);
+}
+
+int add_get_or_insert_config_to_db(DB *db, const int pid, const char *path, const int secret, const int prime) {
+ int cid = config_exists(db, pid, path);
+ if (cid == -2) {
+ db->error = SQL_ERR_SQLITE;
+ return -1;
+ }
+ /* If config doesnt exist insert it and return it's cid */
+ if (cid == -1) {
+ return insert_to_config_table(db, path, secret, prime);
+ }
+
+ /* If it exist it means the user has inserted the same path twice */
+ db->error = SQL_CONFIG_PATH_EXISTS;
+ return -1;
+}
+
+int add_get_or_insert_program_to_db(DB *db, const char *name) {
+ int pid = program_exists(db, name);
+ if (pid == -2) {
+ db->error = SQL_ERR_SQLITE;
+ return -1;
+ }
+ if (pid == -1) {
+ return insert_to_program_table(db,name);
+ }
+ return pid;
+}
+
+int add_transaction_begin(DB *db, const char *progName,
+ const char* confPath, const int secret,
+ const int prime) {
+ __BEGIN_TRANSACTION__
+ int pid = add_get_or_insert_program_to_db(db, progName);
+ if (db->error == SQL_ERR_SQLITE) {
+ printf("Could not insert program to db.\n");
+ close_DB(db);
+ return 0;
+ }
+ int cid = add_get_or_insert_config_to_db(db, pid, confPath, secret, prime);
+ if (db->error == SQL_ERR_SQLITE) {
+ printf("Could not insert config to db.\n");
+ close_DB(db);
+ return 0;
+ }
+ else if (db->error == SQL_CONFIG_PATH_EXISTS) {
+ printf("This config already exists in the database.\n");
+ close_DB(db);
+ return 0;
+ }
+ add_insert_relationship(db, pid, cid);
+ if (db->error == SQL_ERR_SQLITE) {
+ printf("Could not insert config to db.\n");
+ close_DB(db);
+ return 0;
+ }
+ __END_TRANSACTION__
+
+ return 1;
}
generated by cgit v1.2.3 (git 2.25.1) at 2024-11-22 07:34:39 +0000