flush peer respect value on disconnect
[oweals/gnunet.git] / src / fs / gnunet-auto-share.c
index ac38c6d94f808adb1ddaa80565a7c7e61405858f..32725f66f00d73e925f066cf01029b37b4c150c4 100644 (file)
@@ -21,7 +21,7 @@
  * @file fs/gnunet-auto-share.c
  * @brief automatically publish files on GNUnet
  * @author Christian Grothoff
- * 
+ *
  * TODO:
  * - support loading meta data / keywords from resource file
  * - add stability timer (a la buildbot)
 #include "platform.h"
 #include "gnunet_util_lib.h"
 
+#define MIN_FREQUENCY GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_HOURS, 4)
+
+#define MAX_FREQUENCY GNUNET_TIME_UNIT_MINUTES
+
 
 /**
  * Item in our work queue (or in the set of files/directories
@@ -200,30 +204,35 @@ load_state ()
   while (n-- > 0)
   {
     if ( (GNUNET_OK !=
-         GNUNET_BIO_read_string (rh, "filename", &fn, 1024)) || 
+         GNUNET_BIO_read_string (rh, "filename", &fn, 1024)) ||
         (GNUNET_OK !=
          GNUNET_BIO_read (rh, "id", &id, sizeof (struct GNUNET_HashCode))) )
       goto error;
-    wi = GNUNET_malloc (sizeof (struct WorkItem));
+    wi = GNUNET_new (struct WorkItem);
     wi->id = id;
     wi->filename = fn;
+    GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+               "Loaded serialization ID for `%s' is `%s'\n",
+               wi->filename,
+               GNUNET_h2s (&id));
     fn = NULL;
     GNUNET_CRYPTO_hash (wi->filename,
                        strlen (wi->filename),
                        &id);
-    GNUNET_CONTAINER_multihashmap_put (work_finished,
-                                      &id,
-                                      wi,
-                                      GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_ONLY);
+    GNUNET_break (GNUNET_OK ==
+                  GNUNET_CONTAINER_multihashmap_put (work_finished,
+                                                     &id,
+                                                     wi,
+                                                     GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_ONLY));
   }
-  if (GNUNET_OK == 
+  if (GNUNET_OK ==
       GNUNET_BIO_read_close (rh, &emsg))
     return;
   rh = NULL;
  error:
   GNUNET_free_non_null (fn);
   if (NULL != rh)
-    GNUNET_BIO_read_close (rh, &emsg);
+    (void) GNUNET_BIO_read_close (rh, &emsg);
   GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
              _("Failed to load state: %s\n"),
              emsg);
@@ -247,7 +256,11 @@ write_item (void *cls,
   struct GNUNET_BIO_WriteHandle *wh = cls;
   struct WorkItem *wi = value;
 
-  if ( (GNUNET_OK != 
+  GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+             "Saving serialization ID of file `%s' with value `%s'\n",
+             wi->filename,
+             GNUNET_h2s (&wi->id));
+  if ( (GNUNET_OK !=
        GNUNET_BIO_write_string (wh, wi->filename)) ||
        (GNUNET_OK !=
        GNUNET_BIO_write (wh,
@@ -271,6 +284,14 @@ save_state ()
   n = GNUNET_CONTAINER_multihashmap_size (work_finished);
   fn = get_state_file ();
   wh = GNUNET_BIO_write_open (fn);
+  if (NULL == wh)
+  {
+    GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
+               _("Failed to save state to file %s\n"),
+               fn);
+    GNUNET_free (fn);
+    return;
+  }
   if (GNUNET_OK !=
       GNUNET_BIO_write_int32 (wh, n))
   {
@@ -335,21 +356,68 @@ maint_child_death (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 {
   struct WorkItem *wi = cls;
   struct GNUNET_HashCode key;
+  enum GNUNET_OS_ProcessStatusType type;
+  unsigned long code;
+  int ret;
+  char c;
+  const struct GNUNET_DISK_FileHandle *pr;
+
 
   run_task = GNUNET_SCHEDULER_NO_TASK;
-  GNUNET_break (GNUNET_OK ==
-               GNUNET_OS_process_wait (publish_proc));
+  pr = GNUNET_DISK_pipe_handle (sigpipe, GNUNET_DISK_PIPE_END_READ);
+  if (0 == (tc->reason & GNUNET_SCHEDULER_REASON_READ_READY))
+  {
+    /* shutdown scheduled us, ignore! */
+    run_task =
+      GNUNET_SCHEDULER_add_read_file (GNUNET_TIME_UNIT_FOREVER_REL,
+                                     pr, &maint_child_death, wi);
+    return;
+  }
+
+  ret = GNUNET_OS_process_status (publish_proc,
+                                 &type,
+                                 &code);
+  GNUNET_assert (GNUNET_SYSERR != ret);
+  if (GNUNET_NO == ret)
+  {
+    GNUNET_break (0);
+    GNUNET_OS_process_kill (publish_proc, SIGKILL);
+    type = GNUNET_OS_PROCESS_SIGNALED;
+  }
   GNUNET_OS_process_destroy (publish_proc);
   publish_proc = NULL;
-  GNUNET_CRYPTO_hash (wi->filename,
-                     strlen (wi->filename),
-                     &key);
-  GNUNET_CONTAINER_multihashmap_put (work_finished,
-                                    &key,
-                                    wi,
-                                    GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_ONLY);
+  /* consume the signal */
+  GNUNET_break (0 < GNUNET_DISK_file_read (pr, &c, sizeof (c)));
+
+  if (GNUNET_YES == do_shutdown)
+  {
+    GNUNET_free (wi->filename);
+    GNUNET_free (wi);
+    return;
+  }
+  if ( (GNUNET_OS_PROCESS_EXITED == type) &&
+       (0 == code) )
+  {
+    GNUNET_log (GNUNET_ERROR_TYPE_INFO,
+               _("Publication of `%s' done\n"),
+               wi->filename);
+    GNUNET_CRYPTO_hash (wi->filename,
+                       strlen (wi->filename),
+                       &key);
+    GNUNET_break (GNUNET_OK ==
+                  GNUNET_CONTAINER_multihashmap_put (work_finished,
+                                                     &key,
+                                                     wi,
+                                                     GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_ONLY));
+  }
+  else
+  {
+    GNUNET_CONTAINER_DLL_insert_tail (work_head,
+                                     work_tail,
+                                     wi);
+  }
   save_state ();
-  schedule_next_task ();    
+  schedule_next_task ();
 }
 
 
@@ -386,7 +454,7 @@ work (void *cls,
   static char content_prio[20];
   static char repl_level[20];
   struct WorkItem *wi;
-  const struct GNUNET_DISK_FileHandle *pr;  
+  const struct GNUNET_DISK_FileHandle *pr;
   int argc;
 
   run_task = GNUNET_SCHEDULER_NO_TASK;
@@ -418,8 +486,11 @@ work (void *cls,
   argv[argc++] = repl_level;
   argv[argc++] = wi->filename;
   argv[argc] = NULL;
+  GNUNET_log (GNUNET_ERROR_TYPE_INFO,
+             _("Publishing `%s'\n"),
+             wi->filename);
   publish_proc = GNUNET_OS_start_process_vap (GNUNET_YES,
-                                             NULL, NULL,
+                                              0, NULL, NULL, NULL,
                                              "gnunet-publish",
                                              argv);
   if (NULL == publish_proc)
@@ -459,9 +530,6 @@ determine_id (void *cls,
   struct GNUNET_HashCode fx[2];
   struct GNUNET_HashCode ft;
 
-  if (NULL != strstr (filename,
-                     DIR_SEPARATOR_STR ".auto-share"))
-    return GNUNET_OK; /* skip internal file */
   if (0 != STAT (filename, &sbuf))
   {
     GNUNET_log_strerror_file (GNUNET_ERROR_TYPE_WARNING, "stat", filename);
@@ -470,9 +538,12 @@ determine_id (void *cls,
   GNUNET_CRYPTO_hash (filename, strlen (filename), &fx[0]);
   if (!S_ISDIR (sbuf.st_mode))
   {
-    uint64_t fsize = GNUNET_htonll (sbuf.st_size);
+    uint64_t fattr[2];
+
+    fattr[0] = GNUNET_htonll (sbuf.st_size);
+    fattr[0] = GNUNET_htonll (sbuf.st_mtime);
 
-    GNUNET_CRYPTO_hash (&fsize, sizeof (uint64_t), &fx[1]);
+    GNUNET_CRYPTO_hash (fattr, sizeof (fattr), &fx[1]);
   }
   else
   {
@@ -484,7 +555,7 @@ determine_id (void *cls,
   /* use hash here to make hierarchical structure distinct from
      all files on the same level */
   GNUNET_CRYPTO_hash (fx, sizeof (fx), &ft);
-  /* use XOR here so that order of the files in the directory 
+  /* use XOR here so that order of the files in the directory
      does not matter! */
   GNUNET_CRYPTO_hash_xor (&ft, id, id);
   return GNUNET_OK;
@@ -510,6 +581,11 @@ add_file (void *cls,
 
   if (GNUNET_YES == do_shutdown)
     return GNUNET_SYSERR;
+  if ( (NULL != strstr (filename,
+                     "/.auto-share")) ||
+       (NULL != strstr (filename,
+                       "\\.auto-share")) )
+    return GNUNET_OK; /* skip internal file */
   GNUNET_CRYPTO_hash (filename,
                      strlen (filename),
                      &key);
@@ -524,21 +600,22 @@ add_file (void *cls,
                     sizeof (struct GNUNET_HashCode)))
       return GNUNET_OK; /* skip: we did this one already */
     /* contents changed, need to re-do the directory... */
-    GNUNET_CONTAINER_multihashmap_remove (work_finished,
-                                         &key,
-                                         wi);
-    wi->id = id; 
+    GNUNET_assert (GNUNET_YES ==
+                  GNUNET_CONTAINER_multihashmap_remove (work_finished,
+                                                        &key,
+                                                        wi));
   }
   else
   {
-    wi = GNUNET_malloc (sizeof (struct WorkItem));
+    wi = GNUNET_new (struct WorkItem);
     wi->filename = GNUNET_strdup (filename);
   }
+  wi->id = id;
   GNUNET_CONTAINER_DLL_insert (work_head,
                               work_tail,
                               wi);
   if (GNUNET_YES == do_shutdown)
-    return GNUNET_SYSERR; 
+    return GNUNET_SYSERR;
   return GNUNET_OK;
 }
 
@@ -570,18 +647,17 @@ schedule_next_task ()
   struct GNUNET_TIME_Relative delay;
 
   if (GNUNET_YES == do_shutdown)
-    return;  
+    return;
   if (NULL == work_head)
   {
     /* delay by at most 4h, at least 1s, and otherwise in between depending
        on how long it took to scan */
     delay = GNUNET_TIME_absolute_get_duration (start_time);
-    delay = GNUNET_TIME_relative_min (GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_HOURS,
-                                                                    4),
+    delay = GNUNET_TIME_relative_min (MIN_FREQUENCY,
                                      GNUNET_TIME_relative_multiply (delay,
                                                                     100));
     delay = GNUNET_TIME_relative_max (delay,
-                                     GNUNET_TIME_UNIT_MINUTES);
+                                     MAX_FREQUENCY);
     run_task = GNUNET_SCHEDULER_add_delayed (delay,
                                             &scan,
                                             NULL);
@@ -607,7 +683,7 @@ run (void *cls, char *const *args, const char *cfgfile,
 {
   /* check arguments */
   if ((args[0] == NULL) || (args[1] != NULL) ||
-      (GNUNET_YES != GNUNET_DISK_directory_test (args[0])))
+      (GNUNET_YES != GNUNET_DISK_directory_test (args[0], GNUNET_YES)))
   {
     printf (_("You must specify one and only one directory name for automatic publication.\n"));
     ret = -1;
@@ -616,11 +692,11 @@ run (void *cls, char *const *args, const char *cfgfile,
   cfg_filename = GNUNET_strdup (cfgfile);
   cfg = c;
   dir_name = args[0];
-  work_finished = GNUNET_CONTAINER_multihashmap_create (1024);
+  work_finished = GNUNET_CONTAINER_multihashmap_create (1024, GNUNET_NO);
   load_state ();
   run_task = GNUNET_SCHEDULER_add_with_priority (GNUNET_SCHEDULER_PRIORITY_IDLE,
                                                 &scan, NULL);
-  
+
   kill_task =
       GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_FOREVER_REL, &do_stop_task,
                                     NULL);
@@ -633,7 +709,7 @@ run (void *cls, char *const *args, const char *cfgfile,
  * @param cls NULL (unused)
  * @param key key of the item in the map (unused)
  * @param value the 'struct WorkItem' to free
- * @return GNUNET_OK to continue to iterate 
+ * @return GNUNET_OK to continue to iterate
  */
 static int
 free_item (void *cls,
@@ -647,6 +723,7 @@ free_item (void *cls,
   return GNUNET_OK;
 }
 
+
 /**
  * The main function to automatically publish content to GNUnet.
  *
@@ -656,7 +733,7 @@ free_item (void *cls,
  */
 int
 main (int argc, char *const *argv)
-{  
+{
   static const struct GNUNET_GETOPT_CommandLineOption options[] = {
     {'a', "anonymity", "LEVEL",
      gettext_noop ("set the desired LEVEL of sender-anonymity"),
@@ -694,10 +771,13 @@ main (int argc, char *const *argv)
                            gettext_noop
                            ("Automatically publish files from a directory on GNUnet"),
                            options, &run, NULL)) ? ret : 1;
-  (void) GNUNET_CONTAINER_multihashmap_iterate (work_finished,
-                                               &free_item,
-                                               NULL);
-  GNUNET_CONTAINER_multihashmap_destroy (work_finished);
+  if (NULL != work_finished)
+  {
+    (void) GNUNET_CONTAINER_multihashmap_iterate (work_finished,
+                                                 &free_item,
+                                                 NULL);
+    GNUNET_CONTAINER_multihashmap_destroy (work_finished);
+  }
   while (NULL != (wi = work_head))
   {
     GNUNET_CONTAINER_DLL_remove (work_head, work_tail, wi);
@@ -710,6 +790,7 @@ main (int argc, char *const *argv)
   sigpipe = NULL;
   GNUNET_free (cfg_filename);
   cfg_filename = NULL;
+  GNUNET_free ((void*) argv);
   return ok;
 }