fix only cache variable long/short

[oweals/gnunet.git] / src / gns / gnunet-gns.c

diff --git a/src/gns/gnunet-gns.c b/src/gns/gnunet-gns.c
index 1c9c8c5a0ebaff8bf94c2407d3991dda6c0c078e..1f3ce8331585703db23097165b9411a872ada137 100644 (file)
--- a/src/gns/gnunet-gns.c
+++ b/src/gns/gnunet-gns.c
@@ -40,6 +40,11 @@ static const struct GNUNET_CONFIGURATION_Handle *cfg;
  */
 static struct GNUNET_GNS_Handle *gns;
 
+/**
+ * Desired timeout for the lookup (default is no timeout).
+ */
+static struct GNUNET_TIME_Relative timeout;
+
 /**
  * GNS name to lookup. (-u option)
  */
@@ -60,6 +65,11 @@ static char *zone_ego_name;
  */
 static char *public_key;
 
+/**
+ * Set to #GNUNET_YES if we must not use the DHT (only local lookup).
+ */
+static int only_cached;
+
 /**
  * raw output
  */
@@ -206,7 +216,7 @@ lookup_with_keys (const struct GNUNET_CRYPTO_EcdsaPublicKey *pkey,
                                        lookup_name,
                                        pkey,
                                        rtype,
-                                       GNUNET_NO, /* Use DHT */
+                                       only_cached,
                                        shorten_key,
                                        &process_lookup_result,
                                        lookup_name);
@@ -265,6 +275,7 @@ lookup_with_public_key (const struct GNUNET_CRYPTO_EcdsaPublicKey *pkey)
   GNUNET_assert (NULL != pkey);
   pkeym = GNUNET_new (struct GNUNET_CRYPTO_EcdsaPublicKey);
   *pkeym = *pkey;
+  GNUNET_break (NULL == id_op);
   id_op = GNUNET_IDENTITY_get (identity,
                               "gns-short",
                               &identity_shorten_cb,
@@ -327,6 +338,7 @@ identity_master_cb (void *cls,
                    const char *name)
 {
   struct GNUNET_CRYPTO_EcdsaPublicKey pkey;
+  const char *dot;
 
   id_op = NULL;
   if (NULL == ego)
@@ -337,6 +349,11 @@ identity_master_cb (void *cls,
     return;
   }
   GNUNET_IDENTITY_ego_get_public_key (ego, &pkey);
+  /* if the name is of the form 'label.gnu', never go to the DHT */
+  dot = strchr (lookup_name, '.');
+  if ( (NULL != dot) &&
+       (0 == strcasecmp (dot, ".gnu")) )
+    only_cached = GNUNET_YES;
   lookup_with_public_key (&pkey);
 }
 
@@ -364,7 +381,7 @@ run (void *cls, char *const *args, const char *cfgfile,
             _("Failed to connect to GNS\n"));
     return;
   }
-  GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_FOREVER_REL,
+  GNUNET_SCHEDULER_add_delayed (timeout,
                                &do_shutdown, NULL);
   if (NULL != public_key)
   {
@@ -402,6 +419,7 @@ run (void *cls, char *const *args, const char *cfgfile,
   }
   else
   {
+    GNUNET_break (NULL == id_op);
     id_op = GNUNET_IDENTITY_get (identity,
                                 "gns-master",
                                 &identity_master_cb,
@@ -428,6 +446,9 @@ main (int argc, char *const *argv)
     {'t', "type", "TYPE",
       gettext_noop ("Specify the type of the record to lookup"), 1,
       &GNUNET_GETOPT_set_string, &lookup_type},
+    { 'T', "timeout", "DELAY",
+      gettext_noop ("Specify timeout for the lookup"), 1,
+      &GNUNET_GETOPT_set_relative_time, &timeout },
     {'r', "raw", NULL,
       gettext_noop ("No unneeded output"), 0,
       &GNUNET_GETOPT_set_one, &raw},
@@ -441,6 +462,7 @@ main (int argc, char *const *argv)
   };
   int ret;
 
+  timeout = GNUNET_TIME_UNIT_FOREVER_REL;
   if (GNUNET_OK != GNUNET_STRINGS_get_utf8_args (argc, argv, &argc, &argv))
     return 2;