exchange

Base system with REST service to issue digital coins, run by the payment service provider
Log | Files | Refs | Submodules | README | LICENSE

taler-auditor-dbinit.c (4368B)

      1 /*
      2   This file is part of TALER
      3   Copyright (C) 2014, 2015, 2020 Taler Systems SA
      4 
      5   TALER is free software; you can redistribute it and/or modify it under the
      6   terms of the GNU General Public License as published by the Free Software
      7   Foundation; either version 3, or (at your option) any later version.
      8 
      9   TALER is distributed in the hope that it will be useful, but WITHOUT ANY
     10   WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
     11   A PARTICULAR PURPOSE.  See the GNU General Public License for more details.
     12 
     13   You should have received a copy of the GNU General Public License along with
     14   TALER; see the file COPYING.  If not, see <http://www.gnu.org/licenses/>
     15 */
     16 /**
     17  * @file auditor/taler-auditor-dbinit.c
     18  * @brief Create tables for the auditor database.
     19  * @author Florian Dold
     20  * @author Marcello Stanisci
     21  */
     22 #include "platform.h"
     23 #include <gnunet/gnunet_util_lib.h>
     24 #include "taler/taler_util.h"
     25 #include "auditordb_lib.h"
     26 #include "auditor-database/gc.h"
     27 #include "auditor-database/drop_tables.h"
     28 #include "auditor-database/create_tables.h"
     29 
     30 
     31 /**
     32  * Return value from main().
     33  */
     34 static int global_ret;
     35 
     36 /**
     37  * -r option: do restart audits
     38  */
     39 static int restart_db;
     40 
     41 /**
     42  * -R option: do full DB reset
     43  */
     44 static int reset_db;
     45 
     46 /**
     47  * -g option: garbage collect DB reset
     48  */
     49 static int gc_db;
     50 
     51 
     52 /**
     53  * Main function that will be run.
     54  *
     55  * @param cls closure
     56  * @param args remaining command-line arguments
     57  * @param cfgfile name of the configuration file used (for saving, can be NULL!)
     58  * @param cfg configuration
     59  */
     60 static void
     61 run (void *cls,
     62      char *const *args,
     63      const char *cfgfile,
     64      const struct GNUNET_CONFIGURATION_Handle *cfg)
     65 {
     66   struct TALER_AUDITORDB_PostgresContext *pg;
     67 
     68   (void) cls;
     69   (void) args;
     70   (void) cfgfile;
     71   if (NULL ==
     72       (pg = TALER_AUDITORDB_connect (cfg,
     73                                      true)))
     74   {
     75     fprintf (stderr,
     76              "Failed to initialize database plugin.\n");
     77     global_ret = EXIT_NOTINSTALLED;
     78     return;
     79   }
     80   if (reset_db)
     81   {
     82     if (GNUNET_OK !=
     83         TALER_AUDITORDB_drop_tables (pg,
     84                                      GNUNET_YES))
     85       GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
     86                   "Failed to reset database\n");
     87   }
     88   else if (restart_db)
     89   {
     90     if (GNUNET_OK !=
     91         TALER_AUDITORDB_drop_tables (pg,
     92                                      GNUNET_NO))
     93       GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
     94                   "Failed to restart audits\n");
     95   }
     96   if (GNUNET_OK !=
     97       TALER_AUDITORDB_create_tables (cfg,
     98                                      false,
     99                                      0))
    100   {
    101     fprintf (stderr,
    102              "Failed to initialize database.\n");
    103     TALER_AUDITORDB_disconnect (pg);
    104     global_ret = EXIT_NOPERMISSION;
    105     return;
    106   }
    107   if (gc_db)
    108   {
    109     if (GNUNET_SYSERR == TALER_AUDITORDB_gc (pg))
    110       fprintf (stderr,
    111                "Garbage collection failed!\n");
    112   }
    113   TALER_AUDITORDB_disconnect (pg);
    114 }
    115 
    116 
    117 /**
    118  * The main function of the database initialization tool.
    119  * Used to initialize the Taler auditor's database.
    120  *
    121  * @param argc number of arguments from the command line
    122  * @param argv command line arguments
    123  * @return 0 ok, 1 on error
    124  */
    125 int
    126 main (int argc,
    127       char *const *argv)
    128 {
    129   const struct GNUNET_GETOPT_CommandLineOption options[] = {
    130     GNUNET_GETOPT_option_flag ('r',
    131                                "restart",
    132                                "restart audits (DANGEROUS: all audits resume from scratch)",
    133                                &restart_db),
    134     GNUNET_GETOPT_option_flag ('R',
    135                                "reset",
    136                                "reset database (DANGEROUS: all existing data is lost!)",
    137                                &reset_db),
    138     GNUNET_GETOPT_option_flag ('g',
    139                                "gc",
    140                                "garbage collect database",
    141                                &gc_db),
    142     GNUNET_GETOPT_OPTION_END
    143   };
    144   enum GNUNET_GenericReturnValue ret;
    145 
    146   ret = GNUNET_PROGRAM_run (
    147     TALER_AUDITOR_project_data (),
    148     argc, argv,
    149     "taler-auditor-dbinit",
    150     gettext_noop ("Initialize Taler auditor database"),
    151     options,
    152     &run, NULL);
    153   if (GNUNET_SYSERR == ret)
    154     return EXIT_INVALIDARGUMENT;
    155   if (GNUNET_NO == ret)
    156     return EXIT_SUCCESS;
    157   return global_ret;
    158 }
    159 
    160 
    161 /* end of taler-auditor-dbinit.c */