1 files changed, 215 insertions, 0 deletions
diff --git a/src/mint/taler-mint-reservemod.c b/src/mint/taler-mint-reservemod.c
new file mode 100644
index 000000000..3dd94f84b
--- /dev/null
+++ b/src/mint/taler-mint-reservemod.c
@@ -0,0 +1,215 @@
+/*
+  This file is part of TALER
+  (C) 2014 Christian Grothoff (and other contributing authors)
+
+  TALER is free software; you can redistribute it and/or modify it under the
+  terms of the GNU General Public License as published by the Free Software
+  Foundation; either version 3, or (at your option) any later version.
+
+  TALER is distributed in the hope that it will be useful, but WITHOUT ANY
+  WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
+  A PARTICULAR PURPOSE.  See the GNU General Public License for more details.
+
+  You should have received a copy of the GNU General Public License along with
+  TALER; see the file COPYING.  If not, If not, see <http://www.gnu.org/licenses/>
+*/
+
+/**
+ * @file taler-mint-reservemod.c
+ * @brief Modify reserves.
+ * @author Florian Dold
+ * @author Benedikt Mueller
+ */
+
+#include "platform.h"
+#include <gnunet/gnunet_util_lib.h>
+#include <libpq-fe.h>
+#include "taler_util.h"
+#include "taler_signatures.h"
+#include "mint.h"
+
+static char *mintdir;
+static struct GNUNET_CRYPTO_EddsaPublicKey *reserve_pub;
+static struct GNUNET_CONFIGURATION_Handle *kcfg;
+static PGconn *db_conn;
+
+
+
+/**
+ * Create a new or add to existing reserve.
+ * Fails if currencies do not match.
+ * 
+ * @param denom denomination to add
+ *
+ * @return ...
+ */
+int
+reservemod_add (struct TALER_Amount denom)
+{
+  PGresult *result;
+  {
+    const void *param_values[] = { reserve_pub };
+    int param_lengths[] = {sizeof(struct GNUNET_CRYPTO_EddsaPublicKey)};
+    int param_formats[] = {1};
+    result = PQexecParams (db_conn,
+                           "select balance_value, balance_fraction, balance_currency from reserves where reserve_pub=$1 limit 1;",
+                           1, NULL, (const char * const *) param_values, param_lengths, param_formats, 1);
+  }
+
+  if (PGRES_TUPLES_OK != PQresultStatus (result))
+  {
+    fprintf (stderr, "Select failed: %s\n", PQresultErrorMessage (result));
+    return GNUNET_SYSERR;
+  }
+  if (0 == PQntuples (result))
+  {
+    struct GNUNET_TIME_AbsoluteNBO exnbo;
+    exnbo = GNUNET_TIME_absolute_hton (GNUNET_TIME_absolute_add ( GNUNET_TIME_absolute_get (), GNUNET_TIME_UNIT_YEARS));
+
+    uint32_t value = htonl (denom.value);
+    uint32_t fraction = htonl (denom.fraction);
+    const void *param_values[] = {
+      reserve_pub,
+      &value,
+      &fraction,
+      denom.currency, 
+      &exnbo};
+    int param_lengths[] = {32, 4, 4, strlen(denom.currency), 8};
+    int param_formats[] = {1, 1, 1, 1, 1};
+    result = PQexecParams (db_conn,
+                           "insert into reserves (reserve_pub, balance_value, balance_fraction, balance_currency, "
+                           " expiration_date )"
+                           "values ($1,$2,$3,$4,$5);",
+                           5, NULL, (const char **) param_values, param_lengths, param_formats, 1);
+  
+    if (PGRES_COMMAND_OK != PQresultStatus (result))
+    {
+      fprintf (stderr, "Insert failed: %s\n", PQresultErrorMessage (result));
+      return GNUNET_SYSERR;
+    }
+  } 
+  else 
+  {
+    struct TALER_Amount old_denom;
+    struct TALER_Amount new_denom;
+    struct TALER_AmountNBO new_denom_nbo;
+    int denom_indices[] = {0, 1, 2};
+    int param_lengths[] = {4, 4, 32};
+    int param_formats[] = {1, 1, 1};
+    const void *param_values[] = {
+      &new_denom_nbo.value,
+      &new_denom_nbo.fraction,
+      reserve_pub
+    };
+
+    GNUNET_assert (GNUNET_OK == TALER_TALER_DB_extract_amount (result, 0, denom_indices, &old_denom));
+    new_denom = TALER_amount_add (old_denom, denom);
+    new_denom_nbo = TALER_amount_hton (new_denom);
+    result = PQexecParams (db_conn,
+                           "UPDATE reserves "
+                           "SET balance_value = $1, balance_fraction = $2, "
+                           " status_sig = NULL, status_sign_pub = NULL "
+                           "WHERE reserve_pub = $3 ",
+                           3, NULL, (const char **) param_values, param_lengths, param_formats, 1);
+
+    if (PGRES_COMMAND_OK != PQresultStatus (result))
+    {
+      fprintf (stderr, "Update failed: %s\n", PQresultErrorMessage (result));
+      return GNUNET_SYSERR;
+    }
+
+    if (0 != strcmp ("1", PQcmdTuples (result)))
+    {
+      fprintf (stderr, "Update failed (updated '%s' tupes instead of '1')\n",
+               PQcmdTuples (result));
+      return GNUNET_SYSERR;
+    }
+
+  } 
+  return GNUNET_OK; 
+}
+
+
+/**
+ * The main function of the reservemod tool
+ *
+ * @param argc number of arguments from the command line
+ * @param argv command line arguments
+ * @return 0 ok, 1 on error
+ */
+int
+main (int argc, char *const *argv)
+{
+  static char *reserve_pub_str;
+  static char *add_str;
+  static const struct GNUNET_GETOPT_CommandLineOption options[] = {
+    GNUNET_GETOPT_OPTION_HELP ("gnunet-mint-keyup OPTIONS"),
+    {'d', "mint-dir", "DIR",
+     "mint directory with keys to update", 1,
+     &GNUNET_GETOPT_set_filename, &mintdir},
+    {'R', "reserve", "KEY",
+     "reserve (public key) to modify", 1,
+     &GNUNET_GETOPT_set_string, &reserve_pub_str},
+    {'a', "add", "DENOM",
+     "value to add", 1,
+     &GNUNET_GETOPT_set_string, &add_str},
+    GNUNET_GETOPT_OPTION_END
+  };
+  char *TALER_MINT_db_connection_cfg_str;
+
+  GNUNET_assert (GNUNET_OK == GNUNET_log_setup ("taler-mint-keycheck", "WARNING", NULL));
+
+  if (GNUNET_GETOPT_run ("taler-mint-keyup", options, argc, argv) < 0) 
+    return 1;
+  if (NULL == mintdir)
+  {
+    fprintf (stderr, "mint directory not given\n"); 
+    return 1;
+  }
+
+  reserve_pub = GNUNET_new (struct GNUNET_CRYPTO_EddsaPublicKey);
+  if ((NULL == reserve_pub_str) ||
+      (GNUNET_OK != GNUNET_STRINGS_string_to_data (reserve_pub_str,
+                                                   strlen (reserve_pub_str), 
+                                                   reserve_pub,
+                                                   sizeof (struct GNUNET_CRYPTO_EddsaPublicKey))))
+  {
+    fprintf (stderr, "reserve key invalid\n");
+    return 1;
+  }
+
+  kcfg = TALER_MINT_config_load (mintdir);
+  if (NULL == kcfg)
+  {
+    fprintf (stderr, "can't load mint configuration\n");
+    return 1;
+  }
+  if (GNUNET_OK != GNUNET_CONFIGURATION_get_value_string (kcfg, "mint", "db", &TALER_MINT_db_connection_cfg_str))
+  {
+    fprintf (stderr, "db configuration string not found\n");
+    return 42;
+  }
+  db_conn = PQconnectdb (TALER_MINT_db_connection_cfg_str);
+  if (CONNECTION_OK != PQstatus (db_conn))
+  {
+    fprintf (stderr, "db connection failed: %s\n", PQerrorMessage (db_conn));
+    return 1;
+  }
+
+  if (NULL != add_str)
+  {
+    struct TALER_Amount add_value;
+    if (GNUNET_OK != TALER_string_to_amount (add_str, &add_value))
+    {
+      fprintf (stderr, "could not read value\n");
+      return 1;
+    }
+    if (GNUNET_OK != reservemod_add (add_value))
+    {
+      fprintf (stderr, "adding value failed\n");
+      return 1;
+    }
+  }
+  return 0;
+}
+