Merge branch 'credentials' of git+ssh://gnunet.org/gnunet into credentials
[oweals/gnunet.git] / src / my / my.c
index 1ab4a64dbd09a49e336693b2bfc9e8af22f5a73a..459f09b6a4b65227c88ad532db1d4ea18c054357 100644 (file)
@@ -20,6 +20,7 @@
 /**
  * @file my/my.c
  * @brief library to help with access to a MySQL database
+ * @author Christophe Genevey
  * @author Christian Grothoff
  */
 #include "platform.h"
 #include "gnunet_my_lib.h"
 
 
-
 /**
  * Run a prepared SELECT statement.
  *
  * @param mc mysql context
  * @param sh handle to SELECT statment
  * @param params parameters to the statement
- * @return mysql result
+ * @return
+      #GNUNET_YES if we can prepare all statement
+      #GNUNET_SYSERR if we can't prepare all statement
  */
-
- /***** FIXE THIS FUNCTION *****/
 int
 GNUNET_MY_exec_prepared (struct GNUNET_MYSQL_Context *mc,
                          struct GNUNET_MYSQL_StatementHandle *sh,
-                         const struct GNUNET_MY_QueryParam *params)
+                         struct GNUNET_MY_QueryParam *params)
 {
   const struct GNUNET_MY_QueryParam *p;
   unsigned int num;
@@ -55,7 +55,7 @@ GNUNET_MY_exec_prepared (struct GNUNET_MYSQL_Context *mc,
     MYSQL_BIND qbind[num];
     unsigned int off;
 
-    memset(qbind, 0, sizeof(qbind));
+    memset (qbind, 0, sizeof(qbind));
     off = 0;
     for (i=0;NULL != (p = &params[i])->conv;i++)
     {
@@ -64,87 +64,207 @@ GNUNET_MY_exec_prepared (struct GNUNET_MYSQL_Context *mc,
                    p,
                    &qbind[off]))
       {
+        GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
+                    "Conversion for MySQL query failed at offset %u\n",
+                    i);
         return GNUNET_SYSERR;
       }
       off += p->num_params;
     }
-    stmt = GNUNET_MYSQL_statement_get_stmt (mc, sh);
+    stmt = GNUNET_MYSQL_statement_get_stmt (sh);
     if (mysql_stmt_bind_param (stmt,
                                qbind))
     {
-      GNUNET_log_from (GNUNET_ERROR_TYPE_ERROR, "mysql",
+      GNUNET_log_from (GNUNET_ERROR_TYPE_ERROR,
+                       "my",
                        _("`%s' failed at %s:%d with error: %s\n"),
-                       "mysql_stmt_bind_param", __FILE__, __LINE__,
+                       "mysql_stmt_bind_param",
+                       __FILE__, __LINE__,
                        mysql_stmt_error (stmt));
       GNUNET_MYSQL_statements_invalidate (mc);
       return GNUNET_SYSERR;
     }
-  }
-  if (mysql_stmt_execute (stmt))
-  {
-    GNUNET_log_from (GNUNET_ERROR_TYPE_ERROR, "mysql",
-                     _("`%s' failed at %s:%d with error: %s\n"),
-                     "mysql_stmt_execute", __FILE__, __LINE__,
-                     mysql_stmt_error (stmt));
-    GNUNET_MYSQL_statements_invalidate (mc);
-    return GNUNET_SYSERR;
+
+    if (mysql_stmt_execute (stmt))
+    {
+      GNUNET_log_from (GNUNET_ERROR_TYPE_ERROR,
+                       "my",
+                       _("`%s' failed at %s:%d with error: %s\n"),
+                       "mysql_stmt_execute", __FILE__, __LINE__,
+                       mysql_stmt_error (stmt));
+      GNUNET_MYSQL_statements_invalidate (mc);
+      return GNUNET_SYSERR;
+    }
+    GNUNET_MY_cleanup_query (params,
+                             qbind);
   }
   return GNUNET_OK;
 }
 
 
 /**
- * Extract results from a query result according
- * to the given specification. If colums are NULL,
- * the destination is not modified, and #GNUNET_NO is returned4
+ * Free all memory that was allocated in @a qp during
+ * #GNUNET_MY_exec_prepared().
  *
+ * @param qp query specification to clean up
+ * @param qbind array of parameter to clean up
+ */
+void
+GNUNET_MY_cleanup_query (struct GNUNET_MY_QueryParam *qp,
+                         MYSQL_BIND *qbind)
+{
+  unsigned int i;
+
+  for (i=0; NULL != qp[i].conv ;i++)
+    if (NULL != qp[i].cleaner)
+      qp[i].cleaner (qp[i].conv_cls,
+                     &qbind[i]);
+}
+
+
+/**
+ * Extract results from a query result according to the given
+ * specification.  Always fetches the next row.
  *
- * @param result
- * @param row, the row from the result to extract
- * @param result specificatio to extract for
+ * @param sh statement that returned results
+ * @param rs specification to extract for
  * @return
   #GNUNET_YES if all results could be extracted
-    #GNUNET_NO if at least one result was NULL
   #GNUNET_SYSERR if a result was invalid
-*/
*  #GNUNET_YES if all results could be extracted
+ *  #GNUNET_NO if there is no more data in the result set
*  #GNUNET_SYSERR if a result was invalid
+ */
 int
 GNUNET_MY_extract_result (struct GNUNET_MYSQL_StatementHandle *sh,
-                          struct GNUNET_MY_QueryParam *qp,
-                          struct GNUNET_MY_ResultSpec *rs,
-                          int row)
+                          struct GNUNET_MY_ResultSpec *rs)
 {
-  MYSQL_BIND * result;
+  unsigned int num_fields;
   unsigned int i;
-  int had_null = GNUNET_NO;
   int ret;
   MYSQL_STMT *stmt;
 
-  stmt = GNUNET_MYSQL_statement_get_stmt (NULL /* FIXME */, sh);
-  // result = mysql_get_result (stmt);
-  result = NULL;
-  for (i = 0 ; NULL != rs[i].conv ; i++)
+  stmt = GNUNET_MYSQL_statement_get_stmt (sh);
+  if (NULL == stmt)
+  {
+    GNUNET_break (0);
+    return GNUNET_SYSERR;
+  }
+  if (NULL == rs)
   {
-    struct GNUNET_MY_ResultSpec *spec;
+    mysql_stmt_free_result (stmt);
+    return GNUNET_NO;
+  }
 
-    spec = &rs[i];
-    ret = spec->conv (spec->conv_cls,
-                      qp,
-                      result);
+  num_fields = 0;
+  for (i=0;NULL != rs[i].pre_conv;i++)
+    num_fields += rs[i].num_fields;
+
+  if (mysql_stmt_field_count (stmt) != num_fields)
+  {
+    GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
+                "Number of fields missmatch between SQL result and result specification\n");
+    return GNUNET_SYSERR;
+  }
 
-    if (GNUNET_SYSERR == ret)
+  {
+    MYSQL_BIND result[num_fields];
+    unsigned int field_off;
+
+    memset (result, 0, sizeof (MYSQL_BIND) * num_fields);
+    field_off = 0;
+    for (i=0;NULL != rs[i].pre_conv;i++)
+    {
+      struct GNUNET_MY_ResultSpec *rp = &rs[i];
+
+      if (GNUNET_OK !=
+          rp->pre_conv (rp->conv_cls,
+                        rp,
+                        stmt,
+                        field_off,
+                        &result[field_off]))
+
+      {
+        GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
+                    "Pre-conversion for MySQL result failed at offset %u\n",
+                    i);
+        return GNUNET_SYSERR;
+      }
+      field_off += rp->num_fields;
+    }
+
+    if (mysql_stmt_bind_result (stmt, result))
     {
-     //GNUNET_MY_cleanup_result(rs);
+      GNUNET_log_from (GNUNET_ERROR_TYPE_ERROR,
+                       "my",
+                       _("%s failed at %s:%d with error: %s\n"),
+                       "mysql_stmt_bind_result",
+                       __FILE__, __LINE__,
+                       mysql_stmt_error (stmt));
       return GNUNET_SYSERR;
     }
+#if TEST_OPTIMIZATION
+    (void) mysql_stmt_store_result (stmt);
+#endif
+    ret = mysql_stmt_fetch (stmt);
+    if (MYSQL_NO_DATA == ret)
+    {
+      mysql_stmt_free_result (stmt);
+      return GNUNET_NO;
+    }
+    if (1 == ret)
+    {
+      GNUNET_log_from (GNUNET_ERROR_TYPE_ERROR,
+                       "my",
+                       _("%s failed at %s:%d with error: %s\n"),
+                       "mysql_stmt_fetch",
+                       __FILE__, __LINE__,
+                       mysql_stmt_error (stmt));
+      GNUNET_MY_cleanup_result (rs);
+      mysql_stmt_free_result (stmt);
+      return GNUNET_SYSERR;
+    }
+    field_off = 0;
+    for (i=0;NULL != rs[i].post_conv;i++)
+    {
+      struct GNUNET_MY_ResultSpec *rp = &rs[i];
 
-    if (NULL != spec->result_size)
-      *spec->result_size = spec->dst_size;
+      if (NULL != rp->post_conv)
+        if (GNUNET_OK !=
+            rp->post_conv (rp->conv_cls,
+                           rp,
+                           stmt,
+                           field_off,
+                           &result[field_off]))
+        {
+          GNUNET_log (GNUNET_ERROR_TYPE_INFO,
+                      "Post-conversion for MySQL result failed at offset %u\n",
+                      i);
+          mysql_stmt_free_result (stmt);
+          GNUNET_MY_cleanup_result (rs);
+          return GNUNET_SYSERR;
+        }
+      field_off += rp->num_fields;
+    }
   }
+  return GNUNET_OK;
+}
 
-  if (GNUNET_YES == had_null)
-    return GNUNET_NO;
 
-  return GNUNET_OK;
+/**
+ * Free all memory that was allocated in @a rs during
+ * #GNUNET_MY_extract_result().
+ *
+ * @param rs result specification to clean up
+ */
+void
+GNUNET_MY_cleanup_result (struct GNUNET_MY_ResultSpec *rs)
+{
+  unsigned int i;
+
+  for (i=0;NULL != rs[i].post_conv;i++)
+    if (NULL != rs[i].cleaner)
+      rs[i].cleaner (rs[i].conv_cls,
+                     &rs[i]);
 }
 
+
 /* end of my.c */