src/fs/gnunet-search.c

   1 /*
   2      This file is part of GNUnet.
   3      (C) 2001, 2002, 2004, 2005, 2006, 2007, 2009 Christian Grothoff (and other contributing authors)
   4
   5      GNUnet is free software; you can redistribute it and/or modify
   6      it under the terms of the GNU General Public License as published
   7      by the Free Software Foundation; either version 3, or (at your
   8      option) any later version.
   9
  10      GNUnet is distributed in the hope that it will be useful, but
  11      WITHOUT ANY WARRANTY; without even the implied warranty of
  12      MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13      General Public License for more details.
  14
  15      You should have received a copy of the GNU General Public License
  16      along with GNUnet; see the file COPYING.  If not, write to the
  17      Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  18      Boston, MA 02111-1307, USA.
  19 */
  20 /**
  21  * @file fs/gnunet-search.c
  22  * @brief searching for files on GNUnet
  23  * @author Christian Grothoff
  24  * @author Krista Bennett
  25  * @author James Blackwell
  26  * @author Igor Wronsky
  27  */
  28 #include "platform.h"
  29 #include "gnunet_fs_service.h"
  30
  31 static int ret;
  32
  33 static const struct GNUNET_CONFIGURATION_Handle *cfg;
  34
  35 static struct GNUNET_FS_Handle *ctx;
  36
  37 static struct GNUNET_FS_SearchContext *sc;
  38
  39 static char *output_filename;
  40
  41 static struct GNUNET_FS_DirectoryBuilder *db;
  42
  43 static unsigned int anonymity = 1;
  44
  45 /**
  46  * Timeout for the search, 0 means to wait for CTRL-C.
  47  */
  48 static struct GNUNET_TIME_Relative timeout;
  49
  50 static unsigned int results_limit;
  51
  52 static unsigned int results;
  53
  54 static int verbose;
  55
  56 static int local_only;
  57
  58 /**
  59  * Type of a function that libextractor calls for each
  60  * meta data item found.
  61  *
  62  * @param cls closure (user-defined, unused)
  63  * @param plugin_name name of the plugin that produced this value;
  64  *        special values can be used (i.e. '&lt;zlib&gt;' for zlib being
  65  *        used in the main libextractor library and yielding
  66  *        meta data).
  67  * @param type libextractor-type describing the meta data
  68  * @param format basic format information about data
  69  * @param data_mime_type mime-type of data (not of the original file);
  70  *        can be NULL (if mime-type is not known)
  71  * @param data actual meta-data found
  72  * @param data_size number of bytes in data
  73  * @return 0 to continue extracting, 1 to abort
  74  */
  75 static int
  76 item_printer (void *cls, const char *plugin_name, enum EXTRACTOR_MetaType type,
  77               enum EXTRACTOR_MetaFormat format, const char *data_mime_type,
  78               const char *data, size_t data_size)
  79 {
  80   if ((format != EXTRACTOR_METAFORMAT_UTF8) &&
  81       (format != EXTRACTOR_METAFORMAT_C_STRING))
  82     return 0;
  83   if (type == EXTRACTOR_METATYPE_GNUNET_ORIGINAL_FILENAME)
  84     return 0;
  85   printf ("\t%20s: %s\n",
  86           dgettext (LIBEXTRACTOR_GETTEXT_DOMAIN,
  87                     EXTRACTOR_metatype_to_string (type)), data);
  88   return 0;
  89 }
  90
  91
  92 static void
  93 clean_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
  94 {
  95   size_t dsize;
  96   void *ddata;
  97
  98   GNUNET_FS_stop (ctx);
  99   ctx = NULL;
 100   if (output_filename == NULL)
 101     return;
 102   if (GNUNET_OK != GNUNET_FS_directory_builder_finish (db, &dsize, &ddata))
 103   {
 104     GNUNET_break (0);
 105     GNUNET_free (output_filename);
 106     return;
 107   }
 108   if (dsize !=
 109       GNUNET_DISK_fn_write (output_filename, ddata, dsize,
 110                             GNUNET_DISK_PERM_USER_READ |
 111                             GNUNET_DISK_PERM_USER_WRITE))
 112   {
 113     FPRINTF (stderr,
 114              _("Failed to write directory with search results to `%s'\n"),
 115              output_filename);
 116   }
 117   GNUNET_free_non_null (ddata);
 118   GNUNET_free (output_filename);
 119 }
 120
 121
 122 /**
 123  * Called by FS client to give information about the progress of an
 124  * operation.
 125  *
 126  * @param cls closure
 127  * @param info details about the event, specifying the event type
 128  *        and various bits about the event
 129  * @return client-context (for the next progress call
 130  *         for this operation; should be set to NULL for
 131  *         SUSPEND and STOPPED events).  The value returned
 132  *         will be passed to future callbacks in the respective
 133  *         field in the GNUNET_FS_ProgressInfo struct.
 134  */
 135 static void *
 136 progress_cb (void *cls, const struct GNUNET_FS_ProgressInfo *info)
 137 {
 138   static unsigned int cnt;
 139   char *uri;
 140   char *dotdot;
 141   char *filename;
 142
 143   switch (info->status)
 144   {
 145   case GNUNET_FS_STATUS_SEARCH_START:
 146     break;
 147   case GNUNET_FS_STATUS_SEARCH_RESULT:
 148     if (db != NULL)
 149       GNUNET_FS_directory_builder_add (db,
 150                                        info->value.search.specifics.result.uri,
 151                                        info->value.search.specifics.result.meta,
 152                                        NULL);
 153     uri = GNUNET_FS_uri_to_string (info->value.search.specifics.result.uri);
 154     printf ("#%u:\n", cnt++);
 155     filename =
 156         GNUNET_CONTAINER_meta_data_get_by_type (info->value.search.
 157                                                 specifics.result.meta,
 158                                                 EXTRACTOR_METATYPE_GNUNET_ORIGINAL_FILENAME);
 159     if (filename != NULL)
 160     {
 161       while (NULL != (dotdot = strstr (filename, "..")))
 162         dotdot[0] = dotdot[1] = '_';
 163       printf ("gnunet-download -o \"%s\" %s\n", filename, uri);
 164     }
 165     else
 166       printf ("gnunet-download %s\n", uri);
 167     if (verbose)
 168       GNUNET_CONTAINER_meta_data_iterate (info->value.search.specifics.
 169                                           result.meta, &item_printer, NULL);
 170     printf ("\n");
 171     fflush (stdout);
 172     GNUNET_free_non_null (filename);
 173     GNUNET_free (uri);
 174     results++;
 175     if ((results_limit > 0) && (results >= results_limit))
 176       GNUNET_SCHEDULER_shutdown ();
 177     break;
 178   case GNUNET_FS_STATUS_SEARCH_UPDATE:
 179     break;
 180   case GNUNET_FS_STATUS_SEARCH_RESULT_STOPPED:
 181     /* ignore */
 182     break;
 183   case GNUNET_FS_STATUS_SEARCH_ERROR:
 184     FPRINTF (stderr, _("Error searching: %s.\n"),
 185              info->value.search.specifics.error.message);
 186     GNUNET_SCHEDULER_shutdown ();
 187     break;
 188   case GNUNET_FS_STATUS_SEARCH_STOPPED:
 189     GNUNET_SCHEDULER_add_continuation (&clean_task, NULL,
 190                                        GNUNET_SCHEDULER_REASON_PREREQ_DONE);
 191     break;
 192   default:
 193     FPRINTF (stderr, _("Unexpected status: %d\n"), info->status);
 194     break;
 195   }
 196   return NULL;
 197 }
 198
 199
 200 static void
 201 shutdown_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 202 {
 203   if (sc != NULL)
 204   {
 205     GNUNET_FS_search_stop (sc);
 206     sc = NULL;
 207   }
 208 }
 209
 210
 211 /**
 212  * Main function that will be run by the scheduler.
 213  *
 214  * @param cls closure
 215  * @param args remaining command-line arguments
 216  * @param cfgfile name of the configuration file used (for saving, can be NULL!)
 217  * @param c configuration
 218  */
 219 static void
 220 run (void *cls, char *const *args, const char *cfgfile,
 221      const struct GNUNET_CONFIGURATION_Handle *c)
 222 {
 223   struct GNUNET_FS_Uri *uri;
 224   unsigned int argc;
 225   enum GNUNET_FS_SearchOptions options;
 226
 227   argc = 0;
 228   while (NULL != args[argc])
 229     argc++;
 230   uri = GNUNET_FS_uri_ksk_create_from_args (argc, (const char **) args);
 231   if (NULL == uri)
 232   {
 233     FPRINTF (stderr, "%s",  _("Could not create keyword URI from arguments.\n"));
 234     ret = 1;
 235     return;
 236   }
 237   cfg = c;
 238   ctx =
 239       GNUNET_FS_start (cfg, "gnunet-search", &progress_cb, NULL,
 240                        GNUNET_FS_FLAGS_NONE, GNUNET_FS_OPTIONS_END);
 241   if (NULL == ctx)
 242   {
 243     FPRINTF (stderr, _("Could not initialize `%s' subsystem.\n"), "FS");
 244     GNUNET_FS_uri_destroy (uri);
 245     ret = 1;
 246     return;
 247   }
 248   if (output_filename != NULL)
 249     db = GNUNET_FS_directory_builder_create (NULL);
 250   options = GNUNET_FS_SEARCH_OPTION_NONE;
 251   if (local_only)
 252     options |= GNUNET_FS_SEARCH_OPTION_LOOPBACK_ONLY;
 253   sc = GNUNET_FS_search_start (ctx, uri, anonymity, options, NULL);
 254   GNUNET_FS_uri_destroy (uri);
 255   if (NULL == sc)
 256   {
 257     FPRINTF (stderr, "%s",  _("Could not start searching.\n"));
 258     GNUNET_FS_stop (ctx);
 259     ret = 1;
 260     return;
 261   }
 262   if (0 != timeout.rel_value_us)
 263     GNUNET_SCHEDULER_add_delayed (timeout, &shutdown_task, NULL);
 264   else
 265     GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_FOREVER_REL, &shutdown_task,
 266                                   NULL);
 267 }
 268
 269
 270 /**
 271  * The main function to search GNUnet.
 272  *
 273  * @param argc number of arguments from the command line
 274  * @param argv command line arguments
 275  * @return 0 ok, 1 on error
 276  */
 277 int
 278 main (int argc, char *const *argv)
 279 {
 280   static const struct GNUNET_GETOPT_CommandLineOption options[] = {
 281     {'a', "anonymity", "LEVEL",
 282      gettext_noop ("set the desired LEVEL of receiver-anonymity"),
 283      1, &GNUNET_GETOPT_set_uint, &anonymity},
 284     {'n', "no-network", NULL,
 285      gettext_noop ("only search the local peer (no P2P network search)"),
 286      0, &GNUNET_GETOPT_set_one, &local_only},
 287     {'o', "output", "PREFIX",
 288      gettext_noop ("write search results to file starting with PREFIX"),
 289      1, &GNUNET_GETOPT_set_string, &output_filename},
 290     {'t', "timeout", "DELAY",
 291      gettext_noop ("automatically terminate search after DELAY"),
 292      1, &GNUNET_GETOPT_set_relative_time, &timeout},
 293     {'V', "verbose", NULL,
 294      gettext_noop ("be verbose (print progress information)"),
 295      0, &GNUNET_GETOPT_set_one, &verbose},
 296     {'N', "results", "VALUE",
 297      gettext_noop
 298      ("automatically terminate search after VALUE results are found"),
 299      1, &GNUNET_GETOPT_set_uint, &results_limit},
 300     GNUNET_GETOPT_OPTION_END
 301   };
 302
 303   if (GNUNET_OK != GNUNET_STRINGS_get_utf8_args (argc, argv, &argc, &argv))
 304     return 2;
 305
 306   ret = (GNUNET_OK ==
 307          GNUNET_PROGRAM_run (argc, argv, "gnunet-search [OPTIONS] KEYWORD",
 308                              gettext_noop
 309                              ("Search GNUnet for files that were published on GNUnet"),
 310                              options, &run, NULL)) ? ret : 1;
 311   GNUNET_free ((void*) argv);
 312   return ret;
 313 }
 314
 315 /* end of gnunet-search.c */