]> git.sur5r.net Git - bacula/bacula/blobdiff - bacula/src/cats/sql.c
- Convert more atoi to str_to_int64() for DB.
[bacula/bacula] / bacula / src / cats / sql.c
index 26144925acbda794b8aafa645c3067149b5a28bb..e61133530e223ea02298c9f912103e9b1541be7d 100644 (file)
@@ -1,6 +1,6 @@
 /*
  * Bacula Catalog Database interface routines
- * 
+ *
  *     Almost generic set of SQL database interface routines
  *     (with a little more work)
  *
@@ -10,7 +10,7 @@
  */
 
 /*
-   Copyright (C) 2000-2004 Kern Sibbald and John Walker
+   Copyright (C) 2000-2005 Kern Sibbald
 
    This program is free software; you can redistribute it and/or
    modify it under the terms of the GNU General Public License as
@@ -37,7 +37,7 @@
 #include "bacula.h"
 #include "cats.h"
 
-#if    HAVE_MYSQL || HAVE_SQLITE || HAVE_POSTGRESQL
+#if    HAVE_SQLITE3 || HAVE_MYSQL || HAVE_SQLITE || HAVE_POSTGRESQL
 
 uint32_t bacula_db_version = 0;
 
@@ -52,19 +52,19 @@ static int int_handler(void *ctx, int num_fields, char **row)
 {
    uint32_t *val = (uint32_t *)ctx;
 
-   Dmsg1(200, "int_handler starts with row pointing at %x\n", row);
+   Dmsg1(800, "int_handler starts with row pointing at %x\n", row);
 
    if (row[0]) {
-      Dmsg1(200, "int_handler finds '%s'\n", row[0]);
-      *val = atoi(row[0]);
+      Dmsg1(800, "int_handler finds '%s'\n", row[0]);
+      *val = str_to_int64(row[0]);
    } else {
-      Dmsg0(200, "int_handler finds zero\n");
+      Dmsg0(800, "int_handler finds zero\n");
       *val = 0;
    }
-   Dmsg0(200, "int_handler finishes\n");
+   Dmsg0(800, "int_handler finishes\n");
    return 0;
 }
-       
+
 
 
 /* NOTE!!! The following routines expect that the
@@ -74,12 +74,12 @@ static int int_handler(void *ctx, int num_fields, char **row)
 /* Check that the tables correspond to the version we want */
 int check_tables_version(JCR *jcr, B_DB *mdb)
 {
-   char *query = "SELECT VersionId FROM Version";
-  
+   const char *query = "SELECT VersionId FROM Version";
+
    bacula_db_version = 0;
    db_sql_query(mdb, query, int_handler, (void *)&bacula_db_version);
    if (bacula_db_version != BDB_VERSION) {
-      Mmsg(&mdb->errmsg, "Version error for database \"%s\". Wanted %d, got %d\n",
+      Mmsg(mdb->errmsg, "Version error for database \"%s\". Wanted %d, got %d\n",
          mdb->db_name, BDB_VERSION, bacula_db_version);
       Jmsg(jcr, M_FATAL, 0, "%s", mdb->errmsg);
       return 0;
@@ -89,7 +89,7 @@ int check_tables_version(JCR *jcr, B_DB *mdb)
 
 /* Utility routine for queries. The database MUST be locked before calling here. */
 int
-QueryDB(char *file, int line, JCR *jcr, B_DB *mdb, char *cmd)
+QueryDB(const char *file, int line, JCR *jcr, B_DB *mdb, char *cmd)
 {
    int status;
    if ((status=sql_query(mdb, cmd)) != 0) {
@@ -106,13 +106,13 @@ QueryDB(char *file, int line, JCR *jcr, B_DB *mdb, char *cmd)
    return mdb->result != NULL;
 }
 
-/* 
- * Utility routine to do inserts   
- * Returns: 0 on failure      
+/*
+ * Utility routine to do inserts
+ * Returns: 0 on failure
  *         1 on success
  */
 int
-InsertDB(char *file, int line, JCR *jcr, B_DB *mdb, char *cmd)
+InsertDB(const char *file, int line, JCR *jcr, B_DB *mdb, char *cmd)
 {
    if (sql_query(mdb, cmd)) {
       m_msg(file, line, &mdb->errmsg,  _("insert %s failed:\n%s\n"), cmd, sql_strerror(mdb));
@@ -129,7 +129,7 @@ InsertDB(char *file, int line, JCR *jcr, B_DB *mdb, char *cmd)
    }
    if (mdb->num_rows != 1) {
       char ed1[30];
-      m_msg(file, line, &mdb->errmsg, _("Insertion problem: affected_rows=%s\n"), 
+      m_msg(file, line, &mdb->errmsg, _("Insertion problem: affected_rows=%s\n"),
         edit_uint64(mdb->num_rows, ed1));
       if (verbose) {
          j_msg(file, line, jcr, M_INFO, 0, "%s\n", cmd);
@@ -142,10 +142,10 @@ InsertDB(char *file, int line, JCR *jcr, B_DB *mdb, char *cmd)
 
 /* Utility routine for updates.
  *  Returns: 0 on failure
- *          1 on success  
+ *          1 on success
  */
 int
-UpdateDB(char *file, int line, JCR *jcr, B_DB *mdb, char *cmd)
+UpdateDB(const char *file, int line, JCR *jcr, B_DB *mdb, char *cmd)
 {
 
    if (sql_query(mdb, cmd)) {
@@ -157,12 +157,12 @@ UpdateDB(char *file, int line, JCR *jcr, B_DB *mdb, char *cmd)
       return 0;
    }
    mdb->num_rows = sql_affected_rows(mdb);
-   if (mdb->num_rows != 1) {
+   if (mdb->num_rows < 1) {
       char ed1[30];
-      m_msg(file, line, &mdb->errmsg, _("Update problem: affected_rows=%s\n"), 
+      m_msg(file, line, &mdb->errmsg, _("Update problem: affected_rows=%s\n"),
         edit_uint64(mdb->num_rows, ed1));
       if (verbose) {
-         j_msg(file, line, jcr, M_INFO, 0, "%s\n", cmd);
+//       j_msg(file, line, jcr, M_INFO, 0, "%s\n", cmd);
       }
       return 0;
    }
@@ -170,13 +170,13 @@ UpdateDB(char *file, int line, JCR *jcr, B_DB *mdb, char *cmd)
    return 1;
 }
 
-/* Utility routine for deletes  
+/* Utility routine for deletes
  *
  * Returns: -1 on error
  *          n number of rows affected
  */
 int
-DeleteDB(char *file, int line, JCR *jcr, B_DB *mdb, char *cmd)
+DeleteDB(const char *file, int line, JCR *jcr, B_DB *mdb, char *cmd)
 {
 
    if (sql_query(mdb, cmd)) {
@@ -209,7 +209,7 @@ int get_sql_record_max(JCR *jcr, B_DB *mdb)
          Mmsg1(&mdb->errmsg, _("error fetching row: %s\n"), sql_strerror(mdb));
         stat = -1;
       } else {
-        stat = atoi(row[0]);
+        stat = str_to_int64(row[0]);
       }
       sql_free_result(mdb);
    } else {
@@ -232,43 +232,48 @@ char *db_strerror(B_DB *mdb)
  *   thread without blocking, but must be unlocked the number of
  *   times it was locked.
  */
-void _db_lock(char *file, int line, B_DB *mdb)
+void _db_lock(const char *file, int line, B_DB *mdb)
 {
    int errstat;
    if ((errstat=rwl_writelock(&mdb->lock)) != 0) {
+      berrno be;
       e_msg(file, line, M_ABORT, 0, "rwl_writelock failure. ERR=%s\n",
-          strerror(errstat));
+          be.strerror(errstat));
    }
-}    
+}
 
 /*
  * Unlock the database. This can be called multiple times by the
  *   same thread up to the number of times that thread called
  *   db_lock()/
  */
-void _db_unlock(char *file, int line, B_DB *mdb)
+void _db_unlock(const char *file, int line, B_DB *mdb)
 {
    int errstat;
    if ((errstat=rwl_writeunlock(&mdb->lock)) != 0) {
+      berrno be;
       e_msg(file, line, M_ABORT, 0, "rwl_writeunlock failure. ERR=%s\n",
-          strerror(errstat));
+          be.strerror(errstat));
    }
-}    
+}
 
 /*
  * Start a transaction. This groups inserts and makes things
- *  much more efficient. Usually started when inserting 
+ *  much more efficient. Usually started when inserting
  *  file attributes.
  */
 void db_start_transaction(JCR *jcr, B_DB *mdb)
 {
-#ifdef xAVE_SQLITE
+#ifdef HAVE_SQLITE
+   if (!mdb->allow_transactions) {
+      return;
+   }
    db_lock(mdb);
    /* Allow only 10,000 changes per transaction */
    if (mdb->transaction && mdb->changes > 10000) {
       db_end_transaction(jcr, mdb);
    }
-   if (!mdb->transaction) {   
+   if (!mdb->transaction) {
       my_sqlite_query(mdb, "BEGIN");  /* begin transaction */
       Dmsg0(400, "Start SQLite transaction\n");
       mdb->transaction = 1;
@@ -276,11 +281,41 @@ void db_start_transaction(JCR *jcr, B_DB *mdb)
    db_unlock(mdb);
 #endif
 
+/*
+ * This is turned off because transactions break
+ * if multiple simultaneous jobs are run.
+ */
+#ifdef HAVE_POSTGRESQL
+   if (!mdb->allow_transactions) {
+      return;
+   }
+   db_lock(mdb);
+   /* Allow only 25,000 changes per transaction */
+   if (mdb->transaction && mdb->changes > 25000) {
+      db_end_transaction(jcr, mdb);
+   }
+   if (!mdb->transaction) {
+      db_sql_query(mdb, "BEGIN", NULL, NULL);  /* begin transaction */
+      Dmsg0(400, "Start PosgreSQL transaction\n");
+      mdb->transaction = 1;
+   }
+   db_unlock(mdb);
+#endif
 }
 
 void db_end_transaction(JCR *jcr, B_DB *mdb)
 {
-#ifdef xAVE_SQLITE
+   /*
+    * This can be called during thread cleanup and
+    *  the db may already be closed.  So simply return.
+    */
+   if (!mdb) {
+      return;
+   }
+#ifdef HAVE_SQLITE
+   if (!mdb->allow_transactions) {
+      return;
+   }
    db_lock(mdb);
    if (mdb->transaction) {
       my_sqlite_query(mdb, "COMMIT"); /* end transaction */
@@ -290,6 +325,20 @@ void db_end_transaction(JCR *jcr, B_DB *mdb)
    mdb->changes = 0;
    db_unlock(mdb);
 #endif
+
+#ifdef HAVE_POSTGRESQL
+   if (!mdb->allow_transactions) {
+      return;
+   }
+   db_lock(mdb);
+   if (mdb->transaction) {
+      db_sql_query(mdb, "COMMIT", NULL, NULL); /* end transaction */
+      mdb->transaction = 0;
+      Dmsg1(400, "End PostgreSQL transaction changes=%d\n", mdb->changes);
+   }
+   mdb->changes = 0;
+   db_unlock(mdb);
+#endif
 }
 
 /*
@@ -301,7 +350,7 @@ void split_path_and_file(JCR *jcr, B_DB *mdb, const char *fname)
 {
    const char *p, *f;
 
-   /* Find path without the filename.  
+   /* Find path without the filename.
     * I.e. everything after the last / is a "filename".
     * OK, maybe it is a directory name, but we treat it like
     * a filename. If we don't find a / then the whole name
@@ -319,7 +368,7 @@ void split_path_and_file(JCR *jcr, B_DB *mdb, const char *fname)
    }
 
    /* If filename doesn't exist (i.e. root directory), we
-    * simply create a blank name consisting of a single 
+    * simply create a blank name consisting of a single
     * space. This makes handling zero length filenames
     * easier.
     */
@@ -329,12 +378,11 @@ void split_path_and_file(JCR *jcr, B_DB *mdb, const char *fname)
       memcpy(mdb->fname, f, mdb->fnl);   /* copy filename */
       mdb->fname[mdb->fnl] = 0;
    } else {
-      mdb->fname[0] = ' ';            /* blank filename */
-      mdb->fname[1] = 0;
-      mdb->fnl = 1;
+      mdb->fname[0] = 0;
+      mdb->fnl = 0;
    }
 
-   mdb->pnl = f - fname;    
+   mdb->pnl = f - fname;
    if (mdb->pnl > 0) {
       mdb->path = check_pool_memory_size(mdb->path, mdb->pnl+1);
       memcpy(mdb->path, fname, mdb->pnl);
@@ -342,12 +390,11 @@ void split_path_and_file(JCR *jcr, B_DB *mdb, const char *fname)
    } else {
       Mmsg1(&mdb->errmsg, _("Path length is zero. File=%s\n"), fname);
       Jmsg(jcr, M_ERROR, 0, "%s", mdb->errmsg);
-      mdb->path[0] = ' ';
-      mdb->path[1] = 0;
-      mdb->pnl = 1;
+      mdb->path[0] = 0;
+      mdb->pnl = 0;
    }
 
-   Dmsg2(100, "sllit path=%s file=%s\n", mdb->path, mdb->fname);
+   Dmsg2(500, "split path=%s file=%s\n", mdb->path, mdb->fname);
 }
 
 /*
@@ -372,8 +419,8 @@ list_dashes(B_DB *mdb, DB_LIST_HANDLER *send, void *ctx)
 }
 
 /*
- * If full_list is set, we list vertically, otherwise, we 
- * list on one line horizontally.      
+ * If full_list is set, we list vertically, otherwise, we
+ * list on one line horizontally.
  */
 void
 list_result(JCR *jcr, B_DB *mdb, DB_LIST_HANDLER *send, void *ctx, e_list_type type)
@@ -383,17 +430,17 @@ list_result(JCR *jcr, B_DB *mdb, DB_LIST_HANDLER *send, void *ctx, e_list_type t
    int i, col_len, max_len = 0;
    char buf[2000], ewc[30];
 
-   Dmsg0(200, "list_result starts\n");
+   Dmsg0(800, "list_result starts\n");
    if (mdb->result == NULL || sql_num_rows(mdb) == 0) {
       send(ctx, _("No results to list.\n"));
       return;
    }
 
-   Dmsg1(200, "list_result starts looking at %d fields\n", sql_num_fields(mdb));
+   Dmsg1(800, "list_result starts looking at %d fields\n", sql_num_fields(mdb));
    /* determine column display widths */
    sql_field_seek(mdb, 0);
    for (i = 0; i < sql_num_fields(mdb); i++) {
-      Dmsg1(200, "list_result processing field %d\n", i);
+      Dmsg1(800, "list_result processing field %d\n", i);
       field = sql_fetch_field(mdb);
       col_len = strlen(field->name);
       if (type == VERT_LIST) {
@@ -414,17 +461,17 @@ list_result(JCR *jcr, B_DB *mdb, DB_LIST_HANDLER *send, void *ctx, e_list_type t
       }
    }
 
-   Dmsg0(200, "list_result finished first loop\n");
+   Dmsg0(800, "list_result finished first loop\n");
    if (type == VERT_LIST) {
       goto vertical_list;
    }
 
-   Dmsg1(200, "list_result starts second loop looking at %d fields\n", sql_num_fields(mdb));
+   Dmsg1(800, "list_result starts second loop looking at %d fields\n", sql_num_fields(mdb));
    list_dashes(mdb, send, ctx);
    send(ctx, "|");
    sql_field_seek(mdb, 0);
    for (i = 0; i < sql_num_fields(mdb); i++) {
-      Dmsg1(200, "list_result looking at field %d\n", i);
+      Dmsg1(800, "list_result looking at field %d\n", i);
       field = sql_fetch_field(mdb);
       bsnprintf(buf, sizeof(buf), " %-*s |", (int)field->max_length, field->name);
       send(ctx, buf);
@@ -432,7 +479,7 @@ list_result(JCR *jcr, B_DB *mdb, DB_LIST_HANDLER *send, void *ctx, e_list_type t
    send(ctx, "\n");
    list_dashes(mdb, send, ctx);
 
-   Dmsg1(200, "list_result starts third loop looking at %d fields\n", sql_num_fields(mdb));
+   Dmsg1(800, "list_result starts third loop looking at %d fields\n", sql_num_fields(mdb));
    while ((row = sql_fetch_row(mdb)) != NULL) {
       sql_field_seek(mdb, 0);
       send(ctx, "|");
@@ -440,8 +487,8 @@ list_result(JCR *jcr, B_DB *mdb, DB_LIST_HANDLER *send, void *ctx, e_list_type t
         field = sql_fetch_field(mdb);
         if (row[i] == NULL) {
             bsnprintf(buf, sizeof(buf), " %-*s |", (int)field->max_length, "NULL");
-        } else if (IS_NUM(field->type) && !jcr->gui) {
-            bsnprintf(buf, sizeof(buf), " %*s |", (int)field->max_length,       
+        } else if (IS_NUM(field->type) && !jcr->gui && is_an_integer(row[i])) {
+            bsnprintf(buf, sizeof(buf), " %*s |", (int)field->max_length,
                      add_commas(row[i], ewc));
         } else {
             bsnprintf(buf, sizeof(buf), " %-*s |", (int)field->max_length, row[i]);
@@ -454,16 +501,16 @@ list_result(JCR *jcr, B_DB *mdb, DB_LIST_HANDLER *send, void *ctx, e_list_type t
    return;
 
 vertical_list:
-   
-   Dmsg1(200, "list_result starts vertical list at %d fields\n", sql_num_fields(mdb));
+
+   Dmsg1(800, "list_result starts vertical list at %d fields\n", sql_num_fields(mdb));
    while ((row = sql_fetch_row(mdb)) != NULL) {
       sql_field_seek(mdb, 0);
       for (i = 0; i < sql_num_fields(mdb); i++) {
         field = sql_fetch_field(mdb);
         if (row[i] == NULL) {
             bsnprintf(buf, sizeof(buf), " %*s: %s\n", max_len, field->name, "NULL");
-        } else if (IS_NUM(field->type) && !jcr->gui) {
-            bsnprintf(buf, sizeof(buf), " %*s: %s\n", max_len, field->name, 
+        } else if (IS_NUM(field->type) && !jcr->gui && is_an_integer(row[i])) {
+            bsnprintf(buf, sizeof(buf), " %*s: %s\n", max_len, field->name,
                add_commas(row[i], ewc));
         } else {
             bsnprintf(buf, sizeof(buf), " %*s: %s\n", max_len, field->name, row[i]);
@@ -476,4 +523,4 @@ vertical_list:
 }
 
 
-#endif /* HAVE_MYSQL || HAVE_SQLITE || HAVE_POSTGRESQL */
+#endif /* HAVE_SQLITE3 || HAVE_MYSQL || HAVE_SQLITE || HAVE_POSTGRESQL*/