Line data    Source code

       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 "taler/platform.h"
      23              : #include <gnunet/gnunet_util_lib.h>
      24              : #include "taler/taler_util.h"
      25              : #include "taler/taler_auditordb_plugin.h"
      26              : 
      27              : 
      28              : /**
      29              :  * Return value from main().
      30              :  */
      31              : static int global_ret;
      32              : 
      33              : /**
      34              :  * -r option: do restart audits
      35              :  */
      36              : static int restart_db;
      37              : 
      38              : /**
      39              :  * -R option: do full DB reset
      40              :  */
      41              : static int reset_db;
      42              : 
      43              : /**
      44              :  * -g option: garbage collect DB reset
      45              :  */
      46              : static int gc_db;
      47              : 
      48              : 
      49              : /**
      50              :  * Main function that will be run.
      51              :  *
      52              :  * @param cls closure
      53              :  * @param args remaining command-line arguments
      54              :  * @param cfgfile name of the configuration file used (for saving, can be NULL!)
      55              :  * @param cfg configuration
      56              :  */
      57              : static void
      58            5 : run (void *cls,
      59              :      char *const *args,
      60              :      const char *cfgfile,
      61              :      const struct GNUNET_CONFIGURATION_Handle *cfg)
      62              : {
      63              :   struct TALER_AUDITORDB_Plugin *plugin;
      64              : 
      65              :   (void) cls;
      66              :   (void) args;
      67              :   (void) cfgfile;
      68            5 :   if (NULL ==
      69            5 :       (plugin = TALER_AUDITORDB_plugin_load (cfg,
      70              :                                              true)))
      71              :   {
      72            0 :     fprintf (stderr,
      73              :              "Failed to initialize database plugin.\n");
      74            0 :     global_ret = EXIT_NOTINSTALLED;
      75            0 :     return;
      76              :   }
      77            5 :   if (reset_db)
      78              :   {
      79            4 :     if (GNUNET_OK !=
      80            4 :         plugin->drop_tables (plugin->cls,
      81              :                              GNUNET_YES))
      82            0 :       GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
      83              :                   "Failed to reset database\n");
      84              :   }
      85            1 :   else if (restart_db)
      86              :   {
      87            0 :     if (GNUNET_OK !=
      88            0 :         plugin->drop_tables (plugin->cls,
      89              :                              GNUNET_NO))
      90            0 :       GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
      91              :                   "Failed to restart audits\n");
      92              :   }
      93            5 :   if (GNUNET_OK !=
      94            5 :       plugin->create_tables (plugin->cls,
      95              :                              false,
      96              :                              0))
      97              :   {
      98            0 :     fprintf (stderr,
      99              :              "Failed to initialize database.\n");
     100            0 :     TALER_AUDITORDB_plugin_unload (plugin);
     101            0 :     global_ret = EXIT_NOPERMISSION;
     102            0 :     return;
     103              :   }
     104            5 :   if (gc_db)
     105              :   {
     106            0 :     if (GNUNET_SYSERR == plugin->gc (plugin->cls))
     107            0 :       fprintf (stderr,
     108              :                "Garbage collection failed!\n");
     109              :   }
     110            5 :   TALER_AUDITORDB_plugin_unload (plugin);
     111              : }
     112              : 
     113              : 
     114              : /**
     115              :  * The main function of the database initialization tool.
     116              :  * Used to initialize the Taler auditor's database.
     117              :  *
     118              :  * @param argc number of arguments from the command line
     119              :  * @param argv command line arguments
     120              :  * @return 0 ok, 1 on error
     121              :  */
     122              : int
     123            5 : main (int argc,
     124              :       char *const *argv)
     125              : {
     126            5 :   const struct GNUNET_GETOPT_CommandLineOption options[] = {
     127            5 :     GNUNET_GETOPT_option_flag ('r',
     128              :                                "restart",
     129              :                                "restart audits (DANGEROUS: all audits resume from scratch)",
     130              :                                &restart_db),
     131            5 :     GNUNET_GETOPT_option_flag ('R',
     132              :                                "reset",
     133              :                                "reset database (DANGEROUS: all existing data is lost!)",
     134              :                                &reset_db),
     135            5 :     GNUNET_GETOPT_option_flag ('g',
     136              :                                "gc",
     137              :                                "garbage collect database",
     138              :                                &gc_db),
     139              :     GNUNET_GETOPT_OPTION_END
     140              :   };
     141              :   enum GNUNET_GenericReturnValue ret;
     142              : 
     143            5 :   ret = GNUNET_PROGRAM_run (
     144              :     TALER_AUDITOR_project_data (),
     145              :     argc, argv,
     146              :     "taler-auditor-dbinit",
     147              :     gettext_noop ("Initialize Taler auditor database"),
     148              :     options,
     149              :     &run, NULL);
     150            5 :   if (GNUNET_SYSERR == ret)
     151            0 :     return EXIT_INVALIDARGUMENT;
     152            5 :   if (GNUNET_NO == ret)
     153            0 :     return EXIT_SUCCESS;
     154            5 :   return global_ret;
     155              : }
     156              : 
     157              : 
     158              : /* end of taler-auditor-dbinit.c */