challenger-dbinit.c (3979B)
1 /* 2 This file is part of TALER 3 Copyright (C) 2023 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 challengerdb/challenger-dbinit.c 18 * @brief Create tables for the challenger database. 19 * @author Christian Grothoff 20 */ 21 #include "platform.h" 22 #include <gnunet/gnunet_util_lib.h> 23 #include "challenger_util.h" 24 #include "challenger_database_lib.h" 25 #include "challenger-database/create_tables.h" 26 #include "challenger-database/drop_tables.h" 27 #include "challenger-database/gc.h" 28 29 30 /** 31 * Return value from main(). 32 */ 33 static int global_ret; 34 35 /** 36 * -r option: do full DB reset 37 */ 38 static int reset_db; 39 40 /** 41 * -g option: do GC reset 42 */ 43 static int gc_db; 44 45 46 /** 47 * Main function that will be run. 48 * 49 * @param cls closure 50 * @param args remaining command-line arguments 51 * @param cfgfile name of the configuration file used (for saving, can be NULL!) 52 * @param cfg configuration 53 */ 54 static void 55 run (void *cls, 56 char *const *args, 57 const char *cfgfile, 58 const struct GNUNET_CONFIGURATION_Handle *cfg) 59 { 60 struct CHALLENGERDB_PostgresContext *db; 61 62 (void) cls; 63 (void) args; 64 (void) cfgfile; 65 if (NULL == 66 (db = CHALLENGERDB_connect (cfg, 67 true))) 68 { 69 fprintf (stderr, 70 "Failed to initialize database connection.\n"); 71 global_ret = EXIT_NOTINSTALLED; 72 return; 73 } 74 if (reset_db) 75 { 76 if (GNUNET_OK != CHALLENGERDB_drop_tables (db)) 77 { 78 GNUNET_log (GNUNET_ERROR_TYPE_ERROR, 79 "Could not drop tables as requested. Either database was not yet initialized, or permission denied. Consult the logs. Will still try to create new tables.\n"); 80 } 81 } 82 if (GNUNET_OK != 83 CHALLENGERDB_create_tables (db)) 84 { 85 global_ret = EXIT_FAILURE; 86 CHALLENGERDB_disconnect (db); 87 return; 88 } 89 if (gc_db) 90 { 91 struct GNUNET_TIME_Absolute now; 92 93 now = GNUNET_TIME_absolute_get (); 94 if (0 > 95 CHALLENGERDB_gc (db, 96 now)) 97 { 98 fprintf (stderr, 99 "Garbage collection failed!\n"); 100 global_ret = EXIT_FAILURE; 101 } 102 } 103 CHALLENGERDB_disconnect (db); 104 } 105 106 107 /** 108 * The main function of the database initialization tool. 109 * Used to initialize the Challenger' database. 110 * 111 * @param argc number of arguments from the command line 112 * @param argv command line arguments 113 * @return 0 ok, non-zero on error 114 */ 115 int 116 main (int argc, 117 char *const *argv) 118 { 119 struct GNUNET_GETOPT_CommandLineOption options[] = { 120 GNUNET_GETOPT_option_flag ('r', 121 "reset", 122 "reset database (DANGEROUS: all existing data is lost!)", 123 &reset_db), 124 GNUNET_GETOPT_option_flag ('g', 125 "garbagecollect", 126 "remove state data from database", 127 &gc_db), 128 GNUNET_GETOPT_OPTION_END 129 }; 130 enum GNUNET_GenericReturnValue ret; 131 132 ret = GNUNET_PROGRAM_run (CHALLENGER_project_data (), 133 argc, argv, 134 "challenger-dbinit", 135 "Initialize challenger database", 136 options, 137 &run, NULL); 138 if (GNUNET_SYSERR == ret) 139 return EXIT_INVALIDARGUMENT; 140 if (GNUNET_NO == ret) 141 return EXIT_SUCCESS; 142 return global_ret; 143 } 144 145 146 /* end of challenger-dbinit.c */