1 files changed, 224 insertions, 0 deletions

diff --git a/src/messenger/gnunet-service-messenger_operation_store.c b/src/messenger/gnunet-service-messenger_operation_store.c
new file mode 100644
index 000000000..05985ef84
--- /dev/null
+++ b/src/messenger/gnunet-service-messenger_operation_store.c
@@ -0,0 +1,224 @@
+/*
+   This file is part of GNUnet.
+   Copyright (C) 2021 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
+ */
+/**
+ * @author Tobias Frisch
+ * @file src/messenger/gnunet-service-messenger_operation_store.c
+ * @brief GNUnet MESSENGER service
+ */
+
+#include "gnunet-service-messenger_operation_store.h"
+
+#include "gnunet-service-messenger_operation.h"
+#include "gnunet-service-messenger_room.h"
+
+void
+init_operation_store (struct GNUNET_MESSENGER_OperationStore *store, struct GNUNET_MESSENGER_SrvRoom *room)
+{
+  GNUNET_assert((store) && (room));
+
+  store->room = room;
+  store->operations = GNUNET_CONTAINER_multihashmap_create(8, GNUNET_NO);
+}
+
+static int
+iterate_destroy_operations (void *cls, const struct GNUNET_HashCode *key, void *value)
+{
+  struct GNUNET_MESSENGER_Operation *op = value;
+
+  destroy_operation(op);
+
+  return GNUNET_YES;
+}
+
+void
+clear_operation_store (struct GNUNET_MESSENGER_OperationStore *store)
+{
+  GNUNET_assert(store);
+
+  GNUNET_CONTAINER_multihashmap_iterate (store->operations, iterate_destroy_operations, NULL);
+  GNUNET_CONTAINER_multihashmap_destroy(store->operations);
+}
+
+static int
+callback_scan_for_operations (void *cls, const char *filename)
+{
+  struct GNUNET_MESSENGER_OperationStore *store = cls;
+
+  if (GNUNET_YES == GNUNET_DISK_file_test (filename))
+  {
+    char *path;
+
+    GNUNET_asprintf (&path, "%s%c", filename, DIR_SEPARATOR);
+
+    struct GNUNET_MESSENGER_Operation *op = load_operation(store, path);
+
+    if ((op) && (GNUNET_OK != GNUNET_CONTAINER_multihashmap_put(
+        store->operations,
+        &(op->hash), op,
+        GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_FAST)))
+    {
+      destroy_operation(op);
+    }
+
+    GNUNET_free(path);
+  }
+
+  return GNUNET_OK;
+}
+
+void
+load_operation_store (struct GNUNET_MESSENGER_OperationStore *store,
+                      const char *directory)
+{
+  GNUNET_assert ((store) && (directory));
+
+  if (GNUNET_OK == GNUNET_DISK_directory_test (directory, GNUNET_YES))
+    GNUNET_DISK_directory_scan (directory, callback_scan_for_operations, store);
+}
+
+static int
+iterate_save_operations (void *cls, const struct GNUNET_HashCode *key, void *value)
+{
+  const char *save_dir = cls;
+
+  struct GNUNET_MESSENGER_Operation *op = value;
+
+  if (!op)
+    return GNUNET_YES;
+
+  char *op_dir;
+  GNUNET_asprintf (&op_dir, "%s%s.cfg", save_dir, GNUNET_h2s(key));
+  save_operation(op, op_dir);
+
+  GNUNET_free(op_dir);
+  return GNUNET_YES;
+}
+
+void
+save_operation_store (const struct GNUNET_MESSENGER_OperationStore *store,
+                      const char *directory)
+{
+  GNUNET_assert ((store) && (directory));
+
+  char* save_dir;
+  GNUNET_asprintf (&save_dir, "%s%s%c", directory, "operations", DIR_SEPARATOR);
+
+  if ((GNUNET_YES == GNUNET_DISK_directory_test (save_dir, GNUNET_NO)) ||
+      (GNUNET_OK == GNUNET_DISK_directory_create (save_dir)))
+    GNUNET_CONTAINER_multihashmap_iterate (store->operations, iterate_save_operations, save_dir);
+
+  GNUNET_free(save_dir);
+}
+
+enum GNUNET_MESSENGER_OperationType
+get_store_operation_type (const struct GNUNET_MESSENGER_OperationStore *store,
+                          const struct GNUNET_HashCode *hash)
+{
+  GNUNET_assert((store) && (hash));
+
+  struct GNUNET_MESSENGER_Operation *op = GNUNET_CONTAINER_multihashmap_get(store->operations, hash);
+
+  if (!op)
+    return GNUNET_MESSENGER_OP_UNKNOWN;
+
+  return op->type;
+}
+
+int
+use_store_operation (struct GNUNET_MESSENGER_OperationStore *store,
+                     const struct GNUNET_HashCode *hash,
+                     enum GNUNET_MESSENGER_OperationType type,
+                     struct GNUNET_TIME_Relative delay)
+{
+  GNUNET_assert((store) && (hash));
+
+  struct GNUNET_MESSENGER_Operation *op = GNUNET_CONTAINER_multihashmap_get(store->operations, hash);
+
+  if (op)
+    goto use_op;
+
+  op = create_operation(hash);
+
+  if (GNUNET_OK != GNUNET_CONTAINER_multihashmap_put(store->operations, hash, op, GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_FAST))
+  {
+    destroy_operation(op);
+
+    return GNUNET_SYSERR;
+  }
+
+use_op:
+  if ((op->type != GNUNET_MESSENGER_OP_UNKNOWN) &&
+      (type == GNUNET_MESSENGER_OP_DELETE))
+    stop_operation (op);
+
+  return start_operation(op, type, store, delay);
+}
+
+void
+cancel_store_operation (struct GNUNET_MESSENGER_OperationStore *store,
+                        const struct GNUNET_HashCode *hash)
+{
+  GNUNET_assert((store) && (hash));
+
+  struct GNUNET_MESSENGER_Operation *op = GNUNET_CONTAINER_multihashmap_get(store->operations, hash);
+
+  if (!op)
+    return;
+
+  stop_operation(op);
+
+  GNUNET_CONTAINER_multihashmap_remove(store->operations, hash, op);
+
+  destroy_operation(op);
+}
+
+extern void
+callback_room_deletion (struct GNUNET_MESSENGER_SrvRoom *room, const struct GNUNET_HashCode *hash);
+
+extern void
+callback_room_merge (struct GNUNET_MESSENGER_SrvRoom *room, const struct GNUNET_HashCode *hash);
+
+void
+callback_store_operation (struct GNUNET_MESSENGER_OperationStore *store,
+                          enum GNUNET_MESSENGER_OperationType type,
+                          const struct GNUNET_HashCode *hash)
+{
+  GNUNET_assert((store) && (hash));
+
+  struct GNUNET_HashCode op_hash;
+  GNUNET_memcpy(&op_hash, hash, sizeof(op_hash));
+  cancel_store_operation (store, &op_hash);
+
+  struct GNUNET_MESSENGER_SrvRoom *room = store->room;
+
+  switch (type)
+  {
+  case GNUNET_MESSENGER_OP_REQUEST:
+    break;
+  case GNUNET_MESSENGER_OP_DELETE:
+    callback_room_deletion (room, &op_hash);
+    break;
+  case GNUNET_MESSENGER_OP_MERGE:
+    callback_room_merge (room, &op_hash);
+    break;
+  default:
+    break;
+  }
+}