gnunet-dht-put.c

Go to the documentation of this file.

/*
     This file is part of GNUnet.
     Copyright (C) 2001, 2002, 2004, 2005, 2006, 2007, 2009, 2017 GNUnet e.V.
 
     GNUnet is free software: you can redistribute it and/or modify it
     under the terms of the GNU Affero General Public License as published
     by the Free Software Foundation, either version 3 of the License,
     or (at your option) any later version.
 
     GNUnet 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
     Affero General Public License for more details.
 
     You should have received a copy of the GNU Affero General Public License
     along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
     SPDX-License-Identifier: AGPL3.0-or-later
 */
#include "platform.h"
#include "gnunet_dht_service.h"
 
static unsigned int query_type;
 
struct GNUNET_HashCode key;
 
static char *query_key;
 
static int raw_key;
 
static struct GNUNET_TIME_Relative expiration;
 
static unsigned int replication = 5;
 
static unsigned int verbose;
 
static int demultixplex_everywhere;
 
static int record_route;
 
static struct GNUNET_DHT_Handle *dht_handle;
 
 
static const struct GNUNET_CONFIGURATION_Handle *cfg;
 
static int ret;
 
static char *data;
 
 
static void
shutdown_task (void *cls)
{
  if (NULL != dht_handle)
  {
    GNUNET_DHT_disconnect (dht_handle);
    dht_handle = NULL;
  }
}
 
 
static void
message_sent_cont (void *cls)
{
  GNUNET_SCHEDULER_add_now (&shutdown_task, NULL);
}
 
 
static void
run (void *cls,
     char *const *args,
     const char *cfgfile,
     const struct GNUNET_CONFIGURATION_Handle *c)
{
  enum GNUNET_DHT_RouteOption ro;
 
  cfg = c;
  if ((NULL == query_key) || (NULL == data))
  {
    fprintf (stderr, "%s", _ ("Must provide KEY and DATA for DHT put!\n"));
    ret = 1;
    return;
  }
 
  /* Resolve the key before connecting, so that a bad key does not leak
     the DHT handle. */
  if (raw_key)
  {
    if (GNUNET_OK !=
        GNUNET_CRYPTO_hash_from_string (query_key,
                                        &key))
    {
      fprintf (stderr,
               _ ("`%s' is not a valid key\n"),
               query_key);
      ret = 1;
      return;
    }
  }
  else
  {
    GNUNET_CRYPTO_hash (query_key,
                        strlen (query_key),
                        &key);
  }
  if (NULL == (dht_handle = GNUNET_DHT_connect (cfg, 1)))
  {
    fprintf (stderr, _ ("Could not connect to DHT service!\n"));
    ret = 1;
    return;
  }
  if (GNUNET_BLOCK_TYPE_ANY == query_type) /* Type of data not set */
    query_type = GNUNET_BLOCK_TYPE_TEST;
 
  if (verbose)
    fprintf (stderr,
             _ ("Issuing put request for `%s' with data `%s'!\n"),
             query_key,
             data);
  ro = GNUNET_DHT_RO_NONE;
  if (demultixplex_everywhere)
    ro |= GNUNET_DHT_RO_DEMULTIPLEX_EVERYWHERE;
  if (record_route)
    ro |= GNUNET_DHT_RO_RECORD_ROUTE;
  GNUNET_DHT_put (dht_handle,
                  &key,
                  replication,
                  ro,
                  query_type,
                  strlen (data),
                  data,
                  GNUNET_TIME_relative_to_absolute (expiration),
                  &message_sent_cont,
                  NULL);
}
 
 
int
main (int argc, char *const *argv)
{
  struct GNUNET_GETOPT_CommandLineOption options[] = {
    GNUNET_GETOPT_option_string (
      'd',
      "data",
      "DATA",
      gettext_noop (
        "the data to insert under the key"),
      &data),
    GNUNET_GETOPT_option_relative_time (
      'e',
      "expiration",
      "EXPIRATION",
      gettext_noop ("how long to store this entry in the dht (in seconds)"),
      &expiration),
    GNUNET_GETOPT_option_string (
      'k',
      "key",
      "KEY",
      gettext_noop (
        "the string to hash into the query key (see -K to give the key itself)")
      ,
      &query_key),
    GNUNET_GETOPT_option_flag (
      'K',
      "raw-key",
      gettext_noop (
        "interpret -k as the query key itself, in the base32 form GNUnet prints it, instead of hashing it"),
      &raw_key),
    GNUNET_GETOPT_option_flag (
      'x',
      "demultiplex",
      gettext_noop (
        "use DHT's demultiplex everywhere option"),
      &demultixplex_everywhere),
    GNUNET_GETOPT_option_uint (
      'r',
      "replication",
      "LEVEL",
      gettext_noop ("how many replicas to create"),
      &replication),
    GNUNET_GETOPT_option_flag (
      'R',
      "record",
      gettext_noop ("use DHT's record route option"),
      &record_route),
    GNUNET_GETOPT_option_uint (
      't',
      "type",
      "TYPE",
      gettext_noop ("the type to insert data as"),
      &query_type),
    GNUNET_GETOPT_option_verbose (&verbose),
    GNUNET_GETOPT_OPTION_END
  };
 
 
  expiration = GNUNET_TIME_UNIT_HOURS;
  ret = (GNUNET_OK ==
         GNUNET_PROGRAM_run (
           GNUNET_OS_project_data_gnunet (),
           argc,
           argv,
           "gnunet-dht-put",
           gettext_noop (
             "Issue a PUT request to the GNUnet DHT insert DATA under KEY."),
           options,
           &run,
           NULL))
         ? ret
         : 1;
  return ret;
}
 
 
/* end of gnunet-dht-put.c */