3 files changed, 142 insertions, 12 deletions

diff --git a/src/service/namestore/gnunet-service-namestore.c b/src/service/namestore/gnunet-service-namestore.c
index 230583936..f375828dc 100644
--- a/src/service/namestore/gnunet-service-namestore.c
+++ b/src/service/namestore/gnunet-service-namestore.c
@@ -1275,7 +1275,7 @@ check_edit_record_set (void *cls, const struct EditRecordSetMessage *er_msg)
   key_len = ntohs (er_msg->key_len);
   src_size = ntohs (er_msg->gns_header.header.size);
   if (name_len + editor_hint_len + key_len != src_size - sizeof(struct
-                                                                LabelLookupMessage))
+                                                                EditRecordSetMessage))
   {
     GNUNET_break (0);
     return GNUNET_SYSERR;
@@ -1285,10 +1285,10 @@ check_edit_record_set (void *cls, const struct EditRecordSetMessage *er_msg)
 
 
 /**
- * Handles a #GNUNET_MESSAGE_TYPE_NAMESTORE_RECORD_LOOKUP message
+ * Handles a #GNUNET_MESSAGE_TYPE_NAMESTORE_RECORD_SET_EDIT message
  *
  * @param cls client sending the message
- * @param ll_msg message of type `struct LabelLookupMessage`
+ * @param ll_msg message of type `struct EditRecordSetMessage`
  */
 static void
 handle_edit_record_set (void *cls, const struct EditRecordSetMessage *er_msg)
@@ -1381,6 +1381,119 @@ handle_edit_record_set (void *cls, const struct EditRecordSetMessage *er_msg)
 
 
 /**
+ * Handles a #GNUNET_MESSAGE_TYPE_NAMESTORE_RECORD_SET_EDIT_CANCEL message
+ *
+ * @param cls client sending the message
+ * @param er_msg message of type `struct EditRecordSetMessage`
+ * @return #GNUNET_OK if @a er_msg is well-formed
+ */
+static int
+check_edit_record_set_cancel (void *cls, const struct
+                              EditRecordSetCancelMessage *er_msg)
+{
+  uint16_t name_len;
+  uint16_t editor_hint_len;
+  uint16_t editor_hint_repl_len;
+  size_t src_size;
+  size_t key_len;
+
+  (void) cls;
+  name_len = ntohs (er_msg->label_len);
+  editor_hint_len = ntohs (er_msg->editor_hint_len);
+  editor_hint_repl_len = ntohs (er_msg->editor_hint_len);
+  key_len = ntohs (er_msg->key_len);
+  src_size = ntohs (er_msg->gns_header.header.size);
+  if (name_len + editor_hint_len + editor_hint_repl_len + key_len != src_size
+      - sizeof(struct
+               EditRecordSetCancelMessage))
+  {
+    GNUNET_break (0);
+    return GNUNET_SYSERR;
+  }
+  return GNUNET_OK;
+}
+
+
+/**
+ * Handles a #GNUNET_MESSAGE_TYPE_NAMESTORE_RECORD_SET_EDIT_CANCEL message
+ *
+ * @param cls client sending the message
+ * @param ll_msg message of type `struct EditRecordSetCancelMessage`
+ */
+static void
+handle_edit_record_set_cancel (void *cls, const struct
+                               EditRecordSetCancelMessage *er_msg)
+{
+  struct GNUNET_CRYPTO_PrivateKey zone;
+  struct NamestoreClient *nc = cls;
+  struct GNUNET_MQ_Envelope *env;
+  struct NamestoreResponseMessage *rer_msg;
+  const char *name_tmp;
+  const char *editor_hint;
+  const char *editor_hint_repl;
+  char *conv_name;
+  uint16_t name_len;
+  uint16_t editor_hint_len;
+  int res;
+  size_t key_len;
+  size_t kb_read;
+
+  key_len = ntohs (er_msg->key_len);
+  name_len = ntohs (er_msg->label_len);
+  editor_hint_len = ntohs (er_msg->editor_hint_len);
+  if ((GNUNET_SYSERR ==
+       GNUNET_CRYPTO_read_private_key_from_buffer (&er_msg[1],
+                                                   key_len,
+                                                   &zone,
+                                                   &kb_read)) ||
+      (kb_read != key_len))
+  {
+    GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
+                "Error reading private key\n");
+    GNUNET_SERVICE_client_drop (nc->client);
+    return;
+  }
+  name_tmp = (const char *) &er_msg[1] + key_len;
+  editor_hint = (const char *) name_tmp + name_len;
+  editor_hint_repl = (const char *) name_tmp + name_len + editor_hint_len;
+  GNUNET_SERVICE_client_continue (nc->client);
+  GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+              "Received NAMESTORE_RECORD_SET_EDIT message for name `%s'\n",
+              name_tmp);
+
+  conv_name = GNUNET_GNSRECORD_string_normalize (name_tmp);
+  if (NULL == conv_name)
+  {
+    GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
+                "Error converting name `%s'\n",
+                name_tmp);
+    GNUNET_SERVICE_client_drop (nc->client);
+    return;
+  }
+  name_len = strlen (conv_name) + 1;
+  GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+              "Clearing editor hint\n");
+  res = nc->GSN_database->clear_editor_hint (nc->GSN_database->cls,
+                                             editor_hint,
+                                             editor_hint_repl,
+                                             &zone,
+                                             conv_name);
+  GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+              "Clearing editor hint %s\n", (GNUNET_SYSERR == res) ? "failed." :
+              "successful.");
+
+  env =
+    GNUNET_MQ_msg (rer_msg,
+                   GNUNET_MESSAGE_TYPE_NAMESTORE_GENERIC_RESPONSE);
+  rer_msg->gns_header.r_id = er_msg->gns_header.r_id;
+  rer_msg->ec = htons ((GNUNET_OK == res) ? GNUNET_EC_NONE :
+                       GNUNET_EC_NAMESTORE_BACKEND_FAILED);
+  GNUNET_MQ_send (nc->mq, env);
+  GNUNET_free (conv_name);
+}
+
+
+/**
  * Handles a #GNUNET_MESSAGE_TYPE_NAMESTORE_RECORD_LOOKUP message
  *
  * @param cls client sending the message
@@ -2674,6 +2787,10 @@ GNUNET_SERVICE_MAIN (
                          GNUNET_MESSAGE_TYPE_NAMESTORE_RECORD_SET_EDIT,
                          struct EditRecordSetMessage,
                          NULL),
+  GNUNET_MQ_hd_var_size (edit_record_set_cancel,
+                         GNUNET_MESSAGE_TYPE_NAMESTORE_RECORD_SET_EDIT_CANCEL,
+                         struct EditRecordSetCancelMessage,
+                         NULL),
   GNUNET_MQ_hd_var_size (record_lookup,
                          GNUNET_MESSAGE_TYPE_NAMESTORE_RECORD_LOOKUP,
                          struct LabelLookupMessage,
diff --git a/src/service/namestore/namestore_api.c b/src/service/namestore/namestore_api.c
index 435c1622d..45a15c39a 100644
--- a/src/service/namestore/namestore_api.c
+++ b/src/service/namestore/namestore_api.c
@@ -375,8 +375,8 @@ handle_generic_response (void *cls,
   qe = find_qe (h, ntohl (msg->gns_header.r_id));
   res = ntohl (msg->ec);
   LOG (GNUNET_ERROR_TYPE_DEBUG,
-       "Received RECORD_STORE_RESPONSE with result %d\n",
-       res);
+       "Received GENERIC_RESPONSE with result %s\n",
+       GNUNET_ErrorCode_get_hint(res));
   if (NULL == qe)
     return;
   if (NULL != qe->cont)
@@ -1617,8 +1617,7 @@ GNUNET_NAMESTORE_record_set_edit_cancel (struct GNUNET_NAMESTORE_Handle *h,
                                          const char *label,
                                          const char *editor_hint,
                                          const char *editor_hint_replacement,
-                                         GNUNET_SCHEDULER_TaskCallback
-                                         finished_cb,
+                                         GNUNET_NAMESTORE_ContinuationWithStatus finished_cb,
                                          void *finished_cls)
 {
   struct GNUNET_NAMESTORE_QueueEntry *qe;
@@ -1641,6 +1640,8 @@ GNUNET_NAMESTORE_record_set_edit_cancel (struct GNUNET_NAMESTORE_Handle *h,
   qe = GNUNET_new (struct GNUNET_NAMESTORE_QueueEntry);
   qe->h = h;
   qe->op_id = get_op_id (h);
+  qe->cont = finished_cb;
+  qe->cont_cls = finished_cls;
   GNUNET_CONTAINER_DLL_insert_tail (h->op_head, h->op_tail, qe);
 
   key_len = GNUNET_CRYPTO_private_key_get_length (pkey);
diff --git a/src/service/namestore/test_namestore_api_edit_records.c b/src/service/namestore/test_namestore_api_edit_records.c
index 9dce13e33..02e821377 100644
--- a/src/service/namestore/test_namestore_api_edit_records.c
+++ b/src/service/namestore/test_namestore_api_edit_records.c
@@ -21,7 +21,9 @@
  * @file namestore/test_namestore_api_edit_records.c
  * @brief testcase for namestore_api.c: Multiple clients work with record set.
  */
+#include "gnunet_error_codes.h"
 #include "gnunet_namestore_service.h"
+#include "gnunet_scheduler_lib.h"
 #include "gnunet_testing_lib.h"
 
 #define TEST_RECORD_TYPE GNUNET_DNSPARSER_TYPE_TXT
@@ -91,6 +93,15 @@ end (void *cls)
     GNUNET_SCHEDULER_cancel (endbadly_task);
   cleanup ();
   res = 0;
+  GNUNET_SCHEDULER_shutdown ();
+}
+
+
+static void
+cancel_done (void *cls, enum GNUNET_ErrorCode ec)
+{
+  GNUNET_assert (GNUNET_EC_NONE == ec);
+  GNUNET_SCHEDULER_add_now (&end, NULL);
 }
 
 
@@ -102,11 +113,12 @@ begin_cont_b (void *cls,
               GNUNET_GNSRECORD_Data *rd,
               const char *editor_hint)
 {
-  const char *name = cls;
+  char *name = cls;
 
   GNUNET_assert (GNUNET_EC_NONE == ec);
   GNUNET_assert (0 != strcmp (editor_hint, "B"));
-  GNUNET_SCHEDULER_add_now (&end, NULL);
+  nsqe = GNUNET_NAMESTORE_record_set_edit_cancel (nsh2, &privkey, name, "A",
+                                                  "B", &cancel_done, name);
 }
 
 
@@ -118,7 +130,7 @@ begin_cont (void *cls,
             GNUNET_GNSRECORD_Data *rd,
             const char *editor_hint)
 {
-  const char *name = cls;
+  char *name = cls;
 
   GNUNET_assert (GNUNET_EC_NONE == ec);
   GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
@@ -135,7 +147,7 @@ static void
 preload_cont (void *cls,
               enum GNUNET_ErrorCode ec)
 {
-  const char *name = cls;
+  char *name = cls;
 
   GNUNET_assert (NULL != cls);
   nsqe = NULL;
@@ -168,7 +180,7 @@ run (void *cls,
      struct GNUNET_TESTING_Peer *peer)
 {
   struct GNUNET_GNSRECORD_Data rd;
-  const char *name = "dummy";
+  char *name = "dummy";
 
   endbadly_task = GNUNET_SCHEDULER_add_delayed (TIMEOUT,
                                                 &endbadly,